[09/12] Add a test for multiple makecontext calls

Message ID 20180721142035.21059-10-hjl.tools@gmail.com
State New, archived
Headers

Commit Message

H.J. Lu July 21, 2018, 2:20 p.m. UTC
  Check multiple makecontext calls.

	* stdlib/Makefile (tests): Add tst-swapcontext2.
	* stdlib/tst-swapcontext2.c: New test.
---
 stdlib/Makefile           |   2 +-
 stdlib/tst-swapcontext2.c | 108 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 109 insertions(+), 1 deletion(-)
 create mode 100644 stdlib/tst-swapcontext2.c
  

Comments

Carlos O'Donell July 25, 2018, 3:31 a.m. UTC | #1
On 07/21/2018 10:20 AM, H.J. Lu wrote:
> Check multiple makecontext calls.
> 
> 	* stdlib/Makefile (tests): Add tst-swapcontext2.
> 	* stdlib/tst-swapcontext2.c: New test.

OK for 2.28.

Reviewed-by: Carlos O'Donell <carlos@redhat.com>

> ---
>  stdlib/Makefile           |   2 +-
>  stdlib/tst-swapcontext2.c | 108 ++++++++++++++++++++++++++++++++++++++
>  2 files changed, 109 insertions(+), 1 deletion(-)
>  create mode 100644 stdlib/tst-swapcontext2.c
> 
> diff --git a/stdlib/Makefile b/stdlib/Makefile
> index bc8929f2b9..40d07ade20 100644
> --- a/stdlib/Makefile
> +++ b/stdlib/Makefile
> @@ -85,7 +85,7 @@ tests		:= tst-strtol tst-strtod testmb testrand testsort testdiv   \
>  		   test-at_quick_exit-race test-cxa_atexit-race             \
>  		   test-on_exit-race test-dlclose-exit-race 		    \
>  		   tst-makecontext-align test-bz22786 tst-strtod-nan-sign \
> -		   tst-swapcontext1
> +		   tst-swapcontext1 tst-swapcontext2

OK.

>  
>  tests-internal	:= tst-strtod1i tst-strtod3 tst-strtod4 tst-strtod5i \
>  		   tst-tls-atexit tst-tls-atexit-nodelete
> diff --git a/stdlib/tst-swapcontext2.c b/stdlib/tst-swapcontext2.c
> new file mode 100644
> index 0000000000..7802df4868
> --- /dev/null
> +++ b/stdlib/tst-swapcontext2.c
> @@ -0,0 +1,108 @@
> +/* Check multiple makecontext calls.
> +   Copyright (C) 2018 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
> +   <http://www.gnu.org/licenses/>.  */
> +
> +#include <stdio.h>
> +#include <stdlib.h>
> +#include <ucontext.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";
> 

OK.

> +#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);
> +}
> +

OK.

> +__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);
> +}
> +

OK.

> +__attribute__((noinline, noclone))
> +static void
> +func1(void)
> +{
> +  while ( 1 )
> +    {
> +      printf("  \e[31mfunc1: func3(&uctx_func1, &uctx_main, str1)\e[0m\n");
> +      func3( &uctx_func1, &uctx_main, str1, fmt1);
> +    }
> +}
> +

OK.

> +__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);
> +    }
> +}

OK.

> +
> +static int
> +do_test (void)
> +{
> +  char func1_stack[16384];
> +  char func2_stack[16384];
> +  int i;
> +
> +  if (getcontext(&uctx_func1) == -1)
> +    handle_error("getcontext");
> +  uctx_func1.uc_stack.ss_sp = func1_stack;
> +  uctx_func1.uc_stack.ss_size = sizeof(func1_stack);
> +  uctx_func1.uc_link = &uctx_main;
> +  makecontext(&uctx_func1, func1, 0);

OK.

> +
> +  if (getcontext(&uctx_func2) == -1)
> +    handle_error("getcontext");
> +  uctx_func2.uc_stack.ss_sp = func2_stack;
> +  uctx_func2.uc_stack.ss_size = sizeof(func2_stack);
> +  uctx_func2.uc_link = &uctx_func1;
> +  makecontext(&uctx_func2, func2, 0);

OK.

> +
> +  for ( i = 0; i < 4; 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");

OK.

> +    }
> +
> +  printf("main: exiting\n");
> +  exit(EXIT_SUCCESS);
> +}
> +
> +#include <support/test-driver.c>
> 

Cheers,
Carlos.
  

Patch

diff --git a/stdlib/Makefile b/stdlib/Makefile
index bc8929f2b9..40d07ade20 100644
--- a/stdlib/Makefile
+++ b/stdlib/Makefile
@@ -85,7 +85,7 @@  tests		:= tst-strtol tst-strtod testmb testrand testsort testdiv   \
 		   test-at_quick_exit-race test-cxa_atexit-race             \
 		   test-on_exit-race test-dlclose-exit-race 		    \
 		   tst-makecontext-align test-bz22786 tst-strtod-nan-sign \
-		   tst-swapcontext1
+		   tst-swapcontext1 tst-swapcontext2
 
 tests-internal	:= tst-strtod1i tst-strtod3 tst-strtod4 tst-strtod5i \
 		   tst-tls-atexit tst-tls-atexit-nodelete
diff --git a/stdlib/tst-swapcontext2.c b/stdlib/tst-swapcontext2.c
new file mode 100644
index 0000000000..7802df4868
--- /dev/null
+++ b/stdlib/tst-swapcontext2.c
@@ -0,0 +1,108 @@ 
+/* Check multiple makecontext calls.
+   Copyright (C) 2018 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
+   <http://www.gnu.org/licenses/>.  */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <ucontext.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";
+
+#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");
+      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);
+    }
+}
+
+static int
+do_test (void)
+{
+  char func1_stack[16384];
+  char func2_stack[16384];
+  int i;
+
+  if (getcontext(&uctx_func1) == -1)
+    handle_error("getcontext");
+  uctx_func1.uc_stack.ss_sp = func1_stack;
+  uctx_func1.uc_stack.ss_size = sizeof(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 = func2_stack;
+  uctx_func2.uc_stack.ss_size = sizeof(func2_stack);
+  uctx_func2.uc_link = &uctx_func1;
+  makecontext(&uctx_func2, func2, 0);
+
+  for ( i = 0; i < 4; 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");
+    }
+
+  printf("main: exiting\n");
+  exit(EXIT_SUCCESS);
+}
+
+#include <support/test-driver.c>