public inbox for glibc-cvs@sourceware.org help / color / mirror / Atom feed
From: H.J. Lu <hjl@sourceware.org> To: glibc-cvs@sourceware.org Subject: [glibc] Add a setjmp/longjmp test between user contexts Date: Mon, 1 Jan 2024 23:56:46 +0000 (GMT) [thread overview] Message-ID: <20240101235647.174FF385840D@sourceware.org> (raw) https://sourceware.org/git/gitweb.cgi?p=glibc.git;h=8d8ae5eebd2c81064cfe88d877940ea4e6046d16 commit 8d8ae5eebd2c81064cfe88d877940ea4e6046d16 Author: H.J. Lu <hjl.tools@gmail.com> Date: Wed Dec 20 07:34:42 2023 -0800 Add a setjmp/longjmp test between user contexts Verify that setjmp and longjmp work correctly between user contexts. Arrange stacks for uctx_func1 and uctx_func2 so that ____longjmp_chk works when setjmp and longjmp are called from different user contexts. Reviewed-by: Noah Goldstein <goldstein.w.n@gmail.com> Diff: --- stdlib/Makefile | 1 + stdlib/tst-swapcontext2.c | 138 ++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 139 insertions(+) diff --git a/stdlib/Makefile b/stdlib/Makefile index d0173d06a4..f166ec6c80 100644 --- a/stdlib/Makefile +++ b/stdlib/Makefile @@ -256,6 +256,7 @@ tests := \ tst-strtol-locale \ tst-strtoll \ tst-swapcontext1 \ + tst-swapcontext2 \ tst-thread-quick_exit \ tst-tininess \ tst-unsetenv1 \ diff --git a/stdlib/tst-swapcontext2.c b/stdlib/tst-swapcontext2.c new file mode 100644 index 0000000000..f679755649 --- /dev/null +++ b/stdlib/tst-swapcontext2.c @@ -0,0 +1,138 @@ +/* Check setjmp/longjmp between user contexts. + Copyright (C) 2023 Free Software Foundation, Inc. + This file is part of the GNU C Library. + + The GNU C Library is free software; you can redistribute it and/or + modify it under the terms of the GNU Lesser General Public + License as published by the Free Software Foundation; either + version 2.1 of the License, or (at your option) any later version. + + The GNU C Library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Lesser General Public License for more details. + + You should have received a copy of the GNU Lesser General Public + License along with the GNU C Library; if not, see + <https://www.gnu.org/licenses/>. */ + +#include <stdio.h> +#include <stdlib.h> +#include <setjmp.h> +#include <ucontext.h> +#include <stackinfo.h> + +static ucontext_t uctx_main, uctx_func1, uctx_func2; +const char *str1 = "\e[31mswapcontext(&uctx_func1, &uctx_main)\e[0m"; +const char *str2 = "\e[34mswapcontext(&uctx_func2, &uctx_main)\e[0m"; +const char *fmt1 = "\e[31m"; +const char *fmt2 = "\e[34m"; +static jmp_buf jmpbuf; + +#define handle_error(msg) \ + do { perror(msg); exit(EXIT_FAILURE); } while (0) + +__attribute__((noinline, noclone)) +static void +func4(ucontext_t *uocp, ucontext_t *ucp, const char *str, const char *fmt) +{ + printf(" %sfunc4: %s\e[0m\n", fmt, str); + if (swapcontext(uocp, ucp) == -1) + handle_error("swapcontext"); + printf(" %sfunc4: returning\e[0m\n", fmt); +} + +__attribute__((noinline, noclone)) +static void +func3(ucontext_t *uocp, ucontext_t *ucp, const char *str, const char *fmt) +{ + printf(" %sfunc3: func4(uocp, ucp, str)\e[0m\n", fmt); + func4(uocp, ucp, str, fmt); + printf(" %sfunc3: returning\e[0m\n", fmt); +} + +__attribute__((noinline, noclone)) +static void +func1(void) +{ + while ( 1 ) + { + printf(" \e[31mfunc1: func3(&uctx_func1, &uctx_main, str1)\e[0m\n"); + if (setjmp (jmpbuf) != 0) + { + printf(" %sfunc1: from longjmp\e[0m\n", fmt1); + exit (EXIT_SUCCESS); + } + func3( &uctx_func1, &uctx_main, str1, fmt1); + } +} + +__attribute__((noinline, noclone)) +static void +func2(void) +{ + while ( 1 ) + { + printf(" \e[34mfunc2: func3(&uctx_func2, &uctx_main, str2)\e[0m\n"); + func3(&uctx_func2, &uctx_main, str2, fmt2); + printf(" %sfunc2: calling longjmp\e[0m\n", fmt2); + longjmp (jmpbuf, 1); + } +} + +static int +do_test (void) +{ + /* ____longjmp_chk has */ +#if 0 +#ifdef _STACK_GROWS_DOWN +#define called_from(this, saved) ((this) < (saved)) +#else +#define called_from(this, saved) ((this) > (saved)) +#endif + + /* If "env" is from a frame that called us, we're all set. */ + if (called_from(this_frame, saved_frame)) + __longjmp (env, val); +#endif + /* Arrange stacks for uctx_func1 and uctx_func2 so that called_from + is true when setjmp is called from uctx_func1 and longjmp is called + from uctx_func2. */ +#ifdef _STACK_GROWS_DOWN +# define UCTX_FUNC1_STACK 1 +# define UCTX_FUNC2_STACK 0 +#else +# define UCTX_FUNC1_STACK 0 +# define UCTX_FUNC2_STACK 1 +#endif + char func_stack[2][16384]; + int i; + + if (getcontext(&uctx_func1) == -1) + handle_error("getcontext"); + uctx_func1.uc_stack.ss_sp = func_stack[UCTX_FUNC1_STACK]; + uctx_func1.uc_stack.ss_size = sizeof (func_stack[UCTX_FUNC1_STACK]); + uctx_func1.uc_link = &uctx_main; + makecontext(&uctx_func1, func1, 0); + + if (getcontext(&uctx_func2) == -1) + handle_error("getcontext"); + uctx_func2.uc_stack.ss_sp = func_stack[UCTX_FUNC2_STACK]; + uctx_func2.uc_stack.ss_size = sizeof (func_stack[UCTX_FUNC2_STACK]); + uctx_func2.uc_link = &uctx_func1; + makecontext(&uctx_func2, func2, 0); + + for ( i = 0; i < 2; i++ ) + { + if (swapcontext(&uctx_main, &uctx_func1) == -1) + handle_error("swapcontext"); + printf(" \e[35mmain: swapcontext(&uctx_main, &uctx_func2)\n\e[0m"); + if (swapcontext(&uctx_main, &uctx_func2) == -1) + handle_error("swapcontext"); + printf(" \e[35mmain: swapcontext(&uctx_main, &uctx_func1)\n\e[0m"); + } + + exit(EXIT_FAILURE); +} + +#include <support/test-driver.c>
reply other threads:[~2024-01-01 23:56 UTC|newest] Thread overview: [no followups] expand[flat|nested] mbox.gz Atom feed
Reply instructions: You may reply publicly to this message via plain-text email using any one of the following methods: * Save the following mbox file, import it into your mail client, and reply-to-all from there: mbox Avoid top-posting and favor interleaved quoting: https://en.wikipedia.org/wiki/Posting_style#Interleaved_style * Reply using the --to, --cc, and --in-reply-to switches of git-send-email(1): git send-email \ --in-reply-to=20240101235647.174FF385840D@sourceware.org \ --to=hjl@sourceware.org \ --cc=glibc-cvs@sourceware.org \ /path/to/YOUR_REPLY https://kernel.org/pub/software/scm/git/docs/git-send-email.html * If your mail client supports setting the In-Reply-To header via mailto: links, try the mailto: linkBe sure your reply has a Subject: header at the top and a blank line before the message body.
This is a public inbox, see mirroring instructions for how to clone and mirror all data and code used for this inbox; as well as URLs for read-only IMAP folder(s) and NNTP newsgroup(s).