samba-mirror/ctdb/tests/ibwrapper_test.c

/*
 * Unix SMB/CIFS implementation.
 * Test the infiniband wrapper.
 *
 * Copyright (C) Sven Oehme <oehmes@de.ibm.com> 2006
 *
 * Major code contributions by Peter Somogyi <psomogyi@gamax.hu>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <malloc.h>
#include <assert.h>
#include <unistd.h>
#include <signal.h>
#include <sys/time.h>

#include "includes.h"
#include "lib/events/events.h"
#include "ib/ibwrapper.h"

struct ibwtest_ctx {
	int	is_server;
	char	*id; /* my id */

	struct ibw_initattr *attrs;
	int	nattrs;
	char	*opts; /* option string */

	struct sockaddr_in *addrs; /* dynamic array of dest addrs */
	int	naddrs;

	unsigned int	nsec; /* delta times between messages in nanosec */
	unsigned int	sleep_nsec; /* nanosleep in the main loop to emulate overloading */

	int	cnt;

	int	nmsg; /* number of messages to send (client) */

	int	kill_me;
	int	stopping;
	int	error;
	struct ibw_ctx	*ibwctx;

	struct timeval	start_time, end_time;
};

struct ibwtest_conn {
	char	*id;
};

enum testopcode {
	TESTOP_SEND_ID = 1,
	TESTOP_SEND_DATA = 2
};

int ibwtest_nanosleep(unsigned int sleep_nsec)
{
	int nanosleep(const struct timespec *req, struct timespec *rem);
	struct timespec req, rem;

	memset(&req, 0, sizeof(struct timespec));
	memset(&rem, 0, sizeof(struct timespec));
	req.tv_sec = 0;
	req.tv_nsec = sleep_nsec;

	while (nanosleep(&req, &rem) < 0) {
		if (errno != EINTR) {
			DEBUG(0, ("nanosleep ERROR: %d\n", errno));
			return -1;
		}
		memcpy(&req, &rem, sizeof(struct timespec));
	}
	return 0;
}

int ibwtest_connect_everybody(struct ibwtest_ctx *tcx)
{
	struct ibwtest_conn	*pconn = talloc_zero(tcx, struct ibwtest_conn);
	int	i;

	for(i=0; i<tcx->naddrs; i++) {
		if (ibw_connect(tcx->ibwctx, &tcx->addrs[i], pconn)) {
			fprintf(stderr, "ibw_connect error at %d\n", i);
			return -1;
		}
	}
	DEBUG(10, ("sent %d connect request...\n", tcx->naddrs));

	return 0;
}

int ibwtest_send_id(struct ibw_conn *conn)
{
	struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);
	char *buf;
	void *key;
	uint32_t	len;

	DEBUG(10, ("test IBWC_CONNECTED\n"));
	len = sizeof(uint32_t)+strlen(tcx->id)+2;
	if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
		DEBUG(0, ("send_id: ibw_alloc_send_buf failed\n"));
		return -1;
	}

	/* first sizeof(uint32_t) size bytes are for length */
	buf[sizeof(uint32_t)] = (char)TESTOP_SEND_ID;
	strcpy(buf+sizeof(uint32_t)+1, tcx->id);

	if (ibw_send(conn, buf, key, len)) {
		DEBUG(0, ("send_id: ibw_send error\n"));
		return -1;
	}
	return 0;
}

int ibwtest_send_test_msg(struct ibwtest_ctx *tcx, struct ibw_conn *conn, const char *msg)
{
	char *buf, *p;
	void *key;
	uint32_t len;

	if (conn->state!=IBWC_CONNECTED)
		return 0; /* not yet up */

	len = strlen(msg)+2 + sizeof(uint32_t);
	if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
		fprintf(stderr, "send_test_msg: ibw_alloc_send_buf failed\n");
		return -1;
	}

	p = buf;
	p += sizeof(uint32_t);
	p[0] = (char)TESTOP_SEND_DATA;
	p++;
	strcpy(p, msg);

	if (ibw_send(conn, buf, key, len)) {
		DEBUG(0, ("send_test_msg: ibw_send error\n"));
		return -1;
	}
	return 0;
}

int ibwtest_connstate_handler(struct ibw_ctx *ctx, struct ibw_conn *conn)
{
	struct ibwtest_ctx	*tcx = NULL; /* userdata */
	struct ibwtest_conn	*pconn = NULL; /* userdata */

	if (ctx) {
		tcx = talloc_get_type(ctx->ctx_userdata, struct ibwtest_ctx);

		switch(ctx->state) {
		case IBWS_INIT:
			DEBUG(10, ("test IBWS_INIT\n"));
			break;
		case IBWS_READY:
			DEBUG(10, ("test IBWS_READY\n"));
			break;
		case IBWS_CONNECT_REQUEST:
			DEBUG(10, ("test IBWS_CONNECT_REQUEST\n"));
			pconn = talloc_zero(conn, struct ibwtest_conn);
			if (ibw_accept(ctx, conn, pconn)) {
				DEBUG(0, ("error accepting the connect request\n"));
			}
			break;
		case IBWS_STOPPED:
			DEBUG(10, ("test IBWS_STOPPED\n"));
			tcx->kill_me = 1; /* main loop can exit */
			break;
		case IBWS_ERROR:
			DEBUG(10, ("test IBWS_ERROR\n"));
			ibw_stop(tcx->ibwctx);
			break;
		default:
			assert(0);
			break;
		}
	}

	if (conn) {
		pconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);
		switch(conn->state) {
		case IBWC_INIT:
			DEBUG(10, ("test IBWC_INIT\n"));
			break;
		case IBWC_CONNECTED:
			if (gettimeofday(&tcx->start_time, NULL)) {
				DEBUG(0, ("gettimeofday error %d", errno));
				return -1;
			}
			ibwtest_send_id(conn);
			break;
		case IBWC_DISCONNECTED:
			DEBUG(10, ("test IBWC_DISCONNECTED\n"));
			break;
		case IBWC_ERROR:
			DEBUG(10, ("test IBWC_ERROR\n"));
			break;
		default:
			assert(0);
			break;
		}
	}
	return 0;
}

int ibwtest_receive_handler(struct ibw_conn *conn, void *buf, int n)
{
	struct ibwtest_conn *pconn;
	enum testopcode op;
	struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);

	assert(conn!=NULL);
	assert(n>=sizeof(uint32_t)+2);
	pconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);

	op = (enum testopcode)((char *)buf)[sizeof(uint32_t)];
	if (op==TESTOP_SEND_ID) {
		pconn->id = talloc_strdup(pconn, ((char *)buf)+sizeof(uint32_t)+1);
	}
	DEBUG(11, ("[%d]msg from %s: \"%s\"(%d)\n", op,
		pconn->id ? pconn->id : "NULL", ((char *)buf)+sizeof(uint32_t)+1, n));

	if (tcx->is_server) {
		char *buf2;
		void *key2;
		/* bounce message */
		if (ibw_alloc_send_buf(conn, (void **)&buf2, &key2, n)) {
			fprintf(stderr, "ibw_alloc_send_buf error #2\n");
			return -1;
		}
		memcpy(buf2, buf, n);
		if (ibw_send(conn, buf2, key2, n)) {
			fprintf(stderr, "ibw_send error #2\n");
			return -2;
		}
	} else {
		if (tcx->nmsg) {
			char	msg[26];
			sprintf(msg, "hello world %d", tcx->nmsg--);
			tcx->error = ibwtest_send_test_msg(tcx, conn, msg);
			if (tcx->nmsg==0) {
				ibw_stop(tcx->ibwctx);
				tcx->stopping = 1;
			}
		}
	}

	return 0;
}

void ibwtest_timeout_handler(struct event_context *ev, struct timed_event *te, 
	struct timeval t, void *private)
{
	struct ibwtest_ctx *tcx = talloc_get_type(private, struct ibwtest_ctx);

	if (!tcx->is_server) {
		struct ibw_conn *p;
		char	msg[50];

		/* fill it with something variable... */
		sprintf(msg, "hello world %d", tcx->cnt++);

		/* send something to everybody... */
		for(p=tcx->ibwctx->conn_list; p!=NULL; p=p->next) {
			if (p->state==IBWC_CONNECTED) {
				tcx->error = ibwtest_send_test_msg(tcx, p, msg);
			}
		}
	} /* else allow main loop run */
}

static struct ibwtest_ctx *testctx = NULL;

void ibwtest_sigint_handler(int sig)
{
	DEBUG(0, ("got SIGINT\n"));
	if (testctx) {
		if (testctx->stopping) {
			DEBUG(10, ("forcing exit...\n"));
			testctx->kill_me = 1;
		} else {
			ibw_stop(testctx->ibwctx);
			testctx->stopping = 1;
		}
	}
}

int ibwtest_parse_attrs(struct ibwtest_ctx *tcx, char *optext,
	struct ibw_initattr **pattrs, int *nattrs, char op)
{
	int	i = 0, n = 1;
	int	porcess_next = 1;
	char	*p, *q;
	struct ibw_initattr *attrs = NULL;

	*pattrs = NULL;
	for(p = optext; *p!='\0'; p++) {
		if (*p==',')
			n++;
	}

	attrs = (struct ibw_initattr *)talloc_size(tcx,
		n * sizeof(struct ibw_initattr));
	for(p = optext; *p!='\0'; p++) {
		if (porcess_next) {
			attrs[i].name = p;
			q = strchr(p, ':');
			if (q==NULL) {
				fprintf(stderr, "-%c format error\n", op);
				return -1;
			}
			*q = '\0';
			attrs[i].value = q + 1;

			porcess_next = 0;
			i++;
			p = q; /* ++ at end */
		}
		if (*p==',') {
			*p = '\0'; /* ++ at end */
			porcess_next = 1;
		}
	}
	*pattrs = attrs;
	*nattrs = n;

	return 0;
}

int ibwtest_getdests(struct ibwtest_ctx *tcx, char op)
{
	int	i;
	struct ibw_initattr	*attrs = NULL;
	struct sockaddr_in	*p;
	char	*tmp;

	tmp = talloc_strdup(tcx, optarg);
	/* hack to reuse the above ibw_initattr parser */
	if (ibwtest_parse_attrs(tcx, tmp, &attrs, &tcx->naddrs, op))
		return -1;

	tcx->addrs = talloc_size(tcx,
		tcx->naddrs * sizeof(struct sockaddr_in));
	for(i=0; i<tcx->naddrs; i++) {
		p = tcx->addrs + i;
		p->sin_family = AF_INET;
		p->sin_addr.s_addr = inet_addr(attrs[i].name);
		p->sin_port = atoi(attrs[i].value);
	}

	return 0;
}

int ibwtest_init_server(struct ibwtest_ctx *tcx)
{
	if (tcx->naddrs!=1) {
		fprintf(stderr, "incorrect number of addrs(%d!=1)\n", tcx->naddrs);
		return -1;
	}

	if (ibw_bind(tcx->ibwctx, &tcx->addrs[0])) {
		DEBUG(0, ("ERROR: ibw_bind failed\n"));
		return -1;
	}
	
	if (ibw_listen(tcx->ibwctx, 1)) {
		DEBUG(0, ("ERROR: ibw_listen failed\n"));
		return -1;
	}

	/* continued at IBWS_READY */
	return 0;
}

void ibwtest_usage(struct ibwtest_ctx *tcx, char *name)
{
	printf("Usage:\n");
	printf("\t%s -i <id> -o {name:value} -d {addr:port} -t nsec -s\n", name);
	printf("\t-i <id> is a free text, acting as a server id, max 23 chars [mandatory]\n");
	printf("\t-o name1:value1,name2:value2,... is a list of (name, value) pairs\n");
	printf("\t-d addr1:port1,addr2:port2,... is a list of destination ip addresses\n");
	printf("\t-t nsec delta time between sends in nanosec [default %d]\n", tcx->nsec);
	printf("\t\t send message periodically and endless when nsec is non-zero\n");
	printf("\t-s server mode (you have to give exactly one -d address:port in this case)\n");
	printf("\t-n number of messages to send [default %d]\n", tcx->nmsg);
	printf("\t-l nsec time to sleep in the main loop [default %d]\n", tcx->sleep_nsec);
	printf("Press ctrl+C to stop the program.\n");
}

int main(int argc, char *argv[])
{
	int	rc, op;
	int	result = 1, nmsg;
	struct event_context *ev = NULL;
	struct ibwtest_ctx *tcx = NULL;
	float	usec;

	tcx = talloc_zero(NULL, struct ibwtest_ctx);
	memset(tcx, 0, sizeof(struct ibwtest_ctx));
	tcx->nsec = 0;
	tcx->nmsg = 1000;

	/* here is the only case we can't avoid using global... */
	testctx = tcx;
	signal(SIGINT, ibwtest_sigint_handler);

	while ((op=getopt(argc, argv, "i:o:d:m:st:n:l:")) != -1) {
		switch (op) {
		case 'i':
			tcx->id = talloc_strdup(tcx, optarg);
			break;
		case 'o':
			tcx->opts = talloc_strdup(tcx, optarg);
			if (ibwtest_parse_attrs(tcx, tcx->opts, &tcx->attrs,
				&tcx->nattrs, op))
				goto cleanup;
			break;
		case 'd':
			if (ibwtest_getdests(tcx, op))
				goto cleanup;
			break;
		case 's':
			tcx->is_server = 1;
			break;
		case 't':
			tcx->nsec = (unsigned int)atoi(optarg);
			break;
		case 'n':
			tcx->nmsg = atoi(optarg);
			break;
		case 'l':
			tcx->sleep_nsec = (unsigned int)atoi(optarg);
			break;
		default:
			fprintf(stderr, "ERROR: unknown option -%c\n", (char)op);
			ibwtest_usage(tcx, argv[0]);
			goto cleanup;
		}
	}
	if (tcx->id==NULL) {
		ibwtest_usage(tcx, argv[0]);
		goto cleanup;
	}
	nmsg = tcx->nmsg;

	ev = event_context_init(NULL);
	assert(ev);

	tcx->ibwctx = ibw_init(tcx->attrs, tcx->nattrs,
		tcx,
		ibwtest_connstate_handler,
		ibwtest_receive_handler,
		ev
	);
	if (!tcx->ibwctx)
		goto cleanup;

	if (tcx->is_server)
		rc = ibwtest_init_server(tcx);
	else
		rc = ibwtest_connect_everybody(tcx);
	if (rc)
		goto cleanup;

	while(!tcx->kill_me && !tcx->error) {
		if (tcx->nsec) {
			event_add_timed(ev, tcx, timeval_current_ofs(0, tcx->nsec),
				ibwtest_timeout_handler, tcx);
		}

		event_loop_once(ev);

		if (tcx->sleep_nsec) {
			if (ibwtest_nanosleep(tcx->sleep_nsec))
				goto cleanup;
		}
	}

	if (!tcx->is_server && nmsg!=0 && !tcx->error) {
		if (gettimeofday(&tcx->end_time, NULL)) {
			DEBUG(0, ("gettimeofday error %d\n", errno));
			goto cleanup;
		}
		usec = (tcx->end_time.tv_sec - tcx->start_time.tv_sec) * 1000000 +
				(tcx->end_time.tv_usec - tcx->start_time.tv_usec);
		printf("usec: %f, nmsg: %d, usec/nmsg: %f\n",
			usec, nmsg, usec/(float)nmsg);
	}

	if (!tcx->error)
		result = 0; /* everything OK */

cleanup:
	if (tcx)
		talloc_free(tcx);
	if (ev)
		talloc_free(ev);
	DEBUG(0, ("exited with code %d\n", result));
	return result;
}