#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