Annotation of src/usr.bin/ssh/ssh-xmss.c, Revision 1.14
1.14 ! djm 1: /* $OpenBSD: ssh-xmss.c,v 1.13 2022/10/28 00:44:17 djm Exp $*/
1.1 markus 2: /*
3: * Copyright (c) 2017 Stefan-Lukas Gazdag.
4: * Copyright (c) 2017 Markus Friedl.
5: *
6: * Permission to use, copy, modify, and distribute this software for any
7: * purpose with or without fee is hereby granted, provided that the above
8: * copyright notice and this permission notice appear in all copies.
9: *
10: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17: */
18: #define SSHKEY_INTERNAL
19: #include <sys/types.h>
20: #include <limits.h>
21:
1.5 millert 22: #include <stdlib.h>
1.1 markus 23: #include <string.h>
24: #include <stdarg.h>
1.5 millert 25: #include <stdint.h>
1.1 markus 26: #include <unistd.h>
27:
28: #include "log.h"
29: #include "sshbuf.h"
30: #include "sshkey.h"
31: #include "sshkey-xmss.h"
32: #include "ssherr.h"
33: #include "ssh.h"
34:
35: #include "xmss_fast.h"
36:
1.6 djm 37: static void
38: ssh_xmss_cleanup(struct sshkey *k)
39: {
40: freezero(k->xmss_pk, sshkey_xmss_pklen(k));
41: freezero(k->xmss_sk, sshkey_xmss_sklen(k));
42: sshkey_xmss_free_state(k);
43: free(k->xmss_name);
44: free(k->xmss_filename);
45: k->xmss_pk = NULL;
46: k->xmss_sk = NULL;
47: k->xmss_name = NULL;
48: k->xmss_filename = NULL;
49: }
50:
1.7 djm 51: static int
52: ssh_xmss_equal(const struct sshkey *a, const struct sshkey *b)
53: {
54: if (a->xmss_pk == NULL || b->xmss_pk == NULL)
55: return 0;
56: if (sshkey_xmss_pklen(a) != sshkey_xmss_pklen(b))
57: return 0;
58: if (memcmp(a->xmss_pk, b->xmss_pk, sshkey_xmss_pklen(a)) != 0)
59: return 0;
60: return 1;
61: }
62:
1.8 djm 63: static int
64: ssh_xmss_serialize_public(const struct sshkey *key, struct sshbuf *b,
1.11 djm 65: enum sshkey_serialize_rep opts)
1.8 djm 66: {
67: int r;
68:
69: if (key->xmss_name == NULL || key->xmss_pk == NULL ||
70: sshkey_xmss_pklen(key) == 0)
71: return SSH_ERR_INVALID_ARGUMENT;
1.11 djm 72: if ((r = sshbuf_put_cstring(b, key->xmss_name)) != 0 ||
1.8 djm 73: (r = sshbuf_put_string(b, key->xmss_pk,
74: sshkey_xmss_pklen(key))) != 0 ||
75: (r = sshkey_xmss_serialize_pk_info(key, b, opts)) != 0)
76: return r;
77:
78: return 0;
79: }
80:
1.10 djm 81: static int
1.13 djm 82: ssh_xmss_serialize_private(const struct sshkey *key, struct sshbuf *b,
83: enum sshkey_serialize_rep opts)
84: {
85: int r;
86:
87: if (key->xmss_name == NULL)
88: return SSH_ERR_INVALID_ARGUMENT;
89: /* Note: can't reuse ssh_xmss_serialize_public because of sk order */
90: if ((r = sshbuf_put_cstring(b, key->xmss_name)) != 0 ||
91: (r = sshbuf_put_string(b, key->xmss_pk,
92: sshkey_xmss_pklen(key))) != 0 ||
93: (r = sshbuf_put_string(b, key->xmss_sk,
94: sshkey_xmss_sklen(key))) != 0 ||
95: (r = sshkey_xmss_serialize_state_opt(key, b, opts)) != 0)
96: return r;
97:
98: return 0;
99: }
100:
101: static int
1.10 djm 102: ssh_xmss_copy_public(const struct sshkey *from, struct sshkey *to)
103: {
104: int r = SSH_ERR_INTERNAL_ERROR;
105: u_int32_t left;
106: size_t pklen;
107:
108: if ((r = sshkey_xmss_init(to, from->xmss_name)) != 0)
109: return r;
110: if (from->xmss_pk == NULL)
111: return 0; /* XXX SSH_ERR_INTERNAL_ERROR ? */
112:
113: if ((pklen = sshkey_xmss_pklen(from)) == 0 ||
114: sshkey_xmss_pklen(to) != pklen)
115: return SSH_ERR_INTERNAL_ERROR;
116: if ((to->xmss_pk = malloc(pklen)) == NULL)
117: return SSH_ERR_ALLOC_FAIL;
118: memcpy(to->xmss_pk, from->xmss_pk, pklen);
119: /* simulate number of signatures left on pubkey */
120: left = sshkey_xmss_signatures_left(from);
121: if (left)
122: sshkey_xmss_enable_maxsign(to, left);
123: return 0;
124: }
125:
1.11 djm 126: static int
127: ssh_xmss_deserialize_public(const char *ktype, struct sshbuf *b,
128: struct sshkey *key)
129: {
130: size_t len = 0;
131: char *xmss_name = NULL;
132: u_char *pk = NULL;
133: int ret = SSH_ERR_INTERNAL_ERROR;
134:
135: if ((ret = sshbuf_get_cstring(b, &xmss_name, NULL)) != 0)
136: goto out;
137: if ((ret = sshkey_xmss_init(key, xmss_name)) != 0)
138: goto out;
139: if ((ret = sshbuf_get_string(b, &pk, &len)) != 0)
140: goto out;
141: if (len == 0 || len != sshkey_xmss_pklen(key)) {
142: ret = SSH_ERR_INVALID_FORMAT;
143: goto out;
144: }
145: key->xmss_pk = pk;
146: pk = NULL;
147: if (!sshkey_is_cert(key) &&
148: (ret = sshkey_xmss_deserialize_pk_info(key, b)) != 0)
149: goto out;
150: /* success */
151: ret = 0;
152: out:
153: free(xmss_name);
154: freezero(pk, len);
155: return ret;
156: }
157:
1.12 djm 158: static int
1.14 ! djm 159: ssh_xmss_deserialize_private(const char *ktype, struct sshbuf *b,
! 160: struct sshkey *key)
! 161: {
! 162: int r;
! 163: char *xmss_name = NULL;
! 164: size_t pklen = 0, sklen = 0;
! 165: u_char *xmss_pk = NULL, *xmss_sk = NULL;
! 166:
! 167: /* Note: can't reuse ssh_xmss_deserialize_public because of sk order */
! 168: if ((r = sshbuf_get_cstring(b, &xmss_name, NULL)) != 0 ||
! 169: (r = sshbuf_get_string(b, &xmss_pk, &pklen)) != 0 ||
! 170: (r = sshbuf_get_string(b, &xmss_sk, &sklen)) != 0)
! 171: goto out;
! 172: if (!sshkey_is_cert(key) &&
! 173: (r = sshkey_xmss_init(key, xmss_name)) != 0)
! 174: goto out;
! 175: if (pklen != sshkey_xmss_pklen(key) ||
! 176: sklen != sshkey_xmss_sklen(key)) {
! 177: r = SSH_ERR_INVALID_FORMAT;
! 178: goto out;
! 179: }
! 180: key->xmss_pk = xmss_pk;
! 181: key->xmss_sk = xmss_sk;
! 182: xmss_pk = xmss_sk = NULL;
! 183: /* optional internal state */
! 184: if ((r = sshkey_xmss_deserialize_state_opt(key, b)) != 0)
! 185: goto out;
! 186: /* success */
! 187: r = 0;
! 188: out:
! 189: free(xmss_name);
! 190: freezero(xmss_pk, pklen);
! 191: freezero(xmss_sk, sklen);
! 192: return r;
! 193: }
! 194:
! 195: static int
1.12 djm 196: ssh_xmss_sign(struct sshkey *key,
197: u_char **sigp, size_t *lenp,
198: const u_char *data, size_t datalen,
199: const char *alg, const char *sk_provider, const char *sk_pin, u_int compat)
1.1 markus 200: {
201: u_char *sig = NULL;
202: size_t slen = 0, len = 0, required_siglen;
203: unsigned long long smlen;
204: int r, ret;
205: struct sshbuf *b = NULL;
206:
207: if (lenp != NULL)
208: *lenp = 0;
209: if (sigp != NULL)
210: *sigp = NULL;
211:
212: if (key == NULL ||
213: sshkey_type_plain(key->type) != KEY_XMSS ||
214: key->xmss_sk == NULL ||
215: sshkey_xmss_params(key) == NULL)
216: return SSH_ERR_INVALID_ARGUMENT;
217: if ((r = sshkey_xmss_siglen(key, &required_siglen)) != 0)
218: return r;
219: if (datalen >= INT_MAX - required_siglen)
220: return SSH_ERR_INVALID_ARGUMENT;
221: smlen = slen = datalen + required_siglen;
222: if ((sig = malloc(slen)) == NULL)
223: return SSH_ERR_ALLOC_FAIL;
1.4 dtucker 224: if ((r = sshkey_xmss_get_state(key, 1)) != 0)
1.1 markus 225: goto out;
226: if ((ret = xmss_sign(key->xmss_sk, sshkey_xmss_bds_state(key), sig, &smlen,
227: data, datalen, sshkey_xmss_params(key))) != 0 || smlen <= datalen) {
228: r = SSH_ERR_INVALID_ARGUMENT; /* XXX better error? */
229: goto out;
230: }
231: /* encode signature */
232: if ((b = sshbuf_new()) == NULL) {
233: r = SSH_ERR_ALLOC_FAIL;
234: goto out;
235: }
236: if ((r = sshbuf_put_cstring(b, "ssh-xmss@openssh.com")) != 0 ||
237: (r = sshbuf_put_string(b, sig, smlen - datalen)) != 0)
238: goto out;
239: len = sshbuf_len(b);
240: if (sigp != NULL) {
241: if ((*sigp = malloc(len)) == NULL) {
242: r = SSH_ERR_ALLOC_FAIL;
243: goto out;
244: }
245: memcpy(*sigp, sshbuf_ptr(b), len);
246: }
247: if (lenp != NULL)
248: *lenp = len;
249: /* success */
250: r = 0;
251: out:
1.4 dtucker 252: if ((ret = sshkey_xmss_update_state(key, 1)) != 0) {
1.1 markus 253: /* discard signature since we cannot update the state */
254: if (r == 0 && sigp != NULL && *sigp != NULL) {
255: explicit_bzero(*sigp, len);
256: free(*sigp);
257: }
258: if (sigp != NULL)
259: *sigp = NULL;
260: if (lenp != NULL)
261: *lenp = 0;
262: r = ret;
263: }
264: sshbuf_free(b);
1.2 jsg 265: if (sig != NULL)
266: freezero(sig, slen);
1.1 markus 267:
268: return r;
269: }
270:
1.12 djm 271: static int
1.1 markus 272: ssh_xmss_verify(const struct sshkey *key,
1.12 djm 273: const u_char *sig, size_t siglen,
274: const u_char *data, size_t dlen, const char *alg, u_int compat,
275: struct sshkey_sig_details **detailsp)
1.1 markus 276: {
277: struct sshbuf *b = NULL;
278: char *ktype = NULL;
279: const u_char *sigblob;
280: u_char *sm = NULL, *m = NULL;
281: size_t len, required_siglen;
282: unsigned long long smlen = 0, mlen = 0;
283: int r, ret;
284:
285: if (key == NULL ||
286: sshkey_type_plain(key->type) != KEY_XMSS ||
287: key->xmss_pk == NULL ||
288: sshkey_xmss_params(key) == NULL ||
1.12 djm 289: sig == NULL || siglen == 0)
1.1 markus 290: return SSH_ERR_INVALID_ARGUMENT;
291: if ((r = sshkey_xmss_siglen(key, &required_siglen)) != 0)
292: return r;
1.12 djm 293: if (dlen >= INT_MAX - required_siglen)
1.1 markus 294: return SSH_ERR_INVALID_ARGUMENT;
295:
1.12 djm 296: if ((b = sshbuf_from(sig, siglen)) == NULL)
1.1 markus 297: return SSH_ERR_ALLOC_FAIL;
298: if ((r = sshbuf_get_cstring(b, &ktype, NULL)) != 0 ||
299: (r = sshbuf_get_string_direct(b, &sigblob, &len)) != 0)
300: goto out;
301: if (strcmp("ssh-xmss@openssh.com", ktype) != 0) {
302: r = SSH_ERR_KEY_TYPE_MISMATCH;
303: goto out;
304: }
305: if (sshbuf_len(b) != 0) {
306: r = SSH_ERR_UNEXPECTED_TRAILING_DATA;
307: goto out;
308: }
309: if (len != required_siglen) {
310: r = SSH_ERR_INVALID_FORMAT;
311: goto out;
312: }
1.12 djm 313: if (dlen >= SIZE_MAX - len) {
1.1 markus 314: r = SSH_ERR_INVALID_ARGUMENT;
315: goto out;
316: }
1.12 djm 317: smlen = len + dlen;
1.1 markus 318: mlen = smlen;
319: if ((sm = malloc(smlen)) == NULL || (m = malloc(mlen)) == NULL) {
320: r = SSH_ERR_ALLOC_FAIL;
321: goto out;
322: }
323: memcpy(sm, sigblob, len);
1.12 djm 324: memcpy(sm+len, data, dlen);
1.1 markus 325: if ((ret = xmss_sign_open(m, &mlen, sm, smlen,
326: key->xmss_pk, sshkey_xmss_params(key))) != 0) {
1.3 djm 327: debug2_f("xmss_sign_open failed: %d", ret);
1.1 markus 328: }
1.12 djm 329: if (ret != 0 || mlen != dlen) {
1.1 markus 330: r = SSH_ERR_SIGNATURE_INVALID;
331: goto out;
332: }
333: /* XXX compare 'm' and 'data' ? */
334: /* success */
335: r = 0;
336: out:
1.2 jsg 337: if (sm != NULL)
338: freezero(sm, smlen);
339: if (m != NULL)
340: freezero(m, smlen);
1.1 markus 341: sshbuf_free(b);
342: free(ktype);
343: return r;
344: }
1.6 djm 345:
346: static const struct sshkey_impl_funcs sshkey_xmss_funcs = {
347: /* .size = */ NULL,
348: /* .alloc = */ NULL,
349: /* .cleanup = */ ssh_xmss_cleanup,
1.7 djm 350: /* .equal = */ ssh_xmss_equal,
1.8 djm 351: /* .ssh_serialize_public = */ ssh_xmss_serialize_public,
1.11 djm 352: /* .ssh_deserialize_public = */ ssh_xmss_deserialize_public,
1.13 djm 353: /* .ssh_serialize_private = */ ssh_xmss_serialize_private,
1.14 ! djm 354: /* .ssh_deserialize_private = */ ssh_xmss_deserialize_private,
1.9 djm 355: /* .generate = */ sshkey_xmss_generate_private_key,
1.10 djm 356: /* .copy_public = */ ssh_xmss_copy_public,
1.12 djm 357: /* .sign = */ ssh_xmss_sign,
358: /* .verify = */ ssh_xmss_verify,
1.6 djm 359: };
360:
361: const struct sshkey_impl sshkey_xmss_impl = {
362: /* .name = */ "ssh-xmss@openssh.com",
363: /* .shortname = */ "XMSS",
364: /* .sigalg = */ NULL,
365: /* .type = */ KEY_XMSS,
366: /* .nid = */ 0,
367: /* .cert = */ 0,
368: /* .sigonly = */ 0,
369: /* .keybits = */ 256,
370: /* .funcs = */ &sshkey_xmss_funcs,
371: };
372:
373: const struct sshkey_impl sshkey_xmss_cert_impl = {
374: /* .name = */ "ssh-xmss-cert-v01@openssh.com",
375: /* .shortname = */ "XMSS-CERT",
376: /* .sigalg = */ NULL,
377: /* .type = */ KEY_XMSS_CERT,
378: /* .nid = */ 0,
379: /* .cert = */ 1,
380: /* .sigonly = */ 0,
381: /* .keybits = */ 256,
382: /* .funcs = */ &sshkey_xmss_funcs,
383: };