4 * Copyright (C) 2009 Red Hat Inc.
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.
13 #include "qapi/qmp/qint.h"
14 #include "qapi/qmp/qfloat.h"
15 #include "qapi/qmp/qdict.h"
16 #include "qapi/qmp/qbool.h"
17 #include "qapi/qmp/qstring.h"
18 #include "qapi/qmp/qobject.h"
19 #include "qemu/queue.h"
20 #include "qemu-common.h"
22 static void qdict_destroy_obj(QObject
*obj
);
24 static const QType qdict_type
= {
26 .destroy
= qdict_destroy_obj
,
30 * qdict_new(): Create a new QDict
32 * Return strong reference.
34 QDict
*qdict_new(void)
38 qdict
= g_malloc0(sizeof(*qdict
));
39 QOBJECT_INIT(qdict
, &qdict_type
);
45 * qobject_to_qdict(): Convert a QObject into a QDict
47 QDict
*qobject_to_qdict(const QObject
*obj
)
49 if (qobject_type(obj
) != QTYPE_QDICT
)
52 return container_of(obj
, QDict
, base
);
56 * tdb_hash(): based on the hash agorithm from gdbm, via tdb
57 * (from module-init-tools)
59 static unsigned int tdb_hash(const char *name
)
61 unsigned value
; /* Used to compute the hash value. */
62 unsigned i
; /* Used to cycle through random values. */
64 /* Set the initial value from the key size. */
65 for (value
= 0x238F13AF * strlen(name
), i
=0; name
[i
]; i
++)
66 value
= (value
+ (((const unsigned char *)name
)[i
] << (i
*5 % 24)));
68 return (1103515243 * value
+ 12345);
72 * alloc_entry(): allocate a new QDictEntry
74 static QDictEntry
*alloc_entry(const char *key
, QObject
*value
)
78 entry
= g_malloc0(sizeof(*entry
));
79 entry
->key
= g_strdup(key
);
86 * qdict_entry_value(): Return qdict entry value
88 * Return weak reference.
90 QObject
*qdict_entry_value(const QDictEntry
*entry
)
96 * qdict_entry_key(): Return qdict entry key
98 * Return a *pointer* to the string, it has to be duplicated before being
101 const char *qdict_entry_key(const QDictEntry
*entry
)
107 * qdict_find(): List lookup function
109 static QDictEntry
*qdict_find(const QDict
*qdict
,
110 const char *key
, unsigned int bucket
)
114 QLIST_FOREACH(entry
, &qdict
->table
[bucket
], next
)
115 if (!strcmp(entry
->key
, key
))
122 * qdict_put_obj(): Put a new QObject into the dictionary
124 * Insert the pair 'key:value' into 'qdict', if 'key' already exists
125 * its 'value' will be replaced.
127 * This is done by freeing the reference to the stored QObject and
128 * storing the new one in the same entry.
130 * NOTE: ownership of 'value' is transferred to the QDict
132 void qdict_put_obj(QDict
*qdict
, const char *key
, QObject
*value
)
137 bucket
= tdb_hash(key
) % QDICT_BUCKET_MAX
;
138 entry
= qdict_find(qdict
, key
, bucket
);
140 /* replace key's value */
141 qobject_decref(entry
->value
);
142 entry
->value
= value
;
144 /* allocate a new entry */
145 entry
= alloc_entry(key
, value
);
146 QLIST_INSERT_HEAD(&qdict
->table
[bucket
], entry
, next
);
152 * qdict_get(): Lookup for a given 'key'
154 * Return a weak reference to the QObject associated with 'key' if
155 * 'key' is present in the dictionary, NULL otherwise.
157 QObject
*qdict_get(const QDict
*qdict
, const char *key
)
161 entry
= qdict_find(qdict
, key
, tdb_hash(key
) % QDICT_BUCKET_MAX
);
162 return (entry
== NULL
? NULL
: entry
->value
);
166 * qdict_haskey(): Check if 'key' exists
168 * Return 1 if 'key' exists in the dict, 0 otherwise
170 int qdict_haskey(const QDict
*qdict
, const char *key
)
172 unsigned int bucket
= tdb_hash(key
) % QDICT_BUCKET_MAX
;
173 return (qdict_find(qdict
, key
, bucket
) == NULL
? 0 : 1);
177 * qdict_size(): Return the size of the dictionary
179 size_t qdict_size(const QDict
*qdict
)
185 * qdict_get_obj(): Get a QObject of a specific type
187 static QObject
*qdict_get_obj(const QDict
*qdict
, const char *key
,
192 obj
= qdict_get(qdict
, key
);
194 assert(qobject_type(obj
) == type
);
200 * qdict_get_double(): Get an number mapped by 'key'
202 * This function assumes that 'key' exists and it stores a
203 * QFloat or QInt object.
205 * Return number mapped by 'key'.
207 double qdict_get_double(const QDict
*qdict
, const char *key
)
209 QObject
*obj
= qdict_get(qdict
, key
);
212 switch (qobject_type(obj
)) {
214 return qfloat_get_double(qobject_to_qfloat(obj
));
216 return qint_get_int(qobject_to_qint(obj
));
223 * qdict_get_int(): Get an integer mapped by 'key'
225 * This function assumes that 'key' exists and it stores a
228 * Return integer mapped by 'key'.
230 int64_t qdict_get_int(const QDict
*qdict
, const char *key
)
232 QObject
*obj
= qdict_get_obj(qdict
, key
, QTYPE_QINT
);
233 return qint_get_int(qobject_to_qint(obj
));
237 * qdict_get_bool(): Get a bool mapped by 'key'
239 * This function assumes that 'key' exists and it stores a
242 * Return bool mapped by 'key'.
244 bool qdict_get_bool(const QDict
*qdict
, const char *key
)
246 QObject
*obj
= qdict_get_obj(qdict
, key
, QTYPE_QBOOL
);
247 return qbool_get_bool(qobject_to_qbool(obj
));
251 * qdict_get_qlist(): Get the QList mapped by 'key'
253 * This function assumes that 'key' exists and it stores a
256 * Return QList mapped by 'key'.
258 QList
*qdict_get_qlist(const QDict
*qdict
, const char *key
)
260 return qobject_to_qlist(qdict_get_obj(qdict
, key
, QTYPE_QLIST
));
264 * qdict_get_qdict(): Get the QDict mapped by 'key'
266 * This function assumes that 'key' exists and it stores a
269 * Return QDict mapped by 'key'.
271 QDict
*qdict_get_qdict(const QDict
*qdict
, const char *key
)
273 return qobject_to_qdict(qdict_get_obj(qdict
, key
, QTYPE_QDICT
));
277 * qdict_get_str(): Get a pointer to the stored string mapped
280 * This function assumes that 'key' exists and it stores a
283 * Return pointer to the string mapped by 'key'.
285 const char *qdict_get_str(const QDict
*qdict
, const char *key
)
287 QObject
*obj
= qdict_get_obj(qdict
, key
, QTYPE_QSTRING
);
288 return qstring_get_str(qobject_to_qstring(obj
));
292 * qdict_get_try_int(): Try to get integer mapped by 'key'
294 * Return integer mapped by 'key', if it is not present in
295 * the dictionary or if the stored object is not of QInt type
296 * 'def_value' will be returned.
298 int64_t qdict_get_try_int(const QDict
*qdict
, const char *key
,
303 obj
= qdict_get(qdict
, key
);
304 if (!obj
|| qobject_type(obj
) != QTYPE_QINT
)
307 return qint_get_int(qobject_to_qint(obj
));
311 * qdict_get_try_bool(): Try to get a bool mapped by 'key'
313 * Return bool mapped by 'key', if it is not present in the
314 * dictionary or if the stored object is not of QBool type
315 * 'def_value' will be returned.
317 bool qdict_get_try_bool(const QDict
*qdict
, const char *key
, bool def_value
)
321 obj
= qdict_get(qdict
, key
);
322 if (!obj
|| qobject_type(obj
) != QTYPE_QBOOL
)
325 return qbool_get_bool(qobject_to_qbool(obj
));
329 * qdict_get_try_str(): Try to get a pointer to the stored string
332 * Return a pointer to the string mapped by 'key', if it is not present
333 * in the dictionary or if the stored object is not of QString type
334 * NULL will be returned.
336 const char *qdict_get_try_str(const QDict
*qdict
, const char *key
)
340 obj
= qdict_get(qdict
, key
);
341 if (!obj
|| qobject_type(obj
) != QTYPE_QSTRING
)
344 return qstring_get_str(qobject_to_qstring(obj
));
348 * qdict_iter(): Iterate over all the dictionary's stored values.
350 * This function allows the user to provide an iterator, which will be
351 * called for each stored value in the dictionary.
353 void qdict_iter(const QDict
*qdict
,
354 void (*iter
)(const char *key
, QObject
*obj
, void *opaque
),
360 for (i
= 0; i
< QDICT_BUCKET_MAX
; i
++) {
361 QLIST_FOREACH(entry
, &qdict
->table
[i
], next
)
362 iter(entry
->key
, entry
->value
, opaque
);
366 static QDictEntry
*qdict_next_entry(const QDict
*qdict
, int first_bucket
)
370 for (i
= first_bucket
; i
< QDICT_BUCKET_MAX
; i
++) {
371 if (!QLIST_EMPTY(&qdict
->table
[i
])) {
372 return QLIST_FIRST(&qdict
->table
[i
]);
380 * qdict_first(): Return first qdict entry for iteration.
382 const QDictEntry
*qdict_first(const QDict
*qdict
)
384 return qdict_next_entry(qdict
, 0);
388 * qdict_next(): Return next qdict entry in an iteration.
390 const QDictEntry
*qdict_next(const QDict
*qdict
, const QDictEntry
*entry
)
394 ret
= QLIST_NEXT(entry
, next
);
396 unsigned int bucket
= tdb_hash(entry
->key
) % QDICT_BUCKET_MAX
;
397 ret
= qdict_next_entry(qdict
, bucket
+ 1);
404 * qdict_clone_shallow(): Clones a given QDict. Its entries are not copied, but
405 * another reference is added.
407 QDict
*qdict_clone_shallow(const QDict
*src
)
415 for (i
= 0; i
< QDICT_BUCKET_MAX
; i
++) {
416 QLIST_FOREACH(entry
, &src
->table
[i
], next
) {
417 qobject_incref(entry
->value
);
418 qdict_put_obj(dest
, entry
->key
, entry
->value
);
426 * qentry_destroy(): Free all the memory allocated by a QDictEntry
428 static void qentry_destroy(QDictEntry
*e
)
431 assert(e
->key
!= NULL
);
432 assert(e
->value
!= NULL
);
434 qobject_decref(e
->value
);
440 * qdict_del(): Delete a 'key:value' pair from the dictionary
442 * This will destroy all data allocated by this entry.
444 void qdict_del(QDict
*qdict
, const char *key
)
448 entry
= qdict_find(qdict
, key
, tdb_hash(key
) % QDICT_BUCKET_MAX
);
450 QLIST_REMOVE(entry
, next
);
451 qentry_destroy(entry
);
457 * qdict_destroy_obj(): Free all the memory allocated by a QDict
459 static void qdict_destroy_obj(QObject
*obj
)
465 qdict
= qobject_to_qdict(obj
);
467 for (i
= 0; i
< QDICT_BUCKET_MAX
; i
++) {
468 QDictEntry
*entry
= QLIST_FIRST(&qdict
->table
[i
]);
470 QDictEntry
*tmp
= QLIST_NEXT(entry
, next
);
471 QLIST_REMOVE(entry
, next
);
472 qentry_destroy(entry
);
481 * qdict_copy_default(): If no entry mapped by 'key' exists in 'dst' yet, the
482 * value of 'key' in 'src' is copied there (and the refcount increased
485 void qdict_copy_default(QDict
*dst
, QDict
*src
, const char *key
)
489 if (qdict_haskey(dst
, key
)) {
493 val
= qdict_get(src
, key
);
496 qdict_put_obj(dst
, key
, val
);
501 * qdict_set_default_str(): If no entry mapped by 'key' exists in 'dst' yet, a
502 * new QString initialised by 'val' is put there.
504 void qdict_set_default_str(QDict
*dst
, const char *key
, const char *val
)
506 if (qdict_haskey(dst
, key
)) {
510 qdict_put(dst
, key
, qstring_from_str(val
));
513 static void qdict_flatten_qdict(QDict
*qdict
, QDict
*target
,
516 static void qdict_flatten_qlist(QList
*qlist
, QDict
*target
, const char *prefix
)
519 const QListEntry
*entry
;
523 /* This function is never called with prefix == NULL, i.e., it is always
524 * called from within qdict_flatten_q(list|dict)(). Therefore, it does not
525 * need to remove list entries during the iteration (the whole list will be
526 * deleted eventually anyway from qdict_flatten_qdict()). */
529 entry
= qlist_first(qlist
);
531 for (i
= 0; entry
; entry
= qlist_next(entry
), i
++) {
532 value
= qlist_entry_obj(entry
);
533 new_key
= g_strdup_printf("%s.%i", prefix
, i
);
535 if (qobject_type(value
) == QTYPE_QDICT
) {
536 qdict_flatten_qdict(qobject_to_qdict(value
), target
, new_key
);
537 } else if (qobject_type(value
) == QTYPE_QLIST
) {
538 qdict_flatten_qlist(qobject_to_qlist(value
), target
, new_key
);
540 /* All other types are moved to the target unchanged. */
541 qobject_incref(value
);
542 qdict_put_obj(target
, new_key
, value
);
549 static void qdict_flatten_qdict(QDict
*qdict
, QDict
*target
, const char *prefix
)
552 const QDictEntry
*entry
, *next
;
556 entry
= qdict_first(qdict
);
558 while (entry
!= NULL
) {
560 next
= qdict_next(qdict
, entry
);
561 value
= qdict_entry_value(entry
);
566 new_key
= g_strdup_printf("%s.%s", prefix
, entry
->key
);
569 if (qobject_type(value
) == QTYPE_QDICT
) {
570 /* Entries of QDicts are processed recursively, the QDict object
571 * itself disappears. */
572 qdict_flatten_qdict(qobject_to_qdict(value
), target
,
573 new_key
? new_key
: entry
->key
);
575 } else if (qobject_type(value
) == QTYPE_QLIST
) {
576 qdict_flatten_qlist(qobject_to_qlist(value
), target
,
577 new_key
? new_key
: entry
->key
);
580 /* All other objects are moved to the target unchanged. */
581 qobject_incref(value
);
582 qdict_put_obj(target
, new_key
, value
);
589 qdict_del(qdict
, entry
->key
);
591 /* Restart loop after modifying the iterated QDict */
592 entry
= qdict_first(qdict
);
601 * qdict_flatten(): For each nested QDict with key x, all fields with key y
602 * are moved to this QDict and their key is renamed to "x.y". For each nested
603 * QList with key x, the field at index y is moved to this QDict with the key
604 * "x.y" (i.e., the reverse of what qdict_array_split() does).
605 * This operation is applied recursively for nested QDicts and QLists.
607 void qdict_flatten(QDict
*qdict
)
609 qdict_flatten_qdict(qdict
, qdict
, NULL
);
612 /* extract all the src QDict entries starting by start into dst */
613 void qdict_extract_subqdict(QDict
*src
, QDict
**dst
, const char *start
)
616 const QDictEntry
*entry
, *next
;
620 entry
= qdict_first(src
);
622 while (entry
!= NULL
) {
623 next
= qdict_next(src
, entry
);
624 if (strstart(entry
->key
, start
, &p
)) {
625 qobject_incref(entry
->value
);
626 qdict_put_obj(*dst
, p
, entry
->value
);
627 qdict_del(src
, entry
->key
);
633 static int qdict_count_prefixed_entries(const QDict
*src
, const char *start
)
635 const QDictEntry
*entry
;
638 for (entry
= qdict_first(src
); entry
; entry
= qdict_next(src
, entry
)) {
639 if (strstart(entry
->key
, start
, NULL
)) {
640 if (count
== INT_MAX
) {
651 * qdict_array_split(): This function moves array-like elements of a QDict into
652 * a new QList. Every entry in the original QDict with a key "%u" or one
653 * prefixed "%u.", where %u designates an unsigned integer starting at 0 and
654 * incrementally counting up, will be moved to a new QDict at index %u in the
655 * output QList with the key prefix removed, if that prefix is "%u.". If the
656 * whole key is just "%u", the whole QObject will be moved unchanged without
657 * creating a new QDict. The function terminates when there is no entry in the
658 * QDict with a prefix directly (incrementally) following the last one; it also
659 * returns if there are both entries with "%u" and "%u." for the same index %u.
660 * Example: {"0.a": 42, "0.b": 23, "1.x": 0, "4.y": 1, "o.o": 7, "2": 66}
661 * (or {"1.x": 0, "4.y": 1, "0.a": 42, "o.o": 7, "0.b": 23, "2": 66})
662 * => [{"a": 42, "b": 23}, {"x": 0}, 66]
663 * and {"4.y": 1, "o.o": 7} (remainder of the old QDict)
665 void qdict_array_split(QDict
*src
, QList
**dst
)
671 for (i
= 0; i
< UINT_MAX
; i
++) {
675 char indexstr
[32], prefix
[32];
678 snprintf_ret
= snprintf(indexstr
, 32, "%u", i
);
679 assert(snprintf_ret
< 32);
681 subqobj
= qdict_get(src
, indexstr
);
683 snprintf_ret
= snprintf(prefix
, 32, "%u.", i
);
684 assert(snprintf_ret
< 32);
686 /* Overflow is the same as positive non-zero results */
687 is_subqdict
= qdict_count_prefixed_entries(src
, prefix
);
689 // There may be either a single subordinate object (named "%u") or
690 // multiple objects (each with a key prefixed "%u."), but not both.
691 if (!subqobj
== !is_subqdict
) {
696 qdict_extract_subqdict(src
, &subqdict
, prefix
);
697 assert(qdict_size(subqdict
) > 0);
699 qobject_incref(subqobj
);
700 qdict_del(src
, indexstr
);
703 qlist_append_obj(*dst
, subqobj
?: QOBJECT(subqdict
));
708 * qdict_array_entries(): Returns the number of direct array entries if the
709 * sub-QDict of src specified by the prefix in subqdict (or src itself for
710 * prefix == "") is valid as an array, i.e. the length of the created list if
711 * the sub-QDict would become empty after calling qdict_array_split() on it. If
712 * the array is not valid, -EINVAL is returned.
714 int qdict_array_entries(QDict
*src
, const char *subqdict
)
716 const QDictEntry
*entry
;
718 unsigned entries
= 0;
719 size_t subqdict_len
= strlen(subqdict
);
721 assert(!subqdict_len
|| subqdict
[subqdict_len
- 1] == '.');
723 /* qdict_array_split() loops until UINT_MAX, but as we want to return
724 * negative errors, we only have a signed return value here. Any additional
725 * entries will lead to -EINVAL. */
726 for (i
= 0; i
< INT_MAX
; i
++) {
728 int subqdict_entries
;
729 size_t slen
= 32 + subqdict_len
;
730 char indexstr
[slen
], prefix
[slen
];
733 snprintf_ret
= snprintf(indexstr
, slen
, "%s%u", subqdict
, i
);
734 assert(snprintf_ret
< slen
);
736 subqobj
= qdict_get(src
, indexstr
);
738 snprintf_ret
= snprintf(prefix
, slen
, "%s%u.", subqdict
, i
);
739 assert(snprintf_ret
< slen
);
741 subqdict_entries
= qdict_count_prefixed_entries(src
, prefix
);
742 if (subqdict_entries
< 0) {
743 return subqdict_entries
;
746 /* There may be either a single subordinate object (named "%u") or
747 * multiple objects (each with a key prefixed "%u."), but not both. */
748 if (subqobj
&& subqdict_entries
) {
750 } else if (!subqobj
&& !subqdict_entries
) {
754 entries
+= subqdict_entries
? subqdict_entries
: 1;
757 /* Consider everything handled that isn't part of the given sub-QDict */
758 for (entry
= qdict_first(src
); entry
; entry
= qdict_next(src
, entry
)) {
759 if (!strstart(qdict_entry_key(entry
), subqdict
, NULL
)) {
764 /* Anything left in the sub-QDict that wasn't handled? */
765 if (qdict_size(src
) != entries
) {
773 * qdict_join(): Absorb the src QDict into the dest QDict, that is, move all
774 * elements from src to dest.
776 * If an element from src has a key already present in dest, it will not be
777 * moved unless overwrite is true.
779 * If overwrite is true, the conflicting values in dest will be discarded and
780 * replaced by the corresponding values from src.
782 * Therefore, with overwrite being true, the src QDict will always be empty when
783 * this function returns. If overwrite is false, the src QDict will be empty
784 * iff there were no conflicts.
786 void qdict_join(QDict
*dest
, QDict
*src
, bool overwrite
)
788 const QDictEntry
*entry
, *next
;
790 entry
= qdict_first(src
);
792 next
= qdict_next(src
, entry
);
794 if (overwrite
|| !qdict_haskey(dest
, entry
->key
)) {
795 qobject_incref(entry
->value
);
796 qdict_put_obj(dest
, entry
->key
, entry
->value
);
797 qdict_del(src
, entry
->key
);