From 3b90c3c463fdb4d5e474b3b9eec50d1373016e0d Mon Sep 17 00:00:00 2001
From: Roland Schulz <schullzroll@gmail.com>
Date: Mon, 21 May 2018 13:53:44 +0200
Subject: [PATCH] Add virshAllocpagesPagesizeCompleter

Returns list of host page sizes from capabilities XML.

Signed-off-by: Roland Schulz <schullzroll@gmail.com>
---
 tools/virsh-completer.c | 71 +++++++++++++++++++++++++++++++++++++++++
 tools/virsh-completer.h |  4 +++
 tools/virsh-host.c      |  1 +
 3 files changed, 76 insertions(+)

diff --git a/tools/virsh-completer.c b/tools/virsh-completer.c
index eb091070c9..5f8a35b5f8 100644
--- a/tools/virsh-completer.c
+++ b/tools/virsh-completer.c
@@ -27,6 +27,7 @@
 #include "virsh-pool.h"
 #include "virsh-util.h"
 #include "internal.h"
+#include "virutil.h"
 #include "viralloc.h"
 #include "virstring.h"
 #include "virxml.h"
@@ -570,3 +571,73 @@ virshSnapshotNameCompleter(vshControl *ctl,
     virshDomainFree(dom);
     return NULL;
 }
+
+char **
+virshAllocpagesPagesizeCompleter(vshControl *ctl,
+                        const vshCmd *cmd ATTRIBUTE_UNUSED,
+                        unsigned int flags)
+{
+    unsigned long long byteval = 0;
+    xmlXPathContextPtr ctxt = NULL;
+    virshControlPtr priv = ctl->privData;
+    unsigned int npages = 0;
+    xmlNodePtr *pages = NULL;
+    double size = 0;
+    size_t i = 0;
+    const char *suffix = NULL;
+    char *pagesize = NULL;
+    char *cap_xml = NULL;
+    char **ret = NULL;
+    char *unit = NULL;
+
+    virCheckFlags(0, NULL);
+
+    if (!priv->conn || virConnectIsAlive(priv->conn) <= 0)
+        goto error;
+
+    if (!(cap_xml = virConnectGetCapabilities(priv->conn)))
+        goto error;
+
+    if (!(virXMLParseStringCtxt(cap_xml, _("capabilities"), &ctxt)))
+        goto error;
+
+    npages = virXPathNodeSet("/capabilities/host/cpu/pages", ctxt, &pages);
+    if (npages <= 0)
+        goto error;
+
+    if (VIR_ALLOC_N(ret, npages + 1) < 0)
+        goto error;
+
+    for (i = 0; i < npages; i++) {
+        VIR_FREE(pagesize);
+        VIR_FREE(unit);
+        pagesize = virXMLPropString(pages[i], "size");
+        unit = virXMLPropString(pages[i], "unit");
+        if (virStrToLong_ull(pagesize, NULL, 10, &byteval) < 0)
+            goto error;
+        if (virScaleInteger(&byteval, unit, 1024, UINT_MAX) < 0)
+            goto error;
+        size = vshPrettyCapacity(byteval, &suffix);
+        if (virAsprintf(&ret[i], "%.0f%s", size, suffix) < 0)
+            goto error;
+    }
+
+ cleanup:
+    xmlXPathFreeContext(ctxt);
+    for (i = 0; i < npages; i++)
+        VIR_FREE(pages[i]);
+    VIR_FREE(pages);
+    VIR_FREE(cap_xml);
+    VIR_FREE(pagesize);
+    VIR_FREE(unit);
+
+    return ret;
+
+ error:
+    if (ret) {
+        for (i = 0; i < npages; i++)
+            VIR_FREE(ret[i]);
+    }
+    VIR_FREE(ret);
+    goto cleanup;
+}
diff --git a/tools/virsh-completer.h b/tools/virsh-completer.h
index ae9626feaa..c7b181879e 100644
--- a/tools/virsh-completer.h
+++ b/tools/virsh-completer.h
@@ -74,4 +74,8 @@ char ** virshSnapshotNameCompleter(vshControl *ctl,
                                    const vshCmd *cmd,
                                    unsigned int flags);
 
+char ** virshAllocpagesPagesizeCompleter(vshControl *ctl,
+                                    const vshCmd *cmd,
+                                    unsigned int flags);
+
 #endif
diff --git a/tools/virsh-host.c b/tools/virsh-host.c
index ecaf830e35..293f06e9ed 100644
--- a/tools/virsh-host.c
+++ b/tools/virsh-host.c
@@ -472,6 +472,7 @@ static const vshCmdOptDef opts_allocpages[] = {
     {.name = "pagesize",
      .type = VSH_OT_INT,
      .flags = VSH_OFLAG_REQ,
+     .completer = virshAllocpagesPagesizeCompleter,
      .help = N_("page size (in kibibytes)")
     },
     {.name = "pagecount",
-- 
GitLab