/*---------------------------------------------------------------------------*
  Project:  TwlSDK - tests - appjumpTest - Nand-2
  File:     main.c

  Copyright 2008 Nintendo.  All rights reserved.

  These coded instructions, statements, and computer programs contain
  proprietary information of Nintendo of America Inc. and/or Nintendo
  Company Ltd., and are protected by Federal copyright law.  They may
  not be disclosed to third parties or copied or duplicated in any form,
  in whole or in part, without the prior written consent of Nintendo.

  $Date:: 2008-09-03#$
  $Rev: 8251 $
  $Author: nishimoto_takashi $
 *---------------------------------------------------------------------------*/
#include <twl.h>
#include <twl/fatfs.h>
#include <twl/os/common/format_rom.h>
#include <twl/nam.h>
#include <twl/aes.h>
#include <twl/os/common/banner.h>
#include <nitro/nvram.h>
#include <nitro/math/rand.h>

#include "application_jump_private.h"
#include "common.h"
#include "screen.h"

#define TITLE_SHOW_BASE_Y			5
#define TITLE_MAX_SHOW				0x10
#define TITLE_NUM_PAGE 				300

#define ETICKET_NUM_MAX				10

#define ES_ERR_OK					0

// 表示する対象をユーザーアプリだけにする場合
#define USER_APP_ONLY

// デバッグ用
//#define DEBUG_MODE

/*---------------------------------------------------------------------------*
    変数 定義
 *---------------------------------------------------------------------------*/
typedef void* (*NAMUTAlloc)(u32 size);
typedef void  (*NAMUTFree)(void* ptr);

static NAMUTAlloc spAllocFunc;
static NAMUTFree  spFreeFunc;

// キー入力
static KeyInfo  gKey;

// インストールされている NAND アプリの数
static s32 gNandAppNum;
static s32 gNandInstalledAppNum;
static s32 gNandAllAppNum;

// カーソル位置
static s32 gCurPos = 0;

// 選択中の要素
static s32 gCurrentElem;

// ページ数
static u32 gCurrentPage;
static u32 gMaxPage;

// Error
static BOOL gErrorFlg;

// eTicketType
typedef enum ETicketType {
	ETICKET_TYPE_COMMON = 0,
	ETICKET_TYPE_PERSONALIZED = 1
}ETicketType;

typedef struct DataStruct
{
	NAMTitleId		id;
	BOOL			commonTicketFlg;
    BOOL			isSrlFlg;
    u32				numTicket;
    ETicketType		tType[ETICKET_NUM_MAX];
    
} DataStruct;

typedef struct {
	u8               pad1[ 12 ];
    u32              deviceId;
	u8				 pad2[ 216 - 16 ];
} ESTicketView;

typedef s32 ESError;

extern ESError ES_GetTicketViews(u64 titleId, ESTicketView* ticketViewList, u32* ticketViewCnt);

static DataStruct gDataList[TITLE_NUM_PAGE * 2];
static DataStruct gInstalledDataList[TITLE_NUM_PAGE];

#ifdef DEBUG_MODE
static MATHRandContext32 context;
#endif

/*---------------------------------------------------------------------------*
   Prototype
 *---------------------------------------------------------------------------*/
static void DrawScene(DataStruct* list);
static BOOL GetDataStruct(DataStruct* list, DataStruct* Ilist);

static void ConvertTitleIdLo(u8* code, u8* titleid_lo);
static void ConvertGameCode(u8* code, u32 game_code);
static void ConvertInitialCode(u8* code, u32 titleid_lo);

static void* AllocForNAM(u32 size);
static void FreeForNAM(void* ptr);

static s32 GetETicketType(DataStruct* data, ETicketType *pETicketType );
static s32 GetTicketViews(ESTicketView** pptv, u32* pNumTicket, NAMTitleId titleId);

BOOL GetETicketData( void );
void* MyNAMUT_Alloc(u32 size);
void MyNAMUT_Free(void* buffer);

#ifdef USER_APP_ONLY
static void getUserApplication(DataStruct* list);
#endif
/*---------------------------------------------------------------------------*/

/*---------------------------------------------------------------------------*
  Name:         TwlMain

  Description:  メイン関数
 *---------------------------------------------------------------------------*/
void TwlMain(void)
{
	u32 count = 0;
    BOOL errorFlg = FALSE;

	InitCommon();
    InitScreen();

    GX_DispOn();
    GXS_DispOn();

    spAllocFunc = AllocForNAM;
	spFreeFunc = FreeForNAM;

	gNandAppNum = 0;
	gNandInstalledAppNum = 0;
	gNandAllAppNum = 0;
	gErrorFlg = FALSE;
    
	FS_Init( FS_DMA_NOT_USE );
    
    NAM_Init(AllocForNAM, FreeForNAM);

	MI_CpuClear8( gDataList, sizeof(gDataList));
	MI_CpuClear8( gInstalledDataList, sizeof(gInstalledDataList));

#ifdef DEBUG_MODE
    MATH_InitRand32( &context, 15 );
#endif
    
    ClearScreen();

	PutMainScreen( 7, 12, 0xf6, "--- Now Loading ---");
	PutSubScreen(  7, 12, 0xf6, "--- Now Loading ---");

	(void) GetETicketData();

    while(TRUE)
    {
        // キー入力情報取得
		ReadKey(&gKey);

        if (gKey.trg & PAD_KEY_DOWN)
        {
			gCurPos++;
            
            if ( gCurrentPage != gMaxPage )
            {
				if ( gCurPos >= TITLE_MAX_SHOW )
                {
                    gCurrentPage++;
					gCurPos = 0;
                }
            }
            else
            {
                if( (gNandAllAppNum & 0x0f) == 0 )
                {
					if( gCurPos >= TITLE_MAX_SHOW )
                    {
                    	gCurrentPage = 0;
						gCurPos = 0;
                    }
                }
				else if ( gCurPos >= (gNandAllAppNum & 0x0f) ) // バグ
                {
                    gCurrentPage = 0;
					gCurPos = 0;
                }
            }

            OS_TPrintf("↓ gCurPos : %x, gCurrentPage : %x, gNandAllAppNum : %x\n", gCurPos, gCurrentPage, gNandAllAppNum);
        }
        if (gKey.trg & PAD_KEY_UP)
        {
            if( gCurPos == 0)
            {
            	if ( gCurrentPage == 0 )
            	{
                    gCurrentPage = gMaxPage;
                    gCurPos = ((gNandAllAppNum & 0x0f) == 0) ? TITLE_MAX_SHOW - 1 : (gNandAllAppNum & 0x0f) - 1;
            	}
            	else
            	{
                    gCurrentPage--;
					gCurPos = TITLE_MAX_SHOW - 1;
            	}
            }
            else
            {
				gCurPos--;
            }

            OS_TPrintf("↑ gCurPos : %x, gCurrentPage : %x, gNandAllAppNum : %x\n", gCurPos, gCurrentPage, gNandAllAppNum);
		}
        
        if (gKey.trg & PAD_KEY_LEFT)
        {
            if(gCurrentPage == 0)
            {
				gCurrentPage = gMaxPage;
            }
            else
            {
				gCurrentPage--;
            }

			gCurPos = 0;

            OS_TPrintf("← gCurPos : %x, gCurrentPage : %x, gNandAllAppNum : %x\n", gCurPos, gCurrentPage, gNandAllAppNum);
        }
        if (gKey.trg & PAD_KEY_RIGHT)
        {
			if(gCurrentPage == gMaxPage)
            {
				gCurrentPage = 0;
            }
            else
            {
				gCurrentPage++;
            }

            gCurPos = 0;

            OS_TPrintf("→ gCurPos : %x, gCurrentPage : %x, gNandAllAppNum : %x\n", gCurPos, gCurrentPage, gNandAllAppNum);
        }

        // 選択中の要素
        gCurrentElem = (s32)((u32)(gCurrentPage << 4) + (u32)gCurPos);
        
		// 画面描画
		DrawScene(gDataList);
        
        // Ｖブランク待ち
        OS_WaitVBlankIntr();
        
        // 画面クリア
        ClearScreen();
    }

	// Ｖブランク待ち 最後に画面を更新してから終了
    OS_WaitVBlankIntr();
    OS_Terminate();
}


/*---------------------------------------------------------------------------*
  Name:         DrawScene

  Description:  画面描画関数

  メモ：
    0xf0, // 黒				0xf1, // 赤				0xf2, // 緑
  	0xf3, // 青				0xf4, // 黄				0xf5, // ピンク
	0xf6, // 水色			0xf7, // くすんだ赤		0xf8, // くすんだ緑
	0xf9, // くすんだ青 	0xfa, // くすんだ黄色	0xfb, // 紫
    0xfc, // うすい青		0xfd, // 灰色			0xfe, // 濃い灰色
 *---------------------------------------------------------------------------*/
#define COMMON_COLOR			((u8)0xff)
#define PERSONALIZED_COLOR		((u8)0xfc)

#define GAME_CODE_BASE_X		1

static void DrawScene(DataStruct* list)
{
	s32 i;
	u8 init_code[5];
	u8 color;
    u32 start;

    DataStruct* p = list;

	if( gErrorFlg )
    {
		PutMainScreen( 10, 12, 0xf1, "--- Error ---");
		PutSubScreen(  10, 12, 0xf1, "--- Error ---");

        return;
    }

	// 上画面	一覧表示
	PutMainScreen( 1,  0, 0xf2, "------- eTicket Viewer ------- ");
    PutMainScreen( 1,  1, 0xfa, "<Page %d/%d>", (gCurrentPage+1), (gMaxPage+1));
	PutMainScreen( 1,  2, 0xf4, " Game       Ticket Ticket");
    PutMainScreen( 1,  3, 0xf4, "  Code  srl   Num    Type");
	PutMainScreen( 0,  4, 0xff, "--------------------------------");
    
	// カーソル表示
    if( gCurPos <= TITLE_MAX_SHOW ){
		PutMainScreen( 0, gCurPos+TITLE_SHOW_BASE_Y , 0xf1, ">");
    }

	start = (u32)(gCurrentPage << 4);
	p += start;
    
    for ( i=(s32)start; i < (start + TITLE_MAX_SHOW); i++, p++)
    {
		s32 tmp_i;
        
		// そもそも NAND アプリの数が 1ページにも満たない場合は途中で終了する
		if ( i >= gNandAllAppNum )
		{
			break;
		}
        
		tmp_i = (s32)(i & 0xf);
        
		ConvertInitialCode(init_code, NAM_GetTitleIdLo(p->id));
        
    	color = p->commonTicketFlg ? COMMON_COLOR : PERSONALIZED_COLOR;

        // ゲームコード表示
    	PutMainScreen( GAME_CODE_BASE_X, TITLE_SHOW_BASE_Y+tmp_i, color, "%2d:%s", (tmp_i+1), init_code);

        // srlの有無表示
		if(p->isSrlFlg)
        {
			PutMainScreen( GAME_CODE_BASE_X + 9, TITLE_SHOW_BASE_Y+tmp_i, color, "o");
        }
        else
        {
			PutMainScreen( GAME_CODE_BASE_X + 9, TITLE_SHOW_BASE_Y+tmp_i, color, "x");
        }

        // ETicketの数の表示
    	PutMainScreen( GAME_CODE_BASE_X + 12, TITLE_SHOW_BASE_Y+tmp_i, color, "%d", p->numTicket);

        // ETicketのタイプの表示
    	if(p->commonTicketFlg)
    	{
			PutMainScreen(GAME_CODE_BASE_X + 19, TITLE_SHOW_BASE_Y+tmp_i, color, "common");
    	}
    	else
    	{
			PutMainScreen(GAME_CODE_BASE_X + 19, TITLE_SHOW_BASE_Y+tmp_i, color, "personalized");
    	}
    }

    PutMainScreen( 0, TITLE_MAX_SHOW + TITLE_SHOW_BASE_Y, 0xff, "--------------------------------");

    PutMainScreen( 0, TITLE_MAX_SHOW + TITLE_SHOW_BASE_Y + 1, 0xfa, "Up Down Key   : Next Application");
    PutMainScreen( 0, TITLE_MAX_SHOW + TITLE_SHOW_BASE_Y + 2, 0xfa, "Left Right Key: Page Change");

    // 下画面	詳細表示
	ConvertInitialCode(init_code, NAM_GetTitleIdLo(list[gCurrentElem].id));
    PutSubScreen(3,   2, 0xf4, "Selected Title : [ %s ]", init_code);
    PutSubScreen(3,   4, 0xff, "- Ticket List -");

    for( i=0; i < list[gCurrentElem].numTicket; i++){
        if(i > 15)
        {
			break;
        }
        
        PutSubScreen(5, 6+i, 0xf4, "Ticket%d : ", (i+1));

        if(list[gCurrentElem].tType[i] == ETICKET_TYPE_COMMON)
        {
			PutSubScreen(15, 6+i, COMMON_COLOR, "COMMON");
        }
        else
        {
			PutSubScreen(15, 6+i, PERSONALIZED_COLOR, "PERSONALIZED");
        }
    }
}


/*---------------------------------------------------------------------------*
  Name:         GetDataStruct

  Description:  TitleIDリストを取得する関数
    			NAM_GetTitleList と NAM_GetInstalledTitleList を使う
 *---------------------------------------------------------------------------*/
static BOOL GetDataStruct(DataStruct* list, DataStruct* Ilist)
{
	// タイトルIDリストバッファ
	NAMTitleId titleIdList[TITLE_NUM_PAGE];
	s32 i;

    // --- GetTitleList
	if ( NAM_GetTitleList(titleIdList, TITLE_NUM_PAGE) != NAM_OK )
	{
		OS_PutString("NAM_GetTitleList failed.");
		return FALSE;
	}
	
	// データリストの作成
	for ( i=0; i<TITLE_NUM_PAGE; i++, list++)
	{
		// そもそも NAND アプリの数が 1ページにも満たない場合は途中で終了する
		if ( i >= gNandAppNum )
		{
			break;
		}

        OS_TPrintf("id : 0x%08x\n", titleIdList[i]);
		list->id = titleIdList[i];
        list->isSrlFlg = TRUE;
	}

	MI_CpuClear8(titleIdList, sizeof(titleIdList));

	OS_PutString("\n\n");
    
    // --- GetInstalledTitleList
	if ( NAM_GetInstalledTitleList(titleIdList, TITLE_NUM_PAGE) != NAM_OK )
	{
		OS_PutString("NAM_GetInstalledTitleList failed.");
		return FALSE;
	}
	
	// データリストの作成
	for ( i=0; i<TITLE_NUM_PAGE; i++, Ilist++)
	{
		// そもそも NAND アプリの数が 1ページにも満たない場合は途中で終了する
		if ( i >= gNandInstalledAppNum )
		{
			break;
		}

        OS_TPrintf("id : 0x%08x\n", titleIdList[i]);
		Ilist->id 		= titleIdList[i];
        Ilist->isSrlFlg = FALSE;
	}

	return TRUE;
}


/*---------------------------------------------------------------------------*
  Name:         GetETicketType

  Description:  指定された titleID の eTicket タイプを取得する

  Arguments:    titleID: common eTicket かどうかを調べたいタイトルの titleID
	            pETicketType : 結果を格納するESETicketTypeポインタ

  Returns:      NAM_OK  : 取得成功
	            それ以外: 取得失敗
 *---------------------------------------------------------------------------*/
static s32 GetETicketType(DataStruct* data, ETicketType *pETicketType )
{
    s32 result;
	ESTicketView* ptv;
    u32 numTicket;

#ifdef DEBUG_MODE
    u32 i;
#endif
	
	*pETicketType = ETICKET_TYPE_PERSONALIZED;
	
    result = GetTicketViews(&ptv, &numTicket, data->id);

#ifndef DEBUG_MODE
	data->numTicket = numTicket;
#else
	data->numTicket = MATH_Rand32( &context, 10 );
#endif
    
    if( result == NAM_OK )
    {
        if( numTicket > 0 )
        {
            int i;
			// eTicket は、そのままもしくは追加しかありえないので、プリインストールされたアプリでは、必ずCommon eTikcetが存在する。
			// よって、全ての eTicket のうち、ひとつでも deviceId が 0x00000000 なら、common eTicket と判断。
            for( i = 0; i < numTicket; i++ )
            {
				if( ptv[i].deviceId == 0x00000000 )
                {
					*pETicketType = ETICKET_TYPE_COMMON;
                    data->tType[i] = ETICKET_TYPE_COMMON;
				}
                else
                {
					data->tType[i] = ETICKET_TYPE_PERSONALIZED;
                }
			}
		}
        MyNAMUT_Free(ptv);
	}

#ifdef DEBUG_MODE
	for( i=0; i<data->numTicket; i++)
    {
        if( MATH_Rand32( &context, 10 ) % 5 )
        {
			data->tType[i] = ETICKET_TYPE_COMMON;
        }
        else
        {
			data->tType[i] = ETICKET_TYPE_PERSONALIZED;
        }
    }
#endif
    
	return result;
}


/*---------------------------------------------------------------------------*
  Name:         GetTicketViews

  Description:  指定されたタイトルの eTicket を取得
　　　　　　　　※nam_title.c の GetTicketViews 関数をコピペ

  Arguments:    pptv       : 取得成功時に eTicket リストのポインタを格納するポインタ
	            pNumTicket : 取得成功時に eTicket 数を格納するポインタ
                titleID    : eTicket を取得したいタイトルの titleID

  Returns:      NAM_OK     : 取得成功
	            それ以外   : 取得失敗
 *---------------------------------------------------------------------------*/
static s32 GetTicketViews(ESTicketView** pptv, u32* pNumTicket, NAMTitleId titleId)
{
    s32 result;
    u32 numTicket;
    ESTicketView* ptv = NULL;

    result = ES_GetTicketViews(titleId, NULL, &numTicket);

    if( result != ES_ERR_OK )
    {
        return result;
    }

    if( numTicket != 0 )
    {
        ptv = MyNAMUT_Alloc(sizeof(ESTicketView) * numTicket);

        if( ptv == NULL )
        {
            return NAM_NO_MEMORY;
        }

        result = ES_GetTicketViews(titleId, ptv, &numTicket);
    }

    if( result == ES_ERR_OK )
    {
        *pptv = ptv;
        *pNumTicket = numTicket;
    }
    else
    {
        MyNAMUT_Free(ptv);
    }

    return result;
}


/*---------------------------------------------------------------------------*
  Name:         GetETicketData

  Description:  各アプリのETicketデータを取得する
 *---------------------------------------------------------------------------*/
BOOL GetETicketData( void )
{
    s32 result = TRUE;
    s32 i,j;

    // NAND にインポートされているNAND アプリの数を取得する
    if ( (gNandAppNum = NAM_GetNumTitles()) < 0)
    {
		OS_Panic("NAM_GetNumTitles() failed.");
	}
    // 実体があるタイトル数
    if ( (gNandInstalledAppNum = NAM_GetNumInstalledTitles()) < 0)
    {
		OS_Panic("NAM_GetNumInstalledTitles() failed.");
	}
   	// 情報の取得
	if ( !GetDataStruct(gDataList, gInstalledDataList) )
	{
		OS_Panic("GetDataStruct() failed.");
	}

    // NAM_GetTitleList          -- 削除されているがeTicketのみ存在するタイトルがリストアップされない
    // NAM_GetInstalledTitleList -- SRLはあるがeTicketがないタイトルがリストアップされない
    // そのため両者をマージする
    gNandAllAppNum = gNandAppNum;
    for (i=0; i<gNandInstalledAppNum; i++)
    {
        BOOL find = FALSE;
        for (j=0; j<gNandAppNum; j++)
        {
            if (gInstalledDataList[i].id == gDataList[j].id) 
            {
                find = TRUE;
                break;
            }
        }
        if (find == FALSE)
        {
            MI_CpuCopy8(&gInstalledDataList[i], &gDataList[gNandAllAppNum], sizeof(DataStruct));
            gNandAllAppNum++;
        }
    }

#ifdef USER_APP_ONLY
    // ユーザーアプリだけ抽出する
	getUserApplication( gDataList );
#endif

	// 必要なページ数を求める
    gMaxPage = (u32)(gNandAllAppNum >> 4);
	if( gMaxPage != 0 && (gNandAllAppNum & 0xf) == 0 )
    {
		gMaxPage--;
    }
    
    // 現在ページの初期化
    gCurrentPage = 0;

	OS_TPrintf("gNandAllAppNum : %d\n",gNandAllAppNum);
    OS_TPrintf("gMaxPage       : %d\n",gMaxPage);

	if( gNandAllAppNum == 0 )
    {
		gErrorFlg = TRUE;

        return FALSE;
    }
    
    // アプリのETicketデータを取得する
    for (i=0; i<gNandAllAppNum; i++)
    {
		ETicketType eTicketType = ETICKET_TYPE_PERSONALIZED; // default
		
		if( GetETicketType( &gDataList[i], &eTicketType ) != NAM_OK )
		{
			result = FALSE;
		}
		else
		{
			u32 numTicket = 0;

            (void)ES_GetTicketViews( gDataList[i].id, NULL, &numTicket);

            gDataList[i].commonTicketFlg = (eTicketType == ETICKET_TYPE_COMMON) ? TRUE : FALSE;
		}
    }
    
    return result;
}



#ifdef USER_APP_ONLY
/*---------------------------------------------------------------------------*
  Name:         getUserApplication

  Description:  全アプリからユーザーアプリだけを抽出する関数
 *---------------------------------------------------------------------------*/
static void getUserApplication(DataStruct* list)
{
    u32 i;
    s32 count = 0;

    DataStruct* p = list;
	DataStruct* buf;

	u32 buf_size = sizeof(DataStruct) * gNandAllAppNum;
    
    if( gNandAllAppNum != 0 )
    {
        buf = MyNAMUT_Alloc( buf_size );

        if( buf == NULL )
        {
            gErrorFlg = TRUE;
			return;
        }
    }
    
    for( i=0; i<gNandAllAppNum; i++, p++ )
    {
        if(!(p->id & TITLE_ID_APP_TYPE_MASK))
        {
			MI_CpuCopy8(p, &buf[count], sizeof(DataStruct));
            count++;
        }
    }
    
    // ユーザーアプリだけのリストをコピー
	MI_CpuCopy8(buf, list, buf_size);

    // アプリ総数の更新
    gNandAllAppNum = count;

    // バッファの開放
    MyNAMUT_Free(buf);
}
#endif


/*---------------------------------------------------------------------------*
   UTIL 関数
 *---------------------------------------------------------------------------*/
static void ConvertTitleIdLo(u8* code, u8* titleid_lo)
{
	u8 tmp[5];
	s32 i;
	
	for ( i=3; i>=0; i--, titleid_lo++ )
	{
		tmp[i] = *titleid_lo;
		*code = tmp[i];
	}

	// NULL 終端
	*code = 0x00;
}

static void ConvertGameCode(u8* code, u32 game_code)
{
	u8 tmp[5];
	s32 i;
	
	ConvertInitialCode(tmp, game_code);
	
	for ( i=3; i>=0; i--, code++)
	{
		*code = tmp[i];
	}
	
	// NULL 文字終端
	*code = 0x00;
}

static void ConvertInitialCode(u8* code, u32 titleid_lo)
{
	s32 i;
	
	for ( i=0; i<4; i++, code++)
	{
		*code = (u8)(titleid_lo >> (8 * (3-i)));
	}
	
	// NULL文字終端
	*code = 0x00;
}


void* MyNAMUT_Alloc(u32 size)
{
    const u32 allocSize = MATH_ROUNDUP32(size);
    SDK_ASSERTMSG( spAllocFunc != NULL, "NAMUT_Init should be called previously.\n");
    return spAllocFunc(allocSize);
}


static void* AllocForNAM(u32 size)
{
	void* ptr;
	ptr = OS_AllocFromMain(size);
	
	if (ptr == NULL)
	{
		OS_Panic("alloc failed.");
	}
	
	return ptr;
}


void MyNAMUT_Free(void* buffer)
{
    SDK_ASSERTMSG( spFreeFunc != NULL, "NAMUT_Init should be called previously.\n");
    if (buffer)
    {
        spFreeFunc(buffer);
    }
}


static void FreeForNAM(void* ptr)
{
	OS_FreeToMain(ptr);
}


void VBlankIntr(void)
{
    // テキスト表示を更新
    UpdateScreen();

    // IRQ チェックフラグ47をセット
    OS_SetIrqCheckFlag(OS_IE_V_BLANK);
}


/*---------------------------------------------------------------------------*
  End of file
 *---------------------------------------------------------------------------*/