Alessio Di Mauro
128 lines
3.5 KiB
C
128 lines
3.5 KiB
C
#include "token_vendors.h"
|
|
#include "yubico_token.h"
|
|
#include <string.h>
|
|
#include "debug.h"
|
|
static CK_RV COMMON_token_generate_key(ykpiv_state *state, CK_BBOOL rsa, CK_BYTE key, CK_ULONG key_len) {
|
|
// TODO: make a function in ykpiv for this
|
|
unsigned char in_data[5];
|
|
unsigned char data[1024];
|
|
unsigned char templ[] = {0, YKPIV_INS_GENERATE_ASYMMERTRIC, 0, 0};
|
|
unsigned char *certptr;
|
|
unsigned long recv_len = sizeof(data);
|
|
int len_bytes;
|
|
int sw;
|
|
|
|
CK_RV rv;
|
|
|
|
templ[3] = key;
|
|
|
|
in_data[0] = 0xac;
|
|
in_data[1] = 3;
|
|
in_data[2] = 0x80;
|
|
in_data[3] = 1;
|
|
|
|
switch(key_len) {
|
|
case 2048:
|
|
if (rsa == CK_TRUE)
|
|
in_data[4] = YKPIV_ALGO_RSA2048;
|
|
else
|
|
return CKR_FUNCTION_FAILED;
|
|
|
|
break;
|
|
|
|
case 1024:
|
|
if (rsa == CK_TRUE)
|
|
in_data[4] = YKPIV_ALGO_RSA1024;
|
|
else
|
|
return CKR_FUNCTION_FAILED;
|
|
|
|
break;
|
|
|
|
case 256:
|
|
if (rsa == CK_FALSE)
|
|
in_data[4] = YKPIV_ALGO_ECCP256;
|
|
else
|
|
return CKR_FUNCTION_FAILED;
|
|
|
|
break;
|
|
|
|
default:
|
|
return CKR_FUNCTION_FAILED;
|
|
}
|
|
//DBG(("Generating key %x with algorithm %u and length %lu", templ[3], in_data[4], key_len));
|
|
if(ykpiv_transfer_data(state, templ, in_data, sizeof(in_data), data, &recv_len, &sw) != YKPIV_OK ||
|
|
sw != 0x9000)
|
|
return CKR_DEVICE_ERROR;
|
|
|
|
// Create a new empty certificate for the key
|
|
recv_len = sizeof(data);
|
|
if ((rv = do_create_empty_cert(data, recv_len, rsa, key_len, data, &recv_len)) != CKR_OK)
|
|
return rv;
|
|
|
|
if (recv_len < 0x80)
|
|
len_bytes = 1;
|
|
else if (recv_len < 0xff)
|
|
len_bytes = 2;
|
|
else
|
|
len_bytes = 3;
|
|
|
|
certptr = data;
|
|
memmove(data + len_bytes + 1, data, recv_len);
|
|
|
|
*certptr++ = 0x70;
|
|
certptr += set_length(certptr, recv_len);
|
|
certptr += recv_len;
|
|
*certptr++ = 0x71;
|
|
*certptr++ = 1;
|
|
*certptr++ = 0; /* certinfo (gzip etc) */
|
|
*certptr++ = 0xfe; /* LRC */
|
|
*certptr++ = 0;
|
|
|
|
// Store the certificate into the token
|
|
if (ykpiv_save_object(state, key_to_object_id(key), data, (size_t)(certptr - data)) != YKPIV_OK)
|
|
return CKR_DEVICE_ERROR;
|
|
|
|
return CKR_OK;
|
|
}
|
|
|
|
token_vendor_t get_token_vendor(vendor_id_t vid) {
|
|
token_vendor_t v;
|
|
|
|
switch (vid) {
|
|
case YUBICO:
|
|
v.get_token_label = YUBICO_get_token_label;
|
|
v.get_token_manufacturer = YUBICO_get_token_manufacturer;
|
|
v.get_token_model = YUBICO_get_token_model;
|
|
v.get_token_flags = YUBICO_get_token_flags;
|
|
v.get_token_version = YUBICO_get_token_version;
|
|
v.get_token_serial = YUBICO_get_token_serial;
|
|
v.get_token_mechanisms_num = YUBICO_get_token_mechanisms_num;
|
|
v.get_token_mechanism_list = YUBICO_get_token_mechanism_list;
|
|
v.get_token_mechanism_info = YUBICO_get_token_mechanism_info;
|
|
v.get_token_objects_num = YUBICO_get_token_objects_num;
|
|
v.get_token_object_list = YUBICO_get_token_object_list;
|
|
v.get_token_raw_certificate = YUBICO_get_token_raw_certificate;
|
|
v.token_generate_key = COMMON_token_generate_key;
|
|
break;
|
|
|
|
case UNKNOWN:
|
|
default:
|
|
v.get_token_label = NULL;
|
|
v.get_token_manufacturer = NULL;
|
|
v.get_token_model = NULL;
|
|
v.get_token_flags = NULL;
|
|
v.get_token_version = NULL;
|
|
v.get_token_serial = NULL;
|
|
v.get_token_mechanisms_num = NULL;
|
|
v.get_token_mechanism_list = NULL;
|
|
v.get_token_mechanism_info = NULL;
|
|
v.get_token_objects_num = NULL;
|
|
v.get_token_object_list = NULL;
|
|
v.get_token_raw_certificate = NULL;
|
|
v.token_generate_key = NULL;
|
|
}
|
|
|
|
return v;
|
|
|
|
}
|