VirtualBox

source: vbox/trunk/src/VBox/Main/USBProxyService.cpp@ 33868

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

src/*: more spelling fixes (logging), thanks Timeless!

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 37.0 KB
Line 
1/* $Id: USBProxyService.cpp 33595 2010-10-29 10:35:00Z vboxsync $ */
2/** @file
3 * VirtualBox USB Proxy Service (base) class.
4 */
5
6/*
7 * Copyright (C) 2006-2007 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 */
17
18#include "USBProxyService.h"
19#include "HostUSBDeviceImpl.h"
20#include "HostImpl.h"
21#include "MachineImpl.h"
22#include "VirtualBoxImpl.h"
23
24#include "AutoCaller.h"
25#include "Logging.h"
26
27#include <VBox/err.h>
28#include <iprt/asm.h>
29#include <iprt/semaphore.h>
30#include <iprt/thread.h>
31#include <iprt/mem.h>
32#include <iprt/string.h>
33
34
35/**
36 * Initialize data members.
37 */
38USBProxyService::USBProxyService(Host *aHost)
39 : mHost(aHost), mThread(NIL_RTTHREAD), mTerminate(false), mLastError(VINF_SUCCESS), mDevices()
40{
41 LogFlowThisFunc(("aHost=%p\n", aHost));
42}
43
44
45/**
46 * Initialize the object.
47 *
48 * Child classes should override and call this method
49 *
50 * @returns S_OK on success, or COM error status on fatal error.
51 */
52HRESULT USBProxyService::init(void)
53{
54 return S_OK;
55}
56
57
58/**
59 * Empty destructor.
60 */
61USBProxyService::~USBProxyService()
62{
63 LogFlowThisFunc(("\n"));
64 Assert(mThread == NIL_RTTHREAD);
65 mDevices.clear();
66 mTerminate = true;
67 mHost = NULL;
68}
69
70
71/**
72 * Query if the service is active and working.
73 *
74 * @returns true if the service is up running.
75 * @returns false if the service isn't running.
76 */
77bool USBProxyService::isActive(void)
78{
79 return mThread != NIL_RTTHREAD;
80}
81
82
83/**
84 * Get last error.
85 * Can be used to check why the proxy !isActive() upon construction.
86 *
87 * @returns VBox status code.
88 */
89int USBProxyService::getLastError(void)
90{
91 return mLastError;
92}
93
94
95/**
96 * Get last error message.
97 * Can be used to check why the proxy !isActive() upon construction as an
98 * extension to getLastError(). May return a NULL error.
99 *
100 * @param
101 * @returns VBox status code.
102 */
103HRESULT USBProxyService::getLastErrorMessage(BSTR *aError)
104{
105 AssertPtrReturn(aError, E_POINTER);
106 mLastErrorMessage.cloneTo(aError);
107 return S_OK;
108}
109
110
111/**
112 * We're using the Host object lock.
113 *
114 * This is just a temporary measure until all the USB refactoring is
115 * done, probably... For now it help avoiding deadlocks we don't have
116 * time to fix.
117 *
118 * @returns Lock handle.
119 */
120RWLockHandle *USBProxyService::lockHandle() const
121{
122 return mHost->lockHandle();
123}
124
125
126/**
127 * Gets the collection of USB devices, slave of Host::USBDevices.
128 *
129 * This is an interface for the HostImpl::USBDevices property getter.
130 *
131 *
132 * @param aUSBDevices Where to store the pointer to the collection.
133 *
134 * @returns COM status code.
135 *
136 * @remarks The caller must own the write lock of the host object.
137 */
138HRESULT USBProxyService::getDeviceCollection(ComSafeArrayOut(IHostUSBDevice *, aUSBDevices))
139{
140 AssertReturn(isWriteLockOnCurrentThread(), E_FAIL);
141 CheckComArgOutSafeArrayPointerValid(aUSBDevices);
142
143 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
144
145 SafeIfaceArray<IHostUSBDevice> Collection (mDevices);
146 Collection.detachTo(ComSafeArrayOutArg(aUSBDevices));
147
148 return S_OK;
149}
150
151
152/**
153 * Request capture of a specific device.
154 *
155 * This is in an interface for SessionMachine::CaptureUSBDevice(), which is
156 * an internal worker used by Console::AttachUSBDevice() from the VM process.
157 *
158 * When the request is completed, SessionMachine::onUSBDeviceAttach() will
159 * be called for the given machine object.
160 *
161 *
162 * @param aMachine The machine to attach the device to.
163 * @param aId The UUID of the USB device to capture and attach.
164 *
165 * @returns COM status code and error info.
166 *
167 * @remarks This method may operate synchronously as well as asynchronously. In the
168 * former case it will temporarily abandon locks because of IPC.
169 */
170HRESULT USBProxyService::captureDeviceForVM(SessionMachine *aMachine, IN_GUID aId)
171{
172 ComAssertRet(aMachine, E_INVALIDARG);
173 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
174
175 /*
176 * Translate the device id into a device object.
177 */
178 ComObjPtr<HostUSBDevice> pHostDevice = findDeviceById(aId);
179 if (pHostDevice.isNull())
180 return setError(E_INVALIDARG,
181 tr("The USB device with UUID {%RTuuid} is not currently attached to the host"), Guid(aId).raw());
182
183 /*
184 * Try to capture the device
185 */
186 AutoWriteLock DevLock(pHostDevice COMMA_LOCKVAL_SRC_POS);
187 return pHostDevice->requestCaptureForVM(aMachine, true /* aSetError */);
188}
189
190
191/**
192 * Notification from VM process about USB device detaching progress.
193 *
194 * This is in an interface for SessionMachine::DetachUSBDevice(), which is
195 * an internal worker used by Console::DetachUSBDevice() from the VM process.
196 *
197 * @param aMachine The machine which is sending the notification.
198 * @param aId The UUID of the USB device is concerns.
199 * @param aDone \a false for the pre-action notification (necessary
200 * for advancing the device state to avoid confusing
201 * the guest).
202 * \a true for the post-action notification. The device
203 * will be subjected to all filters except those of
204 * of \a Machine.
205 *
206 * @returns COM status code.
207 *
208 * @remarks When \a aDone is \a true this method may end up doing IPC to other
209 * VMs when running filters. In these cases it will temporarily
210 * abandon its locks.
211 */
212HRESULT USBProxyService::detachDeviceFromVM(SessionMachine *aMachine, IN_GUID aId, bool aDone)
213{
214 LogFlowThisFunc(("aMachine=%p{%s} aId={%RTuuid} aDone=%RTbool\n",
215 aMachine,
216 aMachine->getName().c_str(),
217 Guid(aId).raw(),
218 aDone));
219
220 // get a list of all running machines while we're outside the lock
221 // (getOpenedMachines requests locks which are incompatible with the lock of the machines list)
222 SessionMachinesList llOpenedMachines;
223 mHost->parent()->getOpenedMachines(llOpenedMachines);
224
225 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
226
227 ComObjPtr<HostUSBDevice> pHostDevice = findDeviceById(aId);
228 ComAssertRet(!pHostDevice.isNull(), E_FAIL);
229 AutoWriteLock DevLock(pHostDevice COMMA_LOCKVAL_SRC_POS);
230
231 /*
232 * Work the state machine.
233 */
234 LogFlowThisFunc(("id={%RTuuid} state=%s aDone=%RTbool name={%s}\n",
235 pHostDevice->getId().raw(), pHostDevice->getStateName(), aDone, pHostDevice->getName().c_str()));
236 bool fRunFilters = false;
237 HRESULT hrc = pHostDevice->onDetachFromVM(aMachine, aDone, &fRunFilters);
238
239 /*
240 * Run filters if necessary.
241 */
242 if ( SUCCEEDED(hrc)
243 && fRunFilters)
244 {
245 Assert(aDone && pHostDevice->getUnistate() == kHostUSBDeviceState_HeldByProxy && pHostDevice->getMachine().isNull());
246 HRESULT hrc2 = runAllFiltersOnDevice(pHostDevice, llOpenedMachines, aMachine);
247 ComAssertComRC(hrc2);
248 }
249 return hrc;
250}
251
252
253/**
254 * Apply filters for the machine to all eligible USB devices.
255 *
256 * This is in an interface for SessionMachine::CaptureUSBDevice(), which
257 * is an internal worker used by Console::AutoCaptureUSBDevices() from the
258 * VM process at VM startup.
259 *
260 * Matching devices will be attached to the VM and may result IPC back
261 * to the VM process via SessionMachine::onUSBDeviceAttach() depending
262 * on whether the device needs to be captured or not. If capture is
263 * required, SessionMachine::onUSBDeviceAttach() will be called
264 * asynchronously by the USB proxy service thread.
265 *
266 * @param aMachine The machine to capture devices for.
267 *
268 * @returns COM status code, perhaps with error info.
269 *
270 * @remarks Write locks the host object and may temporarily abandon
271 * its locks to perform IPC.
272 */
273HRESULT USBProxyService::autoCaptureDevicesForVM(SessionMachine *aMachine)
274{
275 LogFlowThisFunc(("aMachine=%p{%s}\n",
276 aMachine,
277 aMachine->getName().c_str()));
278 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
279 AutoWriteLock mlock(aMachine COMMA_LOCKVAL_SRC_POS);
280
281 /*
282 * Make a copy of the list because we might have to exit and
283 * re-enter the lock protecting it. (This will not make copies
284 * of any HostUSBDevice objects, only reference them.)
285 */
286 HostUSBDeviceList ListCopy = mDevices;
287
288 for (HostUSBDeviceList::iterator it = ListCopy.begin();
289 it != ListCopy.end();
290 ++it)
291 {
292 ComObjPtr<HostUSBDevice> device = *it;
293 AutoWriteLock devLock(device COMMA_LOCKVAL_SRC_POS);
294 if ( device->getUnistate() == kHostUSBDeviceState_HeldByProxy
295 || device->getUnistate() == kHostUSBDeviceState_Unused
296 || device->getUnistate() == kHostUSBDeviceState_Capturable)
297 runMachineFilters(aMachine, device);
298 }
299
300 return S_OK;
301}
302
303
304/**
305 * Detach all USB devices currently attached to a VM.
306 *
307 * This is in an interface for SessionMachine::DetachAllUSBDevices(), which
308 * is an internal worker used by Console::powerDown() from the VM process
309 * at VM startup, and SessionMachine::uninit() at VM abend.
310 *
311 * This is, like #detachDeviceFromVM(), normally a two stage journey
312 * where \a aDone indicates where we are. In addition we may be called
313 * to clean up VMs that have abended, in which case there will be no
314 * preparatory call. Filters will be applied to the devices in the final
315 * call with the risk that we have to do some IPC when attaching them
316 * to other VMs.
317 *
318 * @param aMachine The machine to detach devices from.
319 *
320 * @returns COM status code, perhaps with error info.
321 *
322 * @remarks Write locks the host object and may temporarily abandon
323 * its locks to perform IPC.
324 */
325HRESULT USBProxyService::detachAllDevicesFromVM(SessionMachine *aMachine, bool aDone, bool aAbnormal)
326{
327 // get a list of all running machines while we're outside the lock
328 // (getOpenedMachines requests locks which are incompatible with the lock of the machines list)
329 SessionMachinesList llOpenedMachines;
330 mHost->parent()->getOpenedMachines(llOpenedMachines);
331
332 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
333
334 /*
335 * Make a copy of the device list (not the HostUSBDevice objects, just
336 * the list) since we may end up performing IPC and temporarily have
337 * to abandon locks when applying filters.
338 */
339 HostUSBDeviceList ListCopy = mDevices;
340
341 for (HostUSBDeviceList::iterator It = ListCopy.begin();
342 It != ListCopy.end();
343 ++It)
344 {
345 ComObjPtr<HostUSBDevice> pHostDevice = *It;
346 AutoWriteLock devLock(pHostDevice COMMA_LOCKVAL_SRC_POS);
347 if (pHostDevice->getMachine() == aMachine)
348 {
349 /*
350 * Same procedure as in detachUSBDevice().
351 */
352 bool fRunFilters = false;
353 HRESULT hrc = pHostDevice->onDetachFromVM(aMachine, aDone, &fRunFilters, aAbnormal);
354 if ( SUCCEEDED(hrc)
355 && fRunFilters)
356 {
357 Assert(aDone && pHostDevice->getUnistate() == kHostUSBDeviceState_HeldByProxy && pHostDevice->getMachine().isNull());
358 HRESULT hrc2 = runAllFiltersOnDevice(pHostDevice, llOpenedMachines, aMachine);
359 ComAssertComRC(hrc2);
360 }
361 }
362 }
363
364 return S_OK;
365}
366
367
368/**
369 * Runs all the filters on the specified device.
370 *
371 * All filters mean global and active VM, with the exception of those
372 * belonging to \a aMachine. If a global ignore filter matched or if
373 * none of the filters matched, the device will be released back to
374 * the host.
375 *
376 * The device calling us here will be in the HeldByProxy, Unused, or
377 * Capturable state. The caller is aware that locks held might have
378 * to be abandond because of IPC and that the device might be in
379 * almost any state upon return.
380 *
381 *
382 * @returns COM status code (only parameter & state checks will fail).
383 * @param aDevice The USB device to apply filters to.
384 * @param aIgnoreMachine The machine to ignore filters from (we've just
385 * detached the device from this machine).
386 *
387 * @note The caller is expected to own both the device and Host write locks,
388 * and be prepared that these locks may be abandond temporarily.
389 */
390HRESULT USBProxyService::runAllFiltersOnDevice(ComObjPtr<HostUSBDevice> &aDevice,
391 SessionMachinesList &llOpenedMachines,
392 SessionMachine *aIgnoreMachine)
393{
394 LogFlowThisFunc(("{%s} ignoring=%p\n", aDevice->getName().c_str(), aIgnoreMachine));
395
396 /*
397 * Verify preconditions.
398 */
399 AssertReturn(isWriteLockOnCurrentThread(), E_FAIL);
400 AssertReturn(aDevice->isWriteLockOnCurrentThread(), E_FAIL);
401 AssertMsgReturn(aDevice->isCapturableOrHeld(), ("{%s} %s\n", aDevice->getName().c_str(), aDevice->getStateName()), E_FAIL);
402
403 /*
404 * Get the lists we'll iterate.
405 */
406 Host::USBDeviceFilterList globalFilters;
407
408 mHost->getUSBFilters(&globalFilters);
409
410 /*
411 * Run global filters filerts first.
412 */
413 bool fHoldIt = false;
414 for (Host::USBDeviceFilterList::const_iterator it = globalFilters.begin();
415 it != globalFilters.end();
416 ++it)
417 {
418 AutoWriteLock filterLock(*it COMMA_LOCKVAL_SRC_POS);
419 const HostUSBDeviceFilter::Data &data = (*it)->getData();
420 if (aDevice->isMatch(data))
421 {
422 USBDeviceFilterAction_T action = USBDeviceFilterAction_Null;
423 (*it)->COMGETTER(Action)(&action);
424 if (action == USBDeviceFilterAction_Ignore)
425 {
426 /*
427 * Release the device to the host and we're done.
428 */
429 aDevice->requestReleaseToHost();
430 return S_OK;
431 }
432 if (action == USBDeviceFilterAction_Hold)
433 {
434 /*
435 * A device held by the proxy needs to be subjected
436 * to the machine filters.
437 */
438 fHoldIt = true;
439 break;
440 }
441 AssertMsgFailed(("action=%d\n", action));
442 }
443 }
444 globalFilters.clear();
445
446 /*
447 * Run the per-machine filters.
448 */
449 for (SessionMachinesList::const_iterator it = llOpenedMachines.begin();
450 it != llOpenedMachines.end();
451 ++it)
452 {
453 ComObjPtr<SessionMachine> pMachine = *it;
454
455 /* Skip the machine the device was just detached from. */
456 if ( aIgnoreMachine
457 && pMachine == aIgnoreMachine)
458 continue;
459
460 /* runMachineFilters takes care of checking the machine state. */
461 if (runMachineFilters(pMachine, aDevice))
462 {
463 LogFlowThisFunc(("{%s} attached to %p\n", aDevice->getName().c_str(), (void *)pMachine));
464 return S_OK;
465 }
466 }
467
468 /*
469 * No matching machine, so request hold or release depending
470 * on global filter match.
471 */
472 if (fHoldIt)
473 aDevice->requestHold();
474 else
475 aDevice->requestReleaseToHost();
476 return S_OK;
477}
478
479
480/**
481 * Runs the USB filters of the machine on the device.
482 *
483 * If a match is found we will request capture for VM. This may cause
484 * us to temporary abandon locks while doing IPC.
485 *
486 * @param aMachine Machine whose filters are to be run.
487 * @param aDevice The USB device in question.
488 * @returns @c true if the device has been or is being attached to the VM, @c false otherwise.
489 *
490 * @note Caller must own the USB and device locks for writing.
491 * @note Locks aMachine for reading.
492 */
493bool USBProxyService::runMachineFilters(SessionMachine *aMachine, ComObjPtr<HostUSBDevice> &aDevice)
494{
495 LogFlowThisFunc(("{%s} aMachine=%p \n", aDevice->getName().c_str(), aMachine));
496
497 /*
498 * Validate preconditions.
499 */
500 AssertReturn(aMachine, false);
501 AssertReturn(isWriteLockOnCurrentThread(), false);
502 AssertReturn(aDevice->isWriteLockOnCurrentThread(), false);
503 /* Let HostUSBDevice::requestCaptureToVM() validate the state. */
504
505 /*
506 * Do the job.
507 */
508 ULONG ulMaskedIfs;
509 if (aMachine->hasMatchingUSBFilter(aDevice, &ulMaskedIfs))
510 {
511 /* try to capture the device */
512 HRESULT hrc = aDevice->requestCaptureForVM(aMachine, false /* aSetError */, ulMaskedIfs);
513 return SUCCEEDED(hrc)
514 || hrc == E_UNEXPECTED /* bad device state, give up */;
515 }
516
517 return false;
518}
519
520
521/**
522 * A filter was inserted / loaded.
523 *
524 * @param aFilter Pointer to the inserted filter.
525 * @return ID of the inserted filter
526 */
527void *USBProxyService::insertFilter(PCUSBFILTER aFilter)
528{
529 // return non-NULL to fake success.
530 NOREF(aFilter);
531 return (void *)1;
532}
533
534
535/**
536 * A filter was removed.
537 *
538 * @param aId ID of the filter to remove
539 */
540void USBProxyService::removeFilter(void *aId)
541{
542 NOREF(aId);
543}
544
545
546/**
547 * A VM is trying to capture a device, do necessary preparations.
548 *
549 * @returns VBox status code.
550 * @param aDevice The device in question.
551 */
552int USBProxyService::captureDevice(HostUSBDevice *aDevice)
553{
554 NOREF(aDevice);
555 return VERR_NOT_IMPLEMENTED;
556}
557
558
559/**
560 * Notification that an async captureDevice() operation completed.
561 *
562 * This is used by the proxy to release temporary filters.
563 *
564 * @returns VBox status code.
565 * @param aDevice The device in question.
566 * @param aSuccess Whether it succeeded or failed.
567 */
568void USBProxyService::captureDeviceCompleted(HostUSBDevice *aDevice, bool aSuccess)
569{
570 NOREF(aDevice);
571 NOREF(aSuccess);
572}
573
574
575/**
576 * The device is going to be detached from a VM.
577 *
578 * @param aDevice The device in question.
579 */
580void USBProxyService::detachingDevice(HostUSBDevice *aDevice)
581{
582 NOREF(aDevice);
583}
584
585
586/**
587 * A VM is releasing a device back to the host.
588 *
589 * @returns VBox status code.
590 * @param aDevice The device in question.
591 */
592int USBProxyService::releaseDevice(HostUSBDevice *aDevice)
593{
594 NOREF(aDevice);
595 return VERR_NOT_IMPLEMENTED;
596}
597
598
599/**
600 * Notification that an async releaseDevice() operation completed.
601 *
602 * This is used by the proxy to release temporary filters.
603 *
604 * @returns VBox status code.
605 * @param aDevice The device in question.
606 * @param aSuccess Whether it succeeded or failed.
607 */
608void USBProxyService::releaseDeviceCompleted(HostUSBDevice *aDevice, bool aSuccess)
609{
610 NOREF(aDevice);
611 NOREF(aSuccess);
612}
613
614
615// Internals
616/////////////////////////////////////////////////////////////////////////////
617
618
619/**
620 * Starts the service.
621 *
622 * @returns VBox status.
623 */
624int USBProxyService::start(void)
625{
626 int rc = VINF_SUCCESS;
627 if (mThread == NIL_RTTHREAD)
628 {
629 /*
630 * Force update before starting the poller thread.
631 */
632 rc = wait(0);
633 if (rc == VERR_TIMEOUT || rc == VERR_INTERRUPTED || RT_SUCCESS(rc))
634 {
635 processChanges();
636
637 /*
638 * Create the poller thread which will look for changes.
639 */
640 mTerminate = false;
641 rc = RTThreadCreate(&mThread, USBProxyService::serviceThread, this,
642 0, RTTHREADTYPE_INFREQUENT_POLLER, RTTHREADFLAGS_WAITABLE, "USBPROXY");
643 AssertRC(rc);
644 if (RT_SUCCESS(rc))
645 LogFlowThisFunc(("started mThread=%RTthrd\n", mThread));
646 else
647 mThread = NIL_RTTHREAD;
648 }
649 mLastError = rc;
650 }
651 else
652 LogFlowThisFunc(("already running, mThread=%RTthrd\n", mThread));
653 return rc;
654}
655
656
657/**
658 * Stops the service.
659 *
660 * @returns VBox status.
661 */
662int USBProxyService::stop(void)
663{
664 int rc = VINF_SUCCESS;
665 if (mThread != NIL_RTTHREAD)
666 {
667 /*
668 * Mark the thread for termination and kick it.
669 */
670 ASMAtomicXchgSize(&mTerminate, true);
671 rc = interruptWait();
672 AssertRC(rc);
673
674 /*
675 * Wait for the thread to finish and then update the state.
676 */
677 rc = RTThreadWait(mThread, 60000, NULL);
678 if (rc == VERR_INVALID_HANDLE)
679 rc = VINF_SUCCESS;
680 if (RT_SUCCESS(rc))
681 {
682 LogFlowThisFunc(("stopped mThread=%RTthrd\n", mThread));
683 mThread = NIL_RTTHREAD;
684 mTerminate = false;
685 }
686 else
687 {
688 AssertRC(rc);
689 mLastError = rc;
690 }
691 }
692 else
693 LogFlowThisFunc(("not active\n"));
694
695 return rc;
696}
697
698
699/**
700 * The service thread created by start().
701 *
702 * @param Thread The thread handle.
703 * @param pvUser Pointer to the USBProxyService instance.
704 */
705/*static*/ DECLCALLBACK(int) USBProxyService::serviceThread(RTTHREAD /* Thread */, void *pvUser)
706{
707 USBProxyService *pThis = (USBProxyService *)pvUser;
708 LogFlowFunc(("pThis=%p\n", pThis));
709 pThis->serviceThreadInit();
710 int rc = VINF_SUCCESS;
711
712 /*
713 * Processing loop.
714 */
715 for (;;)
716 {
717 rc = pThis->wait(RT_INDEFINITE_WAIT);
718 if (RT_FAILURE(rc) && rc != VERR_INTERRUPTED && rc != VERR_TIMEOUT)
719 break;
720 if (pThis->mTerminate)
721 break;
722 pThis->processChanges();
723 }
724
725 pThis->serviceThreadTerm();
726 LogFlowFunc(("returns %Rrc\n", rc));
727 return rc;
728}
729
730
731/**
732 * First call made on the service thread, use it to do
733 * thread initialization.
734 *
735 * The default implementation in USBProxyService just a dummy stub.
736 */
737void USBProxyService::serviceThreadInit(void)
738{
739}
740
741
742/**
743 * Last call made on the service thread, use it to do
744 * thread termination.
745 */
746void USBProxyService::serviceThreadTerm(void)
747{
748}
749
750
751/**
752 * Wait for a change in the USB devices attached to the host.
753 *
754 * The default implementation in USBProxyService just a dummy stub.
755 *
756 * @returns VBox status code. VERR_INTERRUPTED and VERR_TIMEOUT are considered
757 * harmless, while all other error status are fatal.
758 * @param aMillies Number of milliseconds to wait.
759 */
760int USBProxyService::wait(RTMSINTERVAL aMillies)
761{
762 return RTThreadSleep(RT_MIN(aMillies, 250));
763}
764
765
766/**
767 * Interrupt any wait() call in progress.
768 *
769 * The default implementation in USBProxyService just a dummy stub.
770 *
771 * @returns VBox status.
772 */
773int USBProxyService::interruptWait(void)
774{
775 return VERR_NOT_IMPLEMENTED;
776}
777
778
779/**
780 * Sort a list of USB devices.
781 *
782 * @returns Pointer to the head of the sorted doubly linked list.
783 * @param aDevices Head pointer (can be both singly and doubly linked list).
784 */
785static PUSBDEVICE sortDevices(PUSBDEVICE pDevices)
786{
787 PUSBDEVICE pHead = NULL;
788 PUSBDEVICE pTail = NULL;
789 while (pDevices)
790 {
791 /* unlink head */
792 PUSBDEVICE pDev = pDevices;
793 pDevices = pDev->pNext;
794 if (pDevices)
795 pDevices->pPrev = NULL;
796
797 /* find location. */
798 PUSBDEVICE pCur = pTail;
799 while ( pCur
800 && HostUSBDevice::compare(pCur, pDev) > 0)
801 pCur = pCur->pPrev;
802
803 /* insert (after pCur) */
804 pDev->pPrev = pCur;
805 if (pCur)
806 {
807 pDev->pNext = pCur->pNext;
808 pCur->pNext = pDev;
809 if (pDev->pNext)
810 pDev->pNext->pPrev = pDev;
811 else
812 pTail = pDev;
813 }
814 else
815 {
816 pDev->pNext = pHead;
817 if (pHead)
818 pHead->pPrev = pDev;
819 else
820 pTail = pDev;
821 pHead = pDev;
822 }
823 }
824
825 return pHead;
826}
827
828
829/**
830 * Process any relevant changes in the attached USB devices.
831 *
832 * Except for the first call, this is always running on the service thread.
833 */
834void USBProxyService::processChanges(void)
835{
836 LogFlowThisFunc(("\n"));
837
838 /*
839 * Get the sorted list of USB devices.
840 */
841 PUSBDEVICE pDevices = getDevices();
842 pDevices = sortDevices(pDevices);
843
844 // get a list of all running machines while we're outside the lock
845 // (getOpenedMachines requests locks which are incompatible with the lock of the machines list)
846 SessionMachinesList llOpenedMachines;
847 mHost->parent()->getOpenedMachines(llOpenedMachines);
848
849 AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS);
850
851 /*
852 * Compare previous list with the previous list of devices
853 * and merge in any changes while notifying Host.
854 */
855 HostUSBDeviceList::iterator It = this->mDevices.begin();
856 while ( It != mDevices.end()
857 || pDevices)
858 {
859 ComObjPtr<HostUSBDevice> pHostDevice;
860
861 if (It != mDevices.end())
862 pHostDevice = *It;
863
864 /*
865 * Assert that the object is still alive (we still reference it in
866 * the collection and we're the only one who calls uninit() on it.
867 */
868 AutoCaller devCaller(pHostDevice.isNull() ? NULL : pHostDevice);
869 AssertComRC(devCaller.rc());
870
871 /*
872 * Lock the device object since we will read/write its
873 * properties. All Host callbacks also imply the object is locked.
874 */
875 AutoWriteLock devLock(pHostDevice.isNull() ? NULL : pHostDevice
876 COMMA_LOCKVAL_SRC_POS);
877
878 /*
879 * Compare.
880 */
881 int iDiff;
882 if (pHostDevice.isNull())
883 iDiff = 1;
884 else
885 {
886 if (!pDevices)
887 iDiff = -1;
888 else
889 iDiff = pHostDevice->compare(pDevices);
890 }
891 if (!iDiff)
892 {
893 /*
894 * The device still there, update the state and move on. The PUSBDEVICE
895 * structure is eaten by updateDeviceState / HostUSBDevice::updateState().
896 */
897 PUSBDEVICE pCur = pDevices;
898 pDevices = pDevices->pNext;
899 pCur->pPrev = pCur->pNext = NULL;
900
901 bool fRunFilters = false;
902 SessionMachine *pIgnoreMachine = NULL;
903 if (updateDeviceState(pHostDevice, pCur, &fRunFilters, &pIgnoreMachine))
904 deviceChanged(pHostDevice,
905 (fRunFilters ? &llOpenedMachines : NULL),
906 pIgnoreMachine);
907 It++;
908 }
909 else
910 {
911 if (iDiff > 0)
912 {
913 /*
914 * Head of pDevices was attached.
915 */
916 PUSBDEVICE pNew = pDevices;
917 pDevices = pDevices->pNext;
918 pNew->pPrev = pNew->pNext = NULL;
919
920 ComObjPtr<HostUSBDevice> NewObj;
921 NewObj.createObject();
922 NewObj->init(pNew, this);
923 Log(("USBProxyService::processChanges: attached %p {%s} %s / %p:{.idVendor=%#06x, .idProduct=%#06x, .pszProduct=\"%s\", .pszManufacturer=\"%s\"}\n",
924 (HostUSBDevice *)NewObj,
925 NewObj->getName().c_str(),
926 NewObj->getStateName(),
927 pNew,
928 pNew->idVendor,
929 pNew->idProduct,
930 pNew->pszProduct,
931 pNew->pszManufacturer));
932
933 mDevices.insert(It, NewObj);
934
935 /* Not really necessary to lock here, but make Assert checks happy. */
936 AutoWriteLock newDevLock(NewObj COMMA_LOCKVAL_SRC_POS);
937 deviceAdded(NewObj, llOpenedMachines, pNew);
938 }
939 else
940 {
941 /*
942 * Check if the device was actually detached or logically detached
943 * as the result of a re-enumeration.
944 */
945 if (!pHostDevice->wasActuallyDetached())
946 It++;
947 else
948 {
949 It = mDevices.erase(It);
950 deviceRemoved(pHostDevice);
951 Log(("USBProxyService::processChanges: detached %p {%s}\n",
952 (HostUSBDevice *)pHostDevice,
953 pHostDevice->getName().c_str()));
954
955 /* from now on, the object is no more valid,
956 * uninitialize to avoid abuse */
957 devCaller.release();
958 pHostDevice->uninit();
959 }
960 }
961 }
962 } /* while */
963
964 LogFlowThisFunc(("returns void\n"));
965}
966
967
968/**
969 * Get a list of USB device currently attached to the host.
970 *
971 * The default implementation in USBProxyService just a dummy stub.
972 *
973 * @returns Pointer to a list of USB devices.
974 * The list nodes are freed individually by calling freeDevice().
975 */
976PUSBDEVICE USBProxyService::getDevices(void)
977{
978 return NULL;
979}
980
981
982/**
983 * Performs the required actions when a device has been added.
984 *
985 * This means things like running filters and subsequent capturing and
986 * VM attaching. This may result in IPC and temporary lock abandonment.
987 *
988 * @param aDevice The device in question.
989 * @param aUSBDevice The USB device structure.
990 */
991void USBProxyService::deviceAdded(ComObjPtr<HostUSBDevice> &aDevice,
992 SessionMachinesList &llOpenedMachines,
993 PUSBDEVICE aUSBDevice)
994{
995 /*
996 * Validate preconditions.
997 */
998 AssertReturnVoid(isWriteLockOnCurrentThread());
999 AssertReturnVoid(aDevice->isWriteLockOnCurrentThread());
1000 LogFlowThisFunc(("aDevice=%p name={%s} state=%s id={%RTuuid}\n",
1001 (HostUSBDevice *)aDevice,
1002 aDevice->getName().c_str(),
1003 aDevice->getStateName(),
1004 aDevice->getId().raw()));
1005
1006 /*
1007 * Run filters on the device.
1008 */
1009 if (aDevice->isCapturableOrHeld())
1010 {
1011 HRESULT rc = runAllFiltersOnDevice(aDevice, llOpenedMachines, NULL /* aIgnoreMachine */);
1012 AssertComRC(rc);
1013 }
1014
1015 NOREF(aUSBDevice);
1016}
1017
1018
1019/**
1020 * Remove device notification hook for the OS specific code.
1021 *
1022 * This is means things like
1023 *
1024 * @param aDevice The device in question.
1025 */
1026void USBProxyService::deviceRemoved(ComObjPtr<HostUSBDevice> &aDevice)
1027{
1028 /*
1029 * Validate preconditions.
1030 */
1031 AssertReturnVoid(isWriteLockOnCurrentThread());
1032 AssertReturnVoid(aDevice->isWriteLockOnCurrentThread());
1033 LogFlowThisFunc(("aDevice=%p name={%s} state=%s id={%RTuuid}\n",
1034 (HostUSBDevice *)aDevice,
1035 aDevice->getName().c_str(),
1036 aDevice->getStateName(),
1037 aDevice->getId().raw()));
1038
1039 /*
1040 * Detach the device from any machine currently using it,
1041 * reset all data and uninitialize the device object.
1042 */
1043 aDevice->onPhysicalDetached();
1044}
1045
1046
1047/**
1048 * Implement fake capture, ++.
1049 *
1050 * @returns true if there is a state change.
1051 * @param pDevice The device in question.
1052 * @param pUSBDevice The USB device structure for the last enumeration.
1053 * @param aRunFilters Whether or not to run filters.
1054 */
1055bool USBProxyService::updateDeviceStateFake(HostUSBDevice *aDevice, PUSBDEVICE aUSBDevice, bool *aRunFilters, SessionMachine **aIgnoreMachine)
1056{
1057 *aRunFilters = false;
1058 *aIgnoreMachine = NULL;
1059 AssertReturn(aDevice, false);
1060 AssertReturn(aDevice->isWriteLockOnCurrentThread(), false);
1061
1062 /*
1063 * Just hand it to the device, it knows best what needs to be done.
1064 */
1065 return aDevice->updateStateFake(aUSBDevice, aRunFilters, aIgnoreMachine);
1066}
1067
1068
1069/**
1070 * Updates the device state.
1071 *
1072 * This is responsible for calling HostUSBDevice::updateState().
1073 *
1074 * @returns true if there is a state change.
1075 * @param aDevice The device in question.
1076 * @param aUSBDevice The USB device structure for the last enumeration.
1077 * @param aRunFilters Whether or not to run filters.
1078 * @param aIgnoreMachine Machine to ignore when running filters.
1079 */
1080bool USBProxyService::updateDeviceState(HostUSBDevice *aDevice, PUSBDEVICE aUSBDevice, bool *aRunFilters, SessionMachine **aIgnoreMachine)
1081{
1082 AssertReturn(aDevice, false);
1083 AssertReturn(aDevice->isWriteLockOnCurrentThread(), false);
1084
1085 return aDevice->updateState(aUSBDevice, aRunFilters, aIgnoreMachine);
1086}
1087
1088
1089/**
1090 * Handle a device which state changed in some significant way.
1091 *
1092 * This means things like running filters and subsequent capturing and
1093 * VM attaching. This may result in IPC and temporary lock abandonment.
1094 *
1095 * @param aDevice The device.
1096 * @param pllOpenedMachines list of running session machines (VirtualBox::getOpenedMachines()); if NULL, we don't run filters
1097 * @param aIgnoreMachine Machine to ignore when running filters.
1098 */
1099void USBProxyService::deviceChanged(ComObjPtr<HostUSBDevice> &aDevice, SessionMachinesList *pllOpenedMachines, SessionMachine *aIgnoreMachine)
1100{
1101 /*
1102 * Validate preconditions.
1103 */
1104 AssertReturnVoid(isWriteLockOnCurrentThread());
1105 AssertReturnVoid(aDevice->isWriteLockOnCurrentThread());
1106 LogFlowThisFunc(("aDevice=%p name={%s} state=%s id={%RTuuid} aRunFilters=%RTbool aIgnoreMachine=%p\n",
1107 (HostUSBDevice *)aDevice,
1108 aDevice->getName().c_str(),
1109 aDevice->getStateName(),
1110 aDevice->getId().raw(),
1111 (pllOpenedMachines != NULL), // used to be "bool aRunFilters"
1112 aIgnoreMachine));
1113
1114 /*
1115 * Run filters if requested to do so.
1116 */
1117 if (pllOpenedMachines)
1118 {
1119 HRESULT rc = runAllFiltersOnDevice(aDevice, *pllOpenedMachines, aIgnoreMachine);
1120 AssertComRC(rc);
1121 }
1122}
1123
1124
1125
1126/**
1127 * Free all the members of a USB device returned by getDevice().
1128 *
1129 * @param pDevice Pointer to the device.
1130 */
1131/*static*/ void
1132USBProxyService::freeDeviceMembers(PUSBDEVICE pDevice)
1133{
1134 RTStrFree((char *)pDevice->pszManufacturer);
1135 pDevice->pszManufacturer = NULL;
1136 RTStrFree((char *)pDevice->pszProduct);
1137 pDevice->pszProduct = NULL;
1138 RTStrFree((char *)pDevice->pszSerialNumber);
1139 pDevice->pszSerialNumber = NULL;
1140
1141 RTStrFree((char *)pDevice->pszAddress);
1142 pDevice->pszAddress = NULL;
1143#ifdef RT_OS_WINDOWS
1144 RTStrFree(pDevice->pszAltAddress);
1145 pDevice->pszAltAddress = NULL;
1146 RTStrFree(pDevice->pszHubName);
1147 pDevice->pszHubName = NULL;
1148#elif defined(RT_OS_SOLARIS)
1149 RTStrFree(pDevice->pszDevicePath);
1150 pDevice->pszDevicePath = NULL;
1151#endif
1152}
1153
1154
1155/**
1156 * Free one USB device returned by getDevice().
1157 *
1158 * @param pDevice Pointer to the device.
1159 */
1160/*static*/ void
1161USBProxyService::freeDevice(PUSBDEVICE pDevice)
1162{
1163 freeDeviceMembers(pDevice);
1164 RTMemFree(pDevice);
1165}
1166
1167
1168/**
1169 * Initializes a filter with the data from the specified device.
1170 *
1171 * @param aFilter The filter to fill.
1172 * @param aDevice The device to fill it with.
1173 */
1174/*static*/ void
1175USBProxyService::initFilterFromDevice(PUSBFILTER aFilter, HostUSBDevice *aDevice)
1176{
1177 PCUSBDEVICE pDev = aDevice->mUsb;
1178 int vrc;
1179
1180 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_VENDOR_ID, pDev->idVendor, true); AssertRC(vrc);
1181 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_PRODUCT_ID, pDev->idProduct, true); AssertRC(vrc);
1182 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_DEVICE_REV, pDev->bcdDevice, true); AssertRC(vrc);
1183 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_DEVICE_CLASS, pDev->bDeviceClass, true); AssertRC(vrc);
1184 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_DEVICE_SUB_CLASS, pDev->bDeviceSubClass, true); AssertRC(vrc);
1185 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_DEVICE_PROTOCOL, pDev->bDeviceProtocol, true); AssertRC(vrc);
1186 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_PORT, pDev->bPort, true); AssertRC(vrc);
1187 vrc = USBFilterSetNumExact(aFilter, USBFILTERIDX_BUS, pDev->bBus, true); AssertRC(vrc);
1188 if (pDev->pszSerialNumber)
1189 {
1190 vrc = USBFilterSetStringExact(aFilter, USBFILTERIDX_SERIAL_NUMBER_STR, pDev->pszSerialNumber, true);
1191 AssertRC(vrc);
1192 }
1193 if (pDev->pszProduct)
1194 {
1195 vrc = USBFilterSetStringExact(aFilter, USBFILTERIDX_PRODUCT_STR, pDev->pszProduct, true);
1196 AssertRC(vrc);
1197 }
1198 if (pDev->pszManufacturer)
1199 {
1200 vrc = USBFilterSetStringExact(aFilter, USBFILTERIDX_MANUFACTURER_STR, pDev->pszManufacturer, true);
1201 AssertRC(vrc);
1202 }
1203}
1204
1205
1206/**
1207 * Searches the list of devices (mDevices) for the given device.
1208 *
1209 *
1210 * @returns Smart pointer to the device on success, NULL otherwise.
1211 * @param aId The UUID of the device we're looking for.
1212 */
1213ComObjPtr<HostUSBDevice> USBProxyService::findDeviceById(IN_GUID aId)
1214{
1215 Guid Id(aId);
1216 ComObjPtr<HostUSBDevice> Dev;
1217 for (HostUSBDeviceList::iterator It = mDevices.begin();
1218 It != mDevices.end();
1219 ++It)
1220 if ((*It)->getId() == Id)
1221 {
1222 Dev = (*It);
1223 break;
1224 }
1225
1226 return Dev;
1227}
1228
1229/*static*/
1230HRESULT USBProxyService::setError(HRESULT aResultCode, const char *aText, ...)
1231{
1232 va_list va;
1233 va_start(va, aText);
1234 HRESULT rc = VirtualBoxBase::setErrorInternal(aResultCode,
1235 COM_IIDOF(IHost),
1236 "USBProxyService",
1237 Utf8StrFmt(aText, va),
1238 false /* aWarning*/,
1239 true /* aLogIt*/);
1240 va_end(va);
1241 return rc;
1242}
1243
1244/* vi: set tabstop=4 shiftwidth=4 expandtab: */
Note: See TracBrowser for help on using the repository browser.

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