1st "working" ib version.
[vlendec/samba-autobuild/.git] / ctdb / tests / ibwrapper_test.c
1 /*
2  * Unix SMB/CIFS implementation.
3  * Test the infiniband wrapper.
4  *
5  * Copyright (C) Sven Oehme <oehmes@de.ibm.com> 2006
6  *
7  * Major code contributions by Peter Somogyi <psomogyi@gamax.hu>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22  */
23
24 #include <stdlib.h>
25 #include <string.h>
26 #include <stdio.h>
27 #include <errno.h>
28 #include <sys/types.h>
29 #include <netinet/in.h>
30 #include <sys/socket.h>
31 #include <netdb.h>
32 #include <arpa/inet.h>
33 #include <malloc.h>
34 #include <assert.h>
35 #include <unistd.h>
36 #include <signal.h>
37
38 #include "includes.h"
39 #include "lib/events/events.h"
40 #include "ib/ibwrapper.h"
41
42 struct ibwtest_ctx {
43         int     is_server;
44         char    *id; /* my id */
45
46         struct ibw_initattr *attrs;
47         int     nattrs;
48         char    *opts; /* option string */
49
50         struct sockaddr_in *addrs; /* dynamic array of dest addrs */
51         int     naddrs;
52
53         unsigned int    nsec; /* nanosleep between messages */
54
55         int     cnt;
56
57         int     kill_me;
58         struct ibw_ctx  *ibwctx;
59 };
60
61 struct ibwtest_conn {
62         char    *id;
63 };
64
65 enum testopcode {
66         TESTOP_SEND_ID = 1,
67         TESTOP_SEND_DATA = 2
68 };
69
70 int ibwtest_connect_everybody(struct ibwtest_ctx *tcx)
71 {
72         struct ibwtest_conn     *pconn = talloc_zero(tcx, struct ibwtest_conn);
73         int     i;
74
75         for(i=0; i<tcx->naddrs; i++) {
76                 if (ibw_connect(tcx->ibwctx, &tcx->addrs[i], pconn)) {
77                         fprintf(stderr, "ibw_connect error at %d\n", i);
78                         return -1;
79                 }
80         }
81         DEBUG(10, ("sent %d connect request...\n", tcx->naddrs));
82
83         return 0;
84 }
85
86 int ibwtest_send_id(struct ibw_conn *conn)
87 {
88         char *buf;
89         void *key;
90         struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);
91         uint32_t        len;
92
93         DEBUG(10, ("test IBWC_CONNECTED\n"));
94         len = sizeof(uint32_t)+strlen(tcx->id)+2;
95         if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
96                 DEBUG(0, ("send_id: ibw_alloc_send_buf failed\n"));
97                 return -1;
98         }
99
100         /* first sizeof(uint32_t) size bytes are for length */
101         buf[sizeof(uint32_t)] = (char)TESTOP_SEND_ID;
102         strcpy(buf+sizeof(uint32_t)+1, tcx->id);
103
104         if (ibw_send(conn, buf, key, len)) {
105                 DEBUG(0, ("send_id: ibw_send error\n"));
106                 return -1;
107         }
108         return 0;
109 }
110
111 int ibwtest_send_test_msg(struct ibwtest_ctx *tcx, struct ibw_conn *conn, const char *msg)
112 {
113         char *buf, *p;
114         void *key;
115         uint32_t len;
116
117         if (conn->state!=IBWC_CONNECTED)
118                 return 0; /* not yet up */
119
120         len = strlen(msg)+2 + sizeof(uint32_t);
121         if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
122                 fprintf(stderr, "send_test_msg: ibw_alloc_send_buf failed\n");
123                 return -1;
124         }
125
126         p = buf;
127         p += sizeof(uint32_t);
128         p[0] = (char)TESTOP_SEND_DATA;
129         p++;
130         strcpy(p, msg);
131
132         if (ibw_send(conn, buf, key, len)) {
133                 DEBUG(0, ("send_test_msg: ibw_send error\n"));
134                 return -1;
135         }
136         return 0;
137 }
138
139 int ibwtest_connstate_handler(struct ibw_ctx *ctx, struct ibw_conn *conn)
140 {
141         struct ibwtest_ctx      *tcx = NULL; /* userdata */
142         struct ibwtest_conn     *pconn = NULL; /* userdata */
143
144         if (ctx) {
145                 tcx = talloc_get_type(ctx->ctx_userdata, struct ibwtest_ctx);
146
147                 switch(ctx->state) {
148                 case IBWS_INIT:
149                         DEBUG(10, ("test IBWS_INIT\n"));
150                         break;
151                 case IBWS_READY:
152                         DEBUG(10, ("test IBWS_READY\n"));
153                         break;
154                 case IBWS_CONNECT_REQUEST:
155                         DEBUG(10, ("test IBWS_CONNECT_REQUEST\n"));
156                         pconn = talloc_zero(conn, struct ibwtest_conn);
157                         if (ibw_accept(ctx, conn, pconn)) {
158                                 DEBUG(0, ("error accepting the connect request\n"));
159                         }
160                         break;
161                 case IBWS_STOPPED:
162                         DEBUG(10, ("test IBWS_STOPPED\n"));
163                         talloc_free(tcx->ibwctx);
164                         DEBUG(10, ("talloc_free(tcx->ibwctx) DONE\n"));
165                         tcx->kill_me = 1; /* main loop can exit */
166                         break;
167                 case IBWS_ERROR:
168                         DEBUG(10, ("test IBWS_ERROR\n"));
169                         ibw_stop(tcx->ibwctx);
170                         break;
171                 default:
172                         assert(0);
173                         break;
174                 }
175         }
176
177         if (conn) {
178                 pconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);
179                 switch(conn->state) {
180                 case IBWC_INIT:
181                         DEBUG(10, ("test IBWC_INIT\n"));
182                         break;
183                 case IBWC_CONNECTED:
184                         ibwtest_send_id(conn);
185                         break;
186                 case IBWC_DISCONNECTED:
187                         DEBUG(10, ("test IBWC_DISCONNECTED\n"));
188                         break;
189                 case IBWC_ERROR:
190                         DEBUG(10, ("test IBWC_ERROR\n"));
191                         break;
192                 default:
193                         assert(0);
194                         break;
195                 }
196         }
197         return 0;
198 }
199
200 int ibwtest_receive_handler(struct ibw_conn *conn, void *buf, int n)
201 {
202         struct ibwtest_conn *pconn;
203         enum testopcode op;
204         struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);
205
206         assert(conn!=NULL);
207         assert(n>=sizeof(uint32_t)+2);
208         pconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);
209
210         op = (enum testopcode)((char *)buf)[sizeof(uint32_t)];
211         if (op==TESTOP_SEND_ID) {
212                 pconn->id = talloc_strdup(pconn, ((char *)buf)+sizeof(uint32_t)+1);
213         }
214         DEBUG(11, ("[%d]msg from %s: \"%s\"(%d)\n", op,
215                 pconn->id ? pconn->id : "NULL", ((char *)buf)+sizeof(uint32_t)+1, n));
216
217         if (tcx->is_server) {
218                 char *buf2;
219                 void *key2;
220                 /* bounce message */
221                 if (ibw_alloc_send_buf(conn, (void **)&buf2, &key2, n)) {
222                         fprintf(stderr, "ibw_alloc_send_buf error #2\n");
223                         return -1;
224                 }
225                 memcpy(buf2, buf, n);
226                 if (ibw_send(conn, buf2, key2, n)) {
227                         fprintf(stderr, "ibw_send error #2\n");
228                         return -2;
229                 }
230         }
231
232         return 0;
233 }
234
235 void ibwtest_timeout_handler(struct event_context *ev, struct timed_event *te, 
236         struct timeval t, void *private)
237 {
238         struct ibwtest_ctx *tcx = talloc_get_type(private, struct ibwtest_ctx);
239
240         if (!tcx->is_server) {
241                 struct ibw_conn *p;
242                 char    msg[50];
243
244                 /* fill it with something variable... */
245                 sprintf(msg, "hello world %d", tcx->cnt++);
246
247                 /* send something to everybody... */
248                 for(p=tcx->ibwctx->conn_list; p!=NULL; p=p->next) {
249                         ibwtest_send_test_msg(tcx, p, msg);
250                 }
251         } /* else allow main loop run */
252 }
253
254 static struct ibwtest_ctx *testctx = NULL;
255
256 void ibwtest_sigquit_handler(int sig)
257 {
258         DEBUG(0, ("got SIGQUIT\n"));
259         if (testctx)
260                 ibw_stop(testctx->ibwctx);
261 }
262
263 int ibwtest_parse_attrs(struct ibwtest_ctx *tcx, char *optext,
264         struct ibw_initattr **pattrs, int *nattrs, char op)
265 {
266         int     i = 0, n = 1;
267         int     porcess_next = 1;
268         char    *p, *q;
269         struct ibw_initattr *attrs = NULL;
270
271         *pattrs = NULL;
272         for(p = optext; *p!='\0'; p++) {
273                 if (*p==',')
274                         n++;
275         }
276
277         attrs = (struct ibw_initattr *)talloc_size(tcx,
278                 n * sizeof(struct ibw_initattr));
279         for(p = optext; *p!='\0'; p++) {
280                 if (porcess_next) {
281                         attrs[i].name = p;
282                         q = strchr(p, ':');
283                         if (q==NULL) {
284                                 fprintf(stderr, "-%c format error\n", op);
285                                 return -1;
286                         }
287                         *q = '\0';
288                         attrs[i].value = q + 1;
289
290                         porcess_next = 0;
291                         i++;
292                         p = q; /* ++ at end */
293                 }
294                 if (*p==',') {
295                         *p = '\0'; /* ++ at end */
296                         porcess_next = 1;
297                 }
298         }
299         *pattrs = attrs;
300         *nattrs = n;
301
302         return 0;
303 }
304
305 int ibwtest_getdests(struct ibwtest_ctx *tcx, char op)
306 {
307         int     i;
308         struct ibw_initattr     *attrs = NULL;
309         struct sockaddr_in      *p;
310         char    *tmp;
311
312         tmp = talloc_strdup(tcx, optarg);
313         /* hack to reuse the above ibw_initattr parser */
314         if (ibwtest_parse_attrs(tcx, tmp, &attrs, &tcx->naddrs, op))
315                 return -1;
316
317         tcx->addrs = talloc_size(tcx,
318                 tcx->naddrs * sizeof(struct sockaddr_in));
319         for(i=0; i<tcx->naddrs; i++) {
320                 p = tcx->addrs + i;
321                 p->sin_family = AF_INET;
322                 p->sin_addr.s_addr = inet_addr(attrs[i].name);
323                 p->sin_port = atoi(attrs[i].value);
324         }
325
326         return 0;
327 }
328
329 int ibwtest_init_server(struct ibwtest_ctx *tcx)
330 {
331         if (tcx->naddrs!=1) {
332                 fprintf(stderr, "incorrect number of addrs(%d!=1)\n", tcx->naddrs);
333                 return -1;
334         }
335
336         if (ibw_bind(tcx->ibwctx, &tcx->addrs[0])) {
337                 DEBUG(0, ("ERROR: ibw_bind failed\n"));
338                 return -1;
339         }
340         
341         if (ibw_listen(tcx->ibwctx, 1)) {
342                 DEBUG(0, ("ERROR: ibw_listen failed\n"));
343                 return -1;
344         }
345
346         /* continued at IBWS_READY */
347         return 0;
348 }
349
350 void ibwtest_usage(struct ibwtest_ctx *tcx, char *name)
351 {
352         printf("Usage:\n");
353         printf("\t%s -i <id> -o {name:value} -d {addr:port} -t nsec -s\n", name);
354         printf("\t-i <id> is a free text, acting as a server id, max 23 chars [mandatory]\n");
355         printf("\t-o name1:value1,name2:value2,... is a list of (name, value) pairs\n");
356         printf("\t-d addr1:port1,addr2:port2,... is a list of destination ip addresses\n");
357         printf("\t-t nsec delta time between sends in nanosec [default %d]\n", tcx->nsec);
358         printf("\t-s server mode (you have to give exactly one -d address:port in this case)\n");
359         printf("Press ctrl+C to stop the program.\n");
360 }
361
362 int main(int argc, char *argv[])
363 {
364         int     rc, op;
365         int     result = 1;
366         struct event_context *ev = NULL;
367         struct ibwtest_ctx *tcx = NULL;
368
369         tcx = talloc_zero(NULL, struct ibwtest_ctx);
370         memset(tcx, 0, sizeof(struct ibwtest_ctx));
371         tcx->nsec = 1000;
372
373         /* here is the only case we can't avoid using global... */
374         testctx = tcx;
375         signal(SIGQUIT, ibwtest_sigquit_handler);
376
377         while ((op=getopt(argc, argv, "i:o:d:m:st:")) != -1) {
378                 switch (op) {
379                 case 'i':
380                         tcx->id = talloc_strdup(tcx, optarg);
381                         break;
382                 case 'o':
383                         tcx->opts = talloc_strdup(tcx, optarg);
384                         if (ibwtest_parse_attrs(tcx, tcx->opts, &tcx->attrs,
385                                 &tcx->nattrs, op))
386                                 goto cleanup;
387                         break;
388                 case 'd':
389                         if (ibwtest_getdests(tcx, op))
390                                 goto cleanup;
391                         break;
392                 case 's':
393                         tcx->is_server = 1;
394                         break;
395                 case 't':
396                         tcx->nsec = (unsigned int)atoi(optarg);
397                         break;
398                 default:
399                         fprintf(stderr, "ERROR: unknown option -%c\n", (char)op);
400                         ibwtest_usage(tcx, argv[0]);
401                         goto cleanup;
402                 }
403         }
404         if (tcx->id==NULL) {
405                 ibwtest_usage(tcx, argv[0]);
406                 goto cleanup;
407         }
408
409         ev = event_context_init(NULL);
410         assert(ev);
411
412         tcx->ibwctx = ibw_init(tcx->attrs, tcx->nattrs,
413                 tcx,
414                 ibwtest_connstate_handler,
415                 ibwtest_receive_handler,
416                 ev
417         );
418         if (!tcx->ibwctx)
419                 goto cleanup;
420
421         if (tcx->is_server)
422                 rc = ibwtest_init_server(tcx);
423         else
424                 rc = ibwtest_connect_everybody(tcx);
425         if (rc)
426                 goto cleanup;
427
428         while(!tcx->kill_me) {
429                 event_add_timed(ev, tcx, timeval_current_ofs(0, tcx->nsec),
430                         ibwtest_timeout_handler, tcx);
431                 event_loop_once(ev);
432         }
433
434         result = 0; /* everything OK */
435
436 cleanup:
437         if (tcx)
438                 talloc_free(tcx);
439         if (ev)
440                 talloc_free(ev);
441         DEBUG(0, ("exited with code %d\n", result));
442         return result;
443 }