545db7380ec28a8a2d77ae90772ff5a4ecdac1b1
[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, 2004 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, write to the Free Software
19    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
20    02111-1307, USA.  */
21
22 /* We need to include config.h so that we know whether we are building
23    with large file system (LFS) support. */
24 #ifdef HAVE_CONFIG_H
25 #include <config.h>
26 #endif
27
28 #include <stdio.h>
29 #include <stdlib.h>
30 #include <string.h>
31 #include <assert.h>
32 #include <errno.h>
33 #include <unistd.h>
34 #include <errno.h>
35
36 #include <gpgme.h>
37
38 #include "t-support.h"
39
40 \f
41 static void
42 flush_data (gpgme_data_t dh)
43 {
44   char buf[100];
45   int ret;
46   
47   ret = gpgme_data_seek (dh, 0, SEEK_SET);
48   if (ret)
49     fail_if_err (gpgme_error_from_errno (errno));
50   while ((ret = gpgme_data_read (dh, buf, 100)) > 0)
51     fwrite (buf, ret, 1, stdout);
52   if (ret < 0)
53     fail_if_err (gpgme_error_from_errno (errno));
54 }
55
56
57 gpgme_error_t
58 edit_fnc (void *opaque, gpgme_status_code_t status, const char *args, int fd)
59 {
60   char *result = NULL;
61   gpgme_data_t out = (gpgme_data_t) opaque;
62
63   fputs ("[-- Response --]\n", stdout);
64   flush_data (out); 
65
66   fprintf (stdout, "[-- Code: %i, %s --]\n", status, args);
67
68   if (fd >= 0)
69     {
70       if (!strcmp (args, "keyedit.prompt"))
71         {
72           static int step = 0;
73           
74           switch (step)
75             {
76             case 0:
77               result = "fpr";
78               break;
79             case 1:
80               result = "expire";
81               break;
82
83               /* This fixes the primary user ID so the keylisting
84                  tests will have predictable output.  */
85             case 2:
86               result = "1";
87               break;
88             case 3:
89               result = "primary";
90               break;
91
92             default:
93               result = "quit";
94               break;
95             }
96           step++;
97         }
98       else if (!strcmp (args, "keyedit.save.okay"))
99         result = "Y";
100       else if (!strcmp (args, "keygen.valid"))
101         result = "0";
102     }
103
104   if (result)
105     {
106       gpgme_io_write (fd, result, strlen (result));
107       gpgme_io_write (fd, "\n", 1);
108     }
109   return 0;
110 }
111
112
113 int 
114 main (int argc, char **argv)
115 {
116   gpgme_ctx_t ctx;
117   gpgme_error_t err;
118   gpgme_data_t out = NULL;
119   gpgme_key_t key = NULL;
120   const char *pattern = "Alpha";
121   char *agent_info;
122
123   init_gpgme (GPGME_PROTOCOL_OpenPGP);
124
125   err = gpgme_new (&ctx);
126   fail_if_err (err);
127   err = gpgme_data_new (&out);
128   fail_if_err (err);
129
130   agent_info = getenv("GPG_AGENT_INFO");
131   if (!(agent_info && strchr (agent_info, ':')))
132     gpgme_set_passphrase_cb (ctx, passphrase_cb, 0);
133
134   err = gpgme_op_keylist_start (ctx, pattern, 0);
135   fail_if_err (err);
136   err = gpgme_op_keylist_next (ctx, &key);
137   fail_if_err (err);
138   err = gpgme_op_keylist_end (ctx);
139   fail_if_err (err);
140
141   err = gpgme_op_edit (ctx, key, edit_fnc, out, out);
142   fail_if_err (err);
143
144   fputs ("[-- Last response --]\n", stdout);
145   flush_data (out);
146
147   gpgme_data_release (out);
148   gpgme_key_unref (key);
149   gpgme_release (ctx);
150
151   return 0;
152 }