Storage/testcase/VDIoRnd.cpp

470122d12fa43f83f993ae8e93ba8e3cf4c674cdvboxsync/* $Id$ */
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync/** @file
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * VBox HDD container test utility - I/O data generator.
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync */
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync/*
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * Copyright (C) 2011 Oracle Corporation
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync *
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * available from http://www.virtualbox.org. This file is free software;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * you can redistribute it and/or modify it under the terms of the GNU
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * General Public License (GPL) as published by the Free Software
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync */
470122d12fa43f83f993ae8e93ba8e3cf4c674cdvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync#define LOGGROUP LOGGROUP_DEFAULT
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync#include <iprt/log.h>
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync#include <iprt/err.h>
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync#include <iprt/mem.h>
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync#include <iprt/rand.h>
470122d12fa43f83f993ae8e93ba8e3cf4c674cdvboxsync#include <iprt/assert.h>
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync#include "VDIoRnd.h"
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync/**
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync * I/O random data generator instance data.
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync */
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsynctypedef struct VDIORND
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync{
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    /** Pointer to the buffer holding the random data. */
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    uint8_t *pbPattern;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    /** Size of the buffer. */
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    size_t   cbPattern;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    /** RNG */
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    RTRAND   hRand;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync} VDIORND;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsyncint VDIoRndCreate(PPVDIORND ppIoRnd, size_t cbPattern, uint64_t uSeed)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync{
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    PVDIORND pIoRnd = NULL;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    int rc = VINF_SUCCESS;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    AssertPtrReturn(ppIoRnd, VERR_INVALID_POINTER);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    pIoRnd = (PVDIORND)RTMemAllocZ(sizeof(VDIORND));
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    if (pIoRnd)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        pIoRnd->pbPattern = (uint8_t *)RTMemPageAllocZ(cbPattern);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    if (   pIoRnd
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        && pIoRnd->pbPattern)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    {
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        pIoRnd->cbPattern = cbPattern;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        rc = RTRandAdvCreateParkMiller(&pIoRnd->hRand);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        if (RT_SUCCESS(rc))
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        {
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync            RTRandAdvSeed(pIoRnd->hRand, uSeed);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync            RTRandAdvBytes(pIoRnd->hRand, pIoRnd->pbPattern, cbPattern);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        }
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        else
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        {
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync            RTMemPageFree(pIoRnd->pbPattern, cbPattern);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync            RTMemFree(pIoRnd);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        }
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    }
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    else
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    {
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        if (pIoRnd)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync            RTMemFree(pIoRnd);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        rc = VERR_NO_MEMORY;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    }
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    if (RT_SUCCESS(rc))
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        *ppIoRnd = pIoRnd;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    return rc;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync}
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsyncvoid VDIoRndDestroy(PVDIORND pIoRnd)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync{
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    AssertPtrReturnVoid(pIoRnd);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    RTRandAdvDestroy(pIoRnd->hRand);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    RTMemPageFree(pIoRnd->pbPattern, pIoRnd->cbPattern);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    RTMemFree(pIoRnd);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync}
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsyncint VDIoRndGetBuffer(PVDIORND pIoRnd, void **ppv, size_t cb)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync{
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    AssertPtrReturn(pIoRnd, VERR_INVALID_POINTER);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    AssertPtrReturn(ppv, VERR_INVALID_POINTER);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    AssertReturn(cb > 0, VERR_INVALID_PARAMETER);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    if (cb > pIoRnd->cbPattern - 512)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync        return VERR_INVALID_PARAMETER;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    *ppv = pIoRnd->pbPattern + RT_ALIGN_64(RTRandAdvU64Ex(pIoRnd->hRand, 0, pIoRnd->cbPattern - cb - 512), 512);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    return VINF_SUCCESS;
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync}
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsyncuint32_t VDIoRndGetU32Ex(PVDIORND pIoRnd, uint32_t uMin, uint32_t uMax)
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync{
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync    return RTRandAdvU32Ex(pIoRnd->hRand, uMin, uMax);
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync}
39cac49f62afd27b0ffdbb779c7c64211939030cvboxsync