用c++实现对可执行程序的图标修改

http://blog.sina.com.cn/s/blog_406be8d10100l963.html

#include <stdio.h>
#include <windows.h>
#include <tchar.h>

struct ICONDIRENTRY {
	BYTE bWidth;
	BYTE bHeight;
	BYTE bColorCount;
	BYTE bReserved;
	WORD wPlanes;
	WORD wBitCount;
	DWORD dwBytesInRes;
	DWORD dwImageOffset;
};

struct ICONDIR {
	WORD idReserved;
	WORD idType;
	WORD idCount;
//ICONDIRENTRY idEntries;
};

struct GRPICONDIRENTRY {
	BYTE bWidth;
	BYTE bHeight;
	BYTE bColorCount;
	BYTE bReserved;
	WORD wPlanes;
	WORD wBitCount;
	DWORD dwBytesInRes;
	WORD nID;
};

struct GRPICONDIR {
	WORD idReserved;
	WORD idType;
	WORD idCount;
	GRPICONDIRENTRY idEntries;
};

//
//函数说明：修改EXE图标
//
//参    数：IconFile 图标文件
//              ExeFile 被修改的EXE文件
//
//返回值： 成功为True，否则False
/
bool ChangeExeIcon(LPCSTR IconFile, LPCSTR ExeFile) {
	ICONDIR stID;
	ICONDIRENTRY stIDE;
	GRPICONDIR stGID;
	HANDLE hFile;
	DWORD nSize, nGSize, dwReserved;
	HANDLE hUpdate;
	PBYTE pIcon, pGrpIcon;
	BOOL ret;

	hFile = CreateFile(IconFile, GENERIC_READ, NULL, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
	if (hFile == INVALID_HANDLE_VALUE) {
		return false;
	}

	ZeroMemory(&stID, sizeof(ICONDIR));
	ret = ReadFile(hFile, &stID, sizeof(ICONDIR), &dwReserved, NULL);

	ZeroMemory(&stIDE, sizeof(ICONDIRENTRY));
	ret = ReadFile(hFile, &stIDE, sizeof(ICONDIRENTRY), &dwReserved, NULL);

	nSize = stIDE.dwBytesInRes;
	pIcon = (PBYTE) malloc(nSize);
	SetFilePointer(hFile, stIDE.dwImageOffset, NULL, FILE_BEGIN);
	ret = ReadFile(hFile, (LPVOID) pIcon, nSize, &dwReserved, NULL);
	if (!ret) {
		CloseHandle(hFile);
		return false;
	}

	ZeroMemory(&stGID, sizeof(GRPICONDIR));
	stGID.idCount = stID.idCount;
	stGID.idReserved = 0;
	stGID.idType = 1;
	CopyMemory(&stGID.idEntries, &stIDE, 12);
	stGID.idEntries.nID = 0;

	nGSize = sizeof(GRPICONDIR);
	pGrpIcon = (PBYTE) malloc(nGSize);
	CopyMemory(pGrpIcon, &stGID, nGSize);

	hUpdate = BeginUpdateResource(ExeFile, false);
	ret = UpdateResource(hUpdate, RT_GROUP_ICON, MAKEINTRESOURCE(1), 0, (LPVOID) pGrpIcon, nGSize);
	ret = UpdateResource(hUpdate, RT_ICON, MAKEINTRESOURCE(1), 0, (LPVOID) pIcon, nSize);
	EndUpdateResource(hUpdate, false);
	if (!ret) {
		CloseHandle(hFile);
		return false;
	}

	CloseHandle(hFile);
	return true;
}

int main() {
	LPCSTR IconFile, ExeFile;
	char IconFilePath[100], ExeFilePath[100];
	printf("路径中的“\\”请用“\\\\”代替\n请输入图标文件路径：\n");
	scanf("%s", IconFilePath);
	printf("请输入Exe文件路径：\n");
	scanf("%s", ExeFilePath);
	IconFile = LPCSTR(&IconFilePath);
	ExeFile = LPCSTR(&ExeFilePath);

	ChangeExeIcon(IconFile, ExeFile);
	return 0;
}