/*********************************************************/
/*  ptycon: a pty-console bridge                         */
/*  Copyright (C) 2016  Z. Gilboa                        */
/*  Released under GPLv2 and GPLv3; see COPYING.PTYCON.  */
/*********************************************************/

#include <psxtypes/psxtypes.h>
#include <ntapi/ntapi.h>
#include <ptycon/ptycon.h>
#include "ptycon_driver_impl.h"
#include "ptycon_ioctl_impl.h"
#include "ptycon_status_impl.h"

static int32_t ptyc_pty_open(nt_pty * hptm, nt_guid * pty_guid, nt_pty ** hpty)
{
	nt_vfd_dev_name		pty_name;
	nt_oa			oa = {sizeof(oa),0,0,0,0,0};

	ntapi->vfd_dev_name_init(
		&pty_name,pty_guid);

	oa.obj_name = &pty_name.name;
	oa.root_dir = hptm;

	return ntapi->pty_open(
		0,hpty,
		NT_FILE_ALL_ACCESS,
		&oa,0,0,0);
}

static int32_t ptyc_ptm_open(struct ptyc_common_ctx * cctx)
{
	nt_guid pty_guid = TTY_PTM_GUID;

	return ptyc_pty_open(
		0,&pty_guid,
		&cctx->hptm);
}

static int32_t ptyc_pts_open(struct ptyc_common_ctx * cctx)
{
	int32_t status;
	nt_guid pty_guid = TTY_PTS_GUID;

	if ((status = ptyc_grant(cctx->hptm)))
		return status;

	return ptyc_pty_open(
		cctx->hptm,&pty_guid,
		&cctx->hpts);
}

int ptyc_alloc_pty(struct ptyc_driver_ctx * dctx)
{
	int32_t				status;
	struct ptyc_driver_ctx_impl *	ictx;
	struct ptyc_common_ctx *	cctx;

	if (!(ictx = ptyc_get_driver_ictx(dctx)))
		return NT_STATUS_INVALID_HANDLE;

	cctx = &ictx->cctx;

	if (cctx->hpts || cctx->hptm)
		return ptyc_set_status(
			dctx,NT_STATUS_DEVICE_ALREADY_ATTACHED);

	if ((status = ptyc_ptm_open(cctx)))
		return ptyc_set_status(
			dctx,status);

	if (!dctx->units[0]
			&& !(cctx->drvflags & PTYC_DRIVER_DBG_OVEN)
			&& !(cctx->drvflags & PTYC_DRIVER_DBG_RAW))
		return ptyc_set_status(
			dctx,NT_STATUS_SUCCESS);

	if ((status = ptyc_pts_open(cctx)))
		return ptyc_set_status(
			dctx,status);

	return ptyc_set_status(
		dctx,NT_STATUS_SUCCESS);
}

void ptyc_free_pty(struct ptyc_driver_ctx * dctx)
{
	struct ptyc_driver_ctx_impl * ictx;

	if (!(ictx = ptyc_get_driver_ictx(dctx)))
		return;

	if (ictx->cctx.hpts) {
		ntapi->pty_close(ictx->cctx.hpts);
		ictx->cctx.hpts = 0;
	}

	if (ictx->cctx.hptm) {
		ntapi->pty_close(ictx->cctx.hptm);
		ictx->cctx.hptm = 0;
	}
}