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

void ErrorHandler(const char* message)
{
   printf("failed: %s\n", message);
   exit(1);
}

int main(int argc, char* argv[])
{
   if (argc < 2)
   {
     printf("Usage: SetTimestamp <file name>...\n");
     exit(1);
   }

   for (int i=1; i<argc; i++)
   {
     printf("Setting timestamp for file \"%s\"...  ", argv[i]);

     HGLOBAL hResLoad;   // handle to loaded resource
     HMODULE hExe;       // handle to existing .EXE file
     HRSRC hRes;         // handle/ptr. to res. info. in hExe
     HANDLE hUpdateRes;  // update resource handle
     LPVOID lpResLock;   // pointer to resource data
     BOOL result;

     // Load the .EXE file that contains the dialog box you want to copy.
     hExe = LoadLibraryEx(argv[i], NULL, DONT_RESOLVE_DLL_REFERENCES);
     if (hExe == NULL)
     {
         ErrorHandler(TEXT("Could not load exe."));
     }

     // Locate the dialog box resource in the .EXE file.
     hRes = FindResource(hExe, MAKEINTRESOURCE(VS_VERSION_INFO), RT_VERSION);
     if (hRes == NULL)
     {
         ErrorHandler(TEXT("Could not locate version resource."));
     }

     // Load the dialog box into global memory.
     hResLoad = LoadResource(hExe, hRes);
     if (hResLoad == NULL)
     {
         ErrorHandler(TEXT("Could not load resource."));
     }

     // Lock the dialog box into global memory.
     lpResLock = LockResource(hResLoad);
     if (lpResLock == NULL)
     {
         ErrorHandler(TEXT("Could not lock resource."));
     }

     VS_FIXEDFILEINFO* info;
     UINT len;
     if (VerQueryValue(lpResLock, "\\", reinterpret_cast<void**>(&info), &len) == 0)
     {
       ErrorHandler(TEXT("Could not query fixed version info"));
     }

     if (info->dwSignature != 0XFEEF04BDL)
     {
       ErrorHandler(TEXT("Signature check failed"));
     }

     FILETIME time;
     GetSystemTimeAsFileTime(&time);

     info->dwFileDateLS = time.dwLowDateTime;
     info->dwFileDateMS = time.dwHighDateTime;

     DWORD ResSize = SizeofResource(hExe, hRes);

     LPVOID NewBuf = malloc(ResSize);
     memcpy(NewBuf, lpResLock, ResSize);

     // Close file to let it be opened for write
     if (!FreeLibrary(hExe))
     {
         ErrorHandler(TEXT("Could not free executable."));
     }

     // Open the file to which you want to add the dialog box resource.
     hUpdateRes = BeginUpdateResource(argv[i], FALSE);
     if (hUpdateRes == NULL)
     {
         ErrorHandler(TEXT("Could not open file for writing."));
     }

     // Add new version resource to the update list.
     result = UpdateResource(hUpdateRes,
         RT_VERSION,
         MAKEINTRESOURCE(VS_VERSION_INFO),
         MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
         NewBuf,                         // ptr to resource info
         ResSize);       // size of resource info

     if (result == FALSE)
     {
         ErrorHandler(TEXT("Could not add resource."));
     }

     // Write changes to FOOT.EXE and then close it.
     if (!EndUpdateResource(hUpdateRes, FALSE))
     {
         ErrorHandler(TEXT("Could not write changes to file."));
     }

     free(NewBuf);

     printf("success.\n");
   }

}