#include <string.h>
#include <time.h>
#include <stdio.h>
#include <stdlib.h>
 
#ifdef USE_JANSSON
#include <jansson.h>
#endif 
 
 
#include "fips_validation.h"
 
#define NEW_LINE_STR    "#"
#define OP_STR      "GCM "
 
#define PARAM_PREFIX    "["
#define KEYLEN_STR  "Keylen = "
#define IVLEN_STR   "IVlen = "
#define PTLEN_STR   "PTlen = "
#define AADLEN_STR  "AADlen = "
#define TAGLEN_STR  "Taglen = "
 
#define COUNT_STR   "Count = "
#define KEY_STR     "Key = "
#define IV_STR      "IV = "
#define PT_STR      "PT = "
#define CT_STR      "CT = "
#define TAG_STR     "Tag = "
#define AAD_STR     "AAD = "
 
#define OP_ENC_STR  "Encrypt"
#define OP_DEC_STR  "Decrypt"
#define OP_ENC_EXT_STR  "ExtIV"
#define OP_ENC_INT_STR  "IntIV"
 
#define KEYLEN_JSON_STR     "keyLen"
#define IVLEN_JSON_STR      "ivLen"
#define PAYLOADLEN_JSON_STR "payloadLen"
#define AADLEN_JSON_STR     "aadLen"
#define TAGLEN_JSON_STR     "tagLen"
 
#define KEY_JSON_STR    "key"
#define IV_JSON_STR     "iv"
#define PT_JSON_STR     "pt"
#define CT_JSON_STR     "ct"
#define AAD_JSON_STR    "aad"
#define TAG_JSON_STR    "tag"
#define DIR_JSON_STR    "direction"
 
#define OP_ENC_JSON_STR "encrypt"
#define OP_DEC_JSON_STR "decrypt"
 
#define IVGEN_JSON_STR  "ivGen"
#define OP_ENC_EXT_JSON_STR "external"
#define OP_ENC_INT_JSON_STR "internal"
 
#define NEG_TEST_STR    "FAIL"
 
static int
parser_read_gcm_pt_len(const char *key, char *src,
{
    int ret = parser_read_uint32_bit_val(key, src, &vec.pt);
 
    if (ret < 0)
        return ret;
 
    if (info.algo == FIPS_TEST_ALGO_AES_GMAC && vec.pt.len == 0) {
        info.interim_info.gcm_data.is_gmac = 1;
        test_ops.prepare_sym_op = prepare_auth_op;
        test_ops.prepare_sym_xform = prepare_gmac_xform;
    } else {
        info.interim_info.gcm_data.is_gmac = 0;
        test_ops.prepare_sym_op = prepare_aead_op;
        test_ops.prepare_sym_xform = prepare_gcm_xform;
    }
 
    return ret;
}
 
static int
parse_gcm_aad_str(const char *key, char *src,
{
    
    if (info.interim_info.gcm_data.is_gmac) {
        vec.pt.len = vec.aead.aad.len;
        return parse_uint8_known_len_hex_str(key, src, &vec.pt);
    } else 
        return parse_uint8_known_len_hex_str(key, src, &vec.aead.aad);
}
 
static int
parse_gcm_pt_ct_str(const char *key, char *src, struct fips_val *val)
{
    
    if (info.interim_info.gcm_data.gen_iv && vec.iv.len) {
        uint32_t i;
 
        if (!vec.iv.val) {
            if (!vec.iv.val)
                return -ENOMEM;
        }
 
        for (i = 0; i < vec.iv.len; i++) {
            int random = rand();
            vec.iv.val[i] = (uint8_t)random;
        }
    }
 
    
    if (info.interim_info.gcm_data.is_gmac)
        return 0;
 
    return parse_uint8_known_len_hex_str(key, src, val);
}
 
struct fips_test_callback gcm_dec_vectors[] = {
        {KEY_STR, parse_uint8_known_len_hex_str, &vec.aead.key},
        {IV_STR, parse_uint8_known_len_hex_str, &vec.iv},
        {CT_STR, parse_gcm_pt_ct_str, &vec.ct},
        {AAD_STR, parse_gcm_aad_str, &vec.aead.aad},
        {TAG_STR, parse_uint8_known_len_hex_str,
                &vec.aead.digest},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback gcm_interim_vectors[] = {
        {KEYLEN_STR, parser_read_uint32_bit_val, &vec.aead.key},
        {IVLEN_STR, parser_read_uint32_bit_val, &vec.iv},
        {PTLEN_STR, parser_read_gcm_pt_len, &vec.pt},
        {PTLEN_STR, parser_read_uint32_bit_val, &vec.ct},
        {AADLEN_STR, parser_read_uint32_bit_val, &vec.aead.aad},
        {TAGLEN_STR, parser_read_uint32_bit_val,
                &vec.aead.digest},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback gcm_enc_vectors[] = {
        {KEY_STR, parse_uint8_known_len_hex_str, &vec.aead.key},
        {IV_STR, parse_uint8_known_len_hex_str, &vec.iv},
        {PT_STR, parse_gcm_pt_ct_str, &vec.pt},
        {AAD_STR, parse_gcm_aad_str, &vec.aead.aad},
        {NULL, NULL, NULL} 
};
 
#ifdef USE_JANSSON
struct fips_test_callback gcm_dec_json_vectors[] = {
        {KEY_JSON_STR, parse_uint8_known_len_hex_str, &vec.aead.key},
        {IV_JSON_STR, parse_uint8_known_len_hex_str, &vec.iv},
        {CT_JSON_STR, parse_gcm_pt_ct_str, &vec.ct},
        {AAD_JSON_STR, parse_gcm_aad_str, &vec.aead.aad},
        {TAG_JSON_STR, parse_uint8_known_len_hex_str,
                &vec.aead.digest},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback gcm_interim_json_vectors[] = {
        {KEYLEN_JSON_STR, parser_read_uint32_bit_val, &vec.aead.key},
        {IVLEN_JSON_STR, parser_read_uint32_bit_val, &vec.iv},
        {PAYLOADLEN_JSON_STR, parser_read_gcm_pt_len, &vec.pt},
        {PAYLOADLEN_JSON_STR, parser_read_uint32_bit_val, &vec.ct},
        {AADLEN_JSON_STR, parser_read_uint32_bit_val, &vec.aead.aad},
        {TAGLEN_JSON_STR, parser_read_uint32_bit_val,
                &vec.aead.digest},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback gcm_enc_json_vectors[] = {
        {KEY_JSON_STR, parse_uint8_known_len_hex_str, &vec.aead.key},
        {IV_JSON_STR, parse_uint8_known_len_hex_str, &vec.iv},
        {PT_JSON_STR, parse_gcm_pt_ct_str, &vec.pt},
        {AAD_JSON_STR, parse_gcm_aad_str, &vec.aead.aad},
        {NULL, NULL, NULL} 
};
#endif 
 
static int
parse_test_gcm_writeback(struct fips_val *val)
{
    struct fips_val tmp_val;
 
    if (info.op == FIPS_TEST_ENC_AUTH_GEN) {
        
        if (info.interim_info.gcm_data.gen_iv) {
            fprintf(info.fp_wr, "%s", IV_STR);
            tmp_val.val = vec.iv.val;
            tmp_val.len = vec.iv.len;
 
            parse_write_hex_str(&tmp_val);
            vec.iv.val = NULL;
        }
 
        fprintf(info.fp_wr, "%s", CT_STR);
 
        if (!info.interim_info.gcm_data.is_gmac) {
            tmp_val.val = val->val;
            tmp_val.len = vec.pt.len;
 
            parse_write_hex_str(&tmp_val);
        } else
            fprintf(info.fp_wr, "\n");
 
        fprintf(info.fp_wr, "%s", TAG_STR);
 
        tmp_val.val = val->val + vec.pt.len;
        tmp_val.len = val->len - vec.pt.len;
 
        parse_write_hex_str(&tmp_val);
    } else {
            fprintf(info.fp_wr, "%s", PT_STR);
            if (!info.interim_info.gcm_data.is_gmac) {
                tmp_val.val = val->val;
                tmp_val.len = vec.pt.len;
 
                parse_write_hex_str(&tmp_val);
            } else
                fprintf(info.fp_wr, "\n");
        } else
            fprintf(info.fp_wr, "%s\n", NEG_TEST_STR);
    }
 
    return 0;
}
 
int
parse_test_gcm_init(void)
{
    char *tmp;
    uint32_t i;
 
    for (i = 0; i < info.nb_vec_lines; i++) {
        char *line = info.vec[i];
 
        tmp = strstr(line, OP_STR);
        if (tmp) {
            if (strstr(line, OP_ENC_STR)) {
                info.op = FIPS_TEST_ENC_AUTH_GEN;
                info.callbacks = gcm_enc_vectors;
                if (strstr(info.file_name, OP_ENC_INT_STR))
                    info.interim_info.gcm_data.gen_iv = 1;
            } else if (strstr(line, OP_DEC_STR)) {
                info.op = FIPS_TEST_DEC_AUTH_VERIF;
                info.callbacks = gcm_dec_vectors;
            } else
                return -EINVAL;
        }
    }
 
    info.interim_callbacks = gcm_interim_vectors;
    info.parse_writeback = parse_test_gcm_writeback;
 
    return 0;
}
 
#ifdef USE_JANSSON
static int
parse_test_gcm_json_writeback(struct fips_val *val)
{
    struct fips_val tmp_val;
    json_t *tcId, *tag;
 
    tcId = json_object_get(json_info.json_test_case, "tcId");
 
    json_info.json_write_case = json_object();
    json_object_set(json_info.json_write_case, "tcId", tcId);
 
    if (info.op == FIPS_TEST_ENC_AUTH_GEN) {
        json_t *ct;
        if (!info.interim_info.gcm_data.is_gmac) {
            tmp_val.val = val->val;
            tmp_val.len = vec.pt.len;
 
            info.one_line_text[0] = '\0';
            writeback_hex_str("", info.one_line_text, &tmp_val);
            ct = json_string(info.one_line_text);
            json_object_set_new(json_info.json_write_case, CT_JSON_STR, ct);
        }
 
        if (info.interim_info.gcm_data.gen_iv) {
            json_t *iv;
            tmp_val.val = vec.iv.val;
            tmp_val.len = vec.iv.len;
 
            writeback_hex_str("", info.one_line_text, &tmp_val);
            iv = json_string(info.one_line_text);
            json_object_set_new(json_info.json_write_case, IV_JSON_STR, iv);
 
            vec.iv.val = NULL;
        }
 
        tmp_val.val = val->val + vec.pt.len;
        tmp_val.len = val->len - vec.pt.len;
 
        writeback_hex_str("", info.one_line_text, &tmp_val);
        tag = json_string(info.one_line_text);
        json_object_set_new(json_info.json_write_case, TAG_JSON_STR, tag);
    } else {
            if (!info.interim_info.gcm_data.is_gmac) {
                tmp_val.val = val->val;
                tmp_val.len = vec.pt.len;
 
                info.one_line_text[0] = '\0';
                writeback_hex_str("", info.one_line_text, &tmp_val);
                json_object_set_new(json_info.json_write_case, PT_JSON_STR,
                    json_string(info.one_line_text));
            } else {
                json_object_set_new(json_info.json_write_case, "testPassed",
                    json_true());
            }
        } else {
            json_object_set_new(json_info.json_write_case, "testPassed",
                json_false());
        }
    }
 
    return 0;
}
 
int
parse_test_gcm_json_init(void)
{
    json_t *direction_obj;
    const char *direction_str;
 
    direction_obj = json_object_get(json_info.json_test_group, DIR_JSON_STR);
    direction_str = json_string_value(direction_obj);
    info.interim_info.gcm_data.gen_iv = 0;
 
    if (strcmp(direction_str, OP_ENC_JSON_STR) == 0) {
        json_t *ivGen_obj = json_object_get(json_info.json_test_group, IVGEN_JSON_STR);
        const char *ivGen_str = json_string_value(ivGen_obj);
 
        info.op = FIPS_TEST_ENC_AUTH_GEN;
        info.callbacks = gcm_enc_json_vectors;
 
        if (strcmp(ivGen_str, OP_ENC_INT_JSON_STR) == 0)
            info.interim_info.gcm_data.gen_iv = 1;
    } else if (strcmp(direction_str, OP_DEC_JSON_STR) == 0) {
        info.op = FIPS_TEST_DEC_AUTH_VERIF;
        info.callbacks = gcm_dec_json_vectors;
    } else {
        return -EINVAL;
    }
    info.interim_callbacks = gcm_interim_json_vectors;
    info.parse_writeback = parse_test_gcm_json_writeback;
 
    return 0;
}
 
#endif 
@ RTE_CRYPTO_OP_STATUS_SUCCESS
 
void * rte_malloc(const char *type, size_t size, unsigned align) __rte_alloc_size(2) __rte_alloc_align(3) __rte_malloc __rte_dealloc_free