source3/smbd/smb2_sesssetup.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Core SMB2 server
   4
   5    Copyright (C) Stefan Metzmacher 2009
   6    Copyright (C) Jeremy Allison 2010
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "smbd/globals.h"
  24 #include "../libcli/smb/smb_common.h"
  25 #include "../libcli/auth/spnego.h"
  26 #include "../libcli/auth/ntlmssp.h"
  27 #include "ntlmssp_wrap.h"
  28 #include "../librpc/gen_ndr/krb5pac.h"
  29 #include "libads/kerberos_proto.h"
  30
  31 static NTSTATUS smbd_smb2_session_setup(struct smbd_smb2_request *smb2req,
  32                                         uint64_t in_session_id,
  33                                         uint8_t in_security_mode,
  34                                         DATA_BLOB in_security_buffer,
  35                                         uint16_t *out_session_flags,
  36                                         DATA_BLOB *out_security_buffer,
  37                                         uint64_t *out_session_id);
  38
  39 NTSTATUS smbd_smb2_request_process_sesssetup(struct smbd_smb2_request *smb2req)
  40 {
  41         const uint8_t *inhdr;
  42         const uint8_t *inbody;
  43         int i = smb2req->current_idx;
  44         uint8_t *outhdr;
  45         DATA_BLOB outbody;
  46         DATA_BLOB outdyn;
  47         size_t expected_body_size = 0x19;
  48         size_t body_size;
  49         uint64_t in_session_id;
  50         uint8_t in_security_mode;
  51         uint16_t in_security_offset;
  52         uint16_t in_security_length;
  53         DATA_BLOB in_security_buffer;
  54         uint16_t out_session_flags;
  55         uint64_t out_session_id;
  56         uint16_t out_security_offset;
  57         DATA_BLOB out_security_buffer;
  58         NTSTATUS status;
  59
  60         inhdr = (const uint8_t *)smb2req->in.vector[i+0].iov_base;
  61
  62         if (smb2req->in.vector[i+1].iov_len != (expected_body_size & 0xFFFFFFFE)) {
  63                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  64         }
  65
  66         inbody = (const uint8_t *)smb2req->in.vector[i+1].iov_base;
  67
  68         body_size = SVAL(inbody, 0x00);
  69         if (body_size != expected_body_size) {
  70                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  71         }
  72
  73         in_security_offset = SVAL(inbody, 0x0C);
  74         in_security_length = SVAL(inbody, 0x0E);
  75
  76         if (in_security_offset != (SMB2_HDR_BODY + (body_size & 0xFFFFFFFE))) {
  77                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  78         }
  79
  80         if (in_security_length > smb2req->in.vector[i+2].iov_len) {
  81                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  82         }
  83
  84         in_session_id = BVAL(inhdr, SMB2_HDR_SESSION_ID);
  85         in_security_mode = CVAL(inbody, 0x03);
  86         in_security_buffer.data = (uint8_t *)smb2req->in.vector[i+2].iov_base;
  87         in_security_buffer.length = in_security_length;
  88
  89         status = smbd_smb2_session_setup(smb2req,
  90                                          in_session_id,
  91                                          in_security_mode,
  92                                          in_security_buffer,
  93                                          &out_session_flags,
  94                                          &out_security_buffer,
  95                                          &out_session_id);
  96         if (!NT_STATUS_IS_OK(status) &&
  97             !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
  98                 status = nt_status_squash(status);
  99                 return smbd_smb2_request_error(smb2req, status);
 100         }
 101
 102         out_security_offset = SMB2_HDR_BODY + 0x08;
 103
 104         outhdr = (uint8_t *)smb2req->out.vector[i].iov_base;
 105
 106         outbody = data_blob_talloc(smb2req->out.vector, NULL, 0x08);
 107         if (outbody.data == NULL) {
 108                 return smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 109         }
 110
 111         SBVAL(outhdr, SMB2_HDR_SESSION_ID, out_session_id);
 112
 113         SSVAL(outbody.data, 0x00, 0x08 + 1);    /* struct size */
 114         SSVAL(outbody.data, 0x02,
 115               out_session_flags);               /* session flags */
 116         SSVAL(outbody.data, 0x04,
 117               out_security_offset);             /* security buffer offset */
 118         SSVAL(outbody.data, 0x06,
 119               out_security_buffer.length);      /* security buffer length */
 120
 121         outdyn = out_security_buffer;
 122
 123         return smbd_smb2_request_done_ex(smb2req, status, outbody, &outdyn,
 124                                          __location__);
 125 }
 126
 127 static int smbd_smb2_session_destructor(struct smbd_smb2_session *session)
 128 {
 129         if (session->sconn == NULL) {
 130                 return 0;
 131         }
 132
 133         /* first free all tcons */
 134         while (session->tcons.list) {
 135                 talloc_free(session->tcons.list);
 136         }
 137
 138         idr_remove(session->sconn->smb2.sessions.idtree, session->vuid);
 139         DLIST_REMOVE(session->sconn->smb2.sessions.list, session);
 140         invalidate_vuid(session->sconn, session->vuid);
 141
 142         session->vuid = 0;
 143         session->status = NT_STATUS_USER_SESSION_DELETED;
 144         session->sconn = NULL;
 145
 146         return 0;
 147 }
 148
 149 static NTSTATUS setup_ntlmssp_session_info(struct smbd_smb2_session *session,
 150                                 NTSTATUS status)
 151 {
 152         if (NT_STATUS_IS_OK(status)) {
 153                 status = auth_ntlmssp_steal_session_info(session,
 154                                 session->auth_ntlmssp_state,
 155                                 &session->session_info);
 156         } else {
 157                 /* Note that this session_info won't have a session
 158                  * key.  But for map to guest, that's exactly the right
 159                  * thing - we can't reasonably guess the key the
 160                  * client wants, as the password was wrong */
 161                 status = do_map_to_guest(status,
 162                         &session->session_info,
 163                         auth_ntlmssp_get_username(session->auth_ntlmssp_state),
 164                         auth_ntlmssp_get_domain(session->auth_ntlmssp_state));
 165         }
 166         return status;
 167 }
 168
 169 #ifdef HAVE_KRB5
 170 static NTSTATUS smbd_smb2_session_setup_krb5(struct smbd_smb2_session *session,
 171                                         struct smbd_smb2_request *smb2req,
 172                                         uint8_t in_security_mode,
 173                                         const DATA_BLOB *secblob,
 174                                         const char *mechOID,
 175                                         uint16_t *out_session_flags,
 176                                         DATA_BLOB *out_security_buffer,
 177                                         uint64_t *out_session_id)
 178 {
 179         DATA_BLOB ap_rep = data_blob_null;
 180         DATA_BLOB ap_rep_wrapped = data_blob_null;
 181         DATA_BLOB ticket = data_blob_null;
 182         DATA_BLOB session_key = data_blob_null;
 183         DATA_BLOB secblob_out = data_blob_null;
 184         uint8 tok_id[2];
 185         struct PAC_LOGON_INFO *logon_info = NULL;
 186         char *principal = NULL;
 187         char *user = NULL;
 188         char *domain = NULL;
 189         struct passwd *pw = NULL;
 190         NTSTATUS status;
 191         char *real_username;
 192         fstring tmp;
 193         bool username_was_mapped = false;
 194         bool map_domainuser_to_guest = false;
 195
 196         if (!spnego_parse_krb5_wrap(talloc_tos(), *secblob, &ticket, tok_id)) {
 197                 status = NT_STATUS_LOGON_FAILURE;
 198                 goto fail;
 199         }
 200
 201         status = ads_verify_ticket(smb2req, lp_realm(), 0, &ticket,
 202                                    &principal, &logon_info, &ap_rep,
 203                                    &session_key, true);
 204
 205         if (!NT_STATUS_IS_OK(status)) {
 206                 DEBUG(1,("smb2: Failed to verify incoming ticket with error %s!\n",
 207                         nt_errstr(status)));
 208                 if (!NT_STATUS_EQUAL(status, NT_STATUS_TIME_DIFFERENCE_AT_DC)) {
 209                         status = NT_STATUS_LOGON_FAILURE;
 210                 }
 211                 goto fail;
 212         }
 213
 214         status = get_user_from_kerberos_info(talloc_tos(),
 215                                              smb2req->sconn->client_id.name,
 216                                              principal, logon_info,
 217                                              &username_was_mapped,
 218                                              &map_domainuser_to_guest,
 219                                              &user, &domain,
 220                                              &real_username, &pw);
 221         if (!NT_STATUS_IS_OK(status)) {
 222                 goto fail;
 223         }
 224
 225         /* save the PAC data if we have it */
 226         if (logon_info) {
 227                 netsamlogon_cache_store(user, &logon_info->info3);
 228         }
 229
 230         /* setup the string used by %U */
 231         sub_set_smb_name(real_username);
 232
 233         /* reload services so that the new %U is taken into account */
 234         reload_services(smb2req->sconn->msg_ctx, smb2req->sconn->sock, true);
 235
 236         status = make_server_info_krb5(session,
 237                                         user, domain, real_username, pw,
 238                                         logon_info, map_domainuser_to_guest,
 239                                         &session->session_info);
 240         if (!NT_STATUS_IS_OK(status)) {
 241                 DEBUG(1, ("smb2: make_server_info_krb5 failed\n"));
 242                 goto fail;
 243         }
 244
 245
 246         session->session_info->nss_token |= username_was_mapped;
 247
 248         /* we need to build the token for the user. make_session_info_guest()
 249            already does this */
 250
 251         if (!session->session_info->security_token ) {
 252                 status = create_local_token(session->session_info);
 253                 if (!NT_STATUS_IS_OK(status)) {
 254                         DEBUG(10,("smb2: failed to create local token: %s\n",
 255                                 nt_errstr(status)));
 256                         goto fail;
 257                 }
 258         }
 259
 260         if ((in_security_mode & SMB2_NEGOTIATE_SIGNING_REQUIRED) ||
 261              lp_server_signing() == Required) {
 262                 session->do_signing = true;
 263         }
 264
 265         if (session->session_info->guest) {
 266                 /* we map anonymous to guest internally */
 267                 *out_session_flags |= SMB2_SESSION_FLAG_IS_GUEST;
 268                 *out_session_flags |= SMB2_SESSION_FLAG_IS_NULL;
 269                 /* force no signing */
 270                 session->do_signing = false;
 271         }
 272
 273         data_blob_free(&session->session_info->user_session_key);
 274         session->session_info->user_session_key =
 275                         data_blob_talloc(
 276                                 session->session_info,
 277                                 session_key.data,
 278                                 session_key.length);
 279         if (session_key.length > 0) {
 280                 if (session->session_info->user_session_key.data == NULL) {
 281                         status = NT_STATUS_NO_MEMORY;
 282                         goto fail;
 283                 }
 284         }
 285         session->session_key = session->session_info->user_session_key;
 286
 287         session->compat_vuser = talloc_zero(session, user_struct);
 288         if (session->compat_vuser == NULL) {
 289                 status = NT_STATUS_NO_MEMORY;
 290                 goto fail;
 291         }
 292         session->compat_vuser->auth_ntlmssp_state = NULL;
 293         session->compat_vuser->homes_snum = -1;
 294         session->compat_vuser->session_info = session->session_info;
 295         session->compat_vuser->session_keystr = NULL;
 296         session->compat_vuser->vuid = session->vuid;
 297         DLIST_ADD(session->sconn->smb1.sessions.validated_users, session->compat_vuser);
 298
 299         /* This is a potentially untrusted username */
 300         alpha_strcpy(tmp, user, ". _-$", sizeof(tmp));
 301         session->session_info->sanitized_username =
 302                                 talloc_strdup(session->session_info, tmp);
 303
 304         if (!session->session_info->guest) {
 305                 session->compat_vuser->homes_snum =
 306                         register_homes_share(session->session_info->unix_name);
 307         }
 308
 309         if (!session_claim(session->sconn, session->compat_vuser)) {
 310                 DEBUG(1, ("smb2: Failed to claim session "
 311                         "for vuid=%d\n",
 312                         session->compat_vuser->vuid));
 313                 goto fail;
 314         }
 315
 316         session->status = NT_STATUS_OK;
 317
 318         /*
 319          * we attach the session to the request
 320          * so that the response can be signed
 321          */
 322         smb2req->session = session;
 323         if (session->do_signing) {
 324                 smb2req->do_signing = true;
 325         }
 326
 327         global_client_caps |= (CAP_LEVEL_II_OPLOCKS|CAP_STATUS32);
 328         status = NT_STATUS_OK;
 329
 330         /* wrap that up in a nice GSS-API wrapping */
 331         ap_rep_wrapped = spnego_gen_krb5_wrap(talloc_tos(), ap_rep,
 332                                 TOK_ID_KRB_AP_REP);
 333
 334         secblob_out = spnego_gen_auth_response(
 335                                         talloc_tos(),
 336                                         &ap_rep_wrapped,
 337                                         status,
 338                                         mechOID);
 339
 340         *out_security_buffer = data_blob_talloc(smb2req,
 341                                                 secblob_out.data,
 342                                                 secblob_out.length);
 343         if (secblob_out.data && out_security_buffer->data == NULL) {
 344                 status = NT_STATUS_NO_MEMORY;
 345                 goto fail;
 346         }
 347
 348         data_blob_free(&ap_rep);
 349         data_blob_free(&ap_rep_wrapped);
 350         data_blob_free(&ticket);
 351         data_blob_free(&session_key);
 352         data_blob_free(&secblob_out);
 353
 354         *out_session_id = session->vuid;
 355
 356         return NT_STATUS_OK;
 357
 358   fail:
 359
 360         data_blob_free(&ap_rep);
 361         data_blob_free(&ap_rep_wrapped);
 362         data_blob_free(&ticket);
 363         data_blob_free(&session_key);
 364         data_blob_free(&secblob_out);
 365
 366         ap_rep_wrapped = data_blob_null;
 367         secblob_out = spnego_gen_auth_response(
 368                                         talloc_tos(),
 369                                         &ap_rep_wrapped,
 370                                         status,
 371                                         mechOID);
 372
 373         *out_security_buffer = data_blob_talloc(smb2req,
 374                                                 secblob_out.data,
 375                                                 secblob_out.length);
 376         data_blob_free(&secblob_out);
 377         return status;
 378 }
 379 #endif
 380
 381 static NTSTATUS smbd_smb2_spnego_negotiate(struct smbd_smb2_session *session,
 382                                         struct smbd_smb2_request *smb2req,
 383                                         uint8_t in_security_mode,
 384                                         DATA_BLOB in_security_buffer,
 385                                         uint16_t *out_session_flags,
 386                                         DATA_BLOB *out_security_buffer,
 387                                         uint64_t *out_session_id)
 388 {
 389         DATA_BLOB secblob_in = data_blob_null;
 390         DATA_BLOB chal_out = data_blob_null;
 391         char *kerb_mech = NULL;
 392         NTSTATUS status;
 393
 394         /* Ensure we have no old NTLM state around. */
 395         TALLOC_FREE(session->auth_ntlmssp_state);
 396
 397         status = parse_spnego_mechanisms(talloc_tos(), in_security_buffer,
 398                         &secblob_in, &kerb_mech);
 399         if (!NT_STATUS_IS_OK(status)) {
 400                 goto out;
 401         }
 402
 403 #ifdef HAVE_KRB5
 404         if (kerb_mech && ((lp_security()==SEC_ADS) ||
 405                                 USE_KERBEROS_KEYTAB) ) {
 406                 status = smbd_smb2_session_setup_krb5(session,
 407                                 smb2req,
 408                                 in_security_mode,
 409                                 &secblob_in,
 410                                 kerb_mech,
 411                                 out_session_flags,
 412                                 out_security_buffer,
 413                                 out_session_id);
 414
 415                 goto out;
 416         }
 417 #endif
 418
 419         if (kerb_mech) {
 420                 /* The mechtoken is a krb5 ticket, but
 421                  * we need to fall back to NTLM. */
 422
 423                 DEBUG(3,("smb2: Got krb5 ticket in SPNEGO "
 424                         "but set to downgrade to NTLMSSP\n"));
 425
 426                 status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 427         } else {
 428                 /* Fall back to NTLMSSP. */
 429                 status = auth_ntlmssp_start(&session->auth_ntlmssp_state);
 430                 if (!NT_STATUS_IS_OK(status)) {
 431                         goto out;
 432                 }
 433
 434                 status = auth_ntlmssp_update(session->auth_ntlmssp_state,
 435                                              secblob_in,
 436                                              &chal_out);
 437         }
 438
 439         if (!NT_STATUS_IS_OK(status) &&
 440                         !NT_STATUS_EQUAL(status,
 441                                 NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 442                 goto out;
 443         }
 444
 445         *out_security_buffer = spnego_gen_auth_response(smb2req,
 446                                                 &chal_out,
 447                                                 status,
 448                                                 OID_NTLMSSP);
 449         if (out_security_buffer->data == NULL) {
 450                 status = NT_STATUS_NO_MEMORY;
 451                 goto out;
 452         }
 453         *out_session_id = session->vuid;
 454
 455   out:
 456
 457         data_blob_free(&secblob_in);
 458         data_blob_free(&chal_out);
 459         TALLOC_FREE(kerb_mech);
 460         if (!NT_STATUS_IS_OK(status) &&
 461                         !NT_STATUS_EQUAL(status,
 462                                 NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 463                 TALLOC_FREE(session->auth_ntlmssp_state);
 464                 TALLOC_FREE(session);
 465         }
 466         return status;
 467 }
 468
 469 static NTSTATUS smbd_smb2_common_ntlmssp_auth_return(struct smbd_smb2_session *session,
 470                                         struct smbd_smb2_request *smb2req,
 471                                         uint8_t in_security_mode,
 472                                         DATA_BLOB in_security_buffer,
 473                                         uint16_t *out_session_flags,
 474                                         uint64_t *out_session_id)
 475 {
 476         fstring tmp;
 477
 478         if ((in_security_mode & SMB2_NEGOTIATE_SIGNING_REQUIRED) ||
 479             lp_server_signing() == Required) {
 480                 session->do_signing = true;
 481         }
 482
 483         if (session->session_info->guest) {
 484                 /* we map anonymous to guest internally */
 485                 *out_session_flags |= SMB2_SESSION_FLAG_IS_GUEST;
 486                 *out_session_flags |= SMB2_SESSION_FLAG_IS_NULL;
 487                 /* force no signing */
 488                 session->do_signing = false;
 489         }
 490
 491         session->session_key = session->session_info->user_session_key;
 492
 493         session->compat_vuser = talloc_zero(session, user_struct);
 494         if (session->compat_vuser == NULL) {
 495                 TALLOC_FREE(session->auth_ntlmssp_state);
 496                 TALLOC_FREE(session);
 497                 return NT_STATUS_NO_MEMORY;
 498         }
 499         session->compat_vuser->auth_ntlmssp_state = session->auth_ntlmssp_state;
 500         session->compat_vuser->homes_snum = -1;
 501         session->compat_vuser->session_info = session->session_info;
 502         session->compat_vuser->session_keystr = NULL;
 503         session->compat_vuser->vuid = session->vuid;
 504         DLIST_ADD(session->sconn->smb1.sessions.validated_users, session->compat_vuser);
 505
 506         /* This is a potentially untrusted username */
 507         alpha_strcpy(tmp,
 508                      auth_ntlmssp_get_username(session->auth_ntlmssp_state),
 509                      ". _-$",
 510                      sizeof(tmp));
 511         session->session_info->sanitized_username = talloc_strdup(
 512                 session->session_info, tmp);
 513
 514         if (!session->compat_vuser->session_info->guest) {
 515                 session->compat_vuser->homes_snum =
 516                         register_homes_share(session->session_info->unix_name);
 517         }
 518
 519         if (!session_claim(session->sconn, session->compat_vuser)) {
 520                 DEBUG(1, ("smb2: Failed to claim session "
 521                         "for vuid=%d\n",
 522                         session->compat_vuser->vuid));
 523                 TALLOC_FREE(session->auth_ntlmssp_state);
 524                 TALLOC_FREE(session);
 525                 return NT_STATUS_LOGON_FAILURE;
 526         }
 527
 528
 529         session->status = NT_STATUS_OK;
 530
 531         /*
 532          * we attach the session to the request
 533          * so that the response can be signed
 534          */
 535         smb2req->session = session;
 536         if (session->do_signing) {
 537                 smb2req->do_signing = true;
 538         }
 539
 540         global_client_caps |= (CAP_LEVEL_II_OPLOCKS|CAP_STATUS32);
 541
 542         *out_session_id = session->vuid;
 543
 544         return NT_STATUS_OK;
 545 }
 546
 547 static NTSTATUS smbd_smb2_spnego_auth(struct smbd_smb2_session *session,
 548                                         struct smbd_smb2_request *smb2req,
 549                                         uint8_t in_security_mode,
 550                                         DATA_BLOB in_security_buffer,
 551                                         uint16_t *out_session_flags,
 552                                         DATA_BLOB *out_security_buffer,
 553                                         uint64_t *out_session_id)
 554 {
 555         DATA_BLOB auth = data_blob_null;
 556         DATA_BLOB auth_out = data_blob_null;
 557         NTSTATUS status;
 558
 559         if (!spnego_parse_auth(talloc_tos(), in_security_buffer, &auth)) {
 560                 TALLOC_FREE(session);
 561                 return NT_STATUS_LOGON_FAILURE;
 562         }
 563
 564         if (auth.data[0] == ASN1_APPLICATION(0)) {
 565                 /* Might be a second negTokenTarg packet */
 566                 DATA_BLOB secblob_in = data_blob_null;
 567                 char *kerb_mech = NULL;
 568
 569                 status = parse_spnego_mechanisms(talloc_tos(),
 570                                 in_security_buffer,
 571                                 &secblob_in, &kerb_mech);
 572                 if (!NT_STATUS_IS_OK(status)) {
 573                         TALLOC_FREE(session);
 574                         return status;
 575                 }
 576
 577 #ifdef HAVE_KRB5
 578                 if (kerb_mech && ((lp_security()==SEC_ADS) ||
 579                                         USE_KERBEROS_KEYTAB) ) {
 580                         status = smbd_smb2_session_setup_krb5(session,
 581                                         smb2req,
 582                                         in_security_mode,
 583                                         &secblob_in,
 584                                         kerb_mech,
 585                                         out_session_flags,
 586                                         out_security_buffer,
 587                                         out_session_id);
 588
 589                         data_blob_free(&secblob_in);
 590                         TALLOC_FREE(kerb_mech);
 591                         if (!NT_STATUS_IS_OK(status)) {
 592                                 TALLOC_FREE(session);
 593                         }
 594                         return status;
 595                 }
 596 #endif
 597
 598                 /* Can't blunder into NTLMSSP auth if we have
 599                  * a krb5 ticket. */
 600
 601                 if (kerb_mech) {
 602                         DEBUG(3,("smb2: network "
 603                                 "misconfiguration, client sent us a "
 604                                 "krb5 ticket and kerberos security "
 605                                 "not enabled\n"));
 606                         TALLOC_FREE(session);
 607                         data_blob_free(&secblob_in);
 608                         TALLOC_FREE(kerb_mech);
 609                         return NT_STATUS_LOGON_FAILURE;
 610                 }
 611
 612                 data_blob_free(&secblob_in);
 613         }
 614
 615         if (session->auth_ntlmssp_state == NULL) {
 616                 status = auth_ntlmssp_start(&session->auth_ntlmssp_state);
 617                 if (!NT_STATUS_IS_OK(status)) {
 618                         data_blob_free(&auth);
 619                         TALLOC_FREE(session);
 620                         return status;
 621                 }
 622         }
 623
 624         status = auth_ntlmssp_update(session->auth_ntlmssp_state,
 625                                      auth,
 626                                      &auth_out);
 627         /* We need to call setup_ntlmssp_session_info() if status==NT_STATUS_OK,
 628            or if status is anything except NT_STATUS_MORE_PROCESSING_REQUIRED,
 629            as this can trigger map to guest. */
 630         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 631                 status = setup_ntlmssp_session_info(session, status);
 632         }
 633
 634         if (!NT_STATUS_IS_OK(status) &&
 635                         !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 636                 TALLOC_FREE(session->auth_ntlmssp_state);
 637                 data_blob_free(&auth);
 638                 TALLOC_FREE(session);
 639                 return status;
 640         }
 641
 642         data_blob_free(&auth);
 643
 644         *out_security_buffer = spnego_gen_auth_response(smb2req,
 645                                 &auth_out, status, NULL);
 646
 647         if (out_security_buffer->data == NULL) {
 648                 TALLOC_FREE(session->auth_ntlmssp_state);
 649                 TALLOC_FREE(session);
 650                 return NT_STATUS_NO_MEMORY;
 651         }
 652
 653         *out_session_id = session->vuid;
 654
 655         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 656                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
 657         }
 658
 659         /* We're done - claim the session. */
 660         return smbd_smb2_common_ntlmssp_auth_return(session,
 661                                                 smb2req,
 662                                                 in_security_mode,
 663                                                 in_security_buffer,
 664                                                 out_session_flags,
 665                                                 out_session_id);
 666 }
 667
 668 static NTSTATUS smbd_smb2_raw_ntlmssp_auth(struct smbd_smb2_session *session,
 669                                         struct smbd_smb2_request *smb2req,
 670                                         uint8_t in_security_mode,
 671                                         DATA_BLOB in_security_buffer,
 672                                         uint16_t *out_session_flags,
 673                                         DATA_BLOB *out_security_buffer,
 674                                         uint64_t *out_session_id)
 675 {
 676         NTSTATUS status;
 677         DATA_BLOB secblob_out = data_blob_null;
 678
 679         if (session->auth_ntlmssp_state == NULL) {
 680                 status = auth_ntlmssp_start(&session->auth_ntlmssp_state);
 681                 if (!NT_STATUS_IS_OK(status)) {
 682                         TALLOC_FREE(session);
 683                         return status;
 684                 }
 685         }
 686
 687         /* RAW NTLMSSP */
 688         status = auth_ntlmssp_update(session->auth_ntlmssp_state,
 689                                      in_security_buffer,
 690                                      &secblob_out);
 691
 692         if (NT_STATUS_IS_OK(status) ||
 693                         NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 694                 *out_security_buffer = data_blob_talloc(smb2req,
 695                                                 secblob_out.data,
 696                                                 secblob_out.length);
 697                 if (secblob_out.data && out_security_buffer->data == NULL) {
 698                         TALLOC_FREE(session->auth_ntlmssp_state);
 699                         TALLOC_FREE(session);
 700                         return NT_STATUS_NO_MEMORY;
 701                 }
 702         }
 703
 704         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 705                 *out_session_id = session->vuid;
 706                 return status;
 707         }
 708
 709         status = setup_ntlmssp_session_info(session, status);
 710
 711         if (!NT_STATUS_IS_OK(status)) {
 712                 TALLOC_FREE(session->auth_ntlmssp_state);
 713                 TALLOC_FREE(session);
 714                 return status;
 715         }
 716         *out_session_id = session->vuid;
 717
 718         return smbd_smb2_common_ntlmssp_auth_return(session,
 719                                                 smb2req,
 720                                                 in_security_mode,
 721                                                 in_security_buffer,
 722                                                 out_session_flags,
 723                                                 out_session_id);
 724 }
 725
 726 static NTSTATUS smbd_smb2_session_setup(struct smbd_smb2_request *smb2req,
 727                                         uint64_t in_session_id,
 728                                         uint8_t in_security_mode,
 729                                         DATA_BLOB in_security_buffer,
 730                                         uint16_t *out_session_flags,
 731                                         DATA_BLOB *out_security_buffer,
 732                                         uint64_t *out_session_id)
 733 {
 734         struct smbd_smb2_session *session;
 735
 736         *out_session_flags = 0;
 737         *out_session_id = 0;
 738
 739         if (in_session_id == 0) {
 740                 int id;
 741
 742                 /* create a new session */
 743                 session = talloc_zero(smb2req->sconn, struct smbd_smb2_session);
 744                 if (session == NULL) {
 745                         return NT_STATUS_NO_MEMORY;
 746                 }
 747                 session->status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 748                 id = idr_get_new_random(smb2req->sconn->smb2.sessions.idtree,
 749                                         session,
 750                                         smb2req->sconn->smb2.sessions.limit);
 751                 if (id == -1) {
 752                         return NT_STATUS_INSUFFICIENT_RESOURCES;
 753                 }
 754                 session->vuid = id;
 755
 756                 session->tcons.idtree = idr_init(session);
 757                 if (session->tcons.idtree == NULL) {
 758                         return NT_STATUS_NO_MEMORY;
 759                 }
 760                 session->tcons.limit = 0x0000FFFE;
 761                 session->tcons.list = NULL;
 762
 763                 DLIST_ADD_END(smb2req->sconn->smb2.sessions.list, session,
 764                               struct smbd_smb2_session *);
 765                 session->sconn = smb2req->sconn;
 766                 talloc_set_destructor(session, smbd_smb2_session_destructor);
 767         } else {
 768                 void *p;
 769
 770                 /* lookup an existing session */
 771                 p = idr_find(smb2req->sconn->smb2.sessions.idtree, in_session_id);
 772                 if (p == NULL) {
 773                         return NT_STATUS_USER_SESSION_DELETED;
 774                 }
 775                 session = talloc_get_type_abort(p, struct smbd_smb2_session);
 776         }
 777
 778         if (NT_STATUS_IS_OK(session->status)) {
 779                 return NT_STATUS_REQUEST_NOT_ACCEPTED;
 780         }
 781
 782         if (in_security_buffer.data[0] == ASN1_APPLICATION(0)) {
 783                 return smbd_smb2_spnego_negotiate(session,
 784                                                 smb2req,
 785                                                 in_security_mode,
 786                                                 in_security_buffer,
 787                                                 out_session_flags,
 788                                                 out_security_buffer,
 789                                                 out_session_id);
 790         } else if (in_security_buffer.data[0] == ASN1_CONTEXT(1)) {
 791                 return smbd_smb2_spnego_auth(session,
 792                                                 smb2req,
 793                                                 in_security_mode,
 794                                                 in_security_buffer,
 795                                                 out_session_flags,
 796                                                 out_security_buffer,
 797                                                 out_session_id);
 798         } else if (strncmp((char *)(in_security_buffer.data), "NTLMSSP", 7) == 0) {
 799                 return smbd_smb2_raw_ntlmssp_auth(session,
 800                                                 smb2req,
 801                                                 in_security_mode,
 802                                                 in_security_buffer,
 803                                                 out_session_flags,
 804                                                 out_security_buffer,
 805                                                 out_session_id);
 806         }
 807
 808         /* Unknown packet type. */
 809         DEBUG(1,("Unknown packet type %u in smb2 sessionsetup\n",
 810                 (unsigned int)in_security_buffer.data[0] ));
 811         TALLOC_FREE(session->auth_ntlmssp_state);
 812         TALLOC_FREE(session);
 813         return NT_STATUS_LOGON_FAILURE;
 814 }
 815
 816 NTSTATUS smbd_smb2_request_check_session(struct smbd_smb2_request *req)
 817 {
 818         const uint8_t *inhdr;
 819         const uint8_t *outhdr;
 820         int i = req->current_idx;
 821         uint64_t in_session_id;
 822         void *p;
 823         struct smbd_smb2_session *session;
 824         bool chained_fixup = false;
 825
 826         inhdr = (const uint8_t *)req->in.vector[i+0].iov_base;
 827
 828         in_session_id = BVAL(inhdr, SMB2_HDR_SESSION_ID);
 829
 830         if (in_session_id == (0xFFFFFFFFFFFFFFFFLL)) {
 831                 if (req->async) {
 832                         /*
 833                          * async request - fill in session_id from
 834                          * already setup request out.vector[].iov_base.
 835                          */
 836                         outhdr = (const uint8_t *)req->out.vector[i].iov_base;
 837                         in_session_id = BVAL(outhdr, SMB2_HDR_SESSION_ID);
 838                 } else if (i > 2) {
 839                         /*
 840                          * Chained request - fill in session_id from
 841                          * the previous request out.vector[].iov_base.
 842                          */
 843                         outhdr = (const uint8_t *)req->out.vector[i-3].iov_base;
 844                         in_session_id = BVAL(outhdr, SMB2_HDR_SESSION_ID);
 845                         chained_fixup = true;
 846                 }
 847         }
 848
 849         /* lookup an existing session */
 850         p = idr_find(req->sconn->smb2.sessions.idtree, in_session_id);
 851         if (p == NULL) {
 852                 return NT_STATUS_USER_SESSION_DELETED;
 853         }
 854         session = talloc_get_type_abort(p, struct smbd_smb2_session);
 855
 856         if (!NT_STATUS_IS_OK(session->status)) {
 857                 return NT_STATUS_ACCESS_DENIED;
 858         }
 859
 860         set_current_user_info(session->session_info->sanitized_username,
 861                               session->session_info->unix_name,
 862                               session->session_info->info3->base.domain.string);
 863
 864         req->session = session;
 865
 866         if (chained_fixup) {
 867                 /* Fix up our own outhdr. */
 868                 outhdr = (const uint8_t *)req->out.vector[i].iov_base;
 869                 SBVAL(outhdr, SMB2_HDR_SESSION_ID, in_session_id);
 870         }
 871         return NT_STATUS_OK;
 872 }
 873
 874 NTSTATUS smbd_smb2_request_process_logoff(struct smbd_smb2_request *req)
 875 {
 876         const uint8_t *inbody;
 877         int i = req->current_idx;
 878         DATA_BLOB outbody;
 879         size_t expected_body_size = 0x04;
 880         size_t body_size;
 881
 882         if (req->in.vector[i+1].iov_len != (expected_body_size & 0xFFFFFFFE)) {
 883                 return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
 884         }
 885
 886         inbody = (const uint8_t *)req->in.vector[i+1].iov_base;
 887
 888         body_size = SVAL(inbody, 0x00);
 889         if (body_size != expected_body_size) {
 890                 return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
 891         }
 892
 893         /*
 894          * TODO: cancel all outstanding requests on the session
 895          *       and delete all tree connections.
 896          */
 897         smbd_smb2_session_destructor(req->session);
 898         /*
 899          * we may need to sign the response, so we need to keep
 900          * the session until the response is sent to the wire.
 901          */
 902         talloc_steal(req, req->session);
 903
 904         outbody = data_blob_talloc(req->out.vector, NULL, 0x04);
 905         if (outbody.data == NULL) {
 906                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 907         }
 908
 909         SSVAL(outbody.data, 0x00, 0x04);        /* struct size */
 910         SSVAL(outbody.data, 0x02, 0);           /* reserved */
 911
 912         return smbd_smb2_request_done(req, outbody, NULL);
 913 }