Pull bugzilla-7880 into release branch
[linux-2.6] / drivers / infiniband / core / cma.c
1 /*
2  * Copyright (c) 2005 Voltaire Inc.  All rights reserved.
3  * Copyright (c) 2002-2005, Network Appliance, Inc. All rights reserved.
4  * Copyright (c) 1999-2005, Mellanox Technologies, Inc. All rights reserved.
5  * Copyright (c) 2005-2006 Intel Corporation.  All rights reserved.
6  *
7  * This Software is licensed under one of the following licenses:
8  *
9  * 1) under the terms of the "Common Public License 1.0" a copy of which is
10  *    available from the Open Source Initiative, see
11  *    http://www.opensource.org/licenses/cpl.php.
12  *
13  * 2) under the terms of the "The BSD License" a copy of which is
14  *    available from the Open Source Initiative, see
15  *    http://www.opensource.org/licenses/bsd-license.php.
16  *
17  * 3) under the terms of the "GNU General Public License (GPL) Version 2" a
18  *    copy of which is available from the Open Source Initiative, see
19  *    http://www.opensource.org/licenses/gpl-license.php.
20  *
21  * Licensee has the right to choose one of the above licenses.
22  *
23  * Redistributions of source code must retain the above copyright
24  * notice and one of the license notices.
25  *
26  * Redistributions in binary form must reproduce both the above copyright
27  * notice, one of the license notices in the documentation
28  * and/or other materials provided with the distribution.
29  *
30  */
31
32 #include <linux/completion.h>
33 #include <linux/in.h>
34 #include <linux/in6.h>
35 #include <linux/mutex.h>
36 #include <linux/random.h>
37 #include <linux/idr.h>
38 #include <linux/inetdevice.h>
39
40 #include <net/tcp.h>
41
42 #include <rdma/rdma_cm.h>
43 #include <rdma/rdma_cm_ib.h>
44 #include <rdma/ib_cache.h>
45 #include <rdma/ib_cm.h>
46 #include <rdma/ib_sa.h>
47 #include <rdma/iw_cm.h>
48
49 MODULE_AUTHOR("Sean Hefty");
50 MODULE_DESCRIPTION("Generic RDMA CM Agent");
51 MODULE_LICENSE("Dual BSD/GPL");
52
53 #define CMA_CM_RESPONSE_TIMEOUT 20
54 #define CMA_MAX_CM_RETRIES 15
55
56 static void cma_add_one(struct ib_device *device);
57 static void cma_remove_one(struct ib_device *device);
58
59 static struct ib_client cma_client = {
60         .name   = "cma",
61         .add    = cma_add_one,
62         .remove = cma_remove_one
63 };
64
65 static struct ib_sa_client sa_client;
66 static struct rdma_addr_client addr_client;
67 static LIST_HEAD(dev_list);
68 static LIST_HEAD(listen_any_list);
69 static DEFINE_MUTEX(lock);
70 static struct workqueue_struct *cma_wq;
71 static DEFINE_IDR(sdp_ps);
72 static DEFINE_IDR(tcp_ps);
73 static DEFINE_IDR(udp_ps);
74 static DEFINE_IDR(ipoib_ps);
75 static int next_port;
76
77 struct cma_device {
78         struct list_head        list;
79         struct ib_device        *device;
80         struct completion       comp;
81         atomic_t                refcount;
82         struct list_head        id_list;
83 };
84
85 enum cma_state {
86         CMA_IDLE,
87         CMA_ADDR_QUERY,
88         CMA_ADDR_RESOLVED,
89         CMA_ROUTE_QUERY,
90         CMA_ROUTE_RESOLVED,
91         CMA_CONNECT,
92         CMA_DISCONNECT,
93         CMA_ADDR_BOUND,
94         CMA_LISTEN,
95         CMA_DEVICE_REMOVAL,
96         CMA_DESTROYING
97 };
98
99 struct rdma_bind_list {
100         struct idr              *ps;
101         struct hlist_head       owners;
102         unsigned short          port;
103 };
104
105 /*
106  * Device removal can occur at anytime, so we need extra handling to
107  * serialize notifying the user of device removal with other callbacks.
108  * We do this by disabling removal notification while a callback is in process,
109  * and reporting it after the callback completes.
110  */
111 struct rdma_id_private {
112         struct rdma_cm_id       id;
113
114         struct rdma_bind_list   *bind_list;
115         struct hlist_node       node;
116         struct list_head        list;
117         struct list_head        listen_list;
118         struct cma_device       *cma_dev;
119         struct list_head        mc_list;
120
121         enum cma_state          state;
122         spinlock_t              lock;
123         struct completion       comp;
124         atomic_t                refcount;
125         wait_queue_head_t       wait_remove;
126         atomic_t                dev_remove;
127
128         int                     backlog;
129         int                     timeout_ms;
130         struct ib_sa_query      *query;
131         int                     query_id;
132         union {
133                 struct ib_cm_id *ib;
134                 struct iw_cm_id *iw;
135         } cm_id;
136
137         u32                     seq_num;
138         u32                     qkey;
139         u32                     qp_num;
140         u8                      srq;
141 };
142
143 struct cma_multicast {
144         struct rdma_id_private *id_priv;
145         union {
146                 struct ib_sa_multicast *ib;
147         } multicast;
148         struct list_head        list;
149         void                    *context;
150         struct sockaddr         addr;
151         u8                      pad[sizeof(struct sockaddr_in6) -
152                                     sizeof(struct sockaddr)];
153 };
154
155 struct cma_work {
156         struct work_struct      work;
157         struct rdma_id_private  *id;
158         enum cma_state          old_state;
159         enum cma_state          new_state;
160         struct rdma_cm_event    event;
161 };
162
163 union cma_ip_addr {
164         struct in6_addr ip6;
165         struct {
166                 __u32 pad[3];
167                 __u32 addr;
168         } ip4;
169 };
170
171 struct cma_hdr {
172         u8 cma_version;
173         u8 ip_version;  /* IP version: 7:4 */
174         __u16 port;
175         union cma_ip_addr src_addr;
176         union cma_ip_addr dst_addr;
177 };
178
179 struct sdp_hh {
180         u8 bsdh[16];
181         u8 sdp_version; /* Major version: 7:4 */
182         u8 ip_version;  /* IP version: 7:4 */
183         u8 sdp_specific1[10];
184         __u16 port;
185         __u16 sdp_specific2;
186         union cma_ip_addr src_addr;
187         union cma_ip_addr dst_addr;
188 };
189
190 struct sdp_hah {
191         u8 bsdh[16];
192         u8 sdp_version;
193 };
194
195 #define CMA_VERSION 0x00
196 #define SDP_MAJ_VERSION 0x2
197
198 static int cma_comp(struct rdma_id_private *id_priv, enum cma_state comp)
199 {
200         unsigned long flags;
201         int ret;
202
203         spin_lock_irqsave(&id_priv->lock, flags);
204         ret = (id_priv->state == comp);
205         spin_unlock_irqrestore(&id_priv->lock, flags);
206         return ret;
207 }
208
209 static int cma_comp_exch(struct rdma_id_private *id_priv,
210                          enum cma_state comp, enum cma_state exch)
211 {
212         unsigned long flags;
213         int ret;
214
215         spin_lock_irqsave(&id_priv->lock, flags);
216         if ((ret = (id_priv->state == comp)))
217                 id_priv->state = exch;
218         spin_unlock_irqrestore(&id_priv->lock, flags);
219         return ret;
220 }
221
222 static enum cma_state cma_exch(struct rdma_id_private *id_priv,
223                                enum cma_state exch)
224 {
225         unsigned long flags;
226         enum cma_state old;
227
228         spin_lock_irqsave(&id_priv->lock, flags);
229         old = id_priv->state;
230         id_priv->state = exch;
231         spin_unlock_irqrestore(&id_priv->lock, flags);
232         return old;
233 }
234
235 static inline u8 cma_get_ip_ver(struct cma_hdr *hdr)
236 {
237         return hdr->ip_version >> 4;
238 }
239
240 static inline void cma_set_ip_ver(struct cma_hdr *hdr, u8 ip_ver)
241 {
242         hdr->ip_version = (ip_ver << 4) | (hdr->ip_version & 0xF);
243 }
244
245 static inline u8 sdp_get_majv(u8 sdp_version)
246 {
247         return sdp_version >> 4;
248 }
249
250 static inline u8 sdp_get_ip_ver(struct sdp_hh *hh)
251 {
252         return hh->ip_version >> 4;
253 }
254
255 static inline void sdp_set_ip_ver(struct sdp_hh *hh, u8 ip_ver)
256 {
257         hh->ip_version = (ip_ver << 4) | (hh->ip_version & 0xF);
258 }
259
260 static inline int cma_is_ud_ps(enum rdma_port_space ps)
261 {
262         return (ps == RDMA_PS_UDP || ps == RDMA_PS_IPOIB);
263 }
264
265 static void cma_attach_to_dev(struct rdma_id_private *id_priv,
266                               struct cma_device *cma_dev)
267 {
268         atomic_inc(&cma_dev->refcount);
269         id_priv->cma_dev = cma_dev;
270         id_priv->id.device = cma_dev->device;
271         list_add_tail(&id_priv->list, &cma_dev->id_list);
272 }
273
274 static inline void cma_deref_dev(struct cma_device *cma_dev)
275 {
276         if (atomic_dec_and_test(&cma_dev->refcount))
277                 complete(&cma_dev->comp);
278 }
279
280 static void cma_detach_from_dev(struct rdma_id_private *id_priv)
281 {
282         list_del(&id_priv->list);
283         cma_deref_dev(id_priv->cma_dev);
284         id_priv->cma_dev = NULL;
285 }
286
287 static int cma_set_qkey(struct ib_device *device, u8 port_num,
288                         enum rdma_port_space ps,
289                         struct rdma_dev_addr *dev_addr, u32 *qkey)
290 {
291         struct ib_sa_mcmember_rec rec;
292         int ret = 0;
293
294         switch (ps) {
295         case RDMA_PS_UDP:
296                 *qkey = RDMA_UDP_QKEY;
297                 break;
298         case RDMA_PS_IPOIB:
299                 ib_addr_get_mgid(dev_addr, &rec.mgid);
300                 ret = ib_sa_get_mcmember_rec(device, port_num, &rec.mgid, &rec);
301                 *qkey = be32_to_cpu(rec.qkey);
302                 break;
303         default:
304                 break;
305         }
306         return ret;
307 }
308
309 static int cma_acquire_dev(struct rdma_id_private *id_priv)
310 {
311         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
312         struct cma_device *cma_dev;
313         union ib_gid gid;
314         int ret = -ENODEV;
315
316         switch (rdma_node_get_transport(dev_addr->dev_type)) {
317         case RDMA_TRANSPORT_IB:
318                 ib_addr_get_sgid(dev_addr, &gid);
319                 break;
320         case RDMA_TRANSPORT_IWARP:
321                 iw_addr_get_sgid(dev_addr, &gid);
322                 break;
323         default:
324                 return -ENODEV;
325         }
326
327         list_for_each_entry(cma_dev, &dev_list, list) {
328                 ret = ib_find_cached_gid(cma_dev->device, &gid,
329                                          &id_priv->id.port_num, NULL);
330                 if (!ret) {
331                         ret = cma_set_qkey(cma_dev->device,
332                                            id_priv->id.port_num,
333                                            id_priv->id.ps, dev_addr,
334                                            &id_priv->qkey);
335                         if (!ret)
336                                 cma_attach_to_dev(id_priv, cma_dev);
337                         break;
338                 }
339         }
340         return ret;
341 }
342
343 static void cma_deref_id(struct rdma_id_private *id_priv)
344 {
345         if (atomic_dec_and_test(&id_priv->refcount))
346                 complete(&id_priv->comp);
347 }
348
349 static int cma_disable_remove(struct rdma_id_private *id_priv,
350                               enum cma_state state)
351 {
352         unsigned long flags;
353         int ret;
354
355         spin_lock_irqsave(&id_priv->lock, flags);
356         if (id_priv->state == state) {
357                 atomic_inc(&id_priv->dev_remove);
358                 ret = 0;
359         } else
360                 ret = -EINVAL;
361         spin_unlock_irqrestore(&id_priv->lock, flags);
362         return ret;
363 }
364
365 static void cma_enable_remove(struct rdma_id_private *id_priv)
366 {
367         if (atomic_dec_and_test(&id_priv->dev_remove))
368                 wake_up(&id_priv->wait_remove);
369 }
370
371 static int cma_has_cm_dev(struct rdma_id_private *id_priv)
372 {
373         return (id_priv->id.device && id_priv->cm_id.ib);
374 }
375
376 struct rdma_cm_id *rdma_create_id(rdma_cm_event_handler event_handler,
377                                   void *context, enum rdma_port_space ps)
378 {
379         struct rdma_id_private *id_priv;
380
381         id_priv = kzalloc(sizeof *id_priv, GFP_KERNEL);
382         if (!id_priv)
383                 return ERR_PTR(-ENOMEM);
384
385         id_priv->state = CMA_IDLE;
386         id_priv->id.context = context;
387         id_priv->id.event_handler = event_handler;
388         id_priv->id.ps = ps;
389         spin_lock_init(&id_priv->lock);
390         init_completion(&id_priv->comp);
391         atomic_set(&id_priv->refcount, 1);
392         init_waitqueue_head(&id_priv->wait_remove);
393         atomic_set(&id_priv->dev_remove, 0);
394         INIT_LIST_HEAD(&id_priv->listen_list);
395         INIT_LIST_HEAD(&id_priv->mc_list);
396         get_random_bytes(&id_priv->seq_num, sizeof id_priv->seq_num);
397
398         return &id_priv->id;
399 }
400 EXPORT_SYMBOL(rdma_create_id);
401
402 static int cma_init_ud_qp(struct rdma_id_private *id_priv, struct ib_qp *qp)
403 {
404         struct ib_qp_attr qp_attr;
405         int qp_attr_mask, ret;
406
407         qp_attr.qp_state = IB_QPS_INIT;
408         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
409         if (ret)
410                 return ret;
411
412         ret = ib_modify_qp(qp, &qp_attr, qp_attr_mask);
413         if (ret)
414                 return ret;
415
416         qp_attr.qp_state = IB_QPS_RTR;
417         ret = ib_modify_qp(qp, &qp_attr, IB_QP_STATE);
418         if (ret)
419                 return ret;
420
421         qp_attr.qp_state = IB_QPS_RTS;
422         qp_attr.sq_psn = 0;
423         ret = ib_modify_qp(qp, &qp_attr, IB_QP_STATE | IB_QP_SQ_PSN);
424
425         return ret;
426 }
427
428 static int cma_init_conn_qp(struct rdma_id_private *id_priv, struct ib_qp *qp)
429 {
430         struct ib_qp_attr qp_attr;
431         int qp_attr_mask, ret;
432
433         qp_attr.qp_state = IB_QPS_INIT;
434         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
435         if (ret)
436                 return ret;
437
438         return ib_modify_qp(qp, &qp_attr, qp_attr_mask);
439 }
440
441 int rdma_create_qp(struct rdma_cm_id *id, struct ib_pd *pd,
442                    struct ib_qp_init_attr *qp_init_attr)
443 {
444         struct rdma_id_private *id_priv;
445         struct ib_qp *qp;
446         int ret;
447
448         id_priv = container_of(id, struct rdma_id_private, id);
449         if (id->device != pd->device)
450                 return -EINVAL;
451
452         qp = ib_create_qp(pd, qp_init_attr);
453         if (IS_ERR(qp))
454                 return PTR_ERR(qp);
455
456         if (cma_is_ud_ps(id_priv->id.ps))
457                 ret = cma_init_ud_qp(id_priv, qp);
458         else
459                 ret = cma_init_conn_qp(id_priv, qp);
460         if (ret)
461                 goto err;
462
463         id->qp = qp;
464         id_priv->qp_num = qp->qp_num;
465         id_priv->srq = (qp->srq != NULL);
466         return 0;
467 err:
468         ib_destroy_qp(qp);
469         return ret;
470 }
471 EXPORT_SYMBOL(rdma_create_qp);
472
473 void rdma_destroy_qp(struct rdma_cm_id *id)
474 {
475         ib_destroy_qp(id->qp);
476 }
477 EXPORT_SYMBOL(rdma_destroy_qp);
478
479 static int cma_modify_qp_rtr(struct rdma_cm_id *id)
480 {
481         struct ib_qp_attr qp_attr;
482         int qp_attr_mask, ret;
483
484         if (!id->qp)
485                 return 0;
486
487         /* Need to update QP attributes from default values. */
488         qp_attr.qp_state = IB_QPS_INIT;
489         ret = rdma_init_qp_attr(id, &qp_attr, &qp_attr_mask);
490         if (ret)
491                 return ret;
492
493         ret = ib_modify_qp(id->qp, &qp_attr, qp_attr_mask);
494         if (ret)
495                 return ret;
496
497         qp_attr.qp_state = IB_QPS_RTR;
498         ret = rdma_init_qp_attr(id, &qp_attr, &qp_attr_mask);
499         if (ret)
500                 return ret;
501
502         return ib_modify_qp(id->qp, &qp_attr, qp_attr_mask);
503 }
504
505 static int cma_modify_qp_rts(struct rdma_cm_id *id)
506 {
507         struct ib_qp_attr qp_attr;
508         int qp_attr_mask, ret;
509
510         if (!id->qp)
511                 return 0;
512
513         qp_attr.qp_state = IB_QPS_RTS;
514         ret = rdma_init_qp_attr(id, &qp_attr, &qp_attr_mask);
515         if (ret)
516                 return ret;
517
518         return ib_modify_qp(id->qp, &qp_attr, qp_attr_mask);
519 }
520
521 static int cma_modify_qp_err(struct rdma_cm_id *id)
522 {
523         struct ib_qp_attr qp_attr;
524
525         if (!id->qp)
526                 return 0;
527
528         qp_attr.qp_state = IB_QPS_ERR;
529         return ib_modify_qp(id->qp, &qp_attr, IB_QP_STATE);
530 }
531
532 static int cma_ib_init_qp_attr(struct rdma_id_private *id_priv,
533                                struct ib_qp_attr *qp_attr, int *qp_attr_mask)
534 {
535         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
536         int ret;
537
538         ret = ib_find_cached_pkey(id_priv->id.device, id_priv->id.port_num,
539                                   ib_addr_get_pkey(dev_addr),
540                                   &qp_attr->pkey_index);
541         if (ret)
542                 return ret;
543
544         qp_attr->port_num = id_priv->id.port_num;
545         *qp_attr_mask = IB_QP_STATE | IB_QP_PKEY_INDEX | IB_QP_PORT;
546
547         if (cma_is_ud_ps(id_priv->id.ps)) {
548                 qp_attr->qkey = id_priv->qkey;
549                 *qp_attr_mask |= IB_QP_QKEY;
550         } else {
551                 qp_attr->qp_access_flags = 0;
552                 *qp_attr_mask |= IB_QP_ACCESS_FLAGS;
553         }
554         return 0;
555 }
556
557 int rdma_init_qp_attr(struct rdma_cm_id *id, struct ib_qp_attr *qp_attr,
558                        int *qp_attr_mask)
559 {
560         struct rdma_id_private *id_priv;
561         int ret = 0;
562
563         id_priv = container_of(id, struct rdma_id_private, id);
564         switch (rdma_node_get_transport(id_priv->id.device->node_type)) {
565         case RDMA_TRANSPORT_IB:
566                 if (!id_priv->cm_id.ib || cma_is_ud_ps(id_priv->id.ps))
567                         ret = cma_ib_init_qp_attr(id_priv, qp_attr, qp_attr_mask);
568                 else
569                         ret = ib_cm_init_qp_attr(id_priv->cm_id.ib, qp_attr,
570                                                  qp_attr_mask);
571                 if (qp_attr->qp_state == IB_QPS_RTR)
572                         qp_attr->rq_psn = id_priv->seq_num;
573                 break;
574         case RDMA_TRANSPORT_IWARP:
575                 if (!id_priv->cm_id.iw) {
576                         qp_attr->qp_access_flags = 0;
577                         *qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS;
578                 } else
579                         ret = iw_cm_init_qp_attr(id_priv->cm_id.iw, qp_attr,
580                                                  qp_attr_mask);
581                 break;
582         default:
583                 ret = -ENOSYS;
584                 break;
585         }
586
587         return ret;
588 }
589 EXPORT_SYMBOL(rdma_init_qp_attr);
590
591 static inline int cma_zero_addr(struct sockaddr *addr)
592 {
593         struct in6_addr *ip6;
594
595         if (addr->sa_family == AF_INET)
596                 return ZERONET(((struct sockaddr_in *) addr)->sin_addr.s_addr);
597         else {
598                 ip6 = &((struct sockaddr_in6 *) addr)->sin6_addr;
599                 return (ip6->s6_addr32[0] | ip6->s6_addr32[1] |
600                         ip6->s6_addr32[2] | ip6->s6_addr32[3]) == 0;
601         }
602 }
603
604 static inline int cma_loopback_addr(struct sockaddr *addr)
605 {
606         return LOOPBACK(((struct sockaddr_in *) addr)->sin_addr.s_addr);
607 }
608
609 static inline int cma_any_addr(struct sockaddr *addr)
610 {
611         return cma_zero_addr(addr) || cma_loopback_addr(addr);
612 }
613
614 static inline __be16 cma_port(struct sockaddr *addr)
615 {
616         if (addr->sa_family == AF_INET)
617                 return ((struct sockaddr_in *) addr)->sin_port;
618         else
619                 return ((struct sockaddr_in6 *) addr)->sin6_port;
620 }
621
622 static inline int cma_any_port(struct sockaddr *addr)
623 {
624         return !cma_port(addr);
625 }
626
627 static int cma_get_net_info(void *hdr, enum rdma_port_space ps,
628                             u8 *ip_ver, __u16 *port,
629                             union cma_ip_addr **src, union cma_ip_addr **dst)
630 {
631         switch (ps) {
632         case RDMA_PS_SDP:
633                 if (sdp_get_majv(((struct sdp_hh *) hdr)->sdp_version) !=
634                     SDP_MAJ_VERSION)
635                         return -EINVAL;
636
637                 *ip_ver = sdp_get_ip_ver(hdr);
638                 *port   = ((struct sdp_hh *) hdr)->port;
639                 *src    = &((struct sdp_hh *) hdr)->src_addr;
640                 *dst    = &((struct sdp_hh *) hdr)->dst_addr;
641                 break;
642         default:
643                 if (((struct cma_hdr *) hdr)->cma_version != CMA_VERSION)
644                         return -EINVAL;
645
646                 *ip_ver = cma_get_ip_ver(hdr);
647                 *port   = ((struct cma_hdr *) hdr)->port;
648                 *src    = &((struct cma_hdr *) hdr)->src_addr;
649                 *dst    = &((struct cma_hdr *) hdr)->dst_addr;
650                 break;
651         }
652
653         if (*ip_ver != 4 && *ip_ver != 6)
654                 return -EINVAL;
655         return 0;
656 }
657
658 static void cma_save_net_info(struct rdma_addr *addr,
659                               struct rdma_addr *listen_addr,
660                               u8 ip_ver, __u16 port,
661                               union cma_ip_addr *src, union cma_ip_addr *dst)
662 {
663         struct sockaddr_in *listen4, *ip4;
664         struct sockaddr_in6 *listen6, *ip6;
665
666         switch (ip_ver) {
667         case 4:
668                 listen4 = (struct sockaddr_in *) &listen_addr->src_addr;
669                 ip4 = (struct sockaddr_in *) &addr->src_addr;
670                 ip4->sin_family = listen4->sin_family;
671                 ip4->sin_addr.s_addr = dst->ip4.addr;
672                 ip4->sin_port = listen4->sin_port;
673
674                 ip4 = (struct sockaddr_in *) &addr->dst_addr;
675                 ip4->sin_family = listen4->sin_family;
676                 ip4->sin_addr.s_addr = src->ip4.addr;
677                 ip4->sin_port = port;
678                 break;
679         case 6:
680                 listen6 = (struct sockaddr_in6 *) &listen_addr->src_addr;
681                 ip6 = (struct sockaddr_in6 *) &addr->src_addr;
682                 ip6->sin6_family = listen6->sin6_family;
683                 ip6->sin6_addr = dst->ip6;
684                 ip6->sin6_port = listen6->sin6_port;
685
686                 ip6 = (struct sockaddr_in6 *) &addr->dst_addr;
687                 ip6->sin6_family = listen6->sin6_family;
688                 ip6->sin6_addr = src->ip6;
689                 ip6->sin6_port = port;
690                 break;
691         default:
692                 break;
693         }
694 }
695
696 static inline int cma_user_data_offset(enum rdma_port_space ps)
697 {
698         switch (ps) {
699         case RDMA_PS_SDP:
700                 return 0;
701         default:
702                 return sizeof(struct cma_hdr);
703         }
704 }
705
706 static void cma_cancel_route(struct rdma_id_private *id_priv)
707 {
708         switch (rdma_node_get_transport(id_priv->id.device->node_type)) {
709         case RDMA_TRANSPORT_IB:
710                 if (id_priv->query)
711                         ib_sa_cancel_query(id_priv->query_id, id_priv->query);
712                 break;
713         default:
714                 break;
715         }
716 }
717
718 static inline int cma_internal_listen(struct rdma_id_private *id_priv)
719 {
720         return (id_priv->state == CMA_LISTEN) && id_priv->cma_dev &&
721                cma_any_addr(&id_priv->id.route.addr.src_addr);
722 }
723
724 static void cma_destroy_listen(struct rdma_id_private *id_priv)
725 {
726         cma_exch(id_priv, CMA_DESTROYING);
727
728         if (id_priv->cma_dev) {
729                 switch (rdma_node_get_transport(id_priv->id.device->node_type)) {
730                 case RDMA_TRANSPORT_IB:
731                         if (id_priv->cm_id.ib && !IS_ERR(id_priv->cm_id.ib))
732                                 ib_destroy_cm_id(id_priv->cm_id.ib);
733                         break;
734                 case RDMA_TRANSPORT_IWARP:
735                         if (id_priv->cm_id.iw && !IS_ERR(id_priv->cm_id.iw))
736                                 iw_destroy_cm_id(id_priv->cm_id.iw);
737                         break;
738                 default:
739                         break;
740                 }
741                 cma_detach_from_dev(id_priv);
742         }
743         list_del(&id_priv->listen_list);
744
745         cma_deref_id(id_priv);
746         wait_for_completion(&id_priv->comp);
747
748         kfree(id_priv);
749 }
750
751 static void cma_cancel_listens(struct rdma_id_private *id_priv)
752 {
753         struct rdma_id_private *dev_id_priv;
754
755         mutex_lock(&lock);
756         list_del(&id_priv->list);
757
758         while (!list_empty(&id_priv->listen_list)) {
759                 dev_id_priv = list_entry(id_priv->listen_list.next,
760                                          struct rdma_id_private, listen_list);
761                 cma_destroy_listen(dev_id_priv);
762         }
763         mutex_unlock(&lock);
764 }
765
766 static void cma_cancel_operation(struct rdma_id_private *id_priv,
767                                  enum cma_state state)
768 {
769         switch (state) {
770         case CMA_ADDR_QUERY:
771                 rdma_addr_cancel(&id_priv->id.route.addr.dev_addr);
772                 break;
773         case CMA_ROUTE_QUERY:
774                 cma_cancel_route(id_priv);
775                 break;
776         case CMA_LISTEN:
777                 if (cma_any_addr(&id_priv->id.route.addr.src_addr) &&
778                     !id_priv->cma_dev)
779                         cma_cancel_listens(id_priv);
780                 break;
781         default:
782                 break;
783         }
784 }
785
786 static void cma_release_port(struct rdma_id_private *id_priv)
787 {
788         struct rdma_bind_list *bind_list = id_priv->bind_list;
789
790         if (!bind_list)
791                 return;
792
793         mutex_lock(&lock);
794         hlist_del(&id_priv->node);
795         if (hlist_empty(&bind_list->owners)) {
796                 idr_remove(bind_list->ps, bind_list->port);
797                 kfree(bind_list);
798         }
799         mutex_unlock(&lock);
800 }
801
802 static void cma_leave_mc_groups(struct rdma_id_private *id_priv)
803 {
804         struct cma_multicast *mc;
805
806         while (!list_empty(&id_priv->mc_list)) {
807                 mc = container_of(id_priv->mc_list.next,
808                                   struct cma_multicast, list);
809                 list_del(&mc->list);
810                 ib_sa_free_multicast(mc->multicast.ib);
811                 kfree(mc);
812         }
813 }
814
815 void rdma_destroy_id(struct rdma_cm_id *id)
816 {
817         struct rdma_id_private *id_priv;
818         enum cma_state state;
819
820         id_priv = container_of(id, struct rdma_id_private, id);
821         state = cma_exch(id_priv, CMA_DESTROYING);
822         cma_cancel_operation(id_priv, state);
823
824         mutex_lock(&lock);
825         if (id_priv->cma_dev) {
826                 mutex_unlock(&lock);
827                 switch (rdma_node_get_transport(id->device->node_type)) {
828                 case RDMA_TRANSPORT_IB:
829                         if (id_priv->cm_id.ib && !IS_ERR(id_priv->cm_id.ib))
830                                 ib_destroy_cm_id(id_priv->cm_id.ib);
831                         break;
832                 case RDMA_TRANSPORT_IWARP:
833                         if (id_priv->cm_id.iw && !IS_ERR(id_priv->cm_id.iw))
834                                 iw_destroy_cm_id(id_priv->cm_id.iw);
835                         break;
836                 default:
837                         break;
838                 }
839                 cma_leave_mc_groups(id_priv);
840                 mutex_lock(&lock);
841                 cma_detach_from_dev(id_priv);
842         }
843         mutex_unlock(&lock);
844
845         cma_release_port(id_priv);
846         cma_deref_id(id_priv);
847         wait_for_completion(&id_priv->comp);
848
849         kfree(id_priv->id.route.path_rec);
850         kfree(id_priv);
851 }
852 EXPORT_SYMBOL(rdma_destroy_id);
853
854 static int cma_rep_recv(struct rdma_id_private *id_priv)
855 {
856         int ret;
857
858         ret = cma_modify_qp_rtr(&id_priv->id);
859         if (ret)
860                 goto reject;
861
862         ret = cma_modify_qp_rts(&id_priv->id);
863         if (ret)
864                 goto reject;
865
866         ret = ib_send_cm_rtu(id_priv->cm_id.ib, NULL, 0);
867         if (ret)
868                 goto reject;
869
870         return 0;
871 reject:
872         cma_modify_qp_err(&id_priv->id);
873         ib_send_cm_rej(id_priv->cm_id.ib, IB_CM_REJ_CONSUMER_DEFINED,
874                        NULL, 0, NULL, 0);
875         return ret;
876 }
877
878 static int cma_verify_rep(struct rdma_id_private *id_priv, void *data)
879 {
880         if (id_priv->id.ps == RDMA_PS_SDP &&
881             sdp_get_majv(((struct sdp_hah *) data)->sdp_version) !=
882             SDP_MAJ_VERSION)
883                 return -EINVAL;
884
885         return 0;
886 }
887
888 static void cma_set_rep_event_data(struct rdma_cm_event *event,
889                                    struct ib_cm_rep_event_param *rep_data,
890                                    void *private_data)
891 {
892         event->param.conn.private_data = private_data;
893         event->param.conn.private_data_len = IB_CM_REP_PRIVATE_DATA_SIZE;
894         event->param.conn.responder_resources = rep_data->responder_resources;
895         event->param.conn.initiator_depth = rep_data->initiator_depth;
896         event->param.conn.flow_control = rep_data->flow_control;
897         event->param.conn.rnr_retry_count = rep_data->rnr_retry_count;
898         event->param.conn.srq = rep_data->srq;
899         event->param.conn.qp_num = rep_data->remote_qpn;
900 }
901
902 static int cma_ib_handler(struct ib_cm_id *cm_id, struct ib_cm_event *ib_event)
903 {
904         struct rdma_id_private *id_priv = cm_id->context;
905         struct rdma_cm_event event;
906         int ret = 0;
907
908         if (cma_disable_remove(id_priv, CMA_CONNECT))
909                 return 0;
910
911         memset(&event, 0, sizeof event);
912         switch (ib_event->event) {
913         case IB_CM_REQ_ERROR:
914         case IB_CM_REP_ERROR:
915                 event.event = RDMA_CM_EVENT_UNREACHABLE;
916                 event.status = -ETIMEDOUT;
917                 break;
918         case IB_CM_REP_RECEIVED:
919                 event.status = cma_verify_rep(id_priv, ib_event->private_data);
920                 if (event.status)
921                         event.event = RDMA_CM_EVENT_CONNECT_ERROR;
922                 else if (id_priv->id.qp && id_priv->id.ps != RDMA_PS_SDP) {
923                         event.status = cma_rep_recv(id_priv);
924                         event.event = event.status ? RDMA_CM_EVENT_CONNECT_ERROR :
925                                                      RDMA_CM_EVENT_ESTABLISHED;
926                 } else
927                         event.event = RDMA_CM_EVENT_CONNECT_RESPONSE;
928                 cma_set_rep_event_data(&event, &ib_event->param.rep_rcvd,
929                                        ib_event->private_data);
930                 break;
931         case IB_CM_RTU_RECEIVED:
932         case IB_CM_USER_ESTABLISHED:
933                 event.event = RDMA_CM_EVENT_ESTABLISHED;
934                 break;
935         case IB_CM_DREQ_ERROR:
936                 event.status = -ETIMEDOUT; /* fall through */
937         case IB_CM_DREQ_RECEIVED:
938         case IB_CM_DREP_RECEIVED:
939                 if (!cma_comp_exch(id_priv, CMA_CONNECT, CMA_DISCONNECT))
940                         goto out;
941                 event.event = RDMA_CM_EVENT_DISCONNECTED;
942                 break;
943         case IB_CM_TIMEWAIT_EXIT:
944         case IB_CM_MRA_RECEIVED:
945                 /* ignore event */
946                 goto out;
947         case IB_CM_REJ_RECEIVED:
948                 cma_modify_qp_err(&id_priv->id);
949                 event.status = ib_event->param.rej_rcvd.reason;
950                 event.event = RDMA_CM_EVENT_REJECTED;
951                 event.param.conn.private_data = ib_event->private_data;
952                 event.param.conn.private_data_len = IB_CM_REJ_PRIVATE_DATA_SIZE;
953                 break;
954         default:
955                 printk(KERN_ERR "RDMA CMA: unexpected IB CM event: %d",
956                        ib_event->event);
957                 goto out;
958         }
959
960         ret = id_priv->id.event_handler(&id_priv->id, &event);
961         if (ret) {
962                 /* Destroy the CM ID by returning a non-zero value. */
963                 id_priv->cm_id.ib = NULL;
964                 cma_exch(id_priv, CMA_DESTROYING);
965                 cma_enable_remove(id_priv);
966                 rdma_destroy_id(&id_priv->id);
967                 return ret;
968         }
969 out:
970         cma_enable_remove(id_priv);
971         return ret;
972 }
973
974 static struct rdma_id_private *cma_new_conn_id(struct rdma_cm_id *listen_id,
975                                                struct ib_cm_event *ib_event)
976 {
977         struct rdma_id_private *id_priv;
978         struct rdma_cm_id *id;
979         struct rdma_route *rt;
980         union cma_ip_addr *src, *dst;
981         __u16 port;
982         u8 ip_ver;
983
984         if (cma_get_net_info(ib_event->private_data, listen_id->ps,
985                              &ip_ver, &port, &src, &dst))
986                 goto err;
987
988         id = rdma_create_id(listen_id->event_handler, listen_id->context,
989                             listen_id->ps);
990         if (IS_ERR(id))
991                 goto err;
992
993         cma_save_net_info(&id->route.addr, &listen_id->route.addr,
994                           ip_ver, port, src, dst);
995
996         rt = &id->route;
997         rt->num_paths = ib_event->param.req_rcvd.alternate_path ? 2 : 1;
998         rt->path_rec = kmalloc(sizeof *rt->path_rec * rt->num_paths,
999                                GFP_KERNEL);
1000         if (!rt->path_rec)
1001                 goto destroy_id;
1002
1003         rt->path_rec[0] = *ib_event->param.req_rcvd.primary_path;
1004         if (rt->num_paths == 2)
1005                 rt->path_rec[1] = *ib_event->param.req_rcvd.alternate_path;
1006
1007         ib_addr_set_sgid(&rt->addr.dev_addr, &rt->path_rec[0].sgid);
1008         ib_addr_set_dgid(&rt->addr.dev_addr, &rt->path_rec[0].dgid);
1009         ib_addr_set_pkey(&rt->addr.dev_addr, be16_to_cpu(rt->path_rec[0].pkey));
1010         rt->addr.dev_addr.dev_type = RDMA_NODE_IB_CA;
1011
1012         id_priv = container_of(id, struct rdma_id_private, id);
1013         id_priv->state = CMA_CONNECT;
1014         return id_priv;
1015
1016 destroy_id:
1017         rdma_destroy_id(id);
1018 err:
1019         return NULL;
1020 }
1021
1022 static struct rdma_id_private *cma_new_udp_id(struct rdma_cm_id *listen_id,
1023                                               struct ib_cm_event *ib_event)
1024 {
1025         struct rdma_id_private *id_priv;
1026         struct rdma_cm_id *id;
1027         union cma_ip_addr *src, *dst;
1028         __u16 port;
1029         u8 ip_ver;
1030         int ret;
1031
1032         id = rdma_create_id(listen_id->event_handler, listen_id->context,
1033                             listen_id->ps);
1034         if (IS_ERR(id))
1035                 return NULL;
1036
1037
1038         if (cma_get_net_info(ib_event->private_data, listen_id->ps,
1039                              &ip_ver, &port, &src, &dst))
1040                 goto err;
1041
1042         cma_save_net_info(&id->route.addr, &listen_id->route.addr,
1043                           ip_ver, port, src, dst);
1044
1045         ret = rdma_translate_ip(&id->route.addr.src_addr,
1046                                 &id->route.addr.dev_addr);
1047         if (ret)
1048                 goto err;
1049
1050         id_priv = container_of(id, struct rdma_id_private, id);
1051         id_priv->state = CMA_CONNECT;
1052         return id_priv;
1053 err:
1054         rdma_destroy_id(id);
1055         return NULL;
1056 }
1057
1058 static void cma_set_req_event_data(struct rdma_cm_event *event,
1059                                    struct ib_cm_req_event_param *req_data,
1060                                    void *private_data, int offset)
1061 {
1062         event->param.conn.private_data = private_data + offset;
1063         event->param.conn.private_data_len = IB_CM_REQ_PRIVATE_DATA_SIZE - offset;
1064         event->param.conn.responder_resources = req_data->responder_resources;
1065         event->param.conn.initiator_depth = req_data->initiator_depth;
1066         event->param.conn.flow_control = req_data->flow_control;
1067         event->param.conn.retry_count = req_data->retry_count;
1068         event->param.conn.rnr_retry_count = req_data->rnr_retry_count;
1069         event->param.conn.srq = req_data->srq;
1070         event->param.conn.qp_num = req_data->remote_qpn;
1071 }
1072
1073 static int cma_req_handler(struct ib_cm_id *cm_id, struct ib_cm_event *ib_event)
1074 {
1075         struct rdma_id_private *listen_id, *conn_id;
1076         struct rdma_cm_event event;
1077         int offset, ret;
1078
1079         listen_id = cm_id->context;
1080         if (cma_disable_remove(listen_id, CMA_LISTEN))
1081                 return -ECONNABORTED;
1082
1083         memset(&event, 0, sizeof event);
1084         offset = cma_user_data_offset(listen_id->id.ps);
1085         event.event = RDMA_CM_EVENT_CONNECT_REQUEST;
1086         if (cma_is_ud_ps(listen_id->id.ps)) {
1087                 conn_id = cma_new_udp_id(&listen_id->id, ib_event);
1088                 event.param.ud.private_data = ib_event->private_data + offset;
1089                 event.param.ud.private_data_len =
1090                                 IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE - offset;
1091         } else {
1092                 conn_id = cma_new_conn_id(&listen_id->id, ib_event);
1093                 cma_set_req_event_data(&event, &ib_event->param.req_rcvd,
1094                                        ib_event->private_data, offset);
1095         }
1096         if (!conn_id) {
1097                 ret = -ENOMEM;
1098                 goto out;
1099         }
1100
1101         atomic_inc(&conn_id->dev_remove);
1102         mutex_lock(&lock);
1103         ret = cma_acquire_dev(conn_id);
1104         mutex_unlock(&lock);
1105         if (ret)
1106                 goto release_conn_id;
1107
1108         conn_id->cm_id.ib = cm_id;
1109         cm_id->context = conn_id;
1110         cm_id->cm_handler = cma_ib_handler;
1111
1112         ret = conn_id->id.event_handler(&conn_id->id, &event);
1113         if (!ret)
1114                 goto out;
1115
1116         /* Destroy the CM ID by returning a non-zero value. */
1117         conn_id->cm_id.ib = NULL;
1118
1119 release_conn_id:
1120         cma_exch(conn_id, CMA_DESTROYING);
1121         cma_enable_remove(conn_id);
1122         rdma_destroy_id(&conn_id->id);
1123
1124 out:
1125         cma_enable_remove(listen_id);
1126         return ret;
1127 }
1128
1129 static __be64 cma_get_service_id(enum rdma_port_space ps, struct sockaddr *addr)
1130 {
1131         return cpu_to_be64(((u64)ps << 16) + be16_to_cpu(cma_port(addr)));
1132 }
1133
1134 static void cma_set_compare_data(enum rdma_port_space ps, struct sockaddr *addr,
1135                                  struct ib_cm_compare_data *compare)
1136 {
1137         struct cma_hdr *cma_data, *cma_mask;
1138         struct sdp_hh *sdp_data, *sdp_mask;
1139         __u32 ip4_addr;
1140         struct in6_addr ip6_addr;
1141
1142         memset(compare, 0, sizeof *compare);
1143         cma_data = (void *) compare->data;
1144         cma_mask = (void *) compare->mask;
1145         sdp_data = (void *) compare->data;
1146         sdp_mask = (void *) compare->mask;
1147
1148         switch (addr->sa_family) {
1149         case AF_INET:
1150                 ip4_addr = ((struct sockaddr_in *) addr)->sin_addr.s_addr;
1151                 if (ps == RDMA_PS_SDP) {
1152                         sdp_set_ip_ver(sdp_data, 4);
1153                         sdp_set_ip_ver(sdp_mask, 0xF);
1154                         sdp_data->dst_addr.ip4.addr = ip4_addr;
1155                         sdp_mask->dst_addr.ip4.addr = ~0;
1156                 } else {
1157                         cma_set_ip_ver(cma_data, 4);
1158                         cma_set_ip_ver(cma_mask, 0xF);
1159                         cma_data->dst_addr.ip4.addr = ip4_addr;
1160                         cma_mask->dst_addr.ip4.addr = ~0;
1161                 }
1162                 break;
1163         case AF_INET6:
1164                 ip6_addr = ((struct sockaddr_in6 *) addr)->sin6_addr;
1165                 if (ps == RDMA_PS_SDP) {
1166                         sdp_set_ip_ver(sdp_data, 6);
1167                         sdp_set_ip_ver(sdp_mask, 0xF);
1168                         sdp_data->dst_addr.ip6 = ip6_addr;
1169                         memset(&sdp_mask->dst_addr.ip6, 0xFF,
1170                                sizeof sdp_mask->dst_addr.ip6);
1171                 } else {
1172                         cma_set_ip_ver(cma_data, 6);
1173                         cma_set_ip_ver(cma_mask, 0xF);
1174                         cma_data->dst_addr.ip6 = ip6_addr;
1175                         memset(&cma_mask->dst_addr.ip6, 0xFF,
1176                                sizeof cma_mask->dst_addr.ip6);
1177                 }
1178                 break;
1179         default:
1180                 break;
1181         }
1182 }
1183
1184 static int cma_iw_handler(struct iw_cm_id *iw_id, struct iw_cm_event *iw_event)
1185 {
1186         struct rdma_id_private *id_priv = iw_id->context;
1187         struct rdma_cm_event event;
1188         struct sockaddr_in *sin;
1189         int ret = 0;
1190
1191         if (cma_disable_remove(id_priv, CMA_CONNECT))
1192                 return 0;
1193
1194         memset(&event, 0, sizeof event);
1195         switch (iw_event->event) {
1196         case IW_CM_EVENT_CLOSE:
1197                 event.event = RDMA_CM_EVENT_DISCONNECTED;
1198                 break;
1199         case IW_CM_EVENT_CONNECT_REPLY:
1200                 sin = (struct sockaddr_in *) &id_priv->id.route.addr.src_addr;
1201                 *sin = iw_event->local_addr;
1202                 sin = (struct sockaddr_in *) &id_priv->id.route.addr.dst_addr;
1203                 *sin = iw_event->remote_addr;
1204                 switch (iw_event->status) {
1205                 case 0:
1206                         event.event = RDMA_CM_EVENT_ESTABLISHED;
1207                         break;
1208                 case -ECONNRESET:
1209                 case -ECONNREFUSED:
1210                         event.event = RDMA_CM_EVENT_REJECTED;
1211                         break;
1212                 case -ETIMEDOUT:
1213                         event.event = RDMA_CM_EVENT_UNREACHABLE;
1214                         break;
1215                 default:
1216                         event.event = RDMA_CM_EVENT_CONNECT_ERROR;
1217                         break;
1218                 }
1219                 break;
1220         case IW_CM_EVENT_ESTABLISHED:
1221                 event.event = RDMA_CM_EVENT_ESTABLISHED;
1222                 break;
1223         default:
1224                 BUG_ON(1);
1225         }
1226
1227         event.status = iw_event->status;
1228         event.param.conn.private_data = iw_event->private_data;
1229         event.param.conn.private_data_len = iw_event->private_data_len;
1230         ret = id_priv->id.event_handler(&id_priv->id, &event);
1231         if (ret) {
1232                 /* Destroy the CM ID by returning a non-zero value. */
1233                 id_priv->cm_id.iw = NULL;
1234                 cma_exch(id_priv, CMA_DESTROYING);
1235                 cma_enable_remove(id_priv);
1236                 rdma_destroy_id(&id_priv->id);
1237                 return ret;
1238         }
1239
1240         cma_enable_remove(id_priv);
1241         return ret;
1242 }
1243
1244 static int iw_conn_req_handler(struct iw_cm_id *cm_id,
1245                                struct iw_cm_event *iw_event)
1246 {
1247         struct rdma_cm_id *new_cm_id;
1248         struct rdma_id_private *listen_id, *conn_id;
1249         struct sockaddr_in *sin;
1250         struct net_device *dev = NULL;
1251         struct rdma_cm_event event;
1252         int ret;
1253
1254         listen_id = cm_id->context;
1255         if (cma_disable_remove(listen_id, CMA_LISTEN))
1256                 return -ECONNABORTED;
1257
1258         /* Create a new RDMA id for the new IW CM ID */
1259         new_cm_id = rdma_create_id(listen_id->id.event_handler,
1260                                    listen_id->id.context,
1261                                    RDMA_PS_TCP);
1262         if (!new_cm_id) {
1263                 ret = -ENOMEM;
1264                 goto out;
1265         }
1266         conn_id = container_of(new_cm_id, struct rdma_id_private, id);
1267         atomic_inc(&conn_id->dev_remove);
1268         conn_id->state = CMA_CONNECT;
1269
1270         dev = ip_dev_find(iw_event->local_addr.sin_addr.s_addr);
1271         if (!dev) {
1272                 ret = -EADDRNOTAVAIL;
1273                 cma_enable_remove(conn_id);
1274                 rdma_destroy_id(new_cm_id);
1275                 goto out;
1276         }
1277         ret = rdma_copy_addr(&conn_id->id.route.addr.dev_addr, dev, NULL);
1278         if (ret) {
1279                 cma_enable_remove(conn_id);
1280                 rdma_destroy_id(new_cm_id);
1281                 goto out;
1282         }
1283
1284         mutex_lock(&lock);
1285         ret = cma_acquire_dev(conn_id);
1286         mutex_unlock(&lock);
1287         if (ret) {
1288                 cma_enable_remove(conn_id);
1289                 rdma_destroy_id(new_cm_id);
1290                 goto out;
1291         }
1292
1293         conn_id->cm_id.iw = cm_id;
1294         cm_id->context = conn_id;
1295         cm_id->cm_handler = cma_iw_handler;
1296
1297         sin = (struct sockaddr_in *) &new_cm_id->route.addr.src_addr;
1298         *sin = iw_event->local_addr;
1299         sin = (struct sockaddr_in *) &new_cm_id->route.addr.dst_addr;
1300         *sin = iw_event->remote_addr;
1301
1302         memset(&event, 0, sizeof event);
1303         event.event = RDMA_CM_EVENT_CONNECT_REQUEST;
1304         event.param.conn.private_data = iw_event->private_data;
1305         event.param.conn.private_data_len = iw_event->private_data_len;
1306         ret = conn_id->id.event_handler(&conn_id->id, &event);
1307         if (ret) {
1308                 /* User wants to destroy the CM ID */
1309                 conn_id->cm_id.iw = NULL;
1310                 cma_exch(conn_id, CMA_DESTROYING);
1311                 cma_enable_remove(conn_id);
1312                 rdma_destroy_id(&conn_id->id);
1313         }
1314
1315 out:
1316         if (dev)
1317                 dev_put(dev);
1318         cma_enable_remove(listen_id);
1319         return ret;
1320 }
1321
1322 static int cma_ib_listen(struct rdma_id_private *id_priv)
1323 {
1324         struct ib_cm_compare_data compare_data;
1325         struct sockaddr *addr;
1326         __be64 svc_id;
1327         int ret;
1328
1329         id_priv->cm_id.ib = ib_create_cm_id(id_priv->id.device, cma_req_handler,
1330                                             id_priv);
1331         if (IS_ERR(id_priv->cm_id.ib))
1332                 return PTR_ERR(id_priv->cm_id.ib);
1333
1334         addr = &id_priv->id.route.addr.src_addr;
1335         svc_id = cma_get_service_id(id_priv->id.ps, addr);
1336         if (cma_any_addr(addr))
1337                 ret = ib_cm_listen(id_priv->cm_id.ib, svc_id, 0, NULL);
1338         else {
1339                 cma_set_compare_data(id_priv->id.ps, addr, &compare_data);
1340                 ret = ib_cm_listen(id_priv->cm_id.ib, svc_id, 0, &compare_data);
1341         }
1342
1343         if (ret) {
1344                 ib_destroy_cm_id(id_priv->cm_id.ib);
1345                 id_priv->cm_id.ib = NULL;
1346         }
1347
1348         return ret;
1349 }
1350
1351 static int cma_iw_listen(struct rdma_id_private *id_priv, int backlog)
1352 {
1353         int ret;
1354         struct sockaddr_in *sin;
1355
1356         id_priv->cm_id.iw = iw_create_cm_id(id_priv->id.device,
1357                                             iw_conn_req_handler,
1358                                             id_priv);
1359         if (IS_ERR(id_priv->cm_id.iw))
1360                 return PTR_ERR(id_priv->cm_id.iw);
1361
1362         sin = (struct sockaddr_in *) &id_priv->id.route.addr.src_addr;
1363         id_priv->cm_id.iw->local_addr = *sin;
1364
1365         ret = iw_cm_listen(id_priv->cm_id.iw, backlog);
1366
1367         if (ret) {
1368                 iw_destroy_cm_id(id_priv->cm_id.iw);
1369                 id_priv->cm_id.iw = NULL;
1370         }
1371
1372         return ret;
1373 }
1374
1375 static int cma_listen_handler(struct rdma_cm_id *id,
1376                               struct rdma_cm_event *event)
1377 {
1378         struct rdma_id_private *id_priv = id->context;
1379
1380         id->context = id_priv->id.context;
1381         id->event_handler = id_priv->id.event_handler;
1382         return id_priv->id.event_handler(id, event);
1383 }
1384
1385 static void cma_listen_on_dev(struct rdma_id_private *id_priv,
1386                               struct cma_device *cma_dev)
1387 {
1388         struct rdma_id_private *dev_id_priv;
1389         struct rdma_cm_id *id;
1390         int ret;
1391
1392         id = rdma_create_id(cma_listen_handler, id_priv, id_priv->id.ps);
1393         if (IS_ERR(id))
1394                 return;
1395
1396         dev_id_priv = container_of(id, struct rdma_id_private, id);
1397
1398         dev_id_priv->state = CMA_ADDR_BOUND;
1399         memcpy(&id->route.addr.src_addr, &id_priv->id.route.addr.src_addr,
1400                ip_addr_size(&id_priv->id.route.addr.src_addr));
1401
1402         cma_attach_to_dev(dev_id_priv, cma_dev);
1403         list_add_tail(&dev_id_priv->listen_list, &id_priv->listen_list);
1404
1405         ret = rdma_listen(id, id_priv->backlog);
1406         if (ret)
1407                 goto err;
1408
1409         return;
1410 err:
1411         cma_destroy_listen(dev_id_priv);
1412 }
1413
1414 static void cma_listen_on_all(struct rdma_id_private *id_priv)
1415 {
1416         struct cma_device *cma_dev;
1417
1418         mutex_lock(&lock);
1419         list_add_tail(&id_priv->list, &listen_any_list);
1420         list_for_each_entry(cma_dev, &dev_list, list)
1421                 cma_listen_on_dev(id_priv, cma_dev);
1422         mutex_unlock(&lock);
1423 }
1424
1425 static int cma_bind_any(struct rdma_cm_id *id, sa_family_t af)
1426 {
1427         struct sockaddr_in addr_in;
1428
1429         memset(&addr_in, 0, sizeof addr_in);
1430         addr_in.sin_family = af;
1431         return rdma_bind_addr(id, (struct sockaddr *) &addr_in);
1432 }
1433
1434 int rdma_listen(struct rdma_cm_id *id, int backlog)
1435 {
1436         struct rdma_id_private *id_priv;
1437         int ret;
1438
1439         id_priv = container_of(id, struct rdma_id_private, id);
1440         if (id_priv->state == CMA_IDLE) {
1441                 ret = cma_bind_any(id, AF_INET);
1442                 if (ret)
1443                         return ret;
1444         }
1445
1446         if (!cma_comp_exch(id_priv, CMA_ADDR_BOUND, CMA_LISTEN))
1447                 return -EINVAL;
1448
1449         id_priv->backlog = backlog;
1450         if (id->device) {
1451                 switch (rdma_node_get_transport(id->device->node_type)) {
1452                 case RDMA_TRANSPORT_IB:
1453                         ret = cma_ib_listen(id_priv);
1454                         if (ret)
1455                                 goto err;
1456                         break;
1457                 case RDMA_TRANSPORT_IWARP:
1458                         ret = cma_iw_listen(id_priv, backlog);
1459                         if (ret)
1460                                 goto err;
1461                         break;
1462                 default:
1463                         ret = -ENOSYS;
1464                         goto err;
1465                 }
1466         } else
1467                 cma_listen_on_all(id_priv);
1468
1469         return 0;
1470 err:
1471         id_priv->backlog = 0;
1472         cma_comp_exch(id_priv, CMA_LISTEN, CMA_ADDR_BOUND);
1473         return ret;
1474 }
1475 EXPORT_SYMBOL(rdma_listen);
1476
1477 static void cma_query_handler(int status, struct ib_sa_path_rec *path_rec,
1478                               void *context)
1479 {
1480         struct cma_work *work = context;
1481         struct rdma_route *route;
1482
1483         route = &work->id->id.route;
1484
1485         if (!status) {
1486                 route->num_paths = 1;
1487                 *route->path_rec = *path_rec;
1488         } else {
1489                 work->old_state = CMA_ROUTE_QUERY;
1490                 work->new_state = CMA_ADDR_RESOLVED;
1491                 work->event.event = RDMA_CM_EVENT_ROUTE_ERROR;
1492                 work->event.status = status;
1493         }
1494
1495         queue_work(cma_wq, &work->work);
1496 }
1497
1498 static int cma_query_ib_route(struct rdma_id_private *id_priv, int timeout_ms,
1499                               struct cma_work *work)
1500 {
1501         struct rdma_dev_addr *addr = &id_priv->id.route.addr.dev_addr;
1502         struct ib_sa_path_rec path_rec;
1503
1504         memset(&path_rec, 0, sizeof path_rec);
1505         ib_addr_get_sgid(addr, &path_rec.sgid);
1506         ib_addr_get_dgid(addr, &path_rec.dgid);
1507         path_rec.pkey = cpu_to_be16(ib_addr_get_pkey(addr));
1508         path_rec.numb_path = 1;
1509         path_rec.reversible = 1;
1510
1511         id_priv->query_id = ib_sa_path_rec_get(&sa_client, id_priv->id.device,
1512                                 id_priv->id.port_num, &path_rec,
1513                                 IB_SA_PATH_REC_DGID | IB_SA_PATH_REC_SGID |
1514                                 IB_SA_PATH_REC_PKEY | IB_SA_PATH_REC_NUMB_PATH |
1515                                 IB_SA_PATH_REC_REVERSIBLE,
1516                                 timeout_ms, GFP_KERNEL,
1517                                 cma_query_handler, work, &id_priv->query);
1518
1519         return (id_priv->query_id < 0) ? id_priv->query_id : 0;
1520 }
1521
1522 static void cma_work_handler(struct work_struct *_work)
1523 {
1524         struct cma_work *work = container_of(_work, struct cma_work, work);
1525         struct rdma_id_private *id_priv = work->id;
1526         int destroy = 0;
1527
1528         atomic_inc(&id_priv->dev_remove);
1529         if (!cma_comp_exch(id_priv, work->old_state, work->new_state))
1530                 goto out;
1531
1532         if (id_priv->id.event_handler(&id_priv->id, &work->event)) {
1533                 cma_exch(id_priv, CMA_DESTROYING);
1534                 destroy = 1;
1535         }
1536 out:
1537         cma_enable_remove(id_priv);
1538         cma_deref_id(id_priv);
1539         if (destroy)
1540                 rdma_destroy_id(&id_priv->id);
1541         kfree(work);
1542 }
1543
1544 static int cma_resolve_ib_route(struct rdma_id_private *id_priv, int timeout_ms)
1545 {
1546         struct rdma_route *route = &id_priv->id.route;
1547         struct cma_work *work;
1548         int ret;
1549
1550         work = kzalloc(sizeof *work, GFP_KERNEL);
1551         if (!work)
1552                 return -ENOMEM;
1553
1554         work->id = id_priv;
1555         INIT_WORK(&work->work, cma_work_handler);
1556         work->old_state = CMA_ROUTE_QUERY;
1557         work->new_state = CMA_ROUTE_RESOLVED;
1558         work->event.event = RDMA_CM_EVENT_ROUTE_RESOLVED;
1559
1560         route->path_rec = kmalloc(sizeof *route->path_rec, GFP_KERNEL);
1561         if (!route->path_rec) {
1562                 ret = -ENOMEM;
1563                 goto err1;
1564         }
1565
1566         ret = cma_query_ib_route(id_priv, timeout_ms, work);
1567         if (ret)
1568                 goto err2;
1569
1570         return 0;
1571 err2:
1572         kfree(route->path_rec);
1573         route->path_rec = NULL;
1574 err1:
1575         kfree(work);
1576         return ret;
1577 }
1578
1579 int rdma_set_ib_paths(struct rdma_cm_id *id,
1580                       struct ib_sa_path_rec *path_rec, int num_paths)
1581 {
1582         struct rdma_id_private *id_priv;
1583         int ret;
1584
1585         id_priv = container_of(id, struct rdma_id_private, id);
1586         if (!cma_comp_exch(id_priv, CMA_ADDR_RESOLVED, CMA_ROUTE_RESOLVED))
1587                 return -EINVAL;
1588
1589         id->route.path_rec = kmalloc(sizeof *path_rec * num_paths, GFP_KERNEL);
1590         if (!id->route.path_rec) {
1591                 ret = -ENOMEM;
1592                 goto err;
1593         }
1594
1595         memcpy(id->route.path_rec, path_rec, sizeof *path_rec * num_paths);
1596         return 0;
1597 err:
1598         cma_comp_exch(id_priv, CMA_ROUTE_RESOLVED, CMA_ADDR_RESOLVED);
1599         return ret;
1600 }
1601 EXPORT_SYMBOL(rdma_set_ib_paths);
1602
1603 static int cma_resolve_iw_route(struct rdma_id_private *id_priv, int timeout_ms)
1604 {
1605         struct cma_work *work;
1606
1607         work = kzalloc(sizeof *work, GFP_KERNEL);
1608         if (!work)
1609                 return -ENOMEM;
1610
1611         work->id = id_priv;
1612         INIT_WORK(&work->work, cma_work_handler);
1613         work->old_state = CMA_ROUTE_QUERY;
1614         work->new_state = CMA_ROUTE_RESOLVED;
1615         work->event.event = RDMA_CM_EVENT_ROUTE_RESOLVED;
1616         queue_work(cma_wq, &work->work);
1617         return 0;
1618 }
1619
1620 int rdma_resolve_route(struct rdma_cm_id *id, int timeout_ms)
1621 {
1622         struct rdma_id_private *id_priv;
1623         int ret;
1624
1625         id_priv = container_of(id, struct rdma_id_private, id);
1626         if (!cma_comp_exch(id_priv, CMA_ADDR_RESOLVED, CMA_ROUTE_QUERY))
1627                 return -EINVAL;
1628
1629         atomic_inc(&id_priv->refcount);
1630         switch (rdma_node_get_transport(id->device->node_type)) {
1631         case RDMA_TRANSPORT_IB:
1632                 ret = cma_resolve_ib_route(id_priv, timeout_ms);
1633                 break;
1634         case RDMA_TRANSPORT_IWARP:
1635                 ret = cma_resolve_iw_route(id_priv, timeout_ms);
1636                 break;
1637         default:
1638                 ret = -ENOSYS;
1639                 break;
1640         }
1641         if (ret)
1642                 goto err;
1643
1644         return 0;
1645 err:
1646         cma_comp_exch(id_priv, CMA_ROUTE_QUERY, CMA_ADDR_RESOLVED);
1647         cma_deref_id(id_priv);
1648         return ret;
1649 }
1650 EXPORT_SYMBOL(rdma_resolve_route);
1651
1652 static int cma_bind_loopback(struct rdma_id_private *id_priv)
1653 {
1654         struct cma_device *cma_dev;
1655         struct ib_port_attr port_attr;
1656         union ib_gid gid;
1657         u16 pkey;
1658         int ret;
1659         u8 p;
1660
1661         mutex_lock(&lock);
1662         if (list_empty(&dev_list)) {
1663                 ret = -ENODEV;
1664                 goto out;
1665         }
1666         list_for_each_entry(cma_dev, &dev_list, list)
1667                 for (p = 1; p <= cma_dev->device->phys_port_cnt; ++p)
1668                         if (!ib_query_port(cma_dev->device, p, &port_attr) &&
1669                             port_attr.state == IB_PORT_ACTIVE)
1670                                 goto port_found;
1671
1672         p = 1;
1673         cma_dev = list_entry(dev_list.next, struct cma_device, list);
1674
1675 port_found:
1676         ret = ib_get_cached_gid(cma_dev->device, p, 0, &gid);
1677         if (ret)
1678                 goto out;
1679
1680         ret = ib_get_cached_pkey(cma_dev->device, p, 0, &pkey);
1681         if (ret)
1682                 goto out;
1683
1684         ib_addr_set_sgid(&id_priv->id.route.addr.dev_addr, &gid);
1685         ib_addr_set_pkey(&id_priv->id.route.addr.dev_addr, pkey);
1686         id_priv->id.port_num = p;
1687         cma_attach_to_dev(id_priv, cma_dev);
1688 out:
1689         mutex_unlock(&lock);
1690         return ret;
1691 }
1692
1693 static void addr_handler(int status, struct sockaddr *src_addr,
1694                          struct rdma_dev_addr *dev_addr, void *context)
1695 {
1696         struct rdma_id_private *id_priv = context;
1697         struct rdma_cm_event event;
1698
1699         memset(&event, 0, sizeof event);
1700         atomic_inc(&id_priv->dev_remove);
1701
1702         /*
1703          * Grab mutex to block rdma_destroy_id() from removing the device while
1704          * we're trying to acquire it.
1705          */
1706         mutex_lock(&lock);
1707         if (!cma_comp_exch(id_priv, CMA_ADDR_QUERY, CMA_ADDR_RESOLVED)) {
1708                 mutex_unlock(&lock);
1709                 goto out;
1710         }
1711
1712         if (!status && !id_priv->cma_dev)
1713                 status = cma_acquire_dev(id_priv);
1714         mutex_unlock(&lock);
1715
1716         if (status) {
1717                 if (!cma_comp_exch(id_priv, CMA_ADDR_RESOLVED, CMA_ADDR_BOUND))
1718                         goto out;
1719                 event.event = RDMA_CM_EVENT_ADDR_ERROR;
1720                 event.status = status;
1721         } else {
1722                 memcpy(&id_priv->id.route.addr.src_addr, src_addr,
1723                        ip_addr_size(src_addr));
1724                 event.event = RDMA_CM_EVENT_ADDR_RESOLVED;
1725         }
1726
1727         if (id_priv->id.event_handler(&id_priv->id, &event)) {
1728                 cma_exch(id_priv, CMA_DESTROYING);
1729                 cma_enable_remove(id_priv);
1730                 cma_deref_id(id_priv);
1731                 rdma_destroy_id(&id_priv->id);
1732                 return;
1733         }
1734 out:
1735         cma_enable_remove(id_priv);
1736         cma_deref_id(id_priv);
1737 }
1738
1739 static int cma_resolve_loopback(struct rdma_id_private *id_priv)
1740 {
1741         struct cma_work *work;
1742         struct sockaddr_in *src_in, *dst_in;
1743         union ib_gid gid;
1744         int ret;
1745
1746         work = kzalloc(sizeof *work, GFP_KERNEL);
1747         if (!work)
1748                 return -ENOMEM;
1749
1750         if (!id_priv->cma_dev) {
1751                 ret = cma_bind_loopback(id_priv);
1752                 if (ret)
1753                         goto err;
1754         }
1755
1756         ib_addr_get_sgid(&id_priv->id.route.addr.dev_addr, &gid);
1757         ib_addr_set_dgid(&id_priv->id.route.addr.dev_addr, &gid);
1758
1759         if (cma_zero_addr(&id_priv->id.route.addr.src_addr)) {
1760                 src_in = (struct sockaddr_in *)&id_priv->id.route.addr.src_addr;
1761                 dst_in = (struct sockaddr_in *)&id_priv->id.route.addr.dst_addr;
1762                 src_in->sin_family = dst_in->sin_family;
1763                 src_in->sin_addr.s_addr = dst_in->sin_addr.s_addr;
1764         }
1765
1766         work->id = id_priv;
1767         INIT_WORK(&work->work, cma_work_handler);
1768         work->old_state = CMA_ADDR_QUERY;
1769         work->new_state = CMA_ADDR_RESOLVED;
1770         work->event.event = RDMA_CM_EVENT_ADDR_RESOLVED;
1771         queue_work(cma_wq, &work->work);
1772         return 0;
1773 err:
1774         kfree(work);
1775         return ret;
1776 }
1777
1778 static int cma_bind_addr(struct rdma_cm_id *id, struct sockaddr *src_addr,
1779                          struct sockaddr *dst_addr)
1780 {
1781         if (src_addr && src_addr->sa_family)
1782                 return rdma_bind_addr(id, src_addr);
1783         else
1784                 return cma_bind_any(id, dst_addr->sa_family);
1785 }
1786
1787 int rdma_resolve_addr(struct rdma_cm_id *id, struct sockaddr *src_addr,
1788                       struct sockaddr *dst_addr, int timeout_ms)
1789 {
1790         struct rdma_id_private *id_priv;
1791         int ret;
1792
1793         id_priv = container_of(id, struct rdma_id_private, id);
1794         if (id_priv->state == CMA_IDLE) {
1795                 ret = cma_bind_addr(id, src_addr, dst_addr);
1796                 if (ret)
1797                         return ret;
1798         }
1799
1800         if (!cma_comp_exch(id_priv, CMA_ADDR_BOUND, CMA_ADDR_QUERY))
1801                 return -EINVAL;
1802
1803         atomic_inc(&id_priv->refcount);
1804         memcpy(&id->route.addr.dst_addr, dst_addr, ip_addr_size(dst_addr));
1805         if (cma_any_addr(dst_addr))
1806                 ret = cma_resolve_loopback(id_priv);
1807         else
1808                 ret = rdma_resolve_ip(&addr_client, &id->route.addr.src_addr,
1809                                       dst_addr, &id->route.addr.dev_addr,
1810                                       timeout_ms, addr_handler, id_priv);
1811         if (ret)
1812                 goto err;
1813
1814         return 0;
1815 err:
1816         cma_comp_exch(id_priv, CMA_ADDR_QUERY, CMA_ADDR_BOUND);
1817         cma_deref_id(id_priv);
1818         return ret;
1819 }
1820 EXPORT_SYMBOL(rdma_resolve_addr);
1821
1822 static void cma_bind_port(struct rdma_bind_list *bind_list,
1823                           struct rdma_id_private *id_priv)
1824 {
1825         struct sockaddr_in *sin;
1826
1827         sin = (struct sockaddr_in *) &id_priv->id.route.addr.src_addr;
1828         sin->sin_port = htons(bind_list->port);
1829         id_priv->bind_list = bind_list;
1830         hlist_add_head(&id_priv->node, &bind_list->owners);
1831 }
1832
1833 static int cma_alloc_port(struct idr *ps, struct rdma_id_private *id_priv,
1834                           unsigned short snum)
1835 {
1836         struct rdma_bind_list *bind_list;
1837         int port, ret;
1838
1839         bind_list = kzalloc(sizeof *bind_list, GFP_KERNEL);
1840         if (!bind_list)
1841                 return -ENOMEM;
1842
1843         do {
1844                 ret = idr_get_new_above(ps, bind_list, snum, &port);
1845         } while ((ret == -EAGAIN) && idr_pre_get(ps, GFP_KERNEL));
1846
1847         if (ret)
1848                 goto err1;
1849
1850         if (port != snum) {
1851                 ret = -EADDRNOTAVAIL;
1852                 goto err2;
1853         }
1854
1855         bind_list->ps = ps;
1856         bind_list->port = (unsigned short) port;
1857         cma_bind_port(bind_list, id_priv);
1858         return 0;
1859 err2:
1860         idr_remove(ps, port);
1861 err1:
1862         kfree(bind_list);
1863         return ret;
1864 }
1865
1866 static int cma_alloc_any_port(struct idr *ps, struct rdma_id_private *id_priv)
1867 {
1868         struct rdma_bind_list *bind_list;
1869         int port, ret;
1870
1871         bind_list = kzalloc(sizeof *bind_list, GFP_KERNEL);
1872         if (!bind_list)
1873                 return -ENOMEM;
1874
1875 retry:
1876         do {
1877                 ret = idr_get_new_above(ps, bind_list, next_port, &port);
1878         } while ((ret == -EAGAIN) && idr_pre_get(ps, GFP_KERNEL));
1879
1880         if (ret)
1881                 goto err1;
1882
1883         if (port > sysctl_local_port_range[1]) {
1884                 if (next_port != sysctl_local_port_range[0]) {
1885                         idr_remove(ps, port);
1886                         next_port = sysctl_local_port_range[0];
1887                         goto retry;
1888                 }
1889                 ret = -EADDRNOTAVAIL;
1890                 goto err2;
1891         }
1892
1893         if (port == sysctl_local_port_range[1])
1894                 next_port = sysctl_local_port_range[0];
1895         else
1896                 next_port = port + 1;
1897
1898         bind_list->ps = ps;
1899         bind_list->port = (unsigned short) port;
1900         cma_bind_port(bind_list, id_priv);
1901         return 0;
1902 err2:
1903         idr_remove(ps, port);
1904 err1:
1905         kfree(bind_list);
1906         return ret;
1907 }
1908
1909 static int cma_use_port(struct idr *ps, struct rdma_id_private *id_priv)
1910 {
1911         struct rdma_id_private *cur_id;
1912         struct sockaddr_in *sin, *cur_sin;
1913         struct rdma_bind_list *bind_list;
1914         struct hlist_node *node;
1915         unsigned short snum;
1916
1917         sin = (struct sockaddr_in *) &id_priv->id.route.addr.src_addr;
1918         snum = ntohs(sin->sin_port);
1919         if (snum < PROT_SOCK && !capable(CAP_NET_BIND_SERVICE))
1920                 return -EACCES;
1921
1922         bind_list = idr_find(ps, snum);
1923         if (!bind_list)
1924                 return cma_alloc_port(ps, id_priv, snum);
1925
1926         /*
1927          * We don't support binding to any address if anyone is bound to
1928          * a specific address on the same port.
1929          */
1930         if (cma_any_addr(&id_priv->id.route.addr.src_addr))
1931                 return -EADDRNOTAVAIL;
1932
1933         hlist_for_each_entry(cur_id, node, &bind_list->owners, node) {
1934                 if (cma_any_addr(&cur_id->id.route.addr.src_addr))
1935                         return -EADDRNOTAVAIL;
1936
1937                 cur_sin = (struct sockaddr_in *) &cur_id->id.route.addr.src_addr;
1938                 if (sin->sin_addr.s_addr == cur_sin->sin_addr.s_addr)
1939                         return -EADDRINUSE;
1940         }
1941
1942         cma_bind_port(bind_list, id_priv);
1943         return 0;
1944 }
1945
1946 static int cma_get_port(struct rdma_id_private *id_priv)
1947 {
1948         struct idr *ps;
1949         int ret;
1950
1951         switch (id_priv->id.ps) {
1952         case RDMA_PS_SDP:
1953                 ps = &sdp_ps;
1954                 break;
1955         case RDMA_PS_TCP:
1956                 ps = &tcp_ps;
1957                 break;
1958         case RDMA_PS_UDP:
1959                 ps = &udp_ps;
1960                 break;
1961         case RDMA_PS_IPOIB:
1962                 ps = &ipoib_ps;
1963                 break;
1964         default:
1965                 return -EPROTONOSUPPORT;
1966         }
1967
1968         mutex_lock(&lock);
1969         if (cma_any_port(&id_priv->id.route.addr.src_addr))
1970                 ret = cma_alloc_any_port(ps, id_priv);
1971         else
1972                 ret = cma_use_port(ps, id_priv);
1973         mutex_unlock(&lock);
1974
1975         return ret;
1976 }
1977
1978 int rdma_bind_addr(struct rdma_cm_id *id, struct sockaddr *addr)
1979 {
1980         struct rdma_id_private *id_priv;
1981         int ret;
1982
1983         if (addr->sa_family != AF_INET)
1984                 return -EAFNOSUPPORT;
1985
1986         id_priv = container_of(id, struct rdma_id_private, id);
1987         if (!cma_comp_exch(id_priv, CMA_IDLE, CMA_ADDR_BOUND))
1988                 return -EINVAL;
1989
1990         if (!cma_any_addr(addr)) {
1991                 ret = rdma_translate_ip(addr, &id->route.addr.dev_addr);
1992                 if (ret)
1993                         goto err1;
1994
1995                 mutex_lock(&lock);
1996                 ret = cma_acquire_dev(id_priv);
1997                 mutex_unlock(&lock);
1998                 if (ret)
1999                         goto err1;
2000         }
2001
2002         memcpy(&id->route.addr.src_addr, addr, ip_addr_size(addr));
2003         ret = cma_get_port(id_priv);
2004         if (ret)
2005                 goto err2;
2006
2007         return 0;
2008 err2:
2009         if (!cma_any_addr(addr)) {
2010                 mutex_lock(&lock);
2011                 cma_detach_from_dev(id_priv);
2012                 mutex_unlock(&lock);
2013         }
2014 err1:
2015         cma_comp_exch(id_priv, CMA_ADDR_BOUND, CMA_IDLE);
2016         return ret;
2017 }
2018 EXPORT_SYMBOL(rdma_bind_addr);
2019
2020 static int cma_format_hdr(void *hdr, enum rdma_port_space ps,
2021                           struct rdma_route *route)
2022 {
2023         struct sockaddr_in *src4, *dst4;
2024         struct cma_hdr *cma_hdr;
2025         struct sdp_hh *sdp_hdr;
2026
2027         src4 = (struct sockaddr_in *) &route->addr.src_addr;
2028         dst4 = (struct sockaddr_in *) &route->addr.dst_addr;
2029
2030         switch (ps) {
2031         case RDMA_PS_SDP:
2032                 sdp_hdr = hdr;
2033                 if (sdp_get_majv(sdp_hdr->sdp_version) != SDP_MAJ_VERSION)
2034                         return -EINVAL;
2035                 sdp_set_ip_ver(sdp_hdr, 4);
2036                 sdp_hdr->src_addr.ip4.addr = src4->sin_addr.s_addr;
2037                 sdp_hdr->dst_addr.ip4.addr = dst4->sin_addr.s_addr;
2038                 sdp_hdr->port = src4->sin_port;
2039                 break;
2040         default:
2041                 cma_hdr = hdr;
2042                 cma_hdr->cma_version = CMA_VERSION;
2043                 cma_set_ip_ver(cma_hdr, 4);
2044                 cma_hdr->src_addr.ip4.addr = src4->sin_addr.s_addr;
2045                 cma_hdr->dst_addr.ip4.addr = dst4->sin_addr.s_addr;
2046                 cma_hdr->port = src4->sin_port;
2047                 break;
2048         }
2049         return 0;
2050 }
2051
2052 static int cma_sidr_rep_handler(struct ib_cm_id *cm_id,
2053                                 struct ib_cm_event *ib_event)
2054 {
2055         struct rdma_id_private *id_priv = cm_id->context;
2056         struct rdma_cm_event event;
2057         struct ib_cm_sidr_rep_event_param *rep = &ib_event->param.sidr_rep_rcvd;
2058         int ret = 0;
2059
2060         if (cma_disable_remove(id_priv, CMA_CONNECT))
2061                 return 0;
2062
2063         memset(&event, 0, sizeof event);
2064         switch (ib_event->event) {
2065         case IB_CM_SIDR_REQ_ERROR:
2066                 event.event = RDMA_CM_EVENT_UNREACHABLE;
2067                 event.status = -ETIMEDOUT;
2068                 break;
2069         case IB_CM_SIDR_REP_RECEIVED:
2070                 event.param.ud.private_data = ib_event->private_data;
2071                 event.param.ud.private_data_len = IB_CM_SIDR_REP_PRIVATE_DATA_SIZE;
2072                 if (rep->status != IB_SIDR_SUCCESS) {
2073                         event.event = RDMA_CM_EVENT_UNREACHABLE;
2074                         event.status = ib_event->param.sidr_rep_rcvd.status;
2075                         break;
2076                 }
2077                 if (id_priv->qkey != rep->qkey) {
2078                         event.event = RDMA_CM_EVENT_UNREACHABLE;
2079                         event.status = -EINVAL;
2080                         break;
2081                 }
2082                 ib_init_ah_from_path(id_priv->id.device, id_priv->id.port_num,
2083                                      id_priv->id.route.path_rec,
2084                                      &event.param.ud.ah_attr);
2085                 event.param.ud.qp_num = rep->qpn;
2086                 event.param.ud.qkey = rep->qkey;
2087                 event.event = RDMA_CM_EVENT_ESTABLISHED;
2088                 event.status = 0;
2089                 break;
2090         default:
2091                 printk(KERN_ERR "RDMA CMA: unexpected IB CM event: %d",
2092                        ib_event->event);
2093                 goto out;
2094         }
2095
2096         ret = id_priv->id.event_handler(&id_priv->id, &event);
2097         if (ret) {
2098                 /* Destroy the CM ID by returning a non-zero value. */
2099                 id_priv->cm_id.ib = NULL;
2100                 cma_exch(id_priv, CMA_DESTROYING);
2101                 cma_enable_remove(id_priv);
2102                 rdma_destroy_id(&id_priv->id);
2103                 return ret;
2104         }
2105 out:
2106         cma_enable_remove(id_priv);
2107         return ret;
2108 }
2109
2110 static int cma_resolve_ib_udp(struct rdma_id_private *id_priv,
2111                               struct rdma_conn_param *conn_param)
2112 {
2113         struct ib_cm_sidr_req_param req;
2114         struct rdma_route *route;
2115         int ret;
2116
2117         req.private_data_len = sizeof(struct cma_hdr) +
2118                                conn_param->private_data_len;
2119         req.private_data = kzalloc(req.private_data_len, GFP_ATOMIC);
2120         if (!req.private_data)
2121                 return -ENOMEM;
2122
2123         if (conn_param->private_data && conn_param->private_data_len)
2124                 memcpy((void *) req.private_data + sizeof(struct cma_hdr),
2125                        conn_param->private_data, conn_param->private_data_len);
2126
2127         route = &id_priv->id.route;
2128         ret = cma_format_hdr((void *) req.private_data, id_priv->id.ps, route);
2129         if (ret)
2130                 goto out;
2131
2132         id_priv->cm_id.ib = ib_create_cm_id(id_priv->id.device,
2133                                             cma_sidr_rep_handler, id_priv);
2134         if (IS_ERR(id_priv->cm_id.ib)) {
2135                 ret = PTR_ERR(id_priv->cm_id.ib);
2136                 goto out;
2137         }
2138
2139         req.path = route->path_rec;
2140         req.service_id = cma_get_service_id(id_priv->id.ps,
2141                                             &route->addr.dst_addr);
2142         req.timeout_ms = 1 << (CMA_CM_RESPONSE_TIMEOUT - 8);
2143         req.max_cm_retries = CMA_MAX_CM_RETRIES;
2144
2145         ret = ib_send_cm_sidr_req(id_priv->cm_id.ib, &req);
2146         if (ret) {
2147                 ib_destroy_cm_id(id_priv->cm_id.ib);
2148                 id_priv->cm_id.ib = NULL;
2149         }
2150 out:
2151         kfree(req.private_data);
2152         return ret;
2153 }
2154
2155 static int cma_connect_ib(struct rdma_id_private *id_priv,
2156                           struct rdma_conn_param *conn_param)
2157 {
2158         struct ib_cm_req_param req;
2159         struct rdma_route *route;
2160         void *private_data;
2161         int offset, ret;
2162
2163         memset(&req, 0, sizeof req);
2164         offset = cma_user_data_offset(id_priv->id.ps);
2165         req.private_data_len = offset + conn_param->private_data_len;
2166         private_data = kzalloc(req.private_data_len, GFP_ATOMIC);
2167         if (!private_data)
2168                 return -ENOMEM;
2169
2170         if (conn_param->private_data && conn_param->private_data_len)
2171                 memcpy(private_data + offset, conn_param->private_data,
2172                        conn_param->private_data_len);
2173
2174         id_priv->cm_id.ib = ib_create_cm_id(id_priv->id.device, cma_ib_handler,
2175                                             id_priv);
2176         if (IS_ERR(id_priv->cm_id.ib)) {
2177                 ret = PTR_ERR(id_priv->cm_id.ib);
2178                 goto out;
2179         }
2180
2181         route = &id_priv->id.route;
2182         ret = cma_format_hdr(private_data, id_priv->id.ps, route);
2183         if (ret)
2184                 goto out;
2185         req.private_data = private_data;
2186
2187         req.primary_path = &route->path_rec[0];
2188         if (route->num_paths == 2)
2189                 req.alternate_path = &route->path_rec[1];
2190
2191         req.service_id = cma_get_service_id(id_priv->id.ps,
2192                                             &route->addr.dst_addr);
2193         req.qp_num = id_priv->qp_num;
2194         req.qp_type = IB_QPT_RC;
2195         req.starting_psn = id_priv->seq_num;
2196         req.responder_resources = conn_param->responder_resources;
2197         req.initiator_depth = conn_param->initiator_depth;
2198         req.flow_control = conn_param->flow_control;
2199         req.retry_count = conn_param->retry_count;
2200         req.rnr_retry_count = conn_param->rnr_retry_count;
2201         req.remote_cm_response_timeout = CMA_CM_RESPONSE_TIMEOUT;
2202         req.local_cm_response_timeout = CMA_CM_RESPONSE_TIMEOUT;
2203         req.max_cm_retries = CMA_MAX_CM_RETRIES;
2204         req.srq = id_priv->srq ? 1 : 0;
2205
2206         ret = ib_send_cm_req(id_priv->cm_id.ib, &req);
2207 out:
2208         if (ret && !IS_ERR(id_priv->cm_id.ib)) {
2209                 ib_destroy_cm_id(id_priv->cm_id.ib);
2210                 id_priv->cm_id.ib = NULL;
2211         }
2212
2213         kfree(private_data);
2214         return ret;
2215 }
2216
2217 static int cma_connect_iw(struct rdma_id_private *id_priv,
2218                           struct rdma_conn_param *conn_param)
2219 {
2220         struct iw_cm_id *cm_id;
2221         struct sockaddr_in* sin;
2222         int ret;
2223         struct iw_cm_conn_param iw_param;
2224
2225         cm_id = iw_create_cm_id(id_priv->id.device, cma_iw_handler, id_priv);
2226         if (IS_ERR(cm_id)) {
2227                 ret = PTR_ERR(cm_id);
2228                 goto out;
2229         }
2230
2231         id_priv->cm_id.iw = cm_id;
2232
2233         sin = (struct sockaddr_in*) &id_priv->id.route.addr.src_addr;
2234         cm_id->local_addr = *sin;
2235
2236         sin = (struct sockaddr_in*) &id_priv->id.route.addr.dst_addr;
2237         cm_id->remote_addr = *sin;
2238
2239         ret = cma_modify_qp_rtr(&id_priv->id);
2240         if (ret)
2241                 goto out;
2242
2243         iw_param.ord = conn_param->initiator_depth;
2244         iw_param.ird = conn_param->responder_resources;
2245         iw_param.private_data = conn_param->private_data;
2246         iw_param.private_data_len = conn_param->private_data_len;
2247         if (id_priv->id.qp)
2248                 iw_param.qpn = id_priv->qp_num;
2249         else
2250                 iw_param.qpn = conn_param->qp_num;
2251         ret = iw_cm_connect(cm_id, &iw_param);
2252 out:
2253         if (ret && !IS_ERR(cm_id)) {
2254                 iw_destroy_cm_id(cm_id);
2255                 id_priv->cm_id.iw = NULL;
2256         }
2257         return ret;
2258 }
2259
2260 int rdma_connect(struct rdma_cm_id *id, struct rdma_conn_param *conn_param)
2261 {
2262         struct rdma_id_private *id_priv;
2263         int ret;
2264
2265         id_priv = container_of(id, struct rdma_id_private, id);
2266         if (!cma_comp_exch(id_priv, CMA_ROUTE_RESOLVED, CMA_CONNECT))
2267                 return -EINVAL;
2268
2269         if (!id->qp) {
2270                 id_priv->qp_num = conn_param->qp_num;
2271                 id_priv->srq = conn_param->srq;
2272         }
2273
2274         switch (rdma_node_get_transport(id->device->node_type)) {
2275         case RDMA_TRANSPORT_IB:
2276                 if (cma_is_ud_ps(id->ps))
2277                         ret = cma_resolve_ib_udp(id_priv, conn_param);
2278                 else
2279                         ret = cma_connect_ib(id_priv, conn_param);
2280                 break;
2281         case RDMA_TRANSPORT_IWARP:
2282                 ret = cma_connect_iw(id_priv, conn_param);
2283                 break;
2284         default:
2285                 ret = -ENOSYS;
2286                 break;
2287         }
2288         if (ret)
2289                 goto err;
2290
2291         return 0;
2292 err:
2293         cma_comp_exch(id_priv, CMA_CONNECT, CMA_ROUTE_RESOLVED);
2294         return ret;
2295 }
2296 EXPORT_SYMBOL(rdma_connect);
2297
2298 static int cma_accept_ib(struct rdma_id_private *id_priv,
2299                          struct rdma_conn_param *conn_param)
2300 {
2301         struct ib_cm_rep_param rep;
2302         struct ib_qp_attr qp_attr;
2303         int qp_attr_mask, ret;
2304
2305         if (id_priv->id.qp) {
2306                 ret = cma_modify_qp_rtr(&id_priv->id);
2307                 if (ret)
2308                         goto out;
2309
2310                 qp_attr.qp_state = IB_QPS_RTS;
2311                 ret = ib_cm_init_qp_attr(id_priv->cm_id.ib, &qp_attr,
2312                                          &qp_attr_mask);
2313                 if (ret)
2314                         goto out;
2315
2316                 qp_attr.max_rd_atomic = conn_param->initiator_depth;
2317                 ret = ib_modify_qp(id_priv->id.qp, &qp_attr, qp_attr_mask);
2318                 if (ret)
2319                         goto out;
2320         }
2321
2322         memset(&rep, 0, sizeof rep);
2323         rep.qp_num = id_priv->qp_num;
2324         rep.starting_psn = id_priv->seq_num;
2325         rep.private_data = conn_param->private_data;
2326         rep.private_data_len = conn_param->private_data_len;
2327         rep.responder_resources = conn_param->responder_resources;
2328         rep.initiator_depth = conn_param->initiator_depth;
2329         rep.failover_accepted = 0;
2330         rep.flow_control = conn_param->flow_control;
2331         rep.rnr_retry_count = conn_param->rnr_retry_count;
2332         rep.srq = id_priv->srq ? 1 : 0;
2333
2334         ret = ib_send_cm_rep(id_priv->cm_id.ib, &rep);
2335 out:
2336         return ret;
2337 }
2338
2339 static int cma_accept_iw(struct rdma_id_private *id_priv,
2340                   struct rdma_conn_param *conn_param)
2341 {
2342         struct iw_cm_conn_param iw_param;
2343         int ret;
2344
2345         ret = cma_modify_qp_rtr(&id_priv->id);
2346         if (ret)
2347                 return ret;
2348
2349         iw_param.ord = conn_param->initiator_depth;
2350         iw_param.ird = conn_param->responder_resources;
2351         iw_param.private_data = conn_param->private_data;
2352         iw_param.private_data_len = conn_param->private_data_len;
2353         if (id_priv->id.qp) {
2354                 iw_param.qpn = id_priv->qp_num;
2355         } else
2356                 iw_param.qpn = conn_param->qp_num;
2357
2358         return iw_cm_accept(id_priv->cm_id.iw, &iw_param);
2359 }
2360
2361 static int cma_send_sidr_rep(struct rdma_id_private *id_priv,
2362                              enum ib_cm_sidr_status status,
2363                              const void *private_data, int private_data_len)
2364 {
2365         struct ib_cm_sidr_rep_param rep;
2366
2367         memset(&rep, 0, sizeof rep);
2368         rep.status = status;
2369         if (status == IB_SIDR_SUCCESS) {
2370                 rep.qp_num = id_priv->qp_num;
2371                 rep.qkey = id_priv->qkey;
2372         }
2373         rep.private_data = private_data;
2374         rep.private_data_len = private_data_len;
2375
2376         return ib_send_cm_sidr_rep(id_priv->cm_id.ib, &rep);
2377 }
2378
2379 int rdma_accept(struct rdma_cm_id *id, struct rdma_conn_param *conn_param)
2380 {
2381         struct rdma_id_private *id_priv;
2382         int ret;
2383
2384         id_priv = container_of(id, struct rdma_id_private, id);
2385         if (!cma_comp(id_priv, CMA_CONNECT))
2386                 return -EINVAL;
2387
2388         if (!id->qp && conn_param) {
2389                 id_priv->qp_num = conn_param->qp_num;
2390                 id_priv->srq = conn_param->srq;
2391         }
2392
2393         switch (rdma_node_get_transport(id->device->node_type)) {
2394         case RDMA_TRANSPORT_IB:
2395                 if (cma_is_ud_ps(id->ps))
2396                         ret = cma_send_sidr_rep(id_priv, IB_SIDR_SUCCESS,
2397                                                 conn_param->private_data,
2398                                                 conn_param->private_data_len);
2399                 else if (conn_param)
2400                         ret = cma_accept_ib(id_priv, conn_param);
2401                 else
2402                         ret = cma_rep_recv(id_priv);
2403                 break;
2404         case RDMA_TRANSPORT_IWARP:
2405                 ret = cma_accept_iw(id_priv, conn_param);
2406                 break;
2407         default:
2408                 ret = -ENOSYS;
2409                 break;
2410         }
2411
2412         if (ret)
2413                 goto reject;
2414
2415         return 0;
2416 reject:
2417         cma_modify_qp_err(id);
2418         rdma_reject(id, NULL, 0);
2419         return ret;
2420 }
2421 EXPORT_SYMBOL(rdma_accept);
2422
2423 int rdma_notify(struct rdma_cm_id *id, enum ib_event_type event)
2424 {
2425         struct rdma_id_private *id_priv;
2426         int ret;
2427
2428         id_priv = container_of(id, struct rdma_id_private, id);
2429         if (!cma_has_cm_dev(id_priv))
2430                 return -EINVAL;
2431
2432         switch (id->device->node_type) {
2433         case RDMA_NODE_IB_CA:
2434                 ret = ib_cm_notify(id_priv->cm_id.ib, event);
2435                 break;
2436         default:
2437                 ret = 0;
2438                 break;
2439         }
2440         return ret;
2441 }
2442 EXPORT_SYMBOL(rdma_notify);
2443
2444 int rdma_reject(struct rdma_cm_id *id, const void *private_data,
2445                 u8 private_data_len)
2446 {
2447         struct rdma_id_private *id_priv;
2448         int ret;
2449
2450         id_priv = container_of(id, struct rdma_id_private, id);
2451         if (!cma_has_cm_dev(id_priv))
2452                 return -EINVAL;
2453
2454         switch (rdma_node_get_transport(id->device->node_type)) {
2455         case RDMA_TRANSPORT_IB:
2456                 if (cma_is_ud_ps(id->ps))
2457                         ret = cma_send_sidr_rep(id_priv, IB_SIDR_REJECT,
2458                                                 private_data, private_data_len);
2459                 else
2460                         ret = ib_send_cm_rej(id_priv->cm_id.ib,
2461                                              IB_CM_REJ_CONSUMER_DEFINED, NULL,
2462                                              0, private_data, private_data_len);
2463                 break;
2464         case RDMA_TRANSPORT_IWARP:
2465                 ret = iw_cm_reject(id_priv->cm_id.iw,
2466                                    private_data, private_data_len);
2467                 break;
2468         default:
2469                 ret = -ENOSYS;
2470                 break;
2471         }
2472         return ret;
2473 }
2474 EXPORT_SYMBOL(rdma_reject);
2475
2476 int rdma_disconnect(struct rdma_cm_id *id)
2477 {
2478         struct rdma_id_private *id_priv;
2479         int ret;
2480
2481         id_priv = container_of(id, struct rdma_id_private, id);
2482         if (!cma_has_cm_dev(id_priv))
2483                 return -EINVAL;
2484
2485         switch (rdma_node_get_transport(id->device->node_type)) {
2486         case RDMA_TRANSPORT_IB:
2487                 ret = cma_modify_qp_err(id);
2488                 if (ret)
2489                         goto out;
2490                 /* Initiate or respond to a disconnect. */
2491                 if (ib_send_cm_dreq(id_priv->cm_id.ib, NULL, 0))
2492                         ib_send_cm_drep(id_priv->cm_id.ib, NULL, 0);
2493                 break;
2494         case RDMA_TRANSPORT_IWARP:
2495                 ret = iw_cm_disconnect(id_priv->cm_id.iw, 0);
2496                 break;
2497         default:
2498                 ret = -EINVAL;
2499                 break;
2500         }
2501 out:
2502         return ret;
2503 }
2504 EXPORT_SYMBOL(rdma_disconnect);
2505
2506 static int cma_ib_mc_handler(int status, struct ib_sa_multicast *multicast)
2507 {
2508         struct rdma_id_private *id_priv;
2509         struct cma_multicast *mc = multicast->context;
2510         struct rdma_cm_event event;
2511         int ret;
2512
2513         id_priv = mc->id_priv;
2514         if (cma_disable_remove(id_priv, CMA_ADDR_BOUND) &&
2515             cma_disable_remove(id_priv, CMA_ADDR_RESOLVED))
2516                 return 0;
2517
2518         if (!status && id_priv->id.qp)
2519                 status = ib_attach_mcast(id_priv->id.qp, &multicast->rec.mgid,
2520                                          multicast->rec.mlid);
2521
2522         memset(&event, 0, sizeof event);
2523         event.status = status;
2524         event.param.ud.private_data = mc->context;
2525         if (!status) {
2526                 event.event = RDMA_CM_EVENT_MULTICAST_JOIN;
2527                 ib_init_ah_from_mcmember(id_priv->id.device,
2528                                          id_priv->id.port_num, &multicast->rec,
2529                                          &event.param.ud.ah_attr);
2530                 event.param.ud.qp_num = 0xFFFFFF;
2531                 event.param.ud.qkey = be32_to_cpu(multicast->rec.qkey);
2532         } else
2533                 event.event = RDMA_CM_EVENT_MULTICAST_ERROR;
2534
2535         ret = id_priv->id.event_handler(&id_priv->id, &event);
2536         if (ret) {
2537                 cma_exch(id_priv, CMA_DESTROYING);
2538                 cma_enable_remove(id_priv);
2539                 rdma_destroy_id(&id_priv->id);
2540                 return 0;
2541         }
2542
2543         cma_enable_remove(id_priv);
2544         return 0;
2545 }
2546
2547 static void cma_set_mgid(struct rdma_id_private *id_priv,
2548                          struct sockaddr *addr, union ib_gid *mgid)
2549 {
2550         unsigned char mc_map[MAX_ADDR_LEN];
2551         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
2552         struct sockaddr_in *sin = (struct sockaddr_in *) addr;
2553         struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) addr;
2554
2555         if (cma_any_addr(addr)) {
2556                 memset(mgid, 0, sizeof *mgid);
2557         } else if ((addr->sa_family == AF_INET6) &&
2558                    ((be32_to_cpu(sin6->sin6_addr.s6_addr32[0]) & 0xFF10A01B) ==
2559                                                                  0xFF10A01B)) {
2560                 /* IPv6 address is an SA assigned MGID. */
2561                 memcpy(mgid, &sin6->sin6_addr, sizeof *mgid);
2562         } else {
2563                 ip_ib_mc_map(sin->sin_addr.s_addr, mc_map);
2564                 if (id_priv->id.ps == RDMA_PS_UDP)
2565                         mc_map[7] = 0x01;       /* Use RDMA CM signature */
2566                 mc_map[8] = ib_addr_get_pkey(dev_addr) >> 8;
2567                 mc_map[9] = (unsigned char) ib_addr_get_pkey(dev_addr);
2568                 *mgid = *(union ib_gid *) (mc_map + 4);
2569         }
2570 }
2571
2572 static int cma_join_ib_multicast(struct rdma_id_private *id_priv,
2573                                  struct cma_multicast *mc)
2574 {
2575         struct ib_sa_mcmember_rec rec;
2576         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
2577         ib_sa_comp_mask comp_mask;
2578         int ret;
2579
2580         ib_addr_get_mgid(dev_addr, &rec.mgid);
2581         ret = ib_sa_get_mcmember_rec(id_priv->id.device, id_priv->id.port_num,
2582                                      &rec.mgid, &rec);
2583         if (ret)
2584                 return ret;
2585
2586         cma_set_mgid(id_priv, &mc->addr, &rec.mgid);
2587         if (id_priv->id.ps == RDMA_PS_UDP)
2588                 rec.qkey = cpu_to_be32(RDMA_UDP_QKEY);
2589         ib_addr_get_sgid(dev_addr, &rec.port_gid);
2590         rec.pkey = cpu_to_be16(ib_addr_get_pkey(dev_addr));
2591         rec.join_state = 1;
2592
2593         comp_mask = IB_SA_MCMEMBER_REC_MGID | IB_SA_MCMEMBER_REC_PORT_GID |
2594                     IB_SA_MCMEMBER_REC_PKEY | IB_SA_MCMEMBER_REC_JOIN_STATE |
2595                     IB_SA_MCMEMBER_REC_QKEY | IB_SA_MCMEMBER_REC_SL |
2596                     IB_SA_MCMEMBER_REC_FLOW_LABEL |
2597                     IB_SA_MCMEMBER_REC_TRAFFIC_CLASS;
2598
2599         mc->multicast.ib = ib_sa_join_multicast(&sa_client, id_priv->id.device,
2600                                                 id_priv->id.port_num, &rec,
2601                                                 comp_mask, GFP_KERNEL,
2602                                                 cma_ib_mc_handler, mc);
2603         if (IS_ERR(mc->multicast.ib))
2604                 return PTR_ERR(mc->multicast.ib);
2605
2606         return 0;
2607 }
2608
2609 int rdma_join_multicast(struct rdma_cm_id *id, struct sockaddr *addr,
2610                         void *context)
2611 {
2612         struct rdma_id_private *id_priv;
2613         struct cma_multicast *mc;
2614         int ret;
2615
2616         id_priv = container_of(id, struct rdma_id_private, id);
2617         if (!cma_comp(id_priv, CMA_ADDR_BOUND) &&
2618             !cma_comp(id_priv, CMA_ADDR_RESOLVED))
2619                 return -EINVAL;
2620
2621         mc = kmalloc(sizeof *mc, GFP_KERNEL);
2622         if (!mc)
2623                 return -ENOMEM;
2624
2625         memcpy(&mc->addr, addr, ip_addr_size(addr));
2626         mc->context = context;
2627         mc->id_priv = id_priv;
2628
2629         spin_lock(&id_priv->lock);
2630         list_add(&mc->list, &id_priv->mc_list);
2631         spin_unlock(&id_priv->lock);
2632
2633         switch (rdma_node_get_transport(id->device->node_type)) {
2634         case RDMA_TRANSPORT_IB:
2635                 ret = cma_join_ib_multicast(id_priv, mc);
2636                 break;
2637         default:
2638                 ret = -ENOSYS;
2639                 break;
2640         }
2641
2642         if (ret) {
2643                 spin_lock_irq(&id_priv->lock);
2644                 list_del(&mc->list);
2645                 spin_unlock_irq(&id_priv->lock);
2646                 kfree(mc);
2647         }
2648         return ret;
2649 }
2650 EXPORT_SYMBOL(rdma_join_multicast);
2651
2652 void rdma_leave_multicast(struct rdma_cm_id *id, struct sockaddr *addr)
2653 {
2654         struct rdma_id_private *id_priv;
2655         struct cma_multicast *mc;
2656
2657         id_priv = container_of(id, struct rdma_id_private, id);
2658         spin_lock_irq(&id_priv->lock);
2659         list_for_each_entry(mc, &id_priv->mc_list, list) {
2660                 if (!memcmp(&mc->addr, addr, ip_addr_size(addr))) {
2661                         list_del(&mc->list);
2662                         spin_unlock_irq(&id_priv->lock);
2663
2664                         if (id->qp)
2665                                 ib_detach_mcast(id->qp,
2666                                                 &mc->multicast.ib->rec.mgid,
2667                                                 mc->multicast.ib->rec.mlid);
2668                         ib_sa_free_multicast(mc->multicast.ib);
2669                         kfree(mc);
2670                         return;
2671                 }
2672         }
2673         spin_unlock_irq(&id_priv->lock);
2674 }
2675 EXPORT_SYMBOL(rdma_leave_multicast);
2676
2677 static void cma_add_one(struct ib_device *device)
2678 {
2679         struct cma_device *cma_dev;
2680         struct rdma_id_private *id_priv;
2681
2682         cma_dev = kmalloc(sizeof *cma_dev, GFP_KERNEL);
2683         if (!cma_dev)
2684                 return;
2685
2686         cma_dev->device = device;
2687
2688         init_completion(&cma_dev->comp);
2689         atomic_set(&cma_dev->refcount, 1);
2690         INIT_LIST_HEAD(&cma_dev->id_list);
2691         ib_set_client_data(device, &cma_client, cma_dev);
2692
2693         mutex_lock(&lock);
2694         list_add_tail(&cma_dev->list, &dev_list);
2695         list_for_each_entry(id_priv, &listen_any_list, list)
2696                 cma_listen_on_dev(id_priv, cma_dev);
2697         mutex_unlock(&lock);
2698 }
2699
2700 static int cma_remove_id_dev(struct rdma_id_private *id_priv)
2701 {
2702         struct rdma_cm_event event;
2703         enum cma_state state;
2704
2705         /* Record that we want to remove the device */
2706         state = cma_exch(id_priv, CMA_DEVICE_REMOVAL);
2707         if (state == CMA_DESTROYING)
2708                 return 0;
2709
2710         cma_cancel_operation(id_priv, state);
2711         wait_event(id_priv->wait_remove, !atomic_read(&id_priv->dev_remove));
2712
2713         /* Check for destruction from another callback. */
2714         if (!cma_comp(id_priv, CMA_DEVICE_REMOVAL))
2715                 return 0;
2716
2717         memset(&event, 0, sizeof event);
2718         event.event = RDMA_CM_EVENT_DEVICE_REMOVAL;
2719         return id_priv->id.event_handler(&id_priv->id, &event);
2720 }
2721
2722 static void cma_process_remove(struct cma_device *cma_dev)
2723 {
2724         struct rdma_id_private *id_priv;
2725         int ret;
2726
2727         mutex_lock(&lock);
2728         while (!list_empty(&cma_dev->id_list)) {
2729                 id_priv = list_entry(cma_dev->id_list.next,
2730                                      struct rdma_id_private, list);
2731
2732                 if (cma_internal_listen(id_priv)) {
2733                         cma_destroy_listen(id_priv);
2734                         continue;
2735                 }
2736
2737                 list_del_init(&id_priv->list);
2738                 atomic_inc(&id_priv->refcount);
2739                 mutex_unlock(&lock);
2740
2741                 ret = cma_remove_id_dev(id_priv);
2742                 cma_deref_id(id_priv);
2743                 if (ret)
2744                         rdma_destroy_id(&id_priv->id);
2745
2746                 mutex_lock(&lock);
2747         }
2748         mutex_unlock(&lock);
2749
2750         cma_deref_dev(cma_dev);
2751         wait_for_completion(&cma_dev->comp);
2752 }
2753
2754 static void cma_remove_one(struct ib_device *device)
2755 {
2756         struct cma_device *cma_dev;
2757
2758         cma_dev = ib_get_client_data(device, &cma_client);
2759         if (!cma_dev)
2760                 return;
2761
2762         mutex_lock(&lock);
2763         list_del(&cma_dev->list);
2764         mutex_unlock(&lock);
2765
2766         cma_process_remove(cma_dev);
2767         kfree(cma_dev);
2768 }
2769
2770 static int cma_init(void)
2771 {
2772         int ret;
2773
2774         get_random_bytes(&next_port, sizeof next_port);
2775         next_port = ((unsigned int) next_port %
2776                     (sysctl_local_port_range[1] - sysctl_local_port_range[0])) +
2777                     sysctl_local_port_range[0];
2778         cma_wq = create_singlethread_workqueue("rdma_cm");
2779         if (!cma_wq)
2780                 return -ENOMEM;
2781
2782         ib_sa_register_client(&sa_client);
2783         rdma_addr_register_client(&addr_client);
2784
2785         ret = ib_register_client(&cma_client);
2786         if (ret)
2787                 goto err;
2788         return 0;
2789
2790 err:
2791         rdma_addr_unregister_client(&addr_client);
2792         ib_sa_unregister_client(&sa_client);
2793         destroy_workqueue(cma_wq);
2794         return ret;
2795 }
2796
2797 static void cma_cleanup(void)
2798 {
2799         ib_unregister_client(&cma_client);
2800         rdma_addr_unregister_client(&addr_client);
2801         ib_sa_unregister_client(&sa_client);
2802         destroy_workqueue(cma_wq);
2803         idr_destroy(&sdp_ps);
2804         idr_destroy(&tcp_ps);
2805         idr_destroy(&udp_ps);
2806         idr_destroy(&ipoib_ps);
2807 }
2808
2809 module_init(cma_init);
2810 module_exit(cma_cleanup);