MachineImplMoveVM.cpp@ 100841

Last change on this file since 100841 was 98352, checked in by vboxsync, 23 months ago
Main: Fix identifiers containing an incorrect plural of "medium".
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 66.6 KB

Line
1	/* $Id: MachineImplMoveVM.cpp 98352 2023-01-30 19:44:51Z vboxsync $ */
2	/** @file
3	* Implementation of MachineMoveVM
4	*/
5
6	/*
7	* Copyright (C) 2011-2023 Oracle and/or its affiliates.
8	*
9	* This file is part of VirtualBox base platform packages, as
10	* available from https://www.virtualbox.org.
11	*
12	* This program is free software; you can redistribute it and/or
13	* modify it under the terms of the GNU General Public License
14	* as published by the Free Software Foundation, in version 3 of the
15	* License.
16	*
17	* This program is distributed in the hope that it will be useful, but
18	* WITHOUT ANY WARRANTY; without even the implied warranty of
19	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20	* General Public License for more details.
21	*
22	* You should have received a copy of the GNU General Public License
23	* along with this program; if not, see <https://www.gnu.org/licenses>.
24	*
25	* SPDX-License-Identifier: GPL-3.0-only
26	*/
27
28	#define LOG_GROUP LOG_GROUP_MAIN_MACHINE
29	#include <iprt/fs.h>
30	#include <iprt/dir.h>
31	#include <iprt/file.h>
32	#include <iprt/path.h>
33	#include <iprt/cpp/utils.h>
34	#include <iprt/stream.h>
35	#include <VBox/com/ErrorInfo.h>
36
37	#include "MachineImplMoveVM.h"
38	#include "SnapshotImpl.h"
39	#include "MediumFormatImpl.h"
40	#include "VirtualBoxImpl.h"
41	#include "LoggingNew.h"
42
43	typedef std::multimap<Utf8Str, Utf8Str> list_t;
44	typedef std::multimap<Utf8Str, Utf8Str>::const_iterator cit_t;
45	typedef std::multimap<Utf8Str, Utf8Str>::iterator it_t;
46	typedef std::pair <std::multimap<Utf8Str, Utf8Str>::iterator, std::multimap<Utf8Str, Utf8Str>::iterator> rangeRes_t;
47
48	struct fileList_t
49	{
50	HRESULT add(const Utf8Str &folder, const Utf8Str &file)
51	{
52	m_list.insert(std::make_pair(folder, file));
53	return S_OK;
54	}
55
56	HRESULT add(const Utf8Str &fullPath)
57	{
58	Utf8Str folder = fullPath;
59	folder.stripFilename();
60	Utf8Str filename = fullPath;
61	filename.stripPath();
62	m_list.insert(std::make_pair(folder, filename));
63	return S_OK;
64	}
65
66	HRESULT removeFileFromList(const Utf8Str &fullPath)
67	{
68	Utf8Str folder = fullPath;
69	folder.stripFilename();
70	Utf8Str filename = fullPath;
71	filename.stripPath();
72	rangeRes_t res = m_list.equal_range(folder);
73	for (it_t it=res.first; it!=res.second;)
74	{
75	if (it->second.equals(filename))
76	{
77	it_t it2 = it;
78	++it;
79	m_list.erase(it2);
80	}
81	else
82	++it;
83	}
84
85	return S_OK;
86	}
87
88	HRESULT removeFileFromList(const Utf8Str &path, const Utf8Str &fileName)
89	{
90	rangeRes_t res = m_list.equal_range(path);
91	for (it_t it=res.first; it!=res.second;)
92	{
93	if (it->second.equals(fileName))
94	{
95	it_t it2 = it;
96	++it;
97	m_list.erase(it2);
98	}
99	else
100	++it;
101	}
102	return S_OK;
103	}
104
105	HRESULT removeFolderFromList(const Utf8Str &path)
106	{
107	m_list.erase(path);
108	return S_OK;
109	}
110
111	rangeRes_t getFilesInRange(const Utf8Str &path)
112	{
113	rangeRes_t res;
114	res = m_list.equal_range(path);
115	return res;
116	}
117
118	std::list<Utf8Str> getFilesInList(const Utf8Str &path)
119	{
120	std::list<Utf8Str> list_;
121	rangeRes_t res = m_list.equal_range(path);
122	for (it_t it=res.first; it!=res.second; ++it)
123	list_.push_back(it->second);
124	return list_;
125	}
126
127
128	list_t m_list;
129
130	};
131
132
133	HRESULT MachineMoveVM::init()
134	{
135	HRESULT hrc = S_OK;
136
137	Utf8Str strTargetFolder;
138	/* adding a trailing slash if it's needed */
139	{
140	size_t len = m_targetPath.length() + 2;
141	if (len >= RTPATH_MAX)
142	return m_pMachine->setError(VBOX_E_IPRT_ERROR, tr("The destination path exceeds the maximum value."));
143
144	/** @todo r=bird: I need to add a Utf8Str method or iprt/cxx/path.h thingy
145	* for doing this. We need this often and code like this doesn't
146	* need to be repeated and re-optimized in each instance... */
147	char *path = new char [len];
148	RTStrCopy(path, len, m_targetPath.c_str());
149	RTPathEnsureTrailingSeparator(path, len);
150	strTargetFolder = m_targetPath = path;
151	delete[] path;
152	}
153
154	/*
155	* We have a mode which user is able to request
156	* basic mode:
157	* - The images which are solely attached to the VM
158	* and located in the original VM folder will be moved.
159	*
160	* Comment: in the future some other modes can be added.
161	*/
162
163	RTFOFF cbTotal = 0;
164	RTFOFF cbFree = 0;
165	uint32_t cbBlock = 0;
166	uint32_t cbSector = 0;
167
168
169	int vrc = RTFsQuerySizes(strTargetFolder.c_str(), &cbTotal, &cbFree, &cbBlock, &cbSector);
170	if (RT_FAILURE(vrc))
171	return m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
172	tr("Unable to determine free space at move destination ('%s'): %Rrc"),
173	strTargetFolder.c_str(), vrc);
174
175	RTDIR hDir;
176	vrc = RTDirOpen(&hDir, strTargetFolder.c_str());
177	if (RT_FAILURE(vrc))
178	return m_pMachine->setErrorVrc(vrc);
179
180	Utf8Str strTempFile = strTargetFolder + "test.txt";
181	RTFILE hFile;
182	vrc = RTFileOpen(&hFile, strTempFile.c_str(), RTFILE_O_OPEN_CREATE \| RTFILE_O_READWRITE \| RTFILE_O_DENY_NONE);
183	if (RT_FAILURE(vrc))
184	{
185	RTDirClose(hDir);
186	return m_pMachine->setErrorVrc(vrc,
187	tr("Can't create a test file test.txt in the %s. Check the access rights of the destination folder."),
188	strTargetFolder.c_str());
189	}
190
191	/** @todo r=vvp: Do we need to check each return result here? Looks excessively.
192	* And it's not so important for the test file.
193	* bird: I'd just do AssertRC on the same line, though the deletion
194	* of the test is a little important. */
195	vrc = RTFileClose(hFile); AssertRC(vrc);
196	RTFileDelete(strTempFile.c_str());
197	vrc = RTDirClose(hDir); AssertRC(vrc);
198
199	Log2(("blocks: total %RTfoff, free %RTfoff\n", cbTotal, cbFree));
200	Log2(("total space (Kb) %RTfoff (Mb) %RTfoff (Gb) %RTfoff\n", cbTotal/_1K, cbTotal/_1M, cbTotal/_1G));
201	Log2(("total free space (Kb) %RTfoff (Mb) %RTfoff (Gb) %RTfoff\n", cbFree/_1K, cbFree/_1M, cbFree/_1G));
202
203	RTFSPROPERTIES properties;
204	vrc = RTFsQueryProperties(strTargetFolder.c_str(), &properties);
205	if (RT_FAILURE(vrc))
206	return m_pMachine->setErrorVrc(vrc, "RTFsQueryProperties(%s): %Rrc", strTargetFolder.c_str(), vrc);
207
208	Log2(("disk properties: remote=%RTbool read only=%RTbool compressed=%RTbool\n",
209	properties.fRemote, properties.fReadOnly, properties.fCompressed));
210
211	/* Get the original VM path */
212	Utf8Str strSettingsFilePath;
213	Bstr bstr_settingsFilePath;
214	hrc = m_pMachine->COMGETTER(SettingsFilePath)(bstr_settingsFilePath.asOutParam());
215	if (FAILED(hrc))
216	return hrc;
217
218	strSettingsFilePath = bstr_settingsFilePath;
219	strSettingsFilePath.stripFilename();
220
221	m_vmFolders.insert(std::make_pair(VBox_SettingFolder, strSettingsFilePath));
222
223	/* Collect all files from the VM's folder */
224	fileList_t fullFileList;
225	hrc = getFilesList(strSettingsFilePath, fullFileList);
226	if (FAILED(hrc))
227	return hrc;
228
229	/*
230	* Collect all known folders used by the VM:
231	* - log folder;
232	* - state folder;
233	* - snapshot folder.
234	*/
235	Utf8Str strLogFolder;
236	Bstr bstr_logFolder;
237	hrc = m_pMachine->COMGETTER(LogFolder)(bstr_logFolder.asOutParam());
238	if (FAILED(hrc))
239	return hrc;
240
241	strLogFolder = bstr_logFolder;
242	if ( m_type.equals("basic")
243	&& RTPathStartsWith(strLogFolder.c_str(), strSettingsFilePath.c_str()))
244	m_vmFolders.insert(std::make_pair(VBox_LogFolder, strLogFolder));
245
246	Utf8Str strStateFilePath;
247	Bstr bstr_stateFilePath;
248	MachineState_T machineState;
249	hrc = m_pMachine->COMGETTER(State)(&machineState);
250	if (FAILED(hrc))
251	return hrc;
252
253	if (machineState == MachineState_Saved \|\| machineState == MachineState_AbortedSaved)
254	{
255	m_pMachine->COMGETTER(StateFilePath)(bstr_stateFilePath.asOutParam());
256	strStateFilePath = bstr_stateFilePath;
257	strStateFilePath.stripFilename();
258	if ( m_type.equals("basic")
259	&& RTPathStartsWith(strStateFilePath.c_str(), strSettingsFilePath.c_str()))
260	m_vmFolders.insert(std::make_pair(VBox_StateFolder, strStateFilePath));
261	}
262
263	Utf8Str strSnapshotFolder;
264	Bstr bstr_snapshotFolder;
265	hrc = m_pMachine->COMGETTER(SnapshotFolder)(bstr_snapshotFolder.asOutParam());
266	if (FAILED(hrc))
267	return hrc;
268
269	strSnapshotFolder = bstr_snapshotFolder;
270	if ( m_type.equals("basic")
271	&& RTPathStartsWith(strSnapshotFolder.c_str(), strSettingsFilePath.c_str()))
272	m_vmFolders.insert(std::make_pair(VBox_SnapshotFolder, strSnapshotFolder));
273
274	if (m_pMachine->i_isSnapshotMachine())
275	{
276	Bstr bstrSrcMachineId;
277	hrc = m_pMachine->COMGETTER(Id)(bstrSrcMachineId.asOutParam());
278	if (FAILED(hrc))
279	return hrc;
280
281	ComPtr<IMachine> newSrcMachine;
282	hrc = m_pMachine->i_getVirtualBox()->FindMachine(bstrSrcMachineId.raw(), newSrcMachine.asOutParam());
283	if (FAILED(hrc))
284	return hrc;
285	}
286
287	/* Add the current machine and all snapshot machines below this machine
288	* in a list for further processing.
289	*/
290
291	int64_t neededFreeSpace = 0;
292
293	/* Actual file list */
294	fileList_t actualFileList;
295	Utf8Str strTargetImageName;
296
297	machineList.push_back(m_pMachine);
298
299	{
300	ULONG cSnapshots = 0;
301	hrc = m_pMachine->COMGETTER(SnapshotCount)(&cSnapshots);
302	if (FAILED(hrc))
303	return hrc;
304
305	if (cSnapshots > 0)
306	{
307	Utf8Str id;
308	if (m_pMachine->i_isSnapshotMachine())
309	id = m_pMachine->i_getSnapshotId().toString();
310	ComPtr<ISnapshot> pSnapshot;
311	hrc = m_pMachine->FindSnapshot(Bstr(id).raw(), pSnapshot.asOutParam());
312	if (FAILED(hrc))
313	return hrc;
314	hrc = createMachineList(pSnapshot);
315	if (FAILED(hrc))
316	return hrc;
317	}
318	}
319
320	ULONG uCount = 1;//looks like it should be initialized by 1. See assertion in the Progress::setNextOperation()
321	ULONG uTotalWeight = 1;
322
323	/* The lists m_llMedia, m_llSaveStateFiles and m_llNVRAMFiles are filled in the queryMediaForAllStates() */
324	hrc = queryMediaForAllStates();
325	if (FAILED(hrc))
326	return hrc;
327
328	/* Calculate the total size of images. Fill m_finalMediaMap */
329	{ /** The scope here for better reading, apart from that the variables have limited scope too */
330	uint64_t totalMediaSize = 0;
331
332	for (size_t i = 0; i < m_llMedia.size(); ++i)
333	{
334	MEDIUMTASKCHAINMOVE &mtc = m_llMedia.at(i);
335	for (size_t a = mtc.chain.size(); a > 0; --a)
336	{
337	Bstr bstrLocation;
338	Utf8Str name = mtc.chain[a - 1].strBaseName;
339	ComPtr<IMedium> plMedium = mtc.chain[a - 1].pMedium;
340	hrc = plMedium->COMGETTER(Location)(bstrLocation.asOutParam());
341	if (FAILED(hrc))
342	return hrc;
343
344	Utf8Str strLocation = bstrLocation;
345
346	/* if an image is located in the actual VM folder it will be added to the actual list */
347	if (strLocation.startsWith(strSettingsFilePath))
348	{
349	LONG64 cbSize = 0;
350	hrc = plMedium->COMGETTER(Size)(&cbSize);
351	if (FAILED(hrc))
352	return hrc;
353
354	std::pair<std::map<Utf8Str, MEDIUMTASKMOVE>::iterator,bool> ret;
355	ret = m_finalMediaMap.insert(std::make_pair(name, mtc.chain[a - 1]));
356	if (ret.second == true)
357	{
358	/* Calculate progress data */
359	++uCount;
360	uTotalWeight += mtc.chain[a - 1].uWeight;
361	totalMediaSize += (uint64_t)cbSize;
362	Log2(("Image %s was added into the moved list\n", name.c_str()));
363	}
364	}
365	}
366	}
367
368	Log2(("Total Size of images is %lld bytes\n", totalMediaSize));
369	neededFreeSpace += totalMediaSize;
370	}
371
372	/* Prepare data for moving ".sav" files */
373	{
374	uint64_t totalStateSize = 0;
375
376	for (size_t i = 0; i < m_llSaveStateFiles.size(); ++i)
377	{
378	uint64_t cbFile = 0;
379	SNAPFILETASKMOVE &sft = m_llSaveStateFiles.at(i);
380
381	Utf8Str name = sft.strFile;
382	/* if a state file is located in the actual VM folder it will be added to the actual list */
383	if (RTPathStartsWith(name.c_str(), strSettingsFilePath.c_str()))
384	{
385	vrc = RTFileQuerySizeByPath(name.c_str(), &cbFile);
386	if (RT_SUCCESS(vrc))
387	{
388	std::pair<std::map<Utf8Str, SNAPFILETASKMOVE>::iterator,bool> ret;
389	ret = m_finalSaveStateFilesMap.insert(std::make_pair(name, sft));
390	if (ret.second == true)
391	{
392	totalStateSize += cbFile;
393	++uCount;
394	uTotalWeight += sft.uWeight;
395	Log2(("The state file %s was added into the moved list\n", name.c_str()));
396	}
397	}
398	else
399	{
400	Log2(("The state file %s wasn't added into the moved list. Couldn't get the file size.\n",
401	name.c_str()));
402	return m_pMachine->setErrorVrc(vrc,
403	tr("Failed to get file size for '%s': %Rrc"),
404	name.c_str(), vrc);
405	}
406	}
407	}
408
409	neededFreeSpace += totalStateSize;
410	}
411
412	/* Prepare data for moving ".nvram" files */
413	{
414	uint64_t totalNVRAMSize = 0;
415
416	for (size_t i = 0; i < m_llNVRAMFiles.size(); ++i)
417	{
418	uint64_t cbFile = 0;
419	SNAPFILETASKMOVE &sft = m_llNVRAMFiles.at(i);
420
421	Utf8Str name = sft.strFile;
422	/* if a NVRAM file is located in the actual VM folder it will be added to the actual list */
423	if (RTPathStartsWith(name.c_str(), strSettingsFilePath.c_str()))
424	{
425	vrc = RTFileQuerySizeByPath(name.c_str(), &cbFile);
426	if (RT_SUCCESS(vrc))
427	{
428	std::pair<std::map<Utf8Str, SNAPFILETASKMOVE>::iterator,bool> ret;
429	ret = m_finalNVRAMFilesMap.insert(std::make_pair(name, sft));
430	if (ret.second == true)
431	{
432	totalNVRAMSize += cbFile;
433	++uCount;
434	uTotalWeight += sft.uWeight;
435	Log2(("The NVRAM file %s was added into the moved list\n", name.c_str()));
436	}
437	}
438	else
439	{
440	Log2(("The NVRAM file %s wasn't added into the moved list. Couldn't get the file size.\n",
441	name.c_str()));
442	return m_pMachine->setErrorVrc(vrc,
443	tr("Failed to get file size for '%s': %Rrc"),
444	name.c_str(), vrc);
445	}
446	}
447	}
448
449	neededFreeSpace += totalNVRAMSize;
450	}
451
452	/* Prepare data for moving the log files */
453	{
454	Utf8Str strFolder = m_vmFolders[VBox_LogFolder];
455
456	if (RTPathExists(strFolder.c_str()))
457	{
458	uint64_t totalLogSize = 0;
459	hrc = getFolderSize(strFolder, totalLogSize);
460	if (SUCCEEDED(hrc))
461	{
462	neededFreeSpace += totalLogSize;
463	if (cbFree - neededFreeSpace <= _1M)
464	return m_pMachine->setError(E_FAIL,
465	tr("Insufficient disk space available (%RTfoff needed, %RTfoff free)"),
466	neededFreeSpace, cbFree);
467
468	fileList_t filesList;
469	hrc = getFilesList(strFolder, filesList);
470	if (FAILED(hrc))
471	return hrc;
472
473	cit_t it = filesList.m_list.begin();
474	while (it != filesList.m_list.end())
475	{
476	Utf8Str strFile = it->first.c_str();
477	strFile.append(RTPATH_DELIMITER).append(it->second.c_str());
478
479	uint64_t cbFile = 0;
480	vrc = RTFileQuerySizeByPath(strFile.c_str(), &cbFile);
481	if (RT_SUCCESS(vrc))
482	{
483	uCount += 1;
484	uTotalWeight += (ULONG)((cbFile + _1M - 1) / _1M);
485	actualFileList.add(strFile);
486	Log2(("The log file %s added into the moved list\n", strFile.c_str()));
487	}
488	else
489	Log2(("The log file %s wasn't added into the moved list. Couldn't get the file size.\n", strFile.c_str()));
490	++it;
491	}
492	}
493	else
494	return hrc;
495	}
496	else
497	{
498	Log2(("Information: The original log folder %s doesn't exist\n", strFolder.c_str()));
499	hrc = S_OK;//it's not error in this case if there isn't an original log folder
500	}
501	}
502
503	LogRel(("Total space needed is %lld bytes\n", neededFreeSpace));
504	/* Check a target location on enough room */
505	if (cbFree - neededFreeSpace <= _1M)
506	{
507	LogRel(("but free space on destination is %RTfoff\n", cbFree));
508	return m_pMachine->setError(VBOX_E_IPRT_ERROR,
509	tr("Insufficient disk space available (%RTfoff needed, %RTfoff free)"),
510	neededFreeSpace, cbFree);
511	}
512
513	/* Add step for .vbox machine setting file */
514	++uCount;
515	uTotalWeight += 1;
516
517	/* Reserve additional steps in case of failure and rollback all changes */
518	uTotalWeight += uCount;//just add 1 for each possible rollback operation
519	uCount += uCount;//and increase the steps twice
520
521	/* Init Progress instance */
522	{
523	hrc = m_pProgress->init(m_pMachine->i_getVirtualBox(),
524	static_cast<IMachine >(m_pMachine) / aInitiator */,
525	Utf8Str(tr("Moving Machine")),
526	true /* fCancellable */,
527	uCount,
528	uTotalWeight,
529	Utf8Str(tr("Initialize Moving")),
530	1);
531	if (FAILED(hrc))
532	return m_pMachine->setError(hrc,
533	tr("Couldn't correctly setup the progress object for moving VM operation"));
534	}
535
536	/* save all VM data */
537	m_pMachine->i_setModified(Machine::IsModified_MachineData);
538	hrc = m_pMachine->SaveSettings();
539	if (FAILED(hrc))
540	return hrc;
541
542	LogFlowFuncLeave();
543
544	return hrc;
545	}
546
547	void MachineMoveVM::printStateFile(settings::SnapshotsList &snl)
548	{
549	settings::SnapshotsList::iterator it;
550	for (it = snl.begin(); it != snl.end(); ++it)
551	{
552	if (!it->strStateFile.isEmpty())
553	{
554	settings::Snapshot snap = (settings::Snapshot)(*it);
555	Log2(("snap.uuid = %s\n", snap.uuid.toStringCurly().c_str()));
556	Log2(("snap.strStateFile = %s\n", snap.strStateFile.c_str()));
557	}
558
559	if (!it->llChildSnapshots.empty())
560	printStateFile(it->llChildSnapshots);
561	}
562	}
563
564	/* static */
565	DECLCALLBACK(int) MachineMoveVM::updateProgress(unsigned uPercent, void *pvUser)
566	{
567	MachineMoveVM pTask = (MachineMoveVM **)pvUser;
568
569	if ( pTask
570	&& !pTask->m_pProgress.isNull())
571	{
572	BOOL fCanceled;
573	pTask->m_pProgress->COMGETTER(Canceled)(&fCanceled);
574	if (fCanceled)
575	return -1;
576	pTask->m_pProgress->SetCurrentOperationProgress(uPercent);
577	}
578	return VINF_SUCCESS;
579	}
580
581	/* static */
582	DECLCALLBACK(int) MachineMoveVM::copyFileProgress(unsigned uPercentage, void *pvUser)
583	{
584	ComObjPtr<Progress> pProgress = static_cast<ComObjPtr<Progress> >(pvUser);
585
586	BOOL fCanceled = false;
587	HRESULT hrc = pProgress->COMGETTER(Canceled)(&fCanceled);
588	if (FAILED(hrc)) return VERR_GENERAL_FAILURE;
589	/* If canceled by the user tell it to the copy operation. */
590	if (fCanceled) return VERR_CANCELLED;
591	/* Set the new process. */
592	hrc = pProgress->SetCurrentOperationProgress(uPercentage);
593	if (FAILED(hrc)) return VERR_GENERAL_FAILURE;
594
595	return VINF_SUCCESS;
596	}
597
598	/* static */
599	void MachineMoveVM::i_MoveVMThreadTask(MachineMoveVM *task)
600	{
601	LogFlowFuncEnter();
602	HRESULT hrc = S_OK;
603
604	MachineMoveVM *taskMoveVM = task;
605	ComObjPtr<Machine> &machine = taskMoveVM->m_pMachine;
606
607	AutoCaller autoCaller(machine);
608	// if (FAILED(autoCaller.hrc())) return;//Should we return something here?
609
610	Utf8Str strTargetFolder = taskMoveVM->m_targetPath;
611	{
612	Bstr bstrMachineName;
613	hrc = machine->COMGETTER(Name)(bstrMachineName.asOutParam());
614	if (FAILED(hrc))
615	{
616	taskMoveVM->m_result = hrc;
617	if (!taskMoveVM->m_pProgress.isNull())
618	taskMoveVM->m_pProgress->i_notifyComplete(taskMoveVM->m_result);
619	return;
620	}
621	strTargetFolder.append(Utf8Str(bstrMachineName));
622	}
623
624	RTCList<Utf8Str> newFiles; /* All extra created files (save states, ...) */
625	RTCList<Utf8Str> originalFiles; /* All original files except images */
626	typedef std::map<Utf8Str, ComObjPtr<Medium> > MediumMap;
627	MediumMap mapOriginalMedium;
628
629	/*
630	* We have the couple modes which user is able to request
631	* basic mode:
632	* - The images which are solely attached to the VM
633	* and located in the original VM folder will be moved.
634	* All subfolders related to the original VM are also moved from the original location
635	* (Standard - snapshots and logs folders).
636	*
637	* canonical mode:
638	* - All disks tied with the VM will be moved into a new location if it's possible.
639	* All folders related to the original VM are also moved.
640	* This mode is intended to collect all files/images/snapshots related to the VM in the one place.
641	*
642	*/
643
644	/*
645	* A way to handle shareable disk:
646	* Collect the shareable disks attched to the VM.
647	* Get the machines whom the shareable disks attach to.
648	* Return an error if the state of any VM doesn't allow to move a shareable disk and
649	* this disk is located in the VM's folder (it means the disk is intended for "moving").
650	*/
651
652
653	/*
654	* Check new destination whether enough room for the VM or not. if "not" return an error.
655	* Make a copy of VM settings and a list with all files which are moved. Save the list on the disk.
656	* Start "move" operation.
657	* Check the result of operation.
658	* if the operation was successful:
659	* - delete all files in the original VM folder;
660	* - update VM disks info with new location;
661	* - update all other VM if it's needed;
662	* - update global settings
663	*/
664
665	try
666	{
667	/* Move all disks */
668	hrc = taskMoveVM->moveAllDisks(taskMoveVM->m_finalMediaMap, strTargetFolder);
669	if (FAILED(hrc))
670	throw hrc;
671
672	/* Get Machine::Data here because moveAllDisks() change it */
673	Machine::Data *machineData = machine->mData.data();
674	settings::MachineConfigFile *machineConfFile = machineData->pMachineConfigFile;
675
676	/* Copy all save state files. */
677	Utf8Str strTrgSnapshotFolder;
678	{
679	/* When the current snapshot folder is absolute we reset it to the
680	* default relative folder. */
681	if (RTPathStartsWithRoot(machineConfFile->machineUserData.strSnapshotFolder.c_str()))
682	machineConfFile->machineUserData.strSnapshotFolder = "Snapshots";
683	machineConfFile->strStateFile = "";
684
685	/* The absolute name of the snapshot folder. */
686	strTrgSnapshotFolder = Utf8StrFmt("%s%c%s", strTargetFolder.c_str(), RTPATH_DELIMITER,
687	machineConfFile->machineUserData.strSnapshotFolder.c_str());
688
689	/* Check if a snapshot folder is necessary and if so doesn't already
690	* exists. */
691	if ( ( taskMoveVM->m_finalSaveStateFilesMap.size() > 0
692	\|\| taskMoveVM->m_finalNVRAMFilesMap.size() > 1)
693	&& !RTDirExists(strTrgSnapshotFolder.c_str()))
694	{
695	int vrc = RTDirCreateFullPath(strTrgSnapshotFolder.c_str(), 0700);
696	if (RT_FAILURE(vrc))
697	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
698	tr("Could not create snapshots folder '%s' (%Rrc)"),
699	strTrgSnapshotFolder.c_str(), vrc);
700	}
701
702	std::map<Utf8Str, SNAPFILETASKMOVE>::iterator itState = taskMoveVM->m_finalSaveStateFilesMap.begin();
703	while (itState != taskMoveVM->m_finalSaveStateFilesMap.end())
704	{
705	const SNAPFILETASKMOVE &sft = itState->second;
706	const Utf8Str &strTrgSaveState = Utf8StrFmt("%s%c%s", strTrgSnapshotFolder.c_str(), RTPATH_DELIMITER,
707	RTPathFilename(sft.strFile.c_str()));
708
709	/* Move to next sub-operation. */
710	hrc = taskMoveVM->m_pProgress->SetNextOperation(BstrFmt(tr("Copy the save state file '%s' ..."),
711	RTPathFilename(sft.strFile.c_str())).raw(),
712	sft.uWeight);
713	if (FAILED(hrc))
714	throw hrc;
715
716	int vrc = RTFileCopyEx(sft.strFile.c_str(), strTrgSaveState.c_str(), 0,
717	MachineMoveVM::copyFileProgress, &taskMoveVM->m_pProgress);
718	if (RT_FAILURE(vrc))
719	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
720	tr("Could not copy state file '%s' to '%s' (%Rrc)"),
721	sft.strFile.c_str(),
722	strTrgSaveState.c_str(),
723	vrc);
724
725	/* save new file in case of restoring */
726	newFiles.append(strTrgSaveState);
727	/* save original file for deletion in the end */
728	originalFiles.append(sft.strFile);
729	++itState;
730	}
731
732	std::map<Utf8Str, SNAPFILETASKMOVE>::iterator itNVRAM = taskMoveVM->m_finalNVRAMFilesMap.begin();
733	while (itNVRAM != taskMoveVM->m_finalNVRAMFilesMap.end())
734	{
735	const SNAPFILETASKMOVE &sft = itNVRAM->second;
736	const Utf8Str &strTrgNVRAM = Utf8StrFmt("%s%c%s", sft.snapshotUuid.isZero() ? strTargetFolder.c_str() : strTrgSnapshotFolder.c_str(),
737	RTPATH_DELIMITER, RTPathFilename(sft.strFile.c_str()));
738
739	/* Move to next sub-operation. */
740	hrc = taskMoveVM->m_pProgress->SetNextOperation(BstrFmt(tr("Copy the NVRAM file '%s' ..."),
741	RTPathFilename(sft.strFile.c_str())).raw(),
742	sft.uWeight);
743	if (FAILED(hrc))
744	throw hrc;
745
746	int vrc = RTFileCopyEx(sft.strFile.c_str(), strTrgNVRAM.c_str(), 0,
747	MachineMoveVM::copyFileProgress, &taskMoveVM->m_pProgress);
748	if (RT_FAILURE(vrc))
749	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
750	tr("Could not copy NVRAM file '%s' to '%s' (%Rrc)"),
751	sft.strFile.c_str(),
752	strTrgNVRAM.c_str(),
753	vrc);
754
755	/* save new file in case of restoring */
756	newFiles.append(strTrgNVRAM);
757	/* save original file for deletion in the end */
758	originalFiles.append(sft.strFile);
759	++itNVRAM;
760	}
761	}
762
763	/*
764	* Update state file path
765	* very important step!
766	*/
767	Log2(("Update state file path\n"));
768	/** @todo r=klaus: this update is not necessarily matching what the
769	* above code has set as the new folders, so it needs reimplementing */
770	taskMoveVM->updatePathsToStateFiles(taskMoveVM->m_vmFolders[VBox_SettingFolder],
771	strTargetFolder);
772
773	/*
774	* Update NVRAM file paths
775	* very important step!
776	*/
777	Log2(("Update NVRAM paths\n"));
778	/** @todo r=klaus: this update is not necessarily matching what the
779	* above code has set as the new folders, so it needs reimplementing.
780	* What's good about this implementation: it does not look at the
781	* list of NVRAM files, because that only lists the existing ones,
782	* but all paths need fixing. */
783	taskMoveVM->updatePathsToNVRAMFiles(taskMoveVM->m_vmFolders[VBox_SettingFolder],
784	strTargetFolder);
785
786	/*
787	* Moving Machine settings file
788	* The settings file are moved after all disks and snapshots because this file should be updated
789	* with actual information and only then should be moved.
790	*/
791	{
792	Log2(("Copy Machine settings file\n"));
793
794	hrc = taskMoveVM->m_pProgress->SetNextOperation(BstrFmt(tr("Copy Machine settings file '%s' ..."),
795	machineConfFile->machineUserData.strName.c_str()).raw(),
796	1);
797	if (FAILED(hrc))
798	throw hrc;
799
800	Utf8Str strTargetSettingsFilePath = strTargetFolder;
801
802	/* Check a folder existing and create one if it's not */
803	if (!RTDirExists(strTargetSettingsFilePath.c_str()))
804	{
805	int vrc = RTDirCreateFullPath(strTargetSettingsFilePath.c_str(), 0700);
806	if (RT_FAILURE(vrc))
807	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
808	tr("Could not create a home machine folder '%s' (%Rrc)"),
809	strTargetSettingsFilePath.c_str(), vrc);
810
811	Log2(("Created a home machine folder %s\n", strTargetSettingsFilePath.c_str()));
812	}
813
814	/* Create a full path */
815	Bstr bstrMachineName;
816	machine->COMGETTER(Name)(bstrMachineName.asOutParam());
817	if (FAILED(hrc))
818	throw hrc;
819	strTargetSettingsFilePath.append(RTPATH_DELIMITER).append(Utf8Str(bstrMachineName));
820	strTargetSettingsFilePath.append(".vbox");
821
822	Utf8Str strSettingsFilePath;
823	Bstr bstr_settingsFilePath;
824	machine->COMGETTER(SettingsFilePath)(bstr_settingsFilePath.asOutParam());
825	if (FAILED(hrc))
826	throw hrc;
827	strSettingsFilePath = bstr_settingsFilePath;
828
829	int vrc = RTFileCopyEx(strSettingsFilePath.c_str(), strTargetSettingsFilePath.c_str(), 0,
830	MachineMoveVM::copyFileProgress, &taskMoveVM->m_pProgress);
831	if (RT_FAILURE(vrc))
832	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
833	tr("Could not copy the setting file '%s' to '%s' (%Rrc)"),
834	strSettingsFilePath.c_str(),
835	strTargetSettingsFilePath.stripFilename().c_str(),
836	vrc);
837
838	Log2(("The setting file %s has been copied into the folder %s\n",
839	strSettingsFilePath.c_str(), strTargetSettingsFilePath.stripFilename().c_str()));
840
841	/* save new file in case of restoring */
842	newFiles.append(strTargetSettingsFilePath);
843	/* save original file for deletion in the end */
844	originalFiles.append(strSettingsFilePath);
845
846	Utf8Str strPrevSettingsFilePath = strSettingsFilePath;
847	strPrevSettingsFilePath.append("-prev");
848	if (RTFileExists(strPrevSettingsFilePath.c_str()))
849	originalFiles.append(strPrevSettingsFilePath);
850	}
851
852	/* Moving Machine log files */
853	{
854	Log2(("Copy machine log files\n"));
855
856	if (taskMoveVM->m_vmFolders[VBox_LogFolder].isNotEmpty())
857	{
858	/* Check an original log folder existence */
859	if (RTDirExists(taskMoveVM->m_vmFolders[VBox_LogFolder].c_str()))
860	{
861	Utf8Str strTargetLogFolderPath = strTargetFolder;
862	strTargetLogFolderPath.append(RTPATH_DELIMITER).append("Logs");
863
864	/* Check a destination log folder existence and create one if it's not */
865	if (!RTDirExists(strTargetLogFolderPath.c_str()))
866	{
867	int vrc = RTDirCreateFullPath(strTargetLogFolderPath.c_str(), 0700);
868	if (RT_FAILURE(vrc))
869	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
870	tr("Could not create log folder '%s' (%Rrc)"),
871	strTargetLogFolderPath.c_str(), vrc);
872
873	Log2(("Created a log machine folder %s\n", strTargetLogFolderPath.c_str()));
874	}
875
876	fileList_t filesList;
877	taskMoveVM->getFilesList(taskMoveVM->m_vmFolders[VBox_LogFolder], filesList);
878	cit_t it = filesList.m_list.begin();
879	while (it != filesList.m_list.end())
880	{
881	Utf8Str strFullSourceFilePath = it->first.c_str();
882	strFullSourceFilePath.append(RTPATH_DELIMITER).append(it->second.c_str());
883
884	Utf8Str strFullTargetFilePath = strTargetLogFolderPath;
885	strFullTargetFilePath.append(RTPATH_DELIMITER).append(it->second.c_str());
886
887	/* Move to next sub-operation. */
888	hrc = taskMoveVM->m_pProgress->SetNextOperation(BstrFmt(tr("Copying the log file '%s' ..."),
889	RTPathFilename(strFullSourceFilePath.c_str())).raw(),
890	1);
891	if (FAILED(hrc))
892	throw hrc;
893
894	int vrc = RTFileCopyEx(strFullSourceFilePath.c_str(), strFullTargetFilePath.c_str(), 0,
895	MachineMoveVM::copyFileProgress, &taskMoveVM->m_pProgress);
896	if (RT_FAILURE(vrc))
897	throw machine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
898	tr("Could not copy the log file '%s' to '%s' (%Rrc)"),
899	strFullSourceFilePath.c_str(),
900	strFullTargetFilePath.stripFilename().c_str(),
901	vrc);
902
903	Log2(("The log file %s has been copied into the folder %s\n", strFullSourceFilePath.c_str(),
904	strFullTargetFilePath.stripFilename().c_str()));
905
906	/* save new file in case of restoring */
907	newFiles.append(strFullTargetFilePath);
908	/* save original file for deletion in the end */
909	originalFiles.append(strFullSourceFilePath);
910
911	++it;
912	}
913	}
914	}
915	}
916
917	/* save all VM data */
918	hrc = machine->SaveSettings();
919	if (FAILED(hrc))
920	throw hrc;
921
922	Log2(("Update path to XML setting file\n"));
923	Utf8Str strTargetSettingsFilePath = strTargetFolder;
924	Bstr bstrMachineName;
925	hrc = machine->COMGETTER(Name)(bstrMachineName.asOutParam());
926	if (FAILED(hrc))
927	throw hrc;
928	strTargetSettingsFilePath.append(RTPATH_DELIMITER).append(Utf8Str(bstrMachineName)).append(".vbox");
929	machineData->m_strConfigFileFull = strTargetSettingsFilePath;
930	machine->mParent->i_copyPathRelativeToConfig(strTargetSettingsFilePath, machineData->m_strConfigFile);
931
932	/* Marks the global registry for uuid as modified */
933	Guid uuid = machine->mData->mUuid;
934	machine->mParent->i_markRegistryModified(uuid);
935
936	/* for saving the global settings we should hold only the VirtualBox lock */
937	AutoWriteLock vboxLock(machine->mParent COMMA_LOCKVAL_SRC_POS);
938
939	/* Save global settings in the VirtualBox.xml */
940	hrc = machine->mParent->i_saveSettings();
941	if (FAILED(hrc))
942	throw hrc;
943	}
944	catch(HRESULT aRc)
945	{
946	hrc = aRc;
947	taskMoveVM->m_result = hrc;
948	}
949	catch (...)
950	{
951	Log2(("Moving machine to a new destination was failed. Check original and destination places.\n"));
952	hrc = VirtualBoxBase::handleUnexpectedExceptions(machine, RT_SRC_POS);
953	taskMoveVM->m_result = hrc;
954	}
955
956	/* Cleanup on failure */
957	if (FAILED(hrc))
958	{
959	Machine::Data *machineData = machine->mData.data();
960
961	/* Restoring the original media */
962	try
963	{
964	/*
965	* Fix the progress counter
966	* In instance, the whole "move vm" operation is failed on 5th step. But total count is 20.
967	* Where 20 = 2 * 10 operations, where 10 is the real number of operations. And this value was doubled
968	* earlier in the init() exactly for one reason - rollback operation. Because in this case we must do
969	* the same operations but in backward direction.
970	* Thus now we want to correct the progress counter from 5 to 15. Why?
971	* Because we should have evaluated the counter as "20/2 + (20/2 - 5)" = 15 or just "20 - 5" = 15
972	* And because the 5th step failed it shouldn't be counted.
973	* As result, we need to rollback 4 operations.
974	* Thus we start from "operation + 1" and finish when "i < operationCount - operation".
975	*/
976
977	/** @todo r=vvp: Do we need to check each return result here? Looks excessively
978	* and what to do with any failure here? We are already in the rollback action.
979	* Throw only the important errors?
980	* We MUST finish this action anyway to avoid garbage and get the original VM state. */
981	/* ! Apparently we should update the Progress object !*/
982	ULONG operationCount = 0;
983	hrc = taskMoveVM->m_pProgress->COMGETTER(OperationCount)(&operationCount);
984	if (FAILED(hrc))
985	throw hrc;
986	ULONG operation = 0;
987	hrc = taskMoveVM->m_pProgress->COMGETTER(Operation)(&operation);
988	if (FAILED(hrc))
989	throw hrc;
990	Bstr bstrOperationDescription;
991	hrc = taskMoveVM->m_pProgress->COMGETTER(OperationDescription)(bstrOperationDescription.asOutParam());
992	if (FAILED(hrc))
993	throw hrc;
994	Utf8Str strOperationDescription = bstrOperationDescription;
995	ULONG operationPercent = 0;
996	hrc = taskMoveVM->m_pProgress->COMGETTER(OperationPercent)(&operationPercent);
997	if (FAILED(hrc))
998	throw hrc;
999	Bstr bstrMachineName;
1000	hrc = machine->COMGETTER(Name)(bstrMachineName.asOutParam());
1001	if (FAILED(hrc))
1002	throw hrc;
1003
1004	Log2(("Moving machine %s was failed on operation %s\n",
1005	Utf8Str(bstrMachineName.raw()).c_str(), Utf8Str(bstrOperationDescription.raw()).c_str()));
1006
1007	for (ULONG i = operation + 1; i < operationCount - operation; ++i)
1008	taskMoveVM->m_pProgress->SetNextOperation(BstrFmt(tr("Skip the empty operation %d..."), i + 1).raw(), 1);
1009
1010	hrc = taskMoveVM->moveAllDisks(taskMoveVM->m_finalMediaMap);
1011	if (FAILED(hrc))
1012	throw hrc;
1013
1014	/* Revert original paths to the state files */
1015	taskMoveVM->updatePathsToStateFiles(strTargetFolder,
1016	taskMoveVM->m_vmFolders[VBox_SettingFolder]);
1017
1018	/* Revert original paths to the NVRAM files */
1019	taskMoveVM->updatePathsToNVRAMFiles(strTargetFolder,
1020	taskMoveVM->m_vmFolders[VBox_SettingFolder]);
1021
1022	/* Delete all created files. Here we update progress object */
1023	hrc = taskMoveVM->deleteFiles(newFiles);
1024	if (FAILED(hrc))
1025	{
1026	Log2(("Rollback scenario: can't delete new created files. Check the destination folder.\n"));
1027	throw hrc;
1028	}
1029
1030	/* Delete destination folder */
1031	int vrc = RTDirRemove(strTargetFolder.c_str());
1032	if (RT_FAILURE(vrc))
1033	{
1034	Log2(("Rollback scenario: can't delete new destination folder.\n"));
1035	throw machine->setErrorVrc(vrc, tr("Rollback scenario: can't delete new destination folder."));
1036	}
1037
1038	/* save all VM data */
1039	{
1040	AutoWriteLock srcLock(machine COMMA_LOCKVAL_SRC_POS);
1041	srcLock.release();
1042	hrc = machine->SaveSettings();
1043	if (FAILED(hrc))
1044	{
1045	Log2(("Rollback scenario: can't save machine settings.\n"));
1046	throw hrc;
1047	}
1048	srcLock.acquire();
1049	}
1050
1051	/* Restore an original path to XML setting file */
1052	{
1053	Log2(("Rollback scenario: restoration of the original path to XML setting file\n"));
1054	Utf8Str strOriginalSettingsFilePath = taskMoveVM->m_vmFolders[VBox_SettingFolder];
1055	strOriginalSettingsFilePath.append(RTPATH_DELIMITER).append(Utf8Str(bstrMachineName)).append(".vbox");
1056	machineData->m_strConfigFileFull = strOriginalSettingsFilePath;
1057	machine->mParent->i_copyPathRelativeToConfig(strOriginalSettingsFilePath, machineData->m_strConfigFile);
1058	}
1059
1060	/* Marks the global registry for uuid as modified */
1061	{
1062	AutoWriteLock srcLock(machine COMMA_LOCKVAL_SRC_POS);
1063	srcLock.release();
1064	Guid uuid = machine->mData->mUuid;
1065	machine->mParent->i_markRegistryModified(uuid);
1066	srcLock.acquire();
1067	}
1068
1069	/* save the global settings; for that we should hold only the VirtualBox lock */
1070	{
1071	AutoWriteLock vboxLock(machine->mParent COMMA_LOCKVAL_SRC_POS);
1072	hrc = machine->mParent->i_saveSettings();
1073	if (FAILED(hrc))
1074	{
1075	Log2(("Rollback scenario: can't save global settings.\n"));
1076	throw hrc;
1077	}
1078	}
1079	}
1080	catch(HRESULT aRc)
1081	{
1082	hrc = aRc;
1083	Log2(("Rollback scenario: restoration the original media failed. Machine can be corrupted.\n"));
1084	}
1085	catch (...)
1086	{
1087	Log2(("Rollback scenario: restoration the original media failed. Machine can be corrupted.\n"));
1088	hrc = VirtualBoxBase::handleUnexpectedExceptions(machine, RT_SRC_POS);
1089	}
1090	/* In case of failure the progress object on the other side (user side) get notification about operation
1091	completion but the operation percentage may not be set to 100% */
1092	}
1093	else /Operation was successful and now we can delete the original files like the state files, XML setting, log files /
1094	{
1095	/*
1096	* In case of success it's not urgent to update the progress object because we call i_notifyComplete() with
1097	* the success result. As result, the last number of progress operation can be not equal the number of operations
1098	* because we doubled the number of operations for rollback case.
1099	* But if we want to update the progress object corectly it's needed to add all medium moved by standard
1100	* "move medium" logic (for us it's taskMoveVM->m_finalMediaMap) to the current number of operation.
1101	*/
1102
1103	ULONG operationCount = 0;
1104	hrc = taskMoveVM->m_pProgress->COMGETTER(OperationCount)(&operationCount);
1105	ULONG operation = 0;
1106	hrc = taskMoveVM->m_pProgress->COMGETTER(Operation)(&operation);
1107
1108	for (ULONG i = operation; i < operation + taskMoveVM->m_finalMediaMap.size() - 1; ++i)
1109	taskMoveVM->m_pProgress->SetNextOperation(BstrFmt(tr("Skip the empty operation %d..."), i).raw(), 1);
1110
1111	hrc = taskMoveVM->deleteFiles(originalFiles);
1112	if (FAILED(hrc))
1113	Log2(("Forward scenario: can't delete all original files.\n"));
1114
1115	/* delete no longer needed source directories */
1116	if ( taskMoveVM->m_vmFolders[VBox_SnapshotFolder].isNotEmpty()
1117	&& RTDirExists(taskMoveVM->m_vmFolders[VBox_SnapshotFolder].c_str()))
1118	RTDirRemove(taskMoveVM->m_vmFolders[VBox_SnapshotFolder].c_str());
1119
1120	if ( taskMoveVM->m_vmFolders[VBox_LogFolder].isNotEmpty()
1121	&& RTDirExists(taskMoveVM->m_vmFolders[VBox_LogFolder].c_str()))
1122	RTDirRemove(taskMoveVM->m_vmFolders[VBox_LogFolder].c_str());
1123
1124	if ( taskMoveVM->m_vmFolders[VBox_SettingFolder].isNotEmpty()
1125	&& RTDirExists(taskMoveVM->m_vmFolders[VBox_SettingFolder].c_str()))
1126	RTDirRemove(taskMoveVM->m_vmFolders[VBox_SettingFolder].c_str());
1127	}
1128
1129	if (!taskMoveVM->m_pProgress.isNull())
1130	taskMoveVM->m_pProgress->i_notifyComplete(taskMoveVM->m_result);
1131
1132	LogFlowFuncLeave();
1133	}
1134
1135	HRESULT MachineMoveVM::moveAllDisks(const std::map<Utf8Str, MEDIUMTASKMOVE> &listOfDisks,
1136	const Utf8Str &strTargetFolder)
1137	{
1138	HRESULT hrc = S_OK;
1139	ComObjPtr<Machine> &machine = m_pMachine;
1140	Utf8Str strLocation;
1141
1142	AutoWriteLock machineLock(machine COMMA_LOCKVAL_SRC_POS);
1143
1144	try
1145	{
1146	std::map<Utf8Str, MEDIUMTASKMOVE>::const_iterator itMedium = listOfDisks.begin();
1147	while (itMedium != listOfDisks.end())
1148	{
1149	const MEDIUMTASKMOVE &mt = itMedium->second;
1150	ComPtr<IMedium> pMedium = mt.pMedium;
1151	Utf8Str strTargetImageName;
1152	Bstr bstrLocation;
1153	Bstr bstrSrcName;
1154
1155	hrc = pMedium->COMGETTER(Name)(bstrSrcName.asOutParam());
1156	if (FAILED(hrc)) throw hrc;
1157
1158	if (strTargetFolder.isNotEmpty())
1159	{
1160	strTargetImageName = strTargetFolder;
1161	hrc = pMedium->COMGETTER(Location)(bstrLocation.asOutParam());
1162	if (FAILED(hrc)) throw hrc;
1163	strLocation = bstrLocation;
1164
1165	if (mt.fSnapshot == true)
1166	strLocation.stripFilename().stripPath().append(RTPATH_DELIMITER).append(Utf8Str(bstrSrcName));
1167	else
1168	strLocation.stripPath();
1169
1170	strTargetImageName.append(RTPATH_DELIMITER).append(strLocation);
1171	hrc = m_pProgress->SetNextOperation(BstrFmt(tr("Moving medium '%ls' ..."), bstrSrcName.raw()).raw(), mt.uWeight);
1172	if (FAILED(hrc)) throw hrc;
1173	}
1174	else
1175	{
1176	strTargetImageName = mt.strBaseName;//Should contain full path to the image
1177	hrc = m_pProgress->SetNextOperation(BstrFmt(tr("Moving medium '%ls' back..."), bstrSrcName.raw()).raw(), mt.uWeight);
1178	if (FAILED(hrc)) throw hrc;
1179	}
1180
1181
1182
1183	/* consistency: use \ if appropriate on the platform */
1184	RTPathChangeToDosSlashes(strTargetImageName.mutableRaw(), false);
1185
1186	bstrLocation = strTargetImageName.c_str();
1187
1188	MediumType_T mediumType;//immutable, shared, passthrough
1189	hrc = pMedium->COMGETTER(Type)(&mediumType);
1190	if (FAILED(hrc)) throw hrc;
1191
1192	DeviceType_T deviceType;//floppy, hard, DVD
1193	hrc = pMedium->COMGETTER(DeviceType)(&deviceType);
1194	if (FAILED(hrc)) throw hrc;
1195
1196	/* Drop lock early because IMedium::MoveTo needs to get the VirtualBox one. */
1197	machineLock.release();
1198
1199	ComPtr<IProgress> moveDiskProgress;
1200	hrc = pMedium->MoveTo(bstrLocation.raw(), moveDiskProgress.asOutParam());
1201	if (SUCCEEDED(hrc))
1202	{
1203	/* In case of failure moveDiskProgress would be in the invalid state or not initialized at all
1204	* Call i_waitForOtherProgressCompletion only in success
1205	*/
1206	/* Wait until the other process has finished. */
1207	hrc = m_pProgress->WaitForOtherProgressCompletion(moveDiskProgress, 0 /* indefinite wait */);
1208	}
1209
1210	/acquire the lock back/
1211	machineLock.acquire();
1212
1213	if (FAILED(hrc)) throw hrc;
1214
1215	Log2(("Moving %s has been finished\n", strTargetImageName.c_str()));
1216
1217	++itMedium;
1218	}
1219
1220	machineLock.release();
1221	}
1222	catch (HRESULT hrcXcpt)
1223	{
1224	Log2(("Exception during moving the disk %s: %Rhrc\n", strLocation.c_str(), hrcXcpt));
1225	hrc = hrcXcpt;
1226	machineLock.release();
1227	}
1228	catch (...)
1229	{
1230	Log2(("Exception during moving the disk %s\n", strLocation.c_str()));
1231	hrc = VirtualBoxBase::handleUnexpectedExceptions(m_pMachine, RT_SRC_POS);
1232	machineLock.release();
1233	}
1234
1235	return hrc;
1236	}
1237
1238	void MachineMoveVM::updatePathsToStateFiles(const Utf8Str &sourcePath, const Utf8Str &targetPath)
1239	{
1240	ComObjPtr<Snapshot> pSnapshot;
1241	HRESULT hrc = m_pMachine->i_findSnapshotById(Guid() /* zero */, pSnapshot, true);
1242	if (SUCCEEDED(hrc) && !pSnapshot.isNull())
1243	pSnapshot->i_updateSavedStatePaths(sourcePath.c_str(),
1244	targetPath.c_str());
1245	if (m_pMachine->mSSData->strStateFilePath.isNotEmpty())
1246	{
1247	if (RTPathStartsWith(m_pMachine->mSSData->strStateFilePath.c_str(), sourcePath.c_str()))
1248	m_pMachine->mSSData->strStateFilePath = Utf8StrFmt("%s%s",
1249	targetPath.c_str(),
1250	m_pMachine->mSSData->strStateFilePath.c_str() + sourcePath.length());
1251	else
1252	m_pMachine->mSSData->strStateFilePath = Utf8StrFmt("%s%c%s",
1253	targetPath.c_str(),
1254	RTPATH_DELIMITER,
1255	RTPathFilename(m_pMachine->mSSData->strStateFilePath.c_str()));
1256	}
1257	}
1258
1259	void MachineMoveVM::updatePathsToNVRAMFiles(const Utf8Str &sourcePath, const Utf8Str &targetPath)
1260	{
1261	ComObjPtr<Snapshot> pSnapshot;
1262	HRESULT hrc = m_pMachine->i_findSnapshotById(Guid() /* zero */, pSnapshot, true);
1263	if (SUCCEEDED(hrc) && !pSnapshot.isNull())
1264	pSnapshot->i_updateNVRAMPaths(sourcePath.c_str(),
1265	targetPath.c_str());
1266	ComObjPtr<NvramStore> pNvramStore(m_pMachine->mNvramStore);
1267	const Utf8Str NVRAMFile(pNvramStore->i_getNonVolatileStorageFile());
1268	if (NVRAMFile.isNotEmpty())
1269	{
1270	Utf8Str newNVRAMFile;
1271	if (RTPathStartsWith(NVRAMFile.c_str(), sourcePath.c_str()))
1272	newNVRAMFile = Utf8StrFmt("%s%s", targetPath.c_str(), NVRAMFile.c_str() + sourcePath.length());
1273	else
1274	newNVRAMFile = Utf8StrFmt("%s%c%s", targetPath.c_str(), RTPATH_DELIMITER, RTPathFilename(newNVRAMFile.c_str()));
1275	pNvramStore->i_updateNonVolatileStorageFile(newNVRAMFile);
1276	}
1277	}
1278
1279	HRESULT MachineMoveVM::getFilesList(const Utf8Str &strRootFolder, fileList_t &filesList)
1280	{
1281	RTDIR hDir;
1282	HRESULT hrc = S_OK;
1283	int vrc = RTDirOpen(&hDir, strRootFolder.c_str());
1284	if (RT_SUCCESS(vrc))
1285	{
1286	/** @todo r=bird: RTDIRENTRY is big and this function is doing
1287	* unrestrained recursion of arbritrary depth. Four things:
1288	* - Add a depth counter parameter and refuse to go deeper than
1289	* a certain reasonable limit.
1290	* - Split this method into a main and a worker, placing
1291	* RTDIRENTRY on the stack in the main and passing it onto to
1292	* worker as a parameter.
1293	* - RTDirRead may fail for reasons other than
1294	* VERR_NO_MORE_FILES. For instance someone could create an
1295	* entry with a name longer than RTDIRENTRY have space to
1296	* store (windows host with UTF-16 encoding shorter than 255
1297	* chars, but UTF-8 encoding longer than 260).
1298	* - enmType can be RTDIRENTRYTYPE_UNKNOWN if the file system or
1299	* the host doesn't return the information. See
1300	* RTDIRENTRY::enmType. Use RTDirQueryUnknownType() to get the
1301	* actual type. */
1302	RTDIRENTRY DirEntry;
1303	while (RT_SUCCESS(RTDirRead(hDir, &DirEntry, NULL)))
1304	{
1305	if (RTDirEntryIsStdDotLink(&DirEntry))
1306	continue;
1307
1308	if (DirEntry.enmType == RTDIRENTRYTYPE_FILE)
1309	{
1310	Utf8Str fullPath(strRootFolder);
1311	filesList.add(strRootFolder, DirEntry.szName);
1312	}
1313	else if (DirEntry.enmType == RTDIRENTRYTYPE_DIRECTORY)
1314	{
1315	Utf8Str strNextFolder(strRootFolder);
1316	strNextFolder.append(RTPATH_DELIMITER).append(DirEntry.szName);
1317	hrc = getFilesList(strNextFolder, filesList);
1318	if (FAILED(hrc))
1319	break;
1320	}
1321	}
1322
1323	vrc = RTDirClose(hDir);
1324	AssertRC(vrc);
1325	}
1326	else if (vrc == VERR_FILE_NOT_FOUND)
1327	hrc = m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
1328	tr("Folder '%s' doesn't exist (%Rrc)"),
1329	strRootFolder.c_str(), vrc);
1330	else
1331	hrc = m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
1332	tr("Could not open folder '%s' (%Rrc)"),
1333	strRootFolder.c_str(), vrc);
1334
1335	return hrc;
1336	}
1337
1338	HRESULT MachineMoveVM::deleteFiles(const RTCList<Utf8Str> &listOfFiles)
1339	{
1340	HRESULT hrc = S_OK;
1341	/* Delete all created files. */
1342	for (size_t i = 0; i < listOfFiles.size(); ++i)
1343	{
1344	Log2(("Deleting file %s ...\n", listOfFiles.at(i).c_str()));
1345	hrc = m_pProgress->SetNextOperation(BstrFmt(tr("Deleting file %s..."), listOfFiles.at(i).c_str()).raw(), 1);
1346	if (FAILED(hrc)) return hrc;
1347
1348	int vrc = RTFileDelete(listOfFiles.at(i).c_str());
1349	if (RT_FAILURE(vrc))
1350	return m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
1351	tr("Could not delete file '%s' (%Rrc)"),
1352	listOfFiles.at(i).c_str(), vrc);
1353
1354	else
1355	Log2(("File %s has been deleted\n", listOfFiles.at(i).c_str()));
1356	}
1357
1358	return hrc;
1359	}
1360
1361	HRESULT MachineMoveVM::getFolderSize(const Utf8Str &strRootFolder, uint64_t &size)
1362	{
1363	HRESULT hrc = S_OK;
1364	int vrc = 0;
1365	uint64_t totalFolderSize = 0;
1366	fileList_t filesList;
1367
1368	bool ex = RTPathExists(strRootFolder.c_str());
1369	if (ex == true)
1370	{
1371	hrc = getFilesList(strRootFolder, filesList);
1372	if (SUCCEEDED(hrc))
1373	{
1374	cit_t it = filesList.m_list.begin();
1375	while (it != filesList.m_list.end())
1376	{
1377	uint64_t cbFile = 0;
1378	Utf8Str fullPath = it->first;
1379	fullPath.append(RTPATH_DELIMITER).append(it->second);
1380	vrc = RTFileQuerySizeByPath(fullPath.c_str(), &cbFile);
1381	if (RT_SUCCESS(vrc))
1382	{
1383	totalFolderSize += cbFile;
1384	}
1385	else
1386	return m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
1387	tr("Could not get the size of file '%s': %Rrc"),
1388	fullPath.c_str(),
1389	vrc);
1390
1391	++it;
1392	}
1393
1394	size = totalFolderSize;
1395	}
1396	}
1397	else
1398	size = 0;
1399
1400	return hrc;
1401	}
1402
1403	HRESULT MachineMoveVM::queryBaseName(const ComPtr<IMedium> &pMedium, Utf8Str &strBaseName) const
1404	{
1405	ComPtr<IMedium> pBaseMedium;
1406	HRESULT hrc = pMedium->COMGETTER(Base)(pBaseMedium.asOutParam());
1407	if (FAILED(hrc)) return hrc;
1408	Bstr bstrBaseName;
1409	hrc = pBaseMedium->COMGETTER(Name)(bstrBaseName.asOutParam());
1410	if (FAILED(hrc)) return hrc;
1411	strBaseName = bstrBaseName;
1412	return hrc;
1413	}
1414
1415	HRESULT MachineMoveVM::createMachineList(const ComPtr<ISnapshot> &pSnapshot)
1416	{
1417	Bstr name;
1418	HRESULT hrc = pSnapshot->COMGETTER(Name)(name.asOutParam());
1419	if (FAILED(hrc)) return hrc;
1420
1421	ComPtr<IMachine> l_pMachine;
1422	hrc = pSnapshot->COMGETTER(Machine)(l_pMachine.asOutParam());
1423	if (FAILED(hrc)) return hrc;
1424	machineList.push_back((Machine)(IMachine)l_pMachine);
1425
1426	SafeIfaceArray<ISnapshot> sfaChilds;
1427	hrc = pSnapshot->COMGETTER(Children)(ComSafeArrayAsOutParam(sfaChilds));
1428	if (FAILED(hrc)) return hrc;
1429	for (size_t i = 0; i < sfaChilds.size(); ++i)
1430	{
1431	hrc = createMachineList(sfaChilds[i]);
1432	if (FAILED(hrc)) return hrc;
1433	}
1434
1435	return hrc;
1436	}
1437
1438	HRESULT MachineMoveVM::queryMediaForAllStates()
1439	{
1440	/* In this case we create a exact copy of the original VM. This means just
1441	* adding all directly and indirectly attached disk images to the worker
1442	* list. */
1443	HRESULT hrc = S_OK;
1444	for (size_t i = 0; i < machineList.size(); ++i)
1445	{
1446	const ComObjPtr<Machine> &machine = machineList.at(i);
1447
1448	/* Add all attachments (and their parents) of the different
1449	* machines to a worker list. */
1450	SafeIfaceArray<IMediumAttachment> sfaAttachments;
1451	hrc = machine->COMGETTER(MediumAttachments)(ComSafeArrayAsOutParam(sfaAttachments));
1452	if (FAILED(hrc)) return hrc;
1453	for (size_t a = 0; a < sfaAttachments.size(); ++a)
1454	{
1455	const ComPtr<IMediumAttachment> &pAtt = sfaAttachments[a];
1456	DeviceType_T deviceType;//floppy, hard, DVD
1457	hrc = pAtt->COMGETTER(Type)(&deviceType);
1458	if (FAILED(hrc)) return hrc;
1459
1460	/* Valid medium attached? */
1461	ComPtr<IMedium> pMedium;
1462	hrc = pAtt->COMGETTER(Medium)(pMedium.asOutParam());
1463	if (FAILED(hrc)) return hrc;
1464
1465	if (pMedium.isNull())
1466	continue;
1467
1468	Bstr bstrLocation;
1469	hrc = pMedium->COMGETTER(Location)(bstrLocation.asOutParam());
1470	if (FAILED(hrc)) return hrc;
1471
1472	/* Cast to ComObjPtr<Medium> */
1473	ComObjPtr<Medium> pObjMedium = (Medium )(IMedium )pMedium;
1474
1475	/* Check for "read-only" medium in terms that VBox can't create this one */
1476	hrc = isMediumTypeSupportedForMoving(pMedium);
1477	if (FAILED(hrc))
1478	{
1479	if (hrc != S_FALSE)
1480	return hrc;
1481	Log2(("Skipping file %ls because of this medium type hasn't been supported for moving.\n", bstrLocation.raw()));
1482	continue;
1483	}
1484
1485	MEDIUMTASKCHAINMOVE mtc;
1486	mtc.devType = deviceType;
1487	mtc.fAttachLinked = false;
1488	mtc.fCreateDiffs = false;
1489
1490	while (!pMedium.isNull())
1491	{
1492	/* Refresh the state so that the file size get read. */
1493	MediumState_T e;
1494	hrc = pMedium->RefreshState(&e);
1495	if (FAILED(hrc)) return hrc;
1496
1497	LONG64 lSize;
1498	hrc = pMedium->COMGETTER(Size)(&lSize);
1499	if (FAILED(hrc)) return hrc;
1500
1501	MediumType_T mediumType;//immutable, shared, passthrough
1502	hrc = pMedium->COMGETTER(Type)(&mediumType);
1503	if (FAILED(hrc)) return hrc;
1504
1505	hrc = pMedium->COMGETTER(Location)(bstrLocation.asOutParam());
1506	if (FAILED(hrc)) return hrc;
1507
1508	MEDIUMTASKMOVE mt;// = {false, "basename", NULL, 0, 0};
1509	mt.strBaseName = bstrLocation;
1510	Utf8Str const &strFolder = m_vmFolders[VBox_SnapshotFolder];
1511
1512	if (strFolder.isNotEmpty() && RTPathStartsWith(mt.strBaseName.c_str(), strFolder.c_str()))
1513	mt.fSnapshot = true;
1514	else
1515	mt.fSnapshot = false;
1516
1517	mt.uIdx = UINT32_MAX;
1518	mt.pMedium = pMedium;
1519	mt.uWeight = (ULONG)((lSize + _1M - 1) / _1M);
1520	mtc.chain.append(mt);
1521
1522	/* Query next parent. */
1523	hrc = pMedium->COMGETTER(Parent)(pMedium.asOutParam());
1524	if (FAILED(hrc)) return hrc;
1525	}
1526
1527	m_llMedia.append(mtc);
1528	}
1529
1530	/* Add the save state files of this machine if there is one. */
1531	hrc = addSaveState(machine);
1532	if (FAILED(hrc)) return hrc;
1533
1534	/* Add the NVRAM files of this machine if there is one. */
1535	hrc = addNVRAM(machine);
1536	if (FAILED(hrc)) return hrc;
1537	}
1538
1539	/* Build up the index list of the image chain. Unfortunately we can't do
1540	* that in the previous loop, cause there we go from child -> parent and
1541	* didn't know how many are between. */
1542	for (size_t i = 0; i < m_llMedia.size(); ++i)
1543	{
1544	uint32_t uIdx = 0;
1545	MEDIUMTASKCHAINMOVE &mtc = m_llMedia.at(i);
1546	for (size_t a = mtc.chain.size(); a > 0; --a)
1547	mtc.chain[a - 1].uIdx = uIdx++;
1548	}
1549
1550	return hrc;
1551	}
1552
1553	HRESULT MachineMoveVM::addSaveState(const ComObjPtr<Machine> &machine)
1554	{
1555	Bstr bstrSrcSaveStatePath;
1556	HRESULT hrc = machine->COMGETTER(StateFilePath)(bstrSrcSaveStatePath.asOutParam());
1557	if (FAILED(hrc)) return hrc;
1558	if (!bstrSrcSaveStatePath.isEmpty())
1559	{
1560	SNAPFILETASKMOVE sft;
1561
1562	sft.snapshotUuid = machine->i_getSnapshotId();
1563	sft.strFile = bstrSrcSaveStatePath;
1564	uint64_t cbSize;
1565
1566	int vrc = RTFileQuerySizeByPath(sft.strFile.c_str(), &cbSize);
1567	if (RT_FAILURE(vrc))
1568	return m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
1569	tr("Could not get file size of '%s': %Rrc"),
1570	sft.strFile.c_str(),
1571	vrc);
1572
1573	/* same rule as above: count both the data which needs to
1574	* be read and written */
1575	sft.uWeight = (ULONG)(2 * (cbSize + _1M - 1) / _1M);
1576	m_llSaveStateFiles.append(sft);
1577	}
1578	return S_OK;
1579	}
1580
1581	HRESULT MachineMoveVM::addNVRAM(const ComObjPtr<Machine> &machine)
1582	{
1583	ComPtr<INvramStore> pNvramStore;
1584	HRESULT hrc = machine->COMGETTER(NonVolatileStore)(pNvramStore.asOutParam());
1585	if (FAILED(hrc)) return hrc;
1586	Bstr bstrSrcNVRAMPath;
1587	hrc = pNvramStore->COMGETTER(NonVolatileStorageFile)(bstrSrcNVRAMPath.asOutParam());
1588	if (FAILED(hrc)) return hrc;
1589	Utf8Str strSrcNVRAMPath(bstrSrcNVRAMPath);
1590	if (!strSrcNVRAMPath.isEmpty() && RTFileExists(strSrcNVRAMPath.c_str()))
1591	{
1592	SNAPFILETASKMOVE sft;
1593
1594	sft.snapshotUuid = machine->i_getSnapshotId();
1595	sft.strFile = strSrcNVRAMPath;
1596	uint64_t cbSize;
1597
1598	int vrc = RTFileQuerySizeByPath(sft.strFile.c_str(), &cbSize);
1599	if (RT_FAILURE(vrc))
1600	return m_pMachine->setErrorBoth(VBOX_E_IPRT_ERROR, vrc,
1601	tr("Could not get file size of '%s': %Rrc"),
1602	sft.strFile.c_str(),
1603	vrc);
1604
1605	/* same rule as above: count both the data which needs to
1606	* be read and written */
1607	sft.uWeight = (ULONG)(2 * (cbSize + _1M - 1) / _1M);
1608	m_llNVRAMFiles.append(sft);
1609	}
1610	return S_OK;
1611	}
1612
1613	void MachineMoveVM::updateProgressStats(MEDIUMTASKCHAINMOVE &mtc, ULONG &uCount, ULONG &uTotalWeight) const
1614	{
1615
1616	/* Currently the copying of diff images involves reading at least
1617	* the biggest parent in the previous chain. So even if the new
1618	* diff image is small in size, it could need some time to create
1619	* it. Adding the biggest size in the chain should balance this a
1620	* little bit more, i.e. the weight is the sum of the data which
1621	* needs to be read and written. */
1622	ULONG uMaxWeight = 0;
1623	for (size_t e = mtc.chain.size(); e > 0; --e)
1624	{
1625	MEDIUMTASKMOVE &mt = mtc.chain.at(e - 1);
1626	mt.uWeight += uMaxWeight;
1627
1628	/* Calculate progress data */
1629	++uCount;
1630	uTotalWeight += mt.uWeight;
1631
1632	/* Save the max size for better weighting of diff image
1633	* creation. */
1634	uMaxWeight = RT_MAX(uMaxWeight, mt.uWeight);
1635	}
1636	}
1637
1638	HRESULT MachineMoveVM::isMediumTypeSupportedForMoving(const ComPtr<IMedium> &pMedium)
1639	{
1640	Bstr bstrLocation;
1641	HRESULT hrc = pMedium->COMGETTER(Location)(bstrLocation.asOutParam());
1642	if (FAILED(hrc))
1643	return hrc;
1644
1645	DeviceType_T deviceType;
1646	hrc = pMedium->COMGETTER(DeviceType)(&deviceType);
1647	if (FAILED(hrc))
1648	return hrc;
1649
1650	ComPtr<IMediumFormat> mediumFormat;
1651	hrc = pMedium->COMGETTER(MediumFormat)(mediumFormat.asOutParam());
1652	if (FAILED(hrc))
1653	return hrc;
1654
1655	/* Check whether VBox is able to create this medium format or not, i.e. medium can be "read-only" */
1656	Bstr bstrFormatName;
1657	hrc = mediumFormat->COMGETTER(Name)(bstrFormatName.asOutParam());
1658	if (FAILED(hrc))
1659	return hrc;
1660
1661	Utf8Str formatName = Utf8Str(bstrFormatName);
1662	if (formatName.compare("VHDX", Utf8Str::CaseInsensitive) == 0)
1663	{
1664	Log2(("Skipping medium %ls. VHDX format is supported in \"read-only\" mode only.\n", bstrLocation.raw()));
1665	return S_FALSE;
1666	}
1667
1668	/* Check whether medium is represented by file on the disk or not */
1669	ComObjPtr<Medium> pObjMedium = (Medium )(IMedium )pMedium;
1670	if (!pObjMedium->i_isMediumFormatFile())
1671	{
1672	Log2(("Skipping medium %ls because it's not a real file on the disk.\n", bstrLocation.raw()));
1673	return S_FALSE;
1674	}
1675
1676	/* some special checks for DVD */
1677	if (deviceType == DeviceType_DVD)
1678	{
1679	Utf8Str ext = bstrLocation;
1680	/* only ISO image is moved */
1681	if (!ext.endsWith(".iso", Utf8Str::CaseInsensitive))
1682	{
1683	Log2(("Skipping file %ls. Only ISO images are supported for now.\n", bstrLocation.raw()));
1684	return S_FALSE;
1685	}
1686	}
1687
1688	return S_OK;
1689	}

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Main/src-server/MachineImplMoveVM.cpp@ 100841

Download in other formats: