Don't close the TCP FD when an async command completes.
[libpwmd.git] / src / libpwmd.c
blob0a84c1a22c4b74bb8f4e2bf593e2996f363c724d
1 /* vim:tw=78:ts=8:sw=4:set ft=c: */
2 /*
3 Copyright (C) 2006-2009 Ben Kibbey <bjk@luxsci.net>
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02110-1301 USA
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <unistd.h>
22 #include <err.h>
23 #include <errno.h>
24 #include <ctype.h>
25 #include <string.h>
26 #include <sys/socket.h>
27 #include <sys/un.h>
28 #include <signal.h>
29 #include <stdarg.h>
30 #include <string.h>
31 #include <sys/wait.h>
32 #include <fcntl.h>
33 #include <pwd.h>
34 #include <time.h>
35 #include <sys/types.h>
36 #include <limits.h>
37 #include <sys/select.h>
38 #include <netdb.h>
39 #include <netinet/in.h>
40 #include <sys/socket.h>
41 #include <libpwmd.h>
43 #ifdef HAVE_CONFIG_H
44 #include <config.h>
45 #endif
47 #ifdef WITH_LIBPTH
48 #include <pth.h>
49 #endif
51 #ifdef WITH_TCP
52 #define DNS_USE_GETTIMEOFDAY_FOR_ID 1
53 #include <ares.h>
54 #include <arpa/nameser.h>
55 #endif
57 #ifdef HAVE_ASSUAN_H
58 #include <assuan.h>
59 #endif
61 #ifdef HAVE_SETLOCALE
62 #include <locale.h>
63 #endif
65 #include "gettext.h"
66 #define N_(msgid) dgettext("libpwmd", msgid)
68 #include "mem.h"
69 #include "types.h"
71 #ifdef WITH_PINENTRY
72 static pwm_t *gpwm;
73 static int gelapsed, gtimeout;
74 static gpg_error_t pinentry_command(pwm_t *pwm, char **result, const char *cmd);
75 #endif
77 static char *_getpwuid(struct passwd *pwd)
79 size_t size = sysconf(_SC_GETPW_R_SIZE_MAX);
80 struct passwd *result;
81 char *buf;
82 int n;
84 if (size == -1)
85 size = 16384;
87 buf = pwmd_malloc(size);
89 if (!buf)
90 return NULL;
92 n = getpwuid_r(getuid(), pwd, buf, size, &result);
94 if (n) {
95 pwmd_free(buf);
96 errno = n;
97 return NULL;
100 if (!result) {
101 pwmd_free(buf);
102 return NULL;
105 errno = n;
106 return buf;
109 const char *_pwmd_strerror(gpg_error_t e)
111 gpg_err_code_t code = gpg_err_code(e);
113 if (code >= GPG_ERR_USER_1 && code < gpg_err_code(EPWMD_MAX)) {
114 switch (code) {
115 default:
116 return NULL;
117 case GPG_ERR_USER_1:
118 return N_("Unknown error");
119 case GPG_ERR_USER_2:
120 return N_("No cache slots available");
121 case GPG_ERR_USER_3:
122 return N_("Recursion loop");
123 case GPG_ERR_USER_4:
124 return N_("No file is open");
125 case GPG_ERR_USER_5:
126 return N_("General LibXML error");
127 case GPG_ERR_USER_6:
128 return N_("File modified");
129 case GPG_ERR_USER_7:
130 return N_("Access denied");
134 return NULL;
137 const char *pwmd_strerror(gpg_error_t e)
139 const char *p = _pwmd_strerror(e);
141 return p ? p : gpg_strerror(e);
144 int pwmd_strerror_r(gpg_error_t e, char *buf, size_t size)
146 const char *p = _pwmd_strerror(e);
148 if (p) {
149 snprintf(buf, size, "%s", p);
151 if (strlen(p) > size)
152 return ERANGE;
154 return 0;
157 return gpg_strerror_r(e, buf, size);
160 gpg_error_t pwmd_init()
162 static int initialized;
164 if (initialized)
165 return 0;
167 #ifndef MEM_DEBUG
168 xmem_init();
169 #endif
170 #ifdef ENABLE_NLS
171 bindtextdomain("libpwmd", LOCALEDIR);
172 #endif
173 gpg_err_init();
174 assuan_set_malloc_hooks(pwmd_malloc, pwmd_realloc, pwmd_free);
175 assuan_set_assuan_err_source(GPG_ERR_SOURCE_DEFAULT);
176 initialized = 1;
177 return 0;
180 static gpg_error_t _socket_connect_finalize(pwm_t *pwm)
182 int active[2];
183 int n = assuan_get_active_fds(pwm->ctx, 0, active, N_ARRAY(active));
185 pwm->fd = n <= 0 ? -1 : dup(active[0]);
186 #ifdef WITH_PINENTRY
187 pwm->pid = -1;
188 pwm->pinentry_tries = 3;
189 #endif
190 assuan_set_pointer(pwm->ctx, pwm);
192 if (pwm->name)
193 return pwmd_command(pwm, NULL, "OPTION CLIENT NAME=%s", pwm->name);
195 return 0;
198 #ifdef WITH_TCP
199 static int read_hook(assuan_context_t ctx, assuan_fd_t fd, void *data,
200 size_t len, ssize_t *ret)
202 pwm_t *pwm = assuan_get_pointer(ctx);
204 if (!pwm || !pwm->tcp_conn)
205 #ifdef WITH_LIBPTH
206 *ret = pth_read((int)fd, data, len);
207 #else
208 *ret = read((int)fd, data, len);
209 #endif
210 else {
211 do {
212 *ret = libssh2_channel_read(pwm->tcp_conn->channel, data, len);
213 } while (*ret == LIBSSH2_ERROR_EAGAIN);
216 return *ret <= 0 ? 0 : 1;
219 static int write_hook(assuan_context_t ctx, assuan_fd_t fd, const void *data,
220 size_t len, ssize_t *ret)
222 pwm_t *pwm = assuan_get_pointer(ctx);
224 if (!pwm || !pwm->tcp_conn)
225 #ifdef WITH_LIBPTH
226 *ret = pth_write((int)fd, data, len);
227 #else
228 *ret = write((int)fd, data, len);
229 #endif
230 else {
231 do {
232 *ret = libssh2_channel_write(pwm->tcp_conn->channel, data, len);
233 } while (*ret == LIBSSH2_ERROR_EAGAIN);
236 return *ret <= 0 ? 0 : 1;
239 static void _ssh_deinit(pwmd_tcp_conn_t *conn);
240 static void free_tcp_conn(pwmd_tcp_conn_t *conn)
242 if (!conn)
243 return;
245 if (conn->username) {
246 pwmd_free(conn->username);
247 conn->username = NULL;
250 if (conn->known_hosts) {
251 pwmd_free(conn->known_hosts);
252 conn->known_hosts = NULL;
255 if (conn->identity) {
256 pwmd_free(conn->identity);
257 conn->identity = NULL;
260 if (conn->identity_pub) {
261 pwmd_free(conn->identity_pub);
262 conn->identity_pub = NULL;
265 if (conn->host) {
266 pwmd_free(conn->host);
267 conn->host = NULL;
270 if (conn->hostkey) {
271 pwmd_free(conn->hostkey);
272 conn->hostkey = NULL;
275 if (conn->chan) {
276 ares_destroy(conn->chan);
277 conn->chan = NULL;
280 if (conn->he) {
281 ares_free_hostent(conn->he);
282 conn->he = NULL;
285 if (conn->fd >= 0) {
286 close(conn->fd);
287 conn->fd = -1;
290 if (conn->session)
291 _ssh_deinit(conn);
292 else
293 pwmd_free(conn);
296 static void _ssh_deinit(pwmd_tcp_conn_t *conn)
298 if (!conn)
299 return;
301 if (conn->channel)
302 libssh2_channel_free(conn->channel);
304 if (conn->session) {
305 libssh2_session_disconnect(conn->session, "Bye!");
306 libssh2_session_free(conn->session);
309 conn->session = NULL;
310 conn->channel = NULL;
311 free_tcp_conn(conn);
314 static void _ssh_assuan_deinit(assuan_context_t ctx)
316 pwm_t *pwm = assuan_get_pointer(ctx);
318 _ssh_deinit(pwm->tcp_conn);
319 pwm->tcp_conn = NULL;
323 * Sets common options from both pwmd_tcp_connect() and
324 * pwmd_tcp_connect_async().
326 static gpg_error_t init_tcp_conn(pwmd_tcp_conn_t **dst, const char *host,
327 int port, const char *identity, const char *user, const char *hosts,
328 int get)
330 pwmd_tcp_conn_t *conn;
331 gpg_error_t rc = 0;
333 if (get) {
334 if (!host)
335 return GPG_ERR_INV_ARG;
337 else {
338 if (!host || !identity || !hosts)
339 return GPG_ERR_INV_ARG;
342 conn = pwmd_calloc(1, sizeof(pwmd_tcp_conn_t));
344 if (!conn)
345 return gpg_error_from_errno(ENOMEM);
347 conn->port = port == -1 ? 22 : port;
348 conn->host = pwmd_strdup(host);
350 if (!conn->host) {
351 rc = gpg_error_from_errno(ENOMEM);
352 goto fail;
355 if (!get) {
356 struct passwd pw;
357 char *pwbuf = _getpwuid(&pw);
359 if (!pwbuf) {
360 rc = gpg_error_from_errno(errno);
361 goto fail;
364 conn->username = pwmd_strdup(user ? user : pw.pw_name);
365 pwmd_free(pwbuf);
367 if (!conn->username) {
368 rc = gpg_error_from_errno(ENOMEM);
369 goto fail;
372 conn->identity = pwmd_strdup(identity);
374 if (!conn->identity) {
375 rc = gpg_error_from_errno(ENOMEM);
376 goto fail;
379 conn->identity_pub = pwmd_malloc(strlen(conn->identity)+5);
381 if (!conn->identity_pub) {
382 rc = gpg_error_from_errno(ENOMEM);
383 goto fail;
386 sprintf(conn->identity_pub, "%s.pub", conn->identity);
387 conn->known_hosts = pwmd_strdup(hosts);
389 if (!conn->known_hosts) {
390 rc = gpg_error_from_errno(ENOMEM);
391 goto fail;
395 *dst = conn;
396 return 0;
398 fail:
399 free_tcp_conn(conn);
400 return rc;
403 static gpg_error_t do_connect(pwm_t *pwm, int prot, void *addr)
405 struct sockaddr_in their_addr;
407 pwm->tcp_conn->fd = socket(prot, SOCK_STREAM, 0);
409 if (pwm->tcp_conn->fd == -1)
410 return gpg_error_from_syserror();
412 if (pwm->tcp_conn->async)
413 fcntl(pwm->tcp_conn->fd, F_SETFL, O_NONBLOCK);
415 pwm->cmd = ASYNC_CMD_CONNECT;
416 their_addr.sin_family = prot;
417 their_addr.sin_port = htons(pwm->tcp_conn->port);
418 their_addr.sin_addr = *((struct in_addr *)addr);
419 memset(their_addr.sin_zero, '\0', sizeof their_addr.sin_zero);
421 #ifdef WITH_LIBPTH
422 if (pth_connect(pwm->tcp_conn->fd, (struct sockaddr *)&their_addr,
423 sizeof(their_addr)) == -1)
424 #else
425 if (connect(pwm->tcp_conn->fd, (struct sockaddr *)&their_addr,
426 sizeof(their_addr)) == -1)
427 #endif
428 return gpg_error_from_syserror();
430 return 0;
433 static gpg_error_t ares_error_to_pwmd(int status)
435 if (status != ARES_SUCCESS)
436 warnx("%s", ares_strerror(status));
438 switch (status) {
439 case ARES_ENODATA:
440 case ARES_EFORMERR:
441 case ARES_ENOTFOUND:
442 return GPG_ERR_UNKNOWN_HOST;
443 case ARES_ESERVFAIL:
444 return GPG_ERR_EHOSTDOWN;
445 case ARES_ETIMEOUT:
446 return GPG_ERR_TIMEOUT;
447 case ARES_ENOMEM:
448 return gpg_error_from_errno(ENOMEM);
449 case ARES_ECONNREFUSED:
450 return GPG_ERR_ECONNREFUSED;
451 default:
452 /* FIXME ??? */
453 return GPG_ERR_EHOSTUNREACH;
456 return ARES_SUCCESS;
459 static void dns_resolve_cb(void *arg, int status, int timeouts,
460 unsigned char *abuf, int alen)
462 pwm_t *pwm = arg;
463 int rc;
464 struct hostent *he;
466 if (status == ARES_EDESTRUCTION)
467 return;
469 if (status != ARES_SUCCESS) {
470 pwm->tcp_conn->rc = ares_error_to_pwmd(status);
471 return;
474 //FIXME localhost. works with ipv4. maybe local system config error
475 /* Check for an IPv6 address first. */
476 rc = ares_parse_a_reply(abuf, alen, &he, NULL, NULL);
478 if (rc != ARES_SUCCESS) {
479 if (rc != ARES_ENODATA) {
480 pwm->tcp_conn->rc = ares_error_to_pwmd(status);
481 return;
484 rc = ares_parse_aaaa_reply(abuf, alen, &he, NULL, NULL);
486 if (rc != ARES_SUCCESS) {
487 pwm->tcp_conn->rc = ares_error_to_pwmd(status);
488 return;
492 pwm->tcp_conn->he = he;
493 pwm->tcp_conn->rc = do_connect(pwm, he->h_addrtype, he->h_addr);
496 static gpg_error_t _do_pwmd_tcp_connect_async(pwm_t *pwm, const char *host,
497 int port, const char *identity, const char *user,
498 const char *known_hosts, pwmd_async_cmd_t which)
500 pwmd_tcp_conn_t *conn;
501 gpg_error_t rc;
503 rc = init_tcp_conn(&conn, host, port, identity, user, known_hosts,
504 which == ASYNC_CMD_HOSTKEY ? 1 : 0);
506 if (rc)
507 return rc;
509 conn->async = 1;
510 pwm->tcp_conn = conn;
511 pwm->tcp_conn->cmd = which;
513 if (pwm->tcp_conn->cmd == ASYNC_CMD_HOSTKEY)
514 pwm->tcp_conn->get_only = 1;
516 pwm->cmd = ASYNC_CMD_DNS;
517 pwm->state = ASYNC_PROCESS;
518 ares_init(&pwm->tcp_conn->chan);
519 ares_query(pwm->tcp_conn->chan, pwm->tcp_conn->host, ns_c_any, ns_t_any,
520 dns_resolve_cb, pwm);
521 return 0;
524 gpg_error_t pwmd_tcp_connect_async(pwm_t *pwm, const char *host, int port,
525 const char *identity, const char *user, const char *known_hosts)
527 return _do_pwmd_tcp_connect_async(pwm, host, port, identity, user,
528 known_hosts, ASYNC_CMD_CONNECT);
531 static void *_ssh_malloc(size_t size, void **data)
533 return pwmd_malloc(size);
536 static void _ssh_free(void *ptr, void **data)
538 pwmd_free(ptr);
541 static void *_ssh_realloc(void *ptr, size_t size, void **data)
543 return pwmd_realloc(ptr, size);
546 static char *to_hex(const char *str, size_t slen)
548 int i;
549 char *buf = pwmd_malloc(slen*2+1);
551 if (!buf)
552 return NULL;
554 for (i = 0, buf[0] = 0; i < slen; i++) {
555 char tmp[3];
557 sprintf(tmp, "%02x", (unsigned char)str[i]);
558 strcat(buf, tmp);
561 return buf;
564 static int verify_host_key(pwm_t *pwm)
566 FILE *fp = fopen(pwm->tcp_conn->known_hosts, "r");
567 char *buf, *p;
569 if (!fp)
570 return 1;
572 buf = pwmd_malloc(LINE_MAX);
574 if (!buf)
575 goto fail;
577 while ((p = fgets(buf, LINE_MAX, fp))) {
578 if (*p == '#' || isspace(*p))
579 continue;
581 if (p[strlen(p)-1] == '\n')
582 p[strlen(p)-1] = 0;
584 if (!strcmp(buf, pwm->tcp_conn->hostkey))
585 goto done;
588 fail:
589 if (buf)
590 pwmd_free(buf);
592 fclose(fp);
593 return 1;
595 done:
596 pwmd_free(buf);
597 fclose(fp);
598 return 0;
601 static gpg_error_t authenticate_ssh(pwm_t *pwm)
603 const char *fp = libssh2_hostkey_hash(pwm->tcp_conn->session,
604 LIBSSH2_HOSTKEY_HASH_SHA1);
605 char *userauth;
607 pwm->tcp_conn->hostkey = to_hex(fp, 20);
609 if (!pwm->tcp_conn->hostkey)
610 return gpg_error_from_errno(ENOMEM);
612 if (pwm->tcp_conn->get_only)
613 return 0;
615 if (!fp || verify_host_key(pwm))
616 return GPG_ERR_CHECKSUM;
618 userauth = libssh2_userauth_list(pwm->tcp_conn->session,
619 pwm->tcp_conn->username, strlen(pwm->tcp_conn->username));
621 if (!userauth || !strstr(userauth, "publickey"))
622 return GPG_ERR_BAD_PIN_METHOD;
624 if (libssh2_userauth_publickey_fromfile(pwm->tcp_conn->session,
625 pwm->tcp_conn->username, pwm->tcp_conn->identity_pub,
626 pwm->tcp_conn->identity, NULL))
627 return GPG_ERR_BAD_SECKEY;
629 return 0;
632 static gpg_error_t setup_tcp_session(pwm_t *pwm)
634 assuan_context_t ctx;
635 struct assuan_io_hooks io_hooks = {read_hook, write_hook};
636 gpg_error_t rc;
638 pwm->tcp_conn->session = libssh2_session_init_ex(_ssh_malloc, _ssh_free,
639 _ssh_realloc, NULL);
641 if (!pwm->tcp_conn->session) {
642 rc = gpg_error_from_errno(ENOMEM);
643 goto fail;
646 if (libssh2_session_startup(pwm->tcp_conn->session, pwm->tcp_conn->fd)) {
647 rc = GPG_ERR_ASSUAN_SERVER_FAULT;
648 goto fail;
651 rc = authenticate_ssh(pwm);
653 if (rc)
654 goto fail;
656 /* pwmd_get_hostkey(). */
657 if (pwm->tcp_conn->get_only) {
658 pwm->result = pwmd_strdup(pwm->tcp_conn->hostkey);
660 if (!pwm->result) {
661 rc = gpg_error_from_errno(ENOMEM);
662 goto fail;
665 return 0;
668 pwm->tcp_conn->channel = libssh2_channel_open_session(pwm->tcp_conn->session);
670 if (!pwm->tcp_conn->channel) {
671 rc = GPG_ERR_ASSUAN_SERVER_FAULT;
672 goto fail;
675 if (libssh2_channel_shell(pwm->tcp_conn->channel)) {
676 rc = GPG_ERR_ASSUAN_SERVER_FAULT;
677 goto fail;
680 assuan_set_io_hooks(&io_hooks);
681 rc = assuan_socket_connect_fd(&ctx, pwm->tcp_conn->fd, 0, pwm);
683 if (rc)
684 goto fail;
686 assuan_set_finish_handler(ctx, _ssh_assuan_deinit);
687 pwm->ctx = ctx;
688 return _socket_connect_finalize(pwm);
690 fail:
691 free_tcp_conn(pwm->tcp_conn);
692 pwm->tcp_conn = NULL;
693 return rc;
696 static gpg_error_t _do_pwmd_tcp_connect(pwm_t *pwm, const char *host, int port,
697 const char *identity, const char *user, const char *known_hosts, int get)
699 pwmd_tcp_conn_t *conn;
700 gpg_error_t rc;
702 rc = init_tcp_conn(&conn, host, port, identity, user, known_hosts, get);
704 if (rc)
705 return rc;
707 pwm->tcp_conn = conn;
708 pwm->tcp_conn->get_only = get;
709 pwm->cmd = ASYNC_CMD_DNS;
710 ares_init(&pwm->tcp_conn->chan);
711 ares_query(pwm->tcp_conn->chan, pwm->tcp_conn->host, ns_c_any, ns_t_any,
712 dns_resolve_cb, pwm);
714 /* dns_resolve_cb() may have already been called. */
715 if (pwm->tcp_conn->rc) {
716 rc = pwm->tcp_conn->rc;
717 goto fail;
721 * Fake a blocking DNS lookup. libcares does a better job than
722 * getaddrinfo().
724 do {
725 fd_set rfds, wfds;
726 int n;
727 struct timeval tv;
729 FD_ZERO(&rfds);
730 FD_ZERO(&wfds);
731 n = ares_fds(pwm->tcp_conn->chan, &rfds, &wfds);
732 ares_timeout(pwm->tcp_conn->chan, NULL, &tv);
733 #ifdef WITH_LIBPTH
734 n = pth_select(n, &rfds, &wfds, NULL, &tv);
735 #else
736 n = select(n, &rfds, &wfds, NULL, &tv);
737 #endif
739 if (n == -1) {
740 rc = gpg_error_from_syserror();
741 goto fail;
743 else if (n == 0) {
744 rc = GPG_ERR_TIMEOUT;
745 goto fail;
748 ares_process(pwm->tcp_conn->chan, &rfds, &wfds);
750 if (pwm->tcp_conn->rc)
751 break;
752 } while (pwm->cmd == ASYNC_CMD_DNS);
754 if (pwm->tcp_conn->rc) {
755 rc = pwm->tcp_conn->rc;
756 goto fail;
759 return setup_tcp_session(pwm);
761 fail:
762 return rc;
765 gpg_error_t pwmd_tcp_connect(pwm_t *pwm, const char *host, int port,
766 const char *identity, const char *user, const char *known_hosts)
768 return _do_pwmd_tcp_connect(pwm, host, port, identity, user, known_hosts, 0);
771 /* Must free the result with pwmd_free(). */
772 gpg_error_t pwmd_get_hostkey(const char *host, int port, char **result)
774 char *hostkey;
775 pwm_t *pwm = pwmd_new(NULL);
776 gpg_error_t rc;
778 if (!pwm)
779 return gpg_error_from_errno(ENOMEM);
781 rc = _do_pwmd_tcp_connect(pwm, host, port, NULL, NULL, NULL, 1);
783 if (rc) {
784 pwmd_close(pwm);
785 return rc;
788 hostkey = pwmd_strdup(pwm->tcp_conn->hostkey);
790 if (!hostkey)
791 rc = gpg_error_from_errno(ENOMEM);
793 pwmd_close(pwm);
794 *result = hostkey;
795 return rc;
798 gpg_error_t pwmd_get_hostkey_async(pwm_t *pwm, const char *host, int port)
800 return _do_pwmd_tcp_connect_async(pwm, host, port, NULL, NULL, NULL,
801 ASYNC_CMD_HOSTKEY);
803 #endif
805 gpg_error_t pwmd_connect(pwm_t *pwm, const char *path)
807 char *socketpath = NULL;
808 assuan_context_t ctx;
809 struct passwd pw;
810 char *pwbuf = _getpwuid(&pw);
811 gpg_error_t rc;
813 if (!pwbuf)
814 return gpg_error_from_errno(errno);
816 if (!path) {
817 socketpath = (char *)pwmd_malloc(strlen(pw.pw_dir) + strlen("/.pwmd/socket") + 1);
818 sprintf(socketpath, "%s/.pwmd/socket", pw.pw_dir);
820 else
821 socketpath = pwmd_strdup(path);
823 pwmd_free(pwbuf);
824 rc = assuan_socket_connect_ext(&ctx, socketpath, -1, 0);
825 pwmd_free(socketpath);
827 if (rc)
828 return rc;
830 pwm->ctx = ctx;
831 return _socket_connect_finalize(pwm);
834 void pwmd_close(pwm_t *pwm)
836 if (!pwm)
837 return;
839 if (pwm->ctx)
840 assuan_disconnect(pwm->ctx);
842 if (pwm->password)
843 pwmd_free(pwm->password);
845 if (pwm->title)
846 pwmd_free(pwm->title);
848 if (pwm->desc)
849 pwmd_free(pwm->desc);
851 if (pwm->prompt)
852 pwmd_free(pwm->prompt);
854 if (pwm->pinentry_tty)
855 pwmd_free(pwm->pinentry_tty);
857 if (pwm->pinentry_display)
858 pwmd_free(pwm->pinentry_display);
860 if (pwm->pinentry_term)
861 pwmd_free(pwm->pinentry_term);
863 if (pwm->lcctype)
864 pwmd_free(pwm->lcctype);
866 if (pwm->lcmessages)
867 pwmd_free(pwm->lcmessages);
869 if (pwm->filename)
870 pwmd_free(pwm->filename);
872 if (pwm->name)
873 pwmd_free(pwm->name);
875 #ifdef WITH_TCP
876 if (pwm->tcp_conn)
877 free_tcp_conn(pwm->tcp_conn);
878 #endif
880 pwmd_free(pwm);
883 static int mem_realloc_cb(void *data, const void *buffer, size_t len)
885 membuf_t *mem = (membuf_t *)data;
886 void *p;
888 if (!buffer)
889 return 0;
891 if ((p = pwmd_realloc(mem->buf, mem->len + len)) == NULL)
892 return 1;
894 mem->buf = p;
895 memcpy((char *)mem->buf + mem->len, buffer, len);
896 mem->len += len;
897 return 0;
900 static int _inquire_cb(void *data, const char *keyword)
902 pwm_t *pwm = (pwm_t *)data;
903 gpg_error_t rc = 0;
904 int flags = fcntl(pwm->fd, F_GETFL);
906 /* Shouldn't get this far without a callback. */
907 if (!pwm->inquire_func)
908 return GPG_ERR_INV_ARG;
911 * Since the socket file descriptor is probably set to non-blocking, set to
912 * blocking to prevent GPG_ERR_EAGAIN errors. This should be fixes when
913 * asynchronous INQUIRE is supported by either libassuan or a later
914 * libpwmd.
916 fcntl(pwm->fd, F_SETFL, 0);
918 for (;;) {
919 char *result = NULL;
920 size_t len;
921 gpg_error_t arc;
923 rc = pwm->inquire_func(pwm->inquire_data, keyword, rc, &result, &len);
924 rc = gpg_err_code(rc);
926 if (rc == GPG_ERR_EOF || !rc) {
927 if (len <= 0 || !result || !*result) {
928 rc = 0;
929 break;
932 arc = assuan_send_data(pwm->ctx, result, len);
934 if (rc == GPG_ERR_EOF) {
935 rc = arc;
936 break;
939 rc = arc;
941 else if (rc)
942 break;
945 fcntl(pwm->fd, F_SETFL, flags);
946 return rc;
949 static gpg_error_t do_nb_command(pwm_t *pwm, const char *cmd, const char *arg)
951 char *buf;
952 gpg_error_t rc;
953 size_t len = strlen(cmd) + 2;
955 len += arg ? strlen(arg) : 0;
957 if (pwm->state == ASYNC_DONE)
958 pwm->state = ASYNC_INIT;
960 if (pwm->state != ASYNC_INIT)
961 return GPG_ERR_INV_STATE;
963 buf = (char *)pwmd_malloc(len);
965 if (!buf) {
966 rc = gpg_error_from_errno(ENOMEM);
967 goto fail;
970 snprintf(buf, len, "%s %s", cmd, arg ? arg : "");
971 rc = assuan_write_line(pwm->ctx, buf);
972 pwmd_free(buf);
974 if (!rc)
975 pwm->state = ASYNC_PROCESS;
977 fail:
978 return rc;
981 gpg_error_t pwmd_open_async(pwm_t *pwm, const char *filename)
983 if (!pwm || !filename)
984 return GPG_ERR_INV_ARG;
986 /* For pinentry retries. */
987 if (!pwm->is_open_cmd) {
988 if (pwm->filename)
989 pwmd_free(pwm->filename);
991 pwm->filename = pwmd_strdup(filename);
994 pwm->is_open_cmd = 1;
995 pwm->cmd = ASYNC_CMD_OPEN;
996 return do_nb_command(pwm, "OPEN", filename);
999 gpg_error_t pwmd_save_async(pwm_t *pwm)
1001 if (!pwm)
1002 return GPG_ERR_INV_ARG;
1004 pwm->cmd = ASYNC_CMD_SAVE;
1005 return do_nb_command(pwm, "SAVE", NULL);
1008 static gpg_error_t parse_assuan_line(pwm_t *pwm)
1010 gpg_error_t rc;
1011 char *line;
1012 size_t len;
1014 rc = assuan_read_line(pwm->ctx, &line, &len);
1016 if (!rc) {
1017 if (line[0] == 'O' && line[1] == 'K' &&
1018 (line[2] == 0 || line[2] == ' ')) {
1019 pwm->state = ASYNC_DONE;
1021 else if (line[0] == '#') {
1023 else if (line[0] == 'S' && (line[1] == 0 || line[1] == ' ')) {
1024 if (pwm->status_func) {
1025 pwm->status_func(pwm->status_data,
1026 line[1] == 0 ? line+1 : line+2);
1029 else if (line[0] == 'E' && line[1] == 'R' && line[2] == 'R' &&
1030 (line[3] == 0 || line[3] == ' ')) {
1031 line += 4;
1032 rc = atoi(line);
1033 pwm->state = ASYNC_DONE;
1037 return rc;
1040 gpg_error_t pwmd_pending_line(pwm_t *pwm)
1042 if (!pwm)
1043 return GPG_ERR_INV_ARG;
1045 if (!pwm->ctx)
1046 return GPG_ERR_INV_STATE;
1048 return assuan_pending_line(pwm->ctx) ? 0 : GPG_ERR_NO_DATA;
1051 static pwmd_async_t reset_async(pwm_t *pwm, int done)
1053 pwm->state = ASYNC_INIT;
1054 pwm->cmd = ASYNC_CMD_NONE;
1055 pwm->ntries = 0;
1056 #ifdef WITH_PINENTRY
1057 pwm->is_open_cmd = 0;
1059 if (pwm->nb_fd != -1) {
1060 close(pwm->nb_fd);
1061 pwm->nb_fd = -1;
1063 #endif
1064 #ifdef WITH_TCP
1065 if (done && pwm->tcp_conn && pwm->tcp_conn->fd != -1) {
1066 close(pwm->tcp_conn->fd);
1067 pwm->tcp_conn->fd = -1;
1069 #endif
1071 return ASYNC_DONE;
1074 pwmd_async_t pwmd_process(pwm_t *pwm, gpg_error_t *rc, char **result)
1076 fd_set fds;
1077 int n;
1078 struct timeval tv = {0, 0};
1080 *rc = 0;
1082 if (!pwm) {
1083 *rc = GPG_ERR_INV_ARG;
1084 return ASYNC_DONE;
1087 /* When not in a command, this will let libassuan process status messages
1088 * by calling PWMD_OPTION_STATUS_FUNC. The client can poll the file
1089 * descriptor returned by pwmd_get_fd() to determine when this should be
1090 * called or call pwmd_pending_line() to determine whether a buffered line
1091 * needs to be processed. */
1092 if (pwm->cmd == ASYNC_CMD_NONE) {
1093 *rc = assuan_command(pwm, pwm->ctx, NULL, "NOP");
1094 return pwm->state;
1097 /* Fixes pwmd_open/save_async2() when there is a cached or new file. */
1098 if (pwm->state == ASYNC_DONE) {
1099 reset_async(pwm, 0);
1100 return ASYNC_DONE;
1103 if (pwm->state != ASYNC_PROCESS) {
1104 *rc = GPG_ERR_INV_STATE;
1105 return ASYNC_DONE;
1108 #ifdef WITH_TCP
1109 if (pwm->cmd == ASYNC_CMD_DNS) {
1110 fd_set rfds, wfds;
1112 if (pwm->tcp_conn->rc) {
1113 *rc = pwm->tcp_conn->rc;
1114 reset_async(pwm, 1);
1115 return ASYNC_DONE;
1118 FD_ZERO(&rfds);
1119 FD_ZERO(&wfds);
1120 n = ares_fds(pwm->tcp_conn->chan, &rfds, &wfds);
1122 /* Shouldn't happen. */
1123 if (!n)
1124 return pwm->state;
1126 #ifdef WITH_LIBPTH
1127 n = pth_select(n, &rfds, &wfds, NULL, &tv);
1128 #else
1129 n = select(n, &rfds, &wfds, NULL, &tv);
1130 #endif
1132 if (n > 0)
1133 ares_process(pwm->tcp_conn->chan, &rfds, &wfds);
1135 return pwm->state;
1137 else if (pwm->cmd == ASYNC_CMD_CONNECT) {
1138 if (pwm->tcp_conn->rc == GPG_ERR_EINPROGRESS) {
1139 int ret;
1140 socklen_t len = sizeof(int);
1142 FD_ZERO(&fds);
1143 FD_SET(pwm->tcp_conn->fd, &fds);
1144 #ifdef WITH_LIBPTH
1145 n = pth_select(pwm->tcp_conn->fd+1, NULL, &fds, NULL, &tv);
1146 #else
1147 n = select(pwm->tcp_conn->fd+1, NULL, &fds, NULL, &tv);
1148 #endif
1150 if (!n || !FD_ISSET(pwm->tcp_conn->fd, &fds))
1151 return pwm->state;
1152 else if (n == -1) {
1153 *rc = gpg_error_from_syserror();
1154 reset_async(pwm, 1);
1155 return ASYNC_DONE;
1158 ret = getsockopt(pwm->tcp_conn->fd, SOL_SOCKET, SO_ERROR, &n, &len);
1160 if (ret || n) {
1161 *rc = ret ? gpg_error_from_syserror() : gpg_error_from_errno(n);
1162 reset_async(pwm, 1);
1163 return ASYNC_DONE;
1166 else if (pwm->tcp_conn->rc) {
1167 *rc = pwm->tcp_conn->rc;
1168 reset_async(pwm, 1);
1169 return ASYNC_DONE;
1172 fcntl(pwm->tcp_conn->fd, F_SETFL, 0);
1173 *rc = setup_tcp_session(pwm);
1175 if (!*rc) {
1176 switch (pwm->tcp_conn->cmd) {
1177 case ASYNC_CMD_HOSTKEY:
1178 if (!*rc)
1179 *result = pwm->result;
1180 break;
1181 default:
1182 break;
1186 return reset_async(pwm, *rc ? 1 : 0);
1188 #endif
1190 #ifdef WITH_PINENTRY
1191 if (pwm->cmd == ASYNC_CMD_OPEN2 || pwm->cmd == ASYNC_CMD_SAVE2) {
1192 if (pwm->nb_fd == -1) {
1193 *rc = GPG_ERR_INV_STATE;
1194 return reset_async(pwm, 1);
1197 FD_ZERO(&fds);
1198 FD_SET(pwm->nb_fd, &fds);
1199 #ifdef WITH_LIBPTH
1200 n = pth_select(pwm->nb_fd+1, &fds, NULL, NULL, &tv);
1201 #else
1202 n = select(pwm->nb_fd+1, &fds, NULL, NULL, &tv);
1203 #endif
1204 if (n == -1) {
1205 *rc = gpg_error_from_syserror();
1206 return reset_async(pwm, pwm->cmd == ASYNC_CMD_OPEN2 ? 1 : 0);
1209 if (n > 0) {
1210 pwmd_nb_status_t nb;
1211 size_t len = read(pwm->nb_fd, &nb, sizeof(nb));
1213 *rc = nb.error;
1214 return reset_async(pwm, pwm->cmd == ASYNC_CMD_OPEN2 ? 1 : 0);
1217 return pwm->state;
1219 #endif
1221 if (pwm->fd < 0) {
1222 *rc = GPG_ERR_INV_STATE;
1223 return reset_async(pwm, 1);
1226 /* This is for the non-blocking OPEN and SAVE commands. */
1227 FD_ZERO(&fds);
1228 FD_SET(pwm->fd, &fds);
1229 #ifdef WITH_LIBPTH
1230 n = pth_select(pwm->fd+1, &fds, NULL, NULL, &tv);
1231 #else
1232 n = select(pwm->fd+1, &fds, NULL, NULL, &tv);
1233 #endif
1235 if (n == -1) {
1236 *rc = gpg_error_from_syserror();
1237 return reset_async(pwm, 0);
1240 if (n > 0) {
1241 if (FD_ISSET(pwm->fd, &fds)) {
1242 *rc = parse_assuan_line(pwm);
1244 if (*rc)
1245 return reset_async(pwm, 0);
1249 while (!*rc && assuan_pending_line(pwm->ctx))
1250 *rc = parse_assuan_line(pwm);
1252 if (*rc)
1253 return reset_async(pwm, 0);
1255 /* For pinentry retries. */
1256 if (pwm->is_open_cmd && gpg_err_code(*rc) == EPWMD_BADKEY &&
1257 ++pwm->ntries < pwm->pinentry_tries) {
1258 pwm->state = ASYNC_INIT;
1259 *rc = pwmd_open_async(pwm, pwm->filename);
1262 if (*rc)
1263 return reset_async(pwm, 1);
1265 if (pwm->state == ASYNC_DONE) {
1266 reset_async(pwm, 0);
1267 return ASYNC_DONE;
1270 return pwm->state;
1273 static gpg_error_t assuan_command(pwm_t *pwm, assuan_context_t ctx,
1274 char **result, const char *cmd)
1276 membuf_t data;
1277 gpg_error_t rc;
1279 data.len = 0;
1280 data.buf = NULL;
1282 rc = assuan_transact(ctx, cmd, mem_realloc_cb, &data, _inquire_cb, pwm,
1283 pwm->status_func, pwm->status_data);
1285 if (rc) {
1286 if (data.buf) {
1287 pwmd_free(data.buf);
1288 data.buf = NULL;
1291 else {
1292 if (data.buf) {
1293 mem_realloc_cb(&data, "", 1);
1295 if (!result) {
1296 pwmd_free(data.buf);
1297 rc = GPG_ERR_INV_ARG;
1299 else
1300 *result = (char *)data.buf;
1304 return gpg_err_code(rc);
1307 gpg_error_t pwmd_inquire(pwm_t *pwm, const char *cmd, pwmd_inquire_fn fn,
1308 void *data)
1310 if (!pwm || !cmd || !fn)
1311 return GPG_ERR_INV_ARG;
1313 pwm->inquire_func = fn;
1314 pwm->inquire_data = data;
1315 return assuan_command(pwm, pwm->ctx, NULL, cmd);
1318 static gpg_error_t terminate_pinentry(pwm_t *pwm)
1320 #ifndef WITH_PINENTRY
1321 return GPG_ERR_NOT_IMPLEMENTED;
1322 #else
1323 pid_t pid = pwm->pid;
1325 pwm->pid = -1;
1327 if (!pwm || pid == -1)
1328 return GPG_ERR_INV_ARG;
1330 if (kill(pid, 0) == 0) {
1331 if (kill(pid, SIGTERM) == -1) {
1332 if (kill(pid, SIGKILL) == -1)
1333 return gpg_error_from_errno(errno);
1336 else
1337 return gpg_error_from_errno(errno);
1339 return 0;
1340 #endif
1343 #ifdef WITH_PINENTRY
1344 static gpg_error_t set_pinentry_strings(pwm_t *pwm, int which)
1346 char *buf;
1347 char tmp[ASSUAN_LINELENGTH];
1348 gpg_error_t error;
1350 if (!pwm->title)
1351 pwm->title = pwmd_strdup(N_("LibPWMD"));
1353 if (!pwm->prompt)
1354 pwm->prompt = pwmd_strdup(N_("Passphrase:"));
1356 if (!pwm->desc && !which)
1357 pwm->desc = pwmd_strdup(N_("Enter a passphrase."));
1359 if (which == 1) {
1360 snprintf(tmp, sizeof(tmp), "SETERROR %s", N_("Invalid passphrase, please try again."));
1361 buf = pwmd_strdup(tmp);
1363 else if (which == 2) {
1364 snprintf(tmp, sizeof(tmp), "SETERROR %s", N_("Please type the passphrase again for confirmation."));
1365 buf = pwmd_strdup(tmp);
1367 else {
1368 buf = (char *)pwmd_malloc(strlen("SETERROR ") + strlen(pwm->desc) + 1);
1369 sprintf(buf, "SETERROR %s", pwm->desc);
1372 error = pinentry_command(pwm, NULL, buf);
1373 pwmd_free(buf);
1375 if (error)
1376 return error;
1378 buf = (char *)pwmd_malloc(strlen("SETPROMPT ") + strlen(pwm->prompt) + 1);
1379 sprintf(buf, "SETPROMPT %s", pwm->prompt);
1380 error = pinentry_command(pwm, NULL, buf);
1381 pwmd_free(buf);
1383 if (error)
1384 return error;
1386 buf = (char *)pwmd_malloc(strlen("SETDESC ") + strlen(pwm->title) + 1);
1387 sprintf(buf, "SETDESC %s", pwm->title);
1388 error = pinentry_command(pwm, NULL, buf);
1389 pwmd_free(buf);
1390 return error;
1393 static void update_pinentry_settings(pwm_t *pwm)
1395 FILE *fp;
1396 char buf[LINE_MAX];
1397 char *p;
1398 struct passwd pw;
1399 char *pwbuf = _getpwuid(&pw);
1401 if (!pwbuf)
1402 return;
1404 snprintf(buf, sizeof(buf), "%s/.pwmd/pinentry.conf", pw.pw_dir);
1405 pwmd_free(pwbuf);
1407 if ((fp = fopen(buf, "r")) == NULL)
1408 return;
1410 while ((p = fgets(buf, sizeof(buf), fp)) != NULL) {
1411 char name[32], val[256];
1413 if (sscanf(p, " %31[a-zA-Z] = %255s", name, val) != 2)
1414 continue;
1416 if (strcasecmp(name, "TTYNAME") == 0) {
1417 pwmd_free(pwm->pinentry_tty);
1418 pwm->pinentry_tty = pwmd_strdup(val);
1420 else if (strcasecmp(name, "TTYTYPE") == 0) {
1421 pwmd_free(pwm->pinentry_term);
1422 pwm->pinentry_term = pwmd_strdup(val);
1424 else if (strcasecmp(name, "DISPLAY") == 0) {
1425 pwmd_free(pwm->pinentry_display);
1426 pwm->pinentry_display = pwmd_strdup(val);
1428 else if (strcasecmp(name, "PATH") == 0) {
1429 pwmd_free(pwm->pinentry_path);
1430 pwm->pinentry_path = pwmd_strdup(val);
1434 fclose(fp);
1437 static gpg_error_t launch_pinentry(pwm_t *pwm)
1439 int rc;
1440 assuan_context_t ctx;
1441 int child_list[] = {-1};
1442 char *display = getenv("DISPLAY");
1443 const char *argv[10];
1444 const char **p = argv;
1445 int have_display = 0;
1446 char *tty = NULL;
1447 char *ttybuf = NULL;
1449 update_pinentry_settings(pwm);
1451 if (pwm->pinentry_display || display)
1452 have_display = 1;
1453 else {
1454 if (!pwm->pinentry_tty) {
1455 ttybuf = pwmd_malloc(255);
1457 if (!ttybuf)
1458 return gpg_error_from_errno(ENOMEM);
1460 rc = ttyname_r(STDOUT_FILENO, ttybuf, 255);
1462 if (rc) {
1463 pwmd_free(ttybuf);
1464 return gpg_error_from_errno(rc);
1467 tty = ttybuf;
1469 else
1470 tty = pwm->pinentry_tty;
1473 if (!have_display && !tty)
1474 return GPG_ERR_ENOTTY;
1476 *p++ = "pinentry";
1477 *p++ = have_display ? "--display" : "--ttyname";
1478 *p++ = have_display ? pwm->pinentry_display ? pwm->pinentry_display : display : tty;
1480 if (pwm->lcctype) {
1481 *p++ = "--lc-ctype";
1482 *p++ = pwm->lcctype;
1485 if (pwm->lcmessages) {
1486 *p++ = "--lc-messages";
1487 *p++ = pwm->lcmessages;
1490 *p = NULL;
1492 if (!have_display) {
1493 *p++ = "--ttytype";
1494 *p++ = pwm->pinentry_term ? pwm->pinentry_term : getenv("TERM");
1495 *p = NULL;
1498 rc = assuan_pipe_connect(&ctx, pwm->pinentry_path ? pwm->pinentry_path : PINENTRY_PATH, argv, child_list);
1500 if (ttybuf)
1501 pwmd_free(ttybuf);
1503 if (rc)
1504 return rc;
1506 pwm->pid = assuan_get_pid(ctx);
1507 pwm->pctx = ctx;
1508 return set_pinentry_strings(pwm, 0);
1511 static gpg_error_t pinentry_command(pwm_t *pwm, char **result, const char *cmd)
1513 gpg_error_t n;
1515 if (!pwm->pctx) {
1516 n = launch_pinentry(pwm);
1518 if (n)
1519 return n;
1522 return assuan_command(pwm, pwm->pctx, result, cmd);
1525 static void pinentry_disconnect(pwm_t *pwm)
1527 if (pwm->pctx)
1528 assuan_disconnect(pwm->pctx);
1530 pwm->pctx = NULL;
1531 pwm->pid = -1;
1535 * Only called from a child process.
1537 static void catchsig(int sig)
1539 switch (sig) {
1540 case SIGALRM:
1541 if (gelapsed++ >= gtimeout)
1542 terminate_pinentry(gpwm);
1543 else
1544 alarm(1);
1545 break;
1546 default:
1547 break;
1550 #endif
1553 * Borrowed from libassuan.
1555 static char *percent_escape(const char *atext)
1557 const unsigned char *s;
1558 int len = strlen(atext) * 3 + 1;
1559 char *buf = (char *)pwmd_malloc(len), *p = buf;
1561 if (!buf)
1562 return NULL;
1564 for (s=(const unsigned char *)atext; *s; s++) {
1565 if (*s < ' ') {
1566 sprintf (p, "%%%02X", *s);
1567 p += 3;
1569 else
1570 *p++ = *s;
1573 *p = 0;
1574 return buf;
1577 static gpg_error_t send_command(pwm_t *pwm, char **result, const char *cmd)
1579 if (!cmd)
1580 return GPG_ERR_INV_ARG;
1582 return assuan_command(pwm, pwm->ctx, result, cmd);
1585 gpg_error_t pwmd_command_ap(pwm_t *pwm, char **result, const char *cmd,
1586 va_list ap)
1588 char *buf;
1589 size_t len;
1590 gpg_error_t error;
1592 if (!pwm || !cmd)
1593 return GPG_ERR_INV_ARG;
1596 * C99 allows the dst pointer to be null which will calculate the length
1597 * of the would-be result and return it.
1599 len = vsnprintf(NULL, 0, cmd, ap)+1;
1600 buf = (char *)pwmd_malloc(len);
1601 len = vsnprintf(buf, len, cmd, ap);
1602 error = send_command(pwm, result, buf);
1603 pwmd_free(buf);
1604 return error;
1608 * Avoid sending the BYE command here. libassuan will close the file
1609 * descriptor and release the assuan context. Use pwmd_close() instead.
1611 gpg_error_t pwmd_command(pwm_t *pwm, char **result, const char *cmd, ...)
1613 va_list ap;
1614 gpg_error_t error;
1616 if (!pwm || !cmd)
1617 return GPG_ERR_INV_ARG;
1619 if (result)
1620 *result = NULL;
1622 va_start(ap, cmd);
1623 error = pwmd_command_ap(pwm, result, cmd, ap);
1624 va_end(ap);
1625 return error;
1628 #ifdef WITH_PINENTRY
1629 static gpg_error_t do_getpin(pwm_t *pwm, char **result)
1631 if (gtimeout) {
1632 signal(SIGALRM, catchsig);
1633 alarm(1);
1636 *result = NULL;
1637 return pinentry_command(pwm, result, "GETPIN");
1640 static gpg_error_t getpin(pwm_t *pwm, char **result, int *try_n, int which)
1642 int pin_try = *try_n;
1643 gpg_error_t error;
1645 getpin_again:
1646 *try_n = pin_try;
1648 if (pin_try == -1) {
1649 error = set_pinentry_strings(pwm, which);
1651 if (error) {
1652 pinentry_disconnect(pwm);
1653 return error;
1656 else {
1657 if (pwm->pinentry_tries-1 != pin_try) {
1658 error = set_pinentry_strings(pwm, 1);
1660 if (error) {
1661 pinentry_disconnect(pwm);
1662 return error;
1667 error = do_getpin(pwm, result);
1670 * Since there was input cancel any timeout setting.
1672 alarm(0);
1674 if (error) {
1675 if (error == GPG_ERR_CANCELED)
1676 return GPG_ERR_CANCELED;
1678 if (pin_try != -1 && pin_try--)
1679 goto getpin_again;
1681 if (pwm->pctx)
1682 pinentry_disconnect(pwm);
1684 *try_n = pin_try;
1685 return error;
1688 return 0;
1690 #endif
1692 static gpg_error_t do_open_command(pwm_t *pwm, const char *filename, char *password)
1694 char buf[ASSUAN_LINELENGTH];
1695 gpg_error_t error;
1696 char *result = NULL;
1698 snprintf(buf, sizeof(buf), "OPEN %s %s", filename, password ? password : "");
1699 error = send_command(pwm, &result, buf);
1700 memset(buf, 0, sizeof(buf));
1702 if (error && result)
1703 pwmd_free(result);
1705 return error;
1708 static gpg_error_t do_pwmd_open(pwm_t *pwm, const char *filename, int nb)
1710 char *result = NULL;
1711 char *password = NULL;
1712 char path[PATH_MAX];
1713 #ifdef WITH_PINENTRY
1714 int pin_try;
1715 #endif
1716 gpg_error_t rc;
1718 if (!pwm || !filename || !*filename)
1719 return GPG_ERR_INV_ARG;
1721 #ifdef WITH_PINENTRY
1722 pin_try = pwm->pinentry_tries - 1;
1723 #endif
1726 * Avoid calling pinentry if the password is cached on the server or if
1727 * this is a new file.
1729 rc = pwmd_command(pwm, &result, "GETCONFIG data_directory");
1731 if (rc)
1732 return rc;
1734 snprintf(path, sizeof(path), "%s/%s", result, filename);
1735 pwmd_free(result);
1737 if (access(path, R_OK) == -1) {
1738 if (errno == ENOENT)
1739 goto gotpassword;
1742 rc = pwmd_command(pwm, &result, "ISCACHED %s", filename);
1744 if (rc == EPWMD_CACHE_NOT_FOUND) {
1745 if (pwm->passfunc) {
1746 password = (char *)pwm->passfunc(pwm->passdata);
1747 goto gotpassword;
1750 #ifdef WITH_PINENTRY
1752 * Get the password from pinentry.
1754 if (pwm->use_pinentry) {
1756 * Nonblocking is wanted. fork() then return a file descriptor
1757 * that the client can use to read() from.
1759 if (nb) {
1760 int p[2];
1761 pid_t pid;
1762 pwmd_nb_status_t pw;
1764 if (pipe(p) == -1)
1765 return gpg_error_from_syserror();
1767 #ifdef WITH_LIBPTH
1768 pid = pth_fork();
1769 #else
1770 pid = fork();
1771 #endif
1773 switch (pid) {
1774 case 0:
1775 close(p[0]);
1776 pw.fd = p[0];
1778 if (pwm->pinentry_timeout > 0) {
1779 gpwm = pwm;
1780 gtimeout = pwm->pinentry_timeout;
1781 gelapsed = 0;
1784 getpin_nb_again:
1785 rc = getpin(pwm, &password, &pin_try, 0);
1787 if (rc) {
1788 getpin_nb_fail:
1789 if (pwm->pctx)
1790 pinentry_disconnect(pwm);
1792 if (gtimeout && gelapsed >= gtimeout)
1793 rc = GPG_ERR_TIMEOUT;
1795 pw.error = rc;
1796 #ifdef WITH_LIBPTH
1797 pth_write(p[1], &pw, sizeof(pw));
1798 #else
1799 write(p[1], &pw, sizeof(pw));
1800 #endif
1801 close(p[1]);
1802 _exit(1);
1806 * Don't count the time it takes to open the file
1807 * which may have many iterations.
1809 signal(SIGALRM, SIG_DFL);
1810 rc = do_open_command(pwm, filename, password);
1812 if (pwm->pinentry_timeout)
1813 signal(SIGALRM, catchsig);
1815 if (pwm->pctx && rc == EPWMD_BADKEY) {
1816 if (pin_try-- > 0)
1817 goto getpin_nb_again;
1819 goto getpin_nb_fail;
1822 pinentry_disconnect(pwm);
1823 pw.error = 0;
1824 #ifdef WITH_LIBPTH
1825 pth_write(p[1], &pw, sizeof(pw));
1826 #else
1827 write(p[1], &pw, sizeof(pw));
1828 #endif
1829 close(p[1]);
1830 _exit(0);
1831 break;
1832 case -1:
1833 rc = gpg_error_from_syserror();
1834 close(p[0]);
1835 close(p[1]);
1836 return rc;
1837 default:
1838 break;
1841 close(p[1]);
1842 pwm->nb_fd = p[0];
1843 return 0;
1846 else {
1847 #endif
1849 * Not using pinentry and the file was not found
1850 * in the cache.
1852 password = pwm->password;
1853 #ifdef WITH_PINENTRY
1855 #endif
1857 else if (rc)
1858 return rc;
1860 gotpassword:
1861 rc = do_open_command(pwm, filename, password);
1864 * Keep the user defined password set with pwmd_setopt(). The password may
1865 * be needed later (pwmd_save()) depending on the pwmd file cache settings.
1867 if (!pwm->passfunc && password && password != pwm->password)
1868 pwmd_free(password);
1870 #ifdef WITH_PINENTRY
1871 if (rc == EPWMD_BADKEY) {
1872 if (pin_try-- > 0 && !nb) {
1873 rc = pwmd_command(pwm, &result, "OPTION TITLE=%s",
1874 N_("Invalid passphrase, please try again."));
1876 if (rc)
1877 return rc;
1879 goto gotpassword;
1882 if (nb)
1883 pinentry_disconnect(pwm);
1885 return rc;
1887 #endif
1889 if (!rc) {
1890 if (pwm->filename)
1891 pwmd_free(pwm->filename);
1893 pwm->filename = pwmd_strdup(filename);
1896 return rc;
1899 gpg_error_t pwmd_open(pwm_t *pwm, const char *filename)
1901 return do_pwmd_open(pwm, filename, 0);
1904 gpg_error_t pwmd_open_async2(pwm_t *pwm, const char *filename)
1906 #ifndef WITH_PINENTRY
1907 return GPG_ERR_NOT_IMPLEMENTED;
1908 #else
1909 gpg_error_t rc;
1911 pwm->cmd = ASYNC_CMD_OPEN2;
1912 pwm->state = ASYNC_PROCESS;
1913 rc = do_pwmd_open(pwm, filename, 1);
1915 if (rc)
1916 reset_async(pwm, 1);
1918 return rc;
1919 #endif
1922 #ifdef WITH_PINENTRY
1923 static gpg_error_t do_save_getpin(pwm_t *pwm, char **password)
1925 int confirm = 0;
1926 gpg_error_t error;
1927 char *result = NULL;
1928 int pin_try = -1;
1930 again:
1931 error = getpin(pwm, &result, &pin_try, confirm ? 2 : 0);
1933 if (error) {
1934 if (pwm->pctx)
1935 pinentry_disconnect(pwm);
1937 if (*password)
1938 pwmd_free(*password);
1940 return error;
1943 if (!confirm++) {
1944 *password = result;
1945 goto again;
1948 if (strcmp(*password, result)) {
1949 pwmd_free(*password);
1950 pwmd_free(result);
1951 pinentry_disconnect(pwm);
1952 error = EPWMD_BADKEY;
1953 return error;
1956 pwmd_free(result);
1957 pinentry_disconnect(pwm);
1958 return 0;
1960 #endif
1962 static gpg_error_t do_save_command(pwm_t *pwm, char *password)
1964 char buf[ASSUAN_LINELENGTH];
1965 gpg_error_t error;
1966 char *result = NULL;
1968 snprintf(buf, sizeof(buf), "SAVE %s", password ? password : "");
1969 error = send_command(pwm, &result, buf);
1970 memset(&buf, 0, sizeof(buf));
1972 if (error && result)
1973 pwmd_free(result);
1975 return error;
1978 static gpg_error_t do_pwmd_save(pwm_t *pwm, int nb)
1980 char *result = NULL;
1981 char *password = NULL;
1982 gpg_error_t rc;
1984 if (!pwm)
1985 return GPG_ERR_INV_ARG;
1987 if (pwm->use_pinentry || pwm->passfunc) {
1988 rc = pwmd_command(pwm, &result, "ISCACHED %s", pwm->filename);
1990 if (rc == EPWMD_CACHE_NOT_FOUND) {
1991 if (pwm->passfunc)
1992 password = (char *)(*pwm->passfunc)(pwm->passdata);
1993 #ifdef WITH_PINENTRY
1994 else if (pwm->use_pinentry) {
1995 if (nb) {
1996 int p[2];
1997 pid_t pid;
1998 pwmd_nb_status_t pw;
2000 if (pipe(p) == -1)
2001 return gpg_error_from_syserror();
2003 #ifdef WITH_LIBPTH
2004 pid = pth_fork();
2005 #else
2006 pid = fork();
2007 #endif
2009 switch (pid) {
2010 case 0:
2011 close(p[0]);
2012 pw.fd = p[0];
2014 do {
2015 password = NULL;
2016 rc = do_save_getpin(pwm, &password);
2017 } while (rc == EPWMD_BADKEY);
2019 if (rc) {
2020 if (pwm->pctx)
2021 pinentry_disconnect(pwm);
2023 pw.error = rc;
2024 #ifdef WITH_LIBPTH
2025 pth_write(p[1], &pw, sizeof(pw));
2026 #else
2027 write(p[1], &pw, sizeof(pw));
2028 #endif
2029 close(p[1]);
2030 _exit(1);
2033 rc = do_save_command(pwm, password);
2034 pinentry_disconnect(pwm);
2035 pw.error = rc;
2036 #ifdef WITH_LIBPTH
2037 pth_write(p[1], &pw, sizeof(pw));
2038 #else
2039 write(p[1], &pw, sizeof(pw));
2040 #endif
2041 close(p[1]);
2042 _exit(0);
2043 break;
2044 case -1:
2045 rc = gpg_error_from_syserror();
2046 close(p[0]);
2047 close(p[1]);
2048 return rc;
2049 default:
2050 break;
2053 close(p[1]);
2054 pwm->nb_fd = p[0];
2055 return 0;
2058 rc = do_save_getpin(pwm, &password);
2060 if (rc)
2061 return rc;
2063 #endif
2065 else {
2066 if (rc)
2067 return rc;
2070 else
2071 password = pwm->password;
2073 rc = do_save_command(pwm, password);
2075 if (!pwm->passfunc && password && password != pwm->password)
2076 pwmd_free(password);
2078 return rc;
2081 gpg_error_t pwmd_save_async2(pwm_t *pwm)
2083 #ifndef WITH_PINENTRY
2084 return GPG_ERR_NOT_IMPLEMENTED;
2085 #else
2086 gpg_error_t rc;
2088 pwm->cmd = ASYNC_CMD_SAVE2;
2089 pwm->state = ASYNC_PROCESS;
2090 rc = do_pwmd_save(pwm, 1);
2092 if (rc)
2093 reset_async(pwm, 0);
2095 return rc;
2096 #endif
2099 gpg_error_t pwmd_save(pwm_t *pwm)
2101 return do_pwmd_save(pwm, 0);
2104 gpg_error_t pwmd_setopt(pwm_t *pwm, pwmd_option_t opt, ...)
2106 va_list ap;
2107 int n = va_arg(ap, int);
2108 char *result;
2109 char *arg1;
2110 gpg_error_t error = 0;
2112 if (!pwm)
2113 return GPG_ERR_INV_ARG;
2115 va_start(ap, opt);
2117 switch (opt) {
2118 case PWMD_OPTION_STATUS_FUNC:
2119 pwm->status_func = va_arg(ap, pwmd_status_fn);
2120 break;
2121 case PWMD_OPTION_STATUS_DATA:
2122 pwm->status_data = va_arg(ap, void *);
2123 break;
2124 case PWMD_OPTION_PASSWORD_FUNC:
2125 pwm->passfunc = va_arg(ap, pwmd_password_fn);
2126 break;
2127 case PWMD_OPTION_PASSWORD_DATA:
2128 pwm->passdata = va_arg(ap, void *);
2129 break;
2130 case PWMD_OPTION_PASSWORD:
2131 arg1 = va_arg(ap, char *);
2133 if (pwm->password)
2134 pwmd_free(pwm->password);
2136 pwm->password = pwmd_strdup(arg1);
2137 break;
2138 case PWMD_OPTION_PINENTRY:
2139 n = va_arg(ap, int);
2141 if (n != 0 && n != 1) {
2142 va_end(ap);
2143 error = GPG_ERR_INV_VALUE;
2145 else {
2146 pwm->use_pinentry = n;
2147 error = pwmd_command(pwm, &result, "OPTION PINENTRY=%i",
2148 !pwm->use_pinentry);
2150 break;
2151 #ifdef WITH_PINENTRY
2152 case PWMD_OPTION_PINENTRY_TRIES:
2153 n = va_arg(ap, int);
2155 if (n <= 0) {
2156 va_end(ap);
2157 error = GPG_ERR_INV_VALUE;
2159 else
2160 pwm->pinentry_tries = n;
2161 break;
2162 #endif
2163 case PWMD_OPTION_PINENTRY_TIMEOUT:
2164 n = va_arg(ap, int);
2166 if (n < 0) {
2167 va_end(ap);
2168 error = GPG_ERR_INV_VALUE;
2170 else
2171 pwm->pinentry_timeout = n;
2173 if (!pwm->use_pinentry)
2174 error = pwmd_command(pwm, &result, "OPTION TIMEOUT=%i",
2175 pwm->pinentry_timeout);
2176 break;
2177 case PWMD_OPTION_PINENTRY_PATH:
2178 if (pwm->pinentry_path)
2179 pwmd_free(pwm->pinentry_path);
2181 pwm->pinentry_path = pwmd_strdup(va_arg(ap, char *));
2183 if (!pwm->use_pinentry)
2184 error = pwmd_command(pwm, &result, "OPTION PATH=%s",
2185 pwm->pinentry_path);
2186 break;
2187 case PWMD_OPTION_PINENTRY_TTY:
2188 if (pwm->pinentry_tty)
2189 pwmd_free(pwm->pinentry_tty);
2191 pwm->pinentry_tty = pwmd_strdup(va_arg(ap, char *));
2193 if (!pwm->use_pinentry)
2194 error = pwmd_command(pwm, &result, "OPTION TTY=%s",
2195 pwm->pinentry_tty);
2196 break;
2197 case PWMD_OPTION_PINENTRY_DISPLAY:
2198 if (pwm->pinentry_display)
2199 pwmd_free(pwm->pinentry_display);
2201 pwm->pinentry_display = pwmd_strdup(va_arg(ap, char *));
2203 if (!pwm->use_pinentry)
2204 error = pwmd_command(pwm, &result, "OPTION DISPLAY=%s",
2205 pwm->pinentry_display);
2206 break;
2207 case PWMD_OPTION_PINENTRY_TERM:
2208 if (pwm->pinentry_term)
2209 pwmd_free(pwm->pinentry_term);
2211 pwm->pinentry_term = pwmd_strdup(va_arg(ap, char *));
2213 if (!pwm->use_pinentry)
2214 error = pwmd_command(pwm, &result, "OPTION TTYTYPE=%s",
2215 pwm->pinentry_term);
2216 break;
2217 case PWMD_OPTION_PINENTRY_TITLE:
2218 if (pwm->title)
2219 pwmd_free(pwm->title);
2221 pwm->title = percent_escape(va_arg(ap, char *));
2223 if (!pwm->use_pinentry)
2224 error = pwmd_command(pwm, &result, "OPTION TITLE=%s",
2225 pwm->title);
2226 break;
2227 case PWMD_OPTION_PINENTRY_PROMPT:
2228 if (pwm->prompt)
2229 pwmd_free(pwm->prompt);
2231 pwm->prompt = percent_escape(va_arg(ap, char *));
2233 if (!pwm->use_pinentry)
2234 error = pwmd_command(pwm, &result, "OPTION PROMPT=%s",
2235 pwm->prompt);
2236 break;
2237 case PWMD_OPTION_PINENTRY_DESC:
2238 if (pwm->desc)
2239 pwmd_free(pwm->desc);
2241 pwm->desc = percent_escape(va_arg(ap, char *));
2243 if (!pwm->use_pinentry)
2244 error = pwmd_command(pwm, &result, "OPTION DESC=%s",
2245 pwm->desc);
2246 break;
2247 case PWMD_OPTION_PINENTRY_LC_CTYPE:
2248 if (pwm->lcctype)
2249 pwmd_free(pwm->lcctype);
2251 pwm->lcctype = pwmd_strdup(va_arg(ap, char *));
2253 if (!pwm->use_pinentry)
2254 error = pwmd_command(pwm, &result, "OPTION LC_CTYPE=%s",
2255 pwm->lcctype);
2256 break;
2257 case PWMD_OPTION_PINENTRY_LC_MESSAGES:
2258 if (pwm->lcmessages)
2259 pwmd_free(pwm->lcmessages);
2261 pwm->lcmessages = pwmd_strdup(va_arg(ap, char *));
2263 if (!pwm->use_pinentry)
2264 error = pwmd_command(pwm, &result, "OPTION LC_MESSAGES=%s",
2265 pwm->lcmessages);
2266 break;
2267 default:
2268 error = GPG_ERR_NOT_IMPLEMENTED;
2269 break;
2272 va_end(ap);
2273 return error;
2276 gpg_error_t pwmd_get_fd(pwm_t *pwm, int *fd)
2278 if (!pwm)
2279 return GPG_ERR_INV_ARG;
2281 if (pwm->fd == -1)
2282 return GPG_ERR_INV_STATE;
2284 *fd = pwm->fd;
2285 return 0;
2288 gpg_error_t pwmd_get_fd2(pwm_t *pwm, int *fd)
2290 #ifndef WITH_PINENTRY
2291 return GPG_ERR_NOT_IMPLEMENTED;
2292 #else
2293 if (!pwm)
2294 return GPG_ERR_INV_ARG;
2296 if (pwm->nb_fd == -1)
2297 return GPG_ERR_INV_STATE;
2299 *fd = pwm->nb_fd;
2300 return 0;
2301 #endif
2304 pwm_t *pwmd_new(const char *name)
2306 pwm_t *h = pwmd_calloc(1, sizeof(pwm_t));
2308 if (!h)
2309 return NULL;
2311 if (name) {
2312 h->name = pwmd_strdup(name);
2314 if (!h->name) {
2315 pwmd_free(h);
2316 return NULL;
2320 h->fd = -1;
2321 #ifdef WITH_PINENTRY
2322 h->nb_fd = -1;
2323 #endif
2324 return h;
2327 void pwmd_free(void *ptr)
2329 xfree(ptr);
2332 void *pwmd_malloc(size_t size)
2334 return xmalloc(size);
2337 void *pwmd_calloc(size_t nmemb, size_t size)
2339 return xcalloc(nmemb, size);
2342 void *pwmd_realloc(void *ptr, size_t size)
2344 return xrealloc(ptr, size);
2347 char *pwmd_strdup(const char *str)
2349 return xstrdup(str);