X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/6d2010ae8f7a6078e10b361c6962983bab233e0f..c6bf4f310a33a9262d455ea4d3f0630b1255e3fe:/iokit/Kernel/IOEventSource.cpp?ds=sidebyside

diff --git a/iokit/Kernel/IOEventSource.cpp b/iokit/Kernel/IOEventSource.cpp
index 95046dacd..3415fd34a 100644
--- a/iokit/Kernel/IOEventSource.cpp
+++ b/iokit/Kernel/IOEventSource.cpp
@@ -2,7 +2,7 @@
  * Copyright (c) 1998-2000, 2009 Apple Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- * 
+ *
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
@@ -11,10 +11,10 @@
  * unlawful or unlicensed copies of an Apple operating system, or to
  * circumvent, violate, or enable the circumvention or violation of, any
  * terms of an Apple operating system software license agreement.
- * 
+ *
  * Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this file.
- * 
+ *
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
@@ -22,20 +22,21 @@
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- * 
+ *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 /*
-Copyright (c) 1998 Apple Computer, Inc.  All rights reserved.
-
-HISTORY
-    1998-7-13	Godfrey van der Linden(gvdl)
-        Created.
-]*/
+ *  Copyright (c) 1998 Apple Computer, Inc.  All rights reserved.
+ *
+ *  HISTORY
+ *   1998-7-13	Godfrey van der Linden(gvdl)
+ *       Created.
+ *  ]*/
 #include <IOKit/IOLib.h>
 
 #include <IOKit/IOEventSource.h>
 #include <IOKit/IOWorkLoop.h>
+#include <libkern/Block.h>
 
 #define super OSObject
 
@@ -50,7 +51,11 @@ OSMetaClassDefineReservedUnused(IOEventSource, 5);
 OSMetaClassDefineReservedUnused(IOEventSource, 6);
 OSMetaClassDefineReservedUnused(IOEventSource, 7);
 
-bool IOEventSource::checkForWork() { return false; }
+bool
+IOEventSource::checkForWork()
+{
+	return false;
+}
 
 /* inline function implementations */
 
@@ -64,7 +69,7 @@ do { \
 #define IOStatisticsUnregisterCounter() \
 do { \
 	if (reserved) \
-		IOStatistics::unregisterEventSource(reserved->counter); \
+	        IOStatistics::unregisterEventSource(reserved->counter); \
 } while (0)
 
 #define IOStatisticsOpenGate() \
@@ -86,133 +91,207 @@ do { \
 
 #endif /* IOKITSTATS */
 
-void IOEventSource::signalWorkAvailable() 
-{ 
-	workLoop->signalWorkAvailable(); 
+void
+IOEventSource::signalWorkAvailable()
+{
+	workLoop->signalWorkAvailable();
 }
 
-void IOEventSource::openGate() 
-{ 
+void
+IOEventSource::openGate()
+{
 	IOStatisticsOpenGate();
-	workLoop->openGate(); 
+	workLoop->openGate();
 }
 
-void IOEventSource::closeGate()	
-{ 
-	workLoop->closeGate(); 
-	IOStatisticsCloseGate(); 
+void
+IOEventSource::closeGate()
+{
+	workLoop->closeGate();
+	IOStatisticsCloseGate();
 }
 
-bool IOEventSource::tryCloseGate() 
-{ 
-	bool res; 
+bool
+IOEventSource::tryCloseGate()
+{
+	bool res;
 	if ((res = workLoop->tryCloseGate())) {
-		IOStatisticsCloseGate(); 
+		IOStatisticsCloseGate();
 	}
-	return res; 
+	return res;
 }
 
-int IOEventSource::sleepGate(void *event, UInt32 type)
-{ 
-	bool res; 
-	IOStatisticsOpenGate(); 
-	res = workLoop->sleepGate(event, type); 
-	IOStatisticsCloseGate(); 
-	return res; 
+int
+IOEventSource::sleepGate(void *event, UInt32 type)
+{
+	int res;
+	IOStatisticsOpenGate();
+	res = workLoop->sleepGate(event, type);
+	IOStatisticsCloseGate();
+	return res;
 }
 
-int IOEventSource::sleepGate(void *event, AbsoluteTime deadline, UInt32 type)
-{ 
-	bool res; 
-	IOStatisticsOpenGate(); 
+int
+IOEventSource::sleepGate(void *event, AbsoluteTime deadline, UInt32 type)
+{
+	int res;
+	IOStatisticsOpenGate();
 	res = workLoop->sleepGate(event, deadline, type);
-	IOStatisticsCloseGate(); 
-	return res; 
+	IOStatisticsCloseGate();
+	return res;
+}
+
+void
+IOEventSource::wakeupGate(void *event, bool oneThread)
+{
+	workLoop->wakeupGate(event, oneThread);
 }
-  
-void IOEventSource::wakeupGate(void *event, bool oneThread) { workLoop->wakeupGate(event, oneThread); }
 
 
-bool IOEventSource::init(OSObject *inOwner,
-                         Action inAction)
+bool
+IOEventSource::init(OSObject *inOwner,
+    Action inAction)
 {
-    if (!inOwner)
-        return false;
+	if (!inOwner) {
+		return false;
+	}
+
+	owner = inOwner;
 
-    owner = inOwner;
+	if (!super::init()) {
+		return false;
+	}
 
-    if ( !super::init() )
-        return false;
+	(void) setAction(inAction);
+	enabled = true;
 
-    (void) setAction(inAction);
-    enabled = true;
+	if (!reserved) {
+		reserved = IONew(ExpansionData, 1);
+		if (!reserved) {
+			return false;
+		}
+	}
 
-    if(!reserved) {
-        reserved = IONew(ExpansionData, 1);
-        if (!reserved) {
-            return false;
-        }
-    }
+	IOStatisticsRegisterCounter();
 
-    IOStatisticsRegisterCounter();
-		
-    return true;
+	return true;
 }
 
-void IOEventSource::free( void )
+void
+IOEventSource::free( void )
 {
-    IOStatisticsUnregisterCounter();
-	
-    if (reserved)
+	IOStatisticsUnregisterCounter();
+
+	if ((kActionBlock & flags) && actionBlock) {
+		Block_release(actionBlock);
+	}
+
+	if (reserved) {
 		IODelete(reserved, ExpansionData, 1);
+	}
+
+	super::free();
+}
+
+void
+IOEventSource::setRefcon(void *newrefcon)
+{
+	refcon = newrefcon;
+}
+
+void *
+IOEventSource::getRefcon() const
+{
+	return refcon;
+}
+
+IOEventSource::Action
+IOEventSource::getAction() const
+{
+	if (kActionBlock & flags) {
+		return NULL;
+	}
+	return action;
+}
 
-    super::free();
+IOEventSource::ActionBlock
+IOEventSource::getActionBlock(ActionBlock) const
+{
+	if (kActionBlock & flags) {
+		return actionBlock;
+	}
+	return NULL;
 }
 
-IOEventSource::Action IOEventSource::getAction () const { return action; };
+void
+IOEventSource::setAction(Action inAction)
+{
+	if ((kActionBlock & flags) && actionBlock) {
+		Block_release(actionBlock);
+	}
+	action = inAction;
+}
 
-void IOEventSource::setAction(Action inAction)
+void
+IOEventSource::setActionBlock(ActionBlock block)
 {
-    action = inAction;
+	if ((kActionBlock & flags) && actionBlock) {
+		Block_release(actionBlock);
+	}
+	actionBlock = Block_copy(block);
+	flags |= kActionBlock;
 }
 
-IOEventSource *IOEventSource::getNext() const { return eventChainNext; };
+IOEventSource *
+IOEventSource::getNext() const
+{
+	return eventChainNext;
+};
 
-void IOEventSource::setNext(IOEventSource *inNext)
+void
+IOEventSource::setNext(IOEventSource *inNext)
 {
-    eventChainNext = inNext;
+	eventChainNext = inNext;
 }
 
-void IOEventSource::enable()
+void
+IOEventSource::enable()
 {
-    enabled = true;
-    if (workLoop)
-        return signalWorkAvailable();
+	enabled = true;
+	if (workLoop) {
+		return signalWorkAvailable();
+	}
 }
 
-void IOEventSource::disable()
+void
+IOEventSource::disable()
 {
-    enabled = false;
+	enabled = false;
 }
 
-bool IOEventSource::isEnabled() const
+bool
+IOEventSource::isEnabled() const
 {
-    return enabled;
+	return enabled;
 }
 
-void IOEventSource::setWorkLoop(IOWorkLoop *inWorkLoop)
+void
+IOEventSource::setWorkLoop(IOWorkLoop *inWorkLoop)
 {
-    if ( !inWorkLoop )
-        disable();
-    workLoop = inWorkLoop;
+	if (!inWorkLoop) {
+		disable();
+	}
+	workLoop = inWorkLoop;
 }
 
-IOWorkLoop *IOEventSource::getWorkLoop() const
+IOWorkLoop *
+IOEventSource::getWorkLoop() const
 {
-    return workLoop;
+	return workLoop;
 }
 
-bool IOEventSource::onThread() const
+bool
+IOEventSource::onThread() const
 {
-    return (workLoop != 0) && workLoop->onThread();
+	return (workLoop != NULL) && workLoop->onThread();
 }