Function GetModuleCodeAddress

CrySystem/DataProbe.cpp:652–714

///////////////////////////////////////////////////////////////////// Get address of executable code RAM for loaded DLL or EXE. /////////////////////////////////////////////////////////////////////

// Get address of executable code RAM for loaded DLL or EXE.
//////////////////////////////////////////////////////////////////////////
static void GetModuleCodeAddress( void *base,void **pCodeStartAddress,int *nCodeSize )
{
	const IMAGE_DOS_HEADER *dos_head = (IMAGE_DOS_HEADER*)base;

	assert(base);
	assert(pCodeStartAddress);
	assert(nCodeSize);

	*pCodeStartAddress = 0;
	*nCodeSize = 0;

#include <pshpack1.h>     // no padding.
	const struct PEHeader
	{
		DWORD signature;
		IMAGE_FILE_HEADER _head;
		IMAGE_OPTIONAL_HEADER opt_head;
		IMAGE_SECTION_HEADER *section_header;  // actual number in NumberOfSections
	}
	*header;
#include <poppack.h>

	if (dos_head->e_magic != IMAGE_DOS_SIGNATURE)
	{
		// Wrong pointer, not to PE header.
		return;
	}                         
	// verify DOS-EXE-Header
	// after end of DOS-EXE-Header: offset to PE-Header
	header = (PEHeader*)(const void *)((char *)dos_head + dos_head->e_lfanew);

	if (IsBadReadPtr(header, sizeof(*header)))  // start of PE-Header
	{
		// (no PE header, probably DOS executable)
		return;
	}

/*
	CryLog( "[CodeModule] base: %x",base );
	CryLog( "[CodeModule] base of code: %x",header->opt_head.BaseOfCode );
	CryLog( "[CodeModule] size of code: %x",header->opt_head.SizeOfCode );
*/

	void *pCodeAddress = (char*)base + header->opt_head.BaseOfCode;
	if (IsBadReadPtr(pCodeAddress, header->opt_head.SizeOfCode ))
	{
		// Something wrong..., cant read code memory, skip it.
		return;
	}
	*pCodeStartAddress = pCodeAddress;
	*nCodeSize = header->opt_head.SizeOfCode;
	
	/*
	{
		int sect;
		const IMAGE_SECTION_HEADER *section_header;
		for (sect = 0, section_header = header->section_header; sect < header->_head.NumberOfSections; sect++, section_header++)
		{