diff mbox series

syscalls/prctl10: Add basic test for PR_SET/GET_TSC

Message ID 1657106160-2126-1-git-send-email-xuyang2018.jy@fujitsu.com
State New
Headers show
Series syscalls/prctl10: Add basic test for PR_SET/GET_TSC | expand

Commit Message

xuyang2018.jy@fujitsu.com July 6, 2022, 11:16 a.m. UTC
Signed-off-by: Yang Xu <xuyang2018.jy@fujitsu.com>
---
 include/lapi/prctl.h                       |   7 ++
 runtest/syscalls                           |   1 +
 testcases/kernel/syscalls/prctl/.gitignore |   1 +
 testcases/kernel/syscalls/prctl/prctl10.c  | 112 +++++++++++++++++++++
 4 files changed, 121 insertions(+)
 create mode 100644 testcases/kernel/syscalls/prctl/prctl10.c

Comments

xuyang2018.jy@fujitsu.com Aug. 15, 2022, 8:28 a.m. UTC | #1
Hi All

Any comment?

Best Regards
Yang Xu
> Signed-off-by: Yang Xu <xuyang2018.jy@fujitsu.com>
> ---
>   include/lapi/prctl.h                       |   7 ++
>   runtest/syscalls                           |   1 +
>   testcases/kernel/syscalls/prctl/.gitignore |   1 +
>   testcases/kernel/syscalls/prctl/prctl10.c  | 112 +++++++++++++++++++++
>   4 files changed, 121 insertions(+)
>   create mode 100644 testcases/kernel/syscalls/prctl/prctl10.c
> 
> diff --git a/include/lapi/prctl.h b/include/lapi/prctl.h
> index fa5922231..8d3ef5c32 100644
> --- a/include/lapi/prctl.h
> +++ b/include/lapi/prctl.h
> @@ -19,6 +19,13 @@
>   # define PR_SET_SECCOMP  22
>   #endif
>   
> +#ifndef PR_SET_TSC
> +# define PR_GET_TSC 25
> +# define PR_SET_TSC 26
> +# define PR_TSC_ENABLE  1
> +# define PR_TSC_SIGSEGV 2
> +#endif
> +
>   #ifndef PR_SET_TIMERSLACK
>   # define PR_SET_TIMERSLACK 29
>   # define PR_GET_TIMERSLACK 30
> diff --git a/runtest/syscalls b/runtest/syscalls
> index 36fc50aeb..a0935821a 100644
> --- a/runtest/syscalls
> +++ b/runtest/syscalls
> @@ -1004,6 +1004,7 @@ prctl06 prctl06
>   prctl07 prctl07
>   prctl08 prctl08
>   prctl09 prctl09
> +prctl10 prctl10
>   
>   pread01 pread01
>   pread01_64 pread01_64
> diff --git a/testcases/kernel/syscalls/prctl/.gitignore b/testcases/kernel/syscalls/prctl/.gitignore
> index 0f2c9b194..50ee4bf60 100644
> --- a/testcases/kernel/syscalls/prctl/.gitignore
> +++ b/testcases/kernel/syscalls/prctl/.gitignore
> @@ -8,3 +8,4 @@
>   /prctl07
>   /prctl08
>   /prctl09
> +/prctl10
> diff --git a/testcases/kernel/syscalls/prctl/prctl10.c b/testcases/kernel/syscalls/prctl/prctl10.c
> new file mode 100644
> index 000000000..1b6791679
> --- /dev/null
> +++ b/testcases/kernel/syscalls/prctl/prctl10.c
> @@ -0,0 +1,112 @@
> +// SPDX-License-Identifier: GPL-2.0-or-later
> +/*
> + * Copyright (c) 2022 FUJITSU LIMITED. All rights reserved.
> + * Author: Yang Xu <xuyang2018.jy@fujitsu.com>
> + */
> +
> +/*\
> + * [Description]
> + *
> + * Basic test to test behaviour of PR_GET_TSC and PR_SET_TSC.
> + *
> + * Set the state of the flag determining whether the timestamp counter can
> + * be read by the process.
> + *
> + * - Pass PR_TSC_ENABLE to arg2 to allow it to be read.
> + * - Pass PR_TSC_SIGSEGV to arg2 to generate a SIGSEGV when reading TSC.
> + */
> +
> +#include <sys/prctl.h>
> +#include <string.h>
> +#include <stdio.h>
> +#include <setjmp.h>
> +#include "tst_test.h"
> +#include "lapi/prctl.h"
> +
> +#define TCASE_ENTRY(tsc_read_stat) { .name = #tsc_read_stat, .read_stat = tsc_read_stat}
> +
> +static int pass;
> +static sigjmp_buf env;
> +
> +static const char *tsc_read_stat_names[] = {
> +	[0] = "[not set]",
> +	[PR_TSC_ENABLE] = "PR_TSC_ENABLE",
> +	[PR_TSC_SIGSEGV] = "PR_TSC_SIGSEGV",
> +};
> +
> +static struct tcase {
> +	char *name;
> +	int read_stat;
> +} tcases[] = {
> +	TCASE_ENTRY(PR_TSC_ENABLE),
> +	TCASE_ENTRY(PR_TSC_SIGSEGV)
> +};
> +
> +static void sighandler(int sig LTP_ATTRIBUTE_UNUSED)
> +{
> +	pass = 1;
> +	TST_EXP_PASS_SILENT(prctl(PR_SET_TSC, PR_TSC_ENABLE));
> +	siglongjmp(env, 1);
> +}
> +
> +static uint64_t rdtsc(void)
> +{
> +	uint32_t lo, hi;
> +	/* We cannot use "=A", since this would use %rax on x86_64 */
> +	__asm__ __volatile__ ("rdtsc" : "=a" (lo), "=d" (hi));
> +	return (uint64_t)hi << 32 | lo;
> +}
> +
> +static void verify_prctl(unsigned int n)
> +{
> +	struct tcase *tc = &tcases[n];
> +	unsigned long long time1, time2;
> +	int tsc_val = 0;
> +
> +	TST_EXP_PASS_SILENT(prctl(PR_SET_TSC, tc->read_stat));
> +	TST_EXP_PASS_SILENT(prctl(PR_GET_TSC, &tsc_val));
> +	if (tsc_val == tc->read_stat)
> +		tst_res(TPASS, "current state is %s(%d)",
> +				tc->name, tc->read_stat);
> +	else
> +		tst_res(TFAIL, "current state is %s(%d), expect %s(%d)",
> +				tsc_read_stat_names[tsc_val], tsc_val, tc->name, tc->read_stat);
> +
> +	if (tc->read_stat == PR_TSC_SIGSEGV) {
> +		if (sigsetjmp(env, 1) == 0)
> +			rdtsc();
> +
> +		if (pass)
> +			tst_res(TPASS,
> +				"get SIGSEGV signal under PR_TSC_SIGSEGV situation");
> +		else
> +			tst_res(TFAIL,
> +				"don't get SIGSEGV signal under PR_TSC_SIGSEGV situation");
> +		pass = 0;
> +	}
> +
> +	time1 = rdtsc();
> +	time2 = rdtsc();
> +	if (time2 > time1)
> +		tst_res(TPASS, "rdtsc works correctly, %lld ->%lld",
> +			time1, time2);
> +	else
> +		tst_res(TFAIL, "rdtsc works incorrectly, %lld ->%lld",
> +			time1, time2);
> +}
> +
> +static void setup(void)
> +{
> +	SAFE_SIGNAL(SIGSEGV, sighandler);
> +}
> +
> +static struct tst_test test = {
> +	.setup = setup,
> +	.test = verify_prctl,
> +	.tcnt = ARRAY_SIZE(tcases),
> +	.supported_archs = (const char *const []) {
> +		"x86",
> +		"x86_64",
> +		NULL
> +	},
> +};
diff mbox series

Patch

diff --git a/include/lapi/prctl.h b/include/lapi/prctl.h
index fa5922231..8d3ef5c32 100644
--- a/include/lapi/prctl.h
+++ b/include/lapi/prctl.h
@@ -19,6 +19,13 @@ 
 # define PR_SET_SECCOMP  22
 #endif
 
+#ifndef PR_SET_TSC
+# define PR_GET_TSC 25
+# define PR_SET_TSC 26
+# define PR_TSC_ENABLE  1
+# define PR_TSC_SIGSEGV 2
+#endif
+
 #ifndef PR_SET_TIMERSLACK
 # define PR_SET_TIMERSLACK 29
 # define PR_GET_TIMERSLACK 30
diff --git a/runtest/syscalls b/runtest/syscalls
index 36fc50aeb..a0935821a 100644
--- a/runtest/syscalls
+++ b/runtest/syscalls
@@ -1004,6 +1004,7 @@  prctl06 prctl06
 prctl07 prctl07
 prctl08 prctl08
 prctl09 prctl09
+prctl10 prctl10
 
 pread01 pread01
 pread01_64 pread01_64
diff --git a/testcases/kernel/syscalls/prctl/.gitignore b/testcases/kernel/syscalls/prctl/.gitignore
index 0f2c9b194..50ee4bf60 100644
--- a/testcases/kernel/syscalls/prctl/.gitignore
+++ b/testcases/kernel/syscalls/prctl/.gitignore
@@ -8,3 +8,4 @@ 
 /prctl07
 /prctl08
 /prctl09
+/prctl10
diff --git a/testcases/kernel/syscalls/prctl/prctl10.c b/testcases/kernel/syscalls/prctl/prctl10.c
new file mode 100644
index 000000000..1b6791679
--- /dev/null
+++ b/testcases/kernel/syscalls/prctl/prctl10.c
@@ -0,0 +1,112 @@ 
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Copyright (c) 2022 FUJITSU LIMITED. All rights reserved.
+ * Author: Yang Xu <xuyang2018.jy@fujitsu.com>
+ */
+
+/*\
+ * [Description]
+ *
+ * Basic test to test behaviour of PR_GET_TSC and PR_SET_TSC.
+ *
+ * Set the state of the flag determining whether the timestamp counter can
+ * be read by the process.
+ *
+ * - Pass PR_TSC_ENABLE to arg2 to allow it to be read.
+ * - Pass PR_TSC_SIGSEGV to arg2 to generate a SIGSEGV when reading TSC.
+ */
+
+#include <sys/prctl.h>
+#include <string.h>
+#include <stdio.h>
+#include <setjmp.h>
+#include "tst_test.h"
+#include "lapi/prctl.h"
+
+#define TCASE_ENTRY(tsc_read_stat) { .name = #tsc_read_stat, .read_stat = tsc_read_stat}
+
+static int pass;
+static sigjmp_buf env;
+
+static const char *tsc_read_stat_names[] = {
+	[0] = "[not set]",
+	[PR_TSC_ENABLE] = "PR_TSC_ENABLE",
+	[PR_TSC_SIGSEGV] = "PR_TSC_SIGSEGV",
+};
+
+static struct tcase {
+	char *name;
+	int read_stat;
+} tcases[] = {
+	TCASE_ENTRY(PR_TSC_ENABLE),
+	TCASE_ENTRY(PR_TSC_SIGSEGV)
+};
+
+static void sighandler(int sig LTP_ATTRIBUTE_UNUSED)
+{
+	pass = 1;
+	TST_EXP_PASS_SILENT(prctl(PR_SET_TSC, PR_TSC_ENABLE));
+	siglongjmp(env, 1);
+}
+
+static uint64_t rdtsc(void)
+{
+	uint32_t lo, hi;
+	/* We cannot use "=A", since this would use %rax on x86_64 */
+	__asm__ __volatile__ ("rdtsc" : "=a" (lo), "=d" (hi));
+	return (uint64_t)hi << 32 | lo;
+}
+
+static void verify_prctl(unsigned int n)
+{
+	struct tcase *tc = &tcases[n];
+	unsigned long long time1, time2;
+	int tsc_val = 0;
+
+	TST_EXP_PASS_SILENT(prctl(PR_SET_TSC, tc->read_stat));
+	TST_EXP_PASS_SILENT(prctl(PR_GET_TSC, &tsc_val));
+	if (tsc_val == tc->read_stat)
+		tst_res(TPASS, "current state is %s(%d)",
+				tc->name, tc->read_stat);
+	else
+		tst_res(TFAIL, "current state is %s(%d), expect %s(%d)",
+				tsc_read_stat_names[tsc_val], tsc_val, tc->name, tc->read_stat);
+
+	if (tc->read_stat == PR_TSC_SIGSEGV) {
+		if (sigsetjmp(env, 1) == 0)
+			rdtsc();
+
+		if (pass)
+			tst_res(TPASS,
+				"get SIGSEGV signal under PR_TSC_SIGSEGV situation");
+		else
+			tst_res(TFAIL,
+				"don't get SIGSEGV signal under PR_TSC_SIGSEGV situation");
+		pass = 0;
+	}
+
+	time1 = rdtsc();
+	time2 = rdtsc();
+	if (time2 > time1)
+		tst_res(TPASS, "rdtsc works correctly, %lld ->%lld",
+			time1, time2);
+	else
+		tst_res(TFAIL, "rdtsc works incorrectly, %lld ->%lld",
+			time1, time2);
+}
+
+static void setup(void)
+{
+	SAFE_SIGNAL(SIGSEGV, sighandler);
+}
+
+static struct tst_test test = {
+	.setup = setup,
+	.test = verify_prctl,
+	.tcnt = ARRAY_SIZE(tcases),
+	.supported_archs = (const char *const []) {
+		"x86",
+		"x86_64",
+		NULL
+	},
+};