drivers/hid/wacom_sys.c

   1 /*
   2  * drivers/input/tablet/wacom_sys.c
   3  *
   4  *  USB Wacom tablet support - system specific code
   5  */
   6
   7 /*
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  */
  13
  14 #include "wacom_wac.h"
  15 #include "wacom.h"
  16 #include <linux/input/mt.h>
  17
  18 #define WAC_MSG_RETRIES         5
  19 #define WAC_CMD_RETRIES         10
  20
  21 #define DEV_ATTR_RW_PERM (S_IRUGO | S_IWUSR | S_IWGRP)
  22 #define DEV_ATTR_WO_PERM (S_IWUSR | S_IWGRP)
  23 #define DEV_ATTR_RO_PERM (S_IRUSR | S_IRGRP)
  24
  25 static int wacom_get_report(struct hid_device *hdev, u8 type, u8 *buf,
  26                             size_t size, unsigned int retries)
  27 {
  28         int retval;
  29
  30         do {
  31                 retval = hid_hw_raw_request(hdev, buf[0], buf, size, type,
  32                                 HID_REQ_GET_REPORT);
  33         } while ((retval == -ETIMEDOUT || retval == -EAGAIN) && --retries);
  34
  35         if (retval < 0)
  36                 hid_err(hdev, "wacom_get_report: ran out of retries "
  37                         "(last error = %d)\n", retval);
  38
  39         return retval;
  40 }
  41
  42 static int wacom_set_report(struct hid_device *hdev, u8 type, u8 *buf,
  43                             size_t size, unsigned int retries)
  44 {
  45         int retval;
  46
  47         do {
  48                 retval = hid_hw_raw_request(hdev, buf[0], buf, size, type,
  49                                 HID_REQ_SET_REPORT);
  50         } while ((retval == -ETIMEDOUT || retval == -EAGAIN) && --retries);
  51
  52         if (retval < 0)
  53                 hid_err(hdev, "wacom_set_report: ran out of retries "
  54                         "(last error = %d)\n", retval);
  55
  56         return retval;
  57 }
  58
  59 static void wacom_wac_queue_insert(struct hid_device *hdev,
  60                                    struct kfifo_rec_ptr_2 *fifo,
  61                                    u8 *raw_data, int size)
  62 {
  63         bool warned = false;
  64
  65         while (kfifo_avail(fifo) < size) {
  66                 if (!warned)
  67                         hid_warn(hdev, "%s: kfifo has filled, starting to drop events\n", __func__);
  68                 warned = true;
  69
  70                 kfifo_skip(fifo);
  71         }
  72
  73         kfifo_in(fifo, raw_data, size);
  74 }
  75
  76 static void wacom_wac_queue_flush(struct hid_device *hdev,
  77                                   struct kfifo_rec_ptr_2 *fifo)
  78 {
  79         while (!kfifo_is_empty(fifo)) {
  80                 u8 buf[WACOM_PKGLEN_MAX];
  81                 int size;
  82                 int err;
  83
  84                 size = kfifo_out(fifo, buf, sizeof(buf));
  85                 err = hid_report_raw_event(hdev, HID_INPUT_REPORT, buf, size, false);
  86                 if (err) {
  87                         hid_warn(hdev, "%s: unable to flush event due to error %d\n",
  88                                  __func__, err);
  89                 }
  90         }
  91 }
  92
  93 static int wacom_wac_pen_serial_enforce(struct hid_device *hdev,
  94                 struct hid_report *report, u8 *raw_data, int size)
  95 {
  96         struct wacom *wacom = hid_get_drvdata(hdev);
  97         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
  98         struct wacom_features *features = &wacom_wac->features;
  99         bool flush = false;
 100         bool insert = false;
 101         int i, j;
 102
 103         if (wacom_wac->serial[0] || !(features->quirks & WACOM_QUIRK_TOOLSERIAL))
 104                 return 0;
 105
 106         /* Queue events which have invalid tool type or serial number */
 107         for (i = 0; i < report->maxfield; i++) {
 108                 for (j = 0; j < report->field[i]->maxusage; j++) {
 109                         struct hid_field *field = report->field[i];
 110                         struct hid_usage *usage = &field->usage[j];
 111                         unsigned int equivalent_usage = wacom_equivalent_usage(usage->hid);
 112                         unsigned int offset;
 113                         unsigned int size;
 114                         unsigned int value;
 115
 116                         if (equivalent_usage != HID_DG_INRANGE &&
 117                             equivalent_usage != HID_DG_TOOLSERIALNUMBER &&
 118                             equivalent_usage != WACOM_HID_WD_SERIALHI &&
 119                             equivalent_usage != WACOM_HID_WD_TOOLTYPE)
 120                                 continue;
 121
 122                         offset = field->report_offset;
 123                         size = field->report_size;
 124                         value = hid_field_extract(hdev, raw_data+1, offset + j * size, size);
 125
 126                         /* If we go out of range, we need to flush the queue ASAP */
 127                         if (equivalent_usage == HID_DG_INRANGE)
 128                                 value = !value;
 129
 130                         if (value) {
 131                                 flush = true;
 132                                 switch (equivalent_usage) {
 133                                 case HID_DG_TOOLSERIALNUMBER:
 134                                         wacom_wac->serial[0] = value;
 135                                         break;
 136
 137                                 case WACOM_HID_WD_SERIALHI:
 138                                         wacom_wac->serial[0] |= ((__u64)value) << 32;
 139                                         break;
 140
 141                                 case WACOM_HID_WD_TOOLTYPE:
 142                                         wacom_wac->id[0] = value;
 143                                         break;
 144                                 }
 145                         }
 146                         else {
 147                                 insert = true;
 148                         }
 149                 }
 150         }
 151
 152         if (flush)
 153                 wacom_wac_queue_flush(hdev, &wacom_wac->pen_fifo);
 154         else if (insert)
 155                 wacom_wac_queue_insert(hdev, &wacom_wac->pen_fifo, raw_data, size);
 156
 157         return insert && !flush;
 158 }
 159
 160 static int wacom_raw_event(struct hid_device *hdev, struct hid_report *report,
 161                 u8 *raw_data, int size)
 162 {
 163         struct wacom *wacom = hid_get_drvdata(hdev);
 164
 165         if (size > WACOM_PKGLEN_MAX)
 166                 return 1;
 167
 168         if (wacom_wac_pen_serial_enforce(hdev, report, raw_data, size))
 169                 return -1;
 170
 171         memcpy(wacom->wacom_wac.data, raw_data, size);
 172
 173         wacom_wac_irq(&wacom->wacom_wac, size);
 174
 175         return 0;
 176 }
 177
 178 static int wacom_open(struct input_dev *dev)
 179 {
 180         struct wacom *wacom = input_get_drvdata(dev);
 181
 182         return hid_hw_open(wacom->hdev);
 183 }
 184
 185 static void wacom_close(struct input_dev *dev)
 186 {
 187         struct wacom *wacom = input_get_drvdata(dev);
 188
 189         /*
 190          * wacom->hdev should never be null, but surprisingly, I had the case
 191          * once while unplugging the Wacom Wireless Receiver.
 192          */
 193         if (wacom->hdev)
 194                 hid_hw_close(wacom->hdev);
 195 }
 196
 197 /*
 198  * Calculate the resolution of the X or Y axis using hidinput_calc_abs_res.
 199  */
 200 static int wacom_calc_hid_res(int logical_extents, int physical_extents,
 201                                unsigned unit, int exponent)
 202 {
 203         struct hid_field field = {
 204                 .logical_maximum = logical_extents,
 205                 .physical_maximum = physical_extents,
 206                 .unit = unit,
 207                 .unit_exponent = exponent,
 208         };
 209
 210         return hidinput_calc_abs_res(&field, ABS_X);
 211 }
 212
 213 static void wacom_hid_usage_quirk(struct hid_device *hdev,
 214                 struct hid_field *field, struct hid_usage *usage)
 215 {
 216         struct wacom *wacom = hid_get_drvdata(hdev);
 217         struct wacom_features *features = &wacom->wacom_wac.features;
 218         unsigned int equivalent_usage = wacom_equivalent_usage(usage->hid);
 219
 220         /*
 221          * The Dell Canvas 27 needs to be switched to its vendor-defined
 222          * report to provide the best resolution.
 223          */
 224         if (hdev->vendor == USB_VENDOR_ID_WACOM &&
 225             hdev->product == 0x4200 &&
 226             field->application == HID_UP_MSVENDOR) {
 227                 wacom->wacom_wac.mode_report = field->report->id;
 228                 wacom->wacom_wac.mode_value = 2;
 229         }
 230
 231         /*
 232          * ISDv4 devices which predate HID's adoption of the
 233          * HID_DG_BARELSWITCH2 usage use 0x000D0000 in its
 234          * position instead. We can accurately detect if a
 235          * usage with that value should be HID_DG_BARRELSWITCH2
 236          * based on the surrounding usages, which have remained
 237          * constant across generations.
 238          */
 239         if (features->type == HID_GENERIC &&
 240             usage->hid == 0x000D0000 &&
 241             field->application == HID_DG_PEN &&
 242             field->physical == HID_DG_STYLUS) {
 243                 int i = usage->usage_index;
 244
 245                 if (i-4 >= 0 && i+1 < field->maxusage &&
 246                     field->usage[i-4].hid == HID_DG_TIPSWITCH &&
 247                     field->usage[i-3].hid == HID_DG_BARRELSWITCH &&
 248                     field->usage[i-2].hid == HID_DG_ERASER &&
 249                     field->usage[i-1].hid == HID_DG_INVERT &&
 250                     field->usage[i+1].hid == HID_DG_INRANGE) {
 251                         usage->hid = HID_DG_BARRELSWITCH2;
 252                 }
 253         }
 254
 255         /* 2nd-generation Intuos Pro Large has incorrect Y maximum */
 256         if (hdev->vendor == USB_VENDOR_ID_WACOM &&
 257             hdev->product == 0x0358 &&
 258             WACOM_PEN_FIELD(field) &&
 259             equivalent_usage == HID_GD_Y) {
 260                 field->logical_maximum = 43200;
 261         }
 262 }
 263
 264 static void wacom_feature_mapping(struct hid_device *hdev,
 265                 struct hid_field *field, struct hid_usage *usage)
 266 {
 267         struct wacom *wacom = hid_get_drvdata(hdev);
 268         struct wacom_features *features = &wacom->wacom_wac.features;
 269         struct hid_data *hid_data = &wacom->wacom_wac.hid_data;
 270         unsigned int equivalent_usage = wacom_equivalent_usage(usage->hid);
 271         u8 *data;
 272         int ret;
 273         u32 n;
 274
 275         wacom_hid_usage_quirk(hdev, field, usage);
 276
 277         switch (equivalent_usage) {
 278         case HID_DG_CONTACTMAX:
 279                 /* leave touch_max as is if predefined */
 280                 if (!features->touch_max) {
 281                         /* read manually */
 282                         data = kzalloc(2, GFP_KERNEL);
 283                         if (!data)
 284                                 break;
 285                         data[0] = field->report->id;
 286                         ret = wacom_get_report(hdev, HID_FEATURE_REPORT,
 287                                                 data, 2, WAC_CMD_RETRIES);
 288                         if (ret == 2) {
 289                                 features->touch_max = data[1];
 290                         } else {
 291                                 features->touch_max = 16;
 292                                 hid_warn(hdev, "wacom_feature_mapping: "
 293                                          "could not get HID_DG_CONTACTMAX, "
 294                                          "defaulting to %d\n",
 295                                           features->touch_max);
 296                         }
 297                         kfree(data);
 298                 }
 299                 break;
 300         case HID_DG_INPUTMODE:
 301                 /* Ignore if value index is out of bounds. */
 302                 if (usage->usage_index >= field->report_count) {
 303                         dev_err(&hdev->dev, "HID_DG_INPUTMODE out of range\n");
 304                         break;
 305                 }
 306
 307                 hid_data->inputmode = field->report->id;
 308                 hid_data->inputmode_index = usage->usage_index;
 309                 break;
 310
 311         case HID_UP_DIGITIZER:
 312                 if (field->report->id == 0x0B &&
 313                     (field->application == WACOM_HID_G9_PEN ||
 314                      field->application == WACOM_HID_G11_PEN)) {
 315                         wacom->wacom_wac.mode_report = field->report->id;
 316                         wacom->wacom_wac.mode_value = 0;
 317                 }
 318                 break;
 319
 320         case WACOM_HID_WD_DATAMODE:
 321                 wacom->wacom_wac.mode_report = field->report->id;
 322                 wacom->wacom_wac.mode_value = 2;
 323                 break;
 324
 325         case WACOM_HID_UP_G9:
 326         case WACOM_HID_UP_G11:
 327                 if (field->report->id == 0x03 &&
 328                     (field->application == WACOM_HID_G9_TOUCHSCREEN ||
 329                      field->application == WACOM_HID_G11_TOUCHSCREEN)) {
 330                         wacom->wacom_wac.mode_report = field->report->id;
 331                         wacom->wacom_wac.mode_value = 0;
 332                 }
 333                 break;
 334         case WACOM_HID_WD_OFFSETLEFT:
 335         case WACOM_HID_WD_OFFSETTOP:
 336         case WACOM_HID_WD_OFFSETRIGHT:
 337         case WACOM_HID_WD_OFFSETBOTTOM:
 338                 /* read manually */
 339                 n = hid_report_len(field->report);
 340                 data = hid_alloc_report_buf(field->report, GFP_KERNEL);
 341                 if (!data)
 342                         break;
 343                 data[0] = field->report->id;
 344                 ret = wacom_get_report(hdev, HID_FEATURE_REPORT,
 345                                         data, n, WAC_CMD_RETRIES);
 346                 if (ret == n) {
 347                         ret = hid_report_raw_event(hdev, HID_FEATURE_REPORT,
 348                                                    data, n, 0);
 349                 } else {
 350                         hid_warn(hdev, "%s: could not retrieve sensor offsets\n",
 351                                  __func__);
 352                 }
 353                 kfree(data);
 354                 break;
 355         }
 356 }
 357
 358 /*
 359  * Interface Descriptor of wacom devices can be incomplete and
 360  * inconsistent so wacom_features table is used to store stylus
 361  * device's packet lengths, various maximum values, and tablet
 362  * resolution based on product ID's.
 363  *
 364  * For devices that contain 2 interfaces, wacom_features table is
 365  * inaccurate for the touch interface.  Since the Interface Descriptor
 366  * for touch interfaces has pretty complete data, this function exists
 367  * to query tablet for this missing information instead of hard coding in
 368  * an additional table.
 369  *
 370  * A typical Interface Descriptor for a stylus will contain a
 371  * boot mouse application collection that is not of interest and this
 372  * function will ignore it.
 373  *
 374  * It also contains a digitizer application collection that also is not
 375  * of interest since any information it contains would be duplicate
 376  * of what is in wacom_features. Usually it defines a report of an array
 377  * of bytes that could be used as max length of the stylus packet returned.
 378  * If it happens to define a Digitizer-Stylus Physical Collection then
 379  * the X and Y logical values contain valid data but it is ignored.
 380  *
 381  * A typical Interface Descriptor for a touch interface will contain a
 382  * Digitizer-Finger Physical Collection which will define both logical
 383  * X/Y maximum as well as the physical size of tablet. Since touch
 384  * interfaces haven't supported pressure or distance, this is enough
 385  * information to override invalid values in the wacom_features table.
 386  *
 387  * Intuos5 touch interface and 3rd gen Bamboo Touch do not contain useful
 388  * data. We deal with them after returning from this function.
 389  */
 390 static void wacom_usage_mapping(struct hid_device *hdev,
 391                 struct hid_field *field, struct hid_usage *usage)
 392 {
 393         struct wacom *wacom = hid_get_drvdata(hdev);
 394         struct wacom_features *features = &wacom->wacom_wac.features;
 395         bool finger = WACOM_FINGER_FIELD(field);
 396         bool pen = WACOM_PEN_FIELD(field);
 397         unsigned equivalent_usage = wacom_equivalent_usage(usage->hid);
 398
 399         /*
 400         * Requiring Stylus Usage will ignore boot mouse
 401         * X/Y values and some cases of invalid Digitizer X/Y
 402         * values commonly reported.
 403         */
 404         if (pen)
 405                 features->device_type |= WACOM_DEVICETYPE_PEN;
 406         else if (finger)
 407                 features->device_type |= WACOM_DEVICETYPE_TOUCH;
 408         else
 409                 return;
 410
 411         wacom_hid_usage_quirk(hdev, field, usage);
 412
 413         switch (equivalent_usage) {
 414         case HID_GD_X:
 415                 features->x_max = field->logical_maximum;
 416                 if (finger) {
 417                         features->x_phy = field->physical_maximum;
 418                         if ((features->type != BAMBOO_PT) &&
 419                             (features->type != BAMBOO_TOUCH)) {
 420                                 features->unit = field->unit;
 421                                 features->unitExpo = field->unit_exponent;
 422                         }
 423                 }
 424                 break;
 425         case HID_GD_Y:
 426                 features->y_max = field->logical_maximum;
 427                 if (finger) {
 428                         features->y_phy = field->physical_maximum;
 429                         if ((features->type != BAMBOO_PT) &&
 430                             (features->type != BAMBOO_TOUCH)) {
 431                                 features->unit = field->unit;
 432                                 features->unitExpo = field->unit_exponent;
 433                         }
 434                 }
 435                 break;
 436         case HID_DG_TIPPRESSURE:
 437                 if (pen)
 438                         features->pressure_max = field->logical_maximum;
 439                 break;
 440         }
 441
 442         if (features->type == HID_GENERIC)
 443                 wacom_wac_usage_mapping(hdev, field, usage);
 444 }
 445
 446 static void wacom_post_parse_hid(struct hid_device *hdev,
 447                                  struct wacom_features *features)
 448 {
 449         struct wacom *wacom = hid_get_drvdata(hdev);
 450         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
 451
 452         if (features->type == HID_GENERIC) {
 453                 /* Any last-minute generic device setup */
 454                 if (wacom_wac->has_mode_change) {
 455                         if (wacom_wac->is_direct_mode)
 456                                 features->device_type |= WACOM_DEVICETYPE_DIRECT;
 457                         else
 458                                 features->device_type &= ~WACOM_DEVICETYPE_DIRECT;
 459                 }
 460
 461                 if (features->touch_max > 1) {
 462                         if (features->device_type & WACOM_DEVICETYPE_DIRECT)
 463                                 input_mt_init_slots(wacom_wac->touch_input,
 464                                                     wacom_wac->features.touch_max,
 465                                                     INPUT_MT_DIRECT);
 466                         else
 467                                 input_mt_init_slots(wacom_wac->touch_input,
 468                                                     wacom_wac->features.touch_max,
 469                                                     INPUT_MT_POINTER);
 470                 }
 471         }
 472 }
 473
 474 static void wacom_parse_hid(struct hid_device *hdev,
 475                            struct wacom_features *features)
 476 {
 477         struct hid_report_enum *rep_enum;
 478         struct hid_report *hreport;
 479         int i, j;
 480
 481         /* check features first */
 482         rep_enum = &hdev->report_enum[HID_FEATURE_REPORT];
 483         list_for_each_entry(hreport, &rep_enum->report_list, list) {
 484                 for (i = 0; i < hreport->maxfield; i++) {
 485                         /* Ignore if report count is out of bounds. */
 486                         if (hreport->field[i]->report_count < 1)
 487                                 continue;
 488
 489                         for (j = 0; j < hreport->field[i]->maxusage; j++) {
 490                                 wacom_feature_mapping(hdev, hreport->field[i],
 491                                                 hreport->field[i]->usage + j);
 492                         }
 493                 }
 494         }
 495
 496         /* now check the input usages */
 497         rep_enum = &hdev->report_enum[HID_INPUT_REPORT];
 498         list_for_each_entry(hreport, &rep_enum->report_list, list) {
 499
 500                 if (!hreport->maxfield)
 501                         continue;
 502
 503                 for (i = 0; i < hreport->maxfield; i++)
 504                         for (j = 0; j < hreport->field[i]->maxusage; j++)
 505                                 wacom_usage_mapping(hdev, hreport->field[i],
 506                                                 hreport->field[i]->usage + j);
 507         }
 508
 509         wacom_post_parse_hid(hdev, features);
 510 }
 511
 512 static int wacom_hid_set_device_mode(struct hid_device *hdev)
 513 {
 514         struct wacom *wacom = hid_get_drvdata(hdev);
 515         struct hid_data *hid_data = &wacom->wacom_wac.hid_data;
 516         struct hid_report *r;
 517         struct hid_report_enum *re;
 518
 519         if (hid_data->inputmode < 0)
 520                 return 0;
 521
 522         re = &(hdev->report_enum[HID_FEATURE_REPORT]);
 523         r = re->report_id_hash[hid_data->inputmode];
 524         if (r) {
 525                 r->field[0]->value[hid_data->inputmode_index] = 2;
 526                 hid_hw_request(hdev, r, HID_REQ_SET_REPORT);
 527         }
 528         return 0;
 529 }
 530
 531 static int wacom_set_device_mode(struct hid_device *hdev,
 532                                  struct wacom_wac *wacom_wac)
 533 {
 534         u8 *rep_data;
 535         struct hid_report *r;
 536         struct hid_report_enum *re;
 537         u32 length;
 538         int error = -ENOMEM, limit = 0;
 539
 540         if (wacom_wac->mode_report < 0)
 541                 return 0;
 542
 543         re = &(hdev->report_enum[HID_FEATURE_REPORT]);
 544         r = re->report_id_hash[wacom_wac->mode_report];
 545         if (!r)
 546                 return -EINVAL;
 547
 548         rep_data = hid_alloc_report_buf(r, GFP_KERNEL);
 549         if (!rep_data)
 550                 return -ENOMEM;
 551
 552         length = hid_report_len(r);
 553
 554         do {
 555                 rep_data[0] = wacom_wac->mode_report;
 556                 rep_data[1] = wacom_wac->mode_value;
 557
 558                 error = wacom_set_report(hdev, HID_FEATURE_REPORT, rep_data,
 559                                          length, 1);
 560                 if (error >= 0)
 561                         error = wacom_get_report(hdev, HID_FEATURE_REPORT,
 562                                                  rep_data, length, 1);
 563         } while (error >= 0 &&
 564                  rep_data[1] != wacom_wac->mode_report &&
 565                  limit++ < WAC_MSG_RETRIES);
 566
 567         kfree(rep_data);
 568
 569         return error < 0 ? error : 0;
 570 }
 571
 572 static int wacom_bt_query_tablet_data(struct hid_device *hdev, u8 speed,
 573                 struct wacom_features *features)
 574 {
 575         struct wacom *wacom = hid_get_drvdata(hdev);
 576         int ret;
 577         u8 rep_data[2];
 578
 579         switch (features->type) {
 580         case GRAPHIRE_BT:
 581                 rep_data[0] = 0x03;
 582                 rep_data[1] = 0x00;
 583                 ret = wacom_set_report(hdev, HID_FEATURE_REPORT, rep_data, 2,
 584                                         3);
 585
 586                 if (ret >= 0) {
 587                         rep_data[0] = speed == 0 ? 0x05 : 0x06;
 588                         rep_data[1] = 0x00;
 589
 590                         ret = wacom_set_report(hdev, HID_FEATURE_REPORT,
 591                                                 rep_data, 2, 3);
 592
 593                         if (ret >= 0) {
 594                                 wacom->wacom_wac.bt_high_speed = speed;
 595                                 return 0;
 596                         }
 597                 }
 598
 599                 /*
 600                  * Note that if the raw queries fail, it's not a hard failure
 601                  * and it is safe to continue
 602                  */
 603                 hid_warn(hdev, "failed to poke device, command %d, err %d\n",
 604                          rep_data[0], ret);
 605                 break;
 606         case INTUOS4WL:
 607                 if (speed == 1)
 608                         wacom->wacom_wac.bt_features &= ~0x20;
 609                 else
 610                         wacom->wacom_wac.bt_features |= 0x20;
 611
 612                 rep_data[0] = 0x03;
 613                 rep_data[1] = wacom->wacom_wac.bt_features;
 614
 615                 ret = wacom_set_report(hdev, HID_FEATURE_REPORT, rep_data, 2,
 616                                         1);
 617                 if (ret >= 0)
 618                         wacom->wacom_wac.bt_high_speed = speed;
 619                 break;
 620         }
 621
 622         return 0;
 623 }
 624
 625 /*
 626  * Switch the tablet into its most-capable mode. Wacom tablets are
 627  * typically configured to power-up in a mode which sends mouse-like
 628  * reports to the OS. To get absolute position, pressure data, etc.
 629  * from the tablet, it is necessary to switch the tablet out of this
 630  * mode and into one which sends the full range of tablet data.
 631  */
 632 static int _wacom_query_tablet_data(struct wacom *wacom)
 633 {
 634         struct hid_device *hdev = wacom->hdev;
 635         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
 636         struct wacom_features *features = &wacom_wac->features;
 637
 638         if (hdev->bus == BUS_BLUETOOTH)
 639                 return wacom_bt_query_tablet_data(hdev, 1, features);
 640
 641         if (features->type != HID_GENERIC) {
 642                 if (features->device_type & WACOM_DEVICETYPE_TOUCH) {
 643                         if (features->type > TABLETPC) {
 644                                 /* MT Tablet PC touch */
 645                                 wacom_wac->mode_report = 3;
 646                                 wacom_wac->mode_value = 4;
 647                         } else if (features->type == WACOM_24HDT) {
 648                                 wacom_wac->mode_report = 18;
 649                                 wacom_wac->mode_value = 2;
 650                         } else if (features->type == WACOM_27QHDT) {
 651                                 wacom_wac->mode_report = 131;
 652                                 wacom_wac->mode_value = 2;
 653                         } else if (features->type == BAMBOO_PAD) {
 654                                 wacom_wac->mode_report = 2;
 655                                 wacom_wac->mode_value = 2;
 656                         }
 657                 } else if (features->device_type & WACOM_DEVICETYPE_PEN) {
 658                         if (features->type <= BAMBOO_PT) {
 659                                 wacom_wac->mode_report = 2;
 660                                 wacom_wac->mode_value = 2;
 661                         }
 662                 }
 663         }
 664
 665         wacom_set_device_mode(hdev, wacom_wac);
 666
 667         if (features->type == HID_GENERIC)
 668                 return wacom_hid_set_device_mode(hdev);
 669
 670         return 0;
 671 }
 672
 673 static void wacom_retrieve_hid_descriptor(struct hid_device *hdev,
 674                                          struct wacom_features *features)
 675 {
 676         struct wacom *wacom = hid_get_drvdata(hdev);
 677         struct usb_interface *intf = wacom->intf;
 678
 679         /* default features */
 680         features->x_fuzz = 4;
 681         features->y_fuzz = 4;
 682         features->pressure_fuzz = 0;
 683         features->distance_fuzz = 1;
 684         features->tilt_fuzz = 1;
 685
 686         /*
 687          * The wireless device HID is basic and layout conflicts with
 688          * other tablets (monitor and touch interface can look like pen).
 689          * Skip the query for this type and modify defaults based on
 690          * interface number.
 691          */
 692         if (features->type == WIRELESS) {
 693                 if (intf->cur_altsetting->desc.bInterfaceNumber == 0)
 694                         features->device_type = WACOM_DEVICETYPE_WL_MONITOR;
 695                 else
 696                         features->device_type = WACOM_DEVICETYPE_NONE;
 697                 return;
 698         }
 699
 700         wacom_parse_hid(hdev, features);
 701 }
 702
 703 struct wacom_hdev_data {
 704         struct list_head list;
 705         struct kref kref;
 706         struct hid_device *dev;
 707         struct wacom_shared shared;
 708 };
 709
 710 static LIST_HEAD(wacom_udev_list);
 711 static DEFINE_MUTEX(wacom_udev_list_lock);
 712
 713 static bool wacom_are_sibling(struct hid_device *hdev,
 714                 struct hid_device *sibling)
 715 {
 716         struct wacom *wacom = hid_get_drvdata(hdev);
 717         struct wacom_features *features = &wacom->wacom_wac.features;
 718         struct wacom *sibling_wacom = hid_get_drvdata(sibling);
 719         struct wacom_features *sibling_features = &sibling_wacom->wacom_wac.features;
 720         __u32 oVid = features->oVid ? features->oVid : hdev->vendor;
 721         __u32 oPid = features->oPid ? features->oPid : hdev->product;
 722
 723         /* The defined oVid/oPid must match that of the sibling */
 724         if (features->oVid != HID_ANY_ID && sibling->vendor != oVid)
 725                 return false;
 726         if (features->oPid != HID_ANY_ID && sibling->product != oPid)
 727                 return false;
 728
 729         /*
 730          * Devices with the same VID/PID must share the same physical
 731          * device path, while those with different VID/PID must share
 732          * the same physical parent device path.
 733          */
 734         if (hdev->vendor == sibling->vendor && hdev->product == sibling->product) {
 735                 if (!hid_compare_device_paths(hdev, sibling, '/'))
 736                         return false;
 737         } else {
 738                 if (!hid_compare_device_paths(hdev, sibling, '.'))
 739                         return false;
 740         }
 741
 742         /* Skip the remaining heuristics unless you are a HID_GENERIC device */
 743         if (features->type != HID_GENERIC)
 744                 return true;
 745
 746         /*
 747          * Direct-input devices may not be siblings of indirect-input
 748          * devices.
 749          */
 750         if ((features->device_type & WACOM_DEVICETYPE_DIRECT) &&
 751             !(sibling_features->device_type & WACOM_DEVICETYPE_DIRECT))
 752                 return false;
 753
 754         /*
 755          * Indirect-input devices may not be siblings of direct-input
 756          * devices.
 757          */
 758         if (!(features->device_type & WACOM_DEVICETYPE_DIRECT) &&
 759             (sibling_features->device_type & WACOM_DEVICETYPE_DIRECT))
 760                 return false;
 761
 762         /* Pen devices may only be siblings of touch devices */
 763         if ((features->device_type & WACOM_DEVICETYPE_PEN) &&
 764             !(sibling_features->device_type & WACOM_DEVICETYPE_TOUCH))
 765                 return false;
 766
 767         /* Touch devices may only be siblings of pen devices */
 768         if ((features->device_type & WACOM_DEVICETYPE_TOUCH) &&
 769             !(sibling_features->device_type & WACOM_DEVICETYPE_PEN))
 770                 return false;
 771
 772         /*
 773          * No reason could be found for these two devices to NOT be
 774          * siblings, so there's a good chance they ARE siblings
 775          */
 776         return true;
 777 }
 778
 779 static struct wacom_hdev_data *wacom_get_hdev_data(struct hid_device *hdev)
 780 {
 781         struct wacom_hdev_data *data;
 782
 783         /* Try to find an already-probed interface from the same device */
 784         list_for_each_entry(data, &wacom_udev_list, list) {
 785                 if (hid_compare_device_paths(hdev, data->dev, '/')) {
 786                         kref_get(&data->kref);
 787                         return data;
 788                 }
 789         }
 790
 791         /* Fallback to finding devices that appear to be "siblings" */
 792         list_for_each_entry(data, &wacom_udev_list, list) {
 793                 if (wacom_are_sibling(hdev, data->dev)) {
 794                         kref_get(&data->kref);
 795                         return data;
 796                 }
 797         }
 798
 799         return NULL;
 800 }
 801
 802 static void wacom_release_shared_data(struct kref *kref)
 803 {
 804         struct wacom_hdev_data *data =
 805                 container_of(kref, struct wacom_hdev_data, kref);
 806
 807         mutex_lock(&wacom_udev_list_lock);
 808         list_del(&data->list);
 809         mutex_unlock(&wacom_udev_list_lock);
 810
 811         kfree(data);
 812 }
 813
 814 static void wacom_remove_shared_data(void *res)
 815 {
 816         struct wacom *wacom = res;
 817         struct wacom_hdev_data *data;
 818         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
 819
 820         if (wacom_wac->shared) {
 821                 data = container_of(wacom_wac->shared, struct wacom_hdev_data,
 822                                     shared);
 823
 824                 if (wacom_wac->shared->touch == wacom->hdev)
 825                         wacom_wac->shared->touch = NULL;
 826                 else if (wacom_wac->shared->pen == wacom->hdev)
 827                         wacom_wac->shared->pen = NULL;
 828
 829                 kref_put(&data->kref, wacom_release_shared_data);
 830                 wacom_wac->shared = NULL;
 831         }
 832 }
 833
 834 static int wacom_add_shared_data(struct hid_device *hdev)
 835 {
 836         struct wacom *wacom = hid_get_drvdata(hdev);
 837         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
 838         struct wacom_hdev_data *data;
 839         int retval = 0;
 840
 841         mutex_lock(&wacom_udev_list_lock);
 842
 843         data = wacom_get_hdev_data(hdev);
 844         if (!data) {
 845                 data = kzalloc(sizeof(struct wacom_hdev_data), GFP_KERNEL);
 846                 if (!data) {
 847                         retval = -ENOMEM;
 848                         goto out;
 849                 }
 850
 851                 kref_init(&data->kref);
 852                 data->dev = hdev;
 853                 list_add_tail(&data->list, &wacom_udev_list);
 854         }
 855
 856         wacom_wac->shared = &data->shared;
 857
 858         retval = devm_add_action(&hdev->dev, wacom_remove_shared_data, wacom);
 859         if (retval) {
 860                 mutex_unlock(&wacom_udev_list_lock);
 861                 wacom_remove_shared_data(wacom);
 862                 return retval;
 863         }
 864
 865         if (wacom_wac->features.device_type & WACOM_DEVICETYPE_TOUCH)
 866                 wacom_wac->shared->touch = hdev;
 867         else if (wacom_wac->features.device_type & WACOM_DEVICETYPE_PEN)
 868                 wacom_wac->shared->pen = hdev;
 869
 870 out:
 871         mutex_unlock(&wacom_udev_list_lock);
 872         return retval;
 873 }
 874
 875 static int wacom_led_control(struct wacom *wacom)
 876 {
 877         unsigned char *buf;
 878         int retval;
 879         unsigned char report_id = WAC_CMD_LED_CONTROL;
 880         int buf_size = 9;
 881
 882         if (!wacom->led.groups)
 883                 return -ENOTSUPP;
 884
 885         if (wacom->wacom_wac.features.type == REMOTE)
 886                 return -ENOTSUPP;
 887
 888         if (wacom->wacom_wac.pid) { /* wireless connected */
 889                 report_id = WAC_CMD_WL_LED_CONTROL;
 890                 buf_size = 13;
 891         }
 892         else if (wacom->wacom_wac.features.type == INTUOSP2_BT) {
 893                 report_id = WAC_CMD_WL_INTUOSP2;
 894                 buf_size = 51;
 895         }
 896         buf = kzalloc(buf_size, GFP_KERNEL);
 897         if (!buf)
 898                 return -ENOMEM;
 899
 900         if (wacom->wacom_wac.features.type == HID_GENERIC) {
 901                 buf[0] = WAC_CMD_LED_CONTROL_GENERIC;
 902                 buf[1] = wacom->led.llv;
 903                 buf[2] = wacom->led.groups[0].select & 0x03;
 904
 905         } else if ((wacom->wacom_wac.features.type >= INTUOS5S &&
 906             wacom->wacom_wac.features.type <= INTUOSPL)) {
 907                 /*
 908                  * Touch Ring and crop mark LED luminance may take on
 909                  * one of four values:
 910                  *    0 = Low; 1 = Medium; 2 = High; 3 = Off
 911                  */
 912                 int ring_led = wacom->led.groups[0].select & 0x03;
 913                 int ring_lum = (((wacom->led.llv & 0x60) >> 5) - 1) & 0x03;
 914                 int crop_lum = 0;
 915                 unsigned char led_bits = (crop_lum << 4) | (ring_lum << 2) | (ring_led);
 916
 917                 buf[0] = report_id;
 918                 if (wacom->wacom_wac.pid) {
 919                         wacom_get_report(wacom->hdev, HID_FEATURE_REPORT,
 920                                          buf, buf_size, WAC_CMD_RETRIES);
 921                         buf[0] = report_id;
 922                         buf[4] = led_bits;
 923                 } else
 924                         buf[1] = led_bits;
 925         }
 926         else if (wacom->wacom_wac.features.type == INTUOSP2_BT) {
 927                 buf[0] = report_id;
 928                 buf[4] = 100; // Power Connection LED (ORANGE)
 929                 buf[5] = 100; // BT Connection LED (BLUE)
 930                 buf[6] = 100; // Paper Mode (RED?)
 931                 buf[7] = 100; // Paper Mode (GREEN?)
 932                 buf[8] = 100; // Paper Mode (BLUE?)
 933                 buf[9] = wacom->led.llv;
 934                 buf[10] = wacom->led.groups[0].select & 0x03;
 935         }
 936         else {
 937                 int led = wacom->led.groups[0].select | 0x4;
 938
 939                 if (wacom->wacom_wac.features.type == WACOM_21UX2 ||
 940                     wacom->wacom_wac.features.type == WACOM_24HD)
 941                         led |= (wacom->led.groups[1].select << 4) | 0x40;
 942
 943                 buf[0] = report_id;
 944                 buf[1] = led;
 945                 buf[2] = wacom->led.llv;
 946                 buf[3] = wacom->led.hlv;
 947                 buf[4] = wacom->led.img_lum;
 948         }
 949
 950         retval = wacom_set_report(wacom->hdev, HID_FEATURE_REPORT, buf, buf_size,
 951                                   WAC_CMD_RETRIES);
 952         kfree(buf);
 953
 954         return retval;
 955 }
 956
 957 static int wacom_led_putimage(struct wacom *wacom, int button_id, u8 xfer_id,
 958                 const unsigned len, const void *img)
 959 {
 960         unsigned char *buf;
 961         int i, retval;
 962         const unsigned chunk_len = len / 4; /* 4 chunks are needed to be sent */
 963
 964         buf = kzalloc(chunk_len + 3 , GFP_KERNEL);
 965         if (!buf)
 966                 return -ENOMEM;
 967
 968         /* Send 'start' command */
 969         buf[0] = WAC_CMD_ICON_START;
 970         buf[1] = 1;
 971         retval = wacom_set_report(wacom->hdev, HID_FEATURE_REPORT, buf, 2,
 972                                   WAC_CMD_RETRIES);
 973         if (retval < 0)
 974                 goto out;
 975
 976         buf[0] = xfer_id;
 977         buf[1] = button_id & 0x07;
 978         for (i = 0; i < 4; i++) {
 979                 buf[2] = i;
 980                 memcpy(buf + 3, img + i * chunk_len, chunk_len);
 981
 982                 retval = wacom_set_report(wacom->hdev, HID_FEATURE_REPORT,
 983                                           buf, chunk_len + 3, WAC_CMD_RETRIES);
 984                 if (retval < 0)
 985                         break;
 986         }
 987
 988         /* Send 'stop' */
 989         buf[0] = WAC_CMD_ICON_START;
 990         buf[1] = 0;
 991         wacom_set_report(wacom->hdev, HID_FEATURE_REPORT, buf, 2,
 992                          WAC_CMD_RETRIES);
 993
 994 out:
 995         kfree(buf);
 996         return retval;
 997 }
 998
 999 static ssize_t wacom_led_select_store(struct device *dev, int set_id,
1000                                       const char *buf, size_t count)
1001 {
1002         struct hid_device *hdev = to_hid_device(dev);
1003         struct wacom *wacom = hid_get_drvdata(hdev);
1004         unsigned int id;
1005         int err;
1006
1007         err = kstrtouint(buf, 10, &id);
1008         if (err)
1009                 return err;
1010
1011         mutex_lock(&wacom->lock);
1012
1013         wacom->led.groups[set_id].select = id & 0x3;
1014         err = wacom_led_control(wacom);
1015
1016         mutex_unlock(&wacom->lock);
1017
1018         return err < 0 ? err : count;
1019 }
1020
1021 #define DEVICE_LED_SELECT_ATTR(SET_ID)                                  \
1022 static ssize_t wacom_led##SET_ID##_select_store(struct device *dev,     \
1023         struct device_attribute *attr, const char *buf, size_t count)   \
1024 {                                                                       \
1025         return wacom_led_select_store(dev, SET_ID, buf, count);         \
1026 }                                                                       \
1027 static ssize_t wacom_led##SET_ID##_select_show(struct device *dev,      \
1028         struct device_attribute *attr, char *buf)                       \
1029 {                                                                       \
1030         struct hid_device *hdev = to_hid_device(dev);\
1031         struct wacom *wacom = hid_get_drvdata(hdev);                    \
1032         return scnprintf(buf, PAGE_SIZE, "%d\n",                        \
1033                          wacom->led.groups[SET_ID].select);             \
1034 }                                                                       \
1035 static DEVICE_ATTR(status_led##SET_ID##_select, DEV_ATTR_RW_PERM,       \
1036                     wacom_led##SET_ID##_select_show,                    \
1037                     wacom_led##SET_ID##_select_store)
1038
1039 DEVICE_LED_SELECT_ATTR(0);
1040 DEVICE_LED_SELECT_ATTR(1);
1041
1042 static ssize_t wacom_luminance_store(struct wacom *wacom, u8 *dest,
1043                                      const char *buf, size_t count)
1044 {
1045         unsigned int value;
1046         int err;
1047
1048         err = kstrtouint(buf, 10, &value);
1049         if (err)
1050                 return err;
1051
1052         mutex_lock(&wacom->lock);
1053
1054         *dest = value & 0x7f;
1055         err = wacom_led_control(wacom);
1056
1057         mutex_unlock(&wacom->lock);
1058
1059         return err < 0 ? err : count;
1060 }
1061
1062 #define DEVICE_LUMINANCE_ATTR(name, field)                              \
1063 static ssize_t wacom_##name##_luminance_store(struct device *dev,       \
1064         struct device_attribute *attr, const char *buf, size_t count)   \
1065 {                                                                       \
1066         struct hid_device *hdev = to_hid_device(dev);\
1067         struct wacom *wacom = hid_get_drvdata(hdev);                    \
1068                                                                         \
1069         return wacom_luminance_store(wacom, &wacom->led.field,          \
1070                                      buf, count);                       \
1071 }                                                                       \
1072 static ssize_t wacom_##name##_luminance_show(struct device *dev,        \
1073         struct device_attribute *attr, char *buf)                       \
1074 {                                                                       \
1075         struct wacom *wacom = dev_get_drvdata(dev);                     \
1076         return scnprintf(buf, PAGE_SIZE, "%d\n", wacom->led.field);     \
1077 }                                                                       \
1078 static DEVICE_ATTR(name##_luminance, DEV_ATTR_RW_PERM,                  \
1079                    wacom_##name##_luminance_show,                       \
1080                    wacom_##name##_luminance_store)
1081
1082 DEVICE_LUMINANCE_ATTR(status0, llv);
1083 DEVICE_LUMINANCE_ATTR(status1, hlv);
1084 DEVICE_LUMINANCE_ATTR(buttons, img_lum);
1085
1086 static ssize_t wacom_button_image_store(struct device *dev, int button_id,
1087                                         const char *buf, size_t count)
1088 {
1089         struct hid_device *hdev = to_hid_device(dev);
1090         struct wacom *wacom = hid_get_drvdata(hdev);
1091         int err;
1092         unsigned len;
1093         u8 xfer_id;
1094
1095         if (hdev->bus == BUS_BLUETOOTH) {
1096                 len = 256;
1097                 xfer_id = WAC_CMD_ICON_BT_XFER;
1098         } else {
1099                 len = 1024;
1100                 xfer_id = WAC_CMD_ICON_XFER;
1101         }
1102
1103         if (count != len)
1104                 return -EINVAL;
1105
1106         mutex_lock(&wacom->lock);
1107
1108         err = wacom_led_putimage(wacom, button_id, xfer_id, len, buf);
1109
1110         mutex_unlock(&wacom->lock);
1111
1112         return err < 0 ? err : count;
1113 }
1114
1115 #define DEVICE_BTNIMG_ATTR(BUTTON_ID)                                   \
1116 static ssize_t wacom_btnimg##BUTTON_ID##_store(struct device *dev,      \
1117         struct device_attribute *attr, const char *buf, size_t count)   \
1118 {                                                                       \
1119         return wacom_button_image_store(dev, BUTTON_ID, buf, count);    \
1120 }                                                                       \
1121 static DEVICE_ATTR(button##BUTTON_ID##_rawimg, DEV_ATTR_WO_PERM,        \
1122                    NULL, wacom_btnimg##BUTTON_ID##_store)
1123
1124 DEVICE_BTNIMG_ATTR(0);
1125 DEVICE_BTNIMG_ATTR(1);
1126 DEVICE_BTNIMG_ATTR(2);
1127 DEVICE_BTNIMG_ATTR(3);
1128 DEVICE_BTNIMG_ATTR(4);
1129 DEVICE_BTNIMG_ATTR(5);
1130 DEVICE_BTNIMG_ATTR(6);
1131 DEVICE_BTNIMG_ATTR(7);
1132
1133 static struct attribute *cintiq_led_attrs[] = {
1134         &dev_attr_status_led0_select.attr,
1135         &dev_attr_status_led1_select.attr,
1136         NULL
1137 };
1138
1139 static struct attribute_group cintiq_led_attr_group = {
1140         .name = "wacom_led",
1141         .attrs = cintiq_led_attrs,
1142 };
1143
1144 static struct attribute *intuos4_led_attrs[] = {
1145         &dev_attr_status0_luminance.attr,
1146         &dev_attr_status1_luminance.attr,
1147         &dev_attr_status_led0_select.attr,
1148         &dev_attr_buttons_luminance.attr,
1149         &dev_attr_button0_rawimg.attr,
1150         &dev_attr_button1_rawimg.attr,
1151         &dev_attr_button2_rawimg.attr,
1152         &dev_attr_button3_rawimg.attr,
1153         &dev_attr_button4_rawimg.attr,
1154         &dev_attr_button5_rawimg.attr,
1155         &dev_attr_button6_rawimg.attr,
1156         &dev_attr_button7_rawimg.attr,
1157         NULL
1158 };
1159
1160 static struct attribute_group intuos4_led_attr_group = {
1161         .name = "wacom_led",
1162         .attrs = intuos4_led_attrs,
1163 };
1164
1165 static struct attribute *intuos5_led_attrs[] = {
1166         &dev_attr_status0_luminance.attr,
1167         &dev_attr_status_led0_select.attr,
1168         NULL
1169 };
1170
1171 static struct attribute_group intuos5_led_attr_group = {
1172         .name = "wacom_led",
1173         .attrs = intuos5_led_attrs,
1174 };
1175
1176 static struct attribute *generic_led_attrs[] = {
1177         &dev_attr_status0_luminance.attr,
1178         &dev_attr_status_led0_select.attr,
1179         NULL
1180 };
1181
1182 static struct attribute_group generic_led_attr_group = {
1183         .name = "wacom_led",
1184         .attrs = generic_led_attrs,
1185 };
1186
1187 struct wacom_sysfs_group_devres {
1188         struct attribute_group *group;
1189         struct kobject *root;
1190 };
1191
1192 static void wacom_devm_sysfs_group_release(struct device *dev, void *res)
1193 {
1194         struct wacom_sysfs_group_devres *devres = res;
1195         struct kobject *kobj = devres->root;
1196
1197         dev_dbg(dev, "%s: dropping reference to %s\n",
1198                 __func__, devres->group->name);
1199         sysfs_remove_group(kobj, devres->group);
1200 }
1201
1202 static int __wacom_devm_sysfs_create_group(struct wacom *wacom,
1203                                            struct kobject *root,
1204                                            struct attribute_group *group)
1205 {
1206         struct wacom_sysfs_group_devres *devres;
1207         int error;
1208
1209         devres = devres_alloc(wacom_devm_sysfs_group_release,
1210                               sizeof(struct wacom_sysfs_group_devres),
1211                               GFP_KERNEL);
1212         if (!devres)
1213                 return -ENOMEM;
1214
1215         devres->group = group;
1216         devres->root = root;
1217
1218         error = sysfs_create_group(devres->root, group);
1219         if (error) {
1220                 devres_free(devres);
1221                 return error;
1222         }
1223
1224         devres_add(&wacom->hdev->dev, devres);
1225
1226         return 0;
1227 }
1228
1229 static int wacom_devm_sysfs_create_group(struct wacom *wacom,
1230                                          struct attribute_group *group)
1231 {
1232         return __wacom_devm_sysfs_create_group(wacom, &wacom->hdev->dev.kobj,
1233                                                group);
1234 }
1235
1236 enum led_brightness wacom_leds_brightness_get(struct wacom_led *led)
1237 {
1238         struct wacom *wacom = led->wacom;
1239
1240         if (wacom->led.max_hlv)
1241                 return led->hlv * LED_FULL / wacom->led.max_hlv;
1242
1243         if (wacom->led.max_llv)
1244                 return led->llv * LED_FULL / wacom->led.max_llv;
1245
1246         /* device doesn't support brightness tuning */
1247         return LED_FULL;
1248 }
1249
1250 static enum led_brightness __wacom_led_brightness_get(struct led_classdev *cdev)
1251 {
1252         struct wacom_led *led = container_of(cdev, struct wacom_led, cdev);
1253         struct wacom *wacom = led->wacom;
1254
1255         if (wacom->led.groups[led->group].select != led->id)
1256                 return LED_OFF;
1257
1258         return wacom_leds_brightness_get(led);
1259 }
1260
1261 static int wacom_led_brightness_set(struct led_classdev *cdev,
1262                                     enum led_brightness brightness)
1263 {
1264         struct wacom_led *led = container_of(cdev, struct wacom_led, cdev);
1265         struct wacom *wacom = led->wacom;
1266         int error;
1267
1268         mutex_lock(&wacom->lock);
1269
1270         if (!wacom->led.groups || (brightness == LED_OFF &&
1271             wacom->led.groups[led->group].select != led->id)) {
1272                 error = 0;
1273                 goto out;
1274         }
1275
1276         led->llv = wacom->led.llv = wacom->led.max_llv * brightness / LED_FULL;
1277         led->hlv = wacom->led.hlv = wacom->led.max_hlv * brightness / LED_FULL;
1278
1279         wacom->led.groups[led->group].select = led->id;
1280
1281         error = wacom_led_control(wacom);
1282
1283 out:
1284         mutex_unlock(&wacom->lock);
1285
1286         return error;
1287 }
1288
1289 static void wacom_led_readonly_brightness_set(struct led_classdev *cdev,
1290                                                enum led_brightness brightness)
1291 {
1292 }
1293
1294 static int wacom_led_register_one(struct device *dev, struct wacom *wacom,
1295                                   struct wacom_led *led, unsigned int group,
1296                                   unsigned int id, bool read_only)
1297 {
1298         int error;
1299         char *name;
1300
1301         name = devm_kasprintf(dev, GFP_KERNEL,
1302                               "%s::wacom-%d.%d",
1303                               dev_name(dev),
1304                               group,
1305                               id);
1306         if (!name)
1307                 return -ENOMEM;
1308
1309         if (!read_only) {
1310                 led->trigger.name = name;
1311                 error = devm_led_trigger_register(dev, &led->trigger);
1312                 if (error) {
1313                         hid_err(wacom->hdev,
1314                                 "failed to register LED trigger %s: %d\n",
1315                                 led->cdev.name, error);
1316                         return error;
1317                 }
1318         }
1319
1320         led->group = group;
1321         led->id = id;
1322         led->wacom = wacom;
1323         led->llv = wacom->led.llv;
1324         led->hlv = wacom->led.hlv;
1325         led->cdev.name = name;
1326         led->cdev.max_brightness = LED_FULL;
1327         led->cdev.flags = LED_HW_PLUGGABLE;
1328         led->cdev.brightness_get = __wacom_led_brightness_get;
1329         if (!read_only) {
1330                 led->cdev.brightness_set_blocking = wacom_led_brightness_set;
1331                 led->cdev.default_trigger = led->cdev.name;
1332         } else {
1333                 led->cdev.brightness_set = wacom_led_readonly_brightness_set;
1334         }
1335
1336         error = devm_led_classdev_register(dev, &led->cdev);
1337         if (error) {
1338                 hid_err(wacom->hdev,
1339                         "failed to register LED %s: %d\n",
1340                         led->cdev.name, error);
1341                 led->cdev.name = NULL;
1342                 return error;
1343         }
1344
1345         return 0;
1346 }
1347
1348 static void wacom_led_groups_release_one(void *data)
1349 {
1350         struct wacom_group_leds *group = data;
1351
1352         devres_release_group(group->dev, group);
1353 }
1354
1355 static int wacom_led_groups_alloc_and_register_one(struct device *dev,
1356                                                    struct wacom *wacom,
1357                                                    int group_id, int count,
1358                                                    bool read_only)
1359 {
1360         struct wacom_led *leds;
1361         int i, error;
1362
1363         if (group_id >= wacom->led.count || count <= 0)
1364                 return -EINVAL;
1365
1366         if (!devres_open_group(dev, &wacom->led.groups[group_id], GFP_KERNEL))
1367                 return -ENOMEM;
1368
1369         leds = devm_kcalloc(dev, count, sizeof(struct wacom_led), GFP_KERNEL);
1370         if (!leds) {
1371                 error = -ENOMEM;
1372                 goto err;
1373         }
1374
1375         wacom->led.groups[group_id].leds = leds;
1376         wacom->led.groups[group_id].count = count;
1377
1378         for (i = 0; i < count; i++) {
1379                 error = wacom_led_register_one(dev, wacom, &leds[i],
1380                                                group_id, i, read_only);
1381                 if (error)
1382                         goto err;
1383         }
1384
1385         wacom->led.groups[group_id].dev = dev;
1386
1387         devres_close_group(dev, &wacom->led.groups[group_id]);
1388
1389         /*
1390          * There is a bug (?) in devm_led_classdev_register() in which its
1391          * increments the refcount of the parent. If the parent is an input
1392          * device, that means the ref count never reaches 0 when
1393          * devm_input_device_release() gets called.
1394          * This means that the LEDs are still there after disconnect.
1395          * Manually force the release of the group so that the leds are released
1396          * once we are done using them.
1397          */
1398         error = devm_add_action_or_reset(&wacom->hdev->dev,
1399                                          wacom_led_groups_release_one,
1400                                          &wacom->led.groups[group_id]);
1401         if (error)
1402                 return error;
1403
1404         return 0;
1405
1406 err:
1407         devres_release_group(dev, &wacom->led.groups[group_id]);
1408         return error;
1409 }
1410
1411 struct wacom_led *wacom_led_find(struct wacom *wacom, unsigned int group_id,
1412                                  unsigned int id)
1413 {
1414         struct wacom_group_leds *group;
1415
1416         if (group_id >= wacom->led.count)
1417                 return NULL;
1418
1419         group = &wacom->led.groups[group_id];
1420
1421         if (!group->leds)
1422                 return NULL;
1423
1424         id %= group->count;
1425
1426         return &group->leds[id];
1427 }
1428
1429 /**
1430  * wacom_led_next: gives the next available led with a wacom trigger.
1431  *
1432  * returns the next available struct wacom_led which has its default trigger
1433  * or the current one if none is available.
1434  */
1435 struct wacom_led *wacom_led_next(struct wacom *wacom, struct wacom_led *cur)
1436 {
1437         struct wacom_led *next_led;
1438         int group, next;
1439
1440         if (!wacom || !cur)
1441                 return NULL;
1442
1443         group = cur->group;
1444         next = cur->id;
1445
1446         do {
1447                 next_led = wacom_led_find(wacom, group, ++next);
1448                 if (!next_led || next_led == cur)
1449                         return next_led;
1450         } while (next_led->cdev.trigger != &next_led->trigger);
1451
1452         return next_led;
1453 }
1454
1455 static void wacom_led_groups_release(void *data)
1456 {
1457         struct wacom *wacom = data;
1458
1459         wacom->led.groups = NULL;
1460         wacom->led.count = 0;
1461 }
1462
1463 static int wacom_led_groups_allocate(struct wacom *wacom, int count)
1464 {
1465         struct device *dev = &wacom->hdev->dev;
1466         struct wacom_group_leds *groups;
1467         int error;
1468
1469         groups = devm_kcalloc(dev, count, sizeof(struct wacom_group_leds),
1470                               GFP_KERNEL);
1471         if (!groups)
1472                 return -ENOMEM;
1473
1474         error = devm_add_action_or_reset(dev, wacom_led_groups_release, wacom);
1475         if (error)
1476                 return error;
1477
1478         wacom->led.groups = groups;
1479         wacom->led.count = count;
1480
1481         return 0;
1482 }
1483
1484 static int wacom_leds_alloc_and_register(struct wacom *wacom, int group_count,
1485                                          int led_per_group, bool read_only)
1486 {
1487         struct device *dev;
1488         int i, error;
1489
1490         if (!wacom->wacom_wac.pad_input)
1491                 return -EINVAL;
1492
1493         dev = &wacom->wacom_wac.pad_input->dev;
1494
1495         error = wacom_led_groups_allocate(wacom, group_count);
1496         if (error)
1497                 return error;
1498
1499         for (i = 0; i < group_count; i++) {
1500                 error = wacom_led_groups_alloc_and_register_one(dev, wacom, i,
1501                                                                 led_per_group,
1502                                                                 read_only);
1503                 if (error)
1504                         return error;
1505         }
1506
1507         return 0;
1508 }
1509
1510 int wacom_initialize_leds(struct wacom *wacom)
1511 {
1512         int error;
1513
1514         if (!(wacom->wacom_wac.features.device_type & WACOM_DEVICETYPE_PAD))
1515                 return 0;
1516
1517         /* Initialize default values */
1518         switch (wacom->wacom_wac.features.type) {
1519         case HID_GENERIC:
1520                 if (!wacom->generic_has_leds)
1521                         return 0;
1522                 wacom->led.llv = 100;
1523                 wacom->led.max_llv = 100;
1524
1525                 error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1526                 if (error) {
1527                         hid_err(wacom->hdev,
1528                                 "cannot create leds err: %d\n", error);
1529                         return error;
1530                 }
1531
1532                 error = wacom_devm_sysfs_create_group(wacom,
1533                                                       &generic_led_attr_group);
1534                 break;
1535
1536         case INTUOS4S:
1537         case INTUOS4:
1538         case INTUOS4WL:
1539         case INTUOS4L:
1540                 wacom->led.llv = 10;
1541                 wacom->led.hlv = 20;
1542                 wacom->led.max_llv = 127;
1543                 wacom->led.max_hlv = 127;
1544                 wacom->led.img_lum = 10;
1545
1546                 error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1547                 if (error) {
1548                         hid_err(wacom->hdev,
1549                                 "cannot create leds err: %d\n", error);
1550                         return error;
1551                 }
1552
1553                 error = wacom_devm_sysfs_create_group(wacom,
1554                                                       &intuos4_led_attr_group);
1555                 break;
1556
1557         case WACOM_24HD:
1558         case WACOM_21UX2:
1559                 wacom->led.llv = 0;
1560                 wacom->led.hlv = 0;
1561                 wacom->led.img_lum = 0;
1562
1563                 error = wacom_leds_alloc_and_register(wacom, 2, 4, false);
1564                 if (error) {
1565                         hid_err(wacom->hdev,
1566                                 "cannot create leds err: %d\n", error);
1567                         return error;
1568                 }
1569
1570                 error = wacom_devm_sysfs_create_group(wacom,
1571                                                       &cintiq_led_attr_group);
1572                 break;
1573
1574         case INTUOS5S:
1575         case INTUOS5:
1576         case INTUOS5L:
1577         case INTUOSPS:
1578         case INTUOSPM:
1579         case INTUOSPL:
1580                 wacom->led.llv = 32;
1581                 wacom->led.max_llv = 96;
1582
1583                 error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1584                 if (error) {
1585                         hid_err(wacom->hdev,
1586                                 "cannot create leds err: %d\n", error);
1587                         return error;
1588                 }
1589
1590                 error = wacom_devm_sysfs_create_group(wacom,
1591                                                       &intuos5_led_attr_group);
1592                 break;
1593
1594         case INTUOSP2_BT:
1595                 wacom->led.llv = 50;
1596                 wacom->led.max_llv = 100;
1597                 error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1598                 if (error) {
1599                         hid_err(wacom->hdev,
1600                                 "cannot create leds err: %d\n", error);
1601                         return error;
1602                 }
1603                 return 0;
1604
1605         case REMOTE:
1606                 wacom->led.llv = 255;
1607                 wacom->led.max_llv = 255;
1608                 error = wacom_led_groups_allocate(wacom, 5);
1609                 if (error) {
1610                         hid_err(wacom->hdev,
1611                                 "cannot create leds err: %d\n", error);
1612                         return error;
1613                 }
1614                 return 0;
1615
1616         default:
1617                 return 0;
1618         }
1619
1620         if (error) {
1621                 hid_err(wacom->hdev,
1622                         "cannot create sysfs group err: %d\n", error);
1623                 return error;
1624         }
1625
1626         return 0;
1627 }
1628
1629 static void wacom_init_work(struct work_struct *work)
1630 {
1631         struct wacom *wacom = container_of(work, struct wacom, init_work.work);
1632
1633         _wacom_query_tablet_data(wacom);
1634         wacom_led_control(wacom);
1635 }
1636
1637 static void wacom_query_tablet_data(struct wacom *wacom)
1638 {
1639         schedule_delayed_work(&wacom->init_work, msecs_to_jiffies(1000));
1640 }
1641
1642 static enum power_supply_property wacom_battery_props[] = {
1643         POWER_SUPPLY_PROP_MODEL_NAME,
1644         POWER_SUPPLY_PROP_PRESENT,
1645         POWER_SUPPLY_PROP_STATUS,
1646         POWER_SUPPLY_PROP_SCOPE,
1647         POWER_SUPPLY_PROP_CAPACITY
1648 };
1649
1650 static int wacom_battery_get_property(struct power_supply *psy,
1651                                       enum power_supply_property psp,
1652                                       union power_supply_propval *val)
1653 {
1654         struct wacom_battery *battery = power_supply_get_drvdata(psy);
1655         int ret = 0;
1656
1657         switch (psp) {
1658                 case POWER_SUPPLY_PROP_MODEL_NAME:
1659                         val->strval = battery->wacom->wacom_wac.name;
1660                         break;
1661                 case POWER_SUPPLY_PROP_PRESENT:
1662                         val->intval = battery->bat_connected;
1663                         break;
1664                 case POWER_SUPPLY_PROP_SCOPE:
1665                         val->intval = POWER_SUPPLY_SCOPE_DEVICE;
1666                         break;
1667                 case POWER_SUPPLY_PROP_CAPACITY:
1668                         val->intval = battery->battery_capacity;
1669                         break;
1670                 case POWER_SUPPLY_PROP_STATUS:
1671                         if (battery->bat_status != WACOM_POWER_SUPPLY_STATUS_AUTO)
1672                                 val->intval = battery->bat_status;
1673                         else if (battery->bat_charging)
1674                                 val->intval = POWER_SUPPLY_STATUS_CHARGING;
1675                         else if (battery->battery_capacity == 100 &&
1676                                     battery->ps_connected)
1677                                 val->intval = POWER_SUPPLY_STATUS_FULL;
1678                         else if (battery->ps_connected)
1679                                 val->intval = POWER_SUPPLY_STATUS_NOT_CHARGING;
1680                         else
1681                                 val->intval = POWER_SUPPLY_STATUS_DISCHARGING;
1682                         break;
1683                 default:
1684                         ret = -EINVAL;
1685                         break;
1686         }
1687
1688         return ret;
1689 }
1690
1691 static int __wacom_initialize_battery(struct wacom *wacom,
1692                                       struct wacom_battery *battery)
1693 {
1694         static atomic_t battery_no = ATOMIC_INIT(0);
1695         struct device *dev = &wacom->hdev->dev;
1696         struct power_supply_config psy_cfg = { .drv_data = battery, };
1697         struct power_supply *ps_bat;
1698         struct power_supply_desc *bat_desc = &battery->bat_desc;
1699         unsigned long n;
1700         int error;
1701
1702         if (!devres_open_group(dev, bat_desc, GFP_KERNEL))
1703                 return -ENOMEM;
1704
1705         battery->wacom = wacom;
1706
1707         n = atomic_inc_return(&battery_no) - 1;
1708
1709         bat_desc->properties = wacom_battery_props;
1710         bat_desc->num_properties = ARRAY_SIZE(wacom_battery_props);
1711         bat_desc->get_property = wacom_battery_get_property;
1712         sprintf(battery->bat_name, "wacom_battery_%ld", n);
1713         bat_desc->name = battery->bat_name;
1714         bat_desc->type = POWER_SUPPLY_TYPE_USB;
1715         bat_desc->use_for_apm = 0;
1716
1717         ps_bat = devm_power_supply_register(dev, bat_desc, &psy_cfg);
1718         if (IS_ERR(ps_bat)) {
1719                 error = PTR_ERR(ps_bat);
1720                 goto err;
1721         }
1722
1723         power_supply_powers(ps_bat, &wacom->hdev->dev);
1724
1725         battery->battery = ps_bat;
1726
1727         devres_close_group(dev, bat_desc);
1728         return 0;
1729
1730 err:
1731         devres_release_group(dev, bat_desc);
1732         return error;
1733 }
1734
1735 static int wacom_initialize_battery(struct wacom *wacom)
1736 {
1737         if (wacom->wacom_wac.features.quirks & WACOM_QUIRK_BATTERY)
1738                 return __wacom_initialize_battery(wacom, &wacom->battery);
1739
1740         return 0;
1741 }
1742
1743 static void wacom_destroy_battery(struct wacom *wacom)
1744 {
1745         if (wacom->battery.battery) {
1746                 devres_release_group(&wacom->hdev->dev,
1747                                      &wacom->battery.bat_desc);
1748                 wacom->battery.battery = NULL;
1749         }
1750 }
1751
1752 static ssize_t wacom_show_speed(struct device *dev,
1753                                 struct device_attribute
1754                                 *attr, char *buf)
1755 {
1756         struct hid_device *hdev = to_hid_device(dev);
1757         struct wacom *wacom = hid_get_drvdata(hdev);
1758
1759         return snprintf(buf, PAGE_SIZE, "%i\n", wacom->wacom_wac.bt_high_speed);
1760 }
1761
1762 static ssize_t wacom_store_speed(struct device *dev,
1763                                 struct device_attribute *attr,
1764                                 const char *buf, size_t count)
1765 {
1766         struct hid_device *hdev = to_hid_device(dev);
1767         struct wacom *wacom = hid_get_drvdata(hdev);
1768         u8 new_speed;
1769
1770         if (kstrtou8(buf, 0, &new_speed))
1771                 return -EINVAL;
1772
1773         if (new_speed != 0 && new_speed != 1)
1774                 return -EINVAL;
1775
1776         wacom_bt_query_tablet_data(hdev, new_speed, &wacom->wacom_wac.features);
1777
1778         return count;
1779 }
1780
1781 static DEVICE_ATTR(speed, DEV_ATTR_RW_PERM,
1782                 wacom_show_speed, wacom_store_speed);
1783
1784
1785 static ssize_t wacom_show_remote_mode(struct kobject *kobj,
1786                                       struct kobj_attribute *kattr,
1787                                       char *buf, int index)
1788 {
1789         struct device *dev = kobj_to_dev(kobj->parent);
1790         struct hid_device *hdev = to_hid_device(dev);
1791         struct wacom *wacom = hid_get_drvdata(hdev);
1792         u8 mode;
1793
1794         mode = wacom->led.groups[index].select;
1795         return sprintf(buf, "%d\n", mode < 3 ? mode : -1);
1796 }
1797
1798 #define DEVICE_EKR_ATTR_GROUP(SET_ID)                                   \
1799 static ssize_t wacom_show_remote##SET_ID##_mode(struct kobject *kobj,   \
1800                                struct kobj_attribute *kattr, char *buf) \
1801 {                                                                       \
1802         return wacom_show_remote_mode(kobj, kattr, buf, SET_ID);        \
1803 }                                                                       \
1804 static struct kobj_attribute remote##SET_ID##_mode_attr = {             \
1805         .attr = {.name = "remote_mode",                                 \
1806                 .mode = DEV_ATTR_RO_PERM},                              \
1807         .show = wacom_show_remote##SET_ID##_mode,                       \
1808 };                                                                      \
1809 static struct attribute *remote##SET_ID##_serial_attrs[] = {            \
1810         &remote##SET_ID##_mode_attr.attr,                               \
1811         NULL                                                            \
1812 };                                                                      \
1813 static struct attribute_group remote##SET_ID##_serial_group = {         \
1814         .name = NULL,                                                   \
1815         .attrs = remote##SET_ID##_serial_attrs,                         \
1816 }
1817
1818 DEVICE_EKR_ATTR_GROUP(0);
1819 DEVICE_EKR_ATTR_GROUP(1);
1820 DEVICE_EKR_ATTR_GROUP(2);
1821 DEVICE_EKR_ATTR_GROUP(3);
1822 DEVICE_EKR_ATTR_GROUP(4);
1823
1824 static int wacom_remote_create_attr_group(struct wacom *wacom, __u32 serial,
1825                                           int index)
1826 {
1827         int error = 0;
1828         struct wacom_remote *remote = wacom->remote;
1829
1830         remote->remotes[index].group.name = devm_kasprintf(&wacom->hdev->dev,
1831                                                           GFP_KERNEL,
1832                                                           "%d", serial);
1833         if (!remote->remotes[index].group.name)
1834                 return -ENOMEM;
1835
1836         error = __wacom_devm_sysfs_create_group(wacom, remote->remote_dir,
1837                                                 &remote->remotes[index].group);
1838         if (error) {
1839                 remote->remotes[index].group.name = NULL;
1840                 hid_err(wacom->hdev,
1841                         "cannot create sysfs group err: %d\n", error);
1842                 return error;
1843         }
1844
1845         return 0;
1846 }
1847
1848 static int wacom_cmd_unpair_remote(struct wacom *wacom, unsigned char selector)
1849 {
1850         const size_t buf_size = 2;
1851         unsigned char *buf;
1852         int retval;
1853
1854         buf = kzalloc(buf_size, GFP_KERNEL);
1855         if (!buf)
1856                 return -ENOMEM;
1857
1858         buf[0] = WAC_CMD_DELETE_PAIRING;
1859         buf[1] = selector;
1860
1861         retval = wacom_set_report(wacom->hdev, HID_OUTPUT_REPORT, buf,
1862                                   buf_size, WAC_CMD_RETRIES);
1863         kfree(buf);
1864
1865         return retval;
1866 }
1867
1868 static ssize_t wacom_store_unpair_remote(struct kobject *kobj,
1869                                          struct kobj_attribute *attr,
1870                                          const char *buf, size_t count)
1871 {
1872         unsigned char selector = 0;
1873         struct device *dev = kobj_to_dev(kobj->parent);
1874         struct hid_device *hdev = to_hid_device(dev);
1875         struct wacom *wacom = hid_get_drvdata(hdev);
1876         int err;
1877
1878         if (!strncmp(buf, "*\n", 2)) {
1879                 selector = WAC_CMD_UNPAIR_ALL;
1880         } else {
1881                 hid_info(wacom->hdev, "remote: unrecognized unpair code: %s\n",
1882                          buf);
1883                 return -1;
1884         }
1885
1886         mutex_lock(&wacom->lock);
1887
1888         err = wacom_cmd_unpair_remote(wacom, selector);
1889         mutex_unlock(&wacom->lock);
1890
1891         return err < 0 ? err : count;
1892 }
1893
1894 static struct kobj_attribute unpair_remote_attr = {
1895         .attr = {.name = "unpair_remote", .mode = 0200},
1896         .store = wacom_store_unpair_remote,
1897 };
1898
1899 static const struct attribute *remote_unpair_attrs[] = {
1900         &unpair_remote_attr.attr,
1901         NULL
1902 };
1903
1904 static void wacom_remotes_destroy(void *data)
1905 {
1906         struct wacom *wacom = data;
1907         struct wacom_remote *remote = wacom->remote;
1908
1909         if (!remote)
1910                 return;
1911
1912         kobject_put(remote->remote_dir);
1913         kfifo_free(&remote->remote_fifo);
1914         wacom->remote = NULL;
1915 }
1916
1917 static int wacom_initialize_remotes(struct wacom *wacom)
1918 {
1919         int error = 0;
1920         struct wacom_remote *remote;
1921         int i;
1922
1923         if (wacom->wacom_wac.features.type != REMOTE)
1924                 return 0;
1925
1926         remote = devm_kzalloc(&wacom->hdev->dev, sizeof(*wacom->remote),
1927                               GFP_KERNEL);
1928         if (!remote)
1929                 return -ENOMEM;
1930
1931         wacom->remote = remote;
1932
1933         spin_lock_init(&remote->remote_lock);
1934
1935         error = kfifo_alloc(&remote->remote_fifo,
1936                         5 * sizeof(struct wacom_remote_data),
1937                         GFP_KERNEL);
1938         if (error) {
1939                 hid_err(wacom->hdev, "failed allocating remote_fifo\n");
1940                 return -ENOMEM;
1941         }
1942
1943         remote->remotes[0].group = remote0_serial_group;
1944         remote->remotes[1].group = remote1_serial_group;
1945         remote->remotes[2].group = remote2_serial_group;
1946         remote->remotes[3].group = remote3_serial_group;
1947         remote->remotes[4].group = remote4_serial_group;
1948
1949         remote->remote_dir = kobject_create_and_add("wacom_remote",
1950                                                     &wacom->hdev->dev.kobj);
1951         if (!remote->remote_dir)
1952                 return -ENOMEM;
1953
1954         error = sysfs_create_files(remote->remote_dir, remote_unpair_attrs);
1955
1956         if (error) {
1957                 hid_err(wacom->hdev,
1958                         "cannot create sysfs group err: %d\n", error);
1959                 return error;
1960         }
1961
1962         for (i = 0; i < WACOM_MAX_REMOTES; i++) {
1963                 wacom->led.groups[i].select = WACOM_STATUS_UNKNOWN;
1964                 remote->remotes[i].serial = 0;
1965         }
1966
1967         error = devm_add_action_or_reset(&wacom->hdev->dev,
1968                                          wacom_remotes_destroy, wacom);
1969         if (error)
1970                 return error;
1971
1972         return 0;
1973 }
1974
1975 static struct input_dev *wacom_allocate_input(struct wacom *wacom)
1976 {
1977         struct input_dev *input_dev;
1978         struct hid_device *hdev = wacom->hdev;
1979         struct wacom_wac *wacom_wac = &(wacom->wacom_wac);
1980
1981         input_dev = devm_input_allocate_device(&hdev->dev);
1982         if (!input_dev)
1983                 return NULL;
1984
1985         input_dev->name = wacom_wac->features.name;
1986         input_dev->phys = hdev->phys;
1987         input_dev->dev.parent = &hdev->dev;
1988         input_dev->open = wacom_open;
1989         input_dev->close = wacom_close;
1990         input_dev->uniq = hdev->uniq;
1991         input_dev->id.bustype = hdev->bus;
1992         input_dev->id.vendor  = hdev->vendor;
1993         input_dev->id.product = wacom_wac->pid ? wacom_wac->pid : hdev->product;
1994         input_dev->id.version = hdev->version;
1995         input_set_drvdata(input_dev, wacom);
1996
1997         return input_dev;
1998 }
1999
2000 static int wacom_allocate_inputs(struct wacom *wacom)
2001 {
2002         struct wacom_wac *wacom_wac = &(wacom->wacom_wac);
2003
2004         wacom_wac->pen_input = wacom_allocate_input(wacom);
2005         wacom_wac->touch_input = wacom_allocate_input(wacom);
2006         wacom_wac->pad_input = wacom_allocate_input(wacom);
2007         if (!wacom_wac->pen_input ||
2008             !wacom_wac->touch_input ||
2009             !wacom_wac->pad_input)
2010                 return -ENOMEM;
2011
2012         wacom_wac->pen_input->name = wacom_wac->pen_name;
2013         wacom_wac->touch_input->name = wacom_wac->touch_name;
2014         wacom_wac->pad_input->name = wacom_wac->pad_name;
2015
2016         return 0;
2017 }
2018
2019 static int wacom_register_inputs(struct wacom *wacom)
2020 {
2021         struct input_dev *pen_input_dev, *touch_input_dev, *pad_input_dev;
2022         struct wacom_wac *wacom_wac = &(wacom->wacom_wac);
2023         int error = 0;
2024
2025         pen_input_dev = wacom_wac->pen_input;
2026         touch_input_dev = wacom_wac->touch_input;
2027         pad_input_dev = wacom_wac->pad_input;
2028
2029         if (!pen_input_dev || !touch_input_dev || !pad_input_dev)
2030                 return -EINVAL;
2031
2032         error = wacom_setup_pen_input_capabilities(pen_input_dev, wacom_wac);
2033         if (error) {
2034                 /* no pen in use on this interface */
2035                 input_free_device(pen_input_dev);
2036                 wacom_wac->pen_input = NULL;
2037                 pen_input_dev = NULL;
2038         } else {
2039                 error = input_register_device(pen_input_dev);
2040                 if (error)
2041                         goto fail;
2042         }
2043
2044         error = wacom_setup_touch_input_capabilities(touch_input_dev, wacom_wac);
2045         if (error) {
2046                 /* no touch in use on this interface */
2047                 input_free_device(touch_input_dev);
2048                 wacom_wac->touch_input = NULL;
2049                 touch_input_dev = NULL;
2050         } else {
2051                 error = input_register_device(touch_input_dev);
2052                 if (error)
2053                         goto fail;
2054         }
2055
2056         error = wacom_setup_pad_input_capabilities(pad_input_dev, wacom_wac);
2057         if (error) {
2058                 /* no pad in use on this interface */
2059                 input_free_device(pad_input_dev);
2060                 wacom_wac->pad_input = NULL;
2061                 pad_input_dev = NULL;
2062         } else {
2063                 error = input_register_device(pad_input_dev);
2064                 if (error)
2065                         goto fail;
2066         }
2067
2068         return 0;
2069
2070 fail:
2071         wacom_wac->pad_input = NULL;
2072         wacom_wac->touch_input = NULL;
2073         wacom_wac->pen_input = NULL;
2074         return error;
2075 }
2076
2077 /*
2078  * Not all devices report physical dimensions from HID.
2079  * Compute the default from hardcoded logical dimension
2080  * and resolution before driver overwrites them.
2081  */
2082 static void wacom_set_default_phy(struct wacom_features *features)
2083 {
2084         if (features->x_resolution) {
2085                 features->x_phy = (features->x_max * 100) /
2086                                         features->x_resolution;
2087                 features->y_phy = (features->y_max * 100) /
2088                                         features->y_resolution;
2089         }
2090 }
2091
2092 static void wacom_calculate_res(struct wacom_features *features)
2093 {
2094         /* set unit to "100th of a mm" for devices not reported by HID */
2095         if (!features->unit) {
2096                 features->unit = 0x11;
2097                 features->unitExpo = -3;
2098         }
2099
2100         features->x_resolution = wacom_calc_hid_res(features->x_max,
2101                                                     features->x_phy,
2102                                                     features->unit,
2103                                                     features->unitExpo);
2104         features->y_resolution = wacom_calc_hid_res(features->y_max,
2105                                                     features->y_phy,
2106                                                     features->unit,
2107                                                     features->unitExpo);
2108 }
2109
2110 void wacom_battery_work(struct work_struct *work)
2111 {
2112         struct wacom *wacom = container_of(work, struct wacom, battery_work);
2113
2114         if ((wacom->wacom_wac.features.quirks & WACOM_QUIRK_BATTERY) &&
2115              !wacom->battery.battery) {
2116                 wacom_initialize_battery(wacom);
2117         }
2118         else if (!(wacom->wacom_wac.features.quirks & WACOM_QUIRK_BATTERY) &&
2119                  wacom->battery.battery) {
2120                 wacom_destroy_battery(wacom);
2121         }
2122 }
2123
2124 static size_t wacom_compute_pktlen(struct hid_device *hdev)
2125 {
2126         struct hid_report_enum *report_enum;
2127         struct hid_report *report;
2128         size_t size = 0;
2129
2130         report_enum = hdev->report_enum + HID_INPUT_REPORT;
2131
2132         list_for_each_entry(report, &report_enum->report_list, list) {
2133                 size_t report_size = hid_report_len(report);
2134                 if (report_size > size)
2135                         size = report_size;
2136         }
2137
2138         return size;
2139 }
2140
2141 static void wacom_update_name(struct wacom *wacom, const char *suffix)
2142 {
2143         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2144         struct wacom_features *features = &wacom_wac->features;
2145         char name[WACOM_NAME_MAX];
2146
2147         /* Generic devices name unspecified */
2148         if ((features->type == HID_GENERIC) && !strcmp("Wacom HID", features->name)) {
2149                 char *product_name = wacom->hdev->name;
2150
2151                 if (hid_is_using_ll_driver(wacom->hdev, &usb_hid_driver)) {
2152                         struct usb_interface *intf = to_usb_interface(wacom->hdev->dev.parent);
2153                         struct usb_device *dev = interface_to_usbdev(intf);
2154                         product_name = dev->product;
2155                 }
2156
2157                 if (wacom->hdev->bus == BUS_I2C) {
2158                         snprintf(name, sizeof(name), "%s %X",
2159                                  features->name, wacom->hdev->product);
2160                 } else if (strstr(product_name, "Wacom") ||
2161                            strstr(product_name, "wacom") ||
2162                            strstr(product_name, "WACOM")) {
2163                         strlcpy(name, product_name, sizeof(name));
2164                 } else {
2165                         snprintf(name, sizeof(name), "Wacom %s", product_name);
2166                 }
2167
2168                 /* strip out excess whitespaces */
2169                 while (1) {
2170                         char *gap = strstr(name, "  ");
2171                         if (gap == NULL)
2172                                 break;
2173                         /* shift everything including the terminator */
2174                         memmove(gap, gap+1, strlen(gap));
2175                 }
2176
2177                 /* get rid of trailing whitespace */
2178                 if (name[strlen(name)-1] == ' ')
2179                         name[strlen(name)-1] = '\0';
2180         } else {
2181                 strlcpy(name, features->name, sizeof(name));
2182         }
2183
2184         snprintf(wacom_wac->name, sizeof(wacom_wac->name), "%s%s",
2185                  name, suffix);
2186
2187         /* Append the device type to the name */
2188         snprintf(wacom_wac->pen_name, sizeof(wacom_wac->pen_name),
2189                 "%s%s Pen", name, suffix);
2190         snprintf(wacom_wac->touch_name, sizeof(wacom_wac->touch_name),
2191                 "%s%s Finger", name, suffix);
2192         snprintf(wacom_wac->pad_name, sizeof(wacom_wac->pad_name),
2193                 "%s%s Pad", name, suffix);
2194 }
2195
2196 static void wacom_release_resources(struct wacom *wacom)
2197 {
2198         struct hid_device *hdev = wacom->hdev;
2199
2200         if (!wacom->resources)
2201                 return;
2202
2203         devres_release_group(&hdev->dev, wacom);
2204
2205         wacom->resources = false;
2206
2207         wacom->wacom_wac.pen_input = NULL;
2208         wacom->wacom_wac.touch_input = NULL;
2209         wacom->wacom_wac.pad_input = NULL;
2210 }
2211
2212 static void wacom_set_shared_values(struct wacom_wac *wacom_wac)
2213 {
2214         if (wacom_wac->features.device_type & WACOM_DEVICETYPE_TOUCH) {
2215                 wacom_wac->shared->type = wacom_wac->features.type;
2216                 wacom_wac->shared->touch_input = wacom_wac->touch_input;
2217         }
2218
2219         if (wacom_wac->has_mute_touch_switch) {
2220                 wacom_wac->shared->has_mute_touch_switch = true;
2221                 wacom_wac->shared->is_touch_on = true;
2222         }
2223
2224         if (wacom_wac->shared->has_mute_touch_switch &&
2225             wacom_wac->shared->touch_input) {
2226                 set_bit(EV_SW, wacom_wac->shared->touch_input->evbit);
2227                 input_set_capability(wacom_wac->shared->touch_input, EV_SW,
2228                                      SW_MUTE_DEVICE);
2229         }
2230 }
2231
2232 static int wacom_parse_and_register(struct wacom *wacom, bool wireless)
2233 {
2234         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2235         struct wacom_features *features = &wacom_wac->features;
2236         struct hid_device *hdev = wacom->hdev;
2237         int error;
2238         unsigned int connect_mask = HID_CONNECT_HIDRAW;
2239
2240         features->pktlen = wacom_compute_pktlen(hdev);
2241         if (features->pktlen > WACOM_PKGLEN_MAX)
2242                 return -EINVAL;
2243
2244         if (!devres_open_group(&hdev->dev, wacom, GFP_KERNEL))
2245                 return -ENOMEM;
2246
2247         wacom->resources = true;
2248
2249         error = wacom_allocate_inputs(wacom);
2250         if (error)
2251                 goto fail;
2252
2253         /*
2254          * Bamboo Pad has a generic hid handling for the Pen, and we switch it
2255          * into debug mode for the touch part.
2256          * We ignore the other interfaces.
2257          */
2258         if (features->type == BAMBOO_PAD) {
2259                 if (features->pktlen == WACOM_PKGLEN_PENABLED) {
2260                         features->type = HID_GENERIC;
2261                 } else if ((features->pktlen != WACOM_PKGLEN_BPAD_TOUCH) &&
2262                            (features->pktlen != WACOM_PKGLEN_BPAD_TOUCH_USB)) {
2263                         error = -ENODEV;
2264                         goto fail;
2265                 }
2266         }
2267
2268         /* set the default size in case we do not get them from hid */
2269         wacom_set_default_phy(features);
2270
2271         /* Retrieve the physical and logical size for touch devices */
2272         wacom_retrieve_hid_descriptor(hdev, features);
2273         wacom_setup_device_quirks(wacom);
2274
2275         if (features->device_type == WACOM_DEVICETYPE_NONE &&
2276             features->type != WIRELESS) {
2277                 error = features->type == HID_GENERIC ? -ENODEV : 0;
2278
2279                 dev_warn(&hdev->dev, "Unknown device_type for '%s'. %s.",
2280                          hdev->name,
2281                          error ? "Ignoring" : "Assuming pen");
2282
2283                 if (error)
2284                         goto fail;
2285
2286                 features->device_type |= WACOM_DEVICETYPE_PEN;
2287         }
2288
2289         wacom_calculate_res(features);
2290
2291         wacom_update_name(wacom, wireless ? " (WL)" : "");
2292
2293         /* pen only Bamboo neither support touch nor pad */
2294         if ((features->type == BAMBOO_PEN) &&
2295             ((features->device_type & WACOM_DEVICETYPE_TOUCH) ||
2296             (features->device_type & WACOM_DEVICETYPE_PAD))) {
2297                 error = -ENODEV;
2298                 goto fail;
2299         }
2300
2301         error = wacom_add_shared_data(hdev);
2302         if (error)
2303                 goto fail;
2304
2305         if (!(features->device_type & WACOM_DEVICETYPE_WL_MONITOR) &&
2306              (features->quirks & WACOM_QUIRK_BATTERY)) {
2307                 error = wacom_initialize_battery(wacom);
2308                 if (error)
2309                         goto fail;
2310         }
2311
2312         error = wacom_register_inputs(wacom);
2313         if (error)
2314                 goto fail;
2315
2316         if (wacom->wacom_wac.features.device_type & WACOM_DEVICETYPE_PAD) {
2317                 error = wacom_initialize_leds(wacom);
2318                 if (error)
2319                         goto fail;
2320
2321                 error = wacom_initialize_remotes(wacom);
2322                 if (error)
2323                         goto fail;
2324         }
2325
2326         if (features->type == HID_GENERIC)
2327                 connect_mask |= HID_CONNECT_DRIVER;
2328
2329         /* Regular HID work starts now */
2330         error = hid_hw_start(hdev, connect_mask);
2331         if (error) {
2332                 hid_err(hdev, "hw start failed\n");
2333                 goto fail;
2334         }
2335
2336         if (!wireless) {
2337                 /* Note that if query fails it is not a hard failure */
2338                 wacom_query_tablet_data(wacom);
2339         }
2340
2341         /* touch only Bamboo doesn't support pen */
2342         if ((features->type == BAMBOO_TOUCH) &&
2343             (features->device_type & WACOM_DEVICETYPE_PEN)) {
2344                 cancel_delayed_work_sync(&wacom->init_work);
2345                 _wacom_query_tablet_data(wacom);
2346                 error = -ENODEV;
2347                 goto fail_quirks;
2348         }
2349
2350         if (features->device_type & WACOM_DEVICETYPE_WL_MONITOR)
2351                 error = hid_hw_open(hdev);
2352
2353         wacom_set_shared_values(wacom_wac);
2354         devres_close_group(&hdev->dev, wacom);
2355
2356         return 0;
2357
2358 fail_quirks:
2359         hid_hw_stop(hdev);
2360 fail:
2361         wacom_release_resources(wacom);
2362         return error;
2363 }
2364
2365 static void wacom_wireless_work(struct work_struct *work)
2366 {
2367         struct wacom *wacom = container_of(work, struct wacom, wireless_work);
2368         struct usb_device *usbdev = wacom->usbdev;
2369         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2370         struct hid_device *hdev1, *hdev2;
2371         struct wacom *wacom1, *wacom2;
2372         struct wacom_wac *wacom_wac1, *wacom_wac2;
2373         int error;
2374
2375         /*
2376          * Regardless if this is a disconnect or a new tablet,
2377          * remove any existing input and battery devices.
2378          */
2379
2380         wacom_destroy_battery(wacom);
2381
2382         /* Stylus interface */
2383         hdev1 = usb_get_intfdata(usbdev->config->interface[1]);
2384         wacom1 = hid_get_drvdata(hdev1);
2385         wacom_wac1 = &(wacom1->wacom_wac);
2386         wacom_release_resources(wacom1);
2387
2388         /* Touch interface */
2389         hdev2 = usb_get_intfdata(usbdev->config->interface[2]);
2390         wacom2 = hid_get_drvdata(hdev2);
2391         wacom_wac2 = &(wacom2->wacom_wac);
2392         wacom_release_resources(wacom2);
2393
2394         if (wacom_wac->pid == 0) {
2395                 hid_info(wacom->hdev, "wireless tablet disconnected\n");
2396         } else {
2397                 const struct hid_device_id *id = wacom_ids;
2398
2399                 hid_info(wacom->hdev, "wireless tablet connected with PID %x\n",
2400                          wacom_wac->pid);
2401
2402                 while (id->bus) {
2403                         if (id->vendor == USB_VENDOR_ID_WACOM &&
2404                             id->product == wacom_wac->pid)
2405                                 break;
2406                         id++;
2407                 }
2408
2409                 if (!id->bus) {
2410                         hid_info(wacom->hdev, "ignoring unknown PID.\n");
2411                         return;
2412                 }
2413
2414                 /* Stylus interface */
2415                 wacom_wac1->features =
2416                         *((struct wacom_features *)id->driver_data);
2417
2418                 wacom_wac1->pid = wacom_wac->pid;
2419                 hid_hw_stop(hdev1);
2420                 error = wacom_parse_and_register(wacom1, true);
2421                 if (error)
2422                         goto fail;
2423
2424                 /* Touch interface */
2425                 if (wacom_wac1->features.touch_max ||
2426                     (wacom_wac1->features.type >= INTUOSHT &&
2427                     wacom_wac1->features.type <= BAMBOO_PT)) {
2428                         wacom_wac2->features =
2429                                 *((struct wacom_features *)id->driver_data);
2430                         wacom_wac2->pid = wacom_wac->pid;
2431                         hid_hw_stop(hdev2);
2432                         error = wacom_parse_and_register(wacom2, true);
2433                         if (error)
2434                                 goto fail;
2435                 }
2436
2437                 strlcpy(wacom_wac->name, wacom_wac1->name,
2438                         sizeof(wacom_wac->name));
2439                 error = wacom_initialize_battery(wacom);
2440                 if (error)
2441                         goto fail;
2442         }
2443
2444         return;
2445
2446 fail:
2447         wacom_release_resources(wacom1);
2448         wacom_release_resources(wacom2);
2449         return;
2450 }
2451
2452 static void wacom_remote_destroy_one(struct wacom *wacom, unsigned int index)
2453 {
2454         struct wacom_remote *remote = wacom->remote;
2455         u32 serial = remote->remotes[index].serial;
2456         int i;
2457         unsigned long flags;
2458
2459         for (i = 0; i < WACOM_MAX_REMOTES; i++) {
2460                 if (remote->remotes[i].serial == serial) {
2461
2462                         spin_lock_irqsave(&remote->remote_lock, flags);
2463                         remote->remotes[i].registered = false;
2464                         spin_unlock_irqrestore(&remote->remote_lock, flags);
2465
2466                         if (remote->remotes[i].battery.battery)
2467                                 devres_release_group(&wacom->hdev->dev,
2468                                                      &remote->remotes[i].battery.bat_desc);
2469
2470                         if (remote->remotes[i].group.name)
2471                                 devres_release_group(&wacom->hdev->dev,
2472                                                      &remote->remotes[i]);
2473
2474                         remote->remotes[i].serial = 0;
2475                         remote->remotes[i].group.name = NULL;
2476                         remote->remotes[i].battery.battery = NULL;
2477                         wacom->led.groups[i].select = WACOM_STATUS_UNKNOWN;
2478                 }
2479         }
2480 }
2481
2482 static int wacom_remote_create_one(struct wacom *wacom, u32 serial,
2483                                    unsigned int index)
2484 {
2485         struct wacom_remote *remote = wacom->remote;
2486         struct device *dev = &wacom->hdev->dev;
2487         int error, k;
2488
2489         /* A remote can pair more than once with an EKR,
2490          * check to make sure this serial isn't already paired.
2491          */
2492         for (k = 0; k < WACOM_MAX_REMOTES; k++) {
2493                 if (remote->remotes[k].serial == serial)
2494                         break;
2495         }
2496
2497         if (k < WACOM_MAX_REMOTES) {
2498                 remote->remotes[index].serial = serial;
2499                 return 0;
2500         }
2501
2502         if (!devres_open_group(dev, &remote->remotes[index], GFP_KERNEL))
2503                 return -ENOMEM;
2504
2505         error = wacom_remote_create_attr_group(wacom, serial, index);
2506         if (error)
2507                 goto fail;
2508
2509         remote->remotes[index].input = wacom_allocate_input(wacom);
2510         if (!remote->remotes[index].input) {
2511                 error = -ENOMEM;
2512                 goto fail;
2513         }
2514         remote->remotes[index].input->uniq = remote->remotes[index].group.name;
2515         remote->remotes[index].input->name = wacom->wacom_wac.pad_name;
2516
2517         if (!remote->remotes[index].input->name) {
2518                 error = -EINVAL;
2519                 goto fail;
2520         }
2521
2522         error = wacom_setup_pad_input_capabilities(remote->remotes[index].input,
2523                                                    &wacom->wacom_wac);
2524         if (error)
2525                 goto fail;
2526
2527         remote->remotes[index].serial = serial;
2528
2529         error = input_register_device(remote->remotes[index].input);
2530         if (error)
2531                 goto fail;
2532
2533         error = wacom_led_groups_alloc_and_register_one(
2534                                         &remote->remotes[index].input->dev,
2535                                         wacom, index, 3, true);
2536         if (error)
2537                 goto fail;
2538
2539         remote->remotes[index].registered = true;
2540
2541         devres_close_group(dev, &remote->remotes[index]);
2542         return 0;
2543
2544 fail:
2545         devres_release_group(dev, &remote->remotes[index]);
2546         remote->remotes[index].serial = 0;
2547         return error;
2548 }
2549
2550 static int wacom_remote_attach_battery(struct wacom *wacom, int index)
2551 {
2552         struct wacom_remote *remote = wacom->remote;
2553         int error;
2554
2555         if (!remote->remotes[index].registered)
2556                 return 0;
2557
2558         if (remote->remotes[index].battery.battery)
2559                 return 0;
2560
2561         if (wacom->led.groups[index].select == WACOM_STATUS_UNKNOWN)
2562                 return 0;
2563
2564         error = __wacom_initialize_battery(wacom,
2565                                         &wacom->remote->remotes[index].battery);
2566         if (error)
2567                 return error;
2568
2569         return 0;
2570 }
2571
2572 static void wacom_remote_work(struct work_struct *work)
2573 {
2574         struct wacom *wacom = container_of(work, struct wacom, remote_work);
2575         struct wacom_remote *remote = wacom->remote;
2576         struct wacom_remote_data data;
2577         unsigned long flags;
2578         unsigned int count;
2579         u32 serial;
2580         int i;
2581
2582         spin_lock_irqsave(&remote->remote_lock, flags);
2583
2584         count = kfifo_out(&remote->remote_fifo, &data, sizeof(data));
2585
2586         if (count != sizeof(data)) {
2587                 hid_err(wacom->hdev,
2588                         "workitem triggered without status available\n");
2589                 spin_unlock_irqrestore(&remote->remote_lock, flags);
2590                 return;
2591         }
2592
2593         if (!kfifo_is_empty(&remote->remote_fifo))
2594                 wacom_schedule_work(&wacom->wacom_wac, WACOM_WORKER_REMOTE);
2595
2596         spin_unlock_irqrestore(&remote->remote_lock, flags);
2597
2598         for (i = 0; i < WACOM_MAX_REMOTES; i++) {
2599                 serial = data.remote[i].serial;
2600                 if (data.remote[i].connected) {
2601
2602                         if (remote->remotes[i].serial == serial) {
2603                                 wacom_remote_attach_battery(wacom, i);
2604                                 continue;
2605                         }
2606
2607                         if (remote->remotes[i].serial)
2608                                 wacom_remote_destroy_one(wacom, i);
2609
2610                         wacom_remote_create_one(wacom, serial, i);
2611
2612                 } else if (remote->remotes[i].serial) {
2613                         wacom_remote_destroy_one(wacom, i);
2614                 }
2615         }
2616 }
2617
2618 static void wacom_mode_change_work(struct work_struct *work)
2619 {
2620         struct wacom *wacom = container_of(work, struct wacom, mode_change_work);
2621         struct wacom_shared *shared = wacom->wacom_wac.shared;
2622         struct wacom *wacom1 = NULL;
2623         struct wacom *wacom2 = NULL;
2624         bool is_direct = wacom->wacom_wac.is_direct_mode;
2625         int error = 0;
2626
2627         if (shared->pen) {
2628                 wacom1 = hid_get_drvdata(shared->pen);
2629                 wacom_release_resources(wacom1);
2630                 hid_hw_stop(wacom1->hdev);
2631                 wacom1->wacom_wac.has_mode_change = true;
2632                 wacom1->wacom_wac.is_direct_mode = is_direct;
2633         }
2634
2635         if (shared->touch) {
2636                 wacom2 = hid_get_drvdata(shared->touch);
2637                 wacom_release_resources(wacom2);
2638                 hid_hw_stop(wacom2->hdev);
2639                 wacom2->wacom_wac.has_mode_change = true;
2640                 wacom2->wacom_wac.is_direct_mode = is_direct;
2641         }
2642
2643         if (wacom1) {
2644                 error = wacom_parse_and_register(wacom1, false);
2645                 if (error)
2646                         return;
2647         }
2648
2649         if (wacom2) {
2650                 error = wacom_parse_and_register(wacom2, false);
2651                 if (error)
2652                         return;
2653         }
2654
2655         return;
2656 }
2657
2658 static int wacom_probe(struct hid_device *hdev,
2659                 const struct hid_device_id *id)
2660 {
2661         struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
2662         struct usb_device *dev = interface_to_usbdev(intf);
2663         struct wacom *wacom;
2664         struct wacom_wac *wacom_wac;
2665         struct wacom_features *features;
2666         int error;
2667
2668         if (!id->driver_data)
2669                 return -EINVAL;
2670
2671         hdev->quirks |= HID_QUIRK_NO_INIT_REPORTS;
2672
2673         /* hid-core sets this quirk for the boot interface */
2674         hdev->quirks &= ~HID_QUIRK_NOGET;
2675
2676         wacom = devm_kzalloc(&hdev->dev, sizeof(struct wacom), GFP_KERNEL);
2677         if (!wacom)
2678                 return -ENOMEM;
2679
2680         hid_set_drvdata(hdev, wacom);
2681         wacom->hdev = hdev;
2682
2683         wacom_wac = &wacom->wacom_wac;
2684         wacom_wac->features = *((struct wacom_features *)id->driver_data);
2685         features = &wacom_wac->features;
2686
2687         if (features->check_for_hid_type && features->hid_type != hdev->type) {
2688                 error = -ENODEV;
2689                 goto fail;
2690         }
2691
2692         error = kfifo_alloc(&wacom_wac->pen_fifo, WACOM_PKGLEN_MAX, GFP_KERNEL);
2693         if (error)
2694                 goto fail;
2695
2696         wacom_wac->hid_data.inputmode = -1;
2697         wacom_wac->mode_report = -1;
2698
2699         wacom->usbdev = dev;
2700         wacom->intf = intf;
2701         mutex_init(&wacom->lock);
2702         INIT_DELAYED_WORK(&wacom->init_work, wacom_init_work);
2703         INIT_WORK(&wacom->wireless_work, wacom_wireless_work);
2704         INIT_WORK(&wacom->battery_work, wacom_battery_work);
2705         INIT_WORK(&wacom->remote_work, wacom_remote_work);
2706         INIT_WORK(&wacom->mode_change_work, wacom_mode_change_work);
2707
2708         /* ask for the report descriptor to be loaded by HID */
2709         error = hid_parse(hdev);
2710         if (error) {
2711                 hid_err(hdev, "parse failed\n");
2712                 goto fail;
2713         }
2714
2715         error = wacom_parse_and_register(wacom, false);
2716         if (error)
2717                 goto fail;
2718
2719         if (hdev->bus == BUS_BLUETOOTH) {
2720                 error = device_create_file(&hdev->dev, &dev_attr_speed);
2721                 if (error)
2722                         hid_warn(hdev,
2723                                  "can't create sysfs speed attribute err: %d\n",
2724                                  error);
2725         }
2726
2727         return 0;
2728
2729 fail:
2730         hid_set_drvdata(hdev, NULL);
2731         return error;
2732 }
2733
2734 static void wacom_remove(struct hid_device *hdev)
2735 {
2736         struct wacom *wacom = hid_get_drvdata(hdev);
2737         struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2738         struct wacom_features *features = &wacom_wac->features;
2739
2740         if (features->device_type & WACOM_DEVICETYPE_WL_MONITOR)
2741                 hid_hw_close(hdev);
2742
2743         hid_hw_stop(hdev);
2744
2745         cancel_delayed_work_sync(&wacom->init_work);
2746         cancel_work_sync(&wacom->wireless_work);
2747         cancel_work_sync(&wacom->battery_work);
2748         cancel_work_sync(&wacom->remote_work);
2749         cancel_work_sync(&wacom->mode_change_work);
2750         if (hdev->bus == BUS_BLUETOOTH)
2751                 device_remove_file(&hdev->dev, &dev_attr_speed);
2752
2753         /* make sure we don't trigger the LEDs */
2754         wacom_led_groups_release(wacom);
2755
2756         if (wacom->wacom_wac.features.type != REMOTE)
2757                 wacom_release_resources(wacom);
2758
2759         kfifo_free(&wacom_wac->pen_fifo);
2760
2761         hid_set_drvdata(hdev, NULL);
2762 }
2763
2764 #ifdef CONFIG_PM
2765 static int wacom_resume(struct hid_device *hdev)
2766 {
2767         struct wacom *wacom = hid_get_drvdata(hdev);
2768
2769         mutex_lock(&wacom->lock);
2770
2771         /* switch to wacom mode first */
2772         _wacom_query_tablet_data(wacom);
2773         wacom_led_control(wacom);
2774
2775         mutex_unlock(&wacom->lock);
2776
2777         return 0;
2778 }
2779
2780 static int wacom_reset_resume(struct hid_device *hdev)
2781 {
2782         return wacom_resume(hdev);
2783 }
2784 #endif /* CONFIG_PM */
2785
2786 static struct hid_driver wacom_driver = {
2787         .name =         "wacom",
2788         .id_table =     wacom_ids,
2789         .probe =        wacom_probe,
2790         .remove =       wacom_remove,
2791         .report =       wacom_wac_report,
2792 #ifdef CONFIG_PM
2793         .resume =       wacom_resume,
2794         .reset_resume = wacom_reset_resume,
2795 #endif
2796         .raw_event =    wacom_raw_event,
2797 };
2798 module_hid_driver(wacom_driver);
2799
2800 MODULE_VERSION(DRIVER_VERSION);
2801 MODULE_AUTHOR(DRIVER_AUTHOR);
2802 MODULE_DESCRIPTION(DRIVER_DESC);
2803 MODULE_LICENSE("GPL");