rtkaio/tst-aiod3.c

   1 /* Test for notification mechanism in lio_listio.
   2    Copyright (C) 2000, 2002, 2006 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, write to the Free
  17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  18    02111-1307 USA.  */
  19
  20 #include <aio.h>
  21 #include <signal.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25 #include <errno.h>
  26 #include "tst-aiod.h"
  27
  28
  29 static pthread_barrier_t b;
  30
  31
  32 static void
  33 thrfct (sigval_t arg)
  34 {
  35   int e = pthread_barrier_wait (&b);
  36   if (e != 0 && e != PTHREAD_BARRIER_SERIAL_THREAD)
  37     {
  38       puts ("child: barrier_wait failed");
  39       exit (1);
  40     }
  41 }
  42
  43
  44 static int
  45 do_test (int argc, char *argv[])
  46 {
  47   char name[] = "/tmp/aio3.XXXXXX";
  48   int fd;
  49   struct aiocb *arr[1];
  50   struct aiocb cb;
  51   static const char buf[] = "Hello World\n";
  52
  53   fd = mkstemp (name);
  54   if (fd == -1)
  55     {
  56       printf ("cannot open temp name: %m\n");
  57       return 1;
  58     }
  59
  60   unlink (name);
  61
  62   if (pthread_barrier_init (&b, NULL, 2) != 0)
  63     {
  64       puts ("barrier_init failed");
  65       return 1;
  66     }
  67
  68   arr[0] = &cb;
  69
  70   void *p;
  71   int sz = set_o_direct (fd);
  72   if (sz != -1)
  73     {
  74       int err = posix_memalign (&p, sz, sz);
  75       if (err)
  76         {
  77           errno = err;
  78           printf ("cannot allocate memory: %m\n");
  79           return 1;
  80         }
  81       memcpy (p, buf, sizeof (buf) - 1);
  82       memset (p + sizeof (buf) - 1, ' ', sz - sizeof (buf) + 1);
  83       printf ("Using O_DIRECT with block size %d\n", sz);
  84     }
  85   else
  86     {
  87       p = (void *) buf;
  88       sz = sizeof (buf) - 1;
  89     }
  90
  91   cb.aio_fildes = fd;
  92   cb.aio_lio_opcode = LIO_WRITE;
  93   cb.aio_reqprio = 0;
  94   cb.aio_buf = p;
  95   cb.aio_nbytes = sz;
  96   cb.aio_offset = 0;
  97   cb.aio_sigevent.sigev_notify = SIGEV_THREAD;
  98   cb.aio_sigevent.sigev_notify_function = thrfct;
  99   cb.aio_sigevent.sigev_notify_attributes = NULL;
 100   cb.aio_sigevent.sigev_value.sival_ptr = NULL;
 101
 102   if (lio_listio (LIO_NOWAIT, arr, 1, NULL) < 0)
 103     {
 104       if (errno == ENOSYS)
 105         {
 106           puts ("no aio support in this configuration");
 107           return 0;
 108         }
 109       printf ("lio_listio failed: %m\n");
 110       return 1;
 111     }
 112
 113   if (aio_suspend ((const struct aiocb *const *) arr, 1, NULL) < 0)
 114     {
 115       printf ("aio_suspend failed: %m\n");
 116       return 1;
 117     }
 118
 119   int e = pthread_barrier_wait (&b);
 120   if (e != 0 && e != PTHREAD_BARRIER_SERIAL_THREAD)
 121     {
 122       puts ("parent: barrier_wait failed");
 123       return 1;
 124     }
 125
 126   puts ("all OK");
 127
 128   return 0;
 129 }
 130
 131 #include "../test-skeleton.c"