esp: check dma length before reading scsi command(CVE-2016-4441)
[qemu/ar7.git] / tests / check-qdict.c
bloba43056c5dee7fcec56ade1ac8f7fe2aaddb32c6f
1 /*
2 * QDict unit-tests.
4 * Copyright (C) 2009 Red Hat Inc.
6 * Authors:
7 * Luiz Capitulino <lcapitulino@redhat.com>
9 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
10 * See the COPYING.LIB file in the top-level directory.
12 #include "qemu/osdep.h"
13 #include <glib.h>
15 #include "qapi/qmp/qint.h"
16 #include "qapi/qmp/qdict.h"
17 #include "qapi/qmp/qstring.h"
18 #include "qemu-common.h"
21 * Public Interface test-cases
23 * (with some violations to access 'private' data)
26 static void qdict_new_test(void)
28 QDict *qdict;
30 qdict = qdict_new();
31 g_assert(qdict != NULL);
32 g_assert(qdict_size(qdict) == 0);
33 g_assert(qdict->base.refcnt == 1);
34 g_assert(qobject_type(QOBJECT(qdict)) == QTYPE_QDICT);
36 // destroy doesn't exit yet
37 g_free(qdict);
40 static void qdict_put_obj_test(void)
42 QInt *qi;
43 QDict *qdict;
44 QDictEntry *ent;
45 const int num = 42;
47 qdict = qdict_new();
49 // key "" will have tdb hash 12345
50 qdict_put_obj(qdict, "", QOBJECT(qint_from_int(num)));
52 g_assert(qdict_size(qdict) == 1);
53 ent = QLIST_FIRST(&qdict->table[12345 % QDICT_BUCKET_MAX]);
54 qi = qobject_to_qint(ent->value);
55 g_assert(qint_get_int(qi) == num);
57 // destroy doesn't exit yet
58 QDECREF(qi);
59 g_free(ent->key);
60 g_free(ent);
61 g_free(qdict);
64 static void qdict_destroy_simple_test(void)
66 QDict *qdict;
68 qdict = qdict_new();
69 qdict_put_obj(qdict, "num", QOBJECT(qint_from_int(0)));
70 qdict_put_obj(qdict, "str", QOBJECT(qstring_from_str("foo")));
72 QDECREF(qdict);
75 static void qdict_get_test(void)
77 QInt *qi;
78 QObject *obj;
79 const int value = -42;
80 const char *key = "test";
81 QDict *tests_dict = qdict_new();
83 qdict_put(tests_dict, key, qint_from_int(value));
85 obj = qdict_get(tests_dict, key);
86 g_assert(obj != NULL);
88 qi = qobject_to_qint(obj);
89 g_assert(qint_get_int(qi) == value);
91 QDECREF(tests_dict);
94 static void qdict_get_int_test(void)
96 int ret;
97 const int value = 100;
98 const char *key = "int";
99 QDict *tests_dict = qdict_new();
101 qdict_put(tests_dict, key, qint_from_int(value));
103 ret = qdict_get_int(tests_dict, key);
104 g_assert(ret == value);
106 QDECREF(tests_dict);
109 static void qdict_get_try_int_test(void)
111 int ret;
112 const int value = 100;
113 const char *key = "int";
114 QDict *tests_dict = qdict_new();
116 qdict_put(tests_dict, key, qint_from_int(value));
118 ret = qdict_get_try_int(tests_dict, key, 0);
119 g_assert(ret == value);
121 QDECREF(tests_dict);
124 static void qdict_get_str_test(void)
126 const char *p;
127 const char *key = "key";
128 const char *str = "string";
129 QDict *tests_dict = qdict_new();
131 qdict_put(tests_dict, key, qstring_from_str(str));
133 p = qdict_get_str(tests_dict, key);
134 g_assert(p != NULL);
135 g_assert(strcmp(p, str) == 0);
137 QDECREF(tests_dict);
140 static void qdict_get_try_str_test(void)
142 const char *p;
143 const char *key = "key";
144 const char *str = "string";
145 QDict *tests_dict = qdict_new();
147 qdict_put(tests_dict, key, qstring_from_str(str));
149 p = qdict_get_try_str(tests_dict, key);
150 g_assert(p != NULL);
151 g_assert(strcmp(p, str) == 0);
153 QDECREF(tests_dict);
156 static void qdict_defaults_test(void)
158 QDict *dict, *copy;
160 dict = qdict_new();
161 copy = qdict_new();
163 qdict_set_default_str(dict, "foo", "abc");
164 qdict_set_default_str(dict, "foo", "def");
165 g_assert_cmpstr(qdict_get_str(dict, "foo"), ==, "abc");
166 qdict_set_default_str(dict, "bar", "ghi");
168 qdict_copy_default(copy, dict, "foo");
169 g_assert_cmpstr(qdict_get_str(copy, "foo"), ==, "abc");
170 qdict_set_default_str(copy, "bar", "xyz");
171 qdict_copy_default(copy, dict, "bar");
172 g_assert_cmpstr(qdict_get_str(copy, "bar"), ==, "xyz");
174 QDECREF(copy);
175 QDECREF(dict);
178 static void qdict_haskey_not_test(void)
180 QDict *tests_dict = qdict_new();
181 g_assert(qdict_haskey(tests_dict, "test") == 0);
183 QDECREF(tests_dict);
186 static void qdict_haskey_test(void)
188 const char *key = "test";
189 QDict *tests_dict = qdict_new();
191 qdict_put(tests_dict, key, qint_from_int(0));
192 g_assert(qdict_haskey(tests_dict, key) == 1);
194 QDECREF(tests_dict);
197 static void qdict_del_test(void)
199 const char *key = "key test";
200 QDict *tests_dict = qdict_new();
202 qdict_put(tests_dict, key, qstring_from_str("foo"));
203 g_assert(qdict_size(tests_dict) == 1);
205 qdict_del(tests_dict, key);
207 g_assert(qdict_size(tests_dict) == 0);
208 g_assert(qdict_haskey(tests_dict, key) == 0);
210 QDECREF(tests_dict);
213 static void qobject_to_qdict_test(void)
215 QDict *tests_dict = qdict_new();
216 g_assert(qobject_to_qdict(QOBJECT(tests_dict)) == tests_dict);
218 QDECREF(tests_dict);
221 static void qdict_iterapi_test(void)
223 int count;
224 const QDictEntry *ent;
225 QDict *tests_dict = qdict_new();
227 g_assert(qdict_first(tests_dict) == NULL);
229 qdict_put(tests_dict, "key1", qint_from_int(1));
230 qdict_put(tests_dict, "key2", qint_from_int(2));
231 qdict_put(tests_dict, "key3", qint_from_int(3));
233 count = 0;
234 for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
235 g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
236 count++;
239 g_assert(count == qdict_size(tests_dict));
241 /* Do it again to test restarting */
242 count = 0;
243 for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
244 g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
245 count++;
248 g_assert(count == qdict_size(tests_dict));
250 QDECREF(tests_dict);
253 static void qdict_flatten_test(void)
255 QList *list1 = qlist_new();
256 QList *list2 = qlist_new();
257 QDict *dict1 = qdict_new();
258 QDict *dict2 = qdict_new();
259 QDict *dict3 = qdict_new();
262 * Test the flattening of
265 * "e": [
266 * 42,
268 * 23,
269 * 66,
271 * "a": 0,
272 * "b": 1
275 * ],
276 * "f": {
277 * "c": 2,
278 * "d": 3,
279 * },
280 * "g": 4
283 * to
286 * "e.0": 42,
287 * "e.1.0": 23,
288 * "e.1.1": 66,
289 * "e.1.2.a": 0,
290 * "e.1.2.b": 1,
291 * "f.c": 2,
292 * "f.d": 3,
293 * "g": 4
297 qdict_put(dict1, "a", qint_from_int(0));
298 qdict_put(dict1, "b", qint_from_int(1));
300 qlist_append_obj(list1, QOBJECT(qint_from_int(23)));
301 qlist_append_obj(list1, QOBJECT(qint_from_int(66)));
302 qlist_append_obj(list1, QOBJECT(dict1));
303 qlist_append_obj(list2, QOBJECT(qint_from_int(42)));
304 qlist_append_obj(list2, QOBJECT(list1));
306 qdict_put(dict2, "c", qint_from_int(2));
307 qdict_put(dict2, "d", qint_from_int(3));
308 qdict_put_obj(dict3, "e", QOBJECT(list2));
309 qdict_put_obj(dict3, "f", QOBJECT(dict2));
310 qdict_put(dict3, "g", qint_from_int(4));
312 qdict_flatten(dict3);
314 g_assert(qdict_get_int(dict3, "e.0") == 42);
315 g_assert(qdict_get_int(dict3, "e.1.0") == 23);
316 g_assert(qdict_get_int(dict3, "e.1.1") == 66);
317 g_assert(qdict_get_int(dict3, "e.1.2.a") == 0);
318 g_assert(qdict_get_int(dict3, "e.1.2.b") == 1);
319 g_assert(qdict_get_int(dict3, "f.c") == 2);
320 g_assert(qdict_get_int(dict3, "f.d") == 3);
321 g_assert(qdict_get_int(dict3, "g") == 4);
323 g_assert(qdict_size(dict3) == 8);
325 QDECREF(dict3);
328 static void qdict_array_split_test(void)
330 QDict *test_dict = qdict_new();
331 QDict *dict1, *dict2;
332 QInt *int1;
333 QList *test_list;
336 * Test the split of
339 * "1.x": 0,
340 * "4.y": 1,
341 * "0.a": 42,
342 * "o.o": 7,
343 * "0.b": 23,
344 * "2": 66
347 * to
351 * "a": 42,
352 * "b": 23
353 * },
355 * "x": 0
356 * },
357 * 66
360 * and
363 * "4.y": 1,
364 * "o.o": 7
367 * (remaining in the old QDict)
369 * This example is given in the comment of qdict_array_split().
372 qdict_put(test_dict, "1.x", qint_from_int(0));
373 qdict_put(test_dict, "4.y", qint_from_int(1));
374 qdict_put(test_dict, "0.a", qint_from_int(42));
375 qdict_put(test_dict, "o.o", qint_from_int(7));
376 qdict_put(test_dict, "0.b", qint_from_int(23));
377 qdict_put(test_dict, "2", qint_from_int(66));
379 qdict_array_split(test_dict, &test_list);
381 dict1 = qobject_to_qdict(qlist_pop(test_list));
382 dict2 = qobject_to_qdict(qlist_pop(test_list));
383 int1 = qobject_to_qint(qlist_pop(test_list));
385 g_assert(dict1);
386 g_assert(dict2);
387 g_assert(int1);
388 g_assert(qlist_empty(test_list));
390 QDECREF(test_list);
392 g_assert(qdict_get_int(dict1, "a") == 42);
393 g_assert(qdict_get_int(dict1, "b") == 23);
395 g_assert(qdict_size(dict1) == 2);
397 QDECREF(dict1);
399 g_assert(qdict_get_int(dict2, "x") == 0);
401 g_assert(qdict_size(dict2) == 1);
403 QDECREF(dict2);
405 g_assert(qint_get_int(int1) == 66);
407 QDECREF(int1);
409 g_assert(qdict_get_int(test_dict, "4.y") == 1);
410 g_assert(qdict_get_int(test_dict, "o.o") == 7);
412 g_assert(qdict_size(test_dict) == 2);
414 QDECREF(test_dict);
418 * Test the split of
421 * "0": 42,
422 * "1": 23,
423 * "1.x": 84
426 * to
429 * 42
432 * and
435 * "1": 23,
436 * "1.x": 84
439 * That is, test whether splitting stops if there is both an entry with key
440 * of "%u" and other entries with keys prefixed "%u." for the same index.
443 test_dict = qdict_new();
445 qdict_put(test_dict, "0", qint_from_int(42));
446 qdict_put(test_dict, "1", qint_from_int(23));
447 qdict_put(test_dict, "1.x", qint_from_int(84));
449 qdict_array_split(test_dict, &test_list);
451 int1 = qobject_to_qint(qlist_pop(test_list));
453 g_assert(int1);
454 g_assert(qlist_empty(test_list));
456 QDECREF(test_list);
458 g_assert(qint_get_int(int1) == 42);
460 QDECREF(int1);
462 g_assert(qdict_get_int(test_dict, "1") == 23);
463 g_assert(qdict_get_int(test_dict, "1.x") == 84);
465 g_assert(qdict_size(test_dict) == 2);
467 QDECREF(test_dict);
470 static void qdict_array_entries_test(void)
472 QDict *dict = qdict_new();
474 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);
476 qdict_put(dict, "bar", qint_from_int(0));
477 qdict_put(dict, "baz.0", qint_from_int(0));
478 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);
480 qdict_put(dict, "foo.1", qint_from_int(0));
481 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
482 qdict_put(dict, "foo.0", qint_from_int(0));
483 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 2);
484 qdict_put(dict, "foo.bar", qint_from_int(0));
485 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
486 qdict_del(dict, "foo.bar");
488 qdict_put(dict, "foo.2.a", qint_from_int(0));
489 qdict_put(dict, "foo.2.b", qint_from_int(0));
490 qdict_put(dict, "foo.2.c", qint_from_int(0));
491 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 3);
492 g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
494 QDECREF(dict);
496 dict = qdict_new();
497 qdict_put(dict, "1", qint_from_int(0));
498 g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
499 qdict_put(dict, "0", qint_from_int(0));
500 g_assert_cmpint(qdict_array_entries(dict, ""), ==, 2);
501 qdict_put(dict, "bar", qint_from_int(0));
502 g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
503 qdict_del(dict, "bar");
505 qdict_put(dict, "2.a", qint_from_int(0));
506 qdict_put(dict, "2.b", qint_from_int(0));
507 qdict_put(dict, "2.c", qint_from_int(0));
508 g_assert_cmpint(qdict_array_entries(dict, ""), ==, 3);
510 QDECREF(dict);
513 static void qdict_join_test(void)
515 QDict *dict1, *dict2;
516 bool overwrite = false;
517 int i;
519 dict1 = qdict_new();
520 dict2 = qdict_new();
523 /* Test everything once without overwrite and once with */
526 /* Test empty dicts */
527 qdict_join(dict1, dict2, overwrite);
529 g_assert(qdict_size(dict1) == 0);
530 g_assert(qdict_size(dict2) == 0);
533 /* First iteration: Test movement */
534 /* Second iteration: Test empty source and non-empty destination */
535 qdict_put(dict2, "foo", qint_from_int(42));
537 for (i = 0; i < 2; i++) {
538 qdict_join(dict1, dict2, overwrite);
540 g_assert(qdict_size(dict1) == 1);
541 g_assert(qdict_size(dict2) == 0);
543 g_assert(qdict_get_int(dict1, "foo") == 42);
547 /* Test non-empty source and destination without conflict */
548 qdict_put(dict2, "bar", qint_from_int(23));
550 qdict_join(dict1, dict2, overwrite);
552 g_assert(qdict_size(dict1) == 2);
553 g_assert(qdict_size(dict2) == 0);
555 g_assert(qdict_get_int(dict1, "foo") == 42);
556 g_assert(qdict_get_int(dict1, "bar") == 23);
559 /* Test conflict */
560 qdict_put(dict2, "foo", qint_from_int(84));
562 qdict_join(dict1, dict2, overwrite);
564 g_assert(qdict_size(dict1) == 2);
565 g_assert(qdict_size(dict2) == !overwrite);
567 g_assert(qdict_get_int(dict1, "foo") == overwrite ? 84 : 42);
568 g_assert(qdict_get_int(dict1, "bar") == 23);
570 if (!overwrite) {
571 g_assert(qdict_get_int(dict2, "foo") == 84);
575 /* Check the references */
576 g_assert(qdict_get(dict1, "foo")->refcnt == 1);
577 g_assert(qdict_get(dict1, "bar")->refcnt == 1);
579 if (!overwrite) {
580 g_assert(qdict_get(dict2, "foo")->refcnt == 1);
584 /* Clean up */
585 qdict_del(dict1, "foo");
586 qdict_del(dict1, "bar");
588 if (!overwrite) {
589 qdict_del(dict2, "foo");
592 while (overwrite ^= true);
595 QDECREF(dict1);
596 QDECREF(dict2);
600 * Errors test-cases
603 static void qdict_put_exists_test(void)
605 int value;
606 const char *key = "exists";
607 QDict *tests_dict = qdict_new();
609 qdict_put(tests_dict, key, qint_from_int(1));
610 qdict_put(tests_dict, key, qint_from_int(2));
612 value = qdict_get_int(tests_dict, key);
613 g_assert(value == 2);
615 g_assert(qdict_size(tests_dict) == 1);
617 QDECREF(tests_dict);
620 static void qdict_get_not_exists_test(void)
622 QDict *tests_dict = qdict_new();
623 g_assert(qdict_get(tests_dict, "foo") == NULL);
625 QDECREF(tests_dict);
629 * Stress test-case
631 * This is a lot big for a unit-test, but there is no other place
632 * to have it.
635 static void remove_dots(char *string)
637 char *p = strchr(string, ':');
638 if (p)
639 *p = '\0';
642 static QString *read_line(FILE *file, char *key)
644 char value[128];
646 if (fscanf(file, "%127s%127s", key, value) == EOF) {
647 return NULL;
649 remove_dots(key);
650 return qstring_from_str(value);
653 #define reset_file(file) fseek(file, 0L, SEEK_SET)
655 static void qdict_stress_test(void)
657 size_t lines;
658 char key[128];
659 FILE *test_file;
660 QDict *qdict;
661 QString *value;
662 const char *test_file_path = "qdict-test-data.txt";
664 test_file = fopen(test_file_path, "r");
665 g_assert(test_file != NULL);
667 // Create the dict
668 qdict = qdict_new();
669 g_assert(qdict != NULL);
671 // Add everything from the test file
672 for (lines = 0;; lines++) {
673 value = read_line(test_file, key);
674 if (!value)
675 break;
677 qdict_put(qdict, key, value);
679 g_assert(qdict_size(qdict) == lines);
681 // Check if everything is really in there
682 reset_file(test_file);
683 for (;;) {
684 const char *str1, *str2;
686 value = read_line(test_file, key);
687 if (!value)
688 break;
690 str1 = qstring_get_str(value);
692 str2 = qdict_get_str(qdict, key);
693 g_assert(str2 != NULL);
695 g_assert(strcmp(str1, str2) == 0);
697 QDECREF(value);
700 // Delete everything
701 reset_file(test_file);
702 for (;;) {
703 value = read_line(test_file, key);
704 if (!value)
705 break;
707 qdict_del(qdict, key);
708 QDECREF(value);
710 g_assert(qdict_haskey(qdict, key) == 0);
712 fclose(test_file);
714 g_assert(qdict_size(qdict) == 0);
715 QDECREF(qdict);
718 int main(int argc, char **argv)
720 g_test_init(&argc, &argv, NULL);
722 g_test_add_func("/public/new", qdict_new_test);
723 g_test_add_func("/public/put_obj", qdict_put_obj_test);
724 g_test_add_func("/public/destroy_simple", qdict_destroy_simple_test);
726 /* Continue, but now with fixtures */
727 g_test_add_func("/public/get", qdict_get_test);
728 g_test_add_func("/public/get_int", qdict_get_int_test);
729 g_test_add_func("/public/get_try_int", qdict_get_try_int_test);
730 g_test_add_func("/public/get_str", qdict_get_str_test);
731 g_test_add_func("/public/get_try_str", qdict_get_try_str_test);
732 g_test_add_func("/public/defaults", qdict_defaults_test);
733 g_test_add_func("/public/haskey_not", qdict_haskey_not_test);
734 g_test_add_func("/public/haskey", qdict_haskey_test);
735 g_test_add_func("/public/del", qdict_del_test);
736 g_test_add_func("/public/to_qdict", qobject_to_qdict_test);
737 g_test_add_func("/public/iterapi", qdict_iterapi_test);
738 g_test_add_func("/public/flatten", qdict_flatten_test);
739 g_test_add_func("/public/array_split", qdict_array_split_test);
740 g_test_add_func("/public/array_entries", qdict_array_entries_test);
741 g_test_add_func("/public/join", qdict_join_test);
743 g_test_add_func("/errors/put_exists", qdict_put_exists_test);
744 g_test_add_func("/errors/get_not_exists", qdict_get_not_exists_test);
746 /* The Big one */
747 if (g_test_slow()) {
748 g_test_add_func("/stress/test", qdict_stress_test);
751 return g_test_run();