source3/modules/vfs_streams_xattr.c

   1 /*
   2  * Store streams in xattrs
   3  *
   4  * Copyright (C) Volker Lendecke, 2008
   5  *
   6  * Partly based on James Peach's Darwin module, which is
   7  *
   8  * Copyright (C) James Peach 2006-2007
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 3 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  22  */
  23
  24 #include "includes.h"
  25 #include "smbd/smbd.h"
  26 #include "system/filesys.h"
  27 #include "../lib/crypto/md5.h"
  28
  29 #undef DBGC_CLASS
  30 #define DBGC_CLASS DBGC_VFS
  31
  32 struct stream_io {
  33         char *base;
  34         char *xattr_name;
  35         void *fsp_name_ptr;
  36         files_struct *fsp;
  37         vfs_handle_struct *handle;
  38 };
  39
  40 static SMB_INO_T stream_inode(const SMB_STRUCT_STAT *sbuf, const char *sname)
  41 {
  42         struct MD5Context ctx;
  43         unsigned char hash[16];
  44         SMB_INO_T result;
  45         char *upper_sname;
  46
  47         DEBUG(10, ("stream_inode called for %lu/%lu [%s]\n",
  48                    (unsigned long)sbuf->st_ex_dev,
  49                    (unsigned long)sbuf->st_ex_ino, sname));
  50
  51         upper_sname = talloc_strdup_upper(talloc_tos(), sname);
  52         SMB_ASSERT(upper_sname != NULL);
  53
  54         MD5Init(&ctx);
  55         MD5Update(&ctx, (const unsigned char *)&(sbuf->st_ex_dev),
  56                   sizeof(sbuf->st_ex_dev));
  57         MD5Update(&ctx, (const unsigned char *)&(sbuf->st_ex_ino),
  58                   sizeof(sbuf->st_ex_ino));
  59         MD5Update(&ctx, (unsigned char *)upper_sname,
  60                   talloc_get_size(upper_sname)-1);
  61         MD5Final(hash, &ctx);
  62
  63         TALLOC_FREE(upper_sname);
  64
  65         /* Hopefully all the variation is in the lower 4 (or 8) bytes! */
  66         memcpy(&result, hash, sizeof(result));
  67
  68         DEBUG(10, ("stream_inode returns %lu\n", (unsigned long)result));
  69
  70         return result;
  71 }
  72
  73 static ssize_t get_xattr_size(connection_struct *conn,
  74                                 files_struct *fsp,
  75                                 const char *fname,
  76                                 const char *xattr_name)
  77 {
  78         NTSTATUS status;
  79         struct ea_struct ea;
  80         ssize_t result;
  81
  82         status = get_ea_value(talloc_tos(), conn, fsp, fname,
  83                               xattr_name, &ea);
  84
  85         if (!NT_STATUS_IS_OK(status)) {
  86                 return -1;
  87         }
  88
  89         result = ea.value.length-1;
  90         TALLOC_FREE(ea.value.data);
  91         return result;
  92 }
  93
  94 /**
  95  * Given a stream name, populate xattr_name with the xattr name to use for
  96  * accessing the stream.
  97  */
  98 static NTSTATUS streams_xattr_get_name(TALLOC_CTX *ctx,
  99                                        const char *stream_name,
 100                                        char **xattr_name)
 101 {
 102         char *stype;
 103
 104         stype = strchr_m(stream_name + 1, ':');
 105
 106         *xattr_name = talloc_asprintf(ctx, "%s%s",
 107                                       SAMBA_XATTR_DOSSTREAM_PREFIX,
 108                                       stream_name + 1);
 109         if (*xattr_name == NULL) {
 110                 return NT_STATUS_NO_MEMORY;
 111         }
 112
 113         if (stype == NULL) {
 114                 /* Append an explicit stream type if one wasn't specified. */
 115                 *xattr_name = talloc_asprintf(ctx, "%s:$DATA",
 116                                                *xattr_name);
 117                 if (*xattr_name == NULL) {
 118                         return NT_STATUS_NO_MEMORY;
 119                 }
 120         } else {
 121                 /* Normalize the stream type to upercase. */
 122                 if (!strupper_m(strrchr_m(*xattr_name, ':') + 1)) {
 123                         return NT_STATUS_INVALID_PARAMETER;
 124                 }
 125         }
 126
 127         DEBUG(10, ("xattr_name: %s, stream_name: %s\n", *xattr_name,
 128                    stream_name));
 129
 130         return NT_STATUS_OK;
 131 }
 132
 133 static bool streams_xattr_recheck(struct stream_io *sio)
 134 {
 135         NTSTATUS status;
 136         char *xattr_name = NULL;
 137
 138         if (sio->fsp->fsp_name == sio->fsp_name_ptr) {
 139                 return true;
 140         }
 141
 142         if (sio->fsp->fsp_name->stream_name == NULL) {
 143                 /* how can this happen */
 144                 errno = EINVAL;
 145                 return false;
 146         }
 147
 148         status = streams_xattr_get_name(talloc_tos(),
 149                                         sio->fsp->fsp_name->stream_name,
 150                                         &xattr_name);
 151         if (!NT_STATUS_IS_OK(status)) {
 152                 return false;
 153         }
 154
 155         TALLOC_FREE(sio->xattr_name);
 156         TALLOC_FREE(sio->base);
 157         sio->xattr_name = talloc_strdup(VFS_MEMCTX_FSP_EXTENSION(sio->handle, sio->fsp),
 158                                         xattr_name);
 159         sio->base = talloc_strdup(VFS_MEMCTX_FSP_EXTENSION(sio->handle, sio->fsp),
 160                                   sio->fsp->fsp_name->base_name);
 161         sio->fsp_name_ptr = sio->fsp->fsp_name;
 162
 163         TALLOC_FREE(xattr_name);
 164
 165         if ((sio->xattr_name == NULL) || (sio->base == NULL)) {
 166                 return false;
 167         }
 168
 169         return true;
 170 }
 171
 172 /**
 173  * Helper to stat/lstat the base file of an smb_fname.
 174  */
 175 static int streams_xattr_stat_base(vfs_handle_struct *handle,
 176                                    struct smb_filename *smb_fname,
 177                                    bool follow_links)
 178 {
 179         char *tmp_stream_name;
 180         int result;
 181
 182         tmp_stream_name = smb_fname->stream_name;
 183         smb_fname->stream_name = NULL;
 184         if (follow_links) {
 185                 result = SMB_VFS_NEXT_STAT(handle, smb_fname);
 186         } else {
 187                 result = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
 188         }
 189         smb_fname->stream_name = tmp_stream_name;
 190         return result;
 191 }
 192
 193 static int streams_xattr_fstat(vfs_handle_struct *handle, files_struct *fsp,
 194                                SMB_STRUCT_STAT *sbuf)
 195 {
 196         struct smb_filename *smb_fname_base = NULL;
 197         int ret = -1;
 198         struct stream_io *io = (struct stream_io *)
 199                 VFS_FETCH_FSP_EXTENSION(handle, fsp);
 200
 201         DEBUG(10, ("streams_xattr_fstat called for %d\n", fsp->fh->fd));
 202
 203         if (io == NULL || fsp->base_fsp == NULL) {
 204                 return SMB_VFS_NEXT_FSTAT(handle, fsp, sbuf);
 205         }
 206
 207         if (!streams_xattr_recheck(io)) {
 208                 return -1;
 209         }
 210
 211         /* Create an smb_filename with stream_name == NULL. */
 212         smb_fname_base = synthetic_smb_fname(talloc_tos(), io->base,
 213                                              NULL, NULL);
 214         if (smb_fname_base == NULL) {
 215                 errno = ENOMEM;
 216                 return -1;
 217         }
 218
 219         if (lp_posix_pathnames()) {
 220                 ret = SMB_VFS_LSTAT(handle->conn, smb_fname_base);
 221         } else {
 222                 ret = SMB_VFS_STAT(handle->conn, smb_fname_base);
 223         }
 224         *sbuf = smb_fname_base->st;
 225         TALLOC_FREE(smb_fname_base);
 226
 227         if (ret == -1) {
 228                 return -1;
 229         }
 230
 231         sbuf->st_ex_size = get_xattr_size(handle->conn, fsp->base_fsp,
 232                                         io->base, io->xattr_name);
 233         if (sbuf->st_ex_size == -1) {
 234                 return -1;
 235         }
 236
 237         DEBUG(10, ("sbuf->st_ex_size = %d\n", (int)sbuf->st_ex_size));
 238
 239         sbuf->st_ex_ino = stream_inode(sbuf, io->xattr_name);
 240         sbuf->st_ex_mode &= ~S_IFMT;
 241         sbuf->st_ex_mode |= S_IFREG;
 242         sbuf->st_ex_blocks = sbuf->st_ex_size / STAT_ST_BLOCKSIZE + 1;
 243
 244         return 0;
 245 }
 246
 247 static int streams_xattr_stat(vfs_handle_struct *handle,
 248                               struct smb_filename *smb_fname)
 249 {
 250         NTSTATUS status;
 251         int result = -1;
 252         char *xattr_name = NULL;
 253
 254         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 255                 return SMB_VFS_NEXT_STAT(handle, smb_fname);
 256         }
 257
 258         /* Note if lp_posix_paths() is true, we can never
 259          * get here as is_ntfs_stream_smb_fname() is
 260          * always false. So we never need worry about
 261          * not following links here. */
 262
 263         /* If the default stream is requested, just stat the base file. */
 264         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 265                 return streams_xattr_stat_base(handle, smb_fname, true);
 266         }
 267
 268         /* Populate the stat struct with info from the base file. */
 269         if (streams_xattr_stat_base(handle, smb_fname, true) == -1) {
 270                 return -1;
 271         }
 272
 273         /* Derive the xattr name to lookup. */
 274         status = streams_xattr_get_name(talloc_tos(), smb_fname->stream_name,
 275                                         &xattr_name);
 276         if (!NT_STATUS_IS_OK(status)) {
 277                 errno = map_errno_from_nt_status(status);
 278                 return -1;
 279         }
 280
 281         /* Augment the base file's stat information before returning. */
 282         smb_fname->st.st_ex_size = get_xattr_size(handle->conn, NULL,
 283                                                   smb_fname->base_name,
 284                                                   xattr_name);
 285         if (smb_fname->st.st_ex_size == -1) {
 286                 errno = ENOENT;
 287                 result = -1;
 288                 goto fail;
 289         }
 290
 291         smb_fname->st.st_ex_ino = stream_inode(&smb_fname->st, xattr_name);
 292         smb_fname->st.st_ex_mode &= ~S_IFMT;
 293         smb_fname->st.st_ex_mode |= S_IFREG;
 294         smb_fname->st.st_ex_blocks =
 295             smb_fname->st.st_ex_size / STAT_ST_BLOCKSIZE + 1;
 296
 297         result = 0;
 298  fail:
 299         TALLOC_FREE(xattr_name);
 300         return result;
 301 }
 302
 303 static int streams_xattr_lstat(vfs_handle_struct *handle,
 304                                struct smb_filename *smb_fname)
 305 {
 306         NTSTATUS status;
 307         int result = -1;
 308         char *xattr_name = NULL;
 309
 310         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 311                 return SMB_VFS_NEXT_LSTAT(handle, smb_fname);
 312         }
 313
 314         /* If the default stream is requested, just stat the base file. */
 315         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 316                 return streams_xattr_stat_base(handle, smb_fname, false);
 317         }
 318
 319         /* Populate the stat struct with info from the base file. */
 320         if (streams_xattr_stat_base(handle, smb_fname, false) == -1) {
 321                 return -1;
 322         }
 323
 324         /* Derive the xattr name to lookup. */
 325         status = streams_xattr_get_name(talloc_tos(), smb_fname->stream_name,
 326                                         &xattr_name);
 327         if (!NT_STATUS_IS_OK(status)) {
 328                 errno = map_errno_from_nt_status(status);
 329                 return -1;
 330         }
 331
 332         /* Augment the base file's stat information before returning. */
 333         smb_fname->st.st_ex_size = get_xattr_size(handle->conn, NULL,
 334                                                   smb_fname->base_name,
 335                                                   xattr_name);
 336         if (smb_fname->st.st_ex_size == -1) {
 337                 errno = ENOENT;
 338                 result = -1;
 339                 goto fail;
 340         }
 341
 342         smb_fname->st.st_ex_ino = stream_inode(&smb_fname->st, xattr_name);
 343         smb_fname->st.st_ex_mode &= ~S_IFMT;
 344         smb_fname->st.st_ex_mode |= S_IFREG;
 345         smb_fname->st.st_ex_blocks =
 346             smb_fname->st.st_ex_size / STAT_ST_BLOCKSIZE + 1;
 347
 348         result = 0;
 349
 350  fail:
 351         TALLOC_FREE(xattr_name);
 352         return result;
 353 }
 354
 355 static int streams_xattr_open(vfs_handle_struct *handle,
 356                               struct smb_filename *smb_fname,
 357                               files_struct *fsp, int flags, mode_t mode)
 358 {
 359         NTSTATUS status;
 360         struct smb_filename *smb_fname_base = NULL;
 361         struct stream_io *sio;
 362         struct ea_struct ea;
 363         char *xattr_name = NULL;
 364         int baseflags;
 365         int hostfd = -1;
 366
 367         DEBUG(10, ("streams_xattr_open called for %s\n",
 368                    smb_fname_str_dbg(smb_fname)));
 369
 370         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 371                 return SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
 372         }
 373
 374         /* If the default stream is requested, just open the base file. */
 375         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 376                 char *tmp_stream_name;
 377                 int ret;
 378
 379                 tmp_stream_name = smb_fname->stream_name;
 380                 smb_fname->stream_name = NULL;
 381
 382                 ret = SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
 383
 384                 smb_fname->stream_name = tmp_stream_name;
 385
 386                 return ret;
 387         }
 388
 389         status = streams_xattr_get_name(talloc_tos(), smb_fname->stream_name,
 390                                         &xattr_name);
 391         if (!NT_STATUS_IS_OK(status)) {
 392                 errno = map_errno_from_nt_status(status);
 393                 goto fail;
 394         }
 395
 396         /* Create an smb_filename with stream_name == NULL. */
 397         smb_fname_base = synthetic_smb_fname(
 398                 talloc_tos(), smb_fname->base_name, NULL, NULL);
 399         if (smb_fname_base == NULL) {
 400                 errno = ENOMEM;
 401                 goto fail;
 402         }
 403
 404         /*
 405          * We use baseflags to turn off nasty side-effects when opening the
 406          * underlying file.
 407          */
 408         baseflags = flags;
 409         baseflags &= ~O_TRUNC;
 410         baseflags &= ~O_EXCL;
 411         baseflags &= ~O_CREAT;
 412
 413         hostfd = SMB_VFS_OPEN(handle->conn, smb_fname_base, fsp,
 414                               baseflags, mode);
 415
 416         TALLOC_FREE(smb_fname_base);
 417
 418         /* It is legit to open a stream on a directory, but the base
 419          * fd has to be read-only.
 420          */
 421         if ((hostfd == -1) && (errno == EISDIR)) {
 422                 baseflags &= ~O_ACCMODE;
 423                 baseflags |= O_RDONLY;
 424                 hostfd = SMB_VFS_OPEN(handle->conn, smb_fname, fsp, baseflags,
 425                                       mode);
 426         }
 427
 428         if (hostfd == -1) {
 429                 goto fail;
 430         }
 431
 432         status = get_ea_value(talloc_tos(), handle->conn, NULL,
 433                               smb_fname->base_name, xattr_name, &ea);
 434
 435         DEBUG(10, ("get_ea_value returned %s\n", nt_errstr(status)));
 436
 437         if (!NT_STATUS_IS_OK(status)
 438             && !NT_STATUS_EQUAL(status, NT_STATUS_NOT_FOUND)) {
 439                 /*
 440                  * The base file is not there. This is an error even if we got
 441                  * O_CREAT, the higher levels should have created the base
 442                  * file for us.
 443                  */
 444                 DEBUG(10, ("streams_xattr_open: base file %s not around, "
 445                            "returning ENOENT\n", smb_fname->base_name));
 446                 errno = ENOENT;
 447                 goto fail;
 448         }
 449
 450         if (!NT_STATUS_IS_OK(status)) {
 451                 /*
 452                  * The attribute does not exist
 453                  */
 454
 455                 if (flags & O_CREAT) {
 456                         /*
 457                          * Darn, xattrs need at least 1 byte
 458                          */
 459                         char null = '\0';
 460
 461                         DEBUG(10, ("creating attribute %s on file %s\n",
 462                                    xattr_name, smb_fname->base_name));
 463
 464                         if (fsp->base_fsp->fh->fd != -1) {
 465                                 if (SMB_VFS_FSETXATTR(
 466                                         fsp->base_fsp, xattr_name,
 467                                         &null, sizeof(null),
 468                                         flags & O_EXCL ? XATTR_CREATE : 0) == -1) {
 469                                         goto fail;
 470                                 }
 471                         } else {
 472                                 if (SMB_VFS_SETXATTR(
 473                                         handle->conn, smb_fname->base_name,
 474                                         xattr_name, &null, sizeof(null),
 475                                         flags & O_EXCL ? XATTR_CREATE : 0) == -1) {
 476                                         goto fail;
 477                                 }
 478                         }
 479                 }
 480         }
 481
 482         if (flags & O_TRUNC) {
 483                 char null = '\0';
 484                 if (fsp->base_fsp->fh->fd != -1) {
 485                         if (SMB_VFS_FSETXATTR(
 486                                         fsp->base_fsp, xattr_name,
 487                                         &null, sizeof(null),
 488                                         flags & O_EXCL ? XATTR_CREATE : 0) == -1) {
 489                                 goto fail;
 490                         }
 491                 } else {
 492                         if (SMB_VFS_SETXATTR(
 493                                         handle->conn, smb_fname->base_name,
 494                                         xattr_name, &null, sizeof(null),
 495                                         flags & O_EXCL ? XATTR_CREATE : 0) == -1) {
 496                                 goto fail;
 497                         }
 498                 }
 499         }
 500
 501         sio = (struct stream_io *)VFS_ADD_FSP_EXTENSION(handle, fsp,
 502                                                         struct stream_io,
 503                                                         NULL);
 504         if (sio == NULL) {
 505                 errno = ENOMEM;
 506                 goto fail;
 507         }
 508
 509         sio->xattr_name = talloc_strdup(VFS_MEMCTX_FSP_EXTENSION(handle, fsp),
 510                                         xattr_name);
 511         sio->base = talloc_strdup(VFS_MEMCTX_FSP_EXTENSION(handle, fsp),
 512                                   smb_fname->base_name);
 513         sio->fsp_name_ptr = fsp->fsp_name;
 514         sio->handle = handle;
 515         sio->fsp = fsp;
 516
 517         if ((sio->xattr_name == NULL) || (sio->base == NULL)) {
 518                 errno = ENOMEM;
 519                 goto fail;
 520         }
 521
 522         return hostfd;
 523
 524  fail:
 525         if (hostfd >= 0) {
 526                 /*
 527                  * BUGBUGBUG -- we would need to call fd_close_posix here, but
 528                  * we don't have a full fsp yet
 529                  */
 530                 SMB_VFS_CLOSE(fsp);
 531         }
 532
 533         return -1;
 534 }
 535
 536 static int streams_xattr_unlink(vfs_handle_struct *handle,
 537                                 const struct smb_filename *smb_fname)
 538 {
 539         NTSTATUS status;
 540         int ret = -1;
 541         char *xattr_name;
 542
 543         if (!is_ntfs_stream_smb_fname(smb_fname)) {
 544                 return SMB_VFS_NEXT_UNLINK(handle, smb_fname);
 545         }
 546
 547         /* If the default stream is requested, just open the base file. */
 548         if (is_ntfs_default_stream_smb_fname(smb_fname)) {
 549                 struct smb_filename *smb_fname_base = NULL;
 550
 551                 smb_fname_base = cp_smb_filename(talloc_tos(), smb_fname);
 552                 if (smb_fname_base == NULL) {
 553                         errno = ENOMEM;
 554                         return -1;
 555                 }
 556
 557                 ret = SMB_VFS_NEXT_UNLINK(handle, smb_fname_base);
 558
 559                 TALLOC_FREE(smb_fname_base);
 560                 return ret;
 561         }
 562
 563         status = streams_xattr_get_name(talloc_tos(), smb_fname->stream_name,
 564                                         &xattr_name);
 565         if (!NT_STATUS_IS_OK(status)) {
 566                 errno = map_errno_from_nt_status(status);
 567                 goto fail;
 568         }
 569
 570         ret = SMB_VFS_REMOVEXATTR(handle->conn, smb_fname->base_name, xattr_name);
 571
 572         if ((ret == -1) && (errno == ENOATTR)) {
 573                 errno = ENOENT;
 574                 goto fail;
 575         }
 576
 577         ret = 0;
 578
 579  fail:
 580         TALLOC_FREE(xattr_name);
 581         return ret;
 582 }
 583
 584 static int streams_xattr_rename(vfs_handle_struct *handle,
 585                                 const struct smb_filename *smb_fname_src,
 586                                 const struct smb_filename *smb_fname_dst)
 587 {
 588         NTSTATUS status;
 589         int ret = -1;
 590         char *src_xattr_name = NULL;
 591         char *dst_xattr_name = NULL;
 592         bool src_is_stream, dst_is_stream;
 593         ssize_t oret;
 594         ssize_t nret;
 595         struct ea_struct ea;
 596
 597         src_is_stream = is_ntfs_stream_smb_fname(smb_fname_src);
 598         dst_is_stream = is_ntfs_stream_smb_fname(smb_fname_dst);
 599
 600         if (!src_is_stream && !dst_is_stream) {
 601                 return SMB_VFS_NEXT_RENAME(handle, smb_fname_src,
 602                                            smb_fname_dst);
 603         }
 604
 605         /* For now don't allow renames from or to the default stream. */
 606         if (is_ntfs_default_stream_smb_fname(smb_fname_src) ||
 607             is_ntfs_default_stream_smb_fname(smb_fname_dst)) {
 608                 errno = ENOSYS;
 609                 goto done;
 610         }
 611
 612         /* Don't rename if the streams are identical. */
 613         if (strcasecmp_m(smb_fname_src->stream_name,
 614                        smb_fname_dst->stream_name) == 0) {
 615                 goto done;
 616         }
 617
 618         /* Get the xattr names. */
 619         status = streams_xattr_get_name(talloc_tos(),
 620                                         smb_fname_src->stream_name,
 621                                         &src_xattr_name);
 622         if (!NT_STATUS_IS_OK(status)) {
 623                 errno = map_errno_from_nt_status(status);
 624                 goto fail;
 625         }
 626         status = streams_xattr_get_name(talloc_tos(),
 627                                         smb_fname_dst->stream_name,
 628                                         &dst_xattr_name);
 629         if (!NT_STATUS_IS_OK(status)) {
 630                 errno = map_errno_from_nt_status(status);
 631                 goto fail;
 632         }
 633
 634         /* read the old stream */
 635         status = get_ea_value(talloc_tos(), handle->conn, NULL,
 636                               smb_fname_src->base_name, src_xattr_name, &ea);
 637         if (!NT_STATUS_IS_OK(status)) {
 638                 errno = ENOENT;
 639                 goto fail;
 640         }
 641
 642         /* (over)write the new stream */
 643         nret = SMB_VFS_SETXATTR(handle->conn, smb_fname_src->base_name,
 644                                 dst_xattr_name, ea.value.data, ea.value.length,
 645                                 0);
 646         if (nret < 0) {
 647                 if (errno == ENOATTR) {
 648                         errno = ENOENT;
 649                 }
 650                 goto fail;
 651         }
 652
 653         /* remove the old stream */
 654         oret = SMB_VFS_REMOVEXATTR(handle->conn, smb_fname_src->base_name,
 655                                    src_xattr_name);
 656         if (oret < 0) {
 657                 if (errno == ENOATTR) {
 658                         errno = ENOENT;
 659                 }
 660                 goto fail;
 661         }
 662
 663  done:
 664         errno = 0;
 665         ret = 0;
 666  fail:
 667         TALLOC_FREE(src_xattr_name);
 668         TALLOC_FREE(dst_xattr_name);
 669         return ret;
 670 }
 671
 672 static NTSTATUS walk_xattr_streams(connection_struct *conn, files_struct *fsp,
 673                                    const char *fname,
 674                                    bool (*fn)(struct ea_struct *ea,
 675                                               void *private_data),
 676                                    void *private_data)
 677 {
 678         NTSTATUS status;
 679         char **names;
 680         size_t i, num_names;
 681         size_t prefix_len = strlen(SAMBA_XATTR_DOSSTREAM_PREFIX);
 682
 683         status = get_ea_names_from_file(talloc_tos(), conn, fsp, fname,
 684                                         &names, &num_names);
 685         if (!NT_STATUS_IS_OK(status)) {
 686                 return status;
 687         }
 688
 689         for (i=0; i<num_names; i++) {
 690                 struct ea_struct ea;
 691
 692                 if (strncmp(names[i], SAMBA_XATTR_DOSSTREAM_PREFIX,
 693                             prefix_len) != 0) {
 694                         continue;
 695                 }
 696
 697                 status = get_ea_value(names, conn, fsp, fname, names[i], &ea);
 698                 if (!NT_STATUS_IS_OK(status)) {
 699                         DEBUG(10, ("Could not get ea %s for file %s: %s\n",
 700                                    names[i], fname, nt_errstr(status)));
 701                         continue;
 702                 }
 703
 704                 ea.name = talloc_asprintf(ea.value.data, ":%s",
 705                                           names[i] + prefix_len);
 706                 if (ea.name == NULL) {
 707                         DEBUG(0, ("talloc failed\n"));
 708                         continue;
 709                 }
 710
 711                 if (!fn(&ea, private_data)) {
 712                         TALLOC_FREE(ea.value.data);
 713                         return NT_STATUS_OK;
 714                 }
 715
 716                 TALLOC_FREE(ea.value.data);
 717         }
 718
 719         TALLOC_FREE(names);
 720         return NT_STATUS_OK;
 721 }
 722
 723 static bool add_one_stream(TALLOC_CTX *mem_ctx, unsigned int *num_streams,
 724                            struct stream_struct **streams,
 725                            const char *name, off_t size,
 726                            off_t alloc_size)
 727 {
 728         struct stream_struct *tmp;
 729
 730         tmp = talloc_realloc(mem_ctx, *streams, struct stream_struct,
 731                                    (*num_streams)+1);
 732         if (tmp == NULL) {
 733                 return false;
 734         }
 735
 736         tmp[*num_streams].name = talloc_strdup(tmp, name);
 737         if (tmp[*num_streams].name == NULL) {
 738                 return false;
 739         }
 740
 741         tmp[*num_streams].size = size;
 742         tmp[*num_streams].alloc_size = alloc_size;
 743
 744         *streams = tmp;
 745         *num_streams += 1;
 746         return true;
 747 }
 748
 749 struct streaminfo_state {
 750         TALLOC_CTX *mem_ctx;
 751         vfs_handle_struct *handle;
 752         unsigned int num_streams;
 753         struct stream_struct *streams;
 754         NTSTATUS status;
 755 };
 756
 757 static bool collect_one_stream(struct ea_struct *ea, void *private_data)
 758 {
 759         struct streaminfo_state *state =
 760                 (struct streaminfo_state *)private_data;
 761
 762         if (!add_one_stream(state->mem_ctx,
 763                             &state->num_streams, &state->streams,
 764                             ea->name, ea->value.length-1,
 765                             smb_roundup(state->handle->conn,
 766                                         ea->value.length-1))) {
 767                 state->status = NT_STATUS_NO_MEMORY;
 768                 return false;
 769         }
 770
 771         return true;
 772 }
 773
 774 static NTSTATUS streams_xattr_streaminfo(vfs_handle_struct *handle,
 775                                          struct files_struct *fsp,
 776                                          const char *fname,
 777                                          TALLOC_CTX *mem_ctx,
 778                                          unsigned int *pnum_streams,
 779                                          struct stream_struct **pstreams)
 780 {
 781         SMB_STRUCT_STAT sbuf;
 782         int ret;
 783         NTSTATUS status;
 784         struct streaminfo_state state;
 785
 786         if ((fsp != NULL) && (fsp->fh->fd != -1)) {
 787                 ret = SMB_VFS_FSTAT(fsp, &sbuf);
 788         }
 789         else {
 790                 struct smb_filename *smb_fname = NULL;
 791                 smb_fname = synthetic_smb_fname(talloc_tos(), fname, NULL,
 792                                                 NULL);
 793                 if (smb_fname == NULL) {
 794                         return NT_STATUS_NO_MEMORY;
 795                 }
 796                 if (lp_posix_pathnames()) {
 797                         ret = SMB_VFS_LSTAT(handle->conn, smb_fname);
 798                 } else {
 799                         ret = SMB_VFS_STAT(handle->conn, smb_fname);
 800                 }
 801                 sbuf = smb_fname->st;
 802                 TALLOC_FREE(smb_fname);
 803         }
 804
 805         if (ret == -1) {
 806                 return map_nt_error_from_unix(errno);
 807         }
 808
 809         state.streams = *pstreams;
 810         state.num_streams = *pnum_streams;
 811         state.mem_ctx = mem_ctx;
 812         state.handle = handle;
 813         state.status = NT_STATUS_OK;
 814
 815         status = walk_xattr_streams(handle->conn, fsp, fname,
 816                                     collect_one_stream, &state);
 817
 818         if (!NT_STATUS_IS_OK(status)) {
 819                 TALLOC_FREE(state.streams);
 820                 return status;
 821         }
 822
 823         if (!NT_STATUS_IS_OK(state.status)) {
 824                 TALLOC_FREE(state.streams);
 825                 return state.status;
 826         }
 827
 828         *pnum_streams = state.num_streams;
 829         *pstreams = state.streams;
 830
 831         return SMB_VFS_NEXT_STREAMINFO(handle, fsp, fname, mem_ctx, pnum_streams, pstreams);
 832 }
 833
 834 static uint32_t streams_xattr_fs_capabilities(struct vfs_handle_struct *handle,
 835                         enum timestamp_set_resolution *p_ts_res)
 836 {
 837         return SMB_VFS_NEXT_FS_CAPABILITIES(handle, p_ts_res) | FILE_NAMED_STREAMS;
 838 }
 839
 840 static ssize_t streams_xattr_pwrite(vfs_handle_struct *handle,
 841                                     files_struct *fsp, const void *data,
 842                                     size_t n, off_t offset)
 843 {
 844         struct stream_io *sio =
 845                 (struct stream_io *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
 846         struct ea_struct ea;
 847         NTSTATUS status;
 848         int ret;
 849
 850         DEBUG(10, ("streams_xattr_pwrite called for %d bytes\n", (int)n));
 851
 852         if (sio == NULL) {
 853                 return SMB_VFS_NEXT_PWRITE(handle, fsp, data, n, offset);
 854         }
 855
 856         if (!streams_xattr_recheck(sio)) {
 857                 return -1;
 858         }
 859
 860         status = get_ea_value(talloc_tos(), handle->conn, fsp->base_fsp,
 861                               sio->base, sio->xattr_name, &ea);
 862         if (!NT_STATUS_IS_OK(status)) {
 863                 return -1;
 864         }
 865
 866         if ((offset + n) > ea.value.length-1) {
 867                 uint8 *tmp;
 868
 869                 tmp = talloc_realloc(talloc_tos(), ea.value.data, uint8,
 870                                            offset + n + 1);
 871
 872                 if (tmp == NULL) {
 873                         TALLOC_FREE(ea.value.data);
 874                         errno = ENOMEM;
 875                         return -1;
 876                 }
 877                 ea.value.data = tmp;
 878                 ea.value.length = offset + n + 1;
 879                 ea.value.data[offset+n] = 0;
 880         }
 881
 882         memcpy(ea.value.data + offset, data, n);
 883
 884         if (fsp->base_fsp->fh->fd != -1) {
 885                 ret = SMB_VFS_FSETXATTR(fsp->base_fsp,
 886                                 sio->xattr_name,
 887                                 ea.value.data, ea.value.length, 0);
 888         } else {
 889                 ret = SMB_VFS_SETXATTR(fsp->conn,
 890                                        fsp->base_fsp->fsp_name->base_name,
 891                                 sio->xattr_name,
 892                                 ea.value.data, ea.value.length, 0);
 893         }
 894         TALLOC_FREE(ea.value.data);
 895
 896         if (ret == -1) {
 897                 return -1;
 898         }
 899
 900         return n;
 901 }
 902
 903 static ssize_t streams_xattr_pread(vfs_handle_struct *handle,
 904                                    files_struct *fsp, void *data,
 905                                    size_t n, off_t offset)
 906 {
 907         struct stream_io *sio =
 908                 (struct stream_io *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
 909         struct ea_struct ea;
 910         NTSTATUS status;
 911         size_t length, overlap;
 912
 913         DEBUG(10, ("streams_xattr_pread: offset=%d, size=%d\n",
 914                    (int)offset, (int)n));
 915
 916         if (sio == NULL) {
 917                 return SMB_VFS_NEXT_PREAD(handle, fsp, data, n, offset);
 918         }
 919
 920         if (!streams_xattr_recheck(sio)) {
 921                 return -1;
 922         }
 923
 924         status = get_ea_value(talloc_tos(), handle->conn, fsp->base_fsp,
 925                               sio->base, sio->xattr_name, &ea);
 926         if (!NT_STATUS_IS_OK(status)) {
 927                 return -1;
 928         }
 929
 930         length = ea.value.length-1;
 931
 932         DEBUG(10, ("streams_xattr_pread: get_ea_value returned %d bytes\n",
 933                    (int)length));
 934
 935         /* Attempt to read past EOF. */
 936         if (length <= offset) {
 937                 return 0;
 938         }
 939
 940         overlap = (offset + n) > length ? (length - offset) : n;
 941         memcpy(data, ea.value.data + offset, overlap);
 942
 943         TALLOC_FREE(ea.value.data);
 944         return overlap;
 945 }
 946
 947 static int streams_xattr_ftruncate(struct vfs_handle_struct *handle,
 948                                         struct files_struct *fsp,
 949                                         off_t offset)
 950 {
 951         int ret;
 952         uint8 *tmp;
 953         struct ea_struct ea;
 954         NTSTATUS status;
 955         struct stream_io *sio =
 956                 (struct stream_io *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
 957
 958         DEBUG(10, ("streams_xattr_ftruncate called for file %s offset %.0f\n",
 959                    fsp_str_dbg(fsp), (double)offset));
 960
 961         if (sio == NULL) {
 962                 return SMB_VFS_NEXT_FTRUNCATE(handle, fsp, offset);
 963         }
 964
 965         if (!streams_xattr_recheck(sio)) {
 966                 return -1;
 967         }
 968
 969         status = get_ea_value(talloc_tos(), handle->conn, fsp->base_fsp,
 970                               sio->base, sio->xattr_name, &ea);
 971         if (!NT_STATUS_IS_OK(status)) {
 972                 return -1;
 973         }
 974
 975         tmp = talloc_realloc(talloc_tos(), ea.value.data, uint8,
 976                                    offset + 1);
 977
 978         if (tmp == NULL) {
 979                 TALLOC_FREE(ea.value.data);
 980                 errno = ENOMEM;
 981                 return -1;
 982         }
 983
 984         /* Did we expand ? */
 985         if (ea.value.length < offset + 1) {
 986                 memset(&tmp[ea.value.length], '\0',
 987                         offset + 1 - ea.value.length);
 988         }
 989
 990         ea.value.data = tmp;
 991         ea.value.length = offset + 1;
 992         ea.value.data[offset] = 0;
 993
 994         if (fsp->base_fsp->fh->fd != -1) {
 995                 ret = SMB_VFS_FSETXATTR(fsp->base_fsp,
 996                                 sio->xattr_name,
 997                                 ea.value.data, ea.value.length, 0);
 998         } else {
 999                 ret = SMB_VFS_SETXATTR(fsp->conn,
1000                                        fsp->base_fsp->fsp_name->base_name,
1001                                 sio->xattr_name,
1002                                 ea.value.data, ea.value.length, 0);
1003         }
1004
1005         TALLOC_FREE(ea.value.data);
1006
1007         if (ret == -1) {
1008                 return -1;
1009         }
1010
1011         return 0;
1012 }
1013
1014 static int streams_xattr_fallocate(struct vfs_handle_struct *handle,
1015                                         struct files_struct *fsp,
1016                                         enum vfs_fallocate_mode mode,
1017                                         off_t offset,
1018                                         off_t len)
1019 {
1020         struct stream_io *sio =
1021                 (struct stream_io *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
1022
1023         DEBUG(10, ("streams_xattr_fallocate called for file %s offset %.0f"
1024                 "len = %.0f\n",
1025                 fsp_str_dbg(fsp), (double)offset, (double)len));
1026
1027         if (sio == NULL) {
1028                 return SMB_VFS_NEXT_FALLOCATE(handle, fsp, mode, offset, len);
1029         }
1030
1031         if (!streams_xattr_recheck(sio)) {
1032                 return errno;
1033         }
1034
1035         /* Let the pwrite code path handle it. */
1036         return ENOSYS;
1037 }
1038
1039
1040 static struct vfs_fn_pointers vfs_streams_xattr_fns = {
1041         .fs_capabilities_fn = streams_xattr_fs_capabilities,
1042         .open_fn = streams_xattr_open,
1043         .stat_fn = streams_xattr_stat,
1044         .fstat_fn = streams_xattr_fstat,
1045         .lstat_fn = streams_xattr_lstat,
1046         .pread_fn = streams_xattr_pread,
1047         .pwrite_fn = streams_xattr_pwrite,
1048         .unlink_fn = streams_xattr_unlink,
1049         .rename_fn = streams_xattr_rename,
1050         .ftruncate_fn = streams_xattr_ftruncate,
1051         .fallocate_fn = streams_xattr_fallocate,
1052         .streaminfo_fn = streams_xattr_streaminfo,
1053 };
1054
1055 NTSTATUS vfs_streams_xattr_init(void);
1056 NTSTATUS vfs_streams_xattr_init(void)
1057 {
1058         return smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "streams_xattr",
1059                                 &vfs_streams_xattr_fns);
1060 }