*
  */
 
+#include <linux/slab.h>
 #include "ah.h"
+#include "vt.h" /* for prints */
+
+/**
+ * rvt_check_ah - validate the attributes of AH
+ * @ibdev: the ib device
+ * @ah_attr: the attributes of the AH
+ */
+int rvt_check_ah(struct ib_device *ibdev,
+                struct ib_ah_attr *ah_attr)
+{
+       int err;
+       struct ib_port_attr port_attr;
+       struct rvt_dev_info *rdi = ib_to_rvt(ibdev);
+       enum rdma_link_layer link = rdma_port_get_link_layer(ibdev,
+                                                            ah_attr->port_num);
+
+       err = ib_query_port(ibdev, ah_attr->port_num, &port_attr);
+       if (err)
+               return -EINVAL;
+       if (ah_attr->port_num < 1 ||
+           ah_attr->port_num > ibdev->phys_port_cnt)
+               return -EINVAL;
+       if (ah_attr->static_rate != IB_RATE_PORT_CURRENT &&
+           ib_rate_to_mbps(ah_attr->static_rate) < 0)
+               return -EINVAL;
+       if ((ah_attr->ah_flags & IB_AH_GRH) &&
+           ah_attr->grh.sgid_index >= port_attr.gid_tbl_len)
+               return -EINVAL;
+       if (link != IB_LINK_LAYER_ETHERNET) {
+               if (ah_attr->dlid == 0)
+                       return -EINVAL;
+               if (ah_attr->dlid >= RVT_MULTICAST_LID_BASE &&
+                   ah_attr->dlid != RVT_PERMISSIVE_LID &&
+                   !(ah_attr->ah_flags & IB_AH_GRH))
+                       return -EINVAL;
+       }
+       if (rdi->driver_f.check_ah(ibdev, ah_attr))
+               return -EINVAL;
+       return 0;
+}
+EXPORT_SYMBOL(rvt_check_ah);
 
 /**
  * rvt_create_ah - create an address handle
 struct ib_ah *rvt_create_ah(struct ib_pd *pd,
                            struct ib_ah_attr *ah_attr)
 {
-       return ERR_PTR(-EOPNOTSUPP);
+       struct rvt_ah *ah;
+       struct rvt_dev_info *dev = ib_to_rvt(pd->device);
+       unsigned long flags;
+
+       if (rvt_check_ah(pd->device, ah_attr))
+               return ERR_PTR(-EINVAL);
+
+       ah = kmalloc(sizeof(*ah), GFP_ATOMIC);
+       if (!ah)
+               return ERR_PTR(-ENOMEM);
+
+       spin_lock_irqsave(&dev->n_ahs_lock, flags);
+       if (dev->n_ahs_allocated == dev->dparms.props.max_ah) {
+               spin_unlock(&dev->n_ahs_lock);
+               kfree(ah);
+               return ERR_PTR(-ENOMEM);
+       }
+
+       dev->n_ahs_allocated++;
+       spin_unlock_irqrestore(&dev->n_ahs_lock, flags);
+
+       ah->attr = *ah_attr;
+       atomic_set(&ah->refcount, 0);
+
+       return &ah->ibah;
 }
 
 int rvt_destroy_ah(struct ib_ah *ibah)
 {
-       return -EOPNOTSUPP;
+       struct rvt_dev_info *dev = ib_to_rvt(ibah->device);
+       struct rvt_ah *ah = ibah_to_rvtah(ibah);
+       unsigned long flags;
+
+       if (atomic_read(&ah->refcount) != 0)
+               return -EBUSY;
+
+       spin_lock_irqsave(&dev->n_ahs_lock, flags);
+       dev->n_ahs_allocated--;
+       spin_unlock_irqrestore(&dev->n_ahs_lock, flags);
+
+       kfree(ah);
+
+       return 0;
 }
 
 int rvt_modify_ah(struct ib_ah *ibah, struct ib_ah_attr *ah_attr)
 {
-       return -EOPNOTSUPP;
+       struct rvt_ah *ah = ibah_to_rvtah(ibah);
+
+       if (rvt_check_ah(ibah->device, ah_attr))
+               return -EINVAL;
+
+       ah->attr = *ah_attr;
+
+       return 0;
 }
 
 int rvt_query_ah(struct ib_ah *ibah, struct ib_ah_attr *ah_attr)
 {
-       return -EOPNOTSUPP;
+       struct rvt_ah *ah = ibah_to_rvtah(ibah);
+
+       *ah_attr = ah->attr;
+
+       return 0;
 }
 
 
        if ((!rdi->driver_f.port_callback) ||
            (!rdi->driver_f.get_card_name) ||
-           (!rdi->driver_f.get_pci_dev)) {
+           (!rdi->driver_f.get_pci_dev) ||
+           (!rdi->driver_f.check_ah)) {
                return -EINVAL;
        }
 
        CHECK_DRIVER_OVERRIDE(rdi, destroy_ah);
        CHECK_DRIVER_OVERRIDE(rdi, modify_ah);
        CHECK_DRIVER_OVERRIDE(rdi, query_ah);
+       spin_lock_init(&rdi->n_ahs_lock);
+       rdi->n_ahs_allocated = 0;
 
        /* Shared Receive Queue */
        CHECK_DRIVER_OVERRIDE(rdi, create_srq);
 
        int (*port_callback)(struct ib_device *, u8, struct kobject *);
        const char * (*get_card_name)(struct rvt_dev_info *rdi);
        struct pci_dev * (*get_pci_dev)(struct rvt_dev_info *rdi);
+       int (*check_ah)(struct ib_device *, struct ib_ah_attr *);
 };
 
 /* Protection domain */
        int user;               /* non-zero if created from user space */
 };
 
+/* Address handle */
+struct rvt_ah {
+       struct ib_ah ibah;
+       struct ib_ah_attr attr;
+       atomic_t refcount;
+};
+
 struct rvt_dev_info {
        struct ib_device ibdev; /* Keep this first. Nothing above here */
 
        int n_pds_allocated;
        spinlock_t n_pds_lock; /* Protect pd allocated count */
 
+       int n_ahs_allocated;
+       spinlock_t n_ahs_lock; /* Protect ah allocated count */
+
        int flags;
 };
 
        return container_of(ibpd, struct rvt_pd, ibpd);
 }
 
+static inline struct rvt_ah *ibah_to_rvtah(struct ib_ah *ibah)
+{
+       return container_of(ibah, struct rvt_ah, ibah);
+}
+
 static inline struct rvt_dev_info *ib_to_rvt(struct ib_device *ibdev)
 {
        return  container_of(ibdev, struct rvt_dev_info, ibdev);
 
 int rvt_register_device(struct rvt_dev_info *rvd);
 void rvt_unregister_device(struct rvt_dev_info *rvd);
+int rvt_check_ah(struct ib_device *ibdev, struct ib_ah_attr *ah_attr);
 int rvt_rkey_ok(struct rvt_qp *qp, struct rvt_sge *sge,
                u32 len, u64 vaddr, u32 rkey, int acc);
 int rvt_lkey_ok(struct rvt_lkey_table *rkt, struct rvt_pd *pd,