#include <sys/time.h>
#include <sys/timeb.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/shm.h>
#include <sys/mman.h>
#include <linux/wireless.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#include <unistd.h>
#include <stdarg.h>
#include <stdio.h>      /*標準輸入輸出定義*/
#include <stdlib.h>     /*標準函數庫定義*/
#include <unistd.h>     /*Unix 標準函數定義*/
#include <fcntl.h>      /*檔控制定義*/
#include <termios.h>    /*PPSIX 終端控制定義*/
#include <errno.h>      /*錯誤號定義*/
#include <errno.h>
#include <string.h>
#include <time.h>
#include <ctype.h>
#include <ifaddrs.h>
#include <math.h>

#include "../Log/log.h"
#include "../Config.h"
#include "Module_PrimaryComm.h"
#include "PrimaryComm.h"

//------------------------------------------------------------------------------
int tranceive(int fd, uint8_t *cmd, uint8_t cmd_len, uint8_t *rx)
{
    int len;
    //sleep(2); //required to make flush work, for some reason
    tcflush(fd, TCIOFLUSH);
    if (write(fd, cmd, cmd_len) >= cmd_len) {
        usleep(50000);
        len = read(fd, rx, 512);
    } else {
        log_error("Serial command %s response fail.\n", cmd);
    }

    return len;
}

int Query_FW_Ver(uint8_t fd, uint8_t targetAddr, Ver *Ret_Buf)
{
    uint8_t result = FAIL;
    uint8_t tx[7] = {0xaa, 0x00, targetAddr, CMD_QUERY_FW_VER, 0x00, 0x00, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;
    uint8_t len = tranceive(fd, tx, sizeof(tx), rx);

    if (len > 0) {
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3])) {
            memcpy(Ret_Buf->Version_FW, (char *)rx + 6, (rx[4] | rx[5] << 8));
            *(Ret_Buf->Version_FW + 8) = 0x00;
            result = PASS;
        }
    }

    return result;
}

int Query_HW_Ver(uint8_t fd, uint8_t targetAddr, Ver *Ret_Buf)
{
    uint8_t result = FAIL;
    uint8_t tx[7] = {0xaa, 0x00, targetAddr, CMD_QUERY_HW_VER, 0x00, 0x00, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3])) {
            memcpy(Ret_Buf->Version_HW, (char *)rx + 6, (rx[4] | rx[5] << 8));
            //*(Ret_Buf->Version_HW + 8) = 0x00;
            result = PASS;
        }
    }

    return result;
}

int Query_Gpio_Input(uint8_t fd, uint8_t targetAddr, Gpio_in *Ret_Buf)
{
    uint8_t result = FAIL;
    uint8_t tx[7] = {0xaa, 0x00, targetAddr, CMD_QUERY_GPIO_IN, 0x00, 0x00, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;
    uint8_t len = tranceive(fd, tx, sizeof(tx), rx);

    if (len > 0) {
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3])) {
            //log_info("rx[6]:0x%x,rx[7]:0x%x", rx[6], rx[7]);
            Ret_Buf->AC_Connector       = (rx[6] >> 0) & 0x01;
            Ret_Buf->AC_MainBreaker     = (rx[6] >> 1) & 0x01;
            Ret_Buf->SPD                = (rx[6] >> 2) & 0x01;
            Ret_Buf->Door_Open          = (rx[6] >> 3) & 0x01;
            Ret_Buf->GFD[0]             = (rx[6] >> 4) & 0x01;
            Ret_Buf->GFD[1]             = (rx[6] >> 5) & 0x01;
            Ret_Buf->AC_Drop            = (rx[6] >> 6) & 0x01;
            Ret_Buf->Emergency_IO       = (rx[6] >> 7) & 0x01; 

            Ret_Buf->Emergency_Btn      = (rx[7] >> 0) & 0x01;
            Ret_Buf->Button[0]          = (rx[7] >> 1) & 0x01;
            Ret_Buf->Button[1]          = (rx[7] >> 2) & 0x01;
            Ret_Buf->Key[0]             = (rx[7] >> 3) & 0x01;
            Ret_Buf->Key[1]             = (rx[7] >> 4) & 0x01;
            Ret_Buf->Key[2]             = (rx[7] >> 5) & 0x01;
            Ret_Buf->Key[3]             = (rx[7] >> 6) & 0x01;
            Ret_Buf->Button3             = (rx[7] >> 7) & 0x01;
            result = PASS;
        }
    }

    return result;
}

int Config_Gpio_Output(uint8_t fd, uint8_t targetAddr, Gpio_out *Set_Buf)
{
    uint8_t result = FAIL;
    uint8_t tx[9] = {0xaa, 0x00, targetAddr, CMD_CONFIG_GPIO_OUTPUT, 0x01, 0x00, 0x00, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    for (int idx = 0; idx < 2; idx++) {
        tx[6] |= (Set_Buf->Button_LED[idx] ? 0x01 : 0x00) << (0 + idx);
    }

    for (int idx = 0; idx < 4; idx++) {
        tx[6] |= (Set_Buf->System_LED[idx] ? 0x01 : 0x00) << (2 + idx);
    }

    tx[6] |= (Set_Buf->AC_Connector ? 0x01 : 0x00) << 6;
    tx[6] |= (Set_Buf->AC_Breaker ? 0x01 : 0x00) << 7;

    for (int idx = 0; idx < (tx[4] | tx[5] << 8); idx++) {
        chksum ^= tx[6 + idx];
    }
    tx[7] = chksum;

    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        chksum = 0x00;
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                (rx[6] == tx[6])) {
            result = PASS;
        }
    }

    return result;
}

int Config_Rtc_Data(uint8_t fd, uint8_t targetAddr, Rtc *Set_Buf)
{
    uint8_t result = FAIL;
    uint8_t tx[21] = { 0xaa, 0x00, targetAddr, CMD_CONFIG_RTC, 0x0E, 0x00, Set_Buf->RtcData[0], Set_Buf->RtcData[1],
                       Set_Buf->RtcData[2], Set_Buf->RtcData[3], Set_Buf->RtcData[4], Set_Buf->RtcData[5], Set_Buf->RtcData[6], Set_Buf->RtcData[7],
                       Set_Buf->RtcData[8], Set_Buf->RtcData[9], Set_Buf->RtcData[10], Set_Buf->RtcData[11], Set_Buf->RtcData[12], Set_Buf->RtcData[13]
                     };
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    for (int idx = 0; idx < (tx[4] | tx[5] << 8); idx++) {
        chksum ^= tx[6 + idx];
    }
    tx[20] = chksum;

    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        chksum = 0x00;
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                (rx[6] == tx[6])) {
            result = PASS;
        }
    }

    return result;
}

int Config_Model_Name(uint8_t fd, uint8_t targetAddr, uint8_t *modelname)
{
    uint8_t result = FAIL;
    uint8_t tx[21] = {0xaa, 0x00, targetAddr, CMD_CONFIG_MODEL_NAME, 0x0E, 0x00,
                      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
                     };
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    memcpy(tx + 6, modelname, 14);

    for (int idx = 0; idx < (tx[4] | tx[5] << 8); idx++) {
        chksum ^= tx[6 + idx];
    }
    tx[20] = chksum;

//  for(int i = 0; i < 21; i++)
//              printf ("tx = %x \n", tx[i]);
    uint8_t len = tranceive(fd, tx, sizeof(tx), rx);
//  for(int i = 0; i < len; i++)
//                  printf ("rx = %x \n", rx[i]);
    if (len > 6) {
        if (len < 6 + (rx[4] | rx[5] << 8)) {
            return result;
        }

        chksum = 0x00;
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                rx[6] == PASS) {
            result = PASS;
        }
    }

    return result;
}
int Update_Start(uint8_t fd, uint8_t targetAddr, uint32_t crc32)
{
    uint8_t result = FAIL;
    uint8_t tx[11] = {0xaa, 0x00, targetAddr, CMD_UPDATE_START, 0x04, 0x00, (crc32 >> 0) & 0xff, (crc32 >> 8) & 0xff, (crc32 >> 16) & 0xff, (crc32 >> 24) & 0xff, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    for (int idx = 0; idx < (tx[4] | tx[5] << 8); idx++) {
        chksum ^= tx[6 + idx];
    }
    tx[10] = chksum;


    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        chksum = 0x00;
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                (rx[6] == 0x00)) {
            result = PASS;
        }
    }

    return result;
}

int Update_Abord(uint8_t fd, uint8_t targetAddr)
{
    uint8_t result = FAIL;
    uint8_t tx[7] = {0xaa, 0x00, targetAddr, CMD_UPDATE_ABORT, 0x04, 0x00, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                (rx[6] == 0x00)) {
            result = PASS;
        }
    }

    return result;
}

int Update_Transfer(uint8_t fd, uint8_t targetAddr, uint32_t startAddr, uint8_t *data, uint16_t length)
{
    uint8_t result = FAIL;
    uint8_t tx[11 + length];
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    tx[0] = 0xaa;
    tx[1] = 0x00;
    tx[2] = targetAddr;
    tx[3] = CMD_UPDATE_TRANSFER;
    tx[4] = (4 + length) & 0xff;
    tx[5] = ((4 + length) >> 8) & 0xff;
    tx[6] = (startAddr >> 0) & 0xff;
    tx[7] = (startAddr >> 8) & 0xff;
    tx[8] = (startAddr >> 16) & 0xff;
    tx[9] = (startAddr >> 24) & 0xff;
    memcpy(tx + 10, data, length);

    for (int idx = 0; idx < (tx[4] | tx[5] << 8); idx++) {
        chksum ^= tx[6 + idx];
    }
    tx[sizeof(tx) - 1] = chksum;

    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                (rx[6] == 0x00)) {
            result = PASS;
        }
    }

    return result;
}

int Update_Finish(uint8_t fd, uint8_t targetAddr)
{
    uint8_t result = FAIL;
    uint8_t tx[7] = {0xaa, 0x00, targetAddr, CMD_UPDATE_FINISH, 0x04, 0x00, 0x00};
    uint8_t rx[512];
    uint8_t chksum = 0x00;

    if (tranceive(fd, tx, sizeof(tx), rx) > 0) {
        for (int idx = 0; idx < (rx[4] | rx[5] << 8); idx++) {
            chksum ^= rx[6 + idx];
        }

        if ((chksum == rx[6 + (rx[4] | rx[5] << 8)]) &&
                (rx[2] == tx[1]) &&
                (rx[1] == tx[2]) &&
                (rx[3] == tx[3]) &&
                (rx[6] == 0x00)) {
            result = PASS;
        }
    }

    return result;
}