// 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 <string> #include <vector> #include "Common/CommonTypes.h" #include "DiscIO/Volume.h" // --- this volume type is used for encrypted Wii images --- namespace DiscIO { enum class BlobType; enum class Country; enum class Language; enum class Region; enum class Platform; class CVolumeWiiCrypted : public IVolume { public: CVolumeWiiCrypted(std::unique_ptr<IBlobReader> reader, u64 _VolumeOffset, const unsigned char* _pVolumeKey); ~CVolumeWiiCrypted(); bool Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt) const override; bool GetTitleID(u64* buffer) const override; std::vector<u8> GetTMD() const override; std::string GetGameID() const override; std::string GetMakerID() const override; u16 GetRevision() const override; std::string GetInternalName() const override; std::map<Language, std::string> GetLongNames() const override; std::vector<u32> GetBanner(int* width, int* height) const override; u64 GetFSTSize() const override; std::string GetApploaderDate() const override; u8 GetDiscNumber() const override; Platform GetVolumeType() const override; bool SupportsIntegrityCheck() const override { return true; } bool CheckIntegrity() const override; bool ChangePartition(u64 offset) override; Region GetRegion() const override; Country GetCountry() const override; BlobType GetBlobType() const override; u64 GetSize() const override; u64 GetRawSize() const override; private: static const unsigned int s_block_header_size = 0x0400; static const unsigned int s_block_data_size = 0x7C00; static const unsigned int s_block_total_size = s_block_header_size + s_block_data_size; std::unique_ptr<IBlobReader> m_pReader; std::unique_ptr<mbedtls_aes_context> m_AES_ctx; u64 m_VolumeOffset; u64 m_dataOffset; mutable u64 m_LastDecryptedBlockOffset; mutable unsigned char m_LastDecryptedBlock[s_block_data_size]; }; } // namespace