r20949: Looking over some lcov output, try and walk some error paths.
[metze/samba/wip.git] / source4 / auth / ntlmssp / ntlmssp_sign.c
1 /* 
2  *  Unix SMB/CIFS implementation.
3  *  Version 3.0
4  *  NTLMSSP Signing routines
5  *  Copyright (C) Luke Kenneth Casson Leighton 1996-2001
6  *  Copyright (C) Andrew Bartlett <abartlet@samba.org> 2003-2005
7  *  
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *  
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *  
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software Foundation,
20  *  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
21  */
22
23 #include "includes.h"
24 #include "auth/auth.h"
25 #include "auth/ntlmssp/ntlmssp.h"
26 #include "auth/ntlmssp/msrpc_parse.h"
27 #include "lib/crypto/crypto.h"
28 #include "auth/credentials/credentials.h"
29 #include "auth/gensec/gensec.h"
30
31 #define CLI_SIGN "session key to client-to-server signing key magic constant"
32 #define CLI_SEAL "session key to client-to-server sealing key magic constant"
33 #define SRV_SIGN "session key to server-to-client signing key magic constant"
34 #define SRV_SEAL "session key to server-to-client sealing key magic constant"
35
36 /**
37  * Some notes on the NTLM2 code:
38  *
39  * NTLM2 is a AEAD system.  This means that the data encrypted is not
40  * all the data that is signed.  In DCE-RPC case, the headers of the
41  * DCE-RPC packets are also signed.  This prevents some of the
42  * fun-and-games one might have by changing them.
43  *
44  */
45
46 static void calc_ntlmv2_key(TALLOC_CTX *mem_ctx, 
47                             DATA_BLOB *subkey,
48                             DATA_BLOB session_key, 
49                             const char *constant)
50 {
51         struct MD5Context ctx3;
52         *subkey = data_blob_talloc(mem_ctx, NULL, 16);
53         MD5Init(&ctx3);
54         MD5Update(&ctx3, session_key.data, session_key.length);
55         MD5Update(&ctx3, (const uint8_t *)constant, strlen(constant)+1);
56         MD5Final(subkey->data, &ctx3);
57 }
58
59 enum ntlmssp_direction {
60         NTLMSSP_SEND,
61         NTLMSSP_RECEIVE
62 };
63
64 static NTSTATUS ntlmssp_make_packet_signature(struct gensec_ntlmssp_state *gensec_ntlmssp_state,
65                                               TALLOC_CTX *sig_mem_ctx, 
66                                               const uint8_t *data, size_t length, 
67                                               const uint8_t *whole_pdu, size_t pdu_length, 
68                                               enum ntlmssp_direction direction,
69                                               DATA_BLOB *sig, BOOL encrypt_sig)
70 {
71         if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {
72
73                 HMACMD5Context ctx;
74                 uint8_t digest[16];
75                 uint8_t seq_num[4];
76
77                 *sig = data_blob_talloc(sig_mem_ctx, NULL, NTLMSSP_SIG_SIZE);
78                 if (!sig->data) {
79                         return NT_STATUS_NO_MEMORY;
80                 }
81                         
82                 switch (direction) {
83                 case NTLMSSP_SEND:
84                         SIVAL(seq_num, 0, gensec_ntlmssp_state->crypt.ntlm2.send_seq_num);
85                         gensec_ntlmssp_state->crypt.ntlm2.send_seq_num++;
86                         hmac_md5_init_limK_to_64(gensec_ntlmssp_state->crypt.ntlm2.send_sign_key.data, 
87                                                  gensec_ntlmssp_state->crypt.ntlm2.send_sign_key.length, &ctx);
88                         break;
89                 case NTLMSSP_RECEIVE:
90                         SIVAL(seq_num, 0, gensec_ntlmssp_state->crypt.ntlm2.recv_seq_num);
91                         gensec_ntlmssp_state->crypt.ntlm2.recv_seq_num++;
92                         hmac_md5_init_limK_to_64(gensec_ntlmssp_state->crypt.ntlm2.recv_sign_key.data, 
93                                                  gensec_ntlmssp_state->crypt.ntlm2.recv_sign_key.length, &ctx);
94                         break;
95                 }
96                 hmac_md5_update(seq_num, sizeof(seq_num), &ctx);
97                 hmac_md5_update(whole_pdu, pdu_length, &ctx);
98                 hmac_md5_final(digest, &ctx);
99
100                 if (encrypt_sig && gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_KEY_EXCH) {
101                         switch (direction) {
102                         case NTLMSSP_SEND:
103                                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state, digest, 8);
104                                 break;
105                         case NTLMSSP_RECEIVE:
106                                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm2.recv_seal_arcfour_state, digest, 8);
107                                 break;
108                         }
109                 }
110
111                 SIVAL(sig->data, 0, NTLMSSP_SIGN_VERSION);
112                 memcpy(sig->data + 4, digest, 8);
113                 memcpy(sig->data + 12, seq_num, 4);
114
115                 DEBUG(10, ("NTLM2: created signature over %llu bytes of input:\n", (unsigned long long)pdu_length));
116                 dump_data(11, sig->data, sig->length);
117                         
118         } else {
119                 uint32_t crc;
120                 crc = crc32_calc_buffer(data, length);
121                 if (!msrpc_gen(sig_mem_ctx, sig, "dddd", NTLMSSP_SIGN_VERSION, 0, crc, gensec_ntlmssp_state->crypt.ntlm.seq_num)) {
122                         return NT_STATUS_NO_MEMORY;
123                 }
124                 gensec_ntlmssp_state->crypt.ntlm.seq_num++;
125
126                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm.arcfour_state, sig->data+4, sig->length-4);
127
128                 DEBUG(10, ("NTLM1: created signature over %llu bytes of input:\n", (unsigned long long)length));
129                 dump_data(11, sig->data, sig->length);
130         }
131         return NT_STATUS_OK;
132 }
133
134 /* TODO: make this non-public */
135 _PUBLIC_ NTSTATUS gensec_ntlmssp_sign_packet(struct gensec_security *gensec_security, 
136                                     TALLOC_CTX *sig_mem_ctx, 
137                                     const uint8_t *data, size_t length, 
138                                     const uint8_t *whole_pdu, size_t pdu_length, 
139                                     DATA_BLOB *sig)
140 {
141         struct gensec_ntlmssp_state *gensec_ntlmssp_state = gensec_security->private_data;
142
143         return ntlmssp_make_packet_signature(gensec_ntlmssp_state, sig_mem_ctx, 
144                                              data, length, 
145                                              whole_pdu, pdu_length, 
146                                              NTLMSSP_SEND, sig, True);
147 }
148
149 /**
150  * Check the signature of an incoming packet 
151  *
152  */
153
154 NTSTATUS gensec_ntlmssp_check_packet(struct gensec_security *gensec_security, 
155                                      TALLOC_CTX *sig_mem_ctx, 
156                                      const uint8_t *data, size_t length, 
157                                      const uint8_t *whole_pdu, size_t pdu_length, 
158                                      const DATA_BLOB *sig)
159 {
160         struct gensec_ntlmssp_state *gensec_ntlmssp_state = gensec_security->private_data;
161
162         DATA_BLOB local_sig;
163         NTSTATUS nt_status;
164
165         if (!gensec_ntlmssp_state->session_key.length) {
166                 DEBUG(3, ("NO session key, cannot check packet signature\n"));
167                 return NT_STATUS_NO_USER_SESSION_KEY;
168         }
169
170         if (sig->length < 8) {
171                 DEBUG(1, ("NTLMSSP packet check failed due to short signature (%lu bytes)!\n", 
172                           (unsigned long)sig->length));
173         }
174
175         nt_status = ntlmssp_make_packet_signature(gensec_ntlmssp_state, sig_mem_ctx, 
176                                                   data, length, 
177                                                   whole_pdu, pdu_length, 
178                                                   NTLMSSP_RECEIVE, &local_sig, True);
179         
180         if (!NT_STATUS_IS_OK(nt_status)) {
181                 DEBUG(0, ("NTLMSSP packet check failed with %s\n", nt_errstr(nt_status)));
182                 return nt_status;
183         }
184
185         if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {
186                 if (local_sig.length != sig->length ||
187                     memcmp(local_sig.data, 
188                            sig->data, sig->length) != 0) {
189                         DEBUG(5, ("BAD SIG NTLM2: wanted signature over %llu bytes of input:\n", (unsigned long long)pdu_length));
190                         dump_data(5, local_sig.data, local_sig.length);
191                         
192                         DEBUG(5, ("BAD SIG: got signature over %llu bytes of input:\n", (unsigned long long)pdu_length));
193                         dump_data(5, sig->data, sig->length);
194                         
195                         DEBUG(1, ("NTLMSSP NTLM2 packet check failed due to invalid signature on %llu bytes of input!\n", (unsigned long long)pdu_length));
196                         return NT_STATUS_ACCESS_DENIED;
197                 }
198         } else {
199                 if (local_sig.length != sig->length ||
200                     memcmp(local_sig.data + 8, 
201                            sig->data + 8, sig->length - 8) != 0) {
202                         DEBUG(5, ("BAD SIG NTLM1: wanted signature of %llu bytes of input:\n", (unsigned long long)length));
203                         dump_data(5, local_sig.data, local_sig.length);
204                         
205                         DEBUG(5, ("BAD SIG: got signature of %llu bytes of input:\n", (unsigned long long)length));
206                         dump_data(5, sig->data, sig->length);
207                         
208                         DEBUG(1, ("NTLMSSP NTLM1 packet check failed due to invalid signature on %llu bytes of input:\n", (unsigned long long)length));
209                         return NT_STATUS_ACCESS_DENIED;
210                 }
211         }
212         dump_data_pw("checked ntlmssp signature\n", sig->data, sig->length);
213
214         return NT_STATUS_OK;
215 }
216
217
218 /**
219  * Seal data with the NTLMSSP algorithm
220  *
221  */
222
223 NTSTATUS gensec_ntlmssp_seal_packet(struct gensec_security *gensec_security, 
224                                     TALLOC_CTX *sig_mem_ctx, 
225                                     uint8_t *data, size_t length, 
226                                     const uint8_t *whole_pdu, size_t pdu_length, 
227                                     DATA_BLOB *sig)
228 {
229         struct gensec_ntlmssp_state *gensec_ntlmssp_state = gensec_security->private_data;
230         NTSTATUS nt_status;
231         if (!gensec_ntlmssp_state->session_key.length) {
232                 DEBUG(3, ("NO session key, cannot seal packet\n"));
233                 return NT_STATUS_NO_USER_SESSION_KEY;
234         }
235
236         DEBUG(10,("ntlmssp_seal_data: seal\n"));
237         dump_data_pw("ntlmssp clear data\n", data, length);
238         if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {
239                 /* The order of these two operations matters - we must first seal the packet,
240                    then seal the sequence number - this is becouse the send_seal_hash is not
241                    constant, but is is rather updated with each iteration */
242                 nt_status = ntlmssp_make_packet_signature(gensec_ntlmssp_state, sig_mem_ctx, 
243                                                           data, length, 
244                                                           whole_pdu, pdu_length, 
245                                                           NTLMSSP_SEND, sig, False);
246                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state, data, length);
247                 if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_KEY_EXCH) {
248                         arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state, sig->data+4, 8);
249                 }
250         } else {
251                 uint32_t crc;
252                 crc = crc32_calc_buffer(data, length);
253                 if (!msrpc_gen(sig_mem_ctx, sig, "dddd", NTLMSSP_SIGN_VERSION, 0, crc, gensec_ntlmssp_state->crypt.ntlm.seq_num)) {
254                         return NT_STATUS_NO_MEMORY;
255                 }
256
257                 /* The order of these two operations matters - we must
258                    first seal the packet, then seal the sequence
259                    number - this is becouse the ntlmssp_hash is not
260                    constant, but is is rather updated with each
261                    iteration */
262
263                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm.arcfour_state, data, length);
264                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm.arcfour_state, sig->data+4, sig->length-4);
265                 /* increment counter on send */
266                 gensec_ntlmssp_state->crypt.ntlm.seq_num++;
267                 nt_status = NT_STATUS_OK;
268         }
269         dump_data_pw("ntlmssp signature\n", sig->data, sig->length);
270         dump_data_pw("ntlmssp sealed data\n", data, length);
271
272
273         return nt_status;
274 }
275
276 /**
277  * Unseal data with the NTLMSSP algorithm
278  *
279  */
280
281 /*
282   wrappers for the ntlmssp_*() functions
283 */
284 NTSTATUS gensec_ntlmssp_unseal_packet(struct gensec_security *gensec_security, 
285                                       TALLOC_CTX *sig_mem_ctx, 
286                                       uint8_t *data, size_t length, 
287                                       const uint8_t *whole_pdu, size_t pdu_length, 
288                                       const DATA_BLOB *sig)
289 {
290         struct gensec_ntlmssp_state *gensec_ntlmssp_state = gensec_security->private_data;
291         if (!gensec_ntlmssp_state->session_key.length) {
292                 DEBUG(3, ("NO session key, cannot unseal packet\n"));
293                 return NT_STATUS_NO_USER_SESSION_KEY;
294         }
295
296         dump_data_pw("ntlmssp sealed data\n", data, length);
297         if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {
298                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm2.recv_seal_arcfour_state, data, length);
299         } else {
300                 arcfour_crypt_sbox(gensec_ntlmssp_state->crypt.ntlm.arcfour_state, data, length);
301         }
302         dump_data_pw("ntlmssp clear data\n", data, length);
303         return gensec_ntlmssp_check_packet(gensec_security, sig_mem_ctx, data, length, whole_pdu, pdu_length, sig);
304 }
305
306 /**
307    Initialise the state for NTLMSSP signing.
308 */
309 /* TODO: make this non-public */
310 _PUBLIC_ NTSTATUS ntlmssp_sign_init(struct gensec_ntlmssp_state *gensec_ntlmssp_state)
311 {
312         TALLOC_CTX *mem_ctx = talloc_new(gensec_ntlmssp_state);
313
314         if (!mem_ctx) {
315                 return NT_STATUS_NO_MEMORY;
316         }
317
318         DEBUG(3, ("NTLMSSP Sign/Seal - Initialising with flags:\n"));
319         debug_ntlmssp_flags(gensec_ntlmssp_state->neg_flags);
320
321         if (gensec_ntlmssp_state->session_key.length < 8) {
322                 talloc_free(mem_ctx);
323                 DEBUG(3, ("NO session key, cannot intialise signing\n"));
324                 return NT_STATUS_NO_USER_SESSION_KEY;
325         }
326
327         if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2)
328         {
329                 DATA_BLOB weak_session_key = gensec_ntlmssp_state->session_key;
330                 const char *send_sign_const;
331                 const char *send_seal_const;
332                 const char *recv_sign_const;
333                 const char *recv_seal_const;
334
335                 DATA_BLOB send_seal_key;
336                 DATA_BLOB recv_seal_key;
337
338                 switch (gensec_ntlmssp_state->role) {
339                 case NTLMSSP_CLIENT:
340                         send_sign_const = CLI_SIGN;
341                         send_seal_const = CLI_SEAL;
342                         recv_sign_const = SRV_SIGN;
343                         recv_seal_const = SRV_SEAL;
344                         break;
345                 case NTLMSSP_SERVER:
346                         send_sign_const = SRV_SIGN;
347                         send_seal_const = SRV_SEAL;
348                         recv_sign_const = CLI_SIGN;
349                         recv_seal_const = CLI_SEAL;
350                         break;
351                 default:
352                         talloc_free(mem_ctx);
353                         return NT_STATUS_INTERNAL_ERROR;
354                 }
355                 
356                 gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state = talloc(gensec_ntlmssp_state, struct arcfour_state);
357                 NT_STATUS_HAVE_NO_MEMORY(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state);
358                 gensec_ntlmssp_state->crypt.ntlm2.recv_seal_arcfour_state = talloc(gensec_ntlmssp_state, struct arcfour_state);
359                 NT_STATUS_HAVE_NO_MEMORY(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state);
360
361                 /**
362                    Weaken NTLMSSP keys to cope with down-level
363                    clients, servers and export restrictions.
364                    
365                    We probably should have some parameters to control
366                    this, once we get NTLM2 working.
367                 */
368
369                 /* Key weakening was not performed on the master key
370                  * for NTLM2 (in ntlmssp_weaken_keys()), but must be
371                  * done on the encryption subkeys only.  That is why
372                  * we don't have this code for the ntlmv1 case.
373                  */
374
375                 if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_128) {
376                         
377                 } else if (gensec_ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_56) {
378                         weak_session_key.length = 7;
379                 } else { /* forty bits */
380                         weak_session_key.length = 5;
381                 }
382                 dump_data_pw("NTLMSSP weakend master key:\n",
383                              weak_session_key.data, 
384                              weak_session_key.length);
385
386                 /* SEND: sign key */
387                 calc_ntlmv2_key(gensec_ntlmssp_state, 
388                                 &gensec_ntlmssp_state->crypt.ntlm2.send_sign_key, 
389                                 gensec_ntlmssp_state->session_key, send_sign_const);
390                 dump_data_pw("NTLMSSP send sign key:\n",
391                              gensec_ntlmssp_state->crypt.ntlm2.send_sign_key.data, 
392                              gensec_ntlmssp_state->crypt.ntlm2.send_sign_key.length);
393                 
394                 /* SEND: seal ARCFOUR pad */
395                 calc_ntlmv2_key(mem_ctx, 
396                                 &send_seal_key, 
397                                 weak_session_key, send_seal_const);
398                 dump_data_pw("NTLMSSP send seal key:\n",
399                              send_seal_key.data, 
400                              send_seal_key.length);
401                 arcfour_init(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state, 
402                              &send_seal_key);
403                 dump_data_pw("NTLMSSP send sesl hash:\n", 
404                              gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state->sbox, 
405                              sizeof(gensec_ntlmssp_state->crypt.ntlm2.send_seal_arcfour_state->sbox));
406
407                 /* RECV: sign key */
408                 calc_ntlmv2_key(gensec_ntlmssp_state, 
409                                 &gensec_ntlmssp_state->crypt.ntlm2.recv_sign_key, 
410                                 gensec_ntlmssp_state->session_key, recv_sign_const);
411                 dump_data_pw("NTLMSSP recv sign key:\n",
412                              gensec_ntlmssp_state->crypt.ntlm2.recv_sign_key.data, 
413                              gensec_ntlmssp_state->crypt.ntlm2.recv_sign_key.length);
414
415                 /* RECV: seal ARCFOUR pad */
416                 calc_ntlmv2_key(mem_ctx, 
417                                 &recv_seal_key, 
418                                 weak_session_key, recv_seal_const);
419                 dump_data_pw("NTLMSSP recv seal key:\n",
420                              recv_seal_key.data, 
421                              recv_seal_key.length);
422                 arcfour_init(gensec_ntlmssp_state->crypt.ntlm2.recv_seal_arcfour_state, 
423                              &recv_seal_key);
424                 dump_data_pw("NTLMSSP receive seal hash:\n", 
425                              gensec_ntlmssp_state->crypt.ntlm2.recv_seal_arcfour_state->sbox, 
426                              sizeof(gensec_ntlmssp_state->crypt.ntlm2.recv_seal_arcfour_state->sbox));
427
428                 gensec_ntlmssp_state->crypt.ntlm2.send_seq_num = 0;
429                 gensec_ntlmssp_state->crypt.ntlm2.recv_seq_num = 0;
430
431         } else {
432                 DATA_BLOB weak_session_key = ntlmssp_weakend_key(gensec_ntlmssp_state, mem_ctx);
433                 DEBUG(5, ("NTLMSSP Sign/Seal - using NTLM1\n"));
434
435                 gensec_ntlmssp_state->crypt.ntlm.arcfour_state = talloc(gensec_ntlmssp_state, struct arcfour_state);
436                 NT_STATUS_HAVE_NO_MEMORY(gensec_ntlmssp_state->crypt.ntlm.arcfour_state);
437
438                 arcfour_init(gensec_ntlmssp_state->crypt.ntlm.arcfour_state, 
439                              &weak_session_key);
440                 dump_data_pw("NTLMSSP hash:\n", gensec_ntlmssp_state->crypt.ntlm.arcfour_state->sbox,
441                              sizeof(gensec_ntlmssp_state->crypt.ntlm.arcfour_state->sbox));
442
443                 gensec_ntlmssp_state->crypt.ntlm.seq_num = 0;
444         }
445
446         talloc_free(mem_ctx);
447         return NT_STATUS_OK;
448 }
449
450 size_t gensec_ntlmssp_sig_size(struct gensec_security *gensec_security, size_t data_size) 
451 {
452         return NTLMSSP_SIG_SIZE;
453 }
454
455 NTSTATUS gensec_ntlmssp_wrap(struct gensec_security *gensec_security, 
456                              TALLOC_CTX *sig_mem_ctx, 
457                              const DATA_BLOB *in, 
458                              DATA_BLOB *out)
459 {
460         DATA_BLOB sig;
461         NTSTATUS nt_status;
462
463         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
464
465                 *out = data_blob_talloc(sig_mem_ctx, NULL, in->length + NTLMSSP_SIG_SIZE);
466                 if (!out->data) {
467                         return NT_STATUS_NO_MEMORY;
468                 }
469                 memcpy(out->data + NTLMSSP_SIG_SIZE, in->data, in->length);
470                 
471                 nt_status = gensec_ntlmssp_seal_packet(gensec_security, sig_mem_ctx, 
472                                                        out->data + NTLMSSP_SIG_SIZE, 
473                                                        out->length - NTLMSSP_SIG_SIZE, 
474                                                        out->data + NTLMSSP_SIG_SIZE, 
475                                                        out->length - NTLMSSP_SIG_SIZE, 
476                                                        &sig);
477                 
478                 if (NT_STATUS_IS_OK(nt_status)) {
479                         memcpy(out->data, sig.data, NTLMSSP_SIG_SIZE);
480                 }
481                 return nt_status;
482
483         } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
484
485                 *out = data_blob_talloc(sig_mem_ctx, NULL, in->length + NTLMSSP_SIG_SIZE);
486                 if (!out->data) {
487                         return NT_STATUS_NO_MEMORY;
488                 }
489                 memcpy(out->data + NTLMSSP_SIG_SIZE, in->data, in->length);
490
491                 nt_status = gensec_ntlmssp_sign_packet(gensec_security, sig_mem_ctx, 
492                                                        out->data + NTLMSSP_SIG_SIZE, 
493                                                        out->length - NTLMSSP_SIG_SIZE, 
494                                                        out->data + NTLMSSP_SIG_SIZE, 
495                                                        out->length - NTLMSSP_SIG_SIZE, 
496                                                        &sig);
497
498                 if (NT_STATUS_IS_OK(nt_status)) {
499                         memcpy(out->data, sig.data, NTLMSSP_SIG_SIZE);
500                 }
501                 return nt_status;
502
503         } else {
504                 *out = *in;
505                 return NT_STATUS_OK;
506         }
507 }
508
509
510 NTSTATUS gensec_ntlmssp_unwrap(struct gensec_security *gensec_security, 
511                                TALLOC_CTX *sig_mem_ctx, 
512                                const DATA_BLOB *in, 
513                                DATA_BLOB *out)
514 {
515         DATA_BLOB sig;
516
517         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
518                 if (in->length < NTLMSSP_SIG_SIZE) {
519                         return NT_STATUS_INVALID_PARAMETER;
520                 }
521                 sig.data = in->data;
522                 sig.length = NTLMSSP_SIG_SIZE;
523
524                 *out = data_blob_talloc(sig_mem_ctx, in->data + NTLMSSP_SIG_SIZE, in->length - NTLMSSP_SIG_SIZE);
525                 
526                 return gensec_ntlmssp_unseal_packet(gensec_security, sig_mem_ctx, 
527                                                     out->data, out->length, 
528                                                     out->data, out->length, 
529                                                     &sig);
530                                                   
531         } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
532                 if (in->length < NTLMSSP_SIG_SIZE) {
533                         return NT_STATUS_INVALID_PARAMETER;
534                 }
535                 sig.data = in->data;
536                 sig.length = NTLMSSP_SIG_SIZE;
537
538                 *out = data_blob_talloc(sig_mem_ctx, in->data + NTLMSSP_SIG_SIZE, in->length - NTLMSSP_SIG_SIZE);
539                 
540                 return gensec_ntlmssp_check_packet(gensec_security, sig_mem_ctx, 
541                                                    out->data, out->length, 
542                                                    out->data, out->length, 
543                                                    &sig);
544         } else {
545                 *out = *in;
546                 return NT_STATUS_OK;
547         }
548 }
549