/**
* @file configCliObj.c
* @author NXP Semiconductors
* @version 1.0
* @par License
*
* Copyright 2018 NXP
* SPDX-License-Identifier: Apache-2.0
*
* @par Description
* Command line handling 'obj' entry
*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>

// project specific include files
#include "sm_types.h"
#include "sm_apdu.h"
#include "tst_sm_util.h"
#include "tst_a71ch_util.h"
#include "probeAxUtil.h"
#include "configCli.h"
#include "configCmd.h"

#include "axHostCrypto.h"
#include "tstHostCrypto.h"

#define FLOW_VERBOSE_PROBE_A70

#ifdef FLOW_VERBOSE_PROBE_A70
#define FPRINTF(...) printf (__VA_ARGS__)
#else
#define FPRINTF(...)
#endif

// #define DBG_PROBE_A70

#ifdef DBG_PROBE_A70
#define DBGPRINTF(...) printf (__VA_ARGS__)
#else
#define DBGPRINTF(...)
#endif

/**
* Allocate memory
*/
static int AllocateBufferObj(U8 ** buf, U16 * length, int argc, int argCurrent, char **argv) {
    // Do not go beyond the last argument when parsing
    if ((argCurrent + 1) >= argc)
    {
        return AX_CLI_ARG_COUNT_MISTAKE;
    }
    *length = (U16)strlen(argv[argCurrent + 1]);
    *length = ((*length) / 2) + 1;
    // Alloc
    *buf = (U8 *)malloc(sizeof(U8) * (*length));
    if (buf == 0)
    {
        return AX_CLI_DYN_ALLOC_ERROR;
    }

    return AX_CLI_EXEC_OK;
}

//
// \return Return 0 upon success
int a7xConfigCliCmdObj(int argc, char ** argv, U16 *sw)
{
    int argCurrent = 1;
    a71_ObjCmdClass_t ssc = A71_OBJ_UNDEF;
    int offset = 0;
    int bufLen = 0;
    int lineLen = 0;
    int nRet = 0;
    int index = 0;
    int chunkSize = 64;
    int readLength = 0;
    int segments = 0;
    U8 * objData= NULL;
    U16 objDataLen = 0;
    U8 offsetArray[4];
    U16 offsetArrayLen = sizeof(offsetArray);
    U8 readLengthArray[4];
    U16 readLengthArrayLen = sizeof(readLengthArray);
    char szFilename[MAX_FILE_PATH];
    char szLineSize[MAX_FILE_PATH];


    // Do not go beyond the last argument when parsing
    if (argCurrent >= argc)
    {
        a7xConfigCliHelp("a71chConfig");
        return AX_CLI_ARG_COUNT_MISTAKE;
    }

    if (strcmp(argv[argCurrent], "update") == 0) {
        ssc = A71_OBJ_UPDATE;
    }
    else if (strcmp(argv[argCurrent], "write") == 0) {
        ssc = A71_OBJ_WRITE;
    }
    else if (strcmp(argv[argCurrent], "get") == 0) {
        ssc = A71_OBJ_READ;
    }
    else if (strcmp(argv[argCurrent], "erase") == 0) {
        ssc = A71_OBJ_ERASE;
    }
    else {
        printf("%s is an unknown command option.\n", argv[argCurrent]);
        return a7xConfigCliHelp("a71chConfig");
    }
    argCurrent++;

    // Get Index
    nRet = axCliGetInteger("x", "", &index, 0, MAX_OBJECTS_NUM-1, argc, argv, &argCurrent);
    if (nRet != AX_CLI_EXEC_OK) { return nRet; }

    switch (ssc)
    {
    case A71_OBJ_WRITE:
        //obj create -x <int> [ -f <data.txt> | -h <hexvalue_data> | -n <segments> ]
        if (strcmp(argv[argCurrent], "-h") == 0) {
            // Allocate memory for input buffer
            nRet = AllocateBufferObj(&objData, &objDataLen, argc, argCurrent, argv);
            if (nRet != AX_CLI_EXEC_OK) { return nRet; }
        }
        nRet = axCliGetHexString("h", "", objData, &objDataLen, 1, objDataLen, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) {
            if (nRet == AX_CLI_ARG_OPTION_ERROR) {
                szFilename[0] = 0;
                bufLen = sizeof(szFilename) - 1;
                nRet = axCliGetOptionalString("f", "", szFilename, bufLen, argc, argv, &argCurrent);
                if (nRet != AX_CLI_EXEC_OK) {
                    if (nRet == AX_CLI_ARG_OPTION_ERROR) {
                        nRet = axCliGetInteger("n", "", &segments, 0, 127, argc, argv, &argCurrent);
                        if (nRet != AX_CLI_EXEC_OK) { return nRet; }
                        nRet = a7xConfigCmdWriteObjFromSegments(index, segments, sw);
                    }
                    else {
                        return nRet;
                    }
                }
                else {
                    // Read optional file line size
                    lineLen = sizeof(szLineSize) - 1;
                    nRet = axCliGetOptionalString("t", "", szLineSize, lineLen, argc, argv, &argCurrent);
                    if (nRet != AX_CLI_EXEC_OK) { chunkSize = 64; }
                    if (szFilename[0] == 0x0 && nRet == AX_CLI_EXEC_OK) {
                        return AX_CLI_ARG_COUNT_MISTAKE;
                    }
                    szLineSize[lineLen] = 0x0;
                    if (nRet == AX_CLI_EXEC_OK) {
                        if ((strcmp(szLineSize, "hex_16") == 0x0) || strcmp(szLineSize, "Hex_16") == 0x0) {
                            chunkSize = 32;
                        }
                        else if ((strcmp(szLineSize, "hex_32") != 0x0) && strcmp(szLineSize, "Hex_32") != 0x0) {
                            return AX_CLI_ARG_VALUE_ERROR;
                        }
                    }
                    nRet = a7xConfigCmdWriteObjFromfile(index, szFilename, chunkSize, A71_OBJ_WRITE, sw);
                    return nRet;
                }
            }
            else {
                return nRet;
            }
        }
        else {
            nRet = a7xConfigCmdWriteObj(index, objData, objDataLen, sw);
            if (objData) { free(objData); }
        }
        break;
    case A71_OBJ_UPDATE:
        // obj update -x <int> -h <hexvalue_offset> [-f <data.txt> -t [hex_16|hex_32] | -h <hexvalue_data>]
        nRet = axCliGetHexString("h", "", offsetArray, &offsetArrayLen, 2, 2, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) { return nRet; }
        offset = (offsetArray[0] << 8) + (offsetArray[1]);
        if (strcmp(argv[argCurrent], "-h") == 0) {
            // Allocate memory for input buffer
            nRet = AllocateBufferObj(&objData, &objDataLen, argc, argCurrent, argv);
            if (nRet != AX_CLI_EXEC_OK) { return nRet; }
        }
        nRet = axCliGetHexString("h", "", objData, &objDataLen, 1, 4096, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) {
            if (nRet == AX_CLI_ARG_OPTION_ERROR) {
                szFilename[0] = 0;
                bufLen = sizeof(szFilename) - 1;
                nRet = axCliGetString("f", "", szFilename, bufLen, argc, argv, &argCurrent);
                if (nRet != AX_CLI_EXEC_OK) { return nRet; }
                // Read optional file line size
                lineLen = sizeof(szLineSize) - 1;
                nRet = axCliGetOptionalString("t", "", szLineSize, lineLen, argc, argv, &argCurrent);
                if (nRet != AX_CLI_EXEC_OK) { chunkSize = 64; }
                if (szFilename[0] == 0x0 && nRet == AX_CLI_EXEC_OK) {
                    return AX_CLI_ARG_COUNT_MISTAKE;
                }
                szLineSize[lineLen] = 0x0;
                if (nRet == AX_CLI_EXEC_OK) {
                    if ((strcmp(szLineSize, "hex_16") == 0x0) || strcmp(szLineSize, "Hex_16") == 0x0) {
                        chunkSize = 32;
                    }
                    else if ((strcmp(szLineSize, "hex_32") != 0x0) && strcmp(szLineSize, "Hex_32") != 0x0) {
                        return AX_CLI_ARG_VALUE_ERROR;
                    }
                }
                nRet = a7xConfigCmdUpdateObjFromfile(index, offset, szFilename, chunkSize, A71_OBJ_UPDATE, sw);
                return nRet;
            }
        }
        else {
            nRet = a7xConfigCmdUpdateObj(index, offset, objData, objDataLen, sw);
            if (objData) { free(objData); }
        }
        break;
    case A71_OBJ_READ:
        // obj get -x <int> -h <hexvalue_offset> -s <int_size> [ -f <data.txt> ]
        // Read offset
        nRet = axCliGetHexString("h", "", offsetArray, &offsetArrayLen, 2, 2, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) {
            offset = 0x0;
        }
        else {
            offset = (offsetArray[0] << 8) + (offsetArray[1]);
        }
        // Read length of read
        nRet = axCliGetHexString("s", "", readLengthArray, &readLengthArrayLen, 2, 2, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) {
            readLength = 0;
        }
        else {
            readLength = (readLengthArray[0] << 8) + (readLengthArray[1]);
        }
        // Read optional file name
        bufLen = sizeof(szFilename) - 1;
        nRet = axCliGetOptionalString("f", "", szFilename, bufLen, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) { memset(szFilename, 0x0, sizeof(szFilename)); }
        // Read optional file line size
        lineLen = sizeof(szLineSize) - 1;
        nRet = axCliGetOptionalString("t", "", szLineSize, lineLen, argc, argv, &argCurrent);
        if (nRet != AX_CLI_EXEC_OK) { chunkSize = 64; }
        if (szFilename[0] == 0x0 && nRet == AX_CLI_EXEC_OK) {
            return AX_CLI_ARG_COUNT_MISTAKE;
        }
        szLineSize[lineLen] = 0x0;
        if (nRet == AX_CLI_EXEC_OK) {
            if ((strcmp(szLineSize, "hex_16") == 0x0) || strcmp(szLineSize, "Hex_16") == 0x0) {
                chunkSize = 32;
            }
            else if ((strcmp(szLineSize, "hex_32") != 0x0) && strcmp(szLineSize, "Hex_32") != 0x0) {
                return AX_CLI_ARG_VALUE_ERROR;
            }
        }
        // read data
        nRet = a7xConfigCmdReadObj(index, offset, readLength, chunkSize, szFilename, sw);
        break;

    case A71_OBJ_ERASE:
        // obj erase -x <int>
        nRet = a7xConfigCmdEraseObj(index, sw);
        break;

    default:
        // Cannot be triggered
        return AX_CLI_NOT_IMPLEMENTED;
    }

    return nRet;
}