#include "config.h"
#include "DataLog.h"
#include <stdarg.h>
#include <string.h>
#include <wtf/FilePrintStream.h>
#include <wtf/LockedPrintStream.h>
#include <wtf/Threading.h>
#include <mutex>
#include <thread>
#if OS(UNIX) || OS(DARWIN)
#include <unistd.h>
#endif
#define DATA_LOG_TO_FILE 0
#define DATA_LOG_TO_DARWIN_TEMP_DIR 0
#define DATA_LOG_FILENAME "/tmp/WTFLog"
namespace WTF {
static PrintStream* s_file;
static uint64_t s_fileData[(sizeof(FilePrintStream) + 7) / 8];
static void initializeLogFileOnce()
{
FilePrintStream* file = nullptr;
#if DATA_LOG_TO_FILE
const long maxPathLength = 1024;
char filenameSuffix[maxPathLength + 1];
#if PLATFORM(WIN)
_snprintf(filenameSuffix, sizeof(filenameSuffix), ".%d.txt", GetCurrentProcessId());
#else
snprintf(filenameSuffix, sizeof(filenameSuffix), ".%d.txt", getpid());
#endif
#if DATA_LOG_TO_DARWIN_TEMP_DIR
char filenameBuffer[maxPathLength + 1];
unsigned suffixLength = strlen(filenameSuffix);
#if defined(DATA_LOG_FILENAME)
char* logBasename = strrchr(DATA_LOG_FILENAME, '/');
if (!logBasename)
logBasename = (char*)DATA_LOG_FILENAME;
#else
const char* logBasename = "WTFLog";
#endif
const char* filename = nullptr;
bool success = confstr(_CS_DARWIN_USER_TEMP_DIR, filenameBuffer, sizeof(filenameBuffer));
if (success) {
size_t lastComponentLength = strlen(logBasename) + suffixLength;
size_t dirnameLength = strlen(filenameBuffer);
bool shouldAddPathSeparator = filenameBuffer[dirnameLength - 1] != '/' && logBasename[0] != '/';
if (lastComponentLength + shouldAddPathSeparator <= sizeof(filenameBuffer) - dirnameLength - 1) {
if (shouldAddPathSeparator)
strncat(filenameBuffer, "/", 1);
strncat(filenameBuffer, logBasename, sizeof(filenameBuffer) - strlen(filenameBuffer) - 1);
filename = filenameBuffer;
}
}
#elif defined(DATA_LOG_FILENAME)
const char* filename = DATA_LOG_FILENAME;
#else
const char* filename = getenv("WTF_DATA_LOG_FILENAME");
#endif
char actualFilename[maxPathLength + 1];
if (filename) {
#if PLATFORM(WIN)
_snprintf(actualFilename, sizeof(actualFilename), "%s%s", filename, filenameSuffix);
#else
snprintf(actualFilename, sizeof(actualFilename), "%s%s", filename, filenameSuffix);
#endif
file = FilePrintStream::open(actualFilename, "w").release();
if (file)
WTFLogAlways("*** DataLog output to \"%s\" ***\n", actualFilename);
else
WTFLogAlways("Warning: Could not open DataLog file %s for writing.\n", actualFilename);
}
#endif // DATA_LOG_TO_FILE
bool wrapWithLocked = true;
if (!file) {
file = new (s_fileData) FilePrintStream(stderr, FilePrintStream::Borrow);
wrapWithLocked = false;
}
setvbuf(file->file(), 0, _IONBF, 0);
if (wrapWithLocked)
s_file = new LockedPrintStream(std::unique_ptr<FilePrintStream>(file));
else
s_file = file;
}
static void initializeLogFile()
{
static std::once_flag once;
std::call_once(
once,
[] {
initializeLogFileOnce();
});
}
PrintStream& dataFile()
{
initializeLogFile();
return *s_file;
}
void dataLogFV(const char* format, va_list argList)
{
dataFile().vprintf(format, argList);
}
void dataLogF(const char* format, ...)
{
va_list argList;
va_start(argList, format);
dataLogFV(format, argList);
va_end(argList);
}
void dataLogFString(const char* str)
{
dataFile().printf("%s", str);
}
}