ofproto: Avoid passing indeterminate value to rule_insert().
[openvswitch] / ovsdb / ovsdb-client.c
index 6e0068125fcb03070b3f6862d1074d54690ed8ca..df24e371c56deec5bacc85177ad9ff4261a10f9f 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2009 Nicira Networks.
+ * Copyright (c) 2009, 2010 Nicira Networks.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,6 +16,7 @@
 
 #include <config.h>
 
+#include <assert.h>
 #include <errno.h>
 #include <getopt.h>
 #include <limits.h>
 #include "command-line.h"
 #include "column.h"
 #include "compiler.h"
+#include "daemon.h"
 #include "dynamic-string.h"
 #include "json.h"
 #include "jsonrpc.h"
 #include "ovsdb.h"
 #include "ovsdb-error.h"
 #include "stream.h"
+#include "stream-ssl.h"
 #include "table.h"
 #include "timeval.h"
 #include "util.h"
@@ -53,6 +56,9 @@ static int output_width;
 /* --no-headings: Whether table output should include headings. */
 static int output_headings = true;
 
+/* --pretty: Flags to pass to json_to_string(). */
+static int json_flags = JSSF_SORT;
+
 static const struct command all_commands[];
 
 static void usage(void) NO_RETURN;
@@ -61,6 +67,7 @@ static void parse_options(int argc, char *argv[]);
 int
 main(int argc, char *argv[])
 {
+    proctitle_init(argc, argv);
     set_program_name(argv[0]);
     time_init();
     vlog_init();
@@ -73,13 +80,22 @@ main(int argc, char *argv[])
 static void
 parse_options(int argc, char *argv[])
 {
+    enum {
+        OPT_BOOTSTRAP_CA_CERT = UCHAR_MAX + 1
+    };
     static struct option long_options[] = {
         {"wide", no_argument, &output_width, INT_MAX},
         {"format", required_argument, 0, 'f'},
            {"no-headings", no_argument, &output_headings, 0},
+        {"pretty", no_argument, &json_flags, JSSF_PRETTY | JSSF_SORT},
         {"verbose", optional_argument, 0, 'v'},
         {"help", no_argument, 0, 'h'},
         {"version", no_argument, 0, 'V'},
+        DAEMON_LONG_OPTIONS,
+#ifdef HAVE_OPENSSL
+        {"bootstrap-ca-cert", required_argument, 0, OPT_BOOTSTRAP_CA_CERT},
+        STREAM_SSL_LONG_OPTIONS
+#endif
         {0, 0, 0, 0},
     };
     char *short_options = long_options_to_short_options(long_options);
@@ -121,6 +137,16 @@ parse_options(int argc, char *argv[])
             vlog_set_verbosity(optarg);
             break;
 
+        DAEMON_OPTION_HANDLERS
+
+#ifdef HAVE_OPENSSL
+        STREAM_SSL_OPTION_HANDLERS
+
+        case OPT_BOOTSTRAP_CA_CERT:
+            stream_ssl_set_ca_cert_file(optarg, true);
+            break;
+#endif
+
         case '?':
             exit(EXIT_FAILURE);
 
@@ -141,25 +167,29 @@ usage(void)
     printf("%s: Open vSwitch database JSON-RPC client\n"
            "usage: %s [OPTIONS] COMMAND [ARG...]\n"
            "\nValid commands are:\n"
-           "\n  get-schema SERVER\n"
-           "    retrieve schema from SERVER\n"
-           "\n  list-tables SERVER\n"
-           "    list SERVER's tables\n"
-           "\n  list-columns SERVER [TABLE]\n"
-           "    list columns in TABLE (or all tables) on SERVER\n"
+           "\n  list-dbs SERVER\n"
+           "    list databases available on SERVER\n"
+           "\n  get-schema SERVER DATABASE\n"
+           "    retrieve schema for DATABASE from SERVER\n"
+           "\n  list-tables SERVER DATABSE\n"
+           "    list tables for DATABSAE on SERVER\n"
+           "\n  list-columns SERVER DATABASE [TABLE]\n"
+           "    list columns in TABLE (or all tables) in DATABASE on SERVER\n"
            "\n  transact SERVER TRANSACTION\n"
            "    run TRANSACTION (a JSON array of operations) on SERVER\n"
            "    and print the results as JSON on stdout\n"
-           "\n  monitor SERVER TABLE [COLUMN,...] [SELECT,...]\n"
-           "    monitor contents of (COLUMNs in) TABLE on SERVER\n"
+           "\n  monitor SERVER DATABASE TABLE [COLUMN,...] [SELECT,...]\n"
+           "    monitor contents of (COLUMNs in) TABLE in DATABASE on SERVER\n"
            "    Valid SELECTs are: initial, insert, delete, modify\n",
            program_name, program_name);
-    stream_usage("SERVER", true, true);
+    stream_usage("SERVER", true, true, true);
     printf("\nOutput formatting options:\n"
            "  -f, --format=FORMAT         set output formatting to FORMAT\n"
            "                              (\"table\", \"html\", or \"csv\"\n"
            "  --wide                      don't limit TTY lines to 79 bytes\n"
-           "  --no-headings               omit table heading row\n");
+           "  --no-headings               omit table heading row\n"
+           "  --pretty                    pretty-print JSON in output");
+    daemon_usage();
     vlog_usage();
     printf("\nOther options:\n"
            "  -h, --help                  display this help message\n"
@@ -209,7 +239,7 @@ open_jsonrpc(const char *server)
 static void
 print_json(struct json *json)
 {
-    char *string = json_to_string(json, JSSF_SORT);
+    char *string = json_to_string(json, json_flags);
     fputs(string, stdout);
     free(string);
 }
@@ -230,13 +260,16 @@ check_ovsdb_error(struct ovsdb_error *error)
 }
 
 static struct ovsdb_schema *
-fetch_schema_from_rpc(struct jsonrpc *rpc)
+fetch_schema_from_rpc(struct jsonrpc *rpc, const char *database)
 {
     struct jsonrpc_msg *request, *reply;
     struct ovsdb_schema *schema;
     int error;
 
-    request = jsonrpc_create_request("get_schema", json_array_create_empty());
+    request = jsonrpc_create_request("get_schema",
+                                     json_array_create_1(
+                                         json_string_create(database)),
+                                     NULL);
     error = jsonrpc_transact_block(rpc, request, &reply);
     if (error) {
         ovs_fatal(error, "transaction failed");
@@ -248,13 +281,13 @@ fetch_schema_from_rpc(struct jsonrpc *rpc)
 }
 
 static struct ovsdb_schema *
-fetch_schema(const char *server)
+fetch_schema(const char *server, const char *database)
 {
     struct ovsdb_schema *schema;
     struct jsonrpc *rpc;
 
     rpc = open_jsonrpc(server);
-    schema = fetch_schema_from_rpc(rpc);
+    schema = fetch_schema_from_rpc(rpc, database);
     jsonrpc_close(rpc);
 
     return schema;
@@ -542,10 +575,41 @@ table_print(const struct table *table)
     }
 }
 \f
+static void
+do_list_dbs(int argc UNUSED, char *argv[])
+{
+    struct jsonrpc_msg *request, *reply;
+    struct jsonrpc *rpc;
+    int error;
+    size_t i;
+
+    rpc = open_jsonrpc(argv[1]);
+    request = jsonrpc_create_request("list_dbs", json_array_create_empty(),
+                                     NULL);
+    error = jsonrpc_transact_block(rpc, request, &reply);
+    if (error) {
+        ovs_fatal(error, "transaction failed");
+    }
+
+    if (reply->result->type != JSON_ARRAY) {
+        ovs_fatal(0, "list_dbs response is not array");
+    }
+
+    for (i = 0; i < reply->result->u.array.n; i++) {
+        const struct json *name = reply->result->u.array.elems[i];
+
+        if (name->type != JSON_STRING) {
+            ovs_fatal(0, "list_dbs response %zu is not string", i);
+        }
+        puts(name->u.string);
+    }
+    jsonrpc_msg_destroy(reply);
+}
+
 static void
 do_get_schema(int argc UNUSED, char *argv[])
 {
-    struct ovsdb_schema *schema = fetch_schema(argv[1]);
+    struct ovsdb_schema *schema = fetch_schema(argv[1], argv[2]);
     print_and_free_json(ovsdb_schema_to_json(schema));
     ovsdb_schema_destroy(schema);
 }
@@ -557,7 +621,7 @@ do_list_tables(int argc UNUSED, char *argv[])
     struct shash_node *node;
     struct table t;
 
-    schema = fetch_schema(argv[1]);
+    schema = fetch_schema(argv[1], argv[2]);
     table_init(&t);
     table_add_column(&t, "Table");
     table_add_column(&t, "Comment");
@@ -577,12 +641,12 @@ do_list_tables(int argc UNUSED, char *argv[])
 static void
 do_list_columns(int argc UNUSED, char *argv[])
 {
-    const char *table_name = argv[2];
+    const char *table_name = argv[3];
     struct ovsdb_schema *schema;
     struct shash_node *table_node;
     struct table t;
 
-    schema = fetch_schema(argv[1]);
+    schema = fetch_schema(argv[1], argv[2]);
     table_init(&t);
     if (!table_name) {
         table_add_column(&t, "Table");
@@ -597,7 +661,7 @@ do_list_columns(int argc UNUSED, char *argv[])
             struct shash_node *column_node;
 
             SHASH_FOR_EACH (column_node, &ts->columns) {
-                struct ovsdb_column *column = column_node->data;
+                const struct ovsdb_column *column = column_node->data;
                 struct json *type = ovsdb_type_to_json(&column->type);
 
                 table_add_row(&t);
@@ -629,14 +693,14 @@ do_transact(int argc UNUSED, char *argv[])
     transaction = parse_json(argv[2]);
 
     rpc = open_jsonrpc(argv[1]);
-    request = jsonrpc_create_request("transact", transaction);
+    request = jsonrpc_create_request("transact", transaction, NULL);
     error = jsonrpc_transact_block(rpc, request, &reply);
     if (error) {
         ovs_fatal(error, "transaction failed");
     }
     if (reply->error) {
         ovs_fatal(error, "transaction returned error: %s",
-                  json_to_string(reply->error, JSSF_SORT));
+                  json_to_string(reply->error, json_flags));
     }
     print_json(reply->result);
     putchar('\n');
@@ -730,6 +794,9 @@ monitor_print(struct json *table_updates,
 static void
 do_monitor(int argc, char *argv[])
 {
+    const char *server = argv[1];
+    const char *database = argv[2];
+    const char *table_name = argv[3];
     struct ovsdb_column_set columns = OVSDB_COLUMN_SET_INITIALIZER;
     struct ovsdb_table_schema *table;
     struct ovsdb_schema *schema;
@@ -738,25 +805,27 @@ do_monitor(int argc, char *argv[])
     struct json *select, *monitor, *monitor_request, *monitor_requests,
         *request_id;
 
-    rpc = open_jsonrpc(argv[1]);
+    rpc = open_jsonrpc(server);
 
-    schema = fetch_schema_from_rpc(rpc);
-    table = shash_find_data(&schema->tables, argv[2]);
+    schema = fetch_schema_from_rpc(rpc, database);
+    table = shash_find_data(&schema->tables, table_name);
     if (!table) {
-        ovs_fatal(0, "%s: no table named \"%s\"", argv[1], argv[2]);
+        ovs_fatal(0, "%s: %s does not have a table named \"%s\"",
+                  server, database, table_name);
     }
 
-    if (argc >= 4 && *argv[3] != '\0') {
+    if (argc >= 5 && *argv[4] != '\0') {
         char *save_ptr = NULL;
         char *token;
 
-        for (token = strtok_r(argv[3], ",", &save_ptr); token != NULL;
+        for (token = strtok_r(argv[4], ",", &save_ptr); token != NULL;
              token = strtok_r(NULL, ",", &save_ptr)) {
             const struct ovsdb_column *column;
             column = ovsdb_table_schema_get_column(table, token);
             if (!column) {
-                ovs_fatal(0, "%s: table \"%s\" does not have a "
-                          "column named \"%s\"", argv[1], argv[2], token);
+                ovs_fatal(0, "%s: table \"%s\" in %s does not have a "
+                          "column named \"%s\"",
+                          server, table_name, database, token);
             }
             ovsdb_column_set_add(&columns, column);
         }
@@ -771,12 +840,12 @@ do_monitor(int argc, char *argv[])
         }
     }
 
-    if (argc >= 5 && *argv[4] != '\0') {
+    if (argc >= 6 && *argv[5] != '\0') {
         char *save_ptr = NULL;
         char *token;
 
         select = json_object_create();
-        for (token = strtok_r(argv[4], ",", &save_ptr); token != NULL;
+        for (token = strtok_r(argv[5], ",", &save_ptr); token != NULL;
              token = strtok_r(NULL, ",", &save_ptr)) {
             json_object_put(select, token, json_boolean_create(true));
         }
@@ -792,10 +861,11 @@ do_monitor(int argc, char *argv[])
     }
 
     monitor_requests = json_object_create();
-    json_object_put(monitor_requests, argv[2], monitor_request);
+    json_object_put(monitor_requests, table_name, monitor_request);
 
-    monitor = json_array_create_2(json_null_create(), monitor_requests);
-    request = jsonrpc_create_request("monitor", monitor);
+    monitor = json_array_create_3(json_string_create(database),
+                                  json_null_create(), monitor_requests);
+    request = jsonrpc_create_request("monitor", monitor, NULL);
     request_id = json_clone(request->id);
     jsonrpc_send(rpc, request);
     for (;;) {
@@ -804,7 +874,8 @@ do_monitor(int argc, char *argv[])
 
         error = jsonrpc_recv_block(rpc, &msg);
         if (error) {
-            ovs_fatal(error, "%s: receive failed", argv[1]);
+            ovsdb_schema_destroy(schema);
+            ovs_fatal(error, "%s: receive failed", server);
         }
 
         if (msg->type == JSONRPC_REQUEST && !strcmp(msg->method, "echo")) {
@@ -813,6 +884,15 @@ do_monitor(int argc, char *argv[])
         } else if (msg->type == JSONRPC_REPLY
                    && json_equal(msg->id, request_id)) {
             monitor_print(msg->result, table, &columns, true);
+            fflush(stdout);
+            if (get_detach()) {
+                /* daemonize() closes the standard file descriptors.  We output
+                 * to stdout, so we need to save and restore STDOUT_FILENO. */
+                int fd = dup(STDOUT_FILENO);
+                daemonize();
+                dup2(fd, STDOUT_FILENO);
+                close(fd);
+            }
         } else if (msg->type == JSONRPC_NOTIFY
                    && !strcmp(msg->method, "update")) {
             struct json *params = msg->params;
@@ -821,8 +901,10 @@ do_monitor(int argc, char *argv[])
                 && params->u.array.elems[0]->type == JSON_NULL) {
                 monitor_print(params->u.array.elems[1],
                               table, &columns, false);
+                fflush(stdout);
             }
         }
+        jsonrpc_msg_destroy(msg);
     }
 }
 
@@ -833,11 +915,12 @@ do_help(int argc UNUSED, char *argv[] UNUSED)
 }
 
 static const struct command all_commands[] = {
-    { "get-schema", 1, 1, do_get_schema },
-    { "list-tables", 1, 1, do_list_tables },
-    { "list-columns", 1, 2, do_list_columns },
+    { "list-dbs", 1, 1, do_list_dbs },
+    { "get-schema", 2, 2, do_get_schema },
+    { "list-tables", 2, 2, do_list_tables },
+    { "list-columns", 2, 3, do_list_columns },
     { "transact", 2, 2, do_transact },
-    { "monitor", 2, 4, do_monitor },
+    { "monitor", 3, 5, do_monitor },
     { "help", 0, INT_MAX, do_help },
     { NULL, 0, 0, NULL },
 };