s4:rpc_server/remote: pass through DCERPC_PFC_FLAG_CONC_MPX if it was used by the...
[Samba.git] / ctdb / ib / ibwrapper_test.c
blobbc88dad051c4244a7ca44b3869a5e104817f434e
1 /*
2 * Unix SMB/CIFS implementation.
3 * Test the infiniband wrapper.
5 * Copyright (C) Sven Oehme <oehmes@de.ibm.com> 2006
7 * Major code contributions by Peter Somogyi <psomogyi@gamax.hu>
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 3 of the License, or
12 * (at your option) any later version.
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.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, see <http://www.gnu.org/licenses/>.
23 #include "replace.h"
24 #include "system/network.h"
25 #include "system/time.h"
26 #include "system/wait.h"
28 #include <assert.h>
29 #include <talloc.h>
30 #include <tevent.h>
32 #include "lib/util/time.h"
33 #include "lib/util/debug.h"
35 #include "common/logging.h"
37 #include "ib/ibwrapper.h"
39 struct ibwtest_ctx {
40 int is_server;
41 char *id; /* my id */
43 struct ibw_initattr *attrs;
44 int nattrs;
45 char *opts; /* option string */
47 struct sockaddr_in *addrs; /* dynamic array of dest addrs */
48 int naddrs;
50 unsigned int nsec; /* delta times between messages in nanosec */
51 unsigned int sleep_usec; /* microsecs to sleep in the main loop to emulate overloading */
52 uint32_t maxsize; /* maximum variable message size */
54 int cnt;
55 int nsent;
57 int nmsg; /* number of messages to send (client) */
59 int kill_me;
60 int stopping;
61 int error;
62 struct ibw_ctx *ibwctx;
64 struct timeval start_time, end_time;
67 struct ibwtest_conn {
68 char *id;
71 enum testopcode {
72 TESTOP_SEND_ID = 1,
73 TESTOP_SEND_TEXT = 2,
74 TESTOP_SEND_RND = 3
77 static int ibwtest_connect_everybody(struct ibwtest_ctx *tcx)
79 struct ibw_conn *conn;
80 struct ibwtest_conn *tconn = talloc_zero(tcx, struct ibwtest_conn);
81 int i;
83 for(i=0; i<tcx->naddrs; i++) {
84 conn = ibw_conn_new(tcx->ibwctx, tconn);
85 if (ibw_connect(conn, &tcx->addrs[i], tconn)) {
86 fprintf(stderr, "ibw_connect error at %d\n", i);
87 return -1;
90 DEBUG(DEBUG_DEBUG, ("sent %d connect request...\n", tcx->naddrs));
92 return 0;
95 static int ibwtest_send_id(struct ibw_conn *conn)
97 struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);
98 char *buf;
99 void *key;
100 uint32_t len;
102 DEBUG(DEBUG_DEBUG, ("ibwtest_send_id\n"));
103 len = sizeof(uint32_t)+strlen(tcx->id)+2;
104 if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
105 DEBUG(DEBUG_ERR, ("send_id: ibw_alloc_send_buf failed\n"));
106 return -1;
109 /* first sizeof(uint32_t) size bytes are for length */
110 *((uint32_t *)buf) = len;
111 buf[sizeof(uint32_t)] = (char)TESTOP_SEND_ID;
112 strcpy(buf+sizeof(uint32_t)+1, tcx->id);
114 if (ibw_send(conn, buf, key, len)) {
115 DEBUG(DEBUG_ERR, ("send_id: ibw_send error\n"));
116 return -1;
118 tcx->nsent++;
120 return 0;
123 static int ibwtest_send_test_msg(struct ibwtest_ctx *tcx, struct ibw_conn *conn, const char *msg)
125 char *buf, *p;
126 void *key;
127 uint32_t len;
129 if (conn->state!=IBWC_CONNECTED)
130 return 0; /* not yet up */
132 len = strlen(msg) + 2 + sizeof(uint32_t);
133 if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
134 fprintf(stderr, "send_test_msg: ibw_alloc_send_buf failed\n");
135 return -1;
138 *((uint32_t *)buf) = len;
139 p = buf;
140 p += sizeof(uint32_t);
141 p[0] = (char)TESTOP_SEND_TEXT;
142 p++;
143 strcpy(p, msg);
145 if (ibw_send(conn, buf, key, len)) {
146 DEBUG(DEBUG_ERR, ("send_test_msg: ibw_send error\n"));
147 return -1;
149 tcx->nsent++;
151 return 0;
154 static unsigned char ibwtest_fill_random(unsigned char *buf, uint32_t size)
156 uint32_t i = size;
157 unsigned char sum = 0;
158 unsigned char value;
159 while(i) {
160 i--;
161 value = (unsigned char)(256.0 * (rand() / (RAND_MAX + 1.0)));
162 buf[i] = value;
163 sum += value;
165 return sum;
168 static unsigned char ibwtest_get_sum(unsigned char *buf, uint32_t size)
170 uint32_t i = size;
171 unsigned char sum = 0;
173 while(i) {
174 i--;
175 sum += buf[i];
177 return sum;
180 static int ibwtest_do_varsize_scenario_conn_size(struct ibwtest_ctx *tcx, struct ibw_conn *conn, uint32_t size)
182 unsigned char *buf;
183 void *key;
184 uint32_t len;
185 unsigned char sum;
187 len = sizeof(uint32_t) + 1 + size + 1;
188 if (ibw_alloc_send_buf(conn, (void **)&buf, &key, len)) {
189 DEBUG(DEBUG_ERR, ("varsize/ibw_alloc_send_buf failed\n"));
190 return -1;
192 *((uint32_t *)buf) = len;
193 buf[sizeof(uint32_t)] = TESTOP_SEND_RND;
194 sum = ibwtest_fill_random(buf + sizeof(uint32_t) + 1, size);
195 buf[sizeof(uint32_t) + 1 + size] = sum;
196 if (ibw_send(conn, buf, key, len)) {
197 DEBUG(DEBUG_ERR, ("varsize/ibw_send failed\n"));
198 return -1;
200 tcx->nsent++;
202 return 0;
205 static int ibwtest_do_varsize_scenario_conn(struct ibwtest_ctx *tcx, struct ibw_conn *conn)
207 uint32_t size;
208 int i;
210 for(i=0; i<tcx->nmsg; i++)
212 //size = (uint32_t)((float)(tcx->maxsize) * (rand() / (RAND_MAX + 1.0)));
213 size = (uint32_t)((float)(tcx->maxsize) * ((float)(i+1)/(float)tcx->nmsg));
214 if (ibwtest_do_varsize_scenario_conn_size(tcx, conn, size))
215 return -1;
217 return 0;
220 /*int ibwtest_do_varsize_scenario(ibwtest_ctx *tcx)
222 int rc;
223 struct ibw_conn *conn;
225 for(conn=tcx->ibwctx->conn_list; conn!=NULL; conn=conn->next) {
226 if (conn->state==IBWC_CONNECTED) {
227 rc = ibwtest_do_varsize_scenario_conn(tcx, conn);
228 if (rc)
229 tcx->error = rc;
234 static int ibwtest_connstate_handler(struct ibw_ctx *ctx, struct ibw_conn *conn)
236 struct ibwtest_ctx *tcx = NULL; /* userdata */
237 struct ibwtest_conn *tconn = NULL; /* userdata */
239 if (ctx) {
240 tcx = talloc_get_type(ctx->ctx_userdata, struct ibwtest_ctx);
242 switch(ctx->state) {
243 case IBWS_INIT:
244 DEBUG(DEBUG_DEBUG, ("test IBWS_INIT\n"));
245 break;
246 case IBWS_READY:
247 DEBUG(DEBUG_DEBUG, ("test IBWS_READY\n"));
248 break;
249 case IBWS_CONNECT_REQUEST:
250 DEBUG(DEBUG_DEBUG, ("test IBWS_CONNECT_REQUEST\n"));
251 tconn = talloc_zero(conn, struct ibwtest_conn);
252 if (ibw_accept(ctx, conn, tconn)) {
253 DEBUG(DEBUG_ERR, ("error accepting the connect request\n"));
255 break;
256 case IBWS_STOPPED:
257 DEBUG(DEBUG_DEBUG, ("test IBWS_STOPPED\n"));
258 tcx->kill_me = 1; /* main loop can exit */
259 break;
260 case IBWS_ERROR:
261 DEBUG(DEBUG_DEBUG, ("test IBWS_ERROR\n"));
262 ibw_stop(tcx->ibwctx);
263 break;
264 default:
265 assert(0);
266 break;
270 if (conn) {
271 tconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);
272 switch(conn->state) {
273 case IBWC_INIT:
274 DEBUG(DEBUG_DEBUG, ("test IBWC_INIT\n"));
275 break;
276 case IBWC_CONNECTED:
277 if (gettimeofday(&tcx->start_time, NULL)) {
278 DEBUG(DEBUG_ERR, ("gettimeofday error %d", errno));
279 return -1;
281 ibwtest_send_id(conn);
282 break;
283 case IBWC_DISCONNECTED:
284 DEBUG(DEBUG_DEBUG, ("test IBWC_DISCONNECTED\n"));
285 talloc_free(conn);
286 break;
287 case IBWC_ERROR:
288 DEBUG(DEBUG_DEBUG, ("test IBWC_ERROR %s\n", ibw_getLastError()));
289 break;
290 default:
291 assert(0);
292 break;
295 return 0;
298 static int ibwtest_receive_handler(struct ibw_conn *conn, void *buf, int n)
300 struct ibwtest_conn *tconn;
301 enum testopcode op;
302 struct ibwtest_ctx *tcx = talloc_get_type(conn->ctx->ctx_userdata, struct ibwtest_ctx);
303 int rc = 0;
305 assert(conn!=NULL);
306 assert(n>=sizeof(uint32_t)+1);
307 tconn = talloc_get_type(conn->conn_userdata, struct ibwtest_conn);
309 op = (enum testopcode)((char *)buf)[sizeof(uint32_t)];
310 if (op==TESTOP_SEND_ID) {
311 tconn->id = talloc_strdup(tconn, ((char *)buf)+sizeof(uint32_t)+1);
313 if (op==TESTOP_SEND_ID || op==TESTOP_SEND_TEXT) {
314 DEBUG(DEBUG_DEBUG, ("[%d]msg from %s: \"%s\"(%d)\n", op,
315 tconn->id ? tconn->id : "NULL", ((char *)buf)+sizeof(uint32_t)+1, n));
318 if (tcx->is_server) {
319 if (op==TESTOP_SEND_RND) {
320 unsigned char sum;
321 sum = ibwtest_get_sum((unsigned char *)buf + sizeof(uint32_t) + 1,
322 n - sizeof(uint32_t) - 2);
323 DEBUG(DEBUG_DEBUG, ("[%d]msg varsize %u/sum %u from %s\n",
325 (uint32_t)(n - sizeof(uint32_t) - 2),
326 (uint32_t)sum,
327 tconn->id ? tconn->id : "NULL"));
328 if (sum!=((unsigned char *)buf)[n-1]) {
329 DEBUG(DEBUG_ERR, ("ERROR: checksum mismatch %u!=%u\n",
330 (uint32_t)sum, (uint32_t)((unsigned char *)buf)[n-1]));
331 ibw_stop(tcx->ibwctx);
332 goto error;
334 } else if (op!=TESTOP_SEND_ID) {
335 char *buf2;
336 void *key2;
338 /* bounce message regardless what it is */
339 if (ibw_alloc_send_buf(conn, (void **)&buf2, &key2, n)) {
340 fprintf(stderr, "ibw_alloc_send_buf error #2\n");
341 goto error;
343 memcpy(buf2, buf, n);
344 if (ibw_send(conn, buf2, key2, n)) {
345 fprintf(stderr, "ibw_send error #2\n");
346 goto error;
348 tcx->nsent++;
350 } else { /* client: */
351 if (op==TESTOP_SEND_ID && tcx->maxsize) {
352 /* send them in one blow */
353 rc = ibwtest_do_varsize_scenario_conn(tcx, conn);
356 if (tcx->nmsg) {
357 char msg[26];
358 sprintf(msg, "hello world %d", tcx->nmsg--);
359 rc = ibwtest_send_test_msg(tcx, conn, msg);
360 if (tcx->nmsg==0) {
361 ibw_stop(tcx->ibwctx);
362 tcx->stopping = 1;
367 if (rc)
368 tcx->error = rc;
370 return rc;
371 error:
372 return -1;
375 static void ibwtest_timeout_handler(struct tevent_context *ev,
376 struct tevent_timer *te,
377 struct timeval t, void *private_data)
379 struct ibwtest_ctx *tcx = talloc_get_type(private_data, struct ibwtest_ctx);
380 int rc;
382 if (!tcx->is_server) {
383 struct ibw_conn *conn;
384 char msg[50];
386 /* fill it with something variable... */
387 sprintf(msg, "hello world %d", tcx->cnt++);
389 /* send something to everybody... */
390 for(conn=tcx->ibwctx->conn_list; conn!=NULL; conn=conn->next) {
391 if (conn->state==IBWC_CONNECTED) {
392 rc = ibwtest_send_test_msg(tcx, conn, msg);
393 if (rc)
394 tcx->error = rc;
397 } /* else allow main loop run */
400 static struct ibwtest_ctx *testctx = NULL;
402 static void ibwtest_sigint_handler(int sig)
404 DEBUG(DEBUG_ERR, ("got SIGINT\n"));
405 if (testctx) {
406 if (testctx->ibwctx->state==IBWS_READY ||
407 testctx->ibwctx->state==IBWS_CONNECT_REQUEST ||
408 testctx->ibwctx->state==IBWS_ERROR)
410 if (testctx->stopping) {
411 DEBUG(DEBUG_DEBUG, ("forcing exit...\n"));
412 testctx->kill_me = 1;
413 } else {
414 /* mostly expected case */
415 ibw_stop(testctx->ibwctx);
416 testctx->stopping = 1;
418 } else
419 testctx->kill_me = 1;
423 static int ibwtest_parse_attrs(struct ibwtest_ctx *tcx, char *optext,
424 struct ibw_initattr **pattrs, int *nattrs, char op)
426 int i = 0, n = 1;
427 int porcess_next = 1;
428 char *p, *q;
429 struct ibw_initattr *attrs = NULL;
431 *pattrs = NULL;
432 for(p = optext; *p!='\0'; p++) {
433 if (*p==',')
434 n++;
437 attrs = (struct ibw_initattr *)talloc_size(tcx,
438 n * sizeof(struct ibw_initattr));
439 for(p = optext; *p!='\0'; p++) {
440 if (porcess_next) {
441 attrs[i].name = p;
442 q = strchr(p, ':');
443 if (q==NULL) {
444 fprintf(stderr, "-%c format error\n", op);
445 return -1;
447 *q = '\0';
448 attrs[i].value = q + 1;
450 porcess_next = 0;
451 i++;
452 p = q; /* ++ at end */
454 if (*p==',') {
455 *p = '\0'; /* ++ at end */
456 porcess_next = 1;
459 *pattrs = attrs;
460 *nattrs = n;
462 return 0;
465 static int ibwtest_get_address(const char *address, struct in_addr *addr)
467 if (inet_pton(AF_INET, address, addr) <= 0) {
468 struct hostent *he = gethostbyname(address);
469 if (he == NULL || he->h_length > sizeof(*addr)) {
470 DEBUG(DEBUG_ERR, ("invalid nework address '%s'\n", address));
471 return -1;
473 memcpy(addr, he->h_addr, he->h_length);
475 return 0;
478 static int ibwtest_getdests(struct ibwtest_ctx *tcx, char op)
480 int i;
481 struct ibw_initattr *attrs = NULL;
482 struct sockaddr_in *p;
483 char *tmp;
485 tmp = talloc_strdup(tcx, optarg);
486 if (tmp == NULL) return -1;
487 /* hack to reuse the above ibw_initattr parser */
488 if (ibwtest_parse_attrs(tcx, tmp, &attrs, &tcx->naddrs, op))
489 return -1;
491 tcx->addrs = talloc_size(tcx,
492 tcx->naddrs * sizeof(struct sockaddr_in));
493 for(i=0; i<tcx->naddrs; i++) {
494 p = tcx->addrs + i;
495 p->sin_family = AF_INET;
496 if (ibwtest_get_address(attrs[i].name, &p->sin_addr))
497 return -1;
498 p->sin_port = htons(atoi(attrs[i].value));
501 return 0;
504 static int ibwtest_init_server(struct ibwtest_ctx *tcx)
506 if (tcx->naddrs!=1) {
507 fprintf(stderr, "incorrect number of addrs(%d!=1)\n", tcx->naddrs);
508 return -1;
511 if (ibw_bind(tcx->ibwctx, &tcx->addrs[0])) {
512 DEBUG(DEBUG_ERR, ("ERROR: ibw_bind failed\n"));
513 return -1;
516 if (ibw_listen(tcx->ibwctx, 1)) {
517 DEBUG(DEBUG_ERR, ("ERROR: ibw_listen failed\n"));
518 return -1;
521 /* continued at IBWS_READY */
522 return 0;
525 static void ibwtest_usage(struct ibwtest_ctx *tcx, char *name)
527 printf("Usage:\n");
528 printf("\t%s -i <id> -o {name:value} -d {addr:port} -t nsec -s\n", name);
529 printf("\t-i <id> is a free text, acting as a server id, max 23 chars [mandatory]\n");
530 printf("\t-o name1:value1,name2:value2,... is a list of (name, value) pairs\n");
531 printf("\t-a addr1:port1,addr2:port2,... is a list of destination ip addresses\n");
532 printf("\t-t nsec delta time between sends in nanosec [default %d]\n", tcx->nsec);
533 printf("\t\t send message periodically and endless when nsec is non-zero\n");
534 printf("\t-s server mode (you have to give exactly one -d address:port in this case)\n");
535 printf("\t-n number of messages to send [default %d]\n", tcx->nmsg);
536 printf("\t-l usec time to sleep in the main loop [default %d]\n", tcx->sleep_usec);
537 printf("\t-v max variable msg size in bytes [default %d], 0=don't send var. size\n", tcx->maxsize);
538 printf("\t-d LogLevel [default %d]\n", DEBUGLEVEL);
539 printf("Press ctrl+C to stop the program.\n");
542 int main(int argc, char *argv[])
544 int rc, op;
545 int result = 1;
546 struct tevent_context *ev = NULL;
547 struct ibwtest_ctx *tcx = NULL;
548 float usec;
550 tcx = talloc_zero(NULL, struct ibwtest_ctx);
551 memset(tcx, 0, sizeof(struct ibwtest_ctx));
552 tcx->nsec = 0;
553 tcx->nmsg = 1000;
554 DEBUGLEVEL = 0;
556 /* here is the only case we can't avoid using global... */
557 testctx = tcx;
558 signal(SIGINT, ibwtest_sigint_handler);
559 srand((unsigned)time(NULL));
561 while ((op=getopt(argc, argv, "i:o:d:m:st:n:l:v:a:")) != -1) {
562 switch (op) {
563 case 'i':
564 tcx->id = talloc_strdup(tcx, optarg);
565 break;
566 case 'o':
567 tcx->opts = talloc_strdup(tcx, optarg);
568 if (tcx->opts) goto cleanup;
569 if (ibwtest_parse_attrs(tcx, tcx->opts, &tcx->attrs,
570 &tcx->nattrs, op))
571 goto cleanup;
572 break;
573 case 'a':
574 if (ibwtest_getdests(tcx, op))
575 goto cleanup;
576 break;
577 case 's':
578 tcx->is_server = 1;
579 break;
580 case 't':
581 tcx->nsec = (unsigned int)atoi(optarg);
582 break;
583 case 'n':
584 tcx->nmsg = atoi(optarg);
585 break;
586 case 'l':
587 tcx->sleep_usec = (unsigned int)atoi(optarg);
588 break;
589 case 'v':
590 tcx->maxsize = (unsigned int)atoi(optarg);
591 break;
592 case 'd':
593 DEBUGLEVEL = atoi(optarg);
594 break;
595 default:
596 fprintf(stderr, "ERROR: unknown option -%c\n", (char)op);
597 ibwtest_usage(tcx, argv[0]);
598 goto cleanup;
601 if (tcx->id==NULL) {
602 ibwtest_usage(tcx, argv[0]);
603 goto cleanup;
606 ev = tevent_context_init(NULL);
607 assert(ev);
609 tcx->ibwctx = ibw_init(tcx->attrs, tcx->nattrs,
610 tcx,
611 ibwtest_connstate_handler,
612 ibwtest_receive_handler,
615 if (!tcx->ibwctx)
616 goto cleanup;
618 if (tcx->is_server)
619 rc = ibwtest_init_server(tcx);
620 else
621 rc = ibwtest_connect_everybody(tcx);
622 if (rc)
623 goto cleanup;
625 while(!tcx->kill_me && !tcx->error) {
626 if (tcx->nsec) {
627 tevent_add_timer(ev, tcx,
628 timeval_current_ofs(0, tcx->nsec),
629 ibwtest_timeout_handler, tcx);
632 tevent_loop_once(ev);
634 if (tcx->sleep_usec)
635 usleep(tcx->sleep_usec);
638 if (!tcx->is_server && tcx->nsent!=0 && !tcx->error) {
639 if (gettimeofday(&tcx->end_time, NULL)) {
640 DEBUG(DEBUG_ERR, ("gettimeofday error %d\n", errno));
641 goto cleanup;
643 usec = (tcx->end_time.tv_sec - tcx->start_time.tv_sec) * 1000000 +
644 (tcx->end_time.tv_usec - tcx->start_time.tv_usec);
645 printf("usec: %f, nmsg: %d, usec/nmsg: %f\n",
646 usec, tcx->nsent, usec/(float)tcx->nsent);
649 if (!tcx->error)
650 result = 0; /* everything OK */
652 cleanup:
653 if (tcx)
654 talloc_free(tcx);
655 if (ev)
656 talloc_free(ev);
657 DEBUG(DEBUG_ERR, ("exited with code %d\n", result));
658 return result;