// Copyright 2008 Dolphin Emulator Project // Licensed under GPLv2+ // Refer to the license.txt file included. #pragma once #include <map> #include <mbedtls/aes.h> #include <memory> #include <optional> #include <string> #include <vector> #include "Common/CommonTypes.h" #include "Common/Lazy.h" #include "Core/IOS/ES/Formats.h" #include "DiscIO/Filesystem.h" #include "DiscIO/Volume.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 Volume { public: VolumeWii(std::unique_ptr<BlobReader> reader); ~VolumeWii(); bool Read(u64 _Offset, u64 _Length, u8* _pBuffer, const Partition& partition) 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 FileSystem* GetFileSystem(const Partition& partition) const override; std::string GetGameID(const Partition& partition) const override; std::string GetMakerID(const Partition& partition) const override; std::optional<u16> GetRevision(const Partition& partition) const override; std::string GetInternalName(const Partition& partition) const override; std::map<Language, std::string> GetLongNames() const override; std::vector<u32> GetBanner(int* width, int* height) const override; std::string GetApploaderDate(const Partition& partition) const override; std::optional<u8> GetDiscNumber(const Partition& partition) const override; Platform GetVolumeType() const override; bool SupportsIntegrityCheck() const override { return true; } bool CheckIntegrity(const Partition& partition) const override; Region GetRegion() const override; Country GetCountry(const Partition& partition) const override; BlobType GetBlobType() const override; u64 GetSize() const override; u64 GetRawSize() const override; static u64 PartitionOffsetToRawOffset(u64 offset, const Partition& partition); static constexpr unsigned int BLOCK_HEADER_SIZE = 0x0400; static constexpr unsigned int BLOCK_DATA_SIZE = 0x7C00; static constexpr unsigned int BLOCK_TOTAL_SIZE = BLOCK_HEADER_SIZE + BLOCK_DATA_SIZE; protected: u32 GetOffsetShift() const override { return 2; } private: struct PartitionDetails { Common::Lazy<std::unique_ptr<mbedtls_aes_context>> key; Common::Lazy<IOS::ES::TicketReader> ticket; Common::Lazy<IOS::ES::TMDReader> tmd; Common::Lazy<std::unique_ptr<FileSystem>> file_system; u32 type; }; std::unique_ptr<BlobReader> m_pReader; std::map<Partition, PartitionDetails> m_partitions; Partition m_game_partition; mutable u64 m_last_decrypted_block; mutable u8 m_last_decrypted_block_data[BLOCK_DATA_SIZE]; }; } // namespace