a3a7733806589bd1fb031d42a754560cc8efafdb
[gnupg.git] / agent / query.c
1 /* query.c - fork of the pinentry to query stuff from the user
2  *      Copyright (C) 2001, 2002 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 2 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, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #include <errno.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <ctype.h>
27 #include <assert.h>
28 #include <unistd.h>
29 #include <sys/stat.h>
30 #ifdef USE_GNU_PTH
31 # include <pth.h>
32 #endif
33
34 #include "agent.h"
35 #include "i18n.h"
36 #include <assuan.h>
37
38 #ifdef _POSIX_OPEN_MAX
39 #define MAX_OPEN_FDS _POSIX_OPEN_MAX
40 #else
41 #define MAX_OPEN_FDS 20
42 #endif
43
44 static ASSUAN_CONTEXT entry_ctx = NULL;
45 #ifdef USE_GNU_PTH
46 static pth_mutex_t entry_lock = PTH_MUTEX_INIT;
47 #endif
48
49 /* data to be passed to our callbacks */
50 struct entry_parm_s {
51   int lines;
52   size_t size;
53   char *buffer;
54 };
55
56
57
58 \f
59 /* Unlock the pinentry so that another thread can start one and
60    disconnect that pinentry - we do this after the unlock so that a
61    stalled pinentry does not block other threads.  Fixme: We should
62    have a timeout in Assuan for the disconnetc operation. */
63 static int 
64 unlock_pinentry (int rc)
65 {
66   ASSUAN_CONTEXT ctx = entry_ctx;
67
68 #ifdef USE_GNU_PTH
69   if (!pth_mutex_release (&entry_lock))
70     {
71       log_error ("failed to release the entry lock\n");
72       if (!rc)
73         rc = gpg_error (GPG_ERR_INTERNAL);
74     }
75 #endif
76   entry_ctx = NULL;
77   assuan_disconnect (ctx);
78   return rc;
79 }
80
81
82 /* To make sure we leave no secrets in our image after forking of the
83    pinentry, we use this callback. */
84 static void
85 atfork_cb (void *opaque, int where)
86 {
87   if (!where)
88     gcry_control (GCRYCTL_TERM_SECMEM);
89 }
90
91
92 /* Fork off the pin entry if this has not already been done.  Note,
93    that this function must always be used to aquire the lock for the
94    pinentry - we will serialize _all_ pinentry calls.
95  */
96 static int
97 start_pinentry (CTRL ctrl)
98 {
99   int rc;
100   const char *pgmname;
101   ASSUAN_CONTEXT ctx;
102   const char *argv[5];
103   int no_close_list[3];
104   int i;
105
106 #ifdef USE_GNU_PTH
107   if (!pth_mutex_acquire (&entry_lock, 0, NULL))
108     {
109       log_error ("failed to acquire the entry lock\n");
110       return gpg_error (GPG_ERR_INTERNAL);
111     }
112 #endif
113
114   if (entry_ctx)
115     return 0; 
116
117   if (opt.verbose)
118     log_info ("starting a new PIN Entry\n");
119       
120   if (fflush (NULL))
121     {
122       gpg_error_t tmperr = gpg_error (gpg_err_code_from_errno (errno));
123       log_error ("error flushing pending output: %s\n", strerror (errno));
124       return unlock_pinentry (tmperr);
125     }
126
127   if (!opt.pinentry_program || !*opt.pinentry_program)
128     opt.pinentry_program = GNUPG_DEFAULT_PINENTRY;
129   if ( !(pgmname = strrchr (opt.pinentry_program, '/')))
130     pgmname = opt.pinentry_program;
131   else
132     pgmname++;
133
134   argv[0] = pgmname;
135   if (ctrl->display && !opt.keep_display)
136     {
137       argv[1] = "--display";
138       argv[2] = ctrl->display;
139       argv[3] = NULL;
140     }
141   else
142     argv[1] = NULL;
143   
144   i=0;
145   if (!opt.running_detached)
146     {
147       if (log_get_fd () != -1)
148         no_close_list[i++] = log_get_fd ();
149       no_close_list[i++] = fileno (stderr);
150     }
151   no_close_list[i] = -1;
152
153   /* Connect to the pinentry and perform initial handshaking */
154   rc = assuan_pipe_connect2 (&ctx, opt.pinentry_program, (char**)argv,
155                              no_close_list, atfork_cb, NULL);
156   if (rc)
157     {
158       log_error ("can't connect to the PIN entry module: %s\n",
159                  assuan_strerror (rc));
160       return unlock_pinentry (gpg_error (GPG_ERR_NO_PIN_ENTRY));
161     }
162   entry_ctx = ctx;
163
164   if (DBG_ASSUAN)
165     log_debug ("connection to PIN entry established\n");
166
167   rc = assuan_transact (entry_ctx, 
168                         opt.no_grab? "OPTION no-grab":"OPTION grab",
169                         NULL, NULL, NULL, NULL, NULL, NULL);
170   if (rc)
171     return unlock_pinentry (map_assuan_err (rc));
172   if (ctrl->ttyname)
173     {
174       char *optstr;
175       if (asprintf (&optstr, "OPTION ttyname=%s", ctrl->ttyname) < 0 )
176         return unlock_pinentry (out_of_core ());
177       rc = assuan_transact (entry_ctx, optstr, NULL, NULL, NULL, NULL, NULL,
178                             NULL);
179       free (optstr);
180       if (rc)
181         return unlock_pinentry (map_assuan_err (rc));
182     }
183   if (ctrl->ttytype)
184     {
185       char *optstr;
186       if (asprintf (&optstr, "OPTION ttytype=%s", ctrl->ttytype) < 0 )
187         return unlock_pinentry (out_of_core ());
188       rc = assuan_transact (entry_ctx, optstr, NULL, NULL, NULL, NULL, NULL,
189                             NULL);
190       if (rc)
191         return unlock_pinentry (map_assuan_err (rc));
192     }
193   if (ctrl->lc_ctype)
194     {
195       char *optstr;
196       if (asprintf (&optstr, "OPTION lc-ctype=%s", ctrl->lc_ctype) < 0 )
197         return unlock_pinentry (out_of_core ());
198       rc = assuan_transact (entry_ctx, optstr, NULL, NULL, NULL, NULL, NULL,
199                             NULL);
200       if (rc)
201         return unlock_pinentry (map_assuan_err (rc));
202     }
203   if (ctrl->lc_messages)
204     {
205       char *optstr;
206       if (asprintf (&optstr, "OPTION lc-messages=%s", ctrl->lc_messages) < 0 )
207         return unlock_pinentry (out_of_core ());
208       rc = assuan_transact (entry_ctx, optstr, NULL, NULL, NULL, NULL, NULL,
209                             NULL);
210       if (rc)
211         return unlock_pinentry (map_assuan_err (rc));
212     }
213   return 0;
214 }
215
216
217 static AssuanError
218 getpin_cb (void *opaque, const void *buffer, size_t length)
219 {
220   struct entry_parm_s *parm = opaque;
221
222   if (!buffer)
223     return 0;
224
225   /* we expect the pin to fit on one line */
226   if (parm->lines || length >= parm->size)
227     return ASSUAN_Too_Much_Data;
228
229   /* fixme: we should make sure that the assuan buffer is allocated in
230      secure memory or read the response byte by byte */
231   memcpy (parm->buffer, buffer, length);
232   parm->buffer[length] = 0;
233   parm->lines++;
234   return 0;
235 }
236
237
238 static int
239 all_digitsp( const char *s)
240 {
241   for (; *s && *s >= '0' && *s <= '9'; s++)
242     ;
243   return !*s;
244 }  
245
246
247 \f
248 /* Call the Entry and ask for the PIN.  We do check for a valid PIN
249    number here and repeat it as long as we have invalid formed
250    numbers. */
251 int
252 agent_askpin (CTRL ctrl,
253               const char *desc_text, struct pin_entry_info_s *pininfo)
254 {
255   int rc;
256   char line[ASSUAN_LINELENGTH];
257   struct entry_parm_s parm;
258   const char *errtext = NULL;
259   int is_pin = 0;
260
261   if (opt.batch)
262     return 0; /* fixme: we should return BAD PIN */
263
264   if (!pininfo || pininfo->max_length < 1)
265     return gpg_error (GPG_ERR_INV_VALUE);
266   if (!desc_text && pininfo->min_digits)
267     desc_text = _("Please enter your PIN, so that the secret key "
268                   "can be unlocked for this session");
269   else if (!desc_text)
270     desc_text = _("Please enter your passphrase, so that the secret key "
271                   "can be unlocked for this session");
272
273   is_pin = desc_text && strstr (desc_text, "PIN");
274
275   rc = start_pinentry (ctrl);
276   if (rc)
277     return rc;
278
279   snprintf (line, DIM(line)-1, "SETDESC %s", desc_text);
280   line[DIM(line)-1] = 0;
281   rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
282   if (rc)
283     return unlock_pinentry (map_assuan_err (rc));
284
285   rc = assuan_transact (entry_ctx,
286                         is_pin? "SETPROMPT PIN:"
287                               : "SETPROMPT Passphrase:",
288                         NULL, NULL, NULL, NULL, NULL, NULL);
289   if (rc)
290     return unlock_pinentry (map_assuan_err (rc));
291
292   for (;pininfo->failed_tries < pininfo->max_tries; pininfo->failed_tries++)
293     {
294       memset (&parm, 0, sizeof parm);
295       parm.size = pininfo->max_length;
296       parm.buffer = pininfo->pin;
297
298       if (errtext)
299         { 
300           /* fixme: should we show the try count? It must be translated */
301           snprintf (line, DIM(line)-1, "SETERROR %s (try %d of %d)",
302                     errtext, pininfo->failed_tries+1, pininfo->max_tries);
303           line[DIM(line)-1] = 0;
304           rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
305           if (rc)
306             return unlock_pinentry (map_assuan_err (rc));
307           errtext = NULL;
308         }
309       
310       rc = assuan_transact (entry_ctx, "GETPIN", getpin_cb, &parm,
311                             NULL, NULL, NULL, NULL);
312       if (rc == ASSUAN_Too_Much_Data)
313         errtext = is_pin? _("PIN too long")
314                         : _("Passphrase too long");
315       else if (rc)
316         return unlock_pinentry (map_assuan_err (rc));
317
318       if (!errtext && pininfo->min_digits)
319         {
320           /* do some basic checks on the entered PIN. */
321           if (!all_digitsp (pininfo->pin))
322             errtext = _("Invalid characters in PIN");
323           else if (pininfo->max_digits
324                    && strlen (pininfo->pin) > pininfo->max_digits)
325             errtext = _("PIN too long");
326           else if (strlen (pininfo->pin) < pininfo->min_digits)
327             errtext = _("PIN too short");
328         }
329
330       if (!errtext && pininfo->check_cb)
331         {
332           /* More checks by utilizing the optional callback. */
333           pininfo->cb_errtext = NULL;
334           rc = pininfo->check_cb (pininfo);
335           if (rc == -1 && pininfo->cb_errtext)
336             errtext = pininfo->cb_errtext;
337           else if (gpg_err_code (rc) == GPG_ERR_BAD_PASSPHRASE
338                    || gpg_err_code (rc) == GPG_ERR_BAD_PIN)
339             errtext = (is_pin? _("Bad PIN")
340                        : _("Bad Passphrase"));
341           else if (rc)
342             return unlock_pinentry (map_assuan_err (rc));
343         }
344
345       if (!errtext)
346         return unlock_pinentry (0); /* okay, got a PIN or passphrase */
347     }
348
349   return unlock_pinentry (gpg_error (pininfo->min_digits? GPG_ERR_BAD_PIN
350                           : GPG_ERR_BAD_PASSPHRASE));
351 }
352
353
354 \f
355 /* Ask for the passphrase using the supplied arguments.  The
356    passphrase is returned in RETPASS as an hex encoded string to be
357    freed by the caller */
358 int 
359 agent_get_passphrase (CTRL ctrl,
360                       char **retpass, const char *desc, const char *prompt,
361                       const char *errtext)
362 {
363
364   int rc;
365   char line[ASSUAN_LINELENGTH];
366   struct entry_parm_s parm;
367   unsigned char *p, *hexstring;
368   int i;
369
370   *retpass = NULL;
371   if (opt.batch)
372     return gpg_error (GPG_ERR_BAD_PASSPHRASE); 
373
374   rc = start_pinentry (ctrl);
375   if (rc)
376     return rc;
377
378   if (!prompt)
379     prompt = desc && strstr (desc, "PIN")? "PIN": _("Passphrase");
380
381
382   if (desc)
383     snprintf (line, DIM(line)-1, "SETDESC %s", desc);
384   else
385     snprintf (line, DIM(line)-1, "RESET");
386   line[DIM(line)-1] = 0;
387   rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
388   if (rc)
389     return unlock_pinentry (map_assuan_err (rc));
390
391   snprintf (line, DIM(line)-1, "SETPROMPT %s", prompt);
392   line[DIM(line)-1] = 0;
393   rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
394   if (rc)
395     return unlock_pinentry (map_assuan_err (rc));
396
397   if (errtext)
398     {
399       snprintf (line, DIM(line)-1, "SETERROR %s", errtext);
400       line[DIM(line)-1] = 0;
401       rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
402       if (rc)
403         return unlock_pinentry (map_assuan_err (rc));
404     }
405
406   memset (&parm, 0, sizeof parm);
407   parm.size = ASSUAN_LINELENGTH/2 - 5;
408   parm.buffer = gcry_malloc_secure (parm.size+10);
409   if (!parm.buffer)
410     return unlock_pinentry (out_of_core ());
411
412   assuan_begin_confidential (entry_ctx);
413   rc = assuan_transact (entry_ctx, "GETPIN", getpin_cb, &parm, NULL, NULL, NULL, NULL);
414   if (rc)
415     {
416       xfree (parm.buffer);
417       return unlock_pinentry (map_assuan_err (rc));
418     }
419   
420   hexstring = gcry_malloc_secure (strlen (parm.buffer)*2+1);
421   if (!hexstring)
422     {
423       gpg_error_t tmperr = out_of_core ();
424       xfree (parm.buffer);
425       return unlock_pinentry (tmperr);
426     }
427
428   for (i=0, p=parm.buffer; *p; p++, i += 2)
429     sprintf (hexstring+i, "%02X", *p);
430   
431   xfree (parm.buffer);
432   *retpass = hexstring;
433   return unlock_pinentry (0);
434 }
435
436
437 \f
438 /* Pop up the PIN-entry, display the text and the prompt and ask the
439    user to confirm this.  We return 0 for success, ie. the used
440    confirmed it, GPG_ERR_NOT_CONFIRMED for what the text says or an
441    other error. */
442 int 
443 agent_get_confirmation (CTRL ctrl,
444                         const char *desc, const char *ok, const char *cancel)
445 {
446   int rc;
447   char line[ASSUAN_LINELENGTH];
448
449   rc = start_pinentry (ctrl);
450   if (rc)
451     return rc;
452
453   if (desc)
454     snprintf (line, DIM(line)-1, "SETDESC %s", desc);
455   else
456     snprintf (line, DIM(line)-1, "RESET");
457   line[DIM(line)-1] = 0;
458   rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
459   if (rc)
460     return unlock_pinentry (map_assuan_err (rc));
461
462   if (ok)
463     {
464       snprintf (line, DIM(line)-1, "SETOK %s", ok);
465       line[DIM(line)-1] = 0;
466       rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
467       if (rc)
468         return unlock_pinentry (map_assuan_err (rc));
469     }
470   if (cancel)
471     {
472       snprintf (line, DIM(line)-1, "SETCANCEL %s", cancel);
473       line[DIM(line)-1] = 0;
474       rc = assuan_transact (entry_ctx, line, NULL, NULL, NULL, NULL, NULL, NULL);
475       if (rc)
476         return unlock_pinentry (map_assuan_err (rc));
477     }
478
479   rc = assuan_transact (entry_ctx, "CONFIRM", NULL, NULL, NULL, NULL, NULL, NULL);
480   return unlock_pinentry (map_assuan_err (rc));
481 }
482
483
484