/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <errno.h>
#include <security/cryptoki.h>
#include <sys/crypto/ioctl.h>
#include "kernelGlobal.h"
#include "kernelSlot.h"

CK_ULONG	slot_count = 0;
kernel_slot_t	**slot_table;

static CK_RV
kernel_get_slot_number()
{
	CK_RV rv;
	crypto_get_provider_list_t *pl;
	int r;

	pl = malloc(sizeof (crypto_get_provider_list_t));
	if (pl == NULL)
		return (CKR_HOST_MEMORY);

	pl->pl_count = 0;
	while ((r = ioctl(kernel_fd, CRYPTO_GET_PROVIDER_LIST, pl)) < 0) {
		if (errno != EINTR)
			break;
	}
	if (r < 0) {
		rv = CKR_FUNCTION_FAILED;
	} else {
		if (pl->pl_return_value != CRYPTO_SUCCESS) {
			rv = crypto2pkcs11_error_number(pl->pl_return_value);
		} else {
			rv = CKR_OK;
		}
	}

	if (rv == CKR_OK) {
		slot_count = pl->pl_count;
	}

	(void) free(pl);
	return (rv);
}

/*
 * To retrieve the crypto_function_list structure with boolean entries
 * indicating which functions are supported by the hardware provider which
 * is specified by the slot ID.
 */
static CK_RV
kernel_get_func_list(kernel_slot_t *pslot)
{
	CK_RV rv = CKR_OK;
	crypto_get_function_list_t  fl;
	int r;

	fl.fl_provider_id = pslot->sl_provider_id;

	while ((r = ioctl(kernel_fd, CRYPTO_GET_FUNCTION_LIST, &fl)) < 0) {
		if (errno != EINTR)
			break;
	}
	if (r < 0) {
		rv = CKR_FUNCTION_FAILED;
	} else {
		if (fl.fl_return_value == 0) {
			rv = CKR_OK;
		} else {
			rv = crypto2pkcs11_error_number(fl.fl_return_value);
		}
	}

	if (rv != CKR_OK) {
		return (rv);
	}

	pslot->sl_func_list.fl_digest_init = fl.fl_list.fl_digest_init;
	pslot->sl_func_list.fl_digest = fl.fl_list.fl_digest;
	pslot->sl_func_list.fl_digest_update = fl.fl_list.fl_digest_update;
	pslot->sl_func_list.fl_digest_key = fl.fl_list.fl_digest_key;
	pslot->sl_func_list.fl_digest_final = fl.fl_list.fl_digest_final;
	pslot->sl_func_list.fl_encrypt_init = fl.fl_list.fl_encrypt_init;
	pslot->sl_func_list.fl_encrypt = fl.fl_list.fl_encrypt;
	pslot->sl_func_list.fl_encrypt_update = fl.fl_list.fl_encrypt_update;
	pslot->sl_func_list.fl_encrypt_final = fl.fl_list.fl_encrypt_final;
	pslot->sl_func_list.fl_decrypt_init = fl.fl_list.fl_decrypt_init;
	pslot->sl_func_list.fl_decrypt = fl.fl_list.fl_decrypt;
	pslot->sl_func_list.fl_decrypt_update = fl.fl_list.fl_decrypt_update;
	pslot->sl_func_list.fl_decrypt_final = fl.fl_list.fl_decrypt_final;
	pslot->sl_func_list.fl_mac_init = fl.fl_list.fl_mac_init;
	pslot->sl_func_list.fl_mac = fl.fl_list.fl_mac;
	pslot->sl_func_list.fl_mac_update = fl.fl_list.fl_mac_update;
	pslot->sl_func_list.fl_mac_final = fl.fl_list.fl_mac_final;
	pslot->sl_func_list.fl_sign_init = fl.fl_list.fl_sign_init;
	pslot->sl_func_list.fl_sign = fl.fl_list.fl_sign;
	pslot->sl_func_list.fl_sign_update = fl.fl_list.fl_sign_update;
	pslot->sl_func_list.fl_sign_final = fl.fl_list.fl_sign_final;
	pslot->sl_func_list.fl_sign_recover_init =
	    fl.fl_list.fl_sign_recover_init;
	pslot->sl_func_list.fl_sign_recover = fl.fl_list.fl_sign_recover;
	pslot->sl_func_list.fl_digest_encrypt_update =
	    fl.fl_list.fl_digest_encrypt_update;
	pslot->sl_func_list.fl_decrypt_digest_update =
	    fl.fl_list.fl_decrypt_digest_update;
	pslot->sl_func_list.fl_sign_encrypt_update =
	    fl.fl_list.fl_sign_encrypt_update;
	pslot->sl_func_list.fl_decrypt_verify_update =
	    fl.fl_list.fl_decrypt_verify_update;
	pslot->sl_func_list.fl_seed_random = fl.fl_list.fl_seed_random;
	pslot->sl_func_list.fl_generate_random = fl.fl_list.fl_generate_random;
	pslot->sl_func_list.fl_session_open = fl.fl_list.fl_session_open;
	pslot->sl_func_list.fl_session_close = fl.fl_list.fl_session_close;
	pslot->sl_func_list.fl_session_login = fl.fl_list.fl_session_login;
	pslot->sl_func_list.fl_session_logout = fl.fl_list.fl_session_logout;
	pslot->sl_func_list.fl_object_create = fl.fl_list.fl_object_create;
	pslot->sl_func_list.fl_object_copy = fl.fl_list.fl_object_copy;
	pslot->sl_func_list.fl_object_destroy = fl.fl_list.fl_object_destroy;
	pslot->sl_func_list.fl_object_get_size = fl.fl_list.fl_object_get_size;
	pslot->sl_func_list.fl_object_get_attribute_value =
	    fl.fl_list.fl_object_get_attribute_value;
	pslot->sl_func_list.fl_object_set_attribute_value =
	    fl.fl_list.fl_object_set_attribute_value;
	pslot->sl_func_list.fl_object_find_init =
	    fl.fl_list.fl_object_find_init;
	pslot->sl_func_list.fl_object_find = fl.fl_list.fl_object_find;
	pslot->sl_func_list.fl_object_find_final =
	    fl.fl_list.fl_object_find_final;
	pslot->sl_func_list.fl_key_generate = fl.fl_list.fl_key_generate;
	pslot->sl_func_list.fl_key_generate_pair =
	    fl.fl_list.fl_key_generate_pair;
	pslot->sl_func_list.fl_key_wrap = fl.fl_list.fl_key_wrap;
	pslot->sl_func_list.fl_key_unwrap = fl.fl_list.fl_key_unwrap;
	pslot->sl_func_list.fl_init_token = fl.fl_list.fl_init_token;
	pslot->sl_func_list.fl_init_pin = fl.fl_list.fl_init_pin;
	pslot->sl_func_list.fl_set_pin = fl.fl_list.fl_set_pin;

	return (CKR_OK);
}

/*
 * Initialize the slot table.
 *
 * This function is called from C_Initialize() only.  Since C_Initialize()
 * holds the global mutex lock, there is no need to acquire another lock
 * in this routine to protect the slot table.
 */
CK_RV
kernel_slottable_init()
{
	int i, cur_slot_num = 0;
	CK_RV rv = CKR_OK;
	crypto_get_provider_list_t *pl = NULL;
	int r;

	/*
	 * Find out how many slots are presented from kernel hardware
	 * providers. If there is no slot presented, just return.
	 */
	rv = kernel_get_slot_number();
	if (rv != CKR_OK || slot_count == 0) {
		return (rv);
	}

	/* Allocate space for the slot table */
	slot_table = malloc(sizeof (kernel_slot_t *) * slot_count);
	if (slot_table == NULL) {
		return (CKR_HOST_MEMORY);
	}

	/* For each slot, allocate space and initialize the slot's mutex. */
	for (i = 0; i < slot_count; i++) {
		slot_table[i] = malloc(sizeof (kernel_slot_t));
		if (slot_table[i] == NULL) {
			rv = CKR_HOST_MEMORY;
			goto failed;
		}

		slot_table[i]->sl_sess_list = NULL;
		slot_table[i]->sl_tobj_list = NULL;
		slot_table[i]->sl_state = CKU_PUBLIC;

		/* Initialize this slot's mutex */
		if (pthread_mutex_init(&slot_table[i]->sl_mutex, NULL) != 0) {
			rv = CKR_FUNCTION_FAILED;
			(void) free(slot_table[i]);
			goto failed;
		}

		cur_slot_num = i;
	}

	/*
	 * Get the provider ID for each slot from kernel and save it in the
	 * slot table.
	 */
	pl = malloc(slot_count * sizeof (crypto_get_provider_list_t));
	if (pl == NULL) {
		rv = CKR_HOST_MEMORY;
		goto failed;
	}

	pl->pl_count = slot_count;
	while ((r = ioctl(kernel_fd, CRYPTO_GET_PROVIDER_LIST, pl)) < 0) {
		if (errno != EINTR)
			break;
	}
	if (r < 0) {
		rv = CKR_FUNCTION_FAILED;
		goto failed;
	} else {
		if (pl->pl_return_value != CRYPTO_SUCCESS) {
			rv = crypto2pkcs11_error_number(pl->pl_return_value);
			goto failed;
		} else {
			rv = CKR_OK;
		}
	}

	for (i = 0; i < slot_count; i++) {
		slot_table[i]->sl_provider_id = pl->pl_list[i].pe_provider_id;
	}

	/*
	 * Get the function list for each slot from kernel and save it in
	 * the slot table.
	 */
	for (i = 0; i < slot_count; i++) {
		rv = kernel_get_func_list(slot_table[i]);
		if (rv != CKR_OK) {
			goto failed;
		}
	}

	(void) free(pl);
	return (CKR_OK);

failed:
	for (i = 0; i < cur_slot_num; i++) {
		(void) pthread_mutex_destroy(&slot_table[i]->sl_mutex);
		(void) free(slot_table[i]);
	}

	(void) free(slot_table);
	(void) free(pl);
	return (rv);
}