IOHIDLibUserClient.cpp [plain text]
#include <sys/systm.h>
#include <IOKit/IOCommandGate.h>
#include <IOKit/IOMemoryDescriptor.h>
#include <IOKit/IOService.h>
#include <IOKit/IOSyncer.h>
#include <IOKit/IOWorkLoop.h>
#include "IOHIDLibUserClient.h"
#include "IOHIDDevice.h"
#include "IOHIDEventQueue.h"
#define super IOUserClient
struct AsyncParam {
OSAsyncReference fAsyncRef;
UInt32 fMax;
IOMemoryDescriptor *fMem;
IOHIDReportType reportType;
};
struct AsyncGateParam {
OSAsyncReference asyncRef;
IOHIDReportType reportType;
UInt32 reportID;
void * reportBuffer;
UInt32 reportBufferSize;
UInt32 completionTimeOutMS;
};
OSDefineMetaClassAndStructors(IOHIDLibUserClient, IOUserClient);
const IOExternalMethod IOHIDLibUserClient::
sMethods[kIOHIDLibUserClientNumCommands] = {
{ 0,
(IOMethod) &IOHIDLibUserClient::open,
kIOUCScalarIScalarO,
1,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::close,
kIOUCScalarIScalarO,
0,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::createQueue,
kIOUCScalarIScalarO,
2,
1
},
{ 0,
(IOMethod) &IOHIDLibUserClient::disposeQueue,
kIOUCScalarIScalarO,
1,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::addElementToQueue,
kIOUCScalarIScalarO,
3,
1
},
{ 0,
(IOMethod) &IOHIDLibUserClient::removeElementFromQueue,
kIOUCScalarIScalarO,
2,
1
},
{ 0,
(IOMethod) &IOHIDLibUserClient::queueHasElement,
kIOUCScalarIScalarO,
2,
1
},
{ 0,
(IOMethod) &IOHIDLibUserClient::startQueue,
kIOUCScalarIScalarO,
1,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::stopQueue,
kIOUCScalarIScalarO,
1,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::updateElementValue,
kIOUCScalarIScalarO,
1,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::postElementValue,
kIOUCStructIStructO,
0xffffffff,
0
},
{ 0,
(IOMethod) &IOHIDLibUserClient::getReport,
kIOUCScalarIStructO,
2,
0xffffffff
},
{ 0,
(IOMethod) &IOHIDLibUserClient::getReportOOL,
kIOUCStructIStructO,
sizeof(IOHIDReportReq),
sizeof(UInt32)
},
{ 0,
(IOMethod) &IOHIDLibUserClient::setReport,
kIOUCScalarIStructI,
2,
0xffffffff
},
{ 0,
(IOMethod) &IOHIDLibUserClient::setReportOOL,
kIOUCStructIStructO,
sizeof(IOHIDReportReq),
0
}
};
const IOExternalAsyncMethod IOHIDLibUserClient::
sAsyncMethods[kIOHIDLibUserClientNumAsyncCommands] = {
{ 0,
(IOAsyncMethod) &IOHIDLibUserClient::setAsyncPort,
kIOUCScalarIScalarO,
0,
0
},
{ 0,
(IOAsyncMethod) &IOHIDLibUserClient::setQueueAsyncPort,
kIOUCScalarIScalarO,
1,
0
},
{ 0,
(IOAsyncMethod) &IOHIDLibUserClient::asyncGetReport,
kIOUCScalarIScalarO,
5,
0
},
{ 0,
(IOAsyncMethod) &IOHIDLibUserClient::asyncSetReport,
kIOUCScalarIScalarO,
5,
0
}
};
bool IOHIDLibUserClient::
initWithTask(task_t owningTask, void * , UInt32 )
{
if (!super::init())
return false;
fClient = owningTask;
fGate = 0;
fCachedOptionBits = 0;
task_reference (fClient);
fQueueSet = OSSet::withCapacity(4);
if (!fQueueSet)
return false;
return true;
}
IOReturn IOHIDLibUserClient::clientClose(void)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::clientCloseGated));
}
IOReturn IOHIDLibUserClient::clientCloseGated(void)
{
if (fClient) {
task_deallocate(fClient);
fClient = 0;
}
if (fNub) {
OSCollectionIterator * iterator = OSCollectionIterator::withCollection(fQueueSet);
if (iterator)
{
IOHIDEventQueue * queue;
while (queue = (IOHIDEventQueue *)iterator->getNextObject())
{
fNub->stopEventDelivery(queue);
}
iterator->release();
}
fNub->close(this, fCachedOptionBits);
detach(fNub);
}
return kIOReturnSuccess;
}
bool IOHIDLibUserClient::start(IOService *provider)
{
IOWorkLoop *wl = 0;
if (!super::start(provider))
return false;
fNub = OSDynamicCast(IOHIDDevice, provider);
if (!fNub)
return false;
fGate = 0;
wl = getWorkLoop();
if (!wl)
goto ABORT_START;
fGate = IOCommandGate::commandGate(this);
if (!fGate)
goto ABORT_START;
wl->retain();
wl->addEventSource(fGate);
return true;
ABORT_START:
if (fGate) {
wl->removeEventSource(fGate);
wl->release();
fGate->release();
fGate = 0;
}
return false;
}
IOExternalMethod *IOHIDLibUserClient::
getTargetAndMethodForIndex(IOService **target, UInt32 index)
{
if (index < (UInt32) kIOHIDLibUserClientNumCommands)
{
*target = this;
return (IOExternalMethod *) &sMethods[index];
}
else
return 0;
}
IOExternalAsyncMethod * IOHIDLibUserClient::
getAsyncTargetAndMethodForIndex(IOService **target, UInt32 index)
{
if (index < (UInt32) kIOHIDLibUserClientNumAsyncCommands)
{
*target = this;
return (IOExternalAsyncMethod *) &sAsyncMethods[index];
}
else
return 0;
}
IOReturn IOHIDLibUserClient::
setAsyncPort(OSAsyncReference asyncRef, void *, void *, void *,
void *, void *, void *)
{
fWakePort = (mach_port_t) asyncRef[0];
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::
setQueueAsyncPort(OSAsyncReference asyncRef, void * vInQueue, void *, void *,
void *, void *, void *)
{
IOHIDEventQueue * eventQueue = (IOHIDEventQueue *) vInQueue;
fQueuePort = (mach_port_t) asyncRef[0];
if ( !eventQueue )
return kIOReturnBadArgument;
eventQueue->setNotificationPort(fQueuePort);
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::open(void * flags)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::openGated), flags);
}
IOReturn IOHIDLibUserClient::openGated(void * flags)
{
IOReturn ret = kIOReturnSuccess;
IOOptionBits options = (IOOptionBits)flags;
ret = clientHasPrivilege(fClient, kIOClientPrivilegeLocalUser);
if (ret != kIOReturnSuccess)
{
ret = clientHasPrivilege(fClient, kIOClientPrivilegeAdministrator);
if (ret != kIOReturnSuccess)
return ret;
}
if (!fNub->IOService::open(this, options))
return kIOReturnExclusiveAccess;
fCachedOptionBits = options;
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::close()
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::closeGated));
}
IOReturn IOHIDLibUserClient::closeGated()
{
fNub->close(this, fCachedOptionBits);
return kIOReturnSuccess;
}
bool
IOHIDLibUserClient::didTerminate( IOService * provider, IOOptionBits options, bool * defer )
{
fNub->close(this, fCachedOptionBits);
return super::didTerminate(provider, options, defer);
}
void IOHIDLibUserClient::free()
{
IOWorkLoop *wl;
if (fGate)
{
wl = fGate->getWorkLoop();
if (wl)
wl->release();
fGate->release();
fGate = 0;
}
if (fQueueSet)
{
fQueueSet->release();
fQueueSet = 0;
}
if (fNub)
{
fNub = 0;
}
super::free();
}
IOReturn IOHIDLibUserClient::clientMemoryForType (
UInt32 type,
IOOptionBits * options,
IOMemoryDescriptor ** memory )
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::clientMemoryForTypeGated), (void *)type, (void *)options, (void *)memory);
}
IOReturn IOHIDLibUserClient::clientMemoryForTypeGated(
UInt32 type,
IOOptionBits * options,
IOMemoryDescriptor ** memory )
{
IOReturn ret = kIOReturnNoMemory;
IOMemoryDescriptor * memoryToShare = NULL;
IOHIDEventQueue * queue = NULL;
if (type == IOHIDLibUserClientElementValuesType)
{
if (fNub && !isInactive())
memoryToShare = fNub->getMemoryWithCurrentElementValues();
}
else if (queue = OSDynamicCast(IOHIDEventQueue, (OSObject *)type))
{
memoryToShare = queue->getMemoryDescriptor();
}
if (memoryToShare)
{
memoryToShare->retain();
ret = kIOReturnSuccess;
}
*options = 0;
*memory = memoryToShare;
return ret;
}
IOReturn IOHIDLibUserClient::createQueue(void * vInFlags, void * vInDepth, void * vOutQueue)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::createQueueGated), vInFlags, vInDepth, vOutQueue);
}
IOReturn IOHIDLibUserClient::createQueueGated(void * vInFlags, void * vInDepth, void * vOutQueue)
{
UInt32 depth = (UInt32) vInDepth;
void ** outQueue = (void **) vOutQueue;
IOHIDEventQueue * eventQueue = IOHIDEventQueue::withEntries (depth+1, DEFAULT_HID_ENTRY_SIZE);
*outQueue = eventQueue;
fQueueSet->setObject(eventQueue);
eventQueue->release();
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::disposeQueue(void * vInQueue)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::disposeQueueGated), vInQueue);
}
IOReturn IOHIDLibUserClient::disposeQueueGated(void * vInQueue)
{
IOReturn ret = kIOReturnSuccess;
IOHIDEventQueue * queue = (IOHIDEventQueue *) vInQueue;
if (fNub && !isInactive())
ret = fNub->stopEventDelivery (queue);
fQueueSet->removeObject(queue);
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::addElementToQueue(void * vInQueue, void * vInElementCookie, void * vInFlags, void *vSizeChange)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::addElementToQueueGated), vInQueue, vInElementCookie, vInFlags, vSizeChange);
}
IOReturn IOHIDLibUserClient::addElementToQueueGated(void * vInQueue, void * vInElementCookie, void * vInFlags, void *vSizeChange)
{
IOReturn ret = kIOReturnSuccess;
UInt32 size = 0;
int * sizeChange = (int *) vSizeChange;
IOHIDEventQueue * queue = (IOHIDEventQueue *) vInQueue;
IOHIDElementCookie elementCookie = (IOHIDElementCookie) vInElementCookie;
size = (queue) ? queue->getEntrySize() : 0;
if (fNub && !isInactive())
ret = fNub->startEventDelivery (queue, elementCookie);
*sizeChange = (queue && (size != queue->getEntrySize()));
return ret;
}
IOReturn IOHIDLibUserClient::removeElementFromQueue (void * vInQueue, void * vInElementCookie, void * vSizeChange)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::removeElementFromQueueGated), vInQueue, vInElementCookie, vSizeChange);
}
IOReturn IOHIDLibUserClient::removeElementFromQueueGated (void * vInQueue, void * vInElementCookie, void * vSizeChange)
{
IOReturn ret = kIOReturnSuccess;
UInt32 size = 0;
int * sizeChange = (int *) vSizeChange;
IOHIDEventQueue * queue = (IOHIDEventQueue *) vInQueue;
IOHIDElementCookie elementCookie = (IOHIDElementCookie) vInElementCookie;
size = (queue) ? queue->getEntrySize() : 0;
if (fNub && !isInactive())
ret = fNub->stopEventDelivery (queue, elementCookie);
*sizeChange = (queue && (size != queue->getEntrySize()));
return ret;
}
IOReturn IOHIDLibUserClient::queueHasElement (void * vInQueue, void * vInElementCookie, void * vOutHasElement)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::queueHasElementGated), vInQueue, vInElementCookie, vOutHasElement);
}
IOReturn IOHIDLibUserClient::queueHasElementGated (void * vInQueue, void * vInElementCookie, void * vOutHasElement)
{
IOReturn ret = kIOReturnSuccess;
IOHIDEventQueue * queue = (IOHIDEventQueue *) vInQueue;
IOHIDElementCookie elementCookie = (IOHIDElementCookie) vInElementCookie;
int * outHasElement = (int *) vOutHasElement;
bool hasElement = false;
if (fNub && !isInactive())
ret = fNub->checkEventDelivery (queue, elementCookie, &hasElement);
*outHasElement = hasElement;
return ret;
}
IOReturn IOHIDLibUserClient::startQueue (void * vInQueue)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::startQueueGated), vInQueue);
}
IOReturn IOHIDLibUserClient::startQueueGated (void * vInQueue)
{
IOHIDEventQueue * queue = (IOHIDEventQueue *) vInQueue;
queue->start();
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::stopQueue (void * vInQueue)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::stopQueueGated), vInQueue);
}
IOReturn IOHIDLibUserClient::stopQueueGated (void * vInQueue)
{
IOHIDEventQueue * queue = (IOHIDEventQueue *) vInQueue;
queue->stop();
return kIOReturnSuccess;
}
IOReturn IOHIDLibUserClient::updateElementValue (void * cookie)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::updateElementValueGated), cookie);
}
IOReturn IOHIDLibUserClient::updateElementValueGated (void * cookie)
{
IOReturn ret = kIOReturnError;
if (fNub && !isInactive())
ret = fNub->updateElementValues(&cookie, 1);
return ret;
}
IOReturn IOHIDLibUserClient::postElementValue (void * cookies, void * cookiesBytes)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::postElementValueGated), cookies, cookiesBytes);
}
IOReturn IOHIDLibUserClient::postElementValueGated (void * cookies, void * cookiesBytes)
{
IOReturn ret = kIOReturnError;
UInt32 numCookies = ((UInt32)cookiesBytes) / sizeof(UInt32);
if (fNub && !isInactive())
ret = fNub->postElementValues((IOHIDElementCookie *)cookies, numCookies);
return ret;
}
IOReturn IOHIDLibUserClient::getReport (void *vReportType, void *vReportID, void *vReportBuffer, void *vReportBufferSize)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::getReportGated), vReportType, vReportID, vReportBuffer, vReportBufferSize);
}
IOReturn IOHIDLibUserClient::getReportGated (IOHIDReportType reportType,
UInt32 reportID,
void *reportBuffer,
UInt32 *reportBufferSize)
{
IOReturn ret;
IOMemoryDescriptor * mem;
if (fNub && !isInactive())
{
mem = IOMemoryDescriptor::withAddress(reportBuffer, *reportBufferSize, kIODirectionIn);
if(mem)
{
*reportBufferSize = 0;
ret = fNub->getReport(mem, reportType, reportID);
if (ret == kIOReturnSuccess)
fNub->handleReport(mem, reportType, kIOHIDReportOptionNotInterrupt);
*reportBufferSize = mem->getLength();
mem->release();
}
else
ret = kIOReturnNoMemory;
}
else
ret = kIOReturnNotAttached;
return ret;
}
IOReturn IOHIDLibUserClient::getReportOOL( void *vReqIn, void *vSizeOut, void * vInCount, void *vOutCount)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::getReportOOLGated), vReqIn, vSizeOut, vInCount, vOutCount);
}
IOReturn IOHIDLibUserClient::getReportOOLGated( IOHIDReportReq *reqIn,
UInt32 *sizeOut,
IOByteCount inCount,
IOByteCount *outCount)
{
IOReturn ret;
IOMemoryDescriptor * mem;
if (fNub && !isInactive())
{
*sizeOut = 0;
mem = IOMemoryDescriptor::withAddress((vm_address_t)reqIn->reportBuffer, reqIn->reportBufferSize, kIODirectionIn, fClient);
if(mem)
{
ret = mem->prepare();
if(ret == kIOReturnSuccess)
ret = fNub->getReport(mem, (IOHIDReportType)(reqIn->reportType), reqIn->reportID);
if (ret == kIOReturnSuccess)
fNub->handleReport(mem, (IOHIDReportType)(reqIn->reportType), kIOHIDReportOptionNotInterrupt);
*sizeOut = mem->getLength();
mem->complete();
mem->release();
}
else
ret = kIOReturnNoMemory;
}
else
ret = kIOReturnNotAttached;
return ret;
}
IOReturn IOHIDLibUserClient::setReport (void *vReportType, void *vReportID, void *vReportBuffer, void *vReportBufferSize)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::setReportGated), vReportType, vReportID, vReportBuffer, vReportBufferSize);
}
IOReturn IOHIDLibUserClient::setReportGated( IOHIDReportType reportType,
UInt32 reportID,
void *reportBuffer,
UInt32 reportBufferSize)
{
IOReturn ret;
IOMemoryDescriptor * mem;
if (fNub && !isInactive())
{
mem = IOMemoryDescriptor::withAddress(reportBuffer, reportBufferSize, kIODirectionOut);
if(mem)
{
ret = fNub->setReport(mem, reportType, reportID);
if (ret == kIOReturnSuccess)
fNub->handleReport(mem, reportType, kIOHIDReportOptionNotInterrupt);
mem->release();
}
else
ret = kIOReturnNoMemory;
}
else
ret = kIOReturnNotAttached;
return ret;
}
IOReturn IOHIDLibUserClient::setReportOOL (void *vReq, void *vInCount)
{
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::setReportOOLGated), vReq, vInCount);
}
IOReturn IOHIDLibUserClient::setReportOOLGated (IOHIDReportReq *req, IOByteCount inCount)
{
IOReturn ret;
IOMemoryDescriptor * mem;
if (fNub && !isInactive())
{
mem = IOMemoryDescriptor::withAddress((vm_address_t)req->reportBuffer, req->reportBufferSize, kIODirectionOut, fClient);
if(mem)
{
ret = mem->prepare();
if(ret == kIOReturnSuccess)
ret = fNub->setReport(mem, (IOHIDReportType)(req->reportType), req->reportID);
if (ret == kIOReturnSuccess)
fNub->handleReport(mem, (IOHIDReportType)(req->reportType), kIOHIDReportOptionNotInterrupt);
mem->complete();
mem->release();
}
else
ret = kIOReturnNoMemory;
}
else
ret = kIOReturnNotAttached;
return ret;
}
IOReturn IOHIDLibUserClient::asyncGetReport(OSAsyncReference asyncRef,
IOHIDReportType reportType,
UInt32 reportID,
void *reportBuffer,
UInt32 reportBufferSize,
UInt32 completionTimeOutMS)
{
AsyncGateParam param;
bcopy(asyncRef, param.asyncRef, sizeof(OSAsyncReference));
param.reportType = reportType;
param.reportID = reportID;
param.reportBuffer = reportBuffer;
param.reportBufferSize = reportBufferSize;
param.completionTimeOutMS = completionTimeOutMS;
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::asyncGetReportGated), ¶m);
}
IOReturn IOHIDLibUserClient::asyncGetReportGated ( void * param )
{
IOHIDReportType reportType = ((AsyncGateParam *)param)->reportType;
UInt32 reportID = ((AsyncGateParam *)param)->reportID;
void * reportBuffer = ((AsyncGateParam *)param)->reportBuffer;
UInt32 reportBufferSize = ((AsyncGateParam *)param)->reportBufferSize;
UInt32 completionTimeOutMS = ((AsyncGateParam *)param)->completionTimeOutMS;
IOReturn ret;
IOHIDCompletion tap;
IOMemoryDescriptor * mem = NULL;
AsyncParam * pb = NULL;
retain();
if (fNub && !isInactive())
{
do {
mem = IOMemoryDescriptor::withAddress((vm_address_t)reportBuffer, reportBufferSize, kIODirectionIn, fClient);
if(!mem)
{
ret = kIOReturnNoMemory;
break;
}
ret = mem->prepare();
if(ret != kIOReturnSuccess)
break;
pb = (AsyncParam *)IOMalloc(sizeof(AsyncParam));
if(!pb)
{
ret = kIOReturnNoMemory;
break;
}
bcopy(((AsyncGateParam *)param)->asyncRef, pb->fAsyncRef, sizeof(OSAsyncReference));
pb->fMax = reportBufferSize;
pb->fMem = mem;
pb->reportType = reportType;
tap.target = this;
tap.action = OSMemberFunctionCast(IOHIDCompletionAction, this, &IOHIDLibUserClient::ReqComplete);
tap.parameter = pb;
ret = fNub->getReport(mem, reportType, reportID, completionTimeOutMS, &tap);
} while (false);
}
else
ret = kIOReturnNotAttached;
if(ret != kIOReturnSuccess)
{
if(mem)
{
mem->complete();
mem->release();
}
if(pb)
IOFree(pb, sizeof(*pb));
release();
}
return ret;
}
IOReturn IOHIDLibUserClient::asyncSetReport(OSAsyncReference asyncRef,
IOHIDReportType reportType,
UInt32 reportID,
void *reportBuffer,
UInt32 reportBufferSize,
UInt32 completionTimeOutMS)
{
AsyncGateParam param;
bcopy(asyncRef, param.asyncRef, sizeof(OSAsyncReference));
param.reportType = reportType;
param.reportID = reportID;
param.reportBuffer = reportBuffer;
param.reportBufferSize = reportBufferSize;
param.completionTimeOutMS = completionTimeOutMS;
return fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::asyncSetReportGated), ¶m);
}
IOReturn IOHIDLibUserClient::asyncSetReportGated ( void * param )
{
IOHIDReportType reportType = ((AsyncGateParam *)param)->reportType;
UInt32 reportID = ((AsyncGateParam *)param)->reportID;
void * reportBuffer = ((AsyncGateParam *)param)->reportBuffer;
UInt32 reportBufferSize = ((AsyncGateParam *)param)->reportBufferSize;
UInt32 completionTimeOutMS = ((AsyncGateParam *)param)->completionTimeOutMS;
IOReturn ret;
IOHIDCompletion tap;
IOMemoryDescriptor * mem = NULL;
AsyncParam * pb = NULL;
retain();
if (fNub && !isInactive())
{
do {
mem = IOMemoryDescriptor::withAddress((vm_address_t)reportBuffer, reportBufferSize, kIODirectionOut, fClient);
if(!mem)
{
ret = kIOReturnNoMemory;
break;
}
ret = mem->prepare();
if(ret != kIOReturnSuccess)
break;
pb = (AsyncParam *)IOMalloc(sizeof(AsyncParam));
if(!pb)
{
ret = kIOReturnNoMemory;
break;
}
bcopy(((AsyncGateParam *)param)->asyncRef, pb->fAsyncRef, sizeof(OSAsyncReference));
pb->fMax = reportBufferSize;
pb->fMem = mem;
pb->reportType = reportType;
tap.target = this;
tap.action = OSMemberFunctionCast(IOHIDCompletionAction, this, &IOHIDLibUserClient::ReqComplete);
tap.parameter = pb;
ret = fNub->setReport(mem, reportType, reportID, completionTimeOutMS, &tap);
} while (false);
}
else
ret = kIOReturnNotAttached;
if(ret != kIOReturnSuccess)
{
if(mem)
{
mem->complete();
mem->release();
}
if(pb)
IOFree(pb, sizeof(*pb));
release();
}
return ret;
}
void IOHIDLibUserClient::ReqComplete(void *param, IOReturn res, UInt32 remaining)
{
fGate->runAction(OSMemberFunctionCast(IOCommandGate::Action, this, &IOHIDLibUserClient::ReqCompleteGated), param, (void *)res, (void *)remaining);
}
void IOHIDLibUserClient::ReqCompleteGated(void *param, IOReturn res, UInt32 remaining)
{
void * args[1];
AsyncParam * pb = (AsyncParam *)param;
if(res == kIOReturnSuccess)
{
args[0] = (void *)(pb->fMax - remaining);
if (fNub && !isInactive())
fNub->handleReport(pb->fMem, pb->reportType, kIOHIDReportOptionNotInterrupt);
}
else
{
args[0] = 0;
}
if (pb->fMem)
{
pb->fMem->complete();
pb->fMem->release();
}
sendAsyncResult(pb->fAsyncRef, res, args, 1);
IOFree(pb, sizeof(*pb));
release();
}