mirror of
https://github.com/rvtr/ctr_Repair.git
synced 2025-10-31 13:51:08 -04:00
419 lines
14 KiB
C++
419 lines
14 KiB
C++
/*---------------------------------------------------------------------------*
|
||
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
|
||
{
|
||
u8* readBuf;
|
||
}
|
||
|
||
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();
|
||
}
|
||
|
||
#define LIST_MAX 4000
|
||
u8 s_list[LIST_MAX][nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN+1] NN_ATTRIBUTE_ALIGN(32);//シリアルリスト
|
||
u8 sign[256] 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)
|
||
{
|
||
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 初期化に失敗
|
||
|
||
//ファイルバッファ
|
||
const size_t READ_BUFFER_SIZE = 64*1024;
|
||
if (READ_BUFFER_SIZE > appHeap.GetTotalFreeSize())ErrorStop("memory allocate fail");
|
||
readBuf = reinterpret_cast<u8*>(appHeap.Allocate(READ_BUFFER_SIZE,32));
|
||
|
||
|
||
//シリアルチェック
|
||
if( GetSerial() )
|
||
{//シリアル取得成功
|
||
result = nn::fs::MountSdmc();
|
||
if (result.IsSuccess())
|
||
{
|
||
int i,j,n;
|
||
s32 fsize = FileRead(INF_FILE);//設定ファイルがあるか
|
||
if((fsize < 1024) && (fsize > 0))//サイズチェック
|
||
{
|
||
n = 0;
|
||
while(n < fsize)
|
||
{
|
||
if (readBuf[n] == '-'){
|
||
if (n == fsize-1)break;//終端
|
||
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;
|
||
}
|
||
}
|
||
n++;
|
||
}
|
||
}
|
||
|
||
|
||
fsize = FileRead(SIGN_FILE);//署名リード
|
||
if(fsize == 256)//サイズチェック
|
||
{
|
||
memcpy(sign,readBuf,256);//readBuf -> sign
|
||
|
||
{
|
||
|
||
fsize = FileRead(LIST_FILE);//リストリード
|
||
if((fsize < LIST_MAX * (nn::cfg::CTR::CFG_SECURE_INFO_SERIAL_NO_LEN+1)) && (fsize > 0))//サイズチェック
|
||
{
|
||
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];
|
||
n++;
|
||
if ((d==0x0d)||(d==0x0a)) //改行コード
|
||
{
|
||
if (readBuf[n]==0x0a)n++;//CR+LF,LF+LF
|
||
else ErrorStop("list broken");//CRのみは異常、署名時にファイル壊れてた？
|
||
break;
|
||
}
|
||
s_list[list_ct][i] = d;
|
||
}
|
||
if (i>0){
|
||
s_list[list_ct][i] = 0;//終端
|
||
list_ct++;
|
||
if (list_ct>=LIST_MAX)
|
||
{//上限オーバー: おそらくリスト作成ミス
|
||
ErrorStop("List too Long");
|
||
}
|
||
}
|
||
}
|
||
//本体シリアルがリストにあるかチェック
|
||
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_LOG("result = %d\n",result.GetDescription());
|
||
|
||
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();//テストデータへ
|
||
}
|
||
|
||
nn::fs::MountRom(16, 16, buffer, ROMFS_BUFFER_SIZE);
|
||
|
||
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(trg & 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(trg & 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);
|
||
|
||
}
|
||
|