Add source code for ctrtool

2026-07-04 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,175 @@
+#pragma once
+#include <string>
+#include <vector>
+#include <tc/Optional.h>
+#include <tc/io.h>
+
+#include "FirmProcess.h"
+#include "KeyBag.h"
+
+namespace ctrtool {
+
+struct Settings
+{
+	enum FileType
+	{
+		FILE_TYPE_ERROR,
+		FILE_TYPE_NCSD,
+		FILE_TYPE_CIA,
+		FILE_TYPE_NCCH,
+		FILE_TYPE_EXHEADER,
+		FILE_TYPE_EXEFS,
+		FILE_TYPE_ROMFS,
+		FILE_TYPE_FIRM,
+		FILE_TYPE_CERT,
+		FILE_TYPE_TIK,
+		FILE_TYPE_TMD,
+		FILE_TYPE_LZSS,
+		FILE_TYPE_CRR,
+		FILE_TYPE_CRO,
+		FILE_TYPE_IVFC,
+		FILE_TYPE_SMDH
+	};
+
+	struct InputFileOptions
+	{
+		FileType filetype;
+		tc::Optional<tc::io::Path> path;
+	} infile;
+
+	struct Options
+	{
+		bool plain;
+		bool raw;
+		bool verbose;
+		bool verify;
+		bool show_keys;
+		bool is_dev;
+		KeyBag keybag;
+	} opt;
+
+	// LZSS options
+	struct LzssOptions
+	{
+		tc::Optional<tc::io::Path> extract_path;
+	} lzss;
+
+	// NCCH options
+	struct NcchOptions
+	{
+		tc::Optional<tc::io::Path> exheader_path;
+		tc::Optional<tc::io::Path> logo_path;
+		tc::Optional<tc::io::Path> plainregion_path;
+		tc::Optional<tc::io::Path> exefs_path;
+		tc::Optional<tc::io::Path> romfs_path;
+	} ncch;
+
+	// ExHeader options
+	struct ExheaderOptions
+	{
+		bool show_syscalls_as_names;
+	} exheader;
+
+	// ExeFs options
+	struct ExefsOptions
+	{
+		tc::Optional<tc::io::Path> extract_path;
+		bool list_fs;
+		bool decompress_code_partition;
+	} exefs;
+
+	// RomFs options
+	struct RomfsOptions
+	{
+		tc::Optional<tc::io::Path> extract_path;
+		bool list_fs;
+	} romfs;
+
+	// CCI/CIA options
+	struct RomOptions
+	{
+		size_t content_process_index;
+		tc::Optional<tc::io::Path> content_extract_path;
+	} rom;
+
+	// CIA options
+	struct CiaOptions
+	{
+		tc::Optional<tc::io::Path> certs_path;
+		tc::Optional<tc::io::Path> tik_path;
+		tc::Optional<tc::io::Path> tmd_path;
+		tc::Optional<tc::io::Path> meta_path;
+	} cia;
+	
+	// FIRM options
+	struct FirmOptions
+	{
+		tc::Optional<tc::io::Path> extract_path;
+		FirmProcess::FirmwareType firm_type;
+	} firm;
+
+	// CWAV options
+	struct CwavOptions
+	{
+		tc::Optional<tc::io::Path> extract_path;
+		size_t wav_loops;
+	} cwav;
+	
+
+	Settings()
+	{
+		infile.filetype = FILE_TYPE_ERROR;
+		infile.path = tc::Optional<tc::io::Path>();
+
+		opt.plain = false;
+		opt.raw = false;
+		opt.verbose = false;
+		opt.verify = false;
+		opt.show_keys = false;
+		opt.is_dev = false;
+		opt.keybag = KeyBag();
+
+		exheader.show_syscalls_as_names = false;
+
+		exefs.extract_path = tc::Optional<tc::io::Path>();
+		exefs.list_fs = false;
+		exefs.decompress_code_partition = false;
+
+		romfs.extract_path = tc::Optional<tc::io::Path>();
+		romfs.list_fs = false;
+
+		rom.content_process_index = 0;
+		rom.content_extract_path = tc::Optional<tc::io::Path>();
+
+		cia.certs_path = tc::Optional<tc::io::Path>();
+		cia.tik_path = tc::Optional<tc::io::Path>();
+		cia.tmd_path = tc::Optional<tc::io::Path>();
+		cia.meta_path = tc::Optional<tc::io::Path>();
+
+		firm.extract_path = tc::Optional<tc::io::Path>();
+		firm.firm_type = FirmProcess::FirmwareType_Nand;
+
+		cwav.extract_path = tc::Optional<tc::io::Path>();
+		cwav.wav_loops = 0;
+	}
+};
+
+class SettingsInitializer : public Settings
+{
+public:
+	SettingsInitializer(const std::vector<std::string>& args);
+private:
+	void parse_args(const std::vector<std::string>& args);
+	void register_option_handlers();
+	void determine_filetype();
+	void usage_text();
+
+	std::string mModuleLabel;
+
+	bool mShowKeys;
+	tc::Optional<std::string> mFallBackTitleKey;
+	tc::Optional<std::string> mFallBackSeed;
+	tc::Optional<tc::io::Path> mSeedDbPath;
+};
+
+}