rvtr/TwlToolsRED
1
Fork 0
mirror of https://github.com/rvtr/TwlToolsRED.git synced 2025-10-31 06:41:18 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

各ファイル毎のダイジェスト検証結果を出力するように変更

Browse Source 
git-svn-id: file:///Users/lillianskinner/Downloads/platinum/twl/TwlToolsRED@568 7061adef-622a-194b-ae81-725974e89856
This commit is contained in:
n1481 2011-07-07 05:34:31 +00:00
parent a438710879
commit d6c6f1bcff
5 changed files with 59 additions and 192 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

212
build/tools/TamperDetectorForSrl/card_hash.cpp
View File

@ -4,6 +4,7 @@
#include <openssl/hmac.h> /* libcrypto.a */
#include "twl_format_rom.h"
#include "card_hash.h"
#include "entry.h"
#define MATH_SHA1_DIGEST_SIZE (160/8)
@ -100,16 +101,10 @@ void CARDi_Init( CARDRomHashContext *context, RomHeader* header)
// rom image
context->buffer = (u8*)malloc( header->digest1_block_size);
// rom size
{
int i;
u32 rom_size;
for( i=0; i<header->rom_size; i++)
{
rom_size *= 2;
}
context->rom_size = (rom_size * 1024 / 8 * 1024);
}
// 検証OK/NG記録
context->hash_correct = (u8*)malloc( header->digest1_table_size / CARD_ROM_HASH_SIZE);
context->master_hash_correct = (u8*)malloc( header->digest1_table_size / header->digest2_covered_digest1_num / CARD_ROM_HASH_SIZE);
}
@ -142,9 +137,6 @@ u32 CARDi_GetHashSectorIndex(const CARDRomHashContext *context, u32 offset)
return offset / context->bytes_per_sector;
}
//static CARDRomHashBlock* CARDi_TouchRomHashBlock(CARDRomHashContext *context, u32 sector)
//{
//}
void CARDi_CheckHash(CARDRomHashContext *context, FILE* fp, u32 start, u32 size, RomHeader* header)
{
@ -213,8 +205,10 @@ bool Digest1Check(CARDRomHashContext *context, FILE* fp, RomHeader* header, u32
/* ROMデータのHashをDigest1と比較 */
if( !CARDi_CompareHash( context->hash, context->buffer, context->bytes_per_sector))
{
context->hash_correct[digest1_index] = 0; // 結果記録
ret = false;
}
context->hash_correct[digest1_index] = 1; // 結果記録
}
rest -= context->bytes_per_sector;
@ -227,6 +221,7 @@ bool Digest1Check(CARDRomHashContext *context, FILE* fp, RomHeader* header, u32
bool Digest2Check(CARDRomHashContext *context, FILE* fp, RomHeader* header)
{
bool ret = true;
int digest2_index;
int i, j;
int digest1_index_num = (header->digest1_table_size / header->digest1_block_size);
@ -238,11 +233,14 @@ bool Digest2Check(CARDRomHashContext *context, FILE* fp, RomHeader* header)
fread( &(context->hash[j * CARD_ROM_HASH_SIZE]), CARD_ROM_HASH_SIZE, 1, fp);
}
/* Digest1をDigest2と比較 */
if( !CARDi_CompareHash( &(context->master_hash[(i/context->sectors_per_block) * CARD_ROM_HASH_SIZE]),
digest2_index = (i/context->sectors_per_block);
if( !CARDi_CompareHash( &(context->master_hash[digest2_index * CARD_ROM_HASH_SIZE]),
context->hash, (CARD_ROM_HASH_SIZE * context->sectors_per_block)))
{
context->master_hash_correct[digest2_index] = 0;
ret = false;
}
context->master_hash_correct[digest2_index] = 1;
i+= header->digest2_covered_digest1_num;
}
@ -285,184 +283,26 @@ void CARD_CheckHash(CARDRomHashContext *context, RomHeader* header, FILE* fp)
printf( "-----------------------\n\n");
}
/*
static void CARDi_ReadRomHashImageDirect(CARDRomHashContext *context, void *buffer, u32 offset, u32 length)
/* 特定のファイルに対応するダイジェストテーブルが正しいか検証する */
void CARD_CheckFileDigest(CARDRomHashContext *context, MyFileEntry* file_entry, u8* ret_digest1, u8* ret_digest2)
{
const u32 sectunit = context->bytes_per_sector;
const u32 blckunit = context->sectors_per_block;
u32 position = offset;
u32 end = length + offset;
u32 sector = CARDi_GetHashSectorIndex(context, position);
long nowgfp;
FILE* gfp;
while (position < end)
u32 i;
u32 digest1_index_begin = CARDi_GetHashSectorIndex( context, file_entry->top);
u32 digest1_index_end = CARDi_GetHashSectorIndex( context, file_entry->bottom);
*ret_digest1 = 1;
*ret_digest2 = 1;
for( i=digest1_index_begin; i<=digest1_index_end; i++)
{
// 今回取得可能な最小単位のイメージを同期読み込み。
nowgfp = ftell( gfp);
fseek( gfp, position, SEEK_SET);
u32 available = fread( buffer, end - position, 1, gfp);
// 今回の検証中に必要となりそうなブロックを前もってアクセス。
// (void)CARDi_TouchRomHashBlock(context, sector);
// 取得したイメージの正当性を検証。
while (available >= sectunit)
if( !context->hash_correct[i])
{
CARDRomHashBlock *block = CARDi_TouchRomHashBlock(context, sector);
u32 slot = sector - block->index * blckunit;
while ((slot < blckunit) && (available >= sectunit))
{
// 必要ならここでブロック単位のハッシュテーブルを検証。
if (block != context->valid_block)
{
OSIntrMode bak_cpsr = OS_DisableInterrupts();
while (context->loading_block)
{
OS_SleepThread(NULL);
}
if (block == context->loaded_block)
{
context->loaded_block = block->next;
}
(void)OS_RestoreInterrupts(bak_cpsr);
CARDi_CompareHash(&context->master_hash[block->index * CARD_ROM_HASH_SIZE],
block->hash, CARD_ROM_HASH_SIZE * blckunit);
block->next = context->valid_block;
context->valid_block = block;
}
// イメージのハッシュを計算。
CARDi_CompareHash(&block->hash[slot * CARD_ROM_HASH_SIZE], buffer, sectunit);
position += sectunit;
available -= sectunit;
buffer = ((u8 *)buffer) + sectunit;
slot += 1;
sector += 1;
}
*ret_digest1 = 0;
}
}
}
*/
#if 0
/*---------------------------------------------------------------------------*
Name: CARDi_ReadRomHashImageDirect
Description: ハッシュコンテキストへキャッシュせずに転送先へ直接コピー。
Arguments: context : CARDRomHashContext構造体。
buffer : 転送先バッファ。(4バイト整合されている必要がある)
offset : アクセスするROMオフセット。
length : 転送サイズ。
Returns: None.
*---------------------------------------------------------------------------*/
static void CARDi_ReadRomHashImageDirect(CARDRomHashContext *context, void *buffer, u32 offset, u32 length)
{
const u32 sectunit = context->bytes_per_sector;
const u32 blckunit = context->sectors_per_block;
u32 position = offset;
u32 end = length + offset;
u32 sector = CARDi_GetHashSectorIndex(context, position);
while (position < end)
{
// 今回取得可能な最小単位のイメージを同期読み込み。
u32 available = (u32)(*context->ReadSync)(context->userdata, buffer, position, end - position);
// 今回の検証中に必要となりそうなブロックを前もってアクセス。
(void)CARDi_TouchRomHashBlock(context, sector);
// 次回に必要となる分の読み込み準備を要求。
if (context->ReadAsync && (position + available < end))
if( !context->master_hash_correct[i/context->sectors_per_block])
{
(void)(*context->ReadAsync)(context->userdata, NULL, position + available, end - (position + available));
}
// 取得したイメージの正当性を検証。
while (available >= sectunit)
{
CARDRomHashBlock *block = CARDi_TouchRomHashBlock(context, sector);
u32 slot = sector - block->index * blckunit;
while ((slot < blckunit) && (available >= sectunit))
{
// 必要ならここでブロック単位のハッシュテーブルを検証。
if (block != context->valid_block)
{
OSIntrMode bak_cpsr = OS_DisableInterrupts();
while (context->loading_block)
{
OS_SleepThread(NULL);
}
if (block == context->loaded_block)
{
context->loaded_block = block->next;
}
(void)OS_RestoreInterrupts(bak_cpsr);
CARDi_CompareHash(&context->master_hash[block->index * CARD_ROM_HASH_SIZE],
block->hash, CARD_ROM_HASH_SIZE * blckunit);
block->next = context->valid_block;
context->valid_block = block;
}
// イメージのハッシュを計算。
CARDi_CompareHash(&block->hash[slot * CARD_ROM_HASH_SIZE], buffer, sectunit);
position += sectunit;
available -= sectunit;
buffer = ((u8 *)buffer) + sectunit;
slot += 1;
sector += 1;
}
*ret_digest2 = 0;
}
}
}
/*---------------------------------------------------------------------------*
Name: MATH_CalcHMACSHA1
Description: HMAC-SHA-1 を計算する。
Arguments: digest HMAC-SHA-1 値を格納する場所へのポインタ
data 入力データのポインタ
dataLength 入力データ長
key 鍵のポインタ
keyLength 鍵の長さ
Returns: None.
*---------------------------------------------------------------------------*/
void MATH_CalcHMACSHA1(void *digest, const void *bin_ptr, u32 bin_len, const void *key_ptr, u32 key_len)
{
MATHSHA1Context context;
unsigned char hash_buf[ MATH_SHA1_DIGEST_SIZE ]; /* ハッシュ関数から得るハッシュ値 */
MATHiHMACFuncs hash2funcs = {
MATH_SHA1_DIGEST_SIZE,
(512/8),
};
hash2funcs.context = &context;
hash2funcs.hash_buf = hash_buf;
hash2funcs.HashReset = (void (*)(void*)) MATH_SHA1Init;
hash2funcs.HashSetSource = (void (*)(void*, const void*, u32)) MATH_SHA1Update;
hash2funcs.HashGetDigest = (void (*)(void*, void*)) MATH_SHA1GetHash;
MATHi_CalcHMAC(digest, bin_ptr, bin_len, key_ptr, key_len, &hash2funcs);
}
/*---------------------------------------------------------------------------*
Name: CARDi_CompareHash
Description: ハッシュによる正当性チェック。(HMAC-SHA1)
Arguments: hash : 比較基準となるハッシュ値
src : チェック対象のイメージ
len　: チェック対象のサイズ
Returns: None.
*---------------------------------------------------------------------------*/
static void CARDi_CompareHash(const void *hash, void *buffer, u32 length)
{
u8 tmphash[CARD_ROM_HASH_SIZE];
MATH_CalcHMACSHA1(tmphash, buffer, length, CARDiHmacKey, sizeof(CARDiHmacKey));
if (MI_CpuComp8(hash, tmphash, sizeof(tmphash)) != 0)
{
printf("ROM-hash comparation error!\n");
}
}
#endif

7
build/tools/TamperDetectorForSrl/card_hash.h
View File

@ -2,6 +2,8 @@
#ifndef CARD_HASH_H_
#define CARD_HASH_H_
#include "entry.h"
// 現時点でCARDライブラリが妥当と判断した定数。
// パフォーマンス計測の結果によって適宜変更してもよい。
@ -39,7 +41,6 @@ CARDRomRegion;
// セクタも別途リスト構造として管理する必要がある。
typedef struct CARDRomHashContext
{
u32 rom_size;
// ROMヘッダから取得する基本設定
CARDRomRegion area_ntr;
CARDRomRegion area_ltd;
@ -77,6 +78,9 @@ typedef struct CARDRomHashContext
*/
u8 *buffer;
u8 *hash;
u8 *master_hash_correct;
u8 *hash_correct;
}
CARDRomHashContext;
@ -85,6 +89,7 @@ void CARDi_Init( CARDRomHashContext *context, RomHeader* header);
void CARDi_CheckHash(CARDRomHashContext *context, FILE* fp, u32 sect, u32 size, RomHeader* header);
bool Digest2Check(CARDRomHashContext *context, FILE* fp, RomHeader* header);
void CARD_CheckHash(CARDRomHashContext *context, RomHeader* header, FILE* fp);
void CARD_CheckFileDigest(CARDRomHashContext *context, MyFileEntry* file_entry, u8* ret_digest1, u8* ret_digest2);
#endif //CARD_HASH_H_

23
build/tools/TamperDetectorForSrl/checker.cpp
View File

@ -3,6 +3,7 @@
#include "checker.h"
//#include "nitro_romheader.h"
#include "twl_format_rom.h"
#include "card_hash.h"
extern Entry gEntry;
extern Entry mEntry;
@ -844,7 +845,7 @@ void Checker::ExportGenuineBmpFiles( Entry* gEntry, PrintLevel print_enable)
}
/* ディレクトリとファイルをチェックする */
void Checker::CheckAllEntries( Entry* gEntry, Entry* mEntry)
void Checker::CheckAllEntries( CARDRomHashContext *context, Entry* gEntry, Entry* mEntry)
{
{
MyDirEntry *currentEntry = gEntry->dirEntry;
@ -895,6 +896,26 @@ void Checker::CheckAllEntries( Entry* gEntry, Entry* mEntry)
false, PRINT_LEVEL_1) == false)
{
currentEntry->modified = true; // 改竄フラグ
}
{
u8 d1, d2;
CARD_CheckFileDigest( context, hisEntry, &d1, &d2);
if( d1)
{
printf( "（ファイル - digest1 検証はOK）\n");
}
else
{
printf( "（ファイル - digest1 検証はNG）\n");
}
if( d2)
{
printf( "（digest1 - digest2 検証はOK）\n");
}
else
{
printf( "（digest1 - digest2 検証はNG）\n");
}
printf( "\n");
}
}

3
build/tools/TamperDetectorForSrl/checker.h
View File

@ -10,6 +10,7 @@
#include "twl_format_rom.h"
#include "banner.h"
#include "entry.h"
#include "card_hash.h"
typedef struct
{
@ -62,7 +63,7 @@ class Checker
bool FindEntry( u32 fnt_offset, u16 entry_id, RomHeader* headerBuf, FILE* fp, Entry* entry, u16 parent_id, PrintLevel print_enable);
void FindAllocation( u16 entry_id, RomHeader* headerBuf, FILE* fp, Entry* entry, PrintLevel print_enable);
void CheckAllEntries( Entry* gEntry, Entry* mEntry);
void CheckAllEntries( CARDRomHashContext *context, Entry* gEntry, Entry* mEntry);
/* ROM内のBMPファイルを全て切り出して出力する */
void ExportGenuineBmpFiles( Entry* gEntry, PrintLevel print_enable);

6
build/tools/TamperDetectorForSrl/main.cpp
View File

@ -59,6 +59,8 @@ bool int_bits(void)
int main (int argc, char *argv[])
{
CARDRomHashContext context;
// 処理系の unsignedビット数が想定外ならエラー終了（types.hを変更してビルドし直してください）
if( !int_bits())
{
@ -91,8 +93,6 @@ int main (int argc, char *argv[])
// ダイジェスト検証(digest1, digest2)
{
CARDRomHashContext context;
CARDi_Init( &context, &mHeaderBuf);
CARD_CheckHash( &context, &mHeaderBuf, mfp);
}
@ -116,7 +116,7 @@ int main (int argc, char *argv[])
{
mEntry.FollowParent();
mEntry.AutoSetFullPath();
checker.CheckAllEntries( &gEntry, &mEntry);
checker.CheckAllEntries( &context, &gEntry, &mEntry);
}
checker.ExportGenuineBmpFiles( &gEntry, PRINT_LEVEL_0);
/*