Changeset 33057 in vbox

Timestamp:

Oct 12, 2010 12:10:14 PM (14 years ago)

Author:

vboxsync

Message:

Runtime: add a manifest memory writer

Location:

trunk

Files:

• include/iprt/manifest.h (modified) (1 diff)
• src/VBox/Runtime/common/checksum/manifest.cpp (modified) (1 diff)

trunk/include/iprt/manifest.h

@@ -105 +105 @@
                                    PFNRTPROGRESS pfnProgressCallback, void *pvUser);
 
+/**
+ * Creates a manifest file in memory for a set of files. The manifest file
+ * contains SHA1 sums of every provided file and could be used to verify the
+ * data integrity of them.
+ *
+ * @returns iprt status code.
+ *
+ * @param   ppvBuf               Pointer to resulting memory buffer.
+ * @param   pcbSize              Pointer for the size of the memory buffer.
+ * @param   papszFileNames       Array of file names.
+ * @param   papszFileDigests     Array of file digests.
+ * @param   cFiles               Number of entries in papszFileNames and papszFileDigests.
+ */
+RTR3DECL(int) RTManifestWriteFilesBuf(void **ppvBuf, size_t *pcbSize, const char * const *papszFileNames, const char * const *papszFileDigests, size_t cFiles);
+
 /** @} */
 

trunk/src/VBox/Runtime/common/checksum/manifest.cpp

@@ -328 +328 @@
 }
 
+RTR3DECL(int) RTManifestWriteFilesBuf(void **ppvBuf, size_t *pcbSize, const char * const *papszFileNames, const char * const *papszFileDigests, size_t cFiles)
+{
+    /* Validate input */
+    AssertPtrReturn(ppvBuf, VERR_INVALID_POINTER);
+    AssertPtrReturn(pcbSize, VERR_INVALID_POINTER);
+    AssertPtrReturn(papszFileNames, VERR_INVALID_POINTER);
+    AssertPtrReturn(papszFileDigests, VERR_INVALID_POINTER);
+    AssertReturn(cFiles > 0, VERR_INVALID_PARAMETER);
+
+    /* Calculate the size necessary for the memory buffer. */
+    size_t cbSize = 0;
+    size_t cbMaxSize = 0;
+    for (size_t i = 0; i < cFiles; ++i)
+    {
+        size_t cbTmp = strlen(RTPathFilename(papszFileNames[i])) + strlen(papszFileDigests[i]) + 10;
+        cbMaxSize = RT_MAX(cbMaxSize, cbTmp);
+        cbSize += cbTmp;
+    }
+
+    /* Create the memory buffer */
+    void *pvBuf = RTMemAlloc(cbSize);
+    if (!pvBuf)
+        return VERR_NO_MEMORY;
+
+    /* Allocate a temporary string buffer. */
+    char * pszTmp = RTStrAlloc(cbMaxSize + 1);
+    size_t cbPos = 0;
+    for (size_t i = 0; i < cFiles; ++i)
+    {
+        size_t cch = RTStrPrintf(pszTmp, cbMaxSize + 1, "SHA1 (%s)= %s\n", RTPathFilename(papszFileNames[i]), papszFileDigests[i]);
+        memcpy(&((char*)pvBuf)[cbPos], pszTmp, cch);
+        cbPos += cch;
+    }
+    RTStrFree(pszTmp);
+
+    /* Results */
+    *ppvBuf = pvBuf;
+    *pcbSize = cbSize;
+
+    return VINF_SUCCESS;
+}
+