i2p/haiku-beta6
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

scsi & virtio: Clean up IORequest usage.

Browse Source 
* Use TransferredBytes() instead of assuming length.
 * Consolidate checks and invoke io hook instead of scheduler directly.
This commit is contained in:
Augustin Cavalier
2023-04-27 15:19:39 -04:00
parent 8d2c997da2
commit f64b099191
3 changed files with 29 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/add-ons/kernel/drivers/disk/scsi/scsi_cd/scsi_cd.cpp
+17 -39
View File
@@ -748,32 +748,35 @@ cd_free(void* cookie)
static status_t
cd_read(void* cookie, off_t pos, void* buffer, size_t* _length)
cd_io(void* cookie, io_request* request)
{
cd_handle* handle = (cd_handle*)cookie;
size_t length = *_length;
if (handle->info->capacity == 0)
if (handle->info->capacity == 0 || handle->info->io_scheduler == NULL) {
notify_io_request(request, B_DEV_NO_MEDIA);
return B_DEV_NO_MEDIA;
}
return handle->info->io_scheduler->ScheduleRequest(request);
}
static status_t
cd_read(void* cookie, off_t pos, void* buffer, size_t* _length)
{
size_t length = *_length;
IORequest request;
status_t status = request.Init(pos, (addr_t)buffer, length, false, 0);
if (status != B_OK)
return status;
if (handle->info->io_scheduler == NULL)
return B_DEV_NO_MEDIA;
status = handle->info->io_scheduler->ScheduleRequest(&request);
status = cd_io(cookie, &request);
if (status != B_OK)
return status;
status = request.Wait(0, 0);
if (status == B_OK)
*_length = length;
else
dprintf("cd_read(): request.Wait() returned: %s\n", strerror(status));
*_length = request.TransferredBytes();
return status;
}
@@ -781,48 +784,23 @@ cd_read(void* cookie, off_t pos, void* buffer, size_t* _length)
static status_t
cd_write(void* cookie, off_t pos, const void* buffer, size_t* _length)
{
cd_handle* handle = (cd_handle*)cookie;
size_t length = *_length;
if (handle->info->capacity == 0)
return B_DEV_NO_MEDIA;
IORequest request;
status_t status = request.Init(pos, (addr_t)buffer, length, true, 0);
if (status != B_OK)
return status;
if (handle->info->io_scheduler == NULL)
return B_DEV_NO_MEDIA;
status = handle->info->io_scheduler->ScheduleRequest(&request);
status = cd_io(cookie, &request);
if (status != B_OK)
return status;
status = request.Wait(0, 0);
if (status == B_OK)
*_length = length;
else
dprintf("cd_write(): request.Wait() returned: %s\n", strerror(status));
*_length = request.TransferredBytes();
return status;
}
static status_t
cd_io(void* cookie, io_request* request)
{
cd_handle* handle = (cd_handle*)cookie;
if (handle->info->capacity == 0) {
notify_io_request(request, B_DEV_NO_MEDIA);
return B_DEV_NO_MEDIA;
}
return handle->info->io_scheduler->ScheduleRequest(request);
}
static status_t
cd_ioctl(void* cookie, uint32 op, void* buffer, size_t length)
{
src/add-ons/kernel/drivers/disk/scsi/scsi_disk/scsi_disk.cpp
+6 -8
View File
@@ -341,10 +341,9 @@ das_read(void* cookie, off_t pos, void* buffer, size_t* _length)
return status;
status = request.Wait(0, 0);
if (status == B_OK)
*_length = length;
else
dprintf("das_read(): request.Wait() returned: %s\n", strerror(status));
*_length = request.TransferredBytes();
if (status != B_OK)
dprintf("das_read: request.Wait() returned: %s\n", strerror(status));
return status;
}
@@ -366,10 +365,9 @@ das_write(void* cookie, off_t pos, const void* buffer, size_t* _length)
return status;
status = request.Wait(0, 0);
if (status == B_OK)
*_length = length;
else
dprintf("das_write(): request.Wait() returned: %s\n", strerror(status));
*_length = request.TransferredBytes();
if (status != B_OK)
dprintf("das_write: request.Wait() returned: %s\n", strerror(status));
return status;
}
src/add-ons/kernel/drivers/disk/virtual/virtio_block/virtio_block.cpp
+6 -8
View File
@@ -352,10 +352,9 @@ virtio_block_read(void* cookie, off_t pos, void* buffer, size_t* _length)
return status;
status = request.Wait(0, 0);
if (status == B_OK)
*_length = length;
else
dprintf("read(): request.Wait() returned: %s\n", strerror(status));
*_length = request.TransferredBytes();
if (status != B_OK)
dprintf("virtio_block_read: request.Wait() returned: %s\n", strerror(status));
return status;
}
@@ -379,10 +378,9 @@ virtio_block_write(void* cookie, off_t pos, const void* buffer,
return status;
status = request.Wait(0, 0);
if (status == B_OK)
*_length = length;
else
dprintf("write(): request.Wait() returned: %s\n", strerror(status));
*_length = request.TransferredBytes();
if (status != B_OK)
dprintf("virtio_block_write: request.Wait() returned: %s\n", strerror(status));
return status;
}