From 3ac00fe9c3396f07a1416ff7fc855f6f9a3c4304 Mon Sep 17 00:00:00 2001
From: Jing Chen <chjing@google.com>
Date: Fri, 6 Nov 2020 18:36:03 -0800
Subject: Implement command GETNCNT for semctl.

PiperOrigin-RevId: 341154192
---
 pkg/sentry/syscalls/linux/sys_sem.go | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

(limited to 'pkg/sentry/syscalls/linux/sys_sem.go')

diff --git a/pkg/sentry/syscalls/linux/sys_sem.go b/pkg/sentry/syscalls/linux/sys_sem.go
index 067f6be6d..e383a0a87 100644
--- a/pkg/sentry/syscalls/linux/sys_sem.go
+++ b/pkg/sentry/syscalls/linux/sys_sem.go
@@ -139,14 +139,17 @@ func Semctl(t *kernel.Task, args arch.SyscallArguments) (uintptr, *kernel.Syscal
 		return 0, nil, err
 
 	case linux.GETZCNT:
-		v, err := getSemzcnt(t, id, num)
+		v, err := getZCnt(t, id, num)
+		return uintptr(v), nil, err
+
+	case linux.GETNCNT:
+		v, err := getNCnt(t, id, num)
 		return uintptr(v), nil, err
 
 	case linux.IPC_INFO,
 		linux.SEM_INFO,
 		linux.SEM_STAT,
-		linux.SEM_STAT_ANY,
-		linux.GETNCNT:
+		linux.SEM_STAT_ANY:
 
 		t.Kernel().EmitUnimplementedEvent(t)
 		fallthrough
@@ -262,12 +265,22 @@ func getPID(t *kernel.Task, id int32, num int32) (int32, error) {
 	return int32(tg.ID()), nil
 }
 
-func getSemzcnt(t *kernel.Task, id int32, num int32) (uint16, error) {
+func getZCnt(t *kernel.Task, id int32, num int32) (uint16, error) {
+	r := t.IPCNamespace().SemaphoreRegistry()
+	set := r.FindByID(id)
+	if set == nil {
+		return 0, syserror.EINVAL
+	}
+	creds := auth.CredentialsFromContext(t)
+	return set.CountZeroWaiters(num, creds)
+}
+
+func getNCnt(t *kernel.Task, id int32, num int32) (uint16, error) {
 	r := t.IPCNamespace().SemaphoreRegistry()
 	set := r.FindByID(id)
 	if set == nil {
 		return 0, syserror.EINVAL
 	}
 	creds := auth.CredentialsFromContext(t)
-	return set.GetZeroWaiters(num, creds)
+	return set.CountNegativeWaiters(num, creds)
 }
-- 
cgit v1.2.3