uint8_t custom_command(struct u2f_hid_msg * msg)
{
struct atecc_response res;
uint8_t ec;
if (msg->cid != U2FHID_BROADCAST) return 0;
switch(msg->pkt.init.cmd)
{
case U2F_CUSTOM_GET_RNG:
if (atecc_send_recv(ATECC_CMD_RNG,ATECC_RNG_P1,ATECC_RNG_P2,
NULL, 0,
appdata.tmp,
sizeof(appdata.tmp), &res) == 0 )
{
memmove(msg->pkt.init.payload, res.buf, 32);
U2FHID_SET_LEN(msg, 32);
usb_write((uint8_t*)msg, 64);
}
else
{
U2FHID_SET_LEN(msg, 0);
usb_write((uint8_t*)msg, 64);
}
break;
case U2F_CUSTOM_SEED_RNG:
ec = atecc_send_recv(ATECC_CMD_NONCE,ATECC_NONCE_RNG_UPDATE,0,
msg->pkt.init.payload, 20,
appdata.tmp,
sizeof(appdata.tmp), &res);
U2FHID_SET_LEN(msg, 1);
msg->pkt.init.payload[0] = ec == 0 ? 1 : 0;
usb_write((uint8_t*)msg, 64);
break;
case U2F_CUSTOM_WIPE_KEYS:
U2FHID_SET_LEN(msg, 1);
ec=u2f_wipe_keys();
msg->pkt.init.payload[0] = ec == 0 ? 1 : 0;
usb_write((uint8_t*)msg, 64);
break;
case U2F_CUSTOM_WINK:
app_wink(U2F_COLOR_WINK);
break;
default:
return 0;
}
return 1;
}
uint8_t custom_command(struct u2f_hid_msg * msg)
{
struct atecc_response res;
struct u2f_hid_msg * reply = (struct u2f_hid_msg *)appdata.tmp;
uint8_t ec;
if (msg->cid != U2FHID_BROADCAST) return 0;
switch(msg->pkt.init.cmd)
{
case U2F_CUSTOM_GET_SERIAL_NUMBER:
if (atecc_send_recv(ATECC_CMD_READ, ATECC_RW_CONFIG | ATECC_RW_EXT, 0,
NULL, 0, reply->pkt.init.payload, sizeof(reply->pkt.init.payload), &res) == 0 )
{
if (res.len > 15)
res.len = 15;
reply->cid = msg->cid;
reply->pkt.init.cmd = msg->pkt.init.cmd;
U2FHID_SET_LEN(reply, res.len);
usb_write((uint8_t*)reply, 64);
}
else
{
reply->cid = msg->cid;
reply->pkt.init.cmd = msg->pkt.init.cmd;
U2FHID_SET_LEN(reply, 0);
usb_write((uint8_t*)reply, 64);
}
break;
case U2F_CUSTOM_GET_CONFIG:
if (atecc_send_recv(ATECC_CMD_READ, ATECC_RW_CONFIG | ATECC_RW_EXT,
(msg->pkt.init.payload[0] << 8) | msg->pkt.init.payload[1], NULL, 0,
reply->pkt.init.payload, sizeof(reply->pkt.init.payload), &res) == 0)
{
if (res.len > 40)
res.len = 40;
reply->cid = msg->cid;
reply->pkt.init.cmd = msg->pkt.init.cmd;
U2FHID_SET_LEN(reply, res.len);
usb_write((uint8_t*)reply, 64);
}
else
{
U2FHID_SET_LEN(msg, 0);
usb_write((uint8_t*)msg, 64);
}
break;
case U2F_CUSTOM_INIT_CONFIG:
atecc_setup_config();
U2FHID_SET_LEN(msg, 0);
usb_write((uint8_t*)msg, 64);
break;
case U2F_CUSTOM_LOCK_CONFIG:
if (is_locked(appdata.tmp)) {
msg->pkt.init.payload[0] = 0xff;
U2FHID_SET_LEN(msg, 1);
usb_write((uint8_t*)msg, 64);
}
else if (atecc_send_recv(ATECC_CMD_LOCK, ATECC_LOCK_CONFIG,
(msg->pkt.init.payload[0] << 8) | msg->pkt.init.payload[1], NULL, 0,
appdata.tmp, sizeof(appdata.tmp), NULL))
{
msg->pkt.init.payload[0] = 0xfe;
U2FHID_SET_LEN(msg, 1);
usb_write((uint8_t*)msg, 64);
}
else
{
msg->pkt.init.payload[0] = 0x00;
U2FHID_SET_LEN(msg, 1);
usb_write((uint8_t*)msg, 64);
}
break;
case U2F_CUSTOM_GEN_ATT_KEY:
if (atecc_send_recv(ATECC_CMD_GENKEY, ATECC_GENKEY_PRIVATE, U2F_ATTESTATION_KEY_SLOT,
NULL, 0, appdata.tmp, sizeof(appdata.tmp), &res) == 0 && res.len <= 64)
{
U2FHID_SET_LEN(msg, res.len - 1);
memmove(msg->pkt.init.payload, res.buf, res.len - 1);
usb_write((uint8_t*)msg, 64);
}
else
{
U2FHID_SET_LEN(msg, 0);
usb_write((uint8_t*)msg, 64);
}
break;
/*
case U2F_CUSTOM_GET_RNG:
if (atecc_send_recv(ATECC_CMD_RNG,ATECC_RNG_P1,ATECC_RNG_P2,
NULL, 0,
appdata.tmp,
sizeof(appdata.tmp), &res) == 0 )
{
memmove(msg->pkt.init.payload, res.buf, 32);
U2FHID_SET_LEN(msg, 32);
usb_write((uint8_t*)msg, 64);
}
else
{
U2FHID_SET_LEN(msg, 0);
usb_write((uint8_t*)msg, 64);
}
break;
case U2F_CUSTOM_SEED_RNG:
ec = atecc_send_recv(ATECC_CMD_NONCE,ATECC_NONCE_RNG_UPDATE,0,
msg->pkt.init.payload, 20,
appdata.tmp,
sizeof(appdata.tmp), &res);
U2FHID_SET_LEN(msg, 1);
msg->pkt.init.payload[0] = ec == 0 ? 1 : 0;
usb_write((uint8_t*)msg, 64);
break;
*/
case U2F_CUSTOM_WIPE_KEYS:
U2FHID_SET_LEN(msg, 1);
ec=u2f_wipe_keys();
msg->pkt.init.payload[0] = ec == 0 ? 1 : 0;
usb_write((uint8_t*)msg, 64);
break;
case U2F_CUSTOM_ENTER_BOOTLOADER:
USB_Detach();
enterBootloader();
break;
case U2F_CUSTOM_INC_COUNT:
if (atecc_send_recv(ATECC_CMD_COUNTER,
ATECC_COUNTER_INC, ATECC_COUNTER0,NULL,0,
appdata.tmp, sizeof(appdata.tmp), &res) == 0)
{
memmove(msg->pkt.init.payload, res.buf, res.len);
U2FHID_SET_LEN(msg, res.len);
usb_write((uint8_t*)msg, 64);
}
else
{
U2FHID_SET_LEN(msg, 0);
usb_write((uint8_t*)msg, 64);
}
break;
default:
return 0;
}
return 1;
}