mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2026-05-06 03:01:10 -04:00
f06aef4f83
Now with cancel button and an actual progress bar. For simplicity, we do two passes on the progress bar, one for loading the NAND into memory and one for extracting it. The user directory is likely on an SSD, making the extraction pass invisibly fast.
109 lines
3.0 KiB
C++
109 lines
3.0 KiB
C++
// Copyright 2017 Dolphin Emulator Project
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#pragma once
|
|
|
|
#include <array>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <fmt/format.h>
|
|
|
|
#include "Common/CommonTypes.h"
|
|
#include "Common/Crypto/AES.h"
|
|
#include "Common/Swap.h"
|
|
|
|
namespace DiscIO
|
|
{
|
|
class NANDImporter final
|
|
{
|
|
public:
|
|
NANDImporter();
|
|
~NANDImporter();
|
|
|
|
enum class Step
|
|
{
|
|
Loading,
|
|
Extracting,
|
|
};
|
|
// Return true to cancel.
|
|
using UpdateCallback = std::function<bool(Step step, int cur, int max)>;
|
|
|
|
// Extract a NAND image to the configured NAND root.
|
|
// If the associated OTP/SEEPROM dump (keys.bin) is not included in the image,
|
|
// get_otp_dump_path will be called to get a path to it.
|
|
void ImportNANDBin(const std::string& path_to_bin, UpdateCallback update_callback,
|
|
const std::function<std::string()>& get_otp_dump_path);
|
|
bool ExtractCertificates();
|
|
|
|
enum class Type
|
|
{
|
|
File = 1,
|
|
Directory = 2,
|
|
};
|
|
|
|
#pragma pack(push, 1)
|
|
struct NANDFSTEntry
|
|
{
|
|
char name[12];
|
|
u8 mode;
|
|
u8 attr;
|
|
Common::BigEndianValue<u16> sub;
|
|
Common::BigEndianValue<u16> sib;
|
|
Common::BigEndianValue<u32> size;
|
|
Common::BigEndianValue<u32> uid;
|
|
Common::BigEndianValue<u16> gid;
|
|
Common::BigEndianValue<u32> x3;
|
|
};
|
|
static_assert(sizeof(NANDFSTEntry) == 0x20, "Wrong size");
|
|
|
|
struct NANDSuperblock
|
|
{
|
|
std::array<char, 4> magic; // "SFFS"
|
|
Common::BigEndianValue<u32> version;
|
|
Common::BigEndianValue<u32> unknown;
|
|
std::array<Common::BigEndianValue<u16>, 0x8000> fat;
|
|
std::array<NANDFSTEntry, 0x17FF> fst;
|
|
std::array<u8, 0x14> pad;
|
|
};
|
|
static_assert(sizeof(NANDSuperblock) == 0x40000, "Wrong size");
|
|
#pragma pack(pop)
|
|
|
|
private:
|
|
bool ReadNANDBin(const std::string& path_to_bin,
|
|
const std::function<std::string()>& get_otp_dump_path);
|
|
bool FindSuperblock();
|
|
bool ExtractFiles();
|
|
std::string GetPath(const NANDFSTEntry& entry, const std::string& parent_path);
|
|
std::string FormatDebugString(const NANDFSTEntry& entry);
|
|
bool ProcessEntry(u16 entry_number, const std::string& parent_path);
|
|
std::vector<u8> GetEntryData(const NANDFSTEntry& entry) const;
|
|
void ExportKeys();
|
|
|
|
std::string m_nand_root;
|
|
std::vector<u8> m_nand;
|
|
std::vector<u8> m_nand_keys;
|
|
std::unique_ptr<Common::AES::Context> m_aes_ctx;
|
|
std::unique_ptr<NANDSuperblock> m_superblock;
|
|
UpdateCallback m_update_callback;
|
|
u16 m_progress_cur = 0;
|
|
u16 m_progress_max = 0;
|
|
};
|
|
} // namespace DiscIO
|
|
|
|
template <>
|
|
struct fmt::formatter<DiscIO::NANDImporter::NANDFSTEntry>
|
|
{
|
|
constexpr auto parse(format_parse_context& ctx) { return ctx.begin(); }
|
|
template <typename FormatContext>
|
|
auto format(const DiscIO::NANDImporter::NANDFSTEntry& entry, FormatContext& ctx) const
|
|
{
|
|
return fmt::format_to(
|
|
ctx.out(), "{:12.12} {:#010b} {:#04x} {:#06x} {:#06x} {:#010x} {:#010x} {:#06x} {:#010x}",
|
|
entry.name, entry.mode, entry.attr, entry.sub, entry.sib, entry.size, entry.uid, entry.gid,
|
|
entry.x3);
|
|
}
|
|
};
|