/*
 *  IBM eServer eHCA Infiniband device driver for Linux on POWER
 *
 *  MR/MW functions
 *
 *  Authors: Dietmar Decker <ddecker@de.ibm.com>
 *           Christoph Raisch <raisch@de.ibm.com>
 *           Hoang-Nam Nguyen <hnguyen@de.ibm.com>
 *
 *  Copyright (c) 2005 IBM Corporation
 *
 *  All rights reserved.
 *
 *  This source code is distributed under a dual license of GPL v2.0 and OpenIB
 *  BSD.
 *
 * OpenIB BSD License
 *
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <linux/slab.h>
#include <rdma/ib_umem.h>

#include "ehca_iverbs.h"
#include "ehca_mrmw.h"
#include "hcp_if.h"
#include "hipz_hw.h"

#define NUM_CHUNKS(length, chunk_size) \
        (((length) + (chunk_size - 1)) / (chunk_size))

/* max number of rpages (per hcall register_rpages) */
#define MAX_RPAGES 512

/* DMEM toleration management */
#define EHCA_SECTSHIFT        SECTION_SIZE_BITS
#define EHCA_SECTSIZE          (1UL << EHCA_SECTSHIFT)
#define EHCA_HUGEPAGESHIFT     34
#define EHCA_HUGEPAGE_SIZE     (1UL << EHCA_HUGEPAGESHIFT)
#define EHCA_HUGEPAGE_PFN_MASK ((EHCA_HUGEPAGE_SIZE - 1) >> PAGE_SHIFT)
#define EHCA_INVAL_ADDR        0xFFFFFFFFFFFFFFFFULL
#define EHCA_DIR_INDEX_SHIFT 13                   /* 8k Entries in 64k block */
#define EHCA_TOP_INDEX_SHIFT (EHCA_DIR_INDEX_SHIFT * 2)
#define EHCA_MAP_ENTRIES (1 << EHCA_DIR_INDEX_SHIFT)
#define EHCA_TOP_MAP_SIZE (0x10000)               /* currently fixed map size */
#define EHCA_DIR_MAP_SIZE (0x10000)
#define EHCA_ENT_MAP_SIZE (0x10000)
#define EHCA_INDEX_MASK (EHCA_MAP_ENTRIES - 1)

static unsigned long ehca_mr_len;

/*
 * Memory map data structures
 */
struct ehca_dir_bmap {
        u64 ent[EHCA_MAP_ENTRIES];
};
struct ehca_top_bmap {
        struct ehca_dir_bmap *dir[EHCA_MAP_ENTRIES];
};
struct ehca_bmap {
        struct ehca_top_bmap *top[EHCA_MAP_ENTRIES];
};

static struct ehca_bmap *ehca_bmap;

static struct kmem_cache *mr_cache;
static struct kmem_cache *mw_cache;

enum ehca_mr_pgsize {
        EHCA_MR_PGSIZE4K  = 0x1000L,
        EHCA_MR_PGSIZE64K = 0x10000L,
        EHCA_MR_PGSIZE1M  = 0x100000L,
        EHCA_MR_PGSIZE16M = 0x1000000L
};

#define EHCA_MR_PGSHIFT4K  12
#define EHCA_MR_PGSHIFT64K 16
#define EHCA_MR_PGSHIFT1M  20
#define EHCA_MR_PGSHIFT16M 24

static u64 ehca_map_vaddr(void *caddr);

static u32 ehca_encode_hwpage_size(u32 pgsize)
{
        int log = ilog2(pgsize);
        WARN_ON(log < 12 || log > 24 || log & 3);
        return (log - 12) / 4;
}

static u64 ehca_get_max_hwpage_size(struct ehca_shca *shca)
{
        return rounddown_pow_of_two(shca->hca_cap_mr_pgsize);
}

static struct ehca_mr *ehca_mr_new(void)
{
        struct ehca_mr *me;

        me = kmem_cache_zalloc(mr_cache, GFP_KERNEL);
        if (me)
                spin_lock_init(&me->mrlock);
        else
                ehca_gen_err("alloc failed");

        return me;
}

static void ehca_mr_delete(struct ehca_mr *me)
{
        kmem_cache_free(mr_cache, me);
}

static struct ehca_mw *ehca_mw_new(void)
{
        struct ehca_mw *me;

        me = kmem_cache_zalloc(mw_cache, GFP_KERNEL);
        if (me)
                spin_lock_init(&me->mwlock);
        else
                ehca_gen_err("alloc failed");

        return me;
}

static void ehca_mw_delete(struct ehca_mw *me)
{
        kmem_cache_free(mw_cache, me);
}

/*----------------------------------------------------------------------*/

struct ib_mr *ehca_get_dma_mr(struct ib_pd *pd, int mr_access_flags)
{
        struct ib_mr *ib_mr;
        int ret;
        struct ehca_mr *e_maxmr;
        struct ehca_pd *e_pd = container_of(pd, struct ehca_pd, ib_pd);
        struct ehca_shca *shca =
                container_of(pd->device, struct ehca_shca, ib_device);

        if (shca->maxmr) {
                e_maxmr = ehca_mr_new();
                if (!e_maxmr) {
                        ehca_err(&shca->ib_device, "out of memory");
                        ib_mr = ERR_PTR(-ENOMEM);
                        goto get_dma_mr_exit0;
                }

                ret = ehca_reg_maxmr(shca, e_maxmr,
                                     (void *)ehca_map_vaddr((void *)(KERNELBASE + PHYSICAL_START)),
                                     mr_access_flags, e_pd,
                                     &e_maxmr->ib.ib_mr.lkey,
                                     &e_maxmr->ib.ib_mr.rkey);
                if (ret) {
                        ehca_mr_delete(e_maxmr);
                        ib_mr = ERR_PTR(ret);
                        goto get_dma_mr_exit0;
                }
                ib_mr = &e_maxmr->ib.ib_mr;
        } else {
                ehca_err(&shca->ib_device, "no internal max-MR exist!");
                ib_mr = ERR_PTR(-EINVAL);
                goto get_dma_mr_exit0;
        }

get_dma_mr_exit0:
        if (IS_ERR(ib_mr))
                ehca_err(&shca->ib_device, "h_ret=%li pd=%p mr_access_flags=%x",
                         PTR_ERR(ib_mr), pd, mr_access_flags);
        return ib_mr;
} /* end ehca_get_dma_mr() */

/*----------------------------------------------------------------------*/

struct ib_mr *ehca_reg_user_mr(struct ib_pd *pd, u64 start, u64 length,
                               u64 virt, int mr_access_flags,
                               struct ib_udata *udata)
{
        struct ib_mr *ib_mr;
        struct ehca_mr *e_mr;
        struct ehca_shca *shca =
                container_of(pd->device, struct ehca_shca, ib_device);
        struct ehca_pd *e_pd = container_of(pd, struct ehca_pd, ib_pd);
        struct ehca_mr_pginfo pginfo;
        int ret, page_shift;
        u32 num_kpages;
        u32 num_hwpages;
        u64 hwpage_size;

        if (!pd) {
                ehca_gen_err("bad pd=%p", pd);
                return ERR_PTR(-EFAULT);
        }

        if (((mr_access_flags & IB_ACCESS_REMOTE_WRITE) &&
             !(mr_access_flags & IB_ACCESS_LOCAL_WRITE)) ||
            ((mr_access_flags & IB_ACCESS_REMOTE_ATOMIC) &&
             !(mr_access_flags & IB_ACCESS_LOCAL_WRITE))) {
                /*
                 * Remote Write Access requires Local Write Access
                 * Remote Atomic Access requires Local Write Access
                 */
                ehca_err(pd->device, "bad input values: mr_access_flags=%x",
                         mr_access_flags);
                ib_mr = ERR_PTR(-EINVAL);
                goto reg_user_mr_exit0;
        }

        if (length == 0 || virt + length < virt) {
                ehca_err(pd->device, "bad input values: length=%llx "
                         "virt_base=%llx", length, virt);
                ib_mr = ERR_PTR(-EINVAL);
                goto reg_user_mr_exit0;
        }

        e_mr = ehca_mr_new();
        if (!e_mr) {
                ehca_err(pd->device, "out of memory");
                ib_mr = ERR_PTR(-ENOMEM);
                goto reg_user_mr_exit0;
        }

        e_mr->umem = ib_umem_get(pd->uobject->context, start, length,
                                 mr_access_flags, 0);
        if (IS_ERR(e_mr->umem)) {
                ib_mr = (void *)e_mr->umem;
                goto reg_user_mr_exit1;
        }

        if (e_mr->umem->page_size != PAGE_SIZE) {
                ehca_err(pd->device, "page size not supported, "
                         "e_mr->umem->page_size=%x", e_mr->umem->page_size);
                ib_mr = ERR_PTR(-EINVAL);
                goto reg_user_mr_exit2;
        }

        /* determine number of MR pages */
        num_kpages = NUM_CHUNKS((virt % PAGE_SIZE) + length, PAGE_SIZE);
        /* select proper hw_pgsize */
        page_shift = PAGE_SHIFT;
        if (e_mr->umem->hugetlb) {
                /* determine page_shift, clamp between 4K and 16M */
                page_shift = (fls64(length - 1) + 3) & ~3;
                page_shift = min(max(page_shift, EHCA_MR_PGSHIFT4K),
                                 EHCA_MR_PGSHIFT16M);
        }
        hwpage_size = 1UL << page_shift;

        /* now that we have the desired page size, shift until it's
         * supported, too. 4K is always supported, so this terminates.
         */
        while (!(hwpage_size & shca->hca_cap_mr_pgsize))
                hwpage_size >>= 4;

reg_user_mr_fallback:
        num_hwpages = NUM_CHUNKS((virt % hwpage_size) + length, hwpage_size);
        /* register MR on HCA */
        memset(&pginfo, 0, sizeof(pginfo));
        pginfo.type = EHCA_MR_PGI_USER;
        pginfo.hwpage_size = hwpage_size;
        pginfo.num_kpages = num_kpages;
        pginfo.num_hwpages = num_hwpages;
        pginfo.u.usr.region = e_mr->umem;
        pginfo.next_hwpage = ib_umem_offset(e_mr->umem) / hwpage_size;
        pginfo.u.usr.next_sg = pginfo.u.usr.region->sg_head.sgl;
        ret = ehca_reg_mr(shca, e_mr, (u64 *)virt, length, mr_access_flags,
                          e_pd, &pginfo, &e_mr->ib.ib_mr.lkey,
                          &e_mr->ib.ib_mr.rkey, EHCA_REG_MR);
        if (ret == -EINVAL && pginfo.hwpage_size > PAGE_SIZE) {
                ehca_warn(pd->device, "failed to register mr "
                          "with hwpage_size=%llx", hwpage_size);
                ehca_info(pd->device, "try to register mr with "
                          "kpage_size=%lx", PAGE_SIZE);
                /*
                 * this means kpages are not contiguous for a hw page
                 * try kernel page size as fallback solution
                 */
                hwpage_size = PAGE_SIZE;
                goto reg_user_mr_fallback;
        }
        if (ret) {
                ib_mr = ERR_PTR(ret);
                goto reg_user_mr_exit2;
        }

        /* successful registration of all pages */
        return &e_mr->ib.ib_mr;

reg_user_mr_exit2:
        ib_umem_release(e_mr->umem);
reg_user_mr_exit1:
        ehca_mr_delete(e_mr);
reg_user_mr_exit0:
        if (IS_ERR(ib_mr))
                ehca_err(pd->device, "rc=%li pd=%p mr_access_flags=%x udata=%p",
                         PTR_ERR(ib_mr), pd, mr_access_flags, udata);
        return ib_mr;
} /* end ehca_reg_user_mr() */

/*----------------------------------------------------------------------*/

int ehca_dereg_mr(struct ib_mr *mr)
{
        int ret = 0;
        u64 h_ret;
        struct ehca_shca *shca =
                container_of(mr->device, struct ehca_shca, ib_device);
        struct ehca_mr *e_mr = container_of(mr, struct ehca_mr, ib.ib_mr);

        if ((e_mr->flags & EHCA_MR_FLAG_FMR)) {
                ehca_err(mr->device, "not supported for FMR, mr=%p e_mr=%p "
                         "e_mr->flags=%x", mr, e_mr, e_mr->flags);
                ret = -EINVAL;
                goto dereg_mr_exit0;
        } else if (e_mr == shca->maxmr) {
                /* should be impossible, however reject to be sure */
                ehca_err(mr->device, "dereg internal max-MR impossible, mr=%p "
                         "shca->maxmr=%p mr->lkey=%x",
                         mr, shca->maxmr, mr->lkey);
                ret = -EINVAL;
                goto dereg_mr_exit0;
        }

        /* TODO: BUSY: MR still has bound window(s) */
        h_ret = hipz_h_free_resource_mr(shca->ipz_hca_handle, e_mr);
        if (h_ret != H_SUCCESS) {
                ehca_err(mr->device, "hipz_free_mr failed, h_ret=%lli shca=%p "
                         "e_mr=%p hca_hndl=%llx mr_hndl=%llx mr->lkey=%x",
                         h_ret, shca, e_mr, shca->ipz_hca_handle.handle,
                         e_mr->ipz_mr_handle.handle, mr->lkey);
                ret = ehca2ib_return_code(h_ret);
                goto dereg_mr_exit0;
        }

        if (e_mr->umem)
                ib_umem_release(e_mr->umem);

        /* successful deregistration */
        ehca_mr_delete(e_mr);

dereg_mr_exit0:
        if (ret)
                ehca_err(mr->device, "ret=%i mr=%p", ret, mr);
        return ret;
} /* end ehca_dereg_mr() */

/*----------------------------------------------------------------------*/

struct ib_mw *ehca_alloc_mw(struct ib_pd *pd, enum ib_mw_type type,
                            struct ib_udata *udata)
{
        struct ib_mw *ib_mw;
        u64 h_ret;
        struct ehca_mw *e_mw;
        struct ehca_pd *e_pd = container_of(pd, struct ehca_pd, ib_pd);
        struct ehca_shca *shca =
                container_of(pd->device, struct ehca_shca, ib_device);
        struct ehca_mw_hipzout_parms hipzout;

        if (type != IB_MW_TYPE_1)
                return ERR_PTR(-EINVAL);

        e_mw = ehca_mw_new();
        if (!e_mw) {
                ib_mw = ERR_PTR(-ENOMEM);
                goto alloc_mw_exit0;
        }

        h_ret = hipz_h_alloc_resource_mw(shca->ipz_hca_handle, e_mw,
                                         e_pd->fw_pd, &hipzout);
        if (h_ret != H_SUCCESS) {
                ehca_err(pd->device, "hipz_mw_allocate failed, h_ret=%lli "
                         "shca=%p hca_hndl=%llx mw=%p",
                         h_ret, shca, shca->ipz_hca_handle.handle, e_mw);
                ib_mw = ERR_PTR(ehca2ib_return_code(h_ret));
                goto alloc_mw_exit1;
        }
        /* successful MW allocation */
        e_mw->ipz_mw_handle = hipzout.handle;
        e_mw->ib_mw.rkey    = hipzout.rkey;
        return &e_mw->ib_mw;

alloc_mw_exit1:
        ehca_mw_delete(e_mw);
alloc_mw_exit0:
        if (IS_ERR(ib_mw))
                ehca_err(pd->device, "h_ret=%li pd=%p", PTR_ERR(ib_mw), pd);
        return ib_mw;
} /* end ehca_alloc_mw() */

/*----------------------------------------------------------------------*/

int ehca_dealloc_mw(struct ib_mw *mw)
{
        u64 h_ret;
        struct ehca_shca *shca =
                container_of(mw->device, struct ehca_shca, ib_device);
        struct ehca_mw *e_mw = container_of(mw, struct ehca_mw, ib_mw);

        h_ret = hipz_h_free_resource_mw(shca->ipz_hca_handle, e_mw);
        if (h_ret != H_SUCCESS) {
                ehca_err(mw->device, "hipz_free_mw failed, h_ret=%lli shca=%p "
                         "mw=%p rkey=%x hca_hndl=%llx mw_hndl=%llx",
                         h_ret, shca, mw, mw->rkey, shca->ipz_hca_handle.handle,
                         e_mw->ipz_mw_handle.handle);
                return ehca2ib_return_code(h_ret);
        }
        /* successful deallocation */
        ehca_mw_delete(e_mw);
        return 0;
} /* end ehca_dealloc_mw() */

/*----------------------------------------------------------------------*/

struct ib_fmr *ehca_alloc_fmr(struct ib_pd *pd,
                              int mr_access_flags,
                              struct ib_fmr_attr *fmr_attr)
{
        struct ib_fmr *ib_fmr;
        struct ehca_shca *shca =
                container_of(pd->device, struct ehca_shca, ib_device);
        struct ehca_pd *e_pd = container_of(pd, struct ehca_pd, ib_pd);
        struct ehca_mr *e_fmr;
        int ret;
        u32 tmp_lkey, tmp_rkey;
        struct ehca_mr_pginfo pginfo;
        u64 hw_pgsize;

        /* check other parameters */
        if (((mr_access_flags & IB_ACCESS_REMOTE_WRITE) &&
             !(mr_access_flags & IB_ACCESS_LOCAL_WRITE)) ||
            ((mr_access_flags & IB_ACCESS_REMOTE_ATOMIC) &&
             !(mr_access_flags & IB_ACCESS_LOCAL_WRITE))) {
                /*
                 * Remote Write Access requires Local Write Access
                 * Remote Atomic Access requires Local Write Access
                 */
                ehca_err(pd->device, "bad input values: mr_access_flags=%x",
                         mr_access_flags);
                ib_fmr = ERR_PTR(-EINVAL);
                goto alloc_fmr_exit0;
        }
        if (mr_access_flags & IB_ACCESS_MW_BIND) {
                ehca_err(pd->device, "bad input values: mr_access_flags=%x",
                         mr_access_flags);
                ib_fmr = ERR_PTR(-EINVAL);
                goto alloc_fmr_exit0;
        }
        if ((fmr_attr->max_pages == 0) || (fmr_attr->max_maps == 0)) {
                ehca_err(pd->device, "bad input values: fmr_attr->max_pages=%x "
                         "fmr_attr->max_maps=%x fmr_attr->page_shift=%x",
                         fmr_attr->max_pages, fmr_attr->max_maps,
                         fmr_attr->page_shift);
                ib_fmr = ERR_PTR(-EINVAL);
                goto alloc_fmr_exit0;
        }

        hw_pgsize = 1 << fmr_attr->page_shift;
        if (!(hw_pgsize & shca->hca_cap_mr_pgsize)) {
                ehca_err(pd->device, "unsupported fmr_attr->page_shift=%x",
                         fmr_attr->page_shift);
                ib_fmr = ERR_PTR(-EINVAL);
                goto alloc_fmr_exit0;
        }

        e_fmr = ehca_mr_new();
        if (!e_fmr) {
                ib_fmr = ERR_PTR(-ENOMEM);
                goto alloc_fmr_exit0;
        }
        e_fmr->flags |= EHCA_MR_FLAG_FMR;

        /* register MR on HCA */
        memset(&pginfo, 0, sizeof(pginfo));
        pginfo.hwpage_size = hw_pgsize;
        /*
         * pginfo.num_hwpages==0, ie register_rpages() will not be called
         * but deferred to map_phys_fmr()
         */
        ret = ehca_reg_mr(shca, e_fmr, NULL,
                          fmr_attr->max_pages * (1 << fmr_attr->page_shift),
                          mr_access_flags, e_pd, &pginfo,
                          &tmp_lkey, &tmp_rkey, EHCA_REG_MR);
        if (ret) {
                ib_fmr = ERR_PTR(ret);
                goto alloc_fmr_exit1;
        }

        /* successful */
        e_fmr->hwpage_size = hw_pgsize;
        e_fmr->fmr_page_size = 1 << fmr_attr->page_shift;
        e_fmr->fmr_max_pages = fmr_attr->max_pages;
        e_fmr->fmr_max_maps = fmr_attr->max_maps;
        e_fmr->fmr_map_cnt = 0;
        return &e_fmr->ib.ib_fmr;

alloc_fmr_exit1:
        ehca_mr_delete(e_fmr);
alloc_fmr_exit0:
        return ib_fmr;
} /* end ehca_alloc_fmr() */

/*----------------------------------------------------------------------*/

int ehca_map_phys_fmr(struct ib_fmr *fmr,
                      u64 *page_list,
                      int list_len,
                      u64 iova)
{
        int ret;
        struct ehca_shca *shca =
                container_of(fmr->device, struct ehca_shca, ib_device);
        struct ehca_mr *e_fmr = container_of(fmr, struct ehca_mr, ib.ib_fmr);
        struct ehca_pd *e_pd = container_of(fmr->pd, struct ehca_pd, ib_pd);
        struct ehca_mr_pginfo pginfo;
        u32 tmp_lkey, tmp_rkey;

        if (!(e_fmr->flags & EHCA_MR_FLAG_FMR)) {
                ehca_err(fmr->device, "not a FMR, e_fmr=%p e_fmr->flags=%x",
                         e_fmr, e_fmr->flags);
                ret = -EINVAL;
                goto map_phys_fmr_exit0;
        }
        ret = ehca_fmr_check_page_list(e_fmr, page_list, list_len);
        if (ret)
                goto map_phys_fmr_exit0;
        if (iova % e_fmr->fmr_page_size) {
                /* only whole-numbered pages */
                ehca_err(fmr->device, "bad iova, iova=%llx fmr_page_size=%x",
                         iova, e_fmr->fmr_page_size);
                ret = -EINVAL;
                goto map_phys_fmr_exit0;
        }
        if (e_fmr->fmr_map_cnt >= e_fmr->fmr_max_maps) {
                /* HCAD does not limit the maps, however trace this anyway */
                ehca_info(fmr->device, "map limit exceeded, fmr=%p "
                          "e_fmr->fmr_map_cnt=%x e_fmr->fmr_max_maps=%x",
                          fmr, e_fmr->fmr_map_cnt, e_fmr->fmr_max_maps);
        }

        memset(&pginfo, 0, sizeof(pginfo));
        pginfo.type = EHCA_MR_PGI_FMR;
        pginfo.num_kpages = list_len;
        pginfo.hwpage_size = e_fmr->hwpage_size;
        pginfo.num_hwpages =
                list_len * e_fmr->fmr_page_size / pginfo.hwpage_size;
        pginfo.u.fmr.page_list = page_list;
        pginfo.next_hwpage =
                (iova & (e_fmr->fmr_page_size-1)) / pginfo.hwpage_size;
        pginfo.u.fmr.fmr_pgsize = e_fmr->fmr_page_size;

        ret = ehca_rereg_mr(shca, e_fmr, (u64 *)iova,
                            list_len * e_fmr->fmr_page_size,
                            e_fmr->acl, e_pd, &pginfo, &tmp_lkey, &tmp_rkey);
        if (ret)
                goto map_phys_fmr_exit0;

        /* successful reregistration */
        e_fmr->fmr_map_cnt++;
        e_fmr->ib.ib_fmr.lkey = tmp_lkey;
        e_fmr->ib.ib_fmr.rkey = tmp_rkey;
        return 0;

map_phys_fmr_exit0:
        if (ret)
                ehca_err(fmr->device, "ret=%i fmr=%p page_list=%p list_len=%x "
                         "iova=%llx", ret, fmr, page_list, list_len, iova);
        return ret;
} /* end ehca_map_phys_fmr() */

/*----------------------------------------------------------------------*/

int ehca_unmap_fmr(struct list_head *fmr_list)
{
        int ret = 0;
        struct ib_fmr *ib_fmr;
        struct ehca_shca *shca = NULL;
        struct ehca_shca *prev_shca;
        struct ehca_mr *e_fmr;
        u32 num_fmr = 0;
        u32 unmap_fmr_cnt = 0;

        /* check all FMR belong to same SHCA, and check internal flag */
        list_for_each_entry(ib_fmr, fmr_list, list) {
                prev_shca = shca;
                shca = container_of(ib_fmr->device, struct ehca_shca,
                                    ib_device);
                e_fmr = container_of(ib_fmr, struct ehca_mr, ib.ib_fmr);
                if ((shca != prev_shca) && prev_shca) {
                        ehca_err(&shca->ib_device, "SHCA mismatch, shca=%p "
                                 "prev_shca=%p e_fmr=%p",
                                 shca, prev_shca, e_fmr);
                        ret = -EINVAL;
                        goto unmap_fmr_exit0;
                }
                if (!(e_fmr->flags & EHCA_MR_FLAG_FMR)) {
                        ehca_err(&shca->ib_device, "not a FMR, e_fmr=%p "
                                 "e_fmr->flags=%x", e_fmr, e_fmr->flags);
                        ret = -EINVAL;
                        goto unmap_fmr_exit0;
                }
                num_fmr++;
        }

        /* loop over all FMRs to unmap */
        list_for_each_entry(ib_fmr, fmr_list, list) {
                unmap_fmr_cnt++;
                e_fmr = container_of(ib_fmr, struct ehca_mr, ib.ib_fmr);
                shca = container_of(ib_fmr->device, struct ehca_shca,
                                    ib_device);
                ret = ehca_unmap_one_fmr(shca, e_fmr);
                if (ret) {
                        /* unmap failed, stop unmapping of rest of FMRs */
                        ehca_err(&shca->ib_device, "unmap of one FMR failed, "
                                 "stop rest, e_fmr=%p num_fmr=%x "
                                 "unmap_fmr_cnt=%x lkey=%x", e_fmr, num_fmr,
                                 unmap_fmr_cnt, e_fmr->ib.ib_fmr.lkey);
                        goto unmap_fmr_exit0;
                }
        }

unmap_fmr_exit0:
        if (ret)
                ehca_gen_err("ret=%i fmr_list=%p num_fmr=%x unmap_fmr_cnt=%x",
                             ret, fmr_list, num_fmr, unmap_fmr_cnt);
        return ret;
} /* end ehca_unmap_fmr() */

/*----------------------------------------------------------------------*/

int ehca_dealloc_fmr(struct ib_fmr *fmr)
{
        int ret;
        u64 h_ret;
        struct ehca_shca *shca =
                container_of(fmr->device, struct ehca_shca, ib_device);
        struct ehca_mr *e_fmr = container_of(fmr, struct ehca_mr, ib.ib_fmr);

        if (!(e_fmr->flags & EHCA_MR_FLAG_FMR)) {
                ehca_err(fmr->device, "not a FMR, e_fmr=%p e_fmr->flags=%x",
                         e_fmr, e_fmr->flags);
                ret = -EINVAL;
                goto free_fmr_exit0;
        }

        h_ret = hipz_h_free_resource_mr(shca->ipz_hca_handle, e_fmr);
        if (h_ret != H_SUCCESS) {
                ehca_err(fmr->device, "hipz_free_mr failed, h_ret=%lli e_fmr=%p "
                         "hca_hndl=%llx fmr_hndl=%llx fmr->lkey=%x",
                         h_ret, e_fmr, shca->ipz_hca_handle.handle,
                         e_fmr->ipz_mr_handle.handle, fmr->lkey);
                ret = ehca2ib_return_code(h_ret);
                goto free_fmr_exit0;
        }
        /* successful deregistration */
        ehca_mr_delete(e_fmr);
        return 0;

free_fmr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i fmr=%p", ret, fmr);
        return ret;
} /* end ehca_dealloc_fmr() */

/*----------------------------------------------------------------------*/

static int ehca_reg_bmap_mr_rpages(struct ehca_shca *shca,
                                   struct ehca_mr *e_mr,
                                   struct ehca_mr_pginfo *pginfo);

int ehca_reg_mr(struct ehca_shca *shca,
                struct ehca_mr *e_mr,
                u64 *iova_start,
                u64 size,
                int acl,
                struct ehca_pd *e_pd,
                struct ehca_mr_pginfo *pginfo,
                u32 *lkey, /*OUT*/
                u32 *rkey, /*OUT*/
                enum ehca_reg_type reg_type)
{
        int ret;
        u64 h_ret;
        u32 hipz_acl;
        struct ehca_mr_hipzout_parms hipzout;

        ehca_mrmw_map_acl(acl, &hipz_acl);
        ehca_mrmw_set_pgsize_hipz_acl(pginfo->hwpage_size, &hipz_acl);
        if (ehca_use_hp_mr == 1)
                hipz_acl |= 0x00000001;

        h_ret = hipz_h_alloc_resource_mr(shca->ipz_hca_handle, e_mr,
                                         (u64)iova_start, size, hipz_acl,
                                         e_pd->fw_pd, &hipzout);
        if (h_ret != H_SUCCESS) {
                ehca_err(&shca->ib_device, "hipz_alloc_mr failed, h_ret=%lli "
                         "hca_hndl=%llx", h_ret, shca->ipz_hca_handle.handle);
                ret = ehca2ib_return_code(h_ret);
                goto ehca_reg_mr_exit0;
        }

        e_mr->ipz_mr_handle = hipzout.handle;

        if (reg_type == EHCA_REG_BUSMAP_MR)
                ret = ehca_reg_bmap_mr_rpages(shca, e_mr, pginfo);
        else if (reg_type == EHCA_REG_MR)
                ret = ehca_reg_mr_rpages(shca, e_mr, pginfo);
        else
                ret = -EINVAL;

        if (ret)
                goto ehca_reg_mr_exit1;

        /* successful registration */
        e_mr->num_kpages = pginfo->num_kpages;
        e_mr->num_hwpages = pginfo->num_hwpages;
        e_mr->hwpage_size = pginfo->hwpage_size;
        e_mr->start = iova_start;
        e_mr->size = size;
        e_mr->acl = acl;
        *lkey = hipzout.lkey;
        *rkey = hipzout.rkey;
        return 0;

ehca_reg_mr_exit1:
        h_ret = hipz_h_free_resource_mr(shca->ipz_hca_handle, e_mr);
        if (h_ret != H_SUCCESS) {
                ehca_err(&shca->ib_device, "h_ret=%lli shca=%p e_mr=%p "
                         "iova_start=%p size=%llx acl=%x e_pd=%p lkey=%x "
                         "pginfo=%p num_kpages=%llx num_hwpages=%llx ret=%i",
                         h_ret, shca, e_mr, iova_start, size, acl, e_pd,
                         hipzout.lkey, pginfo, pginfo->num_kpages,
                         pginfo->num_hwpages, ret);
                ehca_err(&shca->ib_device, "internal error in ehca_reg_mr, "
                         "not recoverable");
        }
ehca_reg_mr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i shca=%p e_mr=%p "
                         "iova_start=%p size=%llx acl=%x e_pd=%p pginfo=%p "
                         "num_kpages=%llx num_hwpages=%llx",
                         ret, shca, e_mr, iova_start, size, acl, e_pd, pginfo,
                         pginfo->num_kpages, pginfo->num_hwpages);
        return ret;
} /* end ehca_reg_mr() */

/*----------------------------------------------------------------------*/

int ehca_reg_mr_rpages(struct ehca_shca *shca,
                       struct ehca_mr *e_mr,
                       struct ehca_mr_pginfo *pginfo)
{
        int ret = 0;
        u64 h_ret;
        u32 rnum;
        u64 rpage;
        u32 i;
        u64 *kpage;

        if (!pginfo->num_hwpages) /* in case of fmr */
                return 0;

        kpage = ehca_alloc_fw_ctrlblock(GFP_KERNEL);
        if (!kpage) {
                ehca_err(&shca->ib_device, "kpage alloc failed");
                ret = -ENOMEM;
                goto ehca_reg_mr_rpages_exit0;
        }

        /* max MAX_RPAGES ehca mr pages per register call */
        for (i = 0; i < NUM_CHUNKS(pginfo->num_hwpages, MAX_RPAGES); i++) {

                if (i == NUM_CHUNKS(pginfo->num_hwpages, MAX_RPAGES) - 1) {
                        rnum = pginfo->num_hwpages % MAX_RPAGES; /* last shot */
                        if (rnum == 0)
                                rnum = MAX_RPAGES;      /* last shot is full */
                } else
                        rnum = MAX_RPAGES;

                ret = ehca_set_pagebuf(pginfo, rnum, kpage);
                if (ret) {
                        ehca_err(&shca->ib_device, "ehca_set_pagebuf "
                                 "bad rc, ret=%i rnum=%x kpage=%p",
                                 ret, rnum, kpage);
                        goto ehca_reg_mr_rpages_exit1;
                }

                if (rnum > 1) {
                        rpage = __pa(kpage);
                        if (!rpage) {
                                ehca_err(&shca->ib_device, "kpage=%p i=%x",
                                         kpage, i);
                                ret = -EFAULT;
                                goto ehca_reg_mr_rpages_exit1;
                        }
                } else
                        rpage = *kpage;

                h_ret = hipz_h_register_rpage_mr(
                        shca->ipz_hca_handle, e_mr,
                        ehca_encode_hwpage_size(pginfo->hwpage_size),
                        0, rpage, rnum);

                if (i == NUM_CHUNKS(pginfo->num_hwpages, MAX_RPAGES) - 1) {
                        /*
                         * check for 'registration complete'==H_SUCCESS
                         * and for 'page registered'==H_PAGE_REGISTERED
                         */
                        if (h_ret != H_SUCCESS) {
                                ehca_err(&shca->ib_device, "last "
                                         "hipz_reg_rpage_mr failed, h_ret=%lli "
                                         "e_mr=%p i=%x hca_hndl=%llx mr_hndl=%llx"
                                         " lkey=%x", h_ret, e_mr, i,
                                         shca->ipz_hca_handle.handle,
                                         e_mr->ipz_mr_handle.handle,
                                         e_mr->ib.ib_mr.lkey);
                                ret = ehca2ib_return_code(h_ret);
                                break;
                        } else
                                ret = 0;
                } else if (h_ret != H_PAGE_REGISTERED) {
                        ehca_err(&shca->ib_device, "hipz_reg_rpage_mr failed, "
                                 "h_ret=%lli e_mr=%p i=%x lkey=%x hca_hndl=%llx "
                                 "mr_hndl=%llx", h_ret, e_mr, i,
                                 e_mr->ib.ib_mr.lkey,
                                 shca->ipz_hca_handle.handle,
                                 e_mr->ipz_mr_handle.handle);
                        ret = ehca2ib_return_code(h_ret);
                        break;
                } else
                        ret = 0;
        } /* end for(i) */


ehca_reg_mr_rpages_exit1:
        ehca_free_fw_ctrlblock(kpage);
ehca_reg_mr_rpages_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i shca=%p e_mr=%p pginfo=%p "
                         "num_kpages=%llx num_hwpages=%llx", ret, shca, e_mr,
                         pginfo, pginfo->num_kpages, pginfo->num_hwpages);
        return ret;
} /* end ehca_reg_mr_rpages() */

/*----------------------------------------------------------------------*/

inline int ehca_rereg_mr_rereg1(struct ehca_shca *shca,
                                struct ehca_mr *e_mr,
                                u64 *iova_start,
                                u64 size,
                                u32 acl,
                                struct ehca_pd *e_pd,
                                struct ehca_mr_pginfo *pginfo,
                                u32 *lkey, /*OUT*/
                                u32 *rkey) /*OUT*/
{
        int ret;
        u64 h_ret;
        u32 hipz_acl;
        u64 *kpage;
        u64 rpage;
        struct ehca_mr_pginfo pginfo_save;
        struct ehca_mr_hipzout_parms hipzout;

        ehca_mrmw_map_acl(acl, &hipz_acl);
        ehca_mrmw_set_pgsize_hipz_acl(pginfo->hwpage_size, &hipz_acl);

        kpage = ehca_alloc_fw_ctrlblock(GFP_KERNEL);
        if (!kpage) {
                ehca_err(&shca->ib_device, "kpage alloc failed");
                ret = -ENOMEM;
                goto ehca_rereg_mr_rereg1_exit0;
        }

        pginfo_save = *pginfo;
        ret = ehca_set_pagebuf(pginfo, pginfo->num_hwpages, kpage);
        if (ret) {
                ehca_err(&shca->ib_device, "set pagebuf failed, e_mr=%p "
                         "pginfo=%p type=%x num_kpages=%llx num_hwpages=%llx "
                         "kpage=%p", e_mr, pginfo, pginfo->type,
                         pginfo->num_kpages, pginfo->num_hwpages, kpage);
                goto ehca_rereg_mr_rereg1_exit1;
        }
        rpage = __pa(kpage);
        if (!rpage) {
                ehca_err(&shca->ib_device, "kpage=%p", kpage);
                ret = -EFAULT;
                goto ehca_rereg_mr_rereg1_exit1;
        }
        h_ret = hipz_h_reregister_pmr(shca->ipz_hca_handle, e_mr,
                                      (u64)iova_start, size, hipz_acl,
                                      e_pd->fw_pd, rpage, &hipzout);
        if (h_ret != H_SUCCESS) {
                /*
                 * reregistration unsuccessful, try it again with the 3 hCalls,
                 * e.g. this is required in case H_MR_CONDITION
                 * (MW bound or MR is shared)
                 */
                ehca_warn(&shca->ib_device, "hipz_h_reregister_pmr failed "
                          "(Rereg1), h_ret=%lli e_mr=%p", h_ret, e_mr);
                *pginfo = pginfo_save;
                ret = -EAGAIN;
        } else if ((u64 *)hipzout.vaddr != iova_start) {
                ehca_err(&shca->ib_device, "PHYP changed iova_start in "
                         "rereg_pmr, iova_start=%p iova_start_out=%llx e_mr=%p "
                         "mr_handle=%llx lkey=%x lkey_out=%x", iova_start,
                         hipzout.vaddr, e_mr, e_mr->ipz_mr_handle.handle,
                         e_mr->ib.ib_mr.lkey, hipzout.lkey);
                ret = -EFAULT;
        } else {
                /*
                 * successful reregistration
                 * note: start and start_out are identical for eServer HCAs
                 */
                e_mr->num_kpages = pginfo->num_kpages;
                e_mr->num_hwpages = pginfo->num_hwpages;
                e_mr->hwpage_size = pginfo->hwpage_size;
                e_mr->start = iova_start;
                e_mr->size = size;
                e_mr->acl = acl;
                *lkey = hipzout.lkey;
                *rkey = hipzout.rkey;
        }

ehca_rereg_mr_rereg1_exit1:
        ehca_free_fw_ctrlblock(kpage);
ehca_rereg_mr_rereg1_exit0:
        if ( ret && (ret != -EAGAIN) )
                ehca_err(&shca->ib_device, "ret=%i lkey=%x rkey=%x "
                         "pginfo=%p num_kpages=%llx num_hwpages=%llx",
                         ret, *lkey, *rkey, pginfo, pginfo->num_kpages,
                         pginfo->num_hwpages);
        return ret;
} /* end ehca_rereg_mr_rereg1() */

/*----------------------------------------------------------------------*/

int ehca_rereg_mr(struct ehca_shca *shca,
                  struct ehca_mr *e_mr,
                  u64 *iova_start,
                  u64 size,
                  int acl,
                  struct ehca_pd *e_pd,
                  struct ehca_mr_pginfo *pginfo,
                  u32 *lkey,
                  u32 *rkey)
{
        int ret = 0;
        u64 h_ret;
        int rereg_1_hcall = 1; /* 1: use hipz_h_reregister_pmr directly */
        int rereg_3_hcall = 0; /* 1: use 3 hipz calls for reregistration */

        /* first determine reregistration hCall(s) */
        if ((pginfo->num_hwpages > MAX_RPAGES) ||
            (e_mr->num_hwpages > MAX_RPAGES) ||
            (pginfo->num_hwpages > e_mr->num_hwpages)) {
                ehca_dbg(&shca->ib_device, "Rereg3 case, "
                         "pginfo->num_hwpages=%llx e_mr->num_hwpages=%x",
                         pginfo->num_hwpages, e_mr->num_hwpages);
                rereg_1_hcall = 0;
                rereg_3_hcall = 1;
        }

        if (e_mr->flags & EHCA_MR_FLAG_MAXMR) { /* check for max-MR */
                rereg_1_hcall = 0;
                rereg_3_hcall = 1;
                e_mr->flags &= ~EHCA_MR_FLAG_MAXMR;
                ehca_err(&shca->ib_device, "Rereg MR for max-MR! e_mr=%p",
                         e_mr);
        }

        if (rereg_1_hcall) {
                ret = ehca_rereg_mr_rereg1(shca, e_mr, iova_start, size,
                                           acl, e_pd, pginfo, lkey, rkey);
                if (ret) {
                        if (ret == -EAGAIN)

                                rereg_3_hcall = 1;
                        else
                                goto ehca_rereg_mr_exit0;
                }
        }

        if (rereg_3_hcall) {
                struct ehca_mr save_mr;

                /* first deregister old MR */
                h_ret = hipz_h_free_resource_mr(shca->ipz_hca_handle, e_mr);
                if (h_ret != H_SUCCESS) {
                        ehca_err(&shca->ib_device, "hipz_free_mr failed, "
                                 "h_ret=%lli e_mr=%p hca_hndl=%llx mr_hndl=%llx "
                                 "mr->lkey=%x",
                                 h_ret, e_mr, shca->ipz_hca_handle.handle,
                                 e_mr->ipz_mr_handle.handle,
                                 e_mr->ib.ib_mr.lkey);
                        ret = ehca2ib_return_code(h_ret);
                        goto ehca_rereg_mr_exit0;
                }
                /* clean ehca_mr_t, without changing struct ib_mr and lock */
                save_mr = *e_mr;
                ehca_mr_deletenew(e_mr);

                /* set some MR values */
                e_mr->flags = save_mr.flags;
                e_mr->hwpage_size = save_mr.hwpage_size;
                e_mr->fmr_page_size = save_mr.fmr_page_size;
                e_mr->fmr_max_pages = save_mr.fmr_max_pages;
                e_mr->fmr_max_maps = save_mr.fmr_max_maps;
                e_mr->fmr_map_cnt = save_mr.fmr_map_cnt;

                ret = ehca_reg_mr(shca, e_mr, iova_start, size, acl,
                                  e_pd, pginfo, lkey, rkey, EHCA_REG_MR);
                if (ret) {
                        u32 offset = (u64)(&e_mr->flags) - (u64)e_mr;
                        memcpy(&e_mr->flags, &(save_mr.flags),
                               sizeof(struct ehca_mr) - offset);
                        goto ehca_rereg_mr_exit0;
                }
        }

ehca_rereg_mr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i shca=%p e_mr=%p "
                         "iova_start=%p size=%llx acl=%x e_pd=%p pginfo=%p "
                         "num_kpages=%llx lkey=%x rkey=%x rereg_1_hcall=%x "
                         "rereg_3_hcall=%x", ret, shca, e_mr, iova_start, size,
                         acl, e_pd, pginfo, pginfo->num_kpages, *lkey, *rkey,
                         rereg_1_hcall, rereg_3_hcall);
        return ret;
} /* end ehca_rereg_mr() */

/*----------------------------------------------------------------------*/

int ehca_unmap_one_fmr(struct ehca_shca *shca,
                       struct ehca_mr *e_fmr)
{
        int ret = 0;
        u64 h_ret;
        struct ehca_pd *e_pd =
                container_of(e_fmr->ib.ib_fmr.pd, struct ehca_pd, ib_pd);
        struct ehca_mr save_fmr;
        u32 tmp_lkey, tmp_rkey;
        struct ehca_mr_pginfo pginfo;
        struct ehca_mr_hipzout_parms hipzout;
        struct ehca_mr save_mr;

        if (e_fmr->fmr_max_pages <= MAX_RPAGES) {
                /*
                 * note: after using rereg hcall with len=0,
                 * rereg hcall must be used again for registering pages
                 */
                h_ret = hipz_h_reregister_pmr(shca->ipz_hca_handle, e_fmr, 0,
                                              0, 0, e_pd->fw_pd, 0, &hipzout);
                if (h_ret == H_SUCCESS) {
                        /* successful reregistration */
                        e_fmr->start = NULL;
                        e_fmr->size = 0;
                        tmp_lkey = hipzout.lkey;
                        tmp_rkey = hipzout.rkey;
                        return 0;
                }
                /*
                 * should not happen, because length checked above,
                 * FMRs are not shared and no MW bound to FMRs
                 */
                ehca_err(&shca->ib_device, "hipz_reregister_pmr failed "
                         "(Rereg1), h_ret=%lli e_fmr=%p hca_hndl=%llx "
                         "mr_hndl=%llx lkey=%x lkey_out=%x",
                         h_ret, e_fmr, shca->ipz_hca_handle.handle,
                         e_fmr->ipz_mr_handle.handle,
                         e_fmr->ib.ib_fmr.lkey, hipzout.lkey);
                /* try free and rereg */
        }

        /* first free old FMR */
        h_ret = hipz_h_free_resource_mr(shca->ipz_hca_handle, e_fmr);
        if (h_ret != H_SUCCESS) {
                ehca_err(&shca->ib_device, "hipz_free_mr failed, "
                         "h_ret=%lli e_fmr=%p hca_hndl=%llx mr_hndl=%llx "
                         "lkey=%x",
                         h_ret, e_fmr, shca->ipz_hca_handle.handle,
                         e_fmr->ipz_mr_handle.handle,
                         e_fmr->ib.ib_fmr.lkey);
                ret = ehca2ib_return_code(h_ret);
                goto ehca_unmap_one_fmr_exit0;
        }
        /* clean ehca_mr_t, without changing lock */
        save_fmr = *e_fmr;
        ehca_mr_deletenew(e_fmr);

        /* set some MR values */
        e_fmr->flags = save_fmr.flags;
        e_fmr->hwpage_size = save_fmr.hwpage_size;
        e_fmr->fmr_page_size = save_fmr.fmr_page_size;
        e_fmr->fmr_max_pages = save_fmr.fmr_max_pages;
        e_fmr->fmr_max_maps = save_fmr.fmr_max_maps;
        e_fmr->fmr_map_cnt = save_fmr.fmr_map_cnt;
        e_fmr->acl = save_fmr.acl;

        memset(&pginfo, 0, sizeof(pginfo));
        pginfo.type = EHCA_MR_PGI_FMR;
        ret = ehca_reg_mr(shca, e_fmr, NULL,
                          (e_fmr->fmr_max_pages * e_fmr->fmr_page_size),
                          e_fmr->acl, e_pd, &pginfo, &tmp_lkey,
                          &tmp_rkey, EHCA_REG_MR);
        if (ret) {
                u32 offset = (u64)(&e_fmr->flags) - (u64)e_fmr;
                memcpy(&e_fmr->flags, &(save_mr.flags),
                       sizeof(struct ehca_mr) - offset);
        }

ehca_unmap_one_fmr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i tmp_lkey=%x tmp_rkey=%x "
                         "fmr_max_pages=%x",
                         ret, tmp_lkey, tmp_rkey, e_fmr->fmr_max_pages);
        return ret;
} /* end ehca_unmap_one_fmr() */

/*----------------------------------------------------------------------*/

int ehca_reg_smr(struct ehca_shca *shca,
                 struct ehca_mr *e_origmr,
                 struct ehca_mr *e_newmr,
                 u64 *iova_start,
                 int acl,
                 struct ehca_pd *e_pd,
                 u32 *lkey, /*OUT*/
                 u32 *rkey) /*OUT*/
{
        int ret = 0;
        u64 h_ret;
        u32 hipz_acl;
        struct ehca_mr_hipzout_parms hipzout;

        ehca_mrmw_map_acl(acl, &hipz_acl);
        ehca_mrmw_set_pgsize_hipz_acl(e_origmr->hwpage_size, &hipz_acl);

        h_ret = hipz_h_register_smr(shca->ipz_hca_handle, e_newmr, e_origmr,
                                    (u64)iova_start, hipz_acl, e_pd->fw_pd,
                                    &hipzout);
        if (h_ret != H_SUCCESS) {
                ehca_err(&shca->ib_device, "hipz_reg_smr failed, h_ret=%lli "
                         "shca=%p e_origmr=%p e_newmr=%p iova_start=%p acl=%x "
                         "e_pd=%p hca_hndl=%llx mr_hndl=%llx lkey=%x",
                         h_ret, shca, e_origmr, e_newmr, iova_start, acl, e_pd,
                         shca->ipz_hca_handle.handle,
                         e_origmr->ipz_mr_handle.handle,
                         e_origmr->ib.ib_mr.lkey);
                ret = ehca2ib_return_code(h_ret);
                goto ehca_reg_smr_exit0;
        }
        /* successful registration */
        e_newmr->num_kpages = e_origmr->num_kpages;
        e_newmr->num_hwpages = e_origmr->num_hwpages;
        e_newmr->hwpage_size   = e_origmr->hwpage_size;
        e_newmr->start = iova_start;
        e_newmr->size = e_origmr->size;
        e_newmr->acl = acl;
        e_newmr->ipz_mr_handle = hipzout.handle;
        *lkey = hipzout.lkey;
        *rkey = hipzout.rkey;
        return 0;

ehca_reg_smr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i shca=%p e_origmr=%p "
                         "e_newmr=%p iova_start=%p acl=%x e_pd=%p",
                         ret, shca, e_origmr, e_newmr, iova_start, acl, e_pd);
        return ret;
} /* end ehca_reg_smr() */

/*----------------------------------------------------------------------*/
static inline void *ehca_calc_sectbase(int top, int dir, int idx)
{
        unsigned long ret = idx;
        ret |= dir << EHCA_DIR_INDEX_SHIFT;
        ret |= top << EHCA_TOP_INDEX_SHIFT;
        return __va(ret << SECTION_SIZE_BITS);
}

#define ehca_bmap_valid(entry) \
        ((u64)entry != (u64)EHCA_INVAL_ADDR)

static u64 ehca_reg_mr_section(int top, int dir, int idx, u64 *kpage,
                               struct ehca_shca *shca, struct ehca_mr *mr,
                               struct ehca_mr_pginfo *pginfo)
{
        u64 h_ret = 0;
        unsigned long page = 0;
        u64 rpage = __pa(kpage);
        int page_count;

        void *sectbase = ehca_calc_sectbase(top, dir, idx);
        if ((unsigned long)sectbase & (pginfo->hwpage_size - 1)) {
                ehca_err(&shca->ib_device, "reg_mr_section will probably fail:"
                                           "hwpage_size does not fit to "
                                           "section start address");
        }
        page_count = EHCA_SECTSIZE / pginfo->hwpage_size;

        while (page < page_count) {
                u64 rnum;
                for (rnum = 0; (rnum < MAX_RPAGES) && (page < page_count);
                     rnum++) {
                        void *pg = sectbase + ((page++) * pginfo->hwpage_size);
                        kpage[rnum] = __pa(pg);
                }

                h_ret = hipz_h_register_rpage_mr(shca->ipz_hca_handle, mr,
                        ehca_encode_hwpage_size(pginfo->hwpage_size),
                        0, rpage, rnum);

                if ((h_ret != H_SUCCESS) && (h_ret != H_PAGE_REGISTERED)) {
                        ehca_err(&shca->ib_device, "register_rpage_mr failed");
                        return h_ret;
                }
        }
        return h_ret;
}

static u64 ehca_reg_mr_sections(int top, int dir, u64 *kpage,
                                struct ehca_shca *shca, struct ehca_mr *mr,
                                struct ehca_mr_pginfo *pginfo)
{
        u64 hret = H_SUCCESS;
        int idx;

        for (idx = 0; idx < EHCA_MAP_ENTRIES; idx++) {
                if (!ehca_bmap_valid(ehca_bmap->top[top]->dir[dir]->ent[idx]))
                        continue;

                hret = ehca_reg_mr_section(top, dir, idx, kpage, shca, mr,
                                           pginfo);
                if ((hret != H_SUCCESS) && (hret != H_PAGE_REGISTERED))
                                return hret;
        }
        return hret;
}

static u64 ehca_reg_mr_dir_sections(int top, u64 *kpage, struct ehca_shca *shca,
                                    struct ehca_mr *mr,
                                    struct ehca_mr_pginfo *pginfo)
{
        u64 hret = H_SUCCESS;
        int dir;

        for (dir = 0; dir < EHCA_MAP_ENTRIES; dir++) {
                if (!ehca_bmap_valid(ehca_bmap->top[top]->dir[dir]))
                        continue;

                hret = ehca_reg_mr_sections(top, dir, kpage, shca, mr, pginfo);
                if ((hret != H_SUCCESS) && (hret != H_PAGE_REGISTERED))
                                return hret;
        }
        return hret;
}

/* register internal max-MR to internal SHCA */
int ehca_reg_internal_maxmr(
        struct ehca_shca *shca,
        struct ehca_pd *e_pd,
        struct ehca_mr **e_maxmr)  /*OUT*/
{
        int ret;
        struct ehca_mr *e_mr;
        u64 *iova_start;
        u64 size_maxmr;
        struct ehca_mr_pginfo pginfo;
        u32 num_kpages;
        u32 num_hwpages;
        u64 hw_pgsize;

        if (!ehca_bmap) {
                ret = -EFAULT;
                goto ehca_reg_internal_maxmr_exit0;
        }

        e_mr = ehca_mr_new();
        if (!e_mr) {
                ehca_err(&shca->ib_device, "out of memory");
                ret = -ENOMEM;
                goto ehca_reg_internal_maxmr_exit0;
        }
        e_mr->flags |= EHCA_MR_FLAG_MAXMR;

        /* register internal max-MR on HCA */
        size_maxmr = ehca_mr_len;
        iova_start = (u64 *)ehca_map_vaddr((void *)(KERNELBASE + PHYSICAL_START));
        num_kpages = NUM_CHUNKS(((u64)iova_start % PAGE_SIZE) + size_maxmr,
                                PAGE_SIZE);
        hw_pgsize = ehca_get_max_hwpage_size(shca);
        num_hwpages = NUM_CHUNKS(((u64)iova_start % hw_pgsize) + size_maxmr,
                                 hw_pgsize);

        memset(&pginfo, 0, sizeof(pginfo));
        pginfo.type = EHCA_MR_PGI_PHYS;
        pginfo.num_kpages = num_kpages;
        pginfo.num_hwpages = num_hwpages;
        pginfo.hwpage_size = hw_pgsize;
        pginfo.u.phy.addr = 0;
        pginfo.u.phy.size = size_maxmr;

        ret = ehca_reg_mr(shca, e_mr, iova_start, size_maxmr, 0, e_pd,
                          &pginfo, &e_mr->ib.ib_mr.lkey,
                          &e_mr->ib.ib_mr.rkey, EHCA_REG_BUSMAP_MR);
        if (ret) {
                ehca_err(&shca->ib_device, "reg of internal max MR failed, "
                         "e_mr=%p iova_start=%p size_maxmr=%llx num_kpages=%x "
                         "num_hwpages=%x", e_mr, iova_start, size_maxmr,
                         num_kpages, num_hwpages);
                goto ehca_reg_internal_maxmr_exit1;
        }

        /* successful registration of all pages */
        e_mr->ib.ib_mr.device = e_pd->ib_pd.device;
        e_mr->ib.ib_mr.pd = &e_pd->ib_pd;
        e_mr->ib.ib_mr.uobject = NULL;
        atomic_inc(&(e_pd->ib_pd.usecnt));
        *e_maxmr = e_mr;
        return 0;

ehca_reg_internal_maxmr_exit1:
        ehca_mr_delete(e_mr);
ehca_reg_internal_maxmr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i shca=%p e_pd=%p e_maxmr=%p",
                         ret, shca, e_pd, e_maxmr);
        return ret;
} /* end ehca_reg_internal_maxmr() */

/*----------------------------------------------------------------------*/

int ehca_reg_maxmr(struct ehca_shca *shca,
                   struct ehca_mr *e_newmr,
                   u64 *iova_start,
                   int acl,
                   struct ehca_pd *e_pd,
                   u32 *lkey,
                   u32 *rkey)
{
        u64 h_ret;
        struct ehca_mr *e_origmr = shca->maxmr;
        u32 hipz_acl;
        struct ehca_mr_hipzout_parms hipzout;

        ehca_mrmw_map_acl(acl, &hipz_acl);
        ehca_mrmw_set_pgsize_hipz_acl(e_origmr->hwpage_size, &hipz_acl);

        h_ret = hipz_h_register_smr(shca->ipz_hca_handle, e_newmr, e_origmr,
                                    (u64)iova_start, hipz_acl, e_pd->fw_pd,
                                    &hipzout);
        if (h_ret != H_SUCCESS) {
                ehca_err(&shca->ib_device, "hipz_reg_smr failed, h_ret=%lli "
                         "e_origmr=%p hca_hndl=%llx mr_hndl=%llx lkey=%x",
                         h_ret, e_origmr, shca->ipz_hca_handle.handle,
                         e_origmr->ipz_mr_handle.handle,
                         e_origmr->ib.ib_mr.lkey);
                return ehca2ib_return_code(h_ret);
        }
        /* successful registration */
        e_newmr->num_kpages = e_origmr->num_kpages;
        e_newmr->num_hwpages = e_origmr->num_hwpages;
        e_newmr->hwpage_size = e_origmr->hwpage_size;
        e_newmr->start = iova_start;
        e_newmr->size = e_origmr->size;
        e_newmr->acl = acl;
        e_newmr->ipz_mr_handle = hipzout.handle;
        *lkey = hipzout.lkey;
        *rkey = hipzout.rkey;
        return 0;
} /* end ehca_reg_maxmr() */

/*----------------------------------------------------------------------*/

int ehca_dereg_internal_maxmr(struct ehca_shca *shca)
{
        int ret;
        struct ehca_mr *e_maxmr;
        struct ib_pd *ib_pd;

        if (!shca->maxmr) {
                ehca_err(&shca->ib_device, "bad call, shca=%p", shca);
                ret = -EINVAL;
                goto ehca_dereg_internal_maxmr_exit0;
        }

        e_maxmr = shca->maxmr;
        ib_pd = e_maxmr->ib.ib_mr.pd;
        shca->maxmr = NULL; /* remove internal max-MR indication from SHCA */

        ret = ehca_dereg_mr(&e_maxmr->ib.ib_mr);
        if (ret) {
                ehca_err(&shca->ib_device, "dereg internal max-MR failed, "
                         "ret=%i e_maxmr=%p shca=%p lkey=%x",
                         ret, e_maxmr, shca, e_maxmr->ib.ib_mr.lkey);
                shca->maxmr = e_maxmr;
                goto ehca_dereg_internal_maxmr_exit0;
        }

        atomic_dec(&ib_pd->usecnt);

ehca_dereg_internal_maxmr_exit0:
        if (ret)
                ehca_err(&shca->ib_device, "ret=%i shca=%p shca->maxmr=%p",
                         ret, shca, shca->maxmr);
        return ret;
} /* end ehca_dereg_internal_maxmr() */

/*----------------------------------------------------------------------*/

/* check page list of map FMR verb for validness */
int ehca_fmr_check_page_list(struct ehca_mr *e_fmr,
                             u64 *page_list,
                             int list_len)
{
        u32 i;
        u64 *page;

        if ((list_len == 0) || (list_len > e_fmr->fmr_max_pages)) {
                ehca_gen_err("bad list_len, list_len=%x "
                             "e_fmr->fmr_max_pages=%x fmr=%p",
                             list_len, e_fmr->fmr_max_pages, e_fmr);
                return -EINVAL;
        }

        /* each page must be aligned */
        page = page_list;
        for (i = 0; i < list_len; i++) {
                if (*page % e_fmr->fmr_page_size) {
                        ehca_gen_err("bad page, i=%x *page=%llx page=%p fmr=%p "
                                     "fmr_page_size=%x", i, *page, page, e_fmr,
                                     e_fmr->fmr_page_size);
                        return -EINVAL;
                }
                page++;
        }

        return 0;
} /* end ehca_fmr_check_page_list() */

/*----------------------------------------------------------------------*/

/* PAGE_SIZE >= pginfo->hwpage_size */
static int ehca_set_pagebuf_user1(struct ehca_mr_pginfo *pginfo,
                                  u32 number,
                                  u64 *kpage)
{
        int ret = 0;
        u64 pgaddr;
        u32 j = 0;
        int hwpages_per_kpage = PAGE_SIZE / pginfo->hwpage_size;
        struct scatterlist **sg = &pginfo->u.usr.next_sg;

        while (*sg != NULL) {
                pgaddr = page_to_pfn(sg_page(*sg))
                        << PAGE_SHIFT;
                *kpage = pgaddr + (pginfo->next_hwpage *
                                   pginfo->hwpage_size);
                if (!(*kpage)) {
                        ehca_gen_err("pgaddr=%llx "
                                     "sg_dma_address=%llx "
                                     "entry=%llx next_hwpage=%llx",
                                     pgaddr, (u64)sg_dma_address(*sg),
                                     pginfo->u.usr.next_nmap,
                                     pginfo->next_hwpage);
                        return -EFAULT;
                }
                (pginfo->hwpage_cnt)++;
                (pginfo->next_hwpage)++;
                kpage++;
                if (pginfo->next_hwpage % hwpages_per_kpage == 0) {
                        (pginfo->kpage_cnt)++;
                        (pginfo->u.usr.next_nmap)++;
                        pginfo->next_hwpage = 0;
                        *sg = sg_next(*sg);
                }
                j++;
                if (j >= number)
                        break;
        }

        return ret;
}

/*
 * check given pages for contiguous layout
 * last page addr is returned in prev_pgaddr for further check
 */
static int ehca_check_kpages_per_ate(struct scatterlist **sg,
                                     int num_pages,
                                     u64 *prev_pgaddr)
{
        for (; *sg && num_pages > 0; *sg = sg_next(*sg), num_pages--) {
                u64 pgaddr = page_to_pfn(sg_page(*sg)) << PAGE_SHIFT;
                if (ehca_debug_level >= 3)
                        ehca_gen_dbg("chunk_page=%llx value=%016llx", pgaddr,
                                     *(u64 *)__va(pgaddr));
                if (pgaddr - PAGE_SIZE != *prev_pgaddr) {
                        ehca_gen_err("uncontiguous page found pgaddr=%llx "
                                     "prev_pgaddr=%llx entries_left_in_hwpage=%x",
                                     pgaddr, *prev_pgaddr, num_pages);
                        return -EINVAL;
                }
                *prev_pgaddr = pgaddr;
        }
        return 0;
}

/* PAGE_SIZE < pginfo->hwpage_size */
static int ehca_set_pagebuf_user2(struct ehca_mr_pginfo *pginfo,
                                  u32 number,
                                  u64 *kpage)
{
        int ret = 0;
        u64 pgaddr, prev_pgaddr;
        u32 j = 0;
        int kpages_per_hwpage = pginfo->hwpage_size / PAGE_SIZE;
        int nr_kpages = kpages_per_hwpage;
        struct scatterlist **sg = &pginfo->u.usr.next_sg;

        while (*sg != NULL) {

                if (nr_kpages == kpages_per_hwpage) {
                        pgaddr = (page_to_pfn(sg_page(*sg))
                                   << PAGE_SHIFT);
                        *kpage = pgaddr;
                        if (!(*kpage)) {
                                ehca_gen_err("pgaddr=%llx entry=%llx",
                                             pgaddr, pginfo->u.usr.next_nmap);
                                ret = -EFAULT;
                                return ret;
                        }
                        /*
                         * The first page in a hwpage must be aligned;
                         * the first MR page is exempt from this rule.
                         */
                        if (pgaddr & (pginfo->hwpage_size - 1)) {
                                if (pginfo->hwpage_cnt) {
                                        ehca_gen_err(
                                                "invalid alignment "
                                                "pgaddr=%llx entry=%llx "
                                                "mr_pgsize=%llx",
                                                pgaddr, pginfo->u.usr.next_nmap,
                                                pginfo->hwpage_size);
                                        ret = -EFAULT;
                                        return ret;
                                }
                                /* first MR page */
                                pginfo->kpage_cnt =
                                        (pgaddr &
                                         (pginfo->hwpage_size - 1)) >>
                                        PAGE_SHIFT;
                                nr_kpages -= pginfo->kpage_cnt;
                                *kpage = pgaddr &
                                         ~(pginfo->hwpage_size - 1);
                        }
                        if (ehca_debug_level >= 3) {
                                u64 val = *(u64 *)__va(pgaddr);
                                ehca_gen_dbg("kpage=%llx page=%llx "
                                             "value=%016llx",
                                             *kpage, pgaddr, val);
                        }
                        prev_pgaddr = pgaddr;
                        *sg = sg_next(*sg);
                        pginfo->kpage_cnt++;
                        pginfo->u.usr.next_nmap++;
                        nr_kpages--;
                        if (!nr_kpages)
                                goto next_kpage;
                        continue;
                }

                ret = ehca_check_kpages_per_ate(sg, nr_kpages,
                                                &prev_pgaddr);
                if (ret)
                        return ret;
                pginfo->kpage_cnt += nr_kpages;
                pginfo->u.usr.next_nmap += nr_kpages;

next_kpage:
                nr_kpages = kpages_per_hwpage;
                (pginfo->hwpage_cnt)++;
                kpage++;
                j++;
                if (j >= number)
                        break;
        }

        return ret;
}

static int ehca_set_pagebuf_phys(struct ehca_mr_pginfo *pginfo,
                                 u32 number, u64 *kpage)
{
        int ret = 0;
        u64 addr = pginfo->u.phy.addr;
        u64 size = pginfo->u.phy.size;
        u64 num_hw, offs_hw;
        u32 i = 0;

        num_hw  = NUM_CHUNKS((addr % pginfo->hwpage_size) + size,
                                pginfo->hwpage_size);
        offs_hw = (addr & ~(pginfo->hwpage_size - 1)) / pginfo->hwpage_size;

        while (pginfo->next_hwpage < offs_hw + num_hw) {
                /* sanity check */
                if ((pginfo->kpage_cnt >= pginfo->num_kpages) ||
                    (pginfo->hwpage_cnt >= pginfo->num_hwpages)) {
                        ehca_gen_err("kpage_cnt >= num_kpages, "
                                     "kpage_cnt=%llx num_kpages=%llx "
                                     "hwpage_cnt=%llx "
                                     "num_hwpages=%llx i=%x",
                                     pginfo->kpage_cnt,
                                     pginfo->num_kpages,
                                     pginfo->hwpage_cnt,
                                     pginfo->num_hwpages, i);
                        return -EFAULT;
                }
                *kpage = (addr & ~(pginfo->hwpage_size - 1)) +
                         (pginfo->next_hwpage * pginfo->hwpage_size);
                if ( !(*kpage) && addr ) {
                        ehca_gen_err("addr=%llx size=%llx "
                                     "next_hwpage=%llx", addr,
                                     size, pginfo->next_hwpage);
                        return -EFAULT;
                }
                (pginfo->hwpage_cnt)++;
                (pginfo->next_hwpage)++;
                if (PAGE_SIZE >= pginfo->hwpage_size) {
                        if (pginfo->next_hwpage %
                            (PAGE_SIZE / pginfo->hwpage_size) == 0)
                                (pginfo->kpage_cnt)++;
                } else
                        pginfo->kpage_cnt += pginfo->hwpage_size /
                                PAGE_SIZE;
                kpage++;
                i++;
                if (i >= number) break;
        }
        if (pginfo->next_hwpage >= offs_hw + num_hw) {
                pginfo->next_hwpage = 0;
        }

        return ret;
}

static int ehca_set_pagebuf_fmr(struct ehca_mr_pginfo *pginfo,
                                u32 number, u64 *kpage)
{
        int ret = 0;
        u64 *fmrlist;
        u32 i;

        /* loop over desired page_list entries */
        fmrlist = pginfo->u.fmr.page_list + pginfo->u.fmr.next_listelem;
        for (i = 0; i < number; i++) {
                *kpage = (*fmrlist & ~(pginfo->hwpage_size - 1)) +
                           pginfo->next_hwpage * pginfo->hwpage_size;
                if ( !(*kpage) ) {
                        ehca_gen_err("*fmrlist=%llx fmrlist=%p "
                                     "next_listelem=%llx next_hwpage=%llx",
                                     *fmrlist, fmrlist,
                                     pginfo->u.fmr.next_listelem,
                                     pginfo->next_hwpage);
                        return -EFAULT;
                }
                (pginfo->hwpage_cnt)++;
                if (pginfo->u.fmr.fmr_pgsize >= pginfo->hwpage_size) {
                        if (pginfo->next_hwpage %
                            (pginfo->u.fmr.fmr_pgsize /
                             pginfo->hwpage_size) == 0) {
                                (pginfo->kpage_cnt)++;
                                (pginfo->u.fmr.next_listelem)++;
                                fmrlist++;
                                pginfo->next_hwpage = 0;
                        } else
                                (pginfo->next_hwpage)++;
                } else {
                        unsigned int cnt_per_hwpage = pginfo->hwpage_size /
                                pginfo->u.fmr.fmr_pgsize;
                        unsigned int j;
                        u64 prev = *kpage;
                        /* check if adrs are contiguous */
                        for (j = 1; j < cnt_per_hwpage; j++) {
                                u64 p = fmrlist[j] & ~(pginfo->hwpage_size - 1);
                                if (prev + pginfo->u.fmr.fmr_pgsize != p) {
                                        ehca_gen_err("uncontiguous fmr pages "
                                                     "found prev=%llx p=%llx "
                                                     "idx=%x", prev, p, i + j);
                                        return -EINVAL;
                                }
                                prev = p;
                        }
                        pginfo->kpage_cnt += cnt_per_hwpage;
                        pginfo->u.fmr.next_listelem += cnt_per_hwpage;
                        fmrlist += cnt_per_hwpage;
                }
                kpage++;
        }
        return ret;
}

/* setup page buffer from page info */
int ehca_set_pagebuf(struct ehca_mr_pginfo *pginfo,
                     u32 number,
                     u64 *kpage)
{
        int ret;

        switch (pginfo->type) {
        case EHCA_MR_PGI_PHYS:
                ret = ehca_set_pagebuf_phys(pginfo, number, kpage);
                break;
        case EHCA_MR_PGI_USER:
                ret = PAGE_SIZE >= pginfo->hwpage_size ?
                        ehca_set_pagebuf_user1(pginfo, number, kpage) :
                        ehca_set_pagebuf_user2(pginfo, number, kpage);
                break;
        case EHCA_MR_PGI_FMR:
                ret = ehca_set_pagebuf_fmr(pginfo, number, kpage);
                break;
        default:
                ehca_gen_err("bad pginfo->type=%x", pginfo->type);
                ret = -EFAULT;
                break;
        }
        return ret;
} /* end ehca_set_pagebuf() */

/*----------------------------------------------------------------------*/

/*
 * check MR if it is a max-MR, i.e. uses whole memory
 * in case it's a max-MR 1 is returned, else 0
 */
int ehca_mr_is_maxmr(u64 size,
                     u64 *iova_start)
{
        /* a MR is treated as max-MR only if it fits following: */
        if ((size == ehca_mr_len) &&
            (iova_start == (void *)ehca_map_vaddr((void *)(KERNELBASE + PHYSICAL_START)))) {
                ehca_gen_dbg("this is a max-MR");
                return 1;
        } else
                return 0;
} /* end ehca_mr_is_maxmr() */

/*----------------------------------------------------------------------*/

/* map access control for MR/MW. This routine is used for MR and MW. */
void ehca_mrmw_map_acl(int ib_acl,
                       u32 *hipz_acl)
{
        *hipz_acl = 0;
        if (ib_acl & IB_ACCESS_REMOTE_READ)
                *hipz_acl |= HIPZ_ACCESSCTRL_R_READ;
        if (ib_acl & IB_ACCESS_REMOTE_WRITE)
                *hipz_acl |= HIPZ_ACCESSCTRL_R_WRITE;
        if (ib_acl & IB_ACCESS_REMOTE_ATOMIC)
                *hipz_acl |= HIPZ_ACCESSCTRL_R_ATOMIC;
        if (ib_acl & IB_ACCESS_LOCAL_WRITE)
                *hipz_acl |= HIPZ_ACCESSCTRL_L_WRITE;
        if (ib_acl & IB_ACCESS_MW_BIND)
                *hipz_acl |= HIPZ_ACCESSCTRL_MW_BIND;
} /* end ehca_mrmw_map_acl() */

/*----------------------------------------------------------------------*/

/* sets page size in hipz access control for MR/MW. */
void ehca_mrmw_set_pgsize_hipz_acl(u32 pgsize, u32 *hipz_acl) /*INOUT*/
{
        *hipz_acl |= (ehca_encode_hwpage_size(pgsize) << 24);
} /* end ehca_mrmw_set_pgsize_hipz_acl() */

/*----------------------------------------------------------------------*/

/*
 * reverse map access control for MR/MW.
 * This routine is used for MR and MW.
 */
void ehca_mrmw_reverse_map_acl(const u32 *hipz_acl,
                               int *ib_acl) /*OUT*/
{
        *ib_acl = 0;
        if (*hipz_acl & HIPZ_ACCESSCTRL_R_READ)
                *ib_acl |= IB_ACCESS_REMOTE_READ;
        if (*hipz_acl & HIPZ_ACCESSCTRL_R_WRITE)
                *ib_acl |= IB_ACCESS_REMOTE_WRITE;
        if (*hipz_acl & HIPZ_ACCESSCTRL_R_ATOMIC)
                *ib_acl |= IB_ACCESS_REMOTE_ATOMIC;
        if (*hipz_acl & HIPZ_ACCESSCTRL_L_WRITE)
                *ib_acl |= IB_ACCESS_LOCAL_WRITE;
        if (*hipz_acl & HIPZ_ACCESSCTRL_MW_BIND)
                *ib_acl |= IB_ACCESS_MW_BIND;
} /* end ehca_mrmw_reverse_map_acl() */


/*----------------------------------------------------------------------*/

/*
 * MR destructor and constructor
 * used in Reregister MR verb, sets all fields in ehca_mr_t to 0,
 * except struct ib_mr and spinlock
 */
void ehca_mr_deletenew(struct ehca_mr *mr)
{
        mr->flags = 0;
        mr->num_kpages = 0;
        mr->num_hwpages = 0;
        mr->acl = 0;
        mr->start = NULL;
        mr->fmr_page_size = 0;
        mr->fmr_max_pages = 0;
        mr->fmr_max_maps = 0;
        mr->fmr_map_cnt = 0;
        memset(&mr->ipz_mr_handle, 0, sizeof(mr->ipz_mr_handle));
        memset(&mr->galpas, 0, sizeof(mr->galpas));
} /* end ehca_mr_deletenew() */

int ehca_init_mrmw_cache(void)
{
        mr_cache = kmem_cache_create("ehca_cache_mr",
                                     sizeof(struct ehca_mr), 0,
                                     SLAB_HWCACHE_ALIGN,
                                     NULL);
        if (!mr_cache)
                return -ENOMEM;
        mw_cache = kmem_cache_create("ehca_cache_mw",
                                     sizeof(struct ehca_mw), 0,
                                     SLAB_HWCACHE_ALIGN,
                                     NULL);
        if (!mw_cache) {
                kmem_cache_destroy(mr_cache);
                mr_cache = NULL;
                return -ENOMEM;
        }
        return 0;
}

void ehca_cleanup_mrmw_cache(void)
{
        if (mr_cache)
                kmem_cache_destroy(mr_cache);
        if (mw_cache)
                kmem_cache_destroy(mw_cache);
}

static inline int ehca_init_top_bmap(struct ehca_top_bmap *ehca_top_bmap,
                                     int dir)
{
        if (!ehca_bmap_valid(ehca_top_bmap->dir[dir])) {
                ehca_top_bmap->dir[dir] =
                        kmalloc(sizeof(struct ehca_dir_bmap), GFP_KERNEL);
                if (!ehca_top_bmap->dir[dir])
                        return -ENOMEM;
                /* Set map block to 0xFF according to EHCA_INVAL_ADDR */
                memset(ehca_top_bmap->dir[dir], 0xFF, EHCA_ENT_MAP_SIZE);
        }
        return 0;
}

static inline int ehca_init_bmap(struct ehca_bmap *ehca_bmap, int top, int dir)
{
        if (!ehca_bmap_valid(ehca_bmap->top[top])) {
                ehca_bmap->top[top] =
                        kmalloc(sizeof(struct ehca_top_bmap), GFP_KERNEL);
                if (!ehca_bmap->top[top])
                        return -ENOMEM;
                /* Set map block to 0xFF according to EHCA_INVAL_ADDR */
                memset(ehca_bmap->top[top], 0xFF, EHCA_DIR_MAP_SIZE);
        }
        return ehca_init_top_bmap(ehca_bmap->top[top], dir);
}

static inline int ehca_calc_index(unsigned long i, unsigned long s)
{
        return (i >> s) & EHCA_INDEX_MASK;
}

void ehca_destroy_busmap(void)
{
        int top, dir;

        if (!ehca_bmap)
                return;

        for (top = 0; top < EHCA_MAP_ENTRIES; top++) {
                if (!ehca_bmap_valid(ehca_bmap->top[top]))
                        continue;
                for (dir = 0; dir < EHCA_MAP_ENTRIES; dir++) {
                        if (!ehca_bmap_valid(ehca_bmap->top[top]->dir[dir]))
                                continue;

                        kfree(ehca_bmap->top[top]->dir[dir]);
                }

                kfree(ehca_bmap->top[top]);
        }

        kfree(ehca_bmap);
        ehca_bmap = NULL;
}

static int ehca_update_busmap(unsigned long pfn, unsigned long nr_pages)
{
        unsigned long i, start_section, end_section;
        int top, dir, idx;

        if (!nr_pages)
                return 0;

        if (!ehca_bmap) {
                ehca_bmap = kmalloc(sizeof(struct ehca_bmap), GFP_KERNEL);
                if (!ehca_bmap)
                        return -ENOMEM;
                /* Set map block to 0xFF according to EHCA_INVAL_ADDR */
                memset(ehca_bmap, 0xFF, EHCA_TOP_MAP_SIZE);
        }

        start_section = (pfn * PAGE_SIZE) / EHCA_SECTSIZE;
        end_section = ((pfn + nr_pages) * PAGE_SIZE) / EHCA_SECTSIZE;
        for (i = start_section; i < end_section; i++) {
                int ret;
                top = ehca_calc_index(i, EHCA_TOP_INDEX_SHIFT);
                dir = ehca_calc_index(i, EHCA_DIR_INDEX_SHIFT);
                idx = i & EHCA_INDEX_MASK;

                ret = ehca_init_bmap(ehca_bmap, top, dir);
                if (ret) {
                        ehca_destroy_busmap();
                        return ret;
                }
                ehca_bmap->top[top]->dir[dir]->ent[idx] = ehca_mr_len;
                ehca_mr_len += EHCA_SECTSIZE;
        }
        return 0;
}

static int ehca_is_hugepage(unsigned long pfn)
{
        int page_order;

        if (pfn & EHCA_HUGEPAGE_PFN_MASK)
                return 0;

        page_order = compound_order(pfn_to_page(pfn));
        if (page_order + PAGE_SHIFT != EHCA_HUGEPAGESHIFT)
                return 0;

        return 1;
}

static int ehca_create_busmap_callback(unsigned long initial_pfn,
                                       unsigned long total_nr_pages, void *arg)
{
        int ret;
        unsigned long pfn, start_pfn, end_pfn, nr_pages;

        if ((total_nr_pages * PAGE_SIZE) < EHCA_HUGEPAGE_SIZE)
                return ehca_update_busmap(initial_pfn, total_nr_pages);

        /* Given chunk is >= 16GB -> check for hugepages */
        start_pfn = initial_pfn;
        end_pfn = initial_pfn + total_nr_pages;
        pfn = start_pfn;

        while (pfn < end_pfn) {
                if (ehca_is_hugepage(pfn)) {
                        /* Add mem found in front of the hugepage */
                        nr_pages = pfn - start_pfn;
                        ret = ehca_update_busmap(start_pfn, nr_pages);
                        if (ret)
                                return ret;
                        /* Skip the hugepage */
                        pfn += (EHCA_HUGEPAGE_SIZE / PAGE_SIZE);
                        start_pfn = pfn;

                } else
                        pfn += (EHCA_SECTSIZE / PAGE_SIZE);
        }

        /* Add mem found behind the hugepage(s)  */
        nr_pages = pfn - start_pfn;
        return ehca_update_busmap(start_pfn, nr_pages);
}

int ehca_create_busmap(void)
{
        int ret;

        ehca_mr_len = 0;
        ret = walk_system_ram_range(0, 1ULL << MAX_PHYSMEM_BITS, NULL,
                                   ehca_create_busmap_callback);
        return ret;
}

static int ehca_reg_bmap_mr_rpages(struct ehca_shca *shca,
                                   struct ehca_mr *e_mr,
                                   struct ehca_mr_pginfo *pginfo)
{
        int top;
        u64 hret, *kpage;

        kpage = ehca_alloc_fw_ctrlblock(GFP_KERNEL);
        if (!kpage) {
                ehca_err(&shca->ib_device, "kpage alloc failed");
                return -ENOMEM;
        }
        for (top = 0; top < EHCA_MAP_ENTRIES; top++) {
                if (!ehca_bmap_valid(ehca_bmap->top[top]))
                        continue;
                hret = ehca_reg_mr_dir_sections(top, kpage, shca, e_mr, pginfo);
                if ((hret != H_PAGE_REGISTERED) && (hret != H_SUCCESS))
                        break;
        }

        ehca_free_fw_ctrlblock(kpage);

        if (hret == H_SUCCESS)
                return 0; /* Everything is fine */
        else {
                ehca_err(&shca->ib_device, "ehca_reg_bmap_mr_rpages failed, "
                                 "h_ret=%lli e_mr=%p top=%x lkey=%x "
                                 "hca_hndl=%llx mr_hndl=%llx", hret, e_mr, top,
                                 e_mr->ib.ib_mr.lkey,
                                 shca->ipz_hca_handle.handle,
                                 e_mr->ipz_mr_handle.handle);
                return ehca2ib_return_code(hret);
        }
}

static u64 ehca_map_vaddr(void *caddr)
{
        int top, dir, idx;
        unsigned long abs_addr, offset;
        u64 entry;

        if (!ehca_bmap)
                return EHCA_INVAL_ADDR;

        abs_addr = __pa(caddr);
        top = ehca_calc_index(abs_addr, EHCA_TOP_INDEX_SHIFT + EHCA_SECTSHIFT);
        if (!ehca_bmap_valid(ehca_bmap->top[top]))
                return EHCA_INVAL_ADDR;

        dir = ehca_calc_index(abs_addr, EHCA_DIR_INDEX_SHIFT + EHCA_SECTSHIFT);
        if (!ehca_bmap_valid(ehca_bmap->top[top]->dir[dir]))
                return EHCA_INVAL_ADDR;

        idx = ehca_calc_index(abs_addr, EHCA_SECTSHIFT);

        entry = ehca_bmap->top[top]->dir[dir]->ent[idx];
        if (ehca_bmap_valid(entry)) {
                offset = (unsigned long)caddr & (EHCA_SECTSIZE - 1);
                return entry | offset;
        } else
                return EHCA_INVAL_ADDR;
}

static int ehca_dma_mapping_error(struct ib_device *dev, u64 dma_addr)
{
        return dma_addr == EHCA_INVAL_ADDR;
}

static u64 ehca_dma_map_single(struct ib_device *dev, void *cpu_addr,
                               size_t size, enum dma_data_direction direction)
{
        if (cpu_addr)
                return ehca_map_vaddr(cpu_addr);
        else
                return EHCA_INVAL_ADDR;
}

static void ehca_dma_unmap_single(struct ib_device *dev, u64 addr, size_t size,
                                  enum dma_data_direction direction)
{
        /* This is only a stub; nothing to be done here */
}

static u64 ehca_dma_map_page(struct ib_device *dev, struct page *page,
                             unsigned long offset, size_t size,
                             enum dma_data_direction direction)
{
        u64 addr;

        if (offset + size > PAGE_SIZE)
                return EHCA_INVAL_ADDR;

        addr = ehca_map_vaddr(page_address(page));
        if (!ehca_dma_mapping_error(dev, addr))
                addr += offset;

        return addr;
}

static void ehca_dma_unmap_page(struct ib_device *dev, u64 addr, size_t size,
                                enum dma_data_direction direction)
{
        /* This is only a stub; nothing to be done here */
}

static int ehca_dma_map_sg(struct ib_device *dev, struct scatterlist *sgl,
                           int nents, enum dma_data_direction direction)
{
        struct scatterlist *sg;
        int i;

        for_each_sg(sgl, sg, nents, i) {
                u64 addr;
                addr = ehca_map_vaddr(sg_virt(sg));
                if (ehca_dma_mapping_error(dev, addr))
                        return 0;

                sg->dma_address = addr;
                sg->dma_length = sg->length;
        }
        return nents;
}

static void ehca_dma_unmap_sg(struct ib_device *dev, struct scatterlist *sg,
                              int nents, enum dma_data_direction direction)
{
        /* This is only a stub; nothing to be done here */
}

static void ehca_dma_sync_single_for_cpu(struct ib_device *dev, u64 addr,
                                         size_t size,
                                         enum dma_data_direction dir)
{
        dma_sync_single_for_cpu(dev->dma_device, addr, size, dir);
}

static void ehca_dma_sync_single_for_device(struct ib_device *dev, u64 addr,
                                            size_t size,
                                            enum dma_data_direction dir)
{
        dma_sync_single_for_device(dev->dma_device, addr, size, dir);
}

static void *ehca_dma_alloc_coherent(struct ib_device *dev, size_t size,
                                     u64 *dma_handle, gfp_t flag)
{
        struct page *p;
        void *addr = NULL;
        u64 dma_addr;

        p = alloc_pages(flag, get_order(size));
        if (p) {
                addr = page_address(p);
                dma_addr = ehca_map_vaddr(addr);
                if (ehca_dma_mapping_error(dev, dma_addr)) {
                        free_pages((unsigned long)addr, get_order(size));
                        return NULL;
                }
                if (dma_handle)
                        *dma_handle = dma_addr;
                return addr;
        }
        return NULL;
}

static void ehca_dma_free_coherent(struct ib_device *dev, size_t size,
                                   void *cpu_addr, u64 dma_handle)
{
        if (cpu_addr && size)
                free_pages((unsigned long)cpu_addr, get_order(size));
}


struct ib_dma_mapping_ops ehca_dma_mapping_ops = {
        .mapping_error          = ehca_dma_mapping_error,
        .map_single             = ehca_dma_map_single,
        .unmap_single           = ehca_dma_unmap_single,
        .map_page               = ehca_dma_map_page,
        .unmap_page             = ehca_dma_unmap_page,
        .map_sg                 = ehca_dma_map_sg,
        .unmap_sg               = ehca_dma_unmap_sg,
        .sync_single_for_cpu    = ehca_dma_sync_single_for_cpu,
        .sync_single_for_device = ehca_dma_sync_single_for_device,
        .alloc_coherent         = ehca_dma_alloc_coherent,
        .free_coherent          = ehca_dma_free_coherent,
};