#include "sdk.h"

#if defined(__unix__)
#include <sys/time.h>
#include <unistd.h>
#endif

#ifdef VMP_GNU
#elif defined(WIN_DRIVER)
void DriverUnload(PDRIVER_OBJECT driver_object)
{

}

NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObject, PUNICODE_STRING pRegistryPath)
{ 
	pRegistryPath;
	pDriverObject->DriverUnload = DriverUnload;
	return STATUS_SUCCESS;
}
#else
BOOL APIENTRY DllMain(HANDLE hModule, DWORD dwReason, LPVOID lpReserved)
{
	hModule;
	dwReason;
	lpReserved;
	return TRUE;
}
#endif

bool VMP_API VMProtectIsProtected()
{
	return false;
}

void VMP_API VMProtectBegin(const char *) 
{

}

void VMP_API VMProtectBeginMutation(const char *)
{

}

void VMP_API VMProtectBeginVirtualization(const char *) 
{

}

void VMP_API VMProtectBeginUltra(const char *) 
{ 

}

void VMP_API VMProtectBeginVirtualizationLockByKey(const char *) 
{ 

}

void VMP_API VMProtectBeginUltraLockByKey(const char *) 
{ 

}

void VMP_API VMProtectEnd() 
{

}

bool VMP_API VMProtectIsDebuggerPresent(bool) 
{
#ifdef VMP_GNU
	return false;
#elif defined(WIN_DRIVER)
	return false;
#else
	return IsDebuggerPresent() != FALSE; 
#endif
}

bool VMP_API VMProtectIsVirtualMachinePresent() 
{ 
	return false; 
}

bool VMP_API VMProtectIsValidImageCRC()
{ 
	return true; 
}

const char * VMP_API VMProtectDecryptStringA(const char *value) 
{ 
	return value; 
}

const VMP_WCHAR * VMP_API VMProtectDecryptStringW(const VMP_WCHAR *value) 
{ 
	return value; 
}

bool VMP_API VMProtectFreeString(void *) 
{ 
	return true; 
}

int VMP_API VMProtectGetOfflineActivationString(const char *, char *, int) 
{
	return ACTIVATION_OK;
}

int VMP_API VMProtectGetOfflineDeactivationString(const char *, char *, int) 
{
	return ACTIVATION_OK;
}

#ifdef __APPLE__
unsigned long GetTickCount()
{
	const int64_t one_million = 1000 * 1000;
	mach_timebase_info_data_t timebase_info;
	mach_timebase_info(&timebase_info);

	// mach_absolute_time() returns billionth of seconds,
	// so divide by one million to get milliseconds
	return static_cast<uint32_t>((mach_absolute_time() * timebase_info.numer) / (one_million * timebase_info.denom));
}
#endif

#ifdef __unix__
unsigned long GetTickCount()
{
    struct timeval tv;
    gettimeofday(&tv, NULL);
    return tv.tv_sec * 1000 + tv.tv_usec / 1000;
}
#endif

#ifndef WIN_DRIVER
bool g_serial_is_correct = false;
bool g_serial_is_blacklisted = false;
uint32_t g_time_of_start = GetTickCount();
#endif

#ifdef VMP_GNU

#define strcmpi strcasecmp
#define INI_MAX_LINE 1024

size_t strnlen(char *text, size_t maxlen)
{
	const char *last = (const char *)memchr(text, '\0', maxlen);
	return last ? (size_t) (last - text) : maxlen;
}

/* Strip whitespace chars off end of given string, in place. Return s. */
static char* rstrip(char* s)
{
	char* p = s + strlen(s);
	while (p > s && isspace((unsigned char)(*--p)))
		*p = '\0';
	return s;
}

/* Return pointer to first non-whitespace char in given string. */
static char* lskip(const char* s)
{
	while (*s && isspace((unsigned char)(*s)))
		s++;
	return (char*)s;
}

/* Return pointer to first char c or ';' comment in given string, or pointer to
   null at end of string if neither found. ';' must be prefixed by a whitespace
   character to register as a comment. */
static char* find_char_or_comment(const char* s, char c)
{
	int was_whitespace = 0;
	while (*s && *s != c && !(was_whitespace && *s == ';')) {
		was_whitespace = isspace((unsigned char)(*s));
		s++;
	}
	return (char*)s;
}

/* See documentation in header file. */
static int GetPrivateProfileString(const char *section_name, const char *key_name, char *buffer, size_t size, const char *file_name)
{
	if (!buffer || !size)
		return 0;

    FILE* file = fopen(file_name, "r");
    if (!file)
        return 0;

    char line[INI_MAX_LINE];
    char* start;
    char* end;
    char* name;
    char* value;
    int lineno = 0;
	int res = 0;
	bool section_found = false;

    /* Scan through file line by line */
    while (fgets(line, INI_MAX_LINE, file) != NULL) {
        lineno++;

        start = line;
        if (lineno == 1 && (unsigned char)start[0] == 0xEF &&
                           (unsigned char)start[1] == 0xBB &&
                           (unsigned char)start[2] == 0xBF) {
            start += 3;
        }
        start = lskip(rstrip(start));

        if (*start == ';' || *start == '#') {
            /* Per Python ConfigParser, allow '#' comments at start of line */
        } else if (*start == '[') {
            /* A "[section]" line */
            end = find_char_or_comment(start + 1, ']');
            if (*end == ']') {
                *end = '\0';
				if (section_found)
					break;

				section_found = strcmpi(start + 1, section_name) == 0;
            }
        } else if (section_found && *start && *start != ';') {
            /* Not a comment, must be a name[=:]value pair */
            end = find_char_or_comment(start, '=');
            if (*end != '=') {
                end = find_char_or_comment(start, ':');
            }
            if (*end == '=' || *end == ':') {
                *end = '\0';
                name = rstrip(start);
                value = lskip(end + 1);
                end = find_char_or_comment(value, '\0');
                if (*end == ';')
                    *end = '\0';
                rstrip(value);

				if (strcmpi(name, key_name) == 0) {
					strncpy(buffer, value, size);
					res = strnlen(buffer, size);
					break;
				}
            }
        }
    }

	fclose(file);
	return res;
}

static bool GetIniValue(const char *value_name, char *buffer, size_t size)
{
	char file_name[PATH_MAX];
	file_name[0] = 0;
	uint32_t name_size = sizeof(file_name);
#if defined(__APPLE__)
	_NSGetExecutablePath(file_name, &name_size);
#else
	int sz = readlink("/proc/self/exe", file_name, name_size);
	if (sz > 0)
		file_name[sz] = 0;
#endif
	char *p = strrchr(file_name, '/');
	if (p)
		*(p + 1) = 0;
	strncat(file_name, "VMProtectLicense.ini", sizeof(file_name) - (p - file_name));
	return GetPrivateProfileString("TestLicense", value_name, buffer, size, file_name) != 0;
}

static const uint8_t utf8_limits[5] = {0xC0, 0xE0, 0xF0, 0xF8, 0xFC};

static void ConvertUTF8ToUnicode(const uint8_t *src, size_t len, VMP_WCHAR *dest, size_t dest_size)
{
	if (!dest || dest_size == 0) return; // nothing to do

	size_t pos = 0;
	size_t dest_pos = 0;
	while (pos < len && dest_pos < dest_size) {
		uint8_t b = src[pos++];

		if (b < 0x80) {
			dest[dest_pos++] = b;
			continue;
		}

		size_t val_len;
		for (val_len = 0; val_len <= _countof(utf8_limits); val_len++) {
			if (b < utf8_limits[val_len])
				break;
		}

		if (val_len == 0)
			continue;

		uint32_t value = b - utf8_limits[val_len - 1];
		for (size_t i = 0; i < val_len; i++) {
			if (pos == len)
				break;
			b = src[pos++];
			if (b < 0x80 || b >= 0xC0)
				break;
			value <<= 6;
			value |= (b - 0x80);
		}

		if (value < 0x10000) {
			dest[dest_pos++] = static_cast<uint16_t>(value);
		} else if (value <= 0x10FFFF) {
			value -= 0x10000;
			dest[dest_pos++] = static_cast<uint16_t>(0xD800 + (value >> 10));
			dest[dest_pos++] = static_cast<uint16_t>(0xDC00 + (value & 0x3FF));
		}
	}
	
	if (dest_pos < dest_size - 1)
		dest[dest_pos] = 0;
	else
		dest[dest_pos - 1] = 0;
}

static bool GetIniValue(const char *value_name, VMP_WCHAR *buffer, size_t size)
{
	char value[INI_MAX_LINE];
	if (GetIniValue(value_name, value, sizeof(value))) {
		ConvertUTF8ToUnicode(reinterpret_cast<uint8_t *>(value), strlen(value), buffer, size);
		return true;
	}
	if (buffer && size)
		buffer[0] = 0;
	return false;
}

#elif defined(WIN_DRIVER)
#else

static bool GetIniValue(const char *value_name, wchar_t *buffer, size_t size)
{
	wchar_t file_name[MAX_PATH];
	file_name[0] = 0;
	GetModuleFileNameW(NULL, file_name, _countof(file_name));
	wchar_t *p = wcsrchr(file_name, L'\\');
	if (p)
		*(p + 1) = 0;
	wcsncat_s(file_name, L"VMProtectLicense.ini", _countof(file_name));

	wchar_t key_name[1024] = {0};
	MultiByteToWideChar(CP_ACP, 0, value_name, -1, key_name, _countof(key_name));
	return GetPrivateProfileStringW(L"TestLicense", key_name, L"", buffer, static_cast<DWORD>(size), file_name) != 0;
}

static bool GetIniValue(const char *value_name, char *buffer, size_t size)
{
	wchar_t value[2048];
	if (GetIniValue(value_name, value, sizeof(value))) {
		WideCharToMultiByte(CP_ACP, 0, value, -1, buffer, static_cast<int>(size), NULL, NULL);
		return true;
	}
	if (buffer && size)
		buffer[0] = 0;
	return false;
}

#endif

#define MAKEDATE(y, m, d) (DWORD)((y << 16) + (m << 8) + d)

int VMP_API VMProtectGetSerialNumberState()
{
#ifdef WIN_DRIVER
	return SERIAL_STATE_FLAG_INVALID;
#else
	if (!g_serial_is_correct) 
		return SERIAL_STATE_FLAG_INVALID;
	if (g_serial_is_blacklisted) 
		return SERIAL_STATE_FLAG_BLACKLISTED;

	int res = 0;

	char buf[256];
	if (GetIniValue("TimeLimit", buf, sizeof(buf))) {
		int running_time = atoi(buf);
		if (running_time >= 0 && running_time <= 255) {
			uint32_t dw = GetTickCount();
			int d = (dw - g_time_of_start) / 1000 / 60; // minutes
			if (running_time <= d) 
				res |= SERIAL_STATE_FLAG_RUNNING_TIME_OVER;
		}
	}

	if (GetIniValue("ExpDate", buf, sizeof(buf))) {
		int y, m, d;
		if (sscanf_s(buf, "%04d%02d%02d", &y, &m, &d) == 3) {
			uint32_t ini_date = (y << 16) + (static_cast<uint8_t>(m) << 8) + static_cast<uint8_t>(d);
			uint32_t cur_date;
#ifdef VMP_GNU
			time_t rawtime;
			time(&rawtime);
			struct tm local_tm;
			tm *timeinfo = localtime_r(&rawtime, &local_tm);
			cur_date = ((timeinfo->tm_year + 1900) << 16) + (static_cast<uint8_t>(timeinfo->tm_mon + 1) << 8) + static_cast<uint8_t>(timeinfo->tm_mday);
#else
			SYSTEMTIME st;
			GetLocalTime(&st);
			cur_date = (st.wYear << 16) + (static_cast<uint8_t>(st.wMonth) << 8) + static_cast<uint8_t>(st.wDay);
#endif
			if (cur_date > ini_date)
				res |= SERIAL_STATE_FLAG_DATE_EXPIRED;
		}
	}

	if (GetIniValue("MaxBuildDate", buf, sizeof(buf))) {
		int y, m, d;
		if (sscanf_s(buf, "%04d%02d%02d", &y, &m, &d) == 3) {
			uint32_t ini_date = (y << 16) + (static_cast<uint8_t>(m) << 8) + static_cast<uint8_t>(d);
			uint32_t cur_date;
#ifdef VMP_GNU
			time_t rawtime;
			time(&rawtime);
			struct tm local_tm;
			tm *timeinfo = localtime_r(&rawtime, &local_tm);
			cur_date = ((timeinfo->tm_year + 1900) << 16) + (static_cast<uint8_t>(timeinfo->tm_mon + 1) << 8) + static_cast<uint8_t>(timeinfo->tm_mday);
#else
			SYSTEMTIME st;
			GetLocalTime(&st);
			cur_date = (st.wYear << 16) + (static_cast<uint8_t>(st.wMonth) << 8) + static_cast<uint8_t>(st.wDay);
#endif
			if (cur_date > ini_date)
				res |= SERIAL_STATE_FLAG_MAX_BUILD_EXPIRED;
		}
	}

	if (GetIniValue("KeyHWID", buf, sizeof(buf))) {
		char buf2[256];
		GetIniValue("MyHWID", buf2, sizeof(buf2));
		if (strcmp(buf, buf2) != 0) 
			res |= SERIAL_STATE_FLAG_BAD_HWID;
	}

	return res;
#endif
}

int VMP_API VMProtectSetSerialNumber(const char *serial)
{
#ifdef WIN_DRIVER
	serial;
	return SERIAL_STATE_FLAG_INVALID;
#else
	g_serial_is_correct = false;
	g_serial_is_blacklisted = false;
	if (!serial || !serial[0]) 
		return SERIAL_STATE_FLAG_INVALID;

	char buf_serial[2048];
	const char *src = serial;
	char *dst = buf_serial;
	while (*src) {
		char c = *src;
		// check agains base64 alphabet
		if ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || (c >= '0' && c <= '9') || c == '+' || c == '/' || c == '=')
			*dst++ = c;
		src++;
	}
	*dst = 0;

	char ini_serial[2048];
	if (!GetIniValue("AcceptedSerialNumber", ini_serial, sizeof(ini_serial)))
		strcpy_s(ini_serial, "serialnumber");
	g_serial_is_correct = strcmp(buf_serial, ini_serial) == 0;

	if (GetIniValue("BlackListedSerialNumber", ini_serial, sizeof(ini_serial)))
		g_serial_is_blacklisted = strcmp(buf_serial, ini_serial) == 0;

	return VMProtectGetSerialNumberState();
#endif
}

bool VMP_API VMProtectGetSerialNumberData(VMProtectSerialNumberData *data, int size)
{
#ifdef WIN_DRIVER
	data; 
	size;
	return false;
#else
	if (size != sizeof(VMProtectSerialNumberData)) 
		return false;
	memset(data, 0, sizeof(VMProtectSerialNumberData));

	data->nState = VMProtectGetSerialNumberState();
	if (data->nState & (SERIAL_STATE_FLAG_INVALID | SERIAL_STATE_FLAG_BLACKLISTED))
		return true; // do not need to read the rest

	GetIniValue("UserName", data->wUserName, _countof(data->wUserName));
	GetIniValue("EMail", data->wEMail, _countof(data->wEMail));

	char buf[2048];
	if (GetIniValue("TimeLimit", buf, sizeof(buf))) {
		int running_time = atoi(buf);
		if (running_time < 0) 
			running_time = 0;
		else
		if (running_time > 255) 
			running_time = 255;
		data->bRunningTime = static_cast<unsigned char>(running_time);
	}

	if (GetIniValue("ExpDate", buf, sizeof(buf))) {
		int y, m, d;
		if (sscanf_s(buf, "%04d%02d%02d", &y, &m, &d) == 3) {
			data->dtExpire.wYear = static_cast<unsigned short>(y);
			data->dtExpire.bMonth = static_cast<unsigned char>(m);
			data->dtExpire.bDay = static_cast<unsigned char>(d);
		}
	}

	if (GetIniValue("MaxBuildDate", buf, sizeof(buf))) {
		int y, m, d;
		if (sscanf_s(buf, "%04d%02d%02d", &y, &m, &d) == 3) {
			data->dtMaxBuild.wYear = static_cast<unsigned short>(y);
			data->dtMaxBuild.bMonth = static_cast<unsigned char>(m);
			data->dtMaxBuild.bDay = static_cast<unsigned char>(d);
		}
	}

	if (GetIniValue("UserData", buf, sizeof(buf))) {
		size_t len = strlen(buf);
		if (len > 0 && len % 2 == 0 && len <= 255 * 2) // otherwise UserData is empty or has bad length
		{
			for (size_t src = 0, dst = 0; src < len; src++) {
				int v = 0;
				char c = buf[src];

				if (c >= '0' && c <= '9') v = c - '0';
				else if (c >= 'a' && c <= 'f') v = c - 'a' + 10;
				else if (c >= 'A' && c <= 'F') v = c - 'A' + 10;
				else {
					data->nUserDataLength = 0;
					memset(data->bUserData, 0, sizeof(data->bUserData));
					break;
				}

				if (src % 2 == 0) {
					data->bUserData[dst] = static_cast<unsigned char>(v << 4);
				} else {
					data->bUserData[dst] |= v;
					dst++;
					data->nUserDataLength = static_cast<unsigned char>(dst);
				}
			}
		}
	}

	return true;
#endif
}

int VMP_API VMProtectGetCurrentHWID(char *hwid, int size)
{
#ifdef WIN_DRIVER
	hwid;
	size;
	return 0;
#else
	if (hwid && size == 0) 
		return 0;

	char buf[1024];
	if (!GetIniValue("MyHWID", buf, sizeof(buf)))
		strcpy_s(buf, "myhwid");

	int res = static_cast<int>(strlen(buf));
	if (hwid) {
		if (size - 1 < res)
			res = size - 1;
		memcpy(hwid, buf, res);
		hwid[res] = 0;
	}
	return res + 1;
#endif
}

int VMP_API VMProtectActivateLicense(const char *code, char *serial, int size) 
{ 
#ifdef WIN_DRIVER
	code;
	serial;
	size;
	return ACTIVATION_NOT_AVAILABLE;
#else
	if (!code)
		return ACTIVATION_BAD_CODE;

	char buf[2048];
	if (!GetIniValue("AcceptedActivationCode", buf, sizeof(buf)))
		strcpy_s(buf, "activationcode");
	if (strcmp(code, buf) != 0)
		return ACTIVATION_BAD_CODE;

	if (!GetIniValue("AcceptedSerialNumber", buf, sizeof(buf)))
		strcpy_s(buf, "serialnumber");

	int need = static_cast<int>(strlen(buf));
	if (need > size - 1) 
		return ACTIVATION_SMALL_BUFFER;

	strncpy_s(serial, size, buf, _TRUNCATE);
	return ACTIVATION_OK;
#endif
}

int VMP_API VMProtectDeactivateLicense(const char *) 
{ 
#ifdef WIN_DRIVER
	return ACTIVATION_NOT_AVAILABLE;
#else
	return ACTIVATION_OK;
#endif
}