RDMA/hns: Fix refcount leak in hns_roce_mmap
[platform/kernel/linux-starfive.git] / drivers / infiniband / hw / hns / hns_roce_main.c
1 /*
2  * Copyright (c) 2016 Hisilicon Limited.
3  * Copyright (c) 2007, 2008 Mellanox Technologies. All rights reserved.
4  *
5  * This software is available to you under a choice of one of two
6  * licenses.  You may choose to be licensed under the terms of the GNU
7  * General Public License (GPL) Version 2, available from the file
8  * COPYING in the main directory of this source tree, or the
9  * OpenIB.org BSD license below:
10  *
11  *     Redistribution and use in source and binary forms, with or
12  *     without modification, are permitted provided that the following
13  *     conditions are met:
14  *
15  *      - Redistributions of source code must retain the above
16  *        copyright notice, this list of conditions and the following
17  *        disclaimer.
18  *
19  *      - Redistributions in binary form must reproduce the above
20  *        copyright notice, this list of conditions and the following
21  *        disclaimer in the documentation and/or other materials
22  *        provided with the distribution.
23  *
24  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
25  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
26  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
27  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
28  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
29  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
30  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
31  * SOFTWARE.
32  */
33 #include <linux/acpi.h>
34 #include <linux/module.h>
35 #include <linux/pci.h>
36 #include <rdma/ib_addr.h>
37 #include <rdma/ib_smi.h>
38 #include <rdma/ib_user_verbs.h>
39 #include <rdma/ib_cache.h>
40 #include "hns_roce_common.h"
41 #include "hns_roce_device.h"
42 #include "hns_roce_hem.h"
43
44 static int hns_roce_set_mac(struct hns_roce_dev *hr_dev, u32 port,
45                             const u8 *addr)
46 {
47         u8 phy_port;
48         u32 i;
49
50         if (hr_dev->pci_dev->revision >= PCI_REVISION_ID_HIP09)
51                 return 0;
52
53         if (!memcmp(hr_dev->dev_addr[port], addr, ETH_ALEN))
54                 return 0;
55
56         for (i = 0; i < ETH_ALEN; i++)
57                 hr_dev->dev_addr[port][i] = addr[i];
58
59         phy_port = hr_dev->iboe.phy_port[port];
60         return hr_dev->hw->set_mac(hr_dev, phy_port, addr);
61 }
62
63 static int hns_roce_add_gid(const struct ib_gid_attr *attr, void **context)
64 {
65         struct hns_roce_dev *hr_dev = to_hr_dev(attr->device);
66         u32 port = attr->port_num - 1;
67         int ret;
68
69         if (port >= hr_dev->caps.num_ports)
70                 return -EINVAL;
71
72         ret = hr_dev->hw->set_gid(hr_dev, attr->index, &attr->gid, attr);
73
74         return ret;
75 }
76
77 static int hns_roce_del_gid(const struct ib_gid_attr *attr, void **context)
78 {
79         struct hns_roce_dev *hr_dev = to_hr_dev(attr->device);
80         u32 port = attr->port_num - 1;
81         int ret;
82
83         if (port >= hr_dev->caps.num_ports)
84                 return -EINVAL;
85
86         ret = hr_dev->hw->set_gid(hr_dev, attr->index, NULL, NULL);
87
88         return ret;
89 }
90
91 static int handle_en_event(struct hns_roce_dev *hr_dev, u32 port,
92                            unsigned long event)
93 {
94         struct device *dev = hr_dev->dev;
95         struct net_device *netdev;
96         int ret = 0;
97
98         netdev = hr_dev->iboe.netdevs[port];
99         if (!netdev) {
100                 dev_err(dev, "can't find netdev on port(%u)!\n", port);
101                 return -ENODEV;
102         }
103
104         switch (event) {
105         case NETDEV_UP:
106         case NETDEV_CHANGE:
107         case NETDEV_REGISTER:
108         case NETDEV_CHANGEADDR:
109                 ret = hns_roce_set_mac(hr_dev, port, netdev->dev_addr);
110                 break;
111         case NETDEV_DOWN:
112                 /*
113                  * In v1 engine, only support all ports closed together.
114                  */
115                 break;
116         default:
117                 dev_dbg(dev, "NETDEV event = 0x%x!\n", (u32)(event));
118                 break;
119         }
120
121         return ret;
122 }
123
124 static int hns_roce_netdev_event(struct notifier_block *self,
125                                  unsigned long event, void *ptr)
126 {
127         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
128         struct hns_roce_ib_iboe *iboe = NULL;
129         struct hns_roce_dev *hr_dev = NULL;
130         int ret;
131         u32 port;
132
133         hr_dev = container_of(self, struct hns_roce_dev, iboe.nb);
134         iboe = &hr_dev->iboe;
135
136         for (port = 0; port < hr_dev->caps.num_ports; port++) {
137                 if (dev == iboe->netdevs[port]) {
138                         ret = handle_en_event(hr_dev, port, event);
139                         if (ret)
140                                 return NOTIFY_DONE;
141                         break;
142                 }
143         }
144
145         return NOTIFY_DONE;
146 }
147
148 static int hns_roce_setup_mtu_mac(struct hns_roce_dev *hr_dev)
149 {
150         int ret;
151         u8 i;
152
153         for (i = 0; i < hr_dev->caps.num_ports; i++) {
154                 ret = hns_roce_set_mac(hr_dev, i,
155                                        hr_dev->iboe.netdevs[i]->dev_addr);
156                 if (ret)
157                         return ret;
158         }
159
160         return 0;
161 }
162
163 static int hns_roce_query_device(struct ib_device *ib_dev,
164                                  struct ib_device_attr *props,
165                                  struct ib_udata *uhw)
166 {
167         struct hns_roce_dev *hr_dev = to_hr_dev(ib_dev);
168
169         memset(props, 0, sizeof(*props));
170
171         props->fw_ver = hr_dev->caps.fw_ver;
172         props->sys_image_guid = cpu_to_be64(hr_dev->sys_image_guid);
173         props->max_mr_size = (u64)(~(0ULL));
174         props->page_size_cap = hr_dev->caps.page_size_cap;
175         props->vendor_id = hr_dev->vendor_id;
176         props->vendor_part_id = hr_dev->vendor_part_id;
177         props->hw_ver = hr_dev->hw_rev;
178         props->max_qp = hr_dev->caps.num_qps;
179         props->max_qp_wr = hr_dev->caps.max_wqes;
180         props->device_cap_flags = IB_DEVICE_PORT_ACTIVE_EVENT |
181                                   IB_DEVICE_RC_RNR_NAK_GEN;
182         props->max_send_sge = hr_dev->caps.max_sq_sg;
183         props->max_recv_sge = hr_dev->caps.max_rq_sg;
184         props->max_sge_rd = 1;
185         props->max_cq = hr_dev->caps.num_cqs;
186         props->max_cqe = hr_dev->caps.max_cqes;
187         props->max_mr = hr_dev->caps.num_mtpts;
188         props->max_pd = hr_dev->caps.num_pds;
189         props->max_qp_rd_atom = hr_dev->caps.max_qp_dest_rdma;
190         props->max_qp_init_rd_atom = hr_dev->caps.max_qp_init_rdma;
191         props->atomic_cap = hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_ATOMIC ?
192                             IB_ATOMIC_HCA : IB_ATOMIC_NONE;
193         props->max_pkeys = 1;
194         props->local_ca_ack_delay = hr_dev->caps.local_ca_ack_delay;
195         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_SRQ) {
196                 props->max_srq = hr_dev->caps.num_srqs;
197                 props->max_srq_wr = hr_dev->caps.max_srq_wrs;
198                 props->max_srq_sge = hr_dev->caps.max_srq_sges;
199         }
200
201         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_FRMR &&
202             hr_dev->pci_dev->revision >= PCI_REVISION_ID_HIP09) {
203                 props->device_cap_flags |= IB_DEVICE_MEM_MGT_EXTENSIONS;
204                 props->max_fast_reg_page_list_len = HNS_ROCE_FRMR_MAX_PA;
205         }
206
207         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_XRC)
208                 props->device_cap_flags |= IB_DEVICE_XRC;
209
210         return 0;
211 }
212
213 static int hns_roce_query_port(struct ib_device *ib_dev, u32 port_num,
214                                struct ib_port_attr *props)
215 {
216         struct hns_roce_dev *hr_dev = to_hr_dev(ib_dev);
217         struct device *dev = hr_dev->dev;
218         struct net_device *net_dev;
219         unsigned long flags;
220         enum ib_mtu mtu;
221         u32 port;
222
223         port = port_num - 1;
224
225         /* props being zeroed by the caller, avoid zeroing it here */
226
227         props->max_mtu = hr_dev->caps.max_mtu;
228         props->gid_tbl_len = hr_dev->caps.gid_table_len[port];
229         props->port_cap_flags = IB_PORT_CM_SUP | IB_PORT_REINIT_SUP |
230                                 IB_PORT_VENDOR_CLASS_SUP |
231                                 IB_PORT_BOOT_MGMT_SUP;
232         props->max_msg_sz = HNS_ROCE_MAX_MSG_LEN;
233         props->pkey_tbl_len = 1;
234         props->active_width = IB_WIDTH_4X;
235         props->active_speed = 1;
236
237         spin_lock_irqsave(&hr_dev->iboe.lock, flags);
238
239         net_dev = hr_dev->iboe.netdevs[port];
240         if (!net_dev) {
241                 spin_unlock_irqrestore(&hr_dev->iboe.lock, flags);
242                 dev_err(dev, "find netdev %u failed!\n", port);
243                 return -EINVAL;
244         }
245
246         mtu = iboe_get_mtu(net_dev->mtu);
247         props->active_mtu = mtu ? min(props->max_mtu, mtu) : IB_MTU_256;
248         props->state = netif_running(net_dev) && netif_carrier_ok(net_dev) ?
249                                IB_PORT_ACTIVE :
250                                IB_PORT_DOWN;
251         props->phys_state = props->state == IB_PORT_ACTIVE ?
252                                     IB_PORT_PHYS_STATE_LINK_UP :
253                                     IB_PORT_PHYS_STATE_DISABLED;
254
255         spin_unlock_irqrestore(&hr_dev->iboe.lock, flags);
256
257         return 0;
258 }
259
260 static enum rdma_link_layer hns_roce_get_link_layer(struct ib_device *device,
261                                                     u32 port_num)
262 {
263         return IB_LINK_LAYER_ETHERNET;
264 }
265
266 static int hns_roce_query_pkey(struct ib_device *ib_dev, u32 port, u16 index,
267                                u16 *pkey)
268 {
269         if (index > 0)
270                 return -EINVAL;
271
272         *pkey = PKEY_ID;
273
274         return 0;
275 }
276
277 static int hns_roce_modify_device(struct ib_device *ib_dev, int mask,
278                                   struct ib_device_modify *props)
279 {
280         unsigned long flags;
281
282         if (mask & ~IB_DEVICE_MODIFY_NODE_DESC)
283                 return -EOPNOTSUPP;
284
285         if (mask & IB_DEVICE_MODIFY_NODE_DESC) {
286                 spin_lock_irqsave(&to_hr_dev(ib_dev)->sm_lock, flags);
287                 memcpy(ib_dev->node_desc, props->node_desc, NODE_DESC_SIZE);
288                 spin_unlock_irqrestore(&to_hr_dev(ib_dev)->sm_lock, flags);
289         }
290
291         return 0;
292 }
293
294 struct hns_user_mmap_entry *
295 hns_roce_user_mmap_entry_insert(struct ib_ucontext *ucontext, u64 address,
296                                 size_t length,
297                                 enum hns_roce_mmap_type mmap_type)
298 {
299         struct hns_user_mmap_entry *entry;
300         int ret;
301
302         entry = kzalloc(sizeof(*entry), GFP_KERNEL);
303         if (!entry)
304                 return NULL;
305
306         entry->address = address;
307         entry->mmap_type = mmap_type;
308
309         switch (mmap_type) {
310         /* pgoff 0 must be used by DB for compatibility */
311         case HNS_ROCE_MMAP_TYPE_DB:
312                 ret = rdma_user_mmap_entry_insert_exact(
313                                 ucontext, &entry->rdma_entry, length, 0);
314                 break;
315         case HNS_ROCE_MMAP_TYPE_DWQE:
316                 ret = rdma_user_mmap_entry_insert_range(
317                                 ucontext, &entry->rdma_entry, length, 1,
318                                 U32_MAX);
319                 break;
320         default:
321                 ret = -EINVAL;
322                 break;
323         }
324
325         if (ret) {
326                 kfree(entry);
327                 return NULL;
328         }
329
330         return entry;
331 }
332
333 static void hns_roce_dealloc_uar_entry(struct hns_roce_ucontext *context)
334 {
335         if (context->db_mmap_entry)
336                 rdma_user_mmap_entry_remove(
337                         &context->db_mmap_entry->rdma_entry);
338 }
339
340 static int hns_roce_alloc_uar_entry(struct ib_ucontext *uctx)
341 {
342         struct hns_roce_ucontext *context = to_hr_ucontext(uctx);
343         u64 address;
344
345         address = context->uar.pfn << PAGE_SHIFT;
346         context->db_mmap_entry = hns_roce_user_mmap_entry_insert(
347                 uctx, address, PAGE_SIZE, HNS_ROCE_MMAP_TYPE_DB);
348         if (!context->db_mmap_entry)
349                 return -ENOMEM;
350
351         return 0;
352 }
353
354 static int hns_roce_alloc_ucontext(struct ib_ucontext *uctx,
355                                    struct ib_udata *udata)
356 {
357         struct hns_roce_ucontext *context = to_hr_ucontext(uctx);
358         struct hns_roce_dev *hr_dev = to_hr_dev(uctx->device);
359         struct hns_roce_ib_alloc_ucontext_resp resp = {};
360         struct hns_roce_ib_alloc_ucontext ucmd = {};
361         int ret;
362
363         if (!hr_dev->active)
364                 return -EAGAIN;
365
366         resp.qp_tab_size = hr_dev->caps.num_qps;
367         resp.srq_tab_size = hr_dev->caps.num_srqs;
368
369         ret = ib_copy_from_udata(&ucmd, udata,
370                                  min(udata->inlen, sizeof(ucmd)));
371         if (ret)
372                 return ret;
373
374         if (hr_dev->pci_dev->revision >= PCI_REVISION_ID_HIP09)
375                 context->config = ucmd.config & HNS_ROCE_EXSGE_FLAGS;
376
377         if (context->config & HNS_ROCE_EXSGE_FLAGS) {
378                 resp.config |= HNS_ROCE_RSP_EXSGE_FLAGS;
379                 resp.max_inline_data = hr_dev->caps.max_sq_inline;
380         }
381
382         ret = hns_roce_uar_alloc(hr_dev, &context->uar);
383         if (ret)
384                 goto error_fail_uar_alloc;
385
386         ret = hns_roce_alloc_uar_entry(uctx);
387         if (ret)
388                 goto error_fail_uar_entry;
389
390         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_CQ_RECORD_DB ||
391             hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_QP_RECORD_DB) {
392                 INIT_LIST_HEAD(&context->page_list);
393                 mutex_init(&context->page_mutex);
394         }
395
396         resp.cqe_size = hr_dev->caps.cqe_sz;
397
398         ret = ib_copy_to_udata(udata, &resp,
399                                min(udata->outlen, sizeof(resp)));
400         if (ret)
401                 goto error_fail_copy_to_udata;
402
403         return 0;
404
405 error_fail_copy_to_udata:
406         hns_roce_dealloc_uar_entry(context);
407
408 error_fail_uar_entry:
409         ida_free(&hr_dev->uar_ida.ida, (int)context->uar.logic_idx);
410
411 error_fail_uar_alloc:
412         return ret;
413 }
414
415 static void hns_roce_dealloc_ucontext(struct ib_ucontext *ibcontext)
416 {
417         struct hns_roce_ucontext *context = to_hr_ucontext(ibcontext);
418         struct hns_roce_dev *hr_dev = to_hr_dev(ibcontext->device);
419
420         hns_roce_dealloc_uar_entry(context);
421
422         ida_free(&hr_dev->uar_ida.ida, (int)context->uar.logic_idx);
423 }
424
425 static int hns_roce_mmap(struct ib_ucontext *uctx, struct vm_area_struct *vma)
426 {
427         struct rdma_user_mmap_entry *rdma_entry;
428         struct hns_user_mmap_entry *entry;
429         phys_addr_t pfn;
430         pgprot_t prot;
431         int ret;
432
433         rdma_entry = rdma_user_mmap_entry_get_pgoff(uctx, vma->vm_pgoff);
434         if (!rdma_entry)
435                 return -EINVAL;
436
437         entry = to_hns_mmap(rdma_entry);
438         pfn = entry->address >> PAGE_SHIFT;
439
440         switch (entry->mmap_type) {
441         case HNS_ROCE_MMAP_TYPE_DB:
442         case HNS_ROCE_MMAP_TYPE_DWQE:
443                 prot = pgprot_device(vma->vm_page_prot);
444                 break;
445         default:
446                 ret = -EINVAL;
447                 goto out;
448         }
449
450         ret = rdma_user_mmap_io(uctx, vma, pfn, rdma_entry->npages * PAGE_SIZE,
451                                 prot, rdma_entry);
452
453 out:
454         rdma_user_mmap_entry_put(rdma_entry);
455         return ret;
456 }
457
458 static void hns_roce_free_mmap(struct rdma_user_mmap_entry *rdma_entry)
459 {
460         struct hns_user_mmap_entry *entry = to_hns_mmap(rdma_entry);
461
462         kfree(entry);
463 }
464
465 static int hns_roce_port_immutable(struct ib_device *ib_dev, u32 port_num,
466                                    struct ib_port_immutable *immutable)
467 {
468         struct ib_port_attr attr;
469         int ret;
470
471         ret = ib_query_port(ib_dev, port_num, &attr);
472         if (ret)
473                 return ret;
474
475         immutable->pkey_tbl_len = attr.pkey_tbl_len;
476         immutable->gid_tbl_len = attr.gid_tbl_len;
477
478         immutable->max_mad_size = IB_MGMT_MAD_SIZE;
479         immutable->core_cap_flags = RDMA_CORE_PORT_IBA_ROCE;
480         if (to_hr_dev(ib_dev)->caps.flags & HNS_ROCE_CAP_FLAG_ROCE_V1_V2)
481                 immutable->core_cap_flags |= RDMA_CORE_PORT_IBA_ROCE_UDP_ENCAP;
482
483         return 0;
484 }
485
486 static void hns_roce_disassociate_ucontext(struct ib_ucontext *ibcontext)
487 {
488 }
489
490 static void hns_roce_get_fw_ver(struct ib_device *device, char *str)
491 {
492         u64 fw_ver = to_hr_dev(device)->caps.fw_ver;
493         unsigned int major, minor, sub_minor;
494
495         major = upper_32_bits(fw_ver);
496         minor = high_16_bits(lower_32_bits(fw_ver));
497         sub_minor = low_16_bits(fw_ver);
498
499         snprintf(str, IB_FW_VERSION_NAME_MAX, "%u.%u.%04u", major, minor,
500                  sub_minor);
501 }
502
503 static void hns_roce_unregister_device(struct hns_roce_dev *hr_dev)
504 {
505         struct hns_roce_ib_iboe *iboe = &hr_dev->iboe;
506
507         hr_dev->active = false;
508         unregister_netdevice_notifier(&iboe->nb);
509         ib_unregister_device(&hr_dev->ib_dev);
510 }
511
512 static const struct ib_device_ops hns_roce_dev_ops = {
513         .owner = THIS_MODULE,
514         .driver_id = RDMA_DRIVER_HNS,
515         .uverbs_abi_ver = 1,
516         .uverbs_no_driver_id_binding = 1,
517
518         .get_dev_fw_str = hns_roce_get_fw_ver,
519         .add_gid = hns_roce_add_gid,
520         .alloc_pd = hns_roce_alloc_pd,
521         .alloc_ucontext = hns_roce_alloc_ucontext,
522         .create_ah = hns_roce_create_ah,
523         .create_user_ah = hns_roce_create_ah,
524         .create_cq = hns_roce_create_cq,
525         .create_qp = hns_roce_create_qp,
526         .dealloc_pd = hns_roce_dealloc_pd,
527         .dealloc_ucontext = hns_roce_dealloc_ucontext,
528         .del_gid = hns_roce_del_gid,
529         .dereg_mr = hns_roce_dereg_mr,
530         .destroy_ah = hns_roce_destroy_ah,
531         .destroy_cq = hns_roce_destroy_cq,
532         .disassociate_ucontext = hns_roce_disassociate_ucontext,
533         .get_dma_mr = hns_roce_get_dma_mr,
534         .get_link_layer = hns_roce_get_link_layer,
535         .get_port_immutable = hns_roce_port_immutable,
536         .mmap = hns_roce_mmap,
537         .mmap_free = hns_roce_free_mmap,
538         .modify_device = hns_roce_modify_device,
539         .modify_qp = hns_roce_modify_qp,
540         .query_ah = hns_roce_query_ah,
541         .query_device = hns_roce_query_device,
542         .query_pkey = hns_roce_query_pkey,
543         .query_port = hns_roce_query_port,
544         .reg_user_mr = hns_roce_reg_user_mr,
545
546         INIT_RDMA_OBJ_SIZE(ib_ah, hns_roce_ah, ibah),
547         INIT_RDMA_OBJ_SIZE(ib_cq, hns_roce_cq, ib_cq),
548         INIT_RDMA_OBJ_SIZE(ib_pd, hns_roce_pd, ibpd),
549         INIT_RDMA_OBJ_SIZE(ib_qp, hns_roce_qp, ibqp),
550         INIT_RDMA_OBJ_SIZE(ib_ucontext, hns_roce_ucontext, ibucontext),
551 };
552
553 static const struct ib_device_ops hns_roce_dev_mr_ops = {
554         .rereg_user_mr = hns_roce_rereg_user_mr,
555 };
556
557 static const struct ib_device_ops hns_roce_dev_mw_ops = {
558         .alloc_mw = hns_roce_alloc_mw,
559         .dealloc_mw = hns_roce_dealloc_mw,
560
561         INIT_RDMA_OBJ_SIZE(ib_mw, hns_roce_mw, ibmw),
562 };
563
564 static const struct ib_device_ops hns_roce_dev_frmr_ops = {
565         .alloc_mr = hns_roce_alloc_mr,
566         .map_mr_sg = hns_roce_map_mr_sg,
567 };
568
569 static const struct ib_device_ops hns_roce_dev_srq_ops = {
570         .create_srq = hns_roce_create_srq,
571         .destroy_srq = hns_roce_destroy_srq,
572
573         INIT_RDMA_OBJ_SIZE(ib_srq, hns_roce_srq, ibsrq),
574 };
575
576 static const struct ib_device_ops hns_roce_dev_xrcd_ops = {
577         .alloc_xrcd = hns_roce_alloc_xrcd,
578         .dealloc_xrcd = hns_roce_dealloc_xrcd,
579
580         INIT_RDMA_OBJ_SIZE(ib_xrcd, hns_roce_xrcd, ibxrcd),
581 };
582
583 static const struct ib_device_ops hns_roce_dev_restrack_ops = {
584         .fill_res_cq_entry = hns_roce_fill_res_cq_entry,
585         .fill_res_cq_entry_raw = hns_roce_fill_res_cq_entry_raw,
586         .fill_res_qp_entry = hns_roce_fill_res_qp_entry,
587         .fill_res_qp_entry_raw = hns_roce_fill_res_qp_entry_raw,
588         .fill_res_mr_entry = hns_roce_fill_res_mr_entry,
589         .fill_res_mr_entry_raw = hns_roce_fill_res_mr_entry_raw,
590 };
591
592 static int hns_roce_register_device(struct hns_roce_dev *hr_dev)
593 {
594         int ret;
595         struct hns_roce_ib_iboe *iboe = NULL;
596         struct ib_device *ib_dev = NULL;
597         struct device *dev = hr_dev->dev;
598         unsigned int i;
599
600         iboe = &hr_dev->iboe;
601         spin_lock_init(&iboe->lock);
602
603         ib_dev = &hr_dev->ib_dev;
604
605         ib_dev->node_type = RDMA_NODE_IB_CA;
606         ib_dev->dev.parent = dev;
607
608         ib_dev->phys_port_cnt = hr_dev->caps.num_ports;
609         ib_dev->local_dma_lkey = hr_dev->caps.reserved_lkey;
610         ib_dev->num_comp_vectors = hr_dev->caps.num_comp_vectors;
611
612         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_REREG_MR)
613                 ib_set_device_ops(ib_dev, &hns_roce_dev_mr_ops);
614
615         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_MW)
616                 ib_set_device_ops(ib_dev, &hns_roce_dev_mw_ops);
617
618         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_FRMR)
619                 ib_set_device_ops(ib_dev, &hns_roce_dev_frmr_ops);
620
621         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_SRQ) {
622                 ib_set_device_ops(ib_dev, &hns_roce_dev_srq_ops);
623                 ib_set_device_ops(ib_dev, hr_dev->hw->hns_roce_dev_srq_ops);
624         }
625
626         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_XRC)
627                 ib_set_device_ops(ib_dev, &hns_roce_dev_xrcd_ops);
628
629         ib_set_device_ops(ib_dev, hr_dev->hw->hns_roce_dev_ops);
630         ib_set_device_ops(ib_dev, &hns_roce_dev_ops);
631         ib_set_device_ops(ib_dev, &hns_roce_dev_restrack_ops);
632         for (i = 0; i < hr_dev->caps.num_ports; i++) {
633                 if (!hr_dev->iboe.netdevs[i])
634                         continue;
635
636                 ret = ib_device_set_netdev(ib_dev, hr_dev->iboe.netdevs[i],
637                                            i + 1);
638                 if (ret)
639                         return ret;
640         }
641         dma_set_max_seg_size(dev, UINT_MAX);
642         ret = ib_register_device(ib_dev, "hns_%d", dev);
643         if (ret) {
644                 dev_err(dev, "ib_register_device failed!\n");
645                 return ret;
646         }
647
648         ret = hns_roce_setup_mtu_mac(hr_dev);
649         if (ret) {
650                 dev_err(dev, "setup_mtu_mac failed!\n");
651                 goto error_failed_setup_mtu_mac;
652         }
653
654         iboe->nb.notifier_call = hns_roce_netdev_event;
655         ret = register_netdevice_notifier(&iboe->nb);
656         if (ret) {
657                 dev_err(dev, "register_netdevice_notifier failed!\n");
658                 goto error_failed_setup_mtu_mac;
659         }
660
661         hr_dev->active = true;
662         return 0;
663
664 error_failed_setup_mtu_mac:
665         ib_unregister_device(ib_dev);
666
667         return ret;
668 }
669
670 static int hns_roce_init_hem(struct hns_roce_dev *hr_dev)
671 {
672         struct device *dev = hr_dev->dev;
673         int ret;
674
675         ret = hns_roce_init_hem_table(hr_dev, &hr_dev->mr_table.mtpt_table,
676                                       HEM_TYPE_MTPT, hr_dev->caps.mtpt_entry_sz,
677                                       hr_dev->caps.num_mtpts);
678         if (ret) {
679                 dev_err(dev, "failed to init MTPT context memory, aborting.\n");
680                 return ret;
681         }
682
683         ret = hns_roce_init_hem_table(hr_dev, &hr_dev->qp_table.qp_table,
684                                       HEM_TYPE_QPC, hr_dev->caps.qpc_sz,
685                                       hr_dev->caps.num_qps);
686         if (ret) {
687                 dev_err(dev, "failed to init QP context memory, aborting.\n");
688                 goto err_unmap_dmpt;
689         }
690
691         ret = hns_roce_init_hem_table(hr_dev, &hr_dev->qp_table.irrl_table,
692                                       HEM_TYPE_IRRL,
693                                       hr_dev->caps.irrl_entry_sz *
694                                       hr_dev->caps.max_qp_init_rdma,
695                                       hr_dev->caps.num_qps);
696         if (ret) {
697                 dev_err(dev, "failed to init irrl_table memory, aborting.\n");
698                 goto err_unmap_qp;
699         }
700
701         if (hr_dev->caps.trrl_entry_sz) {
702                 ret = hns_roce_init_hem_table(hr_dev,
703                                               &hr_dev->qp_table.trrl_table,
704                                               HEM_TYPE_TRRL,
705                                               hr_dev->caps.trrl_entry_sz *
706                                               hr_dev->caps.max_qp_dest_rdma,
707                                               hr_dev->caps.num_qps);
708                 if (ret) {
709                         dev_err(dev,
710                                 "failed to init trrl_table memory, aborting.\n");
711                         goto err_unmap_irrl;
712                 }
713         }
714
715         ret = hns_roce_init_hem_table(hr_dev, &hr_dev->cq_table.table,
716                                       HEM_TYPE_CQC, hr_dev->caps.cqc_entry_sz,
717                                       hr_dev->caps.num_cqs);
718         if (ret) {
719                 dev_err(dev, "failed to init CQ context memory, aborting.\n");
720                 goto err_unmap_trrl;
721         }
722
723         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_SRQ) {
724                 ret = hns_roce_init_hem_table(hr_dev, &hr_dev->srq_table.table,
725                                               HEM_TYPE_SRQC,
726                                               hr_dev->caps.srqc_entry_sz,
727                                               hr_dev->caps.num_srqs);
728                 if (ret) {
729                         dev_err(dev,
730                                 "failed to init SRQ context memory, aborting.\n");
731                         goto err_unmap_cq;
732                 }
733         }
734
735         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_QP_FLOW_CTRL) {
736                 ret = hns_roce_init_hem_table(hr_dev,
737                                               &hr_dev->qp_table.sccc_table,
738                                               HEM_TYPE_SCCC,
739                                               hr_dev->caps.sccc_sz,
740                                               hr_dev->caps.num_qps);
741                 if (ret) {
742                         dev_err(dev,
743                                 "failed to init SCC context memory, aborting.\n");
744                         goto err_unmap_srq;
745                 }
746         }
747
748         if (hr_dev->caps.qpc_timer_entry_sz) {
749                 ret = hns_roce_init_hem_table(hr_dev, &hr_dev->qpc_timer_table,
750                                               HEM_TYPE_QPC_TIMER,
751                                               hr_dev->caps.qpc_timer_entry_sz,
752                                               hr_dev->caps.qpc_timer_bt_num);
753                 if (ret) {
754                         dev_err(dev,
755                                 "failed to init QPC timer memory, aborting.\n");
756                         goto err_unmap_ctx;
757                 }
758         }
759
760         if (hr_dev->caps.cqc_timer_entry_sz) {
761                 ret = hns_roce_init_hem_table(hr_dev, &hr_dev->cqc_timer_table,
762                                               HEM_TYPE_CQC_TIMER,
763                                               hr_dev->caps.cqc_timer_entry_sz,
764                                               hr_dev->caps.cqc_timer_bt_num);
765                 if (ret) {
766                         dev_err(dev,
767                                 "failed to init CQC timer memory, aborting.\n");
768                         goto err_unmap_qpc_timer;
769                 }
770         }
771
772         if (hr_dev->caps.gmv_entry_sz) {
773                 ret = hns_roce_init_hem_table(hr_dev, &hr_dev->gmv_table,
774                                               HEM_TYPE_GMV,
775                                               hr_dev->caps.gmv_entry_sz,
776                                               hr_dev->caps.gmv_entry_num);
777                 if (ret) {
778                         dev_err(dev,
779                                 "failed to init gmv table memory, ret = %d\n",
780                                 ret);
781                         goto err_unmap_cqc_timer;
782                 }
783         }
784
785         return 0;
786
787 err_unmap_cqc_timer:
788         if (hr_dev->caps.cqc_timer_entry_sz)
789                 hns_roce_cleanup_hem_table(hr_dev, &hr_dev->cqc_timer_table);
790
791 err_unmap_qpc_timer:
792         if (hr_dev->caps.qpc_timer_entry_sz)
793                 hns_roce_cleanup_hem_table(hr_dev, &hr_dev->qpc_timer_table);
794
795 err_unmap_ctx:
796         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_QP_FLOW_CTRL)
797                 hns_roce_cleanup_hem_table(hr_dev,
798                                            &hr_dev->qp_table.sccc_table);
799 err_unmap_srq:
800         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_SRQ)
801                 hns_roce_cleanup_hem_table(hr_dev, &hr_dev->srq_table.table);
802
803 err_unmap_cq:
804         hns_roce_cleanup_hem_table(hr_dev, &hr_dev->cq_table.table);
805
806 err_unmap_trrl:
807         if (hr_dev->caps.trrl_entry_sz)
808                 hns_roce_cleanup_hem_table(hr_dev,
809                                            &hr_dev->qp_table.trrl_table);
810
811 err_unmap_irrl:
812         hns_roce_cleanup_hem_table(hr_dev, &hr_dev->qp_table.irrl_table);
813
814 err_unmap_qp:
815         hns_roce_cleanup_hem_table(hr_dev, &hr_dev->qp_table.qp_table);
816
817 err_unmap_dmpt:
818         hns_roce_cleanup_hem_table(hr_dev, &hr_dev->mr_table.mtpt_table);
819
820         return ret;
821 }
822
823 /**
824  * hns_roce_setup_hca - setup host channel adapter
825  * @hr_dev: pointer to hns roce device
826  * Return : int
827  */
828 static int hns_roce_setup_hca(struct hns_roce_dev *hr_dev)
829 {
830         struct device *dev = hr_dev->dev;
831         int ret;
832
833         spin_lock_init(&hr_dev->sm_lock);
834
835         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_CQ_RECORD_DB ||
836             hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_QP_RECORD_DB) {
837                 INIT_LIST_HEAD(&hr_dev->pgdir_list);
838                 mutex_init(&hr_dev->pgdir_mutex);
839         }
840
841         hns_roce_init_uar_table(hr_dev);
842
843         ret = hns_roce_uar_alloc(hr_dev, &hr_dev->priv_uar);
844         if (ret) {
845                 dev_err(dev, "failed to allocate priv_uar.\n");
846                 goto err_uar_table_free;
847         }
848
849         ret = hns_roce_init_qp_table(hr_dev);
850         if (ret) {
851                 dev_err(dev, "failed to init qp_table.\n");
852                 goto err_uar_table_free;
853         }
854
855         hns_roce_init_pd_table(hr_dev);
856
857         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_XRC)
858                 hns_roce_init_xrcd_table(hr_dev);
859
860         hns_roce_init_mr_table(hr_dev);
861
862         hns_roce_init_cq_table(hr_dev);
863
864         if (hr_dev->caps.flags & HNS_ROCE_CAP_FLAG_SRQ)
865                 hns_roce_init_srq_table(hr_dev);
866
867         return 0;
868
869 err_uar_table_free:
870         ida_destroy(&hr_dev->uar_ida.ida);
871         return ret;
872 }
873
874 static void check_and_get_armed_cq(struct list_head *cq_list, struct ib_cq *cq)
875 {
876         struct hns_roce_cq *hr_cq = to_hr_cq(cq);
877         unsigned long flags;
878
879         spin_lock_irqsave(&hr_cq->lock, flags);
880         if (cq->comp_handler) {
881                 if (!hr_cq->is_armed) {
882                         hr_cq->is_armed = 1;
883                         list_add_tail(&hr_cq->node, cq_list);
884                 }
885         }
886         spin_unlock_irqrestore(&hr_cq->lock, flags);
887 }
888
889 void hns_roce_handle_device_err(struct hns_roce_dev *hr_dev)
890 {
891         struct hns_roce_qp *hr_qp;
892         struct hns_roce_cq *hr_cq;
893         struct list_head cq_list;
894         unsigned long flags_qp;
895         unsigned long flags;
896
897         INIT_LIST_HEAD(&cq_list);
898
899         spin_lock_irqsave(&hr_dev->qp_list_lock, flags);
900         list_for_each_entry(hr_qp, &hr_dev->qp_list, node) {
901                 spin_lock_irqsave(&hr_qp->sq.lock, flags_qp);
902                 if (hr_qp->sq.tail != hr_qp->sq.head)
903                         check_and_get_armed_cq(&cq_list, hr_qp->ibqp.send_cq);
904                 spin_unlock_irqrestore(&hr_qp->sq.lock, flags_qp);
905
906                 spin_lock_irqsave(&hr_qp->rq.lock, flags_qp);
907                 if ((!hr_qp->ibqp.srq) && (hr_qp->rq.tail != hr_qp->rq.head))
908                         check_and_get_armed_cq(&cq_list, hr_qp->ibqp.recv_cq);
909                 spin_unlock_irqrestore(&hr_qp->rq.lock, flags_qp);
910         }
911
912         list_for_each_entry(hr_cq, &cq_list, node)
913                 hns_roce_cq_completion(hr_dev, hr_cq->cqn);
914
915         spin_unlock_irqrestore(&hr_dev->qp_list_lock, flags);
916 }
917
918 int hns_roce_init(struct hns_roce_dev *hr_dev)
919 {
920         struct device *dev = hr_dev->dev;
921         int ret;
922
923         hr_dev->is_reset = false;
924
925         if (hr_dev->hw->cmq_init) {
926                 ret = hr_dev->hw->cmq_init(hr_dev);
927                 if (ret) {
928                         dev_err(dev, "init RoCE Command Queue failed!\n");
929                         return ret;
930                 }
931         }
932
933         ret = hr_dev->hw->hw_profile(hr_dev);
934         if (ret) {
935                 dev_err(dev, "get RoCE engine profile failed!\n");
936                 goto error_failed_cmd_init;
937         }
938
939         ret = hns_roce_cmd_init(hr_dev);
940         if (ret) {
941                 dev_err(dev, "cmd init failed!\n");
942                 goto error_failed_cmd_init;
943         }
944
945         /* EQ depends on poll mode, event mode depends on EQ */
946         ret = hr_dev->hw->init_eq(hr_dev);
947         if (ret) {
948                 dev_err(dev, "eq init failed!\n");
949                 goto error_failed_eq_table;
950         }
951
952         if (hr_dev->cmd_mod) {
953                 ret = hns_roce_cmd_use_events(hr_dev);
954                 if (ret)
955                         dev_warn(dev,
956                                  "Cmd event  mode failed, set back to poll!\n");
957         }
958
959         ret = hns_roce_init_hem(hr_dev);
960         if (ret) {
961                 dev_err(dev, "init HEM(Hardware Entry Memory) failed!\n");
962                 goto error_failed_init_hem;
963         }
964
965         ret = hns_roce_setup_hca(hr_dev);
966         if (ret) {
967                 dev_err(dev, "setup hca failed!\n");
968                 goto error_failed_setup_hca;
969         }
970
971         if (hr_dev->hw->hw_init) {
972                 ret = hr_dev->hw->hw_init(hr_dev);
973                 if (ret) {
974                         dev_err(dev, "hw_init failed!\n");
975                         goto error_failed_engine_init;
976                 }
977         }
978
979         INIT_LIST_HEAD(&hr_dev->qp_list);
980         spin_lock_init(&hr_dev->qp_list_lock);
981         INIT_LIST_HEAD(&hr_dev->dip_list);
982         spin_lock_init(&hr_dev->dip_list_lock);
983
984         ret = hns_roce_register_device(hr_dev);
985         if (ret)
986                 goto error_failed_register_device;
987
988         return 0;
989
990 error_failed_register_device:
991         if (hr_dev->hw->hw_exit)
992                 hr_dev->hw->hw_exit(hr_dev);
993
994 error_failed_engine_init:
995         hns_roce_cleanup_bitmap(hr_dev);
996
997 error_failed_setup_hca:
998         hns_roce_cleanup_hem(hr_dev);
999
1000 error_failed_init_hem:
1001         if (hr_dev->cmd_mod)
1002                 hns_roce_cmd_use_polling(hr_dev);
1003         hr_dev->hw->cleanup_eq(hr_dev);
1004
1005 error_failed_eq_table:
1006         hns_roce_cmd_cleanup(hr_dev);
1007
1008 error_failed_cmd_init:
1009         if (hr_dev->hw->cmq_exit)
1010                 hr_dev->hw->cmq_exit(hr_dev);
1011
1012         return ret;
1013 }
1014
1015 void hns_roce_exit(struct hns_roce_dev *hr_dev)
1016 {
1017         hns_roce_unregister_device(hr_dev);
1018
1019         if (hr_dev->hw->hw_exit)
1020                 hr_dev->hw->hw_exit(hr_dev);
1021         hns_roce_cleanup_bitmap(hr_dev);
1022         hns_roce_cleanup_hem(hr_dev);
1023
1024         if (hr_dev->cmd_mod)
1025                 hns_roce_cmd_use_polling(hr_dev);
1026
1027         hr_dev->hw->cleanup_eq(hr_dev);
1028         hns_roce_cmd_cleanup(hr_dev);
1029         if (hr_dev->hw->cmq_exit)
1030                 hr_dev->hw->cmq_exit(hr_dev);
1031 }
1032
1033 MODULE_LICENSE("Dual BSD/GPL");
1034 MODULE_AUTHOR("Wei Hu <xavier.huwei@huawei.com>");
1035 MODULE_AUTHOR("Nenglong Zhao <zhaonenglong@hisilicon.com>");
1036 MODULE_AUTHOR("Lijun Ou <oulijun@huawei.com>");
1037 MODULE_DESCRIPTION("HNS RoCE Driver");