Reintroduce a check in edit_fnc.
[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             default:
96               result = "quit";
97               break;
98             }
99           step++;
100         }
101       else if (!strcmp (args, "keyedit.save.okay"))
102         result = "Y";
103       else if (!strcmp (args, "keygen.valid"))
104         result = "0";
105     }
106
107   if (result)
108     {
109       write (fd, result, strlen (result));
110       write (fd, "\n", 1);
111     }
112   return 0;
113 }
114
115
116 int 
117 main (int argc, char **argv)
118 {
119   gpgme_ctx_t ctx;
120   gpgme_error_t err;
121   gpgme_data_t out = NULL;
122   gpgme_key_t key = NULL;
123   struct passphrase_cb_info_s info;
124   const char *pattern = "Alpha";
125   char *p;
126
127   do
128     {
129       err = gpgme_new (&ctx);
130       fail_if_err (err);
131       err = gpgme_data_new (&out);
132       fail_if_err (err);
133
134       p = getenv("GPG_AGENT_INFO");
135       if (!(p && strchr (p, ':')))
136         {
137           memset (&info, 0, sizeof info);
138           info.c = ctx;
139           gpgme_set_passphrase_cb (ctx, passphrase_cb, &info);
140         } 
141
142       err = gpgme_op_keylist_start (ctx, pattern, 0);
143       fail_if_err (err);
144       err = gpgme_op_keylist_next (ctx, &key);
145       fail_if_err (err);
146       err = gpgme_op_keylist_end (ctx);
147       fail_if_err (err);
148
149       err = gpgme_op_edit (ctx, key, edit_fnc, out, out);
150       fail_if_err (err);
151
152       fputs ("[-- Last response --]\n", stdout);
153       flush_data (out);
154
155       gpgme_data_release (out);
156       gpgme_key_unref (key);
157       gpgme_release (ctx);
158     }
159   while (argc > 1 && !strcmp( argv[1], "--loop"));
160
161   return 0;
162 }
163
164