[libvirt] [PATCH REPOST 5/5] virt-admin: Introduce commands srv-clients-info and srv-clients-set

[Erik Skultety]

Finally wire-up the client processing controls APIs into virt-admin client.

Signed-off-by: Erik Skultety <eskultet at redhat.com>
---
 tools/virt-admin.c | 177 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 177 insertions(+)

diff --git a/tools/virt-admin.c b/tools/virt-admin.c
index 90ad765..ee14fd4 100644
--- a/tools/virt-admin.c
+++ b/tools/virt-admin.c
@@ -809,6 +809,171 @@ cmdClientDisconnect(vshControl *ctl, const vshCmd *cmd)
     return ret;
 }
 
+/* -------------------
+ * Command srv-clients-info
+ * ------------------------
+ */
+
+static const vshCmdInfo info_srv_clients_info[] = {
+    {.name = "help",
+     .data = N_("get server's client processing controls configuration")
+    },
+    {.name = "desc",
+     .data = N_("Retrieve server's client processing controls configuration.")
+    },
+    {.name = NULL}
+};
+
+static const vshCmdOptDef opts_srv_clients_info[] = {
+    {.name = "server",
+     .type = VSH_OT_DATA,
+     .flags = VSH_OFLAG_REQ,
+     .help = N_("Server to retrieve client processing controls from."),
+    },
+    {.name = NULL}
+};
+
+static bool
+cmdSrvClientsInfo(vshControl *ctl, const vshCmd *cmd)
+{
+    bool ret = false;
+    virTypedParameterPtr params = NULL;
+    int nparams = 0;
+    size_t i;
+    const char *srvname = NULL;
+    virAdmServerPtr srv = NULL;
+    vshAdmControlPtr priv = ctl->privData;
+
+    if (vshCommandOptStringReq(ctl, cmd, "server", &srvname) < 0)
+        return false;
+
+    if (!(srv = virAdmConnectLookupServer(priv->conn, srvname, 0)))
+        goto cleanup;
+
+    if (virAdmServerGetClientProcessingControls(srv, &params,
+                                                &nparams, 0) < 0) {
+        vshError(ctl, "%s", _("Unable to retrieve client processing controls "
+                              "from server's configuration"));
+        goto cleanup;
+    }
+
+    for (i = 0; i < nparams; i++)
+        vshPrint(ctl, "%-15s: %d\n", params[i].field, params[i].value.ui);
+
+    ret = true;
+
+ cleanup:
+    virTypedParamsFree(params, nparams);
+    if (srv)
+        virAdmServerFree(srv);
+    return ret;
+}
+
+/* -----------------------
+ * Command srv-clients-set
+ * -----------------------
+ */
+
+static const vshCmdInfo info_srv_clients_set[] = {
+    {.name = "help",
+     .data = N_("set server's client processing controls configuration")
+    },
+    {.name = "desc",
+     .data = N_("Tune server's client processing controls configuration. "
+                "See OPTIONS for currently supported attributes.")
+    },
+    {.name = NULL}
+};
+
+static const vshCmdOptDef opts_srv_clients_set[] = {
+    {.name = "server",
+     .type = VSH_OT_DATA,
+     .flags = VSH_OFLAG_REQ,
+     .help = N_("Server to alter the client processing controls on."),
+    },
+    {.name = "max-clients",
+     .type = VSH_OT_INT,
+     .help = N_("Change the upper limit to overall number of clients "
+                "connected to the server."),
+    },
+    {.name = "max-unauth-clients",
+     .type = VSH_OT_INT,
+     .help = N_("Change the upper limit to number of clients waiting for "
+                "authentication to be connected to the server"),
+    },
+    {.name = NULL}
+};
+
+static bool
+cmdSrvClientsSet(vshControl *ctl, const vshCmd *cmd)
+{
+    bool ret = false;
+    int rv = 0;
+    unsigned int val, max, unauth_max;
+    int maxparams = 0;
+    int nparams = 0;
+    const char *srvname = NULL;
+    virAdmServerPtr srv = NULL;
+    virTypedParameterPtr params = NULL;
+    vshAdmControlPtr priv = ctl->privData;
+
+    if (vshCommandOptStringReq(ctl, cmd, "server", &srvname) < 0)
+        return false;
+
+#define PARSE_CMD_TYPED_PARAM(NAME, FIELD)                                   \
+    if ((rv = vshCommandOptUInt(ctl, cmd, NAME, &val)) < 0) {                \
+        vshError(ctl, _("Unable to parse integer parameter '%s'"), NAME);    \
+        goto cleanup;                                                        \
+    } else if (rv > 0) {                                                     \
+        if (virTypedParamsAddUInt(&params, &nparams, &maxparams,             \
+                                  FIELD, val) < 0)                           \
+        goto save_error;                                                     \
+    }
+
+    PARSE_CMD_TYPED_PARAM("max-clients", VIR_SERVER_CLIENTS_MAX);
+    PARSE_CMD_TYPED_PARAM("max-unauth-clients", VIR_SERVER_CLIENTS_UNAUTH_MAX);
+
+#undef PARSE_CMD_TYPED_PARAM
+
+    if (!nparams) {
+        vshError(ctl, "%s", _("At least one of options --max-clients, "
+                              "--max-unauth-clients is mandatory"));
+        goto cleanup;
+    }
+
+    if (virTypedParamsGetUInt(params, nparams,
+                              VIR_SERVER_CLIENTS_MAX, &max) &&
+        virTypedParamsGetUInt(params, nparams,
+                              VIR_SERVER_CLIENTS_UNAUTH_MAX, &unauth_max) &&
+        unauth_max > max) {
+        vshError(ctl, "%s", _("--max-unauth-clients must be less than "
+                              "--max-clients"));
+        goto cleanup;
+    }
+
+    if (!(srv = virAdmConnectLookupServer(priv->conn, srvname, 0)))
+        goto cleanup;
+
+    if (virAdmServerSetClientProcessingControls(srv, params, nparams, 0) < 0)
+        goto error;
+
+    ret = true;
+
+ cleanup:
+    virTypedParamsFree(params, nparams);
+    if (srv)
+        virAdmServerFree(srv);
+    return ret;
+
+ save_error:
+    vshSaveLibvirtError();
+
+ error:
+    vshError(ctl, "%s", _("Unable to change server's client processing "
+                          "controls configuration"));
+    goto cleanup;
+}
+
 static void *
 vshAdmConnectionHandler(vshControl *ctl)
 {
@@ -1126,6 +1291,12 @@ static const vshCmdDef monitoringCmds[] = {
      .info = info_client_info,
      .flags = 0
     },
+    {.name = "srv-clients-info",
+     .handler = cmdSrvClientsInfo,
+     .opts = opts_srv_clients_info,
+     .info = info_srv_clients_info,
+     .flags = 0
+    },
     {.name = NULL}
 };
 
@@ -1142,6 +1313,12 @@ static const vshCmdDef managementCmds[] = {
      .info = info_client_disconnect,
      .flags = 0
     },
+    {.name = "srv-clients-set",
+     .handler = cmdSrvClientsSet,
+     .opts = opts_srv_clients_set,
+     .info = info_srv_clients_set,
+     .flags = 0
+    },
     {.name = NULL}
 };
 
-- 
2.5.5