/*
 * Copyright (c) 2006 - 2009 Intel-NE, Inc.  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/moduleparam.h>
#include <linux/random.h>
#include <linux/highmem.h>
#include <asm/byteorder.h>

#include <rdma/ib_verbs.h>
#include <rdma/iw_cm.h>
#include <rdma/ib_user_verbs.h>

#include "nes.h"

#include <rdma/ib_umem.h>

atomic_t mod_qp_timouts;
atomic_t qps_created;
atomic_t sw_qps_destroyed;

static void nes_unregister_ofa_device(struct nes_ib_device *nesibdev);

/**
 * nes_alloc_mw
 */
static struct ib_mw *nes_alloc_mw(struct ib_pd *ibpd) {
        struct nes_pd *nespd = to_nespd(ibpd);
        struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_cqp_request *cqp_request;
        struct nes_mr *nesmr;
        struct ib_mw *ibmw;
        struct nes_hw_cqp_wqe *cqp_wqe;
        int ret;
        u32 stag;
        u32 stag_index = 0;
        u32 next_stag_index = 0;
        u32 driver_key = 0;
        u8 stag_key = 0;

        get_random_bytes(&next_stag_index, sizeof(next_stag_index));
        stag_key = (u8)next_stag_index;

        driver_key = 0;

        next_stag_index >>= 8;
        next_stag_index %= nesadapter->max_mr;

        ret = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs,
                        nesadapter->max_mr, &stag_index, &next_stag_index);
        if (ret) {
                return ERR_PTR(ret);
        }

        nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
        if (!nesmr) {
                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                return ERR_PTR(-ENOMEM);
        }

        stag = stag_index << 8;
        stag |= driver_key;
        stag += (u32)stag_key;

        nes_debug(NES_DBG_MR, "Registering STag 0x%08X, index = 0x%08X\n",
                        stag, stag_index);

        /* Register the region with the adapter */
        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                kfree(nesmr);
                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                return ERR_PTR(-ENOMEM);
        }

        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;

        cqp_wqe->wqe_words[NES_CQP_WQE_OPCODE_IDX] =
                        cpu_to_le32( NES_CQP_ALLOCATE_STAG | NES_CQP_STAG_RIGHTS_REMOTE_READ |
                        NES_CQP_STAG_RIGHTS_REMOTE_WRITE | NES_CQP_STAG_VA_TO |
                        NES_CQP_STAG_REM_ACC_EN);

        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX, (nespd->pd_id & 0x00007fff));
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, stag);

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
                        NES_EVENT_TIMEOUT);
        nes_debug(NES_DBG_MR, "Register STag 0x%08X completed, wait_event_timeout ret = %u,"
                        " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                        stag, ret, cqp_request->major_code, cqp_request->minor_code);
        if ((!ret) || (cqp_request->major_code)) {
                nes_put_cqp_request(nesdev, cqp_request);
                kfree(nesmr);
                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                if (!ret) {
                        return ERR_PTR(-ETIME);
                } else {
                        return ERR_PTR(-ENOMEM);
                }
        }
        nes_put_cqp_request(nesdev, cqp_request);

        nesmr->ibmw.rkey = stag;
        nesmr->mode = IWNES_MEMREG_TYPE_MW;
        ibmw = &nesmr->ibmw;
        nesmr->pbl_4k = 0;
        nesmr->pbls_used = 0;

        return ibmw;
}


/**
 * nes_dealloc_mw
 */
static int nes_dealloc_mw(struct ib_mw *ibmw)
{
        struct nes_mr *nesmr = to_nesmw(ibmw);
        struct nes_vnic *nesvnic = to_nesvnic(ibmw->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_hw_cqp_wqe *cqp_wqe;
        struct nes_cqp_request *cqp_request;
        int err = 0;
        int ret;

        /* Deallocate the window with the adapter */
        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
                return -ENOMEM;
        }
        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;
        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, NES_CQP_DEALLOCATE_STAG);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, ibmw->rkey);

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        nes_debug(NES_DBG_MR, "Waiting for deallocate STag 0x%08X to complete.\n",
                        ibmw->rkey);
        ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
                        NES_EVENT_TIMEOUT);
        nes_debug(NES_DBG_MR, "Deallocate STag completed, wait_event_timeout ret = %u,"
                        " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                        ret, cqp_request->major_code, cqp_request->minor_code);
        if (!ret)
                err = -ETIME;
        else if (cqp_request->major_code)
                err = -EIO;

        nes_put_cqp_request(nesdev, cqp_request);

        nes_free_resource(nesadapter, nesadapter->allocated_mrs,
                        (ibmw->rkey & 0x0fffff00) >> 8);
        kfree(nesmr);

        return err;
}


/**
 * nes_bind_mw
 */
static int nes_bind_mw(struct ib_qp *ibqp, struct ib_mw *ibmw,
                struct ib_mw_bind *ibmw_bind)
{
        u64 u64temp;
        struct nes_vnic *nesvnic = to_nesvnic(ibqp->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        /* struct nes_mr *nesmr = to_nesmw(ibmw); */
        struct nes_qp *nesqp = to_nesqp(ibqp);
        struct nes_hw_qp_wqe *wqe;
        unsigned long flags = 0;
        u32 head;
        u32 wqe_misc = 0;
        u32 qsize;

        if (nesqp->ibqp_state > IB_QPS_RTS)
                return -EINVAL;

        spin_lock_irqsave(&nesqp->lock, flags);

        head = nesqp->hwqp.sq_head;
        qsize = nesqp->hwqp.sq_tail;

        /* Check for SQ overflow */
        if (((head + (2 * qsize) - nesqp->hwqp.sq_tail) % qsize) == (qsize - 1)) {
                spin_unlock_irqrestore(&nesqp->lock, flags);
                return -EINVAL;
        }

        wqe = &nesqp->hwqp.sq_vbase[head];
        /* nes_debug(NES_DBG_MR, "processing sq wqe at %p, head = %u.\n", wqe, head); */
        nes_fill_init_qp_wqe(wqe, nesqp, head);
        u64temp = ibmw_bind->wr_id;
        set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_COMP_SCRATCH_LOW_IDX, u64temp);
        wqe_misc = NES_IWARP_SQ_OP_BIND;

        wqe_misc |= NES_IWARP_SQ_WQE_LOCAL_FENCE;

        if (ibmw_bind->send_flags & IB_SEND_SIGNALED)
                wqe_misc |= NES_IWARP_SQ_WQE_SIGNALED_COMPL;

        if (ibmw_bind->mw_access_flags & IB_ACCESS_REMOTE_WRITE) {
                wqe_misc |= NES_CQP_STAG_RIGHTS_REMOTE_WRITE;
        }
        if (ibmw_bind->mw_access_flags & IB_ACCESS_REMOTE_READ) {
                wqe_misc |= NES_CQP_STAG_RIGHTS_REMOTE_READ;
        }

        set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_MISC_IDX, wqe_misc);
        set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_MR_IDX, ibmw_bind->mr->lkey);
        set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_MW_IDX, ibmw->rkey);
        set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_LENGTH_LOW_IDX,
                        ibmw_bind->length);
        wqe->wqe_words[NES_IWARP_SQ_BIND_WQE_LENGTH_HIGH_IDX] = 0;
        u64temp = (u64)ibmw_bind->addr;
        set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_VA_FBO_LOW_IDX, u64temp);

        head++;
        if (head >= qsize)
                head = 0;

        nesqp->hwqp.sq_head = head;
        barrier();

        nes_write32(nesdev->regs+NES_WQE_ALLOC,
                        (1 << 24) | 0x00800000 | nesqp->hwqp.qp_id);

        spin_unlock_irqrestore(&nesqp->lock, flags);

        return 0;
}


/**
 * nes_alloc_fmr
 */
static struct ib_fmr *nes_alloc_fmr(struct ib_pd *ibpd,
                int ibmr_access_flags,
                struct ib_fmr_attr *ibfmr_attr)
{
        unsigned long flags;
        struct nes_pd *nespd = to_nespd(ibpd);
        struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_fmr *nesfmr;
        struct nes_cqp_request *cqp_request;
        struct nes_hw_cqp_wqe *cqp_wqe;
        int ret;
        u32 stag;
        u32 stag_index = 0;
        u32 next_stag_index = 0;
        u32 driver_key = 0;
        u32 opcode = 0;
        u8 stag_key = 0;
        int i=0;
        struct nes_vpbl vpbl;

        get_random_bytes(&next_stag_index, sizeof(next_stag_index));
        stag_key = (u8)next_stag_index;

        driver_key = 0;

        next_stag_index >>= 8;
        next_stag_index %= nesadapter->max_mr;

        ret = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs,
                        nesadapter->max_mr, &stag_index, &next_stag_index);
        if (ret) {
                goto failed_resource_alloc;
        }

        nesfmr = kzalloc(sizeof(*nesfmr), GFP_KERNEL);
        if (!nesfmr) {
                ret = -ENOMEM;
                goto failed_fmr_alloc;
        }

        nesfmr->nesmr.mode = IWNES_MEMREG_TYPE_FMR;
        if (ibfmr_attr->max_pages == 1) {
                /* use zero length PBL */
                nesfmr->nesmr.pbl_4k = 0;
                nesfmr->nesmr.pbls_used = 0;
        } else if (ibfmr_attr->max_pages <= 32) {
                /* use PBL 256 */
                nesfmr->nesmr.pbl_4k = 0;
                nesfmr->nesmr.pbls_used = 1;
        } else if (ibfmr_attr->max_pages <= 512) {
                /* use 4K PBLs */
                nesfmr->nesmr.pbl_4k = 1;
                nesfmr->nesmr.pbls_used = 1;
        } else {
                /* use two level 4K PBLs */
                /* add support for two level 256B PBLs */
                nesfmr->nesmr.pbl_4k = 1;
                nesfmr->nesmr.pbls_used = 1 + (ibfmr_attr->max_pages >> 9) +
                                ((ibfmr_attr->max_pages & 511) ? 1 : 0);
        }
        /* Register the region with the adapter */
        spin_lock_irqsave(&nesadapter->pbl_lock, flags);

        /* track PBL resources */
        if (nesfmr->nesmr.pbls_used != 0) {
                if (nesfmr->nesmr.pbl_4k) {
                        if (nesfmr->nesmr.pbls_used > nesadapter->free_4kpbl) {
                                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                                ret = -ENOMEM;
                                goto failed_vpbl_avail;
                        } else {
                                nesadapter->free_4kpbl -= nesfmr->nesmr.pbls_used;
                        }
                } else {
                        if (nesfmr->nesmr.pbls_used > nesadapter->free_256pbl) {
                                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                                ret = -ENOMEM;
                                goto failed_vpbl_avail;
                        } else {
                                nesadapter->free_256pbl -= nesfmr->nesmr.pbls_used;
                        }
                }
        }

        /* one level pbl */
        if (nesfmr->nesmr.pbls_used == 0) {
                nesfmr->root_vpbl.pbl_vbase = NULL;
                nes_debug(NES_DBG_MR,  "zero level pbl \n");
        } else if (nesfmr->nesmr.pbls_used == 1) {
                /* can change it to kmalloc & dma_map_single */
                nesfmr->root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
                                &nesfmr->root_vpbl.pbl_pbase);
                if (!nesfmr->root_vpbl.pbl_vbase) {
                        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                        ret = -ENOMEM;
                        goto failed_vpbl_alloc;
                }
                nesfmr->leaf_pbl_cnt = 0;
                nes_debug(NES_DBG_MR, "one level pbl, root_vpbl.pbl_vbase=%p \n",
                                nesfmr->root_vpbl.pbl_vbase);
        }
        /* two level pbl */
        else {
                nesfmr->root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 8192,
                                &nesfmr->root_vpbl.pbl_pbase);
                if (!nesfmr->root_vpbl.pbl_vbase) {
                        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                        ret = -ENOMEM;
                        goto failed_vpbl_alloc;
                }

                nesfmr->leaf_pbl_cnt = nesfmr->nesmr.pbls_used-1;
                nesfmr->root_vpbl.leaf_vpbl = kzalloc(sizeof(*nesfmr->root_vpbl.leaf_vpbl)*1024, GFP_ATOMIC);
                if (!nesfmr->root_vpbl.leaf_vpbl) {
                        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                        ret = -ENOMEM;
                        goto failed_leaf_vpbl_alloc;
                }

                nes_debug(NES_DBG_MR, "two level pbl, root_vpbl.pbl_vbase=%p"
                                " leaf_pbl_cnt=%d root_vpbl.leaf_vpbl=%p\n",
                                nesfmr->root_vpbl.pbl_vbase, nesfmr->leaf_pbl_cnt, nesfmr->root_vpbl.leaf_vpbl);

                for (i=0; i<nesfmr->leaf_pbl_cnt; i++)
                        nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase = NULL;

                for (i=0; i<nesfmr->leaf_pbl_cnt; i++) {
                        vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
                                        &vpbl.pbl_pbase);

                        if (!vpbl.pbl_vbase) {
                                ret = -ENOMEM;
                                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                                goto failed_leaf_vpbl_pages_alloc;
                        }

                        nesfmr->root_vpbl.pbl_vbase[i].pa_low = cpu_to_le32((u32)vpbl.pbl_pbase);
                        nesfmr->root_vpbl.pbl_vbase[i].pa_high = cpu_to_le32((u32)((((u64)vpbl.pbl_pbase)>>32)));
                        nesfmr->root_vpbl.leaf_vpbl[i] = vpbl;

                        nes_debug(NES_DBG_MR, "pbase_low=0x%x, pbase_high=0x%x, vpbl=%p\n",
                                        nesfmr->root_vpbl.pbl_vbase[i].pa_low,
                                        nesfmr->root_vpbl.pbl_vbase[i].pa_high,
                                        &nesfmr->root_vpbl.leaf_vpbl[i]);
                }
        }
        nesfmr->ib_qp = NULL;
        nesfmr->access_rights =0;

        stag = stag_index << 8;
        stag |= driver_key;
        stag += (u32)stag_key;

        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
                ret = -ENOMEM;
                goto failed_leaf_vpbl_pages_alloc;
        }
        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;

        nes_debug(NES_DBG_MR, "Registering STag 0x%08X, index = 0x%08X\n",
                        stag, stag_index);

        opcode = NES_CQP_ALLOCATE_STAG | NES_CQP_STAG_VA_TO | NES_CQP_STAG_MR;

        if (nesfmr->nesmr.pbl_4k == 1)
                opcode |= NES_CQP_STAG_PBL_BLK_SIZE;

        if (ibmr_access_flags & IB_ACCESS_REMOTE_WRITE) {
                opcode |= NES_CQP_STAG_RIGHTS_REMOTE_WRITE |
                                NES_CQP_STAG_RIGHTS_LOCAL_WRITE | NES_CQP_STAG_REM_ACC_EN;
                nesfmr->access_rights |=
                                NES_CQP_STAG_RIGHTS_REMOTE_WRITE | NES_CQP_STAG_RIGHTS_LOCAL_WRITE |
                                NES_CQP_STAG_REM_ACC_EN;
        }

        if (ibmr_access_flags & IB_ACCESS_REMOTE_READ) {
                opcode |= NES_CQP_STAG_RIGHTS_REMOTE_READ |
                                NES_CQP_STAG_RIGHTS_LOCAL_READ | NES_CQP_STAG_REM_ACC_EN;
                nesfmr->access_rights |=
                                NES_CQP_STAG_RIGHTS_REMOTE_READ | NES_CQP_STAG_RIGHTS_LOCAL_READ |
                                NES_CQP_STAG_REM_ACC_EN;
        }

        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX, (nespd->pd_id & 0x00007fff));
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, stag);

        cqp_wqe->wqe_words[NES_CQP_STAG_WQE_PBL_BLK_COUNT_IDX] =
                        cpu_to_le32((nesfmr->nesmr.pbls_used>1) ?
                        (nesfmr->nesmr.pbls_used-1) : nesfmr->nesmr.pbls_used);

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
                        NES_EVENT_TIMEOUT);
        nes_debug(NES_DBG_MR, "Register STag 0x%08X completed, wait_event_timeout ret = %u,"
                        " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                        stag, ret, cqp_request->major_code, cqp_request->minor_code);

        if ((!ret) || (cqp_request->major_code)) {
                nes_put_cqp_request(nesdev, cqp_request);
                ret = (!ret) ? -ETIME : -EIO;
                goto failed_leaf_vpbl_pages_alloc;
        }
        nes_put_cqp_request(nesdev, cqp_request);
        nesfmr->nesmr.ibfmr.lkey = stag;
        nesfmr->nesmr.ibfmr.rkey = stag;
        nesfmr->attr = *ibfmr_attr;

        return &nesfmr->nesmr.ibfmr;

        failed_leaf_vpbl_pages_alloc:
        /* unroll all allocated pages */
        for (i=0; i<nesfmr->leaf_pbl_cnt; i++) {
                if (nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase) {
                        pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase,
                                        nesfmr->root_vpbl.leaf_vpbl[i].pbl_pbase);
                }
        }
        if (nesfmr->root_vpbl.leaf_vpbl)
                kfree(nesfmr->root_vpbl.leaf_vpbl);

        failed_leaf_vpbl_alloc:
        if (nesfmr->leaf_pbl_cnt == 0) {
                if (nesfmr->root_vpbl.pbl_vbase)
                        pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.pbl_vbase,
                                        nesfmr->root_vpbl.pbl_pbase);
        } else
                pci_free_consistent(nesdev->pcidev, 8192, nesfmr->root_vpbl.pbl_vbase,
                                nesfmr->root_vpbl.pbl_pbase);

        failed_vpbl_alloc:
        if (nesfmr->nesmr.pbls_used != 0) {
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                if (nesfmr->nesmr.pbl_4k)
                        nesadapter->free_4kpbl += nesfmr->nesmr.pbls_used;
                else
                        nesadapter->free_256pbl += nesfmr->nesmr.pbls_used;
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
        }

failed_vpbl_avail:
        kfree(nesfmr);

        failed_fmr_alloc:
        nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);

        failed_resource_alloc:
        return ERR_PTR(ret);
}


/**
 * nes_dealloc_fmr
 */
static int nes_dealloc_fmr(struct ib_fmr *ibfmr)
{
        unsigned long flags;
        struct nes_mr *nesmr = to_nesmr_from_ibfmr(ibfmr);
        struct nes_fmr *nesfmr = to_nesfmr(nesmr);
        struct nes_vnic *nesvnic = to_nesvnic(ibfmr->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        int i = 0;
        int rc;

        /* free the resources */
        if (nesfmr->leaf_pbl_cnt == 0) {
                /* single PBL case */
                if (nesfmr->root_vpbl.pbl_vbase)
                        pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.pbl_vbase,
                                        nesfmr->root_vpbl.pbl_pbase);
        } else {
                for (i = 0; i < nesfmr->leaf_pbl_cnt; i++) {
                        pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase,
                                        nesfmr->root_vpbl.leaf_vpbl[i].pbl_pbase);
                }
                kfree(nesfmr->root_vpbl.leaf_vpbl);
                pci_free_consistent(nesdev->pcidev, 8192, nesfmr->root_vpbl.pbl_vbase,
                                nesfmr->root_vpbl.pbl_pbase);
        }
        nesmr->ibmw.device = ibfmr->device;
        nesmr->ibmw.pd = ibfmr->pd;
        nesmr->ibmw.rkey = ibfmr->rkey;
        nesmr->ibmw.uobject = NULL;

        rc = nes_dealloc_mw(&nesmr->ibmw);

        if ((rc == 0) && (nesfmr->nesmr.pbls_used != 0)) {
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                if (nesfmr->nesmr.pbl_4k) {
                        nesadapter->free_4kpbl += nesfmr->nesmr.pbls_used;
                        WARN_ON(nesadapter->free_4kpbl > nesadapter->max_4kpbl);
                } else {
                        nesadapter->free_256pbl += nesfmr->nesmr.pbls_used;
                        WARN_ON(nesadapter->free_256pbl > nesadapter->max_256pbl);
                }
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
        }

        return rc;
}


/**
 * nes_map_phys_fmr
 */
static int nes_map_phys_fmr(struct ib_fmr *ibfmr, u64 *page_list,
                int list_len, u64 iova)
{
        return 0;
}


/**
 * nes_unmap_frm
 */
static int nes_unmap_fmr(struct list_head *ibfmr_list)
{
        return 0;
}



/**
 * nes_query_device
 */
static int nes_query_device(struct ib_device *ibdev, struct ib_device_attr *props)
{
        struct nes_vnic *nesvnic = to_nesvnic(ibdev);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_ib_device *nesibdev = nesvnic->nesibdev;

        memset(props, 0, sizeof(*props));
        memcpy(&props->sys_image_guid, nesvnic->netdev->dev_addr, 6);

        props->fw_ver = nesdev->nesadapter->fw_ver;
        props->device_cap_flags = nesdev->nesadapter->device_cap_flags;
        props->vendor_id = nesdev->nesadapter->vendor_id;
        props->vendor_part_id = nesdev->nesadapter->vendor_part_id;
        props->hw_ver = nesdev->nesadapter->hw_rev;
        props->max_mr_size = 0x80000000;
        props->max_qp = nesibdev->max_qp;
        props->max_qp_wr = nesdev->nesadapter->max_qp_wr - 2;
        props->max_sge = nesdev->nesadapter->max_sge;
        props->max_cq = nesibdev->max_cq;
        props->max_cqe = nesdev->nesadapter->max_cqe - 1;
        props->max_mr = nesibdev->max_mr;
        props->max_mw = nesibdev->max_mr;
        props->max_pd = nesibdev->max_pd;
        props->max_sge_rd = 1;
        switch (nesdev->nesadapter->max_irrq_wr) {
                case 0:
                        props->max_qp_rd_atom = 1;
                        break;
                case 1:
                        props->max_qp_rd_atom = 4;
                        break;
                case 2:
                        props->max_qp_rd_atom = 16;
                        break;
                case 3:
                        props->max_qp_rd_atom = 32;
                        break;
                default:
                        props->max_qp_rd_atom = 0;
        }
        props->max_qp_init_rd_atom = props->max_qp_rd_atom;
        props->atomic_cap = IB_ATOMIC_NONE;
        props->max_map_per_fmr = 1;

        return 0;
}


/**
 * nes_query_port
 */
static int nes_query_port(struct ib_device *ibdev, u8 port, struct ib_port_attr *props)
{
        struct nes_vnic *nesvnic = to_nesvnic(ibdev);
        struct net_device *netdev = nesvnic->netdev;

        memset(props, 0, sizeof(*props));

        props->max_mtu = IB_MTU_4096;

        if (netdev->mtu  >= 4096)
                props->active_mtu = IB_MTU_4096;
        else if (netdev->mtu  >= 2048)
                props->active_mtu = IB_MTU_2048;
        else if (netdev->mtu  >= 1024)
                props->active_mtu = IB_MTU_1024;
        else if (netdev->mtu  >= 512)
                props->active_mtu = IB_MTU_512;
        else
                props->active_mtu = IB_MTU_256;

        props->lid = 1;
        props->lmc = 0;
        props->sm_lid = 0;
        props->sm_sl = 0;
        if (nesvnic->linkup)
                props->state = IB_PORT_ACTIVE;
        else
                props->state = IB_PORT_DOWN;
        props->phys_state = 0;
        props->port_cap_flags = IB_PORT_CM_SUP | IB_PORT_REINIT_SUP |
                        IB_PORT_VENDOR_CLASS_SUP | IB_PORT_BOOT_MGMT_SUP;
        props->gid_tbl_len = 1;
        props->pkey_tbl_len = 1;
        props->qkey_viol_cntr = 0;
        props->active_width = IB_WIDTH_4X;
        props->active_speed = 1;
        props->max_msg_sz = 0x80000000;

        return 0;
}


/**
 * nes_modify_port
 */
static int nes_modify_port(struct ib_device *ibdev, u8 port,
                int port_modify_mask, struct ib_port_modify *props)
{
        return 0;
}


/**
 * nes_query_pkey
 */
static int nes_query_pkey(struct ib_device *ibdev, u8 port, u16 index, u16 *pkey)
{
        *pkey = 0;
        return 0;
}


/**
 * nes_query_gid
 */
static int nes_query_gid(struct ib_device *ibdev, u8 port,
                int index, union ib_gid *gid)
{
        struct nes_vnic *nesvnic = to_nesvnic(ibdev);

        memset(&(gid->raw[0]), 0, sizeof(gid->raw));
        memcpy(&(gid->raw[0]), nesvnic->netdev->dev_addr, 6);

        return 0;
}


/**
 * nes_alloc_ucontext - Allocate the user context data structure. This keeps track
 * of all objects associated with a particular user-mode client.
 */
static struct ib_ucontext *nes_alloc_ucontext(struct ib_device *ibdev,
                struct ib_udata *udata)
{
        struct nes_vnic *nesvnic = to_nesvnic(ibdev);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_alloc_ucontext_req req;
        struct nes_alloc_ucontext_resp uresp;
        struct nes_ucontext *nes_ucontext;
        struct nes_ib_device *nesibdev = nesvnic->nesibdev;


        if (ib_copy_from_udata(&req, udata, sizeof(struct nes_alloc_ucontext_req))) {
                printk(KERN_ERR PFX "Invalid structure size on allocate user context.\n");
                return ERR_PTR(-EINVAL);
        }

        if (req.userspace_ver != NES_ABI_USERSPACE_VER) {
                printk(KERN_ERR PFX "Invalid userspace driver version detected. Detected version %d, should be %d\n",
                        req.userspace_ver, NES_ABI_USERSPACE_VER);
                return ERR_PTR(-EINVAL);
        }


        memset(&uresp, 0, sizeof uresp);

        uresp.max_qps = nesibdev->max_qp;
        uresp.max_pds = nesibdev->max_pd;
        uresp.wq_size = nesdev->nesadapter->max_qp_wr * 2;
        uresp.virtwq = nesadapter->virtwq;
        uresp.kernel_ver = NES_ABI_KERNEL_VER;

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

        nes_ucontext->nesdev = nesdev;
        nes_ucontext->mmap_wq_offset = uresp.max_pds;
        nes_ucontext->mmap_cq_offset = nes_ucontext->mmap_wq_offset +
                        ((sizeof(struct nes_hw_qp_wqe) * uresp.max_qps * 2) + PAGE_SIZE-1) /
                        PAGE_SIZE;


        if (ib_copy_to_udata(udata, &uresp, sizeof uresp)) {
                kfree(nes_ucontext);
                return ERR_PTR(-EFAULT);
        }

        INIT_LIST_HEAD(&nes_ucontext->cq_reg_mem_list);
        INIT_LIST_HEAD(&nes_ucontext->qp_reg_mem_list);
        atomic_set(&nes_ucontext->usecnt, 1);
        return &nes_ucontext->ibucontext;
}


/**
 * nes_dealloc_ucontext
 */
static int nes_dealloc_ucontext(struct ib_ucontext *context)
{
        /* struct nes_vnic *nesvnic = to_nesvnic(context->device); */
        /* struct nes_device *nesdev = nesvnic->nesdev; */
        struct nes_ucontext *nes_ucontext = to_nesucontext(context);

        if (!atomic_dec_and_test(&nes_ucontext->usecnt))
          return 0;
        kfree(nes_ucontext);
        return 0;
}


/**
 * nes_mmap
 */
static int nes_mmap(struct ib_ucontext *context, struct vm_area_struct *vma)
{
        unsigned long index;
        struct nes_vnic *nesvnic = to_nesvnic(context->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        /* struct nes_adapter *nesadapter = nesdev->nesadapter; */
        struct nes_ucontext *nes_ucontext;
        struct nes_qp *nesqp;

        nes_ucontext = to_nesucontext(context);


        if (vma->vm_pgoff >= nes_ucontext->mmap_wq_offset) {
                index = (vma->vm_pgoff - nes_ucontext->mmap_wq_offset) * PAGE_SIZE;
                index /= ((sizeof(struct nes_hw_qp_wqe) * nesdev->nesadapter->max_qp_wr * 2) +
                                PAGE_SIZE-1) & (~(PAGE_SIZE-1));
                if (!test_bit(index, nes_ucontext->allocated_wqs)) {
                        nes_debug(NES_DBG_MMAP, "wq %lu not allocated\n", index);
                        return -EFAULT;
                }
                nesqp = nes_ucontext->mmap_nesqp[index];
                if (nesqp == NULL) {
                        nes_debug(NES_DBG_MMAP, "wq %lu has a NULL QP base.\n", index);
                        return -EFAULT;
                }
                if (remap_pfn_range(vma, vma->vm_start,
                                virt_to_phys(nesqp->hwqp.sq_vbase) >> PAGE_SHIFT,
                                vma->vm_end - vma->vm_start,
                                vma->vm_page_prot)) {
                        nes_debug(NES_DBG_MMAP, "remap_pfn_range failed.\n");
                        return -EAGAIN;
                }
                vma->vm_private_data = nesqp;
                return 0;
        } else {
                index = vma->vm_pgoff;
                if (!test_bit(index, nes_ucontext->allocated_doorbells))
                        return -EFAULT;

                vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
                if (io_remap_pfn_range(vma, vma->vm_start,
                                (nesdev->doorbell_start +
                                ((nes_ucontext->mmap_db_index[index] - nesdev->base_doorbell_index) * 4096))
                                >> PAGE_SHIFT, PAGE_SIZE, vma->vm_page_prot))
                        return -EAGAIN;
                vma->vm_private_data = nes_ucontext;
                return 0;
        }

        return -ENOSYS;
}


/**
 * nes_alloc_pd
 */
static struct ib_pd *nes_alloc_pd(struct ib_device *ibdev,
                struct ib_ucontext *context, struct ib_udata *udata)
{
        struct nes_pd *nespd;
        struct nes_vnic *nesvnic = to_nesvnic(ibdev);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_ucontext *nesucontext;
        struct nes_alloc_pd_resp uresp;
        u32 pd_num = 0;
        int err;

        nes_debug(NES_DBG_PD, "nesvnic=%p, netdev=%p %s, ibdev=%p, context=%p, netdev refcnt=%u\n",
                        nesvnic, nesdev->netdev[0], nesdev->netdev[0]->name, ibdev, context,
                        atomic_read(&nesvnic->netdev->refcnt));

        err = nes_alloc_resource(nesadapter, nesadapter->allocated_pds,
                        nesadapter->max_pd, &pd_num, &nesadapter->next_pd);
        if (err) {
                return ERR_PTR(err);
        }

        nespd = kzalloc(sizeof (struct nes_pd), GFP_KERNEL);
        if (!nespd) {
                nes_free_resource(nesadapter, nesadapter->allocated_pds, pd_num);
                return ERR_PTR(-ENOMEM);
        }

        nes_debug(NES_DBG_PD, "Allocating PD (%p) for ib device %s\n",
                        nespd, nesvnic->nesibdev->ibdev.name);

        nespd->pd_id = (pd_num << (PAGE_SHIFT-12)) + nesadapter->base_pd;

        if (context) {
                nesucontext = to_nesucontext(context);
                nespd->mmap_db_index = find_next_zero_bit(nesucontext->allocated_doorbells,
                                NES_MAX_USER_DB_REGIONS, nesucontext->first_free_db);
                nes_debug(NES_DBG_PD, "find_first_zero_biton doorbells returned %u, mapping pd_id %u.\n",
                                nespd->mmap_db_index, nespd->pd_id);
                if (nespd->mmap_db_index >= NES_MAX_USER_DB_REGIONS) {
                        nes_debug(NES_DBG_PD, "mmap_db_index > MAX\n");
                        nes_free_resource(nesadapter, nesadapter->allocated_pds, pd_num);
                        kfree(nespd);
                        return ERR_PTR(-ENOMEM);
                }

                uresp.pd_id = nespd->pd_id;
                uresp.mmap_db_index = nespd->mmap_db_index;
                if (ib_copy_to_udata(udata, &uresp, sizeof (struct nes_alloc_pd_resp))) {
                        nes_free_resource(nesadapter, nesadapter->allocated_pds, pd_num);
                        kfree(nespd);
                        return ERR_PTR(-EFAULT);
                }

                set_bit(nespd->mmap_db_index, nesucontext->allocated_doorbells);
                nesucontext->mmap_db_index[nespd->mmap_db_index] = nespd->pd_id;
                nesucontext->first_free_db = nespd->mmap_db_index + 1;
        }

        nes_debug(NES_DBG_PD, "PD%u structure located @%p.\n", nespd->pd_id, nespd);
        return &nespd->ibpd;
}


/**
 * nes_dealloc_pd
 */
static int nes_dealloc_pd(struct ib_pd *ibpd)
{
        struct nes_ucontext *nesucontext;
        struct nes_pd *nespd = to_nespd(ibpd);
        struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;

        if ((ibpd->uobject) && (ibpd->uobject->context)) {
                nesucontext = to_nesucontext(ibpd->uobject->context);
                nes_debug(NES_DBG_PD, "Clearing bit %u from allocated doorbells\n",
                                nespd->mmap_db_index);
                clear_bit(nespd->mmap_db_index, nesucontext->allocated_doorbells);
                nesucontext->mmap_db_index[nespd->mmap_db_index] = 0;
                if (nesucontext->first_free_db > nespd->mmap_db_index) {
                        nesucontext->first_free_db = nespd->mmap_db_index;
                }
        }

        nes_debug(NES_DBG_PD, "Deallocating PD%u structure located @%p.\n",
                        nespd->pd_id, nespd);
        nes_free_resource(nesadapter, nesadapter->allocated_pds,
                        (nespd->pd_id-nesadapter->base_pd)>>(PAGE_SHIFT-12));
        kfree(nespd);

        return 0;
}


/**
 * nes_create_ah
 */
static struct ib_ah *nes_create_ah(struct ib_pd *pd, struct ib_ah_attr *ah_attr)
{
        return ERR_PTR(-ENOSYS);
}


/**
 * nes_destroy_ah
 */
static int nes_destroy_ah(struct ib_ah *ah)
{
        return -ENOSYS;
}


/**
 * nes_get_encoded_size
 */
static inline u8 nes_get_encoded_size(int *size)
{
        u8 encoded_size = 0;
        if (*size <= 32) {
                *size = 32;

                encoded_size = 1;
        } else if (*size <= 128) {
                *size = 128;
                encoded_size = 2;
        } else if (*size <= 512) {
                *size = 512;
                encoded_size = 3;
        }
        return (encoded_size);
}



/**
 * nes_setup_virt_qp
 */
static int nes_setup_virt_qp(struct nes_qp *nesqp, struct nes_pbl *nespbl,
                struct nes_vnic *nesvnic, int sq_size, int rq_size)
{
        unsigned long flags;
        void *mem;
        __le64 *pbl = NULL;
        __le64 *tpbl;
        __le64 *pblbuffer;
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        u32 pbl_entries;
        u8 rq_pbl_entries;
        u8 sq_pbl_entries;

        pbl_entries = nespbl->pbl_size >> 3;
        nes_debug(NES_DBG_QP, "Userspace PBL, pbl_size=%u, pbl_entries = %d pbl_vbase=%p, pbl_pbase=%lx\n",
                        nespbl->pbl_size, pbl_entries,
                        (void *)nespbl->pbl_vbase,
                        (unsigned long) nespbl->pbl_pbase);
        pbl = (__le64 *) nespbl->pbl_vbase; /* points to first pbl entry */
        /* now lets set the sq_vbase as well as rq_vbase addrs we will assign */
        /* the first pbl to be fro the rq_vbase... */
        rq_pbl_entries = (rq_size * sizeof(struct nes_hw_qp_wqe)) >> 12;
        sq_pbl_entries = (sq_size * sizeof(struct nes_hw_qp_wqe)) >> 12;
        nesqp->hwqp.sq_pbase = (le32_to_cpu(((__le32 *)pbl)[0])) | ((u64)((le32_to_cpu(((__le32 *)pbl)[1]))) << 32);
        if (!nespbl->page) {
                nes_debug(NES_DBG_QP, "QP nespbl->page is NULL \n");
                kfree(nespbl);
                return -ENOMEM;
        }

        nesqp->hwqp.sq_vbase = kmap(nespbl->page);
        nesqp->page = nespbl->page;
        if (!nesqp->hwqp.sq_vbase) {
                nes_debug(NES_DBG_QP, "QP sq_vbase kmap failed\n");
                kfree(nespbl);
                return -ENOMEM;
        }

        /* Now to get to sq.. we need to calculate how many */
        /* PBL entries were used by the rq.. */
        pbl += sq_pbl_entries;
        nesqp->hwqp.rq_pbase = (le32_to_cpu(((__le32 *)pbl)[0])) | ((u64)((le32_to_cpu(((__le32 *)pbl)[1]))) << 32);
        /* nesqp->hwqp.rq_vbase = bus_to_virt(*pbl); */
        /*nesqp->hwqp.rq_vbase = phys_to_virt(*pbl); */

        nes_debug(NES_DBG_QP, "QP sq_vbase= %p sq_pbase=%lx rq_vbase=%p rq_pbase=%lx\n",
                  nesqp->hwqp.sq_vbase, (unsigned long) nesqp->hwqp.sq_pbase,
                  nesqp->hwqp.rq_vbase, (unsigned long) nesqp->hwqp.rq_pbase);
        spin_lock_irqsave(&nesadapter->pbl_lock, flags);
        if (!nesadapter->free_256pbl) {
                pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
                                nespbl->pbl_pbase);
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                kunmap(nesqp->page);
                kfree(nespbl);
                return -ENOMEM;
        }
        nesadapter->free_256pbl--;
        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);

        nesqp->pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 256, &nesqp->pbl_pbase);
        pblbuffer = nesqp->pbl_vbase;
        if (!nesqp->pbl_vbase) {
                /* memory allocated during nes_reg_user_mr() */
                pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
                                    nespbl->pbl_pbase);
                kfree(nespbl);
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                nesadapter->free_256pbl++;
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                kunmap(nesqp->page);
                return -ENOMEM;
        }
        memset(nesqp->pbl_vbase, 0, 256);
        /* fill in the page address in the pbl buffer.. */
        tpbl = pblbuffer + 16;
        pbl = (__le64 *)nespbl->pbl_vbase;
        while (sq_pbl_entries--)
                *tpbl++ = *pbl++;
        tpbl = pblbuffer;
        while (rq_pbl_entries--)
                *tpbl++ = *pbl++;

        /* done with memory allocated during nes_reg_user_mr() */
        pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
                            nespbl->pbl_pbase);
        kfree(nespbl);

        nesqp->qp_mem_size =
                        max((u32)sizeof(struct nes_qp_context), ((u32)256)) + 256;     /* this is Q2 */
        /* Round up to a multiple of a page */
        nesqp->qp_mem_size += PAGE_SIZE - 1;
        nesqp->qp_mem_size &= ~(PAGE_SIZE - 1);

        mem = pci_alloc_consistent(nesdev->pcidev, nesqp->qp_mem_size,
                        &nesqp->hwqp.q2_pbase);

        if (!mem) {
                pci_free_consistent(nesdev->pcidev, 256, nesqp->pbl_vbase, nesqp->pbl_pbase);
                nesqp->pbl_vbase = NULL;
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                nesadapter->free_256pbl++;
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                kunmap(nesqp->page);
                return -ENOMEM;
        }
        nesqp->hwqp.q2_vbase = mem;
        mem += 256;
        memset(nesqp->hwqp.q2_vbase, 0, 256);
        nesqp->nesqp_context = mem;
        memset(nesqp->nesqp_context, 0, sizeof(*nesqp->nesqp_context));
        nesqp->nesqp_context_pbase = nesqp->hwqp.q2_pbase + 256;

        return 0;
}


/**
 * nes_setup_mmap_qp
 */
static int nes_setup_mmap_qp(struct nes_qp *nesqp, struct nes_vnic *nesvnic,
                int sq_size, int rq_size)
{
        void *mem;
        struct nes_device *nesdev = nesvnic->nesdev;

        nesqp->qp_mem_size = (sizeof(struct nes_hw_qp_wqe) * sq_size) +
                        (sizeof(struct nes_hw_qp_wqe) * rq_size) +
                        max((u32)sizeof(struct nes_qp_context), ((u32)256)) +
                        256; /* this is Q2 */
        /* Round up to a multiple of a page */
        nesqp->qp_mem_size += PAGE_SIZE - 1;
        nesqp->qp_mem_size &= ~(PAGE_SIZE - 1);

        mem = pci_alloc_consistent(nesdev->pcidev, nesqp->qp_mem_size,
                        &nesqp->hwqp.sq_pbase);
        if (!mem)
                return -ENOMEM;
        nes_debug(NES_DBG_QP, "PCI consistent memory for "
                        "host descriptor rings located @ %p (pa = 0x%08lX.) size = %u.\n",
                        mem, (unsigned long)nesqp->hwqp.sq_pbase, nesqp->qp_mem_size);

        memset(mem, 0, nesqp->qp_mem_size);

        nesqp->hwqp.sq_vbase = mem;
        mem += sizeof(struct nes_hw_qp_wqe) * sq_size;

        nesqp->hwqp.rq_vbase = mem;
        nesqp->hwqp.rq_pbase = nesqp->hwqp.sq_pbase +
                        sizeof(struct nes_hw_qp_wqe) * sq_size;
        mem += sizeof(struct nes_hw_qp_wqe) * rq_size;

        nesqp->hwqp.q2_vbase = mem;
        nesqp->hwqp.q2_pbase = nesqp->hwqp.rq_pbase +
                        sizeof(struct nes_hw_qp_wqe) * rq_size;
        mem += 256;
        memset(nesqp->hwqp.q2_vbase, 0, 256);

        nesqp->nesqp_context = mem;
        nesqp->nesqp_context_pbase = nesqp->hwqp.q2_pbase + 256;
        memset(nesqp->nesqp_context, 0, sizeof(*nesqp->nesqp_context));
        return 0;
}


/**
 * nes_free_qp_mem() is to free up the qp's pci_alloc_consistent() memory.
 */
static inline void nes_free_qp_mem(struct nes_device *nesdev,
                struct nes_qp *nesqp, int virt_wqs)
{
        unsigned long flags;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        if (!virt_wqs) {
                pci_free_consistent(nesdev->pcidev, nesqp->qp_mem_size,
                                nesqp->hwqp.sq_vbase, nesqp->hwqp.sq_pbase);
        }else {
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                nesadapter->free_256pbl++;
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                pci_free_consistent(nesdev->pcidev, nesqp->qp_mem_size, nesqp->hwqp.q2_vbase, nesqp->hwqp.q2_pbase);
                pci_free_consistent(nesdev->pcidev, 256, nesqp->pbl_vbase, nesqp->pbl_pbase );
                nesqp->pbl_vbase = NULL;
                kunmap(nesqp->page);
        }
}


/**
 * nes_create_qp
 */
static struct ib_qp *nes_create_qp(struct ib_pd *ibpd,
                struct ib_qp_init_attr *init_attr, struct ib_udata *udata)
{
        u64 u64temp= 0;
        u64 u64nesqp = 0;
        struct nes_pd *nespd = to_nespd(ibpd);
        struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_qp *nesqp;
        struct nes_cq *nescq;
        struct nes_ucontext *nes_ucontext;
        struct nes_hw_cqp_wqe *cqp_wqe;
        struct nes_cqp_request *cqp_request;
        struct nes_create_qp_req req;
        struct nes_create_qp_resp uresp;
        struct nes_pbl  *nespbl = NULL;
        u32 qp_num = 0;
        u32 opcode = 0;
        /* u32 counter = 0; */
        void *mem;
        unsigned long flags;
        int ret;
        int err;
        int virt_wqs = 0;
        int sq_size;
        int rq_size;
        u8 sq_encoded_size;
        u8 rq_encoded_size;
        /* int counter; */

        if (init_attr->create_flags)
                return ERR_PTR(-EINVAL);

        atomic_inc(&qps_created);
        switch (init_attr->qp_type) {
                case IB_QPT_RC:
                        if (nes_drv_opt & NES_DRV_OPT_NO_INLINE_DATA) {
                                init_attr->cap.max_inline_data = 0;
                        } else {
                                init_attr->cap.max_inline_data = 64;
                        }
                        sq_size = init_attr->cap.max_send_wr;
                        rq_size = init_attr->cap.max_recv_wr;

                        /* check if the encoded sizes are OK or not... */
                        sq_encoded_size = nes_get_encoded_size(&sq_size);
                        rq_encoded_size = nes_get_encoded_size(&rq_size);

                        if ((!sq_encoded_size) || (!rq_encoded_size)) {
                                nes_debug(NES_DBG_QP, "ERROR bad rq (%u) or sq (%u) size\n",
                                                rq_size, sq_size);
                                return ERR_PTR(-EINVAL);
                        }

                        init_attr->cap.max_send_wr = sq_size -2;
                        init_attr->cap.max_recv_wr = rq_size -1;
                        nes_debug(NES_DBG_QP, "RQ size=%u, SQ Size=%u\n", rq_size, sq_size);

                        ret = nes_alloc_resource(nesadapter, nesadapter->allocated_qps,
                                        nesadapter->max_qp, &qp_num, &nesadapter->next_qp);
                        if (ret) {
                                return ERR_PTR(ret);
                        }

                        /* Need 512 (actually now 1024) byte alignment on this structure */
                        mem = kzalloc(sizeof(*nesqp)+NES_SW_CONTEXT_ALIGN-1, GFP_KERNEL);
                        if (!mem) {
                                nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                nes_debug(NES_DBG_QP, "Unable to allocate QP\n");
                                return ERR_PTR(-ENOMEM);
                        }
                        u64nesqp = (unsigned long)mem;
                        u64nesqp += ((u64)NES_SW_CONTEXT_ALIGN) - 1;
                        u64temp = ((u64)NES_SW_CONTEXT_ALIGN) - 1;
                        u64nesqp &= ~u64temp;
                        nesqp = (struct nes_qp *)(unsigned long)u64nesqp;
                        /* nes_debug(NES_DBG_QP, "nesqp=%p, allocated buffer=%p.  Rounded to closest %u\n",
                                        nesqp, mem, NES_SW_CONTEXT_ALIGN); */
                        nesqp->allocated_buffer = mem;

                        if (udata) {
                                if (ib_copy_from_udata(&req, udata, sizeof(struct nes_create_qp_req))) {
                                        nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                        kfree(nesqp->allocated_buffer);
                                        nes_debug(NES_DBG_QP, "ib_copy_from_udata() Failed \n");
                                        return NULL;
                                }
                                if (req.user_wqe_buffers) {
                                        virt_wqs = 1;
                                }
                                if ((ibpd->uobject) && (ibpd->uobject->context)) {
                                        nesqp->user_mode = 1;
                                        nes_ucontext = to_nesucontext(ibpd->uobject->context);
                                        if (virt_wqs) {
                                                err = 1;
                                                list_for_each_entry(nespbl, &nes_ucontext->qp_reg_mem_list, list) {
                                                        if (nespbl->user_base == (unsigned long )req.user_wqe_buffers) {
                                                                list_del(&nespbl->list);
                                                                err = 0;
                                                                nes_debug(NES_DBG_QP, "Found PBL for virtual QP. nespbl=%p. user_base=0x%lx\n",
                                                                          nespbl, nespbl->user_base);
                                                                break;
                                                        }
                                                }
                                                if (err) {
                                                        nes_debug(NES_DBG_QP, "Didn't Find PBL for virtual QP. address = %llx.\n",
                                                                  (long long unsigned int)req.user_wqe_buffers);
                                                        nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                                        kfree(nesqp->allocated_buffer);
                                                        return ERR_PTR(-EFAULT);
                                                }
                                        }

                                        nes_ucontext = to_nesucontext(ibpd->uobject->context);
                                        nesqp->mmap_sq_db_index =
                                                find_next_zero_bit(nes_ucontext->allocated_wqs,
                                                                   NES_MAX_USER_WQ_REGIONS, nes_ucontext->first_free_wq);
                                        /* nes_debug(NES_DBG_QP, "find_first_zero_biton wqs returned %u\n",
                                                        nespd->mmap_db_index); */
                                        if (nesqp->mmap_sq_db_index >= NES_MAX_USER_WQ_REGIONS) {
                                                nes_debug(NES_DBG_QP,
                                                          "db index > max user regions, failing create QP\n");
                                                nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                                if (virt_wqs) {
                                                        pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
                                                                            nespbl->pbl_pbase);
                                                        kfree(nespbl);
                                                }
                                                kfree(nesqp->allocated_buffer);
                                                return ERR_PTR(-ENOMEM);
                                        }
                                        set_bit(nesqp->mmap_sq_db_index, nes_ucontext->allocated_wqs);
                                        nes_ucontext->mmap_nesqp[nesqp->mmap_sq_db_index] = nesqp;
                                        nes_ucontext->first_free_wq = nesqp->mmap_sq_db_index + 1;
                                } else {
                                        nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                        kfree(nesqp->allocated_buffer);
                                        return ERR_PTR(-EFAULT);
                                }
                        }
                        err = (!virt_wqs) ? nes_setup_mmap_qp(nesqp, nesvnic, sq_size, rq_size) :
                                        nes_setup_virt_qp(nesqp, nespbl, nesvnic, sq_size, rq_size);
                        if (err) {
                                nes_debug(NES_DBG_QP,
                                          "error geting qp mem code = %d\n", err);
                                nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                kfree(nesqp->allocated_buffer);
                                return ERR_PTR(-ENOMEM);
                        }

                        nesqp->hwqp.sq_size = sq_size;
                        nesqp->hwqp.sq_encoded_size = sq_encoded_size;
                        nesqp->hwqp.sq_head = 1;
                        nesqp->hwqp.rq_size = rq_size;
                        nesqp->hwqp.rq_encoded_size = rq_encoded_size;
                        /* nes_debug(NES_DBG_QP, "nesqp->nesqp_context_pbase = %p\n",
                                        (void *)nesqp->nesqp_context_pbase);
                        */
                        nesqp->hwqp.qp_id = qp_num;
                        nesqp->ibqp.qp_num = nesqp->hwqp.qp_id;
                        nesqp->nespd = nespd;

                        nescq = to_nescq(init_attr->send_cq);
                        nesqp->nesscq = nescq;
                        nescq = to_nescq(init_attr->recv_cq);
                        nesqp->nesrcq = nescq;

                        nesqp->nesqp_context->misc |= cpu_to_le32((u32)PCI_FUNC(nesdev->pcidev->devfn) <<
                                        NES_QPCONTEXT_MISC_PCI_FCN_SHIFT);
                        nesqp->nesqp_context->misc |= cpu_to_le32((u32)nesqp->hwqp.rq_encoded_size <<
                                        NES_QPCONTEXT_MISC_RQ_SIZE_SHIFT);
                        nesqp->nesqp_context->misc |= cpu_to_le32((u32)nesqp->hwqp.sq_encoded_size <<
                                        NES_QPCONTEXT_MISC_SQ_SIZE_SHIFT);
                        if (!udata) {
                                nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_PRIV_EN);
                                nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_FAST_REGISTER_EN);
                        }
                        nesqp->nesqp_context->cqs = cpu_to_le32(nesqp->nesscq->hw_cq.cq_number +
                                        ((u32)nesqp->nesrcq->hw_cq.cq_number << 16));
                        u64temp = (u64)nesqp->hwqp.sq_pbase;
                        nesqp->nesqp_context->sq_addr_low = cpu_to_le32((u32)u64temp);
                        nesqp->nesqp_context->sq_addr_high = cpu_to_le32((u32)(u64temp >> 32));


                        if (!virt_wqs) {
                                u64temp = (u64)nesqp->hwqp.sq_pbase;
                                nesqp->nesqp_context->sq_addr_low = cpu_to_le32((u32)u64temp);
                                nesqp->nesqp_context->sq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
                                u64temp = (u64)nesqp->hwqp.rq_pbase;
                                nesqp->nesqp_context->rq_addr_low = cpu_to_le32((u32)u64temp);
                                nesqp->nesqp_context->rq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
                        } else {
                                u64temp = (u64)nesqp->pbl_pbase;
                                nesqp->nesqp_context->rq_addr_low = cpu_to_le32((u32)u64temp);
                                nesqp->nesqp_context->rq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
                        }

                        /* nes_debug(NES_DBG_QP, "next_qp_nic_index=%u, using nic_index=%d\n",
                                        nesvnic->next_qp_nic_index,
                                        nesvnic->qp_nic_index[nesvnic->next_qp_nic_index]); */
                        spin_lock_irqsave(&nesdev->cqp.lock, flags);
                        nesqp->nesqp_context->misc2 |= cpu_to_le32(
                                        (u32)nesvnic->qp_nic_index[nesvnic->next_qp_nic_index] <<
                                        NES_QPCONTEXT_MISC2_NIC_INDEX_SHIFT);
                        nesvnic->next_qp_nic_index++;
                        if ((nesvnic->next_qp_nic_index > 3) ||
                                        (nesvnic->qp_nic_index[nesvnic->next_qp_nic_index] == 0xf)) {
                                nesvnic->next_qp_nic_index = 0;
                        }
                        spin_unlock_irqrestore(&nesdev->cqp.lock, flags);

                        nesqp->nesqp_context->pd_index_wscale |= cpu_to_le32((u32)nesqp->nespd->pd_id << 16);
                        u64temp = (u64)nesqp->hwqp.q2_pbase;
                        nesqp->nesqp_context->q2_addr_low = cpu_to_le32((u32)u64temp);
                        nesqp->nesqp_context->q2_addr_high = cpu_to_le32((u32)(u64temp >> 32));
                        nesqp->nesqp_context->aeq_token_low =  cpu_to_le32((u32)((unsigned long)(nesqp)));
                        nesqp->nesqp_context->aeq_token_high =  cpu_to_le32((u32)(upper_32_bits((unsigned long)(nesqp))));
                        nesqp->nesqp_context->ird_ord_sizes = cpu_to_le32(NES_QPCONTEXT_ORDIRD_ALSMM |
                                        ((((u32)nesadapter->max_irrq_wr) <<
                                        NES_QPCONTEXT_ORDIRD_IRDSIZE_SHIFT) & NES_QPCONTEXT_ORDIRD_IRDSIZE_MASK));
                        if (disable_mpa_crc) {
                                nes_debug(NES_DBG_QP, "Disabling MPA crc checking due to module option.\n");
                                nesqp->nesqp_context->ird_ord_sizes |= cpu_to_le32(NES_QPCONTEXT_ORDIRD_RNMC);
                        }


                        /* Create the QP */
                        cqp_request = nes_get_cqp_request(nesdev);
                        if (cqp_request == NULL) {
                                nes_debug(NES_DBG_QP, "Failed to get a cqp_request\n");
                                nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                nes_free_qp_mem(nesdev, nesqp,virt_wqs);
                                kfree(nesqp->allocated_buffer);
                                return ERR_PTR(-ENOMEM);
                        }
                        cqp_request->waiting = 1;
                        cqp_wqe = &cqp_request->cqp_wqe;

                        if (!virt_wqs) {
                                opcode = NES_CQP_CREATE_QP | NES_CQP_QP_TYPE_IWARP |
                                        NES_CQP_QP_IWARP_STATE_IDLE;
                        } else {
                                opcode = NES_CQP_CREATE_QP | NES_CQP_QP_TYPE_IWARP | NES_CQP_QP_VIRT_WQS |
                                        NES_CQP_QP_IWARP_STATE_IDLE;
                        }
                        opcode |= NES_CQP_QP_CQS_VALID;
                        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
                        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
                        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX, nesqp->hwqp.qp_id);

                        u64temp = (u64)nesqp->nesqp_context_pbase;
                        set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_QP_WQE_CONTEXT_LOW_IDX, u64temp);

                        atomic_set(&cqp_request->refcount, 2);
                        nes_post_cqp_request(nesdev, cqp_request);

                        /* Wait for CQP */
                        nes_debug(NES_DBG_QP, "Waiting for create iWARP QP%u to complete.\n",
                                        nesqp->hwqp.qp_id);
                        ret = wait_event_timeout(cqp_request->waitq,
                                        (cqp_request->request_done != 0), NES_EVENT_TIMEOUT);
                        nes_debug(NES_DBG_QP, "Create iwarp QP%u completed, wait_event_timeout ret=%u,"
                                        " nesdev->cqp_head = %u, nesdev->cqp.sq_tail = %u,"
                                        " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                                        nesqp->hwqp.qp_id, ret, nesdev->cqp.sq_head, nesdev->cqp.sq_tail,
                                        cqp_request->major_code, cqp_request->minor_code);
                        if ((!ret) || (cqp_request->major_code)) {
                                nes_put_cqp_request(nesdev, cqp_request);
                                nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                nes_free_qp_mem(nesdev, nesqp,virt_wqs);
                                kfree(nesqp->allocated_buffer);
                                if (!ret) {
                                        return ERR_PTR(-ETIME);
                                } else {
                                        return ERR_PTR(-EIO);
                                }
                        }

                        nes_put_cqp_request(nesdev, cqp_request);

                        if (ibpd->uobject) {
                                uresp.mmap_sq_db_index = nesqp->mmap_sq_db_index;
                                uresp.actual_sq_size = sq_size;
                                uresp.actual_rq_size = rq_size;
                                uresp.qp_id = nesqp->hwqp.qp_id;
                                uresp.nes_drv_opt = nes_drv_opt;
                                if (ib_copy_to_udata(udata, &uresp, sizeof uresp)) {
                                        nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
                                        nes_free_qp_mem(nesdev, nesqp,virt_wqs);
                                        kfree(nesqp->allocated_buffer);
                                        return ERR_PTR(-EFAULT);
                                }
                        }

                        nes_debug(NES_DBG_QP, "QP%u structure located @%p.Size = %u.\n",
                                        nesqp->hwqp.qp_id, nesqp, (u32)sizeof(*nesqp));
                        spin_lock_init(&nesqp->lock);
                        init_waitqueue_head(&nesqp->state_waitq);
                        init_waitqueue_head(&nesqp->kick_waitq);
                        nes_add_ref(&nesqp->ibqp);
                        break;
                default:
                        nes_debug(NES_DBG_QP, "Invalid QP type: %d\n", init_attr->qp_type);
                        return ERR_PTR(-EINVAL);
        }

        /* update the QP table */
        nesdev->nesadapter->qp_table[nesqp->hwqp.qp_id-NES_FIRST_QPN] = nesqp;
        nes_debug(NES_DBG_QP, "netdev refcnt=%u\n",
                        atomic_read(&nesvnic->netdev->refcnt));

        return &nesqp->ibqp;
}


/**
 * nes_clean_cq
 */
static void nes_clean_cq(struct nes_qp *nesqp, struct nes_cq *nescq)
{
        u32 cq_head;
        u32 lo;
        u32 hi;
        u64 u64temp;
        unsigned long flags = 0;

        spin_lock_irqsave(&nescq->lock, flags);

        cq_head = nescq->hw_cq.cq_head;
        while (le32_to_cpu(nescq->hw_cq.cq_vbase[cq_head].cqe_words[NES_CQE_OPCODE_IDX]) & NES_CQE_VALID) {
                rmb();
                lo = le32_to_cpu(nescq->hw_cq.cq_vbase[cq_head].cqe_words[NES_CQE_COMP_COMP_CTX_LOW_IDX]);
                hi = le32_to_cpu(nescq->hw_cq.cq_vbase[cq_head].cqe_words[NES_CQE_COMP_COMP_CTX_HIGH_IDX]);
                u64temp = (((u64)hi) << 32) | ((u64)lo);
                u64temp &= ~(NES_SW_CONTEXT_ALIGN-1);
                if (u64temp == (u64)(unsigned long)nesqp) {
                        /* Zero the context value so cqe will be ignored */
                        nescq->hw_cq.cq_vbase[cq_head].cqe_words[NES_CQE_COMP_COMP_CTX_LOW_IDX] = 0;
                        nescq->hw_cq.cq_vbase[cq_head].cqe_words[NES_CQE_COMP_COMP_CTX_HIGH_IDX] = 0;
                }

                if (++cq_head >= nescq->hw_cq.cq_size)
                        cq_head = 0;
        }

        spin_unlock_irqrestore(&nescq->lock, flags);
}


/**
 * nes_destroy_qp
 */
static int nes_destroy_qp(struct ib_qp *ibqp)
{
        struct nes_qp *nesqp = to_nesqp(ibqp);
        struct nes_ucontext *nes_ucontext;
        struct ib_qp_attr attr;
        struct iw_cm_id *cm_id;
        struct iw_cm_event cm_event;
        int ret;

        atomic_inc(&sw_qps_destroyed);
        nesqp->destroyed = 1;

        /* Blow away the connection if it exists. */
        if (nesqp->ibqp_state >= IB_QPS_INIT && nesqp->ibqp_state <= IB_QPS_RTS) {
                /* if (nesqp->ibqp_state == IB_QPS_RTS) { */
                attr.qp_state = IB_QPS_ERR;
                nes_modify_qp(&nesqp->ibqp, &attr, IB_QP_STATE, NULL);
        }

        if (((nesqp->ibqp_state == IB_QPS_INIT) ||
                        (nesqp->ibqp_state == IB_QPS_RTR)) && (nesqp->cm_id)) {
                cm_id = nesqp->cm_id;
                cm_event.event = IW_CM_EVENT_CONNECT_REPLY;
                cm_event.status = IW_CM_EVENT_STATUS_TIMEOUT;
                cm_event.local_addr = cm_id->local_addr;
                cm_event.remote_addr = cm_id->remote_addr;
                cm_event.private_data = NULL;
                cm_event.private_data_len = 0;

                nes_debug(NES_DBG_QP, "Generating a CM Timeout Event for "
                                "QP%u. cm_id = %p, refcount = %u. \n",
                                nesqp->hwqp.qp_id, cm_id, atomic_read(&nesqp->refcount));

                cm_id->rem_ref(cm_id);
                ret = cm_id->event_handler(cm_id, &cm_event);
                if (ret)
                        nes_debug(NES_DBG_QP, "OFA CM event_handler returned, ret=%d\n", ret);
        }

        if (nesqp->user_mode) {
                if ((ibqp->uobject)&&(ibqp->uobject->context)) {
                        nes_ucontext = to_nesucontext(ibqp->uobject->context);
                        clear_bit(nesqp->mmap_sq_db_index, nes_ucontext->allocated_wqs);
                        nes_ucontext->mmap_nesqp[nesqp->mmap_sq_db_index] = NULL;
                        if (nes_ucontext->first_free_wq > nesqp->mmap_sq_db_index) {
                                nes_ucontext->first_free_wq = nesqp->mmap_sq_db_index;
                        }
                }
                if (nesqp->pbl_pbase)
                        kunmap(nesqp->page);
        } else {
                /* Clean any pending completions from the cq(s) */
                if (nesqp->nesscq)
                        nes_clean_cq(nesqp, nesqp->nesscq);

                if ((nesqp->nesrcq) && (nesqp->nesrcq != nesqp->nesscq))
                        nes_clean_cq(nesqp, nesqp->nesrcq);
        }

        nes_rem_ref(&nesqp->ibqp);
        return 0;
}


/**
 * nes_create_cq
 */
static struct ib_cq *nes_create_cq(struct ib_device *ibdev, int entries,
                int comp_vector,
                struct ib_ucontext *context, struct ib_udata *udata)
{
        u64 u64temp;
        struct nes_vnic *nesvnic = to_nesvnic(ibdev);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_cq *nescq;
        struct nes_ucontext *nes_ucontext = NULL;
        struct nes_cqp_request *cqp_request;
        void *mem = NULL;
        struct nes_hw_cqp_wqe *cqp_wqe;
        struct nes_pbl *nespbl = NULL;
        struct nes_create_cq_req req;
        struct nes_create_cq_resp resp;
        u32 cq_num = 0;
        u32 opcode = 0;
        u32 pbl_entries = 1;
        int err;
        unsigned long flags;
        int ret;

        err = nes_alloc_resource(nesadapter, nesadapter->allocated_cqs,
                        nesadapter->max_cq, &cq_num, &nesadapter->next_cq);
        if (err) {
                return ERR_PTR(err);
        }

        nescq = kzalloc(sizeof(struct nes_cq), GFP_KERNEL);
        if (!nescq) {
                nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                nes_debug(NES_DBG_CQ, "Unable to allocate nes_cq struct\n");
                return ERR_PTR(-ENOMEM);
        }

        nescq->hw_cq.cq_size = max(entries + 1, 5);
        nescq->hw_cq.cq_number = cq_num;
        nescq->ibcq.cqe = nescq->hw_cq.cq_size - 1;


        if (context) {
                nes_ucontext = to_nesucontext(context);
                if (ib_copy_from_udata(&req, udata, sizeof (struct nes_create_cq_req))) {
                        nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                        kfree(nescq);
                        return ERR_PTR(-EFAULT);
                }
                nesvnic->mcrq_ucontext = nes_ucontext;
                nes_ucontext->mcrqf = req.mcrqf;
                if (nes_ucontext->mcrqf) {
                        if (nes_ucontext->mcrqf & 0x80000000)
                                nescq->hw_cq.cq_number = nesvnic->nic.qp_id + 28 + 2 * ((nes_ucontext->mcrqf & 0xf) - 1);
                        else if (nes_ucontext->mcrqf & 0x40000000)
                                nescq->hw_cq.cq_number = nes_ucontext->mcrqf & 0xffff;
                        else
                                nescq->hw_cq.cq_number = nesvnic->mcrq_qp_id + nes_ucontext->mcrqf-1;
                        nescq->mcrqf = nes_ucontext->mcrqf;
                        nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                }
                nes_debug(NES_DBG_CQ, "CQ Virtual Address = %08lX, size = %u.\n",
                                (unsigned long)req.user_cq_buffer, entries);
                err = 1;
                list_for_each_entry(nespbl, &nes_ucontext->cq_reg_mem_list, list) {
                        if (nespbl->user_base == (unsigned long )req.user_cq_buffer) {
                                list_del(&nespbl->list);
                                err = 0;
                                nes_debug(NES_DBG_CQ, "Found PBL for virtual CQ. nespbl=%p.\n",
                                                nespbl);
                                break;
                        }
                }
                if (err) {
                        nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                        kfree(nescq);
                        return ERR_PTR(-EFAULT);
                }

                pbl_entries = nespbl->pbl_size >> 3;
                nescq->cq_mem_size = 0;
        } else {
                nescq->cq_mem_size = nescq->hw_cq.cq_size * sizeof(struct nes_hw_cqe);
                nes_debug(NES_DBG_CQ, "Attempting to allocate pci memory (%u entries, %u bytes) for CQ%u.\n",
                                entries, nescq->cq_mem_size, nescq->hw_cq.cq_number);

                /* allocate the physical buffer space */
                mem = pci_alloc_consistent(nesdev->pcidev, nescq->cq_mem_size,
                                &nescq->hw_cq.cq_pbase);
                if (!mem) {
                        printk(KERN_ERR PFX "Unable to allocate pci memory for cq\n");
                        nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                        kfree(nescq);
                        return ERR_PTR(-ENOMEM);
                }

                memset(mem, 0, nescq->cq_mem_size);
                nescq->hw_cq.cq_vbase = mem;
                nescq->hw_cq.cq_head = 0;
                nes_debug(NES_DBG_CQ, "CQ%u virtual address @ %p, phys = 0x%08X\n",
                                nescq->hw_cq.cq_number, nescq->hw_cq.cq_vbase,
                                (u32)nescq->hw_cq.cq_pbase);
        }

        nescq->hw_cq.ce_handler = nes_iwarp_ce_handler;
        spin_lock_init(&nescq->lock);

        /* send CreateCQ request to CQP */
        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_CQ, "Failed to get a cqp_request.\n");
                if (!context)
                        pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
                                        nescq->hw_cq.cq_pbase);
                else {
                        pci_free_consistent(nesdev->pcidev, nespbl->pbl_size,
                                            nespbl->pbl_vbase, nespbl->pbl_pbase);
                        kfree(nespbl);
                }

                nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                kfree(nescq);
                return ERR_PTR(-ENOMEM);
        }
        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;

        opcode = NES_CQP_CREATE_CQ | NES_CQP_CQ_CEQ_VALID |
                        NES_CQP_CQ_CHK_OVERFLOW |
                        NES_CQP_CQ_CEQE_MASK | ((u32)nescq->hw_cq.cq_size << 16);

        spin_lock_irqsave(&nesadapter->pbl_lock, flags);

        if (pbl_entries != 1) {
                if (pbl_entries > 32) {
                        /* use 4k pbl */
                        nes_debug(NES_DBG_CQ, "pbl_entries=%u, use a 4k PBL\n", pbl_entries);
                        if (nesadapter->free_4kpbl == 0) {
                                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                                nes_free_cqp_request(nesdev, cqp_request);
                                if (!context)
                                        pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
                                                        nescq->hw_cq.cq_pbase);
                                else {
                                        pci_free_consistent(nesdev->pcidev, nespbl->pbl_size,
                                                            nespbl->pbl_vbase, nespbl->pbl_pbase);
                                        kfree(nespbl);
                                }
                                nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                                kfree(nescq);
                                return ERR_PTR(-ENOMEM);
                        } else {
                                opcode |= (NES_CQP_CQ_VIRT | NES_CQP_CQ_4KB_CHUNK);
                                nescq->virtual_cq = 2;
                                nesadapter->free_4kpbl--;
                        }
                } else {
                        /* use 256 byte pbl */
                        nes_debug(NES_DBG_CQ, "pbl_entries=%u, use a 256 byte PBL\n", pbl_entries);
                        if (nesadapter->free_256pbl == 0) {
                                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                                nes_free_cqp_request(nesdev, cqp_request);
                                if (!context)
                                        pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
                                                        nescq->hw_cq.cq_pbase);
                                else {
                                        pci_free_consistent(nesdev->pcidev, nespbl->pbl_size,
                                                            nespbl->pbl_vbase, nespbl->pbl_pbase);
                                        kfree(nespbl);
                                }
                                nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                                kfree(nescq);
                                return ERR_PTR(-ENOMEM);
                        } else {
                                opcode |= NES_CQP_CQ_VIRT;
                                nescq->virtual_cq = 1;
                                nesadapter->free_256pbl--;
                        }
                }
        }

        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);

        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX,
                        (nescq->hw_cq.cq_number | ((u32)nesdev->ceq_index << 16)));

        if (context) {
                if (pbl_entries != 1)
                        u64temp = (u64)nespbl->pbl_pbase;
                else
                        u64temp = le64_to_cpu(nespbl->pbl_vbase[0]);
                set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_CQ_WQE_DOORBELL_INDEX_HIGH_IDX,
                                nes_ucontext->mmap_db_index[0]);
        } else {
                u64temp = (u64)nescq->hw_cq.cq_pbase;
                cqp_wqe->wqe_words[NES_CQP_CQ_WQE_DOORBELL_INDEX_HIGH_IDX] = 0;
        }
        set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_CQ_WQE_PBL_LOW_IDX, u64temp);
        cqp_wqe->wqe_words[NES_CQP_CQ_WQE_CQ_CONTEXT_HIGH_IDX] = 0;
        u64temp = (u64)(unsigned long)&nescq->hw_cq;
        cqp_wqe->wqe_words[NES_CQP_CQ_WQE_CQ_CONTEXT_LOW_IDX] =
                        cpu_to_le32((u32)(u64temp >> 1));
        cqp_wqe->wqe_words[NES_CQP_CQ_WQE_CQ_CONTEXT_HIGH_IDX] =
                        cpu_to_le32(((u32)((u64temp) >> 33)) & 0x7FFFFFFF);

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        nes_debug(NES_DBG_CQ, "Waiting for create iWARP CQ%u to complete.\n",
                        nescq->hw_cq.cq_number);
        ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
                        NES_EVENT_TIMEOUT * 2);
        nes_debug(NES_DBG_CQ, "Create iWARP CQ%u completed, wait_event_timeout ret = %d.\n",
                        nescq->hw_cq.cq_number, ret);
        if ((!ret) || (cqp_request->major_code)) {
                nes_put_cqp_request(nesdev, cqp_request);
                if (!context)
                        pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
                                        nescq->hw_cq.cq_pbase);
                else {
                        pci_free_consistent(nesdev->pcidev, nespbl->pbl_size,
                                            nespbl->pbl_vbase, nespbl->pbl_pbase);
                        kfree(nespbl);
                }
                nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                kfree(nescq);
                return ERR_PTR(-EIO);
        }
        nes_put_cqp_request(nesdev, cqp_request);

        if (context) {
                /* free the nespbl */
                pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
                                nespbl->pbl_pbase);
                kfree(nespbl);
                resp.cq_id = nescq->hw_cq.cq_number;
                resp.cq_size = nescq->hw_cq.cq_size;
                resp.mmap_db_index = 0;
                if (ib_copy_to_udata(udata, &resp, sizeof resp)) {
                        nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
                        kfree(nescq);
                        return ERR_PTR(-EFAULT);
                }
        }

        return &nescq->ibcq;
}


/**
 * nes_destroy_cq
 */
static int nes_destroy_cq(struct ib_cq *ib_cq)
{
        struct nes_cq *nescq;
        struct nes_device *nesdev;
        struct nes_vnic *nesvnic;
        struct nes_adapter *nesadapter;
        struct nes_hw_cqp_wqe *cqp_wqe;
        struct nes_cqp_request *cqp_request;
        unsigned long flags;
        u32 opcode = 0;
        int ret;

        if (ib_cq == NULL)
                return 0;

        nescq = to_nescq(ib_cq);
        nesvnic = to_nesvnic(ib_cq->device);
        nesdev = nesvnic->nesdev;
        nesadapter = nesdev->nesadapter;

        nes_debug(NES_DBG_CQ, "Destroy CQ%u\n", nescq->hw_cq.cq_number);

        /* Send DestroyCQ request to CQP */
        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_CQ, "Failed to get a cqp_request.\n");
                return -ENOMEM;
        }
        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;
        opcode = NES_CQP_DESTROY_CQ | (nescq->hw_cq.cq_size << 16);
        spin_lock_irqsave(&nesadapter->pbl_lock, flags);
        if (nescq->virtual_cq == 1) {
                nesadapter->free_256pbl++;
                if (nesadapter->free_256pbl > nesadapter->max_256pbl) {
                        printk(KERN_ERR PFX "%s: free 256B PBLs(%u) has exceeded the max(%u)\n",
                                        __func__, nesadapter->free_256pbl, nesadapter->max_256pbl);
                }
        } else if (nescq->virtual_cq == 2) {
                nesadapter->free_4kpbl++;
                if (nesadapter->free_4kpbl > nesadapter->max_4kpbl) {
                        printk(KERN_ERR PFX "%s: free 4K PBLs(%u) has exceeded the max(%u)\n",
                                        __func__, nesadapter->free_4kpbl, nesadapter->max_4kpbl);
                }
                opcode |= NES_CQP_CQ_4KB_CHUNK;
        }

        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);

        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX,
                (nescq->hw_cq.cq_number | ((u32)PCI_FUNC(nesdev->pcidev->devfn) << 16)));
        if (!nescq->mcrqf)
                nes_free_resource(nesadapter, nesadapter->allocated_cqs, nescq->hw_cq.cq_number);

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        nes_debug(NES_DBG_CQ, "Waiting for destroy iWARP CQ%u to complete.\n",
                        nescq->hw_cq.cq_number);
        ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
                        NES_EVENT_TIMEOUT);
        nes_debug(NES_DBG_CQ, "Destroy iWARP CQ%u completed, wait_event_timeout ret = %u,"
                        " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                        nescq->hw_cq.cq_number, ret, cqp_request->major_code,
                        cqp_request->minor_code);
        if (!ret) {
                nes_debug(NES_DBG_CQ, "iWARP CQ%u destroy timeout expired\n",
                                        nescq->hw_cq.cq_number);
                ret = -ETIME;
        } else if (cqp_request->major_code) {
                nes_debug(NES_DBG_CQ, "iWARP CQ%u destroy failed\n",
                                        nescq->hw_cq.cq_number);
                ret = -EIO;
        } else {
                ret = 0;
        }
        nes_put_cqp_request(nesdev, cqp_request);

        if (nescq->cq_mem_size)
                pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size,
                                    nescq->hw_cq.cq_vbase, nescq->hw_cq.cq_pbase);
        kfree(nescq);

        return ret;
}

/**
 * root_256
 */
static u32 root_256(struct nes_device *nesdev,
                    struct nes_root_vpbl *root_vpbl,
                    struct nes_root_vpbl *new_root,
                    u16 pbl_count_4k)
{
        u64 leaf_pbl;
        int i, j, k;

        if (pbl_count_4k == 1) {
                new_root->pbl_vbase = pci_alloc_consistent(nesdev->pcidev,
                                                512, &new_root->pbl_pbase);

                if (new_root->pbl_vbase == NULL)
                        return 0;

                leaf_pbl = (u64)root_vpbl->pbl_pbase;
                for (i = 0; i < 16; i++) {
                        new_root->pbl_vbase[i].pa_low =
                                cpu_to_le32((u32)leaf_pbl);
                        new_root->pbl_vbase[i].pa_high =
                                cpu_to_le32((u32)((((u64)leaf_pbl) >> 32)));
                        leaf_pbl += 256;
                }
        } else {
                for (i = 3; i >= 0; i--) {
                        j = i * 16;

                        root_vpbl->pbl_vbase[j] = root_vpbl->pbl_vbase[i];
                        leaf_pbl = le32_to_cpu(root_vpbl->pbl_vbase[j].pa_low) +
                            (((u64)le32_to_cpu(root_vpbl->pbl_vbase[j].pa_high))
                                << 32);
                        for (k = 1; k < 16; k++) {
                                leaf_pbl += 256;
                                root_vpbl->pbl_vbase[j + k].pa_low =
                                                cpu_to_le32((u32)leaf_pbl);
                                root_vpbl->pbl_vbase[j + k].pa_high =
                                    cpu_to_le32((u32)((((u64)leaf_pbl) >> 32)));
                        }
                }
        }

        return 1;
}


/**
 * nes_reg_mr
 */
static int nes_reg_mr(struct nes_device *nesdev, struct nes_pd *nespd,
                u32 stag, u64 region_length, struct nes_root_vpbl *root_vpbl,
                dma_addr_t single_buffer, u16 pbl_count_4k,
                u16 residual_page_count_4k, int acc, u64 *iova_start,
                u16 *actual_pbl_cnt, u8 *used_4k_pbls)
{
        struct nes_hw_cqp_wqe *cqp_wqe;
        struct nes_cqp_request *cqp_request;
        unsigned long flags;
        int ret;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        uint pg_cnt = 0;
        u16 pbl_count_256 = 0;
        u16 pbl_count = 0;
        u8  use_256_pbls = 0;
        u8  use_4k_pbls = 0;
        u16 use_two_level = (pbl_count_4k > 1) ? 1 : 0;
        struct nes_root_vpbl new_root = {0, 0, 0};
        u32 opcode = 0;
        u16 major_code;

        /* Register the region with the adapter */
        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
                return -ENOMEM;
        }
        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;

        if (pbl_count_4k) {
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);

                pg_cnt = ((pbl_count_4k - 1) * 512) + residual_page_count_4k;
                pbl_count_256 = (pg_cnt + 31) / 32;
                if (pg_cnt <= 32) {
                        if (pbl_count_256 <= nesadapter->free_256pbl)
                                use_256_pbls = 1;
                        else if (pbl_count_4k <= nesadapter->free_4kpbl)
                                use_4k_pbls = 1;
                } else if (pg_cnt <= 2048) {
                        if (((pbl_count_4k + use_two_level) <= nesadapter->free_4kpbl) &&
                            (nesadapter->free_4kpbl > (nesadapter->max_4kpbl >> 1))) {
                                use_4k_pbls = 1;
                        } else if ((pbl_count_256 + 1) <= nesadapter->free_256pbl) {
                                use_256_pbls = 1;
                                use_two_level = 1;
                        } else if ((pbl_count_4k + use_two_level) <= nesadapter->free_4kpbl) {
                                use_4k_pbls = 1;
                        }
                } else {
                        if ((pbl_count_4k + 1) <= nesadapter->free_4kpbl)
                                use_4k_pbls = 1;
                }

                if (use_256_pbls) {
                        pbl_count = pbl_count_256;
                        nesadapter->free_256pbl -= pbl_count + use_two_level;
                } else if (use_4k_pbls) {
                        pbl_count =  pbl_count_4k;
                        nesadapter->free_4kpbl -= pbl_count + use_two_level;
                } else {
                        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
                        nes_debug(NES_DBG_MR, "Out of Pbls\n");
                        nes_free_cqp_request(nesdev, cqp_request);
                        return -ENOMEM;
                }

                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
        }

        if (use_256_pbls && use_two_level) {
                if (root_256(nesdev, root_vpbl, &new_root, pbl_count_4k) == 1) {
                        if (new_root.pbl_pbase != 0)
                                root_vpbl = &new_root;
                } else {
                        spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                        nesadapter->free_256pbl += pbl_count_256 + use_two_level;
                        use_256_pbls = 0;

                        if (pbl_count_4k == 1)
                                use_two_level = 0;
                        pbl_count = pbl_count_4k;

                        if ((pbl_count_4k + use_two_level) <= nesadapter->free_4kpbl) {
                                nesadapter->free_4kpbl -= pbl_count + use_two_level;
                                use_4k_pbls = 1;
                        }
                        spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);

                        if (use_4k_pbls == 0)
                                return -ENOMEM;
                }
        }

        opcode = NES_CQP_REGISTER_STAG | NES_CQP_STAG_RIGHTS_LOCAL_READ |
                                        NES_CQP_STAG_VA_TO | NES_CQP_STAG_MR;
        if (acc & IB_ACCESS_LOCAL_WRITE)
                opcode |= NES_CQP_STAG_RIGHTS_LOCAL_WRITE;
        if (acc & IB_ACCESS_REMOTE_WRITE)
                opcode |= NES_CQP_STAG_RIGHTS_REMOTE_WRITE | NES_CQP_STAG_REM_ACC_EN;
        if (acc & IB_ACCESS_REMOTE_READ)
                opcode |= NES_CQP_STAG_RIGHTS_REMOTE_READ | NES_CQP_STAG_REM_ACC_EN;
        if (acc & IB_ACCESS_MW_BIND)
                opcode |= NES_CQP_STAG_RIGHTS_WINDOW_BIND | NES_CQP_STAG_REM_ACC_EN;

        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
        set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_VA_LOW_IDX, *iova_start);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_LEN_LOW_IDX, region_length);

        cqp_wqe->wqe_words[NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX] =
                        cpu_to_le32((u32)(region_length >> 8) & 0xff000000);
        cqp_wqe->wqe_words[NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX] |=
                        cpu_to_le32(nespd->pd_id & 0x00007fff);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, stag);

        if (pbl_count == 0) {
                set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PA_LOW_IDX, single_buffer);
        } else {
                set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PA_LOW_IDX, root_vpbl->pbl_pbase);
                set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PBL_BLK_COUNT_IDX, pbl_count);
                set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PBL_LEN_IDX, (pg_cnt * 8));

                if (use_4k_pbls)
                        cqp_wqe->wqe_words[NES_CQP_WQE_OPCODE_IDX] |= cpu_to_le32(NES_CQP_STAG_PBL_BLK_SIZE);
        }
        barrier();

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
                        NES_EVENT_TIMEOUT);
        nes_debug(NES_DBG_MR, "Register STag 0x%08X completed, wait_event_timeout ret = %u,"
                        " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                        stag, ret, cqp_request->major_code, cqp_request->minor_code);
        major_code = cqp_request->major_code;
        nes_put_cqp_request(nesdev, cqp_request);

        if ((!ret || major_code) && pbl_count != 0) {
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                if (use_256_pbls)
                        nesadapter->free_256pbl += pbl_count + use_two_level;
                else if (use_4k_pbls)
                        nesadapter->free_4kpbl += pbl_count + use_two_level;
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
        }
        if (new_root.pbl_pbase)
                pci_free_consistent(nesdev->pcidev, 512, new_root.pbl_vbase,
                                    new_root.pbl_pbase);

        if (!ret)
                return -ETIME;
        else if (major_code)
                return -EIO;

        *actual_pbl_cnt = pbl_count + use_two_level;
        *used_4k_pbls = use_4k_pbls;
        return 0;
}


/**
 * nes_reg_phys_mr
 */
static struct ib_mr *nes_reg_phys_mr(struct ib_pd *ib_pd,
                struct ib_phys_buf *buffer_list, int num_phys_buf, int acc,
                u64 * iova_start)
{
        u64 region_length;
        struct nes_pd *nespd = to_nespd(ib_pd);
        struct nes_vnic *nesvnic = to_nesvnic(ib_pd->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_mr *nesmr;
        struct ib_mr *ibmr;
        struct nes_vpbl vpbl;
        struct nes_root_vpbl root_vpbl;
        u32 stag;
        u32 i;
        unsigned long mask;
        u32 stag_index = 0;
        u32 next_stag_index = 0;
        u32 driver_key = 0;
        u32 root_pbl_index = 0;
        u32 cur_pbl_index = 0;
        int err = 0, pbl_depth = 0;
        int ret = 0;
        u16 pbl_count = 0;
        u8 single_page = 1;
        u8 stag_key = 0;

        pbl_depth = 0;
        region_length = 0;
        vpbl.pbl_vbase = NULL;
        root_vpbl.pbl_vbase = NULL;
        root_vpbl.pbl_pbase = 0;

        get_random_bytes(&next_stag_index, sizeof(next_stag_index));
        stag_key = (u8)next_stag_index;

        driver_key = 0;

        next_stag_index >>= 8;
        next_stag_index %= nesadapter->max_mr;
        if (num_phys_buf > (1024*512)) {
                return ERR_PTR(-E2BIG);
        }

        if ((buffer_list[0].addr ^ *iova_start) & ~PAGE_MASK)
                return ERR_PTR(-EINVAL);

        err = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs, nesadapter->max_mr,
                        &stag_index, &next_stag_index);
        if (err) {
                return ERR_PTR(err);
        }

        nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
        if (!nesmr) {
                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                return ERR_PTR(-ENOMEM);
        }

        for (i = 0; i < num_phys_buf; i++) {

                if ((i & 0x01FF) == 0) {
                        if (root_pbl_index == 1) {
                                /* Allocate the root PBL */
                                root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 8192,
                                                &root_vpbl.pbl_pbase);
                                nes_debug(NES_DBG_MR, "Allocating root PBL, va = %p, pa = 0x%08X\n",
                                                root_vpbl.pbl_vbase, (unsigned int)root_vpbl.pbl_pbase);
                                if (!root_vpbl.pbl_vbase) {
                                        pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
                                                        vpbl.pbl_pbase);
                                        nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                                        kfree(nesmr);
                                        return ERR_PTR(-ENOMEM);
                                }
                                root_vpbl.leaf_vpbl = kzalloc(sizeof(*root_vpbl.leaf_vpbl)*1024, GFP_KERNEL);
                                if (!root_vpbl.leaf_vpbl) {
                                        pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
                                                        root_vpbl.pbl_pbase);
                                        pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
                                                        vpbl.pbl_pbase);
                                        nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                                        kfree(nesmr);
                                        return ERR_PTR(-ENOMEM);
                                }
                                root_vpbl.pbl_vbase[0].pa_low = cpu_to_le32((u32)vpbl.pbl_pbase);
                                root_vpbl.pbl_vbase[0].pa_high =
                                                cpu_to_le32((u32)((((u64)vpbl.pbl_pbase) >> 32)));
                                root_vpbl.leaf_vpbl[0] = vpbl;
                        }
                        /* Allocate a 4K buffer for the PBL */
                        vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
                                        &vpbl.pbl_pbase);
                        nes_debug(NES_DBG_MR, "Allocating leaf PBL, va = %p, pa = 0x%016lX\n",
                                        vpbl.pbl_vbase, (unsigned long)vpbl.pbl_pbase);
                        if (!vpbl.pbl_vbase) {
                                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                                ibmr = ERR_PTR(-ENOMEM);
                                kfree(nesmr);
                                goto reg_phys_err;
                        }
                        /* Fill in the root table */
                        if (1 <= root_pbl_index) {
                                root_vpbl.pbl_vbase[root_pbl_index].pa_low =
                                                cpu_to_le32((u32)vpbl.pbl_pbase);
                                root_vpbl.pbl_vbase[root_pbl_index].pa_high =
                                                cpu_to_le32((u32)((((u64)vpbl.pbl_pbase) >> 32)));
                                root_vpbl.leaf_vpbl[root_pbl_index] = vpbl;
                        }
                        root_pbl_index++;
                        cur_pbl_index = 0;
                }

                mask = !buffer_list[i].size;
                if (i != 0)
                        mask |= buffer_list[i].addr;
                if (i != num_phys_buf - 1)
                        mask |= buffer_list[i].addr + buffer_list[i].size;

                if (mask & ~PAGE_MASK) {
                        nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                        nes_debug(NES_DBG_MR, "Invalid buffer addr or size\n");
                        ibmr = ERR_PTR(-EINVAL);
                        kfree(nesmr);
                        goto reg_phys_err;
                }

                region_length += buffer_list[i].size;
                if ((i != 0) && (single_page)) {
                        if ((buffer_list[i-1].addr+PAGE_SIZE) != buffer_list[i].addr)
                                single_page = 0;
                }
                vpbl.pbl_vbase[cur_pbl_index].pa_low = cpu_to_le32((u32)buffer_list[i].addr & PAGE_MASK);
                vpbl.pbl_vbase[cur_pbl_index++].pa_high =
                                cpu_to_le32((u32)((((u64)buffer_list[i].addr) >> 32)));
        }

        stag = stag_index << 8;
        stag |= driver_key;
        stag += (u32)stag_key;

        nes_debug(NES_DBG_MR, "Registering STag 0x%08X, VA = 0x%016lX,"
                        " length = 0x%016lX, index = 0x%08X\n",
                        stag, (unsigned long)*iova_start, (unsigned long)region_length, stag_index);

        /* Make the leaf PBL the root if only one PBL */
        if (root_pbl_index == 1) {
                root_vpbl.pbl_pbase = vpbl.pbl_pbase;
        }

        if (single_page) {
                pbl_count = 0;
        } else {
                pbl_count = root_pbl_index;
        }
        ret = nes_reg_mr(nesdev, nespd, stag, region_length, &root_vpbl,
                        buffer_list[0].addr, pbl_count, (u16)cur_pbl_index, acc, iova_start,
                        &nesmr->pbls_used, &nesmr->pbl_4k);

        if (ret == 0) {
                nesmr->ibmr.rkey = stag;
                nesmr->ibmr.lkey = stag;
                nesmr->mode = IWNES_MEMREG_TYPE_MEM;
                ibmr = &nesmr->ibmr;
        } else {
                kfree(nesmr);
                ibmr = ERR_PTR(-ENOMEM);
        }

        reg_phys_err:
        /* free the resources */
        if (root_pbl_index == 1) {
                /* single PBL case */
                pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase, vpbl.pbl_pbase);
        } else {
                for (i=0; i<root_pbl_index; i++) {
                        pci_free_consistent(nesdev->pcidev, 4096, root_vpbl.leaf_vpbl[i].pbl_vbase,
                                        root_vpbl.leaf_vpbl[i].pbl_pbase);
                }
                kfree(root_vpbl.leaf_vpbl);
                pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
                                root_vpbl.pbl_pbase);
        }

        return ibmr;
}


/**
 * nes_get_dma_mr
 */
static struct ib_mr *nes_get_dma_mr(struct ib_pd *pd, int acc)
{
        struct ib_phys_buf bl;
        u64 kva = 0;

        nes_debug(NES_DBG_MR, "\n");

        bl.size = (u64)0xffffffffffULL;
        bl.addr = 0;
        return nes_reg_phys_mr(pd, &bl, 1, acc, &kva);
}


/**
 * nes_reg_user_mr
 */
static struct ib_mr *nes_reg_user_mr(struct ib_pd *pd, u64 start, u64 length,
                u64 virt, int acc, struct ib_udata *udata)
{
        u64 iova_start;
        __le64 *pbl;
        u64 region_length;
        dma_addr_t last_dma_addr = 0;
        dma_addr_t first_dma_addr = 0;
        struct nes_pd *nespd = to_nespd(pd);
        struct nes_vnic *nesvnic = to_nesvnic(pd->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct ib_mr *ibmr = ERR_PTR(-EINVAL);
        struct ib_umem_chunk *chunk;
        struct nes_ucontext *nes_ucontext;
        struct nes_pbl *nespbl;
        struct nes_mr *nesmr;
        struct ib_umem *region;
        struct nes_mem_reg_req req;
        struct nes_vpbl vpbl;
        struct nes_root_vpbl root_vpbl;
        int nmap_index, page_index;
        int page_count = 0;
        int err, pbl_depth = 0;
        int chunk_pages;
        int ret;
        u32 stag;
        u32 stag_index = 0;
        u32 next_stag_index;
        u32 driver_key;
        u32 root_pbl_index = 0;
        u32 cur_pbl_index = 0;
        u32 skip_pages;
        u16 pbl_count;
        u8 single_page = 1;
        u8 stag_key;

        region = ib_umem_get(pd->uobject->context, start, length, acc, 0);
        if (IS_ERR(region)) {
                return (struct ib_mr *)region;
        }

        nes_debug(NES_DBG_MR, "User base = 0x%lX, Virt base = 0x%lX, length = %u,"
                        " offset = %u, page size = %u.\n",
                        (unsigned long int)start, (unsigned long int)virt, (u32)length,
                        region->offset, region->page_size);

        skip_pages = ((u32)region->offset) >> 12;

        if (ib_copy_from_udata(&req, udata, sizeof(req)))
                return ERR_PTR(-EFAULT);
        nes_debug(NES_DBG_MR, "Memory Registration type = %08X.\n", req.reg_type);

        switch (req.reg_type) {
                case IWNES_MEMREG_TYPE_MEM:
                        pbl_depth = 0;
                        region_length = 0;
                        vpbl.pbl_vbase = NULL;
                        root_vpbl.pbl_vbase = NULL;
                        root_vpbl.pbl_pbase = 0;

                        get_random_bytes(&next_stag_index, sizeof(next_stag_index));
                        stag_key = (u8)next_stag_index;

                        driver_key = next_stag_index & 0x70000000;

                        next_stag_index >>= 8;
                        next_stag_index %= nesadapter->max_mr;

                        err = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs,
                                        nesadapter->max_mr, &stag_index, &next_stag_index);
                        if (err) {
                                ib_umem_release(region);
                                return ERR_PTR(err);
                        }

                        nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
                        if (!nesmr) {
                                ib_umem_release(region);
                                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                                return ERR_PTR(-ENOMEM);
                        }
                        nesmr->region = region;

                        list_for_each_entry(chunk, &region->chunk_list, list) {
                                nes_debug(NES_DBG_MR, "Chunk: nents = %u, nmap = %u .\n",
                                                chunk->nents, chunk->nmap);
                                for (nmap_index = 0; nmap_index < chunk->nmap; ++nmap_index) {
                                        if (sg_dma_address(&chunk->page_list[nmap_index]) & ~PAGE_MASK) {
                                                ib_umem_release(region);
                                                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                                                nes_debug(NES_DBG_MR, "Unaligned Memory Buffer: 0x%x\n",
                                                                (unsigned int) sg_dma_address(&chunk->page_list[nmap_index]));
                                                ibmr = ERR_PTR(-EINVAL);
                                                kfree(nesmr);
                                                goto reg_user_mr_err;
                                        }

                                        if (!sg_dma_len(&chunk->page_list[nmap_index])) {
                                                ib_umem_release(region);
                                                nes_free_resource(nesadapter, nesadapter->allocated_mrs,
                                                                stag_index);
                                                nes_debug(NES_DBG_MR, "Invalid Buffer Size\n");
                                                ibmr = ERR_PTR(-EINVAL);
                                                kfree(nesmr);
                                                goto reg_user_mr_err;
                                        }

                                        region_length += sg_dma_len(&chunk->page_list[nmap_index]);
                                        chunk_pages = sg_dma_len(&chunk->page_list[nmap_index]) >> 12;
                                        region_length -= skip_pages << 12;
                                        for (page_index=skip_pages; page_index < chunk_pages; page_index++) {
                                                skip_pages = 0;
                                                if ((page_count!=0)&&(page_count<<12)-(region->offset&(4096-1))>=region->length)
                                                        goto enough_pages;
                                                if ((page_count&0x01FF) == 0) {
                                                        if (page_count >= 1024 * 512) {
                                                                ib_umem_release(region);
                                                                nes_free_resource(nesadapter,
                                                                                nesadapter->allocated_mrs, stag_index);
                                                                kfree(nesmr);
                                                                ibmr = ERR_PTR(-E2BIG);
                                                                goto reg_user_mr_err;
                                                        }
                                                        if (root_pbl_index == 1) {
                                                                root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev,
                                                                                8192, &root_vpbl.pbl_pbase);
                                                                nes_debug(NES_DBG_MR, "Allocating root PBL, va = %p, pa = 0x%08X\n",
                                                                                root_vpbl.pbl_vbase, (unsigned int)root_vpbl.pbl_pbase);
                                                                if (!root_vpbl.pbl_vbase) {
                                                                        ib_umem_release(region);
                                                                        pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
                                                                                        vpbl.pbl_pbase);
                                                                        nes_free_resource(nesadapter, nesadapter->allocated_mrs,
                                                                                        stag_index);
                                                                        kfree(nesmr);
                                                                        ibmr = ERR_PTR(-ENOMEM);
                                                                        goto reg_user_mr_err;
                                                                }
                                                                root_vpbl.leaf_vpbl = kzalloc(sizeof(*root_vpbl.leaf_vpbl)*1024,
                                                                                GFP_KERNEL);
                                                                if (!root_vpbl.leaf_vpbl) {
                                                                        ib_umem_release(region);
                                                                        pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
                                                                                        root_vpbl.pbl_pbase);
                                                                        pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
                                                                                        vpbl.pbl_pbase);
                                                                        nes_free_resource(nesadapter, nesadapter->allocated_mrs,
                                                                                        stag_index);
                                                                        kfree(nesmr);
                                                                        ibmr = ERR_PTR(-ENOMEM);
                                                                        goto reg_user_mr_err;
                                                                }
                                                                root_vpbl.pbl_vbase[0].pa_low =
                                                                                cpu_to_le32((u32)vpbl.pbl_pbase);
                                                                root_vpbl.pbl_vbase[0].pa_high =
                                                                                cpu_to_le32((u32)((((u64)vpbl.pbl_pbase) >> 32)));
                                                                root_vpbl.leaf_vpbl[0] = vpbl;
                                                        }
                                                        vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
                                                                        &vpbl.pbl_pbase);
                                                        nes_debug(NES_DBG_MR, "Allocating leaf PBL, va = %p, pa = 0x%08X\n",
                                                                        vpbl.pbl_vbase, (unsigned int)vpbl.pbl_pbase);
                                                        if (!vpbl.pbl_vbase) {
                                                                ib_umem_release(region);
                                                                nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
                                                                ibmr = ERR_PTR(-ENOMEM);
                                                                kfree(nesmr);
                                                                goto reg_user_mr_err;
                                                        }
                                                        if (1 <= root_pbl_index) {
                                                                root_vpbl.pbl_vbase[root_pbl_index].pa_low =
                                                                                cpu_to_le32((u32)vpbl.pbl_pbase);
                                                                root_vpbl.pbl_vbase[root_pbl_index].pa_high =
                                                                                cpu_to_le32((u32)((((u64)vpbl.pbl_pbase)>>32)));
                                                                root_vpbl.leaf_vpbl[root_pbl_index] = vpbl;
                                                        }
                                                        root_pbl_index++;
                                                        cur_pbl_index = 0;
                                                }
                                                if (single_page) {
                                                        if (page_count != 0) {
                                                                if ((last_dma_addr+4096) !=
                                                                                (sg_dma_address(&chunk->page_list[nmap_index])+
                                                                                (page_index*4096)))
                                                                        single_page = 0;
                                                                last_dma_addr = sg_dma_address(&chunk->page_list[nmap_index])+
                                                                                (page_index*4096);
                                                        } else {
                                                                first_dma_addr = sg_dma_address(&chunk->page_list[nmap_index])+
                                                                                (page_index*4096);
                                                                last_dma_addr = first_dma_addr;
                                                        }
                                                }

                                                vpbl.pbl_vbase[cur_pbl_index].pa_low =
                                                                cpu_to_le32((u32)(sg_dma_address(&chunk->page_list[nmap_index])+
                                                                (page_index*4096)));
                                                vpbl.pbl_vbase[cur_pbl_index].pa_high =
                                                                cpu_to_le32((u32)((((u64)(sg_dma_address(&chunk->page_list[nmap_index])+
                                                                (page_index*4096))) >> 32)));
                                                cur_pbl_index++;
                                                page_count++;
                                        }
                                }
                        }
                        enough_pages:
                        nes_debug(NES_DBG_MR, "calculating stag, stag_index=0x%08x, driver_key=0x%08x,"
                                        " stag_key=0x%08x\n",
                                        stag_index, driver_key, stag_key);
                        stag = stag_index << 8;
                        stag |= driver_key;
                        stag += (u32)stag_key;
                        if (stag == 0) {
                                stag = 1;
                        }

                        iova_start = virt;
                        /* Make the leaf PBL the root if only one PBL */
                        if (root_pbl_index == 1) {
                                root_vpbl.pbl_pbase = vpbl.pbl_pbase;
                        }

                        if (single_page) {
                                pbl_count = 0;
                        } else {
                                pbl_count = root_pbl_index;
                                first_dma_addr = 0;
                        }
                        nes_debug(NES_DBG_MR, "Registering STag 0x%08X, VA = 0x%08X, length = 0x%08X,"
                                        " index = 0x%08X, region->length=0x%08llx, pbl_count = %u\n",
                                        stag, (unsigned int)iova_start,
                                        (unsigned int)region_length, stag_index,
                                        (unsigned long long)region->length, pbl_count);
                        ret = nes_reg_mr(nesdev, nespd, stag, region->length, &root_vpbl,
                                         first_dma_addr, pbl_count, (u16)cur_pbl_index, acc,
                                         &iova_start, &nesmr->pbls_used, &nesmr->pbl_4k);

                        nes_debug(NES_DBG_MR, "ret=%d\n", ret);

                        if (ret == 0) {
                                nesmr->ibmr.rkey = stag;
                                nesmr->ibmr.lkey = stag;
                                nesmr->mode = IWNES_MEMREG_TYPE_MEM;
                                ibmr = &nesmr->ibmr;
                        } else {
                                ib_umem_release(region);
                                kfree(nesmr);
                                ibmr = ERR_PTR(-ENOMEM);
                        }

                        reg_user_mr_err:
                        /* free the resources */
                        if (root_pbl_index == 1) {
                                pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
                                                vpbl.pbl_pbase);
                        } else {
                                for (page_index=0; page_index<root_pbl_index; page_index++) {
                                        pci_free_consistent(nesdev->pcidev, 4096,
                                                        root_vpbl.leaf_vpbl[page_index].pbl_vbase,
                                                        root_vpbl.leaf_vpbl[page_index].pbl_pbase);
                                }
                                kfree(root_vpbl.leaf_vpbl);
                                pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
                                                root_vpbl.pbl_pbase);
                        }

                        nes_debug(NES_DBG_MR, "Leaving, ibmr=%p", ibmr);

                        return ibmr;
                case IWNES_MEMREG_TYPE_QP:
                case IWNES_MEMREG_TYPE_CQ:
                        nespbl = kzalloc(sizeof(*nespbl), GFP_KERNEL);
                        if (!nespbl) {
                                nes_debug(NES_DBG_MR, "Unable to allocate PBL\n");
                                ib_umem_release(region);
                                return ERR_PTR(-ENOMEM);
                        }
                        nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
                        if (!nesmr) {
                                ib_umem_release(region);
                                kfree(nespbl);
                                nes_debug(NES_DBG_MR, "Unable to allocate nesmr\n");
                                return ERR_PTR(-ENOMEM);
                        }
                        nesmr->region = region;
                        nes_ucontext = to_nesucontext(pd->uobject->context);
                        pbl_depth = region->length >> 12;
                        pbl_depth += (region->length & (4096-1)) ? 1 : 0;
                        nespbl->pbl_size = pbl_depth*sizeof(u64);
                        if (req.reg_type == IWNES_MEMREG_TYPE_QP) {
                                nes_debug(NES_DBG_MR, "Attempting to allocate QP PBL memory");
                        } else {
                                nes_debug(NES_DBG_MR, "Attempting to allocate CP PBL memory");
                        }

                        nes_debug(NES_DBG_MR, " %u bytes, %u entries.\n",
                                        nespbl->pbl_size, pbl_depth);
                        pbl = pci_alloc_consistent(nesdev->pcidev, nespbl->pbl_size,
                                        &nespbl->pbl_pbase);
                        if (!pbl) {
                                ib_umem_release(region);
                                kfree(nesmr);
                                kfree(nespbl);
                                nes_debug(NES_DBG_MR, "Unable to allocate PBL memory\n");
                                return ERR_PTR(-ENOMEM);
                        }

                        nespbl->pbl_vbase = (u64 *)pbl;
                        nespbl->user_base = start;
                        nes_debug(NES_DBG_MR, "Allocated PBL memory, %u bytes, pbl_pbase=%lx,"
                                        " pbl_vbase=%p user_base=0x%lx\n",
                                  nespbl->pbl_size, (unsigned long) nespbl->pbl_pbase,
                                  (void *) nespbl->pbl_vbase, nespbl->user_base);

                        list_for_each_entry(chunk, &region->chunk_list, list) {
                                for (nmap_index = 0; nmap_index < chunk->nmap; ++nmap_index) {
                                        chunk_pages = sg_dma_len(&chunk->page_list[nmap_index]) >> 12;
                                        chunk_pages += (sg_dma_len(&chunk->page_list[nmap_index]) & (4096-1)) ? 1 : 0;
                                        nespbl->page = sg_page(&chunk->page_list[0]);
                                        for (page_index=0; page_index<chunk_pages; page_index++) {
                                                ((__le32 *)pbl)[0] = cpu_to_le32((u32)
                                                                (sg_dma_address(&chunk->page_list[nmap_index])+
                                                                (page_index*4096)));
                                                ((__le32 *)pbl)[1] = cpu_to_le32(((u64)
                                                                (sg_dma_address(&chunk->page_list[nmap_index])+
                                                                (page_index*4096)))>>32);
                                                nes_debug(NES_DBG_MR, "pbl=%p, *pbl=0x%016llx, 0x%08x%08x\n", pbl,
                                                                (unsigned long long)*pbl,
                                                                le32_to_cpu(((__le32 *)pbl)[1]), le32_to_cpu(((__le32 *)pbl)[0]));
                                                pbl++;
                                        }
                                }
                        }
                        if (req.reg_type == IWNES_MEMREG_TYPE_QP) {
                                list_add_tail(&nespbl->list, &nes_ucontext->qp_reg_mem_list);
                        } else {
                                list_add_tail(&nespbl->list, &nes_ucontext->cq_reg_mem_list);
                        }
                        nesmr->ibmr.rkey = -1;
                        nesmr->ibmr.lkey = -1;
                        nesmr->mode = req.reg_type;
                        return &nesmr->ibmr;
        }

        return ERR_PTR(-ENOSYS);
}


/**
 * nes_dereg_mr
 */
static int nes_dereg_mr(struct ib_mr *ib_mr)
{
        struct nes_mr *nesmr = to_nesmr(ib_mr);
        struct nes_vnic *nesvnic = to_nesvnic(ib_mr->device);
        struct nes_device *nesdev = nesvnic->nesdev;
        struct nes_adapter *nesadapter = nesdev->nesadapter;
        struct nes_hw_cqp_wqe *cqp_wqe;
        struct nes_cqp_request *cqp_request;
        unsigned long flags;
        int ret;
        u16 major_code;
        u16 minor_code;

        if (nesmr->region) {
                ib_umem_release(nesmr->region);
        }
        if (nesmr->mode != IWNES_MEMREG_TYPE_MEM) {
                kfree(nesmr);
                return 0;
        }

        /* Deallocate the region with the adapter */

        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
                return -ENOMEM;
        }
        cqp_request->waiting = 1;
        cqp_wqe = &cqp_request->cqp_wqe;

        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX,
                        NES_CQP_DEALLOCATE_STAG | NES_CQP_STAG_VA_TO |
                        NES_CQP_STAG_DEALLOC_PBLS | NES_CQP_STAG_MR);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, ib_mr->rkey);

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        nes_debug(NES_DBG_MR, "Waiting for deallocate STag 0x%08X completed\n", ib_mr->rkey);
        ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
                        NES_EVENT_TIMEOUT);
        nes_debug(NES_DBG_MR, "Deallocate STag 0x%08X completed, wait_event_timeout ret = %u,"
                        " CQP Major:Minor codes = 0x%04X:0x%04X\n",
                        ib_mr->rkey, ret, cqp_request->major_code, cqp_request->minor_code);

        major_code = cqp_request->major_code;
        minor_code = cqp_request->minor_code;

        nes_put_cqp_request(nesdev, cqp_request);

        if (!ret) {
                nes_debug(NES_DBG_MR, "Timeout waiting to destroy STag,"
                                " ib_mr=%p, rkey = 0x%08X\n",
                                ib_mr, ib_mr->rkey);
                return -ETIME;
        } else if (major_code) {
                nes_debug(NES_DBG_MR, "Error (0x%04X:0x%04X) while attempting"
                                " to destroy STag, ib_mr=%p, rkey = 0x%08X\n",
                                major_code, minor_code, ib_mr, ib_mr->rkey);
                return -EIO;
        }

        if (nesmr->pbls_used != 0) {
                spin_lock_irqsave(&nesadapter->pbl_lock, flags);
                if (nesmr->pbl_4k) {
                        nesadapter->free_4kpbl += nesmr->pbls_used;
                        if (nesadapter->free_4kpbl > nesadapter->max_4kpbl)
                                printk(KERN_ERR PFX "free 4KB PBLs(%u) has "
                                        "exceeded the max(%u)\n",
                                        nesadapter->free_4kpbl,
                                        nesadapter->max_4kpbl);
                } else {
                        nesadapter->free_256pbl += nesmr->pbls_used;
                        if (nesadapter->free_256pbl > nesadapter->max_256pbl)
                                printk(KERN_ERR PFX "free 256B PBLs(%u) has "
                                        "exceeded the max(%u)\n",
                                        nesadapter->free_256pbl,
                                        nesadapter->max_256pbl);
                }
                spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
        }
        nes_free_resource(nesadapter, nesadapter->allocated_mrs,
                        (ib_mr->rkey & 0x0fffff00) >> 8);

        kfree(nesmr);

        return 0;
}


/**
 * show_rev
 */
static ssize_t show_rev(struct device *dev, struct device_attribute *attr,
                        char *buf)
{
        struct nes_ib_device *nesibdev =
                        container_of(dev, struct nes_ib_device, ibdev.dev);
        struct nes_vnic *nesvnic = nesibdev->nesvnic;

        nes_debug(NES_DBG_INIT, "\n");
        return sprintf(buf, "%x\n", nesvnic->nesdev->nesadapter->hw_rev);
}


/**
 * show_fw_ver
 */
static ssize_t show_fw_ver(struct device *dev, struct device_attribute *attr,
                           char *buf)
{
        struct nes_ib_device *nesibdev =
                        container_of(dev, struct nes_ib_device, ibdev.dev);
        struct nes_vnic *nesvnic = nesibdev->nesvnic;

        nes_debug(NES_DBG_INIT, "\n");
        return sprintf(buf, "%u.%u\n",
                (nesvnic->nesdev->nesadapter->firmware_version >> 16),
                (nesvnic->nesdev->nesadapter->firmware_version & 0x000000ff));
}


/**
 * show_hca
 */
static ssize_t show_hca(struct device *dev, struct device_attribute *attr,
                        char *buf)
{
        nes_debug(NES_DBG_INIT, "\n");
        return sprintf(buf, "NES020\n");
}


/**
 * show_board
 */
static ssize_t show_board(struct device *dev, struct device_attribute *attr,
                          char *buf)
{
        nes_debug(NES_DBG_INIT, "\n");
        return sprintf(buf, "%.*s\n", 32, "NES020 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 *nes_dev_attributes[] = {
        &dev_attr_hw_rev,
        &dev_attr_fw_ver,
        &dev_attr_hca_type,
        &dev_attr_board_id
};


/**
 * nes_query_qp
 */
static int nes_query_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
                int attr_mask, struct ib_qp_init_attr *init_attr)
{
        struct nes_qp *nesqp = to_nesqp(ibqp);

        nes_debug(NES_DBG_QP, "\n");

        attr->qp_access_flags = 0;
        attr->cap.max_send_wr = nesqp->hwqp.sq_size;
        attr->cap.max_recv_wr = nesqp->hwqp.rq_size;
        attr->cap.max_recv_sge = 1;
        if (nes_drv_opt & NES_DRV_OPT_NO_INLINE_DATA) {
                init_attr->cap.max_inline_data = 0;
        } else {
                init_attr->cap.max_inline_data = 64;
        }

        init_attr->event_handler = nesqp->ibqp.event_handler;
        init_attr->qp_context = nesqp->ibqp.qp_context;
        init_attr->send_cq = nesqp->ibqp.send_cq;
        init_attr->recv_cq = nesqp->ibqp.recv_cq;
        init_attr->srq = nesqp->ibqp.srq = nesqp->ibqp.srq;
        init_attr->cap = attr->cap;

        return 0;
}


/**
 * nes_hw_modify_qp
 */
int nes_hw_modify_qp(struct nes_device *nesdev, struct nes_qp *nesqp,
                u32 next_iwarp_state, u32 termlen, u32 wait_completion)
{
        struct nes_hw_cqp_wqe *cqp_wqe;
        /* struct iw_cm_id *cm_id = nesqp->cm_id; */
        /* struct iw_cm_event cm_event; */
        struct nes_cqp_request *cqp_request;
        int ret;
        u16 major_code;

        nes_debug(NES_DBG_MOD_QP, "QP%u, refcount=%d\n",
                        nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount));

        cqp_request = nes_get_cqp_request(nesdev);
        if (cqp_request == NULL) {
                nes_debug(NES_DBG_MOD_QP, "Failed to get a cqp_request.\n");
                return -ENOMEM;
        }
        if (wait_completion) {
                cqp_request->waiting = 1;
        } else {
                cqp_request->waiting = 0;
        }
        cqp_wqe = &cqp_request->cqp_wqe;

        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX,
                        NES_CQP_MODIFY_QP | NES_CQP_QP_TYPE_IWARP | next_iwarp_state);
        nes_debug(NES_DBG_MOD_QP, "using next_iwarp_state=%08x, wqe_words=%08x\n",
                        next_iwarp_state, le32_to_cpu(cqp_wqe->wqe_words[NES_CQP_WQE_OPCODE_IDX]));
        nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
        set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX, nesqp->hwqp.qp_id);
        set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_QP_WQE_CONTEXT_LOW_IDX, (u64)nesqp->nesqp_context_pbase);

        /* If sending a terminate message, fill in the length (in words) */
        if (((next_iwarp_state & NES_CQP_QP_IWARP_STATE_MASK) == NES_CQP_QP_IWARP_STATE_TERMINATE) &&
            !(next_iwarp_state & NES_CQP_QP_TERM_DONT_SEND_TERM_MSG)) {
                termlen = ((termlen + 3) >> 2) << NES_CQP_OP_TERMLEN_SHIFT;
                set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_QP_WQE_NEW_MSS_IDX, termlen);
        }

        atomic_set(&cqp_request->refcount, 2);
        nes_post_cqp_request(nesdev, cqp_request);

        /* Wait for CQP */
        if (wait_completion) {
                /* nes_debug(NES_DBG_MOD_QP, "Waiting for modify iWARP QP%u to complete.\n",
                                nesqp->hwqp.qp_id); */
                ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
                                NES_EVENT_TIMEOUT);
                nes_debug(NES_DBG_MOD_QP, "Modify iwarp QP%u completed, wait_event_timeout ret=%u, "
                                "CQP Major:Minor codes = 0x%04X:0x%04X.\n",
                                nesqp->hwqp.qp_id, ret, cqp_request->major_code, cqp_request->minor_code);
                major_code = cqp_request->major_code;
                if (major_code) {
                        nes_debug(NES_DBG_MOD_QP, "Modify iwarp QP%u failed"
                                        "CQP Major:Minor codes = 0x%04X:0x%04X, intended next state = 0x%08X.\n",
                                        nesqp->hwqp.qp_id, cqp_request->major_code,
                                        cqp_request->minor_code, next_iwarp_state);
                }