input/sixpair.c

   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
  23 gboolean option_get_master = TRUE;
  24 char *option_master= NULL;
  25 gboolean option_store_info = TRUE;
  26 const char *option_device = NULL;
  27 gboolean option_quiet = FALSE;
  28
  29 const 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
  38 static gboolean
  39 show_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
  60 static char *
  61 get_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
  88 static gboolean
  89 set_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
 123 static char *
 124 get_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
 146 static int
 147 handle_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
 218 bail:
 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
 231 int 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 }