[libvirt] [PATCH 5/6] Add new disk streaming commands to virsh

[Adam Litke]

Define two new virsh commands: one to control disk streaming and one to print
the status of active disk streams.

* tools/virsh.c: implement the new commands

Signed-off-by: Adam Litke <agl at us.ibm.com>
---
 tools/virsh.c |  134 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 133 insertions(+), 1 deletions(-)

diff --git a/tools/virsh.c b/tools/virsh.c
index f2d2c9d..edffc61 100644
--- a/tools/virsh.c
+++ b/tools/virsh.c
@@ -3999,6 +3999,137 @@ done:
 }
 
 /*
+ * "domstreamdisk" command
+ */
+static const vshCmdInfo info_domstreamdisk[] = {
+    {"help", gettext_noop("Stream data to a disk")},
+    {"desc", gettext_noop("Stream data to a disk connected to a running domain")},
+    { NULL, NULL },
+};
+
+static const vshCmdOptDef opts_domstreamdisk[] = {
+    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
+    {"start", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("Start streaming a disk") },
+    {"stop", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("Stop streaming a disk") },
+    {"incremental", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("Perform an incremental stream") },
+    {"path", VSH_OT_DATA, VSH_OFLAG_REQ, N_("block device")},
+    {"offset", VSH_OT_DATA, VSH_OFLAG_NONE, N_("Device offset for incremental stream")},
+    { NULL, 0, 0, NULL },
+};
+
+static int
+cmdDomStreamDisk(vshControl *ctl, const vshCmd *cmd)
+{
+    virDomainPtr dom;
+    const char *name, *path;
+    unsigned long long offset, next;
+    unsigned int flags, start, stop, incr;
+
+    if (!vshConnectionUsability(ctl, ctl->conn))
+        return FALSE;
+
+    flags = start = stop = incr = 0;
+    if (vshCommandOptBool(cmd, "start")) {
+        start = 1;
+        flags = VIR_STREAM_DISK_START;
+    }
+    if (vshCommandOptBool(cmd, "stop")) {
+        stop = 1;
+        flags = VIR_STREAM_DISK_STOP;
+    }
+    if (vshCommandOptBool(cmd, "incremental")) {
+        incr = 1;
+        flags = VIR_STREAM_DISK_ONE;
+    }
+    if (start + stop + incr != 1) {
+        vshError(ctl, _("Exactly one mode: --start, --stop, --incremental, "
+                        "is required"));
+        return FALSE;
+    }
+
+    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
+        return FALSE;
+
+    if (vshCommandOptString(cmd, "path", &path) < 0)
+        return FALSE;
+
+    if (flags == VIR_STREAM_DISK_ONE) {
+        if (vshCommandOptULongLong(cmd, "offset", &offset) < 0) {
+            vshError(ctl, _("An offset is required for incremental streaming"));
+            virDomainFree(dom);
+            return FALSE;
+        }
+    } else {
+        offset = 0;
+    }
+
+    next = virDomainStreamDisk(dom, path, offset, flags);
+    if (next == (unsigned long long) -1) {
+        virDomainFree(dom);
+        return FALSE;
+    }
+
+    if (flags == VIR_STREAM_DISK_START)
+        vshPrint (ctl, "Stream successfully started\n");
+    else if (flags == VIR_STREAM_DISK_STOP)
+        vshPrint (ctl, "Stream successfully stopped\n");
+    else
+        vshPrint (ctl, "Strem successful.  Continue at offset %llu\n", next);
+
+    virDomainFree(dom);
+    return TRUE;
+}
+
+/*
+ * "domstreamdiskinfo" command
+ */
+static const vshCmdInfo info_domstreamdiskinfo[] = {
+    {"help", gettext_noop("Get disk streaming status for a domain")},
+    {"desc", gettext_noop("Get disk streaming status for a running domain")},
+    { NULL, NULL },
+};
+
+static const vshCmdOptDef opts_domstreamdiskinfo[] = {
+    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
+    { NULL, 0, 0, NULL },
+};
+
+static int
+cmdDomStreamDiskInfo(vshControl *ctl, const vshCmd *cmd)
+{
+    virDomainPtr dom;
+    const char *name;
+    struct _virStreamDiskState streams[VIR_STREAM_DISK_MAX_STREAMS];
+    int nr_streams, i;
+
+    if (!vshConnectionUsability(ctl, ctl->conn))
+        return FALSE;
+
+    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
+        return FALSE;
+
+    nr_streams = virDomainStreamDiskInfo(dom, streams,
+                                        VIR_STREAM_DISK_MAX_STREAMS, 0);
+    if (nr_streams < 0) {
+        vshError(ctl, _("Failed to get disk stream information for domain %s"),
+            name);
+        virDomainFree(dom);
+        return FALSE;
+    }
+
+    vshPrint (ctl, "%-30s %-10s %-10s\n", _("Device"), _("Offset"),
+              _("Size"));
+    vshPrint (ctl, "----------------------------------------------------\n");
+    for (i = 0; i < nr_streams; i++) {
+        vshPrint (ctl, "%-30s %-10llu %-10llu\n", streams[i].path,
+                  streams[i].offset, streams[i].size);
+    }
+
+    virDomainFree(dom);
+    return TRUE;
+}
+
+/*
  * "net-autostart" command
  */
 static const vshCmdInfo info_network_autostart[] = {
@@ -10682,6 +10813,8 @@ static const vshCmdDef domManagementCmds[] = {
     {"domjobabort", cmdDomjobabort, opts_domjobabort, info_domjobabort},
     {"domjobinfo", cmdDomjobinfo, opts_domjobinfo, info_domjobinfo},
     {"domname", cmdDomname, opts_domname, info_domname},
+    {"domstreamdisk", cmdDomStreamDisk, opts_domstreamdisk, info_domstreamdisk},
+    {"domstreamdiskinfo", cmdDomStreamDiskInfo, opts_domstreamdiskinfo, info_domstreamdiskinfo},
     {"domuuid", cmdDomuuid, opts_domuuid, info_domuuid},
     {"domxml-from-native", cmdDomXMLFromNative, opts_domxmlfromnative, info_domxmlfromnative},
     {"domxml-to-native", cmdDomXMLToNative, opts_domxmltonative, info_domxmltonative},
@@ -11279,7 +11412,6 @@ vshCommandOptULongLong(const vshCmd *cmd, const char *name,
     return ret;
 }
 
-
 /*
  * Returns TRUE/FALSE if the option exists
  */
-- 
1.7.3