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 */ |
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> |
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 |
|
|
195 |
/* Enumeration of devices from rdesktop.c */ |
/* Enumeration of devices from rdesktop.c */ |
196 |
/* returns numer of units found and initialized. */ |
/* returns numer of units found and initialized. */ |
208 |
while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES) |
while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES) |
209 |
{ |
{ |
210 |
pos2 = next_arg(optarg, '='); |
pos2 = next_arg(optarg, '='); |
|
strcpy(g_rdpdr_device[*id].name, optarg); |
|
211 |
|
|
212 |
toupper_str(g_rdpdr_device[*id].name); |
strncpy(g_rdpdr_device[*id].name, optarg, sizeof(g_rdpdr_device[*id].name)); |
213 |
|
if (strlen(optarg) > 8) |
214 |
/* add trailing colon to name. */ |
fprintf(stderr, "share name %s truncated to %s\n", optarg, |
215 |
strcat(g_rdpdr_device[*id].name, ":"); |
g_rdpdr_device[*id].name); |
216 |
|
|
217 |
g_rdpdr_device[*id].local_path = xmalloc(strlen(pos2) + 1); |
g_rdpdr_device[*id].local_path = xmalloc(strlen(pos2) + 1); |
218 |
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); |
|
219 |
g_rdpdr_device[*id].device_type = DEVICE_TYPE_DISK; |
g_rdpdr_device[*id].device_type = DEVICE_TYPE_DISK; |
220 |
count++; |
count++; |
221 |
(*id)++; |
(*id)++; |
228 |
/* Opens or creates a file or directory */ |
/* Opens or creates a file or directory */ |
229 |
static NTSTATUS |
static NTSTATUS |
230 |
disk_create(uint32 device_id, uint32 accessmask, uint32 sharemode, uint32 create_disposition, |
disk_create(uint32 device_id, uint32 accessmask, uint32 sharemode, uint32 create_disposition, |
231 |
uint32 flags_and_attributes, char *filename, HANDLE * phandle) |
uint32 flags_and_attributes, char *filename, NTHANDLE * phandle) |
232 |
{ |
{ |
233 |
HANDLE handle; |
NTHANDLE handle; |
234 |
DIR *dirp; |
DIR *dirp; |
235 |
int flags, mode; |
int flags, mode; |
236 |
char path[256]; |
char path[256]; |
384 |
} |
} |
385 |
|
|
386 |
static NTSTATUS |
static NTSTATUS |
387 |
disk_close(HANDLE handle) |
disk_close(NTHANDLE handle) |
388 |
{ |
{ |
389 |
struct fileinfo *pfinfo; |
struct fileinfo *pfinfo; |
390 |
|
|
404 |
} |
} |
405 |
|
|
406 |
static NTSTATUS |
static NTSTATUS |
407 |
disk_read(HANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
disk_read(NTHANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
408 |
{ |
{ |
409 |
int n; |
int n; |
410 |
|
|
441 |
} |
} |
442 |
|
|
443 |
static NTSTATUS |
static NTSTATUS |
444 |
disk_write(HANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
disk_write(NTHANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result) |
445 |
{ |
{ |
446 |
int n; |
int n; |
447 |
|
|
468 |
} |
} |
469 |
|
|
470 |
NTSTATUS |
NTSTATUS |
471 |
disk_query_information(HANDLE handle, uint32 info_class, STREAM out) |
disk_query_information(NTHANDLE handle, uint32 info_class, STREAM out) |
472 |
{ |
{ |
473 |
uint32 file_attributes, ft_high, ft_low; |
uint32 file_attributes, ft_high, ft_low; |
474 |
struct stat filestat; |
struct stat filestat; |
549 |
} |
} |
550 |
|
|
551 |
NTSTATUS |
NTSTATUS |
552 |
disk_set_information(HANDLE handle, uint32 info_class, STREAM in, STREAM out) |
disk_set_information(NTHANDLE handle, uint32 info_class, STREAM in, STREAM out) |
553 |
{ |
{ |
554 |
uint32 device_id, length, file_attributes, ft_high, ft_low; |
uint32 length, file_attributes, ft_high, ft_low; |
555 |
char newname[256], fullpath[256]; |
char newname[256], fullpath[256]; |
556 |
struct fileinfo *pfinfo; |
struct fileinfo *pfinfo; |
557 |
|
|
717 |
if (stat_fs.f_bsize * stat_fs.f_bfree < length) |
if (stat_fs.f_bsize * stat_fs.f_bfree < length) |
718 |
return STATUS_DISK_FULL; |
return STATUS_DISK_FULL; |
719 |
|
|
720 |
/* FIXME: Growing file with ftruncate doesn't |
if (ftruncate_growable(handle, length) != 0) |
|
work with Linux FAT fs */ |
|
|
if (ftruncate(handle, length) != 0) |
|
721 |
{ |
{ |
|
perror("ftruncate"); |
|
722 |
return STATUS_DISK_FULL; |
return STATUS_DISK_FULL; |
723 |
} |
} |
724 |
|
|
800 |
|
|
801 |
|
|
802 |
NTSTATUS |
NTSTATUS |
803 |
disk_query_volume_information(HANDLE handle, uint32 info_class, STREAM out) |
disk_query_volume_information(NTHANDLE handle, uint32 info_class, STREAM out) |
804 |
{ |
{ |
805 |
struct STATFS_T stat_fs; |
struct STATFS_T stat_fs; |
806 |
struct fileinfo *pfinfo; |
struct fileinfo *pfinfo; |
865 |
} |
} |
866 |
|
|
867 |
NTSTATUS |
NTSTATUS |
868 |
disk_query_directory(HANDLE handle, uint32 info_class, char *pattern, STREAM out) |
disk_query_directory(NTHANDLE handle, uint32 info_class, char *pattern, STREAM out) |
869 |
{ |
{ |
870 |
uint32 file_attributes, ft_low, ft_high; |
uint32 file_attributes, ft_low, ft_high; |
871 |
char *dirname, fullpath[256]; |
char *dirname, fullpath[256]; |
975 |
|
|
976 |
|
|
977 |
static NTSTATUS |
static NTSTATUS |
978 |
disk_device_control(HANDLE handle, uint32 request, STREAM in, STREAM out) |
disk_device_control(NTHANDLE handle, uint32 request, STREAM in, STREAM out) |
979 |
{ |
{ |
|
uint32 result; |
|
|
|
|
980 |
if (((request >> 16) != 20) || ((request >> 16) != 9)) |
if (((request >> 16) != 20) || ((request >> 16) != 9)) |
981 |
return STATUS_INVALID_PARAMETER; |
return STATUS_INVALID_PARAMETER; |
982 |
|
|