/*

 * PKCS15 emulation layer for EstEID card issued from December 2018.

 *

 * Copyright (C) 2019, Martin Paljak <martin@martinpaljak.net>

 *

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2.1 of the License, or (at your option) any later version.

 *

 * This library 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

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this library; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <ctype.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include "common/compat_strlcpy.h"

#include "internal.h"

#include "opensc.h"

#include "pkcs15.h"

static int

is_latvian_eid(sc_pkcs15_card_t *p15card)

{

  return p15card->card->type == SC_CARD_TYPE_LATEID_2018 ||

         p15card->card->type == SC_CARD_TYPE_LATEID_2018_V2_2025;

}

static int sc_pkcs15emu_esteid2018_init(sc_pkcs15_card_t *p15card) {

  sc_card_t *card = p15card->card;

  u8 buff[11];

  int r, i, cert_slot = 0;

  size_t field_length = 0, taglen, buflen;

  sc_path_t tmppath;

  /* Read documber number to be used as serial */

  sc_format_path("3F00D003", &tmppath);

  LOG_TEST_RET(card->ctx, sc_select_file(card, &tmppath, NULL), "SELECT docnr");

  r = sc_read_binary(card, 0, buff, 11, 0);

  LOG_TEST_RET(card->ctx, r, "read document number failed");

  const u8 *tag = sc_asn1_find_tag(card->ctx, buff, (size_t)r, 0x04, &taglen);

  if (tag == NULL)

    LOG_FUNC_RETURN(card->ctx, SC_ERROR_INTERNAL);

  for (size_t j = 0; j < taglen; j++)

    if (!isalnum(tag[j]))

      LOG_FUNC_RETURN(card->ctx, SC_ERROR_INTERNAL);

  free(p15card->tokeninfo->serial_number);

  p15card->tokeninfo->serial_number = malloc(taglen + 1);

  if (!p15card->tokeninfo->serial_number)

    LOG_FUNC_RETURN(card->ctx, SC_ERROR_OUT_OF_MEMORY);

  set_string(&p15card->tokeninfo->label, "ID-kaart");

  set_string(&p15card->tokeninfo->manufacturer_id, "IDEMIA");

  p15card->tokeninfo->serial_number = memcpy(p15card->tokeninfo->serial_number, tag, taglen);

  p15card->tokeninfo->serial_number[taglen] = '\0';

  p15card->tokeninfo->flags = SC_PKCS15_TOKEN_READONLY;

  if (is_latvian_eid(p15card)) {

    u8 *buf;

    const u8 *ptr;

    sc_pkcs15_object_t obj = {0};

    sc_pkcs15_cert_info_t *cert_info = NULL;

    if (!p15card->file_app) {

      p15card->file_app = sc_file_new();

    }

    if (!p15card->file_app) {

      LOG_FUNC_RETURN(card->ctx, SC_ERROR_OUT_OF_MEMORY);

    }

    sc_format_path("3F00adf1", &p15card->file_app->path);

    sc_format_path("3F00adf17005", &tmppath);

    r = sc_pkcs15_read_file(p15card, &tmppath, &buf, &buflen, 0);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Reading EF.CDF file failed");

    ptr = buf;

    r = sc_pkcs15_decode_cdf_entry(p15card, &obj, &ptr, &buflen);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Decoding EF.CDF file failed");

    cert_info = (struct sc_pkcs15_cert_info *)obj.data;

    if (cert_info && cert_info->path.len > 0) {

      cert_slot = cert_info->path.value[cert_info->path.len - 1] - 1;

      if (cert_slot != 0 && cert_slot != 1) {

        /* unknown slot, see `cert_paths` below */

        r = SC_ERROR_UNKNOWN_DATA_RECEIVED;

        LOG_TEST_GOTO_ERR(card->ctx, r, "Unknown certificate path");

      }

    }

    sc_pkcs15_free_cert_info(cert_info);

    sc_file_free(p15card->file_app);

    p15card->file_app = NULL;

    free(buf);

  }

  /* add certificates */

  for (i = 0; i < 2; i++) {

    static const char *esteid_cert_names[2] = {"Isikutuvastus", "Allkirjastamine"};

    static const char *cert_paths[2][2] = {

        {"3f00:adf1:3401", "3f00:adf2:341f"},

        {"3f00:adf1:3402", "3f00:adf2:341e"}

        };

    static const u8 esteid_cert_ids[2] = {1, 2};

    struct sc_pkcs15_cert_info cert_info = {

        .id = {.len = 1, .value[0] = esteid_cert_ids[i]}

    };

    struct sc_pkcs15_object cert_obj = {0};

    strlcpy(cert_obj.label, esteid_cert_names[i], sizeof(cert_obj.label));

    sc_format_path(cert_paths[cert_slot][i], &cert_info.path);

    r = sc_pkcs15emu_add_x509_cert(p15card, &cert_obj, &cert_info);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Could not add cert oebjct");

    // Read data from first cert

    if (i != 0)

      continue;

    sc_pkcs15_cert_t *cert = NULL;

    r = sc_pkcs15_read_certificate(p15card, &cert_info, 0, &cert);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Could not read authentication certificate");

    if (cert->key->algorithm == SC_ALGORITHM_EC)

      field_length = cert->key->u.ec.params.field_length;

    static const struct sc_object_id cn_oid = {

        {2, 5, 4, 3, -1}

    };

    u8 *cn_name = NULL;

    size_t cn_len = 0;

    sc_pkcs15_get_name_from_dn(card->ctx, cert->subject, cert->subject_len, &cn_oid, &cn_name, &cn_len);

    sc_pkcs15_free_certificate(cert);

    if (cn_len > 0) {

      char *token_name = (char *)realloc(cn_name, cn_len + 1);

      if (token_name) {

        token_name[cn_len] = '\0';

        free(p15card->tokeninfo->label);

        p15card->tokeninfo->label = token_name;

      } else

        free(cn_name);

    }

  }

  /* add pins */

  static const u8 pin_authid[3] = {1, 2, 3};

  for (i = 0; i < 3; i++) {

    static const char *esteid_pin_names[3] = {"PIN1", "PIN2", "PUK"};

    static const size_t pin_min[2][3] = {

        {4, 5, 8}, // Estonian

        {4, 6, 8}, // Latvian

    };

    static const int esteid_pin_ref[3] = {0x01, 0x85, 0x02};

    static const char *esteid_pin_path[3] = {"3F00", "3F00ADF2", "3F00"};

    static const unsigned int esteid_pin_flags[3] = {

        SC_PKCS15_PIN_FLAG_NEEDS_PADDING | SC_PKCS15_PIN_TYPE_FLAGS_PIN_GLOBAL,

        SC_PKCS15_PIN_FLAG_NEEDS_PADDING | SC_PKCS15_PIN_TYPE_FLAGS_PIN_LOCAL,

        SC_PKCS15_PIN_FLAG_NEEDS_PADDING | SC_PKCS15_PIN_TYPE_FLAGS_PUK_GLOBAL};

    struct sc_pkcs15_auth_info pin_info = {

        .auth_id = {.len = 1, .value[0] = pin_authid[i]},

        .auth_type = SC_PKCS15_PIN_AUTH_TYPE_PIN,

        .attrs = {

            .pin = {

                .reference = esteid_pin_ref[i],

                .flags = esteid_pin_flags[i],

                .type = SC_PKCS15_PIN_TYPE_ASCII_NUMERIC,

                .min_length = pin_min[is_latvian_eid(p15card)][i],

                .stored_length = 12,

                .max_length = 12,

                .pad_char = 0xFF}},

        .tries_left = 3,

        .max_tries = 3

          };

    struct sc_pkcs15_object pin_obj = {

        .flags = esteid_pin_flags[i]};

    sc_format_path(esteid_pin_path[i], &pin_info.path);

    strlcpy(pin_obj.label, esteid_pin_names[i], sizeof(pin_obj.label));

    /* Link normal PINs with PUK */

    if (i < 2) {

      pin_obj.auth_id.len = 1;

      pin_obj.auth_id.value[0] = 3;

    }

    r = sc_pkcs15emu_add_pin_obj(p15card, &pin_obj, &pin_info);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Could not add pin object");

  }

  // trigger PIN counter refresh via pin_cmd

  struct sc_pkcs15_object *objs[3];

  r = sc_pkcs15_get_objects(p15card, SC_PKCS15_TYPE_AUTH, objs, 3);

  if (r != 3) {

    sc_log(card->ctx, "Can not get auth objects");

    goto err;

  }

  for (i = 0; i < r; i++) {

    r = sc_pkcs15_get_pin_info(p15card, objs[i]);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Could not get pin object");

  }

  /* add private keys */

  for (i = 0; i < 2; i++) {

    static const u8 prkey_id[2] = {1, 2};

    static const char *prkey_name[2] = {"Isikutuvastus", "Allkirjastamine"};

    static const char *prkey_path[2] = {"3F00:ADF1", "3F00:ADF2"};

    static const unsigned int prkey_usage[2] = {SC_PKCS15_PRKEY_USAGE_SIGN | SC_PKCS15_PRKEY_USAGE_DERIVE,

        SC_PKCS15_PRKEY_USAGE_NONREPUDIATION};

    static const int prkey_consent[2] = {0, 1};

    static const u8 prkey_ref[2][2] = {

        {0x81, 0x9F}, // Slot 1

        {0x82, 0x9E}, // Slot 2

    };

    struct sc_pkcs15_prkey_info prkey_info = {

        .id = {.len = 1, .value[0] = prkey_id[i]},

        .native = 1,

        .key_reference = prkey_ref[is_latvian_eid(p15card)][i],

        .field_length = field_length,

        .usage = prkey_usage[i]

           };

    struct sc_pkcs15_object prkey_obj = {

        .auth_id = {.len = 1, .value[0] = pin_authid[i]},

        .user_consent = prkey_consent[i],

        .flags = SC_PKCS15_CO_FLAG_PRIVATE

      };

    sc_format_path(prkey_path[i], &prkey_info.path);

    strlcpy(prkey_obj.label, prkey_name[i], sizeof(prkey_obj.label));

    r = sc_pkcs15emu_add_ec_prkey(p15card, &prkey_obj, &prkey_info);

    LOG_TEST_GOTO_ERR(card->ctx, r, "Could not add private key object");

  }

  return SC_SUCCESS;

err:

  sc_pkcs15_card_clear(p15card);

  LOG_FUNC_RETURN(card->ctx, SC_ERROR_INTERNAL);

}

int sc_pkcs15emu_esteid2018_init_ex(sc_pkcs15_card_t *p15card, struct sc_aid *aid) {

  if (p15card->card->type == SC_CARD_TYPE_ESTEID_2018 ||

      p15card->card->type == SC_CARD_TYPE_ESTEID_2018_V2_2025 ||

      is_latvian_eid(p15card))

    return sc_pkcs15emu_esteid2018_init(p15card);

  return SC_ERROR_WRONG_CARD;

}