/*
请使用16位色或以下的普通图标,否则颜色失真。
另外，Exe文件改图标后，可能在该文件夹下看不出来，
这是因为有图标缓存，移动到其他文件夹即可。
或者删除图标缓存（磁盘清理）
*/

#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);

//
//函数说明：修改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;
}