[linux-2.6-block.git] / drivers / usb / input / appletouch.c

/*
 * Apple USB Touchpad (for post-February 2005 PowerBooks and MacBooks) driver
 *
 * Copyright (C) 2001-2004 Greg Kroah-Hartman (greg@kroah.com)
 * Copyright (C) 2005      Johannes Berg (johannes@sipsolutions.net)
 * Copyright (C) 2005      Stelian Pop (stelian@popies.net)
 * Copyright (C) 2005      Frank Arnold (frank@scirocco-5v-turbo.de)
 * Copyright (C) 2005      Peter Osterlund (petero2@telia.com)
 * Copyright (C) 2005      Michael Hanselmann (linux-kernel@hansmi.ch)
 * Copyright (C) 2006      Nicolas Boichat (nicolas@boichat.ch)
 *
 * Thanks to Alex Harper <basilisk@foobox.net> for his inputs.
 *
 * 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.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

#include <linux/kernel.h>
#include <linux/errno.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/module.h>
#include <linux/usb/input.h>

/* Apple has powerbooks which have the keyboard with different Product IDs */
#define APPLE_VENDOR_ID		0x05AC

/* These names come from Info.plist in AppleUSBTrackpad.kext */
#define GEYSER_ANSI_PRODUCT_ID	0x0214
#define GEYSER_ISO_PRODUCT_ID	0x0215
#define GEYSER_JIS_PRODUCT_ID	0x0216

/* MacBook devices */
#define GEYSER3_ANSI_PRODUCT_ID	0x0217
#define GEYSER3_ISO_PRODUCT_ID	0x0218
#define GEYSER3_JIS_PRODUCT_ID	0x0219

#define ATP_DEVICE(prod)					\
	.match_flags = USB_DEVICE_ID_MATCH_DEVICE |		\
		       USB_DEVICE_ID_MATCH_INT_CLASS |		\
		       USB_DEVICE_ID_MATCH_INT_PROTOCOL,	\
	.idVendor = APPLE_VENDOR_ID,				\
	.idProduct = (prod),					\
	.bInterfaceClass = 0x03,				\
	.bInterfaceProtocol = 0x02

/* table of devices that work with this driver */
static struct usb_device_id atp_table [] = {
	{ ATP_DEVICE(0x020E) },
	{ ATP_DEVICE(0x020F) },
	{ ATP_DEVICE(0x030A) },
	{ ATP_DEVICE(0x030B) },

	/* PowerBooks Oct 2005 */
	{ ATP_DEVICE(GEYSER_ANSI_PRODUCT_ID) },
	{ ATP_DEVICE(GEYSER_ISO_PRODUCT_ID) },
	{ ATP_DEVICE(GEYSER_JIS_PRODUCT_ID) },

	{ ATP_DEVICE(GEYSER3_ANSI_PRODUCT_ID) },
	{ ATP_DEVICE(GEYSER3_ISO_PRODUCT_ID) },
	{ ATP_DEVICE(GEYSER3_JIS_PRODUCT_ID) },

	/* Terminating entry */
	{ }
};
MODULE_DEVICE_TABLE (usb, atp_table);

/*
 * number of sensors. Note that only 16 instead of 26 X (horizontal)
 * sensors exist on 12" and 15" PowerBooks. All models have 16 Y
 * (vertical) sensors.
 */
#define ATP_XSENSORS	26
#define ATP_YSENSORS	16

/* amount of fuzz this touchpad generates */
#define ATP_FUZZ	16

/* maximum pressure this driver will report */
#define ATP_PRESSURE	300
/*
 * multiplication factor for the X and Y coordinates.
 * We try to keep the touchpad aspect ratio while still doing only simple
 * arithmetics.
 * The factors below give coordinates like:
 *	0 <= x <  960 on 12" and 15" Powerbooks
 *	0 <= x < 1600 on 17" Powerbooks
 *	0 <= y <  646
 */
#define ATP_XFACT	64
#define ATP_YFACT	43

/*
 * Threshold for the touchpad sensors. Any change less than ATP_THRESHOLD is
 * ignored.
 */
#define ATP_THRESHOLD	 5

/* MacBook Pro (Geyser 3) initialization constants */
#define ATP_GEYSER3_MODE_READ_REQUEST_ID 1
#define ATP_GEYSER3_MODE_WRITE_REQUEST_ID 9
#define ATP_GEYSER3_MODE_REQUEST_VALUE 0x300
#define ATP_GEYSER3_MODE_REQUEST_INDEX 0
#define ATP_GEYSER3_MODE_VENDOR_VALUE 0x04

/* Structure to hold all of our device specific stuff */
struct atp {
	char			phys[64];
	struct usb_device *	udev;		/* usb device */
	struct urb *		urb;		/* usb request block */
	signed char *		data;		/* transferred data */
	int			open;		/* non-zero if opened */
	struct input_dev	*input;		/* input dev */
	int			valid;		/* are the sensors valid ? */
	int			x_old;		/* last reported x/y, */
	int			y_old;		/* used for smoothing */
						/* current value of the sensors */
	signed char		xy_cur[ATP_XSENSORS + ATP_YSENSORS];
						/* last value of the sensors */
	signed char		xy_old[ATP_XSENSORS + ATP_YSENSORS];
						/* accumulated sensors */
	int			xy_acc[ATP_XSENSORS + ATP_YSENSORS];
	int			overflowwarn;	/* overflow warning printed? */
	int			datalen;	/* size of an USB urb transfer */
};

#define dbg_dump(msg, tab) \
	if (debug > 1) {						\
		int i;							\
		printk("appletouch: %s %lld", msg, (long long)jiffies); \
		for (i = 0; i < ATP_XSENSORS + ATP_YSENSORS; i++)	\
			printk(" %02x", tab[i]);			\
		printk("\n");						\
	}

#define dprintk(format, a...)						\
	do {								\
		if (debug) printk(format, ##a);				\
	} while (0)

MODULE_AUTHOR("Johannes Berg, Stelian Pop, Frank Arnold, Michael Hanselmann");
MODULE_DESCRIPTION("Apple PowerBooks USB touchpad driver");
MODULE_LICENSE("GPL");

static int debug = 1;
module_param(debug, int, 0644);
MODULE_PARM_DESC(debug, "Activate debugging output");

/* Checks if the device a Geyser 2 (ANSI, ISO, JIS) */
static inline int atp_is_geyser_2(struct atp *dev)
{
	u16 productId = le16_to_cpu(dev->udev->descriptor.idProduct);

	return (productId == GEYSER_ANSI_PRODUCT_ID) ||
		(productId == GEYSER_ISO_PRODUCT_ID) ||
		(productId == GEYSER_JIS_PRODUCT_ID);
}

static inline int atp_is_geyser_3(struct atp *dev)
{
	u16 productId = le16_to_cpu(dev->udev->descriptor.idProduct);

	return (productId == GEYSER3_ANSI_PRODUCT_ID) ||
		(productId == GEYSER3_ISO_PRODUCT_ID) ||
		(productId == GEYSER3_JIS_PRODUCT_ID);
}

static int atp_calculate_abs(int *xy_sensors, int nb_sensors, int fact,
			     int *z, int *fingers)
{
	int i;
	/* values to calculate mean */
	int pcum = 0, psum = 0;

	*fingers = 0;

	for (i = 0; i < nb_sensors; i++) {
		if (xy_sensors[i] < ATP_THRESHOLD)
			continue;
		if ((i - 1 < 0) || (xy_sensors[i - 1] < ATP_THRESHOLD))
			(*fingers)++;
		pcum += xy_sensors[i] * i;
		psum += xy_sensors[i];
	}

	if (psum > 0) {
		*z = psum;
		return pcum * fact / psum;
	}

	return 0;
}

static inline void atp_report_fingers(struct input_dev *input, int fingers)
{
	input_report_key(input, BTN_TOOL_FINGER, fingers == 1);
	input_report_key(input, BTN_TOOL_DOUBLETAP, fingers == 2);
	input_report_key(input, BTN_TOOL_TRIPLETAP, fingers > 2);
}

static void atp_complete(struct urb* urb)
{
	int x, y, x_z, y_z, x_f, y_f;
	int retval, i, j;
	struct atp *dev = urb->context;

	switch (urb->status) {
	case 0:
		/* success */
		break;
	case -EOVERFLOW:
		if(!dev->overflowwarn) {
			printk("appletouch: OVERFLOW with data "
				"length %d, actual length is %d\n",
				dev->datalen, dev->urb->actual_length);
			dev->overflowwarn = 1;
		}
	case -ECONNRESET:
	case -ENOENT:
	case -ESHUTDOWN:
		/* This urb is terminated, clean up */
		dbg("%s - urb shutting down with status: %d",
		    __FUNCTION__, urb->status);
		return;
	default:
		dbg("%s - nonzero urb status received: %d",
		    __FUNCTION__, urb->status);
		goto exit;
	}

	/* drop incomplete datasets */
	if (dev->urb->actual_length != dev->datalen) {
		dprintk("appletouch: incomplete data package"
			" (first byte: %d, length: %d).\n",
			dev->data[0], dev->urb->actual_length);
		goto exit;
	}

	/* reorder the sensors values */
	if (atp_is_geyser_3(dev)) {
		memset(dev->xy_cur, 0, sizeof(dev->xy_cur));

		/*
		 * The values are laid out like this:
		 * -, Y1, Y2, -, Y3, Y4, -, ..., -, X1, X2, -, X3, X4, ...
		 * '-' is an unused value.
		 */

		/* read X values */
		for (i = 0, j = 19; i < 20; i += 2, j += 3) {
			dev->xy_cur[i] = dev->data[j + 1];
			dev->xy_cur[i + 1] = dev->data[j + 2];
		}
		/* read Y values */
		for (i = 0, j = 1; i < 9; i += 2, j += 3) {
			dev->xy_cur[ATP_XSENSORS + i] = dev->data[j + 1];
			dev->xy_cur[ATP_XSENSORS + i + 1] = dev->data[j + 2];
		}
	} else if (atp_is_geyser_2(dev)) {
		memset(dev->xy_cur, 0, sizeof(dev->xy_cur));

		/*
		 * The values are laid out like this:
		 * Y1, Y2, -, Y3, Y4, -, ..., X1, X2, -, X3, X4, -, ...
		 * '-' is an unused value.
		 */

		/* read X values */
		for (i = 0, j = 19; i < 20; i += 2, j += 3) {
			dev->xy_cur[i] = dev->data[j];
			dev->xy_cur[i + 1] = dev->data[j + 1];
		}

		/* read Y values */
		for (i = 0, j = 1; i < 9; i += 2, j += 3) {
			dev->xy_cur[ATP_XSENSORS + i] = dev->data[j];
			dev->xy_cur[ATP_XSENSORS + i + 1] = dev->data[j + 1];
		}
	} else {
		for (i = 0; i < 8; i++) {
			/* X values */
			dev->xy_cur[i     ] = dev->data[5 * i +  2];
			dev->xy_cur[i +  8] = dev->data[5 * i +  4];
			dev->xy_cur[i + 16] = dev->data[5 * i + 42];
			if (i < 2)
				dev->xy_cur[i + 24] = dev->data[5 * i + 44];

			/* Y values */
			dev->xy_cur[i + 26] = dev->data[5 * i +  1];
			dev->xy_cur[i + 34] = dev->data[5 * i +  3];
		}
	}

	dbg_dump("sample", dev->xy_cur);

	if (!dev->valid) {
		/* first sample */
		dev->valid = 1;
		dev->x_old = dev->y_old = -1;
		memcpy(dev->xy_old, dev->xy_cur, sizeof(dev->xy_old));

		if (atp_is_geyser_3(dev)) /* No 17" Macbooks (yet) */
			goto exit;

		/* 17" Powerbooks have extra X sensors */
		for (i = (atp_is_geyser_2(dev)?15:16); i < ATP_XSENSORS; i++) {
			if (!dev->xy_cur[i]) continue;

			printk("appletouch: 17\" model detected.\n");
			if(atp_is_geyser_2(dev))
				input_set_abs_params(dev->input, ABS_X, 0,
						     (20 - 1) *
						     ATP_XFACT - 1,
						     ATP_FUZZ, 0);
			else
				input_set_abs_params(dev->input, ABS_X, 0,
						     (ATP_XSENSORS - 1) *
						     ATP_XFACT - 1,
						     ATP_FUZZ, 0);

			break;
		}

		goto exit;
	}

	for (i = 0; i < ATP_XSENSORS + ATP_YSENSORS; i++) {
		/* accumulate the change */
		signed char change = dev->xy_old[i] - dev->xy_cur[i];
		dev->xy_acc[i] -= change;

		/* prevent down drifting */
		if (dev->xy_acc[i] < 0)
			dev->xy_acc[i] = 0;
	}

	memcpy(dev->xy_old, dev->xy_cur, sizeof(dev->xy_old));

	dbg_dump("accumulator", dev->xy_acc);

	x = atp_calculate_abs(dev->xy_acc, ATP_XSENSORS,
			      ATP_XFACT, &x_z, &x_f);
	y = atp_calculate_abs(dev->xy_acc + ATP_XSENSORS, ATP_YSENSORS,
			      ATP_YFACT, &y_z, &y_f);

	if (x && y) {
		if (dev->x_old != -1) {
			x = (dev->x_old * 3 + x) >> 2;
			y = (dev->y_old * 3 + y) >> 2;
			dev->x_old = x;
			dev->y_old = y;

			if (debug > 1)
				printk("appletouch: X: %3d Y: %3d "
				       "Xz: %3d Yz: %3d\n",
				       x, y, x_z, y_z);

			input_report_key(dev->input, BTN_TOUCH, 1);
			input_report_abs(dev->input, ABS_X, x);
			input_report_abs(dev->input, ABS_Y, y);
			input_report_abs(dev->input, ABS_PRESSURE,
					 min(ATP_PRESSURE, x_z + y_z));
			atp_report_fingers(dev->input, max(x_f, y_f));
		}
		dev->x_old = x;
		dev->y_old = y;
	}
	else if (!x && !y) {

		dev->x_old = dev->y_old = -1;
		input_report_key(dev->input, BTN_TOUCH, 0);
		input_report_abs(dev->input, ABS_PRESSURE, 0);
		atp_report_fingers(dev->input, 0);

		/* reset the accumulator on release */
		memset(dev->xy_acc, 0, sizeof(dev->xy_acc));
	}

	input_report_key(dev->input, BTN_LEFT,
			 !!dev->data[dev->datalen - 1]);

	input_sync(dev->input);

exit:
	retval = usb_submit_urb(dev->urb, GFP_ATOMIC);
	if (retval) {
		err("%s - usb_submit_urb failed with result %d",
		    __FUNCTION__, retval);
	}
}

static int atp_open(struct input_dev *input)
{
	struct atp *dev = input->private;

	if (usb_submit_urb(dev->urb, GFP_ATOMIC))
		return -EIO;

	dev->open = 1;
	return 0;
}

static void atp_close(struct input_dev *input)
{
	struct atp *dev = input->private;

	usb_kill_urb(dev->urb);
	dev->open = 0;
}

static int atp_probe(struct usb_interface *iface, const struct usb_device_id *id)
{
	struct atp *dev;
	struct input_dev *input_dev;
	struct usb_device *udev = interface_to_usbdev(iface);
	struct usb_host_interface *iface_desc;
	struct usb_endpoint_descriptor *endpoint;
	int int_in_endpointAddr = 0;
	int i, retval = -ENOMEM;


	/* set up the endpoint information */
	/* use only the first interrupt-in endpoint */
	iface_desc = iface->cur_altsetting;
	for (i = 0; i < iface_desc->desc.bNumEndpoints; i++) {
		endpoint = &iface_desc->endpoint[i].desc;
		if (!int_in_endpointAddr && usb_endpoint_is_int_in(endpoint)) {
			/* we found an interrupt in endpoint */
			int_in_endpointAddr = endpoint->bEndpointAddress;
			break;
		}
	}
	if (!int_in_endpointAddr) {
		err("Could not find int-in endpoint");
		return -EIO;
	}

	/* allocate memory for our device state and initialize it */
	dev = kzalloc(sizeof(struct atp), GFP_KERNEL);
	input_dev = input_allocate_device();
	if (!dev || !input_dev) {
		err("Out of memory");
		goto err_free_devs;
	}

	dev->udev = udev;
	dev->input = input_dev;
	dev->overflowwarn = 0;
	if (atp_is_geyser_3(dev))
		dev->datalen = 64;
	else if (atp_is_geyser_2(dev))
		dev->datalen = 64;
	else
		dev->datalen = 81;

	if (atp_is_geyser_3(dev)) {
		/*
		 * By default Geyser 3 device sends standard USB HID mouse
		 * packets (Report ID 2). This code changes device mode, so it
		 * sends raw sensor reports (Report ID 5).
		 */
		char data[8];
		int size;

		size = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0),
			ATP_GEYSER3_MODE_READ_REQUEST_ID,
			USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
			ATP_GEYSER3_MODE_REQUEST_VALUE,
			ATP_GEYSER3_MODE_REQUEST_INDEX, &data, 8, 5000);

		if (size != 8) {
			err("Could not do mode read request from device"
							" (Geyser 3 mode)");
			goto err_free_devs;
		}

		/* Apply the mode switch */
		data[0] = ATP_GEYSER3_MODE_VENDOR_VALUE;

		size = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
			ATP_GEYSER3_MODE_WRITE_REQUEST_ID,
			USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
			ATP_GEYSER3_MODE_REQUEST_VALUE,
			ATP_GEYSER3_MODE_REQUEST_INDEX, &data, 8, 5000);

		if (size != 8) {
			err("Could not do mode write request to device"
							" (Geyser 3 mode)");
			goto err_free_devs;
		}
		printk("appletouch Geyser 3 inited.\n");
	}

	dev->urb = usb_alloc_urb(0, GFP_KERNEL);
	if (!dev->urb) {
		retval = -ENOMEM;
		goto err_free_devs;
	}

	dev->data = usb_buffer_alloc(dev->udev, dev->datalen, GFP_KERNEL,
				     &dev->urb->transfer_dma);
	if (!dev->data) {
		retval = -ENOMEM;
		goto err_free_urb;
	}

	usb_fill_int_urb(dev->urb, udev,
			 usb_rcvintpipe(udev, int_in_endpointAddr),
			 dev->data, dev->datalen, atp_complete, dev, 1);

	usb_make_path(udev, dev->phys, sizeof(dev->phys));
	strlcat(dev->phys, "/input0", sizeof(dev->phys));

	input_dev->name = "appletouch";
	input_dev->phys = dev->phys;
	usb_to_input_id(dev->udev, &input_dev->id);
	input_dev->cdev.dev = &iface->dev;

	input_dev->private = dev;
	input_dev->open = atp_open;
	input_dev->close = atp_close;

	set_bit(EV_ABS, input_dev->evbit);

	if (atp_is_geyser_3(dev)) {
		/*
		 * MacBook have 20 X sensors, 10 Y sensors
		 */
		input_set_abs_params(input_dev, ABS_X, 0,
				     ((20 - 1) * ATP_XFACT) - 1, ATP_FUZZ, 0);
		input_set_abs_params(input_dev, ABS_Y, 0,
				     ((10 - 1) * ATP_YFACT) - 1, ATP_FUZZ, 0);
	} else if (atp_is_geyser_2(dev)) {
		/*
		 * Oct 2005 15" PowerBooks have 15 X sensors, 17" are detected
		 * later.
		 */
		input_set_abs_params(input_dev, ABS_X, 0,
				     ((15 - 1) * ATP_XFACT) - 1, ATP_FUZZ, 0);
		input_set_abs_params(input_dev, ABS_Y, 0,
				     ((9 - 1) * ATP_YFACT) - 1, ATP_FUZZ, 0);
	} else {
		/*
		 * 12" and 15" Powerbooks only have 16 x sensors,
		 * 17" models are detected later.
		 */
		input_set_abs_params(input_dev, ABS_X, 0,
				     (16 - 1) * ATP_XFACT - 1, ATP_FUZZ, 0);
		input_set_abs_params(input_dev, ABS_Y, 0,
				     (ATP_YSENSORS - 1) * ATP_YFACT - 1, ATP_FUZZ, 0);
	}
	input_set_abs_params(input_dev, ABS_PRESSURE, 0, ATP_PRESSURE, 0, 0);

	set_bit(EV_KEY, input_dev->evbit);
	set_bit(BTN_TOUCH, input_dev->keybit);
	set_bit(BTN_TOOL_FINGER, input_dev->keybit);
	set_bit(BTN_TOOL_DOUBLETAP, input_dev->keybit);
	set_bit(BTN_TOOL_TRIPLETAP, input_dev->keybit);
	set_bit(BTN_LEFT, input_dev->keybit);

	input_register_device(dev->input);

	/* save our data pointer in this interface device */
	usb_set_intfdata(iface, dev);

	return 0;

 err_free_urb:
	usb_free_urb(dev->urb);
 err_free_devs:
	usb_set_intfdata(iface, NULL);
	kfree(dev);
	input_free_device(input_dev);
	return retval;
}

static void atp_disconnect(struct usb_interface *iface)
{
	struct atp *dev = usb_get_intfdata(iface);

	usb_set_intfdata(iface, NULL);
	if (dev) {
		usb_kill_urb(dev->urb);
		input_unregister_device(dev->input);
		usb_buffer_free(dev->udev, dev->datalen,
				dev->data, dev->urb->transfer_dma);
		usb_free_urb(dev->urb);
		kfree(dev);
	}
	printk(KERN_INFO "input: appletouch disconnected\n");
}

static int atp_suspend(struct usb_interface *iface, pm_message_t message)
{
	struct atp *dev = usb_get_intfdata(iface);
	usb_kill_urb(dev->urb);
	dev->valid = 0;
	return 0;
}

static int atp_resume(struct usb_interface *iface)
{
	struct atp *dev = usb_get_intfdata(iface);
	if (dev->open && usb_submit_urb(dev->urb, GFP_ATOMIC))
		return -EIO;

	return 0;
}

static struct usb_driver atp_driver = {
	.name		= "appletouch",
	.probe		= atp_probe,
	.disconnect	= atp_disconnect,
	.suspend	= atp_suspend,
	.resume		= atp_resume,
	.id_table	= atp_table,
};

static int __init atp_init(void)
{
	return usb_register(&atp_driver);
}

static void __exit atp_exit(void)
{
	usb_deregister(&atp_driver);
}

module_init(atp_init);
module_exit(atp_exit);
Commit	Line	Data
f7214ff4	1	/*
9effa978	2	* Apple USB Touchpad (for post-February 2005 PowerBooks and MacBooks) driver
f7214ff4 SP	3	*
	4	* Copyright (C) 2001-2004 Greg Kroah-Hartman (greg@kroah.com)
	5	* Copyright (C) 2005 Johannes Berg (johannes@sipsolutions.net)
	6	* Copyright (C) 2005 Stelian Pop (stelian@popies.net)
	7	* Copyright (C) 2005 Frank Arnold (frank@scirocco-5v-turbo.de)
	8	* Copyright (C) 2005 Peter Osterlund (petero2@telia.com)
e1e02c9f	9	* Copyright (C) 2005 Michael Hanselmann (linux-kernel@hansmi.ch)
9effa978	10	* Copyright (C) 2006 Nicolas Boichat (nicolas@boichat.ch)
f7214ff4 SP	11	*
	12	* Thanks to Alex Harper <basilisk@foobox.net> for his inputs.
	13	*
	14	* This program is free software; you can redistribute it and/or modify
	15	* it under the terms of the GNU General Public License as published by
	16	* the Free Software Foundation; either version 2 of the License, or
	17	* (at your option) any later version.
	18	*
	19	* This program is distributed in the hope that it will be useful,
	20	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	21	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	22	* GNU General Public License for more details.
	23	*
	24	* You should have received a copy of the GNU General Public License
	25	* along with this program; if not, write to the Free Software
	26	* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
	27	*
	28	*/
	29
f7214ff4 SP	30	#include <linux/kernel.h>
	31	#include <linux/errno.h>
	32	#include <linux/init.h>
	33	#include <linux/slab.h>
	34	#include <linux/module.h>
ae0dadcf	35	#include <linux/usb/input.h>
f7214ff4 SP	36
	37	/* Apple has powerbooks which have the keyboard with different Product IDs */
	38	#define APPLE_VENDOR_ID 0x05AC
	39
e1e02c9f MH	40	/* These names come from Info.plist in AppleUSBTrackpad.kext */
	41	#define GEYSER_ANSI_PRODUCT_ID 0x0214
	42	#define GEYSER_ISO_PRODUCT_ID 0x0215
	43	#define GEYSER_JIS_PRODUCT_ID 0x0216
	44
9effa978 NB	45	/* MacBook devices */
	46	#define GEYSER3_ANSI_PRODUCT_ID 0x0217
	47	#define GEYSER3_ISO_PRODUCT_ID 0x0218
	48	#define GEYSER3_JIS_PRODUCT_ID 0x0219
	49
f7214ff4	50	#define ATP_DEVICE(prod) \
c5b7c7c3	51	.match_flags = USB_DEVICE_ID_MATCH_DEVICE \| \
f7214ff4 SP	52	USB_DEVICE_ID_MATCH_INT_CLASS \| \
	53	USB_DEVICE_ID_MATCH_INT_PROTOCOL, \
	54	.idVendor = APPLE_VENDOR_ID, \
	55	.idProduct = (prod), \
	56	.bInterfaceClass = 0x03, \
	57	.bInterfaceProtocol = 0x02
	58
	59	/* table of devices that work with this driver */
	60	static struct usb_device_id atp_table [] = {
	61	{ ATP_DEVICE(0x020E) },
	62	{ ATP_DEVICE(0x020F) },
	63	{ ATP_DEVICE(0x030A) },
	64	{ ATP_DEVICE(0x030B) },
e1e02c9f MH	65
	66	/* PowerBooks Oct 2005 */
	67	{ ATP_DEVICE(GEYSER_ANSI_PRODUCT_ID) },
	68	{ ATP_DEVICE(GEYSER_ISO_PRODUCT_ID) },
	69	{ ATP_DEVICE(GEYSER_JIS_PRODUCT_ID) },
	70
9effa978 NB	71	{ ATP_DEVICE(GEYSER3_ANSI_PRODUCT_ID) },
	72	{ ATP_DEVICE(GEYSER3_ISO_PRODUCT_ID) },
	73	{ ATP_DEVICE(GEYSER3_JIS_PRODUCT_ID) },
	74
e1e02c9f MH	75	/* Terminating entry */
e1e02c9f MH	76	{ }
f7214ff4 SP	77	};
	78	MODULE_DEVICE_TABLE (usb, atp_table);
	79
f7214ff4 SP	80	/*
	81	* number of sensors. Note that only 16 instead of 26 X (horizontal)
	82	* sensors exist on 12" and 15" PowerBooks. All models have 16 Y
	83	* (vertical) sensors.
	84	*/
	85	#define ATP_XSENSORS 26
	86	#define ATP_YSENSORS 16
	87
	88	/* amount of fuzz this touchpad generates */
	89	#define ATP_FUZZ 16
	90
	91	/* maximum pressure this driver will report */
	92	#define ATP_PRESSURE 300
	93	/*
	94	* multiplication factor for the X and Y coordinates.
	95	* We try to keep the touchpad aspect ratio while still doing only simple
	96	* arithmetics.
	97	* The factors below give coordinates like:
c5b7c7c3 DT	98	* 0 <= x < 960 on 12" and 15" Powerbooks
	99	* 0 <= x < 1600 on 17" Powerbooks
	100	* 0 <= y < 646
f7214ff4 SP	101	*/
	102	#define ATP_XFACT 64
	103	#define ATP_YFACT 43
	104
	105	/*
	106	* Threshold for the touchpad sensors. Any change less than ATP_THRESHOLD is
	107	* ignored.
	108	*/
	109	#define ATP_THRESHOLD 5
	110
9effa978 NB	111	/* MacBook Pro (Geyser 3) initialization constants */
	112	#define ATP_GEYSER3_MODE_READ_REQUEST_ID 1
	113	#define ATP_GEYSER3_MODE_WRITE_REQUEST_ID 9
	114	#define ATP_GEYSER3_MODE_REQUEST_VALUE 0x300
	115	#define ATP_GEYSER3_MODE_REQUEST_INDEX 0
	116	#define ATP_GEYSER3_MODE_VENDOR_VALUE 0x04
	117
f7214ff4 SP	118	/* Structure to hold all of our device specific stuff */
f7214ff4 SP	119	struct atp {
c5b7c7c3	120	char phys[64];
f7214ff4 SP	121	struct usb_device * udev; /* usb device */
	122	struct urb * urb; /* usb request block */
	123	signed char * data; /* transferred data */
	124	int open; /* non-zero if opened */
c5b7c7c3	125	struct input_dev input; / input dev */
f7214ff4 SP	126	int valid; /* are the sensors valid ? */
	127	int x_old; /* last reported x/y, */
	128	int y_old; /* used for smoothing */
	129	/* current value of the sensors */
	130	signed char xy_cur[ATP_XSENSORS + ATP_YSENSORS];
	131	/* last value of the sensors */
	132	signed char xy_old[ATP_XSENSORS + ATP_YSENSORS];
	133	/* accumulated sensors */
	134	int xy_acc[ATP_XSENSORS + ATP_YSENSORS];
e1e02c9f MH	135	int overflowwarn; /* overflow warning printed? */
e1e02c9f MH	136	int datalen; /* size of an USB urb transfer */
f7214ff4 SP	137	};
	138
	139	#define dbg_dump(msg, tab) \
	140	if (debug > 1) { \
	141	int i; \
	142	printk("appletouch: %s %lld", msg, (long long)jiffies); \
	143	for (i = 0; i < ATP_XSENSORS + ATP_YSENSORS; i++) \
c5b7c7c3 DT	144	printk(" %02x", tab[i]); \
c5b7c7c3 DT	145	printk("\n"); \
f7214ff4 SP	146	}
f7214ff4 SP	147
c5b7c7c3	148	#define dprintk(format, a...) \
f7214ff4 SP	149	do { \
	150	if (debug) printk(format, ##a); \
	151	} while (0)
	152
e1e02c9f	153	MODULE_AUTHOR("Johannes Berg, Stelian Pop, Frank Arnold, Michael Hanselmann");
f7214ff4 SP	154	MODULE_DESCRIPTION("Apple PowerBooks USB touchpad driver");
	155	MODULE_LICENSE("GPL");
	156
	157	static int debug = 1;
	158	module_param(debug, int, 0644);
	159	MODULE_PARM_DESC(debug, "Activate debugging output");
	160
e1e02c9f MH	161	/* Checks if the device a Geyser 2 (ANSI, ISO, JIS) */
	162	static inline int atp_is_geyser_2(struct atp *dev)
	163	{
9effa978	164	u16 productId = le16_to_cpu(dev->udev->descriptor.idProduct);
e1e02c9f MH	165
	166	return (productId == GEYSER_ANSI_PRODUCT_ID) \|\|
	167	(productId == GEYSER_ISO_PRODUCT_ID) \|\|
	168	(productId == GEYSER_JIS_PRODUCT_ID);
	169	}
	170
9effa978 NB	171	static inline int atp_is_geyser_3(struct atp *dev)
	172	{
	173	u16 productId = le16_to_cpu(dev->udev->descriptor.idProduct);
	174
	175	return (productId == GEYSER3_ANSI_PRODUCT_ID) \|\|
	176	(productId == GEYSER3_ISO_PRODUCT_ID) \|\|
	177	(productId == GEYSER3_JIS_PRODUCT_ID);
	178	}
	179
f7214ff4 SP	180	static int atp_calculate_abs(int *xy_sensors, int nb_sensors, int fact,
	181	int z, int fingers)
	182	{
	183	int i;
	184	/* values to calculate mean */
	185	int pcum = 0, psum = 0;
	186
	187	*fingers = 0;
	188
	189	for (i = 0; i < nb_sensors; i++) {
	190	if (xy_sensors[i] < ATP_THRESHOLD)
	191	continue;
	192	if ((i - 1 < 0) \|\| (xy_sensors[i - 1] < ATP_THRESHOLD))
	193	(*fingers)++;
	194	pcum += xy_sensors[i] * i;
	195	psum += xy_sensors[i];
	196	}
	197
	198	if (psum > 0) {
	199	*z = psum;
	200	return pcum * fact / psum;
	201	}
	202
	203	return 0;
	204	}
	205
	206	static inline void atp_report_fingers(struct input_dev *input, int fingers)
	207	{
	208	input_report_key(input, BTN_TOOL_FINGER, fingers == 1);
	209	input_report_key(input, BTN_TOOL_DOUBLETAP, fingers == 2);
	210	input_report_key(input, BTN_TOOL_TRIPLETAP, fingers > 2);
	211	}
	212
7d12e780	213	static void atp_complete(struct urb* urb)
f7214ff4 SP	214	{
f7214ff4 SP	215	int x, y, x_z, y_z, x_f, y_f;
e1e02c9f	216	int retval, i, j;
f7214ff4 SP	217	struct atp *dev = urb->context;
	218
	219	switch (urb->status) {
	220	case 0:
	221	/* success */
	222	break;
e1e02c9f MH	223	case -EOVERFLOW:
	224	if(!dev->overflowwarn) {
	225	printk("appletouch: OVERFLOW with data "
	226	"length %d, actual length is %d\n",
	227	dev->datalen, dev->urb->actual_length);
	228	dev->overflowwarn = 1;
	229	}
f7214ff4 SP	230	case -ECONNRESET:
	231	case -ENOENT:
	232	case -ESHUTDOWN:
	233	/* This urb is terminated, clean up */
	234	dbg("%s - urb shutting down with status: %d",
	235	__FUNCTION__, urb->status);
	236	return;
	237	default:
	238	dbg("%s - nonzero urb status received: %d",
	239	__FUNCTION__, urb->status);
	240	goto exit;
	241	}
	242
	243	/* drop incomplete datasets */
e1e02c9f	244	if (dev->urb->actual_length != dev->datalen) {
9effa978 NB	245	dprintk("appletouch: incomplete data package"
	246	" (first byte: %d, length: %d).\n",
	247	dev->data[0], dev->urb->actual_length);
f7214ff4 SP	248	goto exit;
	249	}
	250
	251	/* reorder the sensors values */
9effa978 NB	252	if (atp_is_geyser_3(dev)) {
	253	memset(dev->xy_cur, 0, sizeof(dev->xy_cur));
	254
	255	/*
	256	* The values are laid out like this:
	257	* -, Y1, Y2, -, Y3, Y4, -, ..., -, X1, X2, -, X3, X4, ...
	258	* '-' is an unused value.
	259	*/
	260
	261	/* read X values */
	262	for (i = 0, j = 19; i < 20; i += 2, j += 3) {
	263	dev->xy_cur[i] = dev->data[j + 1];
	264	dev->xy_cur[i + 1] = dev->data[j + 2];
	265	}
	266	/* read Y values */
	267	for (i = 0, j = 1; i < 9; i += 2, j += 3) {
	268	dev->xy_cur[ATP_XSENSORS + i] = dev->data[j + 1];
	269	dev->xy_cur[ATP_XSENSORS + i + 1] = dev->data[j + 2];
	270	}
	271	} else if (atp_is_geyser_2(dev)) {
e1e02c9f MH	272	memset(dev->xy_cur, 0, sizeof(dev->xy_cur));
	273
	274	/*
	275	* The values are laid out like this:
	276	* Y1, Y2, -, Y3, Y4, -, ..., X1, X2, -, X3, X4, -, ...
	277	* '-' is an unused value.
	278	*/
	279
	280	/* read X values */
	281	for (i = 0, j = 19; i < 20; i += 2, j += 3) {
	282	dev->xy_cur[i] = dev->data[j];
	283	dev->xy_cur[i + 1] = dev->data[j + 1];
	284	}
	285
	286	/* read Y values */
	287	for (i = 0, j = 1; i < 9; i += 2, j += 3) {
	288	dev->xy_cur[ATP_XSENSORS + i] = dev->data[j];
	289	dev->xy_cur[ATP_XSENSORS + i + 1] = dev->data[j + 1];
	290	}
	291	} else {
	292	for (i = 0; i < 8; i++) {
	293	/* X values */
	294	dev->xy_cur[i ] = dev->data[5 * i + 2];
	295	dev->xy_cur[i + 8] = dev->data[5 * i + 4];
	296	dev->xy_cur[i + 16] = dev->data[5 * i + 42];
	297	if (i < 2)
	298	dev->xy_cur[i + 24] = dev->data[5 * i + 44];
	299
	300	/* Y values */
	301	dev->xy_cur[i + 26] = dev->data[5 * i + 1];
	302	dev->xy_cur[i + 34] = dev->data[5 * i + 3];
	303	}
f7214ff4 SP	304	}
	305
	306	dbg_dump("sample", dev->xy_cur);
	307
	308	if (!dev->valid) {
	309	/* first sample */
	310	dev->valid = 1;
	311	dev->x_old = dev->y_old = -1;
	312	memcpy(dev->xy_old, dev->xy_cur, sizeof(dev->xy_old));
	313
9effa978 NB	314	if (atp_is_geyser_3(dev)) /* No 17" Macbooks (yet) */
	315	goto exit;
	316
e1e02c9f MH	317	/* 17" Powerbooks have extra X sensors */
	318	for (i = (atp_is_geyser_2(dev)?15:16); i < ATP_XSENSORS; i++) {
	319	if (!dev->xy_cur[i]) continue;
	320
	321	printk("appletouch: 17\" model detected.\n");
	322	if(atp_is_geyser_2(dev))
	323	input_set_abs_params(dev->input, ABS_X, 0,
	324	(20 - 1) *
	325	ATP_XFACT - 1,
	326	ATP_FUZZ, 0);
	327	else
c5b7c7c3 DT	328	input_set_abs_params(dev->input, ABS_X, 0,
c5b7c7c3 DT	329	(ATP_XSENSORS - 1) *
f7214ff4 SP	330	ATP_XFACT - 1,
f7214ff4 SP	331	ATP_FUZZ, 0);
e1e02c9f MH	332
	333	break;
	334	}
f7214ff4 SP	335
	336	goto exit;
	337	}
	338
	339	for (i = 0; i < ATP_XSENSORS + ATP_YSENSORS; i++) {
	340	/* accumulate the change */
	341	signed char change = dev->xy_old[i] - dev->xy_cur[i];
	342	dev->xy_acc[i] -= change;
	343
	344	/* prevent down drifting */
	345	if (dev->xy_acc[i] < 0)
	346	dev->xy_acc[i] = 0;
	347	}
	348
	349	memcpy(dev->xy_old, dev->xy_cur, sizeof(dev->xy_old));
	350
	351	dbg_dump("accumulator", dev->xy_acc);
	352
	353	x = atp_calculate_abs(dev->xy_acc, ATP_XSENSORS,
	354	ATP_XFACT, &x_z, &x_f);
	355	y = atp_calculate_abs(dev->xy_acc + ATP_XSENSORS, ATP_YSENSORS,
	356	ATP_YFACT, &y_z, &y_f);
	357
	358	if (x && y) {
	359	if (dev->x_old != -1) {
	360	x = (dev->x_old * 3 + x) >> 2;
	361	y = (dev->y_old * 3 + y) >> 2;
	362	dev->x_old = x;
	363	dev->y_old = y;
	364
	365	if (debug > 1)
	366	printk("appletouch: X: %3d Y: %3d "
	367	"Xz: %3d Yz: %3d\n",
	368	x, y, x_z, y_z);
	369
c5b7c7c3 DT	370	input_report_key(dev->input, BTN_TOUCH, 1);
	371	input_report_abs(dev->input, ABS_X, x);
	372	input_report_abs(dev->input, ABS_Y, y);
	373	input_report_abs(dev->input, ABS_PRESSURE,
f7214ff4	374	min(ATP_PRESSURE, x_z + y_z));
c5b7c7c3	375	atp_report_fingers(dev->input, max(x_f, y_f));
f7214ff4 SP	376	}
	377	dev->x_old = x;
	378	dev->y_old = y;
	379	}
	380	else if (!x && !y) {
	381
	382	dev->x_old = dev->y_old = -1;
c5b7c7c3 DT	383	input_report_key(dev->input, BTN_TOUCH, 0);
	384	input_report_abs(dev->input, ABS_PRESSURE, 0);
	385	atp_report_fingers(dev->input, 0);
f7214ff4 SP	386
	387	/* reset the accumulator on release */
	388	memset(dev->xy_acc, 0, sizeof(dev->xy_acc));
	389	}
	390
e1e02c9f MH	391	input_report_key(dev->input, BTN_LEFT,
e1e02c9f MH	392	!!dev->data[dev->datalen - 1]);
f7214ff4	393
c5b7c7c3	394	input_sync(dev->input);
f7214ff4 SP	395
	396	exit:
	397	retval = usb_submit_urb(dev->urb, GFP_ATOMIC);
	398	if (retval) {
	399	err("%s - usb_submit_urb failed with result %d",
	400	__FUNCTION__, retval);
	401	}
	402	}
	403
	404	static int atp_open(struct input_dev *input)
	405	{
	406	struct atp *dev = input->private;
	407
	408	if (usb_submit_urb(dev->urb, GFP_ATOMIC))
	409	return -EIO;
	410
	411	dev->open = 1;
	412	return 0;
	413	}
	414
	415	static void atp_close(struct input_dev *input)
	416	{
	417	struct atp *dev = input->private;
	418
	419	usb_kill_urb(dev->urb);
	420	dev->open = 0;
	421	}
	422
	423	static int atp_probe(struct usb_interface iface, const struct usb_device_id id)
	424	{
c5b7c7c3 DT	425	struct atp *dev;
	426	struct input_dev *input_dev;
	427	struct usb_device *udev = interface_to_usbdev(iface);
f7214ff4 SP	428	struct usb_host_interface *iface_desc;
	429	struct usb_endpoint_descriptor *endpoint;
	430	int int_in_endpointAddr = 0;
	431	int i, retval = -ENOMEM;
	432
f7214ff4 SP	433
	434	/* set up the endpoint information */
	435	/* use only the first interrupt-in endpoint */
	436	iface_desc = iface->cur_altsetting;
	437	for (i = 0; i < iface_desc->desc.bNumEndpoints; i++) {
	438	endpoint = &iface_desc->endpoint[i].desc;
97b107ca	439	if (!int_in_endpointAddr && usb_endpoint_is_int_in(endpoint)) {
f7214ff4 SP	440	/* we found an interrupt in endpoint */
	441	int_in_endpointAddr = endpoint->bEndpointAddress;
	442	break;
	443	}
	444	}
	445	if (!int_in_endpointAddr) {
f7214ff4	446	err("Could not find int-in endpoint");
c5b7c7c3	447	return -EIO;
f7214ff4 SP	448	}
f7214ff4 SP	449
c5b7c7c3 DT	450	/* allocate memory for our device state and initialize it */
	451	dev = kzalloc(sizeof(struct atp), GFP_KERNEL);
	452	input_dev = input_allocate_device();
	453	if (!dev \|\| !input_dev) {
	454	err("Out of memory");
	455	goto err_free_devs;
	456	}
	457
	458	dev->udev = udev;
	459	dev->input = input_dev;
e1e02c9f	460	dev->overflowwarn = 0;
9effa978 NB	461	if (atp_is_geyser_3(dev))
	462	dev->datalen = 64;
	463	else if (atp_is_geyser_2(dev))
	464	dev->datalen = 64;
	465	else
	466	dev->datalen = 81;
	467
	468	if (atp_is_geyser_3(dev)) {
	469	/*
	470	* By default Geyser 3 device sends standard USB HID mouse
	471	* packets (Report ID 2). This code changes device mode, so it
	472	* sends raw sensor reports (Report ID 5).
	473	*/
	474	char data[8];
	475	int size;
	476
	477	size = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0),
	478	ATP_GEYSER3_MODE_READ_REQUEST_ID,
	479	USB_DIR_IN \| USB_TYPE_CLASS \| USB_RECIP_INTERFACE,
	480	ATP_GEYSER3_MODE_REQUEST_VALUE,
	481	ATP_GEYSER3_MODE_REQUEST_INDEX, &data, 8, 5000);
	482
	483	if (size != 8) {
	484	err("Could not do mode read request from device"
	485	" (Geyser 3 mode)");
	486	goto err_free_devs;
	487	}
	488
	489	/* Apply the mode switch */
	490	data[0] = ATP_GEYSER3_MODE_VENDOR_VALUE;
	491
	492	size = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
	493	ATP_GEYSER3_MODE_WRITE_REQUEST_ID,
	494	USB_DIR_OUT \| USB_TYPE_CLASS \| USB_RECIP_INTERFACE,
	495	ATP_GEYSER3_MODE_REQUEST_VALUE,
	496	ATP_GEYSER3_MODE_REQUEST_INDEX, &data, 8, 5000);
	497
	498	if (size != 8) {
	499	err("Could not do mode write request to device"
	500	" (Geyser 3 mode)");
	501	goto err_free_devs;
	502	}
	503	printk("appletouch Geyser 3 inited.\n");
	504	}
f7214ff4 SP	505
	506	dev->urb = usb_alloc_urb(0, GFP_KERNEL);
	507	if (!dev->urb) {
	508	retval = -ENOMEM;
c5b7c7c3	509	goto err_free_devs;
f7214ff4	510	}
c5b7c7c3	511
e1e02c9f	512	dev->data = usb_buffer_alloc(dev->udev, dev->datalen, GFP_KERNEL,
f7214ff4 SP	513	&dev->urb->transfer_dma);
	514	if (!dev->data) {
	515	retval = -ENOMEM;
c5b7c7c3	516	goto err_free_urb;
f7214ff4	517	}
c5b7c7c3 DT	518
	519	usb_fill_int_urb(dev->urb, udev,
	520	usb_rcvintpipe(udev, int_in_endpointAddr),
e1e02c9f	521	dev->data, dev->datalen, atp_complete, dev, 1);
f7214ff4	522
c5b7c7c3 DT	523	usb_make_path(udev, dev->phys, sizeof(dev->phys));
	524	strlcat(dev->phys, "/input0", sizeof(dev->phys));
	525
	526	input_dev->name = "appletouch";
	527	input_dev->phys = dev->phys;
	528	usb_to_input_id(dev->udev, &input_dev->id);
	529	input_dev->cdev.dev = &iface->dev;
f7214ff4	530
c5b7c7c3 DT	531	input_dev->private = dev;
	532	input_dev->open = atp_open;
	533	input_dev->close = atp_close;
f7214ff4	534
c5b7c7c3	535	set_bit(EV_ABS, input_dev->evbit);
f7214ff4	536
9effa978 NB	537	if (atp_is_geyser_3(dev)) {
	538	/*
	539	* MacBook have 20 X sensors, 10 Y sensors
	540	*/
	541	input_set_abs_params(input_dev, ABS_X, 0,
	542	((20 - 1) * ATP_XFACT) - 1, ATP_FUZZ, 0);
	543	input_set_abs_params(input_dev, ABS_Y, 0,
	544	((10 - 1) * ATP_YFACT) - 1, ATP_FUZZ, 0);
	545	} else if (atp_is_geyser_2(dev)) {
e1e02c9f MH	546	/*
	547	* Oct 2005 15" PowerBooks have 15 X sensors, 17" are detected
	548	* later.
	549	*/
	550	input_set_abs_params(input_dev, ABS_X, 0,
	551	((15 - 1) * ATP_XFACT) - 1, ATP_FUZZ, 0);
	552	input_set_abs_params(input_dev, ABS_Y, 0,
	553	((9 - 1) * ATP_YFACT) - 1, ATP_FUZZ, 0);
	554	} else {
	555	/*
	556	* 12" and 15" Powerbooks only have 16 x sensors,
	557	* 17" models are detected later.
	558	*/
	559	input_set_abs_params(input_dev, ABS_X, 0,
	560	(16 - 1) * ATP_XFACT - 1, ATP_FUZZ, 0);
	561	input_set_abs_params(input_dev, ABS_Y, 0,
	562	(ATP_YSENSORS - 1) * ATP_YFACT - 1, ATP_FUZZ, 0);
	563	}
c5b7c7c3	564	input_set_abs_params(input_dev, ABS_PRESSURE, 0, ATP_PRESSURE, 0, 0);
f7214ff4	565
c5b7c7c3 DT	566	set_bit(EV_KEY, input_dev->evbit);
	567	set_bit(BTN_TOUCH, input_dev->keybit);
	568	set_bit(BTN_TOOL_FINGER, input_dev->keybit);
	569	set_bit(BTN_TOOL_DOUBLETAP, input_dev->keybit);
	570	set_bit(BTN_TOOL_TRIPLETAP, input_dev->keybit);
	571	set_bit(BTN_LEFT, input_dev->keybit);
f7214ff4	572
c5b7c7c3	573	input_register_device(dev->input);
f7214ff4	574
c5b7c7c3 DT	575	/* save our data pointer in this interface device */
c5b7c7c3 DT	576	usb_set_intfdata(iface, dev);
f7214ff4 SP	577
	578	return 0;
	579
c5b7c7c3	580	err_free_urb:
f7214ff4	581	usb_free_urb(dev->urb);
c5b7c7c3	582	err_free_devs:
f7214ff4	583	usb_set_intfdata(iface, NULL);
f7214ff4	584	kfree(dev);
c5b7c7c3	585	input_free_device(input_dev);
f7214ff4 SP	586	return retval;
	587	}
	588
	589	static void atp_disconnect(struct usb_interface *iface)
	590	{
	591	struct atp *dev = usb_get_intfdata(iface);
	592
	593	usb_set_intfdata(iface, NULL);
	594	if (dev) {
	595	usb_kill_urb(dev->urb);
c5b7c7c3	596	input_unregister_device(dev->input);
e1e02c9f	597	usb_buffer_free(dev->udev, dev->datalen,
f7214ff4	598	dev->data, dev->urb->transfer_dma);
7af569ab	599	usb_free_urb(dev->urb);
f7214ff4 SP	600	kfree(dev);
	601	}
	602	printk(KERN_INFO "input: appletouch disconnected\n");
	603	}
	604
	605	static int atp_suspend(struct usb_interface *iface, pm_message_t message)
	606	{
	607	struct atp *dev = usb_get_intfdata(iface);
	608	usb_kill_urb(dev->urb);
	609	dev->valid = 0;
	610	return 0;
	611	}
	612
	613	static int atp_resume(struct usb_interface *iface)
	614	{
	615	struct atp *dev = usb_get_intfdata(iface);
	616	if (dev->open && usb_submit_urb(dev->urb, GFP_ATOMIC))
	617	return -EIO;
	618
	619	return 0;
	620	}
	621
	622	static struct usb_driver atp_driver = {
f7214ff4 SP	623	.name = "appletouch",
	624	.probe = atp_probe,
	625	.disconnect = atp_disconnect,
	626	.suspend = atp_suspend,
	627	.resume = atp_resume,
	628	.id_table = atp_table,
	629	};
	630
	631	static int __init atp_init(void)
	632	{
	633	return usb_register(&atp_driver);
	634	}
	635
	636	static void __exit atp_exit(void)
	637	{
	638	usb_deregister(&atp_driver);
	639	}
	640
	641	module_init(atp_init);
	642	module_exit(atp_exit);