diff options
author | midipix <writeonce@midipix.org> | 2017-07-30 21:21:21 +0000 |
---|---|---|
committer | midipix <writeonce@midipix.org> | 2017-07-31 08:43:36 -0400 |
commit | 9faf9a067cfeaa786ac1784fb8bf4eece49922e9 (patch) | |
tree | 46950830693387fa1223c43445b1f1c022408c80 /src/sem/ntapi_sem_connect.c | |
parent | 029ed17d66b232d5e8bea0b3ab543ec8331e563d (diff) | |
download | ntapi-9faf9a067cfeaa786ac1784fb8bf4eece49922e9.tar.bz2 ntapi-9faf9a067cfeaa786ac1784fb8bf4eece49922e9.tar.xz |
subsystem interfaces: integrated sysv semaphore client-side interfaces.
Diffstat (limited to 'src/sem/ntapi_sem_connect.c')
-rw-r--r-- | src/sem/ntapi_sem_connect.c | 335 |
1 files changed, 335 insertions, 0 deletions
diff --git a/src/sem/ntapi_sem_connect.c b/src/sem/ntapi_sem_connect.c new file mode 100644 index 0000000..32c3610 --- /dev/null +++ b/src/sem/ntapi_sem_connect.c @@ -0,0 +1,335 @@ +/********************************************************/ +/* ntapi: Native API core library */ +/* Copyright (C) 2013--2017 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */ +/********************************************************/ + +#include <psxtypes/psxtypes.h> +#include <ntapi/nt_file.h> +#include <ntapi/nt_string.h> +#include <ntapi/nt_atomic.h> +#include <ntapi/nt_port.h> +#include <ntapi/nt_ipc.h> +#include <ntapi/nt_sem.h> +#include <ntapi/ntapi.h> +#include "ntapi_impl.h" + +static const nt_guid g_sempid = NT_IPC_GUID_SEMPID; +static const wchar16_t p_sempid[6] = NT_IPC_OBJDIR_PREFIX_SEMPID; + +static int32_t __semctl_get_service_attr( + nt_rtdata * rtdata, + nt_tty_service_info * semctl) +{ + nt_iosb iosb; + + /* inherited runtime data? */ + if (rtdata->semctl_keys[0]) { + semctl->attr.ver_major = 0; + semctl->attr.ver_minor = 0; + semctl->attr.options = 0; + semctl->attr.flags = 0; + + semctl->attr.type = rtdata->semctl_type; + semctl->attr.subtype = rtdata->semctl_subtype; + + semctl->attr.keys.key[0] = rtdata->semctl_keys[0]; + semctl->attr.keys.key[1] = rtdata->semctl_keys[1]; + semctl->attr.keys.key[2] = rtdata->semctl_keys[2]; + semctl->attr.keys.key[3] = rtdata->semctl_keys[3]; + semctl->attr.keys.key[4] = rtdata->semctl_keys[4]; + semctl->attr.keys.key[5] = rtdata->semctl_keys[5]; + + __ntapi->tt_guid_copy( + &semctl->attr.guid, + &rtdata->semctl_guid); + + return NT_STATUS_SUCCESS; + } + + /* obtain service info */ + return __ntapi->tty_query_information_service( + 0,&iosb, + semctl,&(nt_guid)NT_PORT_GUID_SEMCTL, + 0,0); +} + +static int32_t __semctl_server_connect( + nt_rtdata * rtdata, + nt_tty_service_info * semctl) +{ + int32_t status; + + /* already cononected? */ + if (rtdata->hsemctl) + return NT_STATUS_SUCCESS; + + /* connect */ + if ((status = __ntapi->ipc_connect_by_attr( + &rtdata->hsemctl, + &semctl->attr))) + return status; + + /* update */ + rtdata->semctl_type = semctl->attr.type; + rtdata->semctl_subtype = semctl->attr.subtype; + + rtdata->semctl_keys[0] = semctl->attr.keys.key[0]; + rtdata->semctl_keys[1] = semctl->attr.keys.key[1]; + rtdata->semctl_keys[2] = semctl->attr.keys.key[2]; + rtdata->semctl_keys[3] = semctl->attr.keys.key[3]; + rtdata->semctl_keys[4] = semctl->attr.keys.key[4]; + rtdata->semctl_keys[5] = semctl->attr.keys.key[5]; + + __ntapi->tt_guid_copy( + &rtdata->semctl_guid, + &semctl->attr.guid); + + return NT_STATUS_SUCCESS; +} + +static int32_t __sempid_symlink_set( + nt_rtdata * rtdata, + nt_tty_service_info * semctl) +{ + int32_t status; + void * hpiddir; + nt_port_name svcname; + nt_unicode_string str; + + if (rtdata->hsempid) + return NT_STATUS_SUCCESS; + + if (!rtdata->hsempiddir) { + if ((status = __ntapi->tt_open_ipc_object_directory( + &hpiddir, + NT_DIRECTORY_ALL_ACCESS, + p_sempid,&g_sempid))) + return status; + + if (at_locked_cas((intptr_t *)&rtdata->hsempiddir,0,(intptr_t)hpiddir)) + __ntapi->zw_close(hpiddir); + } + + __ntapi->tt_port_name_from_attr( + &svcname,&semctl->attr); + + str.strlen = (uint16_t)(size_t)(&(((nt_port_name *)0)->null_termination)); + str.maxlen = sizeof(nt_port_name); + str.buffer = svcname.base_named_objects; + + return __ntapi->tt_create_ipc_object_directory_entry( + &rtdata->hsempid, + NT_SEC_STANDARD_RIGHTS_ALL, + rtdata->hsempiddir, + 0,&str, + pe_get_current_process_id()); + +} + +static int32_t __stdcall __sem_open( + void * hipc, + nt_sem_info * sem, + uint32_t access, + nt_object_attributes * oa, + nt_iosb * iosb, + uint32_t share, + uint32_t semslots, + uint32_t key, + uint32_t id, + uint32_t opcode) +{ + int32_t status; + nt_tty_port_msg msg; + nt_iosb siosb; + nt_tty_service_info semctl; + nt_runtime_data * rtdata; + + /* init */ + rtdata = (__ntapi_internals())->rtdata; + + /* semctl service attributes */ + if (!rtdata->hsempid) + if ((status = __semctl_get_service_attr(rtdata,&semctl))) + return status; + + /* semctl server */ + if ((status = __semctl_server_connect(rtdata,&semctl))) + return status; + + /* sempid symlink */ + if ((status = __sempid_symlink_set(rtdata,&semctl))) + return status; + + /* hipc */ + if (!hipc && (opcode == NT_TTY_SEM_ALLOC)) + hipc = (__ntapi_internals())->rtdata->hsemctl; + + /* obtain sem info */ + __ntapi->tt_aligned_block_memset( + &msg,0,sizeof(msg)); + + if (!iosb) + iosb = &siosb; + + msg.header.msg_type = NT_LPC_NEW_MESSAGE; + msg.header.data_size = sizeof(nt_sem_info_msg) - sizeof(msg.header); + msg.header.msg_size = sizeof(msg); + msg.ttyinfo.opcode = opcode; + + msg.seminfo.semkey = (int32_t)key; + msg.seminfo.semid = (int32_t)id; + msg.seminfo.semslots = semslots; + + msg.seminfo.ntattr = oa->obj_attr; + msg.seminfo.ntaccess = access; + msg.seminfo.ntshare = share; + + if ((status = __ntapi->zw_request_wait_reply_port(hipc,&msg,&msg))) + return status; + else if (msg.ttyinfo.status) + return msg.ttyinfo.status; + + iosb->info = sizeof(msg.svcinfo); + iosb->status = NT_STATUS_SUCCESS; + + /* new semaphore? */ + if (opcode == NT_TTY_SEM_ALLOC) + if ((status = __ntapi->ipc_connect_by_attr( + &hipc,&msg.svcinfo.attr))) + return status; + + /* all done */ + __ntapi->tt_aligned_block_memset( + (uintptr_t *)sem, + 0,sizeof(*sem)); + + sem->semkey = msg.svcinfo.key; + sem->semid = msg.svcinfo.id; + + sem->hport = hipc; + + return NT_STATUS_SUCCESS; +} + + +int32_t __ntapi_sem_create( + __in void * hport, + __out nt_sem_info * sem, + __in uint32_t access, + __in nt_object_attributes * oa, + __out nt_iosb * iosb, + __in uint32_t share, + __in uint32_t nslots) +{ + uint32_t key; + + /* validate */ + if (!oa->root_dir) + return NT_STATUS_DIRECTORY_SERVICE_REQUIRED; + + if (oa->obj_name && !oa->obj_name->strlen) { + key = 0; + + } else if (oa->obj_name) { + if (oa->obj_name->strlen != 8 * sizeof(wchar16_t)) + return NT_STATUS_OBJECT_NAME_INVALID; + + if (__ntapi->tt_hex_utf16_to_uint32(oa->obj_name->buffer,&key)) + return NT_STATUS_OBJECT_NAME_INVALID; + + } else { + key = 0; + } + + /* open semaphore */ + return __sem_open( + hport,sem,access, + oa,iosb,share,nslots, + key,0,NT_TTY_SEM_ALLOC); +} + + +int32_t __stdcall __ntapi_sem_open( + __in void * hport, + __out nt_sem_info * sem, + __in uint32_t access, + __in nt_object_attributes * oa, + __out nt_iosb * iosb, + __in uint32_t share, + __in uint32_t nslots) +{ + int32_t status; + uint32_t key; + uint32_t id; + void * hsymlink; + nt_oa ipcoa; + void * hipc; + + /* validate */ + if (!oa->root_dir) + return NT_STATUS_DIRECTORY_SERVICE_REQUIRED; + + if (!oa->obj_name) + return NT_STATUS_INVALID_PARAMETER; + + if (oa->obj_name->strlen != 8 * sizeof(wchar16_t)) + return NT_STATUS_OBJECT_NAME_INVALID; + + if (__ntapi->tt_hex_utf16_to_uint32(oa->obj_name->buffer,&key)) + return NT_STATUS_OBJECT_NAME_INVALID; + + /* open symlink */ + ipcoa.len = sizeof(ipcoa); + ipcoa.root_dir = oa->root_dir; + ipcoa.obj_name = oa->obj_name; + ipcoa.obj_attr = 0; + ipcoa.sec_desc = oa->sec_desc; + ipcoa.sec_qos = oa->sec_qos; + + status = __ntapi->zw_open_symbolic_link_object( + &hsymlink, + NT_SEC_STANDARD_RIGHTS_READ | NT_GENERIC_READ, + &ipcoa); + + switch (status) { + case NT_STATUS_SUCCESS: + break; + + case NT_STATUS_OBJECT_NAME_NOT_FOUND: + case NT_STATUS_OBJECT_PATH_NOT_FOUND: + if (oa->obj_attr & NT_OBJ_OPENIF) + return __sem_open( + hport,sem,access, + oa,iosb,share,nslots, + key,0,NT_TTY_SEM_ALLOC); + else + return status; + + default: + return status; + } + + /* ipc connect */ + status = __ntapi->ipc_connect_by_symlink( + &hipc,hsymlink); + + __ntapi->zw_close( + hsymlink); + + if (status) + return status; + + /* open by id? */ + if (oa->obj_attr & NT_OBJ_OPENLINK) { + id = key; + key = 0; + } else { + id = 0; + } + + return __sem_open( + hipc,sem,access, + oa,iosb,share,nslots, + key,id,NT_TTY_SEM_OPEN); +} |