1*df91c470SKonstantin Taranov // SPDX-License-Identifier: GPL-2.0-only 2*df91c470SKonstantin Taranov /* 3*df91c470SKonstantin Taranov * Copyright (c) 2024, Microsoft Corporation. All rights reserved. 4*df91c470SKonstantin Taranov */ 5*df91c470SKonstantin Taranov 6*df91c470SKonstantin Taranov #include "mana_ib.h" 7*df91c470SKonstantin Taranov 8*df91c470SKonstantin Taranov int mana_ib_create_ah(struct ib_ah *ibah, struct rdma_ah_init_attr *attr, 9*df91c470SKonstantin Taranov struct ib_udata *udata) 10*df91c470SKonstantin Taranov { 11*df91c470SKonstantin Taranov struct mana_ib_dev *mdev = container_of(ibah->device, struct mana_ib_dev, ib_dev); 12*df91c470SKonstantin Taranov struct mana_ib_ah *ah = container_of(ibah, struct mana_ib_ah, ibah); 13*df91c470SKonstantin Taranov struct rdma_ah_attr *ah_attr = attr->ah_attr; 14*df91c470SKonstantin Taranov const struct ib_global_route *grh; 15*df91c470SKonstantin Taranov enum rdma_network_type ntype; 16*df91c470SKonstantin Taranov 17*df91c470SKonstantin Taranov if (ah_attr->type != RDMA_AH_ATTR_TYPE_ROCE || 18*df91c470SKonstantin Taranov !(rdma_ah_get_ah_flags(ah_attr) & IB_AH_GRH)) 19*df91c470SKonstantin Taranov return -EINVAL; 20*df91c470SKonstantin Taranov 21*df91c470SKonstantin Taranov if (udata) 22*df91c470SKonstantin Taranov return -EINVAL; 23*df91c470SKonstantin Taranov 24*df91c470SKonstantin Taranov ah->av = dma_pool_zalloc(mdev->av_pool, GFP_ATOMIC, &ah->dma_handle); 25*df91c470SKonstantin Taranov if (!ah->av) 26*df91c470SKonstantin Taranov return -ENOMEM; 27*df91c470SKonstantin Taranov 28*df91c470SKonstantin Taranov grh = rdma_ah_read_grh(ah_attr); 29*df91c470SKonstantin Taranov ntype = rdma_gid_attr_network_type(grh->sgid_attr); 30*df91c470SKonstantin Taranov 31*df91c470SKonstantin Taranov copy_in_reverse(ah->av->dest_mac, ah_attr->roce.dmac, ETH_ALEN); 32*df91c470SKonstantin Taranov ah->av->udp_src_port = rdma_flow_label_to_udp_sport(grh->flow_label); 33*df91c470SKonstantin Taranov ah->av->hop_limit = grh->hop_limit; 34*df91c470SKonstantin Taranov ah->av->dscp = (grh->traffic_class >> 2) & 0x3f; 35*df91c470SKonstantin Taranov ah->av->is_ipv6 = (ntype == RDMA_NETWORK_IPV6); 36*df91c470SKonstantin Taranov 37*df91c470SKonstantin Taranov if (ah->av->is_ipv6) { 38*df91c470SKonstantin Taranov copy_in_reverse(ah->av->dest_ip, grh->dgid.raw, 16); 39*df91c470SKonstantin Taranov copy_in_reverse(ah->av->src_ip, grh->sgid_attr->gid.raw, 16); 40*df91c470SKonstantin Taranov } else { 41*df91c470SKonstantin Taranov ah->av->dest_ip[10] = 0xFF; 42*df91c470SKonstantin Taranov ah->av->dest_ip[11] = 0xFF; 43*df91c470SKonstantin Taranov copy_in_reverse(&ah->av->dest_ip[12], &grh->dgid.raw[12], 4); 44*df91c470SKonstantin Taranov copy_in_reverse(&ah->av->src_ip[12], &grh->sgid_attr->gid.raw[12], 4); 45*df91c470SKonstantin Taranov } 46*df91c470SKonstantin Taranov 47*df91c470SKonstantin Taranov return 0; 48*df91c470SKonstantin Taranov } 49*df91c470SKonstantin Taranov 50*df91c470SKonstantin Taranov int mana_ib_destroy_ah(struct ib_ah *ibah, u32 flags) 51*df91c470SKonstantin Taranov { 52*df91c470SKonstantin Taranov struct mana_ib_dev *mdev = container_of(ibah->device, struct mana_ib_dev, ib_dev); 53*df91c470SKonstantin Taranov struct mana_ib_ah *ah = container_of(ibah, struct mana_ib_ah, ibah); 54*df91c470SKonstantin Taranov 55*df91c470SKonstantin Taranov dma_pool_free(mdev->av_pool, ah->av, ah->dma_handle); 56*df91c470SKonstantin Taranov 57*df91c470SKonstantin Taranov return 0; 58*df91c470SKonstantin Taranov } 59