main.c

Go to the documentation of this file.

/*
 * Copyright (c) 2006, 2007 Cisco Systems, Inc. All rights reserved.
 * Copyright (c) 2007, 2008 Mellanox Technologies. All rights reserved.
 *
 * This software is available to you under a choice of one of two
 * licenses.  You may choose to be licensed under the terms of the GNU
 * General Public License (GPL) Version 2, available from the file
 * COPYING in the main directory of this source tree, or the
 * OpenIB.org BSD license below:
 *
 *     Redistribution and use in source and binary forms, with or
 *     without modification, are permitted provided that the following
 *     conditions are met:
 *
 *      - Redistributions of source code must retain the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer.
 *
 *      - Redistributions in binary form must reproduce the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer in the documentation and/or other materials
 *        provided with the distribution.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <linux/module.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/errno.h>
#include <linux/netdevice.h>
#include <linux/inetdevice.h>
#include <linux/rtnetlink.h>
#include <linux/if_vlan.h>

#include <rdma/ib_smi.h>
#include <rdma/ib_user_verbs.h>
#include <rdma/ib_addr.h>

#include <linux/mlx4/driver.h>
#include <linux/mlx4/cmd.h>

#include "mlx4_ib.h"
#include "user.h"

#define DRV_NAME    MLX4_IB_DRV_NAME
#define DRV_VERSION "1.0"
#define DRV_RELDATE "April 4, 2008"

MODULE_AUTHOR("Roland Dreier");
MODULE_DESCRIPTION("Mellanox ConnectX HCA InfiniBand driver");
MODULE_LICENSE("Dual BSD/GPL");
MODULE_VERSION(DRV_VERSION);

int mlx4_ib_sm_guid_assign = 1;
module_param_named(sm_guid_assign, mlx4_ib_sm_guid_assign, int, 0444);
MODULE_PARM_DESC(sm_guid_assign, "Enable SM alias_GUID assignment if sm_guid_assign > 0 (Default: 1)");

static const char mlx4_ib_version[] =
    DRV_NAME ": Mellanox ConnectX InfiniBand driver v"
    DRV_VERSION " (" DRV_RELDATE ")\n";

struct update_gid_work {
    struct work_struct  work;
    union ib_gid        gids[128];
    struct mlx4_ib_dev     *dev;
    int         port;
};

static void do_slave_init(struct mlx4_ib_dev *ibdev, int slave, int do_init);

static struct workqueue_struct *wq;

static void init_query_mad(struct ib_smp *mad)
{
    mad->base_version  = 1;
    mad->mgmt_class    = IB_MGMT_CLASS_SUBN_LID_ROUTED;
    mad->class_version = 1;
    mad->method    = IB_MGMT_METHOD_GET;
}

static union ib_gid zgid;

static int mlx4_ib_query_device(struct ib_device *ibdev,
                struct ib_device_attr *props)
{
    struct mlx4_ib_dev *dev = to_mdev(ibdev);
    struct ib_smp *in_mad  = NULL;
    struct ib_smp *out_mad = NULL;
    int err = -ENOMEM;

    in_mad  = kzalloc(sizeof *in_mad, GFP_KERNEL);
    out_mad = kmalloc(sizeof *out_mad, GFP_KERNEL);
    if (!in_mad || !out_mad)
        goto out;

    init_query_mad(in_mad);
    in_mad->attr_id = IB_SMP_ATTR_NODE_INFO;

    err = mlx4_MAD_IFC(to_mdev(ibdev), MLX4_MAD_IFC_IGNORE_KEYS,
               1, NULL, NULL, in_mad, out_mad);
    if (err)
        goto out;

    memset(props, 0, sizeof *props);

    props->fw_ver = dev->dev->caps.fw_ver;
    props->device_cap_flags    = IB_DEVICE_CHANGE_PHY_PORT |
        IB_DEVICE_PORT_ACTIVE_EVENT     |
        IB_DEVICE_SYS_IMAGE_GUID        |
        IB_DEVICE_RC_RNR_NAK_GEN        |
        IB_DEVICE_BLOCK_MULTICAST_LOOPBACK;
    if (dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_BAD_PKEY_CNTR)
        props->device_cap_flags |= IB_DEVICE_BAD_PKEY_CNTR;
    if (dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_BAD_QKEY_CNTR)
        props->device_cap_flags |= IB_DEVICE_BAD_QKEY_CNTR;
    if (dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_APM)
        props->device_cap_flags |= IB_DEVICE_AUTO_PATH_MIG;
    if (dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_UD_AV_PORT)
        props->device_cap_flags |= IB_DEVICE_UD_AV_PORT_ENFORCE;
    if (dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_IPOIB_CSUM)
        props->device_cap_flags |= IB_DEVICE_UD_IP_CSUM;
    if (dev->dev->caps.max_gso_sz && dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_BLH)
        props->device_cap_flags |= IB_DEVICE_UD_TSO;
    if (dev->dev->caps.bmme_flags & MLX4_BMME_FLAG_RESERVED_LKEY)
        props->device_cap_flags |= IB_DEVICE_LOCAL_DMA_LKEY;
    if ((dev->dev->caps.bmme_flags & MLX4_BMME_FLAG_LOCAL_INV) &&
        (dev->dev->caps.bmme_flags & MLX4_BMME_FLAG_REMOTE_INV) &&
        (dev->dev->caps.bmme_flags & MLX4_BMME_FLAG_FAST_REG_WR))
        props->device_cap_flags |= IB_DEVICE_MEM_MGT_EXTENSIONS;
    if (dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_XRC)
        props->device_cap_flags |= IB_DEVICE_XRC;

    props->vendor_id       = be32_to_cpup((__be32 *) (out_mad->data + 36)) &
        0xffffff;
    props->vendor_part_id      = dev->dev->pdev->device;
    props->hw_ver          = be32_to_cpup((__be32 *) (out_mad->data + 32));
    memcpy(&props->sys_image_guid, out_mad->data +  4, 8);

    props->max_mr_size     = ~0ull;
    props->page_size_cap       = dev->dev->caps.page_size_cap;
    props->max_qp          = dev->dev->caps.num_qps - dev->dev->caps.reserved_qps;
    props->max_qp_wr       = dev->dev->caps.max_wqes - MLX4_IB_SQ_MAX_SPARE;
    props->max_sge         = min(dev->dev->caps.max_sq_sg,
                     dev->dev->caps.max_rq_sg);
    props->max_cq          = dev->dev->caps.num_cqs - dev->dev->caps.reserved_cqs;
    props->max_cqe         = dev->dev->caps.max_cqes;
    props->max_mr          = dev->dev->caps.num_mpts - dev->dev->caps.reserved_mrws;
    props->max_pd          = dev->dev->caps.num_pds - dev->dev->caps.reserved_pds;
    props->max_qp_rd_atom      = dev->dev->caps.max_qp_dest_rdma;
    props->max_qp_init_rd_atom = dev->dev->caps.max_qp_init_rdma;
    props->max_res_rd_atom     = props->max_qp_rd_atom * props->max_qp;
    props->max_srq         = dev->dev->caps.num_srqs - dev->dev->caps.reserved_srqs;
    props->max_srq_wr      = dev->dev->caps.max_srq_wqes - 1;
    props->max_srq_sge     = dev->dev->caps.max_srq_sge;
    props->max_fast_reg_page_list_len = MLX4_MAX_FAST_REG_PAGES;
    props->local_ca_ack_delay  = dev->dev->caps.local_ca_ack_delay;
    props->atomic_cap      = dev->dev->caps.flags & MLX4_DEV_CAP_FLAG_ATOMIC ?
        IB_ATOMIC_HCA : IB_ATOMIC_NONE;
    props->masked_atomic_cap   = props->atomic_cap;
    props->max_pkeys       = dev->dev->caps.pkey_table_len[1];
    props->max_mcast_grp       = dev->dev->caps.num_mgms + dev->dev->caps.num_amgms;
    props->max_mcast_qp_attach = dev->dev->caps.num_qp_per_mgm;
    props->max_total_mcast_qp_attach = props->max_mcast_qp_attach *
                       props->max_mcast_grp;
    props->max_map_per_fmr = dev->dev->caps.max_fmr_maps;

out:
    kfree(in_mad);
    kfree(out_mad);

    return err;
}

static enum rdma_link_layer
mlx4_ib_port_link_layer(struct ib_device *device, u8 port_num)
{
    struct mlx4_dev *dev = to_mdev(device)->dev;

    return dev->caps.port_mask[port_num] == MLX4_PORT_TYPE_IB ?
        IB_LINK_LAYER_INFINIBAND : IB_LINK_LAYER_ETHERNET;
}

static int ib_link_query_port(struct ib_device *ibdev, u8 port,
                  struct ib_port_attr *props, int netw_view)
{
    struct ib_smp *in_mad  = NULL;
    struct ib_smp *out_mad = NULL;
    int ext_active_speed;
    int mad_ifc_flags = MLX4_MAD_IFC_IGNORE_KEYS;
    int err = -ENOMEM;

    in_mad  = kzalloc(sizeof *in_mad, GFP_KERNEL);
    out_mad = kmalloc(sizeof *out_mad, GFP_KERNEL);
    if (!in_mad || !out_mad)
        goto out;

    init_query_mad(in_mad);
    in_mad->attr_id  = IB_SMP_ATTR_PORT_INFO;
    in_mad->attr_mod = cpu_to_be32(port);

    if (mlx4_is_mfunc(to_mdev(ibdev)->dev) && netw_view)
        mad_ifc_flags |= MLX4_MAD_IFC_NET_VIEW;

    err = mlx4_MAD_IFC(to_mdev(ibdev), mad_ifc_flags, port, NULL, NULL,
                in_mad, out_mad);
    if (err)
        goto out;


    props->lid      = be16_to_cpup((__be16 *) (out_mad->data + 16));
    props->lmc      = out_mad->data[34] & 0x7;
    props->sm_lid       = be16_to_cpup((__be16 *) (out_mad->data + 18));
    props->sm_sl        = out_mad->data[36] & 0xf;
    props->state        = out_mad->data[32] & 0xf;
    props->phys_state   = out_mad->data[33] >> 4;
    props->port_cap_flags   = be32_to_cpup((__be32 *) (out_mad->data + 20));
    if (netw_view)
        props->gid_tbl_len = out_mad->data[50];
    else
        props->gid_tbl_len = to_mdev(ibdev)->dev->caps.gid_table_len[port];
    props->max_msg_sz   = to_mdev(ibdev)->dev->caps.max_msg_sz;
    props->pkey_tbl_len = to_mdev(ibdev)->dev->caps.pkey_table_len[port];
    props->bad_pkey_cntr    = be16_to_cpup((__be16 *) (out_mad->data + 46));
    props->qkey_viol_cntr   = be16_to_cpup((__be16 *) (out_mad->data + 48));
    props->active_width = out_mad->data[31] & 0xf;
    props->active_speed = out_mad->data[35] >> 4;
    props->max_mtu      = out_mad->data[41] & 0xf;
    props->active_mtu   = out_mad->data[36] >> 4;
    props->subnet_timeout   = out_mad->data[51] & 0x1f;
    props->max_vl_num   = out_mad->data[37] >> 4;
    props->init_type_reply  = out_mad->data[41] >> 4;

    /* Check if extended speeds (EDR/FDR/...) are supported */
    if (props->port_cap_flags & IB_PORT_EXTENDED_SPEEDS_SUP) {
        ext_active_speed = out_mad->data[62] >> 4;

        switch (ext_active_speed) {
        case 1:
            props->active_speed = IB_SPEED_FDR;
            break;
        case 2:
            props->active_speed = IB_SPEED_EDR;
            break;
        }
    }

    /* If reported active speed is QDR, check if is FDR-10 */
    if (props->active_speed == IB_SPEED_QDR) {
        init_query_mad(in_mad);
        in_mad->attr_id = MLX4_ATTR_EXTENDED_PORT_INFO;
        in_mad->attr_mod = cpu_to_be32(port);

        err = mlx4_MAD_IFC(to_mdev(ibdev), mad_ifc_flags, port,
                   NULL, NULL, in_mad, out_mad);
        if (err)
            goto out;

        /* Checking LinkSpeedActive for FDR-10 */
        if (out_mad->data[15] & 0x1)
            props->active_speed = IB_SPEED_FDR10;
    }

    /* Avoid wrong speed value returned by FW if the IB link is down. */
    if (props->state == IB_PORT_DOWN)
         props->active_speed = IB_SPEED_SDR;

out:
    kfree(in_mad);
    kfree(out_mad);
    return err;
}

static u8 state_to_phys_state(enum ib_port_state state)
{
    return state == IB_PORT_ACTIVE ? 5 : 3;
}

static int eth_link_query_port(struct ib_device *ibdev, u8 port,
                   struct ib_port_attr *props, int netw_view)
{

    struct mlx4_ib_dev *mdev = to_mdev(ibdev);
    struct mlx4_ib_iboe *iboe = &mdev->iboe;
    struct net_device *ndev;
    enum ib_mtu tmp;
    struct mlx4_cmd_mailbox *mailbox;
    int err = 0;

    mailbox = mlx4_alloc_cmd_mailbox(mdev->dev);
    if (IS_ERR(mailbox))
        return PTR_ERR(mailbox);

    err = mlx4_cmd_box(mdev->dev, 0, mailbox->dma, port, 0,
               MLX4_CMD_QUERY_PORT, MLX4_CMD_TIME_CLASS_B,
               MLX4_CMD_WRAPPED);
    if (err)
        goto out;

    props->active_width =  (((u8 *)mailbox->buf)[5] == 0x40) ?
                        IB_WIDTH_4X : IB_WIDTH_1X;
    props->active_speed = IB_SPEED_QDR;
    props->port_cap_flags   = IB_PORT_CM_SUP;
    props->gid_tbl_len  = mdev->dev->caps.gid_table_len[port];
    props->max_msg_sz   = mdev->dev->caps.max_msg_sz;
    props->pkey_tbl_len = 1;
    props->max_mtu      = IB_MTU_4096;
    props->max_vl_num   = 2;
    props->state        = IB_PORT_DOWN;
    props->phys_state   = state_to_phys_state(props->state);
    props->active_mtu   = IB_MTU_256;
    spin_lock(&iboe->lock);
    ndev = iboe->netdevs[port - 1];
    if (!ndev)
        goto out_unlock;

    tmp = iboe_get_mtu(ndev->mtu);
    props->active_mtu = tmp ? min(props->max_mtu, tmp) : IB_MTU_256;

    props->state        = (netif_running(ndev) && netif_carrier_ok(ndev)) ?
                    IB_PORT_ACTIVE : IB_PORT_DOWN;
    props->phys_state   = state_to_phys_state(props->state);
out_unlock:
    spin_unlock(&iboe->lock);
out:
    mlx4_free_cmd_mailbox(mdev->dev, mailbox);
    return err;
}

int __mlx4_ib_query_port(struct ib_device *ibdev, u8 port,
             struct ib_port_attr *props, int netw_view)
{
    int err;

    memset(props, 0, sizeof *props);

    err = mlx4_ib_port_link_layer(ibdev, port) == IB_LINK_LAYER_INFINIBAND ?
        ib_link_query_port(ibdev, port, props, netw_view) :
                eth_link_query_port(ibdev, port, props, netw_view);

    return err;
}

static int mlx4_ib_query_port(struct ib_device *ibdev, u8 port,
                  struct ib_port_attr *props)
{
    /* returns host view */
    return __mlx4_ib_query_port(ibdev, port, props, 0);
}

int __mlx4_ib_query_gid(struct ib_device *ibdev, u8 port, int index,
            union ib_gid *gid, int netw_view)
{
    struct ib_smp *in_mad  = NULL;
    struct ib_smp *out_mad = NULL;
    int err = -ENOMEM;
    struct mlx4_ib_dev *dev = to_mdev(ibdev);
    int clear = 0;
    int mad_ifc_flags = MLX4_MAD_IFC_IGNORE_KEYS;

    in_mad  = kzalloc(sizeof *in_mad, GFP_KERNEL);
    out_mad = kmalloc(sizeof *out_mad, GFP_KERNEL);
    if (!in_mad || !out_mad)
        goto out;

    init_query_mad(in_mad);
    in_mad->attr_id  = IB_SMP_ATTR_PORT_INFO;
    in_mad->attr_mod = cpu_to_be32(port);

    if (mlx4_is_mfunc(dev->dev) && netw_view)
        mad_ifc_flags |= MLX4_MAD_IFC_NET_VIEW;

    err = mlx4_MAD_IFC(dev, mad_ifc_flags, port, NULL, NULL, in_mad, out_mad);
    if (err)
        goto out;

    memcpy(gid->raw, out_mad->data + 8, 8);

    if (mlx4_is_mfunc(dev->dev) && !netw_view) {
        if (index) {
            /* For any index > 0, return the null guid */
            err = 0;
            clear = 1;
            goto out;
        }
    }

    init_query_mad(in_mad);
    in_mad->attr_id  = IB_SMP_ATTR_GUID_INFO;
    in_mad->attr_mod = cpu_to_be32(index / 8);

    err = mlx4_MAD_IFC(dev, mad_ifc_flags, port,
               NULL, NULL, in_mad, out_mad);
    if (err)
        goto out;

    memcpy(gid->raw + 8, out_mad->data + (index % 8) * 8, 8);

out:
    if (clear)
        memset(gid->raw + 8, 0, 8);
    kfree(in_mad);
    kfree(out_mad);
    return err;
}

static int iboe_query_gid(struct ib_device *ibdev, u8 port, int index,
              union ib_gid *gid)
{
    struct mlx4_ib_dev *dev = to_mdev(ibdev);

    *gid = dev->iboe.gid_table[port - 1][index];

    return 0;
}

static int mlx4_ib_query_gid(struct ib_device *ibdev, u8 port, int index,
                 union ib_gid *gid)
{
    if (rdma_port_get_link_layer(ibdev, port) == IB_LINK_LAYER_INFINIBAND)
        return __mlx4_ib_query_gid(ibdev, port, index, gid, 0);
    else
        return iboe_query_gid(ibdev, port, index, gid);
}

int __mlx4_ib_query_pkey(struct ib_device *ibdev, u8 port, u16 index,
             u16 *pkey, int netw_view)
{
    struct ib_smp *in_mad  = NULL;
    struct ib_smp *out_mad = NULL;
    int mad_ifc_flags = MLX4_MAD_IFC_IGNORE_KEYS;
    int err = -ENOMEM;

    in_mad  = kzalloc(sizeof *in_mad, GFP_KERNEL);
    out_mad = kmalloc(sizeof *out_mad, GFP_KERNEL);
    if (!in_mad || !out_mad)
        goto out;

    init_query_mad(in_mad);
    in_mad->attr_id  = IB_SMP_ATTR_PKEY_TABLE;
    in_mad->attr_mod = cpu_to_be32(index / 32);

    if (mlx4_is_mfunc(to_mdev(ibdev)->dev) && netw_view)
        mad_ifc_flags |= MLX4_MAD_IFC_NET_VIEW;

    err = mlx4_MAD_IFC(to_mdev(ibdev), mad_ifc_flags, port, NULL, NULL,
               in_mad, out_mad);
    if (err)
        goto out;

    *pkey = be16_to_cpu(((__be16 *) out_mad->data)[index % 32]);

out:
    kfree(in_mad);
    kfree(out_mad);
    return err;
}

static int mlx4_ib_query_pkey(struct ib_device *ibdev, u8 port, u16 index, u16 *pkey)
{
    return __mlx4_ib_query_pkey(ibdev, port, index, pkey, 0);
}

static int mlx4_ib_modify_device(struct ib_device *ibdev, int mask,
                 struct ib_device_modify *props)
{
    struct mlx4_cmd_mailbox *mailbox;
    unsigned long flags;

    if (mask & ~IB_DEVICE_MODIFY_NODE_DESC)
        return -EOPNOTSUPP;

    if (!(mask & IB_DEVICE_MODIFY_NODE_DESC))
        return 0;

    if (mlx4_is_slave(to_mdev(ibdev)->dev))
        return -EOPNOTSUPP;

    spin_lock_irqsave(&to_mdev(ibdev)->sm_lock, flags);
    memcpy(ibdev->node_desc, props->node_desc, 64);
    spin_unlock_irqrestore(&to_mdev(ibdev)->sm_lock, flags);

    /*
     * If possible, pass node desc to FW, so it can generate
     * a 144 trap.  If cmd fails, just ignore.
     */
    mailbox = mlx4_alloc_cmd_mailbox(to_mdev(ibdev)->dev);
    if (IS_ERR(mailbox))
        return 0;

    memset(mailbox->buf, 0, 256);
    memcpy(mailbox->buf, props->node_desc, 64);
    mlx4_cmd(to_mdev(ibdev)->dev, mailbox->dma, 1, 0,
         MLX4_CMD_SET_NODE, MLX4_CMD_TIME_CLASS_A, MLX4_CMD_NATIVE);

    mlx4_free_cmd_mailbox(to_mdev(ibdev)->dev, mailbox);

    return 0;
}

static int mlx4_SET_PORT(struct mlx4_ib_dev *dev, u8 port, int reset_qkey_viols,
             u32 cap_mask)
{
    struct mlx4_cmd_mailbox *mailbox;
    int err;
    u8 is_eth = dev->dev->caps.port_type[port] == MLX4_PORT_TYPE_ETH;

    mailbox = mlx4_alloc_cmd_mailbox(dev->dev);
    if (IS_ERR(mailbox))
        return PTR_ERR(mailbox);

    memset(mailbox->buf, 0, 256);

    if (dev->dev->flags & MLX4_FLAG_OLD_PORT_CMDS) {
        *(u8 *) mailbox->buf         = !!reset_qkey_viols << 6;
        ((__be32 *) mailbox->buf)[2] = cpu_to_be32(cap_mask);
    } else {
        ((u8 *) mailbox->buf)[3]     = !!reset_qkey_viols;
        ((__be32 *) mailbox->buf)[1] = cpu_to_be32(cap_mask);
    }

    err = mlx4_cmd(dev->dev, mailbox->dma, port, is_eth, MLX4_CMD_SET_PORT,
               MLX4_CMD_TIME_CLASS_B, MLX4_CMD_NATIVE);

    mlx4_free_cmd_mailbox(dev->dev, mailbox);
    return err;
}

static int mlx4_ib_modify_port(struct ib_device *ibdev, u8 port, int mask,
                   struct ib_port_modify *props)
{
    struct ib_port_attr attr;
    u32 cap_mask;
    int err;

    mutex_lock(&to_mdev(ibdev)->cap_mask_mutex);

    err = mlx4_ib_query_port(ibdev, port, &attr);
    if (err)
        goto out;

    cap_mask = (attr.port_cap_flags | props->set_port_cap_mask) &
        ~props->clr_port_cap_mask;

    err = mlx4_SET_PORT(to_mdev(ibdev), port,
                !!(mask & IB_PORT_RESET_QKEY_CNTR),
                cap_mask);

out:
    mutex_unlock(&to_mdev(ibdev)->cap_mask_mutex);
    return err;
}

static struct ib_ucontext *mlx4_ib_alloc_ucontext(struct ib_device *ibdev,
                          struct ib_udata *udata)
{
    struct mlx4_ib_dev *dev = to_mdev(ibdev);
    struct mlx4_ib_ucontext *context;
    struct mlx4_ib_alloc_ucontext_resp resp;
    int err;

    if (!dev->ib_active)
        return ERR_PTR(-EAGAIN);

    resp.qp_tab_size      = dev->dev->caps.num_qps;
    resp.bf_reg_size      = dev->dev->caps.bf_reg_size;
    resp.bf_regs_per_page = dev->dev->caps.bf_regs_per_page;

    context = kmalloc(sizeof *context, GFP_KERNEL);
    if (!context)
        return ERR_PTR(-ENOMEM);

    err = mlx4_uar_alloc(to_mdev(ibdev)->dev, &context->uar);
    if (err) {
        kfree(context);
        return ERR_PTR(err);
    }

    INIT_LIST_HEAD(&context->db_page_list);
    mutex_init(&context->db_page_mutex);

    err = ib_copy_to_udata(udata, &resp, sizeof resp);
    if (err) {
        mlx4_uar_free(to_mdev(ibdev)->dev, &context->uar);
        kfree(context);
        return ERR_PTR(-EFAULT);
    }

    return &context->ibucontext;
}

static int mlx4_ib_dealloc_ucontext(struct ib_ucontext *ibcontext)
{
    struct mlx4_ib_ucontext *context = to_mucontext(ibcontext);

    mlx4_uar_free(to_mdev(ibcontext->device)->dev, &context->uar);
    kfree(context);

    return 0;
}

static int mlx4_ib_mmap(struct ib_ucontext *context, struct vm_area_struct *vma)
{
    struct mlx4_ib_dev *dev = to_mdev(context->device);

    if (vma->vm_end - vma->vm_start != PAGE_SIZE)
        return -EINVAL;

    if (vma->vm_pgoff == 0) {
        vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);

        if (io_remap_pfn_range(vma, vma->vm_start,
                       to_mucontext(context)->uar.pfn,
                       PAGE_SIZE, vma->vm_page_prot))
            return -EAGAIN;
    } else if (vma->vm_pgoff == 1 && dev->dev->caps.bf_reg_size != 0) {
        vma->vm_page_prot = pgprot_writecombine(vma->vm_page_prot);

        if (io_remap_pfn_range(vma, vma->vm_start,
                       to_mucontext(context)->uar.pfn +
                       dev->dev->caps.num_uars,
                       PAGE_SIZE, vma->vm_page_prot))
            return -EAGAIN;
    } else
        return -EINVAL;

    return 0;
}

static struct ib_pd *mlx4_ib_alloc_pd(struct ib_device *ibdev,
                      struct ib_ucontext *context,
                      struct ib_udata *udata)
{
    struct mlx4_ib_pd *pd;
    int err;

    pd = kmalloc(sizeof *pd, GFP_KERNEL);
    if (!pd)
        return ERR_PTR(-ENOMEM);

    err = mlx4_pd_alloc(to_mdev(ibdev)->dev, &pd->pdn);
    if (err) {
        kfree(pd);
        return ERR_PTR(err);
    }

    if (context)
        if (ib_copy_to_udata(udata, &pd->pdn, sizeof (__u32))) {
            mlx4_pd_free(to_mdev(ibdev)->dev, pd->pdn);
            kfree(pd);
            return ERR_PTR(-EFAULT);
        }

    return &pd->ibpd;
}

static int mlx4_ib_dealloc_pd(struct ib_pd *pd)
{
    mlx4_pd_free(to_mdev(pd->device)->dev, to_mpd(pd)->pdn);
    kfree(pd);

    return 0;
}

static struct ib_xrcd *mlx4_ib_alloc_xrcd(struct ib_device *ibdev,
                      struct ib_ucontext *context,
                      struct ib_udata *udata)
{
    struct mlx4_ib_xrcd *xrcd;
    int err;

    if (!(to_mdev(ibdev)->dev->caps.flags & MLX4_DEV_CAP_FLAG_XRC))
        return ERR_PTR(-ENOSYS);

    xrcd = kmalloc(sizeof *xrcd, GFP_KERNEL);
    if (!xrcd)
        return ERR_PTR(-ENOMEM);

    err = mlx4_xrcd_alloc(to_mdev(ibdev)->dev, &xrcd->xrcdn);
    if (err)
        goto err1;

    xrcd->pd = ib_alloc_pd(ibdev);
    if (IS_ERR(xrcd->pd)) {
        err = PTR_ERR(xrcd->pd);
        goto err2;
    }

    xrcd->cq = ib_create_cq(ibdev, NULL, NULL, xrcd, 1, 0);
    if (IS_ERR(xrcd->cq)) {
        err = PTR_ERR(xrcd->cq);
        goto err3;
    }

    return &xrcd->ibxrcd;

err3:
    ib_dealloc_pd(xrcd->pd);
err2:
    mlx4_xrcd_free(to_mdev(ibdev)->dev, xrcd->xrcdn);
err1:
    kfree(xrcd);
    return ERR_PTR(err);
}

static int mlx4_ib_dealloc_xrcd(struct ib_xrcd *xrcd)
{
    ib_destroy_cq(to_mxrcd(xrcd)->cq);
    ib_dealloc_pd(to_mxrcd(xrcd)->pd);
    mlx4_xrcd_free(to_mdev(xrcd->device)->dev, to_mxrcd(xrcd)->xrcdn);
    kfree(xrcd);

    return 0;
}

static int add_gid_entry(struct ib_qp *ibqp, union ib_gid *gid)
{
    struct mlx4_ib_qp *mqp = to_mqp(ibqp);
    struct mlx4_ib_dev *mdev = to_mdev(ibqp->device);
    struct mlx4_ib_gid_entry *ge;

    ge = kzalloc(sizeof *ge, GFP_KERNEL);
    if (!ge)
        return -ENOMEM;

    ge->gid = *gid;
    if (mlx4_ib_add_mc(mdev, mqp, gid)) {
        ge->port = mqp->port;
        ge->added = 1;
    }

    mutex_lock(&mqp->mutex);
    list_add_tail(&ge->list, &mqp->gid_list);
    mutex_unlock(&mqp->mutex);

    return 0;
}

int mlx4_ib_add_mc(struct mlx4_ib_dev *mdev, struct mlx4_ib_qp *mqp,
           union ib_gid *gid)
{
    u8 mac[6];
    struct net_device *ndev;
    int ret = 0;

    if (!mqp->port)
        return 0;

    spin_lock(&mdev->iboe.lock);
    ndev = mdev->iboe.netdevs[mqp->port - 1];
    if (ndev)
        dev_hold(ndev);
    spin_unlock(&mdev->iboe.lock);

    if (ndev) {
        rdma_get_mcast_mac((struct in6_addr *)gid, mac);
        rtnl_lock();
        dev_mc_add(mdev->iboe.netdevs[mqp->port - 1], mac);
        ret = 1;
        rtnl_unlock();
        dev_put(ndev);
    }

    return ret;
}

struct mlx4_ib_steering {
    struct list_head list;
    u64 reg_id;
    union ib_gid gid;
};

static int mlx4_ib_mcg_attach(struct ib_qp *ibqp, union ib_gid *gid, u16 lid)
{
    int err;
    struct mlx4_ib_dev *mdev = to_mdev(ibqp->device);
    struct mlx4_ib_qp *mqp = to_mqp(ibqp);
    u64 reg_id;
    struct mlx4_ib_steering *ib_steering = NULL;

    if (mdev->dev->caps.steering_mode ==
        MLX4_STEERING_MODE_DEVICE_MANAGED) {
        ib_steering = kmalloc(sizeof(*ib_steering), GFP_KERNEL);
        if (!ib_steering)
            return -ENOMEM;
    }

    err = mlx4_multicast_attach(mdev->dev, &mqp->mqp, gid->raw, mqp->port,
                    !!(mqp->flags &
                       MLX4_IB_QP_BLOCK_MULTICAST_LOOPBACK),
                    MLX4_PROT_IB_IPV6, &reg_id);
    if (err)
        goto err_malloc;

    err = add_gid_entry(ibqp, gid);
    if (err)
        goto err_add;

    if (ib_steering) {
        memcpy(ib_steering->gid.raw, gid->raw, 16);
        ib_steering->reg_id = reg_id;
        mutex_lock(&mqp->mutex);
        list_add(&ib_steering->list, &mqp->steering_rules);
        mutex_unlock(&mqp->mutex);
    }
    return 0;

err_add:
    mlx4_multicast_detach(mdev->dev, &mqp->mqp, gid->raw,
                  MLX4_PROT_IB_IPV6, reg_id);
err_malloc:
    kfree(ib_steering);

    return err;
}

static struct mlx4_ib_gid_entry *find_gid_entry(struct mlx4_ib_qp *qp, u8 *raw)
{
    struct mlx4_ib_gid_entry *ge;
    struct mlx4_ib_gid_entry *tmp;
    struct mlx4_ib_gid_entry *ret = NULL;

    list_for_each_entry_safe(ge, tmp, &qp->gid_list, list) {
        if (!memcmp(raw, ge->gid.raw, 16)) {
            ret = ge;
            break;
        }
    }

    return ret;
}

static int mlx4_ib_mcg_detach(struct ib_qp *ibqp, union ib_gid *gid, u16 lid)
{
    int err;
    struct mlx4_ib_dev *mdev = to_mdev(ibqp->device);
    struct mlx4_ib_qp *mqp = to_mqp(ibqp);
    u8 mac[6];
    struct net_device *ndev;
    struct mlx4_ib_gid_entry *ge;
    u64 reg_id = 0;

    if (mdev->dev->caps.steering_mode ==
        MLX4_STEERING_MODE_DEVICE_MANAGED) {
        struct mlx4_ib_steering *ib_steering;

        mutex_lock(&mqp->mutex);
        list_for_each_entry(ib_steering, &mqp->steering_rules, list) {
            if (!memcmp(ib_steering->gid.raw, gid->raw, 16)) {
                list_del(&ib_steering->list);
                break;
            }
        }
        mutex_unlock(&mqp->mutex);
        if (&ib_steering->list == &mqp->steering_rules) {
            pr_err("Couldn't find reg_id for mgid. Steering rule is left attached\n");
            return -EINVAL;
        }
        reg_id = ib_steering->reg_id;
        kfree(ib_steering);
    }

    err = mlx4_multicast_detach(mdev->dev, &mqp->mqp, gid->raw,
                    MLX4_PROT_IB_IPV6, reg_id);
    if (err)
        return err;

    mutex_lock(&mqp->mutex);
    ge = find_gid_entry(mqp, gid->raw);
    if (ge) {
        spin_lock(&mdev->iboe.lock);
        ndev = ge->added ? mdev->iboe.netdevs[ge->port - 1] : NULL;
        if (ndev)
            dev_hold(ndev);
        spin_unlock(&mdev->iboe.lock);
        rdma_get_mcast_mac((struct in6_addr *)gid, mac);
        if (ndev) {
            rtnl_lock();
            dev_mc_del(mdev->iboe.netdevs[ge->port - 1], mac);
            rtnl_unlock();
            dev_put(ndev);
        }
        list_del(&ge->list);
        kfree(ge);
    } else
        pr_warn("could not find mgid entry\n");

    mutex_unlock(&mqp->mutex);

    return 0;
}

static int init_node_data(struct mlx4_ib_dev *dev)
{
    struct ib_smp *in_mad  = NULL;
    struct ib_smp *out_mad = NULL;
    int mad_ifc_flags = MLX4_MAD_IFC_IGNORE_KEYS;
    int err = -ENOMEM;

    in_mad  = kzalloc(sizeof *in_mad, GFP_KERNEL);
    out_mad = kmalloc(sizeof *out_mad, GFP_KERNEL);
    if (!in_mad || !out_mad)
        goto out;

    init_query_mad(in_mad);
    in_mad->attr_id = IB_SMP_ATTR_NODE_DESC;
    if (mlx4_is_master(dev->dev))
        mad_ifc_flags |= MLX4_MAD_IFC_NET_VIEW;

    err = mlx4_MAD_IFC(dev, mad_ifc_flags, 1, NULL, NULL, in_mad, out_mad);
    if (err)
        goto out;

    memcpy(dev->ib_dev.node_desc, out_mad->data, 64);

    in_mad->attr_id = IB_SMP_ATTR_NODE_INFO;

    err = mlx4_MAD_IFC(dev, mad_ifc_flags, 1, NULL, NULL, in_mad, out_mad);
    if (err)
        goto out;

    dev->dev->rev_id = be32_to_cpup((__be32 *) (out_mad->data + 32));
    memcpy(&dev->ib_dev.node_guid, out_mad->data + 12, 8);

out:
    kfree(in_mad);
    kfree(out_mad);
    return err;
}

static ssize_t show_hca(struct device *device, struct device_attribute *attr,
            char *buf)
{
    struct mlx4_ib_dev *dev =
        container_of(device, struct mlx4_ib_dev, ib_dev.dev);
    return sprintf(buf, "MT%d\n", dev->dev->pdev->device);
}

static ssize_t show_fw_ver(struct device *device, struct device_attribute *attr,
               char *buf)
{
    struct mlx4_ib_dev *dev =
        container_of(device, struct mlx4_ib_dev, ib_dev.dev);
    return sprintf(buf, "%d.%d.%d\n", (int) (dev->dev->caps.fw_ver >> 32),
               (int) (dev->dev->caps.fw_ver >> 16) & 0xffff,
               (int) dev->dev->caps.fw_ver & 0xffff);
}

static ssize_t show_rev(struct device *device, struct device_attribute *attr,
            char *buf)
{
    struct mlx4_ib_dev *dev =
        container_of(device, struct mlx4_ib_dev, ib_dev.dev);
    return sprintf(buf, "%x\n", dev->dev->rev_id);
}

static ssize_t show_board(struct device *device, struct device_attribute *attr,
              char *buf)
{
    struct mlx4_ib_dev *dev =
        container_of(device, struct mlx4_ib_dev, ib_dev.dev);
    return sprintf(buf, "%.*s\n", MLX4_BOARD_ID_LEN,
               dev->dev->board_id);
}

static DEVICE_ATTR(hw_rev,   S_IRUGO, show_rev,    NULL);
static DEVICE_ATTR(fw_ver,   S_IRUGO, show_fw_ver, NULL);
static DEVICE_ATTR(hca_type, S_IRUGO, show_hca,    NULL);
static DEVICE_ATTR(board_id, S_IRUGO, show_board,  NULL);

static struct device_attribute *mlx4_class_attributes[] = {
    &dev_attr_hw_rev,
    &dev_attr_fw_ver,
    &dev_attr_hca_type,
    &dev_attr_board_id
};

static void mlx4_addrconf_ifid_eui48(u8 *eui, u16 vlan_id, struct net_device *dev)
{
    memcpy(eui, dev->dev_addr, 3);
    memcpy(eui + 5, dev->dev_addr + 3, 3);
    if (vlan_id < 0x1000) {
        eui[3] = vlan_id >> 8;
        eui[4] = vlan_id & 0xff;
    } else {
        eui[3] = 0xff;
        eui[4] = 0xfe;
    }
    eui[0] ^= 2;
}

static void update_gids_task(struct work_struct *work)
{
    struct update_gid_work *gw = container_of(work, struct update_gid_work, work);
    struct mlx4_cmd_mailbox *mailbox;
    union ib_gid *gids;
    int err;
    struct mlx4_dev *dev = gw->dev->dev;

    mailbox = mlx4_alloc_cmd_mailbox(dev);
    if (IS_ERR(mailbox)) {
        pr_warn("update gid table failed %ld\n", PTR_ERR(mailbox));
        return;
    }

    gids = mailbox->buf;
    memcpy(gids, gw->gids, sizeof gw->gids);

    err = mlx4_cmd(dev, mailbox->dma, MLX4_SET_PORT_GID_TABLE << 8 | gw->port,
               1, MLX4_CMD_SET_PORT, MLX4_CMD_TIME_CLASS_B,
               MLX4_CMD_WRAPPED);
    if (err)
        pr_warn("set port command failed\n");
    else {
        memcpy(gw->dev->iboe.gid_table[gw->port - 1], gw->gids, sizeof gw->gids);
        mlx4_ib_dispatch_event(gw->dev, gw->port, IB_EVENT_GID_CHANGE);
    }

    mlx4_free_cmd_mailbox(dev, mailbox);
    kfree(gw);
}

static int update_ipv6_gids(struct mlx4_ib_dev *dev, int port, int clear)
{
    struct net_device *ndev = dev->iboe.netdevs[port - 1];
    struct update_gid_work *work;
    struct net_device *tmp;
    int i;
    u8 *hits;
    int ret;
    union ib_gid gid;
    int free;
    int found;
    int need_update = 0;
    u16 vid;

    work = kzalloc(sizeof *work, GFP_ATOMIC);
    if (!work)
        return -ENOMEM;

    hits = kzalloc(128, GFP_ATOMIC);
    if (!hits) {
        ret = -ENOMEM;
        goto out;
    }

    rcu_read_lock();
    for_each_netdev_rcu(&init_net, tmp) {
        if (ndev && (tmp == ndev || rdma_vlan_dev_real_dev(tmp) == ndev)) {
            gid.global.subnet_prefix = cpu_to_be64(0xfe80000000000000LL);
            vid = rdma_vlan_dev_vlan_id(tmp);
            mlx4_addrconf_ifid_eui48(&gid.raw[8], vid, ndev);
            found = 0;
            free = -1;
            for (i = 0; i < 128; ++i) {
                if (free < 0 &&
                    !memcmp(&dev->iboe.gid_table[port - 1][i], &zgid, sizeof zgid))
                    free = i;
                if (!memcmp(&dev->iboe.gid_table[port - 1][i], &gid, sizeof gid)) {
                    hits[i] = 1;
                    found = 1;
                    break;
                }
            }

            if (!found) {
                if (tmp == ndev &&
                    (memcmp(&dev->iboe.gid_table[port - 1][0],
                        &gid, sizeof gid) ||
                     !memcmp(&dev->iboe.gid_table[port - 1][0],
                         &zgid, sizeof gid))) {
                    dev->iboe.gid_table[port - 1][0] = gid;
                    ++need_update;
                    hits[0] = 1;
                } else if (free >= 0) {
                    dev->iboe.gid_table[port - 1][free] = gid;
                    hits[free] = 1;
                    ++need_update;
                }
            }
        }
    }
    rcu_read_unlock();

    for (i = 0; i < 128; ++i)
        if (!hits[i]) {
            if (memcmp(&dev->iboe.gid_table[port - 1][i], &zgid, sizeof zgid))
                ++need_update;
            dev->iboe.gid_table[port - 1][i] = zgid;
        }

    if (need_update) {
        memcpy(work->gids, dev->iboe.gid_table[port - 1], sizeof work->gids);
        INIT_WORK(&work->work, update_gids_task);
        work->port = port;
        work->dev = dev;
        queue_work(wq, &work->work);
    } else
        kfree(work);

    kfree(hits);
    return 0;

out:
    kfree(work);
    return ret;
}

static void handle_en_event(struct mlx4_ib_dev *dev, int port, unsigned long event)
{
    switch (event) {
    case NETDEV_UP:
    case NETDEV_CHANGEADDR:
        update_ipv6_gids(dev, port, 0);
        break;

    case NETDEV_DOWN:
        update_ipv6_gids(dev, port, 1);
        dev->iboe.netdevs[port - 1] = NULL;
    }
}

static void netdev_added(struct mlx4_ib_dev *dev, int port)
{
    update_ipv6_gids(dev, port, 0);
}

static void netdev_removed(struct mlx4_ib_dev *dev, int port)
{
    update_ipv6_gids(dev, port, 1);
}

static int mlx4_ib_netdev_event(struct notifier_block *this, unsigned long event,
                void *ptr)
{
    struct net_device *dev = ptr;
    struct mlx4_ib_dev *ibdev;
    struct net_device *oldnd;
    struct mlx4_ib_iboe *iboe;
    int port;

    if (!net_eq(dev_net(dev), &init_net))
        return NOTIFY_DONE;

    ibdev = container_of(this, struct mlx4_ib_dev, iboe.nb);
    iboe = &ibdev->iboe;

    spin_lock(&iboe->lock);
    mlx4_foreach_ib_transport_port(port, ibdev->dev) {
        oldnd = iboe->netdevs[port - 1];
        iboe->netdevs[port - 1] =
            mlx4_get_protocol_dev(ibdev->dev, MLX4_PROT_ETH, port);
        if (oldnd != iboe->netdevs[port - 1]) {
            if (iboe->netdevs[port - 1])
                netdev_added(ibdev, port);
            else
                netdev_removed(ibdev, port);
        }
    }

    if (dev == iboe->netdevs[0] ||
        (iboe->netdevs[0] && rdma_vlan_dev_real_dev(dev) == iboe->netdevs[0]))
        handle_en_event(ibdev, 1, event);
    else if (dev == iboe->netdevs[1]
         || (iboe->netdevs[1] && rdma_vlan_dev_real_dev(dev) == iboe->netdevs[1]))
        handle_en_event(ibdev, 2, event);

    spin_unlock(&iboe->lock);

    return NOTIFY_DONE;
}

static void init_pkeys(struct mlx4_ib_dev *ibdev)
{
    int port;
    int slave;
    int i;

    if (mlx4_is_master(ibdev->dev)) {
        for (slave = 0; slave <= ibdev->dev->num_vfs; ++slave) {
            for (port = 1; port <= ibdev->dev->caps.num_ports; ++port) {
                for (i = 0;
                     i < ibdev->dev->phys_caps.pkey_phys_table_len[port];
                     ++i) {
                    ibdev->pkeys.virt2phys_pkey[slave][port - 1][i] =
                    /* master has the identity virt2phys pkey mapping */
                        (slave == mlx4_master_func_num(ibdev->dev) || !i) ? i :
                            ibdev->dev->phys_caps.pkey_phys_table_len[port] - 1;
                    mlx4_sync_pkey_table(ibdev->dev, slave, port, i,
                                 ibdev->pkeys.virt2phys_pkey[slave][port - 1][i]);
                }
            }
        }
        /* initialize pkey cache */
        for (port = 1; port <= ibdev->dev->caps.num_ports; ++port) {
            for (i = 0;
                 i < ibdev->dev->phys_caps.pkey_phys_table_len[port];
                 ++i)
                ibdev->pkeys.phys_pkey_cache[port-1][i] =
                    (i) ? 0 : 0xFFFF;
        }
    }
}

static void mlx4_ib_alloc_eqs(struct mlx4_dev *dev, struct mlx4_ib_dev *ibdev)
{
    char name[32];
    int eq_per_port = 0;
    int added_eqs = 0;
    int total_eqs = 0;
    int i, j, eq;

    /* Legacy mode or comp_pool is not large enough */
    if (dev->caps.comp_pool == 0 ||
        dev->caps.num_ports > dev->caps.comp_pool)
        return;

    eq_per_port = rounddown_pow_of_two(dev->caps.comp_pool/
                    dev->caps.num_ports);

    /* Init eq table */
    added_eqs = 0;
    mlx4_foreach_port(i, dev, MLX4_PORT_TYPE_IB)
        added_eqs += eq_per_port;

    total_eqs = dev->caps.num_comp_vectors + added_eqs;

    ibdev->eq_table = kzalloc(total_eqs * sizeof(int), GFP_KERNEL);
    if (!ibdev->eq_table)
        return;

    ibdev->eq_added = added_eqs;

    eq = 0;
    mlx4_foreach_port(i, dev, MLX4_PORT_TYPE_IB) {
        for (j = 0; j < eq_per_port; j++) {
            sprintf(name, "mlx4-ib-%d-%d@%s",
                i, j, dev->pdev->bus->name);
            /* Set IRQ for specific name (per ring) */
            if (mlx4_assign_eq(dev, name, NULL,
                       &ibdev->eq_table[eq])) {
                /* Use legacy (same as mlx4_en driver) */
                pr_warn("Can't allocate EQ %d; reverting to legacy\n", eq);
                ibdev->eq_table[eq] =
                    (eq % dev->caps.num_comp_vectors);
            }
            eq++;
        }
    }

    /* Fill the reset of the vector with legacy EQ */
    for (i = 0, eq = added_eqs; i < dev->caps.num_comp_vectors; i++)
        ibdev->eq_table[eq++] = i;

    /* Advertise the new number of EQs to clients */
    ibdev->ib_dev.num_comp_vectors = total_eqs;
}

static void mlx4_ib_free_eqs(struct mlx4_dev *dev, struct mlx4_ib_dev *ibdev)
{
    int i;

    /* no additional eqs were added */
    if (!ibdev->eq_table)
        return;

    /* Reset the advertised EQ number */
    ibdev->ib_dev.num_comp_vectors = dev->caps.num_comp_vectors;

    /* Free only the added eqs */
    for (i = 0; i < ibdev->eq_added; i++) {
        /* Don't free legacy eqs if used */
        if (ibdev->eq_table[i] <= dev->caps.num_comp_vectors)
            continue;
        mlx4_release_eq(dev, ibdev->eq_table[i]);
    }

    kfree(ibdev->eq_table);
}

static void *mlx4_ib_add(struct mlx4_dev *dev)
{
    struct mlx4_ib_dev *ibdev;
    int num_ports = 0;
    int i, j;
    int err;
    struct mlx4_ib_iboe *iboe;

    pr_info_once("%s", mlx4_ib_version);

    mlx4_foreach_non_ib_transport_port(i, dev)
        num_ports++;

    if (mlx4_is_mfunc(dev) && num_ports) {
        dev_err(&dev->pdev->dev, "RoCE is not supported over SRIOV as yet\n");
        return NULL;
    }

    num_ports = 0;
    mlx4_foreach_ib_transport_port(i, dev)
        num_ports++;

    /* No point in registering a device with no ports... */
    if (num_ports == 0)
        return NULL;

    ibdev = (struct mlx4_ib_dev *) ib_alloc_device(sizeof *ibdev);
    if (!ibdev) {
        dev_err(&dev->pdev->dev, "Device struct alloc failed\n");
        return NULL;
    }

    iboe = &ibdev->iboe;

    if (mlx4_pd_alloc(dev, &ibdev->priv_pdn))
        goto err_dealloc;

    if (mlx4_uar_alloc(dev, &ibdev->priv_uar))
        goto err_pd;

    ibdev->uar_map = ioremap((phys_addr_t) ibdev->priv_uar.pfn << PAGE_SHIFT,
                 PAGE_SIZE);
    if (!ibdev->uar_map)
        goto err_uar;
    MLX4_INIT_DOORBELL_LOCK(&ibdev->uar_lock);

    ibdev->dev = dev;

    strlcpy(ibdev->ib_dev.name, "mlx4_%d", IB_DEVICE_NAME_MAX);
    ibdev->ib_dev.owner     = THIS_MODULE;
    ibdev->ib_dev.node_type     = RDMA_NODE_IB_CA;
    ibdev->ib_dev.local_dma_lkey    = dev->caps.reserved_lkey;
    ibdev->num_ports        = num_ports;
    ibdev->ib_dev.phys_port_cnt     = ibdev->num_ports;
    ibdev->ib_dev.num_comp_vectors  = dev->caps.num_comp_vectors;
    ibdev->ib_dev.dma_device    = &dev->pdev->dev;

    ibdev->ib_dev.uverbs_abi_ver    = MLX4_IB_UVERBS_ABI_VERSION;
    ibdev->ib_dev.uverbs_cmd_mask   =
        (1ull << IB_USER_VERBS_CMD_GET_CONTEXT)     |
        (1ull << IB_USER_VERBS_CMD_QUERY_DEVICE)    |
        (1ull << IB_USER_VERBS_CMD_QUERY_PORT)      |
        (1ull << IB_USER_VERBS_CMD_ALLOC_PD)        |
        (1ull << IB_USER_VERBS_CMD_DEALLOC_PD)      |
        (1ull << IB_USER_VERBS_CMD_REG_MR)      |
        (1ull << IB_USER_VERBS_CMD_DEREG_MR)        |
        (1ull << IB_USER_VERBS_CMD_CREATE_COMP_CHANNEL) |
        (1ull << IB_USER_VERBS_CMD_CREATE_CQ)       |
        (1ull << IB_USER_VERBS_CMD_RESIZE_CQ)       |
        (1ull << IB_USER_VERBS_CMD_DESTROY_CQ)      |
        (1ull << IB_USER_VERBS_CMD_CREATE_QP)       |
        (1ull << IB_USER_VERBS_CMD_MODIFY_QP)       |
        (1ull << IB_USER_VERBS_CMD_QUERY_QP)        |
        (1ull << IB_USER_VERBS_CMD_DESTROY_QP)      |
        (1ull << IB_USER_VERBS_CMD_ATTACH_MCAST)    |
        (1ull << IB_USER_VERBS_CMD_DETACH_MCAST)    |
        (1ull << IB_USER_VERBS_CMD_CREATE_SRQ)      |
        (1ull << IB_USER_VERBS_CMD_MODIFY_SRQ)      |
        (1ull << IB_USER_VERBS_CMD_QUERY_SRQ)       |
        (1ull << IB_USER_VERBS_CMD_DESTROY_SRQ)     |
        (1ull << IB_USER_VERBS_CMD_CREATE_XSRQ)     |
        (1ull << IB_USER_VERBS_CMD_OPEN_QP);

    ibdev->ib_dev.query_device  = mlx4_ib_query_device;
    ibdev->ib_dev.query_port    = mlx4_ib_query_port;
    ibdev->ib_dev.get_link_layer    = mlx4_ib_port_link_layer;
    ibdev->ib_dev.query_gid     = mlx4_ib_query_gid;
    ibdev->ib_dev.query_pkey    = mlx4_ib_query_pkey;
    ibdev->ib_dev.modify_device = mlx4_ib_modify_device;
    ibdev->ib_dev.modify_port   = mlx4_ib_modify_port;
    ibdev->ib_dev.alloc_ucontext    = mlx4_ib_alloc_ucontext;
    ibdev->ib_dev.dealloc_ucontext  = mlx4_ib_dealloc_ucontext;
    ibdev->ib_dev.mmap      = mlx4_ib_mmap;
    ibdev->ib_dev.alloc_pd      = mlx4_ib_alloc_pd;
    ibdev->ib_dev.dealloc_pd    = mlx4_ib_dealloc_pd;
    ibdev->ib_dev.create_ah     = mlx4_ib_create_ah;
    ibdev->ib_dev.query_ah      = mlx4_ib_query_ah;
    ibdev->ib_dev.destroy_ah    = mlx4_ib_destroy_ah;
    ibdev->ib_dev.create_srq    = mlx4_ib_create_srq;
    ibdev->ib_dev.modify_srq    = mlx4_ib_modify_srq;
    ibdev->ib_dev.query_srq     = mlx4_ib_query_srq;
    ibdev->ib_dev.destroy_srq   = mlx4_ib_destroy_srq;
    ibdev->ib_dev.post_srq_recv = mlx4_ib_post_srq_recv;
    ibdev->ib_dev.create_qp     = mlx4_ib_create_qp;
    ibdev->ib_dev.modify_qp     = mlx4_ib_modify_qp;
    ibdev->ib_dev.query_qp      = mlx4_ib_query_qp;
    ibdev->ib_dev.destroy_qp    = mlx4_ib_destroy_qp;
    ibdev->ib_dev.post_send     = mlx4_ib_post_send;
    ibdev->ib_dev.post_recv     = mlx4_ib_post_recv;
    ibdev->ib_dev.create_cq     = mlx4_ib_create_cq;
    ibdev->ib_dev.modify_cq     = mlx4_ib_modify_cq;
    ibdev->ib_dev.resize_cq     = mlx4_ib_resize_cq;
    ibdev->ib_dev.destroy_cq    = mlx4_ib_destroy_cq;
    ibdev->ib_dev.poll_cq       = mlx4_ib_poll_cq;
    ibdev->ib_dev.req_notify_cq = mlx4_ib_arm_cq;
    ibdev->ib_dev.get_dma_mr    = mlx4_ib_get_dma_mr;
    ibdev->ib_dev.reg_user_mr   = mlx4_ib_reg_user_mr;
    ibdev->ib_dev.dereg_mr      = mlx4_ib_dereg_mr;
    ibdev->ib_dev.alloc_fast_reg_mr = mlx4_ib_alloc_fast_reg_mr;
    ibdev->ib_dev.alloc_fast_reg_page_list = mlx4_ib_alloc_fast_reg_page_list;
    ibdev->ib_dev.free_fast_reg_page_list  = mlx4_ib_free_fast_reg_page_list;
    ibdev->ib_dev.attach_mcast  = mlx4_ib_mcg_attach;
    ibdev->ib_dev.detach_mcast  = mlx4_ib_mcg_detach;
    ibdev->ib_dev.process_mad   = mlx4_ib_process_mad;

    if (!mlx4_is_slave(ibdev->dev)) {
        ibdev->ib_dev.alloc_fmr     = mlx4_ib_fmr_alloc;
        ibdev->ib_dev.map_phys_fmr  = mlx4_ib_map_phys_fmr;
        ibdev->ib_dev.unmap_fmr     = mlx4_ib_unmap_fmr;
        ibdev->ib_dev.dealloc_fmr   = mlx4_ib_fmr_dealloc;
    }

    if (dev->caps.flags & MLX4_DEV_CAP_FLAG_XRC) {
        ibdev->ib_dev.alloc_xrcd = mlx4_ib_alloc_xrcd;
        ibdev->ib_dev.dealloc_xrcd = mlx4_ib_dealloc_xrcd;
        ibdev->ib_dev.uverbs_cmd_mask |=
            (1ull << IB_USER_VERBS_CMD_OPEN_XRCD) |
            (1ull << IB_USER_VERBS_CMD_CLOSE_XRCD);
    }

    mlx4_ib_alloc_eqs(dev, ibdev);

    spin_lock_init(&iboe->lock);

    if (init_node_data(ibdev))
        goto err_map;

    for (i = 0; i < ibdev->num_ports; ++i) {
        if (mlx4_ib_port_link_layer(&ibdev->ib_dev, i + 1) ==
                        IB_LINK_LAYER_ETHERNET) {
            err = mlx4_counter_alloc(ibdev->dev, &ibdev->counters[i]);
            if (err)
                ibdev->counters[i] = -1;
        } else
                ibdev->counters[i] = -1;
    }

    spin_lock_init(&ibdev->sm_lock);
    mutex_init(&ibdev->cap_mask_mutex);

    if (ib_register_device(&ibdev->ib_dev, NULL))
        goto err_counter;

    if (mlx4_ib_mad_init(ibdev))
        goto err_reg;

    if (mlx4_ib_init_sriov(ibdev))
        goto err_mad;

    if (dev->caps.flags & MLX4_DEV_CAP_FLAG_IBOE && !iboe->nb.notifier_call) {
        iboe->nb.notifier_call = mlx4_ib_netdev_event;
        err = register_netdevice_notifier(&iboe->nb);
        if (err)
            goto err_sriov;
    }

    for (j = 0; j < ARRAY_SIZE(mlx4_class_attributes); ++j) {
        if (device_create_file(&ibdev->ib_dev.dev,
                       mlx4_class_attributes[j]))
            goto err_notif;
    }

    ibdev->ib_active = true;

    if (mlx4_is_mfunc(ibdev->dev))
        init_pkeys(ibdev);

    /* create paravirt contexts for any VFs which are active */
    if (mlx4_is_master(ibdev->dev)) {
        for (j = 0; j < MLX4_MFUNC_MAX; j++) {
            if (j == mlx4_master_func_num(ibdev->dev))
                continue;
            if (mlx4_is_slave_active(ibdev->dev, j))
                do_slave_init(ibdev, j, 1);
        }
    }
    return ibdev;

err_notif:
    if (unregister_netdevice_notifier(&ibdev->iboe.nb))
        pr_warn("failure unregistering notifier\n");
    flush_workqueue(wq);

err_sriov:
    mlx4_ib_close_sriov(ibdev);

err_mad:
    mlx4_ib_mad_cleanup(ibdev);

err_reg:
    ib_unregister_device(&ibdev->ib_dev);

err_counter:
    for (; i; --i)
        if (ibdev->counters[i - 1] != -1)
            mlx4_counter_free(ibdev->dev, ibdev->counters[i - 1]);

err_map:
    iounmap(ibdev->uar_map);

err_uar:
    mlx4_uar_free(dev, &ibdev->priv_uar);

err_pd:
    mlx4_pd_free(dev, ibdev->priv_pdn);

err_dealloc:
    ib_dealloc_device(&ibdev->ib_dev);

    return NULL;
}

static void mlx4_ib_remove(struct mlx4_dev *dev, void *ibdev_ptr)
{
    struct mlx4_ib_dev *ibdev = ibdev_ptr;
    int p;

    mlx4_ib_close_sriov(ibdev);
    mlx4_ib_mad_cleanup(ibdev);
    ib_unregister_device(&ibdev->ib_dev);
    if (ibdev->iboe.nb.notifier_call) {
        if (unregister_netdevice_notifier(&ibdev->iboe.nb))
            pr_warn("failure unregistering notifier\n");
        ibdev->iboe.nb.notifier_call = NULL;
    }
    iounmap(ibdev->uar_map);
    for (p = 0; p < ibdev->num_ports; ++p)
        if (ibdev->counters[p] != -1)
            mlx4_counter_free(ibdev->dev, ibdev->counters[p]);
    mlx4_foreach_port(p, dev, MLX4_PORT_TYPE_IB)
        mlx4_CLOSE_PORT(dev, p);

    mlx4_ib_free_eqs(dev, ibdev);

    mlx4_uar_free(dev, &ibdev->priv_uar);
    mlx4_pd_free(dev, ibdev->priv_pdn);
    ib_dealloc_device(&ibdev->ib_dev);
}

static void do_slave_init(struct mlx4_ib_dev *ibdev, int slave, int do_init)
{
    struct mlx4_ib_demux_work **dm = NULL;
    struct mlx4_dev *dev = ibdev->dev;
    int i;
    unsigned long flags;

    if (!mlx4_is_master(dev))
        return;

    dm = kcalloc(dev->caps.num_ports, sizeof *dm, GFP_ATOMIC);
    if (!dm) {
        pr_err("failed to allocate memory for tunneling qp update\n");
        goto out;
    }

    for (i = 0; i < dev->caps.num_ports; i++) {
        dm[i] = kmalloc(sizeof (struct mlx4_ib_demux_work), GFP_ATOMIC);
        if (!dm[i]) {
            pr_err("failed to allocate memory for tunneling qp update work struct\n");
            for (i = 0; i < dev->caps.num_ports; i++) {
                if (dm[i])
                    kfree(dm[i]);
            }
            goto out;
        }
    }
    /* initialize or tear down tunnel QPs for the slave */
    for (i = 0; i < dev->caps.num_ports; i++) {
        INIT_WORK(&dm[i]->work, mlx4_ib_tunnels_update_work);
        dm[i]->port = i + 1;
        dm[i]->slave = slave;
        dm[i]->do_init = do_init;
        dm[i]->dev = ibdev;
        spin_lock_irqsave(&ibdev->sriov.going_down_lock, flags);
        if (!ibdev->sriov.is_going_down)
            queue_work(ibdev->sriov.demux[i].ud_wq, &dm[i]->work);
        spin_unlock_irqrestore(&ibdev->sriov.going_down_lock, flags);
    }
out:
    if (dm)
        kfree(dm);
    return;
}

static void mlx4_ib_event(struct mlx4_dev *dev, void *ibdev_ptr,
              enum mlx4_dev_event event, unsigned long param)
{
    struct ib_event ibev;
    struct mlx4_ib_dev *ibdev = to_mdev((struct ib_device *) ibdev_ptr);
    struct mlx4_eqe *eqe = NULL;
    struct ib_event_work *ew;
    int p = 0;

    if (event == MLX4_DEV_EVENT_PORT_MGMT_CHANGE)
        eqe = (struct mlx4_eqe *)param;
    else
        p = (int) param;

    switch (event) {
    case MLX4_DEV_EVENT_PORT_UP:
        if (p > ibdev->num_ports)
            return;
        if (mlx4_is_master(dev) &&
            rdma_port_get_link_layer(&ibdev->ib_dev, p) ==
            IB_LINK_LAYER_INFINIBAND) {
            mlx4_ib_invalidate_all_guid_record(ibdev, p);
        }
        ibev.event = IB_EVENT_PORT_ACTIVE;
        break;

    case MLX4_DEV_EVENT_PORT_DOWN:
        if (p > ibdev->num_ports)
            return;
        ibev.event = IB_EVENT_PORT_ERR;
        break;

    case MLX4_DEV_EVENT_CATASTROPHIC_ERROR:
        ibdev->ib_active = false;
        ibev.event = IB_EVENT_DEVICE_FATAL;
        break;

    case MLX4_DEV_EVENT_PORT_MGMT_CHANGE:
        ew = kmalloc(sizeof *ew, GFP_ATOMIC);
        if (!ew) {
            pr_err("failed to allocate memory for events work\n");
            break;
        }

        INIT_WORK(&ew->work, handle_port_mgmt_change_event);
        memcpy(&ew->ib_eqe, eqe, sizeof *eqe);
        ew->ib_dev = ibdev;
        /* need to queue only for port owner, which uses GEN_EQE */
        if (mlx4_is_master(dev))
            queue_work(wq, &ew->work);
        else
            handle_port_mgmt_change_event(&ew->work);
        return;

    case MLX4_DEV_EVENT_SLAVE_INIT:
        /* here, p is the slave id */
        do_slave_init(ibdev, p, 1);
        return;

    case MLX4_DEV_EVENT_SLAVE_SHUTDOWN:
        /* here, p is the slave id */
        do_slave_init(ibdev, p, 0);
        return;

    default:
        return;
    }

    ibev.device       = ibdev_ptr;
    ibev.element.port_num = (u8) p;

    ib_dispatch_event(&ibev);
}

static struct mlx4_interface mlx4_ib_interface = {
    .add        = mlx4_ib_add,
    .remove     = mlx4_ib_remove,
    .event      = mlx4_ib_event,
    .protocol   = MLX4_PROT_IB_IPV6
};

static int __init mlx4_ib_init(void)
{
    int err;

    wq = create_singlethread_workqueue("mlx4_ib");
    if (!wq)
        return -ENOMEM;

    err = mlx4_ib_mcg_init();
    if (err)
        goto clean_wq;

    err = mlx4_register_interface(&mlx4_ib_interface);
    if (err)
        goto clean_mcg;

    return 0;

clean_mcg:
    mlx4_ib_mcg_destroy();

clean_wq:
    destroy_workqueue(wq);
    return err;
}

static void __exit mlx4_ib_cleanup(void)
{
    mlx4_unregister_interface(&mlx4_ib_interface);
    mlx4_ib_mcg_destroy();
    destroy_workqueue(wq);
}

module_init(mlx4_ib_init);
module_exit(mlx4_ib_cleanup);