source3/smbd/dir.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Directory handling routines
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Jeremy Allison 2007
   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 "system/filesys.h"
  23 #include "smbd/smbd.h"
  24 #include "smbd/globals.h"
  25 #include "libcli/security/security.h"
  26 #include "lib/util/bitmap.h"
  27 #include "../lib/util/memcache.h"
  28 #include "../librpc/gen_ndr/open_files.h"
  29
  30 /*
  31    This module implements directory related functions for Samba.
  32 */
  33
  34 /* "Special" directory offsets. */
  35 #define END_OF_DIRECTORY_OFFSET ((long)-1)
  36 #define START_OF_DIRECTORY_OFFSET ((long)0)
  37 #define DOT_DOT_DIRECTORY_OFFSET ((long)0x80000000)
  38
  39 /* "Special" directory offsets in 32-bit wire format. */
  40 #define WIRE_END_OF_DIRECTORY_OFFSET ((uint32_t)0xFFFFFFFF)
  41 #define WIRE_START_OF_DIRECTORY_OFFSET ((uint32_t)0)
  42 #define WIRE_DOT_DOT_DIRECTORY_OFFSET ((uint32_t)0x80000000)
  43
  44 /* Make directory handle internals available. */
  45
  46 struct name_cache_entry {
  47         char *name;
  48         long offset;
  49 };
  50
  51 struct smb_Dir {
  52         connection_struct *conn;
  53         DIR *dir;
  54         long offset;
  55         struct smb_filename *dir_smb_fname;
  56         size_t name_cache_size;
  57         struct name_cache_entry *name_cache;
  58         unsigned int name_cache_index;
  59         unsigned int file_number;
  60         files_struct *fsp; /* Back pointer to containing fsp, only
  61                               set from OpenDir_fsp(). */
  62 };
  63
  64 struct dptr_struct {
  65         struct dptr_struct *next, *prev;
  66         int dnum;
  67         uint16_t spid;
  68         struct connection_struct *conn;
  69         struct smb_Dir *dir_hnd;
  70         bool expect_close;
  71         char *wcard;
  72         uint32_t attr;
  73         struct smb_filename *smb_dname;
  74         bool has_wild; /* Set to true if the wcard entry has MS wildcard characters in it. */
  75         bool did_stat; /* Optimisation for non-wcard searches. */
  76         bool priv;     /* Directory handle opened with privilege. */
  77         uint32_t counter;
  78         struct memcache *dptr_cache;
  79 };
  80
  81 static struct smb_Dir *OpenDir_fsp(TALLOC_CTX *mem_ctx, connection_struct *conn,
  82                         files_struct *fsp,
  83                         const char *mask,
  84                         uint32_t attr);
  85
  86 static void DirCacheAdd(struct smb_Dir *dir_hnd, const char *name, long offset);
  87
  88 static int smb_Dir_destructor(struct smb_Dir *dir_hnd);
  89
  90 #define INVALID_DPTR_KEY (-3)
  91
  92 /****************************************************************************
  93  Initialise the dir bitmap.
  94 ****************************************************************************/
  95
  96 bool init_dptrs(struct smbd_server_connection *sconn)
  97 {
  98         if (sconn->searches.dptr_bmap) {
  99                 return true;
 100         }
 101
 102         sconn->searches.dptr_bmap = bitmap_talloc(
 103                 sconn, MAX_DIRECTORY_HANDLES);
 104
 105         if (sconn->searches.dptr_bmap == NULL) {
 106                 return false;
 107         }
 108
 109         return true;
 110 }
 111
 112 /****************************************************************************
 113  Get the struct dptr_struct for a dir index.
 114 ****************************************************************************/
 115
 116 static struct dptr_struct *dptr_get(struct smbd_server_connection *sconn,
 117                                     int key)
 118 {
 119         struct dptr_struct *dptr;
 120
 121         for (dptr = sconn->searches.dirptrs; dptr != NULL; dptr = dptr->next) {
 122                 if(dptr->dnum != key) {
 123                         continue;
 124                 }
 125                 DLIST_PROMOTE(sconn->searches.dirptrs, dptr);
 126                 return dptr;
 127         }
 128         return(NULL);
 129 }
 130
 131 /****************************************************************************
 132  Get the dir path for a dir index.
 133 ****************************************************************************/
 134
 135 const char *dptr_path(struct smbd_server_connection *sconn, int key)
 136 {
 137         struct dptr_struct *dptr = dptr_get(sconn, key);
 138         if (dptr)
 139                 return(dptr->smb_dname->base_name);
 140         return(NULL);
 141 }
 142
 143 /****************************************************************************
 144  Get the dir wcard for a dir index.
 145 ****************************************************************************/
 146
 147 const char *dptr_wcard(struct smbd_server_connection *sconn, int key)
 148 {
 149         struct dptr_struct *dptr = dptr_get(sconn, key);
 150         if (dptr)
 151                 return(dptr->wcard);
 152         return(NULL);
 153 }
 154
 155 /****************************************************************************
 156  Get the dir attrib for a dir index.
 157 ****************************************************************************/
 158
 159 uint16_t dptr_attr(struct smbd_server_connection *sconn, int key)
 160 {
 161         struct dptr_struct *dptr = dptr_get(sconn, key);
 162         if (dptr)
 163                 return(dptr->attr);
 164         return(0);
 165 }
 166
 167 /****************************************************************************
 168  Close all dptrs for a cnum.
 169 ****************************************************************************/
 170
 171 void dptr_closecnum(connection_struct *conn)
 172 {
 173         struct dptr_struct *dptr, *next;
 174         struct smbd_server_connection *sconn = conn->sconn;
 175
 176         if (sconn == NULL) {
 177                 return;
 178         }
 179
 180         for(dptr = sconn->searches.dirptrs; dptr; dptr = next) {
 181                 next = dptr->next;
 182                 if (dptr->conn == conn) {
 183                         files_struct *fsp = dptr->dir_hnd->fsp;
 184                         close_file(NULL, fsp, NORMAL_CLOSE);
 185                         fsp = NULL;
 186                 }
 187         }
 188 }
 189
 190 /****************************************************************************
 191  Create a new dir ptr. If the flag old_handle is true then we must allocate
 192  from the bitmap range 0 - 255 as old SMBsearch directory handles are only
 193  one byte long. If old_handle is false we allocate from the range
 194  256 - MAX_DIRECTORY_HANDLES. We bias the number we return by 1 to ensure
 195  a directory handle is never zero.
 196  wcard must not be zero.
 197 ****************************************************************************/
 198
 199 NTSTATUS dptr_create(connection_struct *conn,
 200                 struct smb_request *req,
 201                 files_struct *fsp,
 202                 bool old_handle,
 203                 bool expect_close,
 204                 uint16_t spid,
 205                 const char *wcard,
 206                 bool wcard_has_wild,
 207                 uint32_t attr,
 208                 struct dptr_struct **dptr_ret)
 209 {
 210         struct smbd_server_connection *sconn = conn->sconn;
 211         struct dptr_struct *dptr = NULL;
 212         struct smb_Dir *dir_hnd;
 213
 214         DBG_INFO("dir=%s\n", fsp_str_dbg(fsp));
 215
 216         if (sconn == NULL) {
 217                 DEBUG(0,("dptr_create: called with fake connection_struct\n"));
 218                 return NT_STATUS_INTERNAL_ERROR;
 219         }
 220
 221         if (!wcard) {
 222                 return NT_STATUS_INVALID_PARAMETER;
 223         }
 224
 225         if (!(fsp->access_mask & SEC_DIR_LIST)) {
 226                 DBG_INFO("dptr_create: directory %s "
 227                         "not open for LIST access\n",
 228                         fsp_str_dbg(fsp));
 229                 return NT_STATUS_ACCESS_DENIED;
 230         }
 231         dir_hnd = OpenDir_fsp(NULL, conn, fsp, wcard, attr);
 232         if (!dir_hnd) {
 233                 return map_nt_error_from_unix(errno);
 234         }
 235
 236         dptr = talloc_zero(NULL, struct dptr_struct);
 237         if(!dptr) {
 238                 DEBUG(0,("talloc fail in dptr_create.\n"));
 239                 TALLOC_FREE(dir_hnd);
 240                 return NT_STATUS_NO_MEMORY;
 241         }
 242
 243         dptr->smb_dname = cp_smb_filename(dptr, fsp->fsp_name);
 244         if (dptr->smb_dname == NULL) {
 245                 TALLOC_FREE(dptr);
 246                 TALLOC_FREE(dir_hnd);
 247                 return NT_STATUS_NO_MEMORY;
 248         }
 249         dptr->conn = conn;
 250         dptr->dir_hnd = dir_hnd;
 251         dptr->spid = spid;
 252         dptr->expect_close = expect_close;
 253         dptr->wcard = talloc_strdup(dptr, wcard);
 254         if (!dptr->wcard) {
 255                 TALLOC_FREE(dptr);
 256                 TALLOC_FREE(dir_hnd);
 257                 return NT_STATUS_NO_MEMORY;
 258         }
 259         if ((req != NULL && req->posix_pathnames) ||
 260                         (wcard[0] == '.' && wcard[1] == 0)) {
 261                 dptr->has_wild = True;
 262         } else {
 263                 dptr->has_wild = wcard_has_wild;
 264         }
 265
 266         dptr->attr = attr;
 267
 268         if (sconn->using_smb2) {
 269                 goto done;
 270         }
 271
 272         if(old_handle) {
 273
 274                 /*
 275                  * This is an old-style SMBsearch request. Ensure the
 276                  * value we return will fit in the range 1-255.
 277                  */
 278
 279                 dptr->dnum = bitmap_find(sconn->searches.dptr_bmap, 0);
 280
 281                 if(dptr->dnum == -1 || dptr->dnum > 254) {
 282                         DBG_ERR("returned %d: Error - all old "
 283                                 "dirptrs in use ?\n",
 284                                 dptr->dnum);
 285                         TALLOC_FREE(dptr);
 286                         TALLOC_FREE(dir_hnd);
 287                         return NT_STATUS_TOO_MANY_OPENED_FILES;
 288                 }
 289         } else {
 290
 291                 /*
 292                  * This is a new-style trans2 request. Allocate from
 293                  * a range that will return 256 - MAX_DIRECTORY_HANDLES.
 294                  */
 295
 296                 dptr->dnum = bitmap_find(sconn->searches.dptr_bmap, 255);
 297
 298                 if(dptr->dnum == -1 || dptr->dnum < 255) {
 299                         DBG_ERR("returned %d: Error - all new "
 300                                 "dirptrs in use ?\n",
 301                                 dptr->dnum);
 302                         TALLOC_FREE(dptr);
 303                         TALLOC_FREE(dir_hnd);
 304                         return NT_STATUS_TOO_MANY_OPENED_FILES;
 305                 }
 306         }
 307
 308         bitmap_set(sconn->searches.dptr_bmap, dptr->dnum);
 309
 310         dptr->dnum += 1; /* Always bias the dnum by one - no zero dnums allowed. */
 311
 312         DLIST_ADD(sconn->searches.dirptrs, dptr);
 313
 314 done:
 315         DBG_INFO("creating new dirptr [%d] for path [%s], expect_close = %d\n",
 316                  dptr->dnum, fsp_str_dbg(fsp), expect_close);
 317
 318         *dptr_ret = dptr;
 319
 320         return NT_STATUS_OK;
 321 }
 322
 323
 324 /****************************************************************************
 325  Wrapper functions to access the lower level directory handles.
 326 ****************************************************************************/
 327
 328 void dptr_CloseDir(files_struct *fsp)
 329 {
 330         struct smbd_server_connection *sconn = NULL;
 331
 332         if (fsp->dptr == NULL) {
 333                 return;
 334         }
 335         sconn = fsp->dptr->conn->sconn;
 336
 337         /*
 338          * The destructor for the struct smb_Dir (fsp->dptr->dir_hnd)
 339          * now handles all resource deallocation.
 340          */
 341
 342         DBG_INFO("closing dptr key %d\n", fsp->dptr->dnum);
 343
 344         if (sconn != NULL && !sconn->using_smb2) {
 345                 DLIST_REMOVE(sconn->searches.dirptrs, fsp->dptr);
 346
 347                 /*
 348                  * Free the dnum in the bitmap. Remember the dnum value is
 349                  * always biased by one with respect to the bitmap.
 350                  */
 351
 352                 if (!bitmap_query(sconn->searches.dptr_bmap,
 353                                   fsp->dptr->dnum - 1))
 354                 {
 355                         DBG_ERR("closing dnum = %d and bitmap not set !\n",
 356                                 fsp->dptr->dnum);
 357                 }
 358
 359                 bitmap_clear(sconn->searches.dptr_bmap, fsp->dptr->dnum - 1);
 360         }
 361
 362         TALLOC_FREE(fsp->dptr->dir_hnd);
 363         TALLOC_FREE(fsp->dptr);
 364 }
 365
 366 void dptr_SeekDir(struct dptr_struct *dptr, long offset)
 367 {
 368         SeekDir(dptr->dir_hnd, offset);
 369 }
 370
 371 long dptr_TellDir(struct dptr_struct *dptr)
 372 {
 373         return TellDir(dptr->dir_hnd);
 374 }
 375
 376 bool dptr_has_wild(struct dptr_struct *dptr)
 377 {
 378         return dptr->has_wild;
 379 }
 380
 381 int dptr_dnum(struct dptr_struct *dptr)
 382 {
 383         return dptr->dnum;
 384 }
 385
 386 bool dptr_get_priv(struct dptr_struct *dptr)
 387 {
 388         return dptr->priv;
 389 }
 390
 391 void dptr_set_priv(struct dptr_struct *dptr)
 392 {
 393         dptr->priv = true;
 394 }
 395
 396 /****************************************************************************
 397  Return the next visible file name, skipping veto'd and invisible files.
 398 ****************************************************************************/
 399
 400 static const char *dptr_normal_ReadDirName(struct dptr_struct *dptr,
 401                                            long *poffset, SMB_STRUCT_STAT *pst,
 402                                            char **ptalloced)
 403 {
 404         /* Normal search for the next file. */
 405         const char *name;
 406         char *talloced = NULL;
 407
 408         while ((name = ReadDirName(dptr->dir_hnd, poffset, pst, &talloced))
 409                != NULL) {
 410                 if (is_visible_file(dptr->conn,
 411                                 dptr->dir_hnd,
 412                                 name,
 413                                 pst,
 414                                 true)) {
 415                         *ptalloced = talloced;
 416                         return name;
 417                 }
 418                 TALLOC_FREE(talloced);
 419         }
 420         return NULL;
 421 }
 422
 423 /****************************************************************************
 424  Return the next visible file name, skipping veto'd and invisible files.
 425 ****************************************************************************/
 426
 427 static char *dptr_ReadDirName(TALLOC_CTX *ctx,
 428                               struct dptr_struct *dptr,
 429                               long *poffset,
 430                               SMB_STRUCT_STAT *pst)
 431 {
 432         struct smb_filename smb_fname_base;
 433         char *name = NULL;
 434         const char *name_temp = NULL;
 435         char *talloced = NULL;
 436         char *pathreal = NULL;
 437         char *found_name = NULL;
 438         int ret;
 439
 440         SET_STAT_INVALID(*pst);
 441
 442         if (dptr->has_wild || dptr->did_stat) {
 443                 name_temp = dptr_normal_ReadDirName(dptr, poffset, pst,
 444                                                     &talloced);
 445                 if (name_temp == NULL) {
 446                         return NULL;
 447                 }
 448                 if (talloced != NULL) {
 449                         return talloc_move(ctx, &talloced);
 450                 }
 451                 return talloc_strdup(ctx, name_temp);
 452         }
 453
 454         /* If poffset is -1 then we know we returned this name before and we
 455          * have no wildcards. We're at the end of the directory. */
 456         if (*poffset == END_OF_DIRECTORY_OFFSET) {
 457                 return NULL;
 458         }
 459
 460         /* We know the stored wcard contains no wildcard characters.
 461          * See if we can match with a stat call. If we can't, then set
 462          * did_stat to true to ensure we only do this once and keep
 463          * searching. */
 464
 465         dptr->did_stat = true;
 466
 467         /* First check if it should be visible. */
 468         if (!is_visible_file(dptr->conn,
 469                         dptr->dir_hnd,
 470                         dptr->wcard,
 471                         pst,
 472                         true)) {
 473                 /* This only returns false if the file was found, but
 474                    is explicitly not visible. Set us to end of
 475                    directory, but return NULL as we know we can't ever
 476                    find it. */
 477                 goto ret;
 478         }
 479
 480         if (VALID_STAT(*pst)) {
 481                 name = talloc_strdup(ctx, dptr->wcard);
 482                 goto ret;
 483         }
 484
 485         pathreal = talloc_asprintf(ctx,
 486                                 "%s/%s",
 487                                 dptr->smb_dname->base_name,
 488                                 dptr->wcard);
 489         if (!pathreal)
 490                 return NULL;
 491
 492         /* Create an smb_filename with stream_name == NULL. */
 493         smb_fname_base = (struct smb_filename) { .base_name = pathreal };
 494
 495         if (SMB_VFS_STAT(dptr->conn, &smb_fname_base) == 0) {
 496                 *pst = smb_fname_base.st;
 497                 name = talloc_strdup(ctx, dptr->wcard);
 498                 goto clean;
 499         } else {
 500                 /* If we get any other error than ENOENT or ENOTDIR
 501                    then the file exists we just can't stat it. */
 502                 if (errno != ENOENT && errno != ENOTDIR) {
 503                         name = talloc_strdup(ctx, dptr->wcard);
 504                         goto clean;
 505                 }
 506         }
 507
 508         /* Stat failed. We know this is authoratiative if we are
 509          * providing case sensitive semantics or the underlying
 510          * filesystem is case sensitive.
 511          */
 512         if (dptr->conn->case_sensitive ||
 513             !(dptr->conn->fs_capabilities & FILE_CASE_SENSITIVE_SEARCH))
 514         {
 515                 goto clean;
 516         }
 517
 518         /*
 519          * Try case-insensitive stat if the fs has the ability. This avoids
 520          * scanning the whole directory.
 521          */
 522         ret = SMB_VFS_GET_REAL_FILENAME(dptr->conn,
 523                                         dptr->smb_dname->base_name,
 524                                         dptr->wcard,
 525                                         ctx,
 526                                         &found_name);
 527         if (ret == 0) {
 528                 name = found_name;
 529                 goto clean;
 530         } else if (errno == ENOENT) {
 531                 /* The case-insensitive lookup was authoritative. */
 532                 goto clean;
 533         }
 534
 535         TALLOC_FREE(pathreal);
 536
 537         name_temp = dptr_normal_ReadDirName(dptr, poffset, pst, &talloced);
 538         if (name_temp == NULL) {
 539                 return NULL;
 540         }
 541         if (talloced != NULL) {
 542                 return talloc_move(ctx, &talloced);
 543         }
 544         return talloc_strdup(ctx, name_temp);
 545
 546 clean:
 547         TALLOC_FREE(pathreal);
 548 ret:
 549         /* We need to set the underlying dir_hnd offset to -1
 550          * also as this function is usually called with the
 551          * output from TellDir. */
 552         dptr->dir_hnd->offset = *poffset = END_OF_DIRECTORY_OFFSET;
 553         return name;
 554 }
 555
 556 /****************************************************************************
 557  Search for a file by name, skipping veto'ed and not visible files.
 558 ****************************************************************************/
 559
 560 bool dptr_SearchDir(struct dptr_struct *dptr, const char *name, long *poffset, SMB_STRUCT_STAT *pst)
 561 {
 562         SET_STAT_INVALID(*pst);
 563
 564         if (!dptr->has_wild && (dptr->dir_hnd->offset == END_OF_DIRECTORY_OFFSET)) {
 565                 /* This is a singleton directory and we're already at the end. */
 566                 *poffset = END_OF_DIRECTORY_OFFSET;
 567                 return False;
 568         }
 569
 570         return SearchDir(dptr->dir_hnd, name, poffset);
 571 }
 572
 573 /****************************************************************************
 574  Map a native directory offset to a 32-bit cookie.
 575 ****************************************************************************/
 576
 577 static uint32_t map_dir_offset_to_wire(struct dptr_struct *dptr, long offset)
 578 {
 579         DATA_BLOB key;
 580         DATA_BLOB val;
 581
 582         if (offset == END_OF_DIRECTORY_OFFSET) {
 583                 return WIRE_END_OF_DIRECTORY_OFFSET;
 584         } else if(offset == START_OF_DIRECTORY_OFFSET) {
 585                 return WIRE_START_OF_DIRECTORY_OFFSET;
 586         } else if (offset == DOT_DOT_DIRECTORY_OFFSET) {
 587                 return WIRE_DOT_DOT_DIRECTORY_OFFSET;
 588         }
 589         if (sizeof(long) == 4) {
 590                 /* 32-bit machine. We can cheat... */
 591                 return (uint32_t)offset;
 592         }
 593         if (dptr->dptr_cache == NULL) {
 594                 /* Lazy initialize cache. */
 595                 dptr->dptr_cache = memcache_init(dptr, 0);
 596                 if (dptr->dptr_cache == NULL) {
 597                         return WIRE_END_OF_DIRECTORY_OFFSET;
 598                 }
 599         } else {
 600                 /* Have we seen this offset before ? */
 601                 key.data = (void *)&offset;
 602                 key.length = sizeof(offset);
 603                 if (memcache_lookup(dptr->dptr_cache,
 604                                         SMB1_SEARCH_OFFSET_MAP,
 605                                         key,
 606                                         &val)) {
 607                         uint32_t wire_offset;
 608                         SMB_ASSERT(val.length == sizeof(wire_offset));
 609                         memcpy(&wire_offset, val.data, sizeof(wire_offset));
 610                         DEBUG(10,("found wire %u <-> offset %ld\n",
 611                                 (unsigned int)wire_offset,
 612                                 (long)offset));
 613                         return wire_offset;
 614                 }
 615         }
 616         /* Allocate a new wire cookie. */
 617         do {
 618                 dptr->counter++;
 619         } while (dptr->counter == WIRE_START_OF_DIRECTORY_OFFSET ||
 620                  dptr->counter == WIRE_END_OF_DIRECTORY_OFFSET ||
 621                  dptr->counter == WIRE_DOT_DOT_DIRECTORY_OFFSET);
 622         /* Store it in the cache. */
 623         key.data = (void *)&offset;
 624         key.length = sizeof(offset);
 625         val.data = (void *)&dptr->counter;
 626         val.length = sizeof(dptr->counter); /* MUST BE uint32_t ! */
 627         memcache_add(dptr->dptr_cache,
 628                         SMB1_SEARCH_OFFSET_MAP,
 629                         key,
 630                         val);
 631         /* And the reverse mapping for lookup from
 632            map_wire_to_dir_offset(). */
 633         memcache_add(dptr->dptr_cache,
 634                         SMB1_SEARCH_OFFSET_MAP,
 635                         val,
 636                         key);
 637         DEBUG(10,("stored wire %u <-> offset %ld\n",
 638                 (unsigned int)dptr->counter,
 639                 (long)offset));
 640         return dptr->counter;
 641 }
 642
 643 /****************************************************************************
 644  Fill the 5 byte server reserved dptr field.
 645 ****************************************************************************/
 646
 647 bool dptr_fill(struct smbd_server_connection *sconn,
 648                char *buf1,unsigned int key)
 649 {
 650         unsigned char *buf = (unsigned char *)buf1;
 651         struct dptr_struct *dptr = dptr_get(sconn, key);
 652         uint32_t wire_offset;
 653         if (!dptr) {
 654                 DEBUG(1,("filling null dirptr %d\n",key));
 655                 return(False);
 656         }
 657         wire_offset = map_dir_offset_to_wire(dptr,TellDir(dptr->dir_hnd));
 658         DEBUG(6,("fill on key %u dirptr 0x%lx now at %d\n",key,
 659                 (long)dptr->dir_hnd,(int)wire_offset));
 660         buf[0] = key;
 661         SIVAL(buf,1,wire_offset);
 662         return(True);
 663 }
 664
 665 /****************************************************************************
 666  Map a 32-bit wire cookie to a native directory offset.
 667 ****************************************************************************/
 668
 669 static long map_wire_to_dir_offset(struct dptr_struct *dptr, uint32_t wire_offset)
 670 {
 671         DATA_BLOB key;
 672         DATA_BLOB val;
 673
 674         if (wire_offset == WIRE_END_OF_DIRECTORY_OFFSET) {
 675                 return END_OF_DIRECTORY_OFFSET;
 676         } else if(wire_offset == WIRE_START_OF_DIRECTORY_OFFSET) {
 677                 return START_OF_DIRECTORY_OFFSET;
 678         } else if (wire_offset == WIRE_DOT_DOT_DIRECTORY_OFFSET) {
 679                 return DOT_DOT_DIRECTORY_OFFSET;
 680         }
 681         if (sizeof(long) == 4) {
 682                 /* 32-bit machine. We can cheat... */
 683                 return (long)wire_offset;
 684         }
 685         if (dptr->dptr_cache == NULL) {
 686                 /* Logic error, cache should be initialized. */
 687                 return END_OF_DIRECTORY_OFFSET;
 688         }
 689         key.data = (void *)&wire_offset;
 690         key.length = sizeof(wire_offset);
 691         if (memcache_lookup(dptr->dptr_cache,
 692                                 SMB1_SEARCH_OFFSET_MAP,
 693                                 key,
 694                                 &val)) {
 695                 /* Found mapping. */
 696                 long offset;
 697                 SMB_ASSERT(val.length == sizeof(offset));
 698                 memcpy(&offset, val.data, sizeof(offset));
 699                 DEBUG(10,("lookup wire %u <-> offset %ld\n",
 700                         (unsigned int)wire_offset,
 701                         (long)offset));
 702                 return offset;
 703         }
 704         return END_OF_DIRECTORY_OFFSET;
 705 }
 706
 707 /****************************************************************************
 708  Return the associated fsp and seek the dir_hnd on it it given the 5 byte
 709  server field.
 710 ****************************************************************************/
 711
 712 files_struct *dptr_fetch_fsp(struct smbd_server_connection *sconn,
 713                                char *buf, int *num)
 714 {
 715         unsigned int key = *(unsigned char *)buf;
 716         struct dptr_struct *dptr = dptr_get(sconn, key);
 717         uint32_t wire_offset;
 718         long seekoff;
 719
 720         if (dptr == NULL) {
 721                 DEBUG(3,("fetched null dirptr %d\n",key));
 722                 return(NULL);
 723         }
 724         *num = key;
 725         wire_offset = IVAL(buf,1);
 726         seekoff = map_wire_to_dir_offset(dptr, wire_offset);
 727         SeekDir(dptr->dir_hnd,seekoff);
 728         DEBUG(3,("fetching dirptr %d for path %s at offset %d\n",
 729                 key, dptr->smb_dname->base_name, (int)seekoff));
 730         return dptr->dir_hnd->fsp;
 731 }
 732
 733 /****************************************************************************
 734  Fetch the fsp associated with the dptr_num.
 735 ****************************************************************************/
 736
 737 files_struct *dptr_fetch_lanman2_fsp(struct smbd_server_connection *sconn,
 738                                        int dptr_num)
 739 {
 740         struct dptr_struct *dptr  = dptr_get(sconn, dptr_num);
 741         if (dptr == NULL) {
 742                 return NULL;
 743         }
 744         DBG_NOTICE("fetching dirptr %d for path %s\n",
 745                 dptr_num,
 746                 dptr->smb_dname->base_name);
 747         return dptr->dir_hnd->fsp;
 748 }
 749
 750 static bool mangle_mask_match(connection_struct *conn,
 751                 const char *filename,
 752                 const char *mask)
 753 {
 754         char mname[13];
 755
 756         if (!name_to_8_3(filename,mname,False,conn->params)) {
 757                 return False;
 758         }
 759         return mask_match_search(mname,mask,False);
 760 }
 761
 762 bool smbd_dirptr_get_entry(TALLOC_CTX *ctx,
 763                            struct dptr_struct *dirptr,
 764                            const char *mask,
 765                            uint32_t dirtype,
 766                            bool dont_descend,
 767                            bool ask_sharemode,
 768                            bool get_dosmode,
 769                            bool (*match_fn)(TALLOC_CTX *ctx,
 770                                             void *private_data,
 771                                             const char *dname,
 772                                             const char *mask,
 773                                             char **_fname),
 774                            bool (*mode_fn)(TALLOC_CTX *ctx,
 775                                            void *private_data,
 776                                            struct smb_filename *smb_fname,
 777                                            bool get_dosmode,
 778                                            uint32_t *_mode),
 779                            void *private_data,
 780                            char **_fname,
 781                            struct smb_filename **_smb_fname,
 782                            uint32_t *_mode,
 783                            long *_prev_offset)
 784 {
 785         connection_struct *conn = dirptr->conn;
 786         size_t slashlen;
 787         size_t pathlen;
 788         const char *dpath = dirptr->smb_dname->base_name;
 789         bool dirptr_path_is_dot = ISDOT(dpath);
 790
 791         *_smb_fname = NULL;
 792         *_mode = 0;
 793
 794         pathlen = strlen(dpath);
 795         slashlen = ( dpath[pathlen-1] != '/') ? 1 : 0;
 796
 797         while (true) {
 798                 long cur_offset;
 799                 long prev_offset;
 800                 SMB_STRUCT_STAT sbuf = { 0 };
 801                 char *dname = NULL;
 802                 bool isdots;
 803                 char *fname = NULL;
 804                 char *pathreal = NULL;
 805                 struct smb_filename smb_fname;
 806                 uint32_t mode = 0;
 807                 bool ok;
 808
 809                 cur_offset = dptr_TellDir(dirptr);
 810                 prev_offset = cur_offset;
 811                 dname = dptr_ReadDirName(ctx, dirptr, &cur_offset, &sbuf);
 812
 813                 DEBUG(6,("smbd_dirptr_get_entry: dirptr 0x%lx now at offset %ld\n",
 814                         (long)dirptr, cur_offset));
 815
 816                 if (dname == NULL) {
 817                         return false;
 818                 }
 819
 820                 isdots = (ISDOT(dname) || ISDOTDOT(dname));
 821                 if (dont_descend && !isdots) {
 822                         TALLOC_FREE(dname);
 823                         continue;
 824                 }
 825
 826                 /*
 827                  * fname may get mangled, dname is never mangled.
 828                  * Whenever we're accessing the filesystem we use
 829                  * pathreal which is composed from dname.
 830                  */
 831
 832                 ok = match_fn(ctx, private_data, dname, mask, &fname);
 833                 if (!ok) {
 834                         TALLOC_FREE(dname);
 835                         continue;
 836                 }
 837
 838                 /*
 839                  * This used to be
 840                  * pathreal = talloc_asprintf(ctx, "%s%s%s", dirptr->path,
 841                  *                            needslash?"/":"", dname);
 842                  * but this was measurably slower than doing the memcpy.
 843                  */
 844
 845                 pathreal = talloc_array(
 846                         ctx, char,
 847                         pathlen + slashlen + talloc_get_size(dname));
 848                 if (!pathreal) {
 849                         TALLOC_FREE(dname);
 850                         TALLOC_FREE(fname);
 851                         return false;
 852                 }
 853
 854                 /*
 855                  * We don't want to pass ./xxx to modules below us so don't
 856                  * add the path if it is just . by itself.
 857                  */
 858                 if (dirptr_path_is_dot) {
 859                         memcpy(pathreal, dname, talloc_get_size(dname));
 860                 } else {
 861                         memcpy(pathreal, dpath, pathlen);
 862                         pathreal[pathlen] = '/';
 863                         memcpy(pathreal + slashlen + pathlen, dname,
 864                                talloc_get_size(dname));
 865                 }
 866
 867                 /* Create smb_fname with NULL stream_name. */
 868                 smb_fname = (struct smb_filename) {
 869                         .base_name = pathreal, .st = sbuf
 870                 };
 871
 872                 ok = mode_fn(ctx, private_data, &smb_fname, get_dosmode, &mode);
 873                 if (!ok) {
 874                         TALLOC_FREE(dname);
 875                         TALLOC_FREE(fname);
 876                         TALLOC_FREE(pathreal);
 877                         continue;
 878                 }
 879
 880                 if (!dir_check_ftype(mode, dirtype)) {
 881                         DEBUG(5,("[%s] attribs 0x%x didn't match 0x%x\n",
 882                                 fname, (unsigned int)mode, (unsigned int)dirtype));
 883                         TALLOC_FREE(dname);
 884                         TALLOC_FREE(fname);
 885                         TALLOC_FREE(pathreal);
 886                         continue;
 887                 }
 888
 889                 if (ask_sharemode && !S_ISDIR(smb_fname.st.st_ex_mode)) {
 890                         struct timespec write_time_ts;
 891                         struct file_id fileid;
 892
 893                         fileid = vfs_file_id_from_sbuf(conn,
 894                                                        &smb_fname.st);
 895                         get_file_infos(fileid, 0, NULL, &write_time_ts);
 896                         if (!is_omit_timespec(&write_time_ts)) {
 897                                 update_stat_ex_mtime(&smb_fname.st,
 898                                                      write_time_ts);
 899                         }
 900                 }
 901
 902                 DEBUG(3,("smbd_dirptr_get_entry mask=[%s] found %s "
 903                         "fname=%s (%s)\n",
 904                         mask, smb_fname_str_dbg(&smb_fname),
 905                         dname, fname));
 906
 907                 if (!conn->sconn->using_smb2) {
 908                         /*
 909                          * The dircache is only needed for SMB1 because SMB1
 910                          * uses a name for the resume wheras SMB2 always
 911                          * continues from the next position (unless it's told to
 912                          * restart or close-and-reopen the listing).
 913                          */
 914                         DirCacheAdd(dirptr->dir_hnd, dname, cur_offset);
 915                 }
 916
 917                 TALLOC_FREE(dname);
 918
 919                 *_smb_fname = cp_smb_filename(ctx, &smb_fname);
 920                 TALLOC_FREE(pathreal);
 921                 if (*_smb_fname == NULL) {
 922                         return false;
 923                 }
 924                 *_fname = fname;
 925                 *_mode = mode;
 926                 *_prev_offset = prev_offset;
 927
 928                 return true;
 929         }
 930
 931         return false;
 932 }
 933
 934 /****************************************************************************
 935  Get an 8.3 directory entry.
 936 ****************************************************************************/
 937
 938 static bool smbd_dirptr_8_3_match_fn(TALLOC_CTX *ctx,
 939                                      void *private_data,
 940                                      const char *dname,
 941                                      const char *mask,
 942                                      char **_fname)
 943 {
 944         connection_struct *conn = (connection_struct *)private_data;
 945
 946         if ((strcmp(mask,"*.*") == 0) ||
 947             mask_match_search(dname, mask, false) ||
 948             mangle_mask_match(conn, dname, mask)) {
 949                 char mname[13];
 950                 const char *fname;
 951                 /*
 952                  * Ensure we can push the original name as UCS2. If
 953                  * not, then just don't return this name.
 954                  */
 955                 NTSTATUS status;
 956                 size_t ret_len = 0;
 957                 size_t len = (strlen(dname) + 2) * 4; /* Allow enough space. */
 958                 uint8_t *tmp = talloc_array(talloc_tos(),
 959                                         uint8_t,
 960                                         len);
 961
 962                 status = srvstr_push(NULL,
 963                         FLAGS2_UNICODE_STRINGS,
 964                         tmp,
 965                         dname,
 966                         len,
 967                         STR_TERMINATE,
 968                         &ret_len);
 969
 970                 TALLOC_FREE(tmp);
 971
 972                 if (!NT_STATUS_IS_OK(status)) {
 973                         return false;
 974                 }
 975
 976                 if (!mangle_is_8_3(dname, false, conn->params)) {
 977                         bool ok = name_to_8_3(dname, mname, false,
 978                                               conn->params);
 979                         if (!ok) {
 980                                 return false;
 981                         }
 982                         fname = mname;
 983                 } else {
 984                         fname = dname;
 985                 }
 986
 987                 *_fname = talloc_strdup(ctx, fname);
 988                 if (*_fname == NULL) {
 989                         return false;
 990                 }
 991
 992                 return true;
 993         }
 994
 995         return false;
 996 }
 997
 998 static bool smbd_dirptr_8_3_mode_fn(TALLOC_CTX *ctx,
 999                                     void *private_data,
1000                                     struct smb_filename *smb_fname,
1001                                     bool get_dosmode,
1002                                     uint32_t *_mode)
1003 {
1004         connection_struct *conn = (connection_struct *)private_data;
1005
1006         if (!VALID_STAT(smb_fname->st)) {
1007                 if ((SMB_VFS_STAT(conn, smb_fname)) != 0) {
1008                         DEBUG(5,("smbd_dirptr_8_3_mode_fn: "
1009                                  "Couldn't stat [%s]. Error "
1010                                  "= %s\n",
1011                                  smb_fname_str_dbg(smb_fname),
1012                                  strerror(errno)));
1013                         return false;
1014                 }
1015         }
1016
1017         *_mode = dos_mode(conn, smb_fname);
1018         return true;
1019 }
1020
1021 bool get_dir_entry(TALLOC_CTX *ctx,
1022                 struct dptr_struct *dirptr,
1023                 const char *mask,
1024                 uint32_t dirtype,
1025                 char **_fname,
1026                 off_t *_size,
1027                 uint32_t *_mode,
1028                 struct timespec *_date,
1029                 bool check_descend,
1030                 bool ask_sharemode)
1031 {
1032         connection_struct *conn = dirptr->conn;
1033         char *fname = NULL;
1034         struct smb_filename *smb_fname = NULL;
1035         uint32_t mode = 0;
1036         long prev_offset;
1037         bool ok;
1038
1039         ok = smbd_dirptr_get_entry(ctx,
1040                                    dirptr,
1041                                    mask,
1042                                    dirtype,
1043                                    check_descend,
1044                                    ask_sharemode,
1045                                    true,
1046                                    smbd_dirptr_8_3_match_fn,
1047                                    smbd_dirptr_8_3_mode_fn,
1048                                    conn,
1049                                    &fname,
1050                                    &smb_fname,
1051                                    &mode,
1052                                    &prev_offset);
1053         if (!ok) {
1054                 return false;
1055         }
1056
1057         *_fname = talloc_move(ctx, &fname);
1058         *_size = smb_fname->st.st_ex_size;
1059         *_mode = mode;
1060         *_date = smb_fname->st.st_ex_mtime;
1061         TALLOC_FREE(smb_fname);
1062         return true;
1063 }
1064
1065 /*******************************************************************
1066  Check to see if a user can read a file. This is only approximate,
1067  it is used as part of the "hide unreadable" option. Don't
1068  use it for anything security sensitive.
1069 ********************************************************************/
1070
1071 static bool user_can_read_file(connection_struct *conn,
1072                                struct smb_filename *smb_fname)
1073 {
1074         NTSTATUS status;
1075         uint32_t rejected_share_access = 0;
1076         uint32_t rejected_mask = 0;
1077         struct security_descriptor *sd = NULL;
1078         uint32_t access_mask = FILE_READ_DATA|
1079                                 FILE_READ_EA|
1080                                 FILE_READ_ATTRIBUTES|
1081                                 SEC_STD_READ_CONTROL;
1082
1083         /*
1084          * Never hide files from the root user.
1085          * We use (uid_t)0 here not sec_initial_uid()
1086          * as make test uses a single user context.
1087          */
1088
1089         if (get_current_uid(conn) == (uid_t)0) {
1090                 return True;
1091         }
1092
1093         /*
1094          * We can't directly use smbd_check_access_rights()
1095          * here, as this implicitly grants FILE_READ_ATTRIBUTES
1096          * which the Windows access-based-enumeration code
1097          * explicitly checks for on the file security descriptor.
1098          * See bug:
1099          *
1100          * https://bugzilla.samba.org/show_bug.cgi?id=10252
1101          *
1102          * and the smb2.acl2.ACCESSBASED test for details.
1103          */
1104
1105         rejected_share_access = access_mask & ~(conn->share_access);
1106         if (rejected_share_access) {
1107                 DEBUG(10, ("rejected share access 0x%x "
1108                         "on %s (0x%x)\n",
1109                         (unsigned int)access_mask,
1110                         smb_fname_str_dbg(smb_fname),
1111                         (unsigned int)rejected_share_access ));
1112                 return false;
1113         }
1114
1115         status = SMB_VFS_GET_NT_ACL(conn,
1116                         smb_fname,
1117                         (SECINFO_OWNER |
1118                          SECINFO_GROUP |
1119                          SECINFO_DACL),
1120                         talloc_tos(),
1121                         &sd);
1122
1123         if (!NT_STATUS_IS_OK(status)) {
1124                 DEBUG(10, ("Could not get acl "
1125                         "on %s: %s\n",
1126                         smb_fname_str_dbg(smb_fname),
1127                         nt_errstr(status)));
1128                 return false;
1129         }
1130
1131         status = se_file_access_check(sd,
1132                                 get_current_nttok(conn),
1133                                 false,
1134                                 access_mask,
1135                                 &rejected_mask);
1136
1137         TALLOC_FREE(sd);
1138
1139         if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED)) {
1140                 DEBUG(10,("rejected bits 0x%x read access for %s\n",
1141                         (unsigned int)rejected_mask,
1142                         smb_fname_str_dbg(smb_fname) ));
1143                 return false;
1144         }
1145         return true;
1146 }
1147
1148 /*******************************************************************
1149  Check to see if a user can write a file (and only files, we do not
1150  check dirs on this one). This is only approximate,
1151  it is used as part of the "hide unwriteable" option. Don't
1152  use it for anything security sensitive.
1153 ********************************************************************/
1154
1155 static bool user_can_write_file(connection_struct *conn,
1156                                 const struct smb_filename *smb_fname)
1157 {
1158         /*
1159          * Never hide files from the root user.
1160          * We use (uid_t)0 here not sec_initial_uid()
1161          * as make test uses a single user context.
1162          */
1163
1164         if (get_current_uid(conn) == (uid_t)0) {
1165                 return True;
1166         }
1167
1168         SMB_ASSERT(VALID_STAT(smb_fname->st));
1169
1170         /* Pseudo-open the file */
1171
1172         if(S_ISDIR(smb_fname->st.st_ex_mode)) {
1173                 return True;
1174         }
1175
1176         return can_write_to_file(conn, smb_fname);
1177 }
1178
1179 /*******************************************************************
1180   Is a file a "special" type ?
1181 ********************************************************************/
1182
1183 static bool file_is_special(connection_struct *conn,
1184                             const struct smb_filename *smb_fname)
1185 {
1186         /*
1187          * Never hide files from the root user.
1188          * We use (uid_t)0 here not sec_initial_uid()
1189          * as make test uses a single user context.
1190          */
1191
1192         if (get_current_uid(conn) == (uid_t)0) {
1193                 return False;
1194         }
1195
1196         SMB_ASSERT(VALID_STAT(smb_fname->st));
1197
1198         if (S_ISREG(smb_fname->st.st_ex_mode) ||
1199             S_ISDIR(smb_fname->st.st_ex_mode) ||
1200             S_ISLNK(smb_fname->st.st_ex_mode))
1201                 return False;
1202
1203         return True;
1204 }
1205
1206 /*******************************************************************
1207  Should the file be seen by the client?
1208  NOTE: A successful return is no guarantee of the file's existence.
1209 ********************************************************************/
1210
1211 bool is_visible_file(connection_struct *conn,
1212                 struct smb_Dir *dir_hnd,
1213                 const char *name,
1214                 SMB_STRUCT_STAT *pst,
1215                 bool use_veto)
1216 {
1217         bool hide_unreadable = lp_hide_unreadable(SNUM(conn));
1218         bool hide_unwriteable = lp_hide_unwriteable_files(SNUM(conn));
1219         bool hide_special = lp_hide_special_files(SNUM(conn));
1220         int hide_new_files_timeout = lp_hide_new_files_timeout(SNUM(conn));
1221         char *entry = NULL;
1222         struct smb_filename *dir_path = dir_hnd->fsp->fsp_name;
1223         struct smb_filename *smb_fname_base = NULL;
1224         bool ret = false;
1225
1226         if ((strcmp(".",name) == 0) || (strcmp("..",name) == 0)) {
1227                 return True; /* . and .. are always visible. */
1228         }
1229
1230         /* If it's a vetoed file, pretend it doesn't even exist */
1231         if (use_veto && IS_VETO_PATH(conn, name)) {
1232                 DEBUG(10,("is_visible_file: file %s is vetoed.\n", name ));
1233                 return False;
1234         }
1235
1236         if (hide_unreadable ||
1237             hide_unwriteable ||
1238             hide_special ||
1239             (hide_new_files_timeout != 0))
1240         {
1241                 entry = talloc_asprintf(talloc_tos(),
1242                                         "%s/%s",
1243                                         dir_path->base_name,
1244                                         name);
1245                 if (!entry) {
1246                         ret = false;
1247                         goto out;
1248                 }
1249
1250                 /* Create an smb_filename with stream_name == NULL. */
1251                 smb_fname_base = synthetic_smb_fname(talloc_tos(),
1252                                                 entry,
1253                                                 NULL,
1254                                                 pst,
1255                                                 0);
1256                 if (smb_fname_base == NULL) {
1257                         ret = false;
1258                         goto out;
1259                 }
1260
1261                 /* If the file name does not exist, there's no point checking
1262                  * the configuration options. We succeed, on the basis that the
1263                  * checks *might* have passed if the file was present.
1264                  */
1265                 if (!VALID_STAT(*pst)) {
1266                         if (SMB_VFS_STAT(conn, smb_fname_base) != 0) {
1267                                 ret = true;
1268                                 goto out;
1269                         }
1270                         *pst = smb_fname_base->st;
1271                 }
1272
1273                 /* Honour _hide unreadable_ option */
1274                 if (hide_unreadable &&
1275                     !user_can_read_file(conn, smb_fname_base)) {
1276                         DEBUG(10,("is_visible_file: file %s is unreadable.\n",
1277                                  entry ));
1278                         ret = false;
1279                         goto out;
1280                 }
1281                 /* Honour _hide unwriteable_ option */
1282                 if (hide_unwriteable && !user_can_write_file(conn,
1283                                                              smb_fname_base)) {
1284                         DEBUG(10,("is_visible_file: file %s is unwritable.\n",
1285                                  entry ));
1286                         ret = false;
1287                         goto out;
1288                 }
1289                 /* Honour _hide_special_ option */
1290                 if (hide_special && file_is_special(conn, smb_fname_base)) {
1291                         DEBUG(10,("is_visible_file: file %s is special.\n",
1292                                  entry ));
1293                         ret = false;
1294                         goto out;
1295                 }
1296
1297                 if (hide_new_files_timeout != 0) {
1298
1299                         double age = timespec_elapsed(
1300                                 &smb_fname_base->st.st_ex_mtime);
1301
1302                         if (age < (double)hide_new_files_timeout) {
1303                                 ret = false;
1304                                 goto out;
1305                         }
1306                 }
1307         }
1308
1309         ret = true;
1310  out:
1311         TALLOC_FREE(smb_fname_base);
1312         TALLOC_FREE(entry);
1313         return ret;
1314 }
1315
1316 static int smb_Dir_destructor(struct smb_Dir *dir_hnd)
1317 {
1318         files_struct *fsp = dir_hnd->fsp;
1319
1320         SMB_VFS_CLOSEDIR(dir_hnd->conn, dir_hnd->dir);
1321         fsp->fh->fd = -1;
1322         if (fsp->dptr != NULL) {
1323                 SMB_ASSERT(fsp->dptr->dir_hnd == dir_hnd);
1324                 fsp->dptr->dir_hnd = NULL;
1325         }
1326         dir_hnd->fsp = NULL;
1327         return 0;
1328 }
1329
1330 /*******************************************************************
1331  Open a directory.
1332 ********************************************************************/
1333
1334 static int smb_Dir_OpenDir_destructor(struct smb_Dir *dir_hnd)
1335 {
1336         files_struct *fsp = dir_hnd->fsp;
1337
1338         smb_Dir_destructor(dir_hnd);
1339         file_free(NULL, fsp);
1340         return 0;
1341 }
1342
1343 struct smb_Dir *OpenDir(TALLOC_CTX *mem_ctx,
1344                         connection_struct *conn,
1345                         const struct smb_filename *smb_dname,
1346                         const char *mask,
1347                         uint32_t attr)
1348 {
1349         struct files_struct *fsp = NULL;
1350         struct smb_Dir *dir_hnd = NULL;
1351         NTSTATUS status;
1352
1353         status = open_internal_dirfsp_at(conn, conn->cwd_fsp, smb_dname, &fsp);
1354         if (!NT_STATUS_IS_OK(status)) {
1355                 return NULL;
1356         }
1357
1358         dir_hnd = OpenDir_fsp(mem_ctx, conn, fsp, mask, attr);
1359         if (dir_hnd == NULL) {
1360                 return NULL;
1361         }
1362
1363         /*
1364          * This overwrites the destructor set by smb_Dir_OpenDir_destructor(),
1365          * but smb_Dir_OpenDir_destructor() calls the OpenDir_fsp() destructor.
1366          */
1367         talloc_set_destructor(dir_hnd, smb_Dir_OpenDir_destructor);
1368         return dir_hnd;
1369 }
1370
1371 /*******************************************************************
1372  Open a directory from an fsp.
1373 ********************************************************************/
1374
1375 static struct smb_Dir *OpenDir_fsp(TALLOC_CTX *mem_ctx, connection_struct *conn,
1376                         files_struct *fsp,
1377                         const char *mask,
1378                         uint32_t attr)
1379 {
1380         struct smb_Dir *dir_hnd = talloc_zero(mem_ctx, struct smb_Dir);
1381
1382         if (!dir_hnd) {
1383                 goto fail;
1384         }
1385
1386         if (!fsp->fsp_flags.is_directory) {
1387                 errno = EBADF;
1388                 goto fail;
1389         }
1390
1391         if (fsp->fh->fd == -1) {
1392                 errno = EBADF;
1393                 goto fail;
1394         }
1395
1396         dir_hnd->conn = conn;
1397
1398         if (!conn->sconn->using_smb2) {
1399                 /*
1400                  * The dircache is only needed for SMB1 because SMB1 uses a name
1401                  * for the resume wheras SMB2 always continues from the next
1402                  * position (unless it's told to restart or close-and-reopen the
1403                  * listing).
1404                  */
1405                 dir_hnd->name_cache_size =
1406                         lp_directory_name_cache_size(SNUM(conn));
1407         }
1408
1409         dir_hnd->dir_smb_fname = cp_smb_filename(dir_hnd, fsp->fsp_name);
1410         if (!dir_hnd->dir_smb_fname) {
1411                 errno = ENOMEM;
1412                 goto fail;
1413         }
1414
1415         dir_hnd->dir = SMB_VFS_FDOPENDIR(fsp, mask, attr);
1416         if (dir_hnd->dir == NULL) {
1417                 goto fail;
1418         }
1419         dir_hnd->fsp = fsp;
1420
1421         talloc_set_destructor(dir_hnd, smb_Dir_destructor);
1422
1423         return dir_hnd;
1424
1425   fail:
1426         TALLOC_FREE(dir_hnd);
1427         return NULL;
1428 }
1429
1430
1431 /*******************************************************************
1432  Read from a directory.
1433  Return directory entry, current offset, and optional stat information.
1434  Don't check for veto or invisible files.
1435 ********************************************************************/
1436
1437 const char *ReadDirName(struct smb_Dir *dir_hnd, long *poffset,
1438                         SMB_STRUCT_STAT *sbuf, char **ptalloced)
1439 {
1440         const char *n;
1441         char *talloced = NULL;
1442         connection_struct *conn = dir_hnd->conn;
1443
1444         /* Cheat to allow . and .. to be the first entries returned. */
1445         if (((*poffset == START_OF_DIRECTORY_OFFSET) ||
1446              (*poffset == DOT_DOT_DIRECTORY_OFFSET)) && (dir_hnd->file_number < 2))
1447         {
1448                 if (dir_hnd->file_number == 0) {
1449                         n = ".";
1450                         *poffset = dir_hnd->offset = START_OF_DIRECTORY_OFFSET;
1451                 } else {
1452                         n = "..";
1453                         *poffset = dir_hnd->offset = DOT_DOT_DIRECTORY_OFFSET;
1454                 }
1455                 dir_hnd->file_number++;
1456                 *ptalloced = NULL;
1457                 return n;
1458         }
1459
1460         if (*poffset == END_OF_DIRECTORY_OFFSET) {
1461                 *poffset = dir_hnd->offset = END_OF_DIRECTORY_OFFSET;
1462                 return NULL;
1463         }
1464
1465         /* A real offset, seek to it. */
1466         SeekDir(dir_hnd, *poffset);
1467
1468         while ((n = vfs_readdirname(conn, dir_hnd->dir, sbuf, &talloced))) {
1469                 /* Ignore . and .. - we've already returned them. */
1470                 if (*n == '.') {
1471                         if ((n[1] == '\0') || (n[1] == '.' && n[2] == '\0')) {
1472                                 TALLOC_FREE(talloced);
1473                                 continue;
1474                         }
1475                 }
1476                 *poffset = dir_hnd->offset = SMB_VFS_TELLDIR(conn, dir_hnd->dir);
1477                 *ptalloced = talloced;
1478                 dir_hnd->file_number++;
1479                 return n;
1480         }
1481         *poffset = dir_hnd->offset = END_OF_DIRECTORY_OFFSET;
1482         *ptalloced = NULL;
1483         return NULL;
1484 }
1485
1486 /*******************************************************************
1487  Rewind to the start.
1488 ********************************************************************/
1489
1490 void RewindDir(struct smb_Dir *dir_hnd, long *poffset)
1491 {
1492         SMB_VFS_REWINDDIR(dir_hnd->conn, dir_hnd->dir);
1493         dir_hnd->file_number = 0;
1494         dir_hnd->offset = START_OF_DIRECTORY_OFFSET;
1495         *poffset = START_OF_DIRECTORY_OFFSET;
1496 }
1497
1498 /*******************************************************************
1499  Seek a dir.
1500 ********************************************************************/
1501
1502 void SeekDir(struct smb_Dir *dirp, long offset)
1503 {
1504         if (offset != dirp->offset) {
1505                 if (offset == START_OF_DIRECTORY_OFFSET) {
1506                         RewindDir(dirp, &offset);
1507                         /*
1508                          * Ok we should really set the file number here
1509                          * to 1 to enable ".." to be returned next. Trouble
1510                          * is I'm worried about callers using SeekDir(dirp,0)
1511                          * as equivalent to RewindDir(). So leave this alone
1512                          * for now.
1513                          */
1514                 } else if  (offset == DOT_DOT_DIRECTORY_OFFSET) {
1515                         RewindDir(dirp, &offset);
1516                         /*
1517                          * Set the file number to 2 - we want to get the first
1518                          * real file entry (the one we return after "..")
1519                          * on the next ReadDir.
1520                          */
1521                         dirp->file_number = 2;
1522                 } else if (offset == END_OF_DIRECTORY_OFFSET) {
1523                         ; /* Don't seek in this case. */
1524                 } else {
1525                         SMB_VFS_SEEKDIR(dirp->conn, dirp->dir, offset);
1526                 }
1527                 dirp->offset = offset;
1528         }
1529 }
1530
1531 /*******************************************************************
1532  Tell a dir position.
1533 ********************************************************************/
1534
1535 long TellDir(struct smb_Dir *dir_hnd)
1536 {
1537         return(dir_hnd->offset);
1538 }
1539
1540 /*******************************************************************
1541  Add an entry into the dcache.
1542 ********************************************************************/
1543
1544 static void DirCacheAdd(struct smb_Dir *dir_hnd, const char *name, long offset)
1545 {
1546         struct name_cache_entry *e;
1547
1548         if (dir_hnd->name_cache_size == 0) {
1549                 return;
1550         }
1551
1552         if (dir_hnd->name_cache == NULL) {
1553                 dir_hnd->name_cache = talloc_zero_array(dir_hnd,
1554                                                 struct name_cache_entry,
1555                                                 dir_hnd->name_cache_size);
1556
1557                 if (dir_hnd->name_cache == NULL) {
1558                         return;
1559                 }
1560         }
1561
1562         dir_hnd->name_cache_index = (dir_hnd->name_cache_index+1) %
1563                                         dir_hnd->name_cache_size;
1564         e = &dir_hnd->name_cache[dir_hnd->name_cache_index];
1565         TALLOC_FREE(e->name);
1566         e->name = talloc_strdup(dir_hnd, name);
1567         e->offset = offset;
1568 }
1569
1570 /*******************************************************************
1571  Find an entry by name. Leave us at the offset after it.
1572  Don't check for veto or invisible files.
1573 ********************************************************************/
1574
1575 bool SearchDir(struct smb_Dir *dir_hnd, const char *name, long *poffset)
1576 {
1577         int i;
1578         const char *entry = NULL;
1579         char *talloced = NULL;
1580         connection_struct *conn = dir_hnd->conn;
1581
1582         /* Search back in the name cache. */
1583         if (dir_hnd->name_cache_size && dir_hnd->name_cache) {
1584                 for (i = dir_hnd->name_cache_index; i >= 0; i--) {
1585                         struct name_cache_entry *e = &dir_hnd->name_cache[i];
1586                         if (e->name && (conn->case_sensitive ? (strcmp(e->name, name) == 0) : strequal(e->name, name))) {
1587                                 *poffset = e->offset;
1588                                 SeekDir(dir_hnd, e->offset);
1589                                 return True;
1590                         }
1591                 }
1592                 for (i = dir_hnd->name_cache_size - 1;
1593                                 i > dir_hnd->name_cache_index; i--) {
1594                         struct name_cache_entry *e = &dir_hnd->name_cache[i];
1595                         if (e->name && (conn->case_sensitive ? (strcmp(e->name, name) == 0) : strequal(e->name, name))) {
1596                                 *poffset = e->offset;
1597                                 SeekDir(dir_hnd, e->offset);
1598                                 return True;
1599                         }
1600                 }
1601         }
1602
1603         /* Not found in the name cache. Rewind directory and start from scratch. */
1604         SMB_VFS_REWINDDIR(conn, dir_hnd->dir);
1605         dir_hnd->file_number = 0;
1606         *poffset = START_OF_DIRECTORY_OFFSET;
1607         while ((entry = ReadDirName(dir_hnd, poffset, NULL, &talloced))) {
1608                 if (conn->case_sensitive ? (strcmp(entry, name) == 0) : strequal(entry, name)) {
1609                         TALLOC_FREE(talloced);
1610                         return True;
1611                 }
1612                 TALLOC_FREE(talloced);
1613         }
1614         return False;
1615 }
1616
1617 struct files_below_forall_state {
1618         char *dirpath;
1619         size_t dirpath_len;
1620         int (*fn)(struct file_id fid, const struct share_mode_data *data,
1621                   void *private_data);
1622         void *private_data;
1623 };
1624
1625 static int files_below_forall_fn(struct file_id fid,
1626                                  const struct share_mode_data *data,
1627                                  void *private_data)
1628 {
1629         struct files_below_forall_state *state = private_data;
1630         char tmpbuf[PATH_MAX];
1631         char *fullpath, *to_free;
1632         size_t len;
1633
1634         len = full_path_tos(data->servicepath, data->base_name,
1635                             tmpbuf, sizeof(tmpbuf),
1636                             &fullpath, &to_free);
1637         if (len == -1) {
1638                 return 0;
1639         }
1640         if (state->dirpath_len >= len) {
1641                 /*
1642                  * Filter files above dirpath
1643                  */
1644                 goto out;
1645         }
1646         if (fullpath[state->dirpath_len] != '/') {
1647                 /*
1648                  * Filter file that don't have a path separator at the end of
1649                  * dirpath's length
1650                  */
1651                 goto out;
1652         }
1653
1654         if (memcmp(state->dirpath, fullpath, state->dirpath_len) != 0) {
1655                 /*
1656                  * Not a parent
1657                  */
1658                 goto out;
1659         }
1660
1661         TALLOC_FREE(to_free);
1662         return state->fn(fid, data, state->private_data);
1663
1664 out:
1665         TALLOC_FREE(to_free);
1666         return 0;
1667 }
1668
1669 static int files_below_forall(connection_struct *conn,
1670                               const struct smb_filename *dir_name,
1671                               int (*fn)(struct file_id fid,
1672                                         const struct share_mode_data *data,
1673                                         void *private_data),
1674                               void *private_data)
1675 {
1676         struct files_below_forall_state state = {
1677                         .fn = fn,
1678                         .private_data = private_data,
1679         };
1680         int ret;
1681         char tmpbuf[PATH_MAX];
1682         char *to_free;
1683
1684         state.dirpath_len = full_path_tos(conn->connectpath,
1685                                           dir_name->base_name,
1686                                           tmpbuf, sizeof(tmpbuf),
1687                                           &state.dirpath, &to_free);
1688         if (state.dirpath_len == -1) {
1689                 return -1;
1690         }
1691
1692         ret = share_mode_forall(files_below_forall_fn, &state);
1693         TALLOC_FREE(to_free);
1694         return ret;
1695 }
1696
1697 struct have_file_open_below_state {
1698         bool found_one;
1699 };
1700
1701 static int have_file_open_below_fn(struct file_id fid,
1702                                    const struct share_mode_data *data,
1703                                    void *private_data)
1704 {
1705         struct have_file_open_below_state *state = private_data;
1706         state->found_one = true;
1707         return 1;
1708 }
1709
1710 bool have_file_open_below(connection_struct *conn,
1711                                  const struct smb_filename *name)
1712 {
1713         struct have_file_open_below_state state = {
1714                 .found_one = false,
1715         };
1716         int ret;
1717
1718         if (!VALID_STAT(name->st)) {
1719                 return false;
1720         }
1721         if (!S_ISDIR(name->st.st_ex_mode)) {
1722                 return false;
1723         }
1724
1725         ret = files_below_forall(conn, name, have_file_open_below_fn, &state);
1726         if (ret == -1) {
1727                 return false;
1728         }
1729
1730         return state.found_one;
1731 }
1732
1733 /*****************************************************************
1734  Is this directory empty ?
1735 *****************************************************************/
1736
1737 NTSTATUS can_delete_directory_fsp(files_struct *fsp)
1738 {
1739         NTSTATUS status = NT_STATUS_OK;
1740         long dirpos = 0;
1741         const char *dname = NULL;
1742         char *talloced = NULL;
1743         SMB_STRUCT_STAT st;
1744         struct connection_struct *conn = fsp->conn;
1745         struct smb_Dir *dir_hnd = OpenDir(talloc_tos(),
1746                                         conn,
1747                                         fsp->fsp_name,
1748                                         NULL,
1749                                         0);
1750
1751         if (!dir_hnd) {
1752                 return map_nt_error_from_unix(errno);
1753         }
1754
1755         while ((dname = ReadDirName(dir_hnd, &dirpos, &st, &talloced))) {
1756                 /* Quick check for "." and ".." */
1757                 if (dname[0] == '.') {
1758                         if (!dname[1] || (dname[1] == '.' && !dname[2])) {
1759                                 TALLOC_FREE(talloced);
1760                                 continue;
1761                         }
1762                 }
1763
1764                 if (!is_visible_file(conn,
1765                                 dir_hnd,
1766                                 dname,
1767                                 &st,
1768                                 True)) {
1769                         TALLOC_FREE(talloced);
1770                         continue;
1771                 }
1772
1773                 DEBUG(10,("got name %s - can't delete\n",
1774                          dname ));
1775                 status = NT_STATUS_DIRECTORY_NOT_EMPTY;
1776                 break;
1777         }
1778         TALLOC_FREE(talloced);
1779         TALLOC_FREE(dir_hnd);
1780
1781         if (!NT_STATUS_IS_OK(status)) {
1782                 return status;
1783         }
1784
1785         if (!(fsp->posix_flags & FSP_POSIX_FLAGS_RENAME) &&
1786             lp_strict_rename(SNUM(conn)) &&
1787             have_file_open_below(fsp->conn, fsp->fsp_name))
1788         {
1789                 return NT_STATUS_ACCESS_DENIED;
1790         }
1791
1792         return NT_STATUS_OK;
1793 }