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);
118