#include <unistd.h>
#include <sys/types.h>
#include <CoreFoundation/CoreFoundation.h>
#include <libkern/OSByteOrder.h>
#include <bootstrap_priv.h>
#include <mach/mach.h>
#include <IOKit/IOKitLib.h>
#include <IOKit/hidsystem/IOHIDLib.h>
#include <IOKit/hid/IOHIDLibPrivate.h>
#include <IOKit/pwr_mgt/IOPMLibPrivate.h>
#include <servers/bootstrap.h>
#include "powermanagement.h"
#include <IOKit/hid/IOHIDEventSystemClient.h>
#include <IOKit/hid/IOHIDServiceKeys.h>
#include <os/log.h>
#if !TARGET_OS_IPHONE
kern_return_t IOFramebufferServerStart( void );
#endif
kern_return_t
IOHIDCreateSharedMemory( io_connect_t connect,
unsigned int version )
{
#if !TARGET_OS_IPHONE
IOFramebufferServerStart();
#endif
uint64_t inData = version;
return IOConnectCallMethod( connect, 0, &inData, 1, NULL, 0, NULL, NULL, NULL, NULL); }
kern_return_t
IOHIDSetEventsEnable( io_connect_t connect,
boolean_t enable )
{
uint64_t inData = enable;
return IOConnectCallMethod( connect, 1, &inData, 1, NULL, 0, NULL, NULL, NULL, NULL); }
kern_return_t
IOHIDSetCursorEnable( io_connect_t connect,
boolean_t enable )
{
uint64_t inData = enable;
return IOConnectCallMethod( connect, 2, &inData, 1, NULL, 0, NULL, NULL, NULL, NULL); }
static bool _IOPMReportSoftwareHIDEvent(UInt32 eventType)
{
mach_port_t newConnection;
kern_return_t kern_result = KERN_SUCCESS;
int allowEvent = true;
kern_result = bootstrap_look_up2(bootstrap_port,
kIOPMServerBootstrapName,
&newConnection,
0,
BOOTSTRAP_PRIVILEGED_SERVER);
if(KERN_SUCCESS == kern_result) {
io_pm_hid_event_report_activity(newConnection, eventType, &allowEvent);
mach_port_deallocate(mach_task_self(), newConnection);
}
return allowEvent;
}
kern_return_t
IOHIDPostEvent( io_connect_t connect,
UInt32 eventType,
IOGPoint location,
const NXEventData * eventData,
UInt32 eventDataVersion,
IOOptionBits eventFlags,
IOOptionBits options )
{
int * eventPid = 0;
size_t dataSize = sizeof(struct evioLLEvent) + sizeof(int);
char data[dataSize];
struct evioLLEvent* event;
UInt32 eventDataSize = sizeof(NXEventData);
int allowEvent = true;
bzero(data, dataSize);
event = (struct evioLLEvent*)data;
event->type = eventType;
event->location = location;
event->flags = eventFlags;
event->setFlags = options;
event->setCursor = options & (kIOHIDSetCursorPosition | kIOHIDSetRelativeCursorPosition);
eventPid = (int *)(event + 1);
*eventPid = getpid();
if ( eventDataVersion < 2 )
{
eventDataSize = 32;
event->setCursor = eventDataVersion; }
if ( eventDataSize < sizeof(event->data) )
{
bcopy( eventData, &(event->data), eventDataSize );
bzero( ((UInt8 *)(&(event->data))) + eventDataSize,
sizeof(event->data) - eventDataSize );
}
else
bcopy( eventData, &event->data, sizeof(event->data) );
allowEvent = _IOPMReportSoftwareHIDEvent(event->type);
if (allowEvent) {
return IOConnectCallMethod(connect, 3, NULL, 0, data, dataSize, NULL, NULL, NULL, NULL); }
else {
return KERN_SUCCESS;
}
}
extern kern_return_t
IOHIDSetCursorBounds( io_connect_t connect, const IOGBounds * bounds )
{
if ( !bounds )
return kIOReturnBadArgument;
return IOConnectCallMethod(connect, 6, NULL, 0, bounds, sizeof(*bounds), NULL, NULL, NULL, NULL); }
extern kern_return_t
IOHIDSetOnScreenCursorBounds( io_connect_t connect, const IOGPoint * point, const IOGBounds * bounds )
{
if ( !bounds || !point )
return kIOReturnBadArgument;
int16_t data[6] = {point->x, point->y, bounds->minx, bounds->miny, bounds->maxx, bounds->maxy};
return IOConnectCallMethod(connect, 12, NULL, 0, data, sizeof(data), NULL, NULL, NULL, NULL); }
kern_return_t
IOHIDSetMouseLocation( io_connect_t connect, int x, int y )
{
NXEventData event;
IOGPoint location = {x,y};
memset(&event, 0, sizeof(event));
return IOHIDPostEvent(connect, NX_NULLEVENT, location, &event, 2, -1, kIOHIDSetCursorPosition);
}
kern_return_t
IOHIDSetFixedMouseLocation( io_connect_t connect, int32_t x, int32_t y )
{
int32_t data[3] = {x, y, 0};
data[2] = getpid();
return IOConnectCallMethod(connect, 4, NULL, 0, data, sizeof(data), NULL, NULL, NULL, NULL); }
kern_return_t
IOHIDGetButtonEventNum( io_connect_t connect,
NXMouseButton button, int * eventNum )
{
kern_return_t err;
uint64_t inData = button;
uint64_t outData;
uint32_t outSize = 1;
err = IOConnectCallMethod(connect, 5, &inData, 1, NULL, 0, &outData, &outSize, NULL, NULL); *eventNum = (int) outData;
return( err);
}
kern_return_t
IOHIDGetStateForSelector( io_connect_t handle, int selector, UInt32 *state )
{
kern_return_t err;
uint64_t inData[1] = {selector};
uint64_t outData[1] = {0};
uint32_t outCount = 1;
err = IOConnectCallMethod(handle, 5, inData, 1, NULL, 0, outData, &outCount, NULL, NULL);
*state = outData[0];
return err;
}
kern_return_t
IOHIDSetStateForSelector( io_connect_t handle, int selector, UInt32 state )
{
kern_return_t err;
uint64_t inData[2] = {selector, state};
uint32_t outCount = 0;
err = IOConnectCallMethod(handle, 6, inData, 2, NULL, 0, NULL, &outCount, NULL, NULL);
return err;
}
kern_return_t
IOHIDGetModifierLockState( io_connect_t handle __unused, int selector, bool *state )
{
kern_return_t err = kIOReturnSuccess;
boolean_t modifierState = false;
IOHIDEventSystemClientRef client = NULL;
CFArrayRef services = NULL;
if (selector != kIOHIDCapsLockState && selector != kIOHIDNumLockState) {
err = kIOReturnBadArgument;
goto exit;
}
client = IOHIDEventSystemClientCreateWithType (kCFAllocatorDefault, kIOHIDEventSystemClientTypePassive, NULL);
if (client == NULL) {
os_log_error (OS_LOG_DEFAULT, "Failed to create event system client");
return kIOReturnError;
}
services = (CFArrayRef) IOHIDEventSystemClientCopyServices (client);
if (services == NULL || CFArrayGetCount(services) == 0) {
err = kIOReturnNoDevice;
goto exit;
}
for (CFIndex index = 0; index < CFArrayGetCount(services); index++) {
IOHIDServiceClientRef service = (IOHIDServiceClientRef) CFArrayGetValueAtIndex(services, index);
if (service && IOHIDServiceClientConformsTo (service, kHIDPage_GenericDesktop, kHIDUsage_GD_Keyboard)) {
CFStringRef key = (selector == kIOHIDCapsLockState) ? CFSTR(kIOHIDServiceCapsLockStateKey) : CFSTR(kIOHIDServiceNumLockStateKey);
CFBooleanRef modifierStateProp = (CFBooleanRef)IOHIDServiceClientCopyProperty(service, key);
if (modifierStateProp) {
modifierState = CFBooleanGetValue(modifierStateProp);
if (modifierState) {
break;
}
CFRelease(modifierStateProp);
}
}
}
exit:
*state = modifierState;
if (services) {
CFRelease(services);
}
if (client) {
CFRelease(client);
}
return err;
}
kern_return_t
IOHIDSetModifierLockState( io_connect_t handle __unused, int selector, bool state )
{
kern_return_t err = kIOReturnSuccess;
IOHIDEventSystemClientRef client = NULL;
CFArrayRef services = NULL;
if (selector != kIOHIDCapsLockState && selector != kIOHIDNumLockState) {
err = kIOReturnBadArgument;
goto exit;
}
client = IOHIDEventSystemClientCreateWithType (kCFAllocatorDefault, kIOHIDEventSystemClientTypePassive, NULL);
if (client == NULL) {
os_log_error (OS_LOG_DEFAULT, "Failed to create event system client");
return kIOReturnError;
}
services = (CFArrayRef) IOHIDEventSystemClientCopyServices (client);
if (services == NULL || CFArrayGetCount(services) == 0) {
err = kIOReturnNoDevice;
goto exit;
}
for (CFIndex index = 0; index < CFArrayGetCount(services); index++) {
IOHIDServiceClientRef service = (IOHIDServiceClientRef) CFArrayGetValueAtIndex(services, index);
if (service && IOHIDServiceClientConformsTo (service, kHIDPage_GenericDesktop, kHIDUsage_GD_Keyboard)) {
CFStringRef key = (selector == kIOHIDCapsLockState) ? CFSTR(kIOHIDServiceCapsLockStateKey) : CFSTR(kIOHIDServiceNumLockStateKey);
IOHIDServiceClientSetProperty(service, key, state ? kCFBooleanTrue : kCFBooleanFalse);
}
}
exit:
if (services) {
CFRelease(services);
}
if (client) {
CFRelease(client);
}
return err;
}
kern_return_t
IOHIDRegisterVirtualDisplay( io_connect_t handle, UInt32 *display_token )
{
kern_return_t err;
uint64_t outData[1] = {0};
uint32_t outCount = 1;
err = IOConnectCallMethod(handle, 7, NULL, 0, NULL, 0, outData, &outCount, NULL, NULL); *display_token = outData[0];
return err;
}
kern_return_t
IOHIDUnregisterVirtualDisplay( io_connect_t handle, UInt32 display_token )
{
kern_return_t err;
uint64_t inData[1] = {display_token};
uint32_t outCount = 0;
err = IOConnectCallMethod(handle, 8, inData, 1, NULL, 0, NULL, &outCount, NULL, NULL);
return err;
}
kern_return_t
IOHIDSetVirtualDisplayBounds( io_connect_t handle, UInt32 display_token, const IOGBounds * bounds )
{
kern_return_t err;
uint64_t inData[5] = {display_token, bounds->minx, bounds->maxx, bounds->miny, bounds->maxy};
uint32_t outCount = 0;
err = IOConnectCallMethod(handle, 9, inData, 5, NULL, 0, NULL, &outCount, NULL, NULL);
return err;
}
kern_return_t
IOHIDGetActivityState( io_connect_t handle, bool *hidActivityIdle )
{
kern_return_t err;
uint64_t outData[1] = {0};
uint32_t outCount = 1;
if (!hidActivityIdle) return kIOReturnBadArgument;
err = IOConnectCallMethod(handle, 10, NULL, 0, NULL, 0, outData, &outCount, NULL, NULL); *hidActivityIdle = outData[0] ? true : false;
return err;
}