PayloadLoaderInstaller/source/InstallerService.h

#pragma once

#include "utils/pugixml.hpp"
#include "common/common.h"
#include <optional>
#include <coreinit/memorymap.h>

class InstallerService {
public:
    enum eResults {
        SUCCESS = 0,
        NO_COMPATIBLE_APP_INSTALLED = -1,
        FAILED_TO_COPY_FILES = -2,
        FAILED_TO_CHECK_HASH_COPIED_FILES = -3,
        SYSTEM_XML_INFORMATION_NOT_FOUND = -4,
        SYSTEM_XML_PARSING_FAILED = -5,
        SYSTEM_XML_HASH_MISMATCH_RESTORE_FAILED = -6,
        SYSTEM_XML_HASH_MISMATCH = -7,
        RPX_HASH_MISMATCH = -8,
        RPX_HASH_MISMATCH_RESTORE_FAILED = -9,
        COS_XML_PARSING_FAILED = -10,
        COS_XML_HASH_MISMATCH = -11,
        COS_XML_HASH_MISMATCH_RESTORE_FAILED = -12,
        MALLOC_FAILED = -13,
        FST_HASH_MISMATCH = -14,
        FST_HASH_MISMATCH_RESTORE_FAILED = -15,
        FST_HEADER_MISMATCH = -16,
        FST_NO_USABLE_SECTION_FOUND = -17,
        FAILED_TO_LOAD_FILE = -18,
    };

    static bool isColdBootAllowed(){
        if (OSIsAddressValid(0x00FFFFF8)) {
            uint64_t bootedFrom = *((uint64_t *) 0x00FFFFF8);
            if (
                    bootedFrom == 0x000500101004E000L || // H&S JPN
                    bootedFrom == 0x000500101004E100L || // H&S USA
                    bootedFrom == 0x000500101004E200L    // H&S EUR
                    ) {
                return true;
            }
        }
        return false;
    }

    static bool isBackupAvailable(const std::string &path);

    static eResults restoreAppFiles(const std::string &path);

    static eResults backupAppFiles(const std::string &path);

    static eResults patchCOS(const std::string &path, char *hash);

    static eResults checkCOS(const std::string &path, char *hash);

    static eResults checkSystemXML(const std::string &path, uint64_t titleId);

    static eResults checkFST(const std::string &path, const char *fstHash);

    static std::optional<appInformation> getInstalledAppInformation();

    static std::optional<uint64_t> getSystemMenuTitleId();

    static std::string ErrorMessage(eResults error);

    static std::string ErrorDescription(eResults error);

    static eResults patchFST(const std::string &path, const char *hash);

    static eResults copyRPX(const std::string &path, const uint8_t *rpx_data, size_t size, const std::string &rpx_hash);

    static eResults patchSystemXML(const std::string &path, uint64_t id);

    static uint64_t getColdbootTitleId(const std::string &path);

    static eResults checkFSTAlreadyValid(const std::string &path, const std::string &hash);

    static eResults checkTMDValid(const std::string &path, const std::string &hash, const std::string &tmdWithCertHash);

    static eResults checkCOSAlreadyValid(const std::string &path, const std::string &hash);

    static eResults checkRPXAlreadyValid(const std::string &path, const std::string &hash);

    static eResults setBootTitle(uint64_t titleId);

private:
    static eResults patchFSTData(uint8_t *fstData, uint32_t size);

    static bool patchCOSXMLData(pugi::xml_document *doc);

    static eResults checkFileHash(const std::string &filePath, const std::string &hash);
};