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 "../auth/gensec/gensec.h"
  27 #include "auth.h"
  28 #include "../lib/tsocket/tsocket.h"
  29 #include "../libcli/security/security.h"
  30 #include "../lib/util/tevent_ntstatus.h"
  31
  32 static struct tevent_req *smbd_smb2_session_setup_send(TALLOC_CTX *mem_ctx,
  33                                         struct tevent_context *ev,
  34                                         struct smbd_smb2_request *smb2req,
  35                                         uint64_t in_session_id,
  36                                         uint8_t in_flags,
  37                                         uint8_t in_security_mode,
  38                                         uint64_t in_previous_session_id,
  39                                         DATA_BLOB in_security_buffer);
  40 static NTSTATUS smbd_smb2_session_setup_recv(struct tevent_req *req,
  41                                         uint16_t *out_session_flags,
  42                                         TALLOC_CTX *mem_ctx,
  43                                         DATA_BLOB *out_security_buffer,
  44                                         uint64_t *out_session_id);
  45
  46 static void smbd_smb2_request_sesssetup_done(struct tevent_req *subreq);
  47
  48 NTSTATUS smbd_smb2_request_process_sesssetup(struct smbd_smb2_request *smb2req)
  49 {
  50         const uint8_t *inhdr;
  51         const uint8_t *inbody;
  52         uint64_t in_session_id;
  53         uint8_t in_flags;
  54         uint8_t in_security_mode;
  55         uint64_t in_previous_session_id;
  56         uint16_t in_security_offset;
  57         uint16_t in_security_length;
  58         DATA_BLOB in_security_buffer;
  59         NTSTATUS status;
  60         struct tevent_req *subreq;
  61
  62         status = smbd_smb2_request_verify_sizes(smb2req, 0x19);
  63         if (!NT_STATUS_IS_OK(status)) {
  64                 return smbd_smb2_request_error(smb2req, status);
  65         }
  66         inhdr = SMBD_SMB2_IN_HDR_PTR(smb2req);
  67         inbody = SMBD_SMB2_IN_BODY_PTR(smb2req);
  68
  69         in_session_id = BVAL(inhdr, SMB2_HDR_SESSION_ID);
  70
  71         in_flags = CVAL(inbody, 0x02);
  72         in_security_mode = CVAL(inbody, 0x03);
  73         /* Capabilities = IVAL(inbody, 0x04) */
  74         /* Channel = IVAL(inbody, 0x08) */
  75         in_security_offset = SVAL(inbody, 0x0C);
  76         in_security_length = SVAL(inbody, 0x0E);
  77         in_previous_session_id = BVAL(inbody, 0x10);
  78
  79         if (in_security_offset != (SMB2_HDR_BODY + SMBD_SMB2_IN_BODY_LEN(smb2req))) {
  80                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  81         }
  82
  83         if (in_security_length > SMBD_SMB2_IN_DYN_LEN(smb2req)) {
  84                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  85         }
  86
  87         in_security_buffer.data = SMBD_SMB2_IN_DYN_PTR(smb2req);
  88         in_security_buffer.length = in_security_length;
  89
  90         subreq = smbd_smb2_session_setup_send(smb2req,
  91                                               smb2req->sconn->ev_ctx,
  92                                               smb2req,
  93                                               in_session_id,
  94                                               in_flags,
  95                                               in_security_mode,
  96                                               in_previous_session_id,
  97                                               in_security_buffer);
  98         if (subreq == NULL) {
  99                 return smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 100         }
 101         tevent_req_set_callback(subreq, smbd_smb2_request_sesssetup_done, smb2req);
 102
 103         return smbd_smb2_request_pending_queue(smb2req, subreq, 500);
 104 }
 105
 106 static void smbd_smb2_request_sesssetup_done(struct tevent_req *subreq)
 107 {
 108         struct smbd_smb2_request *smb2req =
 109                 tevent_req_callback_data(subreq,
 110                 struct smbd_smb2_request);
 111         uint8_t *outhdr;
 112         DATA_BLOB outbody;
 113         DATA_BLOB outdyn;
 114         uint16_t out_session_flags;
 115         uint64_t out_session_id;
 116         uint16_t out_security_offset;
 117         DATA_BLOB out_security_buffer = data_blob_null;
 118         NTSTATUS status;
 119         NTSTATUS error; /* transport error */
 120
 121         status = smbd_smb2_session_setup_recv(subreq,
 122                                               &out_session_flags,
 123                                               smb2req,
 124                                               &out_security_buffer,
 125                                               &out_session_id);
 126         TALLOC_FREE(subreq);
 127         if (!NT_STATUS_IS_OK(status) &&
 128             !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 129                 status = nt_status_squash(status);
 130                 error = smbd_smb2_request_error(smb2req, status);
 131                 if (!NT_STATUS_IS_OK(error)) {
 132                         smbd_server_connection_terminate(smb2req->sconn,
 133                                                          nt_errstr(error));
 134                         return;
 135                 }
 136                 return;
 137         }
 138
 139         out_security_offset = SMB2_HDR_BODY + 0x08;
 140
 141         outhdr = SMBD_SMB2_OUT_HDR_PTR(smb2req);
 142
 143         outbody = data_blob_talloc(smb2req->out.vector, NULL, 0x08);
 144         if (outbody.data == NULL) {
 145                 error = smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 146                 if (!NT_STATUS_IS_OK(error)) {
 147                         smbd_server_connection_terminate(smb2req->sconn,
 148                                                          nt_errstr(error));
 149                         return;
 150                 }
 151                 return;
 152         }
 153
 154         SBVAL(outhdr, SMB2_HDR_SESSION_ID, out_session_id);
 155
 156         SSVAL(outbody.data, 0x00, 0x08 + 1);    /* struct size */
 157         SSVAL(outbody.data, 0x02,
 158               out_session_flags);               /* session flags */
 159         SSVAL(outbody.data, 0x04,
 160               out_security_offset);             /* security buffer offset */
 161         SSVAL(outbody.data, 0x06,
 162               out_security_buffer.length);      /* security buffer length */
 163
 164         outdyn = out_security_buffer;
 165
 166         error = smbd_smb2_request_done_ex(smb2req, status, outbody, &outdyn,
 167                                            __location__);
 168         if (!NT_STATUS_IS_OK(error)) {
 169                 smbd_server_connection_terminate(smb2req->sconn,
 170                                                  nt_errstr(error));
 171                 return;
 172         }
 173 }
 174
 175 static NTSTATUS smbd_smb2_auth_generic_return(struct smbXsrv_session *session,
 176                                         struct smbd_smb2_request *smb2req,
 177                                         uint8_t in_security_mode,
 178                                         struct auth_session_info *session_info,
 179                                         uint16_t *out_session_flags,
 180                                         uint64_t *out_session_id)
 181 {
 182         NTSTATUS status;
 183         bool guest = false;
 184         uint8_t session_key[16];
 185         struct smbXsrv_session *x = session;
 186         struct smbXsrv_connection *conn = session->connection;
 187
 188         if ((in_security_mode & SMB2_NEGOTIATE_SIGNING_REQUIRED) ||
 189             lp_server_signing() == SMB_SIGNING_REQUIRED) {
 190                 x->global->signing_required = true;
 191         }
 192
 193         if (lp_smb_encrypt(-1) == SMB_SIGNING_REQUIRED) {
 194                 x->global->encryption_required = true;
 195         }
 196
 197         if (security_session_user_level(session_info, NULL) < SECURITY_USER) {
 198                 /* we map anonymous to guest internally */
 199                 *out_session_flags |= SMB2_SESSION_FLAG_IS_GUEST;
 200                 *out_session_flags |= SMB2_SESSION_FLAG_IS_NULL;
 201                 /* force no signing */
 202                 x->global->signing_required = false;
 203                 guest = true;
 204         }
 205
 206         if (guest && x->global->encryption_required) {
 207                 DEBUG(1,("reject guest session as encryption is required\n"));
 208                 return NT_STATUS_ACCESS_DENIED;
 209         }
 210
 211         if (!(conn->smb2.server.capabilities & SMB2_CAP_ENCRYPTION)) {
 212                 if (x->global->encryption_required) {
 213                         DEBUG(1,("reject session with dialect[0x%04X] "
 214                                  "as encryption is required\n",
 215                                  conn->smb2.server.dialect));
 216                         return NT_STATUS_ACCESS_DENIED;
 217                 }
 218         }
 219
 220         if (x->global->encryption_required) {
 221                 *out_session_flags |= SMB2_SESSION_FLAG_ENCRYPT_DATA;
 222         }
 223
 224         ZERO_STRUCT(session_key);
 225         memcpy(session_key, session_info->session_key.data,
 226                MIN(session_info->session_key.length, sizeof(session_key)));
 227
 228         x->global->signing_key = data_blob_talloc(x->global,
 229                                                   session_key,
 230                                                   sizeof(session_key));
 231         if (x->global->signing_key.data == NULL) {
 232                 ZERO_STRUCT(session_key);
 233                 return NT_STATUS_NO_MEMORY;
 234         }
 235
 236         if (conn->protocol >= PROTOCOL_SMB2_24) {
 237                 const DATA_BLOB label = data_blob_string_const_null("SMB2AESCMAC");
 238                 const DATA_BLOB context = data_blob_string_const_null("SmbSign");
 239
 240                 smb2_key_derivation(session_key, sizeof(session_key),
 241                                     label.data, label.length,
 242                                     context.data, context.length,
 243                                     x->global->signing_key.data);
 244         }
 245
 246         if (conn->protocol >= PROTOCOL_SMB2_24) {
 247                 const DATA_BLOB label = data_blob_string_const_null("SMB2AESCCM");
 248                 const DATA_BLOB context = data_blob_string_const_null("ServerIn ");
 249
 250                 x->global->decryption_key = data_blob_talloc(x->global,
 251                                                              session_key,
 252                                                              sizeof(session_key));
 253                 if (x->global->decryption_key.data == NULL) {
 254                         ZERO_STRUCT(session_key);
 255                         return NT_STATUS_NO_MEMORY;
 256                 }
 257
 258                 smb2_key_derivation(session_key, sizeof(session_key),
 259                                     label.data, label.length,
 260                                     context.data, context.length,
 261                                     x->global->decryption_key.data);
 262         }
 263
 264         if (conn->protocol >= PROTOCOL_SMB2_24) {
 265                 const DATA_BLOB label = data_blob_string_const_null("SMB2AESCCM");
 266                 const DATA_BLOB context = data_blob_string_const_null("ServerOut");
 267
 268                 x->global->encryption_key = data_blob_talloc(x->global,
 269                                                              session_key,
 270                                                              sizeof(session_key));
 271                 if (x->global->encryption_key.data == NULL) {
 272                         ZERO_STRUCT(session_key);
 273                         return NT_STATUS_NO_MEMORY;
 274                 }
 275
 276                 smb2_key_derivation(session_key, sizeof(session_key),
 277                                     label.data, label.length,
 278                                     context.data, context.length,
 279                                     x->global->encryption_key.data);
 280         }
 281
 282         x->global->application_key = data_blob_dup_talloc(x->global,
 283                                                 x->global->signing_key);
 284         if (x->global->application_key.data == NULL) {
 285                 ZERO_STRUCT(session_key);
 286                 return NT_STATUS_NO_MEMORY;
 287         }
 288
 289         if (conn->protocol >= PROTOCOL_SMB2_24) {
 290                 const DATA_BLOB label = data_blob_string_const_null("SMB2APP");
 291                 const DATA_BLOB context = data_blob_string_const_null("SmbRpc");
 292
 293                 smb2_key_derivation(session_key, sizeof(session_key),
 294                                     label.data, label.length,
 295                                     context.data, context.length,
 296                                     x->global->application_key.data);
 297         }
 298         ZERO_STRUCT(session_key);
 299
 300         x->global->channels[0].signing_key = data_blob_dup_talloc(x->global->channels,
 301                                                 x->global->signing_key);
 302         if (x->global->channels[0].signing_key.data == NULL) {
 303                 return NT_STATUS_NO_MEMORY;
 304         }
 305
 306         data_blob_clear_free(&session_info->session_key);
 307         session_info->session_key = data_blob_dup_talloc(session_info,
 308                                                 x->global->application_key);
 309         if (session_info->session_key.data == NULL) {
 310                 return NT_STATUS_NO_MEMORY;
 311         }
 312
 313         session->compat = talloc_zero(session, struct user_struct);
 314         if (session->compat == NULL) {
 315                 return NT_STATUS_NO_MEMORY;
 316         }
 317         session->compat->session = session;
 318         session->compat->homes_snum = -1;
 319         session->compat->session_info = session_info;
 320         session->compat->session_keystr = NULL;
 321         session->compat->vuid = session->global->session_wire_id;
 322         DLIST_ADD(smb2req->sconn->users, session->compat);
 323         smb2req->sconn->num_users++;
 324
 325         if (security_session_user_level(session_info, NULL) >= SECURITY_USER) {
 326                 session->compat->homes_snum =
 327                         register_homes_share(session_info->unix_info->unix_name);
 328         }
 329
 330         if (!session_claim(smb2req->sconn, session->compat)) {
 331                 DEBUG(1, ("smb2: Failed to claim session "
 332                         "for vuid=%llu\n",
 333                         (unsigned long long)session->compat->vuid));
 334                 return NT_STATUS_LOGON_FAILURE;
 335         }
 336
 337         set_current_user_info(session_info->unix_info->sanitized_username,
 338                               session_info->unix_info->unix_name,
 339                               session_info->info->domain_name);
 340
 341         reload_services(smb2req->sconn, conn_snum_used, true);
 342
 343         session->status = NT_STATUS_OK;
 344         session->global->auth_session_info = session_info;
 345         session->global->auth_session_info_seqnum += 1;
 346         session->global->channels[0].auth_session_info_seqnum =
 347                 session->global->auth_session_info_seqnum;
 348         session->global->expiration_time = gensec_expire_time(session->gensec);
 349
 350         status = smbXsrv_session_update(session);
 351         if (!NT_STATUS_IS_OK(status)) {
 352                 DEBUG(0, ("smb2: Failed to update session for vuid=%llu - %s\n",
 353                           (unsigned long long)session->compat->vuid,
 354                           nt_errstr(status)));
 355                 return NT_STATUS_LOGON_FAILURE;
 356         }
 357
 358         /*
 359          * we attach the session to the request
 360          * so that the response can be signed
 361          */
 362         smb2req->session = session;
 363         if (!guest) {
 364                 smb2req->do_signing = true;
 365         }
 366
 367         global_client_caps |= (CAP_LEVEL_II_OPLOCKS|CAP_STATUS32);
 368
 369         *out_session_id = session->global->session_wire_id;
 370
 371         return NT_STATUS_OK;
 372 }
 373
 374 static NTSTATUS smbd_smb2_reauth_generic_return(struct smbXsrv_session *session,
 375                                         struct smbd_smb2_request *smb2req,
 376                                         struct auth_session_info *session_info,
 377                                         uint16_t *out_session_flags,
 378                                         uint64_t *out_session_id)
 379 {
 380         NTSTATUS status;
 381         struct smbXsrv_session *x = session;
 382         struct smbXsrv_connection *conn = session->connection;
 383
 384         data_blob_clear_free(&session_info->session_key);
 385         session_info->session_key = data_blob_dup_talloc(session_info,
 386                                                 x->global->application_key);
 387         if (session_info->session_key.data == NULL) {
 388                 return NT_STATUS_NO_MEMORY;
 389         }
 390
 391         session->compat->session_info = session_info;
 392         session->compat->vuid = session->global->session_wire_id;
 393
 394         session->compat->homes_snum =
 395                         register_homes_share(session_info->unix_info->unix_name);
 396
 397         set_current_user_info(session_info->unix_info->sanitized_username,
 398                               session_info->unix_info->unix_name,
 399                               session_info->info->domain_name);
 400
 401         reload_services(smb2req->sconn, conn_snum_used, true);
 402
 403         session->status = NT_STATUS_OK;
 404         TALLOC_FREE(session->global->auth_session_info);
 405         session->global->auth_session_info = session_info;
 406         session->global->auth_session_info_seqnum += 1;
 407         session->global->channels[0].auth_session_info_seqnum =
 408                 session->global->auth_session_info_seqnum;
 409         session->global->expiration_time = gensec_expire_time(session->gensec);
 410
 411         status = smbXsrv_session_update(session);
 412         if (!NT_STATUS_IS_OK(status)) {
 413                 DEBUG(0, ("smb2: Failed to update session for vuid=%llu - %s\n",
 414                           (unsigned long long)session->compat->vuid,
 415                           nt_errstr(status)));
 416                 return NT_STATUS_LOGON_FAILURE;
 417         }
 418
 419         conn_clear_vuid_caches(conn->sconn, session->compat->vuid);
 420
 421         *out_session_id = session->global->session_wire_id;
 422
 423         return NT_STATUS_OK;
 424 }
 425
 426 struct smbd_smb2_session_setup_state {
 427         struct tevent_context *ev;
 428         struct smbd_smb2_request *smb2req;
 429         uint64_t in_session_id;
 430         uint8_t in_flags;
 431         uint8_t in_security_mode;
 432         uint64_t in_previous_session_id;
 433         DATA_BLOB in_security_buffer;
 434         struct smbXsrv_session *session;
 435         struct auth_session_info *session_info;
 436         uint16_t out_session_flags;
 437         DATA_BLOB out_security_buffer;
 438         uint64_t out_session_id;
 439 };
 440
 441 static int smbd_smb2_session_setup_state_destructor(struct smbd_smb2_session_setup_state *state)
 442 {
 443         /*
 444          * if state->session is not NULL,
 445          * we remove the session on failure
 446          */
 447         TALLOC_FREE(state->session);
 448         return 0;
 449 }
 450
 451 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq);
 452 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq);
 453
 454 static struct tevent_req *smbd_smb2_session_setup_send(TALLOC_CTX *mem_ctx,
 455                                         struct tevent_context *ev,
 456                                         struct smbd_smb2_request *smb2req,
 457                                         uint64_t in_session_id,
 458                                         uint8_t in_flags,
 459                                         uint8_t in_security_mode,
 460                                         uint64_t in_previous_session_id,
 461                                         DATA_BLOB in_security_buffer)
 462 {
 463         struct tevent_req *req;
 464         struct smbd_smb2_session_setup_state *state;
 465         NTSTATUS status;
 466         NTTIME now = timeval_to_nttime(&smb2req->request_time);
 467         struct tevent_req *subreq;
 468
 469         req = tevent_req_create(mem_ctx, &state,
 470                                 struct smbd_smb2_session_setup_state);
 471         if (req == NULL) {
 472                 return NULL;
 473         }
 474         state->ev = ev;
 475         state->smb2req = smb2req;
 476         state->in_session_id = in_session_id;
 477         state->in_flags = in_flags;
 478         state->in_security_mode = in_security_mode;
 479         state->in_previous_session_id = in_previous_session_id;
 480         state->in_security_buffer = in_security_buffer;
 481
 482         if (in_flags & SMB2_SESSION_FLAG_BINDING) {
 483                 if (smb2req->sconn->conn->protocol < PROTOCOL_SMB2_22) {
 484                         tevent_req_nterror(req, NT_STATUS_REQUEST_NOT_ACCEPTED);
 485                         return tevent_req_post(req, ev);
 486                 }
 487
 488                 /*
 489                  * We do not support multi channel.
 490                  */
 491                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
 492                 return tevent_req_post(req, ev);
 493         }
 494
 495         talloc_set_destructor(state, smbd_smb2_session_setup_state_destructor);
 496
 497         if (state->in_session_id == 0) {
 498                 /* create a new session */
 499                 status = smbXsrv_session_create(state->smb2req->sconn->conn,
 500                                                 now, &state->session);
 501                 if (tevent_req_nterror(req, status)) {
 502                         return tevent_req_post(req, ev);
 503                 }
 504         } else {
 505                 status = smb2srv_session_lookup(state->smb2req->sconn->conn,
 506                                                 state->in_session_id, now,
 507                                                 &state->session);
 508                 if (NT_STATUS_EQUAL(status, NT_STATUS_NETWORK_SESSION_EXPIRED)) {
 509                         status = NT_STATUS_OK;
 510                 }
 511                 if (NT_STATUS_IS_OK(status)) {
 512                         state->session->status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 513                         status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 514                         TALLOC_FREE(state->session->gensec);
 515                 }
 516                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 517                         tevent_req_nterror(req, status);
 518                         return tevent_req_post(req, ev);
 519                 }
 520         }
 521
 522         if (state->session->gensec == NULL) {
 523                 status = auth_generic_prepare(state->session,
 524                                               state->session->connection->remote_address,
 525                                               &state->session->gensec);
 526                 if (tevent_req_nterror(req, status)) {
 527                         return tevent_req_post(req, ev);
 528                 }
 529
 530                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_SESSION_KEY);
 531                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_UNIX_TOKEN);
 532
 533                 status = gensec_start_mech_by_oid(state->session->gensec,
 534                                                   GENSEC_OID_SPNEGO);
 535                 if (tevent_req_nterror(req, status)) {
 536                         return tevent_req_post(req, ev);
 537                 }
 538         }
 539
 540         become_root();
 541         subreq = gensec_update_send(state, state->ev,
 542                                     state->session->gensec,
 543                                     state->in_security_buffer);
 544         unbecome_root();
 545         if (tevent_req_nomem(subreq, req)) {
 546                 return tevent_req_post(req, ev);
 547         }
 548         tevent_req_set_callback(subreq, smbd_smb2_session_setup_gensec_done, req);
 549
 550         return req;
 551 }
 552
 553 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq)
 554 {
 555         struct tevent_req *req =
 556                 tevent_req_callback_data(subreq,
 557                 struct tevent_req);
 558         struct smbd_smb2_session_setup_state *state =
 559                 tevent_req_data(req,
 560                 struct smbd_smb2_session_setup_state);
 561         NTSTATUS status;
 562
 563         become_root();
 564         status = gensec_update_recv(subreq, state,
 565                                     &state->out_security_buffer);
 566         unbecome_root();
 567         TALLOC_FREE(subreq);
 568         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) &&
 569             !NT_STATUS_IS_OK(status)) {
 570                 tevent_req_nterror(req, status);
 571                 return;
 572         }
 573
 574         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 575                 state->out_session_id = state->session->global->session_wire_id;
 576                 /* we want to keep the session */
 577                 state->session = NULL;
 578                 tevent_req_nterror(req, status);
 579                 return;
 580         }
 581
 582         status = gensec_session_info(state->session->gensec,
 583                                      state->session->global,
 584                                      &state->session_info);
 585         if (tevent_req_nterror(req, status)) {
 586                 return;
 587         }
 588
 589         if ((state->in_previous_session_id != 0) &&
 590              (state->session->global->session_wire_id !=
 591               state->in_previous_session_id))
 592         {
 593                 subreq = smb2srv_session_close_previous_send(state, state->ev,
 594                                                 state->session->connection,
 595                                                 state->session_info,
 596                                                 state->in_previous_session_id,
 597                                                 state->session->global->session_wire_id);
 598                 if (tevent_req_nomem(subreq, req)) {
 599                         return;
 600                 }
 601                 tevent_req_set_callback(subreq,
 602                                         smbd_smb2_session_setup_previous_done,
 603                                         req);
 604                 return;
 605         }
 606
 607         if (state->session->global->auth_session_info != NULL) {
 608                 status = smbd_smb2_reauth_generic_return(state->session,
 609                                                          state->smb2req,
 610                                                          state->session_info,
 611                                                          &state->out_session_flags,
 612                                                          &state->out_session_id);
 613                 if (tevent_req_nterror(req, status)) {
 614                         return;
 615                 }
 616                 /* we want to keep the session */
 617                 state->session = NULL;
 618                 tevent_req_done(req);
 619                 return;
 620         }
 621
 622         status = smbd_smb2_auth_generic_return(state->session,
 623                                                state->smb2req,
 624                                                state->in_security_mode,
 625                                                state->session_info,
 626                                                &state->out_session_flags,
 627                                                &state->out_session_id);
 628         if (tevent_req_nterror(req, status)) {
 629                 return;
 630         }
 631
 632         /* we want to keep the session */
 633         state->session = NULL;
 634         tevent_req_done(req);
 635         return;
 636 }
 637
 638 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq)
 639 {
 640         struct tevent_req *req =
 641                 tevent_req_callback_data(subreq,
 642                 struct tevent_req);
 643         struct smbd_smb2_session_setup_state *state =
 644                 tevent_req_data(req,
 645                 struct smbd_smb2_session_setup_state);
 646         NTSTATUS status;
 647
 648         status = smb2srv_session_close_previous_recv(subreq);
 649         TALLOC_FREE(subreq);
 650         if (tevent_req_nterror(req, status)) {
 651                 return;
 652         }
 653
 654         if (state->session->global->auth_session_info != NULL) {
 655                 status = smbd_smb2_reauth_generic_return(state->session,
 656                                                          state->smb2req,
 657                                                          state->session_info,
 658                                                          &state->out_session_flags,
 659                                                          &state->out_session_id);
 660                 if (tevent_req_nterror(req, status)) {
 661                         return;
 662                 }
 663                 /* we want to keep the session */
 664                 state->session = NULL;
 665                 tevent_req_done(req);
 666                 return;
 667         }
 668
 669         status = smbd_smb2_auth_generic_return(state->session,
 670                                                state->smb2req,
 671                                                state->in_security_mode,
 672                                                state->session_info,
 673                                                &state->out_session_flags,
 674                                                &state->out_session_id);
 675         if (tevent_req_nterror(req, status)) {
 676                 return;
 677         }
 678
 679         /* we want to keep the session */
 680         state->session = NULL;
 681         tevent_req_done(req);
 682         return;
 683 }
 684
 685 static NTSTATUS smbd_smb2_session_setup_recv(struct tevent_req *req,
 686                                         uint16_t *out_session_flags,
 687                                         TALLOC_CTX *mem_ctx,
 688                                         DATA_BLOB *out_security_buffer,
 689                                         uint64_t *out_session_id)
 690 {
 691         struct smbd_smb2_session_setup_state *state =
 692                 tevent_req_data(req,
 693                 struct smbd_smb2_session_setup_state);
 694         NTSTATUS status;
 695
 696         if (tevent_req_is_nterror(req, &status)) {
 697                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 698                         tevent_req_received(req);
 699                         return nt_status_squash(status);
 700                 }
 701         } else {
 702                 status = NT_STATUS_OK;
 703         }
 704
 705         *out_session_flags = state->out_session_flags;
 706         *out_security_buffer = state->out_security_buffer;
 707         *out_session_id = state->out_session_id;
 708
 709         talloc_steal(mem_ctx, out_security_buffer->data);
 710         tevent_req_received(req);
 711         return status;
 712 }
 713
 714 NTSTATUS smbd_smb2_request_process_logoff(struct smbd_smb2_request *req)
 715 {
 716         NTSTATUS status;
 717         DATA_BLOB outbody;
 718
 719         status = smbd_smb2_request_verify_sizes(req, 0x04);
 720         if (!NT_STATUS_IS_OK(status)) {
 721                 return smbd_smb2_request_error(req, status);
 722         }
 723
 724         /*
 725          * TODO: cancel all outstanding requests on the session
 726          */
 727         status = smbXsrv_session_logoff(req->session);
 728         if (!NT_STATUS_IS_OK(status)) {
 729                 DEBUG(0, ("smbd_smb2_request_process_logoff: "
 730                           "smbXsrv_session_logoff() failed: %s\n",
 731                           nt_errstr(status)));
 732                 /*
 733                  * If we hit this case, there is something completely
 734                  * wrong, so we better disconnect the transport connection.
 735                  */
 736                 return status;
 737         }
 738
 739         /*
 740          * we may need to sign the response, so we need to keep
 741          * the session until the response is sent to the wire.
 742          */
 743         talloc_steal(req, req->session);
 744
 745         outbody = data_blob_talloc(req->out.vector, NULL, 0x04);
 746         if (outbody.data == NULL) {
 747                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 748         }
 749
 750         SSVAL(outbody.data, 0x00, 0x04);        /* struct size */
 751         SSVAL(outbody.data, 0x02, 0);           /* reserved */
 752
 753         return smbd_smb2_request_done(req, outbody, NULL);
 754 }