/*
 * (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 "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>
/*
************************************************************************************************************
*
*                                             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] ro        :0x%x\n", i, part_info->ro);
		printf("\n");
	}
}

/*
************************************************************************************************************
*
*                                             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");
	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);

    sprite_uichar_printf("CARD OK\n");
	tick_printf("sprite success \n");
	//烧写结束
	__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;
}