Do proper error handling if the socket is closed
[Samba/gebeck_regimport.git] / source3 / modules / vfs_shadow_copy.c
blobfbd2960abaecd281b28832e07608bba658e524a1
1 /*
2 * implementation of an Shadow Copy module
4 * Copyright (C) Stefan Metzmacher 2003-2004
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.
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.
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/>.
20 #include "includes.h"
23 Please read the VFS module Samba-HowTo-Collection.
24 there's a chapter about this module
26 For this share
27 Z:\
29 the ShadowCopies are in this directories
31 Z:\@GMT-2003.08.05-12.00.00\
32 Z:\@GMT-2003.08.05-12.01.00\
33 Z:\@GMT-2003.08.05-12.02.00\
35 e.g.
37 Z:\testfile.txt
38 Z:\@GMT-2003.08.05-12.02.00\testfile.txt
40 or:
42 Z:\testdir\testfile.txt
43 Z:\@GMT-2003.08.05-12.02.00\testdir\testfile.txt
46 Note: Files must differ to be displayed via Windows Explorer!
47 Directories are always displayed...
50 static int vfs_shadow_copy_debug_level = DBGC_VFS;
52 #undef DBGC_CLASS
53 #define DBGC_CLASS vfs_shadow_copy_debug_level
55 #define SHADOW_COPY_PREFIX "@GMT-"
56 #define SHADOW_COPY_SAMPLE "@GMT-2004.02.18-15.44.00"
58 typedef struct {
59 int pos;
60 int num;
61 SMB_STRUCT_DIRENT *dirs;
62 } shadow_copy_Dir;
64 static bool shadow_copy_match_name(const char *name)
66 if (strncmp(SHADOW_COPY_PREFIX,name, sizeof(SHADOW_COPY_PREFIX)-1)==0 &&
67 (strlen(SHADOW_COPY_SAMPLE) == strlen(name))) {
68 return True;
71 return False;
74 static SMB_STRUCT_DIR *shadow_copy_opendir(vfs_handle_struct *handle, const char *fname, const char *mask, uint32 attr)
76 shadow_copy_Dir *dirp;
77 SMB_STRUCT_DIR *p = SMB_VFS_NEXT_OPENDIR(handle,fname,mask,attr);
79 if (!p) {
80 DEBUG(0,("shadow_copy_opendir: SMB_VFS_NEXT_OPENDIR() failed for [%s]\n",fname));
81 return NULL;
84 dirp = SMB_MALLOC_P(shadow_copy_Dir);
85 if (!dirp) {
86 DEBUG(0,("shadow_copy_opendir: Out of memory\n"));
87 SMB_VFS_NEXT_CLOSEDIR(handle,p);
88 return NULL;
91 ZERO_STRUCTP(dirp);
93 while (True) {
94 SMB_STRUCT_DIRENT *d;
96 d = SMB_VFS_NEXT_READDIR(handle, p);
97 if (d == NULL) {
98 break;
101 if (shadow_copy_match_name(d->d_name)) {
102 DEBUG(8,("shadow_copy_opendir: hide [%s]\n",d->d_name));
103 continue;
106 DEBUG(10,("shadow_copy_opendir: not hide [%s]\n",d->d_name));
108 dirp->dirs = SMB_REALLOC_ARRAY(dirp->dirs,SMB_STRUCT_DIRENT, dirp->num+1);
109 if (!dirp->dirs) {
110 DEBUG(0,("shadow_copy_opendir: Out of memory\n"));
111 break;
114 dirp->dirs[dirp->num++] = *d;
117 SMB_VFS_NEXT_CLOSEDIR(handle,p);
118 return((SMB_STRUCT_DIR *)dirp);
121 static SMB_STRUCT_DIRENT *shadow_copy_readdir(vfs_handle_struct *handle, SMB_STRUCT_DIR *_dirp)
123 shadow_copy_Dir *dirp = (shadow_copy_Dir *)_dirp;
125 if (dirp->pos < dirp->num) {
126 return &(dirp->dirs[dirp->pos++]);
129 return NULL;
132 static void shadow_copy_seekdir(struct vfs_handle_struct *handle, SMB_STRUCT_DIR *_dirp, long offset)
134 shadow_copy_Dir *dirp = (shadow_copy_Dir *)_dirp;
136 if (offset < dirp->num) {
137 dirp->pos = offset ;
141 static long shadow_copy_telldir(struct vfs_handle_struct *handle, SMB_STRUCT_DIR *_dirp)
143 shadow_copy_Dir *dirp = (shadow_copy_Dir *)_dirp;
144 return( dirp->pos ) ;
147 static void shadow_copy_rewinddir(struct vfs_handle_struct *handle, SMB_STRUCT_DIR *_dirp)
149 shadow_copy_Dir *dirp = (shadow_copy_Dir *)_dirp;
150 dirp->pos = 0 ;
153 static int shadow_copy_closedir(vfs_handle_struct *handle, SMB_STRUCT_DIR *_dirp)
155 shadow_copy_Dir *dirp = (shadow_copy_Dir *)_dirp;
157 SAFE_FREE(dirp->dirs);
158 SAFE_FREE(dirp);
160 return 0;
163 static int shadow_copy_get_shadow_copy_data(vfs_handle_struct *handle, files_struct *fsp, SHADOW_COPY_DATA *shadow_copy_data, bool labels)
165 SMB_STRUCT_DIR *p = SMB_VFS_NEXT_OPENDIR(handle,fsp->conn->connectpath,NULL,0);
167 shadow_copy_data->num_volumes = 0;
168 shadow_copy_data->labels = NULL;
170 if (!p) {
171 DEBUG(0,("shadow_copy_get_shadow_copy_data: SMB_VFS_NEXT_OPENDIR() failed for [%s]\n",fsp->conn->connectpath));
172 return -1;
175 while (True) {
176 SHADOW_COPY_LABEL *tlabels;
177 SMB_STRUCT_DIRENT *d;
179 d = SMB_VFS_NEXT_READDIR(handle, p);
180 if (d == NULL) {
181 break;
184 /* */
185 if (!shadow_copy_match_name(d->d_name)) {
186 DEBUG(10,("shadow_copy_get_shadow_copy_data: ignore [%s]\n",d->d_name));
187 continue;
190 DEBUG(7,("shadow_copy_get_shadow_copy_data: not ignore [%s]\n",d->d_name));
192 if (!labels) {
193 shadow_copy_data->num_volumes++;
194 continue;
197 tlabels = (SHADOW_COPY_LABEL *)TALLOC_REALLOC(shadow_copy_data->mem_ctx,
198 shadow_copy_data->labels,
199 (shadow_copy_data->num_volumes+1)*sizeof(SHADOW_COPY_LABEL));
200 if (tlabels == NULL) {
201 DEBUG(0,("shadow_copy_get_shadow_copy_data: Out of memory\n"));
202 SMB_VFS_NEXT_CLOSEDIR(handle,p);
203 return -1;
206 snprintf(tlabels[shadow_copy_data->num_volumes++], sizeof(*tlabels), "%s",d->d_name);
208 shadow_copy_data->labels = tlabels;
211 SMB_VFS_NEXT_CLOSEDIR(handle,p);
212 return 0;
215 /* VFS operations structure */
217 static vfs_op_tuple shadow_copy_ops[] = {
218 {SMB_VFS_OP(shadow_copy_opendir), SMB_VFS_OP_OPENDIR, SMB_VFS_LAYER_TRANSPARENT},
219 {SMB_VFS_OP(shadow_copy_readdir), SMB_VFS_OP_READDIR, SMB_VFS_LAYER_TRANSPARENT},
220 {SMB_VFS_OP(shadow_copy_seekdir), SMB_VFS_OP_SEEKDIR, SMB_VFS_LAYER_TRANSPARENT},
221 {SMB_VFS_OP(shadow_copy_telldir), SMB_VFS_OP_TELLDIR, SMB_VFS_LAYER_TRANSPARENT},
222 {SMB_VFS_OP(shadow_copy_rewinddir), SMB_VFS_OP_REWINDDIR, SMB_VFS_LAYER_TRANSPARENT},
223 {SMB_VFS_OP(shadow_copy_closedir), SMB_VFS_OP_CLOSEDIR, SMB_VFS_LAYER_TRANSPARENT},
225 {SMB_VFS_OP(shadow_copy_get_shadow_copy_data), SMB_VFS_OP_GET_SHADOW_COPY_DATA,SMB_VFS_LAYER_OPAQUE},
227 {SMB_VFS_OP(NULL), SMB_VFS_OP_NOOP, SMB_VFS_LAYER_NOOP}
230 NTSTATUS vfs_shadow_copy_init(void);
231 NTSTATUS vfs_shadow_copy_init(void)
233 NTSTATUS ret = smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "shadow_copy", shadow_copy_ops);
235 if (!NT_STATUS_IS_OK(ret))
236 return ret;
238 vfs_shadow_copy_debug_level = debug_add_class("shadow_copy");
239 if (vfs_shadow_copy_debug_level == -1) {
240 vfs_shadow_copy_debug_level = DBGC_VFS;
241 DEBUG(0, ("%s: Couldn't register custom debugging class!\n",
242 "vfs_shadow_copy_init"));
243 } else {
244 DEBUG(10, ("%s: Debug class number of '%s': %d\n",
245 "vfs_shadow_copy_init","shadow_copy",vfs_shadow_copy_debug_level));
248 return ret;