Here's an updated version that defines agent_send_response as returning an int.

Signed-off-by: Sean Hefty <[EMAIL PROTECTED]>


Index: agent.c
===================================================================
--- agent.c     (revision 3794)
+++ agent.c     (working copy)
@@ -36,58 +36,41 @@
  *
  * $Id$
  */
+#include "agent.h"
+#include "smi.h"
 
-#include <linux/dma-mapping.h>
-#include <asm/bug.h>
-
-#include <rdma/ib_smi.h>
+#define SPFX "ib_agent: "
 
-#include "smi.h"
-#include "agent_priv.h"
-#include "mad_priv.h"
-#include "agent.h"
+struct ib_agent_port_private {
+       struct list_head port_list;
+       struct ib_mad_agent *agent[2];
+};
 
-spinlock_t ib_agent_port_list_lock;
+static DEFINE_SPINLOCK(ib_agent_port_list_lock);
 static LIST_HEAD(ib_agent_port_list);
 
-/*
- * Caller must hold ib_agent_port_list_lock
- */
-static inline struct ib_agent_port_private *
-__ib_get_agent_port(struct ib_device *device, int port_num,
-                   struct ib_mad_agent *mad_agent)
+static struct ib_agent_port_private *
+__ib_get_agent_port(struct ib_device *device, int port_num)
 {
        struct ib_agent_port_private *entry;
 
-       BUG_ON(!(!!device ^ !!mad_agent));  /* Exactly one MUST be (!NULL) */
-
-       if (device) {
-               list_for_each_entry(entry, &ib_agent_port_list, port_list) {
-                       if (entry->smp_agent->device == device &&
-                           entry->port_num == port_num)
-                               return entry;
-               }
-       } else {
-               list_for_each_entry(entry, &ib_agent_port_list, port_list) {
-                       if ((entry->smp_agent == mad_agent) ||
-                           (entry->perf_mgmt_agent == mad_agent))
-                               return entry;
-               }
+       list_for_each_entry(entry, &ib_agent_port_list, port_list) {
+               if (entry->agent[0]->device == device &&
+                   entry->agent[0]->port_num == port_num)
+                       return entry;
        }
        return NULL;
 }
 
-static inline struct ib_agent_port_private *
-ib_get_agent_port(struct ib_device *device, int port_num,
-                 struct ib_mad_agent *mad_agent)
+static struct ib_agent_port_private *
+ib_get_agent_port(struct ib_device *device, int port_num)
 {
        struct ib_agent_port_private *entry;
        unsigned long flags;
 
        spin_lock_irqsave(&ib_agent_port_list_lock, flags);
-       entry = __ib_get_agent_port(device, port_num, mad_agent);
+       entry = __ib_get_agent_port(device, port_num);
        spin_unlock_irqrestore(&ib_agent_port_list_lock, flags);
-
        return entry;
 }
 
@@ -99,192 +82,71 @@ int smi_check_local_dr_smp(struct ib_smp
 
        if (smp->mgmt_class != IB_MGMT_CLASS_SUBN_DIRECTED_ROUTE)
                return 1;
-       port_priv = ib_get_agent_port(device, port_num, NULL);
+
+       port_priv = ib_get_agent_port(device, port_num);
        if (!port_priv) {
                printk(KERN_DEBUG SPFX "smi_check_local_dr_smp %s port %d "
-                      "not open\n",
-                      device->name, port_num);
+                      "not open\n", device->name, port_num);
                return 1;
        }
 
-       return smi_check_local_smp(port_priv->smp_agent, smp);
+       return smi_check_local_smp(port_priv->agent[0], smp);
 }
 
-static int agent_mad_send(struct ib_mad_agent *mad_agent,
-                         struct ib_agent_port_private *port_priv,
-                         struct ib_mad_private *mad_priv,
-                         struct ib_grh *grh,
-                         struct ib_wc *wc)
-{
-       struct ib_agent_send_wr *agent_send_wr;
-       struct ib_sge gather_list;
-       struct ib_send_wr send_wr;
-       struct ib_send_wr *bad_send_wr;
-       struct ib_ah_attr ah_attr;
-       unsigned long flags;
-       int ret = 1;
-
-       agent_send_wr = kmalloc(sizeof(*agent_send_wr), GFP_KERNEL);
-       if (!agent_send_wr)
-               goto out;
-       agent_send_wr->mad = mad_priv;
-
-       gather_list.addr = dma_map_single(mad_agent->device->dma_device,
-                                         &mad_priv->mad,
-                                         sizeof(mad_priv->mad),
-                                         DMA_TO_DEVICE);
-       gather_list.length = sizeof(mad_priv->mad);
-       gather_list.lkey = mad_agent->mr->lkey;
-
-       send_wr.next = NULL;
-       send_wr.opcode = IB_WR_SEND;
-       send_wr.sg_list = &gather_list;
-       send_wr.num_sge = 1;
-       send_wr.wr.ud.remote_qpn = wc->src_qp; /* DQPN */
-       send_wr.wr.ud.timeout_ms = 0;
-       send_wr.send_flags = IB_SEND_SIGNALED | IB_SEND_SOLICITED;
-
-       ah_attr.dlid = wc->slid;
-       ah_attr.port_num = mad_agent->port_num;
-       ah_attr.src_path_bits = wc->dlid_path_bits;
-       ah_attr.sl = wc->sl;
-       ah_attr.static_rate = 0;
-       ah_attr.ah_flags = 0; /* No GRH */
-       if (mad_priv->mad.mad.mad_hdr.mgmt_class == IB_MGMT_CLASS_PERF_MGMT) {
-               if (wc->wc_flags & IB_WC_GRH) {
-                       ah_attr.ah_flags = IB_AH_GRH;
-                       /* Should sgid be looked up ? */
-                       ah_attr.grh.sgid_index = 0;
-                       ah_attr.grh.hop_limit = grh->hop_limit;
-                       ah_attr.grh.flow_label = be32_to_cpu(
-                               grh->version_tclass_flow)  & 0xfffff;
-                       ah_attr.grh.traffic_class = (be32_to_cpu(
-                               grh->version_tclass_flow) >> 20) & 0xff;
-                       memcpy(ah_attr.grh.dgid.raw,
-                              grh->sgid.raw,
-                              sizeof(ah_attr.grh.dgid));
-               }
-       }
-
-       agent_send_wr->ah = ib_create_ah(mad_agent->qp->pd, &ah_attr);
-       if (IS_ERR(agent_send_wr->ah)) {
-               printk(KERN_ERR SPFX "No memory for address handle\n");
-               kfree(agent_send_wr);
-               goto out;
-       }
-
-       send_wr.wr.ud.ah = agent_send_wr->ah;
-       if (mad_priv->mad.mad.mad_hdr.mgmt_class == IB_MGMT_CLASS_PERF_MGMT) {
-               send_wr.wr.ud.pkey_index = wc->pkey_index;
-               send_wr.wr.ud.remote_qkey = IB_QP1_QKEY;
-       } else {        /* for SMPs */
-               send_wr.wr.ud.pkey_index = 0;
-               send_wr.wr.ud.remote_qkey = 0;
-       }
-       send_wr.wr.ud.mad_hdr = &mad_priv->mad.mad.mad_hdr;
-       send_wr.wr_id = (unsigned long)agent_send_wr;
-
-       pci_unmap_addr_set(agent_send_wr, mapping, gather_list.addr);
-
-       /* Send */
-       spin_lock_irqsave(&port_priv->send_list_lock, flags);
-       if (ib_post_send_mad(mad_agent, &send_wr, &bad_send_wr)) {
-               spin_unlock_irqrestore(&port_priv->send_list_lock, flags);
-               dma_unmap_single(mad_agent->device->dma_device,
-                                pci_unmap_addr(agent_send_wr, mapping),
-                                sizeof(mad_priv->mad),
-                                DMA_TO_DEVICE);
-               ib_destroy_ah(agent_send_wr->ah);
-               kfree(agent_send_wr);
-       } else {
-               list_add_tail(&agent_send_wr->send_list,
-                             &port_priv->send_posted_list);
-               spin_unlock_irqrestore(&port_priv->send_list_lock, flags);
-               ret = 0;
-       }
-
-out:
-       return ret;
-}
-
-int agent_send(struct ib_mad_private *mad,
-              struct ib_grh *grh,
-              struct ib_wc *wc,
-              struct ib_device *device,
-              int port_num)
+int agent_send_response(struct ib_mad *mad, struct ib_grh *grh,
+                       struct ib_wc *wc, struct ib_device *device,
+                       int port_num, int qpn)
 {
        struct ib_agent_port_private *port_priv;
-       struct ib_mad_agent *mad_agent;
-
-       port_priv = ib_get_agent_port(device, port_num, NULL);
-       if (!port_priv) {
-               printk(KERN_DEBUG SPFX "agent_send %s port %d not open\n",
-                      device->name, port_num);
-               return 1;
-       }
+       struct ib_mad_agent *agent;
+       struct ib_mad_send_buf *send_buf;
+       struct ib_send_wr *bad_wr;
+       struct ib_ah *ah;
+       int ret;
 
-       /* Get mad agent based on mgmt_class in MAD */
-       switch (mad->mad.mad.mad_hdr.mgmt_class) {
-               case IB_MGMT_CLASS_SUBN_DIRECTED_ROUTE:
-               case IB_MGMT_CLASS_SUBN_LID_ROUTED:
-                       mad_agent = port_priv->smp_agent;
-                       break;
-               case IB_MGMT_CLASS_PERF_MGMT:
-                       mad_agent = port_priv->perf_mgmt_agent;
-                       break;
-               default:
-                       return 1;
-       }
+       port_priv = ib_get_agent_port(device, port_num);
+       if (!port_priv)
+               return -ENODEV;
 
-       return agent_mad_send(mad_agent, port_priv, mad, grh, wc);
+       agent = port_priv->agent[qpn];
+       ah = ib_create_ah_from_wc(agent->qp->pd, wc, grh, port_num);
+       if (IS_ERR(ah))
+               return PTR_ERR(ah);
+
+       send_buf = ib_create_send_mad(agent, wc->src_qp, wc->pkey_index, ah, 0,
+                                     sizeof *mad - IB_MGMT_MAD_DATA,
+                                     IB_MGMT_MAD_DATA, GFP_KERNEL);
+       if (IS_ERR(send_buf)) {
+               ret = PTR_ERR(send_buf);
+               goto err1;
+       }
+
+       *send_buf->mad = *mad;
+       if ((ret = ib_post_send_mad(agent, &send_buf->send_wr, &bad_wr)))
+               goto err2;
+       return 0;
+err2:
+       ib_free_send_mad(send_buf);
+err1:
+       ib_destroy_ah(ah);
+       return ret;
 }
 
 static void agent_send_handler(struct ib_mad_agent *mad_agent,
                               struct ib_mad_send_wc *mad_send_wc)
 {
-       struct ib_agent_port_private    *port_priv;
-       struct ib_agent_send_wr         *agent_send_wr;
-       unsigned long                   flags;
-
-       /* Find matching MAD agent */
-       port_priv = ib_get_agent_port(NULL, 0, mad_agent);
-       if (!port_priv) {
-               printk(KERN_ERR SPFX "agent_send_handler: no matching MAD "
-                      "agent %p\n", mad_agent);
-               return;
-       }
+       struct ib_mad_send_buf *send_buf;
 
-       agent_send_wr = (struct ib_agent_send_wr *)(unsigned 
long)mad_send_wc->wr_id;
-       spin_lock_irqsave(&port_priv->send_list_lock, flags);
-       /* Remove completed send from posted send MAD list */
-       list_del(&agent_send_wr->send_list);
-       spin_unlock_irqrestore(&port_priv->send_list_lock, flags);
-
-       dma_unmap_single(mad_agent->device->dma_device,
-                        pci_unmap_addr(agent_send_wr, mapping),
-                        sizeof(agent_send_wr->mad->mad),
-                        DMA_TO_DEVICE);
-
-       ib_destroy_ah(agent_send_wr->ah);
-
-       /* Release allocated memory */
-       kmem_cache_free(ib_mad_cache, agent_send_wr->mad);
-       kfree(agent_send_wr);
+       send_buf = (void *)(unsigned long) mad_send_wc->wr_id;
+       ib_destroy_ah(send_buf->send_wr.wr.ud.ah);
+       ib_free_send_mad(send_buf);
 }
 
 int ib_agent_port_open(struct ib_device *device, int port_num)
 {
-       int ret;
        struct ib_agent_port_private *port_priv;
        unsigned long flags;
-
-       /* First, check if port already open for SMI */
-       port_priv = ib_get_agent_port(device, port_num, NULL);
-       if (port_priv) {
-               printk(KERN_DEBUG SPFX "%s port %d already open\n",
-                      device->name, port_num);
-               return 0;
-       }
+       int ret;
 
        /* Create new device info */
        port_priv = kmalloc(sizeof *port_priv, GFP_KERNEL);
@@ -293,32 +155,25 @@ int ib_agent_port_open(struct ib_device 
                ret = -ENOMEM;
                goto error1;
        }
-
        memset(port_priv, 0, sizeof *port_priv);
-       port_priv->port_num = port_num;
-       spin_lock_init(&port_priv->send_list_lock);
-       INIT_LIST_HEAD(&port_priv->send_posted_list);
-
-       /* Obtain send only MAD agent for SM class (SMI QP) */
-       port_priv->smp_agent = ib_register_mad_agent(device, port_num,
-                                                    IB_QPT_SMI,
-                                                    NULL, 0,
-                                                   &agent_send_handler,
-                                                    NULL, NULL);
 
-       if (IS_ERR(port_priv->smp_agent)) {
-               ret = PTR_ERR(port_priv->smp_agent);
+       /* Obtain send only MAD agent for SMI QP */
+       port_priv->agent[0] = ib_register_mad_agent(device, port_num,
+                                                   IB_QPT_SMI, NULL, 0,
+                                                   &agent_send_handler,
+                                                   NULL, NULL);
+       if (IS_ERR(port_priv->agent[0])) {
+               ret = PTR_ERR(port_priv->agent[0]);
                goto error2;
        }
 
-       /* Obtain send only MAD agent for PerfMgmt class (GSI QP) */
-       port_priv->perf_mgmt_agent = ib_register_mad_agent(device, port_num,
-                                                          IB_QPT_GSI,
-                                                          NULL, 0,
-                                                         &agent_send_handler,
-                                                          NULL, NULL);
-       if (IS_ERR(port_priv->perf_mgmt_agent)) {
-               ret = PTR_ERR(port_priv->perf_mgmt_agent);
+       /* Obtain send only MAD agent for GSI QP */
+       port_priv->agent[1] = ib_register_mad_agent(device, port_num,
+                                                   IB_QPT_GSI, NULL, 0,
+                                                   &agent_send_handler,
+                                                   NULL, NULL);
+       if (IS_ERR(port_priv->agent[1])) {
+               ret = PTR_ERR(port_priv->agent[1]);
                goto error3;
        }
 
@@ -329,7 +184,7 @@ int ib_agent_port_open(struct ib_device 
        return 0;
 
 error3:
-       ib_unregister_mad_agent(port_priv->smp_agent);
+       ib_unregister_mad_agent(port_priv->agent[0]);
 error2:
        kfree(port_priv);
 error1:
@@ -342,7 +197,7 @@ int ib_agent_port_close(struct ib_device
        unsigned long flags;
 
        spin_lock_irqsave(&ib_agent_port_list_lock, flags);
-       port_priv = __ib_get_agent_port(device, port_num, NULL);
+       port_priv = __ib_get_agent_port(device, port_num);
        if (port_priv == NULL) {
                spin_unlock_irqrestore(&ib_agent_port_list_lock, flags);
                printk(KERN_ERR SPFX "Port %d not found\n", port_num);
@@ -351,9 +206,8 @@ int ib_agent_port_close(struct ib_device
        list_del(&port_priv->port_list);
        spin_unlock_irqrestore(&ib_agent_port_list_lock, flags);
 
-       ib_unregister_mad_agent(port_priv->perf_mgmt_agent);
-       ib_unregister_mad_agent(port_priv->smp_agent);
+       ib_unregister_mad_agent(port_priv->agent[1]);
+       ib_unregister_mad_agent(port_priv->agent[0]);
        kfree(port_priv);
-
        return 0;
 }
Index: agent.h
===================================================================
--- agent.h     (revision 3794)
+++ agent.h     (working copy)
@@ -39,17 +39,14 @@
 #ifndef __AGENT_H_
 #define __AGENT_H_
 
-extern spinlock_t ib_agent_port_list_lock;
+#include <rdma/ib_mad.h>
 
-extern int ib_agent_port_open(struct ib_device *device,
-                             int port_num);
+extern int ib_agent_port_open(struct ib_device *device, int port_num);
 
 extern int ib_agent_port_close(struct ib_device *device, int port_num);
 
-extern int agent_send(struct ib_mad_private *mad,
-                     struct ib_grh *grh,
-                     struct ib_wc *wc,
-                     struct ib_device *device,
-                     int port_num);
+extern int agent_send_response(struct ib_mad *mad, struct ib_grh *grh,
+                              struct ib_wc *wc, struct ib_device *device,
+                              int port_num, int qpn);
 
 #endif /* __AGENT_H_ */
Index: smi.h
===================================================================
--- smi.h       (revision 3794)
+++ smi.h       (working copy)
@@ -35,10 +35,11 @@
  *
  * $Id$
  */
-
 #ifndef __SMI_H_
 #define __SMI_H_
 
+#include <rdma/ib_smi.h>
+
 int smi_handle_dr_smp_recv(struct ib_smp *smp,
                           u8 node_type,
                           int port_num,
Index: mad.c
===================================================================
--- mad.c       (revision 3794)
+++ mad.c       (working copy)
@@ -1728,11 +1728,11 @@ local:
                        if (ret & IB_MAD_RESULT_CONSUMED)
                                goto out;
                        if (ret & IB_MAD_RESULT_REPLY) {
-                               /* Send response */
-                               if (!agent_send(response, &recv->grh, wc,
-                                               port_priv->device,
-                                               port_priv->port_num))
-                                       response = NULL;
+                               agent_send_response(&response->mad.mad,
+                                                   &recv->grh, wc,
+                                                   port_priv->device,
+                                                   port_priv->port_num,
+                                                   qp_info->qp->qp_num);
                                goto out;
                        }
                }
@@ -2761,7 +2761,6 @@ static int __init ib_mad_init_module(voi
        int ret;
 
        spin_lock_init(&ib_mad_port_list_lock);
-       spin_lock_init(&ib_agent_port_list_lock);
 
        ib_mad_cache = kmem_cache_create("ib_mad",
                                         sizeof(struct ib_mad_private),



_______________________________________________
openib-general mailing list
[email protected]
http://openib.org/mailman/listinfo/openib-general

To unsubscribe, please visit http://openib.org/mailman/listinfo/openib-general

Reply via email to