/* 
 * 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 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 
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License 
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "common.h"
#include "openpgp.h"
#include "version.h"
#include "files.h"
#include "random.h"
#include "eac.h"
#include "crypto_utils.h"
#include "mbedtls/rsa.h"
#include "mbedtls/ecdsa.h"
#include "mbedtls/ecdh.h"

bool has_pw1 = false;
bool has_pw3 = false;
uint8_t session_pw3[32];

uint8_t openpgp_aid[] = {
    6, 
    0xD2,0x76,0x00,0x01,0x24,0x01,
};

uint8_t openpgp_aid_full[] = {
    16,00, 
    0xD2,0x76,0x00,0x01,0x24,0x01,
    OPGP_VERSION_MAJOR,OPGP_VERSION_MINOR,0xff,0xfe,0xff,0xff,0xff,0xff,0x00,0x00
};

char atr_openpgp[] = { 
    21,
    0x3b,0xda,0x18,0xff,0x81,0xb1,0xfe,0x75,0x1f,0x03,0x00,0x31,0xf5,0x73,0xc0,0x01,0x60,0x00,0x90,0x00,0x1c
};

int openpgp_process_apdu();

void select_file(file_t *pe) {
    if (!pe)
    {
        currentDF = (file_t *)MF;
        currentEF = NULL;
    }
    else if (pe->type & FILE_TYPE_INTERNAL_EF) {
        currentEF = pe;
        currentDF = &file_entries[pe->parent];
    }
    else {
        currentDF = pe;
    }
    if (currentEF == file_openpgp) {
        selected_applet = currentEF;
        //sc_hsm_unload(); //reset auth status
    }
}

static int cmd_select() {
    uint8_t p1 = P1(apdu);
    uint8_t p2 = P2(apdu);
    file_t *pe = NULL;
    uint16_t fid = 0x0;
    
    // Only "first or only occurence" supported 
    //if ((p2 & 0xF3) != 0x00) {
    //    return SW_INCORRECT_P1P2();
    //}
    
    if (apdu.cmd_apdu_data_len >= 2)
        fid = get_uint16_t(apdu.cmd_apdu_data, 0);
        
    //if ((fid & 0xff00) == (KEY_PREFIX << 8))
    //    fid = (PRKD_PREFIX << 8) | (fid & 0xff);
    
    if (!pe) {
        if (p1 == 0x0) { //Select MF, DF or EF - File identifier or absent
            if (apdu.cmd_apdu_data_len == 0) {
            	pe = (file_t *)MF;
            	//ac_fini();
            }
            else if (apdu.cmd_apdu_data_len == 2) {
                if (!(pe = search_by_fid(fid, NULL, SPECIFY_ANY))) {
                    return SW_FILE_NOT_FOUND();
                }
            }
        }
        else if (p1 == 0x01) { //Select child DF - DF identifier
            if (!(pe = search_by_fid(fid, currentDF, SPECIFY_DF))) {
                return SW_FILE_NOT_FOUND();
            }
        }
        else if (p1 == 0x02) { //Select EF under the current DF - EF identifier
            if (!(pe = search_by_fid(fid, currentDF, SPECIFY_EF))) {
                return SW_FILE_NOT_FOUND();
            }
        }
        else if (p1 == 0x03) { //Select parent DF of the current DF - Absent
            if (apdu.cmd_apdu_data_len != 0)
                return SW_FILE_NOT_FOUND();
        }
        else if (p1 == 0x04) { //Select by DF name - e.g., [truncated] application identifier
            if (!(pe = search_by_name(apdu.cmd_apdu_data, apdu.cmd_apdu_data_len))) {
                return SW_FILE_NOT_FOUND();
            }
            if (card_terminated) {
                return set_res_sw (0x62, 0x85);
            }        
        }
        else if (p1 == 0x08) { //Select from the MF - Path without the MF identifier
            if (!(pe = search_by_path(apdu.cmd_apdu_data, apdu.cmd_apdu_data_len, MF))) {
                return SW_FILE_NOT_FOUND();
            }
        }
        else if (p1 == 0x09) { //Select from the current DF - Path without the current DF identifier
            if (!(pe = search_by_path(apdu.cmd_apdu_data, apdu.cmd_apdu_data_len, currentDF))) {
                return SW_FILE_NOT_FOUND();
            }
        }
    }
    if ((p2 & 0xfc) == 0x00 || (p2 & 0xfc) == 0x04) {
        if ((p2 & 0xfc) == 0x04)
            process_fci(pe);
    }
    else
        return SW_INCORRECT_P1P2();
    select_file(pe);
    return SW_OK ();
}

void scan_files() {
    scan_flash();
    file_t *ef;
    if ((ef = search_by_fid(EF_FULL_AID, NULL, SPECIFY_ANY))) {
        ef->data = openpgp_aid_full;
        pico_get_unique_board_id_string(ef->data+12, 4);
    }
    if ((ef = search_by_fid(EF_PW1, NULL, SPECIFY_ANY))) {
        if (!ef->data) {
            TU_LOG1("PW1 is empty. Initializing with default password\r\n");
            const uint8_t def[6] = { 0x31,0x32,0x33,0x34,0x35,0x36 };
            uint8_t dhash[33];
            dhash[0] = sizeof(def);
            double_hash_pin(def, sizeof(def), dhash+1);
            flash_write_data_to_file(ef, dhash, sizeof(dhash));
            
            ef = search_by_fid(EF_PW1_RETRIES, NULL, SPECIFY_ANY);
            if (ef && !ef->data) {
                const uint8_t retries = 3;
                flash_write_data_to_file(ef, &retries, sizeof(retries));
            }
        }
    }
    if ((ef = search_by_fid(EF_RC, NULL, SPECIFY_ANY))) {
        if (!ef->data) {
            TU_LOG1("RC is empty. Initializing with default password\r\n");
            
            const uint8_t def[8] = { 0x31,0x32,0x33,0x34,0x35,0x36,0x37,0x38 };
            uint8_t dhash[33];
            dhash[0] = sizeof(def);
            double_hash_pin(def, sizeof(def), dhash+1);
            flash_write_data_to_file(ef, dhash, sizeof(dhash));
            
            ef = search_by_fid(EF_RC_RETRIES, NULL, SPECIFY_ANY);
            if (ef && !ef->data) {
                const uint8_t retries = 3;
                flash_write_data_to_file(ef, &retries, sizeof(retries));
            }
        }
    }
    if ((ef = search_by_fid(EF_PW3, NULL, SPECIFY_ANY))) {
        if (!ef->data) {
            TU_LOG1("PW3 is empty. Initializing with default password\r\n");
            
            const uint8_t def[8] = { 0x31,0x32,0x33,0x34,0x35,0x36,0x37,0x38 };
            uint8_t dhash[33];
            dhash[0] = sizeof(def);
            double_hash_pin(def, sizeof(def), dhash+1);
            flash_write_data_to_file(ef, dhash, sizeof(dhash));
            
            ef = search_by_fid(EF_PW3_RETRIES, NULL, SPECIFY_ANY);
            if (ef && !ef->data) {
                const uint8_t retries = 3;
                flash_write_data_to_file(ef, &retries, sizeof(retries));
            }
        }
    }
    if ((ef = search_by_fid(EF_SIG_COUNT, NULL, SPECIFY_ANY))) {
        if (!ef->data) {
            TU_LOG1("SigCount is empty. Initializing to zero\r\n");
            const uint8_t def[3] = { 0 };
            flash_write_data_to_file(ef, def, sizeof(def));
        }
    }
    low_flash_available();
}

void init_openpgp() {
    isUserAuthenticated = false;
    has_pw1 = has_pw3 = false;
    scan_files();
    //cmd_select();
}

int openpgp_unload() {
    isUserAuthenticated = false;
    has_pw1 = has_pw3 = false;
    return CCID_OK;
}

app_t *openpgp_select_aid(app_t *a) {
    if (!memcmp(apdu.cmd_apdu_data, openpgp_aid+1, openpgp_aid[0])) {
        a->aid = openpgp_aid;
        a->process_apdu = openpgp_process_apdu;
        a->unload = openpgp_unload;
        init_openpgp();
        return a;
    }
    return NULL;
}

void __attribute__ ((constructor)) openpgp_ctor() { 
    ccid_atr = atr_openpgp;
    register_app(openpgp_select_aid);
}

int parse_do(uint16_t *fids, int mode) {
    int len = 0;
    file_t *ef;
    for (int i = 0; i < fids[0]; i++) {
        if ((ef = search_by_fid(fids[i+1], NULL, SPECIFY_EF))) {
            uint16_t data_len;
            if ((ef->type & FILE_DATA_FUNC) == FILE_DATA_FUNC) {
                data_len = ((int (*)(const file_t *, int))(ef->data))((const file_t *)ef, mode);
            }
            else {
                if (ef->data)
                    data_len = file_read_uint16(ef->data);
                else
                    data_len = 0;
                if (mode == 1) {
                    if (fids[0] > 1) {
                        if (fids[i+1] < 0x0100) {
                            res_APDU[res_APDU_size++] = fids[i+1] & 0xff;
                        }
                        else {
                            res_APDU[res_APDU_size++] = fids[i+1] >> 8;
                            res_APDU[res_APDU_size++] = fids[i+1] & 0xff;
                        }
                        res_APDU_size += format_tlv_len(data_len, res_APDU+res_APDU_size);
                    }
                    if (ef->data)
                        memcpy(res_APDU+res_APDU_size, file_read(ef->data+2), data_len);
                    res_APDU_size += data_len;
                }
            }
            len += data_len;
        }
    }
    return len;
}

int parse_trium(uint16_t fid, uint8_t num, size_t size) {
    for (uint8_t i = 0; i < num; i++) {
        file_t *ef;
        if ((ef = search_by_fid(fid+i, NULL, SPECIFY_EF)) && ef->data) {
            uint16_t data_len = file_read_uint16(ef->data);
            memcpy(res_APDU+res_APDU_size, file_read(ef->data+2), data_len);
            res_APDU_size += data_len;
        }
        else {
            memset(res_APDU+res_APDU_size, 0, size);
            res_APDU_size += size;
        }
    }
    return num*size;
}

int parse_ch_data(const file_t *f, int mode) {
    uint16_t fids[] = {
        3,
        EF_CH_NAME, EF_LANG_PREF, EF_SEX, 
    };
    res_APDU[res_APDU_size++] = EF_CH_DATA & 0xff;
    res_APDU[res_APDU_size++] = 0x82;
    uint8_t *lp = res_APDU+res_APDU_size;
    res_APDU_size += 2;
    uint16_t data_len = parse_do(fids, mode);
    uint16_t lpdif = res_APDU+res_APDU_size-lp-2;
    *lp++ = lpdif >> 8;
    *lp++ = lpdif & 0xff;
    return lpdif+4;
}

int parse_sec_tpl(const file_t *f, int mode) {
    res_APDU[res_APDU_size++] = EF_SEC_TPL & 0xff;
    res_APDU[res_APDU_size++] = 5;
    file_t *ef = search_by_fid(EF_SIG_COUNT, NULL, SPECIFY_ANY);
    if (ef && ef->data) {
        res_APDU[res_APDU_size++] = EF_SIG_COUNT & 0xff;
        res_APDU[res_APDU_size++] = 3;
        memcpy(res_APDU+res_APDU_size, file_read(ef->data+2), 3);
        res_APDU_size += 3;
    }
    return 5+2;
}

int parse_ch_cert(const file_t *f, int mode) {
    return 0;
}

int parse_fp(const file_t *f, int mode) {
    res_APDU[res_APDU_size++] = EF_FP & 0xff;
    res_APDU[res_APDU_size++] = 60;
    return parse_trium(EF_FP_SIG, 3, 20);
}

int parse_cafp(const file_t *f, int mode) {
    res_APDU[res_APDU_size++] = EF_CA_FP & 0xff;
    res_APDU[res_APDU_size++] = 60;
    return parse_trium(EF_FP_CA1, 3, 20);
}

int parse_ts(const file_t *f, int mode) {
    res_APDU[res_APDU_size++] = EF_TS_ALL & 0xff;
    res_APDU[res_APDU_size++] = 12;
    return parse_trium(EF_TS_SIG, 3, 4);    
}

int parse_keyinfo(const file_t *f, int mode) {
    int init_len = res_APDU_size;
    if (res_APDU_size > 0) {
        res_APDU[res_APDU_size++] = EF_KEY_INFO & 0xff;
        res_APDU[res_APDU_size++] = 6;
    }
    res_APDU[res_APDU_size++] = 0x00;
    res_APDU[res_APDU_size++] = 0x00;
    
    res_APDU[res_APDU_size++] = 0x01;
    res_APDU[res_APDU_size++] = 0x00;
    
    res_APDU[res_APDU_size++] = 0x02;
    res_APDU[res_APDU_size++] = 0x00;
    return res_APDU_size-init_len;
}

int parse_pw_status(const file_t *f, int mode) {
    file_t *ef;
    int init_len = res_APDU_size;
    if (res_APDU_size > 0) {
        res_APDU[res_APDU_size++] = EF_PW_STATUS & 0xff;
        res_APDU[res_APDU_size++] = 7;
    }
    res_APDU[res_APDU_size++] = 0x1;
    res_APDU[res_APDU_size++] = 127;
    res_APDU[res_APDU_size++] = 127;
    res_APDU[res_APDU_size++] = 127;
    ef = search_by_fid(EF_PW1_RETRIES, NULL, SPECIFY_ANY);
    if (ef && ef->data) {
        res_APDU[res_APDU_size++] = file_read_uint8(ef->data+2);
    }
    ef = search_by_fid(EF_RC_RETRIES, NULL, SPECIFY_ANY);
    if (ef && ef->data) {
        res_APDU[res_APDU_size++] = file_read_uint8(ef->data+2);
    }
    ef = search_by_fid(EF_PW3_RETRIES, NULL, SPECIFY_ANY);
    if (ef && ef->data) {
        res_APDU[res_APDU_size++] = file_read_uint8(ef->data+2);
    }
    return res_APDU_size-init_len;
}

#define ALGO_RSA        0x01
#define ALGO_ECDH       0x12
#define ALGO_ECDSA      0x13
#define ALGO_AES        0x70
#define ALGO_AES_128    0x71
#define ALGO_AES_192    0x72
#define ALGO_AES_256    0x74

#define ALGO_RSA_1K     0
#define ALGO_RSA_2k     1
#define ALGO_RSA_3K     2
#define ALGO_RSA_4K     3
#define ALGO_X448       4
#define ALGO_P256K1     5
#define ALGO_P256R1     6
#define ALGO_P384R1     7
#define ALGO_P521R1     8
#define ALGO_BP256R1    9
#define ALGO_BP384R1    10
#define ALGO_BP512R1    11
#define ALGO_CV22519    12

static const uint8_t algorithm_attr_x448[] = {
  4,
  ALGO_ECDH,
  /* OID of X448 */
  0x2b, 0x65, 0x6f
};

static const uint8_t algorithm_attr_rsa1k[] = {
  6,
  ALGO_RSA,
  0x04, 0x00,	      /* Length modulus (in bit): 2048 */
  0x00, 0x20,	      /* Length exponent (in bit): 32  */
  0x00		      /* 0: Acceptable format is: P and Q */
};

static const uint8_t algorithm_attr_rsa2k[] = {
  6,
  ALGO_RSA,
  0x08, 0x00,	      /* Length modulus (in bit): 2048 */
  0x00, 0x20,	      /* Length exponent (in bit): 32  */
  0x00		      /* 0: Acceptable format is: P and Q */
};

static const uint8_t algorithm_attr_rsa3k[] = {
  6,
  ALGO_RSA,
  0x0C, 0x00,	      /* Length modulus (in bit): 2048 */
  0x00, 0x20,	      /* Length exponent (in bit): 32  */
  0x00		      /* 0: Acceptable format is: P and Q */
};

static const uint8_t algorithm_attr_rsa4k[] = {
  6,
  ALGO_RSA,
  0x10, 0x00,	      /* Length modulus (in bit): 4096 */
  0x00, 0x20,	      /* Length exponent (in bit): 32  */
  0x00		      /* 0: Acceptable format is: P and Q */
};

static const uint8_t algorithm_attr_p256k1[] = {
  6,
  ALGO_ECDSA,
  0x2b, 0x81, 0x04, 0x00, 0x0a 
};

static const uint8_t algorithm_attr_p256r1[] = {
  9,
  ALGO_ECDSA,
  0x2A,0x86,0x48,0xCE,0x3D,0x03,0x01,0x07
};

static const uint8_t algorithm_attr_p384r1[] = {
  6,
  ALGO_ECDSA,
  0x2B,0x81,0x04,0x00,0x22
};

static const uint8_t algorithm_attr_p521r1[] = {
  6,
  ALGO_ECDSA,
  0x2B,0x81,0x04,0x00,0x23
};

static const uint8_t algorithm_attr_bp256r1[] = {
  10,
  ALGO_ECDSA,
  0x2B,0x24,0x03,0x03,0x02,0x08,0x01,0x01,0x07
};

static const uint8_t algorithm_attr_bp384r1[] = {
  10,
  ALGO_ECDSA,
  0x2B,0x24,0x03,0x03,0x02,0x08,0x01,0x01,0x0B
};

static const uint8_t algorithm_attr_bp512r1[] = {
  10,
  ALGO_ECDSA,
  0x2B,0x24,0x03,0x03,0x02,0x08,0x01,0x01,0x0D
};

static const uint8_t algorithm_attr_cv25519[] = {
  11,
  ALGO_ECDH,
  0x2b, 0x06, 0x01, 0x04, 0x01, 0x97, 0x55, 0x01, 0x05, 0x01
};

static const uint8_t *algorithm_attr[] = {
    algorithm_attr_rsa1k, algorithm_attr_rsa2k, algorithm_attr_rsa3k, algorithm_attr_rsa4k,
    algorithm_attr_x448, algorithm_attr_p256k1,
    algorithm_attr_p256r1, algorithm_attr_p384r1, algorithm_attr_p521r1,
    algorithm_attr_bp256r1, algorithm_attr_bp384r1, algorithm_attr_bp512r1,
    algorithm_attr_cv25519
};

int parse_algo(const uint8_t *algo, uint16_t tag) {
    res_APDU[res_APDU_size++] = tag & 0xff;
    memcpy(res_APDU+res_APDU_size, algo, algo[0]+1);
    res_APDU_size += algo[0]+1;
    return algo[0]+2;
}

int parse_algoinfo(const file_t *f, int mode) {
    uint8_t datalen = 0;
    if (f->fid == EF_ALGO_INFO) {
        res_APDU[res_APDU_size++] = EF_ALGO_INFO & 0xff;
        uint8_t *lp = res_APDU+res_APDU_size;
        res_APDU_size++;
        datalen += parse_algo(algorithm_attr_rsa2k, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_rsa4k, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_p256k1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_p256r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_p384r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_p521r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_bp256r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_bp384r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_bp512r1, EF_ALGO_SIG);
        
        datalen += parse_algo(algorithm_attr_rsa2k, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_rsa4k, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_p256k1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_p256r1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_p384r1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_p521r1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_bp256r1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_bp384r1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_bp512r1, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_cv25519, EF_ALGO_DEC);
        datalen += parse_algo(algorithm_attr_x448, EF_ALGO_DEC);
        
        datalen += parse_algo(algorithm_attr_rsa2k, EF_ALGO_AUT);
        datalen += parse_algo(algorithm_attr_rsa4k, EF_ALGO_AUT);
        datalen += parse_algo(algorithm_attr_p256k1, EF_ALGO_AUT);
        datalen += parse_algo(algorithm_attr_p256r1, EF_ALGO_AUT);
        datalen += parse_algo(algorithm_attr_p384r1, EF_ALGO_AUT);
        datalen += parse_algo(algorithm_attr_p521r1, EF_ALGO_AUT);
        datalen += parse_algo(algorithm_attr_bp256r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_bp384r1, EF_ALGO_SIG);
        datalen += parse_algo(algorithm_attr_bp512r1, EF_ALGO_SIG);
        *lp = res_APDU+res_APDU_size-lp-1;
        datalen = *lp;
    }
    else if (f->fid == EF_ALGO_SIG || f->fid == EF_ALGO_DEC || f->fid == EF_ALGO_AUT) {
        uint16_t fid = 0x1000 | f->fid;
        file_t *ef;
        if (!(ef = search_by_fid(fid, NULL, SPECIFY_EF)) || !ef->data)
            datalen += parse_algo(algorithm_attr_rsa2k, f->fid);
        else {
            uint16_t len = file_read_uint16(ef->data);
            if (res_APDU_size > 0) {
                res_APDU[res_APDU_size++] = f->fid & 0xff;
                res_APDU[res_APDU_size++] = len & 0xff;
                datalen += 2;
            }
            memcpy(res_APDU+res_APDU_size, file_read(ef->data+2), len);
            res_APDU_size += len;
            datalen += len;
        }
    }
    return datalen;
}

int parse_app_data(const file_t *f, int mode) {
    uint16_t fids[] = {
        6,
        EF_FULL_AID, EF_HIST_BYTES, EF_EXLEN_INFO, EF_GFM, EF_DISCRETE_DO, EF_KEY_INFO
    };
    res_APDU[res_APDU_size++] = EF_APP_DATA & 0xff;
    res_APDU[res_APDU_size++] = 0x82;
    uint8_t *lp = res_APDU+res_APDU_size;
    res_APDU_size += 2;
    uint16_t data_len = parse_do(fids, mode);
    uint16_t lpdif = res_APDU+res_APDU_size-lp-2;
    *lp++ = lpdif >> 8;
    *lp++ = lpdif & 0xff;
    return lpdif+4;
}

int parse_discrete_do(const file_t *f, int mode) {
    uint16_t fids[] = {
        8,
        EF_EXT_CAP, EF_ALGO_SIG, EF_ALGO_DEC, EF_ALGO_AUT, EF_PW_STATUS, EF_FP, EF_CA_FP, EF_TS_ALL, //EF_UIF_SIG, EF_UIF_DEC, EF_UIF_AUT
    };
    res_APDU[res_APDU_size++] = EF_DISCRETE_DO & 0xff;
    res_APDU[res_APDU_size++] = 0x82;
    uint8_t *lp = res_APDU+res_APDU_size;
    res_APDU_size += 2;
    uint16_t data_len = parse_do(fids, mode);
    uint16_t lpdif = res_APDU+res_APDU_size-lp-2;
    *lp++ = lpdif >> 8;
    *lp++ = lpdif & 0xff;
    return lpdif+4;
}

static int cmd_get_data() {
    if (apdu.cmd_apdu_data_len > 0)
        return SW_WRONG_LENGTH();
    uint16_t fid = (P1(apdu) << 8) | P2(apdu);
    file_t *ef;
    if (!(ef = search_by_fid(fid, NULL, SPECIFY_EF)))
        return SW_FILE_NOT_FOUND();
    if (!authenticate_action(ef, ACL_OP_READ_SEARCH)) {
        return SW_SECURITY_STATUS_NOT_SATISFIED();
    }
    if (ef->data) {
        uint16_t fids[] = {1,fid};
        uint16_t data_len = parse_do(fids, 1);
        if (apdu.expected_res_size > data_len)
            apdu.expected_res_size = data_len;
    }
    return SW_OK();
}

int pin_reset_retries(const file_t *pin, bool force) {
    if (!pin)
        return CCID_ERR_NULL_PARAM; 
    const file_t *act = search_by_fid(pin->fid+3, NULL, SPECIFY_EF);
    if (!act)
        return CCID_ERR_FILE_NOT_FOUND;
    uint8_t retries = file_read_uint8(act->data+2);
    if (retries == 0 && force == false) //blocked
        return CCID_ERR_BLOCKED;
    retries = 3;
    int r = flash_write_data_to_file((file_t *)act, &retries, sizeof(retries));
    low_flash_available();
    return r;
}

int pin_wrong_retry(const file_t *pin) {
    if (!pin)
        return CCID_ERR_NULL_PARAM; 
    const file_t *act = search_by_fid(pin->fid+3, NULL, SPECIFY_EF);
    if (!act)
        return CCID_ERR_FILE_NOT_FOUND;
    uint8_t retries = file_read_uint8(act->data+2);
    if (retries > 0) {
        retries -= 1;
        int r = flash_write_data_to_file((file_t *)act, &retries, sizeof(retries));
        if (r != CCID_OK)
            return r;
        low_flash_available();
        if (retries == 0)
            return CCID_ERR_BLOCKED;
        return retries;
    }
    return CCID_ERR_BLOCKED;
}

int check_pin(const file_t *pin, const uint8_t *data, size_t len) {
    if (!pin)
        return SW_REFERENCE_NOT_FOUND();
    if (!pin->data) {
        return SW_REFERENCE_NOT_FOUND();
    }
    isUserAuthenticated = false;
    has_pw1 = has_pw3 = false;

    uint8_t dhash[32];
    double_hash_pin(data, len, dhash);
    if (sizeof(dhash) != file_read_uint16(pin->data)-1) //1 byte for pin len
        return SW_CONDITIONS_NOT_SATISFIED();
    if (memcmp(file_read(pin->data+3), dhash, sizeof(dhash)) != 0) {
        int retries;
        if ((retries = pin_wrong_retry(pin)) < CCID_OK)
            return SW_PIN_BLOCKED();
        return set_res_sw(0x63, 0xc0 | retries);
    }

    int r = pin_reset_retries(pin, false);
    if (r == CCID_ERR_BLOCKED)
        return SW_PIN_BLOCKED();
    if (r != CCID_OK)
        return SW_MEMORY_FAILURE();
    isUserAuthenticated = true;
    if (pin->fid == EF_PW1)
        has_pw1 = true;
    else if (pin->fid == EF_PW3) {
        has_pw3 = true;
        hash_multi(data, len, session_pw3);
    }
    return SW_OK();
}

static int cmd_verify() {
    uint8_t p1 = P1(apdu);
    uint8_t p2 = P2(apdu);
    
    if (p1 != 0x0 || (p2 & 0x60) != 0x0)
        return SW_WRONG_P1P2();
    uint8_t qualifier = p2&0x1f;
    uint16_t fid = 0x1000 | p2;
    if (fid == EF_RC && apdu.cmd_apdu_data_len > 0)
        fid = EF_PW1;
    file_t *pw, *retries;
    if (!(pw = search_by_fid(fid, NULL, SPECIFY_EF)))
        return SW_REFERENCE_NOT_FOUND();
    if (!(retries = search_by_fid(fid+3, NULL, SPECIFY_EF)))
        return SW_REFERENCE_NOT_FOUND();
    if (file_read_uint8(pw->data+2) == 0) //not initialized
        return SW_REFERENCE_NOT_FOUND();
    if (apdu.cmd_apdu_data_len > 0) {
        return check_pin(pw, apdu.cmd_apdu_data, apdu.cmd_apdu_data_len);
    }
    if (file_read_uint8(retries->data+2) == 0)
        return SW_PIN_BLOCKED();
    return set_res_sw(0x63, 0xc0 | file_read_uint8(retries->data+2));
}

static 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_FILE_NOT_FOUND();
    if (!authenticate_action(ef, ACL_OP_UPDATE_ERASE)) {
        return SW_SECURITY_STATUS_NOT_SATISFIED();
    }
    if (apdu.cmd_apdu_data_len > 0 && (ef->type & FILE_DATA_FLASH)) {
        int r = flash_write_data_to_file(ef, apdu.cmd_apdu_data, apdu.cmd_apdu_data_len);
        if (r != CCID_OK)
            return SW_MEMORY_FAILURE();
        low_flash_available();
    }
    return SW_OK();
}

static int cmd_change_pin() {
    if (P1(apdu) != 0x0)
        return SW_WRONG_P1P2();
    uint16_t fid = 0x1000 | P2(apdu);
    file_t *pw;
    if (!(pw = search_by_fid(fid, NULL, SPECIFY_EF)))
        return SW_REFERENCE_NOT_FOUND();
    uint8_t pin_len = file_read_uint8(pw->data+2);
    uint16_t r = check_pin(pw, apdu.cmd_apdu_data, pin_len);
    if (r != 0x9000)
        return r;
    uint8_t dhash[33];
    dhash[0] = apdu.cmd_apdu_data_len-pin_len;
    double_hash_pin(apdu.cmd_apdu_data+pin_len, apdu.cmd_apdu_data_len-pin_len, dhash+1);
    flash_write_data_to_file(pw, dhash, sizeof(dhash));
    low_flash_available();
    return SW_OK();
}

static int cmd_reset_retry() {
    if (P2(apdu) != 0x81)
        return SW_REFERENCE_NOT_FOUND();
    if (P1(apdu) == 0x0 || P1(apdu) == 0x2) {
        int newpin_len = 0;
        file_t *pw = NULL;
        if (!(pw = search_by_fid(EF_PW1, NULL, SPECIFY_EF)))
            return SW_REFERENCE_NOT_FOUND();
        if (P1(apdu) == 0x0) {
            file_t *rc, *pw;
            if (!(rc = search_by_fid(EF_RC, NULL, SPECIFY_EF)))
                return SW_REFERENCE_NOT_FOUND();
            uint8_t pin_len = file_read_uint8(rc->data+2);
            if (apdu.cmd_apdu_data_len <= pin_len)
                return SW_WRONG_LENGTH();
            uint16_t r = check_pin(rc, apdu.cmd_apdu_data, pin_len);
            if (r != 0x9000)
                return r;
            newpin_len = apdu.cmd_apdu_data_len-pin_len;
        }
        else if (P1(apdu) == 0x2) {    
            if (!has_pw3)
                return SW_CONDITIONS_NOT_SATISFIED();
            newpin_len = apdu.cmd_apdu_data_len;
        }
        uint8_t dhash[33];
        dhash[0] = newpin_len;
        double_hash_pin(apdu.cmd_apdu_data+(apdu.cmd_apdu_data_len-newpin_len), newpin_len, dhash+1);
        flash_write_data_to_file(pw, dhash, sizeof(dhash));
        if (pin_reset_retries(pw, true) != CCID_OK)
            return SW_MEMORY_FAILURE();
        low_flash_available();
        return SW_OK();
    }
    return SW_INCORRECT_P1P2();
}

int store_keys(void *key_ctx, int type, uint16_t key_id) {
    int r, key_size;
    uint8_t kdata[4096/8]; //worst 
    
    //if (!has_pw3)
    //    return CCID_NO_LOGIN;
    //file_t *pw3 = search_by_fid(EF_PW3, NULL, SPECIFY_EF);
    //if (!pw3)
    //    return CCID_ERR_FILE_NOT_FOUND;
    file_t *ef = search_by_fid(key_id, NULL, SPECIFY_EF);
    if (!ef)
        return CCID_ERR_FILE_NOT_FOUND;
    if (type == ALGO_RSA) {
        mbedtls_rsa_context *rsa = (mbedtls_rsa_context *)key_ctx;
        key_size = mbedtls_mpi_size(&rsa->P)+mbedtls_mpi_size(&rsa->Q);
        mbedtls_mpi_write_binary(&rsa->P, kdata, key_size/2);
        mbedtls_mpi_write_binary(&rsa->Q, kdata+key_size/2, key_size/2);
    }
    else if (type == ALGO_ECDSA || type == ALGO_ECDH) {
        mbedtls_ecdsa_context *ecdsa = (mbedtls_ecdsa_context *)key_ctx;
        key_size = mbedtls_mpi_size(&ecdsa->d);
        kdata[0] = ecdsa->grp.id & 0xff;
        mbedtls_mpi_write_binary(&ecdsa->d, kdata+1, key_size);
        key_size++;
    }
    else if (type & ALGO_AES) {
        if (type == ALGO_AES_128)
            key_size = 16;
        else if (type == ALGO_AES_192)
            key_size = 24;
        else if (type == ALGO_AES_256)
            key_size = 32;
        memcpy(kdata, key_ctx, key_size);
    }
    //r = aes_encrypt_cfb_256(file_read(pw3->data+2), session_pw3, kdata, key_size);
    //if (r != CCID_OK)
    //    return r;
    r = flash_write_data_to_file(ef, kdata, key_size);
    if (r != CCID_OK)
        return r;
    low_flash_available();
    return CCID_OK;
}

int load_private_key_rsa(mbedtls_rsa_context *ctx, file_t *fkey) {
    //wait_button();
    int key_size = file_read_uint16(fkey->data);
    uint8_t kdata[4096/8];
    //if (!has_pw3)
    //    return CCID_NO_LOGIN;
    //file_t *pw3 = search_by_fid(EF_PW3, NULL, SPECIFY_EF);
    //if (!pw3)
    //    return CCID_ERR_FILE_NOT_FOUND;
    memcpy(kdata, file_read(fkey->data+2), key_size);
    //int r = aes_decrypt_cfb_256(file_read(pw3->data+2), session_pw3, kdata, key_size);
    //if (r != CCID_OK)
    //    return r;
    if (mbedtls_mpi_read_binary(&ctx->P, kdata, key_size/2) != 0) {
        mbedtls_rsa_free(ctx);
        return CCID_WRONG_DATA;
    }
    if (mbedtls_mpi_read_binary(&ctx->Q, kdata+key_size/2, key_size/2) != 0) {
        mbedtls_rsa_free(ctx);
        return CCID_WRONG_DATA;
    }
    if (mbedtls_mpi_lset(&ctx->E, 0x10001) != 0) {
        mbedtls_rsa_free(ctx);
        return CCID_EXEC_ERROR;
    }
    if (mbedtls_rsa_import(ctx, NULL, &ctx->P, &ctx->Q, NULL, &ctx->E) != 0) {
        mbedtls_rsa_free(ctx);
        return CCID_WRONG_DATA;
    }
    if (mbedtls_rsa_complete(ctx) != 0) {
        mbedtls_rsa_free(ctx);
        return CCID_WRONG_DATA;
    }
    if (mbedtls_rsa_check_privkey(ctx) != 0) {
        mbedtls_rsa_free(ctx);
        return CCID_WRONG_DATA;
    }
    return CCID_OK;
}

int load_private_key_ecdsa(mbedtls_ecdsa_context *ctx, file_t *fkey) {
    //wait_button();
    int key_size = file_read_uint16(fkey->data);
    uint8_t kdata[67]; //Worst case, 521 bit + 1byte
    memcpy(kdata, file_read(fkey->data+2), key_size);
    mbedtls_ecp_group_id gid = kdata[0];
    int r = mbedtls_ecp_read_key(gid, ctx, kdata+1, key_size-1);
    if (r != 0) {
        mbedtls_ecdsa_free(ctx);
        return CCID_EXEC_ERROR;
    }
    return CCID_OK;
}

mbedtls_ecp_group_id get_ec_group_id_from_attr(const uint8_t *algo, size_t algo_len) {
    if (memcmp(algorithm_attr_p256k1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_SECP256K1;
    else if (memcmp(algorithm_attr_p256r1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_SECP256R1;
    else if (memcmp(algorithm_attr_p384r1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_SECP384R1;
    else if (memcmp(algorithm_attr_p521r1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_SECP521R1;
    else if (memcmp(algorithm_attr_bp256r1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_BP256R1;
    else if (memcmp(algorithm_attr_bp384r1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_BP384R1;
    else if (memcmp(algorithm_attr_bp512r1+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_BP512R1;
    else if (memcmp(algorithm_attr_cv25519+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_CURVE25519;
    else if (memcmp(algorithm_attr_x448+2, algo, algo_len) == 0)
        return MBEDTLS_ECP_DP_CURVE448;
    return MBEDTLS_ECP_DP_NONE;
}

void make_rsa_response(mbedtls_rsa_context *rsa) {
    memcpy(res_APDU, "\x7f\x49\x82\x00\x00", 5);
    res_APDU_size = 5;
    res_APDU[res_APDU_size++] = 0x81;
    res_APDU[res_APDU_size++] = 0x82;
    put_uint16_t(mbedtls_mpi_size(&rsa->N), res_APDU+res_APDU_size); res_APDU_size += 2;
    mbedtls_mpi_write_binary(&rsa->N, res_APDU+res_APDU_size, mbedtls_mpi_size(&rsa->N)); res_APDU_size += mbedtls_mpi_size(&rsa->N);
    res_APDU[res_APDU_size++] = 0x82;
    res_APDU[res_APDU_size++] = mbedtls_mpi_size(&rsa->E) & 0xff;
    mbedtls_mpi_write_binary(&rsa->E, res_APDU+res_APDU_size, mbedtls_mpi_size(&rsa->E)); res_APDU_size += mbedtls_mpi_size(&rsa->E);
    put_uint16_t(res_APDU_size-5, res_APDU+3);
}

void make_ecdsa_response(mbedtls_ecdsa_context *ecdsa) {
    size_t plen = mbedtls_mpi_size(&ecdsa->grp.P);
    memcpy(res_APDU, "\x7f\x49\x81\x00", 4);
    res_APDU_size = 4;
    res_APDU[res_APDU_size++] = 0x86;
    res_APDU[res_APDU_size++] = 0x81;
    res_APDU[res_APDU_size++] = 2*plen+1;
    res_APDU[res_APDU_size++] = 0x04;
    mbedtls_mpi_write_binary(&ecdsa->Q.X, res_APDU+res_APDU_size, plen); res_APDU_size += plen;
    mbedtls_mpi_write_binary(&ecdsa->Q.Y, res_APDU+res_APDU_size, plen); res_APDU_size += plen;
    res_APDU[3] = res_APDU_size-4;
}

static int cmd_keypair_gen() {
    if (P2(apdu) != 0x0)
        return SW_INCORRECT_P1P2();
    if (apdu.cmd_apdu_data_len != 2 && apdu.cmd_apdu_data_len != 5)
        return SW_WRONG_LENGTH();
    
    uint16_t fid = 0x0;
    int r = CCID_OK;
    if (apdu.cmd_apdu_data[0] == 0xB6)
        fid = EF_PK_SIG;
    else if (apdu.cmd_apdu_data[0] == 0xB8)
        fid = EF_PK_DEC;
    else if (apdu.cmd_apdu_data[0] == 0xA4)
        fid = EF_PK_AUT;
    else
        return SW_WRONG_DATA();
    
    file_t *algo_ef = search_by_fid(fid-0x0010, NULL, SPECIFY_EF);
    if (!algo_ef)
        return SW_FILE_NOT_FOUND();
    const uint8_t *algo = algorithm_attr_rsa2k+1;
    uint16_t algo_len = algorithm_attr_rsa2k[0];
    if (algo_ef && algo_ef->data) {
        algo = file_read(algo_ef->data+2);
        algo_len = file_read_uint16(algo_ef->data);
    }
    if (P1(apdu) == 0x80) { //generate
        if (algo[0] == ALGO_RSA) {
            int exponent = 65537, nlen = (algo[1] << 8) | algo[2];
            printf("KEYPAIR RSA %d\r\n",nlen);
            //if (nlen != 2048 && nlen != 4096)
            //    return SW_FUNC_NOT_SUPPORTED();
            mbedtls_rsa_context rsa;
            mbedtls_rsa_init(&rsa);
            uint8_t index = 0;
            r = mbedtls_rsa_gen_key(&rsa, random_gen, &index, nlen, exponent);
            printf("r %d\r\n",r);
            if (r != 0) {
                mbedtls_rsa_free(&rsa);
                return SW_EXEC_ERROR();
            }
            r = store_keys(&rsa, ALGO_RSA, fid);
            printf("r %d\r\n",r);
            make_rsa_response(&rsa);
            mbedtls_rsa_free(&rsa);
            if (r != CCID_OK)
                return SW_EXEC_ERROR();
        }
        else if (algo[0] == ALGO_ECDH || algo[0] == ALGO_ECDSA) {
            printf("KEYPAIR ECDSA\r\n");
            mbedtls_ecp_group_id gid = get_ec_group_id_from_attr(algo+1, algo_len-1);
            if (gid == MBEDTLS_ECP_DP_NONE)
                return SW_FUNC_NOT_SUPPORTED();
            mbedtls_ecdsa_context ecdsa;
            mbedtls_ecdsa_init(&ecdsa);
            uint8_t index = 0;
            r = mbedtls_ecdsa_genkey(&ecdsa, gid, random_gen, &index);
            if (r != 0) {
                mbedtls_ecdsa_free(&ecdsa);
                return SW_EXEC_ERROR();
            }
            r = store_keys(&ecdsa, algo[0], fid);
            make_ecdsa_response(&ecdsa);
            mbedtls_ecdsa_free(&ecdsa);
            if (r != CCID_OK)
                return SW_EXEC_ERROR();
        }
        else
            return SW_FUNC_NOT_SUPPORTED();
        file_t *pbef = search_by_fid(fid+3, NULL, SPECIFY_EF);
        if (!pbef)
            return SW_FILE_NOT_FOUND();
        r = flash_write_data_to_file(pbef, res_APDU, res_APDU_size);
        if (r != CCID_OK)
            return SW_EXEC_ERROR();
        low_flash_available();
        return SW_OK();
    }
    else if (P1(apdu) == 0x81) { //read
        file_t *ef = search_by_fid(fid+3, NULL, SPECIFY_EF);
        if (!ef || !ef->data)
            return SW_FILE_NOT_FOUND();
        res_APDU_size = file_read_uint16(ef->data);
        memcpy(res_APDU, file_read(ef->data+2), res_APDU_size);
        return SW_OK();
    }
    return SW_INCORRECT_P1P2();
}

static int cmd_pso() {
    uint16_t algo_fid = 0x0, pk_fid = 0x0;
    if (P1(apdu) == 0x9E && P2(apdu) == 0x9A) {
        algo_fid = EF_ALGO_PRIV1;
        pk_fid = EF_PK_SIG;
    }
    else if (P1(apdu) == 0x80 && P2(apdu) == 0x86) {
        algo_fid = EF_ALGO_PRIV2;
        pk_fid = EF_PK_DEC;
    }
    else
        return SW_INCORRECT_P1P2();
    file_t *algo_ef = search_by_fid(algo_fid, NULL, SPECIFY_EF);
    if (!algo_ef)
        return SW_FILE_NOT_FOUND();
    const uint8_t *algo = algorithm_attr_rsa2k+1;
    uint16_t algo_len = algorithm_attr_rsa2k[0];
    if (algo_ef && algo_ef->data) {
        algo_len = file_read_uint16(algo_ef->data);
        algo = file_read(algo_ef->data+2);
    }
    file_t *ef = search_by_fid(pk_fid, NULL, SPECIFY_EF);
    if (!ef)
        return SW_FILE_NOT_FOUND();
    int r = CCID_OK;
    int key_size = file_read_uint16(ef->data);
    if (algo[0] == ALGO_RSA) {
        mbedtls_rsa_context ctx;
        mbedtls_rsa_init(&ctx);
        r = load_private_key_rsa(&ctx, ef);
        if (r != CCID_OK)
            return SW_EXEC_ERROR();
        if (P1(apdu) == 0x9E && P2(apdu) == 0x9A) {
            mbedtls_md_type_t md = MBEDTLS_MD_NONE;
            if (memcmp(apdu.cmd_apdu_data, "\x30\x51\x30\x0D\x06\x09\x60\x86\x48\x01\x65\x03\x04\x02\x01\x05\x00",17))
                md = MBEDTLS_MD_SHA256;
            else if (memcmp(apdu.cmd_apdu_data, "\x30\x51\x30\x0D\x06\x09\x60\x86\x48\x01\x65\x03\x04\x02\x02\x05\x00",17))
                md = MBEDTLS_MD_SHA384;
            else if (memcmp(apdu.cmd_apdu_data, "\x30\x51\x30\x0D\x06\x09\x60\x86\x48\x01\x65\x03\x04\x02\x03\x05\x00",17))
                md = MBEDTLS_MD_SHA512;
            uint8_t hash_len = apdu.cmd_apdu_data[18];
            if (md == MBEDTLS_MD_SHA256 && hash_len != 32)
                return SW_WRONG_DATA();
            else if (md == MBEDTLS_MD_SHA384 && hash_len != 48)
                return SW_WRONG_DATA();
            else if (md == MBEDTLS_MD_SHA512 && hash_len != 64)
                return SW_WRONG_DATA();
            const uint8_t *hash = apdu.cmd_apdu_data+19;
            uint8_t *signature = calloc( 1, ctx.len );
            r = mbedtls_rsa_pkcs1_sign(&ctx, random_gen, NULL, md, hash_len, hash, signature);
            memcpy(res_APDU, signature, key_size);
            free(signature);
            mbedtls_rsa_free(&ctx);
            if (r != 0) {
                return SW_EXEC_ERROR();
            }
            res_APDU_size = key_size;
            //apdu.expected_res_size = key_size;
        }
        else if (P1(apdu) == 0x80 && P2(apdu) == 0x86) {
            if (apdu.cmd_apdu_data_len < key_size) //needs padding
                memset(apdu.cmd_apdu_data+apdu.cmd_apdu_data_len, 0, key_size-apdu.cmd_apdu_data_len);
            size_t olen = 0;
            r = mbedtls_rsa_pkcs1_decrypt(&ctx, random_gen, NULL, &olen, apdu.cmd_apdu_data+1, res_APDU, key_size);
            if (r != 0) {
                mbedtls_rsa_free(&ctx);
                return SW_EXEC_ERROR();
            }
            res_APDU_size = olen;
            mbedtls_rsa_free(&ctx);
        }
    }
    else if (algo[0] == ALGO_ECDH || algo[0] == ALGO_ECDSA) {
        if (P1(apdu) == 0x9E && P2(apdu) == 0x9A) {
            mbedtls_ecdsa_context ctx;
            mbedtls_ecdsa_init(&ctx);
            mbedtls_md_type_t md = MBEDTLS_MD_SHA256;
            if (apdu.cmd_apdu_data_len == 32)
                md = MBEDTLS_MD_SHA256;
            else if (apdu.cmd_apdu_data_len == 20)
                md = MBEDTLS_MD_SHA1;
            else if (apdu.cmd_apdu_data_len == 28)
                md = MBEDTLS_MD_SHA224;
            else if (apdu.cmd_apdu_data_len == 48)
                md = MBEDTLS_MD_SHA384;
            else if (apdu.cmd_apdu_data_len == 64)
                md = MBEDTLS_MD_SHA512;
            r = load_private_key_ecdsa(&ctx, ef);
            if (r != CCID_OK)
                return SW_CONDITIONS_NOT_SATISFIED();
            size_t olen = 0;
            uint8_t buf[MBEDTLS_ECDSA_MAX_LEN];
            mbedtls_mpi ri, si;
            mbedtls_mpi_init(&ri);
            mbedtls_mpi_init(&si);
            r = mbedtls_ecdsa_sign(&ctx.grp, &ri, &si, &ctx.d, apdu.cmd_apdu_data, apdu.cmd_apdu_data_len, random_gen, NULL);
            if (r != 0) {
                mbedtls_mpi_free(&ri);
                mbedtls_mpi_free(&si);
                mbedtls_ecdsa_free(&ctx);
                return SW_EXEC_ERROR();
            }
            mbedtls_mpi_write_binary(&ri, res_APDU, mbedtls_mpi_size(&ri)); 
            res_APDU_size = mbedtls_mpi_size(&ri);
            mbedtls_mpi_write_binary(&si, res_APDU+res_APDU_size, mbedtls_mpi_size(&si)); 
            res_APDU_size += mbedtls_mpi_size(&si);
            mbedtls_mpi_free(&ri);
            mbedtls_mpi_free(&si);
            mbedtls_ecdsa_free(&ctx);
        }
        else if (P1(apdu) == 0x80 && P2(apdu) == 0x86) {
            mbedtls_ecdh_context ctx;
            uint8_t kdata[67];
            uint8_t *data = apdu.cmd_apdu_data;
            if (*data++ != 0xA6)
                return SW_WRONG_DATA();
            if (*data == 0x82) data += 3;
            else if (*data == 0x81) data += 2;
            else data++;
            if (*data != 0x7f || *(data+1) != 0x49)
                return SW_WRONG_DATA();
            data += 2;
            if (*data == 0x82) data += 3;
            else if (*data == 0x81) data += 2;
            else data++;
            if (*data++ != 0x86)
                return SW_WRONG_DATA();
            uint16_t dlen = 0;
            if (*data == 0x82) { dlen = (*(data+1) << 8) | *(data+2); data += 3; }
            else if (*data == 0x81) { dlen = *(data+1); data += 2; }
            else { dlen = *data++; }
            if (dlen != 2*key_size-1)
                return SW_WRONG_LENGTH();             
            memcpy(kdata, file_read(ef->data+2), key_size);
            mbedtls_ecdh_init(&ctx);
            mbedtls_ecp_group_id gid = kdata[0];
            r = mbedtls_ecdh_setup(&ctx, gid);
            if (r != 0) {
                mbedtls_ecdh_free(&ctx);
                return SW_DATA_INVALID();
            }
            r = mbedtls_mpi_read_binary(&ctx.ctx.mbed_ecdh.d, kdata+1, key_size-1);
            if (r != 0) {
                mbedtls_ecdh_free(&ctx);
                return SW_DATA_INVALID();
            }
            r = mbedtls_ecdh_read_public(&ctx, data-1, dlen+1);
            if (r != 0) {
                mbedtls_ecdh_free(&ctx);
                return SW_DATA_INVALID();
            }
            size_t olen = 0;
            res_APDU[0] = 0x04;
            r = mbedtls_ecdh_calc_secret(&ctx, &olen, res_APDU+1, MBEDTLS_ECP_MAX_BYTES, random_gen, NULL);
            if (r != 0) {
                mbedtls_ecdh_free(&ctx);
                return SW_EXEC_ERROR();
            }
            res_APDU_size = olen+1;
            mbedtls_ecdh_free(&ctx);
        }
    }
    return SW_OK();
}

static int cmd_terminate_df() {
    if (P1(apdu) != 0x0 && P2(apdu) != 0x0)
        return SW_INCORRECT_P1P2();
    if (apdu.cmd_apdu_data_len != 0)
        return SW_WRONG_LENGTH();
    initialize_flash(true);
    scan_files();
    return SW_OK();
}

static int cmd_activate_file() {
    return SW_OK();
}

static int cmd_challenge() {
    uint8_t *rb = (uint8_t *)random_bytes_get(apdu.expected_res_size);
    if (!rb)
        return SW_WRONG_LENGTH();
    memcpy(res_APDU, rb, apdu.expected_res_size);
    res_APDU_size = apdu.expected_res_size;
    return SW_OK();
}

typedef struct cmd
{
  uint8_t ins;
  int (*cmd_handler)();
} cmd_t;

#define INS_VERIFY          0x20
#define INS_CHANGE_PIN      0x24
#define INS_PSO             0x2A
#define INS_RESET_RETRY     0x2C
#define INS_ACTIVATE_FILE   0x44
#define INS_KEYPAIR_GEN     0x47
#define INS_CHALLENGE       0x84
#define INS_SELECT          0xA4
#define INS_GET_DATA        0xCA
#define INS_PUT_DATA        0xDA
#define INS_TERMINATE_DF    0xE6

static const cmd_t cmds[] = {
    { INS_GET_DATA, cmd_get_data },
    { INS_SELECT, cmd_select },
    { INS_VERIFY, cmd_verify },
    { INS_PUT_DATA, cmd_put_data },
    { INS_CHANGE_PIN, cmd_change_pin },
    { INS_RESET_RETRY, cmd_reset_retry },
    { INS_KEYPAIR_GEN, cmd_keypair_gen },
    { INS_PSO, cmd_pso },
    { INS_TERMINATE_DF, cmd_terminate_df },
    { INS_ACTIVATE_FILE, cmd_activate_file },
    { INS_CHALLENGE, cmd_challenge },
    { 0x00, 0x0}
};

int openpgp_process_apdu() {
    int r = sm_unwrap();
    for (const cmd_t *cmd = cmds; cmd->ins != 0x00; cmd++) {
        if (cmd->ins == INS(apdu)) {
            int r = cmd->cmd_handler();
            sm_wrap();
            return r;
        }
    }
    return SW_INS_NOT_SUPPORTED();
}