1/*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2008 Isilon Inc http://www.isilon.com/
5 * Authors: Doug Rabson <dfr@rabson.org>
6 * Developed with Red Inc: Alfred Perlstein <alfred@freebsd.org>
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30#include <sys/cdefs.h>
31#include "opt_inet6.h"
32
33#include <sys/param.h>
34#include <sys/kernel.h>
35#include <sys/kobj.h>
36#include <sys/lock.h>
37#include <sys/malloc.h>
38#include <sys/mbuf.h>
39#include <sys/module.h>
40#include <sys/mutex.h>
41#include <kgssapi/gssapi.h>
42#include <kgssapi/gssapi_impl.h>
43
44#include "kgss_if.h"
45#include "kcrypto.h"
46
47#define GSS_TOKEN_SENT_BY_ACCEPTOR	1
48#define GSS_TOKEN_SEALED		2
49#define GSS_TOKEN_ACCEPTOR_SUBKEY	4
50
51static gss_OID_desc krb5_mech_oid =
52{9, (void *) "\x2a\x86\x48\x86\xf7\x12\x01\x02\x02" };
53
54struct krb5_data {
55	size_t		kd_length;
56	void		*kd_data;
57};
58
59struct krb5_keyblock {
60	uint16_t	kk_type; /* encryption type */
61	struct krb5_data kk_key; /* key data */
62};
63
64struct krb5_address {
65	uint16_t	ka_type;
66	struct krb5_data ka_addr;
67};
68
69/*
70 * The km_elem array is ordered so that the highest received sequence
71 * number is listed first.
72 */
73struct krb5_msg_order {
74	uint32_t		km_flags;
75	uint32_t		km_start;
76	uint32_t		km_length;
77	uint32_t		km_jitter_window;
78	uint32_t		km_first_seq;
79	uint32_t		*km_elem;
80};
81
82struct krb5_context {
83	struct _gss_ctx_id_t	kc_common;
84	struct mtx		kc_lock;
85	uint32_t		kc_ac_flags;
86	uint32_t		kc_ctx_flags;
87	uint32_t		kc_more_flags;
88#define LOCAL			1
89#define OPEN			2
90#define COMPAT_OLD_DES3		4
91#define COMPAT_OLD_DES3_SELECTED 8
92#define ACCEPTOR_SUBKEY		16
93	struct krb5_address	kc_local_address;
94	struct krb5_address	kc_remote_address;
95	uint16_t		kc_local_port;
96	uint16_t		kc_remote_port;
97	struct krb5_keyblock	kc_keyblock;
98	struct krb5_keyblock	kc_local_subkey;
99	struct krb5_keyblock	kc_remote_subkey;
100	volatile uint32_t	kc_local_seqnumber;
101	uint32_t		kc_remote_seqnumber;
102	uint32_t		kc_keytype;
103	uint32_t		kc_cksumtype;
104	struct krb5_data	kc_source_name;
105	struct krb5_data	kc_target_name;
106	uint32_t		kc_lifetime;
107	struct krb5_msg_order	kc_msg_order;
108	struct krb5_key_state	*kc_tokenkey;
109	struct krb5_key_state	*kc_encryptkey;
110	struct krb5_key_state	*kc_checksumkey;
111
112	struct krb5_key_state	*kc_send_seal_Ke;
113	struct krb5_key_state	*kc_send_seal_Ki;
114	struct krb5_key_state	*kc_send_seal_Kc;
115	struct krb5_key_state	*kc_send_sign_Kc;
116
117	struct krb5_key_state	*kc_recv_seal_Ke;
118	struct krb5_key_state	*kc_recv_seal_Ki;
119	struct krb5_key_state	*kc_recv_seal_Kc;
120	struct krb5_key_state	*kc_recv_sign_Kc;
121};
122
123static uint16_t
124get_uint16(const uint8_t **pp, size_t *lenp)
125{
126	const uint8_t *p = *pp;
127	uint16_t v;
128
129	if (*lenp < 2)
130		return (0);
131
132	v = (p[0] << 8) | p[1];
133	*pp = p + 2;
134	*lenp = *lenp - 2;
135
136	return (v);
137}
138
139static uint32_t
140get_uint32(const uint8_t **pp, size_t *lenp)
141{
142	const uint8_t *p = *pp;
143	uint32_t v;
144
145	if (*lenp < 4)
146		return (0);
147
148	v = (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
149	*pp = p + 4;
150	*lenp = *lenp - 4;
151
152	return (v);
153}
154
155static void
156get_data(const uint8_t **pp, size_t *lenp, struct krb5_data *dp)
157{
158	size_t sz = get_uint32(pp, lenp);
159
160	dp->kd_length = sz;
161	dp->kd_data = malloc(sz, M_GSSAPI, M_WAITOK);
162
163	if (*lenp < sz)
164		sz = *lenp;
165	bcopy(*pp, dp->kd_data, sz);
166	(*pp) += sz;
167	(*lenp) -= sz;
168}
169
170static void
171delete_data(struct krb5_data *dp)
172{
173	if (dp->kd_data) {
174		free(dp->kd_data, M_GSSAPI);
175		dp->kd_length = 0;
176		dp->kd_data = NULL;
177	}
178}
179
180static void
181get_address(const uint8_t **pp, size_t *lenp, struct krb5_address *ka)
182{
183
184	ka->ka_type = get_uint16(pp, lenp);
185	get_data(pp, lenp, &ka->ka_addr);
186}
187
188static void
189delete_address(struct krb5_address *ka)
190{
191	delete_data(&ka->ka_addr);
192}
193
194static void
195get_keyblock(const uint8_t **pp, size_t *lenp, struct krb5_keyblock *kk)
196{
197
198	kk->kk_type = get_uint16(pp, lenp);
199	get_data(pp, lenp, &kk->kk_key);
200}
201
202static void
203delete_keyblock(struct krb5_keyblock *kk)
204{
205	if (kk->kk_key.kd_data)
206		bzero(kk->kk_key.kd_data, kk->kk_key.kd_length);
207	delete_data(&kk->kk_key);
208}
209
210static void
211copy_key(struct krb5_keyblock *from, struct krb5_keyblock **to)
212{
213
214	if (from->kk_key.kd_length)
215		*to = from;
216	else
217		*to = NULL;
218}
219
220/*
221 * Return non-zero if we are initiator.
222 */
223static __inline int
224is_initiator(struct krb5_context *kc)
225{
226	return (kc->kc_more_flags & LOCAL);
227}
228
229/*
230 * Return non-zero if we are acceptor.
231 */
232static __inline int
233is_acceptor(struct krb5_context *kc)
234{
235	return !(kc->kc_more_flags & LOCAL);
236}
237
238static void
239get_initiator_subkey(struct krb5_context *kc, struct krb5_keyblock **kdp)
240{
241
242	if (is_initiator(kc))
243		copy_key(&kc->kc_local_subkey, kdp);
244	else
245		copy_key(&kc->kc_remote_subkey, kdp);
246	if (!*kdp)
247		copy_key(&kc->kc_keyblock, kdp);
248}
249
250static void
251get_acceptor_subkey(struct krb5_context *kc, struct krb5_keyblock **kdp)
252{
253
254	if (is_initiator(kc))
255		copy_key(&kc->kc_remote_subkey, kdp);
256	else
257		copy_key(&kc->kc_local_subkey, kdp);
258}
259
260static OM_uint32
261get_keys(struct krb5_context *kc)
262{
263	struct krb5_keyblock *keydata;
264	struct krb5_encryption_class *ec;
265	struct krb5_key_state *key;
266	int etype;
267
268	keydata = NULL;
269	get_acceptor_subkey(kc, &keydata);
270	if (!keydata)
271		if ((kc->kc_more_flags & ACCEPTOR_SUBKEY) == 0)
272			get_initiator_subkey(kc, &keydata);
273	if (!keydata)
274		return (GSS_S_FAILURE);
275
276	/*
277	 * GSS-API treats all DES etypes the same and all DES3 etypes
278	 * the same.
279	 */
280	switch (keydata->kk_type) {
281	case ETYPE_DES_CBC_CRC:
282	case ETYPE_DES_CBC_MD4:
283	case ETYPE_DES_CBC_MD5:
284		etype = ETYPE_DES_CBC_CRC;
285		break;
286
287	case ETYPE_DES3_CBC_MD5:
288	case ETYPE_DES3_CBC_SHA1:
289	case ETYPE_OLD_DES3_CBC_SHA1:
290		etype = ETYPE_DES3_CBC_SHA1;
291		break;
292
293	default:
294		etype = keydata->kk_type;
295	}
296
297	ec = krb5_find_encryption_class(etype);
298	if (!ec)
299		return (GSS_S_FAILURE);
300
301	key = krb5_create_key(ec);
302	krb5_set_key(key, keydata->kk_key.kd_data);
303	kc->kc_tokenkey = key;
304
305	switch (etype) {
306	case ETYPE_DES_CBC_CRC:
307	case ETYPE_ARCFOUR_HMAC_MD5:
308	case ETYPE_ARCFOUR_HMAC_MD5_56: {
309		/*
310		 * Single DES and ARCFOUR uses a 'derived' key (XOR
311		 * with 0xf0) for encrypting wrap tokens. The original
312		 * key is used for checksums and sequence numbers.
313		 */
314		struct krb5_key_state *ekey;
315		uint8_t *ekp, *kp;
316		int i;
317
318		ekey = krb5_create_key(ec);
319		ekp = ekey->ks_key;
320		kp = key->ks_key;
321		for (i = 0; i < ec->ec_keylen; i++)
322			ekp[i] = kp[i] ^ 0xf0;
323		krb5_set_key(ekey, ekp);
324		kc->kc_encryptkey = ekey;
325		refcount_acquire(&key->ks_refs);
326		kc->kc_checksumkey = key;
327		break;
328	}
329
330	case ETYPE_DES3_CBC_SHA1:
331		/*
332		 * Triple DES uses a RFC 3961 style derived key with
333		 * usage number KG_USAGE_SIGN for checksums. The
334		 * original key is used for encryption and sequence
335		 * numbers.
336		 */
337		kc->kc_checksumkey = krb5_get_checksum_key(key, KG_USAGE_SIGN);
338		refcount_acquire(&key->ks_refs);
339		kc->kc_encryptkey = key;
340		break;
341
342	default:
343		/*
344		 * We need eight derived keys four for sending and
345		 * four for receiving.
346		 */
347		if (is_initiator(kc)) {
348			/*
349			 * We are initiator.
350			 */
351			kc->kc_send_seal_Ke = krb5_get_encryption_key(key,
352			    KG_USAGE_INITIATOR_SEAL);
353			kc->kc_send_seal_Ki = krb5_get_integrity_key(key,
354			    KG_USAGE_INITIATOR_SEAL);
355			kc->kc_send_seal_Kc = krb5_get_checksum_key(key,
356			    KG_USAGE_INITIATOR_SEAL);
357			kc->kc_send_sign_Kc = krb5_get_checksum_key(key,
358			    KG_USAGE_INITIATOR_SIGN);
359
360			kc->kc_recv_seal_Ke = krb5_get_encryption_key(key,
361			    KG_USAGE_ACCEPTOR_SEAL);
362			kc->kc_recv_seal_Ki = krb5_get_integrity_key(key,
363			    KG_USAGE_ACCEPTOR_SEAL);
364			kc->kc_recv_seal_Kc = krb5_get_checksum_key(key,
365			    KG_USAGE_ACCEPTOR_SEAL);
366			kc->kc_recv_sign_Kc = krb5_get_checksum_key(key,
367			    KG_USAGE_ACCEPTOR_SIGN);
368		} else {
369			/*
370			 * We are acceptor.
371			 */
372			kc->kc_send_seal_Ke = krb5_get_encryption_key(key,
373			    KG_USAGE_ACCEPTOR_SEAL);
374			kc->kc_send_seal_Ki = krb5_get_integrity_key(key,
375			    KG_USAGE_ACCEPTOR_SEAL);
376			kc->kc_send_seal_Kc = krb5_get_checksum_key(key,
377			    KG_USAGE_ACCEPTOR_SEAL);
378			kc->kc_send_sign_Kc = krb5_get_checksum_key(key,
379			    KG_USAGE_ACCEPTOR_SIGN);
380
381			kc->kc_recv_seal_Ke = krb5_get_encryption_key(key,
382			    KG_USAGE_INITIATOR_SEAL);
383			kc->kc_recv_seal_Ki = krb5_get_integrity_key(key,
384			    KG_USAGE_INITIATOR_SEAL);
385			kc->kc_recv_seal_Kc = krb5_get_checksum_key(key,
386			    KG_USAGE_INITIATOR_SEAL);
387			kc->kc_recv_sign_Kc = krb5_get_checksum_key(key,
388			    KG_USAGE_INITIATOR_SIGN);
389		}
390		break;
391	}
392
393	return (GSS_S_COMPLETE);
394}
395
396static void
397krb5_init(gss_ctx_id_t ctx)
398{
399	struct krb5_context *kc = (struct krb5_context *)ctx;
400
401	mtx_init(&kc->kc_lock, "krb5 gss lock", NULL, MTX_DEF);
402}
403
404static OM_uint32
405krb5_import(gss_ctx_id_t ctx,
406    enum sec_context_format format,
407    const gss_buffer_t context_token)
408{
409	struct krb5_context *kc = (struct krb5_context *)ctx;
410	OM_uint32 res;
411	const uint8_t *p = (const uint8_t *) context_token->value;
412	size_t len = context_token->length;
413	uint32_t flags;
414	int i;
415
416	/*
417	 * We support heimdal 0.6 and heimdal 1.1
418	 */
419	if (format != KGSS_HEIMDAL_0_6 && format != KGSS_HEIMDAL_1_1)
420		return (GSS_S_DEFECTIVE_TOKEN);
421
422#define SC_LOCAL_ADDRESS	1
423#define SC_REMOTE_ADDRESS	2
424#define SC_KEYBLOCK		4
425#define SC_LOCAL_SUBKEY		8
426#define SC_REMOTE_SUBKEY	16
427
428	/*
429	 * Ensure that the token starts with krb5 oid.
430	 */
431	if (p[0] != 0x00 || p[1] != krb5_mech_oid.length
432	    || len < krb5_mech_oid.length + 2
433	    || bcmp(krb5_mech_oid.elements, p + 2,
434		krb5_mech_oid.length))
435		return (GSS_S_DEFECTIVE_TOKEN);
436	p += krb5_mech_oid.length + 2;
437	len -= krb5_mech_oid.length + 2;
438
439	flags = get_uint32(&p, &len);
440	kc->kc_ac_flags = get_uint32(&p, &len);
441	if (flags & SC_LOCAL_ADDRESS)
442		get_address(&p, &len, &kc->kc_local_address);
443	if (flags & SC_REMOTE_ADDRESS)
444		get_address(&p, &len, &kc->kc_remote_address);
445	kc->kc_local_port = get_uint16(&p, &len);
446	kc->kc_remote_port = get_uint16(&p, &len);
447	if (flags & SC_KEYBLOCK)
448		get_keyblock(&p, &len, &kc->kc_keyblock);
449	if (flags & SC_LOCAL_SUBKEY)
450		get_keyblock(&p, &len, &kc->kc_local_subkey);
451	if (flags & SC_REMOTE_SUBKEY)
452		get_keyblock(&p, &len, &kc->kc_remote_subkey);
453	kc->kc_local_seqnumber = get_uint32(&p, &len);
454	kc->kc_remote_seqnumber = get_uint32(&p, &len);
455	kc->kc_keytype = get_uint32(&p, &len);
456	kc->kc_cksumtype = get_uint32(&p, &len);
457	get_data(&p, &len, &kc->kc_source_name);
458	get_data(&p, &len, &kc->kc_target_name);
459	kc->kc_ctx_flags = get_uint32(&p, &len);
460	kc->kc_more_flags = get_uint32(&p, &len);
461	kc->kc_lifetime = get_uint32(&p, &len);
462	/*
463	 * Heimdal 1.1 adds the message order stuff.
464	 */
465	if (format == KGSS_HEIMDAL_1_1) {
466		kc->kc_msg_order.km_flags = get_uint32(&p, &len);
467		kc->kc_msg_order.km_start = get_uint32(&p, &len);
468		kc->kc_msg_order.km_length = get_uint32(&p, &len);
469		kc->kc_msg_order.km_jitter_window = get_uint32(&p, &len);
470		kc->kc_msg_order.km_first_seq = get_uint32(&p, &len);
471		kc->kc_msg_order.km_elem =
472			malloc(kc->kc_msg_order.km_jitter_window * sizeof(uint32_t),
473			    M_GSSAPI, M_WAITOK);
474		for (i = 0; i < kc->kc_msg_order.km_jitter_window; i++)
475			kc->kc_msg_order.km_elem[i] = get_uint32(&p, &len);
476	} else {
477		kc->kc_msg_order.km_flags = 0;
478	}
479
480	res = get_keys(kc);
481	if (GSS_ERROR(res))
482		return (res);
483
484	/*
485	 * We don't need these anymore.
486	 */
487	delete_keyblock(&kc->kc_keyblock);
488	delete_keyblock(&kc->kc_local_subkey);
489	delete_keyblock(&kc->kc_remote_subkey);
490
491	return (GSS_S_COMPLETE);
492}
493
494static void
495krb5_delete(gss_ctx_id_t ctx, gss_buffer_t output_token)
496{
497	struct krb5_context *kc = (struct krb5_context *)ctx;
498
499	delete_address(&kc->kc_local_address);
500	delete_address(&kc->kc_remote_address);
501	delete_keyblock(&kc->kc_keyblock);
502	delete_keyblock(&kc->kc_local_subkey);
503	delete_keyblock(&kc->kc_remote_subkey);
504	delete_data(&kc->kc_source_name);
505	delete_data(&kc->kc_target_name);
506	if (kc->kc_msg_order.km_elem)
507		free(kc->kc_msg_order.km_elem, M_GSSAPI);
508	if (output_token) {
509		output_token->length = 0;
510		output_token->value = NULL;
511	}
512	if (kc->kc_tokenkey) {
513		krb5_free_key(kc->kc_tokenkey);
514		if (kc->kc_encryptkey) {
515			krb5_free_key(kc->kc_encryptkey);
516			krb5_free_key(kc->kc_checksumkey);
517		} else {
518			krb5_free_key(kc->kc_send_seal_Ke);
519			krb5_free_key(kc->kc_send_seal_Ki);
520			krb5_free_key(kc->kc_send_seal_Kc);
521			krb5_free_key(kc->kc_send_sign_Kc);
522			krb5_free_key(kc->kc_recv_seal_Ke);
523			krb5_free_key(kc->kc_recv_seal_Ki);
524			krb5_free_key(kc->kc_recv_seal_Kc);
525			krb5_free_key(kc->kc_recv_sign_Kc);
526		}
527	}
528	mtx_destroy(&kc->kc_lock);
529}
530
531static gss_OID
532krb5_mech_type(gss_ctx_id_t ctx)
533{
534
535	return (&krb5_mech_oid);
536}
537
538/*
539 * Make a token with the given type and length (the length includes
540 * the TOK_ID), initialising the token header appropriately. Return a
541 * pointer to the TOK_ID of the token.  A new mbuf is allocated with
542 * the framing header plus hlen bytes of space.
543 *
544 * Format is as follows:
545 *
546 *	0x60			[APPLICATION 0] SEQUENCE
547 *	DER encoded length	length of oid + type + inner token length
548 *	0x06 NN <oid data>	OID of mechanism type
549 *	TT TT			TOK_ID
550 *	<inner token>		data for inner token
551 *
552 * 1:		der encoded length
553 */
554static void *
555krb5_make_token(char tok_id[2], size_t hlen, size_t len, struct mbuf **mp)
556{
557	size_t inside_len, len_len, tlen;
558	gss_OID oid = &krb5_mech_oid;
559	struct mbuf *m;
560	uint8_t *p;
561
562	inside_len = 2 + oid->length + len;
563	if (inside_len < 128)
564		len_len = 1;
565	else if (inside_len < 0x100)
566		len_len = 2;
567	else if (inside_len < 0x10000)
568		len_len = 3;
569	else if (inside_len < 0x1000000)
570		len_len = 4;
571	else
572		len_len = 5;
573
574	tlen = 1 + len_len + 2 + oid->length + hlen;
575	KASSERT(tlen <= MLEN, ("token head too large"));
576	MGET(m, M_WAITOK, MT_DATA);
577	M_ALIGN(m, tlen);
578	m->m_len = tlen;
579
580	p = (uint8_t *) m->m_data;
581	*p++ = 0x60;
582	switch (len_len) {
583	case 1:
584		*p++ = inside_len;
585		break;
586	case 2:
587		*p++ = 0x81;
588		*p++ = inside_len;
589		break;
590	case 3:
591		*p++ = 0x82;
592		*p++ = inside_len >> 8;
593		*p++ = inside_len;
594		break;
595	case 4:
596		*p++ = 0x83;
597		*p++ = inside_len >> 16;
598		*p++ = inside_len >> 8;
599		*p++ = inside_len;
600		break;
601	case 5:
602		*p++ = 0x84;
603		*p++ = inside_len >> 24;
604		*p++ = inside_len >> 16;
605		*p++ = inside_len >> 8;
606		*p++ = inside_len;
607		break;
608	}
609
610	*p++ = 0x06;
611	*p++ = oid->length;
612	bcopy(oid->elements, p, oid->length);
613	p += oid->length;
614
615	p[0] = tok_id[0];
616	p[1] = tok_id[1];
617
618	*mp = m;
619
620	return (p);
621}
622
623/*
624 * Verify a token, checking the inner token length and mechanism oid.
625 * pointer to the first byte of the TOK_ID. The length of the
626 * encapsulated data is checked to be at least len bytes; the actual
627 * length of the encapsulated data (including TOK_ID) is returned in
628 * *encap_len.
629 *
630 * If can_pullup is TRUE and the token header is fragmented, we will
631 * rearrange it.
632 *
633 * Format is as follows:
634 *
635 *	0x60			[APPLICATION 0] SEQUENCE
636 *	DER encoded length	length of oid + type + inner token length
637 *	0x06 NN <oid data>	OID of mechanism type
638 *	TT TT			TOK_ID
639 *	<inner token>		data for inner token
640 *
641 * 1:		der encoded length
642 */
643static void *
644krb5_verify_token(char tok_id[2], size_t len, struct mbuf **mp,
645    size_t *encap_len, bool_t can_pullup)
646{
647	struct mbuf *m;
648	size_t tlen, hlen, len_len, inside_len;
649	gss_OID oid = &krb5_mech_oid;
650	uint8_t *p;
651
652	m = *mp;
653	tlen = m_length(m, NULL);
654	if (tlen < 2)
655		return (NULL);
656
657	/*
658	 * Ensure that at least the framing part of the token is
659	 * contigous.
660	 */
661	if (m->m_len < 2) {
662		if (can_pullup)
663			*mp = m = m_pullup(m, 2);
664		else
665			return (NULL);
666	}
667
668	p = m->m_data;
669
670	if (*p++ != 0x60)
671		return (NULL);
672
673	if (*p < 0x80) {
674		inside_len = *p++;
675		len_len = 1;
676	} else {
677		/*
678		 * Ensure there is enough space for the DER encoded length.
679		 */
680		len_len = (*p & 0x7f) + 1;
681		if (tlen < len_len + 1)
682			return (NULL);
683		if (m->m_len < len_len + 1) {
684			if (can_pullup)
685				*mp = m = m_pullup(m, len_len + 1);
686			else
687				return (NULL);
688			p = m->m_data + 1;
689		}
690
691		switch (*p++) {
692		case 0x81:
693			inside_len = *p++;
694			break;
695
696		case 0x82:
697			inside_len = (p[0] << 8) | p[1];
698			p += 2;
699			break;
700
701		case 0x83:
702			inside_len = (p[0] << 16) | (p[1] << 8) | p[2];
703			p += 3;
704			break;
705
706		case 0x84:
707			inside_len = (p[0] << 24) | (p[1] << 16)
708				| (p[2] << 8) | p[3];
709			p += 4;
710			break;
711
712		default:
713			return (NULL);
714		}
715	}
716
717	if (tlen != inside_len + len_len + 1)
718		return (NULL);
719	if (inside_len < 2 + oid->length + len)
720		return (NULL);
721
722	/*
723	 * Now that we know the value of len_len, we can pullup the
724	 * whole header. The header is 1 + len_len + 2 + oid->length +
725	 * len bytes.
726	 */
727	hlen = 1 + len_len + 2 + oid->length + len;
728	if (m->m_len < hlen) {
729		if (can_pullup)
730			*mp = m = m_pullup(m, hlen);
731		else
732			return (NULL);
733		p = m->m_data + 1 + len_len;
734	}
735
736	if (*p++ != 0x06)
737		return (NULL);
738	if (*p++ != oid->length)
739		return (NULL);
740	if (bcmp(oid->elements, p, oid->length))
741		return (NULL);
742	p += oid->length;
743
744	if (p[0] != tok_id[0])
745		return (NULL);
746
747	if (p[1] != tok_id[1])
748		return (NULL);
749
750	*encap_len = inside_len - 2 - oid->length;
751
752	return (p);
753}
754
755static void
756krb5_insert_seq(struct krb5_msg_order *mo, uint32_t seq, int index)
757{
758	int i;
759
760	if (mo->km_length < mo->km_jitter_window)
761		mo->km_length++;
762
763	for (i = mo->km_length - 1; i > index; i--)
764		mo->km_elem[i] = mo->km_elem[i - 1];
765	mo->km_elem[index] = seq;
766}
767
768/*
769 * Check sequence numbers according to RFC 2743 section 1.2.3.
770 */
771static OM_uint32
772krb5_sequence_check(struct krb5_context *kc, uint32_t seq)
773{
774	OM_uint32 res = GSS_S_FAILURE;
775	struct krb5_msg_order *mo = &kc->kc_msg_order;
776	int check_sequence = mo->km_flags & GSS_C_SEQUENCE_FLAG;
777	int check_replay = mo->km_flags & GSS_C_REPLAY_FLAG;
778	int i;
779
780	mtx_lock(&kc->kc_lock);
781
782	/*
783	 * Message is in-sequence with no gap.
784	 */
785	if (mo->km_length == 0 || seq == mo->km_elem[0] + 1) {
786		/*
787		 * This message is received in-sequence with no gaps.
788		 */
789		krb5_insert_seq(mo, seq, 0);
790		res = GSS_S_COMPLETE;
791		goto out;
792	}
793
794	if (seq > mo->km_elem[0]) {
795		/*
796		 * This message is received in-sequence with a gap.
797		 */
798		krb5_insert_seq(mo, seq, 0);
799		if (check_sequence)
800			res = GSS_S_GAP_TOKEN;
801		else
802			res = GSS_S_COMPLETE;
803		goto out;
804	}
805
806	if (seq < mo->km_elem[mo->km_length - 1]) {
807		if (check_replay && !check_sequence)
808			res = GSS_S_OLD_TOKEN;
809		else
810			res = GSS_S_UNSEQ_TOKEN;
811		goto out;
812	}
813
814	for (i = 0; i < mo->km_length; i++) {
815		if (mo->km_elem[i] == seq) {
816			res = GSS_S_DUPLICATE_TOKEN;
817			goto out;
818		}
819		if (mo->km_elem[i] < seq) {
820			/*
821			 * We need to insert this seq here,
822			 */
823			krb5_insert_seq(mo, seq, i);
824			if (check_replay && !check_sequence)
825				res = GSS_S_COMPLETE;
826			else
827				res = GSS_S_UNSEQ_TOKEN;
828			goto out;
829		}
830	}
831
832out:
833	mtx_unlock(&kc->kc_lock);
834
835	return (res);
836}
837
838static uint8_t sgn_alg_des_md5[] = { 0x00, 0x00 };
839static uint8_t seal_alg_des[] = { 0x00, 0x00 };
840static uint8_t sgn_alg_des3_sha1[] = { 0x04, 0x00 };
841static uint8_t seal_alg_des3[] = { 0x02, 0x00 };
842static uint8_t seal_alg_rc4[] = { 0x10, 0x00 };
843static uint8_t sgn_alg_hmac_md5[] = { 0x11, 0x00 };
844
845/*
846 * Return the size of the inner token given the use of the key's
847 * encryption class. For wrap tokens, the length of the padded
848 * plaintext will be added to this.
849 */
850static size_t
851token_length(struct krb5_key_state *key)
852{
853
854	return (16 + key->ks_class->ec_checksumlen);
855}
856
857static OM_uint32
858krb5_get_mic_old(struct krb5_context *kc, struct mbuf *m,
859    struct mbuf **micp, uint8_t sgn_alg[2])
860{
861	struct mbuf *mlast, *mic, *tm;
862	uint8_t *p, dir;
863	size_t tlen, mlen, cklen;
864	uint32_t seq;
865	char buf[8];
866
867	mlen = m_length(m, &mlast);
868
869	tlen = token_length(kc->kc_tokenkey);
870	p = krb5_make_token("\x01\x01", tlen, tlen, &mic);
871	p += 2;			/* TOK_ID */
872	*p++ = sgn_alg[0];	/* SGN_ALG */
873	*p++ = sgn_alg[1];
874
875	*p++ = 0xff;		/* filler */
876	*p++ = 0xff;
877	*p++ = 0xff;
878	*p++ = 0xff;
879
880	/*
881	 * SGN_CKSUM:
882	 *
883	 * Calculate the keyed checksum of the token header plus the
884	 * message.
885	 */
886	cklen = kc->kc_checksumkey->ks_class->ec_checksumlen;
887
888	mic->m_len = p - (uint8_t *) mic->m_data;
889	mic->m_next = m;
890	MGET(tm, M_WAITOK, MT_DATA);
891	tm->m_len = cklen;
892	mlast->m_next = tm;
893
894	krb5_checksum(kc->kc_checksumkey, 15, mic, mic->m_len - 8,
895	    8 + mlen, cklen);
896	bcopy(tm->m_data, p + 8, cklen);
897	mic->m_next = NULL;
898	mlast->m_next = NULL;
899	m_free(tm);
900
901	/*
902	 * SND_SEQ:
903	 *
904	 * Take the four bytes of the sequence number least
905	 * significant first followed by four bytes of direction
906	 * marker (zero for initiator and 0xff for acceptor). Encrypt
907	 * that data using the SGN_CKSUM as IV. Note: ARC4 wants the
908	 * sequence number big-endian.
909	 */
910	seq = atomic_fetchadd_32(&kc->kc_local_seqnumber, 1);
911	if (sgn_alg[0] == 0x11) {
912		p[0] = (seq >> 24);
913		p[1] = (seq >> 16);
914		p[2] = (seq >> 8);
915		p[3] = (seq >> 0);
916	} else {
917		p[0] = (seq >> 0);
918		p[1] = (seq >> 8);
919		p[2] = (seq >> 16);
920		p[3] = (seq >> 24);
921	}
922	if (is_initiator(kc)) {
923		dir = 0;
924	} else {
925		dir = 0xff;
926	}
927	p[4] = dir;
928	p[5] = dir;
929	p[6] = dir;
930	p[7] = dir;
931	bcopy(p + 8, buf, 8);
932
933	/*
934	 * Set the mic buffer to its final size so that the encrypt
935	 * can see the SND_SEQ part.
936	 */
937	mic->m_len += 8 + cklen;
938	krb5_encrypt(kc->kc_tokenkey, mic, mic->m_len - cklen - 8, 8, buf, 8);
939
940	*micp = mic;
941	return (GSS_S_COMPLETE);
942}
943
944static OM_uint32
945krb5_get_mic_new(struct krb5_context *kc,  struct mbuf *m,
946    struct mbuf **micp)
947{
948	struct krb5_key_state *key = kc->kc_send_sign_Kc;
949	struct mbuf *mlast, *mic;
950	uint8_t *p;
951	int flags;
952	size_t mlen, cklen;
953	uint32_t seq;
954
955	mlen = m_length(m, &mlast);
956	cklen = key->ks_class->ec_checksumlen;
957
958	KASSERT(16 + cklen <= MLEN, ("checksum too large for an mbuf"));
959	MGET(mic, M_WAITOK, MT_DATA);
960	M_ALIGN(mic, 16 + cklen);
961	mic->m_len = 16 + cklen;
962	p = mic->m_data;
963
964	/* TOK_ID */
965	p[0] = 0x04;
966	p[1] = 0x04;
967
968	/* Flags */
969	flags = 0;
970	if (is_acceptor(kc))
971		flags |= GSS_TOKEN_SENT_BY_ACCEPTOR;
972	if (kc->kc_more_flags & ACCEPTOR_SUBKEY)
973		flags |= GSS_TOKEN_ACCEPTOR_SUBKEY;
974	p[2] = flags;
975
976	/* Filler */
977	p[3] = 0xff;
978	p[4] = 0xff;
979	p[5] = 0xff;
980	p[6] = 0xff;
981	p[7] = 0xff;
982
983	/* SND_SEQ */
984	p[8] = 0;
985	p[9] = 0;
986	p[10] = 0;
987	p[11] = 0;
988	seq = atomic_fetchadd_32(&kc->kc_local_seqnumber, 1);
989	p[12] = (seq >> 24);
990	p[13] = (seq >> 16);
991	p[14] = (seq >> 8);
992	p[15] = (seq >> 0);
993
994	/*
995	 * SGN_CKSUM:
996	 *
997	 * Calculate the keyed checksum of the message plus the first
998	 * 16 bytes of the token header.
999	 */
1000	mlast->m_next = mic;
1001	krb5_checksum(key, 0, m, 0, mlen + 16, cklen);
1002	mlast->m_next = NULL;
1003
1004	*micp = mic;
1005	return (GSS_S_COMPLETE);
1006}
1007
1008static OM_uint32
1009krb5_get_mic(gss_ctx_id_t ctx, OM_uint32 *minor_status,
1010    gss_qop_t qop_req, struct mbuf *m, struct mbuf **micp)
1011{
1012	struct krb5_context *kc = (struct krb5_context *)ctx;
1013
1014	*minor_status = 0;
1015
1016	if (qop_req != GSS_C_QOP_DEFAULT)
1017		return (GSS_S_BAD_QOP);
1018
1019	if (time_uptime > kc->kc_lifetime)
1020		return (GSS_S_CONTEXT_EXPIRED);
1021
1022	switch (kc->kc_tokenkey->ks_class->ec_type) {
1023	case ETYPE_DES_CBC_CRC:
1024		return (krb5_get_mic_old(kc, m, micp, sgn_alg_des_md5));
1025
1026	case ETYPE_DES3_CBC_SHA1:
1027		return (krb5_get_mic_old(kc, m, micp, sgn_alg_des3_sha1));
1028
1029	case ETYPE_ARCFOUR_HMAC_MD5:
1030	case ETYPE_ARCFOUR_HMAC_MD5_56:
1031		return (krb5_get_mic_old(kc, m, micp, sgn_alg_hmac_md5));
1032
1033	default:
1034		return (krb5_get_mic_new(kc, m, micp));
1035	}
1036
1037	return (GSS_S_FAILURE);
1038}
1039
1040static OM_uint32
1041krb5_verify_mic_old(struct krb5_context *kc, struct mbuf *m, struct mbuf *mic,
1042    uint8_t sgn_alg[2])
1043{
1044	struct mbuf *mlast, *tm;
1045	uint8_t *p, *tp, dir;
1046	size_t mlen, tlen, elen;
1047	size_t cklen;
1048	uint32_t seq;
1049
1050	mlen = m_length(m, &mlast);
1051
1052	tlen = token_length(kc->kc_tokenkey);
1053	p = krb5_verify_token("\x01\x01", tlen, &mic, &elen, FALSE);
1054	if (!p)
1055		return (GSS_S_DEFECTIVE_TOKEN);
1056#if 0
1057	/*
1058	 * Disable this check - heimdal-1.1 generates DES3 MIC tokens
1059	 * that are 2 bytes too big.
1060	 */
1061	if (elen != tlen)
1062		return (GSS_S_DEFECTIVE_TOKEN);
1063#endif
1064	/* TOK_ID */
1065	p += 2;
1066
1067	/* SGN_ALG */
1068	if (p[0] != sgn_alg[0] || p[1] != sgn_alg[1])
1069		return (GSS_S_DEFECTIVE_TOKEN);
1070	p += 2;
1071
1072	if (p[0] != 0xff || p[1] != 0xff || p[2] != 0xff || p[3] != 0xff)
1073		return (GSS_S_DEFECTIVE_TOKEN);
1074	p += 4;
1075
1076	/*
1077	 * SGN_CKSUM:
1078	 *
1079	 * Calculate the keyed checksum of the token header plus the
1080	 * message.
1081	 */
1082	cklen = kc->kc_checksumkey->ks_class->ec_checksumlen;
1083	mic->m_len = p - (uint8_t *) mic->m_data;
1084	mic->m_next = m;
1085	MGET(tm, M_WAITOK, MT_DATA);
1086	tm->m_len = cklen;
1087	mlast->m_next = tm;
1088
1089	krb5_checksum(kc->kc_checksumkey, 15, mic, mic->m_len - 8,
1090	    8 + mlen, cklen);
1091	mic->m_next = NULL;
1092	mlast->m_next = NULL;
1093	if (bcmp(tm->m_data, p + 8, cklen)) {
1094		m_free(tm);
1095		return (GSS_S_BAD_SIG);
1096	}
1097
1098	/*
1099	 * SND_SEQ:
1100	 *
1101	 * Take the four bytes of the sequence number least
1102	 * significant first followed by four bytes of direction
1103	 * marker (zero for initiator and 0xff for acceptor). Encrypt
1104	 * that data using the SGN_CKSUM as IV.  Note: ARC4 wants the
1105	 * sequence number big-endian.
1106	 */
1107	bcopy(p, tm->m_data, 8);
1108	tm->m_len = 8;
1109	krb5_decrypt(kc->kc_tokenkey, tm, 0, 8, p + 8, 8);
1110
1111	tp = tm->m_data;
1112	if (sgn_alg[0] == 0x11) {
1113		seq = tp[3] | (tp[2] << 8) | (tp[1] << 16) | (tp[0] << 24);
1114	} else {
1115		seq = tp[0] | (tp[1] << 8) | (tp[2] << 16) | (tp[3] << 24);
1116	}
1117
1118	if (is_initiator(kc)) {
1119		dir = 0xff;
1120	} else {
1121		dir = 0;
1122	}
1123	if (tp[4] != dir || tp[5] != dir || tp[6] != dir || tp[7] != dir) {
1124		m_free(tm);
1125		return (GSS_S_DEFECTIVE_TOKEN);
1126	}
1127	m_free(tm);
1128
1129	if (kc->kc_msg_order.km_flags &
1130		(GSS_C_REPLAY_FLAG | GSS_C_SEQUENCE_FLAG)) {
1131		return (krb5_sequence_check(kc, seq));
1132	}
1133
1134	return (GSS_S_COMPLETE);
1135}
1136
1137static OM_uint32
1138krb5_verify_mic_new(struct krb5_context *kc, struct mbuf *m, struct mbuf *mic)
1139{
1140	OM_uint32 res;
1141	struct krb5_key_state *key = kc->kc_recv_sign_Kc;
1142	struct mbuf *mlast;
1143	uint8_t *p;
1144	int flags;
1145	size_t mlen, cklen;
1146	char buf[32];
1147
1148	mlen = m_length(m, &mlast);
1149	cklen = key->ks_class->ec_checksumlen;
1150
1151	KASSERT(mic->m_next == NULL, ("MIC should be contiguous"));
1152	if (mic->m_len != 16 + cklen)
1153		return (GSS_S_DEFECTIVE_TOKEN);
1154	p = mic->m_data;
1155
1156	/* TOK_ID */
1157	if (p[0] != 0x04)
1158		return (GSS_S_DEFECTIVE_TOKEN);
1159	if (p[1] != 0x04)
1160		return (GSS_S_DEFECTIVE_TOKEN);
1161
1162	/* Flags */
1163	flags = 0;
1164	if (is_initiator(kc))
1165		flags |= GSS_TOKEN_SENT_BY_ACCEPTOR;
1166	if (kc->kc_more_flags & ACCEPTOR_SUBKEY)
1167		flags |= GSS_TOKEN_ACCEPTOR_SUBKEY;
1168	if (p[2] != flags)
1169		return (GSS_S_DEFECTIVE_TOKEN);
1170
1171	/* Filler */
1172	if (p[3] != 0xff)
1173		return (GSS_S_DEFECTIVE_TOKEN);
1174	if (p[4] != 0xff)
1175		return (GSS_S_DEFECTIVE_TOKEN);
1176	if (p[5] != 0xff)
1177		return (GSS_S_DEFECTIVE_TOKEN);
1178	if (p[6] != 0xff)
1179		return (GSS_S_DEFECTIVE_TOKEN);
1180	if (p[7] != 0xff)
1181		return (GSS_S_DEFECTIVE_TOKEN);
1182
1183	/* SND_SEQ */
1184	if (kc->kc_msg_order.km_flags &
1185		(GSS_C_REPLAY_FLAG | GSS_C_SEQUENCE_FLAG)) {
1186		uint32_t seq;
1187		if (p[8] || p[9] || p[10] || p[11]) {
1188			res = GSS_S_UNSEQ_TOKEN;
1189		} else {
1190			seq = (p[12] << 24) | (p[13] << 16)
1191				| (p[14] << 8) | p[15];
1192			res = krb5_sequence_check(kc, seq);
1193		}
1194		if (GSS_ERROR(res))
1195			return (res);
1196	} else {
1197		res = GSS_S_COMPLETE;
1198	}
1199
1200	/*
1201	 * SGN_CKSUM:
1202	 *
1203	 * Calculate the keyed checksum of the message plus the first
1204	 * 16 bytes of the token header.
1205	 */
1206	m_copydata(mic, 16, cklen, buf);
1207	mlast->m_next = mic;
1208	krb5_checksum(key, 0, m, 0, mlen + 16, cklen);
1209	mlast->m_next = NULL;
1210	if (bcmp(buf, p + 16, cklen)) {
1211		return (GSS_S_BAD_SIG);
1212	}
1213
1214	return (GSS_S_COMPLETE);
1215}
1216
1217static OM_uint32
1218krb5_verify_mic(gss_ctx_id_t ctx, OM_uint32 *minor_status,
1219    struct mbuf *m, struct mbuf *mic, gss_qop_t *qop_state)
1220{
1221	struct krb5_context *kc = (struct krb5_context *)ctx;
1222
1223	*minor_status = 0;
1224	if (qop_state)
1225		*qop_state = GSS_C_QOP_DEFAULT;
1226
1227	if (time_uptime > kc->kc_lifetime)
1228		return (GSS_S_CONTEXT_EXPIRED);
1229
1230	switch (kc->kc_tokenkey->ks_class->ec_type) {
1231	case ETYPE_DES_CBC_CRC:
1232		return (krb5_verify_mic_old(kc, m, mic, sgn_alg_des_md5));
1233
1234	case ETYPE_ARCFOUR_HMAC_MD5:
1235	case ETYPE_ARCFOUR_HMAC_MD5_56:
1236		return (krb5_verify_mic_old(kc, m, mic, sgn_alg_hmac_md5));
1237
1238	case ETYPE_DES3_CBC_SHA1:
1239		return (krb5_verify_mic_old(kc, m, mic, sgn_alg_des3_sha1));
1240
1241	default:
1242		return (krb5_verify_mic_new(kc, m, mic));
1243	}
1244
1245	return (GSS_S_FAILURE);
1246}
1247
1248static OM_uint32
1249krb5_wrap_old(struct krb5_context *kc, int conf_req_flag,
1250    struct mbuf **mp, int *conf_state,
1251    uint8_t sgn_alg[2], uint8_t seal_alg[2])
1252{
1253	struct mbuf *m, *mlast, *tm, *cm, *pm;
1254	size_t mlen, tlen, padlen, datalen;
1255	uint8_t *p, dir;
1256	size_t cklen;
1257	uint8_t buf[8];
1258	uint32_t seq;
1259
1260	/*
1261	 * How many trailing pad bytes do we need?
1262	 */
1263	m = *mp;
1264	mlen = m_length(m, &mlast);
1265	tlen = kc->kc_tokenkey->ks_class->ec_msgblocklen;
1266	padlen = tlen - (mlen % tlen);
1267
1268	/*
1269	 * The data part of the token has eight bytes of random
1270	 * confounder prepended and followed by up to eight bytes of
1271	 * padding bytes each of which is set to the number of padding
1272	 * bytes.
1273	 */
1274	datalen = mlen + 8 + padlen;
1275	tlen = token_length(kc->kc_tokenkey);
1276
1277	p = krb5_make_token("\x02\x01", tlen, datalen + tlen, &tm);
1278	p += 2;			/* TOK_ID */
1279	*p++ = sgn_alg[0];	/* SGN_ALG */
1280	*p++ = sgn_alg[1];
1281	if (conf_req_flag) {
1282		*p++ = seal_alg[0]; /* SEAL_ALG */
1283		*p++ = seal_alg[1];
1284	} else {
1285		*p++ = 0xff;	/* SEAL_ALG = none */
1286		*p++ = 0xff;
1287	}
1288
1289	*p++ = 0xff;		/* filler */
1290	*p++ = 0xff;
1291
1292	/*
1293	 * Copy the padded message data.
1294	 */
1295	if (M_LEADINGSPACE(m) >= 8) {
1296		m->m_data -= 8;
1297		m->m_len += 8;
1298	} else {
1299		MGET(cm, M_WAITOK, MT_DATA);
1300		cm->m_len = 8;
1301		cm->m_next = m;
1302		m = cm;
1303	}
1304	arc4rand(m->m_data, 8, 0);
1305	if (M_TRAILINGSPACE(mlast) >= padlen) {
1306		memset(mlast->m_data + mlast->m_len, padlen, padlen);
1307		mlast->m_len += padlen;
1308	} else {
1309		MGET(pm, M_WAITOK, MT_DATA);
1310		memset(pm->m_data, padlen, padlen);
1311		pm->m_len = padlen;
1312		mlast->m_next = pm;
1313		mlast = pm;
1314	}
1315	tm->m_next = m;
1316
1317	/*
1318	 * SGN_CKSUM:
1319	 *
1320	 * Calculate the keyed checksum of the token header plus the
1321	 * padded message. Fiddle with tm->m_len so that we only
1322	 * checksum the 8 bytes of head that we care about.
1323	 */
1324	cklen = kc->kc_checksumkey->ks_class->ec_checksumlen;
1325	tlen = tm->m_len;
1326	tm->m_len = p - (uint8_t *) tm->m_data;
1327	MGET(cm, M_WAITOK, MT_DATA);
1328	cm->m_len = cklen;
1329	mlast->m_next = cm;
1330	krb5_checksum(kc->kc_checksumkey, 13, tm, tm->m_len - 8,
1331	    datalen + 8, cklen);
1332	tm->m_len = tlen;
1333	mlast->m_next = NULL;
1334	bcopy(cm->m_data, p + 8, cklen);
1335	m_free(cm);
1336
1337	/*
1338	 * SND_SEQ:
1339	 *
1340	 * Take the four bytes of the sequence number least
1341	 * significant first (most significant first for ARCFOUR)
1342	 * followed by four bytes of direction marker (zero for
1343	 * initiator and 0xff for acceptor). Encrypt that data using
1344	 * the SGN_CKSUM as IV.
1345	 */
1346	seq = atomic_fetchadd_32(&kc->kc_local_seqnumber, 1);
1347	if (sgn_alg[0] == 0x11) {
1348		p[0] = (seq >> 24);
1349		p[1] = (seq >> 16);
1350		p[2] = (seq >> 8);
1351		p[3] = (seq >> 0);
1352	} else {
1353		p[0] = (seq >> 0);
1354		p[1] = (seq >> 8);
1355		p[2] = (seq >> 16);
1356		p[3] = (seq >> 24);
1357	}
1358	if (is_initiator(kc)) {
1359		dir = 0;
1360	} else {
1361		dir = 0xff;
1362	}
1363	p[4] = dir;
1364	p[5] = dir;
1365	p[6] = dir;
1366	p[7] = dir;
1367	krb5_encrypt(kc->kc_tokenkey, tm, p - (uint8_t *) tm->m_data,
1368	    8, p + 8, 8);
1369
1370	if (conf_req_flag) {
1371		/*
1372		 * Encrypt the padded message with an IV of zero for
1373		 * DES and DES3, or an IV of the sequence number in
1374		 * big-endian format for ARCFOUR.
1375		 */
1376		if (seal_alg[0] == 0x10) {
1377			buf[0] = (seq >> 24);
1378			buf[1] = (seq >> 16);
1379			buf[2] = (seq >> 8);
1380			buf[3] = (seq >> 0);
1381			krb5_encrypt(kc->kc_encryptkey, m, 0, datalen,
1382			    buf, 4);
1383		} else {
1384			krb5_encrypt(kc->kc_encryptkey, m, 0, datalen,
1385			    NULL, 0);
1386		}
1387	}
1388
1389	if (conf_state)
1390		*conf_state = conf_req_flag;
1391
1392	*mp = tm;
1393	return (GSS_S_COMPLETE);
1394}
1395
1396static OM_uint32
1397krb5_wrap_new(struct krb5_context *kc, int conf_req_flag,
1398    struct mbuf **mp, int *conf_state)
1399{
1400	struct krb5_key_state *Ke = kc->kc_send_seal_Ke;
1401	struct krb5_key_state *Ki = kc->kc_send_seal_Ki;
1402	struct krb5_key_state *Kc = kc->kc_send_seal_Kc;
1403	const struct krb5_encryption_class *ec = Ke->ks_class;
1404	struct mbuf *m, *mlast, *tm;
1405	uint8_t *p;
1406	int flags, EC;
1407	size_t mlen, blen, mblen, cklen, ctlen;
1408	uint32_t seq;
1409	static char zpad[32];
1410
1411	m = *mp;
1412	mlen = m_length(m, &mlast);
1413
1414	blen = ec->ec_blocklen;
1415	mblen = ec->ec_msgblocklen;
1416	cklen = ec->ec_checksumlen;
1417
1418	if (conf_req_flag) {
1419		/*
1420		 * For sealed messages, we need space for 16 bytes of
1421		 * header, blen confounder, plaintext, padding, copy
1422		 * of header and checksum.
1423		 *
1424		 * We pad to mblen (which may be different from
1425		 * blen). If the encryption class is using CTS, mblen
1426		 * will be one (i.e. no padding required).
1427		 */
1428		if (mblen > 1)
1429			EC = mlen % mblen;
1430		else
1431			EC = 0;
1432		ctlen = blen + mlen + EC + 16;
1433
1434		/*
1435		 * Put initial header and confounder before the
1436		 * message.
1437		 */
1438		M_PREPEND(m, 16 + blen, M_WAITOK);
1439
1440		/*
1441		 * Append padding + copy of header and checksum. Try
1442		 * to fit this into the end of the original message,
1443		 * otherwise allocate a trailer.
1444		 */
1445		if (M_TRAILINGSPACE(mlast) >= EC + 16 + cklen) {
1446			tm = NULL;
1447			mlast->m_len += EC + 16 + cklen;
1448		} else {
1449			MGET(tm, M_WAITOK, MT_DATA);
1450			tm->m_len = EC + 16 + cklen;
1451			mlast->m_next = tm;
1452		}
1453	} else {
1454		/*
1455		 * For unsealed messages, we need 16 bytes of header
1456		 * plus space for the plaintext and a checksum. EC is
1457		 * set to the checksum size. We leave space in tm for
1458		 * a copy of the header - this will be trimmed later.
1459		 */
1460		M_PREPEND(m, 16, M_WAITOK);
1461
1462		MGET(tm, M_WAITOK, MT_DATA);
1463		tm->m_len = cklen + 16;
1464		mlast->m_next = tm;
1465		ctlen = 0;
1466		EC = cklen;
1467	}
1468
1469	p = m->m_data;
1470
1471	/* TOK_ID */
1472	p[0] = 0x05;
1473	p[1] = 0x04;
1474
1475	/* Flags */
1476	flags = 0;
1477	if (conf_req_flag)
1478		flags = GSS_TOKEN_SEALED;
1479	if (is_acceptor(kc))
1480		flags |= GSS_TOKEN_SENT_BY_ACCEPTOR;
1481	if (kc->kc_more_flags & ACCEPTOR_SUBKEY)
1482		flags |= GSS_TOKEN_ACCEPTOR_SUBKEY;
1483	p[2] = flags;
1484
1485	/* Filler */
1486	p[3] = 0xff;
1487
1488	/* EC + RRC - set to zero initially */
1489	p[4] = 0;
1490	p[5] = 0;
1491	p[6] = 0;
1492	p[7] = 0;
1493
1494	/* SND_SEQ */
1495	p[8] = 0;
1496	p[9] = 0;
1497	p[10] = 0;
1498	p[11] = 0;
1499	seq = atomic_fetchadd_32(&kc->kc_local_seqnumber, 1);
1500	p[12] = (seq >> 24);
1501	p[13] = (seq >> 16);
1502	p[14] = (seq >> 8);
1503	p[15] = (seq >> 0);
1504
1505	if (conf_req_flag) {
1506		/*
1507		 * Encrypt according to RFC 4121 section 4.2 and RFC
1508		 * 3961 section 5.3. Note: we don't generate tokens
1509		 * with RRC values other than zero. If we did, we
1510		 * should zero RRC in the copied header.
1511		 */
1512		arc4rand(p + 16, blen, 0);
1513		if (EC) {
1514			m_copyback(m, 16 + blen + mlen, EC, zpad);
1515		}
1516		m_copyback(m, 16 + blen + mlen + EC, 16, p);
1517
1518		krb5_checksum(Ki, 0, m, 16, ctlen, cklen);
1519		krb5_encrypt(Ke, m, 16, ctlen, NULL, 0);
1520	} else {
1521		/*
1522		 * The plaintext message is followed by a checksum of
1523		 * the plaintext plus a version of the header where EC
1524		 * and RRC are set to zero. Also, the original EC must
1525		 * be our checksum size.
1526		 */
1527		bcopy(p, tm->m_data, 16);
1528		krb5_checksum(Kc, 0, m, 16, mlen + 16, cklen);
1529		tm->m_data += 16;
1530		tm->m_len -= 16;
1531	}
1532
1533	/*
1534	 * Finally set EC to its actual value
1535	 */
1536	p[4] = EC >> 8;
1537	p[5] = EC;
1538
1539	*mp = m;
1540	return (GSS_S_COMPLETE);
1541}
1542
1543static OM_uint32
1544krb5_wrap(gss_ctx_id_t ctx, OM_uint32 *minor_status,
1545    int conf_req_flag, gss_qop_t qop_req,
1546    struct mbuf **mp, int *conf_state)
1547{
1548	struct krb5_context *kc = (struct krb5_context *)ctx;
1549
1550	*minor_status = 0;
1551	if (conf_state)
1552		*conf_state = 0;
1553
1554	if (qop_req != GSS_C_QOP_DEFAULT)
1555		return (GSS_S_BAD_QOP);
1556
1557	if (time_uptime > kc->kc_lifetime)
1558		return (GSS_S_CONTEXT_EXPIRED);
1559
1560	switch (kc->kc_tokenkey->ks_class->ec_type) {
1561	case ETYPE_DES_CBC_CRC:
1562		return (krb5_wrap_old(kc, conf_req_flag,
1563			mp, conf_state, sgn_alg_des_md5, seal_alg_des));
1564
1565	case ETYPE_ARCFOUR_HMAC_MD5:
1566	case ETYPE_ARCFOUR_HMAC_MD5_56:
1567		return (krb5_wrap_old(kc, conf_req_flag,
1568			mp, conf_state, sgn_alg_hmac_md5, seal_alg_rc4));
1569
1570	case ETYPE_DES3_CBC_SHA1:
1571		return (krb5_wrap_old(kc, conf_req_flag,
1572			mp, conf_state, sgn_alg_des3_sha1, seal_alg_des3));
1573
1574	default:
1575		return (krb5_wrap_new(kc, conf_req_flag, mp, conf_state));
1576	}
1577
1578	return (GSS_S_FAILURE);
1579}
1580
1581static void
1582m_trim(struct mbuf *m, int len)
1583{
1584	struct mbuf *n;
1585	int off;
1586
1587	if (m == NULL)
1588		return;
1589	n = m_getptr(m, len, &off);
1590	if (n) {
1591		n->m_len = off;
1592		if (n->m_next) {
1593			m_freem(n->m_next);
1594			n->m_next = NULL;
1595		}
1596	}
1597}
1598
1599static OM_uint32
1600krb5_unwrap_old(struct krb5_context *kc, struct mbuf **mp, int *conf_state,
1601    uint8_t sgn_alg[2], uint8_t seal_alg[2])
1602{
1603	OM_uint32 res;
1604	struct mbuf *m, *mlast, *hm, *cm, *n;
1605	uint8_t *p, dir;
1606	size_t tlen, elen, datalen, padlen;
1607	size_t cklen;
1608	uint8_t buf[32];
1609	uint32_t seq;
1610	int i, conf;
1611
1612	m = *mp;
1613	m_length(m, &mlast);
1614
1615	tlen = token_length(kc->kc_tokenkey);
1616	cklen = kc->kc_tokenkey->ks_class->ec_checksumlen;
1617
1618	p = krb5_verify_token("\x02\x01", tlen, &m, &elen, TRUE);
1619	*mp = m;
1620	if (!p)
1621		return (GSS_S_DEFECTIVE_TOKEN);
1622	datalen = elen - tlen;
1623
1624	/*
1625	 * Trim the framing header first to make life a little easier
1626	 * later.
1627	 */
1628	m_adj(m, p - (uint8_t *) m->m_data);
1629
1630	/* TOK_ID */
1631	p += 2;
1632
1633	/* SGN_ALG */
1634	if (p[0] != sgn_alg[0] || p[1] != sgn_alg[1])
1635		return (GSS_S_DEFECTIVE_TOKEN);
1636	p += 2;
1637
1638	/* SEAL_ALG */
1639	if (p[0] == seal_alg[0] && p[1] == seal_alg[1])
1640		conf = 1;
1641	else if (p[0] == 0xff && p[1] == 0xff)
1642		conf = 0;
1643	else
1644		return (GSS_S_DEFECTIVE_TOKEN);
1645	p += 2;
1646
1647	if (p[0] != 0xff || p[1] != 0xff)
1648		return (GSS_S_DEFECTIVE_TOKEN);
1649	p += 2;
1650
1651	/*
1652	 * SND_SEQ:
1653	 *
1654	 * Take the four bytes of the sequence number least
1655	 * significant first (most significant for ARCFOUR) followed
1656	 * by four bytes of direction marker (zero for initiator and
1657	 * 0xff for acceptor). Encrypt that data using the SGN_CKSUM
1658	 * as IV.
1659	 */
1660	krb5_decrypt(kc->kc_tokenkey, m, 8, 8, p + 8, 8);
1661	if (sgn_alg[0] == 0x11) {
1662		seq = p[3] | (p[2] << 8) | (p[1] << 16) | (p[0] << 24);
1663	} else {
1664		seq = p[0] | (p[1] << 8) | (p[2] << 16) | (p[3] << 24);
1665	}
1666
1667	if (is_initiator(kc)) {
1668		dir = 0xff;
1669	} else {
1670		dir = 0;
1671	}
1672	if (p[4] != dir || p[5] != dir || p[6] != dir || p[7] != dir)
1673		return (GSS_S_DEFECTIVE_TOKEN);
1674
1675	if (kc->kc_msg_order.km_flags &
1676	    (GSS_C_REPLAY_FLAG | GSS_C_SEQUENCE_FLAG)) {
1677		res = krb5_sequence_check(kc, seq);
1678		if (GSS_ERROR(res))
1679			return (res);
1680	} else {
1681		res = GSS_S_COMPLETE;
1682	}
1683
1684	/*
1685	 * If the token was encrypted, decode it in-place.
1686	 */
1687	if (conf) {
1688		/*
1689		 * Decrypt the padded message with an IV of zero for
1690		 * DES and DES3 or an IV of the big-endian encoded
1691		 * sequence number for ARCFOUR.
1692		 */
1693		if (seal_alg[0] == 0x10) {
1694			krb5_decrypt(kc->kc_encryptkey, m, 16 + cklen,
1695			    datalen, p, 4);
1696		} else {
1697			krb5_decrypt(kc->kc_encryptkey, m, 16 + cklen,
1698			    datalen, NULL, 0);
1699		}
1700	}
1701	if (conf_state)
1702		*conf_state = conf;
1703
1704	/*
1705	 * Check the trailing pad bytes.
1706	 * RFC1964 specifies between 1<->8 bytes, each with a binary value
1707	 * equal to the number of bytes.
1708	 */
1709	if (mlast->m_len > 0)
1710		padlen = mlast->m_data[mlast->m_len - 1];
1711	else {
1712		n = m_getptr(m, tlen + datalen - 1, &i);
1713		/*
1714		 * When the position is exactly equal to the # of data bytes
1715		 * in the mbuf list, m_getptr() will return the last mbuf in
1716		 * the list and an off == m_len for that mbuf, so that case
1717		 * needs to be checked as well as a NULL return.
1718		 */
1719		if (n == NULL || n->m_len == i)
1720			return (GSS_S_DEFECTIVE_TOKEN);
1721		padlen = n->m_data[i];
1722	}
1723	if (padlen < 1 || padlen > 8 || padlen > tlen + datalen)
1724		return (GSS_S_DEFECTIVE_TOKEN);
1725	m_copydata(m, tlen + datalen - padlen, padlen, buf);
1726	for (i = 0; i < padlen; i++) {
1727		if (buf[i] != padlen) {
1728			return (GSS_S_DEFECTIVE_TOKEN);
1729		}
1730	}
1731
1732	/*
1733	 * SGN_CKSUM:
1734	 *
1735	 * Calculate the keyed checksum of the token header plus the
1736	 * padded message. We do a little mbuf surgery to trim out the
1737	 * parts we don't want to checksum.
1738	 */
1739	hm = m;
1740	*mp = m = m_split(m, 16 + cklen, M_WAITOK);
1741	mlast = m_last(m);
1742	hm->m_len = 8;
1743	hm->m_next = m;
1744	MGET(cm, M_WAITOK, MT_DATA);
1745	cm->m_len = cklen;
1746	mlast->m_next = cm;
1747
1748	krb5_checksum(kc->kc_checksumkey, 13, hm, 0, datalen + 8, cklen);
1749	hm->m_next = NULL;
1750	mlast->m_next = NULL;
1751
1752	if (bcmp(cm->m_data, hm->m_data + 16, cklen)) {
1753		m_freem(hm);
1754		m_free(cm);
1755		return (GSS_S_BAD_SIG);
1756	}
1757	m_freem(hm);
1758	m_free(cm);
1759
1760	/*
1761	 * Trim off the confounder and padding.
1762	 */
1763	m_adj(m, 8);
1764	if (mlast->m_len >= padlen) {
1765		mlast->m_len -= padlen;
1766	} else {
1767		m_trim(m, datalen - 8 - padlen);
1768	}
1769
1770	*mp = m;
1771	return (res);
1772}
1773
1774static OM_uint32
1775krb5_unwrap_new(struct krb5_context *kc, struct mbuf **mp, int *conf_state)
1776{
1777	OM_uint32 res;
1778	struct krb5_key_state *Ke = kc->kc_recv_seal_Ke;
1779	struct krb5_key_state *Ki = kc->kc_recv_seal_Ki;
1780	struct krb5_key_state *Kc = kc->kc_recv_seal_Kc;
1781	const struct krb5_encryption_class *ec = Ke->ks_class;
1782	struct mbuf *m, *mlast, *hm, *cm;
1783	uint8_t *p;
1784	int sealed, flags, EC, RRC;
1785	size_t blen, cklen, ctlen, mlen, plen, tlen;
1786	char buf[32], buf2[32];
1787
1788	m = *mp;
1789	mlen = m_length(m, &mlast);
1790
1791	if (mlen <= 16)
1792		return (GSS_S_DEFECTIVE_TOKEN);
1793	if (m->m_len < 16) {
1794		m = m_pullup(m, 16);
1795		*mp = m;
1796	}
1797	p = m->m_data;
1798
1799	/* TOK_ID */
1800	if (p[0] != 0x05)
1801		return (GSS_S_DEFECTIVE_TOKEN);
1802	if (p[1] != 0x04)
1803		return (GSS_S_DEFECTIVE_TOKEN);
1804
1805	/* Flags */
1806	sealed = p[2] & GSS_TOKEN_SEALED;
1807	flags = sealed;
1808	if (is_initiator(kc))
1809		flags |= GSS_TOKEN_SENT_BY_ACCEPTOR;
1810	if (kc->kc_more_flags & ACCEPTOR_SUBKEY)
1811		flags |= GSS_TOKEN_ACCEPTOR_SUBKEY;
1812	if (p[2] != flags)
1813		return (GSS_S_DEFECTIVE_TOKEN);
1814
1815	/* Filler */
1816	if (p[3] != 0xff)
1817		return (GSS_S_DEFECTIVE_TOKEN);
1818
1819	/* EC + RRC */
1820	EC = (p[4] << 8) + p[5];
1821	RRC = (p[6] << 8) + p[7];
1822
1823	/* SND_SEQ */
1824	if (kc->kc_msg_order.km_flags &
1825		(GSS_C_REPLAY_FLAG | GSS_C_SEQUENCE_FLAG)) {
1826		uint32_t seq;
1827		if (p[8] || p[9] || p[10] || p[11]) {
1828			res = GSS_S_UNSEQ_TOKEN;
1829		} else {
1830			seq = (p[12] << 24) | (p[13] << 16)
1831				| (p[14] << 8) | p[15];
1832			res = krb5_sequence_check(kc, seq);
1833		}
1834		if (GSS_ERROR(res))
1835			return (res);
1836	} else {
1837		res = GSS_S_COMPLETE;
1838	}
1839
1840	/*
1841	 * Separate the header before dealing with RRC. We only need
1842	 * to keep the header if the message isn't encrypted.
1843	 */
1844	if (sealed) {
1845		hm = NULL;
1846		m_adj(m, 16);
1847	} else {
1848		hm = m;
1849		*mp = m = m_split(m, 16, M_WAITOK);
1850		mlast = m_last(m);
1851	}
1852
1853	/*
1854	 * Undo the effects of RRC by rotating left.
1855	 */
1856	if (RRC > 0) {
1857		struct mbuf *rm;
1858		size_t rlen;
1859
1860		rlen = mlen - 16;
1861		if (RRC <= sizeof(buf) && m->m_len >= rlen) {
1862			/*
1863			 * Simple case, just rearrange the bytes in m.
1864			 */
1865			bcopy(m->m_data, buf, RRC);
1866			bcopy(m->m_data + RRC, m->m_data, rlen - RRC);
1867			bcopy(buf, m->m_data + rlen - RRC, RRC);
1868		} else {
1869			/*
1870			 * More complicated - rearrange the mbuf
1871			 * chain.
1872			 */
1873			rm = m;
1874			*mp = m = m_split(m, RRC, M_WAITOK);
1875			m_cat(m, rm);
1876			mlast = rm;
1877		}
1878	}
1879
1880	blen = ec->ec_blocklen;
1881	cklen = ec->ec_checksumlen;
1882	if (sealed) {
1883		/*
1884		 * Decrypt according to RFC 4121 section 4.2 and RFC
1885		 * 3961 section 5.3. The message must be large enough
1886		 * for a blocksize confounder, at least one block of
1887		 * cyphertext and a checksum.
1888		 */
1889		if (mlen < 16 + 2*blen + cklen)
1890			return (GSS_S_DEFECTIVE_TOKEN);
1891
1892		ctlen = mlen - 16 - cklen;
1893		krb5_decrypt(Ke, m, 0, ctlen, NULL, 0);
1894
1895		/*
1896		 * The size of the plaintext is ctlen minus blocklen
1897		 * (for the confounder), 16 (for the copy of the token
1898		 * header) and EC (for the filler). The actual
1899		 * plaintext starts after the confounder.
1900		 */
1901		plen = ctlen - blen - 16 - EC;
1902
1903		/*
1904		 * Checksum the padded plaintext.
1905		 */
1906		m_copydata(m, ctlen, cklen, buf);
1907		krb5_checksum(Ki, 0, m, 0, ctlen, cklen);
1908		m_copydata(m, ctlen, cklen, buf2);
1909
1910		if (bcmp(buf, buf2, cklen))
1911			return (GSS_S_BAD_SIG);
1912
1913		/*
1914		 * Trim the message back to just plaintext.
1915		 */
1916		m_adj(m, blen);
1917		tlen = 16 + EC + cklen;
1918		if (mlast->m_len >= tlen) {
1919			mlast->m_len -= tlen;
1920		} else {
1921			m_trim(m, plen);
1922		}
1923	} else {
1924		/*
1925		 * The plaintext message is followed by a checksum of
1926		 * the plaintext plus a version of the header where EC
1927		 * and RRC are set to zero. Also, the original EC must
1928		 * be our checksum size.
1929		 */
1930		if (mlen < 16 + cklen || EC != cklen)
1931			return (GSS_S_DEFECTIVE_TOKEN);
1932
1933		/*
1934		 * The size of the plaintext is simply the message
1935		 * size less header and checksum. The plaintext starts
1936		 * right after the header (which we have saved in hm).
1937		 */
1938		plen = mlen - 16 - cklen;
1939
1940		/*
1941		 * Insert a copy of the header (with EC and RRC set to
1942		 * zero) between the plaintext message and the
1943		 * checksum.
1944		 */
1945		p = hm->m_data;
1946		p[4] = p[5] = p[6] = p[7] = 0;
1947
1948		cm = m_split(m, plen, M_WAITOK);
1949		mlast = m_last(m);
1950		m->m_next = hm;
1951		hm->m_next = cm;
1952
1953		bcopy(cm->m_data, buf, cklen);
1954		krb5_checksum(Kc, 0, m, 0, plen + 16, cklen);
1955		if (bcmp(cm->m_data, buf, cklen))
1956			return (GSS_S_BAD_SIG);
1957
1958		/*
1959		 * The checksum matches, discard all buf the plaintext.
1960		 */
1961		mlast->m_next = NULL;
1962		m_freem(hm);
1963	}
1964
1965	if (conf_state)
1966		*conf_state = (sealed != 0);
1967
1968	return (res);
1969}
1970
1971static OM_uint32
1972krb5_unwrap(gss_ctx_id_t ctx, OM_uint32 *minor_status,
1973    struct mbuf **mp, int *conf_state, gss_qop_t *qop_state)
1974{
1975	struct krb5_context *kc = (struct krb5_context *)ctx;
1976	OM_uint32 maj_stat;
1977
1978	*minor_status = 0;
1979	if (qop_state)
1980		*qop_state = GSS_C_QOP_DEFAULT;
1981	if (conf_state)
1982		*conf_state = 0;
1983
1984	if (time_uptime > kc->kc_lifetime)
1985		return (GSS_S_CONTEXT_EXPIRED);
1986
1987	switch (kc->kc_tokenkey->ks_class->ec_type) {
1988	case ETYPE_DES_CBC_CRC:
1989		maj_stat = krb5_unwrap_old(kc, mp, conf_state,
1990			sgn_alg_des_md5, seal_alg_des);
1991		break;
1992
1993	case ETYPE_ARCFOUR_HMAC_MD5:
1994	case ETYPE_ARCFOUR_HMAC_MD5_56:
1995		maj_stat = krb5_unwrap_old(kc, mp, conf_state,
1996			sgn_alg_hmac_md5, seal_alg_rc4);
1997		break;
1998
1999	case ETYPE_DES3_CBC_SHA1:
2000		maj_stat = krb5_unwrap_old(kc, mp, conf_state,
2001			sgn_alg_des3_sha1, seal_alg_des3);
2002		break;
2003
2004	default:
2005		maj_stat = krb5_unwrap_new(kc, mp, conf_state);
2006		break;
2007	}
2008
2009	if (GSS_ERROR(maj_stat)) {
2010		m_freem(*mp);
2011		*mp = NULL;
2012	}
2013
2014	return (maj_stat);
2015}
2016
2017static OM_uint32
2018krb5_wrap_size_limit(gss_ctx_id_t ctx, OM_uint32 *minor_status,
2019    int conf_req_flag, gss_qop_t qop_req, OM_uint32 req_output_size,
2020    OM_uint32 *max_input_size)
2021{
2022	struct krb5_context *kc = (struct krb5_context *)ctx;
2023	const struct krb5_encryption_class *ec;
2024	OM_uint32 overhead;
2025
2026	*minor_status = 0;
2027	*max_input_size = 0;
2028
2029	if (qop_req != GSS_C_QOP_DEFAULT)
2030		return (GSS_S_BAD_QOP);
2031
2032	ec = kc->kc_tokenkey->ks_class;
2033	switch (ec->ec_type) {
2034	case ETYPE_DES_CBC_CRC:
2035	case ETYPE_DES3_CBC_SHA1:
2036	case ETYPE_ARCFOUR_HMAC_MD5:
2037	case ETYPE_ARCFOUR_HMAC_MD5_56:
2038		/*
2039		 * up to 5 bytes for [APPLICATION 0] SEQUENCE
2040		 * 2 + krb5 oid length
2041		 * 8 bytes of header
2042		 * 8 bytes of confounder
2043		 * maximum of 8 bytes of padding
2044		 * checksum
2045		 */
2046		overhead = 5 + 2 + krb5_mech_oid.length;
2047		overhead += 8 + 8 + ec->ec_msgblocklen;
2048		overhead += ec->ec_checksumlen;
2049		break;
2050
2051	default:
2052		if (conf_req_flag) {
2053			/*
2054			 * 16 byts of header
2055			 * blocklen bytes of confounder
2056			 * up to msgblocklen - 1 bytes of padding
2057			 * 16 bytes for copy of header
2058			 * checksum
2059			 */
2060			overhead = 16 + ec->ec_blocklen;
2061			overhead += ec->ec_msgblocklen - 1;
2062			overhead += 16;
2063			overhead += ec->ec_checksumlen;
2064		} else {
2065			/*
2066			 * 16 bytes of header plus checksum.
2067			 */
2068			overhead = 16 + ec->ec_checksumlen;
2069		}
2070	}
2071
2072	*max_input_size = req_output_size - overhead;
2073
2074	return (GSS_S_COMPLETE);
2075}
2076
2077static kobj_method_t krb5_methods[] = {
2078	KOBJMETHOD(kgss_init,		krb5_init),
2079	KOBJMETHOD(kgss_import,		krb5_import),
2080	KOBJMETHOD(kgss_delete,		krb5_delete),
2081	KOBJMETHOD(kgss_mech_type,	krb5_mech_type),
2082	KOBJMETHOD(kgss_get_mic,	krb5_get_mic),
2083	KOBJMETHOD(kgss_verify_mic,	krb5_verify_mic),
2084	KOBJMETHOD(kgss_wrap,		krb5_wrap),
2085	KOBJMETHOD(kgss_unwrap,		krb5_unwrap),
2086	KOBJMETHOD(kgss_wrap_size_limit, krb5_wrap_size_limit),
2087	{ 0, 0 }
2088};
2089
2090static struct kobj_class krb5_class = {
2091	"kerberosv5",
2092	krb5_methods,
2093	sizeof(struct krb5_context)
2094};
2095
2096/*
2097 * Kernel module glue
2098 */
2099static int
2100kgssapi_krb5_modevent(module_t mod, int type, void *data)
2101{
2102
2103	switch (type) {
2104	case MOD_LOAD:
2105		kgss_install_mech(&krb5_mech_oid, "kerberosv5", &krb5_class);
2106		break;
2107
2108	case MOD_UNLOAD:
2109		kgss_uninstall_mech(&krb5_mech_oid);
2110		break;
2111	}
2112
2113	return (0);
2114}
2115static moduledata_t kgssapi_krb5_mod = {
2116	"kgssapi_krb5",
2117	kgssapi_krb5_modevent,
2118	NULL,
2119};
2120DECLARE_MODULE(kgssapi_krb5, kgssapi_krb5_mod, SI_SUB_VFS, SI_ORDER_ANY);
2121MODULE_DEPEND(kgssapi_krb5, kgssapi, 1, 1, 1);
2122MODULE_DEPEND(kgssapi_krb5, crypto, 1, 1, 1);
2123MODULE_DEPEND(kgssapi_krb5, rc4, 1, 1, 1);
2124MODULE_VERSION(kgssapi_krb5, 1);
2125