1/* To compile
2 * gcc -g -Wall -I../src -I../lib/ -I../include -DSTORAGEDIR=\"/var/lib/bluetooth\" -o sixpair sixpair.c ../src/storage.c ../common/libhelper.a -I../common `pkg-config --libs --cflags glib-2.0 libusb-1.0` -lbluetooth
3 */
4
5#include <unistd.h>
6#include <stdio.h>
7#include <inttypes.h>
8
9#include <sdp.h>
10#include <bluetooth/bluetooth.h>
11#include <bluetooth/sdp_lib.h>
12#include <glib.h>
13#include <libusb.h>
14
15#include "storage.h"
16
17/* Vendor and product ID for the Sixaxis PS3 controller */
18#define VENDOR 0x054c
19#define PRODUCT 0x0268
20
21#define PS3_PNP_RECORD "3601920900000A000100000900013503191124090004350D35061901000900113503190011090006350909656E09006A0901000900093508350619112409010009000D350F350D350619010009001335031900110901002513576972656C65737320436F6E74726F6C6C65720901012513576972656C65737320436F6E74726F6C6C6572090102251B536F6E7920436F6D707574657220456E7465727461696E6D656E740902000901000902010901000902020800090203082109020428010902052801090206359A35980822259405010904A101A102850175089501150026FF00810375019513150025013500450105091901291381027501950D0600FF8103150026FF0005010901A10075089504350046FF0009300931093209358102C0050175089527090181027508953009019102750895300901B102C0A1028502750895300901B102C0A10285EE750895300901B102C0A10285EF750895300901B102C0C0090207350835060904090901000902082800090209280109020A280109020B09010009020C093E8009020D280009020E2800"
22
23gboolean option_get_master = TRUE;
24char *option_master= NULL;
25gboolean option_store_info = TRUE;
26const char *option_device = NULL;
27gboolean option_quiet = FALSE;
28
29const GOptionEntry options[] = {
30	{ "get-master", '\0', 0, G_OPTION_ARG_NONE, &option_get_master, "Get currently set master address", NULL },
31	{ "set-master", '\0', 0, G_OPTION_ARG_STRING, &option_master, "Set master address (\"auto\" for automatic)", NULL },
32	{ "store-info", '\0', 0, G_OPTION_ARG_NONE, &option_store_info, "Store the HID info into the input database", NULL },
33	{ "device", '\0', 0, G_OPTION_ARG_STRING, &option_device, "Only handle one device (default, all supported", NULL },
34	{ "quiet", 'q', 0, G_OPTION_ARG_NONE, &option_quiet, "Quieten the output", NULL },
35	{ NULL }
36};
37
38static gboolean
39show_master (libusb_device_handle *devh, int itfnum)
40{
41	unsigned char msg[8];
42	int res;
43
44	res = libusb_control_transfer (devh,
45				       LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE,
46				       0x01, 0x03f5, itfnum,
47				       (void*) msg, sizeof(msg),
48				       5000);
49
50	if (res < 0) {
51		g_warning ("Getting the master Bluetooth address failed");
52		return FALSE;
53	}
54	g_print ("Current Bluetooth master: %02X:%02X:%02X:%02X:%02X:%02X\n",
55		 msg[2], msg[3], msg[4], msg[5], msg[6], msg[7]);
56
57	return TRUE;
58}
59
60static char *
61get_bdaddr (libusb_device_handle *devh, int itfnum)
62{
63	unsigned char msg[17];
64	char *address;
65	int res;
66
67	res = libusb_control_transfer (devh,
68				       LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE,
69				       0x01, 0x03f2, itfnum,
70				       (void*) msg, sizeof(msg),
71				       5000);
72
73	if (res < 0) {
74		g_warning ("Getting the device Bluetooth address failed");
75		return NULL;
76	}
77
78	address = g_strdup_printf ("%02X:%02X:%02X:%02X:%02X:%02X",
79				   msg[4], msg[5], msg[6], msg[7], msg[8], msg[9]);
80
81	if (option_quiet == FALSE) {
82		g_print ("Device Bluetooth address: %s\n", address);
83	}
84
85	return address;
86}
87
88static gboolean
89set_master_bdaddr (libusb_device_handle *devh, int itfnum, char *host)
90{
91	unsigned char msg[8];
92	int mac[6];
93	int res;
94
95	if (sscanf(host, "%X:%X:%X:%X:%X:%X",
96		   &mac[0],&mac[1],&mac[2],&mac[3],&mac[4],&mac[5]) != 6) {
97		return FALSE;
98	}
99
100	msg[0] = 0x01;
101	msg[1] = 0x00;
102	msg[2] = mac[0];
103	msg[3] = mac[1];
104	msg[4] = mac[2];
105	msg[5] = mac[3];
106	msg[6] = mac[4];
107	msg[7] = mac[5];
108
109	res = libusb_control_transfer (devh,
110				       LIBUSB_ENDPOINT_OUT | LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE,
111				       0x09, 0x03f5, itfnum,
112				       (void*) msg, sizeof(msg),
113				       5000);
114
115	if (res < 0) {
116		g_warning ("Setting the master Bluetooth address failed");
117		return FALSE;
118	}
119
120	return TRUE;
121}
122
123static char *
124get_host_bdaddr (void)
125{
126	FILE *f;
127	int mac[6];
128
129	//FIXME use dbus to get the default adapter
130
131	f = popen("hcitool dev", "r");
132
133	if (f == NULL) {
134		//FIXME
135		return NULL;
136	}
137	if (fscanf(f, "%*s\n%*s %X:%X:%X:%X:%X:%X",
138		   &mac[0],&mac[1],&mac[2],&mac[3],&mac[4],&mac[5]) != 6) {
139		//FIXME
140		return NULL;
141	}
142
143	return g_strdup_printf ("%02X:%02X:%02X:%02X:%02X:%02X", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
144}
145
146static int
147handle_device (libusb_device *dev, struct libusb_config_descriptor *cfg, int itfnum, const struct libusb_interface_descriptor *alt)
148{
149	libusb_device_handle *devh;
150	int res, retval;
151
152	retval = -1;
153
154	if (libusb_open (dev, &devh) < 0) {
155		g_warning ("Can't open device");
156		goto bail;
157	}
158	libusb_detach_kernel_driver (devh, itfnum);
159
160	res = libusb_claim_interface (devh, itfnum);
161	if (res < 0) {
162		g_warning ("Can't claim interface %d", itfnum);
163		goto bail;
164	}
165
166	if (option_get_master != FALSE) {
167		if (show_master (devh, itfnum) == FALSE)
168			goto bail;
169		retval = 0;
170	}
171
172	if (option_master != NULL) {
173		if (strcmp (option_master, "auto") == 0) {
174			g_free (option_master);
175			option_master = get_host_bdaddr ();
176			if (option_master == NULL) {
177				g_warning ("Can't get bdaddr from default device");
178				retval = -1;
179				goto bail;
180			}
181		}
182	} else {
183		option_master = get_host_bdaddr ();
184		if (option_master == NULL) {
185			g_warning ("Can't get bdaddr from default device");
186			retval = -1;
187			goto bail;
188		}
189	}
190
191	if (option_store_info != FALSE) {
192		sdp_record_t *rec;
193		char *device;
194		bdaddr_t dst, src;
195
196		device = get_bdaddr (devh, itfnum);
197		if (device == NULL) {
198			retval = -1;
199			goto bail;
200		}
201
202		rec = record_from_string (PS3_PNP_RECORD);
203		store_record(option_master, device, rec);
204		write_trust(option_master, device, "[all]", TRUE);
205		store_device_id(option_master, device, 0xffff, 0x054c, 0x0268, 0);
206		str2ba(option_master, &src);
207		str2ba(device, &dst);
208		write_device_profiles(&src, &dst, "");
209		write_device_name(&src, &dst, "PLAYSTATION(R)3 Controller");
210		sdp_record_free(rec);
211
212		if (set_master_bdaddr (devh, itfnum, option_master) == FALSE) {
213			retval = -1;
214			goto bail;
215		}
216	}
217
218bail:
219	libusb_release_interface (devh, itfnum);
220	res = libusb_attach_kernel_driver(devh, itfnum);
221	if (res < 0) {
222		//FIXME sometimes the kernel tells us ENOENT, but succeeds anyway...
223		g_warning ("Reattaching the driver failed: %d", res);
224	}
225	if (devh != NULL)
226		libusb_close (devh);
227
228	return retval;
229}
230
231int main (int argc, char **argv)
232{
233	GOptionContext *context;
234	GError *error = NULL;
235	libusb_device **list;
236	ssize_t num_devices, i;
237
238	context = g_option_context_new ("- Manage Sixaxis PS3 controllers");
239	g_option_context_add_main_entries (context, options, NULL);
240	if (g_option_context_parse (context, &argc, &argv, &error) == FALSE) {
241		g_warning ("Couldn't parse command-line options: %s", error->message);
242		return 1;
243	}
244
245	/* Check that the passed bdaddr is correct */
246	if (option_master != NULL && strcmp (option_master, "auto") != 0) {
247		//FIXME check bdaddr
248	}
249
250	libusb_init (NULL);
251
252	/* Find device(s) */
253	num_devices = libusb_get_device_list (NULL, &list);
254	if (num_devices < 0) {
255		g_warning ("libusb_get_device_list failed");
256		return 1;
257	}
258
259	for (i = 0; i < num_devices; i++) {
260		struct libusb_config_descriptor *cfg;
261		libusb_device *dev = list[i];
262		struct libusb_device_descriptor desc;
263		guint8 j;
264
265		if (libusb_get_device_descriptor (dev, &desc) < 0) {
266			g_warning ("libusb_get_device_descriptor failed");
267			continue;
268		}
269
270		/* Here we check for the supported devices */
271		if (desc.idVendor != VENDOR || desc.idProduct != PRODUCT)
272			continue;
273
274		/* Look for the interface number that interests us */
275		for (j = 0; j < desc.bNumConfigurations; j++) {
276			struct libusb_config_descriptor *config;
277			guint8 k;
278
279			libusb_get_config_descriptor (dev, j, &config);
280
281			for (k = 0; k < config->bNumInterfaces; k++) {
282				const struct libusb_interface *itf = &config->interface[k];
283				int l;
284
285				for (l = 0; l < itf->num_altsetting ; l++) {
286					struct libusb_interface_descriptor alt;
287
288					alt = itf->altsetting[l];
289					if (alt.bInterfaceClass == 3) {
290						handle_device (dev, cfg, l, &alt);
291					}
292				}
293			}
294		}
295	}
296
297	return 0;
298}
299
300