LCOV - code coverage report
Current view: top level - src - wait.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 79 88 89.8 %
Date: 2018-11-14 16:53:58 Functions: 6 6 100.0 %

          Line data    Source code
       1             : /* wait.c
       2             :    Copyright (C) 2000 Werner Koch (dd9jn)
       3             :    Copyright (C) 2001, 2002, 2003, 2004, 2005, 2007 g10 Code GmbH
       4             : 
       5             :    This file is part of GPGME.
       6             : 
       7             :    GPGME is free software; you can redistribute it and/or modify it
       8             :    under the terms of the GNU Lesser General Public License as
       9             :    published by the Free Software Foundation; either version 2.1 of
      10             :    the License, or (at your option) any later version.
      11             : 
      12             :    GPGME is distributed in the hope that it will be useful, but
      13             :    WITHOUT ANY WARRANTY; without even the implied warranty of
      14             :    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      15             :    Lesser General Public License for more details.
      16             : 
      17             :    You should have received a copy of the GNU Lesser General Public
      18             :    License along with this program; if not, write to the Free Software
      19             :    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
      20             :    02111-1307, USA.  */
      21             : 
      22             : #if HAVE_CONFIG_H
      23             : #include <config.h>
      24             : #endif
      25             : #include <stdlib.h>
      26             : #include <string.h>
      27             : #include <assert.h>
      28             : #include <errno.h>
      29             : #ifdef HAVE_SYS_TYPES_H
      30             : # include <sys/types.h>
      31             : #endif
      32             : 
      33             : #include "util.h"
      34             : #include "context.h"
      35             : #include "ops.h"
      36             : #include "wait.h"
      37             : #include "sema.h"
      38             : #include "priv-io.h"
      39             : #include "engine.h"
      40             : #include "debug.h"
      41             : 
      42             : 
      43             : void
      44         726 : _gpgme_fd_table_init (fd_table_t fdt)
      45             : {
      46         726 :   fdt->fds = NULL;
      47         726 :   fdt->size = 0;
      48         726 : }
      49             : 
      50             : void
      51         715 : _gpgme_fd_table_deinit (fd_table_t fdt)
      52             : {
      53         715 :   if (fdt->fds)
      54         477 :     free (fdt->fds);
      55         715 : }
      56             : 
      57             : 
      58             : /* XXX We should keep a marker and roll over for speed.  */
      59             : static gpgme_error_t
      60        2451 : fd_table_put (fd_table_t fdt, int fd, int dir, void *opaque, int *idx)
      61             : {
      62             :   unsigned int i, j;
      63             :   struct io_select_fd_s *new_fds;
      64             : 
      65        5396 :   for (i = 0; i < fdt->size; i++)
      66             :     {
      67        4907 :       if (fdt->fds[i].fd == -1)
      68        1962 :         break;
      69             :     }
      70        2451 :   if (i == fdt->size)
      71             :     {
      72             : #define FDT_ALLOCSIZE 10
      73         489 :       new_fds = realloc (fdt->fds, (fdt->size + FDT_ALLOCSIZE)
      74             :                          * sizeof (*new_fds));
      75         489 :       if (!new_fds)
      76           0 :         return gpg_error_from_syserror ();
      77             : 
      78         489 :       fdt->fds = new_fds;
      79         489 :       fdt->size += FDT_ALLOCSIZE;
      80        5379 :       for (j = 0; j < FDT_ALLOCSIZE; j++)
      81        4890 :         fdt->fds[i + j].fd = -1;
      82             :     }
      83             : 
      84        2451 :   fdt->fds[i].fd = fd;
      85        2451 :   fdt->fds[i].for_read = (dir == 1);
      86        2451 :   fdt->fds[i].for_write = (dir == 0);
      87        2451 :   fdt->fds[i].signaled = 0;
      88        2451 :   fdt->fds[i].opaque = opaque;
      89        2451 :   *idx = i;
      90        2451 :   return 0;
      91             : }
      92             : 
      93             : 
      94             : /* Register the file descriptor FD with the handler FNC (which gets
      95             :    FNC_DATA as its first argument) for the direction DIR.  DATA should
      96             :    be the context for which the fd is added.  R_TAG will hold the tag
      97             :    that can be used to remove the fd.  */
      98             : gpgme_error_t
      99        2451 : _gpgme_add_io_cb (void *data, int fd, int dir, gpgme_io_cb_t fnc,
     100             :                   void *fnc_data, void **r_tag)
     101             : {
     102             :   gpgme_error_t err;
     103        2451 :   gpgme_ctx_t ctx = (gpgme_ctx_t) data;
     104             :   fd_table_t fdt;
     105             :   struct wait_item_s *item;
     106             :   struct tag *tag;
     107             : 
     108        2451 :   assert (fnc);
     109        2451 :   assert (ctx);
     110             : 
     111        2451 :   fdt = &ctx->fdt;
     112        2451 :   assert (fdt);
     113             : 
     114        2451 :   tag = malloc (sizeof *tag);
     115        2451 :   if (!tag)
     116           0 :     return gpg_error_from_syserror ();
     117        2451 :   tag->ctx = ctx;
     118             : 
     119             :   /* Allocate a structure to hold information about the handler.  */
     120        2451 :   item = calloc (1, sizeof *item);
     121        2451 :   if (!item)
     122             :     {
     123           0 :       free (tag);
     124           0 :       return gpg_error_from_syserror ();
     125             :     }
     126        2451 :   item->ctx = ctx;
     127        2451 :   item->dir = dir;
     128        2451 :   item->handler = fnc;
     129        2451 :   item->handler_value = fnc_data;
     130             : 
     131        2451 :   err = fd_table_put (fdt, fd, dir, item, &tag->idx);
     132        2451 :   if (err)
     133             :     {
     134           0 :       free (tag);
     135           0 :       free (item);
     136           0 :       return err;
     137             :     }
     138             : 
     139        2451 :   TRACE3 (DEBUG_CTX, "_gpgme_add_io_cb", ctx,
     140             :           "fd %d, dir=%d -> tag=%p", fd, dir, tag);
     141             : 
     142        2451 :   *r_tag = tag;
     143        2451 :   return 0;
     144             : }
     145             : 
     146             : 
     147             : void
     148        2446 : _gpgme_remove_io_cb (void *data)
     149             : {
     150        2446 :   struct tag *tag = data;
     151             :   gpgme_ctx_t ctx;
     152             :   fd_table_t fdt;
     153             :   int idx;
     154             : 
     155        2446 :   assert (tag);
     156        2446 :   ctx = tag->ctx;
     157        2446 :   assert (ctx);
     158        2446 :   fdt = &ctx->fdt;
     159        2446 :   assert (fdt);
     160        2446 :   idx = tag->idx;
     161             : 
     162        2446 :   TRACE2 (DEBUG_CTX, "_gpgme_remove_io_cb", data,
     163             :           "setting fd 0x%x (item=%p) done", fdt->fds[idx].fd,
     164             :           fdt->fds[idx].opaque);
     165             : 
     166        2446 :   free (fdt->fds[idx].opaque);
     167        2446 :   free (tag);
     168             : 
     169             :   /* Free the table entry.  */
     170        2446 :   fdt->fds[idx].fd = -1;
     171        2446 :   fdt->fds[idx].for_read = 0;
     172        2446 :   fdt->fds[idx].for_write = 0;
     173        2446 :   fdt->fds[idx].opaque = NULL;
     174        2446 : }
     175             : 
     176             : 
     177             : /* This is slightly embarrassing.  The problem is that running an I/O
     178             :    callback _may_ influence the status of other file descriptors.  Our
     179             :    own event loops could compensate for that, but the external event
     180             :    loops cannot.  FIXME: We may still want to optimize this a bit when
     181             :    we are called from our own event loops.  So if CHECKED is 1, the
     182             :    check is skipped.  */
     183             : gpgme_error_t
     184        9299 : _gpgme_run_io_cb (struct io_select_fd_s *an_fds, int checked,
     185             :                   gpgme_error_t *op_err)
     186             : {
     187             :   struct wait_item_s *item;
     188             :   struct io_cb_data iocb_data;
     189             :   gpgme_error_t err;
     190             : 
     191        9299 :   item = (struct wait_item_s *) an_fds->opaque;
     192        9299 :   assert (item);
     193             : 
     194        9299 :   if (!checked)
     195             :     {
     196             :       int nr;
     197             :       struct io_select_fd_s fds;
     198             : 
     199        9300 :       TRACE0 (DEBUG_CTX, "_gpgme_run_io_cb", item, "need to check");
     200        9299 :       fds = *an_fds;
     201        9299 :       fds.signaled = 0;
     202             :       /* Just give it a quick poll.  */
     203        9299 :       nr = _gpgme_io_select (&fds, 1, 1);
     204        9300 :       assert (nr <= 1);
     205        9300 :       if (nr < 0)
     206           0 :         return errno;
     207        9300 :       else if (nr == 0)
     208             :         /* The status changed in the meantime, there is nothing left
     209             :            to do.  */
     210           0 :         return 0;
     211             :     }
     212             : 
     213        9299 :   TRACE2 (DEBUG_CTX, "_gpgme_run_io_cb", item, "handler (%p, %d)",
     214             :           item->handler_value, an_fds->fd);
     215             : 
     216        9300 :   iocb_data.handler_value = item->handler_value;
     217        9300 :   iocb_data.op_err = 0;
     218        9300 :   err = item->handler (&iocb_data, an_fds->fd);
     219             : 
     220        9299 :   *op_err = iocb_data.op_err;
     221        9299 :   return err;
     222             : }

Generated by: LCOV version 1.13