Changeset 93895 in vbox for trunk/src/VBox/ValidationKit/utils/TestExecServ

Timestamp:

Feb 23, 2022 12:48:02 PM (3 years ago)

Author:

vboxsync

Message:

Validation Kit/TxS: Implemented (local) copy file support for TxS. Useful for copying stuff around on guests. bugref:10195

File:

• trunk/src/VBox/ValidationKit/utils/TestExecServ/TestExecService.cpp (modified) (4 diffs)

trunk/src/VBox/ValidationKit/utils/TestExecServ/TestExecService.cpp

@@ -660 +660 @@
  * @returns true if valid, false if invalid.
  * @param   pPktHdr             The packet being unpacked.
- * @param   pszArgName           The argument name.
+ * @param   pszArgName          The argument name.
  * @param   psz                 Pointer to the string within pPktHdr.
  * @param   ppszExp             Where to return the expanded string.  Must be
@@ -1007 +1007 @@
 
     RTStrFree(pszPath);
+    return rc;
+}
+
+/**
+ * Copies a file from the source to the destination locally.
+ *
+ * @returns IPRT status code from send.
+ * @param   pPktHdr             The copy file packet.
+ */
+static int txsDoCopyFile(PCTXSPKTHDR pPktHdr)
+{
+    /* After the packet header follows a 32-bit file mode,
+     * the remainder of the packet are two zero terminated paths. */
+    size_t const cbMin = sizeof(TXSPKTHDR) + sizeof(RTFMODE) + 2;
+    if (pPktHdr->cb < cbMin)
+        return txsReplyBadMinSize(pPktHdr, cbMin);
+
+    RTFMODE const fMode = *(RTFMODE const *)(pPktHdr + 1);
+
+    int rc;
+
+    char       *pszSrc;
+    const char *pch;
+    if (txsIsStringValid(pPktHdr, "source", (const char *)(pPktHdr + 1) + sizeof(uint32_t) * 2, &pszSrc, &pch, &rc))
+    {
+        char *pszDst;
+        if (txsIsStringValid(pPktHdr, "dest", pch, &pszDst, NULL /* Check for string termination */, &rc))
+        {
+            rc = RTFileCopy(pszSrc, pszDst);
+            if (RT_SUCCESS(rc))
+            {
+                if (fMode) /* Do we need to set the file mode? */
+                {
+                    rc = RTPathSetMode(pszDst, fMode);
+                    if (RT_FAILURE(rc))
+                        rc = txsReplyRC(pPktHdr, rc, "RTPathSetMode(\"%s\", %#x)", pszDst, fMode);
+                }
+
+                if (RT_SUCCESS(rc))
+                    rc = txsReplyAck(pPktHdr);
+            }
+            else
+                rc = txsReplyRC(pPktHdr, rc, "RTFileCopy");
+            RTStrFree(pszDst);
+        }
+
+        RTStrFree(pszSrc);
+    }
+
     return rc;
 }
@@ -3105 +3154 @@
         else if (txsIsSameOpcode(pPktHdr, "LIST    "))
             rc = txsDoList(pPktHdr);
+        else if (txsIsSameOpcode(pPktHdr, "CPFILE  "))
+            rc = txsDoCopyFile(pPktHdr);
         else if (txsIsSameOpcode(pPktHdr, "PUT FILE"))
             rc = txsDoPutFile(pPktHdr, false /*fHasMode*/);
@@ -3973 +4024 @@
     return rcExit;
 }
-