|
| 1 | +// SPDX-License-Identifier: GPL-2.0-only |
| 2 | +/* |
| 3 | + * Copyright (c) 2022, Microsoft Corporation. All rights reserved. |
| 4 | + */ |
| 5 | + |
| 6 | +#include "mana_ib.h" |
| 7 | +#include <net/mana/mana_auxiliary.h> |
| 8 | + |
| 9 | +MODULE_DESCRIPTION("Microsoft Azure Network Adapter IB driver"); |
| 10 | +MODULE_LICENSE("GPL"); |
| 11 | +MODULE_IMPORT_NS(NET_MANA); |
| 12 | + |
| 13 | +static const struct ib_device_ops mana_ib_dev_ops = { |
| 14 | + .owner = THIS_MODULE, |
| 15 | + .driver_id = RDMA_DRIVER_MANA, |
| 16 | + .uverbs_abi_ver = MANA_IB_UVERBS_ABI_VERSION, |
| 17 | + |
| 18 | + .alloc_pd = mana_ib_alloc_pd, |
| 19 | + .alloc_ucontext = mana_ib_alloc_ucontext, |
| 20 | + .create_cq = mana_ib_create_cq, |
| 21 | + .create_qp = mana_ib_create_qp, |
| 22 | + .create_rwq_ind_table = mana_ib_create_rwq_ind_table, |
| 23 | + .create_wq = mana_ib_create_wq, |
| 24 | + .dealloc_pd = mana_ib_dealloc_pd, |
| 25 | + .dealloc_ucontext = mana_ib_dealloc_ucontext, |
| 26 | + .dereg_mr = mana_ib_dereg_mr, |
| 27 | + .destroy_cq = mana_ib_destroy_cq, |
| 28 | + .destroy_qp = mana_ib_destroy_qp, |
| 29 | + .destroy_rwq_ind_table = mana_ib_destroy_rwq_ind_table, |
| 30 | + .destroy_wq = mana_ib_destroy_wq, |
| 31 | + .disassociate_ucontext = mana_ib_disassociate_ucontext, |
| 32 | + .get_port_immutable = mana_ib_get_port_immutable, |
| 33 | + .mmap = mana_ib_mmap, |
| 34 | + .modify_qp = mana_ib_modify_qp, |
| 35 | + .modify_wq = mana_ib_modify_wq, |
| 36 | + .query_device = mana_ib_query_device, |
| 37 | + .query_gid = mana_ib_query_gid, |
| 38 | + .query_port = mana_ib_query_port, |
| 39 | + .reg_user_mr = mana_ib_reg_user_mr, |
| 40 | + |
| 41 | + INIT_RDMA_OBJ_SIZE(ib_cq, mana_ib_cq, ibcq), |
| 42 | + INIT_RDMA_OBJ_SIZE(ib_pd, mana_ib_pd, ibpd), |
| 43 | + INIT_RDMA_OBJ_SIZE(ib_qp, mana_ib_qp, ibqp), |
| 44 | + INIT_RDMA_OBJ_SIZE(ib_ucontext, mana_ib_ucontext, ibucontext), |
| 45 | + INIT_RDMA_OBJ_SIZE(ib_rwq_ind_table, mana_ib_rwq_ind_table, |
| 46 | + ib_ind_table), |
| 47 | +}; |
| 48 | + |
| 49 | +static int mana_ib_probe(struct auxiliary_device *adev, |
| 50 | + const struct auxiliary_device_id *id) |
| 51 | +{ |
| 52 | + struct mana_adev *madev = container_of(adev, struct mana_adev, adev); |
| 53 | + struct gdma_dev *mdev = madev->mdev; |
| 54 | + struct mana_context *mc; |
| 55 | + struct mana_ib_dev *dev; |
| 56 | + int ret; |
| 57 | + |
| 58 | + mc = mdev->driver_data; |
| 59 | + |
| 60 | + dev = ib_alloc_device(mana_ib_dev, ib_dev); |
| 61 | + if (!dev) |
| 62 | + return -ENOMEM; |
| 63 | + |
| 64 | + ib_set_device_ops(&dev->ib_dev, &mana_ib_dev_ops); |
| 65 | + |
| 66 | + dev->ib_dev.phys_port_cnt = mc->num_ports; |
| 67 | + |
| 68 | + ibdev_dbg(&dev->ib_dev, "mdev=%p id=%d num_ports=%d\n", mdev, |
| 69 | + mdev->dev_id.as_uint32, dev->ib_dev.phys_port_cnt); |
| 70 | + |
| 71 | + dev->gdma_dev = mdev; |
| 72 | + dev->ib_dev.node_type = RDMA_NODE_IB_CA; |
| 73 | + |
| 74 | + /* |
| 75 | + * num_comp_vectors needs to set to the max MSIX index |
| 76 | + * when interrupts and event queues are implemented |
| 77 | + */ |
| 78 | + dev->ib_dev.num_comp_vectors = 1; |
| 79 | + dev->ib_dev.dev.parent = mdev->gdma_context->dev; |
| 80 | + |
| 81 | + ret = ib_register_device(&dev->ib_dev, "mana_%d", |
| 82 | + mdev->gdma_context->dev); |
| 83 | + if (ret) { |
| 84 | + ib_dealloc_device(&dev->ib_dev); |
| 85 | + return ret; |
| 86 | + } |
| 87 | + |
| 88 | + dev_set_drvdata(&adev->dev, dev); |
| 89 | + |
| 90 | + return 0; |
| 91 | +} |
| 92 | + |
| 93 | +static void mana_ib_remove(struct auxiliary_device *adev) |
| 94 | +{ |
| 95 | + struct mana_ib_dev *dev = dev_get_drvdata(&adev->dev); |
| 96 | + |
| 97 | + ib_unregister_device(&dev->ib_dev); |
| 98 | + ib_dealloc_device(&dev->ib_dev); |
| 99 | +} |
| 100 | + |
| 101 | +static const struct auxiliary_device_id mana_id_table[] = { |
| 102 | + { |
| 103 | + .name = "mana.rdma", |
| 104 | + }, |
| 105 | + {}, |
| 106 | +}; |
| 107 | + |
| 108 | +MODULE_DEVICE_TABLE(auxiliary, mana_id_table); |
| 109 | + |
| 110 | +static struct auxiliary_driver mana_driver = { |
| 111 | + .name = "rdma", |
| 112 | + .probe = mana_ib_probe, |
| 113 | + .remove = mana_ib_remove, |
| 114 | + .id_table = mana_id_table, |
| 115 | +}; |
| 116 | + |
| 117 | +module_auxiliary_driver(mana_driver); |
0 commit comments