vserver 1.9.3
[linux-2.6.git] / drivers / input / mouse / psmouse-base.c
index 3d4a300..7fbd926 100644 (file)
 #include "synaptics.h"
 #include "logips2pp.h"
 
+#define DRIVER_DESC    "PS/2 mouse driver"
+
 MODULE_AUTHOR("Vojtech Pavlik <vojtech@suse.cz>");
-MODULE_DESCRIPTION("PS/2 mouse driver");
+MODULE_DESCRIPTION(DRIVER_DESC);
 MODULE_LICENSE("GPL");
 
 static char *psmouse_proto;
@@ -43,9 +45,9 @@ int psmouse_smartscroll = 1;
 module_param_named(smartscroll, psmouse_smartscroll, bool, 0);
 MODULE_PARM_DESC(smartscroll, "Logitech Smartscroll autorepeat, 1 = enabled (default), 0 = disabled.");
 
-unsigned int psmouse_resetafter;
+static unsigned int psmouse_resetafter;
 module_param_named(resetafter, psmouse_resetafter, uint, 0);
-MODULE_PARM_DESC(resetafter, "Reset Synaptics Touchpad after so many bad packets (0 = never).");
+MODULE_PARM_DESC(resetafter, "Reset device after so many bad packets (0 = never).");
 
 __obsolete_setup("psmouse_noext");
 __obsolete_setup("psmouse_resolution=");
@@ -56,15 +58,22 @@ __obsolete_setup("psmouse_rate=");
 static char *psmouse_protocols[] = { "None", "PS/2", "PS2++", "PS2T++", "GenPS/2", "ImPS/2", "ImExPS/2", "SynPS/2"};
 
 /*
- * psmouse_process_packet() analyzes the PS/2 mouse packet contents and
- * reports relevant events to the input module.
+ * psmouse_process_byte() analyzes the PS/2 data stream and reports
+ * relevant events to the input module once full packet has arrived.
  */
 
-static void psmouse_process_packet(struct psmouse *psmouse, struct pt_regs *regs)
+static psmouse_ret_t psmouse_process_byte(struct psmouse *psmouse, struct pt_regs *regs)
 {
        struct input_dev *dev = &psmouse->dev;
        unsigned char *packet = psmouse->packet;
 
+       if (psmouse->pktcnt < 3 + (psmouse->type >= PSMOUSE_GENPS))
+               return PSMOUSE_GOOD_DATA;
+
+/*
+ * Full packet accumulated, process it
+ */
+
        input_regs(dev, regs);
 
 /*
@@ -112,6 +121,8 @@ static void psmouse_process_packet(struct psmouse *psmouse, struct pt_regs *regs
        input_report_rel(dev, REL_Y, packet[2] ? (int) ((packet[0] << 3) & 0x100) - (int) packet[2] : 0);
 
        input_sync(dev);
+
+       return PSMOUSE_FULL_PACKET;
 }
 
 /*
@@ -123,6 +134,7 @@ static irqreturn_t psmouse_interrupt(struct serio *serio,
                unsigned char data, unsigned int flags, struct pt_regs *regs)
 {
        struct psmouse *psmouse = serio->private;
+       psmouse_ret_t rc;
 
        if (psmouse->state == PSMOUSE_IGNORE)
                goto out;
@@ -132,37 +144,67 @@ static irqreturn_t psmouse_interrupt(struct serio *serio,
                        printk(KERN_WARNING "psmouse.c: bad data from KBC -%s%s\n",
                                flags & SERIO_TIMEOUT ? " timeout" : "",
                                flags & SERIO_PARITY ? " bad parity" : "");
-               if (psmouse->acking) {
-                       psmouse->ack = -1;
-                       psmouse->acking = 0;
-               }
-               psmouse->pktcnt = 0;
+               psmouse->nak = 1;
+               clear_bit(PSMOUSE_FLAG_ACK, &psmouse->flags);
+               clear_bit(PSMOUSE_FLAG_CMD,  &psmouse->flags);
+               wake_up_interruptible(&psmouse->wait);
                goto out;
        }
 
-       if (psmouse->acking) {
+       if (test_bit(PSMOUSE_FLAG_ACK, &psmouse->flags)) {
                switch (data) {
                        case PSMOUSE_RET_ACK:
-                               psmouse->ack = 1;
+                               psmouse->nak = 0;
                                break;
+
                        case PSMOUSE_RET_NAK:
-                               psmouse->ack = -1;
+                               psmouse->nak = 1;
                                break;
+
+                       /*
+                        * Workaround for mice which don't ACK the Get ID command.
+                        * These are valid mouse IDs that we recognize.
+                        */
+                       case 0x00:
+                       case 0x03:
+                       case 0x04:
+                               if (test_bit(PSMOUSE_FLAG_WAITID, &psmouse->flags)) {
+                                       psmouse->nak = 0;
+                                       break;
+                               }
+                               /* Fall through */
                        default:
-                               psmouse->ack = 1;       /* Workaround for mice which don't ACK the Get ID command */
-                               if (psmouse->cmdcnt)
-                                       psmouse->cmdbuf[--psmouse->cmdcnt] = data;
-                               break;
+                               goto out;
                }
-               psmouse->acking = 0;
-               goto out;
+
+               if (!psmouse->nak && psmouse->cmdcnt) {
+                       set_bit(PSMOUSE_FLAG_CMD, &psmouse->flags);
+                       set_bit(PSMOUSE_FLAG_CMD1, &psmouse->flags);
+               }
+               clear_bit(PSMOUSE_FLAG_ACK, &psmouse->flags);
+               wake_up_interruptible(&psmouse->wait);
+
+               if (data == PSMOUSE_RET_ACK || data == PSMOUSE_RET_NAK)
+                       goto out;
        }
 
-       if (psmouse->cmdcnt) {
-               psmouse->cmdbuf[--psmouse->cmdcnt] = data;
+       if (test_bit(PSMOUSE_FLAG_CMD, &psmouse->flags)) {
+               if (psmouse->cmdcnt)
+                       psmouse->cmdbuf[--psmouse->cmdcnt] = data;
+
+               if (test_and_clear_bit(PSMOUSE_FLAG_CMD1, &psmouse->flags) && psmouse->cmdcnt)
+                       wake_up_interruptible(&psmouse->wait);
+
+               if (!psmouse->cmdcnt) {
+                       clear_bit(PSMOUSE_FLAG_CMD, &psmouse->flags);
+                       wake_up_interruptible(&psmouse->wait);
+               }
                goto out;
        }
 
+       if (psmouse->state == PSMOUSE_INITIALIZING)
+               goto out;
+
        if (psmouse->state == PSMOUSE_ACTIVATED &&
            psmouse->pktcnt && time_after(jiffies, psmouse->last + HZ/2)) {
                printk(KERN_WARNING "psmouse.c: %s at %s lost synchronization, throwing %d bytes away.\n",
@@ -180,7 +222,7 @@ static irqreturn_t psmouse_interrupt(struct serio *serio,
                if (psmouse->pktcnt == 2) {
                        if (psmouse->packet[1] == PSMOUSE_RET_ID) {
                                psmouse->state = PSMOUSE_IGNORE;
-                               serio_rescan(serio);
+                               serio_reconnect(serio);
                                goto out;
                        }
                        if (psmouse->type == PSMOUSE_SYNAPTICS) {
@@ -193,19 +235,32 @@ static irqreturn_t psmouse_interrupt(struct serio *serio,
                }
        }
 
-       if (psmouse->type == PSMOUSE_SYNAPTICS) {
-               /*
-                * The synaptics driver has its own resync logic,
-                * so it needs to receive all bytes one at a time.
-                */
-               synaptics_process_byte(psmouse, regs);
-               goto out;
-       }
+       rc = psmouse->protocol_handler(psmouse, regs);
 
-       if (psmouse->pktcnt == 3 + (psmouse->type >= PSMOUSE_GENPS)) {
-               psmouse_process_packet(psmouse, regs);
-               psmouse->pktcnt = 0;
-               goto out;
+       switch (rc) {
+               case PSMOUSE_BAD_DATA:
+                       printk(KERN_WARNING "psmouse.c: %s at %s lost sync at byte %d\n",
+                               psmouse->name, psmouse->phys, psmouse->pktcnt);
+                       psmouse->pktcnt = 0;
+
+                       if (++psmouse->out_of_sync == psmouse_resetafter) {
+                               psmouse->state = PSMOUSE_IGNORE;
+                               printk(KERN_NOTICE "psmouse.c: issuing reconnect request\n");
+                               serio_reconnect(psmouse->serio);
+                       }
+                       break;
+
+               case PSMOUSE_FULL_PACKET:
+                       psmouse->pktcnt = 0;
+                       if (psmouse->out_of_sync) {
+                               psmouse->out_of_sync = 0;
+                               printk(KERN_NOTICE "psmouse.c: %s at %s - driver resynched.\n",
+                                       psmouse->name, psmouse->phys);
+                       }
+                       break;
+
+               case PSMOUSE_GOOD_DATA:
+                       break;
        }
 out:
        return IRQ_HANDLED;
@@ -215,74 +270,116 @@ out:
  * psmouse_sendbyte() sends a byte to the mouse, and waits for acknowledge.
  * It doesn't handle retransmission, though it could - because when there would
  * be need for retransmissions, the mouse has to be replaced anyway.
+ *
+ * psmouse_sendbyte() can only be called from a process context
  */
 
 static int psmouse_sendbyte(struct psmouse *psmouse, unsigned char byte)
 {
-       int timeout = 10000; /* 100 msec */
-       psmouse->ack = 0;
-       psmouse->acking = 1;
-
-       if (serio_write(psmouse->serio, byte)) {
-               psmouse->acking = 0;
-               return -1;
-       }
+       psmouse->nak = 1;
+       set_bit(PSMOUSE_FLAG_ACK, &psmouse->flags);
 
-       while (!psmouse->ack && timeout--) udelay(10);
+       if (serio_write(psmouse->serio, byte) == 0)
+               wait_event_interruptible_timeout(psmouse->wait,
+                               !test_bit(PSMOUSE_FLAG_ACK, &psmouse->flags),
+                               msecs_to_jiffies(200));
 
-       return -(psmouse->ack <= 0);
+       clear_bit(PSMOUSE_FLAG_ACK, &psmouse->flags);
+       return -psmouse->nak;
 }
 
 /*
  * psmouse_command() sends a command and its parameters to the mouse,
  * then waits for the response and puts it in the param array.
+ *
+ * psmouse_command() can only be called from a process context
  */
 
 int psmouse_command(struct psmouse *psmouse, unsigned char *param, int command)
 {
-       int timeout = 500000; /* 500 msec */
+       int timeout;
        int send = (command >> 12) & 0xf;
        int receive = (command >> 8) & 0xf;
+       int rc = -1;
        int i;
 
-       psmouse->cmdcnt = receive;
+       timeout = msecs_to_jiffies(command == PSMOUSE_CMD_RESET_BAT ? 4000 : 500);
+
+       clear_bit(PSMOUSE_FLAG_CMD, &psmouse->flags);
+       if (command == PSMOUSE_CMD_GETID)
+               set_bit(PSMOUSE_FLAG_WAITID, &psmouse->flags);
 
-       if (command == PSMOUSE_CMD_RESET_BAT)
-                timeout = 4000000; /* 4 sec */
+       if (receive && param)
+               for (i = 0; i < receive; i++)
+                       psmouse->cmdbuf[(receive - 1) - i] = param[i];
 
-       /* initialize cmdbuf with preset values from param */
-       if (receive)
-          for (i = 0; i < receive; i++)
-               psmouse->cmdbuf[(receive - 1) - i] = param[i];
+       psmouse->cmdcnt = receive;
 
        if (command & 0xff)
                if (psmouse_sendbyte(psmouse, command & 0xff))
-                       return (psmouse->cmdcnt = 0) - 1;
+                       goto out;
 
        for (i = 0; i < send; i++)
                if (psmouse_sendbyte(psmouse, param[i]))
-                       return (psmouse->cmdcnt = 0) - 1;
-
-       while (psmouse->cmdcnt && timeout--) {
-
-               if (psmouse->cmdcnt == 1 && command == PSMOUSE_CMD_RESET_BAT &&
-                               timeout > 100000) /* do not run in a endless loop */
-                       timeout = 100000; /* 1 sec */
+                       goto out;
 
-               if (psmouse->cmdcnt == 1 && command == PSMOUSE_CMD_GETID &&
-                   psmouse->cmdbuf[1] != 0xab && psmouse->cmdbuf[1] != 0xac) {
+       timeout = wait_event_interruptible_timeout(psmouse->wait,
+                               !test_bit(PSMOUSE_FLAG_CMD1, &psmouse->flags), timeout);
+
+       if (psmouse->cmdcnt && timeout > 0) {
+               if (command == PSMOUSE_CMD_RESET_BAT && jiffies_to_msecs(timeout) > 100)
+                       timeout = msecs_to_jiffies(100);
+
+               if (command == PSMOUSE_CMD_GETID &&
+                   psmouse->cmdbuf[receive - 1] != 0xab && psmouse->cmdbuf[receive - 1] != 0xac) {
+                       /*
+                        * Device behind the port is not a keyboard
+                        * so we don't need to wait for the 2nd byte
+                        * of ID response.
+                        */
+                       clear_bit(PSMOUSE_FLAG_CMD, &psmouse->flags);
                        psmouse->cmdcnt = 0;
-                       break;
                }
 
-               udelay(1);
+               wait_event_interruptible_timeout(psmouse->wait,
+                               !test_bit(PSMOUSE_FLAG_CMD, &psmouse->flags), timeout);
        }
 
-       for (i = 0; i < receive; i++)
-               param[i] = psmouse->cmdbuf[(receive - 1) - i];
+       if (param)
+               for (i = 0; i < receive; i++)
+                       param[i] = psmouse->cmdbuf[(receive - 1) - i];
+
+       if (psmouse->cmdcnt && (command != PSMOUSE_CMD_RESET_BAT || psmouse->cmdcnt != 1))
+               goto out;
+
+       rc = 0;
+
+out:
+       clear_bit(PSMOUSE_FLAG_CMD, &psmouse->flags);
+       clear_bit(PSMOUSE_FLAG_CMD1, &psmouse->flags);
+       clear_bit(PSMOUSE_FLAG_WAITID, &psmouse->flags);
+       return rc;
+}
+
+/*
+ * psmouse_sliced_command() sends an extended PS/2 command to the mouse
+ * using sliced syntax, understood by advanced devices, such as Logitech
+ * or Synaptics touchpads. The command is encoded as:
+ * 0xE6 0xE8 rr 0xE8 ss 0xE8 tt 0xE8 uu where (rr*64)+(ss*16)+(tt*4)+uu
+ * is the command.
+ */
+int psmouse_sliced_command(struct psmouse *psmouse, unsigned char command)
+{
+       int i;
+
+       if (psmouse_command(psmouse, NULL, PSMOUSE_CMD_SETSCALE11))
+               return -1;
 
-       if (psmouse->cmdcnt)
-               return (psmouse->cmdcnt = 0) - 1;
+       for (i = 6; i >= 0; i -= 2) {
+               unsigned char d = (command >> i) & 3;
+               if (psmouse_command(psmouse, &d, PSMOUSE_CMD_SETRES))
+                       return -1;
+       }
 
        return 0;
 }
@@ -347,6 +444,8 @@ static int im_explorer_detect(struct psmouse *psmouse)
 {
        unsigned char param[2];
 
+       intellimouse_detect(psmouse);
+
        param[0] = 200;
        psmouse_command(psmouse, param, PSMOUSE_CMD_SETRATE);
        param[0] = 200;
@@ -363,31 +462,31 @@ static int im_explorer_detect(struct psmouse *psmouse)
  * the mouse may have.
  */
 
-static int psmouse_extensions(struct psmouse *psmouse)
+static int psmouse_extensions(struct psmouse *psmouse,
+                             unsigned int max_proto, int set_properties)
 {
        int synaptics_hardware = 0;
 
-       psmouse->vendor = "Generic";
-       psmouse->name = "Mouse";
-       psmouse->model = 0;
-
 /*
  * Try Synaptics TouchPad
  */
-       if (psmouse_max_proto > PSMOUSE_PS2 && synaptics_detect(psmouse)) {
+       if (max_proto > PSMOUSE_PS2 && synaptics_detect(psmouse)) {
                synaptics_hardware = 1;
-               psmouse->vendor = "Synaptics";
-               psmouse->name = "TouchPad";
 
-               if (psmouse_max_proto > PSMOUSE_IMEX) {
-                       if (synaptics_init(psmouse) == 0)
+               if (set_properties) {
+                       psmouse->vendor = "Synaptics";
+                       psmouse->name = "TouchPad";
+               }
+
+               if (max_proto > PSMOUSE_IMEX) {
+                       if (!set_properties || synaptics_init(psmouse) == 0)
                                return PSMOUSE_SYNAPTICS;
 /*
  * Some Synaptics touchpads can emulate extended protocols (like IMPS/2).
  * Unfortunately Logitech/Genius probes confuse some firmware versions so
  * we'll have to skip them.
  */
-                       psmouse_max_proto = PSMOUSE_IMEX;
+                       max_proto = PSMOUSE_IMEX;
                }
 /*
  * Make sure that touchpad is in relative mode, gestures (taps) are enabled
@@ -395,35 +494,46 @@ static int psmouse_extensions(struct psmouse *psmouse)
                synaptics_reset(psmouse);
        }
 
-       if (psmouse_max_proto > PSMOUSE_IMEX && genius_detect(psmouse)) {
-               set_bit(BTN_EXTRA, psmouse->dev.keybit);
-               set_bit(BTN_SIDE, psmouse->dev.keybit);
-               set_bit(REL_WHEEL, psmouse->dev.relbit);
+       if (max_proto > PSMOUSE_IMEX && genius_detect(psmouse)) {
+
+               if (set_properties) {
+                       set_bit(BTN_EXTRA, psmouse->dev.keybit);
+                       set_bit(BTN_SIDE, psmouse->dev.keybit);
+                       set_bit(REL_WHEEL, psmouse->dev.relbit);
+                       psmouse->vendor = "Genius";
+                       psmouse->name = "Wheel Mouse";
+               }
 
-               psmouse->vendor = "Genius";
-               psmouse->name = "Wheel Mouse";
                return PSMOUSE_GENPS;
        }
 
-       if (psmouse_max_proto > PSMOUSE_IMEX) {
-               int type = ps2pp_detect(psmouse);
-               if (type)
+       if (max_proto > PSMOUSE_IMEX) {
+               int type = ps2pp_init(psmouse, set_properties);
+               if (type > PSMOUSE_PS2)
                        return type;
        }
 
-       if (psmouse_max_proto >= PSMOUSE_IMPS && intellimouse_detect(psmouse)) {
-               set_bit(REL_WHEEL, psmouse->dev.relbit);
+       if (max_proto >= PSMOUSE_IMEX && im_explorer_detect(psmouse)) {
 
-               if (psmouse_max_proto >= PSMOUSE_IMEX &&
-                                       im_explorer_detect(psmouse)) {
+               if (set_properties) {
+                       set_bit(REL_WHEEL, psmouse->dev.relbit);
                        set_bit(BTN_SIDE, psmouse->dev.keybit);
                        set_bit(BTN_EXTRA, psmouse->dev.keybit);
+                       if (!psmouse->name)
+                               psmouse->name = "Explorer Mouse";
+               }
+
+               return PSMOUSE_IMEX;
+       }
+
+       if (max_proto >= PSMOUSE_IMPS && intellimouse_detect(psmouse)) {
 
-                       psmouse->name = "Explorer Mouse";
-                       return PSMOUSE_IMEX;
+               if (set_properties) {
+                       set_bit(REL_WHEEL, psmouse->dev.relbit);
+                       if (!psmouse->name)
+                               psmouse->name = "Wheel Mouse";
                }
 
-               psmouse->name = "Wheel Mouse";
                return PSMOUSE_IMPS;
        }
 
@@ -473,12 +583,7 @@ static int psmouse_probe(struct psmouse *psmouse)
        if (psmouse_command(psmouse, NULL, PSMOUSE_CMD_RESET_DIS))
                printk(KERN_WARNING "psmouse.c: Failed to reset mouse on %s\n", psmouse->serio->phys);
 
-/*
- * And here we try to determine if it has any extensions over the
- * basic PS/2 3-button mouse.
- */
-
-       return psmouse->type = psmouse_extensions(psmouse);
+       return 0;
 }
 
 /*
@@ -544,6 +649,21 @@ static void psmouse_initialize(struct psmouse *psmouse)
        psmouse_command(psmouse, param, PSMOUSE_CMD_SETSTREAM);
 }
 
+/*
+ * psmouse_set_state() sets new psmouse state and resets all flags and
+ * counters while holding serio lock so fighting with interrupt handler
+ * is not a concern.
+ */
+
+static void psmouse_set_state(struct psmouse *psmouse, enum psmouse_state new_state)
+{
+       serio_pause_rx(psmouse->serio);
+       psmouse->state = new_state;
+       psmouse->pktcnt = psmouse->cmdcnt = psmouse->out_of_sync = 0;
+       psmouse->flags = 0;
+       serio_continue_rx(psmouse->serio);
+}
+
 /*
  * psmouse_activate() enables the mouse so that we get motion reports from it.
  */
@@ -553,9 +673,24 @@ static void psmouse_activate(struct psmouse *psmouse)
        if (psmouse_command(psmouse, NULL, PSMOUSE_CMD_ENABLE))
                printk(KERN_WARNING "psmouse.c: Failed to enable mouse on %s\n", psmouse->serio->phys);
 
-       psmouse->state = PSMOUSE_ACTIVATED;
+       psmouse_set_state(psmouse, PSMOUSE_ACTIVATED);
+}
+
+
+/*
+ * psmouse_deactivate() puts the mouse into poll mode so that we don't get motion
+ * reports from it unless we explicitely request it.
+ */
+
+static void psmouse_deactivate(struct psmouse *psmouse)
+{
+       if (psmouse_command(psmouse, NULL, PSMOUSE_CMD_DISABLE))
+               printk(KERN_WARNING "psmouse.c: Failed to deactivate mouse on %s\n", psmouse->serio->phys);
+
+       psmouse_set_state(psmouse, PSMOUSE_CMD_MODE);
 }
 
+
 /*
  * psmouse_cleanup() resets the mouse into power-on state.
  */
@@ -573,22 +708,21 @@ static void psmouse_cleanup(struct serio *serio)
 
 static void psmouse_disconnect(struct serio *serio)
 {
-       struct psmouse *psmouse = serio->private;
+       struct psmouse *psmouse, *parent;
 
-       psmouse->state = PSMOUSE_CMD_MODE;
+       psmouse = serio->private;
+       psmouse_set_state(psmouse, PSMOUSE_CMD_MODE);
 
-       if (psmouse->ptport) {
-               if (psmouse->ptport->deactivate)
-                       psmouse->ptport->deactivate(psmouse);
-               __serio_unregister_port(&psmouse->ptport->serio); /* we have serio_sem */
-               kfree(psmouse->ptport);
-               psmouse->ptport = NULL;
+       if (serio->parent && (serio->type & SERIO_TYPE) == SERIO_PS_PSTHRU) {
+               parent = serio->parent->private;
+               if (parent->pt_deactivate)
+                       parent->pt_deactivate(parent);
        }
 
        if (psmouse->disconnect)
                psmouse->disconnect(psmouse);
 
-       psmouse->state = PSMOUSE_IGNORE;
+       psmouse_set_state(psmouse, PSMOUSE_IGNORE);
 
        input_unregister_device(&psmouse->dev);
        serio_close(serio);
@@ -599,42 +733,59 @@ static void psmouse_disconnect(struct serio *serio)
  * psmouse_connect() is a callback from the serio module when
  * an unhandled serio port is found.
  */
-static void psmouse_connect(struct serio *serio, struct serio_dev *dev)
+static void psmouse_connect(struct serio *serio, struct serio_driver *drv)
 {
-       struct psmouse *psmouse;
+       struct psmouse *psmouse, *parent = NULL;
 
        if ((serio->type & SERIO_TYPE) != SERIO_8042 &&
            (serio->type & SERIO_TYPE) != SERIO_PS_PSTHRU)
                return;
 
+       /*
+        * If this is a pass-through port deactivate parent so the device
+        * connected to this port can be successfully identified
+        */
+       if (serio->parent && (serio->type & SERIO_TYPE) == SERIO_PS_PSTHRU) {
+               parent = serio->parent->private;
+               psmouse_deactivate(parent);
+       }
+
        if (!(psmouse = kmalloc(sizeof(struct psmouse), GFP_KERNEL)))
-               return;
+               goto out;
 
        memset(psmouse, 0, sizeof(struct psmouse));
 
+       init_waitqueue_head(&psmouse->wait);
        init_input_dev(&psmouse->dev);
        psmouse->dev.evbit[0] = BIT(EV_KEY) | BIT(EV_REL);
        psmouse->dev.keybit[LONG(BTN_MOUSE)] = BIT(BTN_LEFT) | BIT(BTN_MIDDLE) | BIT(BTN_RIGHT);
        psmouse->dev.relbit[0] = BIT(REL_X) | BIT(REL_Y);
-
-       psmouse->state = PSMOUSE_CMD_MODE;
        psmouse->serio = serio;
        psmouse->dev.private = psmouse;
+       psmouse_set_state(psmouse, PSMOUSE_INITIALIZING);
 
        serio->private = psmouse;
-       if (serio_open(serio, dev)) {
+       if (serio_open(serio, drv)) {
                kfree(psmouse);
                serio->private = NULL;
-               return;
+               goto out;
        }
 
-       if (psmouse_probe(psmouse) <= 0) {
+       if (psmouse_probe(psmouse) < 0) {
                serio_close(serio);
                kfree(psmouse);
                serio->private = NULL;
-               return;
+               goto out;
        }
 
+       psmouse->type = psmouse_extensions(psmouse, psmouse_max_proto, 1);
+       if (!psmouse->vendor)
+               psmouse->vendor = "Generic";
+       if (!psmouse->name)
+               psmouse->name = "Mouse";
+       if (!psmouse->protocol_handler)
+               psmouse->protocol_handler = psmouse_process_byte;
+
        sprintf(psmouse->devname, "%s %s %s",
                psmouse_protocols[psmouse->type], psmouse->vendor, psmouse->name);
        sprintf(psmouse->phys, "%s/input0",
@@ -651,66 +802,88 @@ static void psmouse_connect(struct serio *serio, struct serio_dev *dev)
 
        printk(KERN_INFO "input: %s on %s\n", psmouse->devname, serio->phys);
 
+       psmouse_set_state(psmouse, PSMOUSE_CMD_MODE);
+
        psmouse_initialize(psmouse);
 
-       if (psmouse->ptport) {
-               printk(KERN_INFO "serio: %s port at %s\n", psmouse->ptport->serio.name, psmouse->phys);
-               __serio_register_port(&psmouse->ptport->serio); /* we have serio_sem */
-               if (psmouse->ptport->activate)
-                       psmouse->ptport->activate(psmouse);
+       if (parent && parent->pt_activate)
+               parent->pt_activate(parent);
+
+       if (serio->child) {
+               /*
+                * Nothing to be done here, serio core will detect that
+                * the driver set serio->child and will register it for us.
+                */
+               printk(KERN_INFO "serio: %s port at %s\n", serio->child->name, psmouse->phys);
        }
 
        psmouse_activate(psmouse);
+
+out:
+       /* If this is a pass-through port the parent awaits to be activated */
+       if (parent)
+               psmouse_activate(parent);
 }
 
 
 static int psmouse_reconnect(struct serio *serio)
 {
        struct psmouse *psmouse = serio->private;
-       struct serio_dev *dev = serio->dev;
-       int old_type;
+       struct psmouse *parent = NULL;
+       struct serio_driver *drv = serio->drv;
+       int rc = -1;
 
-       if (!dev || !psmouse) {
+       if (!drv || !psmouse) {
                printk(KERN_DEBUG "psmouse: reconnect request, but serio is disconnected, ignoring...\n");
                return -1;
        }
 
-       old_type = psmouse->type;
+       if (serio->parent && (serio->type & SERIO_TYPE) == SERIO_PS_PSTHRU) {
+               parent = serio->parent->private;
+               psmouse_deactivate(parent);
+       }
+
+       psmouse_set_state(psmouse, PSMOUSE_INITIALIZING);
 
-       psmouse->state = PSMOUSE_CMD_MODE;
-       psmouse->type = psmouse->acking = psmouse->cmdcnt = psmouse->pktcnt = 0;
        if (psmouse->reconnect) {
               if (psmouse->reconnect(psmouse))
-                       return -1;
-       } else if (psmouse_probe(psmouse) != old_type)
-               return -1;
+                       goto out;
+       } else if (psmouse_probe(psmouse) < 0 ||
+                  psmouse->type != psmouse_extensions(psmouse, psmouse_max_proto, 0))
+               goto out;
 
        /* ok, the device type (and capabilities) match the old one,
         * we can continue using it, complete intialization
         */
-       psmouse->type = old_type;
+       psmouse_set_state(psmouse, PSMOUSE_CMD_MODE);
+
        psmouse_initialize(psmouse);
 
-       if (psmouse->ptport) {
-                       if (psmouse_reconnect(&psmouse->ptport->serio)) {
-                       __serio_unregister_port(&psmouse->ptport->serio);
-                       __serio_register_port(&psmouse->ptport->serio);
-                       if (psmouse->ptport->activate)
-                               psmouse->ptport->activate(psmouse);
-               }
-       }
+       if (parent && parent->pt_activate)
+               parent->pt_activate(parent);
 
        psmouse_activate(psmouse);
-       return 0;
+       rc = 0;
+
+out:
+       /* If this is a pass-through port the parent waits to be activated */
+       if (parent)
+               psmouse_activate(parent);
+
+       return rc;
 }
 
 
-static struct serio_dev psmouse_dev = {
-       .interrupt =    psmouse_interrupt,
-       .connect =      psmouse_connect,
-       .reconnect =    psmouse_reconnect,
-       .disconnect =   psmouse_disconnect,
-       .cleanup =      psmouse_cleanup,
+static struct serio_driver psmouse_drv = {
+       .driver         = {
+               .name   = "psmouse",
+       },
+       .description    = DRIVER_DESC,
+       .interrupt      = psmouse_interrupt,
+       .connect        = psmouse_connect,
+       .reconnect      = psmouse_reconnect,
+       .disconnect     = psmouse_disconnect,
+       .cleanup        = psmouse_cleanup,
 };
 
 static inline void psmouse_parse_proto(void)
@@ -730,13 +903,13 @@ static inline void psmouse_parse_proto(void)
 int __init psmouse_init(void)
 {
        psmouse_parse_proto();
-       serio_register_device(&psmouse_dev);
+       serio_register_driver(&psmouse_drv);
        return 0;
 }
 
 void __exit psmouse_exit(void)
 {
-       serio_unregister_device(&psmouse_dev);
+       serio_unregister_driver(&psmouse_drv);
 }
 
 module_init(psmouse_init);