Add source code for ctrtool

2026-07-03 08:49:03 +00:00 · 2022-03-12 16:00:33 +08:00
parent 6ad2f13c50
commit 800f5776bc
681 changed files with 219734 additions and 0 deletions
@@ -0,0 +1,128 @@
+#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, bool show_fs);
+	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 mShowFs;
+	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::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);
+};
+
+}