From: Alexander Mikhalitsyn <[email protected]> Add tests for VMSTATE_VARRAY_OF_POINTER{,_TO_STRUCT}_UINT32_ALLOC.
Signed-off-by: Alexander Mikhalitsyn <[email protected]> --- tests/unit/test-vmstate.c | 157 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 157 insertions(+) diff --git a/tests/unit/test-vmstate.c b/tests/unit/test-vmstate.c index cadbab3c5e2..d10cf34fc75 100644 --- a/tests/unit/test-vmstate.c +++ b/tests/unit/test-vmstate.c @@ -702,6 +702,155 @@ static void test_arr_ptr_prim_0_load(void) } } +static uint8_t wire_arr_ptr_with_nulls[] = { + VMS_NOTNULLPTR_MARKER, + 0x00, 0x00, 0x00, 0x00, + VMS_NULLPTR_MARKER, + VMS_NOTNULLPTR_MARKER, + 0x00, 0x00, 0x00, 0x02, + VMS_NOTNULLPTR_MARKER, + 0x00, 0x00, 0x00, 0x03, + QEMU_VM_EOF +}; + +typedef struct { + uint32_t ar_items_num; + TestStructTriv **ar; +} TestVArrayOfPtrToStuctWithNULLs; + +const VMStateDescription vmsd_arps_with_nulls = { + .name = "test/arps_with_nulls", + .version_id = 1, + .minimum_version_id = 1, + .fields = (const VMStateField[]) { + VMSTATE_VARRAY_OF_POINTER_TO_STRUCT_UINT32_ALLOC( + ar, TestVArrayOfPtrToStuctWithNULLs, ar_items_num, 0, vmsd_tst, TestStructTriv), + VMSTATE_END_OF_LIST() + } +}; + +static void test_arr_ptr_nulls_str_save(void) +{ + TestStructTriv ar[AR_SIZE] = { {.i = 0}, {.i = 1}, {.i = 2}, {.i = 3} }; + TestVArrayOfPtrToStuctWithNULLs sample = {}; + int idx; + + sample.ar_items_num = AR_SIZE; + sample.ar = g_new0(TestStructTriv*, sample.ar_items_num); + sample.ar[0] = g_new0(TestStructTriv, 1); + *sample.ar[0] = ar[0]; + /* note, sample.ar[1] remains NULL */ + sample.ar[2] = g_new0(TestStructTriv, 1); + *sample.ar[2] = ar[2]; + sample.ar[3] = g_new0(TestStructTriv, 1); + *sample.ar[3] = ar[3]; + + save_vmstate(&vmsd_arps_with_nulls, &sample); + compare_vmstate(wire_arr_ptr_with_nulls, sizeof(wire_arr_ptr_with_nulls)); + + for (idx = 0; idx < AR_SIZE; ++idx) { + g_free(sample.ar[idx]); + } + g_free(sample.ar); +} + +static void test_arr_ptr_nulls_str_load(void) +{ + TestStructTriv ar_gt[AR_SIZE] = {{.i = 0}, {.i = 0}, {.i = 2}, {.i = 3} }; + TestVArrayOfPtrToStuctWithNULLs obj = {}; + int idx; + + obj.ar_items_num = AR_SIZE; + obj.ar = g_new0(TestStructTriv*, obj.ar_items_num); + + save_buffer(wire_arr_ptr_with_nulls, sizeof(wire_arr_ptr_with_nulls)); + SUCCESS(load_vmstate_one(&vmsd_arps_with_nulls, &obj, 1, + wire_arr_ptr_with_nulls, sizeof(wire_arr_ptr_with_nulls))); + + for (idx = 0; idx < AR_SIZE; ++idx) { + if (idx == 1) { + g_assert_cmpint((uintptr_t)(obj.ar[idx]), ==, 0); + } else { + /* compare the target array ar with the ground truth array ar_gt */ + g_assert_cmpint(ar_gt[idx].i, ==, obj.ar[idx]->i); + } + } + + for (idx = 0; idx < AR_SIZE; ++idx) { + g_free(obj.ar[idx]); + } + g_free(obj.ar); +} + +typedef struct { + uint32_t ar_items_num; + int32_t **ar; +} TestVArrayOfPtrToIntWithNULLs; + +const VMStateDescription vmsd_arpp_with_nulls = { + .name = "test/arpp_with_nulls", + .version_id = 1, + .minimum_version_id = 1, + .fields = (const VMStateField[]) { + VMSTATE_VARRAY_OF_POINTER_UINT32_ALLOC( + ar, TestVArrayOfPtrToIntWithNULLs, ar_items_num, 0, vmstate_info_int32, int32_t), + VMSTATE_END_OF_LIST() + } +}; + +static void test_arr_ptr_nulls_prim_save(void) +{ + int32_t ar[AR_SIZE] = { 0, 1, 2, 3 }; + TestVArrayOfPtrToIntWithNULLs sample = {}; + int idx; + + sample.ar_items_num = AR_SIZE; + sample.ar = g_new0(int32_t*, sample.ar_items_num); + sample.ar[0] = g_new0(int32_t, 1); + *sample.ar[0] = ar[0]; + /* note, sample.ar[1] remains NULL */ + sample.ar[2] = g_new0(int32_t, 1); + *sample.ar[2] = ar[2]; + sample.ar[3] = g_new0(int32_t, 1); + *sample.ar[3] = ar[3]; + + save_vmstate(&vmsd_arpp_with_nulls, &sample); + compare_vmstate(wire_arr_ptr_with_nulls, sizeof(wire_arr_ptr_with_nulls)); + + for (idx = 0; idx < AR_SIZE; ++idx) { + g_free(sample.ar[idx]); + } + g_free(sample.ar); +} + +static void test_arr_ptr_nulls_prim_load(void) +{ + int32_t ar_gt[AR_SIZE] = { 0, 0, 2, 3 }; + TestVArrayOfPtrToIntWithNULLs obj = {}; + int idx; + + obj.ar_items_num = AR_SIZE; + obj.ar = g_new0(int32_t*, obj.ar_items_num); + + save_buffer(wire_arr_ptr_with_nulls, sizeof(wire_arr_ptr_with_nulls)); + SUCCESS(load_vmstate_one(&vmsd_arpp_with_nulls, &obj, 1, + wire_arr_ptr_with_nulls, sizeof(wire_arr_ptr_with_nulls))); + + for (idx = 0; idx < AR_SIZE; ++idx) { + if (idx == 1) { + g_assert_cmpint((uintptr_t)(obj.ar[idx]), ==, 0); + } else { + /* compare the target array ar with the ground truth array ar_gt */ + g_assert_cmpint(ar_gt[idx], ==, *obj.ar[idx]); + } + } + + for (idx = 0; idx < AR_SIZE; ++idx) { + g_free(obj.ar[idx]); + } + g_free(obj.ar); +} + /* test QTAILQ migration */ typedef struct TestQtailqElement TestQtailqElement; @@ -1568,6 +1717,14 @@ int main(int argc, char **argv) test_arr_ptr_prim_0_save); g_test_add_func("/vmstate/array/ptr/prim/0/load", test_arr_ptr_prim_0_load); + g_test_add_func("/vmstate/array/ptr-nulls/str/save", + test_arr_ptr_nulls_str_save); + g_test_add_func("/vmstate/array/ptr-nulls/str/load", + test_arr_ptr_nulls_str_load); + g_test_add_func("/vmstate/array/ptr-nulls/prim/save", + test_arr_ptr_nulls_prim_save); + g_test_add_func("/vmstate/array/ptr-nulls/prim/load", + test_arr_ptr_nulls_prim_load); g_test_add_func("/vmstate/qtailq/save/saveq", test_save_q); g_test_add_func("/vmstate/qtailq/load/loadq", test_load_q); g_test_add_func("/vmstate/gtree/save/savedomain", test_gtree_save_domain); -- 2.47.3
