/*********************************************************/
/* toksvc: a framework-native token broker service */
/* Copyright (C) 2020 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.TOKSVC. */
/*********************************************************/
#include <ntapi/ntapi.h>
#include <stdio.h>
extern const ntapi_vtbl * toks_ntapi;
int toks_sprintf(char * str, const char * fmt, ...)
{
int ret;
va_list ap;
va_start(ap, fmt);
ret = toks_ntapi->vsprintf(str, fmt, ap);
va_end(ap);
return ret;
}
int toks_snprintf(char * str, size_t n, const char * fmt, ...)
{
int ret;
va_list ap;
va_start(ap, fmt);
ret = toks_ntapi->vsnprintf(str, n, fmt, ap);
va_end(ap);
return ret;
}
int toks_isatty(int fildes)
{
nt_runtime_data * rtdata;
if ((toks_ntapi->tt_get_runtime_data(&rtdata,0)))
return 0;
if (fildes == 0)
return (rtdata->stdin_type == NT_FILE_TYPE_PTY);
else if (fildes == 1)
return (rtdata->stdout_type == NT_FILE_TYPE_PTY);
else if (fildes == 2)
return (rtdata->stderr_type == NT_FILE_TYPE_PTY);
else
return 0;
}
int toks_write(int fd, const void * buf, size_t size)
{
int32_t status;
nt_runtime_data * rtdata;
ntapi_zw_write_file * iofn;
void * hio;
void * hevent;
int fdtype;
nt_iosb iosb;
/* size */
if (size >= 0x80000000)
return NT_STATUS_INVALID_PARAMETER;
/* rtdata */
if (toks_ntapi->tt_get_runtime_data(&rtdata,0))
return NT_STATUS_REINITIALIZATION_NEEDED;
/* hio, io type */
if (fd == STDIN_FILENO) {
hio = rtdata->hstdin;
fdtype = rtdata->stdin_type;
} else if (fd == STDOUT_FILENO) {
hio = rtdata->hstdout;
fdtype = rtdata->stdout_type;
} else if (fd == STDERR_FILENO) {
hio = rtdata->hstderr;
fdtype = rtdata->stderr_type;
} else {
return NT_STATUS_INVALID_PARAMETER;
}
if (!hio)
return NT_STATUS_INVALID_HANDLE;
/* iofn */
iofn = (fdtype == NT_FILE_TYPE_PTY)
? (ntapi_zw_write_file *)toks_ntapi->pty_write
: toks_ntapi->zw_write_file;
/* hevent */
if ((status = toks_ntapi->tt_create_private_event(
&hevent,
NT_NOTIFICATION_EVENT,
NT_EVENT_NOT_SIGNALED)))
return status;
/* iowrite */
iosb.status = NT_STATUS_PENDING;
status = iofn(
hio,hevent,
0,0,&iosb,
(void *)buf,size,
0,0);
/* wait */
if (status == NT_STATUS_PENDING)
toks_ntapi->zw_wait_for_single_object(
hevent,0,0);
/* hevent */
toks_ntapi->zw_close(hevent);
/* ret */
switch (iosb.status) {
case NT_STATUS_SUCCESS:
return iosb.info;
default:
return (iosb.status < 0)
? iosb.status
: -iosb.status;
}
}