djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 1 | /* $OpenBSD: ssh-pkcs11-helper.c,v 1.22 2020/01/25 00:03:36 djm Exp $ */ |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 2 | /* |
| 3 | * Copyright (c) 2010 Markus Friedl. All rights reserved. |
| 4 | * |
| 5 | * Permission to use, copy, modify, and distribute this software for any |
| 6 | * purpose with or without fee is hereby granted, provided that the above |
| 7 | * copyright notice and this permission notice appear in all copies. |
| 8 | * |
| 9 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
| 10 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| 11 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
| 12 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| 13 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
| 14 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
| 15 | * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
| 16 | */ |
| 17 | |
Damien Miller | 8ad0fbd | 2010-02-12 09:49:06 +1100 | [diff] [blame] | 18 | #include "includes.h" |
| 19 | |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 20 | #include <sys/types.h> |
Damien Miller | 8ad0fbd | 2010-02-12 09:49:06 +1100 | [diff] [blame] | 21 | #ifdef HAVE_SYS_TIME_H |
| 22 | # include <sys/time.h> |
| 23 | #endif |
| 24 | |
| 25 | #include "openbsd-compat/sys-queue.h" |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 26 | |
djm@openbsd.org | be02d7c | 2019-09-06 04:53:27 +0000 | [diff] [blame] | 27 | #include <stdlib.h> |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 28 | #include <errno.h> |
Darren Tucker | 6fc7e1c | 2019-10-28 15:53:25 +1100 | [diff] [blame] | 29 | #ifdef HAVE_POLL_H |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 30 | #include <poll.h> |
Darren Tucker | 6fc7e1c | 2019-10-28 15:53:25 +1100 | [diff] [blame] | 31 | #endif |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 32 | #include <stdarg.h> |
| 33 | #include <string.h> |
| 34 | #include <unistd.h> |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 35 | |
| 36 | #include "xmalloc.h" |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 37 | #include "sshbuf.h" |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 38 | #include "log.h" |
| 39 | #include "misc.h" |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 40 | #include "sshkey.h" |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 41 | #include "authfd.h" |
| 42 | #include "ssh-pkcs11.h" |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 43 | #include "ssherr.h" |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 44 | |
Damien Miller | 47f9a41 | 2010-03-14 08:37:49 +1100 | [diff] [blame] | 45 | #ifdef ENABLE_PKCS11 |
| 46 | |
djm@openbsd.org | 670104b | 2019-09-06 05:23:55 +0000 | [diff] [blame] | 47 | #ifdef WITH_OPENSSL |
| 48 | |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 49 | /* borrows code from sftp-server and ssh-agent */ |
| 50 | |
| 51 | struct pkcs11_keyinfo { |
markus@openbsd.org | 54d90ac | 2017-05-30 08:52:19 +0000 | [diff] [blame] | 52 | struct sshkey *key; |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 53 | char *providername, *label; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 54 | TAILQ_ENTRY(pkcs11_keyinfo) next; |
| 55 | }; |
| 56 | |
| 57 | TAILQ_HEAD(, pkcs11_keyinfo) pkcs11_keylist; |
| 58 | |
| 59 | #define MAX_MSG_LENGTH 10240 /*XXX*/ |
| 60 | |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 61 | /* input and output queue */ |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 62 | struct sshbuf *iqueue; |
| 63 | struct sshbuf *oqueue; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 64 | |
| 65 | static void |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 66 | add_key(struct sshkey *k, char *name, char *label) |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 67 | { |
| 68 | struct pkcs11_keyinfo *ki; |
| 69 | |
| 70 | ki = xcalloc(1, sizeof(*ki)); |
| 71 | ki->providername = xstrdup(name); |
| 72 | ki->key = k; |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 73 | ki->label = xstrdup(label); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 74 | TAILQ_INSERT_TAIL(&pkcs11_keylist, ki, next); |
| 75 | } |
| 76 | |
| 77 | static void |
| 78 | del_keys_by_name(char *name) |
| 79 | { |
| 80 | struct pkcs11_keyinfo *ki, *nxt; |
| 81 | |
| 82 | for (ki = TAILQ_FIRST(&pkcs11_keylist); ki; ki = nxt) { |
| 83 | nxt = TAILQ_NEXT(ki, next); |
| 84 | if (!strcmp(ki->providername, name)) { |
| 85 | TAILQ_REMOVE(&pkcs11_keylist, ki, next); |
Darren Tucker | a627d42 | 2013-06-02 07:31:17 +1000 | [diff] [blame] | 86 | free(ki->providername); |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 87 | free(ki->label); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 88 | sshkey_free(ki->key); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 89 | free(ki); |
| 90 | } |
| 91 | } |
| 92 | } |
| 93 | |
| 94 | /* lookup matching 'private' key */ |
markus@openbsd.org | 54d90ac | 2017-05-30 08:52:19 +0000 | [diff] [blame] | 95 | static struct sshkey * |
| 96 | lookup_key(struct sshkey *k) |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 97 | { |
| 98 | struct pkcs11_keyinfo *ki; |
| 99 | |
| 100 | TAILQ_FOREACH(ki, &pkcs11_keylist, next) { |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 101 | debug("check %p %s %s", ki, ki->providername, ki->label); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 102 | if (sshkey_equal(k, ki->key)) |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 103 | return (ki->key); |
| 104 | } |
| 105 | return (NULL); |
| 106 | } |
| 107 | |
| 108 | static void |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 109 | send_msg(struct sshbuf *m) |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 110 | { |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 111 | int r; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 112 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 113 | if ((r = sshbuf_put_stringb(oqueue, m)) != 0) |
| 114 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 115 | } |
| 116 | |
| 117 | static void |
| 118 | process_add(void) |
| 119 | { |
| 120 | char *name, *pin; |
djm@openbsd.org | 93f0210 | 2019-01-20 22:51:37 +0000 | [diff] [blame] | 121 | struct sshkey **keys = NULL; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 122 | int r, i, nkeys; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 123 | u_char *blob; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 124 | size_t blen; |
| 125 | struct sshbuf *msg; |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 126 | char **labels = NULL; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 127 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 128 | if ((msg = sshbuf_new()) == NULL) |
| 129 | fatal("%s: sshbuf_new failed", __func__); |
| 130 | if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0 || |
| 131 | (r = sshbuf_get_cstring(iqueue, &pin, NULL)) != 0) |
| 132 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 133 | if ((nkeys = pkcs11_add_provider(name, pin, &keys, &labels)) > 0) { |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 134 | if ((r = sshbuf_put_u8(msg, |
| 135 | SSH2_AGENT_IDENTITIES_ANSWER)) != 0 || |
| 136 | (r = sshbuf_put_u32(msg, nkeys)) != 0) |
| 137 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 138 | for (i = 0; i < nkeys; i++) { |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 139 | if ((r = sshkey_to_blob(keys[i], &blob, &blen)) != 0) { |
| 140 | debug("%s: sshkey_to_blob: %s", |
| 141 | __func__, ssh_err(r)); |
Damien Miller | f1e44ea | 2013-12-05 10:23:21 +1100 | [diff] [blame] | 142 | continue; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 143 | } |
| 144 | if ((r = sshbuf_put_string(msg, blob, blen)) != 0 || |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 145 | (r = sshbuf_put_cstring(msg, labels[i])) != 0) |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 146 | fatal("%s: buffer error: %s", |
| 147 | __func__, ssh_err(r)); |
Darren Tucker | a627d42 | 2013-06-02 07:31:17 +1000 | [diff] [blame] | 148 | free(blob); |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 149 | add_key(keys[i], name, labels[i]); |
| 150 | free(labels[i]); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 151 | } |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 152 | } else { |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 153 | if ((r = sshbuf_put_u8(msg, SSH_AGENT_FAILURE)) != 0) |
| 154 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
djm@openbsd.org | 93f0210 | 2019-01-20 22:51:37 +0000 | [diff] [blame] | 155 | if ((r = sshbuf_put_u32(msg, -nkeys)) != 0) |
| 156 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 157 | } |
djm@openbsd.org | 89a8d45 | 2020-01-25 00:03:36 +0000 | [diff] [blame] | 158 | free(labels); |
| 159 | free(keys); /* keys themselves are transferred to pkcs11_keylist */ |
Darren Tucker | a627d42 | 2013-06-02 07:31:17 +1000 | [diff] [blame] | 160 | free(pin); |
| 161 | free(name); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 162 | send_msg(msg); |
| 163 | sshbuf_free(msg); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 164 | } |
| 165 | |
| 166 | static void |
| 167 | process_del(void) |
| 168 | { |
| 169 | char *name, *pin; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 170 | struct sshbuf *msg; |
| 171 | int r; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 172 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 173 | if ((msg = sshbuf_new()) == NULL) |
| 174 | fatal("%s: sshbuf_new failed", __func__); |
| 175 | if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0 || |
| 176 | (r = sshbuf_get_cstring(iqueue, &pin, NULL)) != 0) |
| 177 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 178 | del_keys_by_name(name); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 179 | if ((r = sshbuf_put_u8(msg, pkcs11_del_provider(name) == 0 ? |
| 180 | SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE)) != 0) |
| 181 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Darren Tucker | a627d42 | 2013-06-02 07:31:17 +1000 | [diff] [blame] | 182 | free(pin); |
| 183 | free(name); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 184 | send_msg(msg); |
| 185 | sshbuf_free(msg); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 186 | } |
| 187 | |
| 188 | static void |
| 189 | process_sign(void) |
| 190 | { |
| 191 | u_char *blob, *data, *signature = NULL; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 192 | size_t blen, dlen, slen = 0; |
| 193 | int r, ok = -1; |
markus@openbsd.org | 54d90ac | 2017-05-30 08:52:19 +0000 | [diff] [blame] | 194 | struct sshkey *key, *found; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 195 | struct sshbuf *msg; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 196 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 197 | /* XXX support SHA2 signature flags */ |
| 198 | if ((r = sshbuf_get_string(iqueue, &blob, &blen)) != 0 || |
| 199 | (r = sshbuf_get_string(iqueue, &data, &dlen)) != 0 || |
| 200 | (r = sshbuf_get_u32(iqueue, NULL)) != 0) |
| 201 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 202 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 203 | if ((r = sshkey_from_blob(blob, blen, &key)) != 0) |
| 204 | error("%s: sshkey_from_blob: %s", __func__, ssh_err(r)); |
| 205 | else { |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 206 | if ((found = lookup_key(key)) != NULL) { |
Damien Miller | 8668706 | 2014-07-02 15:28:02 +1000 | [diff] [blame] | 207 | #ifdef WITH_OPENSSL |
| 208 | int ret; |
| 209 | |
djm@openbsd.org | 93f0210 | 2019-01-20 22:51:37 +0000 | [diff] [blame] | 210 | if (key->type == KEY_RSA) { |
| 211 | slen = RSA_size(key->rsa); |
| 212 | signature = xmalloc(slen); |
| 213 | ret = RSA_private_encrypt(dlen, data, signature, |
| 214 | found->rsa, RSA_PKCS1_PADDING); |
| 215 | if (ret != -1) { |
| 216 | slen = ret; |
| 217 | ok = 0; |
| 218 | } |
Darren Tucker | 97370f6 | 2019-05-17 10:54:51 +1000 | [diff] [blame] | 219 | #ifdef OPENSSL_HAS_ECC |
djm@openbsd.org | 93f0210 | 2019-01-20 22:51:37 +0000 | [diff] [blame] | 220 | } else if (key->type == KEY_ECDSA) { |
dtucker@openbsd.org | 5c8d14c | 2019-05-16 08:47:27 +0000 | [diff] [blame] | 221 | u_int xslen = ECDSA_size(key->ecdsa); |
| 222 | |
djm@openbsd.org | 93f0210 | 2019-01-20 22:51:37 +0000 | [diff] [blame] | 223 | signature = xmalloc(xslen); |
| 224 | /* "The parameter type is ignored." */ |
| 225 | ret = ECDSA_sign(-1, data, dlen, signature, |
| 226 | &xslen, found->ecdsa); |
| 227 | if (ret != 0) |
| 228 | ok = 0; |
| 229 | else |
| 230 | error("%s: ECDSA_sign" |
| 231 | " returns %d", __func__, ret); |
| 232 | slen = xslen; |
Darren Tucker | 97370f6 | 2019-05-17 10:54:51 +1000 | [diff] [blame] | 233 | #endif /* OPENSSL_HAS_ECC */ |
djm@openbsd.org | 93f0210 | 2019-01-20 22:51:37 +0000 | [diff] [blame] | 234 | } else |
| 235 | error("%s: don't know how to sign with key " |
| 236 | "type %d", __func__, (int)key->type); |
Damien Miller | 8668706 | 2014-07-02 15:28:02 +1000 | [diff] [blame] | 237 | #endif /* WITH_OPENSSL */ |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 238 | } |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 239 | sshkey_free(key); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 240 | } |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 241 | if ((msg = sshbuf_new()) == NULL) |
| 242 | fatal("%s: sshbuf_new failed", __func__); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 243 | if (ok == 0) { |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 244 | if ((r = sshbuf_put_u8(msg, SSH2_AGENT_SIGN_RESPONSE)) != 0 || |
| 245 | (r = sshbuf_put_string(msg, signature, slen)) != 0) |
| 246 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 247 | } else { |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 248 | if ((r = sshbuf_put_u8(msg, SSH2_AGENT_FAILURE)) != 0) |
| 249 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 250 | } |
Darren Tucker | a627d42 | 2013-06-02 07:31:17 +1000 | [diff] [blame] | 251 | free(data); |
| 252 | free(blob); |
| 253 | free(signature); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 254 | send_msg(msg); |
| 255 | sshbuf_free(msg); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 256 | } |
| 257 | |
| 258 | static void |
| 259 | process(void) |
| 260 | { |
| 261 | u_int msg_len; |
| 262 | u_int buf_len; |
| 263 | u_int consumed; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 264 | u_char type; |
| 265 | const u_char *cp; |
| 266 | int r; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 267 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 268 | buf_len = sshbuf_len(iqueue); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 269 | if (buf_len < 5) |
| 270 | return; /* Incomplete message. */ |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 271 | cp = sshbuf_ptr(iqueue); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 272 | msg_len = get_u32(cp); |
| 273 | if (msg_len > MAX_MSG_LENGTH) { |
| 274 | error("bad message len %d", msg_len); |
| 275 | cleanup_exit(11); |
| 276 | } |
| 277 | if (buf_len < msg_len + 4) |
| 278 | return; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 279 | if ((r = sshbuf_consume(iqueue, 4)) != 0 || |
| 280 | (r = sshbuf_get_u8(iqueue, &type)) != 0) |
| 281 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 282 | buf_len -= 4; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 283 | switch (type) { |
| 284 | case SSH_AGENTC_ADD_SMARTCARD_KEY: |
| 285 | debug("process_add"); |
| 286 | process_add(); |
| 287 | break; |
| 288 | case SSH_AGENTC_REMOVE_SMARTCARD_KEY: |
| 289 | debug("process_del"); |
| 290 | process_del(); |
| 291 | break; |
| 292 | case SSH2_AGENTC_SIGN_REQUEST: |
| 293 | debug("process_sign"); |
| 294 | process_sign(); |
| 295 | break; |
| 296 | default: |
| 297 | error("Unknown message %d", type); |
| 298 | break; |
| 299 | } |
| 300 | /* discard the remaining bytes from the current packet */ |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 301 | if (buf_len < sshbuf_len(iqueue)) { |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 302 | error("iqueue grew unexpectedly"); |
| 303 | cleanup_exit(255); |
| 304 | } |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 305 | consumed = buf_len - sshbuf_len(iqueue); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 306 | if (msg_len < consumed) { |
| 307 | error("msg_len %d < consumed %d", msg_len, consumed); |
| 308 | cleanup_exit(255); |
| 309 | } |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 310 | if (msg_len > consumed) { |
| 311 | if ((r = sshbuf_consume(iqueue, msg_len - consumed)) != 0) |
| 312 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
| 313 | } |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 314 | } |
| 315 | |
| 316 | void |
| 317 | cleanup_exit(int i) |
| 318 | { |
| 319 | /* XXX */ |
| 320 | _exit(i); |
| 321 | } |
| 322 | |
djm@openbsd.org | c7670b0 | 2019-01-21 12:53:35 +0000 | [diff] [blame] | 323 | |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 324 | int |
| 325 | main(int argc, char **argv) |
| 326 | { |
djm@openbsd.org | c7670b0 | 2019-01-21 12:53:35 +0000 | [diff] [blame] | 327 | int r, ch, in, out, max, log_stderr = 0; |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 328 | ssize_t len; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 329 | SyslogFacility log_facility = SYSLOG_FACILITY_AUTH; |
| 330 | LogLevel log_level = SYSLOG_LEVEL_ERROR; |
| 331 | char buf[4*4096]; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 332 | extern char *__progname; |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 333 | struct pollfd pfd[2]; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 334 | |
Damien Miller | dfa4156 | 2010-02-12 10:06:28 +1100 | [diff] [blame] | 335 | __progname = ssh_get_progname(argv[0]); |
djm@openbsd.org | c7670b0 | 2019-01-21 12:53:35 +0000 | [diff] [blame] | 336 | seed_rng(); |
| 337 | TAILQ_INIT(&pkcs11_keylist); |
Damien Miller | dfa4156 | 2010-02-12 10:06:28 +1100 | [diff] [blame] | 338 | |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 339 | log_init(__progname, log_level, log_facility, log_stderr); |
| 340 | |
djm@openbsd.org | c7670b0 | 2019-01-21 12:53:35 +0000 | [diff] [blame] | 341 | while ((ch = getopt(argc, argv, "v")) != -1) { |
| 342 | switch (ch) { |
| 343 | case 'v': |
| 344 | log_stderr = 1; |
| 345 | if (log_level == SYSLOG_LEVEL_ERROR) |
| 346 | log_level = SYSLOG_LEVEL_DEBUG1; |
| 347 | else if (log_level < SYSLOG_LEVEL_DEBUG3) |
| 348 | log_level++; |
| 349 | break; |
| 350 | default: |
| 351 | fprintf(stderr, "usage: %s [-v]\n", __progname); |
| 352 | exit(1); |
| 353 | } |
| 354 | } |
| 355 | |
| 356 | log_init(__progname, log_level, log_facility, log_stderr); |
| 357 | |
| 358 | pkcs11_init(0); |
| 359 | |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 360 | in = STDIN_FILENO; |
| 361 | out = STDOUT_FILENO; |
| 362 | |
| 363 | max = 0; |
| 364 | if (in > max) |
| 365 | max = in; |
| 366 | if (out > max) |
| 367 | max = out; |
| 368 | |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 369 | if ((iqueue = sshbuf_new()) == NULL) |
| 370 | fatal("%s: sshbuf_new failed", __func__); |
| 371 | if ((oqueue = sshbuf_new()) == NULL) |
| 372 | fatal("%s: sshbuf_new failed", __func__); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 373 | |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 374 | while (1) { |
| 375 | memset(pfd, 0, sizeof(pfd)); |
| 376 | pfd[0].fd = in; |
| 377 | pfd[1].fd = out; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 378 | |
| 379 | /* |
| 380 | * Ensure that we can read a full buffer and handle |
| 381 | * the worst-case length packet it can generate, |
| 382 | * otherwise apply backpressure by stopping reads. |
| 383 | */ |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 384 | if ((r = sshbuf_check_reserve(iqueue, sizeof(buf))) == 0 && |
| 385 | (r = sshbuf_check_reserve(oqueue, MAX_MSG_LENGTH)) == 0) |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 386 | pfd[0].events = POLLIN; |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 387 | else if (r != SSH_ERR_NO_BUFFER_SPACE) |
| 388 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 389 | |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 390 | if (sshbuf_len(oqueue) > 0) |
| 391 | pfd[1].events = POLLOUT; |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 392 | |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 393 | if ((r = poll(pfd, 2, -1 /* INFTIM */)) <= 0) { |
| 394 | if (r == 0 || errno == EINTR) |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 395 | continue; |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 396 | fatal("poll: %s", strerror(errno)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 397 | } |
| 398 | |
| 399 | /* copy stdin to iqueue */ |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 400 | if ((pfd[0].revents & (POLLIN|POLLERR)) != 0) { |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 401 | len = read(in, buf, sizeof buf); |
| 402 | if (len == 0) { |
| 403 | debug("read eof"); |
| 404 | cleanup_exit(0); |
| 405 | } else if (len < 0) { |
| 406 | error("read: %s", strerror(errno)); |
| 407 | cleanup_exit(1); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 408 | } else if ((r = sshbuf_put(iqueue, buf, len)) != 0) { |
| 409 | fatal("%s: buffer error: %s", |
| 410 | __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 411 | } |
| 412 | } |
| 413 | /* send oqueue to stdout */ |
djm@openbsd.org | 2c22387 | 2019-01-23 02:01:10 +0000 | [diff] [blame] | 414 | if ((pfd[1].revents & (POLLOUT|POLLHUP)) != 0) { |
| 415 | len = write(out, sshbuf_ptr(oqueue), |
| 416 | sshbuf_len(oqueue)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 417 | if (len < 0) { |
| 418 | error("write: %s", strerror(errno)); |
| 419 | cleanup_exit(1); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 420 | } else if ((r = sshbuf_consume(oqueue, len)) != 0) { |
| 421 | fatal("%s: buffer error: %s", |
| 422 | __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 423 | } |
| 424 | } |
| 425 | |
| 426 | /* |
| 427 | * Process requests from client if we can fit the results |
| 428 | * into the output buffer, otherwise stop processing input |
| 429 | * and let the output queue drain. |
| 430 | */ |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 431 | if ((r = sshbuf_check_reserve(oqueue, MAX_MSG_LENGTH)) == 0) |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 432 | process(); |
markus@openbsd.org | b0d3413 | 2018-01-08 15:18:46 +0000 | [diff] [blame] | 433 | else if (r != SSH_ERR_NO_BUFFER_SPACE) |
| 434 | fatal("%s: buffer error: %s", __func__, ssh_err(r)); |
Damien Miller | 7ea845e | 2010-02-12 09:21:02 +1100 | [diff] [blame] | 435 | } |
| 436 | } |
djm@openbsd.org | 670104b | 2019-09-06 05:23:55 +0000 | [diff] [blame] | 437 | |
| 438 | #else /* WITH_OPENSSL */ |
| 439 | void |
| 440 | cleanup_exit(int i) |
| 441 | { |
| 442 | _exit(i); |
| 443 | } |
| 444 | |
| 445 | int |
| 446 | main(int argc, char **argv) |
| 447 | { |
| 448 | fprintf(stderr, "PKCS#11 code is not enabled\n"); |
| 449 | return 1; |
| 450 | } |
| 451 | #endif /* WITH_OPENSSL */ |
Damien Miller | dfa4156 | 2010-02-12 10:06:28 +1100 | [diff] [blame] | 452 | #else /* ENABLE_PKCS11 */ |
| 453 | int |
| 454 | main(int argc, char **argv) |
| 455 | { |
| 456 | extern char *__progname; |
| 457 | |
| 458 | __progname = ssh_get_progname(argv[0]); |
| 459 | log_init(__progname, SYSLOG_LEVEL_ERROR, SYSLOG_FACILITY_AUTH, 0); |
| 460 | fatal("PKCS#11 support disabled at compile time"); |
| 461 | } |
| 462 | #endif /* ENABLE_PKCS11 */ |