wacom_sys.c 21.4 KB
Newer Older
1
/*
2
 * drivers/input/tablet/wacom_sys.c
3
 *
4
 *  USB Wacom tablet support - system specific code
5 6 7 8 9 10 11 12 13 14
 */

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#include "wacom_wac.h"
15
#include "wacom.h"
16

17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
/* defines to get HID report descriptor */
#define HID_DEVICET_HID		(USB_TYPE_CLASS | 0x01)
#define HID_DEVICET_REPORT	(USB_TYPE_CLASS | 0x02)
#define HID_USAGE_UNDEFINED		0x00
#define HID_USAGE_PAGE			0x05
#define HID_USAGE_PAGE_DIGITIZER	0x0d
#define HID_USAGE_PAGE_DESKTOP		0x01
#define HID_USAGE			0x09
#define HID_USAGE_X			0x30
#define HID_USAGE_Y			0x31
#define HID_USAGE_X_TILT		0x3d
#define HID_USAGE_Y_TILT		0x3e
#define HID_USAGE_FINGER		0x22
#define HID_USAGE_STYLUS		0x20
#define HID_COLLECTION			0xc0

enum {
	WCM_UNDEFINED = 0,
	WCM_DESKTOP,
	WCM_DIGITIZER,
};

struct hid_descriptor {
	struct usb_descriptor_header header;
	__le16   bcdHID;
	u8       bCountryCode;
	u8       bNumDescriptors;
	u8       bDescriptorType;
	__le16   wDescriptorLength;
} __attribute__ ((packed));

/* defines to get/set USB message */
49 50
#define USB_REQ_GET_REPORT	0x01
#define USB_REQ_SET_REPORT	0x09
51

52
#define WAC_HID_FEATURE_REPORT	0x03
53
#define WAC_MSG_RETRIES		5
54

55 56 57 58 59 60 61
#define WAC_CMD_LED_CONTROL	0x20
#define WAC_CMD_ICON_START	0x21
#define WAC_CMD_ICON_XFER	0x23
#define WAC_CMD_RETRIES		10

static int wacom_get_report(struct usb_interface *intf, u8 type, u8 id,
			    void *buf, size_t size, unsigned int retries)
62
{
63 64 65 66 67 68 69 70 71 72 73 74 75
	struct usb_device *dev = interface_to_usbdev(intf);
	int retval;

	do {
		retval = usb_control_msg(dev, usb_rcvctrlpipe(dev, 0),
				USB_REQ_GET_REPORT,
				USB_TYPE_CLASS | USB_RECIP_INTERFACE,
				(type << 8) + id,
				intf->altsetting[0].desc.bInterfaceNumber,
				buf, size, 100);
	} while ((retval == -ETIMEDOUT || retval == -EPIPE) && --retries);

	return retval;
76 77
}

78 79
static int wacom_set_report(struct usb_interface *intf, u8 type, u8 id,
			    void *buf, size_t size, unsigned int retries)
80
{
81 82 83 84 85 86 87 88 89 90 91 92 93
	struct usb_device *dev = interface_to_usbdev(intf);
	int retval;

	do {
		retval = usb_control_msg(dev, usb_sndctrlpipe(dev, 0),
				USB_REQ_SET_REPORT,
				USB_TYPE_CLASS | USB_RECIP_INTERFACE,
				(type << 8) + id,
				intf->altsetting[0].desc.bInterfaceNumber,
				buf, size, 1000);
	} while ((retval == -ETIMEDOUT || retval == -EPIPE) && --retries);

	return retval;
94 95
}

96
static void wacom_sys_irq(struct urb *urb)
97 98 99 100 101 102 103 104 105 106 107 108
{
	struct wacom *wacom = urb->context;
	int retval;

	switch (urb->status) {
	case 0:
		/* success */
		break;
	case -ECONNRESET:
	case -ENOENT:
	case -ESHUTDOWN:
		/* this urb is terminated, clean up */
109
		dbg("%s - urb shutting down with status: %d", __func__, urb->status);
110 111
		return;
	default:
112
		dbg("%s - nonzero urb status received: %d", __func__, urb->status);
113 114 115
		goto exit;
	}

116
	wacom_wac_irq(&wacom->wacom_wac, urb->actual_length);
117 118

 exit:
119
	usb_mark_last_busy(wacom->usbdev);
120
	retval = usb_submit_urb(urb, GFP_ATOMIC);
121 122
	if (retval)
		err ("%s - usb_submit_urb failed with result %d",
123
		     __func__, retval);
124 125 126 127
}

static int wacom_open(struct input_dev *dev)
{
128
	struct wacom *wacom = input_get_drvdata(dev);
129
	int retval = 0;
130

131
	if (usb_autopm_get_interface(wacom->intf) < 0)
132
		return -EIO;
133 134

	mutex_lock(&wacom->lock);
135 136

	if (usb_submit_urb(wacom->irq, GFP_KERNEL)) {
137 138
		retval = -EIO;
		goto out;
139 140
	}

141
	wacom->open = true;
142
	wacom->intf->needs_remote_wakeup = 1;
143

144
out:
145
	mutex_unlock(&wacom->lock);
146
	usb_autopm_put_interface(wacom->intf);
147
	return retval;
148 149 150 151
}

static void wacom_close(struct input_dev *dev)
{
152
	struct wacom *wacom = input_get_drvdata(dev);
153 154 155
	int autopm_error;

	autopm_error = usb_autopm_get_interface(wacom->intf);
156

157
	mutex_lock(&wacom->lock);
158
	usb_kill_urb(wacom->irq);
159
	wacom->open = false;
160 161
	wacom->intf->needs_remote_wakeup = 0;
	mutex_unlock(&wacom->lock);
162

163 164
	if (!autopm_error)
		usb_autopm_put_interface(wacom->intf);
165 166
}

167
static int wacom_parse_hid(struct usb_interface *intf, struct hid_descriptor *hid_desc,
168
			   struct wacom_features *features)
169 170
{
	struct usb_device *dev = interface_to_usbdev(intf);
171 172 173
	char limit = 0;
	/* result has to be defined as int for some devices */
	int result = 0;
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
	int i = 0, usage = WCM_UNDEFINED, finger = 0, pen = 0;
	unsigned char *report;

	report = kzalloc(hid_desc->wDescriptorLength, GFP_KERNEL);
	if (!report)
		return -ENOMEM;

	/* retrive report descriptors */
	do {
		result = usb_control_msg(dev, usb_rcvctrlpipe(dev, 0),
			USB_REQ_GET_DESCRIPTOR,
			USB_RECIP_INTERFACE | USB_DIR_IN,
			HID_DEVICET_REPORT << 8,
			intf->altsetting[0].desc.bInterfaceNumber, /* interface */
			report,
			hid_desc->wDescriptorLength,
			5000); /* 5 secs */
191
	} while (result < 0 && limit++ < WAC_MSG_RETRIES);
192

193
	/* No need to parse the Descriptor. It isn't an error though */
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
	if (result < 0)
		goto out;

	for (i = 0; i < hid_desc->wDescriptorLength; i++) {

		switch (report[i]) {
		case HID_USAGE_PAGE:
			switch (report[i + 1]) {
			case HID_USAGE_PAGE_DIGITIZER:
				usage = WCM_DIGITIZER;
				i++;
				break;

			case HID_USAGE_PAGE_DESKTOP:
				usage = WCM_DESKTOP;
				i++;
				break;
			}
			break;

		case HID_USAGE:
			switch (report[i + 1]) {
			case HID_USAGE_X:
				if (usage == WCM_DESKTOP) {
					if (finger) {
219
						features->device_type = BTN_TOOL_FINGER;
220 221 222
						if (features->type == TABLETPC2FG) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_TPC2FG;
223
							features->device_type = BTN_TOOL_DOUBLETAP;
224
						}
225 226 227
						if (features->type == BAMBOO_PT) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_BBTOUCH;
228
							features->device_type = BTN_TOOL_DOUBLETAP;
229 230 231 232 233 234 235 236 237 238 239 240 241 242
							features->x_phy =
								get_unaligned_le16(&report[i + 5]);
							features->x_max =
								get_unaligned_le16(&report[i + 8]);
							i += 15;
						} else {
							features->x_max =
								get_unaligned_le16(&report[i + 3]);
							features->x_phy =
								get_unaligned_le16(&report[i + 6]);
							features->unit = report[i + 9];
							features->unitExpo = report[i + 11];
							i += 12;
						}
243
					} else if (pen) {
244 245
						/* penabled only accepts exact bytes of data */
						if (features->type == TABLETPC2FG)
246
							features->pktlen = WACOM_PKGLEN_GRAPHIRE;
247 248
						if (features->type == BAMBOO_PT)
							features->pktlen = WACOM_PKGLEN_BBFUN;
249
						features->device_type = BTN_TOOL_PEN;
250
						features->x_max =
251
							get_unaligned_le16(&report[i + 3]);
252 253 254 255 256 257 258 259 260 261 262 263 264
						i += 4;
					}
				} else if (usage == WCM_DIGITIZER) {
					/* max pressure isn't reported
					features->pressure_max = (unsigned short)
							(report[i+4] << 8  | report[i + 3]);
					*/
					features->pressure_max = 255;
					i += 4;
				}
				break;

			case HID_USAGE_Y:
265 266
				if (usage == WCM_DESKTOP) {
					if (finger) {
267
						features->device_type = BTN_TOOL_FINGER;
268 269 270
						if (features->type == TABLETPC2FG) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_TPC2FG;
271
							features->device_type = BTN_TOOL_DOUBLETAP;
272
							features->y_max =
273
								get_unaligned_le16(&report[i + 3]);
274
							features->y_phy =
275
								get_unaligned_le16(&report[i + 6]);
276
							i += 7;
277 278 279
						} else if (features->type == BAMBOO_PT) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_BBTOUCH;
280
							features->device_type = BTN_TOOL_DOUBLETAP;
281 282 283 284 285
							features->y_phy =
								get_unaligned_le16(&report[i + 3]);
							features->y_max =
								get_unaligned_le16(&report[i + 6]);
							i += 12;
286 287 288 289
						} else {
							features->y_max =
								features->x_max;
							features->y_phy =
290
								get_unaligned_le16(&report[i + 3]);
291 292 293 294 295
							i += 4;
						}
					} else if (pen) {
						/* penabled only accepts exact bytes of data */
						if (features->type == TABLETPC2FG)
296
							features->pktlen = WACOM_PKGLEN_GRAPHIRE;
297 298
						if (features->type == BAMBOO_PT)
							features->pktlen = WACOM_PKGLEN_BBFUN;
299 300
						features->device_type = BTN_TOOL_PEN;
						features->y_max =
301
							get_unaligned_le16(&report[i + 3]);
302 303 304
						i += 4;
					}
				}
305 306 307 308 309 310 311 312 313 314 315 316 317 318 319
				break;

			case HID_USAGE_FINGER:
				finger = 1;
				i++;
				break;

			case HID_USAGE_STYLUS:
				pen = 1;
				i++;
				break;

			case HID_USAGE_UNDEFINED:
				if (usage == WCM_DESKTOP && finger) /* capacity */
					features->pressure_max =
320
						get_unaligned_le16(&report[i + 3]);
321 322 323 324 325 326
				i += 4;
				break;
			}
			break;

		case HID_COLLECTION:
327
			/* reset UsagePage and Finger */
328 329 330 331 332 333
			finger = usage = 0;
			break;
		}
	}

 out:
334
	result = 0;
335 336 337 338
	kfree(report);
	return result;
}

339
static int wacom_query_tablet_data(struct usb_interface *intf, struct wacom_features *features)
340 341
{
	unsigned char *rep_data;
342 343
	int limit = 0, report_id = 2;
	int error = -ENOMEM;
344

345
	rep_data = kmalloc(4, GFP_KERNEL);
346
	if (!rep_data)
347 348
		return error;

349
	/* ask to report tablet data if it is MT Tablet PC or
350 351
	 * not a Tablet PC */
	if (features->type == TABLETPC2FG) {
352 353 354
		do {
			rep_data[0] = 3;
			rep_data[1] = 4;
355 356
			rep_data[2] = 0;
			rep_data[3] = 0;
357
			report_id = 3;
358 359
			error = wacom_set_report(intf, WAC_HID_FEATURE_REPORT,
						 report_id, rep_data, 4, 1);
360
			if (error >= 0)
361 362 363
				error = wacom_get_report(intf,
						WAC_HID_FEATURE_REPORT,
						report_id, rep_data, 4, 1);
364
		} while ((error < 0 || rep_data[1] != 4) && limit++ < WAC_MSG_RETRIES);
365
	} else if (features->type != TABLETPC) {
366 367 368
		do {
			rep_data[0] = 2;
			rep_data[1] = 2;
369 370
			error = wacom_set_report(intf, WAC_HID_FEATURE_REPORT,
						 report_id, rep_data, 2, 1);
371
			if (error >= 0)
372 373 374
				error = wacom_get_report(intf,
						WAC_HID_FEATURE_REPORT,
						report_id, rep_data, 2, 1);
375
		} while ((error < 0 || rep_data[1] != 2) && limit++ < WAC_MSG_RETRIES);
376
	}
377 378 379 380 381 382

	kfree(rep_data);

	return error < 0 ? error : 0;
}

383 384 385 386 387 388 389
static int wacom_retrieve_hid_descriptor(struct usb_interface *intf,
		struct wacom_features *features)
{
	int error = 0;
	struct usb_host_interface *interface = intf->cur_altsetting;
	struct hid_descriptor *hid_desc;

390
	/* default features */
391
	features->device_type = BTN_TOOL_PEN;
392 393 394 395
	features->x_fuzz = 4;
	features->y_fuzz = 4;
	features->pressure_fuzz = 0;
	features->distance_fuzz = 0;
396

397
	/* only Tablet PCs and Bamboo P&T need to retrieve the info */
398 399
	if ((features->type != TABLETPC) && (features->type != TABLETPC2FG) &&
	    (features->type != BAMBOO_PT))
400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417
		goto out;

	if (usb_get_extra_descriptor(interface, HID_DEVICET_HID, &hid_desc)) {
		if (usb_get_extra_descriptor(&interface->endpoint[0],
				HID_DEVICET_REPORT, &hid_desc)) {
			printk("wacom: can not retrieve extra class descriptor\n");
			error = 1;
			goto out;
		}
	}
	error = wacom_parse_hid(intf, hid_desc, features);
	if (error)
		goto out;

 out:
	return error;
}

418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492
struct wacom_usbdev_data {
	struct list_head list;
	struct kref kref;
	struct usb_device *dev;
	struct wacom_shared shared;
};

static LIST_HEAD(wacom_udev_list);
static DEFINE_MUTEX(wacom_udev_list_lock);

static struct wacom_usbdev_data *wacom_get_usbdev_data(struct usb_device *dev)
{
	struct wacom_usbdev_data *data;

	list_for_each_entry(data, &wacom_udev_list, list) {
		if (data->dev == dev) {
			kref_get(&data->kref);
			return data;
		}
	}

	return NULL;
}

static int wacom_add_shared_data(struct wacom_wac *wacom,
				 struct usb_device *dev)
{
	struct wacom_usbdev_data *data;
	int retval = 0;

	mutex_lock(&wacom_udev_list_lock);

	data = wacom_get_usbdev_data(dev);
	if (!data) {
		data = kzalloc(sizeof(struct wacom_usbdev_data), GFP_KERNEL);
		if (!data) {
			retval = -ENOMEM;
			goto out;
		}

		kref_init(&data->kref);
		data->dev = dev;
		list_add_tail(&data->list, &wacom_udev_list);
	}

	wacom->shared = &data->shared;

out:
	mutex_unlock(&wacom_udev_list_lock);
	return retval;
}

static void wacom_release_shared_data(struct kref *kref)
{
	struct wacom_usbdev_data *data =
		container_of(kref, struct wacom_usbdev_data, kref);

	mutex_lock(&wacom_udev_list_lock);
	list_del(&data->list);
	mutex_unlock(&wacom_udev_list_lock);

	kfree(data);
}

static void wacom_remove_shared_data(struct wacom_wac *wacom)
{
	struct wacom_usbdev_data *data;

	if (wacom->shared) {
		data = container_of(wacom->shared, struct wacom_usbdev_data, shared);
		kref_put(&data->kref, wacom_release_shared_data);
		wacom->shared = NULL;
	}
}

493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680
static int wacom_led_control(struct wacom *wacom)
{
	unsigned char *buf;
	int retval;

	buf = kzalloc(9, GFP_KERNEL);
	if (!buf)
		return -ENOMEM;

	buf[0] = WAC_CMD_LED_CONTROL;
	buf[1] = wacom->led.select >= 0 ? wacom->led.select | 4 : 0;
	buf[2] = wacom->led.llv;
	buf[3] = wacom->led.hlv;
	buf[4] = wacom->led.img_lum;

	retval = wacom_set_report(wacom->intf, 0x03, WAC_CMD_LED_CONTROL,
				  buf, 9, WAC_CMD_RETRIES);
	kfree(buf);

	return retval;
}

static int wacom_led_putimage(struct wacom *wacom, int button_id, const void *img)
{
	unsigned char *buf;
	int i, retval;

	buf = kzalloc(259, GFP_KERNEL);
	if (!buf)
		return -ENOMEM;

	/* Send 'start' command */
	buf[0] = WAC_CMD_ICON_START;
	buf[1] = 1;
	retval = wacom_set_report(wacom->intf, 0x03, WAC_CMD_ICON_START,
				  buf, 2, WAC_CMD_RETRIES);
	if (retval < 0)
		goto out;

	buf[0] = WAC_CMD_ICON_XFER;
	buf[1] = button_id & 0x07;
	for (i = 0; i < 4; i++) {
		buf[2] = i;
		memcpy(buf + 3, img + i * 256, 256);

		retval = wacom_set_report(wacom->intf, 0x03, WAC_CMD_ICON_XFER,
					  buf, 259, WAC_CMD_RETRIES);
		if (retval < 0)
			break;
	}

	/* Send 'stop' */
	buf[0] = WAC_CMD_ICON_START;
	buf[1] = 0;
	wacom_set_report(wacom->intf, 0x03, WAC_CMD_ICON_START,
			 buf, 2, WAC_CMD_RETRIES);

out:
	kfree(buf);
	return retval;
}

static ssize_t wacom_led_select_store(struct device *dev,
				      struct device_attribute *attr,
				      const char *buf, size_t count)
{
	struct wacom *wacom = dev_get_drvdata(dev);
	unsigned int id;
	int err;

	err = kstrtouint(buf, 10, &id);
	if (err)
		return err;

	mutex_lock(&wacom->lock);

	wacom->led.select = id;
	err = wacom_led_control(wacom);

	mutex_unlock(&wacom->lock);

	return err < 0 ? err : count;
}

static DEVICE_ATTR(status_led_select, S_IWUSR, NULL, wacom_led_select_store);

static ssize_t wacom_luminance_store(struct wacom *wacom, u8 *dest,
				     const char *buf, size_t count)
{
	unsigned int value;
	int err;

	err = kstrtouint(buf, 10, &value);
	if (err)
		return err;

	mutex_lock(&wacom->lock);

	*dest = value & 0x7f;
	err = wacom_led_control(wacom);

	mutex_unlock(&wacom->lock);

	return err < 0 ? err : count;
}

#define DEVICE_LUMINANCE_ATTR(name, field)				\
static ssize_t wacom_##name##_luminance_store(struct device *dev,	\
	struct device_attribute *attr, const char *buf, size_t count)	\
{									\
	struct wacom *wacom = dev_get_drvdata(dev);			\
									\
	return wacom_luminance_store(wacom, &wacom->led.field,		\
				     buf, count);			\
}									\
static DEVICE_ATTR(name##_luminance, S_IWUSR,				\
		   NULL, wacom_##name##_luminance_store)

DEVICE_LUMINANCE_ATTR(status0, llv);
DEVICE_LUMINANCE_ATTR(status1, hlv);
DEVICE_LUMINANCE_ATTR(buttons, img_lum);

static ssize_t wacom_button_image_store(struct device *dev, int button_id,
					const char *buf, size_t count)
{
	struct wacom *wacom = dev_get_drvdata(dev);
	int err;

	if (count != 1024)
		return -EINVAL;

	mutex_lock(&wacom->lock);

	err = wacom_led_putimage(wacom, button_id, buf);

	mutex_unlock(&wacom->lock);

	return err < 0 ? err : count;
}

#define DEVICE_BTNIMG_ATTR(BUTTON_ID)					\
static ssize_t wacom_btnimg##BUTTON_ID##_store(struct device *dev,	\
	struct device_attribute *attr, const char *buf, size_t count)	\
{									\
	return wacom_button_image_store(dev, BUTTON_ID, buf, count);	\
}									\
static DEVICE_ATTR(button##BUTTON_ID##_rawimg, S_IWUSR,			\
		   NULL, wacom_btnimg##BUTTON_ID##_store)

DEVICE_BTNIMG_ATTR(0);
DEVICE_BTNIMG_ATTR(1);
DEVICE_BTNIMG_ATTR(2);
DEVICE_BTNIMG_ATTR(3);
DEVICE_BTNIMG_ATTR(4);
DEVICE_BTNIMG_ATTR(5);
DEVICE_BTNIMG_ATTR(6);
DEVICE_BTNIMG_ATTR(7);

static struct attribute *wacom_led_attrs[] = {
	&dev_attr_status0_luminance.attr,
	&dev_attr_status1_luminance.attr,
	&dev_attr_status_led_select.attr,
	&dev_attr_buttons_luminance.attr,
	&dev_attr_button0_rawimg.attr,
	&dev_attr_button1_rawimg.attr,
	&dev_attr_button2_rawimg.attr,
	&dev_attr_button3_rawimg.attr,
	&dev_attr_button4_rawimg.attr,
	&dev_attr_button5_rawimg.attr,
	&dev_attr_button6_rawimg.attr,
	&dev_attr_button7_rawimg.attr,
	NULL
};

static struct attribute_group wacom_led_attr_group = {
	.name = "wacom_led",
	.attrs = wacom_led_attrs,
};

static int wacom_initialize_leds(struct wacom *wacom)
{
	int error;

	if (wacom->wacom_wac.features.type >= INTUOS4 &&
	    wacom->wacom_wac.features.type <= INTUOS4L) {

		/* Initialize default values */
		wacom->led.select = 0;
681
		wacom->led.llv = 10;
682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706
		wacom->led.hlv = 20;
		wacom->led.img_lum = 10;
		wacom_led_control(wacom);

		error = sysfs_create_group(&wacom->intf->dev.kobj,
					   &wacom_led_attr_group);
		if (error) {
			dev_err(&wacom->intf->dev,
				"cannot create sysfs group err: %d\n", error);
			return error;
		}
	}

	return 0;
}

static void wacom_destroy_leds(struct wacom *wacom)
{
	if (wacom->wacom_wac.features.type >= INTUOS4 &&
	    wacom->wacom_wac.features.type <= INTUOS4L) {
		sysfs_remove_group(&wacom->intf->dev.kobj,
				   &wacom_led_attr_group);
	}
}

707 708 709 710 711 712
static int wacom_probe(struct usb_interface *intf, const struct usb_device_id *id)
{
	struct usb_device *dev = interface_to_usbdev(intf);
	struct usb_endpoint_descriptor *endpoint;
	struct wacom *wacom;
	struct wacom_wac *wacom_wac;
713
	struct wacom_features *features;
714
	struct input_dev *input_dev;
715
	int error;
716

717
	if (!id->driver_info)
718 719
		return -EINVAL;

720 721
	wacom = kzalloc(sizeof(struct wacom), GFP_KERNEL);
	input_dev = input_allocate_device();
722
	if (!wacom || !input_dev) {
723 724 725 726
		error = -ENOMEM;
		goto fail1;
	}

727
	wacom_wac = &wacom->wacom_wac;
728 729 730 731
	wacom_wac->features = *((struct wacom_features *)id->driver_info);
	features = &wacom_wac->features;
	if (features->pktlen > WACOM_PKGLEN_MAX) {
		error = -EINVAL;
732
		goto fail1;
733
	}
734

735 736
	wacom_wac->data = usb_alloc_coherent(dev, WACOM_PKGLEN_MAX,
					     GFP_KERNEL, &wacom->data_dma);
737 738
	if (!wacom_wac->data) {
		error = -ENOMEM;
739
		goto fail1;
740
	}
741 742

	wacom->irq = usb_alloc_urb(0, GFP_KERNEL);
743 744
	if (!wacom->irq) {
		error = -ENOMEM;
745
		goto fail2;
746
	}
747 748

	wacom->usbdev = dev;
749 750
	wacom->intf = intf;
	mutex_init(&wacom->lock);
751 752 753
	usb_make_path(dev, wacom->phys, sizeof(wacom->phys));
	strlcat(wacom->phys, "/input0", sizeof(wacom->phys));

754
	wacom_wac->input = input_dev;
755

756 757
	endpoint = &intf->cur_altsetting->endpoint[0].desc;

758 759 760
	/* Retrieve the physical and logical size for OEM devices */
	error = wacom_retrieve_hid_descriptor(intf, features);
	if (error)
761
		goto fail3;
762

763 764
	wacom_setup_device_quirks(features);

765 766
	strlcpy(wacom_wac->name, features->name, sizeof(wacom_wac->name));

767
	if (features->quirks & WACOM_QUIRK_MULTI_INPUT) {
768 769 770 771 772
		/* Append the device type to the name */
		strlcat(wacom_wac->name,
			features->device_type == BTN_TOOL_PEN ?
				" Pen" : " Finger",
			sizeof(wacom_wac->name));
773 774 775 776

		error = wacom_add_shared_data(wacom_wac, dev);
		if (error)
			goto fail3;
777 778
	}

779 780 781 782 783 784
	input_dev->name = wacom_wac->name;
	input_dev->dev.parent = &intf->dev;
	input_dev->open = wacom_open;
	input_dev->close = wacom_close;
	usb_to_input_id(dev, &input_dev->id);
	input_set_drvdata(input_dev, wacom);
785

786
	wacom_setup_input_capabilities(input_dev, wacom_wac);
787 788 789

	usb_fill_int_urb(wacom->irq, dev,
			 usb_rcvintpipe(dev, endpoint->bEndpointAddress),
790
			 wacom_wac->data, features->pktlen,
P
Ping Cheng 已提交
791
			 wacom_sys_irq, wacom, endpoint->bInterval);
792 793 794
	wacom->irq->transfer_dma = wacom->data_dma;
	wacom->irq->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;

795
	error = wacom_initialize_leds(wacom);
796
	if (error)
797
		goto fail4;
798

799 800 801 802
	error = input_register_device(input_dev);
	if (error)
		goto fail5;

803 804
	/* Note that if query fails it is not a hard failure */
	wacom_query_tablet_data(intf, features);
805 806 807 808

	usb_set_intfdata(intf, wacom);
	return 0;

809
 fail5: wacom_destroy_leds(wacom);
810
 fail4:	wacom_remove_shared_data(wacom_wac);
811
 fail3:	usb_free_urb(wacom->irq);
812
 fail2:	usb_free_coherent(dev, WACOM_PKGLEN_MAX, wacom_wac->data, wacom->data_dma);
813
 fail1:	input_free_device(input_dev);
814
	kfree(wacom);
815
	return error;
816 817 818 819
}

static void wacom_disconnect(struct usb_interface *intf)
{
820
	struct wacom *wacom = usb_get_intfdata(intf);
821 822

	usb_set_intfdata(intf, NULL);
823 824

	usb_kill_urb(wacom->irq);
825
	input_unregister_device(wacom->wacom_wac.input);
826
	wacom_destroy_leds(wacom);
827
	usb_free_urb(wacom->irq);
828
	usb_free_coherent(interface_to_usbdev(intf), WACOM_PKGLEN_MAX,
829 830
			wacom->wacom_wac.data, wacom->data_dma);
	wacom_remove_shared_data(&wacom->wacom_wac);
831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847
	kfree(wacom);
}

static int wacom_suspend(struct usb_interface *intf, pm_message_t message)
{
	struct wacom *wacom = usb_get_intfdata(intf);

	mutex_lock(&wacom->lock);
	usb_kill_urb(wacom->irq);
	mutex_unlock(&wacom->lock);

	return 0;
}

static int wacom_resume(struct usb_interface *intf)
{
	struct wacom *wacom = usb_get_intfdata(intf);
848
	struct wacom_features *features = &wacom->wacom_wac.features;
849
	int rv = 0;
850 851

	mutex_lock(&wacom->lock);
852 853 854

	/* switch to wacom mode first */
	wacom_query_tablet_data(intf, features);
855
	wacom_led_control(wacom);
856

857 858
	if (wacom->open && usb_submit_urb(wacom->irq, GFP_NOIO) < 0)
		rv = -EIO;
859

860 861 862 863 864 865 866 867
	mutex_unlock(&wacom->lock);

	return rv;
}

static int wacom_reset_resume(struct usb_interface *intf)
{
	return wacom_resume(intf);
868 869 870 871
}

static struct usb_driver wacom_driver = {
	.name =		"wacom",
872
	.id_table =	wacom_ids,
873 874
	.probe =	wacom_probe,
	.disconnect =	wacom_disconnect,
875 876 877 878
	.suspend =	wacom_suspend,
	.resume =	wacom_resume,
	.reset_resume =	wacom_reset_resume,
	.supports_autosuspend = 1,
879 880 881 882 883
};

static int __init wacom_init(void)
{
	int result;
884

885 886
	result = usb_register(&wacom_driver);
	if (result == 0)
887 888
		printk(KERN_INFO KBUILD_MODNAME ": " DRIVER_VERSION ":"
		       DRIVER_DESC "\n");
889 890 891 892 893 894 895 896 897 898
	return result;
}

static void __exit wacom_exit(void)
{
	usb_deregister(&wacom_driver);
}

module_init(wacom_init);
module_exit(wacom_exit);