mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2024-11-15 22:09:19 -07:00
ca0438e26d
This adds a WiiWad constructor that takes a BlobReader, so that the class can be used with more than just files from the host filesystem. Required for using WiiWad with WADs from update partitions.
113 lines
3.2 KiB
C++
113 lines
3.2 KiB
C++
// Copyright 2009 Dolphin Emulator Project
|
|
// Licensed under GPLv2+
|
|
// Refer to the license.txt file included.
|
|
|
|
#include <cstddef>
|
|
#include <memory>
|
|
#include <optional>
|
|
#include <string>
|
|
|
|
#include "Common/Align.h"
|
|
#include "Common/Assert.h"
|
|
#include "Common/CommonTypes.h"
|
|
#include "Common/FileUtil.h"
|
|
#include "Common/Logging/Log.h"
|
|
#include "DiscIO/Blob.h"
|
|
#include "DiscIO/WiiWad.h"
|
|
|
|
namespace DiscIO
|
|
{
|
|
namespace
|
|
{
|
|
std::vector<u8> CreateWADEntry(BlobReader& reader, u32 size, u64 offset)
|
|
{
|
|
if (size == 0)
|
|
return {};
|
|
|
|
std::vector<u8> buffer(size);
|
|
|
|
if (!reader.Read(offset, size, buffer.data()))
|
|
{
|
|
ERROR_LOG(DISCIO, "WiiWAD: Could not read from file");
|
|
PanicAlertT("WiiWAD: Could not read from file");
|
|
}
|
|
|
|
return buffer;
|
|
}
|
|
|
|
bool IsWiiWAD(BlobReader& reader)
|
|
{
|
|
const std::optional<u32> header_size = reader.ReadSwapped<u32>(0x0);
|
|
const std::optional<u32> header_type = reader.ReadSwapped<u32>(0x4);
|
|
return header_size == u32(0x20) &&
|
|
(header_type == u32(0x49730000) || header_type == u32(0x69620000));
|
|
}
|
|
} // Anonymous namespace
|
|
|
|
WiiWAD::WiiWAD(const std::string& name) : WiiWAD(CreateBlobReader(name))
|
|
{
|
|
}
|
|
|
|
WiiWAD::WiiWAD(std::unique_ptr<BlobReader> blob_reader) : m_reader(std::move(blob_reader))
|
|
{
|
|
if (m_reader)
|
|
m_valid = ParseWAD();
|
|
}
|
|
|
|
WiiWAD::~WiiWAD() = default;
|
|
|
|
bool WiiWAD::ParseWAD()
|
|
{
|
|
if (!IsWiiWAD(*m_reader))
|
|
return false;
|
|
|
|
std::optional<u32> certificate_chain_size = m_reader->ReadSwapped<u32>(0x08);
|
|
std::optional<u32> reserved = m_reader->ReadSwapped<u32>(0x0C);
|
|
std::optional<u32> ticket_size = m_reader->ReadSwapped<u32>(0x10);
|
|
std::optional<u32> tmd_size = m_reader->ReadSwapped<u32>(0x14);
|
|
std::optional<u32> data_app_size = m_reader->ReadSwapped<u32>(0x18);
|
|
std::optional<u32> footer_size = m_reader->ReadSwapped<u32>(0x1C);
|
|
if (!certificate_chain_size || !reserved || !ticket_size || !tmd_size || !data_app_size ||
|
|
!footer_size)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (MAX_LOGLEVEL >= LogTypes::LOG_LEVELS::LDEBUG)
|
|
_dbg_assert_msg_(BOOT, *reserved == 0x00, "WiiWAD: Reserved must be 0x00");
|
|
|
|
u32 offset = 0x40;
|
|
m_certificate_chain = CreateWADEntry(*m_reader, *certificate_chain_size, offset);
|
|
offset += Common::AlignUp(*certificate_chain_size, 0x40);
|
|
m_ticket.SetBytes(CreateWADEntry(*m_reader, *ticket_size, offset));
|
|
offset += Common::AlignUp(*ticket_size, 0x40);
|
|
m_tmd.SetBytes(CreateWADEntry(*m_reader, *tmd_size, offset));
|
|
offset += Common::AlignUp(*tmd_size, 0x40);
|
|
m_data_app_offset = offset;
|
|
m_data_app = CreateWADEntry(*m_reader, *data_app_size, offset);
|
|
offset += Common::AlignUp(*data_app_size, 0x40);
|
|
m_footer = CreateWADEntry(*m_reader, *footer_size, offset);
|
|
offset += Common::AlignUp(*footer_size, 0x40);
|
|
|
|
return true;
|
|
}
|
|
|
|
std::vector<u8> WiiWAD::GetContent(u16 index) const
|
|
{
|
|
u64 offset = m_data_app_offset;
|
|
for (const IOS::ES::Content& content : m_tmd.GetContents())
|
|
{
|
|
const u64 aligned_size = Common::AlignUp(content.size, 0x40);
|
|
if (content.index == index)
|
|
{
|
|
std::vector<u8> data(aligned_size);
|
|
if (!m_reader->Read(offset, aligned_size, data.data()))
|
|
return {};
|
|
return data;
|
|
}
|
|
offset += aligned_size;
|
|
}
|
|
return {};
|
|
}
|
|
} // namespace DiscIO
|