/*
 * Copyright (c) 2023, Arm Limited. All rights reserved.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 *
 */

#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include "erpc_port.h"
#include "erpc_client_start.h"
#include "tfm_erpc.h"
#include "tfm_crypto_defs.h"
#include "psa/client.h"
#include "psa/crypto.h"

#if (!defined(ERPC_TRANSPORT_UART)) && (!defined(ERPC_TRANSPORT_TCP))
#include <stdlib.h>
#include <getopt.h>
#define ARGC_UART 3
#define ARGC_TCP 4
#endif

static void test_call(void)
{
    psa_status_t status;
    uint8_t hash[PSA_HASH_LENGTH(PSA_ALG_SHA_256)] = {0};
    size_t hash_size = sizeof(hash);
    const uint8_t *msg = "test";

    status = psa_hash_compute(PSA_ALG_SHA_256, msg, strlen(msg), hash, hash_size, &hash_size);

    printf("psa_hash_compute: %d\r\n", status);
    printf("> hash_size = %zu\r\n", hash_size);
    printf("> hash = ");
    for (size_t i = 0; i < sizeof(hash); ++i) {
        printf("%02hhX", hash[i]);
    }
    printf("\r\n");
}

int main(int argc, char *argv[])
{
    erpc_transport_t transport;

#if (!defined(ERPC_TRANSPORT_UART)) && (!defined(ERPC_TRANSPORT_TCP))
    int erpc_uart_flag = 0, erpc_tcp_flag = 0;
    char *uart_dev = NULL, *tcp_host = NULL, *tcp_port = NULL;
    struct option erpc_transport_options[] =
    {
        {"UART", no_argument, &erpc_uart_flag, 1},
        {"TCP", no_argument, &erpc_tcp_flag, 1},
        {0, 0, 0, 0}
    };
#endif

#ifdef ERPC_TRANSPORT_UART
    transport = erpc_transport_serial_init(PORT_NAME, 115200);
#elif defined(ERPC_TRANSPORT_TCP)
    transport = erpc_transport_tcp_init(ERPC_HOST, ERPC_PORT, false);
#else
    /* Check if argc is correct */
    if (argc != ARGC_UART && argc != ARGC_TCP) {
        printf("Incorrect argument numbers.\r\n");
        printf("Please input --UART PORT or --TCP HOST PORT\r\n");
        return 1;
    }

    /* Loop check to set _flag and parse arguments */
    while (getopt_long(argc, argv, "", erpc_transport_options, NULL) != -1);
    if (!erpc_uart_flag && !erpc_tcp_flag) {
        printf("No valid transportation layer selected.\r\n");
        return 1;
    } else if(erpc_uart_flag && erpc_tcp_flag) {
        printf("UART and TCP cannot be set at the same time.\r\n");
        return 1;
    } else if (erpc_uart_flag) {
        if (argc - optind != 1) {
            printf("Incorrect argument numbers for --UART.\r\n");
            return 1;
        }
        uart_dev = argv[optind];
    } else if (erpc_tcp_flag) {
        if (argc - optind != 2) {
            printf("Incorrect argument numbers for --TCP.\r\n");
        return 1;
        }
        tcp_host = argv[optind];
        tcp_port = argv[optind + 1];
    }

    /* eRPC transport initialization */
    if (erpc_uart_flag) {
        printf("UART device is setting to %s\r\n",uart_dev);
        transport = erpc_transport_serial_init(uart_dev, 115200);
    } else if (erpc_tcp_flag) {
        printf("TCP host is setting to %s\t",tcp_host);
        printf("TCP port is setting to %s\r\n",tcp_port);
        transport = erpc_transport_tcp_init(tcp_host, atoi(tcp_port), false);
    }
#endif

    if (!transport) {
        printf("eRPC transport init failed!\r\n");
        return 1;
    }

    erpc_client_start(transport);

    printf("psa_framework_version: %d\r\n", psa_framework_version());

    test_call();

    return 0;
}