|
 |
500bc3 |
/*********************************************************/
|
|
|
500bc3 |
/* toksvc: a framework-native token broker service */
|
|
|
d91fa0 |
/* Copyright (C) 2020 SysDeer Technologies, LLC */
|
|
|
500bc3 |
/* Released under GPLv2 and GPLv3; see COPYING.TOKSVC. */
|
|
|
500bc3 |
/*********************************************************/
|
|
|
500bc3 |
|
|
|
500bc3 |
#include <psxtypes/psxtypes.h>
|
|
|
500bc3 |
#include <ntapi/ntapi.h>
|
|
|
500bc3 |
|
|
|
500bc3 |
#include <toksvc/toksvc.h>
|
|
|
500bc3 |
#include "toksvc_daemon_impl.h"
|
|
|
500bc3 |
#include "toksvc_driver_impl.h"
|
|
|
500bc3 |
|
|
|
500bc3 |
#include <psxabi/sys_daemon.h>
|
|
|
500bc3 |
|
|
|
7927ff |
static const nt_guid g_pidany = NT_PROCESS_GUID_PIDANY;
|
|
|
7927ff |
static const wchar16_t p_pidany[6] = NT_PROCESS_OBJDIR_PREFIX_PIDANY;
|
|
|
500bc3 |
|
|
|
500bc3 |
static int32_t toks_daemon_pid_connect(
|
|
|
344692 |
struct toks_daemon_ctx * dctx,
|
|
|
344692 |
void ** hipc)
|
|
|
500bc3 |
{
|
|
|
500bc3 |
int32_t status;
|
|
|
500bc3 |
void * hpidany;
|
|
|
500bc3 |
void * hpiddir;
|
|
|
500bc3 |
nt_oa ipcoa;
|
|
|
500bc3 |
nt_unicode_string pidname;
|
|
|
500bc3 |
wchar16_t pidnamebuf[8];
|
|
|
500bc3 |
|
|
|
500bc3 |
/* hpiddir */
|
|
|
344692 |
if (!(hpiddir = toks_get_driver_hpiddir(dctx->driver_ctx))) {
|
|
|
500bc3 |
if ((status = ntapi->tt_open_ipc_object_directory(
|
|
|
500bc3 |
&hpiddir,
|
|
|
500bc3 |
NT_SEC_READ_CONTROL
|
|
|
500bc3 |
| NT_DIRECTORY_QUERY
|
|
|
500bc3 |
| NT_DIRECTORY_TRAVERSE
|
|
|
500bc3 |
| NT_DIRECTORY_CREATE_OBJECT
|
|
|
500bc3 |
| NT_DIRECTORY_CREATE_SUBDIRECTORY,
|
|
|
500bc3 |
p_pidany,&g_pidany)))
|
|
|
500bc3 |
return status;
|
|
|
500bc3 |
|
|
|
344692 |
toks_set_driver_hpiddir(dctx->driver_ctx,hpiddir);
|
|
|
500bc3 |
}
|
|
|
500bc3 |
|
|
|
500bc3 |
/* pid entry name */
|
|
|
500bc3 |
ntapi->tt_uint32_to_hex_utf16(
|
|
|
344692 |
dctx->reqtokpid,pidnamebuf);
|
|
|
500bc3 |
|
|
|
500bc3 |
pidname.strlen = sizeof(pidnamebuf);
|
|
|
500bc3 |
pidname.maxlen = sizeof(pidnamebuf);;
|
|
|
500bc3 |
pidname.buffer = pidnamebuf;
|
|
|
500bc3 |
|
|
|
500bc3 |
/* open symlink */
|
|
|
500bc3 |
ipcoa.len = sizeof(ipcoa);
|
|
|
500bc3 |
ipcoa.root_dir = hpiddir;
|
|
|
500bc3 |
ipcoa.obj_name = &pidname;
|
|
|
500bc3 |
ipcoa.obj_attr = 0;
|
|
|
500bc3 |
ipcoa.sec_desc = 0;
|
|
|
500bc3 |
ipcoa.sec_qos = 0;
|
|
|
500bc3 |
|
|
|
500bc3 |
if ((status = ntapi->zw_open_symbolic_link_object(
|
|
|
500bc3 |
&hpidany,
|
|
|
500bc3 |
NT_SEC_STANDARD_RIGHTS_READ | NT_GENERIC_READ,
|
|
|
500bc3 |
&ipcoa)))
|
|
|
500bc3 |
return status;
|
|
|
500bc3 |
|
|
|
500bc3 |
status = ntapi->ipc_connect_by_symlink(
|
|
|
a9d102 |
hipc,hpidany,false);
|
|
|
500bc3 |
|
|
|
500bc3 |
ntapi->zw_close(hpidany);
|
|
|
500bc3 |
|
|
|
500bc3 |
return status;
|
|
|
500bc3 |
}
|
|
|
500bc3 |
|
|
|
500bc3 |
int32_t toks_daemon_pidopen(struct toks_daemon_ctx * dctx)
|
|
|
500bc3 |
{
|
|
|
500bc3 |
int32_t status;
|
|
|
500bc3 |
nt_tty_session_msg msg;
|
|
|
500bc3 |
void * hipc;
|
|
|
500bc3 |
|
|
|
344692 |
if ((status = toks_daemon_pid_connect(dctx,&hipc)))
|
|
|
500bc3 |
return status;
|
|
|
500bc3 |
|
|
|
500bc3 |
ntapi->tt_aligned_block_memset(
|
|
|
500bc3 |
&msg,0,sizeof(msg));
|
|
|
500bc3 |
|
|
|
500bc3 |
msg.header.msg_type = NT_LPC_NEW_MESSAGE;
|
|
|
500bc3 |
msg.header.data_size = sizeof(msg.data);
|
|
|
500bc3 |
msg.header.msg_size = sizeof(msg);
|
|
|
500bc3 |
msg.data.ttyinfo.opcode = PSX_DAEMON_IOCTL_PIDQUERY;
|
|
|
500bc3 |
|
|
|
500bc3 |
if ((status = ntapi->zw_request_wait_reply_port(hipc,&msg,&msg)))
|
|
|
500bc3 |
return status;
|
|
|
500bc3 |
else if (msg.data.ttyinfo.status)
|
|
|
500bc3 |
return msg.data.ttyinfo.status;
|
|
|
500bc3 |
|
|
|
344692 |
dctx->reqsyspid = msg.data.sessioninfo.syspid;
|
|
|
500bc3 |
|
|
|
500bc3 |
return NT_STATUS_SUCCESS;
|
|
|
500bc3 |
}
|