summaryrefslogtreecommitdiffstats
path: root/cat/virt-filesystems.c
diff options
context:
space:
mode:
authorRichard W.M. Jones <rjones@redhat.com>2010-11-22 17:33:35 +0000
committerRichard W.M. Jones <rjones@redhat.com>2010-11-23 10:22:08 +0000
commitfbc2555903be8c88ad9430d871cf0d27c8fded1e (patch)
treee4a779915c7e469fc3c78d05ebfb481fbf6093e4 /cat/virt-filesystems.c
parentf6d3d5677194ae7aaea70d43845341d91907b5ee (diff)
downloadlibguestfs-fbc2555903be8c88ad9430d871cf0d27c8fded1e.tar.gz
libguestfs-fbc2555903be8c88ad9430d871cf0d27c8fded1e.tar.xz
libguestfs-fbc2555903be8c88ad9430d871cf0d27c8fded1e.zip
New tool: virt-filesystems
This tool replaces virt-list-filesystems and virt-list-partitions with a new tool written in C with a more uniform command line structure and output. This existing Perl tools are deprecated but remain indefinitely.
Diffstat (limited to 'cat/virt-filesystems.c')
-rw-r--r--cat/virt-filesystems.c870
1 files changed, 870 insertions, 0 deletions
diff --git a/cat/virt-filesystems.c b/cat/virt-filesystems.c
new file mode 100644
index 00000000..580710d6
--- /dev/null
+++ b/cat/virt-filesystems.c
@@ -0,0 +1,870 @@
+/* virt-filesystems
+ * Copyright (C) 2010 Red Hat Inc.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#include <config.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <inttypes.h>
+#include <unistd.h>
+#include <getopt.h>
+#include <assert.h>
+
+#include "c-ctype.h"
+#include "human.h"
+#include "progname.h"
+
+#include "guestfs.h"
+#include "options.h"
+
+/* These globals are shared with options.c. */
+guestfs_h *g;
+
+int read_only = 1;
+int verbose = 0;
+int keys_from_stdin = 0;
+int echo_keys = 0;
+const char *libvirt_uri = NULL;
+int inspector = 0;
+
+static int csv = 0; /* --csv */
+static int human = 0; /* --human-readable|-h */
+
+/* What is selected for output. */
+#define OUTPUT_FILESYSTEMS 1
+#define OUTPUT_FILESYSTEMS_EXTRA 2
+#define OUTPUT_PARTITIONS 4
+#define OUTPUT_BLOCKDEVS 8
+#define OUTPUT_LVS 16
+#define OUTPUT_VGS 32
+#define OUTPUT_PVS 64
+#define OUTPUT_ALL INT_MAX
+static int output = 0;
+
+/* What columns to output. This is in display order. */
+#define COLUMN_NAME 1 /* always shown */
+#define COLUMN_TYPE 2
+#define COLUMN_VFS_TYPE 4 /* if --filesystems */
+#define COLUMN_VFS_LABEL 8 /* if --filesystems */
+#define COLUMN_SIZE 16 /* bytes, or human-readable if -h */
+#define COLUMN_PARENT_NAME 32 /* only for partitions, LVs */
+#define COLUMN_UUID 64 /* if --uuid */
+#define NR_COLUMNS 7
+static int columns;
+
+static char *canonical_device (const char *dev);
+static void do_output_title (void);
+static void do_output (void);
+static void do_output_end (void);
+
+static inline char *
+bad_cast (char const *s)
+{
+ return (char *) s;
+}
+
+static void __attribute__((noreturn))
+usage (int status)
+{
+ if (status != EXIT_SUCCESS)
+ fprintf (stderr, _("Try `%s --help' for more information.\n"),
+ program_name);
+ else {
+ fprintf (stdout,
+ _("%s: list filesystems, partitions, block devices, LVM in a VM\n"
+ "Copyright (C) 2010 Red Hat Inc.\n"
+ "Usage:\n"
+ " %s [--options] -d domname file\n"
+ " %s [--options] -a disk.img [-a disk.img ...] file\n"
+ "Options:\n"
+ " -a|--add image Add image\n"
+ " --all Display everything\n"
+ " --blkdevs|--block-devices\n"
+ " Display block devices\n"
+ " -c|--connect uri Specify libvirt URI for -d option\n"
+ " --csv Output as Comma-Separated Values\n"
+ " -d|--domain guest Add disks from libvirt guest\n"
+ " --echo-keys Don't turn off echo for passphrases\n"
+ " --extra Display swap and data filesystems\n"
+ " --filesystems Display mountable filesystems\n"
+ " --format[=raw|..] Force disk format for -a option\n"
+ " -h|--human-readable Human-readable sizes in --long output\n"
+ " --help Display brief help\n"
+ " --keys-from-stdin Read passphrases from stdin\n"
+ " -l|--long Long output\n"
+ " --lvs|--logvols|--logical-volumes\n"
+ " Display LVM logical volumes\n"
+ " --no-title No title in --long output\n"
+ " --parts|--partitions Display partitions\n"
+ " --pvs|--physvols|--physical-volumes\n"
+ " Display LVM physical volumes\n"
+ " --uuid|--uuids Add UUIDs to --long output\n"
+ " -v|--verbose Verbose messages\n"
+ " -V|--version Display version and exit\n"
+ " --vgs|--volgroups|--volume-groups\n"
+ " Display LVM volume groups\n"
+ " -x Trace libguestfs API calls\n"
+ "For more information, see the manpage %s(1).\n"),
+ program_name, program_name, program_name,
+ program_name);
+ }
+ exit (status);
+}
+
+int
+main (int argc, char *argv[])
+{
+ /* Set global program name that is not polluted with libtool artifacts. */
+ set_program_name (argv[0]);
+
+ setlocale (LC_ALL, "");
+ bindtextdomain (PACKAGE, LOCALEBASEDIR);
+ textdomain (PACKAGE);
+
+ enum { HELP_OPTION = CHAR_MAX + 1 };
+
+ static const char *options = "a:c:d:hlvVx";
+ static const struct option long_options[] = {
+ { "add", 1, 0, 'a' },
+ { "all", 0, 0, 0 },
+ { "blkdevs", 0, 0, 0 },
+ { "block-devices", 0, 0, 0 },
+ { "connect", 1, 0, 'c' },
+ { "csv", 0, 0, 0 },
+ { "domain", 1, 0, 'd' },
+ { "echo-keys", 0, 0, 0 },
+ { "extra", 0, 0, 0 },
+ { "filesystems", 0, 0, 0 },
+ { "format", 2, 0, 0 },
+ { "help", 0, 0, HELP_OPTION },
+ { "human-readable", 0, 0, 'h' },
+ { "keys-from-stdin", 0, 0, 0 },
+ { "long", 0, 0, 'l' },
+ { "logical-volumes", 0, 0, 0 },
+ { "logvols", 0, 0, 0 },
+ { "lvs", 0, 0, 0 },
+ { "no-title", 0, 0, 0 },
+ { "parts", 0, 0, 0 },
+ { "partitions", 0, 0, 0 },
+ { "physical-volumes", 0, 0, 0 },
+ { "physvols", 0, 0, 0 },
+ { "pvs", 0, 0, 0 },
+ { "uuid", 0, 0, 0 },
+ { "uuids", 0, 0, 0 },
+ { "verbose", 0, 0, 'v' },
+ { "version", 0, 0, 'V' },
+ { "vgs", 0, 0, 0 },
+ { "volgroups", 0, 0, 0 },
+ { "volume-groups", 0, 0, 0 },
+ { 0, 0, 0, 0 }
+ };
+ struct drv *drvs = NULL;
+ struct drv *drv;
+ const char *format = NULL;
+ int c;
+ int option_index;
+ int no_title = 0; /* --no-title */
+ int long_mode = 0; /* --long|-l */
+ int uuid = 0; /* --uuid */
+ int title;
+
+ g = guestfs_create ();
+ if (g == NULL) {
+ fprintf (stderr, _("guestfs_create: failed to create handle\n"));
+ exit (EXIT_FAILURE);
+ }
+
+ argv[0] = bad_cast (program_name);
+
+ for (;;) {
+ c = getopt_long (argc, argv, options, long_options, &option_index);
+ if (c == -1) break;
+
+ switch (c) {
+ case 0: /* options which are long only */
+ if (STREQ (long_options[option_index].name, "keys-from-stdin")) {
+ keys_from_stdin = 1;
+ } else if (STREQ (long_options[option_index].name, "echo-keys")) {
+ echo_keys = 1;
+ } else if (STREQ (long_options[option_index].name, "format")) {
+ if (!optarg || STREQ (optarg, ""))
+ format = NULL;
+ else
+ format = optarg;
+ } else if (STREQ (long_options[option_index].name, "all")) {
+ output = OUTPUT_ALL;
+ } else if (STREQ (long_options[option_index].name, "blkdevs") ||
+ STREQ (long_options[option_index].name, "block-devices")) {
+ output |= OUTPUT_BLOCKDEVS;
+ } else if (STREQ (long_options[option_index].name, "csv")) {
+ csv = 1;
+ } else if (STREQ (long_options[option_index].name, "extra")) {
+ output |= OUTPUT_FILESYSTEMS;
+ output |= OUTPUT_FILESYSTEMS_EXTRA;
+ } else if (STREQ (long_options[option_index].name, "filesystems")) {
+ output |= OUTPUT_FILESYSTEMS;
+ } else if (STREQ (long_options[option_index].name, "logical-volumes") ||
+ STREQ (long_options[option_index].name, "logvols") ||
+ STREQ (long_options[option_index].name, "lvs")) {
+ output |= OUTPUT_LVS;
+ } else if (STREQ (long_options[option_index].name, "no-title")) {
+ no_title = 1;
+ } else if (STREQ (long_options[option_index].name, "parts") ||
+ STREQ (long_options[option_index].name, "partitions")) {
+ output |= OUTPUT_PARTITIONS;
+ } else if (STREQ (long_options[option_index].name, "physical-volumes") ||
+ STREQ (long_options[option_index].name, "physvols") ||
+ STREQ (long_options[option_index].name, "pvs")) {
+ output |= OUTPUT_PVS;
+ } else if (STREQ (long_options[option_index].name, "uuid") ||
+ STREQ (long_options[option_index].name, "uuids")) {
+ uuid = 1;
+ } else if (STREQ (long_options[option_index].name, "vgs") ||
+ STREQ (long_options[option_index].name, "volgroups") ||
+ STREQ (long_options[option_index].name, "volume-groups")) {
+ output |= OUTPUT_VGS;
+ } else {
+ fprintf (stderr, _("%s: unknown long option: %s (%d)\n"),
+ program_name, long_options[option_index].name, option_index);
+ exit (EXIT_FAILURE);
+ }
+ break;
+
+ case 'a':
+ OPTION_a;
+ break;
+
+ case 'c':
+ OPTION_c;
+ break;
+
+ case 'd':
+ OPTION_d;
+ break;
+
+ case 'h':
+ human = 1;
+ break;
+
+ case 'l':
+ long_mode = 1;
+ break;
+
+ case 'v':
+ OPTION_v;
+ break;
+
+ case 'V':
+ OPTION_V;
+ break;
+
+ case 'x':
+ OPTION_x;
+ break;
+
+ case HELP_OPTION:
+ usage (EXIT_SUCCESS);
+
+ default:
+ usage (EXIT_FAILURE);
+ }
+ }
+
+ /* These are really constants, but they have to be variables for the
+ * options parsing code. Assert here that they have known-good
+ * values.
+ */
+ assert (read_only == 1);
+ assert (inspector == 0);
+
+ /* Must be no extra arguments on the command line. */
+ if (optind != argc)
+ usage (EXIT_FAILURE);
+
+ /* -h and --csv doesn't make sense. Spreadsheets will corrupt these
+ * fields. (RHBZ#600977).
+ */
+ if (human && csv) {
+ fprintf (stderr, _("%s: you cannot use -h and --csv options together.\n"),
+ program_name);
+ exit (EXIT_FAILURE);
+ }
+
+ /* Nothing selected for output, means --filesystems is implied. */
+ if (output == 0)
+ output = OUTPUT_FILESYSTEMS;
+
+ /* What columns will be displayed? */
+ columns = COLUMN_NAME;
+ if (long_mode) {
+ columns |= COLUMN_TYPE;
+ columns |= COLUMN_SIZE;
+ if ((output & OUTPUT_FILESYSTEMS)) {
+ columns |= COLUMN_VFS_TYPE;
+ columns |= COLUMN_VFS_LABEL;
+ }
+ if ((output & (OUTPUT_PARTITIONS|OUTPUT_LVS)))
+ columns |= COLUMN_PARENT_NAME;
+ if (uuid)
+ columns |= COLUMN_UUID;
+ }
+
+ /* Display title by default only in long mode. */
+ title = long_mode;
+ if (no_title)
+ title = 0;
+
+ /* User must have specified some drives. */
+ if (drvs == NULL)
+ usage (EXIT_FAILURE);
+
+ /* Add drives. */
+ add_drives (drvs, 'a');
+
+ if (guestfs_launch (g) == -1)
+ exit (EXIT_FAILURE);
+
+ /* Free up data structures, no longer needed after this point. */
+ free_drives (drvs);
+
+ if (title)
+ do_output_title ();
+ do_output ();
+ do_output_end ();
+
+ guestfs_close (g);
+
+ exit (EXIT_SUCCESS);
+}
+
+static void do_output_filesystems (void);
+static void do_output_lvs (void);
+static void do_output_vgs (void);
+static void do_output_pvs (void);
+static void do_output_partitions (void);
+static void do_output_blockdevs (void);
+static void write_row (const char *name, const char *type, const char *vfs_type, const char *vfs_label, int64_t size, const char *parent_name, const char *uuid);
+static void write_row_strings (char **strings, size_t len);
+
+static void
+do_output_title (void)
+{
+ const char *headings[NR_COLUMNS];
+ size_t len = 0;
+
+ /* NB. These strings are not localized and must not contain spaces. */
+ if ((columns & COLUMN_NAME))
+ headings[len++] = "Name";
+ if ((columns & COLUMN_TYPE))
+ headings[len++] = "Type";
+ if ((columns & COLUMN_VFS_TYPE))
+ headings[len++] = "VFS";
+ if ((columns & COLUMN_VFS_LABEL))
+ headings[len++] = "Label";
+ if ((columns & COLUMN_SIZE))
+ headings[len++] = "Size";
+ if ((columns & COLUMN_PARENT_NAME))
+ headings[len++] = "Parent";
+ if ((columns & COLUMN_UUID))
+ headings[len++] = "UUID";
+ assert (len <= NR_COLUMNS);
+
+ write_row_strings ((char **) headings, len);
+}
+
+static void
+do_output (void)
+{
+ /* The ordering here is trying to be most specific -> least specific,
+ * although that is not required or guaranteed.
+ */
+ if ((output & OUTPUT_FILESYSTEMS))
+ do_output_filesystems ();
+
+ if ((output & OUTPUT_LVS))
+ do_output_lvs ();
+
+ if ((output & OUTPUT_VGS))
+ do_output_vgs ();
+
+ if ((output & OUTPUT_PVS))
+ do_output_pvs ();
+
+ if ((output & OUTPUT_PARTITIONS))
+ do_output_partitions ();
+
+ if ((output & OUTPUT_BLOCKDEVS))
+ do_output_blockdevs ();
+}
+
+static void
+do_output_filesystems (void)
+{
+ char **fses;
+ size_t i;
+
+ fses = guestfs_list_filesystems (g);
+ if (fses == NULL)
+ exit (EXIT_FAILURE);
+
+ for (i = 0; fses[i] != NULL; i += 2) {
+ char *dev, *vfs_label = NULL, *vfs_uuid = NULL;
+ int64_t size = -1;
+
+ /* Skip swap and unknown, unless --extra flag was given. */
+ if (!(output & OUTPUT_FILESYSTEMS_EXTRA) &&
+ (STREQ (fses[i+1], "swap") || STREQ (fses[i+1], "unknown")))
+ continue;
+
+ dev = canonical_device (fses[i]);
+
+ /* Only bother to look these up if we will be displaying them,
+ * otherwise pass them as NULL.
+ */
+ if ((columns & COLUMN_VFS_LABEL)) {
+ vfs_label = guestfs_vfs_label (g, fses[i]);
+ if (vfs_label == NULL)
+ exit (EXIT_FAILURE);
+ }
+ if ((columns & COLUMN_UUID)) {
+ vfs_uuid = guestfs_vfs_uuid (g, fses[i]);
+ if (vfs_uuid == NULL)
+ exit (EXIT_FAILURE);
+ }
+ if ((columns & COLUMN_SIZE)) {
+ size = guestfs_blockdev_getsize64 (g, fses[i]);
+ if (size == -1)
+ exit (EXIT_FAILURE);
+ }
+
+ write_row (dev, "filesystem",
+ fses[i+1], vfs_label, size, NULL, vfs_uuid);
+
+ free (dev);
+ free (vfs_label);
+ free (vfs_uuid);
+ free (fses[i]);
+ free (fses[i+1]);
+ }
+
+ free (fses);
+}
+
+static void
+do_output_lvs (void)
+{
+ char **lvs;
+ size_t i;
+
+ lvs = guestfs_lvs (g);
+ if (lvs == NULL)
+ exit (EXIT_FAILURE);
+
+ for (i = 0; lvs[i] != NULL; ++i) {
+ char *uuid = NULL, *parent_name = NULL;
+ int64_t size = -1;
+
+ if ((columns & COLUMN_SIZE)) {
+ size = guestfs_blockdev_getsize64 (g, lvs[i]);
+ if (size == -1)
+ exit (EXIT_FAILURE);
+ }
+ if ((columns & COLUMN_UUID)) {
+ uuid = guestfs_lvuuid (g, lvs[i]);
+ if (uuid == NULL)
+ exit (EXIT_FAILURE);
+ }
+ if ((columns & COLUMN_PARENT_NAME)) {
+ parent_name = strdup (lvs[i]);
+ if (parent_name == NULL) {
+ perror ("strdup");
+ exit (EXIT_FAILURE);
+ }
+ char *p = strrchr (parent_name, '/');
+ if (p)
+ *p = '\0';
+ }
+
+ write_row (lvs[i], "lv",
+ NULL, NULL, size, parent_name, uuid);
+
+ free (uuid);
+ free (parent_name);
+ free (lvs[i]);
+ }
+
+ free (lvs);
+}
+
+static void
+do_output_vgs (void)
+{
+ struct guestfs_lvm_vg_list *vgs;
+ size_t i;
+
+ vgs = guestfs_vgs_full (g);
+ if (vgs == NULL)
+ exit (EXIT_FAILURE);
+
+ for (i = 0; i < vgs->len; ++i) {
+ char name[PATH_MAX];
+ char uuid[33];
+
+ strcpy (name, "/dev/");
+ strcpy (&name[5], vgs->val[i].vg_name);
+ memcpy (uuid, vgs->val[i].vg_uuid, 32);
+ uuid[32] = '\0';
+ write_row (name, "vg",
+ NULL, NULL, (int64_t) vgs->val[i].vg_size, NULL, uuid);
+
+ }
+
+ guestfs_free_lvm_vg_list (vgs);
+}
+
+static void
+do_output_pvs (void)
+{
+ struct guestfs_lvm_pv_list *pvs;
+ size_t i;
+
+ pvs = guestfs_pvs_full (g);
+ if (pvs == NULL)
+ exit (EXIT_FAILURE);
+
+ for (i = 0; i < pvs->len; ++i) {
+ char *dev;
+ char uuid[33];
+
+ dev = canonical_device (pvs->val[i].pv_name);
+
+ memcpy (uuid, pvs->val[i].pv_uuid, 32);
+ uuid[32] = '\0';
+ write_row (dev, "pv",
+ NULL, NULL, (int64_t) pvs->val[i].pv_size, NULL, uuid);
+
+ free (dev);
+ }
+
+ guestfs_free_lvm_pv_list (pvs);
+}
+
+static void
+do_output_partitions (void)
+{
+ char **parts;
+ size_t i;
+
+ parts = guestfs_list_partitions (g);
+ if (parts == NULL)
+ exit (EXIT_FAILURE);
+
+ for (i = 0; parts[i] != NULL; ++i) {
+ char *dev, *parent_name = NULL;
+ int64_t size = -1;
+
+ dev = canonical_device (parts[i]);
+
+ if ((columns & COLUMN_SIZE)) {
+ size = guestfs_blockdev_getsize64 (g, parts[i]);
+ if (size == -1)
+ exit (EXIT_FAILURE);
+ }
+ if ((columns & COLUMN_PARENT_NAME)) {
+ parent_name = guestfs_part_to_dev (g, parts[i]);
+ if (parent_name == NULL)
+ exit (EXIT_FAILURE);
+ char *p = canonical_device (parent_name);
+ free (parent_name);
+ parent_name = p;
+ }
+
+ write_row (dev, "partition",
+ NULL, NULL, size, parent_name, NULL);
+
+ free (dev);
+ free (parent_name);
+ free (parts[i]);
+ }
+
+ free (parts);
+}
+
+static void
+do_output_blockdevs (void)
+{
+ char **devices;
+ size_t i;
+
+ devices = guestfs_list_devices (g);
+ if (devices == NULL)
+ exit (EXIT_FAILURE);
+
+ for (i = 0; devices[i] != NULL; ++i) {
+ int64_t size = -1;
+ char *dev;
+
+ dev = canonical_device (devices[i]);
+
+ if ((columns & COLUMN_SIZE)) {
+ size = guestfs_blockdev_getsize64 (g, devices[i]);
+ if (size == -1)
+ exit (EXIT_FAILURE);
+ }
+
+ write_row (dev, "device",
+ NULL, NULL, size, NULL, NULL);
+
+ free (dev);
+ free (devices[i]);
+ }
+
+ free (devices);
+}
+
+/* /dev/vda1 -> /dev/sda. Returns a string which the caller must free. */
+static char *
+canonical_device (const char *dev)
+{
+ char *ret = strdup (dev);
+ if (ret == NULL) {
+ perror ("strdup");
+ exit (EXIT_FAILURE);
+ }
+
+ if (STRPREFIX (ret, "/dev/") &&
+ (ret[5] == 'h' || ret[5] == 'v') &&
+ ret[6] == 'd' &&
+ c_isalpha (ret[7]) &&
+ (c_isdigit (ret[8]) || ret[8] == '\0'))
+ ret[5] = 's';
+
+ return ret;
+}
+
+static void
+write_row (const char *name, const char *type,
+ const char *vfs_type, const char *vfs_label,
+ int64_t size, const char *parent_name, const char *uuid)
+{
+ const char *strings[NR_COLUMNS];
+ size_t len = 0;
+ char hum[LONGEST_HUMAN_READABLE];
+ char num[256];
+
+ if ((columns & COLUMN_NAME))
+ strings[len++] = name;
+ if ((columns & COLUMN_TYPE))
+ strings[len++] = type;
+ if ((columns & COLUMN_VFS_TYPE))
+ strings[len++] = vfs_type;
+ if ((columns & COLUMN_VFS_LABEL))
+ strings[len++] = vfs_label;
+ if ((columns & COLUMN_SIZE)) {
+ if (size >= 0) {
+ if (human) {
+ strings[len++] =
+ human_readable ((uintmax_t) size, hum,
+ human_round_to_nearest|human_autoscale|
+ human_base_1024|human_SI,
+ 1, 1);
+ }
+ else {
+ snprintf (num, sizeof num, "%" PRIi64, size);
+ strings[len++] = num;
+ }
+ }
+ else
+ strings[len++] = NULL;
+ }
+ if ((columns & COLUMN_PARENT_NAME))
+ strings[len++] = parent_name;
+ if ((columns & COLUMN_UUID))
+ strings[len++] = uuid;
+ assert (len <= NR_COLUMNS);
+
+ write_row_strings ((char **) strings, len);
+}
+
+static void add_row (char **strings, size_t len);
+static void write_csv_field (const char *field);
+
+static void
+write_row_strings (char **strings, size_t len)
+{
+ if (!csv) {
+ /* Text mode. Because we want the columns to line up, we can't
+ * output directly, but instead need to save up the rows and
+ * output them at the end.
+ */
+ add_row (strings, len);
+ }
+ else { /* CSV mode: output it directly, quoted */
+ size_t i;
+
+ for (i = 0; i < len; ++i) {
+ if (i > 0)
+ putchar (',');
+ if (strings[i] != NULL)
+ write_csv_field (strings[i]);
+ }
+ putchar ('\n');
+ }
+}
+
+/* Function to quote CSV fields on output without requiring an
+ * external module.
+ */
+static void
+write_csv_field (const char *field)
+{
+ size_t i, len;
+ int needs_quoting = 0;
+
+ len = strlen (field);
+
+ for (i = 0; i < len; ++i) {
+ if (field[i] == ' ' || field[i] == '"' ||
+ field[i] == '\n' || field[i] == ',') {
+ needs_quoting = 1;
+ break;
+ }
+ }
+
+ if (!needs_quoting) {
+ printf ("%s", field);
+ return;
+ }
+
+ /* Quoting for CSV fields. */
+ putchar ('"');
+ for (i = 0; i < len; ++i) {
+ if (field[i] == '"') {
+ putchar ('"');
+ putchar ('"');
+ } else
+ putchar (field[i]);
+ }
+ putchar ('"');
+}
+
+/* This code is only used in text mode (non-CSV output). */
+static char ***rows = NULL;
+static size_t nr_rows = 0;
+static size_t max_width[NR_COLUMNS];
+
+static void
+add_row (char **strings, size_t len)
+{
+ size_t i, slen;
+ char **row;
+
+ assert (len <= NR_COLUMNS);
+
+ row = malloc (sizeof (char *) * len);
+ if (row == NULL) {
+ perror ("malloc");
+ exit (EXIT_FAILURE);
+ }
+
+ for (i = 0; i < len; ++i) {
+ if (strings[i]) {
+ row[i] = strdup (strings[i]);
+ if (row[i] == NULL) {
+ perror ("strdup");
+ exit (EXIT_FAILURE);
+ }
+
+ /* Keep a running total of the max width of each column. */
+ slen = strlen (strings[i]);
+ if (slen == 0)
+ slen = 1; /* because "" is printed as "-" */
+ if (slen > max_width[i])
+ max_width[i] = slen;
+ }
+ else
+ row[i] = NULL;
+ }
+
+ rows = realloc (rows, sizeof (char **) * (nr_rows + 1));
+ if (rows == NULL) {
+ perror ("realloc");
+ exit (EXIT_FAILURE);
+ }
+ rows[nr_rows] = row;
+ nr_rows++;
+}
+
+/* In text mode we saved up all the output so that we can print the
+ * columns aligned.
+ */
+static void
+do_output_end (void)
+{
+ size_t i, j, k, len, space_btwn;
+
+ if (csv)
+ return;
+
+ /* How much space between columns? Try 2 spaces between columns, but
+ * if that just pushes us over 72 columns, use 1 space.
+ */
+ space_btwn = 2;
+ i = 0;
+ for (j = 0; j < NR_COLUMNS; ++j)
+ i += max_width[j] + space_btwn;
+ if (i > 72)
+ space_btwn = 1;
+
+ for (i = 0; i < nr_rows; ++i) {
+ char **row = rows[i];
+
+ k = 0;
+
+ for (j = 0; j < NR_COLUMNS; ++j) {
+ /* Ignore columns which are completely empty. This also deals
+ * with the fact that we didn't remember the length of each row
+ * in add_row above.
+ */
+ if (max_width[j] == 0)
+ continue;
+
+ while (k) {
+ putchar (' ');
+ k--;
+ }
+
+ if (row[j] == NULL || STREQ (row[j], "")) {
+ printf ("-");
+ len = 1;
+ } else {
+ printf ("%s", row[j]);
+ len = strlen (row[j]);
+ }
+ free (row[j]);
+
+ assert (len <= max_width[j]);
+ k = max_width[j] - len + space_btwn;
+ }
+
+ putchar ('\n');
+ free (row);
+ }
+ free (rows);
+}