drivers/infiniband/hw/vmw_pvrdma/pvrdma_verbs.c

   1 /*
   2  * Copyright (c) 2012-2016 VMware, Inc.  All rights reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of EITHER the GNU General Public License
   6  * version 2 as published by the Free Software Foundation or the BSD
   7  * 2-Clause License. This program is distributed in the hope that it
   8  * will be useful, but WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED
   9  * WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
  10  * See the GNU General Public License version 2 for more details at
  11  * http://www.gnu.org/licenses/old-licenses/gpl-2.0.en.html.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program available in the file COPYING in the main
  15  * directory of this source tree.
  16  *
  17  * The BSD 2-Clause License
  18  *
  19  *     Redistribution and use in source and binary forms, with or
  20  *     without modification, are permitted provided that the following
  21  *     conditions are met:
  22  *
  23  *      - Redistributions of source code must retain the above
  24  *        copyright notice, this list of conditions and the following
  25  *        disclaimer.
  26  *
  27  *      - Redistributions in binary form must reproduce the above
  28  *        copyright notice, this list of conditions and the following
  29  *        disclaimer in the documentation and/or other materials
  30  *        provided with the distribution.
  31  *
  32  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  33  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  34  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  35  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  36  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  37  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  38  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  39  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  40  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  41  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  42  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  43  * OF THE POSSIBILITY OF SUCH DAMAGE.
  44  */
  45
  46 #include <asm/page.h>
  47 #include <linux/inet.h>
  48 #include <linux/io.h>
  49 #include <rdma/ib_addr.h>
  50 #include <rdma/ib_smi.h>
  51 #include <rdma/ib_user_verbs.h>
  52 #include <rdma/vmw_pvrdma-abi.h>
  53
  54 #include "pvrdma.h"
  55
  56 /**
  57  * pvrdma_query_device - query device
  58  * @ibdev: the device to query
  59  * @props: the device properties
  60  * @uhw: user data
  61  *
  62  * @return: 0 on success, otherwise negative errno
  63  */
  64 int pvrdma_query_device(struct ib_device *ibdev,
  65                         struct ib_device_attr *props,
  66                         struct ib_udata *uhw)
  67 {
  68         struct pvrdma_dev *dev = to_vdev(ibdev);
  69
  70         if (uhw->inlen || uhw->outlen)
  71                 return -EINVAL;
  72
  73         memset(props, 0, sizeof(*props));
  74
  75         props->fw_ver = dev->dsr->caps.fw_ver;
  76         props->sys_image_guid = dev->dsr->caps.sys_image_guid;
  77         props->max_mr_size = dev->dsr->caps.max_mr_size;
  78         props->page_size_cap = dev->dsr->caps.page_size_cap;
  79         props->vendor_id = dev->dsr->caps.vendor_id;
  80         props->vendor_part_id = dev->pdev->device;
  81         props->hw_ver = dev->dsr->caps.hw_ver;
  82         props->max_qp = dev->dsr->caps.max_qp;
  83         props->max_qp_wr = dev->dsr->caps.max_qp_wr;
  84         props->device_cap_flags = dev->dsr->caps.device_cap_flags;
  85         props->max_send_sge = dev->dsr->caps.max_sge;
  86         props->max_recv_sge = dev->dsr->caps.max_sge;
  87         props->max_sge_rd = PVRDMA_GET_CAP(dev, dev->dsr->caps.max_sge,
  88                                            dev->dsr->caps.max_sge_rd);
  89         props->max_srq = dev->dsr->caps.max_srq;
  90         props->max_srq_wr = dev->dsr->caps.max_srq_wr;
  91         props->max_srq_sge = dev->dsr->caps.max_srq_sge;
  92         props->max_cq = dev->dsr->caps.max_cq;
  93         props->max_cqe = dev->dsr->caps.max_cqe;
  94         props->max_mr = dev->dsr->caps.max_mr;
  95         props->max_pd = dev->dsr->caps.max_pd;
  96         props->max_qp_rd_atom = dev->dsr->caps.max_qp_rd_atom;
  97         props->max_qp_init_rd_atom = dev->dsr->caps.max_qp_init_rd_atom;
  98         props->atomic_cap =
  99                 dev->dsr->caps.atomic_ops &
 100                 (PVRDMA_ATOMIC_OP_COMP_SWAP | PVRDMA_ATOMIC_OP_FETCH_ADD) ?
 101                 IB_ATOMIC_HCA : IB_ATOMIC_NONE;
 102         props->masked_atomic_cap = props->atomic_cap;
 103         props->max_ah = dev->dsr->caps.max_ah;
 104         props->max_pkeys = dev->dsr->caps.max_pkeys;
 105         props->local_ca_ack_delay = dev->dsr->caps.local_ca_ack_delay;
 106         if ((dev->dsr->caps.bmme_flags & PVRDMA_BMME_FLAG_LOCAL_INV) &&
 107             (dev->dsr->caps.bmme_flags & PVRDMA_BMME_FLAG_REMOTE_INV) &&
 108             (dev->dsr->caps.bmme_flags & PVRDMA_BMME_FLAG_FAST_REG_WR)) {
 109                 props->device_cap_flags |= IB_DEVICE_MEM_MGT_EXTENSIONS;
 110                 props->max_fast_reg_page_list_len = PVRDMA_GET_CAP(dev,
 111                                 PVRDMA_MAX_FAST_REG_PAGES,
 112                                 dev->dsr->caps.max_fast_reg_page_list_len);
 113         }
 114
 115         props->device_cap_flags |= IB_DEVICE_PORT_ACTIVE_EVENT |
 116                                    IB_DEVICE_RC_RNR_NAK_GEN;
 117
 118         return 0;
 119 }
 120
 121 /**
 122  * pvrdma_query_port - query device port attributes
 123  * @ibdev: the device to query
 124  * @port: the port number
 125  * @props: the device properties
 126  *
 127  * @return: 0 on success, otherwise negative errno
 128  */
 129 int pvrdma_query_port(struct ib_device *ibdev, u8 port,
 130                       struct ib_port_attr *props)
 131 {
 132         struct pvrdma_dev *dev = to_vdev(ibdev);
 133         union pvrdma_cmd_req req;
 134         union pvrdma_cmd_resp rsp;
 135         struct pvrdma_cmd_query_port *cmd = &req.query_port;
 136         struct pvrdma_cmd_query_port_resp *resp = &rsp.query_port_resp;
 137         int err;
 138
 139         memset(cmd, 0, sizeof(*cmd));
 140         cmd->hdr.cmd = PVRDMA_CMD_QUERY_PORT;
 141         cmd->port_num = port;
 142
 143         err = pvrdma_cmd_post(dev, &req, &rsp, PVRDMA_CMD_QUERY_PORT_RESP);
 144         if (err < 0) {
 145                 dev_warn(&dev->pdev->dev,
 146                          "could not query port, error: %d\n", err);
 147                 return err;
 148         }
 149
 150         /* props being zeroed by the caller, avoid zeroing it here */
 151
 152         props->state = pvrdma_port_state_to_ib(resp->attrs.state);
 153         props->max_mtu = pvrdma_mtu_to_ib(resp->attrs.max_mtu);
 154         props->active_mtu = pvrdma_mtu_to_ib(resp->attrs.active_mtu);
 155         props->gid_tbl_len = resp->attrs.gid_tbl_len;
 156         props->port_cap_flags =
 157                 pvrdma_port_cap_flags_to_ib(resp->attrs.port_cap_flags);
 158         props->port_cap_flags |= IB_PORT_CM_SUP;
 159         props->ip_gids = true;
 160         props->max_msg_sz = resp->attrs.max_msg_sz;
 161         props->bad_pkey_cntr = resp->attrs.bad_pkey_cntr;
 162         props->qkey_viol_cntr = resp->attrs.qkey_viol_cntr;
 163         props->pkey_tbl_len = resp->attrs.pkey_tbl_len;
 164         props->lid = resp->attrs.lid;
 165         props->sm_lid = resp->attrs.sm_lid;
 166         props->lmc = resp->attrs.lmc;
 167         props->max_vl_num = resp->attrs.max_vl_num;
 168         props->sm_sl = resp->attrs.sm_sl;
 169         props->subnet_timeout = resp->attrs.subnet_timeout;
 170         props->init_type_reply = resp->attrs.init_type_reply;
 171         props->active_width = pvrdma_port_width_to_ib(resp->attrs.active_width);
 172         props->active_speed = pvrdma_port_speed_to_ib(resp->attrs.active_speed);
 173         props->phys_state = resp->attrs.phys_state;
 174
 175         return 0;
 176 }
 177
 178 /**
 179  * pvrdma_query_gid - query device gid
 180  * @ibdev: the device to query
 181  * @port: the port number
 182  * @index: the index
 183  * @gid: the device gid value
 184  *
 185  * @return: 0 on success, otherwise negative errno
 186  */
 187 int pvrdma_query_gid(struct ib_device *ibdev, u8 port, int index,
 188                      union ib_gid *gid)
 189 {
 190         struct pvrdma_dev *dev = to_vdev(ibdev);
 191
 192         if (index >= dev->dsr->caps.gid_tbl_len)
 193                 return -EINVAL;
 194
 195         memcpy(gid, &dev->sgid_tbl[index], sizeof(union ib_gid));
 196
 197         return 0;
 198 }
 199
 200 /**
 201  * pvrdma_query_pkey - query device port's P_Key table
 202  * @ibdev: the device to query
 203  * @port: the port number
 204  * @index: the index
 205  * @pkey: the device P_Key value
 206  *
 207  * @return: 0 on success, otherwise negative errno
 208  */
 209 int pvrdma_query_pkey(struct ib_device *ibdev, u8 port, u16 index,
 210                       u16 *pkey)
 211 {
 212         int err = 0;
 213         union pvrdma_cmd_req req;
 214         union pvrdma_cmd_resp rsp;
 215         struct pvrdma_cmd_query_pkey *cmd = &req.query_pkey;
 216
 217         memset(cmd, 0, sizeof(*cmd));
 218         cmd->hdr.cmd = PVRDMA_CMD_QUERY_PKEY;
 219         cmd->port_num = port;
 220         cmd->index = index;
 221
 222         err = pvrdma_cmd_post(to_vdev(ibdev), &req, &rsp,
 223                               PVRDMA_CMD_QUERY_PKEY_RESP);
 224         if (err < 0) {
 225                 dev_warn(&to_vdev(ibdev)->pdev->dev,
 226                          "could not query pkey, error: %d\n", err);
 227                 return err;
 228         }
 229
 230         *pkey = rsp.query_pkey_resp.pkey;
 231
 232         return 0;
 233 }
 234
 235 enum rdma_link_layer pvrdma_port_link_layer(struct ib_device *ibdev,
 236                                             u8 port)
 237 {
 238         return IB_LINK_LAYER_ETHERNET;
 239 }
 240
 241 int pvrdma_modify_device(struct ib_device *ibdev, int mask,
 242                          struct ib_device_modify *props)
 243 {
 244         unsigned long flags;
 245
 246         if (mask & ~(IB_DEVICE_MODIFY_SYS_IMAGE_GUID |
 247                      IB_DEVICE_MODIFY_NODE_DESC)) {
 248                 dev_warn(&to_vdev(ibdev)->pdev->dev,
 249                          "unsupported device modify mask %#x\n", mask);
 250                 return -EOPNOTSUPP;
 251         }
 252
 253         if (mask & IB_DEVICE_MODIFY_NODE_DESC) {
 254                 spin_lock_irqsave(&to_vdev(ibdev)->desc_lock, flags);
 255                 memcpy(ibdev->node_desc, props->node_desc, 64);
 256                 spin_unlock_irqrestore(&to_vdev(ibdev)->desc_lock, flags);
 257         }
 258
 259         if (mask & IB_DEVICE_MODIFY_SYS_IMAGE_GUID) {
 260                 mutex_lock(&to_vdev(ibdev)->port_mutex);
 261                 to_vdev(ibdev)->sys_image_guid =
 262                         cpu_to_be64(props->sys_image_guid);
 263                 mutex_unlock(&to_vdev(ibdev)->port_mutex);
 264         }
 265
 266         return 0;
 267 }
 268
 269 /**
 270  * pvrdma_modify_port - modify device port attributes
 271  * @ibdev: the device to modify
 272  * @port: the port number
 273  * @mask: attributes to modify
 274  * @props: the device properties
 275  *
 276  * @return: 0 on success, otherwise negative errno
 277  */
 278 int pvrdma_modify_port(struct ib_device *ibdev, u8 port, int mask,
 279                        struct ib_port_modify *props)
 280 {
 281         struct ib_port_attr attr;
 282         struct pvrdma_dev *vdev = to_vdev(ibdev);
 283         int ret;
 284
 285         if (mask & ~IB_PORT_SHUTDOWN) {
 286                 dev_warn(&vdev->pdev->dev,
 287                          "unsupported port modify mask %#x\n", mask);
 288                 return -EOPNOTSUPP;
 289         }
 290
 291         mutex_lock(&vdev->port_mutex);
 292         ret = ib_query_port(ibdev, port, &attr);
 293         if (ret)
 294                 goto out;
 295
 296         vdev->port_cap_mask |= props->set_port_cap_mask;
 297         vdev->port_cap_mask &= ~props->clr_port_cap_mask;
 298
 299         if (mask & IB_PORT_SHUTDOWN)
 300                 vdev->ib_active = false;
 301
 302 out:
 303         mutex_unlock(&vdev->port_mutex);
 304         return ret;
 305 }
 306
 307 /**
 308  * pvrdma_alloc_ucontext - allocate ucontext
 309  * @ibdev: the IB device
 310  * @udata: user data
 311  *
 312  * @return: the ib_ucontext pointer on success, otherwise errno.
 313  */
 314 struct ib_ucontext *pvrdma_alloc_ucontext(struct ib_device *ibdev,
 315                                           struct ib_udata *udata)
 316 {
 317         struct pvrdma_dev *vdev = to_vdev(ibdev);
 318         struct pvrdma_ucontext *context;
 319         union pvrdma_cmd_req req;
 320         union pvrdma_cmd_resp rsp;
 321         struct pvrdma_cmd_create_uc *cmd = &req.create_uc;
 322         struct pvrdma_cmd_create_uc_resp *resp = &rsp.create_uc_resp;
 323         struct pvrdma_alloc_ucontext_resp uresp = {0};
 324         int ret;
 325         void *ptr;
 326
 327         if (!vdev->ib_active)
 328                 return ERR_PTR(-EAGAIN);
 329
 330         context = kmalloc(sizeof(*context), GFP_KERNEL);
 331         if (!context)
 332                 return ERR_PTR(-ENOMEM);
 333
 334         context->dev = vdev;
 335         ret = pvrdma_uar_alloc(vdev, &context->uar);
 336         if (ret) {
 337                 kfree(context);
 338                 return ERR_PTR(-ENOMEM);
 339         }
 340
 341         /* get ctx_handle from host */
 342         memset(cmd, 0, sizeof(*cmd));
 343         cmd->pfn = context->uar.pfn;
 344         cmd->hdr.cmd = PVRDMA_CMD_CREATE_UC;
 345         ret = pvrdma_cmd_post(vdev, &req, &rsp, PVRDMA_CMD_CREATE_UC_RESP);
 346         if (ret < 0) {
 347                 dev_warn(&vdev->pdev->dev,
 348                          "could not create ucontext, error: %d\n", ret);
 349                 ptr = ERR_PTR(ret);
 350                 goto err;
 351         }
 352
 353         context->ctx_handle = resp->ctx_handle;
 354
 355         /* copy back to user */
 356         uresp.qp_tab_size = vdev->dsr->caps.max_qp;
 357         ret = ib_copy_to_udata(udata, &uresp, sizeof(uresp));
 358         if (ret) {
 359                 pvrdma_uar_free(vdev, &context->uar);
 360                 context->ibucontext.device = ibdev;
 361                 pvrdma_dealloc_ucontext(&context->ibucontext);
 362                 return ERR_PTR(-EFAULT);
 363         }
 364
 365         return &context->ibucontext;
 366
 367 err:
 368         pvrdma_uar_free(vdev, &context->uar);
 369         kfree(context);
 370         return ptr;
 371 }
 372
 373 /**
 374  * pvrdma_dealloc_ucontext - deallocate ucontext
 375  * @ibcontext: the ucontext
 376  *
 377  * @return: 0 on success, otherwise errno.
 378  */
 379 int pvrdma_dealloc_ucontext(struct ib_ucontext *ibcontext)
 380 {
 381         struct pvrdma_ucontext *context = to_vucontext(ibcontext);
 382         union pvrdma_cmd_req req;
 383         struct pvrdma_cmd_destroy_uc *cmd = &req.destroy_uc;
 384         int ret;
 385
 386         memset(cmd, 0, sizeof(*cmd));
 387         cmd->hdr.cmd = PVRDMA_CMD_DESTROY_UC;
 388         cmd->ctx_handle = context->ctx_handle;
 389
 390         ret = pvrdma_cmd_post(context->dev, &req, NULL, 0);
 391         if (ret < 0)
 392                 dev_warn(&context->dev->pdev->dev,
 393                          "destroy ucontext failed, error: %d\n", ret);
 394
 395         /* Free the UAR even if the device command failed */
 396         pvrdma_uar_free(to_vdev(ibcontext->device), &context->uar);
 397         kfree(context);
 398
 399         return ret;
 400 }
 401
 402 /**
 403  * pvrdma_mmap - create mmap region
 404  * @ibcontext: the user context
 405  * @vma: the VMA
 406  *
 407  * @return: 0 on success, otherwise errno.
 408  */
 409 int pvrdma_mmap(struct ib_ucontext *ibcontext, struct vm_area_struct *vma)
 410 {
 411         struct pvrdma_ucontext *context = to_vucontext(ibcontext);
 412         unsigned long start = vma->vm_start;
 413         unsigned long size = vma->vm_end - vma->vm_start;
 414         unsigned long offset = vma->vm_pgoff << PAGE_SHIFT;
 415
 416         dev_dbg(&context->dev->pdev->dev, "create mmap region\n");
 417
 418         if ((size != PAGE_SIZE) || (offset & ~PAGE_MASK)) {
 419                 dev_warn(&context->dev->pdev->dev,
 420                          "invalid params for mmap region\n");
 421                 return -EINVAL;
 422         }
 423
 424         /* Map UAR to kernel space, VM_LOCKED? */
 425         vma->vm_flags |= VM_DONTCOPY | VM_DONTEXPAND;
 426         vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
 427         if (io_remap_pfn_range(vma, start, context->uar.pfn, size,
 428                                vma->vm_page_prot))
 429                 return -EAGAIN;
 430
 431         return 0;
 432 }
 433
 434 /**
 435  * pvrdma_alloc_pd - allocate protection domain
 436  * @ibdev: the IB device
 437  * @context: user context
 438  * @udata: user data
 439  *
 440  * @return: the ib_pd protection domain pointer on success, otherwise errno.
 441  */
 442 struct ib_pd *pvrdma_alloc_pd(struct ib_device *ibdev,
 443                               struct ib_ucontext *context,
 444                               struct ib_udata *udata)
 445 {
 446         struct pvrdma_pd *pd;
 447         struct pvrdma_dev *dev = to_vdev(ibdev);
 448         union pvrdma_cmd_req req;
 449         union pvrdma_cmd_resp rsp;
 450         struct pvrdma_cmd_create_pd *cmd = &req.create_pd;
 451         struct pvrdma_cmd_create_pd_resp *resp = &rsp.create_pd_resp;
 452         struct pvrdma_alloc_pd_resp pd_resp = {0};
 453         int ret;
 454         void *ptr;
 455
 456         /* Check allowed max pds */
 457         if (!atomic_add_unless(&dev->num_pds, 1, dev->dsr->caps.max_pd))
 458                 return ERR_PTR(-ENOMEM);
 459
 460         pd = kmalloc(sizeof(*pd), GFP_KERNEL);
 461         if (!pd) {
 462                 ptr = ERR_PTR(-ENOMEM);
 463                 goto err;
 464         }
 465
 466         memset(cmd, 0, sizeof(*cmd));
 467         cmd->hdr.cmd = PVRDMA_CMD_CREATE_PD;
 468         cmd->ctx_handle = (context) ? to_vucontext(context)->ctx_handle : 0;
 469         ret = pvrdma_cmd_post(dev, &req, &rsp, PVRDMA_CMD_CREATE_PD_RESP);
 470         if (ret < 0) {
 471                 dev_warn(&dev->pdev->dev,
 472                          "failed to allocate protection domain, error: %d\n",
 473                          ret);
 474                 ptr = ERR_PTR(ret);
 475                 goto freepd;
 476         }
 477
 478         pd->privileged = !context;
 479         pd->pd_handle = resp->pd_handle;
 480         pd->pdn = resp->pd_handle;
 481         pd_resp.pdn = resp->pd_handle;
 482
 483         if (context) {
 484                 if (ib_copy_to_udata(udata, &pd_resp, sizeof(pd_resp))) {
 485                         dev_warn(&dev->pdev->dev,
 486                                  "failed to copy back protection domain\n");
 487                         pvrdma_dealloc_pd(&pd->ibpd);
 488                         return ERR_PTR(-EFAULT);
 489                 }
 490         }
 491
 492         /* u32 pd handle */
 493         return &pd->ibpd;
 494
 495 freepd:
 496         kfree(pd);
 497 err:
 498         atomic_dec(&dev->num_pds);
 499         return ptr;
 500 }
 501
 502 /**
 503  * pvrdma_dealloc_pd - deallocate protection domain
 504  * @pd: the protection domain to be released
 505  *
 506  * @return: 0 on success, otherwise errno.
 507  */
 508 int pvrdma_dealloc_pd(struct ib_pd *pd)
 509 {
 510         struct pvrdma_dev *dev = to_vdev(pd->device);
 511         union pvrdma_cmd_req req;
 512         struct pvrdma_cmd_destroy_pd *cmd = &req.destroy_pd;
 513         int ret;
 514
 515         memset(cmd, 0, sizeof(*cmd));
 516         cmd->hdr.cmd = PVRDMA_CMD_DESTROY_PD;
 517         cmd->pd_handle = to_vpd(pd)->pd_handle;
 518
 519         ret = pvrdma_cmd_post(dev, &req, NULL, 0);
 520         if (ret)
 521                 dev_warn(&dev->pdev->dev,
 522                          "could not dealloc protection domain, error: %d\n",
 523                          ret);
 524
 525         kfree(to_vpd(pd));
 526         atomic_dec(&dev->num_pds);
 527
 528         return 0;
 529 }
 530
 531 /**
 532  * pvrdma_create_ah - create an address handle
 533  * @pd: the protection domain
 534  * @ah_attr: the attributes of the AH
 535  * @udata: user data blob
 536  *
 537  * @return: the ib_ah pointer on success, otherwise errno.
 538  */
 539 struct ib_ah *pvrdma_create_ah(struct ib_pd *pd, struct rdma_ah_attr *ah_attr,
 540                                struct ib_udata *udata)
 541 {
 542         struct pvrdma_dev *dev = to_vdev(pd->device);
 543         struct pvrdma_ah *ah;
 544         const struct ib_global_route *grh;
 545         u8 port_num = rdma_ah_get_port_num(ah_attr);
 546
 547         if (!(rdma_ah_get_ah_flags(ah_attr) & IB_AH_GRH))
 548                 return ERR_PTR(-EINVAL);
 549
 550         grh = rdma_ah_read_grh(ah_attr);
 551         if ((ah_attr->type != RDMA_AH_ATTR_TYPE_ROCE)  ||
 552             rdma_is_multicast_addr((struct in6_addr *)grh->dgid.raw))
 553                 return ERR_PTR(-EINVAL);
 554
 555         if (!atomic_add_unless(&dev->num_ahs, 1, dev->dsr->caps.max_ah))
 556                 return ERR_PTR(-ENOMEM);
 557
 558         ah = kzalloc(sizeof(*ah), GFP_ATOMIC);
 559         if (!ah) {
 560                 atomic_dec(&dev->num_ahs);
 561                 return ERR_PTR(-ENOMEM);
 562         }
 563
 564         ah->av.port_pd = to_vpd(pd)->pd_handle | (port_num << 24);
 565         ah->av.src_path_bits = rdma_ah_get_path_bits(ah_attr);
 566         ah->av.src_path_bits |= 0x80;
 567         ah->av.gid_index = grh->sgid_index;
 568         ah->av.hop_limit = grh->hop_limit;
 569         ah->av.sl_tclass_flowlabel = (grh->traffic_class << 20) |
 570                                       grh->flow_label;
 571         memcpy(ah->av.dgid, grh->dgid.raw, 16);
 572         memcpy(ah->av.dmac, ah_attr->roce.dmac, ETH_ALEN);
 573
 574         ah->ibah.device = pd->device;
 575         ah->ibah.pd = pd;
 576         ah->ibah.uobject = NULL;
 577
 578         return &ah->ibah;
 579 }
 580
 581 /**
 582  * pvrdma_destroy_ah - destroy an address handle
 583  * @ah: the address handle to destroyed
 584  *
 585  * @return: 0 on success.
 586  */
 587 int pvrdma_destroy_ah(struct ib_ah *ah)
 588 {
 589         struct pvrdma_dev *dev = to_vdev(ah->device);
 590
 591         kfree(to_vah(ah));
 592         atomic_dec(&dev->num_ahs);
 593
 594         return 0;
 595 }