Wii-U-Account-Swap/source/switch.cpp

#include <iostream>

#include <string.h>

#include <coreinit/launch.h>
#include <coreinit/screen.h>
#include <coreinit/time.h>
#include <coreinit/thread.h>
#include <mocha/mocha.h>
#include <sysapp/launch.h>
#include <whb/log_console.h>
#include <whb/log.h>
#include <whb/proc.h>

#include "../include/global.h"


void switchAccount(const char* backupFile, const char* accountType) {
    WHBLogPrintf("Switch: You will be swapped to a %s.", accountType);
    WHBLogPrint("---------------------------------------------------------");
    WHBLogConsoleDraw();

    WHBLogPrintf("Switching account.dat to %s.", accountType);
    WHBLogConsoleDraw();
    // Open the account.dat file and switch it to the specified account.
    FILE *backup = fopen(backupFile, "rb");
    if (backup == NULL) {
        WHBLogConsoleSetColor(0x99000000);
        WHBLogPrintf("Error opening %s account backup!", accountType);
        WHBLogConsoleDraw();
        WHBLogPrint("Have you made a backup of this account yet?");
        // Wait 5 seconds, then go back to the menu.
        OSSleepTicks(OSMillisecondsToTicks(5000));
    }
    else {
        WHBLogPrintf("%s account backup opened.", accountType);
        WHBLogConsoleDraw();
        char *buffer = (char *)malloc(BUFFER_SIZE);
        if (buffer == NULL) {
            WHBLogConsoleSetColor(0x99000000);
            WHBLogPrint("Error allocating memory!");
            WHBLogConsoleDraw();
        }
        else {
            WHBLogPrint("Memory was allocated successfully.");
            WHBLogConsoleDraw();
            
            FILE *account = fopen(ACCOUNT_FILE.c_str(), "wb");
            if (account == NULL) {
                WHBLogConsoleSetColor(0x99000000);
                WHBLogPrint("Error opening system account.dat file!");
                WHBLogConsoleDraw();
            }
            else {
                WHBLogPrint("System account.dat file opened.");
                WHBLogConsoleDraw();
                size_t bytesRead = 0;
                while ((bytesRead = fread(buffer, 1, BUFFER_SIZE, backup)) > 0) {
                    fwrite(buffer, 1, bytesRead, account);
                }
                fclose(account);
                WHBLogPrint("System account.dat file restored.");
                // We'll attempt to automatically swap the network using Inkay's configuration.
                FILE *inkay = fopen(INKAY_CONFIG, "wb");
                if (inkay == NULL) {
                    // If we can't open the file, we will move on.
                    WHBLogPrint("Error opening Inkay config file!");
                    WHBLogPrint("Network will not be automatically swapped.");
                    WHBLogConsoleDraw();
                }
                else {
                    WHBLogPrint("Inkay config file opened.");
                    WHBLogConsoleDraw();
                    WHBLogPrintf("Swapping network to %s.", accountType);
                    const char *inkayContent = "{\"storageitems\":{\"connect_to_network\":%d}}";
                    fprintf(inkay, inkayContent, strcmp(accountType, "Pretendo Network ID") == 0 ? 1 : 0);
                    fclose(inkay);
                    WHBLogPrint("Inkay config file edited.");
                    WHBLogConsoleDraw();
                }
                WHBLogConsoleSetColor(0x00990000);
                WHBLogPrint("---------------------------------------------------------");
                WHBLogPrint("The account.dat was restored successfully!");
                WHBLogPrint("Your console will restart in 5 seconds...");
                WHBLogConsoleDraw();
            }
            free(buffer);
        }
        fclose(backup);
        OSSleepTicks(OSMillisecondsToTicks(5000));
        OSForceFullRelaunch();
        SYSLaunchMenu();
        deinitialize();
    }
    printMainMenu();
}