// SPDX-License-Identifier: GPL-2.0+
/*
 *  Copyright IBM Corp. 2001, 2018
 *  Author(s): Robert Burroughs
 *             Eric Rossman (edrossma@us.ibm.com)
 *             Cornelia Huck <cornelia.huck@de.ibm.com>
 *
 *  Hotplug & misc device support: Jochen Roehrig (roehrig@de.ibm.com)
 *  Major cleanup & driver split: Martin Schwidefsky <schwidefsky@de.ibm.com>
 *                                Ralph Wuerthner <rwuerthn@de.ibm.com>
 *  MSGTYPE restruct:             Holger Dengler <hd@linux.vnet.ibm.com>
 *  Multiple device nodes: Harald Freudenberger <freude@linux.ibm.com>
 */

#define pr_fmt(fmt) "zcrypt: " fmt

#include <linux/export.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/interrupt.h>
#include <linux/miscdevice.h>
#include <linux/fs.h>
#include <linux/slab.h>
#include <linux/atomic.h>
#include <linux/uaccess.h>
#include <linux/hw_random.h>
#include <linux/debugfs.h>
#include <linux/cdev.h>
#include <linux/ctype.h>
#include <linux/capability.h>
#include <asm/debug.h>

#define CREATE_TRACE_POINTS
#include <asm/trace/zcrypt.h>

#include "zcrypt_api.h"
#include "zcrypt_debug.h"

#include "zcrypt_msgtype6.h"
#include "zcrypt_msgtype50.h"
#include "zcrypt_ccamisc.h"
#include "zcrypt_ep11misc.h"

/*
 * Module description.
 */
MODULE_AUTHOR("IBM Corporation");
MODULE_DESCRIPTION("Cryptographic Coprocessor interface, " \
                   "Copyright IBM Corp. 2001, 2012");
MODULE_LICENSE("GPL");

unsigned int zcrypt_mempool_threshold = 5;
module_param_named(mempool_threshold, zcrypt_mempool_threshold, uint, 0400);
MODULE_PARM_DESC(mempool_threshold, "CCA and EP11 request/reply mempool minimal items (min: 1)");

/*
 * zcrypt tracepoint functions
 */
EXPORT_TRACEPOINT_SYMBOL(s390_zcrypt_req);
EXPORT_TRACEPOINT_SYMBOL(s390_zcrypt_rep);

DEFINE_SPINLOCK(zcrypt_list_lock);
LIST_HEAD(zcrypt_card_list);

static atomic_t zcrypt_open_count = ATOMIC_INIT(0);

static LIST_HEAD(zcrypt_ops_list);

/* Zcrypt related debug feature stuff. */
debug_info_t *zcrypt_dbf_info;

/*
 * Process a rescan of the transport layer.
 * Runs a synchronous AP bus rescan.
 * Returns true if something has changed (for example the
 * bus scan has found and build up new devices) and it is
 * worth to do a retry. Otherwise false is returned meaning
 * no changes on the AP bus level.
 */
static inline bool zcrypt_process_rescan(void)
{
        return ap_bus_force_rescan();
}

void zcrypt_msgtype_register(struct zcrypt_ops *zops)
{
        list_add_tail(&zops->list, &zcrypt_ops_list);
}

void zcrypt_msgtype_unregister(struct zcrypt_ops *zops)
{
        list_del_init(&zops->list);
}

struct zcrypt_ops *zcrypt_msgtype(unsigned char *name, int variant)
{
        struct zcrypt_ops *zops;

        list_for_each_entry(zops, &zcrypt_ops_list, list)
                if (zops->variant == variant &&
                    (!strncmp(zops->name, name, sizeof(zops->name))))
                        return zops;
        return NULL;
}
EXPORT_SYMBOL(zcrypt_msgtype);

/*
 * Multi device nodes extension functions.
 */

struct zcdn_device;

static void zcdn_device_release(struct device *dev);
static const struct class zcrypt_class = {
        .name = ZCRYPT_NAME,
        .dev_release = zcdn_device_release,
};
static dev_t zcrypt_devt;
static struct cdev zcrypt_cdev;

struct zcdn_device {
        struct device device;
        struct ap_perms perms;
};

#define to_zcdn_dev(x) container_of((x), struct zcdn_device, device)

#define ZCDN_MAX_NAME 32

static int zcdn_create(const char *name);
static int zcdn_destroy(const char *name);

/*
 * Find zcdn device by name.
 * Returns reference to the zcdn device which needs to be released
 * with put_device() after use.
 */
static inline struct zcdn_device *find_zcdndev_by_name(const char *name)
{
        struct device *dev = class_find_device_by_name(&zcrypt_class, name);

        return dev ? to_zcdn_dev(dev) : NULL;
}

/*
 * Find zcdn device by devt value.
 * Returns reference to the zcdn device which needs to be released
 * with put_device() after use.
 */
static inline struct zcdn_device *find_zcdndev_by_devt(dev_t devt)
{
        struct device *dev = class_find_device_by_devt(&zcrypt_class, devt);

        return dev ? to_zcdn_dev(dev) : NULL;
}

static ssize_t ioctlmask_show(struct device *dev,
                              struct device_attribute *attr,
                              char *buf)
{
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);
        int i, n;

        if (mutex_lock_interruptible(&ap_attr_mutex))
                return -ERESTARTSYS;

        n = sysfs_emit(buf, "0x");
        for (i = 0; i < sizeof(zcdndev->perms.ioctlm) / sizeof(long); i++)
                n += sysfs_emit_at(buf, n, "%016lx", zcdndev->perms.ioctlm[i]);
        n += sysfs_emit_at(buf, n, "\n");

        mutex_unlock(&ap_attr_mutex);

        return n;
}

static ssize_t ioctlmask_store(struct device *dev,
                               struct device_attribute *attr,
                               const char *buf, size_t count)
{
        int rc;
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);

        rc = ap_parse_mask_str(buf, zcdndev->perms.ioctlm,
                               AP_IOCTLS, &ap_attr_mutex);
        if (rc)
                return rc;

        return count;
}

static DEVICE_ATTR_RW(ioctlmask);

static ssize_t apmask_show(struct device *dev,
                           struct device_attribute *attr,
                           char *buf)
{
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);
        int i, n;

        if (mutex_lock_interruptible(&ap_attr_mutex))
                return -ERESTARTSYS;

        n = sysfs_emit(buf, "0x");
        for (i = 0; i < sizeof(zcdndev->perms.apm) / sizeof(long); i++)
                n += sysfs_emit_at(buf, n, "%016lx", zcdndev->perms.apm[i]);
        n += sysfs_emit_at(buf, n, "\n");

        mutex_unlock(&ap_attr_mutex);

        return n;
}

static ssize_t apmask_store(struct device *dev,
                            struct device_attribute *attr,
                            const char *buf, size_t count)
{
        int rc;
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);

        rc = ap_parse_mask_str(buf, zcdndev->perms.apm,
                               AP_DEVICES, &ap_attr_mutex);
        if (rc)
                return rc;

        return count;
}

static DEVICE_ATTR_RW(apmask);

static ssize_t aqmask_show(struct device *dev,
                           struct device_attribute *attr,
                           char *buf)
{
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);
        int i, n;

        if (mutex_lock_interruptible(&ap_attr_mutex))
                return -ERESTARTSYS;

        n = sysfs_emit(buf, "0x");
        for (i = 0; i < sizeof(zcdndev->perms.aqm) / sizeof(long); i++)
                n += sysfs_emit_at(buf, n, "%016lx", zcdndev->perms.aqm[i]);
        n += sysfs_emit_at(buf, n, "\n");

        mutex_unlock(&ap_attr_mutex);

        return n;
}

static ssize_t aqmask_store(struct device *dev,
                            struct device_attribute *attr,
                            const char *buf, size_t count)
{
        int rc;
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);

        rc = ap_parse_mask_str(buf, zcdndev->perms.aqm,
                               AP_DOMAINS, &ap_attr_mutex);
        if (rc)
                return rc;

        return count;
}

static DEVICE_ATTR_RW(aqmask);

static ssize_t admask_show(struct device *dev,
                           struct device_attribute *attr,
                           char *buf)
{
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);
        int i, n;

        if (mutex_lock_interruptible(&ap_attr_mutex))
                return -ERESTARTSYS;

        n = sysfs_emit(buf, "0x");
        for (i = 0; i < sizeof(zcdndev->perms.adm) / sizeof(long); i++)
                n += sysfs_emit_at(buf, n, "%016lx", zcdndev->perms.adm[i]);
        n += sysfs_emit_at(buf, n, "\n");

        mutex_unlock(&ap_attr_mutex);

        return n;
}

static ssize_t admask_store(struct device *dev,
                            struct device_attribute *attr,
                            const char *buf, size_t count)
{
        int rc;
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);

        rc = ap_parse_mask_str(buf, zcdndev->perms.adm,
                               AP_DOMAINS, &ap_attr_mutex);
        if (rc)
                return rc;

        return count;
}

static DEVICE_ATTR_RW(admask);

static struct attribute *zcdn_dev_attrs[] = {
        &dev_attr_ioctlmask.attr,
        &dev_attr_apmask.attr,
        &dev_attr_aqmask.attr,
        &dev_attr_admask.attr,
        NULL
};

static struct attribute_group zcdn_dev_attr_group = {
        .attrs = zcdn_dev_attrs
};

static const struct attribute_group *zcdn_dev_attr_groups[] = {
        &zcdn_dev_attr_group,
        NULL
};

static ssize_t zcdn_create_store(const struct class *class,
                                 const struct class_attribute *attr,
                                 const char *buf, size_t count)
{
        int rc;
        char name[ZCDN_MAX_NAME];

        strscpy(name, skip_spaces(buf), sizeof(name));

        rc = zcdn_create(strim(name));

        return rc ? rc : count;
}

static const struct class_attribute class_attr_zcdn_create =
        __ATTR(create, 0600, NULL, zcdn_create_store);

static ssize_t zcdn_destroy_store(const struct class *class,
                                  const struct class_attribute *attr,
                                  const char *buf, size_t count)
{
        int rc;
        char name[ZCDN_MAX_NAME];

        strscpy(name, skip_spaces(buf), sizeof(name));

        rc = zcdn_destroy(strim(name));

        return rc ? rc : count;
}

static const struct class_attribute class_attr_zcdn_destroy =
        __ATTR(destroy, 0600, NULL, zcdn_destroy_store);

static void zcdn_device_release(struct device *dev)
{
        struct zcdn_device *zcdndev = to_zcdn_dev(dev);

        ZCRYPT_DBF_INFO("%s releasing zcdn device %d:%d\n",
                        __func__, MAJOR(dev->devt), MINOR(dev->devt));

        kfree(zcdndev);
}

static int zcdn_create(const char *name)
{
        dev_t devt;
        int i, rc = 0;
        struct zcdn_device *zcdndev;

        if (mutex_lock_interruptible(&ap_attr_mutex))
                return -ERESTARTSYS;

        /* check if device node with this name already exists */
        if (name[0]) {
                zcdndev = find_zcdndev_by_name(name);
                if (zcdndev) {
                        put_device(&zcdndev->device);
                        rc = -EEXIST;
                        goto unlockout;
                }
        }

        /* find an unused minor number */
        for (i = 0; i < ZCRYPT_MAX_MINOR_NODES; i++) {
                devt = MKDEV(MAJOR(zcrypt_devt), MINOR(zcrypt_devt) + i);
                zcdndev = find_zcdndev_by_devt(devt);
                if (zcdndev)
                        put_device(&zcdndev->device);
                else
                        break;
        }
        if (i == ZCRYPT_MAX_MINOR_NODES) {
                rc = -ENOSPC;
                goto unlockout;
        }

        /* alloc and prepare a new zcdn device */
        zcdndev = kzalloc_obj(*zcdndev);
        if (!zcdndev) {
                rc = -ENOMEM;
                goto unlockout;
        }
        zcdndev->device.release = zcdn_device_release;
        zcdndev->device.class = &zcrypt_class;
        zcdndev->device.devt = devt;
        zcdndev->device.groups = zcdn_dev_attr_groups;
        if (name[0])
                rc = dev_set_name(&zcdndev->device, "%s", name);
        else
                rc = dev_set_name(&zcdndev->device, ZCRYPT_NAME "_%d", (int)MINOR(devt));
        if (rc) {
                kfree(zcdndev);
                goto unlockout;
        }
        rc = device_register(&zcdndev->device);
        if (rc) {
                put_device(&zcdndev->device);
                goto unlockout;
        }

        ZCRYPT_DBF_INFO("%s created zcdn device %d:%d\n",
                        __func__, MAJOR(devt), MINOR(devt));

unlockout:
        mutex_unlock(&ap_attr_mutex);
        return rc;
}

static int zcdn_destroy(const char *name)
{
        int rc = 0;
        struct zcdn_device *zcdndev;

        if (mutex_lock_interruptible(&ap_attr_mutex))
                return -ERESTARTSYS;

        /* try to find this zcdn device */
        zcdndev = find_zcdndev_by_name(name);
        if (!zcdndev) {
                rc = -ENOENT;
                goto unlockout;
        }

        /*
         * The zcdn device is not hard destroyed. It is subject to
         * reference counting and thus just needs to be unregistered.
         */
        put_device(&zcdndev->device);
        device_unregister(&zcdndev->device);

unlockout:
        mutex_unlock(&ap_attr_mutex);
        return rc;
}

static void zcdn_destroy_all(void)
{
        int i;
        dev_t devt;
        struct zcdn_device *zcdndev;

        mutex_lock(&ap_attr_mutex);
        for (i = 0; i < ZCRYPT_MAX_MINOR_NODES; i++) {
                devt = MKDEV(MAJOR(zcrypt_devt), MINOR(zcrypt_devt) + i);
                zcdndev = find_zcdndev_by_devt(devt);
                if (zcdndev) {
                        put_device(&zcdndev->device);
                        device_unregister(&zcdndev->device);
                }
        }
        mutex_unlock(&ap_attr_mutex);
}

/*
 * zcrypt_read (): Not supported beyond zcrypt 1.3.1.
 *
 * This function is not supported beyond zcrypt 1.3.1.
 */
static ssize_t zcrypt_read(struct file *filp, char __user *buf,
                           size_t count, loff_t *f_pos)
{
        return -EPERM;
}

/*
 * zcrypt_write(): Not allowed.
 *
 * Write is not allowed
 */
static ssize_t zcrypt_write(struct file *filp, const char __user *buf,
                            size_t count, loff_t *f_pos)
{
        return -EPERM;
}

/*
 * zcrypt_open(): Count number of users.
 *
 * Device open function to count number of users.
 */
static int zcrypt_open(struct inode *inode, struct file *filp)
{
        struct ap_perms *perms = &ap_perms;

        if (filp->f_inode->i_cdev == &zcrypt_cdev) {
                struct zcdn_device *zcdndev;

                if (mutex_lock_interruptible(&ap_attr_mutex))
                        return -ERESTARTSYS;
                zcdndev = find_zcdndev_by_devt(filp->f_inode->i_rdev);
                /* find returns a reference, no get_device() needed */
                mutex_unlock(&ap_attr_mutex);
                if (zcdndev)
                        perms = &zcdndev->perms;
        }
        filp->private_data = (void *)perms;

        atomic_inc(&zcrypt_open_count);
        return stream_open(inode, filp);
}

/*
 * zcrypt_release(): Count number of users.
 *
 * Device close function to count number of users.
 */
static int zcrypt_release(struct inode *inode, struct file *filp)
{
        if (filp->f_inode->i_cdev == &zcrypt_cdev) {
                struct zcdn_device *zcdndev;

                mutex_lock(&ap_attr_mutex);
                zcdndev = find_zcdndev_by_devt(filp->f_inode->i_rdev);
                mutex_unlock(&ap_attr_mutex);
                if (zcdndev) {
                        /* 2 puts here: one for find, one for open */
                        put_device(&zcdndev->device);
                        put_device(&zcdndev->device);
                }
        }

        atomic_dec(&zcrypt_open_count);
        return 0;
}

static inline int zcrypt_check_ioctl(struct ap_perms *perms,
                                     unsigned int cmd)
{
        int rc = -EPERM;
        int ioctlnr = (cmd & _IOC_NRMASK) >> _IOC_NRSHIFT;

        if (ioctlnr > 0 && ioctlnr < AP_IOCTLS) {
                if (test_bit_inv(ioctlnr, perms->ioctlm))
                        rc = 0;
        }

        if (rc)
                ZCRYPT_DBF_WARN("%s ioctl check failed: ioctlnr=0x%04x rc=%d\n",
                                __func__, ioctlnr, rc);

        return rc;
}

static inline bool zcrypt_check_card(struct ap_perms *perms, int card)
{
        return test_bit_inv(card, perms->apm) ? true : false;
}

static inline bool zcrypt_check_queue(struct ap_perms *perms, int queue)
{
        return test_bit_inv(queue, perms->aqm) ? true : false;
}

static inline struct zcrypt_queue *zcrypt_pick_queue(struct zcrypt_card *zc,
                                                     struct zcrypt_queue *zq,
                                                     struct module **pmod,
                                                     unsigned int weight)
{
        if (!zq || !try_module_get(zq->queue->ap_dev.device.driver->owner))
                return NULL;
        zcrypt_card_get(zc);
        zcrypt_queue_get(zq);
        get_device(&zq->queue->ap_dev.device);
        atomic_add(weight, &zc->load);
        atomic_add(weight, &zq->load);
        zq->request_count++;
        *pmod = zq->queue->ap_dev.device.driver->owner;
        return zq;
}

static inline void zcrypt_drop_queue(struct zcrypt_card *zc,
                                     struct zcrypt_queue *zq,
                                     struct module *mod,
                                     unsigned int weight)
{
        zq->request_count--;
        atomic_sub(weight, &zc->load);
        atomic_sub(weight, &zq->load);
        put_device(&zq->queue->ap_dev.device);
        zcrypt_queue_put(zq);
        zcrypt_card_put(zc);
        module_put(mod);
}

static inline bool zcrypt_card_compare(struct zcrypt_card *zc,
                                       struct zcrypt_card *pref_zc,
                                       unsigned int weight,
                                       unsigned int pref_weight)
{
        if (!pref_zc)
                return true;
        weight += atomic_read(&zc->load);
        pref_weight += atomic_read(&pref_zc->load);
        if (weight == pref_weight)
                return atomic64_read(&zc->card->total_request_count) <
                        atomic64_read(&pref_zc->card->total_request_count);
        return weight < pref_weight;
}

static inline bool zcrypt_queue_compare(struct zcrypt_queue *zq,
                                        struct zcrypt_queue *pref_zq,
                                        unsigned int weight,
                                        unsigned int pref_weight)
{
        if (!pref_zq)
                return true;
        weight += atomic_read(&zq->load);
        pref_weight += atomic_read(&pref_zq->load);
        if (weight == pref_weight)
                return zq->queue->total_request_count <
                        pref_zq->queue->total_request_count;
        return weight < pref_weight;
}

/*
 * zcrypt ioctls.
 */
static long zcrypt_rsa_modexpo(struct ap_perms *perms,
                               struct zcrypt_track *tr,
                               struct ica_rsa_modexpo *mex)
{
        struct zcrypt_card *zc, *pref_zc;
        struct zcrypt_queue *zq, *pref_zq;
        struct ap_message ap_msg;
        unsigned int wgt = 0, pref_wgt = 0;
        unsigned int func_code = 0;
        int cpen, qpen, qid = 0, rc;
        struct module *mod;

        trace_s390_zcrypt_req(mex, TP_ICARSAMODEXPO);

        rc = ap_init_apmsg(&ap_msg, 0);
        if (rc)
                goto out;

        if (mex->outputdatalength < mex->inputdatalength) {
                rc = -EINVAL;
                goto out;
        }

        /*
         * As long as outputdatalength is big enough, we can set the
         * outputdatalength equal to the inputdatalength, since that is the
         * number of bytes we will copy in any case
         */
        mex->outputdatalength = mex->inputdatalength;

        rc = get_rsa_modex_fc(mex, &func_code);
        if (rc)
                goto out;

        pref_zc = NULL;
        pref_zq = NULL;
        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                /* Check for usable accelerator or CCA card */
                if (!zc->online || !zc->card->config || zc->card->chkstop ||
                    !(zc->card->hwinfo.accel || zc->card->hwinfo.cca))
                        continue;
                /* Check for size limits */
                if (zc->min_mod_size > mex->inputdatalength ||
                    zc->max_mod_size < mex->inputdatalength)
                        continue;
                /* check if device node has admission for this card */
                if (!zcrypt_check_card(perms, zc->card->id))
                        continue;
                /* get weight index of the card device  */
                wgt = zc->speed_rating[func_code];
                /* penalty if this msg was previously sent via this card */
                cpen = (tr && tr->again_counter && tr->last_qid &&
                        AP_QID_CARD(tr->last_qid) == zc->card->id) ?
                        TRACK_AGAIN_CARD_WEIGHT_PENALTY : 0;
                if (!zcrypt_card_compare(zc, pref_zc, wgt + cpen, pref_wgt))
                        continue;
                for_each_zcrypt_queue(zq, zc) {
                        /* check if device is usable and eligible */
                        if (!zq->online || !zq->ops->rsa_modexpo ||
                            !ap_queue_usable(zq->queue))
                                continue;
                        /* check if device node has admission for this queue */
                        if (!zcrypt_check_queue(perms,
                                                AP_QID_QUEUE(zq->queue->qid)))
                                continue;
                        /* penalty if the msg was previously sent at this qid */
                        qpen = (tr && tr->again_counter && tr->last_qid &&
                                tr->last_qid == zq->queue->qid) ?
                                TRACK_AGAIN_QUEUE_WEIGHT_PENALTY : 0;
                        if (!zcrypt_queue_compare(zq, pref_zq,
                                                  wgt + cpen + qpen, pref_wgt))
                                continue;
                        pref_zc = zc;
                        pref_zq = zq;
                        pref_wgt = wgt + cpen + qpen;
                }
        }
        pref_zq = zcrypt_pick_queue(pref_zc, pref_zq, &mod, wgt);
        spin_unlock(&zcrypt_list_lock);

        if (!pref_zq) {
                pr_debug("no matching queue found => ENODEV\n");
                rc = -ENODEV;
                goto out;
        }

        qid = pref_zq->queue->qid;
        rc = pref_zq->ops->rsa_modexpo(pref_zq, mex, &ap_msg);

        spin_lock(&zcrypt_list_lock);
        zcrypt_drop_queue(pref_zc, pref_zq, mod, wgt);
        spin_unlock(&zcrypt_list_lock);

out:
        ap_release_apmsg(&ap_msg);
        if (tr) {
                tr->last_rc = rc;
                tr->last_qid = qid;
        }
        trace_s390_zcrypt_rep(mex, func_code, rc,
                              AP_QID_CARD(qid), AP_QID_QUEUE(qid),
                              ap_msg.psmid);
        return rc;
}

static long zcrypt_rsa_crt(struct ap_perms *perms,
                           struct zcrypt_track *tr,
                           struct ica_rsa_modexpo_crt *crt)
{
        struct zcrypt_card *zc, *pref_zc;
        struct zcrypt_queue *zq, *pref_zq;
        struct ap_message ap_msg;
        unsigned int wgt = 0, pref_wgt = 0;
        unsigned int func_code = 0;
        int cpen, qpen, qid = 0, rc;
        struct module *mod;

        trace_s390_zcrypt_req(crt, TP_ICARSACRT);

        rc = ap_init_apmsg(&ap_msg, 0);
        if (rc)
                goto out;

        if (crt->outputdatalength < crt->inputdatalength) {
                rc = -EINVAL;
                goto out;
        }

        /*
         * As long as outputdatalength is big enough, we can set the
         * outputdatalength equal to the inputdatalength, since that is the
         * number of bytes we will copy in any case
         */
        crt->outputdatalength = crt->inputdatalength;

        rc = get_rsa_crt_fc(crt, &func_code);
        if (rc)
                goto out;

        pref_zc = NULL;
        pref_zq = NULL;
        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                /* Check for usable accelerator or CCA card */
                if (!zc->online || !zc->card->config || zc->card->chkstop ||
                    !(zc->card->hwinfo.accel || zc->card->hwinfo.cca))
                        continue;
                /* Check for size limits */
                if (zc->min_mod_size > crt->inputdatalength ||
                    zc->max_mod_size < crt->inputdatalength)
                        continue;
                /* check if device node has admission for this card */
                if (!zcrypt_check_card(perms, zc->card->id))
                        continue;
                /* get weight index of the card device  */
                wgt = zc->speed_rating[func_code];
                /* penalty if this msg was previously sent via this card */
                cpen = (tr && tr->again_counter && tr->last_qid &&
                        AP_QID_CARD(tr->last_qid) == zc->card->id) ?
                        TRACK_AGAIN_CARD_WEIGHT_PENALTY : 0;
                if (!zcrypt_card_compare(zc, pref_zc, wgt + cpen, pref_wgt))
                        continue;
                for_each_zcrypt_queue(zq, zc) {
                        /* check if device is usable and eligible */
                        if (!zq->online || !zq->ops->rsa_modexpo_crt ||
                            !ap_queue_usable(zq->queue))
                                continue;
                        /* check if device node has admission for this queue */
                        if (!zcrypt_check_queue(perms,
                                                AP_QID_QUEUE(zq->queue->qid)))
                                continue;
                        /* penalty if the msg was previously sent at this qid */
                        qpen = (tr && tr->again_counter && tr->last_qid &&
                                tr->last_qid == zq->queue->qid) ?
                                TRACK_AGAIN_QUEUE_WEIGHT_PENALTY : 0;
                        if (!zcrypt_queue_compare(zq, pref_zq,
                                                  wgt + cpen + qpen, pref_wgt))
                                continue;
                        pref_zc = zc;
                        pref_zq = zq;
                        pref_wgt = wgt + cpen + qpen;
                }
        }
        pref_zq = zcrypt_pick_queue(pref_zc, pref_zq, &mod, wgt);
        spin_unlock(&zcrypt_list_lock);

        if (!pref_zq) {
                pr_debug("no matching queue found => ENODEV\n");
                rc = -ENODEV;
                goto out;
        }

        qid = pref_zq->queue->qid;
        rc = pref_zq->ops->rsa_modexpo_crt(pref_zq, crt, &ap_msg);

        spin_lock(&zcrypt_list_lock);
        zcrypt_drop_queue(pref_zc, pref_zq, mod, wgt);
        spin_unlock(&zcrypt_list_lock);

out:
        ap_release_apmsg(&ap_msg);
        if (tr) {
                tr->last_rc = rc;
                tr->last_qid = qid;
        }
        trace_s390_zcrypt_rep(crt, func_code, rc,
                              AP_QID_CARD(qid), AP_QID_QUEUE(qid),
                              ap_msg.psmid);
        return rc;
}

static long _zcrypt_send_cprb(u32 xflags, struct ap_perms *perms,
                              struct zcrypt_track *tr,
                              struct ica_xcRB *xcrb)
{
        bool userspace = xflags & ZCRYPT_XFLAG_USERSPACE;
        struct zcrypt_card *zc, *pref_zc;
        struct zcrypt_queue *zq, *pref_zq;
        struct ap_message ap_msg;
        unsigned int wgt = 0, pref_wgt = 0;
        unsigned int func_code = 0;
        unsigned short *domain, tdom;
        int cpen, qpen, qid = 0, rc;
        struct module *mod;

        trace_s390_zcrypt_req(xcrb, TB_ZSECSENDCPRB);

        xcrb->status = 0;

        rc = ap_init_apmsg(&ap_msg, xflags & ZCRYPT_XFLAG_NOMEMALLOC ?
                           AP_MSG_FLAG_MEMPOOL : 0);
        if (rc)
                goto out;

        rc = prep_cca_ap_msg(userspace, xcrb, &ap_msg, &func_code, &domain);
        if (rc)
                goto out;
        print_hex_dump_debug("ccareq: ", DUMP_PREFIX_ADDRESS, 16, 1,
                             ap_msg.msg, ap_msg.len, false);

        tdom = *domain;
        if (perms != &ap_perms && tdom < AP_DOMAINS) {
                if (ap_msg.flags & AP_MSG_FLAG_ADMIN) {
                        if (!test_bit_inv(tdom, perms->adm)) {
                                rc = -ENODEV;
                                goto out;
                        }
                } else if ((ap_msg.flags & AP_MSG_FLAG_USAGE) == 0) {
                        rc = -EOPNOTSUPP;
                        goto out;
                }
        }
        /*
         * If a valid target domain is set and this domain is NOT a usage
         * domain but a control only domain, autoselect target domain.
         */
        if (tdom < AP_DOMAINS &&
            !ap_test_config_usage_domain(tdom) &&
            ap_test_config_ctrl_domain(tdom))
                tdom = AUTOSEL_DOM;

        pref_zc = NULL;
        pref_zq = NULL;
        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                /* Check for usable CCA card */
                if (!zc->online || !zc->card->config || zc->card->chkstop ||
                    !zc->card->hwinfo.cca)
                        continue;
                /* Check for user selected CCA card */
                if (xcrb->user_defined != AUTOSELECT &&
                    xcrb->user_defined != zc->card->id)
                        continue;
                /* check if request size exceeds card max msg size */
                if (ap_msg.len > zc->card->maxmsgsize)
                        continue;
                /* check if device node has admission for this card */
                if (!zcrypt_check_card(perms, zc->card->id))
                        continue;
                /* get weight index of the card device  */
                wgt = speed_idx_cca(func_code) * zc->speed_rating[SECKEY];
                /* penalty if this msg was previously sent via this card */
                cpen = (tr && tr->again_counter && tr->last_qid &&
                        AP_QID_CARD(tr->last_qid) == zc->card->id) ?
                        TRACK_AGAIN_CARD_WEIGHT_PENALTY : 0;
                if (!zcrypt_card_compare(zc, pref_zc, wgt + cpen, pref_wgt))
                        continue;
                for_each_zcrypt_queue(zq, zc) {
                        /* check for device usable and eligible */
                        if (!zq->online || !zq->ops->send_cprb ||
                            !ap_queue_usable(zq->queue) ||
                            (tdom != AUTOSEL_DOM &&
                             tdom != AP_QID_QUEUE(zq->queue->qid)))
                                continue;
                        /* check if device node has admission for this queue */
                        if (!zcrypt_check_queue(perms,
                                                AP_QID_QUEUE(zq->queue->qid)))
                                continue;
                        /* penalty if the msg was previously sent at this qid */
                        qpen = (tr && tr->again_counter && tr->last_qid &&
                                tr->last_qid == zq->queue->qid) ?
                                TRACK_AGAIN_QUEUE_WEIGHT_PENALTY : 0;
                        if (!zcrypt_queue_compare(zq, pref_zq,
                                                  wgt + cpen + qpen, pref_wgt))
                                continue;
                        pref_zc = zc;
                        pref_zq = zq;
                        pref_wgt = wgt + cpen + qpen;
                }
        }
        pref_zq = zcrypt_pick_queue(pref_zc, pref_zq, &mod, wgt);
        spin_unlock(&zcrypt_list_lock);

        if (!pref_zq) {
                pr_debug("no match for address %02x.%04x => ENODEV\n",
                         xcrb->user_defined, *domain);
                rc = -ENODEV;
                goto out;
        }

        /* in case of auto select, provide the correct domain */
        qid = pref_zq->queue->qid;
        if (*domain == AUTOSEL_DOM)
                *domain = AP_QID_QUEUE(qid);

        rc = pref_zq->ops->send_cprb(userspace, pref_zq, xcrb, &ap_msg);
        if (!rc) {
                print_hex_dump_debug("ccarpl: ", DUMP_PREFIX_ADDRESS, 16, 1,
                                     ap_msg.msg, ap_msg.len, false);
        }

        spin_lock(&zcrypt_list_lock);
        zcrypt_drop_queue(pref_zc, pref_zq, mod, wgt);
        spin_unlock(&zcrypt_list_lock);

out:
        ap_release_apmsg(&ap_msg);
        if (tr) {
                tr->last_rc = rc;
                tr->last_qid = qid;
        }
        trace_s390_zcrypt_rep(xcrb, func_code, rc,
                              AP_QID_CARD(qid), AP_QID_QUEUE(qid),
                              ap_msg.psmid);
        return rc;
}

long zcrypt_send_cprb(struct ica_xcRB *xcrb, u32 xflags)
{
        struct zcrypt_track tr;
        int rc;

        memset(&tr, 0, sizeof(tr));

        do {
                rc = _zcrypt_send_cprb(xflags, &ap_perms, &tr, xcrb);
        } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);

        /* on ENODEV failure: retry once again after a requested rescan */
        if (rc == -ENODEV && zcrypt_process_rescan())
                do {
                        rc = _zcrypt_send_cprb(xflags, &ap_perms, &tr, xcrb);
                } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);
        if (rc == -EAGAIN && tr.again_counter >= TRACK_AGAIN_MAX)
                rc = -EIO;
        if (rc)
                pr_debug("rc=%d\n", rc);

        return rc;
}
EXPORT_SYMBOL(zcrypt_send_cprb);

static bool is_desired_ep11_card(unsigned int dev_id,
                                 unsigned short target_num,
                                 struct ep11_target_dev *targets)
{
        while (target_num-- > 0) {
                if (targets->ap_id == dev_id || targets->ap_id == AUTOSEL_AP)
                        return true;
                targets++;
        }
        return false;
}

static bool is_desired_ep11_queue(unsigned int dev_qid,
                                  unsigned short target_num,
                                  struct ep11_target_dev *targets)
{
        int card = AP_QID_CARD(dev_qid), dom = AP_QID_QUEUE(dev_qid);

        while (target_num-- > 0) {
                if ((targets->ap_id == card || targets->ap_id == AUTOSEL_AP) &&
                    (targets->dom_id == dom || targets->dom_id == AUTOSEL_DOM))
                        return true;
                targets++;
        }
        return false;
}

static long _zcrypt_send_ep11_cprb(u32 xflags, struct ap_perms *perms,
                                   struct zcrypt_track *tr,
                                   struct ep11_urb *xcrb)
{
        bool userspace = xflags & ZCRYPT_XFLAG_USERSPACE;
        struct zcrypt_card *zc, *pref_zc;
        struct zcrypt_queue *zq, *pref_zq;
        struct ep11_target_dev *targets = NULL;
        unsigned short target_num;
        unsigned int wgt = 0, pref_wgt = 0;
        unsigned int func_code = 0, domain;
        struct ap_message ap_msg;
        int cpen, qpen, qid = 0, rc;
        struct module *mod;

        trace_s390_zcrypt_req(xcrb, TP_ZSENDEP11CPRB);

        rc = ap_init_apmsg(&ap_msg, xflags & ZCRYPT_XFLAG_NOMEMALLOC ?
                           AP_MSG_FLAG_MEMPOOL : 0);
        if (rc)
                goto out;

        target_num = (unsigned short)xcrb->targets_num;

        /* empty list indicates autoselect (all available targets) */
        rc = -ENOMEM;
        if (target_num != 0) {
                if (userspace) {
                        targets = kzalloc_objs(*targets, target_num);
                        if (!targets)
                                goto out;
                        if (copy_from_user(targets, xcrb->targets,
                                           target_num * sizeof(*targets))) {
                                rc = -EFAULT;
                                goto out;
                        }
                } else {
                        targets = (struct ep11_target_dev __force __kernel *)xcrb->targets;
                }
        }

        rc = prep_ep11_ap_msg(userspace, xcrb, &ap_msg, &func_code, &domain);
        if (rc)
                goto out;
        print_hex_dump_debug("ep11req: ", DUMP_PREFIX_ADDRESS, 16, 1,
                             ap_msg.msg, ap_msg.len, false);

        if (perms != &ap_perms && domain < AUTOSEL_DOM) {
                if (ap_msg.flags & AP_MSG_FLAG_ADMIN) {
                        if (!test_bit_inv(domain, perms->adm)) {
                                rc = -ENODEV;
                                goto out;
                        }
                } else if ((ap_msg.flags & AP_MSG_FLAG_USAGE) == 0) {
                        rc = -EOPNOTSUPP;
                        goto out;
                }
        }

        pref_zc = NULL;
        pref_zq = NULL;
        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                /* Check for usable EP11 card */
                if (!zc->online || !zc->card->config || zc->card->chkstop ||
                    !zc->card->hwinfo.ep11)
                        continue;
                /* Check for user selected EP11 card */
                if (targets &&
                    !is_desired_ep11_card(zc->card->id, target_num, targets))
                        continue;
                /* check if request size exceeds card max msg size */
                if (ap_msg.len > zc->card->maxmsgsize)
                        continue;
                /* check if device node has admission for this card */
                if (!zcrypt_check_card(perms, zc->card->id))
                        continue;
                /* get weight index of the card device  */
                wgt = speed_idx_ep11(func_code) * zc->speed_rating[SECKEY];
                /* penalty if this msg was previously sent via this card */
                cpen = (tr && tr->again_counter && tr->last_qid &&
                        AP_QID_CARD(tr->last_qid) == zc->card->id) ?
                        TRACK_AGAIN_CARD_WEIGHT_PENALTY : 0;
                if (!zcrypt_card_compare(zc, pref_zc, wgt + cpen, pref_wgt))
                        continue;
                for_each_zcrypt_queue(zq, zc) {
                        /* check if device is usable and eligible */
                        if (!zq->online || !zq->ops->send_ep11_cprb ||
                            !ap_queue_usable(zq->queue) ||
                            (targets &&
                             !is_desired_ep11_queue(zq->queue->qid,
                                                    target_num, targets)))
                                continue;
                        /* check if device node has admission for this queue */
                        if (!zcrypt_check_queue(perms,
                                                AP_QID_QUEUE(zq->queue->qid)))
                                continue;
                        /* penalty if the msg was previously sent at this qid */
                        qpen = (tr && tr->again_counter && tr->last_qid &&
                                tr->last_qid == zq->queue->qid) ?
                                TRACK_AGAIN_QUEUE_WEIGHT_PENALTY : 0;
                        if (!zcrypt_queue_compare(zq, pref_zq,
                                                  wgt + cpen + qpen, pref_wgt))
                                continue;
                        pref_zc = zc;
                        pref_zq = zq;
                        pref_wgt = wgt + cpen + qpen;
                }
        }
        pref_zq = zcrypt_pick_queue(pref_zc, pref_zq, &mod, wgt);
        spin_unlock(&zcrypt_list_lock);

        if (!pref_zq) {
                if (targets && target_num == 1) {
                        pr_debug("no match for address %02x.%04x => ENODEV\n",
                                 (int)targets->ap_id, (int)targets->dom_id);
                } else if (targets) {
                        pr_debug("no match for %d target addrs => ENODEV\n",
                                 (int)target_num);
                } else {
                        pr_debug("no match for address ff.ffff => ENODEV\n");
                }
                rc = -ENODEV;
                goto out;
        }

        qid = pref_zq->queue->qid;
        rc = pref_zq->ops->send_ep11_cprb(userspace, pref_zq, xcrb, &ap_msg);
        if (!rc) {
                print_hex_dump_debug("ep11rpl: ", DUMP_PREFIX_ADDRESS, 16, 1,
                                     ap_msg.msg, ap_msg.len, false);
        }

        spin_lock(&zcrypt_list_lock);
        zcrypt_drop_queue(pref_zc, pref_zq, mod, wgt);
        spin_unlock(&zcrypt_list_lock);

out:
        if (userspace)
                kfree(targets);
        ap_release_apmsg(&ap_msg);
        if (tr) {
                tr->last_rc = rc;
                tr->last_qid = qid;
        }
        trace_s390_zcrypt_rep(xcrb, func_code, rc,
                              AP_QID_CARD(qid), AP_QID_QUEUE(qid),
                              ap_msg.psmid);
        return rc;
}

long zcrypt_send_ep11_cprb(struct ep11_urb *xcrb, u32 xflags)
{
        struct zcrypt_track tr;
        int rc;

        memset(&tr, 0, sizeof(tr));

        do {
                rc = _zcrypt_send_ep11_cprb(xflags, &ap_perms, &tr, xcrb);
        } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);

        /* on ENODEV failure: retry once again after a requested rescan */
        if (rc == -ENODEV && zcrypt_process_rescan())
                do {
                        rc = _zcrypt_send_ep11_cprb(xflags, &ap_perms, &tr, xcrb);
                } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);
        if (rc == -EAGAIN && tr.again_counter >= TRACK_AGAIN_MAX)
                rc = -EIO;
        if (rc)
                pr_debug("rc=%d\n", rc);

        return rc;
}
EXPORT_SYMBOL(zcrypt_send_ep11_cprb);

static long zcrypt_rng(char *buffer)
{
        struct zcrypt_card *zc, *pref_zc;
        struct zcrypt_queue *zq, *pref_zq;
        unsigned int wgt = 0, pref_wgt = 0;
        unsigned int func_code = 0;
        struct ap_message ap_msg;
        unsigned int domain;
        int qid = 0, rc = -ENODEV;
        struct module *mod;

        trace_s390_zcrypt_req(buffer, TP_HWRNGCPRB);

        rc = ap_init_apmsg(&ap_msg, 0);
        if (rc)
                goto out;
        rc = prep_rng_ap_msg(&ap_msg, &func_code, &domain);
        if (rc)
                goto out;

        pref_zc = NULL;
        pref_zq = NULL;
        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                /* Check for usable CCA card */
                if (!zc->online || !zc->card->config || zc->card->chkstop ||
                    !zc->card->hwinfo.cca)
                        continue;
                /* get weight index of the card device  */
                wgt = zc->speed_rating[func_code];
                if (!zcrypt_card_compare(zc, pref_zc, wgt, pref_wgt))
                        continue;
                for_each_zcrypt_queue(zq, zc) {
                        /* check if device is usable and eligible */
                        if (!zq->online || !zq->ops->rng ||
                            !ap_queue_usable(zq->queue))
                                continue;
                        if (!zcrypt_queue_compare(zq, pref_zq, wgt, pref_wgt))
                                continue;
                        pref_zc = zc;
                        pref_zq = zq;
                        pref_wgt = wgt;
                }
        }
        pref_zq = zcrypt_pick_queue(pref_zc, pref_zq, &mod, wgt);
        spin_unlock(&zcrypt_list_lock);

        if (!pref_zq) {
                pr_debug("no matching queue found => ENODEV\n");
                rc = -ENODEV;
                goto out;
        }

        qid = pref_zq->queue->qid;
        rc = pref_zq->ops->rng(pref_zq, buffer, &ap_msg);

        spin_lock(&zcrypt_list_lock);
        zcrypt_drop_queue(pref_zc, pref_zq, mod, wgt);
        spin_unlock(&zcrypt_list_lock);

out:
        ap_release_apmsg(&ap_msg);
        trace_s390_zcrypt_rep(buffer, func_code, rc,
                              AP_QID_CARD(qid), AP_QID_QUEUE(qid),
                              ap_msg.psmid);
        return rc;
}

static void zcrypt_device_status_mask(struct zcrypt_device_status *devstatus)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        struct zcrypt_device_status *stat;
        int card, queue;

        memset(devstatus, 0, MAX_ZDEV_ENTRIES
               * sizeof(struct zcrypt_device_status));

        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        card = AP_QID_CARD(zq->queue->qid);
                        if (card >= MAX_ZDEV_CARDIDS)
                                continue;
                        queue = AP_QID_QUEUE(zq->queue->qid);
                        stat = &devstatus[card * AP_DOMAINS + queue];
                        stat->hwtype = zc->card->ap_dev.device_type;
                        stat->functions = zc->card->hwinfo.fac >> 26;
                        stat->qid = zq->queue->qid;
                        stat->online = zq->online ? 0x01 : 0x00;
                }
        }
        spin_unlock(&zcrypt_list_lock);
}

void zcrypt_device_status_mask_ext(struct zcrypt_device_status_ext *devstatus,
                                   int maxcard, int maxqueue)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        struct zcrypt_device_status_ext *stat;
        int card, queue;

        maxcard = min_t(int, maxcard, MAX_ZDEV_CARDIDS_EXT);
        maxqueue = min_t(int, maxqueue, MAX_ZDEV_DOMAINS_EXT);

        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        card = AP_QID_CARD(zq->queue->qid);
                        queue = AP_QID_QUEUE(zq->queue->qid);
                        if (card >= maxcard || queue >= maxqueue)
                                continue;
                        stat = &devstatus[card * maxqueue + queue];
                        stat->hwtype = zc->card->ap_dev.device_type;
                        stat->functions = zc->card->hwinfo.fac >> 26;
                        stat->qid = zq->queue->qid;
                        stat->online = zq->online ? 0x01 : 0x00;
                }
        }
        spin_unlock(&zcrypt_list_lock);
}
EXPORT_SYMBOL(zcrypt_device_status_mask_ext);

int zcrypt_device_status_ext(int card, int queue,
                             struct zcrypt_device_status_ext *devstat)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;

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

        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        if (card == AP_QID_CARD(zq->queue->qid) &&
                            queue == AP_QID_QUEUE(zq->queue->qid)) {
                                devstat->hwtype = zc->card->ap_dev.device_type;
                                devstat->functions = zc->card->hwinfo.fac >> 26;
                                devstat->qid = zq->queue->qid;
                                devstat->online = zq->online ? 0x01 : 0x00;
                                spin_unlock(&zcrypt_list_lock);
                                return 0;
                        }
                }
        }
        spin_unlock(&zcrypt_list_lock);

        return -ENODEV;
}
EXPORT_SYMBOL(zcrypt_device_status_ext);

static void zcrypt_status_mask(char status[], size_t max_adapters)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        int card;

        memset(status, 0, max_adapters);
        spin_lock(&zcrypt_list_lock);
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        card = AP_QID_CARD(zq->queue->qid);
                        if (AP_QID_QUEUE(zq->queue->qid) != ap_domain_index ||
                            card >= max_adapters)
                                continue;
                        status[card] = zc->online ? zc->user_space_type : 0x0d;
                }
        }
        spin_unlock(&zcrypt_list_lock);
}

static void zcrypt_qdepth_mask(char qdepth[], size_t max_adapters)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        int card;

        memset(qdepth, 0, max_adapters);
        spin_lock(&zcrypt_list_lock);
        local_bh_disable();
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        card = AP_QID_CARD(zq->queue->qid);
                        if (AP_QID_QUEUE(zq->queue->qid) != ap_domain_index ||
                            card >= max_adapters)
                                continue;
                        spin_lock(&zq->queue->lock);
                        qdepth[card] =
                                zq->queue->pendingq_count +
                                zq->queue->requestq_count;
                        spin_unlock(&zq->queue->lock);
                }
        }
        local_bh_enable();
        spin_unlock(&zcrypt_list_lock);
}

static void zcrypt_perdev_reqcnt(u32 reqcnt[], size_t max_adapters)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        int card;
        u64 cnt;

        memset(reqcnt, 0, sizeof(int) * max_adapters);
        spin_lock(&zcrypt_list_lock);
        local_bh_disable();
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        card = AP_QID_CARD(zq->queue->qid);
                        if (AP_QID_QUEUE(zq->queue->qid) != ap_domain_index ||
                            card >= max_adapters)
                                continue;
                        spin_lock(&zq->queue->lock);
                        cnt = zq->queue->total_request_count;
                        spin_unlock(&zq->queue->lock);
                        reqcnt[card] = (cnt < UINT_MAX) ? (u32)cnt : UINT_MAX;
                }
        }
        local_bh_enable();
        spin_unlock(&zcrypt_list_lock);
}

static int zcrypt_pendingq_count(void)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        int pendingq_count;

        pendingq_count = 0;
        spin_lock(&zcrypt_list_lock);
        local_bh_disable();
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        if (AP_QID_QUEUE(zq->queue->qid) != ap_domain_index)
                                continue;
                        spin_lock(&zq->queue->lock);
                        pendingq_count += zq->queue->pendingq_count;
                        spin_unlock(&zq->queue->lock);
                }
        }
        local_bh_enable();
        spin_unlock(&zcrypt_list_lock);
        return pendingq_count;
}

static int zcrypt_requestq_count(void)
{
        struct zcrypt_card *zc;
        struct zcrypt_queue *zq;
        int requestq_count;

        requestq_count = 0;
        spin_lock(&zcrypt_list_lock);
        local_bh_disable();
        for_each_zcrypt_card(zc) {
                for_each_zcrypt_queue(zq, zc) {
                        if (AP_QID_QUEUE(zq->queue->qid) != ap_domain_index)
                                continue;
                        spin_lock(&zq->queue->lock);
                        requestq_count += zq->queue->requestq_count;
                        spin_unlock(&zq->queue->lock);
                }
        }
        local_bh_enable();
        spin_unlock(&zcrypt_list_lock);
        return requestq_count;
}

static int icarsamodexpo_ioctl(struct ap_perms *perms, unsigned long arg)
{
        int rc;
        struct zcrypt_track tr;
        struct ica_rsa_modexpo mex;
        struct ica_rsa_modexpo __user *umex = (void __user *)arg;

        memset(&tr, 0, sizeof(tr));
        if (copy_from_user(&mex, umex, sizeof(mex)))
                return -EFAULT;

        do {
                rc = zcrypt_rsa_modexpo(perms, &tr, &mex);
        } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);

        /* on ENODEV failure: retry once again after a requested rescan */
        if (rc == -ENODEV && zcrypt_process_rescan())
                do {
                        rc = zcrypt_rsa_modexpo(perms, &tr, &mex);
                } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);
        if (rc == -EAGAIN && tr.again_counter >= TRACK_AGAIN_MAX)
                rc = -EIO;
        if (rc) {
                pr_debug("ioctl ICARSAMODEXPO rc=%d\n", rc);
                return rc;
        }
        return put_user(mex.outputdatalength, &umex->outputdatalength);
}

static int icarsacrt_ioctl(struct ap_perms *perms, unsigned long arg)
{
        int rc;
        struct zcrypt_track tr;
        struct ica_rsa_modexpo_crt crt;
        struct ica_rsa_modexpo_crt __user *ucrt = (void __user *)arg;

        memset(&tr, 0, sizeof(tr));
        if (copy_from_user(&crt, ucrt, sizeof(crt)))
                return -EFAULT;

        do {
                rc = zcrypt_rsa_crt(perms, &tr, &crt);
        } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);

        /* on ENODEV failure: retry once again after a requested rescan */
        if (rc == -ENODEV && zcrypt_process_rescan())
                do {
                        rc = zcrypt_rsa_crt(perms, &tr, &crt);
                } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);
        if (rc == -EAGAIN && tr.again_counter >= TRACK_AGAIN_MAX)
                rc = -EIO;
        if (rc) {
                pr_debug("ioctl ICARSACRT rc=%d\n", rc);
                return rc;
        }
        return put_user(crt.outputdatalength, &ucrt->outputdatalength);
}

static int zsecsendcprb_ioctl(struct ap_perms *perms, unsigned long arg)
{
        int rc;
        struct ica_xcRB xcrb;
        struct zcrypt_track tr;
        u32 xflags = ZCRYPT_XFLAG_USERSPACE;
        struct ica_xcRB __user *uxcrb = (void __user *)arg;

        memset(&tr, 0, sizeof(tr));
        if (copy_from_user(&xcrb, uxcrb, sizeof(xcrb)))
                return -EFAULT;

        do {
                rc = _zcrypt_send_cprb(xflags, perms, &tr, &xcrb);
        } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);

        /* on ENODEV failure: retry once again after a requested rescan */
        if (rc == -ENODEV && zcrypt_process_rescan())
                do {
                        rc = _zcrypt_send_cprb(xflags, perms, &tr, &xcrb);
                } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);
        if (rc == -EAGAIN && tr.again_counter >= TRACK_AGAIN_MAX)
                rc = -EIO;
        if (rc)
                pr_debug("ioctl ZSENDCPRB rc=%d status=0x%x\n",
                         rc, xcrb.status);
        if (copy_to_user(uxcrb, &xcrb, sizeof(xcrb)))
                return -EFAULT;
        return rc;
}

static int zsendep11cprb_ioctl(struct ap_perms *perms, unsigned long arg)
{
        int rc;
        struct ep11_urb xcrb;
        struct zcrypt_track tr;
        u32 xflags = ZCRYPT_XFLAG_USERSPACE;
        struct ep11_urb __user *uxcrb = (void __user *)arg;

        memset(&tr, 0, sizeof(tr));
        if (copy_from_user(&xcrb, uxcrb, sizeof(xcrb)))
                return -EFAULT;

        do {
                rc = _zcrypt_send_ep11_cprb(xflags, perms, &tr, &xcrb);
        } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);

        /* on ENODEV failure: retry once again after a requested rescan */
        if (rc == -ENODEV && zcrypt_process_rescan())
                do {
                        rc = _zcrypt_send_ep11_cprb(xflags, perms, &tr, &xcrb);
                } while (rc == -EAGAIN && ++tr.again_counter < TRACK_AGAIN_MAX);
        if (rc == -EAGAIN && tr.again_counter >= TRACK_AGAIN_MAX)
                rc = -EIO;
        if (rc)
                pr_debug("ioctl ZSENDEP11CPRB rc=%d\n", rc);
        if (copy_to_user(uxcrb, &xcrb, sizeof(xcrb)))
                return -EFAULT;
        return rc;
}

static long zcrypt_unlocked_ioctl(struct file *filp, unsigned int cmd,
                                  unsigned long arg)
{
        int rc;
        struct ap_perms *perms =
                (struct ap_perms *)filp->private_data;

        rc = zcrypt_check_ioctl(perms, cmd);
        if (rc)
                return rc;

        switch (cmd) {
        case ICARSAMODEXPO:
                return icarsamodexpo_ioctl(perms, arg);
        case ICARSACRT:
                return icarsacrt_ioctl(perms, arg);
        case ZSECSENDCPRB:
                return zsecsendcprb_ioctl(perms, arg);
        case ZSENDEP11CPRB:
                return zsendep11cprb_ioctl(perms, arg);
        case ZCRYPT_DEVICE_STATUS: {
                struct zcrypt_device_status_ext *device_status;
                size_t total_size = MAX_ZDEV_ENTRIES_EXT
                        * sizeof(struct zcrypt_device_status_ext);

                device_status = kvzalloc_objs(struct zcrypt_device_status_ext,
                                              MAX_ZDEV_ENTRIES_EXT);
                if (!device_status)
                        return -ENOMEM;
                zcrypt_device_status_mask_ext(device_status,
                                              MAX_ZDEV_CARDIDS_EXT,
                                              MAX_ZDEV_DOMAINS_EXT);
                if (copy_to_user((char __user *)arg, device_status,
                                 total_size))
                        rc = -EFAULT;
                kvfree(device_status);
                return rc;
        }
        case ZCRYPT_STATUS_MASK: {
                char status[AP_DEVICES];

                zcrypt_status_mask(status, AP_DEVICES);
                if (copy_to_user((char __user *)arg, status, sizeof(status)))
                        return -EFAULT;
                return 0;
        }
        case ZCRYPT_QDEPTH_MASK: {
                char qdepth[AP_DEVICES];

                zcrypt_qdepth_mask(qdepth, AP_DEVICES);
                if (copy_to_user((char __user *)arg, qdepth, sizeof(qdepth)))
                        return -EFAULT;
                return 0;
        }
        case ZCRYPT_PERDEV_REQCNT: {
                u32 *reqcnt;

                reqcnt = kcalloc(AP_DEVICES, sizeof(u32), GFP_KERNEL);
                if (!reqcnt)
                        return -ENOMEM;
                zcrypt_perdev_reqcnt(reqcnt, AP_DEVICES);
                if (copy_to_user((int __user *)arg, reqcnt,
                                 sizeof(u32) * AP_DEVICES))
                        rc = -EFAULT;
                kfree(reqcnt);
                return rc;
        }
        case Z90STAT_REQUESTQ_COUNT:
                return put_user(zcrypt_requestq_count(), (int __user *)arg);
        case Z90STAT_PENDINGQ_COUNT:
                return put_user(zcrypt_pendingq_count(), (int __user *)arg);
        case Z90STAT_TOTALOPEN_COUNT:
                return put_user(atomic_read(&zcrypt_open_count),
                                (int __user *)arg);
        case Z90STAT_DOMAIN_INDEX:
                return put_user(ap_domain_index, (int __user *)arg);
        /*
         * Deprecated ioctls
         */
        case ZDEVICESTATUS: {
                /* the old ioctl supports only 64 adapters */
                struct zcrypt_device_status *device_status;
                size_t total_size = MAX_ZDEV_ENTRIES
                        * sizeof(struct zcrypt_device_status);

                device_status = kzalloc(total_size, GFP_KERNEL);
                if (!device_status)
                        return -ENOMEM;
                zcrypt_device_status_mask(device_status);
                if (copy_to_user((char __user *)arg, device_status,
                                 total_size))
                        rc = -EFAULT;
                kfree(device_status);
                return rc;
        }
        case Z90STAT_STATUS_MASK: {
                /* the old ioctl supports only 64 adapters */
                char status[MAX_ZDEV_CARDIDS];

                zcrypt_status_mask(status, MAX_ZDEV_CARDIDS);
                if (copy_to_user((char __user *)arg, status, sizeof(status)))
                        return -EFAULT;
                return 0;
        }
        case Z90STAT_QDEPTH_MASK: {
                /* the old ioctl supports only 64 adapters */
                char qdepth[MAX_ZDEV_CARDIDS];

                zcrypt_qdepth_mask(qdepth, MAX_ZDEV_CARDIDS);
                if (copy_to_user((char __user *)arg, qdepth, sizeof(qdepth)))
                        return -EFAULT;
                return 0;
        }
        case Z90STAT_PERDEV_REQCNT: {
                /* the old ioctl supports only 64 adapters */
                u32 reqcnt[MAX_ZDEV_CARDIDS];

                zcrypt_perdev_reqcnt(reqcnt, MAX_ZDEV_CARDIDS);
                if (copy_to_user((int __user *)arg, reqcnt, sizeof(reqcnt)))
                        return -EFAULT;
                return 0;
        }
        /* unknown ioctl number */
        default:
                pr_debug("unknown ioctl 0x%08x\n", cmd);
                return -ENOIOCTLCMD;
        }
}

/*
 * Misc device file operations.
 */
static const struct file_operations zcrypt_fops = {
        .owner          = THIS_MODULE,
        .read           = zcrypt_read,
        .write          = zcrypt_write,
        .unlocked_ioctl = zcrypt_unlocked_ioctl,
        .open           = zcrypt_open,
        .release        = zcrypt_release,
};

/*
 * Misc device.
 */
static struct miscdevice zcrypt_misc_device = {
        .minor      = MISC_DYNAMIC_MINOR,
        .name       = "z90crypt",
        .fops       = &zcrypt_fops,
};

static int zcrypt_rng_device_count;
static u32 *zcrypt_rng_buffer;
static int zcrypt_rng_buffer_index;
static DEFINE_MUTEX(zcrypt_rng_mutex);

static int zcrypt_rng_data_read(struct hwrng *rng, u32 *data)
{
        int rc;

        /*
         * We don't need locking here because the RNG API guarantees serialized
         * read method calls.
         */
        if (zcrypt_rng_buffer_index == 0) {
                rc = zcrypt_rng((char *)zcrypt_rng_buffer);
                /* on ENODEV failure: retry once again after an AP bus rescan */
                if (rc == -ENODEV && zcrypt_process_rescan())
                        rc = zcrypt_rng((char *)zcrypt_rng_buffer);
                if (rc < 0)
                        return -EIO;
                zcrypt_rng_buffer_index = rc / sizeof(*data);
        }
        *data = zcrypt_rng_buffer[--zcrypt_rng_buffer_index];
        return sizeof(*data);
}

static struct hwrng zcrypt_rng_dev = {
        .name           = "zcrypt",
        .data_read      = zcrypt_rng_data_read,
        .quality        = 990,
};

int zcrypt_rng_device_add(void)
{
        int rc = 0;

        mutex_lock(&zcrypt_rng_mutex);
        if (zcrypt_rng_device_count == 0) {
                zcrypt_rng_buffer = (u32 *)get_zeroed_page(GFP_KERNEL);
                if (!zcrypt_rng_buffer) {
                        rc = -ENOMEM;
                        goto out;
                }
                zcrypt_rng_buffer_index = 0;
                rc = hwrng_register(&zcrypt_rng_dev);
                if (rc)
                        goto out_free;
                zcrypt_rng_device_count = 1;
        } else {
                zcrypt_rng_device_count++;
        }
        mutex_unlock(&zcrypt_rng_mutex);
        return 0;

out_free:
        free_page((unsigned long)zcrypt_rng_buffer);
out:
        mutex_unlock(&zcrypt_rng_mutex);
        return rc;
}

void zcrypt_rng_device_remove(void)
{
        mutex_lock(&zcrypt_rng_mutex);
        zcrypt_rng_device_count--;
        if (zcrypt_rng_device_count == 0) {
                hwrng_unregister(&zcrypt_rng_dev);
                free_page((unsigned long)zcrypt_rng_buffer);
        }
        mutex_unlock(&zcrypt_rng_mutex);
}

/*
 * Wait until the zcrypt api is operational.
 * The AP bus scan and the binding of ap devices to device drivers is
 * an asynchronous job. This function waits until these initial jobs
 * are done and so the zcrypt api should be ready to serve crypto
 * requests - if there are resources available. The function uses an
 * internal timeout of 30s. The very first caller will either wait for
 * ap bus bindings complete or the timeout happens. This state will be
 * remembered for further callers which will only be blocked until a
 * decision is made (timeout or bindings complete).
 * On timeout -ETIME is returned, on success the return value is 0.
 */
int zcrypt_wait_api_operational(void)
{
        static DEFINE_MUTEX(zcrypt_wait_api_lock);
        static int zcrypt_wait_api_state;
        int rc;

        rc = mutex_lock_interruptible(&zcrypt_wait_api_lock);
        if (rc)
                return rc;

        switch (zcrypt_wait_api_state) {
        case 0:
                /* initial state, invoke wait for the ap bus complete */
                rc = ap_wait_apqn_bindings_complete(
                        msecs_to_jiffies(ZCRYPT_WAIT_BINDINGS_COMPLETE_MS));
                switch (rc) {
                case 0:
                        /* ap bus bindings are complete */
                        zcrypt_wait_api_state = 1;
                        break;
                case -EINTR:
                        /* interrupted, go back to caller */
                        break;
                case -ETIME:
                        /* timeout */
                        ZCRYPT_DBF_WARN("%s ap_wait_init_apqn_bindings_complete()=ETIME\n",
                                        __func__);
                        zcrypt_wait_api_state = -ETIME;
                        break;
                default:
                        /* other failure */
                        pr_debug("ap_wait_init_apqn_bindings_complete()=%d\n", rc);
                        break;
                }
                break;
        case 1:
                /* a previous caller already found ap bus bindings complete */
                rc = 0;
                break;
        default:
                /* a previous caller had timeout or other failure */
                rc = zcrypt_wait_api_state;
                break;
        }

        mutex_unlock(&zcrypt_wait_api_lock);

        return rc;
}
EXPORT_SYMBOL(zcrypt_wait_api_operational);

int __init zcrypt_debug_init(void)
{
        zcrypt_dbf_info = debug_register("zcrypt", 2, 1,
                                         ZCRYPT_DBF_MAX_SPRINTF_ARGS * sizeof(long));
        debug_register_view(zcrypt_dbf_info, &debug_sprintf_view);
        debug_set_level(zcrypt_dbf_info, DBF_ERR);

        return 0;
}

void zcrypt_debug_exit(void)
{
        debug_unregister(zcrypt_dbf_info);
}

static int __init zcdn_init(void)
{
        int rc;

        /* create a new class 'zcrypt' */
        rc = class_register(&zcrypt_class);
        if (rc)
                goto out_class_register_failed;

        /* alloc device minor range */
        rc = alloc_chrdev_region(&zcrypt_devt,
                                 0, ZCRYPT_MAX_MINOR_NODES,
                                 ZCRYPT_NAME);
        if (rc)
                goto out_alloc_chrdev_failed;

        cdev_init(&zcrypt_cdev, &zcrypt_fops);
        zcrypt_cdev.owner = THIS_MODULE;
        rc = cdev_add(&zcrypt_cdev, zcrypt_devt, ZCRYPT_MAX_MINOR_NODES);
        if (rc)
                goto out_cdev_add_failed;

        /* need some class specific sysfs attributes */
        rc = class_create_file(&zcrypt_class, &class_attr_zcdn_create);
        if (rc)
                goto out_class_create_file_1_failed;
        rc = class_create_file(&zcrypt_class, &class_attr_zcdn_destroy);
        if (rc)
                goto out_class_create_file_2_failed;

        return 0;

out_class_create_file_2_failed:
        class_remove_file(&zcrypt_class, &class_attr_zcdn_create);
out_class_create_file_1_failed:
        cdev_del(&zcrypt_cdev);
out_cdev_add_failed:
        unregister_chrdev_region(zcrypt_devt, ZCRYPT_MAX_MINOR_NODES);
out_alloc_chrdev_failed:
        class_unregister(&zcrypt_class);
out_class_register_failed:
        return rc;
}

static void zcdn_exit(void)
{
        class_remove_file(&zcrypt_class, &class_attr_zcdn_create);
        class_remove_file(&zcrypt_class, &class_attr_zcdn_destroy);
        zcdn_destroy_all();
        cdev_del(&zcrypt_cdev);
        unregister_chrdev_region(zcrypt_devt, ZCRYPT_MAX_MINOR_NODES);
        class_unregister(&zcrypt_class);
}

/*
 * zcrypt_api_init(): Module initialization.
 *
 * The module initialization code.
 */
int __init zcrypt_api_init(void)
{
        int rc;

        /* make sure the mempool threshold is >= 1 */
        if (zcrypt_mempool_threshold < 1) {
                rc = -EINVAL;
                goto out;
        }

        rc = zcrypt_debug_init();
        if (rc)
                goto out;

        rc = zcdn_init();
        if (rc)
                goto out_zcdn_init_failed;

        rc = zcrypt_ccamisc_init();
        if (rc)
                goto out_ccamisc_init_failed;

        rc = zcrypt_ep11misc_init();
        if (rc)
                goto out_ep11misc_init_failed;

        /* Register the request sprayer. */
        rc = misc_register(&zcrypt_misc_device);
        if (rc < 0)
                goto out_misc_register_failed;

        zcrypt_msgtype6_init();
        zcrypt_msgtype50_init();

        return 0;

out_misc_register_failed:
        zcrypt_ep11misc_exit();
out_ep11misc_init_failed:
        zcrypt_ccamisc_exit();
out_ccamisc_init_failed:
        zcdn_exit();
out_zcdn_init_failed:
        zcrypt_debug_exit();
out:
        return rc;
}

/*
 * zcrypt_api_exit(): Module termination.
 *
 * The module termination code.
 */
void __exit zcrypt_api_exit(void)
{
        zcdn_exit();
        misc_deregister(&zcrypt_misc_device);
        zcrypt_msgtype6_exit();
        zcrypt_msgtype50_exit();
        zcrypt_ccamisc_exit();
        zcrypt_ep11misc_exit();
        zcrypt_debug_exit();
}

module_init(zcrypt_api_init);
module_exit(zcrypt_api_exit);