/*
 * This file is part of the Pico HSM distribution (https://github.com/polhenarejos/pico-hsm).
 * 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 "sc_hsm.h"
#include "asn1.h"
#include "oid.h"
#include "eac.h"
#include "files.h"
#include "cvc.h"

file_t *ef_puk_aut = NULL;

int cmd_mse() {
    int p1 = P1(apdu);
    int p2 = P2(apdu);
    if (p2 != 0xA4 && p2 != 0xA6 && p2 != 0xAA && p2 != 0xB4 && p2 != 0xB6 && p2 != 0xB8)
        return SW_INCORRECT_P1P2();
    if (p1 & 0x1) { //SET
        uint16_t tag = 0x0;
        uint8_t *tag_data = NULL, *p = NULL;
        size_t tag_len = 0;
        while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) {
            if (tag == 0x80) {
                if (p2 == 0xA4) {
                    if (tag_len == 10 && memcmp(tag_data, OID_ID_CA_ECDH_AES_CBC_CMAC_128, tag_len) == 0)
                        sm_set_protocol(MSE_AES);
                }
            }
            else if (tag == 0x83) {
                if (tag_len == 1) {

                }
                else {
                    if (p2 == 0xB6) {
                        if (puk_store_select_chr(tag_data) == CCID_OK)
                            return SW_OK();
                    }
                    else if (p2 == 0xA4) { /* Aut */
                        for (int i = 0; i < MAX_PUK; i++) {
                            file_t *ef = search_dynamic_file(EF_PUK+i);
                            if (!ef)
                                break;
                            if (ef->data == NULL || file_get_size(ef) == 0)
                                break;
                            size_t chr_len = 0;
                            const uint8_t *chr = cvc_get_chr(file_get_data(ef), file_get_size(ef), &chr_len);
                            if (memcmp(chr, tag_data, chr_len) == 0) {
                                ef_puk_aut = ef;
                                return SW_OK();
                            }
                        }
                    }
                    return SW_REFERENCE_NOT_FOUND();
                }
            }
        }
    }
    else
        return SW_INCORRECT_P1P2();
    return SW_OK();
}