diff --git a/src/internal/toksvc_driver_impl.h b/src/internal/toksvc_driver_impl.h index 12f9b14..fd018a3 100644 --- a/src/internal/toksvc_driver_impl.h +++ b/src/internal/toksvc_driver_impl.h @@ -50,4 +50,7 @@ static inline struct toks_driver_ctx_impl * toks_get_driver_ictx(const struct to return 0; } +int32_t toks_open_file(void ** hfile, void * hat, const char * arg, bool fprivate); +int32_t toks_open_dir(void ** hfile, void * hat, const char * arg, bool fprivate); + #endif diff --git a/src/internal/toksvc_open_impl.c b/src/internal/toksvc_open_impl.c new file mode 100644 index 0000000..a5095cf --- /dev/null +++ b/src/internal/toksvc_open_impl.c @@ -0,0 +1,107 @@ +/*********************************************************/ +/* toksvc: a framework-native token broker service */ +/* Copyright (C) 2020 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.TOKSVC. */ +/*********************************************************/ + +#include +#include + +#include +#include "toksvc_driver_impl.h" + +static int32_t toks_open( + void ** hfile, + void * hat, + const char * arg, + uint32_t options, + bool fprivate) +{ + int32_t status; + nt_oa oa; + nt_iosb iosb; + nt_unicode_string path; + nt_unicode_conversion_params_utf8_to_utf16 params = {0,0,0,0,0,0,0,0,0}; + wchar16_t buffer[4096]; + wchar16_t * wch; + size_t nbytes; + + /* utf-8 --> utf-16 */ + params.src = (const unsigned char *)arg; + params.src_size_in_bytes= ntapi->tt_string_null_offset_multibyte(arg); + params.dst = buffer; + params.dst_size_in_bytes= sizeof(buffer); + + if ((status = ntapi->uc_convert_unicode_stream_utf8_to_utf16(¶ms))) + return status; + + /* convenience */ + for (wch=buffer, nbytes=params.bytes_written; nbytes; ) { + if (*wch == '/') + *wch = '\\'; + + nbytes -= sizeof(wchar16_t); + wch++; + } + + /* path */ + path.maxlen = 0; + path.strlen = (uint16_t)params.bytes_written; + path.buffer = buffer; + + /* oa */ + oa.len = sizeof(nt_oa); + oa.root_dir = (buffer[0]=='\\') ? 0 : hat; + oa.obj_name = &path; + oa.obj_attr = fprivate ? 0 : NT_OBJ_INHERIT; + oa.sec_desc = 0; + oa.sec_qos = 0; + + /* open */ + return ntapi->zw_open_file( + hfile, + NT_SEC_SYNCHRONIZE | NT_FILE_READ_ATTRIBUTES | NT_FILE_READ_DATA, + &oa,&iosb, + NT_FILE_SHARE_READ | NT_FILE_SHARE_WRITE | NT_FILE_SHARE_DELETE, + options | NT_FILE_SYNCHRONOUS_IO_ALERT); +} + +static int32_t toks_open_volume( + void ** hfile, + void * hat, + const char * arg, + uint32_t options, + bool fprivate) +{ + int32_t status; + void * hvolume = 0; + wchar16_t drive = arg[0]; + + if ((arg[1]==':') && ((arg[2]=='\\') || (arg[2]=='/'))) { + if ((status = ntapi->tt_get_dos_drive_root_handle( + &hvolume,drive))) + return status; + + hat = hvolume; + arg = &arg[3]; + } + + status = toks_open( + hfile,hat,arg, + options,fprivate); + + if (hvolume) + ntapi->zw_close(hvolume); + + return status; +} + +int32_t toks_open_file(void ** hfile, void * hat, const char * arg, bool fprivate) +{ + return toks_open_volume(hfile,hat,arg,NT_FILE_NON_DIRECTORY_FILE,fprivate); +} + +int32_t toks_open_dir(void ** hfile, void * hat, const char * arg, bool fprivate) +{ + return toks_open_volume(hfile,hat,arg,NT_FILE_DIRECTORY_FILE,fprivate); +}