[LTP] [PATCH v2] syscalls/nice05: Add testcase for nice() syscall
Zhao Gongyi
zhaogongyi@huawei.com
Sat Jun 18 03:40:14 CEST 2022
Add test verifies that the low nice thread executes more
time than the high nice thread since the two thread binded
on the same cpu.
Signed-off-by: Zhao Gongyi <zhaogongyi@huawei.com>
---
runtest/syscalls | 1 +
testcases/kernel/syscalls/nice/.gitignore | 1 +
testcases/kernel/syscalls/nice/Makefile | 5 +
testcases/kernel/syscalls/nice/nice05.c | 196 ++++++++++++++++++++++
4 files changed, 203 insertions(+)
create mode 100644 testcases/kernel/syscalls/nice/nice05.c
diff --git a/runtest/syscalls b/runtest/syscalls
index 88c31db06..a1f741d6a 100644
--- a/runtest/syscalls
+++ b/runtest/syscalls
@@ -901,6 +901,7 @@ nice01 nice01
nice02 nice02
nice03 nice03
nice04 nice04
+nice05 nice05
open01 open01
open01A symlink01 -T open01
diff --git a/testcases/kernel/syscalls/nice/.gitignore b/testcases/kernel/syscalls/nice/.gitignore
index 9d7a1bb43..58d64779e 100644
--- a/testcases/kernel/syscalls/nice/.gitignore
+++ b/testcases/kernel/syscalls/nice/.gitignore
@@ -2,3 +2,4 @@
/nice02
/nice03
/nice04
+/nice05
diff --git a/testcases/kernel/syscalls/nice/Makefile b/testcases/kernel/syscalls/nice/Makefile
index 044619fb8..9b155295e 100644
--- a/testcases/kernel/syscalls/nice/Makefile
+++ b/testcases/kernel/syscalls/nice/Makefile
@@ -3,6 +3,11 @@
top_srcdir ?= ../../../..
+LTPLIBS = ltpnewipc
+
+nice05: CFLAGS += -pthread
+nice05: LTPLDLIBS += -lltpnewipc
+
include $(top_srcdir)/include/mk/testcases.mk
include $(top_srcdir)/include/mk/generic_leaf_target.mk
diff --git a/testcases/kernel/syscalls/nice/nice05.c b/testcases/kernel/syscalls/nice/nice05.c
new file mode 100644
index 000000000..cdaa96e97
--- /dev/null
+++ b/testcases/kernel/syscalls/nice/nice05.c
@@ -0,0 +1,196 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Copyright(c) 2022 Huawei Technologies Co., Ltd
+ * Author: Li Mengfei <limengfei4@huawei.com>
+ * Zhao Gongyi <zhaogongyi@huawei.com>
+ */
+
+/*\
+ * [Description]
+ *
+ * 1. Create a high nice thread and a low nice thread, the main
+ * thread wake them at the same time
+ * 2. Both threads run on the same CPU
+ * 3. Verify that the low nice thread executes more time than
+ * the high nice thread
+ */
+
+#define _GNU_SOURCE
+#include <pthread.h>
+#include <sys/types.h>
+#include <stdio.h>
+#include "tst_test.h"
+#include "tst_safe_pthread.h"
+#include "tst_safe_sysv_ipc.h"
+#include "lapi/syscalls.h"
+#include "libnewipc.h"
+
+static pthread_barrier_t barrier;
+static int some_cpu;
+static cpu_set_t *set;
+static pid_t *child_tid;
+static key_t shm_key;
+static int shm_id;
+
+static void set_nice(int nice_inc)
+{
+ int orig_nice;
+
+ orig_nice = SAFE_GETPRIORITY(PRIO_PROCESS, 0);
+ TEST(nice(nice_inc));
+
+ if (TST_RET != (orig_nice + nice_inc)) {
+ tst_brk(TFAIL | TTERRNO, "nice(%d) returned %li, expected %i",
+ nice_inc, TST_RET, orig_nice + nice_inc);
+ }
+
+ if (TST_ERR)
+ tst_brk(TFAIL | TTERRNO, "nice(%d) failed", nice_inc);
+
+}
+
+static void *nice_low_thread(void *arg)
+{
+ int number = 0;
+ int ret = 0;
+
+ child_tid[0] = tst_syscall(__NR_gettid);
+ set_nice(*(int *)arg);
+ ret = pthread_barrier_wait(&barrier);
+ if (ret != 0 && ret != PTHREAD_BARRIER_SERIAL_THREAD) {
+ tst_brk(TBROK, "pthread_barrier_wait() returned %s",
+ tst_strerrno(-ret));
+ }
+
+ while (1)
+ number++;
+
+ return NULL;
+}
+
+static void *nice_high_thread(void *arg)
+{
+ int number = 0;
+ int ret = 0;
+
+ child_tid[1] = tst_syscall(__NR_gettid);
+ set_nice(*(int *)arg);
+ ret = pthread_barrier_wait(&barrier);
+ if (ret != 0 && ret != PTHREAD_BARRIER_SERIAL_THREAD) {
+ tst_brk(TBROK, "pthread_barrier_wait() returned %s",
+ tst_strerrno(-ret));
+ }
+
+ while (1)
+ number++;
+
+ return NULL;
+}
+
+static void setup(void)
+{
+ size_t size;
+ int index;
+ int nrcpus = 1024;
+
+ set = CPU_ALLOC(nrcpus);
+ if (!set)
+ tst_brk(TBROK | TERRNO, "CPU_ALLOC()");
+
+ size = CPU_ALLOC_SIZE(nrcpus);
+ CPU_ZERO_S(size, set);
+ if (sched_getaffinity(0, size, set) < 0)
+ tst_brk(TBROK | TERRNO, "sched_getaffinity()");
+
+ for (index = 0; index < (int)size * 8; index++)
+ if (CPU_ISSET_S(index, size, set))
+ some_cpu = index;
+
+ CPU_ZERO_S(size, set);
+ CPU_SET_S(some_cpu, size, set);
+ if (sched_setaffinity(0, size, set) < 0)
+ tst_brk(TBROK | TERRNO, "sched_setaffinity()");
+
+ shm_key = GETIPCKEY();
+ shm_id = SAFE_SHMGET(shm_key, getpagesize(), 0666 | IPC_CREAT);
+ child_tid = SAFE_SHMAT(shm_id, 0, 0);
+}
+
+static void cleanup(void)
+{
+ if (set)
+ CPU_FREE(set);
+
+ if (child_tid)
+ SAFE_SHMDT(child_tid);
+
+ if (shm_id)
+ SAFE_SHMCTL(shm_id, IPC_RMID, 0);
+}
+
+static void verify_nice(void)
+{
+ int ret;
+ int nice_inc_high = -1;
+ int nice_inc_low = -2;
+ char buf[4096] = {0};
+ float sum_exec_runtime[2];
+ pthread_t nice_low, nice_high;
+ pid_t pid;
+
+ pid = SAFE_FORK();
+ if (!pid) {
+ ret = pthread_barrier_init(&barrier, NULL, 3);
+ if (ret != 0) {
+ tst_brk(TBROK, "pthread_barrier_init() returned %s",
+ tst_strerrno(-ret));
+ }
+
+ SAFE_PTHREAD_CREATE(&nice_high, NULL, nice_high_thread,
+ (void *)&nice_inc_high);
+ SAFE_PTHREAD_CREATE(&nice_low, NULL, nice_low_thread,
+ (void *)&nice_inc_low);
+
+ ret = pthread_barrier_wait(&barrier);
+ if (ret != 0 && ret != PTHREAD_BARRIER_SERIAL_THREAD) {
+ tst_brk(TBROK, "pthread_barrier_wait() returned %s",
+ tst_strerrno(-ret));
+ }
+
+ while (1)
+ sleep(1);
+ } else {
+ sleep(3);
+ SAFE_KILL(pid, SIGSTOP);
+
+ sprintf(buf, "/proc/%d/sched", child_tid[0]);
+ SAFE_FILE_SCANF(buf, "%*s %*s %*s %*s %*s"
+ "%*s %*s %*f %*s %*s %*f %*s %*s %f",
+ &sum_exec_runtime[0]);
+
+ sprintf(buf, "/proc/%d/sched", child_tid[1]);
+ SAFE_FILE_SCANF(buf, "%*s %*s %*s %*s %*s"
+ "%*s %*s %*f %*s %*s %*f %*s %*s %f",
+ &sum_exec_runtime[1]);
+
+ if (sum_exec_runtime[0] < sum_exec_runtime[1]) {
+ tst_brk(TFAIL, "nice_low_thread sum_exec_runtime: %f "
+ "nice_high_thread sum_exec_runtime: %f",
+ sum_exec_runtime[0], sum_exec_runtime[1]);
+ } else {
+ tst_res(TPASS, "executes more cycles "
+ "than the high nice thread");
+ }
+
+ SAFE_KILL(pid, SIGKILL);
+ SAFE_WAIT(NULL);
+ }
+}
+
+static struct tst_test test = {
+ .setup = setup,
+ .cleanup = cleanup,
+ .test_all = verify_nice,
+ .needs_root = 1,
+ .forks_child = 1,
+};
--
2.17.1
More information about the ltp
mailing list