Changeset 104975 in vbox

Timestamp:

Jun 20, 2024 10:17:11 AM (3 months ago)

Author:

vboxsync

Message:

Runtime/tools/RTTraceLogTool: Allow saving a tracelog received over network to a local file, bugref:10701

File:

• trunk/src/VBox/Runtime/tools/RTTraceLogTool.cpp (modified) (6 diffs)

trunk/src/VBox/Runtime/tools/RTTraceLogTool.cpp

@@ -44 +44 @@
 #include <iprt/assert.h>
 #include <iprt/errcore.h>
+#include <iprt/file.h>
 #include <iprt/getopt.h>
 #include <iprt/initterm.h>
@@ -100 +101 @@
     /** The TCP server. */
     PRTTCPSERVER                pTcpSrv;
+    /** File to save the read data to. */
+    RTFILE                      hFile;
 } RTTRACELOGTOOLTCP;
 /** Pointer to the TCP server/client state. */
@@ -116 +119 @@
             RTTcpClientClose(pTrcLogTcp->hSock);
     }
+    if (pTrcLogTcp->hFile != NIL_RTFILE)
+    {
+        RTFileClose(pTrcLogTcp->hFile);
+        pTrcLogTcp->hFile = NIL_RTFILE;
+    }
     RTMemFree(pTrcLogTcp);
 }
@@ -135 +143 @@
     if (RT_SUCCESS(rc))
         rc = RTTcpReadNB(pTrcLogTcp->hSock, pvBuf, cbBuf, pcbRead);
+
+    if (   RT_SUCCESS(rc)
+        && pTrcLogTcp->hFile != NIL_RTFILE)
+    {
+        int rc2 = RTFileWrite(pTrcLogTcp->hFile, pvBuf, *pcbRead, NULL);
+        if (RT_FAILURE(rc2))
+            RTMsgError("Failed to write received data to save file: %Rrc\n", rc2);
+    }
 
     return rc;
@@ -158 +174 @@
 static int rtTraceLogToolReaderCreate(PRTTRACELOGRDR phTraceLogRdr, const char *pszInput, const char *pszSave)
 {
-    RT_NOREF(pszSave);
-
     /* Try treating the input as a file first. */
     int rc = RTTraceLogRdrCreateFromFile(phTraceLogRdr, pszInput);
@@ -196 +210 @@
                 if (RT_FAILURE(rc))
                     rtTraceLogTcpDestroy(pTrcLogTcp);
+
+                if (pszSave)
+                {
+                    rc = RTFileOpen(&pTrcLogTcp->hFile, pszSave, RTFILE_O_CREATE_REPLACE | RTFILE_O_DENY_NONE | RTFILE_O_READWRITE);
+                    if (RT_FAILURE(rc))
+                        rtTraceLogTcpDestroy(pTrcLogTcp);
+                }
             }
             else