Tencent2016API.cpp

#include "stdafx.h"
#include <iostream>
#include <string>
#include <winsock2.h>
#include <windows.h>
#include <cstdlib>
#include <nb30.h>
#include "direct.h"
#include <time.h>
#include <comdef.h>
#include <Wbemidl.h>
#include <conio.h>
#include "atlstr.h"
#include "atlbase.h"

using namespace std;

#pragma comment(lib,"Netapi32.lib")
# pragma comment(lib, "wbemuuid.lib")
# pragma comment(lib, "ws2_32.lib")

//不弹框的system，返回执行结果
BOOL system_hide(string CommandLine, string &exe_result)
{       
	HANDLE hRead, hWrite;
    SECURITY_ATTRIBUTES sa;  
	sa.bInheritHandle = TRUE;
    sa.nLength = sizeof(SECURITY_ATTRIBUTES);      
    sa.lpSecurityDescriptor = NULL;     
    if (!CreatePipe(&hRead,&hWrite,&sa,0))      
    {     
        return FALSE;   
    }       
    STARTUPINFOA si;       
    PROCESS_INFORMATION pi;        
    si.cb = sizeof(STARTUPINFO);     
    GetStartupInfoA(&si);       
    si.hStdError = hWrite;     
    si.hStdOutput = hWrite;     
    si.wShowWindow = SW_HIDE;      
    si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;     
    if (!CreateProcessA(NULL, (char *)CommandLine.c_str(), NULL, NULL, TRUE, NULL, NULL, NULL, &si, &pi))       
    {     
		return FALSE;     
    }     
    CloseHandle(hWrite);     
	DWORD bytesRead;
    char buffer[1024] = {0};     	
    while(true)       		
    {     		
        memset(buffer,0,strlen(buffer));  
        if(ReadFile(hRead,buffer,1023,&bytesRead,NULL)==NULL)     			
            break;     
		exe_result += buffer;     		
    }       
    return TRUE;      
}

typedef struct _ASTAT_
{
	ADAPTER_STATUS adapt;
	NAME_BUFFER NameBuff [30];
} ASTAT, *PASTAT;

//获取当前主机第一张网卡的MAC地址
void get_3part_mac(string &mac)
{
	NCB Ncb;
	ASTAT Adapter;
	UCHAR uRetCode;
	LANA_ENUM lenum;
	memset(&Ncb, 0, sizeof(Ncb));
	Ncb.ncb_command = NCBENUM;
	Ncb.ncb_buffer = (UCHAR *)&lenum;
	Ncb.ncb_length = sizeof(lenum);
	uRetCode = Netbios( &Ncb );
	for(int i=0; i < lenum.length ; i++)
	{
		memset(&Ncb, 0, sizeof(Ncb));
		Ncb.ncb_command = NCBRESET;
		Ncb.ncb_lana_num = lenum.lana[i];
		uRetCode = Netbios( &Ncb );
		memset(&Ncb, 0, sizeof(Ncb));
		Ncb.ncb_command = NCBASTAT;
		Ncb.ncb_lana_num = lenum.lana[i];
		strcpy((char *)Ncb.ncb_callname,"*");
		Ncb.ncb_buffer = (unsigned char *) &Adapter;
		Ncb.ncb_length = sizeof(Adapter);
		uRetCode = Netbios( &Ncb );
		if (uRetCode == 0)
		{
			char tmp[128];
			sprintf(tmp,"%02x-%02x-%02x",
				Adapter.adapt.adapter_address[0],
				Adapter.adapt.adapter_address[1],
				Adapter.adapt.adapter_address[2]
				);
			mac = tmp;
		}
	}
}

//通过WMI获取主机信息
BOOL ManageWMIInfo(string &result, string table, wstring wcol)
{	
	HRESULT hres;
	char bord[1024];
	//初始化COM 
	hres = CoInitialize(0);
	//获得WMI连接COM接口  
	IWbemLocator *pLoc = NULL;
	hres = CoCreateInstance(
		CLSID_WbemLocator, 
		0, 
		CLSCTX_INPROC_SERVER, 
		IID_IWbemLocator, (LPVOID *) &pLoc);
	if (FAILED(hres))
	{
		cout << "Failed to create IWbemLocator object."
			<< "Err code = 0x"
			<< hex << hres << endl;
		CoUninitialize();
		return false; 
	}
	//通过连接接口连接WMI的内核对象名ROOT//CIMV2  
	IWbemServices *pSvc = NULL;
	hres = pLoc->ConnectServer(
		_bstr_t(L"ROOT\\CIMV2"), // Object path of WMI namespace
		NULL, // User name. NULL = current user
		NULL, // User password. NULL = current
		0, // Locale. NULL indicates current
		NULL, // Security flags.
		0, // Authority (e.g. Kerberos)
		0, // Context object 
		&pSvc // pointer to IWbemServices proxy
		);
	if (FAILED(hres))
	{
		cout << "Could not connect. Error code = 0x"
			<< hex << hres << endl;
		pLoc->Release(); 
		CoUninitialize();
		return false; 
	}
	//设置请求代理的安全级别   
	hres = CoSetProxyBlanket(
		pSvc, // Indicates the proxy to set
		RPC_C_AUTHN_WINNT, // RPC_C_AUTHN_xxx
		RPC_C_AUTHZ_NONE, // RPC_C_AUTHZ_xxx
		NULL, // Server principal name 
		RPC_C_AUTHN_LEVEL_CALL, // RPC_C_AUTHN_LEVEL_xxx 
		RPC_C_IMP_LEVEL_IMPERSONATE, // RPC_C_IMP_LEVEL_xxx
		NULL, // client identity
		EOAC_NONE // proxy capabilities 
		);
	if (FAILED(hres))
	{
		cout << "Could not set proxy blanket. Error code = 0x"
			<< hex << hres << endl;
		pSvc->Release();
		pLoc->Release(); 
		CoUninitialize();
		return false; 
	}
	//通过请求代理来向WMI发送请求
	IEnumWbemClassObject* pEnumerator = NULL;
	string select = "SELECT * FROM "+ table;
	hres = pSvc->ExecQuery(
		bstr_t("WQL"), 	
		bstr_t(select.c_str()),
		WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, 
		NULL,
		&pEnumerator);
	if (FAILED(hres))
	{
		cout << "Query for Network Adapter Configuration failed."
			<< " Error code = 0x”"
			<< hex << hres << endl;
		pSvc->Release();
		pLoc->Release();
		CoUninitialize();
		return false; 
	}
	//循环枚举所有的结果对象
	ULONG uReturn = 0;
	IWbemClassObject *pclsObj;
	while (pEnumerator)
	{
		HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1, 
			&pclsObj, &uReturn);
		if(0 == uReturn)
		{
			break;
		}
		VARIANT vtProp;
		VariantInit(&vtProp);
		hr = pclsObj->Get(wcol.c_str(), 0, &vtProp, 0, 0);
		if(!FAILED(hr))
		{
			CW2A tmpstr1(vtProp.bstrVal);
			strcpy_s(bord,200,tmpstr1);
			result = bord;
		}
		VariantClear(&vtProp);
	}
	//释放资源  
	pSvc->Release();
	pLoc->Release();
	pEnumerator->Release();
	pclsObj->Release();
	CoUninitialize();
	return true; 
}

//识别Virtual PC异常
DWORD IslnsideVPC_exceptionFilter(LPEXCEPTION_POINTERS ep)
{
	PCONTEXT ctx=ep->ContextRecord;
	ctx->Ebx = -1; //未运行在VPC中
	ctx->Eip += 4; //跳过”call VPC”操作
	return EXCEPTION_CONTINUE_EXECUTION;
}