mirror of
https://github.com/openvswitch/ovs
synced 2025-10-17 14:28:02 +00:00
987 lines
30 KiB
C
987 lines
30 KiB
C
/*
|
||
* Copyright (c) 2009, 2010, 2011 Nicira Networks.
|
||
*
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at:
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
#include <config.h>
|
||
|
||
#include <assert.h>
|
||
#include <ctype.h>
|
||
#include <errno.h>
|
||
#include <getopt.h>
|
||
#include <limits.h>
|
||
#include <signal.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <unistd.h>
|
||
|
||
#include "command-line.h"
|
||
#include "column.h"
|
||
#include "compiler.h"
|
||
#include "daemon.h"
|
||
#include "dirs.h"
|
||
#include "dynamic-string.h"
|
||
#include "json.h"
|
||
#include "jsonrpc.h"
|
||
#include "lib/table.h"
|
||
#include "ovsdb.h"
|
||
#include "ovsdb-data.h"
|
||
#include "ovsdb-error.h"
|
||
#include "sort.h"
|
||
#include "sset.h"
|
||
#include "stream.h"
|
||
#include "stream-ssl.h"
|
||
#include "table.h"
|
||
#include "timeval.h"
|
||
#include "util.h"
|
||
#include "vlog.h"
|
||
|
||
VLOG_DEFINE_THIS_MODULE(ovsdb_client);
|
||
|
||
enum args_needed {
|
||
NEED_NONE, /* No JSON-RPC connection or database name needed. */
|
||
NEED_RPC, /* JSON-RPC connection needed. */
|
||
NEED_DATABASE /* JSON-RPC connection and database name needed. */
|
||
};
|
||
|
||
struct ovsdb_client_command {
|
||
const char *name;
|
||
enum args_needed need;
|
||
int min_args;
|
||
int max_args;
|
||
void (*handler)(struct jsonrpc *rpc, const char *database,
|
||
int argc, char *argv[]);
|
||
};
|
||
|
||
/* Format for table output. */
|
||
static struct table_style table_style = TABLE_STYLE_DEFAULT;
|
||
|
||
static const struct ovsdb_client_command all_commands[];
|
||
|
||
static void usage(void) NO_RETURN;
|
||
static void parse_options(int argc, char *argv[]);
|
||
static struct jsonrpc *open_jsonrpc(const char *server);
|
||
static void fetch_dbs(struct jsonrpc *, struct sset *dbs);
|
||
|
||
int
|
||
main(int argc, char *argv[])
|
||
{
|
||
const struct ovsdb_client_command *command;
|
||
const char *database;
|
||
struct jsonrpc *rpc;
|
||
|
||
proctitle_init(argc, argv);
|
||
set_program_name(argv[0]);
|
||
parse_options(argc, argv);
|
||
signal(SIGPIPE, SIG_IGN);
|
||
|
||
if (optind >= argc) {
|
||
ovs_fatal(0, "missing command name; use --help for help");
|
||
}
|
||
|
||
for (command = all_commands; ; command++) {
|
||
if (!command->name) {
|
||
VLOG_FATAL("unknown command '%s'; use --help for help",
|
||
argv[optind]);
|
||
} else if (!strcmp(command->name, argv[optind])) {
|
||
break;
|
||
}
|
||
}
|
||
optind++;
|
||
|
||
if (command->need != NEED_NONE) {
|
||
if (argc - optind > command->min_args
|
||
&& (isalpha((unsigned char) argv[optind][0])
|
||
&& strchr(argv[optind], ':'))) {
|
||
rpc = open_jsonrpc(argv[optind++]);
|
||
} else {
|
||
char *sock = xasprintf("unix:%s/db.sock", ovs_rundir());
|
||
rpc = open_jsonrpc(sock);
|
||
free(sock);
|
||
}
|
||
} else {
|
||
rpc = NULL;
|
||
}
|
||
|
||
if (command->need == NEED_DATABASE) {
|
||
struct sset dbs;
|
||
|
||
sset_init(&dbs);
|
||
fetch_dbs(rpc, &dbs);
|
||
if (argc - optind > command->min_args
|
||
&& sset_contains(&dbs, argv[optind])) {
|
||
database = argv[optind++];
|
||
} else if (sset_count(&dbs) == 1) {
|
||
database = xstrdup(SSET_FIRST(&dbs));
|
||
} else if (sset_contains(&dbs, "Open_vSwitch")) {
|
||
database = "Open_vSwitch";
|
||
} else {
|
||
ovs_fatal(0, "no default database for `%s' command, please "
|
||
"specify a database name", command->name);
|
||
}
|
||
sset_destroy(&dbs);
|
||
} else {
|
||
database = NULL;
|
||
}
|
||
|
||
if (argc - optind < command->min_args ||
|
||
argc - optind > command->max_args) {
|
||
VLOG_FATAL("invalid syntax for '%s' (use --help for help)",
|
||
command->name);
|
||
}
|
||
|
||
command->handler(rpc, database, argc - optind, argv + optind);
|
||
|
||
jsonrpc_close(rpc);
|
||
|
||
if (ferror(stdout)) {
|
||
VLOG_FATAL("write to stdout failed");
|
||
}
|
||
if (ferror(stderr)) {
|
||
VLOG_FATAL("write to stderr failed");
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static void
|
||
parse_options(int argc, char *argv[])
|
||
{
|
||
enum {
|
||
OPT_BOOTSTRAP_CA_CERT = UCHAR_MAX + 1,
|
||
DAEMON_OPTION_ENUMS,
|
||
TABLE_OPTION_ENUMS
|
||
};
|
||
static struct option long_options[] = {
|
||
{"verbose", optional_argument, NULL, 'v'},
|
||
{"help", no_argument, NULL, 'h'},
|
||
{"version", no_argument, NULL, 'V'},
|
||
DAEMON_LONG_OPTIONS,
|
||
#ifdef HAVE_OPENSSL
|
||
{"bootstrap-ca-cert", required_argument, NULL, OPT_BOOTSTRAP_CA_CERT},
|
||
STREAM_SSL_LONG_OPTIONS,
|
||
#endif
|
||
TABLE_LONG_OPTIONS,
|
||
{NULL, 0, NULL, 0},
|
||
};
|
||
char *short_options = long_options_to_short_options(long_options);
|
||
|
||
for (;;) {
|
||
int c;
|
||
|
||
c = getopt_long(argc, argv, short_options, long_options, NULL);
|
||
if (c == -1) {
|
||
break;
|
||
}
|
||
|
||
switch (c) {
|
||
case 'h':
|
||
usage();
|
||
|
||
case 'V':
|
||
ovs_print_version(0, 0);
|
||
exit(EXIT_SUCCESS);
|
||
|
||
case 'v':
|
||
vlog_set_verbosity(optarg);
|
||
break;
|
||
|
||
DAEMON_OPTION_HANDLERS
|
||
|
||
TABLE_OPTION_HANDLERS(&table_style)
|
||
|
||
STREAM_SSL_OPTION_HANDLERS
|
||
|
||
case OPT_BOOTSTRAP_CA_CERT:
|
||
stream_ssl_set_ca_cert_file(optarg, true);
|
||
break;
|
||
|
||
case '?':
|
||
exit(EXIT_FAILURE);
|
||
|
||
case 0:
|
||
/* getopt_long() already set the value for us. */
|
||
break;
|
||
|
||
default:
|
||
abort();
|
||
}
|
||
}
|
||
free(short_options);
|
||
}
|
||
|
||
static void
|
||
usage(void)
|
||
{
|
||
printf("%s: Open vSwitch database JSON-RPC client\n"
|
||
"usage: %s [OPTIONS] COMMAND [ARG...]\n"
|
||
"\nValid commands are:\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 get-schema-version [SERVER] [DATABASE]\n"
|
||
" retrieve schema for DATABASE from SERVER and report only its\n"
|
||
" version number on stdout\n"
|
||
"\n list-tables [SERVER] [DATABASE]\n"
|
||
" list tables for DATABASE 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] [DATABASE] TABLE [COLUMN,...]...\n"
|
||
" monitor contents of COLUMNs in TABLE in DATABASE on SERVER.\n"
|
||
" COLUMNs may include !initial, !insert, !delete, !modify\n"
|
||
" to avoid seeing the specified kinds of changes.\n"
|
||
"\n dump [SERVER] [DATABASE]\n"
|
||
" dump contents of DATABASE on SERVER to stdout\n"
|
||
"\nThe default SERVER is unix:%s/db.sock.\n"
|
||
"The default DATABASE is Open_vSwitch.\n",
|
||
program_name, program_name, ovs_rundir());
|
||
stream_usage("SERVER", true, true, true);
|
||
printf("\nOutput formatting options:\n"
|
||
" -f, --format=FORMAT set output formatting to FORMAT\n"
|
||
" (\"table\", \"html\", \"csv\", "
|
||
"or \"json\")\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"
|
||
" -V, --version display version information\n");
|
||
exit(EXIT_SUCCESS);
|
||
}
|
||
|
||
static struct json *
|
||
parse_json(const char *s)
|
||
{
|
||
struct json *json = json_from_string(s);
|
||
if (json->type == JSON_STRING) {
|
||
ovs_fatal(0, "\"%s\": %s", s, json->u.string);
|
||
}
|
||
return json;
|
||
}
|
||
|
||
static struct jsonrpc *
|
||
open_jsonrpc(const char *server)
|
||
{
|
||
struct stream *stream;
|
||
int error;
|
||
|
||
error = stream_open_block(jsonrpc_stream_open(server, &stream), &stream);
|
||
if (error == EAFNOSUPPORT) {
|
||
struct pstream *pstream;
|
||
|
||
error = jsonrpc_pstream_open(server, &pstream);
|
||
if (error) {
|
||
ovs_fatal(error, "failed to connect or listen to \"%s\"", server);
|
||
}
|
||
|
||
VLOG_INFO("%s: waiting for connection...", server);
|
||
error = pstream_accept_block(pstream, &stream);
|
||
if (error) {
|
||
ovs_fatal(error, "failed to accept connection on \"%s\"", server);
|
||
}
|
||
|
||
pstream_close(pstream);
|
||
} else if (error) {
|
||
ovs_fatal(error, "failed to connect to \"%s\"", server);
|
||
}
|
||
|
||
return jsonrpc_open(stream);
|
||
}
|
||
|
||
static void
|
||
print_json(struct json *json)
|
||
{
|
||
char *string = json_to_string(json, table_style.json_flags);
|
||
fputs(string, stdout);
|
||
free(string);
|
||
}
|
||
|
||
static void
|
||
print_and_free_json(struct json *json)
|
||
{
|
||
print_json(json);
|
||
json_destroy(json);
|
||
}
|
||
|
||
static void
|
||
check_ovsdb_error(struct ovsdb_error *error)
|
||
{
|
||
if (error) {
|
||
ovs_fatal(0, "%s", ovsdb_error_to_string(error));
|
||
}
|
||
}
|
||
|
||
static struct ovsdb_schema *
|
||
fetch_schema(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_1(
|
||
json_string_create(database)),
|
||
NULL);
|
||
error = jsonrpc_transact_block(rpc, request, &reply);
|
||
if (error) {
|
||
ovs_fatal(error, "transaction failed");
|
||
}
|
||
check_ovsdb_error(ovsdb_schema_from_json(reply->result, &schema));
|
||
jsonrpc_msg_destroy(reply);
|
||
|
||
return schema;
|
||
}
|
||
|
||
static void
|
||
fetch_dbs(struct jsonrpc *rpc, struct sset *dbs)
|
||
{
|
||
struct jsonrpc_msg *request, *reply;
|
||
int error;
|
||
size_t i;
|
||
|
||
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);
|
||
}
|
||
sset_add(dbs, name->u.string);
|
||
}
|
||
jsonrpc_msg_destroy(reply);
|
||
}
|
||
|
||
static void
|
||
do_list_dbs(struct jsonrpc *rpc, const char *database OVS_UNUSED,
|
||
int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
|
||
{
|
||
const char *db_name;
|
||
struct sset dbs;
|
||
|
||
sset_init(&dbs);
|
||
fetch_dbs(rpc, &dbs);
|
||
SSET_FOR_EACH (db_name, &dbs) {
|
||
puts(db_name);
|
||
}
|
||
sset_destroy(&dbs);
|
||
}
|
||
|
||
static void
|
||
do_get_schema(struct jsonrpc *rpc, const char *database,
|
||
int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
|
||
{
|
||
struct ovsdb_schema *schema = fetch_schema(rpc, database);
|
||
print_and_free_json(ovsdb_schema_to_json(schema));
|
||
ovsdb_schema_destroy(schema);
|
||
}
|
||
|
||
static void
|
||
do_get_schema_version(struct jsonrpc *rpc, const char *database,
|
||
int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
|
||
{
|
||
struct ovsdb_schema *schema = fetch_schema(rpc, database);
|
||
puts(schema->version);
|
||
ovsdb_schema_destroy(schema);
|
||
}
|
||
|
||
static void
|
||
do_list_tables(struct jsonrpc *rpc, const char *database,
|
||
int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
|
||
{
|
||
struct ovsdb_schema *schema;
|
||
struct shash_node *node;
|
||
struct table t;
|
||
|
||
schema = fetch_schema(rpc, database);
|
||
table_init(&t);
|
||
table_add_column(&t, "Table");
|
||
SHASH_FOR_EACH (node, &schema->tables) {
|
||
struct ovsdb_table_schema *ts = node->data;
|
||
|
||
table_add_row(&t);
|
||
table_add_cell(&t)->text = xstrdup(ts->name);
|
||
}
|
||
ovsdb_schema_destroy(schema);
|
||
table_print(&t, &table_style);
|
||
}
|
||
|
||
static void
|
||
do_list_columns(struct jsonrpc *rpc, const char *database,
|
||
int argc OVS_UNUSED, char *argv[])
|
||
{
|
||
const char *table_name = argv[0];
|
||
struct ovsdb_schema *schema;
|
||
struct shash_node *table_node;
|
||
struct table t;
|
||
|
||
schema = fetch_schema(rpc, database);
|
||
table_init(&t);
|
||
if (!table_name) {
|
||
table_add_column(&t, "Table");
|
||
}
|
||
table_add_column(&t, "Column");
|
||
table_add_column(&t, "Type");
|
||
SHASH_FOR_EACH (table_node, &schema->tables) {
|
||
struct ovsdb_table_schema *ts = table_node->data;
|
||
|
||
if (!table_name || !strcmp(table_name, ts->name)) {
|
||
struct shash_node *column_node;
|
||
|
||
SHASH_FOR_EACH (column_node, &ts->columns) {
|
||
const struct ovsdb_column *column = column_node->data;
|
||
|
||
table_add_row(&t);
|
||
if (!table_name) {
|
||
table_add_cell(&t)->text = xstrdup(ts->name);
|
||
}
|
||
table_add_cell(&t)->text = xstrdup(column->name);
|
||
table_add_cell(&t)->json = ovsdb_type_to_json(&column->type);
|
||
}
|
||
}
|
||
}
|
||
ovsdb_schema_destroy(schema);
|
||
table_print(&t, &table_style);
|
||
}
|
||
|
||
static void
|
||
do_transact(struct jsonrpc *rpc, const char *database OVS_UNUSED,
|
||
int argc OVS_UNUSED, char *argv[])
|
||
{
|
||
struct jsonrpc_msg *request, *reply;
|
||
struct json *transaction;
|
||
int error;
|
||
|
||
transaction = parse_json(argv[0]);
|
||
|
||
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, table_style.json_flags));
|
||
}
|
||
print_json(reply->result);
|
||
putchar('\n');
|
||
jsonrpc_msg_destroy(reply);
|
||
}
|
||
|
||
static void
|
||
monitor_print_row(struct json *row, const char *type, const char *uuid,
|
||
const struct ovsdb_column_set *columns, struct table *t)
|
||
{
|
||
size_t i;
|
||
|
||
if (!row) {
|
||
ovs_error(0, "missing %s row", type);
|
||
return;
|
||
} else if (row->type != JSON_OBJECT) {
|
||
ovs_error(0, "<row> is not object");
|
||
return;
|
||
}
|
||
|
||
table_add_row(t);
|
||
table_add_cell(t)->text = xstrdup(uuid);
|
||
table_add_cell(t)->text = xstrdup(type);
|
||
for (i = 0; i < columns->n_columns; i++) {
|
||
const struct ovsdb_column *column = columns->columns[i];
|
||
struct json *value = shash_find_data(json_object(row), column->name);
|
||
struct cell *cell = table_add_cell(t);
|
||
if (value) {
|
||
cell->json = json_clone(value);
|
||
cell->type = &column->type;
|
||
}
|
||
}
|
||
}
|
||
|
||
static void
|
||
monitor_print(struct json *table_updates,
|
||
const struct ovsdb_table_schema *table,
|
||
const struct ovsdb_column_set *columns, bool initial)
|
||
{
|
||
struct json *table_update;
|
||
struct shash_node *node;
|
||
struct table t;
|
||
size_t i;
|
||
|
||
table_init(&t);
|
||
|
||
if (table_updates->type != JSON_OBJECT) {
|
||
ovs_error(0, "<table-updates> is not object");
|
||
return;
|
||
}
|
||
table_update = shash_find_data(json_object(table_updates), table->name);
|
||
if (!table_update) {
|
||
return;
|
||
}
|
||
if (table_update->type != JSON_OBJECT) {
|
||
ovs_error(0, "<table-update> is not object");
|
||
return;
|
||
}
|
||
|
||
table_add_column(&t, "row");
|
||
table_add_column(&t, "action");
|
||
for (i = 0; i < columns->n_columns; i++) {
|
||
table_add_column(&t, "%s", columns->columns[i]->name);
|
||
}
|
||
SHASH_FOR_EACH (node, json_object(table_update)) {
|
||
struct json *row_update = node->data;
|
||
struct json *old, *new;
|
||
|
||
if (row_update->type != JSON_OBJECT) {
|
||
ovs_error(0, "<row-update> is not object");
|
||
continue;
|
||
}
|
||
old = shash_find_data(json_object(row_update), "old");
|
||
new = shash_find_data(json_object(row_update), "new");
|
||
if (initial) {
|
||
monitor_print_row(new, "initial", node->name, columns, &t);
|
||
} else if (!old) {
|
||
monitor_print_row(new, "insert", node->name, columns, &t);
|
||
} else if (!new) {
|
||
monitor_print_row(old, "delete", node->name, columns, &t);
|
||
} else {
|
||
monitor_print_row(old, "old", node->name, columns, &t);
|
||
monitor_print_row(new, "new", "", columns, &t);
|
||
}
|
||
}
|
||
table_print(&t, &table_style);
|
||
table_destroy(&t);
|
||
}
|
||
|
||
static void
|
||
add_column(const char *server, const struct ovsdb_column *column,
|
||
struct ovsdb_column_set *columns, struct json *columns_json)
|
||
{
|
||
if (ovsdb_column_set_contains(columns, column->index)) {
|
||
ovs_fatal(0, "%s: column \"%s\" mentioned multiple times",
|
||
server, column->name);
|
||
}
|
||
ovsdb_column_set_add(columns, column);
|
||
json_array_add(columns_json, json_string_create(column->name));
|
||
}
|
||
|
||
static struct json *
|
||
parse_monitor_columns(char *arg, const char *server, const char *database,
|
||
const struct ovsdb_table_schema *table,
|
||
struct ovsdb_column_set *columns)
|
||
{
|
||
bool initial, insert, delete, modify;
|
||
struct json *mr, *columns_json;
|
||
char *save_ptr = NULL;
|
||
char *token;
|
||
|
||
mr = json_object_create();
|
||
columns_json = json_array_create_empty();
|
||
json_object_put(mr, "columns", columns_json);
|
||
|
||
initial = insert = delete = modify = true;
|
||
for (token = strtok_r(arg, ",", &save_ptr); token != NULL;
|
||
token = strtok_r(NULL, ",", &save_ptr)) {
|
||
if (!strcmp(token, "!initial")) {
|
||
initial = false;
|
||
} else if (!strcmp(token, "!insert")) {
|
||
insert = false;
|
||
} else if (!strcmp(token, "!delete")) {
|
||
delete = false;
|
||
} else if (!strcmp(token, "!modify")) {
|
||
modify = false;
|
||
} else {
|
||
const struct ovsdb_column *column;
|
||
|
||
column = ovsdb_table_schema_get_column(table, token);
|
||
if (!column) {
|
||
ovs_fatal(0, "%s: table \"%s\" in %s does not have a "
|
||
"column named \"%s\"",
|
||
server, table->name, database, token);
|
||
}
|
||
add_column(server, column, columns, columns_json);
|
||
}
|
||
}
|
||
|
||
if (columns_json->u.array.n == 0) {
|
||
const struct shash_node **nodes;
|
||
size_t i, n;
|
||
|
||
n = shash_count(&table->columns);
|
||
nodes = shash_sort(&table->columns);
|
||
for (i = 0; i < n; i++) {
|
||
const struct ovsdb_column *column = nodes[i]->data;
|
||
if (column->index != OVSDB_COL_UUID
|
||
&& column->index != OVSDB_COL_VERSION) {
|
||
add_column(server, column, columns, columns_json);
|
||
}
|
||
}
|
||
free(nodes);
|
||
|
||
add_column(server, ovsdb_table_schema_get_column(table,"_version"),
|
||
columns, columns_json);
|
||
}
|
||
|
||
if (!initial || !insert || !delete || !modify) {
|
||
struct json *select = json_object_create();
|
||
json_object_put(select, "initial", json_boolean_create(initial));
|
||
json_object_put(select, "insert", json_boolean_create(insert));
|
||
json_object_put(select, "delete", json_boolean_create(delete));
|
||
json_object_put(select, "modify", json_boolean_create(modify));
|
||
json_object_put(mr, "select", select);
|
||
}
|
||
|
||
return mr;
|
||
}
|
||
|
||
static void
|
||
do_monitor(struct jsonrpc *rpc, const char *database,
|
||
int argc, char *argv[])
|
||
{
|
||
const char *server = jsonrpc_get_name(rpc);
|
||
const char *table_name = argv[0];
|
||
struct ovsdb_column_set columns = OVSDB_COLUMN_SET_INITIALIZER;
|
||
struct ovsdb_table_schema *table;
|
||
struct ovsdb_schema *schema;
|
||
struct jsonrpc_msg *request;
|
||
struct json *monitor, *monitor_request_array,
|
||
*monitor_requests, *request_id;
|
||
|
||
schema = fetch_schema(rpc, database);
|
||
table = shash_find_data(&schema->tables, table_name);
|
||
if (!table) {
|
||
ovs_fatal(0, "%s: %s does not have a table named \"%s\"",
|
||
server, database, table_name);
|
||
}
|
||
|
||
monitor_request_array = json_array_create_empty();
|
||
if (argc > 1) {
|
||
int i;
|
||
|
||
for (i = 1; i < argc; i++) {
|
||
json_array_add(
|
||
monitor_request_array,
|
||
parse_monitor_columns(argv[i], server, database, table,
|
||
&columns));
|
||
}
|
||
} else {
|
||
/* Allocate a writable empty string since parse_monitor_columns() is
|
||
* going to strtok() it and that's risky with literal "". */
|
||
char empty[] = "";
|
||
json_array_add(
|
||
monitor_request_array,
|
||
parse_monitor_columns(empty, server, database, table, &columns));
|
||
}
|
||
|
||
monitor_requests = json_object_create();
|
||
json_object_put(monitor_requests, table_name, monitor_request_array);
|
||
|
||
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 (;;) {
|
||
struct jsonrpc_msg *msg;
|
||
int error;
|
||
|
||
error = jsonrpc_recv_block(rpc, &msg);
|
||
if (error) {
|
||
ovsdb_schema_destroy(schema);
|
||
ovs_fatal(error, "%s: receive failed", server);
|
||
}
|
||
|
||
if (msg->type == JSONRPC_REQUEST && !strcmp(msg->method, "echo")) {
|
||
jsonrpc_send(rpc, jsonrpc_create_reply(json_clone(msg->params),
|
||
msg->id));
|
||
} 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;
|
||
if (params->type == JSON_ARRAY
|
||
&& params->u.array.n == 2
|
||
&& params->u.array.elems[0]->type == JSON_NULL) {
|
||
monitor_print(params->u.array.elems[1],
|
||
table, &columns, false);
|
||
fflush(stdout);
|
||
}
|
||
}
|
||
jsonrpc_msg_destroy(msg);
|
||
}
|
||
}
|
||
|
||
struct dump_table_aux {
|
||
struct ovsdb_datum **data;
|
||
const struct ovsdb_column **columns;
|
||
size_t n_columns;
|
||
};
|
||
|
||
static int
|
||
compare_data(size_t a_y, size_t b_y, size_t x,
|
||
const struct dump_table_aux *aux)
|
||
{
|
||
return ovsdb_datum_compare_3way(&aux->data[a_y][x],
|
||
&aux->data[b_y][x],
|
||
&aux->columns[x]->type);
|
||
}
|
||
|
||
static int
|
||
compare_rows(size_t a_y, size_t b_y, void *aux_)
|
||
{
|
||
struct dump_table_aux *aux = aux_;
|
||
size_t x;
|
||
|
||
/* Skip UUID columns on the first pass, since their values tend to be
|
||
* random and make our results less reproducible. */
|
||
for (x = 0; x < aux->n_columns; x++) {
|
||
if (aux->columns[x]->type.key.type != OVSDB_TYPE_UUID) {
|
||
int cmp = compare_data(a_y, b_y, x, aux);
|
||
if (cmp) {
|
||
return cmp;
|
||
}
|
||
}
|
||
}
|
||
|
||
/* Use UUID columns as tie-breakers. */
|
||
for (x = 0; x < aux->n_columns; x++) {
|
||
if (aux->columns[x]->type.key.type == OVSDB_TYPE_UUID) {
|
||
int cmp = compare_data(a_y, b_y, x, aux);
|
||
if (cmp) {
|
||
return cmp;
|
||
}
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static void
|
||
swap_rows(size_t a_y, size_t b_y, void *aux_)
|
||
{
|
||
struct dump_table_aux *aux = aux_;
|
||
struct ovsdb_datum *tmp = aux->data[a_y];
|
||
aux->data[a_y] = aux->data[b_y];
|
||
aux->data[b_y] = tmp;
|
||
}
|
||
|
||
static int
|
||
compare_columns(const void *a_, const void *b_)
|
||
{
|
||
const struct ovsdb_column *const *ap = a_;
|
||
const struct ovsdb_column *const *bp = b_;
|
||
const struct ovsdb_column *a = *ap;
|
||
const struct ovsdb_column *b = *bp;
|
||
|
||
return strcmp(a->name, b->name);
|
||
}
|
||
|
||
static void
|
||
dump_table(const struct ovsdb_table_schema *ts, struct json_array *rows)
|
||
{
|
||
const struct ovsdb_column **columns;
|
||
size_t n_columns;
|
||
|
||
struct ovsdb_datum **data;
|
||
|
||
struct dump_table_aux aux;
|
||
struct shash_node *node;
|
||
struct table t;
|
||
size_t x, y;
|
||
|
||
/* Sort columns by name, for reproducibility. */
|
||
columns = xmalloc(shash_count(&ts->columns) * sizeof *columns);
|
||
n_columns = 0;
|
||
SHASH_FOR_EACH (node, &ts->columns) {
|
||
struct ovsdb_column *column = node->data;
|
||
if (strcmp(column->name, "_version")) {
|
||
columns[n_columns++] = column;
|
||
}
|
||
}
|
||
qsort(columns, n_columns, sizeof *columns, compare_columns);
|
||
|
||
/* Extract data from table. */
|
||
data = xmalloc(rows->n * sizeof *data);
|
||
for (y = 0; y < rows->n; y++) {
|
||
struct shash *row;
|
||
|
||
if (rows->elems[y]->type != JSON_OBJECT) {
|
||
ovs_fatal(0, "row %zu in table %s response is not a JSON object: "
|
||
"%s", y, ts->name, json_to_string(rows->elems[y], 0));
|
||
}
|
||
row = json_object(rows->elems[y]);
|
||
|
||
data[y] = xmalloc(n_columns * sizeof **data);
|
||
for (x = 0; x < n_columns; x++) {
|
||
const struct json *json = shash_find_data(row, columns[x]->name);
|
||
if (!json) {
|
||
ovs_fatal(0, "row %zu in table %s response lacks %s column",
|
||
y, ts->name, columns[x]->name);
|
||
}
|
||
|
||
check_ovsdb_error(ovsdb_datum_from_json(&data[y][x],
|
||
&columns[x]->type,
|
||
json, NULL));
|
||
}
|
||
}
|
||
|
||
/* Sort rows by column values, for reproducibility. */
|
||
aux.data = data;
|
||
aux.columns = columns;
|
||
aux.n_columns = n_columns;
|
||
sort(rows->n, compare_rows, swap_rows, &aux);
|
||
|
||
/* Add column headings. */
|
||
table_init(&t);
|
||
table_set_caption(&t, xasprintf("%s table", ts->name));
|
||
for (x = 0; x < n_columns; x++) {
|
||
table_add_column(&t, "%s", columns[x]->name);
|
||
}
|
||
|
||
/* Print rows. */
|
||
for (y = 0; y < rows->n; y++) {
|
||
table_add_row(&t);
|
||
for (x = 0; x < n_columns; x++) {
|
||
struct cell *cell = table_add_cell(&t);
|
||
cell->json = ovsdb_datum_to_json(&data[y][x], &columns[x]->type);
|
||
cell->type = &columns[x]->type;
|
||
}
|
||
}
|
||
table_print(&t, &table_style);
|
||
table_destroy(&t);
|
||
}
|
||
|
||
static void
|
||
do_dump(struct jsonrpc *rpc, const char *database,
|
||
int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
|
||
{
|
||
struct jsonrpc_msg *request, *reply;
|
||
struct ovsdb_schema *schema;
|
||
struct json *transaction;
|
||
int error;
|
||
|
||
const struct shash_node **tables;
|
||
size_t n_tables;
|
||
|
||
size_t i;
|
||
|
||
schema = fetch_schema(rpc, database);
|
||
tables = shash_sort(&schema->tables);
|
||
n_tables = shash_count(&schema->tables);
|
||
|
||
/* Construct transaction to retrieve entire database. */
|
||
transaction = json_array_create_1(json_string_create(database));
|
||
for (i = 0; i < n_tables; i++) {
|
||
const struct ovsdb_table_schema *ts = tables[i]->data;
|
||
struct json *op, *columns;
|
||
struct shash_node *node;
|
||
|
||
columns = json_array_create_empty();
|
||
SHASH_FOR_EACH (node, &ts->columns) {
|
||
const struct ovsdb_column *column = node->data;
|
||
|
||
if (strcmp(column->name, "_version")) {
|
||
json_array_add(columns, json_string_create(column->name));
|
||
}
|
||
}
|
||
|
||
op = json_object_create();
|
||
json_object_put_string(op, "op", "select");
|
||
json_object_put_string(op, "table", tables[i]->name);
|
||
json_object_put(op, "where", json_array_create_empty());
|
||
json_object_put(op, "columns", columns);
|
||
json_array_add(transaction, op);
|
||
}
|
||
|
||
/* Send request, get reply. */
|
||
request = jsonrpc_create_request("transact", transaction, NULL);
|
||
error = jsonrpc_transact_block(rpc, request, &reply);
|
||
if (error) {
|
||
ovs_fatal(error, "transaction failed");
|
||
}
|
||
|
||
/* Print database contents. */
|
||
if (reply->result->type != JSON_ARRAY
|
||
|| reply->result->u.array.n != n_tables) {
|
||
ovs_fatal(0, "reply is not array of %zu elements: %s",
|
||
n_tables, json_to_string(reply->result, 0));
|
||
}
|
||
for (i = 0; i < n_tables; i++) {
|
||
const struct ovsdb_table_schema *ts = tables[i]->data;
|
||
const struct json *op_result = reply->result->u.array.elems[i];
|
||
struct json *rows;
|
||
|
||
if (op_result->type != JSON_OBJECT
|
||
|| !(rows = shash_find_data(json_object(op_result), "rows"))
|
||
|| rows->type != JSON_ARRAY) {
|
||
ovs_fatal(0, "%s table reply is not an object with a \"rows\" "
|
||
"member array: %s",
|
||
ts->name, json_to_string(op_result, 0));
|
||
}
|
||
|
||
dump_table(ts, &rows->u.array);
|
||
}
|
||
}
|
||
|
||
static void
|
||
do_help(struct jsonrpc *rpc OVS_UNUSED, const char *database OVS_UNUSED,
|
||
int argc OVS_UNUSED, char *argv[] OVS_UNUSED)
|
||
{
|
||
usage();
|
||
}
|
||
|
||
/* All command handlers (except for "help") are expected to take an optional
|
||
* server socket name (e.g. "unix:...") as their first argument. The socket
|
||
* name argument must be included in max_args (but left out of min_args). The
|
||
* command name and socket name are not included in the arguments passed to the
|
||
* handler: the argv[0] passed to the handler is the first argument after the
|
||
* optional server socket name. The connection to the server is available as
|
||
* global variable 'rpc'. */
|
||
static const struct ovsdb_client_command all_commands[] = {
|
||
{ "list-dbs", NEED_RPC, 0, 0, do_list_dbs },
|
||
{ "get-schema", NEED_DATABASE, 0, 0, do_get_schema },
|
||
{ "get-schema-version", NEED_DATABASE, 0, 0, do_get_schema_version },
|
||
{ "list-tables", NEED_DATABASE, 0, 0, do_list_tables },
|
||
{ "list-columns", NEED_DATABASE, 0, 1, do_list_columns },
|
||
{ "transact", NEED_RPC, 1, 1, do_transact },
|
||
{ "monitor", NEED_DATABASE, 1, INT_MAX, do_monitor },
|
||
{ "dump", NEED_DATABASE, 0, 0, do_dump },
|
||
|
||
{ "help", NEED_NONE, 0, INT_MAX, do_help },
|
||
|
||
{ NULL, 0, 0, 0, NULL },
|
||
};
|