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 <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  */
  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  */
  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
  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);