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