/*-======================- Copyright (c) 2007 -======================-

程序作者:

    Hacker Ans : 駭客任務組Ans軟件工作室
聯系方式:

    QQ:360038060
	[oopfans]群:13161288


修改時間:

    2007-03-09

功能描述:
    系統相關信息

版權聲明:
    許可任何單位,個人隨意使用,拷貝,修改,散布及出售這份代碼,但是必須保
    留此版權信息,以慰藉作者辛勤的勞動,及表明此代碼的來源,如若此份代碼
    有任何BUG,請通知作者,以便彌補作者由于水平所限而導致的一些錯誤和不
    足,謝謝!

-======================- Copyright (c) 2006 -======================-*/

#ifndef _ZSYSINFO
#define _ZSYSINFO

#pragma warning(disable:4786) 
#include <string>
#include <deque>
using namespace std;

#include "ZThunk.h"
#include "Tlhelp32.h"

#include <winsock2.h>
#pragma comment(lib,"WS2_32.lib")

class ZSysInfo  
{
public:

	ZSysInfo();
	~ZSysInfo();
private:

	//此對像用于將回調函數封裝成類成員函數
	ZThunk m_thunk;

/*-------------------------------------------------------------------------*/
private:

	SYSTEMTIME m_SYSTEMTIME;

public:

	//獲取年,月,日,時,分,秒,毫秒(以整型值返回)
	int GetYear();
	int GetMonth();
	int GetDay();
	int GetHour();
	int GetMinute();
	int GetSecond();
	int GetMillisecond();

	/*
	獲取年,月,日,時,分,秒,毫秒(以字符串值返回)
	FillZero:如果不足十是否為零填充
	*/
	string GetYear_str();
	string GetMonth_str(bool FillZero=true);
	string GetDay_str(bool FillZero=true);
	string GetHour_str(bool FillZero=true);
	string GetMinute_str(bool FillZero=true);
	string GetSecond_str(bool FillZero=true);
	string GetMillisecond_str(bool FillZero=true);

/*-------------------------------------------------------------------------*/

	/*
	獲取操作系統版本
	返回值:
		  0 未知
		  1 win 98
		  2 win Me
		  3 win NT 3.51
		  4 win NT 4.0
		  5 win 2000
		  6 win xp
		  7 win 2003
	*/
	static int GetOSVersion();

/*-------------------------------------------------------------------------*/
private:

	//本機名稱
	string m_LocalhostName;

public:

	//獲取本機IP列表
	deque<string> GetLocalhostIP();

	//獲取本機名稱
	string GetLocalhostName();

/*-------------------------------------------------------------------------*/

public:

	//獲取進程進息
	deque<PROCESSENTRY32> GetProcessInfo();
	
	//獲取線程信息
	deque<THREADENTRY32> GetThreadInfo(unsigned int Pid);

	//根據進程名獲取進程ID
	deque<int> GetProcessID(char* szProcessName);

	//根據PID獲取進程名稱
	string GetProcessName(int ProcessID);

	/*
	根據線程ID獲取線程句柄,調用失敗返回0xFFFFFFFF
	win98的Kernel32.dll不含有OpenThread函數,
	故此函數不能應用在win98環境下
	*/
	HANDLE ZSysInfo::GetThreadHandle(unsigned int ThreadID);

private:

	HWND m_HWND;

	typedef struct _CtrlInfo
	{ 
	  long CtrlID; 
	  long CtrlStyle; 
	} CTRLINFO,*PCTRLINFO;

private:

	//枚舉子窗口回調函數
	bool EnumChildProc(HWND hwnd,LPARAM lParam);

public:

	//根據父句柄及控件ID獲取句柄,沒找到則返回0xFFFFFFFF
	HWND FindChildWindow(HWND hParent,long CtrlID,long CtrlStyle);
};

#endif