Add string.h

[proxmark3-svn] / client / crypto / asn1utils.c

//-----------------------------------------------------------------------------
// Copyright (C) 2018 Merlok
//
// This code is licensed to you under the terms of the GNU GPL, version 2 or,
// at your option, any later version. See the LICENSE.txt file for the text of
// the license.
//-----------------------------------------------------------------------------
// asn.1 utils
//-----------------------------------------------------------------------------

#include "asn1utils.h"
#include <ctype.h>
#include <stdlib.h>
#include <mbedtls/asn1.h>
#include "emv/tlv.h"
#include "emv/dump.h"
#include "asn1dump.h"
#include "util.h"
#include "ui.h" // PrintAndLog

int ecdsa_asn1_get_signature(uint8_t *signature, size_t signaturelen, uint8_t *rval, uint8_t *sval) {
        if (!signature || !signaturelen || !rval || !sval)
                return 1;

        int res = 0;
        unsigned char *p = signature;
        const unsigned char *end = p + signaturelen;
        size_t len;
        mbedtls_mpi xmpi;

        if ((res = mbedtls_asn1_get_tag(&p, end, &len, MBEDTLS_ASN1_CONSTRUCTED | MBEDTLS_ASN1_SEQUENCE)) == 0) {
                mbedtls_mpi_init(&xmpi);
                res = mbedtls_asn1_get_mpi(&p, end, &xmpi);
                if (res) {
                        mbedtls_mpi_free(&xmpi);
                        goto exit;
                }
                
                res = mbedtls_mpi_write_binary(&xmpi, rval, 32);
                mbedtls_mpi_free(&xmpi);
                if (res) 
                        goto exit;

                mbedtls_mpi_init(&xmpi);
                res = mbedtls_asn1_get_mpi(&p, end, &xmpi);
                if (res) {
                        mbedtls_mpi_free(&xmpi);
                        goto exit;
                }
                
                res = mbedtls_mpi_write_binary(&xmpi, sval, 32);
                mbedtls_mpi_free(&xmpi);
                if (res) 
                        goto exit;

                // check size
                if (end != p)
                        return 2;
                }

exit:
        return res;
}

static bool print_cb(void *data, const struct tlv *tlv, int level, bool is_leaf) {
        bool candump = true;
        asn1_tag_dump(tlv, stdout, level, &candump);
        if (is_leaf && candump) {
                dump_buffer(tlv->value, tlv->len, stdout, level);
        }

        return true;
}

int asn1_print(uint8_t *asn1buf, size_t asn1buflen, char *indent) {
        
        struct tlvdb *t = NULL;
        t = tlvdb_parse_multi(asn1buf, asn1buflen);
        if (t) {
                tlvdb_visit(t, print_cb, NULL, 0);
                tlvdb_free(t);
        } else {
                PrintAndLogEx(ERR, "Can't parse data as TLV tree.");
                return 1;
        }
        
        return 0;
}