c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync#include "internal/iprt.h"

70dd77b2f7732defac9d929109f20d92676f0570vboxsync#include <iprt/manifest.h>

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync#include <iprt/err.h>

70dd77b2f7732defac9d929109f20d92676f0570vboxsync#include <iprt/file.h>

70dd77b2f7732defac9d929109f20d92676f0570vboxsync#include <iprt/mem.h>

70dd77b2f7732defac9d929109f20d92676f0570vboxsync#include <iprt/path.h>

e9b9aa0cb0196b30917a7756bc124db5e7e9ca76vboxsync#include <iprt/sha.h>

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync#include <iprt/stream.h>

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync#include <iprt/string.h>

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsynctypedef struct RTMANIFESTFILEENTRY

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char *pszManifestFile;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char *pszManifestDigest;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync PRTMANIFESTTEST pTestPattern;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync} RTMANIFESTFILEENTRY;

70dd77b2f7732defac9d929109f20d92676f0570vboxsynctypedef RTMANIFESTFILEENTRY* PRTMANIFESTFILEENTRY;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsynctypedef struct RTMANIFESTCALLBACKDATA

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync{

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync PFNRTPROGRESS pfnProgressCallback;

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync void *pvUser;

5c90ca78888c7bb1632ef4c131cc668b03227271vboxsync size_t cMaxFiles;

5c90ca78888c7bb1632ef4c131cc668b03227271vboxsync size_t cCurrentFile;

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync} RTMANIFESTCALLBACKDATA;

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsynctypedef RTMANIFESTCALLBACKDATA* PRTMANIFESTCALLBACKDATA;

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsyncDECLINLINE(char *) rtManifestPosOfCharInBuf(char const *pv, size_t cb, char c)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

0aa6851583f4cb074ccc163d5b53948b9429ab77vboxsync char *pb = (char *)pv;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync for (; cb; --cb, ++pb)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_UNLIKELY(*pb == c))

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return pb;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return NULL;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync}

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsyncDECLINLINE(size_t) rtManifestIndexOfCharInBuf(char const *pv, size_t cb, char c)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char const *pb = (char const *)pv;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync for (size_t i=0; i < cb; ++i, ++pb)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_UNLIKELY(*pb == c))

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync return i;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return cb;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync}

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsyncint rtSHAProgressCallback(unsigned uPercent, void *pvUser)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

70dd77b2f7732defac9d929109f20d92676f0570vboxsync PRTMANIFESTCALLBACKDATA pData = (PRTMANIFESTCALLBACKDATA)pvUser;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return pData->pfnProgressCallback((unsigned)( (uPercent + (float)pData->cCurrentFile * 100.0)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync / (float)pData->cMaxFiles),

70dd77b2f7732defac9d929109f20d92676f0570vboxsync pData->pvUser);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync}

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsyncRTR3DECL(int) RTManifestVerify(const char *pszManifestFile, PRTMANIFESTTEST paTests, size_t cTests, size_t *piFailed)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Validate input */

70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(pszManifestFile, VERR_INVALID_POINTER);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Open the manifest file */

70dd77b2f7732defac9d929109f20d92676f0570vboxsync RTFILE file;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync int rc = RTFileOpen(&file, pszManifestFile, RTFILE_O_OPEN | RTFILE_O_READ | RTFILE_O_DENY_NONE);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync if (RT_FAILURE(rc))

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return rc;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync void *pvBuf = 0;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync do

70dd77b2f7732defac9d929109f20d92676f0570vboxsync {

70dd77b2f7732defac9d929109f20d92676f0570vboxsync uint64_t cbSize;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = RTFileGetSize(file, &cbSize);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc))

70dd77b2f7732defac9d929109f20d92676f0570vboxsync break;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync /* Cast down for the case size_t < uint64_t. This isn't really correct,

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync size_t cbToRead = (size_t)cbSize;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync pvBuf = RTMemAlloc(cbToRead);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (!pvBuf)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = VERR_NO_MEMORY;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync break;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync size_t cbRead = 0;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = RTFileRead(file, pvBuf, cbToRead, &cbRead);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc))

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync break;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = RTManifestVerifyFilesBuf(pvBuf, cbRead, paTests, cTests, piFailed);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }while (0);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Cleanup */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (pvBuf)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync RTMemFree(pvBuf);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync RTFileClose(file);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return rc;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync}

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsyncRTR3DECL(int) RTManifestVerifyFiles(const char *pszManifestFile, const char * const *papszFiles, size_t cFiles, size_t *piFailed,

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync PFNRTPROGRESS pfnProgressCallback, void *pvUser)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

70dd77b2f7732defac9d929109f20d92676f0570vboxsync /* Validate input */

70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(pszManifestFile, VERR_INVALID_POINTER);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(papszFiles, VERR_INVALID_POINTER);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertPtrNullReturn(pfnProgressCallback, VERR_INVALID_POINTER);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync int rc = VINF_SUCCESS;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync /* Create our compare list */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync PRTMANIFESTTEST paFiles = (PRTMANIFESTTEST)RTMemTmpAllocZ(sizeof(RTMANIFESTTEST) * cFiles);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (!paFiles)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync return VERR_NO_MEMORY;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync RTMANIFESTCALLBACKDATA callback = { pfnProgressCallback, pvUser, cFiles, 0 };

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Fill our compare list */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync for (size_t i = 0; i < cFiles; ++i)

70dd77b2f7732defac9d929109f20d92676f0570vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char *pszDigest;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync if (pfnProgressCallback)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync callback.cCurrentFile = i;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = RTSha1DigestFromFile(papszFiles[i], &pszDigest, rtSHAProgressCallback, &callback);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync }

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync else

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = RTSha1DigestFromFile(papszFiles[i], &pszDigest, NULL, NULL);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc))

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync break;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync paFiles[i].pszTestFile = (char*)papszFiles[i];

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync paFiles[i].pszTestDigest = pszDigest;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Do the verification */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_SUCCESS(rc))

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = RTManifestVerify(pszManifestFile, paFiles, cFiles, piFailed);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Cleanup */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync for (size_t i = 0; i < cFiles; ++i)

70dd77b2f7732defac9d929109f20d92676f0570vboxsync {

70dd77b2f7732defac9d929109f20d92676f0570vboxsync if (paFiles[i].pszTestDigest)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync RTStrFree((char*)paFiles[i].pszTestDigest);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }

70dd77b2f7732defac9d929109f20d92676f0570vboxsync RTMemTmpFree(paFiles);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync return rc;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync}

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsyncRTR3DECL(int) RTManifestWriteFiles(const char *pszManifestFile, const char * const *papszFiles, size_t cFiles,

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync PFNRTPROGRESS pfnProgressCallback, void *pvUser)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Validate input */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertPtrReturn(pszManifestFile, VERR_INVALID_POINTER);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(papszFiles, VERR_INVALID_POINTER);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertPtrNullReturn(pfnProgressCallback, VERR_INVALID_POINTER);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync RTFILE file;

0aa6851583f4cb074ccc163d5b53948b9429ab77vboxsync int rc = RTFileOpen(&file, pszManifestFile, RTFILE_O_CREATE | RTFILE_O_WRITE | RTFILE_O_DENY_ALL);

0aa6851583f4cb074ccc163d5b53948b9429ab77vboxsync if (RT_FAILURE(rc))

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync return rc;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync PRTMANIFESTTEST paFiles = 0;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync void *pvBuf = 0;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync do

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync paFiles = (PRTMANIFESTTEST)RTMemAllocZ(sizeof(RTMANIFESTTEST) * cFiles);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync if (!paFiles)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = VERR_NO_MEMORY;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync break;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync }

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync RTMANIFESTCALLBACKDATA callback = { pfnProgressCallback, pvUser, cFiles, 0 };

70dd77b2f7732defac9d929109f20d92676f0570vboxsync for (size_t i = 0; i < cFiles; ++i)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync paFiles[i].pszTestFile = papszFiles[i];

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Calculate the SHA1 digest of every file */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (pfnProgressCallback)

70dd77b2f7732defac9d929109f20d92676f0570vboxsync {

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync callback.cCurrentFile = i;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = RTSha1DigestFromFile(paFiles[i].pszTestFile, (char**)&paFiles[i].pszTestDigest, rtSHAProgressCallback, &callback);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }

70dd77b2f7732defac9d929109f20d92676f0570vboxsync else

70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = RTSha1DigestFromFile(paFiles[i].pszTestFile, (char**)&paFiles[i].pszTestDigest, NULL, NULL);

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync if (RT_FAILURE(rc))

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync break;

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync }

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_SUCCESS(rc))

70dd77b2f7732defac9d929109f20d92676f0570vboxsync {

70dd77b2f7732defac9d929109f20d92676f0570vboxsync size_t cbSize = 0;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = RTManifestWriteFilesBuf(&pvBuf, &cbSize, paFiles, cFiles);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc))

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync break;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = RTFileWrite(file, pvBuf, cbSize, 0);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }while (0);

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync RTFileClose(file);

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync /* Cleanup */

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync if (pvBuf)

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync RTMemFree(pvBuf);

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync for (size_t i = 0; i < cFiles; ++i)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (paFiles[i].pszTestDigest)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync RTStrFree((char*)paFiles[i].pszTestDigest);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync RTMemFree(paFiles);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Delete the manifest file on failure */

70dd77b2f7732defac9d929109f20d92676f0570vboxsync if (RT_FAILURE(rc))

70dd77b2f7732defac9d929109f20d92676f0570vboxsync RTFileDelete(pszManifestFile);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync return rc;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync}

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsyncRTR3DECL(int) RTManifestVerifyFilesBuf(void *pvBuf, size_t cbSize, PRTMANIFESTTEST paTests, size_t cTests, size_t *piFailed)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{

70dd77b2f7732defac9d929109f20d92676f0570vboxsync /* Validate input */

70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(pvBuf, VERR_INVALID_POINTER);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertReturn(cbSize > 0, VERR_INVALID_PARAMETER);

70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(paTests, VERR_INVALID_POINTER);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertReturn(cTests > 0, VERR_INVALID_PARAMETER);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertPtrNullReturn(piFailed, VERR_INVALID_POINTER);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync int rc = VINF_SUCCESS;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync PRTMANIFESTFILEENTRY paFiles = (PRTMANIFESTFILEENTRY)RTMemTmpAllocZ(sizeof(RTMANIFESTFILEENTRY) * cTests);

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (!paFiles)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync return VERR_NO_MEMORY;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync /* Fill our compare list */

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync for (size_t i = 0; i < cTests; ++i)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync paFiles[i].pTestPattern = &paTests[i];

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char *pcBuf = (char*)pvBuf;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync size_t cbRead = 0;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Parse the manifest file line by line */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync for (;;)

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync if (cbRead >= cbSize)

70dd77b2f7732defac9d929109f20d92676f0570vboxsync break;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync size_t cch = rtManifestIndexOfCharInBuf(pcBuf, cbSize - cbRead, '\n') + 1;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync /* Skip empty lines (UNIX/DOS format) */

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync if ( ( cch == 1

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync && pcBuf[0] == '\n')

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync || ( cch == 2

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync && pcBuf[0] == '\r'

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync && pcBuf[1] == '\n'))

510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync pcBuf += cch;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync cbRead += cch;

70dd77b2f7732defac9d929109f20d92676f0570vboxsync continue;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /** @todo r=bird:

70dd77b2f7732defac9d929109f20d92676f0570vboxsync

70dd77b2f7732defac9d929109f20d92676f0570vboxsync /* Check for the digest algorithm */

70dd77b2f7732defac9d929109f20d92676f0570vboxsync if ( cch < 4

70dd77b2f7732defac9d929109f20d92676f0570vboxsync || !( pcBuf[0] == 'S'

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync && pcBuf[1] == 'H'

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync && pcBuf[2] == 'A'

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync && pcBuf[3] == '1'))

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync {

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Digest unsupported */

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync rc = VERR_MANIFEST_UNSUPPORTED_DIGEST_TYPE;

c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync break;

}

/* Try to find the filename */

char *pszNameStart = rtManifestPosOfCharInBuf(pcBuf, cch, '(');

if (!pszNameStart)

{

rc = VERR_MANIFEST_WRONG_FILE_FORMAT;

break;

}

char *pszNameEnd = rtManifestPosOfCharInBuf(pcBuf, cch, ')');

if (!pszNameEnd)

{

rc = VERR_MANIFEST_WRONG_FILE_FORMAT;

break;

}

/* Copy the filename part */

size_t cchName = pszNameEnd - pszNameStart - 1;

char *pszName = (char *)RTMemTmpAlloc(cchName + 1);

if (!pszName)

{

rc = VERR_NO_MEMORY;

break;

}

memcpy(pszName, pszNameStart + 1, cchName);

pszName[cchName] = '\0';

/* Try to find the digest sum */

char *pszDigestStart = rtManifestPosOfCharInBuf(pcBuf, cch, '=') + 1;

if (!pszDigestStart)

{

RTMemTmpFree(pszName);

rc = VERR_MANIFEST_WRONG_FILE_FORMAT;

break;

}

char *pszDigestEnd = rtManifestPosOfCharInBuf(pcBuf, cch, '\r');

if (!pszDigestEnd)

pszDigestEnd = rtManifestPosOfCharInBuf(pcBuf, cch, '\n');

if (!pszDigestEnd)

{

rc = VERR_MANIFEST_WRONG_FILE_FORMAT;

break;

}

/* Copy the digest part */

size_t cchDigest = pszDigestEnd - pszDigestStart - 1;

char *pszDigest = (char *)RTMemTmpAlloc(cchDigest + 1);

if (!pszDigest)

{

rc = VERR_NO_MEMORY;

break;

}

memcpy(pszDigest, pszDigestStart + 1, cchDigest);

pszDigest[cchDigest] = '\0';

/* Check our file list against the extracted data */

bool fFound = false;

for (size_t i = 0; i < cTests; ++i)

{

if (!RTStrCmp(RTPathFilename(paFiles[i].pTestPattern->pszTestFile), RTStrStrip(pszName)))

{

/* Add the data of the manifest file to the file list */

paFiles[i].pszManifestFile = RTStrDup(RTStrStrip(pszName));

paFiles[i].pszManifestDigest = RTStrDup(RTStrStrip(pszDigest));

fFound = true;

break;

}

}

RTMemTmpFree(pszName);

RTMemTmpFree(pszDigest);

if (!fFound)

{

/* There have to be an entry in the file list */

rc = VERR_MANIFEST_FILE_MISMATCH;

break;

}

pcBuf += cch;

cbRead += cch;

}

if ( rc == VINF_SUCCESS

|| rc == VERR_EOF)

{

rc = VINF_SUCCESS;

for (size_t i = 0; i < cTests; ++i)

{

/* If there is an entry in the file list, which hasn't an

if ( !paFiles[i].pszManifestFile

|| !paFiles[i].pszManifestDigest)

{

rc = VERR_MANIFEST_FILE_MISMATCH;

break;

}

/* Do the manifest SHA1 digest match against the actual digest? */

if (RTStrICmp(paFiles[i].pszManifestDigest, paFiles[i].pTestPattern->pszTestDigest))

{

if (piFailed)

*piFailed = i;

rc = VERR_MANIFEST_DIGEST_MISMATCH;

break;

}

}

}

/* Cleanup */

for (size_t i = 0; i < cTests; ++i)

{

if (paFiles[i].pszManifestFile)

RTStrFree(paFiles[i].pszManifestFile);

if (paFiles[i].pszManifestDigest)

RTStrFree(paFiles[i].pszManifestDigest);

}

RTMemTmpFree(paFiles);

return rc;

}

RTR3DECL(int) RTManifestWriteFilesBuf(void **ppvBuf, size_t *pcbSize, PRTMANIFESTTEST paFiles, size_t cFiles)

{

/* Validate input */

AssertPtrReturn(ppvBuf, VERR_INVALID_POINTER);

AssertPtrReturn(pcbSize, VERR_INVALID_POINTER);

AssertPtrReturn(paFiles, 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(paFiles[i].pszTestFile)) + strlen(paFiles[i].pszTestDigest) + 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(paFiles[i].pszTestFile), paFiles[i].pszTestDigest);

memcpy(&((char*)pvBuf)[cbPos], pszTmp, cch);

cbPos += cch;

}

RTStrFree(pszTmp);

/* Results */

*ppvBuf = pvBuf;

*pcbSize = cbSize;

return VINF_SUCCESS;

}