Signed-off-by: Juan Quintela <quint...@redhat.com> --- 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