mirror of
https://github.com/rvtr/ctr_Repair.git
synced 2025-10-31 13:51:08 -04:00
5dc903446e
git-svn-id: file:///Volumes/Transfer/gigaleak_20231201/2020-05-23%20-%20ctr.7z%20+%20svn_v1.068.zip/ctr/svn/ctr_Repair@611 385bec56-5757-e545-9c3a-d8741f4650f1
308 lines
8.7 KiB
C++
308 lines
8.7 KiB
C++
/*---------------------------------------------------------------------------*
|
|
Project: Horizon
|
|
File: main.cpp
|
|
|
|
Copyright (C)2011 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/version.h>
|
|
#include <nn/ptm/CTR/ptm_ApiSysmenu.h>
|
|
#include <nn/mcu.h>
|
|
#include <nn/cfg/CTR/cfg_Api.h>
|
|
#include <nn/cfg/CTR/cfg_ApiInit.h>
|
|
#include <nn/cfg/CTR/cfg_ApiSys.h>
|
|
#include <nn/ps.h>
|
|
#include <nn/nwm.h>
|
|
#include <nn/ac.h>
|
|
#include <nn/friends.h>
|
|
#include <nn/friends/CTR/friends_ApiPrivate.h>
|
|
#include <nn/fs/CTR/fs_ArchiveTypesForSystem.h>
|
|
#include <nn/fs/CTR/MPCore/fs_FileSystemBasePrivate.h>
|
|
#include <nn/fs/fs_ApiSysSaveData.h>
|
|
#include <nn/cfg/CTR/cfg_ApiNor.h>
|
|
#include <nn/os/os_SharedInfo.h>
|
|
#include <nn/pl/CTR/pl_Version.h>
|
|
#include <nn/fs/fs_ApiSharedExtSaveData.h>
|
|
#include <nn/nim.h>
|
|
#include <nn/ndm.h>
|
|
|
|
#include <nn/crypto/crypto_AesCmac.h>
|
|
#include <nn/crypto/crypto_SwAesCtrContext.h>
|
|
#include <nn/crypto/crypto_Sha256.h>
|
|
#include <nn/crypto/crypto_SwAesCmac.h>
|
|
|
|
#include "Aes_define.h"
|
|
#include "FileTransfer.h"
|
|
#include "CommonLogger.h"
|
|
|
|
#include "demo.h"
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
#include "DrawSystemState.h"
|
|
#include "FileName.h"
|
|
#include "SimplePlayer.h"
|
|
#include "CommonLogger.h"
|
|
#include "SDMountManager.h"
|
|
#include "HeapManager.h"
|
|
#include "common_Types.h"
|
|
#include "VersionDetect.h"
|
|
#include "Util.h"
|
|
#include "CommonLogger.h"
|
|
#include "SdReaderWriter.h"
|
|
|
|
namespace
|
|
{
|
|
|
|
// グラフィックスに割り当てるメモリ
|
|
const size_t s_GxHeapSize = 0x800000;
|
|
|
|
const u32 CONSOLE_WIDTH = 38;
|
|
const u32 CONSOLE_HEIGHT = 24;
|
|
const u32 CONSOLE_MAX_LINE = 1000;
|
|
|
|
const size_t ADD_CMAC_THREAD_STACK_SIZE = 0x4000;
|
|
nn::os::Thread s_AddCmacThread;
|
|
nn::os::StackBuffer<ADD_CMAC_THREAD_STACK_SIZE> s_AddCmacThreadStack;
|
|
|
|
u32 s_AddCmacSuccess = 0;
|
|
u32 s_AddCmacFail = 0;
|
|
|
|
}
|
|
|
|
namespace CmacAdder
|
|
{
|
|
|
|
// ディレクトリ間のコピー
|
|
// アーカイブ越しのコピーが可能
|
|
// アーカイブにマウントした状態で呼び出す必要あり
|
|
// 書き込み先のディレクトリはあらかじめ消去しておくこと。
|
|
// 引数はスラッシュ付き
|
|
// TODO:分割して短くする
|
|
bool AddCmac(const wchar_t * from_path, void* buf, const size_t bufSize)
|
|
{
|
|
using namespace common;
|
|
|
|
nn::fs::Directory from_dir;
|
|
nn::fs::DirectoryEntry entry;
|
|
s32 numread = 0;
|
|
std::wostringstream target_from;
|
|
std::wostringstream target_to;
|
|
std::wostringstream targetDirectory;
|
|
bool ret_value = true;
|
|
|
|
nn::Result result = from_dir.TryInitialize(from_path);
|
|
|
|
if (result.IsFailure())
|
|
{
|
|
COMMON_LOGGER_RESULT_IF_FAILED(result);
|
|
return false;
|
|
}
|
|
|
|
// 出力先ディレクトリを削除してから作成
|
|
const wchar_t* CMAC_DIR = L"CmacAdded";
|
|
targetDirectory << from_path << CMAC_DIR;
|
|
result = nn::fs::TryDeleteDirectoryRecursively(targetDirectory.str().c_str());
|
|
COMMON_LOGGER_RESULT_IF_FAILED(result);
|
|
result = nn::fs::TryCreateDirectory(targetDirectory.str().c_str());
|
|
if (result.IsFailure())
|
|
{
|
|
COMMON_LOGGER_RESULT_IF_FAILED(result);
|
|
return false;
|
|
}
|
|
|
|
while (1)
|
|
{
|
|
result = from_dir.TryRead(&numread, &entry, 1);
|
|
if (result.IsFailure() || numread != 1)
|
|
{
|
|
break;
|
|
}
|
|
|
|
if (std::wcscmp(entry.entryName, L".") == 0 || std::wcscmp(entry.entryName, L"..") == 0)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
target_from.str(L"");
|
|
target_from.clear(std::stringstream::goodbit);
|
|
target_from << from_path << entry.entryName;
|
|
|
|
target_to.str(L"");
|
|
target_to.clear(std::stringstream::goodbit);
|
|
target_to << from_path << CMAC_DIR << L"/" <<entry.entryName;
|
|
|
|
// ディレクトリの場合
|
|
if (entry.attributes.isDirectory)
|
|
{
|
|
continue;
|
|
}
|
|
else if (std::wcscmp(entry.entryName, common::AP_SETTING_FILENAME) == 0)
|
|
{
|
|
continue;
|
|
}
|
|
// ファイルの場合
|
|
// CMACが付いていないと考えてCMACを付加する
|
|
else
|
|
{
|
|
nn::fs::FileInputStream sdFile;
|
|
s64 filesize;
|
|
s32 readsize;
|
|
|
|
// 読み込み対象ファイル開く
|
|
result = sdFile.TryInitialize(target_from.str().c_str());
|
|
COMMON_LOGGER_RETURN_FALSE_IF_FAILED(result);
|
|
|
|
// 読み込み対象ファイルのサイズ取得
|
|
result = sdFile.TryGetSize(&filesize);
|
|
COMMON_LOGGER_RETURN_FALSE_IF_FAILED(result);
|
|
|
|
result = sdFile.TryRead(&readsize, buf, bufSize);
|
|
COMMON_LOGGER_RETURN_FALSE_IF_FAILED(result);
|
|
|
|
common::SdReaderWriter writer;
|
|
result = writer.WriteBufWithCmac(target_to.str().c_str(), buf, readsize);
|
|
COMMON_LOGGER_RETURN_FALSE_IF_FAILED(result);
|
|
sdFile.Finalize();
|
|
s_AddCmacSuccess++;
|
|
}
|
|
}
|
|
from_dir.Finalize();
|
|
return ret_value;
|
|
}
|
|
|
|
|
|
void AddCmacThreadFunc()
|
|
{
|
|
nn::Result result;
|
|
|
|
COMMON_LOGGER("AddCmacThreadFunc Start\n");
|
|
s_AddCmacFail = 0;
|
|
s_AddCmacSuccess = 0;
|
|
|
|
result = common::SdMountManager::Mount();
|
|
|
|
size_t bufSize = common::GetAllocatableSize(AES_BLOCK_SIZE * 2);
|
|
if(bufSize > common::FILE_COPY_HEAP_SIZE)
|
|
{
|
|
bufSize = common::FILE_COPY_HEAP_SIZE;
|
|
}
|
|
|
|
common::HeapManager heap(bufSize, AES_BLOCK_SIZE * 2);
|
|
void* buf = heap.GetAddr();
|
|
if (buf != NULL)
|
|
{
|
|
AddCmac((std::wstring(common::SDMC_ROOT_DIRECTORY_PATH).c_str()),
|
|
buf, bufSize);
|
|
}
|
|
|
|
common::SdMountManager::Unmount();
|
|
|
|
COMMON_LOGGER("AddCmac Thread Finalize\n");
|
|
|
|
COMMON_LOGGER("\n\n");
|
|
COMMON_LOGGER("AddCmac Finished, success = %d, fail = %d\n", s_AddCmacSuccess, s_AddCmacFail);
|
|
}
|
|
|
|
extern "C" void nninitSetupDaemons(void)
|
|
{
|
|
}
|
|
|
|
|
|
extern "C" void nnMain(void)
|
|
{
|
|
nn::Result result;
|
|
|
|
// os の初期化
|
|
nn::os::Initialize();
|
|
|
|
// fs の初期化
|
|
nn::fs::Initialize();
|
|
|
|
// appletの初期化
|
|
nn::applet::Enable( false );
|
|
|
|
// hid の初期化
|
|
result = nn::hid::Initialize();
|
|
NN_ERR_THROW_FATAL_IF_FATAL_ONLY(result);
|
|
|
|
// ヒープの確保
|
|
common::InitializeHeap();
|
|
|
|
// RenderSystem の準備
|
|
common::HeapManager gxHeap(s_GxHeapSize);
|
|
uptr heapForGx = reinterpret_cast<uptr>(gxHeap.GetAddr());
|
|
demo::RenderSystemDrawing s_RenderSystem;
|
|
s_RenderSystem.Initialize(heapForGx, s_GxHeapSize);
|
|
|
|
// ログ描画の初期化
|
|
common::Logger::GetLoggerInstance()->Initialize(CONSOLE_WIDTH, CONSOLE_HEIGHT, CONSOLE_MAX_LINE, &s_RenderSystem);
|
|
|
|
// RenderSystemを作ってからログが出せる
|
|
common::Logger::InitializeEjectThread();
|
|
|
|
COMMON_LOGGER("AddCmac Start\n");
|
|
|
|
// ボタン入力
|
|
nn::hid::PadReader s_PadReader;
|
|
nn::hid::PadStatus padStatus;
|
|
|
|
for(;;)
|
|
{
|
|
s_PadReader.ReadLatest(&padStatus);
|
|
|
|
if(padStatus.trigger & nn::hid::BUTTON_A)
|
|
{
|
|
if(s_AddCmacThread.IsValid() && !s_AddCmacThread.IsAlive())
|
|
{
|
|
s_AddCmacThread.Join();
|
|
s_AddCmacThread.Finalize();
|
|
}
|
|
s_AddCmacThread.Start(AddCmacThreadFunc, s_AddCmacThreadStack);
|
|
}
|
|
|
|
// コンソールスクロール
|
|
if(padStatus.hold & nn::hid::BUTTON_UP)
|
|
{
|
|
common::Logger::GetLoggerInstance()->ScrollUp();
|
|
}
|
|
|
|
// コンソールスクロール
|
|
if(padStatus.hold & nn::hid::BUTTON_DOWN)
|
|
{
|
|
common::Logger::GetLoggerInstance()->ScrollDown();
|
|
}
|
|
|
|
if(padStatus.hold & nn::hid::BUTTON_LEFT)
|
|
{
|
|
common::Logger::GetLoggerInstance()->ScrollToBegin();
|
|
}
|
|
|
|
if(padStatus.hold & nn::hid::BUTTON_RIGHT)
|
|
{
|
|
common::Logger::GetLoggerInstance()->ScrollToEnd();
|
|
}
|
|
|
|
s_RenderSystem.SetRenderTarget(NN_GX_DISPLAY0);
|
|
s_RenderSystem.Clear();
|
|
s_RenderSystem.SetColor(1.f, 1.f, 1.f);
|
|
|
|
common::Logger::GetLoggerInstance()->DrawConsole();
|
|
s_RenderSystem.SwapBuffers();
|
|
|
|
s_RenderSystem.WaitVsync(NN_GX_DISPLAY_BOTH);
|
|
}
|
|
}
|
|
|
|
}
|