| 1 | /* SPDX-License-Identifier: GPL-2.0 OR BSD-2-Clause */ |
| 2 | /* |
| 3 | * Copyright 2018-2025 Amazon.com, Inc. or its affiliates. All rights reserved. |
| 4 | */ |
| 5 | |
| 6 | #ifndef _EFA_H_ |
| 7 | #define _EFA_H_ |
| 8 | |
| 9 | #include <linux/bitops.h> |
| 10 | #include <linux/interrupt.h> |
| 11 | #include <linux/pci.h> |
| 12 | |
| 13 | #include <rdma/efa-abi.h> |
| 14 | #include <rdma/ib_verbs.h> |
| 15 | |
| 16 | #include "efa_com_cmd.h" |
| 17 | |
| 18 | #define DRV_MODULE_NAME "efa" |
| 19 | #define DEVICE_NAME "Elastic Fabric Adapter (EFA)" |
| 20 | |
| 21 | #define EFA_IRQNAME_SIZE 40 |
| 22 | |
| 23 | #define EFA_MGMNT_MSIX_VEC_IDX 0 |
| 24 | #define EFA_COMP_EQS_VEC_BASE 1 |
| 25 | |
| 26 | struct efa_irq { |
| 27 | irq_handler_t handler; |
| 28 | void *data; |
| 29 | u32 irqn; |
| 30 | u32 vector; |
| 31 | cpumask_t affinity_hint_mask; |
| 32 | char name[EFA_IRQNAME_SIZE]; |
| 33 | }; |
| 34 | |
| 35 | /* Don't use anything other than atomic64 */ |
| 36 | struct efa_stats { |
| 37 | atomic64_t alloc_pd_err; |
| 38 | atomic64_t create_qp_err; |
| 39 | atomic64_t create_cq_err; |
| 40 | atomic64_t reg_mr_err; |
| 41 | atomic64_t alloc_ucontext_err; |
| 42 | atomic64_t create_ah_err; |
| 43 | atomic64_t mmap_err; |
| 44 | atomic64_t keep_alive_rcvd; |
| 45 | }; |
| 46 | |
| 47 | struct efa_dev { |
| 48 | struct ib_device ibdev; |
| 49 | struct efa_com_dev edev; |
| 50 | struct pci_dev *pdev; |
| 51 | struct efa_com_get_device_attr_result dev_attr; |
| 52 | |
| 53 | u64 reg_bar_addr; |
| 54 | u64 reg_bar_len; |
| 55 | u64 mem_bar_addr; |
| 56 | u64 mem_bar_len; |
| 57 | u64 db_bar_addr; |
| 58 | u64 db_bar_len; |
| 59 | |
| 60 | u32 num_irq_vectors; |
| 61 | u32 admin_msix_vector_idx; |
| 62 | struct efa_irq admin_irq; |
| 63 | |
| 64 | struct efa_stats stats; |
| 65 | |
| 66 | /* Array of completion EQs */ |
| 67 | struct efa_eq *eqs; |
| 68 | u32 neqs; |
| 69 | |
| 70 | /* Only stores CQs with interrupts enabled */ |
| 71 | struct xarray cqs_xa; |
| 72 | }; |
| 73 | |
| 74 | struct efa_ucontext { |
| 75 | struct ib_ucontext ibucontext; |
| 76 | u16 uarn; |
| 77 | }; |
| 78 | |
| 79 | struct efa_pd { |
| 80 | struct ib_pd ibpd; |
| 81 | u16 pdn; |
| 82 | }; |
| 83 | |
| 84 | struct efa_mr_interconnect_info { |
| 85 | u16 recv_ic_id; |
| 86 | u16 rdma_read_ic_id; |
| 87 | u16 rdma_recv_ic_id; |
| 88 | u8 recv_ic_id_valid : 1; |
| 89 | u8 rdma_read_ic_id_valid : 1; |
| 90 | u8 rdma_recv_ic_id_valid : 1; |
| 91 | }; |
| 92 | |
| 93 | struct efa_mr { |
| 94 | struct ib_mr ibmr; |
| 95 | struct ib_umem *umem; |
| 96 | struct efa_mr_interconnect_info ic_info; |
| 97 | }; |
| 98 | |
| 99 | struct efa_cq { |
| 100 | struct ib_cq ibcq; |
| 101 | struct efa_ucontext *ucontext; |
| 102 | dma_addr_t dma_addr; |
| 103 | void *cpu_addr; |
| 104 | struct rdma_user_mmap_entry *mmap_entry; |
| 105 | struct rdma_user_mmap_entry *db_mmap_entry; |
| 106 | size_t size; |
| 107 | u16 cq_idx; |
| 108 | /* NULL when no interrupts requested */ |
| 109 | struct efa_eq *eq; |
| 110 | struct ib_umem *umem; |
| 111 | }; |
| 112 | |
| 113 | struct efa_qp { |
| 114 | struct ib_qp ibqp; |
| 115 | dma_addr_t rq_dma_addr; |
| 116 | void *rq_cpu_addr; |
| 117 | size_t rq_size; |
| 118 | enum ib_qp_state state; |
| 119 | |
| 120 | /* Used for saving mmap_xa entries */ |
| 121 | struct rdma_user_mmap_entry *sq_db_mmap_entry; |
| 122 | struct rdma_user_mmap_entry *llq_desc_mmap_entry; |
| 123 | struct rdma_user_mmap_entry *rq_db_mmap_entry; |
| 124 | struct rdma_user_mmap_entry *rq_mmap_entry; |
| 125 | |
| 126 | u32 qp_handle; |
| 127 | u32 max_send_wr; |
| 128 | u32 max_recv_wr; |
| 129 | u32 max_send_sge; |
| 130 | u32 max_recv_sge; |
| 131 | u32 max_inline_data; |
| 132 | }; |
| 133 | |
| 134 | struct efa_ah { |
| 135 | struct ib_ah ibah; |
| 136 | u16 ah; |
| 137 | /* dest_addr */ |
| 138 | u8 id[EFA_GID_SIZE]; |
| 139 | }; |
| 140 | |
| 141 | struct efa_eq { |
| 142 | struct efa_com_eq eeq; |
| 143 | struct efa_irq irq; |
| 144 | }; |
| 145 | |
| 146 | int efa_query_device(struct ib_device *ibdev, |
| 147 | struct ib_device_attr *props, |
| 148 | struct ib_udata *udata); |
| 149 | int efa_query_port(struct ib_device *ibdev, u32 port, |
| 150 | struct ib_port_attr *props); |
| 151 | int efa_query_qp(struct ib_qp *ibqp, struct ib_qp_attr *qp_attr, |
| 152 | int qp_attr_mask, |
| 153 | struct ib_qp_init_attr *qp_init_attr); |
| 154 | int efa_query_gid(struct ib_device *ibdev, u32 port, int index, |
| 155 | union ib_gid *gid); |
| 156 | int efa_query_pkey(struct ib_device *ibdev, u32 port, u16 index, |
| 157 | u16 *pkey); |
| 158 | int efa_alloc_pd(struct ib_pd *ibpd, struct ib_udata *udata); |
| 159 | int efa_dealloc_pd(struct ib_pd *ibpd, struct ib_udata *udata); |
| 160 | int efa_destroy_qp(struct ib_qp *ibqp, struct ib_udata *udata); |
| 161 | int efa_create_qp(struct ib_qp *ibqp, struct ib_qp_init_attr *init_attr, |
| 162 | struct ib_udata *udata); |
| 163 | int efa_destroy_cq(struct ib_cq *ibcq, struct ib_udata *udata); |
| 164 | int efa_create_cq(struct ib_cq *ibcq, const struct ib_cq_init_attr *attr, |
| 165 | struct uverbs_attr_bundle *attrs); |
| 166 | int efa_create_cq_umem(struct ib_cq *ibcq, const struct ib_cq_init_attr *attr, |
| 167 | struct ib_umem *umem, struct uverbs_attr_bundle *attrs); |
| 168 | struct ib_mr *efa_reg_mr(struct ib_pd *ibpd, u64 start, u64 length, |
| 169 | u64 virt_addr, int access_flags, |
| 170 | struct ib_dmah *dmah, |
| 171 | struct ib_udata *udata); |
| 172 | struct ib_mr *efa_reg_user_mr_dmabuf(struct ib_pd *ibpd, u64 start, |
| 173 | u64 length, u64 virt_addr, |
| 174 | int fd, int access_flags, |
| 175 | struct ib_dmah *dmah, |
| 176 | struct uverbs_attr_bundle *attrs); |
| 177 | int efa_dereg_mr(struct ib_mr *ibmr, struct ib_udata *udata); |
| 178 | int efa_get_port_immutable(struct ib_device *ibdev, u32 port_num, |
| 179 | struct ib_port_immutable *immutable); |
| 180 | int efa_alloc_ucontext(struct ib_ucontext *ibucontext, struct ib_udata *udata); |
| 181 | void efa_dealloc_ucontext(struct ib_ucontext *ibucontext); |
| 182 | int efa_mmap(struct ib_ucontext *ibucontext, |
| 183 | struct vm_area_struct *vma); |
| 184 | void efa_mmap_free(struct rdma_user_mmap_entry *rdma_entry); |
| 185 | int efa_create_ah(struct ib_ah *ibah, |
| 186 | struct rdma_ah_init_attr *init_attr, |
| 187 | struct ib_udata *udata); |
| 188 | int efa_destroy_ah(struct ib_ah *ibah, u32 flags); |
| 189 | int efa_modify_qp(struct ib_qp *ibqp, struct ib_qp_attr *qp_attr, |
| 190 | int qp_attr_mask, struct ib_udata *udata); |
| 191 | enum rdma_link_layer efa_port_link_layer(struct ib_device *ibdev, |
| 192 | u32 port_num); |
| 193 | struct rdma_hw_stats *efa_alloc_hw_port_stats(struct ib_device *ibdev, u32 port_num); |
| 194 | struct rdma_hw_stats *efa_alloc_hw_device_stats(struct ib_device *ibdev); |
| 195 | int efa_get_hw_stats(struct ib_device *ibdev, struct rdma_hw_stats *stats, |
| 196 | u32 port_num, int index); |
| 197 | |
| 198 | #endif /* _EFA_H_ */ |
| 199 | |