xref: /illumos-gate/usr/src/lib/libsmbfs/smb/ntlmssp.c (revision 142d813a06c6f9a6142e2c276b62129a17a31a65)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
24  * Copyright 2018 Nexenta Systems, Inc.  All rights reserved.
25  */
26 
27 /*
28  * NT Lan Manager Security Support Provider (NTLMSSP)
29  *
30  * Based on information from the "Davenport NTLM" page:
31  * http://davenport.sourceforge.net/ntlm.html
32  */
33 
34 
35 #include <errno.h>
36 #include <stdio.h>
37 #include <stddef.h>
38 #include <stdlib.h>
39 #include <unistd.h>
40 #include <strings.h>
41 #include <netdb.h>
42 #include <libintl.h>
43 #include <xti.h>
44 #include <assert.h>
45 
46 #include <sys/types.h>
47 #include <sys/time.h>
48 #include <sys/byteorder.h>
49 #include <sys/socket.h>
50 #include <sys/fcntl.h>
51 
52 #include <netinet/in.h>
53 #include <netinet/tcp.h>
54 #include <arpa/inet.h>
55 
56 #include <netsmb/smb.h>
57 #include <netsmb/smb_lib.h>
58 #include <netsmb/mchain.h>
59 
60 #include "private.h"
61 #include "charsets.h"
62 #include "smb_crypt.h"
63 #include "spnego.h"
64 #include "derparse.h"
65 #include "ssp.h"
66 #include "ntlm.h"
67 #include "ntlmssp.h"
68 
69 /* A shorter alias for a crazy long name from [MS-NLMP] */
70 #define	NTLMSSP_NEGOTIATE_ESS \
71 	NTLMSSP_NEGOTIATE_EXTENDED_SESSIONSECURITY
72 
73 typedef struct ntlmssp_state {
74 	uint32_t ss_flags;
75 	char *ss_target_name;	/* Primary domain or server name */
76 	struct mbuf *ss_target_info;
77 	uchar_t	ss_ssnkey[NTLM_HASH_SZ];
78 	uchar_t	ss_kxkey[NTLM_HASH_SZ];
79 } ntlmssp_state_t;
80 
81 /*
82  * So called "security buffer".
83  * A lot like an RPC string.
84  */
85 struct sec_buf {
86 	uint16_t sb_length;
87 	uint16_t sb_maxlen;
88 	uint32_t sb_offset;
89 };
90 #define	ID_SZ 8
91 static const char ntlmssp_id[ID_SZ] = "NTLMSSP";
92 
93 static int
94 ntlm_rand_ssn_key(ntlmssp_state_t *ssp_st, struct mbdata *ek_mbp);
95 
96 /*
97  * Get a "security buffer" (header part)
98  */
99 static int
100 md_get_sb_hdr(struct mbdata *mbp, struct sec_buf *sb)
101 {
102 	int err;
103 
104 	(void) md_get_uint16le(mbp, &sb->sb_length);
105 	(void) md_get_uint16le(mbp, &sb->sb_maxlen);
106 	err = md_get_uint32le(mbp, &sb->sb_offset);
107 
108 	return (err);
109 }
110 
111 /*
112  * Get a "security buffer" (data part), where
113  * the data is delivered as an mbuf.
114  */
115 static int
116 md_get_sb_data(struct mbdata *mbp, struct sec_buf *sb, struct mbuf **mp)
117 {
118 	struct mbdata tmp_mb;
119 	int err;
120 
121 	/*
122 	 * Setup tmp_mb to point to the start of the header.
123 	 * This is a dup ref - do NOT free it.
124 	 */
125 	mb_initm(&tmp_mb, mbp->mb_top);
126 
127 	/* Skip data up to the offset. */
128 	err = md_get_mem(&tmp_mb, NULL, sb->sb_offset, MB_MSYSTEM);
129 	if (err)
130 		return (err);
131 
132 	/* Get the data (as an mbuf). */
133 	err = md_get_mbuf(&tmp_mb, sb->sb_maxlen, mp);
134 
135 	return (err);
136 }
137 
138 /*
139  * Put a "security buffer" (header part)
140  */
141 static int
142 mb_put_sb_hdr(struct mbdata *mbp, struct sec_buf *sb)
143 {
144 	int err;
145 
146 	(void) mb_put_uint16le(mbp, sb->sb_length);
147 	(void) mb_put_uint16le(mbp, sb->sb_maxlen);
148 	err = mb_put_uint32le(mbp, sb->sb_offset);
149 
150 	return (err);
151 }
152 
153 /*
154  * Put a "security buffer" (data part), where
155  * the data is an mbuf.  Note: consumes m.
156  */
157 static int
158 mb_put_sb_data(struct mbdata *mbp, struct sec_buf *sb, struct mbuf *m)
159 {
160 	int cnt0;
161 	int err = 0;
162 
163 	sb->sb_offset = cnt0 = mbp->mb_count;
164 	if (m != NULL)
165 		err = mb_put_mbuf(mbp, m);
166 	sb->sb_maxlen = sb->sb_length = mbp->mb_count - cnt0;
167 
168 	return (err);
169 }
170 
171 /*
172  * Put a "security buffer" (data part), where
173  * the data is a string (OEM or unicode).
174  *
175  * The string is NOT null terminated.
176  */
177 static int
178 mb_put_sb_string(struct mbdata *mbp, struct sec_buf *sb,
179 	const char *str, int unicode)
180 {
181 	int err, trim;
182 	struct mbdata tmp_mb;
183 
184 	bzero(&tmp_mb, sizeof (tmp_mb));
185 
186 	if (str != NULL && *str != '\0') {
187 		/*
188 		 * Put the string into a temp. mbuf,
189 		 * then chop off the null terminator
190 		 * before appending to caller's mbp.
191 		 */
192 		err = mb_init(&tmp_mb);
193 		if (err)
194 			return (err);
195 		err = mb_put_string(&tmp_mb, str, unicode);
196 		if (err)
197 			return (err);
198 
199 		trim = (unicode) ? 2 : 1;
200 		if (tmp_mb.mb_cur->m_len < trim)
201 			trim = 0;
202 		tmp_mb.mb_cur->m_len -= trim;
203 	}
204 
205 	err = mb_put_sb_data(mbp, sb, tmp_mb.mb_top);
206 	/*
207 	 * Note: tmp_mb.mb_top (if any) is consumed,
208 	 * so do NOT free it (no mb_done)
209 	 */
210 	return (err);
211 }
212 
213 /*
214  * Build a Type 1 message
215  *
216  * This message has a header section containing offsets to
217  * data later in the message.  We use the common trick of
218  * building it in two parts and then concatenatening.
219  */
220 int
221 ntlmssp_put_type1(struct ssp_ctx *sp, struct mbdata *out_mb)
222 {
223 	struct type1hdr {
224 		char h_id[ID_SZ];
225 		uint32_t h_type;
226 		uint32_t h_flags;
227 		struct sec_buf h_cldom;
228 		struct sec_buf h_wksta;
229 	} hdr;
230 	struct mbdata mb2;	/* 2nd part */
231 	int err;
232 	struct smb_ctx *ctx = sp->smb_ctx;
233 	ntlmssp_state_t *ssp_st = sp->sp_private;
234 
235 	if ((err = mb_init(&mb2)) != 0)
236 		return (err);
237 	mb2.mb_count = sizeof (hdr);
238 
239 	/*
240 	 * The initial negotiation flags represent the union of all
241 	 * options we support.  The server selects from these.
242 	 * See: [MS-NLMP 2.2.2.5 NEGOTIATE]
243 	 */
244 	ssp_st->ss_flags =
245 	    NTLMSSP_NEGOTIATE_UNICODE |
246 	    NTLMSSP_NEGOTIATE_OEM |
247 	    NTLMSSP_REQUEST_TARGET |
248 	    NTLMSSP_NEGOTIATE_SIGN |
249 	    NTLMSSP_NEGOTIATE_SEAL |
250 	    /* NTLMSSP_NEGOTIATE_LM_KEY (never) */
251 	    NTLMSSP_NEGOTIATE_NTLM |
252 	    NTLMSSP_NEGOTIATE_ALWAYS_SIGN |
253 	    NTLMSSP_NEGOTIATE_ESS |
254 	    NTLMSSP_NEGOTIATE_128 |
255 	    NTLMSSP_NEGOTIATE_KEY_EXCH |
256 	    NTLMSSP_NEGOTIATE_56;
257 
258 	if ((ctx->ct_vopt & SMBVOPT_SIGNING_ENABLED) == 0)
259 		ssp_st->ss_flags &= ~NTLMSSP_NEGOTIATE_ALWAYS_SIGN;
260 
261 	bcopy(ntlmssp_id, &hdr.h_id, ID_SZ);
262 	hdr.h_type = NTLMSSP_MSGTYPE_NEGOTIATE;
263 	hdr.h_flags = ssp_st->ss_flags;
264 
265 	/*
266 	 * We could put the client domain, client name strings
267 	 * here, (always in OEM format, upper-case), and set
268 	 * NTLMSSP_NEGOTIATE_OEM_..._SUPPLIED, but Windows
269 	 * leaves these NULL so let's do the same.
270 	 */
271 	(void) mb_put_sb_string(&mb2, &hdr.h_cldom, NULL, 0);
272 	(void) mb_put_sb_string(&mb2, &hdr.h_wksta, NULL, 0);
273 
274 	/*
275 	 * Marshal the header (in LE order)
276 	 * then concatenate the 2nd part.
277 	 */
278 	(void) mb_put_mem(out_mb, &hdr.h_id, ID_SZ, MB_MSYSTEM);
279 	(void) mb_put_uint32le(out_mb, hdr.h_type);
280 	(void) mb_put_uint32le(out_mb, hdr.h_flags);
281 	(void) mb_put_sb_hdr(out_mb, &hdr.h_cldom);
282 	(void) mb_put_sb_hdr(out_mb, &hdr.h_wksta);
283 
284 	err = mb_put_mbuf(out_mb, mb2.mb_top);
285 
286 	return (err);
287 }
288 
289 /*
290  * Parse a Type 2 message
291  */
292 int
293 ntlmssp_get_type2(struct ssp_ctx *sp, struct mbdata *in_mb)
294 {
295 	struct type2hdr {
296 		char h_id[ID_SZ];
297 		uint32_t h_type;
298 		struct sec_buf h_target_name;
299 		uint32_t h_flags;
300 		uint8_t h_challenge[8];
301 		uint32_t h_context[2];		/* optional */
302 		struct sec_buf h_target_info;	/* optional */
303 	} hdr;
304 	struct mbdata top_mb, tmp_mb;
305 	struct mbuf *m;
306 	int err, uc;
307 	int min_hdr_sz = offsetof(struct type2hdr, h_context);
308 	struct smb_ctx *ctx = sp->smb_ctx;
309 	ntlmssp_state_t *ssp_st = sp->sp_private;
310 	char *buf = NULL;
311 
312 	if (m_totlen(in_mb->mb_top) < min_hdr_sz) {
313 		err = EBADRPC;
314 		goto out;
315 	}
316 
317 	/*
318 	 * Save the mbdata pointers before we consume anything.
319 	 * Careful to NOT free this (would be dup. free)
320 	 * We use this below to find data based on offsets
321 	 * from the start of the header.
322 	 */
323 	top_mb = *in_mb;
324 
325 	/* Parse the fixed size header stuff. */
326 	bzero(&hdr, sizeof (hdr));
327 	(void) md_get_mem(in_mb, &hdr.h_id, ID_SZ, MB_MSYSTEM);
328 	(void) md_get_uint32le(in_mb, &hdr.h_type);
329 	if (hdr.h_type != NTLMSSP_MSGTYPE_CHALLENGE) {
330 		err = EPROTO;
331 		goto out;
332 	}
333 	(void) md_get_sb_hdr(in_mb, &hdr.h_target_name);
334 	(void) md_get_uint32le(in_mb, &hdr.h_flags);
335 	(void) md_get_mem(in_mb, &hdr.h_challenge, NTLM_CHAL_SZ, MB_MSYSTEM);
336 
337 	/*
338 	 * Save flags, server challenge for later.
339 	 */
340 	ssp_st->ss_flags = hdr.h_flags;
341 	bcopy(&hdr.h_challenge, ctx->ct_srv_chal, NTLM_CHAL_SZ);
342 
343 	/*
344 	 * Turn off flags that might have been given but
345 	 * that we don't want to send with authenticate.
346 	 */
347 	uc = hdr.h_flags & NTLMSSP_NEGOTIATE_UNICODE;
348 	ssp_st->ss_flags &= ~NTLMSSP_NEGOTIATE_VERSION;
349 
350 	/*
351 	 * Now find out if the optional parts are there.
352 	 */
353 	if ((m_totlen(top_mb.mb_top) > sizeof (hdr)) &&
354 	    (hdr.h_target_name.sb_offset >= sizeof (hdr))) {
355 		(void) md_get_uint32le(in_mb, &hdr.h_context[0]);
356 		(void) md_get_uint32le(in_mb, &hdr.h_context[1]);
357 		(void) md_get_sb_hdr(in_mb, &hdr.h_target_info);
358 	}
359 
360 	/*
361 	 * Get the target name string.  (Server name or
362 	 * Primary domain name.)  First get a copy of the
363 	 * data from the offset/length indicated in the
364 	 * security buffer header; then parse the string.
365 	 */
366 	err = md_get_sb_data(&top_mb, &hdr.h_target_name, &m);
367 	if (err)
368 		goto out;
369 	mb_initm(&tmp_mb, m);
370 	err = md_get_string(&tmp_mb, &ssp_st->ss_target_name, uc);
371 	mb_done(&tmp_mb);
372 
373 	/*
374 	 * Get the target info blob, if present.
375 	 */
376 	if (hdr.h_target_info.sb_offset >= sizeof (hdr)) {
377 		err = md_get_sb_data(&top_mb, &hdr.h_target_info,
378 		    &ssp_st->ss_target_info);
379 	}
380 
381 out:
382 	if (buf != NULL)
383 		free(buf);
384 
385 	return (err);
386 }
387 
388 /*
389  * Build a Type 3 message
390  *
391  * This message has a header section containing offsets to
392  * data later in the message.  We use the common trick of
393  * building it in two parts and then concatenatening.
394  */
395 int
396 ntlmssp_put_type3(struct ssp_ctx *sp, struct mbdata *out_mb)
397 {
398 	struct type3hdr {
399 		char h_id[ID_SZ];
400 		uint32_t h_type;
401 		struct sec_buf h_lm_resp;
402 		struct sec_buf h_nt_resp;
403 		struct sec_buf h_domain;
404 		struct sec_buf h_user;
405 		struct sec_buf h_wksta;
406 		struct sec_buf h_ssn_key;
407 		uint32_t h_flags;
408 		/* Version struct (ommitted) */
409 		uchar_t h_mic[NTLM_HASH_SZ];
410 	} hdr;
411 	struct mbdata lm_mbc;	/* LM response */
412 	struct mbdata nt_mbc;	/* NT response */
413 	struct mbdata ti_mbc;	/* target info */
414 	struct mbdata ek_mbc;	/* encrypted session key */
415 	struct mbdata mb2;	/* payload */
416 	int err, uc;
417 	struct smb_ctx *ctx = sp->smb_ctx;
418 	ntlmssp_state_t *ssp_st = sp->sp_private;
419 	uchar_t *pmic;
420 
421 	bzero(&hdr, sizeof (hdr));
422 	bzero(&lm_mbc, sizeof (lm_mbc));
423 	bzero(&nt_mbc, sizeof (nt_mbc));
424 	bzero(&ti_mbc, sizeof (ti_mbc));
425 	bzero(&ek_mbc, sizeof (ek_mbc));
426 	bzero(&mb2, sizeof (mb2));
427 
428 	/*
429 	 * Fill in the NTLMSSP header, etc.
430 	 */
431 	if ((err = mb_init(&mb2)) != 0)
432 		goto out;
433 	mb2.mb_count = sizeof (hdr);
434 	uc = ssp_st->ss_flags & NTLMSSP_NEGOTIATE_UNICODE;
435 
436 	bcopy(ntlmssp_id, &hdr.h_id, ID_SZ);
437 	hdr.h_type = NTLMSSP_MSGTYPE_AUTHENTICATE;
438 	hdr.h_flags = ssp_st->ss_flags;
439 
440 	/*
441 	 * Put the NTLMv2/LMv2 or NTLM/LM (v1) responses,
442 	 * and compute the session key, etc.
443 	 */
444 	if (ctx->ct_authflags & SMB_AT_ANON) {
445 		/*
446 		 * We're setting up a NULL session, meaning
447 		 * the lm_mbc, nt_mbc parts remain empty.
448 		 * Let's add the "anon" flag (hint), and
449 		 * as we have no OWF hashes, we can't use
450 		 * "extended session security" (_ESS).
451 		 * The SessionBaseKey is all zeros, so
452 		 * the KeyExchangeKey is too.  Otherwise
453 		 * this is like NTLMv2/LMv2
454 		 */
455 		ssp_st->ss_flags |= NTLMSSP_NEGOTIATE_NULL_SESSION;
456 		ssp_st->ss_flags &= ~NTLMSSP_NEGOTIATE_ESS;
457 		hdr.h_flags = ssp_st->ss_flags;
458 		err = 0;
459 		/* KeyExchangeKey = SessionBaseKey = (zeros) */
460 		memset(ssp_st->ss_ssnkey, 0, NTLM_HASH_SZ);
461 		memset(ssp_st->ss_kxkey, 0, NTLM_HASH_SZ);
462 	} else if (ctx->ct_authflags & SMB_AT_NTLM2) {
463 		/*
464 		 * Doing NTLMv2/LMv2
465 		 */
466 		err = ntlm_build_target_info(ctx,
467 		    ssp_st->ss_target_info, &ti_mbc);
468 		if (err)
469 			goto out;
470 		err = ntlm_put_v2_responses(ctx, &ti_mbc,
471 		    &lm_mbc, &nt_mbc, ssp_st->ss_ssnkey);
472 		if (err)
473 			goto out;
474 		/* KeyExchangeKey = SessionBaseKey (v2) */
475 		memcpy(ssp_st->ss_kxkey, ssp_st->ss_ssnkey, NTLM_HASH_SZ);
476 	} else if (ssp_st->ss_flags & NTLMSSP_NEGOTIATE_ESS) {
477 		/*
478 		 * Doing NTLM ("v1x") which is NTLM with
479 		 * "Extended Session Security"
480 		 */
481 		err = ntlm_put_v1x_responses(ctx,
482 		    &lm_mbc, &nt_mbc, ssp_st->ss_ssnkey);
483 		if (err)
484 			goto out;
485 		/*
486 		 * "v1x computes the KeyExchangeKey from both the
487 		 * server and client nonce and (v1) SessionBaseKey.
488 		 */
489 		ntlm2_kxkey(ctx, &lm_mbc, ssp_st->ss_ssnkey,
490 		    ssp_st->ss_kxkey);
491 	} else {
492 		/*
493 		 * Doing plain old NTLM (and LM if enabled)
494 		 */
495 		err = ntlm_put_v1_responses(ctx,
496 		    &lm_mbc, &nt_mbc, ssp_st->ss_ssnkey);
497 		if (err)
498 			goto out;
499 		/* KeyExchangeKey = SessionBaseKey (v1) */
500 		memcpy(ssp_st->ss_kxkey, ssp_st->ss_ssnkey, NTLM_HASH_SZ);
501 	}
502 
503 	/*
504 	 * Compute the "ExportedSessionKey" and (possibly) the
505 	 * "EncryptedRandomSesionKey". [MS-NLMP 3.1.5.1.2]
506 	 */
507 	if (ssp_st->ss_flags & NTLMSSP_NEGOTIATE_KEY_EXCH) {
508 		err = ntlm_rand_ssn_key(ssp_st, &ek_mbc);
509 		if (err)
510 			goto out;
511 	} else {
512 		/* ExportedSessionKey is the KeyExchangeKey */
513 		memcpy(ssp_st->ss_ssnkey, ssp_st->ss_kxkey, NTLM_HASH_SZ);
514 		/* EncryptedRandomSessionKey remains NULL */
515 	}
516 
517 	err = mb_put_sb_data(&mb2, &hdr.h_lm_resp, lm_mbc.mb_top);
518 	lm_mbc.mb_top = NULL; /* consumed */
519 	if (err)
520 		goto out;
521 	err = mb_put_sb_data(&mb2, &hdr.h_nt_resp, nt_mbc.mb_top);
522 	nt_mbc.mb_top = NULL; /* consumed */
523 	if (err)
524 		goto out;
525 
526 	/*
527 	 * Put the "target" (domain), user, workstation
528 	 */
529 	err = mb_put_sb_string(&mb2, &hdr.h_domain, ctx->ct_domain, uc);
530 	if (err)
531 		goto out;
532 	err = mb_put_sb_string(&mb2, &hdr.h_user, ctx->ct_user, uc);
533 	if (err)
534 		goto out;
535 	err = mb_put_sb_string(&mb2, &hdr.h_wksta, ctx->ct_locname, uc);
536 	if (err)
537 		goto out;
538 
539 	/*
540 	 * Put the "Encrypted Random Session Key", if any.
541 	 * (ek_mbc.mb_top may be NULL)
542 	 */
543 	err = mb_put_sb_data(&mb2, &hdr.h_ssn_key, ek_mbc.mb_top);
544 	ek_mbc.mb_top = NULL; /* consumed (if any) */
545 	if (err)
546 		goto out;
547 
548 	/*
549 	 * Marshal the header (in LE order)
550 	 * then concatenate the 2nd part.
551 	 */
552 	(void) mb_put_mem(out_mb, &hdr.h_id, ID_SZ, MB_MSYSTEM);
553 	(void) mb_put_uint32le(out_mb, hdr.h_type);
554 
555 	(void) mb_put_sb_hdr(out_mb, &hdr.h_lm_resp);
556 	(void) mb_put_sb_hdr(out_mb, &hdr.h_nt_resp);
557 
558 	(void) mb_put_sb_hdr(out_mb, &hdr.h_domain);
559 	(void) mb_put_sb_hdr(out_mb, &hdr.h_user);
560 	(void) mb_put_sb_hdr(out_mb, &hdr.h_wksta);
561 
562 	(void) mb_put_sb_hdr(out_mb, &hdr.h_ssn_key);
563 	(void) mb_put_uint32le(out_mb, hdr.h_flags);
564 
565 	/* Put zeros for the MIC - filled in later */
566 	pmic = mb_reserve(out_mb, NTLM_HASH_SZ);
567 
568 	/* Put the payload. */
569 	err = mb_put_mbuf(out_mb, mb2.mb_top);
570 	mb2.mb_top = NULL; /* consumed */
571 
572 	/*
573 	 * Compute the MIC and stuff that in...
574 	 * The MIC is apparently optional.
575 	 */
576 	(void) pmic;
577 
578 out:
579 	mb_done(&mb2);
580 	mb_done(&lm_mbc);
581 	mb_done(&nt_mbc);
582 	mb_done(&ti_mbc);
583 	mb_done(&ek_mbc);
584 
585 	return (err);
586 }
587 
588 /*
589  * Helper for ntlmssp_put_type3 when doing key exchange.
590  *
591  * "ExportedSessionKey" is what we give to the "application"
592  * layer, which in here means the MAC key for SMB signing.
593  * With "key exchange", we replace the ExportedSessionKey
594  * with random data and send that (encrypted) to the peer.
595  */
596 static int
597 ntlm_rand_ssn_key(
598 	ntlmssp_state_t *ssp_st,
599 	struct mbdata *ek_mbp)
600 {
601 
602 	uchar_t *encr_ssn_key;
603 	int err;
604 
605 	if ((err = mb_init(ek_mbp)) != 0)
606 		return (err);
607 	encr_ssn_key = mb_reserve(ek_mbp, NTLM_HASH_SZ);
608 
609 	/* Set "ExportedSessionKey to NONCE(16) */
610 	(void) smb_get_urandom(ssp_st->ss_ssnkey, NTLM_HASH_SZ);
611 
612 	/* Set "EncryptedRandomSessionKey" to RC4(...) */
613 	err = smb_encrypt_RC4(encr_ssn_key, NTLM_HASH_SZ,
614 	    ssp_st->ss_kxkey, NTLM_HASH_SZ,
615 	    ssp_st->ss_ssnkey, NTLM_HASH_SZ);
616 
617 	return (err);
618 }
619 
620 /*
621  * ntlmssp_final
622  *
623  * Called after successful authentication.
624  * Save the session key.
625  */
626 int
627 ntlmssp_final(struct ssp_ctx *sp)
628 {
629 	struct smb_ctx *ctx = sp->smb_ctx;
630 	ntlmssp_state_t *ssp_st = sp->sp_private;
631 	int err = 0;
632 
633 	/*
634 	 * Update/save the session key.
635 	 */
636 	if (ctx->ct_ssnkey_buf != NULL) {
637 		free(ctx->ct_ssnkey_buf);
638 		ctx->ct_ssnkey_buf = NULL;
639 	}
640 	ctx->ct_ssnkey_buf = malloc(NTLM_HASH_SZ);
641 	if (ctx->ct_ssnkey_buf == NULL) {
642 		err = ENOMEM;
643 		goto out;
644 	}
645 	ctx->ct_ssnkey_len = NTLM_HASH_SZ;
646 	memcpy(ctx->ct_ssnkey_buf, ssp_st->ss_ssnkey, NTLM_HASH_SZ);
647 
648 out:
649 	return (err);
650 }
651 
652 /*
653  * ntlmssp_next_token
654  *
655  * See ssp.c: ssp_ctx_next_token
656  */
657 int
658 ntlmssp_next_token(struct ssp_ctx *sp, struct mbdata *in_mb,
659 	struct mbdata *out_mb)
660 {
661 	int err;
662 
663 	if (out_mb == NULL) {
664 		/* final call on successful auth. */
665 		err = ntlmssp_final(sp);
666 		goto out;
667 	}
668 
669 	/* Will build an ouptut token. */
670 	err = mb_init(out_mb);
671 	if (err)
672 		goto out;
673 
674 	/*
675 	 * When called with in_mb == NULL, it means
676 	 * this is the first call for this session,
677 	 * so put a Type 1 (initialize) token.
678 	 */
679 	if (in_mb == NULL) {
680 		err = ntlmssp_put_type1(sp, out_mb);
681 		goto out;
682 	}
683 
684 	/*
685 	 * This is not the first call, so
686 	 * parse the response token we received.
687 	 * It should be a Type 2 (challenge).
688 	 * Then put a Type 3 (authenticate)
689 	 */
690 	err = ntlmssp_get_type2(sp, in_mb);
691 	if (err)
692 		goto out;
693 
694 	err = ntlmssp_put_type3(sp, out_mb);
695 
696 out:
697 	if (err)
698 		DPRINT("ret: %d", err);
699 	return (err);
700 }
701 
702 /*
703  * ntlmssp_ctx_destroy
704  *
705  * Destroy mechanism-specific data.
706  */
707 void
708 ntlmssp_destroy(struct ssp_ctx *sp)
709 {
710 	ntlmssp_state_t *ssp_st;
711 
712 	ssp_st = sp->sp_private;
713 	if (ssp_st != NULL) {
714 		sp->sp_private = NULL;
715 		free(ssp_st->ss_target_name);
716 		m_freem(ssp_st->ss_target_info);
717 		free(ssp_st);
718 	}
719 }
720 
721 /*
722  * ntlmssp_init_clnt
723  *
724  * Initialize a new NTLMSSP client context.
725  */
726 int
727 ntlmssp_init_client(struct ssp_ctx *sp)
728 {
729 	ntlmssp_state_t *ssp_st;
730 	smb_ctx_t *ctx = sp->smb_ctx;
731 
732 	if ((ctx->ct_authflags &
733 	    (SMB_AT_NTLM2 | SMB_AT_NTLM1 | SMB_AT_ANON)) == 0) {
734 		DPRINT("No NTLM authflags");
735 		return (EINVAL);
736 	}
737 
738 	/* Get the client nonce. */
739 	(void) smb_get_urandom(ctx->ct_clnonce, NTLM_CHAL_SZ);
740 
741 	ssp_st = calloc(1, sizeof (*ssp_st));
742 	if (ssp_st == NULL)
743 		return (ENOMEM);
744 
745 	sp->sp_nexttok = ntlmssp_next_token;
746 	sp->sp_destroy = ntlmssp_destroy;
747 	sp->sp_private = ssp_st;
748 
749 	return (0);
750 }
751