manifest.cpp revision 5c90ca78888c7bb1632ef4c131cc668b03227271
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync/* $Id$ */
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync/** @file
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * IPRT - Manifest file handling.
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync */
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync/*
e64031e20c39650a7bc902a3e1aba613b9415deevboxsync * Copyright (C) 2009 Oracle Corporation
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync *
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * available from http://www.virtualbox.org. This file is free software;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * you can redistribute it and/or modify it under the terms of the GNU
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * General Public License (GPL) as published by the Free Software
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync *
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * The contents of this file may alternatively be used under the terms
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * of the Common Development and Distribution License Version 1.0
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * VirtualBox OSE distribution, in which case the provisions of the
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * CDDL are applicable instead of those of the GPL.
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync *
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * You may elect to license modified versions of this file under the
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * terms and conditions of either the GPL or the CDDL or both.
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync */
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
70dd77b2f7732defac9d929109f20d92676f0570vboxsync/*******************************************************************************
70dd77b2f7732defac9d929109f20d92676f0570vboxsync* Header Files *
70dd77b2f7732defac9d929109f20d92676f0570vboxsync*******************************************************************************/
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
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync/*******************************************************************************
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync* Structures and Typedefs *
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync*******************************************************************************/
70dd77b2f7732defac9d929109f20d92676f0570vboxsync/**
70dd77b2f7732defac9d929109f20d92676f0570vboxsync * Internal per file structure used by RTManifestVerify
70dd77b2f7732defac9d929109f20d92676f0570vboxsync */
70dd77b2f7732defac9d929109f20d92676f0570vboxsynctypedef struct RTMANIFESTFILEENTRY
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync{
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char *pszManifestFile;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync char *pszManifestDigest;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync PRTMANIFESTTEST pTestPattern;
70dd77b2f7732defac9d929109f20d92676f0570vboxsync} RTMANIFESTFILEENTRY;
70dd77b2f7732defac9d929109f20d92676f0570vboxsynctypedef RTMANIFESTFILEENTRY* PRTMANIFESTFILEENTRY;
70dd77b2f7732defac9d929109f20d92676f0570vboxsync
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync/**
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync * Internal structure used for the progress callback
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync */
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsynctypedef struct RTMANIFESTCALLBACKDATA
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync{
44918d7b9fb45f0d3e2ba56bf6d2cede9a2ae566vboxsync PFNRTMANIFESTPROGRESS pfnProgressCallback;
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync void *pvUser;
5c90ca78888c7bb1632ef4c131cc668b03227271vboxsync size_t cMaxFiles;
5c90ca78888c7bb1632ef4c131cc668b03227271vboxsync size_t cCurrentFile;
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync} RTMANIFESTCALLBACKDATA;
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsynctypedef RTMANIFESTCALLBACKDATA* PRTMANIFESTCALLBACKDATA;
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsyncint rtSHAProgressCallback(unsigned uPercent, void *pvUser)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync{
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync PRTMANIFESTCALLBACKDATA pData = (PRTMANIFESTCALLBACKDATA)pvUser;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync return pData->pfnProgressCallback((unsigned)((uPercent + (float)pData->cCurrentFile * 100.0) / (float)pData->cMaxFiles), pData->pvUser);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync}
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsyncRTR3DECL(int) RTManifestVerify(const char *pszManifestFile, PRTMANIFESTTEST paTests, size_t cTests, size_t *piFailed)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync{
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Validate input */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync AssertPtrReturn(pszManifestFile, VERR_INVALID_POINTER);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync AssertPtrReturn(paTests, VERR_INVALID_POINTER);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync AssertReturn(cTests > 0, VERR_INVALID_PARAMETER);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Open the manifest file */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync PRTSTREAM pStream;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync int rc = RTStrmOpen(pszManifestFile, "r", &pStream);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RT_FAILURE(rc))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync return rc;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync PRTMANIFESTFILEENTRY paFiles = (PRTMANIFESTFILEENTRY)RTMemTmpAllocZ(sizeof(RTMANIFESTFILEENTRY) * cTests);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!paFiles)
44918d7b9fb45f0d3e2ba56bf6d2cede9a2ae566vboxsync {
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync RTStrmClose(pStream);
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync return VERR_NO_MEMORY;
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync }
44918d7b9fb45f0d3e2ba56bf6d2cede9a2ae566vboxsync
44918d7b9fb45f0d3e2ba56bf6d2cede9a2ae566vboxsync /* Fill our compare list */
44918d7b9fb45f0d3e2ba56bf6d2cede9a2ae566vboxsync for (size_t i = 0; i < cTests; ++i)
510dd1bd989d25f95680fc2a3b51bf11854bc6a4vboxsync paFiles[i].pTestPattern = &paTests[i];
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Parse the manifest file line by line */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char szLine[1024];
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync for (;;)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
0aa6851583f4cb074ccc163d5b53948b9429ab77vboxsync rc = RTStrmGetLine(pStream, szLine, sizeof(szLine));
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc))
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync break;
70dd77b2f7732defac9d929109f20d92676f0570vboxsync size_t cch = strlen(szLine);
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Skip empty lines */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (cch == 0)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync continue;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /** @todo r=bird:
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync * -# The SHA1 test should probably include a blank space check.
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync * -# If there is a specific order to the elements in the string, it would be
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync * good if the delimiter searching checked for it.
70dd77b2f7732defac9d929109f20d92676f0570vboxsync * -# Deal with filenames containing delimiter characters.
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Check for the digest algorithm */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if ( cch < 4
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync || !( szLine[0] == 'S'
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync && szLine[1] == 'H'
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync && szLine[2] == 'A'
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync && szLine[3] == '1'))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Digest unsupported */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_UNSUPPORTED_DIGEST_TYPE;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Try to find the filename */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszNameStart = strchr(szLine, '(');
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!pszNameStart)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_WRONG_FILE_FORMAT;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszNameEnd = strchr(szLine, ')');
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!pszNameEnd)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_WRONG_FILE_FORMAT;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Copy the filename part */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync size_t cchName = pszNameEnd - pszNameStart - 1;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszName = (char *)RTMemTmpAlloc(cchName + 1);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!pszName)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_NO_MEMORY;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync memcpy(pszName, pszNameStart + 1, cchName);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync pszName[cchName] = '\0';
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Try to find the digest sum */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszDigestStart = strchr(szLine, '=');
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!pszDigestStart)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTMemTmpFree(pszName);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_WRONG_FILE_FORMAT;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszDigest = ++pszDigestStart;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Check our file list against the extracted data */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync bool fFound = false;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync for (size_t i = 0; i < cTests; ++i)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!RTStrCmp(RTPathFilename(paFiles[i].pTestPattern->pszTestFile), RTStrStrip(pszName)))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Add the data of the manifest file to the file list */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync paFiles[i].pszManifestFile = RTStrDup(RTStrStrip(pszName));
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync paFiles[i].pszManifestDigest = RTStrDup(RTStrStrip(pszDigest));
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync fFound = true;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTMemTmpFree(pszName);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!fFound)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* There have to be an entry in the file list */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_FILE_MISMATCH;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTStrmClose(pStream);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if ( rc == VINF_SUCCESS
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync || rc == VERR_EOF)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = VINF_SUCCESS;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync for (size_t i = 0; i < cTests; ++i)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* If there is an entry in the file list, which hasn't an
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync * equivalent in the manifest file, its an error. */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if ( !paFiles[i].pszManifestFile
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync || !paFiles[i].pszManifestDigest)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_FILE_MISMATCH;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Do the manifest SHA1 digest match against the actual digest? */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RTStrICmp(paFiles[i].pszManifestDigest, paFiles[i].pTestPattern->pszTestDigest))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (piFailed)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync *piFailed = i;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = VERR_MANIFEST_DIGEST_MISMATCH;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Cleanup */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync for (size_t i = 0; i < cTests; ++i)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (paFiles[i].pszManifestFile)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTStrFree(paFiles[i].pszManifestFile);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (paFiles[i].pszManifestDigest)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTStrFree(paFiles[i].pszManifestDigest);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTMemTmpFree(paFiles);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync return rc;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync}
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsyncRTR3DECL(int) RTManifestVerifyFiles(const char *pszManifestFile, const char * const *papszFiles, size_t cFiles, size_t *piFailed, PFNRTMANIFESTPROGRESS pfnProgressCallback, void *pvUser)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync{
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Validate input */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync AssertPtrReturn(pszManifestFile, VERR_INVALID_POINTER);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync AssertPtrReturn(papszFiles, VERR_INVALID_POINTER);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync AssertPtrNullReturn(pfnProgressCallback, VERR_INVALID_PARAMETER);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync int rc = VINF_SUCCESS;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Create our compare list */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync PRTMANIFESTTEST paFiles = (PRTMANIFESTTEST)RTMemTmpAllocZ(sizeof(RTMANIFESTTEST) * cFiles);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (!paFiles)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync return VERR_NO_MEMORY;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTMANIFESTCALLBACKDATA callback = { pfnProgressCallback, pvUser, cFiles, 0 };
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Fill our compare list */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync for (size_t i = 0; i < cFiles; ++i)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszDigest;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (pfnProgressCallback)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync callback.cCurrentFile = i;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = RTSha1Digest(papszFiles[i], &pszDigest, rtSHAProgressCallback, &callback);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync else
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = RTSha1Digest(papszFiles[i], &pszDigest, NULL, NULL);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RT_FAILURE(rc))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync paFiles[i].pszTestFile = (char*)papszFiles[i];
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync paFiles[i].pszTestDigest = pszDigest;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Do the verification */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RT_SUCCESS(rc))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = RTManifestVerify(pszManifestFile, paFiles, cFiles, piFailed);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Cleanup */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync for (size_t i = 0; i < cFiles; ++i)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (paFiles[i].pszTestDigest)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTStrFree(paFiles[i].pszTestDigest);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTMemTmpFree(paFiles);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync return rc;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync}
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsyncRTR3DECL(int) RTManifestWriteFiles(const char *pszManifestFile, const char * const *papszFiles, size_t cFiles, PFNRTMANIFESTPROGRESS pfnProgressCallback, void *pvUser)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync{
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Validate input */
70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrReturn(pszManifestFile, VERR_INVALID_POINTER);
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync AssertPtrReturn(papszFiles, VERR_INVALID_POINTER);
70dd77b2f7732defac9d929109f20d92676f0570vboxsync AssertPtrNullReturn(pfnProgressCallback, VERR_INVALID_PARAMETER);
70dd77b2f7732defac9d929109f20d92676f0570vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync /* Open a file to stream in */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync PRTSTREAM pStream;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync int rc = RTStrmOpen(pszManifestFile, "w", &pStream);
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc))
70dd77b2f7732defac9d929109f20d92676f0570vboxsync return rc;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTMANIFESTCALLBACKDATA callback = { pfnProgressCallback, pvUser, cFiles, 0 };
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync for (size_t i = 0; i < cFiles; ++i)
70dd77b2f7732defac9d929109f20d92676f0570vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Calculate the SHA1 digest of every file */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync char *pszDigest;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (pfnProgressCallback)
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync {
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync callback.cCurrentFile = i;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = RTSha1Digest(papszFiles[i], &pszDigest, rtSHAProgressCallback, &callback);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync }
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync else
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync rc = RTSha1Digest(papszFiles[i], &pszDigest, NULL, NULL);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RT_FAILURE(rc))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync break;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Add the entry to the manifest file */
70dd77b2f7732defac9d929109f20d92676f0570vboxsync int cch = RTStrmPrintf(pStream, "SHA1 (%s)= %s\n", RTPathFilename(papszFiles[i]), pszDigest);
70dd77b2f7732defac9d929109f20d92676f0570vboxsync RTStrFree(pszDigest);
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RT_UNLIKELY(cch < 0))
70dd77b2f7732defac9d929109f20d92676f0570vboxsync {
70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc = VERR_INTERNAL_ERROR;
70dd77b2f7732defac9d929109f20d92676f0570vboxsync break;
70dd77b2f7732defac9d929109f20d92676f0570vboxsync }
70dd77b2f7732defac9d929109f20d92676f0570vboxsync }
70dd77b2f7732defac9d929109f20d92676f0570vboxsync int rc2 = RTStrmClose(pStream);
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync if (RT_FAILURE(rc2) && RT_SUCCESS(rc))
70dd77b2f7732defac9d929109f20d92676f0570vboxsync rc2 = rc;
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync /* Delete the manifest file on failure */
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync if (RT_FAILURE(rc))
382dd16193dd5770e53a99d3e3f3bc4e96f1ddd8vboxsync RTFileDelete(pszManifestFile);
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync return rc;
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync}
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync
c40afa339b10a23d6fffcbeb7d4572bb494685f0vboxsync