Changeset 76905 in vbox

Timestamp:

Jan 20, 2019 7:49:15 PM (6 years ago)

Author:

vboxsync

Message:

Runtime/testcase/tstRTDvm: Resurrect testcase as an example on how to use the current API

File:

• trunk/src/VBox/Runtime/testcase/tstRTDvm.cpp (modified) (8 diffs)

trunk/src/VBox/Runtime/testcase/tstRTDvm.cpp

@@ -35 +35 @@
 #include <iprt/file.h>
 #include <iprt/string.h>
+#include <iprt/vfs.h>
 
 
@@ -40 +41 @@
 *   Structures and Typedefs                                                                                                      *
 *********************************************************************************************************************************/
-/**
- * Disk structure.
- */
-typedef struct TSTRTDVMDISK
-{
-    /** Flag whether this disk uses the image file or a volume. */
-    bool            fUseImage;
-    /** Data dependent on the flag. */
-    union
-    {
-        /** File handle of the image. */
-        RTVFSFILE   hImage;
-        /** Handle of the volume. */
-        RTDVMVOLUME hVol;
-    };
-} TSTRTDVMDISK, *PTSTRTDVMDISK;
 
 
-#if 0
-static DECLCALLBACK(int) dvmDiskRead(void *pvUser, uint64_t off, void *pvBuf, size_t cbRead)
-{
-    PTSTRTDVMDISK pDisk = (PTSTRTDVMDISK)pvUser;
-
-    if (pDisk->fUseImage)
-        return RTFileReadAt(pDisk->hImage, off, pvBuf, cbRead, NULL);
-    return RTDvmVolumeRead(pDisk->hVol, off, pvBuf, cbRead);
-}
-
-static DECLCALLBACK(int) dvmDiskWrite(void *pvUser, uint64_t off, const void *pvBuf, size_t cbWrite)
-{
-    PTSTRTDVMDISK pDisk = (PTSTRTDVMDISK)pvUser;
-
-    if (pDisk->fUseImage)
-        return RTFileWriteAt(pDisk->hImage, off, pvBuf, cbWrite, NULL);
-    return RTDvmVolumeWrite(pDisk->hVol, off, pvBuf, cbWrite);
-}
-
-static int tstRTDvmVolume(RTTEST hTest, PTSTRTDVMDISK pDisk, uint64_t cb, unsigned cNesting)
+static int tstRTDvmVolume(RTTEST hTest, RTVFSFILE hVfsDisk, unsigned cNesting)
 {
     char szPrefix[100];
@@ -92 +58 @@
     RTTestSubF(hTest, "Create DVM");
     RTDVM hVolMgr;
-    rc = RTDvmCreate(&hVolMgr, dvmDiskRead, dvmDiskWrite, cb, 512, 0, pDisk);
+    rc = RTDvmCreate(&hVolMgr, hVfsDisk, 512, 0 /*fFlags*/);
     if (RT_FAILURE(rc))
     {
@@ -110 +76 @@
         return VINF_SUCCESS;
 
-    RTTestIPrintf(RTTESTLVL_ALWAYS, "%s Successfully opened map with format: %s.\n", szPrefix, RTDvmMapGetFormat(hVolMgr));
+    RTTestIPrintf(RTTESTLVL_ALWAYS, "%s Successfully opened map with format: %s.\n", szPrefix, RTDvmMapGetFormatName(hVolMgr));
 
     /* Dump all volumes. */
@@ -147 +113 @@
          * (think of MBR partitions with a bsdlabel inside)
          */
-        TSTRTDVMDISK Disk;
-        Disk.fUseImage = false;
-        Disk.hVol      = hVol;
-        rc = tstRTDvmVolume(hTest, &Disk, RTDvmVolumeGetSize(hVol), cNesting + 1);
+        RTVFSFILE hVfsVol;
+        rc = RTDvmVolumeCreateVfsFile(hVol, RTFILE_O_OPEN | RTFILE_O_DENY_NONE | RTFILE_O_READWRITE, &hVfsVol);
+        if (RT_SUCCESS(rc))
+        {
+            rc = tstRTDvmVolume(hTest, hVfsVol, cNesting + 1);
+            RTVfsFileRelease(hVfsVol);
+        }
+        else
+            RTTestIFailed("RTDvmVolumeCreateVfsFile -> %Rrc", rc);
 
         RTDVMVOLUME hVolNext;
@@ -171 +142 @@
     return rc;
 }
-#endif
+
 
 int main(int argc, char **argv)
@@ -193 +164 @@
     }
 
-#if 1
-    RTTestFailed(hTest, "Needs updating to RTDvm API changes!");
-#else
-    /* Open image. */
-    RTFILE hFile;
-    uint64_t cb = 0;
-    rc = RTFileOpen(&hFile, argv[1], RTFILE_O_OPEN | RTFILE_O_DENY_NONE | RTFILE_O_READWRITE);
+    RTVFSFILE hVfsDisk;
+    rc = RTVfsFileOpenNormal(argv[1], RTFILE_O_OPEN | RTFILE_O_DENY_NONE | RTFILE_O_READWRITE, &hVfsDisk);
     if (RT_FAILURE(rc))
     {
-        RTTestIFailed("RTFileOpen -> %Rrc", rc);
+        RTTestIFailed("RTVfsFileOpenNormal -> %Rrc", rc);
         return RTTestSummaryAndDestroy(hTest);
     }
 
-    rc = RTFileGetSize(hFile, &cb);
+    uint64_t cb = 0;
+    rc = RTVfsFileGetSize(hVfsDisk, &cb);
     if (   RT_FAILURE(rc)
         || cb % 512 != 0) /* Assume 512 byte sector size. */
     {
-        RTTestIFailed("RTFileGetSize -> %Rrc", rc);
+        RTTestIFailed("RTVfsFileGetSize -> %Rrc", rc);
         return RTTestSummaryAndDestroy(hTest);
     }
 
-    TSTRTDVMDISK Disk;
-
-    Disk.fUseImage = true;
-    Disk.hImage    = hFile;
-    rc = tstRTDvmVolume(hTest, &Disk, cb, 0);
+    rc = tstRTDvmVolume(hTest, hVfsDisk, 0);
 
     RTTESTI_CHECK(rc == VINF_SUCCESS);
@@ -225 +188 @@
      * Summary
      */
-#endif
     return RTTestSummaryAndDestroy(hTest);
 }