core: New function gpgme_op_tofu_policy
[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 <http://www.gnu.org/licenses/>.
18  */
19
20 /* We need to include config.h so that we know whether we are building
21    with large file system (LFS) support. */
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <stdlib.h>
27 #include <stdio.h>
28 #include <string.h>
29 #include <assert.h>
30
31 #include <gpgme.h>
32
33 #define PGM "run-tofu"
34
35 #include "run-support.h"
36
37
38 static int verbose;
39
40
41 static gpg_error_t
42 status_cb (void *opaque, const char *keyword, const char *value)
43 {
44   (void)opaque;
45   fprintf (stderr, "status_cb: %s %s\n", nonnull(keyword), nonnull(value));
46   return 0;
47 }
48
49
50
51 static gpgme_tofu_policy_t
52 parse_policy_string (const char *string)
53 {
54   gpgme_tofu_policy_t policy;
55
56   if (!strcmp (string, "auto"))
57     policy = GPGME_TOFU_POLICY_AUTO;
58   else if (!strcmp (string, "good"))
59     policy = GPGME_TOFU_POLICY_GOOD;
60   else if (!strcmp (string, "bad"))
61     policy = GPGME_TOFU_POLICY_BAD;
62   else if (!strcmp (string, "ask"))
63     policy = GPGME_TOFU_POLICY_ASK;
64   else if (!strcmp (string, "unknown"))
65     policy = GPGME_TOFU_POLICY_UNKNOWN;
66   else
67     {
68       fprintf (stderr, PGM ": invalid policy value '%s'\n", string);
69       exit (1);
70     }
71
72   return policy;
73 }
74
75
76
77 static int
78 show_usage (int ex)
79 {
80   fputs ("usage: " PGM " [options] FPR\n\n"
81          "Options:\n"
82          "  --policy NAME    Set tofu policy for key to NAME\n"
83          "  --verbose        run in verbose mode\n"
84          "  --status         print status lines from the backend\n"
85          , stderr);
86   exit (ex);
87 }
88
89
90 int
91 main (int argc, char **argv)
92 {
93   int last_argc = -1;
94   gpgme_error_t err;
95   gpgme_ctx_t ctx;
96   gpgme_protocol_t protocol = GPGME_PROTOCOL_OpenPGP;
97   int print_status = 0;
98   gpgme_key_t thekey;
99   const char *fpr;
100   const char *policystr = NULL;
101   gpgme_tofu_policy_t policy;
102
103   if (argc)
104     { argc--; argv++; }
105
106   while (argc && last_argc != argc )
107     {
108       last_argc = argc;
109       if (!strcmp (*argv, "--"))
110         {
111           argc--; argv++;
112           break;
113         }
114       else if (!strcmp (*argv, "--help"))
115         show_usage (0);
116       else if (!strcmp (*argv, "--verbose"))
117         {
118           verbose = 1;
119           argc--; argv++;
120         }
121       else if (!strcmp (*argv, "--status"))
122         {
123           print_status = 1;
124           argc--; argv++;
125         }
126       else if (!strcmp (*argv, "--policy"))
127         {
128           argc--; argv++;
129           if (!argc)
130             show_usage (1);
131           policystr = *argv;
132           argc--; argv++;
133         }
134       else if (!strncmp (*argv, "--", 2))
135         show_usage (1);
136     }
137
138   if (argc != 1)
139     show_usage (1);
140   fpr = argv[0];
141
142   init_gpgme (protocol);
143
144   err = gpgme_new (&ctx);
145   fail_if_err (err);
146   gpgme_set_protocol (ctx, protocol);
147   gpgme_set_armor (ctx, 1);
148   if (print_status)
149     {
150       gpgme_set_status_cb (ctx, status_cb, NULL);
151       gpgme_set_ctx_flag (ctx, "full-status", "1");
152     }
153
154   err = gpgme_get_key (ctx, fpr, &thekey, 0);
155   if (err)
156     {
157       fprintf (stderr, PGM ": error getting key '%s': %s\n",
158                fpr, gpg_strerror (err));
159       exit (1);
160     }
161
162   if (policystr)
163     {
164       policy = parse_policy_string (policystr);
165
166       err = gpgme_op_tofu_policy (ctx, thekey, policy);
167       if (err)
168         {
169           fprintf (stderr, PGM ": gpgme_op_tofu_polciy failed: %s\n",
170                    gpg_strerror (err));
171           exit (1);
172         }
173     }
174
175   gpgme_key_unref (thekey);
176   gpgme_release (ctx);
177   return 0;
178 }