20 |
|
|
21 |
#include "disk.h" |
#include "disk.h" |
22 |
|
|
|
#if (defined(sun) && (defined(__svr4__) || defined(__SVR4))) |
|
|
#define SOLARIS |
|
|
#endif |
|
|
|
|
|
#if (defined(SOLARIS) || defined(__hpux)) |
|
|
#define DIRFD(a) ((a)->dd_fd) |
|
|
#else |
|
|
#define DIRFD(a) (dirfd(a)) |
|
|
#endif |
|
|
|
|
23 |
#include <sys/types.h> |
#include <sys/types.h> |
24 |
#include <sys/stat.h> |
#include <sys/stat.h> |
25 |
#include <unistd.h> |
#include <unistd.h> |
31 |
#include <utime.h> |
#include <utime.h> |
32 |
#include <time.h> /* ctime */ |
#include <time.h> /* ctime */ |
33 |
|
|
34 |
|
#if (defined(HAVE_DIRFD) || (HAVE_DECL_DIRFD == 1)) |
35 |
|
#define DIRFD(a) (dirfd(a)) |
36 |
|
#else |
37 |
|
#define DIRFD(a) ((a)->DIR_FD_MEMBER_NAME) |
38 |
|
#endif |
39 |
|
|
40 |
|
/* TODO: let autoconf figure out everything below... */ |
41 |
|
#if (defined(sun) && (defined(__svr4__) || defined(__SVR4))) |
42 |
|
#define SOLARIS |
43 |
|
#endif |
44 |
|
|
45 |
#if (defined(SOLARIS) || defined (__hpux) || defined(__BEOS__)) |
#if (defined(SOLARIS) || defined (__hpux) || defined(__BEOS__)) |
46 |
#include <sys/statvfs.h> /* solaris statvfs */ |
#include <sys/statvfs.h> /* solaris statvfs */ |
47 |
#include <sys/mntent.h> |
/* TODO: Fix mntent-handling for solaris/hpux |
48 |
/* TODO: Fix mntent-handling for solaris */ |
* #include <sys/mntent.h> */ |
49 |
#undef HAVE_MNTENT_H |
#undef HAVE_MNTENT_H |
50 |
#define MNTENT_PATH "/etc/mnttab" |
#define MNTENT_PATH "/etc/mnttab" |
51 |
#define STATFS_FN(path, buf) (statvfs(path,buf)) |
#define STATFS_FN(path, buf) (statvfs(path,buf)) |
52 |
#define STATFS_T statvfs |
#define STATFS_T statvfs |
53 |
#define F_NAMELEN(buf) ((buf).f_namemax) |
#define F_NAMELEN(buf) ((buf).f_namemax) |
54 |
|
|
55 |
#elif (defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__)) |
#elif (defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__) || defined(__APPLE__)) |
56 |
#include <sys/param.h> |
#include <sys/param.h> |
57 |
#include <sys/mount.h> |
#include <sys/mount.h> |
58 |
#define STATFS_FN(path, buf) (statfs(path,buf)) |
#define STATFS_FN(path, buf) (statfs(path,buf)) |
59 |
#define STATFS_T statfs |
#define STATFS_T statfs |
60 |
#define F_NAMELEN(buf) (NAME_MAX) |
#define F_NAMELEN(buf) (NAME_MAX) |
61 |
|
|
62 |
|
#elif (defined(__SGI_IRIX__)) |
63 |
|
#include <sys/types.h> |
64 |
|
#include <sys/statvfs.h> |
65 |
|
#define STATFS_FN(path, buf) (statvfs(path,buf)) |
66 |
|
#define STATFS_T statvfs |
67 |
|
#define F_NAMELEN(buf) ((buf).f_namemax) |
68 |
|
|
69 |
#else |
#else |
70 |
#include <sys/vfs.h> /* linux statfs */ |
#include <sys/vfs.h> /* linux statfs */ |
71 |
#include <mntent.h> |
#include <mntent.h> |
91 |
} FsInfoType; |
} FsInfoType; |
92 |
|
|
93 |
|
|
94 |
time_t |
static time_t |
95 |
get_create_time(struct stat *st) |
get_create_time(struct stat *st) |
96 |
{ |
{ |
97 |
time_t ret, ret1; |
time_t ret, ret1; |
106 |
} |
} |
107 |
|
|
108 |
/* Convert seconds since 1970 to a filetime */ |
/* Convert seconds since 1970 to a filetime */ |
109 |
void |
static void |
110 |
seconds_since_1970_to_filetime(time_t seconds, uint32 * high, uint32 * low) |
seconds_since_1970_to_filetime(time_t seconds, uint32 * high, uint32 * low) |
111 |
{ |
{ |
112 |
unsigned long long ticks; |
unsigned long long ticks; |
117 |
} |
} |
118 |
|
|
119 |
/* Convert seconds since 1970 back to filetime */ |
/* Convert seconds since 1970 back to filetime */ |
120 |
time_t |
static time_t |
121 |
convert_1970_to_filetime(uint32 high, uint32 low) |
convert_1970_to_filetime(uint32 high, uint32 low) |
122 |
{ |
{ |
123 |
unsigned long long ticks; |
unsigned long long ticks; |
132 |
|
|
133 |
} |
} |
134 |
|
|
135 |
|
/* A wrapper for ftruncate which supports growing files, even if the |
136 |
|
native ftruncate doesn't. This is needed on Linux FAT filesystems, |
137 |
|
for example. */ |
138 |
|
static int |
139 |
|
ftruncate_growable(int fd, off_t length) |
140 |
|
{ |
141 |
|
int ret; |
142 |
|
off_t pos; |
143 |
|
static const char zero; |
144 |
|
|
145 |
|
/* Try the simple method first */ |
146 |
|
if ((ret = ftruncate(fd, length)) != -1) |
147 |
|
{ |
148 |
|
return ret; |
149 |
|
} |
150 |
|
|
151 |
|
/* |
152 |
|
* Some kind of error. Perhaps we were trying to grow. Retry |
153 |
|
* in a safe way. |
154 |
|
*/ |
155 |
|
|
156 |
|
/* Get current position */ |
157 |
|
if ((pos = lseek(fd, 0, SEEK_CUR)) == -1) |
158 |
|
{ |
159 |
|
perror("lseek"); |
160 |
|
return -1; |
161 |
|
} |
162 |
|
|
163 |
|
/* Seek to new size */ |
164 |
|
if (lseek(fd, length, SEEK_SET) == -1) |
165 |
|
{ |
166 |
|
perror("lseek"); |
167 |
|
return -1; |
168 |
|
} |
169 |
|
|
170 |
|
/* Write a zero */ |
171 |
|
if (write(fd, &zero, 1) == -1) |
172 |
|
{ |
173 |
|
perror("write"); |
174 |
|
return -1; |
175 |
|
} |
176 |
|
|
177 |
|
/* Truncate. This shouldn't fail. */ |
178 |
|
if (ftruncate(fd, length) == -1) |
179 |
|
{ |
180 |
|
perror("ftruncate"); |
181 |
|
return -1; |
182 |
|
} |
183 |
|
|
184 |
|
/* Restore position */ |
185 |
|
if (lseek(fd, pos, SEEK_SET) == -1) |
186 |
|
{ |
187 |
|
perror("lseek"); |
188 |
|
return -1; |
189 |
|
} |
190 |
|
|
191 |
|
return 0; |
192 |
|
} |
193 |
|
|
194 |
|
/* Just like open(2), but if a open with O_EXCL fails, retry with |
195 |
|
GUARDED semantics. This might be necessary because some filesystems |
196 |
|
(such as NFS filesystems mounted from a unfsd server) doesn't |
197 |
|
support O_EXCL. GUARDED semantics are subject to race conditions, |
198 |
|
but we can live with that. |
199 |
|
*/ |
200 |
|
static int |
201 |
|
open_weak_exclusive(const char *pathname, int flags, mode_t mode) |
202 |
|
{ |
203 |
|
int ret; |
204 |
|
struct stat statbuf; |
205 |
|
|
206 |
|
ret = open(pathname, flags, mode); |
207 |
|
if (ret != -1 || !(flags & O_EXCL)) |
208 |
|
{ |
209 |
|
/* Success, or not using O_EXCL */ |
210 |
|
return ret; |
211 |
|
} |
212 |
|
|
213 |
|
/* An error occured, and we are using O_EXCL. In case the FS |
214 |
|
doesn't support O_EXCL, some kind of error will be |
215 |
|
returned. Unfortunately, we don't know which one. Linux |
216 |
|
2.6.8 seems to return 524, but I cannot find a documented |
217 |
|
#define for this case. So, we'll return only on errors that |
218 |
|
we know aren't related to O_EXCL. */ |
219 |
|
switch (errno) |
220 |
|
{ |
221 |
|
case EACCES: |
222 |
|
case EEXIST: |
223 |
|
case EINTR: |
224 |
|
case EISDIR: |
225 |
|
case ELOOP: |
226 |
|
case ENAMETOOLONG: |
227 |
|
case ENOENT: |
228 |
|
case ENOTDIR: |
229 |
|
return ret; |
230 |
|
} |
231 |
|
|
232 |
|
/* Retry with GUARDED semantics */ |
233 |
|
if (stat(pathname, &statbuf) != -1) |
234 |
|
{ |
235 |
|
/* File exists */ |
236 |
|
errno = EEXIST; |
237 |
|
return -1; |
238 |
|
} |
239 |
|
else |
240 |
|
{ |
241 |
|
return open(pathname, flags & ~O_EXCL, mode); |
242 |
|
} |
243 |
|
} |
244 |
|
|
245 |
/* Enumeration of devices from rdesktop.c */ |
/* Enumeration of devices from rdesktop.c */ |
246 |
/* returns numer of units found and initialized. */ |
/* returns numer of units found and initialized. */ |
258 |
while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES) |
while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES) |
259 |
{ |
{ |
260 |
pos2 = next_arg(optarg, '='); |
pos2 = next_arg(optarg, '='); |
|
strcpy(g_rdpdr_device[*id].name, optarg); |
|
|
|
|
|
toupper_str(g_rdpdr_device[*id].name); |
|
261 |
|
|
262 |
/* add trailing colon to name. */ |
strncpy(g_rdpdr_device[*id].name, optarg, sizeof(g_rdpdr_device[*id].name)); |
263 |
strcat(g_rdpdr_device[*id].name, ":"); |
if (strlen(optarg) > 8) |
264 |
|
fprintf(stderr, "share name %s truncated to %s\n", optarg, |
265 |
|
g_rdpdr_device[*id].name); |
266 |
|
|
267 |
g_rdpdr_device[*id].local_path = xmalloc(strlen(pos2) + 1); |
g_rdpdr_device[*id].local_path = xmalloc(strlen(pos2) + 1); |
268 |
strcpy(g_rdpdr_device[*id].local_path, pos2); |
strcpy(g_rdpdr_device[*id].local_path, pos2); |
|
printf("DISK %s to %s\n", g_rdpdr_device[*id].name, g_rdpdr_device[*id].local_path); |
|
269 |
g_rdpdr_device[*id].device_type = DEVICE_TYPE_DISK; |
g_rdpdr_device[*id].device_type = DEVICE_TYPE_DISK; |
270 |
count++; |
count++; |
271 |
(*id)++; |
(*id)++; |
276 |
} |
} |
277 |
|
|
278 |
/* Opens or creates a file or directory */ |
/* Opens or creates a file or directory */ |
279 |
NTSTATUS |
static NTSTATUS |
280 |
disk_create(uint32 device_id, uint32 accessmask, uint32 sharemode, uint32 create_disposition, |
disk_create(uint32 device_id, uint32 accessmask, uint32 sharemode, uint32 create_disposition, |
281 |
uint32 flags_and_attributes, char *filename, HANDLE * phandle) |
uint32 flags_and_attributes, char *filename, NTHANDLE * phandle) |
282 |
{ |
{ |
283 |
HANDLE handle; |
NTHANDLE handle; |
284 |
DIR *dirp; |
DIR *dirp; |
285 |
int flags, mode; |
int flags, mode; |
286 |
char path[256]; |
char path[256]; |
291 |
flags = 0; |
flags = 0; |
292 |
mode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH; |
mode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH; |
293 |
|
|
294 |
|
if (*filename && filename[strlen(filename) - 1] == '/') |
|
if (filename[strlen(filename) - 1] == '/') |
|
295 |
filename[strlen(filename) - 1] = 0; |
filename[strlen(filename) - 1] = 0; |
296 |
sprintf(path, "%s%s", g_rdpdr_device[device_id].local_path, filename); |
sprintf(path, "%s%s", g_rdpdr_device[device_id].local_path, filename); |
297 |
|
|
384 |
flags |= O_RDONLY; |
flags |= O_RDONLY; |
385 |
} |
} |
386 |
|
|
387 |
handle = open(path, flags, mode); |
handle = open_weak_exclusive(path, flags, mode); |
388 |
if (handle == -1) |
if (handle == -1) |
389 |
{ |
{ |
390 |
switch (errno) |
switch (errno) |
432 |
return STATUS_SUCCESS; |
return STATUS_SUCCESS; |
433 |
} |
} |
434 |
|
|
435 |
NTSTATUS |
static NTSTATUS |
436 |
disk_close(HANDLE handle) |
disk_close(NTHANDLE handle) |
437 |
{ |
{ |
438 |
struct fileinfo *pfinfo; |
struct fileinfo *pfinfo; |
439 |
|
|
452 |
return STATUS_SUCCESS; |
return STATUS_SUCCESS; |
453 |
} |
} |
454 |
|
|
455 |
NTSTATUS |
static NTSTATUS |
456 |
disk_read(HANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
disk_read(NTHANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
457 |
{ |
{ |
458 |
int n; |
int n; |
459 |
|
|
489 |
return STATUS_SUCCESS; |
return STATUS_SUCCESS; |
490 |
} |
} |
491 |
|
|
492 |
NTSTATUS |
static NTSTATUS |
493 |
disk_write(HANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
disk_write(NTHANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
494 |
{ |
{ |
495 |
int n; |
int n; |
496 |
|
|
517 |
} |
} |
518 |
|
|
519 |
NTSTATUS |
NTSTATUS |
520 |
disk_query_information(HANDLE handle, uint32 info_class, STREAM out) |
disk_query_information(NTHANDLE handle, uint32 info_class, STREAM out) |
521 |
{ |
{ |
522 |
uint32 file_attributes, ft_high, ft_low; |
uint32 file_attributes, ft_high, ft_low; |
523 |
struct stat filestat; |
struct stat filestat; |
598 |
} |
} |
599 |
|
|
600 |
NTSTATUS |
NTSTATUS |
601 |
disk_set_information(HANDLE handle, uint32 info_class, STREAM in, STREAM out) |
disk_set_information(NTHANDLE handle, uint32 info_class, STREAM in, STREAM out) |
602 |
{ |
{ |
603 |
uint32 device_id, length, file_attributes, ft_high, ft_low; |
uint32 length, file_attributes, ft_high, ft_low; |
604 |
char newname[256], fullpath[256]; |
char newname[256], fullpath[256]; |
605 |
struct fileinfo *pfinfo; |
struct fileinfo *pfinfo; |
606 |
|
|
721 |
break; |
break; |
722 |
|
|
723 |
case FileDispositionInformation: |
case FileDispositionInformation: |
724 |
|
/* As far as I understand it, the correct |
725 |
//unimpl("IRP Set File Information class: FileDispositionInformation\n"); |
thing to do here is to *schedule* a delete, |
726 |
|
so it will be deleted when the file is |
727 |
|
closed. Subsequent |
728 |
|
FileDispositionInformation requests with |
729 |
|
DeleteFile set to FALSE should unschedule |
730 |
|
the delete. See |
731 |
|
http://www.osronline.com/article.cfm?article=245. Currently, |
732 |
|
we are deleting the file immediately. I |
733 |
|
guess this is a FIXME. */ |
734 |
|
|
735 |
//in_uint32_le(in, delete_on_close); |
//in_uint32_le(in, delete_on_close); |
736 |
// disk_close(handle); |
|
737 |
|
/* Make sure we close the file before |
738 |
|
unlinking it. Not doing so would trigger |
739 |
|
silly-delete if using NFS, which might fail |
740 |
|
on FAT floppies, for example. */ |
741 |
|
disk_close(handle); |
742 |
|
|
743 |
if ((pfinfo->flags_and_attributes & FILE_DIRECTORY_FILE)) // remove a directory |
if ((pfinfo->flags_and_attributes & FILE_DIRECTORY_FILE)) // remove a directory |
744 |
{ |
{ |
745 |
if (rmdir(pfinfo->path) < 0) |
if (rmdir(pfinfo->path) < 0) |
751 |
break; |
break; |
752 |
|
|
753 |
case FileAllocationInformation: |
case FileAllocationInformation: |
754 |
|
/* Fall through to FileEndOfFileInformation, |
755 |
unimpl("IRP Set File Information class: FileAllocationInformation\n"); |
which uses ftrunc. This is like Samba with |
756 |
break; |
"strict allocation = false", and means that |
757 |
|
we won't detect out-of-quota errors, for |
758 |
|
example. */ |
759 |
|
|
760 |
case FileEndOfFileInformation: |
case FileEndOfFileInformation: |
761 |
in_uint8s(in, 28); /* unknown */ |
in_uint8s(in, 28); /* unknown */ |
766 |
if (stat_fs.f_bsize * stat_fs.f_bfree < length) |
if (stat_fs.f_bsize * stat_fs.f_bfree < length) |
767 |
return STATUS_DISK_FULL; |
return STATUS_DISK_FULL; |
768 |
|
|
769 |
//printf("FileEndOfFileInformation length = %d\n", length); |
if (ftruncate_growable(handle, length) != 0) |
770 |
// ???????????? |
{ |
771 |
//unimpl("IRP Set File Information class: FileEndOfFileInformation\n"); |
return STATUS_DISK_FULL; |
772 |
|
} |
773 |
|
|
774 |
break; |
break; |
775 |
default: |
default: |
776 |
|
|
780 |
return STATUS_SUCCESS; |
return STATUS_SUCCESS; |
781 |
} |
} |
782 |
|
|
783 |
FsInfoType * |
static FsInfoType * |
784 |
FsVolumeInfo(char *fpath) |
FsVolumeInfo(char *fpath) |
785 |
{ |
{ |
786 |
|
|
849 |
|
|
850 |
|
|
851 |
NTSTATUS |
NTSTATUS |
852 |
disk_query_volume_information(HANDLE handle, uint32 info_class, STREAM out) |
disk_query_volume_information(NTHANDLE handle, uint32 info_class, STREAM out) |
853 |
{ |
{ |
854 |
struct STATFS_T stat_fs; |
struct STATFS_T stat_fs; |
855 |
struct fileinfo *pfinfo; |
struct fileinfo *pfinfo; |
914 |
} |
} |
915 |
|
|
916 |
NTSTATUS |
NTSTATUS |
917 |
disk_query_directory(HANDLE handle, uint32 info_class, char *pattern, STREAM out) |
disk_query_directory(NTHANDLE handle, uint32 info_class, char *pattern, STREAM out) |
918 |
{ |
{ |
919 |
uint32 file_attributes, ft_low, ft_high; |
uint32 file_attributes, ft_low, ft_high; |
920 |
char *dirname, fullpath[256]; |
char *dirname, fullpath[256]; |
930 |
|
|
931 |
switch (info_class) |
switch (info_class) |
932 |
{ |
{ |
933 |
case 3: //FIXME: Why 3? |
case FileBothDirectoryInformation: |
934 |
|
|
935 |
// If a search pattern is received, remember this pattern, and restart search |
// If a search pattern is received, remember this pattern, and restart search |
936 |
if (pattern[0] != 0) |
if (pattern[0] != 0) |
950 |
// Get information for directory entry |
// Get information for directory entry |
951 |
sprintf(fullpath, "%s/%s", dirname, pdirent->d_name); |
sprintf(fullpath, "%s/%s", dirname, pdirent->d_name); |
952 |
|
|
|
/* JIF |
|
|
printf("Stat: %s\n", fullpath); */ |
|
953 |
if (stat(fullpath, &fstat)) |
if (stat(fullpath, &fstat)) |
954 |
{ |
{ |
955 |
perror("stat"); |
switch (errno) |
956 |
out_uint8(out, 0); |
{ |
957 |
return STATUS_ACCESS_DENIED; |
case ENOENT: |
958 |
|
case ELOOP: |
959 |
|
case EACCES: |
960 |
|
/* These are non-fatal errors. */ |
961 |
|
memset(&fstat, 0, sizeof(fstat)); |
962 |
|
break; |
963 |
|
default: |
964 |
|
/* Fatal error. By returning STATUS_NO_SUCH_FILE, |
965 |
|
the directory list operation will be aborted */ |
966 |
|
perror(fullpath); |
967 |
|
out_uint8(out, 0); |
968 |
|
return STATUS_NO_SUCH_FILE; |
969 |
|
} |
970 |
} |
} |
971 |
|
|
972 |
if (S_ISDIR(fstat.st_mode)) |
if (S_ISDIR(fstat.st_mode)) |
1010 |
break; |
break; |
1011 |
|
|
1012 |
default: |
default: |
1013 |
|
/* FIXME: Support FileDirectoryInformation, |
1014 |
|
FileFullDirectoryInformation, and |
1015 |
|
FileNamesInformation */ |
1016 |
|
|
1017 |
unimpl("IRP Query Directory sub: 0x%x\n", info_class); |
unimpl("IRP Query Directory sub: 0x%x\n", info_class); |
1018 |
return STATUS_INVALID_PARAMETER; |
return STATUS_INVALID_PARAMETER; |
1024 |
|
|
1025 |
|
|
1026 |
static NTSTATUS |
static NTSTATUS |
1027 |
disk_device_control(HANDLE handle, uint32 request, STREAM in, STREAM out) |
disk_device_control(NTHANDLE handle, uint32 request, STREAM in, STREAM out) |
1028 |
{ |
{ |
|
uint32 result; |
|
|
|
|
1029 |
if (((request >> 16) != 20) || ((request >> 16) != 9)) |
if (((request >> 16) != 20) || ((request >> 16) != 9)) |
1030 |
return STATUS_INVALID_PARAMETER; |
return STATUS_INVALID_PARAMETER; |
1031 |
|
|