#include <string.h>
#include <stdio.h>
#include <time.h>
 
 
#include "fips_validation.h"
 
#define MODE_STR    "XTS"
#define ALGO_STR    "test data for "
#define OP_STR      "State"
#define KEY_SIZE_STR    "Key Length : "
 
#define COUNT_STR   "COUNT = "
#define KEY_STR     "Key = "
#define IV_STR      "i = "
#define PT_STR      "PT = "
#define CT_STR      "CT = "
 
#define OP_ENC_STR  "ENCRYPT"
#define OP_DEC_STR  "DECRYPT"
 
#define ALGO_JSON_STR       "algorithm"
#define TESTTYPE_JSON_STR   "testType"
#define DIR_JSON_STR        "direction"
#define KEYLEN_JSON_STR     "keyLen"
#define TWEAKMODE_JSON_STR  "tweakMode"
 
#define KEY_JSON_STR        "key"
#define DATAUNITLEN_JSON_STR    "dataUnitLen"
#define PAYLOADLEN_JSON_STR "payloadLen"
#define TWEAKVALUE_JSON_STR "tweakValue"
#define SEQNUMBER_JSON_STR  "sequenceNumber"
#define PT_JSON_STR "pt"
#define CT_JSON_STR "ct"
 
#define OP_ENC_JSON_STR "encrypt"
#define OP_DEC_JSON_STR "decrypt"
 
static int
parse_interim_xts_enc_dec(const char *key,
{
    if (strcmp(key, OP_ENC_STR) == 0)
        info.op = FIPS_TEST_ENC_AUTH_GEN;
    else if (strcmp(key, OP_DEC_STR) == 0)
        info.op = FIPS_TEST_DEC_AUTH_VERIF;
    else
        return -1;
    return 0;
}
 
struct fips_test_callback xts_tests_vectors[] = {
        {KEY_STR, parse_uint8_hex_str, &vec.cipher_auth.key},
        {IV_STR, parse_uint8_hex_str, &vec.iv},
        {PT_STR, parse_uint8_hex_str, &vec.pt},
        {CT_STR, parse_uint8_hex_str, &vec.ct},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback xts_tests_interim_vectors[] = {
        {OP_ENC_STR, parse_interim_xts_enc_dec, NULL},
        {OP_DEC_STR, parse_interim_xts_enc_dec, NULL},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback xts_writeback_callbacks[] = {
        {COUNT_STR, NULL, NULL},
        {IV_STR, writeback_hex_str, &vec.iv},
        {KEY_STR, writeback_hex_str, &vec.cipher_auth.key},
        {PT_STR, writeback_hex_str, &vec.pt},
        {CT_STR, writeback_hex_str, &vec.ct},
        {NULL, NULL, NULL} 
};
 
#ifdef USE_JANSSON
static int
parser_xts_read_keylen(const char *key, char *src, struct fips_val *val)
{
    int ret;
 
    ret = parser_read_uint32_bit_val(key, src, val);
    if (ret < 0)
        return ret;
 
    val->len *= 2;
    return 0;
}
 
static int
parser_xts_read_tweakval(const char *key, char *src, struct fips_val *val)
{
    char num_str[4] = {0};
    int ret;
 
    if (info.interim_info.xts_data.tweak_mode == XTS_TWEAK_MODE_HEX) {
        ret = parse_uint8_hex_str(key, src, val);
    } else if (info.interim_info.xts_data.tweak_mode == XTS_TWEAK_MODE_NUMBER) {
        snprintf(num_str, 
RTE_DIM(num_str), 
"%x", atoi(src));
 
        ret = parse_uint8_hex_str(key, num_str, val);
    } else {
        ret = -1;
    }
 
    return ret;
}
 
struct fips_test_callback xts_dec_json_vectors[] = {
        {KEY_JSON_STR, parse_uint8_known_len_hex_str, &vec.cipher_auth.key},
        {TWEAKVALUE_JSON_STR, parser_xts_read_tweakval, &vec.iv},
        {CT_JSON_STR, parse_uint8_hex_str, &vec.ct},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback xts_interim_json_vectors[] = {
        {KEYLEN_JSON_STR, parser_xts_read_keylen, &vec.cipher_auth.key},
        {NULL, NULL, NULL} 
};
 
struct fips_test_callback xts_enc_json_vectors[] = {
        {KEY_JSON_STR, parse_uint8_known_len_hex_str, &vec.cipher_auth.key},
        {TWEAKVALUE_JSON_STR, parser_xts_read_tweakval, &vec.iv},
        {SEQNUMBER_JSON_STR, parser_xts_read_tweakval, &vec.iv},
        {PT_JSON_STR, parse_uint8_hex_str, &vec.pt},
        {NULL, NULL, NULL} 
};
 
static int
parse_test_xts_json_writeback(struct fips_val *val)
{
    struct fips_val tmp_val;
    json_t *tcId;
 
    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;
 
        tmp_val.val = val->val;
        tmp_val.len = vec.pt.len;
 
        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);
 
        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);
    } else {
            tmp_val.val = val->val;
            tmp_val.len = vec.ct.len;
 
            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_false());
        }
    }
 
    return 0;
}
 
int
parse_test_xts_json_init(void)
{
    if (json_info.json_test_group) {
        json_t *direction_obj, *tweakmode_obj;
        const char *direction_str, *tweakmode_str;
 
        direction_obj = json_object_get(json_info.json_test_group, DIR_JSON_STR);
        direction_str = json_string_value(direction_obj);
 
        if (strcmp(direction_str, OP_ENC_JSON_STR) == 0) {
            info.op = FIPS_TEST_ENC_AUTH_GEN;
            info.callbacks = xts_enc_json_vectors;
 
        } else if (strcmp(direction_str, OP_DEC_JSON_STR) == 0) {
            info.op = FIPS_TEST_DEC_AUTH_VERIF;
            info.callbacks = xts_dec_json_vectors;
        } else {
            return -EINVAL;
        }
 
        tweakmode_obj = json_object_get(json_info.json_test_group, TWEAKMODE_JSON_STR);
        tweakmode_str = json_string_value(tweakmode_obj);
        if (strcmp(tweakmode_str, "hex") == 0)
            info.interim_info.xts_data.tweak_mode = XTS_TWEAK_MODE_HEX;
        else
            info.interim_info.xts_data.tweak_mode = XTS_TWEAK_MODE_NUMBER;
 
        info.interim_callbacks = xts_interim_json_vectors;
    }
 
    info.parse_writeback = parse_test_xts_json_writeback;
    return 0;
}
#endif 
 
static int
parse_test_xts_writeback(struct fips_val *val)
{
    if (info.op == FIPS_TEST_ENC_AUTH_GEN)
        fprintf(info.fp_wr, "%s", CT_STR);
    else
        fprintf(info.fp_wr, "%s", PT_STR);
 
    parse_write_hex_str(val);
    return 0;
}
 
static int
rsp_test_xts_check(struct fips_val *val)
{
    struct fips_val *data;
    if (info.op == FIPS_TEST_ENC_AUTH_GEN)
        data = &vec.ct;
    else
        data = &vec.pt;
 
    if (memcmp(val->val, data->val, val->len) == 0)
        fprintf(info.fp_wr, "Success\n");
    else
        fprintf(info.fp_wr, "Failed\n");
    return 0;
}
 
int parse_test_xts_init(void)
{
    char *tmp;
    uint32_t i;
    for (i = 0; i < info.nb_vec_lines; i++) {
        char *line = info.vec[i];
        tmp = strstr(line, KEY_SIZE_STR);
        if (tmp) {
            tmp += (strlen(KEY_SIZE_STR) + strlen("AES"));
            if (parser_read_uint32(
                &info.interim_info.aes_data.key_len,
                    tmp) < 0)
                return -EINVAL;
            info.interim_info.aes_data.key_len =
            (info.interim_info.aes_data.key_len*2) / 8;
            continue;
        }
 
    }
    info.parse_writeback = parse_test_xts_writeback;
    info.callbacks = xts_tests_vectors;
    info.interim_callbacks = xts_tests_interim_vectors;
    info.writeback_callbacks = xts_writeback_callbacks;
    info.kat_check = rsp_test_xts_check;
 
    return 0;
}
@ RTE_CRYPTO_OP_STATUS_SUCCESS