win_portable.h

//--------------------------------------------------------------------
// 文件名:		win_portable.h
// 内  容:		
// 说  明:		
// 创建日期:	2010年12月14日
// 创建人:		陆利民
// 版权所有:	苏州蜗牛电子有限公司
//--------------------------------------------------------------------

#ifndef _SYSTEM_WIN_PORTABLE_H
#define _SYSTEM_WIN_PORTABLE_H

#include <windows.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <float.h>
#include <math.h>
#include <objbase.h>

// 日期和时间
struct port_date_time_t
{
	int nYear;
	int nMonth;
	int nDay;
	int nHour;
	int nMinute;
	int nSecond;
	int nMilliseconds;
	int nDayOfWeek;
};

// 内存使用情况
struct port_memory_info_t
{
	double dMemoryLoad;
	double dTotalMemory;
	double dAvailMemory;
};

// 获得本地当前时间
inline bool Port_GetLocalTime(port_date_time_t* dt)
{
	SYSTEMTIME st;
	
	GetLocalTime(&st);
	
	dt->nYear = st.wYear;
	dt->nMonth = st.wMonth;
	dt->nDay = st.wDay;
	dt->nHour = st.wHour;
	dt->nMinute = st.wMinute;
	dt->nSecond = st.wSecond;
	dt->nMilliseconds = st.wMilliseconds;
	dt->nDayOfWeek = st.wDayOfWeek;
	
	return true;
}

// 获得系统计时
inline unsigned int Port_GetTickCount()
{
	return GetTickCount();
}

// 获得当前高精度时刻
inline double Port_GetPerformanceTime()
{
	static LARGE_INTEGER s_u64Freq;
	static BOOL s_init = QueryPerformanceFrequency(&s_u64Freq);

	LARGE_INTEGER u64Tick;

	QueryPerformanceCounter(&u64Tick);

	return (double)u64Tick.QuadPart / (double)s_u64Freq.QuadPart;
}

// 系统休眠
inline void Port_Sleep(unsigned int ms)
{
	Sleep(ms);
}

// 是否非法的浮点数
inline int Port_IsNaN(double value)
{
	return _isnan(value);
}

// 创建唯一标识符
inline const char* Port_CreateGuid(char* buffer, size_t size)
{
	GUID id;
	
	if (CoCreateGuid(&id) != S_OK)
	{
		return NULL;
	}
	
	_snprintf(buffer, size - 1, 
		"%08X%04X%04X%02X%02X%02X%02X%02X%02X%02X%02X",
		id.Data1, id.Data2, id.Data3, 
		id.Data4[0], id.Data4[1], id.Data4[2], id.Data4[3],
		id.Data4[4], id.Data4[5], id.Data4[6], id.Data4[7]);
	buffer[size - 1] = 0;
		
	return buffer;
}

// 弹出信息窗口
inline void Port_PopupMsgBox(const char* info, const char* title)
{
	MessageBoxA(NULL, info, title, MB_OK);
}

// 弹出错误信息窗口
inline void Port_PopupErrorMsgBox(const char* info, const char* title)
{
	MessageBoxA(NULL, info, title, MB_OK | MB_ICONERROR);
}

// 获得当前的进程ID
inline unsigned int Port_GetCurrentProcessId()
{
	return GetCurrentProcessId();
}

// 获得当前执行文件的路径
inline bool Port_GetCurrentModuleFileName(char* buffer, size_t size)
{
	if (GetModuleFileNameA(NULL, buffer, (DWORD)size) == 0)
	{
		buffer[0] = 0;
		return false;
	}
	
	return true;
}

// 终止进程
inline bool Port_KillProcess(unsigned int proc_id)
{
	HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, proc_id);
	
	if (NULL == hProcess)
	{
		return false;
	}
	
	if (!TerminateProcess(hProcess, 0))
	{
		CloseHandle(hProcess);
		return false;
	}
	
	CloseHandle(hProcess);
	
	return true;
}

// 获得内存使用情况
inline bool Port_GetMemoryInfo(port_memory_info_t* pInfo)
{
	memset(pInfo, 0, sizeof(port_memory_info_t));
	
	MEMORYSTATUS ms;

	GlobalMemoryStatus(&ms);
	
	pInfo->dMemoryLoad = (double)ms.dwMemoryLoad / 100.0;
	pInfo->dTotalMemory = (double)ms.dwTotalPhys;
	pInfo->dAvailMemory = (double)ms.dwAvailPhys;
	
	return true;
}

// 获得CPU使用率
inline double Port_GetCpuRate()
{
	// 无法直接获取
	return 0.0;
}

// 初始化守护进程
inline bool Port_InitDaemon()
{
	return true;
}

// 启动控制台程序
inline bool Port_Execute(const char* cmd_line)
{
	char cmd[256];
	size_t size = strlen(cmd_line) + 1;
	
	if (size > sizeof(cmd))
	{
		return false;
	} 
	
	memcpy(cmd, cmd_line, size);
	
	STARTUPINFOA si;
	PROCESS_INFORMATION pi;
	
	memset(&si, 0, sizeof(si));
	si.cb = sizeof(si);
	memset(&pi, 0, sizeof(pi));
	
	BOOL res = CreateProcessA(NULL, cmd, NULL, NULL, FALSE, CREATE_NEW_CONSOLE, 
		NULL, NULL, &si, &pi);

	if (res)
	{
		// 释放资源
		CloseHandle(pi.hThread);
		CloseHandle(pi.hProcess);
	}
	
	return res == TRUE;
}

// 设置当前进程与CPU核心的亲和性
inline bool Port_SetProcessAffinity(int core_id)
{
	if ((core_id < 0) || (core_id >= 64))
	{
		return false;
	}
	
	return SetProcessAffinityMask(GetCurrentProcess(), 0x1 << core_id) == TRUE;
}

#endif // _SYSTEM_WIN_PORTABLE_H