mirror of
https://github.com/DarkMatterCore/nxdumptool.git
synced 2024-10-30 07:21:46 +00:00
506 lines
15 KiB
C
506 lines
15 KiB
C
/*
|
|
* nxdt_log.c
|
|
*
|
|
* Copyright (c) 2020-2024, 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"
|
|
|
|
#if (LOG_LEVEL >= LOG_LEVEL_DEBUG) && (LOG_LEVEL < LOG_LEVEL_NONE)
|
|
|
|
/* Global variables. */
|
|
|
|
static Mutex g_logMutex = 0;
|
|
|
|
static char *g_lastLogMsg = NULL;
|
|
|
|
static FsFile g_logFile = {0};
|
|
static s64 g_logFileOffset = 0;
|
|
|
|
static char *g_logBuffer = NULL;
|
|
static size_t g_logBufferLength = 0;
|
|
|
|
static const char *g_logStrFormat = "[%d-%02d-%02d %02d:%02d:%02d.%09lu] [%s] %s:%d:%s -> ";
|
|
static const char *g_logSessionSeparator = "________________________________________________________________\r\n";
|
|
|
|
static const char *g_logLevelNames[] = {
|
|
[LOG_LEVEL_DEBUG] = "DEBUG",
|
|
[LOG_LEVEL_INFO] = "INFO",
|
|
[LOG_LEVEL_WARNING] = "WARNING",
|
|
[LOG_LEVEL_ERROR] = "ERROR"
|
|
};
|
|
|
|
/* Function prototypes. */
|
|
|
|
static void _logWriteStringToLogFile(const char *src);
|
|
static void _logWriteFormattedStringToLogFile(bool save, u8 level, const char *file_name, int line, const char *func_name, const char *fmt, va_list args);
|
|
|
|
static void _logFlushLogFile(void);
|
|
|
|
static bool logAllocateLogBuffer(void);
|
|
static bool logOpenLogFile(void);
|
|
|
|
static void logWriteStringToNxLink(const char *str);
|
|
|
|
void logWriteStringToLogFile(const char *src)
|
|
{
|
|
SCOPED_LOCK(&g_logMutex) _logWriteStringToLogFile(src);
|
|
}
|
|
|
|
__attribute__((format(printf, 5, 6))) void logWriteFormattedStringToLogFile(u8 level, const char *file_name, int line, const char *func_name, const char *fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
SCOPED_LOCK(&g_logMutex) _logWriteFormattedStringToLogFile(true, level, file_name, line, func_name, fmt, args);
|
|
va_end(args);
|
|
}
|
|
|
|
__attribute__((format(printf, 7, 8))) void logWriteFormattedStringToBuffer(char **dst, size_t *dst_size, u8 level, const char *file_name, int line, const char *func_name, const char *fmt, ...)
|
|
{
|
|
if (!dst || !dst_size || level < LOG_LEVEL || !file_name || !*file_name || !func_name || !*func_name || !fmt || !*fmt) return;
|
|
|
|
va_list args;
|
|
|
|
int str1_len = 0, str2_len = 0;
|
|
size_t log_str_len = 0;
|
|
|
|
char *dst_ptr = *dst, *tmp_str = NULL;
|
|
size_t dst_cur_size = *dst_size, dst_str_len = (dst_ptr ? strlen(dst_ptr) : 0);
|
|
|
|
struct tm ts = {0};
|
|
struct timespec now = {0};
|
|
|
|
/* Sanity check. */
|
|
if (dst_cur_size && dst_str_len >= dst_cur_size) return;
|
|
|
|
va_start(args, fmt);
|
|
|
|
/* Get current time with nanosecond precision. */
|
|
clock_gettime(CLOCK_REALTIME, &now);
|
|
|
|
/* Get local time. */
|
|
localtime_r(&(now.tv_sec), &ts);
|
|
ts.tm_year += 1900;
|
|
ts.tm_mon++;
|
|
|
|
/* Get formatted string length. */
|
|
str1_len = snprintf(NULL, 0, g_logStrFormat, ts.tm_year, ts.tm_mon, ts.tm_mday, ts.tm_hour, ts.tm_min, ts.tm_sec, now.tv_nsec, g_logLevelNames[level], file_name, line, func_name);
|
|
if (str1_len <= 0) goto end;
|
|
|
|
str2_len = vsnprintf(NULL, 0, fmt, args);
|
|
if (str2_len <= 0) goto end;
|
|
|
|
log_str_len = (size_t)(str1_len + str2_len + 3);
|
|
|
|
if (!dst_ptr || !dst_cur_size || log_str_len > (dst_cur_size - dst_str_len))
|
|
{
|
|
/* Update buffer size. */
|
|
dst_cur_size = (dst_str_len + log_str_len);
|
|
|
|
/* Reallocate buffer. */
|
|
tmp_str = realloc(dst_ptr, dst_cur_size);
|
|
if (!tmp_str) goto end;
|
|
|
|
dst_ptr = tmp_str;
|
|
tmp_str = NULL;
|
|
|
|
/* Clear allocated area. */
|
|
memset(dst_ptr + dst_str_len, 0, log_str_len);
|
|
|
|
/* Update pointers. */
|
|
*dst = dst_ptr;
|
|
*dst_size = dst_cur_size;
|
|
}
|
|
|
|
/* Generate formatted string. */
|
|
sprintf(dst_ptr + dst_str_len, g_logStrFormat, ts.tm_year, ts.tm_mon, ts.tm_mday, ts.tm_hour, ts.tm_min, ts.tm_sec, now.tv_nsec, g_logLevelNames[level], file_name, line, func_name);
|
|
vsprintf(dst_ptr + dst_str_len + (size_t)str1_len, fmt, args);
|
|
strcat(dst_ptr, CRLF);
|
|
|
|
end:
|
|
va_end(args);
|
|
}
|
|
|
|
__attribute__((format(printf, 7, 8))) void logWriteBinaryDataToLogFile(const void *data, size_t data_size, u8 level, const char *file_name, int line, const char *func_name, const char *fmt, ...)
|
|
{
|
|
if (!data || !data_size || level < LOG_LEVEL || !file_name || !*file_name || !func_name || !*func_name || !fmt || !*fmt) return;
|
|
|
|
va_list args;
|
|
size_t data_str_size = ((data_size * 2) + 3);
|
|
char *data_str = NULL;
|
|
|
|
/* Allocate memory for the hex string representation of the provided binary data. */
|
|
data_str = calloc(data_str_size, sizeof(char));
|
|
if (!data_str) goto end;
|
|
|
|
/* Generate hex string representation. */
|
|
utilsGenerateHexString(data_str, data_str_size, data, data_size, true);
|
|
strcat(data_str, CRLF);
|
|
|
|
SCOPED_LOCK(&g_logMutex)
|
|
{
|
|
/* Write formatted string. */
|
|
va_start(args, fmt);
|
|
_logWriteFormattedStringToLogFile(false, level, file_name, line, func_name, fmt, args);
|
|
va_end(args);
|
|
|
|
/* Write hex string representation. */
|
|
_logWriteStringToLogFile(data_str);
|
|
}
|
|
|
|
end:
|
|
if (data_str) free(data_str);
|
|
}
|
|
|
|
void logFlushLogFile(void)
|
|
{
|
|
SCOPED_LOCK(&g_logMutex) _logFlushLogFile();
|
|
}
|
|
|
|
void logCloseLogFile(void)
|
|
{
|
|
SCOPED_LOCK(&g_logMutex)
|
|
{
|
|
/* Flush log buffer. */
|
|
_logFlushLogFile();
|
|
|
|
/* Close logfile. */
|
|
if (serviceIsActive(&(g_logFile.s)))
|
|
{
|
|
fsFileClose(&g_logFile);
|
|
memset(&g_logFile, 0, sizeof(FsFile));
|
|
|
|
/* Commit SD card filesystem changes. */
|
|
utilsCommitSdCardFileSystemChanges();
|
|
}
|
|
|
|
/* Free last message bugger. */
|
|
if (g_lastLogMsg)
|
|
{
|
|
free(g_lastLogMsg);
|
|
g_lastLogMsg = NULL;
|
|
}
|
|
|
|
/* Free log buffer. */
|
|
if (g_logBuffer)
|
|
{
|
|
free(g_logBuffer);
|
|
g_logBuffer = NULL;
|
|
}
|
|
|
|
/* Reset logfile offset. */
|
|
g_logFileOffset = 0;
|
|
}
|
|
}
|
|
|
|
char *logGetLastMessage(void)
|
|
{
|
|
char *ret = NULL;
|
|
|
|
SCOPED_LOCK(&g_logMutex)
|
|
{
|
|
if (g_lastLogMsg) ret = strdup(g_lastLogMsg);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
void logControlMutex(bool lock)
|
|
{
|
|
bool locked = mutexIsLockedByCurrentThread(&g_logMutex);
|
|
|
|
if (!locked && lock)
|
|
{
|
|
mutexLock(&g_logMutex);
|
|
} else
|
|
if (locked && !lock)
|
|
{
|
|
mutexUnlock(&g_logMutex);
|
|
}
|
|
}
|
|
|
|
static void _logWriteStringToLogFile(const char *src)
|
|
{
|
|
/* Make sure we have allocated memory for the log buffer and opened the logfile. */
|
|
if (!src || !*src || !logAllocateLogBuffer() || !logOpenLogFile()) return;
|
|
|
|
Result rc = 0;
|
|
size_t src_len = strlen(src), tmp_len = 0;
|
|
|
|
/* Check if the formatted string length is lower than the log buffer size. */
|
|
if (src_len < LOG_BUF_SIZE)
|
|
{
|
|
/* Flush log buffer contents (if needed). */
|
|
if ((g_logBufferLength + src_len) >= LOG_BUF_SIZE)
|
|
{
|
|
_logFlushLogFile();
|
|
if (g_logBufferLength) return;
|
|
}
|
|
|
|
/* Copy string into the log buffer. */
|
|
strcpy(g_logBuffer + g_logBufferLength, src);
|
|
g_logBufferLength += src_len;
|
|
} else {
|
|
/* Flush log buffer. */
|
|
_logFlushLogFile();
|
|
if (g_logBufferLength) return;
|
|
|
|
/* Write string data until it no longer exceeds the log buffer size. */
|
|
while(src_len >= LOG_BUF_SIZE)
|
|
{
|
|
rc = fsFileWrite(&g_logFile, g_logFileOffset, src + tmp_len, LOG_BUF_SIZE, FsWriteOption_Flush);
|
|
if (R_FAILED(rc)) return;
|
|
|
|
g_logFileOffset += LOG_BUF_SIZE;
|
|
tmp_len += LOG_BUF_SIZE;
|
|
src_len -= LOG_BUF_SIZE;
|
|
}
|
|
|
|
/* Copy any remaining data from the string into the log buffer. */
|
|
if (src_len)
|
|
{
|
|
strcpy(g_logBuffer, src + tmp_len);
|
|
g_logBufferLength = src_len;
|
|
}
|
|
}
|
|
|
|
/* Write data to nxlink. */
|
|
logWriteStringToNxLink(src);
|
|
|
|
#if LOG_FORCE_FLUSH == 1
|
|
/* Flush log buffer. */
|
|
_logFlushLogFile();
|
|
#endif
|
|
}
|
|
|
|
static void _logWriteFormattedStringToLogFile(bool save, u8 level, const char *file_name, int line, const char *func_name, const char *fmt, va_list args)
|
|
{
|
|
/* Make sure we have allocated memory for the log buffer and opened the logfile. */
|
|
if (level < LOG_LEVEL || !file_name || !*file_name || !func_name || !*func_name || !fmt || !*fmt || !logAllocateLogBuffer() || !logOpenLogFile()) return;
|
|
|
|
Result rc = 0;
|
|
|
|
int str1_len = 0, str2_len = 0;
|
|
size_t log_str_len = 0;
|
|
|
|
char *tmp_str = NULL;
|
|
size_t tmp_len = 0;
|
|
|
|
struct tm ts = {0};
|
|
struct timespec now = {0};
|
|
|
|
/* Get current time with nanosecond precision. */
|
|
clock_gettime(CLOCK_REALTIME, &now);
|
|
|
|
/* Get local time. */
|
|
localtime_r(&(now.tv_sec), &ts);
|
|
ts.tm_year += 1900;
|
|
ts.tm_mon++;
|
|
|
|
/* Get formatted string length. */
|
|
str1_len = snprintf(NULL, 0, g_logStrFormat, ts.tm_year, ts.tm_mon, ts.tm_mday, ts.tm_hour, ts.tm_min, ts.tm_sec, now.tv_nsec, g_logLevelNames[level], file_name, line, func_name);
|
|
if (str1_len <= 0) return;
|
|
|
|
str2_len = vsnprintf(NULL, 0, fmt, args);
|
|
if (str2_len <= 0) return;
|
|
|
|
log_str_len = (size_t)(str1_len + str2_len + 2);
|
|
|
|
/* Save log message (if needed). */
|
|
if (save)
|
|
{
|
|
if (g_lastLogMsg) free(g_lastLogMsg);
|
|
|
|
tmp_len = (strlen(func_name) + 2);
|
|
|
|
g_lastLogMsg = calloc(tmp_len + (size_t)str2_len + 1, sizeof(char));
|
|
if (g_lastLogMsg)
|
|
{
|
|
sprintf(g_lastLogMsg, "%s: ", func_name);
|
|
vsprintf(g_lastLogMsg + tmp_len, fmt, args);
|
|
}
|
|
|
|
tmp_len = 0;
|
|
}
|
|
|
|
/* Check if the formatted string length is less than the log buffer size. */
|
|
if (log_str_len < LOG_BUF_SIZE)
|
|
{
|
|
/* Flush log buffer contents (if needed). */
|
|
if ((g_logBufferLength + log_str_len) >= LOG_BUF_SIZE)
|
|
{
|
|
_logFlushLogFile();
|
|
if (g_logBufferLength) return;
|
|
}
|
|
|
|
/* Nice and easy string formatting using the log buffer. */
|
|
sprintf(g_logBuffer + g_logBufferLength, g_logStrFormat, ts.tm_year, ts.tm_mon, ts.tm_mday, ts.tm_hour, ts.tm_min, ts.tm_sec, now.tv_nsec, g_logLevelNames[level], file_name, line, func_name);
|
|
vsprintf(g_logBuffer + g_logBufferLength + (size_t)str1_len, fmt, args);
|
|
strcat(g_logBuffer, CRLF);
|
|
|
|
/* Write data to nxlink. */
|
|
logWriteStringToNxLink(g_logBuffer + g_logBufferLength);
|
|
|
|
/* Update log buffer length. */
|
|
g_logBufferLength += log_str_len;
|
|
} else {
|
|
/* Flush log buffer. */
|
|
_logFlushLogFile();
|
|
if (g_logBufferLength) return;
|
|
|
|
/* Allocate memory for a temporary buffer. This will hold the formatted string. */
|
|
tmp_str = calloc(log_str_len + 1, sizeof(char));
|
|
if (!tmp_str) return;
|
|
|
|
/* Generate formatted string. */
|
|
sprintf(tmp_str, g_logStrFormat, ts.tm_year, ts.tm_mon, ts.tm_mday, ts.tm_hour, ts.tm_min, ts.tm_sec, now.tv_nsec, g_logLevelNames[level], file_name, line, func_name);
|
|
vsprintf(tmp_str + (size_t)str1_len, fmt, args);
|
|
strcat(tmp_str, CRLF);
|
|
|
|
/* Write data to nxlink. */
|
|
logWriteStringToNxLink(tmp_str);
|
|
|
|
/* Write formatted string data until it no longer exceeds the log buffer size. */
|
|
while(log_str_len >= LOG_BUF_SIZE)
|
|
{
|
|
rc = fsFileWrite(&g_logFile, g_logFileOffset, tmp_str + tmp_len, LOG_BUF_SIZE, FsWriteOption_Flush);
|
|
if (R_FAILED(rc)) goto end;
|
|
|
|
g_logFileOffset += LOG_BUF_SIZE;
|
|
tmp_len += LOG_BUF_SIZE;
|
|
log_str_len -= LOG_BUF_SIZE;
|
|
}
|
|
|
|
/* Copy any remaining data from the formatted string into the log buffer. */
|
|
if (log_str_len)
|
|
{
|
|
strcpy(g_logBuffer, tmp_str + tmp_len);
|
|
g_logBufferLength = log_str_len;
|
|
}
|
|
}
|
|
|
|
#if LOG_FORCE_FLUSH == 1
|
|
/* Flush log buffer. */
|
|
_logFlushLogFile();
|
|
#endif
|
|
|
|
end:
|
|
if (tmp_str) free(tmp_str);
|
|
}
|
|
|
|
static void _logFlushLogFile(void)
|
|
{
|
|
if (!serviceIsActive(&(g_logFile.s)) || !g_logBuffer || !g_logBufferLength) return;
|
|
|
|
/* Write log buffer contents and flush the written data right away. */
|
|
Result rc = fsFileWrite(&g_logFile, g_logFileOffset, g_logBuffer, g_logBufferLength, FsWriteOption_Flush);
|
|
if (R_SUCCEEDED(rc))
|
|
{
|
|
/* Update global variables. */
|
|
g_logFileOffset += (s64)g_logBufferLength;
|
|
*g_logBuffer = '\0';
|
|
g_logBufferLength = 0;
|
|
}
|
|
}
|
|
|
|
static bool logAllocateLogBuffer(void)
|
|
{
|
|
if (g_logBuffer) return true;
|
|
g_logBuffer = memalign(LOG_BUF_SIZE, LOG_BUF_SIZE);
|
|
return (g_logBuffer != NULL);
|
|
}
|
|
|
|
static bool logOpenLogFile(void)
|
|
{
|
|
if (serviceIsActive(&(g_logFile.s))) return true;
|
|
|
|
Result rc = 0;
|
|
bool use_root = true;
|
|
const char *launch_path = utilsGetLaunchPath();
|
|
char path[FS_MAX_PATH] = {0}, *ptr1 = NULL, *ptr2 = NULL;
|
|
|
|
/* Get SD card FsFileSystem object. */
|
|
FsFileSystem *sdmc_fs = utilsGetSdCardFileSystemObject();
|
|
if (!sdmc_fs) return false;
|
|
|
|
/* Generate logfile path. */
|
|
if (launch_path)
|
|
{
|
|
ptr1 = strchr(launch_path, '/');
|
|
ptr2 = strrchr(launch_path, '/');
|
|
|
|
if (ptr1 && ptr2 && ptr1 != ptr2)
|
|
{
|
|
/* Create logfile in the current working directory. Strip the devoptab device name prefix while we're at it, since we won't need it. */
|
|
snprintf(path, sizeof(path), "%.*s" LOG_FILE_NAME, (int)((ptr2 - ptr1) + 1), ptr1);
|
|
use_root = false;
|
|
}
|
|
}
|
|
|
|
/* Create logfile in the SD card root directory, if needed. */
|
|
if (use_root) sprintf(path, "/" LOG_FILE_NAME);
|
|
|
|
/* Create file. This will fail if the logfile exists, so we don't check its return value. */
|
|
fsFsCreateFile(sdmc_fs, path, 0, 0);
|
|
|
|
/* Open file. */
|
|
rc = fsFsOpenFile(sdmc_fs, path, FsOpenMode_Write | FsOpenMode_Append, &g_logFile);
|
|
if (R_SUCCEEDED(rc))
|
|
{
|
|
/* Get file size. */
|
|
rc = fsFileGetSize(&g_logFile, &g_logFileOffset);
|
|
if (R_SUCCEEDED(rc))
|
|
{
|
|
size_t len = 0;
|
|
|
|
if (!g_logFileOffset)
|
|
{
|
|
/* Write UTF-8 BOM if the logfile is empty. */
|
|
len = strlen(UTF8_BOM);
|
|
rc = fsFileWrite(&g_logFile, g_logFileOffset, UTF8_BOM, len, FsWriteOption_Flush);
|
|
} else {
|
|
/* Write session separator if the logfile isn't empty. */
|
|
len = strlen(g_logSessionSeparator);
|
|
rc = fsFileWrite(&g_logFile, g_logFileOffset, g_logSessionSeparator, len, FsWriteOption_Flush);
|
|
}
|
|
|
|
if (R_SUCCEEDED(rc)) g_logFileOffset += (s64)len;
|
|
}
|
|
}
|
|
|
|
/* Close file if we successfully opened it, but an error occurred afterwards. */
|
|
if (R_FAILED(rc) && serviceIsActive(&(g_logFile.s)))
|
|
{
|
|
fsFileClose(&g_logFile);
|
|
memset(&g_logFile, 0, sizeof(FsFile));
|
|
}
|
|
|
|
return R_SUCCEEDED(rc);
|
|
}
|
|
|
|
static void logWriteStringToNxLink(const char *str)
|
|
{
|
|
int fd = utilsGetNxLinkFileDescriptor();
|
|
if (fd >= 0)
|
|
{
|
|
dprintf(fd, "%s", str);
|
|
fsync(fd);
|
|
}
|
|
}
|
|
|
|
#endif /* (LOG_LEVEL >= LOG_LEVEL_DEBUG) && (LOG_LEVEL < LOG_LEVEL_NONE) */
|