/********************************************************/
/*  ntapi: Native API core library                      */
/*  Copyright (C) 2013--2017  Z. Gilboa                 */
/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */
/********************************************************/

#include <psxtypes/psxtypes.h>
#include <ntapi/nt_port.h>
#include <ntapi/nt_tty.h>
#include <ntapi/ntapi.h>
#include "ntapi_impl.h"
#include "ntapi_pty.h"

int32_t	__stdcall __ntapi_pty_query(
	nt_pty *		pty,
	nt_io_status_block *	iosb,
	void *			pty_info,
	uint32_t		pty_info_length,
	nt_pty_info_class	pty_info_class)
{
	int32_t			status;
	void * 			hport;
	nt_pty_sigctl_msg	msg;
	uintptr_t *		info;
	nt_pty_inherit_info * 	inherit;

	if (pty_info_class>=NT_PTY_INFORMATION_CAP)
		return NT_STATUS_INVALID_INFO_CLASS;
	else if (pty_info_class == NT_PTY_BASIC_INFORMATION)
		return NT_STATUS_NOT_IMPLEMENTED;
	else if ((pty_info_class == NT_PTY_CLIENT_INFORMATION) && (pty_info_length != sizeof(nt_pty_client_info)))
		return NT_STATUS_INVALID_PARAMETER;
	else if ((pty_info_class == NT_PTY_INHERIT_INFORMATION) && (pty_info_length != sizeof(nt_pty_inherit_info)))
		return NT_STATUS_INVALID_PARAMETER;

	__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		= NT_TTY_PTY_QUERY;

	if (pty_info_class == NT_PTY_CLIENT_INFORMATION) {
		hport 				= pty->hport;
		msg.data.ctlinfo.hpty		= pty->hpty;
		msg.data.ctlinfo.luid.high	= pty->luid.high;
		msg.data.ctlinfo.luid.low	= pty->luid.low;
		msg.data.ctlinfo.ctlcode	= pty_info_class;

		__ntapi->tt_guid_copy(
			&msg.data.ctlinfo.guid,
			&pty->guid);

	} else if (pty_info_class == NT_PTY_INHERIT_INFORMATION) {
		msg.data.ctlinfo.hpty		= NT_INVALID_HANDLE_VALUE;
		msg.data.ctlinfo.ctlcode	= pty_info_class;

		inherit = (nt_pty_inherit_info *)pty_info;
		msg.data.ctlinfo.ctxarg[0] = inherit->any[0];
		msg.data.ctlinfo.ctxarg[1] = inherit->any[1];
		msg.data.ctlinfo.ctxarg[2] = inherit->any[2];
		msg.data.ctlinfo.ctxarg[3] = inherit->any[3];

		hport = pty ? pty : __ntapi_internals()->hport_tty_session;
	} else {
		return NT_STATUS_INVALID_INFO_CLASS;
	}

	if ((status = __ntapi->zw_request_wait_reply_port(hport,&msg,&msg)))
		return status;
	else if (msg.data.ttyinfo.status)
		return msg.data.ttyinfo.status;

	iosb->info   = msg.data.ctlinfo.iosb.info;
	iosb->status = msg.data.ctlinfo.iosb.status;

	if (pty_info_class == NT_PTY_CLIENT_INFORMATION) {
		info = (uintptr_t *)pty_info;
		info[0] = msg.data.ctlinfo.ctxarg[0];
		info[1] = msg.data.ctlinfo.ctxarg[1];
		info[2] = msg.data.ctlinfo.ctxarg[2];
		info[3] = msg.data.ctlinfo.ctxarg[3];

	} else if (pty_info_class == NT_PTY_INHERIT_INFORMATION) {
		inherit = (nt_pty_inherit_info *)pty_info;
		inherit->hpty 		= msg.data.ctlinfo.hpty;
		inherit->luid.low 	= msg.data.ctlinfo.luid.low;
		inherit->luid.high 	= msg.data.ctlinfo.luid.high;

		inherit->access		= (uint32_t)msg.data.ctlinfo.ctxarg[0];
		inherit->flags 		= (uint32_t)msg.data.ctlinfo.ctxarg[1];
		inherit->share 		= (uint32_t)msg.data.ctlinfo.ctxarg[2];
		inherit->options 	= (uint32_t)msg.data.ctlinfo.ctxarg[3];

		__ntapi->tt_guid_copy(
			&inherit->guid,
			&msg.data.ctlinfo.guid);
	}

	return NT_STATUS_SUCCESS;
}