s3 OneFS: Add recvfile implementation
[samba.git] / source3 / modules / vfs_onefs.c
1 /*
2  * Unix SMB/CIFS implementation.
3  * Support for OneFS
4  *
5  * Copyright (C) Tim Prouty, 2008
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include "includes.h"
22 #include "onefs.h"
23
24 #undef DBGC_CLASS
25 #define DBGC_CLASS DBGC_VFS
26
27 #define ONEFS_DATA_FASTBUF      10
28
29 struct onefs_vfs_config share_config[ONEFS_DATA_FASTBUF];
30 struct onefs_vfs_config *pshare_config;
31
32 static void onefs_load_faketimestamp_config(struct vfs_handle_struct *handle,
33                                             struct onefs_vfs_config *cfg)
34 {
35         const char **parm;
36         int snum = SNUM(handle->conn);
37
38         parm = lp_parm_string_list(snum, PARM_ONEFS_TYPE, PARM_ATIME_NOW,
39                                    NULL);
40
41         if (parm) {
42                 cfg->init_flags |= ONEFS_VFS_CONFIG_FAKETIMESTAMPS;
43                 set_namearray(&cfg->atime_now_list,*parm);
44         }
45
46         parm = lp_parm_string_list(snum, PARM_ONEFS_TYPE, PARM_CTIME_NOW,
47                                    NULL);
48
49         if (parm) {
50                 cfg->init_flags |= ONEFS_VFS_CONFIG_FAKETIMESTAMPS;
51                 set_namearray(&cfg->ctime_now_list,*parm);
52         }
53
54         parm = lp_parm_string_list(snum, PARM_ONEFS_TYPE, PARM_MTIME_NOW,
55                                    NULL);
56
57         if (parm) {
58                 cfg->init_flags |= ONEFS_VFS_CONFIG_FAKETIMESTAMPS;
59                 set_namearray(&cfg->mtime_now_list,*parm);
60         }
61
62         parm = lp_parm_string_list(snum, PARM_ONEFS_TYPE, PARM_ATIME_STATIC,
63                                    NULL);
64
65         if (parm) {
66                 cfg->init_flags |= ONEFS_VFS_CONFIG_FAKETIMESTAMPS;
67                 set_namearray(&cfg->atime_static_list,*parm);
68         }
69
70         parm = lp_parm_string_list(snum, PARM_ONEFS_TYPE, PARM_MTIME_STATIC,
71                                    NULL);
72
73         if (parm) {
74                 cfg->init_flags |= ONEFS_VFS_CONFIG_FAKETIMESTAMPS;
75                 set_namearray(&cfg->mtime_static_list,*parm);
76         }
77
78         cfg->atime_slop = lp_parm_int(snum, PARM_ONEFS_TYPE, PARM_ATIME_SLOP,0);
79         cfg->ctime_slop = lp_parm_int(snum, PARM_ONEFS_TYPE, PARM_CTIME_SLOP,0);
80         cfg->mtime_slop = lp_parm_int(snum, PARM_ONEFS_TYPE, PARM_MTIME_SLOP,0);
81 }
82
83
84 static int onefs_load_config(struct vfs_handle_struct *handle)
85 {
86         int snum = SNUM(handle->conn);
87         int share_count = lp_numservices();
88
89         if (!pshare_config) {
90
91                 if (share_count <= ONEFS_DATA_FASTBUF)
92                         pshare_config = share_config;
93                 else {
94                         pshare_config =
95                             SMB_MALLOC_ARRAY(struct onefs_vfs_config,
96                                              share_count);
97                         if (!pshare_config) {
98                                 errno = ENOMEM;
99                                 return -1;
100                         }
101
102                         memset(pshare_config, 0,
103                             (sizeof(struct onefs_vfs_config) * share_count));
104                 }
105         }
106
107         if ((pshare_config[snum].init_flags &
108                 ONEFS_VFS_CONFIG_INITIALIZED) == 0) {
109                         pshare_config[snum].init_flags =
110                             ONEFS_VFS_CONFIG_INITIALIZED;
111                         onefs_load_faketimestamp_config(handle,
112                                                         &pshare_config[snum]);
113         }
114
115         return 0;
116 }
117
118 bool onefs_get_config(int snum, int config_type,
119                       struct onefs_vfs_config *cfg)
120 {
121         if (share_config[snum].init_flags & config_type)
122                 *cfg = share_config[snum];
123         else
124                 return false;
125
126         return true;
127 }
128
129 static int onefs_connect(struct vfs_handle_struct *handle, const char *service,
130                          const char *user)
131 {
132         int ret = onefs_load_config(handle);
133
134         if (ret)
135                 return ret;
136
137         return SMB_VFS_NEXT_CONNECT(handle, service, user);
138 }
139
140 static int onefs_mkdir(vfs_handle_struct *handle, const char *path,
141                        mode_t mode)
142 {
143         /* SMB_VFS_MKDIR should never be called in vfs_onefs */
144         SMB_ASSERT(false);
145         return SMB_VFS_NEXT_MKDIR(handle, path, mode);
146 }
147
148 static int onefs_open(vfs_handle_struct *handle, const char *fname,
149                       files_struct *fsp, int flags, mode_t mode)
150 {
151         /* SMB_VFS_OPEN should never be called in vfs_onefs */
152         SMB_ASSERT(false);
153         return SMB_VFS_NEXT_OPEN(handle, fname, fsp, flags, mode);
154 }
155
156 static ssize_t onefs_recvfile(vfs_handle_struct *handle, int fromfd,
157                               files_struct *tofsp, SMB_OFF_T offset,
158                               size_t count)
159 {
160         ssize_t result;
161
162         START_PROFILE_BYTES(syscall_recvfile, count);
163         result = onefs_sys_recvfile(fromfd, tofsp->fh->fd, offset, count);
164         END_PROFILE(syscall_recvfile);
165         return result;
166 }
167
168 static uint64_t onefs_get_alloc_size(struct vfs_handle_struct *handle,
169                                      files_struct *fsp,
170                                      const SMB_STRUCT_STAT *sbuf)
171 {
172         uint64_t result;
173
174         START_PROFILE(syscall_get_alloc_size);
175
176         if(S_ISDIR(sbuf->st_mode)) {
177                 result = 0;
178                 goto out;
179         }
180
181         /* Just use the file size since st_blocks is unreliable on OneFS. */
182         result = get_file_size_stat(sbuf);
183
184         if (fsp && fsp->initial_allocation_size)
185                 result = MAX(result,fsp->initial_allocation_size);
186
187         result = smb_roundup(handle->conn, result);
188
189  out:
190         END_PROFILE(syscall_get_alloc_size);
191         return result;
192 }
193
194 static int onefs_statvfs(vfs_handle_struct *handle, const char *path,
195                          vfs_statvfs_struct *statbuf)
196 {
197         struct statvfs statvfs_buf;
198         int result;
199
200         DEBUG(5, ("Calling SMB_STAT_VFS \n"));
201         result = statvfs(path, &statvfs_buf);
202         ZERO_STRUCTP(statbuf);
203
204         if (!result) {
205                 statbuf->OptimalTransferSize = statvfs_buf.f_iosize;
206                 statbuf->BlockSize = statvfs_buf.f_bsize;
207                 statbuf->TotalBlocks = statvfs_buf.f_blocks;
208                 statbuf->BlocksAvail = statvfs_buf.f_bfree;
209                 statbuf->UserBlocksAvail = statvfs_buf.f_bavail;
210                 statbuf->TotalFileNodes = statvfs_buf.f_files;
211                 statbuf->FreeFileNodes = statvfs_buf.f_ffree;
212                 statbuf->FsIdentifier =
213                     (((uint64_t)statvfs_buf.f_fsid.val[0]<<32) &
214                         0xffffffff00000000LL) |
215                     (uint64_t)statvfs_buf.f_fsid.val[1];
216         }
217         return result;
218 }
219
220 static int onefs_get_real_filename(vfs_handle_struct *handle, const char *path,
221                                    const char *name, TALLOC_CTX *mem_ctx,
222                                    char **found_name)
223 {
224         SMB_STRUCT_STAT sb;
225         struct stat_extra se;
226         int result;
227         char *full_name = NULL;
228
229         ZERO_STRUCT(se);
230         se.se_version = ESTAT_CURRENT_VERSION;
231         se.se_flags = ESTAT_CASE_INSENSITIVE | ESTAT_SYMLINK_NOFOLLOW;
232
233         if (*path != '\0') {
234                 if (!(full_name = talloc_asprintf(mem_ctx, "%s/%s", path, name))) {
235                         errno = ENOMEM;
236                         DEBUG(2, ("talloc_asprintf failed\n"));
237                         result = -1;
238                         goto done;
239                 }
240         }
241
242         if ((result = estat(full_name ? full_name : name, &sb, &se)) != 0) {
243                 DEBUG(2, ("error calling estat: %s\n", strerror(errno)));
244                 goto done;
245         }
246
247         *found_name = talloc_strdup(mem_ctx, se.se_realname);
248         if (*found_name == NULL) {
249                 errno = ENOMEM;
250                 result = -1;
251                 goto done;
252         }
253
254 done:
255         TALLOC_FREE(full_name);
256         return result;
257 }
258
259 static int onefs_ntimes(vfs_handle_struct *handle, const char *fname,
260                         struct smb_file_time *ft)
261 {
262         int flags = 0;
263         struct timespec times[3];
264
265         if (!null_timespec(ft->atime)) {
266                 flags |= VT_ATIME;
267                 times[0] = ft->atime;
268                 DEBUG(6,("**** onefs_ntimes: actime: %s.%d\n",
269                         time_to_asc(convert_timespec_to_time_t(ft->atime)),
270                         ft->atime.tv_nsec));
271         }
272
273         if (!null_timespec(ft->mtime)) {
274                 flags |= VT_MTIME;
275                 times[1] = ft->mtime;
276                 DEBUG(6,("**** onefs_ntimes: modtime: %s.%d\n",
277                         time_to_asc(convert_timespec_to_time_t(ft->mtime)),
278                         ft->mtime.tv_nsec));
279         }
280
281         if (!null_timespec(ft->create_time)) {
282                 flags |= VT_BTIME;
283                 times[2] = ft->create_time;
284                 DEBUG(6,("**** onefs_ntimes: createtime: %s.%d\n",
285                    time_to_asc(convert_timespec_to_time_t(ft->create_time)),
286                    ft->create_time.tv_nsec));
287         }
288
289         return onefs_vtimes_streams(handle, fname, flags, times);
290 }
291
292 static uint32_t onefs_fs_capabilities(struct vfs_handle_struct *handle)
293 {
294         return SMB_VFS_NEXT_FS_CAPABILITIES(handle) | FILE_NAMED_STREAMS;
295 }
296
297 static vfs_op_tuple onefs_ops[] = {
298         {SMB_VFS_OP(onefs_connect), SMB_VFS_OP_CONNECT,
299          SMB_VFS_LAYER_TRANSPARENT},
300         {SMB_VFS_OP(onefs_fs_capabilities), SMB_VFS_OP_FS_CAPABILITIES,
301          SMB_VFS_LAYER_TRANSPARENT},
302         {SMB_VFS_OP(onefs_opendir), SMB_VFS_OP_OPENDIR,
303          SMB_VFS_LAYER_TRANSPARENT},
304         {SMB_VFS_OP(onefs_readdir), SMB_VFS_OP_READDIR,
305          SMB_VFS_LAYER_OPAQUE},
306         {SMB_VFS_OP(onefs_seekdir), SMB_VFS_OP_SEEKDIR,
307          SMB_VFS_LAYER_OPAQUE},
308         {SMB_VFS_OP(onefs_telldir), SMB_VFS_OP_TELLDIR,
309          SMB_VFS_LAYER_OPAQUE},
310         {SMB_VFS_OP(onefs_rewinddir), SMB_VFS_OP_REWINDDIR,
311          SMB_VFS_LAYER_OPAQUE},
312         {SMB_VFS_OP(onefs_mkdir), SMB_VFS_OP_MKDIR,
313          SMB_VFS_LAYER_OPAQUE},
314         {SMB_VFS_OP(onefs_closedir), SMB_VFS_OP_CLOSEDIR,
315          SMB_VFS_LAYER_TRANSPARENT},
316         {SMB_VFS_OP(onefs_init_search_op), SMB_VFS_OP_INIT_SEARCH_OP,
317          SMB_VFS_LAYER_OPAQUE},
318         {SMB_VFS_OP(onefs_open), SMB_VFS_OP_OPEN,
319          SMB_VFS_LAYER_OPAQUE},
320         {SMB_VFS_OP(onefs_create_file), SMB_VFS_OP_CREATE_FILE,
321          SMB_VFS_LAYER_OPAQUE},
322         {SMB_VFS_OP(onefs_close), SMB_VFS_OP_CLOSE,
323          SMB_VFS_LAYER_TRANSPARENT},
324         {SMB_VFS_OP(onefs_recvfile), SMB_VFS_OP_RECVFILE,
325          SMB_VFS_LAYER_OPAQUE},
326         {SMB_VFS_OP(onefs_rename), SMB_VFS_OP_RENAME,
327          SMB_VFS_LAYER_TRANSPARENT},
328         {SMB_VFS_OP(onefs_stat), SMB_VFS_OP_STAT,
329          SMB_VFS_LAYER_TRANSPARENT},
330         {SMB_VFS_OP(onefs_fstat), SMB_VFS_OP_FSTAT,
331          SMB_VFS_LAYER_TRANSPARENT},
332         {SMB_VFS_OP(onefs_lstat), SMB_VFS_OP_LSTAT,
333          SMB_VFS_LAYER_TRANSPARENT},
334         {SMB_VFS_OP(onefs_get_alloc_size), SMB_VFS_OP_GET_ALLOC_SIZE,
335          SMB_VFS_LAYER_OPAQUE},
336         {SMB_VFS_OP(onefs_unlink), SMB_VFS_OP_UNLINK,
337          SMB_VFS_LAYER_TRANSPARENT},
338         {SMB_VFS_OP(onefs_ntimes), SMB_VFS_OP_NTIMES,
339          SMB_VFS_LAYER_OPAQUE},
340         {SMB_VFS_OP(onefs_chflags), SMB_VFS_OP_CHFLAGS,
341          SMB_VFS_LAYER_TRANSPARENT},
342         {SMB_VFS_OP(onefs_streaminfo), SMB_VFS_OP_STREAMINFO,
343          SMB_VFS_LAYER_OPAQUE},
344         {SMB_VFS_OP(onefs_brl_lock_windows), SMB_VFS_OP_BRL_LOCK_WINDOWS,
345          SMB_VFS_LAYER_OPAQUE},
346         {SMB_VFS_OP(onefs_brl_unlock_windows), SMB_VFS_OP_BRL_UNLOCK_WINDOWS,
347          SMB_VFS_LAYER_OPAQUE},
348         {SMB_VFS_OP(onefs_brl_cancel_windows), SMB_VFS_OP_BRL_CANCEL_WINDOWS,
349          SMB_VFS_LAYER_OPAQUE},
350         {SMB_VFS_OP(onefs_fget_nt_acl), SMB_VFS_OP_FGET_NT_ACL,
351          SMB_VFS_LAYER_OPAQUE},
352         {SMB_VFS_OP(onefs_get_nt_acl), SMB_VFS_OP_GET_NT_ACL,
353          SMB_VFS_LAYER_OPAQUE},
354         {SMB_VFS_OP(onefs_fset_nt_acl), SMB_VFS_OP_FSET_NT_ACL,
355          SMB_VFS_LAYER_OPAQUE},
356         {SMB_VFS_OP(onefs_statvfs), SMB_VFS_OP_STATVFS,
357          SMB_VFS_LAYER_OPAQUE},
358         {SMB_VFS_OP(onefs_get_real_filename), SMB_VFS_OP_GET_REAL_FILENAME,
359          SMB_VFS_LAYER_OPAQUE},
360         {SMB_VFS_OP(NULL), SMB_VFS_OP_NOOP, SMB_VFS_LAYER_NOOP}
361 };
362
363 NTSTATUS vfs_onefs_init(void)
364 {
365         return smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "onefs",
366                                 onefs_ops);
367 }