N-InjectLib and N-CodeHook simple example

#include "NCodeHookInstantiation.h"
 
NCodeHookIA32 nCodeHook;
 
#define USER32 "user32.dll"
 
// Without defining at least one export function injection doesn't work
extern "C" __declspec(dllexport) void  tryMe(); 
 
__declspec(dllexport) void tryMe() {
    // empty body
}
 
typedef BOOL (WINAPI *SetWindowTextWPtr) (HWND hWnd,LPCWSTR lpOutputString);
SetWindowTextWPtr origSetWindowTextW = NULL;
 
BOOL WINAPI SetWindowTextWHook(HWND hWnd, LPCWSTR lpString)
{
    return origSetWindowTextW(hWnd, L"***");
}
 
 
BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
        origSetWindowTextW = nCodeHook.createHookByName(USER32, "SetWindowTextW", SetWindowTextWHook);
        break;
 
    case DLL_THREAD_ATTACH:
        break;
 
    case DLL_THREAD_DETACH:
        break;
 
    case DLL_PROCESS_DETACH:
        break;
    }
    return TRUE;
}

#include "Process.h"
#include "IATModifier.h"
#include <windows.h>
 
int main (int argc, char* argv[]) {
 
    STARTUPINFO         siStartupInfo;
    PROCESS_INFORMATION piProcessInfo;
 
    char fullPathToProcess[] = "c:\\windows\\system32\\calc.exe";
    memset(&siStartupInfo, 0, sizeof(siStartupInfo));
    memset(&piProcessInfo, 0, sizeof(piProcessInfo));
 
    siStartupInfo.cb = sizeof(siStartupInfo);
 
    if(CreateProcess(fullPathToProcess,     // Application name
        NULL,                 // Application arguments
        0,
        0,
        FALSE,
        CREATE_DEFAULT_ERROR_MODE | CREATE_SUSPENDED,
        0,
        0,                              // Working directory
        &siStartupInfo,
        &piProcessInfo) == FALSE) {
            // something bad happend
            return 1;
    }
 
    DWORD process_id = piProcessInfo.dwProcessId;         
    Process process(process_id);
    IATModifier iatMod(process, true);    
    iatMod.writeIAT("CodeHook.dll");
 
    ResumeThread(piProcessInfo.hThread);
    WaitForSingleObject(piProcessInfo.hProcess, INFINITE);
 
}