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