VirtualBox

source: vbox/trunk/src/VBox/Main/testcase/tstOVF.cpp@ 30881

Last change on this file since 30881 was 30881, checked in by vboxsync, 14 years ago

Main/OVF: add Appliance::machines[] attribute so that caller can figure out which machines were actually created; improve OVF testcase (not done yet)
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 13.3 KB
Line 
1/* $Id: tstOVF.cpp 30881 2010-07-16 14:34:31Z vboxsync $ */
2/** @file
3 *
4 * tstOVF - testcases for OVF import and export
5 */
6
7/*
8 * Copyright (C) 2010 Oracle Corporation
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
19#include <VBox/com/VirtualBox.h>
20
21#include <VBox/com/com.h>
22#include <VBox/com/array.h>
23#include <VBox/com/string.h>
24#include <VBox/com/ErrorInfo.h>
25#include <VBox/com/errorprint.h>
26#include <VBox/com/EventQueue.h>
27
28#include <iprt/initterm.h>
29#include <iprt/stream.h>
30#include <iprt/file.h>
31#include <iprt/path.h>
32#include <iprt/param.h>
33
34#include <list>
35
36using namespace com;
37
38// main
39///////////////////////////////////////////////////////////////////////////////
40
41/**
42 * Quick hack exception structure.
43 *
44 */
45struct MyError
46{
47 MyError(HRESULT rc,
48 const char *pcsz,
49 IProgress *pProgress = NULL)
50 : m_rc(rc)
51 {
52 m_str = "ERROR: ";
53 m_str += pcsz;
54
55 if (pProgress)
56 {
57 com::ProgressErrorInfo info(pProgress);
58 com::GluePrintErrorInfo(info);
59 }
60 else if (rc)
61 {
62 com::ErrorInfo info;
63 if (!info.isFullAvailable() && !info.isBasicAvailable())
64 com::GluePrintRCMessage(rc);
65 else
66 com::GluePrintErrorInfo(info);
67 }
68 }
69
70 Utf8Str m_str;
71 HRESULT m_rc;
72};
73
74/**
75 * Imports the given OVF file, with all bells and whistles.
76 * Throws MyError on errors.
77 * @param pcszPrefix Descriptive short prefix string for console output.
78 * @param pVirtualBox VirtualBox instance.
79 * @param pcszOVF0 File to import.
80 * @param llMachinesCreated out: UUIDs of machines that were created so that caller can clean up.
81 */
82void importOVF(const char *pcszPrefix,
83 ComPtr<IVirtualBox> &pVirtualBox,
84 const char *pcszOVF0,
85 std::list<Guid> &llMachinesCreated)
86{
87 char szAbsOVF[RTPATH_MAX];
88 RTPathAbs(pcszOVF0, szAbsOVF, sizeof(szAbsOVF));
89
90 RTPrintf("%s: reading appliance \"%s\"...\n", pcszPrefix, szAbsOVF);
91 ComPtr<IAppliance> pAppl;
92 HRESULT rc = pVirtualBox->CreateAppliance(pAppl.asOutParam());
93 if (FAILED(rc)) throw MyError(rc, "failed to create appliance\n");
94
95 ComPtr<IProgress> pProgress;
96 rc = pAppl->Read(Bstr(szAbsOVF), pProgress.asOutParam());
97 if (FAILED(rc)) throw MyError(rc, "Appliance::Read() failed\n");
98 rc = pProgress->WaitForCompletion(-1);
99 if (FAILED(rc)) throw MyError(rc, "Progress::WaitForCompletion() failed\n");
100 LONG rc2;
101 pProgress->COMGETTER(ResultCode)(&rc2);
102 if (FAILED(rc2)) throw MyError(rc2, "Appliance::Read() failed\n", pProgress);
103
104 RTPrintf("%s: interpreting appliance \"%s\"...\n", pcszPrefix, szAbsOVF);
105 rc = pAppl->Interpret();
106 if (FAILED(rc)) throw MyError(rc, "Appliance::Interpret() failed\n");
107
108 com::SafeIfaceArray<IVirtualSystemDescription> aDescriptions;
109 rc = pAppl->COMGETTER(VirtualSystemDescriptions)(ComSafeArrayAsOutParam(aDescriptions));
110 for (uint32_t u = 0;
111 u < aDescriptions.size();
112 ++u)
113 {
114 ComPtr<IVirtualSystemDescription> pVSys = aDescriptions[u];
115
116 com::SafeArray<VirtualSystemDescriptionType_T> aTypes;
117 com::SafeArray<BSTR> aRefs;
118 com::SafeArray<BSTR> aOvfValues;
119 com::SafeArray<BSTR> aVboxValues;
120 com::SafeArray<BSTR> aExtraConfigValues;
121 rc = pVSys->GetDescription(ComSafeArrayAsOutParam(aTypes),
122 ComSafeArrayAsOutParam(aRefs),
123 ComSafeArrayAsOutParam(aOvfValues),
124 ComSafeArrayAsOutParam(aVboxValues),
125 ComSafeArrayAsOutParam(aExtraConfigValues));
126 if (FAILED(rc)) throw MyError(rc, "VirtualSystemDescription::GetDescription() failed\n");
127
128 for (uint32_t u2 = 0;
129 u2 < aTypes.size();
130 ++u2)
131 {
132 const char *pcszType;
133
134 VirtualSystemDescriptionType_T t = aTypes[u2];
135 switch (t)
136 {
137 case VirtualSystemDescriptionType_OS:
138 pcszType = "ostype";
139 break;
140
141 case VirtualSystemDescriptionType_Name:
142 pcszType = "name";
143 break;
144
145 case VirtualSystemDescriptionType_Product:
146 pcszType = "product";
147 break;
148
149 case VirtualSystemDescriptionType_ProductUrl:
150 pcszType = "producturl";
151 break;
152
153 case VirtualSystemDescriptionType_Vendor:
154 pcszType = "vendor";
155 break;
156
157 case VirtualSystemDescriptionType_VendorUrl:
158 pcszType = "vendorurl";
159 break;
160
161 case VirtualSystemDescriptionType_Version:
162 pcszType = "version";
163 break;
164
165 case VirtualSystemDescriptionType_Description:
166 pcszType = "description";
167 break;
168
169 case VirtualSystemDescriptionType_License:
170 pcszType = "license";
171 break;
172
173 case VirtualSystemDescriptionType_CPU:
174 pcszType = "cpu";
175 break;
176
177 case VirtualSystemDescriptionType_Memory:
178 pcszType = "memory";
179 break;
180
181 case VirtualSystemDescriptionType_HardDiskControllerIDE:
182 pcszType = "ide";
183 break;
184
185 case VirtualSystemDescriptionType_HardDiskControllerSATA:
186 pcszType = "sata";
187 break;
188
189 case VirtualSystemDescriptionType_HardDiskControllerSAS:
190 pcszType = "sas";
191 break;
192
193 case VirtualSystemDescriptionType_HardDiskControllerSCSI:
194 pcszType = "scsi";
195 break;
196
197 case VirtualSystemDescriptionType_HardDiskImage:
198 pcszType = "hd";
199 break;
200
201 case VirtualSystemDescriptionType_CDROM:
202 pcszType = "cdrom";
203 break;
204
205 case VirtualSystemDescriptionType_Floppy:
206 pcszType = "floppy";
207 break;
208
209 case VirtualSystemDescriptionType_NetworkAdapter:
210 pcszType = "net";
211 break;
212
213 case VirtualSystemDescriptionType_USBController:
214 pcszType = "usb";
215 break;
216
217 case VirtualSystemDescriptionType_SoundCard:
218 pcszType = "sound";
219 break;
220
221 default:
222 throw MyError(E_UNEXPECTED, "Invalid VirtualSystemDescriptionType\n");
223 break;
224 }
225
226 RTPrintf(" vsys %2u item %2u: type %2d (%s), ovf: \"%ls\", vbox: \"%ls\", extra: \"%ls\"\n",
227 u, u2, t, pcszType,
228 aOvfValues[u2],
229 aVboxValues[u2],
230 aExtraConfigValues[u2]);
231 }
232 }
233
234 RTPrintf("%s: importing %d machine(s)...\n", pcszPrefix, aDescriptions.size());
235 rc = pAppl->ImportMachines(pProgress.asOutParam());
236 if (FAILED(rc)) throw MyError(rc, "Appliance::ImportMachines() failed\n");
237 rc = pProgress->WaitForCompletion(-1);
238 if (FAILED(rc)) throw MyError(rc, "Progress::WaitForCompletion() failed\n");
239 pProgress->COMGETTER(ResultCode)(&rc2);
240 if (FAILED(rc2)) throw MyError(rc2, "Progress::GetResultCode() failed\n");
241
242 com::SafeArray<BSTR> aMachineUUIDs;
243 rc = pAppl->COMGETTER(Machines)(ComSafeArrayAsOutParam(aMachineUUIDs));
244 if (FAILED(rc)) throw MyError(rc, "Appliance::GetMachines() failed\n");
245
246 for (size_t u = 0;
247 u < aMachineUUIDs.size();
248 ++u)
249 {
250 RTPrintf("%s: created machine %u: %ls\n", pcszPrefix, u, aMachineUUIDs[u]);
251 llMachinesCreated.push_back(Guid(Bstr(aMachineUUIDs[u])));
252 }
253
254 RTPrintf("%s: success!\n", pcszPrefix);
255}
256
257/**
258 * Copies ovf-testcases/ovf-dummy.vmdk to the given target and appends that
259 * target as a string to the given list so that the caller can delete it
260 * again later.
261 * @param llFiles2Delete List of strings to append the target file path to.
262 * @param pcszDest Target for dummy VMDK.
263 */
264void copyDummyDiskImage(std::list<Utf8Str> &llFiles2Delete, const char *pcszDest)
265{
266 int vrc = RTFileCopy("ovf-testcases/ovf-dummy.vmdk", pcszDest);
267 if (RT_FAILURE(vrc)) throw MyError(0, Utf8StrFmt("Cannot copy ovf-dummy.vmdk to %s: %Rra\n", pcszDest, vrc).c_str());
268 llFiles2Delete.push_back(pcszDest);
269}
270
271/**
272 *
273 * @param argc
274 * @param argv[]
275 * @return
276 */
277int main(int argc, char *argv[])
278{
279 RTR3Init();
280
281 HRESULT rc = S_OK;
282
283 std::list<Utf8Str> llFiles2Delete;
284 std::list<Guid> llMachinesCreated;
285
286 ComPtr<IVirtualBox> pVirtualBox;
287
288 try
289 {
290 RTPrintf("Initializing COM...\n");
291 rc = com::Initialize();
292 if (FAILED(rc)) throw MyError(rc, "failed to initialize COM!\n");
293
294 ComPtr<ISession> pSession;
295
296 RTPrintf("Creating VirtualBox object...\n");
297 rc = pVirtualBox.createLocalObject(CLSID_VirtualBox);
298 if (FAILED(rc)) throw MyError(rc, "failed to create the VirtualBox object!\n");
299
300 rc = pSession.createInprocObject(CLSID_Session);
301 if (FAILED(rc)) throw MyError(rc, "failed to create a session object!\n");
302
303 // create the event queue
304 // (here it is necessary only to process remaining XPCOM/IPC events after the session is closed)
305 EventQueue eventQ;
306
307 // testcase 1: import ovf-joomla-0.9/joomla-1.1.4-ovf.ovf
308 copyDummyDiskImage(llFiles2Delete, "ovf-testcases/ovf-joomla-0.9/joomla-1.1.4-ovf-0.vmdk");
309 copyDummyDiskImage(llFiles2Delete, "ovf-testcases/ovf-joomla-0.9/joomla-1.1.4-ovf-1.vmdk");
310 importOVF("joomla-0.9", pVirtualBox, "ovf-testcases/ovf-joomla-0.9/joomla-1.1.4-ovf.ovf", llMachinesCreated);
311
312 // testcase 2: import ovf-winxp-vbox-sharedfolders/winxp.ovf
313 copyDummyDiskImage(llFiles2Delete, "ovf-testcases/ovf-winxp-vbox-sharedfolders/Windows 5.1 XP 1 merged.vmdk");
314 copyDummyDiskImage(llFiles2Delete, "ovf-testcases/ovf-winxp-vbox-sharedfolders/smallvdi.vmdk");
315 importOVF("winxp-vbox-sharedfolders", pVirtualBox, "ovf-testcases/ovf-winxp-vbox-sharedfolders/winxp.ovf", llMachinesCreated);
316
317 RTPrintf("Machine imports done, no errors. Cleaning up...\n");
318 }
319 catch (MyError &e)
320 {
321 rc = e.m_rc;
322 RTPrintf("%s", e.m_str.c_str());
323 }
324
325 try
326 {
327 // clean up the machines created
328 for (std::list<Guid>::const_iterator it = llMachinesCreated.begin();
329 it != llMachinesCreated.end();
330 ++it)
331 {
332// const Guid &uuid = *it;
333// Bstr bstrUUID(uuid.toUtf16());
334// ComPtr<IMachine> pMachine;
335// rc = pVirtualBox->GetMachine(bstrUUID, pMachine.asOutParam());
336// if (FAILED(rc)) throw MyError(rc, "VirtualBox::FindMachine() failed\n");
337//
338// SafeIfaceArray<IMediumAttachment> aMediumAttachments;
339// rc = pMachine->COMGETTER(MediumAttachments)(ComSafeArrayAsOutParam(aMediumAttachments));
340// if (FAILED(rc)) throw MyError(rc, "Machine::MediumAttachments() failed\n");
341//
342// for (size_t u2 = 0;
343// u2 < aMediumAttachments.size();
344// ++u2)
345// {
346// ComPtr<IMediumAttachment> pMediumAttachment(aMediumAttachments[u2]);
347// ComPtr<IMedium> pMedium;
348// rc = pMediumAttachment->COMGETTER(Medium)(pMedium.asOutParam());
349// if (FAILED(rc)) throw MyError(rc, "MediumAttachment::GetMedium() failed\n");
350//
351// if (!pMedium.isNull())
352// {
353// Bstr bstrLocation;
354// rc = pMedium->COMGETTER(Location)(bstrLocation.asOutParam());
355// if (FAILED(rc)) throw MyError(rc, "Medium::GetLocation() failed\n");
356//
357// llFiles2Delete.push_back(bstrLocation);
358// }
359// }
360
361 RTPrintf(" Deleting machine %ls...\n", bstrUUID.raw());
362 pVirtualBox->UnregisterMachine(bstrUUID, pMachine.asOutParam());
363 if (FAILED(rc)) throw MyError(rc, "VirtualBox::UnregisterMachine() failed\n");
364
365 if (!pMachine.isNull())
366 {
367 rc = pMachine->DeleteSettings();
368 if (FAILED(rc)) throw MyError(rc, "Machine::DeleteSettings() failed\n");
369 }
370 }
371 }
372 catch (MyError &e)
373 {
374 rc = e.m_rc;
375 RTPrintf("%s", e.m_str.c_str());
376 }
377
378 // clean up the VMDK copies that we made in copyDummyDiskImage()
379 for (std::list<Utf8Str>::const_iterator it = llFiles2Delete.begin();
380 it != llFiles2Delete.end();
381 ++it)
382 {
383 const Utf8Str &strFile = *it;
384 RTFileDelete(strFile.c_str());
385 }
386
387 pVirtualBox.setNull();
388
389 RTPrintf("Shutting down COM...\n");
390 com::Shutdown();
391 RTPrintf ("tstOVF all done!\n");
392
393 return rc;
394}
395
Note: See TracBrowser for help on using the repository browser.

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