]>
Commit | Line | Data |
---|---|---|
681c28a3 | 1 | #include "qemu/osdep.h" |
6b673957 | 2 | #include "qapi/qmp/qdict.h" |
15280c36 | 3 | #include "qapi/qmp/qnum.h" |
6b673957 | 4 | #include "qapi/qmp/qstring.h" |
e688df6b | 5 | #include "qapi/error.h" |
b3db211f | 6 | #include "qapi/qobject-input-visitor.h" |
d98150f0 LE |
7 | #include "tests/test-qapi-types.h" |
8 | #include "tests/test-qapi-visit.h" | |
eb815e24 | 9 | #include "test-qapi-commands.h" |
69ed8366 | 10 | |
1527badb MA |
11 | static QmpCommandList qmp_commands; |
12 | ||
1f7b9f31 | 13 | #if defined(TEST_IF_STRUCT) && defined(TEST_IF_CMD) |
967c8851 MAL |
14 | UserDefThree *qmp_TestIfCmd(TestIfStruct *foo, Error **errp) |
15 | { | |
16 | return NULL; | |
17 | } | |
1f7b9f31 | 18 | #endif |
967c8851 MAL |
19 | |
20 | UserDefThree *qmp_TestCmdReturnDefThree(Error **errp) | |
21 | { | |
22 | return NULL; | |
23 | } | |
24 | ||
69ed8366 MR |
25 | void qmp_user_def_cmd(Error **errp) |
26 | { | |
27 | } | |
28 | ||
7b13f2c2 | 29 | void qmp_test_flags_command(Error **errp) |
1a1b11dc PX |
30 | { |
31 | } | |
32 | ||
ae6bf766 MAL |
33 | void qmp_cmd_success_response(Error **errp) |
34 | { | |
35 | } | |
36 | ||
972a1101 EB |
37 | Empty2 *qmp_user_def_cmd0(Error **errp) |
38 | { | |
39 | return g_new0(Empty2, 1); | |
40 | } | |
41 | ||
69ed8366 MR |
42 | void qmp_user_def_cmd1(UserDefOne * ud1, Error **errp) |
43 | { | |
44 | } | |
45 | ||
ab22ad96 MA |
46 | UserDefTwo *qmp_user_def_cmd2(UserDefOne *ud1a, |
47 | bool has_udb1, UserDefOne *ud1b, | |
48 | Error **errp) | |
69ed8366 MR |
49 | { |
50 | UserDefTwo *ret; | |
7267c094 AL |
51 | UserDefOne *ud1c = g_malloc0(sizeof(UserDefOne)); |
52 | UserDefOne *ud1d = g_malloc0(sizeof(UserDefOne)); | |
69ed8366 MR |
53 | |
54 | ud1c->string = strdup(ud1a->string); | |
ddf21908 | 55 | ud1c->integer = ud1a->integer; |
ab22ad96 | 56 | ud1d->string = strdup(has_udb1 ? ud1b->string : "blah0"); |
ddf21908 | 57 | ud1d->integer = has_udb1 ? ud1b->integer : 0; |
69ed8366 | 58 | |
b6fcf32d EB |
59 | ret = g_new0(UserDefTwo, 1); |
60 | ret->string0 = strdup("blah1"); | |
6446a592 EB |
61 | ret->dict1 = g_new0(UserDefTwoDict, 1); |
62 | ret->dict1->string1 = strdup("blah2"); | |
63 | ret->dict1->dict2 = g_new0(UserDefTwoDictDict, 1); | |
64 | ret->dict1->dict2->userdef = ud1c; | |
65 | ret->dict1->dict2->string = strdup("blah3"); | |
66 | ret->dict1->dict3 = g_new0(UserDefTwoDictDict, 1); | |
67 | ret->dict1->has_dict3 = true; | |
68 | ret->dict1->dict3->userdef = ud1d; | |
69 | ret->dict1->dict3->string = strdup("blah4"); | |
69ed8366 MR |
70 | |
71 | return ret; | |
72 | } | |
73 | ||
cae95eae | 74 | int64_t qmp_guest_get_time(int64_t a, bool has_b, int64_t b, Error **errp) |
c2216a8a MA |
75 | { |
76 | return a + (has_b ? b : 0); | |
77 | } | |
78 | ||
28770e05 MA |
79 | QObject *qmp_guest_sync(QObject *arg, Error **errp) |
80 | { | |
81 | return arg; | |
82 | } | |
83 | ||
c818408e EB |
84 | void qmp_boxed_struct(UserDefZero *arg, Error **errp) |
85 | { | |
86 | } | |
87 | ||
b359f4b2 | 88 | void qmp_boxed_union(UserDefListUnion *arg, Error **errp) |
c818408e EB |
89 | { |
90 | } | |
91 | ||
e3c4c3d7 EB |
92 | __org_qemu_x_Union1 *qmp___org_qemu_x_command(__org_qemu_x_EnumList *a, |
93 | __org_qemu_x_StructList *b, | |
94 | __org_qemu_x_Union2 *c, | |
95 | __org_qemu_x_Alt *d, | |
96 | Error **errp) | |
97 | { | |
98 | __org_qemu_x_Union1 *ret = g_new0(__org_qemu_x_Union1, 1); | |
99 | ||
c363acef | 100 | ret->type = ORG_QEMU_X_UNION1_KIND___ORG_QEMU_X_BRANCH; |
32bafa8f | 101 | ret->u.__org_qemu_x_branch.data = strdup("blah1"); |
e3c4c3d7 | 102 | |
c43567c1 EB |
103 | /* Also test that 'wchar-t' was munged to 'q_wchar_t' */ |
104 | if (b && b->value && !b->value->has_q_wchar_t) { | |
105 | b->value->q_wchar_t = 1; | |
106 | } | |
e3c4c3d7 EB |
107 | return ret; |
108 | } | |
109 | ||
110 | ||
69ed8366 MR |
111 | /* test commands with no input and no return value */ |
112 | static void test_dispatch_cmd(void) | |
113 | { | |
114 | QDict *req = qdict_new(); | |
d43b1694 | 115 | QDict *resp; |
69ed8366 | 116 | |
46f5ac20 | 117 | qdict_put_str(req, "execute", "user_def_cmd"); |
69ed8366 | 118 | |
674ed722 | 119 | resp = qmp_dispatch(&qmp_commands, QOBJECT(req), false); |
69ed8366 | 120 | assert(resp != NULL); |
d43b1694 | 121 | assert(!qdict_haskey(resp, "error")); |
69ed8366 | 122 | |
cb3e7f08 MAL |
123 | qobject_unref(resp); |
124 | qobject_unref(req); | |
69ed8366 MR |
125 | } |
126 | ||
c55f070b PX |
127 | static void test_dispatch_cmd_oob(void) |
128 | { | |
129 | QDict *req = qdict_new(); | |
130 | QDict *resp; | |
131 | ||
132 | qdict_put_str(req, "exec-oob", "test-flags-command"); | |
133 | ||
134 | resp = qmp_dispatch(&qmp_commands, QOBJECT(req), true); | |
135 | assert(resp != NULL); | |
136 | assert(!qdict_haskey(resp, "error")); | |
137 | ||
138 | qobject_unref(resp); | |
139 | qobject_unref(req); | |
140 | } | |
141 | ||
69ed8366 | 142 | /* test commands that return an error due to invalid parameters */ |
51009170 | 143 | static void test_dispatch_cmd_failure(void) |
69ed8366 MR |
144 | { |
145 | QDict *req = qdict_new(); | |
a0067da1 | 146 | QDict *args = qdict_new(); |
d43b1694 | 147 | QDict *resp; |
69ed8366 | 148 | |
46f5ac20 | 149 | qdict_put_str(req, "execute", "user_def_cmd2"); |
69ed8366 | 150 | |
674ed722 | 151 | resp = qmp_dispatch(&qmp_commands, QOBJECT(req), false); |
69ed8366 | 152 | assert(resp != NULL); |
d43b1694 | 153 | assert(qdict_haskey(resp, "error")); |
69ed8366 | 154 | |
cb3e7f08 MAL |
155 | qobject_unref(resp); |
156 | qobject_unref(req); | |
a0067da1 MAL |
157 | |
158 | /* check that with extra arguments it throws an error */ | |
159 | req = qdict_new(); | |
46f5ac20 | 160 | qdict_put_int(args, "a", 66); |
a0067da1 MAL |
161 | qdict_put(req, "arguments", args); |
162 | ||
46f5ac20 | 163 | qdict_put_str(req, "execute", "user_def_cmd"); |
a0067da1 | 164 | |
674ed722 | 165 | resp = qmp_dispatch(&qmp_commands, QOBJECT(req), false); |
a0067da1 | 166 | assert(resp != NULL); |
d43b1694 | 167 | assert(qdict_haskey(resp, "error")); |
a0067da1 | 168 | |
cb3e7f08 MAL |
169 | qobject_unref(resp); |
170 | qobject_unref(req); | |
69ed8366 MR |
171 | } |
172 | ||
ae6bf766 MAL |
173 | static void test_dispatch_cmd_success_response(void) |
174 | { | |
175 | QDict *req = qdict_new(); | |
176 | QDict *resp; | |
177 | ||
178 | qdict_put_str(req, "execute", "cmd-success-response"); | |
179 | resp = qmp_dispatch(&qmp_commands, QOBJECT(req), false); | |
180 | g_assert_null(resp); | |
181 | qobject_unref(req); | |
182 | } | |
183 | ||
357765fe MA |
184 | static QObject *test_qmp_dispatch(QDict *req) |
185 | { | |
357765fe MA |
186 | QDict *resp; |
187 | QObject *ret; | |
188 | ||
d43b1694 | 189 | resp = qmp_dispatch(&qmp_commands, QOBJECT(req), false); |
357765fe MA |
190 | assert(resp && !qdict_haskey(resp, "error")); |
191 | ret = qdict_get(resp, "return"); | |
192 | assert(ret); | |
cb3e7f08 | 193 | qobject_ref(ret); |
d43b1694 | 194 | qobject_unref(resp); |
357765fe MA |
195 | return ret; |
196 | } | |
197 | ||
69ed8366 MR |
198 | /* test commands that involve both input parameters and return values */ |
199 | static void test_dispatch_cmd_io(void) | |
200 | { | |
201 | QDict *req = qdict_new(); | |
202 | QDict *args = qdict_new(); | |
c2216a8a | 203 | QDict *args3 = qdict_new(); |
69ed8366 MR |
204 | QDict *ud1a = qdict_new(); |
205 | QDict *ud1b = qdict_new(); | |
357765fe MA |
206 | QDict *ret, *ret_dict, *ret_dict_dict, *ret_dict_dict_userdef; |
207 | QDict *ret_dict_dict2, *ret_dict_dict2_userdef; | |
01b2ffce MAL |
208 | QNum *ret3; |
209 | int64_t val; | |
69ed8366 | 210 | |
46f5ac20 EB |
211 | qdict_put_int(ud1a, "integer", 42); |
212 | qdict_put_str(ud1a, "string", "hello"); | |
213 | qdict_put_int(ud1b, "integer", 422); | |
214 | qdict_put_str(ud1b, "string", "hello2"); | |
de6e7951 EB |
215 | qdict_put(args, "ud1a", ud1a); |
216 | qdict_put(args, "ud1b", ud1b); | |
217 | qdict_put(req, "arguments", args); | |
46f5ac20 | 218 | qdict_put_str(req, "execute", "user_def_cmd2"); |
69ed8366 | 219 | |
7dc847eb | 220 | ret = qobject_to(QDict, test_qmp_dispatch(req)); |
357765fe | 221 | |
b6fcf32d EB |
222 | assert(!strcmp(qdict_get_str(ret, "string0"), "blah1")); |
223 | ret_dict = qdict_get_qdict(ret, "dict1"); | |
224 | assert(!strcmp(qdict_get_str(ret_dict, "string1"), "blah2")); | |
225 | ret_dict_dict = qdict_get_qdict(ret_dict, "dict2"); | |
357765fe MA |
226 | ret_dict_dict_userdef = qdict_get_qdict(ret_dict_dict, "userdef"); |
227 | assert(qdict_get_int(ret_dict_dict_userdef, "integer") == 42); | |
228 | assert(!strcmp(qdict_get_str(ret_dict_dict_userdef, "string"), "hello")); | |
229 | assert(!strcmp(qdict_get_str(ret_dict_dict, "string"), "blah3")); | |
b6fcf32d | 230 | ret_dict_dict2 = qdict_get_qdict(ret_dict, "dict3"); |
357765fe MA |
231 | ret_dict_dict2_userdef = qdict_get_qdict(ret_dict_dict2, "userdef"); |
232 | assert(qdict_get_int(ret_dict_dict2_userdef, "integer") == 422); | |
233 | assert(!strcmp(qdict_get_str(ret_dict_dict2_userdef, "string"), "hello2")); | |
234 | assert(!strcmp(qdict_get_str(ret_dict_dict2, "string"), "blah4")); | |
cb3e7f08 | 235 | qobject_unref(ret); |
c2216a8a | 236 | |
46f5ac20 | 237 | qdict_put_int(args3, "a", 66); |
c2216a8a | 238 | qdict_put(req, "arguments", args3); |
46f5ac20 | 239 | qdict_put_str(req, "execute", "guest-get-time"); |
c2216a8a | 240 | |
7dc847eb | 241 | ret3 = qobject_to(QNum, test_qmp_dispatch(req)); |
01b2ffce MAL |
242 | g_assert(qnum_get_try_int(ret3, &val)); |
243 | g_assert_cmpint(val, ==, 66); | |
cb3e7f08 | 244 | qobject_unref(ret3); |
c2216a8a | 245 | |
cb3e7f08 | 246 | qobject_unref(req); |
69ed8366 MR |
247 | } |
248 | ||
5cd5f0d0 MR |
249 | /* test generated dealloc functions for generated types */ |
250 | static void test_dealloc_types(void) | |
251 | { | |
252 | UserDefOne *ud1test, *ud1a, *ud1b; | |
253 | UserDefOneList *ud1list; | |
254 | ||
255 | ud1test = g_malloc0(sizeof(UserDefOne)); | |
ddf21908 | 256 | ud1test->integer = 42; |
5cd5f0d0 MR |
257 | ud1test->string = g_strdup("hi there 42"); |
258 | ||
259 | qapi_free_UserDefOne(ud1test); | |
260 | ||
261 | ud1a = g_malloc0(sizeof(UserDefOne)); | |
ddf21908 | 262 | ud1a->integer = 43; |
5cd5f0d0 MR |
263 | ud1a->string = g_strdup("hi there 43"); |
264 | ||
265 | ud1b = g_malloc0(sizeof(UserDefOne)); | |
ddf21908 | 266 | ud1b->integer = 44; |
5cd5f0d0 MR |
267 | ud1b->string = g_strdup("hi there 44"); |
268 | ||
269 | ud1list = g_malloc0(sizeof(UserDefOneList)); | |
270 | ud1list->value = ud1a; | |
271 | ud1list->next = g_malloc0(sizeof(UserDefOneList)); | |
272 | ud1list->next->value = ud1b; | |
273 | ||
274 | qapi_free_UserDefOneList(ud1list); | |
275 | } | |
276 | ||
d98150f0 LE |
277 | /* test generated deallocation on an object whose construction was prematurely |
278 | * terminated due to an error */ | |
279 | static void test_dealloc_partial(void) | |
280 | { | |
281 | static const char text[] = "don't leak me"; | |
282 | ||
283 | UserDefTwo *ud2 = NULL; | |
284 | Error *err = NULL; | |
285 | ||
286 | /* create partial object */ | |
287 | { | |
288 | QDict *ud2_dict; | |
b70ce101 | 289 | Visitor *v; |
d98150f0 LE |
290 | |
291 | ud2_dict = qdict_new(); | |
46f5ac20 | 292 | qdict_put_str(ud2_dict, "string0", text); |
d98150f0 | 293 | |
048abb7b | 294 | v = qobject_input_visitor_new(QOBJECT(ud2_dict)); |
b70ce101 EB |
295 | visit_type_UserDefTwo(v, NULL, &ud2, &err); |
296 | visit_free(v); | |
cb3e7f08 | 297 | qobject_unref(ud2_dict); |
d98150f0 LE |
298 | } |
299 | ||
68ab47e4 | 300 | /* verify that visit_type_XXX() cleans up properly on error */ |
a12a5a1a | 301 | error_free_or_abort(&err); |
68ab47e4 EB |
302 | assert(!ud2); |
303 | ||
304 | /* Manually create a partial object, leaving ud2->dict1 at NULL */ | |
305 | ud2 = g_new0(UserDefTwo, 1); | |
306 | ud2->string0 = g_strdup(text); | |
d98150f0 LE |
307 | |
308 | /* tear down partial object */ | |
309 | qapi_free_UserDefTwo(ud2); | |
310 | } | |
311 | ||
312 | ||
69ed8366 MR |
313 | int main(int argc, char **argv) |
314 | { | |
315 | g_test_init(&argc, &argv, NULL); | |
316 | ||
214e4a5b | 317 | g_test_add_func("/qmp/dispatch_cmd", test_dispatch_cmd); |
c55f070b | 318 | g_test_add_func("/qmp/dispatch_cmd_oob", test_dispatch_cmd_oob); |
214e4a5b MAL |
319 | g_test_add_func("/qmp/dispatch_cmd_failure", test_dispatch_cmd_failure); |
320 | g_test_add_func("/qmp/dispatch_cmd_io", test_dispatch_cmd_io); | |
ae6bf766 MAL |
321 | g_test_add_func("/qmp/dispatch_cmd_success_response", |
322 | test_dispatch_cmd_success_response); | |
214e4a5b MAL |
323 | g_test_add_func("/qmp/dealloc_types", test_dealloc_types); |
324 | g_test_add_func("/qmp/dealloc_partial", test_dealloc_partial); | |
69ed8366 | 325 | |
1527badb | 326 | test_qmp_init_marshal(&qmp_commands); |
69ed8366 MR |
327 | g_test_run(); |
328 | ||
329 | return 0; | |
330 | } |