rt/tst-aio4.c

   1 /* Test for completion signal handling.
   2    Copyright (C) 2000-2015 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, see
  17    <http://www.gnu.org/licenses/>.  */
  18
  19 #include <aio.h>
  20 #include <signal.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <unistd.h>
  24 #include <errno.h>
  25
  26 /* We might need a bit longer timeout.  */
  27 #define TIMEOUT 10 /* sec */
  28
  29 int my_signo;
  30
  31 volatile sig_atomic_t flag;
  32
  33
  34 static void
  35 sighandler (const int signo)
  36 {
  37   flag = signo;
  38 }
  39
  40 static int
  41 wait_flag (void)
  42 {
  43   while (flag == 0)
  44     {
  45       puts ("Sleeping...");
  46       sleep (1);
  47     }
  48
  49   if (flag != my_signo)
  50     {
  51       printf ("signal handler received wrong signal, flag is %d\n", flag);
  52       return 1;
  53     }
  54
  55   return 0;
  56 }
  57
  58 #ifndef SIGRTMIN
  59 # define SIGRTMIN -1
  60 # define SIGRTMAX -1
  61 #endif
  62
  63 static int
  64 do_test (int argc, char *argv[])
  65 {
  66   char name[] = "/tmp/aio4.XXXXXX";
  67   int fd;
  68   struct aiocb *arr[1];
  69   struct aiocb cb;
  70   static const char buf[] = "Hello World\n";
  71   struct aioinit init = {10, 20, 0};
  72   struct sigaction sa;
  73   struct sigevent ev;
  74
  75   if (SIGRTMIN == -1)
  76   {
  77       printf ("RT signals not supported.\n");
  78       return 0;
  79   }
  80
  81   /* Select a signal from the middle of the available choices... */
  82   my_signo = (SIGRTMAX + SIGRTMIN) / 2;
  83
  84   fd = mkstemp (name);
  85   if (fd == -1)
  86     {
  87       printf ("cannot open temp name: %m\n");
  88       return 1;
  89     }
  90
  91   unlink (name);
  92
  93   /* Test also aio_init.  */
  94   aio_init (&init);
  95
  96   arr[0] = &cb;
  97
  98   cb.aio_fildes = fd;
  99   cb.aio_lio_opcode = LIO_WRITE;
 100   cb.aio_reqprio = 0;
 101   cb.aio_buf = (void *) buf;
 102   cb.aio_nbytes = sizeof (buf) - 1;
 103   cb.aio_offset = 0;
 104   cb.aio_sigevent.sigev_notify = SIGEV_SIGNAL;
 105   cb.aio_sigevent.sigev_notify_function = NULL;
 106   cb.aio_sigevent.sigev_notify_attributes = NULL;
 107   cb.aio_sigevent.sigev_signo = my_signo;
 108   cb.aio_sigevent.sigev_value.sival_ptr = NULL;
 109
 110   ev.sigev_notify = SIGEV_SIGNAL;
 111   ev.sigev_notify_function = NULL;
 112   ev.sigev_notify_attributes = NULL;
 113   ev.sigev_signo = my_signo;
 114
 115   sa.sa_handler = sighandler;
 116   sigemptyset (&sa.sa_mask);
 117   sa.sa_flags = SA_RESTART;
 118
 119   if (sigaction (my_signo, &sa, NULL) < 0)
 120     {
 121       printf ("sigaction failed: %m\n");
 122       return 1;
 123     }
 124
 125   flag = 0;
 126   /* First use aio_write.  */
 127   if (aio_write (arr[0]) < 0)
 128     {
 129       if (errno == ENOSYS)
 130         {
 131           puts ("no aio support in this configuration");
 132           return 0;
 133         }
 134       printf ("aio_write failed: %m\n");
 135       return 1;
 136     }
 137
 138   if (wait_flag ())
 139     return 1;
 140
 141   puts ("aio_write OK");
 142
 143   flag = 0;
 144   /* Again with lio_listio.  */
 145   if (lio_listio (LIO_NOWAIT, arr, 1, &ev) < 0)
 146     {
 147       printf ("lio_listio failed: %m\n");
 148       return 1;
 149     }
 150
 151   if (wait_flag ())
 152     return 1;
 153
 154   puts ("all OK");
 155
 156   return 0;
 157 }
 158
 159 #include "../test-skeleton.c"