1 files changed, 117 insertions, 0 deletions
diff --git a/src/sem/ntapi_sem_set.c b/src/sem/ntapi_sem_set.c
new file mode 100644
index 0000000..813de30
--- /dev/null
+++ b/src/sem/ntapi_sem_set.c
@@ -0,0 +1,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);
+}