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

/*  ntcon: free-standing console interface definitions  */

/*  Copyright (C) 2013--2016  SysDeer Technologies, LLC */

/*  Released under GPLv2 and GPLv3; see COPYING.NTCON.  */

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

#include <pemagine/pemagine.h>

#include <ntcon/ntcon.h>

#define NT_STATUS_SUCCESS		0x00000000

#define NT_STATUS_INTERNAL_ERROR	0xC00000E5

#define NTCON_PTR_INIT(x,y) pvtbl->x = \

	(ntcon_##x *)pe_get_procedure_address(hkernel, #y)

ntcon_api

int32_t __fastcall ntcon_vtbl_init(ntcon_vtbl * pvtbl)

{

	void *	hkernel;

	if (!(hkernel = pe_get_kernel32_module_handle()))

		return NT_STATUS_INTERNAL_ERROR;

	/* console handles */

	NTCON_PTR_INIT(alloc_console,				AllocConsole);

	NTCON_PTR_INIT(attach_console,				AttachConsole);

	NTCON_PTR_INIT(free_console,				FreeConsole);

	NTCON_PTR_INIT(get_std_handle,				GetStdHandle);

	/* read/write using std handles */

	NTCON_PTR_INIT(get_number_of_console_input_events,	GetNumberOfConsoleInputEvents);

	NTCON_PTR_INIT(flush_console_input_buffer,		FlushConsoleInputBuffer);

	NTCON_PTR_INIT(peek_console_input_ansi,			PeekConsoleInputA);

	NTCON_PTR_INIT(peek_console_input_utf16,		PeekConsoleInputW);

	NTCON_PTR_INIT(read_console_ansi,			ReadConsoleA);

	NTCON_PTR_INIT(read_console_utf16,			ReadConsoleW);

	NTCON_PTR_INIT(read_console_input_ansi,			ReadConsoleInputA);

	NTCON_PTR_INIT(read_console_input_utf16,		ReadConsoleInputW);

	NTCON_PTR_INIT(read_console_output_ansi,		ReadConsoleOutputA);

	NTCON_PTR_INIT(read_console_output_utf16,		ReadConsoleOutputW);

	NTCON_PTR_INIT(write_console_ansi,			WriteConsoleA);

	NTCON_PTR_INIT(write_console_utf16,			WriteConsoleW);

	NTCON_PTR_INIT(write_console_input_ansi,		WriteConsoleInputA);

	NTCON_PTR_INIT(write_console_input_utf16,		WriteConsoleInputW);

	NTCON_PTR_INIT(write_console_output_ansi,		WriteConsoleOutputA);

	NTCON_PTR_INIT(write_console_output_utf16,		WriteConsoleOutputW);

	/* screen buffers */

	NTCON_PTR_INIT(create_console_screen_buffer,		CreateConsoleScreenBuffer);

	NTCON_PTR_INIT(get_console_screen_buffer_info,		GetConsoleScreenBufferInfo);

	NTCON_PTR_INIT(get_console_screen_buffer_info_ex,	GetConsoleScreenBufferInfoEx);

	NTCON_PTR_INIT(set_console_screen_buffer_info_ex,	SetConsoleScreenBufferInfoEx);

	NTCON_PTR_INIT(set_console_screen_buffer_size,		SetConsoleScreenBufferSize);

	NTCON_PTR_INIT(set_console_active_screen_buffer,	SetConsoleActiveScreenBuffer);

	NTCON_PTR_INIT(set_console_window_info,			SetConsoleWindowInfo);

	NTCON_PTR_INIT(scroll_console_screen_buffer_ansi,	ScrollConsoleScreenBufferA);

	NTCON_PTR_INIT(scroll_console_screen_buffer_utf16,	ScrollConsoleScreenBufferW);

	/* code page */

	NTCON_PTR_INIT(get_console_code_page,			GetConsoleCP);

	NTCON_PTR_INIT(set_console_code_page,			SetConsoleCP);

	/* console control */

	NTCON_PTR_INIT(generate_console_ctrl_event,		GenerateConsoleCtrlEvent);

	NTCON_PTR_INIT(set_console_ctrl_handler,		SetConsoleCtrlHandler);

	/* interface mode */

	NTCON_PTR_INIT(get_console_mode,			GetConsoleMode);

	NTCON_PTR_INIT(set_console_mode,			SetConsoleMode);

	/* console process list */

	NTCON_PTR_INIT(get_console_process_list,		GetConsoleProcessList);

	/* aliases */

	NTCON_PTR_INIT(add_console_alias_ansi,			AddConsoleAliasA);

	NTCON_PTR_INIT(add_console_alias_utf16,			AddConsoleAliasW);

	NTCON_PTR_INIT(get_console_alias_ansi,			GetConsoleAliasA);

	NTCON_PTR_INIT(get_console_alias_utf16,			GetConsoleAliasW);

	NTCON_PTR_INIT(get_console_aliases_ansi,		GetConsoleAliasesA);

	NTCON_PTR_INIT(get_console_aliases_utf16,		GetConsoleAliasesW);

	NTCON_PTR_INIT(get_console_aliases_length_ansi,		GetConsoleAliasesLengthA);

	NTCON_PTR_INIT(get_console_aliases_length_utf16,	GetConsoleAliasesLengthW);

	NTCON_PTR_INIT(get_console_alias_exes_ansi,		GetConsoleAliasExesA);

	NTCON_PTR_INIT(get_console_alias_exes_utf16,		GetConsoleAliasExesW);

	NTCON_PTR_INIT(get_console_alias_exes_length_ansi,	GetConsoleAliasExesLengthA);

	NTCON_PTR_INIT(get_console_alias_exes_length_utf16,	GetConsoleAliasExesLengthW);

	/* output attributes */

	NTCON_PTR_INIT(fill_console_output_attribute,		FillConsoleOutputAttribute);

	NTCON_PTR_INIT(read_console_output_attribute,		ReadConsoleOutputAttribute);

	NTCON_PTR_INIT(write_console_output_attribute,		WriteConsoleOutputAttribute);

	/* output characters */

	NTCON_PTR_INIT(fill_console_output_character_ansi,	FillConsoleOutputCharacterA);

	NTCON_PTR_INIT(fill_console_output_character_utf16,	FillConsoleOutputCharacterW);

	NTCON_PTR_INIT(read_console_output_character_ansi,	ReadConsoleOutputCharacterA);

	NTCON_PTR_INIT(read_console_output_character_utf16,	ReadConsoleOutputCharacterW);

	NTCON_PTR_INIT(write_console_output_character_ansi,	WriteConsoleOutputCharacterA);

	NTCON_PTR_INIT(write_console_output_character_utf16,	WriteConsoleOutputCharacterW);

	/* output code page */

	NTCON_PTR_INIT(get_console_output_code_page,		GetConsoleOutputCP);

	NTCON_PTR_INIT(set_console_output_code_page,		SetConsoleOutputCP);

	/* text attributes */

	NTCON_PTR_INIT(set_console_text_attribute,		SetConsoleTextAttribute);

	/* cursor info */

	NTCON_PTR_INIT(get_console_cursor_info,			GetConsoleCursorInfo);

	NTCON_PTR_INIT(set_console_cursor_info,			SetConsoleCursorInfo);

	NTCON_PTR_INIT(set_console_cursor_position,		SetConsoleCursorPosition);

	/* selection info */

	NTCON_PTR_INIT(get_console_selection_info,		GetConsoleSelectionInfo);

	/* history info */

	NTCON_PTR_INIT(get_console_history_info,		GetConsoleHistoryInfo);

	NTCON_PTR_INIT(set_console_history_info,		SetConsoleHistoryInfo);

	/* font info */

	NTCON_PTR_INIT(get_console_font_size,			GetConsoleFontSize);

	NTCON_PTR_INIT(get_current_console_font,		GetCurrentConsoleFont);

	NTCON_PTR_INIT(get_current_console_font_ex,		GetCurrentConsoleFontEx);

	NTCON_PTR_INIT(set_current_console_font_ex,		SetCurrentConsoleFontEx);

	/* mouse capabilities */

	NTCON_PTR_INIT(get_number_of_console_mouse_buttons,	GetNumberOfConsoleMouseButtons);

	/* window information */

	NTCON_PTR_INIT(get_console_window,			GetConsoleWindow);

	NTCON_PTR_INIT(get_largest_console_window_size,		GetLargestConsoleWindowSize);

	/* console api: display mode */

	NTCON_PTR_INIT(get_console_display_mode,		GetConsoleDisplayMode);

	NTCON_PTR_INIT(set_console_display_mode,		SetConsoleDisplayMode);

	/* console title */

	NTCON_PTR_INIT(get_console_title_ansi,			GetConsoleTitleA);

	NTCON_PTR_INIT(get_console_title_utf16,			GetConsoleTitleW);

	NTCON_PTR_INIT(set_console_title_ansi,			SetConsoleTitleA);

	NTCON_PTR_INIT(set_console_title_utf16,			SetConsoleTitleW);

	NTCON_PTR_INIT(get_console_original_title_ansi,		GetConsoleOriginalTitleA);

	NTCON_PTR_INIT(get_console_original_title_utf16,	GetConsoleOriginalTitleW);

	return NT_STATUS_SUCCESS;

}