source3/modules/vfs_streams_depot.c

   1 /*
   2  * Store streams in a separate subdirectory
   3  *
   4  * Copyright (C) Volker Lendecke, 2007
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include "includes.h"
  21 #include "smbd/smbd.h"
  22 #include "system/filesys.h"
  23
  24 #undef DBGC_CLASS
  25 #define DBGC_CLASS DBGC_VFS
  26
  27 /*
  28  * Excerpt from a mail from tridge:
  29  *
  30  * Volker, what I'm thinking of is this:
  31  * /mount-point/.streams/XX/YY/aaaa.bbbb/namedstream1
  32  * /mount-point/.streams/XX/YY/aaaa.bbbb/namedstream2
  33  *
  34  * where XX/YY is a 2 level hash based on the fsid/inode. "aaaa.bbbb"
  35  * is the fsid/inode. "namedstreamX" is a file named after the stream
  36  * name.
  37  */
  38
  39 static uint32_t hash_fn(DATA_BLOB key)
  40 {
  41         uint32_t value; /* Used to compute the hash value.  */
  42         uint32_t i;     /* Used to cycle through random values. */
  43
  44         /* Set the initial value from the key size. */
  45         for (value = 0x238F13AF * key.length, i=0; i < key.length; i++)
  46                 value = (value + (key.data[i] << (i*5 % 24)));
  47
  48         return (1103515243 * value + 12345);
  49 }
  50
  51 /*
  52  * With the hashing scheme based on the inode we need to protect against
  53  * streams showing up on files with re-used inodes. This can happen if we
  54  * create a stream directory from within Samba, and a local process or NFS
  55  * client deletes the file without deleting the streams directory. When the
  56  * inode is re-used and the stream directory is still around, the streams in
  57  * there would be show up as belonging to the new file.
  58  *
  59  * There are several workarounds for this, probably the easiest one is on
  60  * systems which have a true birthtime stat element: When the file has a later
  61  * birthtime than the streams directory, then we have to recreate the
  62  * directory.
  63  *
  64  * The other workaround is to somehow mark the file as generated by Samba with
  65  * something that a NFS client would not do. The closest one is a special
  66  * xattr value being set. On systems which do not support xattrs, it might be
  67  * an option to put in a special ACL entry for a non-existing group.
  68  */
  69
  70 static bool file_is_valid(vfs_handle_struct *handle, const char *path)
  71 {
  72         char buf;
  73
  74         DEBUG(10, ("file_is_valid (%s) called\n", path));
  75
  76         if (SMB_VFS_GETXATTR(handle->conn, path, SAMBA_XATTR_MARKER,
  77                                   &buf, sizeof(buf)) != sizeof(buf)) {
  78                 DEBUG(10, ("GETXATTR failed: %s\n", strerror(errno)));
  79                 return false;
  80         }
  81
  82         if (buf != '1') {
  83                 DEBUG(10, ("got wrong buffer content: '%c'\n", buf));
  84                 return false;
  85         }
  86
  87         return true;
  88 }
  89
  90 static bool mark_file_valid(vfs_handle_struct *handle, const char *path)
  91 {
  92         char buf = '1';
  93         int ret;
  94
  95         DEBUG(10, ("marking file %s as valid\n", path));
  96
  97         ret = SMB_VFS_SETXATTR(handle->conn, path, SAMBA_XATTR_MARKER,
  98                                     &buf, sizeof(buf), 0);
  99
 100         if (ret == -1) {
 101                 DEBUG(10, ("SETXATTR failed: %s\n", strerror(errno)));
 102                 return false;
 103         }
 104
 105         return true;
 106 }
 107
 108 /**
 109  * Given an smb_filename, determine the stream directory using the file's
 110  * base_name.
 111  */
 112 static char *stream_dir(vfs_handle_struct *handle,
 113                         const struct smb_filename *smb_fname,
 114                         const SMB_STRUCT_STAT *base_sbuf, bool create_it)
 115 {
 116         uint32_t hash;
 117         struct smb_filename *smb_fname_hash = NULL;
 118         char *result = NULL;
 119         SMB_STRUCT_STAT base_sbuf_tmp;
 120         uint8_t first, second;
 121         char *tmp;
 122         char *id_hex;
 123         struct file_id id;
 124         uint8 id_buf[16];
 125         bool check_valid;
 126         const char *rootdir;
 127
 128         check_valid = lp_parm_bool(SNUM(handle->conn),
 129                       "streams_depot", "check_valid", true);
 130
 131         tmp = talloc_asprintf(talloc_tos(), "%s/.streams", handle->conn->cwd);
 132
 133         if (tmp == NULL) {
 134                 errno = ENOMEM;
 135                 goto fail;
 136         }
 137
 138         rootdir = lp_parm_const_string(
 139                 SNUM(handle->conn), "streams_depot", "directory",
 140                 tmp);
 141
 142         /* Stat the base file if it hasn't already been done. */
 143         if (base_sbuf == NULL) {
 144                 struct smb_filename *smb_fname_base;
 145
 146                 smb_fname_base = synthetic_smb_fname(
 147                         talloc_tos(), smb_fname->base_name, NULL, NULL);
 148                 if (smb_fname_base == NULL) {
 149                         errno = ENOMEM;
 150                         goto fail;
 151                 }
 152                 if (SMB_VFS_NEXT_STAT(handle, smb_fname_base) == -1) {
 153                         TALLOC_FREE(smb_fname_base);
 154                         goto fail;
 155                 }
 156                 base_sbuf_tmp = smb_fname_base->st;
 157                 TALLOC_FREE(smb_fname_base);
 158         } else {
 159                 base_sbuf_tmp = *base_sbuf;
 160         }
 161
 162         id = SMB_VFS_FILE_ID_CREATE(handle->conn, &base_sbuf_tmp);
 163
 164         push_file_id_16((char *)id_buf, &id);
 165
 166         hash = hash_fn(data_blob_const(id_buf, sizeof(id_buf)));
 167
 168         first = hash & 0xff;
 169         second = (hash >> 8) & 0xff;
 170
 171         id_hex = hex_encode_talloc(talloc_tos(), id_buf, sizeof(id_buf));
 172
 173         if (id_hex == NULL) {
 174                 errno = ENOMEM;
 175                 goto fail;
 176         }
 177
 178         result = talloc_asprintf(talloc_tos(), "%s/%2.2X/%2.2X/%s", rootdir,
 179                                  first, second, id_hex);
 180
 181         TALLOC_FREE(id_hex);
 182
 183         if (result == NULL) {
 184                 errno = ENOMEM;
 185                 return NULL;
 186         }
 187
 188         smb_fname_hash = synthetic_smb_fname(talloc_tos(), result, NULL, NULL);
 189         if (smb_fname_hash == NULL) {
 190                 errno = ENOMEM;
 191                 goto fail;
 192         }
 193
 194         if (SMB_VFS_NEXT_STAT(handle, smb_fname_hash) == 0) {
 195                 struct smb_filename *smb_fname_new = NULL;
 196                 char *newname;
 197                 bool delete_lost;
 198
 199                 if (!S_ISDIR(smb_fname_hash->st.st_ex_mode)) {
 200                         errno = EINVAL;
 201                         goto fail;
 202                 }
 203
 204                 if (!check_valid ||
 205                     file_is_valid(handle, smb_fname->base_name)) {
 206                         return result;
 207                 }
 208
 209                 /*
 210                  * Someone has recreated a file under an existing inode
 211                  * without deleting the streams directory.
 212                  * Move it away or remove if streams_depot:delete_lost is set.
 213                  */
 214
 215         again:
 216                 delete_lost = lp_parm_bool(SNUM(handle->conn), "streams_depot",
 217                                            "delete_lost", false);
 218
 219                 if (delete_lost) {
 220                         DEBUG(3, ("Someone has recreated a file under an "
 221                               "existing inode. Removing: %s\n",
 222                               smb_fname_hash->base_name));
 223                         recursive_rmdir(talloc_tos(), handle->conn,
 224                                         smb_fname_hash);
 225                         SMB_VFS_NEXT_RMDIR(handle, smb_fname_hash->base_name);
 226                 } else {
 227                         newname = talloc_asprintf(talloc_tos(), "lost-%lu",
 228                                                   random());
 229                         DEBUG(3, ("Someone has recreated a file under an "
 230                               "existing inode. Renaming: %s to: %s\n",
 231                               smb_fname_hash->base_name,
 232                               newname));
 233                         if (newname == NULL) {
 234                                 errno = ENOMEM;
 235                                 goto fail;
 236                         }
 237
 238                         smb_fname_new = synthetic_smb_fname(
 239                                 talloc_tos(), newname, NULL, NULL);
 240                         TALLOC_FREE(newname);
 241                         if (smb_fname_new == NULL) {
 242                                 errno = ENOMEM;
 243                                 goto fail;
 244                         }
 245
 246                         if (SMB_VFS_NEXT_RENAME(handle, smb_fname_hash,
 247                                                 smb_fname_new) == -1) {
 248                                 TALLOC_FREE(smb_fname_new);
 249                                 if ((errno == EEXIST) || (errno == ENOTEMPTY)) {
 250                                         goto again;
 251                                 }
 252                                 goto fail;
 253                         }
 254
 255                         TALLOC_FREE(smb_fname_new);
 256                 }
 257         }
 258
 259         if (!create_it) {
 260                 errno = ENOENT;
 261                 goto fail;
 262         }
 263
 264         if ((SMB_VFS_NEXT_MKDIR(handle, rootdir, 0755) != 0)
 265             && (errno != EEXIST)) {
 266                 goto fail;
 267         }
 268
 269         tmp = talloc_asprintf(result, "%s/%2.2X", rootdir, first);
 270         if (tmp == NULL) {
 271                 errno = ENOMEM;
 272                 goto fail;
 273         }
 274
 275         if ((SMB_VFS_NEXT_MKDIR(handle, tmp, 0755) != 0)
 276             && (errno != EEXIST)) {
 277                 goto fail;
 278         }
 279
 280         TALLOC_FREE(tmp);
 281
 282         tmp = talloc_asprintf(result, "%s/%2.2X/%2.2X", rootdir, first,
 283                               second);
 284         if (tmp == NULL) {
 285                 errno = ENOMEM;
 286                 goto fail;
 287         }
 288
 289         if ((SMB_VFS_NEXT_MKDIR(handle, tmp, 0755) != 0)
 290             && (errno != EEXIST)) {
 291                 goto fail;
 292         }
 293
 294         TALLOC_FREE(tmp);
 295
 296         if ((SMB_VFS_NEXT_MKDIR(handle, result, 0755) != 0)
 297             && (errno != EEXIST)) {
 298                 goto fail;
 299         }
 300
 301         if (check_valid && !mark_file_valid(handle, smb_fname->base_name)) {
 302                 goto fail;
 303         }
 304
 305         TALLOC_FREE(smb_fname_hash);
 306         return result;
 307
 308  fail:
 309         TALLOC_FREE(smb_fname_hash);
 310         TALLOC_FREE(result);
 311         return NULL;
 312 }
 313 /**
 314  * Given a stream name, populate smb_fname_out with the actual location of the
 315  * stream.
 316  */
 317 static NTSTATUS stream_smb_fname(vfs_handle_struct *handle,
 318                                  const struct smb_filename *smb_fname,
 319                                  struct smb_filename **smb_fname_out,
 320                                  bool create_dir)
 321 {
 322         char *dirname, *stream_fname;
 323         const char *stype;
 324         NTSTATUS status;
 325
 326         *smb_fname_out = NULL;
 327
 328         stype = strchr_m(smb_fname->stream_name + 1, ':');
 329
 330         if (stype) {
 331                 if (strcasecmp_m(stype, ":$DATA") != 0) {
 332                         return NT_STATUS_INVALID_PARAMETER;
 333                 }
 334         }
 335
 336         dirname = stream_dir(handle, smb_fname, NULL, create_dir);
 337
 338         if (dirname == NULL) {
 339                 status = map_nt_error_from_unix(errno);
 340                 goto fail;
 341         }
 342
 343         stream_fname = talloc_asprintf(talloc_tos(), "%s/%s", dirname,
 344                                        smb_fname->stream_name);
 345
 346         if (stream_fname == NULL) {
 347                 status = NT_STATUS_NO_MEMORY;
 348                 goto fail;
 349         }
 350
 351         if (stype == NULL) {
 352                 /* Append an explicit stream type if one wasn't specified. */
 353                 stream_fname = talloc_asprintf(talloc_tos(), "%s:$DATA",
 354                                                stream_fname);
 355                 if (stream_fname == NULL) {
 356                         status = NT_STATUS_NO_MEMORY;
 357                         goto fail;
 358                 }
 359         } else {
 360                 /* Normalize the stream type to upercase. */
 361                 if (!strupper_m(strrchr_m(stream_fname, ':') + 1)) {
 362                         status = NT_STATUS_INVALID_PARAMETER;
 363                         goto fail;
 364                 }
 365         }
 366
 367         DEBUG(10, ("stream filename = %s\n", stream_fname));
 368
 369         /* Create an smb_filename with stream_name == NULL. */
 370         *smb_fname_out = synthetic_smb_fname(
 371                 talloc_tos(), stream_fname, NULL, NULL);
 372         if (*smb_fname_out == NULL) {
 373                 return NT_STATUS_NO_MEMORY;
 374         }
 375
 376         return NT_STATUS_OK;
 377
 378  fail:
 379         DEBUG(5, ("stream_name failed: %s\n", strerror(errno)));
 380         TALLOC_FREE(*smb_fname_out);
 381         return status;
 382 }
 383
 384 static NTSTATUS walk_streams(vfs_handle_struct *handle,
 385                              struct smb_filename *smb_fname_base,
 386                              char **pdirname,
 387                              bool (*fn)(const char *dirname,
 388                                         const char *dirent,
 389                                         void *private_data),
 390                              void *private_data)
 391 {
 392         char *dirname;
 393         DIR *dirhandle = NULL;
 394         const char *dirent = NULL;
 395         char *talloced = NULL;
 396
 397         dirname = stream_dir(handle, smb_fname_base, &smb_fname_base->st,
 398                              false);
 399
 400         if (dirname == NULL) {
 401                 if (errno == ENOENT) {
 402                         /*
 403                          * no stream around
 404                          */
 405                         return NT_STATUS_OK;
 406                 }
 407                 return map_nt_error_from_unix(errno);
 408         }
 409
 410         DEBUG(10, ("walk_streams: dirname=%s\n", dirname));
 411
 412         dirhandle = SMB_VFS_NEXT_OPENDIR(handle, dirname, NULL, 0);
 413
 414         if (dirhandle == NULL) {
 415                 TALLOC_FREE(dirname);
 416                 return map_nt_error_from_unix(errno);
 417         }
 418
 419         while ((dirent = vfs_readdirname(handle->conn, dirhandle, NULL,
 420                                          &talloced)) != NULL) {
 421
 422                 if (ISDOT(dirent) || ISDOTDOT(dirent)) {
 423                         TALLOC_FREE(talloced);
 424                         continue;
 425                 }
 426
 427                 DEBUG(10, ("walk_streams: dirent=%s\n", dirent));
 428
 429                 if (!fn(dirname, dirent, private_data)) {
 430                         TALLOC_FREE(talloced);
 431                         break;
 432                 }
 433                 TALLOC_FREE(talloced);
 434         }
 435
 436         SMB_VFS_NEXT_CLOSEDIR(handle, dirhandle);
 437
 438         if (pdirname != NULL) {
 439                 *pdirname = dirname;
 440         }
 441         else {
 442                 TALLOC_FREE(dirname);
 443         }
 444
 445         return NT_STATUS_OK;
 446 }
 447
 448 /**
 449  * Helper to stat/lstat the base file of an smb_fname. This will actually
 450  * fills in the stat struct in smb_filename.
 451  */
 452 static int streams_depot_stat_base(vfs_handle_struct *handle,
 453                                    struct smb_filename *smb_fname,
 454                                    bool follow_links)
 455 {
 456         char *tmp_stream_name;
 457         int result;
 458
 459         tmp_stream_name = smb_fname->stream_name;
 460         smb_fname->stream_name = NULL;
 461         if (follow_links) {
 462                 result = SMB_VFS_NEXT_STAT(handle, smb_fname);
 463         } else {
 464                 result = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
 465         }
 466         smb_fname->stream_name = tmp_stream_name;
 467         return result;
 468 }
 469
 470 static int streams_depot_stat(vfs_handle_struct *handle,
 471                               struct smb_filename *smb_fname)
 472 {
 473         struct smb_filename *smb_fname_stream = NULL;
 474         NTSTATUS status;
 475         int ret = -1;
 476
 477         DEBUG(10, ("streams_depot_stat called for [%s]\n",
 478                    smb_fname_str_dbg(smb_fname)));
 479
 480         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 481                 return SMB_VFS_NEXT_STAT(handle, smb_fname);
 482         }
 483
 484         /* If the default stream is requested, just stat the base file. */
 485         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 486                 return streams_depot_stat_base(handle, smb_fname, true);
 487         }
 488
 489         /* Stat the actual stream now. */
 490         status = stream_smb_fname(handle, smb_fname, &smb_fname_stream,
 491                                   false);
 492         if (!NT_STATUS_IS_OK(status)) {
 493                 ret = -1;
 494                 errno = map_errno_from_nt_status(status);
 495                 goto done;
 496         }
 497
 498         ret = SMB_VFS_NEXT_STAT(handle, smb_fname_stream);
 499
 500         /* Update the original smb_fname with the stat info. */
 501         smb_fname->st = smb_fname_stream->st;
 502  done:
 503         TALLOC_FREE(smb_fname_stream);
 504         return ret;
 505 }
 506
 507
 508
 509 static int streams_depot_lstat(vfs_handle_struct *handle,
 510                                struct smb_filename *smb_fname)
 511 {
 512         struct smb_filename *smb_fname_stream = NULL;
 513         NTSTATUS status;
 514         int ret = -1;
 515
 516         DEBUG(10, ("streams_depot_lstat called for [%s]\n",
 517                    smb_fname_str_dbg(smb_fname)));
 518
 519         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 520                 return SMB_VFS_NEXT_LSTAT(handle, smb_fname);
 521         }
 522
 523         /* If the default stream is requested, just stat the base file. */
 524         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 525                 return streams_depot_stat_base(handle, smb_fname, false);
 526         }
 527
 528         /* Stat the actual stream now. */
 529         status = stream_smb_fname(handle, smb_fname, &smb_fname_stream,
 530                                   false);
 531         if (!NT_STATUS_IS_OK(status)) {
 532                 ret = -1;
 533                 errno = map_errno_from_nt_status(status);
 534                 goto done;
 535         }
 536
 537         ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname_stream);
 538
 539  done:
 540         TALLOC_FREE(smb_fname_stream);
 541         return ret;
 542 }
 543
 544 static int streams_depot_open(vfs_handle_struct *handle,
 545                               struct smb_filename *smb_fname,
 546                               files_struct *fsp, int flags, mode_t mode)
 547 {
 548         struct smb_filename *smb_fname_stream = NULL;
 549         struct smb_filename *smb_fname_base = NULL;
 550         NTSTATUS status;
 551         int ret = -1;
 552
 553         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 554                 return SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
 555         }
 556
 557         /* If the default stream is requested, just open the base file. */
 558         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 559                 char *tmp_stream_name;
 560
 561                 tmp_stream_name = smb_fname->stream_name;
 562                 smb_fname->stream_name = NULL;
 563                 ret = SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
 564                 smb_fname->stream_name = tmp_stream_name;
 565
 566                 return ret;
 567         }
 568
 569         /* Ensure the base file still exists. */
 570         smb_fname_base = synthetic_smb_fname(
 571                 talloc_tos(), smb_fname->base_name, NULL, NULL);
 572         if (smb_fname_base == NULL) {
 573                 ret = -1;
 574                 errno = ENOMEM;
 575                 goto done;
 576         }
 577
 578         ret = SMB_VFS_NEXT_STAT(handle, smb_fname_base);
 579         if (ret == -1) {
 580                 goto done;
 581         }
 582
 583         /* Determine the stream name, and then open it. */
 584         status = stream_smb_fname(handle, smb_fname, &smb_fname_stream, true);
 585         if (!NT_STATUS_IS_OK(status)) {
 586                 ret = -1;
 587                 errno = map_errno_from_nt_status(status);
 588                 goto done;
 589         }
 590
 591         ret = SMB_VFS_NEXT_OPEN(handle, smb_fname_stream, fsp, flags, mode);
 592
 593  done:
 594         TALLOC_FREE(smb_fname_stream);
 595         TALLOC_FREE(smb_fname_base);
 596         return ret;
 597 }
 598
 599 static int streams_depot_unlink(vfs_handle_struct *handle,
 600                                 const struct smb_filename *smb_fname)
 601 {
 602         struct smb_filename *smb_fname_base = NULL;
 603         int ret = -1;
 604
 605         DEBUG(10, ("streams_depot_unlink called for %s\n",
 606                    smb_fname_str_dbg(smb_fname)));
 607
 608         /* If there is a valid stream, just unlink the stream and return. */
 609         if (is_ntfs_stream_smb_fname(smb_fname) &&
 610             !is_ntfs_default_stream_smb_fname(smb_fname)) {
 611                 struct smb_filename *smb_fname_stream = NULL;
 612                 NTSTATUS status;
 613
 614                 status = stream_smb_fname(handle, smb_fname, &smb_fname_stream,
 615                                           false);
 616                 if (!NT_STATUS_IS_OK(status)) {
 617                         errno = map_errno_from_nt_status(status);
 618                         return -1;
 619                 }
 620
 621                 ret = SMB_VFS_NEXT_UNLINK(handle, smb_fname_stream);
 622
 623                 TALLOC_FREE(smb_fname_stream);
 624                 return ret;
 625         }
 626
 627         /*
 628          * We potentially need to delete the per-inode streams directory
 629          */
 630
 631         smb_fname_base = synthetic_smb_fname(
 632                 talloc_tos(), smb_fname->base_name, NULL, NULL);
 633         if (smb_fname_base == NULL) {
 634                 errno = ENOMEM;
 635                 return -1;
 636         }
 637
 638         if (lp_posix_pathnames()) {
 639                 ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname_base);
 640         } else {
 641                 ret = SMB_VFS_NEXT_STAT(handle, smb_fname_base);
 642         }
 643
 644         if (ret == -1) {
 645                 TALLOC_FREE(smb_fname_base);
 646                 return -1;
 647         }
 648
 649         ret = SMB_VFS_NEXT_UNLINK(handle, smb_fname);
 650         if (ret == 0) {
 651                 char *dirname = stream_dir(handle, smb_fname_base,
 652                                            &smb_fname_base->st, false);
 653
 654                 if (dirname != NULL) {
 655                         SMB_VFS_NEXT_RMDIR(handle, dirname);
 656                 }
 657                 TALLOC_FREE(dirname);
 658         }
 659
 660         TALLOC_FREE(smb_fname_base);
 661         return ret;
 662 }
 663
 664 static int streams_depot_rmdir(vfs_handle_struct *handle, const char *path)
 665 {
 666         struct smb_filename *smb_fname_base = NULL;
 667         int ret = -1;
 668
 669         DEBUG(10, ("streams_depot_rmdir called for %s\n", path));
 670
 671         /*
 672          * We potentially need to delete the per-inode streams directory
 673          */
 674
 675         smb_fname_base = synthetic_smb_fname(talloc_tos(), path, NULL, NULL);
 676         if (smb_fname_base == NULL) {
 677                 errno = ENOMEM;
 678                 return -1;
 679         }
 680
 681         if (lp_posix_pathnames()) {
 682                 ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname_base);
 683         } else {
 684                 ret = SMB_VFS_NEXT_STAT(handle, smb_fname_base);
 685         }
 686
 687         if (ret == -1) {
 688                 TALLOC_FREE(smb_fname_base);
 689                 return -1;
 690         }
 691
 692         ret = SMB_VFS_NEXT_RMDIR(handle, path);
 693         if (ret == 0) {
 694                 char *dirname = stream_dir(handle, smb_fname_base,
 695                                            &smb_fname_base->st, false);
 696
 697                 if (dirname != NULL) {
 698                         SMB_VFS_NEXT_RMDIR(handle, dirname);
 699                 }
 700                 TALLOC_FREE(dirname);
 701         }
 702
 703         TALLOC_FREE(smb_fname_base);
 704         return ret;
 705 }
 706
 707 static int streams_depot_rename(vfs_handle_struct *handle,
 708                                 const struct smb_filename *smb_fname_src,
 709                                 const struct smb_filename *smb_fname_dst)
 710 {
 711         struct smb_filename *smb_fname_src_stream = NULL;
 712         struct smb_filename *smb_fname_dst_stream = NULL;
 713         bool src_is_stream, dst_is_stream;
 714         NTSTATUS status;
 715         int ret = -1;
 716
 717         DEBUG(10, ("streams_depot_rename called for %s => %s\n",
 718                    smb_fname_str_dbg(smb_fname_src),
 719                    smb_fname_str_dbg(smb_fname_dst)));
 720
 721         src_is_stream = is_ntfs_stream_smb_fname(smb_fname_src);
 722         dst_is_stream = is_ntfs_stream_smb_fname(smb_fname_dst);
 723
 724         if (!src_is_stream && !dst_is_stream) {
 725                 return SMB_VFS_NEXT_RENAME(handle, smb_fname_src,
 726                                            smb_fname_dst);
 727         }
 728
 729         /* for now don't allow renames from or to the default stream */
 730         if (is_ntfs_default_stream_smb_fname(smb_fname_src) ||
 731             is_ntfs_default_stream_smb_fname(smb_fname_dst)) {
 732                 errno = ENOSYS;
 733                 goto done;
 734         }
 735
 736         status = stream_smb_fname(handle, smb_fname_src, &smb_fname_src_stream,
 737                                   false);
 738         if (!NT_STATUS_IS_OK(status)) {
 739                 errno = map_errno_from_nt_status(status);
 740                 goto done;
 741         }
 742
 743         status = stream_smb_fname(handle, smb_fname_dst,
 744                                   &smb_fname_dst_stream, false);
 745         if (!NT_STATUS_IS_OK(status)) {
 746                 errno = map_errno_from_nt_status(status);
 747                 goto done;
 748         }
 749
 750         ret = SMB_VFS_NEXT_RENAME(handle, smb_fname_src_stream,
 751                                   smb_fname_dst_stream);
 752
 753 done:
 754         TALLOC_FREE(smb_fname_src_stream);
 755         TALLOC_FREE(smb_fname_dst_stream);
 756         return ret;
 757 }
 758
 759 static bool add_one_stream(TALLOC_CTX *mem_ctx, unsigned int *num_streams,
 760                            struct stream_struct **streams,
 761                            const char *name, off_t size,
 762                            off_t alloc_size)
 763 {
 764         struct stream_struct *tmp;
 765
 766         tmp = talloc_realloc(mem_ctx, *streams, struct stream_struct,
 767                                    (*num_streams)+1);
 768         if (tmp == NULL) {
 769                 return false;
 770         }
 771
 772         tmp[*num_streams].name = talloc_strdup(tmp, name);
 773         if (tmp[*num_streams].name == NULL) {
 774                 return false;
 775         }
 776
 777         tmp[*num_streams].size = size;
 778         tmp[*num_streams].alloc_size = alloc_size;
 779
 780         *streams = tmp;
 781         *num_streams += 1;
 782         return true;
 783 }
 784
 785 struct streaminfo_state {
 786         TALLOC_CTX *mem_ctx;
 787         vfs_handle_struct *handle;
 788         unsigned int num_streams;
 789         struct stream_struct *streams;
 790         NTSTATUS status;
 791 };
 792
 793 static bool collect_one_stream(const char *dirname,
 794                                const char *dirent,
 795                                void *private_data)
 796 {
 797         struct streaminfo_state *state =
 798                 (struct streaminfo_state *)private_data;
 799         struct smb_filename *smb_fname = NULL;
 800         char *sname = NULL;
 801         bool ret;
 802
 803         sname = talloc_asprintf(talloc_tos(), "%s/%s", dirname, dirent);
 804         if (sname == NULL) {
 805                 state->status = NT_STATUS_NO_MEMORY;
 806                 ret = false;
 807                 goto out;
 808         }
 809
 810         smb_fname = synthetic_smb_fname(talloc_tos(), sname, NULL, NULL);
 811         if (smb_fname == NULL) {
 812                 state->status = NT_STATUS_NO_MEMORY;
 813                 ret = false;
 814                 goto out;
 815         }
 816
 817         if (SMB_VFS_NEXT_STAT(state->handle, smb_fname) == -1) {
 818                 DEBUG(10, ("Could not stat %s: %s\n", sname,
 819                            strerror(errno)));
 820                 ret = true;
 821                 goto out;
 822         }
 823
 824         if (!add_one_stream(state->mem_ctx,
 825                             &state->num_streams, &state->streams,
 826                             dirent, smb_fname->st.st_ex_size,
 827                             SMB_VFS_GET_ALLOC_SIZE(state->handle->conn, NULL,
 828                                                    &smb_fname->st))) {
 829                 state->status = NT_STATUS_NO_MEMORY;
 830                 ret = false;
 831                 goto out;
 832         }
 833
 834         ret = true;
 835  out:
 836         TALLOC_FREE(sname);
 837         TALLOC_FREE(smb_fname);
 838         return ret;
 839 }
 840
 841 static NTSTATUS streams_depot_streaminfo(vfs_handle_struct *handle,
 842                                          struct files_struct *fsp,
 843                                          const char *fname,
 844                                          TALLOC_CTX *mem_ctx,
 845                                          unsigned int *pnum_streams,
 846                                          struct stream_struct **pstreams)
 847 {
 848         struct smb_filename *smb_fname_base;
 849         int ret;
 850         NTSTATUS status;
 851         struct streaminfo_state state;
 852
 853         smb_fname_base = synthetic_smb_fname(talloc_tos(), fname, NULL, NULL);
 854         if (smb_fname_base == NULL) {
 855                 return NT_STATUS_NO_MEMORY;
 856         }
 857
 858         if ((fsp != NULL) && (fsp->fh->fd != -1)) {
 859                 ret = SMB_VFS_NEXT_FSTAT(handle, fsp, &smb_fname_base->st);
 860         }
 861         else {
 862                 if (lp_posix_pathnames()) {
 863                         ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname_base);
 864                 } else {
 865                         ret = SMB_VFS_NEXT_STAT(handle, smb_fname_base);
 866                 }
 867         }
 868
 869         if (ret == -1) {
 870                 status = map_nt_error_from_unix(errno);
 871                 goto out;
 872         }
 873
 874         state.streams = *pstreams;
 875         state.num_streams = *pnum_streams;
 876         state.mem_ctx = mem_ctx;
 877         state.handle = handle;
 878         state.status = NT_STATUS_OK;
 879
 880         status = walk_streams(handle, smb_fname_base, NULL, collect_one_stream,
 881                               &state);
 882
 883         if (!NT_STATUS_IS_OK(status)) {
 884                 TALLOC_FREE(state.streams);
 885                 goto out;
 886         }
 887
 888         if (!NT_STATUS_IS_OK(state.status)) {
 889                 TALLOC_FREE(state.streams);
 890                 status = state.status;
 891                 goto out;
 892         }
 893
 894         *pnum_streams = state.num_streams;
 895         *pstreams = state.streams;
 896         status = SMB_VFS_NEXT_STREAMINFO(handle, fsp, fname, mem_ctx, pnum_streams, pstreams);
 897
 898  out:
 899         TALLOC_FREE(smb_fname_base);
 900         return status;
 901 }
 902
 903 static uint32_t streams_depot_fs_capabilities(struct vfs_handle_struct *handle,
 904                         enum timestamp_set_resolution *p_ts_res)
 905 {
 906         return SMB_VFS_NEXT_FS_CAPABILITIES(handle, p_ts_res) | FILE_NAMED_STREAMS;
 907 }
 908
 909 static struct vfs_fn_pointers vfs_streams_depot_fns = {
 910         .fs_capabilities_fn = streams_depot_fs_capabilities,
 911         .open_fn = streams_depot_open,
 912         .stat_fn = streams_depot_stat,
 913         .lstat_fn = streams_depot_lstat,
 914         .unlink_fn = streams_depot_unlink,
 915         .rmdir_fn = streams_depot_rmdir,
 916         .rename_fn = streams_depot_rename,
 917         .streaminfo_fn = streams_depot_streaminfo,
 918 };
 919
 920 NTSTATUS vfs_streams_depot_init(void);
 921 NTSTATUS vfs_streams_depot_init(void)
 922 {
 923         return smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "streams_depot",
 924                                 &vfs_streams_depot_fns);
 925 }