[deliverable/linux.git] / drivers / s390 / crypto / zcrypt_cex2a.c

/*
 *  linux/drivers/s390/crypto/zcrypt_cex2a.c
 *
 *  zcrypt 2.1.0
 *
 *  Copyright (C)  2001, 2006 IBM Corporation
 *  Author(s): Robert Burroughs
 *	       Eric Rossman (edrossma@us.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>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <linux/module.h>
#include <linux/init.h>
#include <linux/err.h>
#include <asm/atomic.h>
#include <asm/uaccess.h>

#include "ap_bus.h"
#include "zcrypt_api.h"
#include "zcrypt_error.h"
#include "zcrypt_cex2a.h"

#define CEX2A_MIN_MOD_SIZE	  1	/*    8 bits	*/
#define CEX2A_MAX_MOD_SIZE	256	/* 2048 bits	*/
#define CEX3A_MIN_MOD_SIZE	CEX2A_MIN_MOD_SIZE
#define CEX3A_MAX_MOD_SIZE	CEX2A_MAX_MOD_SIZE

#define CEX2A_SPEED_RATING	970
#define CEX3A_SPEED_RATING	900 /* Fixme: Needs finetuning */

#define CEX2A_MAX_MESSAGE_SIZE	0x390	/* sizeof(struct type50_crb2_msg)    */
#define CEX2A_MAX_RESPONSE_SIZE 0x110	/* max outputdatalength + type80_hdr */

#define CEX3A_MAX_MESSAGE_SIZE	CEX2A_MAX_MESSAGE_SIZE
#define CEX3A_MAX_RESPONSE_SIZE	CEX2A_MAX_RESPONSE_SIZE

#define CEX2A_CLEANUP_TIME	(15*HZ)
#define CEX3A_CLEANUP_TIME	CEX2A_CLEANUP_TIME

static struct ap_device_id zcrypt_cex2a_ids[] = {
	{ AP_DEVICE(AP_DEVICE_TYPE_CEX2A) },
	{ AP_DEVICE(AP_DEVICE_TYPE_CEX3A) },
	{ /* end of list */ },
};

#ifndef CONFIG_ZCRYPT_MONOLITHIC
MODULE_DEVICE_TABLE(ap, zcrypt_cex2a_ids);
MODULE_AUTHOR("IBM Corporation");
MODULE_DESCRIPTION("CEX2A Cryptographic Coprocessor device driver, "
		   "Copyright 2001, 2006 IBM Corporation");
MODULE_LICENSE("GPL");
#endif

static int zcrypt_cex2a_probe(struct ap_device *ap_dev);
static void zcrypt_cex2a_remove(struct ap_device *ap_dev);
static void zcrypt_cex2a_receive(struct ap_device *, struct ap_message *,
				 struct ap_message *);

static struct ap_driver zcrypt_cex2a_driver = {
	.probe = zcrypt_cex2a_probe,
	.remove = zcrypt_cex2a_remove,
	.receive = zcrypt_cex2a_receive,
	.ids = zcrypt_cex2a_ids,
	.request_timeout = CEX2A_CLEANUP_TIME,
};

/**
 * Convert a ICAMEX message to a type50 MEX message.
 *
 * @zdev: crypto device pointer
 * @zreq: crypto request pointer
 * @mex: pointer to user input data
 *
 * Returns 0 on success or -EFAULT.
 */
static int ICAMEX_msg_to_type50MEX_msg(struct zcrypt_device *zdev,
				       struct ap_message *ap_msg,
				       struct ica_rsa_modexpo *mex)
{
	unsigned char *mod, *exp, *inp;
	int mod_len;

	mod_len = mex->inputdatalength;

	if (mod_len <= 128) {
		struct type50_meb1_msg *meb1 = ap_msg->message;
		memset(meb1, 0, sizeof(*meb1));
		ap_msg->length = sizeof(*meb1);
		meb1->header.msg_type_code = TYPE50_TYPE_CODE;
		meb1->header.msg_len = sizeof(*meb1);
		meb1->keyblock_type = TYPE50_MEB1_FMT;
		mod = meb1->modulus + sizeof(meb1->modulus) - mod_len;
		exp = meb1->exponent + sizeof(meb1->exponent) - mod_len;
		inp = meb1->message + sizeof(meb1->message) - mod_len;
	} else {
		struct type50_meb2_msg *meb2 = ap_msg->message;
		memset(meb2, 0, sizeof(*meb2));
		ap_msg->length = sizeof(*meb2);
		meb2->header.msg_type_code = TYPE50_TYPE_CODE;
		meb2->header.msg_len = sizeof(*meb2);
		meb2->keyblock_type = TYPE50_MEB2_FMT;
		mod = meb2->modulus + sizeof(meb2->modulus) - mod_len;
		exp = meb2->exponent + sizeof(meb2->exponent) - mod_len;
		inp = meb2->message + sizeof(meb2->message) - mod_len;
	}

	if (copy_from_user(mod, mex->n_modulus, mod_len) ||
	    copy_from_user(exp, mex->b_key, mod_len) ||
	    copy_from_user(inp, mex->inputdata, mod_len))
		return -EFAULT;
	return 0;
}

/**
 * Convert a ICACRT message to a type50 CRT message.
 *
 * @zdev: crypto device pointer
 * @zreq: crypto request pointer
 * @crt: pointer to user input data
 *
 * Returns 0 on success or -EFAULT.
 */
static int ICACRT_msg_to_type50CRT_msg(struct zcrypt_device *zdev,
				       struct ap_message *ap_msg,
				       struct ica_rsa_modexpo_crt *crt)
{
	int mod_len, short_len, long_len, long_offset;
	unsigned char *p, *q, *dp, *dq, *u, *inp;

	mod_len = crt->inputdatalength;
	short_len = mod_len / 2;
	long_len = mod_len / 2 + 8;

	/*
	 * CEX2A cannot handle p, dp, or U > 128 bytes.
	 * If we have one of these, we need to do extra checking.
	 */
	if (long_len > 128) {
		/*
		 * zcrypt_rsa_crt already checked for the leading
		 * zeroes of np_prime, bp_key and u_mult_inc.
		 */
		long_offset = long_len - 128;
		long_len = 128;
	} else
		long_offset = 0;

	/*
	 * Instead of doing extra work for p, dp, U > 64 bytes, we'll just use
	 * the larger message structure.
	 */
	if (long_len <= 64) {
		struct type50_crb1_msg *crb1 = ap_msg->message;
		memset(crb1, 0, sizeof(*crb1));
		ap_msg->length = sizeof(*crb1);
		crb1->header.msg_type_code = TYPE50_TYPE_CODE;
		crb1->header.msg_len = sizeof(*crb1);
		crb1->keyblock_type = TYPE50_CRB1_FMT;
		p = crb1->p + sizeof(crb1->p) - long_len;
		q = crb1->q + sizeof(crb1->q) - short_len;
		dp = crb1->dp + sizeof(crb1->dp) - long_len;
		dq = crb1->dq + sizeof(crb1->dq) - short_len;
		u = crb1->u + sizeof(crb1->u) - long_len;
		inp = crb1->message + sizeof(crb1->message) - mod_len;
	} else {
		struct type50_crb2_msg *crb2 = ap_msg->message;
		memset(crb2, 0, sizeof(*crb2));
		ap_msg->length = sizeof(*crb2);
		crb2->header.msg_type_code = TYPE50_TYPE_CODE;
		crb2->header.msg_len = sizeof(*crb2);
		crb2->keyblock_type = TYPE50_CRB2_FMT;
		p = crb2->p + sizeof(crb2->p) - long_len;
		q = crb2->q + sizeof(crb2->q) - short_len;
		dp = crb2->dp + sizeof(crb2->dp) - long_len;
		dq = crb2->dq + sizeof(crb2->dq) - short_len;
		u = crb2->u + sizeof(crb2->u) - long_len;
		inp = crb2->message + sizeof(crb2->message) - mod_len;
	}

	if (copy_from_user(p, crt->np_prime + long_offset, long_len) ||
	    copy_from_user(q, crt->nq_prime, short_len) ||
	    copy_from_user(dp, crt->bp_key + long_offset, long_len) ||
	    copy_from_user(dq, crt->bq_key, short_len) ||
	    copy_from_user(u, crt->u_mult_inv + long_offset, long_len) ||
	    copy_from_user(inp, crt->inputdata, mod_len))
		return -EFAULT;


	return 0;
}

/**
 * Copy results from a type 80 reply message back to user space.
 *
 * @zdev: crypto device pointer
 * @reply: reply AP message.
 * @data: pointer to user output data
 * @length: size of user output data
 *
 * Returns 0 on success or -EFAULT.
 */
static int convert_type80(struct zcrypt_device *zdev,
			  struct ap_message *reply,
			  char __user *outputdata,
			  unsigned int outputdatalength)
{
	struct type80_hdr *t80h = reply->message;
	unsigned char *data;

	if (t80h->len < sizeof(*t80h) + outputdatalength) {
		/* The result is too short, the CEX2A card may not do that.. */
		zdev->online = 0;
		return -EAGAIN;	/* repeat the request on a different device. */
	}
	BUG_ON(t80h->len > CEX2A_MAX_RESPONSE_SIZE);
	data = reply->message + t80h->len - outputdatalength;
	if (copy_to_user(outputdata, data, outputdatalength))
		return -EFAULT;
	return 0;
}

static int convert_response(struct zcrypt_device *zdev,
			    struct ap_message *reply,
			    char __user *outputdata,
			    unsigned int outputdatalength)
{
	/* Response type byte is the second byte in the response. */
	switch (((unsigned char *) reply->message)[1]) {
	case TYPE82_RSP_CODE:
	case TYPE88_RSP_CODE:
		return convert_error(zdev, reply);
	case TYPE80_RSP_CODE:
		return convert_type80(zdev, reply,
				      outputdata, outputdatalength);
	default: /* Unknown response type, this should NEVER EVER happen */
		zdev->online = 0;
		return -EAGAIN;	/* repeat the request on a different device. */
	}
}

/**
 * This function is called from the AP bus code after a crypto request
 * "msg" has finished with the reply message "reply".
 * It is called from tasklet context.
 * @ap_dev: pointer to the AP device
 * @msg: pointer to the AP message
 * @reply: pointer to the AP reply message
 */
static void zcrypt_cex2a_receive(struct ap_device *ap_dev,
				 struct ap_message *msg,
				 struct ap_message *reply)
{
	static struct error_hdr error_reply = {
		.type = TYPE82_RSP_CODE,
		.reply_code = REP82_ERROR_MACHINE_FAILURE,
	};
	struct type80_hdr *t80h;
	int length;

	/* Copy the reply message to the request message buffer. */
	if (IS_ERR(reply)) {
		memcpy(msg->message, &error_reply, sizeof(error_reply));
		goto out;
	}
	t80h = reply->message;
	if (t80h->type == TYPE80_RSP_CODE) {
		length = min(CEX2A_MAX_RESPONSE_SIZE, (int) t80h->len);
		memcpy(msg->message, reply->message, length);
	} else
		memcpy(msg->message, reply->message, sizeof error_reply);
out:
	complete((struct completion *) msg->private);
}

static atomic_t zcrypt_step = ATOMIC_INIT(0);

/**
 * The request distributor calls this function if it picked the CEX2A
 * device to handle a modexpo request.
 * @zdev: pointer to zcrypt_device structure that identifies the
 *	  CEX2A device to the request distributor
 * @mex: pointer to the modexpo request buffer
 */
static long zcrypt_cex2a_modexpo(struct zcrypt_device *zdev,
				 struct ica_rsa_modexpo *mex)
{
	struct ap_message ap_msg;
	struct completion work;
	int rc;

	ap_init_message(&ap_msg);
	ap_msg.message = kmalloc(CEX2A_MAX_MESSAGE_SIZE, GFP_KERNEL);
	if (!ap_msg.message)
		return -ENOMEM;
	ap_msg.psmid = (((unsigned long long) current->pid) << 32) +
				atomic_inc_return(&zcrypt_step);
	ap_msg.private = &work;
	rc = ICAMEX_msg_to_type50MEX_msg(zdev, &ap_msg, mex);
	if (rc)
		goto out_free;
	init_completion(&work);
	ap_queue_message(zdev->ap_dev, &ap_msg);
	rc = wait_for_completion_interruptible(&work);
	if (rc == 0)
		rc = convert_response(zdev, &ap_msg, mex->outputdata,
				      mex->outputdatalength);
	else
		/* Signal pending. */
		ap_cancel_message(zdev->ap_dev, &ap_msg);
out_free:
	kfree(ap_msg.message);
	return rc;
}

/**
 * The request distributor calls this function if it picked the CEX2A
 * device to handle a modexpo_crt request.
 * @zdev: pointer to zcrypt_device structure that identifies the
 *	  CEX2A device to the request distributor
 * @crt: pointer to the modexpoc_crt request buffer
 */
static long zcrypt_cex2a_modexpo_crt(struct zcrypt_device *zdev,
				     struct ica_rsa_modexpo_crt *crt)
{
	struct ap_message ap_msg;
	struct completion work;
	int rc;

	ap_init_message(&ap_msg);
	ap_msg.message = kmalloc(CEX2A_MAX_MESSAGE_SIZE, GFP_KERNEL);
	if (!ap_msg.message)
		return -ENOMEM;
	ap_msg.psmid = (((unsigned long long) current->pid) << 32) +
				atomic_inc_return(&zcrypt_step);
	ap_msg.private = &work;
	rc = ICACRT_msg_to_type50CRT_msg(zdev, &ap_msg, crt);
	if (rc)
		goto out_free;
	init_completion(&work);
	ap_queue_message(zdev->ap_dev, &ap_msg);
	rc = wait_for_completion_interruptible(&work);
	if (rc == 0)
		rc = convert_response(zdev, &ap_msg, crt->outputdata,
				      crt->outputdatalength);
	else
		/* Signal pending. */
		ap_cancel_message(zdev->ap_dev, &ap_msg);
out_free:
	kfree(ap_msg.message);
	return rc;
}

/**
 * The crypto operations for a CEX2A card.
 */
static struct zcrypt_ops zcrypt_cex2a_ops = {
	.rsa_modexpo = zcrypt_cex2a_modexpo,
	.rsa_modexpo_crt = zcrypt_cex2a_modexpo_crt,
};

/**
 * Probe function for CEX2A cards. It always accepts the AP device
 * since the bus_match already checked the hardware type.
 * @ap_dev: pointer to the AP device.
 */
static int zcrypt_cex2a_probe(struct ap_device *ap_dev)
{
	struct zcrypt_device *zdev = NULL;
	int rc = 0;

	switch (ap_dev->device_type) {
	case AP_DEVICE_TYPE_CEX2A:
		zdev = zcrypt_device_alloc(CEX2A_MAX_RESPONSE_SIZE);
		if (!zdev)
			return -ENOMEM;
		zdev->user_space_type = ZCRYPT_CEX2A;
		zdev->type_string = "CEX2A";
		zdev->min_mod_size = CEX2A_MIN_MOD_SIZE;
		zdev->max_mod_size = CEX2A_MAX_MOD_SIZE;
		zdev->short_crt = 1;
		zdev->speed_rating = CEX2A_SPEED_RATING;
		break;
	case AP_DEVICE_TYPE_CEX3A:
		zdev = zcrypt_device_alloc(CEX3A_MAX_RESPONSE_SIZE);
		if (!zdev)
			return -ENOMEM;
		zdev->user_space_type = ZCRYPT_CEX3A;
		zdev->type_string = "CEX3A";
		zdev->min_mod_size = CEX3A_MIN_MOD_SIZE;
		zdev->max_mod_size = CEX3A_MAX_MOD_SIZE;
		zdev->short_crt = 1;
		zdev->speed_rating = CEX3A_SPEED_RATING;
		break;
	}
	if (zdev != NULL) {
		zdev->ap_dev = ap_dev;
		zdev->ops = &zcrypt_cex2a_ops;
		zdev->online = 1;
		ap_dev->reply = &zdev->reply;
		ap_dev->private = zdev;
		rc = zcrypt_device_register(zdev);
	}
	if (rc) {
		ap_dev->private = NULL;
		zcrypt_device_free(zdev);
	}
	return rc;
}

/**
 * This is called to remove the extended CEX2A driver information
 * if an AP device is removed.
 */
static void zcrypt_cex2a_remove(struct ap_device *ap_dev)
{
	struct zcrypt_device *zdev = ap_dev->private;

	zcrypt_device_unregister(zdev);
}

int __init zcrypt_cex2a_init(void)
{
	return ap_driver_register(&zcrypt_cex2a_driver, THIS_MODULE, "cex2a");
}

void __exit zcrypt_cex2a_exit(void)
{
	ap_driver_unregister(&zcrypt_cex2a_driver);
}

#ifndef CONFIG_ZCRYPT_MONOLITHIC
module_init(zcrypt_cex2a_init);
module_exit(zcrypt_cex2a_exit);
#endif
Commit	Line	Data
963ed931 MS	1	/*
	2	* linux/drivers/s390/crypto/zcrypt_cex2a.c
	3	*
5432114b	4	* zcrypt 2.1.0
963ed931 MS	5	*
	6	* Copyright (C) 2001, 2006 IBM Corporation
	7	* Author(s): Robert Burroughs
	8	* Eric Rossman (edrossma@us.ibm.com)
	9	*
	10	* Hotplug & misc device support: Jochen Roehrig (roehrig@de.ibm.com)
	11	* Major cleanup & driver split: Martin Schwidefsky <schwidefsky@de.ibm.com>
	12	* Ralph Wuerthner <rwuerthn@de.ibm.com>
	13	*
	14	* This program is free software; you can redistribute it and/or modify
	15	* it under the terms of the GNU General Public License as published by
	16	* the Free Software Foundation; either version 2, or (at your option)
	17	* any later version.
	18	*
	19	* This program is distributed in the hope that it will be useful,
	20	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	21	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	22	* GNU General Public License for more details.
	23	*
	24	* You should have received a copy of the GNU General Public License
	25	* along with this program; if not, write to the Free Software
	26	* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
	27	*/
	28
	29	#include <linux/module.h>
	30	#include <linux/init.h>
	31	#include <linux/err.h>
	32	#include <asm/atomic.h>
	33	#include <asm/uaccess.h>
	34
	35	#include "ap_bus.h"
	36	#include "zcrypt_api.h"
	37	#include "zcrypt_error.h"
	38	#include "zcrypt_cex2a.h"
	39
	40	#define CEX2A_MIN_MOD_SIZE 1 /* 8 bits */
	41	#define CEX2A_MAX_MOD_SIZE 256 /* 2048 bits */
8e89b6be FB	42	#define CEX3A_MIN_MOD_SIZE CEX2A_MIN_MOD_SIZE
8e89b6be FB	43	#define CEX3A_MAX_MOD_SIZE CEX2A_MAX_MOD_SIZE
963ed931 MS	44
963ed931 MS	45	#define CEX2A_SPEED_RATING 970
7a6f5cd0	46	#define CEX3A_SPEED_RATING 900 /* Fixme: Needs finetuning */
963ed931 MS	47
	48	#define CEX2A_MAX_MESSAGE_SIZE 0x390 /* sizeof(struct type50_crb2_msg) */
	49	#define CEX2A_MAX_RESPONSE_SIZE 0x110 /* max outputdatalength + type80_hdr */
	50
8e89b6be FB	51	#define CEX3A_MAX_MESSAGE_SIZE CEX2A_MAX_MESSAGE_SIZE
	52	#define CEX3A_MAX_RESPONSE_SIZE CEX2A_MAX_RESPONSE_SIZE
	53
963ed931	54	#define CEX2A_CLEANUP_TIME (15*HZ)
8e89b6be	55	#define CEX3A_CLEANUP_TIME CEX2A_CLEANUP_TIME
963ed931 MS	56
	57	static struct ap_device_id zcrypt_cex2a_ids[] = {
	58	{ AP_DEVICE(AP_DEVICE_TYPE_CEX2A) },
ffda4f71	59	{ AP_DEVICE(AP_DEVICE_TYPE_CEX3A) },
963ed931 MS	60	{ /* end of list */ },
	61	};
	62
	63	#ifndef CONFIG_ZCRYPT_MONOLITHIC
	64	MODULE_DEVICE_TABLE(ap, zcrypt_cex2a_ids);
	65	MODULE_AUTHOR("IBM Corporation");
	66	MODULE_DESCRIPTION("CEX2A Cryptographic Coprocessor device driver, "
	67	"Copyright 2001, 2006 IBM Corporation");
	68	MODULE_LICENSE("GPL");
	69	#endif
	70
	71	static int zcrypt_cex2a_probe(struct ap_device *ap_dev);
	72	static void zcrypt_cex2a_remove(struct ap_device *ap_dev);
	73	static void zcrypt_cex2a_receive(struct ap_device , struct ap_message ,
	74	struct ap_message *);
	75
	76	static struct ap_driver zcrypt_cex2a_driver = {
	77	.probe = zcrypt_cex2a_probe,
	78	.remove = zcrypt_cex2a_remove,
	79	.receive = zcrypt_cex2a_receive,
	80	.ids = zcrypt_cex2a_ids,
af512ed0	81	.request_timeout = CEX2A_CLEANUP_TIME,
963ed931 MS	82	};
	83
	84	/**
	85	* Convert a ICAMEX message to a type50 MEX message.
	86	*
	87	* @zdev: crypto device pointer
	88	* @zreq: crypto request pointer
	89	* @mex: pointer to user input data
	90	*
	91	* Returns 0 on success or -EFAULT.
	92	*/
	93	static int ICAMEX_msg_to_type50MEX_msg(struct zcrypt_device *zdev,
	94	struct ap_message *ap_msg,
	95	struct ica_rsa_modexpo *mex)
	96	{
	97	unsigned char mod, exp, *inp;
	98	int mod_len;
	99
	100	mod_len = mex->inputdatalength;
	101
	102	if (mod_len <= 128) {
	103	struct type50_meb1_msg *meb1 = ap_msg->message;
	104	memset(meb1, 0, sizeof(*meb1));
	105	ap_msg->length = sizeof(*meb1);
	106	meb1->header.msg_type_code = TYPE50_TYPE_CODE;
	107	meb1->header.msg_len = sizeof(*meb1);
	108	meb1->keyblock_type = TYPE50_MEB1_FMT;
	109	mod = meb1->modulus + sizeof(meb1->modulus) - mod_len;
	110	exp = meb1->exponent + sizeof(meb1->exponent) - mod_len;
	111	inp = meb1->message + sizeof(meb1->message) - mod_len;
	112	} else {
	113	struct type50_meb2_msg *meb2 = ap_msg->message;
	114	memset(meb2, 0, sizeof(*meb2));
	115	ap_msg->length = sizeof(*meb2);
	116	meb2->header.msg_type_code = TYPE50_TYPE_CODE;
	117	meb2->header.msg_len = sizeof(*meb2);
	118	meb2->keyblock_type = TYPE50_MEB2_FMT;
	119	mod = meb2->modulus + sizeof(meb2->modulus) - mod_len;
	120	exp = meb2->exponent + sizeof(meb2->exponent) - mod_len;
	121	inp = meb2->message + sizeof(meb2->message) - mod_len;
	122	}
	123
	124	if (copy_from_user(mod, mex->n_modulus, mod_len) \|\|
	125	copy_from_user(exp, mex->b_key, mod_len) \|\|
	126	copy_from_user(inp, mex->inputdata, mod_len))
	127	return -EFAULT;
	128	return 0;
	129	}
	130
	131	/**
	132	* Convert a ICACRT message to a type50 CRT message.
	133	*
	134	* @zdev: crypto device pointer
	135	* @zreq: crypto request pointer
	136	* @crt: pointer to user input data
	137	*
	138	* Returns 0 on success or -EFAULT.
	139	*/
	140	static int ICACRT_msg_to_type50CRT_msg(struct zcrypt_device *zdev,
	141	struct ap_message *ap_msg,
	142	struct ica_rsa_modexpo_crt *crt)
	143	{
	144	int mod_len, short_len, long_len, long_offset;
	145	unsigned char p, q, dp, dq, u, inp;
146
147	mod_len = crt->inputdatalength;
148	short_len = mod_len / 2;
149	long_len = mod_len / 2 + 8;
150
151	/*
152	* CEX2A cannot handle p, dp, or U > 128 bytes.
153	* If we have one of these, we need to do extra checking.
154	*/
155	if (long_len > 128) {
156	/*
157	* zcrypt_rsa_crt already checked for the leading
158	* zeroes of np_prime, bp_key and u_mult_inc.
159	*/
160	long_offset = long_len - 128;
161	long_len = 128;
162	} else
163	long_offset = 0;
164
165	/*
166	* Instead of doing extra work for p, dp, U > 64 bytes, we'll just use
167	* the larger message structure.
168	*/
169	if (long_len <= 64) {
170	struct type50_crb1_msg *crb1 = ap_msg->message;
171	memset(crb1, 0, sizeof(*crb1));
172	ap_msg->length = sizeof(*crb1);
173	crb1->header.msg_type_code = TYPE50_TYPE_CODE;
174	crb1->header.msg_len = sizeof(*crb1);
175	crb1->keyblock_type = TYPE50_CRB1_FMT;
176	p = crb1->p + sizeof(crb1->p) - long_len;
177	q = crb1->q + sizeof(crb1->q) - short_len;
178	dp = crb1->dp + sizeof(crb1->dp) - long_len;
179	dq = crb1->dq + sizeof(crb1->dq) - short_len;
180	u = crb1->u + sizeof(crb1->u) - long_len;
181	inp = crb1->message + sizeof(crb1->message) - mod_len;
182	} else {
183	struct type50_crb2_msg *crb2 = ap_msg->message;
184	memset(crb2, 0, sizeof(*crb2));
185	ap_msg->length = sizeof(*crb2);
186	crb2->header.msg_type_code = TYPE50_TYPE_CODE;
187	crb2->header.msg_len = sizeof(*crb2);
188	crb2->keyblock_type = TYPE50_CRB2_FMT;
189	p = crb2->p + sizeof(crb2->p) - long_len;
190	q = crb2->q + sizeof(crb2->q) - short_len;
191	dp = crb2->dp + sizeof(crb2->dp) - long_len;
192	dq = crb2->dq + sizeof(crb2->dq) - short_len;
193	u = crb2->u + sizeof(crb2->u) - long_len;
194	inp = crb2->message + sizeof(crb2->message) - mod_len;
195	}
196
197	if (copy_from_user(p, crt->np_prime + long_offset, long_len) \|\|
198	copy_from_user(q, crt->nq_prime, short_len) \|\|
199	copy_from_user(dp, crt->bp_key + long_offset, long_len) \|\|
200	copy_from_user(dq, crt->bq_key, short_len) \|\|
201	copy_from_user(u, crt->u_mult_inv + long_offset, long_len) \|\|
202	copy_from_user(inp, crt->inputdata, mod_len))
203	return -EFAULT;
204
205
206	return 0;
207	}
208
209	/**
210	* Copy results from a type 80 reply message back to user space.
211	*
212	* @zdev: crypto device pointer
213	* @reply: reply AP message.
214	* @data: pointer to user output data
215	* @length: size of user output data
216	*
217	* Returns 0 on success or -EFAULT.
218	*/
219	static int convert_type80(struct zcrypt_device *zdev,
220	struct ap_message *reply,
221	char __user *outputdata,
222	unsigned int outputdatalength)
223	{
224	struct type80_hdr *t80h = reply->message;
225	unsigned char *data;
226
227	if (t80h->len < sizeof(*t80h) + outputdatalength) {
228	/* The result is too short, the CEX2A card may not do that.. */
229	zdev->online = 0;
230	return -EAGAIN; /* repeat the request on a different device. */
231	}
232	BUG_ON(t80h->len > CEX2A_MAX_RESPONSE_SIZE);
233	data = reply->message + t80h->len - outputdatalength;
234	if (copy_to_user(outputdata, data, outputdatalength))
235	return -EFAULT;
236	return 0;
237	}
238
239	static int convert_response(struct zcrypt_device *zdev,
240	struct ap_message *reply,
241	char __user *outputdata,
242	unsigned int outputdatalength)
243	{
244	/* Response type byte is the second byte in the response. */
245	switch (((unsigned char *) reply->message)[1]) {
246	case TYPE82_RSP_CODE:
247	case TYPE88_RSP_CODE:
248	return convert_error(zdev, reply);
249	case TYPE80_RSP_CODE:
250	return convert_type80(zdev, reply,
251	outputdata, outputdatalength);
252	default: /* Unknown response type, this should NEVER EVER happen */
963ed931 MS	253	zdev->online = 0;
	254	return -EAGAIN; /* repeat the request on a different device. */
	255	}
	256	}
	257
	258	/**
	259	* This function is called from the AP bus code after a crypto request
	260	* "msg" has finished with the reply message "reply".
	261	* It is called from tasklet context.
	262	* @ap_dev: pointer to the AP device
	263	* @msg: pointer to the AP message
	264	* @reply: pointer to the AP reply message
	265	*/
	266	static void zcrypt_cex2a_receive(struct ap_device *ap_dev,
	267	struct ap_message *msg,
	268	struct ap_message *reply)
	269	{
	270	static struct error_hdr error_reply = {
	271	.type = TYPE82_RSP_CODE,
	272	.reply_code = REP82_ERROR_MACHINE_FAILURE,
	273	};
21e7b2c4	274	struct type80_hdr *t80h;
963ed931 MS	275	int length;
	276
	277	/* Copy the reply message to the request message buffer. */
21e7b2c4	278	if (IS_ERR(reply)) {
963ed931	279	memcpy(msg->message, &error_reply, sizeof(error_reply));
21e7b2c4 JL	280	goto out;
	281	}
	282	t80h = reply->message;
	283	if (t80h->type == TYPE80_RSP_CODE) {
963ed931 MS	284	length = min(CEX2A_MAX_RESPONSE_SIZE, (int) t80h->len);
	285	memcpy(msg->message, reply->message, length);
	286	} else
	287	memcpy(msg->message, reply->message, sizeof error_reply);
21e7b2c4	288	out:
963ed931 MS	289	complete((struct completion *) msg->private);
	290	}
	291
	292	static atomic_t zcrypt_step = ATOMIC_INIT(0);
	293
	294	/**
	295	* The request distributor calls this function if it picked the CEX2A
	296	* device to handle a modexpo request.
	297	* @zdev: pointer to zcrypt_device structure that identifies the
	298	* CEX2A device to the request distributor
	299	* @mex: pointer to the modexpo request buffer
	300	*/
	301	static long zcrypt_cex2a_modexpo(struct zcrypt_device *zdev,
	302	struct ica_rsa_modexpo *mex)
	303	{
	304	struct ap_message ap_msg;
	305	struct completion work;
	306	int rc;
	307
468ffddf	308	ap_init_message(&ap_msg);
5cbded58	309	ap_msg.message = kmalloc(CEX2A_MAX_MESSAGE_SIZE, GFP_KERNEL);
963ed931 MS	310	if (!ap_msg.message)
	311	return -ENOMEM;
	312	ap_msg.psmid = (((unsigned long long) current->pid) << 32) +
	313	atomic_inc_return(&zcrypt_step);
	314	ap_msg.private = &work;
	315	rc = ICAMEX_msg_to_type50MEX_msg(zdev, &ap_msg, mex);
	316	if (rc)
	317	goto out_free;
	318	init_completion(&work);
	319	ap_queue_message(zdev->ap_dev, &ap_msg);
af512ed0 RW	320	rc = wait_for_completion_interruptible(&work);
af512ed0 RW	321	if (rc == 0)
963ed931 MS	322	rc = convert_response(zdev, &ap_msg, mex->outputdata,
963ed931 MS	323	mex->outputdatalength);
af512ed0 RW	324	else
af512ed0 RW	325	/* Signal pending. */
963ed931	326	ap_cancel_message(zdev->ap_dev, &ap_msg);
963ed931 MS	327	out_free:
	328	kfree(ap_msg.message);
	329	return rc;
	330	}
	331
	332	/**
	333	* The request distributor calls this function if it picked the CEX2A
	334	* device to handle a modexpo_crt request.
	335	* @zdev: pointer to zcrypt_device structure that identifies the
	336	* CEX2A device to the request distributor
	337	* @crt: pointer to the modexpoc_crt request buffer
	338	*/
	339	static long zcrypt_cex2a_modexpo_crt(struct zcrypt_device *zdev,
	340	struct ica_rsa_modexpo_crt *crt)
	341	{
	342	struct ap_message ap_msg;
	343	struct completion work;
	344	int rc;
	345
468ffddf	346	ap_init_message(&ap_msg);
5cbded58	347	ap_msg.message = kmalloc(CEX2A_MAX_MESSAGE_SIZE, GFP_KERNEL);
963ed931 MS	348	if (!ap_msg.message)
	349	return -ENOMEM;
	350	ap_msg.psmid = (((unsigned long long) current->pid) << 32) +
	351	atomic_inc_return(&zcrypt_step);
	352	ap_msg.private = &work;
	353	rc = ICACRT_msg_to_type50CRT_msg(zdev, &ap_msg, crt);
	354	if (rc)
	355	goto out_free;
	356	init_completion(&work);
	357	ap_queue_message(zdev->ap_dev, &ap_msg);
af512ed0 RW	358	rc = wait_for_completion_interruptible(&work);
af512ed0 RW	359	if (rc == 0)
963ed931 MS	360	rc = convert_response(zdev, &ap_msg, crt->outputdata,
963ed931 MS	361	crt->outputdatalength);
af512ed0 RW	362	else
af512ed0 RW	363	/* Signal pending. */
963ed931	364	ap_cancel_message(zdev->ap_dev, &ap_msg);
963ed931 MS	365	out_free:
	366	kfree(ap_msg.message);
	367	return rc;
	368	}
	369
	370	/**
	371	* The crypto operations for a CEX2A card.
	372	*/
	373	static struct zcrypt_ops zcrypt_cex2a_ops = {
	374	.rsa_modexpo = zcrypt_cex2a_modexpo,
	375	.rsa_modexpo_crt = zcrypt_cex2a_modexpo_crt,
	376	};
	377
	378	/**
	379	* Probe function for CEX2A cards. It always accepts the AP device
	380	* since the bus_match already checked the hardware type.
	381	* @ap_dev: pointer to the AP device.
	382	*/
	383	static int zcrypt_cex2a_probe(struct ap_device *ap_dev)
	384	{
8e89b6be FB	385	struct zcrypt_device *zdev = NULL;
	386	int rc = 0;
	387
	388	switch (ap_dev->device_type) {
	389	case AP_DEVICE_TYPE_CEX2A:
	390	zdev = zcrypt_device_alloc(CEX2A_MAX_RESPONSE_SIZE);
	391	if (!zdev)
	392	return -ENOMEM;
	393	zdev->user_space_type = ZCRYPT_CEX2A;
	394	zdev->type_string = "CEX2A";
	395	zdev->min_mod_size = CEX2A_MIN_MOD_SIZE;
	396	zdev->max_mod_size = CEX2A_MAX_MOD_SIZE;
	397	zdev->short_crt = 1;
	398	zdev->speed_rating = CEX2A_SPEED_RATING;
	399	break;
	400	case AP_DEVICE_TYPE_CEX3A:
	401	zdev = zcrypt_device_alloc(CEX3A_MAX_RESPONSE_SIZE);
	402	if (!zdev)
	403	return -ENOMEM;
	404	zdev->user_space_type = ZCRYPT_CEX3A;
	405	zdev->type_string = "CEX3A";
	406	zdev->min_mod_size = CEX3A_MIN_MOD_SIZE;
	407	zdev->max_mod_size = CEX3A_MAX_MOD_SIZE;
	408	zdev->short_crt = 1;
	409	zdev->speed_rating = CEX3A_SPEED_RATING;
	410	break;
	411	}
	412	if (zdev != NULL) {
	413	zdev->ap_dev = ap_dev;
	414	zdev->ops = &zcrypt_cex2a_ops;
	415	zdev->online = 1;
	416	ap_dev->reply = &zdev->reply;
	417	ap_dev->private = zdev;
	418	rc = zcrypt_device_register(zdev);
	419	}
	420	if (rc) {
	421	ap_dev->private = NULL;
	422	zcrypt_device_free(zdev);
	423	}
963ed931 MS	424	return rc;
	425	}
	426
	427	/**
	428	* This is called to remove the extended CEX2A driver information
	429	* if an AP device is removed.
	430	*/
	431	static void zcrypt_cex2a_remove(struct ap_device *ap_dev)
	432	{
	433	struct zcrypt_device *zdev = ap_dev->private;
	434
	435	zcrypt_device_unregister(zdev);
	436	}
	437
	438	int __init zcrypt_cex2a_init(void)
	439	{
	440	return ap_driver_register(&zcrypt_cex2a_driver, THIS_MODULE, "cex2a");
	441	}
	442
	443	void __exit zcrypt_cex2a_exit(void)
	444	{
	445	ap_driver_unregister(&zcrypt_cex2a_driver);
	446	}
	447
	448	#ifndef CONFIG_ZCRYPT_MONOLITHIC
	449	module_init(zcrypt_cex2a_init);
	450	module_exit(zcrypt_cex2a_exit);
	451	#endif