d090345bb80cd49f4dd0d7e26c6f1414cf27a4ad
[rusty/samba.git] / libcli / auth / smbencrypt.c
1 /*
2    Unix SMB/CIFS implementation.
3    SMB parameters and setup
4    Copyright (C) Andrew Tridgell 1992-1998
5    Modified by Jeremy Allison 1995.
6    Copyright (C) Jeremy Allison 1995-2000.
7    Copyright (C) Luke Kennethc Casson Leighton 1996-2000.
8    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2002-2003
9
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "system/time.h"
26 #include "../libcli/auth/msrpc_parse.h"
27 #include "../lib/crypto/crypto.h"
28 #include "../libcli/auth/libcli_auth.h"
29 #include "../librpc/gen_ndr/ntlmssp.h"
30
31 void SMBencrypt_hash(const uint8_t lm_hash[16], const uint8_t *c8, uint8_t p24[24])
32 {
33         uint8_t p21[21];
34
35         memset(p21,'\0',21);
36         memcpy(p21, lm_hash, 16);
37
38         SMBOWFencrypt(p21, c8, p24);
39
40 #ifdef DEBUG_PASSWORD
41         DEBUG(100,("SMBencrypt_hash: lm#, challenge, response\n"));
42         dump_data(100, p21, 16);
43         dump_data(100, c8, 8);
44         dump_data(100, p24, 24);
45 #endif
46 }
47
48 /*
49    This implements the X/Open SMB password encryption
50    It takes a password ('unix' string), a 8 byte "crypt key"
51    and puts 24 bytes of encrypted password into p24
52
53    Returns False if password must have been truncated to create LM hash
54 */
55
56 bool SMBencrypt(const char *passwd, const uint8_t *c8, uint8_t p24[24])
57 {
58         bool ret;
59         uint8_t lm_hash[16];
60
61         ret = E_deshash(passwd, lm_hash);
62         SMBencrypt_hash(lm_hash, c8, p24);
63         return ret;
64 }
65
66 /**
67  * Creates the MD4 Hash of the users password in NT UNICODE.
68  * @param passwd password in 'unix' charset.
69  * @param p16 return password hashed with md4, caller allocated 16 byte buffer
70  */
71
72 bool E_md4hash(const char *passwd, uint8_t p16[16])
73 {
74         size_t len;
75         smb_ucs2_t *wpwd;
76         bool ret;
77
78         ret = push_ucs2_talloc(NULL, &wpwd, passwd, &len);
79         if (!ret || len < 2) {
80                 /* We don't want to return fixed data, as most callers
81                  * don't check */
82                 mdfour(p16, (const uint8_t *)passwd, strlen(passwd));
83                 return false;
84         }
85
86         len -= 2;
87         mdfour(p16, (const uint8_t *)wpwd, len);
88
89         talloc_free(wpwd);
90         return true;
91 }
92
93 /**
94  * Creates the MD5 Hash of a combination of 16 byte salt and 16 byte NT hash.
95  * @param 16 byte salt.
96  * @param 16 byte NT hash.
97  * @param 16 byte return hashed with md5, caller allocated 16 byte buffer
98  */
99
100 void E_md5hash(const uint8_t salt[16], const uint8_t nthash[16], uint8_t hash_out[16])
101 {
102         struct MD5Context tctx;
103         MD5Init(&tctx);
104         MD5Update(&tctx, salt, 16);
105         MD5Update(&tctx, nthash, 16);
106         MD5Final(hash_out, &tctx);
107 }
108
109 /**
110  * Creates the DES forward-only Hash of the users password in DOS ASCII charset
111  * @param passwd password in 'unix' charset.
112  * @param p16 return password hashed with DES, caller allocated 16 byte buffer
113  * @return false if password was > 14 characters, and therefore may be incorrect, otherwise true
114  * @note p16 is filled in regardless
115  */
116
117 bool E_deshash(const char *passwd, uint8_t p16[16])
118 {
119         bool ret;
120         uint8_t dospwd[14];
121         TALLOC_CTX *mem_ctx;
122
123         size_t converted_size;
124
125         char *tmpbuf;
126
127         ZERO_STRUCT(dospwd);
128
129 #if _SAMBA_BUILD_ == 3
130         mem_ctx = talloc_tos();
131 #else
132         mem_ctx = NULL;
133 #endif
134         tmpbuf = strupper_talloc(mem_ctx, passwd);
135         if (tmpbuf == NULL) {
136                 /* Too many callers don't check this result, we need to fill in the buffer with something */
137                 safe_strcpy((char *)dospwd, passwd, sizeof(dospwd)-1);
138                 E_P16(dospwd, p16);
139                 return false;
140         }
141
142         ZERO_STRUCT(dospwd);
143
144         ret = convert_string_error(CH_UNIX, CH_DOS, tmpbuf, strlen(tmpbuf), dospwd, sizeof(dospwd), &converted_size);
145         talloc_free(tmpbuf);
146
147         /* Only the first 14 chars are considered, password need not
148          * be null terminated.  We do this in the error and success
149          * case to avoid returning a fixed 'password' buffer, but
150          * callers should not use it when E_deshash returns false */
151
152         E_P16((const uint8_t *)dospwd, p16);
153
154         ZERO_STRUCT(dospwd);
155
156         return ret;
157 }
158
159 /**
160  * Creates the MD4 and DES (LM) Hash of the users password.
161  * MD4 is of the NT Unicode, DES is of the DOS UPPERCASE password.
162  * @param passwd password in 'unix' charset.
163  * @param nt_p16 return password hashed with md4, caller allocated 16 byte buffer
164  * @param p16 return password hashed with des, caller allocated 16 byte buffer
165  */
166
167 /* Does both the NT and LM owfs of a user's password */
168 void nt_lm_owf_gen(const char *pwd, uint8_t nt_p16[16], uint8_t p16[16])
169 {
170         /* Calculate the MD4 hash (NT compatible) of the password */
171         memset(nt_p16, '\0', 16);
172         E_md4hash(pwd, nt_p16);
173
174 #ifdef DEBUG_PASSWORD
175         DEBUG(100,("nt_lm_owf_gen: pwd, nt#\n"));
176         dump_data(120, (const uint8_t *)pwd, strlen(pwd));
177         dump_data(100, nt_p16, 16);
178 #endif
179
180         E_deshash(pwd, (uint8_t *)p16);
181
182 #ifdef DEBUG_PASSWORD
183         DEBUG(100,("nt_lm_owf_gen: pwd, lm#\n"));
184         dump_data(120, (const uint8_t *)pwd, strlen(pwd));
185         dump_data(100, p16, 16);
186 #endif
187 }
188
189 /* Does both the NTLMv2 owfs of a user's password */
190 bool ntv2_owf_gen(const uint8_t owf[16],
191                   const char *user_in, const char *domain_in,
192                   bool upper_case_domain, /* Transform the domain into UPPER case */
193                   uint8_t kr_buf[16])
194 {
195         smb_ucs2_t *user;
196         smb_ucs2_t *domain;
197         size_t user_byte_len;
198         size_t domain_byte_len;
199         bool ret;
200
201         HMACMD5Context ctx;
202         TALLOC_CTX *mem_ctx = talloc_init("ntv2_owf_gen for %s\\%s", domain_in, user_in);
203
204         if (!mem_ctx) {
205                 return false;
206         }
207
208         if (!user_in) {
209                 user_in = "";
210         }
211
212         if (!domain_in) {
213                 domain_in = "";
214         }
215
216         user_in = strupper_talloc(mem_ctx, user_in);
217         if (user_in == NULL) {
218                 talloc_free(mem_ctx);
219                 return false;
220         }
221
222         if (upper_case_domain) {
223                 domain_in = strupper_talloc(mem_ctx, domain_in);
224                 if (domain_in == NULL) {
225                         talloc_free(mem_ctx);
226                         return false;
227                 }
228         }
229
230         ret = push_ucs2_talloc(mem_ctx, &user, user_in, &user_byte_len );
231         if (!ret) {
232                 DEBUG(0, ("push_uss2_talloc() for user failed)\n"));
233                 talloc_free(mem_ctx);
234                 return false;
235         }
236
237         ret = push_ucs2_talloc(mem_ctx, &domain, domain_in, &domain_byte_len);
238         if (!ret) {
239                 DEBUG(0, ("push_ucs2_talloc() for domain failed\n"));
240                 talloc_free(mem_ctx);
241                 return false;
242         }
243
244         SMB_ASSERT(user_byte_len >= 2);
245         SMB_ASSERT(domain_byte_len >= 2);
246
247         /* We don't want null termination */
248         user_byte_len = user_byte_len - 2;
249         domain_byte_len = domain_byte_len - 2;
250
251         hmac_md5_init_limK_to_64(owf, 16, &ctx);
252         hmac_md5_update((uint8_t *)user, user_byte_len, &ctx);
253         hmac_md5_update((uint8_t *)domain, domain_byte_len, &ctx);
254         hmac_md5_final(kr_buf, &ctx);
255
256 #ifdef DEBUG_PASSWORD
257         DEBUG(100, ("ntv2_owf_gen: user, domain, owfkey, kr\n"));
258         dump_data(100, (uint8_t *)user, user_byte_len);
259         dump_data(100, (uint8_t *)domain, domain_byte_len);
260         dump_data(100, owf, 16);
261         dump_data(100, kr_buf, 16);
262 #endif
263
264         talloc_free(mem_ctx);
265         return true;
266 }
267
268 /* Does the des encryption from the NT or LM MD4 hash. */
269 void SMBOWFencrypt(const uint8_t passwd[16], const uint8_t *c8, uint8_t p24[24])
270 {
271         uint8_t p21[21];
272
273         ZERO_STRUCT(p21);
274
275         memcpy(p21, passwd, 16);
276         E_P24(p21, c8, p24);
277 }
278
279 /* Does the des encryption. */
280
281 void SMBNTencrypt_hash(const uint8_t nt_hash[16], uint8_t *c8, uint8_t *p24)
282 {
283         uint8_t p21[21];
284
285         memset(p21,'\0',21);
286         memcpy(p21, nt_hash, 16);
287         SMBOWFencrypt(p21, c8, p24);
288
289 #ifdef DEBUG_PASSWORD
290         DEBUG(100,("SMBNTencrypt: nt#, challenge, response\n"));
291         dump_data(100, p21, 16);
292         dump_data(100, c8, 8);
293         dump_data(100, p24, 24);
294 #endif
295 }
296
297 /* Does the NT MD4 hash then des encryption. Plaintext version of the above. */
298
299 void SMBNTencrypt(const char *passwd, uint8_t *c8, uint8_t *p24)
300 {
301         uint8_t nt_hash[16];
302         E_md4hash(passwd, nt_hash);
303         SMBNTencrypt_hash(nt_hash, c8, p24);
304 }
305
306
307 /* Does the md5 encryption from the Key Response for NTLMv2. */
308 void SMBOWFencrypt_ntv2(const uint8_t kr[16],
309                         const DATA_BLOB *srv_chal,
310                         const DATA_BLOB *smbcli_chal,
311                         uint8_t resp_buf[16])
312 {
313         HMACMD5Context ctx;
314
315         hmac_md5_init_limK_to_64(kr, 16, &ctx);
316         hmac_md5_update(srv_chal->data, srv_chal->length, &ctx);
317         hmac_md5_update(smbcli_chal->data, smbcli_chal->length, &ctx);
318         hmac_md5_final(resp_buf, &ctx);
319
320 #ifdef DEBUG_PASSWORD
321         DEBUG(100, ("SMBOWFencrypt_ntv2: srv_chal, smbcli_chal, resp_buf\n"));
322         dump_data(100, srv_chal->data, srv_chal->length);
323         dump_data(100, smbcli_chal->data, smbcli_chal->length);
324         dump_data(100, resp_buf, 16);
325 #endif
326 }
327
328 void SMBsesskeygen_ntv2(const uint8_t kr[16],
329                         const uint8_t * nt_resp, uint8_t sess_key[16])
330 {
331         /* a very nice, 128 bit, variable session key */
332
333         HMACMD5Context ctx;
334
335         hmac_md5_init_limK_to_64(kr, 16, &ctx);
336         hmac_md5_update(nt_resp, 16, &ctx);
337         hmac_md5_final((uint8_t *)sess_key, &ctx);
338
339 #ifdef DEBUG_PASSWORD
340         DEBUG(100, ("SMBsesskeygen_ntv2:\n"));
341         dump_data(100, sess_key, 16);
342 #endif
343 }
344
345 void SMBsesskeygen_ntv1(const uint8_t kr[16], uint8_t sess_key[16])
346 {
347         /* yes, this session key does not change - yes, this
348            is a problem - but it is 128 bits */
349
350         mdfour((uint8_t *)sess_key, kr, 16);
351
352 #ifdef DEBUG_PASSWORD
353         DEBUG(100, ("SMBsesskeygen_ntv1:\n"));
354         dump_data(100, sess_key, 16);
355 #endif
356 }
357
358 void SMBsesskeygen_lm_sess_key(const uint8_t lm_hash[16],
359                                const uint8_t lm_resp[24], /* only uses 8 */
360                                uint8_t sess_key[16])
361 {
362         /* Calculate the LM session key (effective length 40 bits,
363            but changes with each session) */
364         uint8_t p24[24];
365         uint8_t partial_lm_hash[14];
366
367         memcpy(partial_lm_hash, lm_hash, 8);
368         memset(partial_lm_hash + 8, 0xbd, 6);
369
370         des_crypt56(p24,   lm_resp, partial_lm_hash,     1);
371         des_crypt56(p24+8, lm_resp, partial_lm_hash + 7, 1);
372
373         memcpy(sess_key, p24, 16);
374
375 #ifdef DEBUG_PASSWORD
376         DEBUG(100, ("SMBsesskeygen_lm_sess_key: \n"));
377         dump_data(100, sess_key, 16);
378 #endif
379 }
380
381 DATA_BLOB NTLMv2_generate_names_blob(TALLOC_CTX *mem_ctx,
382                                      const char *hostname,
383                                      const char *domain)
384 {
385         DATA_BLOB names_blob = data_blob_talloc(mem_ctx, NULL, 0);
386
387         /* Deliberately ignore return here.. */
388         (void)msrpc_gen(mem_ctx, &names_blob,
389                   "aaa",
390                   MsvAvNbDomainName, domain,
391                   MsvAvNbComputerName, hostname,
392                   MsvAvEOL, "");
393         return names_blob;
394 }
395
396 static DATA_BLOB NTLMv2_generate_client_data(TALLOC_CTX *mem_ctx, const DATA_BLOB *names_blob)
397 {
398         uint8_t client_chal[8];
399         DATA_BLOB response = data_blob(NULL, 0);
400         uint8_t long_date[8];
401         NTTIME nttime;
402
403         unix_to_nt_time(&nttime, time(NULL));
404
405         generate_random_buffer(client_chal, sizeof(client_chal));
406
407         push_nttime(long_date, 0, nttime);
408
409         /* See http://www.ubiqx.org/cifs/SMB.html#SMB.8.5 */
410
411         /* Deliberately ignore return here.. */
412         (void)msrpc_gen(mem_ctx, &response, "ddbbdb",
413                   0x00000101,     /* Header  */
414                   0,              /* 'Reserved'  */
415                   long_date, 8,   /* Timestamp */
416                   client_chal, 8, /* client challenge */
417                   0,              /* Unknown */
418                   names_blob->data, names_blob->length);        /* End of name list */
419
420         return response;
421 }
422
423 static DATA_BLOB NTLMv2_generate_response(TALLOC_CTX *out_mem_ctx,
424                                           const uint8_t ntlm_v2_hash[16],
425                                           const DATA_BLOB *server_chal,
426                                           const DATA_BLOB *names_blob)
427 {
428         uint8_t ntlmv2_response[16];
429         DATA_BLOB ntlmv2_client_data;
430         DATA_BLOB final_response;
431
432         TALLOC_CTX *mem_ctx = talloc_named(out_mem_ctx, 0,
433                                            "NTLMv2_generate_response internal context");
434
435         if (!mem_ctx) {
436                 return data_blob(NULL, 0);
437         }
438
439         /* NTLMv2 */
440         /* generate some data to pass into the response function - including
441            the hostname and domain name of the server */
442         ntlmv2_client_data = NTLMv2_generate_client_data(mem_ctx, names_blob);
443
444         /* Given that data, and the challenge from the server, generate a response */
445         SMBOWFencrypt_ntv2(ntlm_v2_hash, server_chal, &ntlmv2_client_data, ntlmv2_response);
446
447         final_response = data_blob_talloc(out_mem_ctx, NULL, sizeof(ntlmv2_response) + ntlmv2_client_data.length);
448
449         memcpy(final_response.data, ntlmv2_response, sizeof(ntlmv2_response));
450
451         memcpy(final_response.data+sizeof(ntlmv2_response),
452                ntlmv2_client_data.data, ntlmv2_client_data.length);
453
454         talloc_free(mem_ctx);
455
456         return final_response;
457 }
458
459 static DATA_BLOB LMv2_generate_response(TALLOC_CTX *mem_ctx,
460                                         const uint8_t ntlm_v2_hash[16],
461                                         const DATA_BLOB *server_chal)
462 {
463         uint8_t lmv2_response[16];
464         DATA_BLOB lmv2_client_data = data_blob_talloc(mem_ctx, NULL, 8);
465         DATA_BLOB final_response = data_blob_talloc(mem_ctx, NULL,24);
466
467         /* LMv2 */
468         /* client-supplied random data */
469         generate_random_buffer(lmv2_client_data.data, lmv2_client_data.length);
470
471         /* Given that data, and the challenge from the server, generate a response */
472         SMBOWFencrypt_ntv2(ntlm_v2_hash, server_chal, &lmv2_client_data, lmv2_response);
473         memcpy(final_response.data, lmv2_response, sizeof(lmv2_response));
474
475         /* after the first 16 bytes is the random data we generated above,
476            so the server can verify us with it */
477         memcpy(final_response.data+sizeof(lmv2_response),
478                lmv2_client_data.data, lmv2_client_data.length);
479
480         data_blob_free(&lmv2_client_data);
481
482         return final_response;
483 }
484
485 bool SMBNTLMv2encrypt_hash(TALLOC_CTX *mem_ctx,
486                            const char *user, const char *domain, const uint8_t nt_hash[16],
487                            const DATA_BLOB *server_chal,
488                            const DATA_BLOB *names_blob,
489                            DATA_BLOB *lm_response, DATA_BLOB *nt_response,
490                            DATA_BLOB *lm_session_key, DATA_BLOB *user_session_key)
491 {
492         uint8_t ntlm_v2_hash[16];
493
494         /* We don't use the NT# directly.  Instead we use it mashed up with
495            the username and domain.
496            This prevents username swapping during the auth exchange
497         */
498         if (!ntv2_owf_gen(nt_hash, user, domain, true, ntlm_v2_hash)) {
499                 return false;
500         }
501
502         if (nt_response) {
503                 *nt_response = NTLMv2_generate_response(mem_ctx,
504                                                         ntlm_v2_hash, server_chal,
505                                                         names_blob);
506                 if (user_session_key) {
507                         *user_session_key = data_blob_talloc(mem_ctx, NULL, 16);
508
509                         /* The NTLMv2 calculations also provide a session key, for signing etc later */
510                         /* use only the first 16 bytes of nt_response for session key */
511                         SMBsesskeygen_ntv2(ntlm_v2_hash, nt_response->data, user_session_key->data);
512                 }
513         }
514
515         /* LMv2 */
516
517         if (lm_response) {
518                 *lm_response = LMv2_generate_response(mem_ctx,
519                                                       ntlm_v2_hash, server_chal);
520                 if (lm_session_key) {
521                         *lm_session_key = data_blob_talloc(mem_ctx, NULL, 16);
522
523                         /* The NTLMv2 calculations also provide a session key, for signing etc later */
524                         /* use only the first 16 bytes of lm_response for session key */
525                         SMBsesskeygen_ntv2(ntlm_v2_hash, lm_response->data, lm_session_key->data);
526                 }
527         }
528
529         return true;
530 }
531
532 bool SMBNTLMv2encrypt(TALLOC_CTX *mem_ctx,
533                       const char *user, const char *domain,
534                       const char *password,
535                       const DATA_BLOB *server_chal,
536                       const DATA_BLOB *names_blob,
537                       DATA_BLOB *lm_response, DATA_BLOB *nt_response,
538                       DATA_BLOB *lm_session_key, DATA_BLOB *user_session_key)
539 {
540         uint8_t nt_hash[16];
541         E_md4hash(password, nt_hash);
542
543         return SMBNTLMv2encrypt_hash(mem_ctx,
544                                      user, domain, nt_hash, server_chal, names_blob,
545                                      lm_response, nt_response, lm_session_key, user_session_key);
546 }
547
548 /***********************************************************
549  encode a password buffer with a unicode password.  The buffer
550  is filled with random data to make it harder to attack.
551 ************************************************************/
552 bool encode_pw_buffer(uint8_t buffer[516], const char *password, int string_flags)
553 {
554         uint8_t new_pw[512];
555         ssize_t new_pw_len;
556
557         /* the incoming buffer can be any alignment. */
558         string_flags |= STR_NOALIGN;
559
560         new_pw_len = push_string(new_pw,
561                                  password,
562                                  sizeof(new_pw), string_flags);
563         if (new_pw_len == -1) {
564                 return false;
565         }
566
567         memcpy(&buffer[512 - new_pw_len], new_pw, new_pw_len);
568
569         generate_random_buffer(buffer, 512 - new_pw_len);
570
571         /*
572          * The length of the new password is in the last 4 bytes of
573          * the data buffer.
574          */
575         SIVAL(buffer, 512, new_pw_len);
576         ZERO_STRUCT(new_pw);
577         return true;
578 }
579
580
581 /***********************************************************
582  decode a password buffer
583  *new_pw_len is the length in bytes of the possibly mulitbyte
584  returned password including termination.
585 ************************************************************/
586
587 bool decode_pw_buffer(TALLOC_CTX *ctx,
588                       uint8_t in_buffer[516],
589                       char **pp_new_pwrd,
590                       size_t *new_pw_len,
591                       charset_t string_charset)
592 {
593         int byte_len=0;
594
595         *pp_new_pwrd = NULL;
596         *new_pw_len = 0;
597
598         /*
599           Warning !!! : This function is called from some rpc call.
600           The password IN the buffer may be a UNICODE string.
601           The password IN new_pwrd is an ASCII string
602           If you reuse that code somewhere else check first.
603         */
604
605         /* The length of the new password is in the last 4 bytes of the data buffer. */
606
607         byte_len = IVAL(in_buffer, 512);
608
609 #ifdef DEBUG_PASSWORD
610         dump_data(100, in_buffer, 516);
611 #endif
612
613         /* Password cannot be longer than the size of the password buffer */
614         if ( (byte_len < 0) || (byte_len > 512)) {
615                 DEBUG(0, ("decode_pw_buffer: incorrect password length (%d).\n", byte_len));
616                 DEBUG(0, ("decode_pw_buffer: check that 'encrypt passwords = yes'\n"));
617                 return false;
618         }
619
620         /* decode into the return buffer. */
621         if (!convert_string_talloc(ctx, string_charset, CH_UNIX,
622                                    &in_buffer[512 - byte_len],
623                                    byte_len,
624                                    (void *)pp_new_pwrd,
625                                    new_pw_len)) {
626                 DEBUG(0, ("decode_pw_buffer: failed to convert incoming password\n"));
627                 return false;
628         }
629
630 #ifdef DEBUG_PASSWORD
631         DEBUG(100,("decode_pw_buffer: new_pwrd: "));
632         dump_data(100, (uint8_t *)*pp_new_pwrd, *new_pw_len);
633         DEBUG(100,("multibyte len:%lu\n", (unsigned long int)*new_pw_len));
634         DEBUG(100,("original char len:%d\n", byte_len/2));
635 #endif
636
637         return true;
638 }
639
640 /***********************************************************
641  Decode an arc4 encrypted password change buffer.
642 ************************************************************/
643
644 void encode_or_decode_arc4_passwd_buffer(unsigned char pw_buf[532], const DATA_BLOB *psession_key)
645 {
646         struct MD5Context tctx;
647         unsigned char key_out[16];
648
649         /* Confounder is last 16 bytes. */
650
651         MD5Init(&tctx);
652         MD5Update(&tctx, &pw_buf[516], 16);
653         MD5Update(&tctx, psession_key->data, psession_key->length);
654         MD5Final(key_out, &tctx);
655         /* arc4 with key_out. */
656         arcfour_crypt(pw_buf, key_out, 516);
657 }
658
659 /***********************************************************
660  encode a password buffer with an already unicode password.  The
661  rest of the buffer is filled with random data to make it harder to attack.
662 ************************************************************/
663 bool set_pw_in_buffer(uint8_t buffer[516], DATA_BLOB *password)
664 {
665         if (password->length > 512) {
666                 return false;
667         }
668
669         memcpy(&buffer[512 - password->length], password->data, password->length);
670
671         generate_random_buffer(buffer, 512 - password->length);
672
673         /*
674          * The length of the new password is in the last 4 bytes of
675          * the data buffer.
676          */
677         SIVAL(buffer, 512, password->length);
678         return true;
679 }
680
681 /***********************************************************
682  decode a password buffer
683  *new_pw_size is the length in bytes of the extracted unicode password
684 ************************************************************/
685 bool extract_pw_from_buffer(TALLOC_CTX *mem_ctx,
686                             uint8_t in_buffer[516], DATA_BLOB *new_pass)
687 {
688         int byte_len=0;
689
690         /* The length of the new password is in the last 4 bytes of the data buffer. */
691
692         byte_len = IVAL(in_buffer, 512);
693
694 #ifdef DEBUG_PASSWORD
695         dump_data(100, in_buffer, 516);
696 #endif
697
698         /* Password cannot be longer than the size of the password buffer */
699         if ( (byte_len < 0) || (byte_len > 512)) {
700                 return false;
701         }
702
703         *new_pass = data_blob_talloc(mem_ctx, &in_buffer[512 - byte_len], byte_len);
704
705         if (!new_pass->data) {
706                 return false;
707         }
708
709         return true;
710 }
711
712
713 /* encode a wkssvc_PasswordBuffer:
714  *
715  * similar to samr_CryptPasswordEx. Different: 8byte confounder (instead of
716  * 16byte), confounder in front of the 516 byte buffer (instead of after that
717  * buffer), calling MD5Update() first with session_key and then with confounder
718  * (vice versa in samr) - Guenther */
719
720 void encode_wkssvc_join_password_buffer(TALLOC_CTX *mem_ctx,
721                                         const char *pwd,
722                                         DATA_BLOB *session_key,
723                                         struct wkssvc_PasswordBuffer **pwd_buf)
724 {
725         uint8_t buffer[516];
726         struct MD5Context ctx;
727         struct wkssvc_PasswordBuffer *my_pwd_buf = NULL;
728         DATA_BLOB confounded_session_key;
729         int confounder_len = 8;
730         uint8_t confounder[8];
731
732         my_pwd_buf = talloc_zero(mem_ctx, struct wkssvc_PasswordBuffer);
733         if (!my_pwd_buf) {
734                 return;
735         }
736
737         confounded_session_key = data_blob_talloc(mem_ctx, NULL, 16);
738
739         encode_pw_buffer(buffer, pwd, STR_UNICODE);
740
741         generate_random_buffer((uint8_t *)confounder, confounder_len);
742
743         MD5Init(&ctx);
744         MD5Update(&ctx, session_key->data, session_key->length);
745         MD5Update(&ctx, confounder, confounder_len);
746         MD5Final(confounded_session_key.data, &ctx);
747
748         arcfour_crypt_blob(buffer, 516, &confounded_session_key);
749
750         memcpy(&my_pwd_buf->data[0], confounder, confounder_len);
751         memcpy(&my_pwd_buf->data[8], buffer, 516);
752
753         data_blob_free(&confounded_session_key);
754
755         *pwd_buf = my_pwd_buf;
756 }
757
758 WERROR decode_wkssvc_join_password_buffer(TALLOC_CTX *mem_ctx,
759                                           struct wkssvc_PasswordBuffer *pwd_buf,
760                                           DATA_BLOB *session_key,
761                                           char **pwd)
762 {
763         uint8_t buffer[516];
764         struct MD5Context ctx;
765         size_t pwd_len;
766
767         DATA_BLOB confounded_session_key;
768
769         int confounder_len = 8;
770         uint8_t confounder[8];
771
772         *pwd = NULL;
773
774         if (!pwd_buf) {
775                 return WERR_BAD_PASSWORD;
776         }
777
778         if (session_key->length != 16) {
779                 DEBUG(10,("invalid session key\n"));
780                 return WERR_BAD_PASSWORD;
781         }
782
783         confounded_session_key = data_blob_talloc(mem_ctx, NULL, 16);
784
785         memcpy(&confounder, &pwd_buf->data[0], confounder_len);
786         memcpy(&buffer, &pwd_buf->data[8], 516);
787
788         MD5Init(&ctx);
789         MD5Update(&ctx, session_key->data, session_key->length);
790         MD5Update(&ctx, confounder, confounder_len);
791         MD5Final(confounded_session_key.data, &ctx);
792
793         arcfour_crypt_blob(buffer, 516, &confounded_session_key);
794
795         if (!decode_pw_buffer(mem_ctx, buffer, pwd, &pwd_len, CH_UTF16)) {
796                 data_blob_free(&confounded_session_key);
797                 return WERR_BAD_PASSWORD;
798         }
799
800         data_blob_free(&confounded_session_key);
801
802         return WERR_OK;
803 }
804