mirror of
https://github.com/DarkMatterCore/nxdumptool.git
synced 2024-10-30 23:41:45 +00:00
375 lines
12 KiB
C
375 lines
12 KiB
C
/*
|
|
* config.c
|
|
*
|
|
* Copyright (c) 2020-2021, DarkMatterCore <pabloacurielz@gmail.com>.
|
|
*
|
|
* This file is part of nxdumptool (https://github.com/DarkMatterCore/nxdumptool).
|
|
*
|
|
* nxdumptool is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* nxdumptool is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "nxdt_utils.h"
|
|
#include "config.h"
|
|
#include "title.h"
|
|
|
|
#define JSON_VALIDATE_FIELD(type, name, ...) \
|
|
if (!strcmp(key, #name)) { \
|
|
if (name##_found || !configValidateJson##type(val, ##__VA_ARGS__)) goto end; \
|
|
name##_found = true; \
|
|
continue; \
|
|
}
|
|
|
|
#define JSON_VALIDATE_OBJECT(type, name) \
|
|
if (!strcmp(key, #name)) { \
|
|
if (name##_found || !configValidateJson##type##Object(val)) goto end; \
|
|
name##_found = true; \
|
|
continue; \
|
|
}
|
|
|
|
#define JSON_GETTER(functype, vartype, jsontype, ...) \
|
|
vartype configGet##functype(const char *path) { \
|
|
vartype ret = (vartype)0; \
|
|
SCOPED_LOCK(&g_configMutex) { \
|
|
if (!g_configInterfaceInit) break; \
|
|
struct json_object *obj = configGetJsonObjectByPath(g_configJson, path); \
|
|
if (!obj || !configValidateJson##functype(obj, ##__VA_ARGS__)) break; \
|
|
ret = (vartype)json_object_get_##jsontype(obj); \
|
|
} \
|
|
return ret; \
|
|
}
|
|
|
|
#define JSON_SETTER(functype, vartype, jsontype, ...) \
|
|
void configSet##functype(const char *path, vartype value) { \
|
|
SCOPED_LOCK(&g_configMutex) { \
|
|
if (!g_configInterfaceInit) break; \
|
|
struct json_object *obj = configGetJsonObjectByPath(g_configJson, path); \
|
|
if (!obj || !configValidateJson##functype(obj, ##__VA_ARGS__)) break; \
|
|
if (json_object_set_##jsontype(obj, value)) { \
|
|
configWriteConfigJson(); \
|
|
} else { \
|
|
LOG_MSG("Failed to update \"%s\"!", path); \
|
|
} \
|
|
} \
|
|
}
|
|
|
|
/* Global variables. */
|
|
|
|
static Mutex g_configMutex = 0;
|
|
static bool g_configInterfaceInit = false;
|
|
|
|
static struct json_object *g_configJson = NULL;
|
|
|
|
/* Function prototypes. */
|
|
|
|
static bool configParseConfigJson(void);
|
|
static void configWriteConfigJson(void);
|
|
static void configFreeConfigJson(void);
|
|
|
|
static struct json_object *configGetJsonObjectByPath(const struct json_object *obj, const char *path);
|
|
|
|
static bool configValidateJsonRootObject(const struct json_object *obj);
|
|
static bool configValidateJsonGameCardObject(const struct json_object *obj);
|
|
static bool configValidateJsonNspObject(const struct json_object *obj);
|
|
static bool configValidateJsonTicketObject(const struct json_object *obj);
|
|
static bool configValidateJsonNcaFsObject(const struct json_object *obj);
|
|
|
|
NX_INLINE bool configValidateJsonBoolean(const struct json_object *obj);
|
|
NX_INLINE bool configValidateJsonInteger(const struct json_object *obj, int lower_boundary, int upper_boundary);
|
|
NX_INLINE bool configValidateJsonObject(const struct json_object *obj);
|
|
|
|
static void configLogJsonError(void);
|
|
|
|
bool configInitialize(void)
|
|
{
|
|
bool ret = false;
|
|
|
|
SCOPED_LOCK(&g_configMutex)
|
|
{
|
|
ret = g_configInterfaceInit;
|
|
if (ret) break;
|
|
|
|
/* Parse JSON config. */
|
|
if (!configParseConfigJson())
|
|
{
|
|
LOG_MSG("Failed to parse JSON configuration!");
|
|
break;
|
|
}
|
|
|
|
/* Update flags. */
|
|
ret = g_configInterfaceInit = true;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
void configExit(void)
|
|
{
|
|
SCOPED_LOCK(&g_configMutex)
|
|
{
|
|
/* Free JSON object. */
|
|
/* We don't need to write it back to the SD card - setter functions do that on their own. */
|
|
configFreeConfigJson();
|
|
|
|
/* Update flag. */
|
|
g_configInterfaceInit = false;
|
|
}
|
|
}
|
|
|
|
JSON_GETTER(Boolean, bool, boolean);
|
|
JSON_SETTER(Boolean, bool, boolean);
|
|
|
|
JSON_GETTER(Integer, int, int, INT32_MIN, INT32_MAX);
|
|
JSON_SETTER(Integer, int, int, INT32_MIN, INT32_MAX);
|
|
|
|
static bool configParseConfigJson(void)
|
|
{
|
|
bool use_default_config = true, ret = false;
|
|
|
|
/* Read config JSON. */
|
|
g_configJson = json_object_from_file(CONFIG_PATH);
|
|
if (!g_configJson)
|
|
{
|
|
configLogJsonError();
|
|
goto end;
|
|
}
|
|
|
|
/* Validate configuration. */
|
|
ret = configValidateJsonRootObject(g_configJson);
|
|
use_default_config = !ret;
|
|
|
|
end:
|
|
if (use_default_config)
|
|
{
|
|
LOG_MSG("Loading default configuration.");
|
|
|
|
/* Free config JSON. */
|
|
configFreeConfigJson();
|
|
|
|
/* Read default config JSON. */
|
|
g_configJson = json_object_from_file(DEFAULT_CONFIG_PATH);
|
|
if (g_configJson)
|
|
{
|
|
configWriteConfigJson();
|
|
ret = true;
|
|
} else {
|
|
configLogJsonError();
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void configWriteConfigJson(void)
|
|
{
|
|
if (!g_configJson) return;
|
|
if (json_object_to_file_ext(CONFIG_PATH, g_configJson, JSON_C_TO_STRING_SPACED | JSON_C_TO_STRING_PRETTY) != 0) configLogJsonError();
|
|
}
|
|
|
|
static void configFreeConfigJson(void)
|
|
{
|
|
if (!g_configJson) return;
|
|
json_object_put(g_configJson);
|
|
g_configJson = NULL;
|
|
}
|
|
|
|
static struct json_object *configGetJsonObjectByPath(const struct json_object *obj, const char *path)
|
|
{
|
|
const struct json_object *parent_obj = obj;
|
|
struct json_object *child_obj = NULL;
|
|
char *path_dup = NULL, *pch = NULL, *state = NULL;
|
|
|
|
if (!configValidateJsonObject(obj) || !path || !*path)
|
|
{
|
|
LOG_MSG("Invalid parameters!");
|
|
return NULL;
|
|
}
|
|
|
|
/* Duplicate path to avoid problems with strtok_r(). */
|
|
if (!(path_dup = strdup(path)))
|
|
{
|
|
LOG_MSG("Unable to duplicate input path! (\"%s\").", path);
|
|
return NULL;
|
|
}
|
|
|
|
pch = strtok_r(path_dup, "/", &state);
|
|
if (!pch)
|
|
{
|
|
LOG_MSG("Failed to tokenize input path! (\"%s\").", path);
|
|
goto end;
|
|
}
|
|
|
|
while(pch)
|
|
{
|
|
if (!json_object_object_get_ex(parent_obj, pch, &child_obj))
|
|
{
|
|
LOG_MSG("Failed to retrieve JSON object by key for \"%s\"! (\"%s\").", pch, path);
|
|
break;
|
|
}
|
|
|
|
pch = strtok_r(NULL, "/", &state);
|
|
if (pch)
|
|
{
|
|
parent_obj = child_obj;
|
|
child_obj = NULL;
|
|
}
|
|
}
|
|
|
|
end:
|
|
if (path_dup) free(path_dup);
|
|
|
|
return child_obj;
|
|
}
|
|
|
|
static bool configValidateJsonRootObject(const struct json_object *obj)
|
|
{
|
|
bool ret = false, overclock_found = false, naming_convention_found = false, dump_destination_found = false, gamecard_found = false;
|
|
bool nsp_found = false, ticket_found = false, nca_fs_found = false;
|
|
|
|
if (!configValidateJsonObject(obj)) goto end;
|
|
|
|
json_object_object_foreach(obj, key, val)
|
|
{
|
|
JSON_VALIDATE_FIELD(Boolean, overclock);
|
|
JSON_VALIDATE_FIELD(Integer, naming_convention, TitleNamingConvention_Full, TitleNamingConvention_Count - 1);
|
|
JSON_VALIDATE_FIELD(Integer, dump_destination, ConfigDumpDestination_SdCard, ConfigDumpDestination_Count - 1);
|
|
JSON_VALIDATE_OBJECT(GameCard, gamecard);
|
|
JSON_VALIDATE_OBJECT(Nsp, nsp);
|
|
JSON_VALIDATE_OBJECT(Ticket, ticket);
|
|
JSON_VALIDATE_OBJECT(NcaFs, nca_fs);
|
|
goto end;
|
|
}
|
|
|
|
ret = (overclock_found && naming_convention_found && dump_destination_found && gamecard_found && nsp_found && ticket_found && nca_fs_found);
|
|
|
|
end:
|
|
return ret;
|
|
}
|
|
|
|
static bool configValidateJsonGameCardObject(const struct json_object *obj)
|
|
{
|
|
bool ret = false, append_key_area_found = false, keep_certificate_found = false, trim_dump_found = false, calculate_checksum_found = false, checksum_lookup_method_found = false;
|
|
|
|
if (!configValidateJsonObject(obj)) goto end;
|
|
|
|
json_object_object_foreach(obj, key, val)
|
|
{
|
|
JSON_VALIDATE_FIELD(Boolean, append_key_area);
|
|
JSON_VALIDATE_FIELD(Boolean, keep_certificate);
|
|
JSON_VALIDATE_FIELD(Boolean, trim_dump);
|
|
JSON_VALIDATE_FIELD(Boolean, calculate_checksum);
|
|
JSON_VALIDATE_FIELD(Integer, checksum_lookup_method, ConfigChecksumLookupMethod_None, ConfigChecksumLookupMethod_Count - 1);
|
|
goto end;
|
|
}
|
|
|
|
ret = (append_key_area_found && keep_certificate_found && trim_dump_found && calculate_checksum_found && checksum_lookup_method_found);
|
|
|
|
end:
|
|
return ret;
|
|
}
|
|
|
|
static bool configValidateJsonNspObject(const struct json_object *obj)
|
|
{
|
|
bool ret = false, set_download_distribution_found = false, remove_console_data_found = false, remove_titlekey_crypto_found = false, replace_acid_key_sig_found = false;
|
|
bool disable_linked_account_requirement_found = false, enable_screenshots_found = false, enable_video_capture_found = false, disable_hdcp_found = false, append_authoringtool_data_found = false, lookup_checksum_found = false;
|
|
|
|
if (!configValidateJsonObject(obj)) goto end;
|
|
|
|
json_object_object_foreach(obj, key, val)
|
|
{
|
|
JSON_VALIDATE_FIELD(Boolean, set_download_distribution);
|
|
JSON_VALIDATE_FIELD(Boolean, remove_console_data);
|
|
JSON_VALIDATE_FIELD(Boolean, remove_titlekey_crypto);
|
|
JSON_VALIDATE_FIELD(Boolean, replace_acid_key_sig);
|
|
JSON_VALIDATE_FIELD(Boolean, disable_linked_account_requirement);
|
|
JSON_VALIDATE_FIELD(Boolean, enable_screenshots);
|
|
JSON_VALIDATE_FIELD(Boolean, enable_video_capture);
|
|
JSON_VALIDATE_FIELD(Boolean, disable_hdcp);
|
|
JSON_VALIDATE_FIELD(Boolean, lookup_checksum);
|
|
JSON_VALIDATE_FIELD(Boolean, append_authoringtool_data);
|
|
goto end;
|
|
}
|
|
|
|
ret = (set_download_distribution_found && remove_console_data_found && remove_titlekey_crypto_found && replace_acid_key_sig_found && disable_linked_account_requirement_found && \
|
|
enable_screenshots_found && enable_video_capture_found && disable_hdcp_found && append_authoringtool_data_found && lookup_checksum_found);
|
|
|
|
end:
|
|
return ret;
|
|
}
|
|
|
|
static bool configValidateJsonTicketObject(const struct json_object *obj)
|
|
{
|
|
bool ret = false, remove_console_data_found = false;
|
|
|
|
if (!configValidateJsonObject(obj)) goto end;
|
|
|
|
json_object_object_foreach(obj, key, val)
|
|
{
|
|
JSON_VALIDATE_FIELD(Boolean, remove_console_data);
|
|
goto end;
|
|
}
|
|
|
|
ret = remove_console_data_found;
|
|
|
|
end:
|
|
return ret;
|
|
}
|
|
|
|
static bool configValidateJsonNcaFsObject(const struct json_object *obj)
|
|
{
|
|
bool ret = false, use_layeredfs_dir_found = false;
|
|
|
|
if (!configValidateJsonObject(obj)) goto end;
|
|
|
|
json_object_object_foreach(obj, key, val)
|
|
{
|
|
JSON_VALIDATE_FIELD(Boolean, use_layeredfs_dir);
|
|
goto end;
|
|
}
|
|
|
|
ret = use_layeredfs_dir_found;
|
|
|
|
end:
|
|
return ret;
|
|
}
|
|
|
|
NX_INLINE bool configValidateJsonBoolean(const struct json_object *obj)
|
|
{
|
|
return (obj != NULL && json_object_is_type(obj, json_type_boolean));
|
|
}
|
|
|
|
NX_INLINE bool configValidateJsonInteger(const struct json_object *obj, int lower_boundary, int upper_boundary)
|
|
{
|
|
if (!obj || !json_object_is_type(obj, json_type_int) || lower_boundary > upper_boundary) return false;
|
|
int val = json_object_get_int(obj);
|
|
return (val >= lower_boundary && val <= upper_boundary);
|
|
}
|
|
|
|
NX_INLINE bool configValidateJsonObject(const struct json_object *obj)
|
|
{
|
|
return (obj != NULL && json_object_is_type(obj, json_type_object) && json_object_object_length(obj) > 0);
|
|
}
|
|
|
|
static void configLogJsonError(void)
|
|
{
|
|
size_t str_len = 0;
|
|
char *str = (char*)json_util_get_last_err(); /* Drop the const. */
|
|
if (!str || !(str_len = strlen(str))) return;
|
|
|
|
/* Remove line breaks. */
|
|
if (str[str_len - 1] == '\n') str[--str_len] = '\0';
|
|
if (str[str_len - 1] == '\r') str[--str_len] = '\0';
|
|
|
|
/* Log error message. */
|
|
LOG_MSG("%s", str);
|
|
}
|