db7b7d5a606ee22bd5484aecc5a4ecf2a3902e19
[gpgme.git] / tests / run-support.h
1 /* run-support.h - Helper routines for run-* test programs.
2    Copyright (C) 2000 Werner Koch (dd9jn)
3    Copyright (C) 2001, 2002, 2003, 2004, 2009 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, see <http://www.gnu.org/licenses/>.
19 */
20
21 #include <unistd.h>
22 #include <errno.h>
23 #include <stdlib.h>
24 #include <locale.h>
25
26 #ifdef HAVE_W32_SYSTEM
27 #include <windows.h>
28 #endif
29
30 #include <gpgme.h>
31
32 #ifndef DIM
33 #define DIM(v)               (sizeof(v)/sizeof((v)[0]))
34 #endif
35
36 #define fail_if_err(err)                                        \
37   do                                                            \
38     {                                                           \
39       if (err)                                                  \
40         {                                                       \
41           fprintf (stderr, PGM": file %s line %d: <%s> %s\n",   \
42                    __FILE__, __LINE__, gpgme_strsource (err),   \
43                    gpgme_strerror (err));                       \
44           exit (1);                                             \
45         }                                                       \
46     }                                                           \
47   while (0)
48
49
50 static const char *
51 nonnull (const char *s)
52 {
53   return s? s :"[none]";
54 }
55
56
57 void
58 print_data (gpgme_data_t dh)
59 {
60 #define BUF_SIZE 512
61   char buf[BUF_SIZE + 1];
62   int ret;
63   
64   ret = gpgme_data_seek (dh, 0, SEEK_SET);
65   if (ret)
66     fail_if_err (gpgme_err_code_from_errno (errno));
67   while ((ret = gpgme_data_read (dh, buf, BUF_SIZE)) > 0)
68     fwrite (buf, ret, 1, stdout);
69   if (ret < 0)
70     fail_if_err (gpgme_err_code_from_errno (errno));
71 }
72
73
74 gpgme_error_t
75 passphrase_cb (void *opaque, const char *uid_hint, const char *passphrase_info,
76                int last_was_bad, int fd)
77 {
78 #ifdef HAVE_W32_SYSTEM
79   DWORD written;
80   WriteFile ((HANDLE) fd, "abc\n", 4, &written, 0);
81 #else
82   int res;
83   char *pass = "abc\n";
84   int passlen = strlen (pass);
85   int off = 0;
86
87   do
88     {
89       res = write (fd, &pass[off], passlen - off);
90       if (res > 0)
91         off += res;
92     }
93   while (res > 0 && off != passlen);
94
95   return off == passlen ? 0 : gpgme_error_from_errno (errno);
96 #endif
97
98   return 0;
99 }
100
101
102 char *
103 make_filename (const char *fname)
104 {
105   const char *srcdir = getenv ("srcdir");
106   char *buf;
107
108   if (!srcdir)
109     srcdir = ".";
110   buf = malloc (strlen(srcdir) + strlen(fname) + 2);
111   if (!buf) 
112     exit (8);
113   strcpy (buf, srcdir);
114   strcat (buf, "/");
115   strcat (buf, fname);
116   return buf;
117 }
118
119
120 void
121 init_gpgme (gpgme_protocol_t proto)
122 {
123   gpgme_error_t err;
124
125   gpgme_check_version (NULL);
126   setlocale (LC_ALL, "");
127   gpgme_set_locale (NULL, LC_CTYPE, setlocale (LC_CTYPE, NULL));
128 #ifndef HAVE_W32_SYSTEM
129   gpgme_set_locale (NULL, LC_MESSAGES, setlocale (LC_MESSAGES, NULL));
130 #endif
131
132   err = gpgme_engine_check_version (proto);
133   fail_if_err (err);
134 }
135
136
137 void
138 print_import_result (gpgme_import_result_t r)
139 {
140   gpgme_import_status_t st;
141
142   for (st=r->imports; st; st = st->next)
143     {
144       printf ("  fpr: %s err: %d (%s) status:", nonnull (st->fpr),
145               st->result, gpgme_strerror (st->result));
146       if (st->status & GPGME_IMPORT_NEW)
147         fputs (" new", stdout);
148       if (st->status & GPGME_IMPORT_UID)
149         fputs (" uid", stdout);
150       if (st->status & GPGME_IMPORT_SIG)
151         fputs (" sig", stdout);
152       if (st->status & GPGME_IMPORT_SUBKEY)
153         fputs (" subkey", stdout);
154       if (st->status & GPGME_IMPORT_SECRET)
155         fputs (" secret", stdout);
156       putchar ('\n');
157     }
158   printf ("key import summary:\n"
159           "        considered: %d\n"
160           "        no user id: %d\n"
161           "          imported: %d\n"
162           "      imported_rsa: %d\n"
163           "         unchanged: %d\n"
164           "      new user ids: %d\n"
165           "       new subkeys: %d\n"
166           "    new signatures: %d\n"
167           "   new revocations: %d\n"
168           "       secret read: %d\n"
169           "   secret imported: %d\n"
170           "  secret unchanged: %d\n"
171           "  skipped new keys: %d\n"
172           "      not imported: %d\n",
173           r->considered,
174           r->no_user_id,
175           r->imported,
176           r->imported_rsa,
177           r->unchanged,
178           r->new_user_ids,
179           r->new_sub_keys,
180           r->new_signatures,
181           r->new_revocations,
182           r->secret_read,
183           r->secret_imported,
184           r->secret_unchanged,
185           r->skipped_new_keys,
186           r->not_imported);
187 }
188