/[rdesktop]/sourceforge.net/trunk/rdesktop/disk.c
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /sourceforge.net/trunk/rdesktop/disk.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 742 by stargo, Fri Aug 6 09:50:34 2004 UTC revision 790 by astrand, Thu Oct 28 07:46:39 2004 UTC
# Line 31  Line 31 
31  #include <utime.h>  #include <utime.h>
32  #include <time.h>               /* ctime */  #include <time.h>               /* ctime */
33    
34  #if defined(HAVE_DIRFD)  #if (defined(HAVE_DIRFD) || (HAVE_DECL_DIRFD == 1))
35  #define DIRFD(a) (dirfd(a))  #define DIRFD(a) (dirfd(a))
36  #else  #else
37  #define DIRFD(a) ((a)->DIR_FD_MEMBER_NAME)  #define DIRFD(a) ((a)->DIR_FD_MEMBER_NAME)
# Line 52  Line 52 
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>
# Line 125  convert_1970_to_filetime(uint32 high, ui Line 132  convert_1970_to_filetime(uint32 high, ui
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. */
# Line 142  disk_enum_devices(uint32 * id, char *opt Line 258  disk_enum_devices(uint32 * id, char *opt
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)++;
# Line 164  disk_enum_devices(uint32 * id, char *opt Line 278  disk_enum_devices(uint32 * id, char *opt
278  /* Opens or creates a file or directory */  /* Opens or creates a file or directory */
279  static 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];
# Line 177  disk_create(uint32 device_id, uint32 acc Line 291  disk_create(uint32 device_id, uint32 acc
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    
# Line 271  disk_create(uint32 device_id, uint32 acc Line 384  disk_create(uint32 device_id, uint32 acc
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)
# Line 320  disk_create(uint32 device_id, uint32 acc Line 433  disk_create(uint32 device_id, uint32 acc
433  }  }
434    
435  static NTSTATUS  static NTSTATUS
436  disk_close(HANDLE handle)  disk_close(NTHANDLE handle)
437  {  {
438          struct fileinfo *pfinfo;          struct fileinfo *pfinfo;
439    
# Line 340  disk_close(HANDLE handle) Line 453  disk_close(HANDLE handle)
453  }  }
454    
455  static 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    
# Line 377  disk_read(HANDLE handle, uint8 * data, u Line 490  disk_read(HANDLE handle, uint8 * data, u
490  }  }
491    
492  static 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    
# Line 404  disk_write(HANDLE handle, uint8 * data, Line 517  disk_write(HANDLE handle, uint8 * data,
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;
# Line 485  disk_query_information(HANDLE handle, ui Line 598  disk_query_information(HANDLE handle, ui
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    
# Line 653  disk_set_information(HANDLE handle, uint Line 766  disk_set_information(HANDLE handle, uint
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                          /* FIXME: Growing file with ftruncate doesn't                          if (ftruncate_growable(handle, length) != 0)
                            work with Linux FAT fs */  
                         if (ftruncate(handle, length) != 0)  
770                          {                          {
                                 perror("ftruncate");  
771                                  return STATUS_DISK_FULL;                                  return STATUS_DISK_FULL;
772                          }                          }
773    
# Line 739  FsVolumeInfo(char *fpath) Line 849  FsVolumeInfo(char *fpath)
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;
# Line 804  disk_query_volume_information(HANDLE han Line 914  disk_query_volume_information(HANDLE han
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];
# Line 914  disk_query_directory(HANDLE handle, uint Line 1024  disk_query_directory(HANDLE handle, uint
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    
Legend:
Removed from v.742  
changed lines
  Added in v.790
  ViewVC Help
Powered by ViewVC 1.1.26