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

/*  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_get_server_snapshot(

	struct toks_driver_ctx *        dctx,

	struct toks_server_snapshot **  snapshot)

{

	int32_t                         status;

	void *                          hport;

	void *                          hfile;

	void *                          hsection;

	void *                          addr;

	size_t                          size;

	nt_oa                           oa;

	nt_large_integer                ssize;

	nt_sd_common_buffer             sd;

	struct _nt_tty_port_msg         msg;

	struct toks_server_snapshot *   ctx;

	struct toks_service_info *      svcinfo;

	uint32_t                        access_owner;

	uint32_t                        access_group;

	uint32_t                        access_other;

	uint32_t                        access_admin;

	hport  = toks_get_driver_hservice(dctx);

	ntapi->tt_aligned_block_memset(

		&msg,0,sizeof(msg));

	msg.header.msg_type  = NT_LPC_NEW_MESSAGE;

	msg.header.data_size = sizeof(msg.ttyinfo) + sizeof(msg.svcdata);

	msg.header.msg_size  = sizeof(msg);

	msg.ttyinfo.opcode   = TOKS_DAEMON_IOCTL;

	msg.ttyinfo.opdata   = TOKS_IOCTL_GET_TOKEN_INFO;

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

		return status;

	else if (msg.ttyinfo.status)

		return msg.ttyinfo.status;

	/* token info */

	hfile = msg.ttyinfo.exarg;

	/* snapshot context */

	if (!(addr = toks_calloc(1,sizeof(*ctx)))) {

		ntapi->zw_close(hfile);

		return NT_STATUS_MEMORY_NOT_ALLOCATED;

	}

	ctx     = (struct toks_server_snapshot *)addr;

	svcinfo = (struct toks_service_info *)addr;

	/* uuid */

	svcinfo->uuid.data1    = (msg.svcdata.meta[0]);

	svcinfo->uuid.data2    = (msg.svcdata.meta[1]  & 0xffff);

	svcinfo->uuid.data3    = (msg.svcdata.meta[1] >> 16);

	svcinfo->uuid.data4[0] = (msg.svcdata.meta[2] & 0xff);

	svcinfo->uuid.data4[1] = (msg.svcdata.meta[2] & 0xff00) >> 8;

	svcinfo->uuid.data4[2] = (msg.svcdata.meta[2] & 0xff0000) >> 16;

	svcinfo->uuid.data4[3] = (msg.svcdata.meta[2] & 0xff000000) >> 24;

	svcinfo->uuid.data4[4] = (msg.svcdata.meta[3] & 0xff);

	svcinfo->uuid.data4[5] = (msg.svcdata.meta[3] & 0xff00) >> 8;

	svcinfo->uuid.data4[6] = (msg.svcdata.meta[3] & 0xff0000) >> 16;

	svcinfo->uuid.data4[7] = (msg.svcdata.meta[3] & 0xff000000) >> 24;

	/* syspid, systid */

	svcinfo->syspid        = msg.svcdata.data[0];

	svcinfo->systid        = msg.svcdata.data[1];

	/* ctrlpid, csyspid */

	svcinfo->ctrlpid       = msg.svcdata.data[2];

	svcinfo->csyspid       = msg.svcdata.data[3];

	/* allocated, available, pending */

	svcinfo->allocated     = msg.svcdata.data[4];

	svcinfo->available     = msg.svcdata.data[5];

	svcinfo->pending       = msg.svcdata.data[6];

	/* used, free */

	svcinfo->tused         = msg.svcdata.data[7];

	svcinfo->tfree         = msg.svcdata.data[8];

	/* sd */

	access_owner  = NT_GENERIC_READ;

	access_group  = NT_SEC_READ_CONTROL | NT_FILE_READ_ATTRIBUTES;

	access_other  = NT_SEC_READ_CONTROL | NT_FILE_READ_ATTRIBUTES;

	access_owner |= NT_SEC_STANDARD_RIGHTS_ALL;

	access_admin  = access_owner;

	ntapi->acl_init_common_descriptor(

		&sd,0,0,0,0,

		access_owner,access_group,access_other,

		access_admin,access_owner,

		0);

	/* section */

	oa.len		= sizeof(oa);

	oa.root_dir	= 0;

	oa.obj_name	= 0;

	oa.obj_attr	= 0;

	oa.sec_desc	= &sd.sd;

	oa.sec_qos	= 0;

	addr		= 0;

	ssize.quad	= 0;

	size		= __offsetof(struct toks_token_ctx,tokens);

	size           += svcinfo->tused * sizeof(struct toks_token_info);

	status = ntapi->zw_create_section(

		&hsection,

		NT_SECTION_QUERY|NT_SECTION_MAP_READ,

		&oa,&ssize,

		NT_PAGE_READONLY,

		NT_SEC_COMMIT,

		hfile);

	if (status) {

		ntapi->zw_close(hfile);

		toks_free(ctx);

		return status;

	};

	/* section mapping */

	status = ntapi->zw_map_view_of_section(

		hsection,

		NT_CURRENT_PROCESS_HANDLE,

		&addr,0,0,0,

		&size,NT_VIEW_UNMAP,

		0,NT_PAGE_READONLY);

	ntapi->zw_close(hfile);

	ntapi->zw_close(hsection);

	if (status) {

		toks_free(ctx);

		return status;

	};

	/* all done */

	ctx->tokctx = addr;

	*snapshot   = ctx;

	return NT_STATUS_SUCCESS;

}

void toks_free_server_snapshot(struct toks_server_snapshot * snapshot)

{

	ntapi->zw_unmap_view_of_section(

		NT_CURRENT_PROCESS_HANDLE,

		(void *)snapshot->tokctx);

	toks_free(snapshot);

}