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

/*  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(const struct ptyc_common_ctx * cctx)

{

	nt_guid pty_guid = TTY_PTM_GUID;

	return ptyc_pty_open(

		0,&pty_guid,

		(nt_pty **)&cctx->hptm);

}

static int32_t ptyc_pts_open(const 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,

		(nt_pty **)&cctx->hpts);

}

int ptyc_alloc_pty(struct ptyc_driver_ctx * dctx)

{

	int32_t status;

	if (dctx->cctx->hpts || dctx->cctx->hptm)

		return ptyc_set_status(

			dctx,NT_STATUS_DEVICE_ALREADY_ATTACHED);

	if ((status = ptyc_ptm_open(dctx->cctx)))

		return ptyc_set_status(

			dctx,status);

	if (!(dctx->cctx->drvflags & PTYC_DRIVER_DBG_OVEN)

			&& !(dctx->cctx->drvflags & PTYC_DRIVER_DBG_RAW))

		return ptyc_set_status(

			dctx,NT_STATUS_SUCCESS);

	if ((status = ptyc_pts_open(dctx->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_common_ctx * cctx;

	cctx = (struct ptyc_common_ctx *)dctx->cctx;

	if (cctx->hpts) {

		ntapi->pty_close(cctx->hpts);

		cctx->hpts = 0;

	}

	if (cctx->hptm) {

		ntapi->pty_close(cctx->hptm);

		cctx->hptm = 0;

	}

}