This patch was for implemention of address handle operation.
It includes three verbs that create ah, query ah and destory
ah. They is completed independently by RoCE driver.

Signed-off-by: Lijun Ou <ouli...@huawei.com>
Signed-off-by: Wei Hu(Xavier) <xavier.hu...@huawei.com>
---
 drivers/infiniband/hw/hns/hns_roce_ah.c     | 109 ++++++++++++++++++++++++++++
 drivers/infiniband/hw/hns/hns_roce_common.h |   4 +
 drivers/infiniband/hw/hns/hns_roce_device.h |  31 +++++++-
 drivers/infiniband/hw/hns/hns_roce_main.c   |   5 ++
 4 files changed, 148 insertions(+), 1 deletion(-)
 create mode 100644 drivers/infiniband/hw/hns/hns_roce_ah.c

diff --git a/drivers/infiniband/hw/hns/hns_roce_ah.c 
b/drivers/infiniband/hw/hns/hns_roce_ah.c
new file mode 100644
index 0000000..9d0eb61
--- /dev/null
+++ b/drivers/infiniband/hw/hns/hns_roce_ah.c
@@ -0,0 +1,109 @@
+/*
+ * Copyright (c) 2016 Hisilicon Limited.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ */
+
+#include <linux/inet.h>
+#include <linux/init.h>
+#include <linux/platform_device.h>
+#include <linux/slab.h>
+#include <rdma/ib_addr.h>
+#include <rdma/ib_cache.h>
+#include "hns_roce_common.h"
+#include "hns_roce_device.h"
+
+#define HNS_ROCE_PORT_NUM_SHIFT                24
+#define HNS_ROCE_VLAN_SL_BIT_MASK      7
+#define HNS_ROCE_VLAN_SL_SHIFT         13
+
+struct ib_ah *hns_roce_create_ah(struct ib_pd *ibpd, struct ib_ah_attr 
*ah_attr)
+{
+       struct hns_roce_dev *hr_dev = to_hr_dev(ibpd->device);
+       struct device *dev = &hr_dev->pdev->dev;
+       struct ib_gid_attr gid_attr;
+       struct hns_roce_ah *ah;
+       u16 vlan_tag = 0xffff;
+       struct in6_addr in6;
+       union ib_gid sgid;
+       int ret;
+
+       ah = kzalloc(sizeof(*ah), GFP_ATOMIC);
+       if (!ah)
+               return ERR_PTR(-ENOMEM);
+
+       /* Get mac address */
+       memcpy(&in6, ah_attr->grh.dgid.raw, sizeof(ah_attr->grh.dgid.raw));
+       if (rdma_is_multicast_addr(&in6))
+               rdma_get_mcast_mac(&in6, ah->av.mac);
+       else
+               memcpy(ah->av.mac, ah_attr->dmac, sizeof(ah_attr->dmac));
+
+       /* Get source gid */
+       ret = ib_get_cached_gid(ibpd->device, ah_attr->port_num,
+                               ah_attr->grh.sgid_index, &sgid, &gid_attr);
+       if (ret) {
+               dev_err(dev, "get sgid failed! ret = %d\n", ret);
+               kfree(ah);
+               return ERR_PTR(ret);
+       }
+
+       if (gid_attr.ndev) {
+               if (is_vlan_dev(gid_attr.ndev))
+                       vlan_tag = vlan_dev_vlan_id(gid_attr.ndev);
+               dev_put(gid_attr.ndev);
+       }
+
+       if (vlan_tag < 0x1000)
+               vlan_tag |= (ah_attr->sl & HNS_ROCE_VLAN_SL_BIT_MASK) <<
+                            HNS_ROCE_VLAN_SL_SHIFT;
+
+       ah->av.port_pd = cpu_to_be32(to_hr_pd(ibpd)->pdn | (ah_attr->port_num <<
+                                    HNS_ROCE_PORT_NUM_SHIFT));
+       ah->av.gid_index = ah_attr->grh.sgid_index;
+       ah->av.vlan = cpu_to_le16(vlan_tag);
+       dev_dbg(dev, "gid_index = 0x%x,vlan = 0x%x\n", ah->av.gid_index,
+               ah->av.vlan);
+
+       if (ah_attr->static_rate)
+               ah->av.stat_rate = IB_RATE_10_GBPS;
+
+       memcpy(ah->av.dgid, ah_attr->grh.dgid.raw, HNS_ROCE_GID_SIZE);
+       ah->av.sl_tclass_flowlabel = cpu_to_le32(ah_attr->sl <<
+                                                HNS_ROCE_SL_SHIFT);
+
+       return &ah->ibah;
+}
+
+int hns_roce_query_ah(struct ib_ah *ibah, struct ib_ah_attr *ah_attr)
+{
+       struct hns_roce_ah *ah = to_hr_ah(ibah);
+
+       memset(ah_attr, 0, sizeof(*ah_attr));
+
+       ah_attr->sl = le32_to_cpu(ah->av.sl_tclass_flowlabel) >>
+                                 HNS_ROCE_SL_SHIFT;
+       ah_attr->port_num = le32_to_cpu(ah->av.port_pd) >>
+                                       HNS_ROCE_PORT_NUM_SHIFT;
+       ah_attr->static_rate = ah->av.stat_rate;
+       ah_attr->ah_flags = IB_AH_GRH;
+       ah_attr->grh.traffic_class = le32_to_cpu(ah->av.sl_tclass_flowlabel) >>
+                                                HNS_ROCE_TCLASS_SHIFT;
+       ah_attr->grh.flow_label = le32_to_cpu(ah->av.sl_tclass_flowlabel) &
+                                             HNS_ROCE_FLOW_LABLE_MASK;
+       ah_attr->grh.hop_limit = ah->av.hop_limit;
+       ah_attr->grh.sgid_index = ah->av.gid_index;
+       memcpy(ah_attr->grh.dgid.raw, ah->av.dgid, HNS_ROCE_GID_SIZE);
+
+       return 0;
+}
+
+int hns_roce_destroy_ah(struct ib_ah *ah)
+{
+       kfree(to_hr_ah(ah));
+
+       return 0;
+}
diff --git a/drivers/infiniband/hw/hns/hns_roce_common.h 
b/drivers/infiniband/hw/hns/hns_roce_common.h
index 9283f05..db32a52 100644
--- a/drivers/infiniband/hw/hns/hns_roce_common.h
+++ b/drivers/infiniband/hw/hns/hns_roce_common.h
@@ -10,6 +10,10 @@
 #ifndef _HNS_ROCE_COMMON_H
 #define _HNS_ROCE_COMMON_H
 
+#ifndef assert
+#define assert(cond)
+#endif
+
 #define roce_writel(value, addr)     writel((value), (addr))
 #define roce_readl(addr)            readl((addr))
 #define roce_raw_write(value, addr) \
diff --git a/drivers/infiniband/hw/hns/hns_roce_device.h 
b/drivers/infiniband/hw/hns/hns_roce_device.h
index 68d78e9..f74786b 100644
--- a/drivers/infiniband/hw/hns/hns_roce_device.h
+++ b/drivers/infiniband/hw/hns/hns_roce_device.h
@@ -24,7 +24,6 @@
 #define MAC_ADDR_OCTET_NUM                     6
 #define HNS_ROCE_MAX_MSG_LEN                   0x80000000
 
-
 #define HNS_ROCE_BA_SIZE                       (32 * 4096)
 
 #define HNS_ROCE_MAX_IRQ_NUM                   34
@@ -35,6 +34,11 @@
 #define HNS_ROCE_AEQE_VEC_NUM                  1
 #define HNS_ROCE_AEQE_OF_VEC_NUM               1
 
+/* 4G/4K = 1M */
+#define HNS_ROCE_SL_SHIFT                      29
+#define HNS_ROCE_TCLASS_SHIFT                  20
+#define HNS_ROCE_FLOW_LABLE_MASK               0xfffff
+
 #define HNS_ROCE_MAX_PORTS                     6
 #define HNS_ROCE_MAX_GID_NUM                   16
 #define HNS_ROCE_GID_SIZE                      16
@@ -208,6 +212,22 @@ struct hns_roce_raq_table {
        struct hns_roce_buf_list        *e_raq_buf;
 };
 
+struct hns_roce_av {
+       __le32      port_pd;
+       u8          gid_index;
+       u8          stat_rate;
+       u8          hop_limit;
+       __le32      sl_tclass_flowlabel;
+       u8          dgid[HNS_ROCE_GID_SIZE];
+       u8          mac[6];
+       __le16      vlan;
+};
+
+struct hns_roce_ah {
+       struct ib_ah            ibah;
+       struct hns_roce_av      av;
+};
+
 struct hns_roce_cmd_context {
        struct completion       done;
        int                     result;
@@ -394,6 +414,11 @@ static inline struct hns_roce_pd *to_hr_pd(struct ib_pd 
*ibpd)
        return container_of(ibpd, struct hns_roce_pd, ibpd);
 }
 
+static inline struct hns_roce_ah *to_hr_ah(struct ib_ah *ibah)
+{
+       return container_of(ibah, struct hns_roce_ah, ibah);
+}
+
 static inline void hns_roce_write64_k(__be32 val[2], void __iomem *dest)
 {
        __raw_writeq(*(u64 *) val, dest);
@@ -441,6 +466,10 @@ int hns_roce_bitmap_alloc_range(struct hns_roce_bitmap 
*bitmap, int cnt,
 void hns_roce_bitmap_free_range(struct hns_roce_bitmap *bitmap, u32 obj,
                                int cnt);
 
+struct ib_ah *hns_roce_create_ah(struct ib_pd *pd, struct ib_ah_attr *ah_attr);
+int hns_roce_query_ah(struct ib_ah *ibah, struct ib_ah_attr *ah_attr);
+int hns_roce_destroy_ah(struct ib_ah *ah);
+
 struct ib_pd *hns_roce_alloc_pd(struct ib_device *ib_dev,
                                struct ib_ucontext *context,
                                struct ib_udata *udata);
diff --git a/drivers/infiniband/hw/hns/hns_roce_main.c 
b/drivers/infiniband/hw/hns/hns_roce_main.c
index eeae944..b9b88f7 100644
--- a/drivers/infiniband/hw/hns/hns_roce_main.c
+++ b/drivers/infiniband/hw/hns/hns_roce_main.c
@@ -605,6 +605,11 @@ int hns_roce_register_device(struct hns_roce_dev *hr_dev)
        ib_dev->alloc_pd                = hns_roce_alloc_pd;
        ib_dev->dealloc_pd              = hns_roce_dealloc_pd;
 
+       /* AH */
+       ib_dev->create_ah               = hns_roce_create_ah;
+       ib_dev->query_ah                = hns_roce_query_ah;
+       ib_dev->destroy_ah              = hns_roce_destroy_ah;
+
        ret = ib_register_device(ib_dev, NULL);
        if (ret) {
                dev_err(dev, "ib_register_device failed!\n");
-- 
1.9.1

Reply via email to