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