| 1 | /* |
| 2 | * String Input Visitor unit-tests. |
| 3 | * |
| 4 | * Copyright (C) 2012 Red Hat Inc. |
| 5 | * |
| 6 | * Authors: |
| 7 | * Paolo Bonzini <pbonzini@redhat.com> (based on test-qobject-input-visitor) |
| 8 | * |
| 9 | * This work is licensed under the terms of the GNU GPL, version 2 or later. |
| 10 | * See the COPYING file in the top-level directory. |
| 11 | */ |
| 12 | |
| 13 | #include "qemu/osdep.h" |
| 14 | |
| 15 | #include "qapi/error.h" |
| 16 | #include "qapi/string-input-visitor.h" |
| 17 | #include "test-qapi-visit.h" |
| 18 | |
| 19 | typedef struct TestInputVisitorData { |
| 20 | Visitor *v; |
| 21 | } TestInputVisitorData; |
| 22 | |
| 23 | static void visitor_input_teardown(TestInputVisitorData *data, |
| 24 | const void *unused) |
| 25 | { |
| 26 | if (data->v) { |
| 27 | visit_free(data->v); |
| 28 | data->v = NULL; |
| 29 | } |
| 30 | } |
| 31 | |
| 32 | /* This is provided instead of a test setup function so that the JSON |
| 33 | string used by the tests are kept in the test functions (and not |
| 34 | int main()) */ |
| 35 | static |
| 36 | Visitor *visitor_input_test_init(TestInputVisitorData *data, |
| 37 | const char *string) |
| 38 | { |
| 39 | visitor_input_teardown(data, NULL); |
| 40 | |
| 41 | data->v = string_input_visitor_new(string); |
| 42 | g_assert(data->v); |
| 43 | return data->v; |
| 44 | } |
| 45 | |
| 46 | static void test_visitor_in_int(TestInputVisitorData *data, |
| 47 | const void *unused) |
| 48 | { |
| 49 | int64_t res = 0, value = -42; |
| 50 | Error *err = NULL; |
| 51 | Visitor *v; |
| 52 | |
| 53 | v = visitor_input_test_init(data, "-42"); |
| 54 | |
| 55 | visit_type_int(v, NULL, &res, &error_abort); |
| 56 | g_assert_cmpint(res, ==, value); |
| 57 | |
| 58 | v = visitor_input_test_init(data, "not an int"); |
| 59 | |
| 60 | visit_type_int(v, NULL, &res, &err); |
| 61 | error_free_or_abort(&err); |
| 62 | |
| 63 | v = visitor_input_test_init(data, ""); |
| 64 | |
| 65 | visit_type_int(v, NULL, &res, &err); |
| 66 | error_free_or_abort(&err); |
| 67 | } |
| 68 | |
| 69 | static void check_ilist(Visitor *v, int64_t *expected, size_t n) |
| 70 | { |
| 71 | int64List *res = NULL; |
| 72 | int64List *tail; |
| 73 | int i; |
| 74 | |
| 75 | visit_type_int64List(v, NULL, &res, &error_abort); |
| 76 | tail = res; |
| 77 | for (i = 0; i < n; i++) { |
| 78 | g_assert(tail); |
| 79 | g_assert_cmpint(tail->value, ==, expected[i]); |
| 80 | tail = tail->next; |
| 81 | } |
| 82 | g_assert(!tail); |
| 83 | |
| 84 | qapi_free_int64List(res); |
| 85 | } |
| 86 | |
| 87 | static void check_ulist(Visitor *v, uint64_t *expected, size_t n) |
| 88 | { |
| 89 | uint64List *res = NULL; |
| 90 | uint64List *tail; |
| 91 | int i; |
| 92 | |
| 93 | visit_type_uint64List(v, NULL, &res, &error_abort); |
| 94 | tail = res; |
| 95 | for (i = 0; i < n; i++) { |
| 96 | g_assert(tail); |
| 97 | g_assert_cmpuint(tail->value, ==, expected[i]); |
| 98 | tail = tail->next; |
| 99 | } |
| 100 | g_assert(!tail); |
| 101 | |
| 102 | qapi_free_uint64List(res); |
| 103 | } |
| 104 | |
| 105 | static void test_visitor_in_intList(TestInputVisitorData *data, |
| 106 | const void *unused) |
| 107 | { |
| 108 | int64_t expect1[] = { 1, 2, 0, 2, 3, 4, 20, 5, 6, 7, |
| 109 | 8, 9, 1, 2, 3, 4, 5, 6, 7, 8 }; |
| 110 | int64_t expect2[] = { 32767, -32768, -32767 }; |
| 111 | int64_t expect3[] = { INT64_MIN, INT64_MAX }; |
| 112 | int64_t expect4[] = { 1 }; |
| 113 | int64_t expect5[] = { INT64_MAX - 2, INT64_MAX - 1, INT64_MAX }; |
| 114 | Error *err = NULL; |
| 115 | int64List *res = NULL; |
| 116 | Visitor *v; |
| 117 | int64_t val; |
| 118 | |
| 119 | /* Valid lists */ |
| 120 | |
| 121 | v = visitor_input_test_init(data, "1,2,0,2-4,20,5-9,1-8"); |
| 122 | check_ilist(v, expect1, ARRAY_SIZE(expect1)); |
| 123 | |
| 124 | v = visitor_input_test_init(data, "32767,-32768--32767"); |
| 125 | check_ilist(v, expect2, ARRAY_SIZE(expect2)); |
| 126 | |
| 127 | v = visitor_input_test_init(data, |
| 128 | "-9223372036854775808,9223372036854775807"); |
| 129 | check_ilist(v, expect3, ARRAY_SIZE(expect3)); |
| 130 | |
| 131 | v = visitor_input_test_init(data, "1-1"); |
| 132 | check_ilist(v, expect4, ARRAY_SIZE(expect4)); |
| 133 | |
| 134 | v = visitor_input_test_init(data, |
| 135 | "9223372036854775805-9223372036854775807"); |
| 136 | check_ilist(v, expect5, ARRAY_SIZE(expect5)); |
| 137 | |
| 138 | /* Value too large */ |
| 139 | |
| 140 | v = visitor_input_test_init(data, "9223372036854775808"); |
| 141 | visit_type_int64List(v, NULL, &res, &err); |
| 142 | error_free_or_abort(&err); |
| 143 | g_assert(!res); |
| 144 | |
| 145 | /* Value too small */ |
| 146 | |
| 147 | v = visitor_input_test_init(data, "-9223372036854775809"); |
| 148 | visit_type_int64List(v, NULL, &res, &err); |
| 149 | error_free_or_abort(&err); |
| 150 | g_assert(!res); |
| 151 | |
| 152 | /* Range not ascending */ |
| 153 | |
| 154 | v = visitor_input_test_init(data, "3-1"); |
| 155 | visit_type_int64List(v, NULL, &res, &err); |
| 156 | error_free_or_abort(&err); |
| 157 | g_assert(!res); |
| 158 | |
| 159 | v = visitor_input_test_init(data, "9223372036854775807-0"); |
| 160 | visit_type_int64List(v, NULL, &res, &err); |
| 161 | error_free_or_abort(&err); |
| 162 | g_assert(!res); |
| 163 | |
| 164 | /* Range too big (65536 is the limit against DOS attacks) */ |
| 165 | |
| 166 | v = visitor_input_test_init(data, "0-65536"); |
| 167 | visit_type_int64List(v, NULL, &res, &err); |
| 168 | error_free_or_abort(&err); |
| 169 | g_assert(!res); |
| 170 | |
| 171 | /* Empty list */ |
| 172 | |
| 173 | v = visitor_input_test_init(data, ""); |
| 174 | visit_type_int64List(v, NULL, &res, &error_abort); |
| 175 | g_assert(!res); |
| 176 | |
| 177 | /* Not a list */ |
| 178 | |
| 179 | v = visitor_input_test_init(data, "not an int list"); |
| 180 | |
| 181 | visit_type_int64List(v, NULL, &res, &err); |
| 182 | error_free_or_abort(&err); |
| 183 | g_assert(!res); |
| 184 | |
| 185 | /* Unvisited list tail */ |
| 186 | |
| 187 | v = visitor_input_test_init(data, "0,2-3"); |
| 188 | |
| 189 | visit_start_list(v, NULL, NULL, 0, &error_abort); |
| 190 | visit_type_int64(v, NULL, &val, &error_abort); |
| 191 | g_assert_cmpint(val, ==, 0); |
| 192 | visit_type_int64(v, NULL, &val, &error_abort); |
| 193 | g_assert_cmpint(val, ==, 2); |
| 194 | |
| 195 | visit_check_list(v, &err); |
| 196 | error_free_or_abort(&err); |
| 197 | visit_end_list(v, NULL); |
| 198 | |
| 199 | /* Visit beyond end of list */ |
| 200 | |
| 201 | v = visitor_input_test_init(data, "0"); |
| 202 | |
| 203 | visit_start_list(v, NULL, NULL, 0, &error_abort); |
| 204 | visit_type_int64(v, NULL, &val, &err); |
| 205 | g_assert_cmpint(val, ==, 0); |
| 206 | visit_type_int64(v, NULL, &val, &err); |
| 207 | error_free_or_abort(&err); |
| 208 | |
| 209 | visit_check_list(v, &error_abort); |
| 210 | visit_end_list(v, NULL); |
| 211 | } |
| 212 | |
| 213 | static void test_visitor_in_uintList(TestInputVisitorData *data, |
| 214 | const void *unused) |
| 215 | { |
| 216 | uint64_t expect1[] = { 1, 2, 0, 2, 3, 4, 20, 5, 6, 7, |
| 217 | 8, 9, 1, 2, 3, 4, 5, 6, 7, 8 }; |
| 218 | uint64_t expect2[] = { 32767, -32768, -32767 }; |
| 219 | uint64_t expect3[] = { INT64_MIN, INT64_MAX }; |
| 220 | uint64_t expect4[] = { 1 }; |
| 221 | uint64_t expect5[] = { UINT64_MAX }; |
| 222 | uint64_t expect6[] = { UINT64_MAX - 2, UINT64_MAX - 1, UINT64_MAX }; |
| 223 | Error *err = NULL; |
| 224 | uint64List *res = NULL; |
| 225 | Visitor *v; |
| 226 | uint64_t val; |
| 227 | |
| 228 | /* Valid lists */ |
| 229 | |
| 230 | v = visitor_input_test_init(data, "1,2,0,2-4,20,5-9,1-8"); |
| 231 | check_ulist(v, expect1, ARRAY_SIZE(expect1)); |
| 232 | |
| 233 | v = visitor_input_test_init(data, "32767,-32768--32767"); |
| 234 | check_ulist(v, expect2, ARRAY_SIZE(expect2)); |
| 235 | |
| 236 | v = visitor_input_test_init(data, |
| 237 | "-9223372036854775808,9223372036854775807"); |
| 238 | check_ulist(v, expect3, ARRAY_SIZE(expect3)); |
| 239 | |
| 240 | v = visitor_input_test_init(data, "1-1"); |
| 241 | check_ulist(v, expect4, ARRAY_SIZE(expect4)); |
| 242 | |
| 243 | v = visitor_input_test_init(data, "18446744073709551615"); |
| 244 | check_ulist(v, expect5, ARRAY_SIZE(expect5)); |
| 245 | |
| 246 | v = visitor_input_test_init(data, |
| 247 | "18446744073709551613-18446744073709551615"); |
| 248 | check_ulist(v, expect6, ARRAY_SIZE(expect6)); |
| 249 | |
| 250 | /* Value too large */ |
| 251 | |
| 252 | v = visitor_input_test_init(data, "18446744073709551616"); |
| 253 | visit_type_uint64List(v, NULL, &res, &err); |
| 254 | error_free_or_abort(&err); |
| 255 | g_assert(!res); |
| 256 | |
| 257 | /* Value too small */ |
| 258 | |
| 259 | v = visitor_input_test_init(data, "-18446744073709551616"); |
| 260 | visit_type_uint64List(v, NULL, &res, &err); |
| 261 | error_free_or_abort(&err); |
| 262 | g_assert(!res); |
| 263 | |
| 264 | /* Range not ascending */ |
| 265 | |
| 266 | v = visitor_input_test_init(data, "3-1"); |
| 267 | visit_type_uint64List(v, NULL, &res, &err); |
| 268 | error_free_or_abort(&err); |
| 269 | g_assert(!res); |
| 270 | |
| 271 | v = visitor_input_test_init(data, "18446744073709551615-0"); |
| 272 | visit_type_uint64List(v, NULL, &res, &err); |
| 273 | error_free_or_abort(&err); |
| 274 | g_assert(!res); |
| 275 | |
| 276 | /* Range too big (65536 is the limit against DOS attacks) */ |
| 277 | |
| 278 | v = visitor_input_test_init(data, "0-65536"); |
| 279 | visit_type_uint64List(v, NULL, &res, &err); |
| 280 | error_free_or_abort(&err); |
| 281 | g_assert(!res); |
| 282 | |
| 283 | /* Empty list */ |
| 284 | |
| 285 | v = visitor_input_test_init(data, ""); |
| 286 | visit_type_uint64List(v, NULL, &res, &error_abort); |
| 287 | g_assert(!res); |
| 288 | |
| 289 | /* Not a list */ |
| 290 | |
| 291 | v = visitor_input_test_init(data, "not an uint list"); |
| 292 | |
| 293 | visit_type_uint64List(v, NULL, &res, &err); |
| 294 | error_free_or_abort(&err); |
| 295 | g_assert(!res); |
| 296 | |
| 297 | /* Unvisited list tail */ |
| 298 | |
| 299 | v = visitor_input_test_init(data, "0,2-3"); |
| 300 | |
| 301 | visit_start_list(v, NULL, NULL, 0, &error_abort); |
| 302 | visit_type_uint64(v, NULL, &val, &error_abort); |
| 303 | g_assert_cmpuint(val, ==, 0); |
| 304 | visit_type_uint64(v, NULL, &val, &error_abort); |
| 305 | g_assert_cmpuint(val, ==, 2); |
| 306 | |
| 307 | visit_check_list(v, &err); |
| 308 | error_free_or_abort(&err); |
| 309 | visit_end_list(v, NULL); |
| 310 | |
| 311 | /* Visit beyond end of list */ |
| 312 | |
| 313 | v = visitor_input_test_init(data, "0"); |
| 314 | |
| 315 | visit_start_list(v, NULL, NULL, 0, &error_abort); |
| 316 | visit_type_uint64(v, NULL, &val, &err); |
| 317 | g_assert_cmpuint(val, ==, 0); |
| 318 | visit_type_uint64(v, NULL, &val, &err); |
| 319 | error_free_or_abort(&err); |
| 320 | |
| 321 | visit_check_list(v, &error_abort); |
| 322 | visit_end_list(v, NULL); |
| 323 | } |
| 324 | |
| 325 | static void test_visitor_in_bool(TestInputVisitorData *data, |
| 326 | const void *unused) |
| 327 | { |
| 328 | bool res = false; |
| 329 | Visitor *v; |
| 330 | |
| 331 | v = visitor_input_test_init(data, "true"); |
| 332 | |
| 333 | visit_type_bool(v, NULL, &res, &error_abort); |
| 334 | g_assert_cmpint(res, ==, true); |
| 335 | |
| 336 | v = visitor_input_test_init(data, "yes"); |
| 337 | |
| 338 | visit_type_bool(v, NULL, &res, &error_abort); |
| 339 | g_assert_cmpint(res, ==, true); |
| 340 | |
| 341 | v = visitor_input_test_init(data, "on"); |
| 342 | |
| 343 | visit_type_bool(v, NULL, &res, &error_abort); |
| 344 | g_assert_cmpint(res, ==, true); |
| 345 | |
| 346 | v = visitor_input_test_init(data, "false"); |
| 347 | |
| 348 | visit_type_bool(v, NULL, &res, &error_abort); |
| 349 | g_assert_cmpint(res, ==, false); |
| 350 | |
| 351 | v = visitor_input_test_init(data, "no"); |
| 352 | |
| 353 | visit_type_bool(v, NULL, &res, &error_abort); |
| 354 | g_assert_cmpint(res, ==, false); |
| 355 | |
| 356 | v = visitor_input_test_init(data, "off"); |
| 357 | |
| 358 | visit_type_bool(v, NULL, &res, &error_abort); |
| 359 | g_assert_cmpint(res, ==, false); |
| 360 | } |
| 361 | |
| 362 | static void test_visitor_in_number(TestInputVisitorData *data, |
| 363 | const void *unused) |
| 364 | { |
| 365 | double res = 0, value = 3.14; |
| 366 | Error *err = NULL; |
| 367 | Visitor *v; |
| 368 | |
| 369 | v = visitor_input_test_init(data, "3.14"); |
| 370 | |
| 371 | visit_type_number(v, NULL, &res, &error_abort); |
| 372 | g_assert_cmpfloat(res, ==, value); |
| 373 | |
| 374 | /* NaN and infinity has to be rejected */ |
| 375 | |
| 376 | v = visitor_input_test_init(data, "NaN"); |
| 377 | |
| 378 | visit_type_number(v, NULL, &res, &err); |
| 379 | error_free_or_abort(&err); |
| 380 | |
| 381 | v = visitor_input_test_init(data, "inf"); |
| 382 | |
| 383 | visit_type_number(v, NULL, &res, &err); |
| 384 | error_free_or_abort(&err); |
| 385 | |
| 386 | } |
| 387 | |
| 388 | static void test_visitor_in_string(TestInputVisitorData *data, |
| 389 | const void *unused) |
| 390 | { |
| 391 | char *res = NULL, *value = (char *) "Q E M U"; |
| 392 | Visitor *v; |
| 393 | |
| 394 | v = visitor_input_test_init(data, value); |
| 395 | |
| 396 | visit_type_str(v, NULL, &res, &error_abort); |
| 397 | g_assert_cmpstr(res, ==, value); |
| 398 | |
| 399 | g_free(res); |
| 400 | } |
| 401 | |
| 402 | static void test_visitor_in_enum(TestInputVisitorData *data, |
| 403 | const void *unused) |
| 404 | { |
| 405 | Visitor *v; |
| 406 | EnumOne i; |
| 407 | |
| 408 | for (i = 0; i < ENUM_ONE__MAX; i++) { |
| 409 | EnumOne res = -1; |
| 410 | |
| 411 | v = visitor_input_test_init(data, EnumOne_str(i)); |
| 412 | |
| 413 | visit_type_EnumOne(v, NULL, &res, &error_abort); |
| 414 | g_assert_cmpint(i, ==, res); |
| 415 | } |
| 416 | } |
| 417 | |
| 418 | /* Try to crash the visitors */ |
| 419 | static void test_visitor_in_fuzz(TestInputVisitorData *data, |
| 420 | const void *unused) |
| 421 | { |
| 422 | int64_t ires; |
| 423 | intList *ilres; |
| 424 | bool bres; |
| 425 | double nres; |
| 426 | char *sres; |
| 427 | EnumOne eres; |
| 428 | Visitor *v; |
| 429 | unsigned int i; |
| 430 | char buf[10000]; |
| 431 | |
| 432 | for (i = 0; i < 100; i++) { |
| 433 | unsigned int j, k; |
| 434 | |
| 435 | j = g_test_rand_int_range(0, sizeof(buf) - 1); |
| 436 | |
| 437 | buf[j] = '\0'; |
| 438 | |
| 439 | for (k = 0; k != j; k++) { |
| 440 | buf[k] = (char)g_test_rand_int_range(0, 256); |
| 441 | } |
| 442 | |
| 443 | v = visitor_input_test_init(data, buf); |
| 444 | visit_type_int(v, NULL, &ires, NULL); |
| 445 | |
| 446 | v = visitor_input_test_init(data, buf); |
| 447 | visit_type_intList(v, NULL, &ilres, NULL); |
| 448 | qapi_free_intList(ilres); |
| 449 | |
| 450 | v = visitor_input_test_init(data, buf); |
| 451 | visit_type_bool(v, NULL, &bres, NULL); |
| 452 | |
| 453 | v = visitor_input_test_init(data, buf); |
| 454 | visit_type_number(v, NULL, &nres, NULL); |
| 455 | |
| 456 | v = visitor_input_test_init(data, buf); |
| 457 | sres = NULL; |
| 458 | visit_type_str(v, NULL, &sres, NULL); |
| 459 | g_free(sres); |
| 460 | |
| 461 | v = visitor_input_test_init(data, buf); |
| 462 | visit_type_EnumOne(v, NULL, &eres, NULL); |
| 463 | } |
| 464 | } |
| 465 | |
| 466 | static void input_visitor_test_add(const char *testpath, |
| 467 | TestInputVisitorData *data, |
| 468 | void (*test_func)(TestInputVisitorData *data, const void *user_data)) |
| 469 | { |
| 470 | g_test_add(testpath, TestInputVisitorData, data, NULL, test_func, |
| 471 | visitor_input_teardown); |
| 472 | } |
| 473 | |
| 474 | int main(int argc, char **argv) |
| 475 | { |
| 476 | TestInputVisitorData in_visitor_data; |
| 477 | |
| 478 | g_test_init(&argc, &argv, NULL); |
| 479 | |
| 480 | input_visitor_test_add("/string-visitor/input/int", |
| 481 | &in_visitor_data, test_visitor_in_int); |
| 482 | input_visitor_test_add("/string-visitor/input/intList", |
| 483 | &in_visitor_data, test_visitor_in_intList); |
| 484 | input_visitor_test_add("/string-visitor/input/uintList", |
| 485 | &in_visitor_data, test_visitor_in_uintList); |
| 486 | input_visitor_test_add("/string-visitor/input/bool", |
| 487 | &in_visitor_data, test_visitor_in_bool); |
| 488 | input_visitor_test_add("/string-visitor/input/number", |
| 489 | &in_visitor_data, test_visitor_in_number); |
| 490 | input_visitor_test_add("/string-visitor/input/string", |
| 491 | &in_visitor_data, test_visitor_in_string); |
| 492 | input_visitor_test_add("/string-visitor/input/enum", |
| 493 | &in_visitor_data, test_visitor_in_enum); |
| 494 | input_visitor_test_add("/string-visitor/input/fuzz", |
| 495 | &in_visitor_data, test_visitor_in_fuzz); |
| 496 | |
| 497 | g_test_run(); |
| 498 | |
| 499 | return 0; |
| 500 | } |