/********************************************************/

/*  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_ldr.h>

#include <ntapi/ntapi.h>

#include "ntapi_impl.h"

int32_t	__stdcall __ntapi_ldr_load_system_dll(

	__in	void *			hsysdir		__optional,

	__in	wchar16_t *		base_name,

	__in	uint16_t		base_name_size,

	__in	uint32_t *		image_flags	__optional,

	__out	void **			image_base)

{

	int32_t			status;

	void *			hkernel32;

	wchar16_t *		wch;

	wchar16_t *		cap;

	wchar16_t *		sysdir;

	nt_unicode_string	nt_sysdir;

	nt_unicode_string	nt_image_name;

	size_t			buffer[0x200/sizeof(size_t)];

	void *			(*load_library_ex_utf16)(

					wchar16_t *,

					void *,

					uint32_t);

	(void)image_flags;

	/* shell-style dos path */

	sysdir = (wchar16_t *)buffer;

	if ((status = __ntapi->tt_get_system_directory_dos_path(

			hsysdir,

			sysdir,sizeof(buffer),

			0,0,&nt_sysdir)))

		return status;

	sysdir = &sysdir[4];

	/* image */

	nt_image_name.strlen = base_name_size;

	nt_image_name.maxlen = base_name_size;

	nt_image_name.buffer = base_name;

	/* the logical way */

	status = __ntapi->ldr_load_dll(

		sysdir,0,

		&nt_image_name,

		image_base);

	/* eight point one big sigh support */

	if (status) {

		cap = &sysdir[sizeof(buffer)/sizeof(wchar16_t)];

		for (wch=sysdir; *wch; wch++)

			(void)0;

		if (&wch[base_name_size/sizeof(wchar16_t)] >= cap)

			return NT_STATUS_BUFFER_TOO_SMALL;

		if (!(hkernel32 = pe_get_kernel32_module_handle()))

			return NT_STATUS_INTERNAL_ERROR;

		if (!(load_library_ex_utf16 = pe_get_procedure_address(

				hkernel32,"LoadLibraryExW")))

			return NT_STATUS_INTERNAL_ERROR;

		__ntapi->tt_memcpy_utf16(

			wch,base_name,

			base_name_size);

		wch[base_name_size/sizeof(wchar16_t)] = 0;

		if ((*image_base = load_library_ex_utf16(sysdir,0,0)))

			status = NT_STATUS_SUCCESS;

	}

	/* downlevel... */

	if (status) {

		if (&wch[base_name_size/sizeof(wchar16_t)] >= &cap[-10])

			return NT_STATUS_BUFFER_TOO_SMALL;

		*wch++ = 'd';

		*wch++ = 'o';

		*wch++ = 'w';

		*wch++ = 'n';

		*wch++ = 'l';

		*wch++ = 'e';

		*wch++ = 'v';

		*wch++ = 'e';

		*wch++ = 'l';

		*wch++ = '\\';

		__ntapi->tt_memcpy_utf16(

			wch,base_name,

			base_name_size);

		wch[base_name_size/sizeof(wchar16_t)] = 0;

		if ((*image_base = load_library_ex_utf16(sysdir,0,0)))

			status = NT_STATUS_SUCCESS;

	}

	return status;

}