The-Homebrew-Cloud/YAWM-ModMii-Edition
2
0
Fork 0
mirror of https://github.com/modmii/YAWM-ModMii-Edition.git synced 2025-02-25 17:03:38 +01:00
Code Issues Packages Projects Releases Wiki Activity
YAWM-ModMii-Edition/source/wad.c
Naim2000 c86808e911 work in progress
2025-02-12 21:28:36 -05:00

1334 lines
30 KiB
C
Raw Blame History

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <ogcsys.h>
#include <ogc/pad.h>
#include <ogc/es.h>
#include <ogc/aes.h>
#include "sys.h"
#include "title.h"
#include "utils.h"
#include "video.h"
#include "wad.h"
#include "wpad.h"
#include "nand.h"
#include "fileops.h"
#include "sha1.h"
#include "menu.h"
#include "iospatch.h"
#include "malloc.h"
// Turn upper and lower into a full title ID
#define TITLE_ID(x,y) (((u64)(x) << 32) | (y))
// Get upper or lower half of a title ID
#define TITLE_UPPER(x) ((u32)((x) >> 32))
// Turn upper and lower into a full title ID
#define TITLE_LOWER(x) ((u32)(x))
const char RegionLookupTable[16] =
{
'J', 'U', 'E', 0, 0, 0, 'K', 0, 0, 0, 0, 0, 0, 0, 0, 0,
};
const u16 VersionList[] =
{
// J U E K
64, 33, 66, // 1.0
128, 97, 130, // 2.0
162, // 2.1
192, 193, 194, // 2.2
224, 225, 226, // 3.0
256, 257, 258, // 3.1
288, 289, 290, // 3.2
352, 353, 354, 326, // 3.3
384, 385, 386, // 3.4
390, // 3.5
416, 417, 418, // 4.0
448, 449, 450, 454, // 4.1
480, 481, 482, 486, // 4.2
512, 513, 514, 518, // 4.3/vWii 1.0.0
544, 545, 546, // vWii 4.0.0
608, 609, 610 // vWii 5.2.0
};
u32 VersionListSize = sizeof(VersionList) / sizeof(VersionList[0]);
const char* VersionLookupTable[7][20] =
{
// 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
{ "", "", "1.0", "", "2.0", "", "2.2", "3.0", "3.1", "3.2", "", "3.3", "3.4", "4.0", "4.1", "4.2", "4.3", "4.3", "", "4.3" },
{ "", "1.0", "", "2.0", "", "", "2.2", "3.0", "3.1", "3.2", "", "3.3", "3.4", "4.0", "4.1", "4.2", "4.3", "4.3", "", "4.3" },
{ "", "", "1.0", "", "2.0", "2.1", "2.2", "3.0", "3.1", "3.2", "", "3.3", "3.4", "4.0", "4.1", "4.2", "4.3", "4.3", "", "4.3" },
{ "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "" },
{ "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "" },
{ "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "" },
{ "", "", "", "", "", "", "", "", "", "", "3.3", "", "3.5", "", "4.1", "4.2", "4.3", "", "", "" },
};
u32 WaitButtons(void);
static u32 gPriiloaderSize = 0;
static bool gForcedInstall = false;
u32 be32(const u8 *p)
{
return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
}
u64 be64(const u8 *p)
{
return ((u64)be32(p) << 32) | be32(p + 4);
}
static void DecEncTxtBuffer(char* buffer)
{
u32 key = 0x73B5DBFA;
s32 i;
for (i = 0; i < 0x100; i++)
{
buffer[i] ^= key & 0xFF;
key = (key << 1) | (key >> 31);
}
}
static bool GetRegionFromTXT(char* region)
{
u32 size = 0;
*region = 0;
char* buffer = (char*)NANDLoadFile("/title/00000001/00000002/data/setting.txt", &size);
if (!buffer)
return false;
DecEncTxtBuffer(buffer);
char* current = strstr(buffer, "AREA");
if(current)
{
char* start = strchr(current, '=');
char* end = strchr(current, '\n');
if (start && end)
{
start++;
if (!strncmp(start, "JPN", 3) || !strncmp(start, "TWN", 3) || !strncmp(start, "ROC", 3))
*region = 'J';
else if (!strncmp(start, "USA", 3) || !strncmp(start, "BRA", 3) || !strncmp(start, "HKG", 3) ||
!strncmp(start, "ASI", 3) || !strncmp(start, "LTN", 3) || !strncmp(start, "SAF", 3))
*region = 'U';
else if (!strncmp(start, "EUR", 3) || !strncmp(start, "AUS", 3))
*region = 'E';
else if (!strncmp(start, "KOR", 3))
*region = 'K';
else if (!strncmp(start, "CHN", 3))
printf("Error! GetRegionFromTXT: Unhandled product area \"CHN\"!\n");
if (*region != 0)
{
free(buffer);
return true;
}
}
}
else
{
printf("Error! GetRegionFromTXT: Item AREA not found!\n");
}
free(buffer);
return false;
}
s32 GetSysMenuRegion(u16* version, char* region)
{
u16 v = 0;
s32 ret = Title_GetVersion(0x100000002LL, &v);
if (ret < 0)
return ret;
if (version)
*version = v;
if(!GetRegionFromTXT(region))
{
printf("\nCouldn't find the region of this system\n");
sleep(5);
return -1;
}
return 0;
}
bool VersionIsOriginal(u16 version)
{
s32 i;
for (i = 0; i < VersionListSize; i++)
{
if (VersionList[i] == version)
return true;
}
return false;
}
const char* GetSysMenuRegionString(const char region)
{
switch (region)
{
case 'J': return "Japan (NTSC-J)";
case 'U': return "USA (NTSC-U/C)";
case 'E': return "Europe (PAL)";
case 'K': return "Korea (NTSC-K)";
}
return "Unknown";
}
const char* GetSysMenuVersionString(u16 version)
{
/*
* ==== ==== === version
* 0000 0010 0110 0000
* = ==== region
*/
return VersionLookupTable[version % 32][version / 32];
};
static u32 GetSysMenuBootContent(void)
{
s32 ret;
u32 cid = 0;
u32 size = 0;
signed_blob *s_tmd = NULL;
ret = ES_GetStoredTMDSize(0x100000002LL, &size);
if (!size)
{
printf("Error! ES_GetStoredTMDSize failed (ret=%i)\n", ret);
return 0;
}
s_tmd = memalign32(size);
if (!s_tmd)
{
printf("Error! Memory allocation failed!\n");
return 0;
}
ret = ES_GetStoredTMD(0x100000002LL, s_tmd, size);
if (ret < 0)
{
printf("Error! ES_GetStoredTMD failed (ret=%i)\n", ret);
free(s_tmd);
return 0;
}
tmd *p_tmd = SIGNATURE_PAYLOAD(s_tmd);
for (int i = 0; i < p_tmd->num_contents; i++)
{
tmd_content* content = &p_tmd->contents[i];
if (content->index == p_tmd->boot_index)
{
cid = content->cid;
break;
}
}
free(s_tmd);
if (!cid) printf("Error! Cannot find system menu boot content!\n");
return cid;
}
bool GetSysMenuExecPath(char path[ISFS_MAXPATH], bool mainDOL)
{
u32 cid = GetSysMenuBootContent();
if (!cid) return false;
if (mainDOL)
cid = (cid & ~0xF00000000) | 0x10000000;
sprintf(path, "/title/00000001/00000002/content/%08x.app", cid);
return true;
}
bool IsPriiloaderInstalled()
{
char path[ISFS_MAXPATH] ATTRIBUTE_ALIGN(0x20);
if (!GetSysMenuExecPath(path, true))
return false;
u32 size = 0;
NANDGetFileSize(path, &size);
return (size > 0);
}
static bool BackUpPriiloader()
{
char path[ISFS_MAXPATH] ATTRIBUTE_ALIGN(0x20);
if (!GetSysMenuExecPath(path, false))
return false;
u32 size = 0;
s32 ret = NANDBackUpFile(path, "/tmp/priiload.app", &size);
if (ret < 0)
{
printf("Error! NANDBackUpFile: Failed! (Error: %d)\n", ret);
return false;
}
ret = NANDGetFileSize("/tmp/priiload.app", &gPriiloaderSize);
return (gPriiloaderSize == size);
}
static bool MoveMenu(bool restore)
{
ATTRIBUTE_ALIGN(0x20)
char srcPath[ISFS_MAXPATH], dstPath[ISFS_MAXPATH];
if (!GetSysMenuBootContent())
return false;
GetSysMenuExecPath(srcPath, restore);
GetSysMenuExecPath(dstPath, !restore);
u32 size = 0;
s32 ret = NANDBackUpFile(srcPath, dstPath, &size);
if (ret < 0)
{
printf("Error! NANDBackUpFile: Failed! (Error: %d)\n", ret);
return false;
}
u32 checkSize = 0;
ret = NANDGetFileSize(dstPath, &checkSize);
return (checkSize == size);
}
static bool RestorePriiloader()
{
char dstPath[ISFS_MAXPATH] ATTRIBUTE_ALIGN(0x20);
if (!GetSysMenuExecPath(dstPath, false));
u32 size = 0;
s32 ret = NANDBackUpFile("/tmp/priiload.app", dstPath, &size);
if (ret < 0)
{
printf("Error! NANDBackUpFile: Failed! (Error: %d)\n", ret);
return false;
}
u32 checkSize = 0;
ret = NANDGetFileSize(dstPath, &checkSize);
return (checkSize == size && checkSize == gPriiloaderSize);
}
static void PrintCleanupResult(s32 result)
{
if (result < 0)
{
switch (result)
{
case -102:
{
printf(" Access denied.\n");
} break;
case -106:
{
printf(" Not found.\n");
} break;
default:
{
printf(" Error: %d\n", result);
} break;
}
}
else
{
printf(" OK!\n");
}
sleep(1);
}
static void CleanupPriiloaderLeftOvers(bool retain)
{
if (!retain)
{
printf("\n\t\tCleanup Priiloader leftover files...\n");
printf("\r\t\t>> Password file...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/password.txt"));
printf("\r\t\t>> Settings file...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/loader.ini"));
printf("\r\t\t>> Ticket...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/ticket"));
printf("\r\t\t>> File: main.nfo...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/main.nfo"));
printf("\r\t\t>> File: main.bin...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/main.bin"));
}
printf("\n\t\tRemoving Priiloader hacks...\n");
printf("\r\t\t>> File: hacks_s.ini...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/hacks_s.ini"));
printf("\r\t\t>> File: hacks.ini...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/hacks.ini"));
printf("\r\t\t>> File: hacksh_s.ini...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/hacksh_s.ini"));
printf("\r\t\t>> File: hackshas.ini...");
PrintCleanupResult(NANDDeleteFile("/title/00000001/00000002/data/hackshas.ini"));
if (retain)
{
printf("\n\t\tPriiloader hacks will be reset!\n");
printf("\t\tRemember to set them again.\n");
}
}
static bool CompareHashes(bool priiloader)
{
ATTRIBUTE_ALIGN(0x20)
char srcPath[ISFS_MAXPATH], dstPath[ISFS_MAXPATH] = "/tmp/priiload.app";
if (!GetSysMenuExecPath(srcPath, false))
return false;
if (!priiloader)
GetSysMenuExecPath(dstPath, true);
u32 sizeA = 0;
u32 sizeB = 0;
u8* dataA = NANDLoadFile(srcPath, &sizeA);
if (!dataA)
return false;
u8* dataB = NANDLoadFile(dstPath, &sizeB);
if (!dataB)
{
free(dataA);
return false;
}
bool ret = (sizeA == sizeB) && !CompareHash(dataA, sizeA, dataB, sizeB);
free(dataA);
free(dataB);
return ret;
}
/* 'WAD Header' structure */
typedef struct {
u32 header_len;
u16 type;
u16 version;
u32 certs_len;
u32 crl_len;
u32 tik_len;
u32 tmd_len;
u32 data_len;
u32 footer_len;
} wadHeader;
/* Variables */
ATTRIBUTE_ALIGN(0x40)
static u8 wadBuffer[16 << 10];
bool __Wad_FixTicket(signed_blob *s_tik)
{
tik* p_tik = SIGNATURE_PAYLOAD(s_tik);
u8 *ckey = ((u8*)s_tik) + 0x1F1;
bool fixvWiiKey = *ckey == 2;
if (*ckey > 1) {
/* Set common key */
*ckey = 0;
/* Fix tickets using vWii Common Key */
if (fixvWiiKey)
{
__aligned(0x20)
aeskey tkeybuf;
u64 iv[2];
memcpy(tkeybuf, p_tik->cipher_title_key, sizeof(aeskey));
iv[0] = p_tik->titleid;
iv[1] = 0;
AES_Decrypt(vWiiCommonKey, 0x10, iv, 0x10, tkeybuf, tkeybuf, sizeof(tkeybuf));
iv[0] = p_tik->titleid;
iv[1] = 0;
ES_Encrypt(ES_KEY_COMMON, (u32 *)iv, tkeybuf, sizeof(tkeybuf), tkeybuf);
memcpy(p_tik->cipher_title_key, tkeybuf, sizeof(tkeybuf));
}
/* Fakesign ticket */
Title_FakesignTik(s_tik);
}
return fixvWiiKey;
}
bool __Wad_VerifyHeader(wadHeader* header)
{
return (header->header_len == 0x20 && header->type == 0x4973 && header->version == 0x0000);
}
const char* wad_strerror(int ec)
{
switch (ec)
{
case 0: return "Success";
case -106: return "Not found";
case -996: return "Read error";
case -998: return "Skipped";
case -999: return "BRICK BLOCKED";
case -1010: return "Wii System memory full!";
case -1017: return "Invalid argument";
case -1020: return "Device ID mismatch";
case -1022: return "Content hash mismatch";
case -1027: return "Issuer not found in the certificate chain";
case -1028: return "Ticket not found";
case -1029: return "Invalid Ticket";
case -1035: return "Newer version already installed";
case -1036: return "Needed IOS missing!";
case -2011: return "No signature check patch?";
/*
* from libogc.
* This rarely happens unless the WAD had an invalid ticket/tmd size
* (certs were not stripped after downloading from NUS maybe?)
*/
case ES_EINVAL: return "Invalid WAD?"; break;
default: return "unknown error";
}
}
// Convenience purposes
u32 hbcList[] = { 0x48415858, 0x4A4F4449, 0xAF1BF516, 0x4C554C5A, 0x4F484243 };
u32 eulaList[] = { 0x48414B45, 0x48414B50, 0x48414B4A, 0x48414B4B };
u32 rgnselList[] = { 0x48414C45, 0x48414C50, 0x48414C4A, 0x48414C4B };
// Some of the safety checks can block region changing
// Entering the Konami code turns this true, so it will
// skip the problematic checks for region changing.
bool skipRegionSafetyCheck = false;
s32 Wad_Install(FILE *fp)
{
s32 ret;
wadHeader *header = {};
signed_blob *p_certs = NULL, *p_crl = NULL, *p_tik = NULL, *p_tmd = NULL;
tmd *tmd_data = NULL;
u64 tid;
tmd_view *view = NULL;
u32 view_size = 0;
u32 boot2version;
u32 cnt, offset = 0;
u32 *installedContents = NULL;
u32 installedContentsCount = 0;
bool retainPriiloader = false;
bool cleanupPriiloader = false;
SetPRButtons(false);
printf("\t\t>> Reading WAD data...\n\n");
ret = FSOPReadOpenFileA(fp, (void*)&header, offset, sizeof(wadHeader));
if (ret < 0)
goto err;
if (!__Wad_VerifyHeader(header))
{
ret = ES_EINVAL;
goto err;
}
offset += round_up(header->header_len, 64);
/*
* IOS won't let you do this unless you actually call ES_ImportBoot
//Don't try to install boot2
__Wad_GetTitleID(fp, header, &tid);
if (tid == TITLE_ID(1, 1))
{
printf("\n I can't let you do that Dave\n");
ret = -999;
goto out;
}
*/
/* WAD certificates */
ret = FSOPReadOpenFileA(fp, (void*)&p_certs, offset, header->certs_len);
if (ret < 0)
goto err;
offset += round_up(header->certs_len, 64);
/* WAD crl */
if (header->crl_len) {
ret = FSOPReadOpenFileA(fp, (void*)&p_crl, offset, header->crl_len);
if (ret < 0)
goto err;
offset += round_up(header->crl_len, 64);
}
/* WAD ticket */
ret = FSOPReadOpenFileA(fp, (void*)&p_tik, offset, header->tik_len);
if (ret < 0)
goto err;
offset += round_up(header->tik_len, 64);
/* WAD TMD */
ret = FSOPReadOpenFileA(fp, (void*)&p_tmd, offset, header->tmd_len);
if (ret < 0)
goto err;
offset += round_up(header->tmd_len, 64);
bool isvWiiTitle = __Wad_FixTicket(p_tik);
/* Get TMD info */
tmd_data = (tmd *)SIGNATURE_PAYLOAD(p_tmd);
tid = tmd_data->title_id;
if (TITLE_UPPER(tmd_data->sys_version) == 0) // IOS
{
bool stubbed = (tmd_data->title_version & 0xFF) == 0 || tmd_data->title_version == 404;
bool isCrossPlatformIOS = (isvWiiTitle || tmd_data->vwii_title) ^ IS_WIIU; // xor is one of my favourite binary operators of all time
if (isCrossPlatformIOS && !skipRegionSafetyCheck)
{
printf(" Cannot install vWii IOS on Wii\n");
printf(" or Wii IOS on vWii.\n");
printf(" I mean maybe you can. I'm not going to find out though\n");
ret = -999;
goto err;
}
ret = Title_GetTMDView(0x100000002LL, &view, &view_size);
if (ret < 0)
{
printf(" Hey, where's Perry?\n\n");
printf(" (No really, where is your Wii System Menu?)\n");
goto err;
}
if (view->sys_version == tmd_data->title_id)
{
if (isCrossPlatformIOS)
{
printf(" Anything but the System menu's IOS...!!!!\n");
ret = -999;
goto err;
}
if (stubbed) {
printf(" I won't install a stub System Menu IOS\n");
ret = -999;
goto err;
}
if (!IS_WIIU && (view->title_version >> 5) >= 15) // 4.2 or later. Again.
{
tik* ticket = (tik*)SIGNATURE_PAYLOAD(p_tik);
__aligned(0x20)
aeskey titlekey;
u64 iv[2] = { tid };
memcpy(titlekey, ticket->cipher_title_key, sizeof(aeskey));
ES_Decrypt(ES_KEY_COMMON, (u32 *)iv, titlekey, sizeof(aeskey), titlekey);
u32 content0_offset = offset;
for (int i = 0; i < tmd_data->num_contents; i++)
{
if (tmd_data->contents[i].index == 0) break;
content0_offset += round_up(tmd_data->contents[i].size, 0x40);
}
__aligned(0x20)
cIOSInfo build_tag = {};
ret = FSOPReadOpenFile(fp, (void*)&build_tag, content0_offset, sizeof(build_tag));
if (ret < 0)
goto err;
iv[0] = 0;
iv[1] = 0;
AES_Decrypt(titlekey, sizeof(aeskey), iv, sizeof(iv), &build_tag, &build_tag, sizeof(build_tag));
if ((build_tag.hdr_magic != CIOS_INFO_MAGIC ||
build_tag.hdr_version != CIOS_INFO_VERSION ||
build_tag.ios_base != 60) && ES_CheckHasKoreanKey())
{
printf(" Installing this System menu IOS will brick your Wii.\n\n");
printf(" Please remove the Korean key via KoreanKii,\n");
printf(" then try again.\n");
ret = -999;
goto err;
}
}
}
if (stubbed)
{
u64 sysvers = 0;
for (int i = 0; i < 4; i++)
{
if ((Title_GetSysVersion(TITLE_ID(0x00010008, eulaList[i]), &sysvers) == 0) && tid == sysvers)
{
printf(" I won't install a stub EULA IOS\n");
ret = -999;
goto err;
}
if ((Title_GetSysVersion(TITLE_ID(0x00010008, rgnselList[i]), &sysvers) == 0) && tid == sysvers)
{
printf(" I won't install a stub Region select IOS\n");
ret = -999;
goto err;
}
}
for (int i = 0; i < 5; i++)
{
if ((Title_GetSysVersion(TITLE_ID(0x00010001, hbcList[i]), &sysvers) == 0) && tid == sysvers)
{
printf(" I won't stub The Homebrew Channel's IOS\n");
ret = -999;
goto err;
}
}
}
}
else // not IOS
{
ret = Title_GetTMDView(tmd_data->sys_version, &view, &view_size);
if (ret < 0)
{
printf(" This title wants IOS%i, but the installed ver-...\n", TITLE_LOWER(tmd_data->sys_version));
printf(" ...is it even installed?\n");
goto err;
}
if ((view->title_version & 0xFF) == 0 || view->title_version == 404)
{
printf(" This title wants IOS%i, but the installed version\n", TITLE_LOWER(tmd_data->sys_version));
printf(" is a stub.\n");
ret = -1036;
goto err;
}
if (tid == 0x100000002LL)
{
if (skipRegionSafetyCheck || gForcedInstall)
goto skipChecks;
char region = 0;
u16 version = 0;
GetSysMenuRegion(&version, &region);
if (isvWiiTitle || tmd_data->vwii_title) // && !IS_WIIU ? :thinking:
{
printf(" I won't install a vWii SM by default.\n\n");
printf(" If you're really sure about what you're doing, next time\n");
printf(" select your device using Konami...\n");
ret = -999;
goto err;
}
if (region == '\0')
{
printf(" Unknown System menu region. What's up with setting.txt?\n");
ret = -999;
goto err;
}
if (!VersionIsOriginal(tmd_data->title_version))
{
printf(" I won't install an unknown SM versions by default.\n\n");
printf(" Are you installing a tweaked system menu?\n");
printf(" If you're really sure what you're doing, next time\n");
printf(" select your device using Konami...\n");
ret = -999;
goto err;
}
if(region != RegionLookupTable[(tmd_data->title_version & 0x0F)])
{
printf("\n I won't install the wrong regions SM by default.\n\n");
printf("\n Are you region changing?\n");
printf("\n If you're really sure what you're doing, next time\n");
printf(" select your device using Konami...\n");
ret = -999;
goto err;
}
if ((tmd_data->title_version & 0x1F) != 0x6 // 0x6 = KR
&& (tmd_data->title_version >> 5) >= 15) // 4.2 or later
{
cIOSInfo ios_info;
if ((Title_GetcIOSInfo(TITLE_LOWER(tmd_data->sys_version), &ios_info) != 0 ||
ios_info.ios_base != 60) && ES_CheckHasKoreanKey())
{
printf(" Installing this System menu will brick your Wii.\n");
printf(" Please remove the Korean key via KoreanKii,\n");
printf(" then try again.\n");
ret = -999;
goto err;
}
}
skipChecks:
if (tmd_data->title_version < 416)
{
ES_GetBoot2Version(&boot2version);
if(boot2version == 4)
{
printf(" This version of the System Menu\n");
printf(" is not compatible with your Wii\n");
ret = -999;
goto err;
}
}
if (!gForcedInstall && IsPriiloaderInstalled())
{
cleanupPriiloader = true;
printf(" Priiloader is installed next to the system menu.\n\n");
printf(" It is recommended to retain Priiloader as it can\n");
printf(" protect your console from being bricked.\n\n");
printf(" Press A to retain Priiloader or B to remove.");
u32 buttons = WaitButtons();
if ((buttons & WPAD_BUTTON_A))
{
retainPriiloader = (BackUpPriiloader() && CompareHashes(true));
if (retainPriiloader)
{
SetPriiloaderOption(true);
Con_ClearLine();
printf("\r[+] Priiloader will be retained.\n");
fflush(stdout);
}
else
{
Con_ClearLine();
printf("\r Couldn't backup Priiloader.\n");
fflush(stdout);
printf("\n Press A to continue or B to skip");
u32 buttons = WaitButtons();
if (!(buttons & WPAD_BUTTON_A))
{
ret = -990;
goto err;
}
}
}
if (!retainPriiloader)
{
SetPriiloaderOption(false);
Con_ClearLine();
printf("\r[+] Priiloader will be removed.\n");
fflush(stdout);
}
}
else
{
SetPriiloaderOption(false);
}
}
if (gForcedInstall)
{
gForcedInstall = false;
cleanupPriiloader = true;
}
}
#if 0
/* Check available space */
if (TITLE_UPPER(tmd_data->title_id) != 0x1 && TITLE_UPPER(tmd_data->title_id) != 0x00010002)
{
u32 totalContentSize = 0;
for (int i = 0; i < tmd_data->num_contents; i++)
{
tmd_content* content = &tmd_data->contents[i];
if (Title_SharedContentPresent(content, sharedContents, sharedContentsCount))
continue;
totalContentSize += round_up(content->size, 16384);
}
u32 free, user_free, blocks_free, user_blocks_free;
Title_GetFreeSpace(totalContentSize, &free, &user_free);
blocks_free = free / 131072, user_blocks_free = user_free / 131072;
if (totalContentSize > free)
}
#endif
printf("\t\t>> Installing ticket...\n");
/* Install ticket */
ret = ES_AddTicket(p_tik, header->tik_len, p_certs, header->certs_len, p_crl, header->crl_len);
if (ret < 0)
goto err;
printf("\t\t>> Installing title...\n");
/* Install title */
ret = ES_AddTitleStart(p_tmd, header->tmd_len, p_certs, header->certs_len, p_crl, header->crl_len);
if (ret < 0)
goto err;
/* Get list of currently installed contents */
Title_GetTMDContents(p_tmd, &installedContents, &installedContentsCount);
/* Install contents */
for (cnt = 0; cnt < tmd_data->num_contents; cnt++)
{
tmd_content *content = &tmd_data->contents[cnt];
u32 idx = 0, len;
s32 cfd;
/* Encrypted content size */
len = round_up(content->size, 64);
if (content->type & 0x8000) // We're only doing shared. Unless we're going to genuinely read the non-shared contents and check the hash against what we're installing
for (int i = 0; i < installedContentsCount; i++)
if (installedContents[i] == content->cid)
{
offset += len;
continue;
}
Con_ClearLine();
printf("\r\t\t>> Installing content #%02d...", content->index);
/* Install content */
ret = cfd = ES_AddContentStart(tmd_data->title_id, content->cid);
if (ret < 0)
goto err;
/* Install content data */
while (idx < len)
{
u32 size;
/* Data length */
size = (len - idx);
if (size > sizeof(wadBuffer))
size = sizeof(wadBuffer);
/* Read data */
ret = FSOPReadOpenFile(fp, wadBuffer, offset, size);
if (ret < 0)
{
ES_AddContentFinish(cfd);
goto err;
}
/* Install data */
ret = ES_AddContentData(cfd, wadBuffer, size);
if (ret < 0)
{
ret = ES_AddContentFinish(cfd);
goto err;
}
/* Increase variables */
idx += size;
offset += size;
}
/* Finish content installation */
ret = ES_AddContentFinish(cfd);
if (ret < 0)
goto err;
}
Con_ClearLine();
printf("\r\t\t>> Finishing installation...");
fflush(stdout);
/* Finish title install */
ret = ES_AddTitleFinish();
if (ret >= 0)
{
printf(" OK!\n");
if (retainPriiloader)
{
printf("\r\t\t>> Moving System Menu...");
if (MoveMenu(false))
{
printf(" OK!\n");
printf("\r\t\t>> Check System Menu executable hashes...");
s32 restoreMenu = 0;
if (CompareHashes(false))
{
printf(" OK!\n");
}
else
{
printf(" Failed!\n");
restoreMenu = 1;
}
printf("\r\t\t>> Restore Priiloader...");
if (!restoreMenu && RestorePriiloader())
{
printf(" OK!\n");
printf("\r\t\t>> Check Priiloader executable hashes...");
if (CompareHashes(true))
{
printf(" OK!\n");
}
else
{
printf(" Failed!\n");
restoreMenu = 2;
}
}
else
{
printf(" Failed!\n");
restoreMenu = 2;
}
if (restoreMenu)
{
printf("\r\t\t>> Restore System Menu...");
bool restored = true;
switch (restoreMenu)
{
case 2:
{
restored = (MoveMenu(true) && CompareHashes(false));
}
case 1:
{
if (restored)
{
char path[ISFS_MAXPATH] ATTRIBUTE_ALIGN(0x20);
GetSysMenuExecPath(path, true);
NANDDeleteFile(path);
}
}
}
if (restored)
{
printf(" OK!\n");
}
else
{
printf(" Failed!\n");
printf("\n\t\t>> Reinstalling System Menu...\n\n");
sleep(3);
printf("\t\t>> Priiloader will be removed!\n\n");
gForcedInstall = true;
cleanupPriiloader = false;
}
}
}
else
{
printf(" Failed!\n");
printf("\n\t\t>> Priiloader will be removed!\n\n");
}
}
if (cleanupPriiloader)
{
CleanupPriiloaderLeftOvers(retainPriiloader);
}
goto out;
}
err:
printf("\n ERROR! (ret = %d)\n", ret);
if (retainPriiloader)
SetPriiloaderOption(false);
if (ret == 0)
ret = -996;
/* Cancel install */
ES_AddTitleCancel();
out:
/* Free memory */
free(header);
free(p_certs);
free(p_crl);
free(p_tik);
free(p_tmd);
free(installedContents);
free(view);
if (gForcedInstall)
return Wad_Install(fp);
SetPRButtons(true);
return ret;
}
s32 Wad_Uninstall(FILE *fp)
{
wadHeader *header = {};
tikview *tikviews = NULL;
signed_blob *s_tik = NULL;
signed_blob *s_tmd = NULL;
u64 tid, sysvers;
u32 viewCnt;
int ret;
SetPRButtons(false);
printf("\t\t>> Reading WAD data...\n");
/* WAD header */
ret = FSOPReadOpenFileA(fp, (void*)&header, 0, sizeof(wadHeader));
if (ret < 0)
{
printf(" ERROR! (ret = %d)\n", ret);
goto out;
}
if (!__Wad_VerifyHeader(header))
{
puts(" Invalid WAD file?");
ret = ES_EINVAL;
goto out;
}
/* Get ticket */
u32 offset = 0;
offset += round_up(header->header_len, 0x40);
offset += round_up(header->certs_len, 0x40);
offset += round_up(header->crl_len, 0x40);
ret = FSOPReadOpenFileA(fp, (void*)&s_tik, offset, header->tik_len);
if (ret < 0) {
printf(" ERROR! (ret = %d)\n", ret);
goto out;
}
bool isvWiiTitle = __Wad_FixTicket(s_tik);
tik *ticket = SIGNATURE_PAYLOAD(s_tik);
tid = ticket->titleid;
offset += round_up(header->tik_len, 0x40);
ret = FSOPReadOpenFileA(fp, (void*)&s_tmd, offset, header->tmd_len);
if (ret < 0) {
printf(" ERROR! (ret = %d)\n", ret);
goto out;
}
tmd* tmd_data = SIGNATURE_PAYLOAD(s_tmd);
//Assorted Checks
if (TITLE_UPPER(tid) == 0x1)
{
if ((ret = Title_GetSysVersion(0x100000002LL, &sysvers)) != 0)
{
printf(" I can't determine the System Menus IOS\n");
printf(" Deleting system titles is disabled\n");
goto out;
}
if (((isvWiiTitle || tmd_data->vwii_title) ^ IS_WIIU) && !skipRegionSafetyCheck) // Only this way around this time // Ehh nvm
{
printf(
" Attempting to uninstall a Wii IOS WAD on vWii.\n"
" or a vWii IOS WAD on Wii.\n\n"
" Maybe you installed one on accident before?\n\n"
" If you're sure about what you're doing, input\n"
" the Konami code on the device screen. Have fun.\n");
ret = -999;
goto out;
}
if (tid == 0x100000001LL)
{
printf(" I won't try to uninstall boot2\n");
ret = -999;
goto out;
}
if (tid == 0x100000002LL)
{
printf(" I won't uninstall the System Menu\n");
ret = -999;
goto out;
}
if (tid == sysvers)
{
printf(" I won't uninstall the System Menu's IOS\n");
ret = -999;
goto out;
}
for (int i = 0; i < 5; i++)
{
if ((Title_GetSysVersion(TITLE_ID(0x00010001, hbcList[5 - i]), &sysvers) == 0) && tid == sysvers)
{
printf(" I won't uninstall the Homebrew Channel's IOS!\n");
ret = -999;
goto out;
}
}
}
char region = 0;
GetSysMenuRegion(NULL, &region);
if (((tid | 0xFF) == 0x0001000848414CFFLL || (tid | 0xFF) == 0x0001000848414BFFLL) && strchr("JEPK", (char)tid) && region == 0)
{
printf(" Unknown System Menu region\n Please check the site for updates\n");
ret = -999;
goto out;
}
if (tid == TITLE_ID(0x10008, 0x48414B00 | region))
{
printf(" I won't uninstall the EULA\n");
ret = -999;
goto out;
}
if (tid == TITLE_ID(0x10008, 0x48414C00 | region))
{
printf(" I won't uninstall rgnsel\n");
ret = -999;
goto out;
}
if (Title_GetSysVersion(0x0001000848414B00LL | region, &sysvers) == 0 && tid == sysvers)
{
printf(" I won't uninstall the EULAs IOS\n");
ret = -999;
goto out;
}
if (Title_GetSysVersion(0x0001000848414C00LL | region, &sysvers) == 0 && tid == sysvers)
{
printf(" I won't uninstall rgnsel's IOS\n");
ret = -999;
goto out;
}
printf("\t\t>> Deleting title contents... ");
/* Delete title contents */
ret = ES_DeleteTitleContent(tid);
if (ret < 0)
printf(" ERROR! (ret = %d)\n", ret);
else
printf(" OK!\n");
printf("\t\t>> Deleting title...");
/* Delete title */
ret = ES_DeleteTitle(tid);
if (ret < 0)
printf(" ERROR! (ret = %d)\n", ret);
else
printf(" OK!\n");
printf("\t\t>> Deleting tickets...");
/* Get ticket views */
ret = Title_GetTicketViews(tid, &tikviews, &viewCnt);
if (ret < 0)
printf(" ERROR! (ret = %d)\n", ret);
/* Delete tickets */
else
{
u32 cnt;
/* Delete all tickets */
for (cnt = 0; cnt < viewCnt; cnt++) {
#if 0
if (view.devicetype)
{
u32 deviceID = 0;
/* If we failed to get the ID or it actually matches, skip this */
if (ES_GetDeviceID(&deviceID) || view.devicetype == deviceID) continue;
}
#endif
ret = ES_DeleteTicket(&tikviews[cnt]);
if (ret < 0)
break;
}
if (ret < 0)
printf(" ERROR! (ret = %d)\n", ret);
else
printf(" OK!\n");
}
free(tikviews);
out:
/* Free memory */
free(header);
SetPRButtons(true);
return ret;
}
Reference in New Issue View Git Blame Copy Permalink