vmstate: Add pre/post_save() hooks

Signed-off-by: Juan Quintela <quintela@redhat.com>
Signed-off-by: Anthony Liguori <aliguori@us.ibm.com>

2 files changed, 8 insertions, 0 deletions

diff --git a/hw/hw.h b/hw/hw.h
index 8e75a04..d64a0e8 100644
--- a/hw/hw.h
+++ b/hw/hw.h
@@ -308,6 +308,8 @@ struct VMStateDescription {
     LoadStateHandler *load_state_old;
     int (*pre_load)(void *opaque);
     int (*post_load)(void *opaque);
+    void (*pre_save)(const void *opaque);
+    void (*post_save)(const void *opaque);
     VMStateField *fields;
 };
 
diff --git a/savevm.c b/savevm.c
index 8ae8d16..c4bee37 100644
--- a/savevm.c
+++ b/savevm.c
@@ -1073,6 +1073,9 @@ void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd,
 {
     VMStateField *field = vmsd->fields;
 
+    if (vmsd->pre_save) {
+        vmsd->pre_save(opaque);
+    }
     while(field->name) {
         const void *base_addr = opaque + field->offset;
         int i, n_elems = 1;
@@ -1096,6 +1099,9 @@ void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd,
         }
         field++;
     }
+    if (vmsd->post_save) {
+        vmsd->post_save(opaque);
+    }
 }
 
 static int vmstate_load(QEMUFile *f, SaveStateEntry *se, int version_id)