/* $Id: manifest3.cpp 39439 2011-11-28 13:56:13Z vboxsync $ */ /** @file * IPRT - Manifest, the bits with the most dependencies. */ /* * Copyright (C) 2010 Oracle Corporation * * This file is part of VirtualBox Open Source Edition (OSE), as * available from http://www.virtualbox.org. This file is free software; * you can redistribute it and/or modify it under the terms of the GNU * General Public License (GPL) as published by the Free Software * Foundation, in version 2 as it comes in the "COPYING" file of the * VirtualBox OSE distribution. VirtualBox OSE is distributed in the * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. * * The contents of this file may alternatively be used under the terms * of the Common Development and Distribution License Version 1.0 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the * VirtualBox OSE distribution, in which case the provisions of the * CDDL are applicable instead of those of the GPL. * * You may elect to license modified versions of this file under the * terms and conditions of either the GPL or the CDDL or both. */ /******************************************************************************* * Header Files * *******************************************************************************/ #include "internal/iprt.h" #include #include #include #include #include #include #include #include #include #include #include /******************************************************************************* * Structures and Typedefs * *******************************************************************************/ /** * Hashes data. * * Used when hashing a file, stream or similar. */ typedef struct RTMANIFESTHASHES { /** The desired attribute types. * Only the hashes indicated by this will be calculated. */ uint32_t fAttrs; /** The size. */ RTFOFF cbStream; /** The MD5 context. */ RTMD5CONTEXT Md5Ctx; /** The SHA-1 context. */ RTSHA1CONTEXT Sha1Ctx; /** The SHA-256 context. */ RTSHA256CONTEXT Sha256Ctx; /** The SHA-512 context. */ RTSHA512CONTEXT Sha512Ctx; /** The MD5 digest. */ uint8_t abMd5Digest[RTMD5_HASH_SIZE]; /** The SHA-1 digest. */ uint8_t abSha1Digest[RTSHA1_HASH_SIZE]; /** The SHA-256 digest. */ uint8_t abSha256Digest[RTSHA256_HASH_SIZE]; /** The SHA-512 digest. */ uint8_t abSha512Digest[RTSHA512_HASH_SIZE]; } RTMANIFESTHASHES; /** Pointer to a the hashes for a stream. */ typedef RTMANIFESTHASHES *PRTMANIFESTHASHES; /** * The internal data of a manifest passthru I/O stream. */ typedef struct RTMANIFESTPTIOS { /** The stream we're reading from or writing to. */ RTVFSIOSTREAM hVfsIos; /** The hashes. */ PRTMANIFESTHASHES pHashes; /** Whether we're reading or writing. */ bool fReadOrWrite; /** Whether we've already added the entry to the manifest. */ bool fAddedEntry; /** The entry name. */ char *pszEntry; /** The manifest to add the entry to. */ RTMANIFEST hManifest; } RTMANIFESTPTIOS; /** Pointer to a the internal data of a manifest passthru I/O stream. */ typedef RTMANIFESTPTIOS *PRTMANIFESTPTIOS; /** * Creates a hashes structure. * * @returns Pointer to a hashes structure. * @param fAttrs The desired hashes, RTMANIFEST_ATTR_XXX. */ static PRTMANIFESTHASHES rtManifestHashesCreate(uint32_t fAttrs) { PRTMANIFESTHASHES pHashes = (PRTMANIFESTHASHES)RTMemTmpAllocZ(sizeof(*pHashes)); if (pHashes) { pHashes->fAttrs = fAttrs; /*pHashes->cbStream = 0;*/ if (fAttrs & RTMANIFEST_ATTR_MD5) RTMd5Init(&pHashes->Md5Ctx); if (fAttrs & RTMANIFEST_ATTR_SHA1) RTSha1Init(&pHashes->Sha1Ctx); if (fAttrs & RTMANIFEST_ATTR_SHA256) RTSha256Init(&pHashes->Sha256Ctx); if (fAttrs & RTMANIFEST_ATTR_SHA512) RTSha512Init(&pHashes->Sha512Ctx); } return pHashes; } /** * Updates the hashes with a block of data. * * @param pHashes The hashes structure. * @param pvBuf The data block. * @param cbBuf The size of the data block. */ static void rtManifestHashesUpdate(PRTMANIFESTHASHES pHashes, void const *pvBuf, size_t cbBuf) { pHashes->cbStream += cbBuf; if (pHashes->fAttrs & RTMANIFEST_ATTR_MD5) RTMd5Update(&pHashes->Md5Ctx, pvBuf, cbBuf); if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA1) RTSha1Update(&pHashes->Sha1Ctx, pvBuf, cbBuf); if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA256) RTSha256Update(&pHashes->Sha256Ctx, pvBuf, cbBuf); if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA512) RTSha512Update(&pHashes->Sha512Ctx, pvBuf, cbBuf); } /** * Finalizes all the hashes. * * @param pHashes The hashes structure. */ static void rtManifestHashesFinal(PRTMANIFESTHASHES pHashes) { if (pHashes->fAttrs & RTMANIFEST_ATTR_MD5) RTMd5Final(pHashes->abMd5Digest, &pHashes->Md5Ctx); if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA1) RTSha1Final(&pHashes->Sha1Ctx, pHashes->abSha1Digest); if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA256) RTSha256Final(&pHashes->Sha256Ctx, pHashes->abSha256Digest); if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA512) RTSha512Final(&pHashes->Sha512Ctx, pHashes->abSha512Digest); } /** * Adds the hashes to a manifest entry. * * @returns IPRT status code. * @param pHashes The hashes structure. * @param hManifest The manifest to add them to. * @param pszEntry The entry name. */ static int rtManifestHashesSetAttrs(PRTMANIFESTHASHES pHashes, RTMANIFEST hManifest, const char *pszEntry) { char szValue[RTSHA512_DIGEST_LEN + 8]; int rc = VINF_SUCCESS; int rc2; if (pHashes->fAttrs & RTMANIFEST_ATTR_SIZE) { RTStrPrintf(szValue, sizeof(szValue), "%RU64", (uint64_t)pHashes->cbStream); rc2 = RTManifestEntrySetAttr(hManifest, pszEntry, "SIZE", szValue, RTMANIFEST_ATTR_SIZE); if (RT_FAILURE(rc2) && RT_SUCCESS(rc)) rc = rc2; } if (pHashes->fAttrs & RTMANIFEST_ATTR_MD5) { rc2 = RTMd5ToString(pHashes->abMd5Digest, szValue, sizeof(szValue)); if (RT_SUCCESS(rc2)) rc2 = RTManifestEntrySetAttr(hManifest, pszEntry, "MD5", szValue, RTMANIFEST_ATTR_MD5); if (RT_FAILURE(rc2) && RT_SUCCESS(rc)) rc = rc2; } if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA1) { rc2 = RTSha1ToString(pHashes->abSha1Digest, szValue, sizeof(szValue)); if (RT_SUCCESS(rc2)) rc2 = RTManifestEntrySetAttr(hManifest, pszEntry, "SHA1", szValue, RTMANIFEST_ATTR_SHA1); if (RT_FAILURE(rc2) && RT_SUCCESS(rc)) rc = rc2; } if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA256) { rc2 = RTSha256ToString(pHashes->abSha256Digest, szValue, sizeof(szValue)); if (RT_SUCCESS(rc2)) rc2 = RTManifestEntrySetAttr(hManifest, pszEntry, "SHA256", szValue, RTMANIFEST_ATTR_SHA256); if (RT_FAILURE(rc2) && RT_SUCCESS(rc)) rc = rc2; } if (pHashes->fAttrs & RTMANIFEST_ATTR_SHA512) { rc2 = RTSha512ToString(pHashes->abSha512Digest, szValue, sizeof(szValue)); if (RT_SUCCESS(rc2)) rc2 = RTManifestEntrySetAttr(hManifest, pszEntry, "SHA512", szValue, RTMANIFEST_ATTR_SHA512); if (RT_FAILURE(rc2) && RT_SUCCESS(rc)) rc = rc2; } return rc; } /** * Destroys the hashes. * * @param pHashes The hashes structure. NULL is ignored. */ static void rtManifestHashesDestroy(PRTMANIFESTHASHES pHashes) { RTMemTmpFree(pHashes); } /* * * M a n i f e s t p a s s t h r u I / O s t r e a m * M a n i f e s t p a s s t h r u I / O s t r e a m * M a n i f e s t p a s s t h r u I / O s t r e a m * */ /** * @interface_method_impl{RTVFSOBJOPS,pfnClose} */ static DECLCALLBACK(int) rtManifestPtIos_Close(void *pvThis) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; int rc = VINF_SUCCESS; if (!pThis->fAddedEntry) { rtManifestHashesFinal(pThis->pHashes); rc = rtManifestHashesSetAttrs(pThis->pHashes, pThis->hManifest, pThis->pszEntry); } RTVfsIoStrmRelease(pThis->hVfsIos); pThis->hVfsIos = NIL_RTVFSIOSTREAM; rtManifestHashesDestroy(pThis->pHashes); pThis->pHashes = NULL; RTStrFree(pThis->pszEntry); pThis->pszEntry = NULL; RTManifestRelease(pThis->hManifest); pThis->hManifest = NIL_RTMANIFEST; return rc; } /** * @interface_method_impl{RTVFSOBJOPS,pfnQueryInfo} */ static DECLCALLBACK(int) rtManifestPtIos_QueryInfo(void *pvThis, PRTFSOBJINFO pObjInfo, RTFSOBJATTRADD enmAddAttr) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; return RTVfsIoStrmQueryInfo(pThis->hVfsIos, pObjInfo, enmAddAttr); } /** * Updates the hashes with a scather/gather buffer. * * @param pThis The passthru I/O stream instance data. * @param pSgBuf The scather/gather buffer. * @param cbLeft The number of bytes to take from the buffer. */ static void rtManifestPtIos_UpdateHashes(PRTMANIFESTPTIOS pThis, PCRTSGBUF pSgBuf, size_t cbLeft) { for (uint32_t iSeg = 0; iSeg < pSgBuf->cSegs; iSeg++) { size_t cbSeg = pSgBuf->paSegs[iSeg].cbSeg; if (cbSeg > cbLeft) cbSeg = cbLeft; rtManifestHashesUpdate(pThis->pHashes, pSgBuf->paSegs[iSeg].pvSeg, cbSeg); cbLeft -= cbSeg; if (!cbLeft) break; } } /** * @interface_method_impl{RTVFSIOSTREAMOPS,pfnRead} */ static DECLCALLBACK(int) rtManifestPtIos_Read(void *pvThis, RTFOFF off, PCRTSGBUF pSgBuf, bool fBlocking, size_t *pcbRead) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; int rc = RTVfsIoStrmSgRead(pThis->hVfsIos, pSgBuf, fBlocking, pcbRead); if (RT_SUCCESS(rc)) rtManifestPtIos_UpdateHashes(pThis, pSgBuf, pcbRead ? *pcbRead : ~(size_t)0); Assert(off == -1); NOREF(off); return rc; } /** * @interface_method_impl{RTVFSIOSTREAMOPS,pfnWrite} */ static DECLCALLBACK(int) rtManifestPtIos_Write(void *pvThis, RTFOFF off, PCRTSGBUF pSgBuf, bool fBlocking, size_t *pcbWritten) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; int rc = RTVfsIoStrmSgWrite(pThis->hVfsIos, pSgBuf, fBlocking, pcbWritten); if (RT_SUCCESS(rc)) rtManifestPtIos_UpdateHashes(pThis, pSgBuf, pcbWritten ? *pcbWritten : ~(size_t)0); Assert(off == -1); NOREF(off); return rc; } /** * @interface_method_impl{RTVFSIOSTREAMOPS,pfnFlush} */ static DECLCALLBACK(int) rtManifestPtIos_Flush(void *pvThis) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; return RTVfsIoStrmFlush(pThis->hVfsIos); } /** * @interface_method_impl{RTVFSIOSTREAMOPS,pfnPollOne} */ static DECLCALLBACK(int) rtManifestPtIos_PollOne(void *pvThis, uint32_t fEvents, RTMSINTERVAL cMillies, bool fIntr, uint32_t *pfRetEvents) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; return RTVfsIoStrmPoll(pThis->hVfsIos, fEvents, cMillies, fIntr, pfRetEvents); } /** * @interface_method_impl{RTVFSIOSTREAMOPS,pfnTell} */ static DECLCALLBACK(int) rtManifestPtIos_Tell(void *pvThis, PRTFOFF poffActual) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis; RTFOFF off = RTVfsIoStrmTell(pThis->hVfsIos); if (off < 0) return (int)off; *poffActual = off; return VINF_SUCCESS; } /** * The manifest passthru I/O stream vtable. */ static RTVFSIOSTREAMOPS g_rtManifestPassthruIosOps = { { /* Obj */ RTVFSOBJOPS_VERSION, RTVFSOBJTYPE_IO_STREAM, "manifest passthru I/O stream", rtManifestPtIos_Close, rtManifestPtIos_QueryInfo, RTVFSOBJOPS_VERSION }, RTVFSIOSTREAMOPS_VERSION, 0, rtManifestPtIos_Read, rtManifestPtIos_Write, rtManifestPtIos_Flush, rtManifestPtIos_PollOne, rtManifestPtIos_Tell, NULL /* Skip */, NULL /* ZeroFill */, RTVFSIOSTREAMOPS_VERSION, }; /** * Add an entry for an I/O stream using a passthru stream. * * The passthru I/O stream will hash all the data read from or written to the * stream and automatically add an entry to the manifest with the desired * attributes when it is released. Alternatively one can call * RTManifestPtIosAddEntryNow() to have more control over exactly when this * action is performed and which status it yields. * * @returns IPRT status code. * @param hManifest The manifest to add the entry to. * @param hVfsIos The I/O stream to pass thru to/from. * @param pszEntry The entry name. * @param fAttrs The attributes to create for this stream. * @param fReadOrWrite Whether it's a read or write I/O stream. * @param phVfsIosPassthru Where to return the new handle. */ RTDECL(int) RTManifestEntryAddPassthruIoStream(RTMANIFEST hManifest, RTVFSIOSTREAM hVfsIos, const char *pszEntry, uint32_t fAttrs, bool fReadOrWrite, PRTVFSIOSTREAM phVfsIosPassthru) { /* * Validate input. */ AssertReturn(fAttrs < RTMANIFEST_ATTR_END, VERR_INVALID_PARAMETER); AssertPtr(pszEntry); AssertPtr(phVfsIosPassthru); uint32_t cRefs = RTManifestRetain(hManifest); AssertReturn(cRefs != UINT32_MAX, VERR_INVALID_HANDLE); cRefs = RTVfsIoStrmRetain(hVfsIos); AssertReturnStmt(cRefs != UINT32_MAX, VERR_INVALID_HANDLE, RTManifestRelease(hManifest)); /* * Create an instace of the passthru I/O stream. */ PRTMANIFESTPTIOS pThis; RTVFSIOSTREAM hVfsPtIos; int rc = RTVfsNewIoStream(&g_rtManifestPassthruIosOps, sizeof(*pThis), fReadOrWrite ? RTFILE_O_READ : RTFILE_O_WRITE, NIL_RTVFS, NIL_RTVFSLOCK, &hVfsPtIos, (void **)&pThis); if (RT_SUCCESS(rc)) { pThis->hVfsIos = hVfsIos; pThis->pHashes = rtManifestHashesCreate(fAttrs); pThis->hManifest = hManifest; pThis->fReadOrWrite = fReadOrWrite; pThis->fAddedEntry = false; pThis->pszEntry = RTStrDup(pszEntry); if (pThis->pszEntry && pThis->pHashes) { *phVfsIosPassthru = hVfsPtIos; return VINF_SUCCESS; } RTVfsIoStrmRelease(hVfsPtIos); } else { RTVfsIoStrmRelease(hVfsIos); RTManifestRelease(hManifest); } return rc; } /** * Adds the entry to the manifest right now. * * @returns IPRT status code. * @param hVfsPtIos The manifest passthru I/O stream returned by * RTManifestEntryAddPassthruIoStream(). */ RTDECL(int) RTManifestPtIosAddEntryNow(RTVFSIOSTREAM hVfsPtIos) { PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)RTVfsIoStreamToPrivate(hVfsPtIos, &g_rtManifestPassthruIosOps); AssertReturn(pThis, VERR_INVALID_HANDLE); AssertReturn(!pThis->fAddedEntry, VERR_WRONG_ORDER); pThis->fAddedEntry = true; rtManifestHashesFinal(pThis->pHashes); return rtManifestHashesSetAttrs(pThis->pHashes, pThis->hManifest, pThis->pszEntry); } /** * Adds an entry for a file with the specified set of attributes. * * @returns IPRT status code. * * @param hManifest The manifest handle. * @param hVfsIos The I/O stream handle of the entry. This will * be processed to its end on successful return. * (Must be positioned at the start to get * the expected results.) * @param pszEntry The entry name. * @param fAttrs The attributes to create for this stream. */ RTDECL(int) RTManifestEntryAddIoStream(RTMANIFEST hManifest, RTVFSIOSTREAM hVfsIos, const char *pszEntry, uint32_t fAttrs) { /* * Note! This is a convenicence function, so just use the available public * methods to get the job done. */ AssertReturn(fAttrs < RTMANIFEST_ATTR_END, VERR_INVALID_PARAMETER); AssertPtr(pszEntry); /* * Allocate and initialize the hash contexts, hash digests and I/O buffer. */ PRTMANIFESTHASHES pHashes = rtManifestHashesCreate(fAttrs); if (!pHashes) return VERR_NO_TMP_MEMORY; int rc; size_t cbBuf = _1M; void *pvBuf = RTMemTmpAlloc(cbBuf); if (RT_UNLIKELY(!pvBuf)) { cbBuf = _4K; pvBuf = RTMemTmpAlloc(cbBuf); } if (RT_LIKELY(pvBuf)) { /* * Process the stream data. */ for (;;) { size_t cbRead; rc = RTVfsIoStrmRead(hVfsIos, pvBuf, cbBuf, true /*fBlocking*/, &cbRead); if ( (rc == VINF_EOF && cbRead == 0) || RT_FAILURE(rc)) break; rtManifestHashesUpdate(pHashes, pvBuf, cbRead); } RTMemTmpFree(pvBuf); if (RT_SUCCESS(rc)) { /* * Add the entry with the finalized hashes. */ rtManifestHashesFinal(pHashes); rc = RTManifestEntryAdd(hManifest, pszEntry); if (RT_SUCCESS(rc)) rc = rtManifestHashesSetAttrs(pHashes, hManifest, pszEntry); } } else { rtManifestHashesDestroy(pHashes); rc = VERR_NO_TMP_MEMORY; } return rc; }