mirror of
https://github.com/rvtr/TwlIPL.git
synced 2025-10-31 06:01:12 -04:00
9a82c1cc2b
git-svn-id: file:///Users/lillianskinner/Downloads/platinum/twl/TwlIPL/trunk@1426 b08762b0-b915-fc4b-9d8c-17b2551a87ff
286 lines
8.3 KiB
C
286 lines
8.3 KiB
C
/*---------------------------------------------------------------------------*
|
|
Project: TwlIPL - tests - CheckDSHashTable
|
|
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:: $
|
|
$Rev$
|
|
$Author$
|
|
*---------------------------------------------------------------------------*/
|
|
|
|
#include <twl.h>
|
|
#include <sysmenu/dht/dht.h>
|
|
|
|
#define HASH_PATH "/sign/DSHashTable.bin"
|
|
|
|
/*
|
|
このサンプルはテスト用であり、CARD関数の代わりにFS関数で
|
|
DHTライブラリが実装されている前提である
|
|
*/
|
|
|
|
#define ROM_PATH "/srl"
|
|
|
|
#define DS_HASH_TABLE_NUMS 3000
|
|
#define DS_HASH_TABLE_SIZE (sizeof(DHTHeader) + DS_HASH_TABLE_NUMS * sizeof(DHTDatabase))
|
|
|
|
/*
|
|
DHTファイル読み込み先
|
|
*/
|
|
static u8 dht_buffer[DS_HASH_TABLE_SIZE] ATTRIBUTE_ALIGN(32);
|
|
static DHTFile *const dht = (DHTFile*)dht_buffer;
|
|
/*
|
|
Phase2用バッファ
|
|
*/
|
|
static DHTPhase2Work p2work;
|
|
|
|
/*
|
|
実際にはアドレス固定
|
|
*/
|
|
#include <twl/ltdmain_begin.h>
|
|
static ROM_Header_Short rom_header ATTRIBUTE_ALIGN(32); // 使うのは DHT_DS_HEADER_SIZE だけ
|
|
static u8 rom_arm9[3*1024*1024] ATTRIBUTE_ALIGN(32);
|
|
static u8 rom_arm7[1024*1024] ATTRIBUTE_ALIGN(32);
|
|
#include <twl/ltdmain_end.h>
|
|
|
|
/*
|
|
デバイスアクセスAPI
|
|
*/
|
|
#if 1
|
|
// FS版 (arg == FSFile*)
|
|
static BOOL ReadImage(void* dest, s32 offset, s32 length, void* arg)
|
|
{
|
|
FSFile* fp = arg;
|
|
s32 result;
|
|
if ( !FS_SeekFile(fp, offset, FS_SEEK_SET) )
|
|
{
|
|
OS_TPrintf("Cannot seek to the offset (%d bytes).\n", offset);
|
|
return FALSE;
|
|
}
|
|
result = FS_ReadFile(fp, dest, length);
|
|
if ( result != length )
|
|
{
|
|
OS_TPrintf("Cannot read the data (%d bytes).\n", length);
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
static BOOL ReadImageEx(SVCHMACSHA1Context* ctx, s32 offset, s32 length, void* arg)
|
|
{
|
|
if ( !ReadImage(p2work.buffer, offset, length, arg) )
|
|
{
|
|
return FALSE;
|
|
}
|
|
DHT_CheckHashPhase2ExUpdate(ctx, p2work.buffer, length);
|
|
return TRUE;
|
|
}
|
|
#else
|
|
#ifdef SDK_ARM9
|
|
#define PAGE_SIZE 512
|
|
// CARD版 (arg == dma no)
|
|
static BOOL ReadImage(void* dest, s32 offset, s32 length, void* arg)
|
|
{
|
|
u32 dma = (u32)arg;
|
|
CARD_ReadRom(dma, (void*)offset, dest, (u32)length);
|
|
return TRUE;
|
|
}
|
|
#else
|
|
// HOTSW版 (arg == CardBootData* cdb)
|
|
#include <hotswTypes.h>
|
|
#include <dsCardCommon.h>
|
|
static BOOL ReadImage(void* dest, s32 offset, s32 length, void* arg)
|
|
{
|
|
HotSwState retval;
|
|
if ( offset % 512 )
|
|
{
|
|
static u8 page_buffer[512];
|
|
u32 page_offset = (u32)(offset & -512);
|
|
u32 buffer_offset = (u32)(offset % 512);
|
|
u32 valid_length = 512 - buffer_offset;
|
|
retval = ReadPageGame((CardBootData*)arg, page_offset, page_buffer, 512);
|
|
if (retval != HOTSW_SUCCESS)
|
|
{
|
|
return FALSE;
|
|
}
|
|
MI_CpuCopy8(page_buffer + buffer_offset, dest, (length < valid_length ? length : valid_length));
|
|
dest = (u8*)dest + valid_length;
|
|
offset += valid_length;
|
|
length -= valid_length;
|
|
if ( length < 0)
|
|
{
|
|
return TRUE;
|
|
}
|
|
}
|
|
retval = ReadPageGame((CardBootData*)arg, (u32)offset, dest, (u32)length);
|
|
return (retval == HOTSW_SUCCESS);
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
static void VBlankIntr( void )
|
|
{
|
|
OS_SetIrqCheckFlag(OS_IE_V_BLANK);
|
|
}
|
|
|
|
/*
|
|
検証本番のうち、実際にはオーバーヘッドにならない処理
|
|
*/
|
|
static BOOL CheckValidation(FSFile* fp)
|
|
{
|
|
const DHTDatabase *db;
|
|
s32 result;
|
|
|
|
// ヘッダの読み込み
|
|
result = FS_ReadFile(fp, &rom_header, DHT_DS_HEADER_SIZE);
|
|
if ( result != DHT_DS_HEADER_SIZE )
|
|
{
|
|
OS_TPrintf("Cannot read ROM header.\n");
|
|
return FALSE;
|
|
}
|
|
// ARM9部分
|
|
if ( rom_header.main_size > sizeof(rom_arm9) )
|
|
{
|
|
OS_TPrintf("Too large main size (%d > %d).\n", rom_header.main_size, sizeof(rom_arm9));
|
|
return FALSE;
|
|
}
|
|
if ( !FS_SeekFile(fp, (s32)rom_header.main_rom_offset, FS_SEEK_SET) )
|
|
{
|
|
OS_TPrintf("Cannot seek to ARM9 static.\n");
|
|
return FALSE;
|
|
}
|
|
result = FS_ReadFile(fp, rom_arm9, (s32)rom_header.main_size);
|
|
if ( result != rom_header.main_size )
|
|
{
|
|
OS_TPrintf("Cannot read ARM9 static.\n");
|
|
return FALSE;
|
|
}
|
|
// ARM7部分
|
|
if ( rom_header.sub_size > sizeof(rom_arm7) )
|
|
{
|
|
OS_TPrintf("Too large sub size (%d > %d).\n", rom_header.sub_size, sizeof(rom_arm7));
|
|
return FALSE;
|
|
}
|
|
if ( !FS_SeekFile(fp, (s32)rom_header.sub_rom_offset, FS_SEEK_SET) )
|
|
{
|
|
OS_TPrintf("Cannot seek to ARM7 static.\n");
|
|
return FALSE;
|
|
}
|
|
result = FS_ReadFile(fp, rom_arm7, (s32)rom_header.sub_size);
|
|
if ( result != rom_header.sub_size )
|
|
{
|
|
OS_TPrintf("Cannot read ARM7 static.\n");
|
|
return FALSE;
|
|
}
|
|
|
|
// データベースの検索
|
|
db = DHT_GetDatabase(dht, &rom_header);
|
|
if ( !db )
|
|
{
|
|
return FALSE;
|
|
}
|
|
//OS_TPrintf("FOUND: 0x%08X: %.4s(%d)\n", db, db->game_code, db->rom_version);
|
|
// ハッシュ計算 (1) - 隠蔽可能なはず
|
|
if ( !DHT_CheckHashPhase1(db->hash[0], &rom_header, rom_arm9, rom_arm7) )
|
|
{
|
|
return FALSE;
|
|
}
|
|
// ハッシュ計算 (2) - 隠蔽は難しいか
|
|
if ( !DHT_CheckHashPhase2(db->hash[1], &rom_header, &p2work, ReadImage, fp) )
|
|
{
|
|
// ハッシュ計算 (2ex) - 隠蔽は難しいか (失敗した場合も見ておく)
|
|
if ( !DHT_CheckHashPhase2Ex(db->hash[1], &rom_header, (DHTPhase2ExWork*)&p2work, ReadImage, ReadImageEx, fp) )
|
|
{
|
|
return FALSE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
// ハッシュ計算 (2ex) - 隠蔽は難しいか
|
|
if ( !DHT_CheckHashPhase2Ex(db->hash[1], &rom_header, (DHTPhase2ExWork*)&p2work, ReadImage, ReadImageEx, fp) )
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
void TwlMain(void)
|
|
{
|
|
OS_Init();
|
|
OS_InitTick();
|
|
|
|
(void)OS_EnableIrq();
|
|
(void)OS_EnableInterrupts();
|
|
|
|
// 割り込み許可----------------------------
|
|
(void)OS_SetIrqFunction(OS_IE_V_BLANK, VBlankIntr);
|
|
(void)OS_EnableIrqMask(OS_IE_V_BLANK);
|
|
(void)GX_VBlankIntr(TRUE);
|
|
|
|
FS_Init(FS_DMA_NOT_USE);
|
|
{
|
|
FSFile file;
|
|
if ( !FS_OpenFileEx(&file, HASH_PATH, FS_FILEMODE_R) )
|
|
{
|
|
OS_TPanic("Cannot open %s.\n", HASH_PATH);
|
|
}
|
|
// 署名ロード
|
|
if ( !DHT_PrepareDatabase(dht, &file) )
|
|
{
|
|
OS_TPanic("Cannot prepare the database.\n");
|
|
}
|
|
FS_CloseFile(&file);
|
|
}
|
|
|
|
// 本番
|
|
{
|
|
FSFile dir;
|
|
FSDirectoryEntryInfo info;
|
|
BOOL result;
|
|
result = FS_OpenDirectory(&dir, ROM_PATH, FS_FILEMODE_R);
|
|
if ( !result )
|
|
{
|
|
OS_TPanic("Cannot open " ROM_PATH ".\n");
|
|
}
|
|
while ( FS_ReadDirectory(&dir, &info) )
|
|
{
|
|
FSFile file;
|
|
OSTick begin;
|
|
char path[FS_ENTRY_LONGNAME_MAX+sizeof(ROM_PATH)+1];
|
|
if ( info.attributes & FS_ATTRIBUTE_IS_DIRECTORY )
|
|
{
|
|
OS_TPrintf("%s is directory.\n", info.longname);
|
|
continue;
|
|
}
|
|
STD_CopyString(path, ROM_PATH "/");
|
|
STD_ConcatenateString(path, info.longname);
|
|
if (!FS_OpenFileEx(&file, path, FS_FILEMODE_R))
|
|
{
|
|
OS_TPrintf("Cannot open %s.\n", path);
|
|
continue;
|
|
}
|
|
OS_TPrintf("\nTrying %s...\n", path);
|
|
begin = OS_GetTick();
|
|
if ( !CheckValidation(&file) )
|
|
{
|
|
OS_TPrintf("Failed. %d msec (includes loading static data).\n", (int)OS_TicksToMilliSeconds(OS_GetTick()-begin));
|
|
}
|
|
else
|
|
{
|
|
OS_TPrintf("Success. %d msec (includes loading static data).\n", (int)OS_TicksToMilliSeconds(OS_GetTick()-begin));
|
|
}
|
|
FS_CloseFile(&file);
|
|
}
|
|
FS_CloseDirectory(&dir);
|
|
}
|
|
|
|
OS_TPrintf("\nDone.\n");
|
|
OS_Terminate();
|
|
}
|
|
|