xref: /titanic_44/usr/src/lib/libsmbfs/smb/ntlmssp.c (revision 4e901881a1e657e1cbf12d7ef9b476ec373e7939)
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 2013 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_NTLM2 \
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_kxkey[NTLM_HASH_SZ];
78 } ntlmssp_state_t;
79 
80 /*
81  * So called "security buffer".
82  * A lot like an RPC string.
83  */
84 struct sec_buf {
85 	uint16_t sb_length;
86 	uint16_t sb_maxlen;
87 	uint32_t sb_offset;
88 };
89 #define	ID_SZ 8
90 static const char ntlmssp_id[ID_SZ] = "NTLMSSP";
91 
92 static int
93 ntlm_rand_ssn_key(struct smb_ctx *ctx,
94 	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 (set below) */
253 	    NTLMSSP_NEGOTIATE_NTLM2 |
254 	    NTLMSSP_NEGOTIATE_128 |
255 	    NTLMSSP_NEGOTIATE_KEY_EXCH |
256 	    NTLMSSP_NEGOTIATE_56;
257 
258 	if (ctx->ct_vcflags & SMBV_WILL_SIGN) {
259 		ssp_st->ss_flags |= NTLMSSP_NEGOTIATE_ALWAYS_SIGN;
260 		ctx->ct_hflags2 |= SMB_FLAGS2_SECURITY_SIGNATURE;
261 	}
262 
263 	bcopy(ntlmssp_id, &hdr.h_id, ID_SZ);
264 	hdr.h_type = NTLMSSP_MSGTYPE_NEGOTIATE;
265 	hdr.h_flags = ssp_st->ss_flags;
266 
267 	/*
268 	 * We could put the client domain, client name strings
269 	 * here, (always in OEM format, upper-case), and set
270 	 * NTLMSSP_NEGOTIATE_OEM_..._SUPPLIED, but Windows
271 	 * leaves these NULL so let's do the same.
272 	 */
273 	(void) mb_put_sb_string(&mb2, &hdr.h_cldom, NULL, 0);
274 	(void) mb_put_sb_string(&mb2, &hdr.h_wksta, NULL, 0);
275 
276 	/*
277 	 * Marshal the header (in LE order)
278 	 * then concatenate the 2nd part.
279 	 */
280 	(void) mb_put_mem(out_mb, &hdr.h_id, ID_SZ, MB_MSYSTEM);
281 	(void) mb_put_uint32le(out_mb, hdr.h_type);
282 	(void) mb_put_uint32le(out_mb, hdr.h_flags);
283 	(void) mb_put_sb_hdr(out_mb, &hdr.h_cldom);
284 	(void) mb_put_sb_hdr(out_mb, &hdr.h_wksta);
285 
286 	err = mb_put_mbuf(out_mb, mb2.mb_top);
287 
288 	return (err);
289 }
290 
291 /*
292  * Parse a Type 2 message
293  */
294 int
295 ntlmssp_get_type2(struct ssp_ctx *sp, struct mbdata *in_mb)
296 {
297 	struct type2hdr {
298 		char h_id[ID_SZ];
299 		uint32_t h_type;
300 		struct sec_buf h_target_name;
301 		uint32_t h_flags;
302 		uint8_t h_challenge[8];
303 		uint32_t h_context[2];		/* optional */
304 		struct sec_buf h_target_info;	/* optional */
305 	} hdr;
306 	struct mbdata top_mb, tmp_mb;
307 	struct mbuf *m;
308 	int err, uc;
309 	int min_hdr_sz = offsetof(struct type2hdr, h_context);
310 	struct smb_ctx *ctx = sp->smb_ctx;
311 	ntlmssp_state_t *ssp_st = sp->sp_private;
312 	char *buf = NULL;
313 
314 	if (m_totlen(in_mb->mb_top) < min_hdr_sz) {
315 		err = EBADRPC;
316 		goto out;
317 	}
318 
319 	/*
320 	 * Save the mbdata pointers before we consume anything.
321 	 * Careful to NOT free this (would be dup. free)
322 	 * We use this below to find data based on offsets
323 	 * from the start of the header.
324 	 */
325 	top_mb = *in_mb;
326 
327 	/* Parse the fixed size header stuff. */
328 	bzero(&hdr, sizeof (hdr));
329 	(void) md_get_mem(in_mb, &hdr.h_id, ID_SZ, MB_MSYSTEM);
330 	(void) md_get_uint32le(in_mb, &hdr.h_type);
331 	if (hdr.h_type != NTLMSSP_MSGTYPE_CHALLENGE) {
332 		err = EPROTO;
333 		goto out;
334 	}
335 	(void) md_get_sb_hdr(in_mb, &hdr.h_target_name);
336 	(void) md_get_uint32le(in_mb, &hdr.h_flags);
337 	(void) md_get_mem(in_mb, &hdr.h_challenge, NTLM_CHAL_SZ, MB_MSYSTEM);
338 
339 	/*
340 	 * Save flags, server challenge for later.
341 	 */
342 	ssp_st->ss_flags = hdr.h_flags;
343 	bcopy(&hdr.h_challenge, ctx->ct_srv_chal, NTLM_CHAL_SZ);
344 
345 	/*
346 	 * Turn off flags that might have been given but
347 	 * that we don't want to send with authenticate.
348 	 */
349 	uc = hdr.h_flags & NTLMSSP_NEGOTIATE_UNICODE;
350 	ssp_st->ss_flags &= ~NTLMSSP_NEGOTIATE_VERSION;
351 
352 	/*
353 	 * Now find out if the optional parts are there.
354 	 */
355 	if ((m_totlen(top_mb.mb_top) > sizeof (hdr)) &&
356 	    (hdr.h_target_name.sb_offset >= sizeof (hdr))) {
357 		(void) md_get_uint32le(in_mb, &hdr.h_context[0]);
358 		(void) md_get_uint32le(in_mb, &hdr.h_context[1]);
359 		(void) md_get_sb_hdr(in_mb, &hdr.h_target_info);
360 	}
361 
362 	/*
363 	 * Get the target name string.  (Server name or
364 	 * Primary domain name.)  First get a copy of the
365 	 * data from the offset/length indicated in the
366 	 * security buffer header; then parse the string.
367 	 */
368 	err = md_get_sb_data(&top_mb, &hdr.h_target_name, &m);
369 	if (err)
370 		goto out;
371 	mb_initm(&tmp_mb, m);
372 	err = md_get_string(&tmp_mb, &ssp_st->ss_target_name, uc);
373 	mb_done(&tmp_mb);
374 
375 	/*
376 	 * Get the target info blob, if present.
377 	 */
378 	if (hdr.h_target_info.sb_offset >= sizeof (hdr)) {
379 		err = md_get_sb_data(&top_mb, &hdr.h_target_info,
380 		    &ssp_st->ss_target_info);
381 	}
382 
383 out:
384 	if (buf != NULL)
385 		free(buf);
386 
387 	return (err);
388 }
389 
390 /*
391  * Build a Type 3 message
392  *
393  * This message has a header section containing offsets to
394  * data later in the message.  We use the common trick of
395  * building it in two parts and then concatenatening.
396  */
397 int
398 ntlmssp_put_type3(struct ssp_ctx *sp, struct mbdata *out_mb)
399 {
400 	struct type3hdr {
401 		char h_id[ID_SZ];
402 		uint32_t h_type;
403 		struct sec_buf h_lm_resp;
404 		struct sec_buf h_nt_resp;
405 		struct sec_buf h_domain;
406 		struct sec_buf h_user;
407 		struct sec_buf h_wksta;
408 		struct sec_buf h_ssn_key;
409 		uint32_t h_flags;
410 		/* Version struct (ommitted) */
411 		uchar_t h_mic[NTLM_HASH_SZ];
412 	} hdr;
413 	struct mbdata lm_mbc;	/* LM response */
414 	struct mbdata nt_mbc;	/* NT response */
415 	struct mbdata ti_mbc;	/* target info */
416 	struct mbdata ek_mbc;	/* encrypted session key */
417 	struct mbdata mb2;	/* payload */
418 	int err, uc;
419 	struct smb_ctx *ctx = sp->smb_ctx;
420 	ntlmssp_state_t *ssp_st = sp->sp_private;
421 	uchar_t *pmic;
422 
423 	bzero(&hdr, sizeof (hdr));
424 	bzero(&lm_mbc, sizeof (lm_mbc));
425 	bzero(&nt_mbc, sizeof (nt_mbc));
426 	bzero(&ti_mbc, sizeof (ti_mbc));
427 	bzero(&ek_mbc, sizeof (ek_mbc));
428 	bzero(&mb2, sizeof (mb2));
429 
430 	/*
431 	 * Fill in the NTLMSSP header, etc.
432 	 */
433 	if ((err = mb_init(&mb2)) != 0)
434 		goto out;
435 	mb2.mb_count = sizeof (hdr);
436 	uc = ssp_st->ss_flags & NTLMSSP_NEGOTIATE_UNICODE;
437 
438 	bcopy(ntlmssp_id, &hdr.h_id, ID_SZ);
439 	hdr.h_type = NTLMSSP_MSGTYPE_AUTHENTICATE;
440 	hdr.h_flags = ssp_st->ss_flags;
441 
442 	/*
443 	 * Put the NTLMv2/LMv2 or NTLM/LM (v1) responses,
444 	 * and compute the session key, etc.
445 	 */
446 	if (ctx->ct_authflags & SMB_AT_ANON) {
447 		/*
448 		 * We're setting up a NULL session, meaning
449 		 * the lm_mbc, nt_mbc parts remain empty.
450 		 * Let's add the "anon" flag (hint).
451 		 * As there is no session key, disable the
452 		 * fancy session key stuff.
453 		 */
454 		hdr.h_flags |= NTLMSSP_NEGOTIATE_NULL_SESSION;
455 		ssp_st->ss_flags &= ~(
456 		    NTLMSSP_NEGOTIATE_NTLM2 |
457 		    NTLMSSP_NEGOTIATE_KEY_EXCH);
458 		err = 0;
459 	} else if (ctx->ct_authflags & SMB_AT_NTLM2) {
460 		/*
461 		 * Doing NTLMv2/LMv2
462 		 */
463 		err = ntlm_build_target_info(ctx,
464 		    ssp_st->ss_target_info, &ti_mbc);
465 		if (err)
466 			goto out;
467 		err = ntlm_put_v2_responses(ctx, &ti_mbc,
468 		    &lm_mbc, &nt_mbc);
469 		if (err)
470 			goto out;
471 		/* The "key exg. key" is the session base key */
472 		memcpy(ssp_st->ss_kxkey, ctx->ct_ssn_key, NTLM_HASH_SZ);
473 
474 	} else if (ssp_st->ss_flags & NTLMSSP_NEGOTIATE_NTLM2) {
475 		/*
476 		 * Doing NTLM ("v1x") which is NTLM with
477 		 * "Extended Session Security"
478 		 */
479 		err = ntlm_put_v1x_responses(ctx,
480 		    &lm_mbc, &nt_mbc);
481 		if (err)
482 			goto out;
483 		/* Compute the "Key exchange key". */
484 		ntlm2_kxkey(ctx, &lm_mbc, ssp_st->ss_kxkey);
485 	} else {
486 		/*
487 		 * Doing plain old NTLM (and LM if enabled)
488 		 */
489 		err = ntlm_put_v1_responses(ctx,
490 		    &lm_mbc, &nt_mbc);
491 		if (err)
492 			goto out;
493 		/* The "key exg. key" is the session base key */
494 		memcpy(ssp_st->ss_kxkey, ctx->ct_ssn_key, NTLM_HASH_SZ);
495 	}
496 
497 	/*
498 	 * Compute the "Exported Session Key" and (possibly)
499 	 * the "Encrypted Random Sesion Key".
500 	 * [MS-NLMP 3.1.5.1.2]
501 	 */
502 	if (ssp_st->ss_flags & NTLMSSP_NEGOTIATE_KEY_EXCH) {
503 		err = ntlm_rand_ssn_key(ctx, ssp_st, &ek_mbc);
504 		if (err)
505 			goto out;
506 	} else {
507 		/* ExportedSessionKey is the KeyExchangeKey */
508 		memcpy(ctx->ct_ssn_key, ssp_st->ss_kxkey, NTLM_HASH_SZ);
509 		/* EncryptedRandomSessionKey remains NULL */
510 	}
511 
512 	err = mb_put_sb_data(&mb2, &hdr.h_lm_resp, lm_mbc.mb_top);
513 	lm_mbc.mb_top = NULL; /* consumed */
514 	if (err)
515 		goto out;
516 	err = mb_put_sb_data(&mb2, &hdr.h_nt_resp, nt_mbc.mb_top);
517 	nt_mbc.mb_top = NULL; /* consumed */
518 	if (err)
519 		goto out;
520 
521 	/*
522 	 * Put the "target" (domain), user, workstation
523 	 */
524 	err = mb_put_sb_string(&mb2, &hdr.h_domain, ctx->ct_domain, uc);
525 	if (err)
526 		goto out;
527 	err = mb_put_sb_string(&mb2, &hdr.h_user, ctx->ct_user, uc);
528 	if (err)
529 		goto out;
530 	err = mb_put_sb_string(&mb2, &hdr.h_wksta, ctx->ct_locname, uc);
531 	if (err)
532 		goto out;
533 
534 	/*
535 	 * Put the "Encrypted Random Session Key", if any.
536 	 * (ek_mbc.mb_top may be NULL)
537 	 */
538 	err = mb_put_sb_data(&mb2, &hdr.h_ssn_key, ek_mbc.mb_top);
539 	ek_mbc.mb_top = NULL; /* consumed (if any) */
540 	if (err)
541 		goto out;
542 
543 	/*
544 	 * Marshal the header (in LE order)
545 	 * then concatenate the 2nd part.
546 	 */
547 	(void) mb_put_mem(out_mb, &hdr.h_id, ID_SZ, MB_MSYSTEM);
548 	(void) mb_put_uint32le(out_mb, hdr.h_type);
549 
550 	(void) mb_put_sb_hdr(out_mb, &hdr.h_lm_resp);
551 	(void) mb_put_sb_hdr(out_mb, &hdr.h_nt_resp);
552 
553 	(void) mb_put_sb_hdr(out_mb, &hdr.h_domain);
554 	(void) mb_put_sb_hdr(out_mb, &hdr.h_user);
555 	(void) mb_put_sb_hdr(out_mb, &hdr.h_wksta);
556 
557 	(void) mb_put_sb_hdr(out_mb, &hdr.h_ssn_key);
558 	(void) mb_put_uint32le(out_mb, hdr.h_flags);
559 
560 	/* Put zeros for the MIC - filled in later */
561 	pmic = mb_reserve(out_mb, NTLM_HASH_SZ);
562 
563 	/* Put the payload. */
564 	err = mb_put_mbuf(out_mb, mb2.mb_top);
565 	mb2.mb_top = NULL; /* consumed */
566 
567 	/*
568 	 * Compute the MIC and stuff that in...
569 	 * The MIC is apparently optional.
570 	 */
571 	(void) pmic;
572 
573 out:
574 	mb_done(&mb2);
575 	mb_done(&lm_mbc);
576 	mb_done(&nt_mbc);
577 	mb_done(&ti_mbc);
578 	mb_done(&ek_mbc);
579 
580 	return (err);
581 }
582 
583 /*
584  * Helper for ntlmssp_put_type3 when doing key exchange.
585  *
586  * "ExportedSessionKey" is what we give to the "application"
587  * layer, which in here means the MAC key for SMB signing.
588  * With "key exchange", we replace the ExportedSessionKey
589  * with random data and send that (encrypted) to the peer.
590  */
591 static int
592 ntlm_rand_ssn_key(
593 	struct smb_ctx *ctx,
594 	ntlmssp_state_t *ssp_st,
595 	struct mbdata *ek_mbp)
596 {
597 
598 	uchar_t *encr_ssn_key;
599 	int err;
600 
601 	if ((err = mb_init(ek_mbp)) != 0)
602 		return (err);
603 	encr_ssn_key = mb_reserve(ek_mbp, NTLM_HASH_SZ);
604 
605 	/* Set "ExportedSessionKey to NONCE(16) */
606 	(void) smb_get_urandom(ctx->ct_ssn_key, NTLM_HASH_SZ);
607 
608 	/* Set "EncryptedRandomSessionKey" to RC4(...) */
609 	err = smb_encrypt_RC4(encr_ssn_key, NTLM_HASH_SZ,
610 	    ssp_st->ss_kxkey, NTLM_HASH_SZ,
611 	    ctx->ct_ssn_key, NTLM_HASH_SZ);
612 
613 	return (err);
614 }
615 
616 /*
617  * ntlmssp_final
618  *
619  * Called after successful authentication.
620  * Setup the MAC key for signing.
621  */
622 int
623 ntlmssp_final(struct ssp_ctx *sp)
624 {
625 	struct smb_ctx *ctx = sp->smb_ctx;
626 	int err = 0;
627 
628 	/*
629 	 * MAC_key is just the session key, but
630 	 * Only on the first successful auth.
631 	 */
632 	if ((ctx->ct_hflags2 & SMB_FLAGS2_SECURITY_SIGNATURE) &&
633 	    (ctx->ct_mackey == NULL)) {
634 		ctx->ct_mackeylen = NTLM_HASH_SZ;
635 		ctx->ct_mackey = malloc(ctx->ct_mackeylen);
636 		if (ctx->ct_mackey == NULL) {
637 			ctx->ct_mackeylen = 0;
638 			err = ENOMEM;
639 			goto out;
640 		}
641 		memcpy(ctx->ct_mackey, ctx->ct_ssn_key, NTLM_HASH_SZ);
642 		/*
643 		 * Apparently, the server used seq. no. zero
644 		 * for our previous message, so next is two.
645 		 */
646 		ctx->ct_mac_seqno = 2;
647 	}
648 
649 out:
650 	return (err);
651 }
652 
653 /*
654  * ntlmssp_next_token
655  *
656  * See ssp.c: ssp_ctx_next_token
657  */
658 int
659 ntlmssp_next_token(struct ssp_ctx *sp, struct mbdata *in_mb,
660 	struct mbdata *out_mb)
661 {
662 	int err;
663 
664 	if (out_mb == NULL) {
665 		/* final call on successful auth. */
666 		err = ntlmssp_final(sp);
667 		goto out;
668 	}
669 
670 	/* Will build an ouptut token. */
671 	err = mb_init(out_mb);
672 	if (err)
673 		goto out;
674 
675 	/*
676 	 * When called with in_mb == NULL, it means
677 	 * this is the first call for this session,
678 	 * so put a Type 1 (initialize) token.
679 	 */
680 	if (in_mb == NULL) {
681 		err = ntlmssp_put_type1(sp, out_mb);
682 		goto out;
683 	}
684 
685 	/*
686 	 * This is not the first call, so
687 	 * parse the response token we received.
688 	 * It should be a Type 2 (challenge).
689 	 * Then put a Type 3 (authenticate)
690 	 */
691 	err = ntlmssp_get_type2(sp, in_mb);
692 	if (err)
693 		goto out;
694 
695 	err = ntlmssp_put_type3(sp, out_mb);
696 
697 out:
698 	if (err)
699 		DPRINT("ret: %d", err);
700 	return (err);
701 }
702 
703 /*
704  * ntlmssp_ctx_destroy
705  *
706  * Destroy mechanism-specific data.
707  */
708 void
709 ntlmssp_destroy(struct ssp_ctx *sp)
710 {
711 	ntlmssp_state_t *ssp_st;
712 
713 	ssp_st = sp->sp_private;
714 	if (ssp_st != NULL) {
715 		sp->sp_private = NULL;
716 		free(ssp_st->ss_target_name);
717 		m_freem(ssp_st->ss_target_info);
718 		free(ssp_st);
719 	}
720 }
721 
722 /*
723  * ntlmssp_init_clnt
724  *
725  * Initialize a new NTLMSSP client context.
726  */
727 int
728 ntlmssp_init_client(struct ssp_ctx *sp)
729 {
730 	ntlmssp_state_t *ssp_st;
731 
732 	if ((sp->smb_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 	ssp_st = calloc(1, sizeof (*ssp_st));
739 	if (ssp_st == NULL)
740 		return (ENOMEM);
741 
742 	sp->sp_nexttok = ntlmssp_next_token;
743 	sp->sp_destroy = ntlmssp_destroy;
744 	sp->sp_private = ssp_st;
745 
746 	return (0);
747 }
748