src/libpwmd.c

   1 /*
   2     Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015
   3     Ben Kibbey <bjk@luxsci.net>
   4
   5     This file is part of libpwmd.
   6
   7     Libpwmd is free software: you can redistribute it and/or modify
   8     it under the terms of the GNU General Public License as published by
   9     the Free Software Foundation, either version 2 of the License, or
  10     (at your option) any later version.
  11
  12     Libpwmd is distributed in the hope that it will be useful,
  13     but WITHOUT ANY WARRANTY; without even the implied warranty of
  14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15     GNU General Public License for more details.
  16
  17     You should have received a copy of the GNU General Public License
  18     along with Libpwmd.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <err.h>
  28 #include <errno.h>
  29 #include <ctype.h>
  30 #include <string.h>
  31 #include <sys/types.h>
  32 #include <sys/socket.h>
  33 #include <sys/un.h>
  34 #include <signal.h>
  35 #include <stdarg.h>
  36 #include <string.h>
  37 #include <sys/wait.h>
  38 #include <fcntl.h>
  39 #include <pwd.h>
  40 #include <time.h>
  41 #include <limits.h>
  42 #include <sys/select.h>
  43 #include <termios.h>
  44 #include <libpwmd.h>
  45
  46 #ifdef HAVE_STRINGS_H
  47 #include <strings.h>
  48 #endif
  49
  50 #ifndef LINE_MAX
  51 #define LINE_MAX        2048
  52 #endif
  53
  54 #include "mem.h"
  55 #include "misc.h"
  56 #include "types.h"
  57
  58 #ifdef WITH_PINENTRY
  59 #include "pinentry.h"
  60 #endif
  61
  62 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
  63 #include <sys/types.h>
  64 #include <sys/socket.h>
  65 #include <netdb.h>
  66 #include <netinet/in.h>
  67 #endif
  68
  69 #define FINISH(rc) (gpg_err_source(rc) == GPG_ERR_SOURCE_UNKNOWN) \
  70                                         ? gpg_error(rc) : rc
  71
  72 typedef struct
  73 {
  74   size_t len;
  75   void *buf;
  76 } membuf_t;
  77
  78 ssize_t
  79 hook_read (assuan_context_t ctx, assuan_fd_t fd, void *data, size_t len)
  80 {
  81 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
  82   pwm_t *pwm = assuan_get_pointer (ctx);
  83
  84 #ifdef WITH_SSH
  85   if (pwm && pwm->tcp && pwm->tcp->ssh)
  86     return read_hook_ssh (pwm->tcp->ssh, fd, data, len);
  87 #endif
  88 #ifdef WITH_GNUTLS
  89   if (pwm && pwm->tcp && pwm->tcp->tls)
  90     return read_hook_tls (pwm, fd, data, len);
  91 #endif
  92 #endif
  93
  94   return read ((int) fd, data, len);
  95 }
  96
  97 ssize_t
  98 hook_write (assuan_context_t ctx, assuan_fd_t fd, const void *data,
  99             size_t len)
 100 {
 101   ssize_t wrote;
 102 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
 103   pwm_t *pwm = assuan_get_pointer (ctx);
 104
 105 #ifdef WITH_SSH
 106   if (pwm && pwm->tcp && pwm->tcp->ssh)
 107     return write_hook_ssh (pwm->tcp->ssh, fd, data, len);
 108 #endif
 109 #ifdef WITH_GNUTLS
 110   if (pwm && pwm->tcp && pwm->tcp->tls)
 111     return write_hook_tls (pwm, fd, data, len);
 112 #endif
 113 #endif
 114
 115   /* libassuan cannot handle EAGAIN when doing writes. */
 116   do
 117     {
 118       wrote = write ((int) fd, data, len);
 119       if (wrote == -1 && errno == EAGAIN)
 120         usleep (50000);
 121     }
 122   while (wrote == -1 && errno == EAGAIN);
 123
 124   return wrote;
 125 }
 126
 127 pid_t
 128 hook_waitpid (assuan_context_t ctx, pid_t pid, int action, int *status,
 129               int options)
 130 {
 131   return waitpid (pid, status, options);
 132 }
 133
 134 gpg_error_t
 135 pwmd_init ()
 136 {
 137   static int initialized;
 138
 139 #ifdef WITH_GNUTLS
 140   // May be called more than once.
 141   gnutls_global_init ();
 142 #endif
 143
 144   if (initialized)
 145     return 0;
 146
 147 #ifndef MEM_DEBUG
 148   _xmem_init ();
 149 #endif
 150 #ifdef ENABLE_NLS
 151   bindtextdomain ("libpwmd", LOCALEDIR);
 152 #endif
 153 #ifdef WITH_SSH
 154   libssh2_init (0);
 155 #endif
 156   gpg_err_init ();
 157   initialized = 1;
 158   return 0;
 159 }
 160
 161 void
 162 pwmd_deinit ()
 163 {
 164 #ifdef WITH_GNUTLS
 165   gnutls_global_deinit ();
 166 #endif
 167 }
 168
 169 gpg_error_t
 170 _connect_finalize (pwm_t * pwm)
 171 {
 172   gpg_error_t rc = 0;
 173   char *result = NULL;
 174   int active[2];
 175   int n = assuan_get_active_fds (pwm->ctx, 0, active, N_ARRAY (active));
 176
 177   if (n <= 0)
 178     return GPG_ERR_EBADFD;
 179
 180   pwm->fd = active[0];
 181 #ifdef WITH_PINENTRY
 182   pwm->pinentry_pid = -1;
 183 #endif
 184
 185   rc = pwmd_command (pwm, &result, NULL, NULL, NULL, "GETINFO VERSION");
 186   if (!rc)
 187     {
 188       pwm->version = strtoul (result, NULL, 16);
 189       pwmd_free (result);
 190     }
 191
 192   if (!rc && pwm->name)
 193     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION NAME=%s",
 194                        pwm->name);
 195
 196   return rc;
 197 }
 198
 199 static gpg_error_t
 200 connect_uds (pwm_t * pwm, const char *path)
 201 {
 202   char *socketpath = NULL;
 203   struct passwd pw;
 204   char *pwbuf;
 205   gpg_error_t rc;
 206
 207   if (!pwm)
 208     return GPG_ERR_INV_ARG;
 209
 210   pwbuf = _getpwuid (&pw);
 211   if (!pwbuf)
 212     return gpg_error_from_syserror ();
 213
 214   if (!path || !*path)
 215     socketpath = pwmd_strdup_printf ("%s/.pwmd/socket", pw.pw_dir);
 216   else
 217     socketpath = _expand_homedir ((char *) path, &pw);
 218
 219   pwmd_free (pwbuf);
 220   if (!socketpath)
 221     return GPG_ERR_ENOMEM;
 222
 223   rc = assuan_socket_connect (pwm->ctx, socketpath, ASSUAN_INVALID_FD, 0);
 224   pwmd_free (socketpath);
 225   return rc ? rc : _connect_finalize (pwm);
 226 }
 227
 228 static gpg_error_t
 229 init_handle (pwm_t * pwm)
 230 {
 231   gpg_error_t rc;
 232   static struct assuan_malloc_hooks mhooks = {
 233     pwmd_malloc, pwmd_realloc, pwmd_free
 234   };
 235   static struct assuan_system_hooks shooks = {
 236     ASSUAN_SYSTEM_HOOKS_VERSION,
 237     __assuan_usleep,
 238     __assuan_pipe,
 239     __assuan_close,
 240     hook_read,
 241     hook_write,
 242     //FIXME
 243     NULL,                       //recvmsg
 244     NULL,                       //sendmsg both are used for FD passing
 245     __assuan_spawn,
 246     hook_waitpid,
 247     __assuan_socketpair,
 248     __assuan_socket,
 249     __assuan_connect
 250   };
 251
 252   rc = assuan_new_ext (&pwm->ctx, GPG_ERR_SOURCE_DEFAULT, &mhooks, NULL,
 253                        NULL);
 254   if (rc)
 255     return rc;
 256
 257   assuan_set_pointer (pwm->ctx, pwm);
 258   assuan_ctx_set_system_hooks (pwm->ctx, &shooks);
 259   return 0;
 260 }
 261
 262 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
 263 void
 264 free_tcp (pwm_t *pwm)
 265 {
 266   struct tcp_s *tcp = pwm->tcp;
 267
 268   if (!tcp)
 269     return;
 270
 271 #ifdef WITH_SSH
 272   _free_ssh_conn (tcp->ssh);
 273 #endif
 274 #ifdef WITH_GNUTLS
 275   tls_free (pwm);
 276 #endif
 277
 278   pwmd_free (tcp->host);
 279   if (tcp->addrs)
 280     {
 281       freeaddrinfo (tcp->addrs);
 282       tcp->addrs = NULL;
 283     }
 284
 285   pwmd_free (tcp);
 286   pwm->tcp = NULL;
 287 }
 288 #endif
 289
 290 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
 291 gpg_error_t
 292 tcp_connect_common (pwm_t * pwm)
 293 {
 294   struct addrinfo hints = { 0 };
 295   int n;
 296   char portstr[6];
 297   gpg_error_t rc = 0;
 298
 299   switch (pwm->prot)
 300     {
 301     case PWMD_IP_ANY:
 302       hints.ai_family = AF_UNSPEC;
 303       break;
 304     case PWMD_IPV4:
 305       hints.ai_family = AF_INET;
 306       break;
 307     case PWMD_IPV6:
 308       hints.ai_family = AF_INET6;
 309       break;
 310     }
 311
 312   hints.ai_socktype = SOCK_STREAM;
 313   snprintf (portstr, sizeof (portstr), "%i", pwm->tcp->port);
 314   n = getaddrinfo (pwm->tcp->host, portstr, &hints, &pwm->tcp->addrs);
 315   if (n)
 316     {
 317       fprintf (stderr, "%s\n", gai_strerror (n));
 318       return GPG_ERR_UNKNOWN_HOST;      //FIXME
 319     }
 320
 321   for (pwm->tcp->addr = pwm->tcp->addrs; pwm->tcp->addr;
 322        pwm->tcp->addr = pwm->tcp->addrs->ai_next)
 323     {
 324       pwm->fd = socket (pwm->tcp->addr->ai_family, SOCK_STREAM, 0);
 325       if (pwm->fd == -1)
 326         {
 327           rc = gpg_error_from_syserror ();
 328           if (pwm->tcp->addr == pwm->tcp->addrs->ai_next)
 329             break;
 330           continue;
 331         }
 332
 333       if (pwm->socket_timeout)
 334         {
 335           if (fcntl (pwm->fd, F_SETFL, O_NONBLOCK) == -1)
 336             {
 337               rc = gpg_error_from_syserror ();
 338               break;
 339             }
 340         }
 341
 342       if (connect (pwm->fd, pwm->tcp->addr->ai_addr,
 343                    pwm->tcp->addr->ai_family == AF_INET6
 344                    ? sizeof (struct sockaddr_in6)
 345                    : sizeof (struct sockaddr)) == -1)
 346         {
 347           int n;
 348           struct timeval tv = { pwm->socket_timeout, 0 };
 349           fd_set wfds;
 350
 351           rc = gpg_error_from_syserror ();
 352           if (gpg_err_code (rc) != GPG_ERR_EINPROGRESS)
 353             {
 354               close (pwm->fd);
 355               pwm->fd = -1;
 356               if (pwm->tcp->addr == pwm->tcp->addrs->ai_next)
 357                 return rc;
 358               continue;
 359             }
 360
 361           FD_ZERO (&wfds);
 362           FD_SET (pwm->fd, &wfds);
 363           n = select (pwm->fd+1, NULL, &wfds, NULL, &tv);
 364           rc = 0;
 365           if (!n)
 366             rc = gpg_error (GPG_ERR_ETIMEDOUT);
 367           else if (n != -1)
 368             {
 369               socklen_t len = sizeof(int);
 370
 371               getsockopt (pwm->fd, SOL_SOCKET, SO_ERROR, &n, &len);
 372               if (n)
 373                 rc = gpg_error_from_errno (n);
 374             }
 375           else if (n == -1)
 376             rc = gpg_error_from_syserror ();
 377
 378           if (rc)
 379             {
 380               close (pwm->fd);
 381               pwm->fd = -1;
 382               if (pwm->tcp->addr == pwm->tcp->addrs->ai_next
 383                   || gpg_err_code (rc) == GPG_ERR_ETIMEDOUT)
 384                 return rc;
 385             }
 386           else
 387             break;
 388         }
 389       else
 390         break;
 391     }
 392
 393   if (!rc && pwm->socket_timeout)
 394     if (fcntl (pwm->fd, F_SETFL, 0) == -1)
 395       rc = gpg_error_from_syserror ();
 396
 397   return rc;
 398 }
 399 #endif
 400
 401 gpg_error_t
 402 pwmd_connect (pwm_t * pwm, const char *url, ...)
 403 {
 404   const char *p = url;
 405   gpg_error_t rc;
 406
 407   if (!pwm)
 408     return FINISH (GPG_ERR_INV_ARG);
 409   else if (!pwm->ctx)
 410     {
 411       rc = init_handle (pwm);
 412       if (rc)
 413         return rc;
 414     }
 415
 416   if (!(pwm->opts & OPT_SIGPIPE))
 417     signal (SIGPIPE, SIG_IGN);
 418
 419 #ifdef WITH_GNUTLS
 420   pwm->tls_error = 0;
 421 #endif
 422   rc = GPG_ERR_UNSUPPORTED_PROTOCOL;
 423
 424   if (p && (*p == '/' || *p == '~'))
 425     rc = connect_uds (pwm, p);
 426   else if (!p || !strncmp (p, "file://", 7))
 427     {
 428       if (p)
 429         p += 7;
 430 #ifdef DEFAULT_PWMD_SOCKET
 431       else
 432         p = DEFAULT_PWMD_SOCKET;
 433 #endif
 434       rc = connect_uds (pwm, p);
 435     }
 436   else if (!strncmp (p, "ssh://", 6) || !strncmp (p, "ssh6://", 7) ||
 437            !strncmp (p, "ssh4://", 7))
 438     {
 439 #ifndef WITH_SSH
 440       return FINISH (GPG_ERR_NOT_IMPLEMENTED);
 441 #else
 442       char *host = NULL;
 443       int port;
 444       char *username = NULL;
 445
 446       if (!strncmp (p, "ssh6://", 7))
 447         {
 448           pwm->prot = PWMD_IPV6;
 449           p += 7;
 450         }
 451       else if (!strncmp (p, "ssh4://", 7))
 452         {
 453           pwm->prot = PWMD_IPV4;
 454           p += 7;
 455         }
 456       else
 457         {
 458           pwm->prot = PWMD_IP_ANY;
 459           p += 6;
 460         }
 461
 462       rc = _parse_ssh_url (p, &host, &port, &username);
 463       if (!rc)
 464         {
 465           va_list ap;
 466           char *identity = NULL;
 467           char *knownhosts = NULL;
 468
 469           va_start (ap, url);
 470           identity = va_arg (ap, char *);
 471
 472           if (!identity && !pwm->use_agent)
 473             rc = GPG_ERR_INV_ARG;
 474           else
 475             knownhosts = va_arg (ap, char *);
 476
 477           va_end (ap);
 478
 479           if (!rc)
 480             rc = _do_ssh_connect (pwm, host, port, identity, username,
 481                                   knownhosts);
 482           if (!rc)
 483             {
 484               rc = _connect_finalize (pwm);
 485               if (rc)
 486                 free_tcp (pwm);
 487             }
 488         }
 489
 490       pwmd_free (host);
 491       pwmd_free (username);
 492       pwm->local_pinentry = 1;
 493       return FINISH (rc);
 494 #endif
 495     }
 496   else if (!strncmp (p, "tls://", 6) || !strncmp (p, "tls6://", 7) ||
 497            !strncmp (p, "tls4://", 7))
 498     {
 499 #ifndef WITH_GNUTLS
 500       return FINISH (GPG_ERR_NOT_IMPLEMENTED);
 501 #else
 502       char *host = NULL;
 503       int port;
 504
 505       if (!strncmp (p, "tls6://", 7))
 506         {
 507           pwm->prot = PWMD_IPV6;
 508           p += 7;
 509         }
 510       else if (!strncmp (p, "tls4://", 7))
 511         {
 512           pwm->prot = PWMD_IPV4;
 513           p += 7;
 514         }
 515       else
 516         {
 517           pwm->prot = PWMD_IP_ANY;
 518           p += 6;
 519         }
 520
 521       rc = _parse_tls_url (p, &host, &port);
 522       if (!rc)
 523         {
 524           va_list ap;
 525           char *clientcert = NULL;
 526           char *clientkey = NULL;
 527           char *cacert = NULL;
 528           char *prio = NULL;
 529           char *server_fp = NULL;
 530
 531           va_start (ap, url);
 532           clientcert = va_arg (ap, char *);
 533
 534           if (!clientcert)
 535             rc = GPG_ERR_INV_ARG;
 536           else
 537             {
 538               clientkey = va_arg (ap, char *);
 539               if (!clientkey)
 540                 rc = GPG_ERR_INV_ARG;
 541               else
 542                 {
 543                   cacert = va_arg (ap, char *);
 544                   if (!cacert)
 545                     rc = GPG_ERR_INV_ARG;
 546                   else
 547                     {
 548                       prio = va_arg (ap, char *);
 549                       server_fp = va_arg (ap, char *);
 550                     }
 551                 }
 552             }
 553
 554           va_end (ap);
 555
 556           if (!rc)
 557             rc = _do_tls_connect (pwm, host, port, clientcert, clientkey,
 558                                   cacert, prio, server_fp, pwm->tls_verify);
 559           if (!rc)
 560             {
 561               rc = _connect_finalize (pwm);
 562               if (rc)
 563                 free_tcp (pwm);
 564             }
 565         }
 566
 567       pwmd_free (host);
 568       pwm->local_pinentry = 1;
 569       return FINISH (rc);
 570 #endif
 571     }
 572
 573   return FINISH (rc);
 574 }
 575
 576 static void
 577 disconnect (pwm_t * pwm)
 578 {
 579   if (!pwm)
 580     return;
 581
 582   if (pwm->ctx)
 583     assuan_release (pwm->ctx);
 584
 585 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
 586   free_tcp (pwm);
 587 #endif
 588   pwm->ctx = NULL;
 589   pwm->fd = -1;
 590 }
 591
 592 void
 593 pwmd_close (pwm_t * pwm)
 594 {
 595   if (!pwm)
 596     return;
 597
 598   disconnect (pwm);
 599   pwmd_free (pwm->pinentry_error);
 600   pwmd_free (pwm->pinentry_desc);
 601   pwmd_free (pwm->pinentry_prompt);
 602   pwmd_free (pwm->pinentry_tty);
 603   pwmd_free (pwm->pinentry_display);
 604   pwmd_free (pwm->pinentry_term);
 605   pwmd_free (pwm->pinentry_lcctype);
 606   pwmd_free (pwm->pinentry_lcmessages);
 607   pwmd_free (pwm->filename);
 608   pwmd_free (pwm->name);
 609   pwmd_free (pwm->passphrase_info);
 610   pwmd_free (pwm->passphrase_hint);
 611
 612 #ifdef WITH_PINENTRY
 613   if (pwm->pctx)
 614     _pinentry_disconnect (pwm);
 615 #endif
 616
 617   pwmd_free (pwm);
 618 }
 619
 620 static gpg_error_t
 621 inquire_realloc_cb (void *data, const void *buffer, size_t len)
 622 {
 623   membuf_t *mem = (membuf_t *) data;
 624   void *p;
 625
 626   if (!buffer)
 627     return 0;
 628
 629   if ((p = pwmd_realloc (mem->buf, mem->len + len)) == NULL)
 630     return gpg_error (GPG_ERR_ENOMEM);
 631
 632   mem->buf = p;
 633   memcpy ((char *) mem->buf + mem->len, buffer, len);
 634   mem->len += len;
 635   return 0;
 636 }
 637
 638 static gpg_error_t
 639 get_password (pwm_t * pwm, char **result, size_t * len,
 640               pwmd_pinentry_t w, int echo)
 641 {
 642   char buf[LINE_MAX] = { 0 }, *p;
 643   struct termios told, tnew;
 644   char *key = NULL;
 645
 646   if (result)
 647     *result = NULL;
 648
 649   if (len)
 650     *len = 0;
 651
 652   if (!isatty (STDIN_FILENO))
 653     {
 654       fprintf (stderr, N_("Input is not from a terminal! Failing.\n"));
 655       return GPG_ERR_ENOTTY;
 656     }
 657
 658   if (!echo)
 659     {
 660       if (tcgetattr (STDIN_FILENO, &told) == -1)
 661         return gpg_error_from_syserror ();
 662
 663       memcpy (&tnew, &told, sizeof (struct termios));
 664       tnew.c_lflag &= ~(ECHO);
 665       tnew.c_lflag |= ICANON | ECHONL;
 666
 667       if (tcsetattr (STDIN_FILENO, TCSANOW, &tnew) == -1)
 668         {
 669           int n = errno;
 670
 671           tcsetattr (STDIN_FILENO, TCSANOW, &told);
 672           return gpg_error_from_errno (n);
 673         }
 674     }
 675
 676   switch (w)
 677     {
 678     case PWMD_PINENTRY_OPEN:
 679       fprintf (stderr, N_("Password for %s: "), pwm->filename);
 680       break;
 681     case PWMD_PINENTRY_OPEN_FAILED:
 682       fprintf (stderr, N_("Invalid password. Password for %s: "),
 683                pwm->filename);
 684       break;
 685     case PWMD_PINENTRY_SAVE:
 686       fprintf (stderr, N_("New password for %s: "), pwm->filename);
 687       break;
 688     case PWMD_PINENTRY_SAVE_CONFIRM:
 689       fprintf (stderr, N_("Confirm password: "));
 690       break;
 691     default:
 692       break;
 693     }
 694
 695   if ((p = fgets (buf, sizeof (buf), stdin)) == NULL)
 696     {
 697       if (!echo)
 698         tcsetattr (STDIN_FILENO, TCSANOW, &told);
 699       return 0;
 700     }
 701
 702   if (!echo)
 703     tcsetattr (STDIN_FILENO, TCSANOW, &told);
 704
 705   if (feof (stdin))
 706     {
 707       clearerr (stdin);
 708       return GPG_ERR_CANCELED;
 709     }
 710
 711   /* Strip the newline character. */
 712   p[strlen (p) - 1] = 0;
 713
 714   if (buf[0])
 715     {
 716       key = pwmd_strdup_printf ("%s", p);
 717       memset (buf, 0, sizeof (buf));
 718       if (!key)
 719         return GPG_ERR_ENOMEM;
 720
 721       if (result)
 722         *result = key;
 723
 724       if (len)
 725         *len = strlen (key);
 726     }
 727   else
 728     {
 729       if (result)
 730         *result = pwmd_strdup ("");
 731
 732       if (len)
 733         *len = 1;
 734     }
 735
 736   return 0;
 737 }
 738
 739 gpg_error_t
 740 pwmd_password (pwm_t * pwm, const char *keyword, char **data, size_t * size)
 741 {
 742   gpg_error_t rc;
 743   int new_password = 0;
 744   char *password = NULL, *newpass = NULL;
 745   int error = 0;
 746
 747   if (data)
 748     *data = NULL;
 749
 750   if (size)
 751     *size = 0;
 752
 753   if (!strcmp (keyword, "NEW_PASSPHRASE"))
 754     new_password = 1;
 755
 756   if (!new_password && pwm->pinentry_try)
 757     error = 1;
 758
 759 again:
 760   if (pwm->disable_pinentry)
 761     {
 762       rc = get_password (pwm, &password, size,
 763                          new_password ? PWMD_PINENTRY_SAVE :
 764                          error ? PWMD_PINENTRY_OPEN_FAILED :
 765                          PWMD_PINENTRY_OPEN, 0);
 766       if (!rc && new_password)
 767         rc = get_password (pwm, &newpass, size, PWMD_PINENTRY_SAVE_CONFIRM,
 768                            0);
 769     }
 770   else
 771     {
 772       pwmd_pinentry_t which;
 773
 774       if (error)
 775         which = new_password
 776           ? PWMD_PINENTRY_SAVE_FAILED : PWMD_PINENTRY_OPEN_FAILED;
 777       else
 778         which = new_password ? PWMD_PINENTRY_SAVE : PWMD_PINENTRY_OPEN;
 779
 780       rc = pwmd_getpin (pwm, pwm->filename, &password, size, which);
 781       if (!rc && new_password)
 782         rc = pwmd_getpin (pwm, pwm->filename, &newpass, size,
 783                           PWMD_PINENTRY_SAVE_CONFIRM);
 784     }
 785
 786   if (!rc && new_password)
 787     {
 788       if ((!password && newpass) || (!newpass && password)
 789           || (newpass && password && strcmp (newpass, password)))
 790         {
 791           if (pwm->disable_pinentry)
 792             fprintf (stderr, N_("Passphrases do not match.\n"));
 793
 794           pwmd_free (password);
 795           pwmd_free (newpass);
 796           password = newpass = NULL;
 797           error = 1;
 798           goto again;
 799         }
 800     }
 801
 802   (void) pwmd_getpin (pwm, pwm->filename, NULL, NULL, PWMD_PINENTRY_CLOSE);
 803   pwmd_free (newpass);
 804   if (!rc && data)
 805     *data = password;
 806
 807   return rc;
 808 }
 809
 810 static gpg_error_t
 811 inquire_cb (void *data, const char *keyword)
 812 {
 813   pwm_t *pwm = (pwm_t *) data;
 814   gpg_error_t rc = 0;
 815   int free_result = 0;
 816   char *result = NULL;
 817   int is_password = 0;
 818   int new_password = 0;
 819
 820   if (!strcmp (keyword, "PASSPHRASE"))
 821     is_password = 1;
 822   else if (!strcmp (keyword, "NEW_PASSPHRASE"))
 823     new_password = 1;
 824
 825   /* Shouldn't get this far without a callback. */
 826   if (!pwm->override_inquire && !pwm->inquire_func
 827       && !is_password && !new_password)
 828     return gpg_error (GPG_ERR_ASS_NO_INQUIRE_CB);
 829
 830   for (;;)
 831     {
 832       size_t len = 0;
 833       gpg_error_t arc;
 834
 835       result = NULL;
 836
 837       if (!pwm->override_inquire && (is_password || new_password))
 838         {
 839           free_result = 1;
 840           rc = pwmd_password (data, keyword, &result, &len);
 841           if (!rc)
 842             rc = GPG_ERR_EOF;
 843         }
 844       else
 845         rc = pwm->inquire_func (pwm->inquire_data, keyword, rc, &result,
 846                                 &len);
 847
 848       /* gpg will truncate a passphrase at the first nil byte which may be bad
 849        * for generated key files. */
 850       if ((!rc || gpg_err_code (rc) == GPG_ERR_EOF)
 851           && (is_password || new_password))
 852         {
 853           if (len && result && *result && strlen (result) != len)
 854             rc = GPG_ERR_INV_PASSPHRASE;
 855         }
 856
 857     cancel:
 858       if (rc && gpg_err_code (rc) != GPG_ERR_EOF)
 859         {
 860 #ifndef LIBASSUAN_2_1_0
 861           gpg_error_t trc = rc;
 862
 863           /* Cancel this inquire. */
 864           rc = assuan_send_data (pwm->ctx, NULL, 1);
 865           if (!rc)
 866             {
 867               char *line;
 868               size_t len;
 869
 870               /* There is a bug (or feature?) in assuan_send_data() that
 871                * when cancelling an inquire the next read from the server is
 872                * not done until the next command making the next command
 873                * fail with GPG_ERR_ASS_UNEXPECTED_CMD.
 874                */
 875               rc = assuan_read_line (pwm->ctx, &line, &len);
 876
 877               /* Restore the original error. This differs from the error
 878                * returned from the pwmd command (GPG_ERR_CANCELED). This
 879                * error is returned to the calling function.
 880                */
 881               if (!rc)
 882                 rc = trc;
 883             }
 884 #endif
 885           break;
 886         }
 887
 888       if (gpg_err_code (rc) == GPG_ERR_EOF || !rc)
 889         {
 890           if (len <= 0 && !result)
 891             {
 892               rc = 0;
 893               break;
 894             }
 895           else if ((len <= 0 && result) || (len && !result))
 896             {
 897               rc = gpg_error (GPG_ERR_INV_ARG);
 898               break;
 899             }
 900
 901           if (pwm->inquire_maxlen
 902               && pwm->inquire_sent + len > pwm->inquire_maxlen)
 903             {
 904               rc = gpg_error (GPG_ERR_TOO_LARGE);
 905               if (!free_result)
 906                 rc = pwm->inquire_func (pwm->inquire_data, keyword, rc,
 907                                         &result, &len);
 908               goto cancel;
 909             }
 910
 911           arc = assuan_send_data (pwm->ctx, result, len);
 912           if (gpg_err_code (rc) == GPG_ERR_EOF)
 913             {
 914               rc = arc;
 915               break;
 916             }
 917
 918           rc = arc;
 919         }
 920       else if (rc)
 921         break;
 922
 923       if (!rc)
 924         {
 925           pwm->inquire_sent += len;
 926
 927           if (pwm->status_func)
 928             {
 929               char buf[ASSUAN_LINELENGTH];
 930
 931               snprintf (buf, sizeof (buf), "XFER %lu %lu", pwm->inquire_sent,
 932                         pwm->inquire_total);
 933               rc = pwm->status_func (pwm->status_data, buf);
 934               if (rc)
 935                 continue;
 936             }
 937         }
 938     }
 939
 940   if (free_result)
 941     pwmd_free (result);
 942
 943   return rc;
 944 }
 945
 946 static gpg_error_t
 947 parse_assuan_line (pwm_t * pwm)
 948 {
 949   gpg_error_t rc;
 950   char *line;
 951   size_t len;
 952
 953   rc = assuan_read_line (pwm->ctx, &line, &len);
 954   if (!rc)
 955     {
 956       if (line[0] == 'O' && line[1] == 'K' &&
 957           (line[2] == 0 || line[2] == ' '))
 958         {
 959         }
 960       else if (line[0] == '#')
 961         {
 962         }
 963       else if (line[0] == 'S' && (line[1] == 0 || line[1] == ' '))
 964         {
 965           if (pwm->status_func)
 966             {
 967               rc = pwm->status_func (pwm->status_data,
 968                                      line[1] == 0 ? line + 1 : line + 2);
 969             }
 970         }
 971       else if (line[0] == 'E' && line[1] == 'R' && line[2] == 'R' &&
 972                (line[3] == 0 || line[3] == ' '))
 973         {
 974           line += 4;
 975           rc = strtol (line, NULL, 10);
 976         }
 977     }
 978
 979   return rc;
 980 }
 981
 982 static void
 983 reset_handle (pwm_t *pwm)
 984 {
 985   pwm->fd = -1;
 986   pwm->pinentry_disabled = 0;
 987 #ifdef WITH_PINENTRY
 988   if (pwm->pctx)
 989     _pinentry_disconnect (pwm);
 990 #endif
 991 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
 992 #ifdef WITH_GNUTLS
 993   pwm->tls_error = 0;
 994 #endif
 995
 996   if (pwm->tcp)
 997     pwm->tcp->rc = 0;
 998 #endif
 999 }
1000
1001 gpg_error_t
1002 pwmd_disconnect (pwm_t * pwm)
1003 {
1004   if (!pwm)
1005     return FINISH (GPG_ERR_INV_ARG);
1006
1007   if (pwm->fd == -1)
1008     return FINISH (GPG_ERR_INV_STATE);
1009
1010   disconnect (pwm);
1011   reset_handle (pwm);
1012   return 0;
1013 }
1014
1015 /* Note that this should only be called when not in a command. */
1016 gpg_error_t
1017 pwmd_process (pwm_t * pwm)
1018 {
1019   gpg_error_t rc = 0;
1020   fd_set fds;
1021   struct timeval tv = { 0, 0 };
1022   int n;
1023
1024   if (!pwm || pwm->fd == -1)
1025     return FINISH (GPG_ERR_INV_ARG);
1026   else if (!pwm->ctx)
1027     return FINISH (GPG_ERR_INV_STATE);
1028
1029   FD_ZERO (&fds);
1030   FD_SET (pwm->fd, &fds);
1031   n = select (pwm->fd + 1, &fds, NULL, NULL, &tv);
1032
1033   if (n == -1)
1034     return FINISH (gpg_error_from_syserror ());
1035
1036   if (n > 0)
1037     {
1038       if (FD_ISSET (pwm->fd, &fds))
1039         rc = parse_assuan_line (pwm);
1040     }
1041
1042   while (!rc && assuan_pending_line (pwm->ctx))
1043     rc = parse_assuan_line (pwm);
1044
1045 #if defined (WITH_SSH) || defined (WITH_GNUTLS)
1046   if (gpg_err_code (rc) == GPG_ERR_EOF && pwm->tcp)
1047     {
1048       close (pwm->fd);
1049       pwm->fd = -1;
1050     }
1051 #endif
1052
1053   return FINISH (rc);
1054 }
1055
1056 static gpg_error_t
1057 status_cb (void *data, const char *line)
1058 {
1059   pwm_t *pwm = data;
1060
1061   if (!strncmp (line, "INQUIRE_MAXLEN ", 15))
1062     pwm->inquire_maxlen = strtol (line + 15, NULL, 10);
1063   else if (!strncmp (line, "PASSPHRASE_HINT ", 16))
1064     {
1065       pwmd_free (pwm->passphrase_hint);
1066       pwm->passphrase_hint = pwmd_strdup (line+16);
1067     }
1068   else if (!strncmp (line, "PASSPHRASE_INFO ", 16))
1069     {
1070       pwmd_free (pwm->passphrase_info);
1071       pwm->passphrase_info = pwmd_strdup (line+16);
1072     }
1073
1074   if (pwm->status_func)
1075     return pwm->status_func (pwm->status_data, line);
1076
1077   return 0;
1078 }
1079
1080 gpg_error_t
1081 _assuan_command (pwm_t * pwm, assuan_context_t ctx,
1082                  char **result, size_t * len, const char *cmd)
1083 {
1084   membuf_t data;
1085   gpg_error_t rc;
1086
1087   if (!cmd || !*cmd)
1088     return FINISH (GPG_ERR_INV_ARG);
1089
1090   if (strlen (cmd) >= ASSUAN_LINELENGTH + 1)
1091     return FINISH (GPG_ERR_LINE_TOO_LONG);
1092
1093   data.len = 0;
1094   data.buf = NULL;
1095   rc = assuan_transact (ctx, cmd, inquire_realloc_cb, &data,
1096 #ifdef WITH_QUALITY
1097                         pwm->pctx == ctx ? pwm->_inquire_func : inquire_cb,
1098                         pwm->pctx == ctx ? pwm->_inquire_data : pwm,
1099 #else
1100                         inquire_cb, pwm,
1101 #endif
1102                         status_cb, pwm);
1103
1104   if (rc)
1105     {
1106       if (data.buf)
1107         {
1108           pwmd_free (data.buf);
1109           data.buf = NULL;
1110         }
1111     }
1112   else
1113     {
1114       if (data.buf)
1115         {
1116           inquire_realloc_cb (&data, "", 1);
1117
1118           if (result)
1119             *result = (char *) data.buf;
1120           else
1121             pwmd_free (data.buf);
1122
1123           if (len)
1124             *len = data.len;
1125         }
1126     }
1127
1128   pwm->inquire_maxlen = 0;
1129   return rc;
1130 }
1131
1132 gpg_error_t
1133 pwmd_command_ap (pwm_t * pwm, char **result, size_t * rlen,
1134                  pwmd_inquire_cb_t func, void *user, const char *cmd,
1135                  va_list ap)
1136 {
1137   char *buf;
1138   size_t len;
1139   va_list ap2;
1140
1141   if (result)
1142     *result = NULL;
1143
1144   if (rlen)
1145     *rlen = 0;
1146
1147   if (!pwm || !cmd)
1148     return FINISH (GPG_ERR_INV_ARG);
1149   if (!pwm->ctx)
1150     return FINISH (GPG_ERR_INV_STATE);
1151
1152   /*
1153    * C99 allows the dst pointer to be null which will calculate the length
1154    * of the would-be result and return it.
1155    */
1156   va_copy (ap2, ap);
1157   len = vsnprintf (NULL, 0, cmd, ap) + 1;
1158   buf = (char *) pwmd_malloc (len);
1159   if (!buf)
1160     {
1161       va_end (ap2);
1162       return FINISH (GPG_ERR_ENOMEM);
1163     }
1164
1165   len = vsnprintf (buf, len, cmd, ap2);
1166   va_end (ap2);
1167
1168   if (buf[strlen (buf) - 1] == '\n')
1169     buf[strlen (buf) - 1] = 0;
1170   if (buf[strlen (buf) - 1] == '\r')
1171     buf[strlen (buf) - 1] = 0;
1172
1173   pwm->inquire_func = func;
1174   pwm->inquire_data = user;
1175   pwm->inquire_sent = 0;
1176   gpg_error_t rc = _assuan_command (pwm, pwm->ctx, result, rlen, buf);
1177   pwmd_free (buf);
1178   return rc;
1179 }
1180
1181 gpg_error_t
1182 pwmd_command (pwm_t * pwm, char **result, size_t * len,
1183               pwmd_inquire_cb_t func, void *user, const char *cmd, ...)
1184 {
1185   va_list ap;
1186
1187   if (result)
1188     *result = NULL;
1189
1190   if (len)
1191     *len = 0;
1192
1193   if (!pwm || !cmd)
1194     return FINISH (GPG_ERR_INV_ARG);
1195   if (!pwm->ctx)
1196     return FINISH (GPG_ERR_INV_STATE);
1197
1198   va_start (ap, cmd);
1199   gpg_error_t rc = pwmd_command_ap (pwm, result, len, func, user, cmd, ap);
1200   va_end (ap);
1201   return rc;
1202 }
1203
1204 static gpg_error_t
1205 send_pinentry_options (pwm_t * pwm)
1206 {
1207   gpg_error_t rc;
1208
1209   // Pwmd >= 3.1 uses gpgme to do the pinentry settings.
1210   if (pwm->version && pwm->version >= 0x030100)
1211     return 0;
1212
1213   rc = pwmd_command (pwm, NULL, NULL, NULL, NULL,
1214                      "OPTION disable-pinentry=0");
1215   if (!rc && pwm->pinentry_tty)
1216     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION TTYNAME=%s",
1217                        pwm->pinentry_tty);
1218
1219   if (!rc && pwm->pinentry_term)
1220     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION TTYTYPE=%s",
1221                        pwm->pinentry_term);
1222
1223   if (!rc && pwm->pinentry_display)
1224     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION DISPLAY=%s",
1225                        pwm->pinentry_display);
1226
1227   if (!rc && pwm->pinentry_desc)
1228     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION DESC=%s",
1229                        pwm->pinentry_desc);
1230
1231   if (!rc && pwm->pinentry_lcctype)
1232     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION LC_CTYPE=%s",
1233                        pwm->pinentry_lcctype);
1234
1235   if (!rc && pwm->pinentry_lcmessages)
1236     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION LC_MESSAGES=%s",
1237                        pwm->pinentry_lcmessages);
1238
1239   if (!rc && pwm->pinentry_timeout >= 0)
1240     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL,
1241                        "OPTION pinentry-timeout=%i", pwm->pinentry_timeout);
1242
1243   return rc;
1244 }
1245
1246 gpg_error_t
1247 pwmd_socket_type (pwm_t * pwm, pwmd_socket_t * result)
1248 {
1249   if (!pwm || !result)
1250     return FINISH (GPG_ERR_INV_ARG);
1251
1252   *result = PWMD_SOCKET_LOCAL;
1253
1254   if (pwm->fd == -1)
1255     return FINISH (GPG_ERR_INV_STATE);
1256
1257 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1258 #ifdef WITH_SSH
1259   if (pwm->tcp && pwm->tcp->ssh)
1260     *result = PWMD_SOCKET_SSH;
1261 #endif
1262 #ifdef WITH_GNUTLS
1263   if (pwm->tcp && pwm->tcp->tls)
1264     *result = PWMD_SOCKET_TLS;
1265 #endif
1266 #endif
1267   return 0;
1268 }
1269
1270 static gpg_error_t
1271 disable_pinentry (pwm_t *pwm, int *disable)
1272 {
1273   gpg_error_t rc;
1274 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1275   int no_pinentry = pwm->disable_pinentry || pwm->tcp || pwm->local_pinentry;
1276 #else
1277   int no_pinentry = pwm->disable_pinentry || pwm->local_pinentry;
1278 #endif
1279
1280   if (disable)
1281     *disable = no_pinentry;
1282
1283   if (pwm->pinentry_disabled && no_pinentry)
1284     return 0;
1285   else if (!pwm->pinentry_disabled && !no_pinentry)
1286     return 0;
1287
1288   rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION disable-pinentry=%i",
1289                      no_pinentry);
1290   if (!rc)
1291     pwm->pinentry_disabled = no_pinentry;
1292
1293   return rc;
1294 }
1295
1296 gpg_error_t
1297 pwmd_open (pwm_t * pwm, const char *filename, pwmd_inquire_cb_t cb,
1298            void *data)
1299 {
1300   gpg_error_t rc = 0;
1301   int no_pinentry = 0;
1302
1303   if (!pwm || !filename || !*filename)
1304     return FINISH (GPG_ERR_INV_ARG);
1305
1306   if (!pwm->ctx)
1307     return FINISH (GPG_ERR_INV_STATE);
1308
1309   rc = disable_pinentry (pwm, &no_pinentry);
1310   if (!rc && !no_pinentry)
1311     rc = send_pinentry_options (pwm);
1312
1313   if (!rc)
1314     {
1315       pwm->pinentry_try = 0;
1316       pwmd_free (pwm->filename);
1317       pwm->filename = pwmd_strdup (filename);
1318
1319       do
1320         {
1321           rc = pwmd_command (pwm, NULL, NULL, cb, data, "OPEN %s%s",
1322                              (pwm->opts & OPT_LOCK_ON_OPEN) ? "--lock " : "",
1323                              filename);
1324         }
1325       while (gpg_err_code (rc) == GPG_ERR_BAD_PASSPHRASE
1326              && pwm->pinentry_disabled
1327              && ++pwm->pinentry_try < pwm->pinentry_tries);
1328
1329       pwm->pinentry_try = 0;
1330
1331       if (rc)
1332         {
1333           pwmd_free (pwm->filename);
1334           pwm->filename = NULL;
1335         }
1336     }
1337
1338   return FINISH (rc);
1339 }
1340
1341 static gpg_error_t
1342 do_pwmd_save_passwd (pwm_t * pwm, const char *args, pwmd_inquire_cb_t cb,
1343                      void *data, int save)
1344 {
1345   gpg_error_t rc = 0;
1346
1347   if (!pwm)
1348     return FINISH (GPG_ERR_INV_ARG);
1349   if (!pwm->ctx)
1350     return FINISH (GPG_ERR_INV_STATE);
1351
1352   rc = disable_pinentry (pwm, NULL);
1353   if (!rc)
1354     rc = pwmd_command (pwm, NULL, NULL, cb, data,
1355                        save ? "SAVE %s" : "PASSWD %s", args ? args : "");
1356
1357   return FINISH (rc);
1358 }
1359
1360 gpg_error_t
1361 pwmd_passwd (pwm_t * pwm, const char *args, pwmd_inquire_cb_t cb, void *data)
1362 {
1363   return do_pwmd_save_passwd (pwm, args, cb, data, 0);
1364 }
1365
1366 gpg_error_t
1367 pwmd_save (pwm_t * pwm, const char *args, pwmd_inquire_cb_t cb, void *data)
1368 {
1369   return do_pwmd_save_passwd (pwm, args, cb, data, 1);
1370 }
1371
1372 static gpg_error_t
1373 pwmd_get_set_opt (pwm_t *pwm, pwmd_option_t opt, int get, va_list ap)
1374 {
1375   int n, *intp;
1376   size_t *sizetp;
1377   char *arg1, **charpp;
1378   gpg_error_t rc = 0;
1379
1380   if (!pwm)
1381     return GPG_ERR_INV_ARG;
1382
1383   switch (opt)
1384     {
1385     case PWMD_OPTION_SERVER_VERSION:
1386       if (!get)
1387         return GPG_ERR_NOT_SUPPORTED;
1388
1389       if (!pwm)
1390         rc = GPG_ERR_INV_ARG;
1391       else if (!pwm->ctx)
1392         rc = GPG_ERR_INV_STATE;
1393       else
1394         {
1395           unsigned *u = va_arg (ap, unsigned *);
1396           *u = pwm->version;
1397         }
1398
1399       break;
1400     case PWMD_OPTION_SIGPIPE:
1401       if (get)
1402         {
1403           intp = va_arg (ap, int *);
1404           *intp = pwm->opts & OPT_SIGPIPE ? 1 : 0;
1405           break;
1406         }
1407
1408       n = va_arg (ap, int);
1409       if (n < 0 || n > 1)
1410         rc = GPG_ERR_INV_VALUE;
1411
1412       if (n)
1413         pwm->opts |= OPT_SIGPIPE;
1414       else
1415         pwm->opts &= ~OPT_SIGPIPE;
1416
1417       break;
1418     case PWMD_OPTION_LOCK_ON_OPEN:
1419       if (get)
1420         {
1421           intp = va_arg (ap, int *);
1422           *intp = pwm->opts & OPT_LOCK_ON_OPEN ? 1 : 0;
1423           break;
1424         }
1425
1426       n = va_arg (ap, int);
1427
1428       if (n < 0 || n > 1)
1429         rc = GPG_ERR_INV_VALUE;
1430
1431       if (n)
1432         pwm->opts |= OPT_LOCK_ON_OPEN;
1433       else
1434         pwm->opts &= ~OPT_LOCK_ON_OPEN;
1435
1436       break;
1437     case PWMD_OPTION_INQUIRE_TOTAL:
1438       if (get)
1439         {
1440           sizetp = va_arg (ap, size_t *);
1441           *sizetp = pwm->inquire_total;
1442           break;
1443         }
1444
1445       pwm->inquire_total = va_arg (ap, size_t);
1446       break;
1447     case PWMD_OPTION_STATUS_CB:
1448       if (get)
1449         {
1450           pwmd_status_cb_t *cb = va_arg (ap, pwmd_status_cb_t *);
1451
1452           *cb = pwm->status_func;
1453           break;
1454         }
1455
1456       pwm->status_func = va_arg (ap, pwmd_status_cb_t);
1457       break;
1458     case PWMD_OPTION_STATUS_DATA:
1459       if (get)
1460         {
1461           void **data = va_arg (ap, void **);
1462
1463           *data = pwm->status_data;
1464           break;
1465         }
1466
1467       pwm->status_data = va_arg (ap, void *);
1468       break;
1469     case PWMD_OPTION_NO_PINENTRY:
1470       if (get)
1471         {
1472           intp = va_arg (ap, int *);
1473           *intp = pwm->disable_pinentry;
1474           break;
1475         }
1476
1477       n = va_arg (ap, int);
1478
1479       if (n < 0 || n > 1)
1480         rc = GPG_ERR_INV_VALUE;
1481       else
1482         pwm->disable_pinentry = n;
1483
1484       if (!rc && pwm->ctx)
1485         rc = disable_pinentry (pwm, NULL);
1486       break;
1487     case PWMD_OPTION_LOCAL_PINENTRY:
1488       if (get)
1489         {
1490           intp = va_arg (ap, int *);
1491           *intp = pwm->local_pinentry;
1492           break;
1493         }
1494
1495       n = va_arg (ap, int);
1496
1497       if (n < 0 || n > 1)
1498         rc = GPG_ERR_INV_VALUE;
1499       else
1500         pwm->local_pinentry = n;
1501
1502       break;
1503     case PWMD_OPTION_PINENTRY_TIMEOUT:
1504       if (get)
1505         {
1506           intp = va_arg (ap, int *);
1507           *intp = pwm->pinentry_timeout;
1508           break;
1509         }
1510
1511       n = va_arg (ap, int);
1512
1513       if (n < 0)
1514         rc = GPG_ERR_INV_VALUE;
1515       else
1516         pwm->pinentry_timeout = n;
1517
1518 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1519       if (!rc && pwm->fd != -1 && !pwm->tcp)
1520 #else
1521       if (!rc && pwm->fd != -1)
1522 #endif
1523         rc = pwmd_command (pwm, NULL, NULL, NULL, NULL,
1524                            "OPTION pinentry-timeout=%i", pwm->pinentry_timeout);
1525       break;
1526     case PWMD_OPTION_PINENTRY_TRIES:
1527       if (get)
1528         {
1529           intp = va_arg (ap, int *);
1530           *intp = pwm->pinentry_tries;
1531           break;
1532         }
1533
1534       n = va_arg (ap, int);
1535       pwm->pinentry_tries = n;
1536       break;
1537     case PWMD_OPTION_PINENTRY_PATH:
1538       if (get)
1539         {
1540           charpp = va_arg (ap, char **);
1541           *charpp = pwm->pinentry_path;
1542           break;
1543         }
1544
1545       arg1 = va_arg (ap, char *);
1546       pwmd_free (pwm->pinentry_path);
1547       pwm->pinentry_path = arg1 ? _expand_homedir (arg1, NULL) : NULL;
1548       break;
1549     case PWMD_OPTION_PINENTRY_TTY:
1550       if (get)
1551         {
1552           charpp = va_arg (ap, char **);
1553           *charpp = pwm->pinentry_tty;
1554           break;
1555         }
1556
1557       arg1 = va_arg (ap, char *);
1558       pwmd_free (pwm->pinentry_tty);
1559       pwm->pinentry_tty = arg1 ? pwmd_strdup (arg1) : NULL;
1560       break;
1561     case PWMD_OPTION_PINENTRY_DISPLAY:
1562       if (get)
1563         {
1564           charpp = va_arg (ap, char **);
1565           *charpp = pwm->pinentry_display;
1566           break;
1567         }
1568
1569       arg1 = va_arg (ap, char *);
1570       pwmd_free (pwm->pinentry_display);
1571       pwm->pinentry_display = arg1 ? pwmd_strdup (arg1) : NULL;
1572       break;
1573     case PWMD_OPTION_PINENTRY_TERM:
1574       if (get)
1575         {
1576           charpp = va_arg (ap, char **);
1577           *charpp = pwm->pinentry_term;
1578           break;
1579         }
1580
1581       arg1 = va_arg (ap, char *);
1582       pwmd_free (pwm->pinentry_term);
1583       pwm->pinentry_term = arg1 ? pwmd_strdup (arg1) : NULL;
1584       break;
1585     case PWMD_OPTION_PINENTRY_ERROR:
1586       if (get)
1587         {
1588           charpp = va_arg (ap, char **);
1589           *charpp = pwm->pinentry_error;
1590           break;
1591         }
1592
1593       arg1 = va_arg (ap, char *);
1594       pwmd_free (pwm->pinentry_error);
1595       pwm->pinentry_error = arg1 ? _percent_escape (arg1) : NULL;
1596       break;
1597     case PWMD_OPTION_PINENTRY_PROMPT:
1598       if (get)
1599         {
1600           charpp = va_arg (ap, char **);
1601           *charpp = pwm->pinentry_prompt;
1602           break;
1603         }
1604
1605       arg1 = va_arg (ap, char *);
1606       pwmd_free (pwm->pinentry_prompt);
1607       pwm->pinentry_prompt = arg1 ? _percent_escape (arg1) : NULL;
1608       break;
1609     case PWMD_OPTION_PINENTRY_DESC:
1610       if (get)
1611         {
1612           charpp = va_arg (ap, char **);
1613           *charpp = pwm->pinentry_desc;
1614           break;
1615         }
1616
1617       arg1 = va_arg (ap, char *);
1618       pwmd_free (pwm->pinentry_desc);
1619       pwm->pinentry_desc = arg1 ? _percent_escape (arg1) : NULL;
1620       break;
1621     case PWMD_OPTION_PINENTRY_LC_CTYPE:
1622       if (get)
1623         {
1624           charpp = va_arg (ap, char **);
1625           *charpp = pwm->pinentry_lcctype;
1626           break;
1627         }
1628
1629       arg1 = va_arg (ap, char *);
1630       pwmd_free (pwm->pinentry_lcctype);
1631       pwm->pinentry_lcctype = arg1 ? pwmd_strdup (arg1) : NULL;
1632       break;
1633     case PWMD_OPTION_PINENTRY_LC_MESSAGES:
1634       if (get)
1635         {
1636           charpp = va_arg (ap, char **);
1637           *charpp = pwm->pinentry_lcmessages;
1638           break;
1639         }
1640
1641       arg1 = va_arg (ap, char *);
1642       pwmd_free (pwm->pinentry_lcmessages);
1643       pwm->pinentry_lcmessages = arg1 ? pwmd_strdup (arg1) : NULL;
1644       break;
1645     case PWMD_OPTION_KNOWNHOST_CB:
1646       if (get)
1647         {
1648           pwmd_knownhost_cb_t *cb = va_arg (ap, pwmd_knownhost_cb_t *);
1649
1650           *cb = pwm->kh_cb;
1651           break;
1652         }
1653
1654       pwm->kh_cb = va_arg (ap, pwmd_knownhost_cb_t);
1655       break;
1656     case PWMD_OPTION_KNOWNHOST_DATA:
1657       if (get)
1658         {
1659           void **data = va_arg (ap, void **);
1660
1661           *data = pwm->kh_data;
1662           break;
1663         }
1664
1665       pwm->kh_data = va_arg (ap, void *);
1666       break;
1667     case PWMD_OPTION_SSH_AGENT:
1668       if (get)
1669         {
1670           intp = va_arg (ap, int *);
1671           *intp = pwm->use_agent;
1672           break;
1673         }
1674
1675       pwm->use_agent = va_arg (ap, int);
1676
1677       if (pwm->use_agent < 0 || pwm->use_agent > 1)
1678         {
1679           pwm->use_agent = 0;
1680           rc = GPG_ERR_INV_VALUE;
1681         }
1682       break;
1683     case PWMD_OPTION_TLS_VERIFY:
1684       if (get)
1685         {
1686           intp = va_arg (ap, int *);
1687           *intp = pwm->tls_verify;
1688           break;
1689         }
1690
1691       pwm->tls_verify = va_arg (ap, int);
1692
1693       if (pwm->tls_verify < 0 || pwm->tls_verify > 1)
1694         {
1695           pwm->tls_verify = 0;
1696           rc = GPG_ERR_INV_VALUE;
1697         }
1698       break;
1699     case PWMD_OPTION_SOCKET_TIMEOUT:
1700       if (get)
1701         {
1702           intp = va_arg (ap, int *);
1703           *intp = pwm->socket_timeout;
1704           break;
1705         }
1706
1707       pwm->socket_timeout = va_arg (ap, int);
1708       if (pwm->socket_timeout < 0)
1709         {
1710           pwm->socket_timeout = 0;
1711           rc = GPG_ERR_INV_VALUE;
1712         }
1713
1714 #ifdef WITH_SSH
1715       if (pwm->tcp && pwm->tcp->ssh && pwm->tcp->ssh->session)
1716         {
1717           pwm->tcp->ssh->timeout = pwm->socket_timeout;
1718           libssh2_session_set_timeout (pwm->tcp->ssh->session,
1719                                        pwm->socket_timeout * 1000);
1720         }
1721 #endif
1722 #ifdef WITH_GNUTLS
1723       if (pwm->tcp && pwm->tcp->tls && pwm->tcp->tls->session)
1724         pwm->tcp->tls->timeout = pwm->socket_timeout;
1725 #endif
1726       break;
1727     case PWMD_OPTION_OVERRIDE_INQUIRE:
1728       if (get)
1729         {
1730           intp = va_arg (ap, int *);
1731           *intp = pwm->override_inquire;
1732           break;
1733         }
1734
1735       pwm->override_inquire = va_arg (ap, int);
1736
1737       if (pwm->override_inquire < 0 || pwm->override_inquire > 1)
1738         {
1739           pwm->override_inquire = 0;
1740           rc = GPG_ERR_INV_VALUE;
1741         }
1742       break;
1743     default:
1744       rc = GPG_ERR_UNKNOWN_OPTION;
1745       break;
1746     }
1747
1748   return FINISH (rc);
1749 }
1750
1751 gpg_error_t
1752 pwmd_setopt (pwm_t * pwm, pwmd_option_t opt, ...)
1753 {
1754   va_list ap;
1755   gpg_error_t rc = 0;
1756
1757   va_start (ap, opt);
1758   rc = pwmd_get_set_opt (pwm, opt, 0, ap);
1759   va_end (ap);
1760   return FINISH (rc);
1761 }
1762
1763 gpg_error_t
1764 pwmd_getopt (pwm_t *pwm, pwmd_option_t opt, ...)
1765 {
1766   va_list ap;
1767   gpg_error_t rc = 0;
1768
1769   va_start (ap, opt);
1770   rc = pwmd_get_set_opt (pwm, opt, 1, ap);
1771   va_end (ap);
1772   return FINISH (rc);
1773 }
1774
1775 gpg_error_t
1776 pwmd_new (const char *name, pwm_t ** pwm)
1777 {
1778   pwm_t *h = pwmd_calloc (1, sizeof (pwm_t));
1779   gpg_error_t rc;
1780
1781   if (!h)
1782     return FINISH (GPG_ERR_ENOMEM);
1783
1784   if (name)
1785     {
1786       h->name = pwmd_strdup (name);
1787       if (!h->name)
1788         {
1789           pwmd_free (h);
1790           return FINISH (GPG_ERR_ENOMEM);
1791         }
1792     }
1793
1794   reset_handle (h);
1795   h->pinentry_timeout = -1;
1796   h->pinentry_tries = 3;
1797 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1798   h->prot = PWMD_IP_ANY;
1799 #endif
1800
1801   if (ttyname (STDOUT_FILENO))
1802     {
1803       char buf[256];
1804
1805       ttyname_r (STDOUT_FILENO, buf, sizeof (buf));
1806       h->pinentry_tty = pwmd_strdup (buf);
1807       if (!h->pinentry_tty)
1808         {
1809           rc = GPG_ERR_ENOMEM;
1810           goto fail;
1811         }
1812     }
1813
1814   if (getenv ("TERM") && h->pinentry_tty)
1815     {
1816       h->pinentry_term = pwmd_strdup (getenv ("TERM"));
1817       if (!h->pinentry_term)
1818         {
1819           rc = GPG_ERR_ENOMEM;
1820           goto fail;
1821         }
1822     }
1823
1824   if (getenv ("DISPLAY"))
1825     {
1826       h->pinentry_display = pwmd_strdup (getenv ("DISPLAY"));
1827       if (!h->pinentry_display)
1828         {
1829           rc = GPG_ERR_ENOMEM;
1830           goto fail;
1831         }
1832     }
1833
1834   update_pinentry_settings (h);
1835   *pwm = h;
1836   return 0;
1837
1838 fail:
1839   pwmd_close (h);
1840   return FINISH (rc);
1841 }
1842
1843 void
1844 pwmd_free (void *ptr)
1845 {
1846   _xfree (ptr);
1847 }
1848
1849 void *
1850 pwmd_malloc (size_t size)
1851 {
1852   return _xmalloc (size);
1853 }
1854
1855 void *
1856 pwmd_calloc (size_t nmemb, size_t size)
1857 {
1858   return _xcalloc (nmemb, size);
1859 }
1860
1861 void *
1862 pwmd_realloc (void *ptr, size_t size)
1863 {
1864   return _xrealloc (ptr, size);
1865 }
1866
1867 char *
1868 pwmd_strdup (const char *str)
1869 {
1870   char *t;
1871   size_t len;
1872   register size_t c;
1873
1874   len = strlen (str);
1875   t = _xmalloc ((len + 1) * sizeof (char));
1876   if (!t)
1877     return NULL;
1878
1879   for (c = 0; c < len; c++)
1880     t[c] = str[c];
1881
1882   t[c] = 0;
1883   return t;
1884 }
1885
1886 char *
1887 pwmd_strdup_printf (const char *fmt, ...)
1888 {
1889   va_list ap, ap2;
1890   int len;
1891   char *buf;
1892
1893   if (!fmt)
1894     return NULL;
1895
1896   va_start (ap, fmt);
1897   va_copy (ap2, ap);
1898   len = vsnprintf (NULL, 0, fmt, ap);
1899   va_end (ap);
1900   buf = pwmd_malloc (++len);
1901   if (buf)
1902     vsnprintf (buf, len, fmt, ap2);
1903
1904   va_end (ap2);
1905   return buf;
1906 }
1907
1908 gpg_error_t
1909 pwmd_getpin (pwm_t * pwm, const char *filename, char **result,
1910              size_t * len, pwmd_pinentry_t which)
1911 {
1912 #ifndef WITH_PINENTRY
1913   return FINISH (GPG_ERR_NOT_IMPLEMENTED);
1914 #else
1915   gpg_error_t rc = _pwmd_getpin (pwm, filename, result, len, which);
1916
1917   return FINISH (rc);
1918 #endif
1919 }
1920
1921 const char *
1922 pwmd_version ()
1923 {
1924   return LIBPWMD_VERSION_STR;
1925 }
1926
1927 unsigned int
1928 pwmd_features ()
1929 {
1930   unsigned int n = 0;
1931
1932 #ifdef WITH_PINENTRY
1933   n |= PWMD_FEATURE_PINENTRY;
1934 #endif
1935 #ifdef WITH_SSH
1936   n |= PWMD_FEATURE_SSH;
1937 #endif
1938 #ifdef WITH_QUALITY
1939   n |= PWMD_FEATURE_CRACK;
1940 #endif
1941 #ifdef WITH_GNUTLS
1942   n |= PWMD_FEATURE_GNUTLS;
1943 #endif
1944   return n;
1945 }
1946
1947 gpg_error_t
1948 pwmd_fd (pwm_t * pwm, int *fd)
1949 {
1950   if (!pwm || !fd)
1951     return FINISH (GPG_ERR_INV_ARG);
1952
1953   if (pwm->fd == -1)
1954     return FINISH (GPG_ERR_INV_STATE);
1955
1956   *fd = pwm->fd;
1957   return 0;
1958 }
1959
1960 void
1961 pwmd_set_pointer (pwm_t *pwm, void *data)
1962 {
1963   pwm->user_data = data;
1964 }
1965
1966 void *
1967 pwmd_get_pointer (pwm_t *pwm)
1968 {
1969   return pwm->user_data;
1970 }
1971
1972 int
1973 pwmd_tls_error (pwm_t *pwm)
1974 {
1975 #ifndef WITH_GNUTLS
1976   return 0;
1977 #else
1978   return pwm ? pwm->tls_error : 0;
1979 #endif
1980 }