-
Notifications
You must be signed in to change notification settings - Fork 0
/
xpad360c.h
365 lines (296 loc) · 8.54 KB
/
xpad360c.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
/*
TODO:
Headsets
NOTES:
I'm forgetting the idea that the controller is HID compliant.
While other drivers do use a filter driver for HID, we do not as it's inconvenient.
It doesn't make anything less painful, especially since Linux doesn't have a HID filter driver
and we'd have to provide our own HID descriptor.
All allocation functions also initialize (which is always what we want).
*/
#pragma once
#include <linux/spinlock.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/usb/input.h>
enum xpad360c_led_t{
XPAD360_LED_OFF,
XPAD360_LED_ALL_BLINKING,
XPAD360_LED_FLASH_ON_1,
XPAD360_LED_FLASH_ON_2,
XPAD360_LED_FLASH_ON_3,
XPAD360_LED_FLASH_ON_4,
XPAD360_LED_ON_1,
XPAD360_LED_ON_2,
XPAD360_LED_ON_3,
XPAD360_LED_ON_4,
XPAD360_LED_ROTATING,
XPAD360_LED_SECTIONAL_BLINKING,
XPAD360_LED_SLOW_SECTIONAL_BLINKING,
XPAD360_LED_ALTERNATING
};
struct packet_work {
struct work_struct work;
struct urb *urb;
struct input_dev *inputdev;
};
/* Our main structure.
Only oddball here is the out urb.
It's implicitly readonly after initialization.
*/
struct xpad360_controller {
struct input_dev *inputdev;
struct urb *in;
struct urb *out;
struct usb_anchor out_anchor;
char path[64];
};
static inline int xpad360c_check_urb(struct urb *urb)
{
struct device *device = &urb->dev->dev;
switch (urb->status) {
case 0:
return true;
case -ECONNRESET:
dev_dbg(device, "Controller has been reset.\n");
break;
case -ESHUTDOWN:
dev_dbg(device, "Controller has shutdown.\n");
break;
case -ENOENT:
dev_dbg(device, "Controller has been poisoned.\n");
break;
default:
dev_dbg(device, "Unknown status returned by controller: %x\n", urb->status);
}
return false;
}
static int xpad360c_controller_open(struct input_dev* inputdev)
{ return 0; }
static void xpad360c_controller_close(struct input_dev* inputdev)
{}
static void xpad360c_input_capabilities(struct input_dev *inputdev)
{
/* TODO: Fix this ugly mess. Send all data at once.
That would be cleaner and much more efficient. */
#define SET_BIT(type) __set_bit(type, inputdev->keybit);
/* Buttons */
__set_bit(EV_KEY, inputdev->evbit);
SET_BIT(BTN_A);
SET_BIT(BTN_B);
SET_BIT(BTN_X);
SET_BIT(BTN_Y);
SET_BIT(BTN_START);
SET_BIT(BTN_SELECT);
SET_BIT(BTN_THUMBL);
SET_BIT(BTN_THUMBR);
SET_BIT(BTN_TL);
SET_BIT(BTN_TR);
SET_BIT(BTN_MODE);
#undef SET_BIT
#define SET_BIT(type) \
__set_bit(type, inputdev->absbit);\
input_set_abs_params(inputdev, type, -32768, 32767, 16, 128);
/* Axis (Sticks)*/
__set_bit(EV_ABS, inputdev->evbit);
SET_BIT(ABS_X);
SET_BIT(ABS_Y);
SET_BIT(ABS_RX);
SET_BIT(ABS_RY);
#undef SET_BIT
#define SET_BIT(type) \
__set_bit(type, inputdev->absbit); \
input_set_abs_params(inputdev, type, 0, 255, 0, 0);
/* Triggers */
SET_BIT(ABS_Z);
SET_BIT(ABS_RZ);
#undef SET_BIT
/* Force Feedback */
__set_bit(EV_FF, inputdev->evbit);
__set_bit(FF_RUMBLE, inputdev->ffbit);
}
void xpad360c_allocate_inputdev(
struct xpad360_controller *controller,
struct usb_device *usbdev,
const char* name,
const char* path)
{
struct input_dev * inputdev;
if (controller->inputdev)
return;
inputdev = devm_input_allocate_device(&usbdev->dev);
if (!inputdev) return;
inputdev->name = name;
inputdev->phys = path;
inputdev->open = xpad360c_controller_open;
inputdev->close = xpad360c_controller_close;
xpad360c_input_capabilities(inputdev);
usb_to_input_id(usbdev, &inputdev->id);
controller->inputdev = inputdev;
}
void xpad360c_destroy_inputdev(struct xpad360_controller *controller)
{
input_unregister_device(controller->inputdev);
controller->inputdev = NULL;
}
/*
* This function is similar for all 360 controllers, only with different offsets.
* Anything uncommon is dealt with in specific modules.
* Each specific module has to deal with its own quirks.
*/
void xpad360c_parse_input(struct input_dev *inputdev, void *_data)
{
u8 *data = _data;
/* start/back buttons */
input_report_key(inputdev, BTN_START, data[0] & 0x10);
input_report_key(inputdev, BTN_SELECT, data[0] & 0x20); /* Back */
/* stick press left/right */
input_report_key(inputdev, BTN_THUMBL, data[0] & 0x40);
input_report_key(inputdev, BTN_THUMBR, data[0] & 0x80);
input_report_key(inputdev, BTN_TL, data[1] & 0x01); /* Left Shoulder */
input_report_key(inputdev, BTN_TR, data[1] & 0x02); /* Right Shoulder */
input_report_key(inputdev, BTN_MODE, data[1] & 0x04); /* Guide */
/* data[8] & 0x08 is a dummy value */
input_report_key(inputdev, BTN_A, data[1] & 0x10);
input_report_key(inputdev, BTN_B, data[1] & 0x20);
input_report_key(inputdev, BTN_X, data[1] & 0x40);
input_report_key(inputdev, BTN_Y, data[1] & 0x80);
input_report_abs(inputdev, ABS_Z, data[2]);
input_report_abs(inputdev, ABS_RZ, data[3]);
/* Left Stick */
input_report_abs(inputdev, ABS_X, (s16)le16_to_cpup((__le16*)&data[4]));
input_report_abs(inputdev, ABS_Y, ~(s16)le16_to_cpup((__le16*)&data[6]));
/* Right Stick */
input_report_abs(inputdev, ABS_RX, (s16)le16_to_cpup((__le16*)&data[8]));
input_report_abs(inputdev, ABS_RY, ~(s16)le16_to_cpup((__le16*)&data[10]));
input_sync(inputdev);
}
/* This allocates and initializes an urb specific for our needs. */
struct urb* xpad360c_allocate_urb(
struct usb_device *usbdev,
int pipe, /* We can construct an endpoint from a pipe... but not the other way around. */
void(*callback)(struct urb*),
gfp_t mem_flags)
{
struct usb_host_endpoint *ep = usb_pipe_endpoint(usbdev, pipe);
struct urb *urb = usb_alloc_urb(0, mem_flags);
if (unlikely(!urb)) {
return NULL;
}
/* Allocate URB buffer */
urb->transfer_buffer =
usb_alloc_coherent(
usbdev,
ep->desc.wMaxPacketSize,
mem_flags,
&(urb->transfer_dma)
);
if (unlikely(!urb->transfer_buffer)) {
goto fail;
}
urb->dev = usbdev;
urb->pipe = pipe;
urb->transfer_buffer_length = ep->desc.wMaxPacketSize;
urb->complete = callback;
urb->interval = ep->desc.bInterval;
urb->start_frame = -1;
urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
return urb;
fail:
usb_free_urb(urb);
return NULL;
}
static inline
struct urb* xpad360c_copy_urb(struct urb *old_urb, gfp_t mem_flags)
{
struct urb* urb =
xpad360c_allocate_urb(old_urb->dev, old_urb->pipe, old_urb->complete, mem_flags);
urb->context = old_urb->context;
return urb;
}
void xpad360c_destroy_urb(struct urb *urb)
{
struct usb_host_endpoint *ep = usb_pipe_endpoint(urb->dev, urb->pipe);
usb_free_coherent(
urb->dev,
ep->desc.wMaxPacketSize,
urb->transfer_buffer,
urb->transfer_dma
);
usb_free_urb(urb);
}
/*
static void xpad360c_complete(struct urb* urb)
{
xpad360c_check_urb(urb);
}
*/
static void xpad360c_dangerous_complete(struct urb *urb)
{
xpad360c_check_urb(urb);
usb_unanchor_urb(urb);
xpad360c_destroy_urb(urb);
}
/* Callers must do the following:
controller *must* be allocated.
They *must* set controller->in->complete themselves.
They must *not* allocate anything else within the xpad360_controller struct.
If the return value is not zero, they must disown the interface.
You must also register anything yourself. This, unfortunately, cannot be abstracted well.
*/
int xpad360c_probe(
struct xpad360_controller *controller,
struct usb_interface *interface,
void (*on_receive)(struct urb* urb),
void (*on_send)(struct urb* urb))
{
struct usb_device * usbdev = interface_to_usbdev(interface);
struct usb_endpoint_descriptor *ep_out = &(interface->cur_altsetting->endpoint[1].desc);
struct usb_endpoint_descriptor *ep_in = &(interface->cur_altsetting->endpoint[0].desc);
int error = 0;
init_usb_anchor(&controller->out_anchor);
/* Initialize common urbs */
controller->out =
xpad360c_allocate_urb(
usbdev,
usb_sndintpipe(usbdev, ep_out->bEndpointAddress),
on_send, GFP_KERNEL
);
if (unlikely(!controller->out)){
goto fail0;
}
controller->in =
xpad360c_allocate_urb(
usbdev,
usb_rcvintpipe(usbdev, ep_in->bEndpointAddress),
on_receive, GFP_KERNEL
);
if (unlikely(!controller->in)){
goto fail1;
}
controller->in->context = controller;
controller->out->context = controller;
error = usb_submit_urb(controller->in, GFP_KERNEL);
if (unlikely(error)) {
goto fail2;
}
goto success;
fail2:
xpad360c_destroy_urb(controller->in);
fail1:
xpad360c_destroy_urb(controller->out);
fail0:
kfree(controller);
success:
return error;
}
/* Callers must at least do the following:
They must *not* deallocate controller->in.
They must *not* deallocate controller->out.
*/
void xpad360c_destroy(struct xpad360_controller *controller)
{
xpad360c_destroy_urb(controller->in);
xpad360c_destroy_urb(controller->out);
}