xref: /openbmc/linux/drivers/infiniband/sw/rxe/rxe_av.c (revision 25fd735a)
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 #include "rxe.h"
8 #include "rxe_loc.h"
9 
rxe_init_av(struct rdma_ah_attr * attr,struct rxe_av * av)10 void rxe_init_av(struct rdma_ah_attr *attr, struct rxe_av *av)
11 {
12 	rxe_av_from_attr(rdma_ah_get_port_num(attr), av, attr);
13 	rxe_av_fill_ip_info(av, attr);
14 	memcpy(av->dmac, attr->roce.dmac, ETH_ALEN);
15 }
16 
chk_attr(void * obj,struct rdma_ah_attr * attr,bool obj_is_ah)17 static int chk_attr(void *obj, struct rdma_ah_attr *attr, bool obj_is_ah)
18 {
19 	const struct ib_global_route *grh = rdma_ah_read_grh(attr);
20 	struct rxe_port *port;
21 	struct rxe_dev *rxe;
22 	struct rxe_qp *qp;
23 	struct rxe_ah *ah;
24 	int type;
25 
26 	if (obj_is_ah) {
27 		ah = obj;
28 		rxe = to_rdev(ah->ibah.device);
29 	} else {
30 		qp = obj;
31 		rxe = to_rdev(qp->ibqp.device);
32 	}
33 
34 	port = &rxe->port;
35 
36 	if (rdma_ah_get_ah_flags(attr) & IB_AH_GRH) {
37 		if (grh->sgid_index > port->attr.gid_tbl_len) {
38 			if (obj_is_ah)
39 				rxe_dbg_ah(ah, "invalid sgid index = %d\n",
40 						grh->sgid_index);
41 			else
42 				rxe_dbg_qp(qp, "invalid sgid index = %d\n",
43 						grh->sgid_index);
44 			return -EINVAL;
45 		}
46 
47 		type = rdma_gid_attr_network_type(grh->sgid_attr);
48 		if (type < RDMA_NETWORK_IPV4 ||
49 		    type > RDMA_NETWORK_IPV6) {
50 			if (obj_is_ah)
51 				rxe_dbg_ah(ah, "invalid network type for rdma_rxe = %d\n",
52 						type);
53 			else
54 				rxe_dbg_qp(qp, "invalid network type for rdma_rxe = %d\n",
55 						type);
56 			return -EINVAL;
57 		}
58 	}
59 
60 	return 0;
61 }
62 
rxe_av_chk_attr(struct rxe_qp * qp,struct rdma_ah_attr * attr)63 int rxe_av_chk_attr(struct rxe_qp *qp, struct rdma_ah_attr *attr)
64 {
65 	return chk_attr(qp, attr, false);
66 }
67 
rxe_ah_chk_attr(struct rxe_ah * ah,struct rdma_ah_attr * attr)68 int rxe_ah_chk_attr(struct rxe_ah *ah, struct rdma_ah_attr *attr)
69 {
70 	return chk_attr(ah, attr, true);
71 }
72 
rxe_av_from_attr(u8 port_num,struct rxe_av * av,struct rdma_ah_attr * attr)73 void rxe_av_from_attr(u8 port_num, struct rxe_av *av,
74 		     struct rdma_ah_attr *attr)
75 {
76 	const struct ib_global_route *grh = rdma_ah_read_grh(attr);
77 
78 	memset(av, 0, sizeof(*av));
79 	memcpy(av->grh.dgid.raw, grh->dgid.raw, sizeof(grh->dgid.raw));
80 	av->grh.flow_label = grh->flow_label;
81 	av->grh.sgid_index = grh->sgid_index;
82 	av->grh.hop_limit = grh->hop_limit;
83 	av->grh.traffic_class = grh->traffic_class;
84 	av->port_num = port_num;
85 }
86 
rxe_av_to_attr(struct rxe_av * av,struct rdma_ah_attr * attr)87 void rxe_av_to_attr(struct rxe_av *av, struct rdma_ah_attr *attr)
88 {
89 	struct ib_global_route *grh = rdma_ah_retrieve_grh(attr);
90 
91 	attr->type = RDMA_AH_ATTR_TYPE_ROCE;
92 
93 	memcpy(grh->dgid.raw, av->grh.dgid.raw, sizeof(av->grh.dgid.raw));
94 	grh->flow_label = av->grh.flow_label;
95 	grh->sgid_index = av->grh.sgid_index;
96 	grh->hop_limit = av->grh.hop_limit;
97 	grh->traffic_class = av->grh.traffic_class;
98 
99 	rdma_ah_set_ah_flags(attr, IB_AH_GRH);
100 	rdma_ah_set_port_num(attr, av->port_num);
101 }
102 
rxe_av_fill_ip_info(struct rxe_av * av,struct rdma_ah_attr * attr)103 void rxe_av_fill_ip_info(struct rxe_av *av, struct rdma_ah_attr *attr)
104 {
105 	const struct ib_gid_attr *sgid_attr = attr->grh.sgid_attr;
106 	int ibtype;
107 	int type;
108 
109 	rdma_gid2ip((struct sockaddr *)&av->sgid_addr, &sgid_attr->gid);
110 	rdma_gid2ip((struct sockaddr *)&av->dgid_addr,
111 		    &rdma_ah_read_grh(attr)->dgid);
112 
113 	ibtype = rdma_gid_attr_network_type(sgid_attr);
114 
115 	switch (ibtype) {
116 	case RDMA_NETWORK_IPV4:
117 		type = RXE_NETWORK_TYPE_IPV4;
118 		break;
119 	case RDMA_NETWORK_IPV6:
120 		type = RXE_NETWORK_TYPE_IPV6;
121 		break;
122 	default:
123 		/* not reached - checked in rxe_av_chk_attr */
124 		type = 0;
125 		break;
126 	}
127 
128 	av->network_type = type;
129 }
130 
rxe_get_av(struct rxe_pkt_info * pkt,struct rxe_ah ** ahp)131 struct rxe_av *rxe_get_av(struct rxe_pkt_info *pkt, struct rxe_ah **ahp)
132 {
133 	struct rxe_ah *ah;
134 	u32 ah_num;
135 
136 	if (ahp)
137 		*ahp = NULL;
138 
139 	if (!pkt || !pkt->qp)
140 		return NULL;
141 
142 	if (qp_type(pkt->qp) == IB_QPT_RC || qp_type(pkt->qp) == IB_QPT_UC)
143 		return &pkt->qp->pri_av;
144 
145 	if (!pkt->wqe)
146 		return NULL;
147 
148 	ah_num = pkt->wqe->wr.wr.ud.ah_num;
149 	if (ah_num) {
150 		/* only new user provider or kernel client */
151 		ah = rxe_pool_get_index(&pkt->rxe->ah_pool, ah_num);
152 		if (!ah) {
153 			rxe_dbg_qp(pkt->qp, "Unable to find AH matching ah_num\n");
154 			return NULL;
155 		}
156 
157 		if (rxe_ah_pd(ah) != pkt->qp->pd) {
158 			rxe_dbg_qp(pkt->qp, "PDs don't match for AH and QP\n");
159 			rxe_put(ah);
160 			return NULL;
161 		}
162 
163 		if (ahp)
164 			*ahp = ah;
165 		else
166 			rxe_put(ah);
167 
168 		return &ah->av;
169 	}
170 
171 	/* only old user provider for UD sends*/
172 	return &pkt->wqe->wr.wr.ud.av;
173 }
174