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 #include "lib/smbd_tevent_queue.h"
  32
  33 static struct tevent_req *smbd_smb2_session_setup_send(TALLOC_CTX *mem_ctx,
  34                                         struct tevent_context *ev,
  35                                         struct smbd_smb2_request *smb2req,
  36                                         uint64_t in_session_id,
  37                                         uint8_t in_flags,
  38                                         uint8_t in_security_mode,
  39                                         uint64_t in_previous_session_id,
  40                                         DATA_BLOB in_security_buffer);
  41 static NTSTATUS smbd_smb2_session_setup_recv(struct tevent_req *req,
  42                                         uint16_t *out_session_flags,
  43                                         TALLOC_CTX *mem_ctx,
  44                                         DATA_BLOB *out_security_buffer,
  45                                         uint64_t *out_session_id);
  46
  47 static void smbd_smb2_request_sesssetup_done(struct tevent_req *subreq);
  48
  49 NTSTATUS smbd_smb2_request_process_sesssetup(struct smbd_smb2_request *smb2req)
  50 {
  51         const uint8_t *inhdr;
  52         const uint8_t *inbody;
  53         uint64_t in_session_id;
  54         uint8_t in_flags;
  55         uint8_t in_security_mode;
  56         uint64_t in_previous_session_id;
  57         uint16_t in_security_offset;
  58         uint16_t in_security_length;
  59         DATA_BLOB in_security_buffer;
  60         NTSTATUS status;
  61         struct tevent_req *subreq;
  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 = SMBD_SMB2_IN_HDR_PTR(smb2req);
  68         inbody = SMBD_SMB2_IN_BODY_PTR(smb2req);
  69
  70         in_session_id = BVAL(inhdr, SMB2_HDR_SESSION_ID);
  71
  72         in_flags = CVAL(inbody, 0x02);
  73         in_security_mode = CVAL(inbody, 0x03);
  74         /* Capabilities = IVAL(inbody, 0x04) */
  75         /* Channel = IVAL(inbody, 0x08) */
  76         in_security_offset = SVAL(inbody, 0x0C);
  77         in_security_length = SVAL(inbody, 0x0E);
  78         in_previous_session_id = BVAL(inbody, 0x10);
  79
  80         if (in_security_offset != (SMB2_HDR_BODY + SMBD_SMB2_IN_BODY_LEN(smb2req))) {
  81                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  82         }
  83
  84         if (in_security_length > SMBD_SMB2_IN_DYN_LEN(smb2req)) {
  85                 return smbd_smb2_request_error(smb2req, NT_STATUS_INVALID_PARAMETER);
  86         }
  87
  88         in_security_buffer.data = SMBD_SMB2_IN_DYN_PTR(smb2req);
  89         in_security_buffer.length = in_security_length;
  90
  91         subreq = smbd_smb2_session_setup_send(smb2req,
  92                                               smb2req->sconn->ev_ctx,
  93                                               smb2req,
  94                                               in_session_id,
  95                                               in_flags,
  96                                               in_security_mode,
  97                                               in_previous_session_id,
  98                                               in_security_buffer);
  99         if (subreq == NULL) {
 100                 return smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 101         }
 102         tevent_req_set_callback(subreq, smbd_smb2_request_sesssetup_done, smb2req);
 103
 104         return smbd_smb2_request_pending_queue(smb2req, subreq, 500);
 105 }
 106
 107 static void smbd_smb2_request_sesssetup_done(struct tevent_req *subreq)
 108 {
 109         struct smbd_smb2_request *smb2req =
 110                 tevent_req_callback_data(subreq,
 111                 struct smbd_smb2_request);
 112         uint8_t *outhdr;
 113         DATA_BLOB outbody;
 114         DATA_BLOB outdyn;
 115         uint16_t out_session_flags;
 116         uint64_t out_session_id;
 117         uint16_t out_security_offset;
 118         DATA_BLOB out_security_buffer = data_blob_null;
 119         NTSTATUS status;
 120         NTSTATUS error; /* transport error */
 121
 122         status = smbd_smb2_session_setup_recv(subreq,
 123                                               &out_session_flags,
 124                                               smb2req,
 125                                               &out_security_buffer,
 126                                               &out_session_id);
 127         TALLOC_FREE(subreq);
 128         if (!NT_STATUS_IS_OK(status) &&
 129             !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 130                 status = nt_status_squash(status);
 131                 error = smbd_smb2_request_error(smb2req, status);
 132                 if (!NT_STATUS_IS_OK(error)) {
 133                         smbd_server_connection_terminate(smb2req->sconn,
 134                                                          nt_errstr(error));
 135                         return;
 136                 }
 137                 return;
 138         }
 139
 140         out_security_offset = SMB2_HDR_BODY + 0x08;
 141
 142         outhdr = SMBD_SMB2_OUT_HDR_PTR(smb2req);
 143
 144         outbody = data_blob_talloc(smb2req->out.vector, NULL, 0x08);
 145         if (outbody.data == NULL) {
 146                 error = smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 147                 if (!NT_STATUS_IS_OK(error)) {
 148                         smbd_server_connection_terminate(smb2req->sconn,
 149                                                          nt_errstr(error));
 150                         return;
 151                 }
 152                 return;
 153         }
 154
 155         SBVAL(outhdr, SMB2_HDR_SESSION_ID, out_session_id);
 156
 157         SSVAL(outbody.data, 0x00, 0x08 + 1);    /* struct size */
 158         SSVAL(outbody.data, 0x02,
 159               out_session_flags);               /* session flags */
 160         SSVAL(outbody.data, 0x04,
 161               out_security_offset);             /* security buffer offset */
 162         SSVAL(outbody.data, 0x06,
 163               out_security_buffer.length);      /* security buffer length */
 164
 165         outdyn = out_security_buffer;
 166
 167         error = smbd_smb2_request_done_ex(smb2req, status, outbody, &outdyn,
 168                                            __location__);
 169         if (!NT_STATUS_IS_OK(error)) {
 170                 smbd_server_connection_terminate(smb2req->sconn,
 171                                                  nt_errstr(error));
 172                 return;
 173         }
 174 }
 175
 176 static NTSTATUS smbd_smb2_auth_generic_return(struct smbXsrv_session *session,
 177                                         struct smbd_smb2_request *smb2req,
 178                                         uint8_t in_security_mode,
 179                                         struct auth_session_info *session_info,
 180                                         uint16_t *out_session_flags,
 181                                         uint64_t *out_session_id)
 182 {
 183         NTSTATUS status;
 184         bool guest = false;
 185         uint8_t session_key[16];
 186         struct smbXsrv_session *x = session;
 187         struct smbXsrv_connection *conn = session->connection;
 188
 189         if ((in_security_mode & SMB2_NEGOTIATE_SIGNING_REQUIRED) ||
 190             lp_server_signing() == SMB_SIGNING_REQUIRED) {
 191                 x->global->signing_required = true;
 192         }
 193
 194         if (lp_smb_encrypt(-1) == SMB_SIGNING_REQUIRED) {
 195                 x->global->encryption_required = true;
 196         }
 197
 198         if (security_session_user_level(session_info, NULL) < SECURITY_USER) {
 199                 /* we map anonymous to guest internally */
 200                 *out_session_flags |= SMB2_SESSION_FLAG_IS_GUEST;
 201                 *out_session_flags |= SMB2_SESSION_FLAG_IS_NULL;
 202                 /* force no signing */
 203                 x->global->signing_required = false;
 204                 guest = true;
 205         }
 206
 207         if (guest && x->global->encryption_required) {
 208                 DEBUG(1,("reject guest session as encryption is required\n"));
 209                 return NT_STATUS_ACCESS_DENIED;
 210         }
 211
 212         if (!(conn->smb2.server.capabilities & SMB2_CAP_ENCRYPTION)) {
 213                 if (x->global->encryption_required) {
 214                         DEBUG(1,("reject session with dialect[0x%04X] "
 215                                  "as encryption is required\n",
 216                                  conn->smb2.server.dialect));
 217                         return NT_STATUS_ACCESS_DENIED;
 218                 }
 219         }
 220
 221         if (x->global->encryption_required) {
 222                 *out_session_flags |= SMB2_SESSION_FLAG_ENCRYPT_DATA;
 223         }
 224
 225         ZERO_STRUCT(session_key);
 226         memcpy(session_key, session_info->session_key.data,
 227                MIN(session_info->session_key.length, sizeof(session_key)));
 228
 229         x->global->signing_key = data_blob_talloc(x->global,
 230                                                   session_key,
 231                                                   sizeof(session_key));
 232         if (x->global->signing_key.data == NULL) {
 233                 ZERO_STRUCT(session_key);
 234                 return NT_STATUS_NO_MEMORY;
 235         }
 236
 237         if (conn->protocol >= PROTOCOL_SMB2_24) {
 238                 const DATA_BLOB label = data_blob_string_const_null("SMB2AESCMAC");
 239                 const DATA_BLOB context = data_blob_string_const_null("SmbSign");
 240
 241                 smb2_key_derivation(session_key, sizeof(session_key),
 242                                     label.data, label.length,
 243                                     context.data, context.length,
 244                                     x->global->signing_key.data);
 245         }
 246
 247         if (conn->protocol >= PROTOCOL_SMB2_24) {
 248                 const DATA_BLOB label = data_blob_string_const_null("SMB2AESCCM");
 249                 const DATA_BLOB context = data_blob_string_const_null("ServerIn ");
 250
 251                 x->global->decryption_key = data_blob_talloc(x->global,
 252                                                              session_key,
 253                                                              sizeof(session_key));
 254                 if (x->global->decryption_key.data == NULL) {
 255                         ZERO_STRUCT(session_key);
 256                         return NT_STATUS_NO_MEMORY;
 257                 }
 258
 259                 smb2_key_derivation(session_key, sizeof(session_key),
 260                                     label.data, label.length,
 261                                     context.data, context.length,
 262                                     x->global->decryption_key.data);
 263         }
 264
 265         if (conn->protocol >= PROTOCOL_SMB2_24) {
 266                 const DATA_BLOB label = data_blob_string_const_null("SMB2AESCCM");
 267                 const DATA_BLOB context = data_blob_string_const_null("ServerOut");
 268
 269                 x->global->encryption_key = data_blob_talloc(x->global,
 270                                                              session_key,
 271                                                              sizeof(session_key));
 272                 if (x->global->encryption_key.data == NULL) {
 273                         ZERO_STRUCT(session_key);
 274                         return NT_STATUS_NO_MEMORY;
 275                 }
 276
 277                 smb2_key_derivation(session_key, sizeof(session_key),
 278                                     label.data, label.length,
 279                                     context.data, context.length,
 280                                     x->global->encryption_key.data);
 281
 282                 generate_random_buffer((uint8_t *)&x->nonce_high, sizeof(x->nonce_high));
 283                 x->nonce_low = 1;
 284         }
 285
 286         x->global->application_key = data_blob_dup_talloc(x->global,
 287                                                 x->global->signing_key);
 288         if (x->global->application_key.data == NULL) {
 289                 ZERO_STRUCT(session_key);
 290                 return NT_STATUS_NO_MEMORY;
 291         }
 292
 293         if (conn->protocol >= PROTOCOL_SMB2_24) {
 294                 const DATA_BLOB label = data_blob_string_const_null("SMB2APP");
 295                 const DATA_BLOB context = data_blob_string_const_null("SmbRpc");
 296
 297                 smb2_key_derivation(session_key, sizeof(session_key),
 298                                     label.data, label.length,
 299                                     context.data, context.length,
 300                                     x->global->application_key.data);
 301         }
 302         ZERO_STRUCT(session_key);
 303
 304         x->global->channels[0].signing_key = data_blob_dup_talloc(x->global->channels,
 305                                                 x->global->signing_key);
 306         if (x->global->channels[0].signing_key.data == NULL) {
 307                 return NT_STATUS_NO_MEMORY;
 308         }
 309
 310         data_blob_clear_free(&session_info->session_key);
 311         session_info->session_key = data_blob_dup_talloc(session_info,
 312                                                 x->global->application_key);
 313         if (session_info->session_key.data == NULL) {
 314                 return NT_STATUS_NO_MEMORY;
 315         }
 316
 317         session->compat = talloc_zero(session, struct user_struct);
 318         if (session->compat == NULL) {
 319                 return NT_STATUS_NO_MEMORY;
 320         }
 321         session->compat->session = session;
 322         session->compat->homes_snum = -1;
 323         session->compat->session_info = session_info;
 324         session->compat->session_keystr = NULL;
 325         session->compat->vuid = session->global->session_wire_id;
 326         DLIST_ADD(smb2req->sconn->users, session->compat);
 327         smb2req->sconn->num_users++;
 328
 329         if (security_session_user_level(session_info, NULL) >= SECURITY_USER) {
 330                 session->compat->homes_snum =
 331                         register_homes_share(session_info->unix_info->unix_name);
 332         }
 333
 334         set_current_user_info(session_info->unix_info->sanitized_username,
 335                               session_info->unix_info->unix_name,
 336                               session_info->info->domain_name);
 337
 338         reload_services(smb2req->sconn, conn_snum_used, true);
 339
 340         session->status = NT_STATUS_OK;
 341         session->global->auth_session_info = session_info;
 342         session->global->auth_session_info_seqnum += 1;
 343         session->global->channels[0].auth_session_info_seqnum =
 344                 session->global->auth_session_info_seqnum;
 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         struct smbXsrv_connection *conn = session->connection;
 387
 388         data_blob_clear_free(&session_info->session_key);
 389         session_info->session_key = data_blob_dup_talloc(session_info,
 390                                                 x->global->application_key);
 391         if (session_info->session_key.data == NULL) {
 392                 return NT_STATUS_NO_MEMORY;
 393         }
 394
 395         session->compat->session_info = session_info;
 396         session->compat->vuid = session->global->session_wire_id;
 397
 398         session->compat->homes_snum =
 399                         register_homes_share(session_info->unix_info->unix_name);
 400
 401         set_current_user_info(session_info->unix_info->sanitized_username,
 402                               session_info->unix_info->unix_name,
 403                               session_info->info->domain_name);
 404
 405         reload_services(smb2req->sconn, conn_snum_used, true);
 406
 407         session->status = NT_STATUS_OK;
 408         TALLOC_FREE(session->global->auth_session_info);
 409         session->global->auth_session_info = session_info;
 410         session->global->auth_session_info_seqnum += 1;
 411         session->global->channels[0].auth_session_info_seqnum =
 412                 session->global->auth_session_info_seqnum;
 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(conn->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         /*
 466          * if state->session is not NULL,
 467          * we remove the session on failure
 468          */
 469         TALLOC_FREE(state->session);
 470         return 0;
 471 }
 472
 473 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq);
 474 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq);
 475
 476 /************************************************************************
 477  We have to tag the state->session pointer with memory talloc'ed
 478  on it to ensure it gets NULL'ed out if the underlying struct smbXsrv_session
 479  is deleted by shutdown whilst this request is in flight.
 480 ************************************************************************/
 481
 482 static NTSTATUS tag_state_session_ptr(struct smbd_smb2_session_setup_state *state)
 483 {
 484         state->pp_self_ref = talloc_zero(state->session,
 485                         struct smbd_smb2_session_setup_state *);
 486         if (state->pp_self_ref == NULL) {
 487                 return NT_STATUS_NO_MEMORY;
 488         }
 489         *state->pp_self_ref = state;
 490         talloc_set_destructor(state->pp_self_ref, pp_self_ref_destructor);
 491         return NT_STATUS_OK;
 492 }
 493
 494 static struct tevent_req *smbd_smb2_session_setup_send(TALLOC_CTX *mem_ctx,
 495                                         struct tevent_context *ev,
 496                                         struct smbd_smb2_request *smb2req,
 497                                         uint64_t in_session_id,
 498                                         uint8_t in_flags,
 499                                         uint8_t in_security_mode,
 500                                         uint64_t in_previous_session_id,
 501                                         DATA_BLOB in_security_buffer)
 502 {
 503         struct tevent_req *req;
 504         struct smbd_smb2_session_setup_state *state;
 505         NTSTATUS status;
 506         NTTIME now = timeval_to_nttime(&smb2req->request_time);
 507         struct tevent_req *subreq;
 508
 509         req = tevent_req_create(mem_ctx, &state,
 510                                 struct smbd_smb2_session_setup_state);
 511         if (req == NULL) {
 512                 return NULL;
 513         }
 514         state->ev = ev;
 515         state->smb2req = smb2req;
 516         state->in_session_id = in_session_id;
 517         state->in_flags = in_flags;
 518         state->in_security_mode = in_security_mode;
 519         state->in_previous_session_id = in_previous_session_id;
 520         state->in_security_buffer = in_security_buffer;
 521
 522         if (in_flags & SMB2_SESSION_FLAG_BINDING) {
 523                 if (smb2req->sconn->conn->protocol < PROTOCOL_SMB2_22) {
 524                         tevent_req_nterror(req, NT_STATUS_REQUEST_NOT_ACCEPTED);
 525                         return tevent_req_post(req, ev);
 526                 }
 527
 528                 /*
 529                  * We do not support multi channel.
 530                  */
 531                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
 532                 return tevent_req_post(req, ev);
 533         }
 534
 535         talloc_set_destructor(state, smbd_smb2_session_setup_state_destructor);
 536
 537         if (state->in_session_id == 0) {
 538                 /* create a new session */
 539                 status = smbXsrv_session_create(state->smb2req->sconn->conn,
 540                                                 now, &state->session);
 541                 if (tevent_req_nterror(req, status)) {
 542                         return tevent_req_post(req, ev);
 543                 }
 544         } else {
 545                 status = smb2srv_session_lookup(state->smb2req->sconn->conn,
 546                                                 state->in_session_id, now,
 547                                                 &state->session);
 548                 if (NT_STATUS_EQUAL(status, NT_STATUS_NETWORK_SESSION_EXPIRED)) {
 549                         status = NT_STATUS_OK;
 550                 }
 551                 if (NT_STATUS_IS_OK(status)) {
 552                         state->session->status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 553                         status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 554                         TALLOC_FREE(state->session->gensec);
 555                 }
 556                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 557                         tevent_req_nterror(req, status);
 558                         return tevent_req_post(req, ev);
 559                 }
 560         }
 561
 562         status = tag_state_session_ptr(state);
 563         if (tevent_req_nterror(req, status)) {
 564                 return tevent_req_post(req, ev);
 565         }
 566
 567         if (state->session->gensec == NULL) {
 568                 status = auth_generic_prepare(state->session,
 569                                               state->session->connection->remote_address,
 570                                               &state->session->gensec);
 571                 if (tevent_req_nterror(req, status)) {
 572                         return tevent_req_post(req, ev);
 573                 }
 574
 575                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_SESSION_KEY);
 576                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_UNIX_TOKEN);
 577
 578                 status = gensec_start_mech_by_oid(state->session->gensec,
 579                                                   GENSEC_OID_SPNEGO);
 580                 if (tevent_req_nterror(req, status)) {
 581                         return tevent_req_post(req, ev);
 582                 }
 583         }
 584
 585         become_root();
 586         subreq = gensec_update_send(state, state->ev,
 587                                     state->session->gensec,
 588                                     state->in_security_buffer);
 589         unbecome_root();
 590         if (tevent_req_nomem(subreq, req)) {
 591                 return tevent_req_post(req, ev);
 592         }
 593         tevent_req_set_callback(subreq, smbd_smb2_session_setup_gensec_done, req);
 594
 595         return req;
 596 }
 597
 598 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq)
 599 {
 600         struct tevent_req *req =
 601                 tevent_req_callback_data(subreq,
 602                 struct tevent_req);
 603         struct smbd_smb2_session_setup_state *state =
 604                 tevent_req_data(req,
 605                 struct smbd_smb2_session_setup_state);
 606         NTSTATUS status;
 607
 608         become_root();
 609         status = gensec_update_recv(subreq, state,
 610                                     &state->out_security_buffer);
 611         unbecome_root();
 612         TALLOC_FREE(subreq);
 613         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) &&
 614             !NT_STATUS_IS_OK(status)) {
 615                 tevent_req_nterror(req, status);
 616                 return;
 617         }
 618
 619         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 620                 state->out_session_id = state->session->global->session_wire_id;
 621                 /* we want to keep the session */
 622                 TALLOC_FREE(state->pp_self_ref);
 623                 tevent_req_nterror(req, status);
 624                 return;
 625         }
 626
 627         status = gensec_session_info(state->session->gensec,
 628                                      state->session->global,
 629                                      &state->session_info);
 630         if (tevent_req_nterror(req, status)) {
 631                 return;
 632         }
 633
 634         if ((state->in_previous_session_id != 0) &&
 635              (state->session->global->session_wire_id !=
 636               state->in_previous_session_id))
 637         {
 638                 subreq = smb2srv_session_close_previous_send(state, state->ev,
 639                                                 state->session->connection,
 640                                                 state->session_info,
 641                                                 state->in_previous_session_id,
 642                                                 state->session->global->session_wire_id);
 643                 if (tevent_req_nomem(subreq, req)) {
 644                         return;
 645                 }
 646                 tevent_req_set_callback(subreq,
 647                                         smbd_smb2_session_setup_previous_done,
 648                                         req);
 649                 return;
 650         }
 651
 652         if (state->session->global->auth_session_info != NULL) {
 653                 status = smbd_smb2_reauth_generic_return(state->session,
 654                                                          state->smb2req,
 655                                                          state->session_info,
 656                                                          &state->out_session_flags,
 657                                                          &state->out_session_id);
 658                 if (tevent_req_nterror(req, status)) {
 659                         return;
 660                 }
 661                 /* we want to keep the session */
 662                 TALLOC_FREE(state->pp_self_ref);
 663                 tevent_req_done(req);
 664                 return;
 665         }
 666
 667         status = smbd_smb2_auth_generic_return(state->session,
 668                                                state->smb2req,
 669                                                state->in_security_mode,
 670                                                state->session_info,
 671                                                &state->out_session_flags,
 672                                                &state->out_session_id);
 673         if (tevent_req_nterror(req, status)) {
 674                 return;
 675         }
 676
 677         /* we want to keep the session */
 678         TALLOC_FREE(state->pp_self_ref);
 679         tevent_req_done(req);
 680         return;
 681 }
 682
 683 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq)
 684 {
 685         struct tevent_req *req =
 686                 tevent_req_callback_data(subreq,
 687                 struct tevent_req);
 688         struct smbd_smb2_session_setup_state *state =
 689                 tevent_req_data(req,
 690                 struct smbd_smb2_session_setup_state);
 691         NTSTATUS status;
 692
 693         status = smb2srv_session_close_previous_recv(subreq);
 694         TALLOC_FREE(subreq);
 695         if (tevent_req_nterror(req, status)) {
 696                 return;
 697         }
 698
 699         if (state->session->global->auth_session_info != NULL) {
 700                 status = smbd_smb2_reauth_generic_return(state->session,
 701                                                          state->smb2req,
 702                                                          state->session_info,
 703                                                          &state->out_session_flags,
 704                                                          &state->out_session_id);
 705                 if (tevent_req_nterror(req, status)) {
 706                         return;
 707                 }
 708                 /* we want to keep the session */
 709                 TALLOC_FREE(state->pp_self_ref);
 710                 tevent_req_done(req);
 711                 return;
 712         }
 713
 714         status = smbd_smb2_auth_generic_return(state->session,
 715                                                state->smb2req,
 716                                                state->in_security_mode,
 717                                                state->session_info,
 718                                                &state->out_session_flags,
 719                                                &state->out_session_id);
 720         if (tevent_req_nterror(req, status)) {
 721                 return;
 722         }
 723
 724         /* we want to keep the session */
 725         TALLOC_FREE(state->pp_self_ref);
 726         tevent_req_done(req);
 727         return;
 728 }
 729
 730 static NTSTATUS smbd_smb2_session_setup_recv(struct tevent_req *req,
 731                                         uint16_t *out_session_flags,
 732                                         TALLOC_CTX *mem_ctx,
 733                                         DATA_BLOB *out_security_buffer,
 734                                         uint64_t *out_session_id)
 735 {
 736         struct smbd_smb2_session_setup_state *state =
 737                 tevent_req_data(req,
 738                 struct smbd_smb2_session_setup_state);
 739         NTSTATUS status;
 740
 741         if (tevent_req_is_nterror(req, &status)) {
 742                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 743                         tevent_req_received(req);
 744                         return nt_status_squash(status);
 745                 }
 746         } else {
 747                 status = NT_STATUS_OK;
 748         }
 749
 750         *out_session_flags = state->out_session_flags;
 751         *out_security_buffer = state->out_security_buffer;
 752         *out_session_id = state->out_session_id;
 753
 754         talloc_steal(mem_ctx, out_security_buffer->data);
 755         tevent_req_received(req);
 756         return status;
 757 }
 758
 759 static struct tevent_req *smbd_smb2_logoff_send(TALLOC_CTX *mem_ctx,
 760                                         struct tevent_context *ev,
 761                                         struct smbd_smb2_request *smb2req);
 762 static NTSTATUS smbd_smb2_logoff_recv(struct tevent_req *req);
 763 static void smbd_smb2_request_logoff_done(struct tevent_req *subreq);
 764
 765 NTSTATUS smbd_smb2_request_process_logoff(struct smbd_smb2_request *req)
 766 {
 767         NTSTATUS status;
 768         struct tevent_req *subreq = NULL;
 769
 770         status = smbd_smb2_request_verify_sizes(req, 0x04);
 771         if (!NT_STATUS_IS_OK(status)) {
 772                 return smbd_smb2_request_error(req, status);
 773         }
 774
 775         subreq = smbd_smb2_logoff_send(req, req->sconn->ev_ctx, req);
 776         if (subreq == NULL) {
 777                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 778         }
 779         tevent_req_set_callback(subreq, smbd_smb2_request_logoff_done, req);
 780
 781         /*
 782          * Wait a long time before going async on this to allow
 783          * requests we're waiting on to finish. Set timeout to 10 secs.
 784          */
 785         return smbd_smb2_request_pending_queue(req, subreq, 10000000);
 786 }
 787
 788 static void smbd_smb2_request_logoff_done(struct tevent_req *subreq)
 789 {
 790         struct smbd_smb2_request *smb2req =
 791                 tevent_req_callback_data(subreq,
 792                 struct smbd_smb2_request);
 793         DATA_BLOB outbody;
 794         NTSTATUS status;
 795         NTSTATUS error;
 796
 797         status = smbd_smb2_logoff_recv(subreq);
 798         TALLOC_FREE(subreq);
 799         if (!NT_STATUS_IS_OK(status)) {
 800                 error = smbd_smb2_request_error(smb2req, status);
 801                 if (!NT_STATUS_IS_OK(error)) {
 802                         smbd_server_connection_terminate(smb2req->sconn,
 803                                                         nt_errstr(error));
 804                         return;
 805                 }
 806                 return;
 807         }
 808
 809         outbody = data_blob_talloc(smb2req->out.vector, NULL, 0x04);
 810         if (outbody.data == NULL) {
 811                 error = smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 812                 if (!NT_STATUS_IS_OK(error)) {
 813                         smbd_server_connection_terminate(smb2req->sconn,
 814                                                         nt_errstr(error));
 815                         return;
 816                 }
 817                 return;
 818         }
 819
 820         SSVAL(outbody.data, 0x00, 0x04);        /* struct size */
 821         SSVAL(outbody.data, 0x02, 0);           /* reserved */
 822
 823         error = smbd_smb2_request_done(smb2req, outbody, NULL);
 824         if (!NT_STATUS_IS_OK(error)) {
 825                 smbd_server_connection_terminate(smb2req->sconn,
 826                                                 nt_errstr(error));
 827                 return;
 828         }
 829 }
 830
 831 struct smbd_smb2_logout_state {
 832         struct smbd_smb2_request *smb2req;
 833         struct tevent_queue *wait_queue;
 834 };
 835
 836 static void smbd_smb2_logoff_wait_done(struct tevent_req *subreq);
 837
 838 static struct tevent_req *smbd_smb2_logoff_send(TALLOC_CTX *mem_ctx,
 839                                         struct tevent_context *ev,
 840                                         struct smbd_smb2_request *smb2req)
 841 {
 842         struct tevent_req *req;
 843         struct smbd_smb2_logout_state *state;
 844         struct tevent_req *subreq;
 845         struct smbd_smb2_request *preq;
 846
 847         req = tevent_req_create(mem_ctx, &state,
 848                         struct smbd_smb2_logout_state);
 849         if (req == NULL) {
 850                 return NULL;
 851         }
 852         state->smb2req = smb2req;
 853
 854         state->wait_queue = tevent_queue_create(state, "logoff_wait_queue");
 855         if (tevent_req_nomem(state->wait_queue, req)) {
 856                 return tevent_req_post(req, ev);
 857         }
 858
 859         /*
 860          * Make sure that no new request will be able to use this session.
 861          */
 862         smb2req->session->status = NT_STATUS_USER_SESSION_DELETED;
 863
 864         for (preq = smb2req->sconn->smb2.requests; preq != NULL; preq = preq->next) {
 865                 if (preq == smb2req) {
 866                         /* Can't cancel current request. */
 867                         continue;
 868                 }
 869                 if (preq->session != smb2req->session) {
 870                         /* Request on different session. */
 871                         continue;
 872                 }
 873
 874                 /*
 875                  * Never cancel anything in a compound
 876                  * request. Way too hard to deal with
 877                  * the result.
 878                  */
 879                 if (!preq->compound_related && preq->subreq != NULL) {
 880                         tevent_req_cancel(preq->subreq);
 881                 }
 882
 883                 /*
 884                  * Now wait until the request is finished.
 885                  *
 886                  * We don't set a callback, as we just want to block the
 887                  * wait queue and the talloc_free() of the request will
 888                  * remove the item from the wait queue.
 889                  */
 890                 subreq = smbd_tevent_queue_wait_send(preq, ev, state->wait_queue);
 891                 if (tevent_req_nomem(subreq, req)) {
 892                         return tevent_req_post(req, ev);
 893                 }
 894         }
 895
 896         /*
 897          * Now we add our own waiter to the end of the queue,
 898          * this way we get notified when all pending requests are finished
 899          * and send to the socket.
 900          */
 901         subreq = smbd_tevent_queue_wait_send(state, ev, state->wait_queue);
 902         if (tevent_req_nomem(subreq, req)) {
 903                 return tevent_req_post(req, ev);
 904         }
 905         tevent_req_set_callback(subreq, smbd_smb2_logoff_wait_done, req);
 906
 907         return req;
 908 }
 909
 910 static void smbd_smb2_logoff_wait_done(struct tevent_req *subreq)
 911 {
 912         struct tevent_req *req = tevent_req_callback_data(
 913                 subreq, struct tevent_req);
 914         struct smbd_smb2_logout_state *state = tevent_req_data(
 915                 req, struct smbd_smb2_logout_state);
 916         NTSTATUS status;
 917
 918         smbd_tevent_queue_wait_recv(subreq);
 919         TALLOC_FREE(subreq);
 920
 921         /*
 922          * As we've been awoken, we may have changed
 923          * uid in the meantime. Ensure we're still
 924          * root (SMB2_OP_LOGOFF has .as_root = true).
 925          */
 926         change_to_root_user();
 927
 928         status = smbXsrv_session_logoff(state->smb2req->session);
 929         if (tevent_req_nterror(req, status)) {
 930                 return;
 931         }
 932
 933         /*
 934          * we may need to sign the response, so we need to keep
 935          * the session until the response is sent to the wire.
 936          */
 937         talloc_steal(state->smb2req, state->smb2req->session);
 938
 939         tevent_req_done(req);
 940 }
 941
 942 static NTSTATUS smbd_smb2_logoff_recv(struct tevent_req *req)
 943 {
 944         return tevent_req_simple_recv_ntstatus(req);
 945 }