ldso: added pe_terminate_current_process().

2 files changed, 33 insertions, 0 deletions

diff --git a/src/internal/pe_os.h b/src/internal/pe_os.h
index ba7da0e..449433e 100644
--- a/src/internal/pe_os.h
+++ b/src/internal/pe_os.h
@@ -158,10 +158,16 @@ typedef int32_t __stdcall os_zw_open_file(
 	__in	uint32_t		share_access,
 	__in	uint32_t		open_options);
 
+
 typedef int32_t __stdcall os_ldr_load_dll(
 	__in	wchar16_t *		image_path	__optional,
 	__in	uint32_t *		image_flags	__optional,
 	__in	struct pe_unicode_str *	image_name,
 	__out	void **			image_base);
 
+
+typedef int32_t __stdcall os_zw_terminate_process(
+	__in	void *		hprocess,
+	__in	int32_t		status);
+
 #endif
diff --git a/src/ldso/pe_terminate_current_process.c b/src/ldso/pe_terminate_current_process.c
new file mode 100644
index 0000000..ce0294d
--- /dev/null
+++ b/src/ldso/pe_terminate_current_process.c
@@ -0,0 +1,27 @@
+/*****************************************************************************/
+/*  pemagination: a (virtual) tour into portable bits and executable bytes   */
+/*  Copyright (C) 2013--2017  Z. Gilboa                                      */
+/*  Released under GPLv2 and GPLv3; see COPYING.PEMAGINE.                    */
+/*****************************************************************************/
+
+#include <psxtypes/psxtypes.h>
+#include <pemagine/pemagine.h>
+#include "pe_os.h"
+
+int32_t pe_terminate_current_process(int32_t estatus)
+{
+	void *				hntdll;
+	os_zw_terminate_process *	zw_terminate_process;
+
+	/* init */
+	if (!(hntdll = pe_get_ntdll_module_handle()))
+		return OS_STATUS_INTERNAL_ERROR;
+
+	if (!(zw_terminate_process = (os_zw_terminate_process *)pe_get_procedure_address(
+			hntdll,"ZwTerminateProcess")))
+		return OS_STATUS_INTERNAL_ERROR;
+
+	return zw_terminate_process(
+		OS_CURRENT_PROCESS_HANDLE,
+		estatus);
+}