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

/*  ptycon: a pty-console bridge                         */

/*  Copyright (C) 2016--2017  SysDeer Technologies, LLC  */

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

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

#include <psxtypes/psxtypes.h>

#include <ntcon/ntcon.h>

#include <ntapi/ntapi.h>

#include <ptycon/ptycon.h>

#include "ptycon_driver_impl.h"

static int32_t ptyc_cat(

	struct ptyc_driver_ctx* dctx,

	void *			hat,

	const char *		unit,

	void *			hevent)

{

	int32_t		status;

	void *		hfile;

	nt_iosb		iosb;

	uintptr_t	nread;

	uintptr_t	buffer[32768/sizeof(uintptr_t)];

	char *		ch;

	if ((status = ptyc_open_file(&hfile,hat,unit,true)))

		return status;

	status = ntapi->zw_read_file(

		hfile,

		0,0,0,

		&iosb,

		buffer,sizeof(buffer),

		0,0);

	while (status == NT_STATUS_SUCCESS) {

		ch    = (char *)buffer;

		nread = iosb.info;

		for ( ; nread; ) {

			status = ntapi->pty_write(

				dctx->cctx->hpts,

				hevent,0,0,&iosb,

				ch,(uint32_t)nread,

				0,0);

			if (status == NT_STATUS_PENDING)

				status = ntapi->zw_wait_for_single_object(

					hevent,NT_SYNC_ALERTABLE,0);

			if (status || iosb.status) {

				ntapi->zw_close(hfile);

				return status ? status : iosb.status;

			}

			ch    += iosb.info;

			nread -= iosb.info;

		}

		status = ntapi->zw_read_file(

			hfile,

			0,0,0,

			&iosb,

			buffer,sizeof(buffer),

			0,0);

	}

	ntapi->zw_close(hfile);

	return (status == NT_STATUS_END_OF_FILE)

		? NT_STATUS_SUCCESS

		: status;

}

int __stdcall ptyc_dbg_cat(struct ptyc_driver_ctx * dctx)

{

	int32_t		status;

	void *		hevent;

	const char **	punit;

	nt_rtdata *	rtdata;

	nt_peb * 	peb;

	void *		hat;

	if (!dctx->units[0])

		return 0;

	if ((status = ntapi->tt_create_private_event(

			&hevent,

			NT_NOTIFICATION_EVENT,

			NT_EVENT_NOT_SIGNALED)))

		return status;

	if ((status = ntapi->tt_get_runtime_data(&rtdata,0)))

		return status;

	if (!(peb = (nt_peb *)pe_get_peb_address()))

		return NT_STATUS_INTERNAL_ERROR;

	if (!peb->process_params)

		return NT_STATUS_INTERNAL_ERROR;

	hat = rtdata->hcwd

		? rtdata->hcwd

		: peb->process_params->cwd_handle;

	for (punit=dctx->units, status=0; *punit && !status; punit++)

		status = ptyc_cat(

			dctx,hat,

			*punit,

			hevent);

	ntapi->zw_close(hevent);

	return status;

}