import KernelEx-4.5.2

vxd/pemanip.h

@@ -0,0 +1,91 @@
+/*
+ *  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