FreeRDP/winpr/libwinpr/utils/wlog/Layout.c

/**
 * WinPR: Windows Portable Runtime
 * WinPR Logger
 *
 * Copyright 2013 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <string.h>
#include <stdarg.h>

#include <winpr/crt.h>
#include <winpr/print.h>
#include <winpr/sysinfo.h>
#include <winpr/environment.h>

#include "wlog.h"

#include "wlog/Layout.h"

#if defined __linux__ && !defined ANDROID
#include <unistd.h>
#include <sys/syscall.h>
#endif

extern const char* WLOG_LEVELS[7];

/**
 * Log Layout
 */

static void WLog_PrintMessagePrefixVA(wLog* log, wLogMessage* message, const char* format, va_list args)
{
	if (!strchr(format, '%'))
		sprintf_s(message->PrefixString, WLOG_MAX_PREFIX_SIZE - 1, "%s", format);
	else
		wvsnprintfx(message->PrefixString, WLOG_MAX_PREFIX_SIZE - 1, format, args);
}

static void WLog_PrintMessagePrefix(wLog* log, wLogMessage* message, const char* format, ...)
{
	va_list args;
	va_start(args, format);
	WLog_PrintMessagePrefixVA(log, message, format, args);
	va_end(args);
}

BOOL WLog_Layout_GetMessagePrefix(wLog* log, wLogLayout* layout, wLogMessage* message)
{
	char* p;
	int index;
	int argc = 0;
	void* args[32];
	char format[256];
	SYSTEMTIME localTime;
	GetLocalTime(&localTime);
	index = 0;
	p = (char*) layout->FormatString;

	while (*p)
	{
		if (*p == '%')
		{
			p++;

			if (*p)
			{
				if ((p[0] == 'l') && (p[1] == 'v')) /* log level */
				{
					args[argc++] = (void*) WLOG_LEVELS[message->Level];
					format[index++] = '%';
					format[index++] = 's';
					p++;
				}
				else if ((p[0] == 'm') && (p[1] == 'n')) /* module name */
				{
					args[argc++] = (void*) log->Name;
					format[index++] = '%';
					format[index++] = 's';
					p++;
				}
				else if ((p[0] == 'f') && (p[1] == 'l')) /* file */
				{
					char* file;
					file = strrchr(message->FileName, '/');

					if (!file)
						file = strrchr(message->FileName, '\\');

					if (file)
						file++;
					else
						file = (char*) message->FileName;

					args[argc++] = (void*) file;
					format[index++] = '%';
					format[index++] = 's';
					p++;
				}
				else if ((p[0] == 'f') && (p[1] == 'n')) /* function */
				{
					args[argc++] = (void*) message->FunctionName;
					format[index++] = '%';
					format[index++] = 's';
					p++;
				}
				else if ((p[0] == 'l') && (p[1] == 'n')) /* line number */
				{
					args[argc++] = (void*)(size_t) message->LineNumber;
					format[index++] = '%';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'p') && (p[1] == 'i') && (p[2] == 'd')) /* process id */
				{
					args[argc++] = (void*)(size_t) GetCurrentProcessId();
					format[index++] = '%';
					format[index++] = 'u';
					p += 2;
				}
				else if ((p[0] == 't') && (p[1] == 'i') && (p[2] == 'd')) /* thread id */
				{
#if defined __linux__ && !defined ANDROID
					/* On Linux we prefer to see the LWP id */
					args[argc++] = (void*)(size_t) syscall(SYS_gettid);
					format[index++] = '%';
					format[index++] = 'l';
					format[index++] = 'd';
#else
					args[argc++] = (void*)(size_t) GetCurrentThreadId();
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '8';
					format[index++] = 'x';
#endif
					p += 2;
				}
				else if ((p[0] == 'y') && (p[1] == 'r')) /* year */
				{
					args[argc++] = (void*)(size_t) localTime.wYear;
					format[index++] = '%';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'm') && (p[1] == 'o')) /* month */
				{
					args[argc++] = (void*)(size_t) localTime.wMonth;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '2';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'd') && (p[1] == 'w')) /* day of week */
				{
					args[argc++] = (void*)(size_t) localTime.wDayOfWeek;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '2';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'd') && (p[1] == 'y')) /* day */
				{
					args[argc++] = (void*)(size_t) localTime.wDay;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '2';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'h') && (p[1] == 'r')) /* hours */
				{
					args[argc++] = (void*)(size_t) localTime.wHour;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '2';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'm') && (p[1] == 'i')) /* minutes */
				{
					args[argc++] = (void*)(size_t) localTime.wMinute;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '2';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 's') && (p[1] == 'e')) /* seconds */
				{
					args[argc++] = (void*)(size_t) localTime.wSecond;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '2';
					format[index++] = 'u';
					p++;
				}
				else if ((p[0] == 'm') && (p[1] == 'l')) /* milliseconds */
				{
					args[argc++] = (void*)(size_t) localTime.wMilliseconds;
					format[index++] = '%';
					format[index++] = '0';
					format[index++] = '3';
					format[index++] = 'u';
					p++;
				}
			}
		}
		else
		{
			format[index++] = *p;
		}

		p++;
	}

	format[index++] = '\0';

	switch (argc)
	{
		case 0:
			WLog_PrintMessagePrefix(log, message, format);
			break;

		case 1:
			WLog_PrintMessagePrefix(log, message, format, args[0]);
			break;

		case 2:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1]);
			break;

		case 3:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2]);
			break;

		case 4:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3]);
			break;

		case 5:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4]);
			break;

		case 6:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5]);
			break;

		case 7:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6]);
			break;

		case 8:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7]);
			break;

		case 9:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8]);
			break;

		case 10:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9]);
			break;

		case 11:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9], args[10]);
			break;

		case 12:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9], args[10],
			                        args[11]);
			break;

		case 13:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9], args[10],
			                        args[11], args[12]);
			break;

		case 14:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9], args[10],
			                        args[11], args[12], args[13]);
			break;

		case 15:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9], args[10],
			                        args[11], args[12], args[13], args[14]);
			break;

		case 16:
			WLog_PrintMessagePrefix(log, message, format, args[0], args[1], args[2], args[3],
			                        args[4], args[5], args[6], args[7], args[8], args[9], args[10],
			                        args[11], args[12], args[13], args[14], args[15]);
			break;
	}

	return TRUE;
}

wLogLayout* WLog_GetLogLayout(wLog* log)
{
	wLogAppender* appender;
	appender = WLog_GetLogAppender(log);
	return appender->Layout;
}

BOOL WLog_Layout_SetPrefixFormat(wLog* log, wLogLayout* layout, const char* format)
{
	free(layout->FormatString);
	layout->FormatString = NULL;

	if (format)
	{
		layout->FormatString = _strdup(format);

		if (!layout->FormatString)
			return FALSE;
	}

	return TRUE;
}

wLogLayout* WLog_Layout_New(wLog* log)
{
	LPCSTR prefix = "WLOG_PREFIX";
	DWORD nSize;
	char* env = NULL;
	wLogLayout* layout;
	layout = (wLogLayout*) calloc(1, sizeof(wLogLayout));

	if (!layout)
		return NULL;

	nSize = GetEnvironmentVariableA(prefix, NULL, 0);

	if (nSize)
	{
		env = (LPSTR) malloc(nSize);

		if (!env)
		{
			free(layout);
			return NULL;
		}

		if (GetEnvironmentVariableA(prefix, env, nSize) != nSize - 1)
		{
			free(env);
			free(layout);
			return NULL;
		}
	}

	if (env)
		layout->FormatString = env;
	else
	{
#ifdef ANDROID
		layout->FormatString = _strdup("[pid=%pid:tid=%tid] - ");
#else
		layout->FormatString = _strdup("[%hr:%mi:%se:%ml] [%pid:%tid] [%lv][%mn] - ");
#endif

		if (!layout->FormatString)
		{
			free(layout);
			return NULL;
		}
	}

	return layout;
}

void WLog_Layout_Free(wLog* log, wLogLayout* layout)
{
	if (layout)
	{
		if (layout->FormatString)
		{
			free(layout->FormatString);
			layout->FormatString = NULL;
		}

		free(layout);
	}
}