117 lines
4.0 KiB
C++
117 lines
4.0 KiB
C++
#include "main.h"
|
|
#include "Hints.h"
|
|
#include "UpdaterCheck.h"
|
|
#include "utils/DownloadUtils.h"
|
|
#include "utils/LatestVersion.h"
|
|
#include "utils/config.h"
|
|
#include <coreinit/title.h>
|
|
#include <malloc.h>
|
|
#include <nn/spm.h>
|
|
#include <notifications/notifications.h>
|
|
#include <rpxloader/rpxloader.h>
|
|
#include <utils/logger.h>
|
|
#include <wups.h>
|
|
|
|
WUPS_PLUGIN_NAME("AromaBasePlugin");
|
|
WUPS_PLUGIN_DESCRIPTION("Implements small patches and checks for Aroma updates.");
|
|
WUPS_PLUGIN_VERSION(PLUGIN_VERSION_FULL);
|
|
WUPS_PLUGIN_AUTHOR("Maschell");
|
|
WUPS_PLUGIN_LICENSE("GPL");
|
|
|
|
WUPS_USE_WUT_DEVOPTAB();
|
|
WUPS_USE_STORAGE("aroma_base_plugin"); // Unique id for the storage api
|
|
|
|
INITIALIZE_PLUGIN() {
|
|
initLogging();
|
|
if (NotificationModule_InitLibrary() != NOTIFICATION_MODULE_RESULT_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("NotificationModule_InitLibrary failed");
|
|
}
|
|
if (RPXLoader_InitLibrary() != RPX_LOADER_RESULT_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("RPXLoader_InitLibrary failed");
|
|
}
|
|
|
|
// Open storage to read values
|
|
WUPSStorageError storageRes = WUPS_OpenStorage();
|
|
if (storageRes == WUPS_STORAGE_ERROR_SUCCESS) {
|
|
wups_storage_item_t *cat_config = nullptr;
|
|
if (WUPS_GetSubItem(nullptr, CAT_CONFIG, &cat_config) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
if (WUPS_CreateSubItem(nullptr, CAT_CONFIG, &cat_config) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
cat_config = nullptr;
|
|
}
|
|
}
|
|
|
|
if (cat_config != nullptr) {
|
|
LOAD_BOOL_FROM_STORAGE(cat_config, USTEALTH_CONFIG_ID, gActivateUStealth);
|
|
LOAD_BOOL_FROM_STORAGE(cat_config, POWEROFFWARNING_CONFIG_ID, gSkip4SecondOffStatusCheck);
|
|
}
|
|
|
|
wups_storage_item_t *cat_other = nullptr;
|
|
if (WUPS_GetSubItem(nullptr, CAT_OTHER, &cat_other) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
if (WUPS_CreateSubItem(nullptr, CAT_OTHER, &cat_other) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
cat_other = nullptr;
|
|
}
|
|
}
|
|
|
|
if (cat_other != nullptr) {
|
|
LOAD_BOOL_FROM_STORAGE(cat_other, CONFIG_MENU_HINT_SHOWN_ID, gConfigMenuHintShown);
|
|
char hash[41];
|
|
memset(hash, 0, sizeof(hash));
|
|
LOAD_STRING_FROM_STORAGE(cat_other, LAST_UPDATE_HASH_ID, hash, sizeof(hash));
|
|
gLastHash = hash;
|
|
}
|
|
|
|
// Close storage
|
|
if (WUPS_CloseStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to close storage");
|
|
}
|
|
} else {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to open storage %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
}
|
|
|
|
ON_APPLICATION_START() {
|
|
initLogging();
|
|
uint64_t titleID = OSGetTitleID();
|
|
if (titleID == 0x0005001010040000L || // Wii U Menu
|
|
titleID == 0x0005001010040100L ||
|
|
titleID == 0x0005001010040200L) {
|
|
|
|
StartHintThread();
|
|
StartUpdaterCheckThread();
|
|
}
|
|
}
|
|
|
|
ON_APPLICATION_ENDS() {
|
|
StopHintThread();
|
|
StopUpdaterCheckThread();
|
|
deinitLogging();
|
|
}
|
|
|
|
DEINITIALIZE_PLUGIN() {
|
|
NotificationModule_DeInitLibrary();
|
|
RPXLoader_DeInitLibrary();
|
|
}
|
|
|
|
DECL_FUNCTION(int32_t, IsStorageMaybePcFormatted, bool *isPcFormatted, nn::spm::StorageIndex *index) {
|
|
// Make sure the index is valid
|
|
int32_t res = real_IsStorageMaybePcFormatted(isPcFormatted, index);
|
|
if (gActivateUStealth && res == 0) {
|
|
// always return false which makes the Wii U menu stop nagging about this drive
|
|
*isPcFormatted = false;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
DECL_FUNCTION(bool, MCP_Get4SecondOffStatus, int32_t handle) {
|
|
if (gSkip4SecondOffStatusCheck) {
|
|
return false;
|
|
}
|
|
|
|
return real_MCP_Get4SecondOffStatus(handle);
|
|
}
|
|
|
|
// Only replace for the Wii U Menu
|
|
WUPS_MUST_REPLACE_FOR_PROCESS(IsStorageMaybePcFormatted, WUPS_LOADER_LIBRARY_NN_SPM, IsStorageMaybePcFormatted__Q2_2nn3spmFPbQ3_2nn3spm12StorageIndex, WUPS_FP_TARGET_PROCESS_WII_U_MENU);
|
|
WUPS_MUST_REPLACE_FOR_PROCESS(MCP_Get4SecondOffStatus, WUPS_LOADER_LIBRARY_COREINIT, MCP_Get4SecondOffStatus, WUPS_FP_TARGET_PROCESS_WII_U_MENU);
|