source3/smbd/smb2_tcon.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Core SMB2 server
   4
   5    Copyright (C) Stefan Metzmacher 2009
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "smbd/smbd.h"
  23 #include "smbd/globals.h"
  24 #include "../libcli/smb/smb_common.h"
  25 #include "../libcli/security/security.h"
  26 #include "auth.h"
  27 #include "lib/param/loadparm.h"
  28 #include "../lib/util/tevent_ntstatus.h"
  29
  30 #undef DBGC_CLASS
  31 #define DBGC_CLASS DBGC_SMB2
  32
  33 static struct tevent_req *smbd_smb2_tree_connect_send(TALLOC_CTX *mem_ctx,
  34                                         struct tevent_context *ev,
  35                                         struct smbd_smb2_request *smb2req,
  36                                         uint16_t in_flags,
  37                                         const char *in_path);
  38 static NTSTATUS smbd_smb2_tree_connect_recv(struct tevent_req *req,
  39                                             uint8_t *out_share_type,
  40                                             uint32_t *out_share_flags,
  41                                             uint32_t *out_capabilities,
  42                                             uint32_t *out_maximal_access,
  43                                             uint32_t *out_tree_id,
  44                                             bool *disconnect);
  45
  46 static void smbd_smb2_request_tcon_done(struct tevent_req *subreq);
  47
  48 NTSTATUS smbd_smb2_request_process_tcon(struct smbd_smb2_request *req)
  49 {
  50         struct smbXsrv_connection *xconn = req->xconn;
  51         const uint8_t *inbody;
  52         uint16_t in_flags;
  53         uint16_t in_path_offset;
  54         uint16_t in_path_length;
  55         DATA_BLOB in_path_buffer;
  56         char *in_path_string;
  57         size_t in_path_string_size;
  58         NTSTATUS status;
  59         bool ok;
  60         struct tevent_req *subreq;
  61
  62         status = smbd_smb2_request_verify_sizes(req, 0x09);
  63         if (!NT_STATUS_IS_OK(status)) {
  64                 return smbd_smb2_request_error(req, status);
  65         }
  66         inbody = SMBD_SMB2_IN_BODY_PTR(req);
  67
  68         if (xconn->protocol >= PROTOCOL_SMB3_11) {
  69                 in_flags = SVAL(inbody, 0x02);
  70         } else {
  71                 in_flags = 0;
  72         }
  73         in_path_offset = SVAL(inbody, 0x04);
  74         in_path_length = SVAL(inbody, 0x06);
  75
  76         if (in_path_offset != (SMB2_HDR_BODY + SMBD_SMB2_IN_BODY_LEN(req))) {
  77                 return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
  78         }
  79
  80         if (in_path_length > SMBD_SMB2_IN_DYN_LEN(req)) {
  81                 return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
  82         }
  83
  84         in_path_buffer.data = SMBD_SMB2_IN_DYN_PTR(req);
  85         in_path_buffer.length = in_path_length;
  86
  87         ok = convert_string_talloc(req, CH_UTF16, CH_UNIX,
  88                                    in_path_buffer.data,
  89                                    in_path_buffer.length,
  90                                    &in_path_string,
  91                                    &in_path_string_size);
  92         if (!ok) {
  93                 return smbd_smb2_request_error(req, NT_STATUS_ILLEGAL_CHARACTER);
  94         }
  95
  96         if (in_path_buffer.length == 0) {
  97                 in_path_string_size = 0;
  98         }
  99
 100         if (strlen(in_path_string) != in_path_string_size) {
 101                 return smbd_smb2_request_error(req, NT_STATUS_BAD_NETWORK_NAME);
 102         }
 103
 104         subreq = smbd_smb2_tree_connect_send(req,
 105                                              req->sconn->ev_ctx,
 106                                              req,
 107                                              in_flags,
 108                                              in_path_string);
 109         if (subreq == NULL) {
 110                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 111         }
 112         tevent_req_set_callback(subreq, smbd_smb2_request_tcon_done, req);
 113
 114         /*
 115          * Avoid sending a STATUS_PENDING message, it's very likely
 116          * the client won't expect that.
 117          */
 118         return smbd_smb2_request_pending_queue(req, subreq, 0);
 119 }
 120
 121 static void smbd_smb2_request_tcon_done(struct tevent_req *subreq)
 122 {
 123         struct smbd_smb2_request *req =
 124                 tevent_req_callback_data(subreq,
 125                 struct smbd_smb2_request);
 126         uint8_t *outhdr;
 127         DATA_BLOB outbody;
 128         uint8_t out_share_type = 0;
 129         uint32_t out_share_flags = 0;
 130         uint32_t out_capabilities = 0;
 131         uint32_t out_maximal_access = 0;
 132         uint32_t out_tree_id = 0;
 133         bool disconnect = false;
 134         NTSTATUS status;
 135         NTSTATUS error;
 136
 137         status = smbd_smb2_tree_connect_recv(subreq,
 138                                              &out_share_type,
 139                                              &out_share_flags,
 140                                              &out_capabilities,
 141                                              &out_maximal_access,
 142                                              &out_tree_id,
 143                                              &disconnect);
 144         TALLOC_FREE(subreq);
 145         if (!NT_STATUS_IS_OK(status)) {
 146                 if (disconnect) {
 147                         smbd_server_connection_terminate(req->xconn,
 148                                                          nt_errstr(status));
 149                         return;
 150                 }
 151                 error = smbd_smb2_request_error(req, status);
 152                 if (!NT_STATUS_IS_OK(error)) {
 153                         smbd_server_connection_terminate(req->xconn,
 154                                                          nt_errstr(error));
 155                         return;
 156                 }
 157                 return;
 158         }
 159
 160         outhdr = SMBD_SMB2_OUT_HDR_PTR(req);
 161
 162         outbody = smbd_smb2_generate_outbody(req, 0x10);
 163         if (outbody.data == NULL) {
 164                 error = smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 165                 if (!NT_STATUS_IS_OK(error)) {
 166                         smbd_server_connection_terminate(req->xconn,
 167                                                          nt_errstr(error));
 168                         return;
 169                 }
 170                 return;
 171         }
 172
 173         SIVAL(outhdr, SMB2_HDR_TID, out_tree_id);
 174
 175         SSVAL(outbody.data, 0x00, 0x10);        /* struct size */
 176         SCVAL(outbody.data, 0x02,
 177               out_share_type);                  /* share type */
 178         SCVAL(outbody.data, 0x03, 0);           /* reserved */
 179         SIVAL(outbody.data, 0x04,
 180               out_share_flags);                 /* share flags */
 181         SIVAL(outbody.data, 0x08,
 182               out_capabilities);                /* capabilities */
 183         SIVAL(outbody.data, 0x0C,
 184               out_maximal_access);              /* maximal access */
 185
 186         error = smbd_smb2_request_done(req, outbody, NULL);
 187         if (!NT_STATUS_IS_OK(error)) {
 188                 smbd_server_connection_terminate(req->xconn,
 189                                                  nt_errstr(error));
 190                 return;
 191         }
 192 }
 193
 194 static NTSTATUS smbd_smb2_tree_connect(struct smbd_smb2_request *req,
 195                                        const char *in_path,
 196                                        uint8_t *out_share_type,
 197                                        uint32_t *out_share_flags,
 198                                        uint32_t *out_capabilities,
 199                                        uint32_t *out_maximal_access,
 200                                        uint32_t *out_tree_id,
 201                                        bool *disconnect)
 202 {
 203         const struct loadparm_substitution *lp_sub =
 204                 loadparm_s3_global_substitution();
 205         struct smbXsrv_connection *conn = req->xconn;
 206         struct smbXsrv_session *session = req->session;
 207         struct auth_session_info *session_info =
 208                 session->global->auth_session_info;
 209         const char *share = in_path;
 210         char *service = NULL;
 211         int snum = -1;
 212         struct smbXsrv_tcon *tcon;
 213         NTTIME now = timeval_to_nttime(&req->request_time);
 214         connection_struct *compat_conn = NULL;
 215         NTSTATUS status;
 216         bool encryption_desired = req->session->global->encryption_flags & SMBXSRV_ENCRYPTION_DESIRED;
 217         bool encryption_required = req->session->global->encryption_flags & SMBXSRV_ENCRYPTION_REQUIRED;
 218         bool guest_session = false;
 219         bool require_signed_tcon = false;
 220         uint32_t session_global_id;
 221         char *share_name = NULL;
 222         uint8_t encryption_flags = 0;
 223
 224         *disconnect = false;
 225
 226         if (strncmp(share, "\\\\", 2) == 0) {
 227                 const char *p = strchr(share+2, '\\');
 228                 if (p) {
 229                         share = p + 1;
 230                 }
 231         }
 232
 233         DEBUG(10,("smbd_smb2_tree_connect: path[%s] share[%s]\n",
 234                   in_path, share));
 235
 236         if (security_session_user_level(session_info, NULL) < SECURITY_USER) {
 237                 guest_session = true;
 238         }
 239
 240         if (conn->protocol >= PROTOCOL_SMB3_11 && !guest_session) {
 241                 require_signed_tcon = true;
 242         }
 243
 244         if (require_signed_tcon && !req->do_encryption && !req->do_signing) {
 245                 DEBUG(1, ("smbd_smb2_tree_connect: reject request to share "
 246                           "[%s] as '%s\\%s' without encryption or signing. "
 247                           "Disconnecting.\n",
 248                           share,
 249                           req->session->global->auth_session_info->info->domain_name,
 250                           req->session->global->auth_session_info->info->account_name));
 251                 *disconnect = true;
 252                 return NT_STATUS_ACCESS_DENIED;
 253         }
 254
 255         service = talloc_strdup(talloc_tos(), share);
 256         if(!service) {
 257                 return NT_STATUS_NO_MEMORY;
 258         }
 259
 260         if (!strlower_m(service)) {
 261                 DEBUG(2, ("strlower_m %s failed\n", service));
 262                 return NT_STATUS_INVALID_PARAMETER;
 263         }
 264
 265         /* TODO: do more things... */
 266         if (strequal(service,HOMES_NAME)) {
 267                 if (session->homes_snum == -1) {
 268                         DEBUG(2, ("[homes] share not available for "
 269                                 "user %s because it was not found "
 270                                 "or created at session setup "
 271                                 "time\n",
 272                                 session_info->unix_info->unix_name));
 273                         return NT_STATUS_BAD_NETWORK_NAME;
 274                 }
 275                 snum = session->homes_snum;
 276         } else if ((session->homes_snum != -1)
 277                    && strequal(service,
 278                         lp_servicename(talloc_tos(), lp_sub, session->homes_snum))) {
 279                 snum = session->homes_snum;
 280         } else {
 281                 snum = find_service(talloc_tos(), service, &service);
 282                 if (!service) {
 283                         return NT_STATUS_NO_MEMORY;
 284                 }
 285         }
 286
 287         if (snum < 0) {
 288                 DEBUG(3,("smbd_smb2_tree_connect: couldn't find service %s\n",
 289                          service));
 290                 return NT_STATUS_BAD_NETWORK_NAME;
 291         }
 292
 293         /* Handle non-DFS clients attempting connections to msdfs proxy */
 294         if (lp_host_msdfs()) {
 295                 char *proxy = lp_msdfs_proxy(talloc_tos(), lp_sub, snum);
 296
 297                 if ((proxy != NULL) && (*proxy != '\0')) {
 298                         DBG_NOTICE("refusing connection to dfs proxy share "
 299                                    "'%s' (pointing to %s)\n",
 300                                    service,
 301                                    proxy);
 302                         TALLOC_FREE(proxy);
 303                         return NT_STATUS_BAD_NETWORK_NAME;
 304                 }
 305                 TALLOC_FREE(proxy);
 306         }
 307
 308         if ((lp_server_smb_encrypt(snum) >= SMB_ENCRYPTION_DESIRED) &&
 309             (conn->smb2.server.cipher != 0))
 310         {
 311                 encryption_desired = true;
 312         }
 313
 314         if (lp_server_smb_encrypt(snum) == SMB_ENCRYPTION_REQUIRED) {
 315                 encryption_desired = true;
 316                 encryption_required = true;
 317         }
 318
 319         if (guest_session && encryption_required) {
 320                 DEBUG(1,("reject guest as encryption is required for service %s\n",
 321                          service));
 322                 return NT_STATUS_ACCESS_DENIED;
 323         }
 324
 325         if (conn->smb2.server.cipher == 0) {
 326                 if (encryption_required) {
 327                         DEBUG(1,("reject tcon with dialect[0x%04X] "
 328                                  "as encryption is required for service %s\n",
 329                                  conn->smb2.server.dialect, service));
 330                         return NT_STATUS_ACCESS_DENIED;
 331                 }
 332         }
 333
 334         if (encryption_desired) {
 335                 encryption_flags |= SMBXSRV_ENCRYPTION_DESIRED;
 336         }
 337         if (encryption_required) {
 338                 encryption_flags |= SMBXSRV_ENCRYPTION_REQUIRED;
 339         }
 340
 341         session_global_id = req->session->global->session_global_id;
 342         share_name = lp_servicename(talloc_tos(), lp_sub, snum);
 343         if (share_name == NULL) {
 344                 return NT_STATUS_NO_MEMORY;
 345         }
 346
 347         if ((lp_max_connections(snum) > 0)
 348             && (count_current_connections(lp_const_servicename(snum), true) >=
 349                 lp_max_connections(snum))) {
 350
 351                 DBG_WARNING("Max connections (%d) exceeded for [%s][%s]\n",
 352                           lp_max_connections(snum),
 353                           lp_const_servicename(snum), share_name);
 354                 TALLOC_FREE(share_name);
 355                 return NT_STATUS_INSUFFICIENT_RESOURCES;
 356         }
 357
 358         /* create a new tcon as child of the session */
 359         status = smb2srv_tcon_create(req->session,
 360                                      session_global_id,
 361                                      encryption_flags,
 362                                      share_name,
 363                                      now, &tcon);
 364         TALLOC_FREE(share_name);
 365         if (!NT_STATUS_IS_OK(status)) {
 366                 return status;
 367         }
 368
 369         compat_conn = make_connection_smb2(req,
 370                                         tcon, snum,
 371                                         "???",
 372                                         &status);
 373         if (compat_conn == NULL) {
 374                 TALLOC_FREE(tcon);
 375                 return status;
 376         }
 377
 378         tcon->compat = talloc_move(tcon, &compat_conn);
 379
 380         tcon->status = NT_STATUS_OK;
 381
 382         if (IS_PRINT(tcon->compat)) {
 383                 *out_share_type = SMB2_SHARE_TYPE_PRINT;
 384         } else if (IS_IPC(tcon->compat)) {
 385                 *out_share_type = SMB2_SHARE_TYPE_PIPE;
 386         } else {
 387                 *out_share_type = SMB2_SHARE_TYPE_DISK;
 388         }
 389
 390         *out_share_flags = 0;
 391
 392         if (lp_msdfs_root(SNUM(tcon->compat)) && lp_host_msdfs()) {
 393                 *out_share_flags |= (SMB2_SHAREFLAG_DFS|SMB2_SHAREFLAG_DFS_ROOT);
 394                 *out_capabilities = SMB2_SHARE_CAP_DFS;
 395         } else {
 396                 *out_capabilities = 0;
 397         }
 398
 399         switch(lp_csc_policy(SNUM(tcon->compat))) {
 400         case CSC_POLICY_MANUAL:
 401                 break;
 402         case CSC_POLICY_DOCUMENTS:
 403                 *out_share_flags |= SMB2_SHAREFLAG_AUTO_CACHING;
 404                 break;
 405         case CSC_POLICY_PROGRAMS:
 406                 *out_share_flags |= SMB2_SHAREFLAG_VDO_CACHING;
 407                 break;
 408         case CSC_POLICY_DISABLE:
 409                 *out_share_flags |= SMB2_SHAREFLAG_NO_CACHING;
 410                 break;
 411         default:
 412                 break;
 413         }
 414
 415         if (lp_hide_unreadable(SNUM(tcon->compat)) ||
 416             lp_hide_unwriteable_files(SNUM(tcon->compat))) {
 417                 *out_share_flags |= SMB2_SHAREFLAG_ACCESS_BASED_DIRECTORY_ENUM;
 418         }
 419
 420         if (encryption_desired) {
 421                 *out_share_flags |= SMB2_SHAREFLAG_ENCRYPT_DATA;
 422         }
 423
 424         /*
 425          * For disk shares we can change the client
 426          * behavior on a cluster...
 427          */
 428         if (*out_share_type == SMB2_SHARE_TYPE_DISK) {
 429                 bool persistent = false; /* persistent handles not implemented yet */
 430                 bool cluster = lp_clustering();
 431                 bool asymmetric = false; /* shares are symmetric by default */
 432                 bool announce;
 433
 434                 /*
 435                  * In a ctdb cluster shares are continuously available,
 436                  * but windows clients mix this with the global persistent
 437                  * handles support.
 438                  *
 439                  * Persistent handles are requested if
 440                  * SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY is present
 441                  * even without SMB2_CAP_PERSISTENT_HANDLES.
 442                  *
 443                  * And SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY is
 444                  * required for SMB2_SHARE_CAP_CLUSTER to have
 445                  * an effect.
 446                  *
 447                  * So we better don't announce this by default
 448                  * until we support persistent handles.
 449                  */
 450                 announce = lp_parm_bool(SNUM(tcon->compat),
 451                                         "smb3 share cap",
 452                                         "CONTINUOUS AVAILABILITY",
 453                                         persistent);
 454                 if (announce) {
 455                         *out_capabilities |= SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY;
 456                 }
 457
 458                 /*
 459                  * ctdb clusters are always scale out...
 460                  */
 461                 announce = lp_parm_bool(SNUM(tcon->compat),
 462                                         "smb3 share cap",
 463                                         "SCALE OUT",
 464                                         cluster);
 465                 if (announce) {
 466                         *out_capabilities |= SMB2_SHARE_CAP_SCALEOUT;
 467                 }
 468
 469                 /*
 470                  * We support the witness service when ctdb is active
 471                  */
 472                 announce = lp_parm_bool(SNUM(tcon->compat),
 473                                         "smb3 share cap",
 474                                         "CLUSTER",
 475                                         cluster);
 476                 if (announce) {
 477                         *out_capabilities |= SMB2_SHARE_CAP_CLUSTER;
 478                 }
 479
 480                 /*
 481                  * Shares in a ctdb cluster are symmetric by design.
 482                  *
 483                  * But it might be useful to let the client use
 484                  * an isolated transport and witness registration for the
 485                  * specific share.
 486                  */
 487                 announce = lp_parm_bool(SNUM(tcon->compat),
 488                                         "smb3 share cap",
 489                                         "ASYMMETRIC",
 490                                         asymmetric);
 491                 if (announce) {
 492                         *out_capabilities |= SMB2_SHARE_CAP_ASYMMETRIC;
 493                 }
 494         }
 495
 496         *out_maximal_access = tcon->compat->share_access;
 497
 498         *out_tree_id = tcon->global->tcon_wire_id;
 499         req->last_tid = tcon->global->tcon_wire_id;
 500
 501         return NT_STATUS_OK;
 502 }
 503
 504 struct smbd_smb2_tree_connect_state {
 505         const char *in_path;
 506         uint8_t out_share_type;
 507         uint32_t out_share_flags;
 508         uint32_t out_capabilities;
 509         uint32_t out_maximal_access;
 510         uint32_t out_tree_id;
 511         bool disconnect;
 512 };
 513
 514 static struct tevent_req *smbd_smb2_tree_connect_send(TALLOC_CTX *mem_ctx,
 515                                         struct tevent_context *ev,
 516                                         struct smbd_smb2_request *smb2req,
 517                                         uint16_t in_flags,
 518                                         const char *in_path)
 519 {
 520         struct tevent_req *req;
 521         struct smbd_smb2_tree_connect_state *state;
 522         NTSTATUS status;
 523
 524         req = tevent_req_create(mem_ctx, &state,
 525                                 struct smbd_smb2_tree_connect_state);
 526         if (req == NULL) {
 527                 return NULL;
 528         }
 529         state->in_path = in_path;
 530
 531         status = smbd_smb2_tree_connect(smb2req,
 532                                         state->in_path,
 533                                         &state->out_share_type,
 534                                         &state->out_share_flags,
 535                                         &state->out_capabilities,
 536                                         &state->out_maximal_access,
 537                                         &state->out_tree_id,
 538                                         &state->disconnect);
 539         if (tevent_req_nterror(req, status)) {
 540                 return tevent_req_post(req, ev);
 541         }
 542
 543         tevent_req_done(req);
 544         return tevent_req_post(req, ev);
 545 }
 546
 547 static NTSTATUS smbd_smb2_tree_connect_recv(struct tevent_req *req,
 548                                             uint8_t *out_share_type,
 549                                             uint32_t *out_share_flags,
 550                                             uint32_t *out_capabilities,
 551                                             uint32_t *out_maximal_access,
 552                                             uint32_t *out_tree_id,
 553                                             bool *disconnect)
 554 {
 555         struct smbd_smb2_tree_connect_state *state =
 556                 tevent_req_data(req,
 557                 struct smbd_smb2_tree_connect_state);
 558         NTSTATUS status;
 559
 560         if (tevent_req_is_nterror(req, &status)) {
 561                 tevent_req_received(req);
 562                 return status;
 563         }
 564
 565         *out_share_type = state->out_share_type;
 566         *out_share_flags = state->out_share_flags;
 567         *out_capabilities = state->out_capabilities;
 568         *out_maximal_access = state->out_maximal_access;
 569         *out_tree_id = state->out_tree_id;
 570         *disconnect = state->disconnect;
 571
 572         tevent_req_received(req);
 573         return NT_STATUS_OK;
 574 }
 575
 576 static struct tevent_req *smbd_smb2_tdis_send(TALLOC_CTX *mem_ctx,
 577                                         struct tevent_context *ev,
 578                                         struct smbd_smb2_request *smb2req);
 579 static NTSTATUS smbd_smb2_tdis_recv(struct tevent_req *req);
 580 static void smbd_smb2_request_tdis_done(struct tevent_req *subreq);
 581
 582 NTSTATUS smbd_smb2_request_process_tdis(struct smbd_smb2_request *req)
 583 {
 584         NTSTATUS status;
 585         struct tevent_req *subreq = NULL;
 586
 587         status = smbd_smb2_request_verify_sizes(req, 0x04);
 588         if (!NT_STATUS_IS_OK(status)) {
 589                 return smbd_smb2_request_error(req, status);
 590         }
 591
 592         subreq = smbd_smb2_tdis_send(req, req->sconn->ev_ctx, req);
 593         if (subreq == NULL) {
 594                 return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
 595         }
 596         tevent_req_set_callback(subreq, smbd_smb2_request_tdis_done, req);
 597
 598         /*
 599          * Avoid sending a STATUS_PENDING message, it's very likely
 600          * the client won't expect that.
 601          */
 602         return smbd_smb2_request_pending_queue(req, subreq, 0);
 603 }
 604
 605 static void smbd_smb2_request_tdis_done(struct tevent_req *subreq)
 606 {
 607         struct smbd_smb2_request *smb2req =
 608                 tevent_req_callback_data(subreq,
 609                 struct smbd_smb2_request);
 610         DATA_BLOB outbody;
 611         NTSTATUS status;
 612         NTSTATUS error;
 613
 614         status = smbd_smb2_tdis_recv(subreq);
 615         TALLOC_FREE(subreq);
 616         if (!NT_STATUS_IS_OK(status)) {
 617                 error = smbd_smb2_request_error(smb2req, status);
 618                 if (!NT_STATUS_IS_OK(error)) {
 619                         smbd_server_connection_terminate(smb2req->xconn,
 620                                                         nt_errstr(error));
 621                         return;
 622                 }
 623                 return;
 624         }
 625
 626         outbody = smbd_smb2_generate_outbody(smb2req, 0x04);
 627         if (outbody.data == NULL) {
 628                 error = smbd_smb2_request_error(smb2req, NT_STATUS_NO_MEMORY);
 629                 if (!NT_STATUS_IS_OK(error)) {
 630                         smbd_server_connection_terminate(smb2req->xconn,
 631                                                         nt_errstr(error));
 632                         return;
 633                 }
 634                 return;
 635         }
 636
 637         SSVAL(outbody.data, 0x00, 0x04);        /* struct size */
 638         SSVAL(outbody.data, 0x02, 0);           /* reserved */
 639
 640         error = smbd_smb2_request_done(smb2req, outbody, NULL);
 641         if (!NT_STATUS_IS_OK(error)) {
 642                 smbd_server_connection_terminate(smb2req->xconn,
 643                                                 nt_errstr(error));
 644                 return;
 645         }
 646 }
 647
 648 struct smbd_smb2_tdis_state {
 649         struct smbd_smb2_request *smb2req;
 650         struct tevent_queue *wait_queue;
 651 };
 652
 653 static void smbd_smb2_tdis_wait_done(struct tevent_req *subreq);
 654
 655 static struct tevent_req *smbd_smb2_tdis_send(TALLOC_CTX *mem_ctx,
 656                                         struct tevent_context *ev,
 657                                         struct smbd_smb2_request *smb2req)
 658 {
 659         struct tevent_req *req;
 660         struct smbd_smb2_tdis_state *state;
 661         struct tevent_req *subreq;
 662         struct smbXsrv_connection *xconn = NULL;
 663
 664         req = tevent_req_create(mem_ctx, &state,
 665                         struct smbd_smb2_tdis_state);
 666         if (req == NULL) {
 667                 return NULL;
 668         }
 669         state->smb2req = smb2req;
 670
 671         state->wait_queue = tevent_queue_create(state, "tdis_wait_queue");
 672         if (tevent_req_nomem(state->wait_queue, req)) {
 673                 return tevent_req_post(req, ev);
 674         }
 675
 676         /*
 677          * Make sure that no new request will be able to use this tcon.
 678          */
 679         smb2req->tcon->status = NT_STATUS_NETWORK_NAME_DELETED;
 680
 681         xconn = smb2req->xconn->client->connections;
 682         for (; xconn != NULL; xconn = xconn->next) {
 683                 struct smbd_smb2_request *preq;
 684
 685                 for (preq = xconn->smb2.requests; preq != NULL; preq = preq->next) {
 686                         if (preq == smb2req) {
 687                                 /* Can't cancel current request. */
 688                                 continue;
 689                         }
 690                         if (preq->tcon != smb2req->tcon) {
 691                                 /* Request on different tcon. */
 692                                 continue;
 693                         }
 694
 695                         if (preq->subreq != NULL) {
 696                                 tevent_req_cancel(preq->subreq);
 697                         }
 698
 699                         /*
 700                          * Now wait until the request is finished.
 701                          *
 702                          * We don't set a callback, as we just want to block the
 703                          * wait queue and the talloc_free() of the request will
 704                          * remove the item from the wait queue.
 705                          */
 706                         subreq = tevent_queue_wait_send(preq, ev, state->wait_queue);
 707                         if (tevent_req_nomem(subreq, req)) {
 708                                 return tevent_req_post(req, ev);
 709                         }
 710                 }
 711         }
 712
 713         /*
 714          * Now we add our own waiter to the end of the queue,
 715          * this way we get notified when all pending requests are finished
 716          * and send to the socket.
 717          */
 718         subreq = tevent_queue_wait_send(state, ev, state->wait_queue);
 719         if (tevent_req_nomem(subreq, req)) {
 720                 return tevent_req_post(req, ev);
 721         }
 722         tevent_req_set_callback(subreq, smbd_smb2_tdis_wait_done, req);
 723
 724         return req;
 725 }
 726
 727 static void smbd_smb2_tdis_wait_done(struct tevent_req *subreq)
 728 {
 729         struct tevent_req *req = tevent_req_callback_data(
 730                 subreq, struct tevent_req);
 731         struct smbd_smb2_tdis_state *state = tevent_req_data(
 732                 req, struct smbd_smb2_tdis_state);
 733         NTSTATUS status;
 734
 735         tevent_queue_wait_recv(subreq);
 736         TALLOC_FREE(subreq);
 737
 738         /*
 739          * As we've been awoken, we may have changed
 740          * uid in the meantime. Ensure we're still
 741          * root (SMB2_OP_TDIS has .as_root = true).
 742          */
 743         change_to_root_user();
 744
 745         status = smbXsrv_tcon_disconnect(state->smb2req->tcon,
 746                                          state->smb2req->tcon->compat->vuid);
 747         if (tevent_req_nterror(req, status)) {
 748                 return;
 749         }
 750
 751         /* We did tear down the tcon. */
 752         TALLOC_FREE(state->smb2req->tcon);
 753         tevent_req_done(req);
 754 }
 755
 756 static NTSTATUS smbd_smb2_tdis_recv(struct tevent_req *req)
 757 {
 758         return tevent_req_simple_recv_ntstatus(req);
 759 }