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 available to you under a choice of one of two
   8  * licenses.  You may choose to be licensed under the terms of the GNU
   9  * General Public License (GPL) Version 2, available from the file
  10  * COPYING in the main directory of this source tree, or the
  11  * OpenIB.org BSD license below:
  12  *
  13  *     Redistribution and use in source and binary forms, with or
  14  *     without modification, are permitted provided that the following
  15  *     conditions are met:
  16  *
  17  *      - Redistributions of source code must retain the above
  18  *        copyright notice, this list of conditions and the following
  19  *        disclaimer.
  20  *
  21  *      - Redistributions in binary form must reproduce the above
  22  *        copyright notice, this list of conditions and the following
  23  *        disclaimer in the documentation and/or other materials
  24  *        provided with the distribution.
  25  *
  26  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  27  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  28  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  29  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  30  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  31  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  32  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  33  * SOFTWARE.
  34  */
  35
  36 #include <linux/completion.h>
  37 #include <linux/in.h>
  38 #include <linux/in6.h>
  39 #include <linux/mutex.h>
  40 #include <linux/random.h>
  41 #include <linux/idr.h>
  42 #include <linux/inetdevice.h>
  43 #include <linux/slab.h>
  44 #include <linux/module.h>
  45 #include <net/route.h>
  46
  47 #include <net/tcp.h>
  48 #include <net/ipv6.h>
  49
  50 #include <rdma/rdma_cm.h>
  51 #include <rdma/rdma_cm_ib.h>
  52 #include <rdma/rdma_netlink.h>
  53 #include <rdma/ib.h>
  54 #include <rdma/ib_cache.h>
  55 #include <rdma/ib_cm.h>
  56 #include <rdma/ib_sa.h>
  57 #include <rdma/iw_cm.h>
  58
  59 MODULE_AUTHOR("Sean Hefty");
  60 MODULE_DESCRIPTION("Generic RDMA CM Agent");
  61 MODULE_LICENSE("Dual BSD/GPL");
  62
  63 #define CMA_CM_RESPONSE_TIMEOUT 20
  64 #define CMA_MAX_CM_RETRIES 15
  65 #define CMA_CM_MRA_SETTING (IB_CM_MRA_FLAG_DELAY | 24)
  66 #define CMA_IBOE_PACKET_LIFETIME 18
  67
  68 static void cma_add_one(struct ib_device *device);
  69 static void cma_remove_one(struct ib_device *device);
  70
  71 static struct ib_client cma_client = {
  72         .name   = "cma",
  73         .add    = cma_add_one,
  74         .remove = cma_remove_one
  75 };
  76
  77 static struct ib_sa_client sa_client;
  78 static struct rdma_addr_client addr_client;
  79 static LIST_HEAD(dev_list);
  80 static LIST_HEAD(listen_any_list);
  81 static DEFINE_MUTEX(lock);
  82 static struct workqueue_struct *cma_wq;
  83 static DEFINE_IDR(tcp_ps);
  84 static DEFINE_IDR(udp_ps);
  85 static DEFINE_IDR(ipoib_ps);
  86 static DEFINE_IDR(ib_ps);
  87
  88 struct cma_device {
  89         struct list_head        list;
  90         struct ib_device        *device;
  91         struct completion       comp;
  92         atomic_t                refcount;
  93         struct list_head        id_list;
  94 };
  95
  96 struct rdma_bind_list {
  97         struct idr              *ps;
  98         struct hlist_head       owners;
  99         unsigned short          port;
 100 };
 101
 102 enum {
 103         CMA_OPTION_AFONLY,
 104 };
 105
 106 /*
 107  * Device removal can occur at anytime, so we need extra handling to
 108  * serialize notifying the user of device removal with other callbacks.
 109  * We do this by disabling removal notification while a callback is in process,
 110  * and reporting it after the callback completes.
 111  */
 112 struct rdma_id_private {
 113         struct rdma_cm_id       id;
 114
 115         struct rdma_bind_list   *bind_list;
 116         struct hlist_node       node;
 117         struct list_head        list; /* listen_any_list or cma_device.list */
 118         struct list_head        listen_list; /* per device listens */
 119         struct cma_device       *cma_dev;
 120         struct list_head        mc_list;
 121
 122         int                     internal_id;
 123         enum rdma_cm_state      state;
 124         spinlock_t              lock;
 125         struct mutex            qp_mutex;
 126
 127         struct completion       comp;
 128         atomic_t                refcount;
 129         struct mutex            handler_mutex;
 130
 131         int                     backlog;
 132         int                     timeout_ms;
 133         struct ib_sa_query      *query;
 134         int                     query_id;
 135         union {
 136                 struct ib_cm_id *ib;
 137                 struct iw_cm_id *iw;
 138         } cm_id;
 139
 140         u32                     seq_num;
 141         u32                     qkey;
 142         u32                     qp_num;
 143         pid_t                   owner;
 144         u32                     options;
 145         u8                      srq;
 146         u8                      tos;
 147         u8                      reuseaddr;
 148         u8                      afonly;
 149 };
 150
 151 struct cma_multicast {
 152         struct rdma_id_private *id_priv;
 153         union {
 154                 struct ib_sa_multicast *ib;
 155         } multicast;
 156         struct list_head        list;
 157         void                    *context;
 158         struct sockaddr_storage addr;
 159         struct kref             mcref;
 160 };
 161
 162 struct cma_work {
 163         struct work_struct      work;
 164         struct rdma_id_private  *id;
 165         enum rdma_cm_state      old_state;
 166         enum rdma_cm_state      new_state;
 167         struct rdma_cm_event    event;
 168 };
 169
 170 struct cma_ndev_work {
 171         struct work_struct      work;
 172         struct rdma_id_private  *id;
 173         struct rdma_cm_event    event;
 174 };
 175
 176 struct iboe_mcast_work {
 177         struct work_struct       work;
 178         struct rdma_id_private  *id;
 179         struct cma_multicast    *mc;
 180 };
 181
 182 union cma_ip_addr {
 183         struct in6_addr ip6;
 184         struct {
 185                 __be32 pad[3];
 186                 __be32 addr;
 187         } ip4;
 188 };
 189
 190 struct cma_hdr {
 191         u8 cma_version;
 192         u8 ip_version;  /* IP version: 7:4 */
 193         __be16 port;
 194         union cma_ip_addr src_addr;
 195         union cma_ip_addr dst_addr;
 196 };
 197
 198 #define CMA_VERSION 0x00
 199
 200 static int cma_comp(struct rdma_id_private *id_priv, enum rdma_cm_state comp)
 201 {
 202         unsigned long flags;
 203         int ret;
 204
 205         spin_lock_irqsave(&id_priv->lock, flags);
 206         ret = (id_priv->state == comp);
 207         spin_unlock_irqrestore(&id_priv->lock, flags);
 208         return ret;
 209 }
 210
 211 static int cma_comp_exch(struct rdma_id_private *id_priv,
 212                          enum rdma_cm_state comp, enum rdma_cm_state exch)
 213 {
 214         unsigned long flags;
 215         int ret;
 216
 217         spin_lock_irqsave(&id_priv->lock, flags);
 218         if ((ret = (id_priv->state == comp)))
 219                 id_priv->state = exch;
 220         spin_unlock_irqrestore(&id_priv->lock, flags);
 221         return ret;
 222 }
 223
 224 static enum rdma_cm_state cma_exch(struct rdma_id_private *id_priv,
 225                                    enum rdma_cm_state exch)
 226 {
 227         unsigned long flags;
 228         enum rdma_cm_state old;
 229
 230         spin_lock_irqsave(&id_priv->lock, flags);
 231         old = id_priv->state;
 232         id_priv->state = exch;
 233         spin_unlock_irqrestore(&id_priv->lock, flags);
 234         return old;
 235 }
 236
 237 static inline u8 cma_get_ip_ver(struct cma_hdr *hdr)
 238 {
 239         return hdr->ip_version >> 4;
 240 }
 241
 242 static inline void cma_set_ip_ver(struct cma_hdr *hdr, u8 ip_ver)
 243 {
 244         hdr->ip_version = (ip_ver << 4) | (hdr->ip_version & 0xF);
 245 }
 246
 247 static void cma_attach_to_dev(struct rdma_id_private *id_priv,
 248                               struct cma_device *cma_dev)
 249 {
 250         atomic_inc(&cma_dev->refcount);
 251         id_priv->cma_dev = cma_dev;
 252         id_priv->id.device = cma_dev->device;
 253         id_priv->id.route.addr.dev_addr.transport =
 254                 rdma_node_get_transport(cma_dev->device->node_type);
 255         list_add_tail(&id_priv->list, &cma_dev->id_list);
 256 }
 257
 258 static inline void cma_deref_dev(struct cma_device *cma_dev)
 259 {
 260         if (atomic_dec_and_test(&cma_dev->refcount))
 261                 complete(&cma_dev->comp);
 262 }
 263
 264 static inline void release_mc(struct kref *kref)
 265 {
 266         struct cma_multicast *mc = container_of(kref, struct cma_multicast, mcref);
 267
 268         kfree(mc->multicast.ib);
 269         kfree(mc);
 270 }
 271
 272 static void cma_release_dev(struct rdma_id_private *id_priv)
 273 {
 274         mutex_lock(&lock);
 275         list_del(&id_priv->list);
 276         cma_deref_dev(id_priv->cma_dev);
 277         id_priv->cma_dev = NULL;
 278         mutex_unlock(&lock);
 279 }
 280
 281 static inline struct sockaddr *cma_src_addr(struct rdma_id_private *id_priv)
 282 {
 283         return (struct sockaddr *) &id_priv->id.route.addr.src_addr;
 284 }
 285
 286 static inline struct sockaddr *cma_dst_addr(struct rdma_id_private *id_priv)
 287 {
 288         return (struct sockaddr *) &id_priv->id.route.addr.dst_addr;
 289 }
 290
 291 static inline unsigned short cma_family(struct rdma_id_private *id_priv)
 292 {
 293         return id_priv->id.route.addr.src_addr.ss_family;
 294 }
 295
 296 static int cma_set_qkey(struct rdma_id_private *id_priv, u32 qkey)
 297 {
 298         struct ib_sa_mcmember_rec rec;
 299         int ret = 0;
 300
 301         if (id_priv->qkey) {
 302                 if (qkey && id_priv->qkey != qkey)
 303                         return -EINVAL;
 304                 return 0;
 305         }
 306
 307         if (qkey) {
 308                 id_priv->qkey = qkey;
 309                 return 0;
 310         }
 311
 312         switch (id_priv->id.ps) {
 313         case RDMA_PS_UDP:
 314         case RDMA_PS_IB:
 315                 id_priv->qkey = RDMA_UDP_QKEY;
 316                 break;
 317         case RDMA_PS_IPOIB:
 318                 ib_addr_get_mgid(&id_priv->id.route.addr.dev_addr, &rec.mgid);
 319                 ret = ib_sa_get_mcmember_rec(id_priv->id.device,
 320                                              id_priv->id.port_num, &rec.mgid,
 321                                              &rec);
 322                 if (!ret)
 323                         id_priv->qkey = be32_to_cpu(rec.qkey);
 324                 break;
 325         default:
 326                 break;
 327         }
 328         return ret;
 329 }
 330
 331 static int find_gid_port(struct ib_device *device, union ib_gid *gid, u8 port_num)
 332 {
 333         int i;
 334         int err;
 335         struct ib_port_attr props;
 336         union ib_gid tmp;
 337
 338         err = ib_query_port(device, port_num, &props);
 339         if (err)
 340                 return err;
 341
 342         for (i = 0; i < props.gid_tbl_len; ++i) {
 343                 err = ib_query_gid(device, port_num, i, &tmp);
 344                 if (err)
 345                         return err;
 346                 if (!memcmp(&tmp, gid, sizeof tmp))
 347                         return 0;
 348         }
 349
 350         return -EADDRNOTAVAIL;
 351 }
 352
 353 static void cma_translate_ib(struct sockaddr_ib *sib, struct rdma_dev_addr *dev_addr)
 354 {
 355         dev_addr->dev_type = ARPHRD_INFINIBAND;
 356         rdma_addr_set_sgid(dev_addr, (union ib_gid *) &sib->sib_addr);
 357         ib_addr_set_pkey(dev_addr, ntohs(sib->sib_pkey));
 358 }
 359
 360 static int cma_translate_addr(struct sockaddr *addr, struct rdma_dev_addr *dev_addr)
 361 {
 362         int ret;
 363
 364         if (addr->sa_family != AF_IB) {
 365                 ret = rdma_translate_ip(addr, dev_addr);
 366         } else {
 367                 cma_translate_ib((struct sockaddr_ib *) addr, dev_addr);
 368                 ret = 0;
 369         }
 370
 371         return ret;
 372 }
 373
 374 static int cma_acquire_dev(struct rdma_id_private *id_priv)
 375 {
 376         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
 377         struct cma_device *cma_dev;
 378         union ib_gid gid, iboe_gid;
 379         int ret = -ENODEV;
 380         u8 port;
 381         enum rdma_link_layer dev_ll = dev_addr->dev_type == ARPHRD_INFINIBAND ?
 382                 IB_LINK_LAYER_INFINIBAND : IB_LINK_LAYER_ETHERNET;
 383
 384         if (dev_ll != IB_LINK_LAYER_INFINIBAND &&
 385             id_priv->id.ps == RDMA_PS_IPOIB)
 386                 return -EINVAL;
 387
 388         mutex_lock(&lock);
 389         iboe_addr_get_sgid(dev_addr, &iboe_gid);
 390         memcpy(&gid, dev_addr->src_dev_addr +
 391                rdma_addr_gid_offset(dev_addr), sizeof gid);
 392         list_for_each_entry(cma_dev, &dev_list, list) {
 393                 for (port = 1; port <= cma_dev->device->phys_port_cnt; ++port) {
 394                         if (rdma_port_get_link_layer(cma_dev->device, port) == dev_ll) {
 395                                 if (rdma_node_get_transport(cma_dev->device->node_type) == RDMA_TRANSPORT_IB &&
 396                                     rdma_port_get_link_layer(cma_dev->device, port) == IB_LINK_LAYER_ETHERNET)
 397                                         ret = find_gid_port(cma_dev->device, &iboe_gid, port);
 398                                 else
 399                                         ret = find_gid_port(cma_dev->device, &gid, port);
 400
 401                                 if (!ret) {
 402                                         id_priv->id.port_num = port;
 403                                         goto out;
 404                                 }
 405                         }
 406                 }
 407         }
 408
 409 out:
 410         if (!ret)
 411                 cma_attach_to_dev(id_priv, cma_dev);
 412
 413         mutex_unlock(&lock);
 414         return ret;
 415 }
 416
 417 /*
 418  * Select the source IB device and address to reach the destination IB address.
 419  */
 420 static int cma_resolve_ib_dev(struct rdma_id_private *id_priv)
 421 {
 422         struct cma_device *cma_dev, *cur_dev;
 423         struct sockaddr_ib *addr;
 424         union ib_gid gid, sgid, *dgid;
 425         u16 pkey, index;
 426         u8 p;
 427         int i;
 428
 429         cma_dev = NULL;
 430         addr = (struct sockaddr_ib *) cma_dst_addr(id_priv);
 431         dgid = (union ib_gid *) &addr->sib_addr;
 432         pkey = ntohs(addr->sib_pkey);
 433
 434         list_for_each_entry(cur_dev, &dev_list, list) {
 435                 if (rdma_node_get_transport(cur_dev->device->node_type) != RDMA_TRANSPORT_IB)
 436                         continue;
 437
 438                 for (p = 1; p <= cur_dev->device->phys_port_cnt; ++p) {
 439                         if (ib_find_cached_pkey(cur_dev->device, p, pkey, &index))
 440                                 continue;
 441
 442                         for (i = 0; !ib_get_cached_gid(cur_dev->device, p, i, &gid); i++) {
 443                                 if (!memcmp(&gid, dgid, sizeof(gid))) {
 444                                         cma_dev = cur_dev;
 445                                         sgid = gid;
 446                                         id_priv->id.port_num = p;
 447                                         goto found;
 448                                 }
 449
 450                                 if (!cma_dev && (gid.global.subnet_prefix ==
 451                                                  dgid->global.subnet_prefix)) {
 452                                         cma_dev = cur_dev;
 453                                         sgid = gid;
 454                                         id_priv->id.port_num = p;
 455                                 }
 456                         }
 457                 }
 458         }
 459
 460         if (!cma_dev)
 461                 return -ENODEV;
 462
 463 found:
 464         cma_attach_to_dev(id_priv, cma_dev);
 465         addr = (struct sockaddr_ib *) cma_src_addr(id_priv);
 466         memcpy(&addr->sib_addr, &sgid, sizeof sgid);
 467         cma_translate_ib(addr, &id_priv->id.route.addr.dev_addr);
 468         return 0;
 469 }
 470
 471 static void cma_deref_id(struct rdma_id_private *id_priv)
 472 {
 473         if (atomic_dec_and_test(&id_priv->refcount))
 474                 complete(&id_priv->comp);
 475 }
 476
 477 static int cma_disable_callback(struct rdma_id_private *id_priv,
 478                                 enum rdma_cm_state state)
 479 {
 480         mutex_lock(&id_priv->handler_mutex);
 481         if (id_priv->state != state) {
 482                 mutex_unlock(&id_priv->handler_mutex);
 483                 return -EINVAL;
 484         }
 485         return 0;
 486 }
 487
 488 struct rdma_cm_id *rdma_create_id(rdma_cm_event_handler event_handler,
 489                                   void *context, enum rdma_port_space ps,
 490                                   enum ib_qp_type qp_type)
 491 {
 492         struct rdma_id_private *id_priv;
 493
 494         id_priv = kzalloc(sizeof *id_priv, GFP_KERNEL);
 495         if (!id_priv)
 496                 return ERR_PTR(-ENOMEM);
 497
 498         id_priv->owner = task_pid_nr(current);
 499         id_priv->state = RDMA_CM_IDLE;
 500         id_priv->id.context = context;
 501         id_priv->id.event_handler = event_handler;
 502         id_priv->id.ps = ps;
 503         id_priv->id.qp_type = qp_type;
 504         spin_lock_init(&id_priv->lock);
 505         mutex_init(&id_priv->qp_mutex);
 506         init_completion(&id_priv->comp);
 507         atomic_set(&id_priv->refcount, 1);
 508         mutex_init(&id_priv->handler_mutex);
 509         INIT_LIST_HEAD(&id_priv->listen_list);
 510         INIT_LIST_HEAD(&id_priv->mc_list);
 511         get_random_bytes(&id_priv->seq_num, sizeof id_priv->seq_num);
 512
 513         return &id_priv->id;
 514 }
 515 EXPORT_SYMBOL(rdma_create_id);
 516
 517 static int cma_init_ud_qp(struct rdma_id_private *id_priv, struct ib_qp *qp)
 518 {
 519         struct ib_qp_attr qp_attr;
 520         int qp_attr_mask, ret;
 521
 522         qp_attr.qp_state = IB_QPS_INIT;
 523         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
 524         if (ret)
 525                 return ret;
 526
 527         ret = ib_modify_qp(qp, &qp_attr, qp_attr_mask);
 528         if (ret)
 529                 return ret;
 530
 531         qp_attr.qp_state = IB_QPS_RTR;
 532         ret = ib_modify_qp(qp, &qp_attr, IB_QP_STATE);
 533         if (ret)
 534                 return ret;
 535
 536         qp_attr.qp_state = IB_QPS_RTS;
 537         qp_attr.sq_psn = 0;
 538         ret = ib_modify_qp(qp, &qp_attr, IB_QP_STATE | IB_QP_SQ_PSN);
 539
 540         return ret;
 541 }
 542
 543 static int cma_init_conn_qp(struct rdma_id_private *id_priv, struct ib_qp *qp)
 544 {
 545         struct ib_qp_attr qp_attr;
 546         int qp_attr_mask, ret;
 547
 548         qp_attr.qp_state = IB_QPS_INIT;
 549         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
 550         if (ret)
 551                 return ret;
 552
 553         return ib_modify_qp(qp, &qp_attr, qp_attr_mask);
 554 }
 555
 556 int rdma_create_qp(struct rdma_cm_id *id, struct ib_pd *pd,
 557                    struct ib_qp_init_attr *qp_init_attr)
 558 {
 559         struct rdma_id_private *id_priv;
 560         struct ib_qp *qp;
 561         int ret;
 562
 563         id_priv = container_of(id, struct rdma_id_private, id);
 564         if (id->device != pd->device)
 565                 return -EINVAL;
 566
 567         qp = ib_create_qp(pd, qp_init_attr);
 568         if (IS_ERR(qp))
 569                 return PTR_ERR(qp);
 570
 571         if (id->qp_type == IB_QPT_UD)
 572                 ret = cma_init_ud_qp(id_priv, qp);
 573         else
 574                 ret = cma_init_conn_qp(id_priv, qp);
 575         if (ret)
 576                 goto err;
 577
 578         id->qp = qp;
 579         id_priv->qp_num = qp->qp_num;
 580         id_priv->srq = (qp->srq != NULL);
 581         return 0;
 582 err:
 583         ib_destroy_qp(qp);
 584         return ret;
 585 }
 586 EXPORT_SYMBOL(rdma_create_qp);
 587
 588 void rdma_destroy_qp(struct rdma_cm_id *id)
 589 {
 590         struct rdma_id_private *id_priv;
 591
 592         id_priv = container_of(id, struct rdma_id_private, id);
 593         mutex_lock(&id_priv->qp_mutex);
 594         ib_destroy_qp(id_priv->id.qp);
 595         id_priv->id.qp = NULL;
 596         mutex_unlock(&id_priv->qp_mutex);
 597 }
 598 EXPORT_SYMBOL(rdma_destroy_qp);
 599
 600 static int cma_modify_qp_rtr(struct rdma_id_private *id_priv,
 601                              struct rdma_conn_param *conn_param)
 602 {
 603         struct ib_qp_attr qp_attr;
 604         int qp_attr_mask, ret;
 605
 606         mutex_lock(&id_priv->qp_mutex);
 607         if (!id_priv->id.qp) {
 608                 ret = 0;
 609                 goto out;
 610         }
 611
 612         /* Need to update QP attributes from default values. */
 613         qp_attr.qp_state = IB_QPS_INIT;
 614         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
 615         if (ret)
 616                 goto out;
 617
 618         ret = ib_modify_qp(id_priv->id.qp, &qp_attr, qp_attr_mask);
 619         if (ret)
 620                 goto out;
 621
 622         qp_attr.qp_state = IB_QPS_RTR;
 623         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
 624         if (ret)
 625                 goto out;
 626
 627         if (conn_param)
 628                 qp_attr.max_dest_rd_atomic = conn_param->responder_resources;
 629         ret = ib_modify_qp(id_priv->id.qp, &qp_attr, qp_attr_mask);
 630 out:
 631         mutex_unlock(&id_priv->qp_mutex);
 632         return ret;
 633 }
 634
 635 static int cma_modify_qp_rts(struct rdma_id_private *id_priv,
 636                              struct rdma_conn_param *conn_param)
 637 {
 638         struct ib_qp_attr qp_attr;
 639         int qp_attr_mask, ret;
 640
 641         mutex_lock(&id_priv->qp_mutex);
 642         if (!id_priv->id.qp) {
 643                 ret = 0;
 644                 goto out;
 645         }
 646
 647         qp_attr.qp_state = IB_QPS_RTS;
 648         ret = rdma_init_qp_attr(&id_priv->id, &qp_attr, &qp_attr_mask);
 649         if (ret)
 650                 goto out;
 651
 652         if (conn_param)
 653                 qp_attr.max_rd_atomic = conn_param->initiator_depth;
 654         ret = ib_modify_qp(id_priv->id.qp, &qp_attr, qp_attr_mask);
 655 out:
 656         mutex_unlock(&id_priv->qp_mutex);
 657         return ret;
 658 }
 659
 660 static int cma_modify_qp_err(struct rdma_id_private *id_priv)
 661 {
 662         struct ib_qp_attr qp_attr;
 663         int ret;
 664
 665         mutex_lock(&id_priv->qp_mutex);
 666         if (!id_priv->id.qp) {
 667                 ret = 0;
 668                 goto out;
 669         }
 670
 671         qp_attr.qp_state = IB_QPS_ERR;
 672         ret = ib_modify_qp(id_priv->id.qp, &qp_attr, IB_QP_STATE);
 673 out:
 674         mutex_unlock(&id_priv->qp_mutex);
 675         return ret;
 676 }
 677
 678 static int cma_ib_init_qp_attr(struct rdma_id_private *id_priv,
 679                                struct ib_qp_attr *qp_attr, int *qp_attr_mask)
 680 {
 681         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
 682         int ret;
 683         u16 pkey;
 684
 685         if (rdma_port_get_link_layer(id_priv->id.device, id_priv->id.port_num) ==
 686             IB_LINK_LAYER_INFINIBAND)
 687                 pkey = ib_addr_get_pkey(dev_addr);
 688         else
 689                 pkey = 0xffff;
 690
 691         ret = ib_find_cached_pkey(id_priv->id.device, id_priv->id.port_num,
 692                                   pkey, &qp_attr->pkey_index);
 693         if (ret)
 694                 return ret;
 695
 696         qp_attr->port_num = id_priv->id.port_num;
 697         *qp_attr_mask = IB_QP_STATE | IB_QP_PKEY_INDEX | IB_QP_PORT;
 698
 699         if (id_priv->id.qp_type == IB_QPT_UD) {
 700                 ret = cma_set_qkey(id_priv, 0);
 701                 if (ret)
 702                         return ret;
 703
 704                 qp_attr->qkey = id_priv->qkey;
 705                 *qp_attr_mask |= IB_QP_QKEY;
 706         } else {
 707                 qp_attr->qp_access_flags = 0;
 708                 *qp_attr_mask |= IB_QP_ACCESS_FLAGS;
 709         }
 710         return 0;
 711 }
 712
 713 int rdma_init_qp_attr(struct rdma_cm_id *id, struct ib_qp_attr *qp_attr,
 714                        int *qp_attr_mask)
 715 {
 716         struct rdma_id_private *id_priv;
 717         int ret = 0;
 718
 719         id_priv = container_of(id, struct rdma_id_private, id);
 720         switch (rdma_node_get_transport(id_priv->id.device->node_type)) {
 721         case RDMA_TRANSPORT_IB:
 722                 if (!id_priv->cm_id.ib || (id_priv->id.qp_type == IB_QPT_UD))
 723                         ret = cma_ib_init_qp_attr(id_priv, qp_attr, qp_attr_mask);
 724                 else
 725                         ret = ib_cm_init_qp_attr(id_priv->cm_id.ib, qp_attr,
 726                                                  qp_attr_mask);
 727                 if (qp_attr->qp_state == IB_QPS_RTR)
 728                         qp_attr->rq_psn = id_priv->seq_num;
 729                 break;
 730         case RDMA_TRANSPORT_IWARP:
 731                 if (!id_priv->cm_id.iw) {
 732                         qp_attr->qp_access_flags = 0;
 733                         *qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS;
 734                 } else
 735                         ret = iw_cm_init_qp_attr(id_priv->cm_id.iw, qp_attr,
 736                                                  qp_attr_mask);
 737                 break;
 738         default:
 739                 ret = -ENOSYS;
 740                 break;
 741         }
 742
 743         return ret;
 744 }
 745 EXPORT_SYMBOL(rdma_init_qp_attr);
 746
 747 static inline int cma_zero_addr(struct sockaddr *addr)
 748 {
 749         switch (addr->sa_family) {
 750         case AF_INET:
 751                 return ipv4_is_zeronet(((struct sockaddr_in *)addr)->sin_addr.s_addr);
 752         case AF_INET6:
 753                 return ipv6_addr_any(&((struct sockaddr_in6 *) addr)->sin6_addr);
 754         case AF_IB:
 755                 return ib_addr_any(&((struct sockaddr_ib *) addr)->sib_addr);
 756         default:
 757                 return 0;
 758         }
 759 }
 760
 761 static inline int cma_loopback_addr(struct sockaddr *addr)
 762 {
 763         switch (addr->sa_family) {
 764         case AF_INET:
 765                 return ipv4_is_loopback(((struct sockaddr_in *) addr)->sin_addr.s_addr);
 766         case AF_INET6:
 767                 return ipv6_addr_loopback(&((struct sockaddr_in6 *) addr)->sin6_addr);
 768         case AF_IB:
 769                 return ib_addr_loopback(&((struct sockaddr_ib *) addr)->sib_addr);
 770         default:
 771                 return 0;
 772         }
 773 }
 774
 775 static inline int cma_any_addr(struct sockaddr *addr)
 776 {
 777         return cma_zero_addr(addr) || cma_loopback_addr(addr);
 778 }
 779
 780 static int cma_addr_cmp(struct sockaddr *src, struct sockaddr *dst)
 781 {
 782         if (src->sa_family != dst->sa_family)
 783                 return -1;
 784
 785         switch (src->sa_family) {
 786         case AF_INET:
 787                 return ((struct sockaddr_in *) src)->sin_addr.s_addr !=
 788                        ((struct sockaddr_in *) dst)->sin_addr.s_addr;
 789         case AF_INET6:
 790                 return ipv6_addr_cmp(&((struct sockaddr_in6 *) src)->sin6_addr,
 791                                      &((struct sockaddr_in6 *) dst)->sin6_addr);
 792         default:
 793                 return ib_addr_cmp(&((struct sockaddr_ib *) src)->sib_addr,
 794                                    &((struct sockaddr_ib *) dst)->sib_addr);
 795         }
 796 }
 797
 798 static __be16 cma_port(struct sockaddr *addr)
 799 {
 800         struct sockaddr_ib *sib;
 801
 802         switch (addr->sa_family) {
 803         case AF_INET:
 804                 return ((struct sockaddr_in *) addr)->sin_port;
 805         case AF_INET6:
 806                 return ((struct sockaddr_in6 *) addr)->sin6_port;
 807         case AF_IB:
 808                 sib = (struct sockaddr_ib *) addr;
 809                 return htons((u16) (be64_to_cpu(sib->sib_sid) &
 810                                     be64_to_cpu(sib->sib_sid_mask)));
 811         default:
 812                 return 0;
 813         }
 814 }
 815
 816 static inline int cma_any_port(struct sockaddr *addr)
 817 {
 818         return !cma_port(addr);
 819 }
 820
 821 static void cma_save_ib_info(struct rdma_cm_id *id, struct rdma_cm_id *listen_id,
 822                              struct ib_sa_path_rec *path)
 823 {
 824         struct sockaddr_ib *listen_ib, *ib;
 825
 826         listen_ib = (struct sockaddr_ib *) &listen_id->route.addr.src_addr;
 827         ib = (struct sockaddr_ib *) &id->route.addr.src_addr;
 828         ib->sib_family = listen_ib->sib_family;
 829         ib->sib_pkey = path->pkey;
 830         ib->sib_flowinfo = path->flow_label;
 831         memcpy(&ib->sib_addr, &path->sgid, 16);
 832         ib->sib_sid = listen_ib->sib_sid;
 833         ib->sib_sid_mask = cpu_to_be64(0xffffffffffffffffULL);
 834         ib->sib_scope_id = listen_ib->sib_scope_id;
 835
 836         ib = (struct sockaddr_ib *) &id->route.addr.dst_addr;
 837         ib->sib_family = listen_ib->sib_family;
 838         ib->sib_pkey = path->pkey;
 839         ib->sib_flowinfo = path->flow_label;
 840         memcpy(&ib->sib_addr, &path->dgid, 16);
 841 }
 842
 843 static void cma_save_ip4_info(struct rdma_cm_id *id, struct rdma_cm_id *listen_id,
 844                               struct cma_hdr *hdr)
 845 {
 846         struct sockaddr_in *listen4, *ip4;
 847
 848         listen4 = (struct sockaddr_in *) &listen_id->route.addr.src_addr;
 849         ip4 = (struct sockaddr_in *) &id->route.addr.src_addr;
 850         ip4->sin_family = listen4->sin_family;
 851         ip4->sin_addr.s_addr = hdr->dst_addr.ip4.addr;
 852         ip4->sin_port = listen4->sin_port;
 853
 854         ip4 = (struct sockaddr_in *) &id->route.addr.dst_addr;
 855         ip4->sin_family = listen4->sin_family;
 856         ip4->sin_addr.s_addr = hdr->src_addr.ip4.addr;
 857         ip4->sin_port = hdr->port;
 858 }
 859
 860 static void cma_save_ip6_info(struct rdma_cm_id *id, struct rdma_cm_id *listen_id,
 861                               struct cma_hdr *hdr)
 862 {
 863         struct sockaddr_in6 *listen6, *ip6;
 864
 865         listen6 = (struct sockaddr_in6 *) &listen_id->route.addr.src_addr;
 866         ip6 = (struct sockaddr_in6 *) &id->route.addr.src_addr;
 867         ip6->sin6_family = listen6->sin6_family;
 868         ip6->sin6_addr = hdr->dst_addr.ip6;
 869         ip6->sin6_port = listen6->sin6_port;
 870
 871         ip6 = (struct sockaddr_in6 *) &id->route.addr.dst_addr;
 872         ip6->sin6_family = listen6->sin6_family;
 873         ip6->sin6_addr = hdr->src_addr.ip6;
 874         ip6->sin6_port = hdr->port;
 875 }
 876
 877 static int cma_save_net_info(struct rdma_cm_id *id, struct rdma_cm_id *listen_id,
 878                              struct ib_cm_event *ib_event)
 879 {
 880         struct cma_hdr *hdr;
 881
 882         if ((listen_id->route.addr.src_addr.ss_family == AF_IB) &&
 883             (ib_event->event == IB_CM_REQ_RECEIVED)) {
 884                 cma_save_ib_info(id, listen_id, ib_event->param.req_rcvd.primary_path);
 885                 return 0;
 886         }
 887
 888         hdr = ib_event->private_data;
 889         if (hdr->cma_version != CMA_VERSION)
 890                 return -EINVAL;
 891
 892         switch (cma_get_ip_ver(hdr)) {
 893         case 4:
 894                 cma_save_ip4_info(id, listen_id, hdr);
 895                 break;
 896         case 6:
 897                 cma_save_ip6_info(id, listen_id, hdr);
 898                 break;
 899         default:
 900                 return -EINVAL;
 901         }
 902         return 0;
 903 }
 904
 905 static inline int cma_user_data_offset(struct rdma_id_private *id_priv)
 906 {
 907         return cma_family(id_priv) == AF_IB ? 0 : sizeof(struct cma_hdr);
 908 }
 909
 910 static void cma_cancel_route(struct rdma_id_private *id_priv)
 911 {
 912         switch (rdma_port_get_link_layer(id_priv->id.device, id_priv->id.port_num)) {
 913         case IB_LINK_LAYER_INFINIBAND:
 914                 if (id_priv->query)
 915                         ib_sa_cancel_query(id_priv->query_id, id_priv->query);
 916                 break;
 917         default:
 918                 break;
 919         }
 920 }
 921
 922 static void cma_cancel_listens(struct rdma_id_private *id_priv)
 923 {
 924         struct rdma_id_private *dev_id_priv;
 925
 926         /*
 927          * Remove from listen_any_list to prevent added devices from spawning
 928          * additional listen requests.
 929          */
 930         mutex_lock(&lock);
 931         list_del(&id_priv->list);
 932
 933         while (!list_empty(&id_priv->listen_list)) {
 934                 dev_id_priv = list_entry(id_priv->listen_list.next,
 935                                          struct rdma_id_private, listen_list);
 936                 /* sync with device removal to avoid duplicate destruction */
 937                 list_del_init(&dev_id_priv->list);
 938                 list_del(&dev_id_priv->listen_list);
 939                 mutex_unlock(&lock);
 940
 941                 rdma_destroy_id(&dev_id_priv->id);
 942                 mutex_lock(&lock);
 943         }
 944         mutex_unlock(&lock);
 945 }
 946
 947 static void cma_cancel_operation(struct rdma_id_private *id_priv,
 948                                  enum rdma_cm_state state)
 949 {
 950         switch (state) {
 951         case RDMA_CM_ADDR_QUERY:
 952                 rdma_addr_cancel(&id_priv->id.route.addr.dev_addr);
 953                 break;
 954         case RDMA_CM_ROUTE_QUERY:
 955                 cma_cancel_route(id_priv);
 956                 break;
 957         case RDMA_CM_LISTEN:
 958                 if (cma_any_addr(cma_src_addr(id_priv)) && !id_priv->cma_dev)
 959                         cma_cancel_listens(id_priv);
 960                 break;
 961         default:
 962                 break;
 963         }
 964 }
 965
 966 static void cma_release_port(struct rdma_id_private *id_priv)
 967 {
 968         struct rdma_bind_list *bind_list = id_priv->bind_list;
 969
 970         if (!bind_list)
 971                 return;
 972
 973         mutex_lock(&lock);
 974         hlist_del(&id_priv->node);
 975         if (hlist_empty(&bind_list->owners)) {
 976                 idr_remove(bind_list->ps, bind_list->port);
 977                 kfree(bind_list);
 978         }
 979         mutex_unlock(&lock);
 980 }
 981
 982 static void cma_leave_mc_groups(struct rdma_id_private *id_priv)
 983 {
 984         struct cma_multicast *mc;
 985
 986         while (!list_empty(&id_priv->mc_list)) {
 987                 mc = container_of(id_priv->mc_list.next,
 988                                   struct cma_multicast, list);
 989                 list_del(&mc->list);
 990                 switch (rdma_port_get_link_layer(id_priv->cma_dev->device, id_priv->id.port_num)) {
 991                 case IB_LINK_LAYER_INFINIBAND:
 992                         ib_sa_free_multicast(mc->multicast.ib);
 993                         kfree(mc);
 994                         break;
 995                 case IB_LINK_LAYER_ETHERNET:
 996                         kref_put(&mc->mcref, release_mc);
 997                         break;
 998                 default:
 999                         break;
1000                 }
1001         }
1002 }
1003
1004 void rdma_destroy_id(struct rdma_cm_id *id)
1005 {
1006         struct rdma_id_private *id_priv;
1007         enum rdma_cm_state state;
1008
1009         id_priv = container_of(id, struct rdma_id_private, id);
1010         state = cma_exch(id_priv, RDMA_CM_DESTROYING);
1011         cma_cancel_operation(id_priv, state);
1012
1013         /*
1014          * Wait for any active callback to finish.  New callbacks will find
1015          * the id_priv state set to destroying and abort.
1016          */
1017         mutex_lock(&id_priv->handler_mutex);
1018         mutex_unlock(&id_priv->handler_mutex);
1019
1020         if (id_priv->cma_dev) {
1021                 switch (rdma_node_get_transport(id_priv->id.device->node_type)) {
1022                 case RDMA_TRANSPORT_IB:
1023                         if (id_priv->cm_id.ib)
1024                                 ib_destroy_cm_id(id_priv->cm_id.ib);
1025                         break;
1026                 case RDMA_TRANSPORT_IWARP:
1027                         if (id_priv->cm_id.iw)
1028                                 iw_destroy_cm_id(id_priv->cm_id.iw);
1029                         break;
1030                 default:
1031                         break;
1032                 }
1033                 cma_leave_mc_groups(id_priv);
1034                 cma_release_dev(id_priv);
1035         }
1036
1037         cma_release_port(id_priv);
1038         cma_deref_id(id_priv);
1039         wait_for_completion(&id_priv->comp);
1040
1041         if (id_priv->internal_id)
1042                 cma_deref_id(id_priv->id.context);
1043
1044         kfree(id_priv->id.route.path_rec);
1045         kfree(id_priv);
1046 }
1047 EXPORT_SYMBOL(rdma_destroy_id);
1048
1049 static int cma_rep_recv(struct rdma_id_private *id_priv)
1050 {
1051         int ret;
1052
1053         ret = cma_modify_qp_rtr(id_priv, NULL);
1054         if (ret)
1055                 goto reject;
1056
1057         ret = cma_modify_qp_rts(id_priv, NULL);
1058         if (ret)
1059                 goto reject;
1060
1061         ret = ib_send_cm_rtu(id_priv->cm_id.ib, NULL, 0);
1062         if (ret)
1063                 goto reject;
1064
1065         return 0;
1066 reject:
1067         cma_modify_qp_err(id_priv);
1068         ib_send_cm_rej(id_priv->cm_id.ib, IB_CM_REJ_CONSUMER_DEFINED,
1069                        NULL, 0, NULL, 0);
1070         return ret;
1071 }
1072
1073 static void cma_set_rep_event_data(struct rdma_cm_event *event,
1074                                    struct ib_cm_rep_event_param *rep_data,
1075                                    void *private_data)
1076 {
1077         event->param.conn.private_data = private_data;
1078         event->param.conn.private_data_len = IB_CM_REP_PRIVATE_DATA_SIZE;
1079         event->param.conn.responder_resources = rep_data->responder_resources;
1080         event->param.conn.initiator_depth = rep_data->initiator_depth;
1081         event->param.conn.flow_control = rep_data->flow_control;
1082         event->param.conn.rnr_retry_count = rep_data->rnr_retry_count;
1083         event->param.conn.srq = rep_data->srq;
1084         event->param.conn.qp_num = rep_data->remote_qpn;
1085 }
1086
1087 static int cma_ib_handler(struct ib_cm_id *cm_id, struct ib_cm_event *ib_event)
1088 {
1089         struct rdma_id_private *id_priv = cm_id->context;
1090         struct rdma_cm_event event;
1091         int ret = 0;
1092
1093         if ((ib_event->event != IB_CM_TIMEWAIT_EXIT &&
1094                 cma_disable_callback(id_priv, RDMA_CM_CONNECT)) ||
1095             (ib_event->event == IB_CM_TIMEWAIT_EXIT &&
1096                 cma_disable_callback(id_priv, RDMA_CM_DISCONNECT)))
1097                 return 0;
1098
1099         memset(&event, 0, sizeof event);
1100         switch (ib_event->event) {
1101         case IB_CM_REQ_ERROR:
1102         case IB_CM_REP_ERROR:
1103                 event.event = RDMA_CM_EVENT_UNREACHABLE;
1104                 event.status = -ETIMEDOUT;
1105                 break;
1106         case IB_CM_REP_RECEIVED:
1107                 if (id_priv->id.qp) {
1108                         event.status = cma_rep_recv(id_priv);
1109                         event.event = event.status ? RDMA_CM_EVENT_CONNECT_ERROR :
1110                                                      RDMA_CM_EVENT_ESTABLISHED;
1111                 } else {
1112                         event.event = RDMA_CM_EVENT_CONNECT_RESPONSE;
1113                 }
1114                 cma_set_rep_event_data(&event, &ib_event->param.rep_rcvd,
1115                                        ib_event->private_data);
1116                 break;
1117         case IB_CM_RTU_RECEIVED:
1118         case IB_CM_USER_ESTABLISHED:
1119                 event.event = RDMA_CM_EVENT_ESTABLISHED;
1120                 break;
1121         case IB_CM_DREQ_ERROR:
1122                 event.status = -ETIMEDOUT; /* fall through */
1123         case IB_CM_DREQ_RECEIVED:
1124         case IB_CM_DREP_RECEIVED:
1125                 if (!cma_comp_exch(id_priv, RDMA_CM_CONNECT,
1126                                    RDMA_CM_DISCONNECT))
1127                         goto out;
1128                 event.event = RDMA_CM_EVENT_DISCONNECTED;
1129                 break;
1130         case IB_CM_TIMEWAIT_EXIT:
1131                 event.event = RDMA_CM_EVENT_TIMEWAIT_EXIT;
1132                 break;
1133         case IB_CM_MRA_RECEIVED:
1134                 /* ignore event */
1135                 goto out;
1136         case IB_CM_REJ_RECEIVED:
1137                 cma_modify_qp_err(id_priv);
1138                 event.status = ib_event->param.rej_rcvd.reason;
1139                 event.event = RDMA_CM_EVENT_REJECTED;
1140                 event.param.conn.private_data = ib_event->private_data;
1141                 event.param.conn.private_data_len = IB_CM_REJ_PRIVATE_DATA_SIZE;
1142                 break;
1143         default:
1144                 printk(KERN_ERR "RDMA CMA: unexpected IB CM event: %d\n",
1145                        ib_event->event);
1146                 goto out;
1147         }
1148
1149         ret = id_priv->id.event_handler(&id_priv->id, &event);
1150         if (ret) {
1151                 /* Destroy the CM ID by returning a non-zero value. */
1152                 id_priv->cm_id.ib = NULL;
1153                 cma_exch(id_priv, RDMA_CM_DESTROYING);
1154                 mutex_unlock(&id_priv->handler_mutex);
1155                 rdma_destroy_id(&id_priv->id);
1156                 return ret;
1157         }
1158 out:
1159         mutex_unlock(&id_priv->handler_mutex);
1160         return ret;
1161 }
1162
1163 static struct rdma_id_private *cma_new_conn_id(struct rdma_cm_id *listen_id,
1164                                                struct ib_cm_event *ib_event)
1165 {
1166         struct rdma_id_private *id_priv;
1167         struct rdma_cm_id *id;
1168         struct rdma_route *rt;
1169         int ret;
1170
1171         id = rdma_create_id(listen_id->event_handler, listen_id->context,
1172                             listen_id->ps, ib_event->param.req_rcvd.qp_type);
1173         if (IS_ERR(id))
1174                 return NULL;
1175
1176         id_priv = container_of(id, struct rdma_id_private, id);
1177         if (cma_save_net_info(id, listen_id, ib_event))
1178                 goto err;
1179
1180         rt = &id->route;
1181         rt->num_paths = ib_event->param.req_rcvd.alternate_path ? 2 : 1;
1182         rt->path_rec = kmalloc(sizeof *rt->path_rec * rt->num_paths,
1183                                GFP_KERNEL);
1184         if (!rt->path_rec)
1185                 goto err;
1186
1187         rt->path_rec[0] = *ib_event->param.req_rcvd.primary_path;
1188         if (rt->num_paths == 2)
1189                 rt->path_rec[1] = *ib_event->param.req_rcvd.alternate_path;
1190
1191         if (cma_any_addr(cma_src_addr(id_priv))) {
1192                 rt->addr.dev_addr.dev_type = ARPHRD_INFINIBAND;
1193                 rdma_addr_set_sgid(&rt->addr.dev_addr, &rt->path_rec[0].sgid);
1194                 ib_addr_set_pkey(&rt->addr.dev_addr, be16_to_cpu(rt->path_rec[0].pkey));
1195         } else {
1196                 ret = cma_translate_addr(cma_src_addr(id_priv), &rt->addr.dev_addr);
1197                 if (ret)
1198                         goto err;
1199         }
1200         rdma_addr_set_dgid(&rt->addr.dev_addr, &rt->path_rec[0].dgid);
1201
1202         id_priv->state = RDMA_CM_CONNECT;
1203         return id_priv;
1204
1205 err:
1206         rdma_destroy_id(id);
1207         return NULL;
1208 }
1209
1210 static struct rdma_id_private *cma_new_udp_id(struct rdma_cm_id *listen_id,
1211                                               struct ib_cm_event *ib_event)
1212 {
1213         struct rdma_id_private *id_priv;
1214         struct rdma_cm_id *id;
1215         int ret;
1216
1217         id = rdma_create_id(listen_id->event_handler, listen_id->context,
1218                             listen_id->ps, IB_QPT_UD);
1219         if (IS_ERR(id))
1220                 return NULL;
1221
1222         id_priv = container_of(id, struct rdma_id_private, id);
1223         if (cma_save_net_info(id, listen_id, ib_event))
1224                 goto err;
1225
1226         if (!cma_any_addr((struct sockaddr *) &id->route.addr.src_addr)) {
1227                 ret = cma_translate_addr(cma_src_addr(id_priv), &id->route.addr.dev_addr);
1228                 if (ret)
1229                         goto err;
1230         }
1231
1232         id_priv->state = RDMA_CM_CONNECT;
1233         return id_priv;
1234 err:
1235         rdma_destroy_id(id);
1236         return NULL;
1237 }
1238
1239 static void cma_set_req_event_data(struct rdma_cm_event *event,
1240                                    struct ib_cm_req_event_param *req_data,
1241                                    void *private_data, int offset)
1242 {
1243         event->param.conn.private_data = private_data + offset;
1244         event->param.conn.private_data_len = IB_CM_REQ_PRIVATE_DATA_SIZE - offset;
1245         event->param.conn.responder_resources = req_data->responder_resources;
1246         event->param.conn.initiator_depth = req_data->initiator_depth;
1247         event->param.conn.flow_control = req_data->flow_control;
1248         event->param.conn.retry_count = req_data->retry_count;
1249         event->param.conn.rnr_retry_count = req_data->rnr_retry_count;
1250         event->param.conn.srq = req_data->srq;
1251         event->param.conn.qp_num = req_data->remote_qpn;
1252 }
1253
1254 static int cma_check_req_qp_type(struct rdma_cm_id *id, struct ib_cm_event *ib_event)
1255 {
1256         return (((ib_event->event == IB_CM_REQ_RECEIVED) &&
1257                  (ib_event->param.req_rcvd.qp_type == id->qp_type)) ||
1258                 ((ib_event->event == IB_CM_SIDR_REQ_RECEIVED) &&
1259                  (id->qp_type == IB_QPT_UD)) ||
1260                 (!id->qp_type));
1261 }
1262
1263 static int cma_req_handler(struct ib_cm_id *cm_id, struct ib_cm_event *ib_event)
1264 {
1265         struct rdma_id_private *listen_id, *conn_id;
1266         struct rdma_cm_event event;
1267         int offset, ret;
1268
1269         listen_id = cm_id->context;
1270         if (!cma_check_req_qp_type(&listen_id->id, ib_event))
1271                 return -EINVAL;
1272
1273         if (cma_disable_callback(listen_id, RDMA_CM_LISTEN))
1274                 return -ECONNABORTED;
1275
1276         memset(&event, 0, sizeof event);
1277         offset = cma_user_data_offset(listen_id);
1278         event.event = RDMA_CM_EVENT_CONNECT_REQUEST;
1279         if (ib_event->event == IB_CM_SIDR_REQ_RECEIVED) {
1280                 conn_id = cma_new_udp_id(&listen_id->id, ib_event);
1281                 event.param.ud.private_data = ib_event->private_data + offset;
1282                 event.param.ud.private_data_len =
1283                                 IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE - offset;
1284         } else {
1285                 conn_id = cma_new_conn_id(&listen_id->id, ib_event);
1286                 cma_set_req_event_data(&event, &ib_event->param.req_rcvd,
1287                                        ib_event->private_data, offset);
1288         }
1289         if (!conn_id) {
1290                 ret = -ENOMEM;
1291                 goto err1;
1292         }
1293
1294         mutex_lock_nested(&conn_id->handler_mutex, SINGLE_DEPTH_NESTING);
1295         ret = cma_acquire_dev(conn_id);
1296         if (ret)
1297                 goto err2;
1298
1299         conn_id->cm_id.ib = cm_id;
1300         cm_id->context = conn_id;
1301         cm_id->cm_handler = cma_ib_handler;
1302
1303         /*
1304          * Protect against the user destroying conn_id from another thread
1305          * until we're done accessing it.
1306          */
1307         atomic_inc(&conn_id->refcount);
1308         ret = conn_id->id.event_handler(&conn_id->id, &event);
1309         if (ret)
1310                 goto err3;
1311
1312         /*
1313          * Acquire mutex to prevent user executing rdma_destroy_id()
1314          * while we're accessing the cm_id.
1315          */
1316         mutex_lock(&lock);
1317         if (cma_comp(conn_id, RDMA_CM_CONNECT) && (conn_id->id.qp_type != IB_QPT_UD))
1318                 ib_send_cm_mra(cm_id, CMA_CM_MRA_SETTING, NULL, 0);
1319         mutex_unlock(&lock);
1320         mutex_unlock(&conn_id->handler_mutex);
1321         mutex_unlock(&listen_id->handler_mutex);
1322         cma_deref_id(conn_id);
1323         return 0;
1324
1325 err3:
1326         cma_deref_id(conn_id);
1327         /* Destroy the CM ID by returning a non-zero value. */
1328         conn_id->cm_id.ib = NULL;
1329 err2:
1330         cma_exch(conn_id, RDMA_CM_DESTROYING);
1331         mutex_unlock(&conn_id->handler_mutex);
1332 err1:
1333         mutex_unlock(&listen_id->handler_mutex);
1334         if (conn_id)
1335                 rdma_destroy_id(&conn_id->id);
1336         return ret;
1337 }
1338
1339 __be64 rdma_get_service_id(struct rdma_cm_id *id, struct sockaddr *addr)
1340 {
1341         if (addr->sa_family == AF_IB)
1342                 return ((struct sockaddr_ib *) addr)->sib_sid;
1343
1344         return cpu_to_be64(((u64)id->ps << 16) + be16_to_cpu(cma_port(addr)));
1345 }
1346 EXPORT_SYMBOL(rdma_get_service_id);
1347
1348 static void cma_set_compare_data(enum rdma_port_space ps, struct sockaddr *addr,
1349                                  struct ib_cm_compare_data *compare)
1350 {
1351         struct cma_hdr *cma_data, *cma_mask;
1352         __be32 ip4_addr;
1353         struct in6_addr ip6_addr;
1354
1355         memset(compare, 0, sizeof *compare);
1356         cma_data = (void *) compare->data;
1357         cma_mask = (void *) compare->mask;
1358
1359         switch (addr->sa_family) {
1360         case AF_INET:
1361                 ip4_addr = ((struct sockaddr_in *) addr)->sin_addr.s_addr;
1362                 cma_set_ip_ver(cma_data, 4);
1363                 cma_set_ip_ver(cma_mask, 0xF);
1364                 if (!cma_any_addr(addr)) {
1365                         cma_data->dst_addr.ip4.addr = ip4_addr;
1366                         cma_mask->dst_addr.ip4.addr = htonl(~0);
1367                 }
1368                 break;
1369         case AF_INET6:
1370                 ip6_addr = ((struct sockaddr_in6 *) addr)->sin6_addr;
1371                 cma_set_ip_ver(cma_data, 6);
1372                 cma_set_ip_ver(cma_mask, 0xF);
1373                 if (!cma_any_addr(addr)) {
1374                         cma_data->dst_addr.ip6 = ip6_addr;
1375                         memset(&cma_mask->dst_addr.ip6, 0xFF,
1376                                sizeof cma_mask->dst_addr.ip6);
1377                 }
1378                 break;
1379         default:
1380                 break;
1381         }
1382 }
1383
1384 static int cma_iw_handler(struct iw_cm_id *iw_id, struct iw_cm_event *iw_event)
1385 {
1386         struct rdma_id_private *id_priv = iw_id->context;
1387         struct rdma_cm_event event;
1388         int ret = 0;
1389         struct sockaddr *laddr = (struct sockaddr *)&iw_event->local_addr;
1390         struct sockaddr *raddr = (struct sockaddr *)&iw_event->remote_addr;
1391
1392         if (cma_disable_callback(id_priv, RDMA_CM_CONNECT))
1393                 return 0;
1394
1395         memset(&event, 0, sizeof event);
1396         switch (iw_event->event) {
1397         case IW_CM_EVENT_CLOSE:
1398                 event.event = RDMA_CM_EVENT_DISCONNECTED;
1399                 break;
1400         case IW_CM_EVENT_CONNECT_REPLY:
1401                 memcpy(cma_src_addr(id_priv), laddr,
1402                        rdma_addr_size(laddr));
1403                 memcpy(cma_dst_addr(id_priv), raddr,
1404                        rdma_addr_size(raddr));
1405                 switch (iw_event->status) {
1406                 case 0:
1407                         event.event = RDMA_CM_EVENT_ESTABLISHED;
1408                         event.param.conn.initiator_depth = iw_event->ird;
1409                         event.param.conn.responder_resources = iw_event->ord;
1410                         break;
1411                 case -ECONNRESET:
1412                 case -ECONNREFUSED:
1413                         event.event = RDMA_CM_EVENT_REJECTED;
1414                         break;
1415                 case -ETIMEDOUT:
1416                         event.event = RDMA_CM_EVENT_UNREACHABLE;
1417                         break;
1418                 default:
1419                         event.event = RDMA_CM_EVENT_CONNECT_ERROR;
1420                         break;
1421                 }
1422                 break;
1423         case IW_CM_EVENT_ESTABLISHED:
1424                 event.event = RDMA_CM_EVENT_ESTABLISHED;
1425                 event.param.conn.initiator_depth = iw_event->ird;
1426                 event.param.conn.responder_resources = iw_event->ord;
1427                 break;
1428         default:
1429                 BUG_ON(1);
1430         }
1431
1432         event.status = iw_event->status;
1433         event.param.conn.private_data = iw_event->private_data;
1434         event.param.conn.private_data_len = iw_event->private_data_len;
1435         ret = id_priv->id.event_handler(&id_priv->id, &event);
1436         if (ret) {
1437                 /* Destroy the CM ID by returning a non-zero value. */
1438                 id_priv->cm_id.iw = NULL;
1439                 cma_exch(id_priv, RDMA_CM_DESTROYING);
1440                 mutex_unlock(&id_priv->handler_mutex);
1441                 rdma_destroy_id(&id_priv->id);
1442                 return ret;
1443         }
1444
1445         mutex_unlock(&id_priv->handler_mutex);
1446         return ret;
1447 }
1448
1449 static int iw_conn_req_handler(struct iw_cm_id *cm_id,
1450                                struct iw_cm_event *iw_event)
1451 {
1452         struct rdma_cm_id *new_cm_id;
1453         struct rdma_id_private *listen_id, *conn_id;
1454         struct net_device *dev = NULL;
1455         struct rdma_cm_event event;
1456         int ret;
1457         struct ib_device_attr attr;
1458         struct sockaddr *laddr = (struct sockaddr *)&iw_event->local_addr;
1459         struct sockaddr *raddr = (struct sockaddr *)&iw_event->remote_addr;
1460
1461         listen_id = cm_id->context;
1462         if (cma_disable_callback(listen_id, RDMA_CM_LISTEN))
1463                 return -ECONNABORTED;
1464
1465         /* Create a new RDMA id for the new IW CM ID */
1466         new_cm_id = rdma_create_id(listen_id->id.event_handler,
1467                                    listen_id->id.context,
1468                                    RDMA_PS_TCP, IB_QPT_RC);
1469         if (IS_ERR(new_cm_id)) {
1470                 ret = -ENOMEM;
1471                 goto out;
1472         }
1473         conn_id = container_of(new_cm_id, struct rdma_id_private, id);
1474         mutex_lock_nested(&conn_id->handler_mutex, SINGLE_DEPTH_NESTING);
1475         conn_id->state = RDMA_CM_CONNECT;
1476
1477         ret = rdma_translate_ip(laddr, &conn_id->id.route.addr.dev_addr);
1478         if (ret) {
1479                 mutex_unlock(&conn_id->handler_mutex);
1480                 rdma_destroy_id(new_cm_id);
1481                 goto out;
1482         }
1483
1484         ret = cma_acquire_dev(conn_id);
1485         if (ret) {
1486                 mutex_unlock(&conn_id->handler_mutex);
1487                 rdma_destroy_id(new_cm_id);
1488                 goto out;
1489         }
1490
1491         conn_id->cm_id.iw = cm_id;
1492         cm_id->context = conn_id;
1493         cm_id->cm_handler = cma_iw_handler;
1494
1495         memcpy(cma_src_addr(conn_id), laddr, rdma_addr_size(laddr));
1496         memcpy(cma_dst_addr(conn_id), raddr, rdma_addr_size(raddr));
1497
1498         ret = ib_query_device(conn_id->id.device, &attr);
1499         if (ret) {
1500                 mutex_unlock(&conn_id->handler_mutex);
1501                 rdma_destroy_id(new_cm_id);
1502                 goto out;
1503         }
1504
1505         memset(&event, 0, sizeof event);
1506         event.event = RDMA_CM_EVENT_CONNECT_REQUEST;
1507         event.param.conn.private_data = iw_event->private_data;
1508         event.param.conn.private_data_len = iw_event->private_data_len;
1509         event.param.conn.initiator_depth = iw_event->ird;
1510         event.param.conn.responder_resources = iw_event->ord;
1511
1512         /*
1513          * Protect against the user destroying conn_id from another thread
1514          * until we're done accessing it.
1515          */
1516         atomic_inc(&conn_id->refcount);
1517         ret = conn_id->id.event_handler(&conn_id->id, &event);
1518         if (ret) {
1519                 /* User wants to destroy the CM ID */
1520                 conn_id->cm_id.iw = NULL;
1521                 cma_exch(conn_id, RDMA_CM_DESTROYING);
1522                 mutex_unlock(&conn_id->handler_mutex);
1523                 cma_deref_id(conn_id);
1524                 rdma_destroy_id(&conn_id->id);
1525                 goto out;
1526         }
1527
1528         mutex_unlock(&conn_id->handler_mutex);
1529         cma_deref_id(conn_id);
1530
1531 out:
1532         if (dev)
1533                 dev_put(dev);
1534         mutex_unlock(&listen_id->handler_mutex);
1535         return ret;
1536 }
1537
1538 static int cma_ib_listen(struct rdma_id_private *id_priv)
1539 {
1540         struct ib_cm_compare_data compare_data;
1541         struct sockaddr *addr;
1542         struct ib_cm_id *id;
1543         __be64 svc_id;
1544         int ret;
1545
1546         id = ib_create_cm_id(id_priv->id.device, cma_req_handler, id_priv);
1547         if (IS_ERR(id))
1548                 return PTR_ERR(id);
1549
1550         id_priv->cm_id.ib = id;
1551
1552         addr = cma_src_addr(id_priv);
1553         svc_id = rdma_get_service_id(&id_priv->id, addr);
1554         if (cma_any_addr(addr) && !id_priv->afonly)
1555                 ret = ib_cm_listen(id_priv->cm_id.ib, svc_id, 0, NULL);
1556         else {
1557                 cma_set_compare_data(id_priv->id.ps, addr, &compare_data);
1558                 ret = ib_cm_listen(id_priv->cm_id.ib, svc_id, 0, &compare_data);
1559         }
1560
1561         if (ret) {
1562                 ib_destroy_cm_id(id_priv->cm_id.ib);
1563                 id_priv->cm_id.ib = NULL;
1564         }
1565
1566         return ret;
1567 }
1568
1569 static int cma_iw_listen(struct rdma_id_private *id_priv, int backlog)
1570 {
1571         int ret;
1572         struct iw_cm_id *id;
1573
1574         id = iw_create_cm_id(id_priv->id.device,
1575                              iw_conn_req_handler,
1576                              id_priv);
1577         if (IS_ERR(id))
1578                 return PTR_ERR(id);
1579
1580         id_priv->cm_id.iw = id;
1581
1582         memcpy(&id_priv->cm_id.iw->local_addr, cma_src_addr(id_priv),
1583                rdma_addr_size(cma_src_addr(id_priv)));
1584
1585         ret = iw_cm_listen(id_priv->cm_id.iw, backlog);
1586
1587         if (ret) {
1588                 iw_destroy_cm_id(id_priv->cm_id.iw);
1589                 id_priv->cm_id.iw = NULL;
1590         }
1591
1592         return ret;
1593 }
1594
1595 static int cma_listen_handler(struct rdma_cm_id *id,
1596                               struct rdma_cm_event *event)
1597 {
1598         struct rdma_id_private *id_priv = id->context;
1599
1600         id->context = id_priv->id.context;
1601         id->event_handler = id_priv->id.event_handler;
1602         return id_priv->id.event_handler(id, event);
1603 }
1604
1605 static void cma_listen_on_dev(struct rdma_id_private *id_priv,
1606                               struct cma_device *cma_dev)
1607 {
1608         struct rdma_id_private *dev_id_priv;
1609         struct rdma_cm_id *id;
1610         int ret;
1611
1612         if (cma_family(id_priv) == AF_IB &&
1613             rdma_node_get_transport(cma_dev->device->node_type) != RDMA_TRANSPORT_IB)
1614                 return;
1615
1616         id = rdma_create_id(cma_listen_handler, id_priv, id_priv->id.ps,
1617                             id_priv->id.qp_type);
1618         if (IS_ERR(id))
1619                 return;
1620
1621         dev_id_priv = container_of(id, struct rdma_id_private, id);
1622
1623         dev_id_priv->state = RDMA_CM_ADDR_BOUND;
1624         memcpy(cma_src_addr(dev_id_priv), cma_src_addr(id_priv),
1625                rdma_addr_size(cma_src_addr(id_priv)));
1626
1627         cma_attach_to_dev(dev_id_priv, cma_dev);
1628         list_add_tail(&dev_id_priv->listen_list, &id_priv->listen_list);
1629         atomic_inc(&id_priv->refcount);
1630         dev_id_priv->internal_id = 1;
1631         dev_id_priv->afonly = id_priv->afonly;
1632
1633         ret = rdma_listen(id, id_priv->backlog);
1634         if (ret)
1635                 printk(KERN_WARNING "RDMA CMA: cma_listen_on_dev, error %d, "
1636                        "listening on device %s\n", ret, cma_dev->device->name);
1637 }
1638
1639 static void cma_listen_on_all(struct rdma_id_private *id_priv)
1640 {
1641         struct cma_device *cma_dev;
1642
1643         mutex_lock(&lock);
1644         list_add_tail(&id_priv->list, &listen_any_list);
1645         list_for_each_entry(cma_dev, &dev_list, list)
1646                 cma_listen_on_dev(id_priv, cma_dev);
1647         mutex_unlock(&lock);
1648 }
1649
1650 void rdma_set_service_type(struct rdma_cm_id *id, int tos)
1651 {
1652         struct rdma_id_private *id_priv;
1653
1654         id_priv = container_of(id, struct rdma_id_private, id);
1655         id_priv->tos = (u8) tos;
1656 }
1657 EXPORT_SYMBOL(rdma_set_service_type);
1658
1659 static void cma_query_handler(int status, struct ib_sa_path_rec *path_rec,
1660                               void *context)
1661 {
1662         struct cma_work *work = context;
1663         struct rdma_route *route;
1664
1665         route = &work->id->id.route;
1666
1667         if (!status) {
1668                 route->num_paths = 1;
1669                 *route->path_rec = *path_rec;
1670         } else {
1671                 work->old_state = RDMA_CM_ROUTE_QUERY;
1672                 work->new_state = RDMA_CM_ADDR_RESOLVED;
1673                 work->event.event = RDMA_CM_EVENT_ROUTE_ERROR;
1674                 work->event.status = status;
1675         }
1676
1677         queue_work(cma_wq, &work->work);
1678 }
1679
1680 static int cma_query_ib_route(struct rdma_id_private *id_priv, int timeout_ms,
1681                               struct cma_work *work)
1682 {
1683         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
1684         struct ib_sa_path_rec path_rec;
1685         ib_sa_comp_mask comp_mask;
1686         struct sockaddr_in6 *sin6;
1687         struct sockaddr_ib *sib;
1688
1689         memset(&path_rec, 0, sizeof path_rec);
1690         rdma_addr_get_sgid(dev_addr, &path_rec.sgid);
1691         rdma_addr_get_dgid(dev_addr, &path_rec.dgid);
1692         path_rec.pkey = cpu_to_be16(ib_addr_get_pkey(dev_addr));
1693         path_rec.numb_path = 1;
1694         path_rec.reversible = 1;
1695         path_rec.service_id = rdma_get_service_id(&id_priv->id, cma_dst_addr(id_priv));
1696
1697         comp_mask = IB_SA_PATH_REC_DGID | IB_SA_PATH_REC_SGID |
1698                     IB_SA_PATH_REC_PKEY | IB_SA_PATH_REC_NUMB_PATH |
1699                     IB_SA_PATH_REC_REVERSIBLE | IB_SA_PATH_REC_SERVICE_ID;
1700
1701         switch (cma_family(id_priv)) {
1702         case AF_INET:
1703                 path_rec.qos_class = cpu_to_be16((u16) id_priv->tos);
1704                 comp_mask |= IB_SA_PATH_REC_QOS_CLASS;
1705                 break;
1706         case AF_INET6:
1707                 sin6 = (struct sockaddr_in6 *) cma_src_addr(id_priv);
1708                 path_rec.traffic_class = (u8) (be32_to_cpu(sin6->sin6_flowinfo) >> 20);
1709                 comp_mask |= IB_SA_PATH_REC_TRAFFIC_CLASS;
1710                 break;
1711         case AF_IB:
1712                 sib = (struct sockaddr_ib *) cma_src_addr(id_priv);
1713                 path_rec.traffic_class = (u8) (be32_to_cpu(sib->sib_flowinfo) >> 20);
1714                 comp_mask |= IB_SA_PATH_REC_TRAFFIC_CLASS;
1715                 break;
1716         }
1717
1718         id_priv->query_id = ib_sa_path_rec_get(&sa_client, id_priv->id.device,
1719                                                id_priv->id.port_num, &path_rec,
1720                                                comp_mask, timeout_ms,
1721                                                GFP_KERNEL, cma_query_handler,
1722                                                work, &id_priv->query);
1723
1724         return (id_priv->query_id < 0) ? id_priv->query_id : 0;
1725 }
1726
1727 static void cma_work_handler(struct work_struct *_work)
1728 {
1729         struct cma_work *work = container_of(_work, struct cma_work, work);
1730         struct rdma_id_private *id_priv = work->id;
1731         int destroy = 0;
1732
1733         mutex_lock(&id_priv->handler_mutex);
1734         if (!cma_comp_exch(id_priv, work->old_state, work->new_state))
1735                 goto out;
1736
1737         if (id_priv->id.event_handler(&id_priv->id, &work->event)) {
1738                 cma_exch(id_priv, RDMA_CM_DESTROYING);
1739                 destroy = 1;
1740         }
1741 out:
1742         mutex_unlock(&id_priv->handler_mutex);
1743         cma_deref_id(id_priv);
1744         if (destroy)
1745                 rdma_destroy_id(&id_priv->id);
1746         kfree(work);
1747 }
1748
1749 static void cma_ndev_work_handler(struct work_struct *_work)
1750 {
1751         struct cma_ndev_work *work = container_of(_work, struct cma_ndev_work, work);
1752         struct rdma_id_private *id_priv = work->id;
1753         int destroy = 0;
1754
1755         mutex_lock(&id_priv->handler_mutex);
1756         if (id_priv->state == RDMA_CM_DESTROYING ||
1757             id_priv->state == RDMA_CM_DEVICE_REMOVAL)
1758                 goto out;
1759
1760         if (id_priv->id.event_handler(&id_priv->id, &work->event)) {
1761                 cma_exch(id_priv, RDMA_CM_DESTROYING);
1762                 destroy = 1;
1763         }
1764
1765 out:
1766         mutex_unlock(&id_priv->handler_mutex);
1767         cma_deref_id(id_priv);
1768         if (destroy)
1769                 rdma_destroy_id(&id_priv->id);
1770         kfree(work);
1771 }
1772
1773 static int cma_resolve_ib_route(struct rdma_id_private *id_priv, int timeout_ms)
1774 {
1775         struct rdma_route *route = &id_priv->id.route;
1776         struct cma_work *work;
1777         int ret;
1778
1779         work = kzalloc(sizeof *work, GFP_KERNEL);
1780         if (!work)
1781                 return -ENOMEM;
1782
1783         work->id = id_priv;
1784         INIT_WORK(&work->work, cma_work_handler);
1785         work->old_state = RDMA_CM_ROUTE_QUERY;
1786         work->new_state = RDMA_CM_ROUTE_RESOLVED;
1787         work->event.event = RDMA_CM_EVENT_ROUTE_RESOLVED;
1788
1789         route->path_rec = kmalloc(sizeof *route->path_rec, GFP_KERNEL);
1790         if (!route->path_rec) {
1791                 ret = -ENOMEM;
1792                 goto err1;
1793         }
1794
1795         ret = cma_query_ib_route(id_priv, timeout_ms, work);
1796         if (ret)
1797                 goto err2;
1798
1799         return 0;
1800 err2:
1801         kfree(route->path_rec);
1802         route->path_rec = NULL;
1803 err1:
1804         kfree(work);
1805         return ret;
1806 }
1807
1808 int rdma_set_ib_paths(struct rdma_cm_id *id,
1809                       struct ib_sa_path_rec *path_rec, int num_paths)
1810 {
1811         struct rdma_id_private *id_priv;
1812         int ret;
1813
1814         id_priv = container_of(id, struct rdma_id_private, id);
1815         if (!cma_comp_exch(id_priv, RDMA_CM_ADDR_RESOLVED,
1816                            RDMA_CM_ROUTE_RESOLVED))
1817                 return -EINVAL;
1818
1819         id->route.path_rec = kmemdup(path_rec, sizeof *path_rec * num_paths,
1820                                      GFP_KERNEL);
1821         if (!id->route.path_rec) {
1822                 ret = -ENOMEM;
1823                 goto err;
1824         }
1825
1826         id->route.num_paths = num_paths;
1827         return 0;
1828 err:
1829         cma_comp_exch(id_priv, RDMA_CM_ROUTE_RESOLVED, RDMA_CM_ADDR_RESOLVED);
1830         return ret;
1831 }
1832 EXPORT_SYMBOL(rdma_set_ib_paths);
1833
1834 static int cma_resolve_iw_route(struct rdma_id_private *id_priv, int timeout_ms)
1835 {
1836         struct cma_work *work;
1837
1838         work = kzalloc(sizeof *work, GFP_KERNEL);
1839         if (!work)
1840                 return -ENOMEM;
1841
1842         work->id = id_priv;
1843         INIT_WORK(&work->work, cma_work_handler);
1844         work->old_state = RDMA_CM_ROUTE_QUERY;
1845         work->new_state = RDMA_CM_ROUTE_RESOLVED;
1846         work->event.event = RDMA_CM_EVENT_ROUTE_RESOLVED;
1847         queue_work(cma_wq, &work->work);
1848         return 0;
1849 }
1850
1851 static int iboe_tos_to_sl(struct net_device *ndev, int tos)
1852 {
1853         int prio;
1854         struct net_device *dev;
1855
1856         prio = rt_tos2priority(tos);
1857         dev = ndev->priv_flags & IFF_802_1Q_VLAN ?
1858                 vlan_dev_real_dev(ndev) : ndev;
1859
1860         if (dev->num_tc)
1861                 return netdev_get_prio_tc_map(dev, prio);
1862
1863 #if IS_ENABLED(CONFIG_VLAN_8021Q)
1864         if (ndev->priv_flags & IFF_802_1Q_VLAN)
1865                 return (vlan_dev_get_egress_qos_mask(ndev, prio) &
1866                         VLAN_PRIO_MASK) >> VLAN_PRIO_SHIFT;
1867 #endif
1868         return 0;
1869 }
1870
1871 static int cma_resolve_iboe_route(struct rdma_id_private *id_priv)
1872 {
1873         struct rdma_route *route = &id_priv->id.route;
1874         struct rdma_addr *addr = &route->addr;
1875         struct cma_work *work;
1876         int ret;
1877         struct net_device *ndev = NULL;
1878         u16 vid;
1879
1880         work = kzalloc(sizeof *work, GFP_KERNEL);
1881         if (!work)
1882                 return -ENOMEM;
1883
1884         work->id = id_priv;
1885         INIT_WORK(&work->work, cma_work_handler);
1886
1887         route->path_rec = kzalloc(sizeof *route->path_rec, GFP_KERNEL);
1888         if (!route->path_rec) {
1889                 ret = -ENOMEM;
1890                 goto err1;
1891         }
1892
1893         route->num_paths = 1;
1894
1895         if (addr->dev_addr.bound_dev_if)
1896                 ndev = dev_get_by_index(&init_net, addr->dev_addr.bound_dev_if);
1897         if (!ndev) {
1898                 ret = -ENODEV;
1899                 goto err2;
1900         }
1901
1902         vid = rdma_vlan_dev_vlan_id(ndev);
1903
1904         iboe_mac_vlan_to_ll(&route->path_rec->sgid, addr->dev_addr.src_dev_addr, vid);
1905         iboe_mac_vlan_to_ll(&route->path_rec->dgid, addr->dev_addr.dst_dev_addr, vid);
1906
1907         route->path_rec->hop_limit = 1;
1908         route->path_rec->reversible = 1;
1909         route->path_rec->pkey = cpu_to_be16(0xffff);
1910         route->path_rec->mtu_selector = IB_SA_EQ;
1911         route->path_rec->sl = iboe_tos_to_sl(ndev, id_priv->tos);
1912         route->path_rec->mtu = iboe_get_mtu(ndev->mtu);
1913         route->path_rec->rate_selector = IB_SA_EQ;
1914         route->path_rec->rate = iboe_get_rate(ndev);
1915         dev_put(ndev);
1916         route->path_rec->packet_life_time_selector = IB_SA_EQ;
1917         route->path_rec->packet_life_time = CMA_IBOE_PACKET_LIFETIME;
1918         if (!route->path_rec->mtu) {
1919                 ret = -EINVAL;
1920                 goto err2;
1921         }
1922
1923         work->old_state = RDMA_CM_ROUTE_QUERY;
1924         work->new_state = RDMA_CM_ROUTE_RESOLVED;
1925         work->event.event = RDMA_CM_EVENT_ROUTE_RESOLVED;
1926         work->event.status = 0;
1927
1928         queue_work(cma_wq, &work->work);
1929
1930         return 0;
1931
1932 err2:
1933         kfree(route->path_rec);
1934         route->path_rec = NULL;
1935 err1:
1936         kfree(work);
1937         return ret;
1938 }
1939
1940 int rdma_resolve_route(struct rdma_cm_id *id, int timeout_ms)
1941 {
1942         struct rdma_id_private *id_priv;
1943         int ret;
1944
1945         id_priv = container_of(id, struct rdma_id_private, id);
1946         if (!cma_comp_exch(id_priv, RDMA_CM_ADDR_RESOLVED, RDMA_CM_ROUTE_QUERY))
1947                 return -EINVAL;
1948
1949         atomic_inc(&id_priv->refcount);
1950         switch (rdma_node_get_transport(id->device->node_type)) {
1951         case RDMA_TRANSPORT_IB:
1952                 switch (rdma_port_get_link_layer(id->device, id->port_num)) {
1953                 case IB_LINK_LAYER_INFINIBAND:
1954                         ret = cma_resolve_ib_route(id_priv, timeout_ms);
1955                         break;
1956                 case IB_LINK_LAYER_ETHERNET:
1957                         ret = cma_resolve_iboe_route(id_priv);
1958                         break;
1959                 default:
1960                         ret = -ENOSYS;
1961                 }
1962                 break;
1963         case RDMA_TRANSPORT_IWARP:
1964                 ret = cma_resolve_iw_route(id_priv, timeout_ms);
1965                 break;
1966         default:
1967                 ret = -ENOSYS;
1968                 break;
1969         }
1970         if (ret)
1971                 goto err;
1972
1973         return 0;
1974 err:
1975         cma_comp_exch(id_priv, RDMA_CM_ROUTE_QUERY, RDMA_CM_ADDR_RESOLVED);
1976         cma_deref_id(id_priv);
1977         return ret;
1978 }
1979 EXPORT_SYMBOL(rdma_resolve_route);
1980
1981 static void cma_set_loopback(struct sockaddr *addr)
1982 {
1983         switch (addr->sa_family) {
1984         case AF_INET:
1985                 ((struct sockaddr_in *) addr)->sin_addr.s_addr = htonl(INADDR_LOOPBACK);
1986                 break;
1987         case AF_INET6:
1988                 ipv6_addr_set(&((struct sockaddr_in6 *) addr)->sin6_addr,
1989                               0, 0, 0, htonl(1));
1990                 break;
1991         default:
1992                 ib_addr_set(&((struct sockaddr_ib *) addr)->sib_addr,
1993                             0, 0, 0, htonl(1));
1994                 break;
1995         }
1996 }
1997
1998 static int cma_bind_loopback(struct rdma_id_private *id_priv)
1999 {
2000         struct cma_device *cma_dev, *cur_dev;
2001         struct ib_port_attr port_attr;
2002         union ib_gid gid;
2003         u16 pkey;
2004         int ret;
2005         u8 p;
2006
2007         cma_dev = NULL;
2008         mutex_lock(&lock);
2009         list_for_each_entry(cur_dev, &dev_list, list) {
2010                 if (cma_family(id_priv) == AF_IB &&
2011                     rdma_node_get_transport(cur_dev->device->node_type) != RDMA_TRANSPORT_IB)
2012                         continue;
2013
2014                 if (!cma_dev)
2015                         cma_dev = cur_dev;
2016
2017                 for (p = 1; p <= cur_dev->device->phys_port_cnt; ++p) {
2018                         if (!ib_query_port(cur_dev->device, p, &port_attr) &&
2019                             port_attr.state == IB_PORT_ACTIVE) {
2020                                 cma_dev = cur_dev;
2021                                 goto port_found;
2022                         }
2023                 }
2024         }
2025
2026         if (!cma_dev) {
2027                 ret = -ENODEV;
2028                 goto out;
2029         }
2030
2031         p = 1;
2032
2033 port_found:
2034         ret = ib_get_cached_gid(cma_dev->device, p, 0, &gid);
2035         if (ret)
2036                 goto out;
2037
2038         ret = ib_get_cached_pkey(cma_dev->device, p, 0, &pkey);
2039         if (ret)
2040                 goto out;
2041
2042         id_priv->id.route.addr.dev_addr.dev_type =
2043                 (rdma_port_get_link_layer(cma_dev->device, p) == IB_LINK_LAYER_INFINIBAND) ?
2044                 ARPHRD_INFINIBAND : ARPHRD_ETHER;
2045
2046         rdma_addr_set_sgid(&id_priv->id.route.addr.dev_addr, &gid);
2047         ib_addr_set_pkey(&id_priv->id.route.addr.dev_addr, pkey);
2048         id_priv->id.port_num = p;
2049         cma_attach_to_dev(id_priv, cma_dev);
2050         cma_set_loopback(cma_src_addr(id_priv));
2051 out:
2052         mutex_unlock(&lock);
2053         return ret;
2054 }
2055
2056 static void addr_handler(int status, struct sockaddr *src_addr,
2057                          struct rdma_dev_addr *dev_addr, void *context)
2058 {
2059         struct rdma_id_private *id_priv = context;
2060         struct rdma_cm_event event;
2061
2062         memset(&event, 0, sizeof event);
2063         mutex_lock(&id_priv->handler_mutex);
2064         if (!cma_comp_exch(id_priv, RDMA_CM_ADDR_QUERY,
2065                            RDMA_CM_ADDR_RESOLVED))
2066                 goto out;
2067
2068         if (!status && !id_priv->cma_dev)
2069                 status = cma_acquire_dev(id_priv);
2070
2071         if (status) {
2072                 if (!cma_comp_exch(id_priv, RDMA_CM_ADDR_RESOLVED,
2073                                    RDMA_CM_ADDR_BOUND))
2074                         goto out;
2075                 event.event = RDMA_CM_EVENT_ADDR_ERROR;
2076                 event.status = status;
2077         } else {
2078                 memcpy(cma_src_addr(id_priv), src_addr, rdma_addr_size(src_addr));
2079                 event.event = RDMA_CM_EVENT_ADDR_RESOLVED;
2080         }
2081
2082         if (id_priv->id.event_handler(&id_priv->id, &event)) {
2083                 cma_exch(id_priv, RDMA_CM_DESTROYING);
2084                 mutex_unlock(&id_priv->handler_mutex);
2085                 cma_deref_id(id_priv);
2086                 rdma_destroy_id(&id_priv->id);
2087                 return;
2088         }
2089 out:
2090         mutex_unlock(&id_priv->handler_mutex);
2091         cma_deref_id(id_priv);
2092 }
2093
2094 static int cma_resolve_loopback(struct rdma_id_private *id_priv)
2095 {
2096         struct cma_work *work;
2097         union ib_gid gid;
2098         int ret;
2099
2100         work = kzalloc(sizeof *work, GFP_KERNEL);
2101         if (!work)
2102                 return -ENOMEM;
2103
2104         if (!id_priv->cma_dev) {
2105                 ret = cma_bind_loopback(id_priv);
2106                 if (ret)
2107                         goto err;
2108         }
2109
2110         rdma_addr_get_sgid(&id_priv->id.route.addr.dev_addr, &gid);
2111         rdma_addr_set_dgid(&id_priv->id.route.addr.dev_addr, &gid);
2112
2113         work->id = id_priv;
2114         INIT_WORK(&work->work, cma_work_handler);
2115         work->old_state = RDMA_CM_ADDR_QUERY;
2116         work->new_state = RDMA_CM_ADDR_RESOLVED;
2117         work->event.event = RDMA_CM_EVENT_ADDR_RESOLVED;
2118         queue_work(cma_wq, &work->work);
2119         return 0;
2120 err:
2121         kfree(work);
2122         return ret;
2123 }
2124
2125 static int cma_resolve_ib_addr(struct rdma_id_private *id_priv)
2126 {
2127         struct cma_work *work;
2128         int ret;
2129
2130         work = kzalloc(sizeof *work, GFP_KERNEL);
2131         if (!work)
2132                 return -ENOMEM;
2133
2134         if (!id_priv->cma_dev) {
2135                 ret = cma_resolve_ib_dev(id_priv);
2136                 if (ret)
2137                         goto err;
2138         }
2139
2140         rdma_addr_set_dgid(&id_priv->id.route.addr.dev_addr, (union ib_gid *)
2141                 &(((struct sockaddr_ib *) &id_priv->id.route.addr.dst_addr)->sib_addr));
2142
2143         work->id = id_priv;
2144         INIT_WORK(&work->work, cma_work_handler);
2145         work->old_state = RDMA_CM_ADDR_QUERY;
2146         work->new_state = RDMA_CM_ADDR_RESOLVED;
2147         work->event.event = RDMA_CM_EVENT_ADDR_RESOLVED;
2148         queue_work(cma_wq, &work->work);
2149         return 0;
2150 err:
2151         kfree(work);
2152         return ret;
2153 }
2154
2155 static int cma_bind_addr(struct rdma_cm_id *id, struct sockaddr *src_addr,
2156                          struct sockaddr *dst_addr)
2157 {
2158         if (!src_addr || !src_addr->sa_family) {
2159                 src_addr = (struct sockaddr *) &id->route.addr.src_addr;
2160                 src_addr->sa_family = dst_addr->sa_family;
2161                 if (dst_addr->sa_family == AF_INET6) {
2162                         ((struct sockaddr_in6 *) src_addr)->sin6_scope_id =
2163                                 ((struct sockaddr_in6 *) dst_addr)->sin6_scope_id;
2164                 } else if (dst_addr->sa_family == AF_IB) {
2165                         ((struct sockaddr_ib *) src_addr)->sib_pkey =
2166                                 ((struct sockaddr_ib *) dst_addr)->sib_pkey;
2167                 }
2168         }
2169         return rdma_bind_addr(id, src_addr);
2170 }
2171
2172 int rdma_resolve_addr(struct rdma_cm_id *id, struct sockaddr *src_addr,
2173                       struct sockaddr *dst_addr, int timeout_ms)
2174 {
2175         struct rdma_id_private *id_priv;
2176         int ret;
2177
2178         id_priv = container_of(id, struct rdma_id_private, id);
2179         if (id_priv->state == RDMA_CM_IDLE) {
2180                 ret = cma_bind_addr(id, src_addr, dst_addr);
2181                 if (ret)
2182                         return ret;
2183         }
2184
2185         if (cma_family(id_priv) != dst_addr->sa_family)
2186                 return -EINVAL;
2187
2188         if (!cma_comp_exch(id_priv, RDMA_CM_ADDR_BOUND, RDMA_CM_ADDR_QUERY))
2189                 return -EINVAL;
2190
2191         atomic_inc(&id_priv->refcount);
2192         memcpy(cma_dst_addr(id_priv), dst_addr, rdma_addr_size(dst_addr));
2193         if (cma_any_addr(dst_addr)) {
2194                 ret = cma_resolve_loopback(id_priv);
2195         } else {
2196                 if (dst_addr->sa_family == AF_IB) {
2197                         ret = cma_resolve_ib_addr(id_priv);
2198                 } else {
2199                         ret = rdma_resolve_ip(&addr_client, cma_src_addr(id_priv),
2200                                               dst_addr, &id->route.addr.dev_addr,
2201                                               timeout_ms, addr_handler, id_priv);
2202                 }
2203         }
2204         if (ret)
2205                 goto err;
2206
2207         return 0;
2208 err:
2209         cma_comp_exch(id_priv, RDMA_CM_ADDR_QUERY, RDMA_CM_ADDR_BOUND);
2210         cma_deref_id(id_priv);
2211         return ret;
2212 }
2213 EXPORT_SYMBOL(rdma_resolve_addr);
2214
2215 int rdma_set_reuseaddr(struct rdma_cm_id *id, int reuse)
2216 {
2217         struct rdma_id_private *id_priv;
2218         unsigned long flags;
2219         int ret;
2220
2221         id_priv = container_of(id, struct rdma_id_private, id);
2222         spin_lock_irqsave(&id_priv->lock, flags);
2223         if (reuse || id_priv->state == RDMA_CM_IDLE) {
2224                 id_priv->reuseaddr = reuse;
2225                 ret = 0;
2226         } else {
2227                 ret = -EINVAL;
2228         }
2229         spin_unlock_irqrestore(&id_priv->lock, flags);
2230         return ret;
2231 }
2232 EXPORT_SYMBOL(rdma_set_reuseaddr);
2233
2234 int rdma_set_afonly(struct rdma_cm_id *id, int afonly)
2235 {
2236         struct rdma_id_private *id_priv;
2237         unsigned long flags;
2238         int ret;
2239
2240         id_priv = container_of(id, struct rdma_id_private, id);
2241         spin_lock_irqsave(&id_priv->lock, flags);
2242         if (id_priv->state == RDMA_CM_IDLE || id_priv->state == RDMA_CM_ADDR_BOUND) {
2243                 id_priv->options |= (1 << CMA_OPTION_AFONLY);
2244                 id_priv->afonly = afonly;
2245                 ret = 0;
2246         } else {
2247                 ret = -EINVAL;
2248         }
2249         spin_unlock_irqrestore(&id_priv->lock, flags);
2250         return ret;
2251 }
2252 EXPORT_SYMBOL(rdma_set_afonly);
2253
2254 static void cma_bind_port(struct rdma_bind_list *bind_list,
2255                           struct rdma_id_private *id_priv)
2256 {
2257         struct sockaddr *addr;
2258         struct sockaddr_ib *sib;
2259         u64 sid, mask;
2260         __be16 port;
2261
2262         addr = cma_src_addr(id_priv);
2263         port = htons(bind_list->port);
2264
2265         switch (addr->sa_family) {
2266         case AF_INET:
2267                 ((struct sockaddr_in *) addr)->sin_port = port;
2268                 break;
2269         case AF_INET6:
2270                 ((struct sockaddr_in6 *) addr)->sin6_port = port;
2271                 break;
2272         case AF_IB:
2273                 sib = (struct sockaddr_ib *) addr;
2274                 sid = be64_to_cpu(sib->sib_sid);
2275                 mask = be64_to_cpu(sib->sib_sid_mask);
2276                 sib->sib_sid = cpu_to_be64((sid & mask) | (u64) ntohs(port));
2277                 sib->sib_sid_mask = cpu_to_be64(~0ULL);
2278                 break;
2279         }
2280         id_priv->bind_list = bind_list;
2281         hlist_add_head(&id_priv->node, &bind_list->owners);
2282 }
2283
2284 static int cma_alloc_port(struct idr *ps, struct rdma_id_private *id_priv,
2285                           unsigned short snum)
2286 {
2287         struct rdma_bind_list *bind_list;
2288         int ret;
2289
2290         bind_list = kzalloc(sizeof *bind_list, GFP_KERNEL);
2291         if (!bind_list)
2292                 return -ENOMEM;
2293
2294         ret = idr_alloc(ps, bind_list, snum, snum + 1, GFP_KERNEL);
2295         if (ret < 0)
2296                 goto err;
2297
2298         bind_list->ps = ps;
2299         bind_list->port = (unsigned short)ret;
2300         cma_bind_port(bind_list, id_priv);
2301         return 0;
2302 err:
2303         kfree(bind_list);
2304         return ret == -ENOSPC ? -EADDRNOTAVAIL : ret;
2305 }
2306
2307 static int cma_alloc_any_port(struct idr *ps, struct rdma_id_private *id_priv)
2308 {
2309         static unsigned int last_used_port;
2310         int low, high, remaining;
2311         unsigned int rover;
2312
2313         inet_get_local_port_range(&init_net, &low, &high);
2314         remaining = (high - low) + 1;
2315         rover = net_random() % remaining + low;
2316 retry:
2317         if (last_used_port != rover &&
2318             !idr_find(ps, (unsigned short) rover)) {
2319                 int ret = cma_alloc_port(ps, id_priv, rover);
2320                 /*
2321                  * Remember previously used port number in order to avoid
2322                  * re-using same port immediately after it is closed.
2323                  */
2324                 if (!ret)
2325                         last_used_port = rover;
2326                 if (ret != -EADDRNOTAVAIL)
2327                         return ret;
2328         }
2329         if (--remaining) {
2330                 rover++;
2331                 if ((rover < low) || (rover > high))
2332                         rover = low;
2333                 goto retry;
2334         }
2335         return -EADDRNOTAVAIL;
2336 }
2337
2338 /*
2339  * Check that the requested port is available.  This is called when trying to
2340  * bind to a specific port, or when trying to listen on a bound port.  In
2341  * the latter case, the provided id_priv may already be on the bind_list, but
2342  * we still need to check that it's okay to start listening.
2343  */
2344 static int cma_check_port(struct rdma_bind_list *bind_list,
2345                           struct rdma_id_private *id_priv, uint8_t reuseaddr)
2346 {
2347         struct rdma_id_private *cur_id;
2348         struct sockaddr *addr, *cur_addr;
2349
2350         addr = cma_src_addr(id_priv);
2351         hlist_for_each_entry(cur_id, &bind_list->owners, node) {
2352                 if (id_priv == cur_id)
2353                         continue;
2354
2355                 if ((cur_id->state != RDMA_CM_LISTEN) && reuseaddr &&
2356                     cur_id->reuseaddr)
2357                         continue;
2358
2359                 cur_addr = cma_src_addr(cur_id);
2360                 if (id_priv->afonly && cur_id->afonly &&
2361                     (addr->sa_family != cur_addr->sa_family))
2362                         continue;
2363
2364                 if (cma_any_addr(addr) || cma_any_addr(cur_addr))
2365                         return -EADDRNOTAVAIL;
2366
2367                 if (!cma_addr_cmp(addr, cur_addr))
2368                         return -EADDRINUSE;
2369         }
2370         return 0;
2371 }
2372
2373 static int cma_use_port(struct idr *ps, struct rdma_id_private *id_priv)
2374 {
2375         struct rdma_bind_list *bind_list;
2376         unsigned short snum;
2377         int ret;
2378
2379         snum = ntohs(cma_port(cma_src_addr(id_priv)));
2380         if (snum < PROT_SOCK && !capable(CAP_NET_BIND_SERVICE))
2381                 return -EACCES;
2382
2383         bind_list = idr_find(ps, snum);
2384         if (!bind_list) {
2385                 ret = cma_alloc_port(ps, id_priv, snum);
2386         } else {
2387                 ret = cma_check_port(bind_list, id_priv, id_priv->reuseaddr);
2388                 if (!ret)
2389                         cma_bind_port(bind_list, id_priv);
2390         }
2391         return ret;
2392 }
2393
2394 static int cma_bind_listen(struct rdma_id_private *id_priv)
2395 {
2396         struct rdma_bind_list *bind_list = id_priv->bind_list;
2397         int ret = 0;
2398
2399         mutex_lock(&lock);
2400         if (bind_list->owners.first->next)
2401                 ret = cma_check_port(bind_list, id_priv, 0);
2402         mutex_unlock(&lock);
2403         return ret;
2404 }
2405
2406 static struct idr *cma_select_inet_ps(struct rdma_id_private *id_priv)
2407 {
2408         switch (id_priv->id.ps) {
2409         case RDMA_PS_TCP:
2410                 return &tcp_ps;
2411         case RDMA_PS_UDP:
2412                 return &udp_ps;
2413         case RDMA_PS_IPOIB:
2414                 return &ipoib_ps;
2415         case RDMA_PS_IB:
2416                 return &ib_ps;
2417         default:
2418                 return NULL;
2419         }
2420 }
2421
2422 static struct idr *cma_select_ib_ps(struct rdma_id_private *id_priv)
2423 {
2424         struct idr *ps = NULL;
2425         struct sockaddr_ib *sib;
2426         u64 sid_ps, mask, sid;
2427
2428         sib = (struct sockaddr_ib *) cma_src_addr(id_priv);
2429         mask = be64_to_cpu(sib->sib_sid_mask) & RDMA_IB_IP_PS_MASK;
2430         sid = be64_to_cpu(sib->sib_sid) & mask;
2431
2432         if ((id_priv->id.ps == RDMA_PS_IB) && (sid == (RDMA_IB_IP_PS_IB & mask))) {
2433                 sid_ps = RDMA_IB_IP_PS_IB;
2434                 ps = &ib_ps;
2435         } else if (((id_priv->id.ps == RDMA_PS_IB) || (id_priv->id.ps == RDMA_PS_TCP)) &&
2436                    (sid == (RDMA_IB_IP_PS_TCP & mask))) {
2437                 sid_ps = RDMA_IB_IP_PS_TCP;
2438                 ps = &tcp_ps;
2439         } else if (((id_priv->id.ps == RDMA_PS_IB) || (id_priv->id.ps == RDMA_PS_UDP)) &&
2440                    (sid == (RDMA_IB_IP_PS_UDP & mask))) {
2441                 sid_ps = RDMA_IB_IP_PS_UDP;
2442                 ps = &udp_ps;
2443         }
2444
2445         if (ps) {
2446                 sib->sib_sid = cpu_to_be64(sid_ps | ntohs(cma_port((struct sockaddr *) sib)));
2447                 sib->sib_sid_mask = cpu_to_be64(RDMA_IB_IP_PS_MASK |
2448                                                 be64_to_cpu(sib->sib_sid_mask));
2449         }
2450         return ps;
2451 }
2452
2453 static int cma_get_port(struct rdma_id_private *id_priv)
2454 {
2455         struct idr *ps;
2456         int ret;
2457
2458         if (cma_family(id_priv) != AF_IB)
2459                 ps = cma_select_inet_ps(id_priv);
2460         else
2461                 ps = cma_select_ib_ps(id_priv);
2462         if (!ps)
2463                 return -EPROTONOSUPPORT;
2464
2465         mutex_lock(&lock);
2466         if (cma_any_port(cma_src_addr(id_priv)))
2467                 ret = cma_alloc_any_port(ps, id_priv);
2468         else
2469                 ret = cma_use_port(ps, id_priv);
2470         mutex_unlock(&lock);
2471
2472         return ret;
2473 }
2474
2475 static int cma_check_linklocal(struct rdma_dev_addr *dev_addr,
2476                                struct sockaddr *addr)
2477 {
2478 #if IS_ENABLED(CONFIG_IPV6)
2479         struct sockaddr_in6 *sin6;
2480
2481         if (addr->sa_family != AF_INET6)
2482                 return 0;
2483
2484         sin6 = (struct sockaddr_in6 *) addr;
2485         if ((ipv6_addr_type(&sin6->sin6_addr) & IPV6_ADDR_LINKLOCAL) &&
2486             !sin6->sin6_scope_id)
2487                         return -EINVAL;
2488
2489         dev_addr->bound_dev_if = sin6->sin6_scope_id;
2490 #endif
2491         return 0;
2492 }
2493
2494 int rdma_listen(struct rdma_cm_id *id, int backlog)
2495 {
2496         struct rdma_id_private *id_priv;
2497         int ret;
2498
2499         id_priv = container_of(id, struct rdma_id_private, id);
2500         if (id_priv->state == RDMA_CM_IDLE) {
2501                 id->route.addr.src_addr.ss_family = AF_INET;
2502                 ret = rdma_bind_addr(id, cma_src_addr(id_priv));
2503                 if (ret)
2504                         return ret;
2505         }
2506
2507         if (!cma_comp_exch(id_priv, RDMA_CM_ADDR_BOUND, RDMA_CM_LISTEN))
2508                 return -EINVAL;
2509
2510         if (id_priv->reuseaddr) {
2511                 ret = cma_bind_listen(id_priv);
2512                 if (ret)
2513                         goto err;
2514         }
2515
2516         id_priv->backlog = backlog;
2517         if (id->device) {
2518                 switch (rdma_node_get_transport(id->device->node_type)) {
2519                 case RDMA_TRANSPORT_IB:
2520                         ret = cma_ib_listen(id_priv);
2521                         if (ret)
2522                                 goto err;
2523                         break;
2524                 case RDMA_TRANSPORT_IWARP:
2525                         ret = cma_iw_listen(id_priv, backlog);
2526                         if (ret)
2527                                 goto err;
2528                         break;
2529                 default:
2530                         ret = -ENOSYS;
2531                         goto err;
2532                 }
2533         } else
2534                 cma_listen_on_all(id_priv);
2535
2536         return 0;
2537 err:
2538         id_priv->backlog = 0;
2539         cma_comp_exch(id_priv, RDMA_CM_LISTEN, RDMA_CM_ADDR_BOUND);
2540         return ret;
2541 }
2542 EXPORT_SYMBOL(rdma_listen);
2543
2544 int rdma_bind_addr(struct rdma_cm_id *id, struct sockaddr *addr)
2545 {
2546         struct rdma_id_private *id_priv;
2547         int ret;
2548
2549         if (addr->sa_family != AF_INET && addr->sa_family != AF_INET6 &&
2550             addr->sa_family != AF_IB)
2551                 return -EAFNOSUPPORT;
2552
2553         id_priv = container_of(id, struct rdma_id_private, id);
2554         if (!cma_comp_exch(id_priv, RDMA_CM_IDLE, RDMA_CM_ADDR_BOUND))
2555                 return -EINVAL;
2556
2557         ret = cma_check_linklocal(&id->route.addr.dev_addr, addr);
2558         if (ret)
2559                 goto err1;
2560
2561         if (!cma_any_addr(addr)) {
2562                 ret = cma_translate_addr(addr, &id->route.addr.dev_addr);
2563                 if (ret)
2564                         goto err1;
2565
2566                 ret = cma_acquire_dev(id_priv);
2567                 if (ret)
2568                         goto err1;
2569         }
2570
2571         memcpy(cma_src_addr(id_priv), addr, rdma_addr_size(addr));
2572         if (!(id_priv->options & (1 << CMA_OPTION_AFONLY))) {
2573                 if (addr->sa_family == AF_INET)
2574                         id_priv->afonly = 1;
2575 #if IS_ENABLED(CONFIG_IPV6)
2576                 else if (addr->sa_family == AF_INET6)
2577                         id_priv->afonly = init_net.ipv6.sysctl.bindv6only;
2578 #endif
2579         }
2580         ret = cma_get_port(id_priv);
2581         if (ret)
2582                 goto err2;
2583
2584         return 0;
2585 err2:
2586         if (id_priv->cma_dev)
2587                 cma_release_dev(id_priv);
2588 err1:
2589         cma_comp_exch(id_priv, RDMA_CM_ADDR_BOUND, RDMA_CM_IDLE);
2590         return ret;
2591 }
2592 EXPORT_SYMBOL(rdma_bind_addr);
2593
2594 static int cma_format_hdr(void *hdr, struct rdma_id_private *id_priv)
2595 {
2596         struct cma_hdr *cma_hdr;
2597
2598         cma_hdr = hdr;
2599         cma_hdr->cma_version = CMA_VERSION;
2600         if (cma_family(id_priv) == AF_INET) {
2601                 struct sockaddr_in *src4, *dst4;
2602
2603                 src4 = (struct sockaddr_in *) cma_src_addr(id_priv);
2604                 dst4 = (struct sockaddr_in *) cma_dst_addr(id_priv);
2605
2606                 cma_set_ip_ver(cma_hdr, 4);
2607                 cma_hdr->src_addr.ip4.addr = src4->sin_addr.s_addr;
2608                 cma_hdr->dst_addr.ip4.addr = dst4->sin_addr.s_addr;
2609                 cma_hdr->port = src4->sin_port;
2610         } else if (cma_family(id_priv) == AF_INET6) {
2611                 struct sockaddr_in6 *src6, *dst6;
2612
2613                 src6 = (struct sockaddr_in6 *) cma_src_addr(id_priv);
2614                 dst6 = (struct sockaddr_in6 *) cma_dst_addr(id_priv);
2615
2616                 cma_set_ip_ver(cma_hdr, 6);
2617                 cma_hdr->src_addr.ip6 = src6->sin6_addr;
2618                 cma_hdr->dst_addr.ip6 = dst6->sin6_addr;
2619                 cma_hdr->port = src6->sin6_port;
2620         }
2621         return 0;
2622 }
2623
2624 static int cma_sidr_rep_handler(struct ib_cm_id *cm_id,
2625                                 struct ib_cm_event *ib_event)
2626 {
2627         struct rdma_id_private *id_priv = cm_id->context;
2628         struct rdma_cm_event event;
2629         struct ib_cm_sidr_rep_event_param *rep = &ib_event->param.sidr_rep_rcvd;
2630         int ret = 0;
2631
2632         if (cma_disable_callback(id_priv, RDMA_CM_CONNECT))
2633                 return 0;
2634
2635         memset(&event, 0, sizeof event);
2636         switch (ib_event->event) {
2637         case IB_CM_SIDR_REQ_ERROR:
2638                 event.event = RDMA_CM_EVENT_UNREACHABLE;
2639                 event.status = -ETIMEDOUT;
2640                 break;
2641         case IB_CM_SIDR_REP_RECEIVED:
2642                 event.param.ud.private_data = ib_event->private_data;
2643                 event.param.ud.private_data_len = IB_CM_SIDR_REP_PRIVATE_DATA_SIZE;
2644                 if (rep->status != IB_SIDR_SUCCESS) {
2645                         event.event = RDMA_CM_EVENT_UNREACHABLE;
2646                         event.status = ib_event->param.sidr_rep_rcvd.status;
2647                         break;
2648                 }
2649                 ret = cma_set_qkey(id_priv, rep->qkey);
2650                 if (ret) {
2651                         event.event = RDMA_CM_EVENT_ADDR_ERROR;
2652                         event.status = ret;
2653                         break;
2654                 }
2655                 ib_init_ah_from_path(id_priv->id.device, id_priv->id.port_num,
2656                                      id_priv->id.route.path_rec,
2657                                      &event.param.ud.ah_attr);
2658                 event.param.ud.qp_num = rep->qpn;
2659                 event.param.ud.qkey = rep->qkey;
2660                 event.event = RDMA_CM_EVENT_ESTABLISHED;
2661                 event.status = 0;
2662                 break;
2663         default:
2664                 printk(KERN_ERR "RDMA CMA: unexpected IB CM event: %d\n",
2665                        ib_event->event);
2666                 goto out;
2667         }
2668
2669         ret = id_priv->id.event_handler(&id_priv->id, &event);
2670         if (ret) {
2671                 /* Destroy the CM ID by returning a non-zero value. */
2672                 id_priv->cm_id.ib = NULL;
2673                 cma_exch(id_priv, RDMA_CM_DESTROYING);
2674                 mutex_unlock(&id_priv->handler_mutex);
2675                 rdma_destroy_id(&id_priv->id);
2676                 return ret;
2677         }
2678 out:
2679         mutex_unlock(&id_priv->handler_mutex);
2680         return ret;
2681 }
2682
2683 static int cma_resolve_ib_udp(struct rdma_id_private *id_priv,
2684                               struct rdma_conn_param *conn_param)
2685 {
2686         struct ib_cm_sidr_req_param req;
2687         struct ib_cm_id *id;
2688         void *private_data;
2689         int offset, ret;
2690
2691         memset(&req, 0, sizeof req);
2692         offset = cma_user_data_offset(id_priv);
2693         req.private_data_len = offset + conn_param->private_data_len;
2694         if (req.private_data_len < conn_param->private_data_len)
2695                 return -EINVAL;
2696
2697         if (req.private_data_len) {
2698                 private_data = kzalloc(req.private_data_len, GFP_ATOMIC);
2699                 if (!private_data)
2700                         return -ENOMEM;
2701         } else {
2702                 private_data = NULL;
2703         }
2704
2705         if (conn_param->private_data && conn_param->private_data_len)
2706                 memcpy(private_data + offset, conn_param->private_data,
2707                        conn_param->private_data_len);
2708
2709         if (private_data) {
2710                 ret = cma_format_hdr(private_data, id_priv);
2711                 if (ret)
2712                         goto out;
2713                 req.private_data = private_data;
2714         }
2715
2716         id = ib_create_cm_id(id_priv->id.device, cma_sidr_rep_handler,
2717                              id_priv);
2718         if (IS_ERR(id)) {
2719                 ret = PTR_ERR(id);
2720                 goto out;
2721         }
2722         id_priv->cm_id.ib = id;
2723
2724         req.path = id_priv->id.route.path_rec;
2725         req.service_id = rdma_get_service_id(&id_priv->id, cma_dst_addr(id_priv));
2726         req.timeout_ms = 1 << (CMA_CM_RESPONSE_TIMEOUT - 8);
2727         req.max_cm_retries = CMA_MAX_CM_RETRIES;
2728
2729         ret = ib_send_cm_sidr_req(id_priv->cm_id.ib, &req);
2730         if (ret) {
2731                 ib_destroy_cm_id(id_priv->cm_id.ib);
2732                 id_priv->cm_id.ib = NULL;
2733         }
2734 out:
2735         kfree(private_data);
2736         return ret;
2737 }
2738
2739 static int cma_connect_ib(struct rdma_id_private *id_priv,
2740                           struct rdma_conn_param *conn_param)
2741 {
2742         struct ib_cm_req_param req;
2743         struct rdma_route *route;
2744         void *private_data;
2745         struct ib_cm_id *id;
2746         int offset, ret;
2747
2748         memset(&req, 0, sizeof req);
2749         offset = cma_user_data_offset(id_priv);
2750         req.private_data_len = offset + conn_param->private_data_len;
2751         if (req.private_data_len < conn_param->private_data_len)
2752                 return -EINVAL;
2753
2754         if (req.private_data_len) {
2755                 private_data = kzalloc(req.private_data_len, GFP_ATOMIC);
2756                 if (!private_data)
2757                         return -ENOMEM;
2758         } else {
2759                 private_data = NULL;
2760         }
2761
2762         if (conn_param->private_data && conn_param->private_data_len)
2763                 memcpy(private_data + offset, conn_param->private_data,
2764                        conn_param->private_data_len);
2765
2766         id = ib_create_cm_id(id_priv->id.device, cma_ib_handler, id_priv);
2767         if (IS_ERR(id)) {
2768                 ret = PTR_ERR(id);
2769                 goto out;
2770         }
2771         id_priv->cm_id.ib = id;
2772
2773         route = &id_priv->id.route;
2774         if (private_data) {
2775                 ret = cma_format_hdr(private_data, id_priv);
2776                 if (ret)
2777                         goto out;
2778                 req.private_data = private_data;
2779         }
2780
2781         req.primary_path = &route->path_rec[0];
2782         if (route->num_paths == 2)
2783                 req.alternate_path = &route->path_rec[1];
2784
2785         req.service_id = rdma_get_service_id(&id_priv->id, cma_dst_addr(id_priv));
2786         req.qp_num = id_priv->qp_num;
2787         req.qp_type = id_priv->id.qp_type;
2788         req.starting_psn = id_priv->seq_num;
2789         req.responder_resources = conn_param->responder_resources;
2790         req.initiator_depth = conn_param->initiator_depth;
2791         req.flow_control = conn_param->flow_control;
2792         req.retry_count = min_t(u8, 7, conn_param->retry_count);
2793         req.rnr_retry_count = min_t(u8, 7, conn_param->rnr_retry_count);
2794         req.remote_cm_response_timeout = CMA_CM_RESPONSE_TIMEOUT;
2795         req.local_cm_response_timeout = CMA_CM_RESPONSE_TIMEOUT;
2796         req.max_cm_retries = CMA_MAX_CM_RETRIES;
2797         req.srq = id_priv->srq ? 1 : 0;
2798
2799         ret = ib_send_cm_req(id_priv->cm_id.ib, &req);
2800 out:
2801         if (ret && !IS_ERR(id)) {
2802                 ib_destroy_cm_id(id);
2803                 id_priv->cm_id.ib = NULL;
2804         }
2805
2806         kfree(private_data);
2807         return ret;
2808 }
2809
2810 static int cma_connect_iw(struct rdma_id_private *id_priv,
2811                           struct rdma_conn_param *conn_param)
2812 {
2813         struct iw_cm_id *cm_id;
2814         int ret;
2815         struct iw_cm_conn_param iw_param;
2816
2817         cm_id = iw_create_cm_id(id_priv->id.device, cma_iw_handler, id_priv);
2818         if (IS_ERR(cm_id))
2819                 return PTR_ERR(cm_id);
2820
2821         id_priv->cm_id.iw = cm_id;
2822
2823         memcpy(&cm_id->local_addr, cma_src_addr(id_priv),
2824                rdma_addr_size(cma_src_addr(id_priv)));
2825         memcpy(&cm_id->remote_addr, cma_dst_addr(id_priv),
2826                rdma_addr_size(cma_dst_addr(id_priv)));
2827
2828         ret = cma_modify_qp_rtr(id_priv, conn_param);
2829         if (ret)
2830                 goto out;
2831
2832         if (conn_param) {
2833                 iw_param.ord = conn_param->initiator_depth;
2834                 iw_param.ird = conn_param->responder_resources;
2835                 iw_param.private_data = conn_param->private_data;
2836                 iw_param.private_data_len = conn_param->private_data_len;
2837                 iw_param.qpn = id_priv->id.qp ? id_priv->qp_num : conn_param->qp_num;
2838         } else {
2839                 memset(&iw_param, 0, sizeof iw_param);
2840                 iw_param.qpn = id_priv->qp_num;
2841         }
2842         ret = iw_cm_connect(cm_id, &iw_param);
2843 out:
2844         if (ret) {
2845                 iw_destroy_cm_id(cm_id);
2846                 id_priv->cm_id.iw = NULL;
2847         }
2848         return ret;
2849 }
2850
2851 int rdma_connect(struct rdma_cm_id *id, struct rdma_conn_param *conn_param)
2852 {
2853         struct rdma_id_private *id_priv;
2854         int ret;
2855
2856         id_priv = container_of(id, struct rdma_id_private, id);
2857         if (!cma_comp_exch(id_priv, RDMA_CM_ROUTE_RESOLVED, RDMA_CM_CONNECT))
2858                 return -EINVAL;
2859
2860         if (!id->qp) {
2861                 id_priv->qp_num = conn_param->qp_num;
2862                 id_priv->srq = conn_param->srq;
2863         }
2864
2865         switch (rdma_node_get_transport(id->device->node_type)) {
2866         case RDMA_TRANSPORT_IB:
2867                 if (id->qp_type == IB_QPT_UD)
2868                         ret = cma_resolve_ib_udp(id_priv, conn_param);
2869                 else
2870                         ret = cma_connect_ib(id_priv, conn_param);
2871                 break;
2872         case RDMA_TRANSPORT_IWARP:
2873                 ret = cma_connect_iw(id_priv, conn_param);
2874                 break;
2875         default:
2876                 ret = -ENOSYS;
2877                 break;
2878         }
2879         if (ret)
2880                 goto err;
2881
2882         return 0;
2883 err:
2884         cma_comp_exch(id_priv, RDMA_CM_CONNECT, RDMA_CM_ROUTE_RESOLVED);
2885         return ret;
2886 }
2887 EXPORT_SYMBOL(rdma_connect);
2888
2889 static int cma_accept_ib(struct rdma_id_private *id_priv,
2890                          struct rdma_conn_param *conn_param)
2891 {
2892         struct ib_cm_rep_param rep;
2893         int ret;
2894
2895         ret = cma_modify_qp_rtr(id_priv, conn_param);
2896         if (ret)
2897                 goto out;
2898
2899         ret = cma_modify_qp_rts(id_priv, conn_param);
2900         if (ret)
2901                 goto out;
2902
2903         memset(&rep, 0, sizeof rep);
2904         rep.qp_num = id_priv->qp_num;
2905         rep.starting_psn = id_priv->seq_num;
2906         rep.private_data = conn_param->private_data;
2907         rep.private_data_len = conn_param->private_data_len;
2908         rep.responder_resources = conn_param->responder_resources;
2909         rep.initiator_depth = conn_param->initiator_depth;
2910         rep.failover_accepted = 0;
2911         rep.flow_control = conn_param->flow_control;
2912         rep.rnr_retry_count = min_t(u8, 7, conn_param->rnr_retry_count);
2913         rep.srq = id_priv->srq ? 1 : 0;
2914
2915         ret = ib_send_cm_rep(id_priv->cm_id.ib, &rep);
2916 out:
2917         return ret;
2918 }
2919
2920 static int cma_accept_iw(struct rdma_id_private *id_priv,
2921                   struct rdma_conn_param *conn_param)
2922 {
2923         struct iw_cm_conn_param iw_param;
2924         int ret;
2925
2926         ret = cma_modify_qp_rtr(id_priv, conn_param);
2927         if (ret)
2928                 return ret;
2929
2930         iw_param.ord = conn_param->initiator_depth;
2931         iw_param.ird = conn_param->responder_resources;
2932         iw_param.private_data = conn_param->private_data;
2933         iw_param.private_data_len = conn_param->private_data_len;
2934         if (id_priv->id.qp) {
2935                 iw_param.qpn = id_priv->qp_num;
2936         } else
2937                 iw_param.qpn = conn_param->qp_num;
2938
2939         return iw_cm_accept(id_priv->cm_id.iw, &iw_param);
2940 }
2941
2942 static int cma_send_sidr_rep(struct rdma_id_private *id_priv,
2943                              enum ib_cm_sidr_status status, u32 qkey,
2944                              const void *private_data, int private_data_len)
2945 {
2946         struct ib_cm_sidr_rep_param rep;
2947         int ret;
2948
2949         memset(&rep, 0, sizeof rep);
2950         rep.status = status;
2951         if (status == IB_SIDR_SUCCESS) {
2952                 ret = cma_set_qkey(id_priv, qkey);
2953                 if (ret)
2954                         return ret;
2955                 rep.qp_num = id_priv->qp_num;
2956                 rep.qkey = id_priv->qkey;
2957         }
2958         rep.private_data = private_data;
2959         rep.private_data_len = private_data_len;
2960
2961         return ib_send_cm_sidr_rep(id_priv->cm_id.ib, &rep);
2962 }
2963
2964 int rdma_accept(struct rdma_cm_id *id, struct rdma_conn_param *conn_param)
2965 {
2966         struct rdma_id_private *id_priv;
2967         int ret;
2968
2969         id_priv = container_of(id, struct rdma_id_private, id);
2970
2971         id_priv->owner = task_pid_nr(current);
2972
2973         if (!cma_comp(id_priv, RDMA_CM_CONNECT))
2974                 return -EINVAL;
2975
2976         if (!id->qp && conn_param) {
2977                 id_priv->qp_num = conn_param->qp_num;
2978                 id_priv->srq = conn_param->srq;
2979         }
2980
2981         switch (rdma_node_get_transport(id->device->node_type)) {
2982         case RDMA_TRANSPORT_IB:
2983                 if (id->qp_type == IB_QPT_UD) {
2984                         if (conn_param)
2985                                 ret = cma_send_sidr_rep(id_priv, IB_SIDR_SUCCESS,
2986                                                         conn_param->qkey,
2987                                                         conn_param->private_data,
2988                                                         conn_param->private_data_len);
2989                         else
2990                                 ret = cma_send_sidr_rep(id_priv, IB_SIDR_SUCCESS,
2991                                                         0, NULL, 0);
2992                 } else {
2993                         if (conn_param)
2994                                 ret = cma_accept_ib(id_priv, conn_param);
2995                         else
2996                                 ret = cma_rep_recv(id_priv);
2997                 }
2998                 break;
2999         case RDMA_TRANSPORT_IWARP:
3000                 ret = cma_accept_iw(id_priv, conn_param);
3001                 break;
3002         default:
3003                 ret = -ENOSYS;
3004                 break;
3005         }
3006
3007         if (ret)
3008                 goto reject;
3009
3010         return 0;
3011 reject:
3012         cma_modify_qp_err(id_priv);
3013         rdma_reject(id, NULL, 0);
3014         return ret;
3015 }
3016 EXPORT_SYMBOL(rdma_accept);
3017
3018 int rdma_notify(struct rdma_cm_id *id, enum ib_event_type event)
3019 {
3020         struct rdma_id_private *id_priv;
3021         int ret;
3022
3023         id_priv = container_of(id, struct rdma_id_private, id);
3024         if (!id_priv->cm_id.ib)
3025                 return -EINVAL;
3026
3027         switch (id->device->node_type) {
3028         case RDMA_NODE_IB_CA:
3029                 ret = ib_cm_notify(id_priv->cm_id.ib, event);
3030                 break;
3031         default:
3032                 ret = 0;
3033                 break;
3034         }
3035         return ret;
3036 }
3037 EXPORT_SYMBOL(rdma_notify);
3038
3039 int rdma_reject(struct rdma_cm_id *id, const void *private_data,
3040                 u8 private_data_len)
3041 {
3042         struct rdma_id_private *id_priv;
3043         int ret;
3044
3045         id_priv = container_of(id, struct rdma_id_private, id);
3046         if (!id_priv->cm_id.ib)
3047                 return -EINVAL;
3048
3049         switch (rdma_node_get_transport(id->device->node_type)) {
3050         case RDMA_TRANSPORT_IB:
3051                 if (id->qp_type == IB_QPT_UD)
3052                         ret = cma_send_sidr_rep(id_priv, IB_SIDR_REJECT, 0,
3053                                                 private_data, private_data_len);
3054                 else
3055                         ret = ib_send_cm_rej(id_priv->cm_id.ib,
3056                                              IB_CM_REJ_CONSUMER_DEFINED, NULL,
3057                                              0, private_data, private_data_len);
3058                 break;
3059         case RDMA_TRANSPORT_IWARP:
3060                 ret = iw_cm_reject(id_priv->cm_id.iw,
3061                                    private_data, private_data_len);
3062                 break;
3063         default:
3064                 ret = -ENOSYS;
3065                 break;
3066         }
3067         return ret;
3068 }
3069 EXPORT_SYMBOL(rdma_reject);
3070
3071 int rdma_disconnect(struct rdma_cm_id *id)
3072 {
3073         struct rdma_id_private *id_priv;
3074         int ret;
3075
3076         id_priv = container_of(id, struct rdma_id_private, id);
3077         if (!id_priv->cm_id.ib)
3078                 return -EINVAL;
3079
3080         switch (rdma_node_get_transport(id->device->node_type)) {
3081         case RDMA_TRANSPORT_IB:
3082                 ret = cma_modify_qp_err(id_priv);
3083                 if (ret)
3084                         goto out;
3085                 /* Initiate or respond to a disconnect. */
3086                 if (ib_send_cm_dreq(id_priv->cm_id.ib, NULL, 0))
3087                         ib_send_cm_drep(id_priv->cm_id.ib, NULL, 0);
3088                 break;
3089         case RDMA_TRANSPORT_IWARP:
3090                 ret = iw_cm_disconnect(id_priv->cm_id.iw, 0);
3091                 break;
3092         default:
3093                 ret = -EINVAL;
3094                 break;
3095         }
3096 out:
3097         return ret;
3098 }
3099 EXPORT_SYMBOL(rdma_disconnect);
3100
3101 static int cma_ib_mc_handler(int status, struct ib_sa_multicast *multicast)
3102 {
3103         struct rdma_id_private *id_priv;
3104         struct cma_multicast *mc = multicast->context;
3105         struct rdma_cm_event event;
3106         int ret;
3107
3108         id_priv = mc->id_priv;
3109         if (cma_disable_callback(id_priv, RDMA_CM_ADDR_BOUND) &&
3110             cma_disable_callback(id_priv, RDMA_CM_ADDR_RESOLVED))
3111                 return 0;
3112
3113         if (!status)
3114                 status = cma_set_qkey(id_priv, be32_to_cpu(multicast->rec.qkey));
3115         mutex_lock(&id_priv->qp_mutex);
3116         if (!status && id_priv->id.qp)
3117                 status = ib_attach_mcast(id_priv->id.qp, &multicast->rec.mgid,
3118                                          be16_to_cpu(multicast->rec.mlid));
3119         mutex_unlock(&id_priv->qp_mutex);
3120
3121         memset(&event, 0, sizeof event);
3122         event.status = status;
3123         event.param.ud.private_data = mc->context;
3124         if (!status) {
3125                 event.event = RDMA_CM_EVENT_MULTICAST_JOIN;
3126                 ib_init_ah_from_mcmember(id_priv->id.device,
3127                                          id_priv->id.port_num, &multicast->rec,
3128                                          &event.param.ud.ah_attr);
3129                 event.param.ud.qp_num = 0xFFFFFF;
3130                 event.param.ud.qkey = be32_to_cpu(multicast->rec.qkey);
3131         } else
3132                 event.event = RDMA_CM_EVENT_MULTICAST_ERROR;
3133
3134         ret = id_priv->id.event_handler(&id_priv->id, &event);
3135         if (ret) {
3136                 cma_exch(id_priv, RDMA_CM_DESTROYING);
3137                 mutex_unlock(&id_priv->handler_mutex);
3138                 rdma_destroy_id(&id_priv->id);
3139                 return 0;
3140         }
3141
3142         mutex_unlock(&id_priv->handler_mutex);
3143         return 0;
3144 }
3145
3146 static void cma_set_mgid(struct rdma_id_private *id_priv,
3147                          struct sockaddr *addr, union ib_gid *mgid)
3148 {
3149         unsigned char mc_map[MAX_ADDR_LEN];
3150         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
3151         struct sockaddr_in *sin = (struct sockaddr_in *) addr;
3152         struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) addr;
3153
3154         if (cma_any_addr(addr)) {
3155                 memset(mgid, 0, sizeof *mgid);
3156         } else if ((addr->sa_family == AF_INET6) &&
3157                    ((be32_to_cpu(sin6->sin6_addr.s6_addr32[0]) & 0xFFF0FFFF) ==
3158                                                                  0xFF10A01B)) {
3159                 /* IPv6 address is an SA assigned MGID. */
3160                 memcpy(mgid, &sin6->sin6_addr, sizeof *mgid);
3161         } else if (addr->sa_family == AF_IB) {
3162                 memcpy(mgid, &((struct sockaddr_ib *) addr)->sib_addr, sizeof *mgid);
3163         } else if ((addr->sa_family == AF_INET6)) {
3164                 ipv6_ib_mc_map(&sin6->sin6_addr, dev_addr->broadcast, mc_map);
3165                 if (id_priv->id.ps == RDMA_PS_UDP)
3166                         mc_map[7] = 0x01;       /* Use RDMA CM signature */
3167                 *mgid = *(union ib_gid *) (mc_map + 4);
3168         } else {
3169                 ip_ib_mc_map(sin->sin_addr.s_addr, dev_addr->broadcast, mc_map);
3170                 if (id_priv->id.ps == RDMA_PS_UDP)
3171                         mc_map[7] = 0x01;       /* Use RDMA CM signature */
3172                 *mgid = *(union ib_gid *) (mc_map + 4);
3173         }
3174 }
3175
3176 static int cma_join_ib_multicast(struct rdma_id_private *id_priv,
3177                                  struct cma_multicast *mc)
3178 {
3179         struct ib_sa_mcmember_rec rec;
3180         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
3181         ib_sa_comp_mask comp_mask;
3182         int ret;
3183
3184         ib_addr_get_mgid(dev_addr, &rec.mgid);
3185         ret = ib_sa_get_mcmember_rec(id_priv->id.device, id_priv->id.port_num,
3186                                      &rec.mgid, &rec);
3187         if (ret)
3188                 return ret;
3189
3190         ret = cma_set_qkey(id_priv, 0);
3191         if (ret)
3192                 return ret;
3193
3194         cma_set_mgid(id_priv, (struct sockaddr *) &mc->addr, &rec.mgid);
3195         rec.qkey = cpu_to_be32(id_priv->qkey);
3196         rdma_addr_get_sgid(dev_addr, &rec.port_gid);
3197         rec.pkey = cpu_to_be16(ib_addr_get_pkey(dev_addr));
3198         rec.join_state = 1;
3199
3200         comp_mask = IB_SA_MCMEMBER_REC_MGID | IB_SA_MCMEMBER_REC_PORT_GID |
3201                     IB_SA_MCMEMBER_REC_PKEY | IB_SA_MCMEMBER_REC_JOIN_STATE |
3202                     IB_SA_MCMEMBER_REC_QKEY | IB_SA_MCMEMBER_REC_SL |
3203                     IB_SA_MCMEMBER_REC_FLOW_LABEL |
3204                     IB_SA_MCMEMBER_REC_TRAFFIC_CLASS;
3205
3206         if (id_priv->id.ps == RDMA_PS_IPOIB)
3207                 comp_mask |= IB_SA_MCMEMBER_REC_RATE |
3208                              IB_SA_MCMEMBER_REC_RATE_SELECTOR |
3209                              IB_SA_MCMEMBER_REC_MTU_SELECTOR |
3210                              IB_SA_MCMEMBER_REC_MTU |
3211                              IB_SA_MCMEMBER_REC_HOP_LIMIT;
3212
3213         mc->multicast.ib = ib_sa_join_multicast(&sa_client, id_priv->id.device,
3214                                                 id_priv->id.port_num, &rec,
3215                                                 comp_mask, GFP_KERNEL,
3216                                                 cma_ib_mc_handler, mc);
3217         return PTR_ERR_OR_ZERO(mc->multicast.ib);
3218 }
3219
3220 static void iboe_mcast_work_handler(struct work_struct *work)
3221 {
3222         struct iboe_mcast_work *mw = container_of(work, struct iboe_mcast_work, work);
3223         struct cma_multicast *mc = mw->mc;
3224         struct ib_sa_multicast *m = mc->multicast.ib;
3225
3226         mc->multicast.ib->context = mc;
3227         cma_ib_mc_handler(0, m);
3228         kref_put(&mc->mcref, release_mc);
3229         kfree(mw);
3230 }
3231
3232 static void cma_iboe_set_mgid(struct sockaddr *addr, union ib_gid *mgid)
3233 {
3234         struct sockaddr_in *sin = (struct sockaddr_in *)addr;
3235         struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)addr;
3236
3237         if (cma_any_addr(addr)) {
3238                 memset(mgid, 0, sizeof *mgid);
3239         } else if (addr->sa_family == AF_INET6) {
3240                 memcpy(mgid, &sin6->sin6_addr, sizeof *mgid);
3241         } else {
3242                 mgid->raw[0] = 0xff;
3243                 mgid->raw[1] = 0x0e;
3244                 mgid->raw[2] = 0;
3245                 mgid->raw[3] = 0;
3246                 mgid->raw[4] = 0;
3247                 mgid->raw[5] = 0;
3248                 mgid->raw[6] = 0;
3249                 mgid->raw[7] = 0;
3250                 mgid->raw[8] = 0;
3251                 mgid->raw[9] = 0;
3252                 mgid->raw[10] = 0xff;
3253                 mgid->raw[11] = 0xff;
3254                 *(__be32 *)(&mgid->raw[12]) = sin->sin_addr.s_addr;
3255         }
3256 }
3257
3258 static int cma_iboe_join_multicast(struct rdma_id_private *id_priv,
3259                                    struct cma_multicast *mc)
3260 {
3261         struct iboe_mcast_work *work;
3262         struct rdma_dev_addr *dev_addr = &id_priv->id.route.addr.dev_addr;
3263         int err;
3264         struct sockaddr *addr = (struct sockaddr *)&mc->addr;
3265         struct net_device *ndev = NULL;
3266
3267         if (cma_zero_addr((struct sockaddr *)&mc->addr))
3268                 return -EINVAL;
3269
3270         work = kzalloc(sizeof *work, GFP_KERNEL);
3271         if (!work)
3272                 return -ENOMEM;
3273
3274         mc->multicast.ib = kzalloc(sizeof(struct ib_sa_multicast), GFP_KERNEL);
3275         if (!mc->multicast.ib) {
3276                 err = -ENOMEM;
3277                 goto out1;
3278         }
3279
3280         cma_iboe_set_mgid(addr, &mc->multicast.ib->rec.mgid);
3281
3282         mc->multicast.ib->rec.pkey = cpu_to_be16(0xffff);
3283         if (id_priv->id.ps == RDMA_PS_UDP)
3284                 mc->multicast.ib->rec.qkey = cpu_to_be32(RDMA_UDP_QKEY);
3285
3286         if (dev_addr->bound_dev_if)
3287                 ndev = dev_get_by_index(&init_net, dev_addr->bound_dev_if);
3288         if (!ndev) {
3289                 err = -ENODEV;
3290                 goto out2;
3291         }
3292         mc->multicast.ib->rec.rate = iboe_get_rate(ndev);
3293         mc->multicast.ib->rec.hop_limit = 1;
3294         mc->multicast.ib->rec.mtu = iboe_get_mtu(ndev->mtu);
3295         dev_put(ndev);
3296         if (!mc->multicast.ib->rec.mtu) {
3297                 err = -EINVAL;
3298                 goto out2;
3299         }
3300         iboe_addr_get_sgid(dev_addr, &mc->multicast.ib->rec.port_gid);
3301         work->id = id_priv;
3302         work->mc = mc;
3303         INIT_WORK(&work->work, iboe_mcast_work_handler);
3304         kref_get(&mc->mcref);
3305         queue_work(cma_wq, &work->work);
3306
3307         return 0;
3308
3309 out2:
3310         kfree(mc->multicast.ib);
3311 out1:
3312         kfree(work);
3313         return err;
3314 }
3315
3316 int rdma_join_multicast(struct rdma_cm_id *id, struct sockaddr *addr,
3317                         void *context)
3318 {
3319         struct rdma_id_private *id_priv;
3320         struct cma_multicast *mc;
3321         int ret;
3322
3323         id_priv = container_of(id, struct rdma_id_private, id);
3324         if (!cma_comp(id_priv, RDMA_CM_ADDR_BOUND) &&
3325             !cma_comp(id_priv, RDMA_CM_ADDR_RESOLVED))
3326                 return -EINVAL;
3327
3328         mc = kmalloc(sizeof *mc, GFP_KERNEL);
3329         if (!mc)
3330                 return -ENOMEM;
3331
3332         memcpy(&mc->addr, addr, rdma_addr_size(addr));
3333         mc->context = context;
3334         mc->id_priv = id_priv;
3335
3336         spin_lock(&id_priv->lock);
3337         list_add(&mc->list, &id_priv->mc_list);
3338         spin_unlock(&id_priv->lock);
3339
3340         switch (rdma_node_get_transport(id->device->node_type)) {
3341         case RDMA_TRANSPORT_IB:
3342                 switch (rdma_port_get_link_layer(id->device, id->port_num)) {
3343                 case IB_LINK_LAYER_INFINIBAND:
3344                         ret = cma_join_ib_multicast(id_priv, mc);
3345                         break;
3346                 case IB_LINK_LAYER_ETHERNET:
3347                         kref_init(&mc->mcref);
3348                         ret = cma_iboe_join_multicast(id_priv, mc);
3349                         break;
3350                 default:
3351                         ret = -EINVAL;
3352                 }
3353                 break;
3354         default:
3355                 ret = -ENOSYS;
3356                 break;
3357         }
3358
3359         if (ret) {
3360                 spin_lock_irq(&id_priv->lock);
3361                 list_del(&mc->list);
3362                 spin_unlock_irq(&id_priv->lock);
3363                 kfree(mc);
3364         }
3365         return ret;
3366 }
3367 EXPORT_SYMBOL(rdma_join_multicast);
3368
3369 void rdma_leave_multicast(struct rdma_cm_id *id, struct sockaddr *addr)
3370 {
3371         struct rdma_id_private *id_priv;
3372         struct cma_multicast *mc;
3373
3374         id_priv = container_of(id, struct rdma_id_private, id);
3375         spin_lock_irq(&id_priv->lock);
3376         list_for_each_entry(mc, &id_priv->mc_list, list) {
3377                 if (!memcmp(&mc->addr, addr, rdma_addr_size(addr))) {
3378                         list_del(&mc->list);
3379                         spin_unlock_irq(&id_priv->lock);
3380
3381                         if (id->qp)
3382                                 ib_detach_mcast(id->qp,
3383                                                 &mc->multicast.ib->rec.mgid,
3384                                                 be16_to_cpu(mc->multicast.ib->rec.mlid));
3385                         if (rdma_node_get_transport(id_priv->cma_dev->device->node_type) == RDMA_TRANSPORT_IB) {
3386                                 switch (rdma_port_get_link_layer(id->device, id->port_num)) {
3387                                 case IB_LINK_LAYER_INFINIBAND:
3388                                         ib_sa_free_multicast(mc->multicast.ib);
3389                                         kfree(mc);
3390                                         break;
3391                                 case IB_LINK_LAYER_ETHERNET:
3392                                         kref_put(&mc->mcref, release_mc);
3393                                         break;
3394                                 default:
3395                                         break;
3396                                 }
3397                         }
3398                         return;
3399                 }
3400         }
3401         spin_unlock_irq(&id_priv->lock);
3402 }
3403 EXPORT_SYMBOL(rdma_leave_multicast);
3404
3405 static int cma_netdev_change(struct net_device *ndev, struct rdma_id_private *id_priv)
3406 {
3407         struct rdma_dev_addr *dev_addr;
3408         struct cma_ndev_work *work;
3409
3410         dev_addr = &id_priv->id.route.addr.dev_addr;
3411
3412         if ((dev_addr->bound_dev_if == ndev->ifindex) &&
3413             memcmp(dev_addr->src_dev_addr, ndev->dev_addr, ndev->addr_len)) {
3414                 printk(KERN_INFO "RDMA CM addr change for ndev %s used by id %p\n",
3415                        ndev->name, &id_priv->id);
3416                 work = kzalloc(sizeof *work, GFP_KERNEL);
3417                 if (!work)
3418                         return -ENOMEM;
3419
3420                 INIT_WORK(&work->work, cma_ndev_work_handler);
3421                 work->id = id_priv;
3422                 work->event.event = RDMA_CM_EVENT_ADDR_CHANGE;
3423                 atomic_inc(&id_priv->refcount);
3424                 queue_work(cma_wq, &work->work);
3425         }
3426
3427         return 0;
3428 }
3429
3430 static int cma_netdev_callback(struct notifier_block *self, unsigned long event,
3431                                void *ptr)
3432 {
3433         struct net_device *ndev = netdev_notifier_info_to_dev(ptr);
3434         struct cma_device *cma_dev;
3435         struct rdma_id_private *id_priv;
3436         int ret = NOTIFY_DONE;
3437
3438         if (dev_net(ndev) != &init_net)
3439                 return NOTIFY_DONE;
3440
3441         if (event != NETDEV_BONDING_FAILOVER)
3442                 return NOTIFY_DONE;
3443
3444         if (!(ndev->flags & IFF_MASTER) || !(ndev->priv_flags & IFF_BONDING))
3445                 return NOTIFY_DONE;
3446
3447         mutex_lock(&lock);
3448         list_for_each_entry(cma_dev, &dev_list, list)
3449                 list_for_each_entry(id_priv, &cma_dev->id_list, list) {
3450                         ret = cma_netdev_change(ndev, id_priv);
3451                         if (ret)
3452                                 goto out;
3453                 }
3454
3455 out:
3456         mutex_unlock(&lock);
3457         return ret;
3458 }
3459
3460 static struct notifier_block cma_nb = {
3461         .notifier_call = cma_netdev_callback
3462 };
3463
3464 static void cma_add_one(struct ib_device *device)
3465 {
3466         struct cma_device *cma_dev;
3467         struct rdma_id_private *id_priv;
3468
3469         cma_dev = kmalloc(sizeof *cma_dev, GFP_KERNEL);
3470         if (!cma_dev)
3471                 return;
3472
3473         cma_dev->device = device;
3474
3475         init_completion(&cma_dev->comp);
3476         atomic_set(&cma_dev->refcount, 1);
3477         INIT_LIST_HEAD(&cma_dev->id_list);
3478         ib_set_client_data(device, &cma_client, cma_dev);
3479
3480         mutex_lock(&lock);
3481         list_add_tail(&cma_dev->list, &dev_list);
3482         list_for_each_entry(id_priv, &listen_any_list, list)
3483                 cma_listen_on_dev(id_priv, cma_dev);
3484         mutex_unlock(&lock);
3485 }
3486
3487 static int cma_remove_id_dev(struct rdma_id_private *id_priv)
3488 {
3489         struct rdma_cm_event event;
3490         enum rdma_cm_state state;
3491         int ret = 0;
3492
3493         /* Record that we want to remove the device */
3494         state = cma_exch(id_priv, RDMA_CM_DEVICE_REMOVAL);
3495         if (state == RDMA_CM_DESTROYING)
3496                 return 0;
3497
3498         cma_cancel_operation(id_priv, state);
3499         mutex_lock(&id_priv->handler_mutex);
3500
3501         /* Check for destruction from another callback. */
3502         if (!cma_comp(id_priv, RDMA_CM_DEVICE_REMOVAL))
3503                 goto out;
3504
3505         memset(&event, 0, sizeof event);
3506         event.event = RDMA_CM_EVENT_DEVICE_REMOVAL;
3507         ret = id_priv->id.event_handler(&id_priv->id, &event);
3508 out:
3509         mutex_unlock(&id_priv->handler_mutex);
3510         return ret;
3511 }
3512
3513 static void cma_process_remove(struct cma_device *cma_dev)
3514 {
3515         struct rdma_id_private *id_priv;
3516         int ret;
3517
3518         mutex_lock(&lock);
3519         while (!list_empty(&cma_dev->id_list)) {
3520                 id_priv = list_entry(cma_dev->id_list.next,
3521                                      struct rdma_id_private, list);
3522
3523                 list_del(&id_priv->listen_list);
3524                 list_del_init(&id_priv->list);
3525                 atomic_inc(&id_priv->refcount);
3526                 mutex_unlock(&lock);
3527
3528                 ret = id_priv->internal_id ? 1 : cma_remove_id_dev(id_priv);
3529                 cma_deref_id(id_priv);
3530                 if (ret)
3531                         rdma_destroy_id(&id_priv->id);
3532
3533                 mutex_lock(&lock);
3534         }
3535         mutex_unlock(&lock);
3536
3537         cma_deref_dev(cma_dev);
3538         wait_for_completion(&cma_dev->comp);
3539 }
3540
3541 static void cma_remove_one(struct ib_device *device)
3542 {
3543         struct cma_device *cma_dev;
3544
3545         cma_dev = ib_get_client_data(device, &cma_client);
3546         if (!cma_dev)
3547                 return;
3548
3549         mutex_lock(&lock);
3550         list_del(&cma_dev->list);
3551         mutex_unlock(&lock);
3552
3553         cma_process_remove(cma_dev);
3554         kfree(cma_dev);
3555 }
3556
3557 static int cma_get_id_stats(struct sk_buff *skb, struct netlink_callback *cb)
3558 {
3559         struct nlmsghdr *nlh;
3560         struct rdma_cm_id_stats *id_stats;
3561         struct rdma_id_private *id_priv;
3562         struct rdma_cm_id *id = NULL;
3563         struct cma_device *cma_dev;
3564         int i_dev = 0, i_id = 0;
3565
3566         /*
3567          * We export all of the IDs as a sequence of messages.  Each
3568          * ID gets its own netlink message.
3569          */
3570         mutex_lock(&lock);
3571
3572         list_for_each_entry(cma_dev, &dev_list, list) {
3573                 if (i_dev < cb->args[0]) {
3574                         i_dev++;
3575                         continue;
3576                 }
3577
3578                 i_id = 0;
3579                 list_for_each_entry(id_priv, &cma_dev->id_list, list) {
3580                         if (i_id < cb->args[1]) {
3581                                 i_id++;
3582                                 continue;
3583                         }
3584
3585                         id_stats = ibnl_put_msg(skb, &nlh, cb->nlh->nlmsg_seq,
3586                                                 sizeof *id_stats, RDMA_NL_RDMA_CM,
3587                                                 RDMA_NL_RDMA_CM_ID_STATS);
3588                         if (!id_stats)
3589                                 goto out;
3590
3591                         memset(id_stats, 0, sizeof *id_stats);
3592                         id = &id_priv->id;
3593                         id_stats->node_type = id->route.addr.dev_addr.dev_type;
3594                         id_stats->port_num = id->port_num;
3595                         id_stats->bound_dev_if =
3596                                 id->route.addr.dev_addr.bound_dev_if;
3597
3598                         if (ibnl_put_attr(skb, nlh,
3599                                           rdma_addr_size(cma_src_addr(id_priv)),
3600                                           cma_src_addr(id_priv),
3601                                           RDMA_NL_RDMA_CM_ATTR_SRC_ADDR))
3602                                 goto out;
3603                         if (ibnl_put_attr(skb, nlh,
3604                                           rdma_addr_size(cma_src_addr(id_priv)),
3605                                           cma_dst_addr(id_priv),
3606                                           RDMA_NL_RDMA_CM_ATTR_DST_ADDR))
3607                                 goto out;
3608
3609                         id_stats->pid           = id_priv->owner;
3610                         id_stats->port_space    = id->ps;
3611                         id_stats->cm_state      = id_priv->state;
3612                         id_stats->qp_num        = id_priv->qp_num;
3613                         id_stats->qp_type       = id->qp_type;
3614
3615                         i_id++;
3616                 }
3617
3618                 cb->args[1] = 0;
3619                 i_dev++;
3620         }
3621
3622 out:
3623         mutex_unlock(&lock);
3624         cb->args[0] = i_dev;
3625         cb->args[1] = i_id;
3626
3627         return skb->len;
3628 }
3629
3630 static const struct ibnl_client_cbs cma_cb_table[] = {
3631         [RDMA_NL_RDMA_CM_ID_STATS] = { .dump = cma_get_id_stats,
3632                                        .module = THIS_MODULE },
3633 };
3634
3635 static int __init cma_init(void)
3636 {
3637         int ret;
3638
3639         cma_wq = create_singlethread_workqueue("rdma_cm");
3640         if (!cma_wq)
3641                 return -ENOMEM;
3642
3643         ib_sa_register_client(&sa_client);
3644         rdma_addr_register_client(&addr_client);
3645         register_netdevice_notifier(&cma_nb);
3646
3647         ret = ib_register_client(&cma_client);
3648         if (ret)
3649                 goto err;
3650
3651         if (ibnl_add_client(RDMA_NL_RDMA_CM, RDMA_NL_RDMA_CM_NUM_OPS, cma_cb_table))
3652                 printk(KERN_WARNING "RDMA CMA: failed to add netlink callback\n");
3653
3654         return 0;
3655
3656 err:
3657         unregister_netdevice_notifier(&cma_nb);
3658         rdma_addr_unregister_client(&addr_client);
3659         ib_sa_unregister_client(&sa_client);
3660         destroy_workqueue(cma_wq);
3661         return ret;
3662 }
3663
3664 static void __exit cma_cleanup(void)
3665 {
3666         ibnl_remove_client(RDMA_NL_RDMA_CM);
3667         ib_unregister_client(&cma_client);
3668         unregister_netdevice_notifier(&cma_nb);
3669         rdma_addr_unregister_client(&addr_client);
3670         ib_sa_unregister_client(&sa_client);
3671         destroy_workqueue(cma_wq);
3672         idr_destroy(&tcp_ps);
3673         idr_destroy(&udp_ps);
3674         idr_destroy(&ipoib_ps);
3675         idr_destroy(&ib_ps);
3676 }
3677
3678 module_init(cma_init);
3679 module_exit(cma_cleanup);