vboxapi.py@ 100842

Last change on this file since 100842 was 98103, checked in by vboxsync, 2 years ago
Copyright year updates by scm.
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 43.3 KB

Line
1	# -- coding: utf-8 --
2	# $Id: vboxapi.py 98103 2023-01-17 14:15:46Z vboxsync $
3	"""
4	VirtualBox Python API Glue.
5	"""
6
7	__copyright__ = \
8	"""
9	Copyright (C) 2009-2023 Oracle and/or its affiliates.
10
11	This file is part of VirtualBox base platform packages, as
12	available from https://www.virtualbox.org.
13
14	This program is free software; you can redistribute it and/or
15	modify it under the terms of the GNU General Public License
16	as published by the Free Software Foundation, in version 3 of the
17	License.
18
19	This program is distributed in the hope that it will be useful, but
20	WITHOUT ANY WARRANTY; without even the implied warranty of
21	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22	General Public License for more details.
23
24	You should have received a copy of the GNU General Public License
25	along with this program; if not, see <https://www.gnu.org/licenses>.
26
27	The contents of this file may alternatively be used under the terms
28	of the Common Development and Distribution License Version 1.0
29	(CDDL), a copy of it is provided in the "COPYING.CDDL" file included
30	in the VirtualBox distribution, in which case the provisions of the
31	CDDL are applicable instead of those of the GPL.
32
33	You may elect to license modified versions of this file under the
34	terms and conditions of either the GPL or the CDDL or both.
35
36	SPDX-License-Identifier: GPL-3.0-only OR CDDL-1.0
37	"""
38	__version__ = "$Revision: 98103 $"
39
40
41	# Note! To set Python bitness on OSX use 'export VERSIONER_PYTHON_PREFER_32_BIT=yes'
42
43
44	# Standard Python imports.
45	import os
46	import sys
47	import traceback
48
49
50	if sys.version_info >= (3, 0):
51	xrange = range
52	long = int
53
54	#
55	# Globals, environment and sys.path changes.
56	#
57	import platform
58	VBoxBinDir = os.environ.get("VBOX_PROGRAM_PATH", None)
59	VBoxSdkDir = os.environ.get("VBOX_SDK_PATH", None)
60
61	if VBoxBinDir is None:
62	if platform.system() == 'Darwin':
63	VBoxBinDir = '/Applications/VirtualBox.app/Contents/MacOS'
64	else: # Will be set by the installer
65	VBoxBinDir = "%VBOX_INSTALL_PATH%"
66	else:
67	VBoxBinDir = os.path.abspath(VBoxBinDir)
68
69	if VBoxSdkDir is None:
70	if platform.system() == 'Darwin':
71	VBoxSdkDir = '/Applications/VirtualBox.app/Contents/MacOS/sdk'
72	else: # Will be set by the installer
73	VBoxSdkDir = "%VBOX_SDK_PATH%"
74	else:
75	VBoxSdkDir = os.path.abspath(VBoxSdkDir)
76
77	os.environ["VBOX_PROGRAM_PATH"] = VBoxBinDir
78	os.environ["VBOX_SDK_PATH"] = VBoxSdkDir
79	sys.path.append(VBoxBinDir)
80
81
82	#
83	# Import the generated VirtualBox constants.
84	#
85	from .VirtualBox_constants import VirtualBoxReflectionInfo
86
87
88	class PerfCollector(object):
89	""" This class provides a wrapper over IPerformanceCollector in order to
90	get more 'pythonic' interface.
91
92	To begin collection of metrics use setup() method.
93
94	To get collected data use query() method.
95
96	It is possible to disable metric collection without changing collection
97	parameters with disable() method. The enable() method resumes metric
98	collection.
99	"""
100
101	def __init__(self, mgr, vbox):
102	""" Initializes the instance.
103
104	"""
105	self.mgr = mgr
106	self.isMscom = (mgr.type == 'MSCOM')
107	self.collector = vbox.performanceCollector
108
109	def setup(self, names, objects, period, nsamples):
110	""" Discards all previously collected values for the specified
111	metrics, sets the period of collection and the number of retained
112	samples, enables collection.
113	"""
114	self.collector.setupMetrics(names, objects, period, nsamples)
115
116	def enable(self, names, objects):
117	""" Resumes metric collection for the specified metrics.
118	"""
119	self.collector.enableMetrics(names, objects)
120
121	def disable(self, names, objects):
122	""" Suspends metric collection for the specified metrics.
123	"""
124	self.collector.disableMetrics(names, objects)
125
126	def query(self, names, objects):
127	""" Retrieves collected metric values as well as some auxiliary
128	information. Returns an array of dictionaries, one dictionary per
129	metric. Each dictionary contains the following entries:
130	'name': metric name
131	'object': managed object this metric associated with
132	'unit': unit of measurement
133	'scale': divide 'values' by this number to get float numbers
134	'values': collected data
135	'values_as_string': pre-processed values ready for 'print' statement
136	"""
137	# Get around the problem with input arrays returned in output
138	# parameters (see #3953) for MSCOM.
139	if self.isMscom:
140	(values, names, objects, names_out, objects_out, units, scales, sequence_numbers,
141	indices, lengths) = self.collector.queryMetricsData(names, objects)
142	else:
143	(values, names_out, objects_out, units, scales, sequence_numbers,
144	indices, lengths) = self.collector.queryMetricsData(names, objects)
145	out = []
146	for i in xrange(0, len(names_out)):
147	scale = int(scales[i])
148	if scale != 1:
149	fmt = '%.2f%s'
150	else:
151	fmt = '%d %s'
152	out.append({
153	'name': str(names_out[i]),
154	'object': str(objects_out[i]),
155	'unit': str(units[i]),
156	'scale': scale,
157	'values': [int(values[j]) for j in xrange(int(indices[i]), int(indices[i]) + int(lengths[i]))],
158	'values_as_string': '[' + ', '.join([fmt % (int(values[j]) / scale, units[i]) for j in
159	xrange(int(indices[i]), int(indices[i]) + int(lengths[i]))]) + ']'
160	})
161	return out
162
163
164	#
165	# Attribute hacks.
166	#
167	def ComifyName(name):
168	return name[0].capitalize() + name[1:]
169
170
171	## This is for saving the original DispatchBaseClass __getattr__ and __setattr__
172	# method references.
173	_g_dCOMForward = {
174	'getattr': None,
175	'setattr': None,
176	}
177
178
179	def _CustomGetAttr(self, sAttr):
180	""" Our getattr replacement for DispatchBaseClass. """
181	# Fastpath.
182	oRet = self.__class__.__dict__.get(sAttr)
183	if oRet is not None:
184	return oRet
185
186	# Try case-insensitivity workaround for class attributes (COM methods).
187	sAttrLower = sAttr.lower()
188	for k in list(self.__class__.__dict__.keys()):
189	if k.lower() == sAttrLower:
190	setattr(self.__class__, sAttr, self.__class__.__dict__[k])
191	return getattr(self, k)
192
193	# Slow path.
194	try:
195	return _g_dCOMForward['getattr'](self, ComifyName(sAttr))
196	except AttributeError:
197	return _g_dCOMForward['getattr'](self, sAttr)
198
199
200	def _CustomSetAttr(self, sAttr, oValue):
201	""" Our setattr replacement for DispatchBaseClass. """
202	try:
203	return _g_dCOMForward['setattr'](self, ComifyName(sAttr), oValue)
204	except AttributeError:
205	return _g_dCOMForward['setattr'](self, sAttr, oValue)
206
207
208	class PlatformBase(object):
209	"""
210	Base class for the platform specific code.
211	"""
212
213	def __init__(self, aoParams):
214	_ = aoParams
215
216	def getVirtualBox(self):
217	"""
218	Gets a the IVirtualBox singleton.
219	"""
220	return None
221
222	def getSessionObject(self):
223	"""
224	Get a session object that can be used for opening machine sessions.
225
226	The oIVBox parameter is an getVirtualBox() return value, i.e. an
227	IVirtualBox reference.
228
229	See also openMachineSession.
230	"""
231	return None
232
233	def getType(self):
234	""" Returns the platform type (class name sans 'Platform'). """
235	return None
236
237	def isRemote(self):
238	"""
239	Returns True if remote (web services) and False if local (COM/XPCOM).
240	"""
241	return False
242
243	def getArray(self, oInterface, sAttrib):
244	"""
245	Retrives the value of the array attribute 'sAttrib' from
246	interface 'oInterface'.
247
248	This is for hiding platform specific differences in attributes
249	returning arrays.
250	"""
251	_ = oInterface
252	_ = sAttrib
253	return None
254
255	def setArray(self, oInterface, sAttrib, aoArray):
256	"""
257	Sets the value (aoArray) of the array attribute 'sAttrib' in
258	interface 'oInterface'.
259
260	This is for hiding platform specific differences in attributes
261	setting arrays.
262	"""
263	_ = oInterface
264	_ = sAttrib
265	_ = aoArray
266	return None
267
268	def initPerThread(self):
269	"""
270	Does backend specific initialization for the calling thread.
271	"""
272	return True
273
274	def deinitPerThread(self):
275	"""
276	Does backend specific uninitialization for the calling thread.
277	"""
278	return True
279
280	def createListener(self, oImplClass, dArgs):
281	"""
282	Instantiates and wraps an active event listener class so it can be
283	passed to an event source for registration.
284
285	oImplClass is a class (type, not instance) which implements
286	IEventListener.
287
288	dArgs is a dictionary with string indexed variables. This may be
289	modified by the method to pass platform specific parameters. Can
290	be None.
291
292	This currently only works on XPCOM. COM support is not possible due to
293	shortcuts taken in the COM bridge code, which is not under our control.
294	Use passive listeners for COM and web services.
295	"""
296	_ = oImplClass
297	_ = dArgs
298	raise Exception("No active listeners for this platform")
299
300	def waitForEvents(self, cMsTimeout):
301	"""
302	Wait for events to arrive and process them.
303
304	The timeout (cMsTimeout) is in milliseconds for how long to wait for
305	events to arrive. A negative value means waiting for ever, while 0
306	does not wait at all.
307
308	Returns 0 if events was processed.
309	Returns 1 if timed out or interrupted in some way.
310	Returns 2 on error (like not supported for web services).
311
312	Raises an exception if the calling thread is not the main thread (the one
313	that initialized VirtualBoxManager) or if the time isn't an integer.
314	"""
315	_ = cMsTimeout
316	return 2
317
318	def interruptWaitEvents(self):
319	"""
320	Interrupt a waitForEvents call.
321	This is normally called from a worker thread to wake up the main thread.
322
323	Returns True on success, False on failure.
324	"""
325	return False
326
327	def deinit(self):
328	"""
329	Unitializes the platform specific backend.
330	"""
331	return None
332
333	def queryInterface(self, oIUnknown, sClassName):
334	"""
335	IUnknown::QueryInterface wrapper.
336
337	oIUnknown is who to ask.
338	sClassName is the name of the interface we're asking for.
339	"""
340	return None
341
342	#
343	# Error (exception) access methods.
344	#
345
346	def xcptGetStatus(self, oXcpt):
347	"""
348	Returns the COM status code from the VBox API given exception.
349	"""
350	return None
351
352	def xcptIsDeadInterface(self, oXcpt):
353	"""
354	Returns True if the exception indicates that the interface is dead, False if not.
355	"""
356	return False
357
358	def xcptIsEqual(self, oXcpt, hrStatus):
359	"""
360	Checks if the exception oXcpt is equal to the COM/XPCOM status code
361	hrStatus.
362
363	The oXcpt parameter can be any kind of object, we'll just return True
364	if it doesn't behave like a our exception class.
365
366	Will not raise any exception as long as hrStatus and self are not bad.
367	"""
368	try:
369	hrXcpt = self.xcptGetStatus(oXcpt)
370	except AttributeError:
371	return False
372	if hrXcpt == hrStatus:
373	return True
374
375	# Fudge for 32-bit signed int conversion.
376	if 0x7fffffff < hrStatus <= 0xffffffff and hrXcpt < 0:
377	if (hrStatus - 0x100000000) == hrXcpt:
378	return True
379	return False
380
381	def xcptGetMessage(self, oXcpt):
382	"""
383	Returns the best error message found in the COM-like exception.
384	Returns None to fall back on xcptToString.
385	Raises exception if oXcpt isn't our kind of exception object.
386	"""
387	return None
388
389	def xcptGetBaseXcpt(self):
390	"""
391	Returns the base exception class.
392	"""
393	return None
394
395	def xcptSetupConstants(self, oDst):
396	"""
397	Copy/whatever all error constants onto oDst.
398	"""
399	return oDst
400
401	@staticmethod
402	def xcptCopyErrorConstants(oDst, oSrc):
403	"""
404	Copy everything that looks like error constants from oDst to oSrc.
405	"""
406	for sAttr in dir(oSrc):
407	if sAttr[0].isupper() and (sAttr[1].isupper() or sAttr[1] == '_'):
408	oAttr = getattr(oSrc, sAttr)
409	if type(oAttr) is int:
410	setattr(oDst, sAttr, oAttr)
411	return oDst
412
413
414	class PlatformMSCOM(PlatformBase):
415	"""
416	Platform specific code for MS COM.
417	"""
418
419	## @name VirtualBox COM Typelib definitions (should be generate)
420	#
421	# @remarks Must be updated when the corresponding VirtualBox.xidl bits
422	# are changed. Fortunately this isn't very often.
423	# @{
424	VBOX_TLB_GUID = '{D7569351-1750-46F0-936E-BD127D5BC264}'
425	VBOX_TLB_LCID = 0
426	VBOX_TLB_MAJOR = 1
427	VBOX_TLB_MINOR = 3
428	## @}
429
430	def __init__(self, dParams):
431	PlatformBase.__init__(self, dParams)
432
433	#
434	# Since the code runs on all platforms, we have to do a lot of
435	# importing here instead of at the top of the file where it's normally located.
436	#
437	from win32com import universal
438	from win32com.client import gencache, DispatchBaseClass
439	from win32com.client import constants, getevents
440	import win32com
441	import pythoncom
442	import win32api
443	import winerror
444	from win32con import DUPLICATE_SAME_ACCESS
445	from win32api import GetCurrentThread, GetCurrentThreadId, DuplicateHandle, GetCurrentProcess
446	import threading
447
448	self.winerror = winerror
449
450	# Setup client impersonation in COM calls.
451	try:
452	pythoncom.CoInitializeSecurity(None,
453	None,
454	None,
455	pythoncom.RPC_C_AUTHN_LEVEL_DEFAULT,
456	pythoncom.RPC_C_IMP_LEVEL_IMPERSONATE,
457	None,
458	pythoncom.EOAC_NONE,
459	None)
460	except:
461	_, oXcpt, _ = sys.exc_info();
462	if isinstance(oXcpt, pythoncom.com_error) and self.xcptGetStatus(oXcpt) == -2147417831: # RPC_E_TOO_LATE
463	print("Warning: CoInitializeSecurity was already called");
464	else:
465	print("Warning: CoInitializeSecurity failed: ", oXcpt);
466
467	# Remember this thread ID and get its handle so we can wait on it in waitForEvents().
468	self.tid = GetCurrentThreadId()
469	pid = GetCurrentProcess()
470	self.aoHandles = [DuplicateHandle(pid, GetCurrentThread(), pid, 0, 0, DUPLICATE_SAME_ACCESS),] # type: list[PyHANDLE]
471
472	# Hack the COM dispatcher base class so we can modify method and
473	# attribute names to match those in xpcom.
474	if _g_dCOMForward['setattr'] is None:
475	_g_dCOMForward['getattr'] = DispatchBaseClass.__dict__['__getattr__']
476	_g_dCOMForward['setattr'] = DispatchBaseClass.__dict__['__setattr__']
477	setattr(DispatchBaseClass, '__getattr__', _CustomGetAttr)
478	setattr(DispatchBaseClass, '__setattr__', _CustomSetAttr)
479
480	# Hack the exception base class so the users doesn't need to check for
481	# XPCOM or COM and do different things.
482	## @todo
483
484	#
485	# Make sure the gencache is correct (we don't quite follow the COM
486	# versioning rules).
487	#
488	self.flushGenPyCache(win32com.client.gencache)
489	win32com.client.gencache.EnsureDispatch('VirtualBox.Session')
490	win32com.client.gencache.EnsureDispatch('VirtualBox.VirtualBox')
491	win32com.client.gencache.EnsureDispatch('VirtualBox.VirtualBoxClient')
492
493	self.oClient = None ##< instance of client used to support lifetime of VBoxSDS
494	self.oIntCv = threading.Condition()
495	self.fInterrupted = False
496
497	_ = dParams
498
499	def flushGenPyCache(self, oGenCache):
500	"""
501	Flushes VBox related files in the win32com gen_py cache.
502
503	This is necessary since we don't follow the typelib versioning rules
504	that everyeone else seems to subscribe to.
505	"""
506	#
507	# The EnsureModule method have broken validation code, it doesn't take
508	# typelib module directories into account. So we brute force them here.
509	# (It's possible the directory approach is from some older pywin
510	# version or the result of runnig makepy or gencache manually, but we
511	# need to cover it as well.)
512	#
513	sName = oGenCache.GetGeneratedFileName(self.VBOX_TLB_GUID, self.VBOX_TLB_LCID,
514	self.VBOX_TLB_MAJOR, self.VBOX_TLB_MINOR)
515	sGenPath = oGenCache.GetGeneratePath()
516	if len(sName) > 36 and len(sGenPath) > 5:
517	sTypelibPath = os.path.join(sGenPath, sName)
518	if os.path.isdir(sTypelibPath):
519	import shutil
520	shutil.rmtree(sTypelibPath, ignore_errors=True)
521
522	#
523	# Ensure that our typelib is valid.
524	#
525	return oGenCache.EnsureModule(self.VBOX_TLB_GUID, self.VBOX_TLB_LCID, self.VBOX_TLB_MAJOR, self.VBOX_TLB_MINOR)
526
527	def getSessionObject(self):
528	import win32com
529	from win32com.client import Dispatch
530	return win32com.client.Dispatch("VirtualBox.Session")
531
532	def getVirtualBox(self):
533	# Caching self.oClient is the trick for SDS. It allows to keep the
534	# VBoxSDS in the memory until the end of PlatformMSCOM lifetme.
535	if self.oClient is None:
536	import win32com
537	from win32com.client import Dispatch
538	self.oClient = win32com.client.Dispatch("VirtualBox.VirtualBoxClient")
539	return self.oClient.virtualBox
540
541	def getType(self):
542	return 'MSCOM'
543
544	def getArray(self, oInterface, sAttrib):
545	return oInterface.__getattr__(sAttrib)
546
547	def setArray(self, oInterface, sAttrib, aoArray):
548	#
549	# HACK ALERT!
550	#
551	# With pywin32 build 218, we're seeing type mismatch errors here for
552	# IGuestSession::environmentChanges (safearray of BSTRs). The Dispatch
553	# object (_oleobj_) seems to get some type conversion wrong and COM
554	# gets upset. So, we redo some of the dispatcher work here, picking
555	# the missing type information from the getter.
556	#
557	oOleObj = getattr(oInterface, '_oleobj_')
558	aPropMapGet = getattr(oInterface, '_prop_map_get_')
559	aPropMapPut = getattr(oInterface, '_prop_map_put_')
560	sComAttrib = sAttrib if sAttrib in aPropMapGet else ComifyName(sAttrib)
561	try:
562	aArgs, aDefaultArgs = aPropMapPut[sComAttrib]
563	aGetArgs = aPropMapGet[sComAttrib]
564	except KeyError: # fallback.
565	return oInterface.__setattr__(sAttrib, aoArray)
566
567	import pythoncom
568	oOleObj.InvokeTypes(aArgs[0], # dispid
569	aArgs[1], # LCID
570	aArgs[2], # DISPATCH_PROPERTYPUT
571	(pythoncom.VT_HRESULT, 0), # retType - or void?
572	(aGetArgs[2],), # argTypes - trick: we get the type from the getter.
573	aoArray,) # The array
574
575	def initPerThread(self):
576	import pythoncom
577	pythoncom.CoInitializeEx(0)
578
579	def deinitPerThread(self):
580	import pythoncom
581	pythoncom.CoUninitialize()
582
583	def createListener(self, oImplClass, dArgs):
584	if True:
585	raise Exception('no active listeners on Windows as PyGatewayBase::QueryInterface() '
586	'returns new gateway objects all the time, thus breaking EventQueue '
587	'assumptions about the listener interface pointer being constants between calls ')
588	# Did this code ever really work?
589	d = {}
590	d['BaseClass'] = oImplClass
591	d['dArgs'] = dArgs
592	d['tlb_guid'] = PlatformMSCOM.VBOX_TLB_GUID
593	d['tlb_major'] = PlatformMSCOM.VBOX_TLB_MAJOR
594	d['tlb_minor'] = PlatformMSCOM.VBOX_TLB_MINOR
595	str_ = ""
596	str_ += "import win32com.server.util\n"
597	str_ += "import pythoncom\n"
598
599	str_ += "class ListenerImpl(BaseClass):\n"
600	str_ += " _com_interfaces_ = ['IEventListener']\n"
601	str_ += " _typelib_guid_ = tlb_guid\n"
602	str_ += " _typelib_version_ = tlb_major, tlb_minor\n"
603	str_ += " _reg_clsctx_ = pythoncom.CLSCTX_INPROC_SERVER\n"
604	# Maybe we'd better implement Dynamic invoke policy, to be more flexible here
605	str_ += " _reg_policy_spec_ = 'win32com.server.policy.EventHandlerPolicy'\n"
606
607	# capitalized version of listener method
608	str_ += " HandleEvent=BaseClass.handleEvent\n"
609	str_ += " def __init__(self): BaseClass.__init__(self, dArgs)\n"
610	str_ += "result = win32com.server.util.wrap(ListenerImpl())\n"
611	exec(str_, d, d)
612	return d['result']
613
614	def waitForEvents(self, timeout):
615	from win32api import GetCurrentThreadId
616	from win32event import INFINITE
617	from win32event import MsgWaitForMultipleObjects, QS_ALLINPUT, WAIT_TIMEOUT, WAIT_OBJECT_0
618	from pythoncom import PumpWaitingMessages
619	import types
620
621	if not isinstance(timeout, int):
622	raise TypeError("The timeout argument is not an integer")
623	if self.tid != GetCurrentThreadId():
624	raise Exception("wait for events from the same thread you inited!")
625
626	if timeout < 0:
627	cMsTimeout = INFINITE
628	else:
629	cMsTimeout = timeout
630	rc = MsgWaitForMultipleObjects(self.aoHandles, 0, cMsTimeout, QS_ALLINPUT)
631	if WAIT_OBJECT_0 <= rc < WAIT_OBJECT_0 + len(self.aoHandles):
632	# is it possible?
633	rc = 2
634	elif rc == WAIT_OBJECT_0 + len(self.aoHandles):
635	# Waiting messages
636	PumpWaitingMessages()
637	rc = 0
638	else:
639	# Timeout
640	rc = 1
641
642	# check for interruption
643	self.oIntCv.acquire()
644	if self.fInterrupted:
645	self.fInterrupted = False
646	rc = 1
647	self.oIntCv.release()
648
649	return rc
650
651	def interruptWaitEvents(self):
652	"""
653	Basically a python implementation of NativeEventQueue::postEvent().
654
655	The magic value must be in sync with the C++ implementation or this
656	won't work.
657
658	Note that because of this method we cannot easily make use of a
659	non-visible Window to handle the message like we would like to do.
660	"""
661	from win32api import PostThreadMessage
662	from win32con import WM_USER
663
664	self.oIntCv.acquire()
665	self.fInterrupted = True
666	self.oIntCv.release()
667	try:
668	PostThreadMessage(self.tid, WM_USER, None, 0xf241b819)
669	except:
670	return False
671	return True
672
673	def deinit(self):
674	for oHandle in self.aoHandles:
675	if oHandle is not None:
676	oHandle.Close();
677	self.oHandle = None;
678
679	del self.oClient;
680	self.oClient = None;
681
682	# This non-sense doesn't pair up with any pythoncom.CoInitialize[Ex].
683	# See @bugref{9037}.
684	#import pythoncom
685	#pythoncom.CoUninitialize()
686
687	def queryInterface(self, oIUnknown, sClassName):
688	from win32com.client import CastTo
689	return CastTo(oIUnknown, sClassName)
690
691	def xcptGetStatus(self, oXcpt):
692	# The DISP_E_EXCEPTION + excptinfo fun needs checking up, only
693	# empirical info on it so far.
694	hrXcpt = oXcpt.hresult
695	if hrXcpt == self.winerror.DISP_E_EXCEPTION:
696	try:
697	hrXcpt = oXcpt.excepinfo[5]
698	except:
699	pass
700	return hrXcpt
701
702	def xcptIsDeadInterface(self, oXcpt):
703	return self.xcptGetStatus(oXcpt) in [
704	0x800706ba, -2147023174, # RPC_S_SERVER_UNAVAILABLE.
705	0x800706be, -2147023170, # RPC_S_CALL_FAILED.
706	0x800706bf, -2147023169, # RPC_S_CALL_FAILED_DNE.
707	0x80010108, -2147417848, # RPC_E_DISCONNECTED.
708	0x800706b5, -2147023179, # RPC_S_UNKNOWN_IF
709	]
710
711	def xcptGetMessage(self, oXcpt):
712	if hasattr(oXcpt, 'excepinfo'):
713	try:
714	if len(oXcpt.excepinfo) >= 3:
715	sRet = oXcpt.excepinfo[2]
716	if len(sRet) > 0:
717	return sRet[0:]
718	except:
719	pass
720	if hasattr(oXcpt, 'strerror'):
721	try:
722	sRet = oXcpt.strerror
723	if len(sRet) > 0:
724	return sRet
725	except:
726	pass
727	return None
728
729	def xcptGetBaseXcpt(self):
730	import pythoncom
731
732	return pythoncom.com_error
733
734	def xcptSetupConstants(self, oDst):
735	import winerror
736
737	oDst = self.xcptCopyErrorConstants(oDst, winerror)
738
739	# XPCOM compatability constants.
740	oDst.NS_OK = oDst.S_OK
741	oDst.NS_ERROR_FAILURE = oDst.E_FAIL
742	oDst.NS_ERROR_ABORT = oDst.E_ABORT
743	oDst.NS_ERROR_NULL_POINTER = oDst.E_POINTER
744	oDst.NS_ERROR_NO_INTERFACE = oDst.E_NOINTERFACE
745	oDst.NS_ERROR_INVALID_ARG = oDst.E_INVALIDARG
746	oDst.NS_ERROR_OUT_OF_MEMORY = oDst.E_OUTOFMEMORY
747	oDst.NS_ERROR_NOT_IMPLEMENTED = oDst.E_NOTIMPL
748	oDst.NS_ERROR_UNEXPECTED = oDst.E_UNEXPECTED
749	return oDst
750
751
752	class PlatformXPCOM(PlatformBase):
753	"""
754	Platform specific code for XPCOM.
755	"""
756
757	def __init__(self, dParams):
758	PlatformBase.__init__(self, dParams)
759	sys.path.append(VBoxSdkDir + '/bindings/xpcom/python/')
760	import xpcom.vboxxpcom
761	import xpcom
762	import xpcom.components
763	_ = dParams
764
765	def getSessionObject(self):
766	import xpcom.components
767	return xpcom.components.classes["@virtualbox.org/Session;1"].createInstance()
768
769	def getVirtualBox(self):
770	import xpcom.components
771	client = xpcom.components.classes["@virtualbox.org/VirtualBoxClient;1"].createInstance()
772	return client.virtualBox
773
774	def getType(self):
775	return 'XPCOM'
776
777	def getArray(self, oInterface, sAttrib):
778	return oInterface.__getattr__('get' + ComifyName(sAttrib))()
779
780	def setArray(self, oInterface, sAttrib, aoArray):
781	return oInterface.__getattr__('set' + ComifyName(sAttrib))(aoArray)
782
783	def initPerThread(self):
784	import xpcom
785	xpcom._xpcom.AttachThread()
786
787	def deinitPerThread(self):
788	import xpcom
789	xpcom._xpcom.DetachThread()
790
791	def createListener(self, oImplClass, dArgs):
792	d = {}
793	d['BaseClass'] = oImplClass
794	d['dArgs'] = dArgs
795	str = ""
796	str += "import xpcom.components\n"
797	str += "class ListenerImpl(BaseClass):\n"
798	str += " _com_interfaces_ = xpcom.components.interfaces.IEventListener\n"
799	str += " def __init__(self): BaseClass.__init__(self, dArgs)\n"
800	str += "result = ListenerImpl()\n"
801	exec(str, d, d)
802	return d['result']
803
804	def waitForEvents(self, timeout):
805	import xpcom
806	return xpcom._xpcom.WaitForEvents(timeout)
807
808	def interruptWaitEvents(self):
809	import xpcom
810	return xpcom._xpcom.InterruptWait()
811
812	def deinit(self):
813	import xpcom
814	xpcom._xpcom.DeinitCOM()
815
816	def queryInterface(self, oIUnknown, sClassName):
817	import xpcom.components
818	return oIUnknown.queryInterface(getattr(xpcom.components.interfaces, sClassName))
819
820	def xcptGetStatus(self, oXcpt):
821	return oXcpt.errno
822
823	def xcptIsDeadInterface(self, oXcpt):
824	return self.xcptGetStatus(oXcpt) in [
825	0x80004004, -2147467260, # NS_ERROR_ABORT
826	0x800706be, -2147023170, # NS_ERROR_CALL_FAILED (RPC_S_CALL_FAILED)
827	]
828
829	def xcptGetMessage(self, oXcpt):
830	if hasattr(oXcpt, 'msg'):
831	try:
832	sRet = oXcpt.msg
833	if len(sRet) > 0:
834	return sRet
835	except:
836	pass
837	return None
838
839	def xcptGetBaseXcpt(self):
840	import xpcom
841	return xpcom.Exception
842
843	def xcptSetupConstants(self, oDst):
844	import xpcom
845	oDst = self.xcptCopyErrorConstants(oDst, xpcom.nsError)
846
847	# COM compatability constants.
848	oDst.E_ACCESSDENIED = -2147024891 # see VBox/com/defs.h
849	oDst.S_OK = oDst.NS_OK
850	oDst.E_FAIL = oDst.NS_ERROR_FAILURE
851	oDst.E_ABORT = oDst.NS_ERROR_ABORT
852	oDst.E_POINTER = oDst.NS_ERROR_NULL_POINTER
853	oDst.E_NOINTERFACE = oDst.NS_ERROR_NO_INTERFACE
854	oDst.E_INVALIDARG = oDst.NS_ERROR_INVALID_ARG
855	oDst.E_OUTOFMEMORY = oDst.NS_ERROR_OUT_OF_MEMORY
856	oDst.E_NOTIMPL = oDst.NS_ERROR_NOT_IMPLEMENTED
857	oDst.E_UNEXPECTED = oDst.NS_ERROR_UNEXPECTED
858	oDst.DISP_E_EXCEPTION = -2147352567 # For COM compatability only.
859	return oDst
860
861
862	class PlatformWEBSERVICE(PlatformBase):
863	"""
864	VirtualBox Web Services API specific code.
865	"""
866
867	def __init__(self, dParams):
868	PlatformBase.__init__(self, dParams)
869	# Import web services stuff. Fix the sys.path the first time.
870	sWebServLib = os.path.join(VBoxSdkDir, 'bindings', 'webservice', 'python', 'lib')
871	if sWebServLib not in sys.path:
872	sys.path.append(sWebServLib)
873	import VirtualBox_wrappers
874	from VirtualBox_wrappers import IWebsessionManager2
875
876	# Initialize instance variables from parameters.
877	if dParams is not None:
878	self.user = dParams.get("user", "")
879	self.password = dParams.get("password", "")
880	self.url = dParams.get("url", "")
881	else:
882	self.user = ""
883	self.password = ""
884	self.url = None
885	self.vbox = None
886	self.wsmgr = None
887
888	#
889	# Base class overrides.
890	#
891
892	def getSessionObject(self):
893	return self.wsmgr.getSessionObject(self.vbox)
894
895	def getVirtualBox(self):
896	return self.connect(self.url, self.user, self.password)
897
898	def getType(self):
899	return 'WEBSERVICE'
900
901	def isRemote(self):
902	""" Returns True if remote VBox host, False if local. """
903	return True
904
905	def getArray(self, oInterface, sAttrib):
906	return oInterface.__getattr__(sAttrib)
907
908	def setArray(self, oInterface, sAttrib, aoArray):
909	return oInterface.__setattr__(sAttrib, aoArray)
910
911	def waitForEvents(self, timeout):
912	# Webservices cannot do that yet
913	return 2
914
915	def interruptWaitEvents(self, timeout):
916	# Webservices cannot do that yet
917	return False
918
919	def deinit(self):
920	try:
921	self.disconnect()
922	except:
923	pass
924
925	def queryInterface(self, oIUnknown, sClassName):
926	d = {}
927	d['oIUnknown'] = oIUnknown
928	str = ""
929	str += "from VirtualBox_wrappers import " + sClassName + "\n"
930	str += "result = " + sClassName + "(oIUnknown.mgr, oIUnknown.handle)\n"
931	# wrong, need to test if class indeed implements this interface
932	exec(str, d, d)
933	return d['result']
934
935	#
936	# Web service specific methods.
937	#
938
939	def connect(self, url, user, passwd):
940	if self.vbox is not None:
941	self.disconnect()
942	from VirtualBox_wrappers import IWebsessionManager2
943
944	if url is None:
945	url = ""
946	self.url = url
947	if user is None:
948	user = ""
949	self.user = user
950	if passwd is None:
951	passwd = ""
952	self.password = passwd
953	self.wsmgr = IWebsessionManager2(self.url)
954	self.vbox = self.wsmgr.logon(self.user, self.password)
955	if not self.vbox.handle:
956	raise Exception("cannot connect to '" + self.url + "' as '" + self.user + "'")
957	return self.vbox
958
959	def disconnect(self):
960	if self.vbox is not None and self.wsmgr is not None:
961	self.wsmgr.logoff(self.vbox)
962	self.vbox = None
963	self.wsmgr = None
964
965
966	## The current (last) exception class.
967	# This is reinitalized whenever VirtualBoxManager is called, so it will hold
968	# the reference to the error exception class for the last platform/style that
969	# was used. Most clients does talk to multiple VBox instance on different
970	# platforms at the same time, so this should be sufficent for most uses and
971	# be way simpler to use than VirtualBoxManager::oXcptClass.
972	CurXcptClass = None
973
974
975	class VirtualBoxManager(object):
976	"""
977	VirtualBox API manager class.
978
979	The API users will have to instantiate this. If no parameters are given,
980	it will default to interface with the VirtualBox running on the local
981	machine. sStyle can be None (default), MSCOM, XPCOM or WEBSERVICES. Most
982	users will either be specifying None or WEBSERVICES.
983
984	The dPlatformParams is an optional dictionary for passing parameters to the
985	WEBSERVICE backend.
986	"""
987
988	class Statuses(object):
989	def __init__(self):
990	pass
991
992	def __init__(self, sStyle=None, dPlatformParams=None):
993	if sStyle is None:
994	if sys.platform == 'win32':
995	sStyle = "MSCOM"
996	else:
997	sStyle = "XPCOM"
998	if sStyle == 'XPCOM':
999	self.platform = PlatformXPCOM(dPlatformParams)
1000	elif sStyle == 'MSCOM':
1001	self.platform = PlatformMSCOM(dPlatformParams)
1002	elif sStyle == 'WEBSERVICE':
1003	self.platform = PlatformWEBSERVICE(dPlatformParams)
1004	else:
1005	raise Exception('Unknown sStyle=%s' % (sStyle,))
1006	self.style = sStyle
1007	self.type = self.platform.getType()
1008	self.remote = self.platform.isRemote()
1009	## VirtualBox API constants (for webservices, enums are symbolic).
1010	self.constants = VirtualBoxReflectionInfo(sStyle == "WEBSERVICE")
1011
1012	## Status constants.
1013	self.statuses = self.platform.xcptSetupConstants(VirtualBoxManager.Statuses())
1014	## @todo Add VBOX_E_XXX to statuses? They're already in constants...
1015	## Dictionary for errToString, built on demand.
1016	self._dErrorValToName = None
1017
1018	## Dictionary for resolving enum values to names, two levels of dictionaries.
1019	## First level is indexed by enum name, the next by value.
1020	self._ddEnumValueToName = {};
1021
1022	## The exception class for the selected platform.
1023	self.oXcptClass = self.platform.xcptGetBaseXcpt()
1024	global CurXcptClass
1025	CurXcptClass = self.oXcptClass
1026
1027	# Get the virtualbox singleton.
1028	try:
1029	vbox = self.platform.getVirtualBox()
1030	except NameError:
1031	print("Installation problem: check that appropriate libs in place")
1032	traceback.print_exc()
1033	raise
1034	except Exception:
1035	_, e, _ = sys.exc_info()
1036	print("init exception: ", e)
1037	traceback.print_exc()
1038
1039	def __del__(self):
1040	self.deinit()
1041
1042	def getPythonApiRevision(self):
1043	"""
1044	Returns a Python API revision number.
1045	This will be incremented when features are added to this file.
1046	"""
1047	return 3
1048
1049	@property
1050	def mgr(self):
1051	"""
1052	This used to be an attribute referring to a session manager class with
1053	only one method called getSessionObject. It moved into this class.
1054	"""
1055	return self
1056
1057	#
1058	# Wrappers for self.platform methods.
1059	#
1060	def getVirtualBox(self):
1061	""" See PlatformBase::getVirtualBox(). """
1062	return self.platform.getVirtualBox()
1063
1064	def getSessionObject(self, oIVBox = None):
1065	""" See PlatformBase::getSessionObject(). """
1066	# ignore parameter which was never needed
1067	_ = oIVBox
1068	return self.platform.getSessionObject()
1069
1070	def getArray(self, oInterface, sAttrib):
1071	""" See PlatformBase::getArray(). """
1072	return self.platform.getArray(oInterface, sAttrib)
1073
1074	def setArray(self, oInterface, sAttrib, aoArray):
1075	""" See PlatformBase::setArray(). """
1076	return self.platform.setArray(oInterface, sAttrib, aoArray)
1077
1078	def createListener(self, oImplClass, dArgs=None):
1079	""" See PlatformBase::createListener(). """
1080	return self.platform.createListener(oImplClass, dArgs)
1081
1082	def waitForEvents(self, cMsTimeout):
1083	""" See PlatformBase::waitForEvents(). """
1084	return self.platform.waitForEvents(cMsTimeout)
1085
1086	def interruptWaitEvents(self):
1087	""" See PlatformBase::interruptWaitEvents(). """
1088	return self.platform.interruptWaitEvents()
1089
1090	def queryInterface(self, oIUnknown, sClassName):
1091	""" See PlatformBase::queryInterface(). """
1092	return self.platform.queryInterface(oIUnknown, sClassName)
1093
1094	#
1095	# Init and uninit.
1096	#
1097	def initPerThread(self):
1098	""" See PlatformBase::deinitPerThread(). """
1099	self.platform.initPerThread()
1100
1101	def deinitPerThread(self):
1102	""" See PlatformBase::deinitPerThread(). """
1103	return self.platform.deinitPerThread()
1104
1105	def deinit(self):
1106	"""
1107	For unitializing the manager.
1108	Do not access it after calling this method.
1109	"""
1110	if hasattr(self, "platform") and self.platform is not None:
1111	self.platform.deinit()
1112	self.platform = None
1113	return True
1114
1115	#
1116	# Utility methods.
1117	#
1118	def openMachineSession(self, oIMachine, fPermitSharing=True):
1119	"""
1120	Attempts to open the a session to the machine.
1121	Returns a session object on success.
1122	Raises exception on failure.
1123	"""
1124	oSession = self.getSessionObject()
1125	if fPermitSharing:
1126	eType = self.constants.LockType_Shared
1127	else:
1128	eType = self.constants.LockType_Write
1129	oIMachine.lockMachine(oSession, eType)
1130	return oSession
1131
1132	def closeMachineSession(self, oSession):
1133	"""
1134	Closes a session opened by openMachineSession.
1135	Ignores None parameters.
1136	"""
1137	if oSession is not None:
1138	oSession.unlockMachine()
1139	return True
1140
1141	def getPerfCollector(self, oIVBox):
1142	"""
1143	Returns a helper class (PerfCollector) for accessing performance
1144	collector goodies. See PerfCollector for details.
1145	"""
1146	return PerfCollector(self, oIVBox)
1147
1148	def getBinDir(self):
1149	"""
1150	Returns the VirtualBox binary directory.
1151	"""
1152	global VBoxBinDir
1153	return VBoxBinDir
1154
1155	def getSdkDir(self):
1156	"""
1157	Returns the VirtualBox SDK directory.
1158	"""
1159	global VBoxSdkDir
1160	return VBoxSdkDir
1161
1162	def getEnumValueName(self, sEnumTypeNm, oEnumValue, fTypePrefix = False):
1163	"""
1164	Returns the name (string) for the corresponding enum value.
1165	"""
1166	# Cache lookup:
1167	dValueNames = self._ddEnumValueToName.get(sEnumTypeNm);
1168	if dValueNames is not None:
1169	sValueName = dValueNames.get(oEnumValue);
1170	if sValueName:
1171	return sValueName if not fTypePrefix else '%s_%s' % (sEnumTypeNm, sValueName);
1172	else:
1173	# Cache miss. Build the reverse lookup dictionary and add it to the cache:
1174	dNamedValues = self.constants.all_values(sEnumTypeNm);
1175	if len(dNamedValues) > 0:
1176
1177	dValueNames = dict();
1178	for sName in dNamedValues:
1179	dValueNames[dNamedValues[sName]] = sName;
1180	self._ddEnumValueToName[sEnumTypeNm] = dValueNames;
1181
1182	# Lookup the value:
1183	sValueName = dValueNames.get(oEnumValue);
1184	if sValueName:
1185	return sValueName if not fTypePrefix else '%s_%s' % (sEnumTypeNm, sValueName);
1186
1187	# Fallback:
1188	return '%s_Unknown_%s' % (sEnumTypeNm, oEnumValue);
1189
1190	#
1191	# Error code utilities.
1192	#
1193	## @todo port to webservices!
1194	def xcptGetStatus(self, oXcpt=None):
1195	"""
1196	Gets the status code from an exception. If the exception parameter
1197	isn't specified, the current exception is examined.
1198	"""
1199	if oXcpt is None:
1200	oXcpt = sys.exc_info()[1]
1201	return self.platform.xcptGetStatus(oXcpt)
1202
1203	def xcptIsDeadInterface(self, oXcpt=None):
1204	"""
1205	Returns True if the exception indicates that the interface is dead,
1206	False if not. If the exception parameter isn't specified, the current
1207	exception is examined.
1208	"""
1209	if oXcpt is None:
1210	oXcpt = sys.exc_info()[1]
1211	return self.platform.xcptIsDeadInterface(oXcpt)
1212
1213	def xcptIsOurXcptKind(self, oXcpt=None):
1214	"""
1215	Checks if the exception is one that could come from the VBox API. If
1216	the exception parameter isn't specified, the current exception is
1217	examined.
1218	"""
1219	if self.oXcptClass is None: # @todo find the exception class for web services!
1220	return False
1221	if oXcpt is None:
1222	oXcpt = sys.exc_info()[1]
1223	return isinstance(oXcpt, self.oXcptClass)
1224
1225	def xcptIsEqual(self, oXcpt, hrStatus):
1226	"""
1227	Checks if the exception oXcpt is equal to the COM/XPCOM status code
1228	hrStatus.
1229
1230	The oXcpt parameter can be any kind of object, we'll just return True
1231	if it doesn't behave like a our exception class. If it's None, we'll
1232	query the current exception and examine that.
1233
1234	Will not raise any exception as long as hrStatus and self are not bad.
1235	"""
1236	if oXcpt is None:
1237	oXcpt = sys.exc_info()[1]
1238	return self.platform.xcptIsEqual(oXcpt, hrStatus)
1239
1240	def xcptIsNotEqual(self, oXcpt, hrStatus):
1241	"""
1242	Negated xcptIsEqual.
1243	"""
1244	return not self.xcptIsEqual(oXcpt, hrStatus)
1245
1246	def xcptToString(self, hrStatusOrXcpt=None):
1247	"""
1248	Converts the specified COM status code, or the status code of the
1249	specified exception, to a C constant string. If the parameter isn't
1250	specified (is None), the current exception is examined.
1251	"""
1252
1253	# Deal with exceptions.
1254	if hrStatusOrXcpt is None or self.xcptIsOurXcptKind(hrStatusOrXcpt):
1255	hrStatus = self.xcptGetStatus(hrStatusOrXcpt)
1256	else:
1257	hrStatus = hrStatusOrXcpt
1258
1259	# Build the dictionary on demand.
1260	if self._dErrorValToName is None:
1261	dErrorValToName = dict()
1262	for sKey in dir(self.statuses):
1263	if sKey[0].isupper():
1264	oValue = getattr(self.statuses, sKey)
1265	if isinstance(oValue, (int, long)):
1266	dErrorValToName[int(oValue)] = sKey
1267	# Always prefer the COM names (see aliasing in platform specific code):
1268	for sKey in ('S_OK', 'E_FAIL', 'E_ABORT', 'E_POINTER', 'E_NOINTERFACE', 'E_INVALIDARG',
1269	'E_OUTOFMEMORY', 'E_NOTIMPL', 'E_UNEXPECTED',):
1270	oValue = getattr(self.statuses, sKey, None)
1271	if oValue is not None:
1272	dErrorValToName[oValue] = sKey
1273	self._dErrorValToName = dErrorValToName
1274
1275	# Do the lookup, falling back on formatting the status number.
1276	try:
1277	sStr = self._dErrorValToName[int(hrStatus)]
1278	except KeyError:
1279	hrLong = long(hrStatus)
1280	sStr = '%#x (%d)' % (hrLong & 0xffffffff, hrLong)
1281	return sStr
1282
1283	def xcptGetMessage(self, oXcpt=None):
1284	"""
1285	Returns the best error message found in the COM-like exception. If the
1286	exception parameter isn't specified, the current exception is examined.
1287	"""
1288	if oXcpt is None:
1289	oXcpt = sys.exc_info()[1]
1290	sRet = self.platform.xcptGetMessage(oXcpt)
1291	if sRet is None:
1292	sRet = self.xcptToString(oXcpt)
1293	return sRet
1294

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

source: vbox/trunk/src/VBox/Main/glue/vboxapi.py@ 100842

Download in other formats: