#ifndef _NT_THREAD_H_

#define _NT_THREAD_H_

#include "nt_abi.h"

#include "nt_compiler.h"

#include "nt_object.h"

#include "bits/nt32/nt_thread_i386.h"

#include "bits/nt64/nt_thread_x86_64.h"

typedef enum _nt_thread_info_class {

	NT_THREAD_BASIC_INFORMATION,

	NT_THREAD_TIMES,

	NT_THREAD_PRIORITY,

	NT_THREAD_BASE_PRIORITY,

	NT_THREAD_AFFINITY_MASK,

	NT_THREAD_IMPERSONATION_TOKEN,

	NT_THREAD_DESCRIPTOR_TABLE_ENTRY,

	NT_THREAD_ENABLE_ALIGNMENT_FAULT_FIXUP,

	NT_THREAD_EVENT_PAIR,

	NT_THREAD_QUERY_SET_WIN32_START_ADDRESS,

	NT_THREAD_ZERO_TLS_CELL,

	NT_THREAD_PERFORMANCE_COUNT,

	NT_THREAD_AM_I_LAST_THREAD,

	NT_THREAD_IDEAL_PROCESSOR,

	NT_THREAD_PRIORITY_BOOST,

	NT_THREAD_SET_TLS_ARRAY_ADDRESS,

	NT_THREAD_IS_IO_PENDING,

	NT_THREAD_HIDE_FROM_DEBUGGER

} nt_thread_info_class;

typedef enum _nt_exception_disposition {

	NT_EXCEPTION_CONTINUE_EXECUTION,

	NT_EXCEPTION_CONTINUE_SEARCH,

	NT_EXCEPTION_NESTED_EXCEPTION,

	NT_EXCEPTION_COLLIDED_UNWIND

} nt_exception_disposition;

/* special handles */

#define NT_CURRENT_THREAD_HANDLE (void *)(uintptr_t)-2

/* thread access bits */

#define NT_THREAD_TERMINATE			0x00000001

#define NT_THREAD_SUSPEND_RESUME		0x00000002

#define NT_THREAD_ALERT				0x00000004 /* fits right in the middle... */

#define NT_THREAD_GET_CONTEXT			0x00000008

#define NT_THREAD_SET_CONTEXT			0x00000010

#define NT_THREAD_SET_INFORMATION		0x00000020

#define NT_THREAD_QUERY_INFORMATION		0x00000040

#define NT_THREAD_SET_THREAD_TOKEN		0x00000080

#define NT_THREAD_IMPERSONATE			0x00000100

#define NT_THREAD_DIRECT_IMPERSONATION		0x00000200

#define NT_THREAD_SYNCHRONIZE			0x00100000

#define NT_THREAD_ALL_ACCESS			NT_THREAD_TERMINATE \

						| NT_THREAD_SUSPEND_RESUME \

						| NT_THREAD_ALERT \

						| NT_THREAD_GET_CONTEXT \

						| NT_THREAD_SET_CONTEXT \

						| NT_THREAD_SET_INFORMATION \

						| NT_THREAD_QUERY_INFORMATION \

						| NT_THREAD_SET_THREAD_TOKEN \

						| NT_THREAD_IMPERSONATE \

						| NT_THREAD_DIRECT_IMPERSONATION \

						| NT_THREAD_SYNCHRONIZE

/* library-specific thread creation flags */

#define NT_THREAD_RUN_IMMEDIATELY		0x00000000

#define NT_CREATE_SUSPENDED			0x00000004

#define NT_CREATE_FIRST_THREAD_OF_PROCESS	0x00008000

#define NT_CREATE_LOCAL_THREAD			0x00010000

#define NT_STACK_SIZE_PARAM_IS_A_RESERVATION	0x00800000

#define NT_CLOSE_THREAD_HANDLE			0x01000000

/* thread context */

#define NT_CONTEXT_JUST_EVERYTHING		(uint32_t)-1

/* source mark: arch-specific code: begin */

#if (__SIZEOF_POINTER__ == 4)

typedef struct _nt_thread_context_i386  nt_thread_context;

#endif

#if (__SIZEOF_POINTER__ == 8)

typedef nt_mcontext_x86_64_t            nt_thread_context;

typedef nt_mcontext_x86_64_t            nt_mcontext_t;

#endif

/* source mark: arch-specific code: end */

typedef struct _nt_user_stack {

	void * fixed_stack_base;

	void * fixed_stack_limit;

	void * expandable_stack_base;

	void * expandable_stack_limit;

	void * expandable_stack_bottom;

} nt_user_stack;

typedef struct _nt_exception_registration_record {

	struct _nt_exception_registration_record *	next;

	nt_exception_disposition *			handler;

} nt_exception_registration_record;

typedef struct _nt_tib {

	nt_exception_registration_record * 	exception_list;

	void * 					stack_base;

	void * 					stack_limit;

	void * 					sub_system_tib;

	union {

		void * 		fiber_data;

		uint32_t 	version;

	};

	void * 					arbitrary_user_pointer;

	struct _nt_tib * 			self;

} nt_tib;

typedef struct _nt_thread_basic_information {

	int32_t		exit_status;

	nt_tib *	teb_base_address;

	nt_client_id	cid;

	intptr_t	affinity_mask;

	int32_t		priority;

	int32_t		base_priority;

} nt_thread_basic_information;

typedef int32_t __stdcall nt_thread_start_routine(void * context);

typedef struct _nt_thread_params {

	__in	void *				hprocess;

	__out	void *				hthread;

	__in	nt_thread_start_routine *	start;

	__in	void *				arg			__optional;

	__in	void *				ext_ctx			__optional;

	__in	size_t				ext_ctx_size;

	__in	nt_object_attributes *		obj_attr		__optional;

	__in	uint32_t			creation_flags;

	__in	uint32_t			stack_zero_bits;

	__in	size_t				stack_size_commit;

	__in	size_t				stack_size_reserve;

	__in	nt_user_stack *			stack_info		__optional;

	__in	nt_thread_context *		reg_context		__optional;

	__out	int32_t				csrss_status;

	__out	uint32_t			thread_id;

	__in	void *				reserved[2];

} nt_thread_params;

typedef void    __stdcall nt_knormal_routine(

	void *	apc_context,

	void *	arg_1st,

	void *	arg_2nd);

typedef int32_t __stdcall ntapi_zw_create_thread(

	__out	void **			hthread,

	__in	uintptr_t		desired_access,

	__in	nt_object_attributes *	obj_attr,

	__in	void *			hprocess,

	__out	nt_client_id *		hclient_id,

	__in	nt_thread_context *	context,

	__in	nt_user_stack *		user_stack,

	__in	uintptr_t		suspended_flag);

typedef int32_t __stdcall ntapi_zw_open_thread(

	__out	void **			hthread,

	__in	uint32_t		desired_access,

	__in	nt_object_attributes *	obj_attr,

	__in	nt_client_id *		hclient_id);

typedef int32_t __stdcall ntapi_zw_terminate_thread(

	__in	void *	hthread,

	__in	int32_t	exit_status);

typedef int32_t __stdcall ntapi_zw_query_information_thread(

	__in	void *			hthread,

	__in	nt_thread_info_class	thread_info_class,

	__out	void *			thread_info,

	__in	size_t			thread_info_length,

	__out	size_t *		returned_length		__optional);

typedef int32_t __stdcall ntapi_zw_set_information_thread(

	__in	void *			hthread,

	__in	nt_thread_info_class	thread_info_class,

	__in	void *			thread_info,

	__in	size_t			thread_info_length);

typedef int32_t __stdcall ntapi_zw_suspend_thread(

	__in	void *		hthread,

	__out	uint32_t *	prev_suspend_count __optional);

typedef int32_t __stdcall ntapi_zw_resume_thread(

	__in	void *		hthread,

	__out	uint32_t *	prev_suspend_count __optional);

typedef int32_t __stdcall ntapi_zw_get_context_thread(

	__in	void *	hthread,

	__out	void *	context);

typedef int32_t __stdcall ntapi_zw_set_context_thread(

	__in	void *	hthread,

	__in	void *	context);

typedef int32_t __stdcall ntapi_zw_queue_apc_thread(

	__in	void *			hthread,

	__in	nt_knormal_routine *	apc_routine,

	__in	void *			apc_context,

	__in	void *			arg_1st,

	__in	void *			arg_2nd);

typedef int32_t __stdcall ntapi_zw_test_alert(void);

typedef int32_t __stdcall ntapi_zw_alert_thread(

	__in	void *	hthread);

typedef int32_t __stdcall ntapi_zw_alert_resume_thread(

	__in	void *		hthread,

	__out	uint32_t *	prev_suspend_count __optional);

typedef int32_t __stdcall ntapi_zw_register_thread_terminate_port(

	__in	void *	port_handle);

typedef int32_t __stdcall ntapi_zw_impersonate_thread(

	__in	void *					hthread,

	__in	void *					target_thread_handle,

	__in	nt_security_quality_of_service *	sec_qos);

typedef int32_t __stdcall ntapi_zw_impersonate_anonymous_token(

	__in	void *	hthread);

/* extension functions */

typedef int32_t __stdcall ntapi_tt_create_local_thread(

	__in_out	nt_thread_params *	params);

typedef int32_t __stdcall ntapi_tt_create_remote_thread(

	__in_out	nt_thread_params *	params);

typedef int32_t __stdcall ntapi_tt_create_thread(

	__in_out	nt_thread_params *	params);

#endif