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 = 0;
 115         uint64_t out_session_id = 0;
 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->xconn,
 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 = smbd_smb2_generate_outbody(smb2req, 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->xconn,
 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->xconn,
 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 *xconn = smb2req->xconn;
 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 (xconn->smb2.server.cipher == 0) {
 212                 if (x->global->encryption_required) {
 213                         DEBUG(1,("reject session with dialect[0x%04X] "
 214                                  "as encryption is required\n",
 215                                  xconn->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 (xconn->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 (xconn->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 (xconn->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                 generate_random_buffer((uint8_t *)&x->nonce_high, sizeof(x->nonce_high));
 282                 x->nonce_low = 1;
 283         }
 284
 285         x->global->application_key = data_blob_dup_talloc(x->global,
 286                                                 x->global->signing_key);
 287         if (x->global->application_key.data == NULL) {
 288                 ZERO_STRUCT(session_key);
 289                 return NT_STATUS_NO_MEMORY;
 290         }
 291
 292         if (xconn->protocol >= PROTOCOL_SMB2_24) {
 293                 const DATA_BLOB label = data_blob_string_const_null("SMB2APP");
 294                 const DATA_BLOB context = data_blob_string_const_null("SmbRpc");
 295
 296                 smb2_key_derivation(session_key, sizeof(session_key),
 297                                     label.data, label.length,
 298                                     context.data, context.length,
 299                                     x->global->application_key.data);
 300         }
 301         ZERO_STRUCT(session_key);
 302
 303         x->global->channels[0].signing_key = data_blob_dup_talloc(x->global->channels,
 304                                                 x->global->signing_key);
 305         if (x->global->channels[0].signing_key.data == NULL) {
 306                 return NT_STATUS_NO_MEMORY;
 307         }
 308
 309         data_blob_clear_free(&session_info->session_key);
 310         session_info->session_key = data_blob_dup_talloc(session_info,
 311                                                 x->global->application_key);
 312         if (session_info->session_key.data == NULL) {
 313                 return NT_STATUS_NO_MEMORY;
 314         }
 315
 316         session->compat = talloc_zero(session, struct user_struct);
 317         if (session->compat == NULL) {
 318                 return NT_STATUS_NO_MEMORY;
 319         }
 320         session->compat->session = session;
 321         session->compat->homes_snum = -1;
 322         session->compat->session_info = session_info;
 323         session->compat->session_keystr = NULL;
 324         session->compat->vuid = session->global->session_wire_id;
 325         DLIST_ADD(smb2req->sconn->users, session->compat);
 326         smb2req->sconn->num_users++;
 327
 328         if (security_session_user_level(session_info, NULL) >= SECURITY_USER) {
 329                 session->compat->homes_snum =
 330                         register_homes_share(session_info->unix_info->unix_name);
 331         }
 332
 333         set_current_user_info(session_info->unix_info->sanitized_username,
 334                               session_info->unix_info->unix_name,
 335                               session_info->info->domain_name);
 336
 337         reload_services(smb2req->sconn, conn_snum_used, true);
 338
 339         session->status = NT_STATUS_OK;
 340         session->global->auth_session_info = session_info;
 341         session->global->auth_session_info_seqnum += 1;
 342         session->global->channels[0].auth_session_info_seqnum =
 343                 session->global->auth_session_info_seqnum;
 344         session->global->auth_time = timeval_to_nttime(&smb2req->request_time);
 345         session->global->expiration_time = gensec_expire_time(session->gensec);
 346
 347         if (!session_claim(session)) {
 348                 DEBUG(1, ("smb2: Failed to claim session "
 349                         "for vuid=%llu\n",
 350                         (unsigned long long)session->compat->vuid));
 351                 return NT_STATUS_LOGON_FAILURE;
 352         }
 353
 354         status = smbXsrv_session_update(session);
 355         if (!NT_STATUS_IS_OK(status)) {
 356                 DEBUG(0, ("smb2: Failed to update session for vuid=%llu - %s\n",
 357                           (unsigned long long)session->compat->vuid,
 358                           nt_errstr(status)));
 359                 return NT_STATUS_LOGON_FAILURE;
 360         }
 361
 362         /*
 363          * we attach the session to the request
 364          * so that the response can be signed
 365          */
 366         smb2req->session = session;
 367         if (!guest) {
 368                 smb2req->do_signing = true;
 369         }
 370
 371         global_client_caps |= (CAP_LEVEL_II_OPLOCKS|CAP_STATUS32);
 372
 373         *out_session_id = session->global->session_wire_id;
 374
 375         return NT_STATUS_OK;
 376 }
 377
 378 static NTSTATUS smbd_smb2_reauth_generic_return(struct smbXsrv_session *session,
 379                                         struct smbd_smb2_request *smb2req,
 380                                         struct auth_session_info *session_info,
 381                                         uint16_t *out_session_flags,
 382                                         uint64_t *out_session_id)
 383 {
 384         NTSTATUS status;
 385         struct smbXsrv_session *x = session;
 386
 387         data_blob_clear_free(&session_info->session_key);
 388         session_info->session_key = data_blob_dup_talloc(session_info,
 389                                                 x->global->application_key);
 390         if (session_info->session_key.data == NULL) {
 391                 return NT_STATUS_NO_MEMORY;
 392         }
 393
 394         session->compat->session_info = session_info;
 395         session->compat->vuid = session->global->session_wire_id;
 396
 397         session->compat->homes_snum =
 398                         register_homes_share(session_info->unix_info->unix_name);
 399
 400         set_current_user_info(session_info->unix_info->sanitized_username,
 401                               session_info->unix_info->unix_name,
 402                               session_info->info->domain_name);
 403
 404         reload_services(smb2req->sconn, conn_snum_used, true);
 405
 406         session->status = NT_STATUS_OK;
 407         TALLOC_FREE(session->global->auth_session_info);
 408         session->global->auth_session_info = session_info;
 409         session->global->auth_session_info_seqnum += 1;
 410         session->global->channels[0].auth_session_info_seqnum =
 411                 session->global->auth_session_info_seqnum;
 412         session->global->auth_time = timeval_to_nttime(&smb2req->request_time);
 413         session->global->expiration_time = gensec_expire_time(session->gensec);
 414
 415         status = smbXsrv_session_update(session);
 416         if (!NT_STATUS_IS_OK(status)) {
 417                 DEBUG(0, ("smb2: Failed to update session for vuid=%llu - %s\n",
 418                           (unsigned long long)session->compat->vuid,
 419                           nt_errstr(status)));
 420                 return NT_STATUS_LOGON_FAILURE;
 421         }
 422
 423         conn_clear_vuid_caches(smb2req->sconn, session->compat->vuid);
 424
 425         if (security_session_user_level(session_info, NULL) >= SECURITY_USER) {
 426                 smb2req->do_signing = true;
 427         }
 428
 429         *out_session_id = session->global->session_wire_id;
 430
 431         return NT_STATUS_OK;
 432 }
 433
 434 struct smbd_smb2_session_setup_state {
 435         struct tevent_context *ev;
 436         struct smbd_smb2_request *smb2req;
 437         uint64_t in_session_id;
 438         uint8_t in_flags;
 439         uint8_t in_security_mode;
 440         uint64_t in_previous_session_id;
 441         DATA_BLOB in_security_buffer;
 442         struct smbXsrv_session *session;
 443         struct auth_session_info *session_info;
 444         uint16_t out_session_flags;
 445         DATA_BLOB out_security_buffer;
 446         uint64_t out_session_id;
 447         /* The following pointer is owned by state->session. */
 448         struct smbd_smb2_session_setup_state **pp_self_ref;
 449 };
 450
 451 static int pp_self_ref_destructor(struct smbd_smb2_session_setup_state **pp_state)
 452 {
 453         (*pp_state)->session = NULL;
 454         /*
 455          * To make things clearer, ensure the pp_self_ref
 456          * pointer is nulled out. We're never going to
 457          * access this again.
 458          */
 459         (*pp_state)->pp_self_ref = NULL;
 460         return 0;
 461 }
 462
 463 static int smbd_smb2_session_setup_state_destructor(struct smbd_smb2_session_setup_state *state)
 464 {
 465         struct smbXsrv_connection *xconn;
 466         struct smbd_smb2_request *preq;
 467
 468         /*
 469          * If state->session is not NULL,
 470          * we move the session from the session table to the request on failure
 471          * so that the error response can be correctly signed, but the session
 472          * is then really deleted when the request is done.
 473          */
 474
 475         if (state->session == NULL) {
 476                 return 0;
 477         }
 478
 479         state->session->status = NT_STATUS_USER_SESSION_DELETED;
 480         state->smb2req->session = talloc_move(state->smb2req, &state->session);
 481
 482         /*
 483          * We own the session now - we don't need the
 484          * tag talloced on session that keeps track of session independently.
 485          */
 486         TALLOC_FREE(state->pp_self_ref);
 487
 488         /*
 489          * We've made this session owned by the current request.
 490          * Ensure that any outstanding requests don't also refer
 491          * to it.
 492          */
 493         xconn = state->smb2req->xconn;
 494
 495         for (preq = xconn->smb2.requests; preq != NULL; preq = preq->next) {
 496                 if (preq == state->smb2req) {
 497                         continue;
 498                 }
 499                 if (preq->session == state->smb2req->session) {
 500                         preq->session = NULL;
 501                         /*
 502                          * If we no longer have a session we can't
 503                          * sign or encrypt replies.
 504                          */
 505                         preq->do_signing = false;
 506                         preq->do_encryption = false;
 507                 }
 508         }
 509
 510         return 0;
 511 }
 512
 513 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq);
 514 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq);
 515 static void smbd_smb2_session_setup_auth_return(struct tevent_req *req);
 516
 517 /************************************************************************
 518  We have to tag the state->session pointer with memory talloc'ed
 519  on it to ensure it gets NULL'ed out if the underlying struct smbXsrv_session
 520  is deleted by shutdown whilst this request is in flight.
 521 ************************************************************************/
 522
 523 static NTSTATUS tag_state_session_ptr(struct smbd_smb2_session_setup_state *state)
 524 {
 525         state->pp_self_ref = talloc_zero(state->session,
 526                         struct smbd_smb2_session_setup_state *);
 527         if (state->pp_self_ref == NULL) {
 528                 return NT_STATUS_NO_MEMORY;
 529         }
 530         *state->pp_self_ref = state;
 531         talloc_set_destructor(state->pp_self_ref, pp_self_ref_destructor);
 532         return NT_STATUS_OK;
 533 }
 534
 535 static struct tevent_req *smbd_smb2_session_setup_send(TALLOC_CTX *mem_ctx,
 536                                         struct tevent_context *ev,
 537                                         struct smbd_smb2_request *smb2req,
 538                                         uint64_t in_session_id,
 539                                         uint8_t in_flags,
 540                                         uint8_t in_security_mode,
 541                                         uint64_t in_previous_session_id,
 542                                         DATA_BLOB in_security_buffer)
 543 {
 544         struct tevent_req *req;
 545         struct smbd_smb2_session_setup_state *state;
 546         NTSTATUS status;
 547         NTTIME now = timeval_to_nttime(&smb2req->request_time);
 548         struct tevent_req *subreq;
 549
 550         req = tevent_req_create(mem_ctx, &state,
 551                                 struct smbd_smb2_session_setup_state);
 552         if (req == NULL) {
 553                 return NULL;
 554         }
 555         state->ev = ev;
 556         state->smb2req = smb2req;
 557         state->in_session_id = in_session_id;
 558         state->in_flags = in_flags;
 559         state->in_security_mode = in_security_mode;
 560         state->in_previous_session_id = in_previous_session_id;
 561         state->in_security_buffer = in_security_buffer;
 562
 563         if (in_flags & SMB2_SESSION_FLAG_BINDING) {
 564                 if (smb2req->xconn->protocol < PROTOCOL_SMB2_22) {
 565                         tevent_req_nterror(req, NT_STATUS_REQUEST_NOT_ACCEPTED);
 566                         return tevent_req_post(req, ev);
 567                 }
 568
 569                 /*
 570                  * We do not support multi channel.
 571                  */
 572                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
 573                 return tevent_req_post(req, ev);
 574         }
 575
 576         talloc_set_destructor(state, smbd_smb2_session_setup_state_destructor);
 577
 578         if (state->in_session_id == 0) {
 579                 /* create a new session */
 580                 status = smbXsrv_session_create(state->smb2req->xconn,
 581                                                 now, &state->session);
 582                 if (tevent_req_nterror(req, status)) {
 583                         return tevent_req_post(req, ev);
 584                 }
 585         } else {
 586                 if (smb2req->session == NULL) {
 587                         tevent_req_nterror(req, NT_STATUS_USER_SESSION_DELETED);
 588                         return tevent_req_post(req, ev);
 589                 }
 590
 591                 state->session = smb2req->session;
 592                 status = state->session->status;
 593                 if (NT_STATUS_EQUAL(status, NT_STATUS_NETWORK_SESSION_EXPIRED)) {
 594                         status = NT_STATUS_OK;
 595                 }
 596                 if (NT_STATUS_IS_OK(status)) {
 597                         state->session->status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 598                         status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 599                         TALLOC_FREE(state->session->gensec);
 600                 }
 601                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 602                         tevent_req_nterror(req, status);
 603                         return tevent_req_post(req, ev);
 604                 }
 605         }
 606
 607         status = tag_state_session_ptr(state);
 608         if (tevent_req_nterror(req, status)) {
 609                 return tevent_req_post(req, ev);
 610         }
 611
 612         if (state->session->gensec == NULL) {
 613                 status = auth_generic_prepare(state->session,
 614                                               state->smb2req->xconn->remote_address,
 615                                               &state->session->gensec);
 616                 if (tevent_req_nterror(req, status)) {
 617                         return tevent_req_post(req, ev);
 618                 }
 619
 620                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_SESSION_KEY);
 621                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_UNIX_TOKEN);
 622
 623                 status = gensec_start_mech_by_oid(state->session->gensec,
 624                                                   GENSEC_OID_SPNEGO);
 625                 if (tevent_req_nterror(req, status)) {
 626                         return tevent_req_post(req, ev);
 627                 }
 628         }
 629
 630         become_root();
 631         subreq = gensec_update_send(state, state->ev,
 632                                     state->session->gensec,
 633                                     state->in_security_buffer);
 634         unbecome_root();
 635         if (tevent_req_nomem(subreq, req)) {
 636                 return tevent_req_post(req, ev);
 637         }
 638         tevent_req_set_callback(subreq, smbd_smb2_session_setup_gensec_done, req);
 639
 640         return req;
 641 }
 642
 643 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq)
 644 {
 645         struct tevent_req *req =
 646                 tevent_req_callback_data(subreq,
 647                 struct tevent_req);
 648         struct smbd_smb2_session_setup_state *state =
 649                 tevent_req_data(req,
 650                 struct smbd_smb2_session_setup_state);
 651         NTSTATUS status;
 652
 653         become_root();
 654         status = gensec_update_recv(subreq, state,
 655                                     &state->out_security_buffer);
 656         unbecome_root();
 657         TALLOC_FREE(subreq);
 658         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) &&
 659             !NT_STATUS_IS_OK(status)) {
 660                 tevent_req_nterror(req, status);
 661                 return;
 662         }
 663
 664         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 665                 state->out_session_id = state->session->global->session_wire_id;
 666                 /* we want to keep the session */
 667                 state->session = NULL;
 668                 TALLOC_FREE(state->pp_self_ref);
 669                 tevent_req_nterror(req, status);
 670                 return;
 671         }
 672
 673         status = gensec_session_info(state->session->gensec,
 674                                      state->session->global,
 675                                      &state->session_info);
 676         if (tevent_req_nterror(req, status)) {
 677                 return;
 678         }
 679
 680         if ((state->in_previous_session_id != 0) &&
 681              (state->session->global->session_wire_id !=
 682               state->in_previous_session_id))
 683         {
 684                 subreq = smb2srv_session_close_previous_send(state, state->ev,
 685                                                 state->smb2req->xconn,
 686                                                 state->session_info,
 687                                                 state->in_previous_session_id,
 688                                                 state->session->global->session_wire_id);
 689                 if (tevent_req_nomem(subreq, req)) {
 690                         return;
 691                 }
 692                 tevent_req_set_callback(subreq,
 693                                         smbd_smb2_session_setup_previous_done,
 694                                         req);
 695                 return;
 696         }
 697
 698         smbd_smb2_session_setup_auth_return(req);
 699 }
 700
 701 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq)
 702 {
 703         struct tevent_req *req =
 704                 tevent_req_callback_data(subreq,
 705                 struct tevent_req);
 706         NTSTATUS status;
 707
 708         status = smb2srv_session_close_previous_recv(subreq);
 709         TALLOC_FREE(subreq);
 710         if (tevent_req_nterror(req, status)) {
 711                 return;
 712         }
 713
 714         smbd_smb2_session_setup_auth_return(req);
 715 }
 716
 717 static void smbd_smb2_session_setup_auth_return(struct tevent_req *req)
 718 {
 719         struct smbd_smb2_session_setup_state *state =
 720                 tevent_req_data(req,
 721                 struct smbd_smb2_session_setup_state);
 722         NTSTATUS status;
 723
 724         if (state->session->global->auth_session_info != NULL) {
 725                 status = smbd_smb2_reauth_generic_return(state->session,
 726                                                          state->smb2req,
 727                                                          state->session_info,
 728                                                          &state->out_session_flags,
 729                                                          &state->out_session_id);
 730                 if (tevent_req_nterror(req, status)) {
 731                         return;
 732                 }
 733                 /* we want to keep the session */
 734                 state->session = NULL;
 735                 TALLOC_FREE(state->pp_self_ref);
 736                 tevent_req_done(req);
 737                 return;
 738         }
 739
 740         status = smbd_smb2_auth_generic_return(state->session,
 741                                                state->smb2req,
 742                                                state->in_security_mode,
 743                                                state->session_info,
 744                                                &state->out_session_flags,
 745                                                &state->out_session_id);
 746         if (tevent_req_nterror(req, status)) {
 747                 return;
 748         }
 749
 750         /* we want to keep the session */
 751         state->session = NULL;
 752         TALLOC_FREE(state->pp_self_ref);
 753         tevent_req_done(req);
 754         return;
 755 }
 756
 757 static NTSTATUS smbd_smb2_session_setup_recv(struct tevent_req *req,
 758                                         uint16_t *out_session_flags,
 759                                         TALLOC_CTX *mem_ctx,
 760                                         DATA_BLOB *out_security_buffer,
 761                                         uint64_t *out_session_id)
 762 {
 763         struct smbd_smb2_session_setup_state *state =
 764                 tevent_req_data(req,
 765                 struct smbd_smb2_session_setup_state);
 766         NTSTATUS status;
 767
 768         if (tevent_req_is_nterror(req, &status)) {
 769                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 770                         tevent_req_received(req);
 771                         return nt_status_squash(status);
 772                 }
 773         } else {
 774                 status = NT_STATUS_OK;
 775         }
 776
 777         *out_session_flags = state->out_session_flags;
 778         *out_security_buffer = state->out_security_buffer;
 779         *out_session_id = state->out_session_id;
 780
 781         talloc_steal(mem_ctx, out_security_buffer->data);
 782         tevent_req_received(req);
 783         return status;
 784 }
 785
 786 static struct tevent_req *smbd_smb2_logoff_send(TALLOC_CTX *mem_ctx,
 787                                         struct tevent_context *ev,
 788                                         struct smbd_smb2_request *smb2req);
 789 static NTSTATUS smbd_smb2_logoff_recv(struct tevent_req *req);
 790 static void smbd_smb2_request_logoff_done(struct tevent_req *subreq);
 791
 792 NTSTATUS smbd_smb2_request_process_logoff(struct smbd_smb2_request *req)
 793 {
 794         NTSTATUS status;
 795         struct tevent_req *subreq = NULL;
 796
 797         status = smbd_smb2_request_verify_sizes(req, 0x04);
 798         if (!NT_STATUS_IS_OK(status)) {
 799                 return smbd_smb2_request_error(req, status);
 800         }
 801
 802         subreq = smbd_smb2_logoff_send(req, req->sconn->ev_ctx, req);
 803         if (subreq == NULL) {
 804                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 805         }
 806         tevent_req_set_callback(subreq, smbd_smb2_request_logoff_done, req);
 807
 808         /*
 809          * Wait a long time before going async on this to allow
 810          * requests we're waiting on to finish. Set timeout to 10 secs.
 811          */
 812         return smbd_smb2_request_pending_queue(req, subreq, 10000000);
 813 }
 814
 815 static void smbd_smb2_request_logoff_done(struct tevent_req *subreq)
 816 {
 817         struct smbd_smb2_request *smb2req =
 818                 tevent_req_callback_data(subreq,
 819                 struct smbd_smb2_request);
 820         DATA_BLOB outbody;
 821         NTSTATUS status;
 822         NTSTATUS error;
 823
 824         status = smbd_smb2_logoff_recv(subreq);
 825         TALLOC_FREE(subreq);
 826         if (!NT_STATUS_IS_OK(status)) {
 827                 error = smbd_smb2_request_error(smb2req, status);
 828                 if (!NT_STATUS_IS_OK(error)) {
 829                         smbd_server_connection_terminate(smb2req->xconn,
 830                                                         nt_errstr(error));
 831                         return;
 832                 }
 833                 return;
 834         }
 835
 836         outbody = smbd_smb2_generate_outbody(smb2req, 0x04);
 837         if (outbody.data == NULL) {
 838                 error = smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 839                 if (!NT_STATUS_IS_OK(error)) {
 840                         smbd_server_connection_terminate(smb2req->xconn,
 841                                                         nt_errstr(error));
 842                         return;
 843                 }
 844                 return;
 845         }
 846
 847         SSVAL(outbody.data, 0x00, 0x04);        /* struct size */
 848         SSVAL(outbody.data, 0x02, 0);           /* reserved */
 849
 850         error = smbd_smb2_request_done(smb2req, outbody, NULL);
 851         if (!NT_STATUS_IS_OK(error)) {
 852                 smbd_server_connection_terminate(smb2req->xconn,
 853                                                 nt_errstr(error));
 854                 return;
 855         }
 856 }
 857
 858 struct smbd_smb2_logout_state {
 859         struct smbd_smb2_request *smb2req;
 860         struct tevent_queue *wait_queue;
 861 };
 862
 863 static void smbd_smb2_logoff_wait_done(struct tevent_req *subreq);
 864
 865 static struct tevent_req *smbd_smb2_logoff_send(TALLOC_CTX *mem_ctx,
 866                                         struct tevent_context *ev,
 867                                         struct smbd_smb2_request *smb2req)
 868 {
 869         struct tevent_req *req;
 870         struct smbd_smb2_logout_state *state;
 871         struct tevent_req *subreq;
 872         struct smbd_smb2_request *preq;
 873         struct smbXsrv_connection *xconn = smb2req->xconn;
 874
 875         req = tevent_req_create(mem_ctx, &state,
 876                         struct smbd_smb2_logout_state);
 877         if (req == NULL) {
 878                 return NULL;
 879         }
 880         state->smb2req = smb2req;
 881
 882         state->wait_queue = tevent_queue_create(state, "logoff_wait_queue");
 883         if (tevent_req_nomem(state->wait_queue, req)) {
 884                 return tevent_req_post(req, ev);
 885         }
 886
 887         /*
 888          * Make sure that no new request will be able to use this session.
 889          */
 890         smb2req->session->status = NT_STATUS_USER_SESSION_DELETED;
 891
 892         for (preq = xconn->smb2.requests; preq != NULL; preq = preq->next) {
 893                 if (preq == smb2req) {
 894                         /* Can't cancel current request. */
 895                         continue;
 896                 }
 897                 if (preq->session != smb2req->session) {
 898                         /* Request on different session. */
 899                         continue;
 900                 }
 901
 902                 /*
 903                  * Never cancel anything in a compound
 904                  * request. Way too hard to deal with
 905                  * the result.
 906                  */
 907                 if (!preq->compound_related && preq->subreq != NULL) {
 908                         tevent_req_cancel(preq->subreq);
 909                 }
 910
 911                 /*
 912                  * Now wait until the request is finished.
 913                  *
 914                  * We don't set a callback, as we just want to block the
 915                  * wait queue and the talloc_free() of the request will
 916                  * remove the item from the wait queue.
 917                  */
 918                 subreq = tevent_queue_wait_send(preq, ev, state->wait_queue);
 919                 if (tevent_req_nomem(subreq, req)) {
 920                         return tevent_req_post(req, ev);
 921                 }
 922         }
 923
 924         /*
 925          * Now we add our own waiter to the end of the queue,
 926          * this way we get notified when all pending requests are finished
 927          * and send to the socket.
 928          */
 929         subreq = tevent_queue_wait_send(state, ev, state->wait_queue);
 930         if (tevent_req_nomem(subreq, req)) {
 931                 return tevent_req_post(req, ev);
 932         }
 933         tevent_req_set_callback(subreq, smbd_smb2_logoff_wait_done, req);
 934
 935         return req;
 936 }
 937
 938 static void smbd_smb2_logoff_wait_done(struct tevent_req *subreq)
 939 {
 940         struct tevent_req *req = tevent_req_callback_data(
 941                 subreq, struct tevent_req);
 942         struct smbd_smb2_logout_state *state = tevent_req_data(
 943                 req, struct smbd_smb2_logout_state);
 944         NTSTATUS status;
 945
 946         tevent_queue_wait_recv(subreq);
 947         TALLOC_FREE(subreq);
 948
 949         /*
 950          * As we've been awoken, we may have changed
 951          * uid in the meantime. Ensure we're still
 952          * root (SMB2_OP_LOGOFF has .as_root = true).
 953          */
 954         change_to_root_user();
 955
 956         status = smbXsrv_session_logoff(state->smb2req->session);
 957         if (tevent_req_nterror(req, status)) {
 958                 return;
 959         }
 960
 961         /*
 962          * we may need to sign the response, so we need to keep
 963          * the session until the response is sent to the wire.
 964          */
 965         talloc_steal(state->smb2req, state->smb2req->session);
 966
 967         tevent_req_done(req);
 968 }
 969
 970 static NTSTATUS smbd_smb2_logoff_recv(struct tevent_req *req)
 971 {
 972         return tevent_req_simple_recv_ntstatus(req);
 973 }