X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=palacios%2Fsrc%2Fdevices%2Fata.h;h=5cdb7372d0136615d04601dc69bdd3740e9d7799;hb=0fe8e057d19e132f88ab71dce8724b2c5585803a;hp=721f2f499e10576dcbca1f39d9c0aa2bbb4f9bb4;hpb=d0886c953a9e5969ee0f54f11bb146bfe888bfe3;p=palacios.releases.git

diff --git a/palacios/src/devices/ata.h b/palacios/src/devices/ata.h
index 721f2f4..5cdb737 100644
--- a/palacios/src/devices/ata.h
+++ b/palacios/src/devices/ata.h
@@ -43,8 +43,8 @@ static void ata_identify_device(struct ide_drive * drive) {
     // Make it the simplest drive possible (1 head, 1 cyl, 1 sect/track)
     drive_id->num_cylinders = drive->num_cylinders;
     drive_id->num_heads = drive->num_heads;
-    drive_id->bytes_per_track = drive->num_sectors * IDE_SECTOR_SIZE;
-    drive_id->bytes_per_sector = IDE_SECTOR_SIZE;
+    drive_id->bytes_per_track = drive->num_sectors * HD_SECTOR_SIZE;
+    drive_id->bytes_per_sector = HD_SECTOR_SIZE;
     drive_id->sectors_per_track = drive->num_sectors;
 
 
@@ -127,6 +127,21 @@ static int ata_read(struct vm_device * dev, struct ide_channel * channel, uint8_
 }
 
 
+static int ata_write(struct vm_device * dev, struct ide_channel * channel, uint8_t * src, uint_t sect_cnt) {
+    struct ide_drive * drive = get_selected_drive(channel);
+
+    PrintDebug("Writing Drive LBA=%d (count=%d)\n", (uint32_t)(drive->current_lba), sect_cnt);
+
+    int ret = drive->hd_ops->write(src, sect_cnt, drive->current_lba, drive->private_data);
+
+    if (ret == -1) {
+	PrintError("IDE: Error writing HD block (LBA=%p)\n", (void *)(addr_t)(drive->current_lba));
+	return -1;
+    }
+
+    return 0;
+}
+
 
 
 static int ata_get_lba(struct vm_device * dev, struct ide_channel * channel, uint64_t * lba) {
@@ -152,11 +167,11 @@ static int ata_get_lba(struct vm_device * dev, struct ide_channel * channel, uin
     lba_addr.buf[3] = channel->drive_head.lba3;
 
 
-    if (lba_addr.addr + (sect_cnt * IDE_SECTOR_SIZE) > 
+    if ((lba_addr.addr + sect_cnt) > 
 	drive->hd_ops->get_capacity(drive->private_data)) {
 	PrintError("IDE: request size exceeds disk capacity (lba=%d) (sect_cnt=%d) (ReadEnd=%d) (capacity=%p)\n", 
 		   lba_addr.addr, sect_cnt, 
-		   lba_addr.addr + (sect_cnt * IDE_SECTOR_SIZE),
+		   lba_addr.addr + (sect_cnt * HD_SECTOR_SIZE),
 		   (void *)(addr_t)(drive->hd_ops->get_capacity(drive->private_data)));
 	return -1;
     }
@@ -183,7 +198,7 @@ static int ata_read_sectors(struct vm_device * dev, struct ide_channel * channel
 	return -1;
     }
 
-    drive->transfer_length = sect_cnt * IDE_SECTOR_SIZE;
+    drive->transfer_length = sect_cnt * HD_SECTOR_SIZE;
     drive->transfer_index = 0;
 
     channel->status.busy = 0;