3822ee191ea5582e586ad55766de316cb38cad79
[jra/samba/.git] / source3 / smbd / seal.c
1 /* 
2    Unix SMB/CIFS implementation.
3    SMB Transport encryption (sealing) code - server code.
4    Copyright (C) Jeremy Allison 2007.
5    
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 3 of the License, or
9    (at your option) any later version.
10    
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "includes.h"
21
22 /******************************************************************************
23  Server side encryption.
24 ******************************************************************************/
25
26 /******************************************************************************
27  Global server state.
28 ******************************************************************************/
29
30 struct smb_srv_trans_enc_ctx {
31         struct smb_trans_enc_state *es;
32         AUTH_NTLMSSP_STATE *auth_ntlmssp_state; /* Must be kept in sync with pointer in ec->ntlmssp_state. */
33 };
34
35 static struct smb_srv_trans_enc_ctx *partial_srv_trans_enc_ctx;
36 static struct smb_srv_trans_enc_ctx *srv_trans_enc_ctx;
37
38 /******************************************************************************
39  Return global enc context - this must change if we ever do multiple contexts.
40 ******************************************************************************/
41
42 uint16_t srv_enc_ctx(void)
43 {
44         return srv_trans_enc_ctx->es->enc_ctx_num;
45 }
46
47 /******************************************************************************
48  Is this an incoming encrypted packet ?
49 ******************************************************************************/
50
51 bool is_encrypted_packet(const uint8_t *inbuf)
52 {
53         NTSTATUS status;
54         uint16_t enc_num;
55
56         /* Ignore non-session messages or non 0xFF'E' messages. */
57         if(CVAL(inbuf,0) || !(inbuf[4] == 0xFF && inbuf[5] == 'E')) {
58                 return false;
59         }
60
61         status = get_enc_ctx_num(inbuf, &enc_num);
62         if (!NT_STATUS_IS_OK(status)) {
63                 return false;
64         }
65
66         /* Encrypted messages are 0xFF'E'<ctx> */
67         if (srv_trans_enc_ctx && enc_num == srv_enc_ctx()) {
68                 return true;
69         }
70         return false;
71 }
72
73 /******************************************************************************
74  Create an auth_ntlmssp_state and ensure pointer copy is correct.
75 ******************************************************************************/
76
77 static NTSTATUS make_auth_ntlmssp(struct smb_srv_trans_enc_ctx *ec)
78 {
79         NTSTATUS status = auth_ntlmssp_start(&ec->auth_ntlmssp_state);
80         if (!NT_STATUS_IS_OK(status)) {
81                 return nt_status_squash(status);
82         }
83
84         /*
85          * We must remember to update the pointer copy for the common
86          * functions after any auth_ntlmssp_start/auth_ntlmssp_end.
87          */
88         ec->es->s.ntlmssp_state = ec->auth_ntlmssp_state->ntlmssp_state;
89         return status;
90 }
91
92 /******************************************************************************
93  Destroy an auth_ntlmssp_state and ensure pointer copy is correct.
94 ******************************************************************************/
95
96 static void destroy_auth_ntlmssp(struct smb_srv_trans_enc_ctx *ec)
97 {
98         /*
99          * We must remember to update the pointer copy for the common
100          * functions after any auth_ntlmssp_start/auth_ntlmssp_end.
101          */
102
103         if (ec->auth_ntlmssp_state) {
104                 auth_ntlmssp_end(&ec->auth_ntlmssp_state);
105                 /* The auth_ntlmssp_end killed this already. */
106                 ec->es->s.ntlmssp_state = NULL;
107         }
108 }
109
110 #if defined(HAVE_GSSAPI) && defined(HAVE_KRB5)
111
112 /******************************************************************************
113  Import a name.
114 ******************************************************************************/
115
116 static NTSTATUS get_srv_gss_creds(const char *service,
117                                 const char *name,
118                                 gss_cred_usage_t cred_type,
119                                 gss_cred_id_t *p_srv_cred)
120 {
121         OM_uint32 ret;
122         OM_uint32 min;
123         gss_name_t srv_name;
124         gss_buffer_desc input_name;
125         char *host_princ_s = NULL;
126         NTSTATUS status = NT_STATUS_OK;
127
128         gss_OID_desc nt_hostbased_service =
129         {10, CONST_DISCARD(char *,"\x2a\x86\x48\x86\xf7\x12\x01\x02\x01\x04")};
130
131         asprintf(&host_princ_s, "%s@%s", service, name);
132         if (host_princ_s == NULL) {
133                 return NT_STATUS_NO_MEMORY;
134         }
135
136         input_name.value = host_princ_s;
137         input_name.length = strlen(host_princ_s) + 1;
138
139         ret = gss_import_name(&min,
140                                 &input_name,
141                                 &nt_hostbased_service,
142                                 &srv_name);
143
144         DEBUG(10,("get_srv_gss_creds: imported name %s\n",
145                 host_princ_s ));
146
147         if (ret != GSS_S_COMPLETE) {
148                 SAFE_FREE(host_princ_s);
149                 return map_nt_error_from_gss(ret, min);
150         }
151
152         /*
153          * We're accessing the krb5.keytab file here.
154          * ensure we have permissions to do so.
155          */
156         become_root();
157
158         ret = gss_acquire_cred(&min,
159                                 srv_name,
160                                 GSS_C_INDEFINITE,
161                                 GSS_C_NULL_OID_SET,
162                                 cred_type,
163                                 p_srv_cred,
164                                 NULL,
165                                 NULL);
166         unbecome_root();
167
168         if (ret != GSS_S_COMPLETE) {
169                 ADS_STATUS adss = ADS_ERROR_GSS(ret, min);
170                 DEBUG(10,("get_srv_gss_creds: gss_acquire_cred failed with %s\n",
171                         ads_errstr(adss)));
172                 status = map_nt_error_from_gss(ret, min);
173         }
174
175         SAFE_FREE(host_princ_s);
176         gss_release_name(&min, &srv_name);
177         return status;
178 }
179
180 /******************************************************************************
181  Create a gss state.
182  Try and get the cifs/server@realm principal first, then fall back to
183  host/server@realm.
184 ******************************************************************************/
185
186 static NTSTATUS make_auth_gss(struct smb_srv_trans_enc_ctx *ec)
187 {
188         NTSTATUS status;
189         gss_cred_id_t srv_cred;
190         fstring fqdn;
191
192         name_to_fqdn(fqdn, global_myname());
193         strlower_m(fqdn);
194
195         status = get_srv_gss_creds("cifs", fqdn, GSS_C_ACCEPT, &srv_cred);
196         if (!NT_STATUS_IS_OK(status)) {
197                 status = get_srv_gss_creds("host", fqdn, GSS_C_ACCEPT, &srv_cred);
198                 if (!NT_STATUS_IS_OK(status)) {
199                         return nt_status_squash(status);
200                 }
201         }
202
203         ec->es->s.gss_state = SMB_MALLOC_P(struct smb_tran_enc_state_gss);
204         if (!ec->es->s.gss_state) {
205                 OM_uint32 min;
206                 gss_release_cred(&min, &srv_cred);
207                 return NT_STATUS_NO_MEMORY;
208         }
209         ZERO_STRUCTP(ec->es->s.gss_state);
210         ec->es->s.gss_state->creds = srv_cred;
211
212         /* No context yet. */
213         ec->es->s.gss_state->gss_ctx = GSS_C_NO_CONTEXT;
214
215         return NT_STATUS_OK;
216 }
217 #endif
218
219 /******************************************************************************
220  Shutdown a server encryption context.
221 ******************************************************************************/
222
223 static void srv_free_encryption_context(struct smb_srv_trans_enc_ctx **pp_ec)
224 {
225         struct smb_srv_trans_enc_ctx *ec = *pp_ec;
226
227         if (!ec) {
228                 return;
229         }
230
231         if (ec->es) {
232                 switch (ec->es->smb_enc_type) {
233                         case SMB_TRANS_ENC_NTLM:
234                                 destroy_auth_ntlmssp(ec);
235                                 break;
236 #if defined(HAVE_GSSAPI) && defined(HAVE_KRB5)
237                         case SMB_TRANS_ENC_GSS:
238                                 break;
239 #endif
240                 }
241                 common_free_encryption_state(&ec->es);
242         }
243
244         SAFE_FREE(ec);
245         *pp_ec = NULL;
246 }
247
248 /******************************************************************************
249  Create a server encryption context.
250 ******************************************************************************/
251
252 static NTSTATUS make_srv_encryption_context(enum smb_trans_enc_type smb_enc_type, struct smb_srv_trans_enc_ctx **pp_ec)
253 {
254         struct smb_srv_trans_enc_ctx *ec;
255
256         *pp_ec = NULL;
257
258         ec = SMB_MALLOC_P(struct smb_srv_trans_enc_ctx);
259         if (!ec) {
260                 return NT_STATUS_NO_MEMORY;
261         }
262         ZERO_STRUCTP(partial_srv_trans_enc_ctx);
263         ec->es = SMB_MALLOC_P(struct smb_trans_enc_state);
264         if (!ec->es) {
265                 SAFE_FREE(ec);
266                 return NT_STATUS_NO_MEMORY;
267         }
268         ZERO_STRUCTP(ec->es);
269         ec->es->smb_enc_type = smb_enc_type;
270         switch (smb_enc_type) {
271                 case SMB_TRANS_ENC_NTLM:
272                         {
273                                 NTSTATUS status = make_auth_ntlmssp(ec);
274                                 if (!NT_STATUS_IS_OK(status)) {
275                                         srv_free_encryption_context(&ec);
276                                         return status;
277                                 }
278                         }
279                         break;
280
281 #if defined(HAVE_GSSAPI) && defined(HAVE_KRB5)
282                 case SMB_TRANS_ENC_GSS:
283                         /* Acquire our credentials by calling gss_acquire_cred here. */
284                         {
285                                 NTSTATUS status = make_auth_gss(ec);
286                                 if (!NT_STATUS_IS_OK(status)) {
287                                         srv_free_encryption_context(&ec);
288                                         return status;
289                                 }
290                         }
291                         break;
292 #endif
293                 default:
294                         srv_free_encryption_context(&ec);
295                         return NT_STATUS_INVALID_PARAMETER;
296         }
297         *pp_ec = ec;
298         return NT_STATUS_OK;
299 }
300
301 /******************************************************************************
302  Free an encryption-allocated buffer.
303 ******************************************************************************/
304
305 void srv_free_enc_buffer(char *buf)
306 {
307         /* We know this is an smb buffer, and we
308          * didn't malloc, only copy, for a keepalive,
309          * so ignore non-session messages. */
310
311         if(CVAL(buf,0)) {
312                 return;
313         }
314
315         if (srv_trans_enc_ctx) {
316                 common_free_enc_buffer(srv_trans_enc_ctx->es, buf);
317         }
318 }
319
320 /******************************************************************************
321  Decrypt an incoming buffer.
322 ******************************************************************************/
323
324 NTSTATUS srv_decrypt_buffer(char *buf)
325 {
326         /* Ignore non-session messages. */
327         if(CVAL(buf,0)) {
328                 return NT_STATUS_OK;
329         }
330
331         if (srv_trans_enc_ctx) {
332                 return common_decrypt_buffer(srv_trans_enc_ctx->es, buf);
333         }
334
335         return NT_STATUS_OK;
336 }
337
338 /******************************************************************************
339  Encrypt an outgoing buffer. Return the encrypted pointer in buf_out.
340 ******************************************************************************/
341
342 NTSTATUS srv_encrypt_buffer(char *buf, char **buf_out)
343 {
344         *buf_out = buf;
345
346         /* Ignore non-session messages. */
347         if(CVAL(buf,0)) {
348                 return NT_STATUS_OK;
349         }
350
351         if (srv_trans_enc_ctx) {
352                 return common_encrypt_buffer(srv_trans_enc_ctx->es, buf, buf_out);
353         }
354         /* Not encrypting. */
355         return NT_STATUS_OK;
356 }
357
358 /******************************************************************************
359  Do the gss encryption negotiation. Parameters are in/out.
360  Until success we do everything on the partial enc ctx.
361 ******************************************************************************/
362
363 #if defined(HAVE_GSSAPI) && defined(HAVE_KRB5)
364 static NTSTATUS srv_enc_spnego_gss_negotiate(unsigned char **ppdata, size_t *p_data_size, DATA_BLOB secblob)
365 {
366         OM_uint32 ret;
367         OM_uint32 min;
368         OM_uint32 flags = 0;
369         gss_buffer_desc in_buf, out_buf;
370         struct smb_tran_enc_state_gss *gss_state;
371         DATA_BLOB auth_reply = data_blob_null;
372         DATA_BLOB response = data_blob_null;
373         NTSTATUS status;
374
375         if (!partial_srv_trans_enc_ctx) {
376                 status = make_srv_encryption_context(SMB_TRANS_ENC_GSS, &partial_srv_trans_enc_ctx);
377                 if (!NT_STATUS_IS_OK(status)) {
378                         return status;
379                 }
380         }
381
382         gss_state = partial_srv_trans_enc_ctx->es->s.gss_state;
383
384         in_buf.value = secblob.data;
385         in_buf.length = secblob.length;
386
387         out_buf.value = NULL;
388         out_buf.length = 0;
389
390         become_root();
391
392         ret = gss_accept_sec_context(&min,
393                                 &gss_state->gss_ctx,
394                                 gss_state->creds,
395                                 &in_buf,
396                                 GSS_C_NO_CHANNEL_BINDINGS,
397                                 NULL,
398                                 NULL,           /* Ignore oids. */
399                                 &out_buf,       /* To return. */
400                                 &flags,
401                                 NULL,           /* Ingore time. */
402                                 NULL);          /* Ignore delegated creds. */
403         unbecome_root();
404
405         status = gss_err_to_ntstatus(ret, min);
406         if (ret != GSS_S_COMPLETE && ret != GSS_S_CONTINUE_NEEDED) {
407                 return status;
408         }
409
410         /* Ensure we've got sign+seal available. */
411         if (ret == GSS_S_COMPLETE) {
412                 if ((flags & (GSS_C_INTEG_FLAG|GSS_C_CONF_FLAG|GSS_C_REPLAY_FLAG|GSS_C_SEQUENCE_FLAG)) !=
413                                 (GSS_C_INTEG_FLAG|GSS_C_CONF_FLAG|GSS_C_REPLAY_FLAG|GSS_C_SEQUENCE_FLAG)) {
414                         DEBUG(0,("srv_enc_spnego_gss_negotiate: quality of service not good enough "
415                                 "for SMB sealing.\n"));
416                         gss_release_buffer(&min, &out_buf);
417                         return NT_STATUS_ACCESS_DENIED;
418                 }
419         }
420
421         auth_reply = data_blob(out_buf.value, out_buf.length);
422         gss_release_buffer(&min, &out_buf);
423
424         /* Wrap in SPNEGO. */
425         response = spnego_gen_auth_response(&auth_reply, status, OID_KERBEROS5);
426         data_blob_free(&auth_reply);
427
428         SAFE_FREE(*ppdata);
429         *ppdata = (unsigned char *)memdup(response.data, response.length);
430         if ((*ppdata) == NULL && response.length > 0) {
431                 status = NT_STATUS_NO_MEMORY;
432         }
433         *p_data_size = response.length;
434
435         data_blob_free(&response);
436
437         return status;
438 }
439 #endif
440
441 /******************************************************************************
442  Do the NTLM SPNEGO (or raw) encryption negotiation. Parameters are in/out.
443  Until success we do everything on the partial enc ctx.
444 ******************************************************************************/
445
446 static NTSTATUS srv_enc_ntlm_negotiate(unsigned char **ppdata, size_t *p_data_size, DATA_BLOB secblob, bool spnego_wrap)
447 {
448         NTSTATUS status;
449         DATA_BLOB chal = data_blob_null;
450         DATA_BLOB response = data_blob_null;
451
452         status = make_srv_encryption_context(SMB_TRANS_ENC_NTLM, &partial_srv_trans_enc_ctx);
453         if (!NT_STATUS_IS_OK(status)) {
454                 return status;
455         }
456
457         status = auth_ntlmssp_update(partial_srv_trans_enc_ctx->auth_ntlmssp_state, secblob, &chal);
458
459         /* status here should be NT_STATUS_MORE_PROCESSING_REQUIRED
460          * for success ... */
461
462         if (spnego_wrap) {
463                 response = spnego_gen_auth_response(&chal, status, OID_NTLMSSP);
464                 data_blob_free(&chal);
465         } else {
466                 /* Return the raw blob. */
467                 response = chal;
468         }
469
470         SAFE_FREE(*ppdata);
471         *ppdata = (unsigned char *)memdup(response.data, response.length);
472         if ((*ppdata) == NULL && response.length > 0) {
473                 status = NT_STATUS_NO_MEMORY;
474         }
475         *p_data_size = response.length;
476         data_blob_free(&response);
477
478         return status;
479 }
480
481 /******************************************************************************
482  Do the SPNEGO encryption negotiation. Parameters are in/out.
483  Based off code in smbd/sesssionsetup.c
484  Until success we do everything on the partial enc ctx.
485 ******************************************************************************/
486
487 static NTSTATUS srv_enc_spnego_negotiate(connection_struct *conn,
488                                         unsigned char **ppdata,
489                                         size_t *p_data_size,
490                                         unsigned char **pparam,
491                                         size_t *p_param_size)
492 {
493         NTSTATUS status;
494         DATA_BLOB blob = data_blob_null;
495         DATA_BLOB secblob = data_blob_null;
496         char *kerb_mech = NULL;
497
498         blob = data_blob_const(*ppdata, *p_data_size);
499
500         status = parse_spnego_mechanisms(blob, &secblob, &kerb_mech);
501         if (!NT_STATUS_IS_OK(status)) {
502                 return nt_status_squash(status);
503         }
504
505         /* We should have no partial context at this point. */
506
507         srv_free_encryption_context(&partial_srv_trans_enc_ctx);
508
509         if (kerb_mech) {
510                 SAFE_FREE(kerb_mech);
511
512 #if defined(HAVE_GSSAPI) && defined(HAVE_KRB5)
513                 status = srv_enc_spnego_gss_negotiate(ppdata, p_data_size, secblob);
514 #else
515                 /* Currently we don't SPNEGO negotiate
516                  * back to NTLMSSP as we do in sessionsetupX. We should... */
517                 return NT_STATUS_LOGON_FAILURE;
518 #endif
519         } else {
520                 status = srv_enc_ntlm_negotiate(ppdata, p_data_size, secblob, true);
521         }
522
523         data_blob_free(&secblob);
524
525         if (!NT_STATUS_EQUAL(status,NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(status)) {
526                 srv_free_encryption_context(&partial_srv_trans_enc_ctx);
527                 return nt_status_squash(status);
528         }
529
530         if (NT_STATUS_IS_OK(status)) {
531                 /* Return the context we're using for this encryption state. */
532                 if (!(*pparam = SMB_MALLOC_ARRAY(unsigned char, 2))) {
533                         return NT_STATUS_NO_MEMORY;
534                 }
535                 SSVAL(*pparam,0,partial_srv_trans_enc_ctx->es->enc_ctx_num);
536                 *p_param_size = 2;
537         }
538
539         return status;
540 }
541
542 /******************************************************************************
543  Complete a SPNEGO encryption negotiation. Parameters are in/out.
544  We only get this for a NTLM auth second stage.
545 ******************************************************************************/
546
547 static NTSTATUS srv_enc_spnego_ntlm_auth(connection_struct *conn,
548                                         unsigned char **ppdata,
549                                         size_t *p_data_size,
550                                         unsigned char **pparam,
551                                         size_t *p_param_size)
552 {
553         NTSTATUS status;
554         DATA_BLOB blob = data_blob_null;
555         DATA_BLOB auth = data_blob_null;
556         DATA_BLOB auth_reply = data_blob_null;
557         DATA_BLOB response = data_blob_null;
558         struct smb_srv_trans_enc_ctx *ec = partial_srv_trans_enc_ctx;
559
560         /* We must have a partial context here. */
561
562         if (!ec || !ec->es || ec->auth_ntlmssp_state == NULL || ec->es->smb_enc_type != SMB_TRANS_ENC_NTLM) {
563                 srv_free_encryption_context(&partial_srv_trans_enc_ctx);
564                 return NT_STATUS_INVALID_PARAMETER;
565         }
566
567         blob = data_blob_const(*ppdata, *p_data_size);
568         if (!spnego_parse_auth(blob, &auth)) {
569                 srv_free_encryption_context(&partial_srv_trans_enc_ctx);
570                 return NT_STATUS_INVALID_PARAMETER;
571         }
572
573         status = auth_ntlmssp_update(ec->auth_ntlmssp_state, auth, &auth_reply);
574         data_blob_free(&auth);
575
576         /* From RFC4178.
577          *
578          *    supportedMech
579          *
580          *          This field SHALL only be present in the first reply from the
581          *                target.
582          * So set mechOID to NULL here.
583          */
584
585         response = spnego_gen_auth_response(&auth_reply, status, NULL);
586         data_blob_free(&auth_reply);
587
588         if (NT_STATUS_IS_OK(status)) {
589                 /* Return the context we're using for this encryption state. */
590                 if (!(*pparam = SMB_MALLOC_ARRAY(unsigned char, 2))) {
591                         return NT_STATUS_NO_MEMORY;
592                 }
593                 SSVAL(*pparam,0,ec->es->enc_ctx_num);
594                 *p_param_size = 2;
595         }
596
597         SAFE_FREE(*ppdata);
598         *ppdata = (unsigned char *)memdup(response.data, response.length);
599         if ((*ppdata) == NULL && response.length > 0)
600                 return NT_STATUS_NO_MEMORY;
601         *p_data_size = response.length;
602         data_blob_free(&response);
603         return status;
604 }
605
606 /******************************************************************************
607  Raw NTLM encryption negotiation. Parameters are in/out.
608  This function does both steps.
609 ******************************************************************************/
610
611 static NTSTATUS srv_enc_raw_ntlm_auth(connection_struct *conn,
612                                         unsigned char **ppdata,
613                                         size_t *p_data_size,
614                                         unsigned char **pparam,
615                                         size_t *p_param_size)
616 {
617         NTSTATUS status;
618         DATA_BLOB blob = data_blob_const(*ppdata, *p_data_size);
619         DATA_BLOB response = data_blob_null;
620         struct smb_srv_trans_enc_ctx *ec;
621
622         if (!partial_srv_trans_enc_ctx) {
623                 /* This is the initial step. */
624                 status = srv_enc_ntlm_negotiate(ppdata, p_data_size, blob, false);
625                 if (!NT_STATUS_EQUAL(status,NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(status)) {
626                         srv_free_encryption_context(&partial_srv_trans_enc_ctx);
627                         return nt_status_squash(status);
628                 }
629                 return status;
630         }
631
632         ec = partial_srv_trans_enc_ctx;
633         if (!ec || !ec->es || ec->auth_ntlmssp_state == NULL || ec->es->smb_enc_type != SMB_TRANS_ENC_NTLM) {
634                 srv_free_encryption_context(&partial_srv_trans_enc_ctx);
635                 return NT_STATUS_INVALID_PARAMETER;
636         }
637
638         /* Second step. */
639         status = auth_ntlmssp_update(partial_srv_trans_enc_ctx->auth_ntlmssp_state, blob, &response);
640
641         if (NT_STATUS_IS_OK(status)) {
642                 /* Return the context we're using for this encryption state. */
643                 if (!(*pparam = SMB_MALLOC_ARRAY(unsigned char, 2))) {
644                         return NT_STATUS_NO_MEMORY;
645                 }
646                 SSVAL(*pparam,0,ec->es->enc_ctx_num);
647                 *p_param_size = 2;
648         }
649
650         /* Return the raw blob. */
651         SAFE_FREE(*ppdata);
652         *ppdata = (unsigned char *)memdup(response.data, response.length);
653         if ((*ppdata) == NULL && response.length > 0)
654                 return NT_STATUS_NO_MEMORY;
655         *p_data_size = response.length;
656         data_blob_free(&response);
657         return status;
658 }
659
660 /******************************************************************************
661  Do the SPNEGO encryption negotiation. Parameters are in/out.
662 ******************************************************************************/
663
664 NTSTATUS srv_request_encryption_setup(connection_struct *conn,
665                                         unsigned char **ppdata,
666                                         size_t *p_data_size,
667                                         unsigned char **pparam,
668                                         size_t *p_param_size)
669 {
670         unsigned char *pdata = *ppdata;
671
672         SAFE_FREE(*pparam);
673         *p_param_size = 0;
674
675         if (*p_data_size < 1) {
676                 return NT_STATUS_INVALID_PARAMETER;
677         }
678
679         if (pdata[0] == ASN1_APPLICATION(0)) {
680                 /* its a negTokenTarg packet */
681                 return srv_enc_spnego_negotiate(conn, ppdata, p_data_size, pparam, p_param_size);
682         }
683
684         if (pdata[0] == ASN1_CONTEXT(1)) {
685                 /* It's an auth packet */
686                 return srv_enc_spnego_ntlm_auth(conn, ppdata, p_data_size, pparam, p_param_size);
687         }
688
689         /* Maybe it's a raw unwrapped auth ? */
690         if (*p_data_size < 7) {
691                 return NT_STATUS_INVALID_PARAMETER;
692         }
693
694         if (strncmp((char *)pdata, "NTLMSSP", 7) == 0) {
695                 return srv_enc_raw_ntlm_auth(conn, ppdata, p_data_size, pparam, p_param_size);
696         }
697
698         DEBUG(1,("srv_request_encryption_setup: Unknown packet\n"));
699
700         return NT_STATUS_LOGON_FAILURE;
701 }
702
703 /******************************************************************************
704  Negotiation was successful - turn on server-side encryption.
705 ******************************************************************************/
706
707 static NTSTATUS check_enc_good(struct smb_srv_trans_enc_ctx *ec)
708 {
709         if (!ec || !ec->es) {
710                 return NT_STATUS_LOGON_FAILURE;
711         }
712
713         if (ec->es->smb_enc_type == SMB_TRANS_ENC_NTLM) {
714                 if ((ec->es->s.ntlmssp_state->neg_flags & (NTLMSSP_NEGOTIATE_SIGN|NTLMSSP_NEGOTIATE_SEAL)) !=
715                                 (NTLMSSP_NEGOTIATE_SIGN|NTLMSSP_NEGOTIATE_SEAL)) {
716                         return NT_STATUS_INVALID_PARAMETER;
717                 }
718         }
719         /* Todo - check gssapi case. */
720
721         return NT_STATUS_OK;
722 }
723
724 /******************************************************************************
725  Negotiation was successful - turn on server-side encryption.
726 ******************************************************************************/
727
728 NTSTATUS srv_encryption_start(connection_struct *conn)
729 {
730         NTSTATUS status;
731
732         /* Check that we are really doing sign+seal. */
733         status = check_enc_good(partial_srv_trans_enc_ctx);
734         if (!NT_STATUS_IS_OK(status)) {
735                 return status;
736         }
737         /* Throw away the context we're using currently (if any). */
738         srv_free_encryption_context(&srv_trans_enc_ctx);
739
740         /* Steal the partial pointer. Deliberate shallow copy. */
741         srv_trans_enc_ctx = partial_srv_trans_enc_ctx;
742         srv_trans_enc_ctx->es->enc_on = true;
743
744         partial_srv_trans_enc_ctx = NULL;
745
746         DEBUG(1,("srv_encryption_start: context negotiated\n"));
747         return NT_STATUS_OK;
748 }
749
750 /******************************************************************************
751  Shutdown all server contexts.
752 ******************************************************************************/
753
754 void server_encryption_shutdown(void)
755 {
756         srv_free_encryption_context(&partial_srv_trans_enc_ctx);
757         srv_free_encryption_context(&srv_trans_enc_ctx);
758 }