/*
    Variables Plugin for Miranda-IM (www.miranda-im.org)
    Copyright 2003-2006 P. Boon

    This program 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 2 of the License, or
    (at your option) any later version.

    This program 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, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include "variables.h"

static TCHAR *parseComputerName(ARGUMENTSINFO *ai)
{
	if (ai->argc != 1)
		return NULL;

	ai->flags |= AIF_DONTPARSE;
	DWORD len = MAX_COMPUTERNAME_LENGTH;
	TCHAR *res = (TCHAR*)mir_calloc((len + 1) * sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	if (!GetComputerName(res, &len)) {
		mir_free(res);
		return NULL;
	}
	return res;
}

#if _WIN32_WINNT>=0x0500
#include <pdh.h>
#include <pdhmsg.h>

static TCHAR *parseCpuLoad(ARGUMENTSINFO *ai)
{

	HQUERY hQuery;
	PDH_FMT_COUNTERVALUE cValue;
	HCOUNTER hCounter;
	TCHAR *szCounter, szVal[32];

	if (ai->argc != 2)
		return NULL;

	if (_tcslen(ai->targv[1]) == 0)
		szCounter = mir_tstrdup(_T("\\Processor(_Total)\\% Processor Time"));
	else {
		int size = (int)_tcslen(ai->targv[1]) + 32;
		szCounter = (TCHAR *)mir_alloc(size * sizeof(TCHAR));
		if (szCounter == NULL)
			return NULL;

		mir_sntprintf(szCounter, size, _T("\\Process(%s)\\%% Processor Time"), ai->targv[1]);
	}
	PDH_STATUS pdhStatus = PdhValidatePath(szCounter);
	if (pdhStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		return NULL;
	}
	pdhStatus = PdhOpenQuery(NULL, 0, &hQuery);
	if (pdhStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		return NULL;
	}
	pdhStatus = PdhAddCounter(hQuery, szCounter, 0, &hCounter);
	if (pdhStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		pdhStatus = PdhCloseQuery(hQuery);
		return NULL;
	}
	pdhStatus = PdhCollectQueryData(hQuery);
	if (pdhStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		PdhRemoveCounter(hCounter);
		pdhStatus = PdhCloseQuery(hQuery);
		return NULL;
	}
	Sleep(100);
	pdhStatus = PdhCollectQueryData(hQuery);
	if (pdhStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		PdhRemoveCounter(hCounter);
		pdhStatus = PdhCloseQuery(hQuery);
		return NULL;
	}
	pdhStatus = PdhGetFormattedCounterValue(hCounter, PDH_FMT_DOUBLE, (LPDWORD)NULL, &cValue);
	if (pdhStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		PdhRemoveCounter(hCounter);
		pdhStatus = PdhCloseQuery(hQuery);
		return NULL;
	}
	if (cValue.CStatus != ERROR_SUCCESS) {
		mir_free(szCounter);
		PdhRemoveCounter(hCounter);
		pdhStatus = PdhCloseQuery(hQuery);
		return NULL;
	}
	mir_sntprintf(szVal, SIZEOF(szVal), _T("%.0f"), cValue.doubleValue);
	//PdhRemoveCounter(*hCounter);
	PdhCloseQuery(hQuery);
	mir_free(szCounter);

	return mir_tstrdup(szVal);
}
#endif

static TCHAR *parseCurrentDate(ARGUMENTSINFO *ai)
{
	TCHAR *szFormat;
	if (ai->argc == 1 || (ai->argc > 1 && _tcslen(ai->targv[1]) == 0))
		szFormat = NULL;
	else
		szFormat = ai->targv[1];

	int len = GetDateFormat(LOCALE_USER_DEFAULT, 0, NULL, szFormat, NULL, 0);
	TCHAR *res = (TCHAR*)mir_alloc((len + 1)*sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	if (GetDateFormat(LOCALE_USER_DEFAULT, 0, NULL, szFormat, res, len) == 0) {
		mir_free(res);
		return NULL;
	}

	return res;
}

static TCHAR *parseCurrentTime(ARGUMENTSINFO *ai)
{
	TCHAR *szFormat;
	if (ai->argc == 1 || (ai->argc > 1) && (_tcslen(ai->targv[1]) == 0))
		szFormat = NULL;
	else
		szFormat = ai->targv[1];

	int len = GetTimeFormat(LOCALE_USER_DEFAULT, 0, NULL, szFormat, NULL, 0);
	TCHAR *res = (TCHAR*)mir_alloc((len + 1)*sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	if (GetTimeFormat(LOCALE_USER_DEFAULT, 0, NULL, szFormat, res, len) == 0) {
		mir_free(res);
		return NULL;
	}

	return res;
}

static TCHAR *parseDirectory(ARGUMENTSINFO *ai)
{
	if (ai->argc < 2 || ai->argc > 3)
		return NULL;

	int depth = 0;
	if (ai->argc == 3)
		depth = ttoi(ai->targv[2]);

	if (depth <= 0)
		return mir_tstrdup(ai->targv[1]);

	size_t bi, ei;
	for (ei = 0; ei < _tcslen(ai->targv[1]); ei++) {
		if (ai->targv[1][ei] == '\\')
			depth--;
		if (!depth)
			break;
	}
	if (ei >= _tcslen(ai->targv[1]))
		return ai->targv[1];

	for (bi = ei - 1; bi > 0; bi--)
	if (ai->targv[1][bi - 1] == '\\')
		break;

	TCHAR *res = (TCHAR*)mir_alloc((ei - bi + 1) * sizeof(TCHAR));
	_tcsncpy(res, ai->targv[1] + bi, ei - bi);
	res[ei - bi] = 0;
	return res;
}

/*
	path, depth
	returns complete path up to "path depth - depth"
	*/
static TCHAR *parseDirectory2(ARGUMENTSINFO *ai)
{
	if (ai->argc < 2 || ai->argc > 3)
		return NULL;

	int depth = 1;
	if (ai->argc == 3)
		depth = ttoi(ai->targv[2]);

	if (depth <= 0)
		return NULL;

	TCHAR *ecur = ai->targv[1] + _tcslen(ai->targv[1]);
	while (depth > 0) {
		while ((*ecur != '\\') && (ecur > ai->targv[1]))
			ecur--;

		if (*ecur != '\\')
			return NULL;

		depth -= 1;
		ecur--;
	}
	TCHAR *res = (TCHAR*)mir_calloc((ecur - ai->targv[1] + 2) * sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	_tcsncpy(res, ai->targv[1], (ecur - ai->targv[1]) + 1);
	return res;
}

static int getTime(TCHAR *szTime, struct tm *date)
{
	// do some extra checks here
	TCHAR *cur = szTime;
	if (cur >= szTime + _tcslen(szTime))
		return -1;

	date->tm_mon = _tcstoul(cur, &cur, 10) - 1;
	cur++;
	if (cur >= szTime + _tcslen(szTime))
		return -1;

	date->tm_mday = _tcstoul(cur, &cur, 10);
	cur++;
	if (cur >= szTime + _tcslen(szTime))
		return -1;

	date->tm_year = _tcstoul(cur, &cur, 10);
	if (date->tm_year > 2000)
		date->tm_year -= 2000;
	else if (date->tm_year > 1900)
		date->tm_year -= 1900;

	date->tm_year = date->tm_year < 38 ? date->tm_year + 100 : date->tm_year;
	cur++;
	if (cur >= szTime + _tcslen(szTime))
		return -1;

	date->tm_hour = _tcstoul(cur, &cur, 10);
	cur++;
	if (cur >= szTime + _tcslen(szTime))
		return -1;

	date->tm_min = _tcstoul(cur, &cur, 10);
	cur++;
	if (cur >= szTime + _tcslen(szTime))
		return -1;

	date->tm_sec = _tcstoul(cur, &cur, 10);
	return 0;
}

static TCHAR *parseDiffTime(ARGUMENTSINFO *ai)
{
	struct tm t0, t1;
	TCHAR szTime[32];
	double diff;

	if (ai->argc != 3)
		return NULL;

	memset(&t0, 0, sizeof(t0));
	memset(&t1, 0, sizeof(t1));
	if (getTime(ai->targv[1], &t0) != 0)
		return NULL;

	if (getTime(ai->targv[2], &t1) != 0)
		return NULL;

	diff = difftime(mktime(&t1), mktime(&t0));
	mir_sntprintf(szTime, SIZEOF(szTime), _T("%.0f"), diff);

	return mir_tstrdup(szTime);
}

static TCHAR *parseDirExists(ARGUMENTSINFO *ai)
{
	if (ai->argc != 2)
		return NULL;

	HANDLE hFile = CreateFile(ai->targv[1], GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
	if (hFile == INVALID_HANDLE_VALUE)
		ai->flags |= AIF_FALSE;
	else
		CloseHandle(hFile);

	return mir_tstrdup(_T(""));
}

static TCHAR *parseEnvironmentVariable(ARGUMENTSINFO *ai)
{
	if (ai->argc != 2)
		return NULL;

	DWORD len = ExpandEnvironmentStrings(ai->targv[1], NULL, 0);
	if (len <= 0)
		return NULL;

	TCHAR *res = (TCHAR*)mir_alloc((len + 1)*sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	memset(res, 0, ((len + 1) * sizeof(TCHAR)));
	if (ExpandEnvironmentStrings(ai->targv[1], res, len) == 0) {
		mir_free(res);
		return NULL;
	}
	return res;
}

static TCHAR *parseFileExists(ARGUMENTSINFO *ai)
{
	if (ai->argc != 2)
		return NULL;

	HANDLE hFile = CreateFile(ai->targv[1], GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
	if (hFile == INVALID_HANDLE_VALUE)
		ai->flags |= AIF_FALSE;
	else
		CloseHandle(hFile);

	return mir_tstrdup(_T(""));
}

static TCHAR *parseFindWindow(ARGUMENTSINFO *ai)
{
	if (ai->argc != 2)
		return NULL;

	HWND hWin = FindWindow(ai->targv[1], NULL);
	if (hWin == NULL)
		return NULL;

	int len = GetWindowTextLength(hWin);
	if (len == 0)
		return NULL;

	TCHAR *res = (TCHAR*)mir_alloc((len + 1)*sizeof(TCHAR));
	memset(res, 0, ((len + 1) * sizeof(TCHAR)));
	GetWindowText(hWin, res, len + 1);
	return res;
}

// 1 = dir
// 2 = filter
// 3 = sperator
// 4 = [f]iles, [d]irs
static TCHAR *parseListDir(ARGUMENTSINFO *ai)
{
	if (ai->argc < 2)
		return NULL;

	TCHAR tszFirst[MAX_PATH], *tszRes, *tszSeperator, *tszFilter;
	tszFirst[0] = 0;
	tszSeperator = _T("\r\n");
	tszFilter = _T("*");
	tszRes = NULL;

	if (ai->argc > 1)
		_tcsncpy(tszFirst, ai->targv[1], SIZEOF(tszFirst) - 1);

	if (ai->argc > 2)
		tszFilter = ai->targv[2];

	if (ai->argc > 3)
		tszSeperator = ai->targv[3];

	BOOL bFiles = TRUE, bDirs = TRUE;
	if (ai->argc > 4) {
		if (*ai->targv[4] == 'f')
			bDirs = FALSE;
		if (*ai->targv[4] == 'd')
			bFiles = FALSE;
	}
	if (tszFirst[_tcslen(tszFirst) - 1] == '\\')
		_tcsncat(tszFirst, tszFilter, SIZEOF(tszFirst) - _tcslen(tszFirst) - 1);
	else {
		_tcsncat(tszFirst, _T("\\"), SIZEOF(tszFirst) - _tcslen(tszFirst) - 1);
		_tcsncat(tszFirst, tszFilter, SIZEOF(tszFirst) - _tcslen(tszFirst) - 1);
	}

	WIN32_FIND_DATA ffd;
	HANDLE hFind = FindFirstFile(tszFirst, &ffd);
	if (hFind == INVALID_HANDLE_VALUE) {
		return NULL;
	}
	if (((ffd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY) && (bDirs)) || ((!(ffd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)) && (bFiles))) {
		tszRes = (TCHAR*)mir_alloc((_tcslen(ffd.cFileName) + _tcslen(tszSeperator) + 1)*sizeof(TCHAR));
		_tcscpy(tszRes, ffd.cFileName);
	}
	while (FindNextFile(hFind, &ffd) != 0) {
		if (((ffd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY) && (bDirs)) || ((!(ffd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)) && (bFiles))) {
			if (tszRes != NULL) {
				_tcscat(tszRes, tszSeperator);
				tszRes = (TCHAR*)mir_realloc(tszRes, (_tcslen(tszRes) + _tcslen(ffd.cFileName) + _tcslen(tszSeperator) + 1)*sizeof(TCHAR));
			}
			else {
				tszRes = (TCHAR*)mir_alloc((_tcslen(ffd.cFileName) + _tcslen(tszSeperator) + 1)*sizeof(TCHAR));
				_tcscpy(tszRes, _T(""));
			}
			_tcscat(tszRes, ffd.cFileName);
		}
	}
	FindClose(hFind);

	return tszRes;
}

#ifndef WINE
static BOOL CALLBACK MyProcessEnumerator(DWORD dwPID, WORD wTask, char *szProcess, LPARAM lParam)
{
	char **szProc = (char **)lParam;
	if ((*szProc != NULL) && (!_stricmp(*szProc, szProcess)))
		*szProc = NULL;

	return TRUE;
}

static TCHAR *parseProcessRunning(ARGUMENTSINFO *ai)
{
	if (ai->argc != 2)
		return NULL;

	char *szProc, *ref;
	szProc = ref = mir_u2a(ai->targv[1]);

	EnumProcs(MyProcessEnumerator, (LPARAM)&szProc);
	if (szProc != NULL)
		ai->flags |= AIF_FALSE;

	mir_free(ref);
	return mir_tstrdup(_T(""));
}
#endif

static TCHAR *parseRegistryValue(ARGUMENTSINFO *ai)
{
	if (ai->argc != 3)
		return NULL;

	DWORD len, type;

	TCHAR *key = mir_tstrdup(ai->targv[1]);
	if (key == NULL)
		return NULL;

	TCHAR *cur = _tcschr(key, '\\');
	if (cur == NULL) {
		mir_free(key);
		return NULL;
	}

	*cur = 0;
	HKEY hKey;
	if (!_tcscmp(key, _T("HKEY_CLASSES_ROOT")))
		hKey = HKEY_CLASSES_ROOT;
	else if (!_tcscmp(key, _T("HKEY_CURRENT_USER")))
		hKey = HKEY_CURRENT_USER;
	else if (!_tcscmp(key, _T("HKEY_LOCAL_MACHINE")))
		hKey = HKEY_LOCAL_MACHINE;
	else if (!_tcscmp(key, _T("HKEY_USERS")))
		hKey = HKEY_USERS;
	else {
		mir_free(key);
		return NULL;
	}
	TCHAR *subKey = cur + 1;
	if (RegOpenKeyEx(hKey, subKey, 0, KEY_READ, &hKey) != ERROR_SUCCESS) {
		mir_free(key);
		return NULL;
	}
	mir_free(key);
	len = MAX_REGVALUE_LENGTH + 1;
	TCHAR *res = (TCHAR*)mir_alloc(len*sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	memset(res, 0, (len * sizeof(TCHAR)));
	int err = RegQueryValueEx(hKey, ai->targv[2], NULL, &type, (BYTE*)res, &len);
	if ((err != ERROR_SUCCESS) || (type != REG_SZ)) {
		RegCloseKey(hKey);
		mir_free(res);
		return NULL;
	}
	RegCloseKey(hKey);

	return res;
}

static int TsToSystemTime(SYSTEMTIME *sysTime, time_t timestamp)
{
	struct tm *pTime = localtime(&timestamp);
	if (pTime == NULL)
		return -1;

	memset(sysTime, 0, sizeof(SYSTEMTIME));
	sysTime->wDay = pTime->tm_mday;
	sysTime->wDayOfWeek = pTime->tm_wday;
	sysTime->wHour = pTime->tm_hour;
	sysTime->wMinute = pTime->tm_min;
	sysTime->wMonth = pTime->tm_mon + 1;
	sysTime->wSecond = pTime->tm_sec;
	sysTime->wYear = pTime->tm_year + 1900;

	return 0;
}

static TCHAR *parseTimestamp2Date(ARGUMENTSINFO *ai)
{
	if (ai->argc <= 1)
		return NULL;

	SYSTEMTIME sysTime;
	TCHAR *szFormat;
	time_t timestamp = ttoi(ai->targv[1]);
	if (timestamp == 0)
		return NULL;

	if ((ai->argc == 2) || ((ai->argc > 2) && (_tcslen(ai->targv[2]) == 0)))
		szFormat = NULL;
	else
		szFormat = ai->targv[2];

	if (TsToSystemTime(&sysTime, timestamp) != 0)
		return NULL;

	int len = GetDateFormat(LOCALE_USER_DEFAULT, 0, &sysTime, szFormat, NULL, 0);
	TCHAR *res = (TCHAR*)mir_calloc((len + 1) * sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	if (GetDateFormat(LOCALE_USER_DEFAULT, 0, &sysTime, szFormat, res, len) == 0) {
		mir_free(res);
		return NULL;
	}

	return res;
}

static TCHAR *parseTimestamp2Time(ARGUMENTSINFO *ai)
{
	if (ai->argc <= 1)
		return NULL;

	SYSTEMTIME sysTime;
	time_t timestamp = ttoi(ai->targv[1]);
	if (timestamp == 0)
		return NULL;

	TCHAR *szFormat;
	if ((ai->argc == 2) || ((ai->argc > 2) && (_tcslen(ai->targv[2]) == 0)))
		szFormat = NULL;
	else
		szFormat = ai->targv[2];

	if (TsToSystemTime(&sysTime, timestamp) != 0)
		return NULL;

	int len = GetTimeFormat(LOCALE_USER_DEFAULT, 0, &sysTime, szFormat, NULL, 0);
	TCHAR *res = (TCHAR*)mir_alloc((len + 1)*sizeof(TCHAR));
	if (res == NULL)
		return NULL;

	if (GetTimeFormat(LOCALE_USER_DEFAULT, 0, &sysTime, szFormat, res, len) == 0) {
		mir_free(res);
		return NULL;
	}

	return res;
}

static TCHAR *parseTextFile(ARGUMENTSINFO *ai)
{
	if (ai->argc != 3)
		return NULL;

	HANDLE hFile = CreateFile(ai->targv[1], GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
	if (hFile == INVALID_HANDLE_VALUE)
		return NULL;

	DWORD fileSz = GetFileSize(hFile, NULL);
	if (fileSz == INVALID_FILE_SIZE) {
		CloseHandle(hFile);
		return NULL;
	}

	int lineNo = ttoi(ai->targv[2]);
	int lineCount, csz;
	unsigned int icur, bufSz;
	DWORD readSz, totalReadSz;
	unsigned long linePos;
	TCHAR tUC, *res;
	BYTE *pBuf, *pCur;
	ReadFile(hFile, &tUC, sizeof(TCHAR), &readSz, NULL);
	if (tUC != (TCHAR)0xFEFF) {
		tUC = 0;
		csz = sizeof(char);
		SetFilePointer(hFile, 0, NULL, FILE_BEGIN);
	}
	else csz = sizeof(TCHAR);

	totalReadSz = 0;
	lineCount = 1;
	if (*ai->targv[2] == '0') {
		// complete file
		bufSz = fileSz + csz;
		pBuf = (PBYTE)mir_calloc(bufSz);
		if (pBuf == NULL) {
			CloseHandle(hFile);
			return NULL;
		}

		if (ReadFile(hFile, pBuf, bufSz - csz, &readSz, NULL) == 0) {
			CloseHandle(hFile);
			mir_free(pBuf);
			return NULL;
		}
		CloseHandle(hFile);

		if (tUC)
			res = (TCHAR*)pBuf;
		else {
			res = mir_a2t((char *)pBuf);
			mir_free(pBuf);
		}


		return res;
	}
	bufSz = TXTFILEBUFSZ*csz;
	pBuf = (PBYTE)mir_calloc(bufSz);
	if (pBuf == NULL) {
		CloseHandle(hFile);
		return NULL;
	}

	// count number of lines
	do {
		memset(pBuf, 0, bufSz);
		if (ReadFile(hFile, pBuf, bufSz - csz, &readSz, NULL) == 0) {
			CloseHandle(hFile);
			mir_free(pBuf);
			return NULL;
		}
		totalReadSz += readSz;
		for (pCur = pBuf; *pCur != '\0'; pCur += csz) {
			if (tUC) {
				if (!_tcsncmp((TCHAR*)pCur, _T("\r\n"), 2)) {
					lineCount++;
					pCur += csz;
				}
				else if (*(TCHAR*)pCur == '\n')
					lineCount++;
			}
			else {
				if (!strncmp((char *)pCur, "\r\n", 2)) {
					lineCount++;
					pCur += csz;
				}
				else if (*(char *)pCur == '\n')
					lineCount++;
			}
		}
	}
	while ((totalReadSz < fileSz) && (readSz > 0));

	if (lineNo < 0)
		lineNo = lineCount + lineNo + 1;
	else if (*ai->targv[2] == 'r')
		lineNo = (rand() % lineCount) + 1;

	totalReadSz = 0;
	lineCount = 1;
	linePos = 0xFFFFFFFF;
	SetFilePointer(hFile, tUC ? csz : 0, NULL, FILE_BEGIN);
	// find the position in the file where the requested line starts
	do {
		if (ReadFile(hFile, pBuf, bufSz - csz, &readSz, NULL) == 0) {
			CloseHandle(hFile);
			return NULL;
		}
		totalReadSz += readSz;
		for (pCur = pBuf; ((pCur < pBuf + bufSz) && (linePos == 0xFFFFFFFF)); pCur += csz) {
			if (lineCount == lineNo)
				linePos = (tUC ? csz : 0) + totalReadSz - readSz + pCur - pBuf;

			if (tUC) {
				if (!_tcsncmp((TCHAR*)pCur, _T("\r\n"), 2)) {
					lineCount++;
					pCur += csz;
				}
				else if (*(TCHAR*)pCur == '\n')
					lineCount++;
			}
			else {
				if (!strncmp((char *)pCur, "\r\n", 2)) {
					lineCount++;
					pCur += csz;
				}
				else if (*(char *)pCur == '\n')
					lineCount++;
			}
		}
		if (((tUC) && (*(TCHAR*)pCur == '\r')) || ((!tUC) && (*(char *)pCur == '\r'))) {
			// in case the \r was at the end of the buffer, \n could be next
			SetFilePointer(hFile, -1 * csz, NULL, FILE_CURRENT);
			totalReadSz -= csz;
		}
	}
	while ((totalReadSz < fileSz) && (readSz > 0));

	if (linePos < 0) {
		CloseHandle(hFile);
		mir_free(pBuf);
		return NULL;
	}
	if (SetFilePointer(hFile, linePos, NULL, FILE_BEGIN) != linePos) {
		CloseHandle(hFile);
		mir_free(pBuf);
		return NULL;
	}
	memset(pBuf, 0, bufSz);
	pCur = pBuf;
	do {
		icur = 0;
		if (ReadFile(hFile, pBuf, bufSz - csz, &readSz, NULL) == 0) {
			mir_free(pBuf);
			CloseHandle(hFile);
			return NULL;
		}
		for (pCur = pBuf; (pCur < pBuf + readSz); pCur += csz) {
			if ((tUC) && ((!_tcsncmp((TCHAR*)pCur, _T("\r\n"), 2)) || (*(TCHAR*)pCur == '\n')) ||
				((!tUC) && (((!strncmp((char *)pCur, "\r\n", 2)) || (*(char *)pCur == '\n'))))) {
				CloseHandle(hFile);
				if (tUC)
					*(TCHAR*)pCur = 0;
				else
					*(char *)pCur = '\0';

				if (tUC)
					res = (TCHAR*)pBuf;
				else {
					res = mir_a2t((char *)pBuf);
					mir_free(pBuf);
				}

				return res;
			}
		}
		if (((DWORD)(linePos + (pCur - pBuf)) == fileSz)) { // eof
			CloseHandle(hFile);

			if (tUC) {
				res = (TCHAR*)pBuf;
			}
			else {
				res = mir_a2t((char *)pBuf);
				mir_free(pBuf);
			}

			return res;
		}
		if (readSz == bufSz - csz) {
			// buffer must be increased
			bufSz += TXTFILEBUFSZ*csz;
			if (((tUC) && (*(TCHAR*)pCur == '\r')) || ((!tUC) && (*(char *)pCur == '\r'))) {
				pCur -= csz;
			}
			icur = pCur - pBuf;
			pBuf = (PBYTE)mir_realloc(pBuf, bufSz);
			pCur = pBuf + icur;
			memset((pCur + 1), 0, (TXTFILEBUFSZ * csz));
		}
	}
	while (readSz > 0);
	CloseHandle(hFile);

	return NULL;
}

#if _WIN32_WINNT>=0x0500
static TCHAR *parseUpTime(ARGUMENTSINFO *ai)
{
	if (ai->argc != 1)
		return NULL;

	HQUERY hQuery;
	PDH_STATUS pdhStatus = PdhOpenQuery(NULL, 0, &hQuery);
	if (pdhStatus != ERROR_SUCCESS)
		return NULL;

	HCOUNTER hCounter;
	pdhStatus = PdhAddCounter(hQuery, _T("\\System\\System Up Time"), 0, &hCounter);
	if (pdhStatus != ERROR_SUCCESS) {
		PdhCloseQuery(hQuery);
		return NULL;
	}
	pdhStatus = PdhCollectQueryData(hQuery);
	if (pdhStatus != ERROR_SUCCESS) {
		PdhRemoveCounter(hCounter);
		PdhCloseQuery(hQuery);
		return NULL;
	}

	PDH_FMT_COUNTERVALUE cValue;
	pdhStatus = PdhGetFormattedCounterValue(hCounter, PDH_FMT_LARGE, (LPDWORD)NULL, &cValue);
	if (pdhStatus != ERROR_SUCCESS) {
		PdhRemoveCounter(hCounter);
		PdhCloseQuery(hQuery);
		return NULL;
	}

	TCHAR szVal[32];
	mir_sntprintf(szVal, SIZEOF(szVal), _T("%u"), cValue.largeValue);
	PdhRemoveCounter(hCounter);
	PdhCloseQuery(hQuery);
	return mir_tstrdup(szVal);
}
#endif

static TCHAR *parseUserName(ARGUMENTSINFO *ai)
{
	if (ai->argc != 1)
		return NULL;

	ai->flags |= AIF_DONTPARSE;
	DWORD len = UNLEN;
	TCHAR *res = (TCHAR*)mir_alloc(len + 1);
	if (res == NULL)
		return NULL;

	memset(res, 0, (len + 1));
	if (!GetUserName(res, &len)) {
		mir_free(res);
		return NULL;
	}
	return res;
}

// clipboard support
static TCHAR *parseClipboard(ARGUMENTSINFO *ai)
{
	if (ai->argc != 1)
		return NULL;

	ai->flags |= AIF_DONTPARSE;

	TCHAR *res = NULL;

	if (IsClipboardFormatAvailable(CF_TEXT)) {
		if (OpenClipboard(NULL)) {
			HANDLE hData = GetClipboardData(CF_UNICODETEXT);
			if (hData != NULL) {
				TCHAR *tszText = (TCHAR*)GlobalLock(hData);
				size_t len = _tcslen(tszText);
				res = (TCHAR*)mir_alloc((len + 1) * sizeof(TCHAR));
				_tcscpy(res, tszText);
				res[len] = 0;
				GlobalUnlock(hData);
			}
			CloseClipboard();
		}
	}

	return res;
}

void registerSystemTokens()
{
	registerIntToken(COMPUTERNAME, parseComputerName, TRF_FIELD, LPGEN("System Functions")"\t"LPGEN("computer name"));
#if _WIN32_WINNT>=0x0500
	registerIntToken(CPULOAD, parseCpuLoad, TRF_FUNCTION, LPGEN("System Functions")"\t(x)\t"LPGEN("CPU load of process x (without extension) (x is optional)"));
#endif
	registerIntToken(CDATE, parseCurrentDate, TRF_FUNCTION, LPGEN("System Functions")"\t(y)\t"LPGEN("current date in format y (y is optional)"));
	registerIntToken(CTIME, parseCurrentTime, TRF_FUNCTION, LPGEN("System Functions")"\t(y)\t"LPGEN("current time in format y (y is optional)"));
	registerIntToken(DIRECTORY, parseDirectory, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("the directory y directories above x"));
	registerIntToken(DIRECTORY2, parseDirectory2, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("strips y directories from x"));
	registerIntToken(DIFFTIME, parseDiffTime, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("number of seconds between date x and y (x and y in format: M/d/yy H:m:s)"));
	registerIntToken(DIREXISTS, parseDirExists, TRF_FUNCTION, LPGEN("System Functions")"\t(x)\t"LPGEN("TRUE if directory x exists"));
	registerIntToken(FILEEXISTS, parseFileExists, TRF_FUNCTION, LPGEN("System Functions")"\t(x)\t"LPGEN("TRUE if file x exists"));
	registerIntToken(FINDWINDOW, parseFindWindow, TRF_FUNCTION, LPGEN("System Functions")"\t(x)\t"LPGEN("window title of first window of class x"));
	registerIntToken(LISTDIR, parseListDir, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y,z)\t"LPGEN("shows files and directories of directory x, with filter y, separated by z (y and z optional)"));
#ifndef WINE
	registerIntToken(PROCESSRUNNING, parseProcessRunning, TRF_FUNCTION, LPGEN("System Functions")"\t(x)\t"LPGEN("TRUE if process x is running"));
#endif
	registerIntToken(REGISTRYVALUE, parseRegistryValue, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("value y from registry key x (REG_SZ (string) values only)"));
	registerIntToken(TIMESTAMP2DATE, parseTimestamp2Date, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("formats timestamp x (seconds since 1/1/1970) in date format y"));
	registerIntToken(TIMESTAMP2TIME, parseTimestamp2Time, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("formats timestamp x (seconds since 1/1/1970) in time format y"));
	registerIntToken(TXTFILE, parseTextFile, TRF_FUNCTION, LPGEN("System Functions")"\t(x,y)\t"LPGEN("y > 0: line number y from file x, y = 0: the whole file, y < 0: line y counted from the end, y = r: random line"));
#if _WIN32_WINNT>=0x0500
	registerIntToken(UPTIME, parseUpTime, TRF_FIELD, LPGEN("System Functions")"\t"LPGEN("uptime in seconds"));
#endif
	if (!ServiceExists(MS_UTILS_REPLACEVARS))
		registerIntToken(ENVIRONMENTVARIABLE, parseEnvironmentVariable, TRF_FUNCTION, LPGEN("Miranda Core OS")"\t(%xxxxxxx%)\t"LPGEN("any environment variable defined in current Windows session (like %systemroot%, %allusersprofile%, etc.)"));
	else {
		registerIntToken(ENVIRONMENTVARIABLE, parseEnvironmentVariable, TRF_FUNCTION, LPGEN("System Functions")"\t(x)\t"LPGEN("expand environment variable x"));
		registerIntToken(USERNAME, parseUserName, TRF_FIELD, LPGEN("System Functions")"\t"LPGEN("user name"));
	}

	srand((unsigned int)GetTickCount());

	registerIntToken(CLIPBOARD, parseClipboard, TRF_FIELD, LPGEN("System Functions")"\t"LPGEN("text from clipboard"));
}