ApplianceImpl.cpp@ 27908

Last change on this file since 27908 was 27908, checked in by vboxsync, 15 years ago
Main/OVF: add c++ ovf namespace
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 44.4 KB

Line
1	/* $Id: ApplianceImpl.cpp 27908 2010-03-31 14:32:46Z vboxsync $ */
2	/** @file
3	*
4	* IAppliance and IVirtualSystem COM class implementations.
5	*/
6
7	/*
8	* Copyright (C) 2008-2010 Sun Microsystems, Inc.
9	*
10	* This file is part of VirtualBox Open Source Edition (OSE), as
11	* available from http://www.virtualbox.org. This file is free software;
12	* you can redistribute it and/or modify it under the terms of the GNU
13	* General Public License (GPL) as published by the Free Software
14	* Foundation, in version 2 as it comes in the "COPYING" file of the
15	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
16	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
17	*
18	* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
19	* Clara, CA 95054 USA or visit http://www.sun.com if you need
20	* additional information or have any questions.
21	*/
22
23	#include <iprt/path.h>
24
25	#include <VBox/com/array.h>
26
27	#include "ApplianceImpl.h"
28	#include "VFSExplorerImpl.h"
29	#include "VirtualBoxImpl.h"
30	#include "GuestOSTypeImpl.h"
31	#include "ProgressImpl.h"
32	#include "MachineImpl.h"
33
34	#include "AutoCaller.h"
35	#include "Logging.h"
36
37	#include "ApplianceImplPrivate.h"
38
39	using namespace std;
40
41	////////////////////////////////////////////////////////////////////////////////
42	//
43	// Internal helpers
44	//
45	////////////////////////////////////////////////////////////////////////////////
46
47	static const struct
48	{
49	ovf::CIMOSType_T cim;
50	const char *pcszVbox;
51	}
52	g_osTypes[] =
53	{
54	{ ovf::CIMOSType_CIMOS_Unknown, SchemaDefs_OSTypeId_Other },
55	{ ovf::CIMOSType_CIMOS_OS2, SchemaDefs_OSTypeId_OS2 },
56	{ ovf::CIMOSType_CIMOS_MSDOS, SchemaDefs_OSTypeId_DOS },
57	{ ovf::CIMOSType_CIMOS_WIN3x, SchemaDefs_OSTypeId_Windows31 },
58	{ ovf::CIMOSType_CIMOS_WIN95, SchemaDefs_OSTypeId_Windows95 },
59	{ ovf::CIMOSType_CIMOS_WIN98, SchemaDefs_OSTypeId_Windows98 },
60	{ ovf::CIMOSType_CIMOS_WINNT, SchemaDefs_OSTypeId_WindowsNT4 },
61	{ ovf::CIMOSType_CIMOS_NetWare, SchemaDefs_OSTypeId_Netware },
62	{ ovf::CIMOSType_CIMOS_NovellOES, SchemaDefs_OSTypeId_Netware },
63	{ ovf::CIMOSType_CIMOS_Solaris, SchemaDefs_OSTypeId_OpenSolaris },
64	{ ovf::CIMOSType_CIMOS_SunOS, SchemaDefs_OSTypeId_OpenSolaris },
65	{ ovf::CIMOSType_CIMOS_FreeBSD, SchemaDefs_OSTypeId_FreeBSD },
66	{ ovf::CIMOSType_CIMOS_NetBSD, SchemaDefs_OSTypeId_NetBSD },
67	{ ovf::CIMOSType_CIMOS_QNX, SchemaDefs_OSTypeId_QNX },
68	{ ovf::CIMOSType_CIMOS_Windows2000, SchemaDefs_OSTypeId_Windows2000 },
69	{ ovf::CIMOSType_CIMOS_WindowsMe, SchemaDefs_OSTypeId_WindowsMe },
70	{ ovf::CIMOSType_CIMOS_OpenBSD, SchemaDefs_OSTypeId_OpenBSD },
71	{ ovf::CIMOSType_CIMOS_WindowsXP, SchemaDefs_OSTypeId_WindowsXP },
72	{ ovf::CIMOSType_CIMOS_WindowsXPEmbedded, SchemaDefs_OSTypeId_WindowsXP },
73	{ ovf::CIMOSType_CIMOS_WindowsEmbeddedforPointofService, SchemaDefs_OSTypeId_WindowsXP },
74	{ ovf::CIMOSType_CIMOS_MicrosoftWindowsServer2003, SchemaDefs_OSTypeId_Windows2003 },
75	{ ovf::CIMOSType_CIMOS_MicrosoftWindowsServer2003_64, SchemaDefs_OSTypeId_Windows2003_64 },
76	{ ovf::CIMOSType_CIMOS_WindowsXP_64, SchemaDefs_OSTypeId_WindowsXP_64 },
77	{ ovf::CIMOSType_CIMOS_WindowsVista, SchemaDefs_OSTypeId_WindowsVista },
78	{ ovf::CIMOSType_CIMOS_WindowsVista_64, SchemaDefs_OSTypeId_WindowsVista_64 },
79	{ ovf::CIMOSType_CIMOS_MicrosoftWindowsServer2008, SchemaDefs_OSTypeId_Windows2008 },
80	{ ovf::CIMOSType_CIMOS_MicrosoftWindowsServer2008_64, SchemaDefs_OSTypeId_Windows2008_64 },
81	{ ovf::CIMOSType_CIMOS_FreeBSD_64, SchemaDefs_OSTypeId_FreeBSD_64 },
82	{ ovf::CIMOSType_CIMOS_RedHatEnterpriseLinux, SchemaDefs_OSTypeId_RedHat },
83	{ ovf::CIMOSType_CIMOS_RedHatEnterpriseLinux_64, SchemaDefs_OSTypeId_RedHat_64 },
84	{ ovf::CIMOSType_CIMOS_Solaris_64, SchemaDefs_OSTypeId_OpenSolaris_64 },
85	{ ovf::CIMOSType_CIMOS_SUSE, SchemaDefs_OSTypeId_OpenSUSE },
86	{ ovf::CIMOSType_CIMOS_SLES, SchemaDefs_OSTypeId_OpenSUSE },
87	{ ovf::CIMOSType_CIMOS_NovellLinuxDesktop, SchemaDefs_OSTypeId_OpenSUSE },
88	{ ovf::CIMOSType_CIMOS_SUSE_64, SchemaDefs_OSTypeId_OpenSUSE_64 },
89	{ ovf::CIMOSType_CIMOS_SLES_64, SchemaDefs_OSTypeId_OpenSUSE_64 },
90	{ ovf::CIMOSType_CIMOS_LINUX, SchemaDefs_OSTypeId_Linux },
91	{ ovf::CIMOSType_CIMOS_SunJavaDesktopSystem, SchemaDefs_OSTypeId_Linux },
92	{ ovf::CIMOSType_CIMOS_TurboLinux, SchemaDefs_OSTypeId_Linux},
93
94	// { ovf::CIMOSType_CIMOS_TurboLinux_64, },
95
96	{ ovf::CIMOSType_CIMOS_Mandriva, SchemaDefs_OSTypeId_Mandriva },
97	{ ovf::CIMOSType_CIMOS_Mandriva_64, SchemaDefs_OSTypeId_Mandriva_64 },
98	{ ovf::CIMOSType_CIMOS_Ubuntu, SchemaDefs_OSTypeId_Ubuntu },
99	{ ovf::CIMOSType_CIMOS_Ubuntu_64, SchemaDefs_OSTypeId_Ubuntu_64 },
100	{ ovf::CIMOSType_CIMOS_Debian, SchemaDefs_OSTypeId_Debian },
101	{ ovf::CIMOSType_CIMOS_Debian_64, SchemaDefs_OSTypeId_Debian_64 },
102	{ ovf::CIMOSType_CIMOS_Linux_2_4_x, SchemaDefs_OSTypeId_Linux24 },
103	{ ovf::CIMOSType_CIMOS_Linux_2_4_x_64, SchemaDefs_OSTypeId_Linux24_64 },
104	{ ovf::CIMOSType_CIMOS_Linux_2_6_x, SchemaDefs_OSTypeId_Linux26 },
105	{ ovf::CIMOSType_CIMOS_Linux_2_6_x_64, SchemaDefs_OSTypeId_Linux26_64 },
106	{ ovf::CIMOSType_CIMOS_Linux_64, SchemaDefs_OSTypeId_Linux26_64 }
107	};
108
109	/* Pattern structure for matching the OS type description field */
110	struct osTypePattern
111	{
112	const char *pcszPattern;
113	const char *pcszVbox;
114	};
115
116	/* These are the 32-Bit ones. They are sorted by priority. */
117	static const osTypePattern g_osTypesPattern[] =
118	{
119	{"Windows NT", SchemaDefs_OSTypeId_WindowsNT4},
120	{"Windows XP", SchemaDefs_OSTypeId_WindowsXP},
121	{"Windows 2000", SchemaDefs_OSTypeId_Windows2000},
122	{"Windows 2003", SchemaDefs_OSTypeId_Windows2003},
123	{"Windows Vista", SchemaDefs_OSTypeId_WindowsVista},
124	{"Windows 2008", SchemaDefs_OSTypeId_Windows2008},
125	{"SUSE", SchemaDefs_OSTypeId_OpenSUSE},
126	{"Novell", SchemaDefs_OSTypeId_OpenSUSE},
127	{"Red Hat", SchemaDefs_OSTypeId_RedHat},
128	{"Mandriva", SchemaDefs_OSTypeId_Mandriva},
129	{"Ubuntu", SchemaDefs_OSTypeId_Ubuntu},
130	{"Debian", SchemaDefs_OSTypeId_Debian},
131	{"QNX", SchemaDefs_OSTypeId_QNX},
132	{"Linux 2.4", SchemaDefs_OSTypeId_Linux24},
133	{"Linux 2.6", SchemaDefs_OSTypeId_Linux26},
134	{"Linux", SchemaDefs_OSTypeId_Linux},
135	{"OpenSolaris", SchemaDefs_OSTypeId_OpenSolaris},
136	{"Solaris", SchemaDefs_OSTypeId_OpenSolaris},
137	{"FreeBSD", SchemaDefs_OSTypeId_FreeBSD},
138	{"NetBSD", SchemaDefs_OSTypeId_NetBSD},
139	{"Windows 95", SchemaDefs_OSTypeId_Windows95},
140	{"Windows 98", SchemaDefs_OSTypeId_Windows98},
141	{"Windows Me", SchemaDefs_OSTypeId_WindowsMe},
142	{"Windows 3.", SchemaDefs_OSTypeId_Windows31},
143	{"DOS", SchemaDefs_OSTypeId_DOS},
144	{"OS2", SchemaDefs_OSTypeId_OS2}
145	};
146
147	/* These are the 64-Bit ones. They are sorted by priority. */
148	static const osTypePattern g_osTypesPattern64[] =
149	{
150	{"Windows XP", SchemaDefs_OSTypeId_WindowsXP_64},
151	{"Windows 2003", SchemaDefs_OSTypeId_Windows2003_64},
152	{"Windows Vista", SchemaDefs_OSTypeId_WindowsVista_64},
153	{"Windows 2008", SchemaDefs_OSTypeId_Windows2008_64},
154	{"SUSE", SchemaDefs_OSTypeId_OpenSUSE_64},
155	{"Novell", SchemaDefs_OSTypeId_OpenSUSE_64},
156	{"Red Hat", SchemaDefs_OSTypeId_RedHat_64},
157	{"Mandriva", SchemaDefs_OSTypeId_Mandriva_64},
158	{"Ubuntu", SchemaDefs_OSTypeId_Ubuntu_64},
159	{"Debian", SchemaDefs_OSTypeId_Debian_64},
160	{"Linux 2.4", SchemaDefs_OSTypeId_Linux24_64},
161	{"Linux 2.6", SchemaDefs_OSTypeId_Linux26_64},
162	{"Linux", SchemaDefs_OSTypeId_Linux26_64},
163	{"OpenSolaris", SchemaDefs_OSTypeId_OpenSolaris_64},
164	{"Solaris", SchemaDefs_OSTypeId_OpenSolaris_64},
165	{"FreeBSD", SchemaDefs_OSTypeId_FreeBSD_64},
166	};
167
168	/**
169	* Private helper func that suggests a VirtualBox guest OS type
170	* for the given OVF operating system type.
171	* @param osTypeVBox
172	* @param c
173	* @param cStr
174	*/
175	void convertCIMOSType2VBoxOSType(Utf8Str &strType, ovf::CIMOSType_T c, const Utf8Str &cStr)
176	{
177	/* First check if the type is other/other_64 */
178	if (c == ovf::CIMOSType_CIMOS_Other)
179	{
180	for (size_t i=0; i < RT_ELEMENTS(g_osTypesPattern); ++i)
181	if (cStr.contains (g_osTypesPattern[i].pcszPattern, Utf8Str::CaseInsensitive))
182	{
183	strType = g_osTypesPattern[i].pcszVbox;
184	return;
185	}
186	}
187	else if (c == ovf::CIMOSType_CIMOS_Other_64)
188	{
189	for (size_t i=0; i < RT_ELEMENTS(g_osTypesPattern64); ++i)
190	if (cStr.contains (g_osTypesPattern64[i].pcszPattern, Utf8Str::CaseInsensitive))
191	{
192	strType = g_osTypesPattern64[i].pcszVbox;
193	return;
194	}
195	}
196
197	for (size_t i = 0; i < RT_ELEMENTS(g_osTypes); ++i)
198	{
199	if (c == g_osTypes[i].cim)
200	{
201	strType = g_osTypes[i].pcszVbox;
202	return;
203	}
204	}
205
206	strType = SchemaDefs_OSTypeId_Other;
207	}
208
209	/**
210	* Private helper func that suggests a VirtualBox guest OS type
211	* for the given OVF operating system type.
212	* @param osTypeVBox
213	* @param c
214	*/
215	ovf::CIMOSType_T convertVBoxOSType2CIMOSType(const char *pcszVbox)
216	{
217	for (size_t i = 0; i < RT_ELEMENTS(g_osTypes); ++i)
218	{
219	if (!RTStrICmp(pcszVbox, g_osTypes[i].pcszVbox))
220	return g_osTypes[i].cim;
221	}
222
223	return ovf::CIMOSType_CIMOS_Other;
224	}
225
226	////////////////////////////////////////////////////////////////////////////////
227	//
228	// IVirtualBox public methods
229	//
230	////////////////////////////////////////////////////////////////////////////////
231
232	// This code is here so we won't have to include the appliance headers in the
233	// IVirtualBox implementation.
234
235	/**
236	* Implementation for IVirtualBox::createAppliance.
237	*
238	* @param anAppliance IAppliance object created if S_OK is returned.
239	* @return S_OK or error.
240	*/
241	STDMETHODIMP VirtualBox::CreateAppliance(IAppliance** anAppliance)
242	{
243	HRESULT rc;
244
245	ComObjPtr<Appliance> appliance;
246	appliance.createObject();
247	rc = appliance->init(this);
248
249	if (SUCCEEDED(rc))
250	appliance.queryInterfaceTo(anAppliance);
251
252	return rc;
253	}
254
255	////////////////////////////////////////////////////////////////////////////////
256	//
257	// Appliance constructor / destructor
258	//
259	////////////////////////////////////////////////////////////////////////////////
260
261	Appliance::Appliance()
262	: mVirtualBox(NULL)
263	{
264	}
265
266	Appliance::~Appliance()
267	{
268	}
269
270	/**
271	* Appliance COM initializer.
272	* @param
273	* @return
274	*/
275	HRESULT Appliance::init(VirtualBox *aVirtualBox)
276	{
277	/* Enclose the state transition NotReady->InInit->Ready */
278	AutoInitSpan autoInitSpan(this);
279	AssertReturn(autoInitSpan.isOk(), E_FAIL);
280
281	/* Weak reference to a VirtualBox object */
282	unconst(mVirtualBox) = aVirtualBox;
283
284	// initialize data
285	m = new Data;
286
287	/* Confirm a successful initialization */
288	autoInitSpan.setSucceeded();
289
290	return S_OK;
291	}
292
293	/**
294	* Appliance COM uninitializer.
295	* @return
296	*/
297	void Appliance::uninit()
298	{
299	/* Enclose the state transition Ready->InUninit->NotReady */
300	AutoUninitSpan autoUninitSpan(this);
301	if (autoUninitSpan.uninitDone())
302	return;
303
304	delete m;
305	m = NULL;
306	}
307
308	////////////////////////////////////////////////////////////////////////////////
309	//
310	// IAppliance public methods
311	//
312	////////////////////////////////////////////////////////////////////////////////
313
314	/**
315	* Public method implementation.
316	* @param
317	* @return
318	*/
319	STDMETHODIMP Appliance::COMGETTER(Path)(BSTR *aPath)
320	{
321	if (!aPath)
322	return E_POINTER;
323
324	AutoCaller autoCaller(this);
325	if (FAILED(autoCaller.rc())) return autoCaller.rc();
326
327	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
328
329	if (!isApplianceIdle())
330	return E_ACCESSDENIED;
331
332	Bstr bstrPath(m->locInfo.strPath);
333	bstrPath.cloneTo(aPath);
334
335	return S_OK;
336	}
337
338	/**
339	* Public method implementation.
340	* @param
341	* @return
342	*/
343	STDMETHODIMP Appliance::COMGETTER(Disks)(ComSafeArrayOut(BSTR, aDisks))
344	{
345	CheckComArgOutSafeArrayPointerValid(aDisks);
346
347	AutoCaller autoCaller(this);
348	if (FAILED(autoCaller.rc())) return autoCaller.rc();
349
350	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
351
352	if (!isApplianceIdle())
353	return E_ACCESSDENIED;
354
355	if (m->pReader) // OVFReader instantiated?
356	{
357	size_t c = m->pReader->m_mapDisks.size();
358	com::SafeArray<BSTR> sfaDisks(c);
359
360	ovf::DiskImagesMap::const_iterator it;
361	size_t i = 0;
362	for (it = m->pReader->m_mapDisks.begin();
363	it != m->pReader->m_mapDisks.end();
364	++it, ++i)
365	{
366	// create a string representing this disk
367	const ovf::DiskImage &d = it->second;
368	char *psz = NULL;
369	RTStrAPrintf(&psz,
370	"%s\t"
371	"%RI64\t"
372	"%RI64\t"
373	"%s\t"
374	"%s\t"
375	"%RI64\t"
376	"%RI64\t"
377	"%s",
378	d.strDiskId.c_str(),
379	d.iCapacity,
380	d.iPopulatedSize,
381	d.strFormat.c_str(),
382	d.strHref.c_str(),
383	d.iSize,
384	d.iChunkSize,
385	d.strCompression.c_str());
386	Utf8Str utf(psz);
387	Bstr bstr(utf);
388	// push to safearray
389	bstr.cloneTo(&sfaDisks[i]);
390	RTStrFree(psz);
391	}
392
393	sfaDisks.detachTo(ComSafeArrayOutArg(aDisks));
394	}
395
396	return S_OK;
397	}
398
399	/**
400	* Public method implementation.
401	* @param
402	* @return
403	*/
404	STDMETHODIMP Appliance::COMGETTER(VirtualSystemDescriptions)(ComSafeArrayOut(IVirtualSystemDescription*, aVirtualSystemDescriptions))
405	{
406	CheckComArgOutSafeArrayPointerValid(aVirtualSystemDescriptions);
407
408	AutoCaller autoCaller(this);
409	if (FAILED(autoCaller.rc())) return autoCaller.rc();
410
411	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
412
413	if (!isApplianceIdle())
414	return E_ACCESSDENIED;
415
416	SafeIfaceArray<IVirtualSystemDescription> sfaVSD(m->virtualSystemDescriptions);
417	sfaVSD.detachTo(ComSafeArrayOutArg(aVirtualSystemDescriptions));
418
419	return S_OK;
420	}
421
422	STDMETHODIMP Appliance::CreateVFSExplorer(IN_BSTR aURI, IVFSExplorer **aExplorer)
423	{
424	CheckComArgOutPointerValid(aExplorer);
425
426	AutoCaller autoCaller(this);
427	if (FAILED(autoCaller.rc())) return autoCaller.rc();
428
429	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
430
431	ComObjPtr<VFSExplorer> explorer;
432	HRESULT rc = S_OK;
433	try
434	{
435	Utf8Str uri(aURI);
436	/* Check which kind of export the user has requested */
437	LocationInfo li;
438	parseURI(uri, li);
439	/* Create the explorer object */
440	explorer.createObject();
441	rc = explorer->init(li.storageType, li.strPath, li.strHostname, li.strUsername, li.strPassword, mVirtualBox);
442	}
443	catch (HRESULT aRC)
444	{
445	rc = aRC;
446	}
447
448	if (SUCCEEDED(rc))
449	/* Return explorer to the caller */
450	explorer.queryInterfaceTo(aExplorer);
451
452	return rc;
453	}
454
455	/**
456	* Public method implementation.
457	* @return
458	*/
459	STDMETHODIMP Appliance::GetWarnings(ComSafeArrayOut(BSTR, aWarnings))
460	{
461	if (ComSafeArrayOutIsNull(aWarnings))
462	return E_POINTER;
463
464	AutoCaller autoCaller(this);
465	if (FAILED(autoCaller.rc())) return autoCaller.rc();
466
467	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
468
469	com::SafeArray<BSTR> sfaWarnings(m->llWarnings.size());
470
471	list<Utf8Str>::const_iterator it;
472	size_t i = 0;
473	for (it = m->llWarnings.begin();
474	it != m->llWarnings.end();
475	++it, ++i)
476	{
477	Bstr bstr = *it;
478	bstr.cloneTo(&sfaWarnings[i]);
479	}
480
481	sfaWarnings.detachTo(ComSafeArrayOutArg(aWarnings));
482
483	return S_OK;
484	}
485
486	////////////////////////////////////////////////////////////////////////////////
487	//
488	// Appliance private methods
489	//
490	////////////////////////////////////////////////////////////////////////////////
491
492	/**
493	* Returns true if the appliance is in "idle" state. This should always be the
494	* case unless an import or export is currently in progress. Similar to machine
495	* states, this permits the Appliance implementation code to let go of the
496	* Appliance object lock while a time-consuming disk conversion is in progress
497	* without exposing the appliance to conflicting calls.
498	*
499	* This sets an error on "this" (the appliance) and returns false if the appliance
500	* is busy. The caller should then return E_ACCESSDENIED.
501	*
502	* Must be called from under the object lock!
503	*
504	* @return
505	*/
506	bool Appliance::isApplianceIdle() const
507	{
508	if (m->state == Data::ApplianceImporting)
509	setError(VBOX_E_INVALID_OBJECT_STATE, "The appliance is busy importing files");
510	else if (m->state == Data::ApplianceExporting)
511	setError(VBOX_E_INVALID_OBJECT_STATE, "The appliance is busy exporting files");
512	else
513	return true;
514
515	return false;
516	}
517
518	HRESULT Appliance::searchUniqueVMName(Utf8Str& aName) const
519	{
520	IMachine *machine = NULL;
521	char *tmpName = RTStrDup(aName.c_str());
522	int i = 1;
523	/* @todo: Maybe too cost-intensive; try to find a lighter way */
524	while (mVirtualBox->FindMachine(Bstr(tmpName), &machine) != VBOX_E_OBJECT_NOT_FOUND)
525	{
526	RTStrFree(tmpName);
527	RTStrAPrintf(&tmpName, "%s_%d", aName.c_str(), i);
528	++i;
529	}
530	aName = tmpName;
531	RTStrFree(tmpName);
532
533	return S_OK;
534	}
535
536	HRESULT Appliance::searchUniqueDiskImageFilePath(Utf8Str& aName) const
537	{
538	IMedium *harddisk = NULL;
539	char *tmpName = RTStrDup(aName.c_str());
540	int i = 1;
541	/* Check if the file exists or if a file with this path is registered
542	* already */
543	/* @todo: Maybe too cost-intensive; try to find a lighter way */
544	while ( RTPathExists(tmpName)
545	\|\| mVirtualBox->FindHardDisk(Bstr(tmpName), &harddisk) != VBOX_E_OBJECT_NOT_FOUND
546	)
547	{
548	RTStrFree(tmpName);
549	char *tmpDir = RTStrDup(aName.c_str());
550	RTPathStripFilename(tmpDir);;
551	char *tmpFile = RTStrDup(RTPathFilename(aName.c_str()));
552	RTPathStripExt(tmpFile);
553	const char *tmpExt = RTPathExt(aName.c_str());
554	RTStrAPrintf(&tmpName, "%s%c%s_%d%s", tmpDir, RTPATH_DELIMITER, tmpFile, i, tmpExt);
555	RTStrFree(tmpFile);
556	RTStrFree(tmpDir);
557	++i;
558	}
559	aName = tmpName;
560	RTStrFree(tmpName);
561
562	return S_OK;
563	}
564
565	/**
566	* Called from the import and export background threads to synchronize the second
567	* background disk thread's progress object with the current progress object so
568	* that the user interface sees progress correctly and that cancel signals are
569	* passed on to the second thread.
570	* @param pProgressThis Progress object of the current thread.
571	* @param pProgressAsync Progress object of asynchronous task running in background.
572	*/
573	void Appliance::waitForAsyncProgress(ComObjPtr<Progress> &pProgressThis,
574	ComPtr<IProgress> &pProgressAsync)
575	{
576	HRESULT rc;
577
578	// now loop until the asynchronous operation completes and then report its result
579	BOOL fCompleted;
580	BOOL fCanceled;
581	ULONG currentPercent;
582	while (SUCCEEDED(pProgressAsync->COMGETTER(Completed(&fCompleted))))
583	{
584	rc = pProgressThis->COMGETTER(Canceled)(&fCanceled);
585	if (FAILED(rc)) throw rc;
586	if (fCanceled)
587	{
588	pProgressAsync->Cancel();
589	break;
590	}
591
592	rc = pProgressAsync->COMGETTER(Percent(&currentPercent));
593	if (FAILED(rc)) throw rc;
594	if (!pProgressThis.isNull())
595	pProgressThis->SetCurrentOperationProgress(currentPercent);
596	if (fCompleted)
597	break;
598
599	/* Make sure the loop is not too tight */
600	rc = pProgressAsync->WaitForCompletion(100);
601	if (FAILED(rc)) throw rc;
602	}
603	// report result of asynchronous operation
604	LONG iRc;
605	rc = pProgressAsync->COMGETTER(ResultCode)(&iRc);
606	if (FAILED(rc)) throw rc;
607
608
609	// if the thread of the progress object has an error, then
610	// retrieve the error info from there, or it'll be lost
611	if (FAILED(iRc))
612	{
613	ProgressErrorInfo info(pProgressAsync);
614	Utf8Str str(info.getText());
615	const char *pcsz = str.c_str();
616	HRESULT rc2 = setError(iRc, pcsz);
617	throw rc2;
618	}
619	}
620
621	void Appliance::addWarning(const char* aWarning, ...)
622	{
623	va_list args;
624	va_start(args, aWarning);
625	Utf8StrFmtVA str(aWarning, args);
626	va_end(args);
627	m->llWarnings.push_back(str);
628	}
629
630	void Appliance::disksWeight(uint32_t &ulTotalMB, uint32_t &cDisks) const
631	{
632	ulTotalMB = 0;
633	cDisks = 0;
634	/* Weigh the disk images according to their sizes */
635	list< ComObjPtr<VirtualSystemDescription> >::const_iterator it;
636	for (it = m->virtualSystemDescriptions.begin();
637	it != m->virtualSystemDescriptions.end();
638	++it)
639	{
640	ComObjPtr<VirtualSystemDescription> vsdescThis = (*it);
641	/* One for every hard disk of the Virtual System */
642	std::list<VirtualSystemDescriptionEntry*> avsdeHDs = vsdescThis->findByType(VirtualSystemDescriptionType_HardDiskImage);
643	std::list<VirtualSystemDescriptionEntry*>::const_iterator itH;
644	for (itH = avsdeHDs.begin();
645	itH != avsdeHDs.end();
646	++itH)
647	{
648	const VirtualSystemDescriptionEntry pHD = itH;
649	ulTotalMB += pHD->ulSizeMB;
650	++cDisks;
651	}
652	}
653
654	}
655
656	HRESULT Appliance::setUpProgressFS(ComObjPtr<Progress> &pProgress, const Bstr &bstrDescription)
657	{
658	HRESULT rc;
659
660	/* Create the progress object */
661	pProgress.createObject();
662
663	/* Weigh the disk images according to their sizes */
664	uint32_t ulTotalMB;
665	uint32_t cDisks;
666	disksWeight(ulTotalMB, cDisks);
667
668	ULONG cOperations = 1 + cDisks; // one op per disk plus 1 for the XML
669
670	ULONG ulTotalOperationsWeight;
671	if (ulTotalMB)
672	{
673	m->ulWeightPerOperation = (ULONG)((double)ulTotalMB * 1 / 100); // use 1% of the progress for the XML
674	ulTotalOperationsWeight = ulTotalMB + m->ulWeightPerOperation;
675	}
676	else
677	{
678	// no disks to export:
679	ulTotalOperationsWeight = 1;
680	m->ulWeightPerOperation = 1;
681	}
682
683	Log(("Setting up progress object: ulTotalMB = %d, cDisks = %d, => cOperations = %d, ulTotalOperationsWeight = %d, m->ulWeightPerOperation = %d\n",
684	ulTotalMB, cDisks, cOperations, ulTotalOperationsWeight, m->ulWeightPerOperation));
685
686	rc = pProgress->init(mVirtualBox, static_cast<IAppliance*>(this),
687	bstrDescription,
688	TRUE /* aCancelable */,
689	cOperations, // ULONG cOperations,
690	ulTotalOperationsWeight, // ULONG ulTotalOperationsWeight,
691	bstrDescription, // CBSTR bstrFirstOperationDescription,
692	m->ulWeightPerOperation); // ULONG ulFirstOperationWeight,
693	return rc;
694	}
695
696	HRESULT Appliance::setUpProgressImportS3(ComObjPtr<Progress> &pProgress, const Bstr &bstrDescription)
697	{
698	HRESULT rc;
699
700	/* Create the progress object */
701	pProgress.createObject();
702
703	/* Weigh the disk images according to their sizes */
704	uint32_t ulTotalMB;
705	uint32_t cDisks;
706	disksWeight(ulTotalMB, cDisks);
707
708	ULONG cOperations = 1 + 1 + 1 + cDisks; // one op per disk plus 1 for init, plus 1 for the manifest file & 1 plus for the import */
709
710	ULONG ulTotalOperationsWeight = ulTotalMB;
711	if (!ulTotalOperationsWeight)
712	// no disks to export:
713	ulTotalOperationsWeight = 1;
714
715	ULONG ulImportWeight = (ULONG)((double)ulTotalOperationsWeight * 50 / 100); // use 50% for import
716	ulTotalOperationsWeight += ulImportWeight;
717
718	m->ulWeightPerOperation = ulImportWeight; /* save for using later */
719
720	ULONG ulInitWeight = (ULONG)((double)ulTotalOperationsWeight * 0.1 / 100); // use 0.1% for init
721	ulTotalOperationsWeight += ulInitWeight;
722
723	Log(("Setting up progress object: ulTotalMB = %d, cDisks = %d, => cOperations = %d, ulTotalOperationsWeight = %d, m->ulWeightPerOperation = %d\n",
724	ulTotalMB, cDisks, cOperations, ulTotalOperationsWeight, m->ulWeightPerOperation));
725
726	rc = pProgress->init(mVirtualBox, static_cast<IAppliance*>(this),
727	bstrDescription,
728	TRUE /* aCancelable */,
729	cOperations, // ULONG cOperations,
730	ulTotalOperationsWeight, // ULONG ulTotalOperationsWeight,
731	Bstr(tr("Init")), // CBSTR bstrFirstOperationDescription,
732	ulInitWeight); // ULONG ulFirstOperationWeight,
733	return rc;
734	}
735
736	HRESULT Appliance::setUpProgressWriteS3(ComObjPtr<Progress> &pProgress, const Bstr &bstrDescription)
737	{
738	HRESULT rc;
739
740	/* Create the progress object */
741	pProgress.createObject();
742
743	/* Weigh the disk images according to their sizes */
744	uint32_t ulTotalMB;
745	uint32_t cDisks;
746	disksWeight(ulTotalMB, cDisks);
747
748	ULONG cOperations = 1 + 1 + 1 + cDisks; // one op per disk plus 1 for the OVF, plus 1 for the mf & 1 plus to the temporary creation */
749
750	ULONG ulTotalOperationsWeight;
751	if (ulTotalMB)
752	{
753	m->ulWeightPerOperation = (ULONG)((double)ulTotalMB * 1 / 100); // use 1% of the progress for OVF file upload (we didn't know the size at this point)
754	ulTotalOperationsWeight = ulTotalMB + m->ulWeightPerOperation;
755	}
756	else
757	{
758	// no disks to export:
759	ulTotalOperationsWeight = 1;
760	m->ulWeightPerOperation = 1;
761	}
762	ULONG ulOVFCreationWeight = (ULONG)((double)ulTotalOperationsWeight * 50.0 / 100.0); /* Use 50% for the creation of the OVF & the disks */
763	ulTotalOperationsWeight += ulOVFCreationWeight;
764
765	Log(("Setting up progress object: ulTotalMB = %d, cDisks = %d, => cOperations = %d, ulTotalOperationsWeight = %d, m->ulWeightPerOperation = %d\n",
766	ulTotalMB, cDisks, cOperations, ulTotalOperationsWeight, m->ulWeightPerOperation));
767
768	rc = pProgress->init(mVirtualBox, static_cast<IAppliance*>(this),
769	bstrDescription,
770	TRUE /* aCancelable */,
771	cOperations, // ULONG cOperations,
772	ulTotalOperationsWeight, // ULONG ulTotalOperationsWeight,
773	bstrDescription, // CBSTR bstrFirstOperationDescription,
774	ulOVFCreationWeight); // ULONG ulFirstOperationWeight,
775	return rc;
776	}
777
778	void Appliance::parseURI(Utf8Str strUri, LocationInfo &locInfo) const
779	{
780	/* Check the URI for the protocol */
781	if (strUri.startsWith("file://", Utf8Str::CaseInsensitive)) /* File based */
782	{
783	locInfo.storageType = VFSType_File;
784	strUri = strUri.substr(sizeof("file://") - 1);
785	}
786	else if (strUri.startsWith("SunCloud://", Utf8Str::CaseInsensitive)) /* Sun Cloud service */
787	{
788	locInfo.storageType = VFSType_S3;
789	strUri = strUri.substr(sizeof("SunCloud://") - 1);
790	}
791	else if (strUri.startsWith("S3://", Utf8Str::CaseInsensitive)) /* S3 service */
792	{
793	locInfo.storageType = VFSType_S3;
794	strUri = strUri.substr(sizeof("S3://") - 1);
795	}
796	else if (strUri.startsWith("webdav://", Utf8Str::CaseInsensitive)) /* webdav service */
797	throw E_NOTIMPL;
798
799	/* Not necessary on a file based URI */
800	if (locInfo.storageType != VFSType_File)
801	{
802	size_t uppos = strUri.find("@"); /* username:password combo */
803	if (uppos != Utf8Str::npos)
804	{
805	locInfo.strUsername = strUri.substr(0, uppos);
806	strUri = strUri.substr(uppos + 1);
807	size_t upos = locInfo.strUsername.find(":");
808	if (upos != Utf8Str::npos)
809	{
810	locInfo.strPassword = locInfo.strUsername.substr(upos + 1);
811	locInfo.strUsername = locInfo.strUsername.substr(0, upos);
812	}
813	}
814	size_t hpos = strUri.find("/"); /* hostname part */
815	if (hpos != Utf8Str::npos)
816	{
817	locInfo.strHostname = strUri.substr(0, hpos);
818	strUri = strUri.substr(hpos);
819	}
820	}
821
822	locInfo.strPath = strUri;
823	}
824
825	void Appliance::parseBucket(Utf8Str &aPath, Utf8Str &aBucket) const
826	{
827	/* Buckets are S3 specific. So parse the bucket out of the file path */
828	if (!aPath.startsWith("/"))
829	throw setError(E_INVALIDARG,
830	tr("The path '%s' must start with /"), aPath.c_str());
831	size_t bpos = aPath.find("/", 1);
832	if (bpos != Utf8Str::npos)
833	{
834	aBucket = aPath.substr(1, bpos - 1); /* The bucket without any slashes */
835	aPath = aPath.substr(bpos); /* The rest of the file path */
836	}
837	/* If there is no bucket name provided reject it */
838	if (aBucket.isEmpty())
839	throw setError(E_INVALIDARG,
840	tr("You doesn't provide a bucket name in the URI '%s'"), aPath.c_str());
841	}
842
843	Utf8Str Appliance::manifestFileName(Utf8Str aPath) const
844	{
845	/* Get the name part */
846	char *pszMfName = RTStrDup(RTPathFilename(aPath.c_str()));
847	/* Strip any extensions */
848	RTPathStripExt(pszMfName);
849	/* Path without the filename */
850	aPath.stripFilename();
851	/* Format the manifest path */
852	Utf8StrFmt strMfFile("%s/%s.mf", aPath.c_str(), pszMfName);
853	RTStrFree(pszMfName);
854	return strMfFile;
855	}
856
857	/**
858	*
859	* @return
860	*/
861	int Appliance::TaskOVF::startThread()
862	{
863	int vrc = RTThreadCreate(NULL, Appliance::taskThreadImportOrExport, this,
864	0, RTTHREADTYPE_MAIN_HEAVY_WORKER, 0,
865	"Appliance::Task");
866
867	ComAssertMsgRCRet(vrc,
868	("Could not create OVF task thread (%Rrc)\n", vrc), E_FAIL);
869
870	return S_OK;
871	}
872
873	/**
874	* Thread function for the thread started in Appliance::readImpl() and Appliance::importImpl()
875	* and Appliance::writeImpl().
876	* This will in turn call Appliance::readFS() or Appliance::readS3() or Appliance::importFS()
877	* or Appliance::importS3() or Appliance::writeFS() or Appliance::writeS3().
878	*
879	* @param aThread
880	* @param pvUser
881	*/
882	/* static */
883	DECLCALLBACK(int) Appliance::taskThreadImportOrExport(RTTHREAD /* aThread /, void pvUser)
884	{
885	std::auto_ptr<TaskOVF> task(static_cast<TaskOVF*>(pvUser));
886	AssertReturn(task.get(), VERR_GENERAL_FAILURE);
887
888	Appliance *pAppliance = task->pAppliance;
889
890	LogFlowFuncEnter();
891	LogFlowFunc(("Appliance %p\n", pAppliance));
892
893	HRESULT taskrc = S_OK;
894
895	switch (task->taskType)
896	{
897	case TaskOVF::Read:
898	if (task->locInfo.storageType == VFSType_File)
899	taskrc = pAppliance->readFS(task->locInfo);
900	else if (task->locInfo.storageType == VFSType_S3)
901	taskrc = pAppliance->readS3(task.get());
902	break;
903
904	case TaskOVF::Import:
905	if (task->locInfo.storageType == VFSType_File)
906	taskrc = pAppliance->importFS(task->locInfo, task->pProgress);
907	else if (task->locInfo.storageType == VFSType_S3)
908	taskrc = pAppliance->importS3(task.get());
909	break;
910
911	case TaskOVF::Write:
912	if (task->locInfo.storageType == VFSType_File)
913	taskrc = pAppliance->writeFS(task->locInfo, task->enFormat, task->pProgress);
914	else if (task->locInfo.storageType == VFSType_S3)
915	taskrc = pAppliance->writeS3(task.get());
916	break;
917	}
918
919	task->rc = taskrc;
920
921	if (!task->pProgress.isNull())
922	task->pProgress->notifyComplete(taskrc);
923
924	LogFlowFuncLeave();
925
926	return VINF_SUCCESS;
927	}
928
929	/* static */
930	int Appliance::TaskOVF::updateProgress(unsigned uPercent, void *pvUser)
931	{
932	Appliance::TaskOVF* pTask = (Appliance::TaskOVF*)pvUser;
933
934	if ( pTask
935	&& !pTask->pProgress.isNull())
936	{
937	BOOL fCanceled;
938	pTask->pProgress->COMGETTER(Canceled)(&fCanceled);
939	if (fCanceled)
940	return -1;
941	pTask->pProgress->SetCurrentOperationProgress(uPercent);
942	}
943	return VINF_SUCCESS;
944	}
945
946	////////////////////////////////////////////////////////////////////////////////
947	//
948	// IVirtualSystemDescription constructor / destructor
949	//
950	////////////////////////////////////////////////////////////////////////////////
951
952	DEFINE_EMPTY_CTOR_DTOR(VirtualSystemDescription)
953
954	/**
955	* COM initializer.
956	* @return
957	*/
958	HRESULT VirtualSystemDescription::init()
959	{
960	/* Enclose the state transition NotReady->InInit->Ready */
961	AutoInitSpan autoInitSpan(this);
962	AssertReturn(autoInitSpan.isOk(), E_FAIL);
963
964	/* Initialize data */
965	m = new Data();
966
967	/* Confirm a successful initialization */
968	autoInitSpan.setSucceeded();
969	return S_OK;
970	}
971
972	/**
973	* COM uninitializer.
974	*/
975
976	void VirtualSystemDescription::uninit()
977	{
978	delete m;
979	m = NULL;
980	}
981
982	////////////////////////////////////////////////////////////////////////////////
983	//
984	// IVirtualSystemDescription public methods
985	//
986	////////////////////////////////////////////////////////////////////////////////
987
988	/**
989	* Public method implementation.
990	* @param
991	* @return
992	*/
993	STDMETHODIMP VirtualSystemDescription::COMGETTER(Count)(ULONG *aCount)
994	{
995	if (!aCount)
996	return E_POINTER;
997
998	AutoCaller autoCaller(this);
999	if (FAILED(autoCaller.rc())) return autoCaller.rc();
1000
1001	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1002
1003	*aCount = (ULONG)m->llDescriptions.size();
1004
1005	return S_OK;
1006	}
1007
1008	/**
1009	* Public method implementation.
1010	* @return
1011	*/
1012	STDMETHODIMP VirtualSystemDescription::GetDescription(ComSafeArrayOut(VirtualSystemDescriptionType_T, aTypes),
1013	ComSafeArrayOut(BSTR, aRefs),
1014	ComSafeArrayOut(BSTR, aOrigValues),
1015	ComSafeArrayOut(BSTR, aVboxValues),
1016	ComSafeArrayOut(BSTR, aExtraConfigValues))
1017	{
1018	if (ComSafeArrayOutIsNull(aTypes) \|\|
1019	ComSafeArrayOutIsNull(aRefs) \|\|
1020	ComSafeArrayOutIsNull(aOrigValues) \|\|
1021	ComSafeArrayOutIsNull(aVboxValues) \|\|
1022	ComSafeArrayOutIsNull(aExtraConfigValues))
1023	return E_POINTER;
1024
1025	AutoCaller autoCaller(this);
1026	if (FAILED(autoCaller.rc())) return autoCaller.rc();
1027
1028	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1029
1030	ULONG c = (ULONG)m->llDescriptions.size();
1031	com::SafeArray<VirtualSystemDescriptionType_T> sfaTypes(c);
1032	com::SafeArray<BSTR> sfaRefs(c);
1033	com::SafeArray<BSTR> sfaOrigValues(c);
1034	com::SafeArray<BSTR> sfaVboxValues(c);
1035	com::SafeArray<BSTR> sfaExtraConfigValues(c);
1036
1037	list<VirtualSystemDescriptionEntry>::const_iterator it;
1038	size_t i = 0;
1039	for (it = m->llDescriptions.begin();
1040	it != m->llDescriptions.end();
1041	++it, ++i)
1042	{
1043	const VirtualSystemDescriptionEntry &vsde = (*it);
1044
1045	sfaTypes[i] = vsde.type;
1046
1047	Bstr bstr = vsde.strRef;
1048	bstr.cloneTo(&sfaRefs[i]);
1049
1050	bstr = vsde.strOvf;
1051	bstr.cloneTo(&sfaOrigValues[i]);
1052
1053	bstr = vsde.strVbox;
1054	bstr.cloneTo(&sfaVboxValues[i]);
1055
1056	bstr = vsde.strExtraConfig;
1057	bstr.cloneTo(&sfaExtraConfigValues[i]);
1058	}
1059
1060	sfaTypes.detachTo(ComSafeArrayOutArg(aTypes));
1061	sfaRefs.detachTo(ComSafeArrayOutArg(aRefs));
1062	sfaOrigValues.detachTo(ComSafeArrayOutArg(aOrigValues));
1063	sfaVboxValues.detachTo(ComSafeArrayOutArg(aVboxValues));
1064	sfaExtraConfigValues.detachTo(ComSafeArrayOutArg(aExtraConfigValues));
1065
1066	return S_OK;
1067	}
1068
1069	/**
1070	* Public method implementation.
1071	* @return
1072	*/
1073	STDMETHODIMP VirtualSystemDescription::GetDescriptionByType(VirtualSystemDescriptionType_T aType,
1074	ComSafeArrayOut(VirtualSystemDescriptionType_T, aTypes),
1075	ComSafeArrayOut(BSTR, aRefs),
1076	ComSafeArrayOut(BSTR, aOrigValues),
1077	ComSafeArrayOut(BSTR, aVboxValues),
1078	ComSafeArrayOut(BSTR, aExtraConfigValues))
1079	{
1080	if (ComSafeArrayOutIsNull(aTypes) \|\|
1081	ComSafeArrayOutIsNull(aRefs) \|\|
1082	ComSafeArrayOutIsNull(aOrigValues) \|\|
1083	ComSafeArrayOutIsNull(aVboxValues) \|\|
1084	ComSafeArrayOutIsNull(aExtraConfigValues))
1085	return E_POINTER;
1086
1087	AutoCaller autoCaller(this);
1088	if (FAILED(autoCaller.rc())) return autoCaller.rc();
1089
1090	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1091
1092	std::list<VirtualSystemDescriptionEntry*> vsd = findByType (aType);
1093	ULONG c = (ULONG)vsd.size();
1094	com::SafeArray<VirtualSystemDescriptionType_T> sfaTypes(c);
1095	com::SafeArray<BSTR> sfaRefs(c);
1096	com::SafeArray<BSTR> sfaOrigValues(c);
1097	com::SafeArray<BSTR> sfaVboxValues(c);
1098	com::SafeArray<BSTR> sfaExtraConfigValues(c);
1099
1100	list<VirtualSystemDescriptionEntry*>::const_iterator it;
1101	size_t i = 0;
1102	for (it = vsd.begin();
1103	it != vsd.end();
1104	++it, ++i)
1105	{
1106	const VirtualSystemDescriptionEntry vsde = (it);
1107
1108	sfaTypes[i] = vsde->type;
1109
1110	Bstr bstr = vsde->strRef;
1111	bstr.cloneTo(&sfaRefs[i]);
1112
1113	bstr = vsde->strOvf;
1114	bstr.cloneTo(&sfaOrigValues[i]);
1115
1116	bstr = vsde->strVbox;
1117	bstr.cloneTo(&sfaVboxValues[i]);
1118
1119	bstr = vsde->strExtraConfig;
1120	bstr.cloneTo(&sfaExtraConfigValues[i]);
1121	}
1122
1123	sfaTypes.detachTo(ComSafeArrayOutArg(aTypes));
1124	sfaRefs.detachTo(ComSafeArrayOutArg(aRefs));
1125	sfaOrigValues.detachTo(ComSafeArrayOutArg(aOrigValues));
1126	sfaVboxValues.detachTo(ComSafeArrayOutArg(aVboxValues));
1127	sfaExtraConfigValues.detachTo(ComSafeArrayOutArg(aExtraConfigValues));
1128
1129	return S_OK;
1130	}
1131
1132	/**
1133	* Public method implementation.
1134	* @return
1135	*/
1136	STDMETHODIMP VirtualSystemDescription::GetValuesByType(VirtualSystemDescriptionType_T aType,
1137	VirtualSystemDescriptionValueType_T aWhich,
1138	ComSafeArrayOut(BSTR, aValues))
1139	{
1140	if (ComSafeArrayOutIsNull(aValues))
1141	return E_POINTER;
1142
1143	AutoCaller autoCaller(this);
1144	if (FAILED(autoCaller.rc())) return autoCaller.rc();
1145
1146	AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
1147
1148	std::list<VirtualSystemDescriptionEntry*> vsd = findByType (aType);
1149	com::SafeArray<BSTR> sfaValues((ULONG)vsd.size());
1150
1151	list<VirtualSystemDescriptionEntry*>::const_iterator it;
1152	size_t i = 0;
1153	for (it = vsd.begin();
1154	it != vsd.end();
1155	++it, ++i)
1156	{
1157	const VirtualSystemDescriptionEntry vsde = (it);
1158
1159	Bstr bstr;
1160	switch (aWhich)
1161	{
1162	case VirtualSystemDescriptionValueType_Reference: bstr = vsde->strRef; break;
1163	case VirtualSystemDescriptionValueType_Original: bstr = vsde->strOvf; break;
1164	case VirtualSystemDescriptionValueType_Auto: bstr = vsde->strVbox; break;
1165	case VirtualSystemDescriptionValueType_ExtraConfig: bstr = vsde->strExtraConfig; break;
1166	}
1167
1168	bstr.cloneTo(&sfaValues[i]);
1169	}
1170
1171	sfaValues.detachTo(ComSafeArrayOutArg(aValues));
1172
1173	return S_OK;
1174	}
1175
1176	/**
1177	* Public method implementation.
1178	* @return
1179	*/
1180	STDMETHODIMP VirtualSystemDescription::SetFinalValues(ComSafeArrayIn(BOOL, aEnabled),
1181	ComSafeArrayIn(IN_BSTR, argVboxValues),
1182	ComSafeArrayIn(IN_BSTR, argExtraConfigValues))
1183	{
1184	#ifndef RT_OS_WINDOWS
1185	NOREF(aEnabledSize);
1186	#endif /* RT_OS_WINDOWS */
1187
1188	CheckComArgSafeArrayNotNull(aEnabled);
1189	CheckComArgSafeArrayNotNull(argVboxValues);
1190	CheckComArgSafeArrayNotNull(argExtraConfigValues);
1191
1192	AutoCaller autoCaller(this);
1193	if (FAILED(autoCaller.rc())) return autoCaller.rc();
1194
1195	AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1196
1197	com::SafeArray<BOOL> sfaEnabled(ComSafeArrayInArg(aEnabled));
1198	com::SafeArray<IN_BSTR> sfaVboxValues(ComSafeArrayInArg(argVboxValues));
1199	com::SafeArray<IN_BSTR> sfaExtraConfigValues(ComSafeArrayInArg(argExtraConfigValues));
1200
1201	if ( (sfaEnabled.size() != m->llDescriptions.size())
1202	\|\| (sfaVboxValues.size() != m->llDescriptions.size())
1203	\|\| (sfaExtraConfigValues.size() != m->llDescriptions.size())
1204	)
1205	return E_INVALIDARG;
1206
1207	list<VirtualSystemDescriptionEntry>::iterator it;
1208	size_t i = 0;
1209	for (it = m->llDescriptions.begin();
1210	it != m->llDescriptions.end();
1211	++it, ++i)
1212	{
1213	VirtualSystemDescriptionEntry& vsde = *it;
1214
1215	if (sfaEnabled[i])
1216	{
1217	vsde.strVbox = sfaVboxValues[i];
1218	vsde.strExtraConfig = sfaExtraConfigValues[i];
1219	}
1220	else
1221	vsde.type = VirtualSystemDescriptionType_Ignore;
1222	}
1223
1224	return S_OK;
1225	}
1226
1227	/**
1228	* Public method implementation.
1229	* @return
1230	*/
1231	STDMETHODIMP VirtualSystemDescription::AddDescription(VirtualSystemDescriptionType_T aType,
1232	IN_BSTR aVboxValue,
1233	IN_BSTR aExtraConfigValue)
1234	{
1235	AutoCaller autoCaller(this);
1236	if (FAILED(autoCaller.rc())) return autoCaller.rc();
1237
1238	AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
1239
1240	addEntry(aType, "", aVboxValue, aVboxValue, 0, aExtraConfigValue);
1241
1242	return S_OK;
1243	}
1244
1245	/**
1246	* Internal method; adds a new description item to the member list.
1247	* @param aType Type of description for the new item.
1248	* @param strRef Reference item; only used with hard disk controllers.
1249	* @param aOrigValue Corresponding original value from OVF.
1250	* @param aAutoValue Initial configuration value (can be overridden by caller with setFinalValues).
1251	* @param ulSizeMB Weight for IProgress
1252	* @param strExtraConfig Extra configuration; meaning dependent on type.
1253	*/
1254	void VirtualSystemDescription::addEntry(VirtualSystemDescriptionType_T aType,
1255	const Utf8Str &strRef,
1256	const Utf8Str &aOrigValue,
1257	const Utf8Str &aAutoValue,
1258	uint32_t ulSizeMB,
1259	const Utf8Str &strExtraConfig /= ""/)
1260	{
1261	VirtualSystemDescriptionEntry vsde;
1262	vsde.ulIndex = (uint32_t)m->llDescriptions.size(); // each entry gets an index so the client side can reference them
1263	vsde.type = aType;
1264	vsde.strRef = strRef;
1265	vsde.strOvf = aOrigValue;
1266	vsde.strVbox = aAutoValue;
1267	vsde.strExtraConfig = strExtraConfig;
1268	vsde.ulSizeMB = ulSizeMB;
1269
1270	m->llDescriptions.push_back(vsde);
1271	}
1272
1273	/**
1274	* Private method; returns a list of description items containing all the items from the member
1275	* description items of this virtual system that match the given type.
1276	* @param aType
1277	* @return
1278	*/
1279	std::list<VirtualSystemDescriptionEntry*> VirtualSystemDescription::findByType(VirtualSystemDescriptionType_T aType)
1280	{
1281	std::list<VirtualSystemDescriptionEntry*> vsd;
1282
1283	list<VirtualSystemDescriptionEntry>::iterator it;
1284	for (it = m->llDescriptions.begin();
1285	it != m->llDescriptions.end();
1286	++it)
1287	{
1288	if (it->type == aType)
1289	vsd.push_back(&(*it));
1290	}
1291
1292	return vsd;
1293	}
1294
1295	/**
1296	* Private method; looks thru the member hardware items for the IDE, SATA, or SCSI controller with
1297	* the given reference ID. Useful when needing the controller for a particular
1298	* virtual disk.
1299	* @param id
1300	* @return
1301	*/
1302	const VirtualSystemDescriptionEntry* VirtualSystemDescription::findControllerFromID(uint32_t id)
1303	{
1304	Utf8Str strRef = Utf8StrFmt("%RI32", id);
1305	list<VirtualSystemDescriptionEntry>::const_iterator it;
1306	for (it = m->llDescriptions.begin();
1307	it != m->llDescriptions.end();
1308	++it)
1309	{
1310	const VirtualSystemDescriptionEntry &d = *it;
1311	switch (d.type)
1312	{
1313	case VirtualSystemDescriptionType_HardDiskControllerIDE:
1314	case VirtualSystemDescriptionType_HardDiskControllerSATA:
1315	case VirtualSystemDescriptionType_HardDiskControllerSCSI:
1316	if (d.strRef == strRef)
1317	return &d;
1318	break;
1319	}
1320	}
1321
1322	return NULL;
1323	}
1324

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

source: vbox/trunk/src/VBox/Main/ApplianceImpl.cpp@ 27908

Download in other formats: