tests/test-ovsdb.c

   1 /*
   2  * Copyright (c) 2009, 2010, 2011 Nicira Networks.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <config.h>
  18
  19 #include <assert.h>
  20 #include <fcntl.h>
  21 #include <getopt.h>
  22 #include <inttypes.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25
  26 #include "command-line.h"
  27 #include "dynamic-string.h"
  28 #include "json.h"
  29 #include "jsonrpc.h"
  30 #include "ovsdb-data.h"
  31 #include "ovsdb-error.h"
  32 #include "ovsdb-idl.h"
  33 #include "ovsdb-types.h"
  34 #include "ovsdb/column.h"
  35 #include "ovsdb/condition.h"
  36 #include "ovsdb/file.h"
  37 #include "ovsdb/log.h"
  38 #include "ovsdb/mutation.h"
  39 #include "ovsdb/ovsdb.h"
  40 #include "ovsdb/query.h"
  41 #include "ovsdb/row.h"
  42 #include "ovsdb/table.h"
  43 #include "ovsdb/transaction.h"
  44 #include "ovsdb/trigger.h"
  45 #include "poll-loop.h"
  46 #include "stream.h"
  47 #include "svec.h"
  48 #include "tests/idltest.h"
  49 #include "timeval.h"
  50 #include "util.h"
  51 #include "vlog.h"
  52
  53 static struct command all_commands[];
  54
  55 static void usage(void) NO_RETURN;
  56 static void parse_options(int argc, char *argv[]);
  57
  58 int
  59 main(int argc, char *argv[])
  60 {
  61     set_program_name(argv[0]);
  62     parse_options(argc, argv);
  63     run_command(argc - optind, argv + optind, all_commands);
  64     return 0;
  65 }
  66
  67 static void
  68 parse_options(int argc, char *argv[])
  69 {
  70     static struct option long_options[] = {
  71         {"timeout", required_argument, NULL, 't'},
  72         {"verbose", optional_argument, NULL, 'v'},
  73         {"help", no_argument, NULL, 'h'},
  74         {NULL, 0, NULL, 0},
  75     };
  76     char *short_options = long_options_to_short_options(long_options);
  77
  78     for (;;) {
  79         unsigned long int timeout;
  80         int c;
  81
  82         c = getopt_long(argc, argv, short_options, long_options, NULL);
  83         if (c == -1) {
  84             break;
  85         }
  86
  87         switch (c) {
  88         case 't':
  89             timeout = strtoul(optarg, NULL, 10);
  90             if (timeout <= 0) {
  91                 ovs_fatal(0, "value %s on -t or --timeout is not at least 1",
  92                           optarg);
  93             } else {
  94                 time_alarm(timeout);
  95             }
  96             break;
  97
  98         case 'h':
  99             usage();
 100
 101         case 'v':
 102             vlog_set_verbosity(optarg);
 103             break;
 104
 105         case '?':
 106             exit(EXIT_FAILURE);
 107
 108         default:
 109             abort();
 110         }
 111     }
 112     free(short_options);
 113 }
 114
 115 static void
 116 usage(void)
 117 {
 118     printf("%s: Open vSwitch database test utility\n"
 119            "usage: %s [OPTIONS] COMMAND [ARG...]\n\n"
 120            "  log-io FILE FLAGS COMMAND...\n"
 121            "    open FILE with FLAGS, run COMMANDs\n"
 122            "  default-atoms\n"
 123            "    test ovsdb_atom_default()\n"
 124            "  default-data\n"
 125            "    test ovsdb_datum_default()\n"
 126            "  parse-atomic-type TYPE\n"
 127            "    parse TYPE as OVSDB atomic type, and re-serialize\n"
 128            "  parse-base-type TYPE\n"
 129            "    parse TYPE as OVSDB base type, and re-serialize\n"
 130            "  parse-type JSON\n"
 131            "    parse JSON as OVSDB type, and re-serialize\n"
 132            "  parse-atoms TYPE ATOM...\n"
 133            "    parse JSON ATOMs as atoms of TYPE, and re-serialize\n"
 134            "  parse-atom-strings TYPE ATOM...\n"
 135            "    parse string ATOMs as atoms of given TYPE, and re-serialize\n"
 136            "  sort-atoms TYPE ATOM...\n"
 137            "    print JSON ATOMs in sorted order\n"
 138            "  parse-data TYPE DATUM...\n"
 139            "    parse JSON DATUMs as data of given TYPE, and re-serialize\n"
 140            "  parse-data-strings TYPE DATUM...\n"
 141            "    parse string DATUMs as data of given TYPE, and re-serialize\n"
 142            "  parse-column NAME OBJECT\n"
 143            "    parse column NAME with info OBJECT, and re-serialize\n"
 144            "  parse-table NAME OBJECT [DEFAULT-IS-ROOT]\n"
 145            "    parse table NAME with info OBJECT\n"
 146            "  parse-row TABLE ROW..., and re-serialize\n"
 147            "    parse each ROW of defined TABLE\n"
 148            "  compare-row TABLE ROW...\n"
 149            "    mutually compare all of the ROWs, print those that are equal\n"
 150            "  parse-conditions TABLE CONDITION...\n"
 151            "    parse each CONDITION on TABLE, and re-serialize\n"
 152            "  evaluate-conditions TABLE [CONDITION,...] [ROW,...]\n"
 153            "    test CONDITIONS on TABLE against each ROW, print results\n"
 154            "  parse-mutations TABLE MUTATION...\n"
 155            "    parse each MUTATION on TABLE, and re-serialize\n"
 156            "  execute-mutations TABLE [MUTATION,...] [ROW,...]\n"
 157            "    execute MUTATIONS on TABLE on each ROW, print results\n"
 158            "  query TABLE [ROW,...] [CONDITION,...]\n"
 159            "    add each ROW to TABLE, then query and print the rows that\n"
 160            "    satisfy each CONDITION.\n"
 161            "  query-distinct TABLE [ROW,...] [CONDITION,...] COLUMNS\n"
 162            "    add each ROW to TABLE, then query and print the rows that\n"
 163            "    satisfy each CONDITION and have distinct COLUMNS.\n"
 164            "  parse-schema JSON\n"
 165            "    parse JSON as an OVSDB schema, and re-serialize\n"
 166            "  transact COMMAND\n"
 167            "    execute each specified transactional COMMAND:\n"
 168            "      commit\n"
 169            "      abort\n"
 170            "      insert UUID I J\n"
 171            "      delete UUID\n"
 172            "      modify UUID I J\n"
 173            "      print\n"
 174            "  execute SCHEMA TRANSACTION...\n"
 175            "    executes each TRANSACTION on an initially empty database\n"
 176            "    the specified SCHEMA\n"
 177            "  trigger SCHEMA TRANSACTION...\n"
 178            "    executes each TRANSACTION on an initially empty database\n"
 179            "    the specified SCHEMA.   A TRANSACTION of the form\n"
 180            "    [\"advance\", NUMBER] advances NUMBER milliseconds in\n"
 181            "    simulated time, for causing triggers to time out.\n"
 182            "  idl SERVER [TRANSACTION...]\n"
 183            "    connect to SERVER and dump the contents of the database\n"
 184            "    as seen initially by the IDL implementation and after\n"
 185            "    executing each TRANSACTION.  (Each TRANSACTION must modify\n"
 186            "    the database or this command will hang.)\n",
 187            program_name, program_name);
 188     vlog_usage();
 189     printf("\nOther options:\n"
 190            "  -t, --timeout=SECS          give up after SECS seconds\n"
 191            "  -h, --help                  display this help message\n");
 192     exit(EXIT_SUCCESS);
 193 }
 194 \f
 195 /* Command helper functions. */
 196
 197 static struct json *
 198 parse_json(const char *s)
 199 {
 200     struct json *json = json_from_string(s);
 201     if (json->type == JSON_STRING) {
 202         ovs_fatal(0, "\"%s\": %s", s, json->u.string);
 203     }
 204     return json;
 205 }
 206
 207 static struct json *
 208 unbox_json(struct json *json)
 209 {
 210     if (json->type == JSON_ARRAY && json->u.array.n == 1) {
 211         struct json *inner = json->u.array.elems[0];
 212         json->u.array.elems[0] = NULL;
 213         json_destroy(json);
 214         return inner;
 215     } else {
 216         return json;
 217     }
 218 }
 219
 220 static void
 221 print_and_free_json(struct json *json)
 222 {
 223     char *string = json_to_string(json, JSSF_SORT);
 224     json_destroy(json);
 225     puts(string);
 226     free(string);
 227 }
 228
 229 static void
 230 print_and_free_ovsdb_error(struct ovsdb_error *error)
 231 {
 232     char *string = ovsdb_error_to_string(error);
 233     ovsdb_error_destroy(error);
 234     puts(string);
 235     free(string);
 236 }
 237
 238 static void
 239 check_ovsdb_error(struct ovsdb_error *error)
 240 {
 241     if (error) {
 242         char *s = ovsdb_error_to_string(error);
 243         ovsdb_error_destroy(error);
 244         ovs_fatal(0, "%s", s);
 245     }
 246 }
 247
 248 static void
 249 die_if_error(char *error)
 250 {
 251     if (error) {
 252         ovs_fatal(0, "%s", error);
 253     }
 254 }
 255 \f
 256 /* Command implementations. */
 257
 258 static void
 259 do_log_io(int argc, char *argv[])
 260 {
 261     const char *name = argv[1];
 262     char *mode_string = argv[2];
 263
 264     struct ovsdb_error *error;
 265     enum ovsdb_log_open_mode mode;
 266     struct ovsdb_log *log;
 267     int i;
 268
 269     if (!strcmp(mode_string, "read-only")) {
 270         mode = OVSDB_LOG_READ_ONLY;
 271     } else if (!strcmp(mode_string, "read/write")) {
 272         mode = OVSDB_LOG_READ_WRITE;
 273     } else if (!strcmp(mode_string, "create")) {
 274         mode = OVSDB_LOG_CREATE;
 275     } else {
 276         ovs_fatal(0, "unknown log-io open mode \"%s\"", mode_string);
 277     }
 278
 279     check_ovsdb_error(ovsdb_log_open(name, mode, -1, &log));
 280     printf("%s: open successful\n", name);
 281
 282     for (i = 3; i < argc; i++) {
 283         const char *command = argv[i];
 284         if (!strcmp(command, "read")) {
 285             struct json *json;
 286
 287             error = ovsdb_log_read(log, &json);
 288             if (!error) {
 289                 printf("%s: read: ", name);
 290                 if (json) {
 291                     print_and_free_json(json);
 292                 } else {
 293                     printf("end of log\n");
 294                 }
 295                 continue;
 296             }
 297         } else if (!strncmp(command, "write:", 6)) {
 298             struct json *json = parse_json(command + 6);
 299             error = ovsdb_log_write(log, json);
 300             json_destroy(json);
 301         } else if (!strcmp(command, "commit")) {
 302             error = ovsdb_log_commit(log);
 303         } else {
 304             ovs_fatal(0, "unknown log-io command \"%s\"", command);
 305         }
 306         if (error) {
 307             char *s = ovsdb_error_to_string(error);
 308             printf("%s: %s failed: %s\n", name, command, s);
 309             free(s);
 310             ovsdb_error_destroy(error);
 311         } else {
 312             printf("%s: %s successful\n", name, command);
 313         }
 314     }
 315
 316     ovsdb_log_close(log);
 317 }
 318
 319 static void
 320 do_default_atoms(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
 321 {
 322     int type;
 323
 324     for (type = 0; type < OVSDB_N_TYPES; type++) {
 325         union ovsdb_atom atom;
 326
 327         if (type == OVSDB_TYPE_VOID) {
 328             continue;
 329         }
 330
 331         printf("%s: ", ovsdb_atomic_type_to_string(type));
 332
 333         ovsdb_atom_init_default(&atom, type);
 334         if (!ovsdb_atom_equals(&atom, ovsdb_atom_default(type), type)) {
 335             printf("wrong\n");
 336             exit(1);
 337         }
 338         ovsdb_atom_destroy(&atom, type);
 339
 340         printf("OK\n");
 341     }
 342 }
 343
 344 static void
 345 do_default_data(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
 346 {
 347     unsigned int n_min;
 348     int key, value;
 349
 350     for (n_min = 0; n_min <= 1; n_min++) {
 351         for (key = 0; key < OVSDB_N_TYPES; key++) {
 352             if (key == OVSDB_TYPE_VOID) {
 353                 continue;
 354             }
 355             for (value = 0; value < OVSDB_N_TYPES; value++) {
 356                 struct ovsdb_datum datum;
 357                 struct ovsdb_type type;
 358
 359                 ovsdb_base_type_init(&type.key, key);
 360                 ovsdb_base_type_init(&type.value, value);
 361                 type.n_min = n_min;
 362                 type.n_max = 1;
 363                 assert(ovsdb_type_is_valid(&type));
 364
 365                 printf("key %s, value %s, n_min %u: ",
 366                        ovsdb_atomic_type_to_string(key),
 367                        ovsdb_atomic_type_to_string(value), n_min);
 368
 369                 ovsdb_datum_init_default(&datum, &type);
 370                 if (!ovsdb_datum_equals(&datum, ovsdb_datum_default(&type),
 371                                         &type)) {
 372                     printf("wrong\n");
 373                     exit(1);
 374                 }
 375                 ovsdb_datum_destroy(&datum, &type);
 376                 ovsdb_type_destroy(&type);
 377
 378                 printf("OK\n");
 379             }
 380         }
 381     }
 382 }
 383
 384 static void
 385 do_parse_atomic_type(int argc OVS_UNUSED, char *argv[])
 386 {
 387     enum ovsdb_atomic_type type;
 388     struct json *json;
 389
 390     json = unbox_json(parse_json(argv[1]));
 391     check_ovsdb_error(ovsdb_atomic_type_from_json(&type, json));
 392     json_destroy(json);
 393     print_and_free_json(ovsdb_atomic_type_to_json(type));
 394 }
 395
 396 static void
 397 do_parse_base_type(int argc OVS_UNUSED, char *argv[])
 398 {
 399     struct ovsdb_base_type base;
 400     struct json *json;
 401
 402     json = unbox_json(parse_json(argv[1]));
 403     check_ovsdb_error(ovsdb_base_type_from_json(&base, json));
 404     json_destroy(json);
 405     print_and_free_json(ovsdb_base_type_to_json(&base));
 406     ovsdb_base_type_destroy(&base);
 407 }
 408
 409 static void
 410 do_parse_type(int argc OVS_UNUSED, char *argv[])
 411 {
 412     struct ovsdb_type type;
 413     struct json *json;
 414
 415     json = unbox_json(parse_json(argv[1]));
 416     check_ovsdb_error(ovsdb_type_from_json(&type, json));
 417     json_destroy(json);
 418     print_and_free_json(ovsdb_type_to_json(&type));
 419     ovsdb_type_destroy(&type);
 420 }
 421
 422 static void
 423 do_parse_atoms(int argc, char *argv[])
 424 {
 425     struct ovsdb_base_type base;
 426     struct json *json;
 427     int i;
 428
 429     json = unbox_json(parse_json(argv[1]));
 430     check_ovsdb_error(ovsdb_base_type_from_json(&base, json));
 431     json_destroy(json);
 432
 433     for (i = 2; i < argc; i++) {
 434         struct ovsdb_error *error;
 435         union ovsdb_atom atom;
 436
 437         json = unbox_json(parse_json(argv[i]));
 438         error = ovsdb_atom_from_json(&atom, &base, json, NULL);
 439         json_destroy(json);
 440
 441         if (error) {
 442             print_and_free_ovsdb_error(error);
 443         } else {
 444             print_and_free_json(ovsdb_atom_to_json(&atom, base.type));
 445             ovsdb_atom_destroy(&atom, base.type);
 446         }
 447     }
 448     ovsdb_base_type_destroy(&base);
 449 }
 450
 451 static void
 452 do_parse_atom_strings(int argc, char *argv[])
 453 {
 454     struct ovsdb_base_type base;
 455     struct json *json;
 456     int i;
 457
 458     json = unbox_json(parse_json(argv[1]));
 459     check_ovsdb_error(ovsdb_base_type_from_json(&base, json));
 460     json_destroy(json);
 461
 462     for (i = 2; i < argc; i++) {
 463         union ovsdb_atom atom;
 464         struct ds out;
 465
 466         die_if_error(ovsdb_atom_from_string(&atom, &base, argv[i], NULL));
 467
 468         ds_init(&out);
 469         ovsdb_atom_to_string(&atom, base.type, &out);
 470         puts(ds_cstr(&out));
 471         ds_destroy(&out);
 472
 473         ovsdb_atom_destroy(&atom, base.type);
 474     }
 475     ovsdb_base_type_destroy(&base);
 476 }
 477
 478 static void
 479 do_parse_data__(int argc, char *argv[],
 480                 struct ovsdb_error *
 481                 (*parse)(struct ovsdb_datum *datum,
 482                          const struct ovsdb_type *type,
 483                          const struct json *json,
 484                          struct ovsdb_symbol_table *symtab))
 485 {
 486     struct ovsdb_type type;
 487     struct json *json;
 488     int i;
 489
 490     json = unbox_json(parse_json(argv[1]));
 491     check_ovsdb_error(ovsdb_type_from_json(&type, json));
 492     json_destroy(json);
 493
 494     for (i = 2; i < argc; i++) {
 495         struct ovsdb_datum datum;
 496
 497         json = unbox_json(parse_json(argv[i]));
 498         check_ovsdb_error(parse(&datum, &type, json, NULL));
 499         json_destroy(json);
 500
 501         print_and_free_json(ovsdb_datum_to_json(&datum, &type));
 502
 503         ovsdb_datum_destroy(&datum, &type);
 504     }
 505     ovsdb_type_destroy(&type);
 506 }
 507
 508 static void
 509 do_parse_data(int argc, char *argv[])
 510 {
 511     do_parse_data__(argc, argv, ovsdb_datum_from_json);
 512 }
 513
 514 static void
 515 do_parse_data_strings(int argc, char *argv[])
 516 {
 517     struct ovsdb_type type;
 518     struct json *json;
 519     int i;
 520
 521     json = unbox_json(parse_json(argv[1]));
 522     check_ovsdb_error(ovsdb_type_from_json(&type, json));
 523     json_destroy(json);
 524
 525     for (i = 2; i < argc; i++) {
 526         struct ovsdb_datum datum;
 527         struct ds out;
 528
 529         die_if_error(ovsdb_datum_from_string(&datum, &type, argv[i], NULL));
 530
 531         ds_init(&out);
 532         ovsdb_datum_to_string(&datum, &type, &out);
 533         puts(ds_cstr(&out));
 534         ds_destroy(&out);
 535
 536         ovsdb_datum_destroy(&datum, &type);
 537     }
 538     ovsdb_type_destroy(&type);
 539 }
 540
 541 static enum ovsdb_atomic_type compare_atoms_atomic_type;
 542
 543 static int
 544 compare_atoms(const void *a_, const void *b_)
 545 {
 546     const union ovsdb_atom *a = a_;
 547     const union ovsdb_atom *b = b_;
 548
 549     return ovsdb_atom_compare_3way(a, b, compare_atoms_atomic_type);
 550 }
 551
 552 static void
 553 do_sort_atoms(int argc OVS_UNUSED, char *argv[])
 554 {
 555     struct ovsdb_base_type base;
 556     union ovsdb_atom *atoms;
 557     struct json *json, **json_atoms;
 558     size_t n_atoms;
 559     int i;
 560
 561     json = unbox_json(parse_json(argv[1]));
 562     check_ovsdb_error(ovsdb_base_type_from_json(&base, json));
 563     json_destroy(json);
 564
 565     json = unbox_json(parse_json(argv[2]));
 566     if (json->type != JSON_ARRAY) {
 567         ovs_fatal(0, "second argument must be array");
 568     }
 569
 570     /* Convert JSON atoms to internal representation. */
 571     n_atoms = json->u.array.n;
 572     atoms = xmalloc(n_atoms * sizeof *atoms);
 573     for (i = 0; i < n_atoms; i++) {
 574         check_ovsdb_error(ovsdb_atom_from_json(&atoms[i], &base,
 575                                                json->u.array.elems[i], NULL));
 576     }
 577     json_destroy(json);
 578
 579     /* Sort atoms. */
 580     compare_atoms_atomic_type = base.type;
 581     qsort(atoms, n_atoms, sizeof *atoms, compare_atoms);
 582
 583     /* Convert internal representation back to JSON. */
 584     json_atoms = xmalloc(n_atoms * sizeof *json_atoms);
 585     for (i = 0; i < n_atoms; i++) {
 586         json_atoms[i] = ovsdb_atom_to_json(&atoms[i], base.type);
 587         ovsdb_atom_destroy(&atoms[i], base.type);
 588     }
 589     print_and_free_json(json_array_create(json_atoms, n_atoms));
 590     free(atoms);
 591     ovsdb_base_type_destroy(&base);
 592 }
 593
 594 static void
 595 do_parse_column(int argc OVS_UNUSED, char *argv[])
 596 {
 597     struct ovsdb_column *column;
 598     struct json *json;
 599
 600     json = parse_json(argv[2]);
 601     check_ovsdb_error(ovsdb_column_from_json(json, argv[1], &column));
 602     json_destroy(json);
 603     print_and_free_json(ovsdb_column_to_json(column));
 604     ovsdb_column_destroy(column);
 605 }
 606
 607 static void
 608 do_parse_table(int argc OVS_UNUSED, char *argv[])
 609 {
 610     struct ovsdb_table_schema *ts;
 611     bool default_is_root;
 612     struct json *json;
 613
 614     default_is_root = argc > 3 && !strcmp(argv[3], "true");
 615
 616     json = parse_json(argv[2]);
 617     check_ovsdb_error(ovsdb_table_schema_from_json(json, argv[1], &ts));
 618     json_destroy(json);
 619     print_and_free_json(ovsdb_table_schema_to_json(ts, default_is_root));
 620     ovsdb_table_schema_destroy(ts);
 621 }
 622
 623 static void
 624 do_parse_rows(int argc, char *argv[])
 625 {
 626     struct ovsdb_column_set all_columns;
 627     struct ovsdb_table_schema *ts;
 628     struct ovsdb_table *table;
 629     struct json *json;
 630     int i;
 631
 632     json = unbox_json(parse_json(argv[1]));
 633     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
 634     json_destroy(json);
 635
 636     table = ovsdb_table_create(ts);
 637     ovsdb_column_set_init(&all_columns);
 638     ovsdb_column_set_add_all(&all_columns, table);
 639
 640     for (i = 2; i < argc; i++) {
 641         struct ovsdb_column_set columns;
 642         struct ovsdb_row *row;
 643
 644         ovsdb_column_set_init(&columns);
 645         row = ovsdb_row_create(table);
 646
 647         json = unbox_json(parse_json(argv[i]));
 648         check_ovsdb_error(ovsdb_row_from_json(row, json, NULL, &columns));
 649         json_destroy(json);
 650
 651         print_and_free_json(ovsdb_row_to_json(row, &all_columns));
 652
 653         if (columns.n_columns) {
 654             struct svec names;
 655             size_t j;
 656             char *s;
 657
 658             svec_init(&names);
 659             for (j = 0; j < columns.n_columns; j++) {
 660                 svec_add(&names, columns.columns[j]->name);
 661             }
 662             svec_sort(&names);
 663             s = svec_join(&names, ", ", "");
 664             puts(s);
 665             free(s);
 666             svec_destroy(&names);
 667         } else {
 668             printf("<none>\n");
 669         }
 670
 671         ovsdb_column_set_destroy(&columns);
 672         ovsdb_row_destroy(row);
 673     }
 674
 675     ovsdb_column_set_destroy(&all_columns);
 676     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
 677 }
 678
 679 static void
 680 do_compare_rows(int argc, char *argv[])
 681 {
 682     struct ovsdb_column_set all_columns;
 683     struct ovsdb_table_schema *ts;
 684     struct ovsdb_table *table;
 685     struct ovsdb_row **rows;
 686     struct json *json;
 687     char **names;
 688     int n_rows;
 689     int i, j;
 690
 691     json = unbox_json(parse_json(argv[1]));
 692     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
 693     json_destroy(json);
 694
 695     table = ovsdb_table_create(ts);
 696     ovsdb_column_set_init(&all_columns);
 697     ovsdb_column_set_add_all(&all_columns, table);
 698
 699     n_rows = argc - 2;
 700     rows = xmalloc(sizeof *rows * n_rows);
 701     names = xmalloc(sizeof *names * n_rows);
 702     for (i = 0; i < n_rows; i++) {
 703         rows[i] = ovsdb_row_create(table);
 704
 705         json = parse_json(argv[i + 2]);
 706         if (json->type != JSON_ARRAY || json->u.array.n != 2
 707             || json->u.array.elems[0]->type != JSON_STRING) {
 708             ovs_fatal(0, "\"%s\" does not have expected form "
 709                       "[\"name\", {data}]", argv[i]);
 710         }
 711         names[i] = xstrdup(json->u.array.elems[0]->u.string);
 712         check_ovsdb_error(ovsdb_row_from_json(rows[i], json->u.array.elems[1],
 713                                               NULL, NULL));
 714         json_destroy(json);
 715     }
 716     for (i = 0; i < n_rows; i++) {
 717         uint32_t i_hash = ovsdb_row_hash_columns(rows[i], &all_columns, 0);
 718         for (j = i + 1; j < n_rows; j++) {
 719             uint32_t j_hash = ovsdb_row_hash_columns(rows[j], &all_columns, 0);
 720             if (ovsdb_row_equal_columns(rows[i], rows[j], &all_columns)) {
 721                 printf("%s == %s\n", names[i], names[j]);
 722                 if (i_hash != j_hash) {
 723                     printf("but hash(%s) != hash(%s)\n", names[i], names[j]);
 724                     abort();
 725                 }
 726             } else if (i_hash == j_hash) {
 727                 printf("hash(%s) == hash(%s)\n", names[i], names[j]);
 728             }
 729         }
 730     }
 731     for (i = 0; i < n_rows; i++) {
 732         ovsdb_row_destroy(rows[i]);
 733         free(names[i]);
 734     }
 735     free(rows);
 736     free(names);
 737
 738     ovsdb_column_set_destroy(&all_columns);
 739     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
 740 }
 741
 742 static void
 743 do_parse_conditions(int argc, char *argv[])
 744 {
 745     struct ovsdb_table_schema *ts;
 746     struct json *json;
 747     int exit_code = 0;
 748     int i;
 749
 750     json = unbox_json(parse_json(argv[1]));
 751     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
 752     json_destroy(json);
 753
 754     for (i = 2; i < argc; i++) {
 755         struct ovsdb_condition cnd;
 756         struct ovsdb_error *error;
 757
 758         json = parse_json(argv[i]);
 759         error = ovsdb_condition_from_json(ts, json, NULL, &cnd);
 760         if (!error) {
 761             print_and_free_json(ovsdb_condition_to_json(&cnd));
 762         } else {
 763             char *s = ovsdb_error_to_string(error);
 764             ovs_error(0, "%s", s);
 765             free(s);
 766             ovsdb_error_destroy(error);
 767             exit_code = 1;
 768         }
 769         json_destroy(json);
 770
 771         ovsdb_condition_destroy(&cnd);
 772     }
 773     ovsdb_table_schema_destroy(ts);
 774
 775     exit(exit_code);
 776 }
 777
 778 static void
 779 do_evaluate_conditions(int argc OVS_UNUSED, char *argv[])
 780 {
 781     struct ovsdb_table_schema *ts;
 782     struct ovsdb_table *table;
 783     struct ovsdb_condition *conditions;
 784     size_t n_conditions;
 785     struct ovsdb_row **rows;
 786     size_t n_rows;
 787     struct json *json;
 788     size_t i, j;
 789
 790     /* Parse table schema, create table. */
 791     json = unbox_json(parse_json(argv[1]));
 792     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
 793     json_destroy(json);
 794
 795     table = ovsdb_table_create(ts);
 796
 797     /* Parse conditions. */
 798     json = parse_json(argv[2]);
 799     if (json->type != JSON_ARRAY) {
 800         ovs_fatal(0, "CONDITION argument is not JSON array");
 801     }
 802     n_conditions = json->u.array.n;
 803     conditions = xmalloc(n_conditions * sizeof *conditions);
 804     for (i = 0; i < n_conditions; i++) {
 805         check_ovsdb_error(ovsdb_condition_from_json(ts, json->u.array.elems[i],
 806                                                     NULL, &conditions[i]));
 807     }
 808     json_destroy(json);
 809
 810     /* Parse rows. */
 811     json = parse_json(argv[3]);
 812     if (json->type != JSON_ARRAY) {
 813         ovs_fatal(0, "ROW argument is not JSON array");
 814     }
 815     n_rows = json->u.array.n;
 816     rows = xmalloc(n_rows * sizeof *rows);
 817     for (i = 0; i < n_rows; i++) {
 818         rows[i] = ovsdb_row_create(table);
 819         check_ovsdb_error(ovsdb_row_from_json(rows[i], json->u.array.elems[i],
 820                                               NULL, NULL));
 821     }
 822     json_destroy(json);
 823
 824     for (i = 0; i < n_conditions; i++) {
 825         printf("condition %2zu:", i);
 826         for (j = 0; j < n_rows; j++) {
 827             bool result = ovsdb_condition_evaluate(rows[j], &conditions[i]);
 828             if (j % 5 == 0) {
 829                 putchar(' ');
 830             }
 831             putchar(result ? 'T' : '-');
 832         }
 833         printf("\n");
 834     }
 835
 836     for (i = 0; i < n_conditions; i++) {
 837         ovsdb_condition_destroy(&conditions[i]);
 838     }
 839     free(conditions);
 840     for (i = 0; i < n_rows; i++) {
 841         ovsdb_row_destroy(rows[i]);
 842     }
 843     free(rows);
 844     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
 845 }
 846
 847 static void
 848 do_parse_mutations(int argc, char *argv[])
 849 {
 850     struct ovsdb_table_schema *ts;
 851     struct json *json;
 852     int exit_code = 0;
 853     int i;
 854
 855     json = unbox_json(parse_json(argv[1]));
 856     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
 857     json_destroy(json);
 858
 859     for (i = 2; i < argc; i++) {
 860         struct ovsdb_mutation_set set;
 861         struct ovsdb_error *error;
 862
 863         json = parse_json(argv[i]);
 864         error = ovsdb_mutation_set_from_json(ts, json, NULL, &set);
 865         if (!error) {
 866             print_and_free_json(ovsdb_mutation_set_to_json(&set));
 867         } else {
 868             char *s = ovsdb_error_to_string(error);
 869             ovs_error(0, "%s", s);
 870             free(s);
 871             ovsdb_error_destroy(error);
 872             exit_code = 1;
 873         }
 874         json_destroy(json);
 875
 876         ovsdb_mutation_set_destroy(&set);
 877     }
 878     ovsdb_table_schema_destroy(ts);
 879
 880     exit(exit_code);
 881 }
 882
 883 static void
 884 do_execute_mutations(int argc OVS_UNUSED, char *argv[])
 885 {
 886     struct ovsdb_table_schema *ts;
 887     struct ovsdb_table *table;
 888     struct ovsdb_mutation_set *sets;
 889     size_t n_sets;
 890     struct ovsdb_row **rows;
 891     size_t n_rows;
 892     struct json *json;
 893     size_t i, j;
 894
 895     /* Parse table schema, create table. */
 896     json = unbox_json(parse_json(argv[1]));
 897     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
 898     json_destroy(json);
 899
 900     table = ovsdb_table_create(ts);
 901
 902     /* Parse mutations. */
 903     json = parse_json(argv[2]);
 904     if (json->type != JSON_ARRAY) {
 905         ovs_fatal(0, "MUTATION argument is not JSON array");
 906     }
 907     n_sets = json->u.array.n;
 908     sets = xmalloc(n_sets * sizeof *sets);
 909     for (i = 0; i < n_sets; i++) {
 910         check_ovsdb_error(ovsdb_mutation_set_from_json(ts,
 911                                                        json->u.array.elems[i],
 912                                                        NULL, &sets[i]));
 913     }
 914     json_destroy(json);
 915
 916     /* Parse rows. */
 917     json = parse_json(argv[3]);
 918     if (json->type != JSON_ARRAY) {
 919         ovs_fatal(0, "ROW argument is not JSON array");
 920     }
 921     n_rows = json->u.array.n;
 922     rows = xmalloc(n_rows * sizeof *rows);
 923     for (i = 0; i < n_rows; i++) {
 924         rows[i] = ovsdb_row_create(table);
 925         check_ovsdb_error(ovsdb_row_from_json(rows[i], json->u.array.elems[i],
 926                                               NULL, NULL));
 927     }
 928     json_destroy(json);
 929
 930     for (i = 0; i < n_sets; i++) {
 931         printf("mutation %2zu:\n", i);
 932         for (j = 0; j < n_rows; j++) {
 933             struct ovsdb_error *error;
 934             struct ovsdb_row *row;
 935
 936             row = ovsdb_row_clone(rows[j]);
 937             error = ovsdb_mutation_set_execute(row, &sets[i]);
 938
 939             printf("row %zu: ", j);
 940             if (error) {
 941                 print_and_free_ovsdb_error(error);
 942             } else {
 943                 struct ovsdb_column_set columns;
 944                 struct shash_node *node;
 945
 946                 ovsdb_column_set_init(&columns);
 947                 SHASH_FOR_EACH (node, &ts->columns) {
 948                     struct ovsdb_column *c = node->data;
 949                     if (!ovsdb_datum_equals(&row->fields[c->index],
 950                                             &rows[j]->fields[c->index],
 951                                             &c->type)) {
 952                         ovsdb_column_set_add(&columns, c);
 953                     }
 954                 }
 955                 if (columns.n_columns) {
 956                     print_and_free_json(ovsdb_row_to_json(row, &columns));
 957                 } else {
 958                     printf("no change\n");
 959                 }
 960                 ovsdb_column_set_destroy(&columns);
 961             }
 962             ovsdb_row_destroy(row);
 963         }
 964         printf("\n");
 965     }
 966
 967     for (i = 0; i < n_sets; i++) {
 968         ovsdb_mutation_set_destroy(&sets[i]);
 969     }
 970     free(sets);
 971     for (i = 0; i < n_rows; i++) {
 972         ovsdb_row_destroy(rows[i]);
 973     }
 974     free(rows);
 975     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
 976 }
 977
 978 struct do_query_cbdata {
 979     struct uuid *row_uuids;
 980     int *counts;
 981     size_t n_rows;
 982 };
 983
 984 static bool
 985 do_query_cb(const struct ovsdb_row *row, void *cbdata_)
 986 {
 987     struct do_query_cbdata *cbdata = cbdata_;
 988     size_t i;
 989
 990     for (i = 0; i < cbdata->n_rows; i++) {
 991         if (uuid_equals(ovsdb_row_get_uuid(row), &cbdata->row_uuids[i])) {
 992             cbdata->counts[i]++;
 993         }
 994     }
 995
 996     return true;
 997 }
 998
 999 static void
1000 do_query(int argc OVS_UNUSED, char *argv[])
1001 {
1002     struct do_query_cbdata cbdata;
1003     struct ovsdb_table_schema *ts;
1004     struct ovsdb_table *table;
1005     struct json *json;
1006     int exit_code = 0;
1007     size_t i;
1008
1009     /* Parse table schema, create table. */
1010     json = unbox_json(parse_json(argv[1]));
1011     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
1012     json_destroy(json);
1013
1014     table = ovsdb_table_create(ts);
1015
1016     /* Parse rows, add to table. */
1017     json = parse_json(argv[2]);
1018     if (json->type != JSON_ARRAY) {
1019         ovs_fatal(0, "ROW argument is not JSON array");
1020     }
1021     cbdata.n_rows = json->u.array.n;
1022     cbdata.row_uuids = xmalloc(cbdata.n_rows * sizeof *cbdata.row_uuids);
1023     cbdata.counts = xmalloc(cbdata.n_rows * sizeof *cbdata.counts);
1024     for (i = 0; i < cbdata.n_rows; i++) {
1025         struct ovsdb_row *row = ovsdb_row_create(table);
1026         uuid_generate(ovsdb_row_get_uuid_rw(row));
1027         check_ovsdb_error(ovsdb_row_from_json(row, json->u.array.elems[i],
1028                                               NULL, NULL));
1029         if (ovsdb_table_get_row(table, ovsdb_row_get_uuid(row))) {
1030             ovs_fatal(0, "duplicate UUID "UUID_FMT" in table",
1031                       UUID_ARGS(ovsdb_row_get_uuid(row)));
1032         }
1033         cbdata.row_uuids[i] = *ovsdb_row_get_uuid(row);
1034         ovsdb_table_put_row(table, row);
1035     }
1036     json_destroy(json);
1037
1038     /* Parse conditions and execute queries. */
1039     json = parse_json(argv[3]);
1040     if (json->type != JSON_ARRAY) {
1041         ovs_fatal(0, "CONDITION argument is not JSON array");
1042     }
1043     for (i = 0; i < json->u.array.n; i++) {
1044         struct ovsdb_condition cnd;
1045         size_t j;
1046
1047         check_ovsdb_error(ovsdb_condition_from_json(ts, json->u.array.elems[i],
1048                                                     NULL, &cnd));
1049
1050         memset(cbdata.counts, 0, cbdata.n_rows * sizeof *cbdata.counts);
1051         ovsdb_query(table, &cnd, do_query_cb, &cbdata);
1052
1053         printf("query %2zu:", i);
1054         for (j = 0; j < cbdata.n_rows; j++) {
1055             if (j % 5 == 0) {
1056                 putchar(' ');
1057             }
1058             if (cbdata.counts[j]) {
1059                 printf("%d", cbdata.counts[j]);
1060                 if (cbdata.counts[j] > 1) {
1061                     /* Dup! */
1062                     exit_code = 1;
1063                 }
1064             } else {
1065                 putchar('-');
1066             }
1067         }
1068         putchar('\n');
1069
1070         ovsdb_condition_destroy(&cnd);
1071     }
1072     json_destroy(json);
1073
1074     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
1075
1076     exit(exit_code);
1077 }
1078
1079 struct do_query_distinct_class {
1080     struct ovsdb_row *example;
1081     int count;
1082 };
1083
1084 struct do_query_distinct_row {
1085     struct uuid uuid;
1086     struct do_query_distinct_class *class;
1087 };
1088
1089 static void
1090 do_query_distinct(int argc OVS_UNUSED, char *argv[])
1091 {
1092     struct ovsdb_column_set columns;
1093     struct ovsdb_table_schema *ts;
1094     struct ovsdb_table *table;
1095     struct do_query_distinct_row *rows;
1096     size_t n_rows;
1097     struct do_query_distinct_class *classes;
1098     size_t n_classes;
1099     struct json *json;
1100     int exit_code = 0;
1101     size_t i;
1102
1103     /* Parse table schema, create table. */
1104     json = unbox_json(parse_json(argv[1]));
1105     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
1106     json_destroy(json);
1107
1108     table = ovsdb_table_create(ts);
1109
1110     /* Parse column set. */
1111     json = parse_json(argv[4]);
1112     check_ovsdb_error(ovsdb_column_set_from_json(json, table->schema,
1113                                                  &columns));
1114     json_destroy(json);
1115
1116     /* Parse rows, add to table. */
1117     json = parse_json(argv[2]);
1118     if (json->type != JSON_ARRAY) {
1119         ovs_fatal(0, "ROW argument is not JSON array");
1120     }
1121     n_rows = json->u.array.n;
1122     rows = xmalloc(n_rows * sizeof *rows);
1123     classes = xmalloc(n_rows * sizeof *classes);
1124     n_classes = 0;
1125     for (i = 0; i < n_rows; i++) {
1126         struct ovsdb_row *row;
1127         size_t j;
1128
1129         /* Parse row. */
1130         row = ovsdb_row_create(table);
1131         uuid_generate(ovsdb_row_get_uuid_rw(row));
1132         check_ovsdb_error(ovsdb_row_from_json(row, json->u.array.elems[i],
1133                                               NULL, NULL));
1134
1135         /* Initialize row and find equivalence class. */
1136         rows[i].uuid = *ovsdb_row_get_uuid(row);
1137         rows[i].class = NULL;
1138         for (j = 0; j < n_classes; j++) {
1139             if (ovsdb_row_equal_columns(row, classes[j].example, &columns)) {
1140                 rows[i].class = &classes[j];
1141                 break;
1142             }
1143         }
1144         if (!rows[i].class) {
1145             rows[i].class = &classes[n_classes];
1146             classes[n_classes].example = ovsdb_row_clone(row);
1147             n_classes++;
1148         }
1149
1150         /* Add row to table. */
1151         if (ovsdb_table_get_row(table, ovsdb_row_get_uuid(row))) {
1152             ovs_fatal(0, "duplicate UUID "UUID_FMT" in table",
1153                       UUID_ARGS(ovsdb_row_get_uuid(row)));
1154         }
1155         ovsdb_table_put_row(table, row);
1156
1157     }
1158     json_destroy(json);
1159
1160     /* Parse conditions and execute queries. */
1161     json = parse_json(argv[3]);
1162     if (json->type != JSON_ARRAY) {
1163         ovs_fatal(0, "CONDITION argument is not JSON array");
1164     }
1165     for (i = 0; i < json->u.array.n; i++) {
1166         struct ovsdb_row_set results;
1167         struct ovsdb_condition cnd;
1168         size_t j;
1169
1170         check_ovsdb_error(ovsdb_condition_from_json(ts, json->u.array.elems[i],
1171                                                     NULL, &cnd));
1172
1173         for (j = 0; j < n_classes; j++) {
1174             classes[j].count = 0;
1175         }
1176         ovsdb_row_set_init(&results);
1177         ovsdb_query_distinct(table, &cnd, &columns, &results);
1178         for (j = 0; j < results.n_rows; j++) {
1179             size_t k;
1180
1181             for (k = 0; k < n_rows; k++) {
1182                 if (uuid_equals(ovsdb_row_get_uuid(results.rows[j]),
1183                                 &rows[k].uuid)) {
1184                     rows[k].class->count++;
1185                 }
1186             }
1187         }
1188         ovsdb_row_set_destroy(&results);
1189
1190         printf("query %2zu:", i);
1191         for (j = 0; j < n_rows; j++) {
1192             int count = rows[j].class->count;
1193
1194             if (j % 5 == 0) {
1195                 putchar(' ');
1196             }
1197             if (count > 1) {
1198                 /* Dup! */
1199                 printf("%d", count);
1200                 exit_code = 1;
1201             } else if (count == 1) {
1202                 putchar("abcdefghijklmnopqrstuvwxyz"[rows[j].class - classes]);
1203             } else {
1204                 putchar('-');
1205             }
1206         }
1207         putchar('\n');
1208
1209         ovsdb_condition_destroy(&cnd);
1210     }
1211     json_destroy(json);
1212
1213     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
1214
1215     exit(exit_code);
1216 }
1217
1218 static void
1219 do_parse_schema(int argc OVS_UNUSED, char *argv[])
1220 {
1221     struct ovsdb_schema *schema;
1222     struct json *json;
1223
1224     json = parse_json(argv[1]);
1225     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
1226     json_destroy(json);
1227     print_and_free_json(ovsdb_schema_to_json(schema));
1228     ovsdb_schema_destroy(schema);
1229 }
1230
1231 static void
1232 do_execute(int argc OVS_UNUSED, char *argv[])
1233 {
1234     struct ovsdb_schema *schema;
1235     struct json *json;
1236     struct ovsdb *db;
1237     int i;
1238
1239     /* Create database. */
1240     json = parse_json(argv[1]);
1241     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
1242     json_destroy(json);
1243     db = ovsdb_create(schema);
1244
1245     for (i = 2; i < argc; i++) {
1246         struct json *params, *result;
1247         char *s;
1248
1249         params = parse_json(argv[i]);
1250         result = ovsdb_execute(db, params, 0, NULL);
1251         s = json_to_string(result, JSSF_SORT);
1252         printf("%s\n", s);
1253         free(s);
1254         json_destroy(params);
1255         json_destroy(result);
1256     }
1257
1258     ovsdb_destroy(db);
1259 }
1260
1261 struct test_trigger {
1262     struct ovsdb_trigger trigger;
1263     int number;
1264 };
1265
1266 static void
1267 do_trigger_dump(struct test_trigger *t, long long int now, const char *title)
1268 {
1269     struct json *result;
1270     char *s;
1271
1272     result = ovsdb_trigger_steal_result(&t->trigger);
1273     s = json_to_string(result, JSSF_SORT);
1274     printf("t=%lld: trigger %d (%s): %s\n", now, t->number, title, s);
1275     free(s);
1276     json_destroy(result);
1277     ovsdb_trigger_destroy(&t->trigger);
1278     free(t);
1279 }
1280
1281 static void
1282 do_trigger(int argc OVS_UNUSED, char *argv[])
1283 {
1284     struct ovsdb_schema *schema;
1285     struct list completions;
1286     struct json *json;
1287     struct ovsdb *db;
1288     long long int now;
1289     int number;
1290     int i;
1291
1292     /* Create database. */
1293     json = parse_json(argv[1]);
1294     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
1295     json_destroy(json);
1296     db = ovsdb_create(schema);
1297
1298     list_init(&completions);
1299     now = 0;
1300     number = 0;
1301     for (i = 2; i < argc; i++) {
1302         struct json *params = parse_json(argv[i]);
1303         if (params->type == JSON_ARRAY
1304             && json_array(params)->n == 2
1305             && json_array(params)->elems[0]->type == JSON_STRING
1306             && !strcmp(json_string(json_array(params)->elems[0]), "advance")
1307             && json_array(params)->elems[1]->type == JSON_INTEGER) {
1308             now += json_integer(json_array(params)->elems[1]);
1309             json_destroy(params);
1310         } else {
1311             struct test_trigger *t = xmalloc(sizeof *t);
1312             ovsdb_trigger_init(db, &t->trigger, params, &completions, now);
1313             t->number = number++;
1314             if (ovsdb_trigger_is_complete(&t->trigger)) {
1315                 do_trigger_dump(t, now, "immediate");
1316             } else {
1317                 printf("t=%lld: new trigger %d\n", now, t->number);
1318             }
1319         }
1320
1321         ovsdb_trigger_run(db, now);
1322         while (!list_is_empty(&completions)) {
1323             do_trigger_dump(CONTAINER_OF(list_pop_front(&completions),
1324                                          struct test_trigger, trigger.node),
1325                             now, "delayed");
1326         }
1327
1328         ovsdb_trigger_wait(db, now);
1329         poll_immediate_wake();
1330         poll_block();
1331     }
1332
1333     ovsdb_destroy(db);
1334 }
1335
1336 static void
1337 do_help(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1338 {
1339     usage();
1340 }
1341 \f
1342 /* "transact" command. */
1343
1344 static struct ovsdb *do_transact_db;
1345 static struct ovsdb_txn *do_transact_txn;
1346 static struct ovsdb_table *do_transact_table;
1347
1348 static void
1349 do_transact_commit(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1350 {
1351     ovsdb_error_destroy(ovsdb_txn_commit(do_transact_txn, false));
1352     do_transact_txn = NULL;
1353 }
1354
1355 static void
1356 do_transact_abort(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1357 {
1358     ovsdb_txn_abort(do_transact_txn);
1359     do_transact_txn = NULL;
1360 }
1361
1362 static void
1363 uuid_from_integer(int integer, struct uuid *uuid)
1364 {
1365     uuid_zero(uuid);
1366     uuid->parts[3] = integer;
1367 }
1368
1369 static const struct ovsdb_row *
1370 do_transact_find_row(const char *uuid_string)
1371 {
1372     const struct ovsdb_row *row;
1373     struct uuid uuid;
1374
1375     uuid_from_integer(atoi(uuid_string), &uuid);
1376     row = ovsdb_table_get_row(do_transact_table, &uuid);
1377     if (!row) {
1378         ovs_fatal(0, "table does not contain row with UUID "UUID_FMT,
1379                   UUID_ARGS(&uuid));
1380     }
1381     return row;
1382 }
1383
1384 static void
1385 do_transact_set_integer(struct ovsdb_row *row, const char *column_name,
1386                         int integer)
1387 {
1388     if (integer != -1) {
1389         const struct ovsdb_column *column;
1390
1391         column = ovsdb_table_schema_get_column(do_transact_table->schema,
1392                                                column_name);
1393         row->fields[column->index].keys[0].integer = integer;
1394     }
1395 }
1396
1397 static int
1398 do_transact_get_integer(const struct ovsdb_row *row, const char *column_name)
1399 {
1400     const struct ovsdb_column *column;
1401
1402     column = ovsdb_table_schema_get_column(do_transact_table->schema,
1403                                            column_name);
1404     return row->fields[column->index].keys[0].integer;
1405 }
1406
1407 static void
1408 do_transact_set_i_j(struct ovsdb_row *row,
1409                     const char *i_string, const char *j_string)
1410 {
1411     do_transact_set_integer(row, "i", atoi(i_string));
1412     do_transact_set_integer(row, "j", atoi(j_string));
1413 }
1414
1415 static void
1416 do_transact_insert(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1417 {
1418     struct ovsdb_row *row;
1419     struct uuid *uuid;
1420
1421     row = ovsdb_row_create(do_transact_table);
1422
1423     /* Set UUID. */
1424     uuid = ovsdb_row_get_uuid_rw(row);
1425     uuid_from_integer(atoi(argv[1]), uuid);
1426     if (ovsdb_table_get_row(do_transact_table, uuid)) {
1427         ovs_fatal(0, "table already contains row with UUID "UUID_FMT,
1428                   UUID_ARGS(uuid));
1429     }
1430
1431     do_transact_set_i_j(row, argv[2], argv[3]);
1432
1433     /* Insert row. */
1434     ovsdb_txn_row_insert(do_transact_txn, row);
1435 }
1436
1437 static void
1438 do_transact_delete(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1439 {
1440     const struct ovsdb_row *row = do_transact_find_row(argv[1]);
1441     ovsdb_txn_row_delete(do_transact_txn, row);
1442 }
1443
1444 static void
1445 do_transact_modify(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1446 {
1447     const struct ovsdb_row *row_ro;
1448     struct ovsdb_row *row_rw;
1449
1450     row_ro = do_transact_find_row(argv[1]);
1451     row_rw = ovsdb_txn_row_modify(do_transact_txn, row_ro);
1452     do_transact_set_i_j(row_rw, argv[2], argv[3]);
1453 }
1454
1455 static int
1456 compare_rows_by_uuid(const void *a_, const void *b_)
1457 {
1458     struct ovsdb_row *const *ap = a_;
1459     struct ovsdb_row *const *bp = b_;
1460
1461     return uuid_compare_3way(ovsdb_row_get_uuid(*ap), ovsdb_row_get_uuid(*bp));
1462 }
1463
1464 static void
1465 do_transact_print(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1466 {
1467     const struct ovsdb_row **rows;
1468     const struct ovsdb_row *row;
1469     size_t n_rows;
1470     size_t i;
1471
1472     n_rows = hmap_count(&do_transact_table->rows);
1473     rows = xmalloc(n_rows * sizeof *rows);
1474     i = 0;
1475     HMAP_FOR_EACH (row, hmap_node, &do_transact_table->rows) {
1476         rows[i++] = row;
1477     }
1478     assert(i == n_rows);
1479
1480     qsort(rows, n_rows, sizeof *rows, compare_rows_by_uuid);
1481
1482     for (i = 0; i < n_rows; i++) {
1483         printf("\n%"PRId32": i=%d, j=%d",
1484                ovsdb_row_get_uuid(rows[i])->parts[3],
1485                do_transact_get_integer(rows[i], "i"),
1486                do_transact_get_integer(rows[i], "j"));
1487     }
1488
1489     free(rows);
1490 }
1491
1492 static void
1493 do_transact(int argc, char *argv[])
1494 {
1495     static const struct command do_transact_commands[] = {
1496         { "commit", 0, 0, do_transact_commit },
1497         { "abort", 0, 0, do_transact_abort },
1498         { "insert", 2, 3, do_transact_insert },
1499         { "delete", 1, 1, do_transact_delete },
1500         { "modify", 2, 3, do_transact_modify },
1501         { "print", 0, 0, do_transact_print },
1502         { NULL, 0, 0, NULL },
1503     };
1504
1505     struct ovsdb_schema *schema;
1506     struct json *json;
1507     int i;
1508
1509     /* Create table. */
1510     json = parse_json("{\"name\": \"testdb\", "
1511                       " \"tables\": "
1512                       "  {\"mytable\": "
1513                       "    {\"columns\": "
1514                       "      {\"i\": {\"type\": \"integer\"}, "
1515                       "       \"j\": {\"type\": \"integer\"}}}}}");
1516     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
1517     json_destroy(json);
1518     do_transact_db = ovsdb_create(schema);
1519     do_transact_table = ovsdb_get_table(do_transact_db, "mytable");
1520     assert(do_transact_table != NULL);
1521
1522     for (i = 1; i < argc; i++) {
1523         struct json *command;
1524         size_t n_args;
1525         char **args;
1526         int j;
1527
1528         command = parse_json(argv[i]);
1529         if (command->type != JSON_ARRAY) {
1530             ovs_fatal(0, "transaction %d must be JSON array "
1531                       "with at least 1 element", i);
1532         }
1533
1534         n_args = command->u.array.n;
1535         args = xmalloc((n_args + 1) * sizeof *args);
1536         for (j = 0; j < n_args; j++) {
1537             struct json *s = command->u.array.elems[j];
1538             if (s->type != JSON_STRING) {
1539                 ovs_fatal(0, "transaction %d argument %d must be JSON string",
1540                           i, j);
1541             }
1542             args[j] = xstrdup(json_string(s));
1543         }
1544         args[n_args] = NULL;
1545
1546         if (!do_transact_txn) {
1547             do_transact_txn = ovsdb_txn_create(do_transact_db);
1548         }
1549
1550         for (j = 0; j < n_args; j++) {
1551             if (j) {
1552                 putchar(' ');
1553             }
1554             fputs(args[j], stdout);
1555         }
1556         fputs(":", stdout);
1557         run_command(n_args, args, do_transact_commands);
1558         putchar('\n');
1559
1560         for (j = 0; j < n_args; j++) {
1561             free(args[j]);
1562         }
1563         free(args);
1564         json_destroy(command);
1565     }
1566     ovsdb_txn_abort(do_transact_txn);
1567     ovsdb_destroy(do_transact_db); /* Also destroys 'schema'. */
1568 }
1569
1570 static int
1571 compare_link1(const void *a_, const void *b_)
1572 {
1573     const struct idltest_link1 *const *ap = a_;
1574     const struct idltest_link1 *const *bp = b_;
1575     const struct idltest_link1 *a = *ap;
1576     const struct idltest_link1 *b = *bp;
1577
1578     return a->i < b->i ? -1 : a->i > b->i;
1579 }
1580
1581 static void
1582 print_idl(struct ovsdb_idl *idl, int step)
1583 {
1584     const struct idltest_simple *s;
1585     const struct idltest_link1 *l1;
1586     const struct idltest_link2 *l2;
1587     int n = 0;
1588
1589     IDLTEST_SIMPLE_FOR_EACH (s, idl) {
1590         size_t i;
1591
1592         printf("%03d: i=%"PRId64" r=%g b=%s s=%s u="UUID_FMT" ia=[",
1593                step, s->i, s->r, s->b ? "true" : "false",
1594                s->s, UUID_ARGS(&s->u));
1595         for (i = 0; i < s->n_ia; i++) {
1596             printf("%s%"PRId64, i ? " " : "", s->ia[i]);
1597         }
1598         printf("] ra=[");
1599         for (i = 0; i < s->n_ra; i++) {
1600             printf("%s%g", i ? " " : "", s->ra[i]);
1601         }
1602         printf("] ba=[");
1603         for (i = 0; i < s->n_ba; i++) {
1604             printf("%s%s", i ? " " : "", s->ba[i] ? "true" : "false");
1605         }
1606         printf("] sa=[");
1607         for (i = 0; i < s->n_sa; i++) {
1608             printf("%s%s", i ? " " : "", s->sa[i]);
1609         }
1610         printf("] ua=[");
1611         for (i = 0; i < s->n_ua; i++) {
1612             printf("%s"UUID_FMT, i ? " " : "", UUID_ARGS(&s->ua[i]));
1613         }
1614         printf("] uuid="UUID_FMT"\n", UUID_ARGS(&s->header_.uuid));
1615         n++;
1616     }
1617     IDLTEST_LINK1_FOR_EACH (l1, idl) {
1618         struct idltest_link1 **links;
1619         size_t i;
1620
1621         printf("%03d: i=%"PRId64" k=", step, l1->i);
1622         if (l1->k) {
1623             printf("%"PRId64, l1->k->i);
1624         }
1625         printf(" ka=[");
1626         links = xmemdup(l1->ka, l1->n_ka * sizeof *l1->ka);
1627         qsort(links, l1->n_ka, sizeof *links, compare_link1);
1628         for (i = 0; i < l1->n_ka; i++) {
1629             printf("%s%"PRId64, i ? " " : "", links[i]->i);
1630         }
1631         free(links);
1632         printf("] l2=");
1633         if (l1->l2) {
1634             printf("%"PRId64, l1->l2->i);
1635         }
1636         printf(" uuid="UUID_FMT"\n", UUID_ARGS(&l1->header_.uuid));
1637         n++;
1638     }
1639     IDLTEST_LINK2_FOR_EACH (l2, idl) {
1640         printf("%03d: i=%"PRId64" l1=", step, l2->i);
1641         if (l2->l1) {
1642             printf("%"PRId64, l2->l1->i);
1643         }
1644         printf(" uuid="UUID_FMT"\n", UUID_ARGS(&l2->header_.uuid));
1645         n++;
1646     }
1647     if (!n) {
1648         printf("%03d: empty\n", step);
1649     }
1650 }
1651
1652 static void
1653 parse_uuids(const struct json *json, struct ovsdb_symbol_table *symtab,
1654             size_t *n)
1655 {
1656     struct uuid uuid;
1657
1658     if (json->type == JSON_STRING && uuid_from_string(&uuid, json->u.string)) {
1659         char *name = xasprintf("#%zu#", *n);
1660         fprintf(stderr, "%s = "UUID_FMT"\n", name, UUID_ARGS(&uuid));
1661         ovsdb_symbol_table_put(symtab, name, &uuid, false);
1662         free(name);
1663         *n += 1;
1664     } else if (json->type == JSON_ARRAY) {
1665         size_t i;
1666
1667         for (i = 0; i < json->u.array.n; i++) {
1668             parse_uuids(json->u.array.elems[i], symtab, n);
1669         }
1670     } else if (json->type == JSON_OBJECT) {
1671         const struct shash_node *node;
1672
1673         SHASH_FOR_EACH (node, json_object(json)) {
1674             parse_uuids(node->data, symtab, n);
1675         }
1676     }
1677 }
1678
1679 static void
1680 substitute_uuids(struct json *json, const struct ovsdb_symbol_table *symtab)
1681 {
1682     if (json->type == JSON_STRING) {
1683         const struct ovsdb_symbol *symbol;
1684
1685         symbol = ovsdb_symbol_table_get(symtab, json->u.string);
1686         if (symbol) {
1687             free(json->u.string);
1688             json->u.string = xasprintf(UUID_FMT, UUID_ARGS(&symbol->uuid));
1689         }
1690     } else if (json->type == JSON_ARRAY) {
1691         size_t i;
1692
1693         for (i = 0; i < json->u.array.n; i++) {
1694             substitute_uuids(json->u.array.elems[i], symtab);
1695         }
1696     } else if (json->type == JSON_OBJECT) {
1697         const struct shash_node *node;
1698
1699         SHASH_FOR_EACH (node, json_object(json)) {
1700             substitute_uuids(node->data, symtab);
1701         }
1702     }
1703 }
1704
1705 static const struct idltest_simple *
1706 idltest_find_simple(struct ovsdb_idl *idl, int i)
1707 {
1708     const struct idltest_simple *s;
1709
1710     IDLTEST_SIMPLE_FOR_EACH (s, idl) {
1711         if (s->i == i) {
1712             return s;
1713         }
1714     }
1715     return NULL;
1716 }
1717
1718 static void
1719 idl_set(struct ovsdb_idl *idl, char *commands, int step)
1720 {
1721     char *cmd, *save_ptr1 = NULL;
1722     struct ovsdb_idl_txn *txn;
1723     enum ovsdb_idl_txn_status status;
1724     bool increment = false;
1725
1726     txn = ovsdb_idl_txn_create(idl);
1727     for (cmd = strtok_r(commands, ",", &save_ptr1); cmd;
1728          cmd = strtok_r(NULL, ",", &save_ptr1)) {
1729         char *save_ptr2 = NULL;
1730         char *name, *arg1, *arg2, *arg3;
1731
1732         name = strtok_r(cmd, " ", &save_ptr2);
1733         arg1 = strtok_r(NULL, " ", &save_ptr2);
1734         arg2 = strtok_r(NULL, " ", &save_ptr2);
1735         arg3 = strtok_r(NULL, " ", &save_ptr2);
1736
1737         if (!strcmp(name, "set")) {
1738             const struct idltest_simple *s;
1739
1740             if (!arg3) {
1741                 ovs_fatal(0, "\"set\" command requires 3 arguments");
1742             }
1743
1744             s = idltest_find_simple(idl, atoi(arg1));
1745             if (!s) {
1746                 ovs_fatal(0, "\"set\" command asks for nonexistent "
1747                           "i=%d", atoi(arg1));
1748             }
1749
1750             if (!strcmp(arg2, "b")) {
1751                 idltest_simple_set_b(s, atoi(arg3));
1752             } else if (!strcmp(arg2, "s")) {
1753                 idltest_simple_set_s(s, arg3);
1754             } else if (!strcmp(arg2, "u")) {
1755                 struct uuid uuid;
1756                 if (!uuid_from_string(&uuid, arg3)) {
1757                     ovs_fatal(0, "\"%s\" is not a valid UUID", arg3);
1758                 }
1759                 idltest_simple_set_u(s, uuid);
1760             } else if (!strcmp(arg2, "r")) {
1761                 idltest_simple_set_r(s, atof(arg3));
1762             } else {
1763                 ovs_fatal(0, "\"set\" command asks for unknown column %s",
1764                           arg2);
1765             }
1766         } else if (!strcmp(name, "insert")) {
1767             struct idltest_simple *s;
1768
1769             if (!arg1 || arg2) {
1770                 ovs_fatal(0, "\"set\" command requires 1 argument");
1771             }
1772
1773             s = idltest_simple_insert(txn);
1774             idltest_simple_set_i(s, atoi(arg1));
1775         } else if (!strcmp(name, "delete")) {
1776             const struct idltest_simple *s;
1777
1778             if (!arg1 || arg2) {
1779                 ovs_fatal(0, "\"set\" command requires 1 argument");
1780             }
1781
1782             s = idltest_find_simple(idl, atoi(arg1));
1783             if (!s) {
1784                 ovs_fatal(0, "\"set\" command asks for nonexistent "
1785                           "i=%d", atoi(arg1));
1786             }
1787             idltest_simple_delete(s);
1788         } else if (!strcmp(name, "verify")) {
1789             const struct idltest_simple *s;
1790
1791             if (!arg2 || arg3) {
1792                 ovs_fatal(0, "\"verify\" command requires 2 arguments");
1793             }
1794
1795             s = idltest_find_simple(idl, atoi(arg1));
1796             if (!s) {
1797                 ovs_fatal(0, "\"verify\" command asks for nonexistent "
1798                           "i=%d", atoi(arg1));
1799             }
1800
1801             if (!strcmp(arg2, "i")) {
1802                 idltest_simple_verify_i(s);
1803             } else if (!strcmp(arg2, "b")) {
1804                 idltest_simple_verify_b(s);
1805             } else if (!strcmp(arg2, "s")) {
1806                 idltest_simple_verify_s(s);
1807             } else if (!strcmp(arg2, "u")) {
1808                 idltest_simple_verify_s(s);
1809             } else if (!strcmp(arg2, "r")) {
1810                 idltest_simple_verify_r(s);
1811             } else {
1812                 ovs_fatal(0, "\"verify\" command asks for unknown column %s",
1813                           arg2);
1814             }
1815         } else if (!strcmp(name, "increment")) {
1816             if (!arg2 || arg3) {
1817                 ovs_fatal(0, "\"increment\" command requires 2 arguments");
1818             }
1819             ovsdb_idl_txn_increment(txn, arg1, arg2, NULL);
1820             increment = true;
1821         } else {
1822             ovs_fatal(0, "unknown command %s", name);
1823         }
1824     }
1825
1826     status = ovsdb_idl_txn_commit_block(txn);
1827     printf("%03d: commit, status=%s",
1828            step, ovsdb_idl_txn_status_to_string(status));
1829     if (increment) {
1830         printf(", increment=%"PRId64,
1831                ovsdb_idl_txn_get_increment_new_value(txn));
1832     }
1833     putchar('\n');
1834     ovsdb_idl_txn_destroy(txn);
1835 }
1836
1837 static void
1838 do_idl(int argc, char *argv[])
1839 {
1840     struct jsonrpc *rpc;
1841     struct ovsdb_idl *idl;
1842     unsigned int seqno = 0;
1843     struct ovsdb_symbol_table *symtab;
1844     size_t n_uuids = 0;
1845     int step = 0;
1846     int error;
1847     int i;
1848
1849     idltest_init();
1850
1851     idl = ovsdb_idl_create(argv[1], &idltest_idl_class, true);
1852     if (argc > 2) {
1853         struct stream *stream;
1854
1855         error = stream_open_block(jsonrpc_stream_open(argv[1], &stream),
1856                                   &stream);
1857         if (error) {
1858             ovs_fatal(error, "failed to connect to \"%s\"", argv[1]);
1859         }
1860         rpc = jsonrpc_open(stream);
1861     } else {
1862         rpc = NULL;
1863     }
1864
1865     setvbuf(stdout, NULL, _IOLBF, 0);
1866
1867     symtab = ovsdb_symbol_table_create();
1868     for (i = 2; i < argc; i++) {
1869         char *arg = argv[i];
1870         struct jsonrpc_msg *request, *reply;
1871
1872         if (*arg == '+') {
1873             /* The previous transaction didn't change anything. */
1874             arg++;
1875         } else {
1876             /* Wait for update. */
1877             while (ovsdb_idl_get_seqno(idl) == seqno && !ovsdb_idl_run(idl)) {
1878                 jsonrpc_run(rpc);
1879
1880                 ovsdb_idl_wait(idl);
1881                 jsonrpc_wait(rpc);
1882                 poll_block();
1883             }
1884
1885             /* Print update. */
1886             print_idl(idl, step++);
1887         }
1888         seqno = ovsdb_idl_get_seqno(idl);
1889
1890         if (!strcmp(arg, "reconnect")) {
1891             printf("%03d: reconnect\n", step++);
1892             ovsdb_idl_force_reconnect(idl);
1893         } else if (arg[0] != '[') {
1894             idl_set(idl, arg, step++);
1895         } else {
1896             struct json *json = parse_json(arg);
1897             substitute_uuids(json, symtab);
1898             request = jsonrpc_create_request("transact", json, NULL);
1899             error = jsonrpc_transact_block(rpc, request, &reply);
1900             if (error) {
1901                 ovs_fatal(error, "jsonrpc transaction failed");
1902             }
1903             printf("%03d: ", step++);
1904             if (reply->result) {
1905                 parse_uuids(reply->result, symtab, &n_uuids);
1906             }
1907             json_destroy(reply->id);
1908             reply->id = NULL;
1909             print_and_free_json(jsonrpc_msg_to_json(reply));
1910         }
1911     }
1912     ovsdb_symbol_table_destroy(symtab);
1913
1914     if (rpc) {
1915         jsonrpc_close(rpc);
1916     }
1917     while (ovsdb_idl_get_seqno(idl) == seqno && !ovsdb_idl_run(idl)) {
1918         ovsdb_idl_wait(idl);
1919         poll_block();
1920     }
1921     print_idl(idl, step++);
1922     ovsdb_idl_destroy(idl);
1923     printf("%03d: done\n", step);
1924 }
1925
1926 static struct command all_commands[] = {
1927     { "log-io", 2, INT_MAX, do_log_io },
1928     { "default-atoms", 0, 0, do_default_atoms },
1929     { "default-data", 0, 0, do_default_data },
1930     { "parse-atomic-type", 1, 1, do_parse_atomic_type },
1931     { "parse-base-type", 1, 1, do_parse_base_type },
1932     { "parse-type", 1, 1, do_parse_type },
1933     { "parse-atoms", 2, INT_MAX, do_parse_atoms },
1934     { "parse-atom-strings", 2, INT_MAX, do_parse_atom_strings },
1935     { "parse-data", 2, INT_MAX, do_parse_data },
1936     { "parse-data-strings", 2, INT_MAX, do_parse_data_strings },
1937     { "sort-atoms", 2, 2, do_sort_atoms },
1938     { "parse-column", 2, 2, do_parse_column },
1939     { "parse-table", 2, 3, do_parse_table },
1940     { "parse-rows", 2, INT_MAX, do_parse_rows },
1941     { "compare-rows", 2, INT_MAX, do_compare_rows },
1942     { "parse-conditions", 2, INT_MAX, do_parse_conditions },
1943     { "evaluate-conditions", 3, 3, do_evaluate_conditions },
1944     { "parse-mutations", 2, INT_MAX, do_parse_mutations },
1945     { "execute-mutations", 3, 3, do_execute_mutations },
1946     { "query", 3, 3, do_query },
1947     { "query-distinct", 4, 4, do_query_distinct },
1948     { "transact", 1, INT_MAX, do_transact },
1949     { "parse-schema", 1, 1, do_parse_schema },
1950     { "execute", 2, INT_MAX, do_execute },
1951     { "trigger", 2, INT_MAX, do_trigger },
1952     { "idl", 1, INT_MAX, do_idl },
1953     { "help", 0, INT_MAX, do_help },
1954     { NULL, 0, 0, NULL },
1955 };