(root)/
glibc-2.38/
stdlib/
tst-setcontext4.c
       1  /* Testcase checks, if setcontext(), swapcontext() restores signal-mask
       2     and if pending signals are delivered after those calls.
       3     Copyright (C) 2018-2023 Free Software Foundation, Inc.
       4     This file is part of the GNU C Library.
       5  
       6     The GNU C Library is free software; you can redistribute it and/or
       7     modify it under the terms of the GNU Lesser General Public
       8     License as published by the Free Software Foundation; either
       9     version 2.1 of the License, or (at your option) any later version.
      10  
      11     The GNU C Library is distributed in the hope that it will be useful,
      12     but WITHOUT ANY WARRANTY; without even the implied warranty of
      13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      14     Lesser General Public License for more details.
      15  
      16     You should have received a copy of the GNU Lesser General Public
      17     License along with the GNU C Library; if not, see
      18     <https://www.gnu.org/licenses/>.  */
      19  
      20  #include <stdio.h>
      21  #include <stdlib.h>
      22  #include <sys/types.h>
      23  #include <signal.h>
      24  #include <ucontext.h>
      25  #include <unistd.h>
      26  
      27  volatile int global;
      28  volatile sig_atomic_t handlerCalled;
      29  
      30  static void
      31  check (const char *funcName)
      32  {
      33    sigset_t set;
      34  
      35    /* check if SIGUSR2 is unblocked after setcontext-call.  */
      36    sigprocmask (SIG_BLOCK, NULL, &set);
      37  
      38    if (sigismember (&set, SIGUSR2) != 0)
      39      {
      40        printf ("FAIL: SIGUSR2 is blocked after %s.\n", funcName);
      41        exit (1);
      42      }
      43  
      44    if (sigismember (&set, SIGUSR1) != 1)
      45      {
      46        printf ("FAIL: SIGUSR1 is not blocked after %s.\n", funcName);
      47        exit (1);
      48      }
      49  }
      50  
      51  static void
      52  signalmask (int how, int signum)
      53  {
      54    sigset_t set;
      55    sigemptyset (&set);
      56    sigaddset (&set, signum);
      57    if (sigprocmask (how, &set, NULL) != 0)
      58      {
      59        printf ("FAIL: sigprocmaks (%d, %d, NULL): %m\n", how, signum);
      60        exit (1);
      61      }
      62  }
      63  
      64  static void
      65  signalpending (int signum, const char *msg)
      66  {
      67    sigset_t set;
      68    sigemptyset (&set);
      69    if (sigpending (&set) != 0)
      70      {
      71        printf ("FAIL: sigpending: %m\n");
      72        exit (1);
      73      }
      74    if (sigismember (&set, SIGUSR2) != 1)
      75      {
      76        printf ("FAIL: Signal %d is not pending %s\n", signum, msg);
      77        exit (1);
      78      }
      79  }
      80  
      81  static void
      82  handler (int __attribute__ ((unused)) signum)
      83  {
      84    handlerCalled ++;
      85  }
      86  
      87  static int
      88  do_test (void)
      89  {
      90    ucontext_t ctx, oldctx;
      91    struct sigaction action;
      92    pid_t pid;
      93  
      94    pid = getpid ();
      95  
      96    /* unblock SIGUSR2 */
      97    signalmask (SIG_UNBLOCK, SIGUSR2);
      98  
      99    /* block SIGUSR1 */
     100    signalmask (SIG_BLOCK, SIGUSR1);
     101  
     102    /* register handler for SIGUSR2  */
     103    action.sa_flags = 0;
     104    action.sa_handler = handler;
     105    sigemptyset (&action.sa_mask);
     106    sigaction (SIGUSR2, &action, NULL);
     107  
     108    if (getcontext (&ctx) != 0)
     109      {
     110        printf ("FAIL: getcontext: %m\n");
     111        exit (1);
     112      }
     113  
     114    global++;
     115  
     116    if (global == 1)
     117      {
     118        puts ("after getcontext");
     119  
     120        /* block SIGUSR2  */
     121        signalmask (SIG_BLOCK, SIGUSR2);
     122  
     123        /* send SIGUSR2 to me  */
     124        handlerCalled = 0;
     125        kill (pid, SIGUSR2);
     126  
     127        /* was SIGUSR2 handler called?  */
     128        if (handlerCalled != 0)
     129  	{
     130  	  puts ("FAIL: signal handler was called, but signal was blocked.");
     131  	  exit (1);
     132  	}
     133  
     134        /* is SIGUSR2 pending?  */
     135        signalpending (SIGUSR2, "before setcontext");
     136  
     137        /* SIGUSR2 will be unblocked by setcontext-call.  */
     138        if (setcontext (&ctx) != 0)
     139  	{
     140  	  printf ("FAIL: setcontext: %m\n");
     141  	  exit (1);
     142  	}
     143      }
     144    else if (global == 2)
     145      {
     146        puts ("after setcontext");
     147  
     148        /* check SIGUSR1/2  */
     149        check ("setcontext");
     150  
     151        /* was SIGUSR2 handler called? */
     152        if (handlerCalled != 1)
     153  	{
     154  	  puts ("FAIL: signal handler was not called after setcontext.");
     155  	  exit (1);
     156  	}
     157  
     158        /* block SIGUSR2 */
     159        signalmask (SIG_BLOCK, SIGUSR2);
     160  
     161        /* send SIGUSR2 to me  */
     162        handlerCalled = 0;
     163        kill (pid, SIGUSR2);
     164  
     165        /* was SIGUSR2 handler called?  */
     166        if (handlerCalled != 0)
     167  	{
     168  	  puts ("FAIL: signal handler was called, but signal was blocked.");
     169  	  exit (1);
     170  	}
     171  
     172        /* is SIGUSR2 pending?  */
     173        signalpending (SIGUSR2, "before swapcontext");
     174  
     175        if (swapcontext (&oldctx, &ctx) != 0)
     176  	{
     177  	  printf ("FAIL: swapcontext: %m\n");
     178  	  exit (1);
     179  	}
     180  
     181        puts ("FAIL: returned from (&oldctx, &ctx)");
     182        exit (1);
     183      }
     184    else if ( global != 3 )
     185      {
     186        puts ("FAIL: 'global' not incremented three times");
     187        exit (1);
     188      }
     189  
     190    puts ("after swapcontext");
     191    /* check SIGUSR1/2  */
     192    check ("swapcontext");
     193  
     194    /* was SIGUSR2 handler called? */
     195    if (handlerCalled != 1)
     196      {
     197        puts ("FAIL: signal handler was not called after swapcontext.");
     198        exit (1);
     199      }
     200  
     201    /* check sigmask in old context of swapcontext-call  */
     202    if (sigismember (&oldctx.uc_sigmask, SIGUSR2) != 1)
     203      {
     204        puts ("FAIL: SIGUSR2 is not blocked in oldctx.uc_sigmask.");
     205        exit (1);
     206      }
     207  
     208    if (sigismember (&oldctx.uc_sigmask, SIGUSR1) != 1)
     209      {
     210        puts ("FAIL: SIGUSR1 is not blocked in oldctx.uc_sigmaks.");
     211        exit (1);
     212      }
     213  
     214    return 0;
     215  }
     216  
     217  #include <support/test-driver.c>