1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Test setjmp(), longjmp()
4 *
5 * Copyright (c) 2021, Heinrich Schuchardt <xypron.glpk@gmx.de>
6 */
7
8 #include <common.h>
9 #include <test/lib.h>
10 #include <test/test.h>
11 #include <test/ut.h>
12 #include <asm/setjmp.h>
13
14 struct test_jmp_buf {
15 jmp_buf env;
16 int val;
17 };
18
19 /**
20 * test_longjmp() - test longjmp function
21 *
22 * @i is passed to longjmp.
23 * @i << 8 is set in the environment structure.
24 *
25 * @env: environment
26 * @i: value passed to longjmp()
27 */
test_longjmp(struct test_jmp_buf * env,int i)28 static noinline void test_longjmp(struct test_jmp_buf *env, int i)
29 {
30 env->val = i << 8;
31 longjmp(env->env, i);
32 }
33
34 /**
35 * test_setjmp() - test setjmp function
36 *
37 * setjmp() will return the value @i passed to longjmp() if @i is non-zero.
38 * For @i == 0 we expect return value 1.
39 *
40 * @i << 8 will be set by test_longjmp in the environment structure.
41 * This value can be used to check that the stack frame is restored.
42 *
43 * We return the XORed values to allow simply check both at once.
44 *
45 * @i: value passed to longjmp()
46 * Return: values return by longjmp()
47 */
test_setjmp(int i)48 static int test_setjmp(int i)
49 {
50 struct test_jmp_buf env;
51 int ret;
52
53 env.val = -1;
54 ret = setjmp(env.env);
55 if (ret)
56 return ret ^ env.val;
57 test_longjmp(&env, i);
58 /* We should not arrive here */
59 return 0x1000;
60 }
61
lib_test_longjmp(struct unit_test_state * uts)62 static int lib_test_longjmp(struct unit_test_state *uts)
63 {
64 int i;
65
66 for (i = -3; i < 0; ++i)
67 ut_asserteq(i ^ (i << 8), test_setjmp(i));
68 ut_asserteq(1, test_setjmp(0));
69 for (i = 1; i < 4; ++i)
70 ut_asserteq(i ^ (i << 8), test_setjmp(i));
71 return 0;
72 }
73 LIB_TEST(lib_test_longjmp, 0);
74