/* -*- c-basic-offset: 8 -*-
rdesktop: A Remote Desktop Protocol client.
Disk Redirection
Copyright (C) Jeroen Meijer 2003
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include "disk.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h> /* open, close */
#include <dirent.h> /* opendir, closedir, readdir */
#include <fnmatch.h>
#include <errno.h> /* errno */
#include <stdio.h>
#include <utime.h>
#include <time.h> /* ctime */
#if (defined(HAVE_DIRFD) || (HAVE_DECL_DIRFD == 1))
#define DIRFD(a) (dirfd(a))
#else
#define DIRFD(a) ((a)->DIR_FD_MEMBER_NAME)
#endif
/* TODO: Fix mntent-handling for solaris
* #include <sys/mntent.h> */
#if (defined(HAVE_MNTENT_H) && defined(HAVE_SETMNTENT))
#include <mntent.h>
#define MNTENT_PATH "/etc/mtab"
#define USE_SETMNTENT
#endif
#ifdef HAVE_SYS_VFS_H
#include <sys/vfs.h>
#endif
#ifdef HAVE_SYS_STATVFS_H
#include <sys/statvfs.h>
#endif
#ifdef HAVE_SYS_STATFS_H
#include <sys/statfs.h>
#endif
#ifdef HAVE_SYS_PARAM_H
#include <sys/param.h>
#endif
#ifdef HAVE_SYS_MOUNT_H
#include <sys/mount.h>
#endif
#include "rdesktop.h"
#ifdef STAT_STATFS3_OSF1
#define STATFS_FN(path, buf) (statfs(path,buf,sizeof(buf)))
#define STATFS_T statfs
#define USE_STATFS
#endif
#ifdef STAT_STATVFS
#define STATFS_FN(path, buf) (statvfs(path,buf))
#define STATFS_T statvfs
#define USE_STATVFS
#endif
#ifdef STAT_STATVFS64
#define STATFS_FN(path, buf) (statvfs64(path,buf))
#define STATFS_T statvfs64
#define USE_STATVFS
#endif
#if (defined(STAT_STATFS2_FS_DATA) || defined(STAT_STATFS2_BSIZE) || defined(STAT_STATFS2_FSIZE))
#define STATFS_FN(path, buf) (statfs(path,buf))
#define STATFS_T statfs
#define USE_STATFS
#endif
#ifdef STAT_STATFS4
#define STATFS_FN(path, buf) (statfs(path,buf,sizeof(buf),0))
#define STATFS_T statfs
#define USE_STATFS
#endif
#if ((defined(USE_STATFS) && defined(HAVE_STRUCT_STATFS_F_NAMEMAX)) || (defined(USE_STATVFS) && defined(HAVE_STRUCT_STATVFS_F_NAMEMAX)))
#define F_NAMELEN(buf) ((buf).f_namemax)
#endif
#if ((defined(USE_STATFS) && defined(HAVE_STRUCT_STATFS_F_NAMELEN)) || (defined(USE_STATVFS) && defined(HAVE_STRUCT_STATVFS_F_NAMELEN)))
#define F_NAMELEN(buf) ((buf).f_namelen)
#endif
#ifndef F_NAMELEN
#define F_NAMELEN(buf) (255)
#endif
/* Dummy statfs fallback */
#ifndef STATFS_T
struct dummy_statfs_t
{
long f_bfree;
long f_bsize;
long f_blocks;
int f_namelen;
int f_namemax;
};
static int
dummy_statfs(struct dummy_statfs_t *buf)
{
buf->f_blocks = 262144;
buf->f_bfree = 131072;
buf->f_bsize = 512;
buf->f_namelen = 255;
buf->f_namemax = 255;
return 0;
}
#define STATFS_T dummy_statfs_t
#define STATFS_FN(path,buf) (dummy_statfs(buf))
#endif
extern RDPDR_DEVICE g_rdpdr_device[];
FILEINFO g_fileinfo[MAX_OPEN_FILES];
BOOL g_notify_stamp = False;
typedef struct
{
char name[PATH_MAX];
char label[PATH_MAX];
unsigned long serial;
char type[PATH_MAX];
} FsInfoType;
static NTSTATUS NotifyInfo(NTHANDLE handle, uint32 info_class, NOTIFY * p);
static time_t
get_create_time(struct stat *st)
{
time_t ret, ret1;
ret = MIN(st->st_ctime, st->st_mtime);
ret1 = MIN(ret, st->st_atime);
if (ret1 != (time_t) 0)
return ret1;
return ret;
}
/* Convert seconds since 1970 to a filetime */
static void
seconds_since_1970_to_filetime(time_t seconds, uint32 * high, uint32 * low)
{
unsigned long long ticks;
ticks = (seconds + 11644473600LL) * 10000000;
*low = (uint32) ticks;
*high = (uint32) (ticks >> 32);
}
/* Convert seconds since 1970 back to filetime */
static time_t
convert_1970_to_filetime(uint32 high, uint32 low)
{
unsigned long long ticks;
time_t val;
ticks = low + (((unsigned long long) high) << 32);
ticks /= 10000000;
ticks -= 11644473600LL;
val = (time_t) ticks;
return (val);
}
/* A wrapper for ftruncate which supports growing files, even if the
native ftruncate doesn't. This is needed on Linux FAT filesystems,
for example. */
static int
ftruncate_growable(int fd, off_t length)
{
int ret;
off_t pos;
static const char zero = 0;
/* Try the simple method first */
if ((ret = ftruncate(fd, length)) != -1)
{
return ret;
}
/*
* Some kind of error. Perhaps we were trying to grow. Retry
* in a safe way.
*/
/* Get current position */
if ((pos = lseek(fd, 0, SEEK_CUR)) == -1)
{
perror("lseek");
return -1;
}
/* Seek to new size */
if (lseek(fd, length, SEEK_SET) == -1)
{
perror("lseek");
return -1;
}
/* Write a zero */
if (write(fd, &zero, 1) == -1)
{
perror("write");
return -1;
}
/* Truncate. This shouldn't fail. */
if (ftruncate(fd, length) == -1)
{
perror("ftruncate");
return -1;
}
/* Restore position */
if (lseek(fd, pos, SEEK_SET) == -1)
{
perror("lseek");
return -1;
}
return 0;
}
/* Just like open(2), but if a open with O_EXCL fails, retry with
GUARDED semantics. This might be necessary because some filesystems
(such as NFS filesystems mounted from a unfsd server) doesn't
support O_EXCL. GUARDED semantics are subject to race conditions,
but we can live with that.
*/
static int
open_weak_exclusive(const char *pathname, int flags, mode_t mode)
{
int ret;
struct stat statbuf;
ret = open(pathname, flags, mode);
if (ret != -1 || !(flags & O_EXCL))
{
/* Success, or not using O_EXCL */
return ret;
}
/* An error occured, and we are using O_EXCL. In case the FS
doesn't support O_EXCL, some kind of error will be
returned. Unfortunately, we don't know which one. Linux
2.6.8 seems to return 524, but I cannot find a documented
#define for this case. So, we'll return only on errors that
we know aren't related to O_EXCL. */
switch (errno)
{
case EACCES:
case EEXIST:
case EINTR:
case EISDIR:
case ELOOP:
case ENAMETOOLONG:
case ENOENT:
case ENOTDIR:
return ret;
}
/* Retry with GUARDED semantics */
if (stat(pathname, &statbuf) != -1)
{
/* File exists */
errno = EEXIST;
return -1;
}
else
{
return open(pathname, flags & ~O_EXCL, mode);
}
}
/* Enumeration of devices from rdesktop.c */
/* returns numer of units found and initialized. */
/* optarg looks like ':h=/mnt/floppy,b=/mnt/usbdevice1' */
/* when it arrives to this function. */
int
disk_enum_devices(uint32 * id, char *optarg)
{
char *pos = optarg;
char *pos2;
int count = 0;
/* skip the first colon */
optarg++;
while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES)
{
pos2 = next_arg(optarg, '=');
strncpy(g_rdpdr_device[*id].name, optarg, sizeof(g_rdpdr_device[*id].name) - 1);
if (strlen(optarg) > (sizeof(g_rdpdr_device[*id].name) - 1))
fprintf(stderr, "share name %s truncated to %s\n", optarg,
g_rdpdr_device[*id].name);
g_rdpdr_device[*id].local_path = (char *) xmalloc(strlen(pos2) + 1);
strcpy(g_rdpdr_device[*id].local_path, pos2);
g_rdpdr_device[*id].device_type = DEVICE_TYPE_DISK;
count++;
(*id)++;
optarg = pos;
}
return count;
}
/* Opens or creates a file or directory */
static NTSTATUS
disk_create(uint32 device_id, uint32 accessmask, uint32 sharemode, uint32 create_disposition,
uint32 flags_and_attributes, char *filename, NTHANDLE * phandle)
{
NTHANDLE handle;
DIR *dirp;
int flags, mode;
char path[PATH_MAX];
struct stat filestat;
handle = 0;
dirp = NULL;
flags = 0;
mode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH;
if (*filename && filename[strlen(filename) - 1] == '/')
filename[strlen(filename) - 1] = 0;
sprintf(path, "%s%s", g_rdpdr_device[device_id].local_path, filename);
switch (create_disposition)
{
case CREATE_ALWAYS:
/* Delete existing file/link. */
unlink(path);
flags |= O_CREAT;
break;
case CREATE_NEW:
/* If the file already exists, then fail. */
flags |= O_CREAT | O_EXCL;
break;
case OPEN_ALWAYS:
/* Create if not already exists. */
flags |= O_CREAT;
break;
case OPEN_EXISTING:
/* Default behaviour */
break;
case TRUNCATE_EXISTING:
/* If the file does not exist, then fail. */
flags |= O_TRUNC;
break;
}
/*printf("Open: \"%s\" flags: %X, accessmask: %X sharemode: %X create disp: %X\n", path, flags_and_attributes, accessmask, sharemode, create_disposition); */
/* Get information about file and set that flag ourselfs */
if ((stat(path, &filestat) == 0) && (S_ISDIR(filestat.st_mode)))
{
if (flags_and_attributes & FILE_NON_DIRECTORY_FILE)
return STATUS_FILE_IS_A_DIRECTORY;
else
flags_and_attributes |= FILE_DIRECTORY_FILE;
}
if (flags_and_attributes & FILE_DIRECTORY_FILE)
{
if (flags & O_CREAT)
{
mkdir(path, mode);
}
dirp = opendir(path);
if (!dirp)
{
switch (errno)
{
case EACCES:
return STATUS_ACCESS_DENIED;
case ENOENT:
return STATUS_NO_SUCH_FILE;
default:
perror("opendir");
return STATUS_NO_SUCH_FILE;
}
}
handle = DIRFD(dirp);
}
else
{
if (accessmask & GENERIC_ALL
|| (accessmask & GENERIC_READ && accessmask & GENERIC_WRITE))
{
flags |= O_RDWR;
}
else if ((accessmask & GENERIC_WRITE) && !(accessmask & GENERIC_READ))
{
flags |= O_WRONLY;
}
else
{
flags |= O_RDONLY;
}
handle = open_weak_exclusive(path, flags, mode);
if (handle == -1)
{
switch (errno)
{
case EISDIR:
return STATUS_FILE_IS_A_DIRECTORY;
case EACCES:
return STATUS_ACCESS_DENIED;
case ENOENT:
return STATUS_NO_SUCH_FILE;
case EEXIST:
return STATUS_OBJECT_NAME_COLLISION;
default:
perror("open");
return STATUS_NO_SUCH_FILE;
}
}
/* all read and writes of files should be non blocking */
if (fcntl(handle, F_SETFL, O_NONBLOCK) == -1)
perror("fcntl");
}
if (handle >= MAX_OPEN_FILES)
{
error("Maximum number of open files (%s) reached. Increase MAX_OPEN_FILES!\n",
handle);
exit(1);
}
if (dirp)
g_fileinfo[handle].pdir = dirp;
else
g_fileinfo[handle].pdir = NULL;
g_fileinfo[handle].device_id = device_id;
g_fileinfo[handle].flags_and_attributes = flags_and_attributes;
g_fileinfo[handle].accessmask = accessmask;
strncpy(g_fileinfo[handle].path, path, PATH_MAX - 1);
g_fileinfo[handle].delete_on_close = False;
g_notify_stamp = True;
*phandle = handle;
return STATUS_SUCCESS;
}
static NTSTATUS
disk_close(NTHANDLE handle)
{
struct fileinfo *pfinfo;
pfinfo = &(g_fileinfo[handle]);
g_notify_stamp = True;
rdpdr_abort_io(handle, 0, STATUS_CANCELLED);
if (pfinfo->pdir)
{
if (closedir(pfinfo->pdir) < 0)
{
perror("closedir");
return STATUS_INVALID_HANDLE;
}
if (pfinfo->delete_on_close)
if (rmdir(pfinfo->path) < 0)
{
perror(pfinfo->path);
return STATUS_ACCESS_DENIED;
}
pfinfo->delete_on_close = False;
}
else
{
if (close(handle) < 0)
{
perror("close");
return STATUS_INVALID_HANDLE;
}
if (pfinfo->delete_on_close)
if (unlink(pfinfo->path) < 0)
{
perror(pfinfo->path);
return STATUS_ACCESS_DENIED;
}
pfinfo->delete_on_close = False;
}
return STATUS_SUCCESS;
}
static NTSTATUS
disk_read(NTHANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result)
{
int n;
#if 0
/* browsing dir ???? */
/* each request is 24 bytes */
if (g_fileinfo[handle].flags_and_attributes & FILE_DIRECTORY_FILE)
{
*result = 0;
return STATUS_SUCCESS;
}
#endif
lseek(handle, offset, SEEK_SET);
n = read(handle, data, length);
if (n < 0)
{
*result = 0;
switch (errno)
{
case EISDIR:
/* Implement 24 Byte directory read ??
with STATUS_NOT_IMPLEMENTED server doesn't read again */
/* return STATUS_FILE_IS_A_DIRECTORY; */
return STATUS_NOT_IMPLEMENTED;
default:
perror("read");
return STATUS_INVALID_PARAMETER;
}
}
*result = n;
return STATUS_SUCCESS;
}
static NTSTATUS
disk_write(NTHANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result)
{
int n;
lseek(handle, offset, SEEK_SET);
n = write(handle, data, length);
if (n < 0)
{
perror("write");
*result = 0;
switch (errno)
{
case ENOSPC:
return STATUS_DISK_FULL;
default:
return STATUS_ACCESS_DENIED;
}
}
*result = n;
return STATUS_SUCCESS;
}
NTSTATUS
disk_query_information(NTHANDLE handle, uint32 info_class, STREAM out)
{
uint32 file_attributes, ft_high, ft_low;
struct stat filestat;
char *path, *filename;
path = g_fileinfo[handle].path;
/* Get information about file */
if (fstat(handle, &filestat) != 0)
{
perror("stat");
out_uint8(out, 0);
return STATUS_ACCESS_DENIED;
}
/* Set file attributes */
file_attributes = 0;
if (S_ISDIR(filestat.st_mode))
file_attributes |= FILE_ATTRIBUTE_DIRECTORY;
filename = 1 + strrchr(path, '/');
if (filename && filename[0] == '.')
file_attributes |= FILE_ATTRIBUTE_HIDDEN;
if (!file_attributes)
file_attributes |= FILE_ATTRIBUTE_NORMAL;
if (!(filestat.st_mode & S_IWUSR))
file_attributes |= FILE_ATTRIBUTE_READONLY;
/* Return requested data */
switch (info_class)
{
case FileBasicInformation:
seconds_since_1970_to_filetime(get_create_time(&filestat), &ft_high,
&ft_low);
out_uint32_le(out, ft_low); /* create_access_time */
out_uint32_le(out, ft_high);
seconds_since_1970_to_filetime(filestat.st_atime, &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* last_access_time */
out_uint32_le(out, ft_high);
seconds_since_1970_to_filetime(filestat.st_mtime, &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* last_write_time */
out_uint32_le(out, ft_high);
seconds_since_1970_to_filetime(filestat.st_ctime, &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* last_change_time */
out_uint32_le(out, ft_high);
out_uint32_le(out, file_attributes);
break;
case FileStandardInformation:
out_uint32_le(out, filestat.st_size); /* Allocation size */
out_uint32_le(out, 0);
out_uint32_le(out, filestat.st_size); /* End of file */
out_uint32_le(out, 0);
out_uint32_le(out, filestat.st_nlink); /* Number of links */
out_uint8(out, 0); /* Delete pending */
out_uint8(out, S_ISDIR(filestat.st_mode) ? 1 : 0); /* Directory */
break;
case FileObjectIdInformation:
out_uint32_le(out, file_attributes); /* File Attributes */
out_uint32_le(out, 0); /* Reparse Tag */
break;
default:
unimpl("IRP Query (File) Information class: 0x%x\n", info_class);
return STATUS_INVALID_PARAMETER;
}
return STATUS_SUCCESS;
}
NTSTATUS
disk_set_information(NTHANDLE handle, uint32 info_class, STREAM in, STREAM out)
{
uint32 length, file_attributes, ft_high, ft_low, delete_on_close;
char newname[PATH_MAX], fullpath[PATH_MAX];
struct fileinfo *pfinfo;
int mode;
struct stat filestat;
time_t write_time, change_time, access_time, mod_time;
struct utimbuf tvs;
struct STATFS_T stat_fs;
pfinfo = &(g_fileinfo[handle]);
g_notify_stamp = True;
switch (info_class)
{
case FileBasicInformation:
write_time = change_time = access_time = 0;
in_uint8s(in, 4); /* Handle of root dir? */
in_uint8s(in, 24); /* unknown */
/* CreationTime */
in_uint32_le(in, ft_low);
in_uint32_le(in, ft_high);
/* AccessTime */
in_uint32_le(in, ft_low);
in_uint32_le(in, ft_high);
if (ft_low || ft_high)
access_time = convert_1970_to_filetime(ft_high, ft_low);
/* WriteTime */
in_uint32_le(in, ft_low);
in_uint32_le(in, ft_high);
if (ft_low || ft_high)
write_time = convert_1970_to_filetime(ft_high, ft_low);
/* ChangeTime */
in_uint32_le(in, ft_low);
in_uint32_le(in, ft_high);
if (ft_low || ft_high)
change_time = convert_1970_to_filetime(ft_high, ft_low);
in_uint32_le(in, file_attributes);
if (fstat(handle, &filestat))
return STATUS_ACCESS_DENIED;
tvs.modtime = filestat.st_mtime;
tvs.actime = filestat.st_atime;
if (access_time)
tvs.actime = access_time;
if (write_time || change_time)
mod_time = MIN(write_time, change_time);
else
mod_time = write_time ? write_time : change_time;
if (mod_time)
tvs.modtime = mod_time;
if (access_time || write_time || change_time)
{
#if WITH_DEBUG_RDP5
printf("FileBasicInformation access time %s",
ctime(&tvs.actime));
printf("FileBasicInformation modification time %s",
ctime(&tvs.modtime));
#endif
if (utime(pfinfo->path, &tvs) && errno != EPERM)
return STATUS_ACCESS_DENIED;
}
if (!file_attributes)
break; /* not valid */
mode = filestat.st_mode;
if (file_attributes & FILE_ATTRIBUTE_READONLY)
mode &= ~(S_IWUSR | S_IWGRP | S_IWOTH);
else
mode |= S_IWUSR;
mode &= 0777;
#if WITH_DEBUG_RDP5
printf("FileBasicInformation set access mode 0%o", mode);
#endif
if (fchmod(handle, mode))
return STATUS_ACCESS_DENIED;
break;
case FileRenameInformation:
in_uint8s(in, 4); /* Handle of root dir? */
in_uint8s(in, 0x1a); /* unknown */
in_uint32_le(in, length);
if (length && (length / 2) < 256)
{
rdp_in_unistr(in, newname, length);
convert_to_unix_filename(newname);
}
else
{
return STATUS_INVALID_PARAMETER;
}
sprintf(fullpath, "%s%s", g_rdpdr_device[pfinfo->device_id].local_path,
newname);
if (rename(pfinfo->path, fullpath) != 0)
{
perror("rename");
return STATUS_ACCESS_DENIED;
}
break;
case FileDispositionInformation:
/* As far as I understand it, the correct
thing to do here is to *schedule* a delete,
so it will be deleted when the file is
closed. Subsequent
FileDispositionInformation requests with
DeleteFile set to FALSE should unschedule
the delete. See
http://www.osronline.com/article.cfm?article=245. */
in_uint32_le(in, delete_on_close);
if (delete_on_close ||
(pfinfo->
accessmask & (FILE_DELETE_ON_CLOSE | FILE_COMPLETE_IF_OPLOCKED)))
{
pfinfo->delete_on_close = True;
}
break;
case FileAllocationInformation:
/* Fall through to FileEndOfFileInformation,
which uses ftrunc. This is like Samba with
"strict allocation = false", and means that
we won't detect out-of-quota errors, for
example. */
case FileEndOfFileInformation:
in_uint8s(in, 28); /* unknown */
in_uint32_le(in, length); /* file size */
/* prevents start of writing if not enough space left on device */
if (STATFS_FN(g_rdpdr_device[pfinfo->device_id].local_path, &stat_fs) == 0)
if (stat_fs.f_bfree * stat_fs.f_bsize < length)
return STATUS_DISK_FULL;
if (ftruncate_growable(handle, length) != 0)
{
return STATUS_DISK_FULL;
}
break;
default:
unimpl("IRP Set File Information class: 0x%x\n", info_class);
return STATUS_INVALID_PARAMETER;
}
return STATUS_SUCCESS;
}
NTSTATUS
disk_check_notify(NTHANDLE handle)
{
struct fileinfo *pfinfo;
NTSTATUS status = STATUS_PENDING;
NOTIFY notify;
pfinfo = &(g_fileinfo[handle]);
if (!pfinfo->pdir)
return STATUS_INVALID_DEVICE_REQUEST;
status = NotifyInfo(handle, pfinfo->info_class, ¬ify);
if (status != STATUS_PENDING)
return status;
if (memcmp(&pfinfo->notify, ¬ify, sizeof(NOTIFY)))
{
/*printf("disk_check_notify found changed event\n"); */
memcpy(&pfinfo->notify, ¬ify, sizeof(NOTIFY));
status = STATUS_NOTIFY_ENUM_DIR;
}
return status;
}
NTSTATUS
disk_create_notify(NTHANDLE handle, uint32 info_class)
{
struct fileinfo *pfinfo;
NTSTATUS ret = STATUS_PENDING;
/* printf("start disk_create_notify info_class %X\n", info_class); */
pfinfo = &(g_fileinfo[handle]);
pfinfo->info_class = info_class;
ret = NotifyInfo(handle, info_class, &pfinfo->notify);
if (info_class & 0x1000)
{ /* ???? */
if (ret == STATUS_PENDING)
return STATUS_SUCCESS;
}
/* printf("disk_create_notify: num_entries %d\n", pfinfo->notify.num_entries); */
return ret;
}
static NTSTATUS
NotifyInfo(NTHANDLE handle, uint32 info_class, NOTIFY * p)
{
struct fileinfo *pfinfo;
struct stat buf;
struct dirent *dp;
char *fullname;
DIR *dpr;
pfinfo = &(g_fileinfo[handle]);
if (fstat(handle, &buf) < 0)
{
perror("NotifyInfo");
return STATUS_ACCESS_DENIED;
}
p->modify_time = buf.st_mtime;
p->status_time = buf.st_ctime;
p->num_entries = 0;
p->total_time = 0;
dpr = opendir(pfinfo->path);
if (!dpr)
{
perror("NotifyInfo");
return STATUS_ACCESS_DENIED;
}
while ((dp = readdir(dpr)))
{
if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))
continue;
p->num_entries++;
fullname = (char *) xmalloc(strlen(pfinfo->path) + strlen(dp->d_name) + 2);
sprintf(fullname, "%s/%s", pfinfo->path, dp->d_name);
if (!stat(fullname, &buf))
{
p->total_time += (buf.st_mtime + buf.st_ctime);
}
xfree(fullname);
}
closedir(dpr);
return STATUS_PENDING;
}
static FsInfoType *
FsVolumeInfo(char *fpath)
{
static FsInfoType info;
#ifdef USE_SETMNTENT
FILE *fdfs;
struct mntent *e;
#endif
/* initialize */
memset(&info, 0, sizeof(info));
strcpy(info.label, "RDESKTOP");
strcpy(info.type, "RDPFS");
#ifdef USE_SETMNTENT
fdfs = setmntent(MNTENT_PATH, "r");
if (!fdfs)
return &info;
while ((e = getmntent(fdfs)))
{
if (str_startswith(e->mnt_dir, fpath))
{
strcpy(info.type, e->mnt_type);
strcpy(info.name, e->mnt_fsname);
if (strstr(e->mnt_opts, "vfat") || strstr(e->mnt_opts, "iso9660"))
{
int fd = open(e->mnt_fsname, O_RDONLY);
if (fd >= 0)
{
unsigned char buf[512];
memset(buf, 0, sizeof(buf));
if (strstr(e->mnt_opts, "vfat"))
/*FAT*/
{
strcpy(info.type, "vfat");
read(fd, buf, sizeof(buf));
info.serial =
(buf[42] << 24) + (buf[41] << 16) +
(buf[40] << 8) + buf[39];
strncpy(info.label, (char *) buf + 43, 10);
info.label[10] = '\0';
}
else if (lseek(fd, 32767, SEEK_SET) >= 0) /* ISO9660 */
{
read(fd, buf, sizeof(buf));
strncpy(info.label, (char *) buf + 41, 32);
info.label[32] = '\0';
/* info.Serial = (buf[128]<<24)+(buf[127]<<16)+(buf[126]<<8)+buf[125]; */
}
close(fd);
}
}
}
}
endmntent(fdfs);
#else
/* initialize */
memset(&info, 0, sizeof(info));
strcpy(info.label, "RDESKTOP");
strcpy(info.type, "RDPFS");
#endif
return &info;
}
NTSTATUS
disk_query_volume_information(NTHANDLE handle, uint32 info_class, STREAM out)
{
struct STATFS_T stat_fs;
struct fileinfo *pfinfo;
FsInfoType *fsinfo;
pfinfo = &(g_fileinfo[handle]);
if (STATFS_FN(pfinfo->path, &stat_fs) != 0)
{
perror("statfs");
return STATUS_ACCESS_DENIED;
}
fsinfo = FsVolumeInfo(pfinfo->path);
switch (info_class)
{
case FileFsVolumeInformation:
out_uint32_le(out, 0); /* volume creation time low */
out_uint32_le(out, 0); /* volume creation time high */
out_uint32_le(out, fsinfo->serial); /* serial */
out_uint32_le(out, 2 * strlen(fsinfo->label)); /* length of string */
out_uint8(out, 0); /* support objects? */
rdp_out_unistr(out, fsinfo->label, 2 * strlen(fsinfo->label) - 2);
break;
case FileFsSizeInformation:
out_uint32_le(out, stat_fs.f_blocks); /* Total allocation units low */
out_uint32_le(out, 0); /* Total allocation high units */
out_uint32_le(out, stat_fs.f_bfree); /* Available allocation units */
out_uint32_le(out, 0); /* Available allowcation units */
out_uint32_le(out, stat_fs.f_bsize / 0x200); /* Sectors per allocation unit */
out_uint32_le(out, 0x200); /* Bytes per sector */
break;
case FileFsAttributeInformation:
out_uint32_le(out, FS_CASE_SENSITIVE | FS_CASE_IS_PRESERVED); /* fs attributes */
out_uint32_le(out, F_NAMELEN(stat_fs)); /* max length of filename */
out_uint32_le(out, 2 * strlen(fsinfo->type)); /* length of fs_type */
rdp_out_unistr(out, fsinfo->type, 2 * strlen(fsinfo->type) - 2);
break;
case FileFsLabelInformation:
case FileFsDeviceInformation:
case FileFsControlInformation:
case FileFsFullSizeInformation:
case FileFsObjectIdInformation:
case FileFsMaximumInformation:
default:
unimpl("IRP Query Volume Information class: 0x%x\n", info_class);
return STATUS_INVALID_PARAMETER;
}
return STATUS_SUCCESS;
}
NTSTATUS
disk_query_directory(NTHANDLE handle, uint32 info_class, char *pattern, STREAM out)
{
uint32 file_attributes, ft_low, ft_high;
char *dirname, fullpath[PATH_MAX];
DIR *pdir;
struct dirent *pdirent;
struct stat fstat;
struct fileinfo *pfinfo;
pfinfo = &(g_fileinfo[handle]);
pdir = pfinfo->pdir;
dirname = pfinfo->path;
file_attributes = 0;
switch (info_class)
{
case FileBothDirectoryInformation:
/* If a search pattern is received, remember this pattern, and restart search */
if (pattern[0] != 0)
{
strncpy(pfinfo->pattern, 1 + strrchr(pattern, '/'), PATH_MAX - 1);
rewinddir(pdir);
}
/* find next dirent matching pattern */
pdirent = readdir(pdir);
while (pdirent && fnmatch(pfinfo->pattern, pdirent->d_name, 0) != 0)
pdirent = readdir(pdir);
if (pdirent == NULL)
return STATUS_NO_MORE_FILES;
/* Get information for directory entry */
sprintf(fullpath, "%s/%s", dirname, pdirent->d_name);
if (stat(fullpath, &fstat))
{
switch (errno)
{
case ENOENT:
case ELOOP:
case EACCES:
/* These are non-fatal errors. */
memset(&fstat, 0, sizeof(fstat));
break;
default:
/* Fatal error. By returning STATUS_NO_SUCH_FILE,
the directory list operation will be aborted */
perror(fullpath);
out_uint8(out, 0);
return STATUS_NO_SUCH_FILE;
}
}
if (S_ISDIR(fstat.st_mode))
file_attributes |= FILE_ATTRIBUTE_DIRECTORY;
if (pdirent->d_name[0] == '.')
file_attributes |= FILE_ATTRIBUTE_HIDDEN;
if (!file_attributes)
file_attributes |= FILE_ATTRIBUTE_NORMAL;
if (!(fstat.st_mode & S_IWUSR))
file_attributes |= FILE_ATTRIBUTE_READONLY;
/* Return requested information */
out_uint8s(out, 8); /* unknown zero */
seconds_since_1970_to_filetime(get_create_time(&fstat), &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* create time */
out_uint32_le(out, ft_high);
seconds_since_1970_to_filetime(fstat.st_atime, &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* last_access_time */
out_uint32_le(out, ft_high);
seconds_since_1970_to_filetime(fstat.st_mtime, &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* last_write_time */
out_uint32_le(out, ft_high);
seconds_since_1970_to_filetime(fstat.st_ctime, &ft_high, &ft_low);
out_uint32_le(out, ft_low); /* change_write_time */
out_uint32_le(out, ft_high);
out_uint32_le(out, fstat.st_size); /* filesize low */
out_uint32_le(out, 0); /* filesize high */
out_uint32_le(out, fstat.st_size); /* filesize low */
out_uint32_le(out, 0); /* filesize high */
out_uint32_le(out, file_attributes);
out_uint8(out, 2 * strlen(pdirent->d_name) + 2); /* unicode length */
out_uint8s(out, 7); /* pad? */
out_uint8(out, 0); /* 8.3 file length */
out_uint8s(out, 2 * 12); /* 8.3 unicode length */
rdp_out_unistr(out, pdirent->d_name, 2 * strlen(pdirent->d_name));
break;
default:
/* FIXME: Support FileDirectoryInformation,
FileFullDirectoryInformation, and
FileNamesInformation */
unimpl("IRP Query Directory sub: 0x%x\n", info_class);
return STATUS_INVALID_PARAMETER;
}
return STATUS_SUCCESS;
}
static NTSTATUS
disk_device_control(NTHANDLE handle, uint32 request, STREAM in, STREAM out)
{
if (((request >> 16) != 20) || ((request >> 16) != 9))
return STATUS_INVALID_PARAMETER;
/* extract operation */
request >>= 2;
request &= 0xfff;
printf("DISK IOCTL %d\n", request);
switch (request)
{
case 25: /* ? */
case 42: /* ? */
default:
unimpl("DISK IOCTL %d\n", request);
return STATUS_INVALID_PARAMETER;
}
return STATUS_SUCCESS;
}
DEVICE_FNS disk_fns = {
disk_create,
disk_close,
disk_read,
disk_write,
disk_device_control /* device_control */
};
syntax highlighted by Code2HTML, v. 0.9.1