* fhandler.h (class fhandler_dev_raw): Add drive information members
drive_size, current_position and bytes_per_sector. (fhandler_dev_floppy::get_drive_info): Declare. * fhandler_floppy.cc (fhandler_dev_floppy::is_eom): Define ERROR_SEEK and ERROR_SECTOR_NOT_FOUND as end-of-medium conditions. (fhandler_dev_floppy::get_drive_info): New method to have one function retrieving drive info. (fhandler_dev_floppy::open): Call get_drive_info to get drive information right from the start. (fhandler_dev_floppy::lseek): Use and set drive information members. Especially keep track of current_position. (fhandler_dev_floppy::ioctl): Ditto. * fhandler_raw.cc (fhandler_dev_raw::write_file): Keep track of current_position. (fhandler_dev_raw::read_file): Ditto. (fhandler_dev_raw::raw_read): Never try to read beyond end-of-medium. (fhandler_dev_raw::dup): Handle new drive information members.
This commit is contained in:
@@ -27,7 +27,9 @@ details. */
|
||||
int
|
||||
fhandler_dev_floppy::is_eom (int win_error)
|
||||
{
|
||||
int ret = (win_error == ERROR_INVALID_PARAMETER);
|
||||
int ret = (win_error == ERROR_INVALID_PARAMETER
|
||||
|| win_error == ERROR_SEEK
|
||||
|| win_error == ERROR_SECTOR_NOT_FOUND);
|
||||
if (ret)
|
||||
debug_printf ("end of medium");
|
||||
return ret;
|
||||
@@ -48,39 +50,11 @@ fhandler_dev_floppy::fhandler_dev_floppy ()
|
||||
}
|
||||
|
||||
int
|
||||
fhandler_dev_floppy::open (int flags, mode_t)
|
||||
fhandler_dev_floppy::get_drive_info (struct hd_geometry *geo)
|
||||
{
|
||||
/* The correct size of the buffer would be 512 bytes,
|
||||
* which is the atomic size, supported by WinNT.
|
||||
* Unfortunately, the performance is worse than
|
||||
* access to file system on same device!
|
||||
* Setting buffer size to a relatively big value
|
||||
* increases performance by means.
|
||||
* The new ioctl call with 'rdevio.h' header file
|
||||
* supports changing this value.
|
||||
*
|
||||
* Let's be smart: Let's take a multiplier of typical tar
|
||||
* and cpio buffer sizes by default!
|
||||
*/
|
||||
devbufsiz = 61440L; /* 512L; */
|
||||
return fhandler_dev_raw::open (flags);
|
||||
}
|
||||
|
||||
_off64_t
|
||||
fhandler_dev_floppy::lseek (_off64_t offset, int whence)
|
||||
{
|
||||
char buf[512];
|
||||
_off64_t drive_size = 0;
|
||||
_off64_t lloffset = offset;
|
||||
_off64_t current_position;
|
||||
_off64_t sector_aligned_offset;
|
||||
_off64_t bytes_left;
|
||||
DWORD low;
|
||||
LONG high = 0;
|
||||
|
||||
DISK_GEOMETRY di;
|
||||
PARTITION_INFORMATION pi;
|
||||
DWORD bytes_read;
|
||||
DWORD bytes_read = 0;
|
||||
|
||||
if (!DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_DRIVE_GEOMETRY,
|
||||
@@ -96,15 +70,16 @@ fhandler_dev_floppy::lseek (_off64_t offset, int whence)
|
||||
di.TracksPerCylinder,
|
||||
di.SectorsPerTrack,
|
||||
di.BytesPerSector);
|
||||
bytes_per_sector = di.BytesPerSector;
|
||||
if (DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_PARTITION_INFO,
|
||||
NULL, 0,
|
||||
&pi, sizeof (pi),
|
||||
&bytes_read, NULL))
|
||||
{
|
||||
debug_printf ("partition info: %ld (%ld)",
|
||||
pi.StartingOffset.LowPart,
|
||||
pi.PartitionLength.LowPart);
|
||||
debug_printf ("partition info: offset %D length %D",
|
||||
pi.StartingOffset.QuadPart,
|
||||
pi.PartitionLength.QuadPart);
|
||||
drive_size = pi.PartitionLength.QuadPart;
|
||||
}
|
||||
else
|
||||
@@ -112,7 +87,53 @@ fhandler_dev_floppy::lseek (_off64_t offset, int whence)
|
||||
drive_size = di.Cylinders.QuadPart * di.TracksPerCylinder *
|
||||
di.SectorsPerTrack * di.BytesPerSector;
|
||||
}
|
||||
debug_printf ("drive size: %ld", drive_size);
|
||||
debug_printf ("drive size: %D", drive_size);
|
||||
if (geo)
|
||||
{
|
||||
geo->heads = di.TracksPerCylinder;
|
||||
geo->sectors = di.SectorsPerTrack;
|
||||
geo->cylinders = di.Cylinders.LowPart;
|
||||
geo->start = pi.StartingOffset.QuadPart >> 9ULL;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
fhandler_dev_floppy::open (int flags, mode_t)
|
||||
{
|
||||
/* The correct size of the buffer would be 512 bytes,
|
||||
* which is the atomic size, supported by WinNT.
|
||||
* Unfortunately, the performance is worse than
|
||||
* access to file system on same device!
|
||||
* Setting buffer size to a relatively big value
|
||||
* increases performance by means.
|
||||
* The new ioctl call with 'rdevio.h' header file
|
||||
* supports changing this value.
|
||||
*
|
||||
* Let's be smart: Let's take a multiplier of typical tar
|
||||
* and cpio buffer sizes by default!
|
||||
*/
|
||||
devbufsiz = 61440L; /* 512L; */
|
||||
int ret = fhandler_dev_raw::open (flags);
|
||||
|
||||
if (ret && get_drive_info (NULL))
|
||||
{
|
||||
close ();
|
||||
return 0;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
_off64_t
|
||||
fhandler_dev_floppy::lseek (_off64_t offset, int whence)
|
||||
{
|
||||
char buf[512];
|
||||
_off64_t lloffset = offset;
|
||||
_off64_t sector_aligned_offset;
|
||||
_off64_t bytes_left;
|
||||
DWORD low;
|
||||
LONG high = 0;
|
||||
|
||||
if (whence == SEEK_END && drive_size > 0)
|
||||
{
|
||||
@@ -122,134 +143,59 @@ fhandler_dev_floppy::lseek (_off64_t offset, int whence)
|
||||
|
||||
if (whence == SEEK_CUR)
|
||||
{
|
||||
low = SetFilePointer (get_handle (), 0, &high, FILE_CURRENT);
|
||||
if (low == INVALID_SET_FILE_POINTER && GetLastError ())
|
||||
{
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
current_position = low + ((_off64_t) high << 32);
|
||||
/* devbufend and devbufstart are always 0 when writing. */
|
||||
current_position -= devbufend - devbufstart;
|
||||
|
||||
lloffset += current_position;
|
||||
lloffset += current_position - (devbufend - devbufstart);
|
||||
whence = SEEK_SET;
|
||||
}
|
||||
|
||||
if (lloffset < 0 ||
|
||||
drive_size > 0 && lloffset > drive_size)
|
||||
if (whence != SEEK_SET
|
||||
|| lloffset < 0
|
||||
|| drive_size > 0 && lloffset >= drive_size)
|
||||
{
|
||||
set_errno (EINVAL);
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* FIXME: sector can possibly be not 512 bytes long */
|
||||
sector_aligned_offset = (lloffset / 512) * 512;
|
||||
sector_aligned_offset = (lloffset / bytes_per_sector) * bytes_per_sector;
|
||||
bytes_left = lloffset - sector_aligned_offset;
|
||||
|
||||
if (whence == SEEK_SET)
|
||||
/* Invalidate buffer. */
|
||||
devbufstart = devbufend = 0;
|
||||
|
||||
low = sector_aligned_offset & UINT32_MAX;
|
||||
high = sector_aligned_offset >> 32;
|
||||
if (SetFilePointer (get_handle (), low, &high, FILE_BEGIN)
|
||||
== INVALID_SET_FILE_POINTER && GetLastError ())
|
||||
{
|
||||
/* Invalidate buffer. */
|
||||
devbufstart = devbufend = 0;
|
||||
|
||||
low = sector_aligned_offset & UINT32_MAX;
|
||||
high = sector_aligned_offset >> 32;
|
||||
if (SetFilePointer (get_handle (), low, &high, FILE_BEGIN)
|
||||
== INVALID_SET_FILE_POINTER && GetLastError ())
|
||||
{
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
|
||||
eom_detected (false);
|
||||
size_t len = bytes_left;
|
||||
raw_read (buf, len);
|
||||
return sector_aligned_offset + bytes_left;
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
|
||||
set_errno (EINVAL);
|
||||
return -1;
|
||||
eom_detected (false);
|
||||
current_position = sector_aligned_offset;
|
||||
if (bytes_left)
|
||||
{
|
||||
size_t len = bytes_left;
|
||||
raw_read (buf, len);
|
||||
}
|
||||
return current_position + bytes_left;
|
||||
}
|
||||
|
||||
int
|
||||
fhandler_dev_floppy::ioctl (unsigned int cmd, void *buf)
|
||||
{
|
||||
DISK_GEOMETRY di;
|
||||
PARTITION_INFORMATION pi;
|
||||
DWORD bytes_read;
|
||||
_off64_t drive_size = 0;
|
||||
_off64_t start = 0;
|
||||
switch (cmd)
|
||||
{
|
||||
case HDIO_GETGEO:
|
||||
{
|
||||
debug_printf ("HDIO_GETGEO");
|
||||
if (!DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_DRIVE_GEOMETRY,
|
||||
NULL, 0,
|
||||
&di, sizeof (di),
|
||||
&bytes_read, NULL))
|
||||
{
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
debug_printf ("disk geometry: (%ld cyl)*(%ld trk)*(%ld sec)*(%ld bps)",
|
||||
di.Cylinders.LowPart,
|
||||
di.TracksPerCylinder,
|
||||
di.SectorsPerTrack,
|
||||
di.BytesPerSector);
|
||||
if (DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_PARTITION_INFO,
|
||||
NULL, 0,
|
||||
&pi, sizeof (pi),
|
||||
&bytes_read, NULL))
|
||||
{
|
||||
debug_printf ("partition info: %ld (%ld)",
|
||||
pi.StartingOffset.LowPart,
|
||||
pi.PartitionLength.LowPart);
|
||||
start = pi.StartingOffset.QuadPart >> 9ULL;
|
||||
}
|
||||
struct hd_geometry *geo = (struct hd_geometry *) buf;
|
||||
geo->heads = di.TracksPerCylinder;
|
||||
geo->sectors = di.SectorsPerTrack;
|
||||
geo->cylinders = di.Cylinders.LowPart;
|
||||
geo->start = start;
|
||||
return 0;
|
||||
return get_drive_info ((struct hd_geometry *) buf);
|
||||
}
|
||||
case BLKGETSIZE:
|
||||
case BLKGETSIZE64:
|
||||
{
|
||||
debug_printf ("BLKGETSIZE");
|
||||
if (!DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_DRIVE_GEOMETRY,
|
||||
NULL, 0,
|
||||
&di, sizeof (di),
|
||||
&bytes_read, NULL))
|
||||
{
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
debug_printf ("disk geometry: (%ld cyl)*(%ld trk)*(%ld sec)*(%ld bps)",
|
||||
di.Cylinders.LowPart,
|
||||
di.TracksPerCylinder,
|
||||
di.SectorsPerTrack,
|
||||
di.BytesPerSector);
|
||||
if (DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_PARTITION_INFO,
|
||||
NULL, 0,
|
||||
&pi, sizeof (pi),
|
||||
&bytes_read, NULL))
|
||||
{
|
||||
debug_printf ("partition info: %ld (%ld)",
|
||||
pi.StartingOffset.LowPart,
|
||||
pi.PartitionLength.LowPart);
|
||||
drive_size = pi.PartitionLength.QuadPart;
|
||||
}
|
||||
else
|
||||
{
|
||||
drive_size = di.Cylinders.QuadPart * di.TracksPerCylinder *
|
||||
di.SectorsPerTrack * di.BytesPerSector;
|
||||
}
|
||||
if (cmd == BLKGETSIZE)
|
||||
*(long *)buf = drive_size >> 9UL;
|
||||
else
|
||||
@@ -268,26 +214,13 @@ fhandler_dev_floppy::ioctl (unsigned int cmd, void *buf)
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
get_drive_info (NULL);
|
||||
return 0;
|
||||
}
|
||||
case BLKSSZGET:
|
||||
{
|
||||
debug_printf ("BLKSSZGET");
|
||||
if (!DeviceIoControl (get_handle (),
|
||||
IOCTL_DISK_GET_DRIVE_GEOMETRY,
|
||||
NULL, 0,
|
||||
&di, sizeof (di),
|
||||
&bytes_read, NULL))
|
||||
{
|
||||
__seterrno ();
|
||||
return -1;
|
||||
}
|
||||
debug_printf ("disk geometry: (%ld cyl)*(%ld trk)*(%ld sec)*(%ld bps)",
|
||||
di.Cylinders.LowPart,
|
||||
di.TracksPerCylinder,
|
||||
di.SectorsPerTrack,
|
||||
di.BytesPerSector);
|
||||
*(int *)buf = di.BytesPerSector;
|
||||
*(int *)buf = bytes_per_sector;
|
||||
return 0;
|
||||
}
|
||||
default:
|
||||
|
Reference in New Issue
Block a user