1acf1f710SJohn Baldwin /*- 2acf1f710SJohn Baldwin * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3acf1f710SJohn Baldwin * 4acf1f710SJohn Baldwin * Copyright (c) 2018 John H. Baldwin <jhb@FreeBSD.org> 5acf1f710SJohn Baldwin * All rights reserved. 6acf1f710SJohn Baldwin * 7acf1f710SJohn Baldwin * Redistribution and use in source and binary forms, with or without 8acf1f710SJohn Baldwin * modification, are permitted provided that the following conditions 9acf1f710SJohn Baldwin * are met: 10acf1f710SJohn Baldwin * 1. Redistributions of source code must retain the above copyright 11acf1f710SJohn Baldwin * notice, this list of conditions and the following disclaimer. 12acf1f710SJohn Baldwin * 2. Redistributions in binary form must reproduce the above copyright 13acf1f710SJohn Baldwin * notice, this list of conditions and the following disclaimer in the 14acf1f710SJohn Baldwin * documentation and/or other materials provided with the distribution. 15acf1f710SJohn Baldwin * 16acf1f710SJohn Baldwin * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17acf1f710SJohn Baldwin * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18acf1f710SJohn Baldwin * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19acf1f710SJohn Baldwin * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20acf1f710SJohn Baldwin * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21acf1f710SJohn Baldwin * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22acf1f710SJohn Baldwin * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23acf1f710SJohn Baldwin * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24acf1f710SJohn Baldwin * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25acf1f710SJohn Baldwin * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26acf1f710SJohn Baldwin * SUCH DAMAGE. 27acf1f710SJohn Baldwin */ 28acf1f710SJohn Baldwin 29acf1f710SJohn Baldwin #include <sys/cdefs.h> 30acf1f710SJohn Baldwin __FBSDID("$FreeBSD$"); 31acf1f710SJohn Baldwin 32acf1f710SJohn Baldwin #include <atf-c.h> 33acf1f710SJohn Baldwin #include <ucontext.h> 34acf1f710SJohn Baldwin 35acf1f710SJohn Baldwin static char uc_stack[16 * 1024]; 36acf1f710SJohn Baldwin 37acf1f710SJohn Baldwin static void 38acf1f710SJohn Baldwin check_1(int arg1) 39acf1f710SJohn Baldwin { 40acf1f710SJohn Baldwin 41acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg1, 1); 42acf1f710SJohn Baldwin } 43acf1f710SJohn Baldwin 44acf1f710SJohn Baldwin ATF_TC_WITHOUT_HEAD(makecontext_arg1); 45acf1f710SJohn Baldwin ATF_TC_BODY(makecontext_arg1, tc) 46acf1f710SJohn Baldwin { 47acf1f710SJohn Baldwin ucontext_t ctx[2]; 48acf1f710SJohn Baldwin 49acf1f710SJohn Baldwin ATF_REQUIRE_EQ(getcontext(&ctx[1]), 0); 50acf1f710SJohn Baldwin ctx[1].uc_stack.ss_sp = uc_stack; 51acf1f710SJohn Baldwin ctx[1].uc_stack.ss_size = sizeof(uc_stack); 52acf1f710SJohn Baldwin ctx[1].uc_link = &ctx[0]; 53acf1f710SJohn Baldwin makecontext(&ctx[1], (void (*)(void))check_1, 1, 1); 54acf1f710SJohn Baldwin 55acf1f710SJohn Baldwin ATF_REQUIRE_EQ(swapcontext(&ctx[0], &ctx[1]), 0); 56acf1f710SJohn Baldwin } 57acf1f710SJohn Baldwin 58acf1f710SJohn Baldwin static void 59acf1f710SJohn Baldwin check_2(int arg1, int arg2) 60acf1f710SJohn Baldwin { 61acf1f710SJohn Baldwin 62acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg1, 1); 63acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg2, 2); 64acf1f710SJohn Baldwin } 65acf1f710SJohn Baldwin 66acf1f710SJohn Baldwin ATF_TC_WITHOUT_HEAD(makecontext_arg2); 67acf1f710SJohn Baldwin ATF_TC_BODY(makecontext_arg2, tc) 68acf1f710SJohn Baldwin { 69acf1f710SJohn Baldwin ucontext_t ctx[2]; 70acf1f710SJohn Baldwin 71acf1f710SJohn Baldwin ATF_REQUIRE_EQ(getcontext(&ctx[1]), 0); 72acf1f710SJohn Baldwin ctx[1].uc_stack.ss_sp = uc_stack; 73acf1f710SJohn Baldwin ctx[1].uc_stack.ss_size = sizeof(uc_stack); 74acf1f710SJohn Baldwin ctx[1].uc_link = &ctx[0]; 75acf1f710SJohn Baldwin makecontext(&ctx[1], (void (*)(void))check_2, 2, 1, 2); 76acf1f710SJohn Baldwin 77acf1f710SJohn Baldwin ATF_REQUIRE_EQ(swapcontext(&ctx[0], &ctx[1]), 0); 78acf1f710SJohn Baldwin } 79acf1f710SJohn Baldwin 80acf1f710SJohn Baldwin static void 81acf1f710SJohn Baldwin check_3(int arg1, int arg2, int arg3) 82acf1f710SJohn Baldwin { 83acf1f710SJohn Baldwin 84acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg1, 1); 85acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg2, 2); 86acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg3, 3); 87acf1f710SJohn Baldwin } 88acf1f710SJohn Baldwin 89acf1f710SJohn Baldwin ATF_TC_WITHOUT_HEAD(makecontext_arg3); 90acf1f710SJohn Baldwin ATF_TC_BODY(makecontext_arg3, tc) 91acf1f710SJohn Baldwin { 92acf1f710SJohn Baldwin ucontext_t ctx[2]; 93acf1f710SJohn Baldwin 94acf1f710SJohn Baldwin ATF_REQUIRE_EQ(getcontext(&ctx[1]), 0); 95acf1f710SJohn Baldwin ctx[1].uc_stack.ss_sp = uc_stack; 96acf1f710SJohn Baldwin ctx[1].uc_stack.ss_size = sizeof(uc_stack); 97acf1f710SJohn Baldwin ctx[1].uc_link = &ctx[0]; 98acf1f710SJohn Baldwin makecontext(&ctx[1], (void (*)(void))check_3, 3, 1, 2, 3); 99acf1f710SJohn Baldwin 100acf1f710SJohn Baldwin ATF_REQUIRE_EQ(swapcontext(&ctx[0], &ctx[1]), 0); 101acf1f710SJohn Baldwin } 102acf1f710SJohn Baldwin 103acf1f710SJohn Baldwin static void 104acf1f710SJohn Baldwin check_4(int arg1, int arg2, int arg3, int arg4) 105acf1f710SJohn Baldwin { 106acf1f710SJohn Baldwin 107acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg1, 1); 108acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg2, 2); 109acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg3, 3); 110acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg4, 4); 111acf1f710SJohn Baldwin } 112acf1f710SJohn Baldwin 113acf1f710SJohn Baldwin ATF_TC_WITHOUT_HEAD(makecontext_arg4); 114acf1f710SJohn Baldwin ATF_TC_BODY(makecontext_arg4, tc) 115acf1f710SJohn Baldwin { 116acf1f710SJohn Baldwin ucontext_t ctx[2]; 117acf1f710SJohn Baldwin 118acf1f710SJohn Baldwin ATF_REQUIRE_EQ(getcontext(&ctx[1]), 0); 119acf1f710SJohn Baldwin ctx[1].uc_stack.ss_sp = uc_stack; 120acf1f710SJohn Baldwin ctx[1].uc_stack.ss_size = sizeof(uc_stack); 121acf1f710SJohn Baldwin ctx[1].uc_link = &ctx[0]; 122acf1f710SJohn Baldwin makecontext(&ctx[1], (void (*)(void))check_4, 4, 1, 2, 3, 4); 123acf1f710SJohn Baldwin 124acf1f710SJohn Baldwin ATF_REQUIRE_EQ(swapcontext(&ctx[0], &ctx[1]), 0); 125acf1f710SJohn Baldwin } 126acf1f710SJohn Baldwin 127acf1f710SJohn Baldwin static void 128acf1f710SJohn Baldwin check_5(int arg1, int arg2, int arg3, int arg4, int arg5) 129acf1f710SJohn Baldwin { 130acf1f710SJohn Baldwin 131acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg1, 1); 132acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg2, 2); 133acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg3, 3); 134acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg4, 4); 135acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg5, 5); 136acf1f710SJohn Baldwin } 137acf1f710SJohn Baldwin 138acf1f710SJohn Baldwin ATF_TC_WITHOUT_HEAD(makecontext_arg5); 139acf1f710SJohn Baldwin ATF_TC_BODY(makecontext_arg5, tc) 140acf1f710SJohn Baldwin { 141acf1f710SJohn Baldwin ucontext_t ctx[2]; 142acf1f710SJohn Baldwin 143acf1f710SJohn Baldwin ATF_REQUIRE_EQ(getcontext(&ctx[1]), 0); 144acf1f710SJohn Baldwin ctx[1].uc_stack.ss_sp = uc_stack; 145acf1f710SJohn Baldwin ctx[1].uc_stack.ss_size = sizeof(uc_stack); 146acf1f710SJohn Baldwin ctx[1].uc_link = &ctx[0]; 147acf1f710SJohn Baldwin makecontext(&ctx[1], (void (*)(void))check_5, 5, 1, 2, 3, 4, 5); 148acf1f710SJohn Baldwin 149acf1f710SJohn Baldwin ATF_REQUIRE_EQ(swapcontext(&ctx[0], &ctx[1]), 0); 150acf1f710SJohn Baldwin } 151acf1f710SJohn Baldwin 152acf1f710SJohn Baldwin static void 153acf1f710SJohn Baldwin check_6(int arg1, int arg2, int arg3, int arg4, int arg5, int arg6) 154acf1f710SJohn Baldwin { 155acf1f710SJohn Baldwin 156acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg1, 1); 157acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg2, 2); 158acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg3, 3); 159acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg4, 4); 160acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg5, 5); 161acf1f710SJohn Baldwin ATF_REQUIRE_EQ(arg6, 6); 162acf1f710SJohn Baldwin } 163acf1f710SJohn Baldwin 164acf1f710SJohn Baldwin ATF_TC_WITHOUT_HEAD(makecontext_arg6); 165acf1f710SJohn Baldwin ATF_TC_BODY(makecontext_arg6, tc) 166acf1f710SJohn Baldwin { 167acf1f710SJohn Baldwin ucontext_t ctx[2]; 168acf1f710SJohn Baldwin 169acf1f710SJohn Baldwin ATF_REQUIRE_EQ(getcontext(&ctx[1]), 0); 170acf1f710SJohn Baldwin ctx[1].uc_stack.ss_sp = uc_stack; 171acf1f710SJohn Baldwin ctx[1].uc_stack.ss_size = sizeof(uc_stack); 172acf1f710SJohn Baldwin ctx[1].uc_link = &ctx[0]; 173acf1f710SJohn Baldwin makecontext(&ctx[1], (void (*)(void))check_6, 6, 1, 2, 3, 4, 5, 6); 174acf1f710SJohn Baldwin 175acf1f710SJohn Baldwin ATF_REQUIRE_EQ(swapcontext(&ctx[0], &ctx[1]), 0); 176acf1f710SJohn Baldwin } 177acf1f710SJohn Baldwin 178acf1f710SJohn Baldwin ATF_TP_ADD_TCS(tp) 179acf1f710SJohn Baldwin { 180acf1f710SJohn Baldwin 181acf1f710SJohn Baldwin ATF_TP_ADD_TC(tp, makecontext_arg1); 182acf1f710SJohn Baldwin ATF_TP_ADD_TC(tp, makecontext_arg2); 183acf1f710SJohn Baldwin ATF_TP_ADD_TC(tp, makecontext_arg3); 184acf1f710SJohn Baldwin ATF_TP_ADD_TC(tp, makecontext_arg4); 185acf1f710SJohn Baldwin ATF_TP_ADD_TC(tp, makecontext_arg5); 186acf1f710SJohn Baldwin ATF_TP_ADD_TC(tp, makecontext_arg6); 187acf1f710SJohn Baldwin 188acf1f710SJohn Baldwin return (atf_no_error()); 189acf1f710SJohn Baldwin } 190