LXR qemu/tests/test-qmp-cmds.c

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