core: Fix duplication of close_notify_handler for gpgsm.
[gpgme.git] / tests / run-tofu.c
1 /* run-tofu.c  - Test tool for Tofu functions
2  * Copyright (C) 2016 g10 Code GmbH
3  *
4  * This file is part of GPGME.
5  *
6  * GPGME is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of
9  * the License, or (at your option) any later version.
10  *
11  * GPGME is distributed in the hope that it will be useful, but
12  * WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this program; if not, see <https://gnu.org/licenses/>.
18  * SPDX-License-Identifier: LGPL-2.1-or-later
19  */
20
21 /* We need to include config.h so that we know whether we are building
22    with large file system (LFS) support. */
23 #ifdef HAVE_CONFIG_H
24 #include <config.h>
25 #endif
26
27 #include <stdlib.h>
28 #include <stdio.h>
29 #include <string.h>
30 #include <assert.h>
31
32 #include <gpgme.h>
33
34 #define PGM "run-tofu"
35
36 #include "run-support.h"
37
38
39 static int verbose;
40
41
42 static gpg_error_t
43 status_cb (void *opaque, const char *keyword, const char *value)
44 {
45   (void)opaque;
46   fprintf (stderr, "status_cb: %s %s\n", nonnull(keyword), nonnull(value));
47   return 0;
48 }
49
50
51
52 static gpgme_tofu_policy_t
53 parse_policy_string (const char *string)
54 {
55   gpgme_tofu_policy_t policy;
56
57   if (!strcmp (string, "auto"))
58     policy = GPGME_TOFU_POLICY_AUTO;
59   else if (!strcmp (string, "good"))
60     policy = GPGME_TOFU_POLICY_GOOD;
61   else if (!strcmp (string, "bad"))
62     policy = GPGME_TOFU_POLICY_BAD;
63   else if (!strcmp (string, "ask"))
64     policy = GPGME_TOFU_POLICY_ASK;
65   else if (!strcmp (string, "unknown"))
66     policy = GPGME_TOFU_POLICY_UNKNOWN;
67   else
68     {
69       fprintf (stderr, PGM ": invalid policy value '%s'\n", string);
70       exit (1);
71     }
72
73   return policy;
74 }
75
76
77
78 static int
79 show_usage (int ex)
80 {
81   fputs ("usage: " PGM " [options] FPR\n\n"
82          "Options:\n"
83          "  --policy NAME    Set tofu policy for key to NAME\n"
84          "  --verbose        run in verbose mode\n"
85          "  --status         print status lines from the backend\n"
86          , stderr);
87   exit (ex);
88 }
89
90
91 int
92 main (int argc, char **argv)
93 {
94   int last_argc = -1;
95   gpgme_error_t err;
96   gpgme_ctx_t ctx;
97   gpgme_protocol_t protocol = GPGME_PROTOCOL_OpenPGP;
98   int print_status = 0;
99   gpgme_key_t thekey;
100   const char *fpr;
101   const char *policystr = NULL;
102   gpgme_tofu_policy_t policy;
103   const char *s;
104
105   if (argc)
106     { argc--; argv++; }
107
108   while (argc && last_argc != argc )
109     {
110       last_argc = argc;
111       if (!strcmp (*argv, "--"))
112         {
113           argc--; argv++;
114           break;
115         }
116       else if (!strcmp (*argv, "--help"))
117         show_usage (0);
118       else if (!strcmp (*argv, "--verbose"))
119         {
120           verbose = 1;
121           argc--; argv++;
122         }
123       else if (!strcmp (*argv, "--status"))
124         {
125           print_status = 1;
126           argc--; argv++;
127         }
128       else if (!strcmp (*argv, "--policy"))
129         {
130           argc--; argv++;
131           if (!argc)
132             show_usage (1);
133           policystr = *argv;
134           argc--; argv++;
135         }
136       else if (!strncmp (*argv, "--", 2))
137         show_usage (1);
138     }
139
140   if (argc != 1)
141     show_usage (1);
142   fpr = argv[0];
143
144   init_gpgme (protocol);
145
146   err = gpgme_new (&ctx);
147   fail_if_err (err);
148   gpgme_set_protocol (ctx, protocol);
149   gpgme_set_armor (ctx, 1);
150
151
152   s = gpgme_get_ctx_flag (ctx, "no_such-flag");
153   if (s)
154     {
155       fprintf (stderr, PGM ": gpgme_get_ctx_flag failed "
156                "(bad name not detected)\n");
157       exit (1);
158     }
159   s = gpgme_get_ctx_flag (ctx, "full-status");
160   if (!s || *s)
161     {
162       fprintf (stderr, PGM ": gpgme_get_ctx_flag failed (wrong false)\n");
163       exit (1);
164     }
165   if (print_status)
166     {
167       gpgme_set_status_cb (ctx, status_cb, NULL);
168       gpgme_set_ctx_flag (ctx, "full-status", "1");
169       s = gpgme_get_ctx_flag (ctx, "full-status");
170       if (!s || strcmp (s, "1"))
171         {
172           fprintf (stderr, PGM ": gpgme_get_ctx_flag fauled (wrong true)\n");
173           exit (1);
174         }
175     }
176
177   err = gpgme_get_key (ctx, fpr, &thekey, 0);
178   if (err)
179     {
180       fprintf (stderr, PGM ": error getting key '%s': %s\n",
181                fpr, gpg_strerror (err));
182       exit (1);
183     }
184
185   if (policystr)
186     {
187       policy = parse_policy_string (policystr);
188
189       err = gpgme_op_tofu_policy (ctx, thekey, policy);
190       if (err)
191         {
192           fprintf (stderr, PGM ": gpgme_op_tofu_polciy failed: %s\n",
193                    gpg_strerror (err));
194           exit (1);
195         }
196     }
197
198   gpgme_key_unref (thekey);
199   gpgme_release (ctx);
200   return 0;
201 }