xnu-4903.231.4.tar.gz

[apple/xnu.git] / iokit / Kernel / IOEventSource.cpp

diff --git a/iokit/Kernel/IOEventSource.cpp b/iokit/Kernel/IOEventSource.cpp
index 95046dacd2d7eed49047077ac535a105e7955fd8..76c2d5032cf529bb7966bc102b133f3ef9908cdc 100644 (file)
--- a/iokit/Kernel/IOEventSource.cpp
+++ b/iokit/Kernel/IOEventSource.cpp
@@ -36,6 +36,7 @@ HISTORY
 
 #include <IOKit/IOEventSource.h>
 #include <IOKit/IOWorkLoop.h>
+#include <libkern/Block.h>
 
 #define super OSObject
 
@@ -114,7 +115,7 @@ bool IOEventSource::tryCloseGate()
 
 int IOEventSource::sleepGate(void *event, UInt32 type)
 { 
-       bool res; 
+       int res; 
        IOStatisticsOpenGate(); 
        res = workLoop->sleepGate(event, type); 
        IOStatisticsCloseGate(); 
@@ -123,7 +124,7 @@ int IOEventSource::sleepGate(void *event, UInt32 type)
 
 int IOEventSource::sleepGate(void *event, AbsoluteTime deadline, UInt32 type)
 { 
-       bool res; 
+       int res; 
        IOStatisticsOpenGate(); 
        res = workLoop->sleepGate(event, deadline, type);
        IOStatisticsCloseGate(); 
@@ -162,6 +163,8 @@ bool IOEventSource::init(OSObject *inOwner,
 void IOEventSource::free( void )
 {
     IOStatisticsUnregisterCounter();
+
+       if ((kActionBlock & flags) && actionBlock) Block_release(actionBlock);
        
     if (reserved)
                IODelete(reserved, ExpansionData, 1);
@@ -169,13 +172,41 @@ void IOEventSource::free( void )
     super::free();
 }
 
-IOEventSource::Action IOEventSource::getAction () const { return action; };
+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);
+}
+
+IOEventSource::ActionBlock IOEventSource::getActionBlock(ActionBlock) const
+{
+       if (kActionBlock & flags) return actionBlock;
+       return (NULL);
+}
 
 void IOEventSource::setAction(Action inAction)
 {
+       if ((kActionBlock & flags) && actionBlock) Block_release(actionBlock);
     action = inAction;
 }
 
+void IOEventSource::setActionBlock(ActionBlock block)
+{
+       if ((kActionBlock & flags) && actionBlock) Block_release(actionBlock);
+       actionBlock = Block_copy(block);
+       flags |= kActionBlock;
+}
+
 IOEventSource *IOEventSource::getNext() const { return eventChainNext; };
 
 void IOEventSource::setNext(IOEventSource *inNext)