sw/rxe/rxe_av.c

   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
  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
  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
  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
  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
  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
  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
 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
 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 }