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