(root)/
glibc-2.38/
rt/
aio_notify.c
       1  /* Notify initiator of AIO request.
       2     Copyright (C) 1997-2023 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     <https://www.gnu.org/licenses/>.  */
      18  
      19  #include <errno.h>
      20  #include <pthreadP.h>
      21  #include <stdlib.h>
      22  #include <unistd.h>
      23  #include <aio_misc.h>
      24  #include <signal.h>
      25  
      26  #if !PTHREAD_IN_LIBC
      27  # define __pthread_attr_init pthread_attr_init
      28  # define __pthread_attr_setdetachstate pthread_attr_setdetachstate
      29  #endif
      30  
      31  #ifndef aio_start_notify_thread
      32  # define aio_start_notify_thread() do { } while (0)
      33  #endif
      34  
      35  struct notify_func
      36    {
      37      void (*func) (sigval_t);
      38      sigval_t value;
      39    };
      40  
      41  static void *
      42  notify_func_wrapper (void *arg)
      43  {
      44    aio_start_notify_thread ();
      45    struct notify_func *const n = arg;
      46    void (*func) (sigval_t) = n->func;
      47    sigval_t value = n->value;
      48    free (n);
      49    (*func) (value);
      50    return NULL;
      51  }
      52  
      53  
      54  int
      55  __aio_notify_only (struct sigevent *sigev)
      56  {
      57    int result = 0;
      58  
      59    /* Send the signal to notify about finished processing of the request.  */
      60    if (__glibc_unlikely (sigev->sigev_notify == SIGEV_THREAD))
      61      {
      62        /* We have to start a thread.  */
      63        pthread_t tid;
      64        pthread_attr_t attr, *pattr;
      65  
      66        pattr = (pthread_attr_t *) sigev->sigev_notify_attributes;
      67        if (pattr == NULL)
      68  	{
      69  	  __pthread_attr_init (&attr);
      70  	  __pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
      71  	  pattr = &attr;
      72  	}
      73  
      74        /* SIGEV may be freed as soon as we return, so we cannot let the
      75  	 notification thread use that pointer.  Even though a sigval_t is
      76  	 only one word and the same size as a void *, we cannot just pass
      77  	 the value through pthread_create as the argument and have the new
      78  	 thread run the user's function directly, because on some machines
      79  	 the calling convention for a union like sigval_t is different from
      80  	 that for a pointer type like void *.  */
      81        struct notify_func *nf = malloc (sizeof *nf);
      82        if (nf == NULL)
      83  	result = -1;
      84        else
      85  	{
      86  	  nf->func = sigev->sigev_notify_function;
      87  	  nf->value = sigev->sigev_value;
      88  	  if (__pthread_create (&tid, pattr, notify_func_wrapper, nf) < 0)
      89  	    {
      90  	      free (nf);
      91  	      result = -1;
      92  	    }
      93  	}
      94      }
      95    else if (sigev->sigev_notify == SIGEV_SIGNAL)
      96      {
      97        /* We have to send a signal.  */
      98  #if _POSIX_REALTIME_SIGNALS > 0
      99        /* Note that the standard gives us the option of using a plain
     100  	 non-queuing signal here when SA_SIGINFO is not set for the signal.  */
     101        if (__aio_sigqueue (sigev->sigev_signo, sigev->sigev_value, getpid ())
     102  	  < 0)
     103  	result = -1;
     104  #else
     105        /* There are no queued signals on this system at all.  */
     106        result = raise (sigev->sigev_signo);
     107  #endif
     108      }
     109  
     110    return result;
     111  }
     112  
     113  
     114  void
     115  __aio_notify (struct requestlist *req)
     116  {
     117    struct waitlist *waitlist;
     118    struct aiocb *aiocbp = &req->aiocbp->aiocb;
     119  
     120    if (__aio_notify_only (&aiocbp->aio_sigevent) != 0)
     121      {
     122        /* XXX What shall we do if already an error is set by
     123  	 read/write/fsync?  */
     124        aiocbp->__error_code = errno;
     125        aiocbp->__return_value = -1;
     126      }
     127  
     128    /* Now also notify possibly waiting threads.  */
     129    waitlist = req->waiting;
     130    while (waitlist != NULL)
     131      {
     132        struct waitlist *next = waitlist->next;
     133  
     134        if (waitlist->sigevp == NULL)
     135  	{
     136  	  if (waitlist->result != NULL && aiocbp->__return_value == -1)
     137  	    *waitlist->result = -1;
     138  
     139  #ifdef DONT_NEED_AIO_MISC_COND
     140  	  AIO_MISC_NOTIFY (waitlist);
     141  #else
     142  	  /* Decrement the counter.  */
     143  	  --*waitlist->counterp;
     144  
     145  	  pthread_cond_signal (waitlist->cond);
     146  #endif
     147  	}
     148        else
     149  	/* This is part of an asynchronous `lio_listio' operation.  If
     150  	   this request is the last one, send the signal.  */
     151  	if (--*waitlist->counterp == 0)
     152  	  {
     153  	    __aio_notify_only (waitlist->sigevp);
     154  	    /* This is tricky.  See lio_listio.c for the reason why
     155  	       this works.  */
     156  	    free ((void *) waitlist->counterp);
     157  	  }
     158  
     159        waitlist = next;
     160      }
     161  }