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

/*  toksvc: a framework-native token broker service      */

/*  Copyright (C) 2020  SysDeer Technologies, LLC        */

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

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

#include <psxtypes/psxtypes.h>

#include <ntapi/ntapi.h>

#include <toksvc/toksvc.h>

#include "toksvc_driver_impl.h"

#include "toksvc_daemon_impl.h"

int32_t toks_client_connect(struct toks_driver_ctx * dctx)

{

	int32_t			status;

	void *			hsvclink;

	void *			hservice;

	void *			hserver;

	struct _nt_tty_sync_msg msg;

	nt_unicode_string	name;

	nt_guid_str_utf16	guid;

	nt_oa			oa;

	nt_sqos			sqos = {

					sizeof(sqos),

					NT_SECURITY_ANONYMOUS,

					NT_SECURITY_TRACKING_STATIC,

					1};

	ntapi->tt_guid_to_string_utf16(

		toks_get_driver_guid(dctx),

		&guid);

	name.strlen = sizeof(guid);

	name.maxlen = 0;

	name.buffer = &guid.lbrace;

	oa.len		= sizeof(oa);

	oa.root_dir	= toks_get_driver_hsvcdir(dctx);

	oa.obj_name	= &nam;;

	oa.obj_attr	= NT_OBJ_INHERIT;

	oa.sec_desc	= 0;

	oa.sec_qos	= &sqo;;

	if ((status = ntapi->zw_open_symbolic_link_object(

			&hsvclink,

			NT_SYMBOLIC_LINK_QUERY,

			&oa)))

		return status;

	if ((status = ntapi->ipc_connect_by_symlink(&hservice,hsvclink,false))) {

		ntapi->zw_close(hsvclink);

		return status;

	}

	ntapi->zw_close(hsvclink);

	ntapi->tt_aligned_block_memset(

		&msg,0,sizeof(msg));

	msg.header.msg_type	= NT_LPC_NEW_MESSAGE;

	msg.header.data_size	= sizeof(msg.data);

	msg.header.msg_size	= sizeof(msg);

	msg.data.ttyinfo.opcode	= TOKS_DAEMON_CIDQUERY;

	if ((status = ntapi->zw_request_wait_reply_port(hservice,&msg,&msg))) {

		ntapi->zw_close(hservice);

		return status;

	} else if (msg.data.ttyinfo.status) {

		ntapi->zw_close(hservice);

		return msg.data.ttyinfo.status;

	}

	oa.len		= sizeof(oa);

	oa.root_dir	= 0;

	oa.obj_name	= 0;

	oa.obj_attr	= 0;

	oa.sec_desc	= 0;

	oa.sec_qos	= &sqo;;

	if ((status = ntapi->zw_open_process(

			&hserver,NT_PROCESS_SYNCHRONIZE,

			&oa,&msg.data.syncinfo.cid))) {

		ntapi->zw_close(hservice);

		return status;

	}

	toks_set_driver_hservice(

		dctx,hservice);

	toks_set_driver_hserver(

		dctx,hserver);

	return NT_STATUS_SUCCESS;

}