tdTeleportLocal1.py

Last change on this file was 106061, checked in by vboxsync, 3 months ago
Copyright year updates by scm.
Property svn:eol-style set to `LF` Property svn:executable set to ``* Property svn:keywords set to `Author Date Id Revision`
File size: 42.5 KB

Line
1	#!/usr/bin/env python
2	# -- coding: utf-8 --
3	# $Id: tdTeleportLocal1.py 106061 2024-09-16 14:03:52Z vboxsync $
4
5	"""
6	VirtualBox Validation Kit - Local teleportation testdriver.
7	"""
8
9	__copyright__ = \
10	"""
11	Copyright (C) 2010-2024 Oracle and/or its affiliates.
12
13	This file is part of VirtualBox base platform packages, as
14	available from https://www.virtualbox.org.
15
16	This program is free software; you can redistribute it and/or
17	modify it under the terms of the GNU General Public License
18	as published by the Free Software Foundation, in version 3 of the
19	License.
20
21	This program is distributed in the hope that it will be useful, but
22	WITHOUT ANY WARRANTY; without even the implied warranty of
23	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
24	General Public License for more details.
25
26	You should have received a copy of the GNU General Public License
27	along with this program; if not, see <https://www.gnu.org/licenses>.
28
29	The contents of this file may alternatively be used under the terms
30	of the Common Development and Distribution License Version 1.0
31	(CDDL), a copy of it is provided in the "COPYING.CDDL" file included
32	in the VirtualBox distribution, in which case the provisions of the
33	CDDL are applicable instead of those of the GPL.
34
35	You may elect to license modified versions of this file under the
36	terms and conditions of either the GPL or the CDDL or both.
37
38	SPDX-License-Identifier: GPL-3.0-only OR CDDL-1.0
39	"""
40	__version__ = "$Revision: 106061 $"
41
42
43	# Standard Python imports.
44	import os;
45	import sys;
46
47	# Only the main script needs to modify the path.
48	try: __file__ # pylint: disable=used-before-assignment
49	except: __file__ = sys.argv[0];
50	g_ksValidationKitDir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))));
51	sys.path.append(g_ksValidationKitDir);
52
53	# Validation Kit imports.
54	from testdriver import reporter;
55	from testdriver import base;
56	from testdriver import vbox;
57
58
59	class tdTeleportLocal1(vbox.TestDriver):
60	"""
61	Local Teleportation Test #1.
62	"""
63
64	def __init__(self):
65	vbox.TestDriver.__init__(self);
66	self.asRsrcs = None;
67
68	self.asTestsDef = ['test1', 'test2'];
69	self.asTests = ['test1', 'test2'];
70	self.asSkipTests = [];
71	self.asTestVMsDef = ['tst-rhel5', 'tst-win2k3ent', 'tst-sol10'];
72	self.asTestVMs = self.asTestVMsDef;
73	self.asSkipVMs = [];
74	self.asVirtModesDef = ['hwvirt', 'hwvirt-np', 'raw',]
75	self.asVirtModes = self.asVirtModesDef
76	self.acCpusDef = [1, 2,]
77	self.acCpus = self.acCpusDef;
78
79	#
80	# Overridden methods.
81	#
82	def showUsage(self):
83	rc = vbox.TestDriver.showUsage(self);
84	reporter.log('');
85	reporter.log('tdTeleportLocal1 Options:');
86	reporter.log(' --virt-modes <m1[:m2[:]]');
87	reporter.log(' Default: %s' % (':'.join(self.asVirtModesDef)));
88	reporter.log(' --cpu-counts <c1[:c2[:]]');
89	reporter.log(' Default: %s' % (':'.join(str(c) for c in self.acCpusDef)));
90	reporter.log(' --test-vms <vm1[:vm2[:...]]>');
91	reporter.log(' Test the specified VMs in the given order. Use this to change');
92	reporter.log(' the execution order or limit the choice of VMs');
93	reporter.log(' Default: %s (all)' % (':'.join(self.asTestVMsDef)));
94	reporter.log(' --skip-vms <vm1[:vm2[:...]]>');
95	reporter.log(' Skip the specified VMs when testing.');
96	reporter.log(' --tests <test1[:test2[:...]]>');
97	reporter.log(' Run the specified tests.');
98	reporter.log(' Default: %s (all)' % (':'.join(self.asTestsDef)));
99	reporter.log(' --skip-tests <test1[:test2[:...]]>');
100	reporter.log(' Skip the specified VMs when testing.');
101	reporter.log(' --quick');
102	reporter.log(' Shorthand for: --virt-modes hwvirt --cpu-counts 1');
103	reporter.log(' --test-vms tst-rhel5:tst-win2k3ent:tst-sol10');
104	return rc;
105
106	def parseOption(self, asArgs, iArg):
107	if asArgs[iArg] == '--virt-modes':
108	iArg += 1;
109	if iArg >= len(asArgs): raise base.InvalidOption('The "--virt-modes" takes a colon separated list of modes');
110	self.asVirtModes = asArgs[iArg].split(':');
111	for s in self.asVirtModes:
112	if s not in self.asVirtModesDef:
113	raise base.InvalidOption('The "--virt-modes" value "%s" is not valid; valid values are: %s' \
114	% (s, ' '.join(self.asVirtModesDef)));
115	elif asArgs[iArg] == '--cpu-counts':
116	iArg += 1;
117	if iArg >= len(asArgs): raise base.InvalidOption('The "--cpu-counts" takes a colon separated list of cpu counts');
118	self.acCpus = [];
119	for s in asArgs[iArg].split(':'):
120	try: c = int(s);
121	except: raise base.InvalidOption('The "--cpu-counts" value "%s" is not an integer' % (s,));
122	if c <= 0: raise base.InvalidOption('The "--cpu-counts" value "%s" is zero or negative' % (s,));
123	self.acCpus.append(c);
124	elif asArgs[iArg] == '--test-vms':
125	iArg += 1;
126	if iArg >= len(asArgs): raise base.InvalidOption('The "--test-vms" takes colon separated list');
127	if asArgs[iArg]:
128	self.asTestVMs = asArgs[iArg].split(':');
129	for s in self.asTestVMs:
130	if s not in self.asTestVMsDef:
131	raise base.InvalidOption('The "--test-vms" value "%s" is not valid; valid values are: %s' \
132	% (s, ' '.join(self.asTestVMsDef)));
133	else:
134	self.asTestVMs = [];
135	elif asArgs[iArg] == '--skip-vms':
136	iArg += 1;
137	if iArg >= len(asArgs): raise base.InvalidOption('The "--skip-vms" takes colon separated list');
138	self.asSkipVMs = asArgs[iArg].split(':');
139	for s in self.asSkipVMs:
140	if s not in self.asTestVMsDef:
141	reporter.log('warning: The "--skip-vms" value "%s" does not specify any of our test VMs.' % (s));
142	elif asArgs[iArg] == '--tests':
143	iArg += 1;
144	if iArg >= len(asArgs): raise base.InvalidOption('The "--tests" takes colon separated list');
145	self.asTests = asArgs[iArg].split(':');
146	for s in self.asTests:
147	if s not in self.asTestsDef:
148	reporter.log('warning: The "--tests" value "%s" does not specify any of our tests.' % (s));
149	elif asArgs[iArg] == '--skip-tests':
150	iArg += 1;
151	if iArg >= len(asArgs): raise base.InvalidOption('The "--skip-tests" takes colon separated list');
152	self.asSkipVMs = asArgs[iArg].split(':');
153	for s in self.asSkipTests:
154	if s not in self.asTestsDef:
155	reporter.log('warning: The "--skip-tests" value "%s" does not specify any of our tests.' % (s));
156	elif asArgs[iArg] == '--quick':
157	self.asVirtModes = ['hwvirt',];
158	self.acCpus = [1,];
159	#self.asTestVMs = ['tst-rhel5', 'tst-win2k3ent', 'tst-sol10',];
160	self.asTestVMs = ['tst-rhel5', ];
161	else:
162	return vbox.TestDriver.parseOption(self, asArgs, iArg);
163	return iArg + 1;
164
165	def completeOptions(self):
166	# Remove skipped VMs from the test VM list.
167	for sVM in self.asSkipVMs:
168	try: self.asTestVMs.remove(sVM);
169	except: pass;
170
171	# Remove skipped tests from the test list.
172	for sTest in self.asSkipTests:
173	try: self.asTests.remove(sTest);
174	except: pass;
175
176	# If no test2, then no test VMs.
177	if 'test2' not in self.asTests:
178	self.asTestVMs = [];
179
180	return vbox.TestDriver.completeOptions(self);
181
182	def getResourceSet(self):
183	# Construct the resource list the first time it's queried.
184	if self.asRsrcs is None:
185	self.asRsrcs = [];
186	if 'tst-rhel5' in self.asTestVMs:
187	self.asRsrcs.append('3.0/tcp/rhel5.vdi');
188	if 'tst-rhel5-64' in self.asTestVMs:
189	self.asRsrcs.append('3.0/tcp/rhel5-64.vdi');
190	if 'tst-sles11' in self.asTestVMs:
191	self.asRsrcs.append('3.0/tcp/sles11.vdi');
192	if 'tst-sles11-64' in self.asTestVMs:
193	self.asRsrcs.append('3.0/tcp/sles11-64.vdi');
194	if 'tst-oel' in self.asTestVMs:
195	self.asRsrcs.append('3.0/tcp/oel.vdi');
196	if 'tst-oel-64' in self.asTestVMs:
197	self.asRsrcs.append('3.0/tcp/oel-64.vdi');
198	if 'tst-win2k3ent' in self.asTestVMs:
199	self.asRsrcs.append('3.0/tcp/win2k3ent-acpi.vdi');
200	if 'tst-win2k3ent-64' in self.asTestVMs:
201	self.asRsrcs.append('3.0/tcp/win2k3ent-64.vdi');
202	if 'tst-win2k8' in self.asTestVMs:
203	self.asRsrcs.append('3.0/tcp/win2k8.vdi');
204	if 'tst-sol10' in self.asTestVMs:
205	self.asRsrcs.append('3.0/tcp/solaris10.vdi');
206	if 'tst-sol11' in self.asTestVMs:
207	self.asRsrcs.append('3.0/tcp/solaris11.vdi');
208	return self.asRsrcs;
209
210	def actionConfig(self):
211	## @todo actionConfig() and getResourceSet() are working on the same
212	# set of VMs as tdNetBenchmark1, creating common base class would be
213	# a good idea.
214
215	# Some stupid trickery to guess the location of the iso. ## fixme - testsuite unzip ++
216	sVBoxValidationKit_iso = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../VBoxValidationKit.iso'));
217	if not os.path.isfile(sVBoxValidationKit_iso):
218	sVBoxValidationKit_iso = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../VBoxTestSuite.iso'));
219	if not os.path.isfile(sVBoxValidationKit_iso):
220	sVBoxValidationKit_iso = '/mnt/ramdisk/vbox/svn/trunk/validationkit/VBoxValidationKit.iso';
221	if not os.path.isfile(sVBoxValidationKit_iso):
222	sVBoxValidationKit_iso = '/mnt/ramdisk/vbox/svn/trunk/testsuite/VBoxTestSuite.iso';
223	if not os.path.isfile(sVBoxValidationKit_iso):
224	sCur = os.getcwd();
225	for i in range(0, 10):
226	sVBoxValidationKit_iso = os.path.join(sCur, 'validationkit/VBoxValidationKit.iso');
227	if os.path.isfile(sVBoxValidationKit_iso):
228	break;
229	sVBoxValidationKit_iso = os.path.join(sCur, 'testsuite/VBoxTestSuite.iso');
230	if os.path.isfile(sVBoxValidationKit_iso):
231	break;
232	sCur = os.path.abspath(os.path.join(sCur, '..'));
233	if i is not None: pass; # shut up pychecker/pylint.
234	if not os.path.isfile(sVBoxValidationKit_iso):
235	sVBoxValidationKit_iso = '/home/bird/validationkit/VBoxValidationKit.iso';
236	if not os.path.isfile(sVBoxValidationKit_iso):
237	sVBoxValidationKit_iso = '/home/bird/testsuite/VBoxTestSuite.iso';
238
239	# Make sure vboxapi has been imported so we can use the constants.
240	if not self.importVBoxApi():
241	return False;
242
243	#
244	# Configure the empty VMs we're going to use for the first tests.
245	#
246
247	if 'test1' in self.asTests:
248	#oVM = self.createTestVMs('tst-empty-hwvirt', 0, sKind = 'Other', fVirtEx = True);
249	#if oVM is None:
250	# return False;
251
252	oVM = self.createTestVMs('tst-empty-raw', 2, sKind = 'Other', fVirtEx = False);
253	if oVM is None:
254	return False;
255
256	#
257	# Configure the VMs we're going to use for the last test.
258	#
259
260	# Linux VMs
261	if 'tst-rhel5' in self.asTestVMs:
262	oVM = self.createTestVMs('tst-rhel5', 1, '3.0/tcp/rhel5.vdi', sKind = 'RedHat', fIoApic = True, \
263	sDvdImage = sVBoxValidationKit_iso);
264	if oVM is None:
265	return False;
266
267	if 'tst-rhel5-64' in self.asTestVMs:
268	oVM = self.createTestVMs('tst-rhel5-64', 1, '3.0/tcp/rhel5-64.vdi', sKind = 'RedHat_64', \
269	sDvdImage = sVBoxValidationKit_iso);
270	if oVM is None:
271	return False;
272
273	if 'tst-sles11' in self.asTestVMs:
274	oVM = self.createTestVMs('tst-sles11', 1, '3.0/tcp/sles11.vdi', sKind = 'OpenSUSE', fIoApic = True, \
275	sDvdImage = sVBoxValidationKit_iso);
276	if oVM is None:
277	return False;
278
279	if 'tst-sles11-64' in self.asTestVMs:
280	oVM = self.createTestVMs('tst-sles11-64', 1, '3.0/tcp/sles11-64.vdi', sKind = 'OpenSUSE_64', \
281	sDvdImage = sVBoxValidationKit_iso);
282	if oVM is None:
283	return False;
284
285	if 'tst-oel' in self.asTestVMs:
286	oVM = self.createTestVMs('tst-oel', 1, '3.0/tcp/oel.vdi', sKind = 'Oracle', fIoApic = True, \
287	sDvdImage = sVBoxValidationKit_iso);
288	if oVM is None:
289	return False;
290
291	if 'tst-oel-64' in self.asTestVMs:
292	oVM = self.createTestVMs('tst-oel-64', 1, '3.0/tcp/oel-64.vdi', sKind = 'Oracle_64', \
293	sDvdImage = sVBoxValidationKit_iso);
294	if oVM is None:
295	return False;
296
297	# Windows VMs
298	if 'tst-win2k3ent' in self.asTestVMs:
299	oVM = self.createTestVMs('tst-win2k3ent', 1, '3.0/tcp/win2k3ent-acpi.vdi', sKind = 'Windows2003', \
300	sDvdImage = sVBoxValidationKit_iso);
301	if oVM is None:
302	return False;
303
304	if 'tst-win2k3ent-64' in self.asTestVMs:
305	oVM = self.createTestVMs('tst-win2k3ent-64', 1, '3.0/tcp/win2k3ent-64.vdi', sKind = 'Windows2003_64', \
306	sDvdImage = sVBoxValidationKit_iso);
307	if oVM is None:
308	return False;
309
310	if 'tst-win2k8' in self.asTestVMs:
311	oVM = self.createTestVMs('tst-win2k8', 1, '3.0/tcp/win2k8.vdi', sKind = 'Windows2008_64', \
312	sDvdImage = sVBoxValidationKit_iso);
313	if oVM is None:
314	return False;
315
316	# Solaris VMs
317	if 'tst-sol10' in self.asTestVMs:
318	oVM = self.createTestVMs('tst-sol10', 1, '3.0/tcp/solaris10.vdi', sKind = 'Solaris_64', \
319	sDvdImage = sVBoxValidationKit_iso);
320	if oVM is None:
321	return False;
322
323	if 'tst-sol11' in self.asTestVMs:
324	oVM = self.createTestVMs('tst-sol11', 1, '3.0/tcp/os2009-11.vdi', sKind = 'Solaris_64', \
325	sDvdImage = sVBoxValidationKit_iso);
326	if oVM is None:
327	return False;
328
329	return True;
330
331	def actionExecute(self):
332	"""
333	Execute the testcase.
334	"""
335	fRc = 'test1' not in self.asTests or self.test1();
336	if fRc: fRc = 'test2' not in self.asTests or self.test2();
337	return fRc;
338
339
340	#
341	# Test config helpers.
342	#
343
344	def createTestVMs(self, sName, iGroup, tArgs, *dKeywordArgs):
345	"""
346	Wrapper around vbox.createTestVM for creating two VMs, the source
347	(sName-1) and target (sName-2).
348
349	Returns the 2nd VM object on success, None on failure.
350	"""
351	sName1 = sName + '-1';
352	oVM = self.createTestVM(sName1, iGroup * 2, tArgs, *dKeywordArgs);
353	if oVM is not None:
354	sName2 = sName + '-2';
355	oVM = self.createTestVM(sName2, iGroup * 2 + 1, tArgs, *dKeywordArgs);
356	return oVM;
357
358
359	#
360	# Test execution helpers.
361	#
362
363	def test2Teleport(self, oVmSrc, oSessionSrc, oVmDst):
364	"""
365	Attempts a teleportation.
366
367	Returns the input parameters for the next test2Teleport call (source
368	and destiation are switched around). The input session is closed and
369	removed from the task list, while the return session is in the list.
370	"""
371
372	# Enable the teleporter of the VM.
373	oSession = self.openSession(oVmDst);
374	fRc = oSession is not None
375	if fRc:
376	fRc = oSession.enableTeleporter();
377	fRc = fRc and oSession.saveSettings();
378	fRc = oSession.close() and fRc and True; # pychecker hack.
379	if fRc:
380	# Start the destination VM.
381	oSessionDst, oProgressDst = self.startVmEx(oVmDst, fWait = False);
382	if oSessionDst is not None:
383	if oProgressDst.waitForOperation(iOperation = -3) == 0:
384
385	# Do the teleportation.
386	try:
387	uDstPort = oVmDst.teleporterPort;
388	except:
389	reporter.logXcpt();
390	uDstPort = 6502;
391	oProgressSrc = oSessionSrc.teleport('localhost', uDstPort, 'password', 1024);
392	if oProgressSrc is not None:
393	oProgressSrc.wait();
394	if oProgressSrc.isSuccess():
395	oProgressDst.wait();
396	if oProgressSrc.isSuccess() and oProgressDst.isSuccess():
397
398	# Terminate the source VM.
399	self.terminateVmBySession(oSessionSrc, oProgressSrc);
400
401	# Return with the source and destination swapped.
402	return oVmDst, oSessionDst, oVmSrc;
403
404	# Failure / bail out.
405	oProgressSrc.logResult();
406	oProgressDst.logResult();
407	self.terminateVmBySession(oSessionDst, oProgressDst);
408	return oVmSrc, oSessionSrc, oVmDst;
409
410	def test2OneCfg(self, sVmBaseName, cCpus, fHwVirt, fNestedPaging):
411	"""
412	Runs the specified VM thru test #1.
413	"""
414
415	# Reconfigure the source VM.
416	oVmSrc = self.getVmByName(sVmBaseName + '-1');
417	fRc = True;
418	oSession = self.openSession(oVmSrc);
419	if oSession is not None:
420	fRc = fRc and oSession.enableVirtExX86(fHwVirt);
421	fRc = fRc and oSession.enableNestedPagingX86(fNestedPaging);
422	fRc = fRc and oSession.setCpuCount(cCpus);
423	fRc = fRc and oSession.setupTeleporter(False, uPort=6501, sPassword='password');
424	fRc = fRc and oSession.saveSettings();
425	fRc = oSession.close() and fRc and True; # pychecker hack.
426	oSession = None;
427	else:
428	fRc = False;
429
430	# Reconfigure the destination VM.
431	oVmDst = self.getVmByName(sVmBaseName + '-2');
432	oSession = self.openSession(oVmDst);
433	if oSession is not None:
434	fRc = fRc and oSession.enableVirtExX86(fHwVirt);
435	fRc = fRc and oSession.enableNestedPagingX86(fNestedPaging);
436	fRc = fRc and oSession.setCpuCount(cCpus);
437	fRc = fRc and oSession.setupTeleporter(True, uPort=6502, sPassword='password');
438	fRc = fRc and oSession.saveSettings();
439	fRc = oSession.close() and fRc and True; # pychecker hack.
440	oSession = None;
441	else:
442	fRc = False;
443
444	# Simple test.
445	if fRc is True:
446	self.logVmInfo(oVmSrc);
447	self.logVmInfo(oVmDst);
448
449	# Start the source VM.
450	oSessionSrc = self.startVm(oVmSrc);
451	if oSessionSrc is not None:
452	# Simple back and forth to test the ice...
453	cTeleportations = 0;
454	oVmSrc, oSessionSrc, oVmDst = self.test2Teleport(oVmSrc, oSessionSrc, oVmDst);
455	if reporter.testErrorCount() == 0:
456	cTeleportations += 1;
457	oVmSrc, oSessionSrc, oVmDst = self.test2Teleport(oVmSrc, oSessionSrc, oVmDst);
458
459	# Teleport back and forth for a while.
460	msStart = base.timestampMilli();
461	while reporter.testErrorCount() == 0:
462	cTeleportations += 1;
463	if oSessionSrc.txsTryConnectViaTcp(2500, 'localhost') is True:
464	break;
465	oVmSrc, oSessionSrc, oVmDst = self.test2Teleport(oVmSrc, oSessionSrc, oVmDst);
466	cMsElapsed = base.timestampMilli() - msStart;
467	if cMsElapsed > 5*60000:
468	reporter.testFailure('TXS did not show up after %u min of teleporting (%u)...' \
469	% (cMsElapsed / 60000.0, cTeleportations));
470	break;
471
472	# Clean up the source VM.
473	self.terminateVmBySession(oSessionSrc)
474	return None;
475
476	def test2OneVM(self, sVmBaseName, asSupVirtModes = None, rSupCpus = range(1, 256)):
477	"""
478	Runs one VM (a pair really) thru the various configurations.
479	"""
480	if asSupVirtModes is None:
481	asSupVirtModes = self.asVirtModes;
482
483	reporter.testStart(sVmBaseName);
484	for cCpus in self.acCpus:
485	if cCpus == 1: reporter.testStart('1 cpu');
486	else: reporter.testStart('%u cpus' % (cCpus));
487
488	for sVirtMode in self.asVirtModes:
489	if sVirtMode == 'raw' and cCpus > 1:
490	continue;
491	if cCpus not in rSupCpus:
492	continue;
493	if sVirtMode not in asSupVirtModes:
494	continue;
495	hsVirtModeDesc = {};
496	hsVirtModeDesc['raw'] = 'Raw-mode';
497	hsVirtModeDesc['hwvirt'] = 'HwVirt';
498	hsVirtModeDesc['hwvirt-np'] = 'NestedPaging';
499	reporter.testStart(hsVirtModeDesc[sVirtMode]);
500
501	fHwVirt = sVirtMode != 'raw';
502	fNestedPaging = sVirtMode == 'hwvirt-np';
503	self.test2OneCfg(sVmBaseName, cCpus, fHwVirt, fNestedPaging);
504
505	reporter.testDone();
506	reporter.testDone();
507	return reporter.testDone()[1] == 0;
508
509	def test2(self):
510	"""
511	Executes test #2.
512	"""
513
514	# Loop thru the test VMs.
515	fRc = True;
516	for sVM in self.asTestVMs:
517	# figure args.
518	asSupVirtModes = None;
519	if sVM in ('tst-sol11', 'tst-sol10'): # 64-bit only
520	asSupVirtModes = ['hwvirt', 'hwvirt-np',];
521
522	# run test on the VM.
523	if not self.test2OneVM(sVM, asSupVirtModes):
524	fRc = False;
525
526	return fRc;
527	#
528	# Test #1
529	#
530
531	def test1ResetVmConfig(self, oVM, fTeleporterEnabled = False):
532	"""
533	Resets the teleportation config for the specified VM.
534	Returns True on success, False on failure.
535	"""
536	oSession = self.openSession(oVM);
537	if oSession is not None:
538	fRc = oSession.setupTeleporter(fTeleporterEnabled, uPort=6502, sPassword='password');
539	fRc = fRc and oSession.saveSettings();
540	if not oSession.close(): fRc = False;
541	oSession = None;
542	else:
543	fRc = False;
544	return fRc;
545
546	def test1Sub7(self, oVmSrc, oVmDst):
547	"""
548	Test the password check.
549	"""
550	reporter.testStart('Bad password');
551	if self.test1ResetVmConfig(oVmSrc, fTeleporterEnabled = False) \
552	and self.test1ResetVmConfig(oVmDst, fTeleporterEnabled = True):
553	# Start the target VM.
554	oSessionDst, oProgressDst = self.startVmEx(oVmDst, fWait = False);
555	if oSessionDst is not None:
556	if oProgressDst.waitForOperation(iOperation = -3) == 0:
557	# Start the source VM.
558	oSessionSrc = self.startVm(oVmSrc);
559	if oSessionSrc is not None:
560	tsPasswords = ('password-bad', 'passwor', 'pass', 'p', '', 'Password', );
561	for sPassword in tsPasswords:
562	reporter.testStart(sPassword);
563	oProgressSrc = oSessionSrc.teleport('localhost', 6502, sPassword);
564	if oProgressSrc:
565	oProgressSrc.wait();
566	reporter.log('src: %s' % oProgressSrc.stringifyResult());
567	if oProgressSrc.isSuccess():
568	reporter.testFailure('IConsole::teleport succeeded with bad password "%s"' % sPassword);
569	elif oProgressSrc.getErrInfoResultCode() != vbox.ComError.E_FAIL:
570	reporter.testFailure('IConsole::teleport returns %s instead of E_FAIL' \
571	% (vbox.ComError.toString(oProgressSrc.getErrInfoResultCode()),));
572	elif oProgressSrc.getErrInfoText() != 'Invalid password':
573	reporter.testFailure('IConsole::teleport returns "%s" instead of "Invalid password"' \
574	% (oProgressSrc.getErrInfoText(),));
575	elif oProgressDst.isCompleted():
576	reporter.testFailure('Destination completed unexpectedly after bad password "%s"' \
577	% sPassword);
578	else:
579	reporter.testFailure('IConsole::teleport failed with password "%s"' % sPassword);
580	if reporter.testDone()[1] != 0:
581	break;
582	self.terminateVmBySession(oSessionSrc, oProgressSrc);
583	self.terminateVmBySession(oSessionDst, oProgressDst);
584	else:
585	reporter.testFailure('reconfig failed');
586	return reporter.testDone()[1] == 0;
587
588	def test1Sub6(self, oVmSrc, oVmDst):
589	"""
590	Misconfigure the target VM and check that teleportation fails with the
591	same status and message on both ends.
592	xTracker: #4813
593	"""
594	reporter.testStart('Misconfiguration & error message');
595	if self.test1ResetVmConfig(oVmSrc, fTeleporterEnabled = False) \
596	and self.test1ResetVmConfig(oVmDst, fTeleporterEnabled = True):
597	# Give the source a bit more RAM.
598	oSession = self.openSession(oVmSrc);
599	if oSession is not None:
600	try: cbMB = oVmSrc.memorySize + 4;
601	except: cbMB = 1; fRc = False;
602	fRc = oSession.setRamSize(cbMB);
603	if not oSession.saveSettings(): fRc = False;
604	if not oSession.close(): fRc = False;
605	oSession = None;
606	else:
607	fRc = False;
608	if fRc:
609	# Start the target VM.
610	oSessionDst, oProgressDst = self.startVmEx(oVmDst, fWait = False);
611	if oSessionDst is not None:
612	if oProgressDst.waitForOperation(iOperation = -3) == 0:
613	# Start the source VM.
614	oSessionSrc = self.startVm(oVmSrc);
615	if oSessionSrc is not None:
616	# Try teleport.
617	oProgressSrc = oSessionSrc.teleport('localhost', 6502, 'password');
618	if oProgressSrc:
619	oProgressSrc.wait();
620	oProgressDst.wait();
621
622	reporter.log('src: %s' % oProgressSrc.stringifyResult());
623	reporter.log('dst: %s' % oProgressDst.stringifyResult());
624
625	# Make sure it failed.
626	if oProgressSrc.isSuccess() and oProgressDst.isSuccess():
627	reporter.testFailure('The teleporation did not fail as expected');
628
629	# Compare the results.
630	if oProgressSrc.getResult() != oProgressDst.getResult():
631	reporter.testFailure('Result differs - src=%s dst=%s' \
632	% (vbox.ComError.toString(oProgressSrc.getResult()),\
633	vbox.ComError.toString(oProgressDst.getResult())));
634	elif oProgressSrc.getErrInfoResultCode() != oProgressDst.getErrInfoResultCode():
635	reporter.testFailure('ErrorInfo::resultCode differs - src=%s dst=%s' \
636	% (vbox.ComError.toString(oProgressSrc.getErrInfoResultCode()),\
637	vbox.ComError.toString(oProgressDst.getErrInfoResultCode())));
638	elif oProgressSrc.getErrInfoText() != oProgressDst.getErrInfoText():
639	reporter.testFailure('ErrorInfo::text differs - src="%s" dst="%s"' \
640	% (oProgressSrc.getErrInfoText(), oProgressDst.getErrInfoText()));
641
642	self.terminateVmBySession(oSessionSrc, oProgressSrc);
643	self.terminateVmBySession(oSessionDst, oProgressDst);
644	self.test1ResetVmConfig(oVmSrc, fTeleporterEnabled = False)
645	self.test1ResetVmConfig(oVmDst, fTeleporterEnabled = True);
646	else:
647	reporter.testFailure('reconfig #2 failed');
648	else:
649	reporter.testFailure('reconfig #1 failed');
650	return reporter.testDone()[1] == 0;
651
652	def test1Sub5(self, oVmSrc, oVmDst):
653	"""
654	Test that basic teleporting works.
655	xTracker: #4749
656	"""
657	reporter.testStart('Simple teleportation');
658	for cSecsX2 in range(0, 10):
659	if self.test1ResetVmConfig(oVmSrc, fTeleporterEnabled = False) \
660	and self.test1ResetVmConfig(oVmDst, fTeleporterEnabled = True):
661	# Start the target VM.
662	oSessionDst, oProgressDst = self.startVmEx(oVmDst, fWait = False);
663	if oSessionDst is not None:
664	if oProgressDst.waitForOperation(iOperation = -3) == 0:
665	# Start the source VM.
666	oSessionSrc = self.startVm(oVmSrc);
667	if oSessionSrc is not None:
668	self.sleep(cSecsX2 / 2);
669	# Try teleport.
670	oProgressSrc = oSessionSrc.teleport('localhost', 6502, 'password');
671	if oProgressSrc:
672	oProgressSrc.wait();
673	oProgressDst.wait();
674
675	self.terminateVmBySession(oSessionSrc, oProgressSrc);
676	self.terminateVmBySession(oSessionDst, oProgressDst);
677	else:
678	reporter.testFailure('reconfig failed');
679	return reporter.testDone()[1] == 0;
680
681	def test1Sub4(self, oVM):
682	"""
683	Test that we can start and cancel a teleportation target.
684	(No source VM trying to connect here.)
685	xTracker: #4965
686	"""
687	reporter.testStart('openRemoteSession cancel');
688	for cSecsX2 in range(0, 10):
689	if self.test1ResetVmConfig(oVM, fTeleporterEnabled = True):
690	oSession, oProgress = self.startVmEx(oVM, fWait = False);
691	if oSession is not None:
692	self.sleep(cSecsX2 / 2);
693	oProgress.cancel();
694	oProgress.wait();
695	self.terminateVmBySession(oSession, oProgress);
696	else:
697	reporter.testFailure('reconfig failed');
698	return reporter.testDone()[1] == 0;
699
700	def test1Sub3(self, oVM):
701	"""
702	Test that starting a teleportation target VM will fail if we give it
703	a bad address to bind to.
704	"""
705	reporter.testStart('bad IMachine::teleporterAddress');
706
707	# re-configure it with a bad bind-to address.
708	fRc = False;
709	oSession = self.openSession(oVM);
710	if oSession is not None:
711	fRc = oSession.setupTeleporter(True, uPort=6502, sAddress='no.such.hostname.should.ever.exist.duh');
712	if not oSession.saveSettings(fClose=True): fRc = False;
713	oSession = None;
714	if fRc:
715	# Try start it.
716	oSession, oProgress = self.startVmEx(oVM, fWait = False);
717	if oSession is not None:
718	oProgress.wait();
719	## TODO: exact error code and look for the IPRT right string.
720	if not oProgress.isCompleted() or oProgress.getResult() >= 0:
721	reporter.testFailure('%s' % (oProgress.stringifyResult(),));
722	self.terminateVmBySession(oSession, oProgress);
723
724	# put back the old teleporter setup.
725	self.test1ResetVmConfig(oVM, fTeleporterEnabled = True);
726	else:
727	reporter.testFailure('reconfig #1 failed');
728	return reporter.testDone()[1] == 0;
729
730	# test1Sub2 - start
731
732	def test1Sub2SetEnabled(self, oSession, fEnabled):
733	""" This should never fail."""
734	try:
735	oSession.o.machine.teleporterEnabled = fEnabled;
736	except:
737	reporter.testFailureXcpt('machine.teleporterEnabled=%s' % (fEnabled,));
738	return False;
739	try:
740	fNew = oSession.o.machine.teleporterEnabled;
741	except:
742	reporter.testFailureXcpt();
743	return False;
744	if fNew != fEnabled:
745	reporter.testFailure('machine.teleporterEnabled=%s but afterwards it is actually %s' % (fEnabled, fNew));
746	return False;
747	return True;
748
749	def test1Sub2SetPassword(self, oSession, sPassword):
750	""" This should never fail."""
751	try:
752	oSession.o.machine.teleporterPassword = sPassword;
753	except:
754	reporter.testFailureXcpt('machine.teleporterPassword=%s' % (sPassword,));
755	return False;
756	try:
757	sNew = oSession.o.machine.teleporterPassword;
758	except:
759	reporter.testFailureXcpt();
760	return False;
761	if sNew != sPassword:
762	reporter.testFailure('machine.teleporterPassword="%s" but afterwards it is actually "%s"' % (sPassword, sNew));
763	return False;
764	return True;
765
766	def test1Sub2SetPort(self, oSession, uPort, fInvalid = False):
767	""" This can fail, thus fInvalid."""
768	if not fInvalid:
769	uOld = uPort;
770	else:
771	try: uOld = oSession.o.machine.teleporterPort;
772	except: return reporter.testFailureXcpt();
773
774	try:
775	oSession.o.machine.teleporterPort = uPort;
776	except Exception as oXcpt:
777	if not fInvalid or vbox.ComError.notEqual(oXcpt, vbox.ComError.E_INVALIDARG):
778	return reporter.testFailureXcpt('machine.teleporterPort=%u' % (uPort,));
779	else:
780	if fInvalid:
781	return reporter.testFailureXcpt('machine.teleporterPort=%u succeeded unexpectedly' % (uPort,));
782
783	try: uNew = oSession.o.machine.teleporterPort;
784	except: return reporter.testFailureXcpt();
785	if uNew != uOld:
786	if not fInvalid:
787	reporter.testFailure('machine.teleporterPort=%u but afterwards it is actually %u' % (uPort, uNew));
788	else:
789	reporter.testFailure('machine.teleporterPort is %u after failure, expected %u' % (uNew, uOld));
790	return False;
791	return True;
792
793	def test1Sub2SetAddress(self, oSession, sAddress):
794	""" This should never fail."""
795	try:
796	oSession.o.machine.teleporterAddress = sAddress;
797	except:
798	reporter.testFailureXcpt('machine.teleporterAddress=%s' % (sAddress,));
799	return False;
800	try:
801	sNew = oSession.o.machine.teleporterAddress;
802	except:
803	reporter.testFailureXcpt();
804	return False;
805	if sNew != sAddress:
806	reporter.testFailure('machine.teleporterAddress="%s" but afterwards it is actually "%s"' % (sAddress, sNew));
807	return False;
808	return True;
809
810	def test1Sub2(self, oVM):
811	"""
812	Test the attributes, making sure that we get exceptions on bad values.
813	"""
814	reporter.testStart('IMachine::teleport*');
815
816	# Save the original teleporter attributes for the discard test.
817	try:
818	sOrgAddress = oVM.teleporterAddress;
819	uOrgPort = oVM.teleporterPort;
820	sOrgPassword = oVM.teleporterPassword;
821	fOrgEnabled = oVM.teleporterEnabled;
822	except:
823	reporter.testFailureXcpt();
824	else:
825	# Open a session and start messing with the properties.
826	oSession = self.openSession(oVM);
827	if oSession is not None:
828	# Anything goes for the address.
829	reporter.testStart('teleporterAddress');
830	self.test1Sub2SetAddress(oSession, '');
831	self.test1Sub2SetAddress(oSession, '1');
832	self.test1Sub2SetAddress(oSession, 'Anything goes! ^&$@!$%^');
833	reporter.testDone();
834
835	# The port is restricted to {0..65535}.
836	reporter.testStart('teleporterPort');
837	for uPort in range(0, 1000) + range(16000, 17000) + range(32000, 33000) + range(65000, 65536):
838	if not self.test1Sub2SetPort(oSession, uPort):
839	break;
840	self.processPendingEvents();
841	reporter.testDone();
842
843	reporter.testStart('teleporterPort negative');
844	self.test1Sub2SetPort(oSession, 65536, True);
845	self.test1Sub2SetPort(oSession, 999999, True);
846	reporter.testDone();
847
848	# Anything goes for the password.
849	reporter.testStart('teleporterPassword');
850	self.test1Sub2SetPassword(oSession, 'password');
851	self.test1Sub2SetPassword(oSession, '');
852	self.test1Sub2SetPassword(oSession, '1');
853	self.test1Sub2SetPassword(oSession, 'Anything goes! ^&$@!$%^');
854	reporter.testDone();
855
856	# Just test that it works.
857	reporter.testStart('teleporterEnabled');
858	self.test1Sub2SetEnabled(oSession, True);
859	self.test1Sub2SetEnabled(oSession, True);
860	self.test1Sub2SetEnabled(oSession, False);
861	self.test1Sub2SetEnabled(oSession, False);
862	reporter.testDone();
863
864	# Finally, discard the changes, close the session and check
865	# that we're back to the originals.
866	if not oSession.discardSettings(True):
867	reporter.testFailure('Failed to discard settings & close the session')
868	else:
869	reporter.testFailure('Failed to open VM session')
870	try:
871	if oVM.teleporterAddress != sOrgAddress: reporter.testFailure('Rollback failed for teleporterAddress');
872	if oVM.teleporterPort != uOrgPort: reporter.testFailure('Rollback failed for teleporterPort');
873	if oVM.teleporterPassword != sOrgPassword: reporter.testFailure('Rollback failed for teleporterPassword');
874	if oVM.teleporterEnabled != fOrgEnabled: reporter.testFailure('Rollback failed for teleporterEnabled');
875	except:
876	reporter.testFailureXcpt();
877	return reporter.testDone()[1] != 0;
878
879	# test1Sub1 - start
880
881	def test1Sub1DoTeleport(self, oSession, sHostname, uPort, sPassword, cMsMaxDowntime, hrcExpected, sTestName):
882	""" Do a bad IConsole::teleport call and check the result."""
883	reporter.testStart(sTestName);
884	fRc = False;
885	try:
886	oProgress = oSession.o.console.teleport(sHostname, uPort, sPassword, cMsMaxDowntime);
887	except vbox.ComException as oXcpt:
888	if vbox.ComError.equal(oXcpt, hrcExpected):
889	fRc = True;
890	else:
891	reporter.testFailure('hresult %s, expected %s' \
892	% (vbox.ComError.toString(oXcpt.hresult),
893	vbox.ComError.toString(hrcExpected)));
894	except Exception as oXcpt:
895	reporter.testFailure('Unexpected exception %s' % (oXcpt));
896	else:
897	reporter.testFailure('Unpexected success');
898	oProgress.cancel();
899	oProgress.wait();
900	reporter.testDone();
901	return fRc;
902
903	def test1Sub1(self, oVM):
904	""" Test simple IConsole::teleport() failure paths. """
905	reporter.testStart('IConsole::teleport');
906	oSession = self.startVm(oVM);
907	if oSession:
908	self.test1Sub1DoTeleport(oSession, 'localhost', 65536, 'password', 10000,
909	vbox.ComError.E_INVALIDARG, 'Bad port value 65536');
910	self.test1Sub1DoTeleport(oSession, 'localhost', 0, 'password', 10000,
911	vbox.ComError.E_INVALIDARG, 'Bad port value 0');
912	self.test1Sub1DoTeleport(oSession, 'localhost', 5000, 'password', 0,
913	vbox.ComError.E_INVALIDARG, 'Bad max downtime');
914	self.test1Sub1DoTeleport(oSession, '', 5000, 'password', 10000,
915	vbox.ComError.E_INVALIDARG, 'No hostname');
916	self.test1Sub1DoTeleport(oSession, 'no.such.hostname.should.ever.exist.duh', 5000, 'password', 0,
917	vbox.ComError.E_INVALIDARG, 'Non-existing host');
918
919	self.terminateVmBySession(oSession)
920	else:
921	reporter.testFailure('startVm');
922	return reporter.testDone()[1] == 0;
923
924
925	def test1(self):
926	"""
927	Executes test #1 - Negative API testing.
928
929	ASSUMES that the VMs are
930	"""
931	reporter.testStart('Test 1');
932
933	# Get the VMs.
934	#oVmHwVirt1 = self.getVmByName('tst-empty-hwvirt-1');
935	#oVmHwVirt2 = self.getVmByName('tst-empty-hwvirt-2');
936	oVmRaw1 = self.getVmByName('tst-empty-raw-1');
937	oVmRaw2 = self.getVmByName('tst-empty-raw-2');
938
939	# Reset their teleportation related configuration.
940	fRc = True;
941	#for oVM in (oVmHwVirt1, oVmHwVirt2, oVmRaw1, oVmRaw2):
942	for oVM in (oVmRaw1, oVmRaw2):
943	if not self.test1ResetVmConfig(oVM): fRc = False;
944
945	# Do the testing (don't both with fRc on the subtests).
946	if fRc:
947	self.test1Sub1(oVmRaw1);
948	self.test1Sub2(oVmRaw2);
949	self.test1Sub3(oVmRaw2);
950	self.test1Sub4(oVmRaw2);
951	self.processPendingEvents();
952	self.test1Sub5(oVmRaw1, oVmRaw2);
953	self.test1Sub6(oVmRaw1, oVmRaw2);
954	self.test1Sub7(oVmRaw1, oVmRaw2);
955	else:
956	reporter.testFailure('Failed to reset the VM configs')
957	return reporter.testDone()[1] == 0;
958
959
960
961	if __name__ == '__main__':
962	sys.exit(tdTeleportLocal1().main(sys.argv));
963

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

source: vbox/trunk/src/VBox/ValidationKit/tests/teleportation/tdTeleportLocal1.py

Download in other formats: