[PATCH] test: self: add setjmp/longjmp/initjmp test

[PATCH] test: self: add setjmp/longjmp/initjmp test

[Ahmad Fatoum]

We depend on sjlj for OP-TEE early loading on ARM32 as well as for
the bthread infrastructure, which powers the state machine behind the
non-blocking USB mass storage gadget implementation.

Let's add a simple test to verify everything continues to work as
expected.

Signed-off-by: Ahmad Fatoum <a.fatoum@pengutronix.de>
---
 test/self/Kconfig  |  5 +++
 test/self/Makefile |  1 +
 test/self/setjmp.c | 98 ++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 104 insertions(+)
 create mode 100644 test/self/setjmp.c

diff --git a/test/self/Kconfig b/test/self/Kconfig
index d1ca6a701df3..dcbfa85896f7 100644
--- a/test/self/Kconfig
+++ b/test/self/Kconfig
@@ -40,6 +40,7 @@ config SELFTEST_ENABLE_ALL
 	imply SELFTEST_MMU
 	imply SELFTEST_REGULATOR
 	imply SELFTEST_STRING
+	imply SELFTEST_SETJMP
 	help
 	  Selects all self-tests compatible with current configuration
 
@@ -87,4 +88,8 @@ config SELFTEST_STRING
 	bool "String library selftest"
 	select VERSION_CMP
 
+config SELFTEST_SETJMP
+	bool "setjmp/longjmp library selftest"
+	depends on ARCH_HAS_SJLJ
+
 endif
diff --git a/test/self/Makefile b/test/self/Makefile
index a66f34671e5a..b6f4395147dd 100644
--- a/test/self/Makefile
+++ b/test/self/Makefile
@@ -12,6 +12,7 @@ obj-$(CONFIG_SELFTEST_JSON) += json.o
 obj-$(CONFIG_SELFTEST_DIGEST) += digest.o
 obj-$(CONFIG_SELFTEST_MMU) += mmu.o
 obj-$(CONFIG_SELFTEST_STRING) += string.o
+obj-$(CONFIG_SELFTEST_SETJMP) += setjmp.o
 
 clean-files := *.dtb *.dtb.S .*.dtc .*.pre .*.dts *.dtb.z
 clean-files += *.dtbo *.dtbo.S .*.dtso
diff --git a/test/self/setjmp.c b/test/self/setjmp.c
new file mode 100644
index 000000000000..1cbb44bdf941
--- /dev/null
+++ b/test/self/setjmp.c
@@ -0,0 +1,98 @@
+// SPDX-License-Identifier: MIT
+// SPDX-FileCopyrightText: Copyright (C) 2003 Hewlett-Packard
+/*
+ * Code contributed by David Mosberger-Tang <davidm@hpl.hp.com>
+ * to libunwind - a platform-independent unwind library
+ */
+
+#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
+
+#include <common.h>
+#include <bselftest.h>
+#include <string.h>
+#include <asm/setjmp.h>
+
+BSELFTEST_GLOBALS();
+
+static __noreturn void raise_longjmp(jmp_buf jbuf, int i, int n)
+{
+	while (i < n)
+		raise_longjmp(jbuf, i + 1, n);
+
+	longjmp(jbuf, n);
+}
+
+static jmp_buf jbuf;
+
+static void __noreturn initjmp_entry(void)
+{
+	volatile u32 arr[256];
+	int i;
+
+	for (i = 0; i < ARRAY_SIZE(arr); i++)
+		writel(i, &arr[i]);
+
+	/* ensure arr[] is allocated on stack */
+	OPTIMIZER_HIDE_VAR(i);
+	if (i == 0)
+		initjmp_entry();
+
+	longjmp(jbuf, 0x1337);
+}
+
+static void test_setjmp(void)
+{
+	void *stack;
+	jmp_buf ijbuf;
+	volatile int i;
+	int ret;
+
+	total_tests += 20;
+
+	for (i = 0; i < 10; ++i) {
+		if ((ret = setjmp(jbuf))) {
+			pr_debug("%s: secondary setjmp() return, ret=%d\n",
+				 __func__, ret);
+
+			if (ret != i + 1) {
+				printf("%s: setjmp() returned %d, expected %d\n",
+					 __func__, ret, i + 1);
+				failed_tests += 2;
+			}
+			continue;
+		}
+
+		pr_debug("%s.%d: done with setjmp(); calling children\n",
+			 __func__, i + 1);
+
+		raise_longjmp(jbuf, 0, i + 1);
+
+		printf("%s: raise_longjmp() returned unexpectedly\n", __func__);
+		failed_tests++;
+	}
+
+	stack = memalign(16, CONFIG_STACK_SIZE);
+	if (WARN_ON(!stack)) {
+		skipped_tests++;
+		return;
+	}
+
+	pr_debug("%s: testing initjmp\n", __func__);
+
+	total_tests += 1;
+
+	/* initialize new context with fresh stack */
+	initjmp(ijbuf, initjmp_entry, stack + CONFIG_STACK_SIZE);
+
+	switch (setjmp(jbuf)) {
+	case 0x1337:
+		break;
+	case 0:
+		longjmp(ijbuf, 0x42);
+	default:
+		failed_tests++;
+	}
+
+	free(stack);
+}
+bselftest(core, test_setjmp);
-- 
2.39.2

Re: [PATCH] test: self: add setjmp/longjmp/initjmp test

[Sascha Hauer]

On Mon, Jun 05, 2023 at 08:27:55AM +0200, Ahmad Fatoum wrote:
> We depend on sjlj for OP-TEE early loading on ARM32 as well as for
> the bthread infrastructure, which powers the state machine behind the
> non-blocking USB mass storage gadget implementation.
> 
> Let's add a simple test to verify everything continues to work as
> expected.
> 
> Signed-off-by: Ahmad Fatoum <a.fatoum@pengutronix.de>
> ---
>  test/self/Kconfig  |  5 +++
>  test/self/Makefile |  1 +
>  test/self/setjmp.c | 98 ++++++++++++++++++++++++++++++++++++++++++++++
>  3 files changed, 104 insertions(+)
>  create mode 100644 test/self/setjmp.c

Applied, thanks

Sascha

> 
> diff --git a/test/self/Kconfig b/test/self/Kconfig
> index d1ca6a701df3..dcbfa85896f7 100644
> --- a/test/self/Kconfig
> +++ b/test/self/Kconfig
> @@ -40,6 +40,7 @@ config SELFTEST_ENABLE_ALL
>  	imply SELFTEST_MMU
>  	imply SELFTEST_REGULATOR
>  	imply SELFTEST_STRING
> +	imply SELFTEST_SETJMP
>  	help
>  	  Selects all self-tests compatible with current configuration
>  
> @@ -87,4 +88,8 @@ config SELFTEST_STRING
>  	bool "String library selftest"
>  	select VERSION_CMP
>  
> +config SELFTEST_SETJMP
> +	bool "setjmp/longjmp library selftest"
> +	depends on ARCH_HAS_SJLJ
> +
>  endif
> diff --git a/test/self/Makefile b/test/self/Makefile
> index a66f34671e5a..b6f4395147dd 100644
> --- a/test/self/Makefile
> +++ b/test/self/Makefile
> @@ -12,6 +12,7 @@ obj-$(CONFIG_SELFTEST_JSON) += json.o
>  obj-$(CONFIG_SELFTEST_DIGEST) += digest.o
>  obj-$(CONFIG_SELFTEST_MMU) += mmu.o
>  obj-$(CONFIG_SELFTEST_STRING) += string.o
> +obj-$(CONFIG_SELFTEST_SETJMP) += setjmp.o
>  
>  clean-files := *.dtb *.dtb.S .*.dtc .*.pre .*.dts *.dtb.z
>  clean-files += *.dtbo *.dtbo.S .*.dtso
> diff --git a/test/self/setjmp.c b/test/self/setjmp.c
> new file mode 100644
> index 000000000000..1cbb44bdf941
> --- /dev/null
> +++ b/test/self/setjmp.c
> @@ -0,0 +1,98 @@
> +// SPDX-License-Identifier: MIT
> +// SPDX-FileCopyrightText: Copyright (C) 2003 Hewlett-Packard
> +/*
> + * Code contributed by David Mosberger-Tang <davidm@hpl.hp.com>
> + * to libunwind - a platform-independent unwind library
> + */
> +
> +#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
> +
> +#include <common.h>
> +#include <bselftest.h>
> +#include <string.h>
> +#include <asm/setjmp.h>
> +
> +BSELFTEST_GLOBALS();
> +
> +static __noreturn void raise_longjmp(jmp_buf jbuf, int i, int n)
> +{
> +	while (i < n)
> +		raise_longjmp(jbuf, i + 1, n);
> +
> +	longjmp(jbuf, n);
> +}
> +
> +static jmp_buf jbuf;
> +
> +static void __noreturn initjmp_entry(void)
> +{
> +	volatile u32 arr[256];
> +	int i;
> +
> +	for (i = 0; i < ARRAY_SIZE(arr); i++)
> +		writel(i, &arr[i]);
> +
> +	/* ensure arr[] is allocated on stack */
> +	OPTIMIZER_HIDE_VAR(i);
> +	if (i == 0)
> +		initjmp_entry();
> +
> +	longjmp(jbuf, 0x1337);
> +}
> +
> +static void test_setjmp(void)
> +{
> +	void *stack;
> +	jmp_buf ijbuf;
> +	volatile int i;
> +	int ret;
> +
> +	total_tests += 20;
> +
> +	for (i = 0; i < 10; ++i) {
> +		if ((ret = setjmp(jbuf))) {
> +			pr_debug("%s: secondary setjmp() return, ret=%d\n",
> +				 __func__, ret);
> +
> +			if (ret != i + 1) {
> +				printf("%s: setjmp() returned %d, expected %d\n",
> +					 __func__, ret, i + 1);
> +				failed_tests += 2;
> +			}
> +			continue;
> +		}
> +
> +		pr_debug("%s.%d: done with setjmp(); calling children\n",
> +			 __func__, i + 1);
> +
> +		raise_longjmp(jbuf, 0, i + 1);
> +
> +		printf("%s: raise_longjmp() returned unexpectedly\n", __func__);
> +		failed_tests++;
> +	}
> +
> +	stack = memalign(16, CONFIG_STACK_SIZE);
> +	if (WARN_ON(!stack)) {
> +		skipped_tests++;
> +		return;
> +	}
> +
> +	pr_debug("%s: testing initjmp\n", __func__);
> +
> +	total_tests += 1;
> +
> +	/* initialize new context with fresh stack */
> +	initjmp(ijbuf, initjmp_entry, stack + CONFIG_STACK_SIZE);
> +
> +	switch (setjmp(jbuf)) {
> +	case 0x1337:
> +		break;
> +	case 0:
> +		longjmp(ijbuf, 0x42);
> +	default:
> +		failed_tests++;
> +	}
> +
> +	free(stack);
> +}
> +bselftest(core, test_setjmp);
> -- 
> 2.39.2
> 
> 
> 

-- 
Pengutronix e.K.                           |                             |
Steuerwalder Str. 21                       | http://www.pengutronix.de/  |
31137 Hildesheim, Germany                  | Phone: +49-5121-206917-0    |
Amtsgericht Hildesheim, HRA 2686           | Fax:   +49-5121-206917-5555 |