VirtualBox

source: vbox/trunk/src/VBox/Main/MediumImpl.cpp@ 29224

Last change on this file since 29224 was 29224, checked in by vboxsync, 15 years ago

Main: turn ISO not found error into warning to allow restoring VMs with inaccessible ISOs

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 202.5 KB
Line 
1/* $Id: MediumImpl.cpp 29224 2010-05-07 15:49:10Z vboxsync $ */
2
3/** @file
4 *
5 * VirtualBox COM class implementation
6 */
7
8/*
9 * Copyright (C) 2008-2010 Oracle Corporation
10 *
11 * This file is part of VirtualBox Open Source Edition (OSE), as
12 * available from http://www.virtualbox.org. This file is free software;
13 * you can redistribute it and/or modify it under the terms of the GNU
14 * General Public License (GPL) as published by the Free Software
15 * Foundation, in version 2 as it comes in the "COPYING" file of the
16 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
17 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
18 */
19
20#include "MediumImpl.h"
21#include "ProgressImpl.h"
22#include "SystemPropertiesImpl.h"
23#include "VirtualBoxImpl.h"
24
25#include "AutoCaller.h"
26#include "Logging.h"
27
28#include <VBox/com/array.h>
29#include <VBox/com/SupportErrorInfo.h>
30
31#include <VBox/err.h>
32#include <VBox/settings.h>
33
34#include <iprt/param.h>
35#include <iprt/path.h>
36#include <iprt/file.h>
37#include <iprt/tcp.h>
38
39#include <VBox/VBoxHDD.h>
40
41#include <algorithm>
42
43////////////////////////////////////////////////////////////////////////////////
44//
45// Medium data definition
46//
47////////////////////////////////////////////////////////////////////////////////
48
49/** Describes how a machine refers to this image. */
50struct BackRef
51{
52 /** Equality predicate for stdc++. */
53 struct EqualsTo : public std::unary_function <BackRef, bool>
54 {
55 explicit EqualsTo(const Guid &aMachineId) : machineId(aMachineId) {}
56
57 bool operator()(const argument_type &aThat) const
58 {
59 return aThat.machineId == machineId;
60 }
61
62 const Guid machineId;
63 };
64
65 typedef std::list<Guid> GuidList;
66
67 BackRef(const Guid &aMachineId,
68 const Guid &aSnapshotId = Guid::Empty)
69 : machineId(aMachineId),
70 fInCurState(aSnapshotId.isEmpty())
71 {
72 if (!aSnapshotId.isEmpty())
73 llSnapshotIds.push_back(aSnapshotId);
74 }
75
76 Guid machineId;
77 bool fInCurState : 1;
78 GuidList llSnapshotIds;
79};
80
81typedef std::list<BackRef> BackRefList;
82
83struct Medium::Data
84{
85 Data()
86 : pVirtualBox(NULL),
87 state(MediumState_NotCreated),
88 size(0),
89 readers(0),
90 preLockState(MediumState_NotCreated),
91 queryInfoSem(NIL_RTSEMEVENTMULTI),
92 queryInfoRunning(false),
93 type(MediumType_Normal),
94 devType(DeviceType_HardDisk),
95 logicalSize(0),
96 hddOpenMode(OpenReadWrite),
97 autoReset(false),
98 setImageId(false),
99 setParentId(false),
100 hostDrive(FALSE),
101 implicit(false),
102 numCreateDiffTasks(0),
103 vdDiskIfaces(NULL)
104 {}
105
106 /** weak VirtualBox parent */
107 VirtualBox * const pVirtualBox;
108
109 const Guid id;
110 Utf8Str strDescription;
111 MediumState_T state;
112 Utf8Str strLocation;
113 Utf8Str strLocationFull;
114 uint64_t size;
115 Utf8Str strLastAccessError;
116
117 // pParent and llChildren are protected by VirtualBox::getMediaTreeLockHandle()
118 ComObjPtr<Medium> pParent;
119 MediaList llChildren; // to add a child, just call push_back; to remove a child, call child->deparent() which does a lookup
120
121 BackRefList backRefs;
122
123 size_t readers;
124 MediumState_T preLockState;
125
126 RTSEMEVENTMULTI queryInfoSem;
127 bool queryInfoRunning : 1;
128
129 const Utf8Str strFormat;
130 ComObjPtr<MediumFormat> formatObj;
131
132 MediumType_T type;
133 DeviceType_T devType;
134 uint64_t logicalSize; /*< In MBytes. */
135
136 HDDOpenMode hddOpenMode;
137
138 BOOL autoReset : 1;
139
140 /** the following members are invalid after changing UUID on open */
141 BOOL setImageId : 1;
142 BOOL setParentId : 1;
143 const Guid imageId;
144 const Guid parentId;
145
146 BOOL hostDrive : 1;
147
148 typedef std::map <Bstr, Bstr> PropertyMap;
149 PropertyMap properties;
150
151 bool implicit : 1;
152
153 uint32_t numCreateDiffTasks;
154
155 Utf8Str vdError; /*< Error remembered by the VD error callback. */
156
157 VDINTERFACE vdIfError;
158 VDINTERFACEERROR vdIfCallsError;
159
160 VDINTERFACE vdIfConfig;
161 VDINTERFACECONFIG vdIfCallsConfig;
162
163 VDINTERFACE vdIfTcpNet;
164 VDINTERFACETCPNET vdIfCallsTcpNet;
165
166 PVDINTERFACE vdDiskIfaces;
167};
168
169////////////////////////////////////////////////////////////////////////////////
170//
171// Globals
172//
173////////////////////////////////////////////////////////////////////////////////
174
175/**
176 * Medium::Task class for asynchronous operations.
177 *
178 * @note Instances of this class must be created using new() because the
179 * task thread function will delete them when the task is complete.
180 *
181 * @note The constructor of this class adds a caller on the managed Medium
182 * object which is automatically released upon destruction.
183 */
184class Medium::Task
185{
186public:
187 Task(Medium *aMedium, Progress *aProgress)
188 : mVDOperationIfaces(NULL),
189 m_pfNeedsSaveSettings(NULL),
190 mMedium(aMedium),
191 mMediumCaller(aMedium),
192 mThread(NIL_RTTHREAD),
193 mProgress(aProgress)
194 {
195 AssertReturnVoidStmt(aMedium, mRC = E_FAIL);
196 mRC = mMediumCaller.rc();
197 if (FAILED(mRC))
198 return;
199
200 /* Set up a per-operation progress interface, can be used freely (for
201 * binary operations you can use it either on the source or target). */
202 mVDIfCallsProgress.cbSize = sizeof(VDINTERFACEPROGRESS);
203 mVDIfCallsProgress.enmInterface = VDINTERFACETYPE_PROGRESS;
204 mVDIfCallsProgress.pfnProgress = vdProgressCall;
205 int vrc = VDInterfaceAdd(&mVDIfProgress,
206 "Medium::Task::vdInterfaceProgress",
207 VDINTERFACETYPE_PROGRESS,
208 &mVDIfCallsProgress,
209 mProgress,
210 &mVDOperationIfaces);
211 AssertRC(vrc);
212 if (RT_FAILURE(vrc))
213 mRC = E_FAIL;
214 }
215
216 // Make all destructors virtual. Just in case.
217 virtual ~Task()
218 {}
219
220 HRESULT rc() const { return mRC; }
221 bool isOk() const { return SUCCEEDED(rc()); }
222
223 static int fntMediumTask(RTTHREAD aThread, void *pvUser);
224
225 bool isAsync() { return mThread != NIL_RTTHREAD; }
226
227 PVDINTERFACE mVDOperationIfaces;
228
229 // Whether the caller needs to call VirtualBox::saveSettings() after
230 // the task function returns. Only used in synchronous (wait) mode;
231 // otherwise the task will save the settings itself.
232 bool *m_pfNeedsSaveSettings;
233
234 const ComObjPtr<Medium> mMedium;
235 AutoCaller mMediumCaller;
236
237 friend HRESULT Medium::runNow(Medium::Task*, bool*);
238
239protected:
240 HRESULT mRC;
241 RTTHREAD mThread;
242
243private:
244 virtual HRESULT handler() = 0;
245
246 const ComObjPtr<Progress> mProgress;
247
248 static DECLCALLBACK(int) vdProgressCall(void *pvUser, unsigned uPercent);
249
250 VDINTERFACE mVDIfProgress;
251 VDINTERFACEPROGRESS mVDIfCallsProgress;
252};
253
254class Medium::CreateBaseTask : public Medium::Task
255{
256public:
257 CreateBaseTask(Medium *aMedium,
258 Progress *aProgress,
259 uint64_t aSize,
260 MediumVariant_T aVariant)
261 : Medium::Task(aMedium, aProgress),
262 mSize(aSize),
263 mVariant(aVariant)
264 {}
265
266 uint64_t mSize;
267 MediumVariant_T mVariant;
268
269private:
270 virtual HRESULT handler();
271};
272
273class Medium::CreateDiffTask : public Medium::Task
274{
275public:
276 CreateDiffTask(Medium *aMedium,
277 Progress *aProgress,
278 Medium *aTarget,
279 MediumVariant_T aVariant,
280 MediumLockList *aMediumLockList,
281 bool fKeepMediumLockList = false)
282 : Medium::Task(aMedium, aProgress),
283 mpMediumLockList(aMediumLockList),
284 mTarget(aTarget),
285 mVariant(aVariant),
286 mTargetCaller(aTarget),
287 mfKeepMediumLockList(fKeepMediumLockList)
288 {
289 AssertReturnVoidStmt(aTarget != NULL, mRC = E_FAIL);
290 mRC = mTargetCaller.rc();
291 if (FAILED(mRC))
292 return;
293 }
294
295 ~CreateDiffTask()
296 {
297 if (!mfKeepMediumLockList && mpMediumLockList)
298 delete mpMediumLockList;
299 }
300
301 MediumLockList *mpMediumLockList;
302
303 const ComObjPtr<Medium> mTarget;
304 MediumVariant_T mVariant;
305
306private:
307 virtual HRESULT handler();
308
309 AutoCaller mTargetCaller;
310 bool mfKeepMediumLockList;
311};
312
313class Medium::CloneTask : public Medium::Task
314{
315public:
316 CloneTask(Medium *aMedium,
317 Progress *aProgress,
318 Medium *aTarget,
319 MediumVariant_T aVariant,
320 Medium *aParent,
321 MediumLockList *aSourceMediumLockList,
322 MediumLockList *aTargetMediumLockList,
323 bool fKeepSourceMediumLockList = false,
324 bool fKeepTargetMediumLockList = false)
325 : Medium::Task(aMedium, aProgress),
326 mTarget(aTarget),
327 mParent(aParent),
328 mpSourceMediumLockList(aSourceMediumLockList),
329 mpTargetMediumLockList(aTargetMediumLockList),
330 mVariant(aVariant),
331 mTargetCaller(aTarget),
332 mParentCaller(aParent),
333 mfKeepSourceMediumLockList(fKeepSourceMediumLockList),
334 mfKeepTargetMediumLockList(fKeepTargetMediumLockList)
335 {
336 AssertReturnVoidStmt(aTarget != NULL, mRC = E_FAIL);
337 mRC = mTargetCaller.rc();
338 if (FAILED(mRC))
339 return;
340 /* aParent may be NULL */
341 mRC = mParentCaller.rc();
342 if (FAILED(mRC))
343 return;
344 AssertReturnVoidStmt(aSourceMediumLockList != NULL, mRC = E_FAIL);
345 AssertReturnVoidStmt(aTargetMediumLockList != NULL, mRC = E_FAIL);
346 }
347
348 ~CloneTask()
349 {
350 if (!mfKeepSourceMediumLockList && mpSourceMediumLockList)
351 delete mpSourceMediumLockList;
352 if (!mfKeepTargetMediumLockList && mpTargetMediumLockList)
353 delete mpTargetMediumLockList;
354 }
355
356 const ComObjPtr<Medium> mTarget;
357 const ComObjPtr<Medium> mParent;
358 MediumLockList *mpSourceMediumLockList;
359 MediumLockList *mpTargetMediumLockList;
360 MediumVariant_T mVariant;
361
362private:
363 virtual HRESULT handler();
364
365 AutoCaller mTargetCaller;
366 AutoCaller mParentCaller;
367 bool mfKeepSourceMediumLockList;
368 bool mfKeepTargetMediumLockList;
369};
370
371class Medium::CompactTask : public Medium::Task
372{
373public:
374 CompactTask(Medium *aMedium,
375 Progress *aProgress,
376 MediumLockList *aMediumLockList,
377 bool fKeepMediumLockList = false)
378 : Medium::Task(aMedium, aProgress),
379 mpMediumLockList(aMediumLockList),
380 mfKeepMediumLockList(fKeepMediumLockList)
381 {
382 AssertReturnVoidStmt(aMediumLockList != NULL, mRC = E_FAIL);
383 }
384
385 ~CompactTask()
386 {
387 if (!mfKeepMediumLockList && mpMediumLockList)
388 delete mpMediumLockList;
389 }
390
391 MediumLockList *mpMediumLockList;
392
393private:
394 virtual HRESULT handler();
395
396 bool mfKeepMediumLockList;
397};
398
399class Medium::ResetTask : public Medium::Task
400{
401public:
402 ResetTask(Medium *aMedium,
403 Progress *aProgress,
404 MediumLockList *aMediumLockList,
405 bool fKeepMediumLockList = false)
406 : Medium::Task(aMedium, aProgress),
407 mpMediumLockList(aMediumLockList),
408 mfKeepMediumLockList(fKeepMediumLockList)
409 {}
410
411 ~ResetTask()
412 {
413 if (!mfKeepMediumLockList && mpMediumLockList)
414 delete mpMediumLockList;
415 }
416
417 MediumLockList *mpMediumLockList;
418
419private:
420 virtual HRESULT handler();
421
422 bool mfKeepMediumLockList;
423};
424
425class Medium::DeleteTask : public Medium::Task
426{
427public:
428 DeleteTask(Medium *aMedium,
429 Progress *aProgress,
430 MediumLockList *aMediumLockList,
431 bool fKeepMediumLockList = false)
432 : Medium::Task(aMedium, aProgress),
433 mpMediumLockList(aMediumLockList),
434 mfKeepMediumLockList(fKeepMediumLockList)
435 {}
436
437 ~DeleteTask()
438 {
439 if (!mfKeepMediumLockList && mpMediumLockList)
440 delete mpMediumLockList;
441 }
442
443 MediumLockList *mpMediumLockList;
444
445private:
446 virtual HRESULT handler();
447
448 bool mfKeepMediumLockList;
449};
450
451class Medium::MergeTask : public Medium::Task
452{
453public:
454 MergeTask(Medium *aMedium,
455 Medium *aTarget,
456 bool fMergeForward,
457 Medium *aParentForTarget,
458 const MediaList &aChildrenToReparent,
459 Progress *aProgress,
460 MediumLockList *aMediumLockList,
461 bool fKeepMediumLockList = false)
462 : Medium::Task(aMedium, aProgress),
463 mTarget(aTarget),
464 mfMergeForward(fMergeForward),
465 mParentForTarget(aParentForTarget),
466 mChildrenToReparent(aChildrenToReparent),
467 mpMediumLockList(aMediumLockList),
468 mTargetCaller(aTarget),
469 mParentForTargetCaller(aParentForTarget),
470 mfChildrenCaller(false),
471 mfKeepMediumLockList(fKeepMediumLockList)
472 {
473 AssertReturnVoidStmt(aMediumLockList != NULL, mRC = E_FAIL);
474 for (MediaList::const_iterator it = mChildrenToReparent.begin();
475 it != mChildrenToReparent.end();
476 ++it)
477 {
478 HRESULT rc2 = (*it)->addCaller();
479 if (FAILED(rc2))
480 {
481 mRC = E_FAIL;
482 for (MediaList::const_iterator it2 = mChildrenToReparent.begin();
483 it2 != it;
484 --it2)
485 {
486 (*it2)->releaseCaller();
487 }
488 return;
489 }
490 }
491 mfChildrenCaller = true;
492 }
493
494 ~MergeTask()
495 {
496 if (!mfKeepMediumLockList && mpMediumLockList)
497 delete mpMediumLockList;
498 if (mfChildrenCaller)
499 {
500 for (MediaList::const_iterator it = mChildrenToReparent.begin();
501 it != mChildrenToReparent.end();
502 ++it)
503 {
504 (*it)->releaseCaller();
505 }
506 }
507 }
508
509 const ComObjPtr<Medium> mTarget;
510 bool mfMergeForward;
511 /* When mChildrenToReparent is empty then mParentForTarget is non-null.
512 * In other words: they are used in different cases. */
513 const ComObjPtr<Medium> mParentForTarget;
514 MediaList mChildrenToReparent;
515 MediumLockList *mpMediumLockList;
516
517private:
518 virtual HRESULT handler();
519
520 AutoCaller mTargetCaller;
521 AutoCaller mParentForTargetCaller;
522 bool mfChildrenCaller;
523 bool mfKeepMediumLockList;
524};
525
526/**
527 * Thread function for time-consuming medium tasks.
528 *
529 * @param pvUser Pointer to the Medium::Task instance.
530 */
531/* static */
532DECLCALLBACK(int) Medium::Task::fntMediumTask(RTTHREAD aThread, void *pvUser)
533{
534 LogFlowFuncEnter();
535 AssertReturn(pvUser, (int)E_INVALIDARG);
536 Medium::Task *pTask = static_cast<Medium::Task *>(pvUser);
537
538 pTask->mThread = aThread;
539
540 HRESULT rc = pTask->handler();
541
542 /* complete the progress if run asynchronously */
543 if (pTask->isAsync())
544 {
545 if (!pTask->mProgress.isNull())
546 pTask->mProgress->notifyComplete(rc);
547 }
548
549 /* pTask is no longer needed, delete it. */
550 delete pTask;
551
552 LogFlowFunc(("rc=%Rhrc\n", rc));
553 LogFlowFuncLeave();
554
555 return (int)rc;
556}
557
558/**
559 * PFNVDPROGRESS callback handler for Task operations.
560 *
561 * @param pvUser Pointer to the Progress instance.
562 * @param uPercent Completetion precentage (0-100).
563 */
564/*static*/
565DECLCALLBACK(int) Medium::Task::vdProgressCall(void *pvUser, unsigned uPercent)
566{
567 Progress *that = static_cast<Progress *>(pvUser);
568
569 if (that != NULL)
570 {
571 /* update the progress object, capping it at 99% as the final percent
572 * is used for additional operations like setting the UUIDs and similar. */
573 HRESULT rc = that->SetCurrentOperationProgress(uPercent * 99 / 100);
574 if (FAILED(rc))
575 {
576 if (rc == E_FAIL)
577 return VERR_CANCELLED;
578 else
579 return VERR_INVALID_STATE;
580 }
581 }
582
583 return VINF_SUCCESS;
584}
585
586/**
587 * Implementation code for the "create base" task.
588 */
589HRESULT Medium::CreateBaseTask::handler()
590{
591 return mMedium->taskCreateBaseHandler(*this);
592}
593
594/**
595 * Implementation code for the "create diff" task.
596 */
597HRESULT Medium::CreateDiffTask::handler()
598{
599 return mMedium->taskCreateDiffHandler(*this);
600}
601
602/**
603 * Implementation code for the "clone" task.
604 */
605HRESULT Medium::CloneTask::handler()
606{
607 return mMedium->taskCloneHandler(*this);
608}
609
610/**
611 * Implementation code for the "compact" task.
612 */
613HRESULT Medium::CompactTask::handler()
614{
615 return mMedium->taskCompactHandler(*this);
616}
617
618/**
619 * Implementation code for the "reset" task.
620 */
621HRESULT Medium::ResetTask::handler()
622{
623 return mMedium->taskResetHandler(*this);
624}
625
626/**
627 * Implementation code for the "delete" task.
628 */
629HRESULT Medium::DeleteTask::handler()
630{
631 return mMedium->taskDeleteHandler(*this);
632}
633
634/**
635 * Implementation code for the "merge" task.
636 */
637HRESULT Medium::MergeTask::handler()
638{
639 return mMedium->taskMergeHandler(*this);
640}
641
642
643////////////////////////////////////////////////////////////////////////////////
644//
645// Medium constructor / destructor
646//
647////////////////////////////////////////////////////////////////////////////////
648
649DEFINE_EMPTY_CTOR_DTOR(Medium)
650
651HRESULT Medium::FinalConstruct()
652{
653 m = new Data;
654
655 /* Initialize the callbacks of the VD error interface */
656 m->vdIfCallsError.cbSize = sizeof(VDINTERFACEERROR);
657 m->vdIfCallsError.enmInterface = VDINTERFACETYPE_ERROR;
658 m->vdIfCallsError.pfnError = vdErrorCall;
659 m->vdIfCallsError.pfnMessage = NULL;
660
661 /* Initialize the callbacks of the VD config interface */
662 m->vdIfCallsConfig.cbSize = sizeof(VDINTERFACECONFIG);
663 m->vdIfCallsConfig.enmInterface = VDINTERFACETYPE_CONFIG;
664 m->vdIfCallsConfig.pfnAreKeysValid = vdConfigAreKeysValid;
665 m->vdIfCallsConfig.pfnQuerySize = vdConfigQuerySize;
666 m->vdIfCallsConfig.pfnQuery = vdConfigQuery;
667
668 /* Initialize the callbacks of the VD TCP interface (we always use the host
669 * IP stack for now) */
670 m->vdIfCallsTcpNet.cbSize = sizeof(VDINTERFACETCPNET);
671 m->vdIfCallsTcpNet.enmInterface = VDINTERFACETYPE_TCPNET;
672 m->vdIfCallsTcpNet.pfnClientConnect = RTTcpClientConnect;
673 m->vdIfCallsTcpNet.pfnClientClose = RTTcpClientClose;
674 m->vdIfCallsTcpNet.pfnSelectOne = RTTcpSelectOne;
675 m->vdIfCallsTcpNet.pfnRead = RTTcpRead;
676 m->vdIfCallsTcpNet.pfnWrite = RTTcpWrite;
677 m->vdIfCallsTcpNet.pfnFlush = RTTcpFlush;
678 m->vdIfCallsTcpNet.pfnGetLocalAddress = RTTcpGetLocalAddress;
679 m->vdIfCallsTcpNet.pfnGetPeerAddress = RTTcpGetPeerAddress;
680
681 /* Initialize the per-disk interface chain */
682 int vrc;
683 vrc = VDInterfaceAdd(&m->vdIfError,
684 "Medium::vdInterfaceError",
685 VDINTERFACETYPE_ERROR,
686 &m->vdIfCallsError, this, &m->vdDiskIfaces);
687 AssertRCReturn(vrc, E_FAIL);
688
689 vrc = VDInterfaceAdd(&m->vdIfConfig,
690 "Medium::vdInterfaceConfig",
691 VDINTERFACETYPE_CONFIG,
692 &m->vdIfCallsConfig, this, &m->vdDiskIfaces);
693 AssertRCReturn(vrc, E_FAIL);
694
695 vrc = VDInterfaceAdd(&m->vdIfTcpNet,
696 "Medium::vdInterfaceTcpNet",
697 VDINTERFACETYPE_TCPNET,
698 &m->vdIfCallsTcpNet, this, &m->vdDiskIfaces);
699 AssertRCReturn(vrc, E_FAIL);
700
701 vrc = RTSemEventMultiCreate(&m->queryInfoSem);
702 AssertRCReturn(vrc, E_FAIL);
703 vrc = RTSemEventMultiSignal(m->queryInfoSem);
704 AssertRCReturn(vrc, E_FAIL);
705
706 return S_OK;
707}
708
709void Medium::FinalRelease()
710{
711 uninit();
712
713 delete m;
714}
715
716/**
717 * Initializes the hard disk object without creating or opening an associated
718 * storage unit.
719 *
720 * For hard disks that don't have the VD_CAP_CREATE_FIXED or
721 * VD_CAP_CREATE_DYNAMIC capability (and therefore cannot be created or deleted
722 * with the means of VirtualBox) the associated storage unit is assumed to be
723 * ready for use so the state of the hard disk object will be set to Created.
724 *
725 * @param aVirtualBox VirtualBox object.
726 * @param aLocation Storage unit location.
727 * @param pfNeedsSaveSettings Optional pointer to a bool that must have been initialized to false and that will be set to true
728 * by this function if the caller should invoke VirtualBox::saveSettings() because the global settings have changed.
729 */
730HRESULT Medium::init(VirtualBox *aVirtualBox,
731 CBSTR aFormat,
732 CBSTR aLocation,
733 bool *pfNeedsSaveSettings)
734{
735 AssertReturn(aVirtualBox != NULL, E_FAIL);
736 AssertReturn(aFormat != NULL && *aFormat != '\0', E_FAIL);
737
738 /* Enclose the state transition NotReady->InInit->Ready */
739 AutoInitSpan autoInitSpan(this);
740 AssertReturn(autoInitSpan.isOk(), E_FAIL);
741
742 HRESULT rc = S_OK;
743
744 /* share VirtualBox weakly (parent remains NULL so far) */
745 unconst(m->pVirtualBox) = aVirtualBox;
746
747 /* no storage yet */
748 m->state = MediumState_NotCreated;
749
750 /* cannot be a host drive */
751 m->hostDrive = FALSE;
752
753 /* No storage unit is created yet, no need to queryInfo() */
754
755 rc = setFormat(aFormat);
756 if (FAILED(rc)) return rc;
757
758 if (m->formatObj->capabilities() & MediumFormatCapabilities_File)
759 {
760 rc = setLocation(aLocation);
761 if (FAILED(rc)) return rc;
762 }
763 else
764 {
765 rc = setLocation(aLocation);
766 if (FAILED(rc)) return rc;
767 }
768
769 if (!(m->formatObj->capabilities() & ( MediumFormatCapabilities_CreateFixed
770 | MediumFormatCapabilities_CreateDynamic))
771 )
772 {
773 /* storage for hard disks of this format can neither be explicitly
774 * created by VirtualBox nor deleted, so we place the hard disk to
775 * Created state here and also add it to the registry */
776 m->state = MediumState_Created;
777 unconst(m->id).create();
778
779 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
780 rc = m->pVirtualBox->registerHardDisk(this, pfNeedsSaveSettings);
781 }
782
783 /* Confirm a successful initialization when it's the case */
784 if (SUCCEEDED(rc))
785 autoInitSpan.setSucceeded();
786
787 return rc;
788}
789
790/**
791 * Initializes the medium object by opening the storage unit at the specified
792 * location. The enOpenMode parameter defines whether the image will be opened
793 * read/write or read-only.
794 *
795 * Note that the UUID, format and the parent of this medium will be
796 * determined when reading the medium storage unit, unless new values are
797 * specified by the parameters. If the detected or set parent is
798 * not known to VirtualBox, then this method will fail.
799 *
800 * @param aVirtualBox VirtualBox object.
801 * @param aLocation Storage unit location.
802 * @param enOpenMode Whether to open the image read/write or read-only.
803 * @param aDeviceType Device type of medium.
804 * @param aSetImageId Whether to set the image UUID or not.
805 * @param aImageId New image UUID if @aSetId is true. Empty string means
806 * create a new UUID, and a zero UUID is invalid.
807 * @param aSetParentId Whether to set the parent UUID or not.
808 * @param aParentId New parent UUID if @aSetParentId is true. Empty string
809 * means create a new UUID, and a zero UUID is valid.
810 */
811HRESULT Medium::init(VirtualBox *aVirtualBox,
812 CBSTR aLocation,
813 HDDOpenMode enOpenMode,
814 DeviceType_T aDeviceType,
815 BOOL aSetImageId,
816 const Guid &aImageId,
817 BOOL aSetParentId,
818 const Guid &aParentId)
819{
820 AssertReturn(aVirtualBox, E_INVALIDARG);
821 AssertReturn(aLocation, E_INVALIDARG);
822
823 /* Enclose the state transition NotReady->InInit->Ready */
824 AutoInitSpan autoInitSpan(this);
825 AssertReturn(autoInitSpan.isOk(), E_FAIL);
826
827 HRESULT rc = S_OK;
828
829 /* share VirtualBox weakly (parent remains NULL so far) */
830 unconst(m->pVirtualBox) = aVirtualBox;
831
832 /* there must be a storage unit */
833 m->state = MediumState_Created;
834
835 /* remember device type for correct unregistering later */
836 m->devType = aDeviceType;
837
838 /* cannot be a host drive */
839 m->hostDrive = FALSE;
840
841 /* remember the open mode (defaults to ReadWrite) */
842 m->hddOpenMode = enOpenMode;
843
844 if (aDeviceType == DeviceType_HardDisk)
845 rc = setLocation(aLocation);
846 else
847 rc = setLocation(aLocation, "RAW");
848 if (FAILED(rc)) return rc;
849
850 /* save the new uuid values, will be used by queryInfo() */
851 m->setImageId = aSetImageId;
852 unconst(m->imageId) = aImageId;
853 m->setParentId = aSetParentId;
854 unconst(m->parentId) = aParentId;
855
856 /* get all the information about the medium from the storage unit */
857 rc = queryInfo();
858
859 if (SUCCEEDED(rc))
860 {
861 /* if the storage unit is not accessible, it's not acceptable for the
862 * newly opened media so convert this into an error */
863 if (m->state == MediumState_Inaccessible)
864 {
865 Assert(!m->strLastAccessError.isEmpty());
866 rc = setError(E_FAIL, m->strLastAccessError.c_str());
867 }
868 else
869 {
870 AssertReturn(!m->id.isEmpty(), E_FAIL);
871
872 /* storage format must be detected by queryInfo() if the medium is accessible */
873 AssertReturn(!m->strFormat.isEmpty(), E_FAIL);
874 }
875 }
876
877 /* Confirm a successful initialization when it's the case */
878 if (SUCCEEDED(rc))
879 autoInitSpan.setSucceeded();
880
881 return rc;
882}
883
884/**
885 * Initializes the medium object by loading its data from the given settings
886 * node. In this mode, the image will always be opened read/write.
887 *
888 * @param aVirtualBox VirtualBox object.
889 * @param aParent Parent medium disk or NULL for a root (base) medium.
890 * @param aDeviceType Device type of the medium.
891 * @param aNode Configuration settings.
892 *
893 * @note Locks VirtualBox for writing, the medium tree for writing.
894 */
895HRESULT Medium::init(VirtualBox *aVirtualBox,
896 Medium *aParent,
897 DeviceType_T aDeviceType,
898 const settings::Medium &data)
899{
900 using namespace settings;
901
902 AssertReturn(aVirtualBox, E_INVALIDARG);
903
904 /* Enclose the state transition NotReady->InInit->Ready */
905 AutoInitSpan autoInitSpan(this);
906 AssertReturn(autoInitSpan.isOk(), E_FAIL);
907
908 HRESULT rc = S_OK;
909
910 /* share VirtualBox and parent weakly */
911 unconst(m->pVirtualBox) = aVirtualBox;
912
913 /* register with VirtualBox/parent early, since uninit() will
914 * unconditionally unregister on failure */
915 if (aParent)
916 {
917 // differencing image: add to parent
918 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
919 m->pParent = aParent;
920 aParent->m->llChildren.push_back(this);
921 }
922
923 /* see below why we don't call queryInfo() (and therefore treat the medium
924 * as inaccessible for now */
925 m->state = MediumState_Inaccessible;
926 m->strLastAccessError = tr("Accessibility check was not yet performed");
927
928 /* required */
929 unconst(m->id) = data.uuid;
930
931 /* assume not a host drive */
932 m->hostDrive = FALSE;
933
934 /* optional */
935 m->strDescription = data.strDescription;
936
937 /* required */
938 if (aDeviceType == DeviceType_HardDisk)
939 {
940 AssertReturn(!data.strFormat.isEmpty(), E_FAIL);
941 rc = setFormat(Bstr(data.strFormat));
942 if (FAILED(rc)) return rc;
943 }
944 else
945 {
946 /// @todo handle host drive settings here as well?
947 if (!data.strFormat.isEmpty())
948 rc = setFormat(Bstr(data.strFormat));
949 else
950 rc = setFormat(Bstr("RAW"));
951 if (FAILED(rc)) return rc;
952 }
953
954 /* optional, only for diffs, default is false;
955 * we can only auto-reset diff images, so they
956 * must not have a parent */
957 if (aParent != NULL)
958 m->autoReset = data.fAutoReset;
959 else
960 m->autoReset = false;
961
962 /* properties (after setting the format as it populates the map). Note that
963 * if some properties are not supported but preseint in the settings file,
964 * they will still be read and accessible (for possible backward
965 * compatibility; we can also clean them up from the XML upon next
966 * XML format version change if we wish) */
967 for (settings::PropertiesMap::const_iterator it = data.properties.begin();
968 it != data.properties.end(); ++it)
969 {
970 const Utf8Str &name = it->first;
971 const Utf8Str &value = it->second;
972 m->properties[Bstr(name)] = Bstr(value);
973 }
974
975 /* required */
976 rc = setLocation(data.strLocation);
977 if (FAILED(rc)) return rc;
978
979 if (aDeviceType == DeviceType_HardDisk)
980 {
981 /* type is only for base hard disks */
982 if (m->pParent.isNull())
983 m->type = data.hdType;
984 }
985 else
986 m->type = MediumType_Writethrough;
987
988 /* remember device type for correct unregistering later */
989 m->devType = aDeviceType;
990
991 LogFlowThisFunc(("m->strLocationFull='%s', m->strFormat=%s, m->id={%RTuuid}\n",
992 m->strLocationFull.raw(), m->strFormat.raw(), m->id.raw()));
993
994 /* Don't call queryInfo() for registered media to prevent the calling
995 * thread (i.e. the VirtualBox server startup thread) from an unexpected
996 * freeze but mark it as initially inaccessible instead. The vital UUID,
997 * location and format properties are read from the registry file above; to
998 * get the actual state and the rest of the data, the user will have to call
999 * COMGETTER(State). */
1000
1001 AutoWriteLock treeLock(aVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1002
1003 /* load all children */
1004 for (settings::MediaList::const_iterator it = data.llChildren.begin();
1005 it != data.llChildren.end();
1006 ++it)
1007 {
1008 const settings::Medium &med = *it;
1009
1010 ComObjPtr<Medium> pHD;
1011 pHD.createObject();
1012 rc = pHD->init(aVirtualBox,
1013 this, // parent
1014 aDeviceType,
1015 med); // child data
1016 if (FAILED(rc)) break;
1017
1018 rc = m->pVirtualBox->registerHardDisk(pHD, NULL /*pfNeedsSaveSettings*/);
1019 if (FAILED(rc)) break;
1020 }
1021
1022 /* Confirm a successful initialization when it's the case */
1023 if (SUCCEEDED(rc))
1024 autoInitSpan.setSucceeded();
1025
1026 return rc;
1027}
1028
1029/**
1030 * Initializes the medium object by providing the host drive information.
1031 * Not used for anything but the host floppy/host DVD case.
1032 *
1033 * @todo optimize all callers to avoid reconstructing objects with the same
1034 * information over and over again - in the typical case each VM referring to
1035 * a particular host drive has its own instance.
1036 *
1037 * @param aVirtualBox VirtualBox object.
1038 * @param aDeviceType Device type of the medium.
1039 * @param aLocation Location of the host drive.
1040 * @param aDescription Comment for this host drive.
1041 *
1042 * @note Locks VirtualBox lock for writing.
1043 */
1044HRESULT Medium::init(VirtualBox *aVirtualBox,
1045 DeviceType_T aDeviceType,
1046 CBSTR aLocation,
1047 CBSTR aDescription)
1048{
1049 ComAssertRet(aDeviceType == DeviceType_DVD || aDeviceType == DeviceType_Floppy, E_INVALIDARG);
1050 ComAssertRet(aLocation, E_INVALIDARG);
1051
1052 /* Enclose the state transition NotReady->InInit->Ready */
1053 AutoInitSpan autoInitSpan(this);
1054 AssertReturn(autoInitSpan.isOk(), E_FAIL);
1055
1056 /* share VirtualBox weakly (parent remains NULL so far) */
1057 unconst(m->pVirtualBox) = aVirtualBox;
1058
1059 /* fake up a UUID which is unique, but also reproducible */
1060 RTUUID uuid;
1061 RTUuidClear(&uuid);
1062 if (aDeviceType == DeviceType_DVD)
1063 memcpy(&uuid.au8[0], "DVD", 3);
1064 else
1065 memcpy(&uuid.au8[0], "FD", 2);
1066 /* use device name, adjusted to the end of uuid, shortened if necessary */
1067 Utf8Str loc(aLocation);
1068 size_t cbLocation = strlen(loc.raw());
1069 if (cbLocation > 12)
1070 memcpy(&uuid.au8[4], loc.raw() + (cbLocation - 12), 12);
1071 else
1072 memcpy(&uuid.au8[4 + 12 - cbLocation], loc.raw(), cbLocation);
1073 unconst(m->id) = uuid;
1074
1075 m->type = MediumType_Writethrough;
1076 m->devType = aDeviceType;
1077 m->state = MediumState_Created;
1078 m->hostDrive = true;
1079 HRESULT rc = setFormat(Bstr("RAW"));
1080 if (FAILED(rc)) return rc;
1081 rc = setLocation(aLocation);
1082 if (FAILED(rc)) return rc;
1083 m->strDescription = aDescription;
1084
1085/// @todo generate uuid (similarly to host network interface uuid) from location and device type
1086
1087 autoInitSpan.setSucceeded();
1088 return S_OK;
1089}
1090
1091/**
1092 * Uninitializes the instance.
1093 *
1094 * Called either from FinalRelease() or by the parent when it gets destroyed.
1095 *
1096 * @note All children of this hard disk get uninitialized by calling their
1097 * uninit() methods.
1098 *
1099 * @note Caller must hold the tree lock of the medium tree this medium is on.
1100 */
1101void Medium::uninit()
1102{
1103 /* Enclose the state transition Ready->InUninit->NotReady */
1104 AutoUninitSpan autoUninitSpan(this);
1105 if (autoUninitSpan.uninitDone())
1106 return;
1107
1108 if (!m->formatObj.isNull())
1109 {
1110 /* remove the caller reference we added in setFormat() */
1111 m->formatObj->releaseCaller();
1112 m->formatObj.setNull();
1113 }
1114
1115 if (m->state == MediumState_Deleting)
1116 {
1117 /* we are being uninitialized after've been deleted by merge.
1118 * Reparenting has already been done so don't touch it here (we are
1119 * now orphans and removeDependentChild() will assert) */
1120 Assert(m->pParent.isNull());
1121 }
1122 else
1123 {
1124 MediaList::iterator it;
1125 for (it = m->llChildren.begin();
1126 it != m->llChildren.end();
1127 ++it)
1128 {
1129 Medium *pChild = *it;
1130 pChild->m->pParent.setNull();
1131 pChild->uninit();
1132 }
1133 m->llChildren.clear(); // this unsets all the ComPtrs and probably calls delete
1134
1135 if (m->pParent)
1136 {
1137 // this is a differencing disk: then remove it from the parent's children list
1138 deparent();
1139 }
1140 }
1141
1142 RTSemEventMultiSignal(m->queryInfoSem);
1143 RTSemEventMultiDestroy(m->queryInfoSem);
1144 m->queryInfoSem = NIL_RTSEMEVENTMULTI;
1145
1146 unconst(m->pVirtualBox) = NULL;
1147}
1148
1149/**
1150 * Internal helper that removes "this" from the list of children of its
1151 * parent. Used in uninit() and other places when reparenting is necessary.
1152 *
1153 * The caller must hold the hard disk tree lock!
1154 */
1155void Medium::deparent()
1156{
1157 MediaList &llParent = m->pParent->m->llChildren;
1158 for (MediaList::iterator it = llParent.begin();
1159 it != llParent.end();
1160 ++it)
1161 {
1162 Medium *pParentsChild = *it;
1163 if (this == pParentsChild)
1164 {
1165 llParent.erase(it);
1166 break;
1167 }
1168 }
1169 m->pParent.setNull();
1170}
1171
1172/**
1173 * Internal helper that removes "this" from the list of children of its
1174 * parent. Used in uninit() and other places when reparenting is necessary.
1175 *
1176 * The caller must hold the hard disk tree lock!
1177 */
1178void Medium::setParent(const ComObjPtr<Medium> &pParent)
1179{
1180 m->pParent = pParent;
1181 if (pParent)
1182 pParent->m->llChildren.push_back(this);
1183}
1184
1185
1186////////////////////////////////////////////////////////////////////////////////
1187//
1188// IMedium public methods
1189//
1190////////////////////////////////////////////////////////////////////////////////
1191
1192STDMETHODIMP Medium::COMGETTER(Id)(BSTR *aId)
1193{
1194 CheckComArgOutPointerValid(aId);
1195
1196 AutoCaller autoCaller(this);
1197 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1198
1199 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1200
1201 m->id.toUtf16().cloneTo(aId);
1202
1203 return S_OK;
1204}
1205
1206STDMETHODIMP Medium::COMGETTER(Description)(BSTR *aDescription)
1207{
1208 CheckComArgOutPointerValid(aDescription);
1209
1210 AutoCaller autoCaller(this);
1211 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1212
1213 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1214
1215 m->strDescription.cloneTo(aDescription);
1216
1217 return S_OK;
1218}
1219
1220STDMETHODIMP Medium::COMSETTER(Description)(IN_BSTR aDescription)
1221{
1222 AutoCaller autoCaller(this);
1223 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1224
1225// AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1226
1227 /// @todo update m->description and save the global registry (and local
1228 /// registries of portable VMs referring to this medium), this will also
1229 /// require to add the mRegistered flag to data
1230
1231 NOREF(aDescription);
1232
1233 ReturnComNotImplemented();
1234}
1235
1236STDMETHODIMP Medium::COMGETTER(State)(MediumState_T *aState)
1237{
1238 CheckComArgOutPointerValid(aState);
1239
1240 AutoCaller autoCaller(this);
1241 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1242
1243 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1244 *aState = m->state;
1245
1246 return S_OK;
1247}
1248
1249
1250STDMETHODIMP Medium::COMGETTER(Location)(BSTR *aLocation)
1251{
1252 CheckComArgOutPointerValid(aLocation);
1253
1254 AutoCaller autoCaller(this);
1255 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1256
1257 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1258
1259 m->strLocationFull.cloneTo(aLocation);
1260
1261 return S_OK;
1262}
1263
1264STDMETHODIMP Medium::COMSETTER(Location)(IN_BSTR aLocation)
1265{
1266 CheckComArgStrNotEmptyOrNull(aLocation);
1267
1268 AutoCaller autoCaller(this);
1269 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1270
1271 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1272
1273 /// @todo NEWMEDIA for file names, add the default extension if no extension
1274 /// is present (using the information from the VD backend which also implies
1275 /// that one more parameter should be passed to setLocation() requesting
1276 /// that functionality since it is only allwed when called from this method
1277
1278 /// @todo NEWMEDIA rename the file and set m->location on success, then save
1279 /// the global registry (and local registries of portable VMs referring to
1280 /// this medium), this will also require to add the mRegistered flag to data
1281
1282 ReturnComNotImplemented();
1283}
1284
1285STDMETHODIMP Medium::COMGETTER(Name)(BSTR *aName)
1286{
1287 CheckComArgOutPointerValid(aName);
1288
1289 AutoCaller autoCaller(this);
1290 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1291
1292 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1293
1294 getName().cloneTo(aName);
1295
1296 return S_OK;
1297}
1298
1299STDMETHODIMP Medium::COMGETTER(DeviceType)(DeviceType_T *aDeviceType)
1300{
1301 CheckComArgOutPointerValid(aDeviceType);
1302
1303 AutoCaller autoCaller(this);
1304 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1305
1306 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1307
1308 *aDeviceType = m->devType;
1309
1310 return S_OK;
1311}
1312
1313STDMETHODIMP Medium::COMGETTER(HostDrive)(BOOL *aHostDrive)
1314{
1315 CheckComArgOutPointerValid(aHostDrive);
1316
1317 AutoCaller autoCaller(this);
1318 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1319
1320 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1321
1322 *aHostDrive = m->hostDrive;
1323
1324 return S_OK;
1325}
1326
1327STDMETHODIMP Medium::COMGETTER(Size)(ULONG64 *aSize)
1328{
1329 CheckComArgOutPointerValid(aSize);
1330
1331 AutoCaller autoCaller(this);
1332 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1333
1334 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1335
1336 *aSize = m->size;
1337
1338 return S_OK;
1339}
1340
1341STDMETHODIMP Medium::COMGETTER(Format)(BSTR *aFormat)
1342{
1343 CheckComArgOutPointerValid(aFormat);
1344
1345 AutoCaller autoCaller(this);
1346 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1347
1348 /* no need to lock, m->strFormat is const */
1349 m->strFormat.cloneTo(aFormat);
1350
1351 return S_OK;
1352}
1353
1354STDMETHODIMP Medium::COMGETTER(MediumFormat)(IMediumFormat **aMediumFormat)
1355{
1356 CheckComArgOutPointerValid(aMediumFormat);
1357
1358 AutoCaller autoCaller(this);
1359 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1360
1361 /* no need to lock, m->formatObj is const */
1362 m->formatObj.queryInterfaceTo(aMediumFormat);
1363
1364 return S_OK;
1365}
1366
1367STDMETHODIMP Medium::COMGETTER(Type)(MediumType_T *aType)
1368{
1369 CheckComArgOutPointerValid(aType);
1370
1371 AutoCaller autoCaller(this);
1372 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1373
1374 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1375
1376 *aType = m->type;
1377
1378 return S_OK;
1379}
1380
1381STDMETHODIMP Medium::COMSETTER(Type)(MediumType_T aType)
1382{
1383 AutoCaller autoCaller(this);
1384 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1385
1386 // we access mParent and members
1387 AutoMultiWriteLock2 mlock(&m->pVirtualBox->getMediaTreeLockHandle(), this->lockHandle() COMMA_LOCKVAL_SRC_POS);
1388
1389 switch (m->state)
1390 {
1391 case MediumState_Created:
1392 case MediumState_Inaccessible:
1393 break;
1394 default:
1395 return setStateError();
1396 }
1397
1398 /** @todo implement this case later */
1399 CheckComArgExpr(aType, aType != MediumType_Shareable);
1400
1401 if (m->type == aType)
1402 {
1403 /* Nothing to do */
1404 return S_OK;
1405 }
1406
1407 /* cannot change the type of a differencing hard disk */
1408 if (m->pParent)
1409 return setError(E_FAIL,
1410 tr("Cannot change the type of hard disk '%s' because it is a differencing hard disk"),
1411 m->strLocationFull.raw());
1412
1413 /* cannot change the type of a hard disk being in use by more than one VM */
1414 if (m->backRefs.size() > 1)
1415 return setError(E_FAIL,
1416 tr("Cannot change the type of hard disk '%s' because it is attached to %d virtual machines"),
1417 m->strLocationFull.raw(), m->backRefs.size());
1418
1419 switch (aType)
1420 {
1421 case MediumType_Normal:
1422 case MediumType_Immutable:
1423 {
1424 /* normal can be easily converted to immutable and vice versa even
1425 * if they have children as long as they are not attached to any
1426 * machine themselves */
1427 break;
1428 }
1429 case MediumType_Writethrough:
1430 case MediumType_Shareable:
1431 {
1432 /* cannot change to writethrough or shareable if there are children */
1433 if (getChildren().size() != 0)
1434 return setError(E_FAIL,
1435 tr("Cannot change type for hard disk '%s' since it has %d child hard disk(s)"),
1436 m->strLocationFull.raw(), getChildren().size());
1437 break;
1438 }
1439 default:
1440 AssertFailedReturn(E_FAIL);
1441 }
1442
1443 m->type = aType;
1444
1445 // saveSettings needs vbox lock
1446 ComObjPtr<VirtualBox> pVirtualBox(m->pVirtualBox);
1447 mlock.leave();
1448 AutoWriteLock alock(m->pVirtualBox COMMA_LOCKVAL_SRC_POS);
1449
1450 HRESULT rc = pVirtualBox->saveSettings();
1451
1452 return rc;
1453}
1454
1455STDMETHODIMP Medium::COMGETTER(Parent)(IMedium **aParent)
1456{
1457 CheckComArgOutPointerValid(aParent);
1458
1459 AutoCaller autoCaller(this);
1460 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1461
1462 /* we access mParent */
1463 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1464
1465 m->pParent.queryInterfaceTo(aParent);
1466
1467 return S_OK;
1468}
1469
1470STDMETHODIMP Medium::COMGETTER(Children)(ComSafeArrayOut(IMedium *, aChildren))
1471{
1472 CheckComArgOutSafeArrayPointerValid(aChildren);
1473
1474 AutoCaller autoCaller(this);
1475 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1476
1477 /* we access children */
1478 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1479
1480 SafeIfaceArray<IMedium> children(this->getChildren());
1481 children.detachTo(ComSafeArrayOutArg(aChildren));
1482
1483 return S_OK;
1484}
1485
1486STDMETHODIMP Medium::COMGETTER(Base)(IMedium **aBase)
1487{
1488 CheckComArgOutPointerValid(aBase);
1489
1490 /* base() will do callers/locking */
1491
1492 getBase().queryInterfaceTo(aBase);
1493
1494 return S_OK;
1495}
1496
1497STDMETHODIMP Medium::COMGETTER(ReadOnly)(BOOL *aReadOnly)
1498{
1499 CheckComArgOutPointerValid(aReadOnly);
1500
1501 AutoCaller autoCaller(this);
1502 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1503
1504 /* isRadOnly() will do locking */
1505
1506 *aReadOnly = isReadOnly();
1507
1508 return S_OK;
1509}
1510
1511STDMETHODIMP Medium::COMGETTER(LogicalSize)(ULONG64 *aLogicalSize)
1512{
1513 CheckComArgOutPointerValid(aLogicalSize);
1514
1515 {
1516 AutoCaller autoCaller(this);
1517 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1518
1519 /* we access mParent */
1520 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1521
1522 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1523
1524 if (m->pParent.isNull())
1525 {
1526 *aLogicalSize = m->logicalSize;
1527
1528 return S_OK;
1529 }
1530 }
1531
1532 /* We assume that some backend may decide to return a meaningless value in
1533 * response to VDGetSize() for differencing hard disks and therefore
1534 * always ask the base hard disk ourselves. */
1535
1536 /* base() will do callers/locking */
1537
1538 return getBase()->COMGETTER(LogicalSize)(aLogicalSize);
1539}
1540
1541STDMETHODIMP Medium::COMGETTER(AutoReset)(BOOL *aAutoReset)
1542{
1543 CheckComArgOutPointerValid(aAutoReset);
1544
1545 AutoCaller autoCaller(this);
1546 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1547
1548 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1549
1550 if (m->pParent)
1551 *aAutoReset = FALSE;
1552
1553 *aAutoReset = m->autoReset;
1554
1555 return S_OK;
1556}
1557
1558STDMETHODIMP Medium::COMSETTER(AutoReset)(BOOL aAutoReset)
1559{
1560 AutoCaller autoCaller(this);
1561 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1562
1563 /* VirtualBox::saveSettings() needs a write lock */
1564 AutoMultiWriteLock2 alock(m->pVirtualBox, this COMMA_LOCKVAL_SRC_POS);
1565
1566 if (m->pParent.isNull())
1567 return setError(VBOX_E_NOT_SUPPORTED,
1568 tr("Hard disk '%s' is not differencing"),
1569 m->strLocationFull.raw());
1570
1571 if (m->autoReset != aAutoReset)
1572 {
1573 m->autoReset = aAutoReset;
1574
1575 return m->pVirtualBox->saveSettings();
1576 }
1577
1578 return S_OK;
1579}
1580STDMETHODIMP Medium::COMGETTER(LastAccessError)(BSTR *aLastAccessError)
1581{
1582 CheckComArgOutPointerValid(aLastAccessError);
1583
1584 AutoCaller autoCaller(this);
1585 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1586
1587 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1588
1589 m->strLastAccessError.cloneTo(aLastAccessError);
1590
1591 return S_OK;
1592}
1593
1594STDMETHODIMP Medium::COMGETTER(MachineIds)(ComSafeArrayOut(BSTR,aMachineIds))
1595{
1596 CheckComArgOutSafeArrayPointerValid(aMachineIds);
1597
1598 AutoCaller autoCaller(this);
1599 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1600
1601 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1602
1603 com::SafeArray<BSTR> machineIds;
1604
1605 if (m->backRefs.size() != 0)
1606 {
1607 machineIds.reset(m->backRefs.size());
1608
1609 size_t i = 0;
1610 for (BackRefList::const_iterator it = m->backRefs.begin();
1611 it != m->backRefs.end(); ++it, ++i)
1612 {
1613 it->machineId.toUtf16().detachTo(&machineIds[i]);
1614 }
1615 }
1616
1617 machineIds.detachTo(ComSafeArrayOutArg(aMachineIds));
1618
1619 return S_OK;
1620}
1621
1622STDMETHODIMP Medium::RefreshState(MediumState_T *aState)
1623{
1624 CheckComArgOutPointerValid(aState);
1625
1626 AutoCaller autoCaller(this);
1627 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1628
1629 /* queryInfo() locks this for writing. */
1630 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1631
1632 HRESULT rc = S_OK;
1633
1634 switch (m->state)
1635 {
1636 case MediumState_Created:
1637 case MediumState_Inaccessible:
1638 case MediumState_LockedRead:
1639 {
1640 rc = queryInfo();
1641 break;
1642 }
1643 default:
1644 break;
1645 }
1646
1647 *aState = m->state;
1648
1649 return rc;
1650}
1651
1652STDMETHODIMP Medium::GetSnapshotIds(IN_BSTR aMachineId,
1653 ComSafeArrayOut(BSTR, aSnapshotIds))
1654{
1655 CheckComArgExpr(aMachineId, Guid(aMachineId).isEmpty() == false);
1656 CheckComArgOutSafeArrayPointerValid(aSnapshotIds);
1657
1658 AutoCaller autoCaller(this);
1659 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1660
1661 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1662
1663 com::SafeArray<BSTR> snapshotIds;
1664
1665 Guid id(aMachineId);
1666 for (BackRefList::const_iterator it = m->backRefs.begin();
1667 it != m->backRefs.end(); ++it)
1668 {
1669 if (it->machineId == id)
1670 {
1671 size_t size = it->llSnapshotIds.size();
1672
1673 /* if the medium is attached to the machine in the current state, we
1674 * return its ID as the first element of the array */
1675 if (it->fInCurState)
1676 ++size;
1677
1678 if (size > 0)
1679 {
1680 snapshotIds.reset(size);
1681
1682 size_t j = 0;
1683 if (it->fInCurState)
1684 it->machineId.toUtf16().detachTo(&snapshotIds[j++]);
1685
1686 for (BackRef::GuidList::const_iterator jt = it->llSnapshotIds.begin();
1687 jt != it->llSnapshotIds.end();
1688 ++jt, ++j)
1689 {
1690 (*jt).toUtf16().detachTo(&snapshotIds[j]);
1691 }
1692 }
1693
1694 break;
1695 }
1696 }
1697
1698 snapshotIds.detachTo(ComSafeArrayOutArg(aSnapshotIds));
1699
1700 return S_OK;
1701}
1702
1703/**
1704 * @note @a aState may be NULL if the state value is not needed (only for
1705 * in-process calls).
1706 */
1707STDMETHODIMP Medium::LockRead(MediumState_T *aState)
1708{
1709 AutoCaller autoCaller(this);
1710 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1711
1712 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1713
1714 /* Wait for a concurrently running queryInfo() to complete */
1715 while (m->queryInfoRunning)
1716 {
1717 alock.leave();
1718 RTSemEventMultiWait(m->queryInfoSem, RT_INDEFINITE_WAIT);
1719 alock.enter();
1720 }
1721
1722 /* return the current state before */
1723 if (aState)
1724 *aState = m->state;
1725
1726 HRESULT rc = S_OK;
1727
1728 switch (m->state)
1729 {
1730 case MediumState_Created:
1731 case MediumState_Inaccessible:
1732 case MediumState_LockedRead:
1733 {
1734 ++m->readers;
1735
1736 ComAssertMsgBreak(m->readers != 0, ("Counter overflow"), rc = E_FAIL);
1737
1738 /* Remember pre-lock state */
1739 if (m->state != MediumState_LockedRead)
1740 m->preLockState = m->state;
1741
1742 LogFlowThisFunc(("Okay - prev state=%d readers=%d\n", m->state, m->readers));
1743 m->state = MediumState_LockedRead;
1744
1745 break;
1746 }
1747 default:
1748 {
1749 LogFlowThisFunc(("Failing - state=%d\n", m->state));
1750 rc = setStateError();
1751 break;
1752 }
1753 }
1754
1755 return rc;
1756}
1757
1758/**
1759 * @note @a aState may be NULL if the state value is not needed (only for
1760 * in-process calls).
1761 */
1762STDMETHODIMP Medium::UnlockRead(MediumState_T *aState)
1763{
1764 AutoCaller autoCaller(this);
1765 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1766
1767 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1768
1769 HRESULT rc = S_OK;
1770
1771 switch (m->state)
1772 {
1773 case MediumState_LockedRead:
1774 {
1775 Assert(m->readers != 0);
1776 --m->readers;
1777
1778 /* Reset the state after the last reader */
1779 if (m->readers == 0)
1780 {
1781 m->state = m->preLockState;
1782 /* There are cases where we inject the deleting state into
1783 * a medium locked for reading. Make sure #unmarkForDeletion()
1784 * gets the right state afterwards. */
1785 if (m->preLockState == MediumState_Deleting)
1786 m->preLockState = MediumState_Created;
1787 }
1788
1789 LogFlowThisFunc(("new state=%d\n", m->state));
1790 break;
1791 }
1792 default:
1793 {
1794 LogFlowThisFunc(("Failing - state=%d\n", m->state));
1795 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
1796 tr("Medium '%s' is not locked for reading"),
1797 m->strLocationFull.raw());
1798 break;
1799 }
1800 }
1801
1802 /* return the current state after */
1803 if (aState)
1804 *aState = m->state;
1805
1806 return rc;
1807}
1808
1809/**
1810 * @note @a aState may be NULL if the state value is not needed (only for
1811 * in-process calls).
1812 */
1813STDMETHODIMP Medium::LockWrite(MediumState_T *aState)
1814{
1815 AutoCaller autoCaller(this);
1816 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1817
1818 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1819
1820 /* Wait for a concurrently running queryInfo() to complete */
1821 while (m->queryInfoRunning)
1822 {
1823 alock.leave();
1824 RTSemEventMultiWait(m->queryInfoSem, RT_INDEFINITE_WAIT);
1825 alock.enter();
1826 }
1827
1828 /* return the current state before */
1829 if (aState)
1830 *aState = m->state;
1831
1832 HRESULT rc = S_OK;
1833
1834 switch (m->state)
1835 {
1836 case MediumState_Created:
1837 case MediumState_Inaccessible:
1838 {
1839 m->preLockState = m->state;
1840
1841 LogFlowThisFunc(("Okay - prev state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
1842 m->state = MediumState_LockedWrite;
1843 break;
1844 }
1845 default:
1846 {
1847 LogFlowThisFunc(("Failing - state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
1848 rc = setStateError();
1849 break;
1850 }
1851 }
1852
1853 return rc;
1854}
1855
1856/**
1857 * @note @a aState may be NULL if the state value is not needed (only for
1858 * in-process calls).
1859 */
1860STDMETHODIMP Medium::UnlockWrite(MediumState_T *aState)
1861{
1862 AutoCaller autoCaller(this);
1863 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1864
1865 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1866
1867 HRESULT rc = S_OK;
1868
1869 switch (m->state)
1870 {
1871 case MediumState_LockedWrite:
1872 {
1873 m->state = m->preLockState;
1874 /* There are cases where we inject the deleting state into
1875 * a medium locked for writing. Make sure #unmarkForDeletion()
1876 * gets the right state afterwards. */
1877 if (m->preLockState == MediumState_Deleting)
1878 m->preLockState = MediumState_Created;
1879 LogFlowThisFunc(("new state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
1880 break;
1881 }
1882 default:
1883 {
1884 LogFlowThisFunc(("Failing - state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
1885 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
1886 tr("Medium '%s' is not locked for writing"),
1887 m->strLocationFull.raw());
1888 break;
1889 }
1890 }
1891
1892 /* return the current state after */
1893 if (aState)
1894 *aState = m->state;
1895
1896 return rc;
1897}
1898
1899STDMETHODIMP Medium::Close()
1900{
1901 // we're accessing parent/child and backrefs, so lock the tree first, then ourselves
1902 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
1903 this->lockHandle()
1904 COMMA_LOCKVAL_SRC_POS);
1905
1906 bool wasCreated = true;
1907 bool fNeedsSaveSettings = false;
1908
1909 switch (m->state)
1910 {
1911 case MediumState_NotCreated:
1912 wasCreated = false;
1913 break;
1914 case MediumState_Created:
1915 case MediumState_Inaccessible:
1916 break;
1917 default:
1918 return setStateError();
1919 }
1920
1921 if (m->backRefs.size() != 0)
1922 return setError(VBOX_E_OBJECT_IN_USE,
1923 tr("Medium '%s' is attached to %d virtual machines"),
1924 m->strLocationFull.raw(), m->backRefs.size());
1925
1926 /* perform extra media-dependent close checks */
1927 HRESULT rc = canClose();
1928 if (FAILED(rc)) return rc;
1929
1930 if (wasCreated)
1931 {
1932 /* remove from the list of known media before performing actual
1933 * uninitialization (to keep the media registry consistent on
1934 * failure to do so) */
1935 rc = unregisterWithVirtualBox(&fNeedsSaveSettings);
1936 if (FAILED(rc)) return rc;
1937 }
1938
1939 // make a copy of VirtualBox pointer which gets nulled by uninit()
1940 ComObjPtr<VirtualBox> pVirtualBox(m->pVirtualBox);
1941
1942 /* Keep the locks held until after uninit, as otherwise the consistency
1943 * of the medium tree cannot be guaranteed. */
1944 uninit();
1945
1946 multilock.release();
1947
1948 if (fNeedsSaveSettings)
1949 {
1950 AutoWriteLock vboxlock(pVirtualBox COMMA_LOCKVAL_SRC_POS);
1951 pVirtualBox->saveSettings();
1952 }
1953
1954 return S_OK;
1955}
1956
1957STDMETHODIMP Medium::GetProperty(IN_BSTR aName, BSTR *aValue)
1958{
1959 CheckComArgStrNotEmptyOrNull(aName);
1960 CheckComArgOutPointerValid(aValue);
1961
1962 AutoCaller autoCaller(this);
1963 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1964
1965 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1966
1967 Data::PropertyMap::const_iterator it = m->properties.find(Bstr(aName));
1968 if (it == m->properties.end())
1969 return setError(VBOX_E_OBJECT_NOT_FOUND,
1970 tr("Property '%ls' does not exist"), aName);
1971
1972 it->second.cloneTo(aValue);
1973
1974 return S_OK;
1975}
1976
1977STDMETHODIMP Medium::SetProperty(IN_BSTR aName, IN_BSTR aValue)
1978{
1979 CheckComArgStrNotEmptyOrNull(aName);
1980
1981 AutoCaller autoCaller(this);
1982 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1983
1984 /* VirtualBox::saveSettings() needs a write lock */
1985 AutoMultiWriteLock2 alock(m->pVirtualBox, this COMMA_LOCKVAL_SRC_POS);
1986
1987 switch (m->state)
1988 {
1989 case MediumState_Created:
1990 case MediumState_Inaccessible:
1991 break;
1992 default:
1993 return setStateError();
1994 }
1995
1996 Data::PropertyMap::iterator it = m->properties.find(Bstr(aName));
1997 if (it == m->properties.end())
1998 return setError(VBOX_E_OBJECT_NOT_FOUND,
1999 tr("Property '%ls' does not exist"),
2000 aName);
2001
2002 if (aValue && !*aValue)
2003 it->second = (const char *)NULL;
2004 else
2005 it->second = aValue;
2006
2007 HRESULT rc = m->pVirtualBox->saveSettings();
2008
2009 return rc;
2010}
2011
2012STDMETHODIMP Medium::GetProperties(IN_BSTR aNames,
2013 ComSafeArrayOut(BSTR, aReturnNames),
2014 ComSafeArrayOut(BSTR, aReturnValues))
2015{
2016 CheckComArgOutSafeArrayPointerValid(aReturnNames);
2017 CheckComArgOutSafeArrayPointerValid(aReturnValues);
2018
2019 AutoCaller autoCaller(this);
2020 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2021
2022 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2023
2024 /// @todo make use of aNames according to the documentation
2025 NOREF(aNames);
2026
2027 com::SafeArray<BSTR> names(m->properties.size());
2028 com::SafeArray<BSTR> values(m->properties.size());
2029 size_t i = 0;
2030
2031 for (Data::PropertyMap::const_iterator it = m->properties.begin();
2032 it != m->properties.end();
2033 ++it)
2034 {
2035 it->first.cloneTo(&names[i]);
2036 it->second.cloneTo(&values[i]);
2037 ++i;
2038 }
2039
2040 names.detachTo(ComSafeArrayOutArg(aReturnNames));
2041 values.detachTo(ComSafeArrayOutArg(aReturnValues));
2042
2043 return S_OK;
2044}
2045
2046STDMETHODIMP Medium::SetProperties(ComSafeArrayIn(IN_BSTR, aNames),
2047 ComSafeArrayIn(IN_BSTR, aValues))
2048{
2049 CheckComArgSafeArrayNotNull(aNames);
2050 CheckComArgSafeArrayNotNull(aValues);
2051
2052 AutoCaller autoCaller(this);
2053 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2054
2055 /* VirtualBox::saveSettings() needs a write lock */
2056 AutoMultiWriteLock2 alock(m->pVirtualBox, this COMMA_LOCKVAL_SRC_POS);
2057
2058 com::SafeArray<IN_BSTR> names(ComSafeArrayInArg(aNames));
2059 com::SafeArray<IN_BSTR> values(ComSafeArrayInArg(aValues));
2060
2061 /* first pass: validate names */
2062 for (size_t i = 0;
2063 i < names.size();
2064 ++i)
2065 {
2066 if (m->properties.find(Bstr(names[i])) == m->properties.end())
2067 return setError(VBOX_E_OBJECT_NOT_FOUND,
2068 tr("Property '%ls' does not exist"), names[i]);
2069 }
2070
2071 /* second pass: assign */
2072 for (size_t i = 0;
2073 i < names.size();
2074 ++i)
2075 {
2076 Data::PropertyMap::iterator it = m->properties.find(Bstr(names[i]));
2077 AssertReturn(it != m->properties.end(), E_FAIL);
2078
2079 if (values[i] && !*values[i])
2080 it->second = (const char *)NULL;
2081 else
2082 it->second = values[i];
2083 }
2084
2085 HRESULT rc = m->pVirtualBox->saveSettings();
2086
2087 return rc;
2088}
2089
2090STDMETHODIMP Medium::CreateBaseStorage(ULONG64 aLogicalSize,
2091 MediumVariant_T aVariant,
2092 IProgress **aProgress)
2093{
2094 CheckComArgOutPointerValid(aProgress);
2095
2096 AutoCaller autoCaller(this);
2097 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2098
2099 HRESULT rc = S_OK;
2100 ComObjPtr <Progress> pProgress;
2101 Medium::Task *pTask = NULL;
2102
2103 try
2104 {
2105 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2106
2107 aVariant = (MediumVariant_T)((unsigned)aVariant & (unsigned)~MediumVariant_Diff);
2108 if ( !(aVariant & MediumVariant_Fixed)
2109 && !(m->formatObj->capabilities() & MediumFormatCapabilities_CreateDynamic))
2110 throw setError(VBOX_E_NOT_SUPPORTED,
2111 tr("Hard disk format '%s' does not support dynamic storage creation"),
2112 m->strFormat.raw());
2113 if ( (aVariant & MediumVariant_Fixed)
2114 && !(m->formatObj->capabilities() & MediumFormatCapabilities_CreateDynamic))
2115 throw setError(VBOX_E_NOT_SUPPORTED,
2116 tr("Hard disk format '%s' does not support fixed storage creation"),
2117 m->strFormat.raw());
2118
2119 if (m->state != MediumState_NotCreated)
2120 throw setStateError();
2121
2122 pProgress.createObject();
2123 rc = pProgress->init(m->pVirtualBox,
2124 static_cast<IMedium*>(this),
2125 (aVariant & MediumVariant_Fixed)
2126 ? BstrFmt(tr("Creating fixed hard disk storage unit '%s'"), m->strLocationFull.raw())
2127 : BstrFmt(tr("Creating dynamic hard disk storage unit '%s'"), m->strLocationFull.raw()),
2128 TRUE /* aCancelable */);
2129 if (FAILED(rc))
2130 throw rc;
2131
2132 /* setup task object to carry out the operation asynchronously */
2133 pTask = new Medium::CreateBaseTask(this, pProgress, aLogicalSize,
2134 aVariant);
2135 rc = pTask->rc();
2136 AssertComRC(rc);
2137 if (FAILED(rc))
2138 throw rc;
2139
2140 m->state = MediumState_Creating;
2141 }
2142 catch (HRESULT aRC) { rc = aRC; }
2143
2144 if (SUCCEEDED(rc))
2145 {
2146 rc = startThread(pTask);
2147
2148 if (SUCCEEDED(rc))
2149 pProgress.queryInterfaceTo(aProgress);
2150 }
2151 else if (pTask != NULL)
2152 delete pTask;
2153
2154 return rc;
2155}
2156
2157STDMETHODIMP Medium::DeleteStorage(IProgress **aProgress)
2158{
2159 CheckComArgOutPointerValid(aProgress);
2160
2161 AutoCaller autoCaller(this);
2162 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2163
2164 bool fNeedsSaveSettings = false;
2165 ComObjPtr <Progress> pProgress;
2166
2167 HRESULT rc = deleteStorage(&pProgress,
2168 false /* aWait */,
2169 &fNeedsSaveSettings);
2170 if (fNeedsSaveSettings)
2171 {
2172 AutoWriteLock vboxlock(m->pVirtualBox COMMA_LOCKVAL_SRC_POS);
2173 m->pVirtualBox->saveSettings();
2174 }
2175
2176 if (SUCCEEDED(rc))
2177 pProgress.queryInterfaceTo(aProgress);
2178
2179 return rc;
2180}
2181
2182STDMETHODIMP Medium::CreateDiffStorage(IMedium *aTarget,
2183 MediumVariant_T aVariant,
2184 IProgress **aProgress)
2185{
2186 CheckComArgNotNull(aTarget);
2187 CheckComArgOutPointerValid(aProgress);
2188
2189 AutoCaller autoCaller(this);
2190 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2191
2192 ComObjPtr<Medium> diff = static_cast<Medium*>(aTarget);
2193
2194 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2195
2196 if (m->type == MediumType_Writethrough)
2197 return setError(E_FAIL,
2198 tr("Hard disk '%s' is Writethrough"),
2199 m->strLocationFull.raw());
2200
2201 /* Apply the normal locking logic to the entire chain. */
2202 MediumLockList *pMediumLockList(new MediumLockList());
2203 HRESULT rc = diff->createMediumLockList(true /* fFailIfInaccessible */,
2204 true /* fMediumWritable */,
2205 this,
2206 *pMediumLockList);
2207 if (FAILED(rc))
2208 {
2209 delete pMediumLockList;
2210 return rc;
2211 }
2212
2213 ComObjPtr <Progress> pProgress;
2214
2215 rc = createDiffStorage(diff, aVariant, pMediumLockList, &pProgress,
2216 false /* aWait */, NULL /* pfNeedsSaveSettings*/);
2217 if (FAILED(rc))
2218 delete pMediumLockList;
2219 else
2220 pProgress.queryInterfaceTo(aProgress);
2221
2222 return rc;
2223}
2224
2225STDMETHODIMP Medium::MergeTo(IMedium *aTarget, IProgress **aProgress)
2226{
2227 CheckComArgNotNull(aTarget);
2228 CheckComArgOutPointerValid(aProgress);
2229 ComAssertRet(aTarget != this, E_INVALIDARG);
2230
2231 AutoCaller autoCaller(this);
2232 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2233
2234 ComObjPtr<Medium> pTarget = static_cast<Medium*>(aTarget);
2235
2236 bool fMergeForward = false;
2237 ComObjPtr<Medium> pParentForTarget;
2238 MediaList childrenToReparent;
2239 MediumLockList *pMediumLockList = NULL;
2240
2241 HRESULT rc = S_OK;
2242
2243 rc = prepareMergeTo(pTarget, NULL, NULL, true, fMergeForward,
2244 pParentForTarget, childrenToReparent, pMediumLockList);
2245 if (FAILED(rc)) return rc;
2246
2247 ComObjPtr <Progress> pProgress;
2248
2249 rc = mergeTo(pTarget, fMergeForward, pParentForTarget, childrenToReparent,
2250 pMediumLockList, &pProgress, false /* aWait */,
2251 NULL /* pfNeedsSaveSettings */);
2252 if (FAILED(rc))
2253 cancelMergeTo(childrenToReparent, pMediumLockList);
2254 else
2255 pProgress.queryInterfaceTo(aProgress);
2256
2257 return rc;
2258}
2259
2260STDMETHODIMP Medium::CloneTo(IMedium *aTarget,
2261 MediumVariant_T aVariant,
2262 IMedium *aParent,
2263 IProgress **aProgress)
2264{
2265 CheckComArgNotNull(aTarget);
2266 CheckComArgOutPointerValid(aProgress);
2267 ComAssertRet(aTarget != this, E_INVALIDARG);
2268
2269 AutoCaller autoCaller(this);
2270 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2271
2272 ComObjPtr<Medium> pTarget = static_cast<Medium*>(aTarget);
2273 ComObjPtr<Medium> pParent;
2274 if (aParent)
2275 pParent = static_cast<Medium*>(aParent);
2276
2277 HRESULT rc = S_OK;
2278 ComObjPtr<Progress> pProgress;
2279 Medium::Task *pTask = NULL;
2280
2281 try
2282 {
2283 // locking: we need the tree lock first because we access parent pointers
2284 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
2285 // and we need to write-lock the images involved
2286 AutoMultiWriteLock3 alock(this, pTarget, pParent COMMA_LOCKVAL_SRC_POS);
2287
2288 if ( pTarget->m->state != MediumState_NotCreated
2289 && pTarget->m->state != MediumState_Created)
2290 throw pTarget->setStateError();
2291
2292 /* Build the source lock list. */
2293 MediumLockList *pSourceMediumLockList(new MediumLockList());
2294 rc = createMediumLockList(true /* fFailIfInaccessible */,
2295 false /* fMediumWritable */,
2296 NULL,
2297 *pSourceMediumLockList);
2298 if (FAILED(rc))
2299 {
2300 delete pSourceMediumLockList;
2301 throw rc;
2302 }
2303
2304 /* Build the target lock list (including the to-be parent chain). */
2305 MediumLockList *pTargetMediumLockList(new MediumLockList());
2306 rc = pTarget->createMediumLockList(true /* fFailIfInaccessible */,
2307 true /* fMediumWritable */,
2308 pParent,
2309 *pTargetMediumLockList);
2310 if (FAILED(rc))
2311 {
2312 delete pSourceMediumLockList;
2313 delete pTargetMediumLockList;
2314 throw rc;
2315 }
2316
2317 rc = pSourceMediumLockList->Lock();
2318 if (FAILED(rc))
2319 {
2320 delete pSourceMediumLockList;
2321 delete pTargetMediumLockList;
2322 throw setError(rc,
2323 tr("Failed to lock source media '%ls'"),
2324 getLocationFull().raw());
2325 }
2326 rc = pTargetMediumLockList->Lock();
2327 if (FAILED(rc))
2328 {
2329 delete pSourceMediumLockList;
2330 delete pTargetMediumLockList;
2331 throw setError(rc,
2332 tr("Failed to lock target media '%ls'"),
2333 pTarget->getLocationFull().raw());
2334 }
2335
2336 pProgress.createObject();
2337 rc = pProgress->init(m->pVirtualBox,
2338 static_cast <IMedium *>(this),
2339 BstrFmt(tr("Creating clone hard disk '%s'"), pTarget->m->strLocationFull.raw()),
2340 TRUE /* aCancelable */);
2341 if (FAILED(rc))
2342 {
2343 delete pSourceMediumLockList;
2344 delete pTargetMediumLockList;
2345 throw rc;
2346 }
2347
2348 /* setup task object to carry out the operation asynchronously */
2349 pTask = new Medium::CloneTask(this, pProgress, pTarget, aVariant,
2350 pParent, pSourceMediumLockList,
2351 pTargetMediumLockList);
2352 rc = pTask->rc();
2353 AssertComRC(rc);
2354 if (FAILED(rc))
2355 throw rc;
2356
2357 if (pTarget->m->state == MediumState_NotCreated)
2358 pTarget->m->state = MediumState_Creating;
2359 }
2360 catch (HRESULT aRC) { rc = aRC; }
2361
2362 if (SUCCEEDED(rc))
2363 {
2364 rc = startThread(pTask);
2365
2366 if (SUCCEEDED(rc))
2367 pProgress.queryInterfaceTo(aProgress);
2368 }
2369 else if (pTask != NULL)
2370 delete pTask;
2371
2372 return rc;
2373}
2374
2375STDMETHODIMP Medium::Compact(IProgress **aProgress)
2376{
2377 CheckComArgOutPointerValid(aProgress);
2378
2379 AutoCaller autoCaller(this);
2380 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2381
2382 HRESULT rc = S_OK;
2383 ComObjPtr <Progress> pProgress;
2384 Medium::Task *pTask = NULL;
2385
2386 try
2387 {
2388 /* We need to lock both the current object, and the tree lock (would
2389 * cause a lock order violation otherwise) for createMediumLockList. */
2390 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
2391 this->lockHandle()
2392 COMMA_LOCKVAL_SRC_POS);
2393
2394 /* Build the medium lock list. */
2395 MediumLockList *pMediumLockList(new MediumLockList());
2396 rc = createMediumLockList(true /* fFailIfInaccessible */ ,
2397 true /* fMediumWritable */,
2398 NULL,
2399 *pMediumLockList);
2400 if (FAILED(rc))
2401 {
2402 delete pMediumLockList;
2403 throw rc;
2404 }
2405
2406 rc = pMediumLockList->Lock();
2407 if (FAILED(rc))
2408 {
2409 delete pMediumLockList;
2410 throw setError(rc,
2411 tr("Failed to lock media when compacting '%ls'"),
2412 getLocationFull().raw());
2413 }
2414
2415 pProgress.createObject();
2416 rc = pProgress->init(m->pVirtualBox,
2417 static_cast <IMedium *>(this),
2418 BstrFmt(tr("Compacting hard disk '%s'"), m->strLocationFull.raw()),
2419 TRUE /* aCancelable */);
2420 if (FAILED(rc))
2421 {
2422 delete pMediumLockList;
2423 throw rc;
2424 }
2425
2426 /* setup task object to carry out the operation asynchronously */
2427 pTask = new Medium::CompactTask(this, pProgress, pMediumLockList);
2428 rc = pTask->rc();
2429 AssertComRC(rc);
2430 if (FAILED(rc))
2431 throw rc;
2432 }
2433 catch (HRESULT aRC) { rc = aRC; }
2434
2435 if (SUCCEEDED(rc))
2436 {
2437 rc = startThread(pTask);
2438
2439 if (SUCCEEDED(rc))
2440 pProgress.queryInterfaceTo(aProgress);
2441 }
2442 else if (pTask != NULL)
2443 delete pTask;
2444
2445 return rc;
2446}
2447
2448STDMETHODIMP Medium::Resize(ULONG64 aLogicalSize, IProgress **aProgress)
2449{
2450 CheckComArgOutPointerValid(aProgress);
2451
2452 AutoCaller autoCaller(this);
2453 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2454
2455 NOREF(aLogicalSize);
2456 NOREF(aProgress);
2457 ReturnComNotImplemented();
2458}
2459
2460STDMETHODIMP Medium::Reset(IProgress **aProgress)
2461{
2462 CheckComArgOutPointerValid(aProgress);
2463
2464 AutoCaller autoCaller(this);
2465 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2466
2467 HRESULT rc = S_OK;
2468 ComObjPtr <Progress> pProgress;
2469 Medium::Task *pTask = NULL;
2470
2471 try
2472 {
2473 /* canClose() needs the tree lock */
2474 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
2475 this->lockHandle()
2476 COMMA_LOCKVAL_SRC_POS);
2477
2478 LogFlowThisFunc(("ENTER for medium %s\n", m->strLocationFull.c_str()));
2479
2480 if (m->pParent.isNull())
2481 throw setError(VBOX_E_NOT_SUPPORTED,
2482 tr("Hard disk '%s' is not differencing"),
2483 m->strLocationFull.raw());
2484
2485 rc = canClose();
2486 if (FAILED(rc))
2487 throw rc;
2488
2489 /* Build the medium lock list. */
2490 MediumLockList *pMediumLockList(new MediumLockList());
2491 rc = createMediumLockList(true /* fFailIfInaccessible */,
2492 true /* fMediumWritable */,
2493 NULL,
2494 *pMediumLockList);
2495 if (FAILED(rc))
2496 {
2497 delete pMediumLockList;
2498 throw rc;
2499 }
2500
2501 rc = pMediumLockList->Lock();
2502 if (FAILED(rc))
2503 {
2504 delete pMediumLockList;
2505 throw setError(rc,
2506 tr("Failed to lock media when resetting '%ls'"),
2507 getLocationFull().raw());
2508 }
2509
2510 pProgress.createObject();
2511 rc = pProgress->init(m->pVirtualBox,
2512 static_cast<IMedium*>(this),
2513 BstrFmt(tr("Resetting differencing hard disk '%s'"), m->strLocationFull.raw()),
2514 FALSE /* aCancelable */);
2515 if (FAILED(rc))
2516 throw rc;
2517
2518 /* setup task object to carry out the operation asynchronously */
2519 pTask = new Medium::ResetTask(this, pProgress, pMediumLockList);
2520 rc = pTask->rc();
2521 AssertComRC(rc);
2522 if (FAILED(rc))
2523 throw rc;
2524 }
2525 catch (HRESULT aRC) { rc = aRC; }
2526
2527 if (SUCCEEDED(rc))
2528 {
2529 rc = startThread(pTask);
2530
2531 if (SUCCEEDED(rc))
2532 pProgress.queryInterfaceTo(aProgress);
2533 }
2534 else
2535 {
2536 /* Note: on success, the task will unlock this */
2537 {
2538 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2539 HRESULT rc2 = UnlockWrite(NULL);
2540 AssertComRC(rc2);
2541 }
2542 if (pTask != NULL)
2543 delete pTask;
2544 }
2545
2546 LogFlowThisFunc(("LEAVE, rc=%Rhrc\n", rc));
2547
2548 return rc;
2549}
2550
2551////////////////////////////////////////////////////////////////////////////////
2552//
2553// Medium internal methods
2554//
2555////////////////////////////////////////////////////////////////////////////////
2556
2557/**
2558 * Internal method to return the medium's parent medium. Must have caller + locking!
2559 * @return
2560 */
2561const ComObjPtr<Medium>& Medium::getParent() const
2562{
2563 return m->pParent;
2564}
2565
2566/**
2567 * Internal method to return the medium's list of child media. Must have caller + locking!
2568 * @return
2569 */
2570const MediaList& Medium::getChildren() const
2571{
2572 return m->llChildren;
2573}
2574
2575/**
2576 * Internal method to return the medium's GUID. Must have caller + locking!
2577 * @return
2578 */
2579const Guid& Medium::getId() const
2580{
2581 return m->id;
2582}
2583
2584/**
2585 * Internal method to return the medium's GUID. Must have caller + locking!
2586 * @return
2587 */
2588MediumState_T Medium::getState() const
2589{
2590 return m->state;
2591}
2592
2593/**
2594 * Internal method to return the medium's location. Must have caller + locking!
2595 * @return
2596 */
2597const Utf8Str& Medium::getLocation() const
2598{
2599 return m->strLocation;
2600}
2601
2602/**
2603 * Internal method to return the medium's full location. Must have caller + locking!
2604 * @return
2605 */
2606const Utf8Str& Medium::getLocationFull() const
2607{
2608 return m->strLocationFull;
2609}
2610
2611/**
2612 * Internal method to return the medium's format string. Must have caller + locking!
2613 * @return
2614 */
2615const Utf8Str& Medium::getFormat() const
2616{
2617 return m->strFormat;
2618}
2619
2620/**
2621 * Internal method to return the medium's format object. Must have caller + locking!
2622 * @return
2623 */
2624const ComObjPtr<MediumFormat> & Medium::getMediumFormat() const
2625{
2626 return m->formatObj;
2627}
2628
2629/**
2630 * Internal method to return the medium's size. Must have caller + locking!
2631 * @return
2632 */
2633uint64_t Medium::getSize() const
2634{
2635 return m->size;
2636}
2637
2638/**
2639 * Adds the given machine and optionally the snapshot to the list of the objects
2640 * this image is attached to.
2641 *
2642 * @param aMachineId Machine ID.
2643 * @param aSnapshotId Snapshot ID; when non-empty, adds a snapshot attachment.
2644 */
2645HRESULT Medium::attachTo(const Guid &aMachineId,
2646 const Guid &aSnapshotId /*= Guid::Empty*/)
2647{
2648 AssertReturn(!aMachineId.isEmpty(), E_FAIL);
2649
2650 LogFlowThisFunc(("ENTER, aMachineId: {%RTuuid}, aSnapshotId: {%RTuuid}\n", aMachineId.raw(), aSnapshotId.raw()));
2651
2652 AutoCaller autoCaller(this);
2653 AssertComRCReturnRC(autoCaller.rc());
2654
2655 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2656
2657 switch (m->state)
2658 {
2659 case MediumState_Created:
2660 case MediumState_Inaccessible:
2661 case MediumState_LockedRead:
2662 case MediumState_LockedWrite:
2663 break;
2664
2665 default:
2666 return setStateError();
2667 }
2668
2669 if (m->numCreateDiffTasks > 0)
2670 return setError(E_FAIL,
2671 tr("Cannot attach hard disk '%s' {%RTuuid}: %u differencing child hard disk(s) are being created"),
2672 m->strLocationFull.raw(),
2673 m->id.raw(),
2674 m->numCreateDiffTasks);
2675
2676 BackRefList::iterator it = std::find_if(m->backRefs.begin(),
2677 m->backRefs.end(),
2678 BackRef::EqualsTo(aMachineId));
2679 if (it == m->backRefs.end())
2680 {
2681 BackRef ref(aMachineId, aSnapshotId);
2682 m->backRefs.push_back(ref);
2683
2684 return S_OK;
2685 }
2686
2687 // if the caller has not supplied a snapshot ID, then we're attaching
2688 // to a machine a medium which represents the machine's current state,
2689 // so set the flag
2690 if (aSnapshotId.isEmpty())
2691 {
2692 /* sanity: no duplicate attachments */
2693 AssertReturn(!it->fInCurState, E_FAIL);
2694 it->fInCurState = true;
2695
2696 return S_OK;
2697 }
2698
2699 // otherwise: a snapshot medium is being attached
2700
2701 /* sanity: no duplicate attachments */
2702 for (BackRef::GuidList::const_iterator jt = it->llSnapshotIds.begin();
2703 jt != it->llSnapshotIds.end();
2704 ++jt)
2705 {
2706 const Guid &idOldSnapshot = *jt;
2707
2708 if (idOldSnapshot == aSnapshotId)
2709 {
2710#ifdef DEBUG
2711 dumpBackRefs();
2712#endif
2713 return setError(E_FAIL,
2714 tr("Cannot attach medium '%s' {%RTuuid} from snapshot '%RTuuid': medium is already in use by this snapshot!"),
2715 m->strLocationFull.raw(),
2716 m->id.raw(),
2717 aSnapshotId.raw(),
2718 idOldSnapshot.raw());
2719 }
2720 }
2721
2722 it->llSnapshotIds.push_back(aSnapshotId);
2723 it->fInCurState = false;
2724
2725 LogFlowThisFuncLeave();
2726
2727 return S_OK;
2728}
2729
2730/**
2731 * Removes the given machine and optionally the snapshot from the list of the
2732 * objects this image is attached to.
2733 *
2734 * @param aMachineId Machine ID.
2735 * @param aSnapshotId Snapshot ID; when non-empty, removes the snapshot
2736 * attachment.
2737 */
2738HRESULT Medium::detachFrom(const Guid &aMachineId,
2739 const Guid &aSnapshotId /*= Guid::Empty*/)
2740{
2741 AssertReturn(!aMachineId.isEmpty(), E_FAIL);
2742
2743 AutoCaller autoCaller(this);
2744 AssertComRCReturnRC(autoCaller.rc());
2745
2746 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2747
2748 BackRefList::iterator it =
2749 std::find_if(m->backRefs.begin(), m->backRefs.end(),
2750 BackRef::EqualsTo(aMachineId));
2751 AssertReturn(it != m->backRefs.end(), E_FAIL);
2752
2753 if (aSnapshotId.isEmpty())
2754 {
2755 /* remove the current state attachment */
2756 it->fInCurState = false;
2757 }
2758 else
2759 {
2760 /* remove the snapshot attachment */
2761 BackRef::GuidList::iterator jt =
2762 std::find(it->llSnapshotIds.begin(), it->llSnapshotIds.end(), aSnapshotId);
2763
2764 AssertReturn(jt != it->llSnapshotIds.end(), E_FAIL);
2765 it->llSnapshotIds.erase(jt);
2766 }
2767
2768 /* if the backref becomes empty, remove it */
2769 if (it->fInCurState == false && it->llSnapshotIds.size() == 0)
2770 m->backRefs.erase(it);
2771
2772 return S_OK;
2773}
2774
2775/**
2776 * Internal method to return the medium's list of backrefs. Must have caller + locking!
2777 * @return
2778 */
2779const Guid* Medium::getFirstMachineBackrefId() const
2780{
2781 if (!m->backRefs.size())
2782 return NULL;
2783
2784 return &m->backRefs.front().machineId;
2785}
2786
2787const Guid* Medium::getFirstMachineBackrefSnapshotId() const
2788{
2789 if (!m->backRefs.size())
2790 return NULL;
2791
2792 const BackRef &ref = m->backRefs.front();
2793 if (!ref.llSnapshotIds.size())
2794 return NULL;
2795
2796 return &ref.llSnapshotIds.front();
2797}
2798
2799#ifdef DEBUG
2800/**
2801 * Debugging helper that gets called after VirtualBox initialization that writes all
2802 * machine backreferences to the debug log.
2803 */
2804void Medium::dumpBackRefs()
2805{
2806 AutoCaller autoCaller(this);
2807 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2808
2809 LogFlowThisFunc(("Dumping backrefs for medium '%s':\n", m->strLocationFull.raw()));
2810
2811 for (BackRefList::iterator it2 = m->backRefs.begin();
2812 it2 != m->backRefs.end();
2813 ++it2)
2814 {
2815 const BackRef &ref = *it2;
2816 LogFlowThisFunc((" Backref from machine {%RTuuid} (fInCurState: %d)\n", ref.machineId.raw(), ref.fInCurState));
2817
2818 for (BackRef::GuidList::const_iterator jt2 = it2->llSnapshotIds.begin();
2819 jt2 != it2->llSnapshotIds.end();
2820 ++jt2)
2821 {
2822 const Guid &id = *jt2;
2823 LogFlowThisFunc((" Backref from snapshot {%RTuuid}\n", id.raw()));
2824 }
2825 }
2826}
2827#endif
2828
2829/**
2830 * Checks if the given change of \a aOldPath to \a aNewPath affects the location
2831 * of this media and updates it if necessary to reflect the new location.
2832 *
2833 * @param aOldPath Old path (full).
2834 * @param aNewPath New path (full).
2835 *
2836 * @note Locks this object for writing.
2837 */
2838HRESULT Medium::updatePath(const char *aOldPath, const char *aNewPath)
2839{
2840 AssertReturn(aOldPath, E_FAIL);
2841 AssertReturn(aNewPath, E_FAIL);
2842
2843 AutoCaller autoCaller(this);
2844 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2845
2846 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2847
2848 LogFlowThisFunc(("locationFull.before='%s'\n", m->strLocationFull.raw()));
2849
2850 const char *pcszMediumPath = m->strLocationFull.c_str();
2851
2852 if (RTPathStartsWith(pcszMediumPath, aOldPath))
2853 {
2854 Utf8Str newPath = Utf8StrFmt("%s%s",
2855 aNewPath,
2856 pcszMediumPath + strlen(aOldPath));
2857 Utf8Str path = newPath;
2858 m->pVirtualBox->calculateRelativePath(path, path);
2859 unconst(m->strLocationFull) = newPath;
2860 unconst(m->strLocation) = path;
2861
2862 LogFlowThisFunc(("locationFull.after='%s'\n", m->strLocationFull.raw()));
2863 }
2864
2865 return S_OK;
2866}
2867
2868/**
2869 * Checks if the given change of \a aOldPath to \a aNewPath affects the location
2870 * of this hard disk or any its child and updates the paths if necessary to
2871 * reflect the new location.
2872 *
2873 * @param aOldPath Old path (full).
2874 * @param aNewPath New path (full).
2875 *
2876 * @note Locks the medium tree for reading, this object and all children for writing.
2877 */
2878void Medium::updatePaths(const char *aOldPath, const char *aNewPath)
2879{
2880 AssertReturnVoid(aOldPath);
2881 AssertReturnVoid(aNewPath);
2882
2883 AutoCaller autoCaller(this);
2884 AssertComRCReturnVoid(autoCaller.rc());
2885
2886 /* we access children() */
2887 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
2888
2889 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2890
2891 updatePath(aOldPath, aNewPath);
2892
2893 /* update paths of all children */
2894 for (MediaList::const_iterator it = getChildren().begin();
2895 it != getChildren().end();
2896 ++it)
2897 {
2898 (*it)->updatePaths(aOldPath, aNewPath);
2899 }
2900}
2901
2902/**
2903 * Returns the base hard disk of the hard disk chain this hard disk is part of.
2904 *
2905 * The base hard disk is found by walking up the parent-child relationship axis.
2906 * If the hard disk doesn't have a parent (i.e. it's a base hard disk), it
2907 * returns itself in response to this method.
2908 *
2909 * @param aLevel Where to store the number of ancestors of this hard disk
2910 * (zero for the base), may be @c NULL.
2911 *
2912 * @note Locks medium tree for reading.
2913 */
2914ComObjPtr<Medium> Medium::getBase(uint32_t *aLevel /*= NULL*/)
2915{
2916 ComObjPtr<Medium> pBase;
2917 uint32_t level;
2918
2919 AutoCaller autoCaller(this);
2920 AssertReturn(autoCaller.isOk(), pBase);
2921
2922 /* we access mParent */
2923 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
2924
2925 pBase = this;
2926 level = 0;
2927
2928 if (m->pParent)
2929 {
2930 for (;;)
2931 {
2932 AutoCaller baseCaller(pBase);
2933 AssertReturn(baseCaller.isOk(), pBase);
2934
2935 if (pBase->m->pParent.isNull())
2936 break;
2937
2938 pBase = pBase->m->pParent;
2939 ++level;
2940 }
2941 }
2942
2943 if (aLevel != NULL)
2944 *aLevel = level;
2945
2946 return pBase;
2947}
2948
2949/**
2950 * Returns @c true if this hard disk cannot be modified because it has
2951 * dependants (children) or is part of the snapshot. Related to the hard disk
2952 * type and posterity, not to the current media state.
2953 *
2954 * @note Locks this object and medium tree for reading.
2955 */
2956bool Medium::isReadOnly()
2957{
2958 AutoCaller autoCaller(this);
2959 AssertComRCReturn(autoCaller.rc(), false);
2960
2961 /* we access children */
2962 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
2963
2964 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2965
2966 switch (m->type)
2967 {
2968 case MediumType_Normal:
2969 {
2970 if (getChildren().size() != 0)
2971 return true;
2972
2973 for (BackRefList::const_iterator it = m->backRefs.begin();
2974 it != m->backRefs.end(); ++it)
2975 if (it->llSnapshotIds.size() != 0)
2976 return true;
2977
2978 return false;
2979 }
2980 case MediumType_Immutable:
2981 return true;
2982 case MediumType_Writethrough:
2983 case MediumType_Shareable:
2984 return false;
2985 default:
2986 break;
2987 }
2988
2989 AssertFailedReturn(false);
2990}
2991
2992/**
2993 * Saves hard disk data by appending a new <HardDisk> child node to the given
2994 * parent node which can be either <HardDisks> or <HardDisk>.
2995 *
2996 * @param data Settings struct to be updated.
2997 *
2998 * @note Locks this object, medium tree and children for reading.
2999 */
3000HRESULT Medium::saveSettings(settings::Medium &data)
3001{
3002 AutoCaller autoCaller(this);
3003 if (FAILED(autoCaller.rc())) return autoCaller.rc();
3004
3005 /* we access mParent */
3006 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3007
3008 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3009
3010 data.uuid = m->id;
3011 data.strLocation = m->strLocation;
3012 data.strFormat = m->strFormat;
3013
3014 /* optional, only for diffs, default is false */
3015 if (m->pParent)
3016 data.fAutoReset = !!m->autoReset;
3017 else
3018 data.fAutoReset = false;
3019
3020 /* optional */
3021 data.strDescription = m->strDescription;
3022
3023 /* optional properties */
3024 data.properties.clear();
3025 for (Data::PropertyMap::const_iterator it = m->properties.begin();
3026 it != m->properties.end();
3027 ++it)
3028 {
3029 /* only save properties that have non-default values */
3030 if (!it->second.isEmpty())
3031 {
3032 Utf8Str name = it->first;
3033 Utf8Str value = it->second;
3034 data.properties[name] = value;
3035 }
3036 }
3037
3038 /* only for base hard disks */
3039 if (m->pParent.isNull())
3040 data.hdType = m->type;
3041
3042 /* save all children */
3043 for (MediaList::const_iterator it = getChildren().begin();
3044 it != getChildren().end();
3045 ++it)
3046 {
3047 settings::Medium med;
3048 HRESULT rc = (*it)->saveSettings(med);
3049 AssertComRCReturnRC(rc);
3050 data.llChildren.push_back(med);
3051 }
3052
3053 return S_OK;
3054}
3055
3056/**
3057 * Compares the location of this hard disk to the given location.
3058 *
3059 * The comparison takes the location details into account. For example, if the
3060 * location is a file in the host's filesystem, a case insensitive comparison
3061 * will be performed for case insensitive filesystems.
3062 *
3063 * @param aLocation Location to compare to (as is).
3064 * @param aResult Where to store the result of comparison: 0 if locations
3065 * are equal, 1 if this object's location is greater than
3066 * the specified location, and -1 otherwise.
3067 */
3068HRESULT Medium::compareLocationTo(const char *aLocation, int &aResult)
3069{
3070 AutoCaller autoCaller(this);
3071 AssertComRCReturnRC(autoCaller.rc());
3072
3073 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3074
3075 Utf8Str locationFull(m->strLocationFull);
3076
3077 /// @todo NEWMEDIA delegate the comparison to the backend?
3078
3079 if (m->formatObj->capabilities() & MediumFormatCapabilities_File)
3080 {
3081 Utf8Str location(aLocation);
3082
3083 /* For locations represented by files, append the default path if
3084 * only the name is given, and then get the full path. */
3085 if (!RTPathHavePath(aLocation))
3086 {
3087 location = Utf8StrFmt("%s%c%s",
3088 m->pVirtualBox->getDefaultHardDiskFolder().raw(),
3089 RTPATH_DELIMITER,
3090 aLocation);
3091 }
3092
3093 int vrc = m->pVirtualBox->calculateFullPath(location, location);
3094 if (RT_FAILURE(vrc))
3095 return setError(E_FAIL,
3096 tr("Invalid hard disk storage file location '%s' (%Rrc)"),
3097 location.raw(),
3098 vrc);
3099
3100 aResult = RTPathCompare(locationFull.c_str(), location.c_str());
3101 }
3102 else
3103 aResult = locationFull.compare(aLocation);
3104
3105 return S_OK;
3106}
3107
3108/**
3109 * Constructs a medium lock list for this medium. The lock is not taken.
3110 *
3111 * @note Locks the medium tree for reading.
3112 *
3113 * @param fMediumWritable Whether to associate a write lock with this medium.
3114 * @param pToBeParent Medium which will become the parent of this medium.
3115 * @param mediumLockList Where to store the resulting list.
3116 */
3117HRESULT Medium::createMediumLockList(bool fFailIfInaccessible,
3118 bool fMediumWritable,
3119 Medium *pToBeParent,
3120 MediumLockList &mediumLockList)
3121{
3122 HRESULT rc = S_OK;
3123
3124 /* we access parent medium objects */
3125 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3126
3127 /* paranoid sanity checking if the medium has a to-be parent medium */
3128 if (pToBeParent)
3129 {
3130 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3131 ComAssertRet(getParent().isNull(), E_FAIL);
3132 ComAssertRet(getChildren().size() == 0, E_FAIL);
3133 }
3134
3135 ErrorInfoKeeper eik;
3136 MultiResult mrc(S_OK);
3137
3138 ComObjPtr<Medium> pMedium = this;
3139 while (!pMedium.isNull())
3140 {
3141 // need write lock for RefreshState if medium is inaccessible
3142 AutoWriteLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
3143
3144 /* Accessibility check must be first, otherwise locking interferes
3145 * with getting the medium state. Lock lists are not created for
3146 * fun, and thus getting the image status is no luxury. */
3147 MediumState_T mediumState = pMedium->getState();
3148 if (mediumState == MediumState_Inaccessible)
3149 {
3150 rc = pMedium->RefreshState(&mediumState);
3151 if (FAILED(rc)) return rc;
3152
3153 if (mediumState == MediumState_Inaccessible)
3154 {
3155 // ignore inaccessible ISO images and silently return S_OK,
3156 // otherwise VM startup (esp. restore) may fail without good reason
3157 if (!fFailIfInaccessible)
3158 return S_OK;
3159
3160 // otherwise report an error
3161 Bstr error;
3162 rc = pMedium->COMGETTER(LastAccessError)(error.asOutParam());
3163 if (FAILED(rc)) return rc;
3164
3165 /* collect multiple errors */
3166 eik.restore();
3167 Assert(!error.isEmpty());
3168 mrc = setError(E_FAIL,
3169 "%ls",
3170 error.raw());
3171 // error message will be something like
3172 // "Could not open the medium ... VD: error VERR_FILE_NOT_FOUND opening image file ... (VERR_FILE_NOT_FOUND).
3173 eik.fetch();
3174 }
3175 }
3176
3177 if (pMedium == this)
3178 mediumLockList.Prepend(pMedium, fMediumWritable);
3179 else
3180 mediumLockList.Prepend(pMedium, false);
3181
3182 pMedium = pMedium->getParent();
3183 if (pMedium.isNull() && pToBeParent)
3184 {
3185 pMedium = pToBeParent;
3186 pToBeParent = NULL;
3187 }
3188 }
3189
3190 return mrc;
3191}
3192
3193/**
3194 * Returns a preferred format for differencing hard disks.
3195 */
3196Bstr Medium::preferredDiffFormat()
3197{
3198 Utf8Str strFormat;
3199
3200 AutoCaller autoCaller(this);
3201 AssertComRCReturn(autoCaller.rc(), strFormat);
3202
3203 /* m->strFormat is const, no need to lock */
3204 strFormat = m->strFormat;
3205
3206 /* check that our own format supports diffs */
3207 if (!(m->formatObj->capabilities() & MediumFormatCapabilities_Differencing))
3208 {
3209 /* use the default format if not */
3210 AutoReadLock propsLock(m->pVirtualBox->systemProperties() COMMA_LOCKVAL_SRC_POS);
3211 strFormat = m->pVirtualBox->getDefaultHardDiskFormat();
3212 }
3213
3214 return strFormat;
3215}
3216
3217/**
3218 * Returns the medium type. Must have caller + locking!
3219 * @return
3220 */
3221MediumType_T Medium::getType() const
3222{
3223 return m->type;
3224}
3225
3226// private methods
3227////////////////////////////////////////////////////////////////////////////////
3228
3229/**
3230 * Returns a short version of the location attribute.
3231 *
3232 * @note Must be called from under this object's read or write lock.
3233 */
3234Utf8Str Medium::getName()
3235{
3236 Utf8Str name = RTPathFilename(m->strLocationFull.c_str());
3237 return name;
3238}
3239
3240/**
3241 * Sets the value of m->strLocation and calculates the value of m->strLocationFull.
3242 *
3243 * Treats non-FS-path locations specially, and prepends the default hard disk
3244 * folder if the given location string does not contain any path information
3245 * at all.
3246 *
3247 * Also, if the specified location is a file path that ends with '/' then the
3248 * file name part will be generated by this method automatically in the format
3249 * '{<uuid>}.<ext>' where <uuid> is a fresh UUID that this method will generate
3250 * and assign to this medium, and <ext> is the default extension for this
3251 * medium's storage format. Note that this procedure requires the media state to
3252 * be NotCreated and will return a failure otherwise.
3253 *
3254 * @param aLocation Location of the storage unit. If the location is a FS-path,
3255 * then it can be relative to the VirtualBox home directory.
3256 * @param aFormat Optional fallback format if it is an import and the format
3257 * cannot be determined.
3258 *
3259 * @note Must be called from under this object's write lock.
3260 */
3261HRESULT Medium::setLocation(const Utf8Str &aLocation, const Utf8Str &aFormat)
3262{
3263 AssertReturn(!aLocation.isEmpty(), E_FAIL);
3264
3265 AutoCaller autoCaller(this);
3266 AssertComRCReturnRC(autoCaller.rc());
3267
3268 /* formatObj may be null only when initializing from an existing path and
3269 * no format is known yet */
3270 AssertReturn( (!m->strFormat.isEmpty() && !m->formatObj.isNull())
3271 || ( autoCaller.state() == InInit
3272 && m->state != MediumState_NotCreated
3273 && m->id.isEmpty()
3274 && m->strFormat.isEmpty()
3275 && m->formatObj.isNull()),
3276 E_FAIL);
3277
3278 /* are we dealing with a new medium constructed using the existing
3279 * location? */
3280 bool isImport = m->strFormat.isEmpty();
3281
3282 if ( isImport
3283 || ( (m->formatObj->capabilities() & MediumFormatCapabilities_File)
3284 && !m->hostDrive))
3285 {
3286 Guid id;
3287
3288 Utf8Str location(aLocation);
3289
3290 if (m->state == MediumState_NotCreated)
3291 {
3292 /* must be a file (formatObj must be already known) */
3293 Assert(m->formatObj->capabilities() & MediumFormatCapabilities_File);
3294
3295 if (RTPathFilename(location.c_str()) == NULL)
3296 {
3297 /* no file name is given (either an empty string or ends with a
3298 * slash), generate a new UUID + file name if the state allows
3299 * this */
3300
3301 ComAssertMsgRet(!m->formatObj->fileExtensions().empty(),
3302 ("Must be at least one extension if it is MediumFormatCapabilities_File\n"),
3303 E_FAIL);
3304
3305 Bstr ext = m->formatObj->fileExtensions().front();
3306 ComAssertMsgRet(!ext.isEmpty(),
3307 ("Default extension must not be empty\n"),
3308 E_FAIL);
3309
3310 id.create();
3311
3312 location = Utf8StrFmt("%s{%RTuuid}.%ls",
3313 location.raw(), id.raw(), ext.raw());
3314 }
3315 }
3316
3317 /* append the default folder if no path is given */
3318 if (!RTPathHavePath(location.c_str()))
3319 location = Utf8StrFmt("%s%c%s",
3320 m->pVirtualBox->getDefaultHardDiskFolder().raw(),
3321 RTPATH_DELIMITER,
3322 location.raw());
3323
3324 /* get the full file name */
3325 Utf8Str locationFull;
3326 int vrc = m->pVirtualBox->calculateFullPath(location, locationFull);
3327 if (RT_FAILURE(vrc))
3328 return setError(VBOX_E_FILE_ERROR,
3329 tr("Invalid medium storage file location '%s' (%Rrc)"),
3330 location.raw(), vrc);
3331
3332 /* detect the backend from the storage unit if importing */
3333 if (isImport)
3334 {
3335 char *backendName = NULL;
3336
3337 /* is it a file? */
3338 {
3339 RTFILE file;
3340 vrc = RTFileOpen(&file, locationFull.c_str(), RTFILE_O_READ | RTFILE_O_OPEN | RTFILE_O_DENY_NONE);
3341 if (RT_SUCCESS(vrc))
3342 RTFileClose(file);
3343 }
3344 if (RT_SUCCESS(vrc))
3345 {
3346 vrc = VDGetFormat(NULL, locationFull.c_str(), &backendName);
3347 }
3348 else if (vrc != VERR_FILE_NOT_FOUND && vrc != VERR_PATH_NOT_FOUND)
3349 {
3350 /* assume it's not a file, restore the original location */
3351 location = locationFull = aLocation;
3352 vrc = VDGetFormat(NULL, locationFull.c_str(), &backendName);
3353 }
3354
3355 if (RT_FAILURE(vrc))
3356 {
3357 if (vrc == VERR_FILE_NOT_FOUND || vrc == VERR_PATH_NOT_FOUND)
3358 return setError(VBOX_E_FILE_ERROR,
3359 tr("Could not find file for the medium '%s' (%Rrc)"),
3360 locationFull.raw(), vrc);
3361 else if (aFormat.isEmpty())
3362 return setError(VBOX_E_IPRT_ERROR,
3363 tr("Could not get the storage format of the medium '%s' (%Rrc)"),
3364 locationFull.raw(), vrc);
3365 else
3366 {
3367 HRESULT rc = setFormat(Bstr(aFormat));
3368 /* setFormat() must not fail since we've just used the backend so
3369 * the format object must be there */
3370 AssertComRCReturnRC(rc);
3371 }
3372 }
3373 else
3374 {
3375 ComAssertRet(backendName != NULL && *backendName != '\0', E_FAIL);
3376
3377 HRESULT rc = setFormat(Bstr(backendName));
3378 RTStrFree(backendName);
3379
3380 /* setFormat() must not fail since we've just used the backend so
3381 * the format object must be there */
3382 AssertComRCReturnRC(rc);
3383 }
3384 }
3385
3386 /* is it still a file? */
3387 if (m->formatObj->capabilities() & MediumFormatCapabilities_File)
3388 {
3389 m->strLocation = location;
3390 m->strLocationFull = locationFull;
3391
3392 if (m->state == MediumState_NotCreated)
3393 {
3394 /* assign a new UUID (this UUID will be used when calling
3395 * VDCreateBase/VDCreateDiff as a wanted UUID). Note that we
3396 * also do that if we didn't generate it to make sure it is
3397 * either generated by us or reset to null */
3398 unconst(m->id) = id;
3399 }
3400 }
3401 else
3402 {
3403 m->strLocation = locationFull;
3404 m->strLocationFull = locationFull;
3405 }
3406 }
3407 else
3408 {
3409 m->strLocation = aLocation;
3410 m->strLocationFull = aLocation;
3411 }
3412
3413 return S_OK;
3414}
3415
3416/**
3417 * Queries information from the image file.
3418 *
3419 * As a result of this call, the accessibility state and data members such as
3420 * size and description will be updated with the current information.
3421 *
3422 * @note This method may block during a system I/O call that checks storage
3423 * accessibility.
3424 *
3425 * @note Locks medium tree for reading and writing (for new diff media checked
3426 * for the first time). Locks mParent for reading. Locks this object for
3427 * writing.
3428 */
3429HRESULT Medium::queryInfo()
3430{
3431 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3432
3433 if ( m->state != MediumState_Created
3434 && m->state != MediumState_Inaccessible
3435 && m->state != MediumState_LockedRead)
3436 return E_FAIL;
3437
3438 HRESULT rc = S_OK;
3439
3440 int vrc = VINF_SUCCESS;
3441
3442 /* check if a blocking queryInfo() call is in progress on some other thread,
3443 * and wait for it to finish if so instead of querying data ourselves */
3444 if (m->queryInfoRunning)
3445 {
3446 Assert( m->state == MediumState_LockedRead
3447 || m->state == MediumState_LockedWrite);
3448
3449 alock.leave();
3450
3451 vrc = RTSemEventMultiWait(m->queryInfoSem, RT_INDEFINITE_WAIT);
3452
3453 alock.enter();
3454
3455 AssertRC(vrc);
3456
3457 return S_OK;
3458 }
3459
3460 bool success = false;
3461 Utf8Str lastAccessError;
3462
3463 /* are we dealing with a new medium constructed using the existing
3464 * location? */
3465 bool isImport = m->id.isEmpty();
3466 unsigned flags = VD_OPEN_FLAGS_INFO;
3467
3468 /* Note that we don't use VD_OPEN_FLAGS_READONLY when opening new
3469 * media because that would prevent necessary modifications
3470 * when opening media of some third-party formats for the first
3471 * time in VirtualBox (such as VMDK for which VDOpen() needs to
3472 * generate an UUID if it is missing) */
3473 if ( (m->hddOpenMode == OpenReadOnly)
3474 || !isImport
3475 )
3476 flags |= VD_OPEN_FLAGS_READONLY;
3477
3478 /* Lock the medium, which makes the behavior much more consistent */
3479 if (flags & VD_OPEN_FLAGS_READONLY)
3480 rc = LockRead(NULL);
3481 else
3482 rc = LockWrite(NULL);
3483 if (FAILED(rc)) return rc;
3484
3485 /* Copies of the input state fields which are not read-only,
3486 * as we're dropping the lock. CAUTION: be extremely careful what
3487 * you do with the contents of this medium object, as you will
3488 * create races if there are concurrent changes. */
3489 Utf8Str format(m->strFormat);
3490 Utf8Str location(m->strLocationFull);
3491 ComObjPtr<MediumFormat> formatObj = m->formatObj;
3492
3493 /* "Output" values which can't be set because the lock isn't held
3494 * at the time the values are determined. */
3495 Guid mediumId = m->id;
3496 uint64_t mediumSize = 0;
3497 uint64_t mediumLogicalSize = 0;
3498
3499 /* leave the lock before a lengthy operation */
3500 vrc = RTSemEventMultiReset(m->queryInfoSem);
3501 AssertRCReturn(vrc, E_FAIL);
3502 m->queryInfoRunning = true;
3503 alock.leave();
3504
3505 try
3506 {
3507 /* skip accessibility checks for host drives */
3508 if (m->hostDrive)
3509 {
3510 success = true;
3511 throw S_OK;
3512 }
3513
3514 PVBOXHDD hdd;
3515 vrc = VDCreate(m->vdDiskIfaces, &hdd);
3516 ComAssertRCThrow(vrc, E_FAIL);
3517
3518 try
3519 {
3520 /** @todo This kind of opening of images is assuming that diff
3521 * images can be opened as base images. Should be documented if
3522 * it must work for all medium format backends. */
3523 vrc = VDOpen(hdd,
3524 format.c_str(),
3525 location.c_str(),
3526 flags,
3527 m->vdDiskIfaces);
3528 if (RT_FAILURE(vrc))
3529 {
3530 lastAccessError = Utf8StrFmt(tr("Could not open the medium '%s'%s"),
3531 location.c_str(), vdError(vrc).c_str());
3532 throw S_OK;
3533 }
3534
3535 if (formatObj->capabilities() & MediumFormatCapabilities_Uuid)
3536 {
3537 /* Modify the UUIDs if necessary. The associated fields are
3538 * not modified by other code, so no need to copy. */
3539 if (m->setImageId)
3540 {
3541 vrc = VDSetUuid(hdd, 0, m->imageId);
3542 ComAssertRCThrow(vrc, E_FAIL);
3543 }
3544 if (m->setParentId)
3545 {
3546 vrc = VDSetParentUuid(hdd, 0, m->parentId);
3547 ComAssertRCThrow(vrc, E_FAIL);
3548 }
3549 /* zap the information, these are no long-term members */
3550 m->setImageId = false;
3551 unconst(m->imageId).clear();
3552 m->setParentId = false;
3553 unconst(m->parentId).clear();
3554
3555 /* check the UUID */
3556 RTUUID uuid;
3557 vrc = VDGetUuid(hdd, 0, &uuid);
3558 ComAssertRCThrow(vrc, E_FAIL);
3559
3560 if (isImport)
3561 {
3562 mediumId = uuid;
3563
3564 if (mediumId.isEmpty() && (m->hddOpenMode == OpenReadOnly))
3565 // only when importing a VDMK that has no UUID, create one in memory
3566 mediumId.create();
3567 }
3568 else
3569 {
3570 Assert(!mediumId.isEmpty());
3571
3572 if (mediumId != uuid)
3573 {
3574 lastAccessError = Utf8StrFmt(
3575 tr("UUID {%RTuuid} of the medium '%s' does not match the value {%RTuuid} stored in the media registry ('%s')"),
3576 &uuid,
3577 location.c_str(),
3578 mediumId.raw(),
3579 m->pVirtualBox->settingsFilePath().c_str());
3580 throw S_OK;
3581 }
3582 }
3583 }
3584 else
3585 {
3586 /* the backend does not support storing UUIDs within the
3587 * underlying storage so use what we store in XML */
3588
3589 /* generate an UUID for an imported UUID-less medium */
3590 if (isImport)
3591 {
3592 if (m->setImageId)
3593 mediumId = m->imageId;
3594 else
3595 mediumId.create();
3596 }
3597 }
3598
3599 /* check the type */
3600 unsigned uImageFlags;
3601 vrc = VDGetImageFlags(hdd, 0, &uImageFlags);
3602 ComAssertRCThrow(vrc, E_FAIL);
3603
3604 if (uImageFlags & VD_IMAGE_FLAGS_DIFF)
3605 {
3606 RTUUID parentId;
3607 vrc = VDGetParentUuid(hdd, 0, &parentId);
3608 ComAssertRCThrow(vrc, E_FAIL);
3609
3610 if (isImport)
3611 {
3612 /* the parent must be known to us. Note that we freely
3613 * call locking methods of mVirtualBox and parent from the
3614 * write lock (breaking the {parent,child} lock order)
3615 * because there may be no concurrent access to the just
3616 * opened hard disk on ther threads yet (and init() will
3617 * fail if this method reporst MediumState_Inaccessible) */
3618
3619 Guid id = parentId;
3620 ComObjPtr<Medium> pParent;
3621 rc = m->pVirtualBox->findHardDisk(&id, NULL,
3622 false /* aSetError */,
3623 &pParent);
3624 if (FAILED(rc))
3625 {
3626 lastAccessError = Utf8StrFmt(
3627 tr("Parent hard disk with UUID {%RTuuid} of the hard disk '%s' is not found in the media registry ('%s')"),
3628 &parentId, location.c_str(),
3629 m->pVirtualBox->settingsFilePath().c_str());
3630 throw S_OK;
3631 }
3632
3633 /* we set mParent & children() */
3634 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3635
3636 Assert(m->pParent.isNull());
3637 m->pParent = pParent;
3638 m->pParent->m->llChildren.push_back(this);
3639 }
3640 else
3641 {
3642 /* we access mParent */
3643 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3644
3645 /* check that parent UUIDs match. Note that there's no need
3646 * for the parent's AutoCaller (our lifetime is bound to
3647 * it) */
3648
3649 if (m->pParent.isNull())
3650 {
3651 lastAccessError = Utf8StrFmt(
3652 tr("Hard disk '%s' is differencing but it is not associated with any parent hard disk in the media registry ('%s')"),
3653 location.c_str(),
3654 m->pVirtualBox->settingsFilePath().c_str());
3655 throw S_OK;
3656 }
3657
3658 AutoReadLock parentLock(m->pParent COMMA_LOCKVAL_SRC_POS);
3659 if ( m->pParent->getState() != MediumState_Inaccessible
3660 && m->pParent->getId() != parentId)
3661 {
3662 lastAccessError = Utf8StrFmt(
3663 tr("Parent UUID {%RTuuid} of the hard disk '%s' does not match UUID {%RTuuid} of its parent hard disk stored in the media registry ('%s')"),
3664 &parentId, location.c_str(),
3665 m->pParent->getId().raw(),
3666 m->pVirtualBox->settingsFilePath().c_str());
3667 throw S_OK;
3668 }
3669
3670 /// @todo NEWMEDIA what to do if the parent is not
3671 /// accessible while the diff is? Probably nothing. The
3672 /// real code will detect the mismatch anyway.
3673 }
3674 }
3675
3676 mediumSize = VDGetFileSize(hdd, 0);
3677 mediumLogicalSize = VDGetSize(hdd, 0) / _1M;
3678
3679 success = true;
3680 }
3681 catch (HRESULT aRC)
3682 {
3683 rc = aRC;
3684 }
3685
3686 VDDestroy(hdd);
3687
3688 }
3689 catch (HRESULT aRC)
3690 {
3691 rc = aRC;
3692 }
3693
3694 alock.enter();
3695
3696 if (isImport)
3697 unconst(m->id) = mediumId;
3698
3699 if (success)
3700 {
3701 m->size = mediumSize;
3702 m->logicalSize = mediumLogicalSize;
3703 m->strLastAccessError.setNull();
3704 }
3705 else
3706 {
3707 m->strLastAccessError = lastAccessError;
3708 LogWarningFunc(("'%s' is not accessible (error='%s', rc=%Rhrc, vrc=%Rrc)\n",
3709 location.c_str(), m->strLastAccessError.c_str(),
3710 rc, vrc));
3711 }
3712
3713 /* inform other callers if there are any */
3714 RTSemEventMultiSignal(m->queryInfoSem);
3715 m->queryInfoRunning = false;
3716
3717 /* Set the proper state according to the result of the check */
3718 if (success)
3719 m->preLockState = MediumState_Created;
3720 else
3721 m->preLockState = MediumState_Inaccessible;
3722
3723 if (flags & VD_OPEN_FLAGS_READONLY)
3724 rc = UnlockRead(NULL);
3725 else
3726 rc = UnlockWrite(NULL);
3727 if (FAILED(rc)) return rc;
3728
3729 return rc;
3730}
3731
3732/**
3733 * Sets the extended error info according to the current media state.
3734 *
3735 * @note Must be called from under this object's write or read lock.
3736 */
3737HRESULT Medium::setStateError()
3738{
3739 HRESULT rc = E_FAIL;
3740
3741 switch (m->state)
3742 {
3743 case MediumState_NotCreated:
3744 {
3745 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3746 tr("Storage for the medium '%s' is not created"),
3747 m->strLocationFull.raw());
3748 break;
3749 }
3750 case MediumState_Created:
3751 {
3752 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3753 tr("Storage for the medium '%s' is already created"),
3754 m->strLocationFull.raw());
3755 break;
3756 }
3757 case MediumState_LockedRead:
3758 {
3759 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3760 tr("Medium '%s' is locked for reading by another task"),
3761 m->strLocationFull.raw());
3762 break;
3763 }
3764 case MediumState_LockedWrite:
3765 {
3766 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3767 tr("Medium '%s' is locked for writing by another task"),
3768 m->strLocationFull.raw());
3769 break;
3770 }
3771 case MediumState_Inaccessible:
3772 {
3773 /* be in sync with Console::powerUpThread() */
3774 if (!m->strLastAccessError.isEmpty())
3775 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3776 tr("Medium '%s' is not accessible. %s"),
3777 m->strLocationFull.raw(), m->strLastAccessError.c_str());
3778 else
3779 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3780 tr("Medium '%s' is not accessible"),
3781 m->strLocationFull.raw());
3782 break;
3783 }
3784 case MediumState_Creating:
3785 {
3786 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3787 tr("Storage for the medium '%s' is being created"),
3788 m->strLocationFull.raw());
3789 break;
3790 }
3791 case MediumState_Deleting:
3792 {
3793 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
3794 tr("Storage for the medium '%s' is being deleted"),
3795 m->strLocationFull.raw());
3796 break;
3797 }
3798 default:
3799 {
3800 AssertFailed();
3801 break;
3802 }
3803 }
3804
3805 return rc;
3806}
3807
3808/**
3809 * Deletes the hard disk storage unit.
3810 *
3811 * If @a aProgress is not NULL but the object it points to is @c null then a new
3812 * progress object will be created and assigned to @a *aProgress on success,
3813 * otherwise the existing progress object is used. If Progress is NULL, then no
3814 * progress object is created/used at all.
3815 *
3816 * When @a aWait is @c false, this method will create a thread to perform the
3817 * delete operation asynchronously and will return immediately. Otherwise, it
3818 * will perform the operation on the calling thread and will not return to the
3819 * caller until the operation is completed. Note that @a aProgress cannot be
3820 * NULL when @a aWait is @c false (this method will assert in this case).
3821 *
3822 * @param aProgress Where to find/store a Progress object to track operation
3823 * completion.
3824 * @param aWait @c true if this method should block instead of creating
3825 * an asynchronous thread.
3826 * @param pfNeedsSaveSettings Optional pointer to a bool that must have been initialized to false and that will be set to true
3827 * by this function if the caller should invoke VirtualBox::saveSettings() because the global settings have changed.
3828 * This only works in "wait" mode; otherwise saveSettings gets called automatically by the thread that was created,
3829 * and this parameter is ignored.
3830 *
3831 * @note Locks mVirtualBox and this object for writing. Locks medium tree for
3832 * writing.
3833 */
3834HRESULT Medium::deleteStorage(ComObjPtr<Progress> *aProgress,
3835 bool aWait,
3836 bool *pfNeedsSaveSettings)
3837{
3838 AssertReturn(aProgress != NULL || aWait == true, E_FAIL);
3839
3840 HRESULT rc = S_OK;
3841 ComObjPtr<Progress> pProgress;
3842 Medium::Task *pTask = NULL;
3843
3844 try
3845 {
3846 /* we're accessing the media tree, and canClose() needs it too */
3847 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
3848 this->lockHandle()
3849 COMMA_LOCKVAL_SRC_POS);
3850 LogFlowThisFunc(("aWait=%RTbool locationFull=%s\n", aWait, getLocationFull().c_str() ));
3851
3852 if ( !(m->formatObj->capabilities() & ( MediumFormatCapabilities_CreateDynamic
3853 | MediumFormatCapabilities_CreateFixed)))
3854 throw setError(VBOX_E_NOT_SUPPORTED,
3855 tr("Hard disk format '%s' does not support storage deletion"),
3856 m->strFormat.raw());
3857
3858 /* Note that we are fine with Inaccessible state too: a) for symmetry
3859 * with create calls and b) because it doesn't really harm to try, if
3860 * it is really inaccessible, the delete operation will fail anyway.
3861 * Accepting Inaccessible state is especially important because all
3862 * registered hard disks are initially Inaccessible upon VBoxSVC
3863 * startup until COMGETTER(RefreshState) is called. Accept Deleting
3864 * state because some callers need to put the image in this state early
3865 * to prevent races. */
3866 switch (m->state)
3867 {
3868 case MediumState_Created:
3869 case MediumState_Deleting:
3870 case MediumState_Inaccessible:
3871 break;
3872 default:
3873 throw setStateError();
3874 }
3875
3876 if (m->backRefs.size() != 0)
3877 {
3878 Utf8Str strMachines;
3879 for (BackRefList::const_iterator it = m->backRefs.begin();
3880 it != m->backRefs.end();
3881 ++it)
3882 {
3883 const BackRef &b = *it;
3884 if (strMachines.length())
3885 strMachines.append(", ");
3886 strMachines.append(b.machineId.toString().c_str());
3887 }
3888#ifdef DEBUG
3889 dumpBackRefs();
3890#endif
3891 throw setError(VBOX_E_OBJECT_IN_USE,
3892 tr("Cannot delete storage: hard disk '%s' is still attached to the following %d virtual machine(s): %s"),
3893 m->strLocationFull.c_str(),
3894 m->backRefs.size(),
3895 strMachines.c_str());
3896 }
3897
3898 rc = canClose();
3899 if (FAILED(rc))
3900 throw rc;
3901
3902 /* go to Deleting state, so that the medium is not actually locked */
3903 if (m->state != MediumState_Deleting)
3904 {
3905 rc = markForDeletion();
3906 if (FAILED(rc))
3907 throw rc;
3908 }
3909
3910 /* Build the medium lock list. */
3911 MediumLockList *pMediumLockList(new MediumLockList());
3912 rc = createMediumLockList(true /* fFailIfInaccessible */,
3913 true /* fMediumWritable */,
3914 NULL,
3915 *pMediumLockList);
3916 if (FAILED(rc))
3917 {
3918 delete pMediumLockList;
3919 throw rc;
3920 }
3921
3922 rc = pMediumLockList->Lock();
3923 if (FAILED(rc))
3924 {
3925 delete pMediumLockList;
3926 throw setError(rc,
3927 tr("Failed to lock media when deleting '%ls'"),
3928 getLocationFull().raw());
3929 }
3930
3931 /* try to remove from the list of known hard disks before performing
3932 * actual deletion (we favor the consistency of the media registry
3933 * which would have been broken if unregisterWithVirtualBox() failed
3934 * after we successfully deleted the storage) */
3935 rc = unregisterWithVirtualBox(pfNeedsSaveSettings);
3936 if (FAILED(rc))
3937 throw rc;
3938 // no longer need lock
3939 multilock.release();
3940
3941 if (aProgress != NULL)
3942 {
3943 /* use the existing progress object... */
3944 pProgress = *aProgress;
3945
3946 /* ...but create a new one if it is null */
3947 if (pProgress.isNull())
3948 {
3949 pProgress.createObject();
3950 rc = pProgress->init(m->pVirtualBox,
3951 static_cast<IMedium*>(this),
3952 BstrFmt(tr("Deleting hard disk storage unit '%s'"), m->strLocationFull.raw()),
3953 FALSE /* aCancelable */);
3954 if (FAILED(rc))
3955 throw rc;
3956 }
3957 }
3958
3959 /* setup task object to carry out the operation sync/async */
3960 pTask = new Medium::DeleteTask(this, pProgress, pMediumLockList);
3961 rc = pTask->rc();
3962 AssertComRC(rc);
3963 if (FAILED(rc))
3964 throw rc;
3965 }
3966 catch (HRESULT aRC) { rc = aRC; }
3967
3968 if (SUCCEEDED(rc))
3969 {
3970 if (aWait)
3971 rc = runNow(pTask, NULL /* pfNeedsSaveSettings*/);
3972 else
3973 rc = startThread(pTask);
3974
3975 if (SUCCEEDED(rc) && aProgress != NULL)
3976 *aProgress = pProgress;
3977
3978 }
3979 else
3980 {
3981 if (pTask)
3982 delete pTask;
3983
3984 /* Undo deleting state if necessary. */
3985 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3986 unmarkForDeletion();
3987 }
3988
3989 return rc;
3990}
3991
3992/**
3993 * Mark a medium for deletion.
3994 *
3995 * @note Caller must hold the write lock on this medium!
3996 */
3997HRESULT Medium::markForDeletion()
3998{
3999 ComAssertRet(this->lockHandle()->isWriteLockOnCurrentThread(), E_FAIL);
4000 switch (m->state)
4001 {
4002 case MediumState_Created:
4003 case MediumState_Inaccessible:
4004 m->preLockState = m->state;
4005 m->state = MediumState_Deleting;
4006 return S_OK;
4007 default:
4008 return setStateError();
4009 }
4010}
4011
4012/**
4013 * Removes the "mark for deletion".
4014 *
4015 * @note Caller must hold the write lock on this medium!
4016 */
4017HRESULT Medium::unmarkForDeletion()
4018{
4019 ComAssertRet(this->lockHandle()->isWriteLockOnCurrentThread(), E_FAIL);
4020 switch (m->state)
4021 {
4022 case MediumState_Deleting:
4023 m->state = m->preLockState;
4024 return S_OK;
4025 default:
4026 return setStateError();
4027 }
4028}
4029
4030/**
4031 * Mark a medium for deletion which is in locked state.
4032 *
4033 * @note Caller must hold the write lock on this medium!
4034 */
4035HRESULT Medium::markLockedForDeletion()
4036{
4037 ComAssertRet(this->lockHandle()->isWriteLockOnCurrentThread(), E_FAIL);
4038 if ( ( m->state == MediumState_LockedRead
4039 || m->state == MediumState_LockedWrite)
4040 && m->preLockState == MediumState_Created)
4041 {
4042 m->preLockState = MediumState_Deleting;
4043 return S_OK;
4044 }
4045 else
4046 return setStateError();
4047}
4048
4049/**
4050 * Removes the "mark for deletion" for a medium in locked state.
4051 *
4052 * @note Caller must hold the write lock on this medium!
4053 */
4054HRESULT Medium::unmarkLockedForDeletion()
4055{
4056 ComAssertRet(this->lockHandle()->isWriteLockOnCurrentThread(), E_FAIL);
4057 if ( ( m->state == MediumState_LockedRead
4058 || m->state == MediumState_LockedWrite)
4059 && m->preLockState == MediumState_Deleting)
4060 {
4061 m->preLockState = MediumState_Created;
4062 return S_OK;
4063 }
4064 else
4065 return setStateError();
4066}
4067
4068/**
4069 * Creates a new differencing storage unit using the given target hard disk's
4070 * format and the location. Note that @c aTarget must be NotCreated.
4071 *
4072 * The @a aMediumLockList parameter contains the associated medium lock list,
4073 * which must be in locked state. If @a aWait is @c true then the caller is
4074 * responsible for unlocking.
4075 *
4076 * If @a aProgress is not NULL but the object it points to is @c null then a
4077 * new progress object will be created and assigned to @a *aProgress on
4078 * success, otherwise the existing progress object is used. If @a aProgress is
4079 * NULL, then no progress object is created/used at all.
4080 *
4081 * When @a aWait is @c false, this method will create a thread to perform the
4082 * create operation asynchronously and will return immediately. Otherwise, it
4083 * will perform the operation on the calling thread and will not return to the
4084 * caller until the operation is completed. Note that @a aProgress cannot be
4085 * NULL when @a aWait is @c false (this method will assert in this case).
4086 *
4087 * @param aTarget Target hard disk.
4088 * @param aVariant Precise image variant to create.
4089 * @param aMediumLockList List of media which should be locked.
4090 * @param aProgress Where to find/store a Progress object to track
4091 * operation completion.
4092 * @param aWait @c true if this method should block instead of
4093 * creating an asynchronous thread.
4094 * @param pfNeedsSaveSettings Optional pointer to a bool that must have been
4095 * initialized to false and that will be set to true
4096 * by this function if the caller should invoke
4097 * VirtualBox::saveSettings() because the global
4098 * settings have changed. This only works in "wait"
4099 * mode; otherwise saveSettings is called
4100 * automatically by the thread that was created,
4101 * and this parameter is ignored.
4102 *
4103 * @note Locks this object and @a aTarget for writing.
4104 */
4105HRESULT Medium::createDiffStorage(ComObjPtr<Medium> &aTarget,
4106 MediumVariant_T aVariant,
4107 MediumLockList *aMediumLockList,
4108 ComObjPtr<Progress> *aProgress,
4109 bool aWait,
4110 bool *pfNeedsSaveSettings)
4111{
4112 AssertReturn(!aTarget.isNull(), E_FAIL);
4113 AssertReturn(aMediumLockList, E_FAIL);
4114 AssertReturn(aProgress != NULL || aWait == true, E_FAIL);
4115
4116 AutoCaller autoCaller(this);
4117 if (FAILED(autoCaller.rc())) return autoCaller.rc();
4118
4119 AutoCaller targetCaller(aTarget);
4120 if (FAILED(targetCaller.rc())) return targetCaller.rc();
4121
4122 HRESULT rc = S_OK;
4123 ComObjPtr<Progress> pProgress;
4124 Medium::Task *pTask = NULL;
4125
4126 try
4127 {
4128 AutoMultiWriteLock2 alock(this, aTarget COMMA_LOCKVAL_SRC_POS);
4129
4130 ComAssertThrow(m->type != MediumType_Writethrough, E_FAIL);
4131 ComAssertThrow(m->state == MediumState_LockedRead, E_FAIL);
4132
4133 if (aTarget->m->state != MediumState_NotCreated)
4134 throw aTarget->setStateError();
4135
4136 /* Check that the hard disk is not attached to the current state of
4137 * any VM referring to it. */
4138 for (BackRefList::const_iterator it = m->backRefs.begin();
4139 it != m->backRefs.end();
4140 ++it)
4141 {
4142 if (it->fInCurState)
4143 {
4144 /* Note: when a VM snapshot is being taken, all normal hard
4145 * disks attached to the VM in the current state will be, as an
4146 * exception, also associated with the snapshot which is about
4147 * to create (see SnapshotMachine::init()) before deassociating
4148 * them from the current state (which takes place only on
4149 * success in Machine::fixupHardDisks()), so that the size of
4150 * snapshotIds will be 1 in this case. The extra condition is
4151 * used to filter out this legal situation. */
4152 if (it->llSnapshotIds.size() == 0)
4153 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4154 tr("Hard disk '%s' is attached to a virtual machine with UUID {%RTuuid}. No differencing hard disks based on it may be created until it is detached"),
4155 m->strLocationFull.raw(), it->machineId.raw());
4156
4157 Assert(it->llSnapshotIds.size() == 1);
4158 }
4159 }
4160
4161 if (aProgress != NULL)
4162 {
4163 /* use the existing progress object... */
4164 pProgress = *aProgress;
4165
4166 /* ...but create a new one if it is null */
4167 if (pProgress.isNull())
4168 {
4169 pProgress.createObject();
4170 rc = pProgress->init(m->pVirtualBox,
4171 static_cast<IMedium*>(this),
4172 BstrFmt(tr("Creating differencing hard disk storage unit '%s'"), aTarget->m->strLocationFull.raw()),
4173 TRUE /* aCancelable */);
4174 if (FAILED(rc))
4175 throw rc;
4176 }
4177 }
4178
4179 /* setup task object to carry out the operation sync/async */
4180 pTask = new Medium::CreateDiffTask(this, pProgress, aTarget, aVariant,
4181 aMediumLockList,
4182 aWait /* fKeepMediumLockList */);
4183 rc = pTask->rc();
4184 AssertComRC(rc);
4185 if (FAILED(rc))
4186 throw rc;
4187
4188 /* register a task (it will deregister itself when done) */
4189 ++m->numCreateDiffTasks;
4190 Assert(m->numCreateDiffTasks != 0); /* overflow? */
4191
4192 aTarget->m->state = MediumState_Creating;
4193 }
4194 catch (HRESULT aRC) { rc = aRC; }
4195
4196 if (SUCCEEDED(rc))
4197 {
4198 if (aWait)
4199 rc = runNow(pTask, pfNeedsSaveSettings);
4200 else
4201 rc = startThread(pTask);
4202
4203 if (SUCCEEDED(rc) && aProgress != NULL)
4204 *aProgress = pProgress;
4205 }
4206 else if (pTask != NULL)
4207 delete pTask;
4208
4209 return rc;
4210}
4211
4212/**
4213 * Prepares this (source) hard disk, target hard disk and all intermediate hard
4214 * disks for the merge operation.
4215 *
4216 * This method is to be called prior to calling the #mergeTo() to perform
4217 * necessary consistency checks and place involved hard disks to appropriate
4218 * states. If #mergeTo() is not called or fails, the state modifications
4219 * performed by this method must be undone by #cancelMergeTo().
4220 *
4221 * See #mergeTo() for more information about merging.
4222 *
4223 * @param pTarget Target hard disk.
4224 * @param aMachineId Allowed machine attachment. NULL means do not check.
4225 * @param aSnapshotId Allowed snapshot attachment. NULL or empty UUID means
4226 * do not check.
4227 * @param fLockMedia Flag whether to lock the medium lock list or not.
4228 * If set to false and the medium lock list locking fails
4229 * later you must call #cancelMergeTo().
4230 * @param fMergeForward Resulting merge direction (out).
4231 * @param pParentForTarget New parent for target medium after merge (out).
4232 * @param aChildrenToReparent List of children of the source which will have
4233 * to be reparented to the target after merge (out).
4234 * @param aMediumLockList Medium locking information (out).
4235 *
4236 * @note Locks medium tree for reading. Locks this object, aTarget and all
4237 * intermediate hard disks for writing.
4238 */
4239HRESULT Medium::prepareMergeTo(const ComObjPtr<Medium> &pTarget,
4240 const Guid *aMachineId,
4241 const Guid *aSnapshotId,
4242 bool fLockMedia,
4243 bool &fMergeForward,
4244 ComObjPtr<Medium> &pParentForTarget,
4245 MediaList &aChildrenToReparent,
4246 MediumLockList * &aMediumLockList)
4247{
4248 AssertReturn(pTarget != NULL, E_FAIL);
4249 AssertReturn(pTarget != this, E_FAIL);
4250
4251 AutoCaller autoCaller(this);
4252 AssertComRCReturnRC(autoCaller.rc());
4253
4254 AutoCaller targetCaller(pTarget);
4255 AssertComRCReturnRC(targetCaller.rc());
4256
4257 HRESULT rc = S_OK;
4258 fMergeForward = false;
4259 pParentForTarget.setNull();
4260 aChildrenToReparent.clear();
4261 Assert(aMediumLockList == NULL);
4262 aMediumLockList = NULL;
4263
4264 try
4265 {
4266 // locking: we need the tree lock first because we access parent pointers
4267 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
4268
4269 /* more sanity checking and figuring out the merge direction */
4270 ComObjPtr<Medium> pMedium = getParent();
4271 while (!pMedium.isNull() && pMedium != pTarget)
4272 pMedium = pMedium->getParent();
4273 if (pMedium == pTarget)
4274 fMergeForward = false;
4275 else
4276 {
4277 pMedium = pTarget->getParent();
4278 while (!pMedium.isNull() && pMedium != this)
4279 pMedium = pMedium->getParent();
4280 if (pMedium == this)
4281 fMergeForward = true;
4282 else
4283 {
4284 Utf8Str tgtLoc;
4285 {
4286 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4287 tgtLoc = pTarget->getLocationFull();
4288 }
4289
4290 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4291 throw setError(E_FAIL,
4292 tr("Hard disks '%s' and '%s' are unrelated"),
4293 m->strLocationFull.raw(), tgtLoc.raw());
4294 }
4295 }
4296
4297 /* Build the lock list. */
4298 aMediumLockList = new MediumLockList();
4299 if (fMergeForward)
4300 rc = pTarget->createMediumLockList(true /* fFailIfInaccessible */,
4301 true /* fMediumWritable */,
4302 NULL,
4303 *aMediumLockList);
4304 else
4305 rc = createMediumLockList(true /* fFailIfInaccessible */,
4306 false /* fMediumWritable */,
4307 NULL,
4308 *aMediumLockList);
4309 if (FAILED(rc))
4310 throw rc;
4311
4312 /* Sanity checking, must be after lock list creation as it depends on
4313 * valid medium states. The medium objects must be accessible. Only
4314 * do this if immediate locking is requested, otherwise it fails when
4315 * we construct a medium lock list for an already running VM. Snapshot
4316 * deletion uses this to simplify its life. */
4317 if (fLockMedia)
4318 {
4319 {
4320 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4321 if (m->state != MediumState_Created)
4322 throw setStateError();
4323 }
4324 {
4325 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4326 if (pTarget->m->state != MediumState_Created)
4327 throw pTarget->setStateError();
4328 }
4329 }
4330
4331 /* check medium attachment and other sanity conditions */
4332 if (fMergeForward)
4333 {
4334 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4335 if (getChildren().size() > 1)
4336 {
4337 throw setError(E_FAIL,
4338 tr("Medium '%s' involved in the merge operation has more than one child medium (%d)"),
4339 m->strLocationFull.raw(), getChildren().size());
4340 }
4341 /* One backreference is only allowed if the machine ID is not empty
4342 * and it matches the machine the image is attached to (including
4343 * the snapshot ID if not empty). */
4344 if ( m->backRefs.size() != 0
4345 && ( !aMachineId
4346 || m->backRefs.size() != 1
4347 || aMachineId->isEmpty()
4348 || *getFirstMachineBackrefId() != *aMachineId
4349 || ( (!aSnapshotId || !aSnapshotId->isEmpty())
4350 && *getFirstMachineBackrefSnapshotId() != *aSnapshotId)))
4351 throw setError(E_FAIL,
4352 tr("Medium '%s' is attached to %d virtual machines"),
4353 m->strLocationFull.raw(), m->backRefs.size());
4354 if (m->type == MediumType_Immutable)
4355 throw setError(E_FAIL,
4356 tr("Medium '%s' is immutable"),
4357 m->strLocationFull.raw());
4358 }
4359 else
4360 {
4361 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4362 if (pTarget->getChildren().size() > 1)
4363 {
4364 throw setError(E_FAIL,
4365 tr("Medium '%s' involved in the merge operation has more than one child medium (%d)"),
4366 pTarget->m->strLocationFull.raw(),
4367 pTarget->getChildren().size());
4368 }
4369 if (pTarget->m->type == MediumType_Immutable)
4370 throw setError(E_FAIL,
4371 tr("Medium '%s' is immutable"),
4372 pTarget->m->strLocationFull.raw());
4373 }
4374 ComObjPtr<Medium> pLast(fMergeForward ? (Medium *)pTarget : this);
4375 ComObjPtr<Medium> pLastIntermediate = pLast->getParent();
4376 for (pLast = pLastIntermediate;
4377 !pLast.isNull() && pLast != pTarget && pLast != this;
4378 pLast = pLast->getParent())
4379 {
4380 AutoReadLock alock(pLast COMMA_LOCKVAL_SRC_POS);
4381 if (pLast->getChildren().size() > 1)
4382 {
4383 throw setError(E_FAIL,
4384 tr("Medium '%s' involved in the merge operation has more than one child medium (%d)"),
4385 pLast->m->strLocationFull.raw(),
4386 pLast->getChildren().size());
4387 }
4388 if (pLast->m->backRefs.size() != 0)
4389 throw setError(E_FAIL,
4390 tr("Medium '%s' is attached to %d virtual machines"),
4391 pLast->m->strLocationFull.raw(),
4392 pLast->m->backRefs.size());
4393
4394 }
4395
4396 /* Update medium states appropriately */
4397 if (m->state == MediumState_Created)
4398 {
4399 rc = markForDeletion();
4400 if (FAILED(rc))
4401 throw rc;
4402 }
4403 else
4404 {
4405 if (fLockMedia)
4406 throw setStateError();
4407 else if ( m->state == MediumState_LockedWrite
4408 || m->state == MediumState_LockedRead)
4409 {
4410 /* Either mark it for deletiion in locked state or allow
4411 * others to have done so. */
4412 if (m->preLockState == MediumState_Created)
4413 markLockedForDeletion();
4414 else if (m->preLockState != MediumState_Deleting)
4415 throw setStateError();
4416 }
4417 else
4418 throw setStateError();
4419 }
4420
4421 if (fMergeForward)
4422 {
4423 /* we will need parent to reparent target */
4424 pParentForTarget = m->pParent;
4425 }
4426 else
4427 {
4428 /* we will need to reparent children of the source */
4429 for (MediaList::const_iterator it = getChildren().begin();
4430 it != getChildren().end();
4431 ++it)
4432 {
4433 pMedium = *it;
4434 if (fLockMedia)
4435 {
4436 rc = pMedium->LockWrite(NULL);
4437 if (FAILED(rc))
4438 throw rc;
4439 }
4440
4441 aChildrenToReparent.push_back(pMedium);
4442 }
4443 }
4444 for (pLast = pLastIntermediate;
4445 !pLast.isNull() && pLast != pTarget && pLast != this;
4446 pLast = pLast->getParent())
4447 {
4448 AutoWriteLock alock(pLast COMMA_LOCKVAL_SRC_POS);
4449 if (pLast->m->state == MediumState_Created)
4450 {
4451 rc = pLast->markForDeletion();
4452 if (FAILED(rc))
4453 throw rc;
4454 }
4455 else
4456 throw pLast->setStateError();
4457 }
4458
4459 /* Tweak the lock list in the backward merge case, as the target
4460 * isn't marked to be locked for writing yet. */
4461 if (!fMergeForward)
4462 {
4463 MediumLockList::Base::iterator lockListBegin =
4464 aMediumLockList->GetBegin();
4465 MediumLockList::Base::iterator lockListEnd =
4466 aMediumLockList->GetEnd();
4467 lockListEnd--;
4468 for (MediumLockList::Base::iterator it = lockListBegin;
4469 it != lockListEnd;
4470 ++it)
4471 {
4472 MediumLock &mediumLock = *it;
4473 if (mediumLock.GetMedium() == pTarget)
4474 {
4475 HRESULT rc2 = mediumLock.UpdateLock(true);
4476 AssertComRC(rc2);
4477 break;
4478 }
4479 }
4480 }
4481
4482 if (fLockMedia)
4483 {
4484 rc = aMediumLockList->Lock();
4485 if (FAILED(rc))
4486 {
4487 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4488 throw setError(rc,
4489 tr("Failed to lock media when merging to '%ls'"),
4490 pTarget->getLocationFull().raw());
4491 }
4492 }
4493 }
4494 catch (HRESULT aRC) { rc = aRC; }
4495
4496 if (FAILED(rc))
4497 {
4498 delete aMediumLockList;
4499 aMediumLockList = NULL;
4500 }
4501
4502 return rc;
4503}
4504
4505/**
4506 * Merges this hard disk to the specified hard disk which must be either its
4507 * direct ancestor or descendant.
4508 *
4509 * Given this hard disk is SOURCE and the specified hard disk is TARGET, we will
4510 * get two varians of the merge operation:
4511 *
4512 * forward merge
4513 * ------------------------->
4514 * [Extra] <- SOURCE <- Intermediate <- TARGET
4515 * Any Del Del LockWr
4516 *
4517 *
4518 * backward merge
4519 * <-------------------------
4520 * TARGET <- Intermediate <- SOURCE <- [Extra]
4521 * LockWr Del Del LockWr
4522 *
4523 * Each diagram shows the involved hard disks on the hard disk chain where
4524 * SOURCE and TARGET belong. Under each hard disk there is a state value which
4525 * the hard disk must have at a time of the mergeTo() call.
4526 *
4527 * The hard disks in the square braces may be absent (e.g. when the forward
4528 * operation takes place and SOURCE is the base hard disk, or when the backward
4529 * merge operation takes place and TARGET is the last child in the chain) but if
4530 * they present they are involved too as shown.
4531 *
4532 * Nor the source hard disk neither intermediate hard disks may be attached to
4533 * any VM directly or in the snapshot, otherwise this method will assert.
4534 *
4535 * The #prepareMergeTo() method must be called prior to this method to place all
4536 * involved to necessary states and perform other consistency checks.
4537 *
4538 * If @a aWait is @c true then this method will perform the operation on the
4539 * calling thread and will not return to the caller until the operation is
4540 * completed. When this method succeeds, all intermediate hard disk objects in
4541 * the chain will be uninitialized, the state of the target hard disk (and all
4542 * involved extra hard disks) will be restored. @a aMediumLockList will not be
4543 * deleted, whether the operation is successful or not. The caller has to do
4544 * this if appropriate. Note that this (source) hard disk is not uninitialized
4545 * because of possible AutoCaller instances held by the caller of this method
4546 * on the current thread. It's therefore the responsibility of the caller to
4547 * call Medium::uninit() after releasing all callers.
4548 *
4549 * If @a aWait is @c false then this method will create a thread to perform the
4550 * operation asynchronously and will return immediately. If the operation
4551 * succeeds, the thread will uninitialize the source hard disk object and all
4552 * intermediate hard disk objects in the chain, reset the state of the target
4553 * hard disk (and all involved extra hard disks) and delete @a aMediumLockList.
4554 * If the operation fails, the thread will only reset the states of all
4555 * involved hard disks and delete @a aMediumLockList.
4556 *
4557 * When this method fails (regardless of the @a aWait mode), it is a caller's
4558 * responsiblity to undo state changes and delete @a aMediumLockList using
4559 * #cancelMergeTo().
4560 *
4561 * If @a aProgress is not NULL but the object it points to is @c null then a new
4562 * progress object will be created and assigned to @a *aProgress on success,
4563 * otherwise the existing progress object is used. If Progress is NULL, then no
4564 * progress object is created/used at all. Note that @a aProgress cannot be
4565 * NULL when @a aWait is @c false (this method will assert in this case).
4566 *
4567 * @param pTarget Target hard disk.
4568 * @param fMergeForward Merge direction.
4569 * @param pParentForTarget New parent for target medium after merge.
4570 * @param aChildrenToReparent List of children of the source which will have
4571 * to be reparented to the target after merge.
4572 * @param aMediumLockList Medium locking information.
4573 * @param aProgress Where to find/store a Progress object to track operation
4574 * completion.
4575 * @param aWait @c true if this method should block instead of creating
4576 * an asynchronous thread.
4577 * @param pfNeedsSaveSettings Optional pointer to a bool that must have been initialized to false and that will be set to true
4578 * by this function if the caller should invoke VirtualBox::saveSettings() because the global settings have changed.
4579 * This only works in "wait" mode; otherwise saveSettings gets called automatically by the thread that was created,
4580 * and this parameter is ignored.
4581 *
4582 * @note Locks the tree lock for writing. Locks the hard disks from the chain
4583 * for writing.
4584 */
4585HRESULT Medium::mergeTo(const ComObjPtr<Medium> &pTarget,
4586 bool fMergeForward,
4587 const ComObjPtr<Medium> &pParentForTarget,
4588 const MediaList &aChildrenToReparent,
4589 MediumLockList *aMediumLockList,
4590 ComObjPtr <Progress> *aProgress,
4591 bool aWait,
4592 bool *pfNeedsSaveSettings)
4593{
4594 AssertReturn(pTarget != NULL, E_FAIL);
4595 AssertReturn(pTarget != this, E_FAIL);
4596 AssertReturn(aMediumLockList != NULL, E_FAIL);
4597 AssertReturn(aProgress != NULL || aWait == true, E_FAIL);
4598
4599 AutoCaller autoCaller(this);
4600 if (FAILED(autoCaller.rc())) return autoCaller.rc();
4601
4602 HRESULT rc = S_OK;
4603 ComObjPtr <Progress> pProgress;
4604 Medium::Task *pTask = NULL;
4605
4606 try
4607 {
4608 if (aProgress != NULL)
4609 {
4610 /* use the existing progress object... */
4611 pProgress = *aProgress;
4612
4613 /* ...but create a new one if it is null */
4614 if (pProgress.isNull())
4615 {
4616 Utf8Str tgtName;
4617 {
4618 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4619 tgtName = pTarget->getName();
4620 }
4621
4622 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4623
4624 pProgress.createObject();
4625 rc = pProgress->init(m->pVirtualBox,
4626 static_cast<IMedium*>(this),
4627 BstrFmt(tr("Merging hard disk '%s' to '%s'"),
4628 getName().raw(),
4629 tgtName.raw()),
4630 TRUE /* aCancelable */);
4631 if (FAILED(rc))
4632 throw rc;
4633 }
4634 }
4635
4636 /* setup task object to carry out the operation sync/async */
4637 pTask = new Medium::MergeTask(this, pTarget, fMergeForward,
4638 pParentForTarget, aChildrenToReparent,
4639 pProgress, aMediumLockList,
4640 aWait /* fKeepMediumLockList */);
4641 rc = pTask->rc();
4642 AssertComRC(rc);
4643 if (FAILED(rc))
4644 throw rc;
4645 }
4646 catch (HRESULT aRC) { rc = aRC; }
4647
4648 if (SUCCEEDED(rc))
4649 {
4650 if (aWait)
4651 rc = runNow(pTask, pfNeedsSaveSettings);
4652 else
4653 rc = startThread(pTask);
4654
4655 if (SUCCEEDED(rc) && aProgress != NULL)
4656 *aProgress = pProgress;
4657 }
4658 else if (pTask != NULL)
4659 delete pTask;
4660
4661 return rc;
4662}
4663
4664/**
4665 * Undoes what #prepareMergeTo() did. Must be called if #mergeTo() is not
4666 * called or fails. Frees memory occupied by @a aMediumLockList and unlocks
4667 * the medium objects in @a aChildrenToReparent.
4668 *
4669 * @param aChildrenToReparent List of children of the source which will have
4670 * to be reparented to the target after merge.
4671 * @param aMediumLockList Medium locking information.
4672 *
4673 * @note Locks the hard disks from the chain for writing.
4674 */
4675void Medium::cancelMergeTo(const MediaList &aChildrenToReparent,
4676 MediumLockList *aMediumLockList)
4677{
4678 AutoCaller autoCaller(this);
4679 AssertComRCReturnVoid(autoCaller.rc());
4680
4681 AssertReturnVoid(aMediumLockList != NULL);
4682
4683 /* Revert media marked for deletion to previous state. */
4684 HRESULT rc;
4685 MediumLockList::Base::const_iterator mediumListBegin =
4686 aMediumLockList->GetBegin();
4687 MediumLockList::Base::const_iterator mediumListEnd =
4688 aMediumLockList->GetEnd();
4689 for (MediumLockList::Base::const_iterator it = mediumListBegin;
4690 it != mediumListEnd;
4691 ++it)
4692 {
4693 const MediumLock &mediumLock = *it;
4694 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
4695 AutoWriteLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
4696
4697 if (pMedium->m->state == MediumState_Deleting)
4698 {
4699 rc = pMedium->unmarkForDeletion();
4700 AssertComRC(rc);
4701 }
4702 }
4703
4704 /* the destructor will do the work */
4705 delete aMediumLockList;
4706
4707 /* unlock the children which had to be reparented */
4708 for (MediaList::const_iterator it = aChildrenToReparent.begin();
4709 it != aChildrenToReparent.end();
4710 ++it)
4711 {
4712 const ComObjPtr<Medium> &pMedium = *it;
4713
4714 AutoWriteLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
4715 pMedium->UnlockWrite(NULL);
4716 }
4717}
4718
4719/**
4720 * Checks that the format ID is valid and sets it on success.
4721 *
4722 * Note that this method will caller-reference the format object on success!
4723 * This reference must be released somewhere to let the MediumFormat object be
4724 * uninitialized.
4725 *
4726 * @note Must be called from under this object's write lock.
4727 */
4728HRESULT Medium::setFormat(CBSTR aFormat)
4729{
4730 /* get the format object first */
4731 {
4732 AutoReadLock propsLock(m->pVirtualBox->systemProperties() COMMA_LOCKVAL_SRC_POS);
4733
4734 unconst(m->formatObj)
4735 = m->pVirtualBox->systemProperties()->mediumFormat(aFormat);
4736 if (m->formatObj.isNull())
4737 return setError(E_INVALIDARG,
4738 tr("Invalid hard disk storage format '%ls'"),
4739 aFormat);
4740
4741 /* reference the format permanently to prevent its unexpected
4742 * uninitialization */
4743 HRESULT rc = m->formatObj->addCaller();
4744 AssertComRCReturnRC(rc);
4745
4746 /* get properties (preinsert them as keys in the map). Note that the
4747 * map doesn't grow over the object life time since the set of
4748 * properties is meant to be constant. */
4749
4750 Assert(m->properties.empty());
4751
4752 for (MediumFormat::PropertyList::const_iterator it =
4753 m->formatObj->properties().begin();
4754 it != m->formatObj->properties().end();
4755 ++it)
4756 {
4757 m->properties.insert(std::make_pair(it->name, Bstr::Null));
4758 }
4759 }
4760
4761 unconst(m->strFormat) = aFormat;
4762
4763 return S_OK;
4764}
4765
4766/**
4767 * @note Also reused by Medium::Reset().
4768 *
4769 * @note Caller must hold the media tree write lock!
4770 */
4771HRESULT Medium::canClose()
4772{
4773 Assert(m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
4774
4775 if (getChildren().size() != 0)
4776 return setError(E_FAIL,
4777 tr("Cannot close medium '%s' because it has %d child hard disk(s)"),
4778 m->strLocationFull.raw(), getChildren().size());
4779
4780 return S_OK;
4781}
4782
4783/**
4784 * Calls either VirtualBox::unregisterImage or VirtualBox::unregisterHardDisk depending
4785 * on the device type of this medium.
4786 *
4787 * @param pfNeedsSaveSettings Optional pointer to a bool that must have been initialized to false and that will be set to true
4788 * by this function if the caller should invoke VirtualBox::saveSettings() because the global settings have changed.
4789 *
4790 * @note Caller must have locked the media tree lock for writing!
4791 */
4792HRESULT Medium::unregisterWithVirtualBox(bool *pfNeedsSaveSettings)
4793{
4794 /* Note that we need to de-associate ourselves from the parent to let
4795 * unregisterHardDisk() properly save the registry */
4796
4797 /* we modify mParent and access children */
4798 Assert(m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
4799
4800 Medium *pParentBackup = m->pParent;
4801 AssertReturn(getChildren().size() == 0, E_FAIL);
4802 if (m->pParent)
4803 deparent();
4804
4805 HRESULT rc = E_FAIL;
4806 switch (m->devType)
4807 {
4808 case DeviceType_DVD:
4809 rc = m->pVirtualBox->unregisterImage(this, DeviceType_DVD, pfNeedsSaveSettings);
4810 break;
4811
4812 case DeviceType_Floppy:
4813 rc = m->pVirtualBox->unregisterImage(this, DeviceType_Floppy, pfNeedsSaveSettings);
4814 break;
4815
4816 case DeviceType_HardDisk:
4817 rc = m->pVirtualBox->unregisterHardDisk(this, pfNeedsSaveSettings);
4818 break;
4819
4820 default:
4821 break;
4822 }
4823
4824 if (FAILED(rc))
4825 {
4826 if (pParentBackup)
4827 {
4828 /* re-associate with the parent as we are still relatives in the
4829 * registry */
4830 m->pParent = pParentBackup;
4831 m->pParent->m->llChildren.push_back(this);
4832 }
4833 }
4834
4835 return rc;
4836}
4837
4838/**
4839 * Returns the last error message collected by the vdErrorCall callback and
4840 * resets it.
4841 *
4842 * The error message is returned prepended with a dot and a space, like this:
4843 * <code>
4844 * ". <error_text> (%Rrc)"
4845 * </code>
4846 * to make it easily appendable to a more general error message. The @c %Rrc
4847 * format string is given @a aVRC as an argument.
4848 *
4849 * If there is no last error message collected by vdErrorCall or if it is a
4850 * null or empty string, then this function returns the following text:
4851 * <code>
4852 * " (%Rrc)"
4853 * </code>
4854 *
4855 * @note Doesn't do any object locking; it is assumed that the caller makes sure
4856 * the callback isn't called by more than one thread at a time.
4857 *
4858 * @param aVRC VBox error code to use when no error message is provided.
4859 */
4860Utf8Str Medium::vdError(int aVRC)
4861{
4862 Utf8Str error;
4863
4864 if (m->vdError.isEmpty())
4865 error = Utf8StrFmt(" (%Rrc)", aVRC);
4866 else
4867 error = Utf8StrFmt(".\n%s", m->vdError.raw());
4868
4869 m->vdError.setNull();
4870
4871 return error;
4872}
4873
4874/**
4875 * Error message callback.
4876 *
4877 * Puts the reported error message to the m->vdError field.
4878 *
4879 * @note Doesn't do any object locking; it is assumed that the caller makes sure
4880 * the callback isn't called by more than one thread at a time.
4881 *
4882 * @param pvUser The opaque data passed on container creation.
4883 * @param rc The VBox error code.
4884 * @param RT_SRC_POS_DECL Use RT_SRC_POS.
4885 * @param pszFormat Error message format string.
4886 * @param va Error message arguments.
4887 */
4888/*static*/
4889DECLCALLBACK(void) Medium::vdErrorCall(void *pvUser, int rc, RT_SRC_POS_DECL,
4890 const char *pszFormat, va_list va)
4891{
4892 NOREF(pszFile); NOREF(iLine); NOREF(pszFunction); /* RT_SRC_POS_DECL */
4893
4894 Medium *that = static_cast<Medium*>(pvUser);
4895 AssertReturnVoid(that != NULL);
4896
4897 if (that->m->vdError.isEmpty())
4898 that->m->vdError =
4899 Utf8StrFmt("%s (%Rrc)", Utf8StrFmtVA(pszFormat, va).raw(), rc);
4900 else
4901 that->m->vdError =
4902 Utf8StrFmt("%s.\n%s (%Rrc)", that->m->vdError.raw(),
4903 Utf8StrFmtVA(pszFormat, va).raw(), rc);
4904}
4905
4906/* static */
4907DECLCALLBACK(bool) Medium::vdConfigAreKeysValid(void *pvUser,
4908 const char * /* pszzValid */)
4909{
4910 Medium *that = static_cast<Medium*>(pvUser);
4911 AssertReturn(that != NULL, false);
4912
4913 /* we always return true since the only keys we have are those found in
4914 * VDBACKENDINFO */
4915 return true;
4916}
4917
4918/* static */
4919DECLCALLBACK(int) Medium::vdConfigQuerySize(void *pvUser, const char *pszName,
4920 size_t *pcbValue)
4921{
4922 AssertReturn(VALID_PTR(pcbValue), VERR_INVALID_POINTER);
4923
4924 Medium *that = static_cast<Medium*>(pvUser);
4925 AssertReturn(that != NULL, VERR_GENERAL_FAILURE);
4926
4927 Data::PropertyMap::const_iterator it =
4928 that->m->properties.find(Bstr(pszName));
4929 if (it == that->m->properties.end())
4930 return VERR_CFGM_VALUE_NOT_FOUND;
4931
4932 /* we interpret null values as "no value" in Medium */
4933 if (it->second.isEmpty())
4934 return VERR_CFGM_VALUE_NOT_FOUND;
4935
4936 *pcbValue = it->second.length() + 1 /* include terminator */;
4937
4938 return VINF_SUCCESS;
4939}
4940
4941/* static */
4942DECLCALLBACK(int) Medium::vdConfigQuery(void *pvUser, const char *pszName,
4943 char *pszValue, size_t cchValue)
4944{
4945 AssertReturn(VALID_PTR(pszValue), VERR_INVALID_POINTER);
4946
4947 Medium *that = static_cast<Medium*>(pvUser);
4948 AssertReturn(that != NULL, VERR_GENERAL_FAILURE);
4949
4950 Data::PropertyMap::const_iterator it =
4951 that->m->properties.find(Bstr(pszName));
4952 if (it == that->m->properties.end())
4953 return VERR_CFGM_VALUE_NOT_FOUND;
4954
4955 Utf8Str value = it->second;
4956 if (value.length() >= cchValue)
4957 return VERR_CFGM_NOT_ENOUGH_SPACE;
4958
4959 /* we interpret null values as "no value" in Medium */
4960 if (it->second.isEmpty())
4961 return VERR_CFGM_VALUE_NOT_FOUND;
4962
4963 memcpy(pszValue, value.c_str(), value.length() + 1);
4964
4965 return VINF_SUCCESS;
4966}
4967
4968/**
4969 * Starts a new thread driven by the appropriate Medium::Task::handler() method.
4970 *
4971 * @note When the task is executed by this method, IProgress::notifyComplete()
4972 * is automatically called for the progress object associated with this
4973 * task when the task is finished to signal the operation completion for
4974 * other threads asynchronously waiting for it.
4975 */
4976HRESULT Medium::startThread(Medium::Task *pTask)
4977{
4978#ifdef VBOX_WITH_MAIN_LOCK_VALIDATION
4979 /* Extreme paranoia: The calling thread should not hold the medium
4980 * tree lock or any medium lock. Since there is no separate lock class
4981 * for medium objects be even more strict: no other object locks. */
4982 Assert(!AutoLockHoldsLocksInClass(LOCKCLASS_LISTOFMEDIA));
4983 Assert(!AutoLockHoldsLocksInClass(getLockingClass()));
4984#endif
4985
4986 /// @todo use a more descriptive task name
4987 int vrc = RTThreadCreate(NULL, Medium::Task::fntMediumTask, pTask,
4988 0, RTTHREADTYPE_MAIN_HEAVY_WORKER, 0,
4989 "Medium::Task");
4990 if (RT_FAILURE(vrc))
4991 {
4992 delete pTask;
4993 ComAssertMsgRCRet(vrc,
4994 ("Could not create Medium::Task thread (%Rrc)\n",
4995 vrc),
4996 E_FAIL);
4997 }
4998
4999 return S_OK;
5000}
5001
5002/**
5003 * Fix the parent UUID of all children to point to this medium as their
5004 * parent.
5005 */
5006HRESULT Medium::fixParentUuidOfChildren(const MediaList &childrenToReparent)
5007{
5008 MediumLockList mediumLockList;
5009 HRESULT rc = createMediumLockList(true /* fFailIfInaccessible */,
5010 false /* fMediumWritable */,
5011 this,
5012 mediumLockList);
5013 AssertComRCReturnRC(rc);
5014
5015 try
5016 {
5017 PVBOXHDD hdd;
5018 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
5019 ComAssertRCThrow(vrc, E_FAIL);
5020
5021 try
5022 {
5023 MediumLockList::Base::iterator lockListBegin =
5024 mediumLockList.GetBegin();
5025 MediumLockList::Base::iterator lockListEnd =
5026 mediumLockList.GetEnd();
5027 for (MediumLockList::Base::iterator it = lockListBegin;
5028 it != lockListEnd;
5029 ++it)
5030 {
5031 MediumLock &mediumLock = *it;
5032 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5033 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5034
5035 // open the image
5036 vrc = VDOpen(hdd,
5037 pMedium->m->strFormat.c_str(),
5038 pMedium->m->strLocationFull.c_str(),
5039 VD_OPEN_FLAGS_READONLY,
5040 pMedium->m->vdDiskIfaces);
5041 if (RT_FAILURE(vrc))
5042 throw vrc;
5043 }
5044
5045 for (MediaList::const_iterator it = childrenToReparent.begin();
5046 it != childrenToReparent.end();
5047 ++it)
5048 {
5049 /* VD_OPEN_FLAGS_INFO since UUID is wrong yet */
5050 vrc = VDOpen(hdd,
5051 (*it)->m->strFormat.c_str(),
5052 (*it)->m->strLocationFull.c_str(),
5053 VD_OPEN_FLAGS_INFO,
5054 (*it)->m->vdDiskIfaces);
5055 if (RT_FAILURE(vrc))
5056 throw vrc;
5057
5058 vrc = VDSetParentUuid(hdd, VD_LAST_IMAGE, m->id);
5059 if (RT_FAILURE(vrc))
5060 throw vrc;
5061
5062 vrc = VDClose(hdd, false /* fDelete */);
5063 if (RT_FAILURE(vrc))
5064 throw vrc;
5065
5066 (*it)->UnlockWrite(NULL);
5067 }
5068 }
5069 catch (HRESULT aRC) { rc = aRC; }
5070 catch (int aVRC)
5071 {
5072 throw setError(E_FAIL,
5073 tr("Could not update medium UUID references to parent '%s' (%s)"),
5074 m->strLocationFull.raw(),
5075 vdError(aVRC).raw());
5076 }
5077
5078 VDDestroy(hdd);
5079 }
5080 catch (HRESULT aRC) { rc = aRC; }
5081
5082 return rc;
5083}
5084
5085/**
5086 * Runs Medium::Task::handler() on the current thread instead of creating
5087 * a new one.
5088 *
5089 * This call implies that it is made on another temporary thread created for
5090 * some asynchronous task. Avoid calling it from a normal thread since the task
5091 * operations are potentially lengthy and will block the calling thread in this
5092 * case.
5093 *
5094 * @note When the task is executed by this method, IProgress::notifyComplete()
5095 * is not called for the progress object associated with this task when
5096 * the task is finished. Instead, the result of the operation is returned
5097 * by this method directly and it's the caller's responsibility to
5098 * complete the progress object in this case.
5099 */
5100HRESULT Medium::runNow(Medium::Task *pTask,
5101 bool *pfNeedsSaveSettings)
5102{
5103#ifdef VBOX_WITH_MAIN_LOCK_VALIDATION
5104 /* Extreme paranoia: The calling thread should not hold the medium
5105 * tree lock or any medium lock. Since there is no separate lock class
5106 * for medium objects be even more strict: no other object locks. */
5107 Assert(!AutoLockHoldsLocksInClass(LOCKCLASS_LISTOFMEDIA));
5108 Assert(!AutoLockHoldsLocksInClass(getLockingClass()));
5109#endif
5110
5111 pTask->m_pfNeedsSaveSettings = pfNeedsSaveSettings;
5112
5113 /* NIL_RTTHREAD indicates synchronous call. */
5114 return (HRESULT)Medium::Task::fntMediumTask(NIL_RTTHREAD, pTask);
5115}
5116
5117/**
5118 * Implementation code for the "create base" task.
5119 *
5120 * This only gets started from Medium::CreateBaseStorage() and always runs
5121 * asynchronously. As a result, we always save the VirtualBox.xml file when
5122 * we're done here.
5123 *
5124 * @param task
5125 * @return
5126 */
5127HRESULT Medium::taskCreateBaseHandler(Medium::CreateBaseTask &task)
5128{
5129 HRESULT rc = S_OK;
5130
5131 /* these parameters we need after creation */
5132 uint64_t size = 0, logicalSize = 0;
5133 bool fGenerateUuid = false;
5134
5135 try
5136 {
5137 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
5138
5139 /* The object may request a specific UUID (through a special form of
5140 * the setLocation() argument). Otherwise we have to generate it */
5141 Guid id = m->id;
5142 fGenerateUuid = id.isEmpty();
5143 if (fGenerateUuid)
5144 {
5145 id.create();
5146 /* VirtualBox::registerHardDisk() will need UUID */
5147 unconst(m->id) = id;
5148 }
5149
5150 Utf8Str format(m->strFormat);
5151 Utf8Str location(m->strLocationFull);
5152 uint64_t capabilities = m->formatObj->capabilities();
5153 ComAssertThrow(capabilities & ( VD_CAP_CREATE_FIXED
5154 | VD_CAP_CREATE_DYNAMIC), E_FAIL);
5155 Assert(m->state == MediumState_Creating);
5156
5157 PVBOXHDD hdd;
5158 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
5159 ComAssertRCThrow(vrc, E_FAIL);
5160
5161 /* unlock before the potentially lengthy operation */
5162 thisLock.leave();
5163
5164 try
5165 {
5166 /* ensure the directory exists */
5167 rc = VirtualBox::ensureFilePathExists(location);
5168 if (FAILED(rc))
5169 throw rc;
5170
5171 PDMMEDIAGEOMETRY geo = { 0, 0, 0 }; /* auto-detect */
5172
5173 vrc = VDCreateBase(hdd,
5174 format.c_str(),
5175 location.c_str(),
5176 task.mSize * _1M,
5177 task.mVariant,
5178 NULL,
5179 &geo,
5180 &geo,
5181 id.raw(),
5182 VD_OPEN_FLAGS_NORMAL,
5183 NULL,
5184 task.mVDOperationIfaces);
5185 if (RT_FAILURE(vrc))
5186 {
5187 throw setError(E_FAIL,
5188 tr("Could not create the hard disk storage unit '%s'%s"),
5189 location.raw(), vdError(vrc).raw());
5190 }
5191
5192 size = VDGetFileSize(hdd, 0);
5193 logicalSize = VDGetSize(hdd, 0) / _1M;
5194 }
5195 catch (HRESULT aRC) { rc = aRC; }
5196
5197 VDDestroy(hdd);
5198 }
5199 catch (HRESULT aRC) { rc = aRC; }
5200
5201 if (SUCCEEDED(rc))
5202 {
5203 /* register with mVirtualBox as the last step and move to
5204 * Created state only on success (leaving an orphan file is
5205 * better than breaking media registry consistency) */
5206 bool fNeedsSaveSettings = false;
5207 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
5208 rc = m->pVirtualBox->registerHardDisk(this, &fNeedsSaveSettings);
5209 treeLock.release();
5210
5211 if (fNeedsSaveSettings)
5212 {
5213 AutoWriteLock vboxlock(m->pVirtualBox COMMA_LOCKVAL_SRC_POS);
5214 m->pVirtualBox->saveSettings();
5215 }
5216 }
5217
5218 // reenter the lock before changing state
5219 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
5220
5221 if (SUCCEEDED(rc))
5222 {
5223 m->state = MediumState_Created;
5224
5225 m->size = size;
5226 m->logicalSize = logicalSize;
5227 }
5228 else
5229 {
5230 /* back to NotCreated on failure */
5231 m->state = MediumState_NotCreated;
5232
5233 /* reset UUID to prevent it from being reused next time */
5234 if (fGenerateUuid)
5235 unconst(m->id).clear();
5236 }
5237
5238 return rc;
5239}
5240
5241/**
5242 * Implementation code for the "create diff" task.
5243 *
5244 * This task always gets started from Medium::createDiffStorage() and can run
5245 * synchronously or asynchronously depending on the "wait" parameter passed to
5246 * that function. If we run synchronously, the caller expects the bool
5247 * *pfNeedsSaveSettings to be set before returning; otherwise (in asynchronous
5248 * mode), we save the settings ourselves.
5249 *
5250 * @param task
5251 * @return
5252 */
5253HRESULT Medium::taskCreateDiffHandler(Medium::CreateDiffTask &task)
5254{
5255 HRESULT rc = S_OK;
5256
5257 bool fNeedsSaveSettings = false;
5258
5259 const ComObjPtr<Medium> &pTarget = task.mTarget;
5260
5261 uint64_t size = 0, logicalSize = 0;
5262 bool fGenerateUuid = false;
5263
5264 try
5265 {
5266 /* Lock both in {parent,child} order. */
5267 AutoMultiWriteLock2 mediaLock(this, pTarget COMMA_LOCKVAL_SRC_POS);
5268
5269 /* The object may request a specific UUID (through a special form of
5270 * the setLocation() argument). Otherwise we have to generate it */
5271 Guid targetId = pTarget->m->id;
5272 fGenerateUuid = targetId.isEmpty();
5273 if (fGenerateUuid)
5274 {
5275 targetId.create();
5276 /* VirtualBox::registerHardDisk() will need UUID */
5277 unconst(pTarget->m->id) = targetId;
5278 }
5279
5280 Guid id = m->id;
5281
5282 Utf8Str targetFormat(pTarget->m->strFormat);
5283 Utf8Str targetLocation(pTarget->m->strLocationFull);
5284 uint64_t capabilities = m->formatObj->capabilities();
5285 ComAssertThrow(capabilities & VD_CAP_CREATE_DYNAMIC, E_FAIL);
5286
5287 Assert(pTarget->m->state == MediumState_Creating);
5288 Assert(m->state == MediumState_LockedRead);
5289
5290 PVBOXHDD hdd;
5291 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
5292 ComAssertRCThrow(vrc, E_FAIL);
5293
5294 /* the two media are now protected by their non-default states;
5295 * unlock the media before the potentially lengthy operation */
5296 mediaLock.leave();
5297
5298 try
5299 {
5300 /* Open all hard disk images in the target chain but the last. */
5301 MediumLockList::Base::const_iterator targetListBegin =
5302 task.mpMediumLockList->GetBegin();
5303 MediumLockList::Base::const_iterator targetListEnd =
5304 task.mpMediumLockList->GetEnd();
5305 for (MediumLockList::Base::const_iterator it = targetListBegin;
5306 it != targetListEnd;
5307 ++it)
5308 {
5309 const MediumLock &mediumLock = *it;
5310 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5311
5312 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5313
5314 /* Skip over the target diff image */
5315 if (pMedium->m->state == MediumState_Creating)
5316 continue;
5317
5318 /* sanity check */
5319 Assert(pMedium->m->state == MediumState_LockedRead);
5320
5321 /* Open all images in appropriate mode. */
5322 vrc = VDOpen(hdd,
5323 pMedium->m->strFormat.c_str(),
5324 pMedium->m->strLocationFull.c_str(),
5325 VD_OPEN_FLAGS_READONLY,
5326 pMedium->m->vdDiskIfaces);
5327 if (RT_FAILURE(vrc))
5328 throw setError(E_FAIL,
5329 tr("Could not open the hard disk storage unit '%s'%s"),
5330 pMedium->m->strLocationFull.raw(),
5331 vdError(vrc).raw());
5332 }
5333
5334 /* ensure the target directory exists */
5335 rc = VirtualBox::ensureFilePathExists(targetLocation);
5336 if (FAILED(rc))
5337 throw rc;
5338
5339 vrc = VDCreateDiff(hdd,
5340 targetFormat.c_str(),
5341 targetLocation.c_str(),
5342 task.mVariant | VD_IMAGE_FLAGS_DIFF,
5343 NULL,
5344 targetId.raw(),
5345 id.raw(),
5346 VD_OPEN_FLAGS_NORMAL,
5347 pTarget->m->vdDiskIfaces,
5348 task.mVDOperationIfaces);
5349 if (RT_FAILURE(vrc))
5350 throw setError(E_FAIL,
5351 tr("Could not create the differencing hard disk storage unit '%s'%s"),
5352 targetLocation.raw(), vdError(vrc).raw());
5353
5354 size = VDGetFileSize(hdd, VD_LAST_IMAGE);
5355 logicalSize = VDGetSize(hdd, VD_LAST_IMAGE) / _1M;
5356 }
5357 catch (HRESULT aRC) { rc = aRC; }
5358
5359 VDDestroy(hdd);
5360 }
5361 catch (HRESULT aRC) { rc = aRC; }
5362
5363 if (SUCCEEDED(rc))
5364 {
5365 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
5366
5367 Assert(pTarget->m->pParent.isNull());
5368
5369 /* associate the child with the parent */
5370 pTarget->m->pParent = this;
5371 m->llChildren.push_back(pTarget);
5372
5373 /** @todo r=klaus neither target nor base() are locked,
5374 * potential race! */
5375 /* diffs for immutable hard disks are auto-reset by default */
5376 pTarget->m->autoReset = (getBase()->m->type == MediumType_Immutable);
5377
5378 /* register with mVirtualBox as the last step and move to
5379 * Created state only on success (leaving an orphan file is
5380 * better than breaking media registry consistency) */
5381 rc = m->pVirtualBox->registerHardDisk(pTarget, &fNeedsSaveSettings);
5382
5383 if (FAILED(rc))
5384 /* break the parent association on failure to register */
5385 deparent();
5386 }
5387
5388 AutoMultiWriteLock2 mediaLock(this, pTarget COMMA_LOCKVAL_SRC_POS);
5389
5390 if (SUCCEEDED(rc))
5391 {
5392 pTarget->m->state = MediumState_Created;
5393
5394 pTarget->m->size = size;
5395 pTarget->m->logicalSize = logicalSize;
5396 }
5397 else
5398 {
5399 /* back to NotCreated on failure */
5400 pTarget->m->state = MediumState_NotCreated;
5401
5402 pTarget->m->autoReset = FALSE;
5403
5404 /* reset UUID to prevent it from being reused next time */
5405 if (fGenerateUuid)
5406 unconst(pTarget->m->id).clear();
5407 }
5408
5409 if (task.isAsync())
5410 {
5411 if (fNeedsSaveSettings)
5412 {
5413 mediaLock.leave();
5414 AutoWriteLock vboxlock(m->pVirtualBox COMMA_LOCKVAL_SRC_POS);
5415 m->pVirtualBox->saveSettings();
5416 }
5417 }
5418 else
5419 // synchronous mode: report save settings result to caller
5420 if (task.m_pfNeedsSaveSettings)
5421 *task.m_pfNeedsSaveSettings = fNeedsSaveSettings;
5422
5423 /* deregister the task registered in createDiffStorage() */
5424 Assert(m->numCreateDiffTasks != 0);
5425 --m->numCreateDiffTasks;
5426
5427 /* Note that in sync mode, it's the caller's responsibility to
5428 * unlock the hard disk */
5429
5430 return rc;
5431}
5432
5433/**
5434 * Implementation code for the "merge" task.
5435 *
5436 * This task always gets started from Medium::mergeTo() and can run
5437 * synchronously or asynchrously depending on the "wait" parameter passed to
5438 * that function. If we run synchronously, the caller expects the bool
5439 * *pfNeedsSaveSettings to be set before returning; otherwise (in asynchronous
5440 * mode), we save the settings ourselves.
5441 *
5442 * @param task
5443 * @return
5444 */
5445HRESULT Medium::taskMergeHandler(Medium::MergeTask &task)
5446{
5447 HRESULT rc = S_OK;
5448
5449 const ComObjPtr<Medium> &pTarget = task.mTarget;
5450
5451 try
5452 {
5453 PVBOXHDD hdd;
5454 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
5455 ComAssertRCThrow(vrc, E_FAIL);
5456
5457 try
5458 {
5459 // Similar code appears in SessionMachine::onlineMergeMedium, so
5460 // if you make any changes below check whether they are applicable
5461 // in that context as well.
5462
5463 unsigned uTargetIdx = VD_LAST_IMAGE;
5464 unsigned uSourceIdx = VD_LAST_IMAGE;
5465 /* Open all hard disks in the chain. */
5466 MediumLockList::Base::iterator lockListBegin =
5467 task.mpMediumLockList->GetBegin();
5468 MediumLockList::Base::iterator lockListEnd =
5469 task.mpMediumLockList->GetEnd();
5470 unsigned i = 0;
5471 for (MediumLockList::Base::iterator it = lockListBegin;
5472 it != lockListEnd;
5473 ++it)
5474 {
5475 MediumLock &mediumLock = *it;
5476 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5477
5478 if (pMedium == this)
5479 uSourceIdx = i;
5480 else if (pMedium == pTarget)
5481 uTargetIdx = i;
5482
5483 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5484
5485 /*
5486 * complex sanity (sane complexity)
5487 *
5488 * The current image must be in the Deleting (image is merged)
5489 * or LockedRead (parent image) state if it is not the target.
5490 * If it is the target it must be in the LockedWrite state.
5491 */
5492 Assert( ( pMedium != pTarget
5493 && ( pMedium->m->state == MediumState_Deleting
5494 || pMedium->m->state == MediumState_LockedRead))
5495 || ( pMedium == pTarget
5496 && pMedium->m->state == MediumState_LockedWrite));
5497
5498 /*
5499 * Image must be the target, in the LockedRead state
5500 * or Deleting state where it is not allowed to be attached
5501 * to a virtual machine.
5502 */
5503 Assert( pMedium == pTarget
5504 || pMedium->m->state == MediumState_LockedRead
5505 || ( pMedium->m->backRefs.size() == 0
5506 && pMedium->m->state == MediumState_Deleting));
5507 /* The source medium must be in Deleting state. */
5508 Assert( pMedium != this
5509 || pMedium->m->state == MediumState_Deleting);
5510
5511 unsigned uOpenFlags = 0;
5512
5513 if ( pMedium->m->state == MediumState_LockedRead
5514 || pMedium->m->state == MediumState_Deleting)
5515 uOpenFlags = VD_OPEN_FLAGS_READONLY;
5516
5517 /* Open the image */
5518 vrc = VDOpen(hdd,
5519 pMedium->m->strFormat.c_str(),
5520 pMedium->m->strLocationFull.c_str(),
5521 uOpenFlags,
5522 pMedium->m->vdDiskIfaces);
5523 if (RT_FAILURE(vrc))
5524 throw vrc;
5525
5526 i++;
5527 }
5528
5529 ComAssertThrow( uSourceIdx != VD_LAST_IMAGE
5530 && uTargetIdx != VD_LAST_IMAGE, E_FAIL);
5531
5532 vrc = VDMerge(hdd, uSourceIdx, uTargetIdx,
5533 task.mVDOperationIfaces);
5534 if (RT_FAILURE(vrc))
5535 throw vrc;
5536
5537 /* update parent UUIDs */
5538 if (!task.mfMergeForward)
5539 {
5540 /* we need to update UUIDs of all source's children
5541 * which cannot be part of the container at once so
5542 * add each one in there individually */
5543 if (task.mChildrenToReparent.size() > 0)
5544 {
5545 for (MediaList::const_iterator it = task.mChildrenToReparent.begin();
5546 it != task.mChildrenToReparent.end();
5547 ++it)
5548 {
5549 /* VD_OPEN_FLAGS_INFO since UUID is wrong yet */
5550 vrc = VDOpen(hdd,
5551 (*it)->m->strFormat.c_str(),
5552 (*it)->m->strLocationFull.c_str(),
5553 VD_OPEN_FLAGS_INFO,
5554 (*it)->m->vdDiskIfaces);
5555 if (RT_FAILURE(vrc))
5556 throw vrc;
5557
5558 vrc = VDSetParentUuid(hdd, VD_LAST_IMAGE,
5559 pTarget->m->id);
5560 if (RT_FAILURE(vrc))
5561 throw vrc;
5562
5563 vrc = VDClose(hdd, false /* fDelete */);
5564 if (RT_FAILURE(vrc))
5565 throw vrc;
5566
5567 (*it)->UnlockWrite(NULL);
5568 }
5569 }
5570 }
5571 }
5572 catch (HRESULT aRC) { rc = aRC; }
5573 catch (int aVRC)
5574 {
5575 throw setError(E_FAIL,
5576 tr("Could not merge the hard disk '%s' to '%s'%s"),
5577 m->strLocationFull.raw(),
5578 pTarget->m->strLocationFull.raw(),
5579 vdError(aVRC).raw());
5580 }
5581
5582 VDDestroy(hdd);
5583 }
5584 catch (HRESULT aRC) { rc = aRC; }
5585
5586 HRESULT rc2;
5587
5588 if (SUCCEEDED(rc))
5589 {
5590 /* all hard disks but the target were successfully deleted by
5591 * VDMerge; reparent the last one and uninitialize deleted media. */
5592
5593 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
5594
5595 if (task.mfMergeForward)
5596 {
5597 /* first, unregister the target since it may become a base
5598 * hard disk which needs re-registration */
5599 rc2 = m->pVirtualBox->unregisterHardDisk(pTarget, NULL /*&fNeedsSaveSettings*/);
5600 AssertComRC(rc2);
5601
5602 /* then, reparent it and disconnect the deleted branch at
5603 * both ends (chain->parent() is source's parent) */
5604 pTarget->deparent();
5605 pTarget->m->pParent = task.mParentForTarget;
5606 if (pTarget->m->pParent)
5607 {
5608 pTarget->m->pParent->m->llChildren.push_back(pTarget);
5609 deparent();
5610 }
5611
5612 /* then, register again */
5613 rc2 = m->pVirtualBox->registerHardDisk(pTarget, NULL /*&fNeedsSaveSettings*/);
5614 AssertComRC(rc2);
5615 }
5616 else
5617 {
5618 Assert(pTarget->getChildren().size() == 1);
5619 Medium *targetChild = pTarget->getChildren().front();
5620
5621 /* disconnect the deleted branch at the elder end */
5622 targetChild->deparent();
5623
5624 /* reparent source's children and disconnect the deleted
5625 * branch at the younger end */
5626 if (task.mChildrenToReparent.size() > 0)
5627 {
5628 /* obey {parent,child} lock order */
5629 AutoWriteLock sourceLock(this COMMA_LOCKVAL_SRC_POS);
5630
5631 for (MediaList::const_iterator it = task.mChildrenToReparent.begin();
5632 it != task.mChildrenToReparent.end();
5633 it++)
5634 {
5635 Medium *pMedium = *it;
5636 AutoWriteLock childLock(pMedium COMMA_LOCKVAL_SRC_POS);
5637
5638 pMedium->deparent(); // removes pMedium from source
5639 pMedium->setParent(pTarget);
5640 }
5641 }
5642 }
5643
5644 /* unregister and uninitialize all hard disks removed by the merge */
5645 MediumLockList::Base::iterator lockListBegin =
5646 task.mpMediumLockList->GetBegin();
5647 MediumLockList::Base::iterator lockListEnd =
5648 task.mpMediumLockList->GetEnd();
5649 for (MediumLockList::Base::iterator it = lockListBegin;
5650 it != lockListEnd;
5651 )
5652 {
5653 MediumLock &mediumLock = *it;
5654 /* Create a real copy of the medium pointer, as the medium
5655 * lock deletion below would invalidate the referenced object. */
5656 const ComObjPtr<Medium> pMedium = mediumLock.GetMedium();
5657
5658 /* The target and all images not merged (readonly) are skipped */
5659 if ( pMedium == pTarget
5660 || pMedium->m->state == MediumState_LockedRead)
5661 {
5662 ++it;
5663 continue;
5664 }
5665
5666 rc2 = pMedium->m->pVirtualBox->unregisterHardDisk(pMedium,
5667 NULL /*pfNeedsSaveSettings*/);
5668 AssertComRC(rc2);
5669
5670 /* now, uninitialize the deleted hard disk (note that
5671 * due to the Deleting state, uninit() will not touch
5672 * the parent-child relationship so we need to
5673 * uninitialize each disk individually) */
5674
5675 /* note that the operation initiator hard disk (which is
5676 * normally also the source hard disk) is a special case
5677 * -- there is one more caller added by Task to it which
5678 * we must release. Also, if we are in sync mode, the
5679 * caller may still hold an AutoCaller instance for it
5680 * and therefore we cannot uninit() it (it's therefore
5681 * the caller's responsibility) */
5682 if (pMedium == this)
5683 {
5684 Assert(getChildren().size() == 0);
5685 Assert(m->backRefs.size() == 0);
5686 task.mMediumCaller.release();
5687 }
5688
5689 /* Delete the medium lock list entry, which also releases the
5690 * caller added by MergeChain before uninit() and updates the
5691 * iterator to point to the right place. */
5692 rc2 = task.mpMediumLockList->RemoveByIterator(it);
5693 AssertComRC(rc2);
5694
5695 if (task.isAsync() || pMedium != this)
5696 pMedium->uninit();
5697 }
5698 }
5699
5700 if (task.isAsync())
5701 {
5702 // in asynchronous mode, save settings now
5703 AutoWriteLock vboxlock(m->pVirtualBox COMMA_LOCKVAL_SRC_POS);
5704 m->pVirtualBox->saveSettings();
5705 }
5706 else
5707 // synchronous mode: report save settings result to caller
5708 if (task.m_pfNeedsSaveSettings)
5709 *task.m_pfNeedsSaveSettings = true;
5710
5711 if (FAILED(rc))
5712 {
5713 /* Here we come if either VDMerge() failed (in which case we
5714 * assume that it tried to do everything to make a further
5715 * retry possible -- e.g. not deleted intermediate hard disks
5716 * and so on) or VirtualBox::saveSettings() failed (where we
5717 * should have the original tree but with intermediate storage
5718 * units deleted by VDMerge()). We have to only restore states
5719 * (through the MergeChain dtor) unless we are run synchronously
5720 * in which case it's the responsibility of the caller as stated
5721 * in the mergeTo() docs. The latter also implies that we
5722 * don't own the merge chain, so release it in this case. */
5723 if (task.isAsync())
5724 {
5725 Assert(task.mChildrenToReparent.size() == 0);
5726 cancelMergeTo(task.mChildrenToReparent, task.mpMediumLockList);
5727 }
5728 }
5729
5730 return rc;
5731}
5732
5733/**
5734 * Implementation code for the "clone" task.
5735 *
5736 * This only gets started from Medium::CloneTo() and always runs asynchronously.
5737 * As a result, we always save the VirtualBox.xml file when we're done here.
5738 *
5739 * @param task
5740 * @return
5741 */
5742HRESULT Medium::taskCloneHandler(Medium::CloneTask &task)
5743{
5744 HRESULT rc = S_OK;
5745
5746 const ComObjPtr<Medium> &pTarget = task.mTarget;
5747 const ComObjPtr<Medium> &pParent = task.mParent;
5748
5749 bool fCreatingTarget = false;
5750
5751 uint64_t size = 0, logicalSize = 0;
5752 bool fGenerateUuid = false;
5753
5754 try
5755 {
5756 /* Lock all in {parent,child} order. The lock is also used as a
5757 * signal from the task initiator (which releases it only after
5758 * RTThreadCreate()) that we can start the job. */
5759 AutoMultiWriteLock3 thisLock(this, pTarget, pParent COMMA_LOCKVAL_SRC_POS);
5760
5761 fCreatingTarget = pTarget->m->state == MediumState_Creating;
5762
5763 /* The object may request a specific UUID (through a special form of
5764 * the setLocation() argument). Otherwise we have to generate it */
5765 Guid targetId = pTarget->m->id;
5766 fGenerateUuid = targetId.isEmpty();
5767 if (fGenerateUuid)
5768 {
5769 targetId.create();
5770 /* VirtualBox::registerHardDisk() will need UUID */
5771 unconst(pTarget->m->id) = targetId;
5772 }
5773
5774 PVBOXHDD hdd;
5775 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
5776 ComAssertRCThrow(vrc, E_FAIL);
5777
5778 try
5779 {
5780 /* Open all hard disk images in the source chain. */
5781 MediumLockList::Base::const_iterator sourceListBegin =
5782 task.mpSourceMediumLockList->GetBegin();
5783 MediumLockList::Base::const_iterator sourceListEnd =
5784 task.mpSourceMediumLockList->GetEnd();
5785 for (MediumLockList::Base::const_iterator it = sourceListBegin;
5786 it != sourceListEnd;
5787 ++it)
5788 {
5789 const MediumLock &mediumLock = *it;
5790 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5791 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5792
5793 /* sanity check */
5794 Assert(pMedium->m->state == MediumState_LockedRead);
5795
5796 /** Open all images in read-only mode. */
5797 vrc = VDOpen(hdd,
5798 pMedium->m->strFormat.c_str(),
5799 pMedium->m->strLocationFull.c_str(),
5800 VD_OPEN_FLAGS_READONLY,
5801 pMedium->m->vdDiskIfaces);
5802 if (RT_FAILURE(vrc))
5803 throw setError(E_FAIL,
5804 tr("Could not open the hard disk storage unit '%s'%s"),
5805 pMedium->m->strLocationFull.raw(),
5806 vdError(vrc).raw());
5807 }
5808
5809 Utf8Str targetFormat(pTarget->m->strFormat);
5810 Utf8Str targetLocation(pTarget->m->strLocationFull);
5811
5812 Assert( pTarget->m->state == MediumState_Creating
5813 || pTarget->m->state == MediumState_LockedWrite);
5814 Assert(m->state == MediumState_LockedRead);
5815 Assert(pParent.isNull() || pParent->m->state == MediumState_LockedRead);
5816
5817 /* unlock before the potentially lengthy operation */
5818 thisLock.leave();
5819
5820 /* ensure the target directory exists */
5821 rc = VirtualBox::ensureFilePathExists(targetLocation);
5822 if (FAILED(rc))
5823 throw rc;
5824
5825 PVBOXHDD targetHdd;
5826 vrc = VDCreate(m->vdDiskIfaces, &targetHdd);
5827 ComAssertRCThrow(vrc, E_FAIL);
5828
5829 try
5830 {
5831 /* Open all hard disk images in the target chain. */
5832 MediumLockList::Base::const_iterator targetListBegin =
5833 task.mpTargetMediumLockList->GetBegin();
5834 MediumLockList::Base::const_iterator targetListEnd =
5835 task.mpTargetMediumLockList->GetEnd();
5836 for (MediumLockList::Base::const_iterator it = targetListBegin;
5837 it != targetListEnd;
5838 ++it)
5839 {
5840 const MediumLock &mediumLock = *it;
5841 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5842
5843 /* If the target medium is not created yet there's no
5844 * reason to open it. */
5845 if (pMedium == pTarget && fCreatingTarget)
5846 continue;
5847
5848 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5849
5850 /* sanity check */
5851 Assert( pMedium->m->state == MediumState_LockedRead
5852 || pMedium->m->state == MediumState_LockedWrite);
5853
5854 /* Open all images in appropriate mode. */
5855 vrc = VDOpen(targetHdd,
5856 pMedium->m->strFormat.c_str(),
5857 pMedium->m->strLocationFull.c_str(),
5858 (pMedium->m->state == MediumState_LockedWrite) ? VD_OPEN_FLAGS_NORMAL : VD_OPEN_FLAGS_READONLY,
5859 pMedium->m->vdDiskIfaces);
5860 if (RT_FAILURE(vrc))
5861 throw setError(E_FAIL,
5862 tr("Could not open the hard disk storage unit '%s'%s"),
5863 pMedium->m->strLocationFull.raw(),
5864 vdError(vrc).raw());
5865 }
5866
5867 /** @todo r=klaus target isn't locked, race getting the state */
5868 vrc = VDCopy(hdd,
5869 VD_LAST_IMAGE,
5870 targetHdd,
5871 targetFormat.c_str(),
5872 (fCreatingTarget) ? targetLocation.raw() : (char *)NULL,
5873 false,
5874 0,
5875 task.mVariant,
5876 targetId.raw(),
5877 NULL,
5878 pTarget->m->vdDiskIfaces,
5879 task.mVDOperationIfaces);
5880 if (RT_FAILURE(vrc))
5881 throw setError(E_FAIL,
5882 tr("Could not create the clone hard disk '%s'%s"),
5883 targetLocation.raw(), vdError(vrc).raw());
5884
5885 size = VDGetFileSize(targetHdd, VD_LAST_IMAGE);
5886 logicalSize = VDGetSize(targetHdd, VD_LAST_IMAGE) / _1M;
5887 }
5888 catch (HRESULT aRC) { rc = aRC; }
5889
5890 VDDestroy(targetHdd);
5891 }
5892 catch (HRESULT aRC) { rc = aRC; }
5893
5894 VDDestroy(hdd);
5895 }
5896 catch (HRESULT aRC) { rc = aRC; }
5897
5898 /* Only do the parent changes for newly created images. */
5899 if (SUCCEEDED(rc) && fCreatingTarget)
5900 {
5901 /* we set mParent & children() */
5902 AutoWriteLock alock2(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
5903
5904 Assert(pTarget->m->pParent.isNull());
5905
5906 if (pParent)
5907 {
5908 /* associate the clone with the parent and deassociate
5909 * from VirtualBox */
5910 pTarget->m->pParent = pParent;
5911 pParent->m->llChildren.push_back(pTarget);
5912
5913 /* register with mVirtualBox as the last step and move to
5914 * Created state only on success (leaving an orphan file is
5915 * better than breaking media registry consistency) */
5916 rc = pParent->m->pVirtualBox->registerHardDisk(pTarget, NULL /* pfNeedsSaveSettings */);
5917
5918 if (FAILED(rc))
5919 /* break parent association on failure to register */
5920 pTarget->deparent(); // removes target from parent
5921 }
5922 else
5923 {
5924 /* just register */
5925 rc = m->pVirtualBox->registerHardDisk(pTarget, NULL /* pfNeedsSaveSettings */);
5926 }
5927 }
5928
5929 if (fCreatingTarget)
5930 {
5931 AutoWriteLock mLock(pTarget COMMA_LOCKVAL_SRC_POS);
5932
5933 if (SUCCEEDED(rc))
5934 {
5935 pTarget->m->state = MediumState_Created;
5936
5937 pTarget->m->size = size;
5938 pTarget->m->logicalSize = logicalSize;
5939 }
5940 else
5941 {
5942 /* back to NotCreated on failure */
5943 pTarget->m->state = MediumState_NotCreated;
5944
5945 /* reset UUID to prevent it from being reused next time */
5946 if (fGenerateUuid)
5947 unconst(pTarget->m->id).clear();
5948 }
5949 }
5950
5951 // now, at the end of this task (always asynchronous), save the settings
5952 {
5953 AutoWriteLock vboxlock(m->pVirtualBox COMMA_LOCKVAL_SRC_POS);
5954 m->pVirtualBox->saveSettings();
5955 }
5956
5957 /* Everything is explicitly unlocked when the task exits,
5958 * as the task destruction also destroys the source chain. */
5959
5960 /* Make sure the source chain is released early. It could happen
5961 * that we get a deadlock in Appliance::Import when Medium::Close
5962 * is called & the source chain is released at the same time. */
5963 task.mpSourceMediumLockList->Clear();
5964
5965 return rc;
5966}
5967
5968/**
5969 * Implementation code for the "delete" task.
5970 *
5971 * This task always gets started from Medium::deleteStorage() and can run
5972 * synchronously or asynchrously depending on the "wait" parameter passed to
5973 * that function.
5974 *
5975 * @param task
5976 * @return
5977 */
5978HRESULT Medium::taskDeleteHandler(Medium::DeleteTask &task)
5979{
5980 NOREF(task);
5981 HRESULT rc = S_OK;
5982
5983 try
5984 {
5985 /* The lock is also used as a signal from the task initiator (which
5986 * releases it only after RTThreadCreate()) that we can start the job */
5987 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
5988
5989 PVBOXHDD hdd;
5990 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
5991 ComAssertRCThrow(vrc, E_FAIL);
5992
5993 Utf8Str format(m->strFormat);
5994 Utf8Str location(m->strLocationFull);
5995
5996 /* unlock before the potentially lengthy operation */
5997 Assert(m->state == MediumState_Deleting);
5998 thisLock.release();
5999
6000 try
6001 {
6002 vrc = VDOpen(hdd,
6003 format.c_str(),
6004 location.c_str(),
6005 VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_INFO,
6006 m->vdDiskIfaces);
6007 if (RT_SUCCESS(vrc))
6008 vrc = VDClose(hdd, true /* fDelete */);
6009
6010 if (RT_FAILURE(vrc))
6011 throw setError(E_FAIL,
6012 tr("Could not delete the hard disk storage unit '%s'%s"),
6013 location.raw(), vdError(vrc).raw());
6014
6015 }
6016 catch (HRESULT aRC) { rc = aRC; }
6017
6018 VDDestroy(hdd);
6019 }
6020 catch (HRESULT aRC) { rc = aRC; }
6021
6022 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
6023
6024 /* go to the NotCreated state even on failure since the storage
6025 * may have been already partially deleted and cannot be used any
6026 * more. One will be able to manually re-open the storage if really
6027 * needed to re-register it. */
6028 m->state = MediumState_NotCreated;
6029
6030 /* Reset UUID to prevent Create* from reusing it again */
6031 unconst(m->id).clear();
6032
6033 return rc;
6034}
6035
6036/**
6037 * Implementation code for the "reset" task.
6038 *
6039 * This always gets started asynchronously from Medium::Reset().
6040 *
6041 * @param task
6042 * @return
6043 */
6044HRESULT Medium::taskResetHandler(Medium::ResetTask &task)
6045{
6046 HRESULT rc = S_OK;
6047
6048 uint64_t size = 0, logicalSize = 0;
6049
6050 try
6051 {
6052 /* The lock is also used as a signal from the task initiator (which
6053 * releases it only after RTThreadCreate()) that we can start the job */
6054 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
6055
6056 /// @todo Below we use a pair of delete/create operations to reset
6057 /// the diff contents but the most efficient way will of course be
6058 /// to add a VDResetDiff() API call
6059
6060 PVBOXHDD hdd;
6061 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
6062 ComAssertRCThrow(vrc, E_FAIL);
6063
6064 Guid id = m->id;
6065 Utf8Str format(m->strFormat);
6066 Utf8Str location(m->strLocationFull);
6067
6068 Medium *pParent = m->pParent;
6069 Guid parentId = pParent->m->id;
6070 Utf8Str parentFormat(pParent->m->strFormat);
6071 Utf8Str parentLocation(pParent->m->strLocationFull);
6072
6073 Assert(m->state == MediumState_LockedWrite);
6074
6075 /* unlock before the potentially lengthy operation */
6076 thisLock.release();
6077
6078 try
6079 {
6080 /* Open all hard disk images in the target chain but the last. */
6081 MediumLockList::Base::const_iterator targetListBegin =
6082 task.mpMediumLockList->GetBegin();
6083 MediumLockList::Base::const_iterator targetListEnd =
6084 task.mpMediumLockList->GetEnd();
6085 for (MediumLockList::Base::const_iterator it = targetListBegin;
6086 it != targetListEnd;
6087 ++it)
6088 {
6089 const MediumLock &mediumLock = *it;
6090 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
6091
6092 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
6093
6094 /* sanity check */
6095 Assert(pMedium->m->state == MediumState_LockedRead);
6096
6097 /* Open all images in appropriate mode. */
6098 vrc = VDOpen(hdd,
6099 pMedium->m->strFormat.c_str(),
6100 pMedium->m->strLocationFull.c_str(),
6101 VD_OPEN_FLAGS_READONLY,
6102 pMedium->m->vdDiskIfaces);
6103 if (RT_FAILURE(vrc))
6104 throw setError(E_FAIL,
6105 tr("Could not open the hard disk storage unit '%s'%s"),
6106 pMedium->m->strLocationFull.raw(),
6107 vdError(vrc).raw());
6108
6109 /* Done when we hit the image which should be reset */
6110 if (pMedium == this)
6111 break;
6112 }
6113
6114 /* first, delete the storage unit */
6115 vrc = VDClose(hdd, true /* fDelete */);
6116 if (RT_FAILURE(vrc))
6117 throw setError(E_FAIL,
6118 tr("Could not delete the hard disk storage unit '%s'%s"),
6119 location.raw(), vdError(vrc).raw());
6120
6121 /* next, create it again */
6122 vrc = VDOpen(hdd,
6123 parentFormat.c_str(),
6124 parentLocation.c_str(),
6125 VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_INFO,
6126 m->vdDiskIfaces);
6127 if (RT_FAILURE(vrc))
6128 throw setError(E_FAIL,
6129 tr("Could not open the hard disk storage unit '%s'%s"),
6130 parentLocation.raw(), vdError(vrc).raw());
6131
6132 vrc = VDCreateDiff(hdd,
6133 format.c_str(),
6134 location.c_str(),
6135 /// @todo use the same image variant as before
6136 VD_IMAGE_FLAGS_NONE,
6137 NULL,
6138 id.raw(),
6139 parentId.raw(),
6140 VD_OPEN_FLAGS_NORMAL,
6141 m->vdDiskIfaces,
6142 task.mVDOperationIfaces);
6143 if (RT_FAILURE(vrc))
6144 throw setError(E_FAIL,
6145 tr("Could not create the differencing hard disk storage unit '%s'%s"),
6146 location.raw(), vdError(vrc).raw());
6147
6148 size = VDGetFileSize(hdd, VD_LAST_IMAGE);
6149 logicalSize = VDGetSize(hdd, VD_LAST_IMAGE) / _1M;
6150 }
6151 catch (HRESULT aRC) { rc = aRC; }
6152
6153 VDDestroy(hdd);
6154 }
6155 catch (HRESULT aRC) { rc = aRC; }
6156
6157 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
6158
6159 m->size = size;
6160 m->logicalSize = logicalSize;
6161
6162 if (task.isAsync())
6163 {
6164 /* unlock ourselves when done */
6165 HRESULT rc2 = UnlockWrite(NULL);
6166 AssertComRC(rc2);
6167 }
6168
6169 /* Note that in sync mode, it's the caller's responsibility to
6170 * unlock the hard disk */
6171
6172 return rc;
6173}
6174
6175/**
6176 * Implementation code for the "compact" task.
6177 *
6178 * @param task
6179 * @return
6180 */
6181HRESULT Medium::taskCompactHandler(Medium::CompactTask &task)
6182{
6183 HRESULT rc = S_OK;
6184
6185 /* Lock all in {parent,child} order. The lock is also used as a
6186 * signal from the task initiator (which releases it only after
6187 * RTThreadCreate()) that we can start the job. */
6188 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
6189
6190 try
6191 {
6192 PVBOXHDD hdd;
6193 int vrc = VDCreate(m->vdDiskIfaces, &hdd);
6194 ComAssertRCThrow(vrc, E_FAIL);
6195
6196 try
6197 {
6198 /* Open all hard disk images in the chain. */
6199 MediumLockList::Base::const_iterator mediumListBegin =
6200 task.mpMediumLockList->GetBegin();
6201 MediumLockList::Base::const_iterator mediumListEnd =
6202 task.mpMediumLockList->GetEnd();
6203 MediumLockList::Base::const_iterator mediumListLast =
6204 mediumListEnd;
6205 mediumListLast--;
6206 for (MediumLockList::Base::const_iterator it = mediumListBegin;
6207 it != mediumListEnd;
6208 ++it)
6209 {
6210 const MediumLock &mediumLock = *it;
6211 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
6212 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
6213
6214 /* sanity check */
6215 if (it == mediumListLast)
6216 Assert(pMedium->m->state == MediumState_LockedWrite);
6217 else
6218 Assert(pMedium->m->state == MediumState_LockedRead);
6219
6220 /** Open all images but last in read-only mode. */
6221 vrc = VDOpen(hdd,
6222 pMedium->m->strFormat.c_str(),
6223 pMedium->m->strLocationFull.c_str(),
6224 (it == mediumListLast) ? VD_OPEN_FLAGS_NORMAL : VD_OPEN_FLAGS_READONLY,
6225 pMedium->m->vdDiskIfaces);
6226 if (RT_FAILURE(vrc))
6227 throw setError(E_FAIL,
6228 tr("Could not open the hard disk storage unit '%s'%s"),
6229 pMedium->m->strLocationFull.raw(),
6230 vdError(vrc).raw());
6231 }
6232
6233 Assert(m->state == MediumState_LockedWrite);
6234
6235 Utf8Str location(m->strLocationFull);
6236
6237 /* unlock before the potentially lengthy operation */
6238 thisLock.leave();
6239
6240 vrc = VDCompact(hdd, VD_LAST_IMAGE, task.mVDOperationIfaces);
6241 if (RT_FAILURE(vrc))
6242 {
6243 if (vrc == VERR_NOT_SUPPORTED)
6244 throw setError(VBOX_E_NOT_SUPPORTED,
6245 tr("Compacting is not yet supported for hard disk '%s'"),
6246 location.raw());
6247 else if (vrc == VERR_NOT_IMPLEMENTED)
6248 throw setError(E_NOTIMPL,
6249 tr("Compacting is not implemented, hard disk '%s'"),
6250 location.raw());
6251 else
6252 throw setError(E_FAIL,
6253 tr("Could not compact hard disk '%s'%s"),
6254 location.raw(),
6255 vdError(vrc).raw());
6256 }
6257 }
6258 catch (HRESULT aRC) { rc = aRC; }
6259
6260 VDDestroy(hdd);
6261 }
6262 catch (HRESULT aRC) { rc = aRC; }
6263
6264 /* Everything is explicitly unlocked when the task exits,
6265 * as the task destruction also destroys the image chain. */
6266
6267 return rc;
6268}
6269
6270/* vi: set tabstop=4 shiftwidth=4 expandtab: */
Note: See TracBrowser for help on using the repository browser.

© 2025 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette