mirror of
https://github.com/rvtr/twl_wrapsdk.git
synced 2025-10-31 06:11:10 -04:00
4aec244198
git-svn-id: file:///Users/lillianskinner/Downloads/platinum/twl/twl_wrapsdk/trunk@115 4ee2a332-4b2b-5046-8439-1ba90f034370
338 lines
10 KiB
C
338 lines
10 KiB
C
/*---------------------------------------------------------------------------*
|
||
Project: NitroSDK - AES - demos - aes-2
|
||
File: main.c
|
||
|
||
Copyright 2007 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.
|
||
|
||
$Log: main.c,v $
|
||
$NoKeywords: $
|
||
*---------------------------------------------------------------------------*/
|
||
#include <twl.h>
|
||
|
||
#define INPUT_DMA 4
|
||
#define OUTPUT_DMA 5
|
||
|
||
static const u128 key = {
|
||
0x01234567,
|
||
0x89abcdef,
|
||
0x01234567,
|
||
0x89abcdef
|
||
};
|
||
static const u128 iv = {
|
||
0x89abcdef,
|
||
0x01234567,
|
||
0x89abcdef,
|
||
0x01234567
|
||
};
|
||
static const u32 gs_data[] ATTRIBUTE_ALIGN(32) = {
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567,
|
||
0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567, 0x89abcdef, 0x01234567
|
||
};
|
||
|
||
static u8 dataA[sizeof(gs_data) + 32] ATTRIBUTE_ALIGN(32);
|
||
static u8 dataB[sizeof(gs_data) + 32] ATTRIBUTE_ALIGN(32);
|
||
|
||
//================================================================================
|
||
static void dump(const char *str, void *ptr, u32 length)
|
||
{
|
||
u8 *data = (u8*)ptr;
|
||
int i;
|
||
OS_TPrintf("\n[%s] (%d bytes):\n\t", str, length);
|
||
for (i = 0; i < length; i++) {
|
||
OS_TPrintf("%02X", *data++);
|
||
if ((i & 0xF) == 0xF) OS_TPrintf("\n\t");
|
||
else OS_TPrintf(" ");
|
||
}
|
||
OS_TPrintf("\n");
|
||
}
|
||
|
||
#define TEST0_USE_DMA_INPUT
|
||
#define TEST0_USE_DMA_OUTPUT
|
||
static void test0(void)
|
||
{
|
||
OSTick begin;
|
||
AESResult result;
|
||
|
||
OS_TPrintf("\n%s() is starting.\n\n", __func__);
|
||
|
||
// <20>Ƃ肠<C682><E882A0><EFBFBD><EFBFBD><EFBFBD>A<EFBFBD><41><EFBFBD>Z<EFBFBD>b<EFBFBD>g<EFBFBD><67><EFBFBD><EFBFBD> (<28><><EFBFBD>W<D68C>͉<EFBFBD><CD89><EFBFBD><EFBFBD>ς<EFBFBD><CF82><EFBFBD><EFBFBD>Ȃ<EFBFBD> (KEY_SEL<45><4C><EFBFBD>W<EFBFBD>X<EFBFBD>^<5E><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>邪<EFBFBD>Ӗ<EFBFBD><D396>͖<EFBFBD><CD96><EFBFBD>))
|
||
result = AES_Reset();
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_Reset (%d).\n", __func__, result);
|
||
}
|
||
|
||
// <20><><EFBFBD><EFBFBD><EFBFBD>ݒ肷<DD92><E882B7>
|
||
result = AES_SetGeneralKey(&key);
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_SetGeneralKey (%d).\n", __func__, result);
|
||
}
|
||
|
||
#ifdef TEST0_USE_DMA_INPUT
|
||
// <20><><EFBFBD><EFBFBD>DMA<4D>ݒ<EFBFBD>
|
||
result = AES_StartDmaSend(INPUT_DMA, gs_data, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartDmaSend (%d).\n", __func__, result);
|
||
}
|
||
#endif
|
||
|
||
#ifdef TEST0_USE_DMA_OUTPUT
|
||
// <20>o<EFBFBD><6F>DMA<4D>ݒ<EFBFBD>
|
||
MI_CpuClear32(dataA, sizeof(dataA));
|
||
DC_FlushRange(dataA, sizeof(dataA));
|
||
result = AES_StartDmaRecv(OUTPUT_DMA, dataA, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartDmaSend (%d).\n", __func__, result);
|
||
}
|
||
#endif
|
||
|
||
begin = OS_GetTick();
|
||
|
||
// AES-CTR<54>G<EFBFBD><47><EFBFBD>R<EFBFBD>[<5B>h<EFBFBD><68><EFBFBD>J<EFBFBD>n<EFBFBD><6E><EFBFBD><EFBFBD>
|
||
result = AES_StartCtrEnc(&iv, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartCtrEnc (%d).\n", __func__, result);
|
||
}
|
||
|
||
#ifdef TEST0_USE_DMA_INPUT
|
||
#ifdef TEST0_USE_DMA_OUTPUT
|
||
// AES<45>o<EFBFBD><6F>DMA<4D><41><EFBFBD><EFBFBD><EFBFBD>҂<EFBFBD>
|
||
result = AES_WaitDma(OUTPUT_DMA);
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_WaitDma (%d).\n", __func__, result);
|
||
}
|
||
#else
|
||
// CPU<50>ŏo<C58F>͂<EFBFBD><CD82>Ă݂<C482>
|
||
result = AES_CpuRecv(dataA, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_CpuRecv (%d).\n", __func__, result);
|
||
}
|
||
#endif
|
||
#else
|
||
#ifdef TEST0_USE_DMA_OUTPUT
|
||
// CPU<50>œ<EFBFBD><C593>͂<EFBFBD><CD82>Ă݂<C482>
|
||
result = AES_CpuSend(gs_data, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_CpuSend (%d).\n", __func__, result);
|
||
}
|
||
#else
|
||
#error "Does not support CPU input and CPU output at same time."
|
||
#endif
|
||
#endif
|
||
|
||
OS_TPrintf("%s: %d usec.\n", __func__, (u32)OS_TicksToMicroSeconds(OS_GetTick()-begin));
|
||
|
||
// <20>o<EFBFBD>͌<EFBFBD><CD8C>ʂ̕\<5C><>
|
||
dump(__func__, dataA, sizeof(gs_data));
|
||
}
|
||
|
||
static void test1(void)
|
||
{
|
||
OSTick begin;
|
||
AESResult result;
|
||
|
||
OS_TPrintf("\n%s() is starting.\n\n", __func__);
|
||
|
||
// <20>Ƃ肠<C682><E882A0><EFBFBD><EFBFBD><EFBFBD>A<EFBFBD><41><EFBFBD>Z<EFBFBD>b<EFBFBD>g<EFBFBD><67><EFBFBD><EFBFBD> (<28><><EFBFBD>W<D68C>͉<EFBFBD><CD89><EFBFBD><EFBFBD>ς<EFBFBD><CF82><EFBFBD><EFBFBD>Ȃ<EFBFBD> (KEY_SEL<45><4C><EFBFBD>W<EFBFBD>X<EFBFBD>^<5E><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>邪<EFBFBD>Ӗ<EFBFBD><D396>͖<EFBFBD><CD96><EFBFBD>))
|
||
result = AES_Reset();
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_Reset (%d).\n", __func__, result);
|
||
}
|
||
|
||
// <20><><EFBFBD><EFBFBD><EFBFBD>ݒ肵<DD92>Ȃ<EFBFBD> (<28>O<EFBFBD><4F><EFBFBD>̒l<CC92><6C><EFBFBD>g<EFBFBD><67><EFBFBD>܂킷)
|
||
|
||
// <20><><EFBFBD><EFBFBD>DMA<4D>ݒ<EFBFBD>
|
||
DC_FlushRange(dataA, sizeof(dataA));
|
||
result = AES_StartDmaSend(INPUT_DMA, dataA, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartDmaSend (%d).\n", __func__, result);
|
||
}
|
||
|
||
// <20>o<EFBFBD>̓o<CD83>b<EFBFBD>t<EFBFBD>@<40>̃N<CC83><4E><EFBFBD>A
|
||
MI_CpuClear32(dataB, sizeof(dataB));
|
||
DC_FlushRange(dataB, sizeof(dataB));
|
||
|
||
begin = OS_GetTick();
|
||
|
||
// AES-CTR<54>f<EFBFBD>R<EFBFBD>[<5B>h<EFBFBD><68><EFBFBD>J<EFBFBD>n<EFBFBD><6E><EFBFBD><EFBFBD>
|
||
result = AES_StartCtrDec(&iv, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartCtrDec (%d).\n", __func__, result);
|
||
}
|
||
|
||
// CPU<50>ŏo<C58F>͂<EFBFBD><CD82>Ă݂<C482>
|
||
result = AES_CpuRecv(dataB, sizeof(gs_data)); // adata + pdata
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_CpuRecv (%d).\n", __func__, result);
|
||
}
|
||
|
||
OS_TPrintf("%s: %d usec.\n", __func__, (u32)OS_TicksToMicroSeconds(OS_GetTick()-begin));
|
||
|
||
// <20>o<EFBFBD>͌<EFBFBD><CD8C>ʂ̕\<5C><>
|
||
dump(__func__, dataB, sizeof(gs_data));
|
||
|
||
result = AES_IsValid();
|
||
if (AES_RESULT_SUCCESS_TRUE != result && AES_RESULT_SUCCESS_FALSE != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_IsValid (%d).\n", __func__, result);
|
||
}
|
||
else
|
||
{
|
||
OS_TPrintf("Result: %s\n", AES_RESULT_SUCCESS_TRUE == result ? "Success" : "Failed");
|
||
}
|
||
}
|
||
|
||
static void test2(void)
|
||
{
|
||
OSTick begin;
|
||
AESResult result;
|
||
|
||
OS_TPrintf("\n%s() is starting.\n\n", __func__);
|
||
|
||
// <20>Ƃ肠<C682><E882A0><EFBFBD><EFBFBD><EFBFBD>A<EFBFBD><41><EFBFBD>Z<EFBFBD>b<EFBFBD>g<EFBFBD><67><EFBFBD><EFBFBD> (<28><><EFBFBD>W<D68C>͉<EFBFBD><CD89><EFBFBD><EFBFBD>ς<EFBFBD><CF82><EFBFBD><EFBFBD>Ȃ<EFBFBD> (KEY_SEL<45><4C><EFBFBD>W<EFBFBD>X<EFBFBD>^<5E><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>邪<EFBFBD>Ӗ<EFBFBD><D396>͖<EFBFBD><CD96><EFBFBD>))
|
||
result = AES_Reset();
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_Reset (%d).\n", __func__, result);
|
||
}
|
||
|
||
// <20><><EFBFBD><EFBFBD><EFBFBD>ݒ肷<DD92><E882B7>
|
||
result = AES_SetSpecialKey(&key);
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_SetSpecialKey (%d).\n", __func__, result);
|
||
}
|
||
|
||
// <20>o<EFBFBD><6F>DMA<4D>ݒ<EFBFBD>
|
||
MI_CpuClear32(dataB, sizeof(dataB));
|
||
DC_FlushRange(dataB, sizeof(dataB));
|
||
result = AES_StartDmaRecv(OUTPUT_DMA, dataB, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartDmaSend (%d).\n", __func__, result);
|
||
}
|
||
|
||
begin = OS_GetTick();
|
||
|
||
// AES-CTR<54>f<EFBFBD>R<EFBFBD>[<5B>h<EFBFBD><68><EFBFBD>J<EFBFBD>n<EFBFBD><6E><EFBFBD><EFBFBD>
|
||
result = AES_StartCtrDec(&iv, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_StartCtrDec (%d).\n", __func__, result);
|
||
}
|
||
|
||
// CPU<50>œ<EFBFBD><C593>͂<EFBFBD><CD82>Ă݂<C482>
|
||
AES_CpuSend(dataA, sizeof(gs_data));
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_CpuSend (%d).\n", __func__, result);
|
||
}
|
||
|
||
OS_TPrintf("%s: %d usec.\n", __func__, (u32)OS_TicksToMicroSeconds(OS_GetTick()-begin));
|
||
|
||
// <20>o<EFBFBD>͌<EFBFBD><CD8C>ʂ̕\<5C><>
|
||
dump(__func__, dataB, sizeof(gs_data));
|
||
|
||
result = AES_IsValid();
|
||
if (AES_RESULT_SUCCESS_TRUE != result && AES_RESULT_SUCCESS_FALSE != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_IsValid (%d).\n", __func__, result);
|
||
}
|
||
else
|
||
{
|
||
OS_TPrintf("Result: %s\n", AES_RESULT_SUCCESS_TRUE == result ? "Success" : "Failed (OK!)");
|
||
}
|
||
}
|
||
|
||
|
||
/*---------------------------------------------------------------------------*
|
||
Name: TwlMain
|
||
|
||
Description: main
|
||
|
||
Arguments: None
|
||
|
||
Returns: None
|
||
*---------------------------------------------------------------------------*/
|
||
void TwlMain()
|
||
{
|
||
AESResult result;
|
||
|
||
OS_Init();
|
||
|
||
OS_Printf("ARM9 starts.\n");
|
||
|
||
OS_InitTick();
|
||
|
||
OS_TPrintf("Debug Info:\n");
|
||
OS_TPrintf("\tdataA = 0x%08X\n", dataA);
|
||
OS_TPrintf("\tdataB = 0x%08X\n", dataB);
|
||
|
||
AES_Init();
|
||
OS_TPrintf("AES_Init wad done.\n");
|
||
|
||
OS_EnableIrq();
|
||
|
||
while (1)
|
||
{
|
||
result = AES_TryLock();
|
||
if (AES_RESULT_SUCCESS_TRUE == result)
|
||
{
|
||
break;
|
||
}
|
||
if (AES_RESULT_SUCCESS_FALSE != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_TryLock (%d).\n", __func__, result);
|
||
}
|
||
OS_Sleep(1);
|
||
}
|
||
OS_TPrintf("AES_TryLock wad done.\n");
|
||
|
||
test0();
|
||
test1();
|
||
test2();
|
||
|
||
result = AES_Unlock();
|
||
if (AES_RESULT_SUCCESS != result)
|
||
{
|
||
OS_TPrintf("%s: Failed to call AES_Unlock (%d).\n", __func__, result);
|
||
}
|
||
|
||
// done
|
||
OS_TPrintf("\nARM9 ends.\n");
|
||
OS_Terminate();
|
||
}
|