/* * Copyright (c) 2009, 2010, 2011, 2012, 2014 Nicira, Inc. * * 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 #include #include #include #include #include #include #include #include #include #include #include "command-line.h" #include "compiler.h" #include "dirs.h" #include "dynamic-string.h" #include "fatal-signal.h" #include "hash.h" #include "json.h" #include "ovsdb-data.h" #include "ovsdb-idl.h" #include "poll-loop.h" #include "process.h" #include "stream.h" #include "stream-ssl.h" #include "smap.h" #include "sset.h" #include "svec.h" #include "vtep/vtep-idl.h" #include "table.h" #include "timeval.h" #include "util.h" #include "openvswitch/vconn.h" #include "openvswitch/vlog.h" VLOG_DEFINE_THIS_MODULE(vtep_ctl); /* vtep_ctl_fatal() also logs the error, so it is preferred in this file. */ #define ovs_fatal please_use_vtep_ctl_fatal_instead_of_ovs_fatal struct vtep_ctl_context; /* A command supported by vtep-ctl. */ struct vtep_ctl_command_syntax { const char *name; /* e.g. "add-ps" */ int min_args; /* Min number of arguments following name. */ int max_args; /* Max number of arguments following name. */ /* If nonnull, calls ovsdb_idl_add_column() or ovsdb_idl_add_table() for * each column or table in ctx->idl that it uses. */ void (*prerequisites)(struct vtep_ctl_context *ctx); /* Does the actual work of the command and puts the command's output, if * any, in ctx->output or ctx->table. * * Alternatively, if some prerequisite of the command is not met and the * caller should wait for something to change and then retry, it may set * ctx->try_again to true. (Only the "wait-until" command currently does * this.) */ void (*run)(struct vtep_ctl_context *ctx); /* If nonnull, called after the transaction has been successfully * committed. ctx->output is the output from the "run" function, which * this function may modify and otherwise postprocess as needed. (Only the * "create" command currently does any postprocessing.) */ void (*postprocess)(struct vtep_ctl_context *ctx); /* A comma-separated list of supported options, e.g. "--a,--b", or the * empty string if the command does not support any options. */ const char *options; enum { RO, RW } mode; /* Does this command modify the database? */ }; struct vtep_ctl_command { /* Data that remains constant after initialization. */ const struct vtep_ctl_command_syntax *syntax; int argc; char **argv; struct shash options; /* Data modified by commands. */ struct ds output; struct table *table; }; /* --db: The database server to contact. */ static const char *db; /* --oneline: Write each command's output as a single line? */ static bool oneline; /* --dry-run: Do not commit any changes. */ static bool dry_run; /* --timeout: Time to wait for a connection to 'db'. */ static int timeout; /* Format for table output. */ static struct table_style table_style = TABLE_STYLE_DEFAULT; /* The IDL we're using and the current transaction, if any. * This is for use by vtep_ctl_exit() only, to allow it to clean up. * Other code should use its context arguments. */ static struct ovsdb_idl *the_idl; static struct ovsdb_idl_txn *the_idl_txn; OVS_NO_RETURN static void vtep_ctl_exit(int status); OVS_NO_RETURN static void vtep_ctl_fatal(const char *, ...) OVS_PRINTF_FORMAT(1, 2); static char *default_db(void); OVS_NO_RETURN static void usage(void); static void parse_options(int argc, char *argv[], struct shash *local_options); static bool might_write_to_db(char **argv); static struct vtep_ctl_command *parse_commands(int argc, char *argv[], struct shash *local_options, size_t *n_commandsp); static void parse_command(int argc, char *argv[], struct shash *local_options, struct vtep_ctl_command *); static const struct vtep_ctl_command_syntax *find_command(const char *name); static void run_prerequisites(struct vtep_ctl_command[], size_t n_commands, struct ovsdb_idl *); static void do_vtep_ctl(const char *args, struct vtep_ctl_command *, size_t n, struct ovsdb_idl *); static const struct vtep_ctl_table_class *get_table(const char *table_name); static void set_column(const struct vtep_ctl_table_class *, const struct ovsdb_idl_row *, const char *arg, struct ovsdb_symbol_table *); static bool is_condition_satisfied(const struct vtep_ctl_table_class *, const struct ovsdb_idl_row *, const char *arg, struct ovsdb_symbol_table *); static struct vtep_ctl_lswitch *find_lswitch(struct vtep_ctl_context *, const char *name, bool must_exist); static const struct vtep_ctl_command_syntax *get_all_commands(void); int main(int argc, char *argv[]) { extern struct vlog_module VLM_reconnect; struct ovsdb_idl *idl; struct vtep_ctl_command *commands; struct shash local_options; unsigned int seqno; size_t n_commands; char *args; set_program_name(argv[0]); fatal_ignore_sigpipe(); vlog_set_levels(NULL, VLF_CONSOLE, VLL_WARN); vlog_set_levels(&VLM_reconnect, VLF_ANY_DESTINATION, VLL_WARN); vteprec_init(); /* Log our arguments. This is often valuable for debugging systems. */ args = process_escape_args(argv); VLOG(might_write_to_db(argv) ? VLL_INFO : VLL_DBG, "Called as %s", args); /* Parse command line. */ shash_init(&local_options); parse_options(argc, argv, &local_options); commands = parse_commands(argc - optind, argv + optind, &local_options, &n_commands); if (timeout) { time_alarm(timeout); } /* Initialize IDL. */ idl = the_idl = ovsdb_idl_create(db, &vteprec_idl_class, false, false); run_prerequisites(commands, n_commands, idl); /* Execute the commands. * * 'seqno' is the database sequence number for which we last tried to * execute our transaction. There's no point in trying to commit more than * once for any given sequence number, because if the transaction fails * it's because the database changed and we need to obtain an up-to-date * view of the database before we try the transaction again. */ seqno = ovsdb_idl_get_seqno(idl); for (;;) { ovsdb_idl_run(idl); if (seqno != ovsdb_idl_get_seqno(idl)) { seqno = ovsdb_idl_get_seqno(idl); do_vtep_ctl(args, commands, n_commands, idl); } if (seqno == ovsdb_idl_get_seqno(idl)) { ovsdb_idl_wait(idl); poll_block(); } } } static struct option * find_option(const char *name, struct option *options, size_t n_options) { size_t i; for (i = 0; i < n_options; i++) { if (!strcmp(options[i].name, name)) { return &options[i]; } } return NULL; } static struct option * add_option(struct option **optionsp, size_t *n_optionsp, size_t *allocated_optionsp) { if (*n_optionsp >= *allocated_optionsp) { *optionsp = x2nrealloc(*optionsp, allocated_optionsp, sizeof **optionsp); } return &(*optionsp)[(*n_optionsp)++]; } static void parse_options(int argc, char *argv[], struct shash *local_options) { enum { OPT_DB = UCHAR_MAX + 1, OPT_ONELINE, OPT_NO_SYSLOG, OPT_DRY_RUN, OPT_PEER_CA_CERT, OPT_LOCAL, VLOG_OPTION_ENUMS, TABLE_OPTION_ENUMS }; static const struct option global_long_options[] = { {"db", required_argument, NULL, OPT_DB}, {"no-syslog", no_argument, NULL, OPT_NO_SYSLOG}, {"dry-run", no_argument, NULL, OPT_DRY_RUN}, {"oneline", no_argument, NULL, OPT_ONELINE}, {"timeout", required_argument, NULL, 't'}, {"help", no_argument, NULL, 'h'}, {"version", no_argument, NULL, 'V'}, VLOG_LONG_OPTIONS, TABLE_LONG_OPTIONS, STREAM_SSL_LONG_OPTIONS, {"peer-ca-cert", required_argument, NULL, OPT_PEER_CA_CERT}, {NULL, 0, NULL, 0}, }; const int n_global_long_options = ARRAY_SIZE(global_long_options) - 1; char *tmp, *short_options; const struct vtep_ctl_command_syntax *p; struct option *options, *o; size_t allocated_options; size_t n_options; size_t i; tmp = ovs_cmdl_long_options_to_short_options(global_long_options); short_options = xasprintf("+%s", tmp); free(tmp); /* We want to parse both global and command-specific options here, but * getopt_long() isn't too convenient for the job. We copy our global * options into a dynamic array, then append all of the command-specific * options. */ options = xmemdup(global_long_options, sizeof global_long_options); allocated_options = ARRAY_SIZE(global_long_options); n_options = n_global_long_options; for (p = get_all_commands(); p->name; p++) { if (p->options[0]) { char *save_ptr = NULL; char *name; char *s; s = xstrdup(p->options); for (name = strtok_r(s, ",", &save_ptr); name != NULL; name = strtok_r(NULL, ",", &save_ptr)) { char *equals; int has_arg; ovs_assert(name[0] == '-' && name[1] == '-' && name[2]); name += 2; equals = strchr(name, '='); if (equals) { has_arg = required_argument; *equals = '\0'; } else { has_arg = no_argument; } o = find_option(name, options, n_options); if (o) { ovs_assert(o - options >= n_global_long_options); ovs_assert(o->has_arg == has_arg); } else { o = add_option(&options, &n_options, &allocated_options); o->name = xstrdup(name); o->has_arg = has_arg; o->flag = NULL; o->val = OPT_LOCAL; } } free(s); } } o = add_option(&options, &n_options, &allocated_options); memset(o, 0, sizeof *o); table_style.format = TF_LIST; for (;;) { int idx; int c; c = getopt_long(argc, argv, short_options, options, &idx); if (c == -1) { break; } switch (c) { case OPT_DB: db = optarg; break; case OPT_ONELINE: oneline = true; break; case OPT_NO_SYSLOG: vlog_set_levels(&VLM_vtep_ctl, VLF_SYSLOG, VLL_WARN); break; case OPT_DRY_RUN: dry_run = true; break; case OPT_LOCAL: if (shash_find(local_options, options[idx].name)) { vtep_ctl_fatal("'%s' option specified multiple times", options[idx].name); } shash_add_nocopy(local_options, xasprintf("--%s", options[idx].name), optarg ? xstrdup(optarg) : NULL); break; case 'h': usage(); case 'V': ovs_print_version(0, 0); printf("DB Schema %s\n", vteprec_get_db_version()); exit(EXIT_SUCCESS); case 't': timeout = strtoul(optarg, NULL, 10); if (timeout < 0) { vtep_ctl_fatal("value %s on -t or --timeout is invalid", optarg); } break; VLOG_OPTION_HANDLERS TABLE_OPTION_HANDLERS(&table_style) STREAM_SSL_OPTION_HANDLERS case OPT_PEER_CA_CERT: stream_ssl_set_peer_ca_cert_file(optarg); break; case '?': exit(EXIT_FAILURE); default: abort(); } } free(short_options); if (!db) { db = default_db(); } for (i = n_global_long_options; options[i].name; i++) { free(CONST_CAST(char *, options[i].name)); } free(options); } static struct vtep_ctl_command * parse_commands(int argc, char *argv[], struct shash *local_options, size_t *n_commandsp) { struct vtep_ctl_command *commands; size_t n_commands, allocated_commands; int i, start; commands = NULL; n_commands = allocated_commands = 0; for (start = i = 0; i <= argc; i++) { if (i == argc || !strcmp(argv[i], "--")) { if (i > start) { if (n_commands >= allocated_commands) { struct vtep_ctl_command *c; commands = x2nrealloc(commands, &allocated_commands, sizeof *commands); for (c = commands; c < &commands[n_commands]; c++) { shash_moved(&c->options); } } parse_command(i - start, &argv[start], local_options, &commands[n_commands++]); } else if (!shash_is_empty(local_options)) { vtep_ctl_fatal("missing command name (use --help for help)"); } start = i + 1; } } if (!n_commands) { vtep_ctl_fatal("missing command name (use --help for help)"); } *n_commandsp = n_commands; return commands; } static void parse_command(int argc, char *argv[], struct shash *local_options, struct vtep_ctl_command *command) { const struct vtep_ctl_command_syntax *p; struct shash_node *node; int n_arg; int i; shash_init(&command->options); shash_swap(local_options, &command->options); for (i = 0; i < argc; i++) { const char *option = argv[i]; const char *equals; char *key, *value; if (option[0] != '-') { break; } equals = strchr(option, '='); if (equals) { key = xmemdup0(option, equals - option); value = xstrdup(equals + 1); } else { key = xstrdup(option); value = NULL; } if (shash_find(&command->options, key)) { vtep_ctl_fatal("'%s' option specified multiple times", argv[i]); } shash_add_nocopy(&command->options, key, value); } if (i == argc) { vtep_ctl_fatal("missing command name (use --help for help)"); } p = find_command(argv[i]); if (!p) { vtep_ctl_fatal("unknown command '%s'; use --help for help", argv[i]); } SHASH_FOR_EACH (node, &command->options) { const char *s = strstr(p->options, node->name); int end = s ? s[strlen(node->name)] : EOF; if (end != '=' && end != ',' && end != ' ' && end != '\0') { vtep_ctl_fatal("'%s' command has no '%s' option", argv[i], node->name); } if ((end == '=') != (node->data != NULL)) { if (end == '=') { vtep_ctl_fatal("missing argument to '%s' option on '%s' " "command", node->name, argv[i]); } else { vtep_ctl_fatal("'%s' option on '%s' does not accept an " "argument", node->name, argv[i]); } } } n_arg = argc - i - 1; if (n_arg < p->min_args) { vtep_ctl_fatal("'%s' command requires at least %d arguments", p->name, p->min_args); } else if (n_arg > p->max_args) { int j; for (j = i + 1; j < argc; j++) { if (argv[j][0] == '-') { vtep_ctl_fatal("'%s' command takes at most %d arguments " "(note that options must precede command " "names and follow a \"--\" argument)", p->name, p->max_args); } } vtep_ctl_fatal("'%s' command takes at most %d arguments", p->name, p->max_args); } command->syntax = p; command->argc = n_arg + 1; command->argv = &argv[i]; } /* Returns the "struct vtep_ctl_command_syntax" for a given command 'name', or a * null pointer if there is none. */ static const struct vtep_ctl_command_syntax * find_command(const char *name) { static struct shash commands = SHASH_INITIALIZER(&commands); if (shash_is_empty(&commands)) { const struct vtep_ctl_command_syntax *p; for (p = get_all_commands(); p->name; p++) { shash_add_assert(&commands, p->name, p); } } return shash_find_data(&commands, name); } static void vtep_ctl_fatal(const char *format, ...) { char *message; va_list args; va_start(args, format); message = xvasprintf(format, args); va_end(args); vlog_set_levels(&VLM_vtep_ctl, VLF_CONSOLE, VLL_OFF); VLOG_ERR("%s", message); ovs_error(0, "%s", message); vtep_ctl_exit(EXIT_FAILURE); } /* Frees the current transaction and the underlying IDL and then calls * exit(status). * * Freeing the transaction and the IDL is not strictly necessary, but it makes * for a clean memory leak report from valgrind in the normal case. That makes * it easier to notice real memory leaks. */ static void vtep_ctl_exit(int status) { if (the_idl_txn) { ovsdb_idl_txn_abort(the_idl_txn); ovsdb_idl_txn_destroy(the_idl_txn); } ovsdb_idl_destroy(the_idl); exit(status); } static void usage(void) { printf("\ %s: VTEP configuration utility\n\ usage: %s [OPTIONS] COMMAND [ARG...]\n\ \n\ Manager commands:\n\ get-manager print the managers\n\ del-manager delete the managers\n\ set-manager TARGET... set the list of managers to TARGET...\n\ \n\ Physical Switch commands:\n\ add-ps PS create a new physical switch named PS\n\ del-ps PS delete PS and all of its ports\n\ list-ps print the names of all the physical switches\n\ ps-exists PS exit 2 if PS does not exist\n\ \n\ Port commands:\n\ list-ports PS print the names of all the ports on PS\n\ add-port PS PORT add network device PORT to PS\n\ del-port PS PORT delete PORT from PS\n\ \n\ Logical Switch commands:\n\ add-ls LS create a new logical switch named LS\n\ del-ls LS delete LS and all of its ports\n\ list-ls print the names of all the logical switches\n\ ls-exists LS exit 2 if LS does not exist\n\ bind-ls PS PORT VLAN LS bind LS to VLAN on PORT\n\ unbind-ls PS PORT VLAN unbind logical switch on VLAN from PORT\n\ list-bindings PS PORT list bindings for PORT on PS\n\ \n\ MAC binding commands:\n\ add-ucast-local LS MAC [ENCAP] IP add ucast local entry in LS\n\ del-ucast-local LS MAC del ucast local entry from LS\n\ add-mcast-local LS MAC [ENCAP] IP add mcast local entry in LS\n\ del-mcast-local LS MAC [ENCAP] IP del mcast local entry from LS\n\ clear-local-macs LS clear local mac entries\n\ list-local-macs LS list local mac entries\n\ add-ucast-remote LS MAC [ENCAP] IP add ucast remote entry in LS\n\ del-ucast-remote LS MAC del ucast remote entry from LS\n\ add-mcast-remote LS MAC [ENCAP] IP add mcast remote entry in LS\n\ del-mcast-remote LS MAC [ENCAP] IP del mcast remote entry from LS\n\ clear-remote-macs LS clear remote mac entries\n\ list-remote-macs LS list remote mac entries\n\ \n\ Database commands:\n\ list TBL [REC] list RECord (or all records) in TBL\n\ find TBL CONDITION... list records satisfying CONDITION in TBL\n\ get TBL REC COL[:KEY] print values of COLumns in RECord in TBL\n\ set TBL REC COL[:KEY]=VALUE set COLumn values in RECord in TBL\n\ add TBL REC COL [KEY=]VALUE add (KEY=)VALUE to COLumn in RECord in TBL\n\ remove TBL REC COL [KEY=]VALUE remove (KEY=)VALUE from COLumn\n\ clear TBL REC COL clear values from COLumn in RECord in TBL\n\ create TBL COL[:KEY]=VALUE create and initialize new record\n\ destroy TBL REC delete RECord from TBL\n\ wait-until TBL REC [COL[:KEY]=VALUE] wait until condition is true\n\ Potentially unsafe database commands require --force option.\n\ \n\ Options:\n\ --db=DATABASE connect to DATABASE\n\ (default: %s)\n\ -t, --timeout=SECS wait at most SECS seconds\n\ --dry-run do not commit changes to database\n\ --oneline print exactly one line of output per command\n", program_name, program_name, default_db()); vlog_usage(); printf("\ --no-syslog equivalent to --verbose=vtep_ctl:syslog:warn\n"); stream_usage("database", true, true, false); printf("\n\ Other options:\n\ -h, --help display this help message\n\ -V, --version display version information\n"); exit(EXIT_SUCCESS); } static char * default_db(void) { static char *def; if (!def) { def = xasprintf("unix:%s/db.sock", ovs_rundir()); } return def; } /* Returns true if it looks like this set of arguments might modify the * database, otherwise false. (Not very smart, so it's prone to false * positives.) */ static bool might_write_to_db(char **argv) { for (; *argv; argv++) { const struct vtep_ctl_command_syntax *p = find_command(*argv); if (p && p->mode == RW) { return true; } } return false; } struct vtep_ctl_context { /* Read-only. */ int argc; char **argv; struct shash options; /* Modifiable state. */ struct ds output; struct table *table; struct ovsdb_idl *idl; struct ovsdb_idl_txn *txn; struct ovsdb_symbol_table *symtab; const struct vteprec_global *vtep_global; bool verified_ports; /* A cache of the contents of the database. * * A command that needs to use any of this information must first * call vtep_ctl_context_populate_cache(). A command that changes * anything that could invalidate the cache must either call * vtep_ctl_context_invalidate_cache() or manually update the cache * to maintain its correctness. */ bool cache_valid; struct shash pswitches; /* Maps from physical switch name to * struct vtep_ctl_pswitch. */ struct shash ports; /* Maps from port name to struct vtep_ctl_port. */ struct shash lswitches; /* Maps from logical switch name to * struct vtep_ctl_lswitch. */ struct shash plocs; /* Maps from "+" to * struct vteprec_physical_locator. */ /* A command may set this member to true if some prerequisite is not met * and the caller should wait for something to change and then retry. */ bool try_again; }; struct vtep_ctl_pswitch { const struct vteprec_physical_switch *ps_cfg; char *name; struct ovs_list ports; /* Contains "struct vteprec_physical_port"s. */ }; struct vtep_ctl_port { struct ovs_list ports_node; /* In struct vtep_ctl_pswitch's 'ports' list. */ const struct vteprec_physical_port *port_cfg; struct vtep_ctl_pswitch *ps; struct shash bindings; /* Maps from vlan to vtep_ctl_lswitch. */ }; struct vtep_ctl_lswitch { const struct vteprec_logical_switch *ls_cfg; char *name; struct shash ucast_local; /* Maps from mac to vteprec_ucast_macs_local. */ struct shash ucast_remote; /* Maps from mac to vteprec_ucast_macs_remote.*/ struct shash mcast_local; /* Maps from mac to vtep_ctl_mcast_mac. */ struct shash mcast_remote; /* Maps from mac to vtep_ctl_mcast_mac. */ }; struct vtep_ctl_mcast_mac { const struct vteprec_mcast_macs_local *local_cfg; const struct vteprec_mcast_macs_remote *remote_cfg; const struct vteprec_physical_locator_set *ploc_set_cfg; struct ovs_list locators; /* Contains 'vtep_ctl_ploc's. */ }; struct vtep_ctl_ploc { struct ovs_list locators_node; /* In struct vtep_ctl_ploc_set's 'locators' list. */ const struct vteprec_physical_locator *ploc_cfg; }; static void verify_ports(struct vtep_ctl_context *ctx) { if (!ctx->verified_ports) { const struct vteprec_physical_switch *ps; vteprec_global_verify_switches(ctx->vtep_global); VTEPREC_PHYSICAL_SWITCH_FOR_EACH (ps, ctx->idl) { vteprec_physical_switch_verify_ports(ps); } ctx->verified_ports = true; } } static struct vtep_ctl_port * add_port_to_cache(struct vtep_ctl_context *ctx, struct vtep_ctl_pswitch *ps, struct vteprec_physical_port *port_cfg) { char *cache_name = xasprintf("%s+%s", ps->name, port_cfg->name); struct vtep_ctl_port *port; port = xmalloc(sizeof *port); list_push_back(&ps->ports, &port->ports_node); port->port_cfg = port_cfg; port->ps = ps; shash_add(&ctx->ports, cache_name, port); free(cache_name); shash_init(&port->bindings); return port; } static void del_cached_port(struct vtep_ctl_context *ctx, struct vtep_ctl_port *port) { char *cache_name = xasprintf("%s+%s", port->ps->name, port->port_cfg->name); list_remove(&port->ports_node); shash_find_and_delete(&ctx->ports, port->port_cfg->name); vteprec_physical_port_delete(port->port_cfg); free(cache_name); free(port); } static struct vtep_ctl_pswitch * add_pswitch_to_cache(struct vtep_ctl_context *ctx, struct vteprec_physical_switch *ps_cfg) { struct vtep_ctl_pswitch *ps = xmalloc(sizeof *ps); ps->ps_cfg = ps_cfg; ps->name = xstrdup(ps_cfg->name); list_init(&ps->ports); shash_add(&ctx->pswitches, ps->name, ps); return ps; } static void vtep_delete_pswitch(const struct vteprec_global *vtep_global, const struct vteprec_physical_switch *ps) { struct vteprec_physical_switch **pswitches; size_t i, n; pswitches = xmalloc(sizeof *vtep_global->switches * vtep_global->n_switches); for (i = n = 0; i < vtep_global->n_switches; i++) { if (vtep_global->switches[i] != ps) { pswitches[n++] = vtep_global->switches[i]; } } vteprec_global_set_switches(vtep_global, pswitches, n); free(pswitches); } static void del_cached_pswitch(struct vtep_ctl_context *ctx, struct vtep_ctl_pswitch *ps) { ovs_assert(list_is_empty(&ps->ports)); if (ps->ps_cfg) { vteprec_physical_switch_delete(ps->ps_cfg); vtep_delete_pswitch(ctx->vtep_global, ps->ps_cfg); } shash_find_and_delete(&ctx->pswitches, ps->name); free(ps->name); free(ps); } static struct vtep_ctl_lswitch * add_lswitch_to_cache(struct vtep_ctl_context *ctx, const struct vteprec_logical_switch *ls_cfg) { struct vtep_ctl_lswitch *ls = xmalloc(sizeof *ls); ls->ls_cfg = ls_cfg; ls->name = xstrdup(ls_cfg->name); shash_add(&ctx->lswitches, ls->name, ls); shash_init(&ls->ucast_local); shash_init(&ls->ucast_remote); shash_init(&ls->mcast_local); shash_init(&ls->mcast_remote); return ls; } static void del_cached_lswitch(struct vtep_ctl_context *ctx, struct vtep_ctl_lswitch *ls) { if (ls->ls_cfg) { vteprec_logical_switch_delete(ls->ls_cfg); } shash_find_and_delete(&ctx->lswitches, ls->name); free(ls->name); free(ls); } static void commit_ls_bindings(struct vtep_ctl_port *port) { struct vteprec_logical_switch **binding_values; int64_t *binding_keys; size_t n_bindings; struct shash_node *node; int i; n_bindings = shash_count(&port->bindings); binding_keys = xmalloc(n_bindings * sizeof *binding_keys); binding_values = xmalloc(n_bindings * sizeof *binding_values); i = 0; SHASH_FOR_EACH(node, &port->bindings) { struct vtep_ctl_lswitch *ls_entry = node->data; binding_keys[i] = strtoll(node->name, NULL, 0); binding_values[i] = (struct vteprec_logical_switch *)ls_entry->ls_cfg; i++; } vteprec_physical_port_set_vlan_bindings(port->port_cfg, binding_keys, binding_values, n_bindings); free(binding_values); free(binding_keys); } static void add_ls_binding_to_cache(struct vtep_ctl_port *port, const char *vlan, struct vtep_ctl_lswitch *ls) { if (shash_find(&port->bindings, vlan)) { vtep_ctl_fatal("multiple bindings for vlan %s", vlan); } shash_add(&port->bindings, vlan, ls); } static void del_cached_ls_binding(struct vtep_ctl_port *port, const char *vlan) { if (!shash_find(&port->bindings, vlan)) { vtep_ctl_fatal("no binding for vlan %s", vlan); } shash_find_and_delete(&port->bindings, vlan); } static struct vteprec_physical_locator * find_ploc(struct vtep_ctl_context *ctx, const char *encap, const char *dst_ip) { struct vteprec_physical_locator *ploc; char *name = xasprintf("%s+%s", encap, dst_ip); ovs_assert(ctx->cache_valid); ploc = shash_find_data(&ctx->plocs, name); free(name); return ploc; } static void add_ploc_to_cache(struct vtep_ctl_context *ctx, struct vteprec_physical_locator *ploc) { char *name = xasprintf("%s+%s", ploc->encapsulation_type, ploc->dst_ip); struct vteprec_physical_locator *orig_ploc; orig_ploc = find_ploc(ctx, ploc->encapsulation_type, ploc->dst_ip); if (!orig_ploc) { shash_add(&ctx->plocs, name, ploc); } free(name); } static void add_ploc_to_mcast_mac(struct vtep_ctl_mcast_mac *mcast_mac, struct vteprec_physical_locator *ploc_cfg) { struct vtep_ctl_ploc *ploc; ploc = xmalloc(sizeof *ploc); ploc->ploc_cfg = ploc_cfg; list_push_back(&mcast_mac->locators, &ploc->locators_node); } static void del_ploc_from_mcast_mac(struct vtep_ctl_mcast_mac *mcast_mac, struct vteprec_physical_locator *ploc_cfg) { struct vtep_ctl_ploc *ploc; LIST_FOR_EACH (ploc, locators_node, &mcast_mac->locators) { if (ploc->ploc_cfg == ploc_cfg) { list_remove(&ploc->locators_node); free(ploc); return; } } } static struct vtep_ctl_mcast_mac * add_mcast_mac_to_cache(struct vtep_ctl_context *ctx, struct vtep_ctl_lswitch *ls, const char *mac, struct vteprec_physical_locator_set *ploc_set_cfg, bool local) { struct vtep_ctl_mcast_mac *mcast_mac; struct shash *mcast_shash; size_t i; mcast_mac = xmalloc(sizeof *mcast_mac); mcast_shash = local ? &ls->mcast_local : &ls->mcast_remote; mcast_mac->ploc_set_cfg = ploc_set_cfg; list_init(&mcast_mac->locators); shash_add(mcast_shash, mac, mcast_mac); for (i = 0; i < ploc_set_cfg->n_locators; i++) { struct vteprec_physical_locator *ploc_cfg; ploc_cfg = ploc_set_cfg->locators[i]; add_ploc_to_mcast_mac(mcast_mac, ploc_cfg); add_ploc_to_cache(ctx, ploc_cfg); } return mcast_mac; } static void vtep_ctl_context_invalidate_cache(struct vtep_ctl_context *ctx) { struct shash_node *node; if (!ctx->cache_valid) { return; } ctx->cache_valid = false; SHASH_FOR_EACH (node, &ctx->pswitches) { struct vtep_ctl_pswitch *ps = node->data; free(ps->name); free(ps); } shash_destroy(&ctx->pswitches); SHASH_FOR_EACH (node, &ctx->ports) { struct vtep_ctl_port *port = node->data; shash_destroy(&port->bindings); } shash_destroy_free_data(&ctx->ports); SHASH_FOR_EACH (node, &ctx->lswitches) { struct vtep_ctl_lswitch *ls = node->data; struct shash_node *node2, *next_node2; shash_destroy(&ls->ucast_local); shash_destroy(&ls->ucast_remote); SHASH_FOR_EACH_SAFE (node2, next_node2, &ls->mcast_local) { struct vtep_ctl_mcast_mac *mcast_mac = node2->data; struct vtep_ctl_ploc *ploc, *next_ploc; LIST_FOR_EACH_SAFE (ploc, next_ploc, locators_node, &mcast_mac->locators) { free(ploc); } free(mcast_mac); } shash_destroy(&ls->mcast_local); SHASH_FOR_EACH_SAFE (node2, next_node2, &ls->mcast_remote) { struct vtep_ctl_mcast_mac *mcast_mac = node2->data; struct vtep_ctl_ploc *ploc, *next_ploc; LIST_FOR_EACH_SAFE (ploc, next_ploc, locators_node, &mcast_mac->locators) { free(ploc); } free(mcast_mac); } shash_destroy(&ls->mcast_remote); free(ls->name); free(ls); } shash_destroy(&ctx->lswitches); shash_destroy(&ctx->plocs); } static void pre_get_info(struct vtep_ctl_context *ctx) { ovsdb_idl_add_column(ctx->idl, &vteprec_global_col_switches); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_switch_col_name); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_switch_col_ports); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_switch_col_tunnels); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_port_col_name); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_port_col_vlan_bindings); ovsdb_idl_add_column(ctx->idl, &vteprec_logical_switch_col_name); ovsdb_idl_add_column(ctx->idl, &vteprec_ucast_macs_local_col_MAC); ovsdb_idl_add_column(ctx->idl, &vteprec_ucast_macs_local_col_locator); ovsdb_idl_add_column(ctx->idl, &vteprec_ucast_macs_local_col_logical_switch); ovsdb_idl_add_column(ctx->idl, &vteprec_ucast_macs_remote_col_MAC); ovsdb_idl_add_column(ctx->idl, &vteprec_ucast_macs_remote_col_locator); ovsdb_idl_add_column(ctx->idl, &vteprec_ucast_macs_remote_col_logical_switch); ovsdb_idl_add_column(ctx->idl, &vteprec_mcast_macs_local_col_MAC); ovsdb_idl_add_column(ctx->idl, &vteprec_mcast_macs_local_col_locator_set); ovsdb_idl_add_column(ctx->idl, &vteprec_mcast_macs_local_col_logical_switch); ovsdb_idl_add_column(ctx->idl, &vteprec_mcast_macs_remote_col_MAC); ovsdb_idl_add_column(ctx->idl, &vteprec_mcast_macs_remote_col_locator_set); ovsdb_idl_add_column(ctx->idl, &vteprec_mcast_macs_remote_col_logical_switch); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_locator_set_col_locators); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_locator_col_dst_ip); ovsdb_idl_add_column(ctx->idl, &vteprec_physical_locator_col_encapsulation_type); ovsdb_idl_add_column(ctx->idl, &vteprec_tunnel_col_local); ovsdb_idl_add_column(ctx->idl, &vteprec_tunnel_col_remote); } static void vtep_ctl_context_populate_cache(struct vtep_ctl_context *ctx) { const struct vteprec_global *vtep_global = ctx->vtep_global; const struct vteprec_logical_switch *ls_cfg; const struct vteprec_ucast_macs_local *ucast_local_cfg; const struct vteprec_ucast_macs_remote *ucast_remote_cfg; const struct vteprec_mcast_macs_local *mcast_local_cfg; const struct vteprec_mcast_macs_remote *mcast_remote_cfg; const struct vteprec_tunnel *tunnel_cfg; struct sset pswitches, ports, lswitches; size_t i; if (ctx->cache_valid) { /* Cache is already populated. */ return; } ctx->cache_valid = true; shash_init(&ctx->pswitches); shash_init(&ctx->ports); shash_init(&ctx->lswitches); shash_init(&ctx->plocs); sset_init(&pswitches); sset_init(&ports); for (i = 0; i < vtep_global->n_switches; i++) { struct vteprec_physical_switch *ps_cfg = vtep_global->switches[i]; struct vtep_ctl_pswitch *ps; size_t j; if (!sset_add(&pswitches, ps_cfg->name)) { VLOG_WARN("%s: database contains duplicate physical switch name", ps_cfg->name); continue; } ps = add_pswitch_to_cache(ctx, ps_cfg); if (!ps) { continue; } for (j = 0; j < ps_cfg->n_ports; j++) { struct vteprec_physical_port *port_cfg = ps_cfg->ports[j]; if (!sset_add(&ports, port_cfg->name)) { /* Duplicate port name. (We will warn about that later.) */ continue; } } } sset_destroy(&pswitches); sset_destroy(&ports); sset_init(&lswitches); VTEPREC_LOGICAL_SWITCH_FOR_EACH (ls_cfg, ctx->idl) { if (!sset_add(&lswitches, ls_cfg->name)) { VLOG_WARN("%s: database contains duplicate logical switch name", ls_cfg->name); continue; } add_lswitch_to_cache(ctx, ls_cfg); } sset_destroy(&lswitches); VTEPREC_UCAST_MACS_LOCAL_FOR_EACH (ucast_local_cfg, ctx->idl) { struct vtep_ctl_lswitch *ls; if (!ucast_local_cfg->logical_switch) { continue; } ls = find_lswitch(ctx, ucast_local_cfg->logical_switch->name, false); if (!ls) { continue; } if (ucast_local_cfg->locator) { add_ploc_to_cache(ctx, ucast_local_cfg->locator); } shash_add(&ls->ucast_local, ucast_local_cfg->MAC, ucast_local_cfg); } VTEPREC_UCAST_MACS_REMOTE_FOR_EACH (ucast_remote_cfg, ctx->idl) { struct vtep_ctl_lswitch *ls; if (!ucast_remote_cfg->logical_switch) { continue; } ls = find_lswitch(ctx, ucast_remote_cfg->logical_switch->name, false); if (!ls) { continue; } if (ucast_remote_cfg->locator) { add_ploc_to_cache(ctx, ucast_remote_cfg->locator); } shash_add(&ls->ucast_remote, ucast_remote_cfg->MAC, ucast_remote_cfg); } VTEPREC_MCAST_MACS_LOCAL_FOR_EACH (mcast_local_cfg, ctx->idl) { struct vtep_ctl_mcast_mac *mcast_mac; struct vtep_ctl_lswitch *ls; if (!mcast_local_cfg->logical_switch) { continue; } ls = find_lswitch(ctx, mcast_local_cfg->logical_switch->name, false); if (!ls) { continue; } mcast_mac = add_mcast_mac_to_cache(ctx, ls, mcast_local_cfg->MAC, mcast_local_cfg->locator_set, true); mcast_mac->local_cfg = mcast_local_cfg; } VTEPREC_MCAST_MACS_REMOTE_FOR_EACH (mcast_remote_cfg, ctx->idl) { struct vtep_ctl_mcast_mac *mcast_mac; struct vtep_ctl_lswitch *ls; if (!mcast_remote_cfg->logical_switch) { continue; } ls = find_lswitch(ctx, mcast_remote_cfg->logical_switch->name, false); if (!ls) { continue; } mcast_mac = add_mcast_mac_to_cache(ctx, ls, mcast_remote_cfg->MAC, mcast_remote_cfg->locator_set, false); mcast_mac->remote_cfg = mcast_remote_cfg; } VTEPREC_TUNNEL_FOR_EACH (tunnel_cfg, ctx->idl) { if (tunnel_cfg->local) { add_ploc_to_cache(ctx, tunnel_cfg->local); } if (tunnel_cfg->remote) { add_ploc_to_cache(ctx, tunnel_cfg->remote); } } sset_init(&pswitches); for (i = 0; i < vtep_global->n_switches; i++) { struct vteprec_physical_switch *ps_cfg = vtep_global->switches[i]; struct vtep_ctl_pswitch *ps; size_t j; if (!sset_add(&pswitches, ps_cfg->name)) { continue; } ps = shash_find_data(&ctx->pswitches, ps_cfg->name); for (j = 0; j < ps_cfg->n_ports; j++) { struct vteprec_physical_port *port_cfg = ps_cfg->ports[j]; struct vtep_ctl_port *port; size_t k; port = shash_find_data(&ctx->ports, port_cfg->name); if (port) { if (port_cfg == port->port_cfg) { VLOG_WARN("%s: port is in multiple physical switches " "(%s and %s)", port_cfg->name, ps->name, port->ps->name); } else { /* Log as an error because this violates the database's * uniqueness constraints, so the database server shouldn't * have allowed it. */ VLOG_ERR("%s: database contains duplicate port name", port_cfg->name); } continue; } port = add_port_to_cache(ctx, ps, port_cfg); for (k = 0; k < port_cfg->n_vlan_bindings; k++) { struct vteprec_logical_switch *ls_cfg; struct vtep_ctl_lswitch *ls; char *vlan; vlan = xasprintf("%"PRId64, port_cfg->key_vlan_bindings[k]); if (shash_find(&port->bindings, vlan)) { vtep_ctl_fatal("multiple bindings for vlan %s", vlan); } ls_cfg = port_cfg->value_vlan_bindings[k]; ls = find_lswitch(ctx, ls_cfg->name, true); shash_add_nocopy(&port->bindings, vlan, ls); } } } sset_destroy(&pswitches); } static struct vtep_ctl_pswitch * find_pswitch(struct vtep_ctl_context *ctx, const char *name, bool must_exist) { struct vtep_ctl_pswitch *ps; ovs_assert(ctx->cache_valid); ps = shash_find_data(&ctx->pswitches, name); if (must_exist && !ps) { vtep_ctl_fatal("no physical switch named %s", name); } vteprec_global_verify_switches(ctx->vtep_global); return ps; } static struct vtep_ctl_port * find_port(struct vtep_ctl_context *ctx, const char *ps_name, const char *port_name, bool must_exist) { char *cache_name = xasprintf("%s+%s", ps_name, port_name); struct vtep_ctl_port *port; ovs_assert(ctx->cache_valid); port = shash_find_data(&ctx->ports, cache_name); if (port && !strcmp(port_name, port->ps->name)) { port = NULL; } free(cache_name); if (must_exist && !port) { vtep_ctl_fatal("no port named %s", port_name); } verify_ports(ctx); return port; } static void pswitch_insert_port(const struct vteprec_physical_switch *ps, struct vteprec_physical_port *port) { struct vteprec_physical_port **ports; size_t i; ports = xmalloc(sizeof *ps->ports * (ps->n_ports + 1)); for (i = 0; i < ps->n_ports; i++) { ports[i] = ps->ports[i]; } ports[ps->n_ports] = port; vteprec_physical_switch_set_ports(ps, ports, ps->n_ports + 1); free(ports); } static void pswitch_delete_port(const struct vteprec_physical_switch *ps, const struct vteprec_physical_port *port) { struct vteprec_physical_port **ports; size_t i, n; ports = xmalloc(sizeof *ps->ports * ps->n_ports); for (i = n = 0; i < ps->n_ports; i++) { if (ps->ports[i] != port) { ports[n++] = ps->ports[i]; } } vteprec_physical_switch_set_ports(ps, ports, n); free(ports); } static void vtep_insert_pswitch(const struct vteprec_global *vtep_global, struct vteprec_physical_switch *ps) { struct vteprec_physical_switch **pswitches; size_t i; pswitches = xmalloc(sizeof *vtep_global->switches * (vtep_global->n_switches + 1)); for (i = 0; i < vtep_global->n_switches; i++) { pswitches[i] = vtep_global->switches[i]; } pswitches[vtep_global->n_switches] = ps; vteprec_global_set_switches(vtep_global, pswitches, vtep_global->n_switches + 1); free(pswitches); } static void cmd_add_ps(struct vtep_ctl_context *ctx) { const char *ps_name = ctx->argv[1]; bool may_exist = shash_find(&ctx->options, "--may-exist") != NULL; struct vteprec_physical_switch *ps; vtep_ctl_context_populate_cache(ctx); if (find_pswitch(ctx, ps_name, false)) { if (!may_exist) { vtep_ctl_fatal("cannot create physical switch %s because it " "already exists", ps_name); } return; } ps = vteprec_physical_switch_insert(ctx->txn); vteprec_physical_switch_set_name(ps, ps_name); vtep_insert_pswitch(ctx->vtep_global, ps); vtep_ctl_context_invalidate_cache(ctx); } static void del_port(struct vtep_ctl_context *ctx, struct vtep_ctl_port *port) { pswitch_delete_port(port->ps->ps_cfg, port->port_cfg); del_cached_port(ctx, port); } static void del_pswitch(struct vtep_ctl_context *ctx, struct vtep_ctl_pswitch *ps) { struct vtep_ctl_port *port, *next_port; LIST_FOR_EACH_SAFE (port, next_port, ports_node, &ps->ports) { del_port(ctx, port); } del_cached_pswitch(ctx, ps); } static void cmd_del_ps(struct vtep_ctl_context *ctx) { bool must_exist = !shash_find(&ctx->options, "--if-exists"); struct vtep_ctl_pswitch *ps; vtep_ctl_context_populate_cache(ctx); ps = find_pswitch(ctx, ctx->argv[1], must_exist); if (ps) { del_pswitch(ctx, ps); } } static void output_sorted(struct svec *svec, struct ds *output) { const char *name; size_t i; svec_sort(svec); SVEC_FOR_EACH (i, name, svec) { ds_put_format(output, "%s\n", name); } } static void cmd_list_ps(struct vtep_ctl_context *ctx) { struct shash_node *node; struct svec pswitches; vtep_ctl_context_populate_cache(ctx); svec_init(&pswitches); SHASH_FOR_EACH (node, &ctx->pswitches) { struct vtep_ctl_pswitch *ps = node->data; svec_add(&pswitches, ps->name); } output_sorted(&pswitches, &ctx->output); svec_destroy(&pswitches); } static void cmd_ps_exists(struct vtep_ctl_context *ctx) { vtep_ctl_context_populate_cache(ctx); if (!find_pswitch(ctx, ctx->argv[1], false)) { vtep_ctl_exit(2); } } static void cmd_list_ports(struct vtep_ctl_context *ctx) { struct vtep_ctl_pswitch *ps; struct vtep_ctl_port *port; struct svec ports; vtep_ctl_context_populate_cache(ctx); ps = find_pswitch(ctx, ctx->argv[1], true); vteprec_physical_switch_verify_ports(ps->ps_cfg); svec_init(&ports); LIST_FOR_EACH (port, ports_node, &ps->ports) { if (strcmp(port->port_cfg->name, ps->name)) { svec_add(&ports, port->port_cfg->name); } } output_sorted(&ports, &ctx->output); svec_destroy(&ports); } static void add_port(struct vtep_ctl_context *ctx, const char *ps_name, const char *port_name, bool may_exist) { struct vtep_ctl_port *vtep_ctl_port; struct vtep_ctl_pswitch *ps; struct vteprec_physical_port *port; vtep_ctl_context_populate_cache(ctx); vtep_ctl_port = find_port(ctx, ps_name, port_name, false); if (vtep_ctl_port) { if (!may_exist) { vtep_ctl_fatal("cannot create a port named %s on %s because a " "port with that name already exists", port_name, ps_name); } return; } ps = find_pswitch(ctx, ps_name, true); port = vteprec_physical_port_insert(ctx->txn); vteprec_physical_port_set_name(port, port_name); pswitch_insert_port(ps->ps_cfg, port); add_port_to_cache(ctx, ps, port); } static void cmd_add_port(struct vtep_ctl_context *ctx) { bool may_exist = shash_find(&ctx->options, "--may-exist") != NULL; add_port(ctx, ctx->argv[1], ctx->argv[2], may_exist); } static void cmd_del_port(struct vtep_ctl_context *ctx) { bool must_exist = !shash_find(&ctx->options, "--if-exists"); struct vtep_ctl_port *port; vtep_ctl_context_populate_cache(ctx); port = find_port(ctx, ctx->argv[1], ctx->argv[2], must_exist); if (port) { if (ctx->argc == 3) { struct vtep_ctl_pswitch *ps; ps = find_pswitch(ctx, ctx->argv[1], true); if (port->ps != ps) { vtep_ctl_fatal("physical switch %s does not have a port %s", ctx->argv[1], ctx->argv[2]); } } del_port(ctx, port); } } static struct vtep_ctl_lswitch * find_lswitch(struct vtep_ctl_context *ctx, const char *name, bool must_exist) { struct vtep_ctl_lswitch *ls; ovs_assert(ctx->cache_valid); ls = shash_find_data(&ctx->lswitches, name); if (must_exist && !ls) { vtep_ctl_fatal("no logical switch named %s", name); } return ls; } static void cmd_add_ls(struct vtep_ctl_context *ctx) { const char *ls_name = ctx->argv[1]; bool may_exist = shash_find(&ctx->options, "--may-exist") != NULL; struct vteprec_logical_switch *ls; vtep_ctl_context_populate_cache(ctx); if (find_lswitch(ctx, ls_name, false)) { if (!may_exist) { vtep_ctl_fatal("cannot create logical switch %s because it " "already exists", ls_name); } return; } ls = vteprec_logical_switch_insert(ctx->txn); vteprec_logical_switch_set_name(ls, ls_name); vtep_ctl_context_invalidate_cache(ctx); } static void del_lswitch(struct vtep_ctl_context *ctx, struct vtep_ctl_lswitch *ls) { del_cached_lswitch(ctx, ls); } static void cmd_del_ls(struct vtep_ctl_context *ctx) { bool must_exist = !shash_find(&ctx->options, "--if-exists"); struct vtep_ctl_lswitch *ls; vtep_ctl_context_populate_cache(ctx); ls = find_lswitch(ctx, ctx->argv[1], must_exist); if (ls) { del_lswitch(ctx, ls); } } static void cmd_list_ls(struct vtep_ctl_context *ctx) { struct shash_node *node; struct svec lswitches; vtep_ctl_context_populate_cache(ctx); svec_init(&lswitches); SHASH_FOR_EACH (node, &ctx->lswitches) { struct vtep_ctl_lswitch *ls = node->data; svec_add(&lswitches, ls->name); } output_sorted(&lswitches, &ctx->output); svec_destroy(&lswitches); } static void cmd_ls_exists(struct vtep_ctl_context *ctx) { vtep_ctl_context_populate_cache(ctx); if (!find_lswitch(ctx, ctx->argv[1], false)) { vtep_ctl_exit(2); } } static void cmd_list_bindings(struct vtep_ctl_context *ctx) { const struct shash_node *node; struct vtep_ctl_port *port; struct svec bindings; vtep_ctl_context_populate_cache(ctx); port = find_port(ctx, ctx->argv[1], ctx->argv[2], true); svec_init(&bindings); SHASH_FOR_EACH (node, &port->bindings) { struct vtep_ctl_lswitch *lswitch = node->data; char *binding; binding = xasprintf("%04lld %s", strtoll(node->name, NULL, 0), lswitch->name); svec_add_nocopy(&bindings, binding); } output_sorted(&bindings, &ctx->output); svec_destroy(&bindings); } static void cmd_bind_ls(struct vtep_ctl_context *ctx) { struct vtep_ctl_lswitch *ls; struct vtep_ctl_port *port; const char *vlan; vtep_ctl_context_populate_cache(ctx); port = find_port(ctx, ctx->argv[1], ctx->argv[2], true); vlan = ctx->argv[3]; ls = find_lswitch(ctx, ctx->argv[4], true); add_ls_binding_to_cache(port, vlan, ls); commit_ls_bindings(port); vtep_ctl_context_invalidate_cache(ctx); } static void cmd_unbind_ls(struct vtep_ctl_context *ctx) { struct vtep_ctl_port *port; const char *vlan; vtep_ctl_context_populate_cache(ctx); port = find_port(ctx, ctx->argv[1], ctx->argv[2], true); vlan = ctx->argv[3]; del_cached_ls_binding(port, vlan); commit_ls_bindings(port); vtep_ctl_context_invalidate_cache(ctx); } static void add_ucast_entry(struct vtep_ctl_context *ctx, bool local) { struct vtep_ctl_lswitch *ls; const char *mac; const char *encap; const char *dst_ip; struct vteprec_physical_locator *ploc_cfg; vtep_ctl_context_populate_cache(ctx); ls = find_lswitch(ctx, ctx->argv[1], true); mac = ctx->argv[2]; if (ctx->argc == 4) { encap = "vxlan_over_ipv4"; dst_ip = ctx->argv[3]; } else { encap = ctx->argv[3]; dst_ip = ctx->argv[4]; } ploc_cfg = find_ploc(ctx, encap, dst_ip); if (!ploc_cfg) { ploc_cfg = vteprec_physical_locator_insert(ctx->txn); vteprec_physical_locator_set_dst_ip(ploc_cfg, dst_ip); vteprec_physical_locator_set_encapsulation_type(ploc_cfg, encap); add_ploc_to_cache(ctx, ploc_cfg); } if (local) { struct vteprec_ucast_macs_local *ucast_cfg; ucast_cfg = shash_find_data(&ls->ucast_local, mac); if (!ucast_cfg) { ucast_cfg = vteprec_ucast_macs_local_insert(ctx->txn); vteprec_ucast_macs_local_set_MAC(ucast_cfg, mac); vteprec_ucast_macs_local_set_logical_switch(ucast_cfg, ls->ls_cfg); shash_add(&ls->ucast_local, mac, ucast_cfg); } vteprec_ucast_macs_local_set_locator(ucast_cfg, ploc_cfg); } else { struct vteprec_ucast_macs_remote *ucast_cfg; ucast_cfg = shash_find_data(&ls->ucast_remote, mac); if (!ucast_cfg) { ucast_cfg = vteprec_ucast_macs_remote_insert(ctx->txn); vteprec_ucast_macs_remote_set_MAC(ucast_cfg, mac); vteprec_ucast_macs_remote_set_logical_switch(ucast_cfg, ls->ls_cfg); shash_add(&ls->ucast_remote, mac, ucast_cfg); } vteprec_ucast_macs_remote_set_locator(ucast_cfg, ploc_cfg); } vtep_ctl_context_invalidate_cache(ctx); } static void cmd_add_ucast_local(struct vtep_ctl_context *ctx) { add_ucast_entry(ctx, true); } static void cmd_add_ucast_remote(struct vtep_ctl_context *ctx) { add_ucast_entry(ctx, false); } static void del_ucast_entry(struct vtep_ctl_context *ctx, bool local) { struct vtep_ctl_lswitch *ls; struct shash *ucast_shash; struct shash_node *node; vtep_ctl_context_populate_cache(ctx); ls = find_lswitch(ctx, ctx->argv[1], true); ucast_shash = local ? &ls->ucast_local : &ls->ucast_remote; node = shash_find(ucast_shash, ctx->argv[2]); if (!node) { return; } if (local) { struct vteprec_ucast_macs_local *ucast_cfg = node->data; vteprec_ucast_macs_local_delete(ucast_cfg); } else { struct vteprec_ucast_macs_remote *ucast_cfg = node->data; vteprec_ucast_macs_remote_delete(ucast_cfg); } shash_delete(ucast_shash, node); vtep_ctl_context_invalidate_cache(ctx); } static void cmd_del_ucast_local(struct vtep_ctl_context *ctx) { del_ucast_entry(ctx, true); } static void cmd_del_ucast_remote(struct vtep_ctl_context *ctx) { del_ucast_entry(ctx, false); } static void commit_mcast_entries(struct vtep_ctl_mcast_mac *mcast_mac) { struct vtep_ctl_ploc *ploc; struct vteprec_physical_locator **locators = NULL; size_t n_locators; int i; n_locators = list_size(&mcast_mac->locators); ovs_assert(n_locators); locators = xmalloc(n_locators * sizeof *locators); i = 0; LIST_FOR_EACH (ploc, locators_node, &mcast_mac->locators) { locators[i] = (struct vteprec_physical_locator *)ploc->ploc_cfg; i++; } vteprec_physical_locator_set_set_locators(mcast_mac->ploc_set_cfg, locators, n_locators); free(locators); } static void add_mcast_entry(struct vtep_ctl_context *ctx, struct vtep_ctl_lswitch *ls, const char *mac, const char *encap, const char *dst_ip, bool local) { struct shash *mcast_shash; struct vtep_ctl_mcast_mac *mcast_mac; struct vteprec_physical_locator *ploc_cfg; struct vteprec_physical_locator_set *ploc_set_cfg; mcast_shash = local ? &ls->mcast_local : &ls->mcast_remote; /* Physical locator sets are immutable, so allocate a new one. */ ploc_set_cfg = vteprec_physical_locator_set_insert(ctx->txn); mcast_mac = shash_find_data(mcast_shash, mac); if (!mcast_mac) { mcast_mac = add_mcast_mac_to_cache(ctx, ls, mac, ploc_set_cfg, local); if (local) { mcast_mac->local_cfg = vteprec_mcast_macs_local_insert(ctx->txn); vteprec_mcast_macs_local_set_MAC(mcast_mac->local_cfg, mac); vteprec_mcast_macs_local_set_locator_set(mcast_mac->local_cfg, ploc_set_cfg); vteprec_mcast_macs_local_set_logical_switch(mcast_mac->local_cfg, ls->ls_cfg); mcast_mac->remote_cfg = NULL; } else { mcast_mac->remote_cfg = vteprec_mcast_macs_remote_insert(ctx->txn); vteprec_mcast_macs_remote_set_MAC(mcast_mac->remote_cfg, mac); vteprec_mcast_macs_remote_set_locator_set(mcast_mac->remote_cfg, ploc_set_cfg); vteprec_mcast_macs_remote_set_logical_switch(mcast_mac->remote_cfg, ls->ls_cfg); mcast_mac->local_cfg = NULL; } } else { mcast_mac->ploc_set_cfg = ploc_set_cfg; if (local) { vteprec_mcast_macs_local_set_locator_set(mcast_mac->local_cfg, ploc_set_cfg); } else { vteprec_mcast_macs_remote_set_locator_set(mcast_mac->remote_cfg, ploc_set_cfg); } } ploc_cfg = find_ploc(ctx, encap, dst_ip); if (!ploc_cfg) { ploc_cfg = vteprec_physical_locator_insert(ctx->txn); vteprec_physical_locator_set_dst_ip(ploc_cfg, dst_ip); vteprec_physical_locator_set_encapsulation_type(ploc_cfg, encap); add_ploc_to_cache(ctx, ploc_cfg); } add_ploc_to_mcast_mac(mcast_mac, ploc_cfg); commit_mcast_entries(mcast_mac); } static void del_mcast_entry(struct vtep_ctl_context *ctx, struct vtep_ctl_lswitch *ls, const char *mac, const char *encap, const char *dst_ip, bool local) { struct vtep_ctl_mcast_mac *mcast_mac; struct shash *mcast_shash; struct vteprec_physical_locator *ploc_cfg; struct vteprec_physical_locator_set *ploc_set_cfg; mcast_shash = local ? &ls->mcast_local : &ls->mcast_remote; mcast_mac = shash_find_data(mcast_shash, mac); if (!mcast_mac) { return; } ploc_cfg = find_ploc(ctx, encap, dst_ip); if (!ploc_cfg) { /* Couldn't find the physical locator, so just ignore. */ return; } /* Physical locator sets are immutable, so allocate a new one. */ ploc_set_cfg = vteprec_physical_locator_set_insert(ctx->txn); mcast_mac->ploc_set_cfg = ploc_set_cfg; del_ploc_from_mcast_mac(mcast_mac, ploc_cfg); if (list_is_empty(&mcast_mac->locators)) { struct shash_node *node = shash_find(mcast_shash, mac); vteprec_physical_locator_set_delete(ploc_set_cfg); if (local) { vteprec_mcast_macs_local_delete(mcast_mac->local_cfg); } else { vteprec_mcast_macs_remote_delete(mcast_mac->remote_cfg); } free(node->data); shash_delete(mcast_shash, node); } else { if (local) { vteprec_mcast_macs_local_set_locator_set(mcast_mac->local_cfg, ploc_set_cfg); } else { vteprec_mcast_macs_remote_set_locator_set(mcast_mac->remote_cfg, ploc_set_cfg); } commit_mcast_entries(mcast_mac); } } static void add_del_mcast_entry(struct vtep_ctl_context *ctx, bool add, bool local) { struct vtep_ctl_lswitch *ls; const char *mac; const char *encap; const char *dst_ip; vtep_ctl_context_populate_cache(ctx); ls = find_lswitch(ctx, ctx->argv[1], true); mac = ctx->argv[2]; if (ctx->argc == 4) { encap = "vxlan_over_ipv4"; dst_ip = ctx->argv[3]; } else { encap = ctx->argv[3]; dst_ip = ctx->argv[4]; } if (add) { add_mcast_entry(ctx, ls, mac, encap, dst_ip, local); } else { del_mcast_entry(ctx, ls, mac, encap, dst_ip, local); } vtep_ctl_context_invalidate_cache(ctx); } static void cmd_add_mcast_local(struct vtep_ctl_context *ctx) { add_del_mcast_entry(ctx, true, true); } static void cmd_add_mcast_remote(struct vtep_ctl_context *ctx) { add_del_mcast_entry(ctx, true, false); } static void cmd_del_mcast_local(struct vtep_ctl_context *ctx) { add_del_mcast_entry(ctx, false, true); } static void cmd_del_mcast_remote(struct vtep_ctl_context *ctx) { add_del_mcast_entry(ctx, false, false); } static void clear_macs(struct vtep_ctl_context *ctx, bool local) { struct vtep_ctl_lswitch *ls; const struct shash_node *node; struct shash *ucast_shash; struct shash *mcast_shash; vtep_ctl_context_populate_cache(ctx); ls = find_lswitch(ctx, ctx->argv[1], true); ucast_shash = local ? &ls->ucast_local : &ls->ucast_remote; mcast_shash = local ? &ls->mcast_local : &ls->mcast_remote; SHASH_FOR_EACH (node, ucast_shash) { if (local) { struct vteprec_ucast_macs_local *ucast_cfg = node->data; vteprec_ucast_macs_local_delete(ucast_cfg); } else { struct vteprec_ucast_macs_remote *ucast_cfg = node->data; vteprec_ucast_macs_remote_delete(ucast_cfg); } } SHASH_FOR_EACH (node, mcast_shash) { struct vtep_ctl_mcast_mac *mcast_mac = node->data; if (local) { vteprec_mcast_macs_local_delete(mcast_mac->local_cfg); } else { vteprec_mcast_macs_remote_delete(mcast_mac->remote_cfg); } } vtep_ctl_context_invalidate_cache(ctx); } static void cmd_clear_local_macs(struct vtep_ctl_context *ctx) { clear_macs(ctx, true); } static void cmd_clear_remote_macs(struct vtep_ctl_context *ctx) { clear_macs(ctx, false); } static void list_macs(struct vtep_ctl_context *ctx, bool local) { struct vtep_ctl_lswitch *ls; const struct shash_node *node; struct shash *ucast_shash; struct svec ucast_macs; struct shash *mcast_shash; struct svec mcast_macs; vtep_ctl_context_populate_cache(ctx); ls = find_lswitch(ctx, ctx->argv[1], true); ucast_shash = local ? &ls->ucast_local : &ls->ucast_remote; mcast_shash = local ? &ls->mcast_local : &ls->mcast_remote; svec_init(&ucast_macs); SHASH_FOR_EACH (node, ucast_shash) { struct vteprec_ucast_macs_local *ucast_local = node->data; struct vteprec_ucast_macs_remote *ucast_remote = node->data; struct vteprec_physical_locator *ploc_cfg; char *entry; ploc_cfg = local ? ucast_local->locator : ucast_remote->locator; entry = xasprintf(" %s -> %s/%s", node->name, ploc_cfg->encapsulation_type, ploc_cfg->dst_ip); svec_add_nocopy(&ucast_macs, entry); } ds_put_format(&ctx->output, "ucast-mac-%s\n", local ? "local" : "remote"); output_sorted(&ucast_macs, &ctx->output); ds_put_char(&ctx->output, '\n'); svec_destroy(&ucast_macs); svec_init(&mcast_macs); SHASH_FOR_EACH (node, mcast_shash) { struct vtep_ctl_mcast_mac *mcast_mac = node->data; struct vtep_ctl_ploc *ploc; char *entry; LIST_FOR_EACH (ploc, locators_node, &mcast_mac->locators) { entry = xasprintf(" %s -> %s/%s", node->name, ploc->ploc_cfg->encapsulation_type, ploc->ploc_cfg->dst_ip); svec_add_nocopy(&mcast_macs, entry); } } ds_put_format(&ctx->output, "mcast-mac-%s\n", local ? "local" : "remote"); output_sorted(&mcast_macs, &ctx->output); ds_put_char(&ctx->output, '\n'); svec_destroy(&mcast_macs); } static void cmd_list_local_macs(struct vtep_ctl_context *ctx) { list_macs(ctx, true); } static void cmd_list_remote_macs(struct vtep_ctl_context *ctx) { list_macs(ctx, false); } static void verify_managers(const struct vteprec_global *vtep_global) { size_t i; vteprec_global_verify_managers(vtep_global); for (i = 0; i < vtep_global->n_managers; ++i) { const struct vteprec_manager *mgr = vtep_global->managers[i]; vteprec_manager_verify_target(mgr); } } static void pre_manager(struct vtep_ctl_context *ctx) { ovsdb_idl_add_column(ctx->idl, &vteprec_global_col_managers); ovsdb_idl_add_column(ctx->idl, &vteprec_manager_col_target); } static void cmd_get_manager(struct vtep_ctl_context *ctx) { const struct vteprec_global *vtep_global = ctx->vtep_global; struct svec targets; size_t i; verify_managers(vtep_global); /* Print the targets in sorted order for reproducibility. */ svec_init(&targets); for (i = 0; i < vtep_global->n_managers; i++) { svec_add(&targets, vtep_global->managers[i]->target); } svec_sort_unique(&targets); for (i = 0; i < targets.n; i++) { ds_put_format(&ctx->output, "%s\n", targets.names[i]); } svec_destroy(&targets); } static void delete_managers(const struct vtep_ctl_context *ctx) { const struct vteprec_global *vtep_global = ctx->vtep_global; size_t i; /* Delete Manager rows pointed to by 'managers' column. */ for (i = 0; i < vtep_global->n_managers; i++) { vteprec_manager_delete(vtep_global->managers[i]); } /* Delete 'Manager' row refs in 'managers' column. */ vteprec_global_set_managers(vtep_global, NULL, 0); } static void cmd_del_manager(struct vtep_ctl_context *ctx) { const struct vteprec_global *vtep_global = ctx->vtep_global; verify_managers(vtep_global); delete_managers(ctx); } static void insert_managers(struct vtep_ctl_context *ctx, char *targets[], size_t n) { struct vteprec_manager **managers; size_t i; /* Insert each manager in a new row in Manager table. */ managers = xmalloc(n * sizeof *managers); for (i = 0; i < n; i++) { if (stream_verify_name(targets[i]) && pstream_verify_name(targets[i])) { VLOG_WARN("target type \"%s\" is possibly erroneous", targets[i]); } managers[i] = vteprec_manager_insert(ctx->txn); vteprec_manager_set_target(managers[i], targets[i]); } /* Store uuids of new Manager rows in 'managers' column. */ vteprec_global_set_managers(ctx->vtep_global, managers, n); free(managers); } static void cmd_set_manager(struct vtep_ctl_context *ctx) { const size_t n = ctx->argc - 1; verify_managers(ctx->vtep_global); delete_managers(ctx); insert_managers(ctx, &ctx->argv[1], n); } /* Parameter commands. */ struct vtep_ctl_row_id { const struct ovsdb_idl_table_class *table; const struct ovsdb_idl_column *name_column; const struct ovsdb_idl_column *uuid_column; }; struct vtep_ctl_table_class { struct ovsdb_idl_table_class *class; struct vtep_ctl_row_id row_ids[2]; }; static const struct vtep_ctl_table_class tables[] = { {&vteprec_table_global, {{&vteprec_table_global, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_logical_binding_stats, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_logical_switch, {{&vteprec_table_logical_switch, &vteprec_logical_switch_col_name, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_ucast_macs_local, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_ucast_macs_remote, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_mcast_macs_local, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_mcast_macs_remote, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_manager, {{&vteprec_table_manager, &vteprec_manager_col_target, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_physical_locator, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_physical_locator_set, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_physical_port, {{&vteprec_table_physical_port, &vteprec_physical_port_col_name, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_physical_switch, {{&vteprec_table_physical_switch, &vteprec_physical_switch_col_name, NULL}, {NULL, NULL, NULL}}}, {&vteprec_table_tunnel, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}}, {NULL, {{NULL, NULL, NULL}, {NULL, NULL, NULL}}} }; static void die_if_error(char *error) { if (error) { vtep_ctl_fatal("%s", error); } } static int to_lower_and_underscores(unsigned c) { return c == '-' ? '_' : tolower(c); } static unsigned int score_partial_match(const char *name, const char *s) { int score; if (!strcmp(name, s)) { return UINT_MAX; } for (score = 0; ; score++, name++, s++) { if (to_lower_and_underscores(*name) != to_lower_and_underscores(*s)) { break; } else if (*name == '\0') { return UINT_MAX - 1; } } return *s == '\0' ? score : 0; } static const struct vtep_ctl_table_class * get_table(const char *table_name) { const struct vtep_ctl_table_class *table; const struct vtep_ctl_table_class *best_match = NULL; unsigned int best_score = 0; for (table = tables; table->class; table++) { unsigned int score = score_partial_match(table->class->name, table_name); if (score > best_score) { best_match = table; best_score = score; } else if (score == best_score) { best_match = NULL; } } if (best_match) { return best_match; } else if (best_score) { vtep_ctl_fatal("multiple table names match \"%s\"", table_name); } else { vtep_ctl_fatal("unknown table \"%s\"", table_name); } } static const struct vtep_ctl_table_class * pre_get_table(struct vtep_ctl_context *ctx, const char *table_name) { const struct vtep_ctl_table_class *table_class; int i; table_class = get_table(table_name); ovsdb_idl_add_table(ctx->idl, table_class->class); for (i = 0; i < ARRAY_SIZE(table_class->row_ids); i++) { const struct vtep_ctl_row_id *id = &table_class->row_ids[i]; if (id->table) { ovsdb_idl_add_table(ctx->idl, id->table); } if (id->name_column) { ovsdb_idl_add_column(ctx->idl, id->name_column); } if (id->uuid_column) { ovsdb_idl_add_column(ctx->idl, id->uuid_column); } } return table_class; } static const struct ovsdb_idl_row * get_row_by_id(struct vtep_ctl_context *ctx, const struct vtep_ctl_table_class *table, const struct vtep_ctl_row_id *id, const char *record_id) { const struct ovsdb_idl_row *referrer, *final; if (!id->table) { return NULL; } if (!id->name_column) { if (strcmp(record_id, ".")) { return NULL; } referrer = ovsdb_idl_first_row(ctx->idl, id->table); if (!referrer || ovsdb_idl_next_row(referrer)) { return NULL; } } else { const struct ovsdb_idl_row *row; referrer = NULL; for (row = ovsdb_idl_first_row(ctx->idl, id->table); row != NULL; row = ovsdb_idl_next_row(row)) { const struct ovsdb_datum *name; name = ovsdb_idl_get(row, id->name_column, OVSDB_TYPE_STRING, OVSDB_TYPE_VOID); if (name->n == 1 && !strcmp(name->keys[0].string, record_id)) { if (referrer) { vtep_ctl_fatal("multiple rows in %s match \"%s\"", table->class->name, record_id); } referrer = row; } } } if (!referrer) { return NULL; } final = NULL; if (id->uuid_column) { const struct ovsdb_datum *uuid; ovsdb_idl_txn_verify(referrer, id->uuid_column); uuid = ovsdb_idl_get(referrer, id->uuid_column, OVSDB_TYPE_UUID, OVSDB_TYPE_VOID); if (uuid->n == 1) { final = ovsdb_idl_get_row_for_uuid(ctx->idl, table->class, &uuid->keys[0].uuid); } } else { final = referrer; } return final; } static const struct ovsdb_idl_row * get_row (struct vtep_ctl_context *ctx, const struct vtep_ctl_table_class *table, const char *record_id) { const struct ovsdb_idl_row *row; struct uuid uuid; if (uuid_from_string(&uuid, record_id)) { row = ovsdb_idl_get_row_for_uuid(ctx->idl, table->class, &uuid); } else { int i; for (i = 0; i < ARRAY_SIZE(table->row_ids); i++) { row = get_row_by_id(ctx, table, &table->row_ids[i], record_id); if (row) { break; } } } return row; } static const struct ovsdb_idl_row * must_get_row(struct vtep_ctl_context *ctx, const struct vtep_ctl_table_class *table, const char *record_id) { const struct ovsdb_idl_row *row = get_row(ctx, table, record_id); if (!row) { vtep_ctl_fatal("no row \"%s\" in table %s", record_id, table->class->name); } return row; } static char * get_column(const struct vtep_ctl_table_class *table, const char *column_name, const struct ovsdb_idl_column **columnp) { const struct ovsdb_idl_column *best_match = NULL; unsigned int best_score = 0; size_t i; for (i = 0; i < table->class->n_columns; i++) { const struct ovsdb_idl_column *column = &table->class->columns[i]; unsigned int score = score_partial_match(column->name, column_name); if (score > best_score) { best_match = column; best_score = score; } else if (score == best_score) { best_match = NULL; } } *columnp = best_match; if (best_match) { return NULL; } else if (best_score) { return xasprintf("%s contains more than one column whose name " "matches \"%s\"", table->class->name, column_name); } else { return xasprintf("%s does not contain a column whose name matches " "\"%s\"", table->class->name, column_name); } } static struct ovsdb_symbol * create_symbol(struct ovsdb_symbol_table *symtab, const char *id, bool *newp) { struct ovsdb_symbol *symbol; if (id[0] != '@') { vtep_ctl_fatal("row id \"%s\" does not begin with \"@\"", id); } if (newp) { *newp = ovsdb_symbol_table_get(symtab, id) == NULL; } symbol = ovsdb_symbol_table_insert(symtab, id); if (symbol->created) { vtep_ctl_fatal("row id \"%s\" may only be specified on one --id option", id); } symbol->created = true; return symbol; } static void pre_get_column(struct vtep_ctl_context *ctx, const struct vtep_ctl_table_class *table, const char *column_name, const struct ovsdb_idl_column **columnp) { die_if_error(get_column(table, column_name, columnp)); ovsdb_idl_add_column(ctx->idl, *columnp); } static char * missing_operator_error(const char *arg, const char **allowed_operators, size_t n_allowed) { struct ds s; ds_init(&s); ds_put_format(&s, "%s: argument does not end in ", arg); ds_put_format(&s, "\"%s\"", allowed_operators[0]); if (n_allowed == 2) { ds_put_format(&s, " or \"%s\"", allowed_operators[1]); } else if (n_allowed > 2) { size_t i; for (i = 1; i < n_allowed - 1; i++) { ds_put_format(&s, ", \"%s\"", allowed_operators[i]); } ds_put_format(&s, ", or \"%s\"", allowed_operators[i]); } ds_put_format(&s, " followed by a value."); return ds_steal_cstr(&s); } /* Breaks 'arg' apart into a number of fields in the following order: * * - The name of a column in 'table', stored into '*columnp'. The column * name may be abbreviated. * * - Optionally ':' followed by a key string. The key is stored as a * malloc()'d string into '*keyp', or NULL if no key is present in * 'arg'. * * - If 'valuep' is nonnull, an operator followed by a value string. The * allowed operators are the 'n_allowed' string in 'allowed_operators', * or just "=" if 'n_allowed' is 0. If 'operatorp' is nonnull, then the * index of the operator within 'allowed_operators' is stored into * '*operatorp'. The value is stored as a malloc()'d string into * '*valuep', or NULL if no value is present in 'arg'. * * On success, returns NULL. On failure, returned a malloc()'d string error * message and stores NULL into all of the nonnull output arguments. */ static char * OVS_WARN_UNUSED_RESULT parse_column_key_value(const char *arg, const struct vtep_ctl_table_class *table, const struct ovsdb_idl_column **columnp, char **keyp, int *operatorp, const char **allowed_operators, size_t n_allowed, char **valuep) { const char *p = arg; char *column_name; char *error; ovs_assert(!(operatorp && !valuep)); *keyp = NULL; if (valuep) { *valuep = NULL; } /* Parse column name. */ error = ovsdb_token_parse(&p, &column_name); if (error) { goto error; } if (column_name[0] == '\0') { free(column_name); error = xasprintf("%s: missing column name", arg); goto error; } error = get_column(table, column_name, columnp); free(column_name); if (error) { goto error; } /* Parse key string. */ if (*p == ':') { p++; error = ovsdb_token_parse(&p, keyp); if (error) { goto error; } } /* Parse value string. */ if (valuep) { size_t best_len; size_t i; int best; if (!allowed_operators) { static const char *equals = "="; allowed_operators = = n_allowed = 1; } best = -1; best_len = 0; for (i = 0; i < n_allowed; i++) { const char *op = allowed_operators[i]; size_t op_len = strlen(op); if (op_len > best_len && !strncmp(op, p, op_len) && p[op_len]) { best_len = op_len; best = i; } } if (best < 0) { error = missing_operator_error(arg, allowed_operators, n_allowed); goto error; } if (operatorp) { *operatorp = best; } *valuep = xstrdup(p + best_len); } else { if (*p != '\0') { error = xasprintf("%s: trailing garbage \"%s\" in argument", arg, p); goto error; } } return NULL; error: *columnp = NULL; free(*keyp); *keyp = NULL; if (valuep) { free(*valuep); *valuep = NULL; if (operatorp) { *operatorp = -1; } } return error; } static const struct ovsdb_idl_column * pre_parse_column_key_value(struct vtep_ctl_context *ctx, const char *arg, const struct vtep_ctl_table_class *table) { const struct ovsdb_idl_column *column; const char *p; char *column_name; p = arg; die_if_error(ovsdb_token_parse(&p, &column_name)); if (column_name[0] == '\0') { vtep_ctl_fatal("%s: missing column name", arg); } pre_get_column(ctx, table, column_name, &column); free(column_name); return column; } static void check_mutable(const struct vtep_ctl_table_class *table, const struct ovsdb_idl_column *column) { if (!column->mutable) { vtep_ctl_fatal("cannot modify read-only column %s in table %s", column->name, table->class->name); } } static void pre_cmd_get(struct vtep_ctl_context *ctx) { const char *id = shash_find_data(&ctx->options, "--id"); const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; int i; /* Using "get" without --id or a column name could possibly make sense. * Maybe, for example, a vtep-ctl run wants to assert that a row exists. * But it is unlikely that an interactive user would want to do that, so * issue a warning if we're running on a terminal. */ if (!id && ctx->argc <= 3 && isatty(STDOUT_FILENO)) { VLOG_WARN("\"get\" command without row arguments or \"--id\" is " "possibly erroneous"); } table = pre_get_table(ctx, table_name); for (i = 3; i < ctx->argc; i++) { if (!strcasecmp(ctx->argv[i], "_uuid") || !strcasecmp(ctx->argv[i], "-uuid")) { continue; } pre_parse_column_key_value(ctx, ctx->argv[i], table); } } static void cmd_get(struct vtep_ctl_context *ctx) { const char *id = shash_find_data(&ctx->options, "--id"); bool if_exists = shash_find(&ctx->options, "--if-exists"); const char *table_name = ctx->argv[1]; const char *record_id = ctx->argv[2]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_row *row; struct ds *out = &ctx->output; int i; table = get_table(table_name); row = must_get_row(ctx, table, record_id); if (id) { struct ovsdb_symbol *symbol; bool new; symbol = create_symbol(ctx->symtab, id, &new); if (!new) { vtep_ctl_fatal("row id \"%s\" specified on \"get\" command was used " "before it was defined", id); } symbol->uuid = row->uuid; /* This symbol refers to a row that already exists, so disable warnings * about it being unreferenced. */ symbol->strong_ref = true; } for (i = 3; i < ctx->argc; i++) { const struct ovsdb_idl_column *column; const struct ovsdb_datum *datum; char *key_string; /* Special case for obtaining the UUID of a row. We can't just do this * through parse_column_key_value() below since it returns a "struct * ovsdb_idl_column" and the UUID column doesn't have one. */ if (!strcasecmp(ctx->argv[i], "_uuid") || !strcasecmp(ctx->argv[i], "-uuid")) { ds_put_format(out, UUID_FMT"\n", UUID_ARGS(&row->uuid)); continue; } die_if_error(parse_column_key_value(ctx->argv[i], table, &column, &key_string, NULL, NULL, 0, NULL)); ovsdb_idl_txn_verify(row, column); datum = ovsdb_idl_read(row, column); if (key_string) { union ovsdb_atom key; unsigned int idx; if (column->type.value.type == OVSDB_TYPE_VOID) { vtep_ctl_fatal("cannot specify key to get for non-map column %s", column->name); } die_if_error(ovsdb_atom_from_string(&key, &column->type.key, key_string, ctx->symtab)); idx = ovsdb_datum_find_key(datum, &key, column->type.key.type); if (idx == UINT_MAX) { if (!if_exists) { vtep_ctl_fatal("no key \"%s\" in %s record \"%s\" column %s", key_string, table->class->name, record_id, column->name); } } else { ovsdb_atom_to_string(&datum->values[idx], column->type.value.type, out); } ovsdb_atom_destroy(&key, column->type.key.type); } else { ovsdb_datum_to_string(datum, &column->type, out); } ds_put_char(out, '\n'); free(key_string); } } static void parse_column_names(const char *column_names, const struct vtep_ctl_table_class *table, const struct ovsdb_idl_column ***columnsp, size_t *n_columnsp) { const struct ovsdb_idl_column **columns; size_t n_columns; if (!column_names) { size_t i; n_columns = table->class->n_columns + 1; columns = xmalloc(n_columns * sizeof *columns); columns[0] = NULL; for (i = 0; i < table->class->n_columns; i++) { columns[i + 1] = &table->class->columns[i]; } } else { char *s = xstrdup(column_names); size_t allocated_columns; char *save_ptr = NULL; char *column_name; columns = NULL; allocated_columns = n_columns = 0; for (column_name = strtok_r(s, ", ", &save_ptr); column_name; column_name = strtok_r(NULL, ", ", &save_ptr)) { const struct ovsdb_idl_column *column; if (!strcasecmp(column_name, "_uuid")) { column = NULL; } else { die_if_error(get_column(table, column_name, &column)); } if (n_columns >= allocated_columns) { columns = x2nrealloc(columns, &allocated_columns, sizeof *columns); } columns[n_columns++] = column; } free(s); if (!n_columns) { vtep_ctl_fatal("must specify at least one column name"); } } *columnsp = columns; *n_columnsp = n_columns; } static void pre_list_columns(struct vtep_ctl_context *ctx, const struct vtep_ctl_table_class *table, const char *column_names) { const struct ovsdb_idl_column **columns; size_t n_columns; size_t i; parse_column_names(column_names, table, &columns, &n_columns); for (i = 0; i < n_columns; i++) { if (columns[i]) { ovsdb_idl_add_column(ctx->idl, columns[i]); } } free(columns); } static void pre_cmd_list(struct vtep_ctl_context *ctx) { const char *column_names = shash_find_data(&ctx->options, "--columns"); const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; table = pre_get_table(ctx, table_name); pre_list_columns(ctx, table, column_names); } static struct table * list_make_table(const struct ovsdb_idl_column **columns, size_t n_columns) { struct table *out; size_t i; out = xmalloc(sizeof *out); table_init(out); for (i = 0; i < n_columns; i++) { const struct ovsdb_idl_column *column = columns[i]; const char *column_name = column ? column->name : "_uuid"; table_add_column(out, "%s", column_name); } return out; } static void list_record(const struct ovsdb_idl_row *row, const struct ovsdb_idl_column **columns, size_t n_columns, struct table *out) { size_t i; table_add_row(out); for (i = 0; i < n_columns; i++) { const struct ovsdb_idl_column *column = columns[i]; struct cell *cell = table_add_cell(out); if (!column) { struct ovsdb_datum datum; union ovsdb_atom atom; atom.uuid = row->uuid; datum.keys = &atom; datum.values = NULL; datum.n = 1; cell->json = ovsdb_datum_to_json(&datum, &ovsdb_type_uuid); cell->type = &ovsdb_type_uuid; } else { const struct ovsdb_datum *datum = ovsdb_idl_read(row, column); cell->json = ovsdb_datum_to_json(datum, &column->type); cell->type = &column->type; } } } static void cmd_list(struct vtep_ctl_context *ctx) { const char *column_names = shash_find_data(&ctx->options, "--columns"); const struct ovsdb_idl_column **columns; const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; struct table *out; size_t n_columns; int i; table = get_table(table_name); parse_column_names(column_names, table, &columns, &n_columns); out = ctx->table = list_make_table(columns, n_columns); if (ctx->argc > 2) { for (i = 2; i < ctx->argc; i++) { list_record(must_get_row(ctx, table, ctx->argv[i]), columns, n_columns, out); } } else { const struct ovsdb_idl_row *row; for (row = ovsdb_idl_first_row(ctx->idl, table->class); row != NULL; row = ovsdb_idl_next_row(row)) { list_record(row, columns, n_columns, out); } } free(columns); } static void pre_cmd_find(struct vtep_ctl_context *ctx) { const char *column_names = shash_find_data(&ctx->options, "--columns"); const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; int i; table = pre_get_table(ctx, table_name); pre_list_columns(ctx, table, column_names); for (i = 2; i < ctx->argc; i++) { pre_parse_column_key_value(ctx, ctx->argv[i], table); } } static void cmd_find(struct vtep_ctl_context *ctx) { const char *column_names = shash_find_data(&ctx->options, "--columns"); const struct ovsdb_idl_column **columns; const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_row *row; struct table *out; size_t n_columns; table = get_table(table_name); parse_column_names(column_names, table, &columns, &n_columns); out = ctx->table = list_make_table(columns, n_columns); for (row = ovsdb_idl_first_row(ctx->idl, table->class); row; row = ovsdb_idl_next_row(row)) { int i; for (i = 2; i < ctx->argc; i++) { if (!is_condition_satisfied(table, row, ctx->argv[i], ctx->symtab)) { goto next_row; } } list_record(row, columns, n_columns, out); next_row: ; } free(columns); } static void pre_cmd_set(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; int i; table = pre_get_table(ctx, table_name); for (i = 3; i < ctx->argc; i++) { const struct ovsdb_idl_column *column; column = pre_parse_column_key_value(ctx, ctx->argv[i], table); check_mutable(table, column); } } static void set_column(const struct vtep_ctl_table_class *table, const struct ovsdb_idl_row *row, const char *arg, struct ovsdb_symbol_table *symtab) { const struct ovsdb_idl_column *column; char *key_string, *value_string; char *error; error = parse_column_key_value(arg, table, &column, &key_string, NULL, NULL, 0, &value_string); die_if_error(error); if (!value_string) { vtep_ctl_fatal("%s: missing value", arg); } if (key_string) { union ovsdb_atom key, value; struct ovsdb_datum datum; if (column->type.value.type == OVSDB_TYPE_VOID) { vtep_ctl_fatal("cannot specify key to set for non-map column %s", column->name); } die_if_error(ovsdb_atom_from_string(&key, &column->type.key, key_string, symtab)); die_if_error(ovsdb_atom_from_string(&value, &column->type.value, value_string, symtab)); ovsdb_datum_init_empty(&datum); ovsdb_datum_add_unsafe(&datum, &key, &value, &column->type); ovsdb_atom_destroy(&key, column->type.key.type); ovsdb_atom_destroy(&value, column->type.value.type); ovsdb_datum_union(&datum, ovsdb_idl_read(row, column), &column->type, false); ovsdb_idl_txn_write(row, column, &datum); } else { struct ovsdb_datum datum; die_if_error(ovsdb_datum_from_string(&datum, &column->type, value_string, symtab)); ovsdb_idl_txn_write(row, column, &datum); } free(key_string); free(value_string); } static void cmd_set(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *record_id = ctx->argv[2]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_row *row; int i; table = get_table(table_name); row = must_get_row(ctx, table, record_id); for (i = 3; i < ctx->argc; i++) { set_column(table, row, ctx->argv[i], ctx->symtab); } } static void pre_cmd_add(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *column_name = ctx->argv[3]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_column *column; table = pre_get_table(ctx, table_name); pre_get_column(ctx, table, column_name, &column); check_mutable(table, column); } static void cmd_add(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *record_id = ctx->argv[2]; const char *column_name = ctx->argv[3]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_column *column; const struct ovsdb_idl_row *row; const struct ovsdb_type *type; struct ovsdb_datum old; int i; table = get_table(table_name); row = must_get_row(ctx, table, record_id); die_if_error(get_column(table, column_name, &column)); type = &column->type; ovsdb_datum_clone(&old, ovsdb_idl_read(row, column), &column->type); for (i = 4; i < ctx->argc; i++) { struct ovsdb_type add_type; struct ovsdb_datum add; add_type = *type; add_type.n_min = 1; add_type.n_max = UINT_MAX; die_if_error(ovsdb_datum_from_string(&add, &add_type, ctx->argv[i], ctx->symtab)); ovsdb_datum_union(&old, &add, type, false); ovsdb_datum_destroy(&add, type); } if (old.n > type->n_max) { vtep_ctl_fatal("\"add\" operation would put %u %s in column %s of " "table %s but the maximum number is %u", old.n, type->value.type == OVSDB_TYPE_VOID ? "values" : "pairs", column->name, table->class->name, type->n_max); } ovsdb_idl_txn_verify(row, column); ovsdb_idl_txn_write(row, column, &old); } static void pre_cmd_remove(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *column_name = ctx->argv[3]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_column *column; table = pre_get_table(ctx, table_name); pre_get_column(ctx, table, column_name, &column); check_mutable(table, column); } static void cmd_remove(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *record_id = ctx->argv[2]; const char *column_name = ctx->argv[3]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_column *column; const struct ovsdb_idl_row *row; const struct ovsdb_type *type; struct ovsdb_datum old; int i; table = get_table(table_name); row = must_get_row(ctx, table, record_id); die_if_error(get_column(table, column_name, &column)); type = &column->type; ovsdb_datum_clone(&old, ovsdb_idl_read(row, column), &column->type); for (i = 4; i < ctx->argc; i++) { struct ovsdb_type rm_type; struct ovsdb_datum rm; char *error; rm_type = *type; rm_type.n_min = 1; rm_type.n_max = UINT_MAX; error = ovsdb_datum_from_string(&rm, &rm_type, ctx->argv[i], ctx->symtab); if (error && ovsdb_type_is_map(&rm_type)) { rm_type.value.type = OVSDB_TYPE_VOID; die_if_error(ovsdb_datum_from_string(&rm, &rm_type, ctx->argv[i], ctx->symtab)); } free(error); ovsdb_datum_subtract(&old, type, &rm, &rm_type); ovsdb_datum_destroy(&rm, &rm_type); } if (old.n < type->n_min) { vtep_ctl_fatal("\"remove\" operation would put %u %s in column %s of " "table %s but the minimum number is %u", old.n, type->value.type == OVSDB_TYPE_VOID ? "values" : "pairs", column->name, table->class->name, type->n_min); } ovsdb_idl_txn_verify(row, column); ovsdb_idl_txn_write(row, column, &old); } static void pre_cmd_clear(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; int i; table = pre_get_table(ctx, table_name); for (i = 3; i < ctx->argc; i++) { const struct ovsdb_idl_column *column; pre_get_column(ctx, table, ctx->argv[i], &column); check_mutable(table, column); } } static void cmd_clear(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *record_id = ctx->argv[2]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_row *row; int i; table = get_table(table_name); row = must_get_row(ctx, table, record_id); for (i = 3; i < ctx->argc; i++) { const struct ovsdb_idl_column *column; const struct ovsdb_type *type; struct ovsdb_datum datum; die_if_error(get_column(table, ctx->argv[i], &column)); type = &column->type; if (type->n_min > 0) { vtep_ctl_fatal("\"clear\" operation cannot be applied to column %s " "of table %s, which is not allowed to be empty", column->name, table->class->name); } ovsdb_datum_init_empty(&datum); ovsdb_idl_txn_write(row, column, &datum); } } static void pre_create(struct vtep_ctl_context *ctx) { const char *id = shash_find_data(&ctx->options, "--id"); const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; table = get_table(table_name); if (!id && !table->class->is_root) { VLOG_WARN("applying \"create\" command to table %s without --id " "option will have no effect", table->class->name); } } static void cmd_create(struct vtep_ctl_context *ctx) { const char *id = shash_find_data(&ctx->options, "--id"); const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table = get_table(table_name); const struct ovsdb_idl_row *row; const struct uuid *uuid; int i; if (id) { struct ovsdb_symbol *symbol = create_symbol(ctx->symtab, id, NULL); if (table->class->is_root) { /* This table is in the root set, meaning that rows created in it * won't disappear even if they are unreferenced, so disable * warnings about that by pretending that there is a reference. */ symbol->strong_ref = true; } uuid = &symbol->uuid; } else { uuid = NULL; } row = ovsdb_idl_txn_insert(ctx->txn, table->class, uuid); for (i = 2; i < ctx->argc; i++) { set_column(table, row, ctx->argv[i], ctx->symtab); } ds_put_format(&ctx->output, UUID_FMT, UUID_ARGS(&row->uuid)); } /* This function may be used as the 'postprocess' function for commands that * insert new rows into the database. It expects that the command's 'run' * function prints the UUID reported by ovsdb_idl_txn_insert() as the command's * sole output. It replaces that output by the row's permanent UUID assigned * by the database server and appends a new-line. * * Currently we use this only for "create", because the higher-level commands * are supposed to be independent of the actual structure of the VTEP * configuration. */ static void post_create(struct vtep_ctl_context *ctx) { const struct uuid *real; struct uuid dummy; if (!uuid_from_string(&dummy, ds_cstr(&ctx->output))) { OVS_NOT_REACHED(); } real = ovsdb_idl_txn_get_insert_uuid(ctx->txn, &dummy); if (real) { ds_clear(&ctx->output); ds_put_format(&ctx->output, UUID_FMT, UUID_ARGS(real)); } ds_put_char(&ctx->output, '\n'); } static void pre_cmd_destroy(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; pre_get_table(ctx, table_name); } static void cmd_destroy(struct vtep_ctl_context *ctx) { bool must_exist = !shash_find(&ctx->options, "--if-exists"); bool delete_all = shash_find(&ctx->options, "--all"); const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; int i; table = get_table(table_name); if (delete_all && ctx->argc > 2) { vtep_ctl_fatal("--all and records argument should not be specified together"); } if (delete_all && !must_exist) { vtep_ctl_fatal("--all and --if-exists should not be specified together"); } if (delete_all) { const struct ovsdb_idl_row *row; const struct ovsdb_idl_row *next_row; for (row = ovsdb_idl_first_row(ctx->idl, table->class); row;) { next_row = ovsdb_idl_next_row(row); ovsdb_idl_txn_delete(row); row = next_row; } } else { for (i = 2; i < ctx->argc; i++) { const struct ovsdb_idl_row *row; row = (must_exist ? must_get_row : get_row)(ctx, table, ctx->argv[i]); if (row) { ovsdb_idl_txn_delete(row); } } } } #define RELOPS \ RELOP(RELOP_EQ, "=") \ RELOP(RELOP_NE, "!=") \ RELOP(RELOP_LT, "<") \ RELOP(RELOP_GT, ">") \ RELOP(RELOP_LE, "<=") \ RELOP(RELOP_GE, ">=") \ RELOP(RELOP_SET_EQ, "{=}") \ RELOP(RELOP_SET_NE, "{!=}") \ RELOP(RELOP_SET_LT, "{<}") \ RELOP(RELOP_SET_GT, "{>}") \ RELOP(RELOP_SET_LE, "{<=}") \ RELOP(RELOP_SET_GE, "{>=}") enum relop { #define RELOP(ENUM, STRING) ENUM, RELOPS #undef RELOP }; static bool is_set_operator(enum relop op) { return (op == RELOP_SET_EQ || op == RELOP_SET_NE || op == RELOP_SET_LT || op == RELOP_SET_GT || op == RELOP_SET_LE || op == RELOP_SET_GE); } static bool evaluate_relop(const struct ovsdb_datum *a, const struct ovsdb_datum *b, const struct ovsdb_type *type, enum relop op) { switch (op) { case RELOP_EQ: case RELOP_SET_EQ: return ovsdb_datum_compare_3way(a, b, type) == 0; case RELOP_NE: case RELOP_SET_NE: return ovsdb_datum_compare_3way(a, b, type) != 0; case RELOP_LT: return ovsdb_datum_compare_3way(a, b, type) < 0; case RELOP_GT: return ovsdb_datum_compare_3way(a, b, type) > 0; case RELOP_LE: return ovsdb_datum_compare_3way(a, b, type) <= 0; case RELOP_GE: return ovsdb_datum_compare_3way(a, b, type) >= 0; case RELOP_SET_LT: return b->n > a->n && ovsdb_datum_includes_all(a, b, type); case RELOP_SET_GT: return a->n > b->n && ovsdb_datum_includes_all(b, a, type); case RELOP_SET_LE: return ovsdb_datum_includes_all(a, b, type); case RELOP_SET_GE: return ovsdb_datum_includes_all(b, a, type); default: OVS_NOT_REACHED(); } } static bool is_condition_satisfied(const struct vtep_ctl_table_class *table, const struct ovsdb_idl_row *row, const char *arg, struct ovsdb_symbol_table *symtab) { static const char *operators[] = { #define RELOP(ENUM, STRING) STRING, RELOPS #undef RELOP }; const struct ovsdb_idl_column *column; const struct ovsdb_datum *have_datum; char *key_string, *value_string; struct ovsdb_type type; int operator; bool retval; char *error; error = parse_column_key_value(arg, table, &column, &key_string, &operator, operators, ARRAY_SIZE(operators), &value_string); die_if_error(error); if (!value_string) { vtep_ctl_fatal("%s: missing value", arg); } type = column->type; type.n_max = UINT_MAX; have_datum = ovsdb_idl_read(row, column); if (key_string) { union ovsdb_atom want_key; struct ovsdb_datum b; unsigned int idx; if (column->type.value.type == OVSDB_TYPE_VOID) { vtep_ctl_fatal("cannot specify key to check for non-map column %s", column->name); } die_if_error(ovsdb_atom_from_string(&want_key, &column->type.key, key_string, symtab)); type.key = type.value; type.value.type = OVSDB_TYPE_VOID; die_if_error(ovsdb_datum_from_string(&b, &type, value_string, symtab)); idx = ovsdb_datum_find_key(have_datum, &want_key, column->type.key.type); if (idx == UINT_MAX && !is_set_operator(operator)) { retval = false; } else { struct ovsdb_datum a; if (idx != UINT_MAX) { a.n = 1; a.keys = &have_datum->values[idx]; a.values = NULL; } else { a.n = 0; a.keys = NULL; a.values = NULL; } retval = evaluate_relop(&a, &b, &type, operator); } ovsdb_atom_destroy(&want_key, column->type.key.type); ovsdb_datum_destroy(&b, &type); } else { struct ovsdb_datum want_datum; die_if_error(ovsdb_datum_from_string(&want_datum, &column->type, value_string, symtab)); retval = evaluate_relop(have_datum, &want_datum, &type, operator); ovsdb_datum_destroy(&want_datum, &column->type); } free(key_string); free(value_string); return retval; } static void pre_cmd_wait_until(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const struct vtep_ctl_table_class *table; int i; table = pre_get_table(ctx, table_name); for (i = 3; i < ctx->argc; i++) { pre_parse_column_key_value(ctx, ctx->argv[i], table); } } static void cmd_wait_until(struct vtep_ctl_context *ctx) { const char *table_name = ctx->argv[1]; const char *record_id = ctx->argv[2]; const struct vtep_ctl_table_class *table; const struct ovsdb_idl_row *row; int i; table = get_table(table_name); row = get_row(ctx, table, record_id); if (!row) { ctx->try_again = true; return; } for (i = 3; i < ctx->argc; i++) { if (!is_condition_satisfied(table, row, ctx->argv[i], ctx->symtab)) { ctx->try_again = true; return; } } } /* Prepares 'ctx', which has already been initialized with * vtep_ctl_context_init(), for processing 'command'. */ static void vtep_ctl_context_init_command(struct vtep_ctl_context *ctx, struct vtep_ctl_command *command) { ctx->argc = command->argc; ctx->argv = command->argv; ctx->options = command->options; ds_swap(&ctx->output, &command->output); ctx->table = command->table; ctx->verified_ports = false; ctx->try_again = false; } /* Prepares 'ctx' for processing commands, initializing its members with the * values passed in as arguments. * * If 'command' is nonnull, calls vtep_ctl_context_init_command() to prepare for * that particular command. */ static void vtep_ctl_context_init(struct vtep_ctl_context *ctx, struct vtep_ctl_command *command, struct ovsdb_idl *idl, struct ovsdb_idl_txn *txn, const struct vteprec_global *vtep_global, struct ovsdb_symbol_table *symtab) { if (command) { vtep_ctl_context_init_command(ctx, command); } ctx->idl = idl; ctx->txn = txn; ctx->vtep_global = vtep_global; ctx->symtab = symtab; ctx->cache_valid = false; } /* Completes processing of 'command' within 'ctx'. */ static void vtep_ctl_context_done_command(struct vtep_ctl_context *ctx, struct vtep_ctl_command *command) { ds_swap(&ctx->output, &command->output); command->table = ctx->table; } /* Finishes up with 'ctx'. * * If command is nonnull, first calls vtep_ctl_context_done_command() to complete * processing that command within 'ctx'. */ static void vtep_ctl_context_done(struct vtep_ctl_context *ctx, struct vtep_ctl_command *command) { if (command) { vtep_ctl_context_done_command(ctx, command); } } static void run_prerequisites(struct vtep_ctl_command *commands, size_t n_commands, struct ovsdb_idl *idl) { struct vtep_ctl_command *c; ovsdb_idl_add_table(idl, &vteprec_table_global); for (c = commands; c < &commands[n_commands]; c++) { if (c->syntax->prerequisites) { struct vtep_ctl_context ctx; ds_init(&c->output); c->table = NULL; vtep_ctl_context_init(&ctx, c, idl, NULL, NULL, NULL); (c->syntax->prerequisites)(&ctx); vtep_ctl_context_done(&ctx, c); ovs_assert(!c->output.string); ovs_assert(!c->table); } } } static void do_vtep_ctl(const char *args, struct vtep_ctl_command *commands, size_t n_commands, struct ovsdb_idl *idl) { struct ovsdb_idl_txn *txn; const struct vteprec_global *vtep_global; enum ovsdb_idl_txn_status status; struct ovsdb_symbol_table *symtab; struct vtep_ctl_context ctx; struct vtep_ctl_command *c; struct shash_node *node; char *error = NULL; txn = the_idl_txn = ovsdb_idl_txn_create(idl); if (dry_run) { ovsdb_idl_txn_set_dry_run(txn); } ovsdb_idl_txn_add_comment(txn, "vtep-ctl: %s", args); vtep_global = vteprec_global_first(idl); if (!vtep_global) { /* XXX add verification that table is empty */ vtep_global = vteprec_global_insert(txn); } symtab = ovsdb_symbol_table_create(); for (c = commands; c < &commands[n_commands]; c++) { ds_init(&c->output); c->table = NULL; } vtep_ctl_context_init(&ctx, NULL, idl, txn, vtep_global, symtab); for (c = commands; c < &commands[n_commands]; c++) { vtep_ctl_context_init_command(&ctx, c); if (c->syntax->run) { (c->syntax->run)(&ctx); } vtep_ctl_context_done_command(&ctx, c); if (ctx.try_again) { vtep_ctl_context_done(&ctx, NULL); goto try_again; } } vtep_ctl_context_done(&ctx, NULL); SHASH_FOR_EACH (node, &symtab->sh) { struct ovsdb_symbol *symbol = node->data; if (!symbol->created) { vtep_ctl_fatal("row id \"%s\" is referenced but never created " "(e.g. with \"-- --id=%s create ...\")", node->name, node->name); } if (!symbol->strong_ref) { if (!symbol->weak_ref) { VLOG_WARN("row id \"%s\" was created but no reference to it " "was inserted, so it will not actually appear in " "the database", node->name); } else { VLOG_WARN("row id \"%s\" was created but only a weak " "reference to it was inserted, so it will not " "actually appear in the database", node->name); } } } status = ovsdb_idl_txn_commit_block(txn); if (status == TXN_UNCHANGED || status == TXN_SUCCESS) { for (c = commands; c < &commands[n_commands]; c++) { if (c->syntax->postprocess) { struct vtep_ctl_context ctx; vtep_ctl_context_init(&ctx, c, idl, txn, vtep_global, symtab); (c->syntax->postprocess)(&ctx); vtep_ctl_context_done(&ctx, c); } } } error = xstrdup(ovsdb_idl_txn_get_error(txn)); ovsdb_idl_txn_destroy(txn); txn = the_idl_txn = NULL; switch (status) { case TXN_UNCOMMITTED: case TXN_INCOMPLETE: OVS_NOT_REACHED(); case TXN_ABORTED: /* Should not happen--we never call ovsdb_idl_txn_abort(). */ vtep_ctl_fatal("transaction aborted"); case TXN_UNCHANGED: case TXN_SUCCESS: break; case TXN_TRY_AGAIN: goto try_again; case TXN_ERROR: vtep_ctl_fatal("transaction error: %s", error); case TXN_NOT_LOCKED: /* Should not happen--we never call ovsdb_idl_set_lock(). */ vtep_ctl_fatal("database not locked"); default: OVS_NOT_REACHED(); } free(error); ovsdb_symbol_table_destroy(symtab); for (c = commands; c < &commands[n_commands]; c++) { struct ds *ds = &c->output; if (c->table) { table_print(c->table, &table_style); } else if (oneline) { size_t j; ds_chomp(ds, '\n'); for (j = 0; j < ds->length; j++) { int ch = ds->string[j]; switch (ch) { case '\n': fputs("\\n", stdout); break; case '\\': fputs("\\\\", stdout); break; default: putchar(ch); } } putchar('\n'); } else { fputs(ds_cstr(ds), stdout); } ds_destroy(&c->output); table_destroy(c->table); free(c->table); shash_destroy_free_data(&c->options); } free(commands); ovsdb_idl_destroy(idl); exit(EXIT_SUCCESS); try_again: /* Our transaction needs to be rerun, or a prerequisite was not met. Free * resources and return so that the caller can try again. */ if (txn) { ovsdb_idl_txn_abort(txn); ovsdb_idl_txn_destroy(txn); } ovsdb_symbol_table_destroy(symtab); for (c = commands; c < &commands[n_commands]; c++) { ds_destroy(&c->output); table_destroy(c->table); free(c->table); } free(error); } static const struct vtep_ctl_command_syntax all_commands[] = { /* Physical Switch commands. */ {"add-ps", 1, 1, pre_get_info, cmd_add_ps, NULL, "--may-exist", RW}, {"del-ps", 1, 1, pre_get_info, cmd_del_ps, NULL, "--if-exists", RW}, {"list-ps", 0, 0, pre_get_info, cmd_list_ps, NULL, "", RO}, {"ps-exists", 1, 1, pre_get_info, cmd_ps_exists, NULL, "", RO}, /* Port commands. */ {"list-ports", 1, 1, pre_get_info, cmd_list_ports, NULL, "", RO}, {"add-port", 2, 2, pre_get_info, cmd_add_port, NULL, "--may-exist", RW}, {"del-port", 2, 2, pre_get_info, cmd_del_port, NULL, "--if-exists", RW}, /* Logical Switch commands. */ {"add-ls", 1, 1, pre_get_info, cmd_add_ls, NULL, "--may-exist", RW}, {"del-ls", 1, 1, pre_get_info, cmd_del_ls, NULL, "--if-exists", RW}, {"list-ls", 0, 0, pre_get_info, cmd_list_ls, NULL, "", RO}, {"ls-exists", 1, 1, pre_get_info, cmd_ls_exists, NULL, "", RO}, {"list-bindings", 2, 2, pre_get_info, cmd_list_bindings, NULL, "", RO}, {"bind-ls", 4, 4, pre_get_info, cmd_bind_ls, NULL, "", RO}, {"unbind-ls", 3, 3, pre_get_info, cmd_unbind_ls, NULL, "", RO}, /* MAC binding commands. */ {"add-ucast-local", 3, 4, pre_get_info, cmd_add_ucast_local, NULL, "", RW}, {"del-ucast-local", 2, 2, pre_get_info, cmd_del_ucast_local, NULL, "", RW}, {"add-mcast-local", 3, 4, pre_get_info, cmd_add_mcast_local, NULL, "", RW}, {"del-mcast-local", 3, 4, pre_get_info, cmd_del_mcast_local, NULL, "", RW}, {"clear-local-macs", 1, 1, pre_get_info, cmd_clear_local_macs, NULL, "", RO}, {"list-local-macs", 1, 1, pre_get_info, cmd_list_local_macs, NULL, "", RO}, {"add-ucast-remote", 3, 4, pre_get_info, cmd_add_ucast_remote, NULL, "", RW}, {"del-ucast-remote", 2, 2, pre_get_info, cmd_del_ucast_remote, NULL, "", RW}, {"add-mcast-remote", 3, 4, pre_get_info, cmd_add_mcast_remote, NULL, "", RW}, {"del-mcast-remote", 3, 4, pre_get_info, cmd_del_mcast_remote, NULL, "", RW}, {"clear-remote-macs", 1, 1, pre_get_info, cmd_clear_remote_macs, NULL, "", RO}, {"list-remote-macs", 1, 1, pre_get_info, cmd_list_remote_macs, NULL, "", RO}, /* Manager commands. */ {"get-manager", 0, 0, pre_manager, cmd_get_manager, NULL, "", RO}, {"del-manager", 0, 0, pre_manager, cmd_del_manager, NULL, "", RW}, {"set-manager", 1, INT_MAX, pre_manager, cmd_set_manager, NULL, "", RW}, /* Database commands. */ {"comment", 0, INT_MAX, NULL, NULL, NULL, "", RO}, {"get", 2, INT_MAX, pre_cmd_get, cmd_get, NULL, "--if-exists,--id=", RO}, {"list", 1, INT_MAX, pre_cmd_list, cmd_list, NULL, "--columns=", RO}, {"find", 1, INT_MAX, pre_cmd_find, cmd_find, NULL, "--columns=", RO}, {"set", 3, INT_MAX, pre_cmd_set, cmd_set, NULL, "", RW}, {"add", 4, INT_MAX, pre_cmd_add, cmd_add, NULL, "", RW}, {"remove", 4, INT_MAX, pre_cmd_remove, cmd_remove, NULL, "", RW}, {"clear", 3, INT_MAX, pre_cmd_clear, cmd_clear, NULL, "", RW}, {"create", 2, INT_MAX, pre_create, cmd_create, post_create, "--id=", RW}, {"destroy", 1, INT_MAX, pre_cmd_destroy, cmd_destroy, NULL, "--if-exists,--all", RW}, {"wait-until", 2, INT_MAX, pre_cmd_wait_until, cmd_wait_until, NULL, "", RO}, {NULL, 0, 0, NULL, NULL, NULL, NULL, RO}, }; static const struct vtep_ctl_command_syntax * get_all_commands(void) { return all_commands; }