/*---------------------------------------------------------------------------*
  Project:  Horizon
  File:     main.cpp

  Copyright (C)2009 Nintendo Co., Ltd.  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.

  $Rev$
 *---------------------------------------------------------------------------*/

#include <nn.h>
#include <nn/fs.h>
#include <nn/cfg/CTR/cfg_ApiSys.h>
#include <nn/cfg/CTR/cfg_ApiInit.h>
#include <nn/am/am_ApiLocalImporter.h>
#include <nn/am/am_ApiSystemMenu.h>
#include <nn/applet.h>
#include <nn/ndm.h>
#include <nn/crypto/crypto_RsaKey.h>
#include <nn/ps/ctr/ps_Api.h>
#include <stdio.h>
#include <string.h>
#include "demo.h"
#include "../common/ver.h"
#include "../common/sleep.h"
#include "../common/common.h"
#include "../common/shfnt.h"
#include "test_data.h"

#define INF_FILE "sdmc:/csm_inf.txt"

#define KEY_LENGTH 2048
#define LIST_FILE "sdmc:/csm_list.txt"
#define SIGN_FILE "sdmc:/csm_sign.dat"
//#define PUBK_FILE "sdmc:/csm_key_public.der"
#define PRVK_FILE "sdmc:/csm_key_private.der"


extern "C" {
    extern u8* PUBLIC_KEY_BEGIN[];
    extern u8* PUBLIC_KEY_END[];

    const void* PUBLIC_KEY      = PUBLIC_KEY_BEGIN;
}

demo::RenderSystemDrawing s_RenderSystem;
nn::fnd::ExpHeap appHeap;
uptr heapForGx;


namespace
{
    bit8 readBuf[64 * 1024];
}

bool ImportFile(nn::fs::MediaType mediaType, wchar_t* filename)
{
    nn::Result result;

    nn::fs::FileOutputStream* stream;
    result = nn::am::BeginImportProgram(&stream, mediaType);
    if (result.IsSuccess())
    {
        //NN_LOG("Importing: %ls...", filename);
        nn::fs::FileInputStream in(filename);

        while(s32 read = in.Read(readBuf, sizeof(readBuf)))
        {
            stream->Write(readBuf, read);
        }
        result = nn::am::EndImportProgram(stream);
    }
    return result.IsSuccess();
}


u8 s_list[200][nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN+1] NN_ATTRIBUTE_ALIGN(32);//シリアルリスト
u8 sign[1024*10] NN_ATTRIBUTE_ALIGN(32);//署名
//u8 key[1024*10] NN_ATTRIBUTE_ALIGN(32);//署名
u8 serialNo[nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN+1];

bool GetSerial(){
    nn::cfg::CTR::init::Initialize();
    nn::cfg::CTR::system::Initialize();
    nn::Result result = nn::cfg::CTR::system::GetSerialNo(serialNo);
    serialNo[nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN] = 0;
    nn::cfg::CTR::system::Finalize();
    nn::cfg::CTR::init::Finalize();
    return result.IsSuccess();
}



//#define setColor(r,g,b,a) s_RenderSystem.SetColor(r,g,b,a)
void setColor(f32 r,f32 g,f32 b,f32 a){ shf_SetColor(r,g,b,a);}
//#define setText(x,y,s) s_RenderSystem.DrawText(x,y,s)
void drawText(u16 x,u16 y,char *s){ shf_DrawText_0( x,y,s);}
//上はASCIIのみ対応したTextWriterへのラッパ
//日本語表示の場合、WideTextWriterを使う(要sft_側の関数追加)


nn::hid::PadReader *hpr;
//---------------------------------------------------------------- 入力待ち
nn::hid::PadStatus padStatus;
u32 WaitKey(u32 mask)
{

    while(1){
        hpr->ReadLatest(&padStatus);
        if(padStatus.trigger & mask)return padStatus.trigger;
        CheckSysBreak();
    }
}


void endfunc()
{
    nn::hid::Finalize();
    SharedFontFinalize();
    nngxWaitVSync(NN_GX_DISPLAY_BOTH);//SDK2.0以降では不要かも？
    s_RenderSystem.Finalize();
}

//署名検証
bool VeriRsa(u8* src,size_t sz,u8* sig,u8* pub)
{
    nn::crypto::RsaKey key;
    //nn::crypto::Initialize();
    nn::ps::CTR::Initialize();
    nn::Result res = key.InitializePublicKey(pub,KEY_LENGTH);
    NN_LOG("inipubkey desc= %d\n",res.GetDescription());
    if (res.IsSuccess())
    {
        res = nn::ps::CTR::VerifyRsaSha256(src,sz,sig,key);
        NN_LOG("verisig desc= %d\n",res.GetDescription());
    }
    //nn::crypto::Finalize();
    nn::ps::CTR::Finalize();
    return res.IsSuccess();
}

bool VeriRsa(u8* src,size_t sz,u8* sig)
{
    nn::crypto::RsaKey key;
    //nn::crypto::Initialize();
    nn::ps::CTR::Initialize();
    nn::Result res = key.InitializePublicKey(PUBLIC_KEY_BEGIN,KEY_LENGTH);
    NN_LOG("inipubkey desc= %d\n",res.GetDescription());
    if (res.IsSuccess())
    {
        res = nn::ps::CTR::VerifyRsaSha256(src,sz,sig,key);
        NN_LOG("verisig desc= %d\n",res.GetDescription());
    }
    //nn::crypto::Finalize();
    nn::ps::CTR::Finalize();
    return res.IsSuccess();
}



void ErrorStop(char *s)
{
    shf_SetScale(0.7,0.7);
    setColor(1.0, 0.0, 0.0,1.0);
    drawText(10,100,s);
    drawText(10,140,"Push Power Button & Power Off");
    s_RenderSystem.SwapBuffers();
    WaitKey(0);
}


nn::fs::FileInputStream fi;
//ファイルリード
s32 FileRead(char* fname)
{
    s32 size;
    nn::Result result = fi.TryInitialize(fname);
    if (result.IsFailure()){//open error
        //NN_LOG("desc= %d\n",result.GetDescription());
        return 0;
    }
    result = fi.TryRead(&size,readBuf,sizeof(readBuf));
    fi.Finalize();
    if (result.IsFailure()){
        //NN_LOG("desc= %d\n",result.GetDescription());
        return 0;
    }
    return size;
}


char ver[16]; 
char seri[32]; 
void nnMain( void )
{
    nn::Result result;
    bool flg_applove=false;
    bool flg_test=false;
    
    nn::os::Initialize();
    nn::fs::Initialize();

    //DEA-SUPにて推奨のフリーズ暫定対策：無線デーモンを停止 (2011.3.1 現在)
    //ただし、スリープ時の"いつのまに通信"は止まらない
    //無線は使わないので本体横スイッチ切っとくのが確実
    nn::ndm::SuspendScheduler();

    //中断処理の準備
    InitSysBreak((uptr)endfunc);
    // グラフィックスライブラリの初期化は、以降で行わなければならない
    // 他、アプリケーションの初期化処理

    nn::hid::Initialize();
    nn::hid::PadReader padReader;
    hpr = &padReader;

    //result = nn::ns::CTR::InitializeForShell();
    //if (result.IsSuccess()) 
    result = nn::am::InitializeForLocalImporter();

    nn::applet::DisableSleep();//スリープ非対応

    const size_t ROMFS_BUFFER_SIZE = 1024 * 64;
    static char buffer[ROMFS_BUFFER_SIZE];

    // ヒープの確保
    appHeap.Initialize(nn::os::GetDeviceMemoryAddress(), nn::os::GetDeviceMemorySize(), nn::os::ALLOCATE_OPTION_LINEAR);

    const u32 s_GxHeapSize = 0x800000;
    // RenderSystem の準備
    heapForGx = reinterpret_cast<uptr>(appHeap.Allocate(s_GxHeapSize));
    s_RenderSystem.Initialize(heapForGx, s_GxHeapSize);
    SharedFontInit();

    // 共有フォントの種類を取得
    //nn::pl::SharedFontType sftype = nn::pl::GetSharedFontType();

    // スリープ要求に対する返答を有効にする
    // また、蓋の状態チェックを行い蓋が閉じられているならスリープ要求が発生する
//    nn::applet::EnableSleep(true);

    s_RenderSystem.SetRenderTarget(NN_GX_DISPLAY0);
    s_RenderSystem.Clear();
//    s_RenderSystem.SetFontSize(12);
    shf_SetScale(0.8,0.6);
    setColor(1.0,1.0,1.0,1.0);
    drawText(20,20,"CTR Card Savedata Mover Setup");
    s_RenderSystem.SwapBuffers();

    shf_SetScale(0.5,0.5);
    sprintf(ver,"version %.2f",VERSION);
    drawText(250,40,ver);

    shf_SetScale(0.7,0.7);
    if (result.IsFailure())ErrorStop("Initialize Error");//am 初期化に失敗

//シリアルチェック
    if( GetSerial() )
    {//シリアル取得成功
        result = nn::fs::MountSdmc();
        if (result.IsSuccess())
        {
            int i,j,n;
            s32 fsize = FileRead(INF_FILE);
            if(fsize > 0)//設定ファイルがある
            {
                n = 0;
                while(n < fsize)
                {
                    if (readBuf[n++] == '-'){
                        switch (readBuf[n++]){
                            case 'p':// -p ..  本体シリアル表示
                                strcpy(seri,"s/n: ");
                                strcat(seri,(char *)serialNo);
                                shf_SetScale(0.5,0.5);
                                drawText(10,40,(char*)seri);
                                shf_SetScale(0.7,0.7);
                                break;
                            case 't':// -t ..  テストメニュー有効
                                flg_test = true;
                                break;
                        }
                    }
                }
            }

    
            fsize = FileRead(SIGN_FILE);
            if(fsize > 0)//署名リード成功
            {
                if (fsize>sizeof(sign))fsize = sizeof(sign);
                memcpy(sign,readBuf,fsize);

                //fsize = FileRead(PUBK_FILE);
                //if(fsize > 0)//公開鍵リード成功
                {
                    //if (fsize>sizeof(key))fsize = sizeof(key);
                    //memcpy(key,readBuf,fsize);
                    
                    fsize = FileRead(LIST_FILE);
                    if(fsize > 0)//リストリード成功
                    {

                        //if (VeriRsa(readBuf,fsize,sign,key))
                        if (VeriRsa(readBuf,fsize,sign))
                        {//署名検証OK
                            int list_ct = 0;
                            n=0;
                            u8 d;
                            while(n < fsize)//リスト作成
                            {
                                if ((fsize-n) > nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN)j = nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN;
                                else j = fsize-n;
                                for (i=0;i<j;i++)
                                {
                                    d = readBuf[n++];
                                    if ((d==0x0d)||(d==0x0a))
                                    {
                                       if (readBuf[n]==0x0a)n++;
                                       break;
                                    }
                                    s_list[list_ct][i] = d;
                                }
                                if (i>0){
                                    s_list[list_ct][i] = 0;//終端
                                    list_ct++;
                                }
                            }
                            //本体シリアルがリストにあるかチェック
                            n =0;
                            int len = strlen((char*)serialNo);
                            if (len > nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN)len = nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN;
                            while(n<list_ct){
                                for(i=0;i<len;i++)
                                {
                                    if (serialNo[i] != s_list[n][i])break;
                                }
                                if (i == len)
                                {
                                    flg_applove = true;
                                    break;
                                }
                            }
                        }//Verify
                    }//LIST_FILE
                }//PUBK_FILE
            }//SIGN_FILE
            nn::fs::Unmount("sdmc:");
        }//MountSdmc
    }//GetSerial

    nn::fs::MountRom(16, 16, buffer, ROMFS_BUFFER_SIZE);

    u32 mask = 0;
    
    if (flg_applove)//インポート許可本体
    {
        drawText(80,100,"Push X : Import");
        drawText(80,140,"Push Y : Delete");
        mask |= nn::hid::BUTTON_X | nn::hid::BUTTON_Y;
    }

    if (flg_test)//テストメニュー
    {
        drawText(80,160,"Push R : Test Data");
        mask |= nn::hid::BUTTON_R;
    }
    s_RenderSystem.SwapBuffers();

    if (mask == 0)ErrorStop("Serial number Check Failed");

    u32 trg = WaitKey(mask);
    if (trg & nn::hid::BUTTON_R){
        shf_SetScale(0.6,0.6);
        TestMain();//テストデータへ
    }
    shf_SetScale(0.8,0.8);
    // CIA のプログラム ID を取得する。
    nn::am::ProgramInfo programInfo;
    if (nn::am::GetProgramInfoFromCia(&programInfo, ROMFS_IMPORTEE_PATH).IsFailure())
    {
        s_RenderSystem.Clear();
        setColor(1.0, 0.0, 0.0,1.0);
        drawText(10,50,"cia infomation Error");
    
    } else
    {
      s_RenderSystem.Clear();
      if(padStatus.trigger & nn::hid::BUTTON_X)
          drawText(10,120,"Importing");
          else drawText(10,120,"Delete");
      s_RenderSystem.SwapBuffers();

     // 既に存在するものをインポートするとエラーが返ってくるので、あらかじめ消しておく。
      nn::am::DeleteProgram(nn::fs::MEDIA_TYPE_NAND, programInfo.id);
     // タイトル鍵が違うケースに対応するために、チケットも消しておく。
      nn::am::DeleteTicket(programInfo.id);

      if(padStatus.trigger & nn::hid::BUTTON_X)
      {
        // CIA をインポート
        if (ImportFile(nn::fs::MEDIA_TYPE_NAND, ROMFS_IMPORTEE_PATH)==false)
        {
            s_RenderSystem.Clear();
            setColor(1.0, 0.0, 0.0,1.0);
            drawText(10,50,"Impoprt fail");
            s_RenderSystem.SwapBuffers();
        }
      }
      s_RenderSystem.Clear();
      drawText(10,50,"Complate");
    }

      drawText(10,120,"Push Power Button & Power Off");
      s_RenderSystem.SwapBuffers();

      WaitKey(0);

}