/* roca.c - ROCA (CVE-2017-15361) fingerprint checker.
 * Written by Rob Stradling (based on https://github.com/crocs-muni/roca/blob/master/roca/detect.py)
 * Copyright (C) 2017-2018 Sectigo Limited
 * modified 2018 iceman  (dropped openssl bignum,  now use mbedtls lib)
 * modified 2018 merlok
 *
 * 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, either version 2 of the License, or
 * (at your option) any later version.
 *
 * 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/>.
 */
//-----------------------------------------------------------------------------
// EMV roca commands
//-----------------------------------------------------------------------------

#include "emv_roca.h"

#include "ui.h"  // Print...
#include "mbedtls/bignum.h"

static uint8_t g_primes[ROCA_PRINTS_LENGTH] = {
    11, 13, 17, 19, 37, 53, 61, 71, 73, 79, 97, 103, 107, 109, 127, 151, 157
};

mbedtls_mpi g_prints[ROCA_PRINTS_LENGTH];

static void rocacheck_init(void) {

    for (int i = 0; i < ROCA_PRINTS_LENGTH; i++)
        mbedtls_mpi_init(&g_prints[i]);

    mbedtls_mpi_read_string(&g_prints[0], 10, "1026");
    mbedtls_mpi_read_string(&g_prints[1], 10, "5658");
    mbedtls_mpi_read_string(&g_prints[2], 10, "107286");
    mbedtls_mpi_read_string(&g_prints[3], 10, "199410");
    mbedtls_mpi_read_string(&g_prints[4], 10, "67109890");
    mbedtls_mpi_read_string(&g_prints[5], 10, "5310023542746834");
    mbedtls_mpi_read_string(&g_prints[6], 10, "1455791217086302986");
    mbedtls_mpi_read_string(&g_prints[7], 10, "20052041432995567486");
    mbedtls_mpi_read_string(&g_prints[8], 10, "6041388139249378920330");
    mbedtls_mpi_read_string(&g_prints[9], 10, "207530445072488465666");
    mbedtls_mpi_read_string(&g_prints[10], 10, "79228162521181866724264247298");
    mbedtls_mpi_read_string(&g_prints[11], 10, "1760368345969468176824550810518");
    mbedtls_mpi_read_string(&g_prints[12], 10, "50079290986288516948354744811034");
    mbedtls_mpi_read_string(&g_prints[13], 10, "473022961816146413042658758988474");
    mbedtls_mpi_read_string(&g_prints[14], 10, "144390480366845522447407333004847678774");
    mbedtls_mpi_read_string(&g_prints[15], 10, "1800793591454480341970779146165214289059119882");
    mbedtls_mpi_read_string(&g_prints[16], 10, "126304807362733370595828809000324029340048915994");
}

static void rocacheck_cleanup(void) {
    for (int i = 0; i < ROCA_PRINTS_LENGTH; i++)
        mbedtls_mpi_free(&g_prints[i]);
}

static int bitand_is_zero(mbedtls_mpi *a, mbedtls_mpi *b) {

    for (int i = 0; i < mbedtls_mpi_bitlen(a); i++) {

        if (mbedtls_mpi_get_bit(a, i) && mbedtls_mpi_get_bit(b, i))
            return 0;
    }
    return 1;
}


static mbedtls_mpi_uint mpi_get_uint(const mbedtls_mpi *X) {

    if (X->n == 1 && X->s > 0) {
        return X->p[0];
    }
    printf("ZERRRRO!!!\n");
    return 0;
}

/*
static void print_mpi(const char *msg, int radix, const mbedtls_mpi *X) {

    char Xchar[400] = {0};
    size_t len = 0;

    mbedtls_mpi_write_string(X, radix, Xchar, sizeof(Xchar), &len);
    printf("%s[%zu] %s\n", msg, len, Xchar);
}
*/
bool emv_rocacheck(const unsigned char *buf, size_t buflen, bool verbose) {

    mbedtls_mpi t_modulus;
    mbedtls_mpi_init(&t_modulus);

    bool ret = false;

    rocacheck_init();

    MBEDTLS_MPI_CHK(mbedtls_mpi_read_binary(&t_modulus, buf, buflen));

    for (int i = 0; i < ROCA_PRINTS_LENGTH; i++) {

        mbedtls_mpi t_temp;
        mbedtls_mpi t_prime;
        mbedtls_mpi g_one;

        mbedtls_mpi_init(&t_temp);
        mbedtls_mpi_init(&t_prime);
        mbedtls_mpi_init(&g_one);

        MBEDTLS_MPI_CHK(mbedtls_mpi_read_string(&g_one, 10, "1"));

        MBEDTLS_MPI_CHK(mbedtls_mpi_add_int(&t_prime, &t_prime, g_primes[i]));

        MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&t_temp, &t_modulus, &t_prime));

        MBEDTLS_MPI_CHK(mbedtls_mpi_shift_l(&g_one, mpi_get_uint(&t_temp)));

        if (bitand_is_zero(&g_one, &g_prints[i])) {
            if (verbose)
                PrintAndLogEx(FAILED, "No fingerprint found.\n");
            goto cleanup;
        }

        mbedtls_mpi_free(&g_one);
        mbedtls_mpi_free(&t_temp);
        mbedtls_mpi_free(&t_prime);
    }

    ret = true;
    if (verbose)
        PrintAndLogEx(SUCCESS, "Fingerprint found!\n");

cleanup:
    mbedtls_mpi_free(&t_modulus);

    rocacheck_cleanup();
    return ret;
}

int roca_self_test(void) {
    int ret = 0;

    PrintAndLogEx(NORMAL, "");
    PrintAndLogEx(INFO, "ROCA check vulnerability tests");

    // positive
    uint8_t keyp[] = "\x94\x4e\x13\x20\x8a\x28\x0c\x37\xef\xc3\x1c\x31\x14\x48\x5e\x59"\
                     "\x01\x92\xad\xbb\x8e\x11\xc8\x7c\xad\x60\xcd\xef\x00\x37\xce\x99"\
                     "\x27\x83\x30\xd3\xf4\x71\xa2\x53\x8f\xa6\x67\x80\x2e\xd2\xa3\xc4"\
                     "\x4a\x8b\x7d\xea\x82\x6e\x88\x8d\x0a\xa3\x41\xfd\x66\x4f\x7f\xa7";


    if (emv_rocacheck(keyp, 64, false)) {
        PrintAndLogEx(SUCCESS, "Weak modulus   [ %s]", _GREEN_("PASS"));
    } else {
        ret++;
        PrintAndLogEx(FAILED, "Weak modulus   [ %s]", _RED_("Fail"));
    }

    // negative
    uint8_t keyn[] = "\x84\x4e\x13\x20\x8a\x28\x0c\x37\xef\xc3\x1c\x31\x14\x48\x5e\x59"\
                     "\x01\x92\xad\xbb\x8e\x11\xc8\x7c\xad\x60\xcd\xef\x00\x37\xce\x99"\
                     "\x27\x83\x30\xd3\xf4\x71\xa2\x53\x8f\xa6\x67\x80\x2e\xd2\xa3\xc4"\
                     "\x4a\x8b\x7d\xea\x82\x6e\x88\x8d\x0a\xa3\x41\xfd\x66\x4f\x7f\xa7";

    if (emv_rocacheck(keyn, 64, false)) {
        ret++;
        PrintAndLogEx(FAILED, "Strong modulus [ %s]", _RED_("Fail"));
    } else {
        PrintAndLogEx(SUCCESS, "Strong modulus [ %s]", _GREEN_("PASS"));
    }

    return ret;
}