634911af3d40312e2ca9b45235ce4780133ad7cd
[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 g10 Code GmbH
4  *
5  * This file is part of GPGME.
6  *
7  * GPGME is free software; you can redistribute it and/or modify
8  * it 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,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
20  */
21
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <assert.h>
26 #include <errno.h>
27 #include <sys/select.h>
28
29 #include <gpgme.h>
30
31 #define fail_if_err(a) do { if(a) {                                       \
32                                fprintf (stderr, "%s:%d: GpgmeError %s\n", \
33                                 __FILE__, __LINE__, gpgme_strerror(a));   \
34                                 exit (1); }                               \
35                              } while(0)
36
37 static void
38 print_op_info (GpgmeCtx ctx)
39 {
40   char *s = gpgme_get_op_info (ctx, 0);
41
42   if (!s)
43     puts ("<!-- no operation info available -->");
44   else
45     {
46       puts (s);
47       free (s);
48     }
49 }
50
51
52 static void
53 print_data (GpgmeData dh)
54 {
55   char buf[100];
56   size_t nread;
57   GpgmeError err;
58
59   err = gpgme_data_rewind (dh);
60   fail_if_err (err);
61   while (!(err = gpgme_data_read (dh, buf, 100, &nread)))
62     fwrite (buf, nread, 1, stdout);
63   if (err != GPGME_EOF) 
64     fail_if_err (err);
65 }
66
67
68 /* Stripped down version of gpgme/wait.c.  */
69
70 struct op_result
71 {
72   int done;
73   GpgmeError err;
74 };
75
76 struct op_result op_result;
77
78 struct one_fd
79 {
80   int fd;
81   int dir;
82   GpgmeIOCb fnc;
83   void *fnc_data;
84 };
85
86 #define FDLIST_MAX 32
87 struct one_fd fdlist[FDLIST_MAX];
88
89 GpgmeError
90 add_io_cb (void *data, int fd, int dir, GpgmeIOCb fnc, void *fnc_data,
91            void **r_tag)
92 {
93   struct one_fd *fds = data;
94   int i;
95
96   for (i = 0; i < FDLIST_MAX; i++)
97     {
98       if (fds[i].fd == -1)
99         {
100           fds[i].fd = fd;
101           fds[i].dir = dir;
102           fds[i].fnc = fnc;
103           fds[i].fnc_data = fnc_data;
104           break;
105         }
106     }
107   if (i == FDLIST_MAX)
108     return GPGME_General_Error;
109   *r_tag = &fds[i];
110   return 0;
111 }
112
113 void
114 remove_io_cb (void *tag)
115 {
116   struct one_fd *fd = tag;
117
118   fd->fd = -1;
119 }
120
121 void
122 io_event (void *data, GpgmeEventIO type, void *type_data)
123 {
124   struct op_result *result = data;
125
126   if (type == GPGME_EVENT_DONE)
127     {
128       result->done = 1;
129       result->err = * (GpgmeError *) type_data;
130     }
131 }
132
133 int
134 do_select (void)
135 {
136   fd_set rfds;
137   fd_set wfds;
138   int i, n;
139   int any = 0;
140
141   FD_ZERO (&rfds);
142   FD_ZERO (&wfds);
143   for (i = 0; i < FDLIST_MAX; i++)
144     if (fdlist[i].fd != -1)
145       FD_SET (fdlist[i].fd, fdlist[i].dir ? &rfds : &wfds);
146
147   do
148     {
149       n = select (FD_SETSIZE, &rfds, &wfds, NULL, 0);
150     }
151   while (n < 0 && errno == EINTR);
152
153   if (n < 0)
154     return n;   /* Error or timeout.  */
155
156   for (i = 0; i < FDLIST_MAX && n; i++)
157     {
158       if (fdlist[i].fd != -1)
159         {
160           if (FD_ISSET (fdlist[i].fd, fdlist[i].dir ? &rfds : &wfds))
161             {
162               assert (n);
163               n--;
164               any = 1;
165               (*fdlist[i].fnc) (fdlist[i].fnc_data, fdlist[i].fd);
166             }
167         }
168     }
169   return any;
170 }
171
172 int
173 my_wait (void)
174 {
175   int n;
176
177   do
178     {
179       n = do_select ();
180     }
181   while (n >= 0 && !op_result.done);
182   return 0;
183 }
184
185 struct GpgmeIOCbs io_cbs =
186   {
187     add_io_cb,
188     fdlist,
189     remove_io_cb,
190     io_event,
191     &op_result
192   };
193
194 int 
195 main (int argc, char *argv[])
196 {
197   GpgmeCtx ctx;
198   GpgmeError err;
199   GpgmeData in, out;
200   GpgmeRecipients rset;
201   int i;
202
203   for (i = 0; i < FDLIST_MAX; i++)
204     fdlist[i].fd = -1;
205
206   err = gpgme_check_engine ();
207   fail_if_err (err);
208   puts (gpgme_get_engine_info ());
209
210   do
211     {
212       err = gpgme_new (&ctx);
213       fail_if_err (err);
214       gpgme_set_armor (ctx, 1);
215       gpgme_set_io_cbs (ctx, &io_cbs);
216       op_result.done = 0;
217
218       err = gpgme_data_new_from_mem (&in, "Hallo Leute\n", 12, 0);
219       fail_if_err (err);
220
221       err = gpgme_data_new (&out);
222       fail_if_err (err);
223
224       err = gpgme_recipients_new (&rset);
225       fail_if_err (err);
226       err = gpgme_recipients_add_name_with_validity (rset, "Bob",
227                                                      GPGME_VALIDITY_FULL);
228       fail_if_err (err);
229       err = gpgme_recipients_add_name_with_validity (rset, "Alpha",
230                                                      GPGME_VALIDITY_FULL);
231       fail_if_err (err);
232
233       err = gpgme_op_encrypt_start (ctx, rset, in, out);
234       fail_if_err (err);
235
236       my_wait ();
237       fail_if_err (op_result.err);
238       print_op_info (ctx);
239       fail_if_err (err);
240
241       fflush (NULL);
242       fputs ("Begin Result:\n", stdout);
243       print_data (out);
244       fputs ("End Result.\n", stdout);
245    
246       gpgme_recipients_release (rset);
247       gpgme_data_release (in);
248       gpgme_data_release (out);
249       gpgme_release (ctx);
250     }
251   while (argc > 1 && !strcmp (argv[1], "--loop"));
252
253   return 0;
254 }
255
256