// Copyright 2008 Dolphin Emulator Project // SPDX-License-Identifier: GPL-2.0-or-later #pragma once #include <array> #include <functional> #include <map> #include <memory> #include <optional> #include <string> #include <vector> #include "Common/CommonTypes.h" #include "Common/Crypto/SHA1.h" #include "Common/Lazy.h" #include "Core/IOS/ES/Formats.h" #include "DiscIO/Filesystem.h" #include "DiscIO/Volume.h" #include "DiscIO/VolumeDisc.h" #include "Common/Crypto/AES.h" namespace DiscIO { class BlobReader; enum class BlobType; enum class Country; class FileSystem; enum class Language; enum class Region; enum class Platform; class VolumeWii : public VolumeDisc { public: static constexpr size_t AES_KEY_SIZE = Common::AES::Context::KEY_SIZE; static constexpr u32 BLOCKS_PER_GROUP = 0x40; static constexpr u64 BLOCK_HEADER_SIZE = 0x0400; static constexpr u64 BLOCK_DATA_SIZE = 0x7C00; static constexpr u64 BLOCK_TOTAL_SIZE = BLOCK_HEADER_SIZE + BLOCK_DATA_SIZE; static constexpr u64 GROUP_HEADER_SIZE = BLOCK_HEADER_SIZE * BLOCKS_PER_GROUP; static constexpr u64 GROUP_DATA_SIZE = BLOCK_DATA_SIZE * BLOCKS_PER_GROUP; static constexpr u64 GROUP_TOTAL_SIZE = GROUP_HEADER_SIZE + GROUP_DATA_SIZE; struct HashBlock { std::array<Common::SHA1::Digest, 31> h0; std::array<u8, 20> padding_0; std::array<Common::SHA1::Digest, 8> h1; std::array<u8, 32> padding_1; std::array<Common::SHA1::Digest, 8> h2; std::array<u8, 32> padding_2; }; static_assert(sizeof(HashBlock) == BLOCK_HEADER_SIZE); VolumeWii(std::unique_ptr<BlobReader> reader); ~VolumeWii(); bool Read(u64 offset, u64 length, u8* buffer, const Partition& partition) const override; bool HasWiiHashes() const override; bool HasWiiEncryption() const override; std::vector<Partition> GetPartitions() const override; Partition GetGamePartition() const override; std::optional<u32> GetPartitionType(const Partition& partition) const override; std::optional<u64> GetTitleID(const Partition& partition) const override; const IOS::ES::TicketReader& GetTicket(const Partition& partition) const override; const IOS::ES::TMDReader& GetTMD(const Partition& partition) const override; const std::vector<u8>& GetCertificateChain(const Partition& partition) const override; const FileSystem* GetFileSystem(const Partition& partition) const override; static u64 OffsetInHashedPartitionToRawOffset(u64 offset, const Partition& partition, u64 partition_data_offset); u64 PartitionOffsetToRawOffset(u64 offset, const Partition& partition) const override; std::string GetGameTDBID(const Partition& partition = PARTITION_NONE) const override; std::map<Language, std::string> GetLongNames() const override; std::vector<u32> GetBanner(u32* width, u32* height) const override; Platform GetVolumeType() const override; bool IsDatelDisc() const override; bool CheckH3TableIntegrity(const Partition& partition) const override; bool CheckBlockIntegrity(u64 block_index, const u8* encrypted_data, const Partition& partition) const override; bool CheckBlockIntegrity(u64 block_index, const Partition& partition) const override; Region GetRegion() const override; BlobType GetBlobType() const override; u64 GetDataSize() const override; DataSizeType GetDataSizeType() const override; u64 GetRawSize() const override; const BlobReader& GetBlobReader() const override; std::array<u8, 20> GetSyncHash() const override; // The in parameter can either contain all the data to begin with, // or read_function can write data into the in parameter when called. // The latter lets reading run in parallel with hashing. // This function returns false iff read_function returns false. static bool HashGroup(const std::array<u8, BLOCK_DATA_SIZE> in[BLOCKS_PER_GROUP], HashBlock out[BLOCKS_PER_GROUP], const std::function<bool(size_t block)>& read_function = {}); static bool EncryptGroup(u64 offset, u64 partition_data_offset, u64 partition_data_decrypted_size, const std::array<u8, AES_KEY_SIZE>& key, BlobReader* blob, std::array<u8, GROUP_TOTAL_SIZE>* out, const std::function<void(HashBlock hash_blocks[BLOCKS_PER_GROUP])>& hash_exception_callback = {}); static void DecryptBlockHashes(const u8* in, HashBlock* out, Common::AES::Context* aes_context); static void DecryptBlockData(const u8* in, u8* out, Common::AES::Context* aes_context); protected: u32 GetOffsetShift() const override { return 2; } private: struct PartitionDetails { Common::Lazy<std::unique_ptr<Common::AES::Context>> key; Common::Lazy<IOS::ES::TicketReader> ticket; Common::Lazy<IOS::ES::TMDReader> tmd; Common::Lazy<std::vector<u8>> cert_chain; Common::Lazy<std::vector<u8>> h3_table; Common::Lazy<std::unique_ptr<FileSystem>> file_system; Common::Lazy<u64> data_offset; u32 type = 0; }; std::unique_ptr<BlobReader> m_reader; std::map<Partition, PartitionDetails> m_partitions; Partition m_game_partition; bool m_has_hashes; bool m_has_encryption; mutable u64 m_last_decrypted_block; mutable u8 m_last_decrypted_block_data[BLOCK_DATA_SIZE]{}; }; } // namespace DiscIO