LCOV - code coverage report
Current view: top level - tests/gpg - t-eventloop.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 81 83 97.6 %
Date: 2018-11-14 16:53:58 Functions: 6 6 100.0 %

          Line data    Source code
       1             : /* t-eventloop.c - Regression test.
       2             :    Copyright (C) 2000 Werner Koch (dd9jn)
       3             :    Copyright (C) 2001, 2002, 2003, 2004 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             : /* We need to include config.h so that we know whether we are building
      23             :    with large file system (LFS) support. */
      24             : #ifdef HAVE_CONFIG_H
      25             : #include <config.h>
      26             : #endif
      27             : 
      28             : #include <stdio.h>
      29             : #include <stdlib.h>
      30             : #include <string.h>
      31             : #include <assert.h>
      32             : #include <errno.h>
      33             : #include <sys/types.h>
      34             : #include <sys/select.h>
      35             : 
      36             : #include <gpgme.h>
      37             : 
      38             : #include "t-support.h"
      39             : 
      40             : 
      41             : /* Stripped down version of gpgme/wait.c.  */
      42             : 
      43             : struct op_result
      44             : {
      45             :   int done;
      46             :   gpgme_error_t err;
      47             : };
      48             : 
      49             : struct op_result op_result;
      50             : 
      51             : struct one_fd
      52             : {
      53             :   int fd;
      54             :   int dir;
      55             :   gpgme_io_cb_t fnc;
      56             :   void *fnc_data;
      57             : };
      58             : 
      59             : #define FDLIST_MAX 32
      60             : struct one_fd fdlist[FDLIST_MAX];
      61             : 
      62             : gpgme_error_t
      63           4 : add_io_cb (void *data, int fd, int dir, gpgme_io_cb_t fnc, void *fnc_data,
      64             :            void **r_tag)
      65             : {
      66           4 :   struct one_fd *fds = data;
      67             :   int i;
      68             : 
      69          10 :   for (i = 0; i < FDLIST_MAX; i++)
      70             :     {
      71          10 :       if (fds[i].fd == -1)
      72             :         {
      73           4 :           fds[i].fd = fd;
      74           4 :           fds[i].dir = dir;
      75           4 :           fds[i].fnc = fnc;
      76           4 :           fds[i].fnc_data = fnc_data;
      77           4 :           break;
      78             :         }
      79             :     }
      80           4 :   if (i == FDLIST_MAX)
      81           0 :     return gpgme_err_make (GPG_ERR_SOURCE_USER_1, GPG_ERR_GENERAL);
      82           4 :   *r_tag = &fds[i];
      83           4 :   return 0;
      84             : }
      85             : 
      86             : void
      87           4 : remove_io_cb (void *tag)
      88             : {
      89           4 :   struct one_fd *fd = tag;
      90             : 
      91           4 :   fd->fd = -1;
      92           4 : }
      93             : 
      94             : void
      95           2 : io_event (void *data, gpgme_event_io_t type, void *type_data)
      96             : {
      97           2 :   struct op_result *result = data;
      98             : 
      99           2 :   if (type == GPGME_EVENT_DONE)
     100             :     {
     101           1 :       result->done = 1;
     102           1 :       result->err = * (gpgme_error_t *) type_data;
     103             :     }
     104           2 : }
     105             : 
     106             : 
     107             : int
     108          16 : do_select (void)
     109             : {
     110             :   fd_set rfds;
     111             :   fd_set wfds;
     112             :   int i, n;
     113          16 :   int any = 0;
     114             :   struct timeval tv;
     115             : 
     116          16 :   FD_ZERO (&rfds);
     117          16 :   FD_ZERO (&wfds);
     118         528 :   for (i = 0; i < FDLIST_MAX; i++)
     119         512 :     if (fdlist[i].fd != -1)
     120          50 :       FD_SET (fdlist[i].fd, fdlist[i].dir ? &rfds : &wfds);
     121             : 
     122          16 :   tv.tv_sec = 0;
     123          16 :   tv.tv_usec = 1000;
     124             : 
     125             :   do
     126             :     {
     127          16 :       n = select (FD_SETSIZE, &rfds, &wfds, NULL, &tv);
     128             :     }
     129          16 :   while (n < 0 && errno == EINTR);
     130             : 
     131          16 :   if (n < 0)
     132           0 :     return n;   /* Error or timeout.  */
     133             : 
     134          36 :   for (i = 0; i < FDLIST_MAX && n; i++)
     135             :     {
     136          20 :       if (fdlist[i].fd != -1)
     137             :         {
     138          20 :           if (FD_ISSET (fdlist[i].fd, fdlist[i].dir ? &rfds : &wfds))
     139             :             {
     140          12 :               assert (n);
     141          12 :               n--;
     142          12 :               any = 1;
     143          12 :               (*fdlist[i].fnc) (fdlist[i].fnc_data, fdlist[i].fd);
     144             :             }
     145             :         }
     146             :     }
     147          16 :   return any;
     148             : }
     149             : 
     150             : int
     151          16 : my_wait (void)
     152             : {
     153             :   int n;
     154             : 
     155             :   do
     156             :     {
     157          16 :       n = do_select ();
     158             :     }
     159          16 :   while (n >= 0 && !op_result.done);
     160           1 :   return 0;
     161             : }
     162             : 
     163             : 
     164             : struct gpgme_io_cbs io_cbs =
     165             :   {
     166             :     add_io_cb,
     167             :     fdlist,
     168             :     remove_io_cb,
     169             :     io_event,
     170             :     &op_result
     171             :   };
     172             : 
     173             : 
     174             : int
     175           1 : main (void)
     176             : {
     177             :   gpgme_ctx_t ctx;
     178             :   gpgme_error_t err;
     179             :   gpgme_data_t in, out;
     180           1 :   gpgme_key_t key[3] = { NULL, NULL, NULL };
     181             :   int i;
     182             : 
     183           1 :   init_gpgme (GPGME_PROTOCOL_OpenPGP);
     184             : 
     185          33 :   for (i = 0; i < FDLIST_MAX; i++)
     186          32 :     fdlist[i].fd = -1;
     187             : 
     188           1 :   err = gpgme_engine_check_version (GPGME_PROTOCOL_OpenPGP);
     189           1 :   fail_if_err (err);
     190             : 
     191           1 :   err = gpgme_new (&ctx);
     192           1 :   fail_if_err (err);
     193           1 :   gpgme_set_armor (ctx, 1);
     194           1 :   gpgme_set_io_cbs (ctx, &io_cbs);
     195           1 :   op_result.done = 0;
     196             : 
     197           1 :   err = gpgme_data_new_from_mem (&in, "Hallo Leute\n", 12, 0);
     198           1 :   fail_if_err (err);
     199             : 
     200           1 :   err = gpgme_data_new (&out);
     201           1 :   fail_if_err (err);
     202             : 
     203           1 :   err = gpgme_get_key (ctx, "A0FF4590BB6122EDEF6E3C542D727CC768697734",
     204             :                        &key[0], 0);
     205           1 :   fail_if_err (err);
     206           1 :   err = gpgme_get_key (ctx, "D695676BDCEDCC2CDD6152BCFE180B1DA9E3B0B2",
     207             :                        &key[1], 0);
     208           1 :   fail_if_err (err);
     209             : 
     210           1 :   err = gpgme_op_encrypt_start (ctx, key, GPGME_ENCRYPT_ALWAYS_TRUST, in, out);
     211           1 :   fail_if_err (err);
     212             : 
     213           1 :   my_wait ();
     214           1 :   fail_if_err (op_result.err);
     215           1 :   fail_if_err (err);
     216             : 
     217           1 :   fflush (NULL);
     218           1 :   fputs ("Begin Result:\n", stdout);
     219           1 :   print_data (out);
     220           1 :   fputs ("End Result.\n", stdout);
     221             : 
     222           1 :   gpgme_key_unref (key[0]);
     223           1 :   gpgme_key_unref (key[1]);
     224           1 :   gpgme_data_release (in);
     225           1 :   gpgme_data_release (out);
     226           1 :   gpgme_release (ctx);
     227             : 
     228           1 :   return 0;
     229             : }

Generated by: LCOV version 1.13