Blame src/daemon/ntapi_dsr_internal_connection.c

dd89bb
/********************************************************/
dd89bb
/*  ntapi: Native API core library                      */
4256e2
/*  Copyright (C) 2013--2016  Z. Gilboa                 */
dd89bb
/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */
dd89bb
/********************************************************/
dd89bb
6c2ba1
#include <ntapi/nt_atomic.h>
dd89bb
#include <ntapi/nt_status.h>
dd89bb
#include <ntapi/nt_thread.h>
dd89bb
#include <ntapi/nt_port.h>
dd89bb
#include <ntapi/nt_daemon.h>
dd89bb
#include <ntapi/ntapi.h>
dd89bb
#include "ntapi_impl.h"
dd89bb
cd4191
int32_t __ntapi_tt_seh_frame(void *, void *, void *, int32_t (*)(nt_daemon_params *));
cd4191
a5dc9f
static int32_t __stdcall __ntapi_dsr_internal_client_connect_fail(
6c2ba1
	nt_daemon_params *	params,
6c2ba1
	int32_t			status)
6c2ba1
{
6c2ba1
	at_store_32(
6c2ba1
		&params->exit_code_internal_client,
6c2ba1
		status);
6c2ba1
6c2ba1
	return __ntapi->zw_terminate_thread(
6c2ba1
		NT_CURRENT_THREAD_HANDLE,
6c2ba1
		status);
6c2ba1
}
6c2ba1
dd89bb
/* __ntapi_dsr_connect_internal_client executes in the daemon's dedicated thread */
dd89bb
int32_t __stdcall __ntapi_dsr_connect_internal_client(nt_daemon_params * params)
dd89bb
{
6c2ba1
	int32_t			status;
dd89bb
	intptr_t		port_id;
dd89bb
	nt_port_message		port_msg;
dd89bb
	nt_large_integer	timeout;
a5dc9f
	void *			hport;
a5dc9f
	void *			hready;
a5dc9f
a5dc9f
	/* avoid out-of-scope use */
a5dc9f
	hready = params->hthread_internal_client;
dd89bb
6c2ba1
	/* report state */
6c2ba1
	at_store_32(
6c2ba1
		&params->exit_code_daemon_start,
6c2ba1
		NT_STATUS_MORE_PROCESSING_REQUIRED);
dd89bb
dd89bb
	/* timeout-enabled first connection */
dd89bb
	timeout.quad = NT_DSR_INIT_MAX_WAIT;
dd89bb
6c2ba1
	if ((status = __ntapi->zw_reply_wait_receive_port_ex(
6f3f3f
			params->hport_daemon,
6f3f3f
			&port_id,
6f3f3f
			(nt_port_message *)0,
6f3f3f
			(nt_port_message *)&port_msg,
6f3f3f
			&timeout)))
a5dc9f
		return status;
dd89bb
dd89bb
	/* the internal client must be first */
dd89bb
	if (port_msg.client_id.process_id != pe_get_current_process_id())
6f3f3f
		return NT_STATUS_PORT_CONNECTION_REFUSED;
dd89bb
dd89bb
	/* accept connection request */
6c2ba1
	if ((status = __ntapi->zw_accept_connect_port(
a5dc9f
			&hport,
6f3f3f
			port_msg.client_id.process_id,
6f3f3f
			(nt_port_message *)&port_msg,
6f3f3f
			NT_LPC_ACCEPT_CONNECTION,
6f3f3f
			(nt_port_section_write *)0,
6f3f3f
			(nt_port_section_read *)0)))
a5dc9f
		return status;
dd89bb
dd89bb
	/* finalize connection */
a5dc9f
	if ((status = __ntapi->zw_complete_connect_port(hport)))
a5dc9f
		return status;
a5dc9f
a5dc9f
	/* await client thread task completion */
a5dc9f
	__ntapi->zw_wait_for_single_object(
a5dc9f
		hready,NT_SYNC_NON_ALERTABLE,0);
dd89bb
a5dc9f
	/* all done */
a5dc9f
	return NT_STATUS_SUCCESS;
dd89bb
}
dd89bb
dd89bb
dd89bb
/* __ntapi_dsr_internal_client_connect executes in its own temporary thread */
cd4191
static int32_t __ntapi_dsr_internal_client_connect_impl(nt_daemon_params * params)
dd89bb
{
6c2ba1
	int32_t				status;
dd89bb
	nt_unicode_string		server_name;
dd89bb
	nt_object_attributes		oa;
dd89bb
	nt_security_quality_of_service	sqos;
dd89bb
6c2ba1
	/* report status */
6c2ba1
	at_store_32(
6c2ba1
		&params->exit_code_internal_client,
6c2ba1
		NT_STATUS_MORE_PROCESSING_REQUIRED);
dd89bb
dd89bb
	/* init server_name */
dd89bb
	server_name.strlen = (uint16_t)__ntapi->tt_string_null_offset_short((const int16_t *)params->port_name);
dd89bb
	server_name.maxlen = 0;
dd89bb
	server_name.buffer = (uint16_t *)params->port_name;
dd89bb
dd89bb
	/* init security structure */
dd89bb
	sqos.length 			= sizeof(sqos);
dd89bb
	sqos.impersonation_level	= NT_SECURITY_IMPERSONATION;
dd89bb
	sqos.context_tracking_mode	= NT_SECURITY_TRACKING_DYNAMIC;
dd89bb
	sqos.effective_only		= 1;
dd89bb
dd89bb
	/* init the port's object attributes */
dd89bb
	oa.len		= sizeof(oa);
dd89bb
	oa.root_dir	= (void *)0;
dd89bb
	oa.obj_name	= &server_name;
dd89bb
	oa.obj_attr	= 0;
dd89bb
	oa.sec_desc	= (nt_security_descriptor *)0;
dd89bb
	oa.sec_qos	= &sqo;;
dd89bb
dd89bb
	/* establish internal connection */
6c2ba1
	if ((status = __ntapi->zw_connect_port(
6c2ba1
			&params->hport_internal_client,
6c2ba1
			&server_name,
6c2ba1
			&sqos,
6c2ba1
			0,0,0,0,0)))
a5dc9f
		return __ntapi_dsr_internal_client_connect_fail(
6c2ba1
			params,status);
dd89bb
a5dc9f
	/* update port info */
dd89bb
	if (params->pport_internal_client)
a5dc9f
		at_store(
a5dc9f
			(intptr_t *)params->pport_internal_client,
a5dc9f
			(intptr_t)params->hport_internal_client);
a5dc9f
a5dc9f
	/* report status */
a5dc9f
	at_store_32(
a5dc9f
		&params->exit_code_internal_client,
a5dc9f
		status);
dd89bb
dd89bb
	/* signal the 'internal-client-is-ready' event */
6c2ba1
	status = __ntapi->zw_set_event(
dd89bb
		params->hevent_internal_client_ready,
dd89bb
		0);
dd89bb
dd89bb
	/* exit the task-specific thread */
a5dc9f
	return __ntapi->zw_terminate_thread(
a5dc9f
		NT_CURRENT_THREAD_HANDLE,
a5dc9f
		status);
dd89bb
}
cd4191
cd4191
int32_t __stdcall __ntapi_dsr_internal_client_connect(nt_daemon_params * params)
cd4191
{
cd4191
	return __ntapi_tt_seh_frame(
cd4191
		params,0,0,
cd4191
		__ntapi_dsr_internal_client_connect_impl);
cd4191
}