Message ID | 20100211201944.GA25584@amt.cnet |
---|---|
State | New |
Headers | show |
On 02/11/2010 02:19 PM, Marcelo Tosatti wrote: > Save/restore information necessary to continue in progress PIO/ATAPI CMD > transfers. > > This includes the IO buffer. > > Signed-off-by: Marcelo Tosatti<mtosatti@redhat.com> > Applied. Thanks. Regards, Anthony Liguori > diff --git a/hw/ide/core.c b/hw/ide/core.c > index b6643e8..64aebc2 100644 > --- a/hw/ide/core.c > +++ b/hw/ide/core.c > @@ -2640,6 +2640,7 @@ void ide_init2(IDEBus *bus, DriveInfo *hd0, DriveInfo *hd1, > s->unit = i; > s->drive_serial = drive_serial++; > s->io_buffer = qemu_blockalign(s->bs, IDE_DMA_BUF_SECTORS*512 + 4); > + s->io_buffer_total_len = IDE_DMA_BUF_SECTORS*512 + 4; > s->smart_selftest_data = qemu_blockalign(s->bs, 512); > s->sector_write_timer = qemu_new_timer(vm_clock, > ide_sector_write_timer_cb, s); > @@ -2674,6 +2675,25 @@ static bool is_identify_set(void *opaque, int version_id) > return s->identify_set != 0; > } > > +static EndTransferFunc* transfer_end_table[] = { > + ide_sector_read, > + ide_sector_write, > + ide_transfer_stop, > + ide_atapi_cmd_reply_end, > + ide_atapi_cmd, > +}; > + > +static int transfer_end_table_idx(EndTransferFunc *fn) > +{ > + int i; > + > + for (i = 0; i< ARRAY_SIZE(transfer_end_table); i++) > + if (transfer_end_table[i] == fn) > + return i; > + > + return -1; > +} > + > static int ide_drive_post_load(void *opaque, int version_id) > { > IDEState *s = opaque; > @@ -2684,14 +2704,42 @@ static int ide_drive_post_load(void *opaque, int version_id) > s->cdrom_changed = 1; > } > } > + > + if (s->cur_io_buffer_len) { > + s->end_transfer_func = transfer_end_table[s->end_transfer_fn_idx]; > + s->data_ptr = s->io_buffer + s->cur_io_buffer_offset; > + s->data_end = s->data_ptr + s->cur_io_buffer_len; > + } > + > return 0; > } > > +static void ide_drive_pre_save(void *opaque) > +{ > + IDEState *s = opaque; > + > + s->cur_io_buffer_len = 0; > + > + if (!(s->status& DRQ_STAT)) > + return; > + > + s->cur_io_buffer_offset = s->data_ptr - s->io_buffer; > + s->cur_io_buffer_len = s->data_end - s->data_ptr; > + > + s->end_transfer_fn_idx = transfer_end_table_idx(s->end_transfer_func); > + if (s->end_transfer_fn_idx == -1) { > + fprintf(stderr, "%s: invalid end_transfer_func for DRQ_STAT\n", > + __func__); > + s->end_transfer_fn_idx = 2; > + } > +} > + > const VMStateDescription vmstate_ide_drive = { > .name = "ide_drive", > - .version_id = 3, > + .version_id = 4, > .minimum_version_id = 0, > .minimum_version_id_old = 0, > + .pre_save = ide_drive_pre_save, > .post_load = ide_drive_post_load, > .fields = (VMStateField []) { > VMSTATE_INT32(mult_sectors, IDEState), > @@ -2714,7 +2762,14 @@ const VMStateDescription vmstate_ide_drive = { > VMSTATE_UINT8(sense_key, IDEState), > VMSTATE_UINT8(asc, IDEState), > VMSTATE_UINT8_V(cdrom_changed, IDEState, 3), > - /* XXX: if a transfer is pending, we do not save it yet */ > + VMSTATE_INT32_V(req_nb_sectors, IDEState, 4), > + VMSTATE_VARRAY_INT32(io_buffer, IDEState, io_buffer_total_len, 4, > + vmstate_info_uint8, uint8_t), > + VMSTATE_INT32_V(cur_io_buffer_offset, IDEState, 4), > + VMSTATE_INT32_V(cur_io_buffer_len, IDEState, 4), > + VMSTATE_UINT8_V(end_transfer_fn_idx, IDEState, 4), > + VMSTATE_INT32_V(elementary_transfer_size, IDEState, 4), > + VMSTATE_INT32_V(packet_transfer_size, IDEState, 4), > VMSTATE_END_OF_LIST() > } > }; > diff --git a/hw/ide/internal.h b/hw/ide/internal.h > index 1cc4b55..8615d1a 100644 > --- a/hw/ide/internal.h > +++ b/hw/ide/internal.h > @@ -417,6 +417,11 @@ struct IDEState { > uint8_t *data_ptr; > uint8_t *data_end; > uint8_t *io_buffer; > + /* PIO save/restore */ > + int32_t io_buffer_total_len; > + int cur_io_buffer_offset; > + int cur_io_buffer_len; > + uint8_t end_transfer_fn_idx; > QEMUTimer *sector_write_timer; /* only used for win2k install hack */ > uint32_t irq_count; /* counts IRQs when using win2k install hack */ > /* CF-ATA extended error */ > > > >
diff --git a/hw/ide/core.c b/hw/ide/core.c index b6643e8..64aebc2 100644 --- a/hw/ide/core.c +++ b/hw/ide/core.c @@ -2640,6 +2640,7 @@ void ide_init2(IDEBus *bus, DriveInfo *hd0, DriveInfo *hd1, s->unit = i; s->drive_serial = drive_serial++; s->io_buffer = qemu_blockalign(s->bs, IDE_DMA_BUF_SECTORS*512 + 4); + s->io_buffer_total_len = IDE_DMA_BUF_SECTORS*512 + 4; s->smart_selftest_data = qemu_blockalign(s->bs, 512); s->sector_write_timer = qemu_new_timer(vm_clock, ide_sector_write_timer_cb, s); @@ -2674,6 +2675,25 @@ static bool is_identify_set(void *opaque, int version_id) return s->identify_set != 0; } +static EndTransferFunc* transfer_end_table[] = { + ide_sector_read, + ide_sector_write, + ide_transfer_stop, + ide_atapi_cmd_reply_end, + ide_atapi_cmd, +}; + +static int transfer_end_table_idx(EndTransferFunc *fn) +{ + int i; + + for (i = 0; i < ARRAY_SIZE(transfer_end_table); i++) + if (transfer_end_table[i] == fn) + return i; + + return -1; +} + static int ide_drive_post_load(void *opaque, int version_id) { IDEState *s = opaque; @@ -2684,14 +2704,42 @@ static int ide_drive_post_load(void *opaque, int version_id) s->cdrom_changed = 1; } } + + if (s->cur_io_buffer_len) { + s->end_transfer_func = transfer_end_table[s->end_transfer_fn_idx]; + s->data_ptr = s->io_buffer + s->cur_io_buffer_offset; + s->data_end = s->data_ptr + s->cur_io_buffer_len; + } + return 0; } +static void ide_drive_pre_save(void *opaque) +{ + IDEState *s = opaque; + + s->cur_io_buffer_len = 0; + + if (!(s->status & DRQ_STAT)) + return; + + s->cur_io_buffer_offset = s->data_ptr - s->io_buffer; + s->cur_io_buffer_len = s->data_end - s->data_ptr; + + s->end_transfer_fn_idx = transfer_end_table_idx(s->end_transfer_func); + if (s->end_transfer_fn_idx == -1) { + fprintf(stderr, "%s: invalid end_transfer_func for DRQ_STAT\n", + __func__); + s->end_transfer_fn_idx = 2; + } +} + const VMStateDescription vmstate_ide_drive = { .name = "ide_drive", - .version_id = 3, + .version_id = 4, .minimum_version_id = 0, .minimum_version_id_old = 0, + .pre_save = ide_drive_pre_save, .post_load = ide_drive_post_load, .fields = (VMStateField []) { VMSTATE_INT32(mult_sectors, IDEState), @@ -2714,7 +2762,14 @@ const VMStateDescription vmstate_ide_drive = { VMSTATE_UINT8(sense_key, IDEState), VMSTATE_UINT8(asc, IDEState), VMSTATE_UINT8_V(cdrom_changed, IDEState, 3), - /* XXX: if a transfer is pending, we do not save it yet */ + VMSTATE_INT32_V(req_nb_sectors, IDEState, 4), + VMSTATE_VARRAY_INT32(io_buffer, IDEState, io_buffer_total_len, 4, + vmstate_info_uint8, uint8_t), + VMSTATE_INT32_V(cur_io_buffer_offset, IDEState, 4), + VMSTATE_INT32_V(cur_io_buffer_len, IDEState, 4), + VMSTATE_UINT8_V(end_transfer_fn_idx, IDEState, 4), + VMSTATE_INT32_V(elementary_transfer_size, IDEState, 4), + VMSTATE_INT32_V(packet_transfer_size, IDEState, 4), VMSTATE_END_OF_LIST() } }; diff --git a/hw/ide/internal.h b/hw/ide/internal.h index 1cc4b55..8615d1a 100644 --- a/hw/ide/internal.h +++ b/hw/ide/internal.h @@ -417,6 +417,11 @@ struct IDEState { uint8_t *data_ptr; uint8_t *data_end; uint8_t *io_buffer; + /* PIO save/restore */ + int32_t io_buffer_total_len; + int cur_io_buffer_offset; + int cur_io_buffer_len; + uint8_t end_transfer_fn_idx; QEMUTimer *sector_write_timer; /* only used for win2k install hack */ uint32_t irq_count; /* counts IRQs when using win2k install hack */ /* CF-ATA extended error */
Save/restore information necessary to continue in progress PIO/ATAPI CMD transfers. This includes the IO buffer. Signed-off-by: Marcelo Tosatti <mtosatti@redhat.com>