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