VirtualBox

source: vbox/trunk/src/VBox/Main/src-server/MediumImpl.cpp@ 48366

Last change on this file since 48366 was 48297, checked in by vboxsync, 11 years ago

Main/Medium: redesign API level medium locking, needed conversions from MediaList to MediumLockLists in several places, forced cleanups elsewhere, too
Main/Token: introduced token objects for controlling the unlocking, will be used as a general concept in the future
Main/Snapshot: snapshot deletion needed significant cleanups as it was still using many shortcuts, directly calling the API to lock media instead of using lock lists. Now much better, and the online snapshot deletion is also a lot cleaner as it no longer passes unnecessary parameters around which are already known in the machine/snapshot code
Main/MediumLock: small improvements, now has a mode which skips locking already locked media, needed by the Snapshot code where we have overlapping lock lists and have to update the original one instead
Main/Console+Session+Machine: follow-up changes for the online snapshot merging parameter passing simplification, plus an unrelated lock order violation fix in Machine which happens only for inaccessible machines
Main/testcase: update correspondingly
doc: update SDK reference

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 282.3 KB
Line 
1/* $Id: MediumImpl.cpp 48297 2013-09-05 09:57:44Z vboxsync $ */
2/** @file
3 * VirtualBox COM class implementation
4 */
5
6/*
7 * Copyright (C) 2008-2013 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 */
17
18#include "MediumImpl.h"
19#include "TokenImpl.h"
20#include "ProgressImpl.h"
21#include "SystemPropertiesImpl.h"
22#include "VirtualBoxImpl.h"
23
24#include "AutoCaller.h"
25#include "Logging.h"
26
27#include <VBox/com/array.h>
28#include "VBox/com/MultiResult.h"
29#include "VBox/com/ErrorInfo.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#include <iprt/cpp/utils.h>
39
40#include <VBox/vd.h>
41
42#include <algorithm>
43#include <list>
44
45typedef std::list<Guid> GuidList;
46
47////////////////////////////////////////////////////////////////////////////////
48//
49// Medium data definition
50//
51////////////////////////////////////////////////////////////////////////////////
52
53/** Describes how a machine refers to this medium. */
54struct BackRef
55{
56 /** Equality predicate for stdc++. */
57 struct EqualsTo : public std::unary_function <BackRef, bool>
58 {
59 explicit EqualsTo(const Guid &aMachineId) : machineId(aMachineId) {}
60
61 bool operator()(const argument_type &aThat) const
62 {
63 return aThat.machineId == machineId;
64 }
65
66 const Guid machineId;
67 };
68
69 BackRef(const Guid &aMachineId,
70 const Guid &aSnapshotId = Guid::Empty)
71 : machineId(aMachineId),
72 fInCurState(aSnapshotId.isZero())
73 {
74 if (aSnapshotId.isValid() && !aSnapshotId.isZero())
75 llSnapshotIds.push_back(aSnapshotId);
76 }
77
78 Guid machineId;
79 bool fInCurState : 1;
80 GuidList llSnapshotIds;
81};
82
83typedef std::list<BackRef> BackRefList;
84
85struct Medium::Data
86{
87 Data()
88 : pVirtualBox(NULL),
89 state(MediumState_NotCreated),
90 variant(MediumVariant_Standard),
91 size(0),
92 readers(0),
93 preLockState(MediumState_NotCreated),
94 queryInfoSem(LOCKCLASS_MEDIUMQUERY),
95 queryInfoRunning(false),
96 type(MediumType_Normal),
97 devType(DeviceType_HardDisk),
98 logicalSize(0),
99 hddOpenMode(OpenReadWrite),
100 autoReset(false),
101 hostDrive(false),
102 implicit(false),
103 uOpenFlagsDef(VD_OPEN_FLAGS_IGNORE_FLUSH),
104 numCreateDiffTasks(0),
105 vdDiskIfaces(NULL),
106 vdImageIfaces(NULL)
107 { }
108
109 /** weak VirtualBox parent */
110 VirtualBox * const pVirtualBox;
111
112 // pParent and llChildren are protected by VirtualBox::getMediaTreeLockHandle()
113 ComObjPtr<Medium> pParent;
114 MediaList llChildren; // to add a child, just call push_back; to remove a child, call child->deparent() which does a lookup
115
116 GuidList llRegistryIDs; // media registries in which this medium is listed
117
118 const Guid id;
119 Utf8Str strDescription;
120 MediumState_T state;
121 MediumVariant_T variant;
122 Utf8Str strLocationFull;
123 uint64_t size;
124 Utf8Str strLastAccessError;
125
126 BackRefList backRefs;
127
128 size_t readers;
129 MediumState_T preLockState;
130
131 /** Special synchronization for operations which must wait for
132 * Medium::queryInfo in another thread to complete. Using a SemRW is
133 * not quite ideal, but at least it is subject to the lock validator,
134 * unlike the SemEventMulti which we had here for many years. Catching
135 * possible deadlocks is more important than a tiny bit of efficiency. */
136 RWLockHandle queryInfoSem;
137 bool queryInfoRunning : 1;
138
139 const Utf8Str strFormat;
140 ComObjPtr<MediumFormat> formatObj;
141
142 MediumType_T type;
143 DeviceType_T devType;
144 uint64_t logicalSize;
145
146 HDDOpenMode hddOpenMode;
147
148 bool autoReset : 1;
149
150 /** New UUID to be set on the next Medium::queryInfo call. */
151 const Guid uuidImage;
152 /** New parent UUID to be set on the next Medium::queryInfo call. */
153 const Guid uuidParentImage;
154
155 bool hostDrive : 1;
156
157 settings::StringsMap mapProperties;
158
159 bool implicit : 1;
160
161 /** Default flags passed to VDOpen(). */
162 unsigned uOpenFlagsDef;
163
164 uint32_t numCreateDiffTasks;
165
166 Utf8Str vdError; /*< Error remembered by the VD error callback. */
167
168 VDINTERFACEERROR vdIfError;
169
170 VDINTERFACECONFIG vdIfConfig;
171
172 VDINTERFACETCPNET vdIfTcpNet;
173
174 PVDINTERFACE vdDiskIfaces;
175 PVDINTERFACE vdImageIfaces;
176};
177
178typedef struct VDSOCKETINT
179{
180 /** Socket handle. */
181 RTSOCKET hSocket;
182} VDSOCKETINT, *PVDSOCKETINT;
183
184////////////////////////////////////////////////////////////////////////////////
185//
186// Globals
187//
188////////////////////////////////////////////////////////////////////////////////
189
190/**
191 * Medium::Task class for asynchronous operations.
192 *
193 * @note Instances of this class must be created using new() because the
194 * task thread function will delete them when the task is complete.
195 *
196 * @note The constructor of this class adds a caller on the managed Medium
197 * object which is automatically released upon destruction.
198 */
199class Medium::Task
200{
201public:
202 Task(Medium *aMedium, Progress *aProgress)
203 : mVDOperationIfaces(NULL),
204 mMedium(aMedium),
205 mMediumCaller(aMedium),
206 mThread(NIL_RTTHREAD),
207 mProgress(aProgress),
208 mVirtualBoxCaller(NULL)
209 {
210 AssertReturnVoidStmt(aMedium, mRC = E_FAIL);
211 mRC = mMediumCaller.rc();
212 if (FAILED(mRC))
213 return;
214
215 /* Get strong VirtualBox reference, see below. */
216 VirtualBox *pVirtualBox = aMedium->m->pVirtualBox;
217 mVirtualBox = pVirtualBox;
218 mVirtualBoxCaller.attach(pVirtualBox);
219 mRC = mVirtualBoxCaller.rc();
220 if (FAILED(mRC))
221 return;
222
223 /* Set up a per-operation progress interface, can be used freely (for
224 * binary operations you can use it either on the source or target). */
225 mVDIfProgress.pfnProgress = vdProgressCall;
226 int vrc = VDInterfaceAdd(&mVDIfProgress.Core,
227 "Medium::Task::vdInterfaceProgress",
228 VDINTERFACETYPE_PROGRESS,
229 mProgress,
230 sizeof(VDINTERFACEPROGRESS),
231 &mVDOperationIfaces);
232 AssertRC(vrc);
233 if (RT_FAILURE(vrc))
234 mRC = E_FAIL;
235 }
236
237 // Make all destructors virtual. Just in case.
238 virtual ~Task()
239 {}
240
241 HRESULT rc() const { return mRC; }
242 bool isOk() const { return SUCCEEDED(rc()); }
243
244 static int fntMediumTask(RTTHREAD aThread, void *pvUser);
245
246 bool isAsync() { return mThread != NIL_RTTHREAD; }
247
248 PVDINTERFACE mVDOperationIfaces;
249
250 const ComObjPtr<Medium> mMedium;
251 AutoCaller mMediumCaller;
252
253 friend HRESULT Medium::runNow(Medium::Task*);
254
255protected:
256 HRESULT mRC;
257 RTTHREAD mThread;
258
259private:
260 virtual HRESULT handler() = 0;
261
262 const ComObjPtr<Progress> mProgress;
263
264 static DECLCALLBACK(int) vdProgressCall(void *pvUser, unsigned uPercent);
265
266 VDINTERFACEPROGRESS mVDIfProgress;
267
268 /* Must have a strong VirtualBox reference during a task otherwise the
269 * reference count might drop to 0 while a task is still running. This
270 * would result in weird behavior, including deadlocks due to uninit and
271 * locking order issues. The deadlock often is not detectable because the
272 * uninit uses event semaphores which sabotages deadlock detection. */
273 ComObjPtr<VirtualBox> mVirtualBox;
274 AutoCaller mVirtualBoxCaller;
275};
276
277class Medium::CreateBaseTask : public Medium::Task
278{
279public:
280 CreateBaseTask(Medium *aMedium,
281 Progress *aProgress,
282 uint64_t aSize,
283 MediumVariant_T aVariant)
284 : Medium::Task(aMedium, aProgress),
285 mSize(aSize),
286 mVariant(aVariant)
287 {}
288
289 uint64_t mSize;
290 MediumVariant_T mVariant;
291
292private:
293 virtual HRESULT handler();
294};
295
296class Medium::CreateDiffTask : public Medium::Task
297{
298public:
299 CreateDiffTask(Medium *aMedium,
300 Progress *aProgress,
301 Medium *aTarget,
302 MediumVariant_T aVariant,
303 MediumLockList *aMediumLockList,
304 bool fKeepMediumLockList = false)
305 : Medium::Task(aMedium, aProgress),
306 mpMediumLockList(aMediumLockList),
307 mTarget(aTarget),
308 mVariant(aVariant),
309 mTargetCaller(aTarget),
310 mfKeepMediumLockList(fKeepMediumLockList)
311 {
312 AssertReturnVoidStmt(aTarget != NULL, mRC = E_FAIL);
313 mRC = mTargetCaller.rc();
314 if (FAILED(mRC))
315 return;
316 }
317
318 ~CreateDiffTask()
319 {
320 if (!mfKeepMediumLockList && mpMediumLockList)
321 delete mpMediumLockList;
322 }
323
324 MediumLockList *mpMediumLockList;
325
326 const ComObjPtr<Medium> mTarget;
327 MediumVariant_T mVariant;
328
329private:
330 virtual HRESULT handler();
331
332 AutoCaller mTargetCaller;
333 bool mfKeepMediumLockList;
334};
335
336class Medium::CloneTask : public Medium::Task
337{
338public:
339 CloneTask(Medium *aMedium,
340 Progress *aProgress,
341 Medium *aTarget,
342 MediumVariant_T aVariant,
343 Medium *aParent,
344 uint32_t idxSrcImageSame,
345 uint32_t idxDstImageSame,
346 MediumLockList *aSourceMediumLockList,
347 MediumLockList *aTargetMediumLockList,
348 bool fKeepSourceMediumLockList = false,
349 bool fKeepTargetMediumLockList = false)
350 : Medium::Task(aMedium, aProgress),
351 mTarget(aTarget),
352 mParent(aParent),
353 mpSourceMediumLockList(aSourceMediumLockList),
354 mpTargetMediumLockList(aTargetMediumLockList),
355 mVariant(aVariant),
356 midxSrcImageSame(idxSrcImageSame),
357 midxDstImageSame(idxDstImageSame),
358 mTargetCaller(aTarget),
359 mParentCaller(aParent),
360 mfKeepSourceMediumLockList(fKeepSourceMediumLockList),
361 mfKeepTargetMediumLockList(fKeepTargetMediumLockList)
362 {
363 AssertReturnVoidStmt(aTarget != NULL, mRC = E_FAIL);
364 mRC = mTargetCaller.rc();
365 if (FAILED(mRC))
366 return;
367 /* aParent may be NULL */
368 mRC = mParentCaller.rc();
369 if (FAILED(mRC))
370 return;
371 AssertReturnVoidStmt(aSourceMediumLockList != NULL, mRC = E_FAIL);
372 AssertReturnVoidStmt(aTargetMediumLockList != NULL, mRC = E_FAIL);
373 }
374
375 ~CloneTask()
376 {
377 if (!mfKeepSourceMediumLockList && mpSourceMediumLockList)
378 delete mpSourceMediumLockList;
379 if (!mfKeepTargetMediumLockList && mpTargetMediumLockList)
380 delete mpTargetMediumLockList;
381 }
382
383 const ComObjPtr<Medium> mTarget;
384 const ComObjPtr<Medium> mParent;
385 MediumLockList *mpSourceMediumLockList;
386 MediumLockList *mpTargetMediumLockList;
387 MediumVariant_T mVariant;
388 uint32_t midxSrcImageSame;
389 uint32_t midxDstImageSame;
390
391private:
392 virtual HRESULT handler();
393
394 AutoCaller mTargetCaller;
395 AutoCaller mParentCaller;
396 bool mfKeepSourceMediumLockList;
397 bool mfKeepTargetMediumLockList;
398};
399
400class Medium::CompactTask : public Medium::Task
401{
402public:
403 CompactTask(Medium *aMedium,
404 Progress *aProgress,
405 MediumLockList *aMediumLockList,
406 bool fKeepMediumLockList = false)
407 : Medium::Task(aMedium, aProgress),
408 mpMediumLockList(aMediumLockList),
409 mfKeepMediumLockList(fKeepMediumLockList)
410 {
411 AssertReturnVoidStmt(aMediumLockList != NULL, mRC = E_FAIL);
412 }
413
414 ~CompactTask()
415 {
416 if (!mfKeepMediumLockList && mpMediumLockList)
417 delete mpMediumLockList;
418 }
419
420 MediumLockList *mpMediumLockList;
421
422private:
423 virtual HRESULT handler();
424
425 bool mfKeepMediumLockList;
426};
427
428class Medium::ResizeTask : public Medium::Task
429{
430public:
431 ResizeTask(Medium *aMedium,
432 uint64_t aSize,
433 Progress *aProgress,
434 MediumLockList *aMediumLockList,
435 bool fKeepMediumLockList = false)
436 : Medium::Task(aMedium, aProgress),
437 mSize(aSize),
438 mpMediumLockList(aMediumLockList),
439 mfKeepMediumLockList(fKeepMediumLockList)
440 {
441 AssertReturnVoidStmt(aMediumLockList != NULL, mRC = E_FAIL);
442 }
443
444 ~ResizeTask()
445 {
446 if (!mfKeepMediumLockList && mpMediumLockList)
447 delete mpMediumLockList;
448 }
449
450 uint64_t mSize;
451 MediumLockList *mpMediumLockList;
452
453private:
454 virtual HRESULT handler();
455
456 bool mfKeepMediumLockList;
457};
458
459class Medium::ResetTask : public Medium::Task
460{
461public:
462 ResetTask(Medium *aMedium,
463 Progress *aProgress,
464 MediumLockList *aMediumLockList,
465 bool fKeepMediumLockList = false)
466 : Medium::Task(aMedium, aProgress),
467 mpMediumLockList(aMediumLockList),
468 mfKeepMediumLockList(fKeepMediumLockList)
469 {}
470
471 ~ResetTask()
472 {
473 if (!mfKeepMediumLockList && mpMediumLockList)
474 delete mpMediumLockList;
475 }
476
477 MediumLockList *mpMediumLockList;
478
479private:
480 virtual HRESULT handler();
481
482 bool mfKeepMediumLockList;
483};
484
485class Medium::DeleteTask : public Medium::Task
486{
487public:
488 DeleteTask(Medium *aMedium,
489 Progress *aProgress,
490 MediumLockList *aMediumLockList,
491 bool fKeepMediumLockList = false)
492 : Medium::Task(aMedium, aProgress),
493 mpMediumLockList(aMediumLockList),
494 mfKeepMediumLockList(fKeepMediumLockList)
495 {}
496
497 ~DeleteTask()
498 {
499 if (!mfKeepMediumLockList && mpMediumLockList)
500 delete mpMediumLockList;
501 }
502
503 MediumLockList *mpMediumLockList;
504
505private:
506 virtual HRESULT handler();
507
508 bool mfKeepMediumLockList;
509};
510
511class Medium::MergeTask : public Medium::Task
512{
513public:
514 MergeTask(Medium *aMedium,
515 Medium *aTarget,
516 bool fMergeForward,
517 Medium *aParentForTarget,
518 MediumLockList *aChildrenToReparent,
519 Progress *aProgress,
520 MediumLockList *aMediumLockList,
521 bool fKeepMediumLockList = false)
522 : Medium::Task(aMedium, aProgress),
523 mTarget(aTarget),
524 mfMergeForward(fMergeForward),
525 mParentForTarget(aParentForTarget),
526 mpChildrenToReparent(aChildrenToReparent),
527 mpMediumLockList(aMediumLockList),
528 mTargetCaller(aTarget),
529 mParentForTargetCaller(aParentForTarget),
530 mfKeepMediumLockList(fKeepMediumLockList)
531 {
532 AssertReturnVoidStmt(aMediumLockList != NULL, mRC = E_FAIL);
533 }
534
535 ~MergeTask()
536 {
537 if (!mfKeepMediumLockList && mpMediumLockList)
538 delete mpMediumLockList;
539 if (mpChildrenToReparent)
540 delete mpChildrenToReparent;
541 }
542
543 const ComObjPtr<Medium> mTarget;
544 bool mfMergeForward;
545 /* When mpChildrenToReparent is null then mParentForTarget is non-null and
546 * vice versa. In other words: they are used in different cases. */
547 const ComObjPtr<Medium> mParentForTarget;
548 MediumLockList *mpChildrenToReparent;
549 MediumLockList *mpMediumLockList;
550
551private:
552 virtual HRESULT handler();
553
554 AutoCaller mTargetCaller;
555 AutoCaller mParentForTargetCaller;
556 bool mfKeepMediumLockList;
557};
558
559class Medium::ExportTask : public Medium::Task
560{
561public:
562 ExportTask(Medium *aMedium,
563 Progress *aProgress,
564 const char *aFilename,
565 MediumFormat *aFormat,
566 MediumVariant_T aVariant,
567 VDINTERFACEIO *aVDImageIOIf,
568 void *aVDImageIOUser,
569 MediumLockList *aSourceMediumLockList,
570 bool fKeepSourceMediumLockList = false)
571 : Medium::Task(aMedium, aProgress),
572 mpSourceMediumLockList(aSourceMediumLockList),
573 mFilename(aFilename),
574 mFormat(aFormat),
575 mVariant(aVariant),
576 mfKeepSourceMediumLockList(fKeepSourceMediumLockList)
577 {
578 AssertReturnVoidStmt(aSourceMediumLockList != NULL, mRC = E_FAIL);
579
580 mVDImageIfaces = aMedium->m->vdImageIfaces;
581 if (aVDImageIOIf)
582 {
583 int vrc = VDInterfaceAdd(&aVDImageIOIf->Core, "Medium::vdInterfaceIO",
584 VDINTERFACETYPE_IO, aVDImageIOUser,
585 sizeof(VDINTERFACEIO), &mVDImageIfaces);
586 AssertRCReturnVoidStmt(vrc, mRC = E_FAIL);
587 }
588 }
589
590 ~ExportTask()
591 {
592 if (!mfKeepSourceMediumLockList && mpSourceMediumLockList)
593 delete mpSourceMediumLockList;
594 }
595
596 MediumLockList *mpSourceMediumLockList;
597 Utf8Str mFilename;
598 ComObjPtr<MediumFormat> mFormat;
599 MediumVariant_T mVariant;
600 PVDINTERFACE mVDImageIfaces;
601
602private:
603 virtual HRESULT handler();
604
605 bool mfKeepSourceMediumLockList;
606};
607
608class Medium::ImportTask : public Medium::Task
609{
610public:
611 ImportTask(Medium *aMedium,
612 Progress *aProgress,
613 const char *aFilename,
614 MediumFormat *aFormat,
615 MediumVariant_T aVariant,
616 VDINTERFACEIO *aVDImageIOIf,
617 void *aVDImageIOUser,
618 Medium *aParent,
619 MediumLockList *aTargetMediumLockList,
620 bool fKeepTargetMediumLockList = false)
621 : Medium::Task(aMedium, aProgress),
622 mFilename(aFilename),
623 mFormat(aFormat),
624 mVariant(aVariant),
625 mParent(aParent),
626 mpTargetMediumLockList(aTargetMediumLockList),
627 mParentCaller(aParent),
628 mfKeepTargetMediumLockList(fKeepTargetMediumLockList)
629 {
630 AssertReturnVoidStmt(aTargetMediumLockList != NULL, mRC = E_FAIL);
631 /* aParent may be NULL */
632 mRC = mParentCaller.rc();
633 if (FAILED(mRC))
634 return;
635
636 mVDImageIfaces = aMedium->m->vdImageIfaces;
637 if (aVDImageIOIf)
638 {
639 int vrc = VDInterfaceAdd(&aVDImageIOIf->Core, "Medium::vdInterfaceIO",
640 VDINTERFACETYPE_IO, aVDImageIOUser,
641 sizeof(VDINTERFACEIO), &mVDImageIfaces);
642 AssertRCReturnVoidStmt(vrc, mRC = E_FAIL);
643 }
644 }
645
646 ~ImportTask()
647 {
648 if (!mfKeepTargetMediumLockList && mpTargetMediumLockList)
649 delete mpTargetMediumLockList;
650 }
651
652 Utf8Str mFilename;
653 ComObjPtr<MediumFormat> mFormat;
654 MediumVariant_T mVariant;
655 const ComObjPtr<Medium> mParent;
656 MediumLockList *mpTargetMediumLockList;
657 PVDINTERFACE mVDImageIfaces;
658
659private:
660 virtual HRESULT handler();
661
662 AutoCaller mParentCaller;
663 bool mfKeepTargetMediumLockList;
664};
665
666/**
667 * Thread function for time-consuming medium tasks.
668 *
669 * @param pvUser Pointer to the Medium::Task instance.
670 */
671/* static */
672DECLCALLBACK(int) Medium::Task::fntMediumTask(RTTHREAD aThread, void *pvUser)
673{
674 LogFlowFuncEnter();
675 AssertReturn(pvUser, (int)E_INVALIDARG);
676 Medium::Task *pTask = static_cast<Medium::Task *>(pvUser);
677
678 pTask->mThread = aThread;
679
680 HRESULT rc = pTask->handler();
681
682 /* complete the progress if run asynchronously */
683 if (pTask->isAsync())
684 {
685 if (!pTask->mProgress.isNull())
686 pTask->mProgress->notifyComplete(rc);
687 }
688
689 /* pTask is no longer needed, delete it. */
690 delete pTask;
691
692 LogFlowFunc(("rc=%Rhrc\n", rc));
693 LogFlowFuncLeave();
694
695 return (int)rc;
696}
697
698/**
699 * PFNVDPROGRESS callback handler for Task operations.
700 *
701 * @param pvUser Pointer to the Progress instance.
702 * @param uPercent Completion percentage (0-100).
703 */
704/*static*/
705DECLCALLBACK(int) Medium::Task::vdProgressCall(void *pvUser, unsigned uPercent)
706{
707 Progress *that = static_cast<Progress *>(pvUser);
708
709 if (that != NULL)
710 {
711 /* update the progress object, capping it at 99% as the final percent
712 * is used for additional operations like setting the UUIDs and similar. */
713 HRESULT rc = that->SetCurrentOperationProgress(uPercent * 99 / 100);
714 if (FAILED(rc))
715 {
716 if (rc == E_FAIL)
717 return VERR_CANCELLED;
718 else
719 return VERR_INVALID_STATE;
720 }
721 }
722
723 return VINF_SUCCESS;
724}
725
726/**
727 * Implementation code for the "create base" task.
728 */
729HRESULT Medium::CreateBaseTask::handler()
730{
731 return mMedium->taskCreateBaseHandler(*this);
732}
733
734/**
735 * Implementation code for the "create diff" task.
736 */
737HRESULT Medium::CreateDiffTask::handler()
738{
739 return mMedium->taskCreateDiffHandler(*this);
740}
741
742/**
743 * Implementation code for the "clone" task.
744 */
745HRESULT Medium::CloneTask::handler()
746{
747 return mMedium->taskCloneHandler(*this);
748}
749
750/**
751 * Implementation code for the "compact" task.
752 */
753HRESULT Medium::CompactTask::handler()
754{
755 return mMedium->taskCompactHandler(*this);
756}
757
758/**
759 * Implementation code for the "resize" task.
760 */
761HRESULT Medium::ResizeTask::handler()
762{
763 return mMedium->taskResizeHandler(*this);
764}
765
766
767/**
768 * Implementation code for the "reset" task.
769 */
770HRESULT Medium::ResetTask::handler()
771{
772 return mMedium->taskResetHandler(*this);
773}
774
775/**
776 * Implementation code for the "delete" task.
777 */
778HRESULT Medium::DeleteTask::handler()
779{
780 return mMedium->taskDeleteHandler(*this);
781}
782
783/**
784 * Implementation code for the "merge" task.
785 */
786HRESULT Medium::MergeTask::handler()
787{
788 return mMedium->taskMergeHandler(*this);
789}
790
791/**
792 * Implementation code for the "export" task.
793 */
794HRESULT Medium::ExportTask::handler()
795{
796 return mMedium->taskExportHandler(*this);
797}
798
799/**
800 * Implementation code for the "import" task.
801 */
802HRESULT Medium::ImportTask::handler()
803{
804 return mMedium->taskImportHandler(*this);
805}
806
807////////////////////////////////////////////////////////////////////////////////
808//
809// Medium constructor / destructor
810//
811////////////////////////////////////////////////////////////////////////////////
812
813DEFINE_EMPTY_CTOR_DTOR(Medium)
814
815HRESULT Medium::FinalConstruct()
816{
817 m = new Data;
818
819 /* Initialize the callbacks of the VD error interface */
820 m->vdIfError.pfnError = vdErrorCall;
821 m->vdIfError.pfnMessage = NULL;
822
823 /* Initialize the callbacks of the VD config interface */
824 m->vdIfConfig.pfnAreKeysValid = vdConfigAreKeysValid;
825 m->vdIfConfig.pfnQuerySize = vdConfigQuerySize;
826 m->vdIfConfig.pfnQuery = vdConfigQuery;
827
828 /* Initialize the callbacks of the VD TCP interface (we always use the host
829 * IP stack for now) */
830 m->vdIfTcpNet.pfnSocketCreate = vdTcpSocketCreate;
831 m->vdIfTcpNet.pfnSocketDestroy = vdTcpSocketDestroy;
832 m->vdIfTcpNet.pfnClientConnect = vdTcpClientConnect;
833 m->vdIfTcpNet.pfnClientClose = vdTcpClientClose;
834 m->vdIfTcpNet.pfnIsClientConnected = vdTcpIsClientConnected;
835 m->vdIfTcpNet.pfnSelectOne = vdTcpSelectOne;
836 m->vdIfTcpNet.pfnRead = vdTcpRead;
837 m->vdIfTcpNet.pfnWrite = vdTcpWrite;
838 m->vdIfTcpNet.pfnSgWrite = vdTcpSgWrite;
839 m->vdIfTcpNet.pfnFlush = vdTcpFlush;
840 m->vdIfTcpNet.pfnSetSendCoalescing = vdTcpSetSendCoalescing;
841 m->vdIfTcpNet.pfnGetLocalAddress = vdTcpGetLocalAddress;
842 m->vdIfTcpNet.pfnGetPeerAddress = vdTcpGetPeerAddress;
843 m->vdIfTcpNet.pfnSelectOneEx = NULL;
844 m->vdIfTcpNet.pfnPoke = NULL;
845
846 /* Initialize the per-disk interface chain (could be done more globally,
847 * but it's not wasting much time or space so it's not worth it). */
848 int vrc;
849 vrc = VDInterfaceAdd(&m->vdIfError.Core,
850 "Medium::vdInterfaceError",
851 VDINTERFACETYPE_ERROR, this,
852 sizeof(VDINTERFACEERROR), &m->vdDiskIfaces);
853 AssertRCReturn(vrc, E_FAIL);
854
855 /* Initialize the per-image interface chain */
856 vrc = VDInterfaceAdd(&m->vdIfConfig.Core,
857 "Medium::vdInterfaceConfig",
858 VDINTERFACETYPE_CONFIG, this,
859 sizeof(VDINTERFACECONFIG), &m->vdImageIfaces);
860 AssertRCReturn(vrc, E_FAIL);
861
862 vrc = VDInterfaceAdd(&m->vdIfTcpNet.Core,
863 "Medium::vdInterfaceTcpNet",
864 VDINTERFACETYPE_TCPNET, this,
865 sizeof(VDINTERFACETCPNET), &m->vdImageIfaces);
866 AssertRCReturn(vrc, E_FAIL);
867
868 return BaseFinalConstruct();
869}
870
871void Medium::FinalRelease()
872{
873 uninit();
874
875 delete m;
876
877 BaseFinalRelease();
878}
879
880/**
881 * Initializes an empty hard disk object without creating or opening an associated
882 * storage unit.
883 *
884 * This gets called by VirtualBox::CreateHardDisk() in which case uuidMachineRegistry
885 * is empty since starting with VirtualBox 4.0, we no longer add opened media to a
886 * registry automatically (this is deferred until the medium is attached to a machine).
887 *
888 * This also gets called when VirtualBox creates diff images; in this case uuidMachineRegistry
889 * is set to the registry of the parent image to make sure they all end up in the same
890 * file.
891 *
892 * For hard disks that don't have the MediumFormatCapabilities_CreateFixed or
893 * MediumFormatCapabilities_CreateDynamic capability (and therefore cannot be created or deleted
894 * with the means of VirtualBox) the associated storage unit is assumed to be
895 * ready for use so the state of the hard disk object will be set to Created.
896 *
897 * @param aVirtualBox VirtualBox object.
898 * @param aFormat
899 * @param aLocation Storage unit location.
900 * @param uuidMachineRegistry The registry to which this medium should be added (global registry UUID or machine UUID or empty if none).
901 */
902HRESULT Medium::init(VirtualBox *aVirtualBox,
903 const Utf8Str &aFormat,
904 const Utf8Str &aLocation,
905 const Guid &uuidMachineRegistry)
906{
907 AssertReturn(aVirtualBox != NULL, E_FAIL);
908 AssertReturn(!aFormat.isEmpty(), E_FAIL);
909
910 /* Enclose the state transition NotReady->InInit->Ready */
911 AutoInitSpan autoInitSpan(this);
912 AssertReturn(autoInitSpan.isOk(), E_FAIL);
913
914 HRESULT rc = S_OK;
915
916 unconst(m->pVirtualBox) = aVirtualBox;
917
918 if (uuidMachineRegistry.isValid() && !uuidMachineRegistry.isZero())
919 m->llRegistryIDs.push_back(uuidMachineRegistry);
920
921 /* no storage yet */
922 m->state = MediumState_NotCreated;
923
924 /* cannot be a host drive */
925 m->hostDrive = false;
926
927 /* No storage unit is created yet, no need to call Medium::queryInfo */
928
929 rc = setFormat(aFormat);
930 if (FAILED(rc)) return rc;
931
932 rc = setLocation(aLocation);
933 if (FAILED(rc)) return rc;
934
935 if (!(m->formatObj->i_getCapabilities() & ( MediumFormatCapabilities_CreateFixed
936 | MediumFormatCapabilities_CreateDynamic))
937 )
938 {
939 /* Storage for hard disks of this format can neither be explicitly
940 * created by VirtualBox nor deleted, so we place the hard disk to
941 * Inaccessible state here and also add it to the registry. The
942 * state means that one has to use RefreshState() to update the
943 * medium format specific fields. */
944 m->state = MediumState_Inaccessible;
945 // create new UUID
946 unconst(m->id).create();
947
948 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
949 ComObjPtr<Medium> pMedium;
950
951 /*
952 * Check whether the UUID is taken already and create a new one
953 * if required.
954 * Try this only a limited amount of times in case the PRNG is broken
955 * in some way to prevent an endless loop.
956 */
957 for (unsigned i = 0; i < 5; i++)
958 {
959 bool fInUse;
960
961 fInUse = m->pVirtualBox->isMediaUuidInUse(m->id, DeviceType_HardDisk);
962 if (fInUse)
963 {
964 // create new UUID
965 unconst(m->id).create();
966 }
967 else
968 break;
969 }
970
971 rc = m->pVirtualBox->registerMedium(this, &pMedium, DeviceType_HardDisk);
972 Assert(this == pMedium || FAILED(rc));
973 }
974
975 /* Confirm a successful initialization when it's the case */
976 if (SUCCEEDED(rc))
977 autoInitSpan.setSucceeded();
978
979 return rc;
980}
981
982/**
983 * Initializes the medium object by opening the storage unit at the specified
984 * location. The enOpenMode parameter defines whether the medium will be opened
985 * read/write or read-only.
986 *
987 * This gets called by VirtualBox::OpenMedium() and also by
988 * Machine::AttachDevice() and createImplicitDiffs() when new diff
989 * images are created.
990 *
991 * There is no registry for this case since starting with VirtualBox 4.0, we
992 * no longer add opened media to a registry automatically (this is deferred
993 * until the medium is attached to a machine).
994 *
995 * For hard disks, the UUID, format and the parent of this medium will be
996 * determined when reading the medium storage unit. For DVD and floppy images,
997 * which have no UUIDs in their storage units, new UUIDs are created.
998 * If the detected or set parent is not known to VirtualBox, then this method
999 * will fail.
1000 *
1001 * @param aVirtualBox VirtualBox object.
1002 * @param aLocation Storage unit location.
1003 * @param enOpenMode Whether to open the medium read/write or read-only.
1004 * @param fForceNewUuid Whether a new UUID should be set to avoid duplicates.
1005 * @param aDeviceType Device type of medium.
1006 */
1007HRESULT Medium::init(VirtualBox *aVirtualBox,
1008 const Utf8Str &aLocation,
1009 HDDOpenMode enOpenMode,
1010 bool fForceNewUuid,
1011 DeviceType_T aDeviceType)
1012{
1013 AssertReturn(aVirtualBox, E_INVALIDARG);
1014 AssertReturn(!aLocation.isEmpty(), E_INVALIDARG);
1015
1016 HRESULT rc = S_OK;
1017
1018 {
1019 /* Enclose the state transition NotReady->InInit->Ready */
1020 AutoInitSpan autoInitSpan(this);
1021 AssertReturn(autoInitSpan.isOk(), E_FAIL);
1022
1023 unconst(m->pVirtualBox) = aVirtualBox;
1024
1025 /* there must be a storage unit */
1026 m->state = MediumState_Created;
1027
1028 /* remember device type for correct unregistering later */
1029 m->devType = aDeviceType;
1030
1031 /* cannot be a host drive */
1032 m->hostDrive = false;
1033
1034 /* remember the open mode (defaults to ReadWrite) */
1035 m->hddOpenMode = enOpenMode;
1036
1037 if (aDeviceType == DeviceType_DVD)
1038 m->type = MediumType_Readonly;
1039 else if (aDeviceType == DeviceType_Floppy)
1040 m->type = MediumType_Writethrough;
1041
1042 rc = setLocation(aLocation);
1043 if (FAILED(rc)) return rc;
1044
1045 /* get all the information about the medium from the storage unit */
1046 if (fForceNewUuid)
1047 unconst(m->uuidImage).create();
1048
1049 m->state = MediumState_Inaccessible;
1050 m->strLastAccessError = tr("Accessibility check was not yet performed");
1051
1052 /* Confirm a successful initialization before the call to queryInfo.
1053 * Otherwise we can end up with a AutoCaller deadlock because the
1054 * medium becomes visible but is not marked as initialized. Causes
1055 * locking trouble (e.g. trying to save media registries) which is
1056 * hard to solve. */
1057 autoInitSpan.setSucceeded();
1058 }
1059
1060 /* we're normal code from now on, no longer init */
1061 AutoCaller autoCaller(this);
1062 if (FAILED(autoCaller.rc()))
1063 return autoCaller.rc();
1064
1065 /* need to call queryInfo immediately to correctly place the medium in
1066 * the respective media tree and update other information such as uuid */
1067 rc = queryInfo(fForceNewUuid /* fSetImageId */, false /* fSetParentId */);
1068 if (SUCCEEDED(rc))
1069 {
1070 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1071
1072 /* if the storage unit is not accessible, it's not acceptable for the
1073 * newly opened media so convert this into an error */
1074 if (m->state == MediumState_Inaccessible)
1075 {
1076 Assert(!m->strLastAccessError.isEmpty());
1077 rc = setError(E_FAIL, "%s", m->strLastAccessError.c_str());
1078 alock.release();
1079 autoCaller.release();
1080 uninit();
1081 }
1082 else
1083 {
1084 AssertStmt(!m->id.isZero(),
1085 alock.release(); autoCaller.release(); uninit(); return E_FAIL);
1086
1087 /* storage format must be detected by Medium::queryInfo if the
1088 * medium is accessible */
1089 AssertStmt(!m->strFormat.isEmpty(),
1090 alock.release(); autoCaller.release(); uninit(); return E_FAIL);
1091 }
1092 }
1093 else
1094 {
1095 /* opening this image failed, mark the object as dead */
1096 autoCaller.release();
1097 uninit();
1098 }
1099
1100 return rc;
1101}
1102
1103/**
1104 * Initializes the medium object by loading its data from the given settings
1105 * node. In this mode, the medium will always be opened read/write.
1106 *
1107 * In this case, since we're loading from a registry, uuidMachineRegistry is
1108 * always set: it's either the global registry UUID or a machine UUID when
1109 * loading from a per-machine registry.
1110 *
1111 * @param aVirtualBox VirtualBox object.
1112 * @param aParent Parent medium disk or NULL for a root (base) medium.
1113 * @param aDeviceType Device type of the medium.
1114 * @param uuidMachineRegistry The registry to which this medium should be added (global registry UUID or machine UUID).
1115 * @param aNode Configuration settings.
1116 * @param strMachineFolder The machine folder with which to resolve relative paths; if empty, then we use the VirtualBox home directory
1117 *
1118 * @note Locks the medium tree for writing.
1119 */
1120HRESULT Medium::init(VirtualBox *aVirtualBox,
1121 Medium *aParent,
1122 DeviceType_T aDeviceType,
1123 const Guid &uuidMachineRegistry,
1124 const settings::Medium &data,
1125 const Utf8Str &strMachineFolder)
1126{
1127 using namespace settings;
1128
1129 AssertReturn(aVirtualBox, E_INVALIDARG);
1130
1131 /* Enclose the state transition NotReady->InInit->Ready */
1132 AutoInitSpan autoInitSpan(this);
1133 AssertReturn(autoInitSpan.isOk(), E_FAIL);
1134
1135 HRESULT rc = S_OK;
1136
1137 unconst(m->pVirtualBox) = aVirtualBox;
1138
1139 if (uuidMachineRegistry.isValid() && !uuidMachineRegistry.isZero())
1140 m->llRegistryIDs.push_back(uuidMachineRegistry);
1141
1142 /* register with VirtualBox/parent early, since uninit() will
1143 * unconditionally unregister on failure */
1144 if (aParent)
1145 {
1146 // differencing medium: add to parent
1147 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1148 m->pParent = aParent;
1149 aParent->m->llChildren.push_back(this);
1150 }
1151
1152 /* see below why we don't call Medium::queryInfo (and therefore treat
1153 * the medium as inaccessible for now */
1154 m->state = MediumState_Inaccessible;
1155 m->strLastAccessError = tr("Accessibility check was not yet performed");
1156
1157 /* required */
1158 unconst(m->id) = data.uuid;
1159
1160 /* assume not a host drive */
1161 m->hostDrive = false;
1162
1163 /* optional */
1164 m->strDescription = data.strDescription;
1165
1166 /* required */
1167 if (aDeviceType == DeviceType_HardDisk)
1168 {
1169 AssertReturn(!data.strFormat.isEmpty(), E_FAIL);
1170 rc = setFormat(data.strFormat);
1171 if (FAILED(rc)) return rc;
1172 }
1173 else
1174 {
1175 /// @todo handle host drive settings here as well?
1176 if (!data.strFormat.isEmpty())
1177 rc = setFormat(data.strFormat);
1178 else
1179 rc = setFormat("RAW");
1180 if (FAILED(rc)) return rc;
1181 }
1182
1183 /* optional, only for diffs, default is false; we can only auto-reset
1184 * diff media so they must have a parent */
1185 if (aParent != NULL)
1186 m->autoReset = data.fAutoReset;
1187 else
1188 m->autoReset = false;
1189
1190 /* properties (after setting the format as it populates the map). Note that
1191 * if some properties are not supported but present in the settings file,
1192 * they will still be read and accessible (for possible backward
1193 * compatibility; we can also clean them up from the XML upon next
1194 * XML format version change if we wish) */
1195 for (settings::StringsMap::const_iterator it = data.properties.begin();
1196 it != data.properties.end();
1197 ++it)
1198 {
1199 const Utf8Str &name = it->first;
1200 const Utf8Str &value = it->second;
1201 m->mapProperties[name] = value;
1202 }
1203
1204 /* try to decrypt an optional iSCSI initiator secret */
1205 settings::StringsMap::const_iterator itCph = data.properties.find("InitiatorSecretEncrypted");
1206 if ( itCph != data.properties.end()
1207 && !itCph->second.isEmpty())
1208 {
1209 Utf8Str strPlaintext;
1210 int vrc = m->pVirtualBox->decryptSetting(&strPlaintext, itCph->second);
1211 if (RT_SUCCESS(vrc))
1212 m->mapProperties["InitiatorSecret"] = strPlaintext;
1213 }
1214
1215 Utf8Str strFull;
1216 if (m->formatObj->i_getCapabilities() & MediumFormatCapabilities_File)
1217 {
1218 // compose full path of the medium, if it's not fully qualified...
1219 // slightly convoluted logic here. If the caller has given us a
1220 // machine folder, then a relative path will be relative to that:
1221 if ( !strMachineFolder.isEmpty()
1222 && !RTPathStartsWithRoot(data.strLocation.c_str())
1223 )
1224 {
1225 strFull = strMachineFolder;
1226 strFull += RTPATH_SLASH;
1227 strFull += data.strLocation;
1228 }
1229 else
1230 {
1231 // Otherwise use the old VirtualBox "make absolute path" logic:
1232 rc = m->pVirtualBox->calculateFullPath(data.strLocation, strFull);
1233 if (FAILED(rc)) return rc;
1234 }
1235 }
1236 else
1237 strFull = data.strLocation;
1238
1239 rc = setLocation(strFull);
1240 if (FAILED(rc)) return rc;
1241
1242 if (aDeviceType == DeviceType_HardDisk)
1243 {
1244 /* type is only for base hard disks */
1245 if (m->pParent.isNull())
1246 m->type = data.hdType;
1247 }
1248 else if (aDeviceType == DeviceType_DVD)
1249 m->type = MediumType_Readonly;
1250 else
1251 m->type = MediumType_Writethrough;
1252
1253 /* remember device type for correct unregistering later */
1254 m->devType = aDeviceType;
1255
1256 LogFlowThisFunc(("m->strLocationFull='%s', m->strFormat=%s, m->id={%RTuuid}\n",
1257 m->strLocationFull.c_str(), m->strFormat.c_str(), m->id.raw()));
1258
1259 /* Don't call Medium::queryInfo for registered media to prevent the calling
1260 * thread (i.e. the VirtualBox server startup thread) from an unexpected
1261 * freeze but mark it as initially inaccessible instead. The vital UUID,
1262 * location and format properties are read from the registry file above; to
1263 * get the actual state and the rest of the data, the user will have to call
1264 * COMGETTER(State). */
1265
1266 AutoWriteLock treeLock(aVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1267
1268 /* load all children */
1269 for (settings::MediaList::const_iterator it = data.llChildren.begin();
1270 it != data.llChildren.end();
1271 ++it)
1272 {
1273 const settings::Medium &med = *it;
1274
1275 ComObjPtr<Medium> pHD;
1276 pHD.createObject();
1277 rc = pHD->init(aVirtualBox,
1278 this, // parent
1279 aDeviceType,
1280 uuidMachineRegistry,
1281 med, // child data
1282 strMachineFolder);
1283 if (FAILED(rc)) break;
1284
1285 rc = m->pVirtualBox->registerMedium(pHD, &pHD, DeviceType_HardDisk);
1286 if (FAILED(rc)) break;
1287 }
1288
1289 /* Confirm a successful initialization when it's the case */
1290 if (SUCCEEDED(rc))
1291 autoInitSpan.setSucceeded();
1292
1293 return rc;
1294}
1295
1296/**
1297 * Initializes the medium object by providing the host drive information.
1298 * Not used for anything but the host floppy/host DVD case.
1299 *
1300 * There is no registry for this case.
1301 *
1302 * @param aVirtualBox VirtualBox object.
1303 * @param aDeviceType Device type of the medium.
1304 * @param aLocation Location of the host drive.
1305 * @param aDescription Comment for this host drive.
1306 *
1307 * @note Locks VirtualBox lock for writing.
1308 */
1309HRESULT Medium::init(VirtualBox *aVirtualBox,
1310 DeviceType_T aDeviceType,
1311 const Utf8Str &aLocation,
1312 const Utf8Str &aDescription /* = Utf8Str::Empty */)
1313{
1314 ComAssertRet(aDeviceType == DeviceType_DVD || aDeviceType == DeviceType_Floppy, E_INVALIDARG);
1315 ComAssertRet(!aLocation.isEmpty(), E_INVALIDARG);
1316
1317 /* Enclose the state transition NotReady->InInit->Ready */
1318 AutoInitSpan autoInitSpan(this);
1319 AssertReturn(autoInitSpan.isOk(), E_FAIL);
1320
1321 unconst(m->pVirtualBox) = aVirtualBox;
1322
1323 // We do not store host drives in VirtualBox.xml or anywhere else, so if we want
1324 // host drives to be identifiable by UUID and not give the drive a different UUID
1325 // every time VirtualBox starts, we need to fake a reproducible UUID here:
1326 RTUUID uuid;
1327 RTUuidClear(&uuid);
1328 if (aDeviceType == DeviceType_DVD)
1329 memcpy(&uuid.au8[0], "DVD", 3);
1330 else
1331 memcpy(&uuid.au8[0], "FD", 2);
1332 /* use device name, adjusted to the end of uuid, shortened if necessary */
1333 size_t lenLocation = aLocation.length();
1334 if (lenLocation > 12)
1335 memcpy(&uuid.au8[4], aLocation.c_str() + (lenLocation - 12), 12);
1336 else
1337 memcpy(&uuid.au8[4 + 12 - lenLocation], aLocation.c_str(), lenLocation);
1338 unconst(m->id) = uuid;
1339
1340 if (aDeviceType == DeviceType_DVD)
1341 m->type = MediumType_Readonly;
1342 else
1343 m->type = MediumType_Writethrough;
1344 m->devType = aDeviceType;
1345 m->state = MediumState_Created;
1346 m->hostDrive = true;
1347 HRESULT rc = setFormat("RAW");
1348 if (FAILED(rc)) return rc;
1349 rc = setLocation(aLocation);
1350 if (FAILED(rc)) return rc;
1351 m->strDescription = aDescription;
1352
1353 autoInitSpan.setSucceeded();
1354 return S_OK;
1355}
1356
1357/**
1358 * Uninitializes the instance.
1359 *
1360 * Called either from FinalRelease() or by the parent when it gets destroyed.
1361 *
1362 * @note All children of this medium get uninitialized by calling their
1363 * uninit() methods.
1364 */
1365void Medium::uninit()
1366{
1367 /* Enclose the state transition Ready->InUninit->NotReady */
1368 AutoUninitSpan autoUninitSpan(this);
1369 if (autoUninitSpan.uninitDone())
1370 return;
1371
1372 if (!m->formatObj.isNull())
1373 {
1374 /* remove the caller reference we added in setFormat() */
1375 m->formatObj->releaseCaller();
1376 m->formatObj.setNull();
1377 }
1378
1379 if (m->state == MediumState_Deleting)
1380 {
1381 /* This medium has been already deleted (directly or as part of a
1382 * merge). Reparenting has already been done. */
1383 Assert(m->pParent.isNull());
1384 }
1385 else
1386 {
1387 MediaList::iterator it;
1388 for (it = m->llChildren.begin();
1389 it != m->llChildren.end();
1390 ++it)
1391 {
1392 Medium *pChild = *it;
1393 pChild->m->pParent.setNull();
1394 pChild->uninit();
1395 }
1396 m->llChildren.clear(); // this unsets all the ComPtrs and probably calls delete
1397
1398 if (m->pParent)
1399 {
1400 // this is a differencing disk: then remove it from the parent's children list
1401 deparent();
1402 }
1403 }
1404
1405 unconst(m->pVirtualBox) = NULL;
1406}
1407
1408/**
1409 * Internal helper that removes "this" from the list of children of its
1410 * parent. Used in uninit() and other places when reparenting is necessary.
1411 *
1412 * The caller must hold the medium tree lock!
1413 */
1414void Medium::deparent()
1415{
1416 MediaList &llParent = m->pParent->m->llChildren;
1417 for (MediaList::iterator it = llParent.begin();
1418 it != llParent.end();
1419 ++it)
1420 {
1421 Medium *pParentsChild = *it;
1422 if (this == pParentsChild)
1423 {
1424 llParent.erase(it);
1425 break;
1426 }
1427 }
1428 m->pParent.setNull();
1429}
1430
1431/**
1432 * Internal helper that removes "this" from the list of children of its
1433 * parent. Used in uninit() and other places when reparenting is necessary.
1434 *
1435 * The caller must hold the medium tree lock!
1436 */
1437void Medium::setParent(const ComObjPtr<Medium> &pParent)
1438{
1439 m->pParent = pParent;
1440 if (pParent)
1441 pParent->m->llChildren.push_back(this);
1442}
1443
1444
1445////////////////////////////////////////////////////////////////////////////////
1446//
1447// IMedium public methods
1448//
1449////////////////////////////////////////////////////////////////////////////////
1450
1451STDMETHODIMP Medium::COMGETTER(Id)(BSTR *aId)
1452{
1453 CheckComArgOutPointerValid(aId);
1454
1455 AutoCaller autoCaller(this);
1456 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1457
1458 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1459
1460 m->id.toUtf16().cloneTo(aId);
1461
1462 return S_OK;
1463}
1464
1465STDMETHODIMP Medium::COMGETTER(Description)(BSTR *aDescription)
1466{
1467 CheckComArgOutPointerValid(aDescription);
1468
1469 AutoCaller autoCaller(this);
1470 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1471
1472 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1473
1474 m->strDescription.cloneTo(aDescription);
1475
1476 return S_OK;
1477}
1478
1479STDMETHODIMP Medium::COMSETTER(Description)(IN_BSTR aDescription)
1480{
1481 AutoCaller autoCaller(this);
1482 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1483
1484// AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1485
1486 /// @todo update m->description and save the global registry (and local
1487 /// registries of portable VMs referring to this medium), this will also
1488 /// require to add the mRegistered flag to data
1489
1490 NOREF(aDescription);
1491
1492 ReturnComNotImplemented();
1493}
1494
1495STDMETHODIMP Medium::COMGETTER(State)(MediumState_T *aState)
1496{
1497 CheckComArgOutPointerValid(aState);
1498
1499 AutoCaller autoCaller(this);
1500 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1501
1502 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1503 *aState = m->state;
1504
1505 return S_OK;
1506}
1507
1508STDMETHODIMP Medium::COMGETTER(Variant)(ComSafeArrayOut(MediumVariant_T, aVariant))
1509{
1510 CheckComArgOutSafeArrayPointerValid(aVariant);
1511
1512 AutoCaller autoCaller(this);
1513 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1514
1515 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1516
1517 SafeArray<MediumVariant_T> variants(sizeof(MediumVariant_T)*8);
1518
1519 for (ULONG i = 0; i < variants.size(); ++i)
1520 {
1521 ULONG temp = m->variant;
1522 temp &= 1<<i;
1523 variants [i] = (MediumVariant_T)temp;
1524 }
1525
1526 variants.detachTo(ComSafeArrayOutArg(aVariant));
1527
1528 return S_OK;
1529}
1530
1531STDMETHODIMP Medium::COMGETTER(Location)(BSTR *aLocation)
1532{
1533 CheckComArgOutPointerValid(aLocation);
1534
1535 AutoCaller autoCaller(this);
1536 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1537
1538 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1539
1540 m->strLocationFull.cloneTo(aLocation);
1541
1542 return S_OK;
1543}
1544
1545STDMETHODIMP Medium::COMSETTER(Location)(IN_BSTR aLocation)
1546{
1547 CheckComArgStrNotEmptyOrNull(aLocation);
1548
1549 AutoCaller autoCaller(this);
1550 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1551
1552 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1553
1554 /// @todo NEWMEDIA for file names, add the default extension if no extension
1555 /// is present (using the information from the VD backend which also implies
1556 /// that one more parameter should be passed to setLocation() requesting
1557 /// that functionality since it is only allowed when called from this method
1558
1559 /// @todo NEWMEDIA rename the file and set m->location on success, then save
1560 /// the global registry (and local registries of portable VMs referring to
1561 /// this medium), this will also require to add the mRegistered flag to data
1562
1563 ReturnComNotImplemented();
1564}
1565
1566STDMETHODIMP Medium::COMGETTER(Name)(BSTR *aName)
1567{
1568 CheckComArgOutPointerValid(aName);
1569
1570 AutoCaller autoCaller(this);
1571 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1572
1573 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1574
1575 getName().cloneTo(aName);
1576
1577 return S_OK;
1578}
1579
1580STDMETHODIMP Medium::COMGETTER(DeviceType)(DeviceType_T *aDeviceType)
1581{
1582 CheckComArgOutPointerValid(aDeviceType);
1583
1584 AutoCaller autoCaller(this);
1585 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1586
1587 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1588
1589 *aDeviceType = m->devType;
1590
1591 return S_OK;
1592}
1593
1594STDMETHODIMP Medium::COMGETTER(HostDrive)(BOOL *aHostDrive)
1595{
1596 CheckComArgOutPointerValid(aHostDrive);
1597
1598 AutoCaller autoCaller(this);
1599 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1600
1601 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1602
1603 *aHostDrive = m->hostDrive;
1604
1605 return S_OK;
1606}
1607
1608STDMETHODIMP Medium::COMGETTER(Size)(LONG64 *aSize)
1609{
1610 CheckComArgOutPointerValid(aSize);
1611
1612 AutoCaller autoCaller(this);
1613 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1614
1615 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1616
1617 *aSize = m->size;
1618
1619 return S_OK;
1620}
1621
1622STDMETHODIMP Medium::COMGETTER(Format)(BSTR *aFormat)
1623{
1624 CheckComArgOutPointerValid(aFormat);
1625
1626 AutoCaller autoCaller(this);
1627 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1628
1629 /* no need to lock, m->strFormat is const */
1630 m->strFormat.cloneTo(aFormat);
1631
1632 return S_OK;
1633}
1634
1635STDMETHODIMP Medium::COMGETTER(MediumFormat)(IMediumFormat **aMediumFormat)
1636{
1637 CheckComArgOutPointerValid(aMediumFormat);
1638
1639 AutoCaller autoCaller(this);
1640 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1641
1642 /* no need to lock, m->formatObj is const */
1643 m->formatObj.queryInterfaceTo(aMediumFormat);
1644
1645 return S_OK;
1646}
1647
1648STDMETHODIMP Medium::COMGETTER(Type)(MediumType_T *aType)
1649{
1650 CheckComArgOutPointerValid(aType);
1651
1652 AutoCaller autoCaller(this);
1653 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1654
1655 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1656
1657 *aType = m->type;
1658
1659 return S_OK;
1660}
1661
1662STDMETHODIMP Medium::COMSETTER(Type)(MediumType_T aType)
1663{
1664 AutoCaller autoCaller(this);
1665 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1666
1667 // we access mParent and members
1668 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1669 AutoWriteLock mlock(this COMMA_LOCKVAL_SRC_POS);
1670
1671 switch (m->state)
1672 {
1673 case MediumState_Created:
1674 case MediumState_Inaccessible:
1675 break;
1676 default:
1677 return setStateError();
1678 }
1679
1680 if (m->type == aType)
1681 {
1682 /* Nothing to do */
1683 return S_OK;
1684 }
1685
1686 DeviceType_T devType = getDeviceType();
1687 // DVD media can only be readonly.
1688 if (devType == DeviceType_DVD && aType != MediumType_Readonly)
1689 return setError(VBOX_E_INVALID_OBJECT_STATE,
1690 tr("Cannot change the type of DVD medium '%s'"),
1691 m->strLocationFull.c_str());
1692 // Floppy media can only be writethrough or readonly.
1693 if ( devType == DeviceType_Floppy
1694 && aType != MediumType_Writethrough
1695 && aType != MediumType_Readonly)
1696 return setError(VBOX_E_INVALID_OBJECT_STATE,
1697 tr("Cannot change the type of floppy medium '%s'"),
1698 m->strLocationFull.c_str());
1699
1700 /* cannot change the type of a differencing medium */
1701 if (m->pParent)
1702 return setError(VBOX_E_INVALID_OBJECT_STATE,
1703 tr("Cannot change the type of medium '%s' because it is a differencing medium"),
1704 m->strLocationFull.c_str());
1705
1706 /* Cannot change the type of a medium being in use by more than one VM.
1707 * If the change is to Immutable or MultiAttach then it must not be
1708 * directly attached to any VM, otherwise the assumptions about indirect
1709 * attachment elsewhere are violated and the VM becomes inaccessible.
1710 * Attaching an immutable medium triggers the diff creation, and this is
1711 * vital for the correct operation. */
1712 if ( m->backRefs.size() > 1
1713 || ( ( aType == MediumType_Immutable
1714 || aType == MediumType_MultiAttach)
1715 && m->backRefs.size() > 0))
1716 return setError(VBOX_E_INVALID_OBJECT_STATE,
1717 tr("Cannot change the type of medium '%s' because it is attached to %d virtual machines"),
1718 m->strLocationFull.c_str(), m->backRefs.size());
1719
1720 switch (aType)
1721 {
1722 case MediumType_Normal:
1723 case MediumType_Immutable:
1724 case MediumType_MultiAttach:
1725 {
1726 /* normal can be easily converted to immutable and vice versa even
1727 * if they have children as long as they are not attached to any
1728 * machine themselves */
1729 break;
1730 }
1731 case MediumType_Writethrough:
1732 case MediumType_Shareable:
1733 case MediumType_Readonly:
1734 {
1735 /* cannot change to writethrough, shareable or readonly
1736 * if there are children */
1737 if (getChildren().size() != 0)
1738 return setError(VBOX_E_OBJECT_IN_USE,
1739 tr("Cannot change type for medium '%s' since it has %d child media"),
1740 m->strLocationFull.c_str(), getChildren().size());
1741 if (aType == MediumType_Shareable)
1742 {
1743 MediumVariant_T variant = getVariant();
1744 if (!(variant & MediumVariant_Fixed))
1745 return setError(VBOX_E_INVALID_OBJECT_STATE,
1746 tr("Cannot change type for medium '%s' to 'Shareable' since it is a dynamic medium storage unit"),
1747 m->strLocationFull.c_str());
1748 }
1749 else if (aType == MediumType_Readonly && devType == DeviceType_HardDisk)
1750 {
1751 // Readonly hard disks are not allowed, this medium type is reserved for
1752 // DVDs and floppy images at the moment. Later we might allow readonly hard
1753 // disks, but that's extremely unusual and many guest OSes will have trouble.
1754 return setError(VBOX_E_INVALID_OBJECT_STATE,
1755 tr("Cannot change type for medium '%s' to 'Readonly' since it is a hard disk"),
1756 m->strLocationFull.c_str());
1757 }
1758 break;
1759 }
1760 default:
1761 AssertFailedReturn(E_FAIL);
1762 }
1763
1764 if (aType == MediumType_MultiAttach)
1765 {
1766 // This type is new with VirtualBox 4.0 and therefore requires settings
1767 // version 1.11 in the settings backend. Unfortunately it is not enough to do
1768 // the usual routine in MachineConfigFile::bumpSettingsVersionIfNeeded() for
1769 // two reasons: The medium type is a property of the media registry tree, which
1770 // can reside in the global config file (for pre-4.0 media); we would therefore
1771 // possibly need to bump the global config version. We don't want to do that though
1772 // because that might make downgrading to pre-4.0 impossible.
1773 // As a result, we can only use these two new types if the medium is NOT in the
1774 // global registry:
1775 const Guid &uuidGlobalRegistry = m->pVirtualBox->getGlobalRegistryId();
1776 if (isInRegistry(uuidGlobalRegistry))
1777 return setError(VBOX_E_INVALID_OBJECT_STATE,
1778 tr("Cannot change type for medium '%s': the media type 'MultiAttach' can only be used "
1779 "on media registered with a machine that was created with VirtualBox 4.0 or later"),
1780 m->strLocationFull.c_str());
1781 }
1782
1783 m->type = aType;
1784
1785 // save the settings
1786 mlock.release();
1787 treeLock.release();
1788 markRegistriesModified();
1789 m->pVirtualBox->saveModifiedRegistries();
1790
1791 return S_OK;
1792}
1793
1794STDMETHODIMP Medium::COMGETTER(AllowedTypes)(ComSafeArrayOut(MediumType_T, aAllowedTypes))
1795{
1796 CheckComArgOutSafeArrayPointerValid(aAllowedTypes);
1797 NOREF(aAllowedTypes);
1798#ifndef RT_OS_WINDOWS
1799 NOREF(aAllowedTypesSize);
1800#endif
1801
1802 AutoCaller autoCaller(this);
1803 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1804
1805 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1806
1807 ReturnComNotImplemented();
1808}
1809
1810STDMETHODIMP Medium::COMGETTER(Parent)(IMedium **aParent)
1811{
1812 CheckComArgOutPointerValid(aParent);
1813
1814 AutoCaller autoCaller(this);
1815 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1816
1817 /* we access mParent */
1818 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1819
1820 m->pParent.queryInterfaceTo(aParent);
1821
1822 return S_OK;
1823}
1824
1825STDMETHODIMP Medium::COMGETTER(Children)(ComSafeArrayOut(IMedium *, aChildren))
1826{
1827 CheckComArgOutSafeArrayPointerValid(aChildren);
1828
1829 AutoCaller autoCaller(this);
1830 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1831
1832 /* we access children */
1833 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
1834
1835 SafeIfaceArray<IMedium> children(this->getChildren());
1836 children.detachTo(ComSafeArrayOutArg(aChildren));
1837
1838 return S_OK;
1839}
1840
1841STDMETHODIMP Medium::COMGETTER(Base)(IMedium **aBase)
1842{
1843 CheckComArgOutPointerValid(aBase);
1844
1845 /* base() will do callers/locking */
1846
1847 getBase().queryInterfaceTo(aBase);
1848
1849 return S_OK;
1850}
1851
1852STDMETHODIMP Medium::COMGETTER(ReadOnly)(BOOL *aReadOnly)
1853{
1854 CheckComArgOutPointerValid(aReadOnly);
1855
1856 AutoCaller autoCaller(this);
1857 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1858
1859 /* isReadOnly() will do locking */
1860
1861 *aReadOnly = isReadOnly();
1862
1863 return S_OK;
1864}
1865
1866STDMETHODIMP Medium::COMGETTER(LogicalSize)(LONG64 *aLogicalSize)
1867{
1868 CheckComArgOutPointerValid(aLogicalSize);
1869
1870 AutoCaller autoCaller(this);
1871 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1872
1873 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1874
1875 *aLogicalSize = m->logicalSize;
1876
1877 return S_OK;
1878}
1879
1880STDMETHODIMP Medium::COMGETTER(AutoReset)(BOOL *aAutoReset)
1881{
1882 CheckComArgOutPointerValid(aAutoReset);
1883
1884 AutoCaller autoCaller(this);
1885 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1886
1887 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1888
1889 if (m->pParent.isNull())
1890 *aAutoReset = FALSE;
1891 else
1892 *aAutoReset = m->autoReset;
1893
1894 return S_OK;
1895}
1896
1897STDMETHODIMP Medium::COMSETTER(AutoReset)(BOOL aAutoReset)
1898{
1899 AutoCaller autoCaller(this);
1900 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1901
1902 AutoWriteLock mlock(this COMMA_LOCKVAL_SRC_POS);
1903
1904 if (m->pParent.isNull())
1905 return setError(VBOX_E_NOT_SUPPORTED,
1906 tr("Medium '%s' is not differencing"),
1907 m->strLocationFull.c_str());
1908
1909 if (m->autoReset != !!aAutoReset)
1910 {
1911 m->autoReset = !!aAutoReset;
1912
1913 // save the settings
1914 mlock.release();
1915 markRegistriesModified();
1916 m->pVirtualBox->saveModifiedRegistries();
1917 }
1918
1919 return S_OK;
1920}
1921
1922STDMETHODIMP Medium::COMGETTER(LastAccessError)(BSTR *aLastAccessError)
1923{
1924 CheckComArgOutPointerValid(aLastAccessError);
1925
1926 AutoCaller autoCaller(this);
1927 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1928
1929 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1930
1931 m->strLastAccessError.cloneTo(aLastAccessError);
1932
1933 return S_OK;
1934}
1935
1936STDMETHODIMP Medium::COMGETTER(MachineIds)(ComSafeArrayOut(BSTR,aMachineIds))
1937{
1938 CheckComArgOutSafeArrayPointerValid(aMachineIds);
1939
1940 AutoCaller autoCaller(this);
1941 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1942
1943 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1944
1945 com::SafeArray<BSTR> machineIds;
1946
1947 if (m->backRefs.size() != 0)
1948 {
1949 machineIds.reset(m->backRefs.size());
1950
1951 size_t i = 0;
1952 for (BackRefList::const_iterator it = m->backRefs.begin();
1953 it != m->backRefs.end(); ++it, ++i)
1954 {
1955 it->machineId.toUtf16().detachTo(&machineIds[i]);
1956 }
1957 }
1958
1959 machineIds.detachTo(ComSafeArrayOutArg(aMachineIds));
1960
1961 return S_OK;
1962}
1963
1964STDMETHODIMP Medium::SetIds(BOOL aSetImageId,
1965 IN_BSTR aImageId,
1966 BOOL aSetParentId,
1967 IN_BSTR aParentId)
1968{
1969 AutoCaller autoCaller(this);
1970 if (FAILED(autoCaller.rc())) return autoCaller.rc();
1971
1972 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1973
1974 switch (m->state)
1975 {
1976 case MediumState_Created:
1977 break;
1978 default:
1979 return setStateError();
1980 }
1981
1982 Guid imageId, parentId;
1983 if (aSetImageId)
1984 {
1985 if (Bstr(aImageId).isEmpty())
1986 imageId.create();
1987 else
1988 {
1989 imageId = Guid(aImageId);
1990 if (!imageId.isValid())
1991 return setError(E_INVALIDARG, tr("Argument %s is invalid"), "aImageId");
1992 }
1993 }
1994 if (aSetParentId)
1995 {
1996 if (Bstr(aParentId).isEmpty())
1997 parentId.create();
1998 else
1999 parentId = Guid(aParentId);
2000 }
2001
2002 unconst(m->uuidImage) = imageId;
2003 unconst(m->uuidParentImage) = parentId;
2004
2005 // must not hold any locks before calling Medium::queryInfo
2006 alock.release();
2007
2008 HRESULT rc = queryInfo(!!aSetImageId /* fSetImageId */,
2009 !!aSetParentId /* fSetParentId */);
2010
2011 return rc;
2012}
2013
2014STDMETHODIMP Medium::RefreshState(MediumState_T *aState)
2015{
2016 CheckComArgOutPointerValid(aState);
2017
2018 AutoCaller autoCaller(this);
2019 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2020
2021 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2022
2023 HRESULT rc = S_OK;
2024
2025 switch (m->state)
2026 {
2027 case MediumState_Created:
2028 case MediumState_Inaccessible:
2029 case MediumState_LockedRead:
2030 {
2031 // must not hold any locks before calling Medium::queryInfo
2032 alock.release();
2033
2034 rc = queryInfo(false /* fSetImageId */, false /* fSetParentId */);
2035
2036 alock.acquire();
2037 break;
2038 }
2039 default:
2040 break;
2041 }
2042
2043 *aState = m->state;
2044
2045 return rc;
2046}
2047
2048STDMETHODIMP Medium::GetSnapshotIds(IN_BSTR aMachineId,
2049 ComSafeArrayOut(BSTR, aSnapshotIds))
2050{
2051 CheckComArgExpr(aMachineId, Guid(aMachineId).isValid());
2052 CheckComArgOutSafeArrayPointerValid(aSnapshotIds);
2053
2054 AutoCaller autoCaller(this);
2055 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2056
2057 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2058
2059 com::SafeArray<BSTR> snapshotIds;
2060
2061 Guid id(aMachineId);
2062 for (BackRefList::const_iterator it = m->backRefs.begin();
2063 it != m->backRefs.end(); ++it)
2064 {
2065 if (it->machineId == id)
2066 {
2067 size_t size = it->llSnapshotIds.size();
2068
2069 /* if the medium is attached to the machine in the current state, we
2070 * return its ID as the first element of the array */
2071 if (it->fInCurState)
2072 ++size;
2073
2074 if (size > 0)
2075 {
2076 snapshotIds.reset(size);
2077
2078 size_t j = 0;
2079 if (it->fInCurState)
2080 it->machineId.toUtf16().detachTo(&snapshotIds[j++]);
2081
2082 for (GuidList::const_iterator jt = it->llSnapshotIds.begin();
2083 jt != it->llSnapshotIds.end();
2084 ++jt, ++j)
2085 {
2086 (*jt).toUtf16().detachTo(&snapshotIds[j]);
2087 }
2088 }
2089
2090 break;
2091 }
2092 }
2093
2094 snapshotIds.detachTo(ComSafeArrayOutArg(aSnapshotIds));
2095
2096 return S_OK;
2097}
2098
2099STDMETHODIMP Medium::LockRead(IToken **aToken)
2100{
2101 CheckComArgNotNull(aToken);
2102
2103 AutoCaller autoCaller(this);
2104 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2105
2106 /* Must not hold the object lock, as we need control over it below. */
2107 Assert(!isWriteLockOnCurrentThread());
2108 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2109
2110 /* Wait for a concurrently running Medium::queryInfo to complete. */
2111 if (m->queryInfoRunning)
2112 {
2113 /* Must not hold the media tree lock, as Medium::queryInfo needs this
2114 * lock and thus we would run into a deadlock here. */
2115 Assert(!m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
2116 while (m->queryInfoRunning)
2117 {
2118 alock.release();
2119 {
2120 AutoReadLock qlock(m->queryInfoSem COMMA_LOCKVAL_SRC_POS);
2121 }
2122 alock.acquire();
2123 }
2124 }
2125
2126 HRESULT rc = S_OK;
2127
2128 switch (m->state)
2129 {
2130 case MediumState_Created:
2131 case MediumState_Inaccessible:
2132 case MediumState_LockedRead:
2133 {
2134 ++m->readers;
2135
2136 ComAssertMsgBreak(m->readers != 0, ("Counter overflow"), rc = E_FAIL);
2137
2138 /* Remember pre-lock state */
2139 if (m->state != MediumState_LockedRead)
2140 m->preLockState = m->state;
2141
2142 LogFlowThisFunc(("Okay - prev state=%d readers=%d\n", m->state, m->readers));
2143 m->state = MediumState_LockedRead;
2144
2145 ComObjPtr<MediumLockToken> pToken;
2146 rc = pToken.createObject();
2147 if (SUCCEEDED(rc))
2148 rc = pToken->init(this, false /* fWrite */);
2149 if (FAILED(rc))
2150 {
2151 --m->readers;
2152 if (m->readers == 0)
2153 m->state = m->preLockState;
2154 return rc;
2155 }
2156
2157 pToken.queryInterfaceTo(aToken);
2158 break;
2159 }
2160 default:
2161 {
2162 LogFlowThisFunc(("Failing - state=%d\n", m->state));
2163 rc = setStateError();
2164 break;
2165 }
2166 }
2167
2168 return rc;
2169}
2170
2171/**
2172 * @note @a aState may be NULL if the state value is not needed (only for
2173 * in-process calls).
2174 */
2175HRESULT Medium::unlockRead(MediumState_T *aState)
2176{
2177 AutoCaller autoCaller(this);
2178 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2179
2180 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2181
2182 HRESULT rc = S_OK;
2183
2184 switch (m->state)
2185 {
2186 case MediumState_LockedRead:
2187 {
2188 ComAssertMsgBreak(m->readers != 0, ("Counter underflow"), rc = E_FAIL);
2189 --m->readers;
2190
2191 /* Reset the state after the last reader */
2192 if (m->readers == 0)
2193 {
2194 m->state = m->preLockState;
2195 /* There are cases where we inject the deleting state into
2196 * a medium locked for reading. Make sure #unmarkForDeletion()
2197 * gets the right state afterwards. */
2198 if (m->preLockState == MediumState_Deleting)
2199 m->preLockState = MediumState_Created;
2200 }
2201
2202 LogFlowThisFunc(("new state=%d\n", m->state));
2203 break;
2204 }
2205 default:
2206 {
2207 LogFlowThisFunc(("Failing - state=%d\n", m->state));
2208 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
2209 tr("Medium '%s' is not locked for reading"),
2210 m->strLocationFull.c_str());
2211 break;
2212 }
2213 }
2214
2215 /* return the current state after */
2216 if (aState)
2217 *aState = m->state;
2218
2219 return rc;
2220}
2221
2222STDMETHODIMP Medium::LockWrite(IToken **aToken)
2223{
2224 CheckComArgNotNull(aToken);
2225
2226 AutoCaller autoCaller(this);
2227 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2228
2229 /* Must not hold the object lock, as we need control over it below. */
2230 Assert(!isWriteLockOnCurrentThread());
2231 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2232
2233 /* Wait for a concurrently running Medium::queryInfo to complete. */
2234 if (m->queryInfoRunning)
2235 {
2236 /* Must not hold the media tree lock, as Medium::queryInfo needs this
2237 * lock and thus we would run into a deadlock here. */
2238 Assert(!m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
2239 while (m->queryInfoRunning)
2240 {
2241 alock.release();
2242 {
2243 AutoReadLock qlock(m->queryInfoSem COMMA_LOCKVAL_SRC_POS);
2244 }
2245 alock.acquire();
2246 }
2247 }
2248
2249 HRESULT rc = S_OK;
2250
2251 switch (m->state)
2252 {
2253 case MediumState_Created:
2254 case MediumState_Inaccessible:
2255 {
2256 m->preLockState = m->state;
2257
2258 LogFlowThisFunc(("Okay - prev state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
2259 m->state = MediumState_LockedWrite;
2260
2261 ComObjPtr<MediumLockToken> pToken;
2262 rc = pToken.createObject();
2263 if (SUCCEEDED(rc))
2264 rc = pToken->init(this, true /* fWrite */);
2265 if (FAILED(rc))
2266 {
2267 m->state = m->preLockState;
2268 return rc;
2269 }
2270
2271 pToken.queryInterfaceTo(aToken);
2272 break;
2273 }
2274 default:
2275 {
2276 LogFlowThisFunc(("Failing - state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
2277 rc = setStateError();
2278 break;
2279 }
2280 }
2281
2282 return rc;
2283}
2284
2285/**
2286 * @note @a aState may be NULL if the state value is not needed (only for
2287 * in-process calls).
2288 */
2289HRESULT Medium::unlockWrite(MediumState_T *aState)
2290{
2291 AutoCaller autoCaller(this);
2292 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2293
2294 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2295
2296 HRESULT rc = S_OK;
2297
2298 switch (m->state)
2299 {
2300 case MediumState_LockedWrite:
2301 {
2302 m->state = m->preLockState;
2303 /* There are cases where we inject the deleting state into
2304 * a medium locked for writing. Make sure #unmarkForDeletion()
2305 * gets the right state afterwards. */
2306 if (m->preLockState == MediumState_Deleting)
2307 m->preLockState = MediumState_Created;
2308 LogFlowThisFunc(("new state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
2309 break;
2310 }
2311 default:
2312 {
2313 LogFlowThisFunc(("Failing - state=%d locationFull=%s\n", m->state, getLocationFull().c_str()));
2314 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
2315 tr("Medium '%s' is not locked for writing"),
2316 m->strLocationFull.c_str());
2317 break;
2318 }
2319 }
2320
2321 /* return the current state after */
2322 if (aState)
2323 *aState = m->state;
2324
2325 return rc;
2326}
2327
2328STDMETHODIMP Medium::Close()
2329{
2330 AutoCaller autoCaller(this);
2331 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2332
2333 // make a copy of VirtualBox pointer which gets nulled by uninit()
2334 ComObjPtr<VirtualBox> pVirtualBox(m->pVirtualBox);
2335
2336 MultiResult mrc = close(autoCaller);
2337
2338 pVirtualBox->saveModifiedRegistries();
2339
2340 return mrc;
2341}
2342
2343STDMETHODIMP Medium::GetProperty(IN_BSTR aName, BSTR *aValue)
2344{
2345 CheckComArgStrNotEmptyOrNull(aName);
2346 CheckComArgOutPointerValid(aValue);
2347
2348 AutoCaller autoCaller(this);
2349 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2350
2351 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2352
2353 settings::StringsMap::const_iterator it = m->mapProperties.find(Utf8Str(aName));
2354 if (it == m->mapProperties.end())
2355 return setError(VBOX_E_OBJECT_NOT_FOUND,
2356 tr("Property '%ls' does not exist"), aName);
2357
2358 it->second.cloneTo(aValue);
2359
2360 return S_OK;
2361}
2362
2363STDMETHODIMP Medium::SetProperty(IN_BSTR aName, IN_BSTR aValue)
2364{
2365 CheckComArgStrNotEmptyOrNull(aName);
2366
2367 AutoCaller autoCaller(this);
2368 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2369
2370 AutoWriteLock mlock(this COMMA_LOCKVAL_SRC_POS);
2371
2372 switch (m->state)
2373 {
2374 case MediumState_Created:
2375 case MediumState_Inaccessible:
2376 break;
2377 default:
2378 return setStateError();
2379 }
2380
2381 Utf8Str strName(aName);
2382 Utf8Str strValue(aValue);
2383 settings::StringsMap::iterator it = m->mapProperties.find(strName);
2384 if (!strName.startsWith("Special/"))
2385 {
2386 if (it == m->mapProperties.end())
2387 return setError(VBOX_E_OBJECT_NOT_FOUND,
2388 tr("Property '%s' does not exist"),
2389 strName.c_str());
2390 it->second = strValue;
2391 }
2392 else
2393 {
2394 if (it == m->mapProperties.end())
2395 {
2396 if (!strValue.isEmpty())
2397 m->mapProperties[strName] = strValue;
2398 }
2399 else
2400 {
2401 if (!strValue.isEmpty())
2402 it->second = aValue;
2403 else
2404 m->mapProperties.erase(it);
2405 }
2406 }
2407
2408 // save the settings
2409 mlock.release();
2410 markRegistriesModified();
2411 m->pVirtualBox->saveModifiedRegistries();
2412
2413 return S_OK;
2414}
2415
2416STDMETHODIMP Medium::GetProperties(IN_BSTR aNames,
2417 ComSafeArrayOut(BSTR, aReturnNames),
2418 ComSafeArrayOut(BSTR, aReturnValues))
2419{
2420 CheckComArgOutSafeArrayPointerValid(aReturnNames);
2421 CheckComArgOutSafeArrayPointerValid(aReturnValues);
2422
2423 AutoCaller autoCaller(this);
2424 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2425
2426 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
2427
2428 /// @todo make use of aNames according to the documentation
2429 NOREF(aNames);
2430
2431 com::SafeArray<BSTR> names(m->mapProperties.size());
2432 com::SafeArray<BSTR> values(m->mapProperties.size());
2433 size_t i = 0;
2434
2435 for (settings::StringsMap::const_iterator it = m->mapProperties.begin();
2436 it != m->mapProperties.end();
2437 ++it)
2438 {
2439 it->first.cloneTo(&names[i]);
2440 it->second.cloneTo(&values[i]);
2441 ++i;
2442 }
2443
2444 names.detachTo(ComSafeArrayOutArg(aReturnNames));
2445 values.detachTo(ComSafeArrayOutArg(aReturnValues));
2446
2447 return S_OK;
2448}
2449
2450STDMETHODIMP Medium::SetProperties(ComSafeArrayIn(IN_BSTR, aNames),
2451 ComSafeArrayIn(IN_BSTR, aValues))
2452{
2453 CheckComArgSafeArrayNotNull(aNames);
2454 CheckComArgSafeArrayNotNull(aValues);
2455
2456 AutoCaller autoCaller(this);
2457 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2458
2459 AutoWriteLock mlock(this COMMA_LOCKVAL_SRC_POS);
2460
2461 com::SafeArray<IN_BSTR> names(ComSafeArrayInArg(aNames));
2462 com::SafeArray<IN_BSTR> values(ComSafeArrayInArg(aValues));
2463
2464 /* first pass: validate names */
2465 for (size_t i = 0;
2466 i < names.size();
2467 ++i)
2468 {
2469 Utf8Str strName(names[i]);
2470 if ( !strName.startsWith("Special/")
2471 && m->mapProperties.find(strName) == m->mapProperties.end())
2472 return setError(VBOX_E_OBJECT_NOT_FOUND,
2473 tr("Property '%s' does not exist"), strName.c_str());
2474 }
2475
2476 /* second pass: assign */
2477 for (size_t i = 0;
2478 i < names.size();
2479 ++i)
2480 {
2481 Utf8Str strName(names[i]);
2482 Utf8Str strValue(values[i]);
2483 settings::StringsMap::iterator it = m->mapProperties.find(strName);
2484 if (!strName.startsWith("Special/"))
2485 {
2486 AssertReturn(it != m->mapProperties.end(), E_FAIL);
2487 it->second = strValue;
2488 }
2489 else
2490 {
2491 if (it == m->mapProperties.end())
2492 {
2493 if (!strValue.isEmpty())
2494 m->mapProperties[strName] = strValue;
2495 }
2496 else
2497 {
2498 if (!strValue.isEmpty())
2499 it->second = strValue;
2500 else
2501 m->mapProperties.erase(it);
2502 }
2503 }
2504 }
2505
2506 // save the settings
2507 mlock.release();
2508 markRegistriesModified();
2509 m->pVirtualBox->saveModifiedRegistries();
2510
2511 return S_OK;
2512}
2513
2514STDMETHODIMP Medium::CreateBaseStorage(LONG64 aLogicalSize,
2515 ComSafeArrayIn(MediumVariant_T, aVariant),
2516 IProgress **aProgress)
2517{
2518 CheckComArgSafeArrayNotNull(aVariant);
2519 CheckComArgOutPointerValid(aProgress);
2520 if (aLogicalSize < 0)
2521 return setError(E_INVALIDARG, tr("The medium size argument (%lld) is negative"), aLogicalSize);
2522
2523 AutoCaller autoCaller(this);
2524 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2525
2526 HRESULT rc = S_OK;
2527 ComObjPtr<Progress> pProgress;
2528 Medium::Task *pTask = NULL;
2529
2530 try
2531 {
2532 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2533
2534 ULONG mediumVariantFlags = 0;
2535
2536 if (aVariant)
2537 {
2538 com::SafeArray<MediumVariant_T> variants(ComSafeArrayInArg(aVariant));
2539 for (size_t i = 0; i < variants.size(); i++)
2540 mediumVariantFlags |= variants[i];
2541 }
2542
2543 mediumVariantFlags &= ((unsigned)~MediumVariant_Diff);
2544
2545 if ( !(mediumVariantFlags & MediumVariant_Fixed)
2546 && !(m->formatObj->i_getCapabilities() & MediumFormatCapabilities_CreateDynamic))
2547 throw setError(VBOX_E_NOT_SUPPORTED,
2548 tr("Medium format '%s' does not support dynamic storage creation"),
2549 m->strFormat.c_str());
2550
2551 if ( (mediumVariantFlags & MediumVariant_Fixed)
2552 && !(m->formatObj->i_getCapabilities() & MediumFormatCapabilities_CreateDynamic))
2553 throw setError(VBOX_E_NOT_SUPPORTED,
2554 tr("Medium format '%s' does not support fixed storage creation"),
2555 m->strFormat.c_str());
2556
2557 if (m->state != MediumState_NotCreated)
2558 throw setStateError();
2559
2560 pProgress.createObject();
2561 rc = pProgress->init(m->pVirtualBox,
2562 static_cast<IMedium*>(this),
2563 (mediumVariantFlags & MediumVariant_Fixed)
2564 ? BstrFmt(tr("Creating fixed medium storage unit '%s'"), m->strLocationFull.c_str()).raw()
2565 : BstrFmt(tr("Creating dynamic medium storage unit '%s'"), m->strLocationFull.c_str()).raw(),
2566 TRUE /* aCancelable */);
2567 if (FAILED(rc))
2568 throw rc;
2569
2570 /* setup task object to carry out the operation asynchronously */
2571 pTask = new Medium::CreateBaseTask(this, pProgress, aLogicalSize,
2572 (MediumVariant_T)mediumVariantFlags);
2573 //(MediumVariant_T)aVariant);
2574 rc = pTask->rc();
2575 AssertComRC(rc);
2576 if (FAILED(rc))
2577 throw rc;
2578
2579 m->state = MediumState_Creating;
2580 }
2581 catch (HRESULT aRC) { rc = aRC; }
2582
2583 if (SUCCEEDED(rc))
2584 {
2585 rc = startThread(pTask);
2586
2587 if (SUCCEEDED(rc))
2588 pProgress.queryInterfaceTo(aProgress);
2589 }
2590 else if (pTask != NULL)
2591 delete pTask;
2592
2593 return rc;
2594}
2595
2596STDMETHODIMP Medium::DeleteStorage(IProgress **aProgress)
2597{
2598 CheckComArgOutPointerValid(aProgress);
2599
2600 AutoCaller autoCaller(this);
2601 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2602
2603 ComObjPtr<Progress> pProgress;
2604
2605 MultiResult mrc = deleteStorage(&pProgress,
2606 false /* aWait */);
2607 /* Must save the registries in any case, since an entry was removed. */
2608 m->pVirtualBox->saveModifiedRegistries();
2609
2610 if (SUCCEEDED(mrc))
2611 pProgress.queryInterfaceTo(aProgress);
2612
2613 return mrc;
2614}
2615
2616STDMETHODIMP Medium::CreateDiffStorage(IMedium *aTarget,
2617 ComSafeArrayIn(MediumVariant_T, aVariant),
2618 IProgress **aProgress)
2619{
2620 CheckComArgNotNull(aTarget);
2621 CheckComArgOutPointerValid(aProgress);
2622 CheckComArgSafeArrayNotNull(aVariant);
2623
2624 AutoCaller autoCaller(this);
2625 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2626
2627 ComObjPtr<Medium> diff = static_cast<Medium*>(aTarget);
2628
2629 // locking: we need the tree lock first because we access parent pointers
2630 AutoMultiWriteLock3 alock(&m->pVirtualBox->getMediaTreeLockHandle(),
2631 this->lockHandle(), diff->lockHandle() COMMA_LOCKVAL_SRC_POS);
2632
2633 if (m->type == MediumType_Writethrough)
2634 return setError(VBOX_E_INVALID_OBJECT_STATE,
2635 tr("Medium type of '%s' is Writethrough"),
2636 m->strLocationFull.c_str());
2637 else if (m->type == MediumType_Shareable)
2638 return setError(VBOX_E_INVALID_OBJECT_STATE,
2639 tr("Medium type of '%s' is Shareable"),
2640 m->strLocationFull.c_str());
2641 else if (m->type == MediumType_Readonly)
2642 return setError(VBOX_E_INVALID_OBJECT_STATE,
2643 tr("Medium type of '%s' is Readonly"),
2644 m->strLocationFull.c_str());
2645
2646 /* Apply the normal locking logic to the entire chain. */
2647 MediumLockList *pMediumLockList(new MediumLockList());
2648 alock.release();
2649 HRESULT rc = diff->createMediumLockList(true /* fFailIfInaccessible */,
2650 true /* fMediumLockWrite */,
2651 this,
2652 *pMediumLockList);
2653 alock.acquire();
2654 if (FAILED(rc))
2655 {
2656 delete pMediumLockList;
2657 return rc;
2658 }
2659
2660 alock.release();
2661 rc = pMediumLockList->Lock();
2662 alock.acquire();
2663 if (FAILED(rc))
2664 {
2665 delete pMediumLockList;
2666
2667 return setError(rc, tr("Could not lock medium when creating diff '%s'"),
2668 diff->getLocationFull().c_str());
2669 }
2670
2671 Guid parentMachineRegistry;
2672 if (getFirstRegistryMachineId(parentMachineRegistry))
2673 {
2674 /* since this medium has been just created it isn't associated yet */
2675 diff->m->llRegistryIDs.push_back(parentMachineRegistry);
2676 alock.release();
2677 diff->markRegistriesModified();
2678 alock.acquire();
2679 }
2680
2681 alock.release();
2682
2683 ComObjPtr<Progress> pProgress;
2684
2685 ULONG mediumVariantFlags = 0;
2686
2687 if (aVariant)
2688 {
2689 com::SafeArray<MediumVariant_T> variants(ComSafeArrayInArg(aVariant));
2690 for (size_t i = 0; i < variants.size(); i++)
2691 mediumVariantFlags |= variants[i];
2692 }
2693
2694 rc = createDiffStorage(diff, (MediumVariant_T)mediumVariantFlags, pMediumLockList,
2695 &pProgress, false /* aWait */);
2696 if (FAILED(rc))
2697 delete pMediumLockList;
2698 else
2699 pProgress.queryInterfaceTo(aProgress);
2700
2701 return rc;
2702}
2703
2704STDMETHODIMP Medium::MergeTo(IMedium *aTarget, IProgress **aProgress)
2705{
2706 CheckComArgNotNull(aTarget);
2707 CheckComArgOutPointerValid(aProgress);
2708 ComAssertRet(aTarget != this, E_INVALIDARG);
2709
2710 AutoCaller autoCaller(this);
2711 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2712
2713 ComObjPtr<Medium> pTarget = static_cast<Medium*>(aTarget);
2714
2715 bool fMergeForward = false;
2716 ComObjPtr<Medium> pParentForTarget;
2717 MediumLockList *pChildrenToReparent = NULL;
2718 MediumLockList *pMediumLockList = NULL;
2719
2720 HRESULT rc = S_OK;
2721
2722 rc = prepareMergeTo(pTarget, NULL, NULL, true, fMergeForward,
2723 pParentForTarget, pChildrenToReparent, pMediumLockList);
2724 if (FAILED(rc)) return rc;
2725
2726 ComObjPtr<Progress> pProgress;
2727
2728 rc = mergeTo(pTarget, fMergeForward, pParentForTarget, pChildrenToReparent,
2729 pMediumLockList, &pProgress, false /* aWait */);
2730 if (FAILED(rc))
2731 cancelMergeTo(pChildrenToReparent, pMediumLockList);
2732 else
2733 pProgress.queryInterfaceTo(aProgress);
2734
2735 return rc;
2736}
2737
2738STDMETHODIMP Medium::CloneToBase(IMedium *aTarget,
2739 ComSafeArrayIn(MediumVariant_T, aVariant),
2740 IProgress **aProgress)
2741{
2742 int rc = S_OK;
2743 CheckComArgNotNull(aTarget);
2744 CheckComArgOutPointerValid(aProgress);
2745 CheckComArgSafeArrayNotNull(aVariant);
2746
2747 com::SafeArray<MediumVariant_T> variants(ComSafeArrayInArg(aVariant));
2748
2749 rc = CloneTo(aTarget, ComSafeArrayAsInParam(variants), NULL, aProgress);
2750 return rc;
2751}
2752
2753STDMETHODIMP Medium::CloneTo(IMedium *aTarget,
2754 ComSafeArrayIn(MediumVariant_T, aVariant),
2755 IMedium *aParent,
2756 IProgress **aProgress)
2757{
2758 CheckComArgNotNull(aTarget);
2759 CheckComArgOutPointerValid(aProgress);
2760 CheckComArgSafeArrayNotNull(aVariant);
2761
2762 ComAssertRet(aTarget != this, E_INVALIDARG);
2763
2764 AutoCaller autoCaller(this);
2765 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2766
2767 ComObjPtr<Medium> pTarget = static_cast<Medium*>(aTarget);
2768 ComObjPtr<Medium> pParent;
2769 if (aParent)
2770 pParent = static_cast<Medium*>(aParent);
2771
2772 HRESULT rc = S_OK;
2773 ComObjPtr<Progress> pProgress;
2774 Medium::Task *pTask = NULL;
2775
2776 try
2777 {
2778 // locking: we need the tree lock first because we access parent pointers
2779 // and we need to write-lock the media involved
2780 uint32_t cHandles = 3;
2781 LockHandle* pHandles[4] = { &m->pVirtualBox->getMediaTreeLockHandle(),
2782 this->lockHandle(),
2783 pTarget->lockHandle() };
2784 /* Only add parent to the lock if it is not null */
2785 if (!pParent.isNull())
2786 pHandles[cHandles++] = pParent->lockHandle();
2787 AutoWriteLock alock(cHandles,
2788 pHandles
2789 COMMA_LOCKVAL_SRC_POS);
2790
2791 if ( pTarget->m->state != MediumState_NotCreated
2792 && pTarget->m->state != MediumState_Created)
2793 throw pTarget->setStateError();
2794
2795 /* Build the source lock list. */
2796 MediumLockList *pSourceMediumLockList(new MediumLockList());
2797 alock.release();
2798 rc = createMediumLockList(true /* fFailIfInaccessible */,
2799 false /* fMediumLockWrite */,
2800 NULL,
2801 *pSourceMediumLockList);
2802 alock.acquire();
2803 if (FAILED(rc))
2804 {
2805 delete pSourceMediumLockList;
2806 throw rc;
2807 }
2808
2809 /* Build the target lock list (including the to-be parent chain). */
2810 MediumLockList *pTargetMediumLockList(new MediumLockList());
2811 alock.release();
2812 rc = pTarget->createMediumLockList(true /* fFailIfInaccessible */,
2813 true /* fMediumLockWrite */,
2814 pParent,
2815 *pTargetMediumLockList);
2816 alock.acquire();
2817 if (FAILED(rc))
2818 {
2819 delete pSourceMediumLockList;
2820 delete pTargetMediumLockList;
2821 throw rc;
2822 }
2823
2824 alock.release();
2825 rc = pSourceMediumLockList->Lock();
2826 alock.acquire();
2827 if (FAILED(rc))
2828 {
2829 delete pSourceMediumLockList;
2830 delete pTargetMediumLockList;
2831 throw setError(rc,
2832 tr("Failed to lock source media '%s'"),
2833 getLocationFull().c_str());
2834 }
2835 alock.release();
2836 rc = pTargetMediumLockList->Lock();
2837 alock.acquire();
2838 if (FAILED(rc))
2839 {
2840 delete pSourceMediumLockList;
2841 delete pTargetMediumLockList;
2842 throw setError(rc,
2843 tr("Failed to lock target media '%s'"),
2844 pTarget->getLocationFull().c_str());
2845 }
2846
2847 pProgress.createObject();
2848 rc = pProgress->init(m->pVirtualBox,
2849 static_cast <IMedium *>(this),
2850 BstrFmt(tr("Creating clone medium '%s'"), pTarget->m->strLocationFull.c_str()).raw(),
2851 TRUE /* aCancelable */);
2852 if (FAILED(rc))
2853 {
2854 delete pSourceMediumLockList;
2855 delete pTargetMediumLockList;
2856 throw rc;
2857 }
2858
2859 ULONG mediumVariantFlags = 0;
2860
2861 if (aVariant)
2862 {
2863 com::SafeArray<MediumVariant_T> variants(ComSafeArrayInArg(aVariant));
2864 for (size_t i = 0; i < variants.size(); i++)
2865 mediumVariantFlags |= variants[i];
2866 }
2867
2868 /* setup task object to carry out the operation asynchronously */
2869 pTask = new Medium::CloneTask(this, pProgress, pTarget,
2870 (MediumVariant_T)mediumVariantFlags,
2871 pParent, UINT32_MAX, UINT32_MAX,
2872 pSourceMediumLockList, pTargetMediumLockList);
2873 rc = pTask->rc();
2874 AssertComRC(rc);
2875 if (FAILED(rc))
2876 throw rc;
2877
2878 if (pTarget->m->state == MediumState_NotCreated)
2879 pTarget->m->state = MediumState_Creating;
2880 }
2881 catch (HRESULT aRC) { rc = aRC; }
2882
2883 if (SUCCEEDED(rc))
2884 {
2885 rc = startThread(pTask);
2886
2887 if (SUCCEEDED(rc))
2888 pProgress.queryInterfaceTo(aProgress);
2889 }
2890 else if (pTask != NULL)
2891 delete pTask;
2892
2893 return rc;
2894}
2895
2896STDMETHODIMP Medium::Compact(IProgress **aProgress)
2897{
2898 CheckComArgOutPointerValid(aProgress);
2899
2900 AutoCaller autoCaller(this);
2901 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2902
2903 HRESULT rc = S_OK;
2904 ComObjPtr<Progress> pProgress;
2905 Medium::Task *pTask = NULL;
2906
2907 try
2908 {
2909 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2910
2911 /* Build the medium lock list. */
2912 MediumLockList *pMediumLockList(new MediumLockList());
2913 alock.release();
2914 rc = createMediumLockList(true /* fFailIfInaccessible */ ,
2915 true /* fMediumLockWrite */,
2916 NULL,
2917 *pMediumLockList);
2918 alock.acquire();
2919 if (FAILED(rc))
2920 {
2921 delete pMediumLockList;
2922 throw rc;
2923 }
2924
2925 alock.release();
2926 rc = pMediumLockList->Lock();
2927 alock.acquire();
2928 if (FAILED(rc))
2929 {
2930 delete pMediumLockList;
2931 throw setError(rc,
2932 tr("Failed to lock media when compacting '%s'"),
2933 getLocationFull().c_str());
2934 }
2935
2936 pProgress.createObject();
2937 rc = pProgress->init(m->pVirtualBox,
2938 static_cast <IMedium *>(this),
2939 BstrFmt(tr("Compacting medium '%s'"), m->strLocationFull.c_str()).raw(),
2940 TRUE /* aCancelable */);
2941 if (FAILED(rc))
2942 {
2943 delete pMediumLockList;
2944 throw rc;
2945 }
2946
2947 /* setup task object to carry out the operation asynchronously */
2948 pTask = new Medium::CompactTask(this, pProgress, pMediumLockList);
2949 rc = pTask->rc();
2950 AssertComRC(rc);
2951 if (FAILED(rc))
2952 throw rc;
2953 }
2954 catch (HRESULT aRC) { rc = aRC; }
2955
2956 if (SUCCEEDED(rc))
2957 {
2958 rc = startThread(pTask);
2959
2960 if (SUCCEEDED(rc))
2961 pProgress.queryInterfaceTo(aProgress);
2962 }
2963 else if (pTask != NULL)
2964 delete pTask;
2965
2966 return rc;
2967}
2968
2969STDMETHODIMP Medium::Resize(LONG64 aLogicalSize, IProgress **aProgress)
2970{
2971 CheckComArgOutPointerValid(aProgress);
2972
2973 AutoCaller autoCaller(this);
2974 if (FAILED(autoCaller.rc())) return autoCaller.rc();
2975
2976 HRESULT rc = S_OK;
2977 ComObjPtr<Progress> pProgress;
2978 Medium::Task *pTask = NULL;
2979
2980 try
2981 {
2982 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
2983
2984 /* Build the medium lock list. */
2985 MediumLockList *pMediumLockList(new MediumLockList());
2986 alock.release();
2987 rc = createMediumLockList(true /* fFailIfInaccessible */ ,
2988 true /* fMediumLockWrite */,
2989 NULL,
2990 *pMediumLockList);
2991 alock.acquire();
2992 if (FAILED(rc))
2993 {
2994 delete pMediumLockList;
2995 throw rc;
2996 }
2997
2998 alock.release();
2999 rc = pMediumLockList->Lock();
3000 alock.acquire();
3001 if (FAILED(rc))
3002 {
3003 delete pMediumLockList;
3004 throw setError(rc,
3005 tr("Failed to lock media when compacting '%s'"),
3006 getLocationFull().c_str());
3007 }
3008
3009 pProgress.createObject();
3010 rc = pProgress->init(m->pVirtualBox,
3011 static_cast <IMedium *>(this),
3012 BstrFmt(tr("Compacting medium '%s'"), m->strLocationFull.c_str()).raw(),
3013 TRUE /* aCancelable */);
3014 if (FAILED(rc))
3015 {
3016 delete pMediumLockList;
3017 throw rc;
3018 }
3019
3020 /* setup task object to carry out the operation asynchronously */
3021 pTask = new Medium::ResizeTask(this, aLogicalSize, pProgress, pMediumLockList);
3022 rc = pTask->rc();
3023 AssertComRC(rc);
3024 if (FAILED(rc))
3025 throw rc;
3026 }
3027 catch (HRESULT aRC) { rc = aRC; }
3028
3029 if (SUCCEEDED(rc))
3030 {
3031 rc = startThread(pTask);
3032
3033 if (SUCCEEDED(rc))
3034 pProgress.queryInterfaceTo(aProgress);
3035 }
3036 else if (pTask != NULL)
3037 delete pTask;
3038
3039 return rc;
3040}
3041
3042STDMETHODIMP Medium::Reset(IProgress **aProgress)
3043{
3044 CheckComArgOutPointerValid(aProgress);
3045
3046 AutoCaller autoCaller(this);
3047 if (FAILED(autoCaller.rc())) return autoCaller.rc();
3048
3049 HRESULT rc = S_OK;
3050 ComObjPtr<Progress> pProgress;
3051 Medium::Task *pTask = NULL;
3052
3053 try
3054 {
3055 /* canClose() needs the tree lock */
3056 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
3057 this->lockHandle()
3058 COMMA_LOCKVAL_SRC_POS);
3059
3060 LogFlowThisFunc(("ENTER for medium %s\n", m->strLocationFull.c_str()));
3061
3062 if (m->pParent.isNull())
3063 throw setError(VBOX_E_NOT_SUPPORTED,
3064 tr("Medium type of '%s' is not differencing"),
3065 m->strLocationFull.c_str());
3066
3067 rc = canClose();
3068 if (FAILED(rc))
3069 throw rc;
3070
3071 /* Build the medium lock list. */
3072 MediumLockList *pMediumLockList(new MediumLockList());
3073 multilock.release();
3074 rc = createMediumLockList(true /* fFailIfInaccessible */,
3075 true /* fMediumLockWrite */,
3076 NULL,
3077 *pMediumLockList);
3078 multilock.acquire();
3079 if (FAILED(rc))
3080 {
3081 delete pMediumLockList;
3082 throw rc;
3083 }
3084
3085 multilock.release();
3086 rc = pMediumLockList->Lock();
3087 multilock.acquire();
3088 if (FAILED(rc))
3089 {
3090 delete pMediumLockList;
3091 throw setError(rc,
3092 tr("Failed to lock media when resetting '%s'"),
3093 getLocationFull().c_str());
3094 }
3095
3096 pProgress.createObject();
3097 rc = pProgress->init(m->pVirtualBox,
3098 static_cast<IMedium*>(this),
3099 BstrFmt(tr("Resetting differencing medium '%s'"), m->strLocationFull.c_str()).raw(),
3100 FALSE /* aCancelable */);
3101 if (FAILED(rc))
3102 throw rc;
3103
3104 /* setup task object to carry out the operation asynchronously */
3105 pTask = new Medium::ResetTask(this, pProgress, pMediumLockList);
3106 rc = pTask->rc();
3107 AssertComRC(rc);
3108 if (FAILED(rc))
3109 throw rc;
3110 }
3111 catch (HRESULT aRC) { rc = aRC; }
3112
3113 if (SUCCEEDED(rc))
3114 {
3115 rc = startThread(pTask);
3116
3117 if (SUCCEEDED(rc))
3118 pProgress.queryInterfaceTo(aProgress);
3119 }
3120 else if (pTask != NULL)
3121 delete pTask;
3122
3123 LogFlowThisFunc(("LEAVE, rc=%Rhrc\n", rc));
3124
3125 return rc;
3126}
3127
3128////////////////////////////////////////////////////////////////////////////////
3129//
3130// Medium public internal methods
3131//
3132////////////////////////////////////////////////////////////////////////////////
3133
3134/**
3135 * Internal method to return the medium's parent medium. Must have caller + locking!
3136 * @return
3137 */
3138const ComObjPtr<Medium>& Medium::getParent() const
3139{
3140 return m->pParent;
3141}
3142
3143/**
3144 * Internal method to return the medium's list of child media. Must have caller + locking!
3145 * @return
3146 */
3147const MediaList& Medium::getChildren() const
3148{
3149 return m->llChildren;
3150}
3151
3152/**
3153 * Internal method to return the medium's GUID. Must have caller + locking!
3154 * @return
3155 */
3156const Guid& Medium::getId() const
3157{
3158 return m->id;
3159}
3160
3161/**
3162 * Internal method to return the medium's state. Must have caller + locking!
3163 * @return
3164 */
3165MediumState_T Medium::getState() const
3166{
3167 return m->state;
3168}
3169
3170/**
3171 * Internal method to return the medium's variant. Must have caller + locking!
3172 * @return
3173 */
3174MediumVariant_T Medium::getVariant() const
3175{
3176 return m->variant;
3177}
3178
3179/**
3180 * Internal method which returns true if this medium represents a host drive.
3181 * @return
3182 */
3183bool Medium::isHostDrive() const
3184{
3185 return m->hostDrive;
3186}
3187
3188/**
3189 * Internal method to return the medium's full location. Must have caller + locking!
3190 * @return
3191 */
3192const Utf8Str& Medium::getLocationFull() const
3193{
3194 return m->strLocationFull;
3195}
3196
3197/**
3198 * Internal method to return the medium's format string. Must have caller + locking!
3199 * @return
3200 */
3201const Utf8Str& Medium::getFormat() const
3202{
3203 return m->strFormat;
3204}
3205
3206/**
3207 * Internal method to return the medium's format object. Must have caller + locking!
3208 * @return
3209 */
3210const ComObjPtr<MediumFormat>& Medium::getMediumFormat() const
3211{
3212 return m->formatObj;
3213}
3214
3215/**
3216 * Internal method that returns true if the medium is represented by a file on the host disk
3217 * (and not iSCSI or something).
3218 * @return
3219 */
3220bool Medium::isMediumFormatFile() const
3221{
3222 if ( m->formatObj
3223 && (m->formatObj->i_getCapabilities() & MediumFormatCapabilities_File)
3224 )
3225 return true;
3226 return false;
3227}
3228
3229/**
3230 * Internal method to return the medium's size. Must have caller + locking!
3231 * @return
3232 */
3233uint64_t Medium::getSize() const
3234{
3235 return m->size;
3236}
3237
3238/**
3239 * Returns the medium device type. Must have caller + locking!
3240 * @return
3241 */
3242DeviceType_T Medium::getDeviceType() const
3243{
3244 return m->devType;
3245}
3246
3247/**
3248 * Returns the medium type. Must have caller + locking!
3249 * @return
3250 */
3251MediumType_T Medium::getType() const
3252{
3253 return m->type;
3254}
3255
3256/**
3257 * Returns a short version of the location attribute.
3258 *
3259 * @note Must be called from under this object's read or write lock.
3260 */
3261Utf8Str Medium::getName()
3262{
3263 Utf8Str name = RTPathFilename(m->strLocationFull.c_str());
3264 return name;
3265}
3266
3267/**
3268 * This adds the given UUID to the list of media registries in which this
3269 * medium should be registered. The UUID can either be a machine UUID,
3270 * to add a machine registry, or the global registry UUID as returned by
3271 * VirtualBox::getGlobalRegistryId().
3272 *
3273 * Note that for hard disks, this method does nothing if the medium is
3274 * already in another registry to avoid having hard disks in more than
3275 * one registry, which causes trouble with keeping diff images in sync.
3276 * See getFirstRegistryMachineId() for details.
3277 *
3278 * If fRecurse == true, then the media tree lock must be held for reading.
3279 *
3280 * @param id
3281 * @param fRecurse If true, recurses into child media to make sure the whole tree has registries in sync.
3282 * @return true if the registry was added; false if the given id was already on the list.
3283 */
3284bool Medium::addRegistry(const Guid& id, bool fRecurse)
3285{
3286 AutoCaller autoCaller(this);
3287 if (FAILED(autoCaller.rc()))
3288 return false;
3289 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3290
3291 bool fAdd = true;
3292
3293 // hard disks cannot be in more than one registry
3294 if ( m->devType == DeviceType_HardDisk
3295 && m->llRegistryIDs.size() > 0)
3296 fAdd = false;
3297
3298 // no need to add the UUID twice
3299 if (fAdd)
3300 {
3301 for (GuidList::const_iterator it = m->llRegistryIDs.begin();
3302 it != m->llRegistryIDs.end();
3303 ++it)
3304 {
3305 if ((*it) == id)
3306 {
3307 fAdd = false;
3308 break;
3309 }
3310 }
3311 }
3312
3313 if (fAdd)
3314 m->llRegistryIDs.push_back(id);
3315
3316 if (fRecurse)
3317 {
3318 // Get private list of children and release medium lock straight away.
3319 MediaList llChildren(m->llChildren);
3320 alock.release();
3321
3322 for (MediaList::iterator it = llChildren.begin();
3323 it != llChildren.end();
3324 ++it)
3325 {
3326 Medium *pChild = *it;
3327 fAdd |= pChild->addRegistry(id, true);
3328 }
3329 }
3330
3331 return fAdd;
3332}
3333
3334/**
3335 * Removes the given UUID from the list of media registry UUIDs. Returns true
3336 * if found or false if not.
3337 *
3338 * If fRecurse == true, then the media tree lock must be held for reading.
3339 *
3340 * @param id
3341 * @param fRecurse If true, recurses into child media to make sure the whole tree has registries in sync.
3342 * @return
3343 */
3344bool Medium::removeRegistry(const Guid& id, bool fRecurse)
3345{
3346 AutoCaller autoCaller(this);
3347 if (FAILED(autoCaller.rc()))
3348 return false;
3349 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3350
3351 bool fRemove = false;
3352
3353 for (GuidList::iterator it = m->llRegistryIDs.begin();
3354 it != m->llRegistryIDs.end();
3355 ++it)
3356 {
3357 if ((*it) == id)
3358 {
3359 m->llRegistryIDs.erase(it);
3360 fRemove = true;
3361 break;
3362 }
3363 }
3364
3365 if (fRecurse)
3366 {
3367 // Get private list of children and release medium lock straight away.
3368 MediaList llChildren(m->llChildren);
3369 alock.release();
3370
3371 for (MediaList::iterator it = llChildren.begin();
3372 it != llChildren.end();
3373 ++it)
3374 {
3375 Medium *pChild = *it;
3376 fRemove |= pChild->removeRegistry(id, true);
3377 }
3378 }
3379
3380 return fRemove;
3381}
3382
3383/**
3384 * Returns true if id is in the list of media registries for this medium.
3385 *
3386 * Must have caller + read locking!
3387 *
3388 * @param id
3389 * @return
3390 */
3391bool Medium::isInRegistry(const Guid& id)
3392{
3393 for (GuidList::const_iterator it = m->llRegistryIDs.begin();
3394 it != m->llRegistryIDs.end();
3395 ++it)
3396 {
3397 if (*it == id)
3398 return true;
3399 }
3400
3401 return false;
3402}
3403
3404/**
3405 * Internal method to return the medium's first registry machine (i.e. the machine in whose
3406 * machine XML this medium is listed).
3407 *
3408 * Every attached medium must now (4.0) reside in at least one media registry, which is identified
3409 * by a UUID. This is either a machine UUID if the machine is from 4.0 or newer, in which case
3410 * machines have their own media registries, or it is the pseudo-UUID of the VirtualBox
3411 * object if the machine is old and still needs the global registry in VirtualBox.xml.
3412 *
3413 * By definition, hard disks may only be in one media registry, in which all its children
3414 * will be stored as well. Otherwise we run into problems with having keep multiple registries
3415 * in sync. (This is the "cloned VM" case in which VM1 may link to the disks of VM2; in this
3416 * case, only VM2's registry is used for the disk in question.)
3417 *
3418 * If there is no medium registry, particularly if the medium has not been attached yet, this
3419 * does not modify uuid and returns false.
3420 *
3421 * ISOs and RAWs, by contrast, can be in more than one repository to make things easier for
3422 * the user.
3423 *
3424 * Must have caller + locking!
3425 *
3426 * @param uuid Receives first registry machine UUID, if available.
3427 * @return true if uuid was set.
3428 */
3429bool Medium::getFirstRegistryMachineId(Guid &uuid) const
3430{
3431 if (m->llRegistryIDs.size())
3432 {
3433 uuid = m->llRegistryIDs.front();
3434 return true;
3435 }
3436 return false;
3437}
3438
3439/**
3440 * Marks all the registries in which this medium is registered as modified.
3441 */
3442void Medium::markRegistriesModified()
3443{
3444 AutoCaller autoCaller(this);
3445 if (FAILED(autoCaller.rc())) return;
3446
3447 // Get local copy, as keeping the lock over VirtualBox::markRegistryModified
3448 // causes trouble with the lock order
3449 GuidList llRegistryIDs;
3450 {
3451 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3452 llRegistryIDs = m->llRegistryIDs;
3453 }
3454
3455 /* Save the error information now, the implicit restore when this goes
3456 * out of scope will throw away spurious additional errors created below. */
3457 ErrorInfoKeeper eik;
3458 for (GuidList::const_iterator it = llRegistryIDs.begin();
3459 it != llRegistryIDs.end();
3460 ++it)
3461 {
3462 m->pVirtualBox->markRegistryModified(*it);
3463 }
3464}
3465
3466/**
3467 * Adds the given machine and optionally the snapshot to the list of the objects
3468 * this medium is attached to.
3469 *
3470 * @param aMachineId Machine ID.
3471 * @param aSnapshotId Snapshot ID; when non-empty, adds a snapshot attachment.
3472 */
3473HRESULT Medium::addBackReference(const Guid &aMachineId,
3474 const Guid &aSnapshotId /*= Guid::Empty*/)
3475{
3476 AssertReturn(aMachineId.isValid(), E_FAIL);
3477
3478 LogFlowThisFunc(("ENTER, aMachineId: {%RTuuid}, aSnapshotId: {%RTuuid}\n", aMachineId.raw(), aSnapshotId.raw()));
3479
3480 AutoCaller autoCaller(this);
3481 AssertComRCReturnRC(autoCaller.rc());
3482
3483 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3484
3485 switch (m->state)
3486 {
3487 case MediumState_Created:
3488 case MediumState_Inaccessible:
3489 case MediumState_LockedRead:
3490 case MediumState_LockedWrite:
3491 break;
3492
3493 default:
3494 return setStateError();
3495 }
3496
3497 if (m->numCreateDiffTasks > 0)
3498 return setError(VBOX_E_OBJECT_IN_USE,
3499 tr("Cannot attach medium '%s' {%RTuuid}: %u differencing child media are being created"),
3500 m->strLocationFull.c_str(),
3501 m->id.raw(),
3502 m->numCreateDiffTasks);
3503
3504 BackRefList::iterator it = std::find_if(m->backRefs.begin(),
3505 m->backRefs.end(),
3506 BackRef::EqualsTo(aMachineId));
3507 if (it == m->backRefs.end())
3508 {
3509 BackRef ref(aMachineId, aSnapshotId);
3510 m->backRefs.push_back(ref);
3511
3512 return S_OK;
3513 }
3514
3515 // if the caller has not supplied a snapshot ID, then we're attaching
3516 // to a machine a medium which represents the machine's current state,
3517 // so set the flag
3518
3519 if (aSnapshotId.isZero())
3520 {
3521 /* sanity: no duplicate attachments */
3522 if (it->fInCurState)
3523 return setError(VBOX_E_OBJECT_IN_USE,
3524 tr("Cannot attach medium '%s' {%RTuuid}: medium is already associated with the current state of machine uuid {%RTuuid}!"),
3525 m->strLocationFull.c_str(),
3526 m->id.raw(),
3527 aMachineId.raw());
3528 it->fInCurState = true;
3529
3530 return S_OK;
3531 }
3532
3533 // otherwise: a snapshot medium is being attached
3534
3535 /* sanity: no duplicate attachments */
3536 for (GuidList::const_iterator jt = it->llSnapshotIds.begin();
3537 jt != it->llSnapshotIds.end();
3538 ++jt)
3539 {
3540 const Guid &idOldSnapshot = *jt;
3541
3542 if (idOldSnapshot == aSnapshotId)
3543 {
3544#ifdef DEBUG
3545 dumpBackRefs();
3546#endif
3547 return setError(VBOX_E_OBJECT_IN_USE,
3548 tr("Cannot attach medium '%s' {%RTuuid} from snapshot '%RTuuid': medium is already in use by this snapshot!"),
3549 m->strLocationFull.c_str(),
3550 m->id.raw(),
3551 aSnapshotId.raw());
3552 }
3553 }
3554
3555 it->llSnapshotIds.push_back(aSnapshotId);
3556 // Do not touch fInCurState, as the image may be attached to the current
3557 // state *and* a snapshot, otherwise we lose the current state association!
3558
3559 LogFlowThisFuncLeave();
3560
3561 return S_OK;
3562}
3563
3564/**
3565 * Removes the given machine and optionally the snapshot from the list of the
3566 * objects this medium is attached to.
3567 *
3568 * @param aMachineId Machine ID.
3569 * @param aSnapshotId Snapshot ID; when non-empty, removes the snapshot
3570 * attachment.
3571 */
3572HRESULT Medium::removeBackReference(const Guid &aMachineId,
3573 const Guid &aSnapshotId /*= Guid::Empty*/)
3574{
3575 AssertReturn(aMachineId.isValid(), E_FAIL);
3576
3577 AutoCaller autoCaller(this);
3578 AssertComRCReturnRC(autoCaller.rc());
3579
3580 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3581
3582 BackRefList::iterator it =
3583 std::find_if(m->backRefs.begin(), m->backRefs.end(),
3584 BackRef::EqualsTo(aMachineId));
3585 AssertReturn(it != m->backRefs.end(), E_FAIL);
3586
3587 if (aSnapshotId.isZero())
3588 {
3589 /* remove the current state attachment */
3590 it->fInCurState = false;
3591 }
3592 else
3593 {
3594 /* remove the snapshot attachment */
3595 GuidList::iterator jt = std::find(it->llSnapshotIds.begin(),
3596 it->llSnapshotIds.end(),
3597 aSnapshotId);
3598
3599 AssertReturn(jt != it->llSnapshotIds.end(), E_FAIL);
3600 it->llSnapshotIds.erase(jt);
3601 }
3602
3603 /* if the backref becomes empty, remove it */
3604 if (it->fInCurState == false && it->llSnapshotIds.size() == 0)
3605 m->backRefs.erase(it);
3606
3607 return S_OK;
3608}
3609
3610/**
3611 * Internal method to return the medium's list of backrefs. Must have caller + locking!
3612 * @return
3613 */
3614const Guid* Medium::getFirstMachineBackrefId() const
3615{
3616 if (!m->backRefs.size())
3617 return NULL;
3618
3619 return &m->backRefs.front().machineId;
3620}
3621
3622/**
3623 * Internal method which returns a machine that either this medium or one of its children
3624 * is attached to. This is used for finding a replacement media registry when an existing
3625 * media registry is about to be deleted in VirtualBox::unregisterMachine().
3626 *
3627 * Must have caller + locking, *and* caller must hold the media tree lock!
3628 * @return
3629 */
3630const Guid* Medium::getAnyMachineBackref() const
3631{
3632 if (m->backRefs.size())
3633 return &m->backRefs.front().machineId;
3634
3635 for (MediaList::iterator it = m->llChildren.begin();
3636 it != m->llChildren.end();
3637 ++it)
3638 {
3639 Medium *pChild = *it;
3640 // recurse for this child
3641 const Guid* puuid;
3642 if ((puuid = pChild->getAnyMachineBackref()))
3643 return puuid;
3644 }
3645
3646 return NULL;
3647}
3648
3649const Guid* Medium::getFirstMachineBackrefSnapshotId() const
3650{
3651 if (!m->backRefs.size())
3652 return NULL;
3653
3654 const BackRef &ref = m->backRefs.front();
3655 if (!ref.llSnapshotIds.size())
3656 return NULL;
3657
3658 return &ref.llSnapshotIds.front();
3659}
3660
3661size_t Medium::getMachineBackRefCount() const
3662{
3663 return m->backRefs.size();
3664}
3665
3666#ifdef DEBUG
3667/**
3668 * Debugging helper that gets called after VirtualBox initialization that writes all
3669 * machine backreferences to the debug log.
3670 */
3671void Medium::dumpBackRefs()
3672{
3673 AutoCaller autoCaller(this);
3674 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3675
3676 LogFlowThisFunc(("Dumping backrefs for medium '%s':\n", m->strLocationFull.c_str()));
3677
3678 for (BackRefList::iterator it2 = m->backRefs.begin();
3679 it2 != m->backRefs.end();
3680 ++it2)
3681 {
3682 const BackRef &ref = *it2;
3683 LogFlowThisFunc((" Backref from machine {%RTuuid} (fInCurState: %d)\n", ref.machineId.raw(), ref.fInCurState));
3684
3685 for (GuidList::const_iterator jt2 = it2->llSnapshotIds.begin();
3686 jt2 != it2->llSnapshotIds.end();
3687 ++jt2)
3688 {
3689 const Guid &id = *jt2;
3690 LogFlowThisFunc((" Backref from snapshot {%RTuuid}\n", id.raw()));
3691 }
3692 }
3693}
3694#endif
3695
3696/**
3697 * Checks if the given change of \a aOldPath to \a aNewPath affects the location
3698 * of this media and updates it if necessary to reflect the new location.
3699 *
3700 * @param aOldPath Old path (full).
3701 * @param aNewPath New path (full).
3702 *
3703 * @note Locks this object for writing.
3704 */
3705HRESULT Medium::updatePath(const Utf8Str &strOldPath, const Utf8Str &strNewPath)
3706{
3707 AssertReturn(!strOldPath.isEmpty(), E_FAIL);
3708 AssertReturn(!strNewPath.isEmpty(), E_FAIL);
3709
3710 AutoCaller autoCaller(this);
3711 if (FAILED(autoCaller.rc())) return autoCaller.rc();
3712
3713 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
3714
3715 LogFlowThisFunc(("locationFull.before='%s'\n", m->strLocationFull.c_str()));
3716
3717 const char *pcszMediumPath = m->strLocationFull.c_str();
3718
3719 if (RTPathStartsWith(pcszMediumPath, strOldPath.c_str()))
3720 {
3721 Utf8Str newPath(strNewPath);
3722 newPath.append(pcszMediumPath + strOldPath.length());
3723 unconst(m->strLocationFull) = newPath;
3724
3725 LogFlowThisFunc(("locationFull.after='%s'\n", m->strLocationFull.c_str()));
3726 // we changed something
3727 return S_OK;
3728 }
3729
3730 // no change was necessary, signal error which the caller needs to interpret
3731 return VBOX_E_FILE_ERROR;
3732}
3733
3734/**
3735 * Returns the base medium of the media chain this medium is part of.
3736 *
3737 * The base medium is found by walking up the parent-child relationship axis.
3738 * If the medium doesn't have a parent (i.e. it's a base medium), it
3739 * returns itself in response to this method.
3740 *
3741 * @param aLevel Where to store the number of ancestors of this medium
3742 * (zero for the base), may be @c NULL.
3743 *
3744 * @note Locks medium tree for reading.
3745 */
3746ComObjPtr<Medium> Medium::getBase(uint32_t *aLevel /*= NULL*/)
3747{
3748 ComObjPtr<Medium> pBase;
3749 uint32_t level;
3750
3751 AutoCaller autoCaller(this);
3752 AssertReturn(autoCaller.isOk(), pBase);
3753
3754 /* we access mParent */
3755 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3756
3757 pBase = this;
3758 level = 0;
3759
3760 if (m->pParent)
3761 {
3762 for (;;)
3763 {
3764 AutoCaller baseCaller(pBase);
3765 AssertReturn(baseCaller.isOk(), pBase);
3766
3767 if (pBase->m->pParent.isNull())
3768 break;
3769
3770 pBase = pBase->m->pParent;
3771 ++level;
3772 }
3773 }
3774
3775 if (aLevel != NULL)
3776 *aLevel = level;
3777
3778 return pBase;
3779}
3780
3781/**
3782 * Returns @c true if this medium cannot be modified because it has
3783 * dependents (children) or is part of the snapshot. Related to the medium
3784 * type and posterity, not to the current media state.
3785 *
3786 * @note Locks this object and medium tree for reading.
3787 */
3788bool Medium::isReadOnly()
3789{
3790 AutoCaller autoCaller(this);
3791 AssertComRCReturn(autoCaller.rc(), false);
3792
3793 /* we access children */
3794 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3795
3796 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3797
3798 switch (m->type)
3799 {
3800 case MediumType_Normal:
3801 {
3802 if (getChildren().size() != 0)
3803 return true;
3804
3805 for (BackRefList::const_iterator it = m->backRefs.begin();
3806 it != m->backRefs.end(); ++it)
3807 if (it->llSnapshotIds.size() != 0)
3808 return true;
3809
3810 if (m->variant & MediumVariant_VmdkStreamOptimized)
3811 return true;
3812
3813 return false;
3814 }
3815 case MediumType_Immutable:
3816 case MediumType_MultiAttach:
3817 return true;
3818 case MediumType_Writethrough:
3819 case MediumType_Shareable:
3820 case MediumType_Readonly: /* explicit readonly media has no diffs */
3821 return false;
3822 default:
3823 break;
3824 }
3825
3826 AssertFailedReturn(false);
3827}
3828
3829/**
3830 * Internal method to return the medium's size. Must have caller + locking!
3831 * @return
3832 */
3833void Medium::updateId(const Guid &id)
3834{
3835 unconst(m->id) = id;
3836}
3837
3838/**
3839 * Saves medium data by appending a new child node to the given
3840 * parent XML settings node.
3841 *
3842 * @param data Settings struct to be updated.
3843 * @param strHardDiskFolder Folder for which paths should be relative.
3844 *
3845 * @note Locks this object, medium tree and children for reading.
3846 */
3847HRESULT Medium::saveSettings(settings::Medium &data,
3848 const Utf8Str &strHardDiskFolder)
3849{
3850 AutoCaller autoCaller(this);
3851 if (FAILED(autoCaller.rc())) return autoCaller.rc();
3852
3853 /* we access mParent */
3854 AutoReadLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
3855
3856 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3857
3858 data.uuid = m->id;
3859
3860 // make path relative if needed
3861 if ( !strHardDiskFolder.isEmpty()
3862 && RTPathStartsWith(m->strLocationFull.c_str(), strHardDiskFolder.c_str())
3863 )
3864 data.strLocation = m->strLocationFull.substr(strHardDiskFolder.length() + 1);
3865 else
3866 data.strLocation = m->strLocationFull;
3867 data.strFormat = m->strFormat;
3868
3869 /* optional, only for diffs, default is false */
3870 if (m->pParent)
3871 data.fAutoReset = m->autoReset;
3872 else
3873 data.fAutoReset = false;
3874
3875 /* optional */
3876 data.strDescription = m->strDescription;
3877
3878 /* optional properties */
3879 data.properties.clear();
3880
3881 /* handle iSCSI initiator secrets transparently */
3882 bool fHaveInitiatorSecretEncrypted = false;
3883 Utf8Str strCiphertext;
3884 settings::StringsMap::const_iterator itPln = m->mapProperties.find("InitiatorSecret");
3885 if ( itPln != m->mapProperties.end()
3886 && !itPln->second.isEmpty())
3887 {
3888 /* Encrypt the plain secret. If that does not work (i.e. no or wrong settings key
3889 * specified), just use the encrypted secret (if there is any). */
3890 int rc = m->pVirtualBox->encryptSetting(itPln->second, &strCiphertext);
3891 if (RT_SUCCESS(rc))
3892 fHaveInitiatorSecretEncrypted = true;
3893 }
3894 for (settings::StringsMap::const_iterator it = m->mapProperties.begin();
3895 it != m->mapProperties.end();
3896 ++it)
3897 {
3898 /* only save properties that have non-default values */
3899 if (!it->second.isEmpty())
3900 {
3901 const Utf8Str &name = it->first;
3902 const Utf8Str &value = it->second;
3903 /* do NOT store the plain InitiatorSecret */
3904 if ( !fHaveInitiatorSecretEncrypted
3905 || !name.equals("InitiatorSecret"))
3906 data.properties[name] = value;
3907 }
3908 }
3909 if (fHaveInitiatorSecretEncrypted)
3910 data.properties["InitiatorSecretEncrypted"] = strCiphertext;
3911
3912 /* only for base media */
3913 if (m->pParent.isNull())
3914 data.hdType = m->type;
3915
3916 /* save all children */
3917 for (MediaList::const_iterator it = getChildren().begin();
3918 it != getChildren().end();
3919 ++it)
3920 {
3921 settings::Medium med;
3922 HRESULT rc = (*it)->saveSettings(med, strHardDiskFolder);
3923 AssertComRCReturnRC(rc);
3924 data.llChildren.push_back(med);
3925 }
3926
3927 return S_OK;
3928}
3929
3930/**
3931 * Constructs a medium lock list for this medium. The lock is not taken.
3932 *
3933 * @note Caller MUST NOT hold the media tree or medium lock.
3934 *
3935 * @param fFailIfInaccessible If true, this fails with an error if a medium is inaccessible. If false,
3936 * inaccessible media are silently skipped and not locked (i.e. their state remains "Inaccessible");
3937 * this is necessary for a VM's removable media VM startup for which we do not want to fail.
3938 * @param fMediumLockWrite Whether to associate a write lock with this medium.
3939 * @param pToBeParent Medium which will become the parent of this medium.
3940 * @param mediumLockList Where to store the resulting list.
3941 */
3942HRESULT Medium::createMediumLockList(bool fFailIfInaccessible,
3943 bool fMediumLockWrite,
3944 Medium *pToBeParent,
3945 MediumLockList &mediumLockList)
3946{
3947 Assert(!m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
3948 Assert(!isWriteLockOnCurrentThread());
3949
3950 AutoCaller autoCaller(this);
3951 if (FAILED(autoCaller.rc())) return autoCaller.rc();
3952
3953 HRESULT rc = S_OK;
3954
3955 /* paranoid sanity checking if the medium has a to-be parent medium */
3956 if (pToBeParent)
3957 {
3958 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
3959 ComAssertRet(getParent().isNull(), E_FAIL);
3960 ComAssertRet(getChildren().size() == 0, E_FAIL);
3961 }
3962
3963 ErrorInfoKeeper eik;
3964 MultiResult mrc(S_OK);
3965
3966 ComObjPtr<Medium> pMedium = this;
3967 while (!pMedium.isNull())
3968 {
3969 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
3970
3971 /* Accessibility check must be first, otherwise locking interferes
3972 * with getting the medium state. Lock lists are not created for
3973 * fun, and thus getting the medium status is no luxury. */
3974 MediumState_T mediumState = pMedium->getState();
3975 if (mediumState == MediumState_Inaccessible)
3976 {
3977 alock.release();
3978 rc = pMedium->queryInfo(false /* fSetImageId */, false /* fSetParentId */);
3979 alock.acquire();
3980 if (FAILED(rc)) return rc;
3981
3982 mediumState = pMedium->getState();
3983 if (mediumState == MediumState_Inaccessible)
3984 {
3985 // ignore inaccessible ISO media and silently return S_OK,
3986 // otherwise VM startup (esp. restore) may fail without good reason
3987 if (!fFailIfInaccessible)
3988 return S_OK;
3989
3990 // otherwise report an error
3991 Bstr error;
3992 rc = pMedium->COMGETTER(LastAccessError)(error.asOutParam());
3993 if (FAILED(rc)) return rc;
3994
3995 /* collect multiple errors */
3996 eik.restore();
3997 Assert(!error.isEmpty());
3998 mrc = setError(E_FAIL,
3999 "%ls",
4000 error.raw());
4001 // error message will be something like
4002 // "Could not open the medium ... VD: error VERR_FILE_NOT_FOUND opening image file ... (VERR_FILE_NOT_FOUND).
4003 eik.fetch();
4004 }
4005 }
4006
4007 if (pMedium == this)
4008 mediumLockList.Prepend(pMedium, fMediumLockWrite);
4009 else
4010 mediumLockList.Prepend(pMedium, false);
4011
4012 pMedium = pMedium->getParent();
4013 if (pMedium.isNull() && pToBeParent)
4014 {
4015 pMedium = pToBeParent;
4016 pToBeParent = NULL;
4017 }
4018 }
4019
4020 return mrc;
4021}
4022
4023/**
4024 * Creates a new differencing storage unit using the format of the given target
4025 * medium and the location. Note that @c aTarget must be NotCreated.
4026 *
4027 * The @a aMediumLockList parameter contains the associated medium lock list,
4028 * which must be in locked state. If @a aWait is @c true then the caller is
4029 * responsible for unlocking.
4030 *
4031 * If @a aProgress is not NULL but the object it points to is @c null then a
4032 * new progress object will be created and assigned to @a *aProgress on
4033 * success, otherwise the existing progress object is used. If @a aProgress is
4034 * NULL, then no progress object is created/used at all.
4035 *
4036 * When @a aWait is @c false, this method will create a thread to perform the
4037 * create operation asynchronously and will return immediately. Otherwise, it
4038 * will perform the operation on the calling thread and will not return to the
4039 * caller until the operation is completed. Note that @a aProgress cannot be
4040 * NULL when @a aWait is @c false (this method will assert in this case).
4041 *
4042 * @param aTarget Target medium.
4043 * @param aVariant Precise medium variant to create.
4044 * @param aMediumLockList List of media which should be locked.
4045 * @param aProgress Where to find/store a Progress object to track
4046 * operation completion.
4047 * @param aWait @c true if this method should block instead of
4048 * creating an asynchronous thread.
4049 *
4050 * @note Locks this object and @a aTarget for writing.
4051 */
4052HRESULT Medium::createDiffStorage(ComObjPtr<Medium> &aTarget,
4053 MediumVariant_T aVariant,
4054 MediumLockList *aMediumLockList,
4055 ComObjPtr<Progress> *aProgress,
4056 bool aWait)
4057{
4058 AssertReturn(!aTarget.isNull(), E_FAIL);
4059 AssertReturn(aMediumLockList, E_FAIL);
4060 AssertReturn(aProgress != NULL || aWait == true, E_FAIL);
4061
4062 AutoCaller autoCaller(this);
4063 if (FAILED(autoCaller.rc())) return autoCaller.rc();
4064
4065 AutoCaller targetCaller(aTarget);
4066 if (FAILED(targetCaller.rc())) return targetCaller.rc();
4067
4068 HRESULT rc = S_OK;
4069 ComObjPtr<Progress> pProgress;
4070 Medium::Task *pTask = NULL;
4071
4072 try
4073 {
4074 AutoMultiWriteLock2 alock(this, aTarget COMMA_LOCKVAL_SRC_POS);
4075
4076 ComAssertThrow( m->type != MediumType_Writethrough
4077 && m->type != MediumType_Shareable
4078 && m->type != MediumType_Readonly, E_FAIL);
4079 ComAssertThrow(m->state == MediumState_LockedRead, E_FAIL);
4080
4081 if (aTarget->m->state != MediumState_NotCreated)
4082 throw aTarget->setStateError();
4083
4084 /* Check that the medium is not attached to the current state of
4085 * any VM referring to it. */
4086 for (BackRefList::const_iterator it = m->backRefs.begin();
4087 it != m->backRefs.end();
4088 ++it)
4089 {
4090 if (it->fInCurState)
4091 {
4092 /* Note: when a VM snapshot is being taken, all normal media
4093 * attached to the VM in the current state will be, as an
4094 * exception, also associated with the snapshot which is about
4095 * to create (see SnapshotMachine::init()) before deassociating
4096 * them from the current state (which takes place only on
4097 * success in Machine::fixupHardDisks()), so that the size of
4098 * snapshotIds will be 1 in this case. The extra condition is
4099 * used to filter out this legal situation. */
4100 if (it->llSnapshotIds.size() == 0)
4101 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4102 tr("Medium '%s' is attached to a virtual machine with UUID {%RTuuid}. No differencing media based on it may be created until it is detached"),
4103 m->strLocationFull.c_str(), it->machineId.raw());
4104
4105 Assert(it->llSnapshotIds.size() == 1);
4106 }
4107 }
4108
4109 if (aProgress != NULL)
4110 {
4111 /* use the existing progress object... */
4112 pProgress = *aProgress;
4113
4114 /* ...but create a new one if it is null */
4115 if (pProgress.isNull())
4116 {
4117 pProgress.createObject();
4118 rc = pProgress->init(m->pVirtualBox,
4119 static_cast<IMedium*>(this),
4120 BstrFmt(tr("Creating differencing medium storage unit '%s'"), aTarget->m->strLocationFull.c_str()).raw(),
4121 TRUE /* aCancelable */);
4122 if (FAILED(rc))
4123 throw rc;
4124 }
4125 }
4126
4127 /* setup task object to carry out the operation sync/async */
4128 pTask = new Medium::CreateDiffTask(this, pProgress, aTarget, aVariant,
4129 aMediumLockList,
4130 aWait /* fKeepMediumLockList */);
4131 rc = pTask->rc();
4132 AssertComRC(rc);
4133 if (FAILED(rc))
4134 throw rc;
4135
4136 /* register a task (it will deregister itself when done) */
4137 ++m->numCreateDiffTasks;
4138 Assert(m->numCreateDiffTasks != 0); /* overflow? */
4139
4140 aTarget->m->state = MediumState_Creating;
4141 }
4142 catch (HRESULT aRC) { rc = aRC; }
4143
4144 if (SUCCEEDED(rc))
4145 {
4146 if (aWait)
4147 rc = runNow(pTask);
4148 else
4149 rc = startThread(pTask);
4150
4151 if (SUCCEEDED(rc) && aProgress != NULL)
4152 *aProgress = pProgress;
4153 }
4154 else if (pTask != NULL)
4155 delete pTask;
4156
4157 return rc;
4158}
4159
4160/**
4161 * Returns a preferred format for differencing media.
4162 */
4163Utf8Str Medium::getPreferredDiffFormat()
4164{
4165 AutoCaller autoCaller(this);
4166 AssertComRCReturn(autoCaller.rc(), Utf8Str::Empty);
4167
4168 /* check that our own format supports diffs */
4169 if (!(m->formatObj->i_getCapabilities() & MediumFormatCapabilities_Differencing))
4170 {
4171 /* use the default format if not */
4172 Utf8Str tmp;
4173 m->pVirtualBox->getDefaultHardDiskFormat(tmp);
4174 return tmp;
4175 }
4176
4177 /* m->strFormat is const, no need to lock */
4178 return m->strFormat;
4179}
4180
4181/**
4182 * Implementation for the public Medium::Close() with the exception of calling
4183 * VirtualBox::saveRegistries(), in case someone wants to call this for several
4184 * media.
4185 *
4186 * After this returns with success, uninit() has been called on the medium, and
4187 * the object is no longer usable ("not ready" state).
4188 *
4189 * @param autoCaller AutoCaller instance which must have been created on the caller's stack for this medium. This gets released here
4190 * upon which the Medium instance gets uninitialized.
4191 * @return
4192 */
4193HRESULT Medium::close(AutoCaller &autoCaller)
4194{
4195 // we're accessing parent/child and backrefs, so lock the tree first, then ourselves
4196 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
4197 this->lockHandle()
4198 COMMA_LOCKVAL_SRC_POS);
4199
4200 LogFlowFunc(("ENTER for %s\n", getLocationFull().c_str()));
4201
4202 bool wasCreated = true;
4203
4204 switch (m->state)
4205 {
4206 case MediumState_NotCreated:
4207 wasCreated = false;
4208 break;
4209 case MediumState_Created:
4210 case MediumState_Inaccessible:
4211 break;
4212 default:
4213 return setStateError();
4214 }
4215
4216 if (m->backRefs.size() != 0)
4217 return setError(VBOX_E_OBJECT_IN_USE,
4218 tr("Medium '%s' cannot be closed because it is still attached to %d virtual machines"),
4219 m->strLocationFull.c_str(), m->backRefs.size());
4220
4221 // perform extra media-dependent close checks
4222 HRESULT rc = canClose();
4223 if (FAILED(rc)) return rc;
4224
4225 if (wasCreated)
4226 {
4227 // remove from the list of known media before performing actual
4228 // uninitialization (to keep the media registry consistent on
4229 // failure to do so)
4230 rc = unregisterWithVirtualBox();
4231 if (FAILED(rc)) return rc;
4232
4233 multilock.release();
4234 markRegistriesModified();
4235 // Release the AutoCalleri now, as otherwise uninit() will simply hang.
4236 // Needs to be done before saving the registry, as otherwise there
4237 // may be a deadlock with someone else closing this object while we're
4238 // in saveModifiedRegistries(), which needs the media tree lock, which
4239 // the other thread holds until after uninit() below.
4240 /// @todo redesign the locking here, as holding the locks over uninit causes lock order trouble which the lock validator can't detect
4241 autoCaller.release();
4242 m->pVirtualBox->saveModifiedRegistries();
4243 multilock.acquire();
4244 }
4245 else
4246 {
4247 // release the AutoCaller, as otherwise uninit() will simply hang
4248 autoCaller.release();
4249 }
4250
4251 // Keep the locks held until after uninit, as otherwise the consistency
4252 // of the medium tree cannot be guaranteed.
4253 uninit();
4254
4255 LogFlowFuncLeave();
4256
4257 return rc;
4258}
4259
4260/**
4261 * Deletes the medium storage unit.
4262 *
4263 * If @a aProgress is not NULL but the object it points to is @c null then a new
4264 * progress object will be created and assigned to @a *aProgress on success,
4265 * otherwise the existing progress object is used. If Progress is NULL, then no
4266 * progress object is created/used at all.
4267 *
4268 * When @a aWait is @c false, this method will create a thread to perform the
4269 * delete operation asynchronously and will return immediately. Otherwise, it
4270 * will perform the operation on the calling thread and will not return to the
4271 * caller until the operation is completed. Note that @a aProgress cannot be
4272 * NULL when @a aWait is @c false (this method will assert in this case).
4273 *
4274 * @param aProgress Where to find/store a Progress object to track operation
4275 * completion.
4276 * @param aWait @c true if this method should block instead of creating
4277 * an asynchronous thread.
4278 *
4279 * @note Locks mVirtualBox and this object for writing. Locks medium tree for
4280 * writing.
4281 */
4282HRESULT Medium::deleteStorage(ComObjPtr<Progress> *aProgress,
4283 bool aWait)
4284{
4285 AssertReturn(aProgress != NULL || aWait == true, E_FAIL);
4286
4287 AutoCaller autoCaller(this);
4288 if (FAILED(autoCaller.rc())) return autoCaller.rc();
4289
4290 HRESULT rc = S_OK;
4291 ComObjPtr<Progress> pProgress;
4292 Medium::Task *pTask = NULL;
4293
4294 try
4295 {
4296 /* we're accessing the media tree, and canClose() needs it too */
4297 AutoMultiWriteLock2 multilock(&m->pVirtualBox->getMediaTreeLockHandle(),
4298 this->lockHandle()
4299 COMMA_LOCKVAL_SRC_POS);
4300 LogFlowThisFunc(("aWait=%RTbool locationFull=%s\n", aWait, getLocationFull().c_str() ));
4301
4302 if ( !(m->formatObj->i_getCapabilities() & ( MediumFormatCapabilities_CreateDynamic
4303 | MediumFormatCapabilities_CreateFixed)))
4304 throw setError(VBOX_E_NOT_SUPPORTED,
4305 tr("Medium format '%s' does not support storage deletion"),
4306 m->strFormat.c_str());
4307
4308 /* Note that we are fine with Inaccessible state too: a) for symmetry
4309 * with create calls and b) because it doesn't really harm to try, if
4310 * it is really inaccessible, the delete operation will fail anyway.
4311 * Accepting Inaccessible state is especially important because all
4312 * registered media are initially Inaccessible upon VBoxSVC startup
4313 * until COMGETTER(RefreshState) is called. Accept Deleting state
4314 * because some callers need to put the medium in this state early
4315 * to prevent races. */
4316 switch (m->state)
4317 {
4318 case MediumState_Created:
4319 case MediumState_Deleting:
4320 case MediumState_Inaccessible:
4321 break;
4322 default:
4323 throw setStateError();
4324 }
4325
4326 if (m->backRefs.size() != 0)
4327 {
4328 Utf8Str strMachines;
4329 for (BackRefList::const_iterator it = m->backRefs.begin();
4330 it != m->backRefs.end();
4331 ++it)
4332 {
4333 const BackRef &b = *it;
4334 if (strMachines.length())
4335 strMachines.append(", ");
4336 strMachines.append(b.machineId.toString().c_str());
4337 }
4338#ifdef DEBUG
4339 dumpBackRefs();
4340#endif
4341 throw setError(VBOX_E_OBJECT_IN_USE,
4342 tr("Cannot delete storage: medium '%s' is still attached to the following %d virtual machine(s): %s"),
4343 m->strLocationFull.c_str(),
4344 m->backRefs.size(),
4345 strMachines.c_str());
4346 }
4347
4348 rc = canClose();
4349 if (FAILED(rc))
4350 throw rc;
4351
4352 /* go to Deleting state, so that the medium is not actually locked */
4353 if (m->state != MediumState_Deleting)
4354 {
4355 rc = markForDeletion();
4356 if (FAILED(rc))
4357 throw rc;
4358 }
4359
4360 /* Build the medium lock list. */
4361 MediumLockList *pMediumLockList(new MediumLockList());
4362 multilock.release();
4363 rc = createMediumLockList(true /* fFailIfInaccessible */,
4364 true /* fMediumLockWrite */,
4365 NULL,
4366 *pMediumLockList);
4367 multilock.acquire();
4368 if (FAILED(rc))
4369 {
4370 delete pMediumLockList;
4371 throw rc;
4372 }
4373
4374 multilock.release();
4375 rc = pMediumLockList->Lock();
4376 multilock.acquire();
4377 if (FAILED(rc))
4378 {
4379 delete pMediumLockList;
4380 throw setError(rc,
4381 tr("Failed to lock media when deleting '%s'"),
4382 getLocationFull().c_str());
4383 }
4384
4385 /* try to remove from the list of known media before performing
4386 * actual deletion (we favor the consistency of the media registry
4387 * which would have been broken if unregisterWithVirtualBox() failed
4388 * after we successfully deleted the storage) */
4389 rc = unregisterWithVirtualBox();
4390 if (FAILED(rc))
4391 throw rc;
4392 // no longer need lock
4393 multilock.release();
4394 markRegistriesModified();
4395
4396 if (aProgress != NULL)
4397 {
4398 /* use the existing progress object... */
4399 pProgress = *aProgress;
4400
4401 /* ...but create a new one if it is null */
4402 if (pProgress.isNull())
4403 {
4404 pProgress.createObject();
4405 rc = pProgress->init(m->pVirtualBox,
4406 static_cast<IMedium*>(this),
4407 BstrFmt(tr("Deleting medium storage unit '%s'"), m->strLocationFull.c_str()).raw(),
4408 FALSE /* aCancelable */);
4409 if (FAILED(rc))
4410 throw rc;
4411 }
4412 }
4413
4414 /* setup task object to carry out the operation sync/async */
4415 pTask = new Medium::DeleteTask(this, pProgress, pMediumLockList);
4416 rc = pTask->rc();
4417 AssertComRC(rc);
4418 if (FAILED(rc))
4419 throw rc;
4420 }
4421 catch (HRESULT aRC) { rc = aRC; }
4422
4423 if (SUCCEEDED(rc))
4424 {
4425 if (aWait)
4426 rc = runNow(pTask);
4427 else
4428 rc = startThread(pTask);
4429
4430 if (SUCCEEDED(rc) && aProgress != NULL)
4431 *aProgress = pProgress;
4432
4433 }
4434 else
4435 {
4436 if (pTask)
4437 delete pTask;
4438
4439 /* Undo deleting state if necessary. */
4440 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
4441 /* Make sure that any error signalled by unmarkForDeletion() is not
4442 * ending up in the error list (if the caller uses MultiResult). It
4443 * usually is spurious, as in most cases the medium hasn't been marked
4444 * for deletion when the error was thrown above. */
4445 ErrorInfoKeeper eik;
4446 unmarkForDeletion();
4447 }
4448
4449 return rc;
4450}
4451
4452/**
4453 * Mark a medium for deletion.
4454 *
4455 * @note Caller must hold the write lock on this medium!
4456 */
4457HRESULT Medium::markForDeletion()
4458{
4459 ComAssertRet(isWriteLockOnCurrentThread(), E_FAIL);
4460 switch (m->state)
4461 {
4462 case MediumState_Created:
4463 case MediumState_Inaccessible:
4464 m->preLockState = m->state;
4465 m->state = MediumState_Deleting;
4466 return S_OK;
4467 default:
4468 return setStateError();
4469 }
4470}
4471
4472/**
4473 * Removes the "mark for deletion".
4474 *
4475 * @note Caller must hold the write lock on this medium!
4476 */
4477HRESULT Medium::unmarkForDeletion()
4478{
4479 ComAssertRet(isWriteLockOnCurrentThread(), E_FAIL);
4480 switch (m->state)
4481 {
4482 case MediumState_Deleting:
4483 m->state = m->preLockState;
4484 return S_OK;
4485 default:
4486 return setStateError();
4487 }
4488}
4489
4490/**
4491 * Mark a medium for deletion which is in locked state.
4492 *
4493 * @note Caller must hold the write lock on this medium!
4494 */
4495HRESULT Medium::markLockedForDeletion()
4496{
4497 ComAssertRet(isWriteLockOnCurrentThread(), E_FAIL);
4498 if ( ( m->state == MediumState_LockedRead
4499 || m->state == MediumState_LockedWrite)
4500 && m->preLockState == MediumState_Created)
4501 {
4502 m->preLockState = MediumState_Deleting;
4503 return S_OK;
4504 }
4505 else
4506 return setStateError();
4507}
4508
4509/**
4510 * Removes the "mark for deletion" for a medium in locked state.
4511 *
4512 * @note Caller must hold the write lock on this medium!
4513 */
4514HRESULT Medium::unmarkLockedForDeletion()
4515{
4516 ComAssertRet(isWriteLockOnCurrentThread(), E_FAIL);
4517 if ( ( m->state == MediumState_LockedRead
4518 || m->state == MediumState_LockedWrite)
4519 && m->preLockState == MediumState_Deleting)
4520 {
4521 m->preLockState = MediumState_Created;
4522 return S_OK;
4523 }
4524 else
4525 return setStateError();
4526}
4527
4528/**
4529 * Queries the preferred merge direction from this to the other medium, i.e.
4530 * the one which requires the least amount of I/O and therefore time and
4531 * disk consumption.
4532 *
4533 * @returns Status code.
4534 * @retval E_FAIL in case determining the merge direction fails for some reason,
4535 * for example if getting the size of the media fails. There is no
4536 * error set though and the caller is free to continue to find out
4537 * what was going wrong later. Leaves fMergeForward unset.
4538 * @retval VBOX_E_INVALID_OBJECT_STATE if both media are not related to each other
4539 * An error is set.
4540 * @param pOther The other medium to merge with.
4541 * @param fMergeForward Resulting preferred merge direction (out).
4542 */
4543HRESULT Medium::queryPreferredMergeDirection(const ComObjPtr<Medium> &pOther,
4544 bool &fMergeForward)
4545{
4546 AssertReturn(pOther != NULL, E_FAIL);
4547 AssertReturn(pOther != this, E_FAIL);
4548
4549 AutoCaller autoCaller(this);
4550 AssertComRCReturnRC(autoCaller.rc());
4551
4552 AutoCaller otherCaller(pOther);
4553 AssertComRCReturnRC(otherCaller.rc());
4554
4555 HRESULT rc = S_OK;
4556 bool fThisParent = false; /**<< Flag whether this medium is the parent of pOther. */
4557
4558 try
4559 {
4560 // locking: we need the tree lock first because we access parent pointers
4561 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
4562
4563 /* more sanity checking and figuring out the current merge direction */
4564 ComObjPtr<Medium> pMedium = getParent();
4565 while (!pMedium.isNull() && pMedium != pOther)
4566 pMedium = pMedium->getParent();
4567 if (pMedium == pOther)
4568 fThisParent = false;
4569 else
4570 {
4571 pMedium = pOther->getParent();
4572 while (!pMedium.isNull() && pMedium != this)
4573 pMedium = pMedium->getParent();
4574 if (pMedium == this)
4575 fThisParent = true;
4576 else
4577 {
4578 Utf8Str tgtLoc;
4579 {
4580 AutoReadLock alock(pOther COMMA_LOCKVAL_SRC_POS);
4581 tgtLoc = pOther->getLocationFull();
4582 }
4583
4584 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4585 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4586 tr("Media '%s' and '%s' are unrelated"),
4587 m->strLocationFull.c_str(), tgtLoc.c_str());
4588 }
4589 }
4590
4591 /*
4592 * Figure out the preferred merge direction. The current way is to
4593 * get the current sizes of file based images and select the merge
4594 * direction depending on the size.
4595 *
4596 * Can't use the VD API to get current size here as the media might
4597 * be write locked by a running VM. Resort to RTFileQuerySize().
4598 */
4599 int vrc = VINF_SUCCESS;
4600 uint64_t cbMediumThis = 0;
4601 uint64_t cbMediumOther = 0;
4602
4603 if (isMediumFormatFile() && pOther->isMediumFormatFile())
4604 {
4605 vrc = RTFileQuerySize(this->getLocationFull().c_str(), &cbMediumThis);
4606 if (RT_SUCCESS(vrc))
4607 {
4608 vrc = RTFileQuerySize(pOther->getLocationFull().c_str(),
4609 &cbMediumOther);
4610 }
4611
4612 if (RT_FAILURE(vrc))
4613 rc = E_FAIL;
4614 else
4615 {
4616 /*
4617 * Check which merge direction might be more optimal.
4618 * This method is not bullet proof of course as there might
4619 * be overlapping blocks in the images so the file size is
4620 * not the best indicator but it is good enough for our purpose
4621 * and everything else is too complicated, especially when the
4622 * media are used by a running VM.
4623 */
4624 bool fMergeIntoThis = cbMediumThis > cbMediumOther;
4625 fMergeForward = fMergeIntoThis ^ fThisParent;
4626 }
4627 }
4628 }
4629 catch (HRESULT aRC) { rc = aRC; }
4630
4631 return rc;
4632}
4633
4634/**
4635 * Prepares this (source) medium, target medium and all intermediate media
4636 * for the merge operation.
4637 *
4638 * This method is to be called prior to calling the #mergeTo() to perform
4639 * necessary consistency checks and place involved media to appropriate
4640 * states. If #mergeTo() is not called or fails, the state modifications
4641 * performed by this method must be undone by #cancelMergeTo().
4642 *
4643 * See #mergeTo() for more information about merging.
4644 *
4645 * @param pTarget Target medium.
4646 * @param aMachineId Allowed machine attachment. NULL means do not check.
4647 * @param aSnapshotId Allowed snapshot attachment. NULL or empty UUID means
4648 * do not check.
4649 * @param fLockMedia Flag whether to lock the medium lock list or not.
4650 * If set to false and the medium lock list locking fails
4651 * later you must call #cancelMergeTo().
4652 * @param fMergeForward Resulting merge direction (out).
4653 * @param pParentForTarget New parent for target medium after merge (out).
4654 * @param aChildrenToReparent Medium lock list containing all children of the
4655 * source which will have to be reparented to the target
4656 * after merge (out).
4657 * @param aMediumLockList Medium locking information (out).
4658 *
4659 * @note Locks medium tree for reading. Locks this object, aTarget and all
4660 * intermediate media for writing.
4661 */
4662HRESULT Medium::prepareMergeTo(const ComObjPtr<Medium> &pTarget,
4663 const Guid *aMachineId,
4664 const Guid *aSnapshotId,
4665 bool fLockMedia,
4666 bool &fMergeForward,
4667 ComObjPtr<Medium> &pParentForTarget,
4668 MediumLockList * &aChildrenToReparent,
4669 MediumLockList * &aMediumLockList)
4670{
4671 AssertReturn(pTarget != NULL, E_FAIL);
4672 AssertReturn(pTarget != this, E_FAIL);
4673
4674 AutoCaller autoCaller(this);
4675 AssertComRCReturnRC(autoCaller.rc());
4676
4677 AutoCaller targetCaller(pTarget);
4678 AssertComRCReturnRC(targetCaller.rc());
4679
4680 HRESULT rc = S_OK;
4681 fMergeForward = false;
4682 pParentForTarget.setNull();
4683 Assert(aChildrenToReparent == NULL);
4684 aChildrenToReparent = NULL;
4685 Assert(aMediumLockList == NULL);
4686 aMediumLockList = NULL;
4687
4688 try
4689 {
4690 // locking: we need the tree lock first because we access parent pointers
4691 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
4692
4693 /* more sanity checking and figuring out the merge direction */
4694 ComObjPtr<Medium> pMedium = getParent();
4695 while (!pMedium.isNull() && pMedium != pTarget)
4696 pMedium = pMedium->getParent();
4697 if (pMedium == pTarget)
4698 fMergeForward = false;
4699 else
4700 {
4701 pMedium = pTarget->getParent();
4702 while (!pMedium.isNull() && pMedium != this)
4703 pMedium = pMedium->getParent();
4704 if (pMedium == this)
4705 fMergeForward = true;
4706 else
4707 {
4708 Utf8Str tgtLoc;
4709 {
4710 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4711 tgtLoc = pTarget->getLocationFull();
4712 }
4713
4714 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4715 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4716 tr("Media '%s' and '%s' are unrelated"),
4717 m->strLocationFull.c_str(), tgtLoc.c_str());
4718 }
4719 }
4720
4721 /* Build the lock list. */
4722 aMediumLockList = new MediumLockList();
4723 treeLock.release();
4724 if (fMergeForward)
4725 rc = pTarget->createMediumLockList(true /* fFailIfInaccessible */,
4726 true /* fMediumLockWrite */,
4727 NULL,
4728 *aMediumLockList);
4729 else
4730 rc = createMediumLockList(true /* fFailIfInaccessible */,
4731 false /* fMediumLockWrite */,
4732 NULL,
4733 *aMediumLockList);
4734 treeLock.acquire();
4735 if (FAILED(rc))
4736 throw rc;
4737
4738 /* Sanity checking, must be after lock list creation as it depends on
4739 * valid medium states. The medium objects must be accessible. Only
4740 * do this if immediate locking is requested, otherwise it fails when
4741 * we construct a medium lock list for an already running VM. Snapshot
4742 * deletion uses this to simplify its life. */
4743 if (fLockMedia)
4744 {
4745 {
4746 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4747 if (m->state != MediumState_Created)
4748 throw setStateError();
4749 }
4750 {
4751 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4752 if (pTarget->m->state != MediumState_Created)
4753 throw pTarget->setStateError();
4754 }
4755 }
4756
4757 /* check medium attachment and other sanity conditions */
4758 if (fMergeForward)
4759 {
4760 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
4761 if (getChildren().size() > 1)
4762 {
4763 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4764 tr("Medium '%s' involved in the merge operation has more than one child medium (%d)"),
4765 m->strLocationFull.c_str(), getChildren().size());
4766 }
4767 /* One backreference is only allowed if the machine ID is not empty
4768 * and it matches the machine the medium is attached to (including
4769 * the snapshot ID if not empty). */
4770 if ( m->backRefs.size() != 0
4771 && ( !aMachineId
4772 || m->backRefs.size() != 1
4773 || aMachineId->isZero()
4774 || *getFirstMachineBackrefId() != *aMachineId
4775 || ( (!aSnapshotId || !aSnapshotId->isZero())
4776 && *getFirstMachineBackrefSnapshotId() != *aSnapshotId)))
4777 throw setError(VBOX_E_OBJECT_IN_USE,
4778 tr("Medium '%s' is attached to %d virtual machines"),
4779 m->strLocationFull.c_str(), m->backRefs.size());
4780 if (m->type == MediumType_Immutable)
4781 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4782 tr("Medium '%s' is immutable"),
4783 m->strLocationFull.c_str());
4784 if (m->type == MediumType_MultiAttach)
4785 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4786 tr("Medium '%s' is multi-attach"),
4787 m->strLocationFull.c_str());
4788 }
4789 else
4790 {
4791 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4792 if (pTarget->getChildren().size() > 1)
4793 {
4794 throw setError(VBOX_E_OBJECT_IN_USE,
4795 tr("Medium '%s' involved in the merge operation has more than one child medium (%d)"),
4796 pTarget->m->strLocationFull.c_str(),
4797 pTarget->getChildren().size());
4798 }
4799 if (pTarget->m->type == MediumType_Immutable)
4800 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4801 tr("Medium '%s' is immutable"),
4802 pTarget->m->strLocationFull.c_str());
4803 if (pTarget->m->type == MediumType_MultiAttach)
4804 throw setError(VBOX_E_INVALID_OBJECT_STATE,
4805 tr("Medium '%s' is multi-attach"),
4806 pTarget->m->strLocationFull.c_str());
4807 }
4808 ComObjPtr<Medium> pLast(fMergeForward ? (Medium *)pTarget : this);
4809 ComObjPtr<Medium> pLastIntermediate = pLast->getParent();
4810 for (pLast = pLastIntermediate;
4811 !pLast.isNull() && pLast != pTarget && pLast != this;
4812 pLast = pLast->getParent())
4813 {
4814 AutoReadLock alock(pLast COMMA_LOCKVAL_SRC_POS);
4815 if (pLast->getChildren().size() > 1)
4816 {
4817 throw setError(VBOX_E_OBJECT_IN_USE,
4818 tr("Medium '%s' involved in the merge operation has more than one child medium (%d)"),
4819 pLast->m->strLocationFull.c_str(),
4820 pLast->getChildren().size());
4821 }
4822 if (pLast->m->backRefs.size() != 0)
4823 throw setError(VBOX_E_OBJECT_IN_USE,
4824 tr("Medium '%s' is attached to %d virtual machines"),
4825 pLast->m->strLocationFull.c_str(),
4826 pLast->m->backRefs.size());
4827
4828 }
4829
4830 /* Update medium states appropriately */
4831 {
4832 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
4833
4834 if (m->state == MediumState_Created)
4835 {
4836 rc = markForDeletion();
4837 if (FAILED(rc))
4838 throw rc;
4839 }
4840 else
4841 {
4842 if (fLockMedia)
4843 throw setStateError();
4844 else if ( m->state == MediumState_LockedWrite
4845 || m->state == MediumState_LockedRead)
4846 {
4847 /* Either mark it for deletion in locked state or allow
4848 * others to have done so. */
4849 if (m->preLockState == MediumState_Created)
4850 markLockedForDeletion();
4851 else if (m->preLockState != MediumState_Deleting)
4852 throw setStateError();
4853 }
4854 else
4855 throw setStateError();
4856 }
4857 }
4858
4859 if (fMergeForward)
4860 {
4861 /* we will need parent to reparent target */
4862 pParentForTarget = getParent();
4863 }
4864 else
4865 {
4866 /* we will need to reparent children of the source */
4867 aChildrenToReparent = new MediumLockList();
4868 for (MediaList::const_iterator it = getChildren().begin();
4869 it != getChildren().end();
4870 ++it)
4871 {
4872 pMedium = *it;
4873 aChildrenToReparent->Append(pMedium, true /* fLockWrite */);
4874 }
4875 if (fLockMedia && aChildrenToReparent)
4876 {
4877 treeLock.release();
4878 rc = aChildrenToReparent->Lock();
4879 treeLock.acquire();
4880 if (FAILED(rc))
4881 throw rc;
4882 }
4883 }
4884 for (pLast = pLastIntermediate;
4885 !pLast.isNull() && pLast != pTarget && pLast != this;
4886 pLast = pLast->getParent())
4887 {
4888 AutoWriteLock alock(pLast COMMA_LOCKVAL_SRC_POS);
4889 if (pLast->m->state == MediumState_Created)
4890 {
4891 rc = pLast->markForDeletion();
4892 if (FAILED(rc))
4893 throw rc;
4894 }
4895 else
4896 throw pLast->setStateError();
4897 }
4898
4899 /* Tweak the lock list in the backward merge case, as the target
4900 * isn't marked to be locked for writing yet. */
4901 if (!fMergeForward)
4902 {
4903 MediumLockList::Base::iterator lockListBegin =
4904 aMediumLockList->GetBegin();
4905 MediumLockList::Base::iterator lockListEnd =
4906 aMediumLockList->GetEnd();
4907 lockListEnd--;
4908 for (MediumLockList::Base::iterator it = lockListBegin;
4909 it != lockListEnd;
4910 ++it)
4911 {
4912 MediumLock &mediumLock = *it;
4913 if (mediumLock.GetMedium() == pTarget)
4914 {
4915 HRESULT rc2 = mediumLock.UpdateLock(true);
4916 AssertComRC(rc2);
4917 break;
4918 }
4919 }
4920 }
4921
4922 if (fLockMedia)
4923 {
4924 treeLock.release();
4925 rc = aMediumLockList->Lock();
4926 treeLock.acquire();
4927 if (FAILED(rc))
4928 {
4929 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
4930 throw setError(rc,
4931 tr("Failed to lock media when merging to '%s'"),
4932 pTarget->getLocationFull().c_str());
4933 }
4934 }
4935 }
4936 catch (HRESULT aRC) { rc = aRC; }
4937
4938 if (FAILED(rc))
4939 {
4940 if (aMediumLockList)
4941 {
4942 delete aMediumLockList;
4943 aMediumLockList = NULL;
4944 }
4945 if (aChildrenToReparent)
4946 {
4947 delete aChildrenToReparent;
4948 aChildrenToReparent = NULL;
4949 }
4950 }
4951
4952 return rc;
4953}
4954
4955/**
4956 * Merges this medium to the specified medium which must be either its
4957 * direct ancestor or descendant.
4958 *
4959 * Given this medium is SOURCE and the specified medium is TARGET, we will
4960 * get two variants of the merge operation:
4961 *
4962 * forward merge
4963 * ------------------------->
4964 * [Extra] <- SOURCE <- Intermediate <- TARGET
4965 * Any Del Del LockWr
4966 *
4967 *
4968 * backward merge
4969 * <-------------------------
4970 * TARGET <- Intermediate <- SOURCE <- [Extra]
4971 * LockWr Del Del LockWr
4972 *
4973 * Each diagram shows the involved media on the media chain where
4974 * SOURCE and TARGET belong. Under each medium there is a state value which
4975 * the medium must have at a time of the mergeTo() call.
4976 *
4977 * The media in the square braces may be absent (e.g. when the forward
4978 * operation takes place and SOURCE is the base medium, or when the backward
4979 * merge operation takes place and TARGET is the last child in the chain) but if
4980 * they present they are involved too as shown.
4981 *
4982 * Neither the source medium nor intermediate media may be attached to
4983 * any VM directly or in the snapshot, otherwise this method will assert.
4984 *
4985 * The #prepareMergeTo() method must be called prior to this method to place all
4986 * involved to necessary states and perform other consistency checks.
4987 *
4988 * If @a aWait is @c true then this method will perform the operation on the
4989 * calling thread and will not return to the caller until the operation is
4990 * completed. When this method succeeds, all intermediate medium objects in
4991 * the chain will be uninitialized, the state of the target medium (and all
4992 * involved extra media) will be restored. @a aMediumLockList will not be
4993 * deleted, whether the operation is successful or not. The caller has to do
4994 * this if appropriate. Note that this (source) medium is not uninitialized
4995 * because of possible AutoCaller instances held by the caller of this method
4996 * on the current thread. It's therefore the responsibility of the caller to
4997 * call Medium::uninit() after releasing all callers.
4998 *
4999 * If @a aWait is @c false then this method will create a thread to perform the
5000 * operation asynchronously and will return immediately. If the operation
5001 * succeeds, the thread will uninitialize the source medium object and all
5002 * intermediate medium objects in the chain, reset the state of the target
5003 * medium (and all involved extra media) and delete @a aMediumLockList.
5004 * If the operation fails, the thread will only reset the states of all
5005 * involved media and delete @a aMediumLockList.
5006 *
5007 * When this method fails (regardless of the @a aWait mode), it is a caller's
5008 * responsibility to undo state changes and delete @a aMediumLockList using
5009 * #cancelMergeTo().
5010 *
5011 * If @a aProgress is not NULL but the object it points to is @c null then a new
5012 * progress object will be created and assigned to @a *aProgress on success,
5013 * otherwise the existing progress object is used. If Progress is NULL, then no
5014 * progress object is created/used at all. Note that @a aProgress cannot be
5015 * NULL when @a aWait is @c false (this method will assert in this case).
5016 *
5017 * @param pTarget Target medium.
5018 * @param fMergeForward Merge direction.
5019 * @param pParentForTarget New parent for target medium after merge.
5020 * @param aChildrenToReparent List of children of the source which will have
5021 * to be reparented to the target after merge.
5022 * @param aMediumLockList Medium locking information.
5023 * @param aProgress Where to find/store a Progress object to track operation
5024 * completion.
5025 * @param aWait @c true if this method should block instead of creating
5026 * an asynchronous thread.
5027 *
5028 * @note Locks the tree lock for writing. Locks the media from the chain
5029 * for writing.
5030 */
5031HRESULT Medium::mergeTo(const ComObjPtr<Medium> &pTarget,
5032 bool fMergeForward,
5033 const ComObjPtr<Medium> &pParentForTarget,
5034 MediumLockList *aChildrenToReparent,
5035 MediumLockList *aMediumLockList,
5036 ComObjPtr<Progress> *aProgress,
5037 bool aWait)
5038{
5039 AssertReturn(pTarget != NULL, E_FAIL);
5040 AssertReturn(pTarget != this, E_FAIL);
5041 AssertReturn(aMediumLockList != NULL, E_FAIL);
5042 AssertReturn(aProgress != NULL || aWait == true, E_FAIL);
5043
5044 AutoCaller autoCaller(this);
5045 if (FAILED(autoCaller.rc())) return autoCaller.rc();
5046
5047 AutoCaller targetCaller(pTarget);
5048 AssertComRCReturnRC(targetCaller.rc());
5049
5050 HRESULT rc = S_OK;
5051 ComObjPtr<Progress> pProgress;
5052 Medium::Task *pTask = NULL;
5053
5054 try
5055 {
5056 if (aProgress != NULL)
5057 {
5058 /* use the existing progress object... */
5059 pProgress = *aProgress;
5060
5061 /* ...but create a new one if it is null */
5062 if (pProgress.isNull())
5063 {
5064 Utf8Str tgtName;
5065 {
5066 AutoReadLock alock(pTarget COMMA_LOCKVAL_SRC_POS);
5067 tgtName = pTarget->getName();
5068 }
5069
5070 AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
5071
5072 pProgress.createObject();
5073 rc = pProgress->init(m->pVirtualBox,
5074 static_cast<IMedium*>(this),
5075 BstrFmt(tr("Merging medium '%s' to '%s'"),
5076 getName().c_str(),
5077 tgtName.c_str()).raw(),
5078 TRUE /* aCancelable */);
5079 if (FAILED(rc))
5080 throw rc;
5081 }
5082 }
5083
5084 /* setup task object to carry out the operation sync/async */
5085 pTask = new Medium::MergeTask(this, pTarget, fMergeForward,
5086 pParentForTarget, aChildrenToReparent,
5087 pProgress, aMediumLockList,
5088 aWait /* fKeepMediumLockList */);
5089 rc = pTask->rc();
5090 AssertComRC(rc);
5091 if (FAILED(rc))
5092 throw rc;
5093 }
5094 catch (HRESULT aRC) { rc = aRC; }
5095
5096 if (SUCCEEDED(rc))
5097 {
5098 if (aWait)
5099 rc = runNow(pTask);
5100 else
5101 rc = startThread(pTask);
5102
5103 if (SUCCEEDED(rc) && aProgress != NULL)
5104 *aProgress = pProgress;
5105 }
5106 else if (pTask != NULL)
5107 delete pTask;
5108
5109 return rc;
5110}
5111
5112/**
5113 * Undoes what #prepareMergeTo() did. Must be called if #mergeTo() is not
5114 * called or fails. Frees memory occupied by @a aMediumLockList and unlocks
5115 * the medium objects in @a aChildrenToReparent.
5116 *
5117 * @param aChildrenToReparent List of children of the source which will have
5118 * to be reparented to the target after merge.
5119 * @param aMediumLockList Medium locking information.
5120 *
5121 * @note Locks the media from the chain for writing.
5122 */
5123void Medium::cancelMergeTo(MediumLockList *aChildrenToReparent,
5124 MediumLockList *aMediumLockList)
5125{
5126 AutoCaller autoCaller(this);
5127 AssertComRCReturnVoid(autoCaller.rc());
5128
5129 AssertReturnVoid(aMediumLockList != NULL);
5130
5131 /* Revert media marked for deletion to previous state. */
5132 HRESULT rc;
5133 MediumLockList::Base::const_iterator mediumListBegin =
5134 aMediumLockList->GetBegin();
5135 MediumLockList::Base::const_iterator mediumListEnd =
5136 aMediumLockList->GetEnd();
5137 for (MediumLockList::Base::const_iterator it = mediumListBegin;
5138 it != mediumListEnd;
5139 ++it)
5140 {
5141 const MediumLock &mediumLock = *it;
5142 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5143 AutoWriteLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5144
5145 if (pMedium->m->state == MediumState_Deleting)
5146 {
5147 rc = pMedium->unmarkForDeletion();
5148 AssertComRC(rc);
5149 }
5150 }
5151
5152 /* the destructor will do the work */
5153 delete aMediumLockList;
5154
5155 /* unlock the children which had to be reparented, the destructor will do
5156 * the work */
5157 if (aChildrenToReparent)
5158 delete aChildrenToReparent;
5159}
5160
5161/**
5162 * Fix the parent UUID of all children to point to this medium as their
5163 * parent.
5164 */
5165HRESULT Medium::fixParentUuidOfChildren(MediumLockList *pChildrenToReparent)
5166{
5167 Assert(!isWriteLockOnCurrentThread());
5168 Assert(!m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
5169 MediumLockList mediumLockList;
5170 HRESULT rc = createMediumLockList(true /* fFailIfInaccessible */,
5171 false /* fMediumLockWrite */,
5172 this,
5173 mediumLockList);
5174 AssertComRCReturnRC(rc);
5175
5176 try
5177 {
5178 PVBOXHDD hdd;
5179 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
5180 ComAssertRCThrow(vrc, E_FAIL);
5181
5182 try
5183 {
5184 MediumLockList::Base::iterator lockListBegin =
5185 mediumLockList.GetBegin();
5186 MediumLockList::Base::iterator lockListEnd =
5187 mediumLockList.GetEnd();
5188 for (MediumLockList::Base::iterator it = lockListBegin;
5189 it != lockListEnd;
5190 ++it)
5191 {
5192 MediumLock &mediumLock = *it;
5193 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
5194 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
5195
5196 // open the medium
5197 vrc = VDOpen(hdd,
5198 pMedium->m->strFormat.c_str(),
5199 pMedium->m->strLocationFull.c_str(),
5200 VD_OPEN_FLAGS_READONLY | m->uOpenFlagsDef,
5201 pMedium->m->vdImageIfaces);
5202 if (RT_FAILURE(vrc))
5203 throw vrc;
5204 }
5205
5206 MediumLockList::Base::iterator childrenBegin = pChildrenToReparent->GetBegin();
5207 MediumLockList::Base::iterator childrenEnd = pChildrenToReparent->GetEnd();
5208 for (MediumLockList::Base::iterator it = childrenBegin;
5209 it != childrenEnd;
5210 ++it)
5211 {
5212 Medium *pMedium = it->GetMedium();
5213 /* VD_OPEN_FLAGS_INFO since UUID is wrong yet */
5214 vrc = VDOpen(hdd,
5215 pMedium->m->strFormat.c_str(),
5216 pMedium->m->strLocationFull.c_str(),
5217 VD_OPEN_FLAGS_INFO | m->uOpenFlagsDef,
5218 pMedium->m->vdImageIfaces);
5219 if (RT_FAILURE(vrc))
5220 throw vrc;
5221
5222 vrc = VDSetParentUuid(hdd, VD_LAST_IMAGE, m->id.raw());
5223 if (RT_FAILURE(vrc))
5224 throw vrc;
5225
5226 vrc = VDClose(hdd, false /* fDelete */);
5227 if (RT_FAILURE(vrc))
5228 throw vrc;
5229 }
5230 }
5231 catch (HRESULT aRC) { rc = aRC; }
5232 catch (int aVRC)
5233 {
5234 rc = setError(E_FAIL,
5235 tr("Could not update medium UUID references to parent '%s' (%s)"),
5236 m->strLocationFull.c_str(),
5237 vdError(aVRC).c_str());
5238 }
5239
5240 VDDestroy(hdd);
5241 }
5242 catch (HRESULT aRC) { rc = aRC; }
5243
5244 return rc;
5245}
5246
5247/**
5248 * Used by IAppliance to export disk images.
5249 *
5250 * @param aFilename Filename to create (UTF8).
5251 * @param aFormat Medium format for creating @a aFilename.
5252 * @param aVariant Which exact image format variant to use
5253 * for the destination image.
5254 * @param aVDImageIOCallbacks Pointer to the callback table for a
5255 * VDINTERFACEIO interface. May be NULL.
5256 * @param aVDImageIOUser Opaque data for the callbacks.
5257 * @param aProgress Progress object to use.
5258 * @return
5259 * @note The source format is defined by the Medium instance.
5260 */
5261HRESULT Medium::exportFile(const char *aFilename,
5262 const ComObjPtr<MediumFormat> &aFormat,
5263 MediumVariant_T aVariant,
5264 PVDINTERFACEIO aVDImageIOIf, void *aVDImageIOUser,
5265 const ComObjPtr<Progress> &aProgress)
5266{
5267 AssertPtrReturn(aFilename, E_INVALIDARG);
5268 AssertReturn(!aFormat.isNull(), E_INVALIDARG);
5269 AssertReturn(!aProgress.isNull(), E_INVALIDARG);
5270
5271 AutoCaller autoCaller(this);
5272 if (FAILED(autoCaller.rc())) return autoCaller.rc();
5273
5274 HRESULT rc = S_OK;
5275 Medium::Task *pTask = NULL;
5276
5277 try
5278 {
5279 // This needs no extra locks besides what is done in the called methods.
5280
5281 /* Build the source lock list. */
5282 MediumLockList *pSourceMediumLockList(new MediumLockList());
5283 rc = createMediumLockList(true /* fFailIfInaccessible */,
5284 false /* fMediumLockWrite */,
5285 NULL,
5286 *pSourceMediumLockList);
5287 if (FAILED(rc))
5288 {
5289 delete pSourceMediumLockList;
5290 throw rc;
5291 }
5292
5293 rc = pSourceMediumLockList->Lock();
5294 if (FAILED(rc))
5295 {
5296 delete pSourceMediumLockList;
5297 throw setError(rc,
5298 tr("Failed to lock source media '%s'"),
5299 getLocationFull().c_str());
5300 }
5301
5302 /* setup task object to carry out the operation asynchronously */
5303 pTask = new Medium::ExportTask(this, aProgress, aFilename, aFormat,
5304 aVariant, aVDImageIOIf,
5305 aVDImageIOUser, pSourceMediumLockList);
5306 rc = pTask->rc();
5307 AssertComRC(rc);
5308 if (FAILED(rc))
5309 throw rc;
5310 }
5311 catch (HRESULT aRC) { rc = aRC; }
5312
5313 if (SUCCEEDED(rc))
5314 rc = startThread(pTask);
5315 else if (pTask != NULL)
5316 delete pTask;
5317
5318 return rc;
5319}
5320
5321/**
5322 * Used by IAppliance to import disk images.
5323 *
5324 * @param aFilename Filename to read (UTF8).
5325 * @param aFormat Medium format for reading @a aFilename.
5326 * @param aVariant Which exact image format variant to use
5327 * for the destination image.
5328 * @param aVDImageIOCallbacks Pointer to the callback table for a
5329 * VDINTERFACEIO interface. May be NULL.
5330 * @param aVDImageIOUser Opaque data for the callbacks.
5331 * @param aParent Parent medium. May be NULL.
5332 * @param aProgress Progress object to use.
5333 * @return
5334 * @note The destination format is defined by the Medium instance.
5335 */
5336HRESULT Medium::importFile(const char *aFilename,
5337 const ComObjPtr<MediumFormat> &aFormat,
5338 MediumVariant_T aVariant,
5339 PVDINTERFACEIO aVDImageIOIf, void *aVDImageIOUser,
5340 const ComObjPtr<Medium> &aParent,
5341 const ComObjPtr<Progress> &aProgress)
5342{
5343 AssertPtrReturn(aFilename, E_INVALIDARG);
5344 AssertReturn(!aFormat.isNull(), E_INVALIDARG);
5345 AssertReturn(!aProgress.isNull(), E_INVALIDARG);
5346
5347 AutoCaller autoCaller(this);
5348 if (FAILED(autoCaller.rc())) return autoCaller.rc();
5349
5350 HRESULT rc = S_OK;
5351 Medium::Task *pTask = NULL;
5352
5353 try
5354 {
5355 // locking: we need the tree lock first because we access parent pointers
5356 // and we need to write-lock the media involved
5357 uint32_t cHandles = 2;
5358 LockHandle* pHandles[3] = { &m->pVirtualBox->getMediaTreeLockHandle(),
5359 this->lockHandle() };
5360 /* Only add parent to the lock if it is not null */
5361 if (!aParent.isNull())
5362 pHandles[cHandles++] = aParent->lockHandle();
5363 AutoWriteLock alock(cHandles,
5364 pHandles
5365 COMMA_LOCKVAL_SRC_POS);
5366
5367 if ( m->state != MediumState_NotCreated
5368 && m->state != MediumState_Created)
5369 throw setStateError();
5370
5371 /* Build the target lock list. */
5372 MediumLockList *pTargetMediumLockList(new MediumLockList());
5373 alock.release();
5374 rc = createMediumLockList(true /* fFailIfInaccessible */,
5375 true /* fMediumLockWrite */,
5376 aParent,
5377 *pTargetMediumLockList);
5378 alock.acquire();
5379 if (FAILED(rc))
5380 {
5381 delete pTargetMediumLockList;
5382 throw rc;
5383 }
5384
5385 alock.release();
5386 rc = pTargetMediumLockList->Lock();
5387 alock.acquire();
5388 if (FAILED(rc))
5389 {
5390 delete pTargetMediumLockList;
5391 throw setError(rc,
5392 tr("Failed to lock target media '%s'"),
5393 getLocationFull().c_str());
5394 }
5395
5396 /* setup task object to carry out the operation asynchronously */
5397 pTask = new Medium::ImportTask(this, aProgress, aFilename, aFormat,
5398 aVariant, aVDImageIOIf,
5399 aVDImageIOUser, aParent,
5400 pTargetMediumLockList);
5401 rc = pTask->rc();
5402 AssertComRC(rc);
5403 if (FAILED(rc))
5404 throw rc;
5405
5406 if (m->state == MediumState_NotCreated)
5407 m->state = MediumState_Creating;
5408 }
5409 catch (HRESULT aRC) { rc = aRC; }
5410
5411 if (SUCCEEDED(rc))
5412 rc = startThread(pTask);
5413 else if (pTask != NULL)
5414 delete pTask;
5415
5416 return rc;
5417}
5418
5419/**
5420 * Internal version of the public CloneTo API which allows to enable certain
5421 * optimizations to improve speed during VM cloning.
5422 *
5423 * @param aTarget Target medium
5424 * @param aVariant Which exact image format variant to use
5425 * for the destination image.
5426 * @param aParent Parent medium. May be NULL.
5427 * @param aProgress Progress object to use.
5428 * @param idxSrcImageSame The last image in the source chain which has the
5429 * same content as the given image in the destination
5430 * chain. Use UINT32_MAX to disable this optimization.
5431 * @param idxDstImageSame The last image in the destination chain which has the
5432 * same content as the given image in the source chain.
5433 * Use UINT32_MAX to disable this optimization.
5434 * @return
5435 */
5436HRESULT Medium::cloneToEx(const ComObjPtr<Medium> &aTarget, ULONG aVariant,
5437 const ComObjPtr<Medium> &aParent, IProgress **aProgress,
5438 uint32_t idxSrcImageSame, uint32_t idxDstImageSame)
5439{
5440 CheckComArgNotNull(aTarget);
5441 CheckComArgOutPointerValid(aProgress);
5442 ComAssertRet(aTarget != this, E_INVALIDARG);
5443
5444 AutoCaller autoCaller(this);
5445 if (FAILED(autoCaller.rc())) return autoCaller.rc();
5446
5447 HRESULT rc = S_OK;
5448 ComObjPtr<Progress> pProgress;
5449 Medium::Task *pTask = NULL;
5450
5451 try
5452 {
5453 // locking: we need the tree lock first because we access parent pointers
5454 // and we need to write-lock the media involved
5455 uint32_t cHandles = 3;
5456 LockHandle* pHandles[4] = { &m->pVirtualBox->getMediaTreeLockHandle(),
5457 this->lockHandle(),
5458 aTarget->lockHandle() };
5459 /* Only add parent to the lock if it is not null */
5460 if (!aParent.isNull())
5461 pHandles[cHandles++] = aParent->lockHandle();
5462 AutoWriteLock alock(cHandles,
5463 pHandles
5464 COMMA_LOCKVAL_SRC_POS);
5465
5466 if ( aTarget->m->state != MediumState_NotCreated
5467 && aTarget->m->state != MediumState_Created)
5468 throw aTarget->setStateError();
5469
5470 /* Build the source lock list. */
5471 MediumLockList *pSourceMediumLockList(new MediumLockList());
5472 alock.release();
5473 rc = createMediumLockList(true /* fFailIfInaccessible */,
5474 false /* fMediumLockWrite */,
5475 NULL,
5476 *pSourceMediumLockList);
5477 alock.acquire();
5478 if (FAILED(rc))
5479 {
5480 delete pSourceMediumLockList;
5481 throw rc;
5482 }
5483
5484 /* Build the target lock list (including the to-be parent chain). */
5485 MediumLockList *pTargetMediumLockList(new MediumLockList());
5486 alock.release();
5487 rc = aTarget->createMediumLockList(true /* fFailIfInaccessible */,
5488 true /* fMediumLockWrite */,
5489 aParent,
5490 *pTargetMediumLockList);
5491 alock.acquire();
5492 if (FAILED(rc))
5493 {
5494 delete pSourceMediumLockList;
5495 delete pTargetMediumLockList;
5496 throw rc;
5497 }
5498
5499 alock.release();
5500 rc = pSourceMediumLockList->Lock();
5501 alock.acquire();
5502 if (FAILED(rc))
5503 {
5504 delete pSourceMediumLockList;
5505 delete pTargetMediumLockList;
5506 throw setError(rc,
5507 tr("Failed to lock source media '%s'"),
5508 getLocationFull().c_str());
5509 }
5510 alock.release();
5511 rc = pTargetMediumLockList->Lock();
5512 alock.acquire();
5513 if (FAILED(rc))
5514 {
5515 delete pSourceMediumLockList;
5516 delete pTargetMediumLockList;
5517 throw setError(rc,
5518 tr("Failed to lock target media '%s'"),
5519 aTarget->getLocationFull().c_str());
5520 }
5521
5522 pProgress.createObject();
5523 rc = pProgress->init(m->pVirtualBox,
5524 static_cast <IMedium *>(this),
5525 BstrFmt(tr("Creating clone medium '%s'"), aTarget->m->strLocationFull.c_str()).raw(),
5526 TRUE /* aCancelable */);
5527 if (FAILED(rc))
5528 {
5529 delete pSourceMediumLockList;
5530 delete pTargetMediumLockList;
5531 throw rc;
5532 }
5533
5534 /* setup task object to carry out the operation asynchronously */
5535 pTask = new Medium::CloneTask(this, pProgress, aTarget,
5536 (MediumVariant_T)aVariant,
5537 aParent, idxSrcImageSame,
5538 idxDstImageSame, pSourceMediumLockList,
5539 pTargetMediumLockList);
5540 rc = pTask->rc();
5541 AssertComRC(rc);
5542 if (FAILED(rc))
5543 throw rc;
5544
5545 if (aTarget->m->state == MediumState_NotCreated)
5546 aTarget->m->state = MediumState_Creating;
5547 }
5548 catch (HRESULT aRC) { rc = aRC; }
5549
5550 if (SUCCEEDED(rc))
5551 {
5552 rc = startThread(pTask);
5553
5554 if (SUCCEEDED(rc))
5555 pProgress.queryInterfaceTo(aProgress);
5556 }
5557 else if (pTask != NULL)
5558 delete pTask;
5559
5560 return rc;
5561}
5562
5563////////////////////////////////////////////////////////////////////////////////
5564//
5565// Private methods
5566//
5567////////////////////////////////////////////////////////////////////////////////
5568
5569/**
5570 * Queries information from the medium.
5571 *
5572 * As a result of this call, the accessibility state and data members such as
5573 * size and description will be updated with the current information.
5574 *
5575 * @note This method may block during a system I/O call that checks storage
5576 * accessibility.
5577 *
5578 * @note Caller MUST NOT hold the media tree or medium lock.
5579 *
5580 * @note Locks mParent for reading. Locks this object for writing.
5581 *
5582 * @param fSetImageId Whether to reset the UUID contained in the image file to the UUID in the medium instance data (see SetIDs())
5583 * @param fSetParentId Whether to reset the parent UUID contained in the image file to the parent UUID in the medium instance data (see SetIDs())
5584 * @return
5585 */
5586HRESULT Medium::queryInfo(bool fSetImageId, bool fSetParentId)
5587{
5588 Assert(!isWriteLockOnCurrentThread());
5589 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
5590
5591 if ( m->state != MediumState_Created
5592 && m->state != MediumState_Inaccessible
5593 && m->state != MediumState_LockedRead)
5594 return E_FAIL;
5595
5596 HRESULT rc = S_OK;
5597
5598 int vrc = VINF_SUCCESS;
5599
5600 /* check if a blocking queryInfo() call is in progress on some other thread,
5601 * and wait for it to finish if so instead of querying data ourselves */
5602 if (m->queryInfoRunning)
5603 {
5604 Assert( m->state == MediumState_LockedRead
5605 || m->state == MediumState_LockedWrite);
5606
5607 while (m->queryInfoRunning)
5608 {
5609 alock.release();
5610 {
5611 AutoReadLock qlock(m->queryInfoSem COMMA_LOCKVAL_SRC_POS);
5612 }
5613 alock.acquire();
5614 }
5615
5616 return S_OK;
5617 }
5618
5619 bool success = false;
5620 Utf8Str lastAccessError;
5621
5622 /* are we dealing with a new medium constructed using the existing
5623 * location? */
5624 bool isImport = m->id.isZero();
5625 unsigned uOpenFlags = VD_OPEN_FLAGS_INFO;
5626
5627 /* Note that we don't use VD_OPEN_FLAGS_READONLY when opening new
5628 * media because that would prevent necessary modifications
5629 * when opening media of some third-party formats for the first
5630 * time in VirtualBox (such as VMDK for which VDOpen() needs to
5631 * generate an UUID if it is missing) */
5632 if ( m->hddOpenMode == OpenReadOnly
5633 || m->type == MediumType_Readonly
5634 || (!isImport && !fSetImageId && !fSetParentId)
5635 )
5636 uOpenFlags |= VD_OPEN_FLAGS_READONLY;
5637
5638 /* Open shareable medium with the appropriate flags */
5639 if (m->type == MediumType_Shareable)
5640 uOpenFlags |= VD_OPEN_FLAGS_SHAREABLE;
5641
5642 /* Lock the medium, which makes the behavior much more consistent */
5643 alock.release();
5644 ComPtr<IToken> pToken;
5645 if (uOpenFlags & (VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_SHAREABLE))
5646 rc = LockRead(pToken.asOutParam());
5647 else
5648 rc = LockWrite(pToken.asOutParam());
5649 if (FAILED(rc)) return rc;
5650 alock.acquire();
5651
5652 /* Copies of the input state fields which are not read-only,
5653 * as we're dropping the lock. CAUTION: be extremely careful what
5654 * you do with the contents of this medium object, as you will
5655 * create races if there are concurrent changes. */
5656 Utf8Str format(m->strFormat);
5657 Utf8Str location(m->strLocationFull);
5658 ComObjPtr<MediumFormat> formatObj = m->formatObj;
5659
5660 /* "Output" values which can't be set because the lock isn't held
5661 * at the time the values are determined. */
5662 Guid mediumId = m->id;
5663 uint64_t mediumSize = 0;
5664 uint64_t mediumLogicalSize = 0;
5665
5666 /* Flag whether a base image has a non-zero parent UUID and thus
5667 * need repairing after it was closed again. */
5668 bool fRepairImageZeroParentUuid = false;
5669
5670 /* release the object lock before a lengthy operation, and take the
5671 * opportunity to have a media tree lock, too, which isn't held initially */
5672 m->queryInfoRunning = true;
5673 alock.release();
5674 Assert(!isWriteLockOnCurrentThread());
5675 Assert(!m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
5676 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
5677 treeLock.release();
5678
5679 /* Note that taking the queryInfoSem after leaving the object lock above
5680 * can lead to short spinning of the loops waiting for queryInfo() to
5681 * complete. This is unavoidable since the other order causes a lock order
5682 * violation: here it would be requesting the object lock (at the beginning
5683 * of the method), then queryInfoSem, and below the other way round. */
5684 AutoWriteLock qlock(m->queryInfoSem COMMA_LOCKVAL_SRC_POS);
5685
5686 try
5687 {
5688 /* skip accessibility checks for host drives */
5689 if (m->hostDrive)
5690 {
5691 success = true;
5692 throw S_OK;
5693 }
5694
5695 PVBOXHDD hdd;
5696 vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
5697 ComAssertRCThrow(vrc, E_FAIL);
5698
5699 try
5700 {
5701 /** @todo This kind of opening of media is assuming that diff
5702 * media can be opened as base media. Should be documented that
5703 * it must work for all medium format backends. */
5704 vrc = VDOpen(hdd,
5705 format.c_str(),
5706 location.c_str(),
5707 uOpenFlags | m->uOpenFlagsDef,
5708 m->vdImageIfaces);
5709 if (RT_FAILURE(vrc))
5710 {
5711 lastAccessError = Utf8StrFmt(tr("Could not open the medium '%s'%s"),
5712 location.c_str(), vdError(vrc).c_str());
5713 throw S_OK;
5714 }
5715
5716 if (formatObj->i_getCapabilities() & MediumFormatCapabilities_Uuid)
5717 {
5718 /* Modify the UUIDs if necessary. The associated fields are
5719 * not modified by other code, so no need to copy. */
5720 if (fSetImageId)
5721 {
5722 alock.acquire();
5723 vrc = VDSetUuid(hdd, 0, m->uuidImage.raw());
5724 alock.release();
5725 if (RT_FAILURE(vrc))
5726 {
5727 lastAccessError = Utf8StrFmt(tr("Could not update the UUID of medium '%s'%s"),
5728 location.c_str(), vdError(vrc).c_str());
5729 throw S_OK;
5730 }
5731 mediumId = m->uuidImage;
5732 }
5733 if (fSetParentId)
5734 {
5735 alock.acquire();
5736 vrc = VDSetParentUuid(hdd, 0, m->uuidParentImage.raw());
5737 alock.release();
5738 if (RT_FAILURE(vrc))
5739 {
5740 lastAccessError = Utf8StrFmt(tr("Could not update the parent UUID of medium '%s'%s"),
5741 location.c_str(), vdError(vrc).c_str());
5742 throw S_OK;
5743 }
5744 }
5745 /* zap the information, these are no long-term members */
5746 alock.acquire();
5747 unconst(m->uuidImage).clear();
5748 unconst(m->uuidParentImage).clear();
5749 alock.release();
5750
5751 /* check the UUID */
5752 RTUUID uuid;
5753 vrc = VDGetUuid(hdd, 0, &uuid);
5754 ComAssertRCThrow(vrc, E_FAIL);
5755
5756 if (isImport)
5757 {
5758 mediumId = uuid;
5759
5760 if (mediumId.isZero() && (m->hddOpenMode == OpenReadOnly))
5761 // only when importing a VDMK that has no UUID, create one in memory
5762 mediumId.create();
5763 }
5764 else
5765 {
5766 Assert(!mediumId.isZero());
5767
5768 if (mediumId != uuid)
5769 {
5770 /** @todo r=klaus this always refers to VirtualBox.xml as the medium registry, even for new VMs */
5771 lastAccessError = Utf8StrFmt(
5772 tr("UUID {%RTuuid} of the medium '%s' does not match the value {%RTuuid} stored in the media registry ('%s')"),
5773 &uuid,
5774 location.c_str(),
5775 mediumId.raw(),
5776 m->pVirtualBox->settingsFilePath().c_str());
5777 throw S_OK;
5778 }
5779 }
5780 }
5781 else
5782 {
5783 /* the backend does not support storing UUIDs within the
5784 * underlying storage so use what we store in XML */
5785
5786 if (fSetImageId)
5787 {
5788 /* set the UUID if an API client wants to change it */
5789 alock.acquire();
5790 mediumId = m->uuidImage;
5791 alock.release();
5792 }
5793 else if (isImport)
5794 {
5795 /* generate an UUID for an imported UUID-less medium */
5796 mediumId.create();
5797 }
5798 }
5799
5800 /* set the image uuid before the below parent uuid handling code
5801 * might place it somewhere in the media tree, so that the medium
5802 * UUID is valid at this point */
5803 alock.acquire();
5804 if (isImport || fSetImageId)
5805 unconst(m->id) = mediumId;
5806 alock.release();
5807
5808 /* get the medium variant */
5809 unsigned uImageFlags;
5810 vrc = VDGetImageFlags(hdd, 0, &uImageFlags);
5811 ComAssertRCThrow(vrc, E_FAIL);
5812 alock.acquire();
5813 m->variant = (MediumVariant_T)uImageFlags;
5814 alock.release();
5815
5816 /* check/get the parent uuid and update corresponding state */
5817 if (uImageFlags & VD_IMAGE_FLAGS_DIFF)
5818 {
5819 RTUUID parentId;
5820 vrc = VDGetParentUuid(hdd, 0, &parentId);
5821 ComAssertRCThrow(vrc, E_FAIL);
5822
5823 /* streamOptimized VMDK images are only accepted as base
5824 * images, as this allows automatic repair of OVF appliances.
5825 * Since such images don't support random writes they will not
5826 * be created for diff images. Only an overly smart user might
5827 * manually create this case. Too bad for him. */
5828 if ( (isImport || fSetParentId)
5829 && !(uImageFlags & VD_VMDK_IMAGE_FLAGS_STREAM_OPTIMIZED))
5830 {
5831 /* the parent must be known to us. Note that we freely
5832 * call locking methods of mVirtualBox and parent, as all
5833 * relevant locks must be already held. There may be no
5834 * concurrent access to the just opened medium on other
5835 * threads yet (and init() will fail if this method reports
5836 * MediumState_Inaccessible) */
5837
5838 ComObjPtr<Medium> pParent;
5839 if (RTUuidIsNull(&parentId))
5840 rc = VBOX_E_OBJECT_NOT_FOUND;
5841 else
5842 rc = m->pVirtualBox->findHardDiskById(Guid(parentId), false /* aSetError */, &pParent);
5843 if (FAILED(rc))
5844 {
5845 if (fSetImageId && !fSetParentId)
5846 {
5847 /* If the image UUID gets changed for an existing
5848 * image then the parent UUID can be stale. In such
5849 * cases clear the parent information. The parent
5850 * information may/will be re-set later if the
5851 * API client wants to adjust a complete medium
5852 * hierarchy one by one. */
5853 rc = S_OK;
5854 alock.acquire();
5855 RTUuidClear(&parentId);
5856 vrc = VDSetParentUuid(hdd, 0, &parentId);
5857 alock.release();
5858 ComAssertRCThrow(vrc, E_FAIL);
5859 }
5860 else
5861 {
5862 lastAccessError = Utf8StrFmt(tr("Parent medium with UUID {%RTuuid} of the medium '%s' is not found in the media registry ('%s')"),
5863 &parentId, location.c_str(),
5864 m->pVirtualBox->settingsFilePath().c_str());
5865 throw S_OK;
5866 }
5867 }
5868
5869 /* we set mParent & children() */
5870 treeLock.acquire();
5871
5872 if (m->pParent)
5873 deparent();
5874 setParent(pParent);
5875
5876 treeLock.release();
5877 }
5878 else
5879 {
5880 /* we access mParent */
5881 treeLock.acquire();
5882
5883 /* check that parent UUIDs match. Note that there's no need
5884 * for the parent's AutoCaller (our lifetime is bound to
5885 * it) */
5886
5887 if (m->pParent.isNull())
5888 {
5889 /* Due to a bug in VDCopy() in VirtualBox 3.0.0-3.0.14
5890 * and 3.1.0-3.1.8 there are base images out there
5891 * which have a non-zero parent UUID. No point in
5892 * complaining about them, instead automatically
5893 * repair the problem. Later we can bring back the
5894 * error message, but we should wait until really
5895 * most users have repaired their images, either with
5896 * VBoxFixHdd or this way. */
5897#if 1
5898 fRepairImageZeroParentUuid = true;
5899#else /* 0 */
5900 lastAccessError = Utf8StrFmt(
5901 tr("Medium type of '%s' is differencing but it is not associated with any parent medium in the media registry ('%s')"),
5902 location.c_str(),
5903 m->pVirtualBox->settingsFilePath().c_str());
5904 treeLock.release();
5905 throw S_OK;
5906#endif /* 0 */
5907 }
5908
5909 {
5910 AutoReadLock parentLock(m->pParent COMMA_LOCKVAL_SRC_POS);
5911 if ( !fRepairImageZeroParentUuid
5912 && m->pParent->getState() != MediumState_Inaccessible
5913 && m->pParent->getId() != parentId)
5914 {
5915 /** @todo r=klaus this always refers to VirtualBox.xml as the medium registry, even for new VMs */
5916 lastAccessError = Utf8StrFmt(
5917 tr("Parent UUID {%RTuuid} of the medium '%s' does not match UUID {%RTuuid} of its parent medium stored in the media registry ('%s')"),
5918 &parentId, location.c_str(),
5919 m->pParent->getId().raw(),
5920 m->pVirtualBox->settingsFilePath().c_str());
5921 parentLock.release();
5922 treeLock.release();
5923 throw S_OK;
5924 }
5925 }
5926
5927 /// @todo NEWMEDIA what to do if the parent is not
5928 /// accessible while the diff is? Probably nothing. The
5929 /// real code will detect the mismatch anyway.
5930
5931 treeLock.release();
5932 }
5933 }
5934
5935 mediumSize = VDGetFileSize(hdd, 0);
5936 mediumLogicalSize = VDGetSize(hdd, 0);
5937
5938 success = true;
5939 }
5940 catch (HRESULT aRC)
5941 {
5942 rc = aRC;
5943 }
5944
5945 vrc = VDDestroy(hdd);
5946 if (RT_FAILURE(vrc))
5947 {
5948 lastAccessError = Utf8StrFmt(tr("Could not update and close the medium '%s'%s"),
5949 location.c_str(), vdError(vrc).c_str());
5950 success = false;
5951 throw S_OK;
5952 }
5953 }
5954 catch (HRESULT aRC)
5955 {
5956 rc = aRC;
5957 }
5958
5959 treeLock.acquire();
5960 alock.acquire();
5961
5962 if (success)
5963 {
5964 m->size = mediumSize;
5965 m->logicalSize = mediumLogicalSize;
5966 m->strLastAccessError.setNull();
5967 }
5968 else
5969 {
5970 m->strLastAccessError = lastAccessError;
5971 LogWarningFunc(("'%s' is not accessible (error='%s', rc=%Rhrc, vrc=%Rrc)\n",
5972 location.c_str(), m->strLastAccessError.c_str(),
5973 rc, vrc));
5974 }
5975
5976 /* unblock anyone waiting for the queryInfo results */
5977 qlock.release();
5978 m->queryInfoRunning = false;
5979
5980 /* Set the proper state according to the result of the check */
5981 if (success)
5982 m->preLockState = MediumState_Created;
5983 else
5984 m->preLockState = MediumState_Inaccessible;
5985
5986 pToken->Abandon();
5987 pToken.setNull();
5988
5989 if (FAILED(rc)) return rc;
5990
5991 /* If this is a base image which incorrectly has a parent UUID set,
5992 * repair the image now by zeroing the parent UUID. This is only done
5993 * when we have structural information from a config file, on import
5994 * this is not possible. If someone would accidentally call openMedium
5995 * with a diff image before the base is registered this would destroy
5996 * the diff. Not acceptable. */
5997 if (fRepairImageZeroParentUuid)
5998 {
5999 rc = LockWrite(pToken.asOutParam());
6000 if (FAILED(rc)) return rc;
6001
6002 alock.release();
6003
6004 try
6005 {
6006 PVBOXHDD hdd;
6007 vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
6008 ComAssertRCThrow(vrc, E_FAIL);
6009
6010 try
6011 {
6012 vrc = VDOpen(hdd,
6013 format.c_str(),
6014 location.c_str(),
6015 (uOpenFlags & ~VD_OPEN_FLAGS_READONLY) | m->uOpenFlagsDef,
6016 m->vdImageIfaces);
6017 if (RT_FAILURE(vrc))
6018 throw S_OK;
6019
6020 RTUUID zeroParentUuid;
6021 RTUuidClear(&zeroParentUuid);
6022 vrc = VDSetParentUuid(hdd, 0, &zeroParentUuid);
6023 ComAssertRCThrow(vrc, E_FAIL);
6024 }
6025 catch (HRESULT aRC)
6026 {
6027 rc = aRC;
6028 }
6029
6030 VDDestroy(hdd);
6031 }
6032 catch (HRESULT aRC)
6033 {
6034 rc = aRC;
6035 }
6036
6037 pToken->Abandon();
6038 pToken.setNull();
6039 if (FAILED(rc)) return rc;
6040 }
6041
6042 return rc;
6043}
6044
6045/**
6046 * Performs extra checks if the medium can be closed and returns S_OK in
6047 * this case. Otherwise, returns a respective error message. Called by
6048 * Close() under the medium tree lock and the medium lock.
6049 *
6050 * @note Also reused by Medium::Reset().
6051 *
6052 * @note Caller must hold the media tree write lock!
6053 */
6054HRESULT Medium::canClose()
6055{
6056 Assert(m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
6057
6058 if (getChildren().size() != 0)
6059 return setError(VBOX_E_OBJECT_IN_USE,
6060 tr("Cannot close medium '%s' because it has %d child media"),
6061 m->strLocationFull.c_str(), getChildren().size());
6062
6063 return S_OK;
6064}
6065
6066/**
6067 * Unregisters this medium with mVirtualBox. Called by close() under the medium tree lock.
6068 *
6069 * @note Caller must have locked the media tree lock for writing!
6070 */
6071HRESULT Medium::unregisterWithVirtualBox()
6072{
6073 /* Note that we need to de-associate ourselves from the parent to let
6074 * unregisterMedium() properly save the registry */
6075
6076 /* we modify mParent and access children */
6077 Assert(m->pVirtualBox->getMediaTreeLockHandle().isWriteLockOnCurrentThread());
6078
6079 Medium *pParentBackup = m->pParent;
6080 AssertReturn(getChildren().size() == 0, E_FAIL);
6081 if (m->pParent)
6082 deparent();
6083
6084 HRESULT rc = m->pVirtualBox->unregisterMedium(this);
6085 if (FAILED(rc))
6086 {
6087 if (pParentBackup)
6088 {
6089 // re-associate with the parent as we are still relatives in the registry
6090 m->pParent = pParentBackup;
6091 m->pParent->m->llChildren.push_back(this);
6092 }
6093 }
6094
6095 return rc;
6096}
6097
6098/**
6099 * Like SetProperty but do not trigger a settings store. Only for internal use!
6100 */
6101HRESULT Medium::setPropertyDirect(const Utf8Str &aName, const Utf8Str &aValue)
6102{
6103 AutoCaller autoCaller(this);
6104 if (FAILED(autoCaller.rc())) return autoCaller.rc();
6105
6106 AutoWriteLock mlock(this COMMA_LOCKVAL_SRC_POS);
6107
6108 switch (m->state)
6109 {
6110 case MediumState_Created:
6111 case MediumState_Inaccessible:
6112 break;
6113 default:
6114 return setStateError();
6115 }
6116
6117 m->mapProperties[aName] = aValue;
6118
6119 return S_OK;
6120}
6121
6122/**
6123 * Sets the extended error info according to the current media state.
6124 *
6125 * @note Must be called from under this object's write or read lock.
6126 */
6127HRESULT Medium::setStateError()
6128{
6129 HRESULT rc = E_FAIL;
6130
6131 switch (m->state)
6132 {
6133 case MediumState_NotCreated:
6134 {
6135 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6136 tr("Storage for the medium '%s' is not created"),
6137 m->strLocationFull.c_str());
6138 break;
6139 }
6140 case MediumState_Created:
6141 {
6142 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6143 tr("Storage for the medium '%s' is already created"),
6144 m->strLocationFull.c_str());
6145 break;
6146 }
6147 case MediumState_LockedRead:
6148 {
6149 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6150 tr("Medium '%s' is locked for reading by another task"),
6151 m->strLocationFull.c_str());
6152 break;
6153 }
6154 case MediumState_LockedWrite:
6155 {
6156 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6157 tr("Medium '%s' is locked for writing by another task"),
6158 m->strLocationFull.c_str());
6159 break;
6160 }
6161 case MediumState_Inaccessible:
6162 {
6163 /* be in sync with Console::powerUpThread() */
6164 if (!m->strLastAccessError.isEmpty())
6165 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6166 tr("Medium '%s' is not accessible. %s"),
6167 m->strLocationFull.c_str(), m->strLastAccessError.c_str());
6168 else
6169 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6170 tr("Medium '%s' is not accessible"),
6171 m->strLocationFull.c_str());
6172 break;
6173 }
6174 case MediumState_Creating:
6175 {
6176 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6177 tr("Storage for the medium '%s' is being created"),
6178 m->strLocationFull.c_str());
6179 break;
6180 }
6181 case MediumState_Deleting:
6182 {
6183 rc = setError(VBOX_E_INVALID_OBJECT_STATE,
6184 tr("Storage for the medium '%s' is being deleted"),
6185 m->strLocationFull.c_str());
6186 break;
6187 }
6188 default:
6189 {
6190 AssertFailed();
6191 break;
6192 }
6193 }
6194
6195 return rc;
6196}
6197
6198/**
6199 * Sets the value of m->strLocationFull. The given location must be a fully
6200 * qualified path; relative paths are not supported here.
6201 *
6202 * As a special exception, if the specified location is a file path that ends with '/'
6203 * then the file name part will be generated by this method automatically in the format
6204 * '{<uuid>}.<ext>' where <uuid> is a fresh UUID that this method will generate
6205 * and assign to this medium, and <ext> is the default extension for this
6206 * medium's storage format. Note that this procedure requires the media state to
6207 * be NotCreated and will return a failure otherwise.
6208 *
6209 * @param aLocation Location of the storage unit. If the location is a FS-path,
6210 * then it can be relative to the VirtualBox home directory.
6211 * @param aFormat Optional fallback format if it is an import and the format
6212 * cannot be determined.
6213 *
6214 * @note Must be called from under this object's write lock.
6215 */
6216HRESULT Medium::setLocation(const Utf8Str &aLocation,
6217 const Utf8Str &aFormat /* = Utf8Str::Empty */)
6218{
6219 AssertReturn(!aLocation.isEmpty(), E_FAIL);
6220
6221 AutoCaller autoCaller(this);
6222 AssertComRCReturnRC(autoCaller.rc());
6223
6224 /* formatObj may be null only when initializing from an existing path and
6225 * no format is known yet */
6226 AssertReturn( (!m->strFormat.isEmpty() && !m->formatObj.isNull())
6227 || ( autoCaller.state() == InInit
6228 && m->state != MediumState_NotCreated
6229 && m->id.isZero()
6230 && m->strFormat.isEmpty()
6231 && m->formatObj.isNull()),
6232 E_FAIL);
6233
6234 /* are we dealing with a new medium constructed using the existing
6235 * location? */
6236 bool isImport = m->strFormat.isEmpty();
6237
6238 if ( isImport
6239 || ( (m->formatObj->i_getCapabilities() & MediumFormatCapabilities_File)
6240 && !m->hostDrive))
6241 {
6242 Guid id;
6243
6244 Utf8Str locationFull(aLocation);
6245
6246 if (m->state == MediumState_NotCreated)
6247 {
6248 /* must be a file (formatObj must be already known) */
6249 Assert(m->formatObj->i_getCapabilities() & MediumFormatCapabilities_File);
6250
6251 if (RTPathFilename(aLocation.c_str()) == NULL)
6252 {
6253 /* no file name is given (either an empty string or ends with a
6254 * slash), generate a new UUID + file name if the state allows
6255 * this */
6256
6257 ComAssertMsgRet(!m->formatObj->i_getFileExtensions().empty(),
6258 ("Must be at least one extension if it is MediumFormatCapabilities_File\n"),
6259 E_FAIL);
6260
6261 Utf8Str strExt = m->formatObj->i_getFileExtensions().front();
6262 ComAssertMsgRet(!strExt.isEmpty(),
6263 ("Default extension must not be empty\n"),
6264 E_FAIL);
6265
6266 id.create();
6267
6268 locationFull = Utf8StrFmt("%s{%RTuuid}.%s",
6269 aLocation.c_str(), id.raw(), strExt.c_str());
6270 }
6271 }
6272
6273 // we must always have full paths now (if it refers to a file)
6274 if ( ( m->formatObj.isNull()
6275 || m->formatObj->i_getCapabilities() & MediumFormatCapabilities_File)
6276 && !RTPathStartsWithRoot(locationFull.c_str()))
6277 return setError(VBOX_E_FILE_ERROR,
6278 tr("The given path '%s' is not fully qualified"),
6279 locationFull.c_str());
6280
6281 /* detect the backend from the storage unit if importing */
6282 if (isImport)
6283 {
6284 VDTYPE enmType = VDTYPE_INVALID;
6285 char *backendName = NULL;
6286
6287 int vrc = VINF_SUCCESS;
6288
6289 /* is it a file? */
6290 {
6291 RTFILE file;
6292 vrc = RTFileOpen(&file, locationFull.c_str(), RTFILE_O_READ | RTFILE_O_OPEN | RTFILE_O_DENY_NONE);
6293 if (RT_SUCCESS(vrc))
6294 RTFileClose(file);
6295 }
6296 if (RT_SUCCESS(vrc))
6297 {
6298 vrc = VDGetFormat(NULL /* pVDIfsDisk */, NULL /* pVDIfsImage */,
6299 locationFull.c_str(), &backendName, &enmType);
6300 }
6301 else if ( vrc != VERR_FILE_NOT_FOUND
6302 && vrc != VERR_PATH_NOT_FOUND
6303 && vrc != VERR_ACCESS_DENIED
6304 && locationFull != aLocation)
6305 {
6306 /* assume it's not a file, restore the original location */
6307 locationFull = aLocation;
6308 vrc = VDGetFormat(NULL /* pVDIfsDisk */, NULL /* pVDIfsImage */,
6309 locationFull.c_str(), &backendName, &enmType);
6310 }
6311
6312 if (RT_FAILURE(vrc))
6313 {
6314 if (vrc == VERR_ACCESS_DENIED)
6315 return setError(VBOX_E_FILE_ERROR,
6316 tr("Permission problem accessing the file for the medium '%s' (%Rrc)"),
6317 locationFull.c_str(), vrc);
6318 else if (vrc == VERR_FILE_NOT_FOUND || vrc == VERR_PATH_NOT_FOUND)
6319 return setError(VBOX_E_FILE_ERROR,
6320 tr("Could not find file for the medium '%s' (%Rrc)"),
6321 locationFull.c_str(), vrc);
6322 else if (aFormat.isEmpty())
6323 return setError(VBOX_E_IPRT_ERROR,
6324 tr("Could not get the storage format of the medium '%s' (%Rrc)"),
6325 locationFull.c_str(), vrc);
6326 else
6327 {
6328 HRESULT rc = setFormat(aFormat);
6329 /* setFormat() must not fail since we've just used the backend so
6330 * the format object must be there */
6331 AssertComRCReturnRC(rc);
6332 }
6333 }
6334 else if ( enmType == VDTYPE_INVALID
6335 || m->devType != convertToDeviceType(enmType))
6336 {
6337 /*
6338 * The user tried to use a image as a device which is not supported
6339 * by the backend.
6340 */
6341 return setError(E_FAIL,
6342 tr("The medium '%s' can't be used as the requested device type"),
6343 locationFull.c_str());
6344 }
6345 else
6346 {
6347 ComAssertRet(backendName != NULL && *backendName != '\0', E_FAIL);
6348
6349 HRESULT rc = setFormat(backendName);
6350 RTStrFree(backendName);
6351
6352 /* setFormat() must not fail since we've just used the backend so
6353 * the format object must be there */
6354 AssertComRCReturnRC(rc);
6355 }
6356 }
6357
6358 m->strLocationFull = locationFull;
6359
6360 /* is it still a file? */
6361 if ( (m->formatObj->i_getCapabilities() & MediumFormatCapabilities_File)
6362 && (m->state == MediumState_NotCreated)
6363 )
6364 /* assign a new UUID (this UUID will be used when calling
6365 * VDCreateBase/VDCreateDiff as a wanted UUID). Note that we
6366 * also do that if we didn't generate it to make sure it is
6367 * either generated by us or reset to null */
6368 unconst(m->id) = id;
6369 }
6370 else
6371 m->strLocationFull = aLocation;
6372
6373 return S_OK;
6374}
6375
6376/**
6377 * Checks that the format ID is valid and sets it on success.
6378 *
6379 * Note that this method will caller-reference the format object on success!
6380 * This reference must be released somewhere to let the MediumFormat object be
6381 * uninitialized.
6382 *
6383 * @note Must be called from under this object's write lock.
6384 */
6385HRESULT Medium::setFormat(const Utf8Str &aFormat)
6386{
6387 /* get the format object first */
6388 {
6389 SystemProperties *pSysProps = m->pVirtualBox->getSystemProperties();
6390 AutoReadLock propsLock(pSysProps COMMA_LOCKVAL_SRC_POS);
6391
6392 unconst(m->formatObj) = pSysProps->mediumFormat(aFormat);
6393 if (m->formatObj.isNull())
6394 return setError(E_INVALIDARG,
6395 tr("Invalid medium storage format '%s'"),
6396 aFormat.c_str());
6397
6398 /* reference the format permanently to prevent its unexpected
6399 * uninitialization */
6400 HRESULT rc = m->formatObj->addCaller();
6401 AssertComRCReturnRC(rc);
6402
6403 /* get properties (preinsert them as keys in the map). Note that the
6404 * map doesn't grow over the object life time since the set of
6405 * properties is meant to be constant. */
6406
6407 Assert(m->mapProperties.empty());
6408
6409 for (MediumFormat::PropertyArray::const_iterator it = m->formatObj->i_getProperties().begin();
6410 it != m->formatObj->i_getProperties().end();
6411 ++it)
6412 {
6413 m->mapProperties.insert(std::make_pair(it->strName, Utf8Str::Empty));
6414 }
6415 }
6416
6417 unconst(m->strFormat) = aFormat;
6418
6419 return S_OK;
6420}
6421
6422/**
6423 * Converts the Medium device type to the VD type.
6424 */
6425VDTYPE Medium::convertDeviceType()
6426{
6427 VDTYPE enmType;
6428
6429 switch (m->devType)
6430 {
6431 case DeviceType_HardDisk:
6432 enmType = VDTYPE_HDD;
6433 break;
6434 case DeviceType_DVD:
6435 enmType = VDTYPE_DVD;
6436 break;
6437 case DeviceType_Floppy:
6438 enmType = VDTYPE_FLOPPY;
6439 break;
6440 default:
6441 ComAssertFailedRet(VDTYPE_INVALID);
6442 }
6443
6444 return enmType;
6445}
6446
6447/**
6448 * Converts from the VD type to the medium type.
6449 */
6450DeviceType_T Medium::convertToDeviceType(VDTYPE enmType)
6451{
6452 DeviceType_T devType;
6453
6454 switch (enmType)
6455 {
6456 case VDTYPE_HDD:
6457 devType = DeviceType_HardDisk;
6458 break;
6459 case VDTYPE_DVD:
6460 devType = DeviceType_DVD;
6461 break;
6462 case VDTYPE_FLOPPY:
6463 devType = DeviceType_Floppy;
6464 break;
6465 default:
6466 ComAssertFailedRet(DeviceType_Null);
6467 }
6468
6469 return devType;
6470}
6471
6472/**
6473 * Returns the last error message collected by the vdErrorCall callback and
6474 * resets it.
6475 *
6476 * The error message is returned prepended with a dot and a space, like this:
6477 * <code>
6478 * ". <error_text> (%Rrc)"
6479 * </code>
6480 * to make it easily appendable to a more general error message. The @c %Rrc
6481 * format string is given @a aVRC as an argument.
6482 *
6483 * If there is no last error message collected by vdErrorCall or if it is a
6484 * null or empty string, then this function returns the following text:
6485 * <code>
6486 * " (%Rrc)"
6487 * </code>
6488 *
6489 * @note Doesn't do any object locking; it is assumed that the caller makes sure
6490 * the callback isn't called by more than one thread at a time.
6491 *
6492 * @param aVRC VBox error code to use when no error message is provided.
6493 */
6494Utf8Str Medium::vdError(int aVRC)
6495{
6496 Utf8Str error;
6497
6498 if (m->vdError.isEmpty())
6499 error = Utf8StrFmt(" (%Rrc)", aVRC);
6500 else
6501 error = Utf8StrFmt(".\n%s", m->vdError.c_str());
6502
6503 m->vdError.setNull();
6504
6505 return error;
6506}
6507
6508/**
6509 * Error message callback.
6510 *
6511 * Puts the reported error message to the m->vdError field.
6512 *
6513 * @note Doesn't do any object locking; it is assumed that the caller makes sure
6514 * the callback isn't called by more than one thread at a time.
6515 *
6516 * @param pvUser The opaque data passed on container creation.
6517 * @param rc The VBox error code.
6518 * @param RT_SRC_POS_DECL Use RT_SRC_POS.
6519 * @param pszFormat Error message format string.
6520 * @param va Error message arguments.
6521 */
6522/*static*/
6523DECLCALLBACK(void) Medium::vdErrorCall(void *pvUser, int rc, RT_SRC_POS_DECL,
6524 const char *pszFormat, va_list va)
6525{
6526 NOREF(pszFile); NOREF(iLine); NOREF(pszFunction); /* RT_SRC_POS_DECL */
6527
6528 Medium *that = static_cast<Medium*>(pvUser);
6529 AssertReturnVoid(that != NULL);
6530
6531 if (that->m->vdError.isEmpty())
6532 that->m->vdError =
6533 Utf8StrFmt("%s (%Rrc)", Utf8Str(pszFormat, va).c_str(), rc);
6534 else
6535 that->m->vdError =
6536 Utf8StrFmt("%s.\n%s (%Rrc)", that->m->vdError.c_str(),
6537 Utf8Str(pszFormat, va).c_str(), rc);
6538}
6539
6540/* static */
6541DECLCALLBACK(bool) Medium::vdConfigAreKeysValid(void *pvUser,
6542 const char * /* pszzValid */)
6543{
6544 Medium *that = static_cast<Medium*>(pvUser);
6545 AssertReturn(that != NULL, false);
6546
6547 /* we always return true since the only keys we have are those found in
6548 * VDBACKENDINFO */
6549 return true;
6550}
6551
6552/* static */
6553DECLCALLBACK(int) Medium::vdConfigQuerySize(void *pvUser,
6554 const char *pszName,
6555 size_t *pcbValue)
6556{
6557 AssertReturn(VALID_PTR(pcbValue), VERR_INVALID_POINTER);
6558
6559 Medium *that = static_cast<Medium*>(pvUser);
6560 AssertReturn(that != NULL, VERR_GENERAL_FAILURE);
6561
6562 settings::StringsMap::const_iterator it = that->m->mapProperties.find(Utf8Str(pszName));
6563 if (it == that->m->mapProperties.end())
6564 return VERR_CFGM_VALUE_NOT_FOUND;
6565
6566 /* we interpret null values as "no value" in Medium */
6567 if (it->second.isEmpty())
6568 return VERR_CFGM_VALUE_NOT_FOUND;
6569
6570 *pcbValue = it->second.length() + 1 /* include terminator */;
6571
6572 return VINF_SUCCESS;
6573}
6574
6575/* static */
6576DECLCALLBACK(int) Medium::vdConfigQuery(void *pvUser,
6577 const char *pszName,
6578 char *pszValue,
6579 size_t cchValue)
6580{
6581 AssertReturn(VALID_PTR(pszValue), VERR_INVALID_POINTER);
6582
6583 Medium *that = static_cast<Medium*>(pvUser);
6584 AssertReturn(that != NULL, VERR_GENERAL_FAILURE);
6585
6586 settings::StringsMap::const_iterator it = that->m->mapProperties.find(Utf8Str(pszName));
6587 if (it == that->m->mapProperties.end())
6588 return VERR_CFGM_VALUE_NOT_FOUND;
6589
6590 /* we interpret null values as "no value" in Medium */
6591 if (it->second.isEmpty())
6592 return VERR_CFGM_VALUE_NOT_FOUND;
6593
6594 const Utf8Str &value = it->second;
6595 if (value.length() >= cchValue)
6596 return VERR_CFGM_NOT_ENOUGH_SPACE;
6597
6598 memcpy(pszValue, value.c_str(), value.length() + 1);
6599
6600 return VINF_SUCCESS;
6601}
6602
6603DECLCALLBACK(int) Medium::vdTcpSocketCreate(uint32_t fFlags, PVDSOCKET pSock)
6604{
6605 PVDSOCKETINT pSocketInt = NULL;
6606
6607 if ((fFlags & VD_INTERFACETCPNET_CONNECT_EXTENDED_SELECT) != 0)
6608 return VERR_NOT_SUPPORTED;
6609
6610 pSocketInt = (PVDSOCKETINT)RTMemAllocZ(sizeof(VDSOCKETINT));
6611 if (!pSocketInt)
6612 return VERR_NO_MEMORY;
6613
6614 pSocketInt->hSocket = NIL_RTSOCKET;
6615 *pSock = pSocketInt;
6616 return VINF_SUCCESS;
6617}
6618
6619DECLCALLBACK(int) Medium::vdTcpSocketDestroy(VDSOCKET Sock)
6620{
6621 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6622
6623 if (pSocketInt->hSocket != NIL_RTSOCKET)
6624 RTTcpClientCloseEx(pSocketInt->hSocket, false /*fGracefulShutdown*/);
6625
6626 RTMemFree(pSocketInt);
6627
6628 return VINF_SUCCESS;
6629}
6630
6631DECLCALLBACK(int) Medium::vdTcpClientConnect(VDSOCKET Sock, const char *pszAddress, uint32_t uPort)
6632{
6633 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6634
6635 return RTTcpClientConnect(pszAddress, uPort, &pSocketInt->hSocket);
6636}
6637
6638DECLCALLBACK(int) Medium::vdTcpClientClose(VDSOCKET Sock)
6639{
6640 int rc = VINF_SUCCESS;
6641 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6642
6643 rc = RTTcpClientCloseEx(pSocketInt->hSocket, false /*fGracefulShutdown*/);
6644 pSocketInt->hSocket = NIL_RTSOCKET;
6645 return rc;
6646}
6647
6648DECLCALLBACK(bool) Medium::vdTcpIsClientConnected(VDSOCKET Sock)
6649{
6650 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6651 return pSocketInt->hSocket != NIL_RTSOCKET;
6652}
6653
6654DECLCALLBACK(int) Medium::vdTcpSelectOne(VDSOCKET Sock, RTMSINTERVAL cMillies)
6655{
6656 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6657 return RTTcpSelectOne(pSocketInt->hSocket, cMillies);
6658}
6659
6660DECLCALLBACK(int) Medium::vdTcpRead(VDSOCKET Sock, void *pvBuffer, size_t cbBuffer, size_t *pcbRead)
6661{
6662 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6663 return RTTcpRead(pSocketInt->hSocket, pvBuffer, cbBuffer, pcbRead);
6664}
6665
6666DECLCALLBACK(int) Medium::vdTcpWrite(VDSOCKET Sock, const void *pvBuffer, size_t cbBuffer)
6667{
6668 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6669 return RTTcpWrite(pSocketInt->hSocket, pvBuffer, cbBuffer);
6670}
6671
6672DECLCALLBACK(int) Medium::vdTcpSgWrite(VDSOCKET Sock, PCRTSGBUF pSgBuf)
6673{
6674 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6675 return RTTcpSgWrite(pSocketInt->hSocket, pSgBuf);
6676}
6677
6678DECLCALLBACK(int) Medium::vdTcpFlush(VDSOCKET Sock)
6679{
6680 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6681 return RTTcpFlush(pSocketInt->hSocket);
6682}
6683
6684DECLCALLBACK(int) Medium::vdTcpSetSendCoalescing(VDSOCKET Sock, bool fEnable)
6685{
6686 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6687 return RTTcpSetSendCoalescing(pSocketInt->hSocket, fEnable);
6688}
6689
6690DECLCALLBACK(int) Medium::vdTcpGetLocalAddress(VDSOCKET Sock, PRTNETADDR pAddr)
6691{
6692 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6693 return RTTcpGetLocalAddress(pSocketInt->hSocket, pAddr);
6694}
6695
6696DECLCALLBACK(int) Medium::vdTcpGetPeerAddress(VDSOCKET Sock, PRTNETADDR pAddr)
6697{
6698 PVDSOCKETINT pSocketInt = (PVDSOCKETINT)Sock;
6699 return RTTcpGetPeerAddress(pSocketInt->hSocket, pAddr);
6700}
6701
6702/**
6703 * Starts a new thread driven by the appropriate Medium::Task::handler() method.
6704 *
6705 * @note When the task is executed by this method, IProgress::notifyComplete()
6706 * is automatically called for the progress object associated with this
6707 * task when the task is finished to signal the operation completion for
6708 * other threads asynchronously waiting for it.
6709 */
6710HRESULT Medium::startThread(Medium::Task *pTask)
6711{
6712#ifdef VBOX_WITH_MAIN_LOCK_VALIDATION
6713 /* Extreme paranoia: The calling thread should not hold the medium
6714 * tree lock or any medium lock. Since there is no separate lock class
6715 * for medium objects be even more strict: no other object locks. */
6716 Assert(!AutoLockHoldsLocksInClass(LOCKCLASS_LISTOFMEDIA));
6717 Assert(!AutoLockHoldsLocksInClass(getLockingClass()));
6718#endif
6719
6720 /// @todo use a more descriptive task name
6721 int vrc = RTThreadCreate(NULL, Medium::Task::fntMediumTask, pTask,
6722 0, RTTHREADTYPE_MAIN_HEAVY_WORKER, 0,
6723 "Medium::Task");
6724 if (RT_FAILURE(vrc))
6725 {
6726 delete pTask;
6727 return setError(E_FAIL, "Could not create Medium::Task thread (%Rrc)\n", vrc);
6728 }
6729
6730 return S_OK;
6731}
6732
6733/**
6734 * Runs Medium::Task::handler() on the current thread instead of creating
6735 * a new one.
6736 *
6737 * This call implies that it is made on another temporary thread created for
6738 * some asynchronous task. Avoid calling it from a normal thread since the task
6739 * operations are potentially lengthy and will block the calling thread in this
6740 * case.
6741 *
6742 * @note When the task is executed by this method, IProgress::notifyComplete()
6743 * is not called for the progress object associated with this task when
6744 * the task is finished. Instead, the result of the operation is returned
6745 * by this method directly and it's the caller's responsibility to
6746 * complete the progress object in this case.
6747 */
6748HRESULT Medium::runNow(Medium::Task *pTask)
6749{
6750#ifdef VBOX_WITH_MAIN_LOCK_VALIDATION
6751 /* Extreme paranoia: The calling thread should not hold the medium
6752 * tree lock or any medium lock. Since there is no separate lock class
6753 * for medium objects be even more strict: no other object locks. */
6754 Assert(!AutoLockHoldsLocksInClass(LOCKCLASS_LISTOFMEDIA));
6755 Assert(!AutoLockHoldsLocksInClass(getLockingClass()));
6756#endif
6757
6758 /* NIL_RTTHREAD indicates synchronous call. */
6759 return (HRESULT)Medium::Task::fntMediumTask(NIL_RTTHREAD, pTask);
6760}
6761
6762/**
6763 * Implementation code for the "create base" task.
6764 *
6765 * This only gets started from Medium::CreateBaseStorage() and always runs
6766 * asynchronously. As a result, we always save the VirtualBox.xml file when
6767 * we're done here.
6768 *
6769 * @param task
6770 * @return
6771 */
6772HRESULT Medium::taskCreateBaseHandler(Medium::CreateBaseTask &task)
6773{
6774 HRESULT rc = S_OK;
6775
6776 /* these parameters we need after creation */
6777 uint64_t size = 0, logicalSize = 0;
6778 MediumVariant_T variant = MediumVariant_Standard;
6779 bool fGenerateUuid = false;
6780
6781 try
6782 {
6783 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
6784
6785 /* The object may request a specific UUID (through a special form of
6786 * the setLocation() argument). Otherwise we have to generate it */
6787 Guid id = m->id;
6788
6789 fGenerateUuid = id.isZero();
6790 if (fGenerateUuid)
6791 {
6792 id.create();
6793 /* VirtualBox::registerMedium() will need UUID */
6794 unconst(m->id) = id;
6795 }
6796
6797 Utf8Str format(m->strFormat);
6798 Utf8Str location(m->strLocationFull);
6799 uint64_t capabilities = m->formatObj->i_getCapabilities();
6800 ComAssertThrow(capabilities & ( MediumFormatCapabilities_CreateFixed
6801 | MediumFormatCapabilities_CreateDynamic), E_FAIL);
6802 Assert(m->state == MediumState_Creating);
6803
6804 PVBOXHDD hdd;
6805 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
6806 ComAssertRCThrow(vrc, E_FAIL);
6807
6808 /* unlock before the potentially lengthy operation */
6809 thisLock.release();
6810
6811 try
6812 {
6813 /* ensure the directory exists */
6814 if (capabilities & MediumFormatCapabilities_File)
6815 {
6816 rc = VirtualBox::ensureFilePathExists(location, !(task.mVariant & MediumVariant_NoCreateDir) /* fCreate */);
6817 if (FAILED(rc))
6818 throw rc;
6819 }
6820
6821 VDGEOMETRY geo = { 0, 0, 0 }; /* auto-detect */
6822
6823 vrc = VDCreateBase(hdd,
6824 format.c_str(),
6825 location.c_str(),
6826 task.mSize,
6827 task.mVariant & ~MediumVariant_NoCreateDir,
6828 NULL,
6829 &geo,
6830 &geo,
6831 id.raw(),
6832 VD_OPEN_FLAGS_NORMAL | m->uOpenFlagsDef,
6833 m->vdImageIfaces,
6834 task.mVDOperationIfaces);
6835 if (RT_FAILURE(vrc))
6836 throw setError(VBOX_E_FILE_ERROR,
6837 tr("Could not create the medium storage unit '%s'%s"),
6838 location.c_str(), vdError(vrc).c_str());
6839
6840 size = VDGetFileSize(hdd, 0);
6841 logicalSize = VDGetSize(hdd, 0);
6842 unsigned uImageFlags;
6843 vrc = VDGetImageFlags(hdd, 0, &uImageFlags);
6844 if (RT_SUCCESS(vrc))
6845 variant = (MediumVariant_T)uImageFlags;
6846 }
6847 catch (HRESULT aRC) { rc = aRC; }
6848
6849 VDDestroy(hdd);
6850 }
6851 catch (HRESULT aRC) { rc = aRC; }
6852
6853 if (SUCCEEDED(rc))
6854 {
6855 /* register with mVirtualBox as the last step and move to
6856 * Created state only on success (leaving an orphan file is
6857 * better than breaking media registry consistency) */
6858 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
6859 ComObjPtr<Medium> pMedium;
6860 rc = m->pVirtualBox->registerMedium(this, &pMedium, DeviceType_HardDisk);
6861 Assert(this == pMedium);
6862 }
6863
6864 // re-acquire the lock before changing state
6865 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
6866
6867 if (SUCCEEDED(rc))
6868 {
6869 m->state = MediumState_Created;
6870
6871 m->size = size;
6872 m->logicalSize = logicalSize;
6873 m->variant = variant;
6874
6875 thisLock.release();
6876 markRegistriesModified();
6877 if (task.isAsync())
6878 {
6879 // in asynchronous mode, save settings now
6880 m->pVirtualBox->saveModifiedRegistries();
6881 }
6882 }
6883 else
6884 {
6885 /* back to NotCreated on failure */
6886 m->state = MediumState_NotCreated;
6887
6888 /* reset UUID to prevent it from being reused next time */
6889 if (fGenerateUuid)
6890 unconst(m->id).clear();
6891 }
6892
6893 return rc;
6894}
6895
6896/**
6897 * Implementation code for the "create diff" task.
6898 *
6899 * This task always gets started from Medium::createDiffStorage() and can run
6900 * synchronously or asynchronously depending on the "wait" parameter passed to
6901 * that function. If we run synchronously, the caller expects the medium
6902 * registry modification to be set before returning; otherwise (in asynchronous
6903 * mode), we save the settings ourselves.
6904 *
6905 * @param task
6906 * @return
6907 */
6908HRESULT Medium::taskCreateDiffHandler(Medium::CreateDiffTask &task)
6909{
6910 HRESULT rcTmp = S_OK;
6911
6912 const ComObjPtr<Medium> &pTarget = task.mTarget;
6913
6914 uint64_t size = 0, logicalSize = 0;
6915 MediumVariant_T variant = MediumVariant_Standard;
6916 bool fGenerateUuid = false;
6917
6918 try
6919 {
6920 /* Lock both in {parent,child} order. */
6921 AutoMultiWriteLock2 mediaLock(this, pTarget COMMA_LOCKVAL_SRC_POS);
6922
6923 /* The object may request a specific UUID (through a special form of
6924 * the setLocation() argument). Otherwise we have to generate it */
6925 Guid targetId = pTarget->m->id;
6926
6927 fGenerateUuid = targetId.isZero();
6928 if (fGenerateUuid)
6929 {
6930 targetId.create();
6931 /* VirtualBox::registerMedium() will need UUID */
6932 unconst(pTarget->m->id) = targetId;
6933 }
6934
6935 Guid id = m->id;
6936
6937 Utf8Str targetFormat(pTarget->m->strFormat);
6938 Utf8Str targetLocation(pTarget->m->strLocationFull);
6939 uint64_t capabilities = pTarget->m->formatObj->i_getCapabilities();
6940 ComAssertThrow(capabilities & MediumFormatCapabilities_CreateDynamic, E_FAIL);
6941
6942 Assert(pTarget->m->state == MediumState_Creating);
6943 Assert(m->state == MediumState_LockedRead);
6944
6945 PVBOXHDD hdd;
6946 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
6947 ComAssertRCThrow(vrc, E_FAIL);
6948
6949 /* the two media are now protected by their non-default states;
6950 * unlock the media before the potentially lengthy operation */
6951 mediaLock.release();
6952
6953 try
6954 {
6955 /* Open all media in the target chain but the last. */
6956 MediumLockList::Base::const_iterator targetListBegin =
6957 task.mpMediumLockList->GetBegin();
6958 MediumLockList::Base::const_iterator targetListEnd =
6959 task.mpMediumLockList->GetEnd();
6960 for (MediumLockList::Base::const_iterator it = targetListBegin;
6961 it != targetListEnd;
6962 ++it)
6963 {
6964 const MediumLock &mediumLock = *it;
6965 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
6966
6967 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
6968
6969 /* Skip over the target diff medium */
6970 if (pMedium->m->state == MediumState_Creating)
6971 continue;
6972
6973 /* sanity check */
6974 Assert(pMedium->m->state == MediumState_LockedRead);
6975
6976 /* Open all media in appropriate mode. */
6977 vrc = VDOpen(hdd,
6978 pMedium->m->strFormat.c_str(),
6979 pMedium->m->strLocationFull.c_str(),
6980 VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_INFO | m->uOpenFlagsDef,
6981 pMedium->m->vdImageIfaces);
6982 if (RT_FAILURE(vrc))
6983 throw setError(VBOX_E_FILE_ERROR,
6984 tr("Could not open the medium storage unit '%s'%s"),
6985 pMedium->m->strLocationFull.c_str(),
6986 vdError(vrc).c_str());
6987 }
6988
6989 /* ensure the target directory exists */
6990 if (capabilities & MediumFormatCapabilities_File)
6991 {
6992 HRESULT rc = VirtualBox::ensureFilePathExists(targetLocation, !(task.mVariant & MediumVariant_NoCreateDir) /* fCreate */);
6993 if (FAILED(rc))
6994 throw rc;
6995 }
6996
6997 vrc = VDCreateDiff(hdd,
6998 targetFormat.c_str(),
6999 targetLocation.c_str(),
7000 (task.mVariant & ~MediumVariant_NoCreateDir) | VD_IMAGE_FLAGS_DIFF,
7001 NULL,
7002 targetId.raw(),
7003 id.raw(),
7004 VD_OPEN_FLAGS_NORMAL | m->uOpenFlagsDef,
7005 pTarget->m->vdImageIfaces,
7006 task.mVDOperationIfaces);
7007 if (RT_FAILURE(vrc))
7008 throw setError(VBOX_E_FILE_ERROR,
7009 tr("Could not create the differencing medium storage unit '%s'%s"),
7010 targetLocation.c_str(), vdError(vrc).c_str());
7011
7012 size = VDGetFileSize(hdd, VD_LAST_IMAGE);
7013 logicalSize = VDGetSize(hdd, VD_LAST_IMAGE);
7014 unsigned uImageFlags;
7015 vrc = VDGetImageFlags(hdd, 0, &uImageFlags);
7016 if (RT_SUCCESS(vrc))
7017 variant = (MediumVariant_T)uImageFlags;
7018 }
7019 catch (HRESULT aRC) { rcTmp = aRC; }
7020
7021 VDDestroy(hdd);
7022 }
7023 catch (HRESULT aRC) { rcTmp = aRC; }
7024
7025 MultiResult mrc(rcTmp);
7026
7027 if (SUCCEEDED(mrc))
7028 {
7029 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
7030
7031 Assert(pTarget->m->pParent.isNull());
7032
7033 /* associate the child with the parent */
7034 pTarget->m->pParent = this;
7035 m->llChildren.push_back(pTarget);
7036
7037 /** @todo r=klaus neither target nor base() are locked,
7038 * potential race! */
7039 /* diffs for immutable media are auto-reset by default */
7040 pTarget->m->autoReset = (getBase()->m->type == MediumType_Immutable);
7041
7042 /* register with mVirtualBox as the last step and move to
7043 * Created state only on success (leaving an orphan file is
7044 * better than breaking media registry consistency) */
7045 ComObjPtr<Medium> pMedium;
7046 mrc = m->pVirtualBox->registerMedium(pTarget, &pMedium, DeviceType_HardDisk);
7047 Assert(pTarget == pMedium);
7048
7049 if (FAILED(mrc))
7050 /* break the parent association on failure to register */
7051 deparent();
7052 }
7053
7054 AutoMultiWriteLock2 mediaLock(this, pTarget COMMA_LOCKVAL_SRC_POS);
7055
7056 if (SUCCEEDED(mrc))
7057 {
7058 pTarget->m->state = MediumState_Created;
7059
7060 pTarget->m->size = size;
7061 pTarget->m->logicalSize = logicalSize;
7062 pTarget->m->variant = variant;
7063 }
7064 else
7065 {
7066 /* back to NotCreated on failure */
7067 pTarget->m->state = MediumState_NotCreated;
7068
7069 pTarget->m->autoReset = false;
7070
7071 /* reset UUID to prevent it from being reused next time */
7072 if (fGenerateUuid)
7073 unconst(pTarget->m->id).clear();
7074 }
7075
7076 // deregister the task registered in createDiffStorage()
7077 Assert(m->numCreateDiffTasks != 0);
7078 --m->numCreateDiffTasks;
7079
7080 mediaLock.release();
7081 markRegistriesModified();
7082 if (task.isAsync())
7083 {
7084 // in asynchronous mode, save settings now
7085 m->pVirtualBox->saveModifiedRegistries();
7086 }
7087
7088 /* Note that in sync mode, it's the caller's responsibility to
7089 * unlock the medium. */
7090
7091 return mrc;
7092}
7093
7094/**
7095 * Implementation code for the "merge" task.
7096 *
7097 * This task always gets started from Medium::mergeTo() and can run
7098 * synchronously or asynchronously depending on the "wait" parameter passed to
7099 * that function. If we run synchronously, the caller expects the medium
7100 * registry modification to be set before returning; otherwise (in asynchronous
7101 * mode), we save the settings ourselves.
7102 *
7103 * @param task
7104 * @return
7105 */
7106HRESULT Medium::taskMergeHandler(Medium::MergeTask &task)
7107{
7108 HRESULT rcTmp = S_OK;
7109
7110 const ComObjPtr<Medium> &pTarget = task.mTarget;
7111
7112 try
7113 {
7114 PVBOXHDD hdd;
7115 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
7116 ComAssertRCThrow(vrc, E_FAIL);
7117
7118 try
7119 {
7120 // Similar code appears in SessionMachine::onlineMergeMedium, so
7121 // if you make any changes below check whether they are applicable
7122 // in that context as well.
7123
7124 unsigned uTargetIdx = VD_LAST_IMAGE;
7125 unsigned uSourceIdx = VD_LAST_IMAGE;
7126 /* Open all media in the chain. */
7127 MediumLockList::Base::iterator lockListBegin =
7128 task.mpMediumLockList->GetBegin();
7129 MediumLockList::Base::iterator lockListEnd =
7130 task.mpMediumLockList->GetEnd();
7131 unsigned i = 0;
7132 for (MediumLockList::Base::iterator it = lockListBegin;
7133 it != lockListEnd;
7134 ++it)
7135 {
7136 MediumLock &mediumLock = *it;
7137 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
7138
7139 if (pMedium == this)
7140 uSourceIdx = i;
7141 else if (pMedium == pTarget)
7142 uTargetIdx = i;
7143
7144 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
7145
7146 /*
7147 * complex sanity (sane complexity)
7148 *
7149 * The current medium must be in the Deleting (medium is merged)
7150 * or LockedRead (parent medium) state if it is not the target.
7151 * If it is the target it must be in the LockedWrite state.
7152 */
7153 Assert( ( pMedium != pTarget
7154 && ( pMedium->m->state == MediumState_Deleting
7155 || pMedium->m->state == MediumState_LockedRead))
7156 || ( pMedium == pTarget
7157 && pMedium->m->state == MediumState_LockedWrite));
7158
7159 /*
7160 * Medium must be the target, in the LockedRead state
7161 * or Deleting state where it is not allowed to be attached
7162 * to a virtual machine.
7163 */
7164 Assert( pMedium == pTarget
7165 || pMedium->m->state == MediumState_LockedRead
7166 || ( pMedium->m->backRefs.size() == 0
7167 && pMedium->m->state == MediumState_Deleting));
7168 /* The source medium must be in Deleting state. */
7169 Assert( pMedium != this
7170 || pMedium->m->state == MediumState_Deleting);
7171
7172 unsigned uOpenFlags = VD_OPEN_FLAGS_NORMAL;
7173
7174 if ( pMedium->m->state == MediumState_LockedRead
7175 || pMedium->m->state == MediumState_Deleting)
7176 uOpenFlags = VD_OPEN_FLAGS_READONLY;
7177 if (pMedium->m->type == MediumType_Shareable)
7178 uOpenFlags |= VD_OPEN_FLAGS_SHAREABLE;
7179
7180 /* Open the medium */
7181 vrc = VDOpen(hdd,
7182 pMedium->m->strFormat.c_str(),
7183 pMedium->m->strLocationFull.c_str(),
7184 uOpenFlags | m->uOpenFlagsDef,
7185 pMedium->m->vdImageIfaces);
7186 if (RT_FAILURE(vrc))
7187 throw vrc;
7188
7189 i++;
7190 }
7191
7192 ComAssertThrow( uSourceIdx != VD_LAST_IMAGE
7193 && uTargetIdx != VD_LAST_IMAGE, E_FAIL);
7194
7195 vrc = VDMerge(hdd, uSourceIdx, uTargetIdx,
7196 task.mVDOperationIfaces);
7197 if (RT_FAILURE(vrc))
7198 throw vrc;
7199
7200 /* update parent UUIDs */
7201 if (!task.mfMergeForward)
7202 {
7203 /* we need to update UUIDs of all source's children
7204 * which cannot be part of the container at once so
7205 * add each one in there individually */
7206 if (task.mpChildrenToReparent)
7207 {
7208 MediumLockList::Base::iterator childrenBegin = task.mpChildrenToReparent->GetBegin();
7209 MediumLockList::Base::iterator childrenEnd = task.mpChildrenToReparent->GetEnd();
7210 for (MediumLockList::Base::iterator it = childrenBegin;
7211 it != childrenEnd;
7212 ++it)
7213 {
7214 Medium *pMedium = it->GetMedium();
7215 /* VD_OPEN_FLAGS_INFO since UUID is wrong yet */
7216 vrc = VDOpen(hdd,
7217 pMedium->m->strFormat.c_str(),
7218 pMedium->m->strLocationFull.c_str(),
7219 VD_OPEN_FLAGS_INFO | m->uOpenFlagsDef,
7220 pMedium->m->vdImageIfaces);
7221 if (RT_FAILURE(vrc))
7222 throw vrc;
7223
7224 vrc = VDSetParentUuid(hdd, VD_LAST_IMAGE,
7225 pTarget->m->id.raw());
7226 if (RT_FAILURE(vrc))
7227 throw vrc;
7228
7229 vrc = VDClose(hdd, false /* fDelete */);
7230 if (RT_FAILURE(vrc))
7231 throw vrc;
7232 }
7233 }
7234 }
7235 }
7236 catch (HRESULT aRC) { rcTmp = aRC; }
7237 catch (int aVRC)
7238 {
7239 rcTmp = setError(VBOX_E_FILE_ERROR,
7240 tr("Could not merge the medium '%s' to '%s'%s"),
7241 m->strLocationFull.c_str(),
7242 pTarget->m->strLocationFull.c_str(),
7243 vdError(aVRC).c_str());
7244 }
7245
7246 VDDestroy(hdd);
7247 }
7248 catch (HRESULT aRC) { rcTmp = aRC; }
7249
7250 ErrorInfoKeeper eik;
7251 MultiResult mrc(rcTmp);
7252 HRESULT rc2;
7253
7254 if (SUCCEEDED(mrc))
7255 {
7256 /* all media but the target were successfully deleted by
7257 * VDMerge; reparent the last one and uninitialize deleted media. */
7258
7259 AutoWriteLock treeLock(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
7260
7261 if (task.mfMergeForward)
7262 {
7263 /* first, unregister the target since it may become a base
7264 * medium which needs re-registration */
7265 rc2 = m->pVirtualBox->unregisterMedium(pTarget);
7266 AssertComRC(rc2);
7267
7268 /* then, reparent it and disconnect the deleted branch at
7269 * both ends (chain->parent() is source's parent) */
7270 pTarget->deparent();
7271 pTarget->m->pParent = task.mParentForTarget;
7272 if (pTarget->m->pParent)
7273 {
7274 pTarget->m->pParent->m->llChildren.push_back(pTarget);
7275 deparent();
7276 }
7277
7278 /* then, register again */
7279 ComObjPtr<Medium> pMedium;
7280 rc2 = m->pVirtualBox->registerMedium(pTarget, &pMedium,
7281 DeviceType_HardDisk);
7282 AssertComRC(rc2);
7283 }
7284 else
7285 {
7286 Assert(pTarget->getChildren().size() == 1);
7287 Medium *targetChild = pTarget->getChildren().front();
7288
7289 /* disconnect the deleted branch at the elder end */
7290 targetChild->deparent();
7291
7292 /* reparent source's children and disconnect the deleted
7293 * branch at the younger end */
7294 if (task.mpChildrenToReparent)
7295 {
7296 /* obey {parent,child} lock order */
7297 AutoWriteLock sourceLock(this COMMA_LOCKVAL_SRC_POS);
7298
7299 MediumLockList::Base::iterator childrenBegin = task.mpChildrenToReparent->GetBegin();
7300 MediumLockList::Base::iterator childrenEnd = task.mpChildrenToReparent->GetEnd();
7301 for (MediumLockList::Base::iterator it = childrenBegin;
7302 it != childrenEnd;
7303 ++it)
7304 {
7305 Medium *pMedium = it->GetMedium();
7306 AutoWriteLock childLock(pMedium COMMA_LOCKVAL_SRC_POS);
7307
7308 pMedium->deparent(); // removes pMedium from source
7309 pMedium->setParent(pTarget);
7310 }
7311 }
7312 }
7313
7314 /* unregister and uninitialize all media removed by the merge */
7315 MediumLockList::Base::iterator lockListBegin =
7316 task.mpMediumLockList->GetBegin();
7317 MediumLockList::Base::iterator lockListEnd =
7318 task.mpMediumLockList->GetEnd();
7319 for (MediumLockList::Base::iterator it = lockListBegin;
7320 it != lockListEnd;
7321 )
7322 {
7323 MediumLock &mediumLock = *it;
7324 /* Create a real copy of the medium pointer, as the medium
7325 * lock deletion below would invalidate the referenced object. */
7326 const ComObjPtr<Medium> pMedium = mediumLock.GetMedium();
7327
7328 /* The target and all media not merged (readonly) are skipped */
7329 if ( pMedium == pTarget
7330 || pMedium->m->state == MediumState_LockedRead)
7331 {
7332 ++it;
7333 continue;
7334 }
7335
7336 rc2 = pMedium->m->pVirtualBox->unregisterMedium(pMedium);
7337 AssertComRC(rc2);
7338
7339 /* now, uninitialize the deleted medium (note that
7340 * due to the Deleting state, uninit() will not touch
7341 * the parent-child relationship so we need to
7342 * uninitialize each disk individually) */
7343
7344 /* note that the operation initiator medium (which is
7345 * normally also the source medium) is a special case
7346 * -- there is one more caller added by Task to it which
7347 * we must release. Also, if we are in sync mode, the
7348 * caller may still hold an AutoCaller instance for it
7349 * and therefore we cannot uninit() it (it's therefore
7350 * the caller's responsibility) */
7351 if (pMedium == this)
7352 {
7353 Assert(getChildren().size() == 0);
7354 Assert(m->backRefs.size() == 0);
7355 task.mMediumCaller.release();
7356 }
7357
7358 /* Delete the medium lock list entry, which also releases the
7359 * caller added by MergeChain before uninit() and updates the
7360 * iterator to point to the right place. */
7361 rc2 = task.mpMediumLockList->RemoveByIterator(it);
7362 AssertComRC(rc2);
7363
7364 if (task.isAsync() || pMedium != this)
7365 pMedium->uninit();
7366 }
7367 }
7368
7369 markRegistriesModified();
7370 if (task.isAsync())
7371 {
7372 // in asynchronous mode, save settings now
7373 eik.restore();
7374 m->pVirtualBox->saveModifiedRegistries();
7375 eik.fetch();
7376 }
7377
7378 if (FAILED(mrc))
7379 {
7380 /* Here we come if either VDMerge() failed (in which case we
7381 * assume that it tried to do everything to make a further
7382 * retry possible -- e.g. not deleted intermediate media
7383 * and so on) or VirtualBox::saveRegistries() failed (where we
7384 * should have the original tree but with intermediate storage
7385 * units deleted by VDMerge()). We have to only restore states
7386 * (through the MergeChain dtor) unless we are run synchronously
7387 * in which case it's the responsibility of the caller as stated
7388 * in the mergeTo() docs. The latter also implies that we
7389 * don't own the merge chain, so release it in this case. */
7390 if (task.isAsync())
7391 cancelMergeTo(task.mpChildrenToReparent, task.mpMediumLockList);
7392 }
7393
7394 return mrc;
7395}
7396
7397/**
7398 * Implementation code for the "clone" task.
7399 *
7400 * This only gets started from Medium::CloneTo() and always runs asynchronously.
7401 * As a result, we always save the VirtualBox.xml file when we're done here.
7402 *
7403 * @param task
7404 * @return
7405 */
7406HRESULT Medium::taskCloneHandler(Medium::CloneTask &task)
7407{
7408 HRESULT rcTmp = S_OK;
7409
7410 const ComObjPtr<Medium> &pTarget = task.mTarget;
7411 const ComObjPtr<Medium> &pParent = task.mParent;
7412
7413 bool fCreatingTarget = false;
7414
7415 uint64_t size = 0, logicalSize = 0;
7416 MediumVariant_T variant = MediumVariant_Standard;
7417 bool fGenerateUuid = false;
7418
7419 try
7420 {
7421 /* Lock all in {parent,child} order. The lock is also used as a
7422 * signal from the task initiator (which releases it only after
7423 * RTThreadCreate()) that we can start the job. */
7424 AutoMultiWriteLock3 thisLock(this, pTarget, pParent COMMA_LOCKVAL_SRC_POS);
7425
7426 fCreatingTarget = pTarget->m->state == MediumState_Creating;
7427
7428 /* The object may request a specific UUID (through a special form of
7429 * the setLocation() argument). Otherwise we have to generate it */
7430 Guid targetId = pTarget->m->id;
7431
7432 fGenerateUuid = targetId.isZero();
7433 if (fGenerateUuid)
7434 {
7435 targetId.create();
7436 /* VirtualBox::registerMedium() will need UUID */
7437 unconst(pTarget->m->id) = targetId;
7438 }
7439
7440 PVBOXHDD hdd;
7441 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
7442 ComAssertRCThrow(vrc, E_FAIL);
7443
7444 try
7445 {
7446 /* Open all media in the source chain. */
7447 MediumLockList::Base::const_iterator sourceListBegin =
7448 task.mpSourceMediumLockList->GetBegin();
7449 MediumLockList::Base::const_iterator sourceListEnd =
7450 task.mpSourceMediumLockList->GetEnd();
7451 for (MediumLockList::Base::const_iterator it = sourceListBegin;
7452 it != sourceListEnd;
7453 ++it)
7454 {
7455 const MediumLock &mediumLock = *it;
7456 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
7457 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
7458
7459 /* sanity check */
7460 Assert(pMedium->m->state == MediumState_LockedRead);
7461
7462 /** Open all media in read-only mode. */
7463 vrc = VDOpen(hdd,
7464 pMedium->m->strFormat.c_str(),
7465 pMedium->m->strLocationFull.c_str(),
7466 VD_OPEN_FLAGS_READONLY | m->uOpenFlagsDef,
7467 pMedium->m->vdImageIfaces);
7468 if (RT_FAILURE(vrc))
7469 throw setError(VBOX_E_FILE_ERROR,
7470 tr("Could not open the medium storage unit '%s'%s"),
7471 pMedium->m->strLocationFull.c_str(),
7472 vdError(vrc).c_str());
7473 }
7474
7475 Utf8Str targetFormat(pTarget->m->strFormat);
7476 Utf8Str targetLocation(pTarget->m->strLocationFull);
7477 uint64_t capabilities = pTarget->m->formatObj->i_getCapabilities();
7478
7479 Assert( pTarget->m->state == MediumState_Creating
7480 || pTarget->m->state == MediumState_LockedWrite);
7481 Assert(m->state == MediumState_LockedRead);
7482 Assert( pParent.isNull()
7483 || pParent->m->state == MediumState_LockedRead);
7484
7485 /* unlock before the potentially lengthy operation */
7486 thisLock.release();
7487
7488 /* ensure the target directory exists */
7489 if (capabilities & MediumFormatCapabilities_File)
7490 {
7491 HRESULT rc = VirtualBox::ensureFilePathExists(targetLocation, !(task.mVariant & MediumVariant_NoCreateDir) /* fCreate */);
7492 if (FAILED(rc))
7493 throw rc;
7494 }
7495
7496 PVBOXHDD targetHdd;
7497 vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &targetHdd);
7498 ComAssertRCThrow(vrc, E_FAIL);
7499
7500 try
7501 {
7502 /* Open all media in the target chain. */
7503 MediumLockList::Base::const_iterator targetListBegin =
7504 task.mpTargetMediumLockList->GetBegin();
7505 MediumLockList::Base::const_iterator targetListEnd =
7506 task.mpTargetMediumLockList->GetEnd();
7507 for (MediumLockList::Base::const_iterator it = targetListBegin;
7508 it != targetListEnd;
7509 ++it)
7510 {
7511 const MediumLock &mediumLock = *it;
7512 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
7513
7514 /* If the target medium is not created yet there's no
7515 * reason to open it. */
7516 if (pMedium == pTarget && fCreatingTarget)
7517 continue;
7518
7519 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
7520
7521 /* sanity check */
7522 Assert( pMedium->m->state == MediumState_LockedRead
7523 || pMedium->m->state == MediumState_LockedWrite);
7524
7525 unsigned uOpenFlags = VD_OPEN_FLAGS_NORMAL;
7526 if (pMedium->m->state != MediumState_LockedWrite)
7527 uOpenFlags = VD_OPEN_FLAGS_READONLY;
7528 if (pMedium->m->type == MediumType_Shareable)
7529 uOpenFlags |= VD_OPEN_FLAGS_SHAREABLE;
7530
7531 /* Open all media in appropriate mode. */
7532 vrc = VDOpen(targetHdd,
7533 pMedium->m->strFormat.c_str(),
7534 pMedium->m->strLocationFull.c_str(),
7535 uOpenFlags | m->uOpenFlagsDef,
7536 pMedium->m->vdImageIfaces);
7537 if (RT_FAILURE(vrc))
7538 throw setError(VBOX_E_FILE_ERROR,
7539 tr("Could not open the medium storage unit '%s'%s"),
7540 pMedium->m->strLocationFull.c_str(),
7541 vdError(vrc).c_str());
7542 }
7543
7544 /** @todo r=klaus target isn't locked, race getting the state */
7545 if (task.midxSrcImageSame == UINT32_MAX)
7546 {
7547 vrc = VDCopy(hdd,
7548 VD_LAST_IMAGE,
7549 targetHdd,
7550 targetFormat.c_str(),
7551 (fCreatingTarget) ? targetLocation.c_str() : (char *)NULL,
7552 false /* fMoveByRename */,
7553 0 /* cbSize */,
7554 task.mVariant & ~MediumVariant_NoCreateDir,
7555 targetId.raw(),
7556 VD_OPEN_FLAGS_NORMAL | m->uOpenFlagsDef,
7557 NULL /* pVDIfsOperation */,
7558 pTarget->m->vdImageIfaces,
7559 task.mVDOperationIfaces);
7560 }
7561 else
7562 {
7563 vrc = VDCopyEx(hdd,
7564 VD_LAST_IMAGE,
7565 targetHdd,
7566 targetFormat.c_str(),
7567 (fCreatingTarget) ? targetLocation.c_str() : (char *)NULL,
7568 false /* fMoveByRename */,
7569 0 /* cbSize */,
7570 task.midxSrcImageSame,
7571 task.midxDstImageSame,
7572 task.mVariant & ~MediumVariant_NoCreateDir,
7573 targetId.raw(),
7574 VD_OPEN_FLAGS_NORMAL | m->uOpenFlagsDef,
7575 NULL /* pVDIfsOperation */,
7576 pTarget->m->vdImageIfaces,
7577 task.mVDOperationIfaces);
7578 }
7579 if (RT_FAILURE(vrc))
7580 throw setError(VBOX_E_FILE_ERROR,
7581 tr("Could not create the clone medium '%s'%s"),
7582 targetLocation.c_str(), vdError(vrc).c_str());
7583
7584 size = VDGetFileSize(targetHdd, VD_LAST_IMAGE);
7585 logicalSize = VDGetSize(targetHdd, VD_LAST_IMAGE);
7586 unsigned uImageFlags;
7587 vrc = VDGetImageFlags(targetHdd, 0, &uImageFlags);
7588 if (RT_SUCCESS(vrc))
7589 variant = (MediumVariant_T)uImageFlags;
7590 }
7591 catch (HRESULT aRC) { rcTmp = aRC; }
7592
7593 VDDestroy(targetHdd);
7594 }
7595 catch (HRESULT aRC) { rcTmp = aRC; }
7596
7597 VDDestroy(hdd);
7598 }
7599 catch (HRESULT aRC) { rcTmp = aRC; }
7600
7601 ErrorInfoKeeper eik;
7602 MultiResult mrc(rcTmp);
7603
7604 /* Only do the parent changes for newly created media. */
7605 if (SUCCEEDED(mrc) && fCreatingTarget)
7606 {
7607 /* we set mParent & children() */
7608 AutoWriteLock alock2(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
7609
7610 Assert(pTarget->m->pParent.isNull());
7611
7612 if (pParent)
7613 {
7614 /* associate the clone with the parent and deassociate
7615 * from VirtualBox */
7616 pTarget->m->pParent = pParent;
7617 pParent->m->llChildren.push_back(pTarget);
7618
7619 /* register with mVirtualBox as the last step and move to
7620 * Created state only on success (leaving an orphan file is
7621 * better than breaking media registry consistency) */
7622 eik.restore();
7623 ComObjPtr<Medium> pMedium;
7624 mrc = pParent->m->pVirtualBox->registerMedium(pTarget, &pMedium,
7625 DeviceType_HardDisk);
7626 Assert( FAILED(mrc)
7627 || pTarget == pMedium);
7628 eik.fetch();
7629
7630 if (FAILED(mrc))
7631 /* break parent association on failure to register */
7632 pTarget->deparent(); // removes target from parent
7633 }
7634 else
7635 {
7636 /* just register */
7637 eik.restore();
7638 ComObjPtr<Medium> pMedium;
7639 mrc = m->pVirtualBox->registerMedium(pTarget, &pMedium,
7640 DeviceType_HardDisk);
7641 Assert( FAILED(mrc)
7642 || pTarget == pMedium);
7643 eik.fetch();
7644 }
7645 }
7646
7647 if (fCreatingTarget)
7648 {
7649 AutoWriteLock mLock(pTarget COMMA_LOCKVAL_SRC_POS);
7650
7651 if (SUCCEEDED(mrc))
7652 {
7653 pTarget->m->state = MediumState_Created;
7654
7655 pTarget->m->size = size;
7656 pTarget->m->logicalSize = logicalSize;
7657 pTarget->m->variant = variant;
7658 }
7659 else
7660 {
7661 /* back to NotCreated on failure */
7662 pTarget->m->state = MediumState_NotCreated;
7663
7664 /* reset UUID to prevent it from being reused next time */
7665 if (fGenerateUuid)
7666 unconst(pTarget->m->id).clear();
7667 }
7668 }
7669
7670 // now, at the end of this task (always asynchronous), save the settings
7671 if (SUCCEEDED(mrc))
7672 {
7673 // save the settings
7674 markRegistriesModified();
7675 /* collect multiple errors */
7676 eik.restore();
7677 m->pVirtualBox->saveModifiedRegistries();
7678 eik.fetch();
7679 }
7680
7681 /* Everything is explicitly unlocked when the task exits,
7682 * as the task destruction also destroys the source chain. */
7683
7684 /* Make sure the source chain is released early. It could happen
7685 * that we get a deadlock in Appliance::Import when Medium::Close
7686 * is called & the source chain is released at the same time. */
7687 task.mpSourceMediumLockList->Clear();
7688
7689 return mrc;
7690}
7691
7692/**
7693 * Implementation code for the "delete" task.
7694 *
7695 * This task always gets started from Medium::deleteStorage() and can run
7696 * synchronously or asynchronously depending on the "wait" parameter passed to
7697 * that function.
7698 *
7699 * @param task
7700 * @return
7701 */
7702HRESULT Medium::taskDeleteHandler(Medium::DeleteTask &task)
7703{
7704 NOREF(task);
7705 HRESULT rc = S_OK;
7706
7707 try
7708 {
7709 /* The lock is also used as a signal from the task initiator (which
7710 * releases it only after RTThreadCreate()) that we can start the job */
7711 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
7712
7713 PVBOXHDD hdd;
7714 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
7715 ComAssertRCThrow(vrc, E_FAIL);
7716
7717 Utf8Str format(m->strFormat);
7718 Utf8Str location(m->strLocationFull);
7719
7720 /* unlock before the potentially lengthy operation */
7721 Assert(m->state == MediumState_Deleting);
7722 thisLock.release();
7723
7724 try
7725 {
7726 vrc = VDOpen(hdd,
7727 format.c_str(),
7728 location.c_str(),
7729 VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_INFO | m->uOpenFlagsDef,
7730 m->vdImageIfaces);
7731 if (RT_SUCCESS(vrc))
7732 vrc = VDClose(hdd, true /* fDelete */);
7733
7734 if (RT_FAILURE(vrc))
7735 throw setError(VBOX_E_FILE_ERROR,
7736 tr("Could not delete the medium storage unit '%s'%s"),
7737 location.c_str(), vdError(vrc).c_str());
7738
7739 }
7740 catch (HRESULT aRC) { rc = aRC; }
7741
7742 VDDestroy(hdd);
7743 }
7744 catch (HRESULT aRC) { rc = aRC; }
7745
7746 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
7747
7748 /* go to the NotCreated state even on failure since the storage
7749 * may have been already partially deleted and cannot be used any
7750 * more. One will be able to manually re-open the storage if really
7751 * needed to re-register it. */
7752 m->state = MediumState_NotCreated;
7753
7754 /* Reset UUID to prevent Create* from reusing it again */
7755 unconst(m->id).clear();
7756
7757 return rc;
7758}
7759
7760/**
7761 * Implementation code for the "reset" task.
7762 *
7763 * This always gets started asynchronously from Medium::Reset().
7764 *
7765 * @param task
7766 * @return
7767 */
7768HRESULT Medium::taskResetHandler(Medium::ResetTask &task)
7769{
7770 HRESULT rc = S_OK;
7771
7772 uint64_t size = 0, logicalSize = 0;
7773 MediumVariant_T variant = MediumVariant_Standard;
7774
7775 try
7776 {
7777 /* The lock is also used as a signal from the task initiator (which
7778 * releases it only after RTThreadCreate()) that we can start the job */
7779 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
7780
7781 /// @todo Below we use a pair of delete/create operations to reset
7782 /// the diff contents but the most efficient way will of course be
7783 /// to add a VDResetDiff() API call
7784
7785 PVBOXHDD hdd;
7786 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
7787 ComAssertRCThrow(vrc, E_FAIL);
7788
7789 Guid id = m->id;
7790 Utf8Str format(m->strFormat);
7791 Utf8Str location(m->strLocationFull);
7792
7793 Medium *pParent = m->pParent;
7794 Guid parentId = pParent->m->id;
7795 Utf8Str parentFormat(pParent->m->strFormat);
7796 Utf8Str parentLocation(pParent->m->strLocationFull);
7797
7798 Assert(m->state == MediumState_LockedWrite);
7799
7800 /* unlock before the potentially lengthy operation */
7801 thisLock.release();
7802
7803 try
7804 {
7805 /* Open all media in the target chain but the last. */
7806 MediumLockList::Base::const_iterator targetListBegin =
7807 task.mpMediumLockList->GetBegin();
7808 MediumLockList::Base::const_iterator targetListEnd =
7809 task.mpMediumLockList->GetEnd();
7810 for (MediumLockList::Base::const_iterator it = targetListBegin;
7811 it != targetListEnd;
7812 ++it)
7813 {
7814 const MediumLock &mediumLock = *it;
7815 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
7816
7817 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
7818
7819 /* sanity check, "this" is checked above */
7820 Assert( pMedium == this
7821 || pMedium->m->state == MediumState_LockedRead);
7822
7823 /* Open all media in appropriate mode. */
7824 vrc = VDOpen(hdd,
7825 pMedium->m->strFormat.c_str(),
7826 pMedium->m->strLocationFull.c_str(),
7827 VD_OPEN_FLAGS_READONLY | m->uOpenFlagsDef,
7828 pMedium->m->vdImageIfaces);
7829 if (RT_FAILURE(vrc))
7830 throw setError(VBOX_E_FILE_ERROR,
7831 tr("Could not open the medium storage unit '%s'%s"),
7832 pMedium->m->strLocationFull.c_str(),
7833 vdError(vrc).c_str());
7834
7835 /* Done when we hit the media which should be reset */
7836 if (pMedium == this)
7837 break;
7838 }
7839
7840 /* first, delete the storage unit */
7841 vrc = VDClose(hdd, true /* fDelete */);
7842 if (RT_FAILURE(vrc))
7843 throw setError(VBOX_E_FILE_ERROR,
7844 tr("Could not delete the medium storage unit '%s'%s"),
7845 location.c_str(), vdError(vrc).c_str());
7846
7847 /* next, create it again */
7848 vrc = VDOpen(hdd,
7849 parentFormat.c_str(),
7850 parentLocation.c_str(),
7851 VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_INFO | m->uOpenFlagsDef,
7852 m->vdImageIfaces);
7853 if (RT_FAILURE(vrc))
7854 throw setError(VBOX_E_FILE_ERROR,
7855 tr("Could not open the medium storage unit '%s'%s"),
7856 parentLocation.c_str(), vdError(vrc).c_str());
7857
7858 vrc = VDCreateDiff(hdd,
7859 format.c_str(),
7860 location.c_str(),
7861 /// @todo use the same medium variant as before
7862 VD_IMAGE_FLAGS_NONE,
7863 NULL,
7864 id.raw(),
7865 parentId.raw(),
7866 VD_OPEN_FLAGS_NORMAL,
7867 m->vdImageIfaces,
7868 task.mVDOperationIfaces);
7869 if (RT_FAILURE(vrc))
7870 throw setError(VBOX_E_FILE_ERROR,
7871 tr("Could not create the differencing medium storage unit '%s'%s"),
7872 location.c_str(), vdError(vrc).c_str());
7873
7874 size = VDGetFileSize(hdd, VD_LAST_IMAGE);
7875 logicalSize = VDGetSize(hdd, VD_LAST_IMAGE);
7876 unsigned uImageFlags;
7877 vrc = VDGetImageFlags(hdd, 0, &uImageFlags);
7878 if (RT_SUCCESS(vrc))
7879 variant = (MediumVariant_T)uImageFlags;
7880 }
7881 catch (HRESULT aRC) { rc = aRC; }
7882
7883 VDDestroy(hdd);
7884 }
7885 catch (HRESULT aRC) { rc = aRC; }
7886
7887 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
7888
7889 m->size = size;
7890 m->logicalSize = logicalSize;
7891 m->variant = variant;
7892
7893 /* Everything is explicitly unlocked when the task exits,
7894 * as the task destruction also destroys the media chain. */
7895
7896 return rc;
7897}
7898
7899/**
7900 * Implementation code for the "compact" task.
7901 *
7902 * @param task
7903 * @return
7904 */
7905HRESULT Medium::taskCompactHandler(Medium::CompactTask &task)
7906{
7907 HRESULT rc = S_OK;
7908
7909 /* Lock all in {parent,child} order. The lock is also used as a
7910 * signal from the task initiator (which releases it only after
7911 * RTThreadCreate()) that we can start the job. */
7912 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
7913
7914 try
7915 {
7916 PVBOXHDD hdd;
7917 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
7918 ComAssertRCThrow(vrc, E_FAIL);
7919
7920 try
7921 {
7922 /* Open all media in the chain. */
7923 MediumLockList::Base::const_iterator mediumListBegin =
7924 task.mpMediumLockList->GetBegin();
7925 MediumLockList::Base::const_iterator mediumListEnd =
7926 task.mpMediumLockList->GetEnd();
7927 MediumLockList::Base::const_iterator mediumListLast =
7928 mediumListEnd;
7929 mediumListLast--;
7930 for (MediumLockList::Base::const_iterator it = mediumListBegin;
7931 it != mediumListEnd;
7932 ++it)
7933 {
7934 const MediumLock &mediumLock = *it;
7935 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
7936 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
7937
7938 /* sanity check */
7939 if (it == mediumListLast)
7940 Assert(pMedium->m->state == MediumState_LockedWrite);
7941 else
7942 Assert(pMedium->m->state == MediumState_LockedRead);
7943
7944 /* Open all media but last in read-only mode. Do not handle
7945 * shareable media, as compaction and sharing are mutually
7946 * exclusive. */
7947 vrc = VDOpen(hdd,
7948 pMedium->m->strFormat.c_str(),
7949 pMedium->m->strLocationFull.c_str(),
7950 m->uOpenFlagsDef | (it == mediumListLast ? VD_OPEN_FLAGS_NORMAL : VD_OPEN_FLAGS_READONLY),
7951 pMedium->m->vdImageIfaces);
7952 if (RT_FAILURE(vrc))
7953 throw setError(VBOX_E_FILE_ERROR,
7954 tr("Could not open the medium storage unit '%s'%s"),
7955 pMedium->m->strLocationFull.c_str(),
7956 vdError(vrc).c_str());
7957 }
7958
7959 Assert(m->state == MediumState_LockedWrite);
7960
7961 Utf8Str location(m->strLocationFull);
7962
7963 /* unlock before the potentially lengthy operation */
7964 thisLock.release();
7965
7966 vrc = VDCompact(hdd, VD_LAST_IMAGE, task.mVDOperationIfaces);
7967 if (RT_FAILURE(vrc))
7968 {
7969 if (vrc == VERR_NOT_SUPPORTED)
7970 throw setError(VBOX_E_NOT_SUPPORTED,
7971 tr("Compacting is not yet supported for medium '%s'"),
7972 location.c_str());
7973 else if (vrc == VERR_NOT_IMPLEMENTED)
7974 throw setError(E_NOTIMPL,
7975 tr("Compacting is not implemented, medium '%s'"),
7976 location.c_str());
7977 else
7978 throw setError(VBOX_E_FILE_ERROR,
7979 tr("Could not compact medium '%s'%s"),
7980 location.c_str(),
7981 vdError(vrc).c_str());
7982 }
7983 }
7984 catch (HRESULT aRC) { rc = aRC; }
7985
7986 VDDestroy(hdd);
7987 }
7988 catch (HRESULT aRC) { rc = aRC; }
7989
7990 /* Everything is explicitly unlocked when the task exits,
7991 * as the task destruction also destroys the media chain. */
7992
7993 return rc;
7994}
7995
7996/**
7997 * Implementation code for the "resize" task.
7998 *
7999 * @param task
8000 * @return
8001 */
8002HRESULT Medium::taskResizeHandler(Medium::ResizeTask &task)
8003{
8004 HRESULT rc = S_OK;
8005
8006 uint64_t size = 0, logicalSize = 0;
8007
8008 try
8009 {
8010 /* The lock is also used as a signal from the task initiator (which
8011 * releases it only after RTThreadCreate()) that we can start the job */
8012 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
8013
8014 PVBOXHDD hdd;
8015 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
8016 ComAssertRCThrow(vrc, E_FAIL);
8017
8018 try
8019 {
8020 /* Open all media in the chain. */
8021 MediumLockList::Base::const_iterator mediumListBegin =
8022 task.mpMediumLockList->GetBegin();
8023 MediumLockList::Base::const_iterator mediumListEnd =
8024 task.mpMediumLockList->GetEnd();
8025 MediumLockList::Base::const_iterator mediumListLast =
8026 mediumListEnd;
8027 mediumListLast--;
8028 for (MediumLockList::Base::const_iterator it = mediumListBegin;
8029 it != mediumListEnd;
8030 ++it)
8031 {
8032 const MediumLock &mediumLock = *it;
8033 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
8034 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
8035
8036 /* sanity check */
8037 if (it == mediumListLast)
8038 Assert(pMedium->m->state == MediumState_LockedWrite);
8039 else
8040 Assert(pMedium->m->state == MediumState_LockedRead);
8041
8042 /* Open all media but last in read-only mode. Do not handle
8043 * shareable media, as compaction and sharing are mutually
8044 * exclusive. */
8045 vrc = VDOpen(hdd,
8046 pMedium->m->strFormat.c_str(),
8047 pMedium->m->strLocationFull.c_str(),
8048 m->uOpenFlagsDef | (it == mediumListLast ? VD_OPEN_FLAGS_NORMAL : VD_OPEN_FLAGS_READONLY),
8049 pMedium->m->vdImageIfaces);
8050 if (RT_FAILURE(vrc))
8051 throw setError(VBOX_E_FILE_ERROR,
8052 tr("Could not open the medium storage unit '%s'%s"),
8053 pMedium->m->strLocationFull.c_str(),
8054 vdError(vrc).c_str());
8055 }
8056
8057 Assert(m->state == MediumState_LockedWrite);
8058
8059 Utf8Str location(m->strLocationFull);
8060
8061 /* unlock before the potentially lengthy operation */
8062 thisLock.release();
8063
8064 VDGEOMETRY geo = {0, 0, 0}; /* auto */
8065 vrc = VDResize(hdd, task.mSize, &geo, &geo, task.mVDOperationIfaces);
8066 if (RT_FAILURE(vrc))
8067 {
8068 if (vrc == VERR_NOT_SUPPORTED)
8069 throw setError(VBOX_E_NOT_SUPPORTED,
8070 tr("Resizing to new size %llu is not yet supported for medium '%s'"),
8071 task.mSize, location.c_str());
8072 else if (vrc == VERR_NOT_IMPLEMENTED)
8073 throw setError(E_NOTIMPL,
8074 tr("Resiting is not implemented, medium '%s'"),
8075 location.c_str());
8076 else
8077 throw setError(VBOX_E_FILE_ERROR,
8078 tr("Could not resize medium '%s'%s"),
8079 location.c_str(),
8080 vdError(vrc).c_str());
8081 }
8082 size = VDGetFileSize(hdd, VD_LAST_IMAGE);
8083 logicalSize = VDGetSize(hdd, VD_LAST_IMAGE);
8084 }
8085 catch (HRESULT aRC) { rc = aRC; }
8086
8087 VDDestroy(hdd);
8088 }
8089 catch (HRESULT aRC) { rc = aRC; }
8090
8091 if (SUCCEEDED(rc))
8092 {
8093 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
8094 m->size = size;
8095 m->logicalSize = logicalSize;
8096 }
8097
8098 /* Everything is explicitly unlocked when the task exits,
8099 * as the task destruction also destroys the media chain. */
8100
8101 return rc;
8102}
8103
8104/**
8105 * Implementation code for the "export" task.
8106 *
8107 * This only gets started from Medium::exportFile() and always runs
8108 * asynchronously. It doesn't touch anything configuration related, so
8109 * we never save the VirtualBox.xml file here.
8110 *
8111 * @param task
8112 * @return
8113 */
8114HRESULT Medium::taskExportHandler(Medium::ExportTask &task)
8115{
8116 HRESULT rc = S_OK;
8117
8118 try
8119 {
8120 /* Lock all in {parent,child} order. The lock is also used as a
8121 * signal from the task initiator (which releases it only after
8122 * RTThreadCreate()) that we can start the job. */
8123 AutoWriteLock thisLock(this COMMA_LOCKVAL_SRC_POS);
8124
8125 PVBOXHDD hdd;
8126 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
8127 ComAssertRCThrow(vrc, E_FAIL);
8128
8129 try
8130 {
8131 /* Open all media in the source chain. */
8132 MediumLockList::Base::const_iterator sourceListBegin =
8133 task.mpSourceMediumLockList->GetBegin();
8134 MediumLockList::Base::const_iterator sourceListEnd =
8135 task.mpSourceMediumLockList->GetEnd();
8136 for (MediumLockList::Base::const_iterator it = sourceListBegin;
8137 it != sourceListEnd;
8138 ++it)
8139 {
8140 const MediumLock &mediumLock = *it;
8141 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
8142 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
8143
8144 /* sanity check */
8145 Assert(pMedium->m->state == MediumState_LockedRead);
8146
8147 /* Open all media in read-only mode. */
8148 vrc = VDOpen(hdd,
8149 pMedium->m->strFormat.c_str(),
8150 pMedium->m->strLocationFull.c_str(),
8151 VD_OPEN_FLAGS_READONLY | m->uOpenFlagsDef,
8152 pMedium->m->vdImageIfaces);
8153 if (RT_FAILURE(vrc))
8154 throw setError(VBOX_E_FILE_ERROR,
8155 tr("Could not open the medium storage unit '%s'%s"),
8156 pMedium->m->strLocationFull.c_str(),
8157 vdError(vrc).c_str());
8158 }
8159
8160 Utf8Str targetFormat(task.mFormat->i_getId());
8161 Utf8Str targetLocation(task.mFilename);
8162 uint64_t capabilities = task.mFormat->i_getCapabilities();
8163
8164 Assert(m->state == MediumState_LockedRead);
8165
8166 /* unlock before the potentially lengthy operation */
8167 thisLock.release();
8168
8169 /* ensure the target directory exists */
8170 if (capabilities & MediumFormatCapabilities_File)
8171 {
8172 rc = VirtualBox::ensureFilePathExists(targetLocation, !(task.mVariant & MediumVariant_NoCreateDir) /* fCreate */);
8173 if (FAILED(rc))
8174 throw rc;
8175 }
8176
8177 PVBOXHDD targetHdd;
8178 vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &targetHdd);
8179 ComAssertRCThrow(vrc, E_FAIL);
8180
8181 try
8182 {
8183 vrc = VDCopy(hdd,
8184 VD_LAST_IMAGE,
8185 targetHdd,
8186 targetFormat.c_str(),
8187 targetLocation.c_str(),
8188 false /* fMoveByRename */,
8189 0 /* cbSize */,
8190 task.mVariant & ~MediumVariant_NoCreateDir,
8191 NULL /* pDstUuid */,
8192 VD_OPEN_FLAGS_NORMAL | VD_OPEN_FLAGS_SEQUENTIAL,
8193 NULL /* pVDIfsOperation */,
8194 task.mVDImageIfaces,
8195 task.mVDOperationIfaces);
8196 if (RT_FAILURE(vrc))
8197 throw setError(VBOX_E_FILE_ERROR,
8198 tr("Could not create the clone medium '%s'%s"),
8199 targetLocation.c_str(), vdError(vrc).c_str());
8200 }
8201 catch (HRESULT aRC) { rc = aRC; }
8202
8203 VDDestroy(targetHdd);
8204 }
8205 catch (HRESULT aRC) { rc = aRC; }
8206
8207 VDDestroy(hdd);
8208 }
8209 catch (HRESULT aRC) { rc = aRC; }
8210
8211 /* Everything is explicitly unlocked when the task exits,
8212 * as the task destruction also destroys the source chain. */
8213
8214 /* Make sure the source chain is released early, otherwise it can
8215 * lead to deadlocks with concurrent IAppliance activities. */
8216 task.mpSourceMediumLockList->Clear();
8217
8218 return rc;
8219}
8220
8221/**
8222 * Implementation code for the "import" task.
8223 *
8224 * This only gets started from Medium::importFile() and always runs
8225 * asynchronously. It potentially touches the media registry, so we
8226 * always save the VirtualBox.xml file when we're done here.
8227 *
8228 * @param task
8229 * @return
8230 */
8231HRESULT Medium::taskImportHandler(Medium::ImportTask &task)
8232{
8233 HRESULT rcTmp = S_OK;
8234
8235 const ComObjPtr<Medium> &pParent = task.mParent;
8236
8237 bool fCreatingTarget = false;
8238
8239 uint64_t size = 0, logicalSize = 0;
8240 MediumVariant_T variant = MediumVariant_Standard;
8241 bool fGenerateUuid = false;
8242
8243 try
8244 {
8245 /* Lock all in {parent,child} order. The lock is also used as a
8246 * signal from the task initiator (which releases it only after
8247 * RTThreadCreate()) that we can start the job. */
8248 AutoMultiWriteLock2 thisLock(this, pParent COMMA_LOCKVAL_SRC_POS);
8249
8250 fCreatingTarget = m->state == MediumState_Creating;
8251
8252 /* The object may request a specific UUID (through a special form of
8253 * the setLocation() argument). Otherwise we have to generate it */
8254 Guid targetId = m->id;
8255
8256 fGenerateUuid = targetId.isZero();
8257 if (fGenerateUuid)
8258 {
8259 targetId.create();
8260 /* VirtualBox::registerMedium() will need UUID */
8261 unconst(m->id) = targetId;
8262 }
8263
8264
8265 PVBOXHDD hdd;
8266 int vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &hdd);
8267 ComAssertRCThrow(vrc, E_FAIL);
8268
8269 try
8270 {
8271 /* Open source medium. */
8272 vrc = VDOpen(hdd,
8273 task.mFormat->i_getId().c_str(),
8274 task.mFilename.c_str(),
8275 VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_SEQUENTIAL | m->uOpenFlagsDef,
8276 task.mVDImageIfaces);
8277 if (RT_FAILURE(vrc))
8278 throw setError(VBOX_E_FILE_ERROR,
8279 tr("Could not open the medium storage unit '%s'%s"),
8280 task.mFilename.c_str(),
8281 vdError(vrc).c_str());
8282
8283 Utf8Str targetFormat(m->strFormat);
8284 Utf8Str targetLocation(m->strLocationFull);
8285 uint64_t capabilities = task.mFormat->i_getCapabilities();
8286
8287 Assert( m->state == MediumState_Creating
8288 || m->state == MediumState_LockedWrite);
8289 Assert( pParent.isNull()
8290 || pParent->m->state == MediumState_LockedRead);
8291
8292 /* unlock before the potentially lengthy operation */
8293 thisLock.release();
8294
8295 /* ensure the target directory exists */
8296 if (capabilities & MediumFormatCapabilities_File)
8297 {
8298 HRESULT rc = VirtualBox::ensureFilePathExists(targetLocation, !(task.mVariant & MediumVariant_NoCreateDir) /* fCreate */);
8299 if (FAILED(rc))
8300 throw rc;
8301 }
8302
8303 PVBOXHDD targetHdd;
8304 vrc = VDCreate(m->vdDiskIfaces, convertDeviceType(), &targetHdd);
8305 ComAssertRCThrow(vrc, E_FAIL);
8306
8307 try
8308 {
8309 /* Open all media in the target chain. */
8310 MediumLockList::Base::const_iterator targetListBegin =
8311 task.mpTargetMediumLockList->GetBegin();
8312 MediumLockList::Base::const_iterator targetListEnd =
8313 task.mpTargetMediumLockList->GetEnd();
8314 for (MediumLockList::Base::const_iterator it = targetListBegin;
8315 it != targetListEnd;
8316 ++it)
8317 {
8318 const MediumLock &mediumLock = *it;
8319 const ComObjPtr<Medium> &pMedium = mediumLock.GetMedium();
8320
8321 /* If the target medium is not created yet there's no
8322 * reason to open it. */
8323 if (pMedium == this && fCreatingTarget)
8324 continue;
8325
8326 AutoReadLock alock(pMedium COMMA_LOCKVAL_SRC_POS);
8327
8328 /* sanity check */
8329 Assert( pMedium->m->state == MediumState_LockedRead
8330 || pMedium->m->state == MediumState_LockedWrite);
8331
8332 unsigned uOpenFlags = VD_OPEN_FLAGS_NORMAL;
8333 if (pMedium->m->state != MediumState_LockedWrite)
8334 uOpenFlags = VD_OPEN_FLAGS_READONLY;
8335 if (pMedium->m->type == MediumType_Shareable)
8336 uOpenFlags |= VD_OPEN_FLAGS_SHAREABLE;
8337
8338 /* Open all media in appropriate mode. */
8339 vrc = VDOpen(targetHdd,
8340 pMedium->m->strFormat.c_str(),
8341 pMedium->m->strLocationFull.c_str(),
8342 uOpenFlags | m->uOpenFlagsDef,
8343 pMedium->m->vdImageIfaces);
8344 if (RT_FAILURE(vrc))
8345 throw setError(VBOX_E_FILE_ERROR,
8346 tr("Could not open the medium storage unit '%s'%s"),
8347 pMedium->m->strLocationFull.c_str(),
8348 vdError(vrc).c_str());
8349 }
8350
8351 /** @todo r=klaus target isn't locked, race getting the state */
8352 vrc = VDCopy(hdd,
8353 VD_LAST_IMAGE,
8354 targetHdd,
8355 targetFormat.c_str(),
8356 (fCreatingTarget) ? targetLocation.c_str() : (char *)NULL,
8357 false /* fMoveByRename */,
8358 0 /* cbSize */,
8359 task.mVariant & ~MediumVariant_NoCreateDir,
8360 targetId.raw(),
8361 VD_OPEN_FLAGS_NORMAL,
8362 NULL /* pVDIfsOperation */,
8363 m->vdImageIfaces,
8364 task.mVDOperationIfaces);
8365 if (RT_FAILURE(vrc))
8366 throw setError(VBOX_E_FILE_ERROR,
8367 tr("Could not create the clone medium '%s'%s"),
8368 targetLocation.c_str(), vdError(vrc).c_str());
8369
8370 size = VDGetFileSize(targetHdd, VD_LAST_IMAGE);
8371 logicalSize = VDGetSize(targetHdd, VD_LAST_IMAGE);
8372 unsigned uImageFlags;
8373 vrc = VDGetImageFlags(targetHdd, 0, &uImageFlags);
8374 if (RT_SUCCESS(vrc))
8375 variant = (MediumVariant_T)uImageFlags;
8376 }
8377 catch (HRESULT aRC) { rcTmp = aRC; }
8378
8379 VDDestroy(targetHdd);
8380 }
8381 catch (HRESULT aRC) { rcTmp = aRC; }
8382
8383 VDDestroy(hdd);
8384 }
8385 catch (HRESULT aRC) { rcTmp = aRC; }
8386
8387 ErrorInfoKeeper eik;
8388 MultiResult mrc(rcTmp);
8389
8390 /* Only do the parent changes for newly created media. */
8391 if (SUCCEEDED(mrc) && fCreatingTarget)
8392 {
8393 /* we set mParent & children() */
8394 AutoWriteLock alock2(m->pVirtualBox->getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS);
8395
8396 Assert(m->pParent.isNull());
8397
8398 if (pParent)
8399 {
8400 /* associate the clone with the parent and deassociate
8401 * from VirtualBox */
8402 m->pParent = pParent;
8403 pParent->m->llChildren.push_back(this);
8404
8405 /* register with mVirtualBox as the last step and move to
8406 * Created state only on success (leaving an orphan file is
8407 * better than breaking media registry consistency) */
8408 eik.restore();
8409 ComObjPtr<Medium> pMedium;
8410 mrc = pParent->m->pVirtualBox->registerMedium(this, &pMedium,
8411 DeviceType_HardDisk);
8412 Assert(this == pMedium);
8413 eik.fetch();
8414
8415 if (FAILED(mrc))
8416 /* break parent association on failure to register */
8417 this->deparent(); // removes target from parent
8418 }
8419 else
8420 {
8421 /* just register */
8422 eik.restore();
8423 ComObjPtr<Medium> pMedium;
8424 mrc = m->pVirtualBox->registerMedium(this, &pMedium, DeviceType_HardDisk);
8425 Assert(this == pMedium);
8426 eik.fetch();
8427 }
8428 }
8429
8430 if (fCreatingTarget)
8431 {
8432 AutoWriteLock mLock(this COMMA_LOCKVAL_SRC_POS);
8433
8434 if (SUCCEEDED(mrc))
8435 {
8436 m->state = MediumState_Created;
8437
8438 m->size = size;
8439 m->logicalSize = logicalSize;
8440 m->variant = variant;
8441 }
8442 else
8443 {
8444 /* back to NotCreated on failure */
8445 m->state = MediumState_NotCreated;
8446
8447 /* reset UUID to prevent it from being reused next time */
8448 if (fGenerateUuid)
8449 unconst(m->id).clear();
8450 }
8451 }
8452
8453 // now, at the end of this task (always asynchronous), save the settings
8454 {
8455 // save the settings
8456 markRegistriesModified();
8457 /* collect multiple errors */
8458 eik.restore();
8459 m->pVirtualBox->saveModifiedRegistries();
8460 eik.fetch();
8461 }
8462
8463 /* Everything is explicitly unlocked when the task exits,
8464 * as the task destruction also destroys the target chain. */
8465
8466 /* Make sure the target chain is released early, otherwise it can
8467 * lead to deadlocks with concurrent IAppliance activities. */
8468 task.mpTargetMediumLockList->Clear();
8469
8470 return mrc;
8471}
8472
8473/* vi: set tabstop=4 shiftwidth=4 expandtab: */
Note: See TracBrowser for help on using the repository browser.

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