From 211606379164bbca5da15ac0e46e8feaf6f438e6 Mon Sep 17 00:00:00 2001
From: Peter Krempa <pkrempa@redhat.com>
Date: Thu, 1 Apr 2021 16:09:54 +0200
Subject: [PATCH] virCommandToString: Allow stripping command path
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

In tests we don't want to use the full path to commands as it's
unpleasant to keep that working on all systems.

Add an integrated way to strip the prefix which will be used to replace
virTestClearCommandPath() as a more systemic solution.

Signed-off-by: Peter Krempa <pkrempa@redhat.com>
Reviewed-by: Ján Tomko <jtomko@redhat.com>
Reviewed-by: Pavel Hrdina <phrdina@redhat.com>
---
 src/libvirt_private.syms |  1 +
 src/util/vircommand.c    | 23 ++++++++++++++++++++---
 src/util/vircommand.h    |  4 ++++
 3 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/src/libvirt_private.syms b/src/libvirt_private.syms
index c7b8d18e42..fb79884bf3 100644
--- a/src/libvirt_private.syms
+++ b/src/libvirt_private.syms
@@ -2030,6 +2030,7 @@ virCommandSetUID;
 virCommandSetUmask;
 virCommandSetWorkingDirectory;
 virCommandToString;
+virCommandToStringFull;
 virCommandWait;
 virCommandWriteArgLog;
 virFork;
diff --git a/src/util/vircommand.c b/src/util/vircommand.c
index eb70f8cf85..07e64c92dd 100644
--- a/src/util/vircommand.c
+++ b/src/util/vircommand.c
@@ -2055,9 +2055,10 @@ virCommandWriteArgLog(virCommandPtr cmd, int logfd)
 
 
 /**
- * virCommandToString:
+ * virCommandToStringFull:
  * @cmd: the command to convert
  * @linebreaks: true to break line after each env var or option
+ * @stripCommandPath: strip the path leading to the binary of @cmd
  *
  * Call after adding all arguments and environment settings, but
  * before Run/RunAsync, to return a string representation of the
@@ -2067,11 +2068,15 @@ virCommandWriteArgLog(virCommandPtr cmd, int logfd)
  * Caller is responsible for freeing the resulting string.
  */
 char *
-virCommandToString(virCommandPtr cmd, bool linebreaks)
+virCommandToStringFull(virCommandPtr cmd,
+                       bool linebreaks,
+                       bool stripCommandPath)
 {
     size_t i;
     g_auto(virBuffer) buf = VIR_BUFFER_INITIALIZER;
     bool prevopt = false;
+    const char *command = cmd->args[0];
+    g_autofree char *basename = NULL;
 
     /* Cannot assume virCommandRun will be called; so report the error
      * now.  If virCommandRun is called, it will report the same error. */
@@ -2097,7 +2102,11 @@ virCommandToString(virCommandPtr cmd, bool linebreaks)
         if (linebreaks)
             virBufferAddLit(&buf, "\\\n");
     }
-    virBufferEscapeShell(&buf, cmd->args[0]);
+
+    if (stripCommandPath)
+        command = basename = g_path_get_basename(command);
+
+    virBufferEscapeShell(&buf, command);
     for (i = 1; i < cmd->nargs; i++) {
         virBufferAddChar(&buf, ' ');
         if (linebreaks) {
@@ -2116,6 +2125,14 @@ virCommandToString(virCommandPtr cmd, bool linebreaks)
 }
 
 
+char *
+virCommandToString(virCommandPtr cmd,
+                   bool linebreaks)
+{
+    return virCommandToStringFull(cmd, linebreaks, false);
+}
+
+
 int
 virCommandGetArgList(virCommandPtr cmd,
                      char ***args,
diff --git a/src/util/vircommand.h b/src/util/vircommand.h
index a00f30f51f..5a700f8e15 100644
--- a/src/util/vircommand.h
+++ b/src/util/vircommand.h
@@ -171,6 +171,10 @@ void virCommandWriteArgLog(virCommandPtr cmd,
                            int logfd);
 
 char *virCommandToString(virCommandPtr cmd, bool linebreaks) G_GNUC_WARN_UNUSED_RESULT;
+char *virCommandToStringFull(virCommandPtr cmd,
+                             bool linebreaks,
+                             bool stripCommandPath);
+
 int virCommandGetArgList(virCommandPtr cmd, char ***args, size_t *nargs);
 
 int virCommandExec(virCommandPtr cmd, gid_t *groups, int ngroups) G_GNUC_WARN_UNUSED_RESULT;
-- 
2.47.2