drivers/input/misc/ati_remote2.c

   1 /*
   2  * ati_remote2 - ATI/Philips USB RF remote driver
   3  *
   4  * Copyright (C) 2005-2008 Ville Syrjala <syrjala@sci.fi>
   5  * Copyright (C) 2007-2008 Peter Stokes <linux@dadeos.co.uk>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License version 2
   9  * as published by the Free Software Foundation.
  10  */
  11
  12 #include <linux/usb/input.h>
  13 #include <linux/slab.h>
  14
  15 #define DRIVER_DESC    "ATI/Philips USB RF remote driver"
  16 #define DRIVER_VERSION "0.3"
  17
  18 MODULE_DESCRIPTION(DRIVER_DESC);
  19 MODULE_VERSION(DRIVER_VERSION);
  20 MODULE_AUTHOR("Ville Syrjala <syrjala@sci.fi>");
  21 MODULE_LICENSE("GPL");
  22
  23 /*
  24  * ATI Remote Wonder II Channel Configuration
  25  *
  26  * The remote control can by assigned one of sixteen "channels" in order to facilitate
  27  * the use of multiple remote controls within range of each other.
  28  * A remote's "channel" may be altered by pressing and holding the "PC" button for
  29  * approximately 3 seconds, after which the button will slowly flash the count of the
  30  * currently configured "channel", using the numeric keypad enter a number between 1 and
  31  * 16 and then press the "PC" button again, the button will slowly flash the count of the
  32  * newly configured "channel".
  33  */
  34
  35 enum {
  36         ATI_REMOTE2_MAX_CHANNEL_MASK = 0xFFFF,
  37         ATI_REMOTE2_MAX_MODE_MASK = 0x1F,
  38 };
  39
  40 static int ati_remote2_set_mask(const char *val,
  41                                 const struct kernel_param *kp,
  42                                 unsigned int max)
  43 {
  44         unsigned long mask;
  45         int ret;
  46
  47         if (!val)
  48                 return -EINVAL;
  49
  50         ret = strict_strtoul(val, 0, &mask);
  51         if (ret)
  52                 return ret;
  53
  54         if (mask & ~max)
  55                 return -EINVAL;
  56
  57         *(unsigned int *)kp->arg = mask;
  58
  59         return 0;
  60 }
  61
  62 static int ati_remote2_set_channel_mask(const char *val,
  63                                         const struct kernel_param *kp)
  64 {
  65         pr_debug("%s()\n", __func__);
  66
  67         return ati_remote2_set_mask(val, kp, ATI_REMOTE2_MAX_CHANNEL_MASK);
  68 }
  69
  70 static int ati_remote2_get_channel_mask(char *buffer,
  71                                         const struct kernel_param *kp)
  72 {
  73         pr_debug("%s()\n", __func__);
  74
  75         return sprintf(buffer, "0x%04x", *(unsigned int *)kp->arg);
  76 }
  77
  78 static int ati_remote2_set_mode_mask(const char *val,
  79                                      const struct kernel_param *kp)
  80 {
  81         pr_debug("%s()\n", __func__);
  82
  83         return ati_remote2_set_mask(val, kp, ATI_REMOTE2_MAX_MODE_MASK);
  84 }
  85
  86 static int ati_remote2_get_mode_mask(char *buffer,
  87                                      const struct kernel_param *kp)
  88 {
  89         pr_debug("%s()\n", __func__);
  90
  91         return sprintf(buffer, "0x%02x", *(unsigned int *)kp->arg);
  92 }
  93
  94 static unsigned int channel_mask = ATI_REMOTE2_MAX_CHANNEL_MASK;
  95 #define param_check_channel_mask(name, p) __param_check(name, p, unsigned int)
  96 static struct kernel_param_ops param_ops_channel_mask = {
  97         .set = ati_remote2_set_channel_mask,
  98         .get = ati_remote2_get_channel_mask,
  99 };
 100 module_param(channel_mask, channel_mask, 0644);
 101 MODULE_PARM_DESC(channel_mask, "Bitmask of channels to accept <15:Channel16>...<1:Channel2><0:Channel1>");
 102
 103 static unsigned int mode_mask = ATI_REMOTE2_MAX_MODE_MASK;
 104 #define param_check_mode_mask(name, p) __param_check(name, p, unsigned int)
 105 static struct kernel_param_ops param_ops_mode_mask = {
 106         .set = ati_remote2_set_mode_mask,
 107         .get = ati_remote2_get_mode_mask,
 108 };
 109 module_param(mode_mask, mode_mask, 0644);
 110 MODULE_PARM_DESC(mode_mask, "Bitmask of modes to accept <4:PC><3:AUX4><2:AUX3><1:AUX2><0:AUX1>");
 111
 112 static struct usb_device_id ati_remote2_id_table[] = {
 113         { USB_DEVICE(0x0471, 0x0602) }, /* ATI Remote Wonder II */
 114         { }
 115 };
 116 MODULE_DEVICE_TABLE(usb, ati_remote2_id_table);
 117
 118 static DEFINE_MUTEX(ati_remote2_mutex);
 119
 120 enum {
 121         ATI_REMOTE2_OPENED = 0x1,
 122         ATI_REMOTE2_SUSPENDED = 0x2,
 123 };
 124
 125 enum {
 126         ATI_REMOTE2_AUX1,
 127         ATI_REMOTE2_AUX2,
 128         ATI_REMOTE2_AUX3,
 129         ATI_REMOTE2_AUX4,
 130         ATI_REMOTE2_PC,
 131         ATI_REMOTE2_MODES,
 132 };
 133
 134 static const struct {
 135         u8  hw_code;
 136         u16 keycode;
 137 } ati_remote2_key_table[] = {
 138         { 0x00, KEY_0 },
 139         { 0x01, KEY_1 },
 140         { 0x02, KEY_2 },
 141         { 0x03, KEY_3 },
 142         { 0x04, KEY_4 },
 143         { 0x05, KEY_5 },
 144         { 0x06, KEY_6 },
 145         { 0x07, KEY_7 },
 146         { 0x08, KEY_8 },
 147         { 0x09, KEY_9 },
 148         { 0x0c, KEY_POWER },
 149         { 0x0d, KEY_MUTE },
 150         { 0x10, KEY_VOLUMEUP },
 151         { 0x11, KEY_VOLUMEDOWN },
 152         { 0x20, KEY_CHANNELUP },
 153         { 0x21, KEY_CHANNELDOWN },
 154         { 0x28, KEY_FORWARD },
 155         { 0x29, KEY_REWIND },
 156         { 0x2c, KEY_PLAY },
 157         { 0x30, KEY_PAUSE },
 158         { 0x31, KEY_STOP },
 159         { 0x37, KEY_RECORD },
 160         { 0x38, KEY_DVD },
 161         { 0x39, KEY_TV },
 162         { 0x3f, KEY_PROG1 }, /* AUX1-AUX4 and PC */
 163         { 0x54, KEY_MENU },
 164         { 0x58, KEY_UP },
 165         { 0x59, KEY_DOWN },
 166         { 0x5a, KEY_LEFT },
 167         { 0x5b, KEY_RIGHT },
 168         { 0x5c, KEY_OK },
 169         { 0x78, KEY_A },
 170         { 0x79, KEY_B },
 171         { 0x7a, KEY_C },
 172         { 0x7b, KEY_D },
 173         { 0x7c, KEY_E },
 174         { 0x7d, KEY_F },
 175         { 0x82, KEY_ENTER },
 176         { 0x8e, KEY_VENDOR },
 177         { 0x96, KEY_COFFEE },
 178         { 0xa9, BTN_LEFT },
 179         { 0xaa, BTN_RIGHT },
 180         { 0xbe, KEY_QUESTION },
 181         { 0xd0, KEY_EDIT },
 182         { 0xd5, KEY_FRONT },
 183         { 0xf9, KEY_INFO },
 184 };
 185
 186 struct ati_remote2 {
 187         struct input_dev *idev;
 188         struct usb_device *udev;
 189
 190         struct usb_interface *intf[2];
 191         struct usb_endpoint_descriptor *ep[2];
 192         struct urb *urb[2];
 193         void *buf[2];
 194         dma_addr_t buf_dma[2];
 195
 196         unsigned long jiffies;
 197         int mode;
 198
 199         char name[64];
 200         char phys[64];
 201
 202         /* Each mode (AUX1-AUX4 and PC) can have an independent keymap. */
 203         u16 keycode[ATI_REMOTE2_MODES][ARRAY_SIZE(ati_remote2_key_table)];
 204
 205         unsigned int flags;
 206
 207         unsigned int channel_mask;
 208         unsigned int mode_mask;
 209 };
 210
 211 static int ati_remote2_probe(struct usb_interface *interface, const struct usb_device_id *id);
 212 static void ati_remote2_disconnect(struct usb_interface *interface);
 213 static int ati_remote2_suspend(struct usb_interface *interface, pm_message_t message);
 214 static int ati_remote2_resume(struct usb_interface *interface);
 215 static int ati_remote2_reset_resume(struct usb_interface *interface);
 216 static int ati_remote2_pre_reset(struct usb_interface *interface);
 217 static int ati_remote2_post_reset(struct usb_interface *interface);
 218
 219 static struct usb_driver ati_remote2_driver = {
 220         .name       = "ati_remote2",
 221         .probe      = ati_remote2_probe,
 222         .disconnect = ati_remote2_disconnect,
 223         .id_table   = ati_remote2_id_table,
 224         .suspend    = ati_remote2_suspend,
 225         .resume     = ati_remote2_resume,
 226         .reset_resume = ati_remote2_reset_resume,
 227         .pre_reset  = ati_remote2_pre_reset,
 228         .post_reset = ati_remote2_post_reset,
 229         .supports_autosuspend = 1,
 230 };
 231
 232 static int ati_remote2_submit_urbs(struct ati_remote2 *ar2)
 233 {
 234         int r;
 235
 236         r = usb_submit_urb(ar2->urb[0], GFP_KERNEL);
 237         if (r) {
 238                 dev_err(&ar2->intf[0]->dev,
 239                         "%s(): usb_submit_urb() = %d\n", __func__, r);
 240                 return r;
 241         }
 242         r = usb_submit_urb(ar2->urb[1], GFP_KERNEL);
 243         if (r) {
 244                 usb_kill_urb(ar2->urb[0]);
 245                 dev_err(&ar2->intf[1]->dev,
 246                         "%s(): usb_submit_urb() = %d\n", __func__, r);
 247                 return r;
 248         }
 249
 250         return 0;
 251 }
 252
 253 static void ati_remote2_kill_urbs(struct ati_remote2 *ar2)
 254 {
 255         usb_kill_urb(ar2->urb[1]);
 256         usb_kill_urb(ar2->urb[0]);
 257 }
 258
 259 static int ati_remote2_open(struct input_dev *idev)
 260 {
 261         struct ati_remote2 *ar2 = input_get_drvdata(idev);
 262         int r;
 263
 264         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
 265
 266         r = usb_autopm_get_interface(ar2->intf[0]);
 267         if (r) {
 268                 dev_err(&ar2->intf[0]->dev,
 269                         "%s(): usb_autopm_get_interface() = %d\n", __func__, r);
 270                 goto fail1;
 271         }
 272
 273         mutex_lock(&ati_remote2_mutex);
 274
 275         if (!(ar2->flags & ATI_REMOTE2_SUSPENDED)) {
 276                 r = ati_remote2_submit_urbs(ar2);
 277                 if (r)
 278                         goto fail2;
 279         }
 280
 281         ar2->flags |= ATI_REMOTE2_OPENED;
 282
 283         mutex_unlock(&ati_remote2_mutex);
 284
 285         usb_autopm_put_interface(ar2->intf[0]);
 286
 287         return 0;
 288
 289  fail2:
 290         mutex_unlock(&ati_remote2_mutex);
 291         usb_autopm_put_interface(ar2->intf[0]);
 292  fail1:
 293         return r;
 294 }
 295
 296 static void ati_remote2_close(struct input_dev *idev)
 297 {
 298         struct ati_remote2 *ar2 = input_get_drvdata(idev);
 299
 300         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
 301
 302         mutex_lock(&ati_remote2_mutex);
 303
 304         if (!(ar2->flags & ATI_REMOTE2_SUSPENDED))
 305                 ati_remote2_kill_urbs(ar2);
 306
 307         ar2->flags &= ~ATI_REMOTE2_OPENED;
 308
 309         mutex_unlock(&ati_remote2_mutex);
 310 }
 311
 312 static void ati_remote2_input_mouse(struct ati_remote2 *ar2)
 313 {
 314         struct input_dev *idev = ar2->idev;
 315         u8 *data = ar2->buf[0];
 316         int channel, mode;
 317
 318         channel = data[0] >> 4;
 319
 320         if (!((1 << channel) & ar2->channel_mask))
 321                 return;
 322
 323         mode = data[0] & 0x0F;
 324
 325         if (mode > ATI_REMOTE2_PC) {
 326                 dev_err(&ar2->intf[0]->dev,
 327                         "Unknown mode byte (%02x %02x %02x %02x)\n",
 328                         data[3], data[2], data[1], data[0]);
 329                 return;
 330         }
 331
 332         if (!((1 << mode) & ar2->mode_mask))
 333                 return;
 334
 335         input_event(idev, EV_REL, REL_X, (s8) data[1]);
 336         input_event(idev, EV_REL, REL_Y, (s8) data[2]);
 337         input_sync(idev);
 338 }
 339
 340 static int ati_remote2_lookup(unsigned int hw_code)
 341 {
 342         int i;
 343
 344         for (i = 0; i < ARRAY_SIZE(ati_remote2_key_table); i++)
 345                 if (ati_remote2_key_table[i].hw_code == hw_code)
 346                         return i;
 347
 348         return -1;
 349 }
 350
 351 static void ati_remote2_input_key(struct ati_remote2 *ar2)
 352 {
 353         struct input_dev *idev = ar2->idev;
 354         u8 *data = ar2->buf[1];
 355         int channel, mode, hw_code, index;
 356
 357         channel = data[0] >> 4;
 358
 359         if (!((1 << channel) & ar2->channel_mask))
 360                 return;
 361
 362         mode = data[0] & 0x0F;
 363
 364         if (mode > ATI_REMOTE2_PC) {
 365                 dev_err(&ar2->intf[1]->dev,
 366                         "Unknown mode byte (%02x %02x %02x %02x)\n",
 367                         data[3], data[2], data[1], data[0]);
 368                 return;
 369         }
 370
 371         hw_code = data[2];
 372         if (hw_code == 0x3f) {
 373                 /*
 374                  * For some incomprehensible reason the mouse pad generates
 375                  * events which look identical to the events from the last
 376                  * pressed mode key. Naturally we don't want to generate key
 377                  * events for the mouse pad so we filter out any subsequent
 378                  * events from the same mode key.
 379                  */
 380                 if (ar2->mode == mode)
 381                         return;
 382
 383                 if (data[1] == 0)
 384                         ar2->mode = mode;
 385         }
 386
 387         if (!((1 << mode) & ar2->mode_mask))
 388                 return;
 389
 390         index = ati_remote2_lookup(hw_code);
 391         if (index < 0) {
 392                 dev_err(&ar2->intf[1]->dev,
 393                         "Unknown code byte (%02x %02x %02x %02x)\n",
 394                         data[3], data[2], data[1], data[0]);
 395                 return;
 396         }
 397
 398         switch (data[1]) {
 399         case 0: /* release */
 400                 break;
 401         case 1: /* press */
 402                 ar2->jiffies = jiffies + msecs_to_jiffies(idev->rep[REP_DELAY]);
 403                 break;
 404         case 2: /* repeat */
 405
 406                 /* No repeat for mouse buttons. */
 407                 if (ar2->keycode[mode][index] == BTN_LEFT ||
 408                     ar2->keycode[mode][index] == BTN_RIGHT)
 409                         return;
 410
 411                 if (!time_after_eq(jiffies, ar2->jiffies))
 412                         return;
 413
 414                 ar2->jiffies = jiffies + msecs_to_jiffies(idev->rep[REP_PERIOD]);
 415                 break;
 416         default:
 417                 dev_err(&ar2->intf[1]->dev,
 418                         "Unknown state byte (%02x %02x %02x %02x)\n",
 419                         data[3], data[2], data[1], data[0]);
 420                 return;
 421         }
 422
 423         input_event(idev, EV_KEY, ar2->keycode[mode][index], data[1]);
 424         input_sync(idev);
 425 }
 426
 427 static void ati_remote2_complete_mouse(struct urb *urb)
 428 {
 429         struct ati_remote2 *ar2 = urb->context;
 430         int r;
 431
 432         switch (urb->status) {
 433         case 0:
 434                 usb_mark_last_busy(ar2->udev);
 435                 ati_remote2_input_mouse(ar2);
 436                 break;
 437         case -ENOENT:
 438         case -EILSEQ:
 439         case -ECONNRESET:
 440         case -ESHUTDOWN:
 441                 dev_dbg(&ar2->intf[0]->dev,
 442                         "%s(): urb status = %d\n", __func__, urb->status);
 443                 return;
 444         default:
 445                 usb_mark_last_busy(ar2->udev);
 446                 dev_err(&ar2->intf[0]->dev,
 447                         "%s(): urb status = %d\n", __func__, urb->status);
 448         }
 449
 450         r = usb_submit_urb(urb, GFP_ATOMIC);
 451         if (r)
 452                 dev_err(&ar2->intf[0]->dev,
 453                         "%s(): usb_submit_urb() = %d\n", __func__, r);
 454 }
 455
 456 static void ati_remote2_complete_key(struct urb *urb)
 457 {
 458         struct ati_remote2 *ar2 = urb->context;
 459         int r;
 460
 461         switch (urb->status) {
 462         case 0:
 463                 usb_mark_last_busy(ar2->udev);
 464                 ati_remote2_input_key(ar2);
 465                 break;
 466         case -ENOENT:
 467         case -EILSEQ:
 468         case -ECONNRESET:
 469         case -ESHUTDOWN:
 470                 dev_dbg(&ar2->intf[1]->dev,
 471                         "%s(): urb status = %d\n", __func__, urb->status);
 472                 return;
 473         default:
 474                 usb_mark_last_busy(ar2->udev);
 475                 dev_err(&ar2->intf[1]->dev,
 476                         "%s(): urb status = %d\n", __func__, urb->status);
 477         }
 478
 479         r = usb_submit_urb(urb, GFP_ATOMIC);
 480         if (r)
 481                 dev_err(&ar2->intf[1]->dev,
 482                         "%s(): usb_submit_urb() = %d\n", __func__, r);
 483 }
 484
 485 static int ati_remote2_getkeycode(struct input_dev *idev,
 486                                   struct input_keymap_entry *ke)
 487 {
 488         struct ati_remote2 *ar2 = input_get_drvdata(idev);
 489         unsigned int mode;
 490         int offset;
 491         unsigned int index;
 492         unsigned int scancode;
 493
 494         if (ke->flags & INPUT_KEYMAP_BY_INDEX) {
 495                 index = ke->index;
 496                 if (index >= ATI_REMOTE2_MODES *
 497                                 ARRAY_SIZE(ati_remote2_key_table))
 498                         return -EINVAL;
 499
 500                 mode = ke->index / ARRAY_SIZE(ati_remote2_key_table);
 501                 offset = ke->index % ARRAY_SIZE(ati_remote2_key_table);
 502                 scancode = (mode << 8) + ati_remote2_key_table[offset].hw_code;
 503         } else {
 504                 if (input_scancode_to_scalar(ke, &scancode))
 505                         return -EINVAL;
 506
 507                 mode = scancode >> 8;
 508                 if (mode > ATI_REMOTE2_PC)
 509                         return -EINVAL;
 510
 511                 offset = ati_remote2_lookup(scancode & 0xff);
 512                 if (offset < 0)
 513                         return -EINVAL;
 514
 515                 index = mode * ARRAY_SIZE(ati_remote2_key_table) + offset;
 516         }
 517
 518         ke->keycode = ar2->keycode[mode][offset];
 519         ke->len = sizeof(scancode);
 520         memcpy(&ke->scancode, &scancode, sizeof(scancode));
 521         ke->index = index;
 522
 523         return 0;
 524 }
 525
 526 static int ati_remote2_setkeycode(struct input_dev *idev,
 527                                   const struct input_keymap_entry *ke,
 528                                   unsigned int *old_keycode)
 529 {
 530         struct ati_remote2 *ar2 = input_get_drvdata(idev);
 531         unsigned int mode;
 532         int offset;
 533         unsigned int index;
 534         unsigned int scancode;
 535
 536         if (ke->flags & INPUT_KEYMAP_BY_INDEX) {
 537                 if (ke->index >= ATI_REMOTE2_MODES *
 538                                 ARRAY_SIZE(ati_remote2_key_table))
 539                         return -EINVAL;
 540
 541                 mode = ke->index / ARRAY_SIZE(ati_remote2_key_table);
 542                 offset = ke->index % ARRAY_SIZE(ati_remote2_key_table);
 543         } else {
 544                 if (input_scancode_to_scalar(ke, &scancode))
 545                         return -EINVAL;
 546
 547                 mode = scancode >> 8;
 548                 if (mode > ATI_REMOTE2_PC)
 549                         return -EINVAL;
 550
 551                 offset = ati_remote2_lookup(scancode & 0xff);
 552                 if (offset < 0)
 553                         return -EINVAL;
 554         }
 555
 556         *old_keycode = ar2->keycode[mode][offset];
 557         ar2->keycode[mode][offset] = ke->keycode;
 558         __set_bit(ke->keycode, idev->keybit);
 559
 560         for (mode = 0; mode < ATI_REMOTE2_MODES; mode++) {
 561                 for (index = 0; index < ARRAY_SIZE(ati_remote2_key_table); index++) {
 562                         if (ar2->keycode[mode][index] == *old_keycode)
 563                                 return 0;
 564                 }
 565         }
 566
 567         __clear_bit(*old_keycode, idev->keybit);
 568
 569         return 0;
 570 }
 571
 572 static int ati_remote2_input_init(struct ati_remote2 *ar2)
 573 {
 574         struct input_dev *idev;
 575         int index, mode, retval;
 576
 577         idev = input_allocate_device();
 578         if (!idev)
 579                 return -ENOMEM;
 580
 581         ar2->idev = idev;
 582         input_set_drvdata(idev, ar2);
 583
 584         idev->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_REP) | BIT_MASK(EV_REL);
 585         idev->keybit[BIT_WORD(BTN_MOUSE)] = BIT_MASK(BTN_LEFT) |
 586                 BIT_MASK(BTN_RIGHT);
 587         idev->relbit[0] = BIT_MASK(REL_X) | BIT_MASK(REL_Y);
 588
 589         for (mode = 0; mode < ATI_REMOTE2_MODES; mode++) {
 590                 for (index = 0; index < ARRAY_SIZE(ati_remote2_key_table); index++) {
 591                         ar2->keycode[mode][index] = ati_remote2_key_table[index].keycode;
 592                         __set_bit(ar2->keycode[mode][index], idev->keybit);
 593                 }
 594         }
 595
 596         /* AUX1-AUX4 and PC generate the same scancode. */
 597         index = ati_remote2_lookup(0x3f);
 598         ar2->keycode[ATI_REMOTE2_AUX1][index] = KEY_PROG1;
 599         ar2->keycode[ATI_REMOTE2_AUX2][index] = KEY_PROG2;
 600         ar2->keycode[ATI_REMOTE2_AUX3][index] = KEY_PROG3;
 601         ar2->keycode[ATI_REMOTE2_AUX4][index] = KEY_PROG4;
 602         ar2->keycode[ATI_REMOTE2_PC][index] = KEY_PC;
 603         __set_bit(KEY_PROG1, idev->keybit);
 604         __set_bit(KEY_PROG2, idev->keybit);
 605         __set_bit(KEY_PROG3, idev->keybit);
 606         __set_bit(KEY_PROG4, idev->keybit);
 607         __set_bit(KEY_PC, idev->keybit);
 608
 609         idev->rep[REP_DELAY]  = 250;
 610         idev->rep[REP_PERIOD] = 33;
 611
 612         idev->open = ati_remote2_open;
 613         idev->close = ati_remote2_close;
 614
 615         idev->getkeycode = ati_remote2_getkeycode;
 616         idev->setkeycode = ati_remote2_setkeycode;
 617
 618         idev->name = ar2->name;
 619         idev->phys = ar2->phys;
 620
 621         usb_to_input_id(ar2->udev, &idev->id);
 622         idev->dev.parent = &ar2->udev->dev;
 623
 624         retval = input_register_device(idev);
 625         if (retval)
 626                 input_free_device(idev);
 627
 628         return retval;
 629 }
 630
 631 static int ati_remote2_urb_init(struct ati_remote2 *ar2)
 632 {
 633         struct usb_device *udev = ar2->udev;
 634         int i, pipe, maxp;
 635
 636         for (i = 0; i < 2; i++) {
 637                 ar2->buf[i] = usb_alloc_coherent(udev, 4, GFP_KERNEL, &ar2->buf_dma[i]);
 638                 if (!ar2->buf[i])
 639                         return -ENOMEM;
 640
 641                 ar2->urb[i] = usb_alloc_urb(0, GFP_KERNEL);
 642                 if (!ar2->urb[i])
 643                         return -ENOMEM;
 644
 645                 pipe = usb_rcvintpipe(udev, ar2->ep[i]->bEndpointAddress);
 646                 maxp = usb_maxpacket(udev, pipe, usb_pipeout(pipe));
 647                 maxp = maxp > 4 ? 4 : maxp;
 648
 649                 usb_fill_int_urb(ar2->urb[i], udev, pipe, ar2->buf[i], maxp,
 650                                  i ? ati_remote2_complete_key : ati_remote2_complete_mouse,
 651                                  ar2, ar2->ep[i]->bInterval);
 652                 ar2->urb[i]->transfer_dma = ar2->buf_dma[i];
 653                 ar2->urb[i]->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
 654         }
 655
 656         return 0;
 657 }
 658
 659 static void ati_remote2_urb_cleanup(struct ati_remote2 *ar2)
 660 {
 661         int i;
 662
 663         for (i = 0; i < 2; i++) {
 664                 usb_free_urb(ar2->urb[i]);
 665                 usb_free_coherent(ar2->udev, 4, ar2->buf[i], ar2->buf_dma[i]);
 666         }
 667 }
 668
 669 static int ati_remote2_setup(struct ati_remote2 *ar2, unsigned int ch_mask)
 670 {
 671         int r, i, channel;
 672
 673         /*
 674          * Configure receiver to only accept input from remote "channel"
 675          *  channel == 0  -> Accept input from any remote channel
 676          *  channel == 1  -> Only accept input from remote channel 1
 677          *  channel == 2  -> Only accept input from remote channel 2
 678          *  ...
 679          *  channel == 16 -> Only accept input from remote channel 16
 680          */
 681
 682         channel = 0;
 683         for (i = 0; i < 16; i++) {
 684                 if ((1 << i) & ch_mask) {
 685                         if (!(~(1 << i) & ch_mask))
 686                                 channel = i + 1;
 687                         break;
 688                 }
 689         }
 690
 691         r = usb_control_msg(ar2->udev, usb_sndctrlpipe(ar2->udev, 0),
 692                             0x20,
 693                             USB_DIR_OUT | USB_TYPE_VENDOR | USB_RECIP_INTERFACE,
 694                             channel, 0x0, NULL, 0, USB_CTRL_SET_TIMEOUT);
 695         if (r) {
 696                 dev_err(&ar2->udev->dev, "%s - failed to set channel due to error: %d\n",
 697                         __func__, r);
 698                 return r;
 699         }
 700
 701         return 0;
 702 }
 703
 704 static ssize_t ati_remote2_show_channel_mask(struct device *dev,
 705                                              struct device_attribute *attr,
 706                                              char *buf)
 707 {
 708         struct usb_device *udev = to_usb_device(dev);
 709         struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
 710         struct ati_remote2 *ar2 = usb_get_intfdata(intf);
 711
 712         return sprintf(buf, "0x%04x\n", ar2->channel_mask);
 713 }
 714
 715 static ssize_t ati_remote2_store_channel_mask(struct device *dev,
 716                                               struct device_attribute *attr,
 717                                               const char *buf, size_t count)
 718 {
 719         struct usb_device *udev = to_usb_device(dev);
 720         struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
 721         struct ati_remote2 *ar2 = usb_get_intfdata(intf);
 722         unsigned long mask;
 723         int r;
 724
 725         if (strict_strtoul(buf, 0, &mask))
 726                 return -EINVAL;
 727
 728         if (mask & ~ATI_REMOTE2_MAX_CHANNEL_MASK)
 729                 return -EINVAL;
 730
 731         r = usb_autopm_get_interface(ar2->intf[0]);
 732         if (r) {
 733                 dev_err(&ar2->intf[0]->dev,
 734                         "%s(): usb_autopm_get_interface() = %d\n", __func__, r);
 735                 return r;
 736         }
 737
 738         mutex_lock(&ati_remote2_mutex);
 739
 740         if (mask != ar2->channel_mask && !ati_remote2_setup(ar2, mask))
 741                 ar2->channel_mask = mask;
 742
 743         mutex_unlock(&ati_remote2_mutex);
 744
 745         usb_autopm_put_interface(ar2->intf[0]);
 746
 747         return count;
 748 }
 749
 750 static ssize_t ati_remote2_show_mode_mask(struct device *dev,
 751                                           struct device_attribute *attr,
 752                                           char *buf)
 753 {
 754         struct usb_device *udev = to_usb_device(dev);
 755         struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
 756         struct ati_remote2 *ar2 = usb_get_intfdata(intf);
 757
 758         return sprintf(buf, "0x%02x\n", ar2->mode_mask);
 759 }
 760
 761 static ssize_t ati_remote2_store_mode_mask(struct device *dev,
 762                                            struct device_attribute *attr,
 763                                            const char *buf, size_t count)
 764 {
 765         struct usb_device *udev = to_usb_device(dev);
 766         struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
 767         struct ati_remote2 *ar2 = usb_get_intfdata(intf);
 768         unsigned long mask;
 769
 770         if (strict_strtoul(buf, 0, &mask))
 771                 return -EINVAL;
 772
 773         if (mask & ~ATI_REMOTE2_MAX_MODE_MASK)
 774                 return -EINVAL;
 775
 776         ar2->mode_mask = mask;
 777
 778         return count;
 779 }
 780
 781 static DEVICE_ATTR(channel_mask, 0644, ati_remote2_show_channel_mask,
 782                    ati_remote2_store_channel_mask);
 783
 784 static DEVICE_ATTR(mode_mask, 0644, ati_remote2_show_mode_mask,
 785                    ati_remote2_store_mode_mask);
 786
 787 static struct attribute *ati_remote2_attrs[] = {
 788         &dev_attr_channel_mask.attr,
 789         &dev_attr_mode_mask.attr,
 790         NULL,
 791 };
 792
 793 static struct attribute_group ati_remote2_attr_group = {
 794         .attrs = ati_remote2_attrs,
 795 };
 796
 797 static int ati_remote2_probe(struct usb_interface *interface, const struct usb_device_id *id)
 798 {
 799         struct usb_device *udev = interface_to_usbdev(interface);
 800         struct usb_host_interface *alt = interface->cur_altsetting;
 801         struct ati_remote2 *ar2;
 802         int r;
 803
 804         if (alt->desc.bInterfaceNumber)
 805                 return -ENODEV;
 806
 807         ar2 = kzalloc(sizeof (struct ati_remote2), GFP_KERNEL);
 808         if (!ar2)
 809                 return -ENOMEM;
 810
 811         ar2->udev = udev;
 812
 813         ar2->intf[0] = interface;
 814         ar2->ep[0] = &alt->endpoint[0].desc;
 815
 816         ar2->intf[1] = usb_ifnum_to_if(udev, 1);
 817         r = usb_driver_claim_interface(&ati_remote2_driver, ar2->intf[1], ar2);
 818         if (r)
 819                 goto fail1;
 820         alt = ar2->intf[1]->cur_altsetting;
 821         ar2->ep[1] = &alt->endpoint[0].desc;
 822
 823         r = ati_remote2_urb_init(ar2);
 824         if (r)
 825                 goto fail2;
 826
 827         ar2->channel_mask = channel_mask;
 828         ar2->mode_mask = mode_mask;
 829
 830         r = ati_remote2_setup(ar2, ar2->channel_mask);
 831         if (r)
 832                 goto fail2;
 833
 834         usb_make_path(udev, ar2->phys, sizeof(ar2->phys));
 835         strlcat(ar2->phys, "/input0", sizeof(ar2->phys));
 836
 837         strlcat(ar2->name, "ATI Remote Wonder II", sizeof(ar2->name));
 838
 839         r = sysfs_create_group(&udev->dev.kobj, &ati_remote2_attr_group);
 840         if (r)
 841                 goto fail2;
 842
 843         r = ati_remote2_input_init(ar2);
 844         if (r)
 845                 goto fail3;
 846
 847         usb_set_intfdata(interface, ar2);
 848
 849         interface->needs_remote_wakeup = 1;
 850
 851         return 0;
 852
 853  fail3:
 854         sysfs_remove_group(&udev->dev.kobj, &ati_remote2_attr_group);
 855  fail2:
 856         ati_remote2_urb_cleanup(ar2);
 857         usb_driver_release_interface(&ati_remote2_driver, ar2->intf[1]);
 858  fail1:
 859         kfree(ar2);
 860
 861         return r;
 862 }
 863
 864 static void ati_remote2_disconnect(struct usb_interface *interface)
 865 {
 866         struct ati_remote2 *ar2;
 867         struct usb_host_interface *alt = interface->cur_altsetting;
 868
 869         if (alt->desc.bInterfaceNumber)
 870                 return;
 871
 872         ar2 = usb_get_intfdata(interface);
 873         usb_set_intfdata(interface, NULL);
 874
 875         input_unregister_device(ar2->idev);
 876
 877         sysfs_remove_group(&ar2->udev->dev.kobj, &ati_remote2_attr_group);
 878
 879         ati_remote2_urb_cleanup(ar2);
 880
 881         usb_driver_release_interface(&ati_remote2_driver, ar2->intf[1]);
 882
 883         kfree(ar2);
 884 }
 885
 886 static int ati_remote2_suspend(struct usb_interface *interface,
 887                                pm_message_t message)
 888 {
 889         struct ati_remote2 *ar2;
 890         struct usb_host_interface *alt = interface->cur_altsetting;
 891
 892         if (alt->desc.bInterfaceNumber)
 893                 return 0;
 894
 895         ar2 = usb_get_intfdata(interface);
 896
 897         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
 898
 899         mutex_lock(&ati_remote2_mutex);
 900
 901         if (ar2->flags & ATI_REMOTE2_OPENED)
 902                 ati_remote2_kill_urbs(ar2);
 903
 904         ar2->flags |= ATI_REMOTE2_SUSPENDED;
 905
 906         mutex_unlock(&ati_remote2_mutex);
 907
 908         return 0;
 909 }
 910
 911 static int ati_remote2_resume(struct usb_interface *interface)
 912 {
 913         struct ati_remote2 *ar2;
 914         struct usb_host_interface *alt = interface->cur_altsetting;
 915         int r = 0;
 916
 917         if (alt->desc.bInterfaceNumber)
 918                 return 0;
 919
 920         ar2 = usb_get_intfdata(interface);
 921
 922         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
 923
 924         mutex_lock(&ati_remote2_mutex);
 925
 926         if (ar2->flags & ATI_REMOTE2_OPENED)
 927                 r = ati_remote2_submit_urbs(ar2);
 928
 929         if (!r)
 930                 ar2->flags &= ~ATI_REMOTE2_SUSPENDED;
 931
 932         mutex_unlock(&ati_remote2_mutex);
 933
 934         return r;
 935 }
 936
 937 static int ati_remote2_reset_resume(struct usb_interface *interface)
 938 {
 939         struct ati_remote2 *ar2;
 940         struct usb_host_interface *alt = interface->cur_altsetting;
 941         int r = 0;
 942
 943         if (alt->desc.bInterfaceNumber)
 944                 return 0;
 945
 946         ar2 = usb_get_intfdata(interface);
 947
 948         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
 949
 950         mutex_lock(&ati_remote2_mutex);
 951
 952         r = ati_remote2_setup(ar2, ar2->channel_mask);
 953         if (r)
 954                 goto out;
 955
 956         if (ar2->flags & ATI_REMOTE2_OPENED)
 957                 r = ati_remote2_submit_urbs(ar2);
 958
 959         if (!r)
 960                 ar2->flags &= ~ATI_REMOTE2_SUSPENDED;
 961
 962  out:
 963         mutex_unlock(&ati_remote2_mutex);
 964
 965         return r;
 966 }
 967
 968 static int ati_remote2_pre_reset(struct usb_interface *interface)
 969 {
 970         struct ati_remote2 *ar2;
 971         struct usb_host_interface *alt = interface->cur_altsetting;
 972
 973         if (alt->desc.bInterfaceNumber)
 974                 return 0;
 975
 976         ar2 = usb_get_intfdata(interface);
 977
 978         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
 979
 980         mutex_lock(&ati_remote2_mutex);
 981
 982         if (ar2->flags == ATI_REMOTE2_OPENED)
 983                 ati_remote2_kill_urbs(ar2);
 984
 985         return 0;
 986 }
 987
 988 static int ati_remote2_post_reset(struct usb_interface *interface)
 989 {
 990         struct ati_remote2 *ar2;
 991         struct usb_host_interface *alt = interface->cur_altsetting;
 992         int r = 0;
 993
 994         if (alt->desc.bInterfaceNumber)
 995                 return 0;
 996
 997         ar2 = usb_get_intfdata(interface);
 998
 999         dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
1000
1001         if (ar2->flags == ATI_REMOTE2_OPENED)
1002                 r = ati_remote2_submit_urbs(ar2);
1003
1004         mutex_unlock(&ati_remote2_mutex);
1005
1006         return r;
1007 }
1008
1009 static int __init ati_remote2_init(void)
1010 {
1011         int r;
1012
1013         r = usb_register(&ati_remote2_driver);
1014         if (r)
1015                 printk(KERN_ERR "ati_remote2: usb_register() = %d\n", r);
1016         else
1017                 printk(KERN_INFO "ati_remote2: " DRIVER_DESC " " DRIVER_VERSION "\n");
1018
1019         return r;
1020 }
1021
1022 static void __exit ati_remote2_exit(void)
1023 {
1024         usb_deregister(&ati_remote2_driver);
1025 }
1026
1027 module_init(ati_remote2_init);
1028 module_exit(ati_remote2_exit);