1 /* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
2 /*
3  * Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved.
4  * Copyright (c) 2015 System Fabric Works, Inc. All rights reserved.
5  */
6 
7 #ifndef RXE_LOC_H
8 #define RXE_LOC_H
9 
10 /* rxe_av.c */
11 void rxe_init_av(struct rdma_ah_attr *attr, struct rxe_av *av);
12 
13 int rxe_av_chk_attr(struct rxe_dev *rxe, struct rdma_ah_attr *attr);
14 
15 void rxe_av_from_attr(u8 port_num, struct rxe_av *av,
16 		     struct rdma_ah_attr *attr);
17 
18 void rxe_av_to_attr(struct rxe_av *av, struct rdma_ah_attr *attr);
19 
20 void rxe_av_fill_ip_info(struct rxe_av *av, struct rdma_ah_attr *attr);
21 
22 struct rxe_av *rxe_get_av(struct rxe_pkt_info *pkt);
23 
24 /* rxe_cq.c */
25 int rxe_cq_chk_attr(struct rxe_dev *rxe, struct rxe_cq *cq,
26 		    int cqe, int comp_vector);
27 
28 int rxe_cq_from_init(struct rxe_dev *rxe, struct rxe_cq *cq, int cqe,
29 		     int comp_vector, struct ib_udata *udata,
30 		     struct rxe_create_cq_resp __user *uresp);
31 
32 int rxe_cq_resize_queue(struct rxe_cq *cq, int new_cqe,
33 			struct rxe_resize_cq_resp __user *uresp,
34 			struct ib_udata *udata);
35 
36 int rxe_cq_post(struct rxe_cq *cq, struct rxe_cqe *cqe, int solicited);
37 
38 void rxe_cq_disable(struct rxe_cq *cq);
39 
40 void rxe_cq_cleanup(struct rxe_pool_entry *arg);
41 
42 /* rxe_mcast.c */
43 int rxe_mcast_get_grp(struct rxe_dev *rxe, union ib_gid *mgid,
44 		      struct rxe_mc_grp **grp_p);
45 
46 int rxe_mcast_add_grp_elem(struct rxe_dev *rxe, struct rxe_qp *qp,
47 			   struct rxe_mc_grp *grp);
48 
49 int rxe_mcast_drop_grp_elem(struct rxe_dev *rxe, struct rxe_qp *qp,
50 			    union ib_gid *mgid);
51 
52 void rxe_drop_all_mcast_groups(struct rxe_qp *qp);
53 
54 void rxe_mc_cleanup(struct rxe_pool_entry *arg);
55 
56 /* rxe_mmap.c */
57 struct rxe_mmap_info {
58 	struct list_head	pending_mmaps;
59 	struct ib_ucontext	*context;
60 	struct kref		ref;
61 	void			*obj;
62 
63 	struct mminfo info;
64 };
65 
66 void rxe_mmap_release(struct kref *ref);
67 
68 struct rxe_mmap_info *rxe_create_mmap_info(struct rxe_dev *dev, u32 size,
69 					   struct ib_udata *udata, void *obj);
70 
71 int rxe_mmap(struct ib_ucontext *context, struct vm_area_struct *vma);
72 
73 /* rxe_mr.c */
74 enum copy_direction {
75 	to_mr_obj,
76 	from_mr_obj,
77 };
78 
79 void rxe_mr_init_dma(struct rxe_pd *pd, int access, struct rxe_mr *mr);
80 
81 int rxe_mr_init_user(struct rxe_pd *pd, u64 start, u64 length, u64 iova,
82 		     int access, struct ib_udata *udata, struct rxe_mr *mr);
83 
84 int rxe_mr_init_fast(struct rxe_pd *pd, int max_pages, struct rxe_mr *mr);
85 
86 int rxe_mr_copy(struct rxe_mr *mr, u64 iova, void *addr, int length,
87 		enum copy_direction dir, u32 *crcp);
88 
89 int copy_data(struct rxe_pd *pd, int access,
90 	      struct rxe_dma_info *dma, void *addr, int length,
91 	      enum copy_direction dir, u32 *crcp);
92 
93 void *iova_to_vaddr(struct rxe_mr *mr, u64 iova, int length);
94 
95 enum lookup_type {
96 	lookup_local,
97 	lookup_remote,
98 };
99 
100 struct rxe_mr *lookup_mr(struct rxe_pd *pd, int access, u32 key,
101 			 enum lookup_type type);
102 
103 int mr_check_range(struct rxe_mr *mr, u64 iova, size_t length);
104 
105 void rxe_mr_cleanup(struct rxe_pool_entry *arg);
106 
107 int advance_dma_data(struct rxe_dma_info *dma, unsigned int length);
108 
109 /* rxe_net.c */
110 void rxe_loopback(struct sk_buff *skb);
111 int rxe_send(struct rxe_pkt_info *pkt, struct sk_buff *skb);
112 struct sk_buff *rxe_init_packet(struct rxe_dev *rxe, struct rxe_av *av,
113 				int paylen, struct rxe_pkt_info *pkt);
114 int rxe_prepare(struct rxe_pkt_info *pkt, struct sk_buff *skb, u32 *crc);
115 const char *rxe_parent_name(struct rxe_dev *rxe, unsigned int port_num);
116 int rxe_mcast_add(struct rxe_dev *rxe, union ib_gid *mgid);
117 int rxe_mcast_delete(struct rxe_dev *rxe, union ib_gid *mgid);
118 
119 /* rxe_qp.c */
120 int rxe_qp_chk_init(struct rxe_dev *rxe, struct ib_qp_init_attr *init);
121 
122 int rxe_qp_from_init(struct rxe_dev *rxe, struct rxe_qp *qp, struct rxe_pd *pd,
123 		     struct ib_qp_init_attr *init,
124 		     struct rxe_create_qp_resp __user *uresp,
125 		     struct ib_pd *ibpd, struct ib_udata *udata);
126 
127 int rxe_qp_to_init(struct rxe_qp *qp, struct ib_qp_init_attr *init);
128 
129 int rxe_qp_chk_attr(struct rxe_dev *rxe, struct rxe_qp *qp,
130 		    struct ib_qp_attr *attr, int mask);
131 
132 int rxe_qp_from_attr(struct rxe_qp *qp, struct ib_qp_attr *attr,
133 		     int mask, struct ib_udata *udata);
134 
135 int rxe_qp_to_attr(struct rxe_qp *qp, struct ib_qp_attr *attr, int mask);
136 
137 void rxe_qp_error(struct rxe_qp *qp);
138 
139 void rxe_qp_destroy(struct rxe_qp *qp);
140 
141 void rxe_qp_cleanup(struct rxe_pool_entry *arg);
142 
143 static inline int qp_num(struct rxe_qp *qp)
144 {
145 	return qp->ibqp.qp_num;
146 }
147 
148 static inline enum ib_qp_type qp_type(struct rxe_qp *qp)
149 {
150 	return qp->ibqp.qp_type;
151 }
152 
153 static inline enum ib_qp_state qp_state(struct rxe_qp *qp)
154 {
155 	return qp->attr.qp_state;
156 }
157 
158 static inline int qp_mtu(struct rxe_qp *qp)
159 {
160 	if (qp->ibqp.qp_type == IB_QPT_RC || qp->ibqp.qp_type == IB_QPT_UC)
161 		return qp->attr.path_mtu;
162 	else
163 		return IB_MTU_4096;
164 }
165 
166 static inline int rcv_wqe_size(int max_sge)
167 {
168 	return sizeof(struct rxe_recv_wqe) +
169 		max_sge * sizeof(struct ib_sge);
170 }
171 
172 void free_rd_atomic_resource(struct rxe_qp *qp, struct resp_res *res);
173 
174 static inline void rxe_advance_resp_resource(struct rxe_qp *qp)
175 {
176 	qp->resp.res_head++;
177 	if (unlikely(qp->resp.res_head == qp->attr.max_dest_rd_atomic))
178 		qp->resp.res_head = 0;
179 }
180 
181 void retransmit_timer(struct timer_list *t);
182 void rnr_nak_timer(struct timer_list *t);
183 
184 /* rxe_srq.c */
185 #define IB_SRQ_INIT_MASK (~IB_SRQ_LIMIT)
186 
187 int rxe_srq_chk_attr(struct rxe_dev *rxe, struct rxe_srq *srq,
188 		     struct ib_srq_attr *attr, enum ib_srq_attr_mask mask);
189 
190 int rxe_srq_from_init(struct rxe_dev *rxe, struct rxe_srq *srq,
191 		      struct ib_srq_init_attr *init, struct ib_udata *udata,
192 		      struct rxe_create_srq_resp __user *uresp);
193 
194 int rxe_srq_from_attr(struct rxe_dev *rxe, struct rxe_srq *srq,
195 		      struct ib_srq_attr *attr, enum ib_srq_attr_mask mask,
196 		      struct rxe_modify_srq_cmd *ucmd, struct ib_udata *udata);
197 
198 void rxe_dealloc(struct ib_device *ib_dev);
199 
200 int rxe_completer(void *arg);
201 int rxe_requester(void *arg);
202 int rxe_responder(void *arg);
203 
204 u32 rxe_icrc_hdr(struct rxe_pkt_info *pkt, struct sk_buff *skb);
205 
206 void rxe_resp_queue_pkt(struct rxe_qp *qp, struct sk_buff *skb);
207 
208 void rxe_comp_queue_pkt(struct rxe_qp *qp, struct sk_buff *skb);
209 
210 static inline unsigned int wr_opcode_mask(int opcode, struct rxe_qp *qp)
211 {
212 	return rxe_wr_opcode_info[opcode].mask[qp->ibqp.qp_type];
213 }
214 
215 static inline int rxe_xmit_packet(struct rxe_qp *qp, struct rxe_pkt_info *pkt,
216 				  struct sk_buff *skb)
217 {
218 	int err;
219 	int is_request = pkt->mask & RXE_REQ_MASK;
220 	struct rxe_dev *rxe = to_rdev(qp->ibqp.device);
221 
222 	if ((is_request && (qp->req.state != QP_STATE_READY)) ||
223 	    (!is_request && (qp->resp.state != QP_STATE_READY))) {
224 		pr_info("Packet dropped. QP is not in ready state\n");
225 		goto drop;
226 	}
227 
228 	if (pkt->mask & RXE_LOOPBACK_MASK) {
229 		memcpy(SKB_TO_PKT(skb), pkt, sizeof(*pkt));
230 		rxe_loopback(skb);
231 		err = 0;
232 	} else {
233 		err = rxe_send(pkt, skb);
234 	}
235 
236 	if (err) {
237 		rxe->xmit_errors++;
238 		rxe_counter_inc(rxe, RXE_CNT_SEND_ERR);
239 		return err;
240 	}
241 
242 	if ((qp_type(qp) != IB_QPT_RC) &&
243 	    (pkt->mask & RXE_END_MASK)) {
244 		pkt->wqe->state = wqe_state_done;
245 		rxe_run_task(&qp->comp.task, 1);
246 	}
247 
248 	rxe_counter_inc(rxe, RXE_CNT_SENT_PKTS);
249 	goto done;
250 
251 drop:
252 	kfree_skb(skb);
253 	err = 0;
254 done:
255 	return err;
256 }
257 
258 #endif /* RXE_LOC_H */
259