//
// Copyright (C) 2015-2018  Markus Hiienkari <mhiienka@niksula.hut.fi>
//
// This file is part of Open Source Scan Converter project.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//

#include <string.h>
#include <unistd.h>
#include "firmware.h"
#include "sdcard.h"
#include "flash.h"
#include "controls.h"
#include "tvp7002.h"
#include "av_controller.h"
#include "lcd.h"
#include "utils.h"
#include "menu.h"
#include "altera_avalon_pio_regs.h"

extern char menu_row1[LCD_ROW_LEN+1], menu_row2[LCD_ROW_LEN+1];
extern alt_u16 rc_keymap[REMOTE_MAX_KEYS];
extern SD_DEV sdcard_dev;
extern alt_u16 sys_ctrl;

static int check_fw_header(alt_u8 *databuf, fw_hdr *hdr)
{
    alt_u32 crcval, tmp;

    strncpy(hdr->fw_key, (char*)databuf, 4);
    if (strncmp(hdr->fw_key, "OSSC", 4))
        return FW_IMAGE_ERROR;

    hdr->version_major = databuf[4];
    hdr->version_minor = databuf[5];
    strncpy(hdr->version_suffix, (char*)(databuf+6), 8);
    hdr->version_suffix[7] = 0;

    memcpy(&tmp, databuf+14, 4);
    hdr->hdr_len = bswap32(tmp);
    memcpy(&tmp, databuf+18, 4);
    hdr->data_len = bswap32(tmp);
    memcpy(&tmp, databuf+22, 4);
    hdr->data_crc = bswap32(tmp);
    // Always at bytes [508-511]
    memcpy(&tmp, databuf+508, 4);
    hdr->hdr_crc = bswap32(tmp);

    if (hdr->hdr_len < 26 || hdr->hdr_len > 508)
        return FW_HDR_ERROR;

    crcval = crc32(databuf, hdr->hdr_len, 1);

    if (crcval != hdr->hdr_crc)
        return FW_HDR_CRC_ERROR;

    return 0;
}

static int check_fw_image(alt_u32 offset, alt_u32 size, alt_u32 golden_crc, alt_u8 *tmpbuf)
{
    alt_u32 crcval=0, i, bytes_to_read;
    SDRESULTS res;

    for (i=0; i<size; i=i+SD_BLK_SIZE) {
        bytes_to_read = ((size-i < SD_BLK_SIZE) ? (size-i) : SD_BLK_SIZE);
        res = SD_Read(&sdcard_dev, tmpbuf, (offset+i)/SD_BLK_SIZE, 0, bytes_to_read);

        if (res != SD_OK)
            return -res;

        crcval = crc32(tmpbuf, bytes_to_read, (i==0));
    }

    if (crcval != golden_crc)
        return FW_DATA_CRC_ERROR;

    return 0;
}

int fw_update()
{
    int retval, i;
    int retries = FW_UPDATE_RETRIES;
    char *errmsg;
    alt_u8 databuf[SD_BLK_SIZE];
    alt_u32 btn_vec;
    alt_u32 bytes_to_rw;
    fw_hdr fw_header;

#ifdef CHECK_STACK_USE
    // estimate stack usage, assuming around here is the worst case (due to 512B databuf)
    alt_u32 sp;
    asm volatile("mov %0, sp" : "=r"(sp));
    sniprintf(menu_row1, LCD_ROW_LEN+1, "Stack size:");
    sniprintf(menu_row2, LCD_ROW_LEN+1, "%lu bytes", (ONCHIP_MEMORY2_0_BASE+ONCHIP_MEMORY2_0_SIZE_VALUE)-sp);
    ui_disp_menu(1);
    usleep(1000000);
#endif

    retval = check_sdcard(databuf);
    SPI_CS_High();
    if (retval != 0) {
        retval = -retval;
        goto failure;
    }

    retval = check_fw_header(databuf, &fw_header);
    if (retval != 0)
        goto failure;

    sniprintf(menu_row1, LCD_ROW_LEN+1, "Validating data");
    sniprintf(menu_row2, LCD_ROW_LEN+1, "%u bytes", (unsigned)fw_header.data_len);
    ui_disp_menu(1);
    retval = check_fw_image(512, fw_header.data_len, fw_header.data_crc, databuf);
    if (retval != 0)
        goto failure;

    sniprintf(menu_row1, LCD_ROW_LEN+1, "%u.%.2u%s%s", fw_header.version_major, fw_header.version_minor, (fw_header.version_suffix[0] == 0) ? "" : "-", fw_header.version_suffix);
    strncpy(menu_row2, "Update? 1=Y, 2=N", LCD_ROW_LEN+1);
    ui_disp_menu(1);

    while (1) {
        btn_vec = IORD_ALTERA_AVALON_PIO_DATA(PIO_1_BASE) & RC_MASK;

        if (btn_vec == rc_keymap[RC_BTN1]) {
            break;
        } else if (btn_vec == rc_keymap[RC_BTN2]) {
            retval = FW_UPD_CANCELLED;
            goto failure;
        }

        usleep(WAITLOOP_SLEEP_US);
    }

    //disable video output
    tvp_disable_output();
    sys_ctrl |= VIDGEN_OFF;
    IOWR_ALTERA_AVALON_PIO_DATA(PIO_0_BASE, sys_ctrl);
    usleep(10000);

    strncpy(menu_row1, "Updating FW", LCD_ROW_LEN+1);
update_init:
    strncpy(menu_row2, "please wait...", LCD_ROW_LEN+1);
    ui_disp_menu(1);

    retval = copy_sd_to_flash(512/SD_BLK_SIZE, 0, fw_header.data_len, databuf);
    if (retval != 0)
        goto failure;

    strncpy(menu_row1, "Verifying flash", LCD_ROW_LEN+1);
    strncpy(menu_row2, "please wait...", LCD_ROW_LEN+1);
    ui_disp_menu(1);
    retval = verify_flash(0, fw_header.data_len, fw_header.data_crc, databuf);
    if (retval != 0)
        goto failure;

    SPI_CS_High();

    strncpy(menu_row1, "Firmware updated", LCD_ROW_LEN+1);
    strncpy(menu_row2, "please restart", LCD_ROW_LEN+1);
    ui_disp_menu(1);
    while (1) {}

    return 0;

failure:
    SPI_CS_High();

    switch (retval) {
        case SD_NOINIT:
            errmsg = "No SD card det.";
            break;
        case FW_IMAGE_ERROR:
            errmsg = "Invalid image";
            break;
        case FW_HDR_ERROR:
            errmsg = "Invalid header";
            break;
        case FW_HDR_CRC_ERROR:
            errmsg = "Invalid hdr CRC";
            break;
        case FW_DATA_CRC_ERROR:
            errmsg = "Invalid data CRC";
            break;
        case FW_UPD_CANCELLED:
            errmsg = "Update cancelled";
            break;
        case -FLASH_VERIFY_ERROR:
            errmsg = "Flash verif fail";
            break;
        default:
            errmsg = "SD/Flash error";
            break;
    }
    strncpy(menu_row2, errmsg, LCD_ROW_LEN+1);
    ui_disp_menu(1);
    usleep(1000000);

    // Critical error, retry update
    if ((retval < 0) && (retries > 0)) {
        sniprintf(menu_row1, LCD_ROW_LEN+1, "Retrying update");
        retries--;
        goto update_init;
    }

    render_osd_page();
    return -1;
}