diff --git a/include/pemagine/pemagine.h b/include/pemagine/pemagine.h index 781ef59..e88e244 100644 --- a/include/pemagine/pemagine.h +++ b/include/pemagine/pemagine.h @@ -286,6 +286,10 @@ pe_api int32_t pe_open_physical_parent_directory( uint32_t open_options); +pe_api int32_t pe_terminate_current_process( + int32_t estatus); + + #ifdef __cplusplus } #endif diff --git a/project/common.mk b/project/common.mk index bf0e5af..719394e 100644 --- a/project/common.mk +++ b/project/common.mk @@ -16,6 +16,7 @@ API_SRCS = \ src/ldso/pe_load_framework_loader.c \ src/ldso/pe_open_image_from_addr.c \ src/ldso/pe_open_physical_parent_directory.c \ + src/ldso/pe_terminate_current_process.c \ src/meta/pe_get_image_stack_heap_info.c \ src/meta/pe_get_symbol_module_info.c \ src/meta/pe_get_symbol_name.c \ 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 +#include +#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); +}