This source file includes following definitions.
- NormalizeAxis
- NormalizeTrigger
- NormalizeXboxOneTrigger
- NormalizeXbox360ButtonData
- NormalizeXboxOneButtonData
- control_endpoint_
- OpenDevice
- SetLEDPattern
- GetVendorId
- GetProductId
- GetControllerType
- WriteComplete
- GotData
- ProcessXbox360Packet
- ProcessXboxOnePacket
- QueueRead
- IOError
- WriteXboxOneInit
- port_
- DeviceAdded
- DeviceRemoved
- RegisterForNotifications
- RegisterForDeviceNotifications
- UnregisterFromNotifications
- ControllerForLocation
- AddController
- RemoveController
- RemoveControllerByLocationID
- XboxControllerGotData
- XboxControllerError
#include "content/browser/gamepad/xbox_data_fetcher_mac.h"
#include <algorithm>
#include <cmath>
#include <limits>
#include <CoreFoundation/CoreFoundation.h>
#include <IOKit/IOCFPlugIn.h>
#include <IOKit/IOKitLib.h>
#include <IOKit/usb/IOUSBLib.h>
#include <IOKit/usb/USB.h>
#include "base/logging.h"
#include "base/mac/foundation_util.h"
namespace {
const int kVendorMicrosoft = 0x045e;
const int kProductXbox360Controller = 0x028e;
const int kProductXboxOneController = 0x02d1;
const int kXbox360ReadEndpoint = 1;
const int kXbox360ControlEndpoint = 2;
const int kXboxOneReadEndpoint = 2;
const int kXboxOneControlEndpoint = 1;
enum {
STATUS_MESSAGE_BUTTONS = 0,
STATUS_MESSAGE_LED = 1,
STATUS_MESSAGE_RUMBLE = 3,
};
enum {
XBOX_ONE_STATUS_MESSAGE_BUTTONS = 0x20,
};
enum {
CONTROL_MESSAGE_SET_RUMBLE = 0,
CONTROL_MESSAGE_SET_LED = 1,
};
#pragma pack(push, 1)
struct Xbox360ButtonData {
bool dpad_up : 1;
bool dpad_down : 1;
bool dpad_left : 1;
bool dpad_right : 1;
bool start : 1;
bool back : 1;
bool stick_left_click : 1;
bool stick_right_click : 1;
bool bumper_left : 1;
bool bumper_right : 1;
bool guide : 1;
bool dummy1 : 1;
bool a : 1;
bool b : 1;
bool x : 1;
bool y : 1;
uint8 trigger_left;
uint8 trigger_right;
int16 stick_left_x;
int16 stick_left_y;
int16 stick_right_x;
int16 stick_right_y;
uint32 dummy2;
uint16 dummy3;
};
struct XboxOneButtonData {
bool sync : 1;
bool dummy1 : 1;
bool start : 1;
bool back : 1;
bool a : 1;
bool b : 1;
bool x : 1;
bool y : 1;
bool dpad_up : 1;
bool dpad_down : 1;
bool dpad_left : 1;
bool dpad_right : 1;
bool bumper_left : 1;
bool bumper_right : 1;
bool stick_left_click : 1;
bool stick_right_click : 1;
uint16 trigger_left;
uint16 trigger_right;
int16 stick_left_x;
int16 stick_left_y;
int16 stick_right_x;
int16 stick_right_y;
};
#pragma pack(pop)
COMPILE_ASSERT(sizeof(Xbox360ButtonData) == 18, xbox_button_data_wrong_size);
COMPILE_ASSERT(sizeof(XboxOneButtonData) == 14, xbox_button_data_wrong_size);
const int16 kLeftThumbDeadzone = 7849;
const int16 kRightThumbDeadzone = 8689;
const uint8 kXbox360TriggerDeadzone = 30;
const uint16 kXboxOneTriggerMax = 1023;
const uint16 kXboxOneTriggerDeadzone = 120;
void NormalizeAxis(int16 x,
int16 y,
int16 deadzone,
float* x_out,
float* y_out) {
float x_val = x;
float y_val = y;
float real_magnitude = std::sqrt(x_val * x_val + y_val * y_val);
if (real_magnitude > deadzone) {
float magnitude = std::min(32767.0f, real_magnitude);
magnitude -= deadzone;
float ratio = (magnitude / (32767 - deadzone)) / real_magnitude;
*x_out = x_val * ratio;
*y_out = -y_val * ratio;
} else {
*x_out = *y_out = 0.0f;
}
}
float NormalizeTrigger(uint8 value) {
return value < kXbox360TriggerDeadzone ? 0 :
static_cast<float>(value - kXbox360TriggerDeadzone) /
(std::numeric_limits<uint8>::max() - kXbox360TriggerDeadzone);
}
float NormalizeXboxOneTrigger(uint16 value) {
return value < kXboxOneTriggerDeadzone ? 0 :
static_cast<float>(value - kXboxOneTriggerDeadzone) /
(kXboxOneTriggerMax - kXboxOneTriggerDeadzone);
}
void NormalizeXbox360ButtonData(const Xbox360ButtonData& data,
XboxController::Data* normalized_data) {
normalized_data->buttons[0] = data.a;
normalized_data->buttons[1] = data.b;
normalized_data->buttons[2] = data.x;
normalized_data->buttons[3] = data.y;
normalized_data->buttons[4] = data.bumper_left;
normalized_data->buttons[5] = data.bumper_right;
normalized_data->buttons[6] = data.back;
normalized_data->buttons[7] = data.start;
normalized_data->buttons[8] = data.stick_left_click;
normalized_data->buttons[9] = data.stick_right_click;
normalized_data->buttons[10] = data.dpad_up;
normalized_data->buttons[11] = data.dpad_down;
normalized_data->buttons[12] = data.dpad_left;
normalized_data->buttons[13] = data.dpad_right;
normalized_data->buttons[14] = data.guide;
normalized_data->triggers[0] = NormalizeTrigger(data.trigger_left);
normalized_data->triggers[1] = NormalizeTrigger(data.trigger_right);
NormalizeAxis(data.stick_left_x,
data.stick_left_y,
kLeftThumbDeadzone,
&normalized_data->axes[0],
&normalized_data->axes[1]);
NormalizeAxis(data.stick_right_x,
data.stick_right_y,
kRightThumbDeadzone,
&normalized_data->axes[2],
&normalized_data->axes[3]);
}
void NormalizeXboxOneButtonData(const XboxOneButtonData& data,
XboxController::Data* normalized_data) {
normalized_data->buttons[0] = data.a;
normalized_data->buttons[1] = data.b;
normalized_data->buttons[2] = data.x;
normalized_data->buttons[3] = data.y;
normalized_data->buttons[4] = data.bumper_left;
normalized_data->buttons[5] = data.bumper_right;
normalized_data->buttons[6] = data.back;
normalized_data->buttons[7] = data.start;
normalized_data->buttons[8] = data.stick_left_click;
normalized_data->buttons[9] = data.stick_right_click;
normalized_data->buttons[10] = data.dpad_up;
normalized_data->buttons[11] = data.dpad_down;
normalized_data->buttons[12] = data.dpad_left;
normalized_data->buttons[13] = data.dpad_right;
normalized_data->buttons[14] = data.sync;
normalized_data->triggers[0] = NormalizeXboxOneTrigger(data.trigger_left);
normalized_data->triggers[1] = NormalizeXboxOneTrigger(data.trigger_right);
NormalizeAxis(data.stick_left_x,
data.stick_left_y,
kLeftThumbDeadzone,
&normalized_data->axes[0],
&normalized_data->axes[1]);
NormalizeAxis(data.stick_right_x,
data.stick_right_y,
kRightThumbDeadzone,
&normalized_data->axes[2],
&normalized_data->axes[3]);
}
}
XboxController::XboxController(Delegate* delegate)
: device_(NULL),
interface_(NULL),
device_is_open_(false),
interface_is_open_(false),
read_buffer_size_(0),
led_pattern_(LED_NUM_PATTERNS),
location_id_(0),
delegate_(delegate),
controller_type_(UNKNOWN_CONTROLLER),
read_endpoint_(0),
control_endpoint_(0) {
}
XboxController::~XboxController() {
if (source_)
CFRunLoopSourceInvalidate(source_);
if (interface_ && interface_is_open_)
(*interface_)->USBInterfaceClose(interface_);
if (device_ && device_is_open_)
(*device_)->USBDeviceClose(device_);
}
bool XboxController::OpenDevice(io_service_t service) {
IOCFPlugInInterface **plugin;
SInt32 score;
kern_return_t kr =
IOCreatePlugInInterfaceForService(service,
kIOUSBDeviceUserClientTypeID,
kIOCFPlugInInterfaceID,
&plugin,
&score);
if (kr != KERN_SUCCESS)
return false;
base::mac::ScopedIOPluginInterface<IOCFPlugInInterface> plugin_ref(plugin);
HRESULT res =
(*plugin)->QueryInterface(plugin,
CFUUIDGetUUIDBytes(kIOUSBDeviceInterfaceID320),
(LPVOID *)&device_);
if (!SUCCEEDED(res) || !device_)
return false;
UInt16 vendor_id;
kr = (*device_)->GetDeviceVendor(device_, &vendor_id);
if (kr != KERN_SUCCESS || vendor_id != kVendorMicrosoft)
return false;
UInt16 product_id;
kr = (*device_)->GetDeviceProduct(device_, &product_id);
if (kr != KERN_SUCCESS)
return false;
IOUSBFindInterfaceRequest request;
switch (product_id) {
case kProductXbox360Controller:
controller_type_ = XBOX_360_CONTROLLER;
read_endpoint_ = kXbox360ReadEndpoint;
control_endpoint_ = kXbox360ControlEndpoint;
request.bInterfaceClass = 255;
request.bInterfaceSubClass = 93;
request.bInterfaceProtocol = 1;
request.bAlternateSetting = kIOUSBFindInterfaceDontCare;
break;
case kProductXboxOneController:
controller_type_ = XBOX_ONE_CONTROLLER;
read_endpoint_ = kXboxOneReadEndpoint;
control_endpoint_ = kXboxOneControlEndpoint;
request.bInterfaceClass = 255;
request.bInterfaceSubClass = 71;
request.bInterfaceProtocol = 208;
request.bAlternateSetting = kIOUSBFindInterfaceDontCare;
break;
default:
return false;
}
kr = (*device_)->USBDeviceOpen(device_);
if (kr != KERN_SUCCESS)
return false;
device_is_open_ = true;
IOUSBConfigurationDescriptorPtr config_desc;
kr = (*device_)->GetConfigurationDescriptorPtr(device_, 0, &config_desc);
if (kr != KERN_SUCCESS)
return false;
kr = (*device_)->SetConfiguration(device_, config_desc->bConfigurationValue);
if (kr != KERN_SUCCESS)
return false;
io_iterator_t iter;
kr = (*device_)->CreateInterfaceIterator(device_, &request, &iter);
if (kr != KERN_SUCCESS)
return false;
base::mac::ScopedIOObject<io_iterator_t> iter_ref(iter);
io_service_t usb_interface = IOIteratorNext(iter);
if (!usb_interface)
return false;
IOCFPlugInInterface **plugin_interface;
kr = IOCreatePlugInInterfaceForService(usb_interface,
kIOUSBInterfaceUserClientTypeID,
kIOCFPlugInInterfaceID,
&plugin_interface,
&score);
if (kr != KERN_SUCCESS || !plugin_interface)
return false;
base::mac::ScopedIOPluginInterface<IOCFPlugInInterface> interface_ref(
plugin_interface);
do {
IOObjectRelease(usb_interface);
} while ((usb_interface = IOIteratorNext(iter)));
res = (*plugin_interface)->QueryInterface(
plugin_interface,
CFUUIDGetUUIDBytes(kIOUSBInterfaceInterfaceID300),
(LPVOID *)&interface_);
if (!SUCCEEDED(res) || !interface_)
return false;
kr = (*interface_)->USBInterfaceOpen(interface_);
if (kr != KERN_SUCCESS)
return false;
interface_is_open_ = true;
CFRunLoopSourceRef source_ref;
kr = (*interface_)->CreateInterfaceAsyncEventSource(interface_, &source_ref);
if (kr != KERN_SUCCESS || !source_ref)
return false;
source_.reset(source_ref);
CFRunLoopAddSource(CFRunLoopGetCurrent(), source_, kCFRunLoopDefaultMode);
uint8 num_endpoints;
kr = (*interface_)->GetNumEndpoints(interface_, &num_endpoints);
if (kr != KERN_SUCCESS || num_endpoints < 2)
return false;
for (int i = 1; i <= 2; i++) {
uint8 direction;
uint8 number;
uint8 transfer_type;
uint16 max_packet_size;
uint8 interval;
kr = (*interface_)->GetPipeProperties(interface_,
i,
&direction,
&number,
&transfer_type,
&max_packet_size,
&interval);
if (kr != KERN_SUCCESS || transfer_type != kUSBInterrupt) {
return false;
}
if (i == read_endpoint_) {
if (direction != kUSBIn)
return false;
read_buffer_.reset(new uint8[max_packet_size]);
read_buffer_size_ = max_packet_size;
QueueRead();
} else if (i == control_endpoint_) {
if (direction != kUSBOut)
return false;
if (controller_type_ == XBOX_ONE_CONTROLLER)
WriteXboxOneInit();
}
}
kr = (*device_)->GetLocationID(device_, &location_id_);
if (kr != KERN_SUCCESS)
return false;
return true;
}
void XboxController::SetLEDPattern(LEDPattern pattern) {
led_pattern_ = pattern;
const UInt8 length = 3;
UInt8* buffer = new UInt8[length];
buffer[0] = static_cast<UInt8>(CONTROL_MESSAGE_SET_LED);
buffer[1] = length;
buffer[2] = static_cast<UInt8>(pattern);
kern_return_t kr = (*interface_)->WritePipeAsync(interface_,
control_endpoint_,
buffer,
(UInt32)length,
WriteComplete,
buffer);
if (kr != KERN_SUCCESS) {
delete[] buffer;
IOError();
return;
}
}
int XboxController::GetVendorId() const {
return kVendorMicrosoft;
}
int XboxController::GetProductId() const {
if (controller_type_ == XBOX_360_CONTROLLER)
return kProductXbox360Controller;
else
return kProductXboxOneController;
}
XboxController::ControllerType XboxController::GetControllerType() const {
return controller_type_;
}
void XboxController::WriteComplete(void* context, IOReturn result, void* arg0) {
UInt8* buffer = static_cast<UInt8*>(context);
delete[] buffer;
if (result != kIOReturnSuccess)
return;
}
void XboxController::GotData(void* context, IOReturn result, void* arg0) {
size_t bytes_read = reinterpret_cast<size_t>(arg0);
XboxController* controller = static_cast<XboxController*>(context);
if (result != kIOReturnSuccess) {
controller->IOError();
return;
}
if (controller->GetControllerType() == XBOX_360_CONTROLLER)
controller->ProcessXbox360Packet(bytes_read);
else
controller->ProcessXboxOnePacket(bytes_read);
controller->QueueRead();
}
void XboxController::ProcessXbox360Packet(size_t length) {
if (length < 2)
return;
DCHECK(length <= read_buffer_size_);
if (length > read_buffer_size_) {
IOError();
return;
}
uint8* buffer = read_buffer_.get();
if (buffer[1] != length)
return;
uint8 type = buffer[0];
buffer += 2;
length -= 2;
switch (type) {
case STATUS_MESSAGE_BUTTONS: {
if (length != sizeof(Xbox360ButtonData))
return;
Xbox360ButtonData* data = reinterpret_cast<Xbox360ButtonData*>(buffer);
Data normalized_data;
NormalizeXbox360ButtonData(*data, &normalized_data);
delegate_->XboxControllerGotData(this, normalized_data);
break;
}
case STATUS_MESSAGE_LED:
if (length != 3)
return;
if (led_pattern_ == LED_NUM_PATTERNS && buffer[0] < LED_NUM_PATTERNS)
led_pattern_ = static_cast<LEDPattern>(buffer[0]);
break;
default:
break;
}
}
void XboxController::ProcessXboxOnePacket(size_t length) {
if (length < 2)
return;
DCHECK(length <= read_buffer_size_);
if (length > read_buffer_size_) {
IOError();
return;
}
uint8* buffer = read_buffer_.get();
uint8 type = buffer[0];
buffer += 4;
length -= 4;
switch (type) {
case XBOX_ONE_STATUS_MESSAGE_BUTTONS: {
if (length != sizeof(XboxOneButtonData))
return;
XboxOneButtonData* data = reinterpret_cast<XboxOneButtonData*>(buffer);
Data normalized_data;
NormalizeXboxOneButtonData(*data, &normalized_data);
delegate_->XboxControllerGotData(this, normalized_data);
break;
}
default:
break;
}
}
void XboxController::QueueRead() {
kern_return_t kr = (*interface_)->ReadPipeAsync(interface_,
read_endpoint_,
read_buffer_.get(),
read_buffer_size_,
GotData,
this);
if (kr != KERN_SUCCESS)
IOError();
}
void XboxController::IOError() {
delegate_->XboxControllerError(this);
}
void XboxController::WriteXboxOneInit() {
const UInt8 length = 2;
UInt8* buffer = new UInt8[length];
buffer[0] = 0x05;
buffer[1] = 0x20;
kern_return_t kr = (*interface_)->WritePipeAsync(interface_,
control_endpoint_,
buffer,
(UInt32)length,
WriteComplete,
buffer);
if (kr != KERN_SUCCESS) {
delete[] buffer;
IOError();
return;
}
}
XboxDataFetcher::XboxDataFetcher(Delegate* delegate)
: delegate_(delegate),
listening_(false),
source_(NULL),
port_(NULL) {
}
XboxDataFetcher::~XboxDataFetcher() {
while (!controllers_.empty()) {
RemoveController(*controllers_.begin());
}
UnregisterFromNotifications();
}
void XboxDataFetcher::DeviceAdded(void* context, io_iterator_t iterator) {
DCHECK(context);
XboxDataFetcher* fetcher = static_cast<XboxDataFetcher*>(context);
io_service_t ref;
while ((ref = IOIteratorNext(iterator))) {
base::mac::ScopedIOObject<io_service_t> scoped_ref(ref);
XboxController* controller = new XboxController(fetcher);
if (controller->OpenDevice(ref)) {
fetcher->AddController(controller);
} else {
delete controller;
}
}
}
void XboxDataFetcher::DeviceRemoved(void* context, io_iterator_t iterator) {
DCHECK(context);
XboxDataFetcher* fetcher = static_cast<XboxDataFetcher*>(context);
io_service_t ref;
while ((ref = IOIteratorNext(iterator))) {
base::mac::ScopedIOObject<io_service_t> scoped_ref(ref);
base::ScopedCFTypeRef<CFNumberRef> number(
base::mac::CFCastStrict<CFNumberRef>(
IORegistryEntryCreateCFProperty(ref,
CFSTR(kUSBDevicePropertyLocationID),
kCFAllocatorDefault,
kNilOptions)));
UInt32 location_id = 0;
CFNumberGetValue(number, kCFNumberSInt32Type, &location_id);
fetcher->RemoveControllerByLocationID(location_id);
}
}
bool XboxDataFetcher::RegisterForNotifications() {
if (listening_)
return true;
port_ = IONotificationPortCreate(kIOMasterPortDefault);
if (!port_)
return false;
source_ = IONotificationPortGetRunLoopSource(port_);
if (!source_)
return false;
CFRunLoopAddSource(CFRunLoopGetCurrent(), source_, kCFRunLoopDefaultMode);
listening_ = true;
if (!RegisterForDeviceNotifications(
kVendorMicrosoft, kProductXboxOneController,
&xbox_one_device_added_iter_,
&xbox_one_device_removed_iter_))
return false;
if (!RegisterForDeviceNotifications(
kVendorMicrosoft, kProductXbox360Controller,
&xbox_360_device_added_iter_,
&xbox_360_device_removed_iter_))
return false;
return true;
}
bool XboxDataFetcher::RegisterForDeviceNotifications(
int vendor_id,
int product_id,
base::mac::ScopedIOObject<io_iterator_t>* added_iter,
base::mac::ScopedIOObject<io_iterator_t>* removed_iter) {
base::ScopedCFTypeRef<CFNumberRef> vendor_cf(CFNumberCreate(
kCFAllocatorDefault, kCFNumberSInt32Type, &vendor_id));
base::ScopedCFTypeRef<CFNumberRef> product_cf(CFNumberCreate(
kCFAllocatorDefault, kCFNumberSInt32Type, &product_id));
base::ScopedCFTypeRef<CFMutableDictionaryRef> matching_dict(
IOServiceMatching(kIOUSBDeviceClassName));
if (!matching_dict)
return false;
CFDictionarySetValue(matching_dict, CFSTR(kUSBVendorID), vendor_cf);
CFDictionarySetValue(matching_dict, CFSTR(kUSBProductID), product_cf);
CFRetain(matching_dict);
io_iterator_t device_added_iter;
IOReturn ret;
ret = IOServiceAddMatchingNotification(port_,
kIOFirstMatchNotification,
matching_dict,
DeviceAdded,
this,
&device_added_iter);
added_iter->reset(device_added_iter);
if (ret != kIOReturnSuccess) {
LOG(ERROR) << "Error listening for Xbox controller add events: " << ret;
return false;
}
DeviceAdded(this, added_iter->get());
CFRetain(matching_dict);
io_iterator_t device_removed_iter;
ret = IOServiceAddMatchingNotification(port_,
kIOTerminatedNotification,
matching_dict,
DeviceRemoved,
this,
&device_removed_iter);
removed_iter->reset(device_removed_iter);
if (ret != kIOReturnSuccess) {
LOG(ERROR) << "Error listening for Xbox controller remove events: " << ret;
return false;
}
DeviceRemoved(this, removed_iter->get());
return true;
}
void XboxDataFetcher::UnregisterFromNotifications() {
if (!listening_)
return;
listening_ = false;
if (source_)
CFRunLoopSourceInvalidate(source_);
if (port_)
IONotificationPortDestroy(port_);
port_ = NULL;
}
XboxController* XboxDataFetcher::ControllerForLocation(UInt32 location_id) {
for (std::set<XboxController*>::iterator i = controllers_.begin();
i != controllers_.end();
++i) {
if ((*i)->location_id() == location_id)
return *i;
}
return NULL;
}
void XboxDataFetcher::AddController(XboxController* controller) {
DCHECK(!ControllerForLocation(controller->location_id()))
<< "Controller with location ID " << controller->location_id()
<< " already exists in the set of controllers.";
controllers_.insert(controller);
delegate_->XboxDeviceAdd(controller);
}
void XboxDataFetcher::RemoveController(XboxController* controller) {
delegate_->XboxDeviceRemove(controller);
controllers_.erase(controller);
delete controller;
}
void XboxDataFetcher::RemoveControllerByLocationID(uint32 location_id) {
XboxController* controller = NULL;
for (std::set<XboxController*>::iterator i = controllers_.begin();
i != controllers_.end();
++i) {
if ((*i)->location_id() == location_id) {
controller = *i;
break;
}
}
if (controller)
RemoveController(controller);
}
void XboxDataFetcher::XboxControllerGotData(XboxController* controller,
const XboxController::Data& data) {
delegate_->XboxValueChanged(controller, data);
}
void XboxDataFetcher::XboxControllerError(XboxController* controller) {
RemoveController(controller);
}