/* * String Input Visitor unit-tests. * * Copyright (C) 2012 Red Hat Inc. * * Authors: * Paolo Bonzini <pbonzini@redhat.com> (based on test-qobject-input-visitor) * * This work is licensed under the terms of the GNU GPL, version 2 or later. * See the COPYING file in the top-level directory. */ #include "qemu/osdep.h" #include "qemu-common.h" #include "qapi/error.h" #include "qapi/string-input-visitor.h" #include "test-qapi-visit.h" typedef struct TestInputVisitorData { Visitor *v; } TestInputVisitorData; static void visitor_input_teardown(TestInputVisitorData *data, const void *unused) { if (data->v) { visit_free(data->v); data->v = NULL; } } /* This is provided instead of a test setup function so that the JSON string used by the tests are kept in the test functions (and not int main()) */ static Visitor *visitor_input_test_init(TestInputVisitorData *data, const char *string) { visitor_input_teardown(data, NULL); data->v = string_input_visitor_new(string); g_assert(data->v); return data->v; } static void test_visitor_in_int(TestInputVisitorData *data, const void *unused) { int64_t res = 0, value = -42; Error *err = NULL; Visitor *v; v = visitor_input_test_init(data, "-42"); visit_type_int(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, value); v = visitor_input_test_init(data, "not an int"); visit_type_int(v, NULL, &res, &err); error_free_or_abort(&err); v = visitor_input_test_init(data, ""); visit_type_int(v, NULL, &res, &err); error_free_or_abort(&err); } static void check_ilist(Visitor *v, int64_t *expected, size_t n) { int64List *res = NULL; int64List *tail; int i; visit_type_int64List(v, NULL, &res, &error_abort); tail = res; for (i = 0; i < n; i++) { g_assert(tail); g_assert_cmpint(tail->value, ==, expected[i]); tail = tail->next; } g_assert(!tail); qapi_free_int64List(res); } static void check_ulist(Visitor *v, uint64_t *expected, size_t n) { uint64List *res = NULL; uint64List *tail; int i; visit_type_uint64List(v, NULL, &res, &error_abort); tail = res; for (i = 0; i < n; i++) { g_assert(tail); g_assert_cmpuint(tail->value, ==, expected[i]); tail = tail->next; } g_assert(!tail); qapi_free_uint64List(res); } static void test_visitor_in_intList(TestInputVisitorData *data, const void *unused) { int64_t expect1[] = { 1, 2, 0, 2, 3, 4, 20, 5, 6, 7, 8, 9, 1, 2, 3, 4, 5, 6, 7, 8 }; int64_t expect2[] = { 32767, -32768, -32767 }; int64_t expect3[] = { INT64_MIN, INT64_MAX }; int64_t expect4[] = { 1 }; int64_t expect5[] = { INT64_MAX - 2, INT64_MAX - 1, INT64_MAX }; Error *err = NULL; int64List *res = NULL; Visitor *v; int64_t val; /* Valid lists */ v = visitor_input_test_init(data, "1,2,0,2-4,20,5-9,1-8"); check_ilist(v, expect1, ARRAY_SIZE(expect1)); v = visitor_input_test_init(data, "32767,-32768--32767"); check_ilist(v, expect2, ARRAY_SIZE(expect2)); v = visitor_input_test_init(data, "-9223372036854775808,9223372036854775807"); check_ilist(v, expect3, ARRAY_SIZE(expect3)); v = visitor_input_test_init(data, "1-1"); check_ilist(v, expect4, ARRAY_SIZE(expect4)); v = visitor_input_test_init(data, "9223372036854775805-9223372036854775807"); check_ilist(v, expect5, ARRAY_SIZE(expect5)); /* Value too large */ v = visitor_input_test_init(data, "9223372036854775808"); visit_type_int64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Value too small */ v = visitor_input_test_init(data, "-9223372036854775809"); visit_type_int64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Range not ascending */ v = visitor_input_test_init(data, "3-1"); visit_type_int64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); v = visitor_input_test_init(data, "9223372036854775807-0"); visit_type_int64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Range too big (65536 is the limit against DOS attacks) */ v = visitor_input_test_init(data, "0-65536"); visit_type_int64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Empty list */ v = visitor_input_test_init(data, ""); visit_type_int64List(v, NULL, &res, &error_abort); g_assert(!res); /* Not a list */ v = visitor_input_test_init(data, "not an int list"); visit_type_int64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Unvisited list tail */ v = visitor_input_test_init(data, "0,2-3"); visit_start_list(v, NULL, NULL, 0, &error_abort); visit_type_int64(v, NULL, &val, &error_abort); g_assert_cmpint(val, ==, 0); visit_type_int64(v, NULL, &val, &error_abort); g_assert_cmpint(val, ==, 2); visit_check_list(v, &err); error_free_or_abort(&err); visit_end_list(v, NULL); /* Visit beyond end of list */ v = visitor_input_test_init(data, "0"); visit_start_list(v, NULL, NULL, 0, &error_abort); visit_type_int64(v, NULL, &val, &err); g_assert_cmpint(val, ==, 0); visit_type_int64(v, NULL, &val, &err); error_free_or_abort(&err); visit_check_list(v, &error_abort); visit_end_list(v, NULL); } static void test_visitor_in_uintList(TestInputVisitorData *data, const void *unused) { uint64_t expect1[] = { 1, 2, 0, 2, 3, 4, 20, 5, 6, 7, 8, 9, 1, 2, 3, 4, 5, 6, 7, 8 }; uint64_t expect2[] = { 32767, -32768, -32767 }; uint64_t expect3[] = { INT64_MIN, INT64_MAX }; uint64_t expect4[] = { 1 }; uint64_t expect5[] = { UINT64_MAX }; uint64_t expect6[] = { UINT64_MAX - 2, UINT64_MAX - 1, UINT64_MAX }; Error *err = NULL; uint64List *res = NULL; Visitor *v; uint64_t val; /* Valid lists */ v = visitor_input_test_init(data, "1,2,0,2-4,20,5-9,1-8"); check_ulist(v, expect1, ARRAY_SIZE(expect1)); v = visitor_input_test_init(data, "32767,-32768--32767"); check_ulist(v, expect2, ARRAY_SIZE(expect2)); v = visitor_input_test_init(data, "-9223372036854775808,9223372036854775807"); check_ulist(v, expect3, ARRAY_SIZE(expect3)); v = visitor_input_test_init(data, "1-1"); check_ulist(v, expect4, ARRAY_SIZE(expect4)); v = visitor_input_test_init(data, "18446744073709551615"); check_ulist(v, expect5, ARRAY_SIZE(expect5)); v = visitor_input_test_init(data, "18446744073709551613-18446744073709551615"); check_ulist(v, expect6, ARRAY_SIZE(expect6)); /* Value too large */ v = visitor_input_test_init(data, "18446744073709551616"); visit_type_uint64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Value too small */ v = visitor_input_test_init(data, "-18446744073709551616"); visit_type_uint64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Range not ascending */ v = visitor_input_test_init(data, "3-1"); visit_type_uint64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); v = visitor_input_test_init(data, "18446744073709551615-0"); visit_type_uint64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Range too big (65536 is the limit against DOS attacks) */ v = visitor_input_test_init(data, "0-65536"); visit_type_uint64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Empty list */ v = visitor_input_test_init(data, ""); visit_type_uint64List(v, NULL, &res, &error_abort); g_assert(!res); /* Not a list */ v = visitor_input_test_init(data, "not an uint list"); visit_type_uint64List(v, NULL, &res, &err); error_free_or_abort(&err); g_assert(!res); /* Unvisited list tail */ v = visitor_input_test_init(data, "0,2-3"); visit_start_list(v, NULL, NULL, 0, &error_abort); visit_type_uint64(v, NULL, &val, &error_abort); g_assert_cmpuint(val, ==, 0); visit_type_uint64(v, NULL, &val, &error_abort); g_assert_cmpuint(val, ==, 2); visit_check_list(v, &err); error_free_or_abort(&err); visit_end_list(v, NULL); /* Visit beyond end of list */ v = visitor_input_test_init(data, "0"); visit_start_list(v, NULL, NULL, 0, &error_abort); visit_type_uint64(v, NULL, &val, &err); g_assert_cmpuint(val, ==, 0); visit_type_uint64(v, NULL, &val, &err); error_free_or_abort(&err); visit_check_list(v, &error_abort); visit_end_list(v, NULL); } static void test_visitor_in_bool(TestInputVisitorData *data, const void *unused) { Error *err = NULL; bool res = false; Visitor *v; v = visitor_input_test_init(data, "true"); visit_type_bool(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, true); v = visitor_input_test_init(data, "yes"); visit_type_bool(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, true); v = visitor_input_test_init(data, "on"); visit_type_bool(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, true); v = visitor_input_test_init(data, "false"); visit_type_bool(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, false); v = visitor_input_test_init(data, "no"); visit_type_bool(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, false); v = visitor_input_test_init(data, "off"); visit_type_bool(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(res, ==, false); } static void test_visitor_in_number(TestInputVisitorData *data, const void *unused) { double res = 0, value = 3.14; Error *err = NULL; Visitor *v; v = visitor_input_test_init(data, "3.14"); visit_type_number(v, NULL, &res, &err); g_assert(!err); g_assert_cmpfloat(res, ==, value); /* NaN and infinity has to be rejected */ v = visitor_input_test_init(data, "NaN"); visit_type_number(v, NULL, &res, &err); error_free_or_abort(&err); v = visitor_input_test_init(data, "inf"); visit_type_number(v, NULL, &res, &err); error_free_or_abort(&err); } static void test_visitor_in_string(TestInputVisitorData *data, const void *unused) { char *res = NULL, *value = (char *) "Q E M U"; Error *err = NULL; Visitor *v; v = visitor_input_test_init(data, value); visit_type_str(v, NULL, &res, &err); g_assert(!err); g_assert_cmpstr(res, ==, value); g_free(res); } static void test_visitor_in_enum(TestInputVisitorData *data, const void *unused) { Error *err = NULL; Visitor *v; EnumOne i; for (i = 0; i < ENUM_ONE__MAX; i++) { EnumOne res = -1; v = visitor_input_test_init(data, EnumOne_str(i)); visit_type_EnumOne(v, NULL, &res, &err); g_assert(!err); g_assert_cmpint(i, ==, res); } } /* Try to crash the visitors */ static void test_visitor_in_fuzz(TestInputVisitorData *data, const void *unused) { int64_t ires; intList *ilres; bool bres; double nres; char *sres; EnumOne eres; Visitor *v; unsigned int i; char buf[10000]; for (i = 0; i < 100; i++) { unsigned int j, k; j = g_test_rand_int_range(0, sizeof(buf) - 1); buf[j] = '\0'; for (k = 0; k != j; k++) { buf[k] = (char)g_test_rand_int_range(0, 256); } v = visitor_input_test_init(data, buf); visit_type_int(v, NULL, &ires, NULL); v = visitor_input_test_init(data, buf); visit_type_intList(v, NULL, &ilres, NULL); qapi_free_intList(ilres); v = visitor_input_test_init(data, buf); visit_type_bool(v, NULL, &bres, NULL); v = visitor_input_test_init(data, buf); visit_type_number(v, NULL, &nres, NULL); v = visitor_input_test_init(data, buf); sres = NULL; visit_type_str(v, NULL, &sres, NULL); g_free(sres); v = visitor_input_test_init(data, buf); visit_type_EnumOne(v, NULL, &eres, NULL); } } static void input_visitor_test_add(const char *testpath, TestInputVisitorData *data, void (*test_func)(TestInputVisitorData *data, const void *user_data)) { g_test_add(testpath, TestInputVisitorData, data, NULL, test_func, visitor_input_teardown); } int main(int argc, char **argv) { TestInputVisitorData in_visitor_data; g_test_init(&argc, &argv, NULL); input_visitor_test_add("/string-visitor/input/int", &in_visitor_data, test_visitor_in_int); input_visitor_test_add("/string-visitor/input/intList", &in_visitor_data, test_visitor_in_intList); input_visitor_test_add("/string-visitor/input/uintList", &in_visitor_data, test_visitor_in_uintList); input_visitor_test_add("/string-visitor/input/bool", &in_visitor_data, test_visitor_in_bool); input_visitor_test_add("/string-visitor/input/number", &in_visitor_data, test_visitor_in_number); input_visitor_test_add("/string-visitor/input/string", &in_visitor_data, test_visitor_in_string); input_visitor_test_add("/string-visitor/input/enum", &in_visitor_data, test_visitor_in_enum); input_visitor_test_add("/string-visitor/input/fuzz", &in_visitor_data, test_visitor_in_fuzz); g_test_run(); return 0; }