/**************************************************************************************************

    Phyplus Microelectronics Limited confidential and proprietary.
    All rights reserved.

    IMPORTANT: All rights of this software belong to Phyplus Microelectronics
    Limited ("Phyplus"). Your use of this Software is limited to those
    specific rights granted under  the terms of the business contract, the
    confidential agreement, the non-disclosure agreement and any other forms
    of agreements as a customer or a partner of Phyplus. You may not use this
    Software unless you agree to abide by the terms of these agreements.
    You acknowledge that the Software may not be modified, copied,
    distributed or disclosed unless embedded on a Phyplus Bluetooth Low Energy
    (BLE) integrated circuit, either as a product or is integrated into your
    products.  Other than for the aforementioned purposes, you may not use,
    reproduce, copy, prepare derivative works of, modify, distribute, perform,
    display or sell this Software and/or its documentation for any purposes.

    YOU FURTHER ACKNOWLEDGE AND AGREE THAT THE SOFTWARE AND DOCUMENTATION ARE
    PROVIDED AS IS WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESS OR IMPLIED,
    INCLUDING WITHOUT LIMITATION, ANY WARRANTY OF MERCHANTABILITY, TITLE,
    NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT SHALL
    PHYPLUS OR ITS SUBSIDIARIES BE LIABLE OR OBLIGATED UNDER CONTRACT,
    NEGLIGENCE, STRICT LIABILITY, CONTRIBUTION, BREACH OF WARRANTY, OR OTHER
    LEGAL EQUITABLE THEORY ANY DIRECT OR INDIRECT DAMAGES OR EXPENSES
    INCLUDING BUT NOT LIMITED TO ANY INCIDENTAL, SPECIAL, INDIRECT, PUNITIVE
    OR CONSEQUENTIAL DAMAGES, LOST PROFITS OR LOST DATA, COST OF PROCUREMENT
    OF SUBSTITUTE GOODS, TECHNOLOGY, SERVICES, OR ANY CLAIMS BY THIRD PARTIES
    (INCLUDING BUT NOT LIMITED TO ANY DEFENSE THEREOF), OR OTHER SIMILAR COSTS.

**************************************************************************************************/



#include "bcomdef.h"
#include "OSAL.h"
#include "OSAL_PwrMgr.h"
#include "OSAL_bufmgr.h"
#include "gatt.h"
#include "ll.h"
#include "ll_common.h"
#include "hci.h"
#include "gapgattserver.h"
#include "gattservapp.h"
#include "central.h"
#include "gapbondmgr.h"
#include "simpleGATTprofile_ota.h"
#include "ota_mesh_master.h"
#include "timer.h"
#include "log.h"
#include "ll_def.h"
#include "global_config.h"
#include "flash.h"
#include "rflib.h"
#include "otam_cmd.h"
#include "stdio.h"
#include "stdlib.h"
#include "string.h"
#include "otam_protocol.h"
#include "ota_mesh.h"
#include "ota_app_service.h"
#include "ota_protocol.h"
#include "ota_flash.h"
#include "ota_flash_mesh.h"
//#include "otam_1clk_ota.h"
#include "crc16.h"
#include "error.h"

#define OTAF_BASE_ADDR          0x11000000
#define OTAF_END_ADDR           0x1107ffff

enum
{
    OTAM_ST_DISCONNECT = 0,
    OTAM_ST_CONNECTED,  //connected, idle
    OTAM_ST_WAIT_STARTED,
    OTAM_ST_PARAM,
    OTAM_ST_WAIT_PARTITION_INFO,
    OTAM_ST_DATA,
    OTAM_ST_COMPLETE,
    OTAM_ST_ERROR,
    OTAM_ST_CANCELING,  //wait stop
};





typedef struct
{
    ota_fw_t         fw;

    //data cache for a block transmit
    uint32_t          cache_size;
    uint32_t          cache_offset;
    uint8_t           cache_retry;
    uint8_t*          cache;
} otam_fw_t;


typedef struct
{
    uint8_t             state;
    uint8_t             run_mode;
    bool                reset_mode;
    uint16_t            mtu_a;
    uint16_t            burst_size;
    uint8_t             opcode;
    otam_fw_t           fw;
    otam_proto_meth_t   method;
} otam_proto_ctx_t;


static otam_proto_ctx_t s_otap_ctx_t;

static void print_hex (const uint8* data, uint16 len)
{
    //return;
    #if(DEBUG_INFO > 1)
    uint16 i;
    char strdata[5];

    for (i = 0; i < len - 1; i++)
    {
        //if(i %16 ==  0 && i >0)
        //  LOG("\n");
        sprintf(strdata, "%.2x", data[i]);
        AT_LOG("%s ",strdata);
    }

    sprintf(strdata, "%.2x", data[i]);
    AT_LOG("%s\n",strdata);
    #endif
}




static int otam_proto_ctx_reset(uint8_t st)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_proto_meth_t* method = &(pctx->method);
    pctx->state = st;
    pctx->opcode = 0;
    memset(&(pctx->fw), 0, sizeof(otam_fw_t));
    method->clear();
    return PPlus_SUCCESS;
}

static void otam_proto_disconnect(void* param)
{
    otam_proto_ctx_reset(OTAM_ST_DISCONNECT);
}
static void otam_proto_connect(void* param)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_proto_conn_param_t* pconn =  (otam_proto_conn_param_t*)param;
    pctx->state = OTAM_ST_CONNECTED;
    pctx->mtu_a = pconn->mtu -3;
    pctx->opcode = 0xff;

    if(pconn->run_mode == OTAC_RUNMODE_APP)
    {
        pctx->run_mode = OTAC_RUNMODE_APP;
        pctx->reset_mode = 0;
    }
    else if(pconn->run_mode == OTAC_RUNMODE_OTA)
    {
        pctx->run_mode = OTAC_RUNMODE_OTA;

        if(pctx->reset_mode == OTAC_RUNMODE_OTARES)
            pctx->run_mode = OTAC_RUNMODE_OTARES;

        pctx->reset_mode = 0;
    }
    else
    {
        pctx->run_mode = 0;
        pctx->reset_mode = 0;
    }
}

static int send_patition_info(void)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_fw_t* pfw = &(pctx->fw);
    otam_proto_meth_t* method = &(pctx->method);
    uint8_t data[20];

    if(pctx->state != OTAM_ST_WAIT_STARTED && pctx->state != OTAM_ST_DATA)
    {
        return PPlus_ERR_INVALID_STATE;
    }

    //partition cmd: 02 ID FA FA FA FA RA RA RA RA SZ SZ SZ SZ CS CS
    //            ID: index
    //            FA: flash address
    //            RA: run address
    //            SZ: partition size
    //            CS: checksum
    uint32_t val;
    uint16_t offset = 0;
    uint32_t flash_addr = 0;
    uint32_t run_addr = pfw->fw.part[pfw->fw.part_current].run_addr;

    if(run_addr > OTAF_BASE_ADDR && run_addr < OTAF_END_ADDR )
    {
        flash_addr = run_addr;
    }
    else
    {
        //calculate store address in flash
        for(int i = 0; i<pfw->fw.part_current; i++ )
        {
            if(pfw->fw.part[i].run_addr > OTAF_BASE_ADDR && pfw->fw.part[i].run_addr < OTAF_END_ADDR)
                continue;

            val = pfw->fw.part[i].size +3;
            val = val - (val%4);
            flash_addr += val;
        }
    }

    data[offset ++] = OTA_CMD_PARTITION_INFO;
    data[offset ++] = pfw->fw.part_current;
    val = flash_addr;
    data[offset ++] = (uint8_t)(val&0xff);
    data[offset ++] = (uint8_t)((val>>8)&0xff);
    data[offset ++] = (uint8_t)((val>>16)&0xff);
    data[offset ++] = (uint8_t)((val>>24)&0xff);
    val = run_addr;
    data[offset ++] = (uint8_t)(val&0xff);
    data[offset ++] = (uint8_t)((val>>8)&0xff);
    data[offset ++] = (uint8_t)((val>>16)&0xff);
    data[offset ++] = (uint8_t)((val>>24)&0xff);
    val = pfw->fw.part[pfw->fw.part_current].size;
    data[offset ++] = (uint8_t)(val&0xff);
    data[offset ++] = (uint8_t)((val>>8)&0xff);
    data[offset ++] = (uint8_t)((val>>16)&0xff);
    data[offset ++] = (uint8_t)((val>>24)&0xff);
    val = (uint32_t)(pfw->fw.part[pfw->fw.part_current].checksum);
    data[offset ++] = (uint8_t)(val&0xff);
    data[offset ++] = (uint8_t)((val>>8)&0xff);
    pctx->state = OTAM_ST_WAIT_PARTITION_INFO;
    pfw->cache_offset = 0;
    pfw->cache_size = 0;
    pfw->cache_retry = 0;
    pfw->fw.offset = 0;
    return method->write_cmd(data, offset, 1000);
}


static int load_data_cache(void)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_fw_t* pfw = &(pctx->fw);
    ota_fw_part_t* ppart = &(pfw->fw.part[pfw->fw.part_current]);
    uint16_t mtu_a = pctx->mtu_a;
    uint32_t size = mtu_a * pctx->burst_size;

    if(pfw->fw.offset + size > ppart->size)
    {
        size = ppart->size - pfw->fw.offset;
    }

    //memset(pfw->cache, 0, (ATT_MTU_SIZE-3));
    pfw->cache = (uint8_t*)(ppart->flash_addr + pfw->fw.offset+OTAFM_FW_OTA_DATA_ADDR);
    pfw->cache_size = size;
    pfw->cache_offset = 0;
    pfw->cache_retry = 0;
    pfw->fw.offset += size;
    return PPlus_SUCCESS;
}

static int send_data(void)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_fw_t* pfw = &(pctx->fw);
    otam_proto_meth_t* method = &(pctx->method);

    if(!(method->write_data))
        return PPlus_ERR_NOT_REGISTED;

    if(pctx->state != OTAM_ST_DATA)
        return PPlus_ERR_INVALID_STATE;

    int ret = PPlus_SUCCESS;
    uint16_t size = 0;
    uint16_t mtu_a = pctx->mtu_a;

    while(pfw->cache_size - pfw->cache_offset)
    {
        size = mtu_a;

        if((pfw->cache_size - pfw->cache_offset) < mtu_a)
            size = pfw->cache_size - pfw->cache_offset;

        ret = method->write_data(pfw->cache + pfw->cache_offset, size);

        if(ret != PPlus_SUCCESS)
        {
            method->write_data_delay(2);
            return PPlus_SUCCESS;
        }

        pfw->cache_offset += size;
    }

    return PPlus_SUCCESS;
}

void print_version(uint8_t* vinfo)
{
}

static void handle_app_notify_event(void* param, uint8_t len)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;

    switch(pctx->opcode)
    {
    case OTAAPP_CMD_START_OTA:
        break;

    case OTAAPP_CMD_VER:
        print_version(param);
        break;

    default:
        break;
    }
}

static void handle_ota_notify_event(void* param, uint8_t len)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_fw_t* pfw = &(pctx->fw);
    uint8_t* pnotify = (uint8_t*)param;
    otam_proto_meth_t* method = &(pctx->method);
    int retval = pnotify[0];

    if(len == 1)  //fatal error
    {
        otam_proto_ctx_reset(OTAM_ST_CONNECTED);
        return;
    }

    LOG("OTA Notif %x, %x\n",pnotify[0],pnotify[1]);
    print_hex(pnotify, len);

    switch(pnotify[1]) //response type
    {
    case OTA_RSP_START_OTA:
    {
        if(retval == PPlus_SUCCESS && pctx->state == OTAM_ST_WAIT_STARTED)
        {
            send_patition_info();
        }
        else
        {
            pctx->state = OTAM_ST_ERROR;
        }

        break;
    }

    case OTA_RSP_PARAM:
    case OTA_RSP_BLOCK_INFO:
    case OTA_RSP_BLOCK_COMPLETE:
    {
        pctx->state = OTAM_ST_ERROR;
        break;
    }

    case OTA_RSP_OTA_COMPLETE:
    {
        uint8_t data[20];

        if(method->write_cmd)
        {
            data[0] = OTA_CMD_REBOOT;
            data[1] = 1;
            pctx->state = OTAM_ST_COMPLETE;
            LOG("OTA completed!!\n");
            method->write_cmd(data, 2, 1000);
        }

        break;
    }

    case OTA_RSP_PARTITION_INFO:
    {
        pctx->state = OTAM_ST_DATA;

        if(pfw->fw.part_current)
        {
            LOG(" ");
        }

        load_data_cache();
        send_data();
        break;
    }

    case OTA_RSP_PARTITION_COMPLETE:
    {
        pfw->fw.total_offset += pfw->cache_size;
        pfw->fw.part_current ++;
        send_patition_info();
        break;
    }

    case OTA_RSP_BLOCK_BURST:
    {
        if(pctx->state != OTAM_ST_DATA)
        {
            pctx->state = OTAM_ST_ERROR;
            break;
        }

        if(retval == PPlus_SUCCESS)
        {
            load_data_cache();
            send_data();
        }
        else if(retval == PPlus_ERR_OTA_BAD_DATA)
        {
            //case block data is not completed, retry block data
            pfw->cache_retry++;
            pfw->cache_offset = 0;

            if(pfw->cache_retry > 3)
            {
                pctx->state = OTAM_ST_ERROR;
                break;
            }

            send_data();
        }
        else
        {
            pctx->state = OTAM_ST_ERROR;
        }

        break;
    }

    case OTA_RSP_REBOOT:
    {
        LOG("[OTA_RSP_REBOOT]GAPCentralRole_TerminateLink\n");
        GAPCentralRole_TerminateLink(0);
    }

    case OTA_RSP_ERASE:
    case OTA_RSP_ERROR:
    default:
    {
        pctx->state = OTAM_ST_ERROR;
        break;
    }
    }
}


void otamProtocol_event(otap_evt_t* pev)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;

    switch(pev->ev)
    {
    case OTAP_EVT_DISCONNECTED:
        otam_proto_disconnect(pev->data);
        break;

    case OTAP_EVT_CONNECTED:
        otam_proto_connect(pev->data);
        break;

    case OTAP_EVT_NOTIFY:
    {
        if(pctx->run_mode == OTAC_RUNMODE_APP)
            handle_app_notify_event(pev->data, pev->len);
        else
            handle_ota_notify_event(pev->data, pev->len);

        break;
    }

    case OTAP_EVT_DATA_WR_DELAY:
    {
        send_data();
        break;
    }

    case OTAP_EVT_BLE_TIMEOUT:
    default:
    {
        //pctx->state = OTAM_ST_ERROR;
        break;
    }
    }
}
#define OTAM_FW_DATA_ADDR   0x11040000
#define OTAM_FW_DATA_ADDR1  0x11060000

/*
    word      | desc:
    0         | flag: "OTAF"
    1         | partition number
    i*2 + 2   | run address
    i*2 + 3   | size
    N*2 +2    | data area

*/

int load_fw(uint8_t fw_id)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_fw_t* pfw = &(pctx->fw);
    uint32_t faddr = fw_id == 0 ? OTAM_FW_DATA_ADDR : OTAM_FW_DATA_ADDR1;
    uint8_t* pdata = (uint8_t*) (faddr);
    uint32_t* pdata32 = (uint32_t*) pdata;
    uint32_t offset = 0;
    LOG("load fw %x\n", faddr);
    memset((void*)pfw, 0,sizeof(otam_fw_t));

    if(!((char)(pdata[0]) == 'O' && (char)(pdata[1]) == 'T' &&(char)(pdata[2]) == 'A'&&(char)(pdata[3]) == 'F'))
    {
        return PPlus_ERR_INVALID_DATA;
    }

    pfw->fw.part_num = (uint8_t)(pdata32[1]);
    offset = 2 * 4 + 2 * 4 * pfw->fw.part_num;
    pfw->fw.total_size = 0;

    for (uint8_t i = 0; i < pfw->fw.part_num; i++)
    {
        pfw->fw.part[i].run_addr = pdata32[i*2+2];
        pfw->fw.part[i].size = pdata32[i*2+3];
        pfw->fw.part[i].flash_addr = faddr + offset;
        pfw->fw.part[i].checksum = crc16(0, (const volatile void* )(pfw->fw.part[i].flash_addr), pfw->fw.part[i].size);
        offset += pfw->fw.part[i].size;
        pfw->fw.total_size += pfw->fw.part[i].size;
    }

    pfw->fw.total_size += 0;
    return PPlus_SUCCESS;
}

int load_res(void)
{
    return PPlus_ERR_NOT_IMPLEMENTED;
}


int otamProtocol_start_ota(ota_fw_t* pffw)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_fw_t* pfw = &(pctx->fw);
    otam_proto_meth_t* method = &(pctx->method);
    uint8_t data[20];
    memcpy(&(pfw->fw), pffw, sizeof(ota_fw_t));
    pfw->fw.offset = 0;
    pfw->fw.part_current = 0;
    pfw->fw.total_offset = 0;

    if(pctx->state < OTAM_ST_CONNECTED)
        return PPlus_ERR_BLE_NOT_READY;

    if(pctx->run_mode != OTAC_RUNMODE_OTA)
    {
        return PPlus_ERR_INVALID_STATE;
    }

    if(method->write_cmd)
    {
        pctx->state = OTAM_ST_WAIT_STARTED;
        pfw->fw.part_current = 0;

        if(pctx->mtu_a == 20)
        {
            pctx->burst_size = OTA_BURST_SIZE_DEFAULT;
            data[0] = OTA_CMD_START_OTA;
            data[1] = pfw->fw.part_num;
            data[2] = 0;
        }
        else
        {
            pctx->burst_size = 0xffff;
            data[0] = OTA_CMD_START_OTA;
            data[1] = pfw->fw.part_num;
            data[2] = OTA_BURST_SIZE_HISPEED;
        }

        return method->write_cmd(data, 3, 1000);
    }

    return PPlus_ERR_NOT_REGISTED;
}


int otamProtocol_stop_ota(void)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_proto_meth_t* method = &(pctx->method);
    uint8_t data[20];

    if(method->write_cmd)
    {
        data[0] = OTA_CMD_START_OTA;
        data[1] = 0xff;
        data[2] = 0;
        pctx->state = OTAM_ST_CANCELING;
        return method->write_cmd(data, 3, 1000);
    }

    return PPlus_ERR_NOT_REGISTED;
}

int otamProtocol_app_start_ota(uint8_t mode)
{
    otam_proto_ctx_t* pctx = &s_otap_ctx_t;
    otam_proto_meth_t* method = &(pctx->method);
    uint8_t data[20];

    if(pctx->run_mode != OTAC_RUNMODE_APP)
        return PPlus_ERR_INVALID_STATE;

    if(mode > OTA_MODE_RESOURCE)
        return PPlus_ERR_INVALID_PARAM;

    if(method->write_cmd)
    {
        data[0] = OTAAPP_CMD_START_OTA;
        data[1] = mode;
        data[2] = 1;
        return method->write_cmd(data, 3, 0);
    }

    return PPlus_ERR_NOT_REGISTED;
}

int otamProtocol_init(otam_proto_meth_t* method)
{
    memset(&s_otap_ctx_t, 0, sizeof(s_otap_ctx_t));
    s_otap_ctx_t.method = *method;
    return PPlus_SUCCESS;
}

