[LTP] [PATCH v2] sched_setattr/sched_setattr01: Add new testcase for sched_setattr

[Cui Bixuan <cuibixuan@huawei.com>]

Add new testcase for sched_setattr

Signed-off-by: Cui Bixuan <cuibixuan@huawei.com>
---
V2:
    * Delete #include "linux_syscall_numbers.h";
    * Change run_deadline() to do_test();
    * Initialize structure attr at the beginning in sched_setattr01.c;
    * Delete 'errno = 0' in sched_setattr_verify();
    * Change some print message

 runtest/sched                                      |    1 +
 runtest/syscalls                                   |    1 +
 testcases/kernel/syscalls/.gitignore               |    1 +
 testcases/kernel/syscalls/sched_setattr/Makefile   |   25 ++++
 .../syscalls/sched_setattr/sched_setattr01.c       |  136 ++++++++++++++++++++
 5 files changed, 164 insertions(+), 0 deletions(-)
 create mode 100644 testcases/kernel/syscalls/sched_setattr/Makefile
 create mode 100644 testcases/kernel/syscalls/sched_setattr/sched_setattr01.c

diff --git a/runtest/sched b/runtest/sched
index 4b8a1df..89398df 100644
--- a/runtest/sched
+++ b/runtest/sched
@@ -9,6 +9,7 @@ trace_sched01		trace_sched -c 1
 hackbench01 hackbench 50 process 1000
 hackbench02 hackbench 20 thread 1000

+sched_setattr01 sched_setattr01
 sched_getattr01 sched_getattr01
 sched_getattr02 sched_getattr02

diff --git a/runtest/syscalls b/runtest/syscalls
index 724ab66..469baee 100644
--- a/runtest/syscalls
+++ b/runtest/syscalls
@@ -922,6 +922,7 @@ sched_yield01 sched_yield01
 sched_setaffinity01 sched_setaffinity01
 sched_getaffinity01 sched_getaffinity01

+sched_setattr01 sched_setattr01
 sched_getattr01 sched_getattr01
 sched_getattr02 sched_getattr02

diff --git a/testcases/kernel/syscalls/.gitignore b/testcases/kernel/syscalls/.gitignore
index aba112b..c53f0d4 100644
--- a/testcases/kernel/syscalls/.gitignore
+++ b/testcases/kernel/syscalls/.gitignore
@@ -752,6 +752,7 @@
 /sched_rr_get_interval/sched_rr_get_interval02
 /sched_rr_get_interval/sched_rr_get_interval03
 /sched_setaffinity/sched_setaffinity01
+/sched_setattr/sched_setattr01
 /sched_setparam/sched_setparam01
 /sched_setparam/sched_setparam02
 /sched_setparam/sched_setparam03
diff --git a/testcases/kernel/syscalls/sched_setattr/Makefile b/testcases/kernel/syscalls/sched_setattr/Makefile
new file mode 100644
index 0000000..30718a9
--- /dev/null
+++ b/testcases/kernel/syscalls/sched_setattr/Makefile
@@ -0,0 +1,25 @@
+#
+#  Copyright (c) Huawei Technologies Co., Ltd., 2015
+#
+#  This program is free software;  you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation; either version 2 of the License, or
+#  (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY;  without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
+#  the GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program;  if not, write to the Free Software
+#  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#
+
+top_srcdir		?= ../../../..
+
+include $(top_srcdir)/include/mk/testcases.mk
+
+CFLAGS			+= -pthread
+
+include $(top_srcdir)/include/mk/generic_leaf_target.mk
diff --git a/testcases/kernel/syscalls/sched_setattr/sched_setattr01.c b/testcases/kernel/syscalls/sched_setattr/sched_setattr01.c
new file mode 100644
index 0000000..63cf890
--- /dev/null
+++ b/testcases/kernel/syscalls/sched_setattr/sched_setattr01.c
@@ -0,0 +1,136 @@
+/*
+ * Copyright (c) Huawei Technologies Co., Ltd., 2015
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
+ * the GNU General Public License for more details.
+ */
+ /* Description:
+ *   Verify that:
+ *              1) sched_setattr succeed with correct parameters
+ *              2) sched_setattr fails with unused pid
+ *              3) sched_setattr fails with invalid address
+ *              4) sched_setattr fails with invalid flag
+ */
+
+#define _GNU_SOURCE
+#include <unistd.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <linux/unistd.h>
+#include <linux/kernel.h>
+#include <linux/types.h>
+#include <sys/syscall.h>
+#include <pthread.h>
+#include <sched.h>
+#include <errno.h>
+
+#include "test.h"
+#include "lapi/sched.h"
+
+char *TCID = "sched_setattr01";
+
+#define SCHED_DEADLINE	6
+#define RUNTIME_VAL 10000000
+#define PERIOD_VAL 30000000
+#define DEADLINE_VAL 30000000
+
+static pid_t pid;
+static pid_t unused_pid;
+
+static struct sched_attr attr = {
+	.size = sizeof(struct sched_attr),
+	.sched_flags = 0,
+	.sched_nice = 0,
+	.sched_priority = 0,
+
+	.sched_policy = SCHED_DEADLINE,
+	.sched_runtime = RUNTIME_VAL,
+	.sched_period = PERIOD_VAL,
+	.sched_deadline = DEADLINE_VAL,
+};
+
+static struct test_case {
+	pid_t *pid;
+	struct sched_attr *a;
+	unsigned int flags;
+	int exp_return;
+	int exp_errno;
+} test_cases[] = {
+	{&pid, &attr, 0, 0, 0},
+	{&unused_pid, &attr, 0, -1, ESRCH},
+	{&pid, NULL, 0, -1, EINVAL},
+	{&pid, &attr, 1000, -1, EINVAL}
+};
+
+static void setup(void);
+static void sched_setattr_verify(const struct test_case *test);
+
+int TST_TOTAL = ARRAY_SIZE(test_cases);
+
+void *do_test(void *data LTP_ATTRIBUTE_UNUSED)
+{
+	int i;
+
+	for (i = 0; i < TST_TOTAL; i++)
+		sched_setattr_verify(&test_cases[i]);
+
+	return NULL;
+}
+
+static void sched_setattr_verify(const struct test_case *test)
+{
+	TEST(sched_setattr(*(test->pid), test->a, test->flags));
+
+	if (TEST_RETURN != test->exp_return) {
+		tst_resm(TFAIL | TTERRNO, "sched_setattr() return "
+			"unexpectedly: expected: %d", test->exp_return);
+		return;
+	}
+
+	if (TEST_ERRNO == test->exp_errno) {
+		tst_resm(TPASS | TTERRNO,
+			"sched_setattr() works as expected");
+		return;
+	}
+
+	tst_resm(TFAIL | TTERRNO, "sched_setattr() return unexpectedly "
+		": expected: %d - %s",
+		test->exp_errno, tst_strerrno(test->exp_errno));
+}
+
+int main(int argc, char **argv)
+{
+	pthread_t thread;
+	int lc;
+
+	tst_parse_opts(argc, argv, NULL, NULL);
+
+	setup();
+
+	for (lc = 0; TEST_LOOPING(lc); lc++) {
+		pthread_create(&thread, NULL, do_test, NULL);
+		pthread_join(thread, NULL);
+	}
+
+	tst_exit();
+}
+
+void setup(void)
+{
+	unused_pid = tst_get_unused_pid(setup);
+
+	tst_require_root();
+
+	if ((tst_kvercmp(3, 14, 0)) < 0)
+		tst_brkm(TCONF, NULL, "EDF needs kernel 3.14 or higher");
+
+	TEST_PAUSE;
+}
-- 
1.6.0.2 .