qemu/tests/test-string-input-visitor.c
<<
>>
Prefs
   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 "qemu-common.h"
  16#include "qapi/error.h"
  17#include "qapi/string-input-visitor.h"
  18#include "test-qapi-types.h"
  19#include "test-qapi-visit.h"
  20#include "qapi/qmp/types.h"
  21
  22typedef struct TestInputVisitorData {
  23    Visitor *v;
  24} TestInputVisitorData;
  25
  26static void visitor_input_teardown(TestInputVisitorData *data,
  27                                   const void *unused)
  28{
  29    if (data->v) {
  30        visit_free(data->v);
  31        data->v = NULL;
  32    }
  33}
  34
  35/* This is provided instead of a test setup function so that the JSON
  36   string used by the tests are kept in the test functions (and not
  37   int main()) */
  38static
  39Visitor *visitor_input_test_init(TestInputVisitorData *data,
  40                                 const char *string)
  41{
  42    visitor_input_teardown(data, NULL);
  43
  44    data->v = string_input_visitor_new(string);
  45    g_assert(data->v);
  46    return data->v;
  47}
  48
  49static void test_visitor_in_int(TestInputVisitorData *data,
  50                                const void *unused)
  51{
  52    int64_t res = 0, value = -42;
  53    Error *err = NULL;
  54    Visitor *v;
  55
  56    v = visitor_input_test_init(data, "-42");
  57
  58    visit_type_int(v, NULL, &res, &err);
  59    g_assert(!err);
  60    g_assert_cmpint(res, ==, value);
  61
  62    v = visitor_input_test_init(data, "not an int");
  63
  64    visit_type_int(v, NULL, &res, &err);
  65    error_free_or_abort(&err);
  66
  67    v = visitor_input_test_init(data, "");
  68
  69    visit_type_int(v, NULL, &res, &err);
  70    error_free_or_abort(&err);
  71}
  72
  73static void check_ilist(Visitor *v, int64_t *expected, size_t n)
  74{
  75    int64List *res = NULL;
  76    int64List *tail;
  77    int i;
  78
  79    visit_type_int64List(v, NULL, &res, &error_abort);
  80    tail = res;
  81    for (i = 0; i < n; i++) {
  82        g_assert(tail);
  83        g_assert_cmpint(tail->value, ==, expected[i]);
  84        tail = tail->next;
  85    }
  86    g_assert(!tail);
  87
  88    qapi_free_int64List(res);
  89}
  90
  91static void check_ulist(Visitor *v, uint64_t *expected, size_t n)
  92{
  93    uint64List *res = NULL;
  94    uint64List *tail;
  95    int i;
  96
  97    /* BUG: unsigned numbers above INT64_MAX don't work */
  98    for (i = 0; i < n; i++) {
  99        if (expected[i] > INT64_MAX) {
 100            Error *err = NULL;
 101            visit_type_uint64List(v, NULL, &res, &err);
 102            error_free_or_abort(&err);
 103            return;
 104        }
 105    }
 106
 107    visit_type_uint64List(v, NULL, &res, &error_abort);
 108    tail = res;
 109    for (i = 0; i < n; i++) {
 110        g_assert(tail);
 111        g_assert_cmpuint(tail->value, ==, expected[i]);
 112        tail = tail->next;
 113    }
 114    g_assert(!tail);
 115
 116    qapi_free_uint64List(res);
 117}
 118
 119static void test_visitor_in_intList(TestInputVisitorData *data,
 120                                    const void *unused)
 121{
 122    /* Note: the visitor *sorts* ranges *unsigned* */
 123    int64_t expect1[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 20 };
 124    int64_t expect2[] = { 32767, -32768, -32767 };
 125    int64_t expect3[] = { INT64_MAX, INT64_MIN };
 126    uint64_t expect4[] = { UINT64_MAX };
 127    Error *err = NULL;
 128    int64List *res = NULL;
 129    int64List *tail;
 130    Visitor *v;
 131    int64_t val;
 132
 133    /* Valid lists */
 134
 135    v = visitor_input_test_init(data, "1,2,0,2-4,20,5-9,1-8");
 136    check_ilist(v, expect1, ARRAY_SIZE(expect1));
 137
 138    v = visitor_input_test_init(data, "32767,-32768--32767");
 139    check_ilist(v, expect2, ARRAY_SIZE(expect2));
 140
 141    v = visitor_input_test_init(data,
 142                                "-9223372036854775808,9223372036854775807");
 143    check_ilist(v, expect3, ARRAY_SIZE(expect3));
 144
 145    v = visitor_input_test_init(data, "18446744073709551615");
 146    check_ulist(v, expect4, ARRAY_SIZE(expect4));
 147
 148    /* Empty list */
 149
 150    v = visitor_input_test_init(data, "");
 151    visit_type_int64List(v, NULL, &res, &error_abort);
 152    g_assert(!res);
 153
 154    /* Not a list */
 155
 156    v = visitor_input_test_init(data, "not an int list");
 157
 158    visit_type_int64List(v, NULL, &res, &err);
 159    error_free_or_abort(&err);
 160    g_assert(!res);
 161
 162    /* Unvisited list tail */
 163
 164    v = visitor_input_test_init(data, "0,2-3");
 165
 166    /* Would be simpler if the visitor genuinely supported virtual walks */
 167    visit_start_list(v, NULL, (GenericList **)&res, sizeof(*res),
 168                     &error_abort);
 169    tail = res;
 170    visit_type_int64(v, NULL, &tail->value, &error_abort);
 171    g_assert_cmpint(tail->value, ==, 0);
 172    tail = (int64List *)visit_next_list(v, (GenericList *)tail, sizeof(*res));
 173    g_assert(tail);
 174    visit_type_int64(v, NULL, &tail->value, &error_abort);
 175    g_assert_cmpint(tail->value, ==, 2);
 176    tail = (int64List *)visit_next_list(v, (GenericList *)tail, sizeof(*res));
 177    g_assert(tail);
 178
 179    visit_check_list(v, &err);
 180    error_free_or_abort(&err);
 181    visit_end_list(v, (void **)&res);
 182
 183    qapi_free_int64List(res);
 184
 185    /* Visit beyond end of list */
 186    v = visitor_input_test_init(data, "0");
 187
 188    visit_start_list(v, NULL, (GenericList **)&res, sizeof(*res),
 189                     &error_abort);
 190    tail = res;
 191    visit_type_int64(v, NULL, &tail->value, &err);
 192    g_assert_cmpint(tail->value, ==, 0);
 193    visit_type_int64(v, NULL, &val, &err);
 194    g_assert_cmpint(val, ==, 1); /* BUG */
 195    visit_check_list(v, &error_abort);
 196    visit_end_list(v, (void **)&res);
 197
 198    qapi_free_int64List(res);
 199}
 200
 201static void test_visitor_in_bool(TestInputVisitorData *data,
 202                                 const void *unused)
 203{
 204    Error *err = NULL;
 205    bool res = false;
 206    Visitor *v;
 207
 208    v = visitor_input_test_init(data, "true");
 209
 210    visit_type_bool(v, NULL, &res, &err);
 211    g_assert(!err);
 212    g_assert_cmpint(res, ==, true);
 213
 214    v = visitor_input_test_init(data, "yes");
 215
 216    visit_type_bool(v, NULL, &res, &err);
 217    g_assert(!err);
 218    g_assert_cmpint(res, ==, true);
 219
 220    v = visitor_input_test_init(data, "on");
 221
 222    visit_type_bool(v, NULL, &res, &err);
 223    g_assert(!err);
 224    g_assert_cmpint(res, ==, true);
 225
 226    v = visitor_input_test_init(data, "false");
 227
 228    visit_type_bool(v, NULL, &res, &err);
 229    g_assert(!err);
 230    g_assert_cmpint(res, ==, false);
 231
 232    v = visitor_input_test_init(data, "no");
 233
 234    visit_type_bool(v, NULL, &res, &err);
 235    g_assert(!err);
 236    g_assert_cmpint(res, ==, false);
 237
 238    v = visitor_input_test_init(data, "off");
 239
 240    visit_type_bool(v, NULL, &res, &err);
 241    g_assert(!err);
 242    g_assert_cmpint(res, ==, false);
 243}
 244
 245static void test_visitor_in_number(TestInputVisitorData *data,
 246                                   const void *unused)
 247{
 248    double res = 0, value = 3.14;
 249    Error *err = NULL;
 250    Visitor *v;
 251
 252    v = visitor_input_test_init(data, "3.14");
 253
 254    visit_type_number(v, NULL, &res, &err);
 255    g_assert(!err);
 256    g_assert_cmpfloat(res, ==, value);
 257}
 258
 259static void test_visitor_in_string(TestInputVisitorData *data,
 260                                   const void *unused)
 261{
 262    char *res = NULL, *value = (char *) "Q E M U";
 263    Error *err = NULL;
 264    Visitor *v;
 265
 266    v = visitor_input_test_init(data, value);
 267
 268    visit_type_str(v, NULL, &res, &err);
 269    g_assert(!err);
 270    g_assert_cmpstr(res, ==, value);
 271
 272    g_free(res);
 273}
 274
 275static void test_visitor_in_enum(TestInputVisitorData *data,
 276                                 const void *unused)
 277{
 278    Error *err = NULL;
 279    Visitor *v;
 280    EnumOne i;
 281
 282    for (i = 0; EnumOne_lookup[i]; i++) {
 283        EnumOne res = -1;
 284
 285        v = visitor_input_test_init(data, EnumOne_lookup[i]);
 286
 287        visit_type_EnumOne(v, NULL, &res, &err);
 288        g_assert(!err);
 289        g_assert_cmpint(i, ==, res);
 290    }
 291}
 292
 293/* Try to crash the visitors */
 294static void test_visitor_in_fuzz(TestInputVisitorData *data,
 295                                 const void *unused)
 296{
 297    int64_t ires;
 298    intList *ilres;
 299    bool bres;
 300    double nres;
 301    char *sres;
 302    EnumOne eres;
 303    Visitor *v;
 304    unsigned int i;
 305    char buf[10000];
 306
 307    for (i = 0; i < 100; i++) {
 308        unsigned int j;
 309
 310        j = g_test_rand_int_range(0, sizeof(buf) - 1);
 311
 312        buf[j] = '\0';
 313
 314        if (j != 0) {
 315            for (j--; j != 0; j--) {
 316                buf[j - 1] = (char)g_test_rand_int_range(0, 256);
 317            }
 318        }
 319
 320        v = visitor_input_test_init(data, buf);
 321        visit_type_int(v, NULL, &ires, NULL);
 322
 323        v = visitor_input_test_init(data, buf);
 324        visit_type_intList(v, NULL, &ilres, NULL);
 325        qapi_free_intList(ilres);
 326
 327        v = visitor_input_test_init(data, buf);
 328        visit_type_bool(v, NULL, &bres, NULL);
 329
 330        v = visitor_input_test_init(data, buf);
 331        visit_type_number(v, NULL, &nres, NULL);
 332
 333        v = visitor_input_test_init(data, buf);
 334        sres = NULL;
 335        visit_type_str(v, NULL, &sres, NULL);
 336        g_free(sres);
 337
 338        v = visitor_input_test_init(data, buf);
 339        visit_type_EnumOne(v, NULL, &eres, NULL);
 340    }
 341}
 342
 343static void input_visitor_test_add(const char *testpath,
 344                                   TestInputVisitorData *data,
 345                                   void (*test_func)(TestInputVisitorData *data, const void *user_data))
 346{
 347    g_test_add(testpath, TestInputVisitorData, data, NULL, test_func,
 348               visitor_input_teardown);
 349}
 350
 351int main(int argc, char **argv)
 352{
 353    TestInputVisitorData in_visitor_data;
 354
 355    g_test_init(&argc, &argv, NULL);
 356
 357    input_visitor_test_add("/string-visitor/input/int",
 358                           &in_visitor_data, test_visitor_in_int);
 359    input_visitor_test_add("/string-visitor/input/intList",
 360                           &in_visitor_data, test_visitor_in_intList);
 361    input_visitor_test_add("/string-visitor/input/bool",
 362                           &in_visitor_data, test_visitor_in_bool);
 363    input_visitor_test_add("/string-visitor/input/number",
 364                           &in_visitor_data, test_visitor_in_number);
 365    input_visitor_test_add("/string-visitor/input/string",
 366                            &in_visitor_data, test_visitor_in_string);
 367    input_visitor_test_add("/string-visitor/input/enum",
 368                            &in_visitor_data, test_visitor_in_enum);
 369    input_visitor_test_add("/string-visitor/input/fuzz",
 370                            &in_visitor_data, test_visitor_in_fuzz);
 371
 372    g_test_run();
 373
 374    return 0;
 375}
 376