mirror of
https://github.com/DarkStore-3DS/Project_CTR.git
synced 2026-07-03 00:39:14 +00:00
Jakcron
128 lines
3.8 KiB
C++
128 lines
3.8 KiB
C++
#pragma once
|
|
#include "types.h"
|
|
#include "KeyBag.h"
|
|
#include "NcchProcess.h"
|
|
#include <tc/Optional.h>
|
|
#include <tc/io/IFileSystem.h>
|
|
#include <ntd/n3ds/cia.h>
|
|
|
|
#include <ntd/n3ds/es/RsaSigner.h>
|
|
#include <ntd/n3ds/es/Certificate.h>
|
|
#include <ntd/n3ds/es/Ticket.h>
|
|
#include <ntd/n3ds/es/TitleMetaData.h>
|
|
|
|
namespace ctrtool {
|
|
|
|
class CiaProcess
|
|
{
|
|
public:
|
|
CiaProcess();
|
|
|
|
void setInputStream(const std::shared_ptr<tc::io::IStream>& input_stream);
|
|
void setKeyBag(const ctrtool::KeyBag& key_bag);
|
|
void setCliOutputMode(bool show_header_info);
|
|
void setVerboseMode(bool verbose);
|
|
void setVerifyMode(bool verify);
|
|
void setCertExtractPath(const tc::io::Path& extract_path);
|
|
void setTikExtractPath(const tc::io::Path& extract_path);
|
|
void setTmdExtractPath(const tc::io::Path& extract_path);
|
|
void setContentExtractPath(const tc::io::Path& extract_path);
|
|
void setFooterExtractPath(const tc::io::Path& extract_path);
|
|
void setContentIndex(size_t index);
|
|
|
|
// ncch settings passed on
|
|
void setRawMode(bool raw);
|
|
void setPlainMode(bool plain);
|
|
void setShowSyscallName(bool show_name);
|
|
void setNcchRegionProcessOutputMode(NcchProcess::NcchRegion region, bool show_info, bool show_fs, const tc::Optional<tc::io::Path>& bin_extract_path, const tc::Optional<tc::io::Path>& fs_extract_path);
|
|
|
|
void process();
|
|
private:
|
|
std::string mModuleLabel;
|
|
|
|
// input args
|
|
std::shared_ptr<tc::io::IStream> mInputStream;
|
|
ctrtool::KeyBag mKeyBag;
|
|
bool mShowHeaderInfo;
|
|
bool mVerbose;
|
|
bool mVerify;
|
|
tc::Optional<tc::io::Path> mCertExtractPath;
|
|
tc::Optional<tc::io::Path> mTikExtractPath;
|
|
tc::Optional<tc::io::Path> mTmdExtractPath;
|
|
tc::Optional<tc::io::Path> mContentExtractPath;
|
|
tc::Optional<tc::io::Path> mFooterExtractPath;
|
|
size_t mContentIndex;
|
|
|
|
// process variables
|
|
ntd::n3ds::CiaHeader mHeader;
|
|
std::map<std::string, std::shared_ptr<ntd::n3ds::es::ISigner>> mIssuerSigner;
|
|
std::map<std::string, std::shared_ptr<ntd::n3ds::es::ISigner>> mCertImportedIssuerSigner;
|
|
std::vector<ntd::n3ds::es::Certificate> mCertChain;
|
|
std::vector<ValidState> mCertSigValid;
|
|
ntd::n3ds::es::Ticket mTicket;
|
|
ValidState mTicketSigValid;
|
|
ntd::n3ds::es::TitleMetaData mTitleMetaData;
|
|
ValidState mTitleMetaDataSigValid;
|
|
tc::Optional<KeyBag::Aes128Key> mDecryptedTitleKey;
|
|
bool mIsTwlTitle;
|
|
ctrtool::NcchProcess mNcchProcess;
|
|
std::shared_ptr<tc::io::IFileSystem> mFsReader;
|
|
|
|
struct CiaSectionInfo
|
|
{
|
|
int64_t offset;
|
|
int64_t size;
|
|
|
|
CiaSectionInfo() : offset(0), size(0) {}
|
|
} mCertSizeInfo, mTikSizeInfo, mTmdSizeInfo, mContentSizeInfo, mFooterSizeInfo;
|
|
|
|
struct ContentInfo
|
|
{
|
|
int64_t offset;
|
|
int64_t size;
|
|
uint32_t cid;
|
|
uint16_t cindex;
|
|
bool is_encrypted;
|
|
bool is_hashed;
|
|
std::array<byte_t, 32> hash;
|
|
int valid_state;
|
|
|
|
ContentInfo() :
|
|
offset(0),
|
|
size(0),
|
|
cid(0),
|
|
cindex(0),
|
|
is_encrypted(false),
|
|
is_hashed(false),
|
|
valid_state(ValidState::Unchecked)
|
|
{
|
|
memset(hash.data(), 0, hash.size());
|
|
}
|
|
};
|
|
std::map<size_t, ContentInfo> mContentInfo;
|
|
|
|
// helper methods
|
|
void importIssuerProfiles();
|
|
void importHeader();
|
|
void verifyMetadata();
|
|
void verifyContent();
|
|
void printHeader();
|
|
void extractCia();
|
|
void copyStream(const std::shared_ptr<tc::io::IStream>& in, const std::shared_ptr<tc::io::IStream>& out);
|
|
void processContent();
|
|
|
|
void createContentIv(std::array<byte_t, 16>& content_iv, uint16_t index);
|
|
|
|
bool isTwlTitle(uint64_t title_id);
|
|
|
|
// string utils
|
|
std::string getValidString(byte_t validstate);
|
|
std::string getCiaTypeString(uint16_t type);
|
|
std::string getFormatVersionString(uint16_t format_version);
|
|
std::string getTruncatedBytesString(const byte_t* data, size_t len, bool do_not_truncate = false);
|
|
std::string getSigTypeString(brd::es::ESSigType sig_type);
|
|
std::string getCertificatePublicKeyTypeString(brd::es::ESCertPubKeyType public_key_type);
|
|
std::string getTitleVersionString(uint16_t version);
|
|
};
|
|
|
|
} |