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 == '/')
 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       tcsetattr (STDIN_FILENO, TCSANOW, &told);
 698       return 0;
 699     }
 700
 701   if (!echo)
 702     tcsetattr (STDIN_FILENO, TCSANOW, &told);
 703
 704   if (feof (stdin))
 705     {
 706       clearerr (stdin);
 707       return GPG_ERR_CANCELED;
 708     }
 709
 710   /* Strip the newline character. */
 711   p[strlen (p) - 1] = 0;
 712
 713   if (buf[0])
 714     {
 715       key = pwmd_strdup_printf ("%s", p);
 716       memset (buf, 0, sizeof (buf));
 717       if (!key)
 718         return GPG_ERR_ENOMEM;
 719
 720       if (result)
 721         *result = key;
 722
 723       if (len)
 724         *len = strlen (key);
 725     }
 726   else
 727     {
 728       if (result)
 729         *result = pwmd_strdup ("");
 730
 731       if (len)
 732         *len = 1;
 733     }
 734
 735   return 0;
 736 }
 737
 738 gpg_error_t
 739 pwmd_password (pwm_t * pwm, const char *keyword, char **data, size_t * size)
 740 {
 741   gpg_error_t rc;
 742   int new_password = 0;
 743   char *password = NULL, *newpass = NULL;
 744   int error = 0;
 745
 746   if (data)
 747     *data = NULL;
 748
 749   if (size)
 750     *size = 0;
 751
 752   if (!strcmp (keyword, "NEW_PASSPHRASE"))
 753     new_password = 1;
 754
 755   if (!new_password && pwm->pinentry_try)
 756     error = 1;
 757
 758 again:
 759   if (pwm->disable_pinentry)
 760     {
 761       rc = get_password (pwm, &password, size,
 762                          new_password ? PWMD_PINENTRY_SAVE :
 763                          error ? PWMD_PINENTRY_OPEN_FAILED :
 764                          PWMD_PINENTRY_OPEN, 0);
 765       if (!rc && new_password)
 766         rc = get_password (pwm, &newpass, size, PWMD_PINENTRY_SAVE_CONFIRM,
 767                            0);
 768     }
 769   else
 770     {
 771       pwmd_pinentry_t which;
 772
 773       if (error)
 774         which = new_password
 775           ? PWMD_PINENTRY_SAVE_FAILED : PWMD_PINENTRY_OPEN_FAILED;
 776       else
 777         which = new_password ? PWMD_PINENTRY_SAVE : PWMD_PINENTRY_OPEN;
 778
 779       rc = pwmd_getpin (pwm, pwm->filename, &password, size, which);
 780       if (!rc && new_password)
 781         rc = pwmd_getpin (pwm, pwm->filename, &newpass, size,
 782                           PWMD_PINENTRY_SAVE_CONFIRM);
 783     }
 784
 785   if (!rc && new_password)
 786     {
 787       if ((!password && newpass) || (!newpass && password)
 788           || (newpass && password && strcmp (newpass, password)))
 789         {
 790           if (pwm->disable_pinentry)
 791             fprintf (stderr, N_("Passphrases do not match.\n"));
 792
 793           pwmd_free (password);
 794           pwmd_free (newpass);
 795           password = newpass = NULL;
 796           error = 1;
 797           goto again;
 798         }
 799     }
 800
 801   (void) pwmd_getpin (pwm, pwm->filename, NULL, NULL, PWMD_PINENTRY_CLOSE);
 802   pwmd_free (newpass);
 803   if (!rc && data)
 804     *data = password;
 805
 806   return rc;
 807 }
 808
 809 static gpg_error_t
 810 inquire_cb (void *data, const char *keyword)
 811 {
 812   pwm_t *pwm = (pwm_t *) data;
 813   gpg_error_t rc = 0;
 814   int free_result = 0;
 815   char *result = NULL;
 816   int is_password = 0;
 817   int new_password = 0;
 818
 819   if (!strcmp (keyword, "PASSPHRASE"))
 820     is_password = 1;
 821   else if (!strcmp (keyword, "NEW_PASSPHRASE"))
 822     new_password = 1;
 823
 824   /* Shouldn't get this far without a callback. */
 825   if (!pwm->override_inquire && !pwm->inquire_func
 826       && !is_password && !new_password)
 827     return gpg_error (GPG_ERR_ASS_NO_INQUIRE_CB);
 828
 829   for (;;)
 830     {
 831       size_t len = 0;
 832       gpg_error_t arc;
 833
 834       result = NULL;
 835
 836       if (!pwm->override_inquire && (is_password || new_password))
 837         {
 838           free_result = 1;
 839           rc = pwmd_password (data, keyword, &result, &len);
 840           if (!rc)
 841             rc = GPG_ERR_EOF;
 842         }
 843       else
 844         rc = pwm->inquire_func (pwm->inquire_data, keyword, rc, &result,
 845                                 &len);
 846
 847     cancel:
 848       if (rc && gpg_err_code (rc) != GPG_ERR_EOF)
 849         {
 850 #ifndef LIBASSUAN_2_1_0
 851           gpg_error_t trc = rc;
 852
 853           /* Cancel this inquire. */
 854           rc = assuan_send_data (pwm->ctx, NULL, 1);
 855           if (!rc)
 856             {
 857               char *line;
 858               size_t len;
 859
 860               /* There is a bug (or feature?) in assuan_send_data() that
 861                * when cancelling an inquire the next read from the server is
 862                * not done until the next command making the next command
 863                * fail with GPG_ERR_ASS_UNEXPECTED_CMD.
 864                */
 865               rc = assuan_read_line (pwm->ctx, &line, &len);
 866
 867               /* Restore the original error. This differs from the error
 868                * returned from the pwmd command (GPG_ERR_CANCELED). This
 869                * error is returned to the calling function.
 870                */
 871               if (!rc)
 872                 rc = trc;
 873             }
 874 #endif
 875           break;
 876         }
 877
 878       if (gpg_err_code (rc) == GPG_ERR_EOF || !rc)
 879         {
 880           if (len <= 0 && !result)
 881             {
 882               rc = 0;
 883               break;
 884             }
 885           else if ((len <= 0 && result) || (len && !result))
 886             {
 887               rc = gpg_error (GPG_ERR_INV_ARG);
 888               break;
 889             }
 890
 891           if (pwm->inquire_maxlen
 892               && pwm->inquire_sent + len > pwm->inquire_maxlen)
 893             {
 894               rc = gpg_error (GPG_ERR_TOO_LARGE);
 895               if (!free_result)
 896                 rc = pwm->inquire_func (pwm->inquire_data, keyword, rc,
 897                                         &result, &len);
 898               goto cancel;
 899             }
 900
 901           arc = assuan_send_data (pwm->ctx, result, len);
 902           if (gpg_err_code (rc) == GPG_ERR_EOF)
 903             {
 904               rc = arc;
 905               break;
 906             }
 907
 908           rc = arc;
 909         }
 910       else if (rc)
 911         break;
 912
 913       if (!rc)
 914         {
 915           pwm->inquire_sent += len;
 916
 917           if (pwm->status_func)
 918             {
 919               char buf[ASSUAN_LINELENGTH];
 920
 921               snprintf (buf, sizeof (buf), "XFER %lu %lu", pwm->inquire_sent,
 922                         pwm->inquire_total);
 923               rc = pwm->status_func (pwm->status_data, buf);
 924               if (rc)
 925                 continue;
 926             }
 927         }
 928     }
 929
 930   if (free_result)
 931     pwmd_free (result);
 932
 933   return rc;
 934 }
 935
 936 static gpg_error_t
 937 parse_assuan_line (pwm_t * pwm)
 938 {
 939   gpg_error_t rc;
 940   char *line;
 941   size_t len;
 942
 943   rc = assuan_read_line (pwm->ctx, &line, &len);
 944   if (!rc)
 945     {
 946       if (line[0] == 'O' && line[1] == 'K' &&
 947           (line[2] == 0 || line[2] == ' '))
 948         {
 949         }
 950       else if (line[0] == '#')
 951         {
 952         }
 953       else if (line[0] == 'S' && (line[1] == 0 || line[1] == ' '))
 954         {
 955           if (pwm->status_func)
 956             {
 957               rc = pwm->status_func (pwm->status_data,
 958                                      line[1] == 0 ? line + 1 : line + 2);
 959             }
 960         }
 961       else if (line[0] == 'E' && line[1] == 'R' && line[2] == 'R' &&
 962                (line[3] == 0 || line[3] == ' '))
 963         {
 964           line += 4;
 965           rc = strtol (line, NULL, 10);
 966         }
 967     }
 968
 969   return rc;
 970 }
 971
 972 static void
 973 reset_handle (pwm_t *pwm)
 974 {
 975   pwm->fd = -1;
 976   pwm->pinentry_disabled = 0;
 977 #ifdef WITH_PINENTRY
 978   if (pwm->pctx)
 979     _pinentry_disconnect (pwm);
 980 #endif
 981 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
 982 #ifdef WITH_GNUTLS
 983   pwm->tls_error = 0;
 984 #endif
 985
 986   if (pwm->tcp)
 987     pwm->tcp->rc = 0;
 988 #endif
 989 }
 990
 991 gpg_error_t
 992 pwmd_disconnect (pwm_t * pwm)
 993 {
 994   if (!pwm)
 995     return FINISH (GPG_ERR_INV_ARG);
 996
 997   if (pwm->fd == -1)
 998     return FINISH (GPG_ERR_INV_STATE);
 999
1000   disconnect (pwm);
1001   reset_handle (pwm);
1002   return 0;
1003 }
1004
1005 /* Note that this should only be called when not in a command. */
1006 gpg_error_t
1007 pwmd_process (pwm_t * pwm)
1008 {
1009   gpg_error_t rc = 0;
1010   fd_set fds;
1011   struct timeval tv = { 0, 0 };
1012   int n;
1013
1014   if (!pwm || pwm->fd == -1)
1015     return FINISH (GPG_ERR_INV_ARG);
1016   else if (!pwm->ctx)
1017     return FINISH (GPG_ERR_INV_STATE);
1018
1019   FD_ZERO (&fds);
1020   FD_SET (pwm->fd, &fds);
1021   n = select (pwm->fd + 1, &fds, NULL, NULL, &tv);
1022
1023   if (n == -1)
1024     return FINISH (gpg_error_from_syserror ());
1025
1026   if (n > 0)
1027     {
1028       if (FD_ISSET (pwm->fd, &fds))
1029         rc = parse_assuan_line (pwm);
1030     }
1031
1032   while (!rc && assuan_pending_line (pwm->ctx))
1033     rc = parse_assuan_line (pwm);
1034
1035 #if defined (WITH_SSH) || defined (WITH_GNUTLS)
1036   if (gpg_err_code (rc) == GPG_ERR_EOF && pwm->tcp)
1037     {
1038       close (pwm->fd);
1039       pwm->fd = -1;
1040     }
1041 #endif
1042
1043   return FINISH (rc);
1044 }
1045
1046 static gpg_error_t
1047 status_cb (void *data, const char *line)
1048 {
1049   pwm_t *pwm = data;
1050
1051   if (!strncmp (line, "INQUIRE_MAXLEN ", 15))
1052     pwm->inquire_maxlen = strtol (line + 15, NULL, 10);
1053   else if (!strncmp (line, "PASSPHRASE_HINT ", 16))
1054     {
1055       pwmd_free (pwm->passphrase_hint);
1056       pwm->passphrase_hint = pwmd_strdup (line+16);
1057     }
1058   else if (!strncmp (line, "PASSPHRASE_INFO ", 16))
1059     {
1060       pwmd_free (pwm->passphrase_info);
1061       pwm->passphrase_info = pwmd_strdup (line+16);
1062     }
1063
1064   if (pwm->status_func)
1065     return pwm->status_func (pwm->status_data, line);
1066
1067   return 0;
1068 }
1069
1070 gpg_error_t
1071 _assuan_command (pwm_t * pwm, assuan_context_t ctx,
1072                  char **result, size_t * len, const char *cmd)
1073 {
1074   membuf_t data;
1075   gpg_error_t rc;
1076
1077   if (!cmd || !*cmd)
1078     return FINISH (GPG_ERR_INV_ARG);
1079
1080   if (strlen (cmd) >= ASSUAN_LINELENGTH + 1)
1081     return FINISH (GPG_ERR_LINE_TOO_LONG);
1082
1083   data.len = 0;
1084   data.buf = NULL;
1085   rc = assuan_transact (ctx, cmd, inquire_realloc_cb, &data,
1086 #ifdef WITH_QUALITY
1087                         pwm->pctx == ctx ? pwm->_inquire_func : inquire_cb,
1088                         pwm->pctx == ctx ? pwm->_inquire_data : pwm,
1089 #else
1090                         inquire_cb, pwm,
1091 #endif
1092                         status_cb, pwm);
1093
1094   if (rc)
1095     {
1096       if (data.buf)
1097         {
1098           pwmd_free (data.buf);
1099           data.buf = NULL;
1100         }
1101     }
1102   else
1103     {
1104       if (data.buf)
1105         {
1106           inquire_realloc_cb (&data, "", 1);
1107
1108           if (result)
1109             *result = (char *) data.buf;
1110           else
1111             pwmd_free (data.buf);
1112
1113           if (len)
1114             *len = data.len;
1115         }
1116     }
1117
1118   pwm->inquire_maxlen = 0;
1119   return rc;
1120 }
1121
1122 gpg_error_t
1123 pwmd_command_ap (pwm_t * pwm, char **result, size_t * rlen,
1124                  pwmd_inquire_cb_t func, void *user, const char *cmd,
1125                  va_list ap)
1126 {
1127   char *buf;
1128   size_t len;
1129   va_list ap2;
1130
1131   if (result)
1132     *result = NULL;
1133
1134   if (rlen)
1135     *rlen = 0;
1136
1137   if (!pwm || !cmd)
1138     return FINISH (GPG_ERR_INV_ARG);
1139   if (!pwm->ctx)
1140     return FINISH (GPG_ERR_INV_STATE);
1141
1142   /*
1143    * C99 allows the dst pointer to be null which will calculate the length
1144    * of the would-be result and return it.
1145    */
1146   va_copy (ap2, ap);
1147   len = vsnprintf (NULL, 0, cmd, ap) + 1;
1148   buf = (char *) pwmd_malloc (len);
1149   if (!buf)
1150     {
1151       va_end (ap2);
1152       return FINISH (GPG_ERR_ENOMEM);
1153     }
1154
1155   len = vsnprintf (buf, len, cmd, ap2);
1156   va_end (ap2);
1157
1158   if (buf[strlen (buf) - 1] == '\n')
1159     buf[strlen (buf) - 1] = 0;
1160   if (buf[strlen (buf) - 1] == '\r')
1161     buf[strlen (buf) - 1] = 0;
1162
1163   pwm->inquire_func = func;
1164   pwm->inquire_data = user;
1165   pwm->inquire_sent = 0;
1166   gpg_error_t rc = _assuan_command (pwm, pwm->ctx, result, rlen, buf);
1167   pwmd_free (buf);
1168   return rc;
1169 }
1170
1171 gpg_error_t
1172 pwmd_command (pwm_t * pwm, char **result, size_t * len,
1173               pwmd_inquire_cb_t func, void *user, const char *cmd, ...)
1174 {
1175   va_list ap;
1176
1177   if (result)
1178     *result = NULL;
1179
1180   if (len)
1181     *len = 0;
1182
1183   if (!pwm || !cmd)
1184     return FINISH (GPG_ERR_INV_ARG);
1185   if (!pwm->ctx)
1186     return FINISH (GPG_ERR_INV_STATE);
1187
1188   va_start (ap, cmd);
1189   gpg_error_t rc = pwmd_command_ap (pwm, result, len, func, user, cmd, ap);
1190   va_end (ap);
1191   return rc;
1192 }
1193
1194 static gpg_error_t
1195 send_pinentry_options (pwm_t * pwm)
1196 {
1197   gpg_error_t rc;
1198
1199   // Pwmd >= 3.1 uses gpgme to do the pinentry settings.
1200   if (pwm->version && pwm->version >= 0x030100)
1201     return 0;
1202
1203   rc = pwmd_command (pwm, NULL, NULL, NULL, NULL,
1204                      "OPTION disable-pinentry=0");
1205   if (!rc && pwm->pinentry_tty)
1206     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION TTYNAME=%s",
1207                        pwm->pinentry_tty);
1208
1209   if (!rc && pwm->pinentry_term)
1210     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION TTYTYPE=%s",
1211                        pwm->pinentry_term);
1212
1213   if (!rc && pwm->pinentry_display)
1214     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION DISPLAY=%s",
1215                        pwm->pinentry_display);
1216
1217   if (!rc && pwm->pinentry_desc)
1218     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION DESC=%s",
1219                        pwm->pinentry_desc);
1220
1221   if (!rc && pwm->pinentry_lcctype)
1222     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION LC_CTYPE=%s",
1223                        pwm->pinentry_lcctype);
1224
1225   if (!rc && pwm->pinentry_lcmessages)
1226     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION LC_MESSAGES=%s",
1227                        pwm->pinentry_lcmessages);
1228
1229   if (!rc && pwm->pinentry_timeout >= 0)
1230     rc = pwmd_command (pwm, NULL, NULL, NULL, NULL,
1231                        "OPTION pinentry-timeout=%i", pwm->pinentry_timeout);
1232
1233   return rc;
1234 }
1235
1236 gpg_error_t
1237 pwmd_socket_type (pwm_t * pwm, pwmd_socket_t * result)
1238 {
1239   if (!pwm || !result)
1240     return FINISH (GPG_ERR_INV_ARG);
1241
1242   *result = PWMD_SOCKET_LOCAL;
1243
1244   if (pwm->fd == -1)
1245     return FINISH (GPG_ERR_INV_STATE);
1246
1247 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1248 #ifdef WITH_SSH
1249   if (pwm->tcp && pwm->tcp->ssh)
1250     *result = PWMD_SOCKET_SSH;
1251 #endif
1252 #ifdef WITH_GNUTLS
1253   if (pwm->tcp && pwm->tcp->tls)
1254     *result = PWMD_SOCKET_TLS;
1255 #endif
1256 #endif
1257   return 0;
1258 }
1259
1260 static gpg_error_t
1261 disable_pinentry (pwm_t *pwm, int *disable)
1262 {
1263   gpg_error_t rc;
1264 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1265   int no_pinentry = pwm->disable_pinentry || pwm->tcp || pwm->local_pinentry;
1266 #else
1267   int no_pinentry = pwm->disable_pinentry || pwm->local_pinentry;
1268 #endif
1269
1270   if (disable)
1271     *disable = no_pinentry;
1272
1273   if (pwm->pinentry_disabled && no_pinentry)
1274     return 0;
1275   else if (!pwm->pinentry_disabled && !no_pinentry)
1276     return 0;
1277
1278   rc = pwmd_command (pwm, NULL, NULL, NULL, NULL, "OPTION disable-pinentry=%i",
1279                      no_pinentry);
1280   if (!rc)
1281     pwm->pinentry_disabled = no_pinentry;
1282
1283   return rc;
1284 }
1285
1286 gpg_error_t
1287 pwmd_open (pwm_t * pwm, const char *filename, pwmd_inquire_cb_t cb,
1288            void *data)
1289 {
1290   gpg_error_t rc = 0;
1291   int no_pinentry = 0;
1292
1293   if (!pwm || !filename || !*filename)
1294     return FINISH (GPG_ERR_INV_ARG);
1295
1296   if (!pwm->ctx)
1297     return FINISH (GPG_ERR_INV_STATE);
1298
1299   rc = disable_pinentry (pwm, &no_pinentry);
1300   if (!rc && !no_pinentry)
1301     rc = send_pinentry_options (pwm);
1302
1303   if (!rc)
1304     {
1305       pwm->pinentry_try = 0;
1306       pwmd_free (pwm->filename);
1307       pwm->filename = pwmd_strdup (filename);
1308
1309       do
1310         {
1311           rc = pwmd_command (pwm, NULL, NULL, cb, data, "OPEN %s%s",
1312                              (pwm->opts & OPT_LOCK_ON_OPEN) ? "--lock " : "",
1313                              filename);
1314         }
1315       while (gpg_err_code (rc) == GPG_ERR_BAD_PASSPHRASE
1316              && pwm->pinentry_disabled
1317              && ++pwm->pinentry_try < pwm->pinentry_tries);
1318
1319       pwm->pinentry_try = 0;
1320
1321       if (rc)
1322         {
1323           pwmd_free (pwm->filename);
1324           pwm->filename = NULL;
1325         }
1326     }
1327
1328   return FINISH (rc);
1329 }
1330
1331 static gpg_error_t
1332 do_pwmd_save_passwd (pwm_t * pwm, const char *args, pwmd_inquire_cb_t cb,
1333                      void *data, int save)
1334 {
1335   gpg_error_t rc = 0;
1336
1337   if (!pwm)
1338     return FINISH (GPG_ERR_INV_ARG);
1339   if (!pwm->ctx)
1340     return FINISH (GPG_ERR_INV_STATE);
1341
1342   rc = disable_pinentry (pwm, NULL);
1343   if (!rc)
1344     rc = pwmd_command (pwm, NULL, NULL, cb, data,
1345                        save ? "SAVE %s" : "PASSWD %s", args ? args : "");
1346
1347   return FINISH (rc);
1348 }
1349
1350 gpg_error_t
1351 pwmd_passwd (pwm_t * pwm, const char *args, pwmd_inquire_cb_t cb, void *data)
1352 {
1353   return do_pwmd_save_passwd (pwm, args, cb, data, 0);
1354 }
1355
1356 gpg_error_t
1357 pwmd_save (pwm_t * pwm, const char *args, pwmd_inquire_cb_t cb, void *data)
1358 {
1359   return do_pwmd_save_passwd (pwm, args, cb, data, 1);
1360 }
1361
1362 static gpg_error_t
1363 pwmd_get_set_opt (pwm_t *pwm, pwmd_option_t opt, int get, va_list ap)
1364 {
1365   int n, *intp;
1366   size_t *sizetp;
1367   char *arg1, **charpp;
1368   gpg_error_t rc = 0;
1369
1370   if (!pwm)
1371     return GPG_ERR_INV_ARG;
1372
1373   switch (opt)
1374     {
1375     case PWMD_OPTION_SIGPIPE:
1376       if (get)
1377         {
1378           intp = va_arg (ap, int *);
1379           *intp = pwm->opts & OPT_SIGPIPE ? 1 : 0;
1380           break;
1381         }
1382
1383       n = va_arg (ap, int);
1384       if (n < 0 || n > 1)
1385         rc = GPG_ERR_INV_VALUE;
1386
1387       if (n)
1388         pwm->opts |= OPT_SIGPIPE;
1389       else
1390         pwm->opts &= ~OPT_SIGPIPE;
1391
1392       break;
1393     case PWMD_OPTION_LOCK_ON_OPEN:
1394       if (get)
1395         {
1396           intp = va_arg (ap, int *);
1397           *intp = pwm->opts & OPT_LOCK_ON_OPEN ? 1 : 0;
1398           break;
1399         }
1400
1401       n = va_arg (ap, int);
1402
1403       if (n < 0 || n > 1)
1404         rc = GPG_ERR_INV_VALUE;
1405
1406       if (n)
1407         pwm->opts |= OPT_LOCK_ON_OPEN;
1408       else
1409         pwm->opts &= ~OPT_LOCK_ON_OPEN;
1410
1411       break;
1412     case PWMD_OPTION_INQUIRE_TOTAL:
1413       if (get)
1414         {
1415           sizetp = va_arg (ap, size_t *);
1416           *sizetp = pwm->inquire_total;
1417           break;
1418         }
1419
1420       pwm->inquire_total = va_arg (ap, size_t);
1421       break;
1422     case PWMD_OPTION_STATUS_CB:
1423       if (get)
1424         {
1425           pwmd_status_cb_t *cb = va_arg (ap, pwmd_status_cb_t *);
1426
1427           *cb = pwm->status_func;
1428           break;
1429         }
1430
1431       pwm->status_func = va_arg (ap, pwmd_status_cb_t);
1432       break;
1433     case PWMD_OPTION_STATUS_DATA:
1434       if (get)
1435         {
1436           void **data = va_arg (ap, void **);
1437
1438           *data = pwm->status_data;
1439           break;
1440         }
1441
1442       pwm->status_data = va_arg (ap, void *);
1443       break;
1444     case PWMD_OPTION_NO_PINENTRY:
1445       if (get)
1446         {
1447           intp = va_arg (ap, int *);
1448           *intp = pwm->disable_pinentry;
1449           break;
1450         }
1451
1452       n = va_arg (ap, int);
1453
1454       if (n < 0 || n > 1)
1455         rc = GPG_ERR_INV_VALUE;
1456       else
1457         pwm->disable_pinentry = n;
1458
1459       if (!rc && pwm->ctx)
1460         rc = disable_pinentry (pwm, NULL);
1461       break;
1462     case PWMD_OPTION_LOCAL_PINENTRY:
1463       if (get)
1464         {
1465           intp = va_arg (ap, int *);
1466           *intp = pwm->local_pinentry;
1467           break;
1468         }
1469
1470       n = va_arg (ap, int);
1471
1472       if (n < 0 || n > 1)
1473         rc = GPG_ERR_INV_VALUE;
1474       else
1475         pwm->local_pinentry = n;
1476
1477       break;
1478     case PWMD_OPTION_PINENTRY_TIMEOUT:
1479       if (get)
1480         {
1481           intp = va_arg (ap, int *);
1482           *intp = pwm->pinentry_timeout;
1483           break;
1484         }
1485
1486       n = va_arg (ap, int);
1487
1488       if (n < 0)
1489         rc = GPG_ERR_INV_VALUE;
1490       else
1491         pwm->pinentry_timeout = n;
1492
1493 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1494       if (!rc && pwm->fd != -1 && !pwm->tcp)
1495 #else
1496       if (!rc && pwm->fd != -1)
1497 #endif
1498         rc = pwmd_command (pwm, NULL, NULL, NULL, NULL,
1499                            "OPTION pinentry-timeout=%i", pwm->pinentry_timeout);
1500       break;
1501     case PWMD_OPTION_PINENTRY_TRIES:
1502       if (get)
1503         {
1504           intp = va_arg (ap, int *);
1505           *intp = pwm->pinentry_tries;
1506           break;
1507         }
1508
1509       n = va_arg (ap, int);
1510       pwm->pinentry_tries = n;
1511       break;
1512     case PWMD_OPTION_PINENTRY_PATH:
1513       if (get)
1514         {
1515           charpp = va_arg (ap, char **);
1516           *charpp = pwm->pinentry_path;
1517           break;
1518         }
1519
1520       arg1 = va_arg (ap, char *);
1521       pwmd_free (pwm->pinentry_path);
1522       pwm->pinentry_path = arg1 ? _expand_homedir (arg1, NULL) : NULL;
1523       break;
1524     case PWMD_OPTION_PINENTRY_TTY:
1525       if (get)
1526         {
1527           charpp = va_arg (ap, char **);
1528           *charpp = pwm->pinentry_tty;
1529           break;
1530         }
1531
1532       arg1 = va_arg (ap, char *);
1533       pwmd_free (pwm->pinentry_tty);
1534       pwm->pinentry_tty = arg1 ? pwmd_strdup (arg1) : NULL;
1535       break;
1536     case PWMD_OPTION_PINENTRY_DISPLAY:
1537       if (get)
1538         {
1539           charpp = va_arg (ap, char **);
1540           *charpp = pwm->pinentry_display;
1541           break;
1542         }
1543
1544       arg1 = va_arg (ap, char *);
1545       pwmd_free (pwm->pinentry_display);
1546       pwm->pinentry_display = arg1 ? pwmd_strdup (arg1) : NULL;
1547       break;
1548     case PWMD_OPTION_PINENTRY_TERM:
1549       if (get)
1550         {
1551           charpp = va_arg (ap, char **);
1552           *charpp = pwm->pinentry_term;
1553           break;
1554         }
1555
1556       arg1 = va_arg (ap, char *);
1557       pwmd_free (pwm->pinentry_term);
1558       pwm->pinentry_term = arg1 ? pwmd_strdup (arg1) : NULL;
1559       break;
1560     case PWMD_OPTION_PINENTRY_ERROR:
1561       if (get)
1562         {
1563           charpp = va_arg (ap, char **);
1564           *charpp = pwm->pinentry_error;
1565           break;
1566         }
1567
1568       arg1 = va_arg (ap, char *);
1569       pwmd_free (pwm->pinentry_error);
1570       pwm->pinentry_error = arg1 ? _percent_escape (arg1) : NULL;
1571       break;
1572     case PWMD_OPTION_PINENTRY_PROMPT:
1573       if (get)
1574         {
1575           charpp = va_arg (ap, char **);
1576           *charpp = pwm->pinentry_prompt;
1577           break;
1578         }
1579
1580       arg1 = va_arg (ap, char *);
1581       pwmd_free (pwm->pinentry_prompt);
1582       pwm->pinentry_prompt = arg1 ? _percent_escape (arg1) : NULL;
1583       break;
1584     case PWMD_OPTION_PINENTRY_DESC:
1585       if (get)
1586         {
1587           charpp = va_arg (ap, char **);
1588           *charpp = pwm->pinentry_desc;
1589           break;
1590         }
1591
1592       arg1 = va_arg (ap, char *);
1593       pwmd_free (pwm->pinentry_desc);
1594       pwm->pinentry_desc = arg1 ? _percent_escape (arg1) : NULL;
1595       break;
1596     case PWMD_OPTION_PINENTRY_LC_CTYPE:
1597       if (get)
1598         {
1599           charpp = va_arg (ap, char **);
1600           *charpp = pwm->pinentry_lcctype;
1601           break;
1602         }
1603
1604       arg1 = va_arg (ap, char *);
1605       pwmd_free (pwm->pinentry_lcctype);
1606       pwm->pinentry_lcctype = arg1 ? pwmd_strdup (arg1) : NULL;
1607       break;
1608     case PWMD_OPTION_PINENTRY_LC_MESSAGES:
1609       if (get)
1610         {
1611           charpp = va_arg (ap, char **);
1612           *charpp = pwm->pinentry_lcmessages;
1613           break;
1614         }
1615
1616       arg1 = va_arg (ap, char *);
1617       pwmd_free (pwm->pinentry_lcmessages);
1618       pwm->pinentry_lcmessages = arg1 ? pwmd_strdup (arg1) : NULL;
1619       break;
1620     case PWMD_OPTION_KNOWNHOST_CB:
1621       if (get)
1622         {
1623           pwmd_knownhost_cb_t *cb = va_arg (ap, pwmd_knownhost_cb_t *);
1624
1625           *cb = pwm->kh_cb;
1626           break;
1627         }
1628
1629       pwm->kh_cb = va_arg (ap, pwmd_knownhost_cb_t);
1630       break;
1631     case PWMD_OPTION_KNOWNHOST_DATA:
1632       if (get)
1633         {
1634           void **data = va_arg (ap, void **);
1635
1636           *data = pwm->kh_data;
1637           break;
1638         }
1639
1640       pwm->kh_data = va_arg (ap, void *);
1641       break;
1642     case PWMD_OPTION_SSH_AGENT:
1643       if (get)
1644         {
1645           intp = va_arg (ap, int *);
1646           *intp = pwm->use_agent;
1647           break;
1648         }
1649
1650       pwm->use_agent = va_arg (ap, int);
1651
1652       if (pwm->use_agent < 0 || pwm->use_agent > 1)
1653         {
1654           pwm->use_agent = 0;
1655           rc = GPG_ERR_INV_VALUE;
1656         }
1657       break;
1658     case PWMD_OPTION_TLS_VERIFY:
1659       if (get)
1660         {
1661           intp = va_arg (ap, int *);
1662           *intp = pwm->tls_verify;
1663           break;
1664         }
1665
1666       pwm->tls_verify = va_arg (ap, int);
1667
1668       if (pwm->tls_verify < 0 || pwm->tls_verify > 1)
1669         {
1670           pwm->tls_verify = 0;
1671           rc = GPG_ERR_INV_VALUE;
1672         }
1673       break;
1674     case PWMD_OPTION_SOCKET_TIMEOUT:
1675       if (get)
1676         {
1677           intp = va_arg (ap, int *);
1678           *intp = pwm->socket_timeout;
1679           break;
1680         }
1681
1682       pwm->socket_timeout = va_arg (ap, int);
1683       if (pwm->socket_timeout < 0)
1684         {
1685           pwm->socket_timeout = 0;
1686           rc = GPG_ERR_INV_VALUE;
1687         }
1688
1689 #ifdef WITH_SSH
1690       if (pwm->tcp && pwm->tcp->ssh && pwm->tcp->ssh->session)
1691         {
1692           pwm->tcp->ssh->timeout = pwm->socket_timeout;
1693           libssh2_session_set_timeout (pwm->tcp->ssh->session,
1694                                        pwm->socket_timeout * 1000);
1695         }
1696 #endif
1697 #ifdef WITH_GNUTLS
1698       if (pwm->tcp && pwm->tcp->tls && pwm->tcp->tls->session)
1699         pwm->tcp->tls->timeout = pwm->socket_timeout;
1700 #endif
1701       break;
1702     case PWMD_OPTION_OVERRIDE_INQUIRE:
1703       if (get)
1704         {
1705           intp = va_arg (ap, int *);
1706           *intp = pwm->override_inquire;
1707           break;
1708         }
1709
1710       pwm->override_inquire = va_arg (ap, int);
1711
1712       if (pwm->override_inquire < 0 || pwm->override_inquire > 1)
1713         {
1714           pwm->override_inquire = 0;
1715           rc = GPG_ERR_INV_VALUE;
1716         }
1717       break;
1718     default:
1719       rc = GPG_ERR_UNKNOWN_OPTION;
1720       break;
1721     }
1722
1723   return FINISH (rc);
1724 }
1725
1726 gpg_error_t
1727 pwmd_setopt (pwm_t * pwm, pwmd_option_t opt, ...)
1728 {
1729   va_list ap;
1730   gpg_error_t rc = 0;
1731
1732   va_start (ap, opt);
1733   rc = pwmd_get_set_opt (pwm, opt, 0, ap);
1734   va_end (ap);
1735   return FINISH (rc);
1736 }
1737
1738 gpg_error_t
1739 pwmd_getopt (pwm_t *pwm, pwmd_option_t opt, ...)
1740 {
1741   va_list ap;
1742   gpg_error_t rc = 0;
1743
1744   va_start (ap, opt);
1745   rc = pwmd_get_set_opt (pwm, opt, 1, ap);
1746   va_end (ap);
1747   return FINISH (rc);
1748 }
1749
1750 gpg_error_t
1751 pwmd_new (const char *name, pwm_t ** pwm)
1752 {
1753   pwm_t *h = pwmd_calloc (1, sizeof (pwm_t));
1754   gpg_error_t rc;
1755
1756   if (!h)
1757     return FINISH (GPG_ERR_ENOMEM);
1758
1759   if (name)
1760     {
1761       h->name = pwmd_strdup (name);
1762       if (!h->name)
1763         {
1764           pwmd_free (h);
1765           return FINISH (GPG_ERR_ENOMEM);
1766         }
1767     }
1768
1769   reset_handle (h);
1770   h->pinentry_timeout = -1;
1771   h->pinentry_tries = 3;
1772 #if defined(WITH_SSH) || defined(WITH_GNUTLS)
1773   h->prot = PWMD_IP_ANY;
1774 #endif
1775
1776   if (ttyname (STDOUT_FILENO))
1777     {
1778       char buf[256];
1779
1780       ttyname_r (STDOUT_FILENO, buf, sizeof (buf));
1781       h->pinentry_tty = pwmd_strdup (buf);
1782       if (!h->pinentry_tty)
1783         {
1784           rc = GPG_ERR_ENOMEM;
1785           goto fail;
1786         }
1787     }
1788
1789   if (getenv ("TERM") && h->pinentry_tty)
1790     {
1791       h->pinentry_term = pwmd_strdup (getenv ("TERM"));
1792       if (!h->pinentry_term)
1793         {
1794           rc = GPG_ERR_ENOMEM;
1795           goto fail;
1796         }
1797     }
1798
1799   if (getenv ("DISPLAY"))
1800     {
1801       h->pinentry_display = pwmd_strdup (getenv ("DISPLAY"));
1802       if (!h->pinentry_display)
1803         {
1804           rc = GPG_ERR_ENOMEM;
1805           goto fail;
1806         }
1807     }
1808
1809   update_pinentry_settings (h);
1810   *pwm = h;
1811   return 0;
1812
1813 fail:
1814   pwmd_close (h);
1815   return FINISH (rc);
1816 }
1817
1818 void
1819 pwmd_free (void *ptr)
1820 {
1821   _xfree (ptr);
1822 }
1823
1824 void *
1825 pwmd_malloc (size_t size)
1826 {
1827   return _xmalloc (size);
1828 }
1829
1830 void *
1831 pwmd_calloc (size_t nmemb, size_t size)
1832 {
1833   return _xcalloc (nmemb, size);
1834 }
1835
1836 void *
1837 pwmd_realloc (void *ptr, size_t size)
1838 {
1839   return _xrealloc (ptr, size);
1840 }
1841
1842 char *
1843 pwmd_strdup (const char *str)
1844 {
1845   char *t;
1846   size_t len;
1847   register size_t c;
1848
1849   len = strlen (str);
1850   t = _xmalloc ((len + 1) * sizeof (char));
1851   if (!t)
1852     return NULL;
1853
1854   for (c = 0; c < len; c++)
1855     t[c] = str[c];
1856
1857   t[c] = 0;
1858   return t;
1859 }
1860
1861 char *
1862 pwmd_strdup_printf (const char *fmt, ...)
1863 {
1864   va_list ap, ap2;
1865   int len;
1866   char *buf;
1867
1868   if (!fmt)
1869     return NULL;
1870
1871   va_start (ap, fmt);
1872   va_copy (ap2, ap);
1873   len = vsnprintf (NULL, 0, fmt, ap);
1874   va_end (ap);
1875   buf = pwmd_malloc (++len);
1876   if (buf)
1877     vsnprintf (buf, len, fmt, ap2);
1878
1879   va_end (ap2);
1880   return buf;
1881 }
1882
1883 gpg_error_t
1884 pwmd_getpin (pwm_t * pwm, const char *filename, char **result,
1885              size_t * len, pwmd_pinentry_t which)
1886 {
1887 #ifndef WITH_PINENTRY
1888   return FINISH (GPG_ERR_NOT_IMPLEMENTED);
1889 #else
1890   gpg_error_t rc = _pwmd_getpin (pwm, filename, result, len, which);
1891
1892   return FINISH (rc);
1893 #endif
1894 }
1895
1896 const char *
1897 pwmd_version ()
1898 {
1899   return LIBPWMD_VERSION_STR;
1900 }
1901
1902 unsigned int
1903 pwmd_features ()
1904 {
1905   unsigned int n = 0;
1906
1907 #ifdef WITH_PINENTRY
1908   n |= PWMD_FEATURE_PINENTRY;
1909 #endif
1910 #ifdef WITH_SSH
1911   n |= PWMD_FEATURE_SSH;
1912 #endif
1913 #ifdef WITH_QUALITY
1914   n |= PWMD_FEATURE_CRACK;
1915 #endif
1916 #ifdef WITH_GNUTLS
1917   n |= PWMD_FEATURE_GNUTLS;
1918 #endif
1919   return n;
1920 }
1921
1922 gpg_error_t
1923 pwmd_fd (pwm_t * pwm, int *fd)
1924 {
1925   if (!pwm || !fd)
1926     return FINISH (GPG_ERR_INV_ARG);
1927
1928   if (pwm->fd == -1)
1929     return FINISH (GPG_ERR_INV_STATE);
1930
1931   *fd = pwm->fd;
1932   return 0;
1933 }
1934
1935 void
1936 pwmd_set_pointer (pwm_t *pwm, void *data)
1937 {
1938   pwm->user_data = data;
1939 }
1940
1941 void *
1942 pwmd_get_pointer (pwm_t *pwm)
1943 {
1944   return pwm->user_data;
1945 }
1946
1947 int
1948 pwmd_tls_error (pwm_t *pwm)
1949 {
1950 #ifndef WITH_GNUTLS
1951   return 0;
1952 #else
1953   return pwm ? pwm->tls_error : 0;
1954 #endif
1955 }