990876d7f17c7647f35c2d40e1af5fb546295319
[gpgme.git] / tests / gpg / t-eventloop.c
1 /* t-eventloop.c  - regression test
2    Copyright (C) 2000 Werner Koch (dd9jn)
3    Copyright (C) 2001, 2002, 2003 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 General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (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    General Public License for more details.
16  
17    You should have received a copy of the GNU General Public License
18    along with GPGME; if not, write to the Free Software Foundation,
19    Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <assert.h>
25 #include <errno.h>
26 #include <sys/select.h>
27
28 #include <gpgme.h>
29
30 #define fail_if_err(err)                                        \
31   do                                                            \
32     {                                                           \
33       if (err)                                                  \
34         {                                                       \
35           fprintf (stderr, "%s:%d: gpgme_error_t %s\n",         \
36                    __FILE__, __LINE__, gpgme_strerror (err));   \
37           exit (1);                                             \
38         }                                                       \
39     }                                                           \
40   while (0)
41
42
43 static void
44 print_data (gpgme_data_t dh)
45 {
46   char buf[100];
47   int ret;
48   
49   ret = gpgme_data_seek (dh, 0, SEEK_SET);
50   if (ret)
51     fail_if_err (GPGME_File_Error);
52   while ((ret = gpgme_data_read (dh, buf, 100)) > 0)
53     fwrite (buf, ret, 1, stdout);
54   if (ret < 0)
55     fail_if_err (GPGME_File_Error);
56 }
57
58
59 /* Stripped down version of gpgme/wait.c.  */
60
61 struct op_result
62 {
63   int done;
64   gpgme_error_t err;
65 };
66
67 struct op_result op_result;
68
69 struct one_fd
70 {
71   int fd;
72   int dir;
73   gpgme_io_cb_t fnc;
74   void *fnc_data;
75 };
76
77 #define FDLIST_MAX 32
78 struct one_fd fdlist[FDLIST_MAX];
79
80 gpgme_error_t
81 add_io_cb (void *data, int fd, int dir, gpgme_io_cb_t fnc, void *fnc_data,
82            void **r_tag)
83 {
84   struct one_fd *fds = data;
85   int i;
86
87   for (i = 0; i < FDLIST_MAX; i++)
88     {
89       if (fds[i].fd == -1)
90         {
91           fds[i].fd = fd;
92           fds[i].dir = dir;
93           fds[i].fnc = fnc;
94           fds[i].fnc_data = fnc_data;
95           break;
96         }
97     }
98   if (i == FDLIST_MAX)
99     return GPGME_General_Error;
100   *r_tag = &fds[i];
101   return 0;
102 }
103
104 void
105 remove_io_cb (void *tag)
106 {
107   struct one_fd *fd = tag;
108
109   fd->fd = -1;
110 }
111
112 void
113 io_event (void *data, gpgme_event_io_t type, void *type_data)
114 {
115   struct op_result *result = data;
116
117   if (type == GPGME_EVENT_DONE)
118     {
119       result->done = 1;
120       result->err = * (gpgme_error_t *) type_data;
121     }
122 }
123
124
125 int
126 do_select (void)
127 {
128   fd_set rfds;
129   fd_set wfds;
130   int i, n;
131   int any = 0;
132
133   FD_ZERO (&rfds);
134   FD_ZERO (&wfds);
135   for (i = 0; i < FDLIST_MAX; i++)
136     if (fdlist[i].fd != -1)
137       FD_SET (fdlist[i].fd, fdlist[i].dir ? &rfds : &wfds);
138
139   do
140     {
141       n = select (FD_SETSIZE, &rfds, &wfds, NULL, 0);
142     }
143   while (n < 0 && errno == EINTR);
144
145   if (n < 0)
146     return n;   /* Error or timeout.  */
147
148   for (i = 0; i < FDLIST_MAX && n; i++)
149     {
150       if (fdlist[i].fd != -1)
151         {
152           if (FD_ISSET (fdlist[i].fd, fdlist[i].dir ? &rfds : &wfds))
153             {
154               assert (n);
155               n--;
156               any = 1;
157               (*fdlist[i].fnc) (fdlist[i].fnc_data, fdlist[i].fd);
158             }
159         }
160     }
161   return any;
162 }
163
164 int
165 my_wait (void)
166 {
167   int n;
168
169   do
170     {
171       n = do_select ();
172     }
173   while (n >= 0 && !op_result.done);
174   return 0;
175 }
176
177
178 struct gpgme_io_cbs io_cbs =
179   {
180     add_io_cb,
181     fdlist,
182     remove_io_cb,
183     io_event,
184     &op_result
185   };
186
187
188 int 
189 main (int argc, char *argv[])
190 {
191   gpgme_ctx_t ctx;
192   gpgme_error_t err;
193   gpgme_data_t in, out;
194   gpgme_key_t key[3] = { NULL, NULL, NULL };
195   int i;
196
197   for (i = 0; i < FDLIST_MAX; i++)
198     fdlist[i].fd = -1;
199
200   err = gpgme_engine_check_version (GPGME_PROTOCOL_OpenPGP);
201   fail_if_err (err);
202
203   err = gpgme_new (&ctx);
204   fail_if_err (err);
205   gpgme_set_armor (ctx, 1);
206   gpgme_set_io_cbs (ctx, &io_cbs);
207   op_result.done = 0;
208
209   err = gpgme_data_new_from_mem (&in, "Hallo Leute\n", 12, 0);
210   fail_if_err (err);
211
212   err = gpgme_data_new (&out);
213   fail_if_err (err);
214
215   err = gpgme_get_key (ctx, "A0FF4590BB6122EDEF6E3C542D727CC768697734",
216                        &key[0], 0);
217   fail_if_err (err);
218   err = gpgme_get_key (ctx, "D695676BDCEDCC2CDD6152BCFE180B1DA9E3B0B2",
219                        &key[1], 0);
220   fail_if_err (err);
221
222   err = gpgme_op_encrypt_start (ctx, key, GPGME_ENCRYPT_ALWAYS_TRUST, in, out);
223   fail_if_err (err);
224
225   my_wait ();
226   fail_if_err (op_result.err);
227   fail_if_err (err);
228
229   fflush (NULL);
230   fputs ("Begin Result:\n", stdout);
231   print_data (out);
232   fputs ("End Result.\n", stdout);
233    
234   gpgme_key_unref (key[0]);
235   gpgme_key_unref (key[1]);
236   gpgme_data_release (in);
237   gpgme_data_release (out);
238   gpgme_release (ctx);
239
240   return 0;
241 }