Change stack size for Wince.
[gnupg.git] / agent / preset-passphrase.c
1 /* preset-passphrase.c - A tool to preset a passphrase.
2  *      Copyright (C) 2004 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <config.h>
21
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <stddef.h>
25 #include <stdarg.h>
26 #include <string.h>
27 #include <errno.h>
28 #include <assert.h>
29 #include <sys/stat.h>
30 #include <unistd.h>
31 #ifdef HAVE_LOCALE_H
32 #include <locale.h>
33 #endif
34 #ifdef HAVE_LANGINFO_CODESET
35 #include <langinfo.h>
36 #endif
37 #ifdef HAVE_DOSISH_SYSTEM
38 #include <fcntl.h> /* for setmode() */
39 #endif
40 #ifdef HAVE_W32_SYSTEM
41 #include <windows.h>  /* To initialize the sockets.  fixme */
42 #endif
43
44 #define JNLIB_NEED_LOG_LOGV
45 #include "agent.h"
46 #include "simple-pwquery.h"
47 #include "i18n.h"
48 #include "sysutils.h"
49
50
51 enum cmd_and_opt_values 
52 { aNull = 0,
53   oVerbose        = 'v',
54   oPassphrase     = 'P',
55
56   oPreset         = 'c',
57   oForget         = 'f',
58   
59   oNoVerbose = 500,
60
61   oHomedir,
62
63 aTest };
64
65
66 static const char *opt_homedir;
67 static const char *opt_passphrase;
68
69 static ARGPARSE_OPTS opts[] = {
70   
71   { 301, NULL, 0, N_("@Options:\n ") },
72
73   { oVerbose, "verbose",   0, "verbose" },
74   { oPassphrase, "passphrase", 2, "|STRING|use passphrase STRING" },
75   { oPreset,  "preset",   256, "preset passphrase"},
76   { oForget,  "forget",  256, "forget passphrase"},
77
78   { oHomedir, "homedir", 2, "@" }, 
79   {0}
80 };
81
82
83 static const char *
84 my_strusage (int level)
85 {
86   const char *p;
87   switch (level)
88     {
89     case 11: p = "gpg-preset-passphrase (GnuPG)";
90       break;
91     case 13: p = VERSION; break;
92     case 17: p = PRINTABLE_OS_NAME; break;
93     case 19: p = _("Please report bugs to <@EMAIL@>.\n"); break;
94
95     case 1:
96     case 40: 
97       p =  _("Usage: gpg-preset-passphrase [options] KEYGRIP (-h for help)\n");
98       break;
99     case 41:
100       p = _("Syntax: gpg-preset-passphrase [options] KEYGRIP\n"
101                     "Password cache maintenance\n");
102     break;
103     
104     default: p = NULL;
105     }
106   return p;
107 }
108
109
110 \f
111
112 /* Include the implementation of map_spwq_error.  */
113 MAP_SPWQ_ERROR_IMPL
114       
115
116 static void
117 preset_passphrase (const char *keygrip)
118 {
119   int  rc;
120   char *line;
121   /* FIXME: Use secure memory.  */
122   char passphrase[500];
123   char *passphrase_esc;
124
125   if (!opt_passphrase)
126     {
127       rc = read (0, passphrase, sizeof (passphrase) - 1);
128       if (rc < 0)
129         {
130           log_error ("reading passphrase failed: %s\n",
131                      gpg_strerror (gpg_error_from_syserror ()));
132           return;
133         }
134       passphrase[rc] = '\0';
135       line = strchr (passphrase, '\n');
136       if (line)
137         {
138           if (line > passphrase && line[-1] == '\r')
139             line--;
140           *line = '\0';
141         }
142
143       /* FIXME: How to handle empty passwords?  */
144     }
145
146   {
147     const char *s = opt_passphrase ? opt_passphrase : passphrase;
148     passphrase_esc = bin2hex (s, strlen (s), NULL);
149   }
150   if (!passphrase_esc)
151     {
152       log_error ("can not escape string: %s\n",
153                  gpg_strerror (gpg_error_from_syserror ()));
154       return;
155     }
156
157   rc = asprintf (&line, "PRESET_PASSPHRASE %s -1 %s\n", keygrip,
158                  passphrase_esc);
159   wipememory (passphrase_esc, strlen (passphrase_esc));
160   xfree (passphrase_esc);
161
162   if (rc < 0)
163     {
164       log_error ("caching passphrase failed: %s\n",
165                  gpg_strerror (gpg_error_from_syserror ()));
166       return;
167     }
168   if (!opt_passphrase)
169     wipememory (passphrase, sizeof (passphrase));
170
171   rc = map_spwq_error (simple_query (line));
172   if (rc)
173     {
174       log_error ("caching passphrase failed: %s\n", gpg_strerror (rc));
175       return;
176     }
177
178   wipememory (line, strlen (line));
179   xfree (line);
180 }
181
182
183 static void
184 forget_passphrase (const char *keygrip)
185 {
186   int rc;
187   char *line;
188
189   rc = asprintf (&line, "CLEAR_PASSPHRASE %s\n", keygrip);
190   if (rc < 0)
191     rc = gpg_error_from_syserror ();
192   else
193     rc = map_spwq_error (simple_query (line));
194   if (rc)
195     {
196       log_error ("clearing passphrase failed: %s\n", gpg_strerror (rc));
197       return;
198     }
199
200   xfree (line);
201 }
202
203 \f
204 int
205 main (int argc, char **argv)
206 {
207   ARGPARSE_ARGS pargs;
208   int cmd = 0;
209   const char *keygrip = NULL;
210
211   set_strusage (my_strusage);
212   log_set_prefix ("gpg-preset-passphrase", 1); 
213
214   /* Make sure that our subsystems are ready.  */
215   i18n_init ();
216   init_common_subsystems (&argc, &argv);
217
218   opt_homedir = default_homedir ();
219
220   pargs.argc = &argc;
221   pargs.argv = &argv;
222   pargs.flags=  1;  /* (do not remove the args) */
223   while (arg_parse (&pargs, opts) )
224     {
225       switch (pargs.r_opt)
226         {
227         case oVerbose: opt.verbose++; break;
228         case oHomedir: opt_homedir = pargs.r.ret_str; break;
229
230         case oPreset: cmd = oPreset; break;
231         case oForget: cmd = oForget; break;
232         case oPassphrase: opt_passphrase = pargs.r.ret_str; break;
233           
234         default : pargs.err = 2; break;
235         }
236     }
237   if (log_get_errorcount(0))
238     exit(2);
239
240   if (argc == 1)
241     keygrip = *argv;
242   else
243     usage (1);
244
245   /* Tell simple-pwquery about the the standard socket name.  */
246   {
247     char *tmp = make_filename (opt_homedir, "S.gpg-agent", NULL);
248     simple_pw_set_socket (tmp);
249     xfree (tmp);
250   }
251
252   if (cmd == oPreset)
253     preset_passphrase (keygrip);
254   else if (cmd == oForget)
255     forget_passphrase (keygrip);
256   else
257     log_error ("one of the options --preset or --forget must be given\n");
258
259   agent_exit (0);
260   return 8; /*NOTREACHED*/
261 }
262
263
264 void
265 agent_exit (int rc)
266 {
267   rc = rc? rc : log_get_errorcount(0)? 2 : 0;
268   exit (rc);
269 }