test/lib/longjmp.c

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