Blob Blame History Raw
/********************************************************/
/* ntapi: Native API core library */
/* Copyright (C) 2013--2017 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */
/********************************************************/
#include <psxtypes/psxtypes.h>
#include <ntapi/nt_file.h>
#include <ntapi/ntapi.h>
#include "ntapi_impl.h"
ssize_t __cdecl __ntapi_log_write(
__in void * hfile,
__in const void * buf,
__in size_t bytes)
{
nt_iosb iosb;
int32_t status;
status = __ntapi->zw_write_file(
hfile,
(void *)0,
(nt_io_apc_routine *)0,
(void *)0,
&iosb,
(void *)buf,
(uint32_t)bytes,
(nt_large_integer *)0,
(uint32_t *)0);
if (status == NT_STATUS_SUCCESS)
return iosb.info;
else
return -1;
}
int32_t __cdecl __ntapi_log_fn_call(
__in void * hfile __optional,
__in const char * fn_caller_name,
__in void * fn_callee_addr,
__in uintptr_t fn_ret,
__in ntapi_log_write* pfn_log_write __optional,
__in const char * source __optional,
__in int line __optional)
{
struct pe_ldr_tbl_entry * image_meta;
void * image_base;
char * fn_name;
ssize_t bytes;
char log_buf[2048];
if (!pfn_log_write)
pfn_log_write = __ntapi_log_write;
image_meta = pe_get_symbol_module_info(fn_callee_addr);
fn_name = (char *)0;
if (image_meta)
image_base = image_meta->dll_base;
else
image_base = (void *)0;
if (image_base)
fn_name = pe_get_symbol_name(
image_base,
fn_callee_addr);
if (source && fn_name)
bytes = __ntapi->sprintf(
log_buf,
"%s: (%s:%d):\n"
"--> %s returned 0x%08x\n\n",
fn_caller_name, source, line, fn_name, fn_ret);
else if (fn_name)
bytes = __ntapi->sprintf(
log_buf,
"%s: %s returned 0x%08x\n\n",
fn_caller_name, fn_name, fn_ret);
else if (source)
bytes = __ntapi->sprintf(
log_buf,
"%s: (%s:%d):\n"
"--> calling 0x%08x returned 0x%08x\n\n",
fn_caller_name, source, line, fn_callee_addr, fn_ret);
else
bytes = __ntapi->sprintf(
log_buf,
"%s: calling 0x%08x returned 0x%08x\n\n",
fn_caller_name, fn_callee_addr, fn_ret);
if (bytes) {
bytes = __ntapi->strlen(log_buf);
if (bytes == pfn_log_write(hfile,log_buf,bytes))
return NT_STATUS_SUCCESS;
else
return NT_STATUS_UNSUCCESSFUL;
} else
return NT_STATUS_UNSUCCESSFUL;
}
int32_t __cdecl __ntapi_log_msg(
__in void * hfile __optional,
__in const char * source __optional,
__in int line __optional,
__in const char * fn_caller_name,
__in const char * fmt,
__in uintptr_t arg1,
__in uintptr_t arg2,
__in uintptr_t arg3,
__in uintptr_t arg4,
__in uintptr_t arg5,
__in uintptr_t arg6,
__in ntapi_log_write* pfn_log_write __optional)
{
char * buffer;
ssize_t bytes;
char log_buf[2048];
if (!pfn_log_write)
pfn_log_write = __ntapi_log_write;
bytes = 0;
buffer = log_buf;
if (source)
bytes = __ntapi->sprintf(
buffer,
"%s: (%s:%d):\n--> ",
fn_caller_name,source,line);
else if (fn_caller_name)
bytes = __ntapi->sprintf(
buffer,
"%s: ",
fn_caller_name);
else
log_buf[0] = '\0';
if (bytes >= 0)
buffer += __ntapi->strlen(log_buf);
else
return NT_STATUS_UNSUCCESSFUL;
bytes = __ntapi->sprintf(buffer,fmt,arg1,arg2,arg3,arg4,arg5,arg6);
if (bytes) {
bytes = __ntapi->strlen(log_buf);
if (bytes == pfn_log_write(hfile,log_buf,bytes))
return NT_STATUS_SUCCESS;
else
return NT_STATUS_UNSUCCESSFUL;
} else
return NT_STATUS_UNSUCCESSFUL;
}