9238503eea8b0495ee595cb5cef5d49b155a07a4
[gpgme.git] / tests / gpg / t-edit.c
1 /* t-edit.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
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 <unistd.h>
28
29 #include <gpgme.h>
30
31 struct passphrase_cb_info_s {
32     gpgme_ctx_t c;
33     int did_it;
34 };
35
36
37 #define fail_if_err(a) do { if(a) { int my_errno = errno; \
38             fprintf (stderr, "%s:%d: gpgme_error_t %s\n", \
39                  __FILE__, __LINE__, gpgme_strerror(a));   \
40             if ((a) == GPGME_File_Error)                       \
41                    fprintf (stderr, "\terrno=`%s'\n", strerror (my_errno)); \
42                    exit (1); }                               \
43                              } while(0)
44
45 static void
46 flush_data (gpgme_data_t dh)
47 {
48   char buf[100];
49   int ret;
50   
51   ret = gpgme_data_seek (dh, 0, SEEK_SET);
52   if (ret)
53     fail_if_err (GPGME_File_Error);
54   while ((ret = gpgme_data_read (dh, buf, 100)) > 0)
55     fwrite (buf, ret, 1, stdout);
56   if (ret < 0)
57     fail_if_err (GPGME_File_Error);
58 }
59
60
61 static gpgme_error_t
62 passphrase_cb (void *opaque, const char *uid_hint, const char *passphrase_info,
63                int last_was_bad, int fd)
64 {
65   write (fd, "abc\n", 4);
66   return 0;
67 }
68
69
70 gpgme_error_t
71 edit_fnc (void *opaque, gpgme_status_code_t status, const char *args, int fd)
72 {
73   char *result = NULL;
74   gpgme_data_t out = (gpgme_data_t) opaque;
75
76   fputs ("[-- Response --]\n", stdout);
77   flush_data (out); 
78
79   fprintf (stdout, "[-- Code: %i, %s --]\n", status, args);
80
81   if (fd >= 0)
82     {
83       if (!strcmp (args, "keyedit.prompt"))
84         {
85           static int step = 0;
86           
87           switch (step)
88             {
89             case 0:
90               result = "fpr";
91               break;
92             case 1:
93               result = "expire";
94               break;
95
96               /* This fixes the primary user ID so the keylisting
97                  tests will have predictable output.  */
98             case 2:
99               result = "1";
100               break;
101             case 3:
102               result = "primary";
103               break;
104
105             default:
106               result = "quit";
107               break;
108             }
109           step++;
110         }
111       else if (!strcmp (args, "keyedit.save.okay"))
112         result = "Y";
113       else if (!strcmp (args, "keygen.valid"))
114         result = "0";
115     }
116
117   if (result)
118     {
119       write (fd, result, strlen (result));
120       write (fd, "\n", 1);
121     }
122   return 0;
123 }
124
125
126 int 
127 main (int argc, char **argv)
128 {
129   gpgme_ctx_t ctx;
130   gpgme_error_t err;
131   gpgme_data_t out = NULL;
132   gpgme_key_t key = NULL;
133   struct passphrase_cb_info_s info;
134   const char *pattern = "Alpha";
135   char *p;
136
137   do
138     {
139       err = gpgme_new (&ctx);
140       fail_if_err (err);
141       err = gpgme_data_new (&out);
142       fail_if_err (err);
143
144       p = getenv("GPG_AGENT_INFO");
145       if (!(p && strchr (p, ':')))
146         {
147           memset (&info, 0, sizeof info);
148           info.c = ctx;
149           gpgme_set_passphrase_cb (ctx, passphrase_cb, &info);
150         } 
151
152       err = gpgme_op_keylist_start (ctx, pattern, 0);
153       fail_if_err (err);
154       err = gpgme_op_keylist_next (ctx, &key);
155       fail_if_err (err);
156       err = gpgme_op_keylist_end (ctx);
157       fail_if_err (err);
158
159       err = gpgme_op_edit (ctx, key, edit_fnc, out, out);
160       fail_if_err (err);
161
162       fputs ("[-- Last response --]\n", stdout);
163       flush_data (out);
164
165       gpgme_data_release (out);
166       gpgme_key_unref (key);
167       gpgme_release (ctx);
168     }
169   while (argc > 1 && !strcmp( argv[1], "--loop"));
170
171   return 0;
172 }
173
174