/*
 * (C) Copyright 2007-2013
 * Allwinner Technology Co., Ltd. <www.allwinnertech.com>
 * Jerry Wang <wangflord@allwinnertech.com>
 *
 * See file CREDITS for list of people who contributed to this
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */
#include <config.h>
#include <common.h>
#include <sunxi_mbr.h>
#include <sys_config.h>
#include <sys_config_old.h>
#include "sprite_card.h"
#include "sprite_download.h"
#include "sprite_erase.h"
#include <private_boot0.h>
#include <malloc.h>
#include <sunxi_board.h>
#include <fdt_support.h>
#include "../board/sunxi/cartoon/sprite_char/sprite_char_i.h"

extern _ui_char_info_t  ui_char_info;
extern int sprite_led_exit(int status);

#ifdef CONFIG_SUNXI_DISPLAY
void sprite_card_finish_tip(void)
{
    int column  = sprite_source.screen_width/2/ui_char_info.word_size - 3;
    int row  = sprite_source.screen_height/2/ui_char_info.word_size - 3;
    sprite_uichar_printf_ex("CARD OK",row,column);
}
#endif
/*
************************************************************************************************************
*
*                                             function
*
*    name          :
*
*    parmeters     :
*
*    return        :
*
*    note          :
*
*
************************************************************************************************************
*/
void __dump_dlmap(sunxi_download_info *dl_info)
{
	dl_one_part_info		*part_info;
	u32 i;
	char buffer[32];

	printf("*************DOWNLOAD MAP DUMP************\n");
	printf("total download part %d\n", dl_info->download_count);
	printf("\n");
	for(part_info = dl_info->one_part_info, i=0;i<dl_info->download_count;i++, part_info++)
	{
		memset(buffer, 0, 32);
		memcpy(buffer, part_info->name, 16);
		printf("download part[%d] name          :%s\n", i, buffer);
		memset(buffer, 0, 32);
		memcpy(buffer, part_info->dl_filename, 16);
		printf("download part[%d] download file :%s\n", i, buffer);
		memset(buffer, 0, 32);
		memcpy(buffer, part_info->vf_filename, 16);
		printf("download part[%d] verify file   :%s\n", i, buffer);
		printf("download part[%d] lenlo         :0x%x\n", i, part_info->lenlo);
		printf("download part[%d] addrlo        :0x%x\n", i, part_info->addrlo);
		printf("download part[%d] encrypt       :0x%x\n", i, part_info->encrypt);
		printf("download part[%d] verify        :0x%x\n", i, part_info->verify);
		printf("\n");
	}
}


void __dump_mbr(sunxi_mbr_t *mbr_info)
{
	sunxi_partition  		*part_info;
	u32 i;
	char buffer[32];

	printf("*************MBR DUMP***************\n");
	printf("total mbr part %d\n", mbr_info->PartCount);
	printf("\n");
	for(part_info = mbr_info->array, i=0;i<mbr_info->PartCount;i++, part_info++)
	{
		memset(buffer, 0, 32);
		memcpy(buffer, part_info->name, 16);
		printf("part[%d] name      :%s\n", i, buffer);
		memset(buffer, 0, 32);
		memcpy(buffer, part_info->classname, 16);
		printf("part[%d] classname :%s\n", i, buffer);
		printf("part[%d] addrlo    :0x%x\n", i, part_info->addrlo);
		printf("part[%d] lenlo     :0x%x\n", i, part_info->lenlo);
		printf("part[%d] user_type :0x%x\n", i, part_info->user_type);
		printf("part[%d] keydata   :0x%x\n", i, part_info->keydata);
		printf("part[%d] backup_flag   :0x%x\n", i, part_info->backup_flag);
		printf("part[%d] ro        :0x%x\n", i, part_info->ro);
		printf("\n");
	}
}

#ifdef CONFIG_BACKUP_PARTITION
static    int work_addr = 0;
unsigned  int get_backup_flag_addr(void)
{
	return work_addr;
}

unsigned  int set_part_back_work(sunxi_mbr_t *mbr_info, unsigned int work_addr, int work)
{

	sunxi_partition         *part_info;
	u32                     i;
	char work_buffer[512] = "not_set";
	int                     ret  = -1;
	int                     used = 0;
	ret = script_parser_fetch("backup_part", "used", (int *)&used, sizeof(int) / 4);
	if (ret || !used)
	{
		printf("%s:[backup_part] no use\n",__func__);
		return 0;
	}

	if((mbr_info != NULL))
	{
		for(part_info = mbr_info->array, i=0;i<mbr_info->PartCount;i++, part_info++)
		{
			if(0x8000 == part_info->backup_flag)
			{
				work_addr = part_info->addrlo + part_info->lenlo - 1;
				printf("%s: @ part_info->name = %s, work_addr = 0x%x\n",__func__, part_info->name,work_addr);
			}
		}
	}else{
		debug("mbr_info == NULL!\n");
	}

	if(!work_addr)
	{
		//bad usage
		printf("use %s error!:: mbr_info = %p, work_addr = 0x%x, work = %d\n",__func__,mbr_info,work_addr,work);
		return -1;
	}

	if(!work)//set no_work
	{

		printf("%s: set no_work here\n",__func__);
		strcpy(work_buffer,"no_work");
		if(sunxi_sprite_write(work_addr, 1, work_buffer) != 1)//write to the last block
		{
			printf("sunxi sprite error: download backup_flag error : start 0x%x, sectors 0x%x\n",work_addr, 1);
			return -1;
		}
	}
	else//set work
	{
		printf("%s: set work here\n",__func__);
		strcpy(work_buffer,"work");
		if(sunxi_sprite_write(work_addr, 1, work_buffer) != 1)//write to the last block
		{
			printf("sunxi sprite error: download backup_flag error : start 0x%x, sectors 0x%x\n",work_addr, 1);
			return -1;
		}
	}
	printf("%s:  end  work_addr: 0x%x\n", __func__, work_addr);
	return work_addr;
}
#endif

/*
************************************************************************************************************
*
*                                             function
*
*    name          :
*
*    parmeters     :	workmode: 升级模式选择：0，卡上固件形式升级；1，文件形式升级
*
*						name    : 文件升级时的名词
*
*    return        :
*
*    note          :
*
*
************************************************************************************************************
*/
int sunxi_card_sprite_main(int workmode, char *name)
{
	int production_media;					//升级介质
	uchar  img_mbr[1024 * 1024];			//mbr
	sunxi_download_info  dl_map;			//dlinfo
	int    sprite_next_work;
	int nodeoffset;
	int mbr_num = SUNXI_MBR_COPY_NUM;

	tick_printf("sunxi sprite begin\n");
	//获取当前是量产介质是nand或者卡
	production_media = get_boot_storage_type();
	//启动动画显示
	sprite_cartoon_create();
	//检查固件合法性
	if(sprite_card_firmware_probe(name))
    {
    	printf("sunxi sprite firmware probe fail\n");

    	return -1;
    }

	sprite_cartoon_upgrade(5);
	tick_printf("firmware probe ok\n");
	//获取dl_map文件，用于指引下载的数据
	tick_printf("fetch download map\n");
	if(sprite_card_fetch_download_map(&dl_map))
	{
		printf("sunxi sprite error : fetch download map error\n");

		return -1;
	}
	__dump_dlmap(&dl_map);
	//获取mbr
	tick_printf("fetch mbr\n");
	if(sprite_card_fetch_mbr(&img_mbr))
	{
		printf("sunxi sprite error : fetch mbr error\n");

		return -1;
	}
	__dump_mbr((sunxi_mbr_t *)img_mbr);
	//根据mbr，决定擦除时候是否要保留数据
	tick_printf("begin to erase flash\n");
	nand_get_mbr((char *)img_mbr, 16 * 1024);

	if(sunxi_sprite_erase_flash(img_mbr))
	{
		printf("sunxi sprite error: erase flash err\n");

		return -1;
	}
	tick_printf("successed in erasing flash\n");
#ifdef CONFIG_BACKUP_PARTITION
	work_addr = set_part_back_work((sunxi_mbr_t *)img_mbr, 0, 0);
#endif
	if (production_media == STORAGE_NOR)
	{
		mbr_num = 1;
	}
	if(sunxi_sprite_download_mbr(img_mbr, sizeof(sunxi_mbr_t) * mbr_num))
	{
		printf("sunxi sprite error: download mbr err\n");

		return -1;
	}
	sprite_cartoon_upgrade(10);
	tick_printf("begin to download part\n");
	//开始烧写分区
	if(sunxi_sprite_deal_part(&dl_map))
	{
		printf("sunxi sprite error : download part error\n");

		return -1;
	}
	tick_printf("successed in downloading part\n");
	sprite_cartoon_upgrade(80);
	sunxi_sprite_exit(1);

	if(sunxi_sprite_deal_uboot(production_media))
	{
		printf("sunxi sprite error : download uboot error\n");

		return -1;
	}
	tick_printf("successed in downloading uboot\n");
	sprite_cartoon_upgrade(90);
	if(sunxi_sprite_deal_boot0(production_media))
	{
		printf("sunxi sprite error : download boot0 error\n");

		return -1;
	}
	tick_printf("successed in downloading boot0\n");
	sprite_cartoon_upgrade(100);

#ifdef CONFIG_SUNXI_DISPLAY
    sprite_card_finish_tip();
#endif
	tick_printf("sprite success \n");
	sprite_led_exit(0);
	//烧写结束
	__msdelay(3000);
	//处理烧写完成后的动作

	nodeoffset =  fdt_path_offset(working_fdt,FDT_PATH_CARD_BOOT);
	if(nodeoffset < 0 )
	{
		printf("get card_boot para fail\n");
		return -1;
	}
	if(fdt_getprop_u32(working_fdt, nodeoffset,"next_work",(uint32_t *) &sprite_next_work)< 0)
	//if(script_parser_fetch("card_boot", "next_work", &sprite_next_work, 1))
	{
		sprite_next_work = SUNXI_UPDATE_NEXT_ACTION_SHUTDOWN;
		//sprite_next_work = SUNXI_UPDATE_NEXT_ACTION_REUPDATE;
	}
	if(sprite_next_work == SUNXI_UPDATA_NEXT_ACTION_SPRITE_TEST)
	{
		//erase the magic from boot0 head
        printf("try to earse boot0 head :");
		char* boot0_head_buf = NULL;
		boot0_file_head_t  *boot0_head = NULL;
		boot0_head_buf = (char *)malloc(1024);
		memset(boot0_head_buf , 0x00, 1024);
		if(!sunxi_flash_phyread(BOOT0_SDMMC_START_ADDR, 1024/512, boot0_head_buf))
		{
			printf("sunxi_sprite_error : read boot0 head failed\n");
		}
		boot0_head = (boot0_file_head_t *)boot0_head_buf;
        printf("%s \n",boot0_head->boot_head.magic);
		memset(boot0_head->boot_head.magic , 0x00 , 8);
		printf("=====check_sum======%x \n",boot0_head->boot_head.check_sum);
        if(!sunxi_flash_phywrite(BOOT0_SDMMC_START_ADDR, 1024/512, boot0_head_buf))
		{
			printf("sunxi_sprite_error : can not write boot0 head into card \n");
		}
	}

	sunxi_update_subsequent_processing(sprite_next_work);

	return 0;
}