86 lines
3.0 KiB
C
86 lines
3.0 KiB
C
/*
|
|
* This file is part of the Pico OpenPGP distribution (https://github.com/polhenarejos/pico-openpgp).
|
|
* Copyright (c) 2022 Pol Henarejos.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, version 3.
|
|
*
|
|
* 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
|
|
* Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "openpgp.h"
|
|
|
|
int cmd_put_data() {
|
|
uint16_t fid = (P1(apdu) << 8) | P2(apdu);
|
|
file_t *ef;
|
|
if (fid == EF_RESET_CODE) {
|
|
fid = EF_RC;
|
|
}
|
|
else if (fid == EF_ALGO_SIG || fid == EF_ALGO_DEC || fid == EF_ALGO_AUT) {
|
|
fid |= 0x1000;
|
|
}
|
|
if (!(ef = search_by_fid(fid, NULL, SPECIFY_EF))) {
|
|
return SW_REFERENCE_NOT_FOUND();
|
|
}
|
|
if (!authenticate_action(ef, ACL_OP_UPDATE_ERASE)) {
|
|
return SW_SECURITY_STATUS_NOT_SATISFIED();
|
|
}
|
|
if ((fid == EF_PRIV_DO_1 || fid == EF_PRIV_DO_3) && (!has_pw2 && !has_pw3)) {
|
|
return SW_SECURITY_STATUS_NOT_SATISFIED();
|
|
}
|
|
if (!(fid == EF_PRIV_DO_1 || fid == EF_PRIV_DO_3) && !has_pw3) {
|
|
return SW_SECURITY_STATUS_NOT_SATISFIED();
|
|
}
|
|
if (fid == EF_PW_STATUS) {
|
|
fid = EF_PW_PRIV;
|
|
apdu.nc = 4; //we silently ommit the reset parameters
|
|
}
|
|
if (currentEF && currentEF->fid == fid) { // previously selected same EF
|
|
ef = currentEF;
|
|
}
|
|
if (ef->type & FILE_DATA_FLASH) {
|
|
int r = 0;
|
|
if (apdu.nc > 0) {
|
|
if (fid == EF_RC) {
|
|
has_rc = false;
|
|
if ((r = load_dek()) != PICOKEY_OK) {
|
|
return SW_EXEC_ERROR();
|
|
}
|
|
uint8_t dhash[33];
|
|
dhash[0] = apdu.nc;
|
|
double_hash_pin(apdu.data, apdu.nc, dhash + 1);
|
|
r = file_put_data(ef, dhash, sizeof(dhash));
|
|
|
|
file_t *tf = search_by_fid(EF_DEK, NULL, SPECIFY_EF);
|
|
if (!tf) {
|
|
return SW_REFERENCE_NOT_FOUND();
|
|
}
|
|
uint8_t def[IV_SIZE + 32 + 32 + 32 + 32];
|
|
memcpy(def, file_get_data(tf), file_get_size(tf));
|
|
hash_multi(apdu.data, apdu.nc, session_rc);
|
|
memcpy(def + IV_SIZE + 32, dek + IV_SIZE, 32);
|
|
aes_encrypt_cfb_256(session_rc, def, def + IV_SIZE + 32, 32);
|
|
r = file_put_data(tf, def, sizeof(def));
|
|
}
|
|
else {
|
|
r = file_put_data(ef, apdu.data, apdu.nc);
|
|
}
|
|
if (r != PICOKEY_OK) {
|
|
return SW_MEMORY_FAILURE();
|
|
}
|
|
low_flash_available();
|
|
}
|
|
else {
|
|
delete_file(ef);
|
|
}
|
|
}
|
|
return SW_OK();
|
|
}
|