xref: /netbsd-src/crypto/external/bsd/openssh/dist/krl.c (revision a629fefc36f2e87b36355a611e948fafe62680b4)
1*a629fefcSchristos /*	$NetBSD: krl.c,v 1.23 2023/10/25 20:19:57 christos Exp $	*/
2*a629fefcSchristos /* $OpenBSD: krl.c,v 1.59 2023/07/17 05:22:30 djm Exp $ */
379976551Schristos 
4d2a9b9efSchristos /*
5d2a9b9efSchristos  * Copyright (c) 2012 Damien Miller <djm@mindrot.org>
6d2a9b9efSchristos  *
7d2a9b9efSchristos  * Permission to use, copy, modify, and distribute this software for any
8d2a9b9efSchristos  * purpose with or without fee is hereby granted, provided that the above
9d2a9b9efSchristos  * copyright notice and this permission notice appear in all copies.
10d2a9b9efSchristos  *
11d2a9b9efSchristos  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12d2a9b9efSchristos  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13d2a9b9efSchristos  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14d2a9b9efSchristos  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15d2a9b9efSchristos  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16d2a9b9efSchristos  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17d2a9b9efSchristos  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18d2a9b9efSchristos  */
19d2a9b9efSchristos 
20ce11a51fSchristos #include "includes.h"
21*a629fefcSchristos __RCSID("$NetBSD: krl.c,v 1.23 2023/10/25 20:19:57 christos Exp $");
22*a629fefcSchristos 
23d2a9b9efSchristos #include <sys/types.h>
24d2a9b9efSchristos #include <sys/tree.h>
25d2a9b9efSchristos #include <sys/queue.h>
26d2a9b9efSchristos 
27d2a9b9efSchristos #include <errno.h>
28d2a9b9efSchristos #include <fcntl.h>
29d2a9b9efSchristos #include <limits.h>
30d2a9b9efSchristos #include <string.h>
31d2a9b9efSchristos #include <time.h>
32d2a9b9efSchristos #include <unistd.h>
33cd4ada6aSchristos #include <stdlib.h>
34d2a9b9efSchristos 
35e4d43b82Schristos #include "sshbuf.h"
36e4d43b82Schristos #include "ssherr.h"
37e4d43b82Schristos #include "sshkey.h"
38d2a9b9efSchristos #include "authfile.h"
39d2a9b9efSchristos #include "misc.h"
40d2a9b9efSchristos #include "log.h"
41e4d43b82Schristos #include "digest.h"
42e4d43b82Schristos #include "bitmap.h"
438db691beSchristos #include "utf8.h"
44d2a9b9efSchristos 
45d2a9b9efSchristos #include "krl.h"
46d2a9b9efSchristos 
47d2a9b9efSchristos /* #define DEBUG_KRL */
48d2a9b9efSchristos #ifdef DEBUG_KRL
4917418e98Schristos # define KRL_DBG(x) debug3_f x
50d2a9b9efSchristos #else
51d2a9b9efSchristos # define KRL_DBG(x)
52d2a9b9efSchristos #endif
53d2a9b9efSchristos 
54d2a9b9efSchristos /*
55d2a9b9efSchristos  * Trees of revoked serial numbers, key IDs and keys. This allows
56d2a9b9efSchristos  * quick searching, querying and producing lists in canonical order.
57d2a9b9efSchristos  */
58d2a9b9efSchristos 
59d2a9b9efSchristos /* Tree of serial numbers. XXX make smarter: really need a real sparse bitmap */
60d2a9b9efSchristos struct revoked_serial {
61d2a9b9efSchristos 	u_int64_t lo, hi;
62d2a9b9efSchristos 	RB_ENTRY(revoked_serial) tree_entry;
63d2a9b9efSchristos };
64d2a9b9efSchristos static int serial_cmp(struct revoked_serial *a, struct revoked_serial *b);
65d2a9b9efSchristos RB_HEAD(revoked_serial_tree, revoked_serial);
66b592f463Schristos RB_GENERATE_STATIC(revoked_serial_tree, revoked_serial, tree_entry, serial_cmp)
67d2a9b9efSchristos 
68d2a9b9efSchristos /* Tree of key IDs */
69d2a9b9efSchristos struct revoked_key_id {
70d2a9b9efSchristos 	char *key_id;
71d2a9b9efSchristos 	RB_ENTRY(revoked_key_id) tree_entry;
72d2a9b9efSchristos };
73d2a9b9efSchristos static int key_id_cmp(struct revoked_key_id *a, struct revoked_key_id *b);
74d2a9b9efSchristos RB_HEAD(revoked_key_id_tree, revoked_key_id);
75b592f463Schristos RB_GENERATE_STATIC(revoked_key_id_tree, revoked_key_id, tree_entry, key_id_cmp)
76d2a9b9efSchristos 
77d2a9b9efSchristos /* Tree of blobs (used for keys and fingerprints) */
78d2a9b9efSchristos struct revoked_blob {
79d2a9b9efSchristos 	u_char *blob;
80e4d43b82Schristos 	size_t len;
81d2a9b9efSchristos 	RB_ENTRY(revoked_blob) tree_entry;
82d2a9b9efSchristos };
83d2a9b9efSchristos static int blob_cmp(struct revoked_blob *a, struct revoked_blob *b);
84d2a9b9efSchristos RB_HEAD(revoked_blob_tree, revoked_blob);
85b592f463Schristos RB_GENERATE_STATIC(revoked_blob_tree, revoked_blob, tree_entry, blob_cmp)
86d2a9b9efSchristos 
87d2a9b9efSchristos /* Tracks revoked certs for a single CA */
88d2a9b9efSchristos struct revoked_certs {
89e4d43b82Schristos 	struct sshkey *ca_key;
90d2a9b9efSchristos 	struct revoked_serial_tree revoked_serials;
91d2a9b9efSchristos 	struct revoked_key_id_tree revoked_key_ids;
92d2a9b9efSchristos 	TAILQ_ENTRY(revoked_certs) entry;
93d2a9b9efSchristos };
94d2a9b9efSchristos TAILQ_HEAD(revoked_certs_list, revoked_certs);
95d2a9b9efSchristos 
96d2a9b9efSchristos struct ssh_krl {
97d2a9b9efSchristos 	u_int64_t krl_version;
98d2a9b9efSchristos 	u_int64_t generated_date;
99d2a9b9efSchristos 	u_int64_t flags;
100d2a9b9efSchristos 	char *comment;
101d2a9b9efSchristos 	struct revoked_blob_tree revoked_keys;
102d2a9b9efSchristos 	struct revoked_blob_tree revoked_sha1s;
103aa36fcacSchristos 	struct revoked_blob_tree revoked_sha256s;
104d2a9b9efSchristos 	struct revoked_certs_list revoked_certs;
105d2a9b9efSchristos };
106d2a9b9efSchristos 
107d2a9b9efSchristos /* Return equal if a and b overlap */
108d2a9b9efSchristos static int
serial_cmp(struct revoked_serial * a,struct revoked_serial * b)109d2a9b9efSchristos serial_cmp(struct revoked_serial *a, struct revoked_serial *b)
110d2a9b9efSchristos {
111d2a9b9efSchristos 	if (a->hi >= b->lo && a->lo <= b->hi)
112d2a9b9efSchristos 		return 0;
113d2a9b9efSchristos 	return a->lo < b->lo ? -1 : 1;
114d2a9b9efSchristos }
115d2a9b9efSchristos 
116d2a9b9efSchristos static int
key_id_cmp(struct revoked_key_id * a,struct revoked_key_id * b)117d2a9b9efSchristos key_id_cmp(struct revoked_key_id *a, struct revoked_key_id *b)
118d2a9b9efSchristos {
119d2a9b9efSchristos 	return strcmp(a->key_id, b->key_id);
120d2a9b9efSchristos }
121d2a9b9efSchristos 
122d2a9b9efSchristos static int
blob_cmp(struct revoked_blob * a,struct revoked_blob * b)123d2a9b9efSchristos blob_cmp(struct revoked_blob *a, struct revoked_blob *b)
124d2a9b9efSchristos {
125d2a9b9efSchristos 	int r;
126d2a9b9efSchristos 
127d2a9b9efSchristos 	if (a->len != b->len) {
128ee85abc4Schristos 		if ((r = memcmp(a->blob, b->blob, MINIMUM(a->len, b->len))) != 0)
129d2a9b9efSchristos 			return r;
130d2a9b9efSchristos 		return a->len > b->len ? 1 : -1;
131d2a9b9efSchristos 	} else
132d2a9b9efSchristos 		return memcmp(a->blob, b->blob, a->len);
133d2a9b9efSchristos }
134d2a9b9efSchristos 
135d2a9b9efSchristos struct ssh_krl *
ssh_krl_init(void)136d2a9b9efSchristos ssh_krl_init(void)
137d2a9b9efSchristos {
138d2a9b9efSchristos 	struct ssh_krl *krl;
139d2a9b9efSchristos 
140d2a9b9efSchristos 	if ((krl = calloc(1, sizeof(*krl))) == NULL)
141d2a9b9efSchristos 		return NULL;
142d2a9b9efSchristos 	RB_INIT(&krl->revoked_keys);
143d2a9b9efSchristos 	RB_INIT(&krl->revoked_sha1s);
144aa36fcacSchristos 	RB_INIT(&krl->revoked_sha256s);
145d2a9b9efSchristos 	TAILQ_INIT(&krl->revoked_certs);
146d2a9b9efSchristos 	return krl;
147d2a9b9efSchristos }
148d2a9b9efSchristos 
149d2a9b9efSchristos static void
revoked_certs_free(struct revoked_certs * rc)150d2a9b9efSchristos revoked_certs_free(struct revoked_certs *rc)
151d2a9b9efSchristos {
152d2a9b9efSchristos 	struct revoked_serial *rs, *trs;
153d2a9b9efSchristos 	struct revoked_key_id *rki, *trki;
154d2a9b9efSchristos 
155d2a9b9efSchristos 	RB_FOREACH_SAFE(rs, revoked_serial_tree, &rc->revoked_serials, trs) {
156d2a9b9efSchristos 		RB_REMOVE(revoked_serial_tree, &rc->revoked_serials, rs);
157d2a9b9efSchristos 		free(rs);
158d2a9b9efSchristos 	}
159d2a9b9efSchristos 	RB_FOREACH_SAFE(rki, revoked_key_id_tree, &rc->revoked_key_ids, trki) {
160d2a9b9efSchristos 		RB_REMOVE(revoked_key_id_tree, &rc->revoked_key_ids, rki);
161d2a9b9efSchristos 		free(rki->key_id);
162d2a9b9efSchristos 		free(rki);
163d2a9b9efSchristos 	}
164e4d43b82Schristos 	sshkey_free(rc->ca_key);
165d2a9b9efSchristos }
166d2a9b9efSchristos 
167d2a9b9efSchristos void
ssh_krl_free(struct ssh_krl * krl)168d2a9b9efSchristos ssh_krl_free(struct ssh_krl *krl)
169d2a9b9efSchristos {
170d2a9b9efSchristos 	struct revoked_blob *rb, *trb;
171d2a9b9efSchristos 	struct revoked_certs *rc, *trc;
172d2a9b9efSchristos 
173d2a9b9efSchristos 	if (krl == NULL)
174d2a9b9efSchristos 		return;
175d2a9b9efSchristos 
176d2a9b9efSchristos 	free(krl->comment);
177d2a9b9efSchristos 	RB_FOREACH_SAFE(rb, revoked_blob_tree, &krl->revoked_keys, trb) {
178d2a9b9efSchristos 		RB_REMOVE(revoked_blob_tree, &krl->revoked_keys, rb);
179d2a9b9efSchristos 		free(rb->blob);
180d2a9b9efSchristos 		free(rb);
181d2a9b9efSchristos 	}
182d2a9b9efSchristos 	RB_FOREACH_SAFE(rb, revoked_blob_tree, &krl->revoked_sha1s, trb) {
183d2a9b9efSchristos 		RB_REMOVE(revoked_blob_tree, &krl->revoked_sha1s, rb);
184d2a9b9efSchristos 		free(rb->blob);
185d2a9b9efSchristos 		free(rb);
186d2a9b9efSchristos 	}
187aa36fcacSchristos 	RB_FOREACH_SAFE(rb, revoked_blob_tree, &krl->revoked_sha256s, trb) {
188aa36fcacSchristos 		RB_REMOVE(revoked_blob_tree, &krl->revoked_sha256s, rb);
189aa36fcacSchristos 		free(rb->blob);
190aa36fcacSchristos 		free(rb);
191aa36fcacSchristos 	}
192d2a9b9efSchristos 	TAILQ_FOREACH_SAFE(rc, &krl->revoked_certs, entry, trc) {
193d2a9b9efSchristos 		TAILQ_REMOVE(&krl->revoked_certs, rc, entry);
194d2a9b9efSchristos 		revoked_certs_free(rc);
195d2a9b9efSchristos 	}
196b1066cf3Schristos 	free(krl);
197d2a9b9efSchristos }
198d2a9b9efSchristos 
199d2a9b9efSchristos void
ssh_krl_set_version(struct ssh_krl * krl,u_int64_t version)200d2a9b9efSchristos ssh_krl_set_version(struct ssh_krl *krl, u_int64_t version)
201d2a9b9efSchristos {
202d2a9b9efSchristos 	krl->krl_version = version;
203d2a9b9efSchristos }
204d2a9b9efSchristos 
205e4d43b82Schristos int
ssh_krl_set_comment(struct ssh_krl * krl,const char * comment)206d2a9b9efSchristos ssh_krl_set_comment(struct ssh_krl *krl, const char *comment)
207d2a9b9efSchristos {
208d2a9b9efSchristos 	free(krl->comment);
209d2a9b9efSchristos 	if ((krl->comment = strdup(comment)) == NULL)
210e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
211e4d43b82Schristos 	return 0;
212d2a9b9efSchristos }
213d2a9b9efSchristos 
214d2a9b9efSchristos /*
215d2a9b9efSchristos  * Find the revoked_certs struct for a CA key. If allow_create is set then
216d2a9b9efSchristos  * create a new one in the tree if one did not exist already.
217d2a9b9efSchristos  */
218d2a9b9efSchristos static int
revoked_certs_for_ca_key(struct ssh_krl * krl,const struct sshkey * ca_key,struct revoked_certs ** rcp,int allow_create)219e4d43b82Schristos revoked_certs_for_ca_key(struct ssh_krl *krl, const struct sshkey *ca_key,
220d2a9b9efSchristos     struct revoked_certs **rcp, int allow_create)
221d2a9b9efSchristos {
222d2a9b9efSchristos 	struct revoked_certs *rc;
223e4d43b82Schristos 	int r;
224d2a9b9efSchristos 
225d2a9b9efSchristos 	*rcp = NULL;
226d2a9b9efSchristos 	TAILQ_FOREACH(rc, &krl->revoked_certs, entry) {
227e4d43b82Schristos 		if ((ca_key == NULL && rc->ca_key == NULL) ||
228e4d43b82Schristos 		    sshkey_equal(rc->ca_key, ca_key)) {
229d2a9b9efSchristos 			*rcp = rc;
230d2a9b9efSchristos 			return 0;
231d2a9b9efSchristos 		}
232d2a9b9efSchristos 	}
233d2a9b9efSchristos 	if (!allow_create)
234d2a9b9efSchristos 		return 0;
235d2a9b9efSchristos 	/* If this CA doesn't exist in the list then add it now */
236d2a9b9efSchristos 	if ((rc = calloc(1, sizeof(*rc))) == NULL)
237e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
238e4d43b82Schristos 	if (ca_key == NULL)
239e4d43b82Schristos 		rc->ca_key = NULL;
240e4d43b82Schristos 	else if ((r = sshkey_from_private(ca_key, &rc->ca_key)) != 0) {
241d2a9b9efSchristos 		free(rc);
242e4d43b82Schristos 		return r;
243d2a9b9efSchristos 	}
244d2a9b9efSchristos 	RB_INIT(&rc->revoked_serials);
245d2a9b9efSchristos 	RB_INIT(&rc->revoked_key_ids);
246d2a9b9efSchristos 	TAILQ_INSERT_TAIL(&krl->revoked_certs, rc, entry);
24717418e98Schristos 	KRL_DBG(("new CA %s", ca_key == NULL ? "*" : sshkey_type(ca_key)));
248d2a9b9efSchristos 	*rcp = rc;
249d2a9b9efSchristos 	return 0;
250d2a9b9efSchristos }
251d2a9b9efSchristos 
252d2a9b9efSchristos static int
insert_serial_range(struct revoked_serial_tree * rt,u_int64_t lo,u_int64_t hi)253d2a9b9efSchristos insert_serial_range(struct revoked_serial_tree *rt, u_int64_t lo, u_int64_t hi)
254d2a9b9efSchristos {
255d2a9b9efSchristos 	struct revoked_serial rs, *ers, *crs, *irs;
256d2a9b9efSchristos 
25717418e98Schristos 	KRL_DBG(("insert %"PRIu64":%"PRIu64, lo, hi));
2588a4530f9Schristos 	memset(&rs, 0, sizeof(rs));
259d2a9b9efSchristos 	rs.lo = lo;
260d2a9b9efSchristos 	rs.hi = hi;
261d2a9b9efSchristos 	ers = RB_NFIND(revoked_serial_tree, rt, &rs);
262d2a9b9efSchristos 	if (ers == NULL || serial_cmp(ers, &rs) != 0) {
263d2a9b9efSchristos 		/* No entry matches. Just insert */
264d2a9b9efSchristos 		if ((irs = malloc(sizeof(rs))) == NULL)
265e4d43b82Schristos 			return SSH_ERR_ALLOC_FAIL;
266d2a9b9efSchristos 		memcpy(irs, &rs, sizeof(*irs));
267d2a9b9efSchristos 		ers = RB_INSERT(revoked_serial_tree, rt, irs);
268d2a9b9efSchristos 		if (ers != NULL) {
26917418e98Schristos 			KRL_DBG(("bad: ers != NULL"));
270d2a9b9efSchristos 			/* Shouldn't happen */
271d2a9b9efSchristos 			free(irs);
272e4d43b82Schristos 			return SSH_ERR_INTERNAL_ERROR;
273d2a9b9efSchristos 		}
274d2a9b9efSchristos 		ers = irs;
275d2a9b9efSchristos 	} else {
27617418e98Schristos 		KRL_DBG(("overlap found %"PRIu64":%"PRIu64,
277d2a9b9efSchristos 		    ers->lo, ers->hi));
278d2a9b9efSchristos 		/*
279d2a9b9efSchristos 		 * The inserted entry overlaps an existing one. Grow the
280d2a9b9efSchristos 		 * existing entry.
281d2a9b9efSchristos 		 */
282d2a9b9efSchristos 		if (ers->lo > lo)
283d2a9b9efSchristos 			ers->lo = lo;
284d2a9b9efSchristos 		if (ers->hi < hi)
285d2a9b9efSchristos 			ers->hi = hi;
286d2a9b9efSchristos 	}
287e4d43b82Schristos 
288d2a9b9efSchristos 	/*
289d2a9b9efSchristos 	 * The inserted or revised range might overlap or abut adjacent ones;
290d2a9b9efSchristos 	 * coalesce as necessary.
291d2a9b9efSchristos 	 */
292d2a9b9efSchristos 
293d2a9b9efSchristos 	/* Check predecessors */
294d2a9b9efSchristos 	while ((crs = RB_PREV(revoked_serial_tree, rt, ers)) != NULL) {
29517418e98Schristos 		KRL_DBG(("pred %"PRIu64":%"PRIu64,
296ce11a51fSchristos 		    crs->lo, crs->hi));
297d2a9b9efSchristos 		if (ers->lo != 0 && crs->hi < ers->lo - 1)
298d2a9b9efSchristos 			break;
299d2a9b9efSchristos 		/* This entry overlaps. */
300d2a9b9efSchristos 		if (crs->lo < ers->lo) {
301d2a9b9efSchristos 			ers->lo = crs->lo;
30217418e98Schristos 			KRL_DBG(("pred extend %"PRIu64":%"PRIu64,
303d2a9b9efSchristos 			    ers->lo, ers->hi));
304d2a9b9efSchristos 		}
305d2a9b9efSchristos 		RB_REMOVE(revoked_serial_tree, rt, crs);
306d2a9b9efSchristos 		free(crs);
307d2a9b9efSchristos 	}
308d2a9b9efSchristos 	/* Check successors */
309d2a9b9efSchristos 	while ((crs = RB_NEXT(revoked_serial_tree, rt, ers)) != NULL) {
31017418e98Schristos 		KRL_DBG(("succ %"PRIu64":%"PRIu64, crs->lo, crs->hi));
311d2a9b9efSchristos 		if (ers->hi != (u_int64_t)-1 && crs->lo > ers->hi + 1)
312d2a9b9efSchristos 			break;
313d2a9b9efSchristos 		/* This entry overlaps. */
314d2a9b9efSchristos 		if (crs->hi > ers->hi) {
315d2a9b9efSchristos 			ers->hi = crs->hi;
31617418e98Schristos 			KRL_DBG(("succ extend %"PRIu64":%"PRIu64,
317d2a9b9efSchristos 			    ers->lo, ers->hi));
318d2a9b9efSchristos 		}
319d2a9b9efSchristos 		RB_REMOVE(revoked_serial_tree, rt, crs);
320d2a9b9efSchristos 		free(crs);
321d2a9b9efSchristos 	}
32217418e98Schristos 	KRL_DBG(("done, final %"PRIu64":%"PRIu64, ers->lo, ers->hi));
323d2a9b9efSchristos 	return 0;
324d2a9b9efSchristos }
325d2a9b9efSchristos 
326d2a9b9efSchristos int
ssh_krl_revoke_cert_by_serial(struct ssh_krl * krl,const struct sshkey * ca_key,u_int64_t serial)327e4d43b82Schristos ssh_krl_revoke_cert_by_serial(struct ssh_krl *krl, const struct sshkey *ca_key,
328d2a9b9efSchristos     u_int64_t serial)
329d2a9b9efSchristos {
330d2a9b9efSchristos 	return ssh_krl_revoke_cert_by_serial_range(krl, ca_key, serial, serial);
331d2a9b9efSchristos }
332d2a9b9efSchristos 
333d2a9b9efSchristos int
ssh_krl_revoke_cert_by_serial_range(struct ssh_krl * krl,const struct sshkey * ca_key,u_int64_t lo,u_int64_t hi)334e4d43b82Schristos ssh_krl_revoke_cert_by_serial_range(struct ssh_krl *krl,
335e4d43b82Schristos     const struct sshkey *ca_key, u_int64_t lo, u_int64_t hi)
336d2a9b9efSchristos {
337d2a9b9efSchristos 	struct revoked_certs *rc;
338e4d43b82Schristos 	int r;
339d2a9b9efSchristos 
340d2a9b9efSchristos 	if (lo > hi || lo == 0)
341e4d43b82Schristos 		return SSH_ERR_INVALID_ARGUMENT;
342e4d43b82Schristos 	if ((r = revoked_certs_for_ca_key(krl, ca_key, &rc, 1)) != 0)
343e4d43b82Schristos 		return r;
344d2a9b9efSchristos 	return insert_serial_range(&rc->revoked_serials, lo, hi);
345d2a9b9efSchristos }
346d2a9b9efSchristos 
347d2a9b9efSchristos int
ssh_krl_revoke_cert_by_key_id(struct ssh_krl * krl,const struct sshkey * ca_key,const char * key_id)348e4d43b82Schristos ssh_krl_revoke_cert_by_key_id(struct ssh_krl *krl, const struct sshkey *ca_key,
349d2a9b9efSchristos     const char *key_id)
350d2a9b9efSchristos {
351d2a9b9efSchristos 	struct revoked_key_id *rki, *erki;
352d2a9b9efSchristos 	struct revoked_certs *rc;
353e4d43b82Schristos 	int r;
354d2a9b9efSchristos 
355e4d43b82Schristos 	if ((r = revoked_certs_for_ca_key(krl, ca_key, &rc, 1)) != 0)
356e4d43b82Schristos 		return r;
357d2a9b9efSchristos 
35817418e98Schristos 	KRL_DBG(("revoke %s", key_id));
359d2a9b9efSchristos 	if ((rki = calloc(1, sizeof(*rki))) == NULL ||
360d2a9b9efSchristos 	    (rki->key_id = strdup(key_id)) == NULL) {
361d2a9b9efSchristos 		free(rki);
362e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
363d2a9b9efSchristos 	}
364d2a9b9efSchristos 	erki = RB_INSERT(revoked_key_id_tree, &rc->revoked_key_ids, rki);
365d2a9b9efSchristos 	if (erki != NULL) {
366d2a9b9efSchristos 		free(rki->key_id);
367d2a9b9efSchristos 		free(rki);
368d2a9b9efSchristos 	}
369d2a9b9efSchristos 	return 0;
370d2a9b9efSchristos }
371d2a9b9efSchristos 
372d2a9b9efSchristos /* Convert "key" to a public key blob without any certificate information */
373d2a9b9efSchristos static int
plain_key_blob(const struct sshkey * key,u_char ** blob,size_t * blen)374e4d43b82Schristos plain_key_blob(const struct sshkey *key, u_char **blob, size_t *blen)
375d2a9b9efSchristos {
376e4d43b82Schristos 	struct sshkey *kcopy;
377d2a9b9efSchristos 	int r;
378d2a9b9efSchristos 
379e4d43b82Schristos 	if ((r = sshkey_from_private(key, &kcopy)) != 0)
380e4d43b82Schristos 		return r;
381e4d43b82Schristos 	if (sshkey_is_cert(kcopy)) {
382e4d43b82Schristos 		if ((r = sshkey_drop_cert(kcopy)) != 0) {
383e4d43b82Schristos 			sshkey_free(kcopy);
384e4d43b82Schristos 			return r;
385d2a9b9efSchristos 		}
386d2a9b9efSchristos 	}
387e4d43b82Schristos 	r = sshkey_to_blob(kcopy, blob, blen);
388e4d43b82Schristos 	sshkey_free(kcopy);
3898a4530f9Schristos 	return r;
390d2a9b9efSchristos }
391d2a9b9efSchristos 
392d2a9b9efSchristos /* Revoke a key blob. Ownership of blob is transferred to the tree */
393d2a9b9efSchristos static int
revoke_blob(struct revoked_blob_tree * rbt,u_char * blob,size_t len)394e4d43b82Schristos revoke_blob(struct revoked_blob_tree *rbt, u_char *blob, size_t len)
395d2a9b9efSchristos {
396d2a9b9efSchristos 	struct revoked_blob *rb, *erb;
397d2a9b9efSchristos 
398d2a9b9efSchristos 	if ((rb = calloc(1, sizeof(*rb))) == NULL)
399e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
400d2a9b9efSchristos 	rb->blob = blob;
401d2a9b9efSchristos 	rb->len = len;
402d2a9b9efSchristos 	erb = RB_INSERT(revoked_blob_tree, rbt, rb);
403d2a9b9efSchristos 	if (erb != NULL) {
404d2a9b9efSchristos 		free(rb->blob);
405d2a9b9efSchristos 		free(rb);
406d2a9b9efSchristos 	}
407d2a9b9efSchristos 	return 0;
408d2a9b9efSchristos }
409d2a9b9efSchristos 
410d2a9b9efSchristos int
ssh_krl_revoke_key_explicit(struct ssh_krl * krl,const struct sshkey * key)411e4d43b82Schristos ssh_krl_revoke_key_explicit(struct ssh_krl *krl, const struct sshkey *key)
412d2a9b9efSchristos {
413d2a9b9efSchristos 	u_char *blob;
414e4d43b82Schristos 	size_t len;
415e4d43b82Schristos 	int r;
416d2a9b9efSchristos 
41717418e98Schristos 	debug3_f("revoke type %s", sshkey_type(key));
418e4d43b82Schristos 	if ((r = plain_key_blob(key, &blob, &len)) != 0)
419e4d43b82Schristos 		return r;
420d2a9b9efSchristos 	return revoke_blob(&krl->revoked_keys, blob, len);
421d2a9b9efSchristos }
422d2a9b9efSchristos 
423aa36fcacSchristos static int
revoke_by_hash(struct revoked_blob_tree * target,const u_char * p,size_t len)424aa36fcacSchristos revoke_by_hash(struct revoked_blob_tree *target, const u_char *p, size_t len)
425d2a9b9efSchristos {
426d2a9b9efSchristos 	u_char *blob;
427e4d43b82Schristos 	int r;
428d2a9b9efSchristos 
429aa36fcacSchristos 	/* need to copy hash, as revoke_blob steals ownership */
430aa36fcacSchristos 	if ((blob = malloc(len)) == NULL)
431aa36fcacSchristos 		return SSH_ERR_SYSTEM_ERROR;
432aa36fcacSchristos 	memcpy(blob, p, len);
433aa36fcacSchristos 	if ((r = revoke_blob(target, blob, len)) != 0) {
434aa36fcacSchristos 		free(blob);
435e4d43b82Schristos 		return r;
436aa36fcacSchristos 	}
437aa36fcacSchristos 	return 0;
438aa36fcacSchristos }
439aa36fcacSchristos 
440aa36fcacSchristos int
ssh_krl_revoke_key_sha1(struct ssh_krl * krl,const u_char * p,size_t len)441aa36fcacSchristos ssh_krl_revoke_key_sha1(struct ssh_krl *krl, const u_char *p, size_t len)
442aa36fcacSchristos {
44317418e98Schristos 	debug3_f("revoke by sha1");
444aa36fcacSchristos 	if (len != 20)
445aa36fcacSchristos 		return SSH_ERR_INVALID_FORMAT;
446aa36fcacSchristos 	return revoke_by_hash(&krl->revoked_sha1s, p, len);
447aa36fcacSchristos }
448aa36fcacSchristos 
449aa36fcacSchristos int
ssh_krl_revoke_key_sha256(struct ssh_krl * krl,const u_char * p,size_t len)450aa36fcacSchristos ssh_krl_revoke_key_sha256(struct ssh_krl *krl, const u_char *p, size_t len)
451aa36fcacSchristos {
45217418e98Schristos 	debug3_f("revoke by sha256");
453aa36fcacSchristos 	if (len != 32)
454aa36fcacSchristos 		return SSH_ERR_INVALID_FORMAT;
455aa36fcacSchristos 	return revoke_by_hash(&krl->revoked_sha256s, p, len);
456d2a9b9efSchristos }
457d2a9b9efSchristos 
458d2a9b9efSchristos int
ssh_krl_revoke_key(struct ssh_krl * krl,const struct sshkey * key)459e4d43b82Schristos ssh_krl_revoke_key(struct ssh_krl *krl, const struct sshkey *key)
460d2a9b9efSchristos {
461aa36fcacSchristos 	/* XXX replace with SHA256? */
462e4d43b82Schristos 	if (!sshkey_is_cert(key))
463aa36fcacSchristos 		return ssh_krl_revoke_key_explicit(krl, key);
464d2a9b9efSchristos 
4658395c133Schristos 	if (key->cert->serial == 0) {
466d2a9b9efSchristos 		return ssh_krl_revoke_cert_by_key_id(krl,
467d2a9b9efSchristos 		    key->cert->signature_key,
468d2a9b9efSchristos 		    key->cert->key_id);
469d2a9b9efSchristos 	} else {
470d2a9b9efSchristos 		return ssh_krl_revoke_cert_by_serial(krl,
471d2a9b9efSchristos 		    key->cert->signature_key,
472d2a9b9efSchristos 		    key->cert->serial);
473d2a9b9efSchristos 	}
474d2a9b9efSchristos }
475d2a9b9efSchristos 
476d2a9b9efSchristos /*
477e4d43b82Schristos  * Select the most compact section type to emit next in a KRL based on
478e4d43b82Schristos  * the current section type, the run length of contiguous revoked serial
479d2a9b9efSchristos  * numbers and the gaps from the last and to the next revoked serial.
480d2a9b9efSchristos  * Applies a mostly-accurate bit cost model to select the section type
481d2a9b9efSchristos  * that will minimise the size of the resultant KRL.
482d2a9b9efSchristos  */
483d2a9b9efSchristos static int
choose_next_state(int current_state,u_int64_t contig,int final,u_int64_t last_gap,u_int64_t next_gap,int * force_new_section)484d2a9b9efSchristos choose_next_state(int current_state, u_int64_t contig, int final,
485d2a9b9efSchristos     u_int64_t last_gap, u_int64_t next_gap, int *force_new_section)
486d2a9b9efSchristos {
487d2a9b9efSchristos 	int new_state;
488d2a9b9efSchristos 	u_int64_t cost, cost_list, cost_range, cost_bitmap, cost_bitmap_restart;
489d2a9b9efSchristos 
490d2a9b9efSchristos 	/*
491d2a9b9efSchristos 	 * Avoid unsigned overflows.
492d2a9b9efSchristos 	 * The limits are high enough to avoid confusing the calculations.
493d2a9b9efSchristos 	 */
494ee85abc4Schristos 	contig = MINIMUM(contig, 1ULL<<31);
495ee85abc4Schristos 	last_gap = MINIMUM(last_gap, 1ULL<<31);
496ee85abc4Schristos 	next_gap = MINIMUM(next_gap, 1ULL<<31);
497d2a9b9efSchristos 
498d2a9b9efSchristos 	/*
499d2a9b9efSchristos 	 * Calculate the cost to switch from the current state to candidates.
500d2a9b9efSchristos 	 * NB. range sections only ever contain a single range, so their
501d2a9b9efSchristos 	 * switching cost is independent of the current_state.
502d2a9b9efSchristos 	 */
503d2a9b9efSchristos 	cost_list = cost_bitmap = cost_bitmap_restart = 0;
504d2a9b9efSchristos 	cost_range = 8;
505d2a9b9efSchristos 	switch (current_state) {
506d2a9b9efSchristos 	case KRL_SECTION_CERT_SERIAL_LIST:
507d2a9b9efSchristos 		cost_bitmap_restart = cost_bitmap = 8 + 64;
508d2a9b9efSchristos 		break;
509d2a9b9efSchristos 	case KRL_SECTION_CERT_SERIAL_BITMAP:
510d2a9b9efSchristos 		cost_list = 8;
511d2a9b9efSchristos 		cost_bitmap_restart = 8 + 64;
512d2a9b9efSchristos 		break;
513d2a9b9efSchristos 	case KRL_SECTION_CERT_SERIAL_RANGE:
514d2a9b9efSchristos 	case 0:
515d2a9b9efSchristos 		cost_bitmap_restart = cost_bitmap = 8 + 64;
516d2a9b9efSchristos 		cost_list = 8;
517d2a9b9efSchristos 	}
518d2a9b9efSchristos 
519d2a9b9efSchristos 	/* Estimate base cost in bits of each section type */
520d2a9b9efSchristos 	cost_list += 64 * contig + (final ? 0 : 8+64);
521d2a9b9efSchristos 	cost_range += (2 * 64) + (final ? 0 : 8+64);
522ee85abc4Schristos 	cost_bitmap += last_gap + contig + (final ? 0 : MINIMUM(next_gap, 8+64));
523ee85abc4Schristos 	cost_bitmap_restart += contig + (final ? 0 : MINIMUM(next_gap, 8+64));
524d2a9b9efSchristos 
525d2a9b9efSchristos 	/* Convert to byte costs for actual comparison */
526d2a9b9efSchristos 	cost_list = (cost_list + 7) / 8;
527d2a9b9efSchristos 	cost_bitmap = (cost_bitmap + 7) / 8;
528d2a9b9efSchristos 	cost_bitmap_restart = (cost_bitmap_restart + 7) / 8;
529d2a9b9efSchristos 	cost_range = (cost_range + 7) / 8;
530d2a9b9efSchristos 
531d2a9b9efSchristos 	/* Now pick the best choice */
532d2a9b9efSchristos 	*force_new_section = 0;
533d2a9b9efSchristos 	new_state = KRL_SECTION_CERT_SERIAL_BITMAP;
534d2a9b9efSchristos 	cost = cost_bitmap;
535d2a9b9efSchristos 	if (cost_range < cost) {
536d2a9b9efSchristos 		new_state = KRL_SECTION_CERT_SERIAL_RANGE;
537d2a9b9efSchristos 		cost = cost_range;
538d2a9b9efSchristos 	}
539d2a9b9efSchristos 	if (cost_list < cost) {
540d2a9b9efSchristos 		new_state = KRL_SECTION_CERT_SERIAL_LIST;
541d2a9b9efSchristos 		cost = cost_list;
542d2a9b9efSchristos 	}
543d2a9b9efSchristos 	if (cost_bitmap_restart < cost) {
544d2a9b9efSchristos 		new_state = KRL_SECTION_CERT_SERIAL_BITMAP;
545d2a9b9efSchristos 		*force_new_section = 1;
546d2a9b9efSchristos 		cost = cost_bitmap_restart;
547d2a9b9efSchristos 	}
54817418e98Schristos 	KRL_DBG(("contig %llu last_gap %llu next_gap %llu final %d, costs:"
54900a838c4Schristos 	    "list %llu range %llu bitmap %llu new bitmap %llu, "
55017418e98Schristos 	    "selected 0x%02x%s", (long long unsigned)contig,
55100a838c4Schristos 	    (long long unsigned)last_gap, (long long unsigned)next_gap, final,
55200a838c4Schristos 	    (long long unsigned)cost_list, (long long unsigned)cost_range,
55300a838c4Schristos 	    (long long unsigned)cost_bitmap,
55400a838c4Schristos 	    (long long unsigned)cost_bitmap_restart, new_state,
555e4d43b82Schristos 	    *force_new_section ? " restart" : ""));
556d2a9b9efSchristos 	return new_state;
557d2a9b9efSchristos }
558d2a9b9efSchristos 
559e4d43b82Schristos static int
put_bitmap(struct sshbuf * buf,struct bitmap * bitmap)560e4d43b82Schristos put_bitmap(struct sshbuf *buf, struct bitmap *bitmap)
561e4d43b82Schristos {
562e4d43b82Schristos 	size_t len;
563e4d43b82Schristos 	u_char *blob;
564e4d43b82Schristos 	int r;
565e4d43b82Schristos 
566e4d43b82Schristos 	len = bitmap_nbytes(bitmap);
567e4d43b82Schristos 	if ((blob = malloc(len)) == NULL)
568e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
569e4d43b82Schristos 	if (bitmap_to_string(bitmap, blob, len) != 0) {
570e4d43b82Schristos 		free(blob);
571e4d43b82Schristos 		return SSH_ERR_INTERNAL_ERROR;
572e4d43b82Schristos 	}
573e4d43b82Schristos 	r = sshbuf_put_bignum2_bytes(buf, blob, len);
574e4d43b82Schristos 	free(blob);
575e4d43b82Schristos 	return r;
576e4d43b82Schristos }
577e4d43b82Schristos 
578d2a9b9efSchristos /* Generate a KRL_SECTION_CERTIFICATES KRL section */
579d2a9b9efSchristos static int
revoked_certs_generate(struct revoked_certs * rc,struct sshbuf * buf)580e4d43b82Schristos revoked_certs_generate(struct revoked_certs *rc, struct sshbuf *buf)
581d2a9b9efSchristos {
582e4d43b82Schristos 	int final, force_new_sect, r = SSH_ERR_INTERNAL_ERROR;
583d2a9b9efSchristos 	u_int64_t i, contig, gap, last = 0, bitmap_start = 0;
584d2a9b9efSchristos 	struct revoked_serial *rs, *nrs;
585d2a9b9efSchristos 	struct revoked_key_id *rki;
586d2a9b9efSchristos 	int next_state, state = 0;
587e4d43b82Schristos 	struct sshbuf *sect;
588e4d43b82Schristos 	struct bitmap *bitmap = NULL;
589d2a9b9efSchristos 
590e4d43b82Schristos 	if ((sect = sshbuf_new()) == NULL)
591e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
592d2a9b9efSchristos 
593e4d43b82Schristos 	/* Store the header: optional CA scope key, reserved */
594e4d43b82Schristos 	if (rc->ca_key == NULL) {
595e4d43b82Schristos 		if ((r = sshbuf_put_string(buf, NULL, 0)) != 0)
596e4d43b82Schristos 			goto out;
597e4d43b82Schristos 	} else {
598e4d43b82Schristos 		if ((r = sshkey_puts(rc->ca_key, buf)) != 0)
599e4d43b82Schristos 			goto out;
600e4d43b82Schristos 	}
601e4d43b82Schristos 	if ((r = sshbuf_put_string(buf, NULL, 0)) != 0)
602e4d43b82Schristos 		goto out;
603d2a9b9efSchristos 
604d2a9b9efSchristos 	/* Store the revoked serials.  */
605d2a9b9efSchristos 	for (rs = RB_MIN(revoked_serial_tree, &rc->revoked_serials);
606d2a9b9efSchristos 	     rs != NULL;
607d2a9b9efSchristos 	     rs = RB_NEXT(revoked_serial_tree, &rc->revoked_serials, rs)) {
60817418e98Schristos 		KRL_DBG(("serial %llu:%llu state 0x%02x",
60900a838c4Schristos 		    (long long unsigned)rs->lo, (long long unsigned)rs->hi,
610e4d43b82Schristos 		    state));
611d2a9b9efSchristos 
612d2a9b9efSchristos 		/* Check contiguous length and gap to next section (if any) */
613d2a9b9efSchristos 		nrs = RB_NEXT(revoked_serial_tree, &rc->revoked_serials, rs);
614d2a9b9efSchristos 		final = nrs == NULL;
615d2a9b9efSchristos 		gap = nrs == NULL ? 0 : nrs->lo - rs->hi;
616d2a9b9efSchristos 		contig = 1 + (rs->hi - rs->lo);
617d2a9b9efSchristos 
618d2a9b9efSchristos 		/* Choose next state based on these */
619d2a9b9efSchristos 		next_state = choose_next_state(state, contig, final,
620d2a9b9efSchristos 		    state == 0 ? 0 : rs->lo - last, gap, &force_new_sect);
621d2a9b9efSchristos 
622d2a9b9efSchristos 		/*
623d2a9b9efSchristos 		 * If the current section is a range section or has a different
624d2a9b9efSchristos 		 * type to the next section, then finish it off now.
625d2a9b9efSchristos 		 */
626d2a9b9efSchristos 		if (state != 0 && (force_new_sect || next_state != state ||
627d2a9b9efSchristos 		    state == KRL_SECTION_CERT_SERIAL_RANGE)) {
62817418e98Schristos 			KRL_DBG(("finish state 0x%02x", state));
629d2a9b9efSchristos 			switch (state) {
630d2a9b9efSchristos 			case KRL_SECTION_CERT_SERIAL_LIST:
631d2a9b9efSchristos 			case KRL_SECTION_CERT_SERIAL_RANGE:
632d2a9b9efSchristos 				break;
633d2a9b9efSchristos 			case KRL_SECTION_CERT_SERIAL_BITMAP:
634e4d43b82Schristos 				if ((r = put_bitmap(sect, bitmap)) != 0)
635e4d43b82Schristos 					goto out;
636e4d43b82Schristos 				bitmap_free(bitmap);
637d2a9b9efSchristos 				bitmap = NULL;
638d2a9b9efSchristos 				break;
639d2a9b9efSchristos 			}
640e4d43b82Schristos 			if ((r = sshbuf_put_u8(buf, state)) != 0 ||
641e4d43b82Schristos 			    (r = sshbuf_put_stringb(buf, sect)) != 0)
642e4d43b82Schristos 				goto out;
643e4d43b82Schristos 			sshbuf_reset(sect);
644d2a9b9efSchristos 		}
645d2a9b9efSchristos 
646d2a9b9efSchristos 		/* If we are starting a new section then prepare it now */
647d2a9b9efSchristos 		if (next_state != state || force_new_sect) {
64817418e98Schristos 			KRL_DBG(("start state 0x%02x",
649e4d43b82Schristos 			    next_state));
650d2a9b9efSchristos 			state = next_state;
651e4d43b82Schristos 			sshbuf_reset(sect);
652d2a9b9efSchristos 			switch (state) {
653d2a9b9efSchristos 			case KRL_SECTION_CERT_SERIAL_LIST:
654d2a9b9efSchristos 			case KRL_SECTION_CERT_SERIAL_RANGE:
655d2a9b9efSchristos 				break;
656d2a9b9efSchristos 			case KRL_SECTION_CERT_SERIAL_BITMAP:
657e4d43b82Schristos 				if ((bitmap = bitmap_new()) == NULL) {
658e4d43b82Schristos 					r = SSH_ERR_ALLOC_FAIL;
659d2a9b9efSchristos 					goto out;
660e4d43b82Schristos 				}
661d2a9b9efSchristos 				bitmap_start = rs->lo;
662e4d43b82Schristos 				if ((r = sshbuf_put_u64(sect,
663e4d43b82Schristos 				    bitmap_start)) != 0)
664e4d43b82Schristos 					goto out;
665d2a9b9efSchristos 				break;
666d2a9b9efSchristos 			}
667d2a9b9efSchristos 		}
668d2a9b9efSchristos 
669d2a9b9efSchristos 		/* Perform section-specific processing */
670d2a9b9efSchristos 		switch (state) {
671d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_LIST:
672e4d43b82Schristos 			for (i = 0; i < contig; i++) {
673e4d43b82Schristos 				if ((r = sshbuf_put_u64(sect, rs->lo + i)) != 0)
674e4d43b82Schristos 					goto out;
675e4d43b82Schristos 			}
676d2a9b9efSchristos 			break;
677d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_RANGE:
678e4d43b82Schristos 			if ((r = sshbuf_put_u64(sect, rs->lo)) != 0 ||
679e4d43b82Schristos 			    (r = sshbuf_put_u64(sect, rs->hi)) != 0)
680e4d43b82Schristos 				goto out;
681d2a9b9efSchristos 			break;
682d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_BITMAP:
683d2a9b9efSchristos 			if (rs->lo - bitmap_start > INT_MAX) {
68417418e98Schristos 				error_f("insane bitmap gap");
685d2a9b9efSchristos 				goto out;
686d2a9b9efSchristos 			}
687d2a9b9efSchristos 			for (i = 0; i < contig; i++) {
688e4d43b82Schristos 				if (bitmap_set_bit(bitmap,
689e4d43b82Schristos 				    rs->lo + i - bitmap_start) != 0) {
690e4d43b82Schristos 					r = SSH_ERR_ALLOC_FAIL;
691d2a9b9efSchristos 					goto out;
692d2a9b9efSchristos 				}
693e4d43b82Schristos 			}
694d2a9b9efSchristos 			break;
695d2a9b9efSchristos 		}
696d2a9b9efSchristos 		last = rs->hi;
697d2a9b9efSchristos 	}
698d2a9b9efSchristos 	/* Flush the remaining section, if any */
699d2a9b9efSchristos 	if (state != 0) {
70017418e98Schristos 		KRL_DBG(("serial final flush for state 0x%02x", state));
701d2a9b9efSchristos 		switch (state) {
702d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_LIST:
703d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_RANGE:
704d2a9b9efSchristos 			break;
705d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_BITMAP:
706e4d43b82Schristos 			if ((r = put_bitmap(sect, bitmap)) != 0)
707e4d43b82Schristos 				goto out;
708e4d43b82Schristos 			bitmap_free(bitmap);
709d2a9b9efSchristos 			bitmap = NULL;
710d2a9b9efSchristos 			break;
711d2a9b9efSchristos 		}
712e4d43b82Schristos 		if ((r = sshbuf_put_u8(buf, state)) != 0 ||
713e4d43b82Schristos 		    (r = sshbuf_put_stringb(buf, sect)) != 0)
714e4d43b82Schristos 			goto out;
715d2a9b9efSchristos 	}
71617418e98Schristos 	KRL_DBG(("serial done "));
717d2a9b9efSchristos 
718d2a9b9efSchristos 	/* Now output a section for any revocations by key ID */
719e4d43b82Schristos 	sshbuf_reset(sect);
720d2a9b9efSchristos 	RB_FOREACH(rki, revoked_key_id_tree, &rc->revoked_key_ids) {
72117418e98Schristos 		KRL_DBG(("key ID %s", rki->key_id));
722e4d43b82Schristos 		if ((r = sshbuf_put_cstring(sect, rki->key_id)) != 0)
723e4d43b82Schristos 			goto out;
724d2a9b9efSchristos 	}
725e4d43b82Schristos 	if (sshbuf_len(sect) != 0) {
726e4d43b82Schristos 		if ((r = sshbuf_put_u8(buf, KRL_SECTION_CERT_KEY_ID)) != 0 ||
727e4d43b82Schristos 		    (r = sshbuf_put_stringb(buf, sect)) != 0)
728e4d43b82Schristos 			goto out;
729d2a9b9efSchristos 	}
730d2a9b9efSchristos 	r = 0;
731d2a9b9efSchristos  out:
732e4d43b82Schristos 	bitmap_free(bitmap);
733e4d43b82Schristos 	sshbuf_free(sect);
734d2a9b9efSchristos 	return r;
735d2a9b9efSchristos }
736d2a9b9efSchristos 
737d2a9b9efSchristos int
ssh_krl_to_blob(struct ssh_krl * krl,struct sshbuf * buf)738*a629fefcSchristos ssh_krl_to_blob(struct ssh_krl *krl, struct sshbuf *buf)
739d2a9b9efSchristos {
740e4d43b82Schristos 	int r = SSH_ERR_INTERNAL_ERROR;
741d2a9b9efSchristos 	struct revoked_certs *rc;
742d2a9b9efSchristos 	struct revoked_blob *rb;
743e4d43b82Schristos 	struct sshbuf *sect;
744e4d43b82Schristos 	u_char *sblob = NULL;
745d2a9b9efSchristos 
746d2a9b9efSchristos 	if (krl->generated_date == 0)
747d2a9b9efSchristos 		krl->generated_date = time(NULL);
748d2a9b9efSchristos 
749e4d43b82Schristos 	if ((sect = sshbuf_new()) == NULL)
750e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
751d2a9b9efSchristos 
752d2a9b9efSchristos 	/* Store the header */
753e4d43b82Schristos 	if ((r = sshbuf_put(buf, KRL_MAGIC, sizeof(KRL_MAGIC) - 1)) != 0 ||
754e4d43b82Schristos 	    (r = sshbuf_put_u32(buf, KRL_FORMAT_VERSION)) != 0 ||
755e4d43b82Schristos 	    (r = sshbuf_put_u64(buf, krl->krl_version)) != 0 ||
75679976551Schristos 	    (r = sshbuf_put_u64(buf, krl->generated_date)) != 0 ||
757e4d43b82Schristos 	    (r = sshbuf_put_u64(buf, krl->flags)) != 0 ||
758e4d43b82Schristos 	    (r = sshbuf_put_string(buf, NULL, 0)) != 0 ||
759e4d43b82Schristos 	    (r = sshbuf_put_cstring(buf, krl->comment)) != 0)
760e4d43b82Schristos 		goto out;
761d2a9b9efSchristos 
762d2a9b9efSchristos 	/* Store sections for revoked certificates */
763d2a9b9efSchristos 	TAILQ_FOREACH(rc, &krl->revoked_certs, entry) {
764e4d43b82Schristos 		sshbuf_reset(sect);
765e4d43b82Schristos 		if ((r = revoked_certs_generate(rc, sect)) != 0)
766d2a9b9efSchristos 			goto out;
767e4d43b82Schristos 		if ((r = sshbuf_put_u8(buf, KRL_SECTION_CERTIFICATES)) != 0 ||
768e4d43b82Schristos 		    (r = sshbuf_put_stringb(buf, sect)) != 0)
769e4d43b82Schristos 			goto out;
770d2a9b9efSchristos 	}
771d2a9b9efSchristos 
772d2a9b9efSchristos 	/* Finally, output sections for revocations by public key/hash */
773e4d43b82Schristos 	sshbuf_reset(sect);
774d2a9b9efSchristos 	RB_FOREACH(rb, revoked_blob_tree, &krl->revoked_keys) {
77517418e98Schristos 		KRL_DBG(("key len %zu ", rb->len));
776e4d43b82Schristos 		if ((r = sshbuf_put_string(sect, rb->blob, rb->len)) != 0)
777e4d43b82Schristos 			goto out;
778d2a9b9efSchristos 	}
779e4d43b82Schristos 	if (sshbuf_len(sect) != 0) {
780e4d43b82Schristos 		if ((r = sshbuf_put_u8(buf, KRL_SECTION_EXPLICIT_KEY)) != 0 ||
781e4d43b82Schristos 		    (r = sshbuf_put_stringb(buf, sect)) != 0)
782e4d43b82Schristos 			goto out;
783d2a9b9efSchristos 	}
784e4d43b82Schristos 	sshbuf_reset(sect);
785d2a9b9efSchristos 	RB_FOREACH(rb, revoked_blob_tree, &krl->revoked_sha1s) {
78617418e98Schristos 		KRL_DBG(("hash len %zu ", rb->len));
787e4d43b82Schristos 		if ((r = sshbuf_put_string(sect, rb->blob, rb->len)) != 0)
788e4d43b82Schristos 			goto out;
789d2a9b9efSchristos 	}
790e4d43b82Schristos 	if (sshbuf_len(sect) != 0) {
791e4d43b82Schristos 		if ((r = sshbuf_put_u8(buf,
792e4d43b82Schristos 		    KRL_SECTION_FINGERPRINT_SHA1)) != 0 ||
793e4d43b82Schristos 		    (r = sshbuf_put_stringb(buf, sect)) != 0)
794e4d43b82Schristos 			goto out;
795d2a9b9efSchristos 	}
796aa36fcacSchristos 	sshbuf_reset(sect);
797aa36fcacSchristos 	RB_FOREACH(rb, revoked_blob_tree, &krl->revoked_sha256s) {
79817418e98Schristos 		KRL_DBG(("hash len %zu ", rb->len));
799aa36fcacSchristos 		if ((r = sshbuf_put_string(sect, rb->blob, rb->len)) != 0)
800aa36fcacSchristos 			goto out;
801aa36fcacSchristos 	}
802aa36fcacSchristos 	if (sshbuf_len(sect) != 0) {
803aa36fcacSchristos 		if ((r = sshbuf_put_u8(buf,
804aa36fcacSchristos 		    KRL_SECTION_FINGERPRINT_SHA256)) != 0 ||
805aa36fcacSchristos 		    (r = sshbuf_put_stringb(buf, sect)) != 0)
806aa36fcacSchristos 			goto out;
807aa36fcacSchristos 	}
808*a629fefcSchristos 	/* success */
809d2a9b9efSchristos 	r = 0;
810d2a9b9efSchristos  out:
811d2a9b9efSchristos 	free(sblob);
812e4d43b82Schristos 	sshbuf_free(sect);
813d2a9b9efSchristos 	return r;
814d2a9b9efSchristos }
815d2a9b9efSchristos 
816d2a9b9efSchristos static void
format_timestamp(u_int64_t timestamp,char * ts,size_t nts)817d2a9b9efSchristos format_timestamp(u_int64_t timestamp, char *ts, size_t nts)
818d2a9b9efSchristos {
819d2a9b9efSchristos 	time_t t;
820d2a9b9efSchristos 	struct tm *tm;
821d2a9b9efSchristos 
822d2a9b9efSchristos 	t = timestamp;
823d2a9b9efSchristos 	tm = localtime(&t);
824e4d43b82Schristos 	if (tm == NULL)
825e4d43b82Schristos 		strlcpy(ts, "<INVALID>", nts);
826e4d43b82Schristos 	else {
827d2a9b9efSchristos 		*ts = '\0';
828d2a9b9efSchristos 		strftime(ts, nts, "%Y%m%dT%H%M%S", tm);
829d2a9b9efSchristos 	}
830e4d43b82Schristos }
831d2a9b9efSchristos 
832d2a9b9efSchristos static int
cert_extension_subsection(struct sshbuf * subsect,struct ssh_krl * krl)833*a629fefcSchristos cert_extension_subsection(struct sshbuf *subsect, struct ssh_krl *krl)
834*a629fefcSchristos {
835*a629fefcSchristos 	int r = SSH_ERR_INTERNAL_ERROR;
836*a629fefcSchristos 	u_char critical = 1;
837*a629fefcSchristos 	struct sshbuf *value = NULL;
838*a629fefcSchristos 	char *name = NULL;
839*a629fefcSchristos 
840*a629fefcSchristos 	if ((r = sshbuf_get_cstring(subsect, &name, NULL)) != 0 ||
841*a629fefcSchristos 	    (r = sshbuf_get_u8(subsect, &critical)) != 0 ||
842*a629fefcSchristos 	    (r = sshbuf_froms(subsect, &value)) != 0) {
843*a629fefcSchristos 		debug_fr(r, "parse");
844*a629fefcSchristos 		error("KRL has invalid certificate extension subsection");
845*a629fefcSchristos 		r = SSH_ERR_INVALID_FORMAT;
846*a629fefcSchristos 		goto out;
847*a629fefcSchristos 	}
848*a629fefcSchristos 	if (sshbuf_len(subsect) != 0) {
849*a629fefcSchristos 		error("KRL has invalid certificate extension subsection: "
850*a629fefcSchristos 		    "trailing data");
851*a629fefcSchristos 		r = SSH_ERR_INVALID_FORMAT;
852*a629fefcSchristos 		goto out;
853*a629fefcSchristos 	}
854*a629fefcSchristos 	debug_f("cert extension %s critical %u len %zu",
855*a629fefcSchristos 	    name, critical, sshbuf_len(value));
856*a629fefcSchristos 	/* no extensions are currently supported */
857*a629fefcSchristos 	if (critical) {
858*a629fefcSchristos 		error("KRL contains unsupported critical certificate "
859*a629fefcSchristos 		    "subsection \"%s\"", name);
860*a629fefcSchristos 		r = SSH_ERR_FEATURE_UNSUPPORTED;
861*a629fefcSchristos 		goto out;
862*a629fefcSchristos 	}
863*a629fefcSchristos 	/* success */
864*a629fefcSchristos 	r = 0;
865*a629fefcSchristos  out:
866*a629fefcSchristos 	free(name);
867*a629fefcSchristos 	sshbuf_free(value);
868*a629fefcSchristos 	return r;
869*a629fefcSchristos }
870*a629fefcSchristos 
871*a629fefcSchristos static int
parse_revoked_certs(struct sshbuf * buf,struct ssh_krl * krl)872e4d43b82Schristos parse_revoked_certs(struct sshbuf *buf, struct ssh_krl *krl)
873d2a9b9efSchristos {
874e4d43b82Schristos 	int r = SSH_ERR_INTERNAL_ERROR;
875e4d43b82Schristos 	u_char type;
8768a4530f9Schristos 	const u_char *blob;
877e4d43b82Schristos 	size_t blen, nbits;
878e4d43b82Schristos 	struct sshbuf *subsect = NULL;
879d2a9b9efSchristos 	u_int64_t serial, serial_lo, serial_hi;
880e4d43b82Schristos 	struct bitmap *bitmap = NULL;
881d2a9b9efSchristos 	char *key_id = NULL;
882e4d43b82Schristos 	struct sshkey *ca_key = NULL;
883d2a9b9efSchristos 
884e4d43b82Schristos 	if ((subsect = sshbuf_new()) == NULL)
885e4d43b82Schristos 		return SSH_ERR_ALLOC_FAIL;
886d2a9b9efSchristos 
887e4d43b82Schristos 	/* Header: key, reserved */
888e4d43b82Schristos 	if ((r = sshbuf_get_string_direct(buf, &blob, &blen)) != 0 ||
889e4d43b82Schristos 	    (r = sshbuf_skip_string(buf)) != 0)
890d2a9b9efSchristos 		goto out;
891e4d43b82Schristos 	if (blen != 0 && (r = sshkey_from_blob(blob, blen, &ca_key)) != 0)
892e4d43b82Schristos 		goto out;
893e4d43b82Schristos 
894e4d43b82Schristos 	while (sshbuf_len(buf) > 0) {
895e4d43b82Schristos 		sshbuf_free(subsect);
896e4d43b82Schristos 		subsect = NULL;
897e4d43b82Schristos 		if ((r = sshbuf_get_u8(buf, &type)) != 0 ||
898e4d43b82Schristos 		    (r = sshbuf_froms(buf, &subsect)) != 0)
899d2a9b9efSchristos 			goto out;
90017418e98Schristos 		KRL_DBG(("subsection type 0x%02x", type));
901e4d43b82Schristos 		/* sshbuf_dump(subsect, stderr); */
902d2a9b9efSchristos 
903d2a9b9efSchristos 		switch (type) {
904d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_LIST:
905e4d43b82Schristos 			while (sshbuf_len(subsect) > 0) {
906e4d43b82Schristos 				if ((r = sshbuf_get_u64(subsect, &serial)) != 0)
907d2a9b9efSchristos 					goto out;
908e4d43b82Schristos 				if ((r = ssh_krl_revoke_cert_by_serial(krl,
909e4d43b82Schristos 				    ca_key, serial)) != 0)
910d2a9b9efSchristos 					goto out;
911d2a9b9efSchristos 			}
912d2a9b9efSchristos 			break;
913d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_RANGE:
914e4d43b82Schristos 			if ((r = sshbuf_get_u64(subsect, &serial_lo)) != 0 ||
915e4d43b82Schristos 			    (r = sshbuf_get_u64(subsect, &serial_hi)) != 0)
916d2a9b9efSchristos 				goto out;
917e4d43b82Schristos 			if ((r = ssh_krl_revoke_cert_by_serial_range(krl,
918e4d43b82Schristos 			    ca_key, serial_lo, serial_hi)) != 0)
919d2a9b9efSchristos 				goto out;
920d2a9b9efSchristos 			break;
921d2a9b9efSchristos 		case KRL_SECTION_CERT_SERIAL_BITMAP:
922e4d43b82Schristos 			if ((bitmap = bitmap_new()) == NULL) {
923e4d43b82Schristos 				r = SSH_ERR_ALLOC_FAIL;
924d2a9b9efSchristos 				goto out;
925d2a9b9efSchristos 			}
926e4d43b82Schristos 			if ((r = sshbuf_get_u64(subsect, &serial_lo)) != 0 ||
927e4d43b82Schristos 			    (r = sshbuf_get_bignum2_bytes_direct(subsect,
928e4d43b82Schristos 			    &blob, &blen)) != 0)
929e4d43b82Schristos 				goto out;
930e4d43b82Schristos 			if (bitmap_from_string(bitmap, blob, blen) != 0) {
931e4d43b82Schristos 				r = SSH_ERR_INVALID_FORMAT;
932d2a9b9efSchristos 				goto out;
933d2a9b9efSchristos 			}
934e4d43b82Schristos 			nbits = bitmap_nbits(bitmap);
935e4d43b82Schristos 			for (serial = 0; serial < (u_int64_t)nbits; serial++) {
936d2a9b9efSchristos 				if (serial > 0 && serial_lo + serial == 0) {
93717418e98Schristos 					error_f("bitmap wraps u64");
938e4d43b82Schristos 					r = SSH_ERR_INVALID_FORMAT;
939d2a9b9efSchristos 					goto out;
940d2a9b9efSchristos 				}
941e4d43b82Schristos 				if (!bitmap_test_bit(bitmap, serial))
942d2a9b9efSchristos 					continue;
943e4d43b82Schristos 				if ((r = ssh_krl_revoke_cert_by_serial(krl,
944e4d43b82Schristos 				    ca_key, serial_lo + serial)) != 0)
945d2a9b9efSchristos 					goto out;
946d2a9b9efSchristos 			}
947e4d43b82Schristos 			bitmap_free(bitmap);
948d2a9b9efSchristos 			bitmap = NULL;
949d2a9b9efSchristos 			break;
950d2a9b9efSchristos 		case KRL_SECTION_CERT_KEY_ID:
951e4d43b82Schristos 			while (sshbuf_len(subsect) > 0) {
952e4d43b82Schristos 				if ((r = sshbuf_get_cstring(subsect,
953e4d43b82Schristos 				    &key_id, NULL)) != 0)
954d2a9b9efSchristos 					goto out;
955e4d43b82Schristos 				if ((r = ssh_krl_revoke_cert_by_key_id(krl,
956e4d43b82Schristos 				    ca_key, key_id)) != 0)
957d2a9b9efSchristos 					goto out;
958d2a9b9efSchristos 				free(key_id);
959d2a9b9efSchristos 				key_id = NULL;
960d2a9b9efSchristos 			}
961d2a9b9efSchristos 			break;
962*a629fefcSchristos 		case KRL_SECTION_CERT_EXTENSION:
963*a629fefcSchristos 			if ((r = cert_extension_subsection(subsect, krl)) != 0)
964*a629fefcSchristos 				goto out;
965*a629fefcSchristos 			break;
966d2a9b9efSchristos 		default:
967d2a9b9efSchristos 			error("Unsupported KRL certificate section %u", type);
968e4d43b82Schristos 			r = SSH_ERR_INVALID_FORMAT;
969d2a9b9efSchristos 			goto out;
970d2a9b9efSchristos 		}
971e4d43b82Schristos 		if (sshbuf_len(subsect) > 0) {
972d2a9b9efSchristos 			error("KRL certificate section contains unparsed data");
973e4d43b82Schristos 			r = SSH_ERR_INVALID_FORMAT;
974d2a9b9efSchristos 			goto out;
975d2a9b9efSchristos 		}
976d2a9b9efSchristos 	}
977d2a9b9efSchristos 
978e4d43b82Schristos 	r = 0;
979d2a9b9efSchristos  out:
980d2a9b9efSchristos 	if (bitmap != NULL)
981e4d43b82Schristos 		bitmap_free(bitmap);
982d2a9b9efSchristos 	free(key_id);
983e4d43b82Schristos 	sshkey_free(ca_key);
984e4d43b82Schristos 	sshbuf_free(subsect);
985e4d43b82Schristos 	return r;
986d2a9b9efSchristos }
987d2a9b9efSchristos 
988aa36fcacSchristos static int
blob_section(struct sshbuf * sect,struct revoked_blob_tree * target_tree,size_t expected_len)989aa36fcacSchristos blob_section(struct sshbuf *sect, struct revoked_blob_tree *target_tree,
990aa36fcacSchristos     size_t expected_len)
991aa36fcacSchristos {
992aa36fcacSchristos 	u_char *rdata = NULL;
993aa36fcacSchristos 	size_t rlen = 0;
994aa36fcacSchristos 	int r;
995aa36fcacSchristos 
996aa36fcacSchristos 	while (sshbuf_len(sect) > 0) {
997aa36fcacSchristos 		if ((r = sshbuf_get_string(sect, &rdata, &rlen)) != 0)
998aa36fcacSchristos 			return r;
999aa36fcacSchristos 		if (expected_len != 0 && rlen != expected_len) {
100017418e98Schristos 			error_f("bad length");
1001aa36fcacSchristos 			free(rdata);
1002aa36fcacSchristos 			return SSH_ERR_INVALID_FORMAT;
1003aa36fcacSchristos 		}
1004aa36fcacSchristos 		if ((r = revoke_blob(target_tree, rdata, rlen)) != 0) {
1005aa36fcacSchristos 			free(rdata);
1006aa36fcacSchristos 			return r;
1007aa36fcacSchristos 		}
1008aa36fcacSchristos 	}
1009aa36fcacSchristos 	return 0;
1010aa36fcacSchristos }
1011d2a9b9efSchristos 
1012*a629fefcSchristos static int
extension_section(struct sshbuf * sect,struct ssh_krl * krl)1013*a629fefcSchristos extension_section(struct sshbuf *sect, struct ssh_krl *krl)
1014*a629fefcSchristos {
1015*a629fefcSchristos 	int r = SSH_ERR_INTERNAL_ERROR;
1016*a629fefcSchristos 	u_char critical = 1;
1017*a629fefcSchristos 	struct sshbuf *value = NULL;
1018*a629fefcSchristos 	char *name = NULL;
1019*a629fefcSchristos 
1020*a629fefcSchristos 	if ((r = sshbuf_get_cstring(sect, &name, NULL)) != 0 ||
1021*a629fefcSchristos 	    (r = sshbuf_get_u8(sect, &critical)) != 0 ||
1022*a629fefcSchristos 	    (r = sshbuf_froms(sect, &value)) != 0) {
1023*a629fefcSchristos 		debug_fr(r, "parse");
1024*a629fefcSchristos 		error("KRL has invalid extension section");
1025*a629fefcSchristos 		r = SSH_ERR_INVALID_FORMAT;
1026*a629fefcSchristos 		goto out;
1027*a629fefcSchristos 	}
1028*a629fefcSchristos 	if (sshbuf_len(sect) != 0) {
1029*a629fefcSchristos 		error("KRL has invalid extension section: trailing data");
1030*a629fefcSchristos 		r = SSH_ERR_INVALID_FORMAT;
1031*a629fefcSchristos 		goto out;
1032*a629fefcSchristos 	}
1033*a629fefcSchristos 	debug_f("extension %s critical %u len %zu",
1034*a629fefcSchristos 	    name, critical, sshbuf_len(value));
1035*a629fefcSchristos 	/* no extensions are currently supported */
1036*a629fefcSchristos 	if (critical) {
1037*a629fefcSchristos 		error("KRL contains unsupported critical section \"%s\"", name);
1038*a629fefcSchristos 		r = SSH_ERR_FEATURE_UNSUPPORTED;
1039*a629fefcSchristos 		goto out;
1040*a629fefcSchristos 	}
1041*a629fefcSchristos 	/* success */
1042*a629fefcSchristos 	r = 0;
1043*a629fefcSchristos  out:
1044*a629fefcSchristos 	free(name);
1045*a629fefcSchristos 	sshbuf_free(value);
1046*a629fefcSchristos 	return r;
1047*a629fefcSchristos }
1048*a629fefcSchristos 
1049*a629fefcSchristos /* Attempt to parse a KRL */
1050d2a9b9efSchristos int
ssh_krl_from_blob(struct sshbuf * buf,struct ssh_krl ** krlp)1051*a629fefcSchristos ssh_krl_from_blob(struct sshbuf *buf, struct ssh_krl **krlp)
1052d2a9b9efSchristos {
1053e4d43b82Schristos 	struct sshbuf *copy = NULL, *sect = NULL;
1054e4d43b82Schristos 	struct ssh_krl *krl = NULL;
1055d2a9b9efSchristos 	char timestamp[64];
1056*a629fefcSchristos 	int r = SSH_ERR_INTERNAL_ERROR;
1057aa36fcacSchristos 	u_char type;
1058e4d43b82Schristos 	u_int format_version;
1059d2a9b9efSchristos 
1060d2a9b9efSchristos 	*krlp = NULL;
1061*a629fefcSchristos 
1062*a629fefcSchristos 	/* KRL must begin with magic string */
1063*a629fefcSchristos 	if ((r = sshbuf_cmp(buf, 0, KRL_MAGIC, sizeof(KRL_MAGIC) - 1)) != 0) {
1064*a629fefcSchristos 		debug2_f("bad KRL magic header");
1065e4d43b82Schristos 		return SSH_ERR_KRL_BAD_MAGIC;
1066d2a9b9efSchristos 	}
1067d2a9b9efSchristos 
1068d2a9b9efSchristos 	if ((krl = ssh_krl_init()) == NULL) {
106917418e98Schristos 		error_f("alloc failed");
1070d2a9b9efSchristos 		goto out;
1071d2a9b9efSchristos 	}
1072*a629fefcSchristos 	/* Don't modify buffer */
1073*a629fefcSchristos 	if ((copy = sshbuf_fromb(buf)) == NULL) {
1074*a629fefcSchristos 		r = SSH_ERR_ALLOC_FAIL;
1075*a629fefcSchristos 		goto out;
1076*a629fefcSchristos 	}
1077*a629fefcSchristos 	if ((r = sshbuf_consume(copy, sizeof(KRL_MAGIC) - 1)) != 0 ||
1078*a629fefcSchristos 	    (r = sshbuf_get_u32(copy, &format_version)) != 0)
1079d2a9b9efSchristos 		goto out;
1080d2a9b9efSchristos 	if (format_version != KRL_FORMAT_VERSION) {
1081*a629fefcSchristos 		error_f("unsupported KRL format version %u", format_version);
1082e4d43b82Schristos 		r = SSH_ERR_INVALID_FORMAT;
1083d2a9b9efSchristos 		goto out;
1084d2a9b9efSchristos 	}
1085e4d43b82Schristos 	if ((r = sshbuf_get_u64(copy, &krl->krl_version)) != 0 ||
1086e4d43b82Schristos 	    (r = sshbuf_get_u64(copy, &krl->generated_date)) != 0 ||
1087e4d43b82Schristos 	    (r = sshbuf_get_u64(copy, &krl->flags)) != 0 ||
1088e4d43b82Schristos 	    (r = sshbuf_skip_string(copy)) != 0 ||
1089*a629fefcSchristos 	    (r = sshbuf_get_cstring(copy, &krl->comment, NULL)) != 0) {
1090*a629fefcSchristos 		error_fr(r, "parse KRL header");
1091d2a9b9efSchristos 		goto out;
1092*a629fefcSchristos 	}
1093d2a9b9efSchristos 	format_timestamp(krl->generated_date, timestamp, sizeof(timestamp));
109400a838c4Schristos 	debug("KRL version %llu generated at %s%s%s",
109500a838c4Schristos 	    (long long unsigned)krl->krl_version, timestamp,
109600a838c4Schristos 	    *krl->comment ? ": " : "", krl->comment);
1097d2a9b9efSchristos 
1098*a629fefcSchristos 	/* Parse and load the KRL sections. */
1099e4d43b82Schristos 	while (sshbuf_len(copy) > 0) {
1100e4d43b82Schristos 		sshbuf_free(sect);
1101e4d43b82Schristos 		sect = NULL;
1102e4d43b82Schristos 		if ((r = sshbuf_get_u8(copy, &type)) != 0 ||
1103e4d43b82Schristos 		    (r = sshbuf_froms(copy, &sect)) != 0)
1104e4d43b82Schristos 			goto out;
1105*a629fefcSchristos 		KRL_DBG(("section 0x%02x", type));
1106d2a9b9efSchristos 
1107d2a9b9efSchristos 		switch (type) {
1108d2a9b9efSchristos 		case KRL_SECTION_CERTIFICATES:
1109e4d43b82Schristos 			if ((r = parse_revoked_certs(sect, krl)) != 0)
1110d2a9b9efSchristos 				goto out;
1111d2a9b9efSchristos 			break;
1112d2a9b9efSchristos 		case KRL_SECTION_EXPLICIT_KEY:
1113aa36fcacSchristos 			if ((r = blob_section(sect,
1114aa36fcacSchristos 			    &krl->revoked_keys, 0)) != 0)
1115aa36fcacSchristos 				goto out;
1116aa36fcacSchristos 			break;
1117d2a9b9efSchristos 		case KRL_SECTION_FINGERPRINT_SHA1:
1118aa36fcacSchristos 			if ((r = blob_section(sect,
1119aa36fcacSchristos 			    &krl->revoked_sha1s, 20)) != 0)
1120d2a9b9efSchristos 				goto out;
1121aa36fcacSchristos 			break;
1122aa36fcacSchristos 		case KRL_SECTION_FINGERPRINT_SHA256:
1123aa36fcacSchristos 			if ((r = blob_section(sect,
1124aa36fcacSchristos 			    &krl->revoked_sha256s, 32)) != 0)
1125d2a9b9efSchristos 				goto out;
1126d2a9b9efSchristos 			break;
1127*a629fefcSchristos 		case KRL_SECTION_EXTENSION:
1128*a629fefcSchristos 			if ((r = extension_section(sect, krl)) != 0)
1129*a629fefcSchristos 				goto out;
1130*a629fefcSchristos 			break;
1131d2a9b9efSchristos 		case KRL_SECTION_SIGNATURE:
1132d2a9b9efSchristos 			/* Handled above, but still need to stay in synch */
113341768fc1Schristos 			sshbuf_free(sect);
1134e4d43b82Schristos 			sect = NULL;
1135e4d43b82Schristos 			if ((r = sshbuf_skip_string(copy)) != 0)
1136d2a9b9efSchristos 				goto out;
1137d2a9b9efSchristos 			break;
1138d2a9b9efSchristos 		default:
1139d2a9b9efSchristos 			error("Unsupported KRL section %u", type);
1140e4d43b82Schristos 			r = SSH_ERR_INVALID_FORMAT;
1141d2a9b9efSchristos 			goto out;
1142d2a9b9efSchristos 		}
114379976551Schristos 		if (sect != NULL && sshbuf_len(sect) > 0) {
1144d2a9b9efSchristos 			error("KRL section contains unparsed data");
1145e4d43b82Schristos 			r = SSH_ERR_INVALID_FORMAT;
1146d2a9b9efSchristos 			goto out;
1147d2a9b9efSchristos 		}
1148d2a9b9efSchristos 	}
1149d2a9b9efSchristos 
1150*a629fefcSchristos 	/* Success */
1151d2a9b9efSchristos 	*krlp = krl;
1152e4d43b82Schristos 	r = 0;
1153d2a9b9efSchristos  out:
1154e4d43b82Schristos 	if (r != 0)
1155d2a9b9efSchristos 		ssh_krl_free(krl);
1156e4d43b82Schristos 	sshbuf_free(copy);
1157e4d43b82Schristos 	sshbuf_free(sect);
1158e4d43b82Schristos 	return r;
1159d2a9b9efSchristos }
1160d2a9b9efSchristos 
1161e4d43b82Schristos /* Checks certificate serial number and key ID revocation */
1162d2a9b9efSchristos static int
is_cert_revoked(const struct sshkey * key,struct revoked_certs * rc)1163e4d43b82Schristos is_cert_revoked(const struct sshkey *key, struct revoked_certs *rc)
1164d2a9b9efSchristos {
1165d2a9b9efSchristos 	struct revoked_serial rs, *ers;
1166d2a9b9efSchristos 	struct revoked_key_id rki, *erki;
1167d2a9b9efSchristos 
1168d2a9b9efSchristos 	/* Check revocation by cert key ID */
11698a4530f9Schristos 	memset(&rki, 0, sizeof(rki));
1170d2a9b9efSchristos 	rki.key_id = key->cert->key_id;
1171d2a9b9efSchristos 	erki = RB_FIND(revoked_key_id_tree, &rc->revoked_key_ids, &rki);
1172d2a9b9efSchristos 	if (erki != NULL) {
117317418e98Schristos 		KRL_DBG(("revoked by key ID"));
1174e4d43b82Schristos 		return SSH_ERR_KEY_REVOKED;
1175d2a9b9efSchristos 	}
1176d2a9b9efSchristos 
1177d2a9b9efSchristos 	/*
11788395c133Schristos 	 * Zero serials numbers are ignored (it's the default when the
11798395c133Schristos 	 * CA doesn't specify one).
1180d2a9b9efSchristos 	 */
11818395c133Schristos 	if (key->cert->serial == 0)
1182d2a9b9efSchristos 		return 0;
1183d2a9b9efSchristos 
11848a4530f9Schristos 	memset(&rs, 0, sizeof(rs));
1185d2a9b9efSchristos 	rs.lo = rs.hi = key->cert->serial;
1186d2a9b9efSchristos 	ers = RB_FIND(revoked_serial_tree, &rc->revoked_serials, &rs);
1187d2a9b9efSchristos 	if (ers != NULL) {
118817418e98Schristos 		KRL_DBG(("revoked serial %llu matched %llu:%llu",
1189d2a9b9efSchristos 		    key->cert->serial, ers->lo, ers->hi));
1190e4d43b82Schristos 		return SSH_ERR_KEY_REVOKED;
1191d2a9b9efSchristos 	}
1192e4d43b82Schristos 	return 0;
1193e4d43b82Schristos }
1194d2a9b9efSchristos 
1195e4d43b82Schristos /* Checks whether a given key/cert is revoked. Does not check its CA */
1196e4d43b82Schristos static int
is_key_revoked(struct ssh_krl * krl,const struct sshkey * key)1197e4d43b82Schristos is_key_revoked(struct ssh_krl *krl, const struct sshkey *key)
1198e4d43b82Schristos {
1199e4d43b82Schristos 	struct revoked_blob rb, *erb;
1200e4d43b82Schristos 	struct revoked_certs *rc;
1201e4d43b82Schristos 	int r;
1202e4d43b82Schristos 
1203e4d43b82Schristos 	/* Check explicitly revoked hashes first */
1204e4d43b82Schristos 	memset(&rb, 0, sizeof(rb));
1205e4d43b82Schristos 	if ((r = sshkey_fingerprint_raw(key, SSH_DIGEST_SHA1,
1206e4d43b82Schristos 	    &rb.blob, &rb.len)) != 0)
1207e4d43b82Schristos 		return r;
1208e4d43b82Schristos 	erb = RB_FIND(revoked_blob_tree, &krl->revoked_sha1s, &rb);
1209e4d43b82Schristos 	free(rb.blob);
1210e4d43b82Schristos 	if (erb != NULL) {
121117418e98Schristos 		KRL_DBG(("revoked by key SHA1"));
1212e4d43b82Schristos 		return SSH_ERR_KEY_REVOKED;
1213e4d43b82Schristos 	}
1214aa36fcacSchristos 	memset(&rb, 0, sizeof(rb));
1215aa36fcacSchristos 	if ((r = sshkey_fingerprint_raw(key, SSH_DIGEST_SHA256,
1216aa36fcacSchristos 	    &rb.blob, &rb.len)) != 0)
1217aa36fcacSchristos 		return r;
1218aa36fcacSchristos 	erb = RB_FIND(revoked_blob_tree, &krl->revoked_sha256s, &rb);
1219aa36fcacSchristos 	free(rb.blob);
1220aa36fcacSchristos 	if (erb != NULL) {
122117418e98Schristos 		KRL_DBG(("revoked by key SHA256"));
1222aa36fcacSchristos 		return SSH_ERR_KEY_REVOKED;
1223aa36fcacSchristos 	}
1224e4d43b82Schristos 
1225e4d43b82Schristos 	/* Next, explicit keys */
1226e4d43b82Schristos 	memset(&rb, 0, sizeof(rb));
1227e4d43b82Schristos 	if ((r = plain_key_blob(key, &rb.blob, &rb.len)) != 0)
1228e4d43b82Schristos 		return r;
1229e4d43b82Schristos 	erb = RB_FIND(revoked_blob_tree, &krl->revoked_keys, &rb);
1230e4d43b82Schristos 	free(rb.blob);
1231e4d43b82Schristos 	if (erb != NULL) {
123217418e98Schristos 		KRL_DBG(("revoked by explicit key"));
1233e4d43b82Schristos 		return SSH_ERR_KEY_REVOKED;
1234e4d43b82Schristos 	}
1235e4d43b82Schristos 
1236e4d43b82Schristos 	if (!sshkey_is_cert(key))
1237e4d43b82Schristos 		return 0;
1238e4d43b82Schristos 
1239e4d43b82Schristos 	/* Check cert revocation for the specified CA */
1240e4d43b82Schristos 	if ((r = revoked_certs_for_ca_key(krl, key->cert->signature_key,
1241e4d43b82Schristos 	    &rc, 0)) != 0)
1242e4d43b82Schristos 		return r;
1243e4d43b82Schristos 	if (rc != NULL) {
1244e4d43b82Schristos 		if ((r = is_cert_revoked(key, rc)) != 0)
1245e4d43b82Schristos 			return r;
1246e4d43b82Schristos 	}
1247e4d43b82Schristos 	/* Check cert revocation for the wildcard CA */
1248e4d43b82Schristos 	if ((r = revoked_certs_for_ca_key(krl, NULL, &rc, 0)) != 0)
1249e4d43b82Schristos 		return r;
1250e4d43b82Schristos 	if (rc != NULL) {
1251e4d43b82Schristos 		if ((r = is_cert_revoked(key, rc)) != 0)
1252e4d43b82Schristos 			return r;
1253e4d43b82Schristos 	}
1254e4d43b82Schristos 
125517418e98Schristos 	KRL_DBG(("%llu no match", key->cert->serial));
1256d2a9b9efSchristos 	return 0;
1257d2a9b9efSchristos }
1258d2a9b9efSchristos 
1259d2a9b9efSchristos int
ssh_krl_check_key(struct ssh_krl * krl,const struct sshkey * key)1260e4d43b82Schristos ssh_krl_check_key(struct ssh_krl *krl, const struct sshkey *key)
1261d2a9b9efSchristos {
1262d2a9b9efSchristos 	int r;
1263d2a9b9efSchristos 
126417418e98Schristos 	KRL_DBG(("checking key"));
1265d2a9b9efSchristos 	if ((r = is_key_revoked(krl, key)) != 0)
1266d2a9b9efSchristos 		return r;
1267e4d43b82Schristos 	if (sshkey_is_cert(key)) {
126817418e98Schristos 		debug2_f("checking CA key");
1269d2a9b9efSchristos 		if ((r = is_key_revoked(krl, key->cert->signature_key)) != 0)
1270d2a9b9efSchristos 			return r;
1271d2a9b9efSchristos 	}
127217418e98Schristos 	KRL_DBG(("key okay"));
1273d2a9b9efSchristos 	return 0;
1274d2a9b9efSchristos }
1275d2a9b9efSchristos 
1276d2a9b9efSchristos int
ssh_krl_file_contains_key(const char * path,const struct sshkey * key)1277e4d43b82Schristos ssh_krl_file_contains_key(const char *path, const struct sshkey *key)
1278d2a9b9efSchristos {
1279e4d43b82Schristos 	struct sshbuf *krlbuf = NULL;
1280e4d43b82Schristos 	struct ssh_krl *krl = NULL;
1281ed75d7a8Schristos 	int oerrno = 0, r;
1282d2a9b9efSchristos 
1283d2a9b9efSchristos 	if (path == NULL)
1284d2a9b9efSchristos 		return 0;
1285ed75d7a8Schristos 	if ((r = sshbuf_load_file(path, &krlbuf)) != 0) {
1286e4d43b82Schristos 		oerrno = errno;
1287e4d43b82Schristos 		goto out;
1288d2a9b9efSchristos 	}
1289*a629fefcSchristos 	if ((r = ssh_krl_from_blob(krlbuf, &krl)) != 0)
1290e4d43b82Schristos 		goto out;
129117418e98Schristos 	debug2_f("checking KRL %s", path);
1292e4d43b82Schristos 	r = ssh_krl_check_key(krl, key);
1293e4d43b82Schristos  out:
1294e4d43b82Schristos 	sshbuf_free(krlbuf);
1295d2a9b9efSchristos 	ssh_krl_free(krl);
1296e4d43b82Schristos 	if (r != 0)
1297e4d43b82Schristos 		errno = oerrno;
1298e4d43b82Schristos 	return r;
1299d2a9b9efSchristos }
13008db691beSchristos 
13018db691beSchristos int
krl_dump(struct ssh_krl * krl,FILE * f)13028db691beSchristos krl_dump(struct ssh_krl *krl, FILE *f)
13038db691beSchristos {
13048db691beSchristos 	struct sshkey *key = NULL;
13058db691beSchristos 	struct revoked_blob *rb;
13068db691beSchristos 	struct revoked_certs *rc;
13078db691beSchristos 	struct revoked_serial *rs;
13088db691beSchristos 	struct revoked_key_id *rki;
13098db691beSchristos 	int r, ret = 0;
13108db691beSchristos 	char *fp, timestamp[64];
13118db691beSchristos 
13128db691beSchristos 	/* Try to print in a KRL spec-compatible format */
13138db691beSchristos 	format_timestamp(krl->generated_date, timestamp, sizeof(timestamp));
13148db691beSchristos 	fprintf(f, "# KRL version %llu\n",
13158db691beSchristos 	    (unsigned long long)krl->krl_version);
13168db691beSchristos 	fprintf(f, "# Generated at %s\n", timestamp);
13178db691beSchristos 	if (krl->comment != NULL && *krl->comment != '\0') {
13188db691beSchristos 		r = INT_MAX;
13198db691beSchristos 		asmprintf(&fp, INT_MAX, &r, "%s", krl->comment);
13208db691beSchristos 		fprintf(f, "# Comment: %s\n", fp);
13218db691beSchristos 		free(fp);
13228db691beSchristos 	}
13238db691beSchristos 	fputc('\n', f);
13248db691beSchristos 
13258db691beSchristos 	RB_FOREACH(rb, revoked_blob_tree, &krl->revoked_keys) {
13268db691beSchristos 		if ((r = sshkey_from_blob(rb->blob, rb->len, &key)) != 0) {
13278db691beSchristos 			ret = SSH_ERR_INVALID_FORMAT;
132817418e98Schristos 			error_r(r, "parse KRL key");
13298db691beSchristos 			continue;
13308db691beSchristos 		}
13318db691beSchristos 		if ((fp = sshkey_fingerprint(key, SSH_FP_HASH_DEFAULT,
13328db691beSchristos 		    SSH_FP_DEFAULT)) == NULL) {
13338db691beSchristos 			ret = SSH_ERR_INVALID_FORMAT;
13348db691beSchristos 			error("sshkey_fingerprint failed");
13358db691beSchristos 			continue;
13368db691beSchristos 		}
1337e160b4e8Schristos 		fprintf(f, "hash: %s # %s\n", fp, sshkey_ssh_name(key));
13388db691beSchristos 		free(fp);
13398db691beSchristos 		free(key);
13408db691beSchristos 	}
13418db691beSchristos 	RB_FOREACH(rb, revoked_blob_tree, &krl->revoked_sha256s) {
13428db691beSchristos 		fp = tohex(rb->blob, rb->len);
13438db691beSchristos 		fprintf(f, "hash: SHA256:%s\n", fp);
13448db691beSchristos 		free(fp);
13458db691beSchristos 	}
13468db691beSchristos 	RB_FOREACH(rb, revoked_blob_tree, &krl->revoked_sha1s) {
13478db691beSchristos 		/*
13488db691beSchristos 		 * There is not KRL spec keyword for raw SHA1 hashes, so
13498db691beSchristos 		 * print them as comments.
13508db691beSchristos 		 */
13518db691beSchristos 		fp = tohex(rb->blob, rb->len);
13528db691beSchristos 		fprintf(f, "# hash SHA1:%s\n", fp);
13538db691beSchristos 		free(fp);
13548db691beSchristos 	}
13558db691beSchristos 
13568db691beSchristos 	TAILQ_FOREACH(rc, &krl->revoked_certs, entry) {
13578db691beSchristos 		fputc('\n', f);
13588db691beSchristos 		if (rc->ca_key == NULL)
13598db691beSchristos 			fprintf(f, "# Wildcard CA\n");
13608db691beSchristos 		else {
13618db691beSchristos 			if ((fp = sshkey_fingerprint(rc->ca_key,
13628db691beSchristos 			    SSH_FP_HASH_DEFAULT, SSH_FP_DEFAULT)) == NULL) {
13638db691beSchristos 				ret = SSH_ERR_INVALID_FORMAT;
13648db691beSchristos 				error("sshkey_fingerprint failed");
13658db691beSchristos 				continue;
13668db691beSchristos 			}
13678db691beSchristos 			fprintf(f, "# CA key %s %s\n",
13688db691beSchristos 			    sshkey_ssh_name(rc->ca_key), fp);
13698db691beSchristos 			free(fp);
13708db691beSchristos 		}
13718db691beSchristos 		RB_FOREACH(rs, revoked_serial_tree, &rc->revoked_serials) {
13728db691beSchristos 			if (rs->lo == rs->hi) {
13738db691beSchristos 				fprintf(f, "serial: %llu\n",
13748db691beSchristos 				    (unsigned long long)rs->lo);
13758db691beSchristos 			} else {
13768db691beSchristos 				fprintf(f, "serial: %llu-%llu\n",
13778db691beSchristos 				    (unsigned long long)rs->lo,
13788db691beSchristos 				    (unsigned long long)rs->hi);
13798db691beSchristos 			}
13808db691beSchristos 		}
13818db691beSchristos 		RB_FOREACH(rki, revoked_key_id_tree, &rc->revoked_key_ids) {
13828db691beSchristos 			/*
13838db691beSchristos 			 * We don't want key IDs with embedded newlines to
13848db691beSchristos 			 * mess up the display.
13858db691beSchristos 			 */
13868db691beSchristos 			r = INT_MAX;
13878db691beSchristos 			asmprintf(&fp, INT_MAX, &r, "%s", rki->key_id);
13888db691beSchristos 			fprintf(f, "id: %s\n", fp);
13898db691beSchristos 			free(fp);
13908db691beSchristos 		}
13918db691beSchristos 	}
13928db691beSchristos 	return ret;
13938db691beSchristos }
1394