ovsdb: Add replication support and refactor files in terms of replication.
[openvswitch] / tests / test-ovsdb.c
1 /*
2  * Copyright (c) 2009 Nicira Networks.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <config.h>
18
19 #include <assert.h>
20 #include <fcntl.h>
21 #include <getopt.h>
22 #include <inttypes.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25
26 #include "command-line.h"
27 #include "json.h"
28 #include "ovsdb-data.h"
29 #include "ovsdb-error.h"
30 #include "ovsdb-types.h"
31 #include "ovsdb/column.h"
32 #include "ovsdb/condition.h"
33 #include "ovsdb/file.h"
34 #include "ovsdb/log.h"
35 #include "ovsdb/ovsdb.h"
36 #include "ovsdb/query.h"
37 #include "ovsdb/row.h"
38 #include "ovsdb/table.h"
39 #include "ovsdb/transaction.h"
40 #include "ovsdb/trigger.h"
41 #include "poll-loop.h"
42 #include "svec.h"
43 #include "timeval.h"
44 #include "util.h"
45 #include "vlog.h"
46
47 static struct command all_commands[];
48
49 static void usage(void) NO_RETURN;
50 static void parse_options(int argc, char *argv[]);
51
52 int
53 main(int argc, char *argv[])
54 {
55     set_program_name(argv[0]);
56     time_init();
57     vlog_init();
58     parse_options(argc, argv);
59     run_command(argc - optind, argv + optind, all_commands);
60     return 0;
61 }
62
63 static void
64 parse_options(int argc, char *argv[])
65 {
66     static struct option long_options[] = {
67         {"verbose", optional_argument, 0, 'v'},
68         {"help", no_argument, 0, 'h'},
69         {0, 0, 0, 0},
70     };
71     char *short_options = long_options_to_short_options(long_options);
72
73     for (;;) {
74         int c = getopt_long(argc, argv, short_options, long_options, NULL);
75         if (c == -1) {
76             break;
77         }
78
79         switch (c) {
80         case 'h':
81             usage();
82
83         case 'v':
84             vlog_set_verbosity(optarg);
85             break;
86
87         case '?':
88             exit(EXIT_FAILURE);
89
90         default:
91             abort();
92         }
93     }
94     free(short_options);
95 }
96
97 static void
98 usage(void)
99 {
100     printf("%s: Open vSwitch database test utility\n"
101            "usage: %s [OPTIONS] COMMAND [ARG...]\n\n"
102            "  log-io FILE FLAGS COMMAND...\n"
103            "    open FILE with FLAGS, run COMMANDs\n"
104            "  parse-atomic-type TYPE\n"
105            "    parse TYPE as OVSDB atomic type, and re-serialize\n"
106            "  parse-type JSON\n"
107            "    parse JSON as OVSDB type, and re-serialize\n"
108            "  parse-atoms TYPE ATOM...\n"
109            "    parse ATOMs as atoms of given TYPE, and re-serialize\n"
110            "  sort-atoms TYPE ATOM...\n"
111            "    print ATOMs in sorted order, and re-serialize\n"
112            "  parse-data TYPE DATUM...\n"
113            "    parse DATUMs as data of given TYPE, and re-serialize\n"
114            "  parse-column NAME OBJECT\n"
115            "    parse column NAME with info OBJECT, and re-serialize\n"
116            "  parse-table NAME OBJECT\n"
117            "    parse table NAME with info OBJECT\n"
118            "  parse-row TABLE ROW..., and re-serialize\n"
119            "    parse each ROW of defined TABLE\n"
120            "  compare-row TABLE ROW...\n"
121            "    mutually compare all of the ROWs, print those that are equal\n"
122            "  parse-conditions TABLE CONDITION...\n"
123            "    parse each CONDITION on TABLE, and re-serialize\n"
124            "  evaluate-conditions TABLE [CONDITION,...] [ROW,...]\n"
125            "    test CONDITIONS on TABLE against each ROW, print results\n"
126            "  query TABLE [ROW,...] [CONDITION,...]\n"
127            "    add each ROW to TABLE, then query and print the rows that\n"
128            "    satisfy each CONDITION.\n"
129            "  query-distinct TABLE [ROW,...] [CONDITION,...] COLUMNS\n"
130            "    add each ROW to TABLE, then query and print the rows that\n"
131            "    satisfy each CONDITION and have distinct COLUMNS.\n"
132            "  transact COMMAND\n"
133            "    execute each specified transactional COMMAND:\n"
134            "      commit\n"
135            "      abort\n"
136            "      insert UUID I J\n"
137            "      delete UUID\n"
138            "      modify UUID I J\n"
139            "      print\n"
140            "  execute SCHEMA TRANSACTION...\n"
141            "    executes each TRANSACTION on an initially empty database\n"
142            "    the specified SCHEMA\n"
143            "  trigger SCHEMA TRANSACTION...\n"
144            "    executes each TRANSACTION on an initially empty database\n"
145            "    the specified SCHEMA.   A TRANSACTION of the form\n"
146            "    [\"advance\", NUMBER] advances NUMBER milliseconds in\n"
147            "    simulated time, for causing triggers to time out.\n",
148            program_name, program_name);
149     vlog_usage();
150     printf("\nOther options:\n"
151            "  -h, --help                  display this help message\n");
152     exit(EXIT_SUCCESS);
153 }
154 \f
155 /* Command helper functions. */
156
157 static struct json *
158 parse_json(const char *s)
159 {
160     struct json *json = json_from_string(s);
161     if (json->type == JSON_STRING) {
162         ovs_fatal(0, "\"%s\": %s", s, json->u.string);
163     }
164     return json;
165 }
166
167 static struct json *
168 unbox_json(struct json *json)
169 {
170     if (json->type == JSON_ARRAY && json->u.array.n == 1) {
171         struct json *inner = json->u.array.elems[0];
172         json->u.array.elems[0] = NULL;
173         json_destroy(json);
174         return inner;
175     } else {
176         return json;
177     }
178 }
179
180 static void
181 print_and_free_json(struct json *json)
182 {
183     char *string = json_to_string(json, JSSF_SORT);
184     json_destroy(json);
185     puts(string);
186     free(string);
187 }
188
189 static void
190 check_ovsdb_error(struct ovsdb_error *error)
191 {
192     if (error) {
193         ovs_fatal(0, "%s", ovsdb_error_to_string(error));
194     }
195 }
196 \f
197 /* Command implementations. */
198
199 static void
200 do_log_io(int argc, char *argv[])
201 {
202     const char *name = argv[1];
203     char *mode = argv[2];
204
205     struct ovsdb_error *error;
206     struct ovsdb_log *log;
207     char *save_ptr = NULL;
208     const char *token;
209     int flags;
210     int i;
211
212     for (flags = 0, token = strtok_r(mode, " |", &save_ptr); token != NULL;
213          token = strtok_r(NULL, " |", &save_ptr))
214     {
215         if (!strcmp(token, "O_RDONLY")) {
216             flags |= O_RDONLY;
217         } else if (!strcmp(token, "O_RDWR")) {
218             flags |= O_RDWR;
219         } else if (!strcmp(token, "O_TRUNC")) {
220             flags |= O_TRUNC;
221         } else if (!strcmp(token, "O_CREAT")) {
222             flags |= O_CREAT;
223         } else if (!strcmp(token, "O_EXCL")) {
224             flags |= O_EXCL;
225         } else if (!strcmp(token, "O_TRUNC")) {
226             flags |= O_TRUNC;
227         }
228     }
229
230     check_ovsdb_error(ovsdb_log_open(name, flags, &log));
231     printf("%s: open successful\n", name);
232
233     for (i = 3; i < argc; i++) {
234         const char *command = argv[i];
235         if (!strcmp(command, "read")) {
236             struct json *json;
237
238             error = ovsdb_log_read(log, &json);
239             if (!error) {
240                 printf("%s: read: ", name);
241                 if (json) {
242                     print_and_free_json(json);
243                 } else {
244                     printf("end of log\n");
245                 }
246                 continue;
247             }
248         } else if (!strncmp(command, "write:", 6)) {
249             struct json *json = parse_json(command + 6);
250             error = ovsdb_log_write(log, json);
251             json_destroy(json);
252         } else if (!strcmp(command, "commit")) {
253             error = ovsdb_log_commit(log);
254         } else {
255             ovs_fatal(0, "unknown log-io command \"%s\"", command);
256         }
257         if (error) {
258             char *s = ovsdb_error_to_string(error);
259             printf("%s: %s failed: %s\n", name, command, s);
260             free(s);
261         } else {
262             printf("%s: %s successful\n", name, command);
263         }
264     }
265
266     ovsdb_log_close(log);
267 }
268
269 static void
270 do_parse_atomic_type(int argc UNUSED, char *argv[])
271 {
272     enum ovsdb_atomic_type type;
273     struct json *json;
274
275     json = unbox_json(parse_json(argv[1]));
276     check_ovsdb_error(ovsdb_atomic_type_from_json(&type, json));
277     json_destroy(json);
278     print_and_free_json(ovsdb_atomic_type_to_json(type));
279 }
280
281 static void
282 do_parse_type(int argc UNUSED, char *argv[])
283 {
284     struct ovsdb_type type;
285     struct json *json;
286
287     json = unbox_json(parse_json(argv[1]));
288     check_ovsdb_error(ovsdb_type_from_json(&type, json));
289     json_destroy(json);
290     print_and_free_json(ovsdb_type_to_json(&type));
291 }
292
293 static void
294 do_parse_atoms(int argc, char *argv[])
295 {
296     enum ovsdb_atomic_type type;
297     struct json *json;
298     int i;
299
300     json = unbox_json(parse_json(argv[1]));
301     check_ovsdb_error(ovsdb_atomic_type_from_json(&type, json));
302     json_destroy(json);
303
304     for (i = 2; i < argc; i++) {
305         union ovsdb_atom atom;
306
307         json = unbox_json(parse_json(argv[i]));
308         check_ovsdb_error(ovsdb_atom_from_json(&atom, type, json, NULL));
309         json_destroy(json);
310
311         print_and_free_json(ovsdb_atom_to_json(&atom, type));
312
313         ovsdb_atom_destroy(&atom, type);
314     }
315 }
316
317 static void
318 do_parse_data(int argc, char *argv[])
319 {
320     struct ovsdb_type type;
321     struct json *json;
322     int i;
323
324     json = unbox_json(parse_json(argv[1]));
325     check_ovsdb_error(ovsdb_type_from_json(&type, json));
326     json_destroy(json);
327
328     for (i = 2; i < argc; i++) {
329         struct ovsdb_datum datum;
330
331         json = unbox_json(parse_json(argv[i]));
332         check_ovsdb_error(ovsdb_datum_from_json(&datum, &type, json, NULL));
333         json_destroy(json);
334
335         print_and_free_json(ovsdb_datum_to_json(&datum, &type));
336
337         ovsdb_datum_destroy(&datum, &type);
338     }
339 }
340
341 static enum ovsdb_atomic_type compare_atoms_atomic_type;
342
343 static int
344 compare_atoms(const void *a_, const void *b_)
345 {
346     const union ovsdb_atom *a = a_;
347     const union ovsdb_atom *b = b_;
348
349     return ovsdb_atom_compare_3way(a, b, compare_atoms_atomic_type);
350 }
351
352 static void
353 do_sort_atoms(int argc UNUSED, char *argv[])
354 {
355     enum ovsdb_atomic_type type;
356     union ovsdb_atom *atoms;
357     struct json *json, **json_atoms;
358     size_t n_atoms;
359     int i;
360
361     json = unbox_json(parse_json(argv[1]));
362     check_ovsdb_error(ovsdb_atomic_type_from_json(&type, json));
363     json_destroy(json);
364
365     json = unbox_json(parse_json(argv[2]));
366     if (json->type != JSON_ARRAY) {
367         ovs_fatal(0, "second argument must be array");
368     }
369
370     /* Convert JSON atoms to internal representation. */
371     n_atoms = json->u.array.n;
372     atoms = xmalloc(n_atoms * sizeof *atoms);
373     for (i = 0; i < n_atoms; i++) {
374         check_ovsdb_error(ovsdb_atom_from_json(&atoms[i], type,
375                                                json->u.array.elems[i], NULL));
376     }
377     json_destroy(json);
378
379     /* Sort atoms. */
380     compare_atoms_atomic_type = type;
381     qsort(atoms, n_atoms, sizeof *atoms, compare_atoms);
382
383     /* Convert internal representation back to JSON. */
384     json_atoms = xmalloc(n_atoms * sizeof *json_atoms);
385     for (i = 0; i < n_atoms; i++) {
386         json_atoms[i] = ovsdb_atom_to_json(&atoms[i], type);
387         ovsdb_atom_destroy(&atoms[i], type);
388     }
389     print_and_free_json(json_array_create(json_atoms, n_atoms));
390 }
391
392 static void
393 do_parse_column(int argc UNUSED, char *argv[])
394 {
395     struct ovsdb_column *column;
396     struct json *json;
397
398     json = parse_json(argv[2]);
399     check_ovsdb_error(ovsdb_column_from_json(json, argv[1], &column));
400     json_destroy(json);
401     print_and_free_json(ovsdb_column_to_json(column));
402     ovsdb_column_destroy(column);
403 }
404
405 static void
406 do_parse_table(int argc UNUSED, char *argv[])
407 {
408     struct ovsdb_table_schema *ts;
409     struct json *json;
410
411     json = parse_json(argv[2]);
412     check_ovsdb_error(ovsdb_table_schema_from_json(json, argv[1], &ts));
413     json_destroy(json);
414     print_and_free_json(ovsdb_table_schema_to_json(ts));
415     ovsdb_table_schema_destroy(ts);
416 }
417
418 static void
419 do_parse_rows(int argc, char *argv[])
420 {
421     struct ovsdb_column_set all_columns;
422     struct ovsdb_table_schema *ts;
423     struct ovsdb_table *table;
424     struct json *json;
425     int i;
426
427     json = unbox_json(parse_json(argv[1]));
428     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
429     json_destroy(json);
430
431     table = ovsdb_table_create(ts);
432     ovsdb_column_set_init(&all_columns);
433     ovsdb_column_set_add_all(&all_columns, table);
434
435     for (i = 2; i < argc; i++) {
436         struct ovsdb_column_set columns;
437         struct ovsdb_row *row;
438
439         ovsdb_column_set_init(&columns);
440         row = ovsdb_row_create(table);
441
442         json = unbox_json(parse_json(argv[i]));
443         check_ovsdb_error(ovsdb_row_from_json(row, json, NULL, &columns));
444         json_destroy(json);
445
446         print_and_free_json(ovsdb_row_to_json(row, &all_columns));
447
448         if (columns.n_columns) {
449             struct svec names;
450             size_t j;
451             char *s;
452
453             svec_init(&names);
454             for (j = 0; j < columns.n_columns; j++) {
455                 svec_add(&names, columns.columns[j]->name);
456             }
457             svec_sort(&names);
458             s = svec_join(&names, ", ", "");
459             puts(s);
460             free(s);
461             svec_destroy(&names);
462         } else {
463             printf("<none>\n");
464         }
465
466         ovsdb_column_set_destroy(&columns);
467         ovsdb_row_destroy(row);
468     }
469
470     ovsdb_column_set_destroy(&all_columns);
471     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
472 }
473
474 static void
475 do_compare_rows(int argc, char *argv[])
476 {
477     struct ovsdb_column_set all_columns;
478     struct ovsdb_table_schema *ts;
479     struct ovsdb_table *table;
480     struct ovsdb_row **rows;
481     struct json *json;
482     char **names;
483     int n_rows;
484     int i, j;
485
486     json = unbox_json(parse_json(argv[1]));
487     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
488     json_destroy(json);
489
490     table = ovsdb_table_create(ts);
491     ovsdb_column_set_init(&all_columns);
492     ovsdb_column_set_add_all(&all_columns, table);
493
494     n_rows = argc - 2;
495     rows = xmalloc(sizeof *rows * n_rows);
496     names = xmalloc(sizeof *names * n_rows);
497     for (i = 0; i < n_rows; i++) {
498         rows[i] = ovsdb_row_create(table);
499
500         json = parse_json(argv[i + 2]);
501         if (json->type != JSON_ARRAY || json->u.array.n != 2
502             || json->u.array.elems[0]->type != JSON_STRING) {
503             ovs_fatal(0, "\"%s\" does not have expected form "
504                       "[\"name\", {data}]", argv[i]);
505         }
506         names[i] = xstrdup(json->u.array.elems[0]->u.string);
507         check_ovsdb_error(ovsdb_row_from_json(rows[i], json->u.array.elems[1],
508                                               NULL, NULL));
509         json_destroy(json);
510     }
511     for (i = 0; i < n_rows; i++) {
512         uint32_t i_hash = ovsdb_row_hash_columns(rows[i], &all_columns, 0);
513         for (j = i + 1; j < n_rows; j++) {
514             uint32_t j_hash = ovsdb_row_hash_columns(rows[j], &all_columns, 0);
515             if (ovsdb_row_equal_columns(rows[i], rows[j], &all_columns)) {
516                 printf("%s == %s\n", names[i], names[j]);
517                 if (i_hash != j_hash) {
518                     printf("but hash(%s) != hash(%s)\n", names[i], names[j]);
519                     abort();
520                 }
521             } else if (i_hash == j_hash) {
522                 printf("hash(%s) == hash(%s)\n", names[i], names[j]);
523             }
524         }
525     }
526     free(rows);
527     free(names);
528
529     ovsdb_column_set_destroy(&all_columns);
530     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
531 }
532
533 static void
534 do_parse_conditions(int argc, char *argv[])
535 {
536     struct ovsdb_table_schema *ts;
537     struct json *json;
538     int exit_code = 0;
539     int i;
540
541     json = unbox_json(parse_json(argv[1]));
542     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
543     json_destroy(json);
544
545     for (i = 2; i < argc; i++) {
546         struct ovsdb_condition cnd;
547         struct ovsdb_error *error;
548
549         json = parse_json(argv[i]);
550         error = ovsdb_condition_from_json(ts, json, NULL, &cnd);
551         if (!error) {
552             print_and_free_json(ovsdb_condition_to_json(&cnd));
553         } else {
554             char *s = ovsdb_error_to_string(error);
555             ovs_error(0, "%s", s);
556             free(s);
557             ovsdb_error_destroy(error);
558             exit_code = 1;
559         }
560         json_destroy(json);
561
562         ovsdb_condition_destroy(&cnd);
563     }
564     ovsdb_table_schema_destroy(ts);
565
566     exit(exit_code);
567 }
568
569 static void
570 do_evaluate_conditions(int argc UNUSED, char *argv[])
571 {
572     struct ovsdb_table_schema *ts;
573     struct ovsdb_table *table;
574     struct ovsdb_condition *conditions;
575     size_t n_conditions;
576     struct ovsdb_row **rows;
577     size_t n_rows;
578     struct json *json;
579     size_t i, j;
580
581     /* Parse table schema, create table. */
582     json = unbox_json(parse_json(argv[1]));
583     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
584     json_destroy(json);
585
586     table = ovsdb_table_create(ts);
587
588     /* Parse conditions. */
589     json = parse_json(argv[2]);
590     if (json->type != JSON_ARRAY) {
591         ovs_fatal(0, "CONDITION argument is not JSON array");
592     }
593     n_conditions = json->u.array.n;
594     conditions = xmalloc(n_conditions * sizeof *conditions);
595     for (i = 0; i < n_conditions; i++) {
596         check_ovsdb_error(ovsdb_condition_from_json(ts, json->u.array.elems[i],
597                                                     NULL, &conditions[i]));
598     }
599     json_destroy(json);
600
601     /* Parse rows. */
602     json = parse_json(argv[3]);
603     if (json->type != JSON_ARRAY) {
604         ovs_fatal(0, "ROW argument is not JSON array");
605     }
606     n_rows = json->u.array.n;
607     rows = xmalloc(n_rows * sizeof *rows);
608     for (i = 0; i < n_rows; i++) {
609         rows[i] = ovsdb_row_create(table);
610         check_ovsdb_error(ovsdb_row_from_json(rows[i], json->u.array.elems[i],
611                                               NULL, NULL));
612     }
613     json_destroy(json);
614
615     for (i = 0; i < n_conditions; i++) {
616         printf("condition %2d:", i);
617         for (j = 0; j < n_rows; j++) {
618             bool result = ovsdb_condition_evaluate(rows[j], &conditions[i]);
619             if (j % 5 == 0) {
620                 putchar(' ');
621             }
622             putchar(result ? 'T' : '-');
623         }
624         printf("\n");
625     }
626
627     for (i = 0; i < n_conditions; i++) {
628         ovsdb_condition_destroy(&conditions[i]);
629     }
630     for (i = 0; i < n_rows; i++) {
631         ovsdb_row_destroy(rows[i]);
632     }
633     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
634 }
635
636 struct do_query_cbdata {
637     struct uuid *row_uuids;
638     int *counts;
639     size_t n_rows;
640 };
641
642 static bool
643 do_query_cb(const struct ovsdb_row *row, void *cbdata_)
644 {
645     struct do_query_cbdata *cbdata = cbdata_;
646     size_t i;
647
648     for (i = 0; i < cbdata->n_rows; i++) {
649         if (uuid_equals(ovsdb_row_get_uuid(row), &cbdata->row_uuids[i])) {
650             cbdata->counts[i]++;
651         }
652     }
653
654     return true;
655 }
656
657 static void
658 do_query(int argc UNUSED, char *argv[])
659 {
660     struct do_query_cbdata cbdata;
661     struct ovsdb_table_schema *ts;
662     struct ovsdb_table *table;
663     struct json *json;
664     int exit_code = 0;
665     size_t i;
666
667     /* Parse table schema, create table. */
668     json = unbox_json(parse_json(argv[1]));
669     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
670     json_destroy(json);
671
672     table = ovsdb_table_create(ts);
673
674     /* Parse rows, add to table. */
675     json = parse_json(argv[2]);
676     if (json->type != JSON_ARRAY) {
677         ovs_fatal(0, "ROW argument is not JSON array");
678     }
679     cbdata.n_rows = json->u.array.n;
680     cbdata.row_uuids = xmalloc(cbdata.n_rows * sizeof *cbdata.row_uuids);
681     cbdata.counts = xmalloc(cbdata.n_rows * sizeof *cbdata.counts);
682     for (i = 0; i < cbdata.n_rows; i++) {
683         struct ovsdb_row *row = ovsdb_row_create(table);
684         uuid_generate(ovsdb_row_get_uuid_rw(row));
685         check_ovsdb_error(ovsdb_row_from_json(row, json->u.array.elems[i],
686                                               NULL, NULL));
687         if (ovsdb_table_get_row(table, ovsdb_row_get_uuid(row))) {
688             ovs_fatal(0, "duplicate UUID "UUID_FMT" in table",
689                       UUID_ARGS(ovsdb_row_get_uuid(row)));
690         }
691         cbdata.row_uuids[i] = *ovsdb_row_get_uuid(row);
692         ovsdb_table_put_row(table, row);
693     }
694     json_destroy(json);
695
696     /* Parse conditions and execute queries. */
697     json = parse_json(argv[3]);
698     if (json->type != JSON_ARRAY) {
699         ovs_fatal(0, "CONDITION argument is not JSON array");
700     }
701     for (i = 0; i < json->u.array.n; i++) {
702         struct ovsdb_condition cnd;
703         size_t j;
704
705         check_ovsdb_error(ovsdb_condition_from_json(ts, json->u.array.elems[i],
706                                                     NULL, &cnd));
707
708         memset(cbdata.counts, 0, cbdata.n_rows * sizeof *cbdata.counts);
709         ovsdb_query(table, &cnd, do_query_cb, &cbdata);
710
711         printf("query %2d:", i);
712         for (j = 0; j < cbdata.n_rows; j++) {
713             if (j % 5 == 0) {
714                 putchar(' ');
715             }
716             if (cbdata.counts[j]) {
717                 printf("%d", cbdata.counts[j]);
718                 if (cbdata.counts[j] > 1) {
719                     /* Dup! */
720                     exit_code = 1;
721                 }
722             } else {
723                 putchar('-');
724             }
725         }
726         putchar('\n');
727
728         ovsdb_condition_destroy(&cnd);
729     }
730     json_destroy(json);
731
732     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
733
734     exit(exit_code);
735 }
736
737 struct do_query_distinct_class {
738     struct ovsdb_row *example;
739     int count;
740 };
741
742 struct do_query_distinct_row {
743     struct uuid uuid;
744     struct do_query_distinct_class *class;
745 };
746
747 static void
748 do_query_distinct(int argc UNUSED, char *argv[])
749 {
750     struct ovsdb_column_set columns;
751     struct ovsdb_table_schema *ts;
752     struct ovsdb_table *table;
753     struct do_query_distinct_row *rows;
754     size_t n_rows;
755     struct do_query_distinct_class *classes;
756     size_t n_classes;
757     struct json *json;
758     int exit_code = 0;
759     size_t i, j, k;
760
761     /* Parse table schema, create table. */
762     json = unbox_json(parse_json(argv[1]));
763     check_ovsdb_error(ovsdb_table_schema_from_json(json, "mytable", &ts));
764     json_destroy(json);
765
766     table = ovsdb_table_create(ts);
767
768     /* Parse column set. */
769     json = parse_json(argv[4]);
770     ovsdb_column_set_from_json(json, table, &columns);
771     json_destroy(json);
772
773     /* Parse rows, add to table. */
774     json = parse_json(argv[2]);
775     if (json->type != JSON_ARRAY) {
776         ovs_fatal(0, "ROW argument is not JSON array");
777     }
778     n_rows = json->u.array.n;
779     rows = xmalloc(n_rows * sizeof *rows);
780     classes = xmalloc(n_rows * sizeof *classes);
781     n_classes = 0;
782     for (i = 0; i < n_rows; i++) {
783         struct ovsdb_row *row;
784         size_t j;
785
786         /* Parse row. */
787         row = ovsdb_row_create(table);
788         uuid_generate(ovsdb_row_get_uuid_rw(row));
789         check_ovsdb_error(ovsdb_row_from_json(row, json->u.array.elems[i],
790                                               NULL, NULL));
791
792         /* Initialize row and find equivalence class. */
793         rows[i].uuid = *ovsdb_row_get_uuid(row);
794         rows[i].class = NULL;
795         for (j = 0; j < n_classes; j++) {
796             if (ovsdb_row_equal_columns(row, classes[j].example, &columns)) {
797                 rows[i].class = &classes[j];
798                 break;
799             }
800         }
801         if (!rows[i].class) {
802             rows[i].class = &classes[n_classes];
803             classes[n_classes].example = ovsdb_row_clone(row);
804             n_classes++;
805         }
806
807         /* Add row to table. */
808         if (ovsdb_table_get_row(table, ovsdb_row_get_uuid(row))) {
809             ovs_fatal(0, "duplicate UUID "UUID_FMT" in table",
810                       UUID_ARGS(ovsdb_row_get_uuid(row)));
811         }
812         ovsdb_table_put_row(table, row);
813
814     }
815     json_destroy(json);
816
817     /* Parse conditions and execute queries. */
818     json = parse_json(argv[3]);
819     if (json->type != JSON_ARRAY) {
820         ovs_fatal(0, "CONDITION argument is not JSON array");
821     }
822     for (i = 0; i < json->u.array.n; i++) {
823         struct ovsdb_row_set results;
824         struct ovsdb_condition cnd;
825
826         check_ovsdb_error(ovsdb_condition_from_json(ts, json->u.array.elems[i],
827                                                     NULL, &cnd));
828
829         for (j = 0; j < n_classes; j++) {
830             classes[j].count = 0;
831         }
832         ovsdb_row_set_init(&results);
833         ovsdb_query_distinct(table, &cnd, &columns, &results);
834         for (j = 0; j < results.n_rows; j++) {
835             for (k = 0; k < n_rows; k++) {
836                 if (uuid_equals(ovsdb_row_get_uuid(results.rows[j]),
837                                 &rows[k].uuid)) {
838                     rows[k].class->count++;
839                 }
840             }
841         }
842         ovsdb_row_set_destroy(&results);
843
844         printf("query %2d:", i);
845         for (j = 0; j < n_rows; j++) {
846             int count = rows[j].class->count;
847
848             if (j % 5 == 0) {
849                 putchar(' ');
850             }
851             if (count > 1) {
852                 /* Dup! */
853                 printf("%d", count);
854                 exit_code = 1;
855             } else if (count == 1) {
856                 putchar("abcdefghijklmnopqrstuvwxyz"[rows[j].class - classes]);
857             } else {
858                 putchar('-');
859             }
860         }
861         putchar('\n');
862
863         ovsdb_condition_destroy(&cnd);
864     }
865     json_destroy(json);
866
867     ovsdb_table_destroy(table); /* Also destroys 'ts'. */
868
869     exit(exit_code);
870 }
871
872 static void
873 do_execute(int argc UNUSED, char *argv[])
874 {
875     struct ovsdb_schema *schema;
876     struct json *json;
877     struct ovsdb *db;
878     int i;
879
880     /* Create database. */
881     json = parse_json(argv[1]);
882     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
883     json_destroy(json);
884     db = ovsdb_create(schema);
885
886     for (i = 2; i < argc; i++) {
887         struct json *params, *result;
888         char *s;
889
890         params = parse_json(argv[i]);
891         result = ovsdb_execute(db, params, 0, NULL);
892         s = json_to_string(result, JSSF_SORT);
893         printf("%s\n", s);
894         json_destroy(params);
895         json_destroy(result);
896     }
897
898     ovsdb_destroy(db);
899 }
900
901 struct test_trigger {
902     struct ovsdb_trigger trigger;
903     int number;
904 };
905
906 static void
907 do_trigger_dump(struct test_trigger *t, long long int now, const char *title)
908 {
909     struct json *result;
910     char *s;
911
912     result = ovsdb_trigger_steal_result(&t->trigger);
913     s = json_to_string(result, JSSF_SORT);
914     printf("t=%lld: trigger %d (%s): %s\n", now, t->number, title, s);
915     json_destroy(result);
916     ovsdb_trigger_destroy(&t->trigger);
917     free(t);
918 }
919
920 static void
921 do_trigger(int argc UNUSED, char *argv[])
922 {
923     struct ovsdb_schema *schema;
924     struct list completions;
925     struct json *json;
926     struct ovsdb *db;
927     long long int now;
928     int number;
929     int i;
930
931     /* Create database. */
932     json = parse_json(argv[1]);
933     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
934     json_destroy(json);
935     db = ovsdb_create(schema);
936
937     list_init(&completions);
938     now = 0;
939     number = 0;
940     for (i = 2; i < argc; i++) {
941         struct json *params = parse_json(argv[i]);
942         if (params->type == JSON_ARRAY
943             && json_array(params)->n == 2
944             && json_array(params)->elems[0]->type == JSON_STRING
945             && !strcmp(json_string(json_array(params)->elems[0]), "advance")
946             && json_array(params)->elems[1]->type == JSON_INTEGER) {
947             now += json_integer(json_array(params)->elems[1]);
948             json_destroy(params);
949         } else {
950             struct test_trigger *t = xmalloc(sizeof *t);
951             ovsdb_trigger_init(db, &t->trigger, params, &completions, now);
952             t->number = number++;
953             if (ovsdb_trigger_is_complete(&t->trigger)) {
954                 do_trigger_dump(t, now, "immediate");
955             } else {
956                 printf("t=%lld: new trigger %d\n", now, t->number);
957             }
958         }
959
960         ovsdb_trigger_run(db, now);
961         while (!list_is_empty(&completions)) {
962             do_trigger_dump(CONTAINER_OF(list_pop_front(&completions),
963                                          struct test_trigger, trigger.node),
964                             now, "delayed");
965         }
966
967         ovsdb_trigger_wait(db, now);
968         poll_immediate_wake();
969         poll_block();
970     }
971
972     ovsdb_destroy(db);
973 }
974
975 static void
976 do_help(int argc UNUSED, char *argv[] UNUSED)
977 {
978     usage();
979 }
980 \f
981 /* "transact" command. */
982
983 static struct ovsdb *do_transact_db;
984 static struct ovsdb_txn *do_transact_txn;
985 static struct ovsdb_table *do_transact_table;
986
987 static void
988 do_transact_commit(int argc UNUSED, char *argv[] UNUSED)
989 {
990     ovsdb_txn_commit(do_transact_txn, false);
991     do_transact_txn = NULL;
992 }
993
994 static void
995 do_transact_abort(int argc UNUSED, char *argv[] UNUSED)
996 {
997     ovsdb_txn_abort(do_transact_txn);
998     do_transact_txn = NULL;
999 }
1000
1001 static void
1002 uuid_from_integer(int integer, struct uuid *uuid)
1003 {
1004     uuid_zero(uuid);
1005     uuid->parts[3] = integer;
1006 }
1007
1008 static const struct ovsdb_row *
1009 do_transact_find_row(const char *uuid_string)
1010 {
1011     const struct ovsdb_row *row;
1012     struct uuid uuid;
1013
1014     uuid_from_integer(atoi(uuid_string), &uuid);
1015     row = ovsdb_table_get_row(do_transact_table, &uuid);
1016     if (!row) {
1017         ovs_fatal(0, "table does not contain row with UUID "UUID_FMT,
1018                   UUID_ARGS(&uuid));
1019     }
1020     return row;
1021 }
1022
1023 static void
1024 do_transact_set_integer(struct ovsdb_row *row, const char *column_name,
1025                         int integer)
1026 {
1027     if (integer != -1) {
1028         const struct ovsdb_column *column;
1029
1030         column = ovsdb_table_schema_get_column(do_transact_table->schema,
1031                                                column_name);
1032         row->fields[column->index].keys[0].integer = integer;
1033     }
1034 }
1035
1036 static int
1037 do_transact_get_integer(const struct ovsdb_row *row, const char *column_name)
1038 {
1039     const struct ovsdb_column *column;
1040
1041     column = ovsdb_table_schema_get_column(do_transact_table->schema,
1042                                            column_name);
1043     return row->fields[column->index].keys[0].integer;
1044 }
1045
1046 static void
1047 do_transact_set_i_j(struct ovsdb_row *row,
1048                     const char *i_string, const char *j_string)
1049 {
1050     do_transact_set_integer(row, "i", atoi(i_string));
1051     do_transact_set_integer(row, "j", atoi(j_string));
1052 }
1053
1054 static void
1055 do_transact_insert(int argc UNUSED, char *argv[] UNUSED)
1056 {
1057     struct ovsdb_row *row;
1058     struct uuid *uuid;
1059
1060     row = ovsdb_row_create(do_transact_table);
1061
1062     /* Set UUID. */
1063     uuid = ovsdb_row_get_uuid_rw(row);
1064     uuid_from_integer(atoi(argv[1]), uuid);
1065     if (ovsdb_table_get_row(do_transact_table, uuid)) {
1066         ovs_fatal(0, "table already contains row with UUID "UUID_FMT,
1067                   UUID_ARGS(uuid));
1068     }
1069
1070     do_transact_set_i_j(row, argv[2], argv[3]);
1071
1072     /* Insert row. */
1073     ovsdb_txn_row_insert(do_transact_txn, row);
1074 }
1075
1076 static void
1077 do_transact_delete(int argc UNUSED, char *argv[] UNUSED)
1078 {
1079     const struct ovsdb_row *row = do_transact_find_row(argv[1]);
1080     ovsdb_txn_row_delete(do_transact_txn, row);
1081 }
1082
1083 static void
1084 do_transact_modify(int argc UNUSED, char *argv[] UNUSED)
1085 {
1086     const struct ovsdb_row *row_ro;
1087     struct ovsdb_row *row_rw;
1088
1089     row_ro = do_transact_find_row(argv[1]);
1090     row_rw = ovsdb_txn_row_modify(do_transact_txn, row_ro);
1091     do_transact_set_i_j(row_rw, argv[2], argv[3]);
1092 }
1093
1094 static int
1095 compare_rows_by_uuid(const void *a_, const void *b_)
1096 {
1097     struct ovsdb_row *const *ap = a_;
1098     struct ovsdb_row *const *bp = b_;
1099
1100     return uuid_compare_3way(ovsdb_row_get_uuid(*ap), ovsdb_row_get_uuid(*bp));
1101 }
1102
1103 static void
1104 do_transact_print(int argc UNUSED, char *argv[] UNUSED)
1105 {
1106     const struct ovsdb_row **rows;
1107     const struct ovsdb_row *row;
1108     size_t n_rows;
1109     size_t i;
1110
1111     n_rows = hmap_count(&do_transact_table->rows);
1112     rows = xmalloc(n_rows * sizeof *rows);
1113     i = 0;
1114     HMAP_FOR_EACH (row, struct ovsdb_row, hmap_node,
1115                    &do_transact_table->rows) {
1116         rows[i++] = row;
1117     }
1118     assert(i == n_rows);
1119
1120     qsort(rows, n_rows, sizeof *rows, compare_rows_by_uuid);
1121
1122     for (i = 0; i < n_rows; i++) {
1123         printf("\n%"PRId32": i=%d, j=%d",
1124                ovsdb_row_get_uuid(rows[i])->parts[3],
1125                do_transact_get_integer(rows[i], "i"),
1126                do_transact_get_integer(rows[i], "j"));
1127     }
1128
1129     free(rows);
1130 }
1131
1132 static void
1133 do_transact(int argc, char *argv[])
1134 {
1135     static const struct command do_transact_commands[] = {
1136         { "commit", 0, 0, do_transact_commit },
1137         { "abort", 0, 0, do_transact_abort },
1138         { "insert", 2, 3, do_transact_insert },
1139         { "delete", 1, 1, do_transact_delete },
1140         { "modify", 2, 3, do_transact_modify },
1141         { "print", 0, 0, do_transact_print },
1142         { NULL, 0, 0, NULL },
1143     };
1144
1145     struct ovsdb_schema *schema;
1146     struct json *json;
1147     int i;
1148
1149     /* Create table. */
1150     json = parse_json("{\"name\": \"testdb\", "
1151                       " \"tables\": "
1152                       "  {\"mytable\": "
1153                       "    {\"columns\": "
1154                       "      {\"i\": {\"type\": \"integer\"}, "
1155                       "       \"j\": {\"type\": \"integer\"}}}}}");
1156     check_ovsdb_error(ovsdb_schema_from_json(json, &schema));
1157     json_destroy(json);
1158     do_transact_db = ovsdb_create(schema);
1159     do_transact_table = ovsdb_get_table(do_transact_db, "mytable");
1160     assert(do_transact_table != NULL);
1161
1162     for (i = 1; i < argc; i++) {
1163         struct json *command;
1164         size_t n_args;
1165         char **args;
1166         int j;
1167
1168         command = parse_json(argv[i]);
1169         if (command->type != JSON_ARRAY) {
1170             ovs_fatal(0, "transaction %d must be JSON array "
1171                       "with at least 1 element", i);
1172         }
1173
1174         n_args = command->u.array.n;
1175         args = xmalloc((n_args + 1) * sizeof *args);
1176         for (j = 0; j < n_args; j++) {
1177             struct json *s = command->u.array.elems[j];
1178             if (s->type != JSON_STRING) {
1179                 ovs_fatal(0, "transaction %d argument %d must be JSON string",
1180                           i, j);
1181             }
1182             args[j] = xstrdup(json_string(s));
1183         }
1184         args[n_args] = NULL;
1185
1186         if (!do_transact_txn) {
1187             do_transact_txn = ovsdb_txn_create(do_transact_db);
1188         }
1189
1190         for (j = 0; j < n_args; j++) {
1191             if (j) {
1192                 putchar(' ');
1193             }
1194             fputs(args[j], stdout);
1195         }
1196         fputs(":", stdout);
1197         run_command(n_args, args, do_transact_commands);
1198         putchar('\n');
1199
1200         for (j = 0; j < n_args; j++) {
1201             free(args[j]);
1202         }
1203         free(args);
1204         json_destroy(command);
1205     }
1206     ovsdb_txn_abort(do_transact_txn);
1207     ovsdb_destroy(do_transact_db); /* Also destroys 'schema'. */
1208 }
1209
1210 static struct command all_commands[] = {
1211     { "log-io", 2, INT_MAX, do_log_io },
1212     { "parse-atomic-type", 1, 1, do_parse_atomic_type },
1213     { "parse-type", 1, 1, do_parse_type },
1214     { "parse-atoms", 2, INT_MAX, do_parse_atoms },
1215     { "parse-data", 2, INT_MAX, do_parse_data },
1216     { "sort-atoms", 2, 2, do_sort_atoms },
1217     { "parse-column", 2, 2, do_parse_column },
1218     { "parse-table", 2, 2, do_parse_table },
1219     { "parse-rows", 2, INT_MAX, do_parse_rows },
1220     { "compare-rows", 2, INT_MAX, do_compare_rows },
1221     { "parse-conditions", 2, INT_MAX, do_parse_conditions },
1222     { "evaluate-conditions", 3, 3, do_evaluate_conditions },
1223     { "query", 3, 3, do_query },
1224     { "query-distinct", 4, 4, do_query_distinct },
1225     { "transact", 1, INT_MAX, do_transact },
1226     { "execute", 2, INT_MAX, do_execute },
1227     { "trigger", 2, INT_MAX, do_trigger },
1228     { "help", 0, INT_MAX, do_help },
1229     { NULL, 0, 0, NULL },
1230 };