Raw impl. of ibwrapper test tool.

(basic functional test, compilable, untested)
Adjusted makefile and ibwrapper state checking.

(This used to be ctdb commit 82410f42fe5d428eda0ba96c11774e142f672d24)

3 files changed, 432 insertions, 3 deletions

diff --git a/ctdb/Makefile.in b/ctdb/Makefile.in
index 814c107e8f..6ee701a49b 100644
--- a/ctdb/Makefile.in
+++ b/ctdb/Makefile.in
@@ -31,7 +31,7 @@ CTDB_OBJ = $(CTDB_COMMON_OBJ) $(CTDB_TCP_OBJ) $(CTDB_IB_OBJ)
 
 OBJS = @TDBOBJ@ @TALLOCOBJ@ @LIBREPLACEOBJ@ $(EXTRA_OBJ) $(EVENTS_OBJ) $(CTDB_OBJ)
 
-BINS = bin/ctdb_test
+BINS = bin/ctdb_test bin/ibwrapper_test
 
 DIRS = lib bin
 
@@ -54,6 +54,10 @@ bin/ctdb_test: $(OBJS) ctdb_test.o
 	@echo Linking $@
 	@$(CC) $(CFLAGS) -o $@ ctdb_test.o $(OBJS) $(LIB_FLAGS)
 
+bin/ibwrapper_test: $(OBJS) tests/ibwrapper_test.o
+	@echo Linking $@
+	@$(CC) $(CFLAGS) -o $@ tests/ibwrapper_test.o $(OBJS) $(LIB_FLAGS)
+
 clean:
 	rm -f *.o */*.o */*/*.o
 	rm -f $(BINS)
diff --git a/ctdb/ib/ibwrapper.c b/ctdb/ib/ibwrapper.c
index 998a0167d7..b70b6caad6 100644
--- a/ctdb/ib/ibwrapper.c
+++ b/ctdb/ib/ibwrapper.c
@@ -743,8 +743,6 @@ int ibw_connect(struct ibw_ctx *ctx, struct sockaddr_in *serv_addr, void *conn_u
 		return rc;
 	}
 
-	assert(ctx->state==IBWS_READY);
-
 	rc = rdma_resolve_addr(pconn->cm_id, NULL, (struct sockaddr *) &serv_addr, 2000);
 	if (rc) {
 		sprintf(ibw_lasterr, "rdma_resolve_addr error %d\n", rc);
diff --git a/ctdb/tests/ibwrapper_test.c b/ctdb/tests/ibwrapper_test.c
new file mode 100644
index 0000000000..eb2a5469ff
--- /dev/null
+++ b/ctdb/tests/ibwrapper_test.c
@@ -0,0 +1,427 @@
+/*
+ * 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 "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;
+
+	int	max_msg_size;
+	unsigned int	nsec; /* nanosleep between messages */
+
+	int	cnt;
+
+	int	kill_me;
+	struct ibw_ctx	*ibwctx;
+};
+
+struct ibwtest_conn {
+	char	*id;
+};
+
+enum testopcode {
+	TESTOP_SEND_ID = 1,
+	TESTOP_SEND_DATA = 2
+};
+
+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)
+{
+	char *buf;
+	void *key;
+	struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);
+
+	DEBUG(10, ("test IBWC_CONNECTED\n"));
+	if (ibw_alloc_send_buf(conn, (void **)&buf, &key)) {
+		DEBUG(0, ("send_id: ibw_alloc_send_buf failed\n"));
+		return -1;
+	}
+	
+	buf[0] = (char)TESTOP_SEND_ID;
+	strcpy(buf+1, tcx->id);
+
+	if (ibw_send(conn, buf, key, strlen(buf+1))) {
+		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;
+	void *key;
+
+	if (ibw_alloc_send_buf(conn, (void **)&buf, &key)) {
+		fprintf(stderr, "send_test_msg: ibw_alloc_send_buf failed\n");
+		return -1;
+	}
+
+	buf[0] = (char)TESTOP_SEND_DATA;
+	assert(strlen(msg)<tcx->max_msg_size-1);
+	strcpy(buf+1, msg);
+	
+	if (ibw_send(conn, buf, key, strlen(buf+1))) {
+		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"));
+			talloc_free(tcx->ibwctx);
+			DEBUG(10, ("talloc_free(tcx->ibwctx) DONE\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:
+			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);
+	pconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);
+
+	op = (enum testopcode)((char *)buf)[0];
+	DEBUG(11, ("[%d]msg from %s: \"%s\"(%d)\n", op,
+		pconn->id ? pconn->id : NULL, ((char *)buf)+1, n));
+
+	if (tcx->is_server) {
+		char *buf2;
+		void *key2;
+		/* bounce message */
+		if (ibw_alloc_send_buf(conn, (void **)&buf2, &key2)) {
+			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;
+		}
+	}
+
+	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) {
+			ibwtest_send_test_msg(tcx, p, msg);
+		}
+	} /* else allow main loop run */
+}
+
+static struct ibwtest_ctx *testctx = NULL;
+
+void ibwtest_sigquit_handler(int sig)
+{
+	DEBUG(0, ("got SIGQUIT\n"));
+	if (testctx)
+		ibw_stop(testctx->ibwctx);
+}
+
+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++;
+		}
+		if (*p==',') {
+			*p = '\0';
+			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_addr.s_addr = inet_addr(attrs[i].name);
+		p->sin_port = atoi(attrs[i].value);
+		p->sin_family = AF_INET;
+	}
+
+	return 0;
+}
+
+int ibwtest_init_server(struct ibwtest_ctx *tcx)
+{
+	if (tcx->naddrs!=1) {
+		fprintf(stderr, "incorrecr 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;
+	}
+
+	/* 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} -m max_msg_size -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-m max_msg_size maximum message size [default %d]\n", tcx->max_msg_size);
+	printf("\t-t nsec delta time between sends in nanosec [default %d]\n", tcx->nsec);
+	printf("\t-s server mode (you have to give exactly one -d address:port in this case)\n");
+	printf("Press ctrl+C to stop the program.\n");
+}
+
+int main(int argc, char *argv[])
+{
+	int	rc, op;
+	int	result = 1;
+	struct event_context *ev = NULL;
+	struct ibwtest_ctx *tcx = NULL;
+
+	tcx = talloc_zero(NULL, struct ibwtest_ctx);
+	memset(tcx, 0, sizeof(struct ibwtest_ctx));
+	tcx->max_msg_size = 256;
+	tcx->nsec = 1000;
+
+	/* here is the only case we can't avoid using global... */
+	testctx = tcx;
+	signal(SIGQUIT, ibwtest_sigquit_handler);
+
+	while ((op=getopt(argc, argv, "i:o:d:m:s")) != -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 'm':
+			tcx->max_msg_size = atoi(optarg);
+			break;
+		case 's':
+			tcx->is_server = 1;
+			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;
+	}
+
+	ev = event_context_init(NULL);
+	assert(ev);
+
+	tcx->ibwctx = ibw_init(tcx->attrs, tcx->nattrs,
+		tcx,
+		ibwtest_connstate_handler,
+		ibwtest_receive_handler,
+		ev,
+		tcx->max_msg_size
+	);
+	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) {
+		event_add_timed(ev, tcx, timeval_current_ofs(0, tcx->nsec),
+			ibwtest_timeout_handler, tcx);
+		event_loop_once(ev);
+	}
+
+	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;
+}