The-Homebrew-Cloud/dolphin
2
0
Fork 0
mirror of https://github.com/dolphin-emu/dolphin.git synced 2025-01-26 15:55:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

WiiSave: Use std::array instead of C arrays

Browse Source 
Makes copying, comparing more readable
This commit is contained in:
Léo Lam 2018-05-13 13:34:02 +02:00
parent a93d816c28
commit 87e5cd73dc
2 changed files with 46 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
Source/Core/Core/HW/WiiSave.cpp
View File

@ -30,10 +30,14 @@
#include "Common/NandPaths.h" #include "Common/NandPaths.h"
#include "Common/StringUtil.h" #include "Common/StringUtil.h"
constexpr u8 s_sd_key[16] = {0xAB, 0x01, 0xB9, 0xD8, 0xE1, 0x62, 0x2B, 0x08, using Md5 = std::array<u8, 0x10>;
0xAF, 0xBA, 0xD8, 0x4D, 0xBF, 0xC2, 0xA5, 0x5D};
constexpr u8 s_md5_blanker[16] = {0x0E, 0x65, 0x37, 0x81, 0x99, 0xBE, 0x45, 0x17, constexpr std::array<u8, 0x10> s_sd_initial_iv{{0x21, 0x67, 0x12, 0xE6, 0xAA, 0x1F, 0x68, 0x9F,
0xAB, 0x06, 0xEC, 0x22, 0x45, 0x1A, 0x57, 0x93}; 0x95, 0xC5, 0xA2, 0x23, 0x24, 0xDC, 0x6A, 0x98}};
constexpr std::array<u8, 0x10> s_sd_key{{0xAB, 0x01, 0xB9, 0xD8, 0xE1, 0x62, 0x2B, 0x08, 0xAF, 0xBA,
0xD8, 0x4D, 0xBF, 0xC2, 0xA5, 0x5D}};
constexpr Md5 s_md5_blanker{{0x0E, 0x65, 0x37, 0x81, 0x99, 0xBE, 0x45, 0x17, 0xAB, 0x06, 0xEC, 0x22,
0x45, 0x1A, 0x57, 0x93}};
constexpr u32 s_ng_id = 0x0403AC68; constexpr u32 s_ng_id = 0x0403AC68;
bool WiiSave::Import(const std::string& filename) bool WiiSave::Import(const std::string& filename)
@ -86,10 +90,10 @@ std::pair<size_t, std::string> WiiSave::ExportAll()
return {exported_save_count, File::GetUserPath(D_USER_IDX) + "private/wii/title/"}; return {exported_save_count, File::GetUserPath(D_USER_IDX) + "private/wii/title/"};
} }
WiiSave::WiiSave(std::string filename) : m_encrypted_save_path(std::move(filename)), m_valid{true} WiiSave::WiiSave(std::string filename)
: m_sd_iv{s_sd_initial_iv}, m_encrypted_save_path(std::move(filename)), m_valid{true}
{ {
memcpy(m_sd_iv, "\x21\x67\x12\xE6\xAA\x1F\x68\x9F\x95\xC5\xA2\x23\x24\xDC\x6A\x98", 0x10); mbedtls_aes_setkey_dec(&m_aes_ctx, s_sd_key.data(), 128);
mbedtls_aes_setkey_dec(&m_aes_ctx, s_sd_key, 128);
} }
bool WiiSave::Import() bool WiiSave::Import()
@ -101,10 +105,9 @@ bool WiiSave::Import()
return m_valid; return m_valid;
} }
WiiSave::WiiSave(u64 title_id) : m_title_id{title_id} WiiSave::WiiSave(u64 title_id) : m_sd_iv{s_sd_initial_iv}, m_title_id{title_id}
{ {
memcpy(m_sd_iv, "\x21\x67\x12\xE6\xAA\x1F\x68\x9F\x95\xC5\xA2\x23\x24\xDC\x6A\x98", 0x10); mbedtls_aes_setkey_enc(&m_aes_ctx, s_sd_key.data(), 128);
mbedtls_aes_setkey_enc(&m_aes_ctx, s_sd_key, 128);
if (getPaths(true)) if (getPaths(true))
m_valid = true; m_valid = true;
@ -136,7 +139,7 @@ void WiiSave::ReadHDR()
} }
data_file.Close(); data_file.Close();
mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_DECRYPT, HEADER_SZ, m_sd_iv, mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_DECRYPT, HEADER_SZ, m_sd_iv.data(),
(const u8*)&m_encrypted_header, (u8*)&m_header); (const u8*)&m_encrypted_header, (u8*)&m_header);
u32 banner_size = m_header.hdr.banner_size; u32 banner_size = m_header.hdr.banner_size;
if ((banner_size < FULL_BNR_MIN) || (banner_size > FULL_BNR_MAX) || if ((banner_size < FULL_BNR_MIN) || (banner_size > FULL_BNR_MAX) ||
@ -148,16 +151,15 @@ void WiiSave::ReadHDR()
} }
m_title_id = m_header.hdr.save_game_title; m_title_id = m_header.hdr.save_game_title;
u8 md5_file[16]; Md5 md5_file = m_header.hdr.md5;
u8 md5_calc[16]; m_header.hdr.md5 = s_md5_blanker;
memcpy(md5_file, m_header.hdr.md5, 0x10); Md5 md5_calc;
memcpy(m_header.hdr.md5, s_md5_blanker, 0x10); mbedtls_md5((u8*)&m_header, HEADER_SZ, md5_calc.data());
mbedtls_md5((u8*)&m_header, HEADER_SZ, md5_calc); if (md5_file == md5_calc)
if (memcmp(md5_file, md5_calc, 0x10))
{ {
ERROR_LOG(CONSOLE, "MD5 mismatch\n %016" PRIx64 "%016" PRIx64 " != %016" PRIx64 "%016" PRIx64, ERROR_LOG(CONSOLE, "MD5 mismatch\n %016" PRIx64 "%016" PRIx64 " != %016" PRIx64 "%016" PRIx64,
Common::swap64(md5_file), Common::swap64(md5_file + 8), Common::swap64(md5_calc), Common::swap64(md5_file.data()), Common::swap64(md5_file.data() + 8),
Common::swap64(md5_calc + 8)); Common::swap64(md5_calc.data()), Common::swap64(md5_calc.data() + 8));
m_valid = false; m_valid = false;
} }
@ -193,7 +195,7 @@ void WiiSave::WriteHDR()
m_header.hdr.banner_size = banner_size; m_header.hdr.banner_size = banner_size;
m_header.hdr.save_game_title = m_title_id; m_header.hdr.save_game_title = m_title_id;
memcpy(m_header.hdr.md5, s_md5_blanker, 0x10); m_header.hdr.md5 = s_md5_blanker;
m_header.hdr.permissions = 0x3C; m_header.hdr.permissions = 0x3C;
File::IOFile banner_file(banner_file_path, "rb"); File::IOFile banner_file(banner_file_path, "rb");
@ -206,12 +208,12 @@ void WiiSave::WriteHDR()
// remove nocopy flag // remove nocopy flag
m_header.banner[7] &= ~1; m_header.banner[7] &= ~1;
u8 md5_calc[16]; Md5 md5_calc;
mbedtls_md5((u8*)&m_header, HEADER_SZ, md5_calc); mbedtls_md5((u8*)&m_header, HEADER_SZ, md5_calc.data());
memcpy(m_header.hdr.md5, md5_calc, 0x10); m_header.hdr.md5 = std::move(md5_calc);
mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_ENCRYPT, HEADER_SZ, m_sd_iv, (const u8*)&m_header, mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_ENCRYPT, HEADER_SZ, m_sd_iv.data(),
(u8*)&m_encrypted_header); (const u8*)&m_header, (u8*)&m_encrypted_header);
File::IOFile data_file(m_encrypted_save_path, "wb"); File::IOFile data_file(m_encrypted_save_path, "wb");
if (!data_file.WriteBytes(&m_encrypted_header, HEADER_SZ)) if (!data_file.WriteBytes(&m_encrypted_header, HEADER_SZ))
@ -306,7 +308,7 @@ void WiiSave::ImportWiiSaveFiles()
for (u32 i = 0; i < m_bk_hdr.number_of_files; ++i) for (u32 i = 0; i < m_bk_hdr.number_of_files; ++i)
{ {
memset(&file_hdr_tmp, 0, FILE_HDR_SZ); memset(&file_hdr_tmp, 0, FILE_HDR_SZ);
memset(m_iv, 0, 0x10); m_iv.fill(0);
if (!data_file.ReadBytes(&file_hdr_tmp, FILE_HDR_SZ)) if (!data_file.ReadBytes(&file_hdr_tmp, FILE_HDR_SZ))
{ {
@ -323,7 +325,7 @@ void WiiSave::ImportWiiSaveFiles()
{ {
// Allows files in subfolders to be escaped properly (ex: "nocopy/data00") // Allows files in subfolders to be escaped properly (ex: "nocopy/data00")
// Special characters in path components will be escaped such as /../ // Special characters in path components will be escaped such as /../
std::string file_path = Common::EscapePath(reinterpret_cast<const char*>(file_hdr_tmp.name)); std::string file_path = Common::EscapePath(file_hdr_tmp.name.data());
std::string file_path_full = m_wii_title_path + '/' + file_path; std::string file_path_full = m_wii_title_path + '/' + file_path;
File::CreateFullPath(file_path_full); File::CreateFullPath(file_path_full);
@ -341,8 +343,8 @@ void WiiSave::ImportWiiSaveFiles()
break; break;
} }
memcpy(m_iv, file_hdr_tmp.iv, 0x10); m_iv = file_hdr_tmp.iv;
mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_DECRYPT, file_size_rounded, m_iv, mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_DECRYPT, file_size_rounded, m_iv.data(),
static_cast<const u8*>(file_data_enc.data()), file_data.data()); static_cast<const u8*>(file_data_enc.data()), file_data.data());
INFO_LOG(CONSOLE, "Creating file %s", file_path_full.c_str()); INFO_LOG(CONSOLE, "Creating file %s", file_path_full.c_str());
@ -405,7 +407,7 @@ void WiiSave::ExportWiiSaveFiles()
m_valid = false; m_valid = false;
return; return;
} }
strncpy((char*)file_hdr_tmp.name, name.c_str(), sizeof(file_hdr_tmp.name)); std::strncpy(file_hdr_tmp.name.data(), name.c_str(), file_hdr_tmp.name.size());
{ {
File::IOFile fpData_bin(m_encrypted_save_path, "ab"); File::IOFile fpData_bin(m_encrypted_save_path, "ab");
@ -436,8 +438,9 @@ void WiiSave::ExportWiiSaveFiles()
m_valid = false; m_valid = false;
} }
mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_ENCRYPT, file_size_rounded, file_hdr_tmp.iv, mbedtls_aes_crypt_cbc(&m_aes_ctx, MBEDTLS_AES_ENCRYPT, file_size_rounded,
static_cast<const u8*>(file_data.data()), file_data_enc.data()); file_hdr_tmp.iv.data(), static_cast<const u8*>(file_data.data()),
file_data_enc.data());
File::IOFile fpData_bin(m_encrypted_save_path, "ab"); File::IOFile fpData_bin(m_encrypted_save_path, "ab");
if (!fpData_bin.WriteBytes(file_data_enc.data(), file_size_rounded)) if (!fpData_bin.WriteBytes(file_data_enc.data(), file_size_rounded))

19
Source/Core/Core/HW/WiiSave.h
View File

@ -4,6 +4,7 @@
#pragma once #pragma once
#include <array>
#include <mbedtls/aes.h> #include <mbedtls/aes.h>
#include <string> #include <string>
#include <utility> #include <utility>
@ -45,14 +46,14 @@ private:
u32* num_files, u32* size_files); u32* num_files, u32* size_files);
mbedtls_aes_context m_aes_ctx; mbedtls_aes_context m_aes_ctx;
u8 m_sd_iv[0x10]; std::array<u8, 0x10> m_sd_iv;
std::vector<std::string> m_files_list; std::vector<std::string> m_files_list;
std::string m_encrypted_save_path; std::string m_encrypted_save_path;
std::string m_wii_title_path; std::string m_wii_title_path;
u8 m_iv[0x10]; std::array<u8, 0x10> m_iv;
u64 m_title_id; u64 m_title_id;
@ -88,7 +89,7 @@ private:
Common::BigEndianValue<u32> banner_size; // (0x72A0 or 0xF0A0, also seen 0xBAA0) Common::BigEndianValue<u32> banner_size; // (0x72A0 or 0xF0A0, also seen 0xBAA0)
u8 permissions; u8 permissions;
u8 unk1; // maybe permissions is a be16 u8 unk1; // maybe permissions is a be16
u8 md5[0x10]; // md5 of plaintext header with md5 blanker applied std::array<u8, 0x10> md5; // md5 of plaintext header with md5 blanker applied
Common::BigEndianValue<u16> unk2; Common::BigEndianValue<u16> unk2;
}; };
@ -110,10 +111,10 @@ private:
Common::BigEndianValue<u32> unk1; Common::BigEndianValue<u32> unk1;
Common::BigEndianValue<u32> unk2; Common::BigEndianValue<u32> unk2;
Common::BigEndianValue<u32> total_size; Common::BigEndianValue<u32> total_size;
u8 unk3[64]; std::array<u8, 64> unk3;
Common::BigEndianValue<u64> save_game_title; Common::BigEndianValue<u64> save_game_title;
u8 mac_address[6]; std::array<u8, 6> mac_address;
u8 padding[0x12]; std::array<u8, 0x12> padding;
}; };
struct FileHDR // encrypted struct FileHDR // encrypted
@ -123,9 +124,9 @@ private:
u8 permissions; u8 permissions;
u8 attrib; u8 attrib;
u8 type; // (1=file, 2=directory) u8 type; // (1=file, 2=directory)
u8 name[0x45]; std::array<char, 0x45> name;
u8 iv[0x10]; std::array<u8, 0x10> iv;
u8 unk[0x20]; std::array<u8, 0x20> unk;
}; };
#pragma pack(pop) #pragma pack(pop)