diff options
Diffstat (limited to 'src/VBox/Main/src-all/ProgressImpl.cpp')
| -rw-r--r-- | src/VBox/Main/src-all/ProgressImpl.cpp | 1215 |
1 files changed, 1215 insertions, 0 deletions
diff --git a/src/VBox/Main/src-all/ProgressImpl.cpp b/src/VBox/Main/src-all/ProgressImpl.cpp new file mode 100644 index 00000000..a4fc20f6 --- /dev/null +++ b/src/VBox/Main/src-all/ProgressImpl.cpp @@ -0,0 +1,1215 @@ +/* $Id: ProgressImpl.cpp $ */ +/** @file + * VirtualBox Progress COM class implementation + */ + +/* + * Copyright (C) 2006-2022 Oracle and/or its affiliates. + * + * This file is part of VirtualBox base platform packages, as + * available from https://www.virtualbox.org. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation, in version 3 of the + * License. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, see <https://www.gnu.org/licenses>. + * + * SPDX-License-Identifier: GPL-3.0-only + */ + +#define LOG_GROUP LOG_GROUP_MAIN_PROGRESS +#include <iprt/types.h> + +#if defined(VBOX_WITH_XPCOM) +#include <nsIServiceManager.h> +#include <nsIExceptionService.h> +#include <nsCOMPtr.h> +#endif /* defined(VBOX_WITH_XPCOM) */ + +#include "ProgressImpl.h" + +#if !defined(VBOX_COM_INPROC) +# include "VirtualBoxImpl.h" +#endif +#include "VirtualBoxErrorInfoImpl.h" + +#include <iprt/time.h> +#include <iprt/semaphore.h> +#include <iprt/cpp/utils.h> + +#include <iprt/errcore.h> + +#include "AutoCaller.h" +#include "LoggingNew.h" +#include "VBoxEvents.h" + + +Progress::Progress() +#if !defined(VBOX_COM_INPROC) + : mParent(NULL) +#endif +{ +} + +Progress::~Progress() +{ +} + + +HRESULT Progress::FinalConstruct() +{ + mCancelable = FALSE; + mCompleted = FALSE; + mCanceled = FALSE; + mResultCode = S_OK; + + m_cOperations + = m_ulTotalOperationsWeight + = m_ulOperationsCompletedWeight + = m_ulCurrentOperation + = m_ulCurrentOperationWeight + = m_ulOperationPercent + = m_cMsTimeout + = 0; + + // get creation timestamp + m_ullTimestamp = RTTimeMilliTS(); + + m_pfnCancelCallback = NULL; + m_pvCancelUserArg = NULL; + + mCompletedSem = NIL_RTSEMEVENTMULTI; + mWaitersCount = 0; + + return Progress::BaseFinalConstruct(); +} + +void Progress::FinalRelease() +{ + uninit(); + BaseFinalRelease(); +} + +// public initializer/uninitializer for internal purposes only +//////////////////////////////////////////////////////////////////////////////// + +/** + * Initializes the normal progress object. With this variant, one can have + * an arbitrary number of sub-operation which IProgress can analyze to + * have a weighted progress computed. + * + * For example, say that one IProgress is supposed to track the cloning + * of two hard disk images, which are 100 MB and 1000 MB in size, respectively, + * and each of these hard disks should be one sub-operation of the IProgress. + * + * Obviously the progress would be misleading if the progress displayed 50% + * after the smaller image was cloned and would then take much longer for + * the second half. + * + * With weighted progress, one can invoke the following calls: + * + * 1) create progress object with cOperations = 2 and ulTotalOperationsWeight = + * 1100 (100 MB plus 1100, but really the weights can be any ULONG); pass + * in ulFirstOperationWeight = 100 for the first sub-operation + * + * 2) Then keep calling setCurrentOperationProgress() with a percentage + * for the first image; the total progress will increase up to a value + * of 9% (100MB / 1100MB * 100%). + * + * 3) Then call setNextOperation with the second weight (1000 for the megabytes + * of the second disk). + * + * 4) Then keep calling setCurrentOperationProgress() with a percentage for + * the second image, where 100% of the operation will then yield a 100% + * progress of the entire task. + * + * Weighting is optional; you can simply assign a weight of 1 to each operation + * and pass ulTotalOperationsWeight == cOperations to this constructor (but + * for that variant and for backwards-compatibility a simpler constructor exists + * in ProgressImpl.h as well). + * + * Even simpler, if you need no sub-operations at all, pass in cOperations = + * ulTotalOperationsWeight = ulFirstOperationWeight = 1. + * + * @param aParent Parent object (only for server-side Progress objects). + * @param aInitiator Initiator of the task (for server-side objects. Can be + * NULL which means initiator = parent, otherwise must not + * be NULL). + * @param aDescription Overall task description. + * @param aCancelable Flag whether the task maybe canceled. + * @param cOperations Number of operations within this task (at least 1). + * @param ulTotalOperationsWeight Total weight of operations; must be the sum of ulFirstOperationWeight and + * what is later passed with each subsequent setNextOperation() call. + * @param aFirstOperationDescription Description of the first operation. + * @param ulFirstOperationWeight Weight of first sub-operation. + */ +HRESULT Progress::init( +#if !defined(VBOX_COM_INPROC) + VirtualBox *aParent, +#endif + IUnknown *aInitiator, + const Utf8Str &aDescription, + BOOL aCancelable, + ULONG cOperations, + ULONG ulTotalOperationsWeight, + const Utf8Str &aFirstOperationDescription, + ULONG ulFirstOperationWeight) +{ + LogFlowThisFunc(("aDescription=\"%s\", cOperations=%d, ulTotalOperationsWeight=%d, aFirstOperationDescription=\"%s\", ulFirstOperationWeight=%d\n", + aDescription.c_str(), + cOperations, + ulTotalOperationsWeight, + aFirstOperationDescription.c_str(), + ulFirstOperationWeight)); + + AssertReturn(ulTotalOperationsWeight >= 1, E_INVALIDARG); + + /* Enclose the state transition NotReady->InInit->Ready */ + AutoInitSpan autoInitSpan(this); + AssertReturn(autoInitSpan.isOk(), E_FAIL); + + HRESULT hrc = unconst(pEventSource).createObject(); + if (FAILED(hrc)) + return hrc; + + hrc = pEventSource->init(); + if (FAILED(hrc)) + return hrc; + +#if !defined(VBOX_COM_INPROC) + AssertReturn(aParent, E_INVALIDARG); +#else + AssertReturn(aInitiator, E_INVALIDARG); +#endif + +#if !defined(VBOX_COM_INPROC) + /* share parent weakly */ + unconst(mParent) = aParent; +#endif + +#if !defined(VBOX_COM_INPROC) + /* assign (and therefore addref) initiator only if it is not VirtualBox + * (to avoid cycling); otherwise mInitiator will remain null which means + * that it is the same as the parent */ + if (aInitiator) + { + ComObjPtr<VirtualBox> pVirtualBox(mParent); + if (!(pVirtualBox == aInitiator)) + unconst(mInitiator) = aInitiator; + } +#else + unconst(mInitiator) = aInitiator; +#endif + + unconst(mId).create(); + +#if !defined(VBOX_COM_INPROC) + /* add to the global collection of progress operations (note: after + * creating mId) */ + mParent->i_addProgress(this); +#endif + + unconst(mDescription) = aDescription; + + mCancelable = aCancelable; + + m_cOperations = cOperations; + m_ulTotalOperationsWeight = ulTotalOperationsWeight; + m_ulOperationsCompletedWeight = 0; + m_ulCurrentOperation = 0; + m_operationDescription = aFirstOperationDescription; + m_ulCurrentOperationWeight = ulFirstOperationWeight; + m_ulOperationPercent = 0; + + int vrc = RTSemEventMultiCreate(&mCompletedSem); + ComAssertRCRet(vrc, E_FAIL); + + RTSemEventMultiReset(mCompletedSem); + + /* Confirm a successful initialization. */ + autoInitSpan.setSucceeded(); + + return S_OK; +} + +/** + * Initializes the sub-progress object that represents a specific operation of + * the whole task. + * + * Objects initialized with this method are then combined together into the + * single task using a Progress instance, so it doesn't require the + * parent, initiator, description and doesn't create an ID. Note that calling + * respective getter methods on an object initialized with this method is + * useless. Such objects are used only to provide a separate wait semaphore and + * store individual operation descriptions. + * + * @param aCancelable Flag whether the task maybe canceled. + * @param aOperationCount Number of sub-operations within this task (at least 1). + * @param aOperationDescription Description of the individual operation. + */ +HRESULT Progress::init(BOOL aCancelable, + ULONG aOperationCount, + const Utf8Str &aOperationDescription) +{ + LogFlowThisFunc(("aOperationDescription=\"%s\"\n", aOperationDescription.c_str())); + + /* Enclose the state transition NotReady->InInit->Ready */ + AutoInitSpan autoInitSpan(this); + AssertReturn(autoInitSpan.isOk(), E_FAIL); + + mCancelable = aCancelable; + + // for this variant we assume for now that all operations are weighed "1" + // and equal total weight = operation count + m_cOperations = aOperationCount; + m_ulTotalOperationsWeight = aOperationCount; + m_ulOperationsCompletedWeight = 0; + m_ulCurrentOperation = 0; + m_operationDescription = aOperationDescription; + m_ulCurrentOperationWeight = 1; + m_ulOperationPercent = 0; + + int vrc = RTSemEventMultiCreate(&mCompletedSem); + ComAssertRCRet(vrc, E_FAIL); + + RTSemEventMultiReset(mCompletedSem); + + /* Confirm a successful initialization. */ + autoInitSpan.setSucceeded(); + + return S_OK; +} + + +/** + * Uninitializes the instance and sets the ready flag to FALSE. + * + * Called either from FinalRelease() or by the parent when it gets destroyed. + */ +void Progress::uninit() +{ + LogFlowThisFunc(("\n")); + + /* Enclose the state transition Ready->InUninit->NotReady */ + AutoUninitSpan autoUninitSpan(this); + if (autoUninitSpan.uninitDone()) + return; + + /* wake up all threads still waiting on occasion */ + if (mWaitersCount > 0) + { + LogFlow(("WARNING: There are still %d threads waiting for '%s' completion!\n", + mWaitersCount, mDescription.c_str())); + RTSemEventMultiSignal(mCompletedSem); + } + + RTSemEventMultiDestroy(mCompletedSem); + + /* release initiator (effective only if mInitiator has been assigned in init()) */ + unconst(mInitiator).setNull(); + +#if !defined(VBOX_COM_INPROC) + if (mParent) + { + /* remove the added progress on failure to complete the initialization */ + if (autoUninitSpan.initFailed() && mId.isValid() && !mId.isZero()) + mParent->i_removeProgress(mId.ref()); + + unconst(mParent) = NULL; + } +#endif +} + + +// public methods only for internal purposes +//////////////////////////////////////////////////////////////////////////////// + +/** + * Marks the whole task as complete and sets the result code. + * + * If the result code indicates a failure (|FAILED(@a aResultCode)|) then this + * method will import the error info from the current thread and assign it to + * the errorInfo attribute (it will return an error if no info is available in + * such case). + * + * If the result code indicates a success (|SUCCEEDED(@a aResultCode)|) then + * the current operation is set to the last. + * + * Note that this method may be called only once for the given Progress object. + * Subsequent calls will assert. + * + * @param aResultCode Operation result code. + */ +HRESULT Progress::i_notifyComplete(HRESULT aResultCode) +{ + HRESULT hrc; + ComPtr<IVirtualBoxErrorInfo> errorInfo; + if (FAILED(aResultCode)) + { + /* try to import error info from the current thread */ +#if !defined(VBOX_WITH_XPCOM) + ComPtr<IErrorInfo> err; + hrc = ::GetErrorInfo(0, err.asOutParam()); + if (hrc == S_OK && err) + hrc = err.queryInterfaceTo(errorInfo.asOutParam()); +#else /* !defined(VBOX_WITH_XPCOM) */ + nsCOMPtr<nsIExceptionService> es; + es = do_GetService(NS_EXCEPTIONSERVICE_CONTRACTID, &hrc); + if (NS_SUCCEEDED(hrc)) + { + nsCOMPtr <nsIExceptionManager> em; + hrc = es->GetCurrentExceptionManager(getter_AddRefs(em)); + if (NS_SUCCEEDED(hrc)) + { + ComPtr<nsIException> ex; + hrc = em->GetCurrentException(ex.asOutParam()); + if (NS_SUCCEEDED(hrc) && ex) + hrc = ex.queryInterfaceTo(errorInfo.asOutParam()); + } + } +#endif /* !defined(VBOX_WITH_XPCOM) */ + } + + return i_notifyCompleteWorker(aResultCode, errorInfo); +} + +/** + * Wrapper around Progress:notifyCompleteV. + */ +HRESULT Progress::i_notifyComplete(HRESULT aResultCode, + const GUID &aIID, + const char *pcszComponent, + const char *aText, + ...) +{ + va_list va; + va_start(va, aText); + HRESULT hrc = i_notifyCompleteV(aResultCode, aIID, pcszComponent, aText, va); + va_end(va); + return hrc; +} + +/** + * Marks the operation as complete and attaches full error info. + * + * @param aResultCode Operation result (error) code, must not be S_OK. + * @param aIID IID of the interface that defines the error. + * @param pcszComponent Name of the component that generates the error. + * @param aText Error message (must not be null), an RTStrPrintf-like + * format string in UTF-8 encoding. + * @param va List of arguments for the format string. + */ +HRESULT Progress::i_notifyCompleteV(HRESULT aResultCode, + const GUID &aIID, + const char *pcszComponent, + const char *aText, + va_list va) +{ + /* expected to be used only in case of error */ + Assert(FAILED(aResultCode)); + + Utf8Str text(aText, va); + ComObjPtr<VirtualBoxErrorInfo> errorInfo; + HRESULT hrc = errorInfo.createObject(); + AssertComRCReturnRC(hrc); + errorInfo->init(aResultCode, aIID, pcszComponent, text); + + return i_notifyCompleteWorker(aResultCode, errorInfo); +} + +/** + * Wrapper around Progress:notifyCompleteBothV. + */ +HRESULT Progress::i_notifyCompleteBoth(HRESULT aResultCode, + int vrc, + const GUID &aIID, + const char *pcszComponent, + const char *aText, + ...) +{ + va_list va; + va_start(va, aText); + HRESULT hrc = i_notifyCompleteBothV(aResultCode, vrc, aIID, pcszComponent, aText, va); + va_end(va); + return hrc; +} + +/** + * Marks the operation as complete and attaches full error info. + * + * @param aResultCode Operation result (error) code, must not be S_OK. + * @param vrc VBox status code to associate with the error. + * @param aIID IID of the interface that defines the error. + * @param pszComponent Name of the component that generates the error. + * @param pszFormat Error message (must not be null), an RTStrPrintf-like + * format string in UTF-8 encoding. + * @param va List of arguments for the format string. + */ +HRESULT Progress::i_notifyCompleteBothV(HRESULT aResultCode, + int vrc, + const GUID &aIID, + const char *pszComponent, + const char *pszFormat, + va_list va) +{ + /* expected to be used only in case of error */ + Assert(FAILED(aResultCode)); + + Utf8Str text(pszFormat, va); + ComObjPtr<VirtualBoxErrorInfo> errorInfo; + HRESULT hrc = errorInfo.createObject(); + AssertComRCReturnRC(hrc); + errorInfo->initEx(aResultCode, vrc, aIID, pszComponent, text); + + return i_notifyCompleteWorker(aResultCode, errorInfo); +} + +/** + * Sets the cancelation callback, checking for cancelation first. + * + * @returns Success indicator. + * @retval true on success. + * @retval false if the progress object has already been canceled or is in an + * invalid state + * + * @param pfnCallback The function to be called upon cancelation. + * @param pvUser The callback argument. + */ +bool Progress::i_setCancelCallback(void (*pfnCallback)(void *), void *pvUser) +{ + AutoCaller autoCaller(this); + AssertReturn(autoCaller.isOk(), false); + + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + i_checkForAutomaticTimeout(); + if (mCanceled) + return false; + + m_pvCancelUserArg = pvUser; + m_pfnCancelCallback = pfnCallback; + return true; +} + +/** + * @callback_method_impl{FNRTPROGRESS, + * Works the progress of the current operation.} + */ +/*static*/ DECLCALLBACK(int) Progress::i_iprtProgressCallback(unsigned uPercentage, void *pvUser) +{ + Progress *pThis = (Progress *)pvUser; + + /* + * Same as setCurrentOperationProgress, except we don't fail on mCompleted. + */ + AutoWriteLock alock(pThis COMMA_LOCKVAL_SRC_POS); + int vrc = VINF_SUCCESS; + if (!pThis->mCompleted) + { + pThis->i_checkForAutomaticTimeout(); + if (!pThis->mCanceled) + { + if (uPercentage > pThis->m_ulOperationPercent) + pThis->setCurrentOperationProgress(uPercentage); + } + else + { + Assert(pThis->mCancelable); + vrc = VERR_CANCELLED; + } + } + /* else ignored */ + return vrc; +} + +/** + * @callback_method_impl{FNVDPROGRESS, + * Progress::i_iprtProgressCallback with parameters switched around.} + */ +/*static*/ DECLCALLBACK(int) Progress::i_vdProgressCallback(void *pvUser, unsigned uPercentage) +{ + return i_iprtProgressCallback(uPercentage, pvUser); +} + + +// IProgress properties +///////////////////////////////////////////////////////////////////////////// + +HRESULT Progress::getId(com::Guid &aId) +{ + /* mId is constant during life time, no need to lock */ + aId = mId; + + return S_OK; +} + +HRESULT Progress::getDescription(com::Utf8Str &aDescription) +{ + /* mDescription is constant during life time, no need to lock */ + aDescription = mDescription; + + return S_OK; +} +HRESULT Progress::getInitiator(ComPtr<IUnknown> &aInitiator) +{ + /* mInitiator/mParent are constant during life time, no need to lock */ +#if !defined(VBOX_COM_INPROC) + if (mInitiator) + mInitiator.queryInterfaceTo(aInitiator.asOutParam()); + else + { + ComObjPtr<VirtualBox> pVirtualBox(mParent); + pVirtualBox.queryInterfaceTo(aInitiator.asOutParam()); + } +#else + mInitiator.queryInterfaceTo(aInitiator.asOutParam()); +#endif + + return S_OK; +} + +HRESULT Progress::getCancelable(BOOL *aCancelable) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aCancelable = mCancelable; + + return S_OK; +} + +HRESULT Progress::getPercent(ULONG *aPercent) +{ + /* i_checkForAutomaticTimeout requires a write lock. */ + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (mCompleted && SUCCEEDED(mResultCode)) + *aPercent = 100; + else + { + ULONG ulPercent = (ULONG)i_calcTotalPercent(); + // do not report 100% until we're really really done with everything + // as the Qt GUI dismisses progress dialogs in that case + if ( ulPercent == 100 + && ( m_ulOperationPercent < 100 + || (m_ulCurrentOperation < m_cOperations -1) + ) + ) + *aPercent = 99; + else + *aPercent = ulPercent; + } + + i_checkForAutomaticTimeout(); + + return S_OK; +} + +HRESULT Progress::getTimeRemaining(LONG *aTimeRemaining) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (mCompleted) + *aTimeRemaining = 0; + else + { + double dPercentDone = i_calcTotalPercent(); + if (dPercentDone < 1) + *aTimeRemaining = -1; // unreliable, or avoid division by 0 below + else + { + uint64_t ullTimeNow = RTTimeMilliTS(); + uint64_t ullTimeElapsed = ullTimeNow - m_ullTimestamp; + uint64_t ullTimeTotal = (uint64_t)((double)ullTimeElapsed * 100 / dPercentDone); + uint64_t ullTimeRemaining = (ullTimeTotal < ullTimeElapsed) ? 0 : ullTimeTotal - ullTimeElapsed; + +// LogFunc(("dPercentDone = %RI32, ullTimeNow = %RI64, ullTimeElapsed = %RI64, ullTimeTotal = %RI64, ullTimeRemaining = %RI64\n", +// (uint32_t)dPercentDone, ullTimeNow, ullTimeElapsed, ullTimeTotal, ullTimeRemaining)); + + *aTimeRemaining = (LONG)(RT_MIN(ullTimeRemaining, RT_MS_1HOUR_64*24*365) / 1000); + } + } + + return S_OK; +} + +HRESULT Progress::getCompleted(BOOL *aCompleted) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aCompleted = mCompleted; + + return S_OK; +} + +HRESULT Progress::getCanceled(BOOL *aCanceled) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aCanceled = mCanceled; + + return S_OK; +} + +HRESULT Progress::getResultCode(LONG *aResultCode) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (!mCompleted) + return setError(E_FAIL, tr("Result code is not available, operation is still in progress")); + + *aResultCode = (LONG)mResultCode; + + return S_OK; +} + +HRESULT Progress::getErrorInfo(ComPtr<IVirtualBoxErrorInfo> &aErrorInfo) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (!mCompleted) + return setError(E_FAIL, tr("Error info is not available, operation is still in progress")); + + mErrorInfo.queryInterfaceTo(aErrorInfo.asOutParam()); + + return S_OK; +} + +HRESULT Progress::getOperationCount(ULONG *aOperationCount) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aOperationCount = m_cOperations; + + return S_OK; +} + +HRESULT Progress::getOperation(ULONG *aOperation) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aOperation = m_ulCurrentOperation; + + return S_OK; +} + +HRESULT Progress::getOperationDescription(com::Utf8Str &aOperationDescription) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + aOperationDescription = m_operationDescription; + + return S_OK; +} + +HRESULT Progress::getOperationPercent(ULONG *aOperationPercent) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (mCompleted && SUCCEEDED(mResultCode)) + *aOperationPercent = 100; + else + *aOperationPercent = m_ulOperationPercent; + + return S_OK; +} + +HRESULT Progress::getOperationWeight(ULONG *aOperationWeight) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aOperationWeight = m_ulCurrentOperationWeight; + + return S_OK; +} + +HRESULT Progress::getTimeout(ULONG *aTimeout) +{ + AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); + + *aTimeout = m_cMsTimeout; + + return S_OK; +} + +HRESULT Progress::setTimeout(ULONG aTimeout) +{ + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (!mCancelable) + return setError(VBOX_E_INVALID_OBJECT_STATE, tr("Operation cannot be canceled")); + m_cMsTimeout = aTimeout; + + return S_OK; +} + +HRESULT Progress::getEventSource(ComPtr<IEventSource> &aEventSource) +{ + /* event source is const, no need to lock */ + pEventSource.queryInterfaceTo(aEventSource.asOutParam()); + return S_OK; +} + + +// IProgress methods +///////////////////////////////////////////////////////////////////////////// + +/** + * @note XPCOM: when this method is not called on the main XPCOM thread, it + * simply blocks the thread until mCompletedSem is signalled. If the + * thread has its own event queue (hmm, what for?) that it must run, then + * calling this method will definitely freeze event processing. + */ +HRESULT Progress::waitForCompletion(LONG aTimeout) +{ + LogFlowThisFuncEnter(); + LogFlowThisFunc(("aTimeout=%d\n", aTimeout)); + + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + /* if we're already completed, take a shortcut */ + if (!mCompleted && aTimeout != 0) + { + RTMSINTERVAL cMsWait = aTimeout < 0 ? RT_INDEFINITE_WAIT : (RTMSINTERVAL)aTimeout; + uint64_t msLast = aTimeout < 0 ? 0 : RTTimeMilliTS(); + + for (;;) + { + mWaitersCount++; + alock.release(); + int vrc = RTSemEventMultiWait(mCompletedSem, cMsWait); + alock.acquire(); + mWaitersCount--; + + /* the last waiter resets the semaphore */ + if (mWaitersCount == 0) + RTSemEventMultiReset(mCompletedSem); + + if (RT_FAILURE(vrc) && vrc != VERR_TIMEOUT) + return setErrorBoth(VBOX_E_IPRT_ERROR, vrc, tr("Failed to wait for the task completion (%Rrc)"), vrc); + + if (mCompleted) + break; + + if (aTimeout >= 0) + { + uint64_t msNow = RTTimeMilliTS(); + uint64_t cMsElapsed = msNow - msLast; + if (cMsWait <= cMsElapsed) + break; + cMsWait -= (RTMSINTERVAL)cMsElapsed; + msLast = msNow; + } + } + } + + LogFlowThisFuncLeave(); + return S_OK; +} + +/** + * @note XPCOM: when this method is not called on the main XPCOM thread, it + * simply blocks the thread until mCompletedSem is signalled. If the + * thread has its own event queue (hmm, what for?) that it must run, then + * calling this method will definitely freeze event processing. + */ +HRESULT Progress::waitForOperationCompletion(ULONG aOperation, LONG aTimeout) + +{ + LogFlowThisFuncEnter(); + LogFlowThisFunc(("aOperation=%d, aTimeout=%d\n", aOperation, aTimeout)); + + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + CheckComArgExpr(aOperation, aOperation < m_cOperations); + + /* if we're already completed or if the given operation is already done, + * then take a shortcut */ + if ( !mCompleted + && aOperation >= m_ulCurrentOperation + && aTimeout != 0) + { + RTMSINTERVAL cMsWait = aTimeout < 0 ? RT_INDEFINITE_WAIT : (RTMSINTERVAL)aTimeout; + uint64_t msLast = aTimeout < 0 ? 0 : RTTimeMilliTS(); + + for (;;) + { + mWaitersCount ++; + alock.release(); + int vrc = RTSemEventMultiWait(mCompletedSem, cMsWait); + alock.acquire(); + mWaitersCount--; + + /* the last waiter resets the semaphore */ + if (mWaitersCount == 0) + RTSemEventMultiReset(mCompletedSem); + + if (RT_FAILURE(vrc) && vrc != VERR_TIMEOUT) + return setErrorBoth(E_FAIL, vrc, tr("Failed to wait for the operation completion (%Rrc)"), vrc); + + if (mCompleted || aOperation >= m_ulCurrentOperation) + break; + + if (aTimeout >= 0) + { + uint64_t msNow = RTTimeMilliTS(); + uint64_t cMsElapsed = msNow - msLast; + if (cMsWait <= cMsElapsed) + break; + cMsWait -= (RTMSINTERVAL)cMsElapsed; + msLast = msNow; + } + } + } + + LogFlowThisFuncLeave(); + return S_OK; +} + +HRESULT Progress::cancel() +{ + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (!mCancelable) + return setError(VBOX_E_INVALID_OBJECT_STATE, tr("Operation cannot be canceled")); + + if (!mCanceled) + { + LogThisFunc(("Canceling\n")); + mCanceled = TRUE; + if (m_pfnCancelCallback) + m_pfnCancelCallback(m_pvCancelUserArg); + + } + else + LogThisFunc(("Already canceled\n")); + + return S_OK; +} + + +// IInternalProgressControl methods +///////////////////////////////////////////////////////////////////////////// + +/** + * Updates the percentage value of the current operation. + * + * @param aPercent New percentage value of the operation in progress + * (in range [0, 100]). + */ +HRESULT Progress::setCurrentOperationProgress(ULONG aPercent) +{ + AssertMsgReturn(aPercent <= 100, ("%u\n", aPercent), E_INVALIDARG); + + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + i_checkForAutomaticTimeout(); + if (mCancelable && mCanceled) + AssertReturn(!mCompleted, E_FAIL); + AssertReturn(!mCompleted && !mCanceled, E_FAIL); + + if (m_ulOperationPercent != aPercent) + { + m_ulOperationPercent = aPercent; + ULONG actualPercent = 0; + getPercent(&actualPercent); + ::FireProgressPercentageChangedEvent(pEventSource, mId.toString(), (LONG)actualPercent); + } + + return S_OK; +} + +HRESULT Progress::waitForOtherProgressCompletion(const ComPtr<IProgress> &aProgressOther, + ULONG aTimeoutMS) +{ + LogFlowThisFuncEnter(); + + /* Note: no locking needed, because we just use public methods. */ + + BOOL fCancelable = FALSE; + BOOL fCompleted = FALSE; + BOOL fCanceled = FALSE; + ULONG prevPercent = UINT32_MAX; + ULONG currentPercent = 0; + ULONG cOp = 0; + /* Is the async process cancelable? */ + HRESULT hrc = aProgressOther->COMGETTER(Cancelable)(&fCancelable); + if (FAILED(hrc)) return hrc; + + uint64_t u64StopTime = UINT64_MAX; + if (aTimeoutMS > 0) + u64StopTime = RTTimeMilliTS() + aTimeoutMS; + /* Loop as long as the sync process isn't completed. */ + while (SUCCEEDED(aProgressOther->COMGETTER(Completed(&fCompleted)))) + { + /* We can forward any cancel request to the async process only when + * it is cancelable. */ + if (fCancelable) + { + hrc = COMGETTER(Canceled)(&fCanceled); + if (FAILED(hrc)) return hrc; + if (fCanceled) + { + hrc = aProgressOther->Cancel(); + if (FAILED(hrc)) return hrc; + } + } + /* Even if the user canceled the process, we have to wait until the + async task has finished his work (cleanup and such). Otherwise there + will be sync trouble (still wrong state, dead locks, ...) on the + used objects. So just do nothing, but wait for the complete + notification. */ + if (!fCanceled) + { + /* Check if the current operation has changed. It is also possible that + * in the meantime more than one async operation was finished. So we + * have to loop as long as we reached the same operation count. */ + ULONG curOp; + for (;;) + { + hrc = aProgressOther->COMGETTER(Operation(&curOp)); + if (FAILED(hrc)) return hrc; + if (cOp != curOp) + { + Bstr bstr; + ULONG currentWeight; + hrc = aProgressOther->COMGETTER(OperationDescription(bstr.asOutParam())); + if (FAILED(hrc)) return hrc; + hrc = aProgressOther->COMGETTER(OperationWeight(¤tWeight)); + if (FAILED(hrc)) return hrc; + hrc = SetNextOperation(bstr.raw(), currentWeight); + if (FAILED(hrc)) return hrc; + ++cOp; + } + else + break; + } + + hrc = aProgressOther->COMGETTER(OperationPercent(¤tPercent)); + if (FAILED(hrc)) return hrc; + if (currentPercent != prevPercent) + { + prevPercent = currentPercent; + hrc = SetCurrentOperationProgress(currentPercent); + if (FAILED(hrc)) return hrc; + } + } + if (fCompleted) + break; + + if (aTimeoutMS != 0) + { + /* Make sure the loop is not too tight */ + uint64_t u64Now = RTTimeMilliTS(); + uint64_t u64RemainingMS = u64StopTime - u64Now; + if (u64RemainingMS < 10) + u64RemainingMS = 10; + else if (u64RemainingMS > 200) + u64RemainingMS = 200; + hrc = aProgressOther->WaitForCompletion((LONG)u64RemainingMS); + if (FAILED(hrc)) return hrc; + + if (RTTimeMilliTS() >= u64StopTime) + return VBOX_E_TIMEOUT; + } + else + { + /* Make sure the loop is not too tight */ + hrc = aProgressOther->WaitForCompletion(200); + if (FAILED(hrc)) return hrc; + } + } + + /* Transfer error information if applicable and report the error status + * back to the caller to make this as easy as possible. */ + LONG iRc; + hrc = aProgressOther->COMGETTER(ResultCode)(&iRc); + if (FAILED(hrc)) return hrc; + if (FAILED((HRESULT)iRc)) + { + setError(ProgressErrorInfo(aProgressOther)); + hrc = (HRESULT)iRc; + } + + LogFlowThisFuncLeave(); + return hrc; +} + +/** + * Signals that the current operation is successfully completed and advances to + * the next operation. The operation percentage is reset to 0. + * + * @param aNextOperationDescription Description of the next operation. + * @param aNextOperationsWeight Weight of the next operation. + * + * @note The current operation must not be the last one. + */ +HRESULT Progress::setNextOperation(const com::Utf8Str &aNextOperationDescription, ULONG aNextOperationsWeight) +{ + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (mCanceled) + return E_FAIL; + AssertReturn(!mCompleted, E_FAIL); + AssertReturn(m_ulCurrentOperation + 1 < m_cOperations, E_FAIL); + + ++m_ulCurrentOperation; + m_ulOperationsCompletedWeight += m_ulCurrentOperationWeight; + + m_operationDescription = aNextOperationDescription; + m_ulCurrentOperationWeight = aNextOperationsWeight; + m_ulOperationPercent = 0; + + LogThisFunc(("%s: aNextOperationsWeight = %d; m_ulCurrentOperation is now %d, m_ulOperationsCompletedWeight is now %d\n", + m_operationDescription.c_str(), aNextOperationsWeight, m_ulCurrentOperation, m_ulOperationsCompletedWeight)); + + /* wake up all waiting threads */ + if (mWaitersCount > 0) + RTSemEventMultiSignal(mCompletedSem); + + ULONG actualPercent = 0; + getPercent(&actualPercent); + ::FireProgressPercentageChangedEvent(pEventSource, mId.toString(), (LONG)actualPercent); + + return S_OK; +} + +/** + * Notify the progress object that we're almost at the point of no return. + * + * This atomically checks for and disables cancelation. Calls to + * IProgress::Cancel() made after a successful call to this method will fail + * and the user can be told. While this isn't entirely clean behavior, it + * prevents issues with an irreversible actually operation succeeding while the + * user believe it was rolled back. + * + * @returns COM error status. + * @retval S_OK on success. + * @retval E_FAIL if the progress object has already been canceled or is in an + * invalid state + */ +HRESULT Progress::notifyPointOfNoReturn(void) +{ + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + if (mCanceled) + { + LogThisFunc(("returns failure\n")); + return E_FAIL; + } + + mCancelable = FALSE; + LogThisFunc(("returns success\n")); + return S_OK; +} + +/** + * Marks the operation as complete and attaches full error info. + * + * This is where the actual work is done, the related methods all end up here. + * + * @param aResultCode Operation result (error) code, must not be S_OK. + * @param aErrorInfo List of arguments for the format string. + */ +HRESULT Progress::notifyComplete(LONG aResultCode, const ComPtr<IVirtualBoxErrorInfo> &aErrorInfo) +{ + return i_notifyCompleteWorker((HRESULT)aResultCode, aErrorInfo); +} + + +// private internal helpers +///////////////////////////////////////////////////////////////////////////// + +/** + * Marks the operation as complete and attaches full error info. + * + * This is where the actual work is done, the related methods all end up here. + * + * @param aResultCode Operation result (error) code, must not be S_OK. + * @param aErrorInfo List of arguments for the format string. + * + * @note This is just notifyComplete with the correct aResultCode type. + */ +HRESULT Progress::i_notifyCompleteWorker(HRESULT aResultCode, const ComPtr<IVirtualBoxErrorInfo> &aErrorInfo) +{ + LogThisFunc(("aResultCode=%Rhrc\n", aResultCode)); + /* on failure we expect error info, on success there must be none */ + AssertMsg(FAILED(aResultCode) ^ aErrorInfo.isNull(), + ("No error info but trying to set a failed result (%08X/%Rhrc)!\n", aResultCode, aResultCode)); + + AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); + + AssertReturn(mCompleted == FALSE, E_FAIL); + + if (mCanceled && SUCCEEDED(aResultCode)) + aResultCode = E_FAIL; + + mCompleted = TRUE; + mResultCode = aResultCode; + if (SUCCEEDED(aResultCode)) + { + m_ulCurrentOperation = m_cOperations - 1; /* last operation */ + m_ulOperationPercent = 100; + } + mErrorInfo = aErrorInfo; + +#if !defined(VBOX_COM_INPROC) + /* remove from the global collection of pending progress operations */ + if (mParent) + mParent->i_removeProgress(mId.ref()); +#endif + + /* wake up all waiting threads */ + if (mWaitersCount > 0) + RTSemEventMultiSignal(mCompletedSem); + + ::FireProgressTaskCompletedEvent(pEventSource, mId.toString()); + + return S_OK; +} + +/** + * Internal helper to compute the total percent value based on the member values and + * returns it as a "double". This is used both by GetPercent (which returns it as a + * rounded ULONG) and GetTimeRemaining(). + * + * Requires locking by the caller! + * + * @return fractional percentage as a double value. + */ +double Progress::i_calcTotalPercent() +{ + // avoid division by zero + if (m_ulTotalOperationsWeight == 0) + return 0.0; + + double dPercent = ( (double)m_ulOperationsCompletedWeight // weight of operations that have been completed + + ((double)m_ulOperationPercent * + (double)m_ulCurrentOperationWeight / 100.0) // plus partial weight of the current operation + ) * 100.0 / (double)m_ulTotalOperationsWeight; + + return dPercent; +} + +/** + * Internal helper for automatically timing out the operation. + * + * The caller must hold the object write lock. + */ +void Progress::i_checkForAutomaticTimeout(void) +{ + AssertReturnVoid(isWriteLockOnCurrentThread()); + + if ( m_cMsTimeout + && mCancelable + && !mCanceled + && RTTimeMilliTS() - m_ullTimestamp > m_cMsTimeout) + Cancel(); +} |