[Qemu-devel] [PATCH 098/124] vmstate: Test for VMSTATE_PARTIAL_VBUFFER

[Juan Quintela]

Signed-off-by: Juan Quintela <address@hidden>
---
 tests/test-vmstate.c | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)

diff --git a/tests/test-vmstate.c b/tests/test-vmstate.c
index c16b0f0..31acb15 100644
--- a/tests/test-vmstate.c
+++ b/tests/test-vmstate.c
@@ -971,6 +971,80 @@ static void test_buffer_test(void)
     FAILURE(memcmp(obj.partial, obj_buffer.partial, sizeof(obj.partial)));
 }

+typedef struct TestVBuffer {
+    int32_t size;
+    uint8_t  *buffer;
+} TestVBuffer;
+
+static const VMStateDescription vmstate_vbuffer_simple = {
+    .name = "vbuffer/simple",
+    .version_id = 1,
+    .minimum_version_id = 1,
+    .minimum_version_id_old = 1,
+    .fields = (VMStateField[]) {
+        VMSTATE_INT32_EQUAL(size, TestVBuffer),
+        VMSTATE_PARTIAL_VBUFFER(buffer, TestVBuffer, size),
+        VMSTATE_END_OF_LIST()
+    }
+};
+
+uint8_t wire_vbuffer_simple[] = {
+    /* size */    0x00, 0x00, 0x00, 0x05,
+    /* buffer */  0x01, 0x02, 0x03, 0x04, 0x05,
+    QEMU_VM_EOF, /* just to ensure we won't get EOF reported prematurely */
+};
+
+static void obj_vbuffer_copy(void *arg1, void *arg2)
+{
+    TestVBuffer *target = arg1;
+    TestVBuffer *source = arg2;
+    int i;
+
+    target->size = source->size;
+    for (i = 0; i < VMSTATE_ARRAY_SIZE; i++) {
+        target->buffer[i] = source->buffer[i];
+    }
+}
+
+static TestVBuffer *create_vbuffer(void)
+{
+    TestVBuffer *obj = g_malloc0(sizeof(*obj));
+    obj->size = VMSTATE_ARRAY_SIZE;
+    obj->buffer = g_malloc0(VMSTATE_ARRAY_SIZE);
+
+    return obj;
+}
+
+static TestVBuffer *create_vbuffer_init(void)
+{
+    TestVBuffer *obj = create_vbuffer();
+    int i;
+
+    for (i = 0; i < VMSTATE_ARRAY_SIZE; i++) {
+        obj->buffer[i] = i + 1;
+    }
+    return obj;
+}
+
+static void test_vbuffer_simple(void)
+{
+    TestVBuffer *obj, *obj_clone, *obj_vbuffer;
+
+    obj_vbuffer = create_vbuffer_init();
+    obj = create_vbuffer();
+    obj_clone = create_vbuffer();
+
+    save_vmstate(&vmstate_vbuffer_simple, obj_vbuffer);
+
+    compare_vmstate(wire_vbuffer_simple, sizeof(wire_vbuffer_simple));
+
+    SUCCESS(load_vmstate(&vmstate_vbuffer_simple, obj, obj_clone,
+                         obj_vbuffer_copy, 1, wire_vbuffer_simple,
+                         sizeof(wire_vbuffer_simple)));
+
+    SUCCESS(memcmp(obj->buffer, obj_vbuffer->buffer, VMSTATE_ARRAY_SIZE));
+}
+
 typedef struct TestVersioned {
     uint32_t a, b, c, e;
     uint64_t d, f;
@@ -1209,6 +1283,7 @@ int main(int argc, char **argv)
     g_test_add_func("/vmstate/array/sub", test_array_sub);
     g_test_add_func("/vmstate/buffer/simple", test_buffer_simple);
     g_test_add_func("/vmstate/buffer/test", test_buffer_test);
+    g_test_add_func("/vmstate/vbuffer/simple", test_vbuffer_simple);
     g_test_run();

     close(temp_fd);
-- 
1.9.0