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