mirror of
https://github.com/wiiu-env/ftpiiu_plugin.git
synced 2024-06-16 17:28:48 +02:00
231 lines
8.7 KiB
C++
231 lines
8.7 KiB
C++
#include "main.h"
|
|
#include "BackgroundThread.hpp"
|
|
#include "utils/logger.h"
|
|
#include "virtualpath.h"
|
|
#include <coreinit/cache.h>
|
|
#include <cstring>
|
|
#include <mocha/mocha.h>
|
|
#include <nn/ac.h>
|
|
#include <stdio.h>
|
|
#include <wups.h>
|
|
#include <wups/config/WUPSConfigItemBoolean.h>
|
|
#include <wups/config/WUPSConfigItemStub.h>
|
|
|
|
WUPS_PLUGIN_NAME("FTPiiU");
|
|
WUPS_PLUGIN_DESCRIPTION("FTP Server");
|
|
WUPS_PLUGIN_VERSION(VERSION_FULL);
|
|
WUPS_PLUGIN_AUTHOR("Maschell");
|
|
WUPS_PLUGIN_LICENSE("GPL");
|
|
|
|
WUPS_USE_WUT_DEVOPTAB();
|
|
WUPS_USE_STORAGE("ftpiiu"); // Unqiue id for the storage api
|
|
|
|
uint32_t hostIpAddress = 0;
|
|
|
|
BackgroundThread *thread = nullptr;
|
|
|
|
#define FTPIIU_ENABLED_STRING "enabled"
|
|
#define SYSTEM_FILES_ALLOWED_STRING "systemFilesAllowed"
|
|
|
|
bool gFTPServerEnabled = true;
|
|
bool gSystemFilesAllowed = false;
|
|
bool gMochaPathsWereMounted = false;
|
|
|
|
MochaUtilsStatus MountWrapper(const char *mount, const char *dev, const char *mountTo) {
|
|
auto res = Mocha_MountFS(mount, dev, mountTo);
|
|
if (res == MOCHA_RESULT_ALREADY_EXISTS) {
|
|
res = Mocha_MountFS(mount, nullptr, mountTo);
|
|
}
|
|
if (res == MOCHA_RESULT_SUCCESS) {
|
|
std::string mountPath = std::string(mount) + ":/";
|
|
VirtualMountDevice(mountPath.c_str());
|
|
DEBUG_FUNCTION_LINE_VERBOSE("Mounted %s", mountPath.c_str());
|
|
} else {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to mount %s: %s [%d]", mount, Mocha_GetStatusStr(res), res);
|
|
}
|
|
return res;
|
|
}
|
|
void startServer();
|
|
void stopServer();
|
|
|
|
/* Entry point */
|
|
ON_APPLICATION_START() {
|
|
nn::ac::Initialize();
|
|
nn::ac::ConnectAsync();
|
|
hostIpAddress = 0;
|
|
nn::ac::GetAssignedAddress(&hostIpAddress);
|
|
initLogging();
|
|
|
|
if (gFTPServerEnabled) {
|
|
startServer();
|
|
}
|
|
}
|
|
|
|
INITIALIZE_PLUGIN() {
|
|
// Open storage to read values
|
|
WUPSStorageError storageRes = WUPS_OpenStorage();
|
|
if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to open storage %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
} else {
|
|
if ((storageRes = WUPS_GetBool(nullptr, FTPIIU_ENABLED_STRING, &gFTPServerEnabled)) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
// Add the value to the storage if it's missing.
|
|
if (WUPS_StoreBool(nullptr, FTPIIU_ENABLED_STRING, gFTPServerEnabled) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store bool");
|
|
}
|
|
} else if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to get bool %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
if ((storageRes = WUPS_GetBool(nullptr, SYSTEM_FILES_ALLOWED_STRING, &gSystemFilesAllowed)) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
// Add the value to the storage if it's missing.
|
|
if (WUPS_StoreBool(nullptr, SYSTEM_FILES_ALLOWED_STRING, gSystemFilesAllowed) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store bool");
|
|
}
|
|
} else if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to get bool %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
|
|
// Close storage
|
|
if (WUPS_CloseStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to close storage");
|
|
}
|
|
}
|
|
thread = nullptr;
|
|
}
|
|
|
|
void startServer() {
|
|
if (!thread) {
|
|
//!*******************************************************************
|
|
//! Initialize FS *
|
|
//!*******************************************************************
|
|
if (gSystemFilesAllowed) {
|
|
MochaUtilsStatus res;
|
|
if ((res = Mocha_InitLibrary()) == MOCHA_RESULT_SUCCESS) {
|
|
MountWrapper("slccmpt01", "/dev/slccmpt01", "/vol/storage_slccmpt01");
|
|
MountWrapper("storage_odd_tickets", nullptr, "/vol/storage_odd01");
|
|
MountWrapper("storage_odd_updates", nullptr, "/vol/storage_odd02");
|
|
MountWrapper("storage_odd_content", nullptr, "/vol/storage_odd03");
|
|
MountWrapper("storage_odd_content2", nullptr, "/vol/storage_odd04");
|
|
MountWrapper("storage_slc", "/dev/slc01", "/vol/storage_slc01");
|
|
Mocha_MountFS("storage_mlc", nullptr, "/vol/storage_mlc01");
|
|
Mocha_MountFS("storage_usb", nullptr, "/vol/storage_usb01");
|
|
gMochaPathsWereMounted = true;
|
|
} else {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to init libmocha: %s [%d]", Mocha_GetStatusStr(res), res);
|
|
}
|
|
}
|
|
|
|
MountVirtualDevices();
|
|
|
|
thread = BackgroundThread::getInstance();
|
|
OSMemoryBarrier();
|
|
}
|
|
}
|
|
|
|
void stopServer() {
|
|
BackgroundThread::destroyInstance();
|
|
if (gMochaPathsWereMounted) {
|
|
Mocha_UnmountFS("slccmpt01");
|
|
Mocha_UnmountFS("storage_odd_tickets");
|
|
Mocha_UnmountFS("storage_odd_updates");
|
|
Mocha_UnmountFS("storage_odd_content");
|
|
Mocha_UnmountFS("storage_odd_content2");
|
|
Mocha_UnmountFS("storage_slc");
|
|
Mocha_UnmountFS("storage_mlc");
|
|
Mocha_UnmountFS("storage_usb");
|
|
gMochaPathsWereMounted = false;
|
|
}
|
|
|
|
DEBUG_FUNCTION_LINE("Unmount virtual paths");
|
|
UnmountVirtualPaths();
|
|
|
|
thread = nullptr;
|
|
}
|
|
|
|
void gFTPServerRunningChanged(ConfigItemBoolean *item, bool newValue) {
|
|
DEBUG_FUNCTION_LINE("New value in gFTPServerEnabled: %d", newValue);
|
|
gFTPServerEnabled = newValue;
|
|
if (!gFTPServerEnabled) {
|
|
stopServer();
|
|
} else {
|
|
startServer();
|
|
}
|
|
// If the value has changed, we store it in the storage.
|
|
auto res = WUPS_StoreInt(nullptr, FTPIIU_ENABLED_STRING, gFTPServerEnabled);
|
|
if (res != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store gFTPServerEnabled: %s (%d)", WUPS_GetStorageStatusStr(res), res);
|
|
}
|
|
}
|
|
|
|
void gSystemFilesAllowedChanged(ConfigItemBoolean *item, bool newValue) {
|
|
DEBUG_FUNCTION_LINE("New value in gFTPServerEnabled: %d", newValue);
|
|
if (thread != nullptr) { // If the server is already running we need to restart it.
|
|
stopServer();
|
|
gSystemFilesAllowed = newValue;
|
|
startServer();
|
|
} else {
|
|
gSystemFilesAllowed = newValue;
|
|
}
|
|
// If the value has changed, we store it in the storage.
|
|
auto res = WUPS_StoreInt(nullptr, SYSTEM_FILES_ALLOWED_STRING, gSystemFilesAllowed);
|
|
if (res != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store gSystemFilesAllowed: %s (%d)", WUPS_GetStorageStatusStr(res), res);
|
|
}
|
|
}
|
|
|
|
WUPS_GET_CONFIG() {
|
|
// We open the storage, so we can persist the configuration the user did.
|
|
if (WUPS_OpenStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to open storage");
|
|
return 0;
|
|
}
|
|
nn::ac::GetAssignedAddress(&hostIpAddress);
|
|
|
|
WUPSConfigHandle config;
|
|
WUPSConfig_CreateHandled(&config, "FTPiiU");
|
|
|
|
WUPSConfigCategoryHandle setting;
|
|
WUPSConfig_AddCategoryByNameHandled(config, "Settings", &setting);
|
|
|
|
WUPSConfigItemBoolean_AddToCategoryHandled(config, setting, FTPIIU_ENABLED_STRING, "Enable FTPiiU", gFTPServerEnabled, &gFTPServerRunningChanged);
|
|
WUPSConfigItemBoolean_AddToCategoryHandled(config, setting, SYSTEM_FILES_ALLOWED_STRING, "Allow access to system files", gSystemFilesAllowed, &gSystemFilesAllowedChanged);
|
|
|
|
WUPSConfigCategoryHandle info;
|
|
WUPSConfig_AddCategoryByNameHandled(config, "==========", &info);
|
|
WUPSConfigItemStub_AddToCategoryHandled(config, info, "info", "Press B to go Back");
|
|
|
|
WUPSConfigCategoryHandle info1;
|
|
char ipSettings[50];
|
|
if (hostIpAddress != 0) {
|
|
snprintf(ipSettings, 50, "IP of your console is %u.%u.%u.%u. Port %i",
|
|
(hostIpAddress >> 24) & 0xFF,
|
|
(hostIpAddress >> 16) & 0xFF,
|
|
(hostIpAddress >> 8) & 0xFF,
|
|
(hostIpAddress >> 0) & 0xFF,
|
|
PORT);
|
|
} else {
|
|
snprintf(ipSettings, 50, "The console is not connected to a network.");
|
|
}
|
|
WUPSConfig_AddCategoryByNameHandled(config, ipSettings, &info1);
|
|
WUPSConfigItemStub_AddToCategoryHandled(config, info1, "info1", "Press B to go Back");
|
|
|
|
WUPSConfigCategoryHandle info2;
|
|
char portSettings[50];
|
|
snprintf(portSettings, 50, "You can connect with empty credentials");
|
|
WUPSConfig_AddCategoryByNameHandled(config, portSettings, &info2);
|
|
WUPSConfigItemStub_AddToCategoryHandled(config, info2, "info2", "Press B to go Back");
|
|
|
|
return config;
|
|
}
|
|
|
|
WUPS_CONFIG_CLOSED() {
|
|
// Save all changes
|
|
if (WUPS_CloseStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to close storage");
|
|
}
|
|
}
|
|
|
|
ON_APPLICATION_ENDS() {
|
|
stopServer();
|
|
deinitLogging();
|
|
}
|