source: branches/samba-3.5.x/source3/modules/vfs_streams_depot.c

Last change on this file was 414, checked in by Herwig Bauernfeind, 15 years ago

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