2 * Unix Pageant, more or less similar to ssh-agent.
12 #include <sys/types.h>
17 #define PUTTY_DO_GLOBALS /* actually _define_ globals */
23 SockAddr unix_sock_addr(const char *path);
24 Socket new_unix_listener(SockAddr listenaddr, Plug plug);
26 void fatalbox(char *p, ...)
29 fprintf(stderr, "FATAL ERROR: ");
31 vfprintf(stderr, p, ap);
36 void modalfatalbox(char *p, ...)
39 fprintf(stderr, "FATAL ERROR: ");
41 vfprintf(stderr, p, ap);
46 void nonfatal(char *p, ...)
49 fprintf(stderr, "ERROR: ");
51 vfprintf(stderr, p, ap);
55 void connection_fatal(void *frontend, char *p, ...)
58 fprintf(stderr, "FATAL ERROR: ");
60 vfprintf(stderr, p, ap);
65 void cmdline_error(char *p, ...)
68 fprintf(stderr, "pageant: ");
70 vfprintf(stderr, p, ap);
76 FILE *pageant_logfp = NULL;
77 void pageant_log(void *ctx, const char *fmt, va_list ap)
82 fprintf(pageant_logfp, "pageant: ");
83 vfprintf(pageant_logfp, fmt, ap);
84 fprintf(pageant_logfp, "\n");
88 * In Pageant our selects are synchronous, so these functions are
91 int uxsel_input_add(int fd, int rwx) { return 0; }
92 void uxsel_input_remove(int id) { }
97 void random_save_seed(void) {}
98 void random_destroy_seed(void) {}
99 void noise_ultralight(unsigned long data) {}
100 char *platform_default_s(const char *name) { return NULL; }
101 int platform_default_i(const char *name, int def) { return def; }
102 FontSpec *platform_default_fontspec(const char *name) { return fontspec_new(""); }
103 Filename *platform_default_filename(const char *name) { return filename_from_str(""); }
104 char *x_get_default(const char *key) { return NULL; }
105 void log_eventlog(void *handle, const char *event) {}
108 * Short description of parameters.
110 static void usage(void)
112 printf("Pageant: SSH agent\n");
118 static void version(void)
120 printf("pageant: %s\n", ver);
124 void keylist_update(void)
126 /* Nothing needs doing in Unix Pageant */
129 #define PAGEANT_DIR_PREFIX "/tmp/pageant"
131 const char *const appname = "Pageant";
133 static int time_to_die = FALSE;
135 /* Stub functions to permit linking against x11fwd.c. These never get
136 * used, because in LIFE_X11 mode we connect to the X server using a
137 * straightforward Socket and don't try to create an ersatz SSH
139 int sshfwd_write(struct ssh_channel *c, char *data, int len) { return 0; }
140 void sshfwd_write_eof(struct ssh_channel *c) { }
141 void sshfwd_unclean_close(struct ssh_channel *c, const char *err) { }
142 void sshfwd_unthrottle(struct ssh_channel *c, int bufsize) {}
143 Conf *sshfwd_get_conf(struct ssh_channel *c) { return NULL; }
144 void sshfwd_x11_sharing_handover(struct ssh_channel *c,
145 void *share_cs, void *share_chan,
146 const char *peer_addr, int peer_port,
147 int endian, int protomajor, int protominor,
148 const void *initial_data, int initial_len) {}
149 void sshfwd_x11_is_local(struct ssh_channel *c) {}
152 * These functions are part of the plug for our connection to the X
153 * display, so they do get called. They needn't actually do anything,
154 * except that x11_closing has to signal back to the main loop that
155 * it's time to terminate.
157 static void x11_log(Plug p, int type, SockAddr addr, int port,
158 const char *error_msg, int error_code) {}
159 static int x11_receive(Plug plug, int urgent, char *data, int len) {return 0;}
160 static void x11_sent(Plug plug, int bufsize) {}
161 static int x11_closing(Plug plug, const char *error_msg, int error_code,
167 struct X11Connection {
168 const struct plug_function_table *fn;
172 void pageant_print_env(int pid)
174 printf("SSH_AUTH_SOCK=%s; export SSH_AUTH_SOCK;\n"
175 "SSH_AGENT_PID=%d; export SSH_AGENT_PID;\n",
176 socketname, (int)pid);
179 void pageant_fork_and_print_env(int retain_tty)
185 } else if (pid != 0) {
186 pageant_print_env(pid);
191 * Having forked off, we now daemonise ourselves as best we can.
192 * It's good practice in general to setsid() ourself out of any
193 * process group we didn't want to be part of, and to chdir("/")
194 * to avoid holding any directories open that we don't need in
195 * case someone wants to umount them; also, we should definitely
196 * close standard output (because it will very likely be pointing
197 * at a pipe from which some parent process is trying to read our
198 * environment variable dump, so if we hold open another copy of
199 * it then that process will never finish reading). We close
200 * standard input too on general principles, but not standard
201 * error, since we might need to shout a panicky error message
204 if (chdir("/") < 0) {
205 /* should there be an error condition, nothing we can do about
211 /* Get out of our previous process group, to avoid being
212 * blasted by passing signals. But keep our controlling tty,
213 * so we can keep checking to see if we still have one. */
216 /* Do that, but also leave our entire session and detach from
217 * the controlling tty (if any). */
224 void sigchld(int signum)
226 if (write(signalpipe[1], "x", 1) <= 0)
227 /* not much we can do about it */;
230 #define TTY_LIFE_POLL_INTERVAL (TICKSPERSEC * 30)
231 void *dummy_timer_ctx;
232 static void tty_life_timer(void *ctx, unsigned long now)
234 schedule_timer(TTY_LIFE_POLL_INTERVAL, tty_life_timer, &dummy_timer_ctx);
241 KEYACT_CLIENT_DEL_ALL,
243 KEYACT_CLIENT_PUBLIC_OPENSSH,
246 struct cmdline_key_action {
247 struct cmdline_key_action *next;
249 const char *filename;
252 int is_agent_action(keyact action)
254 return action == KEYACT_AGENT_LOAD;
257 struct cmdline_key_action *keyact_head = NULL, *keyact_tail = NULL;
259 void add_keyact(keyact action, const char *filename)
261 struct cmdline_key_action *a = snew(struct cmdline_key_action);
263 a->filename = filename;
266 keyact_tail->next = a;
272 int have_controlling_tty(void)
274 int fd = open("/dev/tty", O_RDONLY);
276 if (errno != ENXIO) {
277 perror("/dev/tty: open");
287 char **exec_args = NULL;
289 LIFE_UNSPEC, LIFE_X11, LIFE_TTY, LIFE_DEBUG, LIFE_PERM, LIFE_EXEC
290 } life = LIFE_UNSPEC;
291 const char *display = NULL;
293 static char *askpass(const char *comment)
295 if (have_controlling_tty()) {
297 prompts_t *p = new_prompts(NULL);
298 p->to_server = FALSE;
299 p->name = dupstr("Pageant passphrase prompt");
301 dupprintf("Enter passphrase to load key '%s': ", comment),
303 ret = console_get_userpass_input(p, NULL, 0);
307 perror("pageant: unable to read passphrase");
311 char *passphrase = dupstr(p->prompts[0]->result);
315 } else if (display) {
316 char *prompt, *passphrase;
320 char *gtk_askpass_main(const char *display, const char *wintitle,
321 const char *prompt, int *success);
323 prompt = dupprintf("Enter passphrase to load key '%s': ", comment);
324 passphrase = gtk_askpass_main(display,
325 "Pageant passphrase prompt",
329 /* return value is error message */
330 fprintf(stderr, "%s\n", passphrase);
336 fprintf(stderr, "no way to read a passphrase without tty or "
342 static int unix_add_keyfile(const char *filename_str)
344 Filename *filename = filename_from_str(filename_str);
351 * Try without a passphrase.
353 status = pageant_add_keyfile(filename, NULL, &err);
354 if (status == PAGEANT_ACTION_OK) {
356 } else if (status == PAGEANT_ACTION_FAILURE) {
357 fprintf(stderr, "pageant: %s: %s\n", filename_str, err);
363 * And now try prompting for a passphrase.
366 char *passphrase = askpass(err);
372 status = pageant_add_keyfile(filename, passphrase, &err);
374 smemclr(passphrase, strlen(passphrase));
378 if (status == PAGEANT_ACTION_OK) {
380 } else if (status == PAGEANT_ACTION_FAILURE) {
381 fprintf(stderr, "pageant: %s: %s\n", filename_str, err);
389 filename_free(filename);
393 void key_list_callback(void *ctx, const char *fingerprint,
394 const char *comment, struct pageant_pubkey *key)
396 printf("%s %s\n", fingerprint, comment);
399 struct key_find_ctx {
401 int match_fp, match_comment;
402 struct pageant_pubkey *found;
406 int match_fingerprint_string(const char *string, const char *fingerprint)
410 /* Find the hash in the fingerprint string. It'll be the word at the end. */
411 hash = strrchr(fingerprint, ' ');
415 /* Now see if the search string is a prefix of the full hash,
416 * neglecting colons and case differences. */
418 while (*string == ':') string++;
419 while (*hash == ':') hash++;
422 if (tolower((unsigned char)*string) != tolower((unsigned char)*hash))
429 void key_find_callback(void *vctx, const char *fingerprint,
430 const char *comment, struct pageant_pubkey *key)
432 struct key_find_ctx *ctx = (struct key_find_ctx *)vctx;
434 if ((ctx->match_comment && !strcmp(ctx->string, comment)) ||
435 (ctx->match_fp && match_fingerprint_string(ctx->string, fingerprint)))
438 ctx->found = pageant_pubkey_copy(key);
443 struct pageant_pubkey *find_key(const char *string, char **retstr)
445 struct key_find_ctx actx, *ctx = &actx;
446 struct pageant_pubkey key_in, *key_ret;
447 int try_file = TRUE, try_fp = TRUE, try_comment = TRUE;
448 int file_errors = FALSE;
451 * Trim off disambiguating prefixes telling us how to interpret
452 * the provided string.
454 if (!strncmp(string, "file:", 5)) {
456 try_fp = try_comment = FALSE;
457 file_errors = TRUE; /* also report failure to load the file */
458 } else if (!strncmp(string, "comment:", 8)) {
460 try_file = try_fp = FALSE;
461 } else if (!strncmp(string, "fp:", 3)) {
463 try_file = try_comment = FALSE;
464 } else if (!strncmp(string, "fingerprint:", 12)) {
466 try_file = try_comment = FALSE;
470 * Try interpreting the string as a key file name.
473 Filename *fn = filename_from_str(string);
474 int keytype = key_type(fn);
475 if (keytype == SSH_KEYTYPE_SSH1 ||
476 keytype == SSH_KEYTYPE_SSH1_PUBLIC) {
479 if (!rsakey_pubblob(fn, &key_in.blob, &key_in.bloblen,
482 *retstr = dupprintf("unable to load file '%s': %s",
490 * If we've successfully loaded the file, stop here - we
491 * already have a key blob and need not go to the agent to
494 key_in.ssh_version = 1;
495 key_ret = pageant_pubkey_copy(&key_in);
499 } else if (keytype == SSH_KEYTYPE_SSH2 ||
500 keytype == SSH_KEYTYPE_SSH2_PUBLIC_RFC4716 ||
501 keytype == SSH_KEYTYPE_SSH2_PUBLIC_OPENSSH) {
504 if ((key_in.blob = ssh2_userkey_loadpub(fn, NULL,
506 NULL, &error)) == NULL) {
508 *retstr = dupprintf("unable to load file '%s': %s",
516 * If we've successfully loaded the file, stop here - we
517 * already have a key blob and need not go to the agent to
520 key_in.ssh_version = 2;
521 key_ret = pageant_pubkey_copy(&key_in);
527 *retstr = dupprintf("unable to load key file '%s': %s",
528 string, key_type_to_str(keytype));
537 * Failing that, go through the keys in the agent, and match
538 * against fingerprints and comments as appropriate.
540 ctx->string = string;
541 ctx->match_fp = try_fp;
542 ctx->match_comment = try_comment;
545 if (pageant_enum_keys(key_find_callback, ctx, retstr) ==
546 PAGEANT_ACTION_FAILURE)
549 if (ctx->nfound == 0) {
550 *retstr = dupstr("no key matched");
553 } else if (ctx->nfound > 1) {
554 *retstr = dupstr("multiple keys matched");
556 pageant_pubkey_free(ctx->found);
564 void run_client(void)
566 const struct cmdline_key_action *act;
567 struct pageant_pubkey *key;
571 if (!agent_exists()) {
572 fprintf(stderr, "pageant: no agent running to talk to\n");
576 for (act = keyact_head; act; act = act->next) {
577 switch (act->action) {
578 case KEYACT_CLIENT_ADD:
579 if (!unix_add_keyfile(act->filename))
582 case KEYACT_CLIENT_LIST:
583 if (pageant_enum_keys(key_list_callback, NULL, &retstr) ==
584 PAGEANT_ACTION_FAILURE) {
585 fprintf(stderr, "pageant: listing keys: %s\n", retstr);
590 case KEYACT_CLIENT_DEL:
592 if (!(key = find_key(act->filename, &retstr)) ||
593 pageant_delete_key(key, &retstr) == PAGEANT_ACTION_FAILURE) {
594 fprintf(stderr, "pageant: deleting key '%s': %s\n",
595 act->filename, retstr);
600 pageant_pubkey_free(key);
602 case KEYACT_CLIENT_PUBLIC_OPENSSH:
603 case KEYACT_CLIENT_PUBLIC:
605 if (!(key = find_key(act->filename, &retstr))) {
606 fprintf(stderr, "pageant: finding key '%s': %s\n",
607 act->filename, retstr);
611 FILE *fp = stdout; /* FIXME: add a -o option? */
613 if (key->ssh_version == 1) {
615 memset(&rkey, 0, sizeof(rkey));
616 rkey.comment = dupstr(key->comment);
617 makekey(key->blob, key->bloblen, &rkey, NULL, 0);
618 ssh1_write_pubkey(fp, &rkey);
621 ssh2_write_pubkey(fp, key->comment, key->blob,key->bloblen,
622 (act->action == KEYACT_CLIENT_PUBLIC ?
623 SSH_KEYTYPE_SSH2_PUBLIC_RFC4716 :
624 SSH_KEYTYPE_SSH2_PUBLIC_OPENSSH));
626 pageant_pubkey_free(key);
629 case KEYACT_CLIENT_DEL_ALL:
630 if (pageant_delete_all_keys(&retstr) == PAGEANT_ACTION_FAILURE) {
631 fprintf(stderr, "pageant: deleting all keys: %s\n", retstr);
637 assert(0 && "Invalid client action found");
648 char *username, *socketdir;
649 struct pageant_listen_state *pl;
654 int i, fdcount, fdsize, fdstate;
655 int termination_pid = -1;
658 const struct cmdline_key_action *act;
661 fdcount = fdsize = 0;
666 * Start by loading any keys provided on the command line.
668 for (act = keyact_head; act; act = act->next) {
669 assert(act->action == KEYACT_AGENT_LOAD);
670 if (!unix_add_keyfile(act->filename))
677 * Set up a listening socket and run Pageant on it.
679 username = get_username();
680 socketdir = dupprintf("%s.%s", PAGEANT_DIR_PREFIX, username);
682 assert(*socketdir == '/');
683 if ((err = make_dir_and_check_ours(socketdir)) != NULL) {
684 fprintf(stderr, "pageant: %s: %s\n", socketdir, err);
687 socketname = dupprintf("%s/pageant.%d", socketdir, (int)getpid());
688 pl = pageant_listener_new();
689 sock = new_unix_listener(unix_sock_addr(socketname), (Plug)pl);
690 if ((err = sk_socket_error(sock)) != NULL) {
691 fprintf(stderr, "pageant: %s: %s\n", socketname, err);
694 pageant_listener_got_socket(pl, sock);
697 conf_set_int(conf, CONF_proxy_type, PROXY_NONE);
700 * Lifetime preparations.
702 signalpipe[0] = signalpipe[1] = -1;
703 if (life == LIFE_X11) {
704 struct X11Display *disp;
708 struct X11Connection *conn;
710 static const struct plug_function_table fn_table = {
719 fprintf(stderr, "pageant: no DISPLAY for -X mode\n");
722 disp = x11_setup_display(display, conf);
724 conn = snew(struct X11Connection);
725 conn->fn = &fn_table;
726 s = new_connection(sk_addr_dup(disp->addr),
727 disp->realhost, disp->port,
728 0, 1, 0, 0, (Plug)conn, conf);
729 if ((err = sk_socket_error(s)) != NULL) {
730 fprintf(stderr, "pageant: unable to connect to X server: %s", err);
733 greeting = x11_make_greeting('B', 11, 0, disp->localauthproto,
735 disp->localauthdatalen,
736 NULL, 0, &greetinglen);
737 sk_write(s, greeting, greetinglen);
738 smemclr(greeting, greetinglen);
741 pageant_fork_and_print_env(FALSE);
742 } else if (life == LIFE_TTY) {
743 schedule_timer(TTY_LIFE_POLL_INTERVAL,
744 tty_life_timer, &dummy_timer_ctx);
745 pageant_fork_and_print_env(TRUE);
746 } else if (life == LIFE_PERM) {
747 pageant_fork_and_print_env(FALSE);
748 } else if (life == LIFE_DEBUG) {
749 pageant_print_env(getpid());
750 pageant_logfp = stdout;
751 } else if (life == LIFE_EXEC) {
757 * Set up the pipe we'll use to tell us about SIGCHLD.
759 if (pipe(signalpipe) < 0) {
763 putty_signal(SIGCHLD, sigchld);
769 } else if (pid == 0) {
770 setenv("SSH_AUTH_SOCK", socketname, TRUE);
771 setenv("SSH_AGENT_PID", dupprintf("%d", (int)agentpid), TRUE);
772 execvp(exec_args[0], exec_args);
776 termination_pid = pid;
781 * Now we've decided on our logging arrangements, pass them on to
784 pageant_listener_set_logfn(pl, NULL, pageant_logfp ? pageant_log : NULL);
786 now = GETTICKCOUNT();
788 while (!time_to_die) {
789 fd_set rset, wset, xset;
800 if (signalpipe[0] >= 0) {
801 FD_SET_MAX(signalpipe[0], maxfd, rset);
804 /* Count the currently active fds. */
806 for (fd = first_fd(&fdstate, &rwx); fd >= 0;
807 fd = next_fd(&fdstate, &rwx)) i++;
809 /* Expand the fdlist buffer if necessary. */
812 fdlist = sresize(fdlist, fdsize, int);
816 * Add all currently open fds to the select sets, and store
817 * them in fdlist as well.
820 for (fd = first_fd(&fdstate, &rwx); fd >= 0;
821 fd = next_fd(&fdstate, &rwx)) {
822 fdlist[fdcount++] = fd;
824 FD_SET_MAX(fd, maxfd, rset);
826 FD_SET_MAX(fd, maxfd, wset);
828 FD_SET_MAX(fd, maxfd, xset);
831 if (toplevel_callback_pending()) {
835 ret = select(maxfd, &rset, &wset, &xset, &tv);
836 } else if (run_timers(now, &next)) {
842 now = GETTICKCOUNT();
843 if (now - then > next - then)
847 tv.tv_sec = ticks / 1000;
848 tv.tv_usec = ticks % 1000 * 1000;
849 ret = select(maxfd, &rset, &wset, &xset, &tv);
853 now = GETTICKCOUNT();
855 ret = select(maxfd, &rset, &wset, &xset, NULL);
858 if (ret < 0 && errno == EINTR)
866 if (life == LIFE_TTY) {
868 * Every time we wake up (whether it was due to tty_timer
869 * elapsing or for any other reason), poll to see if we
870 * still have a controlling terminal. If we don't, then
871 * our containing tty session has ended, so it's time to
872 * clean up and leave.
874 if (!have_controlling_tty()) {
880 for (i = 0; i < fdcount; i++) {
883 * We must process exceptional notifications before
884 * ordinary readability ones, or we may go straight
885 * past the urgent marker.
887 if (FD_ISSET(fd, &xset))
888 select_result(fd, 4);
889 if (FD_ISSET(fd, &rset))
890 select_result(fd, 1);
891 if (FD_ISSET(fd, &wset))
892 select_result(fd, 2);
895 if (signalpipe[0] >= 0 && FD_ISSET(signalpipe[0], &rset)) {
897 if (read(signalpipe[0], c, 1) <= 0)
899 /* ignore its value; it'll be `x' */
903 pid = waitpid(-1, &status, WNOHANG);
906 if (pid == termination_pid)
911 run_toplevel_callbacks();
915 * When we come here, we're terminating, and should clean up our
916 * Unix socket file if possible.
918 if (unlink(socketname) < 0) {
919 fprintf(stderr, "pageant: %s: %s\n", socketname, strerror(errno));
924 int main(int argc, char **argv)
926 int doing_opts = TRUE;
927 keyact curr_keyact = KEYACT_AGENT_LOAD;
930 * Process the command line.
934 if (*p == '-' && doing_opts) {
935 if (!strcmp(p, "-V") || !strcmp(p, "--version")) {
937 } else if (!strcmp(p, "--help")) {
940 } else if (!strcmp(p, "-v")) {
941 pageant_logfp = stderr;
942 } else if (!strcmp(p, "-a")) {
943 curr_keyact = KEYACT_CLIENT_ADD;
944 } else if (!strcmp(p, "-d")) {
945 curr_keyact = KEYACT_CLIENT_DEL;
946 } else if (!strcmp(p, "-D")) {
947 add_keyact(KEYACT_CLIENT_DEL_ALL, NULL);
948 } else if (!strcmp(p, "-l")) {
949 add_keyact(KEYACT_CLIENT_LIST, NULL);
950 } else if (!strcmp(p, "--public")) {
951 curr_keyact = KEYACT_CLIENT_PUBLIC;
952 } else if (!strcmp(p, "--public-openssh")) {
953 curr_keyact = KEYACT_CLIENT_PUBLIC_OPENSSH;
954 } else if (!strcmp(p, "-X")) {
956 } else if (!strcmp(p, "-T")) {
958 } else if (!strcmp(p, "--debug")) {
960 } else if (!strcmp(p, "--permanent")) {
962 } else if (!strcmp(p, "--exec")) {
964 /* Now all subsequent arguments go to the exec command. */
967 argc = 0; /* force end of option processing */
969 fprintf(stderr, "pageant: expected a command "
973 } else if (!strcmp(p, "--")) {
978 * Non-option arguments (apart from those after --exec,
979 * which are treated specially above) are interpreted as
980 * the names of private key files to either add or delete
983 add_keyact(curr_keyact, p);
987 if (life == LIFE_EXEC && !exec_args) {
988 fprintf(stderr, "pageant: expected a command with --exec\n");
993 * Block SIGPIPE, so that we'll get EPIPE individually on
994 * particular network connections that go wrong.
996 putty_signal(SIGPIPE, SIG_IGN);
1002 display = getenv("DISPLAY");
1003 if (display && !*display)
1008 * Now distinguish our two main running modes. Either we're
1009 * actually starting up an agent, in which case we should have a
1010 * lifetime mode, and no key actions of KEYACT_CLIENT_* type; or
1011 * else we're contacting an existing agent to add or remove keys,
1012 * in which case we should have no lifetime mode, and no key
1013 * actions of KEYACT_AGENT_* type.
1016 int has_agent_actions = FALSE;
1017 int has_client_actions = FALSE;
1018 int has_lifetime = FALSE;
1019 const struct cmdline_key_action *act;
1021 for (act = keyact_head; act; act = act->next) {
1022 if (is_agent_action(act->action))
1023 has_agent_actions = TRUE;
1025 has_client_actions = TRUE;
1027 if (life != LIFE_UNSPEC)
1028 has_lifetime = TRUE;
1030 if (has_lifetime && has_client_actions) {
1031 fprintf(stderr, "pageant: client key actions (-a, -d, -D, -l, -L)"
1032 " do not go with an agent lifetime option\n");
1035 if (!has_lifetime && has_agent_actions) {
1036 fprintf(stderr, "pageant: expected an agent lifetime option with"
1037 " bare key file arguments\n");
1040 if (!has_lifetime && !has_client_actions) {
1041 fprintf(stderr, "pageant: expected an agent lifetime option"
1042 " or a client key action\n");
1048 } else if (has_client_actions) {