1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/********************************************************/
/* 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 int32_t __sem_set_return(
intptr_t * hlock,
int32_t status,
int32_t cmd)
{
if (cmd == NT_SEM_CMD_SETALL)
at_store(hlock,0);
return status;
}
int32_t __stdcall __ntapi_sem_set(
__in nt_sem_info * sem,
__out nt_io_status_block * iosb,
__in void * sem_info,
__in uint32_t sem_info_length,
__in int32_t sem_ipc_cmd)
{
int32_t status;
void * hsection;
void * secaddr;
size_t secsize;
nt_sem_info_msg msg;
intptr_t * hlock;
/* validate */
if (!iosb)
return NT_STATUS_INVALID_PARAMETER;
else if (!sem_info)
return NT_STATUS_INVALID_PARAMETER;
else if (!sem_info_length)
return NT_STATUS_INVALID_PARAMETER;
else if (sem_ipc_cmd != NT_SEM_CMD_SETALL)
if (sem_info_length != sizeof(nt_sem_info))
return NT_STATUS_INFO_LENGTH_MISMATCH;
/* section */
hsection = 0;
secaddr = 0;
secsize = 0;
/* SETALL */
if (sem_ipc_cmd == NT_SEM_CMD_SETALL) {
if (sem->section_addr) {
hsection = sem->section;
secaddr = sem->section_addr;
secsize = sem->section_size;
} else if ((status = __ntapi->ipc_init_section_by_port(
sem->hport,&hsection,
&secaddr,&secsize)))
return status;
/* data size */
if (secsize < sem_info_length)
return NT_STATUS_DATA_OVERRUN;
/* lock */
hlock = &(__ntapi_internals()->hlock);
if (at_locked_cas(hlock,0,1))
return NT_STATUS_RESOURCE_NOT_OWNED;
/* data copy */
__ntapi->tt_generic_memcpy(
secaddr,sem_info,
sem_info_length);
} else
hlock = 0;
/* msg */
__ntapi->tt_aligned_block_memset(
&msg,0,sizeof(msg));
msg.header.msg_type = NT_LPC_NEW_MESSAGE;
msg.header.data_size = sizeof(msg.data);
msg.header.msg_size = sizeof(msg);
msg.data.ttyinfo.opcode = NT_TTY_SEM_SET;
msg.data.seminfo.semcmd = sem_ipc_cmd;
msg.data.seminfo.semkey = sem->semkey;
msg.data.seminfo.semid = sem->semid;
msg.data.seminfo.semnum = sem->semnum;
msg.data.seminfo.semval = sem->semval;
msg.data.seminfo.section_addr = secaddr;
msg.data.seminfo.section_size = sem_info_length;
if ((status = __ntapi->zw_request_wait_reply_port(sem->hport,&msg,&msg)))
return __sem_set_return(hlock,status,sem_ipc_cmd);
else if (msg.data.ttyinfo.status)
return __sem_set_return(hlock,msg.data.ttyinfo.status,sem_ipc_cmd);
/* reply */
iosb->status = NT_STATUS_SUCCESS;
iosb->info = 0;
return __sem_set_return(hlock,NT_STATUS_SUCCESS,sem_ipc_cmd);
}
|
