diff options
Diffstat (limited to 'src/VBox/Runtime/generic/RTFileCopyByHandlesEx-generic.cpp')
| -rw-r--r-- | src/VBox/Runtime/generic/RTFileCopyByHandlesEx-generic.cpp | 161 |
1 files changed, 161 insertions, 0 deletions
diff --git a/src/VBox/Runtime/generic/RTFileCopyByHandlesEx-generic.cpp b/src/VBox/Runtime/generic/RTFileCopyByHandlesEx-generic.cpp new file mode 100644 index 00000000..12f40c28 --- /dev/null +++ b/src/VBox/Runtime/generic/RTFileCopyByHandlesEx-generic.cpp @@ -0,0 +1,161 @@ +/* $Id: RTFileCopyByHandlesEx-generic.cpp $ */ +/** @file + * IPRT - RTFileCopyByHandlesEx, generic implementation. + */ + +/* + * Copyright (C) 2006-2020 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/file.h> + +#include <iprt/alloca.h> +#include <iprt/assert.h> +#include <iprt/mem.h> +#include <iprt/errcore.h> + + +RTDECL(int) RTFileCopyByHandlesEx(RTFILE hFileSrc, RTFILE hFileDst, PFNRTPROGRESS pfnProgress, void *pvUser) +{ + /* + * Validate input. + */ + AssertMsgReturn(RTFileIsValid(hFileSrc), ("hFileSrc=%RTfile\n", hFileSrc), VERR_INVALID_PARAMETER); + AssertMsgReturn(RTFileIsValid(hFileDst), ("hFileDst=%RTfile\n", hFileDst), VERR_INVALID_PARAMETER); + AssertMsgReturn(!pfnProgress || VALID_PTR(pfnProgress), ("pfnProgress=%p\n", pfnProgress), VERR_INVALID_PARAMETER); + + /* + * Save file offset. + */ + uint64_t offSrcSaved; + int rc = RTFileSeek(hFileSrc, 0, RTFILE_SEEK_CURRENT, &offSrcSaved); + if (RT_FAILURE(rc)) + return rc; + + /* + * Get the file size and figure out how much we'll copy at a time. + */ + uint64_t cbSrc; + rc = RTFileQuerySize(hFileSrc, &cbSrc); + if (RT_FAILURE(rc)) + return rc; + + uint64_t cbChunk = cbSrc; + if (pfnProgress && cbSrc > _1M) + { + cbChunk /= 100; + if (cbChunk > _64M) + cbChunk = RT_ALIGN_64(cbChunk, _2M); + else + cbChunk = RT_ALIGN_64(cbChunk, _128K); + } + + /* + * Prepare buffers. + */ + RTFILECOPYPARTBUFSTATE BufState; + rc = RTFileCopyPartPrep(&BufState, cbChunk); + if (RT_SUCCESS(rc)) + { + /* + * Prepare the destination file. + */ + uint64_t cbDst; + rc = RTFileQuerySize(hFileDst, &cbDst); + if (RT_SUCCESS(rc) && cbDst > cbSrc) + rc = RTFileSetSize(hFileDst, cbSrc); + if (RT_SUCCESS(rc) && cbDst < cbSrc) + { + rc = RTFileSetAllocationSize(hFileDst, cbSrc, RTFILE_ALLOC_SIZE_F_DEFAULT); + if (rc == VERR_NOT_SUPPORTED) + rc = RTFileSetSize(hFileDst, cbSrc); + } + if (RT_SUCCESS(rc)) + { + /* + * Copy loop that works till we reach EOF. + */ + RTFOFF off = 0; + RTFOFF cbPercent = cbSrc / 100; + RTFOFF offNextPercent = pfnProgress ? cbPercent : RTFOFF_MAX; + unsigned uPercentage = pfnProgress ? 0 : 100; + for (;;) + { + /* + * Copy a block. + */ + uint64_t cbCopied = 0; + rc = RTFileCopyPartEx(hFileSrc, off, hFileDst, off, cbChunk, 0 /*fFlags*/, &BufState, &cbCopied); + if (RT_FAILURE(rc)) + break; + if (cbCopied == 0) + { + /* + * We reached the EOF. Complete the copy operation. + */ + rc = RTFileSetSize(hFileDst, off); + if (RT_SUCCESS(rc)) + rc = RTFileCopyAttributes(hFileSrc, hFileDst, 0); + break; + } + + /* + * Advance and work the progress callback. + */ + off += cbCopied; + if ( off >= offNextPercent + && pfnProgress + && uPercentage < 99) + { + do + { + uPercentage++; + offNextPercent += cbPercent; + } while ( offNextPercent <= off + && uPercentage < 99); + rc = pfnProgress(uPercentage, pvUser); + if (RT_FAILURE(rc)) + break; + } + } + } + + RTFileCopyPartCleanup(&BufState); + + /* + * 100%. + */ + if ( pfnProgress + && RT_SUCCESS(rc)) + rc = pfnProgress(100, pvUser); + } + + /* + * Restore source position. + */ + RTFileSeek(hFileSrc, offSrcSaved, RTFILE_SEEK_BEGIN, NULL); + return rc; +} + |