1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *  Copyright (c) 1998-2001 Vojtech Pavlik
4  *
5  *   Based on the work of:
6  *	Trystan Larey-Williams
7  */
8 
9 /*
10  * ThrustMaster DirectConnect (BSP) joystick family driver for Linux
11  */
12 
13 /*
14  */
15 
16 #include <linux/delay.h>
17 #include <linux/kernel.h>
18 #include <linux/slab.h>
19 #include <linux/module.h>
20 #include <linux/gameport.h>
21 #include <linux/input.h>
22 #include <linux/jiffies.h>
23 
24 #define DRIVER_DESC	"ThrustMaster DirectConnect joystick driver"
25 
26 MODULE_AUTHOR("Vojtech Pavlik <vojtech@ucw.cz>");
27 MODULE_DESCRIPTION(DRIVER_DESC);
28 MODULE_LICENSE("GPL");
29 
30 #define TMDC_MAX_START		600	/* 600 us */
31 #define TMDC_MAX_STROBE		60	/* 60 us */
32 #define TMDC_MAX_LENGTH		13
33 
34 #define TMDC_MODE_M3DI		1
35 #define TMDC_MODE_3DRP		3
36 #define TMDC_MODE_AT		4
37 #define TMDC_MODE_FM		8
38 #define TMDC_MODE_FGP		163
39 
40 #define TMDC_BYTE_ID		10
41 #define TMDC_BYTE_REV		11
42 #define TMDC_BYTE_DEF		12
43 
44 #define TMDC_ABS		7
45 #define TMDC_ABS_HAT		4
46 #define TMDC_BTN		16
47 
48 static const unsigned char tmdc_byte_a[16] = { 0, 1, 3, 4, 6, 7 };
49 static const unsigned char tmdc_byte_d[16] = { 2, 5, 8, 9 };
50 
51 static const signed char tmdc_abs[TMDC_ABS] =
52 	{ ABS_X, ABS_Y, ABS_RUDDER, ABS_THROTTLE, ABS_RX, ABS_RY, ABS_RZ };
53 static const signed char tmdc_abs_hat[TMDC_ABS_HAT] =
54 	{ ABS_HAT0X, ABS_HAT0Y, ABS_HAT1X, ABS_HAT1Y };
55 static const signed char tmdc_abs_at[TMDC_ABS] =
56 	{ ABS_X, ABS_Y, ABS_RUDDER, -1, ABS_THROTTLE };
57 static const signed char tmdc_abs_fm[TMDC_ABS] =
58 	{ ABS_RX, ABS_RY, ABS_X, ABS_Y };
59 
60 static const short tmdc_btn_pad[TMDC_BTN] =
61 	{ BTN_A, BTN_B, BTN_C, BTN_X, BTN_Y, BTN_Z, BTN_START, BTN_SELECT, BTN_TL, BTN_TR };
62 static const short tmdc_btn_joy[TMDC_BTN] =
63 	{ BTN_TRIGGER, BTN_THUMB, BTN_TOP, BTN_TOP2, BTN_BASE, BTN_BASE2, BTN_THUMB2, BTN_PINKIE,
64 	  BTN_BASE3, BTN_BASE4, BTN_A, BTN_B, BTN_C, BTN_X, BTN_Y, BTN_Z };
65 static const short tmdc_btn_fm[TMDC_BTN] =
66         { BTN_TRIGGER, BTN_C, BTN_B, BTN_A, BTN_THUMB, BTN_X, BTN_Y, BTN_Z, BTN_TOP, BTN_TOP2 };
67 static const short tmdc_btn_at[TMDC_BTN] =
68         { BTN_TRIGGER, BTN_THUMB2, BTN_PINKIE, BTN_THUMB, BTN_BASE6, BTN_BASE5, BTN_BASE4,
69           BTN_BASE3, BTN_BASE2, BTN_BASE };
70 
71 static const struct {
72         int x;
73         int y;
74 } tmdc_hat_to_axis[] = {{ 0, 0}, { 1, 0}, { 0,-1}, {-1, 0}, { 0, 1}};
75 
76 static const struct tmdc_model {
77 	unsigned char id;
78 	const char *name;
79 	char abs;
80 	char hats;
81 	char btnc[4];
82 	char btno[4];
83 	const signed char *axes;
84 	const short *buttons;
85 } tmdc_models[] = {
86 	{   1, "ThrustMaster Millennium 3D Inceptor",	  6, 2, { 4, 2 }, { 4, 6 }, tmdc_abs, tmdc_btn_joy },
87 	{   3, "ThrustMaster Rage 3D Gamepad",		  2, 0, { 8, 2 }, { 0, 0 }, tmdc_abs, tmdc_btn_pad },
88 	{   4, "ThrustMaster Attack Throttle",		  5, 2, { 4, 6 }, { 4, 2 }, tmdc_abs_at, tmdc_btn_at },
89 	{   8, "ThrustMaster FragMaster",		  4, 0, { 8, 2 }, { 0, 0 }, tmdc_abs_fm, tmdc_btn_fm },
90 	{ 163, "Thrustmaster Fusion GamePad",		  2, 0, { 8, 2 }, { 0, 0 }, tmdc_abs, tmdc_btn_pad },
91 	{   0, "Unknown %d-axis, %d-button TM device %d", 0, 0, { 0, 0 }, { 0, 0 }, tmdc_abs, tmdc_btn_joy }
92 };
93 
94 
95 struct tmdc_port {
96 	struct input_dev *dev;
97 	char name[64];
98 	char phys[32];
99 	int mode;
100 	const signed char *abs;
101 	const short *btn;
102 	unsigned char absc;
103 	unsigned char btnc[4];
104 	unsigned char btno[4];
105 };
106 
107 struct tmdc {
108 	struct gameport *gameport;
109 	struct tmdc_port *port[2];
110 #if 0
111 	struct input_dev *dev[2];
112 	char name[2][64];
113 	char phys[2][32];
114 	int mode[2];
115 	signed char *abs[2];
116 	short *btn[2];
117 	unsigned char absc[2];
118 	unsigned char btnc[2][4];
119 	unsigned char btno[2][4];
120 #endif
121 	int reads;
122 	int bads;
123 	unsigned char exists;
124 };
125 
126 /*
127  * tmdc_read_packet() reads a ThrustMaster packet.
128  */
129 
tmdc_read_packet(struct gameport * gameport,unsigned char data[2][TMDC_MAX_LENGTH])130 static int tmdc_read_packet(struct gameport *gameport, unsigned char data[2][TMDC_MAX_LENGTH])
131 {
132 	unsigned char u, v, w, x;
133 	unsigned long flags;
134 	int i[2], j[2], t[2], p, k;
135 
136 	p = gameport_time(gameport, TMDC_MAX_STROBE);
137 
138 	for (k = 0; k < 2; k++) {
139 		t[k] = gameport_time(gameport, TMDC_MAX_START);
140 		i[k] = j[k] = 0;
141 	}
142 
143 	local_irq_save(flags);
144 	gameport_trigger(gameport);
145 
146 	w = gameport_read(gameport) >> 4;
147 
148 	do {
149 		x = w;
150 		w = gameport_read(gameport) >> 4;
151 
152 		for (k = 0, v = w, u = x; k < 2; k++, v >>= 2, u >>= 2) {
153 			if (~v & u & 2) {
154 				if (t[k] <= 0 || i[k] >= TMDC_MAX_LENGTH) continue;
155 				t[k] = p;
156 				if (j[k] == 0) {				 /* Start bit */
157 					if (~v & 1) t[k] = 0;
158 					data[k][i[k]] = 0; j[k]++; continue;
159 				}
160 				if (j[k] == 9) {				/* Stop bit */
161 					if (v & 1) t[k] = 0;
162 					j[k] = 0; i[k]++; continue;
163 				}
164 				data[k][i[k]] |= (~v & 1) << (j[k]++ - 1);	/* Data bit */
165 			}
166 			t[k]--;
167 		}
168 	} while (t[0] > 0 || t[1] > 0);
169 
170 	local_irq_restore(flags);
171 
172 	return (i[0] == TMDC_MAX_LENGTH) | ((i[1] == TMDC_MAX_LENGTH) << 1);
173 }
174 
tmdc_parse_packet(struct tmdc_port * port,unsigned char * data)175 static int tmdc_parse_packet(struct tmdc_port *port, unsigned char *data)
176 {
177 	int i, k, l;
178 
179 	if (data[TMDC_BYTE_ID] != port->mode)
180 		return -1;
181 
182 	for (i = 0; i < port->absc; i++) {
183 		if (port->abs[i] < 0)
184 			return 0;
185 
186 		input_report_abs(port->dev, port->abs[i], data[tmdc_byte_a[i]]);
187 	}
188 
189 	switch (port->mode) {
190 
191 		case TMDC_MODE_M3DI:
192 
193 			i = tmdc_byte_d[0];
194 			input_report_abs(port->dev, ABS_HAT0X, ((data[i] >> 3) & 1) - ((data[i] >> 1) & 1));
195 			input_report_abs(port->dev, ABS_HAT0Y, ((data[i] >> 2) & 1) - ( data[i]       & 1));
196 			break;
197 
198 		case TMDC_MODE_AT:
199 
200 			i = tmdc_byte_a[3];
201 			input_report_abs(port->dev, ABS_HAT0X, tmdc_hat_to_axis[(data[i] - 141) / 25].x);
202 			input_report_abs(port->dev, ABS_HAT0Y, tmdc_hat_to_axis[(data[i] - 141) / 25].y);
203 			break;
204 
205 	}
206 
207 	for (k = l = 0; k < 4; k++) {
208 		for (i = 0; i < port->btnc[k]; i++)
209 			input_report_key(port->dev, port->btn[i + l],
210 				((data[tmdc_byte_d[k]] >> (i + port->btno[k])) & 1));
211 		l += port->btnc[k];
212 	}
213 
214 	input_sync(port->dev);
215 
216 	return 0;
217 }
218 
219 /*
220  * tmdc_poll() reads and analyzes ThrustMaster joystick data.
221  */
222 
tmdc_poll(struct gameport * gameport)223 static void tmdc_poll(struct gameport *gameport)
224 {
225 	unsigned char data[2][TMDC_MAX_LENGTH];
226 	struct tmdc *tmdc = gameport_get_drvdata(gameport);
227 	unsigned char r, bad = 0;
228 	int i;
229 
230 	tmdc->reads++;
231 
232 	if ((r = tmdc_read_packet(tmdc->gameport, data)) != tmdc->exists)
233 		bad = 1;
234 	else {
235 		for (i = 0; i < 2; i++) {
236 			if (r & (1 << i) & tmdc->exists) {
237 
238 				if (tmdc_parse_packet(tmdc->port[i], data[i]))
239 					bad = 1;
240 			}
241 		}
242 	}
243 
244 	tmdc->bads += bad;
245 }
246 
tmdc_open(struct input_dev * dev)247 static int tmdc_open(struct input_dev *dev)
248 {
249 	struct tmdc *tmdc = input_get_drvdata(dev);
250 
251 	gameport_start_polling(tmdc->gameport);
252 	return 0;
253 }
254 
tmdc_close(struct input_dev * dev)255 static void tmdc_close(struct input_dev *dev)
256 {
257 	struct tmdc *tmdc = input_get_drvdata(dev);
258 
259 	gameport_stop_polling(tmdc->gameport);
260 }
261 
tmdc_setup_port(struct tmdc * tmdc,int idx,unsigned char * data)262 static int tmdc_setup_port(struct tmdc *tmdc, int idx, unsigned char *data)
263 {
264 	const struct tmdc_model *model;
265 	struct tmdc_port *port;
266 	struct input_dev *input_dev;
267 	int i, j, b = 0;
268 	int err;
269 
270 	tmdc->port[idx] = port = kzalloc(sizeof (struct tmdc_port), GFP_KERNEL);
271 	input_dev = input_allocate_device();
272 	if (!port || !input_dev) {
273 		err = -ENOMEM;
274 		goto fail;
275 	}
276 
277 	port->mode = data[TMDC_BYTE_ID];
278 
279 	for (model = tmdc_models; model->id && model->id != port->mode; model++)
280 		/* empty */;
281 
282 	port->abs = model->axes;
283 	port->btn = model->buttons;
284 
285 	if (!model->id) {
286 		port->absc = data[TMDC_BYTE_DEF] >> 4;
287 		for (i = 0; i < 4; i++)
288 			port->btnc[i] = i < (data[TMDC_BYTE_DEF] & 0xf) ? 8 : 0;
289 	} else {
290 		port->absc = model->abs;
291 		for (i = 0; i < 4; i++)
292 			port->btnc[i] = model->btnc[i];
293 	}
294 
295 	for (i = 0; i < 4; i++)
296 		port->btno[i] = model->btno[i];
297 
298 	snprintf(port->name, sizeof(port->name), model->name,
299 		 port->absc, (data[TMDC_BYTE_DEF] & 0xf) << 3, port->mode);
300 	snprintf(port->phys, sizeof(port->phys), "%s/input%d", tmdc->gameport->phys, i);
301 
302 	port->dev = input_dev;
303 
304 	input_dev->name = port->name;
305 	input_dev->phys = port->phys;
306 	input_dev->id.bustype = BUS_GAMEPORT;
307 	input_dev->id.vendor = GAMEPORT_ID_VENDOR_THRUSTMASTER;
308 	input_dev->id.product = model->id;
309 	input_dev->id.version = 0x0100;
310 	input_dev->dev.parent = &tmdc->gameport->dev;
311 
312 	input_set_drvdata(input_dev, tmdc);
313 
314 	input_dev->open = tmdc_open;
315 	input_dev->close = tmdc_close;
316 
317 	input_dev->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
318 
319 	for (i = 0; i < port->absc && i < TMDC_ABS; i++)
320 		if (port->abs[i] >= 0)
321 			input_set_abs_params(input_dev, port->abs[i], 8, 248, 2, 4);
322 
323 	for (i = 0; i < model->hats && i < TMDC_ABS_HAT; i++)
324 		input_set_abs_params(input_dev, tmdc_abs_hat[i], -1, 1, 0, 0);
325 
326 	for (i = 0; i < 4; i++) {
327 		for (j = 0; j < port->btnc[i] && j < TMDC_BTN; j++)
328 			set_bit(port->btn[j + b], input_dev->keybit);
329 		b += port->btnc[i];
330 	}
331 
332 	err = input_register_device(port->dev);
333 	if (err)
334 		goto fail;
335 
336 	return 0;
337 
338  fail:	input_free_device(input_dev);
339 	kfree(port);
340 	return err;
341 }
342 
343 /*
344  * tmdc_probe() probes for ThrustMaster type joysticks.
345  */
346 
tmdc_connect(struct gameport * gameport,struct gameport_driver * drv)347 static int tmdc_connect(struct gameport *gameport, struct gameport_driver *drv)
348 {
349 	unsigned char data[2][TMDC_MAX_LENGTH];
350 	struct tmdc *tmdc;
351 	int i;
352 	int err;
353 
354 	if (!(tmdc = kzalloc(sizeof(struct tmdc), GFP_KERNEL)))
355 		return -ENOMEM;
356 
357 	tmdc->gameport = gameport;
358 
359 	gameport_set_drvdata(gameport, tmdc);
360 
361 	err = gameport_open(gameport, drv, GAMEPORT_MODE_RAW);
362 	if (err)
363 		goto fail1;
364 
365 	if (!(tmdc->exists = tmdc_read_packet(gameport, data))) {
366 		err = -ENODEV;
367 		goto fail2;
368 	}
369 
370 	gameport_set_poll_handler(gameport, tmdc_poll);
371 	gameport_set_poll_interval(gameport, 20);
372 
373 	for (i = 0; i < 2; i++) {
374 		if (tmdc->exists & (1 << i)) {
375 
376 			err = tmdc_setup_port(tmdc, i, data[i]);
377 			if (err)
378 				goto fail3;
379 		}
380 	}
381 
382 	return 0;
383 
384  fail3: while (--i >= 0) {
385 		if (tmdc->port[i]) {
386 			input_unregister_device(tmdc->port[i]->dev);
387 			kfree(tmdc->port[i]);
388 		}
389 	}
390  fail2:	gameport_close(gameport);
391  fail1:	gameport_set_drvdata(gameport, NULL);
392 	kfree(tmdc);
393 	return err;
394 }
395 
tmdc_disconnect(struct gameport * gameport)396 static void tmdc_disconnect(struct gameport *gameport)
397 {
398 	struct tmdc *tmdc = gameport_get_drvdata(gameport);
399 	int i;
400 
401 	for (i = 0; i < 2; i++) {
402 		if (tmdc->port[i]) {
403 			input_unregister_device(tmdc->port[i]->dev);
404 			kfree(tmdc->port[i]);
405 		}
406 	}
407 	gameport_close(gameport);
408 	gameport_set_drvdata(gameport, NULL);
409 	kfree(tmdc);
410 }
411 
412 static struct gameport_driver tmdc_drv = {
413 	.driver		= {
414 		.name	= "tmdc",
415 		.owner	= THIS_MODULE,
416 	},
417 	.description	= DRIVER_DESC,
418 	.connect	= tmdc_connect,
419 	.disconnect	= tmdc_disconnect,
420 };
421 
422 module_gameport_driver(tmdc_drv);
423