forked from dolphin-emu/dolphin
		
	
		
			
				
	
	
		
			167 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			167 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
// Copyright 2009 Dolphin Emulator Project
 | 
						|
// Licensed under GPLv2+
 | 
						|
// Refer to the license.txt file included.
 | 
						|
 | 
						|
#include <cstddef>
 | 
						|
#include <cstring>
 | 
						|
#include <locale>
 | 
						|
#include <map>
 | 
						|
#include <memory>
 | 
						|
#include <string>
 | 
						|
#include <utility>
 | 
						|
#include <vector>
 | 
						|
 | 
						|
#include "Common/Align.h"
 | 
						|
#include "Common/Assert.h"
 | 
						|
#include "Common/CommonTypes.h"
 | 
						|
#include "Common/Logging/Log.h"
 | 
						|
#include "Common/MsgHandler.h"
 | 
						|
#include "Common/StringUtil.h"
 | 
						|
#include "DiscIO/Blob.h"
 | 
						|
#include "DiscIO/Enums.h"
 | 
						|
#include "DiscIO/Volume.h"
 | 
						|
#include "DiscIO/VolumeWad.h"
 | 
						|
 | 
						|
namespace DiscIO
 | 
						|
{
 | 
						|
CVolumeWAD::CVolumeWAD(std::unique_ptr<IBlobReader> reader) : m_reader(std::move(reader))
 | 
						|
{
 | 
						|
  _assert_(m_reader);
 | 
						|
 | 
						|
  // Source: http://wiibrew.org/wiki/WAD_files
 | 
						|
  ReadSwapped(0x00, &m_hdr_size, false);
 | 
						|
  ReadSwapped(0x08, &m_cert_size, false);
 | 
						|
  ReadSwapped(0x10, &m_tick_size, false);
 | 
						|
  ReadSwapped(0x14, &m_tmd_size, false);
 | 
						|
  ReadSwapped(0x18, &m_data_size, false);
 | 
						|
 | 
						|
  m_offset = Common::AlignUp(m_hdr_size, 0x40) + Common::AlignUp(m_cert_size, 0x40);
 | 
						|
  m_tmd_offset = Common::AlignUp(m_hdr_size, 0x40) + Common::AlignUp(m_cert_size, 0x40) +
 | 
						|
                 Common::AlignUp(m_tick_size, 0x40);
 | 
						|
  m_opening_bnr_offset =
 | 
						|
      m_tmd_offset + Common::AlignUp(m_tmd_size, 0x40) + Common::AlignUp(m_data_size, 0x40);
 | 
						|
 | 
						|
  if (m_tmd_size > 1024 * 1024 * 4)
 | 
						|
  {
 | 
						|
    ERROR_LOG(DISCIO, "TMD is too large: %u bytes", m_tmd_size);
 | 
						|
    return;
 | 
						|
  }
 | 
						|
 | 
						|
  std::vector<u8> tmd_buffer(m_tmd_size);
 | 
						|
  Read(m_tmd_offset, m_tmd_size, tmd_buffer.data(), false);
 | 
						|
  m_tmd.SetBytes(std::move(tmd_buffer));
 | 
						|
}
 | 
						|
 | 
						|
CVolumeWAD::~CVolumeWAD()
 | 
						|
{
 | 
						|
}
 | 
						|
 | 
						|
bool CVolumeWAD::Read(u64 offset, u64 length, u8* buffer, bool decrypt) const
 | 
						|
{
 | 
						|
  if (decrypt)
 | 
						|
    PanicAlertT("Tried to decrypt data from a non-Wii volume");
 | 
						|
 | 
						|
  return m_reader->Read(offset, length, buffer);
 | 
						|
}
 | 
						|
 | 
						|
Region CVolumeWAD::GetRegion() const
 | 
						|
{
 | 
						|
  if (!m_tmd.IsValid())
 | 
						|
    return Region::UNKNOWN_REGION;
 | 
						|
  return m_tmd.GetRegion();
 | 
						|
}
 | 
						|
 | 
						|
Country CVolumeWAD::GetCountry() const
 | 
						|
{
 | 
						|
  if (!m_tmd.IsValid())
 | 
						|
    return Country::COUNTRY_UNKNOWN;
 | 
						|
 | 
						|
  u8 country_code = static_cast<u8>(m_tmd.GetTitleId() & 0xff);
 | 
						|
  if (country_code == 2)  // SYSMENU
 | 
						|
    return TypicalCountryForRegion(GetSysMenuRegion(m_tmd.GetTitleVersion()));
 | 
						|
 | 
						|
  return CountrySwitch(country_code);
 | 
						|
}
 | 
						|
 | 
						|
IOS::ES::TMDReader CVolumeWAD::GetTMD() const
 | 
						|
{
 | 
						|
  return m_tmd;
 | 
						|
}
 | 
						|
 | 
						|
std::string CVolumeWAD::GetGameID() const
 | 
						|
{
 | 
						|
  return m_tmd.GetGameID();
 | 
						|
}
 | 
						|
 | 
						|
std::string CVolumeWAD::GetMakerID() const
 | 
						|
{
 | 
						|
  char temp[2];
 | 
						|
  if (!Read(0x198 + m_tmd_offset, 2, (u8*)temp))
 | 
						|
    return "00";
 | 
						|
 | 
						|
  // Some weird channels use 0x0000 in place of the MakerID, so we need a check here
 | 
						|
  const std::locale& c_locale = std::locale::classic();
 | 
						|
  if (!std::isprint(temp[0], c_locale) || !std::isprint(temp[1], c_locale))
 | 
						|
    return "00";
 | 
						|
 | 
						|
  return DecodeString(temp);
 | 
						|
}
 | 
						|
 | 
						|
bool CVolumeWAD::GetTitleID(u64* buffer) const
 | 
						|
{
 | 
						|
  return ReadSwapped(m_offset + 0x01DC, buffer, false);
 | 
						|
}
 | 
						|
 | 
						|
u16 CVolumeWAD::GetRevision() const
 | 
						|
{
 | 
						|
  if (!m_tmd.IsValid())
 | 
						|
    return 0;
 | 
						|
 | 
						|
  return m_tmd.GetTitleVersion();
 | 
						|
}
 | 
						|
 | 
						|
Platform CVolumeWAD::GetVolumeType() const
 | 
						|
{
 | 
						|
  return Platform::WII_WAD;
 | 
						|
}
 | 
						|
 | 
						|
std::map<Language, std::string> CVolumeWAD::GetLongNames() const
 | 
						|
{
 | 
						|
  if (!m_tmd.IsValid() || !IOS::ES::IsChannel(m_tmd.GetTitleId()))
 | 
						|
    return {};
 | 
						|
 | 
						|
  std::vector<u8> name_data(NAMES_TOTAL_BYTES);
 | 
						|
  if (!Read(m_opening_bnr_offset + 0x9C, NAMES_TOTAL_BYTES, name_data.data()))
 | 
						|
    return std::map<Language, std::string>();
 | 
						|
  return ReadWiiNames(name_data);
 | 
						|
}
 | 
						|
 | 
						|
std::vector<u32> CVolumeWAD::GetBanner(int* width, int* height) const
 | 
						|
{
 | 
						|
  *width = 0;
 | 
						|
  *height = 0;
 | 
						|
 | 
						|
  u64 title_id;
 | 
						|
  if (!GetTitleID(&title_id))
 | 
						|
    return std::vector<u32>();
 | 
						|
 | 
						|
  return GetWiiBanner(width, height, title_id);
 | 
						|
}
 | 
						|
 | 
						|
BlobType CVolumeWAD::GetBlobType() const
 | 
						|
{
 | 
						|
  return m_reader->GetBlobType();
 | 
						|
}
 | 
						|
 | 
						|
u64 CVolumeWAD::GetSize() const
 | 
						|
{
 | 
						|
  return m_reader->GetDataSize();
 | 
						|
}
 | 
						|
 | 
						|
u64 CVolumeWAD::GetRawSize() const
 | 
						|
{
 | 
						|
  return m_reader->GetRawSize();
 | 
						|
}
 | 
						|
 | 
						|
}  // namespace
 |