pintos-os.org Git - openvswitch/blob - ovsdb/ovsdb-client.c

   1 /*
   2  * Copyright (c) 2009, 2010 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 <errno.h>
  21 #include <getopt.h>
  22 #include <limits.h>
  23 #include <signal.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <unistd.h>
  27
  28 #include "command-line.h"
  29 #include "column.h"
  30 #include "compiler.h"
  31 #include "daemon.h"
  32 #include "dynamic-string.h"
  33 #include "json.h"
  34 #include "jsonrpc.h"
  35 #include "ovsdb.h"
  36 #include "ovsdb-data.h"
  37 #include "ovsdb-error.h"
  38 #include "sort.h"
  39 #include "stream.h"
  40 #include "stream-ssl.h"
  41 #include "table.h"
  42 #include "timeval.h"
  43 #include "util.h"
  44 #include "vlog.h"
  45
  46 VLOG_DEFINE_THIS_MODULE(ovsdb_client);
  47
  48 /* --format: Output formatting. */
  49 static enum {
  50     FMT_TABLE,                  /* Textual table. */
  51     FMT_HTML,                   /* HTML table. */
  52     FMT_CSV,                    /* Comma-separated lines. */
  53     FMT_JSON                    /* JSON. */
  54 } output_format;
  55
  56 /* --no-headings: Whether table output should include headings. */
  57 static int output_headings = true;
  58
  59 /* --pretty: Flags to pass to json_to_string(). */
  60 static int json_flags = JSSF_SORT;
  61
  62 /* --data: Format of data in output tables. */
  63 static enum {
  64     DF_STRING,                  /* String format. */
  65     DF_JSON,                    /* JSON. */
  66 } data_format;
  67
  68 static const struct command all_commands[];
  69
  70 static void usage(void) NO_RETURN;
  71 static void parse_options(int argc, char *argv[]);
  72
  73 int
  74 main(int argc, char *argv[])
  75 {
  76     proctitle_init(argc, argv);
  77     set_program_name(argv[0]);
  78     parse_options(argc, argv);
  79     signal(SIGPIPE, SIG_IGN);
  80     run_command(argc - optind, argv + optind, all_commands);
  81     return 0;
  82 }
  83
  84 static void
  85 parse_options(int argc, char *argv[])
  86 {
  87     enum {
  88         OPT_BOOTSTRAP_CA_CERT = UCHAR_MAX + 1,
  89         DAEMON_OPTION_ENUMS,
  90     };
  91     static struct option long_options[] = {
  92         {"format", required_argument, 0, 'f'},
  93         {"data", required_argument, 0, 'd'},
  94         {"no-headings", no_argument, &output_headings, 0},
  95         {"pretty", no_argument, &json_flags, JSSF_PRETTY | JSSF_SORT},
  96         {"verbose", optional_argument, 0, 'v'},
  97         {"help", no_argument, 0, 'h'},
  98         {"version", no_argument, 0, 'V'},
  99         DAEMON_LONG_OPTIONS,
 100 #ifdef HAVE_OPENSSL
 101         {"bootstrap-ca-cert", required_argument, 0, OPT_BOOTSTRAP_CA_CERT},
 102         STREAM_SSL_LONG_OPTIONS
 103 #endif
 104         {0, 0, 0, 0},
 105     };
 106     char *short_options = long_options_to_short_options(long_options);
 107
 108     for (;;) {
 109         int c;
 110
 111         c = getopt_long(argc, argv, short_options, long_options, NULL);
 112         if (c == -1) {
 113             break;
 114         }
 115
 116         switch (c) {
 117         case 'f':
 118             if (!strcmp(optarg, "table")) {
 119                 output_format = FMT_TABLE;
 120             } else if (!strcmp(optarg, "html")) {
 121                 output_format = FMT_HTML;
 122             } else if (!strcmp(optarg, "csv")) {
 123                 output_format = FMT_CSV;
 124             } else if (!strcmp(optarg, "json")) {
 125                 output_format = FMT_JSON;
 126             } else {
 127                 ovs_fatal(0, "unknown output format \"%s\"", optarg);
 128             }
 129             break;
 130
 131         case 'd':
 132             if (!strcmp(optarg, "string")) {
 133                 data_format = DF_STRING;
 134             } else if (!strcmp(optarg, "json")) {
 135                 data_format = DF_JSON;
 136             } else {
 137                 ovs_fatal(0, "unknown data format \"%s\"", optarg);
 138             }
 139             break;
 140
 141         case 'h':
 142             usage();
 143
 144         case 'V':
 145             OVS_PRINT_VERSION(0, 0);
 146             exit(EXIT_SUCCESS);
 147
 148         case 'v':
 149             vlog_set_verbosity(optarg);
 150             break;
 151
 152         DAEMON_OPTION_HANDLERS
 153
 154 #ifdef HAVE_OPENSSL
 155         STREAM_SSL_OPTION_HANDLERS
 156
 157         case OPT_BOOTSTRAP_CA_CERT:
 158             stream_ssl_set_ca_cert_file(optarg, true);
 159             break;
 160 #endif
 161
 162         case '?':
 163             exit(EXIT_FAILURE);
 164
 165         case 0:
 166             /* getopt_long() already set the value for us. */
 167             break;
 168
 169         default:
 170             abort();
 171         }
 172     }
 173     free(short_options);
 174 }
 175
 176 static void
 177 usage(void)
 178 {
 179     printf("%s: Open vSwitch database JSON-RPC client\n"
 180            "usage: %s [OPTIONS] COMMAND [ARG...]\n"
 181            "\nValid commands are:\n"
 182            "\n  list-dbs SERVER\n"
 183            "    list databases available on SERVER\n"
 184            "\n  get-schema SERVER DATABASE\n"
 185            "    retrieve schema for DATABASE from SERVER\n"
 186            "\n  get-schema-version SERVER DATABASE\n"
 187            "    retrieve schema for DATABASE from SERVER and report only its\n"
 188            "    version number on stdout\n"
 189            "\n  list-tables SERVER DATABASE\n"
 190            "    list tables for DATABASE on SERVER\n"
 191            "\n  list-columns SERVER DATABASE [TABLE]\n"
 192            "    list columns in TABLE (or all tables) in DATABASE on SERVER\n"
 193            "\n  transact SERVER TRANSACTION\n"
 194            "    run TRANSACTION (a JSON array of operations) on SERVER\n"
 195            "    and print the results as JSON on stdout\n"
 196            "\n  monitor SERVER DATABASE TABLE [COLUMN,...]...\n"
 197            "    monitor contents of COLUMNs in TABLE in DATABASE on SERVER.\n"
 198            "    COLUMNs may include !initial, !insert, !delete, !modify\n"
 199            "    to avoid seeing the specified kinds of changes.\n"
 200            "\n  dump SERVER DATABASE\n"
 201            "    dump contents of DATABASE on SERVER to stdout\n",
 202            program_name, program_name);
 203     stream_usage("SERVER", true, true, true);
 204     printf("\nOutput formatting options:\n"
 205            "  -f, --format=FORMAT         set output formatting to FORMAT\n"
 206            "                              (\"table\", \"html\", \"csv\", "
 207            "or \"json\")\n"
 208            "  --no-headings               omit table heading row\n"
 209            "  --pretty                    pretty-print JSON in output");
 210     daemon_usage();
 211     vlog_usage();
 212     printf("\nOther options:\n"
 213            "  -h, --help                  display this help message\n"
 214            "  -V, --version               display version information\n");
 215     exit(EXIT_SUCCESS);
 216 }
 217 \f
 218 static struct json *
 219 parse_json(const char *s)
 220 {
 221     struct json *json = json_from_string(s);
 222     if (json->type == JSON_STRING) {
 223         ovs_fatal(0, "\"%s\": %s", s, json->u.string);
 224     }
 225     return json;
 226 }
 227
 228 static struct jsonrpc *
 229 open_jsonrpc(const char *server)
 230 {
 231     struct stream *stream;
 232     int error;
 233
 234     error = stream_open_block(jsonrpc_stream_open(server, &stream), &stream);
 235     if (error == EAFNOSUPPORT) {
 236         struct pstream *pstream;
 237
 238         error = jsonrpc_pstream_open(server, &pstream);
 239         if (error) {
 240             ovs_fatal(error, "failed to connect or listen to \"%s\"", server);
 241         }
 242
 243         VLOG_INFO("%s: waiting for connection...", server);
 244         error = pstream_accept_block(pstream, &stream);
 245         if (error) {
 246             ovs_fatal(error, "failed to accept connection on \"%s\"", server);
 247         }
 248
 249         pstream_close(pstream);
 250     } else if (error) {
 251         ovs_fatal(error, "failed to connect to \"%s\"", server);
 252     }
 253
 254     return jsonrpc_open(stream);
 255 }
 256
 257 static void
 258 print_json(struct json *json)
 259 {
 260     char *string = json_to_string(json, json_flags);
 261     fputs(string, stdout);
 262     free(string);
 263 }
 264
 265 static void
 266 print_and_free_json(struct json *json)
 267 {
 268     print_json(json);
 269     json_destroy(json);
 270 }
 271
 272 static void
 273 check_ovsdb_error(struct ovsdb_error *error)
 274 {
 275     if (error) {
 276         ovs_fatal(0, "%s", ovsdb_error_to_string(error));
 277     }
 278 }
 279
 280 static struct ovsdb_schema *
 281 fetch_schema_from_rpc(struct jsonrpc *rpc, const char *database)
 282 {
 283     struct jsonrpc_msg *request, *reply;
 284     struct ovsdb_schema *schema;
 285     int error;
 286
 287     request = jsonrpc_create_request("get_schema",
 288                                      json_array_create_1(
 289                                          json_string_create(database)),
 290                                      NULL);
 291     error = jsonrpc_transact_block(rpc, request, &reply);
 292     if (error) {
 293         ovs_fatal(error, "transaction failed");
 294     }
 295     check_ovsdb_error(ovsdb_schema_from_json(reply->result, &schema));
 296     jsonrpc_msg_destroy(reply);
 297
 298     return schema;
 299 }
 300
 301 static struct ovsdb_schema *
 302 fetch_schema(const char *server, const char *database)
 303 {
 304     struct ovsdb_schema *schema;
 305     struct jsonrpc *rpc;
 306
 307     rpc = open_jsonrpc(server);
 308     schema = fetch_schema_from_rpc(rpc, database);
 309     jsonrpc_close(rpc);
 310
 311     return schema;
 312 }
 313 \f
 314 struct column {
 315     char *heading;
 316 };
 317
 318 struct cell {
 319     /* Literal text. */
 320     char *text;
 321
 322     /* JSON. */
 323     struct json *json;
 324     const struct ovsdb_type *type;
 325 };
 326
 327 static const char *
 328 cell_to_text(const struct cell *cell_)
 329 {
 330     struct cell *cell = (struct cell *) cell_;
 331     if (!cell->text) {
 332         if (cell->json) {
 333             if (data_format == DF_JSON || !cell->type) {
 334                 cell->text = json_to_string(cell->json, JSSF_SORT);
 335             } else if (data_format == DF_STRING) {
 336                 struct ovsdb_datum datum;
 337                 struct ovsdb_error *error;
 338                 struct ds s;
 339
 340                 error = ovsdb_datum_from_json(&datum, cell->type, cell->json,
 341                                               NULL);
 342                 if (!error) {
 343                     ds_init(&s);
 344                     ovsdb_datum_to_string(&datum, cell->type, &s);
 345                     ovsdb_datum_destroy(&datum, cell->type);
 346                     cell->text = ds_steal_cstr(&s);
 347                 } else {
 348                     cell->text = json_to_string(cell->json, JSSF_SORT);
 349                 }
 350             } else {
 351                 NOT_REACHED();
 352             }
 353         } else {
 354             cell->text = xstrdup("");
 355         }
 356     }
 357
 358     return cell->text;
 359 }
 360
 361 static void
 362 cell_destroy(struct cell *cell)
 363 {
 364     free(cell->text);
 365     json_destroy(cell->json);
 366 }
 367
 368 struct table {
 369     struct cell *cells;
 370     struct column *columns;
 371     size_t n_columns, allocated_columns;
 372     size_t n_rows, allocated_rows;
 373     size_t current_column;
 374     char *caption;
 375 };
 376
 377 static void
 378 table_init(struct table *table)
 379 {
 380     memset(table, 0, sizeof *table);
 381 }
 382
 383 static void
 384 table_destroy(struct table *table)
 385 {
 386     size_t i;
 387
 388     for (i = 0; i < table->n_columns; i++) {
 389         free(table->columns[i].heading);
 390     }
 391     free(table->columns);
 392
 393     for (i = 0; i < table->n_columns * table->n_rows; i++) {
 394         cell_destroy(&table->cells[i]);
 395     }
 396     free(table->cells);
 397
 398     free(table->caption);
 399 }
 400
 401 static void
 402 table_set_caption(struct table *table, char *caption)
 403 {
 404     free(table->caption);
 405     table->caption = caption;
 406 }
 407
 408 static void
 409 table_add_column(struct table *table, const char *heading, ...)
 410     PRINTF_FORMAT(2, 3);
 411
 412 static void
 413 table_add_column(struct table *table, const char *heading, ...)
 414 {
 415     struct column *column;
 416     va_list args;
 417
 418     assert(!table->n_rows);
 419     if (table->n_columns >= table->allocated_columns) {
 420         table->columns = x2nrealloc(table->columns, &table->allocated_columns,
 421                                     sizeof *table->columns);
 422     }
 423     column = &table->columns[table->n_columns++];
 424
 425     va_start(args, heading);
 426     column->heading = xvasprintf(heading, args);
 427     va_end(args);
 428 }
 429
 430 static struct cell *
 431 table_cell__(const struct table *table, size_t row, size_t column)
 432 {
 433     return &table->cells[column + row * table->n_columns];
 434 }
 435
 436 static void
 437 table_add_row(struct table *table)
 438 {
 439     size_t x, y;
 440
 441     if (table->n_rows >= table->allocated_rows) {
 442         table->cells = x2nrealloc(table->cells, &table->allocated_rows,
 443                                   table->n_columns * sizeof *table->cells);
 444     }
 445
 446     y = table->n_rows++;
 447     table->current_column = 0;
 448     for (x = 0; x < table->n_columns; x++) {
 449         struct cell *cell = table_cell__(table, y, x);
 450         memset(cell, 0, sizeof *cell);
 451     }
 452 }
 453
 454 static struct cell *
 455 table_add_cell(struct table *table)
 456 {
 457     size_t x, y;
 458
 459     assert(table->n_rows > 0);
 460     assert(table->current_column < table->n_columns);
 461
 462     x = table->current_column++;
 463     y = table->n_rows - 1;
 464
 465     return table_cell__(table, y, x);
 466 }
 467
 468 static void
 469 table_print_table_line__(struct ds *line)
 470 {
 471     puts(ds_cstr(line));
 472     ds_clear(line);
 473 }
 474
 475 static void
 476 table_print_table__(const struct table *table)
 477 {
 478     static int n = 0;
 479     struct ds line = DS_EMPTY_INITIALIZER;
 480     int *widths;
 481     size_t x, y;
 482
 483     if (n++ > 0) {
 484         putchar('\n');
 485     }
 486
 487     if (table->caption) {
 488         puts(table->caption);
 489     }
 490
 491     widths = xmalloc(table->n_columns * sizeof *widths);
 492     for (x = 0; x < table->n_columns; x++) {
 493         const struct column *column = &table->columns[x];
 494
 495         widths[x] = strlen(column->heading);
 496         for (y = 0; y < table->n_rows; y++) {
 497             const char *text = cell_to_text(table_cell__(table, y, x));
 498             size_t length = strlen(text);
 499
 500             if (length > widths[x])
 501                 widths[x] = length;
 502         }
 503     }
 504
 505     if (output_headings) {
 506         for (x = 0; x < table->n_columns; x++) {
 507             const struct column *column = &table->columns[x];
 508             if (x) {
 509                 ds_put_char(&line, ' ');
 510             }
 511             ds_put_format(&line, "%-*s", widths[x], column->heading);
 512         }
 513         table_print_table_line__(&line);
 514
 515         for (x = 0; x < table->n_columns; x++) {
 516             if (x) {
 517                 ds_put_char(&line, ' ');
 518             }
 519             ds_put_char_multiple(&line, '-', widths[x]);
 520         }
 521         table_print_table_line__(&line);
 522     }
 523
 524     for (y = 0; y < table->n_rows; y++) {
 525         for (x = 0; x < table->n_columns; x++) {
 526             const char *text = cell_to_text(table_cell__(table, y, x));
 527             if (x) {
 528                 ds_put_char(&line, ' ');
 529             }
 530             ds_put_format(&line, "%-*s", widths[x], text);
 531         }
 532         table_print_table_line__(&line);
 533     }
 534
 535     ds_destroy(&line);
 536     free(widths);
 537 }
 538
 539 static void
 540 table_escape_html_text__(const char *s, size_t n)
 541 {
 542     size_t i;
 543
 544     for (i = 0; i < n; i++) {
 545         char c = s[i];
 546
 547         switch (c) {
 548         case '&':
 549             fputs("&amp;", stdout);
 550             break;
 551         case '<':
 552             fputs("&lt;", stdout);
 553             break;
 554         case '>':
 555             fputs("&gt;", stdout);
 556             break;
 557         case '"':
 558             fputs("&quot;", stdout);
 559             break;
 560         default:
 561             putchar(c);
 562             break;
 563         }
 564     }
 565 }
 566
 567 static void
 568 table_print_html_cell__(const char *element, const char *content)
 569 {
 570     const char *p;
 571
 572     printf("    <%s>", element);
 573     for (p = content; *p; ) {
 574         struct uuid uuid;
 575
 576         if (uuid_from_string_prefix(&uuid, p)) {
 577             printf("<a href=\"#%.*s\">%.*s</a>", UUID_LEN, p, 8, p);
 578             p += UUID_LEN;
 579         } else {
 580             table_escape_html_text__(p, 1);
 581             p++;
 582         }
 583     }
 584     printf("</%s>\n", element);
 585 }
 586
 587 static void
 588 table_print_html__(const struct table *table)
 589 {
 590     size_t x, y;
 591
 592     fputs("<table border=1>\n", stdout);
 593
 594     if (table->caption) {
 595         table_print_html_cell__("caption", table->caption);
 596     }
 597
 598     if (output_headings) {
 599         fputs("  <tr>\n", stdout);
 600         for (x = 0; x < table->n_columns; x++) {
 601             const struct column *column = &table->columns[x];
 602             table_print_html_cell__("th", column->heading);
 603         }
 604         fputs("  </tr>\n", stdout);
 605     }
 606
 607     for (y = 0; y < table->n_rows; y++) {
 608         fputs("  <tr>\n", stdout);
 609         for (x = 0; x < table->n_columns; x++) {
 610             const char *content = cell_to_text(table_cell__(table, y, x));
 611
 612             if (!strcmp(table->columns[x].heading, "_uuid")) {
 613                 fputs("    <td><a name=\"", stdout);
 614                 table_escape_html_text__(content, strlen(content));
 615                 fputs("\">", stdout);
 616                 table_escape_html_text__(content, 8);
 617                 fputs("</a></td>\n", stdout);
 618             } else {
 619                 table_print_html_cell__("td", content);
 620             }
 621         }
 622         fputs("  </tr>\n", stdout);
 623     }
 624
 625     fputs("</table>\n", stdout);
 626 }
 627
 628 static void
 629 table_print_csv_cell__(const char *content)
 630 {
 631     const char *p;
 632
 633     if (!strpbrk(content, "\n\",")) {
 634         fputs(content, stdout);
 635     } else {
 636         putchar('"');
 637         for (p = content; *p != '\0'; p++) {
 638             switch (*p) {
 639             case '"':
 640                 fputs("\"\"", stdout);
 641                 break;
 642             default:
 643                 putchar(*p);
 644                 break;
 645             }
 646         }
 647         putchar('"');
 648     }
 649 }
 650
 651 static void
 652 table_print_csv__(const struct table *table)
 653 {
 654     static int n = 0;
 655     size_t x, y;
 656
 657     if (n++ > 0) {
 658         putchar('\n');
 659     }
 660
 661     if (table->caption) {
 662         puts(table->caption);
 663     }
 664
 665     if (output_headings) {
 666         for (x = 0; x < table->n_columns; x++) {
 667             const struct column *column = &table->columns[x];
 668             if (x) {
 669                 putchar(',');
 670             }
 671             table_print_csv_cell__(column->heading);
 672         }
 673         putchar('\n');
 674     }
 675
 676     for (y = 0; y < table->n_rows; y++) {
 677         for (x = 0; x < table->n_columns; x++) {
 678             if (x) {
 679                 putchar(',');
 680             }
 681             table_print_csv_cell__(cell_to_text(table_cell__(table, y, x)));
 682         }
 683         putchar('\n');
 684     }
 685 }
 686
 687 static void
 688 table_print_json__(const struct table *table)
 689 {
 690     struct json *json, *headings, *data;
 691     size_t x, y;
 692     char *s;
 693
 694     json = json_object_create();
 695     if (table->caption) {
 696         json_object_put_string(json, "caption", table->caption);
 697     }
 698
 699     headings = json_array_create_empty();
 700     for (x = 0; x < table->n_columns; x++) {
 701         const struct column *column = &table->columns[x];
 702         json_array_add(headings, json_string_create(column->heading));
 703     }
 704     json_object_put(json, "headings", headings);
 705
 706     data = json_array_create_empty();
 707     for (y = 0; y < table->n_rows; y++) {
 708         struct json *row = json_array_create_empty();
 709         for (x = 0; x < table->n_columns; x++) {
 710             const struct cell *cell = table_cell__(table, y, x);
 711             if (cell->text) {
 712                 json_array_add(row, json_string_create(cell->text));
 713             } else {
 714                 json_array_add(row, json_clone(cell->json));
 715             }
 716         }
 717         json_array_add(data, row);
 718     }
 719     json_object_put(json, "data", data);
 720
 721     s = json_to_string(json, json_flags);
 722     json_destroy(json);
 723     puts(s);
 724     free(s);
 725 }
 726
 727 static void
 728 table_print(const struct table *table)
 729 {
 730     switch (output_format) {
 731     case FMT_TABLE:
 732         table_print_table__(table);
 733         break;
 734
 735     case FMT_HTML:
 736         table_print_html__(table);
 737         break;
 738
 739     case FMT_CSV:
 740         table_print_csv__(table);
 741         break;
 742
 743     case FMT_JSON:
 744         table_print_json__(table);
 745         break;
 746     }
 747 }
 748 \f
 749 static void
 750 do_list_dbs(int argc OVS_UNUSED, char *argv[])
 751 {
 752     struct jsonrpc_msg *request, *reply;
 753     struct jsonrpc *rpc;
 754     int error;
 755     size_t i;
 756
 757     rpc = open_jsonrpc(argv[1]);
 758     request = jsonrpc_create_request("list_dbs", json_array_create_empty(),
 759                                      NULL);
 760     error = jsonrpc_transact_block(rpc, request, &reply);
 761     if (error) {
 762         ovs_fatal(error, "transaction failed");
 763     }
 764
 765     if (reply->result->type != JSON_ARRAY) {
 766         ovs_fatal(0, "list_dbs response is not array");
 767     }
 768
 769     for (i = 0; i < reply->result->u.array.n; i++) {
 770         const struct json *name = reply->result->u.array.elems[i];
 771
 772         if (name->type != JSON_STRING) {
 773             ovs_fatal(0, "list_dbs response %zu is not string", i);
 774         }
 775         puts(name->u.string);
 776     }
 777     jsonrpc_msg_destroy(reply);
 778 }
 779
 780 static void
 781 do_get_schema(int argc OVS_UNUSED, char *argv[])
 782 {
 783     struct ovsdb_schema *schema = fetch_schema(argv[1], argv[2]);
 784     print_and_free_json(ovsdb_schema_to_json(schema));
 785     ovsdb_schema_destroy(schema);
 786 }
 787
 788 static void
 789 do_get_schema_version(int argc OVS_UNUSED, char *argv[])
 790 {
 791     struct ovsdb_schema *schema = fetch_schema(argv[1], argv[2]);
 792     puts(schema->version);
 793     ovsdb_schema_destroy(schema);
 794 }
 795
 796 static void
 797 do_list_tables(int argc OVS_UNUSED, char *argv[])
 798 {
 799     struct ovsdb_schema *schema;
 800     struct shash_node *node;
 801     struct table t;
 802
 803     schema = fetch_schema(argv[1], argv[2]);
 804     table_init(&t);
 805     table_add_column(&t, "Table");
 806     SHASH_FOR_EACH (node, &schema->tables) {
 807         struct ovsdb_table_schema *ts = node->data;
 808
 809         table_add_row(&t);
 810         table_add_cell(&t)->text = xstrdup(ts->name);
 811     }
 812     ovsdb_schema_destroy(schema);
 813     table_print(&t);
 814 }
 815
 816 static void
 817 do_list_columns(int argc OVS_UNUSED, char *argv[])
 818 {
 819     const char *table_name = argv[3];
 820     struct ovsdb_schema *schema;
 821     struct shash_node *table_node;
 822     struct table t;
 823
 824     schema = fetch_schema(argv[1], argv[2]);
 825     table_init(&t);
 826     if (!table_name) {
 827         table_add_column(&t, "Table");
 828     }
 829     table_add_column(&t, "Column");
 830     table_add_column(&t, "Type");
 831     SHASH_FOR_EACH (table_node, &schema->tables) {
 832         struct ovsdb_table_schema *ts = table_node->data;
 833
 834         if (!table_name || !strcmp(table_name, ts->name)) {
 835             struct shash_node *column_node;
 836
 837             SHASH_FOR_EACH (column_node, &ts->columns) {
 838                 const struct ovsdb_column *column = column_node->data;
 839
 840                 table_add_row(&t);
 841                 if (!table_name) {
 842                     table_add_cell(&t)->text = xstrdup(ts->name);
 843                 }
 844                 table_add_cell(&t)->text = xstrdup(column->name);
 845                 table_add_cell(&t)->json = ovsdb_type_to_json(&column->type);
 846             }
 847         }
 848     }
 849     ovsdb_schema_destroy(schema);
 850     table_print(&t);
 851 }
 852
 853 static void
 854 do_transact(int argc OVS_UNUSED, char *argv[])
 855 {
 856     struct jsonrpc_msg *request, *reply;
 857     struct json *transaction;
 858     struct jsonrpc *rpc;
 859     int error;
 860
 861     transaction = parse_json(argv[2]);
 862
 863     rpc = open_jsonrpc(argv[1]);
 864     request = jsonrpc_create_request("transact", transaction, NULL);
 865     error = jsonrpc_transact_block(rpc, request, &reply);
 866     if (error) {
 867         ovs_fatal(error, "transaction failed");
 868     }
 869     if (reply->error) {
 870         ovs_fatal(error, "transaction returned error: %s",
 871                   json_to_string(reply->error, json_flags));
 872     }
 873     print_json(reply->result);
 874     putchar('\n');
 875     jsonrpc_msg_destroy(reply);
 876     jsonrpc_close(rpc);
 877 }
 878
 879 static void
 880 monitor_print_row(struct json *row, const char *type, const char *uuid,
 881                   const struct ovsdb_column_set *columns, struct table *t)
 882 {
 883     size_t i;
 884
 885     if (!row) {
 886         ovs_error(0, "missing %s row", type);
 887         return;
 888     } else if (row->type != JSON_OBJECT) {
 889         ovs_error(0, "<row> is not object");
 890         return;
 891     }
 892
 893     table_add_row(t);
 894     table_add_cell(t)->text = xstrdup(uuid);
 895     table_add_cell(t)->text = xstrdup(type);
 896     for (i = 0; i < columns->n_columns; i++) {
 897         const struct ovsdb_column *column = columns->columns[i];
 898         struct json *value = shash_find_data(json_object(row), column->name);
 899         struct cell *cell = table_add_cell(t);
 900         if (value) {
 901             cell->json = json_clone(value);
 902             cell->type = &column->type;
 903         }
 904     }
 905 }
 906
 907 static void
 908 monitor_print(struct json *table_updates,
 909               const struct ovsdb_table_schema *table,
 910               const struct ovsdb_column_set *columns, bool initial)
 911 {
 912     struct json *table_update;
 913     struct shash_node *node;
 914     struct table t;
 915     size_t i;
 916
 917     table_init(&t);
 918
 919     if (table_updates->type != JSON_OBJECT) {
 920         ovs_error(0, "<table-updates> is not object");
 921         return;
 922     }
 923     table_update = shash_find_data(json_object(table_updates), table->name);
 924     if (!table_update) {
 925         return;
 926     }
 927     if (table_update->type != JSON_OBJECT) {
 928         ovs_error(0, "<table-update> is not object");
 929         return;
 930     }
 931
 932     table_add_column(&t, "row");
 933     table_add_column(&t, "action");
 934     for (i = 0; i < columns->n_columns; i++) {
 935         table_add_column(&t, "%s", columns->columns[i]->name);
 936     }
 937     SHASH_FOR_EACH (node, json_object(table_update)) {
 938         struct json *row_update = node->data;
 939         struct json *old, *new;
 940
 941         if (row_update->type != JSON_OBJECT) {
 942             ovs_error(0, "<row-update> is not object");
 943             continue;
 944         }
 945         old = shash_find_data(json_object(row_update), "old");
 946         new = shash_find_data(json_object(row_update), "new");
 947         if (initial) {
 948             monitor_print_row(new, "initial", node->name, columns, &t);
 949         } else if (!old) {
 950             monitor_print_row(new, "insert", node->name, columns, &t);
 951         } else if (!new) {
 952             monitor_print_row(old, "delete", node->name, columns, &t);
 953         } else {
 954             monitor_print_row(old, "old", node->name, columns, &t);
 955             monitor_print_row(new, "new", "", columns, &t);
 956         }
 957     }
 958     table_print(&t);
 959     table_destroy(&t);
 960 }
 961
 962 static void
 963 add_column(const char *server, const struct ovsdb_column *column,
 964            struct ovsdb_column_set *columns, struct json *columns_json)
 965 {
 966     if (ovsdb_column_set_contains(columns, column->index)) {
 967         ovs_fatal(0, "%s: column \"%s\" mentioned multiple times",
 968                   server, column->name);
 969     }
 970     ovsdb_column_set_add(columns, column);
 971     json_array_add(columns_json, json_string_create(column->name));
 972 }
 973
 974 static struct json *
 975 parse_monitor_columns(char *arg, const char *server, const char *database,
 976                       const struct ovsdb_table_schema *table,
 977                       struct ovsdb_column_set *columns)
 978 {
 979     bool initial, insert, delete, modify;
 980     struct json *mr, *columns_json;
 981     char *save_ptr = NULL;
 982     char *token;
 983
 984     mr = json_object_create();
 985     columns_json = json_array_create_empty();
 986     json_object_put(mr, "columns", columns_json);
 987
 988     initial = insert = delete = modify = true;
 989     for (token = strtok_r(arg, ",", &save_ptr); token != NULL;
 990          token = strtok_r(NULL, ",", &save_ptr)) {
 991         if (!strcmp(token, "!initial")) {
 992             initial = false;
 993         } else if (!strcmp(token, "!insert")) {
 994             insert = false;
 995         } else if (!strcmp(token, "!delete")) {
 996             delete = false;
 997         } else if (!strcmp(token, "!modify")) {
 998             modify = false;
 999         } else {
1000             const struct ovsdb_column *column;
1001
1002             column = ovsdb_table_schema_get_column(table, token);
1003             if (!column) {
1004                 ovs_fatal(0, "%s: table \"%s\" in %s does not have a "
1005                           "column named \"%s\"",
1006                           server, table->name, database, token);
1007             }
1008             add_column(server, column, columns, columns_json);
1009         }
1010     }
1011
1012     if (columns_json->u.array.n == 0) {
1013         const struct shash_node **nodes;
1014         size_t i, n;
1015
1016         n = shash_count(&table->columns);
1017         nodes = shash_sort(&table->columns);
1018         for (i = 0; i < n; i++) {
1019             const struct ovsdb_column *column = nodes[i]->data;
1020             if (column->index != OVSDB_COL_UUID
1021                 && column->index != OVSDB_COL_VERSION) {
1022                 add_column(server, column, columns, columns_json);
1023             }
1024         }
1025         free(nodes);
1026
1027         add_column(server, ovsdb_table_schema_get_column(table,"_version"),
1028                    columns, columns_json);
1029     }
1030
1031     if (!initial || !insert || !delete || !modify) {
1032         struct json *select = json_object_create();
1033         json_object_put(select, "initial", json_boolean_create(initial));
1034         json_object_put(select, "insert", json_boolean_create(insert));
1035         json_object_put(select, "delete", json_boolean_create(delete));
1036         json_object_put(select, "modify", json_boolean_create(modify));
1037         json_object_put(mr, "select", select);
1038     }
1039
1040     return mr;
1041 }
1042
1043 static void
1044 do_monitor(int argc, char *argv[])
1045 {
1046     const char *server = argv[1];
1047     const char *database = argv[2];
1048     const char *table_name = argv[3];
1049     struct ovsdb_column_set columns = OVSDB_COLUMN_SET_INITIALIZER;
1050     struct ovsdb_table_schema *table;
1051     struct ovsdb_schema *schema;
1052     struct jsonrpc_msg *request;
1053     struct jsonrpc *rpc;
1054     struct json *monitor, *monitor_request_array,
1055         *monitor_requests, *request_id;
1056
1057     rpc = open_jsonrpc(server);
1058
1059     schema = fetch_schema_from_rpc(rpc, database);
1060     table = shash_find_data(&schema->tables, table_name);
1061     if (!table) {
1062         ovs_fatal(0, "%s: %s does not have a table named \"%s\"",
1063                   server, database, table_name);
1064     }
1065
1066     monitor_request_array = json_array_create_empty();
1067     if (argc > 4) {
1068         int i;
1069
1070         for (i = 4; i < argc; i++) {
1071             json_array_add(
1072                 monitor_request_array,
1073                 parse_monitor_columns(argv[i], server, database, table,
1074                                       &columns));
1075         }
1076     } else {
1077         /* Allocate a writable empty string since parse_monitor_columns() is
1078          * going to strtok() it and that's risky with literal "". */
1079         char empty[] = "";
1080         json_array_add(
1081             monitor_request_array,
1082             parse_monitor_columns(empty, server, database, table, &columns));
1083     }
1084
1085     monitor_requests = json_object_create();
1086     json_object_put(monitor_requests, table_name, monitor_request_array);
1087
1088     monitor = json_array_create_3(json_string_create(database),
1089                                   json_null_create(), monitor_requests);
1090     request = jsonrpc_create_request("monitor", monitor, NULL);
1091     request_id = json_clone(request->id);
1092     jsonrpc_send(rpc, request);
1093     for (;;) {
1094         struct jsonrpc_msg *msg;
1095         int error;
1096
1097         error = jsonrpc_recv_block(rpc, &msg);
1098         if (error) {
1099             ovsdb_schema_destroy(schema);
1100             ovs_fatal(error, "%s: receive failed", server);
1101         }
1102
1103         if (msg->type == JSONRPC_REQUEST && !strcmp(msg->method, "echo")) {
1104             jsonrpc_send(rpc, jsonrpc_create_reply(json_clone(msg->params),
1105                                                    msg->id));
1106         } else if (msg->type == JSONRPC_REPLY
1107                    && json_equal(msg->id, request_id)) {
1108             monitor_print(msg->result, table, &columns, true);
1109             fflush(stdout);
1110             if (get_detach()) {
1111                 /* daemonize() closes the standard file descriptors.  We output
1112                  * to stdout, so we need to save and restore STDOUT_FILENO. */
1113                 int fd = dup(STDOUT_FILENO);
1114                 daemonize();
1115                 dup2(fd, STDOUT_FILENO);
1116                 close(fd);
1117             }
1118         } else if (msg->type == JSONRPC_NOTIFY
1119                    && !strcmp(msg->method, "update")) {
1120             struct json *params = msg->params;
1121             if (params->type == JSON_ARRAY
1122                 && params->u.array.n == 2
1123                 && params->u.array.elems[0]->type == JSON_NULL) {
1124                 monitor_print(params->u.array.elems[1],
1125                               table, &columns, false);
1126                 fflush(stdout);
1127             }
1128         }
1129         jsonrpc_msg_destroy(msg);
1130     }
1131 }
1132
1133 struct dump_table_aux {
1134     struct ovsdb_datum **data;
1135     const struct ovsdb_column **columns;
1136     size_t n_columns;
1137 };
1138
1139 static int
1140 compare_data(size_t a_y, size_t b_y, size_t x,
1141              const struct dump_table_aux *aux)
1142 {
1143     return ovsdb_datum_compare_3way(&aux->data[a_y][x],
1144                                     &aux->data[b_y][x],
1145                                     &aux->columns[x]->type);
1146 }
1147
1148 static int
1149 compare_rows(size_t a_y, size_t b_y, void *aux_)
1150 {
1151     struct dump_table_aux *aux = aux_;
1152     size_t x;
1153
1154     /* Skip UUID columns on the first pass, since their values tend to be
1155      * random and make our results less reproducible. */
1156     for (x = 0; x < aux->n_columns; x++) {
1157         if (aux->columns[x]->type.key.type != OVSDB_TYPE_UUID) {
1158             int cmp = compare_data(a_y, b_y, x, aux);
1159             if (cmp) {
1160                 return cmp;
1161             }
1162         }
1163     }
1164
1165     /* Use UUID columns as tie-breakers. */
1166     for (x = 0; x < aux->n_columns; x++) {
1167         if (aux->columns[x]->type.key.type == OVSDB_TYPE_UUID) {
1168             int cmp = compare_data(a_y, b_y, x, aux);
1169             if (cmp) {
1170                 return cmp;
1171             }
1172         }
1173     }
1174
1175     return 0;
1176 }
1177
1178 static void
1179 swap_rows(size_t a_y, size_t b_y, void *aux_)
1180 {
1181     struct dump_table_aux *aux = aux_;
1182     struct ovsdb_datum *tmp = aux->data[a_y];
1183     aux->data[a_y] = aux->data[b_y];
1184     aux->data[b_y] = tmp;
1185 }
1186
1187 static int
1188 compare_columns(const void *a_, const void *b_)
1189 {
1190     const struct ovsdb_column *const *ap = a_;
1191     const struct ovsdb_column *const *bp = b_;
1192     const struct ovsdb_column *a = *ap;
1193     const struct ovsdb_column *b = *bp;
1194
1195     return strcmp(a->name, b->name);
1196 }
1197
1198 static void
1199 dump_table(const struct ovsdb_table_schema *ts, struct json_array *rows)
1200 {
1201     const struct ovsdb_column **columns;
1202     size_t n_columns;
1203
1204     struct ovsdb_datum **data;
1205
1206     struct dump_table_aux aux;
1207     struct shash_node *node;
1208     struct table t;
1209     size_t x, y;
1210
1211     /* Sort columns by name, for reproducibility. */
1212     columns = xmalloc(shash_count(&ts->columns) * sizeof *columns);
1213     n_columns = 0;
1214     SHASH_FOR_EACH (node, &ts->columns) {
1215         struct ovsdb_column *column = node->data;
1216         if (strcmp(column->name, "_version")) {
1217             columns[n_columns++] = column;
1218         }
1219     }
1220     qsort(columns, n_columns, sizeof *columns, compare_columns);
1221
1222     /* Extract data from table. */
1223     data = xmalloc(rows->n * sizeof *data);
1224     for (y = 0; y < rows->n; y++) {
1225         struct shash *row;
1226
1227         if (rows->elems[y]->type != JSON_OBJECT) {
1228             ovs_fatal(0,  "row %zu in table %s response is not a JSON object: "
1229                       "%s", y, ts->name, json_to_string(rows->elems[y], 0));
1230         }
1231         row = json_object(rows->elems[y]);
1232
1233         data[y] = xmalloc(n_columns * sizeof **data);
1234         for (x = 0; x < n_columns; x++) {
1235             const struct json *json = shash_find_data(row, columns[x]->name);
1236             if (!json) {
1237                 ovs_fatal(0, "row %zu in table %s response lacks %s column",
1238                           y, ts->name, columns[x]->name);
1239             }
1240
1241             check_ovsdb_error(ovsdb_datum_from_json(&data[y][x],
1242                                                     &columns[x]->type,
1243                                                     json, NULL));
1244         }
1245     }
1246
1247     /* Sort rows by column values, for reproducibility. */
1248     aux.data = data;
1249     aux.columns = columns;
1250     aux.n_columns = n_columns;
1251     sort(rows->n, compare_rows, swap_rows, &aux);
1252
1253     /* Add column headings. */
1254     table_init(&t);
1255     table_set_caption(&t, xasprintf("%s table", ts->name));
1256     for (x = 0; x < n_columns; x++) {
1257         table_add_column(&t, "%s", columns[x]->name);
1258     }
1259
1260     /* Print rows. */
1261     for (y = 0; y < rows->n; y++) {
1262         table_add_row(&t);
1263         for (x = 0; x < n_columns; x++) {
1264             struct cell *cell = table_add_cell(&t);
1265             cell->json = ovsdb_datum_to_json(&data[y][x], &columns[x]->type);
1266             cell->type = &columns[x]->type;
1267         }
1268     }
1269     table_print(&t);
1270     table_destroy(&t);
1271 }
1272
1273 static void
1274 do_dump(int argc OVS_UNUSED, char *argv[])
1275 {
1276     const char *server = argv[1];
1277     const char *database = argv[2];
1278
1279     struct jsonrpc_msg *request, *reply;
1280     struct ovsdb_schema *schema;
1281     struct json *transaction;
1282     struct jsonrpc *rpc;
1283     int error;
1284
1285     const struct shash_node **tables;
1286     size_t n_tables;
1287
1288     size_t i;
1289
1290     rpc = open_jsonrpc(server);
1291
1292     schema = fetch_schema_from_rpc(rpc, database);
1293     tables = shash_sort(&schema->tables);
1294     n_tables = shash_count(&schema->tables);
1295
1296     /* Construct transaction to retrieve entire database. */
1297     transaction = json_array_create_1(json_string_create(database));
1298     for (i = 0; i < n_tables; i++) {
1299         const struct ovsdb_table_schema *ts = tables[i]->data;
1300         struct json *op, *columns;
1301         struct shash_node *node;
1302
1303         columns = json_array_create_empty();
1304         SHASH_FOR_EACH (node, &ts->columns) {
1305             const struct ovsdb_column *column = node->data;
1306
1307             if (strcmp(column->name, "_version")) {
1308                 json_array_add(columns, json_string_create(column->name));
1309             }
1310         }
1311
1312         op = json_object_create();
1313         json_object_put_string(op, "op", "select");
1314         json_object_put_string(op, "table", tables[i]->name);
1315         json_object_put(op, "where", json_array_create_empty());
1316         json_object_put(op, "columns", columns);
1317         json_array_add(transaction, op);
1318     }
1319
1320     /* Send request, get reply. */
1321     request = jsonrpc_create_request("transact", transaction, NULL);
1322     error = jsonrpc_transact_block(rpc, request, &reply);
1323     if (error) {
1324         ovs_fatal(error, "transaction failed");
1325     }
1326
1327     /* Print database contents. */
1328     if (reply->result->type != JSON_ARRAY
1329         || reply->result->u.array.n != n_tables) {
1330         ovs_fatal(0, "reply is not array of %zu elements: %s",
1331                   n_tables, json_to_string(reply->result, 0));
1332     }
1333     for (i = 0; i < n_tables; i++) {
1334         const struct ovsdb_table_schema *ts = tables[i]->data;
1335         const struct json *op_result = reply->result->u.array.elems[i];
1336         struct json *rows;
1337
1338         if (op_result->type != JSON_OBJECT
1339             || !(rows = shash_find_data(json_object(op_result), "rows"))
1340             || rows->type != JSON_ARRAY) {
1341             ovs_fatal(0, "%s table reply is not an object with a \"rows\" "
1342                       "member array: %s",
1343                       ts->name, json_to_string(op_result, 0));
1344         }
1345
1346         dump_table(ts, &rows->u.array);
1347     }
1348 }
1349
1350 static void
1351 do_help(int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
1352 {
1353     usage();
1354 }
1355
1356 static const struct command all_commands[] = {
1357     { "list-dbs", 1, 1, do_list_dbs },
1358     { "get-schema", 2, 2, do_get_schema },
1359     { "get-schema-version", 2, 2, do_get_schema_version },
1360     { "list-tables", 2, 2, do_list_tables },
1361     { "list-columns", 2, 3, do_list_columns },
1362     { "transact", 2, 2, do_transact },
1363     { "monitor", 3, INT_MAX, do_monitor },
1364     { "dump", 2, 2, do_dump },
1365     { "help", 0, INT_MAX, do_help },
1366     { NULL, 0, 0, NULL },
1367 };