source3/modules/vfs_fruit.c

   1 /*
   2  * OS X and Netatalk interoperability VFS module for Samba-3.x
   3  *
   4  * Copyright (C) Ralph Boehme, 2013, 2014
   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 "MacExtensions.h"
  22 #include "smbd/smbd.h"
  23 #include "system/filesys.h"
  24 #include "lib/util/time.h"
  25 #include "../lib/crypto/md5.h"
  26 #include "system/shmem.h"
  27 #include "locking/proto.h"
  28 #include "smbd/globals.h"
  29 #include "messages.h"
  30 #include "libcli/security/security.h"
  31 #include "../libcli/smb/smb2_create_ctx.h"
  32 #include "lib/util/sys_rw.h"
  33 #include "lib/util/tevent_ntstatus.h"
  34 #include "lib/util/tevent_unix.h"
  35 #include "offload_token.h"
  36 #include "string_replace.h"
  37
  38 /*
  39  * Enhanced OS X and Netatalk compatibility
  40  * ========================================
  41  *
  42  * This modules takes advantage of vfs_streams_xattr and
  43  * vfs_catia. VFS modules vfs_fruit and vfs_streams_xattr must be
  44  * loaded in the correct order:
  45  *
  46  *   vfs modules = catia fruit streams_xattr
  47  *
  48  * The module intercepts the OS X special streams "AFP_AfpInfo" and
  49  * "AFP_Resource" and handles them in a special way. All other named
  50  * streams are deferred to vfs_streams_xattr.
  51  *
  52  * The OS X client maps all NTFS illegal characters to the Unicode
  53  * private range. This module optionally stores the charcters using
  54  * their native ASCII encoding using vfs_catia. If you're not enabling
  55  * this feature, you can skip catia from vfs modules.
  56  *
  57  * Finally, open modes are optionally checked against Netatalk AFP
  58  * share modes.
  59  *
  60  * The "AFP_AfpInfo" named stream is a binary blob containing OS X
  61  * extended metadata for files and directories. This module optionally
  62  * reads and stores this metadata in a way compatible with Netatalk 3
  63  * which stores the metadata in an EA "org.netatalk.metadata". Cf
  64  * source3/include/MacExtensions.h for a description of the binary
  65  * blobs content.
  66  *
  67  * The "AFP_Resource" named stream may be arbitrarily large, thus it
  68  * can't be stored in an xattr on most filesystem. ZFS on Solaris is
  69  * the only available filesystem where xattrs can be of any size and
  70  * the OS supports using the file APIs for xattrs.
  71  *
  72  * The AFP_Resource stream is stored in an AppleDouble file prepending
  73  * "._" to the filename. On Solaris with ZFS the stream is optionally
  74  * stored in an EA "org.netatalk.resource".
  75  *
  76  *
  77  * Extended Attributes
  78  * ===================
  79  *
  80  * The OS X SMB client sends xattrs as ADS too. For xattr interop with
  81  * other protocols you may want to adjust the xattr names the VFS
  82  * module vfs_streams_xattr uses for storing ADS's. This defaults to
  83  * user.DosStream.ADS_NAME:$DATA and can be changed by specifying
  84  * these module parameters:
  85  *
  86  *   streams_xattr:prefix = user.
  87  *   streams_xattr:store_stream_type = false
  88  *
  89  *
  90  * TODO
  91  * ====
  92  *
  93  * - log diagnostic if any needed VFS module is not loaded
  94  *   (eg with lp_vfs_objects())
  95  * - add tests
  96  */
  97
  98 static int vfs_fruit_debug_level = DBGC_VFS;
  99
 100 static struct global_fruit_config {
 101         bool nego_aapl; /* client negotiated AAPL */
 102
 103 } global_fruit_config;
 104
 105 #undef DBGC_CLASS
 106 #define DBGC_CLASS vfs_fruit_debug_level
 107
 108 #define FRUIT_PARAM_TYPE_NAME "fruit"
 109 #define ADOUBLE_NAME_PREFIX "._"
 110
 111 #define NETATALK_META_XATTR "org.netatalk.Metadata"
 112 #define NETATALK_RSRC_XATTR "org.netatalk.ResourceFork"
 113
 114 #if defined(HAVE_ATTROPEN)
 115 #define AFPINFO_EA_NETATALK NETATALK_META_XATTR
 116 #define AFPRESOURCE_EA_NETATALK NETATALK_RSRC_XATTR
 117 #else
 118 #define AFPINFO_EA_NETATALK "user." NETATALK_META_XATTR
 119 #define AFPRESOURCE_EA_NETATALK "user." NETATALK_RSRC_XATTR
 120 #endif
 121
 122 enum apple_fork {APPLE_FORK_DATA, APPLE_FORK_RSRC};
 123
 124 enum fruit_rsrc {FRUIT_RSRC_STREAM, FRUIT_RSRC_ADFILE, FRUIT_RSRC_XATTR};
 125 enum fruit_meta {FRUIT_META_STREAM, FRUIT_META_NETATALK};
 126 enum fruit_locking {FRUIT_LOCKING_NETATALK, FRUIT_LOCKING_NONE};
 127 enum fruit_encoding {FRUIT_ENC_NATIVE, FRUIT_ENC_PRIVATE};
 128
 129 struct fruit_config_data {
 130         enum fruit_rsrc rsrc;
 131         enum fruit_meta meta;
 132         enum fruit_locking locking;
 133         enum fruit_encoding encoding;
 134         bool use_aapl;          /* config from smb.conf */
 135         bool use_copyfile;
 136         bool readdir_attr_enabled;
 137         bool unix_info_enabled;
 138         bool copyfile_enabled;
 139         bool veto_appledouble;
 140         bool posix_rename;
 141         bool aapl_zero_file_id;
 142         const char *model;
 143         bool time_machine;
 144         off_t time_machine_max_size;
 145
 146         /*
 147          * Additional options, all enabled by default,
 148          * possibly useful for analyzing performance. The associated
 149          * operations with each of them may be expensive, so having
 150          * the chance to disable them individually gives a chance
 151          * tweaking the setup for the particular usecase.
 152          */
 153         bool readdir_attr_rsize;
 154         bool readdir_attr_finder_info;
 155         bool readdir_attr_max_access;
 156 };
 157
 158 static const struct enum_list fruit_rsrc[] = {
 159         {FRUIT_RSRC_STREAM, "stream"}, /* pass on to vfs_streams_xattr */
 160         {FRUIT_RSRC_ADFILE, "file"}, /* ._ AppleDouble file */
 161         {FRUIT_RSRC_XATTR, "xattr"}, /* Netatalk compatible xattr (ZFS only) */
 162         { -1, NULL}
 163 };
 164
 165 static const struct enum_list fruit_meta[] = {
 166         {FRUIT_META_STREAM, "stream"}, /* pass on to vfs_streams_xattr */
 167         {FRUIT_META_NETATALK, "netatalk"}, /* Netatalk compatible xattr */
 168         { -1, NULL}
 169 };
 170
 171 static const struct enum_list fruit_locking[] = {
 172         {FRUIT_LOCKING_NETATALK, "netatalk"}, /* synchronize locks with Netatalk */
 173         {FRUIT_LOCKING_NONE, "none"},
 174         { -1, NULL}
 175 };
 176
 177 static const struct enum_list fruit_encoding[] = {
 178         {FRUIT_ENC_NATIVE, "native"}, /* map unicode private chars to ASCII */
 179         {FRUIT_ENC_PRIVATE, "private"}, /* keep unicode private chars */
 180         { -1, NULL}
 181 };
 182
 183 static const char *fruit_catia_maps =
 184         "0x01:0xf001,0x02:0xf002,0x03:0xf003,0x04:0xf004,"
 185         "0x05:0xf005,0x06:0xf006,0x07:0xf007,0x08:0xf008,"
 186         "0x09:0xf009,0x0a:0xf00a,0x0b:0xf00b,0x0c:0xf00c,"
 187         "0x0d:0xf00d,0x0e:0xf00e,0x0f:0xf00f,0x10:0xf010,"
 188         "0x11:0xf011,0x12:0xf012,0x13:0xf013,0x14:0xf014,"
 189         "0x15:0xf015,0x16:0xf016,0x17:0xf017,0x18:0xf018,"
 190         "0x19:0xf019,0x1a:0xf01a,0x1b:0xf01b,0x1c:0xf01c,"
 191         "0x1d:0xf01d,0x1e:0xf01e,0x1f:0xf01f,"
 192         "0x22:0xf020,0x2a:0xf021,0x3a:0xf022,0x3c:0xf023,"
 193         "0x3e:0xf024,0x3f:0xf025,0x5c:0xf026,0x7c:0xf027,"
 194         "0x0d:0xf00d";
 195
 196 /*****************************************************************************
 197  * Defines, functions and data structures that deal with AppleDouble
 198  *****************************************************************************/
 199
 200 /*
 201  * There are two AppleDouble blobs we deal with:
 202  *
 203  * - ADOUBLE_META - AppleDouble blob used by Netatalk for storing
 204  *   metadata in an xattr
 205  *
 206  * - ADOUBLE_RSRC - AppleDouble blob used by OS X and Netatalk in
 207  *   ._ files
 208  */
 209 typedef enum {ADOUBLE_META, ADOUBLE_RSRC} adouble_type_t;
 210
 211 /* Version info */
 212 #define AD_VERSION2     0x00020000
 213 #define AD_VERSION      AD_VERSION2
 214
 215 /*
 216  * AppleDouble entry IDs.
 217  */
 218 #define ADEID_DFORK         1
 219 #define ADEID_RFORK         2
 220 #define ADEID_NAME          3
 221 #define ADEID_COMMENT       4
 222 #define ADEID_ICONBW        5
 223 #define ADEID_ICONCOL       6
 224 #define ADEID_FILEI         7
 225 #define ADEID_FILEDATESI    8
 226 #define ADEID_FINDERI       9
 227 #define ADEID_MACFILEI      10
 228 #define ADEID_PRODOSFILEI   11
 229 #define ADEID_MSDOSFILEI    12
 230 #define ADEID_SHORTNAME     13
 231 #define ADEID_AFPFILEI      14
 232 #define ADEID_DID           15
 233
 234 /* Private Netatalk entries */
 235 #define ADEID_PRIVDEV       16
 236 #define ADEID_PRIVINO       17
 237 #define ADEID_PRIVSYN       18
 238 #define ADEID_PRIVID        19
 239 #define ADEID_MAX           (ADEID_PRIVID + 1)
 240
 241 /*
 242  * These are the real ids for the private entries,
 243  * as stored in the adouble file
 244  */
 245 #define AD_DEV              0x80444556
 246 #define AD_INO              0x80494E4F
 247 #define AD_SYN              0x8053594E
 248 #define AD_ID               0x8053567E
 249
 250 /* Number of actually used entries */
 251 #define ADEID_NUM_XATTR      8
 252 #define ADEID_NUM_DOT_UND    2
 253 #define ADEID_NUM_RSRC_XATTR 1
 254
 255 /* AppleDouble magic */
 256 #define AD_APPLESINGLE_MAGIC 0x00051600
 257 #define AD_APPLEDOUBLE_MAGIC 0x00051607
 258 #define AD_MAGIC             AD_APPLEDOUBLE_MAGIC
 259
 260 /* Sizes of relevant entry bits */
 261 #define ADEDLEN_MAGIC       4
 262 #define ADEDLEN_VERSION     4
 263 #define ADEDLEN_FILLER      16
 264 #define AD_FILLER_TAG       "Netatalk        " /* should be 16 bytes */
 265 #define ADEDLEN_NENTRIES    2
 266 #define AD_HEADER_LEN       (ADEDLEN_MAGIC + ADEDLEN_VERSION + \
 267                              ADEDLEN_FILLER + ADEDLEN_NENTRIES) /* 26 */
 268 #define AD_ENTRY_LEN_EID    4
 269 #define AD_ENTRY_LEN_OFF    4
 270 #define AD_ENTRY_LEN_LEN    4
 271 #define AD_ENTRY_LEN (AD_ENTRY_LEN_EID + AD_ENTRY_LEN_OFF + AD_ENTRY_LEN_LEN)
 272
 273 /* Field widths */
 274 #define ADEDLEN_NAME            255
 275 #define ADEDLEN_COMMENT         200
 276 #define ADEDLEN_FILEI           16
 277 #define ADEDLEN_FINDERI         32
 278 #define ADEDLEN_FILEDATESI      16
 279 #define ADEDLEN_SHORTNAME       12 /* length up to 8.3 */
 280 #define ADEDLEN_AFPFILEI        4
 281 #define ADEDLEN_MACFILEI        4
 282 #define ADEDLEN_PRODOSFILEI     8
 283 #define ADEDLEN_MSDOSFILEI      2
 284 #define ADEDLEN_DID             4
 285 #define ADEDLEN_PRIVDEV         8
 286 #define ADEDLEN_PRIVINO         8
 287 #define ADEDLEN_PRIVSYN         8
 288 #define ADEDLEN_PRIVID          4
 289
 290 /* Offsets */
 291 #define ADEDOFF_MAGIC         0
 292 #define ADEDOFF_VERSION       (ADEDOFF_MAGIC + ADEDLEN_MAGIC)
 293 #define ADEDOFF_FILLER        (ADEDOFF_VERSION + ADEDLEN_VERSION)
 294 #define ADEDOFF_NENTRIES      (ADEDOFF_FILLER + ADEDLEN_FILLER)
 295
 296 #define ADEDOFF_FINDERI_XATTR    (AD_HEADER_LEN + \
 297                                   (ADEID_NUM_XATTR * AD_ENTRY_LEN))
 298 #define ADEDOFF_COMMENT_XATTR    (ADEDOFF_FINDERI_XATTR    + ADEDLEN_FINDERI)
 299 #define ADEDOFF_FILEDATESI_XATTR (ADEDOFF_COMMENT_XATTR    + ADEDLEN_COMMENT)
 300 #define ADEDOFF_AFPFILEI_XATTR   (ADEDOFF_FILEDATESI_XATTR + \
 301                                   ADEDLEN_FILEDATESI)
 302 #define ADEDOFF_PRIVDEV_XATTR    (ADEDOFF_AFPFILEI_XATTR   + ADEDLEN_AFPFILEI)
 303 #define ADEDOFF_PRIVINO_XATTR    (ADEDOFF_PRIVDEV_XATTR    + ADEDLEN_PRIVDEV)
 304 #define ADEDOFF_PRIVSYN_XATTR    (ADEDOFF_PRIVINO_XATTR    + ADEDLEN_PRIVINO)
 305 #define ADEDOFF_PRIVID_XATTR     (ADEDOFF_PRIVSYN_XATTR    + ADEDLEN_PRIVSYN)
 306
 307 #define ADEDOFF_FINDERI_DOT_UND  (AD_HEADER_LEN + \
 308                                   (ADEID_NUM_DOT_UND * AD_ENTRY_LEN))
 309 #define ADEDOFF_RFORK_DOT_UND    (ADEDOFF_FINDERI_DOT_UND + ADEDLEN_FINDERI)
 310
 311 #define AD_DATASZ_XATTR (AD_HEADER_LEN + \
 312                          (ADEID_NUM_XATTR * AD_ENTRY_LEN) + \
 313                          ADEDLEN_FINDERI + ADEDLEN_COMMENT + \
 314                          ADEDLEN_FILEDATESI + ADEDLEN_AFPFILEI + \
 315                          ADEDLEN_PRIVDEV + ADEDLEN_PRIVINO + \
 316                          ADEDLEN_PRIVSYN + ADEDLEN_PRIVID)
 317
 318 #if AD_DATASZ_XATTR != 402
 319 #error bad size for AD_DATASZ_XATTR
 320 #endif
 321
 322 #define AD_DATASZ_DOT_UND (AD_HEADER_LEN + \
 323                            (ADEID_NUM_DOT_UND * AD_ENTRY_LEN) + \
 324                            ADEDLEN_FINDERI)
 325 #if AD_DATASZ_DOT_UND != 82
 326 #error bad size for AD_DATASZ_DOT_UND
 327 #endif
 328
 329 /*
 330  * Sharemode locks fcntl() offsets
 331  */
 332 #if _FILE_OFFSET_BITS == 64 || defined(HAVE_LARGEFILE)
 333 #define AD_FILELOCK_BASE (UINT64_C(0x7FFFFFFFFFFFFFFF) - 9)
 334 #else
 335 #define AD_FILELOCK_BASE (UINT32_C(0x7FFFFFFF) - 9)
 336 #endif
 337 #define BYTELOCK_MAX (AD_FILELOCK_BASE - 1)
 338
 339 #define AD_FILELOCK_OPEN_WR        (AD_FILELOCK_BASE + 0)
 340 #define AD_FILELOCK_OPEN_RD        (AD_FILELOCK_BASE + 1)
 341 #define AD_FILELOCK_RSRC_OPEN_WR   (AD_FILELOCK_BASE + 2)
 342 #define AD_FILELOCK_RSRC_OPEN_RD   (AD_FILELOCK_BASE + 3)
 343 #define AD_FILELOCK_DENY_WR        (AD_FILELOCK_BASE + 4)
 344 #define AD_FILELOCK_DENY_RD        (AD_FILELOCK_BASE + 5)
 345 #define AD_FILELOCK_RSRC_DENY_WR   (AD_FILELOCK_BASE + 6)
 346 #define AD_FILELOCK_RSRC_DENY_RD   (AD_FILELOCK_BASE + 7)
 347 #define AD_FILELOCK_OPEN_NONE      (AD_FILELOCK_BASE + 8)
 348 #define AD_FILELOCK_RSRC_OPEN_NONE (AD_FILELOCK_BASE + 9)
 349
 350 /* Time stuff we overload the bits a little */
 351 #define AD_DATE_CREATE         0
 352 #define AD_DATE_MODIFY         4
 353 #define AD_DATE_BACKUP         8
 354 #define AD_DATE_ACCESS        12
 355 #define AD_DATE_MASK          (AD_DATE_CREATE | AD_DATE_MODIFY | \
 356                                AD_DATE_BACKUP | AD_DATE_ACCESS)
 357 #define AD_DATE_UNIX          (1 << 10)
 358 #define AD_DATE_START         0x80000000
 359 #define AD_DATE_DELTA         946684800
 360 #define AD_DATE_FROM_UNIX(x)  (htonl((x) - AD_DATE_DELTA))
 361 #define AD_DATE_TO_UNIX(x)    (ntohl(x) + AD_DATE_DELTA)
 362
 363 #define AD_XATTR_HDR_MAGIC    0x41545452 /* 'ATTR' */
 364 #define AD_XATTR_MAX_ENTRIES  1024 /* Some arbitrarily enforced limit */
 365 #define AD_XATTR_HDR_SIZE     36
 366 #define AD_XATTR_MAX_HDR_SIZE 65536
 367
 368 /* Accessor macros */
 369 #define ad_getentrylen(ad,eid)     ((ad)->ad_eid[(eid)].ade_len)
 370 #define ad_getentryoff(ad,eid)     ((ad)->ad_eid[(eid)].ade_off)
 371 #define ad_setentrylen(ad,eid,len) ((ad)->ad_eid[(eid)].ade_len = (len))
 372 #define ad_setentryoff(ad,eid,off) ((ad)->ad_eid[(eid)].ade_off = (off))
 373
 374 /*
 375  * Both struct ad_xattr_header and struct ad_xattr_entry describe the in memory
 376  * representation as well as the on-disk format.
 377  *
 378  * The ad_xattr_header follows the FinderInfo data in the FinderInfo entry if
 379  * the length of the FinderInfo entry is larger then 32 bytes. It is then
 380  * preceeded with 2 bytes padding.
 381  *
 382  * Cf: https://opensource.apple.com/source/xnu/xnu-4570.1.46/bsd/vfs/vfs_xattr.c
 383  */
 384
 385 struct ad_xattr_header {
 386         uint32_t adx_magic;        /* ATTR_HDR_MAGIC */
 387         uint32_t adx_debug_tag;    /* for debugging == file id of owning file */
 388         uint32_t adx_total_size;   /* file offset of end of attribute header + entries + data */
 389         uint32_t adx_data_start;   /* file offset to attribute data area */
 390         uint32_t adx_data_length;  /* length of attribute data area */
 391         uint32_t adx_reserved[3];
 392         uint16_t adx_flags;
 393         uint16_t adx_num_attrs;
 394 };
 395
 396 /* On-disk entries are aligned on 4 byte boundaries */
 397 struct ad_xattr_entry {
 398         uint32_t adx_offset;    /* file offset to data */
 399         uint32_t adx_length;    /* size of attribute data */
 400         uint16_t adx_flags;
 401         uint8_t  adx_namelen;   /* included the NULL terminator */
 402         char    *adx_name;      /* NULL-terminated UTF-8 name */
 403 };
 404
 405 struct ad_entry {
 406         size_t ade_off;
 407         size_t ade_len;
 408 };
 409
 410 struct adouble {
 411         vfs_handle_struct        *ad_handle;
 412         int                       ad_fd;
 413         bool                      ad_opened;
 414         adouble_type_t            ad_type;
 415         uint32_t                  ad_magic;
 416         uint32_t                  ad_version;
 417         struct ad_entry           ad_eid[ADEID_MAX];
 418         char                     *ad_data;
 419         struct ad_xattr_header    adx_header;
 420         struct ad_xattr_entry    *adx_entries;
 421 };
 422
 423 struct ad_entry_order {
 424         uint32_t id, offset, len;
 425 };
 426
 427 /* Netatalk AppleDouble metadata xattr */
 428 static const
 429 struct ad_entry_order entry_order_meta_xattr[ADEID_NUM_XATTR + 1] = {
 430         {ADEID_FINDERI,    ADEDOFF_FINDERI_XATTR,    ADEDLEN_FINDERI},
 431         {ADEID_COMMENT,    ADEDOFF_COMMENT_XATTR,    0},
 432         {ADEID_FILEDATESI, ADEDOFF_FILEDATESI_XATTR, ADEDLEN_FILEDATESI},
 433         {ADEID_AFPFILEI,   ADEDOFF_AFPFILEI_XATTR,   ADEDLEN_AFPFILEI},
 434         {ADEID_PRIVDEV,    ADEDOFF_PRIVDEV_XATTR,    0},
 435         {ADEID_PRIVINO,    ADEDOFF_PRIVINO_XATTR,    0},
 436         {ADEID_PRIVSYN,    ADEDOFF_PRIVSYN_XATTR,    0},
 437         {ADEID_PRIVID,     ADEDOFF_PRIVID_XATTR,     0},
 438         {0, 0, 0}
 439 };
 440
 441 /* AppleDouble resource fork file (the ones prefixed by "._") */
 442 static const
 443 struct ad_entry_order entry_order_dot_und[ADEID_NUM_DOT_UND + 1] = {
 444         {ADEID_FINDERI,    ADEDOFF_FINDERI_DOT_UND,  ADEDLEN_FINDERI},
 445         {ADEID_RFORK,      ADEDOFF_RFORK_DOT_UND,    0},
 446         {0, 0, 0}
 447 };
 448
 449 /*
 450  * Fake AppleDouble entry oder for resource fork xattr.  The xattr
 451  * isn't an AppleDouble file, it simply contains the resource data,
 452  * but in order to be able to use some API calls like ad_getentryoff()
 453  * we build a fake/helper struct adouble with this entry order struct.
 454  */
 455 static const
 456 struct ad_entry_order entry_order_rsrc_xattr[ADEID_NUM_RSRC_XATTR + 1] = {
 457         {ADEID_RFORK, 0, 0},
 458         {0, 0, 0}
 459 };
 460
 461 /* Conversion from enumerated id to on-disk AppleDouble id */
 462 #define AD_EID_DISK(a) (set_eid[a])
 463 static const uint32_t set_eid[] = {
 464         0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
 465         AD_DEV, AD_INO, AD_SYN, AD_ID
 466 };
 467
 468 struct fio {
 469         /* tcon config handle */
 470         struct fruit_config_data *config;
 471
 472         /* Denote stream type, meta or rsrc */
 473         adouble_type_t type;
 474 };
 475
 476 /*
 477  * Forward declarations
 478  */
 479 static struct adouble *ad_init(TALLOC_CTX *ctx, vfs_handle_struct *handle,
 480                                adouble_type_t type);
 481 static int ad_set(struct adouble *ad, const struct smb_filename *smb_fname);
 482 static int ad_fset(struct adouble *ad, files_struct *fsp);
 483 static int adouble_path(TALLOC_CTX *ctx,
 484                         const struct smb_filename *smb_fname__in,
 485                         struct smb_filename **ppsmb_fname_out);
 486 static AfpInfo *afpinfo_new(TALLOC_CTX *ctx);
 487 static ssize_t afpinfo_pack(const AfpInfo *ai, char *buf);
 488 static AfpInfo *afpinfo_unpack(TALLOC_CTX *ctx, const void *data);
 489
 490
 491 /**
 492  * Return a pointer to an AppleDouble entry
 493  *
 494  * Returns NULL if the entry is not present
 495  **/
 496 static char *ad_get_entry(const struct adouble *ad, int eid)
 497 {
 498         off_t off = ad_getentryoff(ad, eid);
 499         size_t len = ad_getentrylen(ad, eid);
 500
 501         if (off == 0 || len == 0) {
 502                 return NULL;
 503         }
 504
 505         return ad->ad_data + off;
 506 }
 507
 508 /**
 509  * Get a date
 510  **/
 511 static int ad_getdate(const struct adouble *ad,
 512                       unsigned int dateoff,
 513                       uint32_t *date)
 514 {
 515         bool xlate = (dateoff & AD_DATE_UNIX);
 516         char *p = NULL;
 517
 518         dateoff &= AD_DATE_MASK;
 519         p = ad_get_entry(ad, ADEID_FILEDATESI);
 520         if (p == NULL) {
 521                 return -1;
 522         }
 523
 524         if (dateoff > AD_DATE_ACCESS) {
 525             return -1;
 526         }
 527
 528         memcpy(date, p + dateoff, sizeof(uint32_t));
 529
 530         if (xlate) {
 531                 *date = AD_DATE_TO_UNIX(*date);
 532         }
 533         return 0;
 534 }
 535
 536 /**
 537  * Set a date
 538  **/
 539 static int ad_setdate(struct adouble *ad, unsigned int dateoff, uint32_t date)
 540 {
 541         bool xlate = (dateoff & AD_DATE_UNIX);
 542         char *p = NULL;
 543
 544         p = ad_get_entry(ad, ADEID_FILEDATESI);
 545         if (p == NULL) {
 546                 return -1;
 547         }
 548
 549         dateoff &= AD_DATE_MASK;
 550         if (xlate) {
 551                 date = AD_DATE_FROM_UNIX(date);
 552         }
 553
 554         if (dateoff > AD_DATE_ACCESS) {
 555                 return -1;
 556         }
 557
 558         memcpy(p + dateoff, &date, sizeof(date));
 559
 560         return 0;
 561 }
 562
 563
 564 /**
 565  * Map on-disk AppleDouble id to enumerated id
 566  **/
 567 static uint32_t get_eid(uint32_t eid)
 568 {
 569         if (eid <= 15) {
 570                 return eid;
 571         }
 572
 573         switch (eid) {
 574         case AD_DEV:
 575                 return ADEID_PRIVDEV;
 576         case AD_INO:
 577                 return ADEID_PRIVINO;
 578         case AD_SYN:
 579                 return ADEID_PRIVSYN;
 580         case AD_ID:
 581                 return ADEID_PRIVID;
 582         default:
 583                 break;
 584         }
 585
 586         return 0;
 587 }
 588
 589 /**
 590  * Pack AppleDouble structure into data buffer
 591  **/
 592 static bool ad_pack(struct adouble *ad)
 593 {
 594         uint32_t       eid;
 595         uint16_t       nent;
 596         uint32_t       bufsize;
 597         uint32_t       offset = 0;
 598
 599         bufsize = talloc_get_size(ad->ad_data);
 600         if (bufsize < AD_DATASZ_DOT_UND) {
 601                 DBG_ERR("bad buffer size [0x%" PRIx32 "]\n", bufsize);
 602                 return false;
 603         }
 604
 605         if (offset + ADEDLEN_MAGIC < offset ||
 606                         offset + ADEDLEN_MAGIC >= bufsize) {
 607                 return false;
 608         }
 609         RSIVAL(ad->ad_data, offset, ad->ad_magic);
 610         offset += ADEDLEN_MAGIC;
 611
 612         if (offset + ADEDLEN_VERSION < offset ||
 613                         offset + ADEDLEN_VERSION >= bufsize) {
 614                 return false;
 615         }
 616         RSIVAL(ad->ad_data, offset, ad->ad_version);
 617         offset += ADEDLEN_VERSION;
 618
 619         if (offset + ADEDLEN_FILLER < offset ||
 620                         offset + ADEDLEN_FILLER >= bufsize) {
 621                 return false;
 622         }
 623         if (ad->ad_type == ADOUBLE_RSRC) {
 624                 memcpy(ad->ad_data + offset, AD_FILLER_TAG, ADEDLEN_FILLER);
 625         }
 626         offset += ADEDLEN_FILLER;
 627
 628         if (offset + ADEDLEN_NENTRIES < offset ||
 629                         offset + ADEDLEN_NENTRIES >= bufsize) {
 630                 return false;
 631         }
 632         offset += ADEDLEN_NENTRIES;
 633
 634         for (eid = 0, nent = 0; eid < ADEID_MAX; eid++) {
 635                 if (ad->ad_eid[eid].ade_off == 0) {
 636                         /*
 637                          * ade_off is also used as indicator whether a
 638                          * specific entry is used or not
 639                          */
 640                         continue;
 641                 }
 642
 643                 if (offset + AD_ENTRY_LEN_EID < offset ||
 644                                 offset + AD_ENTRY_LEN_EID >= bufsize) {
 645                         return false;
 646                 }
 647                 RSIVAL(ad->ad_data, offset, AD_EID_DISK(eid));
 648                 offset += AD_ENTRY_LEN_EID;
 649
 650                 if (offset + AD_ENTRY_LEN_OFF < offset ||
 651                                 offset + AD_ENTRY_LEN_OFF >= bufsize) {
 652                         return false;
 653                 }
 654                 RSIVAL(ad->ad_data, offset, ad->ad_eid[eid].ade_off);
 655                 offset += AD_ENTRY_LEN_OFF;
 656
 657                 if (offset + AD_ENTRY_LEN_LEN < offset ||
 658                                 offset + AD_ENTRY_LEN_LEN >= bufsize) {
 659                         return false;
 660                 }
 661                 RSIVAL(ad->ad_data, offset, ad->ad_eid[eid].ade_len);
 662                 offset += AD_ENTRY_LEN_LEN;
 663
 664                 nent++;
 665         }
 666
 667         if (ADEDOFF_NENTRIES + 2 >= bufsize) {
 668                 return false;
 669         }
 670         RSSVAL(ad->ad_data, ADEDOFF_NENTRIES, nent);
 671
 672         return true;
 673 }
 674
 675 static bool ad_unpack_xattrs(struct adouble *ad)
 676 {
 677         struct ad_xattr_header *h = &ad->adx_header;
 678         const char *p = ad->ad_data;
 679         uint32_t hoff;
 680         uint32_t i;
 681
 682         if (ad_getentrylen(ad, ADEID_FINDERI) <= ADEDLEN_FINDERI) {
 683                 return true;
 684         }
 685
 686         /* 2 bytes padding */
 687         hoff = ad_getentryoff(ad, ADEID_FINDERI) + ADEDLEN_FINDERI + 2;
 688
 689         h->adx_magic       = RIVAL(p, hoff + 0);
 690         h->adx_debug_tag   = RIVAL(p, hoff + 4); /* Not used -> not checked */
 691         h->adx_total_size  = RIVAL(p, hoff + 8);
 692         h->adx_data_start  = RIVAL(p, hoff + 12);
 693         h->adx_data_length = RIVAL(p, hoff + 16);
 694         h->adx_flags       = RSVAL(p, hoff + 32); /* Not used -> not checked */
 695         h->adx_num_attrs   = RSVAL(p, hoff + 34);
 696
 697         if (h->adx_magic != AD_XATTR_HDR_MAGIC) {
 698                 DBG_ERR("Bad magic: 0x%" PRIx32 "\n", h->adx_magic);
 699                 return false;
 700         }
 701
 702         if (h->adx_total_size > ad_getentryoff(ad, ADEID_RFORK)) {
 703                 DBG_ERR("Bad total size: 0x%" PRIx32 "\n", h->adx_total_size);
 704                 return false;
 705         }
 706         if (h->adx_total_size > AD_XATTR_MAX_HDR_SIZE) {
 707                 DBG_ERR("Bad total size: 0x%" PRIx32 "\n", h->adx_total_size);
 708                 return false;
 709         }
 710
 711         if (h->adx_data_start < (hoff + AD_XATTR_HDR_SIZE)) {
 712                 DBG_ERR("Bad start: 0x%" PRIx32 "\n", h->adx_data_start);
 713                 return false;
 714         }
 715
 716         if ((h->adx_data_start + h->adx_data_length) < h->adx_data_start) {
 717                 DBG_ERR("Bad length: %" PRIu32 "\n", h->adx_data_length);
 718                 return false;
 719         }
 720         if ((h->adx_data_start + h->adx_data_length) >
 721             ad->adx_header.adx_total_size)
 722         {
 723                 DBG_ERR("Bad length: %" PRIu32 "\n", h->adx_data_length);
 724                 return false;
 725         }
 726
 727         if (h->adx_num_attrs > AD_XATTR_MAX_ENTRIES) {
 728                 DBG_ERR("Bad num xattrs: %" PRIu16 "\n", h->adx_num_attrs);
 729                 return false;
 730         }
 731
 732         if (h->adx_num_attrs == 0) {
 733                 return true;
 734         }
 735
 736         ad->adx_entries = talloc_zero_array(
 737                 ad, struct ad_xattr_entry, h->adx_num_attrs);
 738         if (ad->adx_entries == NULL) {
 739                 return false;
 740         }
 741
 742         hoff += AD_XATTR_HDR_SIZE;
 743
 744         for (i = 0; i < h->adx_num_attrs; i++) {
 745                 struct ad_xattr_entry *e = &ad->adx_entries[i];
 746
 747                 hoff = (hoff + 3) & ~3;
 748
 749                 e->adx_offset  = RIVAL(p, hoff + 0);
 750                 e->adx_length  = RIVAL(p, hoff + 4);
 751                 e->adx_flags   = RSVAL(p, hoff + 8);
 752                 e->adx_namelen = *(p + hoff + 10);
 753
 754                 if (e->adx_offset >= ad->adx_header.adx_total_size) {
 755                         DBG_ERR("Bad adx_offset: %" PRIx32 "\n",
 756                                 e->adx_offset);
 757                         return false;
 758                 }
 759
 760                 if ((e->adx_offset + e->adx_length) < e->adx_offset) {
 761                         DBG_ERR("Bad adx_length: %" PRIx32 "\n",
 762                                 e->adx_length);
 763                         return false;
 764                 }
 765
 766                 if ((e->adx_offset + e->adx_length) >
 767                     ad->adx_header.adx_total_size)
 768                 {
 769                         DBG_ERR("Bad adx_length: %" PRIx32 "\n",
 770                                 e->adx_length);
 771                         return false;
 772                 }
 773
 774                 if (e->adx_namelen == 0) {
 775                         DBG_ERR("Bad adx_namelen: %" PRIx32 "\n",
 776                                 e->adx_namelen);
 777                         return false;
 778                 }
 779                 if ((hoff + 11 + e->adx_namelen) < hoff + 11) {
 780                         DBG_ERR("Bad adx_namelen: %" PRIx32 "\n",
 781                                 e->adx_namelen);
 782                         return false;
 783                 }
 784                 if ((hoff + 11 + e->adx_namelen) >
 785                     ad->adx_header.adx_data_start)
 786                 {
 787                         DBG_ERR("Bad adx_namelen: %" PRIx32 "\n",
 788                                 e->adx_namelen);
 789                         return false;
 790                 }
 791
 792                 e->adx_name = talloc_strndup(ad->adx_entries,
 793                                              p + hoff + 11,
 794                                              e->adx_namelen);
 795                 if (e->adx_name == NULL) {
 796                         return false;
 797                 }
 798
 799                 DBG_DEBUG("xattr [%s] offset [0x%x] size [0x%x]\n",
 800                           e->adx_name, e->adx_offset, e->adx_length);
 801                 dump_data(10, (uint8_t *)(ad->ad_data + e->adx_offset),
 802                           e->adx_length);
 803
 804                 hoff += 11 + e->adx_namelen;
 805         }
 806
 807         return true;
 808 }
 809
 810 /**
 811  * Unpack an AppleDouble blob into a struct adoble
 812  **/
 813 static bool ad_unpack(struct adouble *ad, const size_t nentries,
 814                       size_t filesize)
 815 {
 816         size_t bufsize = talloc_get_size(ad->ad_data);
 817         size_t adentries, i;
 818         uint32_t eid, len, off;
 819         bool ok;
 820
 821         /*
 822          * The size of the buffer ad->ad_data is checked when read, so
 823          * we wouldn't have to check our own offsets, a few extra
 824          * checks won't hurt though. We have to check the offsets we
 825          * read from the buffer anyway.
 826          */
 827
 828         if (bufsize < (AD_HEADER_LEN + (AD_ENTRY_LEN * nentries))) {
 829                 DEBUG(1, ("bad size\n"));
 830                 return false;
 831         }
 832
 833         ad->ad_magic = RIVAL(ad->ad_data, 0);
 834         ad->ad_version = RIVAL(ad->ad_data, ADEDOFF_VERSION);
 835         if ((ad->ad_magic != AD_MAGIC) || (ad->ad_version != AD_VERSION)) {
 836                 DEBUG(1, ("wrong magic or version\n"));
 837                 return false;
 838         }
 839
 840         adentries = RSVAL(ad->ad_data, ADEDOFF_NENTRIES);
 841         if (adentries != nentries) {
 842                 DEBUG(1, ("invalid number of entries: %zu\n",
 843                           adentries));
 844                 return false;
 845         }
 846
 847         /* now, read in the entry bits */
 848         for (i = 0; i < adentries; i++) {
 849                 eid = RIVAL(ad->ad_data, AD_HEADER_LEN + (i * AD_ENTRY_LEN));
 850                 eid = get_eid(eid);
 851                 off = RIVAL(ad->ad_data, AD_HEADER_LEN + (i * AD_ENTRY_LEN) + 4);
 852                 len = RIVAL(ad->ad_data, AD_HEADER_LEN + (i * AD_ENTRY_LEN) + 8);
 853
 854                 if (!eid || eid >= ADEID_MAX) {
 855                         DEBUG(1, ("bogus eid %d\n", eid));
 856                         return false;
 857                 }
 858
 859                 /*
 860                  * All entries other than the resource fork are
 861                  * expected to be read into the ad_data buffer, so
 862                  * ensure the specified offset is within that bound
 863                  */
 864                 if ((off > bufsize) && (eid != ADEID_RFORK)) {
 865                         DEBUG(1, ("bogus eid %d: off: %" PRIu32 ", len: %" PRIu32 "\n",
 866                                   eid, off, len));
 867                         return false;
 868                 }
 869
 870                 /*
 871                  * All entries besides FinderInfo and resource fork
 872                  * must fit into the buffer. FinderInfo is special as
 873                  * it may be larger then the default 32 bytes (if it
 874                  * contains marshalled xattrs), but we will fixup that
 875                  * in ad_convert(). And the resource fork is never
 876                  * accessed directly by the ad_data buf (also see
 877                  * comment above) anyway.
 878                  */
 879                 if ((eid != ADEID_RFORK) &&
 880                     (eid != ADEID_FINDERI) &&
 881                     ((off + len) > bufsize)) {
 882                         DEBUG(1, ("bogus eid %d: off: %" PRIu32 ", len: %" PRIu32 "\n",
 883                                   eid, off, len));
 884                         return false;
 885                 }
 886
 887                 /*
 888                  * That would be obviously broken
 889                  */
 890                 if (off > filesize) {
 891                         DEBUG(1, ("bogus eid %d: off: %" PRIu32 ", len: %" PRIu32 "\n",
 892                                   eid, off, len));
 893                         return false;
 894                 }
 895
 896                 /*
 897                  * Check for any entry that has its end beyond the
 898                  * filesize.
 899                  */
 900                 if (off + len < off) {
 901                         DEBUG(1, ("offset wrap in eid %d: off: %" PRIu32
 902                                   ", len: %" PRIu32 "\n",
 903                                   eid, off, len));
 904                         return false;
 905
 906                 }
 907                 if (off + len > filesize) {
 908                         /*
 909                          * If this is the resource fork entry, we fix
 910                          * up the length, for any other entry we bail
 911                          * out.
 912                          */
 913                         if (eid != ADEID_RFORK) {
 914                                 DEBUG(1, ("bogus eid %d: off: %" PRIu32
 915                                           ", len: %" PRIu32 "\n",
 916                                           eid, off, len));
 917                                 return false;
 918                         }
 919
 920                         /*
 921                          * Fixup the resource fork entry by limiting
 922                          * the size to entryoffset - filesize.
 923                          */
 924                         len = filesize - off;
 925                         DEBUG(1, ("Limiting ADEID_RFORK: off: %" PRIu32
 926                                   ", len: %" PRIu32 "\n", off, len));
 927                 }
 928
 929                 ad->ad_eid[eid].ade_off = off;
 930                 ad->ad_eid[eid].ade_len = len;
 931         }
 932
 933         ok = ad_unpack_xattrs(ad);
 934         if (!ok) {
 935                 return false;
 936         }
 937
 938         return true;
 939 }
 940
 941 static bool ad_convert_xattr(struct adouble *ad,
 942                              const struct smb_filename *smb_fname,
 943                              char *map)
 944 {
 945         static struct char_mappings **string_replace_cmaps = NULL;
 946         uint16_t i;
 947         int saved_errno = 0;
 948         NTSTATUS status;
 949
 950         if (ad->adx_header.adx_num_attrs == 0) {
 951                 return true;
 952         }
 953
 954         if (string_replace_cmaps == NULL) {
 955                 const char **mappings = NULL;
 956
 957                 mappings = str_list_make_v3_const(
 958                         talloc_tos(), fruit_catia_maps, NULL);
 959                 if (mappings == NULL) {
 960                         return false;
 961                 }
 962                 string_replace_cmaps = string_replace_init_map(mappings);
 963                 TALLOC_FREE(mappings);
 964         }
 965
 966         for (i = 0; i < ad->adx_header.adx_num_attrs; i++) {
 967                 struct ad_xattr_entry *e = &ad->adx_entries[i];
 968                 char *mapped_name = NULL;
 969                 char *tmp = NULL;
 970                 struct smb_filename *stream_name = NULL;
 971                 files_struct *fsp = NULL;
 972                 ssize_t nwritten;
 973
 974                 status = string_replace_allocate(ad->ad_handle->conn,
 975                                                  e->adx_name,
 976                                                  string_replace_cmaps,
 977                                                  talloc_tos(),
 978                                                  &mapped_name,
 979                                                  vfs_translate_to_windows);
 980                 if (!NT_STATUS_IS_OK(status) &&
 981                     !NT_STATUS_EQUAL(status, NT_STATUS_NONE_MAPPED))
 982                 {
 983                         DBG_ERR("string_replace_allocate failed\n");
 984                         return -1;
 985                 }
 986
 987                 tmp = mapped_name;
 988                 mapped_name = talloc_asprintf(talloc_tos(), ":%s", tmp);
 989                 TALLOC_FREE(tmp);
 990                 if (mapped_name == NULL) {
 991                         return -1;
 992                 }
 993
 994                 stream_name = synthetic_smb_fname(talloc_tos(),
 995                                                   smb_fname->base_name,
 996                                                   mapped_name,
 997                                                   NULL,
 998                                                   smb_fname->flags);
 999                 TALLOC_FREE(mapped_name);
1000                 if (stream_name == NULL) {
1001                         DBG_ERR("synthetic_smb_fname failed\n");
1002                         return -1;
1003                 }
1004
1005                 DBG_DEBUG("stream_name: %s\n", smb_fname_str_dbg(stream_name));
1006
1007                 status = SMB_VFS_CREATE_FILE(
1008                         ad->ad_handle->conn,            /* conn */
1009                         NULL,                           /* req */
1010                         0,                              /* root_dir_fid */
1011                         stream_name,                    /* fname */
1012                         FILE_GENERIC_WRITE,             /* access_mask */
1013                         FILE_SHARE_READ | FILE_SHARE_WRITE, /* share_access */
1014                         FILE_OPEN_IF,                   /* create_disposition */
1015                         0,                              /* create_options */
1016                         0,                              /* file_attributes */
1017                         INTERNAL_OPEN_ONLY,             /* oplock_request */
1018                         NULL,                           /* lease */
1019                         0,                              /* allocation_size */
1020                         0,                              /* private_flags */
1021                         NULL,                           /* sd */
1022                         NULL,                           /* ea_list */
1023                         &fsp,                           /* result */
1024                         NULL,                           /* psbuf */
1025                         NULL, NULL);                    /* create context */
1026                 TALLOC_FREE(stream_name);
1027                 if (!NT_STATUS_IS_OK(status)) {
1028                         DBG_ERR("SMB_VFS_CREATE_FILE failed\n");
1029                         return -1;
1030                 }
1031
1032                 nwritten = SMB_VFS_PWRITE(fsp,
1033                                           map + e->adx_offset,
1034                                           e->adx_length,
1035                                           0);
1036                 if (nwritten == -1) {
1037                         DBG_ERR("SMB_VFS_PWRITE failed\n");
1038                         saved_errno = errno;
1039                         close_file(NULL, fsp, ERROR_CLOSE);
1040                         errno = saved_errno;
1041                         return -1;
1042                 }
1043
1044                 status = close_file(NULL, fsp, NORMAL_CLOSE);
1045                 if (!NT_STATUS_IS_OK(status)) {
1046                         return -1;
1047                 }
1048                 fsp = NULL;
1049         }
1050
1051         return true;
1052 }
1053
1054 /**
1055  * Convert from Apple's ._ file to Netatalk
1056  *
1057  * Apple's AppleDouble may contain a FinderInfo entry longer then 32
1058  * bytes containing packed xattrs. Netatalk can't deal with that, so
1059  * we simply discard the packed xattrs.
1060  *
1061  * @return -1 in case an error occurred, 0 if no conversion was done, 1
1062  * otherwise
1063  **/
1064 static int ad_convert(struct adouble *ad,
1065                       const struct smb_filename *smb_fname,
1066                       int fd)
1067 {
1068         int rc = 0;
1069         char *map = MAP_FAILED;
1070         size_t origlen;
1071         bool ok;
1072
1073         origlen = ad_getentryoff(ad, ADEID_RFORK) +
1074                 ad_getentrylen(ad, ADEID_RFORK);
1075
1076         /* FIXME: direct use of mmap(), vfs_aio_fork does it too */
1077         map = mmap(NULL, origlen, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0);
1078         if (map == MAP_FAILED) {
1079                 DEBUG(2, ("mmap AppleDouble: %s\n", strerror(errno)));
1080                 rc = -1;
1081                 goto exit;
1082         }
1083
1084         ok = ad_convert_xattr(ad, smb_fname, map);
1085         if (!ok) {
1086                 munmap(map, origlen);
1087                 return -1;
1088         }
1089
1090         if (ad_getentrylen(ad, ADEID_RFORK) > 0) {
1091                 memmove(map + ad_getentryoff(ad, ADEID_FINDERI) + ADEDLEN_FINDERI,
1092                         map + ad_getentryoff(ad, ADEID_RFORK),
1093                         ad_getentrylen(ad, ADEID_RFORK));
1094         }
1095
1096         ad_setentrylen(ad, ADEID_FINDERI, ADEDLEN_FINDERI);
1097         ad_setentryoff(ad, ADEID_RFORK,
1098                        ad_getentryoff(ad, ADEID_FINDERI) + ADEDLEN_FINDERI);
1099
1100         /*
1101          * FIXME: direct ftruncate(), but we don't have a fsp for the
1102          * VFS call
1103          */
1104         rc = ftruncate(fd, ad_getentryoff(ad, ADEID_RFORK)
1105                        + ad_getentrylen(ad, ADEID_RFORK));
1106
1107 exit:
1108         if (map != MAP_FAILED) {
1109                 munmap(map, origlen);
1110         }
1111         return rc;
1112 }
1113
1114 /**
1115  * Read and parse Netatalk AppleDouble metadata xattr
1116  **/
1117 static ssize_t ad_read_meta(struct adouble *ad,
1118                                 const struct smb_filename *smb_fname)
1119 {
1120         int      rc = 0;
1121         ssize_t  ealen;
1122         bool     ok;
1123
1124         DEBUG(10, ("reading meta xattr for %s\n", smb_fname->base_name));
1125
1126         ealen = SMB_VFS_GETXATTR(ad->ad_handle->conn, smb_fname,
1127                                  AFPINFO_EA_NETATALK, ad->ad_data,
1128                                  AD_DATASZ_XATTR);
1129         if (ealen == -1) {
1130                 switch (errno) {
1131                 case ENOATTR:
1132                 case ENOENT:
1133                         if (errno == ENOATTR) {
1134                                 errno = ENOENT;
1135                         }
1136                         rc = -1;
1137                         goto exit;
1138                 default:
1139                         DEBUG(2, ("error reading meta xattr: %s\n",
1140                                   strerror(errno)));
1141                         rc = -1;
1142                         goto exit;
1143                 }
1144         }
1145         if (ealen != AD_DATASZ_XATTR) {
1146                 DEBUG(2, ("bad size %zd\n", ealen));
1147                 errno = EINVAL;
1148                 rc = -1;
1149                 goto exit;
1150         }
1151
1152         /* Now parse entries */
1153         ok = ad_unpack(ad, ADEID_NUM_XATTR, AD_DATASZ_XATTR);
1154         if (!ok) {
1155                 DEBUG(2, ("invalid AppleDouble metadata xattr\n"));
1156                 errno = EINVAL;
1157                 rc = -1;
1158                 goto exit;
1159         }
1160
1161         if (!ad_getentryoff(ad, ADEID_FINDERI)
1162             || !ad_getentryoff(ad, ADEID_COMMENT)
1163             || !ad_getentryoff(ad, ADEID_FILEDATESI)
1164             || !ad_getentryoff(ad, ADEID_AFPFILEI)
1165             || !ad_getentryoff(ad, ADEID_PRIVDEV)
1166             || !ad_getentryoff(ad, ADEID_PRIVINO)
1167             || !ad_getentryoff(ad, ADEID_PRIVSYN)
1168             || !ad_getentryoff(ad, ADEID_PRIVID)) {
1169                 DEBUG(2, ("invalid AppleDouble metadata xattr\n"));
1170                 errno = EINVAL;
1171                 rc = -1;
1172                 goto exit;
1173         }
1174
1175 exit:
1176         DEBUG(10, ("reading meta xattr for %s, rc: %d\n",
1177                 smb_fname->base_name, rc));
1178
1179         if (rc != 0) {
1180                 ealen = -1;
1181                 if (errno == EINVAL) {
1182                         become_root();
1183                         removexattr(smb_fname->base_name, AFPINFO_EA_NETATALK);
1184                         unbecome_root();
1185                         errno = ENOENT;
1186                 }
1187         }
1188         return ealen;
1189 }
1190
1191 static int ad_open_rsrc_xattr(const struct smb_filename *smb_fname,
1192                                 int flags,
1193                                 mode_t mode)
1194 {
1195 #ifdef HAVE_ATTROPEN
1196         /* FIXME: direct Solaris xattr syscall */
1197         return attropen(smb_fname->base_name,
1198                         AFPRESOURCE_EA_NETATALK, flags, mode);
1199 #else
1200         errno = ENOSYS;
1201         return -1;
1202 #endif
1203 }
1204
1205 static int ad_open_rsrc_adouble(const struct smb_filename *smb_fname,
1206                                 int flags,
1207                                 mode_t mode)
1208 {
1209         int ret;
1210         int fd;
1211         struct smb_filename *adp_smb_fname = NULL;
1212
1213         ret = adouble_path(talloc_tos(), smb_fname, &adp_smb_fname);
1214         if (ret != 0) {
1215                 return -1;
1216         }
1217
1218         fd = open(adp_smb_fname->base_name, flags, mode);
1219         TALLOC_FREE(adp_smb_fname);
1220
1221         return fd;
1222 }
1223
1224 static int ad_open_rsrc(vfs_handle_struct *handle,
1225                         const struct smb_filename *smb_fname,
1226                         int flags,
1227                         mode_t mode)
1228 {
1229         struct fruit_config_data *config = NULL;
1230         int fd;
1231
1232         SMB_VFS_HANDLE_GET_DATA(handle, config,
1233                                 struct fruit_config_data, return -1);
1234
1235         if (config->rsrc == FRUIT_RSRC_XATTR) {
1236                 fd = ad_open_rsrc_xattr(smb_fname, flags, mode);
1237         } else {
1238                 fd = ad_open_rsrc_adouble(smb_fname, flags, mode);
1239         }
1240
1241         return fd;
1242 }
1243
1244 /*
1245  * Here's the deal: for ADOUBLE_META we can do without an fd as we can issue
1246  * path based xattr calls. For ADOUBLE_RSRC however we need a full-fledged fd
1247  * for file IO on the ._ file.
1248  */
1249 static int ad_open(vfs_handle_struct *handle,
1250                    struct adouble *ad,
1251                    files_struct *fsp,
1252                    const struct smb_filename *smb_fname,
1253                    int flags,
1254                    mode_t mode)
1255 {
1256         int fd;
1257
1258         DBG_DEBUG("Path [%s] type [%s]\n", smb_fname->base_name,
1259                   ad->ad_type == ADOUBLE_META ? "meta" : "rsrc");
1260
1261         if (ad->ad_type == ADOUBLE_META) {
1262                 return 0;
1263         }
1264
1265         if ((fsp != NULL) && (fsp->fh != NULL) && (fsp->fh->fd != -1)) {
1266                 ad->ad_fd = fsp->fh->fd;
1267                 ad->ad_opened = false;
1268                 return 0;
1269         }
1270
1271         fd = ad_open_rsrc(handle, smb_fname, flags, mode);
1272         if (fd == -1) {
1273                 return -1;
1274         }
1275         ad->ad_opened = true;
1276         ad->ad_fd = fd;
1277
1278         DBG_DEBUG("Path [%s] type [%s] fd [%d]\n",
1279                   smb_fname->base_name,
1280                   ad->ad_type == ADOUBLE_META ? "meta" : "rsrc", fd);
1281
1282         return 0;
1283 }
1284
1285 static ssize_t ad_read_rsrc_xattr(struct adouble *ad)
1286 {
1287         int ret;
1288         SMB_STRUCT_STAT st;
1289
1290         /* FIXME: direct sys_fstat(), don't have an fsp */
1291         ret = sys_fstat(ad->ad_fd, &st,
1292                         lp_fake_directory_create_times(
1293                                 SNUM(ad->ad_handle->conn)));
1294         if (ret != 0) {
1295                 return -1;
1296         }
1297
1298         ad_setentrylen(ad, ADEID_RFORK, st.st_ex_size);
1299         return st.st_ex_size;
1300 }
1301
1302 static ssize_t ad_read_rsrc_adouble(struct adouble *ad,
1303                                 const struct smb_filename *smb_fname)
1304 {
1305         SMB_STRUCT_STAT sbuf;
1306         char *p_ad = NULL;
1307         AfpInfo *ai = NULL;
1308         DATA_BLOB aiblob;
1309         struct smb_filename *stream_name = NULL;
1310         files_struct *fsp = NULL;
1311         ssize_t len;
1312         size_t size;
1313         ssize_t nwritten;
1314         NTSTATUS status;
1315         int saved_errno = 0;
1316         int ret;
1317         bool ok;
1318
1319         ret = sys_fstat(ad->ad_fd, &sbuf, lp_fake_directory_create_times(
1320                                 SNUM(ad->ad_handle->conn)));
1321         if (ret != 0) {
1322                 return -1;
1323         }
1324
1325         /*
1326          * AppleDouble file header content and size, two cases:
1327          *
1328          * - without xattrs it is exactly AD_DATASZ_DOT_UND (82) bytes large
1329          * - with embedded xattrs it can be larger, up to AD_XATTR_MAX_HDR_SIZE
1330          *
1331          * Read as much as we can up to AD_XATTR_MAX_HDR_SIZE.
1332          */
1333         size = sbuf.st_ex_size;
1334         if (size > talloc_array_length(ad->ad_data)) {
1335                 if (size > AD_XATTR_MAX_HDR_SIZE) {
1336                         size = AD_XATTR_MAX_HDR_SIZE;
1337                 }
1338                 p_ad = talloc_realloc(ad, ad->ad_data, char, size);
1339                 if (p_ad == NULL) {
1340                         return -1;
1341                 }
1342                 ad->ad_data = p_ad;
1343         }
1344
1345         len = sys_pread(ad->ad_fd, ad->ad_data,
1346                         talloc_array_length(ad->ad_data), 0);
1347         if (len != talloc_array_length(ad->ad_data)) {
1348                 DBG_NOTICE("%s %s: bad size: %zd\n",
1349                            smb_fname->base_name, strerror(errno), len);
1350                 return -1;
1351         }
1352
1353         /* Now parse entries */
1354         ok = ad_unpack(ad, ADEID_NUM_DOT_UND, sbuf.st_ex_size);
1355         if (!ok) {
1356                 DBG_ERR("invalid AppleDouble resource %s\n",
1357                         smb_fname->base_name);
1358                 errno = EINVAL;
1359                 return -1;
1360         }
1361
1362         if ((ad_getentryoff(ad, ADEID_FINDERI) != ADEDOFF_FINDERI_DOT_UND)
1363             || (ad_getentrylen(ad, ADEID_FINDERI) < ADEDLEN_FINDERI)
1364             || (ad_getentryoff(ad, ADEID_RFORK) < ADEDOFF_RFORK_DOT_UND)) {
1365                 DBG_ERR("invalid AppleDouble resource %s\n",
1366                         smb_fname->base_name);
1367                 errno = EINVAL;
1368                 return -1;
1369         }
1370
1371         if (ad_getentrylen(ad, ADEID_FINDERI) == ADEDLEN_FINDERI) {
1372                 return len;
1373         }
1374
1375         /*
1376          * Try to fixup AppleDouble files created by OS X with xattrs
1377          * appended to the ADEID_FINDERI entry. We simply remove the
1378          * xattrs blob, this means any fancy xattr that was stored
1379          * there is lost.
1380          */
1381
1382         ret = ad_convert(ad, smb_fname, ad->ad_fd);
1383         if (ret != 0) {
1384                 DBG_WARNING("Failed to convert [%s]\n", smb_fname->base_name);
1385                 return len;
1386         }
1387
1388         ok = ad_pack(ad);
1389         if (!ok) {
1390                 DBG_WARNING("ad_pack [%s] failed\n", smb_fname->base_name);
1391                 return -1;
1392         }
1393
1394         len = sys_pwrite(ad->ad_fd, ad->ad_data, AD_DATASZ_DOT_UND, 0);
1395         if (len != AD_DATASZ_DOT_UND) {
1396                 DBG_ERR("%s: bad size: %zd\n", smb_fname->base_name, len);
1397                 return -1;
1398         }
1399
1400         p_ad = ad_get_entry(ad, ADEID_FINDERI);
1401         if (p_ad == NULL) {
1402                 return -1;
1403         }
1404
1405         ai = afpinfo_new(talloc_tos());
1406         if (ai == NULL) {
1407                 return -1;
1408         }
1409
1410         memcpy(ai->afpi_FinderInfo, p_ad, ADEDLEN_FINDERI);
1411
1412         aiblob = data_blob_talloc(talloc_tos(), NULL, AFP_INFO_SIZE);
1413         if (aiblob.data == NULL) {
1414                 TALLOC_FREE(ai);
1415                 return -1;
1416         }
1417
1418         size = afpinfo_pack(ai, (char *)aiblob.data);
1419         TALLOC_FREE(ai);
1420         if (size != AFP_INFO_SIZE) {
1421                 return -1;
1422         }
1423
1424         stream_name = synthetic_smb_fname(talloc_tos(),
1425                                           smb_fname->base_name,
1426                                           AFPINFO_STREAM,
1427                                           NULL,
1428                                           smb_fname->flags);
1429         if (stream_name == NULL) {
1430                 data_blob_free(&aiblob);
1431                 DBG_ERR("synthetic_smb_fname failed\n");
1432                 return -1;
1433         }
1434
1435         DBG_DEBUG("stream_name: %s\n", smb_fname_str_dbg(stream_name));
1436
1437         status = SMB_VFS_CREATE_FILE(
1438                 ad->ad_handle->conn,            /* conn */
1439                 NULL,                           /* req */
1440                 0,                              /* root_dir_fid */
1441                 stream_name,                    /* fname */
1442                 FILE_GENERIC_WRITE,             /* access_mask */
1443                 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share_access */
1444                 FILE_OPEN_IF,                   /* create_disposition */
1445                 0,                              /* create_options */
1446                 0,                              /* file_attributes */
1447                 INTERNAL_OPEN_ONLY,             /* oplock_request */
1448                 NULL,                           /* lease */
1449                 0,                              /* allocation_size */
1450                 0,                              /* private_flags */
1451                 NULL,                           /* sd */
1452                 NULL,                           /* ea_list */
1453                 &fsp,                           /* result */
1454                 NULL,                           /* psbuf */
1455                 NULL, NULL);                    /* create context */
1456         TALLOC_FREE(stream_name);
1457         if (!NT_STATUS_IS_OK(status)) {
1458                 DBG_ERR("SMB_VFS_CREATE_FILE failed\n");
1459                 return -1;
1460         }
1461
1462         nwritten = SMB_VFS_PWRITE(fsp,
1463                                   aiblob.data,
1464                                   aiblob.length,
1465                                   0);
1466         if (nwritten == -1) {
1467                 DBG_ERR("SMB_VFS_PWRITE failed\n");
1468                 saved_errno = errno;
1469                 close_file(NULL, fsp, ERROR_CLOSE);
1470                 errno = saved_errno;
1471                 return -1;
1472         }
1473
1474         status = close_file(NULL, fsp, NORMAL_CLOSE);
1475         if (!NT_STATUS_IS_OK(status)) {
1476                 return -1;
1477         }
1478         fsp = NULL;
1479
1480         return len;
1481 }
1482
1483 /**
1484  * Read and parse resource fork, either ._ AppleDouble file or xattr
1485  **/
1486 static ssize_t ad_read_rsrc(struct adouble *ad,
1487                         const struct smb_filename *smb_fname)
1488 {
1489         struct fruit_config_data *config = NULL;
1490         ssize_t len;
1491
1492         SMB_VFS_HANDLE_GET_DATA(ad->ad_handle, config,
1493                                 struct fruit_config_data, return -1);
1494
1495         if (config->rsrc == FRUIT_RSRC_XATTR) {
1496                 len = ad_read_rsrc_xattr(ad);
1497         } else {
1498                 len = ad_read_rsrc_adouble(ad, smb_fname);
1499         }
1500
1501         return len;
1502 }
1503
1504 /**
1505  * Read and unpack an AppleDouble metadata xattr or resource
1506  **/
1507 static ssize_t ad_read(struct adouble *ad, const struct smb_filename *smb_fname)
1508 {
1509         switch (ad->ad_type) {
1510         case ADOUBLE_META:
1511                 return ad_read_meta(ad, smb_fname);
1512         case ADOUBLE_RSRC:
1513                 return ad_read_rsrc(ad, smb_fname);
1514         default:
1515                 return -1;
1516         }
1517 }
1518
1519 static int adouble_destructor(struct adouble *ad)
1520 {
1521         if ((ad->ad_fd != -1) && ad->ad_opened) {
1522                 close(ad->ad_fd);
1523                 ad->ad_fd = -1;
1524         }
1525         return 0;
1526 }
1527
1528 /**
1529  * Allocate a struct adouble without initialiing it
1530  *
1531  * The struct is either hang of the fsp extension context or if fsp is
1532  * NULL from ctx.
1533  *
1534  * @param[in] ctx        talloc context
1535  * @param[in] handle     vfs handle
1536  * @param[in] type       type of AppleDouble, ADOUBLE_META or ADOUBLE_RSRC
1537  *
1538  * @return               adouble handle
1539  **/
1540 static struct adouble *ad_alloc(TALLOC_CTX *ctx, vfs_handle_struct *handle,
1541                                 adouble_type_t type)
1542 {
1543         int rc = 0;
1544         size_t adsize = 0;
1545         struct adouble *ad;
1546         struct fruit_config_data *config;
1547
1548         SMB_VFS_HANDLE_GET_DATA(handle, config,
1549                                 struct fruit_config_data, return NULL);
1550
1551         switch (type) {
1552         case ADOUBLE_META:
1553                 adsize = AD_DATASZ_XATTR;
1554                 break;
1555         case ADOUBLE_RSRC:
1556                 if (config->rsrc == FRUIT_RSRC_ADFILE) {
1557                         adsize = AD_DATASZ_DOT_UND;
1558                 }
1559                 break;
1560         default:
1561                 return NULL;
1562         }
1563
1564         ad = talloc_zero(ctx, struct adouble);
1565         if (ad == NULL) {
1566                 rc = -1;
1567                 goto exit;
1568         }
1569
1570         if (adsize) {
1571                 ad->ad_data = talloc_zero_array(ad, char, adsize);
1572                 if (ad->ad_data == NULL) {
1573                         rc = -1;
1574                         goto exit;
1575                 }
1576         }
1577
1578         ad->ad_handle = handle;
1579         ad->ad_type = type;
1580         ad->ad_magic = AD_MAGIC;
1581         ad->ad_version = AD_VERSION;
1582         ad->ad_fd = -1;
1583
1584         talloc_set_destructor(ad, adouble_destructor);
1585
1586 exit:
1587         if (rc != 0) {
1588                 TALLOC_FREE(ad);
1589         }
1590         return ad;
1591 }
1592
1593 /**
1594  * Allocate and initialize a new struct adouble
1595  *
1596  * @param[in] ctx        talloc context
1597  * @param[in] handle     vfs handle
1598  * @param[in] type       type of AppleDouble, ADOUBLE_META or ADOUBLE_RSRC
1599  *
1600  * @return               adouble handle, initialized
1601  **/
1602 static struct adouble *ad_init(TALLOC_CTX *ctx, vfs_handle_struct *handle,
1603                                adouble_type_t type)
1604 {
1605         int rc = 0;
1606         const struct ad_entry_order  *eid;
1607         struct adouble *ad = NULL;
1608         struct fruit_config_data *config;
1609         time_t t = time(NULL);
1610
1611         SMB_VFS_HANDLE_GET_DATA(handle, config,
1612                                 struct fruit_config_data, return NULL);
1613
1614         switch (type) {
1615         case ADOUBLE_META:
1616                 eid = entry_order_meta_xattr;
1617                 break;
1618         case ADOUBLE_RSRC:
1619                 if (config->rsrc == FRUIT_RSRC_ADFILE) {
1620                         eid = entry_order_dot_und;
1621                 } else {
1622                         eid = entry_order_rsrc_xattr;
1623                 }
1624                 break;
1625         default:
1626                 return NULL;
1627         }
1628
1629         ad = ad_alloc(ctx, handle, type);
1630         if (ad == NULL) {
1631                 return NULL;
1632         }
1633
1634         while (eid->id) {
1635                 ad->ad_eid[eid->id].ade_off = eid->offset;
1636                 ad->ad_eid[eid->id].ade_len = eid->len;
1637                 eid++;
1638         }
1639
1640         /* put something sane in the date fields */
1641         ad_setdate(ad, AD_DATE_CREATE | AD_DATE_UNIX, t);
1642         ad_setdate(ad, AD_DATE_MODIFY | AD_DATE_UNIX, t);
1643         ad_setdate(ad, AD_DATE_ACCESS | AD_DATE_UNIX, t);
1644         ad_setdate(ad, AD_DATE_BACKUP, htonl(AD_DATE_START));
1645
1646         if (rc != 0) {
1647                 TALLOC_FREE(ad);
1648         }
1649         return ad;
1650 }
1651
1652 static struct adouble *ad_get_internal(TALLOC_CTX *ctx,
1653                                        vfs_handle_struct *handle,
1654                                        files_struct *fsp,
1655                                        const struct smb_filename *smb_fname,
1656                                        adouble_type_t type)
1657 {
1658         int rc = 0;
1659         ssize_t len;
1660         struct adouble *ad = NULL;
1661         int mode;
1662
1663         if (fsp != NULL) {
1664                 smb_fname = fsp->base_fsp->fsp_name;
1665         }
1666
1667         DEBUG(10, ("ad_get(%s) called for %s\n",
1668                    type == ADOUBLE_META ? "meta" : "rsrc",
1669                    smb_fname->base_name));
1670
1671         ad = ad_alloc(ctx, handle, type);
1672         if (ad == NULL) {
1673                 rc = -1;
1674                 goto exit;
1675         }
1676
1677         /* Try rw first so we can use the fd in ad_convert() */
1678         mode = O_RDWR;
1679
1680         rc = ad_open(handle, ad, fsp, smb_fname, mode, 0);
1681         if (rc == -1 && ((errno == EROFS) || (errno == EACCES))) {
1682                 mode = O_RDONLY;
1683                 rc = ad_open(handle, ad, fsp, smb_fname, mode, 0);
1684         }
1685         if (rc == -1) {
1686                 DBG_DEBUG("ad_open [%s] error [%s]\n",
1687                           smb_fname->base_name, strerror(errno));
1688                 goto exit;
1689
1690         }
1691
1692         len = ad_read(ad, smb_fname);
1693         if (len == -1) {
1694                 DEBUG(10, ("error reading AppleDouble for %s\n",
1695                         smb_fname->base_name));
1696                 rc = -1;
1697                 goto exit;
1698         }
1699
1700 exit:
1701         DEBUG(10, ("ad_get(%s) for %s returning %d\n",
1702                   type == ADOUBLE_META ? "meta" : "rsrc",
1703                   smb_fname->base_name, rc));
1704
1705         if (rc != 0) {
1706                 TALLOC_FREE(ad);
1707         }
1708         return ad;
1709 }
1710
1711 /**
1712  * Return AppleDouble data for a file
1713  *
1714  * @param[in] ctx      talloc context
1715  * @param[in] handle   vfs handle
1716  * @param[in] smb_fname pathname to file or directory
1717  * @param[in] type     type of AppleDouble, ADOUBLE_META or ADOUBLE_RSRC
1718  *
1719  * @return             talloced struct adouble or NULL on error
1720  **/
1721 static struct adouble *ad_get(TALLOC_CTX *ctx,
1722                               vfs_handle_struct *handle,
1723                               const struct smb_filename *smb_fname,
1724                               adouble_type_t type)
1725 {
1726         return ad_get_internal(ctx, handle, NULL, smb_fname, type);
1727 }
1728
1729 /**
1730  * Return AppleDouble data for a file
1731  *
1732  * @param[in] ctx      talloc context
1733  * @param[in] handle   vfs handle
1734  * @param[in] fsp      fsp to use for IO
1735  * @param[in] type     type of AppleDouble, ADOUBLE_META or ADOUBLE_RSRC
1736  *
1737  * @return             talloced struct adouble or NULL on error
1738  **/
1739 static struct adouble *ad_fget(TALLOC_CTX *ctx, vfs_handle_struct *handle,
1740                                files_struct *fsp, adouble_type_t type)
1741 {
1742         return ad_get_internal(ctx, handle, fsp, NULL, type);
1743 }
1744
1745 /**
1746  * Set AppleDouble metadata on a file or directory
1747  *
1748  * @param[in] ad      adouble handle
1749  *
1750  * @param[in] smb_fname    pathname to file or directory
1751  *
1752  * @return            status code, 0 means success
1753  **/
1754 static int ad_set(struct adouble *ad, const struct smb_filename *smb_fname)
1755 {
1756         bool ok;
1757         int ret;
1758
1759         DBG_DEBUG("Path [%s]\n", smb_fname->base_name);
1760
1761         if (ad->ad_type != ADOUBLE_META) {
1762                 DBG_ERR("ad_set on [%s] used with ADOUBLE_RSRC\n",
1763                         smb_fname->base_name);
1764                 return -1;
1765         }
1766
1767         ok = ad_pack(ad);
1768         if (!ok) {
1769                 return -1;
1770         }
1771
1772         ret = SMB_VFS_SETXATTR(ad->ad_handle->conn,
1773                                smb_fname,
1774                                AFPINFO_EA_NETATALK,
1775                                ad->ad_data,
1776                                AD_DATASZ_XATTR, 0);
1777
1778         DBG_DEBUG("Path [%s] ret [%d]\n", smb_fname->base_name, ret);
1779
1780         return ret;
1781 }
1782
1783 /**
1784  * Set AppleDouble metadata on a file or directory
1785  *
1786  * @param[in] ad      adouble handle
1787  * @param[in] fsp     file handle
1788  *
1789  * @return            status code, 0 means success
1790  **/
1791 static int ad_fset(struct adouble *ad, files_struct *fsp)
1792 {
1793         int rc = -1;
1794         ssize_t len;
1795         bool ok;
1796
1797         DBG_DEBUG("Path [%s]\n", fsp_str_dbg(fsp));
1798
1799         if ((fsp == NULL)
1800             || (fsp->fh == NULL)
1801             || (fsp->fh->fd == -1))
1802         {
1803                 smb_panic("bad fsp");
1804         }
1805
1806         ok = ad_pack(ad);
1807         if (!ok) {
1808                 return -1;
1809         }
1810
1811         switch (ad->ad_type) {
1812         case ADOUBLE_META:
1813                 rc = SMB_VFS_NEXT_SETXATTR(ad->ad_handle,
1814                                            fsp->fsp_name,
1815                                            AFPINFO_EA_NETATALK,
1816                                            ad->ad_data,
1817                                            AD_DATASZ_XATTR, 0);
1818                 break;
1819
1820         case ADOUBLE_RSRC:
1821                 len = SMB_VFS_NEXT_PWRITE(ad->ad_handle,
1822                                           fsp,
1823                                           ad->ad_data,
1824                                           AD_DATASZ_DOT_UND,
1825                                           0);
1826                 if (len != AD_DATASZ_DOT_UND) {
1827                         DBG_ERR("short write on %s: %zd", fsp_str_dbg(fsp), len);
1828                         return -1;
1829                 }
1830                 rc = 0;
1831                 break;
1832
1833         default:
1834                 return -1;
1835         }
1836
1837         DBG_DEBUG("Path [%s] rc [%d]\n", fsp_str_dbg(fsp), rc);
1838
1839         return rc;
1840 }
1841
1842 /*****************************************************************************
1843  * Helper functions
1844  *****************************************************************************/
1845
1846 static bool is_afpinfo_stream(const struct smb_filename *smb_fname)
1847 {
1848         if (strncasecmp_m(smb_fname->stream_name,
1849                           AFPINFO_STREAM_NAME,
1850                           strlen(AFPINFO_STREAM_NAME)) == 0) {
1851                 return true;
1852         }
1853         return false;
1854 }
1855
1856 static bool is_afpresource_stream(const struct smb_filename *smb_fname)
1857 {
1858         if (strncasecmp_m(smb_fname->stream_name,
1859                           AFPRESOURCE_STREAM_NAME,
1860                           strlen(AFPRESOURCE_STREAM_NAME)) == 0) {
1861                 return true;
1862         }
1863         return false;
1864 }
1865
1866 /**
1867  * Test whether stream is an Apple stream, not used atm
1868  **/
1869 #if 0
1870 static bool is_apple_stream(const struct smb_filename *smb_fname)
1871 {
1872         if (is_afpinfo_stream(smb_fname)) {
1873                 return true;
1874         }
1875         if (is_afpresource_stream(smb_fname)) {
1876                 return true;
1877         }
1878         return false;
1879 }
1880 #endif
1881
1882 /**
1883  * Initialize config struct from our smb.conf config parameters
1884  **/
1885 static int init_fruit_config(vfs_handle_struct *handle)
1886 {
1887         struct fruit_config_data *config;
1888         int enumval;
1889         const char *tm_size_str = NULL;
1890
1891         config = talloc_zero(handle->conn, struct fruit_config_data);
1892         if (!config) {
1893                 DEBUG(1, ("talloc_zero() failed\n"));
1894                 errno = ENOMEM;
1895                 return -1;
1896         }
1897
1898         /*
1899          * Versions up to Samba 4.5.x had a spelling bug in the
1900          * fruit:resource option calling lp_parm_enum with
1901          * "res*s*ource" (ie two s).
1902          *
1903          * In Samba 4.6 we accept both the wrong and the correct
1904          * spelling, in Samba 4.7 the bad spelling will be removed.
1905          */
1906         enumval = lp_parm_enum(SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME,
1907                                "ressource", fruit_rsrc, FRUIT_RSRC_ADFILE);
1908         if (enumval == -1) {
1909                 DEBUG(1, ("value for %s: resource type unknown\n",
1910                           FRUIT_PARAM_TYPE_NAME));
1911                 return -1;
1912         }
1913         config->rsrc = (enum fruit_rsrc)enumval;
1914
1915         enumval = lp_parm_enum(SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME,
1916                                "resource", fruit_rsrc, enumval);
1917         if (enumval == -1) {
1918                 DEBUG(1, ("value for %s: resource type unknown\n",
1919                           FRUIT_PARAM_TYPE_NAME));
1920                 return -1;
1921         }
1922         config->rsrc = (enum fruit_rsrc)enumval;
1923
1924         enumval = lp_parm_enum(SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME,
1925                                "metadata", fruit_meta, FRUIT_META_NETATALK);
1926         if (enumval == -1) {
1927                 DEBUG(1, ("value for %s: metadata type unknown\n",
1928                           FRUIT_PARAM_TYPE_NAME));
1929                 return -1;
1930         }
1931         config->meta = (enum fruit_meta)enumval;
1932
1933         enumval = lp_parm_enum(SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME,
1934                                "locking", fruit_locking, FRUIT_LOCKING_NONE);
1935         if (enumval == -1) {
1936                 DEBUG(1, ("value for %s: locking type unknown\n",
1937                           FRUIT_PARAM_TYPE_NAME));
1938                 return -1;
1939         }
1940         config->locking = (enum fruit_locking)enumval;
1941
1942         enumval = lp_parm_enum(SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME,
1943                                "encoding", fruit_encoding, FRUIT_ENC_PRIVATE);
1944         if (enumval == -1) {
1945                 DEBUG(1, ("value for %s: encoding type unknown\n",
1946                           FRUIT_PARAM_TYPE_NAME));
1947                 return -1;
1948         }
1949         config->encoding = (enum fruit_encoding)enumval;
1950
1951         if (config->rsrc == FRUIT_RSRC_ADFILE) {
1952                 config->veto_appledouble = lp_parm_bool(SNUM(handle->conn),
1953                                                         FRUIT_PARAM_TYPE_NAME,
1954                                                         "veto_appledouble",
1955                                                         true);
1956         }
1957
1958         config->use_aapl = lp_parm_bool(
1959                 -1, FRUIT_PARAM_TYPE_NAME, "aapl", true);
1960
1961         config->time_machine = lp_parm_bool(
1962                 SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME, "time machine", false);
1963
1964         config->unix_info_enabled = lp_parm_bool(
1965                 -1, FRUIT_PARAM_TYPE_NAME, "nfs_aces", true);
1966
1967         config->use_copyfile = lp_parm_bool(-1, FRUIT_PARAM_TYPE_NAME,
1968                                            "copyfile", false);
1969
1970         config->posix_rename = lp_parm_bool(
1971                 SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME, "posix_rename", true);
1972
1973         config->aapl_zero_file_id =
1974             lp_parm_bool(-1, FRUIT_PARAM_TYPE_NAME, "zero_file_id", true);
1975
1976         config->readdir_attr_rsize = lp_parm_bool(
1977                 SNUM(handle->conn), "readdir_attr", "aapl_rsize", true);
1978
1979         config->readdir_attr_finder_info = lp_parm_bool(
1980                 SNUM(handle->conn), "readdir_attr", "aapl_finder_info", true);
1981
1982         config->readdir_attr_max_access = lp_parm_bool(
1983                 SNUM(handle->conn), "readdir_attr", "aapl_max_access", true);
1984
1985         config->model = lp_parm_const_string(
1986                 -1, FRUIT_PARAM_TYPE_NAME, "model", "MacSamba");
1987
1988         tm_size_str = lp_parm_const_string(
1989                 SNUM(handle->conn), FRUIT_PARAM_TYPE_NAME,
1990                 "time machine max size", NULL);
1991         if (tm_size_str != NULL) {
1992                 config->time_machine_max_size = conv_str_size(tm_size_str);
1993         }
1994
1995         SMB_VFS_HANDLE_SET_DATA(handle, config,
1996                                 NULL, struct fruit_config_data,
1997                                 return -1);
1998
1999         return 0;
2000 }
2001
2002 /**
2003  * Prepend "._" to a basename
2004  * Return a new struct smb_filename with stream_name == NULL.
2005  **/
2006 static int adouble_path(TALLOC_CTX *ctx,
2007                         const struct smb_filename *smb_fname_in,
2008                         struct smb_filename **pp_smb_fname_out)
2009 {
2010         char *parent;
2011         const char *base;
2012         struct smb_filename *smb_fname = cp_smb_filename(ctx,
2013                                                 smb_fname_in);
2014
2015         if (smb_fname == NULL) {
2016                 return -1;
2017         }
2018
2019         /* We need streamname to be NULL */
2020         TALLOC_FREE(smb_fname->stream_name);
2021
2022         /* And we're replacing base_name. */
2023         TALLOC_FREE(smb_fname->base_name);
2024
2025         if (!parent_dirname(smb_fname, smb_fname_in->base_name,
2026                                 &parent, &base)) {
2027                 TALLOC_FREE(smb_fname);
2028                 return -1;
2029         }
2030
2031         smb_fname->base_name = talloc_asprintf(smb_fname,
2032                                         "%s/._%s", parent, base);
2033         if (smb_fname->base_name == NULL) {
2034                 TALLOC_FREE(smb_fname);
2035                 return -1;
2036         }
2037
2038         *pp_smb_fname_out = smb_fname;
2039
2040         return 0;
2041 }
2042
2043 /**
2044  * Allocate and initialize an AfpInfo struct
2045  **/
2046 static AfpInfo *afpinfo_new(TALLOC_CTX *ctx)
2047 {
2048         AfpInfo *ai = talloc_zero(ctx, AfpInfo);
2049         if (ai == NULL) {
2050                 return NULL;
2051         }
2052         ai->afpi_Signature = AFP_Signature;
2053         ai->afpi_Version = AFP_Version;
2054         ai->afpi_BackupTime = AD_DATE_START;
2055         return ai;
2056 }
2057
2058 /**
2059  * Pack an AfpInfo struct into a buffer
2060  *
2061  * Buffer size must be at least AFP_INFO_SIZE
2062  * Returns size of packed buffer
2063  **/
2064 static ssize_t afpinfo_pack(const AfpInfo *ai, char *buf)
2065 {
2066         memset(buf, 0, AFP_INFO_SIZE);
2067
2068         RSIVAL(buf, 0, ai->afpi_Signature);
2069         RSIVAL(buf, 4, ai->afpi_Version);
2070         RSIVAL(buf, 12, ai->afpi_BackupTime);
2071         memcpy(buf + 16, ai->afpi_FinderInfo, sizeof(ai->afpi_FinderInfo));
2072
2073         return AFP_INFO_SIZE;
2074 }
2075
2076 /**
2077  * Unpack a buffer into a AfpInfo structure
2078  *
2079  * Buffer size must be at least AFP_INFO_SIZE
2080  * Returns allocated AfpInfo struct
2081  **/
2082 static AfpInfo *afpinfo_unpack(TALLOC_CTX *ctx, const void *data)
2083 {
2084         AfpInfo *ai = talloc_zero(ctx, AfpInfo);
2085         if (ai == NULL) {
2086                 return NULL;
2087         }
2088
2089         ai->afpi_Signature = RIVAL(data, 0);
2090         ai->afpi_Version = RIVAL(data, 4);
2091         ai->afpi_BackupTime = RIVAL(data, 12);
2092         memcpy(ai->afpi_FinderInfo, (const char *)data + 16,
2093                sizeof(ai->afpi_FinderInfo));
2094
2095         if (ai->afpi_Signature != AFP_Signature
2096             || ai->afpi_Version != AFP_Version) {
2097                 DEBUG(1, ("Bad AfpInfo signature or version\n"));
2098                 TALLOC_FREE(ai);
2099         }
2100
2101         return ai;
2102 }
2103
2104 /**
2105  * Fake an inode number from the md5 hash of the (xattr) name
2106  **/
2107 static SMB_INO_T fruit_inode(const SMB_STRUCT_STAT *sbuf, const char *sname)
2108 {
2109         MD5_CTX ctx;
2110         unsigned char hash[16];
2111         SMB_INO_T result;
2112         char *upper_sname;
2113
2114         upper_sname = talloc_strdup_upper(talloc_tos(), sname);
2115         SMB_ASSERT(upper_sname != NULL);
2116
2117         MD5Init(&ctx);
2118         MD5Update(&ctx, (const unsigned char *)&(sbuf->st_ex_dev),
2119                   sizeof(sbuf->st_ex_dev));
2120         MD5Update(&ctx, (const unsigned char *)&(sbuf->st_ex_ino),
2121                   sizeof(sbuf->st_ex_ino));
2122         MD5Update(&ctx, (unsigned char *)upper_sname,
2123                   talloc_get_size(upper_sname)-1);
2124         MD5Final(hash, &ctx);
2125
2126         TALLOC_FREE(upper_sname);
2127
2128         /* Hopefully all the variation is in the lower 4 (or 8) bytes! */
2129         memcpy(&result, hash, sizeof(result));
2130
2131         DEBUG(10, ("fruit_inode \"%s\": ino=0x%llu\n",
2132                    sname, (unsigned long long)result));
2133
2134         return result;
2135 }
2136
2137 static bool add_fruit_stream(TALLOC_CTX *mem_ctx, unsigned int *num_streams,
2138                              struct stream_struct **streams,
2139                              const char *name, off_t size,
2140                              off_t alloc_size)
2141 {
2142         struct stream_struct *tmp;
2143
2144         tmp = talloc_realloc(mem_ctx, *streams, struct stream_struct,
2145                              (*num_streams)+1);
2146         if (tmp == NULL) {
2147                 return false;
2148         }
2149
2150         tmp[*num_streams].name = talloc_asprintf(tmp, "%s:$DATA", name);
2151         if (tmp[*num_streams].name == NULL) {
2152                 return false;
2153         }
2154
2155         tmp[*num_streams].size = size;
2156         tmp[*num_streams].alloc_size = alloc_size;
2157
2158         *streams = tmp;
2159         *num_streams += 1;
2160         return true;
2161 }
2162
2163 static bool filter_empty_rsrc_stream(unsigned int *num_streams,
2164                                      struct stream_struct **streams)
2165 {
2166         struct stream_struct *tmp = *streams;
2167         unsigned int i;
2168
2169         if (*num_streams == 0) {
2170                 return true;
2171         }
2172
2173         for (i = 0; i < *num_streams; i++) {
2174                 if (strequal_m(tmp[i].name, AFPRESOURCE_STREAM)) {
2175                         break;
2176                 }
2177         }
2178
2179         if (i == *num_streams) {
2180                 return true;
2181         }
2182
2183         if (tmp[i].size > 0) {
2184                 return true;
2185         }
2186
2187         TALLOC_FREE(tmp[i].name);
2188         if (*num_streams - 1 > i) {
2189                 memmove(&tmp[i], &tmp[i+1],
2190                         (*num_streams - i - 1) * sizeof(struct stream_struct));
2191         }
2192
2193         *num_streams -= 1;
2194         return true;
2195 }
2196
2197 static bool del_fruit_stream(TALLOC_CTX *mem_ctx, unsigned int *num_streams,
2198                              struct stream_struct **streams,
2199                              const char *name)
2200 {
2201         struct stream_struct *tmp = *streams;
2202         unsigned int i;
2203
2204         if (*num_streams == 0) {
2205                 return true;
2206         }
2207
2208         for (i = 0; i < *num_streams; i++) {
2209                 if (strequal_m(tmp[i].name, name)) {
2210                         break;
2211                 }
2212         }
2213
2214         if (i == *num_streams) {
2215                 return true;
2216         }
2217
2218         TALLOC_FREE(tmp[i].name);
2219         if (*num_streams - 1 > i) {
2220                 memmove(&tmp[i], &tmp[i+1],
2221                         (*num_streams - i - 1) * sizeof(struct stream_struct));
2222         }
2223
2224         *num_streams -= 1;
2225         return true;
2226 }
2227
2228 static bool ad_empty_finderinfo(const struct adouble *ad)
2229 {
2230         int cmp;
2231         char emptybuf[ADEDLEN_FINDERI] = {0};
2232         char *fi = NULL;
2233
2234         fi = ad_get_entry(ad, ADEID_FINDERI);
2235         if (fi == NULL) {
2236                 DBG_ERR("Missing FinderInfo in struct adouble [%p]\n", ad);
2237                 return false;
2238         }
2239
2240         cmp = memcmp(emptybuf, fi, ADEDLEN_FINDERI);
2241         return (cmp == 0);
2242 }
2243
2244 static bool ai_empty_finderinfo(const AfpInfo *ai)
2245 {
2246         int cmp;
2247         char emptybuf[ADEDLEN_FINDERI] = {0};
2248
2249         cmp = memcmp(emptybuf, &ai->afpi_FinderInfo[0], ADEDLEN_FINDERI);
2250         return (cmp == 0);
2251 }
2252
2253 /**
2254  * Update btime with btime from Netatalk
2255  **/
2256 static void update_btime(vfs_handle_struct *handle,
2257                          struct smb_filename *smb_fname)
2258 {
2259         uint32_t t;
2260         struct timespec creation_time = {0};
2261         struct adouble *ad;
2262         struct fruit_config_data *config = NULL;
2263
2264         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
2265                                 return);
2266
2267         switch (config->meta) {
2268         case FRUIT_META_STREAM:
2269                 return;
2270         case FRUIT_META_NETATALK:
2271                 /* Handled below */
2272                 break;
2273         default:
2274                 DBG_ERR("Unexpected meta config [%d]\n", config->meta);
2275                 return;
2276         }
2277
2278         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_META);
2279         if (ad == NULL) {
2280                 return;
2281         }
2282         if (ad_getdate(ad, AD_DATE_UNIX | AD_DATE_CREATE, &t) != 0) {
2283                 TALLOC_FREE(ad);
2284                 return;
2285         }
2286         TALLOC_FREE(ad);
2287
2288         creation_time.tv_sec = convert_uint32_t_to_time_t(t);
2289         update_stat_ex_create_time(&smb_fname->st, creation_time);
2290
2291         return;
2292 }
2293
2294 /**
2295  * Map an access mask to a Netatalk single byte byte range lock
2296  **/
2297 static off_t access_to_netatalk_brl(enum apple_fork fork_type,
2298                                     uint32_t access_mask)
2299 {
2300         off_t offset;
2301
2302         switch (access_mask) {
2303         case FILE_READ_DATA:
2304                 offset = AD_FILELOCK_OPEN_RD;
2305                 break;
2306
2307         case FILE_WRITE_DATA:
2308         case FILE_APPEND_DATA:
2309                 offset = AD_FILELOCK_OPEN_WR;
2310                 break;
2311
2312         default:
2313                 offset = AD_FILELOCK_OPEN_NONE;
2314                 break;
2315         }
2316
2317         if (fork_type == APPLE_FORK_RSRC) {
2318                 if (offset == AD_FILELOCK_OPEN_NONE) {
2319                         offset = AD_FILELOCK_RSRC_OPEN_NONE;
2320                 } else {
2321                         offset += 2;
2322                 }
2323         }
2324
2325         return offset;
2326 }
2327
2328 /**
2329  * Map a deny mode to a Netatalk brl
2330  **/
2331 static off_t denymode_to_netatalk_brl(enum apple_fork fork_type,
2332                                       uint32_t deny_mode)
2333 {
2334         off_t offset;
2335
2336         switch (deny_mode) {
2337         case DENY_READ:
2338                 offset = AD_FILELOCK_DENY_RD;
2339                 break;
2340
2341         case DENY_WRITE:
2342                 offset = AD_FILELOCK_DENY_WR;
2343                 break;
2344
2345         default:
2346                 smb_panic("denymode_to_netatalk_brl: bad deny mode\n");
2347         }
2348
2349         if (fork_type == APPLE_FORK_RSRC) {
2350                 offset += 2;
2351         }
2352
2353         return offset;
2354 }
2355
2356 /**
2357  * Call fcntl() with an exclusive F_GETLK request in order to
2358  * determine if there's an exisiting shared lock
2359  *
2360  * @return true if the requested lock was found or any error occurred
2361  *         false if the lock was not found
2362  **/
2363 static bool test_netatalk_lock(files_struct *fsp, off_t in_offset)
2364 {
2365         bool result;
2366         off_t offset = in_offset;
2367         off_t len = 1;
2368         int type = F_WRLCK;
2369         pid_t pid;
2370
2371         result = SMB_VFS_GETLOCK(fsp, &offset, &len, &type, &pid);
2372         if (result == false) {
2373                 return true;
2374         }
2375
2376         if (type != F_UNLCK) {
2377                 return true;
2378         }
2379
2380         return false;
2381 }
2382
2383 static NTSTATUS fruit_check_access(vfs_handle_struct *handle,
2384                                    files_struct *fsp,
2385                                    uint32_t access_mask,
2386                                    uint32_t deny_mode)
2387 {
2388         NTSTATUS status = NT_STATUS_OK;
2389         struct byte_range_lock *br_lck = NULL;
2390         bool open_for_reading, open_for_writing, deny_read, deny_write;
2391         off_t off;
2392         bool have_read = false;
2393         int flags;
2394
2395         /* FIXME: hardcoded data fork, add resource fork */
2396         enum apple_fork fork_type = APPLE_FORK_DATA;
2397
2398         DEBUG(10, ("fruit_check_access: %s, am: %s/%s, dm: %s/%s\n",
2399                   fsp_str_dbg(fsp),
2400                   access_mask & FILE_READ_DATA ? "READ" :"-",
2401                   access_mask & FILE_WRITE_DATA ? "WRITE" : "-",
2402                   deny_mode & DENY_READ ? "DENY_READ" : "-",
2403                   deny_mode & DENY_WRITE ? "DENY_WRITE" : "-"));
2404
2405         if (fsp->fh->fd == -1) {
2406                 return NT_STATUS_OK;
2407         }
2408
2409         flags = fcntl(fsp->fh->fd, F_GETFL);
2410         if (flags == -1) {
2411                 DBG_ERR("fcntl get flags [%s] fd [%d] failed [%s]\n",
2412                         fsp_str_dbg(fsp), fsp->fh->fd, strerror(errno));
2413                 return map_nt_error_from_unix(errno);
2414         }
2415
2416         if (flags & (O_RDONLY|O_RDWR)) {
2417                 /*
2418                  * Applying fcntl read locks requires an fd opened for
2419                  * reading. This means we won't be applying locks for
2420                  * files openend write-only, but what can we do...
2421                  */
2422                 have_read = true;
2423         }
2424
2425         /*
2426          * Check read access and deny read mode
2427          */
2428         if ((access_mask & FILE_READ_DATA) || (deny_mode & DENY_READ)) {
2429                 /* Check access */
2430                 open_for_reading = test_netatalk_lock(
2431                         fsp, access_to_netatalk_brl(fork_type, FILE_READ_DATA));
2432
2433                 deny_read = test_netatalk_lock(
2434                         fsp, denymode_to_netatalk_brl(fork_type, DENY_READ));
2435
2436                 DEBUG(10, ("read: %s, deny_write: %s\n",
2437                           open_for_reading == true ? "yes" : "no",
2438                           deny_read == true ? "yes" : "no"));
2439
2440                 if (((access_mask & FILE_READ_DATA) && deny_read)
2441                     || ((deny_mode & DENY_READ) && open_for_reading)) {
2442                         return NT_STATUS_SHARING_VIOLATION;
2443                 }
2444
2445                 /* Set locks */
2446                 if ((access_mask & FILE_READ_DATA) && have_read) {
2447                         off = access_to_netatalk_brl(fork_type, FILE_READ_DATA);
2448                         br_lck = do_lock(
2449                                 handle->conn->sconn->msg_ctx, fsp,
2450                                 fsp->op->global->open_persistent_id, 1, off,
2451                                 READ_LOCK, POSIX_LOCK, false,
2452                                 &status, NULL);
2453
2454                         if (!NT_STATUS_IS_OK(status))  {
2455                                 return status;
2456                         }
2457                         TALLOC_FREE(br_lck);
2458                 }
2459
2460                 if ((deny_mode & DENY_READ) && have_read) {
2461                         off = denymode_to_netatalk_brl(fork_type, DENY_READ);
2462                         br_lck = do_lock(
2463                                 handle->conn->sconn->msg_ctx, fsp,
2464                                 fsp->op->global->open_persistent_id, 1, off,
2465                                 READ_LOCK, POSIX_LOCK, false,
2466                                 &status, NULL);
2467
2468                         if (!NT_STATUS_IS_OK(status)) {
2469                                 return status;
2470                         }
2471                         TALLOC_FREE(br_lck);
2472                 }
2473         }
2474
2475         /*
2476          * Check write access and deny write mode
2477          */
2478         if ((access_mask & FILE_WRITE_DATA) || (deny_mode & DENY_WRITE)) {
2479                 /* Check access */
2480                 open_for_writing = test_netatalk_lock(
2481                         fsp, access_to_netatalk_brl(fork_type, FILE_WRITE_DATA));
2482
2483                 deny_write = test_netatalk_lock(
2484                         fsp, denymode_to_netatalk_brl(fork_type, DENY_WRITE));
2485
2486                 DEBUG(10, ("write: %s, deny_write: %s\n",
2487                           open_for_writing == true ? "yes" : "no",
2488                           deny_write == true ? "yes" : "no"));
2489
2490                 if (((access_mask & FILE_WRITE_DATA) && deny_write)
2491                     || ((deny_mode & DENY_WRITE) && open_for_writing)) {
2492                         return NT_STATUS_SHARING_VIOLATION;
2493                 }
2494
2495                 /* Set locks */
2496                 if ((access_mask & FILE_WRITE_DATA) && have_read) {
2497                         off = access_to_netatalk_brl(fork_type, FILE_WRITE_DATA);
2498                         br_lck = do_lock(
2499                                 handle->conn->sconn->msg_ctx, fsp,
2500                                 fsp->op->global->open_persistent_id, 1, off,
2501                                 READ_LOCK, POSIX_LOCK, false,
2502                                 &status, NULL);
2503
2504                         if (!NT_STATUS_IS_OK(status)) {
2505                                 return status;
2506                         }
2507                         TALLOC_FREE(br_lck);
2508
2509                 }
2510                 if ((deny_mode & DENY_WRITE) && have_read) {
2511                         off = denymode_to_netatalk_brl(fork_type, DENY_WRITE);
2512                         br_lck = do_lock(
2513                                 handle->conn->sconn->msg_ctx, fsp,
2514                                 fsp->op->global->open_persistent_id, 1, off,
2515                                 READ_LOCK, POSIX_LOCK, false,
2516                                 &status, NULL);
2517
2518                         if (!NT_STATUS_IS_OK(status)) {
2519                                 return status;
2520                         }
2521                         TALLOC_FREE(br_lck);
2522                 }
2523         }
2524
2525         TALLOC_FREE(br_lck);
2526
2527         return status;
2528 }
2529
2530 static NTSTATUS check_aapl(vfs_handle_struct *handle,
2531                            struct smb_request *req,
2532                            const struct smb2_create_blobs *in_context_blobs,
2533                            struct smb2_create_blobs *out_context_blobs)
2534 {
2535         struct fruit_config_data *config;
2536         NTSTATUS status;
2537         struct smb2_create_blob *aapl = NULL;
2538         uint32_t cmd;
2539         bool ok;
2540         uint8_t p[16];
2541         DATA_BLOB blob = data_blob_talloc(req, NULL, 0);
2542         uint64_t req_bitmap, client_caps;
2543         uint64_t server_caps = SMB2_CRTCTX_AAPL_UNIX_BASED;
2544         smb_ucs2_t *model;
2545         size_t modellen;
2546
2547         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
2548                                 return NT_STATUS_UNSUCCESSFUL);
2549
2550         if (!config->use_aapl
2551             || in_context_blobs == NULL
2552             || out_context_blobs == NULL) {
2553                 return NT_STATUS_OK;
2554         }
2555
2556         aapl = smb2_create_blob_find(in_context_blobs,
2557                                      SMB2_CREATE_TAG_AAPL);
2558         if (aapl == NULL) {
2559                 return NT_STATUS_OK;
2560         }
2561
2562         if (aapl->data.length != 24) {
2563                 DEBUG(1, ("unexpected AAPL ctxt length: %ju\n",
2564                           (uintmax_t)aapl->data.length));
2565                 return NT_STATUS_INVALID_PARAMETER;
2566         }
2567
2568         cmd = IVAL(aapl->data.data, 0);
2569         if (cmd != SMB2_CRTCTX_AAPL_SERVER_QUERY) {
2570                 DEBUG(1, ("unsupported AAPL cmd: %d\n", cmd));
2571                 return NT_STATUS_INVALID_PARAMETER;
2572         }
2573
2574         req_bitmap = BVAL(aapl->data.data, 8);
2575         client_caps = BVAL(aapl->data.data, 16);
2576
2577         SIVAL(p, 0, SMB2_CRTCTX_AAPL_SERVER_QUERY);
2578         SIVAL(p, 4, 0);
2579         SBVAL(p, 8, req_bitmap);
2580         ok = data_blob_append(req, &blob, p, 16);
2581         if (!ok) {
2582                 return NT_STATUS_UNSUCCESSFUL;
2583         }
2584
2585         if (req_bitmap & SMB2_CRTCTX_AAPL_SERVER_CAPS) {
2586                 if ((client_caps & SMB2_CRTCTX_AAPL_SUPPORTS_READ_DIR_ATTR) &&
2587                     (handle->conn->tcon->compat->fs_capabilities & FILE_NAMED_STREAMS)) {
2588                         server_caps |= SMB2_CRTCTX_AAPL_SUPPORTS_READ_DIR_ATTR;
2589                         config->readdir_attr_enabled = true;
2590                 }
2591
2592                 if (config->use_copyfile) {
2593                         server_caps |= SMB2_CRTCTX_AAPL_SUPPORTS_OSX_COPYFILE;
2594                         config->copyfile_enabled = true;
2595                 }
2596
2597                 /*
2598                  * The client doesn't set the flag, so we can't check
2599                  * for it and just set it unconditionally
2600                  */
2601                 if (config->unix_info_enabled) {
2602                         server_caps |= SMB2_CRTCTX_AAPL_SUPPORTS_NFS_ACE;
2603                 }
2604
2605                 SBVAL(p, 0, server_caps);
2606                 ok = data_blob_append(req, &blob, p, 8);
2607                 if (!ok) {
2608                         return NT_STATUS_UNSUCCESSFUL;
2609                 }
2610         }
2611
2612         if (req_bitmap & SMB2_CRTCTX_AAPL_VOLUME_CAPS) {
2613                 int val = lp_case_sensitive(SNUM(handle->conn->tcon->compat));
2614                 uint64_t caps = 0;
2615
2616                 switch (val) {
2617                 case Auto:
2618                         break;
2619
2620                 case True:
2621                         caps |= SMB2_CRTCTX_AAPL_CASE_SENSITIVE;
2622                         break;
2623
2624                 default:
2625                         break;
2626                 }
2627
2628                 if (config->time_machine) {
2629                         caps |= SMB2_CRTCTX_AAPL_FULL_SYNC;
2630                 }
2631
2632                 SBVAL(p, 0, caps);
2633
2634                 ok = data_blob_append(req, &blob, p, 8);
2635                 if (!ok) {
2636                         return NT_STATUS_UNSUCCESSFUL;
2637                 }
2638         }
2639
2640         if (req_bitmap & SMB2_CRTCTX_AAPL_MODEL_INFO) {
2641                 ok = convert_string_talloc(req,
2642                                            CH_UNIX, CH_UTF16LE,
2643                                            config->model, strlen(config->model),
2644                                            &model, &modellen);
2645                 if (!ok) {
2646                         return NT_STATUS_UNSUCCESSFUL;
2647                 }
2648
2649                 SIVAL(p, 0, 0);
2650                 SIVAL(p + 4, 0, modellen);
2651                 ok = data_blob_append(req, &blob, p, 8);
2652                 if (!ok) {
2653                         talloc_free(model);
2654                         return NT_STATUS_UNSUCCESSFUL;
2655                 }
2656
2657                 ok = data_blob_append(req, &blob, model, modellen);
2658                 talloc_free(model);
2659                 if (!ok) {
2660                         return NT_STATUS_UNSUCCESSFUL;
2661                 }
2662         }
2663
2664         status = smb2_create_blob_add(out_context_blobs,
2665                                       out_context_blobs,
2666                                       SMB2_CREATE_TAG_AAPL,
2667                                       blob);
2668         if (NT_STATUS_IS_OK(status)) {
2669                 global_fruit_config.nego_aapl = true;
2670                 if (config->aapl_zero_file_id) {
2671                         aapl_force_zero_file_id(handle->conn->sconn);
2672                 }
2673         }
2674
2675         return status;
2676 }
2677
2678 static bool readdir_attr_meta_finderi_stream(
2679         struct vfs_handle_struct *handle,
2680         const struct smb_filename *smb_fname,
2681         AfpInfo *ai)
2682 {
2683         struct smb_filename *stream_name = NULL;
2684         files_struct *fsp = NULL;
2685         ssize_t nread;
2686         NTSTATUS status;
2687         int ret;
2688         bool ok;
2689         uint8_t buf[AFP_INFO_SIZE];
2690
2691         stream_name = synthetic_smb_fname(talloc_tos(),
2692                                           smb_fname->base_name,
2693                                           AFPINFO_STREAM_NAME,
2694                                           NULL, smb_fname->flags);
2695         if (stream_name == NULL) {
2696                 return false;
2697         }
2698
2699         ret = SMB_VFS_STAT(handle->conn, stream_name);
2700         if (ret != 0) {
2701                 return false;
2702         }
2703
2704         status = SMB_VFS_CREATE_FILE(
2705                 handle->conn,                           /* conn */
2706                 NULL,                                   /* req */
2707                 0,                                      /* root_dir_fid */
2708                 stream_name,                            /* fname */
2709                 FILE_READ_DATA,                         /* access_mask */
2710                 (FILE_SHARE_READ | FILE_SHARE_WRITE |   /* share_access */
2711                         FILE_SHARE_DELETE),
2712                 FILE_OPEN,                              /* create_disposition*/
2713                 0,                                      /* create_options */
2714                 0,                                      /* file_attributes */
2715                 INTERNAL_OPEN_ONLY,                     /* oplock_request */
2716                 NULL,                                   /* lease */
2717                 0,                                      /* allocation_size */
2718                 0,                                      /* private_flags */
2719                 NULL,                                   /* sd */
2720                 NULL,                                   /* ea_list */
2721                 &fsp,                                   /* result */
2722                 NULL,                                   /* pinfo */
2723                 NULL, NULL);                            /* create context */
2724
2725         TALLOC_FREE(stream_name);
2726
2727         if (!NT_STATUS_IS_OK(status)) {
2728                 return false;
2729         }
2730
2731         nread = SMB_VFS_PREAD(fsp, &buf[0], AFP_INFO_SIZE, 0);
2732         if (nread != AFP_INFO_SIZE) {
2733                 DBG_ERR("short read [%s] [%zd/%d]\n",
2734                         smb_fname_str_dbg(stream_name), nread, AFP_INFO_SIZE);
2735                 ok = false;
2736                 goto fail;
2737         }
2738
2739         memcpy(&ai->afpi_FinderInfo[0], &buf[AFP_OFF_FinderInfo],
2740                AFP_FinderSize);
2741
2742         ok = true;
2743
2744 fail:
2745         if (fsp != NULL) {
2746                 close_file(NULL, fsp, NORMAL_CLOSE);
2747         }
2748
2749         return ok;
2750 }
2751
2752 static bool readdir_attr_meta_finderi_netatalk(
2753         struct vfs_handle_struct *handle,
2754         const struct smb_filename *smb_fname,
2755         AfpInfo *ai)
2756 {
2757         struct adouble *ad = NULL;
2758         char *p = NULL;
2759
2760         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_META);
2761         if (ad == NULL) {
2762                 return false;
2763         }
2764
2765         p = ad_get_entry(ad, ADEID_FINDERI);
2766         if (p == NULL) {
2767                 DBG_ERR("No ADEID_FINDERI for [%s]\n", smb_fname->base_name);
2768                 TALLOC_FREE(ad);
2769                 return false;
2770         }
2771
2772         memcpy(&ai->afpi_FinderInfo[0], p, AFP_FinderSize);
2773         TALLOC_FREE(ad);
2774         return true;
2775 }
2776
2777 static bool readdir_attr_meta_finderi(struct vfs_handle_struct *handle,
2778                                       const struct smb_filename *smb_fname,
2779                                       struct readdir_attr_data *attr_data)
2780 {
2781         struct fruit_config_data *config = NULL;
2782         uint32_t date_added;
2783         AfpInfo ai = {0};
2784         bool ok;
2785
2786         SMB_VFS_HANDLE_GET_DATA(handle, config,
2787                                 struct fruit_config_data,
2788                                 return false);
2789
2790         switch (config->meta) {
2791         case FRUIT_META_NETATALK:
2792                 ok = readdir_attr_meta_finderi_netatalk(
2793                         handle, smb_fname, &ai);
2794                 break;
2795
2796         case FRUIT_META_STREAM:
2797                 ok = readdir_attr_meta_finderi_stream(
2798                         handle, smb_fname, &ai);
2799                 break;
2800
2801         default:
2802                 DBG_ERR("Unexpected meta config [%d]\n", config->meta);
2803                 return false;
2804         }
2805
2806         if (!ok) {
2807                 /* Don't bother with errors, it's likely ENOENT */
2808                 return true;
2809         }
2810
2811         if (S_ISREG(smb_fname->st.st_ex_mode)) {
2812                 /* finder_type */
2813                 memcpy(&attr_data->attr_data.aapl.finder_info[0],
2814                        &ai.afpi_FinderInfo[0], 4);
2815
2816                 /* finder_creator */
2817                 memcpy(&attr_data->attr_data.aapl.finder_info[0] + 4,
2818                        &ai.afpi_FinderInfo[4], 4);
2819         }
2820
2821         /* finder_flags */
2822         memcpy(&attr_data->attr_data.aapl.finder_info[0] + 8,
2823                &ai.afpi_FinderInfo[8], 2);
2824
2825         /* finder_ext_flags */
2826         memcpy(&attr_data->attr_data.aapl.finder_info[0] + 10,
2827                &ai.afpi_FinderInfo[24], 2);
2828
2829         /* creation date */
2830         date_added = convert_time_t_to_uint32_t(
2831                 smb_fname->st.st_ex_btime.tv_sec - AD_DATE_DELTA);
2832
2833         RSIVAL(&attr_data->attr_data.aapl.finder_info[0], 12, date_added);
2834
2835         return true;
2836 }
2837
2838 static uint64_t readdir_attr_rfork_size_adouble(
2839         struct vfs_handle_struct *handle,
2840         const struct smb_filename *smb_fname)
2841 {
2842         struct adouble *ad = NULL;
2843         uint64_t rfork_size;
2844
2845         ad = ad_get(talloc_tos(), handle, smb_fname,
2846                     ADOUBLE_RSRC);
2847         if (ad == NULL) {
2848                 return 0;
2849         }
2850
2851         rfork_size = ad_getentrylen(ad, ADEID_RFORK);
2852         TALLOC_FREE(ad);
2853
2854         return rfork_size;
2855 }
2856
2857 static uint64_t readdir_attr_rfork_size_stream(
2858         struct vfs_handle_struct *handle,
2859         const struct smb_filename *smb_fname)
2860 {
2861         struct smb_filename *stream_name = NULL;
2862         int ret;
2863         uint64_t rfork_size;
2864
2865         stream_name = synthetic_smb_fname(talloc_tos(),
2866                                           smb_fname->base_name,
2867                                           AFPRESOURCE_STREAM_NAME,
2868                                           NULL, 0);
2869         if (stream_name == NULL) {
2870                 return 0;
2871         }
2872
2873         ret = SMB_VFS_STAT(handle->conn, stream_name);
2874         if (ret != 0) {
2875                 TALLOC_FREE(stream_name);
2876                 return 0;
2877         }
2878
2879         rfork_size = stream_name->st.st_ex_size;
2880         TALLOC_FREE(stream_name);
2881
2882         return rfork_size;
2883 }
2884
2885 static uint64_t readdir_attr_rfork_size(struct vfs_handle_struct *handle,
2886                                         const struct smb_filename *smb_fname)
2887 {
2888         struct fruit_config_data *config = NULL;
2889         uint64_t rfork_size;
2890
2891         SMB_VFS_HANDLE_GET_DATA(handle, config,
2892                                 struct fruit_config_data,
2893                                 return 0);
2894
2895         switch (config->rsrc) {
2896         case FRUIT_RSRC_ADFILE:
2897         case FRUIT_RSRC_XATTR:
2898                 rfork_size = readdir_attr_rfork_size_adouble(handle,
2899                                                              smb_fname);
2900                 break;
2901
2902         case FRUIT_META_STREAM:
2903                 rfork_size = readdir_attr_rfork_size_stream(handle,
2904                                                             smb_fname);
2905                 break;
2906
2907         default:
2908                 DBG_ERR("Unexpected rsrc config [%d]\n", config->rsrc);
2909                 rfork_size = 0;
2910                 break;
2911         }
2912
2913         return rfork_size;
2914 }
2915
2916 static NTSTATUS readdir_attr_macmeta(struct vfs_handle_struct *handle,
2917                                      const struct smb_filename *smb_fname,
2918                                      struct readdir_attr_data *attr_data)
2919 {
2920         NTSTATUS status = NT_STATUS_OK;
2921         struct fruit_config_data *config = NULL;
2922         bool ok;
2923
2924         SMB_VFS_HANDLE_GET_DATA(handle, config,
2925                                 struct fruit_config_data,
2926                                 return NT_STATUS_UNSUCCESSFUL);
2927
2928
2929         /* Ensure we return a default value in the creation_date field */
2930         RSIVAL(&attr_data->attr_data.aapl.finder_info, 12, AD_DATE_START);
2931
2932         /*
2933          * Resource fork length
2934          */
2935
2936         if (config->readdir_attr_rsize) {
2937                 uint64_t rfork_size;
2938
2939                 rfork_size = readdir_attr_rfork_size(handle, smb_fname);
2940                 attr_data->attr_data.aapl.rfork_size = rfork_size;
2941         }
2942
2943         /*
2944          * FinderInfo
2945          */
2946
2947         if (config->readdir_attr_finder_info) {
2948                 ok = readdir_attr_meta_finderi(handle, smb_fname, attr_data);
2949                 if (!ok) {
2950                         status = NT_STATUS_INTERNAL_ERROR;
2951                 }
2952         }
2953
2954         return status;
2955 }
2956
2957 static NTSTATUS remove_virtual_nfs_aces(struct security_descriptor *psd)
2958 {
2959         NTSTATUS status;
2960         uint32_t i;
2961
2962         if (psd->dacl == NULL) {
2963                 return NT_STATUS_OK;
2964         }
2965
2966         for (i = 0; i < psd->dacl->num_aces; i++) {
2967                 /* MS NFS style mode/uid/gid */
2968                 int cmp = dom_sid_compare_domain(
2969                                 &global_sid_Unix_NFS,
2970                                 &psd->dacl->aces[i].trustee);
2971                 if (cmp != 0) {
2972                         /* Normal ACE entry. */
2973                         continue;
2974                 }
2975
2976                 /*
2977                  * security_descriptor_dacl_del()
2978                  * *must* return NT_STATUS_OK as we know
2979                  * we have something to remove.
2980                  */
2981
2982                 status = security_descriptor_dacl_del(psd,
2983                                 &psd->dacl->aces[i].trustee);
2984                 if (!NT_STATUS_IS_OK(status)) {
2985                         DBG_WARNING("failed to remove MS NFS style ACE: %s\n",
2986                                 nt_errstr(status));
2987                         return status;
2988                 }
2989
2990                 /*
2991                  * security_descriptor_dacl_del() may delete more
2992                  * then one entry subsequent to this one if the
2993                  * SID matches, but we only need to ensure that
2994                  * we stay looking at the same element in the array.
2995                  */
2996                 i--;
2997         }
2998         return NT_STATUS_OK;
2999 }
3000
3001 /* Search MS NFS style ACE with UNIX mode */
3002 static NTSTATUS check_ms_nfs(vfs_handle_struct *handle,
3003                              files_struct *fsp,
3004                              struct security_descriptor *psd,
3005                              mode_t *pmode,
3006                              bool *pdo_chmod)
3007 {
3008         uint32_t i;
3009         struct fruit_config_data *config = NULL;
3010
3011         *pdo_chmod = false;
3012
3013         SMB_VFS_HANDLE_GET_DATA(handle, config,
3014                                 struct fruit_config_data,
3015                                 return NT_STATUS_UNSUCCESSFUL);
3016
3017         if (!global_fruit_config.nego_aapl) {
3018                 return NT_STATUS_OK;
3019         }
3020         if (psd->dacl == NULL || !config->unix_info_enabled) {
3021                 return NT_STATUS_OK;
3022         }
3023
3024         for (i = 0; i < psd->dacl->num_aces; i++) {
3025                 if (dom_sid_compare_domain(
3026                             &global_sid_Unix_NFS_Mode,
3027                             &psd->dacl->aces[i].trustee) == 0) {
3028                         *pmode = (mode_t)psd->dacl->aces[i].trustee.sub_auths[2];
3029                         *pmode &= (S_IRWXU | S_IRWXG | S_IRWXO);
3030                         *pdo_chmod = true;
3031
3032                         DEBUG(10, ("MS NFS chmod request %s, %04o\n",
3033                                    fsp_str_dbg(fsp), (unsigned)(*pmode)));
3034                         break;
3035                 }
3036         }
3037
3038         /*
3039          * Remove any incoming virtual ACE entries generated by
3040          * fruit_fget_nt_acl().
3041          */
3042
3043         return remove_virtual_nfs_aces(psd);
3044 }
3045
3046 /****************************************************************************
3047  * VFS ops
3048  ****************************************************************************/
3049
3050 static int fruit_connect(vfs_handle_struct *handle,
3051                          const char *service,
3052                          const char *user)
3053 {
3054         int rc;
3055         char *list = NULL, *newlist = NULL;
3056         struct fruit_config_data *config;
3057
3058         DEBUG(10, ("fruit_connect\n"));
3059
3060         rc = SMB_VFS_NEXT_CONNECT(handle, service, user);
3061         if (rc < 0) {
3062                 return rc;
3063         }
3064
3065         rc = init_fruit_config(handle);
3066         if (rc != 0) {
3067                 return rc;
3068         }
3069
3070         SMB_VFS_HANDLE_GET_DATA(handle, config,
3071                                 struct fruit_config_data, return -1);
3072
3073         if (config->veto_appledouble) {
3074                 list = lp_veto_files(talloc_tos(), SNUM(handle->conn));
3075
3076                 if (list) {
3077                         if (strstr(list, "/" ADOUBLE_NAME_PREFIX "*/") == NULL) {
3078                                 newlist = talloc_asprintf(
3079                                         list,
3080                                         "%s/" ADOUBLE_NAME_PREFIX "*/",
3081                                         list);
3082                                 lp_do_parameter(SNUM(handle->conn),
3083                                                 "veto files",
3084                                                 newlist);
3085                         }
3086                 } else {
3087                         lp_do_parameter(SNUM(handle->conn),
3088                                         "veto files",
3089                                         "/" ADOUBLE_NAME_PREFIX "*/");
3090                 }
3091
3092                 TALLOC_FREE(list);
3093         }
3094
3095         if (config->encoding == FRUIT_ENC_NATIVE) {
3096                 lp_do_parameter(SNUM(handle->conn),
3097                                 "catia:mappings",
3098                                 fruit_catia_maps);
3099         }
3100
3101         if (config->time_machine) {
3102                 DBG_NOTICE("Enabling durable handles for Time Machine "
3103                            "support on [%s]\n", service);
3104                 lp_do_parameter(SNUM(handle->conn), "durable handles", "yes");
3105                 lp_do_parameter(SNUM(handle->conn), "kernel oplocks", "no");
3106                 lp_do_parameter(SNUM(handle->conn), "kernel share modes", "no");
3107                 if (!lp_strict_sync(SNUM(handle->conn))) {
3108                         DBG_WARNING("Time Machine without strict sync is not "
3109                                     "recommended!\n");
3110                 }
3111                 lp_do_parameter(SNUM(handle->conn), "posix locking", "no");
3112         }
3113
3114         return rc;
3115 }
3116
3117 static int fruit_open_meta_stream(vfs_handle_struct *handle,
3118                                   struct smb_filename *smb_fname,
3119                                   files_struct *fsp,
3120                                   int flags,
3121                                   mode_t mode)
3122 {
3123         AfpInfo *ai = NULL;
3124         char afpinfo_buf[AFP_INFO_SIZE];
3125         ssize_t len, written;
3126         int hostfd = -1;
3127         int rc = -1;
3128
3129         hostfd = SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
3130         if (hostfd == -1) {
3131                 return -1;
3132         }
3133
3134         if (!(flags & (O_CREAT | O_TRUNC))) {
3135                 return hostfd;
3136         }
3137
3138         ai = afpinfo_new(talloc_tos());
3139         if (ai == NULL) {
3140                 rc = -1;
3141                 goto fail;
3142         }
3143
3144         len = afpinfo_pack(ai, afpinfo_buf);
3145         if (len != AFP_INFO_SIZE) {
3146                 rc = -1;
3147                 goto fail;
3148         }
3149
3150         /* Set fd, needed in SMB_VFS_NEXT_PWRITE() */
3151         fsp->fh->fd = hostfd;
3152
3153         written = SMB_VFS_NEXT_PWRITE(handle, fsp, afpinfo_buf,
3154                                       AFP_INFO_SIZE, 0);
3155         fsp->fh->fd = -1;
3156         if (written != AFP_INFO_SIZE) {
3157                 DBG_ERR("bad write [%zd/%d]\n", written, AFP_INFO_SIZE);
3158                 rc = -1;
3159                 goto fail;
3160         }
3161
3162         rc = 0;
3163
3164 fail:
3165         DBG_DEBUG("rc=%d, fd=%d\n", rc, hostfd);
3166
3167         if (rc != 0) {
3168                 int saved_errno = errno;
3169                 if (hostfd >= 0) {
3170                         fsp->fh->fd = hostfd;
3171                         SMB_VFS_NEXT_CLOSE(handle, fsp);
3172                 }
3173                 hostfd = -1;
3174                 errno = saved_errno;
3175         }
3176         return hostfd;
3177 }
3178
3179 static int fruit_open_meta_netatalk(vfs_handle_struct *handle,
3180                                     struct smb_filename *smb_fname,
3181                                     files_struct *fsp,
3182                                     int flags,
3183                                     mode_t mode)
3184 {
3185         int rc;
3186         int fakefd = -1;
3187         struct adouble *ad = NULL;
3188         int fds[2];
3189
3190         DBG_DEBUG("Path [%s]\n", smb_fname_str_dbg(smb_fname));
3191
3192         /*
3193          * Return a valid fd, but ensure any attempt to use it returns an error
3194          * (EPIPE). All operations on the smb_fname or the fsp will use path
3195          * based syscalls.
3196          */
3197         rc = pipe(fds);
3198         if (rc != 0) {
3199                 goto exit;
3200         }
3201         fakefd = fds[0];
3202         close(fds[1]);
3203
3204         if (flags & (O_CREAT | O_TRUNC)) {
3205                 /*
3206                  * The attribute does not exist or needs to be truncated,
3207                  * create an AppleDouble EA
3208                  */
3209                 ad = ad_init(fsp, handle, ADOUBLE_META);
3210                 if (ad == NULL) {
3211                         rc = -1;
3212                         goto exit;
3213                 }
3214
3215                 rc = ad_set(ad, fsp->fsp_name);
3216                 if (rc != 0) {
3217                         rc = -1;
3218                         goto exit;
3219                 }
3220
3221                 TALLOC_FREE(ad);
3222         }
3223
3224 exit:
3225         DEBUG(10, ("fruit_open meta rc=%d, fd=%d\n", rc, fakefd));
3226         if (rc != 0) {
3227                 int saved_errno = errno;
3228                 if (fakefd >= 0) {
3229                         close(fakefd);
3230                 }
3231                 fakefd = -1;
3232                 errno = saved_errno;
3233         }
3234         return fakefd;
3235 }
3236
3237 static int fruit_open_meta(vfs_handle_struct *handle,
3238                            struct smb_filename *smb_fname,
3239                            files_struct *fsp, int flags, mode_t mode)
3240 {
3241         int fd;
3242         struct fruit_config_data *config = NULL;
3243         struct fio *fio = NULL;
3244
3245         DBG_DEBUG("path [%s]\n", smb_fname_str_dbg(smb_fname));
3246
3247         SMB_VFS_HANDLE_GET_DATA(handle, config,
3248                                 struct fruit_config_data, return -1);
3249
3250         switch (config->meta) {
3251         case FRUIT_META_STREAM:
3252                 fd = fruit_open_meta_stream(handle, smb_fname,
3253                                             fsp, flags, mode);
3254                 break;
3255
3256         case FRUIT_META_NETATALK:
3257                 fd = fruit_open_meta_netatalk(handle, smb_fname,
3258                                               fsp, flags, mode);
3259                 break;
3260
3261         default:
3262                 DBG_ERR("Unexpected meta config [%d]\n", config->meta);
3263                 return -1;
3264         }
3265
3266         DBG_DEBUG("path [%s] fd [%d]\n", smb_fname_str_dbg(smb_fname), fd);
3267
3268         if (fd == -1) {
3269                 return -1;
3270         }
3271
3272         fio = VFS_ADD_FSP_EXTENSION(handle, fsp, struct fio, NULL);
3273         fio->type = ADOUBLE_META;
3274         fio->config = config;
3275
3276         return fd;
3277 }
3278
3279 static int fruit_open_rsrc_adouble(vfs_handle_struct *handle,
3280                                    struct smb_filename *smb_fname,
3281                                    files_struct *fsp,
3282                                    int flags,
3283                                    mode_t mode)
3284 {
3285         int rc = 0;
3286         struct adouble *ad = NULL;
3287         struct smb_filename *smb_fname_base = NULL;
3288         struct fruit_config_data *config = NULL;
3289         int hostfd = -1;
3290
3291         SMB_VFS_HANDLE_GET_DATA(handle, config,
3292                                 struct fruit_config_data, return -1);
3293
3294         if ((!(flags & O_CREAT)) &&
3295             S_ISDIR(fsp->base_fsp->fsp_name->st.st_ex_mode))
3296         {
3297                 /* sorry, but directories don't habe a resource fork */
3298                 rc = -1;
3299                 goto exit;
3300         }
3301
3302         rc = adouble_path(talloc_tos(), smb_fname, &smb_fname_base);
3303         if (rc != 0) {
3304                 goto exit;
3305         }
3306
3307         /* Sanitize flags */
3308         if (flags & O_WRONLY) {
3309                 /* We always need read access for the metadata header too */
3310                 flags &= ~O_WRONLY;
3311                 flags |= O_RDWR;
3312         }
3313
3314         hostfd = SMB_VFS_NEXT_OPEN(handle, smb_fname_base, fsp,
3315                                    flags, mode);
3316         if (hostfd == -1) {
3317                 rc = -1;
3318                 goto exit;
3319         }
3320
3321         if (flags & (O_CREAT | O_TRUNC)) {
3322                 ad = ad_init(fsp, handle, ADOUBLE_RSRC);
3323                 if (ad == NULL) {
3324                         rc = -1;
3325                         goto exit;
3326                 }
3327
3328                 fsp->fh->fd = hostfd;
3329
3330                 rc = ad_fset(ad, fsp);
3331                 fsp->fh->fd = -1;
3332                 if (rc != 0) {
3333                         rc = -1;
3334                         goto exit;
3335                 }
3336                 TALLOC_FREE(ad);
3337         }
3338
3339 exit:
3340
3341         TALLOC_FREE(smb_fname_base);
3342
3343         DEBUG(10, ("fruit_open resource fork: rc=%d, fd=%d\n", rc, hostfd));
3344         if (rc != 0) {
3345                 int saved_errno = errno;
3346                 if (hostfd >= 0) {
3347                         /*
3348                          * BUGBUGBUG -- we would need to call
3349                          * fd_close_posix here, but we don't have a
3350                          * full fsp yet
3351                          */
3352                         fsp->fh->fd = hostfd;
3353                         SMB_VFS_CLOSE(fsp);
3354                 }
3355                 hostfd = -1;
3356                 errno = saved_errno;
3357         }
3358         return hostfd;
3359 }
3360
3361 static int fruit_open_rsrc_xattr(vfs_handle_struct *handle,
3362                                  struct smb_filename *smb_fname,
3363                                  files_struct *fsp,
3364                                  int flags,
3365                                  mode_t mode)
3366 {
3367 #ifdef HAVE_ATTROPEN
3368         int fd = -1;
3369
3370         fd = attropen(smb_fname->base_name,
3371                       AFPRESOURCE_EA_NETATALK,
3372                       flags,
3373                       mode);
3374         if (fd == -1) {
3375                 return -1;
3376         }
3377
3378         return fd;
3379
3380 #else
3381         errno = ENOSYS;
3382         return -1;
3383 #endif
3384 }
3385
3386 static int fruit_open_rsrc(vfs_handle_struct *handle,
3387                            struct smb_filename *smb_fname,
3388                            files_struct *fsp, int flags, mode_t mode)
3389 {
3390         int fd;
3391         struct fruit_config_data *config = NULL;
3392         struct fio *fio = NULL;
3393
3394         DBG_DEBUG("Path [%s]\n", smb_fname_str_dbg(smb_fname));
3395
3396         SMB_VFS_HANDLE_GET_DATA(handle, config,
3397                                 struct fruit_config_data, return -1);
3398
3399         if (((flags & O_ACCMODE) == O_RDONLY)
3400             && (flags & O_CREAT)
3401             && !VALID_STAT(fsp->fsp_name->st))
3402         {
3403                 /*
3404                  * This means the stream doesn't exist. macOS SMB server fails
3405                  * this with NT_STATUS_OBJECT_NAME_NOT_FOUND, so must we. Cf bug
3406                  * 12565 and the test for this combination in
3407                  * test_rfork_create().
3408                  */
3409                 errno = ENOENT;
3410                 return -1;
3411         }
3412
3413         switch (config->rsrc) {
3414         case FRUIT_RSRC_STREAM:
3415                 fd = SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
3416                 break;
3417
3418         case FRUIT_RSRC_ADFILE:
3419                 fd = fruit_open_rsrc_adouble(handle, smb_fname,
3420                                              fsp, flags, mode);
3421                 break;
3422
3423         case FRUIT_RSRC_XATTR:
3424                 fd = fruit_open_rsrc_xattr(handle, smb_fname,
3425                                            fsp, flags, mode);
3426                 break;
3427
3428         default:
3429                 DBG_ERR("Unexpected rsrc config [%d]\n", config->rsrc);
3430                 return -1;
3431         }
3432
3433         DBG_DEBUG("Path [%s] fd [%d]\n", smb_fname_str_dbg(smb_fname), fd);
3434
3435         if (fd == -1) {
3436                 return -1;
3437         }
3438
3439         fio = VFS_ADD_FSP_EXTENSION(handle, fsp, struct fio, NULL);
3440         fio->type = ADOUBLE_RSRC;
3441         fio->config = config;
3442
3443         return fd;
3444 }
3445
3446 static int fruit_open(vfs_handle_struct *handle,
3447                       struct smb_filename *smb_fname,
3448                       files_struct *fsp, int flags, mode_t mode)
3449 {
3450         int fd;
3451
3452         DBG_DEBUG("Path [%s]\n", smb_fname_str_dbg(smb_fname));
3453
3454         if (!is_ntfs_stream_smb_fname(smb_fname)) {
3455                 return SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
3456         }
3457
3458         if (is_afpinfo_stream(smb_fname)) {
3459                 fd = fruit_open_meta(handle, smb_fname, fsp, flags, mode);
3460         } else if (is_afpresource_stream(smb_fname)) {
3461                 fd = fruit_open_rsrc(handle, smb_fname, fsp, flags, mode);
3462         } else {
3463                 fd = SMB_VFS_NEXT_OPEN(handle, smb_fname, fsp, flags, mode);
3464         }
3465
3466         DBG_DEBUG("Path [%s] fd [%d]\n", smb_fname_str_dbg(smb_fname), fd);
3467
3468         return fd;
3469 }
3470
3471 static int fruit_rename(struct vfs_handle_struct *handle,
3472                         const struct smb_filename *smb_fname_src,
3473                         const struct smb_filename *smb_fname_dst)
3474 {
3475         int rc = -1;
3476         struct fruit_config_data *config = NULL;
3477         struct smb_filename *src_adp_smb_fname = NULL;
3478         struct smb_filename *dst_adp_smb_fname = NULL;
3479
3480         SMB_VFS_HANDLE_GET_DATA(handle, config,
3481                                 struct fruit_config_data, return -1);
3482
3483         if (!VALID_STAT(smb_fname_src->st)) {
3484                 DBG_ERR("Need valid stat for [%s]\n",
3485                         smb_fname_str_dbg(smb_fname_src));
3486                 return -1;
3487         }
3488
3489         rc = SMB_VFS_NEXT_RENAME(handle, smb_fname_src, smb_fname_dst);
3490         if (rc != 0) {
3491                 return -1;
3492         }
3493
3494         if ((config->rsrc != FRUIT_RSRC_ADFILE) ||
3495             (!S_ISREG(smb_fname_src->st.st_ex_mode)))
3496         {
3497                 return 0;
3498         }
3499
3500         rc = adouble_path(talloc_tos(), smb_fname_src, &src_adp_smb_fname);
3501         if (rc != 0) {
3502                 goto done;
3503         }
3504
3505         rc = adouble_path(talloc_tos(), smb_fname_dst, &dst_adp_smb_fname);
3506         if (rc != 0) {
3507                 goto done;
3508         }
3509
3510         DBG_DEBUG("%s -> %s\n",
3511                   smb_fname_str_dbg(src_adp_smb_fname),
3512                   smb_fname_str_dbg(dst_adp_smb_fname));
3513
3514         rc = SMB_VFS_NEXT_RENAME(handle, src_adp_smb_fname, dst_adp_smb_fname);
3515         if (errno == ENOENT) {
3516                 rc = 0;
3517         }
3518
3519 done:
3520         TALLOC_FREE(src_adp_smb_fname);
3521         TALLOC_FREE(dst_adp_smb_fname);
3522         return rc;
3523 }
3524
3525 static int fruit_unlink_meta_stream(vfs_handle_struct *handle,
3526                                     const struct smb_filename *smb_fname)
3527 {
3528         return SMB_VFS_NEXT_UNLINK(handle, smb_fname);
3529 }
3530
3531 static int fruit_unlink_meta_netatalk(vfs_handle_struct *handle,
3532                                       const struct smb_filename *smb_fname)
3533 {
3534         return SMB_VFS_REMOVEXATTR(handle->conn,
3535                                    smb_fname,
3536                                    AFPINFO_EA_NETATALK);
3537 }
3538
3539 static int fruit_unlink_meta(vfs_handle_struct *handle,
3540                              const struct smb_filename *smb_fname)
3541 {
3542         struct fruit_config_data *config = NULL;
3543         int rc;
3544
3545         SMB_VFS_HANDLE_GET_DATA(handle, config,
3546                                 struct fruit_config_data, return -1);
3547
3548         switch (config->meta) {
3549         case FRUIT_META_STREAM:
3550                 rc = fruit_unlink_meta_stream(handle, smb_fname);
3551                 break;
3552
3553         case FRUIT_META_NETATALK:
3554                 rc = fruit_unlink_meta_netatalk(handle, smb_fname);
3555                 break;
3556
3557         default:
3558                 DBG_ERR("Unsupported meta config [%d]\n", config->meta);
3559                 return -1;
3560         }
3561
3562         return rc;
3563 }
3564
3565 static int fruit_unlink_rsrc_stream(vfs_handle_struct *handle,
3566                                     const struct smb_filename *smb_fname,
3567                                     bool force_unlink)
3568 {
3569         int ret;
3570
3571         if (!force_unlink) {
3572                 struct smb_filename *smb_fname_cp = NULL;
3573                 off_t size;
3574
3575                 smb_fname_cp = cp_smb_filename(talloc_tos(), smb_fname);
3576                 if (smb_fname_cp == NULL) {
3577                         return -1;
3578                 }
3579
3580                 /*
3581                  * 0 byte resource fork streams are not listed by
3582                  * vfs_streaminfo, as a result stream cleanup/deletion of file
3583                  * deletion doesn't remove the resourcefork stream.
3584                  */
3585
3586                 ret = SMB_VFS_NEXT_STAT(handle, smb_fname_cp);
3587                 if (ret != 0) {
3588                         TALLOC_FREE(smb_fname_cp);
3589                         DBG_ERR("stat [%s] failed [%s]\n",
3590                                 smb_fname_str_dbg(smb_fname_cp), strerror(errno));
3591                         return -1;
3592                 }
3593
3594                 size = smb_fname_cp->st.st_ex_size;
3595                 TALLOC_FREE(smb_fname_cp);
3596
3597                 if (size > 0) {
3598                         /* OS X ignores resource fork stream delete requests */
3599                         return 0;
3600                 }
3601         }
3602
3603         ret = SMB_VFS_NEXT_UNLINK(handle, smb_fname);
3604         if ((ret != 0) && (errno == ENOENT) && force_unlink) {
3605                 ret = 0;
3606         }
3607
3608         return ret;
3609 }
3610
3611 static int fruit_unlink_rsrc_adouble(vfs_handle_struct *handle,
3612                                      const struct smb_filename *smb_fname,
3613                                      bool force_unlink)
3614 {
3615         int rc;
3616         struct adouble *ad = NULL;
3617         struct smb_filename *adp_smb_fname = NULL;
3618
3619         if (!force_unlink) {
3620                 ad = ad_get(talloc_tos(), handle, smb_fname,
3621                             ADOUBLE_RSRC);
3622                 if (ad == NULL) {
3623                         errno = ENOENT;
3624                         return -1;
3625                 }
3626
3627
3628                 /*
3629                  * 0 byte resource fork streams are not listed by
3630                  * vfs_streaminfo, as a result stream cleanup/deletion of file
3631                  * deletion doesn't remove the resourcefork stream.
3632                  */
3633
3634                 if (ad_getentrylen(ad, ADEID_RFORK) > 0) {
3635                         /* OS X ignores resource fork stream delete requests */
3636                         TALLOC_FREE(ad);
3637                         return 0;
3638                 }
3639
3640                 TALLOC_FREE(ad);
3641         }
3642
3643         rc = adouble_path(talloc_tos(), smb_fname, &adp_smb_fname);
3644         if (rc != 0) {
3645                 return -1;
3646         }
3647
3648         rc = SMB_VFS_NEXT_UNLINK(handle, adp_smb_fname);
3649         TALLOC_FREE(adp_smb_fname);
3650         if ((rc != 0) && (errno == ENOENT) && force_unlink) {
3651                 rc = 0;
3652         }
3653
3654         return rc;
3655 }
3656
3657 static int fruit_unlink_rsrc_xattr(vfs_handle_struct *handle,
3658                                    const struct smb_filename *smb_fname,
3659                                    bool force_unlink)
3660 {
3661         /*
3662          * OS X ignores resource fork stream delete requests, so nothing to do
3663          * here. Removing the file will remove the xattr anyway, so we don't
3664          * have to take care of removing 0 byte resource forks that could be
3665          * left behind.
3666          */
3667         return 0;
3668 }
3669
3670 static int fruit_unlink_rsrc(vfs_handle_struct *handle,
3671                              const struct smb_filename *smb_fname,
3672                              bool force_unlink)
3673 {
3674         struct fruit_config_data *config = NULL;
3675         int rc;
3676
3677         SMB_VFS_HANDLE_GET_DATA(handle, config,
3678                                 struct fruit_config_data, return -1);
3679
3680         switch (config->rsrc) {
3681         case FRUIT_RSRC_STREAM:
3682                 rc = fruit_unlink_rsrc_stream(handle, smb_fname, force_unlink);
3683                 break;
3684
3685         case FRUIT_RSRC_ADFILE:
3686                 rc = fruit_unlink_rsrc_adouble(handle, smb_fname, force_unlink);
3687                 break;
3688
3689         case FRUIT_RSRC_XATTR:
3690                 rc = fruit_unlink_rsrc_xattr(handle, smb_fname, force_unlink);
3691                 break;
3692
3693         default:
3694                 DBG_ERR("Unsupported rsrc config [%d]\n", config->rsrc);
3695                 return -1;
3696         }
3697
3698         return rc;
3699 }
3700
3701 static int fruit_unlink(vfs_handle_struct *handle,
3702                         const struct smb_filename *smb_fname)
3703 {
3704         int rc;
3705         struct fruit_config_data *config = NULL;
3706         struct smb_filename *rsrc_smb_fname = NULL;
3707
3708         SMB_VFS_HANDLE_GET_DATA(handle, config,
3709                                 struct fruit_config_data, return -1);
3710
3711         if (is_afpinfo_stream(smb_fname)) {
3712                 return fruit_unlink_meta(handle, smb_fname);
3713         } else if (is_afpresource_stream(smb_fname)) {
3714                 return fruit_unlink_rsrc(handle, smb_fname, false);
3715         } if (is_ntfs_stream_smb_fname(smb_fname)) {
3716                 return SMB_VFS_NEXT_UNLINK(handle, smb_fname);
3717         }
3718
3719         /*
3720          * A request to delete the base file. Because 0 byte resource
3721          * fork streams are not listed by fruit_streaminfo,
3722          * delete_all_streams() can't remove 0 byte resource fork
3723          * streams, so we have to cleanup this here.
3724          */
3725         rsrc_smb_fname = synthetic_smb_fname(talloc_tos(),
3726                                              smb_fname->base_name,
3727                                              AFPRESOURCE_STREAM_NAME,
3728                                              NULL,
3729                                              smb_fname->flags);
3730         if (rsrc_smb_fname == NULL) {
3731                 return -1;
3732         }
3733
3734         rc = fruit_unlink_rsrc(handle, rsrc_smb_fname, true);
3735         if ((rc != 0) && (errno != ENOENT)) {
3736                 DBG_ERR("Forced unlink of [%s] failed [%s]\n",
3737                         smb_fname_str_dbg(rsrc_smb_fname), strerror(errno));
3738                 TALLOC_FREE(rsrc_smb_fname);
3739                 return -1;
3740         }
3741         TALLOC_FREE(rsrc_smb_fname);
3742
3743         return SMB_VFS_NEXT_UNLINK(handle, smb_fname);
3744 }
3745
3746 static int fruit_chmod(vfs_handle_struct *handle,
3747                        const struct smb_filename *smb_fname,
3748                        mode_t mode)
3749 {
3750         int rc = -1;
3751         struct fruit_config_data *config = NULL;
3752         struct smb_filename *smb_fname_adp = NULL;
3753
3754         rc = SMB_VFS_NEXT_CHMOD(handle, smb_fname, mode);
3755         if (rc != 0) {
3756                 return rc;
3757         }
3758
3759         SMB_VFS_HANDLE_GET_DATA(handle, config,
3760                                 struct fruit_config_data, return -1);
3761
3762         if (config->rsrc != FRUIT_RSRC_ADFILE) {
3763                 return 0;
3764         }
3765
3766         if (!VALID_STAT(smb_fname->st)) {
3767                 return 0;
3768         }
3769
3770         if (!S_ISREG(smb_fname->st.st_ex_mode)) {
3771                 return 0;
3772         }
3773
3774         rc = adouble_path(talloc_tos(), smb_fname, &smb_fname_adp);
3775         if (rc != 0) {
3776                 return -1;
3777         }
3778
3779         DEBUG(10, ("fruit_chmod: %s\n", smb_fname_adp->base_name));
3780
3781         rc = SMB_VFS_NEXT_CHMOD(handle, smb_fname_adp, mode);
3782         if (errno == ENOENT) {
3783                 rc = 0;
3784         }
3785
3786         TALLOC_FREE(smb_fname_adp);
3787         return rc;
3788 }
3789
3790 static int fruit_chown(vfs_handle_struct *handle,
3791                        const struct smb_filename *smb_fname,
3792                        uid_t uid,
3793                        gid_t gid)
3794 {
3795         int rc = -1;
3796         struct fruit_config_data *config = NULL;
3797         struct smb_filename *adp_smb_fname = NULL;
3798
3799         rc = SMB_VFS_NEXT_CHOWN(handle, smb_fname, uid, gid);
3800         if (rc != 0) {
3801                 return rc;
3802         }
3803
3804         SMB_VFS_HANDLE_GET_DATA(handle, config,
3805                                 struct fruit_config_data, return -1);
3806
3807         if (config->rsrc != FRUIT_RSRC_ADFILE) {
3808                 return 0;
3809         }
3810
3811         if (!VALID_STAT(smb_fname->st)) {
3812                 return 0;
3813         }
3814
3815         if (!S_ISREG(smb_fname->st.st_ex_mode)) {
3816                 return 0;
3817         }
3818
3819         rc = adouble_path(talloc_tos(), smb_fname, &adp_smb_fname);
3820         if (rc != 0) {
3821                 goto done;
3822         }
3823
3824         DEBUG(10, ("fruit_chown: %s\n", adp_smb_fname->base_name));
3825
3826         rc = SMB_VFS_NEXT_CHOWN(handle, adp_smb_fname, uid, gid);
3827         if (errno == ENOENT) {
3828                 rc = 0;
3829         }
3830
3831  done:
3832         TALLOC_FREE(adp_smb_fname);
3833         return rc;
3834 }
3835
3836 static int fruit_rmdir(struct vfs_handle_struct *handle,
3837                         const struct smb_filename *smb_fname)
3838 {
3839         DIR *dh = NULL;
3840         struct dirent *de;
3841         struct fruit_config_data *config;
3842
3843         SMB_VFS_HANDLE_GET_DATA(handle, config,
3844                                 struct fruit_config_data, return -1);
3845
3846         if (config->rsrc != FRUIT_RSRC_ADFILE) {
3847                 goto exit_rmdir;
3848         }
3849
3850         /*
3851          * Due to there is no way to change bDeleteVetoFiles variable
3852          * from this module, need to clean up ourselves
3853          */
3854
3855         dh = SMB_VFS_OPENDIR(handle->conn, smb_fname, NULL, 0);
3856         if (dh == NULL) {
3857                 goto exit_rmdir;
3858         }
3859
3860         while ((de = SMB_VFS_READDIR(handle->conn, dh, NULL)) != NULL) {
3861                 int match;
3862                 struct adouble *ad = NULL;
3863                 char *p = NULL;
3864                 struct smb_filename *ad_smb_fname = NULL;
3865                 int ret;
3866
3867                 match = strncmp(de->d_name,
3868                                 ADOUBLE_NAME_PREFIX,
3869                                 strlen(ADOUBLE_NAME_PREFIX));
3870                 if (match != 0) {
3871                         continue;
3872                 }
3873
3874                 p = talloc_asprintf(talloc_tos(), "%s/%s",
3875                                     smb_fname->base_name, de->d_name);
3876                 if (p == NULL) {
3877                         DBG_ERR("talloc_asprintf failed\n");
3878                         return -1;
3879                 }
3880
3881                 ad_smb_fname = synthetic_smb_fname(talloc_tos(), p,
3882                                                     NULL, NULL,
3883                                                     smb_fname->flags);
3884                 TALLOC_FREE(p);
3885                 if (ad_smb_fname == NULL) {
3886                         DBG_ERR("synthetic_smb_fname failed\n");
3887                         return -1;
3888                 }
3889
3890                 /*
3891                  * Check whether it's a valid AppleDouble file, if
3892                  * yes, delete it, ignore it otherwise.
3893                  */
3894                 ad = ad_get(talloc_tos(), handle, ad_smb_fname, ADOUBLE_RSRC);
3895                 if (ad == NULL) {
3896                         TALLOC_FREE(ad_smb_fname);
3897                         TALLOC_FREE(p);
3898                         continue;
3899                 }
3900                 TALLOC_FREE(ad);
3901
3902                 ret = SMB_VFS_NEXT_UNLINK(handle, ad_smb_fname);
3903                 if (ret != 0) {
3904                         DBG_ERR("Deleting [%s] failed\n",
3905                                 smb_fname_str_dbg(ad_smb_fname));
3906                 }
3907                 TALLOC_FREE(ad_smb_fname);
3908         }
3909
3910 exit_rmdir:
3911         if (dh) {
3912                 SMB_VFS_CLOSEDIR(handle->conn, dh);
3913         }
3914         return SMB_VFS_NEXT_RMDIR(handle, smb_fname);
3915 }
3916
3917 static ssize_t fruit_pread_meta_stream(vfs_handle_struct *handle,
3918                                        files_struct *fsp, void *data,
3919                                        size_t n, off_t offset)
3920 {
3921         ssize_t nread;
3922         int ret;
3923
3924         nread = SMB_VFS_NEXT_PREAD(handle, fsp, data, n, offset);
3925
3926         if (nread == n) {
3927                 return nread;
3928         }
3929
3930         DBG_ERR("Removing [%s] after short read [%zd]\n",
3931                 fsp_str_dbg(fsp), nread);
3932
3933         ret = SMB_VFS_NEXT_UNLINK(handle, fsp->fsp_name);
3934         if (ret != 0) {
3935                 DBG_ERR("Removing [%s] failed\n", fsp_str_dbg(fsp));
3936                 return -1;
3937         }
3938
3939         errno = EINVAL;
3940         return -1;
3941 }
3942
3943 static ssize_t fruit_pread_meta_adouble(vfs_handle_struct *handle,
3944                                         files_struct *fsp, void *data,
3945                                         size_t n, off_t offset)
3946 {
3947         AfpInfo *ai = NULL;
3948         struct adouble *ad = NULL;
3949         char afpinfo_buf[AFP_INFO_SIZE];
3950         char *p = NULL;
3951         ssize_t nread;
3952
3953         ai = afpinfo_new(talloc_tos());
3954         if (ai == NULL) {
3955                 return -1;
3956         }
3957
3958         ad = ad_fget(talloc_tos(), handle, fsp, ADOUBLE_META);
3959         if (ad == NULL) {
3960                 nread = -1;
3961                 goto fail;
3962         }
3963
3964         p = ad_get_entry(ad, ADEID_FINDERI);
3965         if (p == NULL) {
3966                 DBG_ERR("No ADEID_FINDERI for [%s]\n", fsp_str_dbg(fsp));
3967                 nread = -1;
3968                 goto fail;
3969         }
3970
3971         memcpy(&ai->afpi_FinderInfo[0], p, ADEDLEN_FINDERI);
3972
3973         nread = afpinfo_pack(ai, afpinfo_buf);
3974         if (nread != AFP_INFO_SIZE) {
3975                 nread = -1;
3976                 goto fail;
3977         }
3978
3979         memcpy(data, afpinfo_buf, n);
3980         nread = n;
3981
3982 fail:
3983         TALLOC_FREE(ai);
3984         return nread;
3985 }
3986
3987 static ssize_t fruit_pread_meta(vfs_handle_struct *handle,
3988                                 files_struct *fsp, void *data,
3989                                 size_t n, off_t offset)
3990 {
3991         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
3992         ssize_t nread;
3993         ssize_t to_return;
3994
3995         /*
3996          * OS X has a off-by-1 error in the offset calculation, so we're
3997          * bug compatible here. It won't hurt, as any relevant real
3998          * world read requests from the AFP_AfpInfo stream will be
3999          * offset=0 n=60. offset is ignored anyway, see below.
4000          */
4001         if ((offset < 0) || (offset >= AFP_INFO_SIZE + 1)) {
4002                 return 0;
4003         }
4004
4005         /* Yes, macOS always reads from offset 0 */
4006         offset = 0;
4007         to_return = MIN(n, AFP_INFO_SIZE);
4008
4009         switch (fio->config->meta) {
4010         case FRUIT_META_STREAM:
4011                 nread = fruit_pread_meta_stream(handle, fsp, data,
4012                                                 to_return, offset);
4013                 break;
4014
4015         case FRUIT_META_NETATALK:
4016                 nread = fruit_pread_meta_adouble(handle, fsp, data,
4017                                                  to_return, offset);
4018                 break;
4019
4020         default:
4021                 DBG_ERR("Unexpected meta config [%d]\n", fio->config->meta);
4022                 return -1;
4023         }
4024
4025         return nread;
4026 }
4027
4028 static ssize_t fruit_pread_rsrc_stream(vfs_handle_struct *handle,
4029                                        files_struct *fsp, void *data,
4030                                        size_t n, off_t offset)
4031 {
4032         return SMB_VFS_NEXT_PREAD(handle, fsp, data, n, offset);
4033 }
4034
4035 static ssize_t fruit_pread_rsrc_xattr(vfs_handle_struct *handle,
4036                                       files_struct *fsp, void *data,
4037                                       size_t n, off_t offset)
4038 {
4039         return SMB_VFS_NEXT_PREAD(handle, fsp, data, n, offset);
4040 }
4041
4042 static ssize_t fruit_pread_rsrc_adouble(vfs_handle_struct *handle,
4043                                         files_struct *fsp, void *data,
4044                                         size_t n, off_t offset)
4045 {
4046         struct adouble *ad = NULL;
4047         ssize_t nread;
4048
4049         ad = ad_fget(talloc_tos(), handle, fsp, ADOUBLE_RSRC);
4050         if (ad == NULL) {
4051                 return -1;
4052         }
4053
4054         nread = SMB_VFS_NEXT_PREAD(handle, fsp, data, n,
4055                                    offset + ad_getentryoff(ad, ADEID_RFORK));
4056
4057         TALLOC_FREE(ad);
4058         return nread;
4059 }
4060
4061 static ssize_t fruit_pread_rsrc(vfs_handle_struct *handle,
4062                                 files_struct *fsp, void *data,
4063                                 size_t n, off_t offset)
4064 {
4065         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4066         ssize_t nread;
4067
4068         if (fio == NULL) {
4069                 errno = EINVAL;
4070                 return -1;
4071         }
4072
4073         switch (fio->config->rsrc) {
4074         case FRUIT_RSRC_STREAM:
4075                 nread = fruit_pread_rsrc_stream(handle, fsp, data, n, offset);
4076                 break;
4077
4078         case FRUIT_RSRC_ADFILE:
4079                 nread = fruit_pread_rsrc_adouble(handle, fsp, data, n, offset);
4080                 break;
4081
4082         case FRUIT_RSRC_XATTR:
4083                 nread = fruit_pread_rsrc_xattr(handle, fsp, data, n, offset);
4084                 break;
4085
4086         default:
4087                 DBG_ERR("Unexpected rsrc config [%d]\n", fio->config->rsrc);
4088                 return -1;
4089         }
4090
4091         return nread;
4092 }
4093
4094 static ssize_t fruit_pread(vfs_handle_struct *handle,
4095                            files_struct *fsp, void *data,
4096                            size_t n, off_t offset)
4097 {
4098         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4099         ssize_t nread;
4100
4101         DBG_DEBUG("Path [%s] offset=%"PRIdMAX", size=%zd\n",
4102                   fsp_str_dbg(fsp), (intmax_t)offset, n);
4103
4104         if (fio == NULL) {
4105                 return SMB_VFS_NEXT_PREAD(handle, fsp, data, n, offset);
4106         }
4107
4108         if (fio->type == ADOUBLE_META) {
4109                 nread = fruit_pread_meta(handle, fsp, data, n, offset);
4110         } else {
4111                 nread = fruit_pread_rsrc(handle, fsp, data, n, offset);
4112         }
4113
4114         DBG_DEBUG("Path [%s] nread [%zd]\n", fsp_str_dbg(fsp), nread);
4115         return nread;
4116 }
4117
4118 static bool fruit_must_handle_aio_stream(struct fio *fio)
4119 {
4120         if (fio == NULL) {
4121                 return false;
4122         };
4123
4124         if ((fio->type == ADOUBLE_META) &&
4125             (fio->config->meta == FRUIT_META_NETATALK))
4126         {
4127                 return true;
4128         }
4129
4130         if ((fio->type == ADOUBLE_RSRC) &&
4131             (fio->config->rsrc == FRUIT_RSRC_ADFILE))
4132         {
4133                 return true;
4134         }
4135
4136         return false;
4137 }
4138
4139 struct fruit_pread_state {
4140         ssize_t nread;
4141         struct vfs_aio_state vfs_aio_state;
4142 };
4143
4144 static void fruit_pread_done(struct tevent_req *subreq);
4145
4146 static struct tevent_req *fruit_pread_send(
4147         struct vfs_handle_struct *handle,
4148         TALLOC_CTX *mem_ctx,
4149         struct tevent_context *ev,
4150         struct files_struct *fsp,
4151         void *data,
4152         size_t n, off_t offset)
4153 {
4154         struct tevent_req *req = NULL;
4155         struct tevent_req *subreq = NULL;
4156         struct fruit_pread_state *state = NULL;
4157         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4158
4159         req = tevent_req_create(mem_ctx, &state,
4160                                 struct fruit_pread_state);
4161         if (req == NULL) {
4162                 return NULL;
4163         }
4164
4165         if (fruit_must_handle_aio_stream(fio)) {
4166                 state->nread = SMB_VFS_PREAD(fsp, data, n, offset);
4167                 if (state->nread != n) {
4168                         if (state->nread != -1) {
4169                                 errno = EIO;
4170                         }
4171                         tevent_req_error(req, errno);
4172                         return tevent_req_post(req, ev);
4173                 }
4174                 tevent_req_done(req);
4175                 return tevent_req_post(req, ev);
4176         }
4177
4178         subreq = SMB_VFS_NEXT_PREAD_SEND(state, ev, handle, fsp,
4179                                          data, n, offset);
4180         if (tevent_req_nomem(req, subreq)) {
4181                 return tevent_req_post(req, ev);
4182         }
4183         tevent_req_set_callback(subreq, fruit_pread_done, req);
4184         return req;
4185 }
4186
4187 static void fruit_pread_done(struct tevent_req *subreq)
4188 {
4189         struct tevent_req *req = tevent_req_callback_data(
4190                 subreq, struct tevent_req);
4191         struct fruit_pread_state *state = tevent_req_data(
4192                 req, struct fruit_pread_state);
4193
4194         state->nread = SMB_VFS_PREAD_RECV(subreq, &state->vfs_aio_state);
4195         TALLOC_FREE(subreq);
4196
4197         if (tevent_req_error(req, state->vfs_aio_state.error)) {
4198                 return;
4199         }
4200         tevent_req_done(req);
4201 }
4202
4203 static ssize_t fruit_pread_recv(struct tevent_req *req,
4204                                         struct vfs_aio_state *vfs_aio_state)
4205 {
4206         struct fruit_pread_state *state = tevent_req_data(
4207                 req, struct fruit_pread_state);
4208
4209         if (tevent_req_is_unix_error(req, &vfs_aio_state->error)) {
4210                 return -1;
4211         }
4212
4213         *vfs_aio_state = state->vfs_aio_state;
4214         return state->nread;
4215 }
4216
4217 static ssize_t fruit_pwrite_meta_stream(vfs_handle_struct *handle,
4218                                         files_struct *fsp, const void *data,
4219                                         size_t n, off_t offset)
4220 {
4221         AfpInfo *ai = NULL;
4222         size_t nwritten;
4223         bool ok;
4224
4225         ai = afpinfo_unpack(talloc_tos(), data);
4226         if (ai == NULL) {
4227                 return -1;
4228         }
4229
4230         nwritten = SMB_VFS_NEXT_PWRITE(handle, fsp, data, n, offset);
4231         if (nwritten != n) {
4232                 return -1;
4233         }
4234
4235         if (!ai_empty_finderinfo(ai)) {
4236                 return n;
4237         }
4238
4239         ok = set_delete_on_close(
4240                         fsp,
4241                         true,
4242                         handle->conn->session_info->security_token,
4243                         handle->conn->session_info->unix_token);
4244         if (!ok) {
4245                 DBG_ERR("set_delete_on_close on [%s] failed\n",
4246                         fsp_str_dbg(fsp));
4247                 return -1;
4248         }
4249
4250         return n;
4251 }
4252
4253 static ssize_t fruit_pwrite_meta_netatalk(vfs_handle_struct *handle,
4254                                           files_struct *fsp, const void *data,
4255                                           size_t n, off_t offset)
4256 {
4257         struct adouble *ad = NULL;
4258         AfpInfo *ai = NULL;
4259         char *p = NULL;
4260         int ret;
4261         bool ok;
4262
4263         ai = afpinfo_unpack(talloc_tos(), data);
4264         if (ai == NULL) {
4265                 return -1;
4266         }
4267
4268         ad = ad_fget(talloc_tos(), handle, fsp, ADOUBLE_META);
4269         if (ad == NULL) {
4270                 ad = ad_init(talloc_tos(), handle, ADOUBLE_META);
4271                 if (ad == NULL) {
4272                         return -1;
4273                 }
4274         }
4275         p = ad_get_entry(ad, ADEID_FINDERI);
4276         if (p == NULL) {
4277                 DBG_ERR("No ADEID_FINDERI for [%s]\n", fsp_str_dbg(fsp));
4278                 TALLOC_FREE(ad);
4279                 return -1;
4280         }
4281
4282         memcpy(p, &ai->afpi_FinderInfo[0], ADEDLEN_FINDERI);
4283
4284         ret = ad_fset(ad, fsp);
4285         if (ret != 0) {
4286                 DBG_ERR("ad_pwrite [%s] failed\n", fsp_str_dbg(fsp));
4287                 TALLOC_FREE(ad);
4288                 return -1;
4289         }
4290
4291         TALLOC_FREE(ad);
4292
4293         if (!ai_empty_finderinfo(ai)) {
4294                 return n;
4295         }
4296
4297         ok = set_delete_on_close(
4298                 fsp,
4299                 true,
4300                 handle->conn->session_info->security_token,
4301                 handle->conn->session_info->unix_token);
4302         if (!ok) {
4303                 DBG_ERR("set_delete_on_close on [%s] failed\n",
4304                         fsp_str_dbg(fsp));
4305                 return -1;
4306         }
4307
4308         return n;
4309 }
4310
4311 static ssize_t fruit_pwrite_meta(vfs_handle_struct *handle,
4312                                  files_struct *fsp, const void *data,
4313                                  size_t n, off_t offset)
4314 {
4315         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4316         ssize_t nwritten;
4317
4318         /*
4319          * Writing an all 0 blob to the metadata stream
4320          * results in the stream being removed on a macOS
4321          * server. This ensures we behave the same and it
4322          * verified by the "delete AFP_AfpInfo by writing all
4323          * 0" test.
4324          */
4325         if (n != AFP_INFO_SIZE || offset != 0) {
4326                 DBG_ERR("unexpected offset=%jd or size=%jd\n",
4327                         (intmax_t)offset, (intmax_t)n);
4328                 return -1;
4329         }
4330
4331         switch (fio->config->meta) {
4332         case FRUIT_META_STREAM:
4333                 nwritten = fruit_pwrite_meta_stream(handle, fsp, data,
4334                                                     n, offset);
4335                 break;
4336
4337         case FRUIT_META_NETATALK:
4338                 nwritten = fruit_pwrite_meta_netatalk(handle, fsp, data,
4339                                                       n, offset);
4340                 break;
4341
4342         default:
4343                 DBG_ERR("Unexpected meta config [%d]\n", fio->config->meta);
4344                 return -1;
4345         }
4346
4347         return nwritten;
4348 }
4349
4350 static ssize_t fruit_pwrite_rsrc_stream(vfs_handle_struct *handle,
4351                                         files_struct *fsp, const void *data,
4352                                         size_t n, off_t offset)
4353 {
4354         return SMB_VFS_NEXT_PWRITE(handle, fsp, data, n, offset);
4355 }
4356
4357 static ssize_t fruit_pwrite_rsrc_xattr(vfs_handle_struct *handle,
4358                                        files_struct *fsp, const void *data,
4359                                        size_t n, off_t offset)
4360 {
4361         return SMB_VFS_NEXT_PWRITE(handle, fsp, data, n, offset);
4362 }
4363
4364 static ssize_t fruit_pwrite_rsrc_adouble(vfs_handle_struct *handle,
4365                                          files_struct *fsp, const void *data,
4366                                          size_t n, off_t offset)
4367 {
4368         struct adouble *ad = NULL;
4369         ssize_t nwritten;
4370         int ret;
4371
4372         ad = ad_fget(talloc_tos(), handle, fsp, ADOUBLE_RSRC);
4373         if (ad == NULL) {
4374                 DBG_ERR("ad_get [%s] failed\n", fsp_str_dbg(fsp));
4375                 return -1;
4376         }
4377
4378         nwritten = SMB_VFS_NEXT_PWRITE(handle, fsp, data, n,
4379                                        offset + ad_getentryoff(ad, ADEID_RFORK));
4380         if (nwritten != n) {
4381                 DBG_ERR("Short write on [%s] [%zd/%zd]\n",
4382                         fsp_str_dbg(fsp), nwritten, n);
4383                 TALLOC_FREE(ad);
4384                 return -1;
4385         }
4386
4387         if ((n + offset) > ad_getentrylen(ad, ADEID_RFORK)) {
4388                 ad_setentrylen(ad, ADEID_RFORK, n + offset);
4389                 ret = ad_fset(ad, fsp);
4390                 if (ret != 0) {
4391                         DBG_ERR("ad_pwrite [%s] failed\n", fsp_str_dbg(fsp));
4392                         TALLOC_FREE(ad);
4393                         return -1;
4394                 }
4395         }
4396
4397         TALLOC_FREE(ad);
4398         return n;
4399 }
4400
4401 static ssize_t fruit_pwrite_rsrc(vfs_handle_struct *handle,
4402                                  files_struct *fsp, const void *data,
4403                                  size_t n, off_t offset)
4404 {
4405         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4406         ssize_t nwritten;
4407
4408         switch (fio->config->rsrc) {
4409         case FRUIT_RSRC_STREAM:
4410                 nwritten = fruit_pwrite_rsrc_stream(handle, fsp, data, n, offset);
4411                 break;
4412
4413         case FRUIT_RSRC_ADFILE:
4414                 nwritten = fruit_pwrite_rsrc_adouble(handle, fsp, data, n, offset);
4415                 break;
4416
4417         case FRUIT_RSRC_XATTR:
4418                 nwritten = fruit_pwrite_rsrc_xattr(handle, fsp, data, n, offset);
4419                 break;
4420
4421         default:
4422                 DBG_ERR("Unexpected rsrc config [%d]\n", fio->config->rsrc);
4423                 return -1;
4424         }
4425
4426         return nwritten;
4427 }
4428
4429 static ssize_t fruit_pwrite(vfs_handle_struct *handle,
4430                             files_struct *fsp, const void *data,
4431                             size_t n, off_t offset)
4432 {
4433         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4434         ssize_t nwritten;
4435
4436         DBG_DEBUG("Path [%s] offset=%"PRIdMAX", size=%zd\n",
4437                   fsp_str_dbg(fsp), (intmax_t)offset, n);
4438
4439         if (fio == NULL) {
4440                 return SMB_VFS_NEXT_PWRITE(handle, fsp, data, n, offset);
4441         }
4442
4443         if (fio->type == ADOUBLE_META) {
4444                 nwritten = fruit_pwrite_meta(handle, fsp, data, n, offset);
4445         } else {
4446                 nwritten = fruit_pwrite_rsrc(handle, fsp, data, n, offset);
4447         }
4448
4449         DBG_DEBUG("Path [%s] nwritten=%zd\n", fsp_str_dbg(fsp), nwritten);
4450         return nwritten;
4451 }
4452
4453 struct fruit_pwrite_state {
4454         ssize_t nwritten;
4455         struct vfs_aio_state vfs_aio_state;
4456 };
4457
4458 static void fruit_pwrite_done(struct tevent_req *subreq);
4459
4460 static struct tevent_req *fruit_pwrite_send(
4461         struct vfs_handle_struct *handle,
4462         TALLOC_CTX *mem_ctx,
4463         struct tevent_context *ev,
4464         struct files_struct *fsp,
4465         const void *data,
4466         size_t n, off_t offset)
4467 {
4468         struct tevent_req *req = NULL;
4469         struct tevent_req *subreq = NULL;
4470         struct fruit_pwrite_state *state = NULL;
4471         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4472
4473         req = tevent_req_create(mem_ctx, &state,
4474                                 struct fruit_pwrite_state);
4475         if (req == NULL) {
4476                 return NULL;
4477         }
4478
4479         if (fruit_must_handle_aio_stream(fio)) {
4480                 state->nwritten = SMB_VFS_PWRITE(fsp, data, n, offset);
4481                 if (state->nwritten != n) {
4482                         if (state->nwritten != -1) {
4483                                 errno = EIO;
4484                         }
4485                         tevent_req_error(req, errno);
4486                         return tevent_req_post(req, ev);
4487                 }
4488                 tevent_req_done(req);
4489                 return tevent_req_post(req, ev);
4490         }
4491
4492         subreq = SMB_VFS_NEXT_PWRITE_SEND(state, ev, handle, fsp,
4493                                           data, n, offset);
4494         if (tevent_req_nomem(req, subreq)) {
4495                 return tevent_req_post(req, ev);
4496         }
4497         tevent_req_set_callback(subreq, fruit_pwrite_done, req);
4498         return req;
4499 }
4500
4501 static void fruit_pwrite_done(struct tevent_req *subreq)
4502 {
4503         struct tevent_req *req = tevent_req_callback_data(
4504                 subreq, struct tevent_req);
4505         struct fruit_pwrite_state *state = tevent_req_data(
4506                 req, struct fruit_pwrite_state);
4507
4508         state->nwritten = SMB_VFS_PWRITE_RECV(subreq, &state->vfs_aio_state);
4509         TALLOC_FREE(subreq);
4510
4511         if (tevent_req_error(req, state->vfs_aio_state.error)) {
4512                 return;
4513         }
4514         tevent_req_done(req);
4515 }
4516
4517 static ssize_t fruit_pwrite_recv(struct tevent_req *req,
4518                                          struct vfs_aio_state *vfs_aio_state)
4519 {
4520         struct fruit_pwrite_state *state = tevent_req_data(
4521                 req, struct fruit_pwrite_state);
4522
4523         if (tevent_req_is_unix_error(req, &vfs_aio_state->error)) {
4524                 return -1;
4525         }
4526
4527         *vfs_aio_state = state->vfs_aio_state;
4528         return state->nwritten;
4529 }
4530
4531 /**
4532  * Helper to stat/lstat the base file of an smb_fname.
4533  */
4534 static int fruit_stat_base(vfs_handle_struct *handle,
4535                            struct smb_filename *smb_fname,
4536                            bool follow_links)
4537 {
4538         char *tmp_stream_name;
4539         int rc;
4540
4541         tmp_stream_name = smb_fname->stream_name;
4542         smb_fname->stream_name = NULL;
4543         if (follow_links) {
4544                 rc = SMB_VFS_NEXT_STAT(handle, smb_fname);
4545         } else {
4546                 rc = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
4547         }
4548         smb_fname->stream_name = tmp_stream_name;
4549         return rc;
4550 }
4551
4552 static int fruit_stat_meta_stream(vfs_handle_struct *handle,
4553                                   struct smb_filename *smb_fname,
4554                                   bool follow_links)
4555 {
4556         int ret;
4557
4558         if (follow_links) {
4559                 ret = SMB_VFS_NEXT_STAT(handle, smb_fname);
4560         } else {
4561                 ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
4562         }
4563
4564         return ret;
4565 }
4566
4567 static int fruit_stat_meta_netatalk(vfs_handle_struct *handle,
4568                                     struct smb_filename *smb_fname,
4569                                     bool follow_links)
4570 {
4571         struct adouble *ad = NULL;
4572
4573         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_META);
4574         if (ad == NULL) {
4575                 DBG_INFO("fruit_stat_meta %s: %s\n",
4576                          smb_fname_str_dbg(smb_fname), strerror(errno));
4577                 errno = ENOENT;
4578                 return -1;
4579         }
4580         TALLOC_FREE(ad);
4581
4582         /* Populate the stat struct with info from the base file. */
4583         if (fruit_stat_base(handle, smb_fname, follow_links) == -1) {
4584                 return -1;
4585         }
4586         smb_fname->st.st_ex_size = AFP_INFO_SIZE;
4587         smb_fname->st.st_ex_ino = fruit_inode(&smb_fname->st,
4588                                               smb_fname->stream_name);
4589         return 0;
4590 }
4591
4592 static int fruit_stat_meta(vfs_handle_struct *handle,
4593                            struct smb_filename *smb_fname,
4594                            bool follow_links)
4595 {
4596         struct fruit_config_data *config = NULL;
4597         int ret;
4598
4599         SMB_VFS_HANDLE_GET_DATA(handle, config,
4600                                 struct fruit_config_data, return -1);
4601
4602         switch (config->meta) {
4603         case FRUIT_META_STREAM:
4604                 ret = fruit_stat_meta_stream(handle, smb_fname, follow_links);
4605                 break;
4606
4607         case FRUIT_META_NETATALK:
4608                 ret = fruit_stat_meta_netatalk(handle, smb_fname, follow_links);
4609                 break;
4610
4611         default:
4612                 DBG_ERR("Unexpected meta config [%d]\n", config->meta);
4613                 return -1;
4614         }
4615
4616         return ret;
4617 }
4618
4619 static int fruit_stat_rsrc_netatalk(vfs_handle_struct *handle,
4620                                     struct smb_filename *smb_fname,
4621                                     bool follow_links)
4622 {
4623         struct adouble *ad = NULL;
4624         int ret;
4625
4626         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_RSRC);
4627         if (ad == NULL) {
4628                 errno = ENOENT;
4629                 return -1;
4630         }
4631
4632         /* Populate the stat struct with info from the base file. */
4633         ret = fruit_stat_base(handle, smb_fname, follow_links);
4634         if (ret != 0) {
4635                 TALLOC_FREE(ad);
4636                 return -1;
4637         }
4638
4639         smb_fname->st.st_ex_size = ad_getentrylen(ad, ADEID_RFORK);
4640         smb_fname->st.st_ex_ino = fruit_inode(&smb_fname->st,
4641                                               smb_fname->stream_name);
4642         TALLOC_FREE(ad);
4643         return 0;
4644 }
4645
4646 static int fruit_stat_rsrc_stream(vfs_handle_struct *handle,
4647                                   struct smb_filename *smb_fname,
4648                                   bool follow_links)
4649 {
4650         int ret;
4651
4652         if (follow_links) {
4653                 ret = SMB_VFS_NEXT_STAT(handle, smb_fname);
4654         } else {
4655                 ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
4656         }
4657
4658         return ret;
4659 }
4660
4661 static int fruit_stat_rsrc_xattr(vfs_handle_struct *handle,
4662                                  struct smb_filename *smb_fname,
4663                                  bool follow_links)
4664 {
4665 #ifdef HAVE_ATTROPEN
4666         int ret;
4667         int fd = -1;
4668
4669         /* Populate the stat struct with info from the base file. */
4670         ret = fruit_stat_base(handle, smb_fname, follow_links);
4671         if (ret != 0) {
4672                 return -1;
4673         }
4674
4675         fd = attropen(smb_fname->base_name,
4676                       AFPRESOURCE_EA_NETATALK,
4677                       O_RDONLY);
4678         if (fd == -1) {
4679                 return 0;
4680         }
4681
4682         ret = sys_fstat(fd, &smb_fname->st, false);
4683         if (ret != 0) {
4684                 close(fd);
4685                 DBG_ERR("fstat [%s:%s] failed\n", smb_fname->base_name,
4686                         AFPRESOURCE_EA_NETATALK);
4687                 return -1;
4688         }
4689         close(fd);
4690         fd = -1;
4691
4692         smb_fname->st.st_ex_ino = fruit_inode(&smb_fname->st,
4693                                               smb_fname->stream_name);
4694
4695         return ret;
4696
4697 #else
4698         errno = ENOSYS;
4699         return -1;
4700 #endif
4701 }
4702
4703 static int fruit_stat_rsrc(vfs_handle_struct *handle,
4704                            struct smb_filename *smb_fname,
4705                            bool follow_links)
4706 {
4707         struct fruit_config_data *config = NULL;
4708         int ret;
4709
4710         DBG_DEBUG("Path [%s]\n", smb_fname_str_dbg(smb_fname));
4711
4712         SMB_VFS_HANDLE_GET_DATA(handle, config,
4713                                 struct fruit_config_data, return -1);
4714
4715         switch (config->rsrc) {
4716         case FRUIT_RSRC_STREAM:
4717                 ret = fruit_stat_rsrc_stream(handle, smb_fname, follow_links);
4718                 break;
4719
4720         case FRUIT_RSRC_XATTR:
4721                 ret = fruit_stat_rsrc_xattr(handle, smb_fname, follow_links);
4722                 break;
4723
4724         case FRUIT_RSRC_ADFILE:
4725                 ret = fruit_stat_rsrc_netatalk(handle, smb_fname, follow_links);
4726                 break;
4727
4728         default:
4729                 DBG_ERR("Unexpected rsrc config [%d]\n", config->rsrc);
4730                 return -1;
4731         }
4732
4733         return ret;
4734 }
4735
4736 static int fruit_stat(vfs_handle_struct *handle,
4737                       struct smb_filename *smb_fname)
4738 {
4739         int rc = -1;
4740
4741         DEBUG(10, ("fruit_stat called for %s\n",
4742                    smb_fname_str_dbg(smb_fname)));
4743
4744         if (!is_ntfs_stream_smb_fname(smb_fname)
4745             || is_ntfs_default_stream_smb_fname(smb_fname)) {
4746                 rc = SMB_VFS_NEXT_STAT(handle, smb_fname);
4747                 if (rc == 0) {
4748                         update_btime(handle, smb_fname);
4749                 }
4750                 return rc;
4751         }
4752
4753         /*
4754          * Note if lp_posix_paths() is true, we can never
4755          * get here as is_ntfs_stream_smb_fname() is
4756          * always false. So we never need worry about
4757          * not following links here.
4758          */
4759
4760         if (is_afpinfo_stream(smb_fname)) {
4761                 rc = fruit_stat_meta(handle, smb_fname, true);
4762         } else if (is_afpresource_stream(smb_fname)) {
4763                 rc = fruit_stat_rsrc(handle, smb_fname, true);
4764         } else {
4765                 return SMB_VFS_NEXT_STAT(handle, smb_fname);
4766         }
4767
4768         if (rc == 0) {
4769                 update_btime(handle, smb_fname);
4770                 smb_fname->st.st_ex_mode &= ~S_IFMT;
4771                 smb_fname->st.st_ex_mode |= S_IFREG;
4772                 smb_fname->st.st_ex_blocks =
4773                         smb_fname->st.st_ex_size / STAT_ST_BLOCKSIZE + 1;
4774         }
4775         return rc;
4776 }
4777
4778 static int fruit_lstat(vfs_handle_struct *handle,
4779                        struct smb_filename *smb_fname)
4780 {
4781         int rc = -1;
4782
4783         DEBUG(10, ("fruit_lstat called for %s\n",
4784                    smb_fname_str_dbg(smb_fname)));
4785
4786         if (!is_ntfs_stream_smb_fname(smb_fname)
4787             || is_ntfs_default_stream_smb_fname(smb_fname)) {
4788                 rc = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
4789                 if (rc == 0) {
4790                         update_btime(handle, smb_fname);
4791                 }
4792                 return rc;
4793         }
4794
4795         if (is_afpinfo_stream(smb_fname)) {
4796                 rc = fruit_stat_meta(handle, smb_fname, false);
4797         } else if (is_afpresource_stream(smb_fname)) {
4798                 rc = fruit_stat_rsrc(handle, smb_fname, false);
4799         } else {
4800                 return SMB_VFS_NEXT_LSTAT(handle, smb_fname);
4801         }
4802
4803         if (rc == 0) {
4804                 update_btime(handle, smb_fname);
4805                 smb_fname->st.st_ex_mode &= ~S_IFMT;
4806                 smb_fname->st.st_ex_mode |= S_IFREG;
4807                 smb_fname->st.st_ex_blocks =
4808                         smb_fname->st.st_ex_size / STAT_ST_BLOCKSIZE + 1;
4809         }
4810         return rc;
4811 }
4812
4813 static int fruit_fstat_meta_stream(vfs_handle_struct *handle,
4814                                    files_struct *fsp,
4815                                    SMB_STRUCT_STAT *sbuf)
4816 {
4817         return SMB_VFS_NEXT_FSTAT(handle, fsp, sbuf);
4818 }
4819
4820 static int fruit_fstat_meta_netatalk(vfs_handle_struct *handle,
4821                                      files_struct *fsp,
4822                                      SMB_STRUCT_STAT *sbuf)
4823 {
4824         int ret;
4825
4826         ret = fruit_stat_base(handle, fsp->base_fsp->fsp_name, false);
4827         if (ret != 0) {
4828                 return -1;
4829         }
4830
4831         *sbuf = fsp->base_fsp->fsp_name->st;
4832         sbuf->st_ex_size = AFP_INFO_SIZE;
4833         sbuf->st_ex_ino = fruit_inode(sbuf, fsp->fsp_name->stream_name);
4834
4835         return 0;
4836 }
4837
4838 static int fruit_fstat_meta(vfs_handle_struct *handle,
4839                             files_struct *fsp,
4840                             SMB_STRUCT_STAT *sbuf,
4841                             struct fio *fio)
4842 {
4843         int ret;
4844
4845         DBG_DEBUG("Path [%s]\n", fsp_str_dbg(fsp));
4846
4847         switch (fio->config->meta) {
4848         case FRUIT_META_STREAM:
4849                 ret = fruit_fstat_meta_stream(handle, fsp, sbuf);
4850                 break;
4851
4852         case FRUIT_META_NETATALK:
4853                 ret = fruit_fstat_meta_netatalk(handle, fsp, sbuf);
4854                 break;
4855
4856         default:
4857                 DBG_ERR("Unexpected meta config [%d]\n", fio->config->meta);
4858                 return -1;
4859         }
4860
4861         DBG_DEBUG("Path [%s] ret [%d]\n", fsp_str_dbg(fsp), ret);
4862         return ret;
4863 }
4864
4865 static int fruit_fstat_rsrc_xattr(vfs_handle_struct *handle,
4866                                   files_struct *fsp,
4867                                   SMB_STRUCT_STAT *sbuf)
4868 {
4869         return SMB_VFS_NEXT_FSTAT(handle, fsp, sbuf);
4870 }
4871
4872 static int fruit_fstat_rsrc_stream(vfs_handle_struct *handle,
4873                                    files_struct *fsp,
4874                                    SMB_STRUCT_STAT *sbuf)
4875 {
4876         return SMB_VFS_NEXT_FSTAT(handle, fsp, sbuf);
4877 }
4878
4879 static int fruit_fstat_rsrc_adouble(vfs_handle_struct *handle,
4880                                     files_struct *fsp,
4881                                     SMB_STRUCT_STAT *sbuf)
4882 {
4883         struct adouble *ad = NULL;
4884         int ret;
4885
4886         /* Populate the stat struct with info from the base file. */
4887         ret = fruit_stat_base(handle, fsp->base_fsp->fsp_name, false);
4888         if (ret == -1) {
4889                 return -1;
4890         }
4891
4892         ad = ad_get(talloc_tos(), handle,
4893                     fsp->base_fsp->fsp_name,
4894                     ADOUBLE_RSRC);
4895         if (ad == NULL) {
4896                 DBG_ERR("ad_get [%s] failed [%s]\n",
4897                         fsp_str_dbg(fsp), strerror(errno));
4898                 return -1;
4899         }
4900
4901         *sbuf = fsp->base_fsp->fsp_name->st;
4902         sbuf->st_ex_size = ad_getentrylen(ad, ADEID_RFORK);
4903         sbuf->st_ex_ino = fruit_inode(sbuf, fsp->fsp_name->stream_name);
4904
4905         TALLOC_FREE(ad);
4906         return 0;
4907 }
4908
4909 static int fruit_fstat_rsrc(vfs_handle_struct *handle, files_struct *fsp,
4910                             SMB_STRUCT_STAT *sbuf, struct fio *fio)
4911 {
4912         int ret;
4913
4914         switch (fio->config->rsrc) {
4915         case FRUIT_RSRC_STREAM:
4916                 ret = fruit_fstat_rsrc_stream(handle, fsp, sbuf);
4917                 break;
4918
4919         case FRUIT_RSRC_ADFILE:
4920                 ret = fruit_fstat_rsrc_adouble(handle, fsp, sbuf);
4921                 break;
4922
4923         case FRUIT_RSRC_XATTR:
4924                 ret = fruit_fstat_rsrc_xattr(handle, fsp, sbuf);
4925                 break;
4926
4927         default:
4928                 DBG_ERR("Unexpected rsrc config [%d]\n", fio->config->rsrc);
4929                 return -1;
4930         }
4931
4932         return ret;
4933 }
4934
4935 static int fruit_fstat(vfs_handle_struct *handle, files_struct *fsp,
4936                        SMB_STRUCT_STAT *sbuf)
4937 {
4938         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
4939         int rc;
4940
4941         if (fio == NULL) {
4942                 return SMB_VFS_NEXT_FSTAT(handle, fsp, sbuf);
4943         }
4944
4945         DBG_DEBUG("Path [%s]\n", fsp_str_dbg(fsp));
4946
4947         if (fio->type == ADOUBLE_META) {
4948                 rc = fruit_fstat_meta(handle, fsp, sbuf, fio);
4949         } else {
4950                 rc = fruit_fstat_rsrc(handle, fsp, sbuf, fio);
4951         }
4952
4953         if (rc == 0) {
4954                 sbuf->st_ex_mode &= ~S_IFMT;
4955                 sbuf->st_ex_mode |= S_IFREG;
4956                 sbuf->st_ex_blocks = sbuf->st_ex_size / STAT_ST_BLOCKSIZE + 1;
4957         }
4958
4959         DBG_DEBUG("Path [%s] rc [%d] size [%"PRIdMAX"]\n",
4960                   fsp_str_dbg(fsp), rc, (intmax_t)sbuf->st_ex_size);
4961         return rc;
4962 }
4963
4964 static NTSTATUS delete_invalid_meta_stream(
4965         vfs_handle_struct *handle,
4966         const struct smb_filename *smb_fname,
4967         TALLOC_CTX *mem_ctx,
4968         unsigned int *pnum_streams,
4969         struct stream_struct **pstreams)
4970 {
4971         struct smb_filename *sname = NULL;
4972         int ret;
4973         bool ok;
4974
4975         ok = del_fruit_stream(mem_ctx, pnum_streams, pstreams, AFPINFO_STREAM);
4976         if (!ok) {
4977                 return NT_STATUS_INTERNAL_ERROR;
4978         }
4979
4980         sname = synthetic_smb_fname(talloc_tos(),
4981                                     smb_fname->base_name,
4982                                     AFPINFO_STREAM_NAME,
4983                                     NULL, 0);
4984         if (sname == NULL) {
4985                 return NT_STATUS_NO_MEMORY;
4986         }
4987
4988         ret = SMB_VFS_NEXT_UNLINK(handle, sname);
4989         TALLOC_FREE(sname);
4990         if (ret != 0) {
4991                 DBG_ERR("Removing [%s] failed\n", smb_fname_str_dbg(sname));
4992                 return map_nt_error_from_unix(errno);
4993         }
4994
4995         return NT_STATUS_OK;
4996 }
4997
4998 static NTSTATUS fruit_streaminfo_meta_stream(
4999         vfs_handle_struct *handle,
5000         struct files_struct *fsp,
5001         const struct smb_filename *smb_fname,
5002         TALLOC_CTX *mem_ctx,
5003         unsigned int *pnum_streams,
5004         struct stream_struct **pstreams)
5005 {
5006         struct stream_struct *stream = *pstreams;
5007         unsigned int num_streams = *pnum_streams;
5008         struct smb_filename *sname = NULL;
5009         char *full_name = NULL;
5010         uint32_t name_hash;
5011         struct share_mode_lock *lck = NULL;
5012         struct file_id id = {0};
5013         bool delete_on_close_set;
5014         int i;
5015         int ret;
5016         NTSTATUS status;
5017         bool ok;
5018
5019         for (i = 0; i < num_streams; i++) {
5020                 if (strequal_m(stream[i].name, AFPINFO_STREAM)) {
5021                         break;
5022                 }
5023         }
5024
5025         if (i == num_streams) {
5026                 return NT_STATUS_OK;
5027         }
5028
5029         if (stream[i].size != AFP_INFO_SIZE) {
5030                 DBG_ERR("Removing invalid AFPINFO_STREAM size [%jd] from [%s]\n",
5031                         (intmax_t)stream[i].size, smb_fname_str_dbg(smb_fname));
5032
5033                 return delete_invalid_meta_stream(handle, smb_fname, mem_ctx,
5034                                                   pnum_streams, pstreams);
5035         }
5036
5037         /*
5038          * Now check if there's a delete-on-close pending on the stream. If so,
5039          * hide the stream. This behaviour was verified against a macOS 10.12
5040          * SMB server.
5041          */
5042
5043         sname = synthetic_smb_fname(talloc_tos(),
5044                                     smb_fname->base_name,
5045                                     AFPINFO_STREAM_NAME,
5046                                     NULL, 0);
5047         if (sname == NULL) {
5048                 status = NT_STATUS_NO_MEMORY;
5049                 goto out;
5050         }
5051
5052         ret = SMB_VFS_NEXT_STAT(handle, sname);
5053         if (ret != 0) {
5054                 status = map_nt_error_from_unix(errno);
5055                 goto out;
5056         }
5057
5058         id = SMB_VFS_NEXT_FILE_ID_CREATE(handle, &sname->st);
5059
5060         lck = get_existing_share_mode_lock(talloc_tos(), id);
5061         if (lck == NULL) {
5062                 status = NT_STATUS_OK;
5063                 goto out;
5064         }
5065
5066         full_name = talloc_asprintf(talloc_tos(),
5067                                     "%s%s",
5068                                     sname->base_name,
5069                                     AFPINFO_STREAM);
5070         if (full_name == NULL) {
5071                 status = NT_STATUS_NO_MEMORY;
5072                 goto out;
5073         }
5074
5075         status = file_name_hash(handle->conn, full_name, &name_hash);
5076         if (!NT_STATUS_IS_OK(status)) {
5077                 goto out;
5078         }
5079
5080         delete_on_close_set = is_delete_on_close_set(lck, name_hash);
5081         if (delete_on_close_set) {
5082                 ok = del_fruit_stream(mem_ctx,
5083                                       pnum_streams,
5084                                       pstreams,
5085                                       AFPINFO_STREAM);
5086                 if (!ok) {
5087                         status = NT_STATUS_INTERNAL_ERROR;
5088                         goto out;
5089                 }
5090         }
5091
5092         status  = NT_STATUS_OK;
5093
5094 out:
5095         TALLOC_FREE(sname);
5096         TALLOC_FREE(lck);
5097         TALLOC_FREE(full_name);
5098         return status;
5099 }
5100
5101 static NTSTATUS fruit_streaminfo_meta_netatalk(
5102         vfs_handle_struct *handle,
5103         struct files_struct *fsp,
5104         const struct smb_filename *smb_fname,
5105         TALLOC_CTX *mem_ctx,
5106         unsigned int *pnum_streams,
5107         struct stream_struct **pstreams)
5108 {
5109         struct stream_struct *stream = *pstreams;
5110         unsigned int num_streams = *pnum_streams;
5111         struct adouble *ad = NULL;
5112         bool is_fi_empty;
5113         int i;
5114         bool ok;
5115
5116         /* Remove the Netatalk xattr from the list */
5117         ok = del_fruit_stream(mem_ctx, pnum_streams, pstreams,
5118                               ":" NETATALK_META_XATTR ":$DATA");
5119         if (!ok) {
5120                 return NT_STATUS_NO_MEMORY;
5121         }
5122
5123         /*
5124          * Check if there's a AFPINFO_STREAM from the VFS streams
5125          * backend and if yes, remove it from the list
5126          */
5127         for (i = 0; i < num_streams; i++) {
5128                 if (strequal_m(stream[i].name, AFPINFO_STREAM)) {
5129                         break;
5130                 }
5131         }
5132
5133         if (i < num_streams) {
5134                 DBG_WARNING("Unexpected AFPINFO_STREAM on [%s]\n",
5135                             smb_fname_str_dbg(smb_fname));
5136
5137                 ok = del_fruit_stream(mem_ctx, pnum_streams, pstreams,
5138                                       AFPINFO_STREAM);
5139                 if (!ok) {
5140                         return NT_STATUS_INTERNAL_ERROR;
5141                 }
5142         }
5143
5144         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_META);
5145         if (ad == NULL) {
5146                 return NT_STATUS_OK;
5147         }
5148
5149         is_fi_empty = ad_empty_finderinfo(ad);
5150         TALLOC_FREE(ad);
5151
5152         if (is_fi_empty) {
5153                 return NT_STATUS_OK;
5154         }
5155
5156         ok = add_fruit_stream(mem_ctx, pnum_streams, pstreams,
5157                               AFPINFO_STREAM_NAME, AFP_INFO_SIZE,
5158                               smb_roundup(handle->conn, AFP_INFO_SIZE));
5159         if (!ok) {
5160                 return NT_STATUS_NO_MEMORY;
5161         }
5162
5163         return NT_STATUS_OK;
5164 }
5165
5166 static NTSTATUS fruit_streaminfo_meta(vfs_handle_struct *handle,
5167                                       struct files_struct *fsp,
5168                                       const struct smb_filename *smb_fname,
5169                                       TALLOC_CTX *mem_ctx,
5170                                       unsigned int *pnum_streams,
5171                                       struct stream_struct **pstreams)
5172 {
5173         struct fruit_config_data *config = NULL;
5174         NTSTATUS status;
5175
5176         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
5177                                 return NT_STATUS_INTERNAL_ERROR);
5178
5179         switch (config->meta) {
5180         case FRUIT_META_NETATALK:
5181                 status = fruit_streaminfo_meta_netatalk(handle, fsp, smb_fname,
5182                                                         mem_ctx, pnum_streams,
5183                                                         pstreams);
5184                 break;
5185
5186         case FRUIT_META_STREAM:
5187                 status = fruit_streaminfo_meta_stream(handle, fsp, smb_fname,
5188                                                       mem_ctx, pnum_streams,
5189                                                       pstreams);
5190                 break;
5191
5192         default:
5193                 return NT_STATUS_INTERNAL_ERROR;
5194         }
5195
5196         return status;
5197 }
5198
5199 static NTSTATUS fruit_streaminfo_rsrc_stream(
5200         vfs_handle_struct *handle,
5201         struct files_struct *fsp,
5202         const struct smb_filename *smb_fname,
5203         TALLOC_CTX *mem_ctx,
5204         unsigned int *pnum_streams,
5205         struct stream_struct **pstreams)
5206 {
5207         bool ok;
5208
5209         ok = filter_empty_rsrc_stream(pnum_streams, pstreams);
5210         if (!ok) {
5211                 DBG_ERR("Filtering resource stream failed\n");
5212                 return NT_STATUS_INTERNAL_ERROR;
5213         }
5214         return NT_STATUS_OK;
5215 }
5216
5217 static NTSTATUS fruit_streaminfo_rsrc_xattr(
5218         vfs_handle_struct *handle,
5219         struct files_struct *fsp,
5220         const struct smb_filename *smb_fname,
5221         TALLOC_CTX *mem_ctx,
5222         unsigned int *pnum_streams,
5223         struct stream_struct **pstreams)
5224 {
5225         bool ok;
5226
5227         ok = filter_empty_rsrc_stream(pnum_streams, pstreams);
5228         if (!ok) {
5229                 DBG_ERR("Filtering resource stream failed\n");
5230                 return NT_STATUS_INTERNAL_ERROR;
5231         }
5232         return NT_STATUS_OK;
5233 }
5234
5235 static NTSTATUS fruit_streaminfo_rsrc_adouble(
5236         vfs_handle_struct *handle,
5237         struct files_struct *fsp,
5238         const struct smb_filename *smb_fname,
5239         TALLOC_CTX *mem_ctx,
5240         unsigned int *pnum_streams,
5241         struct stream_struct **pstreams)
5242 {
5243         struct stream_struct *stream = *pstreams;
5244         unsigned int num_streams = *pnum_streams;
5245         struct adouble *ad = NULL;
5246         bool ok;
5247         size_t rlen;
5248         int i;
5249
5250         /*
5251          * Check if there's a AFPRESOURCE_STREAM from the VFS streams backend
5252          * and if yes, remove it from the list
5253          */
5254         for (i = 0; i < num_streams; i++) {
5255                 if (strequal_m(stream[i].name, AFPRESOURCE_STREAM)) {
5256                         break;
5257                 }
5258         }
5259
5260         if (i < num_streams) {
5261                 DBG_WARNING("Unexpected AFPRESOURCE_STREAM on [%s]\n",
5262                             smb_fname_str_dbg(smb_fname));
5263
5264                 ok = del_fruit_stream(mem_ctx, pnum_streams, pstreams,
5265                                       AFPRESOURCE_STREAM);
5266                 if (!ok) {
5267                         return NT_STATUS_INTERNAL_ERROR;
5268                 }
5269         }
5270
5271         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_RSRC);
5272         if (ad == NULL) {
5273                 return NT_STATUS_OK;
5274         }
5275
5276         rlen = ad_getentrylen(ad, ADEID_RFORK);
5277         TALLOC_FREE(ad);
5278
5279         if (rlen == 0) {
5280                 return NT_STATUS_OK;
5281         }
5282
5283         ok = add_fruit_stream(mem_ctx, pnum_streams, pstreams,
5284                               AFPRESOURCE_STREAM_NAME, rlen,
5285                               smb_roundup(handle->conn, rlen));
5286         if (!ok) {
5287                 return NT_STATUS_NO_MEMORY;
5288         }
5289
5290         return NT_STATUS_OK;
5291 }
5292
5293 static NTSTATUS fruit_streaminfo_rsrc(vfs_handle_struct *handle,
5294                                       struct files_struct *fsp,
5295                                       const struct smb_filename *smb_fname,
5296                                       TALLOC_CTX *mem_ctx,
5297                                       unsigned int *pnum_streams,
5298                                       struct stream_struct **pstreams)
5299 {
5300         struct fruit_config_data *config = NULL;
5301         NTSTATUS status;
5302
5303         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
5304                                 return NT_STATUS_INTERNAL_ERROR);
5305
5306         switch (config->rsrc) {
5307         case FRUIT_RSRC_STREAM:
5308                 status = fruit_streaminfo_rsrc_stream(handle, fsp, smb_fname,
5309                                                       mem_ctx, pnum_streams,
5310                                                       pstreams);
5311                 break;
5312
5313         case FRUIT_RSRC_XATTR:
5314                 status = fruit_streaminfo_rsrc_xattr(handle, fsp, smb_fname,
5315                                                      mem_ctx, pnum_streams,
5316                                                      pstreams);
5317                 break;
5318
5319         case FRUIT_RSRC_ADFILE:
5320                 status = fruit_streaminfo_rsrc_adouble(handle, fsp, smb_fname,
5321                                                        mem_ctx, pnum_streams,
5322                                                        pstreams);
5323                 break;
5324
5325         default:
5326                 return NT_STATUS_INTERNAL_ERROR;
5327         }
5328
5329         return status;
5330 }
5331
5332 static NTSTATUS fruit_streaminfo(vfs_handle_struct *handle,
5333                                  struct files_struct *fsp,
5334                                  const struct smb_filename *smb_fname,
5335                                  TALLOC_CTX *mem_ctx,
5336                                  unsigned int *pnum_streams,
5337                                  struct stream_struct **pstreams)
5338 {
5339         struct fruit_config_data *config = NULL;
5340         NTSTATUS status;
5341
5342         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
5343                                 return NT_STATUS_UNSUCCESSFUL);
5344
5345         DBG_DEBUG("Path [%s]\n", smb_fname_str_dbg(smb_fname));
5346
5347         status = SMB_VFS_NEXT_STREAMINFO(handle, fsp, smb_fname, mem_ctx,
5348                                          pnum_streams, pstreams);
5349         if (!NT_STATUS_IS_OK(status)) {
5350                 return status;
5351         }
5352
5353         status = fruit_streaminfo_meta(handle, fsp, smb_fname,
5354                                        mem_ctx, pnum_streams, pstreams);
5355         if (!NT_STATUS_IS_OK(status)) {
5356                 return status;
5357         }
5358
5359         status = fruit_streaminfo_rsrc(handle, fsp, smb_fname,
5360                                        mem_ctx, pnum_streams, pstreams);
5361         if (!NT_STATUS_IS_OK(status)) {
5362                 return status;
5363         }
5364
5365         return NT_STATUS_OK;
5366 }
5367
5368 static int fruit_ntimes(vfs_handle_struct *handle,
5369                         const struct smb_filename *smb_fname,
5370                         struct smb_file_time *ft)
5371 {
5372         int rc = 0;
5373         struct adouble *ad = NULL;
5374         struct fruit_config_data *config = NULL;
5375
5376         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
5377                                 return -1);
5378
5379         if ((config->meta != FRUIT_META_NETATALK) ||
5380             null_timespec(ft->create_time))
5381         {
5382                 return SMB_VFS_NEXT_NTIMES(handle, smb_fname, ft);
5383         }
5384
5385         DEBUG(10,("set btime for %s to %s\n", smb_fname_str_dbg(smb_fname),
5386                  time_to_asc(convert_timespec_to_time_t(ft->create_time))));
5387
5388         ad = ad_get(talloc_tos(), handle, smb_fname, ADOUBLE_META);
5389         if (ad == NULL) {
5390                 goto exit;
5391         }
5392
5393         ad_setdate(ad, AD_DATE_CREATE | AD_DATE_UNIX,
5394                    convert_time_t_to_uint32_t(ft->create_time.tv_sec));
5395
5396         rc = ad_set(ad, smb_fname);
5397
5398 exit:
5399
5400         TALLOC_FREE(ad);
5401         if (rc != 0) {
5402                 DEBUG(1, ("fruit_ntimes: %s\n", smb_fname_str_dbg(smb_fname)));
5403                 return -1;
5404         }
5405         return SMB_VFS_NEXT_NTIMES(handle, smb_fname, ft);
5406 }
5407
5408 static int fruit_fallocate(struct vfs_handle_struct *handle,
5409                            struct files_struct *fsp,
5410                            uint32_t mode,
5411                            off_t offset,
5412                            off_t len)
5413 {
5414         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
5415
5416         if (fio == NULL) {
5417                 return SMB_VFS_NEXT_FALLOCATE(handle, fsp, mode, offset, len);
5418         }
5419
5420         /* Let the pwrite code path handle it. */
5421         errno = ENOSYS;
5422         return -1;
5423 }
5424
5425 static int fruit_ftruncate_rsrc_xattr(struct vfs_handle_struct *handle,
5426                                       struct files_struct *fsp,
5427                                       off_t offset)
5428 {
5429         if (offset == 0) {
5430                 return SMB_VFS_FREMOVEXATTR(fsp, AFPRESOURCE_EA_NETATALK);
5431         }
5432
5433 #ifdef HAVE_ATTROPEN
5434         return SMB_VFS_NEXT_FTRUNCATE(handle, fsp, offset);
5435 #endif
5436         return 0;
5437 }
5438
5439 static int fruit_ftruncate_rsrc_adouble(struct vfs_handle_struct *handle,
5440                                         struct files_struct *fsp,
5441                                         off_t offset)
5442 {
5443         int rc;
5444         struct adouble *ad = NULL;
5445         off_t ad_off;
5446
5447         ad = ad_fget(talloc_tos(), handle, fsp, ADOUBLE_RSRC);
5448         if (ad == NULL) {
5449                 DBG_DEBUG("ad_get [%s] failed [%s]\n",
5450                           fsp_str_dbg(fsp), strerror(errno));
5451                 return -1;
5452         }
5453
5454         ad_off = ad_getentryoff(ad, ADEID_RFORK);
5455
5456         rc = ftruncate(fsp->fh->fd, offset + ad_off);
5457         if (rc != 0) {
5458                 TALLOC_FREE(ad);
5459                 return -1;
5460         }
5461
5462         ad_setentrylen(ad, ADEID_RFORK, offset);
5463
5464         rc = ad_fset(ad, fsp);
5465         if (rc != 0) {
5466                 DBG_ERR("ad_fset [%s] failed [%s]\n",
5467                         fsp_str_dbg(fsp), strerror(errno));
5468                 TALLOC_FREE(ad);
5469                 return -1;
5470         }
5471
5472         TALLOC_FREE(ad);
5473         return 0;
5474 }
5475
5476 static int fruit_ftruncate_rsrc_stream(struct vfs_handle_struct *handle,
5477                                        struct files_struct *fsp,
5478                                        off_t offset)
5479 {
5480         if (offset == 0) {
5481                 return SMB_VFS_NEXT_UNLINK(handle, fsp->fsp_name);
5482         }
5483
5484         return SMB_VFS_NEXT_FTRUNCATE(handle, fsp, offset);
5485 }
5486
5487 static int fruit_ftruncate_rsrc(struct vfs_handle_struct *handle,
5488                                 struct files_struct *fsp,
5489                                 off_t offset)
5490 {
5491         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
5492         int ret;
5493
5494         switch (fio->config->rsrc) {
5495         case FRUIT_RSRC_XATTR:
5496                 ret = fruit_ftruncate_rsrc_xattr(handle, fsp, offset);
5497                 break;
5498
5499         case FRUIT_RSRC_ADFILE:
5500                 ret = fruit_ftruncate_rsrc_adouble(handle, fsp, offset);
5501                 break;
5502
5503         case FRUIT_RSRC_STREAM:
5504                 ret = fruit_ftruncate_rsrc_stream(handle, fsp, offset);
5505                 break;
5506
5507         default:
5508                 DBG_ERR("Unexpected rsrc config [%d]\n", fio->config->rsrc);
5509                 return -1;
5510         }
5511
5512
5513         return ret;
5514 }
5515
5516 static int fruit_ftruncate_meta(struct vfs_handle_struct *handle,
5517                                 struct files_struct *fsp,
5518                                 off_t offset)
5519 {
5520         if (offset > 60) {
5521                 DBG_WARNING("ftruncate %s to %jd",
5522                             fsp_str_dbg(fsp), (intmax_t)offset);
5523                 /* OS X returns NT_STATUS_ALLOTTED_SPACE_EXCEEDED  */
5524                 errno = EOVERFLOW;
5525                 return -1;
5526         }
5527
5528         /* OS X returns success but does nothing  */
5529         DBG_INFO("ignoring ftruncate %s to %jd\n",
5530                  fsp_str_dbg(fsp), (intmax_t)offset);
5531         return 0;
5532 }
5533
5534 static int fruit_ftruncate(struct vfs_handle_struct *handle,
5535                            struct files_struct *fsp,
5536                            off_t offset)
5537 {
5538         struct fio *fio = (struct fio *)VFS_FETCH_FSP_EXTENSION(handle, fsp);
5539         int ret;
5540
5541         DBG_DEBUG("Path [%s] offset [%"PRIdMAX"]\n", fsp_str_dbg(fsp),
5542                   (intmax_t)offset);
5543
5544         if (fio == NULL) {
5545                 return SMB_VFS_NEXT_FTRUNCATE(handle, fsp, offset);
5546         }
5547
5548         if (fio->type == ADOUBLE_META) {
5549                 ret = fruit_ftruncate_meta(handle, fsp, offset);
5550         } else {
5551                 ret = fruit_ftruncate_rsrc(handle, fsp, offset);
5552         }
5553
5554         DBG_DEBUG("Path [%s] result [%d]\n", fsp_str_dbg(fsp), ret);
5555         return ret;
5556 }
5557
5558 static NTSTATUS fruit_create_file(vfs_handle_struct *handle,
5559                                   struct smb_request *req,
5560                                   uint16_t root_dir_fid,
5561                                   struct smb_filename *smb_fname,
5562                                   uint32_t access_mask,
5563                                   uint32_t share_access,
5564                                   uint32_t create_disposition,
5565                                   uint32_t create_options,
5566                                   uint32_t file_attributes,
5567                                   uint32_t oplock_request,
5568                                   struct smb2_lease *lease,
5569                                   uint64_t allocation_size,
5570                                   uint32_t private_flags,
5571                                   struct security_descriptor *sd,
5572                                   struct ea_list *ea_list,
5573                                   files_struct **result,
5574                                   int *pinfo,
5575                                   const struct smb2_create_blobs *in_context_blobs,
5576                                   struct smb2_create_blobs *out_context_blobs)
5577 {
5578         NTSTATUS status;
5579         struct fruit_config_data *config = NULL;
5580         files_struct *fsp = NULL;
5581
5582         status = check_aapl(handle, req, in_context_blobs, out_context_blobs);
5583         if (!NT_STATUS_IS_OK(status)) {
5584                 goto fail;
5585         }
5586
5587         SMB_VFS_HANDLE_GET_DATA(handle, config, struct fruit_config_data,
5588                                 return NT_STATUS_UNSUCCESSFUL);
5589
5590         status = SMB_VFS_NEXT_CREATE_FILE(
5591                 handle, req, root_dir_fid, smb_fname,
5592                 access_mask, share_access,
5593                 create_disposition, create_options,
5594                 file_attributes, oplock_request,
5595                 lease,
5596                 allocation_size, private_flags,
5597                 sd, ea_list, result,
5598                 pinfo, in_context_blobs, out_context_blobs);
5599         if (!NT_STATUS_IS_OK(status)) {
5600                 return status;
5601         }
5602
5603         fsp = *result;
5604
5605         if (global_fruit_config.nego_aapl) {
5606                 if (config->posix_rename && fsp->is_directory) {
5607                         /*
5608                          * Enable POSIX directory rename behaviour
5609                          */
5610                         fsp->posix_flags |= FSP_POSIX_FLAGS_RENAME;
5611                 }
5612         }
5613
5614         /*
5615          * If this is a plain open for existing files, opening an 0
5616          * byte size resource fork MUST fail with
5617          * NT_STATUS_OBJECT_NAME_NOT_FOUND.
5618          *
5619          * Cf the vfs_fruit torture tests in test_rfork_create().
5620          */
5621         if (is_afpresource_stream(fsp->fsp_name) &&
5622             create_disposition == FILE_OPEN)
5623         {
5624                 if (fsp->fsp_name->st.st_ex_size == 0) {
5625                         status = NT_STATUS_OBJECT_NAME_NOT_FOUND;
5626                         goto fail;
5627                 }
5628         }
5629
5630         if (is_ntfs_stream_smb_fname(smb_fname)
5631             || fsp->is_directory) {
5632                 return status;
5633         }
5634
5635         if (config->locking == FRUIT_LOCKING_NETATALK) {
5636                 status = fruit_check_access(
5637                         handle, *result,
5638                         access_mask,
5639                         map_share_mode_to_deny_mode(share_access, 0));
5640                 if (!NT_STATUS_IS_OK(status)) {
5641                         goto fail;
5642                 }
5643         }
5644
5645         return status;
5646
5647 fail:
5648         DEBUG(10, ("fruit_create_file: %s\n", nt_errstr(status)));
5649
5650         if (fsp) {
5651                 close_file(req, fsp, ERROR_CLOSE);
5652                 *result = fsp = NULL;
5653         }
5654
5655         return status;
5656 }
5657
5658 static NTSTATUS fruit_readdir_attr(struct vfs_handle_struct *handle,
5659                                    const struct smb_filename *fname,
5660                                    TALLOC_CTX *mem_ctx,
5661                                    struct readdir_attr_data **pattr_data)
5662 {
5663         struct fruit_config_data *config = NULL;
5664         struct readdir_attr_data *attr_data;
5665         NTSTATUS status;
5666
5667         SMB_VFS_HANDLE_GET_DATA(handle, config,
5668                                 struct fruit_config_data,
5669                                 return NT_STATUS_UNSUCCESSFUL);
5670
5671         if (!global_fruit_config.nego_aapl) {
5672                 return SMB_VFS_NEXT_READDIR_ATTR(handle, fname, mem_ctx, pattr_data);
5673         }
5674
5675         DEBUG(10, ("fruit_readdir_attr %s\n", fname->base_name));
5676
5677         *pattr_data = talloc_zero(mem_ctx, struct readdir_attr_data);
5678         if (*pattr_data == NULL) {
5679                 return NT_STATUS_UNSUCCESSFUL;
5680         }
5681         attr_data = *pattr_data;
5682         attr_data->type = RDATTR_AAPL;
5683
5684         /*
5685          * Mac metadata: compressed FinderInfo, resource fork length
5686          * and creation date
5687          */
5688         status = readdir_attr_macmeta(handle, fname, attr_data);
5689         if (!NT_STATUS_IS_OK(status)) {
5690                 /*
5691                  * Error handling is tricky: if we return failure from
5692                  * this function, the corresponding directory entry
5693                  * will to be passed to the client, so we really just
5694                  * want to error out on fatal errors.
5695                  */
5696                 if  (!NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED)) {
5697                         goto fail;
5698                 }
5699         }
5700
5701         /*
5702          * UNIX mode
5703          */
5704         if (config->unix_info_enabled) {
5705                 attr_data->attr_data.aapl.unix_mode = fname->st.st_ex_mode;
5706         }
5707
5708         /*
5709          * max_access
5710          */
5711         if (!config->readdir_attr_max_access) {
5712                 attr_data->attr_data.aapl.max_access = FILE_GENERIC_ALL;
5713         } else {
5714                 status = smbd_calculate_access_mask(
5715                         handle->conn,
5716                         fname,
5717                         false,
5718                         SEC_FLAG_MAXIMUM_ALLOWED,
5719                         &attr_data->attr_data.aapl.max_access);
5720                 if (!NT_STATUS_IS_OK(status)) {
5721                         goto fail;
5722                 }
5723         }
5724
5725         return NT_STATUS_OK;
5726
5727 fail:
5728         DEBUG(1, ("fruit_readdir_attr %s, error: %s\n",
5729                   fname->base_name, nt_errstr(status)));
5730         TALLOC_FREE(*pattr_data);
5731         return status;
5732 }
5733
5734 static NTSTATUS fruit_fget_nt_acl(vfs_handle_struct *handle,
5735                                   files_struct *fsp,
5736                                   uint32_t security_info,
5737                                   TALLOC_CTX *mem_ctx,
5738                                   struct security_descriptor **ppdesc)
5739 {
5740         NTSTATUS status;
5741         struct security_ace ace;
5742         struct dom_sid sid;
5743         struct fruit_config_data *config;
5744
5745         SMB_VFS_HANDLE_GET_DATA(handle, config,
5746                                 struct fruit_config_data,
5747                                 return NT_STATUS_UNSUCCESSFUL);
5748
5749         status = SMB_VFS_NEXT_FGET_NT_ACL(handle, fsp, security_info,
5750                                           mem_ctx, ppdesc);
5751         if (!NT_STATUS_IS_OK(status)) {
5752                 return status;
5753         }
5754
5755         /*
5756          * Add MS NFS style ACEs with uid, gid and mode
5757          */
5758         if (!global_fruit_config.nego_aapl) {
5759                 return NT_STATUS_OK;
5760         }
5761         if (!config->unix_info_enabled) {
5762                 return NT_STATUS_OK;
5763         }
5764
5765         /* First remove any existing ACE's with NFS style mode/uid/gid SIDs. */
5766         status = remove_virtual_nfs_aces(*ppdesc);
5767         if (!NT_STATUS_IS_OK(status)) {
5768                 DBG_WARNING("failed to remove MS NFS style ACEs\n");
5769                 return status;
5770         }
5771
5772         /* MS NFS style mode */
5773         sid_compose(&sid, &global_sid_Unix_NFS_Mode, fsp->fsp_name->st.st_ex_mode);
5774         init_sec_ace(&ace, &sid, SEC_ACE_TYPE_ACCESS_DENIED, 0, 0);
5775         status = security_descriptor_dacl_add(*ppdesc, &ace);
5776         if (!NT_STATUS_IS_OK(status)) {
5777                 DEBUG(1,("failed to add MS NFS style ACE\n"));
5778                 return status;
5779         }
5780
5781         /* MS NFS style uid */
5782         sid_compose(&sid, &global_sid_Unix_NFS_Users, fsp->fsp_name->st.st_ex_uid);
5783         init_sec_ace(&ace, &sid, SEC_ACE_TYPE_ACCESS_DENIED, 0, 0);
5784         status = security_descriptor_dacl_add(*ppdesc, &ace);
5785         if (!NT_STATUS_IS_OK(status)) {
5786                 DEBUG(1,("failed to add MS NFS style ACE\n"));
5787                 return status;
5788         }
5789
5790         /* MS NFS style gid */
5791         sid_compose(&sid, &global_sid_Unix_NFS_Groups, fsp->fsp_name->st.st_ex_gid);
5792         init_sec_ace(&ace, &sid, SEC_ACE_TYPE_ACCESS_DENIED, 0, 0);
5793         status = security_descriptor_dacl_add(*ppdesc, &ace);
5794         if (!NT_STATUS_IS_OK(status)) {
5795                 DEBUG(1,("failed to add MS NFS style ACE\n"));
5796                 return status;
5797         }
5798
5799         return NT_STATUS_OK;
5800 }
5801
5802 static NTSTATUS fruit_fset_nt_acl(vfs_handle_struct *handle,
5803                                   files_struct *fsp,
5804                                   uint32_t security_info_sent,
5805                                   const struct security_descriptor *orig_psd)
5806 {
5807         NTSTATUS status;
5808         bool do_chmod;
5809         mode_t ms_nfs_mode = 0;
5810         int result;
5811         struct security_descriptor *psd = NULL;
5812         uint32_t orig_num_aces = 0;
5813
5814         if (orig_psd->dacl != NULL) {
5815                 orig_num_aces = orig_psd->dacl->num_aces;
5816         }
5817
5818         psd = security_descriptor_copy(talloc_tos(), orig_psd);
5819         if (psd == NULL) {
5820                 return NT_STATUS_NO_MEMORY;
5821         }
5822
5823         DBG_DEBUG("fruit_fset_nt_acl: %s\n", fsp_str_dbg(fsp));
5824
5825         status = check_ms_nfs(handle, fsp, psd, &ms_nfs_mode, &do_chmod);
5826         if (!NT_STATUS_IS_OK(status)) {
5827                 DEBUG(1, ("fruit_fset_nt_acl: check_ms_nfs failed%s\n", fsp_str_dbg(fsp)));
5828                 TALLOC_FREE(psd);
5829                 return status;
5830         }
5831
5832         /*
5833          * If only ms_nfs ACE entries were sent, ensure we set the DACL
5834          * sent/present flags correctly now we've removed them.
5835          */
5836
5837         if (orig_num_aces != 0) {
5838                 /*
5839                  * Are there any ACE's left ?
5840                  */
5841                 if (psd->dacl->num_aces == 0) {
5842                         /* No - clear the DACL sent/present flags. */
5843                         security_info_sent &= ~SECINFO_DACL;
5844                         psd->type &= ~SEC_DESC_DACL_PRESENT;
5845                 }
5846         }
5847
5848         status = SMB_VFS_NEXT_FSET_NT_ACL(handle, fsp, security_info_sent, psd);
5849         if (!NT_STATUS_IS_OK(status)) {
5850                 DEBUG(1, ("fruit_fset_nt_acl: SMB_VFS_NEXT_FSET_NT_ACL failed%s\n", fsp_str_dbg(fsp)));
5851                 TALLOC_FREE(psd);
5852                 return status;
5853         }
5854
5855         if (do_chmod) {
5856                 if (fsp->fh->fd != -1) {
5857                         result = SMB_VFS_FCHMOD(fsp, ms_nfs_mode);
5858                 } else {
5859                         result = SMB_VFS_CHMOD(fsp->conn,
5860                                                fsp->fsp_name,
5861                                                ms_nfs_mode);
5862                 }
5863
5864                 if (result != 0) {
5865                         DEBUG(1, ("chmod: %s, result: %d, %04o error %s\n", fsp_str_dbg(fsp),
5866                                   result, (unsigned)ms_nfs_mode,
5867                                   strerror(errno)));
5868                         status = map_nt_error_from_unix(errno);
5869                         TALLOC_FREE(psd);
5870                         return status;
5871                 }
5872         }
5873
5874         TALLOC_FREE(psd);
5875         return NT_STATUS_OK;
5876 }
5877
5878 static struct vfs_offload_ctx *fruit_offload_ctx;
5879
5880 struct fruit_offload_read_state {
5881         struct vfs_handle_struct *handle;
5882         struct tevent_context *ev;
5883         files_struct *fsp;
5884         uint32_t fsctl;
5885         DATA_BLOB token;
5886 };
5887
5888 static void fruit_offload_read_done(struct tevent_req *subreq);
5889
5890 static struct tevent_req *fruit_offload_read_send(
5891         TALLOC_CTX *mem_ctx,
5892         struct tevent_context *ev,
5893         struct vfs_handle_struct *handle,
5894         files_struct *fsp,
5895         uint32_t fsctl,
5896         uint32_t ttl,
5897         off_t offset,
5898         size_t to_copy)
5899 {
5900         struct tevent_req *req = NULL;
5901         struct tevent_req *subreq = NULL;
5902         struct fruit_offload_read_state *state = NULL;
5903
5904         req = tevent_req_create(mem_ctx, &state,
5905                                 struct fruit_offload_read_state);
5906         if (req == NULL) {
5907                 return NULL;
5908         }
5909         *state = (struct fruit_offload_read_state) {
5910                 .handle = handle,
5911                 .ev = ev,
5912                 .fsp = fsp,
5913                 .fsctl = fsctl,
5914         };
5915
5916         subreq = SMB_VFS_NEXT_OFFLOAD_READ_SEND(mem_ctx, ev, handle, fsp,
5917                                                 fsctl, ttl, offset, to_copy);
5918         if (tevent_req_nomem(subreq, req)) {
5919                 return tevent_req_post(req, ev);
5920         }
5921         tevent_req_set_callback(subreq, fruit_offload_read_done, req);
5922         return req;
5923 }
5924
5925 static void fruit_offload_read_done(struct tevent_req *subreq)
5926 {
5927         struct tevent_req *req = tevent_req_callback_data(
5928                 subreq, struct tevent_req);
5929         struct fruit_offload_read_state *state = tevent_req_data(
5930                 req, struct fruit_offload_read_state);
5931         NTSTATUS status;
5932
5933         status = SMB_VFS_NEXT_OFFLOAD_READ_RECV(subreq,
5934                                                 state->handle,
5935                                                 state,
5936                                                 &state->token);
5937         TALLOC_FREE(subreq);
5938         if (tevent_req_nterror(req, status)) {
5939                 return;
5940         }
5941
5942         if (state->fsctl != FSCTL_SRV_REQUEST_RESUME_KEY) {
5943                 tevent_req_done(req);
5944                 return;
5945         }
5946
5947         status = vfs_offload_token_ctx_init(state->fsp->conn->sconn->client,
5948                                             &fruit_offload_ctx);
5949         if (tevent_req_nterror(req, status)) {
5950                 return;
5951         }
5952
5953         status = vfs_offload_token_db_store_fsp(fruit_offload_ctx,
5954                                                 state->fsp,
5955                                                 &state->token);
5956         if (tevent_req_nterror(req, status)) {
5957                 return;
5958         }
5959
5960         tevent_req_done(req);
5961         return;
5962 }
5963
5964 static NTSTATUS fruit_offload_read_recv(struct tevent_req *req,
5965                                         struct vfs_handle_struct *handle,
5966                                         TALLOC_CTX *mem_ctx,
5967                                         DATA_BLOB *token)
5968 {
5969         struct fruit_offload_read_state *state = tevent_req_data(
5970                 req, struct fruit_offload_read_state);
5971         NTSTATUS status;
5972
5973         if (tevent_req_is_nterror(req, &status)) {
5974                 tevent_req_received(req);
5975                 return status;
5976         }
5977
5978         token->length = state->token.length;
5979         token->data = talloc_move(mem_ctx, &state->token.data);
5980
5981         tevent_req_received(req);
5982         return NT_STATUS_OK;
5983 }
5984
5985 struct fruit_offload_write_state {
5986         struct vfs_handle_struct *handle;
5987         off_t copied;
5988         struct files_struct *src_fsp;
5989         struct files_struct *dst_fsp;
5990         bool is_copyfile;
5991 };
5992
5993 static void fruit_offload_write_done(struct tevent_req *subreq);
5994 static struct tevent_req *fruit_offload_write_send(struct vfs_handle_struct *handle,
5995                                                 TALLOC_CTX *mem_ctx,
5996                                                 struct tevent_context *ev,
5997                                                 uint32_t fsctl,
5998                                                 DATA_BLOB *token,
5999                                                 off_t transfer_offset,
6000                                                 struct files_struct *dest_fsp,
6001                                                 off_t dest_off,
6002                                                 off_t num)
6003 {
6004         struct tevent_req *req, *subreq;
6005         struct fruit_offload_write_state *state;
6006         NTSTATUS status;
6007         struct fruit_config_data *config;
6008         off_t src_off = transfer_offset;
6009         files_struct *src_fsp = NULL;
6010         off_t to_copy = num;
6011         bool copyfile_enabled = false;
6012
6013         DEBUG(10,("soff: %ju, doff: %ju, len: %ju\n",
6014                   (uintmax_t)src_off, (uintmax_t)dest_off, (uintmax_t)num));
6015
6016         SMB_VFS_HANDLE_GET_DATA(handle, config,
6017                                 struct fruit_config_data,
6018                                 return NULL);
6019
6020         req = tevent_req_create(mem_ctx, &state,
6021                                 struct fruit_offload_write_state);
6022         if (req == NULL) {
6023                 return NULL;
6024         }
6025         state->handle = handle;
6026         state->dst_fsp = dest_fsp;
6027
6028         switch (fsctl) {
6029         case FSCTL_SRV_COPYCHUNK:
6030         case FSCTL_SRV_COPYCHUNK_WRITE:
6031                 copyfile_enabled = config->copyfile_enabled;
6032                 break;
6033         default:
6034                 break;
6035         }
6036
6037         /*
6038          * Check if this a OS X copyfile style copychunk request with
6039          * a requested chunk count of 0 that was translated to a
6040          * offload_write_send VFS call overloading the parameters src_off
6041          * = dest_off = num = 0.
6042          */
6043         if (copyfile_enabled && num == 0 && src_off == 0 && dest_off == 0) {
6044                 status = vfs_offload_token_db_fetch_fsp(
6045                         fruit_offload_ctx, token, &src_fsp);
6046                 if (tevent_req_nterror(req, status)) {
6047                         return tevent_req_post(req, ev);
6048                 }
6049                 state->src_fsp = src_fsp;
6050
6051                 status = vfs_stat_fsp(src_fsp);
6052                 if (tevent_req_nterror(req, status)) {
6053                         return tevent_req_post(req, ev);
6054                 }
6055
6056                 to_copy = src_fsp->fsp_name->st.st_ex_size;
6057                 state->is_copyfile = true;
6058         }
6059
6060         subreq = SMB_VFS_NEXT_OFFLOAD_WRITE_SEND(handle,
6061                                               mem_ctx,
6062                                               ev,
6063                                               fsctl,
6064                                               token,
6065                                               transfer_offset,
6066                                               dest_fsp,
6067                                               dest_off,
6068                                               to_copy);
6069         if (tevent_req_nomem(subreq, req)) {
6070                 return tevent_req_post(req, ev);
6071         }
6072
6073         tevent_req_set_callback(subreq, fruit_offload_write_done, req);
6074         return req;
6075 }
6076
6077 static void fruit_offload_write_done(struct tevent_req *subreq)
6078 {
6079         struct tevent_req *req = tevent_req_callback_data(
6080                 subreq, struct tevent_req);
6081         struct fruit_offload_write_state *state = tevent_req_data(
6082                 req, struct fruit_offload_write_state);
6083         NTSTATUS status;
6084         unsigned int num_streams = 0;
6085         struct stream_struct *streams = NULL;
6086         unsigned int i;
6087         struct smb_filename *src_fname_tmp = NULL;
6088         struct smb_filename *dst_fname_tmp = NULL;
6089
6090         status = SMB_VFS_NEXT_OFFLOAD_WRITE_RECV(state->handle,
6091                                               subreq,
6092                                               &state->copied);
6093         TALLOC_FREE(subreq);
6094         if (tevent_req_nterror(req, status)) {
6095                 return;
6096         }
6097
6098         if (!state->is_copyfile) {
6099                 tevent_req_done(req);
6100                 return;
6101         }
6102
6103         /*
6104          * Now copy all remaining streams. We know the share supports
6105          * streams, because we're in vfs_fruit. We don't do this async
6106          * because streams are few and small.
6107          */
6108         status = vfs_streaminfo(state->handle->conn, state->src_fsp,
6109                                 state->src_fsp->fsp_name,
6110                                 req, &num_streams, &streams);
6111         if (tevent_req_nterror(req, status)) {
6112                 return;
6113         }
6114
6115         if (num_streams == 1) {
6116                 /* There is always one stream, ::$DATA. */
6117                 tevent_req_done(req);
6118                 return;
6119         }
6120
6121         for (i = 0; i < num_streams; i++) {
6122                 DEBUG(10, ("%s: stream: '%s'/%zu\n",
6123                           __func__, streams[i].name, (size_t)streams[i].size));
6124
6125                 src_fname_tmp = synthetic_smb_fname(
6126                         req,
6127                         state->src_fsp->fsp_name->base_name,
6128                         streams[i].name,
6129                         NULL,
6130                         state->src_fsp->fsp_name->flags);
6131                 if (tevent_req_nomem(src_fname_tmp, req)) {
6132                         return;
6133                 }
6134
6135                 if (is_ntfs_default_stream_smb_fname(src_fname_tmp)) {
6136                         TALLOC_FREE(src_fname_tmp);
6137                         continue;
6138                 }
6139
6140                 dst_fname_tmp = synthetic_smb_fname(
6141                         req,
6142                         state->dst_fsp->fsp_name->base_name,
6143                         streams[i].name,
6144                         NULL,
6145                         state->dst_fsp->fsp_name->flags);
6146                 if (tevent_req_nomem(dst_fname_tmp, req)) {
6147                         TALLOC_FREE(src_fname_tmp);
6148                         return;
6149                 }
6150
6151                 status = copy_file(req,
6152                                    state->handle->conn,
6153                                    src_fname_tmp,
6154                                    dst_fname_tmp,
6155                                    OPENX_FILE_CREATE_IF_NOT_EXIST,
6156                                    0, false);
6157                 if (!NT_STATUS_IS_OK(status)) {
6158                         DEBUG(1, ("%s: copy %s to %s failed: %s\n", __func__,
6159                                   smb_fname_str_dbg(src_fname_tmp),
6160                                   smb_fname_str_dbg(dst_fname_tmp),
6161                                   nt_errstr(status)));
6162                         TALLOC_FREE(src_fname_tmp);
6163                         TALLOC_FREE(dst_fname_tmp);
6164                         tevent_req_nterror(req, status);
6165                         return;
6166                 }
6167
6168                 TALLOC_FREE(src_fname_tmp);
6169                 TALLOC_FREE(dst_fname_tmp);
6170         }
6171
6172         TALLOC_FREE(streams);
6173         TALLOC_FREE(src_fname_tmp);
6174         TALLOC_FREE(dst_fname_tmp);
6175         tevent_req_done(req);
6176 }
6177
6178 static NTSTATUS fruit_offload_write_recv(struct vfs_handle_struct *handle,
6179                                       struct tevent_req *req,
6180                                       off_t *copied)
6181 {
6182         struct fruit_offload_write_state *state = tevent_req_data(
6183                 req, struct fruit_offload_write_state);
6184         NTSTATUS status;
6185
6186         if (tevent_req_is_nterror(req, &status)) {
6187                 DEBUG(1, ("server side copy chunk failed: %s\n",
6188                           nt_errstr(status)));
6189                 *copied = 0;
6190                 tevent_req_received(req);
6191                 return status;
6192         }
6193
6194         *copied = state->copied;
6195         tevent_req_received(req);
6196
6197         return NT_STATUS_OK;
6198 }
6199
6200 static char *fruit_get_bandsize_line(char **lines, int numlines)
6201 {
6202         static regex_t re;
6203         static bool re_initialized = false;
6204         int i;
6205         int ret;
6206
6207         if (!re_initialized) {
6208                 ret = regcomp(&re, "^[[:blank:]]*<key>band-size</key>$", 0);
6209                 if (ret != 0) {
6210                         return NULL;
6211                 }
6212                 re_initialized = true;
6213         }
6214
6215         for (i = 0; i < numlines; i++) {
6216                 regmatch_t matches[1];
6217
6218                 ret = regexec(&re, lines[i], 1, matches, 0);
6219                 if (ret == 0) {
6220                         /*
6221                          * Check if the match was on the last line, sa we want
6222                          * the subsequent line.
6223                          */
6224                         if (i + 1 == numlines) {
6225                                 return NULL;
6226                         }
6227                         return lines[i + 1];
6228                 }
6229                 if (ret != REG_NOMATCH) {
6230                         return NULL;
6231                 }
6232         }
6233
6234         return NULL;
6235 }
6236
6237 static bool fruit_get_bandsize_from_line(char *line, size_t *_band_size)
6238 {
6239         static regex_t re;
6240         static bool re_initialized = false;
6241         regmatch_t matches[2];
6242         uint64_t band_size;
6243         int ret;
6244         bool ok;
6245
6246         if (!re_initialized) {
6247                 ret = regcomp(&re,
6248                               "^[[:blank:]]*"
6249                               "<integer>\\([[:digit:]]*\\)</integer>$",
6250                               0);
6251                 if (ret != 0) {
6252                         return false;
6253                 }
6254                 re_initialized = true;
6255         }
6256
6257         ret = regexec(&re, line, 2, matches, 0);
6258         if (ret != 0) {
6259                 DBG_ERR("regex failed [%s]\n", line);
6260                 return false;
6261         }
6262
6263         line[matches[1].rm_eo] = '\0';
6264
6265         ok = conv_str_u64(&line[matches[1].rm_so], &band_size);
6266         if (!ok) {
6267                 return false;
6268         }
6269         *_band_size = (size_t)band_size;
6270         return true;
6271 }
6272
6273 /*
6274  * This reads and parses an Info.plist from a TM sparsebundle looking for the
6275  * "band-size" key and value.
6276  */
6277 static bool fruit_get_bandsize(vfs_handle_struct *handle,
6278                                const char *dir,
6279                                size_t *band_size)
6280 {
6281 #define INFO_PLIST_MAX_SIZE 64*1024
6282         char *plist = NULL;
6283         struct smb_filename *smb_fname = NULL;
6284         files_struct *fsp = NULL;
6285         uint8_t *file_data = NULL;
6286         char **lines = NULL;
6287         char *band_size_line = NULL;
6288         size_t plist_file_size;
6289         ssize_t nread;
6290         int numlines;
6291         int ret;
6292         bool ok = false;
6293         NTSTATUS status;
6294
6295         plist = talloc_asprintf(talloc_tos(),
6296                                 "%s/%s/Info.plist",
6297                                 handle->conn->connectpath,
6298                                 dir);
6299         if (plist == NULL) {
6300                 ok = false;
6301                 goto out;
6302         }
6303
6304         smb_fname = synthetic_smb_fname(talloc_tos(), plist, NULL, NULL, 0);
6305         if (smb_fname == NULL) {
6306                 ok = false;
6307                 goto out;
6308         }
6309
6310         ret = SMB_VFS_NEXT_LSTAT(handle, smb_fname);
6311         if (ret != 0) {
6312                 DBG_INFO("Ignoring Sparsebundle without Info.plist [%s]\n", dir);
6313                 ok = true;
6314                 goto out;
6315         }
6316
6317         plist_file_size = smb_fname->st.st_ex_size;
6318
6319         if (plist_file_size > INFO_PLIST_MAX_SIZE) {
6320                 DBG_INFO("%s is too large, ignoring\n", plist);
6321                 ok = true;
6322                 goto out;
6323         }
6324
6325         status = SMB_VFS_NEXT_CREATE_FILE(
6326                 handle,                         /* conn */
6327                 NULL,                           /* req */
6328                 0,                              /* root_dir_fid */
6329                 smb_fname,                      /* fname */
6330                 FILE_GENERIC_READ,              /* access_mask */
6331                 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share_access */
6332                 FILE_OPEN,                      /* create_disposition */
6333                 0,                              /* create_options */
6334                 0,                              /* file_attributes */
6335                 INTERNAL_OPEN_ONLY,             /* oplock_request */
6336                 NULL,                           /* lease */
6337                 0,                              /* allocation_size */
6338                 0,                              /* private_flags */
6339                 NULL,                           /* sd */
6340                 NULL,                           /* ea_list */
6341                 &fsp,                           /* result */
6342                 NULL,                           /* psbuf */
6343                 NULL, NULL);                    /* create context */
6344         if (!NT_STATUS_IS_OK(status)) {
6345                 DBG_INFO("Opening [%s] failed [%s]\n",
6346                          smb_fname_str_dbg(smb_fname), nt_errstr(status));
6347                 ok = false;
6348                 goto out;
6349         }
6350
6351         file_data = talloc_array(talloc_tos(), uint8_t, plist_file_size);
6352         if (file_data == NULL) {
6353                 ok = false;
6354                 goto out;
6355         }
6356
6357         nread = SMB_VFS_NEXT_PREAD(handle, fsp, file_data, plist_file_size, 0);
6358         if (nread != plist_file_size) {
6359                 DBG_ERR("Short read on [%s]: %zu/%zd\n",
6360                         fsp_str_dbg(fsp), nread, plist_file_size);
6361                 ok = false;
6362                 goto out;
6363
6364         }
6365
6366         status = close_file(NULL, fsp, NORMAL_CLOSE);
6367         fsp = NULL;
6368         if (!NT_STATUS_IS_OK(status)) {
6369                 DBG_ERR("close_file failed: %s\n", nt_errstr(status));
6370                 ok = false;
6371                 goto out;
6372         }
6373
6374         lines = file_lines_parse((char *)file_data,
6375                                  plist_file_size,
6376                                  &numlines,
6377                                  talloc_tos());
6378         if (lines == NULL) {
6379                 ok = false;
6380                 goto out;
6381         }
6382
6383         band_size_line = fruit_get_bandsize_line(lines, numlines);
6384         if (band_size_line == NULL) {
6385                 DBG_ERR("Didn't find band-size key in [%s]\n",
6386                         smb_fname_str_dbg(smb_fname));
6387                 ok = false;
6388                 goto out;
6389         }
6390
6391         ok = fruit_get_bandsize_from_line(band_size_line, band_size);
6392         if (!ok) {
6393                 DBG_ERR("fruit_get_bandsize_from_line failed\n");
6394                 goto out;
6395         }
6396
6397         DBG_DEBUG("Parsed band-size [%zu] for [%s]\n", *band_size, plist);
6398
6399 out:
6400         if (fsp != NULL) {
6401                 status = close_file(NULL, fsp, NORMAL_CLOSE);
6402                 if (!NT_STATUS_IS_OK(status)) {
6403                         DBG_ERR("close_file failed: %s\n", nt_errstr(status));
6404                 }
6405                 fsp = NULL;
6406         }
6407         TALLOC_FREE(plist);
6408         TALLOC_FREE(smb_fname);
6409         TALLOC_FREE(file_data);
6410         TALLOC_FREE(lines);
6411         return ok;
6412 }
6413
6414 struct fruit_disk_free_state {
6415         off_t total_size;
6416 };
6417
6418 static bool fruit_get_num_bands(vfs_handle_struct *handle,
6419                                 char *bundle,
6420                                 size_t *_nbands)
6421 {
6422         char *path = NULL;
6423         struct smb_filename *bands_dir = NULL;
6424         DIR *d = NULL;
6425         struct dirent *e = NULL;
6426         size_t nbands;
6427         int ret;
6428
6429         path = talloc_asprintf(talloc_tos(),
6430                                "%s/%s/bands",
6431                                handle->conn->connectpath,
6432                                bundle);
6433         if (path == NULL) {
6434                 return false;
6435         }
6436
6437         bands_dir = synthetic_smb_fname(talloc_tos(),
6438                                         path,
6439                                         NULL,
6440                                         NULL,
6441                                         0);
6442         TALLOC_FREE(path);
6443         if (bands_dir == NULL) {
6444                 return false;
6445         }
6446
6447         d = SMB_VFS_NEXT_OPENDIR(handle, bands_dir, NULL, 0);
6448         if (d == NULL) {
6449                 TALLOC_FREE(bands_dir);
6450                 return false;
6451         }
6452
6453         nbands = 0;
6454
6455         for (e = SMB_VFS_NEXT_READDIR(handle, d, NULL);
6456              e != NULL;
6457              e = SMB_VFS_NEXT_READDIR(handle, d, NULL))
6458         {
6459                 if (ISDOT(e->d_name) || ISDOTDOT(e->d_name)) {
6460                         continue;
6461                 }
6462                 nbands++;
6463         }
6464
6465         ret = SMB_VFS_NEXT_CLOSEDIR(handle, d);
6466         if (ret != 0) {
6467                 TALLOC_FREE(bands_dir);
6468                 return false;
6469         }
6470
6471         DBG_DEBUG("%zu bands in [%s]\n", nbands, smb_fname_str_dbg(bands_dir));
6472
6473         TALLOC_FREE(bands_dir);
6474
6475         *_nbands = nbands;
6476         return true;
6477 }
6478
6479 static bool fruit_tmsize_do_dirent(vfs_handle_struct *handle,
6480                                    struct fruit_disk_free_state *state,
6481                                    struct dirent *e)
6482 {
6483         bool ok;
6484         char *p = NULL;
6485         size_t sparsebundle_strlen = strlen("sparsebundle");
6486         size_t bandsize = 0;
6487         size_t nbands;
6488         off_t tm_size;
6489
6490         p = strstr(e->d_name, "sparsebundle");
6491         if (p == NULL) {
6492                 return true;
6493         }
6494
6495         if (p[sparsebundle_strlen] != '\0') {
6496                 return true;
6497         }
6498
6499         DBG_DEBUG("Processing sparsebundle [%s]\n", e->d_name);
6500
6501         ok = fruit_get_bandsize(handle, e->d_name, &bandsize);
6502         if (!ok) {
6503                 /*
6504                  * Beware of race conditions: this may be an uninitialized
6505                  * Info.plist that a client is just creating. We don't want let
6506                  * this to trigger complete failure.
6507                  */
6508                 DBG_ERR("Processing sparsebundle [%s] failed\n", e->d_name);
6509                 return true;
6510         }
6511
6512         ok = fruit_get_num_bands(handle, e->d_name, &nbands);
6513         if (!ok) {
6514                 /*
6515                  * Beware of race conditions: this may be a backup sparsebundle
6516                  * in an early stage lacking a bands subdirectory. We don't want
6517                  * let this to trigger complete failure.
6518                  */
6519                 DBG_ERR("Processing sparsebundle [%s] failed\n", e->d_name);
6520                 return true;
6521         }
6522
6523         if (bandsize > SIZE_MAX/nbands) {
6524                 DBG_ERR("tmsize overflow: bandsize [%zu] nbands [%zu]\n",
6525                         bandsize, nbands);
6526                 return false;
6527         }
6528         tm_size = bandsize * nbands;
6529
6530         if (state->total_size + tm_size < state->total_size) {
6531                 DBG_ERR("tmsize overflow: bandsize [%zu] nbands [%zu]\n",
6532                         bandsize, nbands);
6533                 return false;
6534         }
6535
6536         state->total_size += tm_size;
6537
6538         DBG_DEBUG("[%s] tm_size [%jd] total_size [%jd]\n",
6539                   e->d_name, (intmax_t)tm_size, (intmax_t)state->total_size);
6540
6541         return true;
6542 }
6543
6544 /**
6545  * Calculate used size of a TimeMachine volume
6546  *
6547  * This assumes that the volume is used only for TimeMachine.
6548  *
6549  * - readdir(basedir of share), then
6550  * - for every element that matches regex "^\(.*\)\.sparsebundle$" :
6551  * - parse "\1.sparsebundle/Info.plist" and read the band-size XML key
6552  * - count band files in "\1.sparsebundle/bands/"
6553  * - calculate used size of all bands: band_count * band_size
6554  **/
6555 static uint64_t fruit_disk_free(vfs_handle_struct *handle,
6556                                 const struct smb_filename *smb_fname,
6557                                 uint64_t *_bsize,
6558                                 uint64_t *_dfree,
6559                                 uint64_t *_dsize)
6560 {
6561         struct fruit_config_data *config = NULL;
6562         struct fruit_disk_free_state state = {0};
6563         DIR *d = NULL;
6564         struct dirent *e = NULL;
6565         uint64_t dfree;
6566         uint64_t dsize;
6567         int ret;
6568         bool ok;
6569
6570         SMB_VFS_HANDLE_GET_DATA(handle, config,
6571                                 struct fruit_config_data,
6572                                 return UINT64_MAX);
6573
6574         if (!config->time_machine ||
6575             config->time_machine_max_size == 0)
6576         {
6577                 return SMB_VFS_NEXT_DISK_FREE(handle,
6578                                               smb_fname,
6579                                               _bsize,
6580                                               _dfree,
6581                                               _dsize);
6582         }
6583
6584         d = SMB_VFS_NEXT_OPENDIR(handle, smb_fname, NULL, 0);
6585         if (d == NULL) {
6586                 return UINT64_MAX;
6587         }
6588
6589         for (e = SMB_VFS_NEXT_READDIR(handle, d, NULL);
6590              e != NULL;
6591              e = SMB_VFS_NEXT_READDIR(handle, d, NULL))
6592         {
6593                 ok = fruit_tmsize_do_dirent(handle, &state, e);
6594                 if (!ok) {
6595                         SMB_VFS_NEXT_CLOSEDIR(handle, d);
6596                         return UINT64_MAX;
6597                 }
6598         }
6599
6600         ret = SMB_VFS_NEXT_CLOSEDIR(handle, d);
6601         if (ret != 0) {
6602                 return UINT64_MAX;
6603         }
6604
6605         dsize = config->time_machine_max_size / 512;
6606         dfree = dsize - (state.total_size / 512);
6607         if (dfree > dsize) {
6608                 dfree = 0;
6609         }
6610
6611         *_bsize = 512;
6612         *_dsize = dsize;
6613         *_dfree = dfree;
6614         return dfree / 2;
6615 }
6616
6617 static struct vfs_fn_pointers vfs_fruit_fns = {
6618         .connect_fn = fruit_connect,
6619         .disk_free_fn = fruit_disk_free,
6620
6621         /* File operations */
6622         .chmod_fn = fruit_chmod,
6623         .chown_fn = fruit_chown,
6624         .unlink_fn = fruit_unlink,
6625         .rename_fn = fruit_rename,
6626         .rmdir_fn = fruit_rmdir,
6627         .open_fn = fruit_open,
6628         .pread_fn = fruit_pread,
6629         .pwrite_fn = fruit_pwrite,
6630         .pread_send_fn = fruit_pread_send,
6631         .pread_recv_fn = fruit_pread_recv,
6632         .pwrite_send_fn = fruit_pwrite_send,
6633         .pwrite_recv_fn = fruit_pwrite_recv,
6634         .stat_fn = fruit_stat,
6635         .lstat_fn = fruit_lstat,
6636         .fstat_fn = fruit_fstat,
6637         .streaminfo_fn = fruit_streaminfo,
6638         .ntimes_fn = fruit_ntimes,
6639         .ftruncate_fn = fruit_ftruncate,
6640         .fallocate_fn = fruit_fallocate,
6641         .create_file_fn = fruit_create_file,
6642         .readdir_attr_fn = fruit_readdir_attr,
6643         .offload_read_send_fn = fruit_offload_read_send,
6644         .offload_read_recv_fn = fruit_offload_read_recv,
6645         .offload_write_send_fn = fruit_offload_write_send,
6646         .offload_write_recv_fn = fruit_offload_write_recv,
6647
6648         /* NT ACL operations */
6649         .fget_nt_acl_fn = fruit_fget_nt_acl,
6650         .fset_nt_acl_fn = fruit_fset_nt_acl,
6651 };
6652
6653 static_decl_vfs;
6654 NTSTATUS vfs_fruit_init(TALLOC_CTX *ctx)
6655 {
6656         NTSTATUS ret = smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "fruit",
6657                                         &vfs_fruit_fns);
6658         if (!NT_STATUS_IS_OK(ret)) {
6659                 return ret;
6660         }
6661
6662         vfs_fruit_debug_level = debug_add_class("fruit");
6663         if (vfs_fruit_debug_level == -1) {
6664                 vfs_fruit_debug_level = DBGC_VFS;
6665                 DEBUG(0, ("%s: Couldn't register custom debugging class!\n",
6666                           "vfs_fruit_init"));
6667         } else {
6668                 DEBUG(10, ("%s: Debug class number of '%s': %d\n",
6669                            "vfs_fruit_init","fruit",vfs_fruit_debug_level));
6670         }
6671
6672         return ret;
6673 }