Dynamic Libraries with Delayed Function Loading

#include <windows.h>

#include "macros.h"
#include "hooks.h"

#include "delayedfunc.h"

const char *called_proc = NULL;

FARPROC rtn_addr;

DWORD eax_save;
DWORD ebx_save;
DWORD ecx_save;
DWORD edx_save;
DWORD esi_save;
DWORD edi_save;
DWORD esp_save;
DWORD ebp_save;


__declspec(naked)
void proxy_proc_LoadLibrary()
{
	SAVE_RTN_ADDRESS
	SAVE_REGISTERS
	
	{
		HMODULE lib = LoadLibrary("payload_dll.dll");
		if (lib)
		{
			FARPROC payload_proc = GetProcAddress(lib, called_proc);
			if (payload_proc)
			{
				hook_proc_start();

				LOAD_REGISTERS
				__asm
				{
					add esp, 4
					call payload_proc
					sub esp, 4
				}
				SAVE_REGISTERS

				hook_proc_end();
			}

			FreeLibrary(lib);
			lib = NULL;
		}
	}

	LOAD_RTN_ADDRESS
	LOAD_REGISTERS
	__asm ret
}

__declspec(naked)
void proxy_proc()
{
	SAVE_RTN_ADDRESS
	SAVE_REGISTERS
	
	{
		FARPROC payload_proc = Delayed_Function_Load("payload_dll.dll", called_proc);
		if (payload_proc)
		{
			hook_proc_start();

			LOAD_REGISTERS
			__asm
			{
				add esp, 4
				call payload_proc
				sub esp, 4
			}
			SAVE_REGISTERS

			hook_proc_end();
		}
		Delayed_Function_Free(payload_proc);
		// Instead of freeing the function now, you can
		// substitute the JMP address in the export table
		// of the function in proxy_dll with payload_proc.
		// The memory can be freed when proxy_dll is unloaded.
	}

	LOAD_RTN_ADDRESS
	LOAD_REGISTERS
	__asm ret
}

#include "exports.inl"