[deliverable/linux.git] / drivers / input / tablet / wacom_sys.c

/*
 * drivers/input/tablet/wacom_sys.c
 *
 *  USB Wacom tablet support - system specific code
 */

/*
 * 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"
#include "wacom.h"

/* 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 */
#define USB_REQ_GET_REPORT	0x01
#define USB_REQ_SET_REPORT	0x09
#define WAC_HID_FEATURE_REPORT	0x03

static int usb_get_report(struct usb_interface *intf, unsigned char type,
				unsigned char id, void *buf, int size)
{
	return usb_control_msg(interface_to_usbdev(intf),
		usb_rcvctrlpipe(interface_to_usbdev(intf), 0),
		USB_REQ_GET_REPORT, USB_TYPE_CLASS | USB_RECIP_INTERFACE,
		(type << 8) + id, intf->altsetting[0].desc.bInterfaceNumber,
		buf, size, 100);
}

static int usb_set_report(struct usb_interface *intf, unsigned char type,
				unsigned char id, void *buf, int size)
{
	return usb_control_msg(interface_to_usbdev(intf),
		usb_sndctrlpipe(interface_to_usbdev(intf), 0),
                USB_REQ_SET_REPORT, USB_TYPE_CLASS | USB_RECIP_INTERFACE,
                (type << 8) + id, intf->altsetting[0].desc.bInterfaceNumber,
		buf, size, 1000);
}

static void wacom_sys_irq(struct urb *urb)
{
	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 */
		dbg("%s - urb shutting down with status: %d", __func__, urb->status);
		return;
	default:
		dbg("%s - nonzero urb status received: %d", __func__, urb->status);
		goto exit;
	}

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

 exit:
	usb_mark_last_busy(wacom->usbdev);
	retval = usb_submit_urb(urb, GFP_ATOMIC);
	if (retval)
		err ("%s - usb_submit_urb failed with result %d",
		     __func__, retval);
}

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

	if (usb_autopm_get_interface(wacom->intf) < 0)
		return -EIO;

	mutex_lock(&wacom->lock);

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

	wacom->open = true;
	wacom->intf->needs_remote_wakeup = 1;

out:
	mutex_unlock(&wacom->lock);
	usb_autopm_put_interface(wacom->intf);
	return retval;
}

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

	autopm_error = usb_autopm_get_interface(wacom->intf);

	mutex_lock(&wacom->lock);
	usb_kill_urb(wacom->irq);
	wacom->open = false;
	wacom->intf->needs_remote_wakeup = 0;
	mutex_unlock(&wacom->lock);

	if (!autopm_error)
		usb_autopm_put_interface(wacom->intf);
}

static int wacom_parse_hid(struct usb_interface *intf, struct hid_descriptor *hid_desc,
			   struct wacom_features *features)
{
	struct usb_device *dev = interface_to_usbdev(intf);
	char limit = 0;
	/* result has to be defined as int for some devices */
	int result = 0;
	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 */
	} while (result < 0 && limit++ < 5);

	/* No need to parse the Descriptor. It isn't an error though */
	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) {
						features->device_type = BTN_TOOL_FINGER;
						if (features->type == TABLETPC2FG) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_TPC2FG;
							features->device_type = BTN_TOOL_DOUBLETAP;
						}
						if (features->type == BAMBOO_PT) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_BBTOUCH;
							features->device_type = BTN_TOOL_DOUBLETAP;
							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;
						}
					} else if (pen) {
						/* penabled only accepts exact bytes of data */
						if (features->type == TABLETPC2FG)
							features->pktlen = WACOM_PKGLEN_GRAPHIRE;
						if (features->type == BAMBOO_PT)
							features->pktlen = WACOM_PKGLEN_BBFUN;
						features->device_type = BTN_TOOL_PEN;
						features->x_max =
							get_unaligned_le16(&report[i + 3]);
						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:
				if (usage == WCM_DESKTOP) {
					if (finger) {
						features->device_type = BTN_TOOL_FINGER;
						if (features->type == TABLETPC2FG) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_TPC2FG;
							features->device_type = BTN_TOOL_DOUBLETAP;
							features->y_max =
								get_unaligned_le16(&report[i + 3]);
							features->y_phy =
								get_unaligned_le16(&report[i + 6]);
							i += 7;
						} else if (features->type == BAMBOO_PT) {
							/* need to reset back */
							features->pktlen = WACOM_PKGLEN_BBTOUCH;
							features->device_type = BTN_TOOL_DOUBLETAP;
							features->y_phy =
								get_unaligned_le16(&report[i + 3]);
							features->y_max =
								get_unaligned_le16(&report[i + 6]);
							i += 12;
						} else {
							features->y_max =
								features->x_max;
							features->y_phy =
								get_unaligned_le16(&report[i + 3]);
							i += 4;
						}
					} else if (pen) {
						/* penabled only accepts exact bytes of data */
						if (features->type == TABLETPC2FG)
							features->pktlen = WACOM_PKGLEN_GRAPHIRE;
						if (features->type == BAMBOO_PT)
							features->pktlen = WACOM_PKGLEN_BBFUN;
						features->device_type = BTN_TOOL_PEN;
						features->y_max =
							get_unaligned_le16(&report[i + 3]);
						i += 4;
					}
				}
				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 =
						get_unaligned_le16(&report[i + 3]);
				i += 4;
				break;
			}
			break;

		case HID_COLLECTION:
			/* reset UsagePage and Finger */
			finger = usage = 0;
			break;
		}
	}

 out:
	result = 0;
	kfree(report);
	return result;
}

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

	rep_data = kmalloc(2, GFP_KERNEL);
	if (!rep_data)
		return error;

	/* ask to report tablet data if it is 2FGT Tablet PC or
	 * not a Tablet PC */
	if (features->type == TABLETPC2FG) {
		do {
			rep_data[0] = 3;
			rep_data[1] = 4;
			report_id = 3;
			error = usb_set_report(intf, WAC_HID_FEATURE_REPORT,
				report_id, rep_data, 2);
			if (error >= 0)
				error = usb_get_report(intf,
					WAC_HID_FEATURE_REPORT, report_id,
					rep_data, 3);
		} while ((error < 0 || rep_data[1] != 4) && limit++ < 5);
	} else if (features->type != TABLETPC) {
		do {
			rep_data[0] = 2;
			rep_data[1] = 2;
			error = usb_set_report(intf, WAC_HID_FEATURE_REPORT,
				report_id, rep_data, 2);
			if (error >= 0)
				error = usb_get_report(intf,
					WAC_HID_FEATURE_REPORT, report_id,
					rep_data, 2);
		} while ((error < 0 || rep_data[1] != 2) && limit++ < 5);
	}

	kfree(rep_data);

	return error < 0 ? error : 0;
}

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;

	/* default features */
	features->device_type = BTN_TOOL_PEN;
	features->x_fuzz = 4;
	features->y_fuzz = 4;
	features->pressure_fuzz = 0;
	features->distance_fuzz = 0;

	/* only Tablet PCs and Bamboo P&T need to retrieve the info */
	if ((features->type != TABLETPC) && (features->type != TABLETPC2FG) &&
	    (features->type != BAMBOO_PT))
		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;
}

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;
	}
}

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;
	struct wacom_features *features;
	struct input_dev *input_dev;
	int error;

	if (!id->driver_info)
		return -EINVAL;

	wacom = kzalloc(sizeof(struct wacom), GFP_KERNEL);
	input_dev = input_allocate_device();
	if (!wacom || !input_dev) {
		error = -ENOMEM;
		goto fail1;
	}

	wacom_wac = &wacom->wacom_wac;
	wacom_wac->features = *((struct wacom_features *)id->driver_info);
	features = &wacom_wac->features;
	if (features->pktlen > WACOM_PKGLEN_MAX) {
		error = -EINVAL;
		goto fail1;
	}

	wacom_wac->data = usb_alloc_coherent(dev, WACOM_PKGLEN_MAX,
					     GFP_KERNEL, &wacom->data_dma);
	if (!wacom_wac->data) {
		error = -ENOMEM;
		goto fail1;
	}

	wacom->irq = usb_alloc_urb(0, GFP_KERNEL);
	if (!wacom->irq) {
		error = -ENOMEM;
		goto fail2;
	}

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

	wacom_wac->input = input_dev;

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

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

	wacom_setup_device_quirks(features);

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

	if (features->quirks & WACOM_QUIRK_MULTI_INPUT) {
		/* Append the device type to the name */
		strlcat(wacom_wac->name,
			features->device_type == BTN_TOOL_PEN ?
				" Pen" : " Finger",
			sizeof(wacom_wac->name));

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

	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);

	wacom_setup_input_capabilities(input_dev, wacom_wac);

	usb_fill_int_urb(wacom->irq, dev,
			 usb_rcvintpipe(dev, endpoint->bEndpointAddress),
			 wacom_wac->data, features->pktlen,
			 wacom_sys_irq, wacom, endpoint->bInterval);
	wacom->irq->transfer_dma = wacom->data_dma;
	wacom->irq->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;

	error = input_register_device(input_dev);
	if (error)
		goto fail4;

	/* Note that if query fails it is not a hard failure */
	wacom_query_tablet_data(intf, features);

	usb_set_intfdata(intf, wacom);
	return 0;

 fail4:	wacom_remove_shared_data(wacom_wac);
 fail3:	usb_free_urb(wacom->irq);
 fail2:	usb_free_coherent(dev, WACOM_PKGLEN_MAX, wacom_wac->data, wacom->data_dma);
 fail1:	input_free_device(input_dev);
	kfree(wacom);
	return error;
}

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

	usb_set_intfdata(intf, NULL);

	usb_kill_urb(wacom->irq);
	input_unregister_device(wacom->wacom_wac.input);
	usb_free_urb(wacom->irq);
	usb_free_coherent(interface_to_usbdev(intf), WACOM_PKGLEN_MAX,
			wacom->wacom_wac.data, wacom->data_dma);
	wacom_remove_shared_data(&wacom->wacom_wac);
	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);
	struct wacom_features *features = &wacom->wacom_wac.features;
	int rv;

	mutex_lock(&wacom->lock);

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

	if (wacom->open)
		rv = usb_submit_urb(wacom->irq, GFP_NOIO);
	else
		rv = 0;

	mutex_unlock(&wacom->lock);

	return rv;
}

static int wacom_reset_resume(struct usb_interface *intf)
{
	return wacom_resume(intf);
}

static struct usb_driver wacom_driver = {
	.name =		"wacom",
	.id_table =	wacom_ids,
	.probe =	wacom_probe,
	.disconnect =	wacom_disconnect,
	.suspend =	wacom_suspend,
	.resume =	wacom_resume,
	.reset_resume =	wacom_reset_resume,
	.supports_autosuspend = 1,
};

static int __init wacom_init(void)
{
	int result;

	result = usb_register(&wacom_driver);
	if (result == 0)
		printk(KERN_INFO KBUILD_MODNAME ": " DRIVER_VERSION ":"
		       DRIVER_DESC "\n");
	return result;
}

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

module_init(wacom_init);
module_exit(wacom_exit);
Commit	Line	Data
3bea733a	1	/*
4104d13f	2	* drivers/input/tablet/wacom_sys.c
3bea733a	3	*
232f5693	4	* USB Wacom tablet support - system specific code
3bea733a PC	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
3bea733a	14	#include "wacom_wac.h"
51269fe8	15	#include "wacom.h"
3bea733a	16
545f4e99 PC	17	/* defines to get HID report descriptor */
	18	#define HID_DEVICET_HID (USB_TYPE_CLASS \| 0x01)
	19	#define HID_DEVICET_REPORT (USB_TYPE_CLASS \| 0x02)
	20	#define HID_USAGE_UNDEFINED 0x00
	21	#define HID_USAGE_PAGE 0x05
	22	#define HID_USAGE_PAGE_DIGITIZER 0x0d
	23	#define HID_USAGE_PAGE_DESKTOP 0x01
	24	#define HID_USAGE 0x09
	25	#define HID_USAGE_X 0x30
	26	#define HID_USAGE_Y 0x31
	27	#define HID_USAGE_X_TILT 0x3d
	28	#define HID_USAGE_Y_TILT 0x3e
	29	#define HID_USAGE_FINGER 0x22
	30	#define HID_USAGE_STYLUS 0x20
	31	#define HID_COLLECTION 0xc0
	32
	33	enum {
	34	WCM_UNDEFINED = 0,
	35	WCM_DESKTOP,
	36	WCM_DIGITIZER,
	37	};
	38
	39	struct hid_descriptor {
	40	struct usb_descriptor_header header;
	41	__le16 bcdHID;
	42	u8 bCountryCode;
	43	u8 bNumDescriptors;
	44	u8 bDescriptorType;
	45	__le16 wDescriptorLength;
	46	} __attribute__ ((packed));
	47
	48	/* defines to get/set USB message */
3bea733a PC	49	#define USB_REQ_GET_REPORT 0x01
3bea733a PC	50	#define USB_REQ_SET_REPORT 0x09
545f4e99	51	#define WAC_HID_FEATURE_REPORT 0x03
3bea733a PC	52
	53	static int usb_get_report(struct usb_interface *intf, unsigned char type,
	54	unsigned char id, void *buf, int size)
	55	{
	56	return usb_control_msg(interface_to_usbdev(intf),
	57	usb_rcvctrlpipe(interface_to_usbdev(intf), 0),
	58	USB_REQ_GET_REPORT, USB_TYPE_CLASS \| USB_RECIP_INTERFACE,
	59	(type << 8) + id, intf->altsetting[0].desc.bInterfaceNumber,
	60	buf, size, 100);
	61	}
	62
	63	static int usb_set_report(struct usb_interface *intf, unsigned char type,
	64	unsigned char id, void *buf, int size)
	65	{
	66	return usb_control_msg(interface_to_usbdev(intf),
	67	usb_sndctrlpipe(interface_to_usbdev(intf), 0),
	68	USB_REQ_SET_REPORT, USB_TYPE_CLASS \| USB_RECIP_INTERFACE,
	69	(type << 8) + id, intf->altsetting[0].desc.bInterfaceNumber,
	70	buf, size, 1000);
	71	}
	72
54c9b226	73	static void wacom_sys_irq(struct urb *urb)
3bea733a PC	74	{
3bea733a PC	75	struct wacom *wacom = urb->context;
3bea733a PC	76	int retval;
	77
	78	switch (urb->status) {
	79	case 0:
	80	/* success */
	81	break;
	82	case -ECONNRESET:
	83	case -ENOENT:
	84	case -ESHUTDOWN:
	85	/* this urb is terminated, clean up */
ea3e6c59	86	dbg("%s - urb shutting down with status: %d", __func__, urb->status);
3bea733a PC	87	return;
3bea733a PC	88	default:
ea3e6c59	89	dbg("%s - nonzero urb status received: %d", __func__, urb->status);
3bea733a PC	90	goto exit;
	91	}
	92
95dd3b30	93	wacom_wac_irq(&wacom->wacom_wac, urb->actual_length);
3bea733a PC	94
3bea733a PC	95	exit:
e7224094	96	usb_mark_last_busy(wacom->usbdev);
73a97f4f	97	retval = usb_submit_urb(urb, GFP_ATOMIC);
3bea733a PC	98	if (retval)
3bea733a PC	99	err ("%s - usb_submit_urb failed with result %d",
ea3e6c59	100	__func__, retval);
3bea733a PC	101	}
3bea733a PC	102
3bea733a PC	103	static int wacom_open(struct input_dev *dev)
3bea733a PC	104	{
7791bdae	105	struct wacom *wacom = input_get_drvdata(dev);
f6cd3783	106	int retval = 0;
3bea733a	107
f6cd3783	108	if (usb_autopm_get_interface(wacom->intf) < 0)
3bea733a	109	return -EIO;
f6cd3783 DT	110
f6cd3783 DT	111	mutex_lock(&wacom->lock);
e7224094 ON	112
e7224094 ON	113	if (usb_submit_urb(wacom->irq, GFP_KERNEL)) {
f6cd3783 DT	114	retval = -EIO;
f6cd3783 DT	115	goto out;
e7224094 ON	116	}
e7224094 ON	117
51269fe8	118	wacom->open = true;
e7224094	119	wacom->intf->needs_remote_wakeup = 1;
3bea733a	120
f6cd3783	121	out:
e7224094	122	mutex_unlock(&wacom->lock);
62ecae09	123	usb_autopm_put_interface(wacom->intf);
f6cd3783	124	return retval;
3bea733a PC	125	}
	126
	127	static void wacom_close(struct input_dev *dev)
	128	{
7791bdae	129	struct wacom *wacom = input_get_drvdata(dev);
62ecae09 DT	130	int autopm_error;
	131
	132	autopm_error = usb_autopm_get_interface(wacom->intf);
3bea733a	133
e7224094	134	mutex_lock(&wacom->lock);
3bea733a	135	usb_kill_urb(wacom->irq);
51269fe8	136	wacom->open = false;
e7224094 ON	137	wacom->intf->needs_remote_wakeup = 0;
e7224094 ON	138	mutex_unlock(&wacom->lock);
f6cd3783	139
62ecae09 DT	140	if (!autopm_error)
62ecae09 DT	141	usb_autopm_put_interface(wacom->intf);
3bea733a PC	142	}
3bea733a PC	143
545f4e99	144	static int wacom_parse_hid(struct usb_interface intf, struct hid_descriptor hid_desc,
ec67bbed	145	struct wacom_features *features)
545f4e99 PC	146	{
545f4e99 PC	147	struct usb_device *dev = interface_to_usbdev(intf);
ec67bbed PC	148	char limit = 0;
	149	/* result has to be defined as int for some devices */
	150	int result = 0;
545f4e99 PC	151	int i = 0, usage = WCM_UNDEFINED, finger = 0, pen = 0;
	152	unsigned char *report;
	153
	154	report = kzalloc(hid_desc->wDescriptorLength, GFP_KERNEL);
	155	if (!report)
	156	return -ENOMEM;
	157
	158	/* retrive report descriptors */
	159	do {
	160	result = usb_control_msg(dev, usb_rcvctrlpipe(dev, 0),
	161	USB_REQ_GET_DESCRIPTOR,
	162	USB_RECIP_INTERFACE \| USB_DIR_IN,
	163	HID_DEVICET_REPORT << 8,
	164	intf->altsetting[0].desc.bInterfaceNumber, /* interface */
	165	report,
	166	hid_desc->wDescriptorLength,
	167	5000); /* 5 secs */
	168	} while (result < 0 && limit++ < 5);
	169
384318ec	170	/* No need to parse the Descriptor. It isn't an error though */
545f4e99 PC	171	if (result < 0)
	172	goto out;
	173
	174	for (i = 0; i < hid_desc->wDescriptorLength; i++) {
	175
	176	switch (report[i]) {
	177	case HID_USAGE_PAGE:
	178	switch (report[i + 1]) {
	179	case HID_USAGE_PAGE_DIGITIZER:
	180	usage = WCM_DIGITIZER;
	181	i++;
	182	break;
	183
	184	case HID_USAGE_PAGE_DESKTOP:
	185	usage = WCM_DESKTOP;
	186	i++;
	187	break;
	188	}
	189	break;
	190
	191	case HID_USAGE:
	192	switch (report[i + 1]) {
	193	case HID_USAGE_X:
	194	if (usage == WCM_DESKTOP) {
	195	if (finger) {
84eb5aa6	196	features->device_type = BTN_TOOL_FINGER;
ec67bbed PC	197	if (features->type == TABLETPC2FG) {
	198	/* need to reset back */
	199	features->pktlen = WACOM_PKGLEN_TPC2FG;
84eb5aa6	200	features->device_type = BTN_TOOL_DOUBLETAP;
ec67bbed	201	}
4a88081e PC	202	if (features->type == BAMBOO_PT) {
	203	/* need to reset back */
	204	features->pktlen = WACOM_PKGLEN_BBTOUCH;
84eb5aa6	205	features->device_type = BTN_TOOL_DOUBLETAP;
4a88081e PC	206	features->x_phy =
	207	get_unaligned_le16(&report[i + 5]);
	208	features->x_max =
	209	get_unaligned_le16(&report[i + 8]);
	210	i += 15;
	211	} else {
	212	features->x_max =
	213	get_unaligned_le16(&report[i + 3]);
	214	features->x_phy =
	215	get_unaligned_le16(&report[i + 6]);
	216	features->unit = report[i + 9];
	217	features->unitExpo = report[i + 11];
	218	i += 12;
	219	}
545f4e99	220	} else if (pen) {
ec67bbed PC	221	/* penabled only accepts exact bytes of data */
ec67bbed PC	222	if (features->type == TABLETPC2FG)
57e413d9	223	features->pktlen = WACOM_PKGLEN_GRAPHIRE;
4a88081e PC	224	if (features->type == BAMBOO_PT)
4a88081e PC	225	features->pktlen = WACOM_PKGLEN_BBFUN;
ec67bbed	226	features->device_type = BTN_TOOL_PEN;
545f4e99	227	features->x_max =
252f7769	228	get_unaligned_le16(&report[i + 3]);
545f4e99 PC	229	i += 4;
	230	}
	231	} else if (usage == WCM_DIGITIZER) {
	232	/* max pressure isn't reported
	233	features->pressure_max = (unsigned short)
	234	(report[i+4] << 8 \| report[i + 3]);
	235	*/
	236	features->pressure_max = 255;
	237	i += 4;
	238	}
	239	break;
	240
	241	case HID_USAGE_Y:
ec67bbed PC	242	if (usage == WCM_DESKTOP) {
ec67bbed PC	243	if (finger) {
84eb5aa6	244	features->device_type = BTN_TOOL_FINGER;
ec67bbed PC	245	if (features->type == TABLETPC2FG) {
	246	/* need to reset back */
	247	features->pktlen = WACOM_PKGLEN_TPC2FG;
84eb5aa6	248	features->device_type = BTN_TOOL_DOUBLETAP;
ec67bbed	249	features->y_max =
252f7769	250	get_unaligned_le16(&report[i + 3]);
ec67bbed	251	features->y_phy =
252f7769	252	get_unaligned_le16(&report[i + 6]);
ec67bbed	253	i += 7;
4a88081e PC	254	} else if (features->type == BAMBOO_PT) {
	255	/* need to reset back */
	256	features->pktlen = WACOM_PKGLEN_BBTOUCH;
84eb5aa6	257	features->device_type = BTN_TOOL_DOUBLETAP;
4a88081e PC	258	features->y_phy =
	259	get_unaligned_le16(&report[i + 3]);
	260	features->y_max =
	261	get_unaligned_le16(&report[i + 6]);
	262	i += 12;
ec67bbed PC	263	} else {
	264	features->y_max =
	265	features->x_max;
	266	features->y_phy =
252f7769	267	get_unaligned_le16(&report[i + 3]);
ec67bbed PC	268	i += 4;
	269	}
	270	} else if (pen) {
	271	/* penabled only accepts exact bytes of data */
	272	if (features->type == TABLETPC2FG)
57e413d9	273	features->pktlen = WACOM_PKGLEN_GRAPHIRE;
4a88081e PC	274	if (features->type == BAMBOO_PT)
4a88081e PC	275	features->pktlen = WACOM_PKGLEN_BBFUN;
ec67bbed PC	276	features->device_type = BTN_TOOL_PEN;
ec67bbed PC	277	features->y_max =
252f7769	278	get_unaligned_le16(&report[i + 3]);
ec67bbed PC	279	i += 4;
	280	}
	281	}
545f4e99 PC	282	break;
	283
	284	case HID_USAGE_FINGER:
	285	finger = 1;
	286	i++;
	287	break;
	288
	289	case HID_USAGE_STYLUS:
	290	pen = 1;
	291	i++;
	292	break;
	293
	294	case HID_USAGE_UNDEFINED:
	295	if (usage == WCM_DESKTOP && finger) /* capacity */
	296	features->pressure_max =
252f7769	297	get_unaligned_le16(&report[i + 3]);
545f4e99 PC	298	i += 4;
	299	break;
	300	}
	301	break;
	302
	303	case HID_COLLECTION:
ec67bbed	304	/* reset UsagePage and Finger */
545f4e99 PC	305	finger = usage = 0;
	306	break;
	307	}
	308	}
	309
545f4e99	310	out:
384318ec	311	result = 0;
545f4e99 PC	312	kfree(report);
	313	return result;
	314	}
	315
ec67bbed	316	static int wacom_query_tablet_data(struct usb_interface intf, struct wacom_features features)
3b7307c2 DT	317	{
3b7307c2 DT	318	unsigned char *rep_data;
ec67bbed PC	319	int limit = 0, report_id = 2;
ec67bbed PC	320	int error = -ENOMEM;
3b7307c2 DT	321
	322	rep_data = kmalloc(2, GFP_KERNEL);
	323	if (!rep_data)
ec67bbed PC	324	return error;
ec67bbed PC	325
3dc9f40d CB	326	/* ask to report tablet data if it is 2FGT Tablet PC or
	327	* not a Tablet PC */
	328	if (features->type == TABLETPC2FG) {
ec67bbed PC	329	do {
	330	rep_data[0] = 3;
	331	rep_data[1] = 4;
	332	report_id = 3;
	333	error = usb_set_report(intf, WAC_HID_FEATURE_REPORT,
	334	report_id, rep_data, 2);
	335	if (error >= 0)
	336	error = usb_get_report(intf,
	337	WAC_HID_FEATURE_REPORT, report_id,
	338	rep_data, 3);
	339	} while ((error < 0 \|\| rep_data[1] != 4) && limit++ < 5);
3dc9f40d	340	} else if (features->type != TABLETPC) {
ec67bbed PC	341	do {
	342	rep_data[0] = 2;
	343	rep_data[1] = 2;
	344	error = usb_set_report(intf, WAC_HID_FEATURE_REPORT,
	345	report_id, rep_data, 2);
	346	if (error >= 0)
	347	error = usb_get_report(intf,
	348	WAC_HID_FEATURE_REPORT, report_id,
	349	rep_data, 2);
	350	} while ((error < 0 \|\| rep_data[1] != 2) && limit++ < 5);
	351	}
3b7307c2 DT	352
	353	kfree(rep_data);
	354
	355	return error < 0 ? error : 0;
	356	}
	357
ec67bbed PC	358	static int wacom_retrieve_hid_descriptor(struct usb_interface *intf,
	359	struct wacom_features *features)
	360	{
	361	int error = 0;
	362	struct usb_host_interface *interface = intf->cur_altsetting;
	363	struct hid_descriptor *hid_desc;
	364
fed87e65	365	/* default features */
ec67bbed	366	features->device_type = BTN_TOOL_PEN;
fed87e65 HR	367	features->x_fuzz = 4;
	368	features->y_fuzz = 4;
	369	features->pressure_fuzz = 0;
	370	features->distance_fuzz = 0;
ec67bbed	371
3dc9f40d	372	/* only Tablet PCs and Bamboo P&T need to retrieve the info */
4a88081e PC	373	if ((features->type != TABLETPC) && (features->type != TABLETPC2FG) &&
4a88081e PC	374	(features->type != BAMBOO_PT))
ec67bbed PC	375	goto out;
	376
	377	if (usb_get_extra_descriptor(interface, HID_DEVICET_HID, &hid_desc)) {
	378	if (usb_get_extra_descriptor(&interface->endpoint[0],
	379	HID_DEVICET_REPORT, &hid_desc)) {
	380	printk("wacom: can not retrieve extra class descriptor\n");
	381	error = 1;
	382	goto out;
	383	}
	384	}
	385	error = wacom_parse_hid(intf, hid_desc, features);
	386	if (error)
	387	goto out;
	388
ec67bbed PC	389	out:
	390	return error;
	391	}
	392
4492efff PC	393	struct wacom_usbdev_data {
	394	struct list_head list;
	395	struct kref kref;
	396	struct usb_device *dev;
	397	struct wacom_shared shared;
	398	};
	399
	400	static LIST_HEAD(wacom_udev_list);
	401	static DEFINE_MUTEX(wacom_udev_list_lock);
	402
	403	static struct wacom_usbdev_data wacom_get_usbdev_data(struct usb_device dev)
	404	{
	405	struct wacom_usbdev_data *data;
	406
	407	list_for_each_entry(data, &wacom_udev_list, list) {
	408	if (data->dev == dev) {
	409	kref_get(&data->kref);
	410	return data;
	411	}
	412	}
	413
	414	return NULL;
	415	}
	416
	417	static int wacom_add_shared_data(struct wacom_wac *wacom,
	418	struct usb_device *dev)
	419	{
	420	struct wacom_usbdev_data *data;
	421	int retval = 0;
	422
	423	mutex_lock(&wacom_udev_list_lock);
	424
	425	data = wacom_get_usbdev_data(dev);
	426	if (!data) {
	427	data = kzalloc(sizeof(struct wacom_usbdev_data), GFP_KERNEL);
	428	if (!data) {
	429	retval = -ENOMEM;
	430	goto out;
	431	}
	432
	433	kref_init(&data->kref);
	434	data->dev = dev;
	435	list_add_tail(&data->list, &wacom_udev_list);
	436	}
	437
	438	wacom->shared = &data->shared;
	439
	440	out:
	441	mutex_unlock(&wacom_udev_list_lock);
	442	return retval;
	443	}
	444
	445	static void wacom_release_shared_data(struct kref *kref)
	446	{
	447	struct wacom_usbdev_data *data =
	448	container_of(kref, struct wacom_usbdev_data, kref);
	449
	450	mutex_lock(&wacom_udev_list_lock);
	451	list_del(&data->list);
	452	mutex_unlock(&wacom_udev_list_lock);
	453
	454	kfree(data);
	455	}
	456
457	static void wacom_remove_shared_data(struct wacom_wac *wacom)
458	{
459	struct wacom_usbdev_data *data;
460
461	if (wacom->shared) {
462	data = container_of(wacom->shared, struct wacom_usbdev_data, shared);
463	kref_put(&data->kref, wacom_release_shared_data);
464	wacom->shared = NULL;
465	}
466	}
467
3bea733a PC	468	static int wacom_probe(struct usb_interface intf, const struct usb_device_id id)
	469	{
	470	struct usb_device *dev = interface_to_usbdev(intf);
	471	struct usb_endpoint_descriptor *endpoint;
	472	struct wacom *wacom;
	473	struct wacom_wac *wacom_wac;
e33da8a5	474	struct wacom_features *features;
3bea733a	475	struct input_dev *input_dev;
e33da8a5	476	int error;
3bea733a	477
e33da8a5	478	if (!id->driver_info)
b036f6fb BB	479	return -EINVAL;
b036f6fb BB	480
3bea733a	481	wacom = kzalloc(sizeof(struct wacom), GFP_KERNEL);
3bea733a	482	input_dev = input_allocate_device();
51269fe8	483	if (!wacom \|\| !input_dev) {
e33da8a5 JC	484	error = -ENOMEM;
	485	goto fail1;
	486	}
	487
51269fe8	488	wacom_wac = &wacom->wacom_wac;
e33da8a5 JC	489	wacom_wac->features = ((struct wacom_features )id->driver_info);
	490	features = &wacom_wac->features;
	491	if (features->pktlen > WACOM_PKGLEN_MAX) {
	492	error = -EINVAL;
3bea733a	493	goto fail1;
e33da8a5	494	}
3bea733a	495
997ea58e DM	496	wacom_wac->data = usb_alloc_coherent(dev, WACOM_PKGLEN_MAX,
997ea58e DM	497	GFP_KERNEL, &wacom->data_dma);
e33da8a5 JC	498	if (!wacom_wac->data) {
e33da8a5 JC	499	error = -ENOMEM;
3bea733a	500	goto fail1;
e33da8a5	501	}
3bea733a PC	502
3bea733a PC	503	wacom->irq = usb_alloc_urb(0, GFP_KERNEL);
e33da8a5 JC	504	if (!wacom->irq) {
e33da8a5 JC	505	error = -ENOMEM;
3bea733a	506	goto fail2;
e33da8a5	507	}
3bea733a PC	508
3bea733a PC	509	wacom->usbdev = dev;
e7224094 ON	510	wacom->intf = intf;
e7224094 ON	511	mutex_init(&wacom->lock);
3bea733a PC	512	usb_make_path(dev, wacom->phys, sizeof(wacom->phys));
	513	strlcat(wacom->phys, "/input0", sizeof(wacom->phys));
	514
8da23fc1	515	wacom_wac->input = input_dev;
3bea733a	516
545f4e99 PC	517	endpoint = &intf->cur_altsetting->endpoint[0].desc;
545f4e99 PC	518
ec67bbed PC	519	/* Retrieve the physical and logical size for OEM devices */
	520	error = wacom_retrieve_hid_descriptor(intf, features);
	521	if (error)
4b6d4434	522	goto fail3;
545f4e99	523
bc73dd39 HR	524	wacom_setup_device_quirks(features);
bc73dd39 HR	525
49b764ae PC	526	strlcpy(wacom_wac->name, features->name, sizeof(wacom_wac->name));
49b764ae PC	527
bc73dd39	528	if (features->quirks & WACOM_QUIRK_MULTI_INPUT) {
49b764ae PC	529	/* Append the device type to the name */
	530	strlcat(wacom_wac->name,
	531	features->device_type == BTN_TOOL_PEN ?
	532	" Pen" : " Finger",
	533	sizeof(wacom_wac->name));
4492efff PC	534
	535	error = wacom_add_shared_data(wacom_wac, dev);
	536	if (error)
	537	goto fail3;
49b764ae PC	538	}
49b764ae PC	539
8da23fc1 DT	540	input_dev->name = wacom_wac->name;
	541	input_dev->dev.parent = &intf->dev;
	542	input_dev->open = wacom_open;
	543	input_dev->close = wacom_close;
	544	usb_to_input_id(dev, &input_dev->id);
	545	input_set_drvdata(input_dev, wacom);
e33da8a5	546
8da23fc1	547	wacom_setup_input_capabilities(input_dev, wacom_wac);
3bea733a	548
3bea733a PC	549	usb_fill_int_urb(wacom->irq, dev,
3bea733a PC	550	usb_rcvintpipe(dev, endpoint->bEndpointAddress),
ec67bbed	551	wacom_wac->data, features->pktlen,
8d32e3ae	552	wacom_sys_irq, wacom, endpoint->bInterval);
3bea733a PC	553	wacom->irq->transfer_dma = wacom->data_dma;
	554	wacom->irq->transfer_flags \|= URB_NO_TRANSFER_DMA_MAP;
	555
8da23fc1	556	error = input_register_device(input_dev);
5014186d	557	if (error)
4492efff	558	goto fail4;
3bea733a	559
ec67bbed PC	560	/* Note that if query fails it is not a hard failure */
ec67bbed PC	561	wacom_query_tablet_data(intf, features);
3bea733a PC	562
	563	usb_set_intfdata(intf, wacom);
	564	return 0;
	565
4492efff	566	fail4: wacom_remove_shared_data(wacom_wac);
5014186d	567	fail3: usb_free_urb(wacom->irq);
997ea58e	568	fail2: usb_free_coherent(dev, WACOM_PKGLEN_MAX, wacom_wac->data, wacom->data_dma);
5014186d	569	fail1: input_free_device(input_dev);
3bea733a	570	kfree(wacom);
5014186d	571	return error;
3bea733a PC	572	}
	573
	574	static void wacom_disconnect(struct usb_interface *intf)
	575	{
e7224094	576	struct wacom *wacom = usb_get_intfdata(intf);
3bea733a PC	577
3bea733a PC	578	usb_set_intfdata(intf, NULL);
e7224094 ON	579
e7224094 ON	580	usb_kill_urb(wacom->irq);
8da23fc1	581	input_unregister_device(wacom->wacom_wac.input);
e7224094	582	usb_free_urb(wacom->irq);
997ea58e	583	usb_free_coherent(interface_to_usbdev(intf), WACOM_PKGLEN_MAX,
51269fe8 DT	584	wacom->wacom_wac.data, wacom->data_dma);
51269fe8 DT	585	wacom_remove_shared_data(&wacom->wacom_wac);
e7224094 ON	586	kfree(wacom);
	587	}
	588
	589	static int wacom_suspend(struct usb_interface *intf, pm_message_t message)
	590	{
	591	struct wacom *wacom = usb_get_intfdata(intf);
	592
	593	mutex_lock(&wacom->lock);
	594	usb_kill_urb(wacom->irq);
	595	mutex_unlock(&wacom->lock);
	596
	597	return 0;
	598	}
	599
	600	static int wacom_resume(struct usb_interface *intf)
	601	{
	602	struct wacom *wacom = usb_get_intfdata(intf);
51269fe8	603	struct wacom_features *features = &wacom->wacom_wac.features;
e7224094 ON	604	int rv;
	605
	606	mutex_lock(&wacom->lock);
38101475 PC	607
	608	/* switch to wacom mode first */
	609	wacom_query_tablet_data(intf, features);
	610
	611	if (wacom->open)
e7224094	612	rv = usb_submit_urb(wacom->irq, GFP_NOIO);
38101475	613	else
e7224094	614	rv = 0;
38101475	615
e7224094 ON	616	mutex_unlock(&wacom->lock);
	617
	618	return rv;
	619	}
	620
	621	static int wacom_reset_resume(struct usb_interface *intf)
	622	{
	623	return wacom_resume(intf);
3bea733a PC	624	}
	625
	626	static struct usb_driver wacom_driver = {
	627	.name = "wacom",
b036f6fb	628	.id_table = wacom_ids,
3bea733a PC	629	.probe = wacom_probe,
3bea733a PC	630	.disconnect = wacom_disconnect,
e7224094 ON	631	.suspend = wacom_suspend,
	632	.resume = wacom_resume,
	633	.reset_resume = wacom_reset_resume,
	634	.supports_autosuspend = 1,
3bea733a PC	635	};
	636
	637	static int __init wacom_init(void)
	638	{
	639	int result;
b036f6fb	640
3bea733a PC	641	result = usb_register(&wacom_driver);
3bea733a PC	642	if (result == 0)
899ef6e7 GKH	643	printk(KERN_INFO KBUILD_MODNAME ": " DRIVER_VERSION ":"
899ef6e7 GKH	644	DRIVER_DESC "\n");
3bea733a PC	645	return result;
	646	}
	647
	648	static void __exit wacom_exit(void)
	649	{
	650	usb_deregister(&wacom_driver);
	651	}
	652
	653	module_init(wacom_init);
	654	module_exit(wacom_exit);