1 | #!/usr/bin/env python
|
---|
2 | # -*- coding: utf-8 -*-
|
---|
3 | # $Id: tdTreeDepth1.py 106061 2024-09-16 14:03:52Z vboxsync $
|
---|
4 |
|
---|
5 | """
|
---|
6 | VirtualBox Validation Kit - Medium and Snapshot Tree Depth Test #1
|
---|
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 | import random
|
---|
47 | import time
|
---|
48 | import gc
|
---|
49 |
|
---|
50 | # Only the main script needs to modify the path.
|
---|
51 | try: __file__ # pylint: disable=used-before-assignment
|
---|
52 | except: __file__ = sys.argv[0]
|
---|
53 | g_ksValidationKitDir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
---|
54 | sys.path.append(g_ksValidationKitDir)
|
---|
55 |
|
---|
56 | # Validation Kit imports.
|
---|
57 | from testdriver import base
|
---|
58 | from testdriver import reporter
|
---|
59 | from testdriver import vboxcon
|
---|
60 |
|
---|
61 |
|
---|
62 | class SubTstDrvTreeDepth1(base.SubTestDriverBase):
|
---|
63 | """
|
---|
64 | Sub-test driver for Medium and Snapshot Tree Depth Test #1.
|
---|
65 | """
|
---|
66 |
|
---|
67 | def __init__(self, oTstDrv):
|
---|
68 | base.SubTestDriverBase.__init__(self, oTstDrv, 'tree-depth', 'Media and Snapshot tree depths');
|
---|
69 |
|
---|
70 | def testIt(self):
|
---|
71 | """
|
---|
72 | Execute the sub-testcase.
|
---|
73 | """
|
---|
74 | return self.testMediumTreeDepth() \
|
---|
75 | and self.testSnapshotTreeDepth();
|
---|
76 |
|
---|
77 | def getNumOfAttachedHDs(self, oVM, sController):
|
---|
78 | """
|
---|
79 | Helper routine for counting the hard disks, including differencing disks,
|
---|
80 | attached to the specified countroller.
|
---|
81 | """
|
---|
82 | try:
|
---|
83 | aoMediumAttachments = oVM.getMediumAttachmentsOfController(sController);
|
---|
84 | except:
|
---|
85 | reporter.errorXcpt('oVM.getMediumAttachmentsOfController("%s") failed' % (sController));
|
---|
86 | return -1;
|
---|
87 |
|
---|
88 | cDisks = 0;
|
---|
89 | for oAttachment in aoMediumAttachments:
|
---|
90 | oMedium = oAttachment.medium;
|
---|
91 | if oMedium.deviceType is vboxcon.DeviceType_HardDisk:
|
---|
92 | cDisks = cDisks + 1;
|
---|
93 | reporter.log2('base medium = %s cDisks = %d' % (oMedium.location, cDisks));
|
---|
94 | oParentMedium = oMedium.parent;
|
---|
95 | while oParentMedium is not None:
|
---|
96 | cDisks = cDisks + 1;
|
---|
97 | reporter.log2('parent medium = %s cDisks = %d' % (oParentMedium.location, cDisks));
|
---|
98 | oParentMedium = oParentMedium.parent;
|
---|
99 | return cDisks;
|
---|
100 |
|
---|
101 | def getNumOfHDsFromHardDisksArray(self):
|
---|
102 | """
|
---|
103 | Helper routine for counting the hard disks that belong to this VM which are
|
---|
104 | contained in the global IVirtualBox::hardDisks[] array (which contains base media
|
---|
105 | only).
|
---|
106 | """
|
---|
107 | cDisks = 0;
|
---|
108 | aoHDs = self.oTstDrv.oVBoxMgr.getArray(self.oTstDrv.oVBox, 'hardDisks');
|
---|
109 | # Walk the IVirtualBox::hardDisks[] array for hard disks belonging to this VM.
|
---|
110 | # The array may contain entries from other VMs created by previous API tessts
|
---|
111 | # which haven't unregistered and deleted their VM's configuration. The location
|
---|
112 | # attribute of each disk looks similar to:
|
---|
113 | # /var/tmp/VBoxTestTmp/VBoxUserHome/Machines/vm 2/tdAppliance1-t2-disk1.vmdk
|
---|
114 | # So we compare the directory where the disk is located, i.e. the 'basefolder',
|
---|
115 | # with our VM's basefolder which looks similar to:
|
---|
116 | # /var/tmp/VBoxTestTmp/VBoxUserHome/Machines/test-medium
|
---|
117 | # to determine if the disk belongs to us.
|
---|
118 | for oHD in aoHDs:
|
---|
119 | sHDBaseFolder = os.path.dirname(oHD.location);
|
---|
120 | sVMBaseFolder = os.path.join(self.oTstDrv.oVBox.systemProperties.defaultMachineFolder, 'test-medium');
|
---|
121 | reporter.log2('HDBaseFolder = %s VMBaseFolder = %s' % (sHDBaseFolder, sVMBaseFolder));
|
---|
122 | if sHDBaseFolder== sVMBaseFolder:
|
---|
123 | cDisks = cDisks + 1;
|
---|
124 | return cDisks;
|
---|
125 |
|
---|
126 | def unregisterVM(self, oVM, enmCleanupMode):
|
---|
127 | """
|
---|
128 | Helper routine to unregister the VM using the CleanupMode specified.
|
---|
129 | """
|
---|
130 | reporter.log('unregistering VM using %s' % enmCleanupMode);
|
---|
131 | if enmCleanupMode == 'DetachAllReturnHardDisksOnly':
|
---|
132 | try:
|
---|
133 | aoHDs = oVM.unregister(vboxcon.CleanupMode_DetachAllReturnHardDisksOnly);
|
---|
134 | except:
|
---|
135 | return reporter.logXcpt('unregister(CleanupMode_DetachAllReturnHardDisksOnly) failed');
|
---|
136 | for oHD in aoHDs:
|
---|
137 | reporter.log2('closing medium = %s' % (oHD.location));
|
---|
138 | oHD.close();
|
---|
139 | aoHDs = None;
|
---|
140 | elif enmCleanupMode == 'UnregisterOnly':
|
---|
141 | try:
|
---|
142 | aoHDs = oVM.unregister(vboxcon.CleanupMode_UnregisterOnly);
|
---|
143 | except:
|
---|
144 | return reporter.logXcpt('unregister(CleanupMode_UnregisterOnly) failed');
|
---|
145 | if aoHDs:
|
---|
146 | return reporter.error('unregister(CleanupMode_UnregisterOnly) failed: returned %d disks' %
|
---|
147 | len(aoHDs));
|
---|
148 | else:
|
---|
149 | return reporter.error('unregisterVM: unexpected CleanupMode "%s"' % enmCleanupMode);
|
---|
150 |
|
---|
151 | return True;
|
---|
152 |
|
---|
153 | def openAndRegisterMachine(self, sSettingsFile):
|
---|
154 | """
|
---|
155 | Helper routine which opens a VM and registers it.
|
---|
156 | """
|
---|
157 | reporter.log('opening VM using configuration file = %s, testing config reading' % (sSettingsFile));
|
---|
158 | try:
|
---|
159 | if self.oTstDrv.fpApiVer >= 7.0:
|
---|
160 | # Needs a password parameter since 7.0.
|
---|
161 | oVM = self.oTstDrv.oVBox.openMachine(sSettingsFile, "");
|
---|
162 | else:
|
---|
163 | oVM = self.oTstDrv.oVBox.openMachine(sSettingsFile);
|
---|
164 | except:
|
---|
165 | reporter.logXcpt('openMachine(%s) failed' % (sSettingsFile));
|
---|
166 | return None;
|
---|
167 |
|
---|
168 | if not oVM:
|
---|
169 | return None;
|
---|
170 |
|
---|
171 | try:
|
---|
172 | self.oTstDrv.oVBox.registerMachine(oVM);
|
---|
173 | except:
|
---|
174 | reporter.logXcpt('registerMachine(%s) failed' % (sSettingsFile));
|
---|
175 | return None;
|
---|
176 |
|
---|
177 | return oVM;
|
---|
178 |
|
---|
179 | #
|
---|
180 | # Test execution helpers.
|
---|
181 | #
|
---|
182 |
|
---|
183 | def testMediumTreeDepth(self):
|
---|
184 | """
|
---|
185 | Test medium tree depth.
|
---|
186 | """
|
---|
187 | reporter.testStart('mediumTreeDepth')
|
---|
188 |
|
---|
189 | oVM = self.oTstDrv.createTestVMOnly('test-medium', 'Other')
|
---|
190 | if oVM is None:
|
---|
191 | return False;
|
---|
192 |
|
---|
193 | # Save the path to the VM's settings file while oVM is valid (needed for
|
---|
194 | # openMachine() later when oVM is gone).
|
---|
195 | sSettingsFile = oVM.settingsFilePath
|
---|
196 |
|
---|
197 | oSession = self.oTstDrv.openSession(oVM)
|
---|
198 | if oSession is None:
|
---|
199 | return False;
|
---|
200 |
|
---|
201 | # create chain with up to 64 disk images (medium tree depth limit)
|
---|
202 | cImages = random.randrange(1, 64);
|
---|
203 | reporter.log('Creating chain with %d disk images' % (cImages))
|
---|
204 | for i in range(1, cImages + 1):
|
---|
205 | sHddPath = os.path.join(self.oTstDrv.sScratchPath, 'Test' + str(i) + '.vdi')
|
---|
206 | if i == 1:
|
---|
207 | oHd = oSession.createBaseHd(sHddPath, cb=1024*1024)
|
---|
208 | else:
|
---|
209 | oHd = oSession.createDiffHd(oHd, sHddPath)
|
---|
210 | if oHd is None:
|
---|
211 | return False;
|
---|
212 |
|
---|
213 | # modify the VM config, attach HDD
|
---|
214 | sController='SATA Controller';
|
---|
215 | fRc = oSession.attachHd(sHddPath, sController, fImmutable=False, fForceResource=False);
|
---|
216 | if fRc:
|
---|
217 | fRc = oSession.saveSettings(fClose=True);
|
---|
218 | oSession = None;
|
---|
219 | if not fRc:
|
---|
220 | return False;
|
---|
221 |
|
---|
222 | # Verify that the number of hard disks attached to the VM's only storage
|
---|
223 | # controller equals the number of disks created above.
|
---|
224 | cDisks = self.getNumOfAttachedHDs(oVM, sController);
|
---|
225 | reporter.log('Initial state: Number of hard disks attached = %d (should equal disks created = %d)' % (cDisks, cImages));
|
---|
226 | if cImages != cDisks:
|
---|
227 | reporter.error('Created %d disk images but found %d disks attached' % (cImages, cDisks));
|
---|
228 |
|
---|
229 | # unregister the VM using "DetachAllReturnHardDisksOnly" and confirm all
|
---|
230 | # hard disks were returned and subsequently closed
|
---|
231 | fRc = self.unregisterVM(oVM, 'DetachAllReturnHardDisksOnly');
|
---|
232 | if not fRc:
|
---|
233 | return False;
|
---|
234 | oVM = None;
|
---|
235 |
|
---|
236 | # If there are no base media belonging to this VM in the global IVirtualBox::hardDisks[]
|
---|
237 | # array (expected) then there are no leftover child images either.
|
---|
238 | cNumDisks = self.getNumOfHDsFromHardDisksArray();
|
---|
239 | reporter.log('After unregister(DetachAllReturnHardDisksOnly): API reports %d base images (should be zero)' % (cNumDisks));
|
---|
240 | if cNumDisks != 0:
|
---|
241 | reporter.error('Got %d initial base images, expected zero (0)' % (cNumDisks));
|
---|
242 |
|
---|
243 | # re-register to test loading of settings
|
---|
244 | oVM = self.openAndRegisterMachine(sSettingsFile);
|
---|
245 | if oVM is None:
|
---|
246 | return False;
|
---|
247 |
|
---|
248 | # Verify that the number of hard disks attached to the VM's only storage
|
---|
249 | # controller equals the number of disks created above.
|
---|
250 | cDisks = self.getNumOfAttachedHDs(oVM, sController);
|
---|
251 | reporter.log('After openMachine()+registerMachine(): Number of hard disks attached = %d '
|
---|
252 | '(should equal disks created = %d)' % (cDisks, cImages));
|
---|
253 | if cImages != cDisks:
|
---|
254 | reporter.error('Created %d disk images but after openMachine()+registerMachine() found %d disks attached' %
|
---|
255 | (cImages, cDisks));
|
---|
256 |
|
---|
257 | fRc = self.unregisterVM(oVM, 'UnregisterOnly');
|
---|
258 | if not fRc:
|
---|
259 | return False;
|
---|
260 | oVM = None;
|
---|
261 |
|
---|
262 | # When unregistering a VM with CleanupMode_UnregisterOnly the associated medium's
|
---|
263 | # objects will be closed when the corresponding Machine object ('oVM') is
|
---|
264 | # uninitialized. Assigning the 'None' object to 'oVM' will cause Python to delete
|
---|
265 | # the object when the garbage collector runs however this can take several seconds
|
---|
266 | # so we invoke the Python garbage collector manually here so we don't have to wait.
|
---|
267 | try:
|
---|
268 | gc.collect();
|
---|
269 | except:
|
---|
270 | reporter.logXcpt();
|
---|
271 |
|
---|
272 | reporter.log('Waiting three seconds for Machine::uninit() to be called to close the attached disks');
|
---|
273 | # Fudge factor: Machine::uninit() will be invoked when the oVM object is processed
|
---|
274 | # by the garbage collector above but it may take a few moments to close up to 64
|
---|
275 | # disks.
|
---|
276 | time.sleep(3);
|
---|
277 |
|
---|
278 | # If there are no base media belonging to this VM in the global IVirtualBox::hardDisks[]
|
---|
279 | # array (expected) then there are no leftover child images either.
|
---|
280 | cNumDisks = self.getNumOfHDsFromHardDisksArray();
|
---|
281 | reporter.log('After unregister(UnregisterOnly): API reports %d base images (should be zero)' % (cNumDisks));
|
---|
282 | if cNumDisks != 0:
|
---|
283 | reporter.error('Got %d base images after unregistering, expected zero (0)' % (cNumDisks));
|
---|
284 |
|
---|
285 | return reporter.testDone()[1] == 0;
|
---|
286 |
|
---|
287 | def testSnapshotTreeDepth(self):
|
---|
288 | """
|
---|
289 | Test snapshot tree depth.
|
---|
290 | """
|
---|
291 | reporter.testStart('snapshotTreeDepth')
|
---|
292 |
|
---|
293 | oVM = self.oTstDrv.createTestVMOnly('test-snap', 'Other');
|
---|
294 | if oVM is None:
|
---|
295 | return False;
|
---|
296 |
|
---|
297 | # Save the path to the VM's settings file while oVM is valid (needed for
|
---|
298 | # openMachine() later when oVM is gone).
|
---|
299 | sSettingsFile = oVM.settingsFilePath
|
---|
300 |
|
---|
301 | # modify the VM config, create and attach empty HDD
|
---|
302 | oSession = self.oTstDrv.openSession(oVM)
|
---|
303 | if oSession is None:
|
---|
304 | return False;
|
---|
305 | sHddPath = os.path.join(self.oTstDrv.sScratchPath, 'TestSnapEmpty.vdi')
|
---|
306 | sController='SATA Controller';
|
---|
307 | fRc = oSession.createAndAttachHd(sHddPath, cb=1024*1024, sController=sController, fImmutable=False);
|
---|
308 | fRc = fRc and oSession.saveSettings();
|
---|
309 | if not fRc:
|
---|
310 | return False;
|
---|
311 |
|
---|
312 | # take up to 200 snapshots (250 is the snapshot tree depth limit (settings.h:SETTINGS_SNAPSHOT_DEPTH_MAX))
|
---|
313 | cSnapshots = random.randrange(1, 200);
|
---|
314 | reporter.log('Taking %d snapshots' % (cSnapshots));
|
---|
315 | for i in range(1, cSnapshots + 1):
|
---|
316 | fRc = oSession.takeSnapshot('Snapshot ' + str(i));
|
---|
317 | if not fRc:
|
---|
318 | return False;
|
---|
319 | fRc = oSession.close() and fRc and True;
|
---|
320 | oSession = None;
|
---|
321 | reporter.log('API reports %d snapshots (should equal snapshots created = %d)' % (oVM.snapshotCount, cSnapshots));
|
---|
322 | if oVM.snapshotCount != cSnapshots:
|
---|
323 | reporter.error('Got %d initial snapshots, expected %d' % (oVM.snapshotCount, cSnapshots));
|
---|
324 |
|
---|
325 | # unregister the VM using "DetachAllReturnHardDisksOnly" and confirm all
|
---|
326 | # hard disks were returned and subsequently closed
|
---|
327 | fRc = self.unregisterVM(oVM, 'DetachAllReturnHardDisksOnly');
|
---|
328 | if not fRc:
|
---|
329 | return False;
|
---|
330 | oVM = None;
|
---|
331 |
|
---|
332 | # If there are no base media belonging to this VM in the global IVirtualBox::hardDisks[]
|
---|
333 | # array (expected) then there are no leftover child images either.
|
---|
334 | cNumDisks = self.getNumOfHDsFromHardDisksArray();
|
---|
335 | reporter.log('After unregister(DetachAllReturnHardDisksOnly): API reports %d base images (should be zero)' % (cNumDisks));
|
---|
336 | fRc = fRc and cNumDisks == 0
|
---|
337 | if cNumDisks != 0:
|
---|
338 | reporter.error('Got %d initial base images, expected zero (0)' % (cNumDisks));
|
---|
339 |
|
---|
340 | # re-register to test loading of settings
|
---|
341 | oVM = self.openAndRegisterMachine(sSettingsFile);
|
---|
342 | if oVM is None:
|
---|
343 | return False;
|
---|
344 |
|
---|
345 | # Verify that the number of hard disks attached to the VM's only storage
|
---|
346 | # controller equals the number of disks created above.
|
---|
347 | reporter.log('After openMachine()+registerMachine(): Number of snapshots of VM = %d '
|
---|
348 | '(should equal snapshots created = %d)' % (oVM.snapshotCount, cSnapshots));
|
---|
349 | if oVM.snapshotCount != cSnapshots:
|
---|
350 | reporter.error('Created %d snapshots but after openMachine()+registerMachine() found %d snapshots' %
|
---|
351 | (cSnapshots, oVM.snapshotCount));
|
---|
352 |
|
---|
353 | fRc = self.unregisterVM(oVM, 'UnregisterOnly');
|
---|
354 | if not fRc:
|
---|
355 | return False;
|
---|
356 | oVM = None;
|
---|
357 |
|
---|
358 | # When unregistering a VM with CleanupMode_UnregisterOnly the associated medium's
|
---|
359 | # objects will be closed when the corresponding Machine object ('oVM') is
|
---|
360 | # uninitialized. Assigning the 'None' object to 'oVM' will cause Python to delete
|
---|
361 | # the object when the garbage collector runs however this can take several seconds
|
---|
362 | # so we invoke the Python garbage collector manually here so we don't have to wait.
|
---|
363 | try:
|
---|
364 | gc.collect();
|
---|
365 | except:
|
---|
366 | reporter.logXcpt();
|
---|
367 |
|
---|
368 | reporter.log('Waiting three seconds for Machine::uninit() to be called to close the attached disks');
|
---|
369 | # Fudge factor: Machine::uninit() will be invoked when the oVM object is processed
|
---|
370 | # by the garbage collector above but it may take a few moments to close up to 200
|
---|
371 | # snapshots.
|
---|
372 | time.sleep(3);
|
---|
373 |
|
---|
374 | # If there are no base media belonging to this VM in the global IVirtualBox::hardDisks[]
|
---|
375 | # array (expected) then there are no leftover child images either.
|
---|
376 | cNumDisks = self.getNumOfHDsFromHardDisksArray();
|
---|
377 | reporter.log('After unregister(UnregisterOnly): API reports %d base images (should be zero)' % (cNumDisks));
|
---|
378 | if cNumDisks != 0:
|
---|
379 | reporter.error('Got %d base images after unregistering, expected zero (0)' % (cNumDisks));
|
---|
380 |
|
---|
381 | return reporter.testDone()[1] == 0
|
---|
382 |
|
---|
383 |
|
---|
384 | if __name__ == '__main__':
|
---|
385 | sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
---|
386 | from tdApi1 import tdApi1; # pylint: disable=relative-import
|
---|
387 | sys.exit(tdApi1([SubTstDrvTreeDepth1]).main(sys.argv))
|
---|