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