mirror of
https://github.com/UzixLS/KernelEx.git
synced 2025-07-19 07:21:20 +03:00
92 lines
2.3 KiB
C++
Executable File
92 lines
2.3 KiB
C++
Executable File
/*
|
|
* KernelEx
|
|
* Copyright (C) 2008-2009, 2011, Xeno86
|
|
*
|
|
* This file is part of KernelEx source code.
|
|
*
|
|
* KernelEx is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published
|
|
* by the Free Software Foundation; version 2 of the License.
|
|
*
|
|
* KernelEx is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with GNU Make; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*
|
|
*/
|
|
|
|
#ifndef _SETUP_PEMANIP_H
|
|
#define _SETUP_PEMANIP_H
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <winnt.h>
|
|
|
|
class PEmanip
|
|
{
|
|
public:
|
|
PEmanip();
|
|
PEmanip(void* memory);
|
|
bool HasTarget();
|
|
bool OpenMemory(void* memory);
|
|
void Close();
|
|
IMAGE_NT_HEADERS* GetPEHeader();
|
|
IMAGE_SECTION_HEADER* RvaToSection(DWORD Rva);
|
|
void* RvaToPointer(DWORD Rva);
|
|
DWORD PointerToRva(void* pointer);
|
|
void* GetSectionByName(const char* name);
|
|
DWORD GetSectionSize(const char* name);
|
|
bool AllocSectionSpace(const char* name, int needed_space, void** ptr, int align = 1);
|
|
DWORD GetExportedAPI(const char* func);
|
|
DWORD GetImageBase();
|
|
void* CreateBackup();
|
|
static unsigned long ALIGN(unsigned long x, unsigned long y);
|
|
|
|
private:
|
|
void ZeroInit();
|
|
IMAGE_SECTION_HEADER* FindSectionByName(const char* name);
|
|
|
|
bool has_target;
|
|
int target_len;
|
|
IMAGE_DOS_HEADER* MZh;
|
|
IMAGE_NT_HEADERS* PEh;
|
|
IMAGE_SECTION_HEADER* section_hdrs;
|
|
DWORD image_base;
|
|
};
|
|
|
|
inline bool PEmanip::HasTarget()
|
|
{
|
|
return has_target;
|
|
}
|
|
|
|
inline IMAGE_NT_HEADERS* PEmanip::GetPEHeader()
|
|
{
|
|
return PEh;
|
|
}
|
|
|
|
inline unsigned long PEmanip::ALIGN(unsigned long x, unsigned long y)
|
|
{
|
|
return (x + y - 1) & (~(y - 1));
|
|
}
|
|
|
|
inline DWORD PEmanip::GetImageBase()
|
|
{
|
|
return image_base;
|
|
}
|
|
|
|
inline void* PEmanip::RvaToPointer(DWORD Rva)
|
|
{
|
|
return (void*)(Rva + image_base);
|
|
}
|
|
|
|
inline DWORD PEmanip::PointerToRva(void* pointer)
|
|
{
|
|
return (DWORD) pointer - image_base;
|
|
}
|
|
|
|
#endif
|