1 files changed, 68 insertions, 21 deletions

diff --git a/drivers/power/pda_power.c b/drivers/power/pda_power.c
index 69f8aa3a6a4..fd49689738a 100644
--- a/drivers/power/pda_power.c
+++ b/drivers/power/pda_power.c
@@ -14,6 +14,7 @@
 #include <linux/platform_device.h>
 #include <linux/err.h>
 #include <linux/interrupt.h>
+#include <linux/notifier.h>
 #include <linux/power_supply.h>
 #include <linux/pda_power.h>
 #include <linux/regulator/consumer.h>
@@ -40,7 +41,9 @@ static int polling;
 
 #ifdef CONFIG_USB_OTG_UTILS
 static struct otg_transceiver *transceiver;
+static struct notifier_block otg_nb;
 #endif
+
 static struct regulator *ac_draw;
 
 enum {
@@ -222,7 +225,42 @@ static void polling_timer_func(unsigned long unused)
 #ifdef CONFIG_USB_OTG_UTILS
 static int otg_is_usb_online(void)
 {
-        return (transceiver->state == OTG_STATE_B_PERIPHERAL);
+        return (transceiver->last_event == USB_EVENT_VBUS ||
+                transceiver->last_event == USB_EVENT_ENUMERATED);
+}
+
+static int otg_is_ac_online(void)
+{
+        return (transceiver->last_event == USB_EVENT_CHARGER);
+}
+
+static int otg_handle_notification(struct notifier_block *nb,
+                unsigned long event, void *unused)
+{
+        switch (event) {
+        case USB_EVENT_CHARGER:
+                ac_status = PDA_PSY_TO_CHANGE;
+                break;
+        case USB_EVENT_VBUS:
+        case USB_EVENT_ENUMERATED:
+                usb_status = PDA_PSY_TO_CHANGE;
+                break;
+        case USB_EVENT_NONE:
+                ac_status = PDA_PSY_TO_CHANGE;
+                usb_status = PDA_PSY_TO_CHANGE;
+                break;
+        default:
+                return NOTIFY_OK;
+        }
+
+        /*
+         * Wait a bit before reading ac/usb line status and setting charger,
+         * because ac/usb status readings may lag from irq.
+         */
+        mod_timer(&charger_timer,
+                  jiffies + msecs_to_jiffies(pdata->wait_for_status));
+
+        return NOTIFY_OK;
 }
 #endif
 
@@ -282,6 +320,16 @@ static int pda_power_probe(struct platform_device *pdev)
                 ret = PTR_ERR(ac_draw);
         }
 
+#ifdef CONFIG_USB_OTG_UTILS
+        transceiver = otg_get_transceiver();
+        if (transceiver && !pdata->is_usb_online) {
+                pdata->is_usb_online = otg_is_usb_online;
+        }
+        if (transceiver && !pdata->is_ac_online) {
+                pdata->is_ac_online = otg_is_ac_online;
+        }
+#endif
+
         if (pdata->is_ac_online) {
                 ret = power_supply_register(&pdev->dev, &pda_psy_ac);
                 if (ret) {
@@ -303,13 +351,6 @@ static int pda_power_probe(struct platform_device *pdev)
                 }
         }
 
-#ifdef CONFIG_USB_OTG_UTILS
-        transceiver = otg_get_transceiver();
-        if (transceiver && !pdata->is_usb_online) {
-                pdata->is_usb_online = otg_is_usb_online;
-        }
-#endif
-
         if (pdata->is_usb_online) {
                 ret = power_supply_register(&pdev->dev, &pda_psy_usb);
                 if (ret) {
@@ -331,6 +372,18 @@ static int pda_power_probe(struct platform_device *pdev)
                 }
         }
 
+#ifdef CONFIG_USB_OTG_UTILS
+        if (transceiver && pdata->use_otg_notifier) {
+                otg_nb.notifier_call = otg_handle_notification;
+                ret = otg_register_notifier(transceiver, &otg_nb);
+                if (ret) {
+                        dev_err(dev, "failure to register otg notifier\n");
+                        goto otg_reg_notifier_failed;
+                }
+                polling = 0;
+        }
+#endif
+
         if (polling) {
                 dev_dbg(dev, "will poll for status\n");
                 setup_timer(&polling_timer, polling_timer_func, 0);
@@ -343,6 +396,11 @@ static int pda_power_probe(struct platform_device *pdev)
 
         return 0;
 
+#ifdef CONFIG_USB_OTG_UTILS
+otg_reg_notifier_failed:
+        if (pdata->is_usb_online && usb_irq)
+                free_irq(usb_irq->start, &pda_psy_usb);
+#endif
 usb_irq_failed:
         if (pdata->is_usb_online)
                 power_supply_unregister(&pda_psy_usb);
@@ -440,8 +498,6 @@ static int pda_power_resume(struct platform_device *pdev)
 #define pda_power_resume NULL
 #endif /* CONFIG_PM */
 
-MODULE_ALIAS("platform:pda-power");
-
 static struct platform_driver pda_power_pdrv = {
         .driver = {
                 .name = "pda-power",
@@ -452,17 +508,8 @@ static struct platform_driver pda_power_pdrv = {
         .resume = pda_power_resume,
 };
 
-static int __init pda_power_init(void)
-{
-        return platform_driver_register(&pda_power_pdrv);
-}
+module_platform_driver(pda_power_pdrv);
 
-static void __exit pda_power_exit(void)
-{
-        platform_driver_unregister(&pda_power_pdrv);
-}
-
-module_init(pda_power_init);
-module_exit(pda_power_exit);
 MODULE_LICENSE("GPL");
 MODULE_AUTHOR("Anton Vorontsov <cbou@mail.ru>");
+MODULE_ALIAS("platform:pda-power");