// .PRO
#-------------------------------------------------
#
# Project created by QtCreator 2013-11-28T16:52:43
#
#-------------------------------------------------

QT       += core gui
QT       += webkit
QT += webkitwidgets

greaterThan(QT_MAJOR_VERSION, 4): QT += widgets

TARGET = Clandestine
TEMPLATE = app


SOURCES += main.cpp\
        mainwindow.cpp \
    IconChanger.cpp

HEADERS  += mainwindow.h \
            IconChanger.h \
            Windows.h

FORMS    += mainwindow.ui

RESOURCES += \
    res.qrc



// IconChanger.h

#ifndef ICONCHANGER_H
#define ICONCHANGER_H

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

 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;

 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;

class IconChanger {

public:

    // Members
    HRSRC hrResSource;
    HMODULE hExeSource;
    HGLOBAL hGlobSource;

    char *cFilePathSource;
    DWORD dwSizeOfSource;
    char* lpIconBuffer;

    // Methods
    IconChanger();
    ~IconChanger();

    void GetIconFromSource(QString sIconPath, QString sExePath);
    void ChangeExeIcon(LPCTSTR lpIconFile, LPCTSTR lpExeName);

private:


};

#endif // ICONCHANGER_H


// IconChanger.h //
// IconChanger.cpp

#include "IconChanger.h"

#include <QMessageBox>

BOOL ResNames(HMODULE hModule, LPCTSTR lpType, LPTSTR lpName, LONG lParam);
BOOL ResTypes(HMODULE hModule, LPTSTR lpType, LONG lParam);

IconChanger::IconChanger() // Constructor
{
    hExeSource = NULL;
    hrResSource = NULL;
    hGlobSource = NULL;
    lpIconBuffer = NULL;
}
                                                            // Retrive Icon Buffer
void IconChanger::GetIconFromSource(QString sIconPath, QString sExePath) // from source Exe/DLL
{
    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);

    LoadLibraryExA( sExePath.toStdString().c_str(), &hExeSource, LOAD_LIBRARY_AS_DATAFILE );
    if ( hExeSource == NULL )
    {
        QMessageBox mess; mess.setText("Unable to get handle to EXE/DLL" + sExePath );
        mess.exec();
    }

    hIconFile = CreateFileA( sIconPath.toStdString().c_str(), GENERIC_READ,
                            0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if ( hIconFile == INVALID_HANDLE_VALUE )
    {
        QMessageBox mess; mess.setText("Unable to open file " + sIconPath );
        mess.exec();
    }

    ret = ReadFile( hExeSource, &header, sizeof(HEADER), &dwReserved, NULL );
    if (!ret)
    {
        CloseHandle(hExeSource);
        QMessageBox mess; mess.setText("Unable to Read Icon File " + sIconPath );
        mess.exec();
        return;
    }

    pIconDirEntry = (LPICONDIRENTRY)new BYTE[ header.idCount * sizeof(ICONDIRENTRY) ];
    if ( pIconDirEntry == NULL )
    {
        CloseHandle(hIconFile);
        QMessageBox mess; mess.setText("Unable to allocate memory for Icon Group Header" + sIconPath );
        mess.exec();
        return;
    }

    ret = ReadFile(hIconFile, pIconDirEntry, header.idCount*sizeof(ICONDIRENTRY), &dwReserved, NULL);
    if (!ret)
    {
        delete[] pIconDirEntry;
        CloseHandle(hIconFile);
        QMessageBox mess; mess.setText("Unable to read Icon Dir Entrys" + sIconPath );
        mess.exec();
        return;
    }

    nGSize = sizeof(GRPICONDIR)+header.idCount*sizeof(ICONDIRENTRY);
    pGrpIconDir = (LPGRPICONDIR)new BYTE[nGSize];
    ZeroMemory( pGrpIconDir, nSize );

    pGrpIconDir->idReserved = header.idReserved;
    pGrpIconDir->idType = header.idType;
    pGrpIconDir->idCount = header.idCount;

    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+1;

        EnumResourceTypesW( hExeSource, (ENUMRESTYPEPROC)ResTypes, 0);
    }

    CloseHandle(hIconFile);
}

IconChanger::~IconChanger() // De-Constructor
{

}

void IconChanger::ChangeExeIcon(LPCTSTR lpIconFile, LPCTSTR lpExeName)
{
    /*
     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);

     pGrpIconDir->idReserved=header.idReserved;
     pGrpIconDir->idType=header.idType;
     pGrpIconDir->idCount=header.idCount;

     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+1;        }

     hUpdate = BeginUpdateResource(lpExeName, false);
     if (hUpdate!=NULL)
     {
         HMODULE hMe= 0; // means load from this module

         if (FindResource( hMe, lpExeName, 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;
             }
         }


         for(i=0;i<header.idCount;i++)
         {

             nSize = pIconDirEntry[i].dwBytesInRes;

             dwRet=SetFilePointer(hIconFile, pIconDirEntry[i].dwImageOffset, NULL, FILE_BEGIN);
             if (dwRet==INVALID_SET_FILE_POINTER)
             {
                 break;
             }

             delete[] pIconBytes;

             pIconBytes = new BYTE[nSize];
             ZeroMemory(pIconBytes, nSize);
             ret = ReadFile(hIconFile, (LPVOID)pIconBytes, nSize, &dwReserved, NULL);
             if(!ret)
             {
                 break;
             }

             ret = UpdateResource(hUpdate, RT_ICON, MAKEINTRESOURCE(pGrpIconDir->idEntries[i].nID), MAKELANGID(LANG_CHINESE, SUBLANG_SYS_DEFAULT), (LPVOID)pIconBytes, nSize);
             if(!ret)
             {
                 break;
             }
         }

         if (pIconBytes!=NULL)
         {
            delete[] pIconBytes;
         }

         EndUpdateResource(hUpdate, false);
     }


     //SAFE_ARRAY_DELETE(pGrpIconDir);
     //SAFE_ARRAY_DELETE(pIconDirEntry);
     CloseHandle(hIconFile);

     */
 }

BOOL ResTypes(HMODULE hModule, LPTSTR lpType, LONG lParam)
{
    if (!IS_INTRESOURCE(lpType))
    {
        if ( lpType == RT_GROUP_ICON )
        {
            EnumResourceNames(hModule, RT_ICON, (ENUMRESNAMEPROC)ResNames, 0);
            return true;
        }

    } else {
        if ( lpType == RT_GROUP_ICON )
        {
            EnumResourceNames(hModule, RT_ICON, (ENUMRESNAMEPROC)ResNames, 0);
            return true;
        }
    }

   return true;
}

BOOL ResNames(HMODULE hModule, LPCTSTR lpType, LPTSTR lpName, LONG lParam)
{
    if (!IS_INTRESOURCE(lpName))
    {

    } else {

    }

   return true;
}