tests/test-string-input-visitor.c

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