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_wrap_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_wrap_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_wrap_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_wrap_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                 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 (conn->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->expiration_time = gensec_expire_time(session->gensec);
 345
 346         if (!session_claim(session)) {
 347                 DEBUG(1, ("smb2: Failed to claim session "
 348                         "for vuid=%llu\n",
 349                         (unsigned long long)session->compat->vuid));
 350                 return NT_STATUS_LOGON_FAILURE;
 351         }
 352
 353         status = smbXsrv_session_update(session);
 354         if (!NT_STATUS_IS_OK(status)) {
 355                 DEBUG(0, ("smb2: Failed to update session for vuid=%llu - %s\n",
 356                           (unsigned long long)session->compat->vuid,
 357                           nt_errstr(status)));
 358                 return NT_STATUS_LOGON_FAILURE;
 359         }
 360
 361         /*
 362          * we attach the session to the request
 363          * so that the response can be signed
 364          */
 365         if (!guest) {
 366                 smb2req->do_signing = true;
 367         }
 368
 369         global_client_caps |= (CAP_LEVEL_II_OPLOCKS|CAP_STATUS32);
 370
 371         *out_session_id = session->global->session_wire_id;
 372
 373         return NT_STATUS_OK;
 374 }
 375
 376 static NTSTATUS smbd_smb2_reauth_generic_return(struct smbXsrv_session *session,
 377                                         struct smbd_smb2_request *smb2req,
 378                                         struct auth_session_info *session_info,
 379                                         uint16_t *out_session_flags,
 380                                         uint64_t *out_session_id)
 381 {
 382         NTSTATUS status;
 383         struct smbXsrv_session *x = session;
 384         struct smbXsrv_connection *conn = session->connection;
 385
 386         data_blob_clear_free(&session_info->session_key);
 387         session_info->session_key = data_blob_dup_talloc(session_info,
 388                                                 x->global->application_key);
 389         if (session_info->session_key.data == NULL) {
 390                 return NT_STATUS_NO_MEMORY;
 391         }
 392
 393         session->compat->session_info = session_info;
 394         session->compat->vuid = session->global->session_wire_id;
 395
 396         session->compat->homes_snum =
 397                         register_homes_share(session_info->unix_info->unix_name);
 398
 399         set_current_user_info(session_info->unix_info->sanitized_username,
 400                               session_info->unix_info->unix_name,
 401                               session_info->info->domain_name);
 402
 403         reload_services(smb2req->sconn, conn_snum_used, true);
 404
 405         session->status = NT_STATUS_OK;
 406         TALLOC_FREE(session->global->auth_session_info);
 407         session->global->auth_session_info = session_info;
 408         session->global->auth_session_info_seqnum += 1;
 409         session->global->channels[0].auth_session_info_seqnum =
 410                 session->global->auth_session_info_seqnum;
 411         session->global->expiration_time = gensec_expire_time(session->gensec);
 412
 413         status = smbXsrv_session_update(session);
 414         if (!NT_STATUS_IS_OK(status)) {
 415                 DEBUG(0, ("smb2: Failed to update session for vuid=%llu - %s\n",
 416                           (unsigned long long)session->compat->vuid,
 417                           nt_errstr(status)));
 418                 return NT_STATUS_LOGON_FAILURE;
 419         }
 420
 421         conn_clear_vuid_caches(conn->sconn, session->compat->vuid);
 422
 423         if (security_session_user_level(session_info, NULL) >= SECURITY_USER) {
 424                 smb2req->do_signing = true;
 425         }
 426
 427         *out_session_id = session->global->session_wire_id;
 428
 429         return NT_STATUS_OK;
 430 }
 431
 432 struct smbd_smb2_session_setup_state {
 433         struct tevent_context *ev;
 434         struct smbd_smb2_request *smb2req;
 435         uint64_t in_session_id;
 436         uint8_t in_flags;
 437         uint8_t in_security_mode;
 438         uint64_t in_previous_session_id;
 439         DATA_BLOB in_security_buffer;
 440         struct smbXsrv_session *session;
 441         struct auth_session_info *session_info;
 442         uint16_t out_session_flags;
 443         DATA_BLOB out_security_buffer;
 444         uint64_t out_session_id;
 445 };
 446
 447 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq);
 448 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq);
 449
 450 static struct tevent_req *smbd_smb2_session_setup_send(TALLOC_CTX *mem_ctx,
 451                                         struct tevent_context *ev,
 452                                         struct smbd_smb2_request *smb2req,
 453                                         uint64_t in_session_id,
 454                                         uint8_t in_flags,
 455                                         uint8_t in_security_mode,
 456                                         uint64_t in_previous_session_id,
 457                                         DATA_BLOB in_security_buffer)
 458 {
 459         struct tevent_req *req;
 460         struct smbd_smb2_session_setup_state *state;
 461         NTSTATUS status;
 462         NTTIME now = timeval_to_nttime(&smb2req->request_time);
 463         struct tevent_req *subreq;
 464
 465         req = tevent_req_create(mem_ctx, &state,
 466                                 struct smbd_smb2_session_setup_state);
 467         if (req == NULL) {
 468                 return NULL;
 469         }
 470         state->ev = ev;
 471         state->smb2req = smb2req;
 472         state->in_session_id = in_session_id;
 473         state->in_flags = in_flags;
 474         state->in_security_mode = in_security_mode;
 475         state->in_previous_session_id = in_previous_session_id;
 476         state->in_security_buffer = in_security_buffer;
 477
 478         if (in_flags & SMB2_SESSION_FLAG_BINDING) {
 479                 if (smb2req->sconn->conn->protocol < PROTOCOL_SMB2_22) {
 480                         tevent_req_nterror(req, NT_STATUS_REQUEST_NOT_ACCEPTED);
 481                         return tevent_req_post(req, ev);
 482                 }
 483
 484                 /*
 485                  * We do not support multi channel.
 486                  */
 487                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
 488                 return tevent_req_post(req, ev);
 489         }
 490
 491         if (state->in_session_id == 0) {
 492                 /* create a new session */
 493                 status = smbXsrv_session_create(state->smb2req->sconn->conn,
 494                                                 now, &state->session);
 495                 if (tevent_req_nterror(req, status)) {
 496                         return tevent_req_post(req, ev);
 497                 }
 498                 smb2req->session = state->session;
 499         } else {
 500                 status = smb2srv_session_lookup(state->smb2req->sconn->conn,
 501                                                 state->in_session_id, now,
 502                                                 &state->session);
 503                 if (NT_STATUS_EQUAL(status, NT_STATUS_NETWORK_SESSION_EXPIRED)) {
 504                         status = NT_STATUS_OK;
 505                 }
 506                 if (NT_STATUS_IS_OK(status)) {
 507                         state->session->status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 508                         status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 509                         TALLOC_FREE(state->session->gensec);
 510                 }
 511                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 512                         tevent_req_nterror(req, status);
 513                         return tevent_req_post(req, ev);
 514                 }
 515         }
 516
 517         if (state->session->gensec == NULL) {
 518                 status = auth_generic_prepare(state->session,
 519                                               state->session->connection->remote_address,
 520                                               &state->session->gensec);
 521                 if (tevent_req_nterror(req, status)) {
 522                         return tevent_req_post(req, ev);
 523                 }
 524
 525                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_SESSION_KEY);
 526                 gensec_want_feature(state->session->gensec, GENSEC_FEATURE_UNIX_TOKEN);
 527
 528                 status = gensec_start_mech_by_oid(state->session->gensec,
 529                                                   GENSEC_OID_SPNEGO);
 530                 if (tevent_req_nterror(req, status)) {
 531                         return tevent_req_post(req, ev);
 532                 }
 533         }
 534
 535         become_root();
 536         subreq = gensec_update_send(state, state->ev,
 537                                     state->session->gensec,
 538                                     state->in_security_buffer);
 539         unbecome_root();
 540         if (tevent_req_nomem(subreq, req)) {
 541                 return tevent_req_post(req, ev);
 542         }
 543         tevent_req_set_callback(subreq, smbd_smb2_session_setup_gensec_done, req);
 544
 545         return req;
 546 }
 547
 548 static void smbd_smb2_session_setup_gensec_done(struct tevent_req *subreq)
 549 {
 550         struct tevent_req *req =
 551                 tevent_req_callback_data(subreq,
 552                 struct tevent_req);
 553         struct smbd_smb2_session_setup_state *state =
 554                 tevent_req_data(req,
 555                 struct smbd_smb2_session_setup_state);
 556         NTSTATUS status;
 557
 558         become_root();
 559         status = gensec_update_recv(subreq, state,
 560                                     &state->out_security_buffer);
 561         unbecome_root();
 562         TALLOC_FREE(subreq);
 563         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) &&
 564             !NT_STATUS_IS_OK(status)) {
 565                 tevent_req_nterror(req, status);
 566                 return;
 567         }
 568
 569         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 570                 state->out_session_id = state->session->global->session_wire_id;
 571                 tevent_req_nterror(req, status);
 572                 return;
 573         }
 574
 575         status = gensec_session_info(state->session->gensec,
 576                                      state->session->global,
 577                                      &state->session_info);
 578         if (tevent_req_nterror(req, status)) {
 579                 return;
 580         }
 581
 582         if ((state->in_previous_session_id != 0) &&
 583              (state->session->global->session_wire_id !=
 584               state->in_previous_session_id))
 585         {
 586                 subreq = smb2srv_session_close_previous_send(state, state->ev,
 587                                                 state->session->connection,
 588                                                 state->session_info,
 589                                                 state->in_previous_session_id,
 590                                                 state->session->global->session_wire_id);
 591                 if (tevent_req_nomem(subreq, req)) {
 592                         return;
 593                 }
 594                 tevent_req_set_callback(subreq,
 595                                         smbd_smb2_session_setup_previous_done,
 596                                         req);
 597                 return;
 598         }
 599
 600         if (state->session->global->auth_session_info != NULL) {
 601                 status = smbd_smb2_reauth_generic_return(state->session,
 602                                                          state->smb2req,
 603                                                          state->session_info,
 604                                                          &state->out_session_flags,
 605                                                          &state->out_session_id);
 606                 if (tevent_req_nterror(req, status)) {
 607                         return;
 608                 }
 609                 tevent_req_done(req);
 610                 return;
 611         }
 612
 613         status = smbd_smb2_auth_generic_return(state->session,
 614                                                state->smb2req,
 615                                                state->in_security_mode,
 616                                                state->session_info,
 617                                                &state->out_session_flags,
 618                                                &state->out_session_id);
 619         if (tevent_req_nterror(req, status)) {
 620                 return;
 621         }
 622
 623         tevent_req_done(req);
 624         return;
 625 }
 626
 627 static void smbd_smb2_session_setup_previous_done(struct tevent_req *subreq)
 628 {
 629         struct tevent_req *req =
 630                 tevent_req_callback_data(subreq,
 631                 struct tevent_req);
 632         struct smbd_smb2_session_setup_state *state =
 633                 tevent_req_data(req,
 634                 struct smbd_smb2_session_setup_state);
 635         NTSTATUS status;
 636
 637         status = smb2srv_session_close_previous_recv(subreq);
 638         TALLOC_FREE(subreq);
 639         if (tevent_req_nterror(req, status)) {
 640                 return;
 641         }
 642
 643         if (state->session->global->auth_session_info != NULL) {
 644                 status = smbd_smb2_reauth_generic_return(state->session,
 645                                                          state->smb2req,
 646                                                          state->session_info,
 647                                                          &state->out_session_flags,
 648                                                          &state->out_session_id);
 649                 if (tevent_req_nterror(req, status)) {
 650                         return;
 651                 }
 652                 tevent_req_done(req);
 653                 return;
 654         }
 655
 656         status = smbd_smb2_auth_generic_return(state->session,
 657                                                state->smb2req,
 658                                                state->in_security_mode,
 659                                                state->session_info,
 660                                                &state->out_session_flags,
 661                                                &state->out_session_id);
 662         if (tevent_req_nterror(req, status)) {
 663                 return;
 664         }
 665
 666         tevent_req_done(req);
 667         return;
 668 }
 669
 670 static NTSTATUS smbd_smb2_session_setup_recv(struct tevent_req *req,
 671                                         uint16_t *out_session_flags,
 672                                         TALLOC_CTX *mem_ctx,
 673                                         DATA_BLOB *out_security_buffer,
 674                                         uint64_t *out_session_id)
 675 {
 676         struct smbd_smb2_session_setup_state *state =
 677                 tevent_req_data(req,
 678                 struct smbd_smb2_session_setup_state);
 679         NTSTATUS status;
 680
 681         if (tevent_req_is_nterror(req, &status)) {
 682                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 683                         tevent_req_received(req);
 684                         return nt_status_squash(status);
 685                 }
 686         } else {
 687                 status = NT_STATUS_OK;
 688         }
 689
 690         *out_session_flags = state->out_session_flags;
 691         *out_security_buffer = state->out_security_buffer;
 692         *out_session_id = state->out_session_id;
 693
 694         talloc_steal(mem_ctx, out_security_buffer->data);
 695         tevent_req_received(req);
 696         return status;
 697 }
 698
 699 struct smbd_smb2_session_setup_wrap_state {
 700         struct tevent_context *ev;
 701         struct smbd_smb2_request *smb2req;
 702         uint64_t in_session_id;
 703         uint8_t in_flags;
 704         uint8_t in_security_mode;
 705         uint64_t in_previous_session_id;
 706         DATA_BLOB in_security_buffer;
 707         uint16_t out_session_flags;
 708         DATA_BLOB out_security_buffer;
 709         uint64_t out_session_id;
 710         NTSTATUS error;
 711 };
 712
 713 static void smbd_smb2_session_setup_wrap_setup_done(struct tevent_req *subreq);
 714 static void smbd_smb2_session_setup_wrap_shutdown_done(struct tevent_req *subreq);
 715
 716 static struct tevent_req *smbd_smb2_session_setup_wrap_send(TALLOC_CTX *mem_ctx,
 717                                         struct tevent_context *ev,
 718                                         struct smbd_smb2_request *smb2req,
 719                                         uint64_t in_session_id,
 720                                         uint8_t in_flags,
 721                                         uint8_t in_security_mode,
 722                                         uint64_t in_previous_session_id,
 723                                         DATA_BLOB in_security_buffer)
 724 {
 725         struct tevent_req *req;
 726         struct smbd_smb2_session_setup_wrap_state *state;
 727         struct tevent_req *subreq;
 728
 729         req = tevent_req_create(mem_ctx, &state,
 730                                 struct smbd_smb2_session_setup_wrap_state);
 731         if (req == NULL) {
 732                 return NULL;
 733         }
 734         state->ev = ev;
 735         state->smb2req = smb2req;
 736         state->in_session_id = in_session_id;
 737         state->in_flags = in_flags;
 738         state->in_security_mode = in_security_mode;
 739         state->in_previous_session_id = in_previous_session_id;
 740         state->in_security_buffer = in_security_buffer;
 741
 742         subreq = smbd_smb2_session_setup_send(state, state->ev,
 743                                               state->smb2req,
 744                                               state->in_session_id,
 745                                               state->in_flags,
 746                                               state->in_security_mode,
 747                                               state->in_previous_session_id,
 748                                               state->in_security_buffer);
 749         if (tevent_req_nomem(subreq, req)) {
 750                 return tevent_req_post(req, ev);
 751         }
 752         tevent_req_set_callback(subreq,
 753                                 smbd_smb2_session_setup_wrap_setup_done, req);
 754
 755         return req;
 756 }
 757
 758 static void smbd_smb2_session_setup_wrap_setup_done(struct tevent_req *subreq)
 759 {
 760         struct tevent_req *req =
 761                 tevent_req_callback_data(subreq,
 762                 struct tevent_req);
 763         struct smbd_smb2_session_setup_wrap_state *state =
 764                 tevent_req_data(req,
 765                 struct smbd_smb2_session_setup_wrap_state);
 766         NTSTATUS status;
 767
 768         status = smbd_smb2_session_setup_recv(subreq,
 769                                               &state->out_session_flags,
 770                                               state,
 771                                               &state->out_security_buffer,
 772                                               &state->out_session_id);
 773         TALLOC_FREE(subreq);
 774         if (NT_STATUS_IS_OK(status)) {
 775                 tevent_req_done(req);
 776                 return;
 777         }
 778         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 779                 tevent_req_nterror(req, status);
 780                 return;
 781         }
 782
 783         if (state->smb2req->session == NULL) {
 784                 tevent_req_nterror(req, status);
 785                 return;
 786         }
 787
 788         state->error = status;
 789
 790         subreq = smb2srv_session_shutdown_send(state, state->ev,
 791                                                state->smb2req->session,
 792                                                state->smb2req);
 793         if (tevent_req_nomem(subreq, req)) {
 794                 return;
 795         }
 796         tevent_req_set_callback(subreq,
 797                                 smbd_smb2_session_setup_wrap_shutdown_done,
 798                                 req);
 799 }
 800
 801 static void smbd_smb2_session_setup_wrap_shutdown_done(struct tevent_req *subreq)
 802 {
 803         struct tevent_req *req =
 804                 tevent_req_callback_data(subreq,
 805                 struct tevent_req);
 806         struct smbd_smb2_session_setup_wrap_state *state =
 807                 tevent_req_data(req,
 808                 struct smbd_smb2_session_setup_wrap_state);
 809         NTSTATUS status;
 810
 811         status = smb2srv_session_shutdown_recv(subreq);
 812         TALLOC_FREE(subreq);
 813         if (tevent_req_nterror(req, status)) {
 814                 return;
 815         }
 816
 817         /*
 818          * we may need to sign the response, so we need to keep
 819          * the session until the response is sent to the wire.
 820          */
 821         talloc_steal(state->smb2req, state->smb2req->session);
 822
 823         tevent_req_nterror(req, state->error);
 824 }
 825
 826 static NTSTATUS smbd_smb2_session_setup_wrap_recv(struct tevent_req *req,
 827                                         uint16_t *out_session_flags,
 828                                         TALLOC_CTX *mem_ctx,
 829                                         DATA_BLOB *out_security_buffer,
 830                                         uint64_t *out_session_id)
 831 {
 832         struct smbd_smb2_session_setup_wrap_state *state =
 833                 tevent_req_data(req,
 834                 struct smbd_smb2_session_setup_wrap_state);
 835         NTSTATUS status;
 836
 837         if (tevent_req_is_nterror(req, &status)) {
 838                 if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 839                         tevent_req_received(req);
 840                         return nt_status_squash(status);
 841                 }
 842         } else {
 843                 status = NT_STATUS_OK;
 844         }
 845
 846         *out_session_flags = state->out_session_flags;
 847         *out_security_buffer = state->out_security_buffer;
 848         *out_session_id = state->out_session_id;
 849
 850         talloc_steal(mem_ctx, out_security_buffer->data);
 851         tevent_req_received(req);
 852         return status;
 853 }
 854
 855 static struct tevent_req *smbd_smb2_logoff_send(TALLOC_CTX *mem_ctx,
 856                                         struct tevent_context *ev,
 857                                         struct smbd_smb2_request *smb2req);
 858 static NTSTATUS smbd_smb2_logoff_recv(struct tevent_req *req);
 859 static void smbd_smb2_request_logoff_done(struct tevent_req *subreq);
 860
 861 NTSTATUS smbd_smb2_request_process_logoff(struct smbd_smb2_request *req)
 862 {
 863         NTSTATUS status;
 864         struct tevent_req *subreq = NULL;
 865
 866         status = smbd_smb2_request_verify_sizes(req, 0x04);
 867         if (!NT_STATUS_IS_OK(status)) {
 868                 return smbd_smb2_request_error(req, status);
 869         }
 870
 871         subreq = smbd_smb2_logoff_send(req, req->sconn->ev_ctx, req);
 872         if (subreq == NULL) {
 873                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 874         }
 875         tevent_req_set_callback(subreq, smbd_smb2_request_logoff_done, req);
 876
 877         /*
 878          * Wait a long time before going async on this to allow
 879          * requests we're waiting on to finish. Set timeout to 10 secs.
 880          */
 881         return smbd_smb2_request_pending_queue(req, subreq, 10000000);
 882 }
 883
 884 static void smbd_smb2_request_logoff_done(struct tevent_req *subreq)
 885 {
 886         struct smbd_smb2_request *smb2req =
 887                 tevent_req_callback_data(subreq,
 888                 struct smbd_smb2_request);
 889         DATA_BLOB outbody;
 890         NTSTATUS status;
 891         NTSTATUS error;
 892
 893         status = smbd_smb2_logoff_recv(subreq);
 894         TALLOC_FREE(subreq);
 895         if (!NT_STATUS_IS_OK(status)) {
 896                 error = smbd_smb2_request_error(smb2req, status);
 897                 if (!NT_STATUS_IS_OK(error)) {
 898                         smbd_server_connection_terminate(smb2req->sconn,
 899                                                         nt_errstr(error));
 900                         return;
 901                 }
 902                 return;
 903         }
 904
 905         outbody = data_blob_talloc(smb2req->out.vector, NULL, 0x04);
 906         if (outbody.data == NULL) {
 907                 error = smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 908                 if (!NT_STATUS_IS_OK(error)) {
 909                         smbd_server_connection_terminate(smb2req->sconn,
 910                                                         nt_errstr(error));
 911                         return;
 912                 }
 913                 return;
 914         }
 915
 916         SSVAL(outbody.data, 0x00, 0x04);        /* struct size */
 917         SSVAL(outbody.data, 0x02, 0);           /* reserved */
 918
 919         error = smbd_smb2_request_done(smb2req, outbody, NULL);
 920         if (!NT_STATUS_IS_OK(error)) {
 921                 smbd_server_connection_terminate(smb2req->sconn,
 922                                                 nt_errstr(error));
 923                 return;
 924         }
 925 }
 926
 927 struct smbd_smb2_logoff_state {
 928         struct smbd_smb2_request *smb2req;
 929 };
 930
 931 static void smbd_smb2_logoff_shutdown_done(struct tevent_req *subreq);
 932
 933 static struct tevent_req *smbd_smb2_logoff_send(TALLOC_CTX *mem_ctx,
 934                                         struct tevent_context *ev,
 935                                         struct smbd_smb2_request *smb2req)
 936 {
 937         struct tevent_req *req;
 938         struct smbd_smb2_logoff_state *state;
 939         struct tevent_req *subreq;
 940
 941         req = tevent_req_create(mem_ctx, &state,
 942                         struct smbd_smb2_logoff_state);
 943         if (req == NULL) {
 944                 return NULL;
 945         }
 946         state->smb2req = smb2req;
 947
 948         subreq = smb2srv_session_shutdown_send(state, ev,
 949                                                smb2req->session,
 950                                                smb2req);
 951         if (tevent_req_nomem(subreq, req)) {
 952                 return tevent_req_post(req, ev);
 953         }
 954         tevent_req_set_callback(subreq, smbd_smb2_logoff_shutdown_done, req);
 955
 956         return req;
 957 }
 958
 959 static void smbd_smb2_logoff_shutdown_done(struct tevent_req *subreq)
 960 {
 961         struct tevent_req *req = tevent_req_callback_data(
 962                 subreq, struct tevent_req);
 963         struct smbd_smb2_logoff_state *state = tevent_req_data(
 964                 req, struct smbd_smb2_logoff_state);
 965         NTSTATUS status;
 966
 967         status = smb2srv_session_shutdown_recv(subreq);
 968         if (tevent_req_nterror(req, status)) {
 969                 return;
 970         }
 971         TALLOC_FREE(subreq);
 972
 973         /*
 974          * As we've been awoken, we may have changed
 975          * uid in the meantime. Ensure we're still
 976          * root (SMB2_OP_LOGOFF has .as_root = true).
 977          */
 978         change_to_root_user();
 979
 980         status = smbXsrv_session_logoff(state->smb2req->session);
 981         if (tevent_req_nterror(req, status)) {
 982                 return;
 983         }
 984
 985         /*
 986          * we may need to sign the response, so we need to keep
 987          * the session until the response is sent to the wire.
 988          */
 989         talloc_steal(state->smb2req, state->smb2req->session);
 990
 991         tevent_req_done(req);
 992 }
 993
 994 static NTSTATUS smbd_smb2_logoff_recv(struct tevent_req *req)
 995 {
 996         return tevent_req_simple_recv_ntstatus(req);
 997 }