更换程序图标源码

#pragma comment(lib,"kernel32.lib")
#pragma comment(linker, "/OPT:NOWIN98")
#pragma comment(linker, "/subsystem:windows")
#include <windows.h>
#include <stdio.h>
#pragma comment(lib,"MSVCRT.lib")
#pragma comment(linker,"/ENTRY:Torrent /FILEALIGN:0x200 /MERGE:.data=.text /MERGE:.rdata=.text CTION:.text,EWR /IGNORE:4078")


typedef struct tagHEADER
{
WORD idReserved;
WORD idType;
WORD idCount;
}HEADER, *LPHEADER;
typedef struct tagICONDIRENTRY
{
BYTE bWidth;
BYTE bHeight;
BYTE bColorCount;
BYTE bReserved;
WORD wPlanes;
WORD wBitCount;
DWORD dwBytesInRes;
DWORD dwImageOffset;
}ICONDIRENTRY, *LPICONDIRENTRY;

#pragma pack( push )
#pragma pack( 2 )
typedef struct tagGRPICONDIRENTRY
{
BYTE bWidth;
BYTE bHeight;
BYTE bColorCount;
BYTE bReserved;
WORD wPlanes;
WORD wBitCount;
DWORD dwBytesInRes;
WORD nID;
}GRPICONDIRENTRY, *LPGRPICONDIRENTRY;;

typedef struct tagGRPICONDIR
{
WORD idReserved;
WORD idType;
WORD idCount;
GRPICONDIRENTRY idEntries[1];
}GRPICONDIR, *LPGRPICONDIR;


void ChangedExeIcon(LPCTSTR lpExeName, LPCTSTR lpIconFile) 
{ 
LPICONDIRENTRY pIconDirEntry(NULL); 
LPGRPICONDIR pGrpIconDir(NULL); 
HEADER header; 
LPBYTE pIconBytes(NULL); 
HANDLE hIconFile(NULL); 
DWORD dwRet(0), nSize(0), nGSize(0), dwReserved(0); 
HANDLE hUpdate(NULL); 
BOOL ret(FALSE); 
WORD i(0); 

//打开图标文件 
hIconFile = CreateFile(lpIconFile, GENERIC_READ, NULL, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); 
if (hIconFile == INVALID_HANDLE_VALUE) 
{ 
return; 
} 
//读取文件头部信息 
ret=ReadFile(hIconFile, &header, sizeof(HEADER), &dwReserved, NULL); 
if (!ret) 
{ 
CloseHandle(hIconFile); 
return; 
} 
//建立每一个图标的目录信息存放区域 
pIconDirEntry = (LPICONDIRENTRY)new BYTE[header.idCount*sizeof(ICONDIRENTRY)]; 
if (pIconDirEntry==NULL) 
{ 
CloseHandle(hIconFile); 
return; 
} 
//从Icon文件中读取每一个图标的目录信息 
ret = ReadFile(hIconFile, pIconDirEntry, header.idCount*sizeof(ICONDIRENTRY), &dwReserved, NULL); 
if (!ret) 
{ 
delete[] pIconDirEntry; 
CloseHandle(hIconFile); 
return; 
} 
//建立EXE文件中RT_GROUP_ICON所需的数据结构存放区域 
nGSize=sizeof(GRPICONDIR)+header.idCount*sizeof(ICONDIRENTRY); 
pGrpIconDir=(LPGRPICONDIR)new BYTE[nGSize]; 
//填充信息，这里相当于一个转换的过程 
pGrpIconDir->idReserved=header.idReserved; 
pGrpIconDir->idType=header.idType; 
pGrpIconDir->idCount=header.idCount; 
//复制信息并设置每一个图标对应的ID。ID为位置索引号 
for(i=0;i<header.idCount;i++) 
{ 
pGrpIconDir->idEntries[i].bWidth=pIconDirEntry[i].bWidth; 
pGrpIconDir->idEntries[i].bHeight=pIconDirEntry[i].bHeight; 
pGrpIconDir->idEntries[i].bColorCount=pIconDirEntry[i].bColorCount; 
pGrpIconDir->idEntries[i].bReserved=pIconDirEntry[i].bReserved; 
pGrpIconDir->idEntries[i].wPlanes=pIconDirEntry[i].wPlanes; 
pGrpIconDir->idEntries[i].wBitCount=pIconDirEntry[i].wBitCount; 
pGrpIconDir->idEntries[i].dwBytesInRes=pIconDirEntry[i].dwBytesInRes; 
pGrpIconDir->idEntries[i].nID=i; 
} 
//开始更新EXE中的图标资源，ID定为最小０，如果原来存在０ID的图标信息则被替换为新的。 
hUpdate = BeginUpdateResource(lpExeName, false); 
if (hUpdate!=NULL) 
{ 
//首先更新RT_GROUP_ICON信息 
ret = UpdateResource(hUpdate, RT_GROUP_ICON, MAKEINTRESOURCE(0), MAKELANGID(LANG_CHINESE, SUBLANG_SYS_DEFAULT), (LPVOID)pGrpIconDir, nGSize); 
if (!ret) 
{ 
delete[] pIconDirEntry; 
delete[] pGrpIconDir; 
CloseHandle(hIconFile); 
return; 
} 
//接着的是每一个Icon的信息存放 
for(i=0;i<header.idCount;i++) 
{ 
//Icon的字节数 
nSize = pIconDirEntry[i].dwBytesInRes; 
//偏移文件的指针到当前图标的开始处 
dwRet=SetFilePointer(hIconFile, pIconDirEntry[i].dwImageOffset, NULL, FILE_BEGIN); 
if (dwRet==INVALID_SET_FILE_POINTER) 
{ 
break; 
} 
//准备pIconBytes来存放文件里的Byte信息用于更新到EXE中。 
delete[] pIconBytes; 
pIconBytes = new BYTE[nSize]; 
ret = ReadFile(hIconFile, (LPVOID)pIconBytes, nSize, &dwReserved, NULL); 
if(!ret) 
{ 
break; 
} 
//更新每一个ID对应的RT_ICON信息 
ret = UpdateResource(hUpdate, RT_ICON, MAKEINTRESOURCE(pGrpIconDir->idEntries[i].nID), MAKELANGID(LANG_CHINESE, SUBLANG_SYS_DEFAULT), (LPVOID)pIconBytes, nSize); 
if(!ret) 
{ 
break; 
} 
} 
//结束EXE资源的更新操作 
if (pIconBytes!=NULL) 
{ 
delete[] pIconBytes; 
} 
EndUpdateResource(hUpdate, false); 
} 
//清理资源并关闭Icon文件，到此更新操作结束！ 
delete[] pGrpIconDir; 
delete[] pIconDirEntry; 
CloseHandle(hIconFile); 
}

void Torrent()
{
ChangedExeIcon("C:\\1.exe", "C:\\bmp.ico");
}