Imported from /home/lorry/working-area/delta_VirtualBox/VirtualBox-4.3.10.tar.bz2.HEADVirtualBox-4.3.10master

1 files changed, 201 insertions, 0 deletions

diff --git a/src/VBox/Runtime/common/checksum/RTSha256Digest.cpp b/src/VBox/Runtime/common/checksum/RTSha256Digest.cpp
new file mode 100644
index 00000000..5b8679c4
--- /dev/null
+++ b/src/VBox/Runtime/common/checksum/RTSha256Digest.cpp
@@ -0,0 +1,201 @@
+/** @file
+ * IPRT - SHA256 digest creation
+ */
+
+/*
+ * Copyright (C) 2009-2013 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 <iprt/sha.h>
+
+#include <iprt/alloca.h>
+#include <iprt/assert.h>
+#include <iprt/mem.h>
+#include <iprt/string.h>
+#include <iprt/file.h>
+
+#include <openssl/sha.h>
+
+
+RTR3DECL(int) RTSha256Digest(void* pvBuf, size_t cbBuf, char **ppszDigest, PFNRTPROGRESS pfnProgressCallback, void *pvUser)
+{
+    /* Validate input */
+    AssertPtrReturn(pvBuf, VERR_INVALID_POINTER);
+    AssertPtrReturn(ppszDigest, VERR_INVALID_POINTER);
+    AssertPtrNullReturn(pfnProgressCallback, VERR_INVALID_PARAMETER);
+
+    int rc = VINF_SUCCESS;
+    *ppszDigest = NULL;
+
+    /* Initialize OpenSSL. */
+    SHA256_CTX ctx;
+    if (!SHA256_Init(&ctx))
+        return VERR_INTERNAL_ERROR;
+
+    /* Buffer size for progress callback */
+    double rdMulti = 100.0 / cbBuf;
+
+    /* Working buffer */
+    char *pvTmp = (char*)pvBuf;
+
+    /* Process the memory in blocks */
+    size_t cbRead;
+    size_t cbReadTotal = 0;
+    for (;;)
+    {
+        cbRead = RT_MIN(cbBuf - cbReadTotal, _1M);
+        if(!SHA256_Update(&ctx, pvTmp, cbRead))
+        {
+            rc = VERR_INTERNAL_ERROR;
+            break;
+        }
+        cbReadTotal += cbRead;
+        pvTmp += cbRead;
+
+        /* Call the progress callback if one is defined */
+        if (pfnProgressCallback)
+        {
+            rc = pfnProgressCallback((unsigned)(cbReadTotal * rdMulti), pvUser);
+            if (RT_FAILURE(rc))
+                break; /* canceled */
+        }
+        /* Finished? */
+        if (cbReadTotal == cbBuf)
+            break;
+    }
+    if (RT_SUCCESS(rc))
+    {
+        /* Finally calculate & format the SHA256 sum */
+        unsigned char auchDig[RTSHA256_HASH_SIZE];
+        if (!SHA256_Final(auchDig, &ctx))
+            return VERR_INTERNAL_ERROR;
+
+        char *pszDigest;
+        rc = RTStrAllocEx(&pszDigest, RTSHA256_DIGEST_LEN + 1);
+        if (RT_SUCCESS(rc))
+        {
+            rc = RTSha256ToString(auchDig, pszDigest, RTSHA256_DIGEST_LEN + 1);
+            if (RT_SUCCESS(rc))
+                *ppszDigest = pszDigest;
+            else
+                RTStrFree(pszDigest);
+        }
+    }
+
+    return rc;
+}
+
+RTR3DECL(int) RTSha256DigestFromFile(const char *pszFile, char **ppszDigest, PFNRTPROGRESS pfnProgressCallback, void *pvUser)
+{
+    /* Validate input */
+    AssertPtrReturn(pszFile, VERR_INVALID_POINTER);
+    AssertPtrReturn(ppszDigest, VERR_INVALID_POINTER);
+    AssertPtrNullReturn(pfnProgressCallback, VERR_INVALID_PARAMETER);
+
+    *ppszDigest = NULL;
+
+    /* Initialize OpenSSL. */
+    SHA256_CTX ctx;
+    if (!SHA256_Init(&ctx))
+        return VERR_INTERNAL_ERROR;
+
+    /* Open the file to calculate a SHA256 sum of */
+    RTFILE hFile;
+    int rc = RTFileOpen(&hFile, pszFile, RTFILE_O_OPEN | RTFILE_O_READ | RTFILE_O_DENY_WRITE);
+    if (RT_FAILURE(rc))
+        return rc;
+
+    /* Fetch the file size. Only needed if there is a progress callback. */
+    double rdMulti = 0;
+    if (pfnProgressCallback)
+    {
+        uint64_t cbFile;
+        rc = RTFileGetSize(hFile, &cbFile);
+        if (RT_FAILURE(rc))
+        {
+            RTFileClose(hFile);
+            return rc;
+        }
+        rdMulti = 100.0 / cbFile;
+    }
+
+    /* Allocate a reasonably large buffer, fall back on a tiny one. */
+    void  *pvBufFree;
+    size_t cbBuf = _1M;
+    void  *pvBuf = pvBufFree = RTMemTmpAlloc(cbBuf);
+    if (!pvBuf)
+    {
+        cbBuf = 0x1000;
+        pvBuf = alloca(cbBuf);
+    }
+
+    /* Read that file in blocks */
+    size_t cbRead;
+    size_t cbReadTotal = 0;
+    for (;;)
+    {
+        rc = RTFileRead(hFile, pvBuf, cbBuf, &cbRead);
+        if (RT_FAILURE(rc) || !cbRead)
+            break;
+        if(!SHA256_Update(&ctx, pvBuf, cbRead))
+        {
+            rc = VERR_INTERNAL_ERROR;
+            break;
+        }
+        cbReadTotal += cbRead;
+
+        /* Call the progress callback if one is defined */
+        if (pfnProgressCallback)
+        {
+            rc = pfnProgressCallback((unsigned)(cbReadTotal * rdMulti), pvUser);
+            if (RT_FAILURE(rc))
+                break; /* canceled */
+        }
+    }
+    RTMemTmpFree(pvBufFree);
+    RTFileClose(hFile);
+
+    if (RT_FAILURE(rc))
+        return rc;
+
+    /* Finally calculate & format the SHA256 sum */
+    unsigned char auchDig[RTSHA256_HASH_SIZE];
+    if (!SHA256_Final(auchDig, &ctx))
+        return VERR_INTERNAL_ERROR;
+
+    char *pszDigest;
+    rc = RTStrAllocEx(&pszDigest, RTSHA256_DIGEST_LEN + 1);
+    if (RT_SUCCESS(rc))
+    {
+        rc = RTSha256ToString(auchDig, pszDigest, RTSHA256_DIGEST_LEN + 1);
+        if (RT_SUCCESS(rc))
+            *ppszDigest = pszDigest;
+        else
+            RTStrFree(pszDigest);
+    }
+
+    return rc;
+}
+