[linux.git] / drivers / s390 / crypto / zcrypt_cex4.c

// SPDX-License-Identifier: GPL-2.0
/*
 *  Copyright IBM Corp. 2012
 *  Author(s): Holger Dengler <[email protected]>
 */

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

#include "ap_bus.h"
#include "zcrypt_api.h"
#include "zcrypt_msgtype6.h"
#include "zcrypt_msgtype50.h"
#include "zcrypt_error.h"
#include "zcrypt_cex4.h"

#define CEX4A_MIN_MOD_SIZE	  1	/*    8 bits	*/
#define CEX4A_MAX_MOD_SIZE_2K	256	/* 2048 bits	*/
#define CEX4A_MAX_MOD_SIZE_4K	512	/* 4096 bits	*/

#define CEX4C_MIN_MOD_SIZE	 16	/*  256 bits	*/
#define CEX4C_MAX_MOD_SIZE	512	/* 4096 bits	*/

#define CEX4A_MAX_MESSAGE_SIZE	MSGTYPE50_CRB3_MAX_MSG_SIZE
#define CEX4C_MAX_MESSAGE_SIZE	MSGTYPE06_MAX_MSG_SIZE

/* Waiting time for requests to be processed.
 * Currently there are some types of request which are not deterministic.
 * But the maximum time limit managed by the stomper code is set to 60sec.
 * Hence we have to wait at least that time period.
 */
#define CEX4_CLEANUP_TIME	(900*HZ)

MODULE_AUTHOR("IBM Corporation");
MODULE_DESCRIPTION("CEX4/CEX5/CEX6 Cryptographic Card device driver, " \
		   "Copyright IBM Corp. 2018");
MODULE_LICENSE("GPL");

static struct ap_device_id zcrypt_cex4_card_ids[] = {
	{ .dev_type = AP_DEVICE_TYPE_CEX4,
	  .match_flags = AP_DEVICE_ID_MATCH_CARD_TYPE },
	{ .dev_type = AP_DEVICE_TYPE_CEX5,
	  .match_flags = AP_DEVICE_ID_MATCH_CARD_TYPE },
	{ .dev_type = AP_DEVICE_TYPE_CEX6,
	  .match_flags = AP_DEVICE_ID_MATCH_CARD_TYPE },
	{ /* end of list */ },
};

MODULE_DEVICE_TABLE(ap, zcrypt_cex4_card_ids);

static struct ap_device_id zcrypt_cex4_queue_ids[] = {
	{ .dev_type = AP_DEVICE_TYPE_CEX4,
	  .match_flags = AP_DEVICE_ID_MATCH_QUEUE_TYPE },
	{ .dev_type = AP_DEVICE_TYPE_CEX5,
	  .match_flags = AP_DEVICE_ID_MATCH_QUEUE_TYPE },
	{ .dev_type = AP_DEVICE_TYPE_CEX6,
	  .match_flags = AP_DEVICE_ID_MATCH_QUEUE_TYPE },
	{ /* end of list */ },
};

MODULE_DEVICE_TABLE(ap, zcrypt_cex4_queue_ids);

/**
 * Probe function for CEX4/CEX5/CEX6 card device. 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_cex4_card_probe(struct ap_device *ap_dev)
{
	/*
	 * Normalized speed ratings per crypto adapter
	 * MEX_1k, MEX_2k, MEX_4k, CRT_1k, CRT_2k, CRT_4k, RNG, SECKEY
	 */
	static const int CEX4A_SPEED_IDX[] = {
		 14, 19, 249, 42, 228, 1458, 0, 0};
	static const int CEX5A_SPEED_IDX[] = {
		  8,  9,  20, 18,  66,	458, 0, 0};
	static const int CEX6A_SPEED_IDX[] = {
		  6,  9,  20, 17,  65,	438, 0, 0};

	static const int CEX4C_SPEED_IDX[] = {
		 59,  69, 308, 83, 278, 2204, 209, 40};
	static const int CEX5C_SPEED_IDX[] = {
		 24,  31,  50, 37,  90,  479,  27, 10};
	static const int CEX6C_SPEED_IDX[] = {
		 16,  20,  32, 27,  77,  455,  23,  9};

	static const int CEX4P_SPEED_IDX[] = {
		224, 313, 3560, 359, 605, 2827, 0, 50};
	static const int CEX5P_SPEED_IDX[] = {
		 63,  84,  156,  83, 142,  533, 0, 10};
	static const int CEX6P_SPEED_IDX[] = {
		 55,  70,  121,  73, 129,  522, 0,  9};

	struct ap_card *ac = to_ap_card(&ap_dev->device);
	struct zcrypt_card *zc;
	int rc = 0;

	zc = zcrypt_card_alloc();
	if (!zc)
		return -ENOMEM;
	zc->card = ac;
	ac->private = zc;
	if (ap_test_bit(&ac->functions, AP_FUNC_ACCEL)) {
		if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX4) {
			zc->type_string = "CEX4A";
			zc->user_space_type = ZCRYPT_CEX4;
			memcpy(zc->speed_rating, CEX4A_SPEED_IDX,
			       sizeof(CEX4A_SPEED_IDX));
		} else if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX5) {
			zc->type_string = "CEX5A";
			zc->user_space_type = ZCRYPT_CEX5;
			memcpy(zc->speed_rating, CEX5A_SPEED_IDX,
			       sizeof(CEX5A_SPEED_IDX));
		} else {
			zc->type_string = "CEX6A";
			zc->user_space_type = ZCRYPT_CEX6;
			memcpy(zc->speed_rating, CEX6A_SPEED_IDX,
			       sizeof(CEX6A_SPEED_IDX));
		}
		zc->min_mod_size = CEX4A_MIN_MOD_SIZE;
		if (ap_test_bit(&ac->functions, AP_FUNC_MEX4K) &&
		    ap_test_bit(&ac->functions, AP_FUNC_CRT4K)) {
			zc->max_mod_size = CEX4A_MAX_MOD_SIZE_4K;
			zc->max_exp_bit_length =
				CEX4A_MAX_MOD_SIZE_4K;
		} else {
			zc->max_mod_size = CEX4A_MAX_MOD_SIZE_2K;
			zc->max_exp_bit_length =
				CEX4A_MAX_MOD_SIZE_2K;
		}
	} else if (ap_test_bit(&ac->functions, AP_FUNC_COPRO)) {
		if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX4) {
			zc->type_string = "CEX4C";
			/* wrong user space type, must be CEX4
			 * just keep it for cca compatibility
			 */
			zc->user_space_type = ZCRYPT_CEX3C;
			memcpy(zc->speed_rating, CEX4C_SPEED_IDX,
			       sizeof(CEX4C_SPEED_IDX));
		} else if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX5) {
			zc->type_string = "CEX5C";
			/* wrong user space type, must be CEX5
			 * just keep it for cca compatibility
			 */
			zc->user_space_type = ZCRYPT_CEX3C;
			memcpy(zc->speed_rating, CEX5C_SPEED_IDX,
			       sizeof(CEX5C_SPEED_IDX));
		} else {
			zc->type_string = "CEX6C";
			/* wrong user space type, must be CEX6
			 * just keep it for cca compatibility
			 */
			zc->user_space_type = ZCRYPT_CEX3C;
			memcpy(zc->speed_rating, CEX6C_SPEED_IDX,
			       sizeof(CEX6C_SPEED_IDX));
		}
		zc->min_mod_size = CEX4C_MIN_MOD_SIZE;
		zc->max_mod_size = CEX4C_MAX_MOD_SIZE;
		zc->max_exp_bit_length = CEX4C_MAX_MOD_SIZE;
	} else if (ap_test_bit(&ac->functions, AP_FUNC_EP11)) {
		if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX4) {
			zc->type_string = "CEX4P";
			zc->user_space_type = ZCRYPT_CEX4;
			memcpy(zc->speed_rating, CEX4P_SPEED_IDX,
			       sizeof(CEX4P_SPEED_IDX));
		} else if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX5) {
			zc->type_string = "CEX5P";
			zc->user_space_type = ZCRYPT_CEX5;
			memcpy(zc->speed_rating, CEX5P_SPEED_IDX,
			       sizeof(CEX5P_SPEED_IDX));
		} else {
			zc->type_string = "CEX6P";
			zc->user_space_type = ZCRYPT_CEX6;
			memcpy(zc->speed_rating, CEX6P_SPEED_IDX,
			       sizeof(CEX6P_SPEED_IDX));
		}
		zc->min_mod_size = CEX4C_MIN_MOD_SIZE;
		zc->max_mod_size = CEX4C_MAX_MOD_SIZE;
		zc->max_exp_bit_length = CEX4C_MAX_MOD_SIZE;
	} else {
		zcrypt_card_free(zc);
		return -ENODEV;
	}
	zc->online = 1;

	rc = zcrypt_card_register(zc);
	if (rc) {
		ac->private = NULL;
		zcrypt_card_free(zc);
	}

	return rc;
}

/**
 * This is called to remove the CEX4/CEX5/CEX6 card driver information
 * if an AP card device is removed.
 */
static void zcrypt_cex4_card_remove(struct ap_device *ap_dev)
{
	struct zcrypt_card *zc = to_ap_card(&ap_dev->device)->private;

	if (zc)
		zcrypt_card_unregister(zc);
}

static struct ap_driver zcrypt_cex4_card_driver = {
	.probe = zcrypt_cex4_card_probe,
	.remove = zcrypt_cex4_card_remove,
	.ids = zcrypt_cex4_card_ids,
	.flags = AP_DRIVER_FLAG_DEFAULT,
};

/**
 * Probe function for CEX4/CEX5/CEX6 queue device. 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_cex4_queue_probe(struct ap_device *ap_dev)
{
	struct ap_queue *aq = to_ap_queue(&ap_dev->device);
	struct zcrypt_queue *zq;
	int rc;

	if (ap_test_bit(&aq->card->functions, AP_FUNC_ACCEL)) {
		zq = zcrypt_queue_alloc(CEX4A_MAX_MESSAGE_SIZE);
		if (!zq)
			return -ENOMEM;
		zq->ops = zcrypt_msgtype(MSGTYPE50_NAME,
					 MSGTYPE50_VARIANT_DEFAULT);
	} else if (ap_test_bit(&aq->card->functions, AP_FUNC_COPRO)) {
		zq = zcrypt_queue_alloc(CEX4C_MAX_MESSAGE_SIZE);
		if (!zq)
			return -ENOMEM;
		zq->ops = zcrypt_msgtype(MSGTYPE06_NAME,
					 MSGTYPE06_VARIANT_DEFAULT);
	} else if (ap_test_bit(&aq->card->functions, AP_FUNC_EP11)) {
		zq = zcrypt_queue_alloc(CEX4C_MAX_MESSAGE_SIZE);
		if (!zq)
			return -ENOMEM;
		zq->ops = zcrypt_msgtype(MSGTYPE06_NAME,
					 MSGTYPE06_VARIANT_EP11);
	} else {
		return -ENODEV;
	}
	zq->queue = aq;
	zq->online = 1;
	atomic_set(&zq->load, 0);
	ap_queue_init_reply(aq, &zq->reply);
	aq->request_timeout = CEX4_CLEANUP_TIME,
	aq->private = zq;
	rc = zcrypt_queue_register(zq);
	if (rc) {
		aq->private = NULL;
		zcrypt_queue_free(zq);
	}

	return rc;
}

/**
 * This is called to remove the CEX4/CEX5/CEX6 queue driver
 * information if an AP queue device is removed.
 */
static void zcrypt_cex4_queue_remove(struct ap_device *ap_dev)
{
	struct ap_queue *aq = to_ap_queue(&ap_dev->device);
	struct zcrypt_queue *zq = aq->private;

	if (zq)
		zcrypt_queue_unregister(zq);
}

static struct ap_driver zcrypt_cex4_queue_driver = {
	.probe = zcrypt_cex4_queue_probe,
	.remove = zcrypt_cex4_queue_remove,
	.suspend = ap_queue_suspend,
	.resume = ap_queue_resume,
	.ids = zcrypt_cex4_queue_ids,
	.flags = AP_DRIVER_FLAG_DEFAULT,
};

int __init zcrypt_cex4_init(void)
{
	int rc;

	rc = ap_driver_register(&zcrypt_cex4_card_driver,
				THIS_MODULE, "cex4card");
	if (rc)
		return rc;

	rc = ap_driver_register(&zcrypt_cex4_queue_driver,
				THIS_MODULE, "cex4queue");
	if (rc)
		ap_driver_unregister(&zcrypt_cex4_card_driver);

	return rc;
}

void __exit zcrypt_cex4_exit(void)
{
	ap_driver_unregister(&zcrypt_cex4_queue_driver);
	ap_driver_unregister(&zcrypt_cex4_card_driver);
}

module_init(zcrypt_cex4_init);
module_exit(zcrypt_cex4_exit);
Commit	Line	Data
812141a9	1	// SPDX-License-Identifier: GPL-2.0
1e2076f4 HD	2	/*
	3	* Copyright IBM Corp. 2012
	4	* Author(s): Holger Dengler <[email protected]>
	5	*/
	6
	7	#include <linux/module.h>
	8	#include <linux/slab.h>
	9	#include <linux/init.h>
	10	#include <linux/err.h>
	11	#include <linux/atomic.h>
	12	#include <linux/uaccess.h>
e28d2af4	13	#include <linux/mod_devicetable.h>
1e2076f4 HD	14
	15	#include "ap_bus.h"
	16	#include "zcrypt_api.h"
	17	#include "zcrypt_msgtype6.h"
	18	#include "zcrypt_msgtype50.h"
	19	#include "zcrypt_error.h"
	20	#include "zcrypt_cex4.h"
	21
	22	#define CEX4A_MIN_MOD_SIZE 1 /* 8 bits */
	23	#define CEX4A_MAX_MOD_SIZE_2K 256 /* 2048 bits */
	24	#define CEX4A_MAX_MOD_SIZE_4K 512 /* 4096 bits */
	25
	26	#define CEX4C_MIN_MOD_SIZE 16 /* 256 bits */
	27	#define CEX4C_MAX_MOD_SIZE 512 /* 4096 bits */
	28
1e2076f4 HD	29	#define CEX4A_MAX_MESSAGE_SIZE MSGTYPE50_CRB3_MAX_MSG_SIZE
	30	#define CEX4C_MAX_MESSAGE_SIZE MSGTYPE06_MAX_MSG_SIZE
	31
91f3e3ea IT	32	/* Waiting time for requests to be processed.
	33	* Currently there are some types of request which are not deterministic.
	34	* But the maximum time limit managed by the stomper code is set to 60sec.
	35	* Hence we have to wait at least that time period.
	36	*/
c50a160c	37	#define CEX4_CLEANUP_TIME (900*HZ)
1e2076f4	38
1e2076f4	39	MODULE_AUTHOR("IBM Corporation");
ee410de8 HF	40	MODULE_DESCRIPTION("CEX4/CEX5/CEX6 Cryptographic Card device driver, " \
ee410de8 HF	41	"Copyright IBM Corp. 2018");
1e2076f4 HD	42	MODULE_LICENSE("GPL");
1e2076f4 HD	43
e28d2af4 IT	44	static struct ap_device_id zcrypt_cex4_card_ids[] = {
	45	{ .dev_type = AP_DEVICE_TYPE_CEX4,
	46	.match_flags = AP_DEVICE_ID_MATCH_CARD_TYPE },
	47	{ .dev_type = AP_DEVICE_TYPE_CEX5,
	48	.match_flags = AP_DEVICE_ID_MATCH_CARD_TYPE },
21214b04 HF	49	{ .dev_type = AP_DEVICE_TYPE_CEX6,
21214b04 HF	50	.match_flags = AP_DEVICE_ID_MATCH_CARD_TYPE },
e28d2af4 IT	51	{ /* end of list */ },
	52	};
	53
	54	MODULE_DEVICE_TABLE(ap, zcrypt_cex4_card_ids);
1e2076f4	55
e28d2af4 IT	56	static struct ap_device_id zcrypt_cex4_queue_ids[] = {
	57	{ .dev_type = AP_DEVICE_TYPE_CEX4,
	58	.match_flags = AP_DEVICE_ID_MATCH_QUEUE_TYPE },
	59	{ .dev_type = AP_DEVICE_TYPE_CEX5,
	60	.match_flags = AP_DEVICE_ID_MATCH_QUEUE_TYPE },
21214b04 HF	61	{ .dev_type = AP_DEVICE_TYPE_CEX6,
21214b04 HF	62	.match_flags = AP_DEVICE_ID_MATCH_QUEUE_TYPE },
e28d2af4	63	{ /* end of list */ },
1e2076f4 HD	64	};
1e2076f4 HD	65
e28d2af4 IT	66	MODULE_DEVICE_TABLE(ap, zcrypt_cex4_queue_ids);
e28d2af4 IT	67
1e2076f4	68	/**
ee410de8 HF	69	* Probe function for CEX4/CEX5/CEX6 card device. It always
	70	* accepts the AP device since the bus_match already checked
	71	* the hardware type.
1e2076f4 HD	72	* @ap_dev: pointer to the AP device.
1e2076f4 HD	73	*/
e28d2af4	74	static int zcrypt_cex4_card_probe(struct ap_device *ap_dev)
1e2076f4	75	{
34a15167 IT	76	/*
	77	* Normalized speed ratings per crypto adapter
	78	* MEX_1k, MEX_2k, MEX_4k, CRT_1k, CRT_2k, CRT_4k, RNG, SECKEY
	79	*/
e28d2af4	80	static const int CEX4A_SPEED_IDX[] = {
21214b04	81	14, 19, 249, 42, 228, 1458, 0, 0};
e28d2af4	82	static const int CEX5A_SPEED_IDX[] = {
21214b04 HF	83	8, 9, 20, 18, 66, 458, 0, 0};
	84	static const int CEX6A_SPEED_IDX[] = {
	85	6, 9, 20, 17, 65, 438, 0, 0};
	86
e28d2af4	87	static const int CEX4C_SPEED_IDX[] = {
21214b04	88	59, 69, 308, 83, 278, 2204, 209, 40};
e28d2af4	89	static const int CEX5C_SPEED_IDX[] = {
21214b04 HF	90	24, 31, 50, 37, 90, 479, 27, 10};
	91	static const int CEX6C_SPEED_IDX[] = {
	92	16, 20, 32, 27, 77, 455, 23, 9};
	93
e28d2af4	94	static const int CEX4P_SPEED_IDX[] = {
21214b04	95	224, 313, 3560, 359, 605, 2827, 0, 50};
e28d2af4	96	static const int CEX5P_SPEED_IDX[] = {
21214b04 HF	97	63, 84, 156, 83, 142, 533, 0, 10};
	98	static const int CEX6P_SPEED_IDX[] = {
	99	55, 70, 121, 73, 129, 522, 0, 9};
e28d2af4 IT	100
	101	struct ap_card *ac = to_ap_card(&ap_dev->device);
	102	struct zcrypt_card *zc;
1e2076f4 HD	103	int rc = 0;
1e2076f4 HD	104
e28d2af4 IT	105	zc = zcrypt_card_alloc();
	106	if (!zc)
	107	return -ENOMEM;
	108	zc->card = ac;
	109	ac->private = zc;
	110	if (ap_test_bit(&ac->functions, AP_FUNC_ACCEL)) {
	111	if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX4) {
	112	zc->type_string = "CEX4A";
	113	zc->user_space_type = ZCRYPT_CEX4;
	114	memcpy(zc->speed_rating, CEX4A_SPEED_IDX,
	115	sizeof(CEX4A_SPEED_IDX));
21214b04	116	} else if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX5) {
e28d2af4 IT	117	zc->type_string = "CEX5A";
	118	zc->user_space_type = ZCRYPT_CEX5;
	119	memcpy(zc->speed_rating, CEX5A_SPEED_IDX,
	120	sizeof(CEX5A_SPEED_IDX));
21214b04 HF	121	} else {
	122	zc->type_string = "CEX6A";
	123	zc->user_space_type = ZCRYPT_CEX6;
	124	memcpy(zc->speed_rating, CEX6A_SPEED_IDX,
	125	sizeof(CEX6A_SPEED_IDX));
1e2076f4	126	}
e28d2af4 IT	127	zc->min_mod_size = CEX4A_MIN_MOD_SIZE;
	128	if (ap_test_bit(&ac->functions, AP_FUNC_MEX4K) &&
	129	ap_test_bit(&ac->functions, AP_FUNC_CRT4K)) {
	130	zc->max_mod_size = CEX4A_MAX_MOD_SIZE_4K;
	131	zc->max_exp_bit_length =
	132	CEX4A_MAX_MOD_SIZE_4K;
	133	} else {
	134	zc->max_mod_size = CEX4A_MAX_MOD_SIZE_2K;
	135	zc->max_exp_bit_length =
	136	CEX4A_MAX_MOD_SIZE_2K;
	137	}
	138	} else if (ap_test_bit(&ac->functions, AP_FUNC_COPRO)) {
	139	if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX4) {
	140	zc->type_string = "CEX4C";
	141	/* wrong user space type, must be CEX4
	142	* just keep it for cca compatibility
	143	*/
	144	zc->user_space_type = ZCRYPT_CEX3C;
	145	memcpy(zc->speed_rating, CEX4C_SPEED_IDX,
	146	sizeof(CEX4C_SPEED_IDX));
21214b04	147	} else if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX5) {
e28d2af4 IT	148	zc->type_string = "CEX5C";
	149	/* wrong user space type, must be CEX5
	150	* just keep it for cca compatibility
	151	*/
	152	zc->user_space_type = ZCRYPT_CEX3C;
	153	memcpy(zc->speed_rating, CEX5C_SPEED_IDX,
	154	sizeof(CEX5C_SPEED_IDX));
21214b04 HF	155	} else {
	156	zc->type_string = "CEX6C";
	157	/* wrong user space type, must be CEX6
	158	* just keep it for cca compatibility
	159	*/
	160	zc->user_space_type = ZCRYPT_CEX3C;
	161	memcpy(zc->speed_rating, CEX6C_SPEED_IDX,
	162	sizeof(CEX6C_SPEED_IDX));
e28d2af4 IT	163	}
	164	zc->min_mod_size = CEX4C_MIN_MOD_SIZE;
	165	zc->max_mod_size = CEX4C_MAX_MOD_SIZE;
	166	zc->max_exp_bit_length = CEX4C_MAX_MOD_SIZE;
	167	} else if (ap_test_bit(&ac->functions, AP_FUNC_EP11)) {
	168	if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX4) {
	169	zc->type_string = "CEX4P";
	170	zc->user_space_type = ZCRYPT_CEX4;
	171	memcpy(zc->speed_rating, CEX4P_SPEED_IDX,
	172	sizeof(CEX4P_SPEED_IDX));
21214b04	173	} else if (ac->ap_dev.device_type == AP_DEVICE_TYPE_CEX5) {
e28d2af4 IT	174	zc->type_string = "CEX5P";
	175	zc->user_space_type = ZCRYPT_CEX5;
	176	memcpy(zc->speed_rating, CEX5P_SPEED_IDX,
	177	sizeof(CEX5P_SPEED_IDX));
21214b04 HF	178	} else {
	179	zc->type_string = "CEX6P";
	180	zc->user_space_type = ZCRYPT_CEX6;
	181	memcpy(zc->speed_rating, CEX6P_SPEED_IDX,
	182	sizeof(CEX6P_SPEED_IDX));
e28d2af4 IT	183	}
	184	zc->min_mod_size = CEX4C_MIN_MOD_SIZE;
	185	zc->max_mod_size = CEX4C_MAX_MOD_SIZE;
	186	zc->max_exp_bit_length = CEX4C_MAX_MOD_SIZE;
	187	} else {
	188	zcrypt_card_free(zc);
1e2076f4	189	return -ENODEV;
e28d2af4 IT	190	}
	191	zc->online = 1;
	192
	193	rc = zcrypt_card_register(zc);
1e2076f4	194	if (rc) {
e28d2af4 IT	195	ac->private = NULL;
e28d2af4 IT	196	zcrypt_card_free(zc);
1e2076f4	197	}
e28d2af4	198
1e2076f4 HD	199	return rc;
	200	}
	201
	202	/**
ee410de8	203	* This is called to remove the CEX4/CEX5/CEX6 card driver information
e28d2af4	204	* if an AP card device is removed.
1e2076f4	205	*/
e28d2af4	206	static void zcrypt_cex4_card_remove(struct ap_device *ap_dev)
1e2076f4	207	{
e28d2af4	208	struct zcrypt_card *zc = to_ap_card(&ap_dev->device)->private;
1e2076f4	209
e28d2af4 IT	210	if (zc)
	211	zcrypt_card_unregister(zc);
	212	}
	213
	214	static struct ap_driver zcrypt_cex4_card_driver = {
	215	.probe = zcrypt_cex4_card_probe,
	216	.remove = zcrypt_cex4_card_remove,
	217	.ids = zcrypt_cex4_card_ids,
7e0bdbe5	218	.flags = AP_DRIVER_FLAG_DEFAULT,
e28d2af4 IT	219	};
	220
	221	/**
ee410de8 HF	222	* Probe function for CEX4/CEX5/CEX6 queue device. It always
	223	* accepts the AP device since the bus_match already checked
	224	* the hardware type.
e28d2af4 IT	225	* @ap_dev: pointer to the AP device.
	226	*/
	227	static int zcrypt_cex4_queue_probe(struct ap_device *ap_dev)
	228	{
	229	struct ap_queue *aq = to_ap_queue(&ap_dev->device);
	230	struct zcrypt_queue *zq;
	231	int rc;
	232
	233	if (ap_test_bit(&aq->card->functions, AP_FUNC_ACCEL)) {
	234	zq = zcrypt_queue_alloc(CEX4A_MAX_MESSAGE_SIZE);
	235	if (!zq)
	236	return -ENOMEM;
	237	zq->ops = zcrypt_msgtype(MSGTYPE50_NAME,
	238	MSGTYPE50_VARIANT_DEFAULT);
	239	} else if (ap_test_bit(&aq->card->functions, AP_FUNC_COPRO)) {
	240	zq = zcrypt_queue_alloc(CEX4C_MAX_MESSAGE_SIZE);
	241	if (!zq)
	242	return -ENOMEM;
	243	zq->ops = zcrypt_msgtype(MSGTYPE06_NAME,
	244	MSGTYPE06_VARIANT_DEFAULT);
	245	} else if (ap_test_bit(&aq->card->functions, AP_FUNC_EP11)) {
	246	zq = zcrypt_queue_alloc(CEX4C_MAX_MESSAGE_SIZE);
	247	if (!zq)
	248	return -ENOMEM;
	249	zq->ops = zcrypt_msgtype(MSGTYPE06_NAME,
	250	MSGTYPE06_VARIANT_EP11);
	251	} else {
	252	return -ENODEV;
1e2076f4	253	}
e28d2af4 IT	254	zq->queue = aq;
	255	zq->online = 1;
	256	atomic_set(&zq->load, 0);
	257	ap_queue_init_reply(aq, &zq->reply);
	258	aq->request_timeout = CEX4_CLEANUP_TIME,
	259	aq->private = zq;
	260	rc = zcrypt_queue_register(zq);
	261	if (rc) {
	262	aq->private = NULL;
	263	zcrypt_queue_free(zq);
	264	}
	265
	266	return rc;
1e2076f4 HD	267	}
1e2076f4 HD	268
e28d2af4	269	/**
ee410de8 HF	270	* This is called to remove the CEX4/CEX5/CEX6 queue driver
ee410de8 HF	271	* information if an AP queue device is removed.
e28d2af4 IT	272	*/
	273	static void zcrypt_cex4_queue_remove(struct ap_device *ap_dev)
	274	{
	275	struct ap_queue *aq = to_ap_queue(&ap_dev->device);
	276	struct zcrypt_queue *zq = aq->private;
	277
e28d2af4 IT	278	if (zq)
	279	zcrypt_queue_unregister(zq);
	280	}
	281
	282	static struct ap_driver zcrypt_cex4_queue_driver = {
	283	.probe = zcrypt_cex4_queue_probe,
	284	.remove = zcrypt_cex4_queue_remove,
	285	.suspend = ap_queue_suspend,
	286	.resume = ap_queue_resume,
	287	.ids = zcrypt_cex4_queue_ids,
7e0bdbe5	288	.flags = AP_DRIVER_FLAG_DEFAULT,
e28d2af4 IT	289	};
e28d2af4 IT	290
1e2076f4 HD	291	int __init zcrypt_cex4_init(void)
1e2076f4 HD	292	{
e28d2af4 IT	293	int rc;
	294
	295	rc = ap_driver_register(&zcrypt_cex4_card_driver,
	296	THIS_MODULE, "cex4card");
	297	if (rc)
	298	return rc;
	299
	300	rc = ap_driver_register(&zcrypt_cex4_queue_driver,
	301	THIS_MODULE, "cex4queue");
	302	if (rc)
	303	ap_driver_unregister(&zcrypt_cex4_card_driver);
	304
	305	return rc;
1e2076f4 HD	306	}
	307
	308	void __exit zcrypt_cex4_exit(void)
	309	{
e28d2af4 IT	310	ap_driver_unregister(&zcrypt_cex4_queue_driver);
e28d2af4 IT	311	ap_driver_unregister(&zcrypt_cex4_card_driver);
1e2076f4 HD	312	}
	313
	314	module_init(zcrypt_cex4_init);
	315	module_exit(zcrypt_cex4_exit);