174 lines · c
1// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB2/*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 10void 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 17static 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 else42 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 else54 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 63int rxe_av_chk_attr(struct rxe_qp *qp, struct rdma_ah_attr *attr)64{65 return chk_attr(qp, attr, false);66}67 68int rxe_ah_chk_attr(struct rxe_ah *ah, struct rdma_ah_attr *attr)69{70 return chk_attr(ah, attr, true);71}72 73void 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 87void 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 103void 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 131struct 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 else166 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