1st "working" ib version.
[vlendec/samba-autobuild/.git] / ctdb / ib / ibwrapper.c
1 /*
2  * Unix SMB/CIFS implementation.
3  * Wrap Infiniband calls.
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
37 #include "includes.h"
38 #include "lib/events/events.h"
39 #include "ibwrapper.h"
40
41 #include <rdma/rdma_cma.h>
42
43 #include "ibwrapper_internal.h"
44 #include "lib/util/dlinklist.h"
45
46 #define IBW_LASTERR_BUFSIZE 512
47 static char ibw_lasterr[IBW_LASTERR_BUFSIZE];
48
49 static void ibw_event_handler_verbs(struct event_context *ev,
50         struct fd_event *fde, uint16_t flags, void *private_data);
51 static int ibw_fill_cq(struct ibw_conn *conn);
52 static inline int ibw_wc_recv(struct ibw_conn *conn, struct ibv_wc *wc);
53 static inline int ibw_wc_send(struct ibw_conn *conn, struct ibv_wc *wc);
54
55 static void *ibw_alloc_mr(struct ibw_ctx_priv *pctx, struct ibw_conn_priv *pconn,
56         uint32_t n, struct ibv_mr **ppmr)
57 {
58         void *buf;
59
60         DEBUG(10, ("ibw_alloc_mr(cmid=%p, n=%u)\n", pconn->cm_id, n));
61         buf = memalign(pctx->pagesize, n);
62         if (!buf) {
63                 sprintf(ibw_lasterr, "couldn't allocate memory\n");
64                 return NULL;
65         }
66
67         *ppmr = ibv_reg_mr(pconn->pd, buf, n, IBV_ACCESS_LOCAL_WRITE);
68         if (!*ppmr) {
69                 sprintf(ibw_lasterr, "couldn't allocate mr\n");
70                 free(buf);
71                 return NULL;
72         }
73
74         return buf;
75 }
76
77 static void ibw_free_mr(char **ppbuf, struct ibv_mr **ppmr)
78 {
79         DEBUG(10, ("ibw_free_mr(%u %u)\n", (uint32_t)*ppbuf, (uint32_t)*ppmr));
80         if (*ppmr!=NULL) {
81                 ibv_dereg_mr(*ppmr);
82                 *ppmr = NULL;
83         }
84         if (*ppbuf) {
85                 free(*ppbuf);
86                 *ppbuf = NULL;
87         }
88 }
89
90 static int ibw_init_memory(struct ibw_conn *conn)
91 {
92         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
93         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
94         struct ibw_opts *opts = &pctx->opts;
95         int     i;
96         struct ibw_wr   *p;
97
98         DEBUG(10, ("ibw_init_memory(cmid: %p)\n", pconn->cm_id));
99         pconn->buf_send = ibw_alloc_mr(pctx, pconn,
100                 opts->max_send_wr * opts->avg_send_size, &pconn->mr_send);
101         if (!pconn->buf_send) {
102                 sprintf(ibw_lasterr, "couldn't allocate work send buf\n");
103                 return -1;
104         }
105
106         pconn->buf_recv = ibw_alloc_mr(pctx, pconn,
107                 opts->max_recv_wr * opts->recv_bufsize, &pconn->mr_recv);
108         if (!pconn->buf_recv) {
109                 sprintf(ibw_lasterr, "couldn't allocate work recv buf\n");
110                 return -1;
111         }
112
113         pconn->wr_index = talloc_size(pconn, opts->max_send_wr * sizeof(struct ibw_wr *));
114         assert(pconn->wr_index!=NULL);
115
116         for(i=0; i<opts->max_send_wr; i++) {
117                 p = pconn->wr_index[i] = talloc_zero(pconn, struct ibw_wr);
118                 p->msg = pconn->buf_send + (i * opts->avg_send_size);
119                 p->wr_id = i;
120
121                 DLIST_ADD(pconn->wr_list_avail, p);
122         }
123
124         return 0;
125 }
126
127 static int ibw_ctx_priv_destruct(struct ibw_ctx_priv *pctx)
128 {
129         DEBUG(10, ("ibw_ctx_priv_destruct(%u)\n", (uint32_t)pctx));
130
131         /* destroy cm */
132         if (pctx->cm_channel) {
133                 rdma_destroy_event_channel(pctx->cm_channel);
134                 pctx->cm_channel = NULL;
135         }
136         if (pctx->cm_channel_event) {
137                 /* TODO: do we have to do this here? */
138                 talloc_free(pctx->cm_channel_event);
139                 pctx->cm_channel_event = NULL;
140         }
141         if (pctx->cm_id) {
142                 rdma_destroy_id(pctx->cm_id);
143                 pctx->cm_id = NULL;
144         }
145
146         return 0;
147 }
148
149 static int ibw_ctx_destruct(struct ibw_ctx *ctx)
150 {
151         DEBUG(10, ("ibw_ctx_destruct(%u)\n", (uint32_t)ctx));
152         return 0;
153 }
154
155 static int ibw_conn_priv_destruct(struct ibw_conn_priv *pconn)
156 {
157         DEBUG(10, ("ibw_conn_priv_destruct(%u, cmid: %p)\n",
158                 (uint32_t)pconn, pconn->cm_id));
159
160         /* free memory regions */
161         ibw_free_mr(&pconn->buf_send, &pconn->mr_send);
162         ibw_free_mr(&pconn->buf_recv, &pconn->mr_recv);
163
164         /* pconn->wr_index is freed by talloc */
165         /* pconn->wr_index[i] are freed by talloc */
166
167         /* destroy verbs */
168         if (pconn->cm_id->qp) {
169                 ibv_destroy_qp(pconn->cm_id->qp);
170                 pconn->cm_id->qp = NULL;
171         }
172         if (pconn->cq) {
173                 ibv_destroy_cq(pconn->cq);
174                 pconn->cq = NULL;
175         }
176         if (pconn->verbs_channel) {
177                 ibv_destroy_comp_channel(pconn->verbs_channel);
178                 pconn->verbs_channel = NULL;
179         }
180         if (pconn->verbs_channel_event) {
181                 /* TODO: do we have to do this here? */
182                 talloc_free(pconn->verbs_channel_event);
183                 pconn->verbs_channel_event = NULL;
184         }
185         if (pconn->pd) {
186                 ibv_dealloc_pd(pconn->pd);
187                 pconn->pd = NULL;
188         }
189         if (pconn->cm_id) {
190                 rdma_destroy_id(pconn->cm_id);
191                 pconn->cm_id = NULL;
192         }
193         return 0;
194 }
195
196 static int ibw_conn_destruct(struct ibw_conn *conn)
197 {
198         DEBUG(10, ("ibw_conn_destruct(%u)\n", (uint32_t)conn));
199         
200         /* important here: ctx is a talloc _parent_ */
201         DLIST_REMOVE(conn->ctx->conn_list, conn);
202         return 0;
203 }
204
205 static struct ibw_conn *ibw_conn_new(struct ibw_ctx *ctx)
206 {
207         struct ibw_conn *conn;
208         struct ibw_conn_priv *pconn;
209
210         conn = talloc_zero(ctx, struct ibw_conn);
211         assert(conn!=NULL);
212         talloc_set_destructor(conn, ibw_conn_destruct);
213
214         pconn = talloc_zero(ctx, struct ibw_conn_priv);
215         assert(pconn!=NULL);
216         talloc_set_destructor(pconn, ibw_conn_priv_destruct);
217
218         conn->ctx = ctx;
219         conn->internal = (void *)pconn;
220
221         DLIST_ADD(ctx->conn_list, conn);
222
223         return conn;
224 }
225
226 static int ibw_setup_cq_qp(struct ibw_conn *conn)
227 {
228         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
229         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
230         struct ibv_qp_init_attr init_attr;
231         int rc;
232
233         DEBUG(10, ("ibw_setup_cq_qp(cmid: %p)\n", pconn->cm_id));
234
235         /* init verbs */
236         pconn->verbs_channel = ibv_create_comp_channel(pconn->cm_id->verbs);
237         if (!pconn->verbs_channel) {
238                 sprintf(ibw_lasterr, "ibv_create_comp_channel failed %d\n", errno);
239                 return -1;
240         }
241         DEBUG(10, ("created channel %p\n", pconn->verbs_channel));
242
243         pconn->verbs_channel_event = event_add_fd(pctx->ectx, conn,
244                 pconn->verbs_channel->fd, EVENT_FD_READ, ibw_event_handler_verbs, conn);
245
246         pconn->pd = ibv_alloc_pd(pconn->cm_id->verbs);
247         if (!pconn->pd) {
248                 sprintf(ibw_lasterr, "ibv_alloc_pd failed %d\n", errno);
249                 return -1;
250         }
251         DEBUG(10, ("created pd %p\n", pconn->pd));
252
253         /* init mr */
254         if (ibw_init_memory(conn))
255                 return -1;
256
257         /* init cq */
258         pconn->cq = ibv_create_cq(pconn->cm_id->verbs,
259                 pctx->opts.max_recv_wr + pctx->opts.max_send_wr,
260                 conn, pconn->verbs_channel, 0);
261         if (pconn->cq==NULL) {
262                 sprintf(ibw_lasterr, "ibv_create_cq failed\n");
263                 return -1;
264         }
265
266         rc = ibv_req_notify_cq(pconn->cq, 0);
267         if (rc) {
268                 sprintf(ibw_lasterr, "ibv_req_notify_cq failed with %d\n", rc);
269                 return rc;
270         }
271
272         /* init qp */
273         memset(&init_attr, 0, sizeof(init_attr));
274         init_attr.cap.max_send_wr = pctx->opts.max_send_wr;
275         init_attr.cap.max_recv_wr = pctx->opts.max_recv_wr;
276         init_attr.cap.max_recv_sge = 1;
277         init_attr.cap.max_send_sge = 1;
278         init_attr.qp_type = IBV_QPT_RC;
279         init_attr.send_cq = pconn->cq;
280         init_attr.recv_cq = pconn->cq;
281
282         rc = rdma_create_qp(pconn->cm_id, pconn->pd, &init_attr);
283         if (rc) {
284                 sprintf(ibw_lasterr, "rdma_create_qp failed with %d\n", rc);
285                 return rc;
286         }
287         /* elase result is in pconn->cm_id->qp */
288
289         return ibw_fill_cq(conn);
290 }
291
292 static int ibw_refill_cq_recv(struct ibw_conn *conn)
293 {
294         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
295         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
296         int     rc;
297         struct ibv_sge list = {
298                 .addr   = (uintptr_t) NULL,
299                 .length = pctx->opts.recv_bufsize,
300                 .lkey   = pconn->mr_recv->lkey
301         };
302         struct ibv_recv_wr wr = {
303                 .wr_id      = 0,
304                 .sg_list    = &list,
305                 .num_sge    = 1,
306         };
307         struct ibv_recv_wr *bad_wr;
308
309         DEBUG(10, ("ibw_refill_cq_recv(cmid: %p)\n", pconn->cm_id));
310
311         list.addr = (uintptr_t) pconn->buf_recv + pctx->opts.recv_bufsize * pconn->recv_index;
312         wr.wr_id = pconn->recv_index;
313         pconn->recv_index = (pconn->recv_index + 1) % pctx->opts.max_recv_wr;
314
315         rc = ibv_post_recv(pconn->cm_id->qp, &wr, &bad_wr);
316         if (rc) {
317                 sprintf(ibw_lasterr, "ibv_post_recv failed with %d\n", rc);
318                 DEBUG(0, (ibw_lasterr));
319                 return -2;
320         }
321
322         return 0;
323 }
324
325 static int ibw_fill_cq(struct ibw_conn *conn)
326 {
327         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
328         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
329         int     i, rc;
330         struct ibv_sge list = {
331                 .addr   = (uintptr_t) NULL,
332                 .length = pctx->opts.recv_bufsize,
333                 .lkey   = pconn->mr_recv->lkey
334         };
335         struct ibv_recv_wr wr = {
336                 .wr_id      = 0,
337                 .sg_list    = &list,
338                 .num_sge    = 1,
339         };
340         struct ibv_recv_wr *bad_wr;
341
342         DEBUG(10, ("ibw_fill_cq(cmid: %p)\n", pconn->cm_id));
343
344         for(i = pctx->opts.max_recv_wr; i!=0; i--) {
345                 list.addr = (uintptr_t) pconn->buf_recv + pctx->opts.recv_bufsize * pconn->recv_index;
346                 wr.wr_id = pconn->recv_index;
347                 pconn->recv_index = (pconn->recv_index + 1) % pctx->opts.max_recv_wr;
348
349                 rc = ibv_post_recv(pconn->cm_id->qp, &wr, &bad_wr);
350                 if (rc) {
351                         sprintf(ibw_lasterr, "ibv_post_recv failed with %d\n", rc);
352                         DEBUG(0, (ibw_lasterr));
353                         return -2;
354                 }
355         }
356
357         return 0;
358 }
359
360 static int ibw_manage_connect(struct ibw_conn *conn, struct rdma_cm_id *cma_id)
361 {
362         struct rdma_conn_param conn_param;
363         int     rc;
364
365         DEBUG(10, ("ibw_manage_connect(cmid: %p)\n", cma_id));
366         rc = ibw_setup_cq_qp(conn);
367         if (rc)
368                 return -1;
369
370         /* cm connect */
371         memset(&conn_param, 0, sizeof conn_param);
372         conn_param.responder_resources = 1;
373         conn_param.initiator_depth = 1;
374         conn_param.retry_count = 10;
375
376         rc = rdma_connect(cma_id, &conn_param);
377         if (rc)
378                 sprintf(ibw_lasterr, "rdma_connect error %d\n", rc);
379
380         return rc;
381 }
382
383 static void ibw_event_handler_cm(struct event_context *ev,
384         struct fd_event *fde, uint16_t flags, void *private_data)
385 {
386         int     rc;
387         struct ibw_ctx  *ctx = talloc_get_type(private_data, struct ibw_ctx);
388         struct ibw_ctx_priv *pctx = talloc_get_type(ctx->internal, struct ibw_ctx_priv);
389         struct ibw_conn *conn = NULL;
390         struct ibw_conn_priv *pconn = NULL;
391         struct rdma_cm_id *cma_id = NULL;
392         struct rdma_cm_event *event = NULL;
393
394         assert(ctx!=NULL);
395
396         rc = rdma_get_cm_event(pctx->cm_channel, &event);
397         if (rc) {
398                 ctx->state = IBWS_ERROR;
399                 sprintf(ibw_lasterr, "rdma_get_cm_event error %d\n", rc);
400                 goto error;
401         }
402         cma_id = event->id;
403
404         DEBUG(10, ("cma_event type %d cma_id %p (%s)\n", event->event, cma_id,
405                   (cma_id == pctx->cm_id) ? "parent" : "child"));
406
407         switch (event->event) {
408         case RDMA_CM_EVENT_ADDR_RESOLVED:
409                 /* continuing from ibw_connect ... */
410                 rc = rdma_resolve_route(cma_id, 2000);
411                 if (rc) {
412                         sprintf(ibw_lasterr, "rdma_resolve_route error %d\n", rc);
413                         goto error;
414                 }
415                 /* continued at RDMA_CM_EVENT_ROUTE_RESOLVED */
416                 break;
417
418         case RDMA_CM_EVENT_ROUTE_RESOLVED:
419                 /* after RDMA_CM_EVENT_ADDR_RESOLVED: */
420                 assert(cma_id->context!=NULL);
421                 conn = talloc_get_type(cma_id->context, struct ibw_conn);
422
423                 rc = ibw_manage_connect(conn, cma_id);
424                 if (rc)
425                         goto error;
426
427                 break;
428
429         case RDMA_CM_EVENT_CONNECT_REQUEST:
430                 ctx->state = IBWS_CONNECT_REQUEST;
431                 conn = ibw_conn_new(ctx);
432                 pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
433                 pconn->cm_id = cma_id; /* !!! event will be freed but id not */
434                 cma_id->context = (void *)conn;
435                 DEBUG(10, ("pconn->cm_id %p\n", pconn->cm_id));
436
437                 if (ibw_setup_cq_qp(conn))
438                         goto error;
439
440                 conn->state = IBWC_INIT;
441                 pctx->connstate_func(ctx, conn);
442
443                 /* continued at ibw_accept when invoked by the func above */
444                 if (!pconn->is_accepted) {
445                         talloc_free(conn);
446                         DEBUG(10, ("pconn->cm_id %p wasn't accepted\n", pconn->cm_id));
447                 }
448
449                 /* TODO: clarify whether if it's needed by upper layer: */
450                 ctx->state = IBWS_READY;
451                 pctx->connstate_func(ctx, NULL);
452
453                 /* NOTE: more requests can arrive until RDMA_CM_EVENT_ESTABLISHED ! */
454                 break;
455
456         case RDMA_CM_EVENT_ESTABLISHED:
457                 /* expected after ibw_accept and ibw_connect[not directly] */
458                 DEBUG(0, ("ESTABLISHED (conn: %u)\n", (unsigned int)cma_id->context));
459                 conn = talloc_get_type(cma_id->context, struct ibw_conn);
460                 assert(conn!=NULL); /* important assumption */
461
462                 /* client conn is up */
463                 conn->state = IBWC_CONNECTED;
464
465                 /* both ctx and conn have changed */
466                 pctx->connstate_func(ctx, conn);
467                 break;
468
469         case RDMA_CM_EVENT_ADDR_ERROR:
470         case RDMA_CM_EVENT_ROUTE_ERROR:
471         case RDMA_CM_EVENT_CONNECT_ERROR:
472         case RDMA_CM_EVENT_UNREACHABLE:
473         case RDMA_CM_EVENT_REJECTED:
474                 sprintf(ibw_lasterr, "cma event %d, error %d\n", event->event, event->status);
475                 goto error;
476
477         case RDMA_CM_EVENT_DISCONNECTED:
478                 if (cma_id!=pctx->cm_id) {
479                         DEBUG(0, ("client DISCONNECT event\n"));
480                         conn = talloc_get_type(cma_id->context, struct ibw_conn);
481                         conn->state = IBWC_DISCONNECTED;
482                         pctx->connstate_func(NULL, conn);
483
484                         talloc_free(conn);
485
486                         /* if we are the last... */
487                         if (ctx->conn_list==NULL)
488                                 rdma_disconnect(pctx->cm_id);
489                 } else {
490                         DEBUG(0, ("server DISCONNECT event\n"));
491                         ctx->state = IBWS_STOPPED; /* ??? TODO: try it... */
492                         /* talloc_free(ctx) should be called within or after this func */
493                         pctx->connstate_func(ctx, NULL);
494                 }
495                 break;
496
497         case RDMA_CM_EVENT_DEVICE_REMOVAL:
498                 sprintf(ibw_lasterr, "cma detected device removal!\n");
499                 goto error;
500
501         default:
502                 sprintf(ibw_lasterr, "unknown event %d\n", event->event);
503                 goto error;
504         }
505
506         if ((rc=rdma_ack_cm_event(event))) {
507                 sprintf(ibw_lasterr, "rdma_ack_cm_event failed with %d\n", rc);
508                 goto error;
509         }
510
511         return;
512 error:
513         DEBUG(0, ("cm event handler: %s", ibw_lasterr));
514         if (cma_id!=pctx->cm_id) {
515                 conn = talloc_get_type(cma_id->context, struct ibw_conn);
516                 if (conn)
517                         conn->state = IBWC_ERROR;
518                 pctx->connstate_func(NULL, conn);
519         } else {
520                 ctx->state = IBWS_ERROR;
521                 pctx->connstate_func(ctx, NULL);
522         }
523 }
524
525 static void ibw_event_handler_verbs(struct event_context *ev,
526         struct fd_event *fde, uint16_t flags, void *private_data)
527 {
528         struct ibw_conn *conn = talloc_get_type(private_data, struct ibw_conn);
529         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
530         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
531
532         struct ibv_wc wc;
533         int rc;
534         struct ibv_cq *ev_cq;
535         void          *ev_ctx;
536
537         DEBUG(10, ("ibw_event_handler_verbs(%u)\n", (uint32_t)flags));
538
539         /* TODO: check whether if it's good to have more channels here... */
540         rc = ibv_get_cq_event(pconn->verbs_channel, &ev_cq, &ev_ctx);
541         if (rc) {
542                 sprintf(ibw_lasterr, "Failed to get cq_event with %d\n", rc);
543                 goto error;
544         }
545         if (ev_cq != pconn->cq) {
546                 sprintf(ibw_lasterr, "ev_cq(%u) != pconn->cq(%u)\n",
547                         (unsigned int)ev_cq, (unsigned int)pconn->cq);
548                 goto error;
549         }
550         rc = ibv_req_notify_cq(pconn->cq, 0);
551         if (rc) {
552                 sprintf(ibw_lasterr, "Couldn't request CQ notification (%d)\n", rc);
553                 goto error;
554         }
555
556         while((rc=ibv_poll_cq(pconn->cq, 1, &wc))==1) {
557                 if (wc.status) {
558                         sprintf(ibw_lasterr, "cq completion failed status %d\n",
559                                 wc.status);
560                         goto error;
561                 }
562
563                 switch(wc.opcode) {
564                 case IBV_WC_SEND:
565                         DEBUG(10, ("send completion\n"));
566                         if (ibw_wc_send(conn, &wc))
567                                 goto error;
568                         break;
569
570                 case IBV_WC_RDMA_WRITE:
571                         DEBUG(10, ("rdma write completion\n"));
572                         break;
573         
574                 case IBV_WC_RDMA_READ:
575                         DEBUG(10, ("rdma read completion\n"));
576                         break;
577
578                 case IBV_WC_RECV:
579                         DEBUG(10, ("recv completion\n"));
580                         if (ibw_wc_recv(conn, &wc))
581                                 goto error;
582                         break;
583
584                 default:
585                         sprintf(ibw_lasterr, "unknown completion %d\n", wc.opcode);
586                         goto error;
587                 }
588         }
589         if (rc!=0) {
590                 sprintf(ibw_lasterr, "ibv_poll_cq error %d\n", rc);
591                 goto error;
592         }
593
594         return;
595 error:
596         DEBUG(0, (ibw_lasterr));
597         conn->state = IBWC_ERROR;
598         pctx->connstate_func(NULL, conn);
599 }
600
601 static inline int ibw_wc_send(struct ibw_conn *conn, struct ibv_wc *wc)
602 {
603         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
604         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
605         struct ibw_wr   *p;
606         int     send_index;
607
608         DEBUG(10, ("ibw_wc_send(cmid: %p, wr_id: %u, bl: %u)\n",
609                 pconn->cm_id, (uint32_t)wc->wr_id, (uint32_t)wc->byte_len));
610
611         assert(pconn->cm_id->qp->qp_num==wc->qp_num);
612         assert(wc->wr_id >= pctx->opts.max_recv_wr);
613         send_index = wc->wr_id - pctx->opts.max_recv_wr;
614         pconn->wr_sent--;
615
616         if (send_index < pctx->opts.max_send_wr) {
617                 DEBUG(10, ("ibw_wc_send#1 %u\n", (int)wc->wr_id));
618                 p = pconn->wr_index[send_index];
619                 if (p->msg_large)
620                         ibw_free_mr(&p->msg_large, &p->mr_large);
621                 DLIST_REMOVE(pconn->wr_list_used, p);
622                 DLIST_ADD(pconn->wr_list_avail, p);
623         } else { /* "extra" request - not optimized */
624                 DEBUG(10, ("ibw_wc_send#2 %u\n", (int)wc->wr_id));
625                 for(p=pconn->extra_sent; p!=NULL; p=p->next)
626                         if (p->wr_id==(int)wc->wr_id)
627                                 break;
628                 if (p==NULL) {
629                         sprintf(ibw_lasterr, "failed to find wr_id %d\n", (int)wc->wr_id);
630                                 return -1;
631                 }
632                 ibw_free_mr(&p->msg_large, &p->mr_large);
633                 DLIST_REMOVE(pconn->extra_sent, p);
634                 DLIST_ADD(pconn->extra_avail, p);
635         }
636
637         if (pconn->queue) {
638                 DEBUG(10, ("ibw_wc_send#queue %u\n", (int)wc->wr_id));
639                 p = pconn->queue;
640                 DLIST_REMOVE(pconn->queue, p);
641
642                 assert(p->queued_msg!=NULL);
643                 ibw_send(conn, p->queued_msg, p, ntohl(*(uint32_t *)p->queued_msg));
644                 p->queued_msg = NULL;
645         }
646
647         return 0;
648 }
649
650 static inline int ibw_append_to_part(struct ibw_conn_priv *pconn,
651         struct ibw_part *part, char **pp, uint32_t add_len, int info)
652 {
653         DEBUG(10, ("ibw_append_to_part: cmid=%p, (bs=%u, len=%u, tr=%u), al=%u, i=%u\n",
654                 pconn->cm_id, part->bufsize, part->len, part->to_read, add_len, info));
655
656         /* allocate more if necessary - it's an "evergrowing" buffer... */
657         if (part->len + add_len > part->bufsize) {
658                 if (part->buf==NULL) {
659                         assert(part->len==0);
660                         part->buf = talloc_size(pconn, add_len);
661                         if (part->buf==NULL) {
662                                 sprintf(ibw_lasterr, "recv talloc_size error (%u) #%d\n",
663                                         add_len, info);
664                                 return -1;
665                         }
666                         part->bufsize = add_len;
667                 } else {
668                         part->buf = talloc_realloc_size(pconn,
669                                 part->buf, part->len + add_len);
670                         if (part->buf==NULL) {
671                                 sprintf(ibw_lasterr, "recv realloc error (%u + %u) #%d\n",
672                                         part->len, add_len, info);
673                                 return -1;
674                         }
675                 }
676                 part->bufsize = part->len + add_len;
677         }
678
679         /* consume pp */
680         memcpy(part->buf + part->len, *pp, add_len);
681         *pp += add_len;
682         part->len += add_len;
683         part->to_read -= add_len;
684
685         return 0;
686 }
687
688 static inline int ibw_wc_mem_threshold(struct ibw_conn_priv *pconn,
689         struct ibw_part *part, uint32_t threshold)
690 {
691         DEBUG(10, ("ibw_wc_mem_threshold: cmid=%p, (bs=%u, len=%u, tr=%u), thr=%u\n",
692                 pconn->cm_id, part->bufsize, part->len, part->to_read, threshold));
693
694         if (part->bufsize > threshold) {
695                 DEBUG(3, ("ibw_wc_mem_threshold: cmid=%p, %u > %u\n",
696                         pconn->cm_id, part->bufsize, threshold));
697                 talloc_free(part->buf);
698                 part->buf = talloc_size(pconn, threshold);
699                 if (part->buf==NULL) {
700                         sprintf(ibw_lasterr, "talloc_size failed\n");
701                         return -1;
702                 }
703                 part->bufsize = threshold;
704         }
705         return 0;
706 }
707
708 static inline int ibw_wc_recv(struct ibw_conn *conn, struct ibv_wc *wc)
709 {
710         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
711         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
712         struct ibw_part *part = &pconn->part;
713         char    *p;
714         uint32_t        remain = wc->byte_len;
715
716         DEBUG(10, ("ibw_wc_recv: cmid=%p, wr_id: %u, bl: %u\n",
717                 pconn->cm_id, (uint32_t)wc->wr_id, remain));
718
719         assert(pconn->cm_id->qp->qp_num==wc->qp_num);
720         assert((int)wc->wr_id < pctx->opts.max_recv_wr);
721         assert(wc->byte_len <= pctx->opts.recv_bufsize);
722
723         p = pconn->buf_recv + ((int)wc->wr_id * pctx->opts.recv_bufsize);
724
725         while(remain) {
726                 /* here always true: (part->len!=0 && part->to_read!=0) ||
727                         (part->len==0 && part->to_read==0) */
728                 if (part->len) { /* is there a partial msg to be continued? */
729                         int read_len = (part->to_read<=remain) ? part->to_read : remain;
730                         if (ibw_append_to_part(pconn, part, &p, read_len, 421))
731                                 goto error;
732                         remain -= read_len;
733
734                         if (part->len<=sizeof(uint32_t) && part->to_read==0) {
735                                 assert(part->len==sizeof(uint32_t));
736                                 /* set it again now... */
737                                 part->to_read = ntohl(*((uint32_t *)(part->buf)));
738                                 if (part->to_read<sizeof(uint32_t)) {
739                                         sprintf(ibw_lasterr, "got msglen=%u #2\n", part->to_read);
740                                         goto error;
741                                 }
742                                 part->to_read -= sizeof(uint32_t); /* it's already read */
743                         }
744
745                         if (part->to_read==0) {
746                                 pctx->receive_func(conn, part->buf, part->len);
747                                 part->len = 0; /* tells not having partial data (any more) */
748                                 if (ibw_wc_mem_threshold(pconn, part, pctx->opts.recv_threshold))
749                                         goto error;
750                         }
751                 } else {
752                         if (remain>=sizeof(uint32_t)) {
753                                 uint32_t msglen = ntohl(*(uint32_t *)p);
754                                 if (msglen<sizeof(uint32_t)) {
755                                         sprintf(ibw_lasterr, "got msglen=%u\n", msglen);
756                                         goto error;
757                                 }
758
759                                 /* mostly awaited case: */
760                                 if (msglen<=remain) {
761                                         pctx->receive_func(conn, p, msglen);
762                                         p += msglen;
763                                         remain -= msglen;
764                                 } else {
765                                         part->to_read = msglen;
766                                         /* part->len is already 0 */
767                                         if (ibw_append_to_part(pconn, part, &p, remain, 422))
768                                                 goto error;
769                                         remain = 0; /* to be continued ... */
770                                         /* part->to_read > 0 here */
771                                 }
772                         } else { /* edge case: */
773                                 part->to_read = sizeof(uint32_t);
774                                 /* part->len is already 0 */
775                                 if (ibw_append_to_part(pconn, part, &p, remain, 423))
776                                         goto error;
777                                 remain = 0;
778                                 /* part->to_read > 0 here */
779                         }
780                 }
781         } /* <remain> is always decreased at least by 1 */
782
783         if (ibw_refill_cq_recv(conn))
784                 goto error;
785
786         return 0;
787
788 error:
789         DEBUG(0, ("ibw_wc_recv error: %s", ibw_lasterr));
790         return -1;
791 }
792
793 static int ibw_process_init_attrs(struct ibw_initattr *attr, int nattr, struct ibw_opts *opts)
794 {
795         int     i;
796         const char *name, *value;
797
798         DEBUG(10, ("ibw_process_init_attrs: nattr: %d\n", nattr));
799
800         opts->max_send_wr = 256;
801         opts->max_recv_wr = 1024;
802         opts->avg_send_size = 1024;
803         opts->recv_bufsize = 256;
804         opts->recv_threshold = 1 * 1024 * 1024;
805
806         for(i=0; i<nattr; i++) {
807                 name = attr[i].name;
808                 value = attr[i].value;
809
810                 assert(name!=NULL && value!=NULL);
811                 if (strcmp(name, "max_send_wr")==0)
812                         opts->max_send_wr = atoi(value);
813                 else if (strcmp(name, "max_recv_wr")==0)
814                         opts->max_recv_wr = atoi(value);
815                 else if (strcmp(name, "avg_send_size")==0)
816                         opts->avg_send_size = atoi(value);
817                 else if (strcmp(name, "recv_bufsize")==0)
818                         opts->recv_bufsize = atoi(value);
819                 else if (strcmp(name, "recv_threshold")==0)
820                         opts->recv_threshold = atoi(value);
821                 else {
822                         sprintf(ibw_lasterr, "ibw_init: unknown name %s\n", name);
823                         return -1;
824                 }
825         }
826         return 0;
827 }
828
829 struct ibw_ctx *ibw_init(struct ibw_initattr *attr, int nattr,
830         void *ctx_userdata,
831         ibw_connstate_fn_t ibw_connstate,
832         ibw_receive_fn_t ibw_receive,
833         struct event_context *ectx)
834 {
835         struct ibw_ctx *ctx = talloc_zero(NULL, struct ibw_ctx);
836         struct ibw_ctx_priv *pctx;
837         int     rc;
838
839         DEBUG(10, ("ibw_init(ctx_userdata: %u, ectx: %u)\n",
840                 (uint32_t)ctx_userdata, (uint32_t)ectx));
841
842         /* initialize basic data structures */
843         memset(ibw_lasterr, 0, IBW_LASTERR_BUFSIZE);
844
845         assert(ctx!=NULL);
846         ibw_lasterr[0] = '\0';
847         talloc_set_destructor(ctx, ibw_ctx_destruct);
848         ctx->ctx_userdata = ctx_userdata;
849
850         pctx = talloc_zero(ctx, struct ibw_ctx_priv);
851         talloc_set_destructor(pctx, ibw_ctx_priv_destruct);
852         ctx->internal = (void *)pctx;
853         assert(pctx!=NULL);
854
855         pctx->connstate_func = ibw_connstate;
856         pctx->receive_func = ibw_receive;
857
858         pctx->ectx = ectx;
859
860         /* process attributes */
861         if (ibw_process_init_attrs(attr, nattr, &pctx->opts))
862                 goto cleanup;
863
864         /* init cm */
865         pctx->cm_channel = rdma_create_event_channel();
866         if (!pctx->cm_channel) {
867                 sprintf(ibw_lasterr, "rdma_create_event_channel error %d\n", errno);
868                 goto cleanup;
869         }
870
871         pctx->cm_channel_event = event_add_fd(pctx->ectx, pctx,
872                 pctx->cm_channel->fd, EVENT_FD_READ, ibw_event_handler_cm, ctx);
873
874         rc = rdma_create_id(pctx->cm_channel, &pctx->cm_id, ctx, RDMA_PS_TCP);
875         if (rc) {
876                 rc = errno;
877                 sprintf(ibw_lasterr, "rdma_create_id error %d\n", rc);
878                 goto cleanup;
879         }
880         DEBUG(10, ("created cm_id %p\n", pctx->cm_id));
881
882         pctx->pagesize = sysconf(_SC_PAGESIZE);
883
884         return ctx;
885         /* don't put code here */
886 cleanup:
887         DEBUG(0, (ibw_lasterr));
888
889         if (ctx)
890                 talloc_free(ctx);
891
892         return NULL;
893 }
894
895 int ibw_stop(struct ibw_ctx *ctx)
896 {
897         struct ibw_conn *p;
898
899         DEBUG(10, ("ibw_stop\n"));
900         for(p=ctx->conn_list; p!=NULL; p=p->next) {
901                 if (ctx->state==IBWC_ERROR || ctx->state==IBWC_CONNECTED) {
902                         if (ibw_disconnect(p))
903                                 return -1;
904                 }
905         }
906
907         return 0;
908 }
909
910 int ibw_bind(struct ibw_ctx *ctx, struct sockaddr_in *my_addr)
911 {
912         struct ibw_ctx_priv *pctx = (struct ibw_ctx_priv *)ctx->internal;
913         int     rc;
914
915         DEBUG(10, ("ibw_bind: addr=%s, port=%u\n",
916                 inet_ntoa(my_addr->sin_addr), my_addr->sin_port));
917         rc = rdma_bind_addr(pctx->cm_id, (struct sockaddr *) my_addr);
918         if (rc) {
919                 sprintf(ibw_lasterr, "rdma_bind_addr error %d\n", rc);
920                 DEBUG(0, (ibw_lasterr));
921                 return rc;
922         }
923         DEBUG(10, ("rdma_bind_addr successful\n"));
924
925         return 0;
926 }
927
928 int ibw_listen(struct ibw_ctx *ctx, int backlog)
929 {
930         struct ibw_ctx_priv *pctx = talloc_get_type(ctx->internal, struct ibw_ctx_priv);
931         int     rc;
932
933         DEBUG(10, ("ibw_listen\n"));
934         rc = rdma_listen(pctx->cm_id, backlog);
935         if (rc) {
936                 sprintf(ibw_lasterr, "rdma_listen failed: %d\n", rc);
937                 DEBUG(0, (ibw_lasterr));
938                 return rc;
939         }
940
941         return 0;
942 }
943
944 int ibw_accept(struct ibw_ctx *ctx, struct ibw_conn *conn, void *conn_userdata)
945 {
946         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
947         struct rdma_conn_param  conn_param;
948         int     rc;
949
950         DEBUG(10, ("ibw_accept: cmid=%p\n", pconn->cm_id));
951         conn->conn_userdata = conn_userdata;
952
953         memset(&conn_param, 0, sizeof(struct rdma_conn_param));
954         conn_param.responder_resources = 1;
955         conn_param.initiator_depth = 1;
956         rc = rdma_accept(pconn->cm_id, &conn_param);
957         if (rc) {
958                 sprintf(ibw_lasterr, "rdma_accept failed %d\n", rc);
959                 DEBUG(0, (ibw_lasterr));
960                 return -1;;
961         }
962
963         pconn->is_accepted = 1;
964
965         /* continued at RDMA_CM_EVENT_ESTABLISHED */
966
967         return 0;
968 }
969
970 int ibw_connect(struct ibw_ctx *ctx, struct sockaddr_in *serv_addr, void *conn_userdata)
971 {
972         struct ibw_ctx_priv *pctx = talloc_get_type(ctx->internal, struct ibw_ctx_priv);
973         struct ibw_conn *conn = NULL;
974         struct ibw_conn_priv *pconn = NULL;
975         int     rc;
976
977         conn = ibw_conn_new(ctx);
978         conn->conn_userdata = conn_userdata;
979         pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
980         DEBUG(10, ("ibw_connect: addr=%s, port=%u\n", inet_ntoa(serv_addr->sin_addr), serv_addr->sin_port));
981
982         /* init cm */
983         rc = rdma_create_id(pctx->cm_channel, &pconn->cm_id, conn, RDMA_PS_TCP);
984         if (rc) {
985                 rc = errno;
986                 sprintf(ibw_lasterr, "ibw_connect/rdma_create_id error %d\n", rc);
987                 return rc;
988         }
989         DEBUG(10, ("ibw_connect: rdma_create_id succeeded, cm_id=%p\n", pconn->cm_id));
990
991         rc = rdma_resolve_addr(pconn->cm_id, NULL, (struct sockaddr *) serv_addr, 2000);
992         if (rc) {
993                 sprintf(ibw_lasterr, "rdma_resolve_addr error %d\n", rc);
994                 DEBUG(0, (ibw_lasterr));
995                 return -1;
996         }
997
998         /* continued at RDMA_CM_EVENT_ADDR_RESOLVED */
999
1000         return 0;
1001 }
1002
1003 int ibw_disconnect(struct ibw_conn *conn)
1004 {
1005         int     rc;
1006         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
1007         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
1008
1009         DEBUG(10, ("ibw_disconnect: cmid=%p\n", pconn->cm_id));
1010
1011         rc = rdma_disconnect(pctx->cm_id);
1012         if (rc) {
1013                 sprintf(ibw_lasterr, "ibw_disconnect failed with %d", rc);
1014                 DEBUG(0, (ibw_lasterr));
1015                 return rc;
1016         }
1017
1018         /* continued at RDMA_CM_EVENT_DISCONNECTED */
1019
1020         return 0;
1021 }
1022
1023 int ibw_alloc_send_buf(struct ibw_conn *conn, void **buf, void **key, uint32_t len)
1024 {
1025         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
1026         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
1027         struct ibw_wr *p = pconn->wr_list_avail;
1028
1029         if (p!=NULL) {
1030                 DEBUG(10, ("ibw_alloc_send_buf#1: cmid=%p, len=%d\n", pconn->cm_id, len));
1031
1032                 DLIST_REMOVE(pconn->wr_list_avail, p);
1033                 DLIST_ADD(pconn->wr_list_used, p);
1034
1035                 if (len <= pctx->opts.avg_send_size) {
1036                         *buf = (void *)p->msg;
1037                 } else {
1038                         p->msg_large = ibw_alloc_mr(pctx, pconn, len, &p->mr_large);
1039                         if (!p->msg_large) {
1040                                 sprintf(ibw_lasterr, "ibw_alloc_mr#1 failed\n");
1041                                 goto error;
1042                         }
1043                         *buf = (void *)p->msg_large;
1044                 }
1045         } else {
1046                 DEBUG(10, ("ibw_alloc_send_buf#2: cmid=%p, len=%d\n", pconn->cm_id, len));
1047                 /* not optimized */
1048                 p = pconn->extra_avail;
1049                 if (!p) {
1050                         p = pconn->extra_avail = talloc_zero(pconn, struct ibw_wr);
1051                         if (p==NULL) {
1052                                 sprintf(ibw_lasterr, "talloc_zero failed (emax: %u)", pconn->extra_max);
1053                                 goto error;
1054                         }
1055                         p->wr_id = pctx->opts.max_send_wr + pconn->extra_max;
1056                         pconn->extra_max++;
1057                         switch(pconn->extra_max) {
1058                                 case 1: DEBUG(2, ("warning: queue performed\n")); break;
1059                                 case 10: DEBUG(0, ("warning: queue reached 10\n")); break;
1060                                 case 100: DEBUG(0, ("warning: queue reached 100\n")); break;
1061                                 case 1000: DEBUG(0, ("warning: queue reached 1000\n")); break;
1062                                 default: break;
1063                         }
1064                 }
1065                 DLIST_REMOVE(pconn->extra_avail, p);
1066
1067                 p->msg_large = ibw_alloc_mr(pctx, pconn, len, &p->mr_large);
1068                 if (!p->msg_large) {
1069                         sprintf(ibw_lasterr, "ibw_alloc_mr#2 failed");
1070                         goto error;
1071                 }
1072                 *buf = (void *)p->msg_large;
1073         }
1074
1075         *key = (void *)p;
1076
1077         return 0;
1078 error:
1079         DEBUG(0, ("ibw_alloc_send_buf error: %s\n", ibw_lasterr));
1080         return -1;
1081 }
1082
1083
1084 int ibw_send(struct ibw_conn *conn, void *buf, void *key, uint32_t len)
1085 {
1086         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
1087         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
1088         struct ibw_wr *p = talloc_get_type(key, struct ibw_wr);
1089         int     rc;
1090
1091         *((uint32_t *)buf) = htonl(len);
1092
1093         /* can we send it right now? */
1094         if (pconn->wr_sent<=pctx->opts.max_send_wr) {
1095                 struct ibv_sge list = {
1096                         .addr   = (uintptr_t) NULL,
1097                         .length = len,
1098                         .lkey   = 0
1099                 };
1100                 struct ibv_send_wr wr = {
1101                         .wr_id      = p->wr_id + pctx->opts.max_recv_wr,
1102                         .sg_list    = &list,
1103                         .num_sge    = 1,
1104                         .opcode     = IBV_WR_SEND,
1105                         .send_flags = IBV_SEND_SIGNALED,
1106                 };
1107                 struct ibv_send_wr *bad_wr;
1108
1109                 DEBUG(10, ("ibw_send#1(cmid: %p, wrid: %u, n: %d)\n",
1110                         pconn->cm_id, (uint32_t)wr.wr_id, len));
1111
1112                 list.addr = (uintptr_t)buf;
1113                 if (p->msg_large==NULL) {
1114                         list.lkey = pconn->mr_send->lkey;
1115                 } else {
1116                         assert(p->mr_large!=NULL);
1117                         list.lkey = p->mr_large->lkey;
1118                 }
1119
1120                 rc = ibv_post_send(pconn->cm_id->qp, &wr, &bad_wr);
1121                 if (rc) {
1122                         sprintf(ibw_lasterr, "ibv_post_send error %d (%d)\n",
1123                                 rc, pconn->wr_sent);
1124                         DEBUG(0, (ibw_lasterr));
1125                 } else {
1126                         /* good case */
1127                         if (p->wr_id>=pctx->opts.max_send_wr) {
1128                                 /* we don't have prepared index for this, so that
1129                                  * we will have to find this later on */
1130                                 DLIST_ADD(pconn->extra_sent, p);
1131                         }
1132                         pconn->wr_sent++;
1133                 }
1134                 return rc;
1135         } /* else put the request into our own queue: */
1136
1137         DEBUG(10, ("ibw_send#2(cmid: %p, len: %u)\n", pconn->cm_id, len));
1138
1139         /* to be sent by ibw_wc_send */
1140         DLIST_ADD_END(pconn->queue, p, struct ibw_wr *); /* TODO: optimize */
1141         p->queued_msg = buf;
1142
1143         return 0;
1144 }
1145
1146 int ibw_cancel_send_buf(struct ibw_conn *conn, void *buf, void *key)
1147 {
1148         struct ibw_ctx_priv *pctx = talloc_get_type(conn->ctx->internal, struct ibw_ctx_priv);
1149         struct ibw_conn_priv *pconn = talloc_get_type(conn->internal, struct ibw_conn_priv);
1150         struct ibw_wr *p = talloc_get_type(key, struct ibw_wr);
1151
1152         assert(p!=NULL);
1153         assert(buf!=NULL);
1154         assert(conn!=NULL);
1155
1156         if (p->msg_large)
1157                 ibw_free_mr(&p->msg_large, &p->mr_large);
1158
1159         /* parallel case */
1160         if (p->wr_id < pctx->opts.max_send_wr) {
1161                 DEBUG(10, ("ibw_cancel_send_buf#1 %u", (int)p->wr_id));
1162                 DLIST_REMOVE(pconn->wr_list_used, p);
1163                 DLIST_ADD(pconn->wr_list_avail, p);
1164         } else { /* "extra" packet */
1165                 DEBUG(10, ("ibw_cancel_send_buf#2 %u", (int)p->wr_id));
1166                 DLIST_ADD(pconn->extra_avail, p);
1167         }
1168
1169         return 0;
1170 }
1171
1172 const char *ibw_getLastError(void)
1173 {
1174         return ibw_lasterr;
1175 }