/*
  *  Copyright (C) 2009-2010, Lars-Peter Clausen <lars@metafoo.de>
+ *  Copyright (C) 2010, Paul Cercueil <paul@crapouillou.net>
  *      JZ4740 SoC RTC driver
  *
  *  This program is free software; you can redistribute it and/or modify it
 
        ret = jz4740_rtc_reg_write(rtc, JZ_REG_RTC_SEC_ALARM, secs);
        if (!ret)
-               ret = jz4740_rtc_ctrl_set_bits(rtc, JZ_RTC_CTRL_AE, alrm->enabled);
+               ret = jz4740_rtc_ctrl_set_bits(rtc,
+                       JZ_RTC_CTRL_AE | JZ_RTC_CTRL_AF_IRQ, alrm->enabled);
 
        return ret;
 }
 
        platform_set_drvdata(pdev, rtc);
 
+       device_init_wakeup(&pdev->dev, 1);
+
        rtc->rtc = rtc_device_register(pdev->name, &pdev->dev, &jz4740_rtc_ops,
                                        THIS_MODULE);
        if (IS_ERR(rtc->rtc)) {
        return 0;
 }
 
+
+#ifdef CONFIG_PM
+static int jz4740_rtc_suspend(struct device *dev)
+{
+       struct jz4740_rtc *rtc = dev_get_drvdata(dev);
+
+       if (device_may_wakeup(dev))
+               enable_irq_wake(rtc->irq);
+       return 0;
+}
+
+static int jz4740_rtc_resume(struct device *dev)
+{
+       struct jz4740_rtc *rtc = dev_get_drvdata(dev);
+
+       if (device_may_wakeup(dev))
+               disable_irq_wake(rtc->irq);
+       return 0;
+}
+
+static const struct dev_pm_ops jz4740_pm_ops = {
+       .suspend = jz4740_rtc_suspend,
+       .resume  = jz4740_rtc_resume,
+};
+#define JZ4740_RTC_PM_OPS (&jz4740_pm_ops)
+
+#else
+#define JZ4740_RTC_PM_OPS NULL
+#endif  /* CONFIG_PM */
+
 struct platform_driver jz4740_rtc_driver = {
-       .probe = jz4740_rtc_probe,
-       .remove = __devexit_p(jz4740_rtc_remove),
-       .driver = {
-               .name = "jz4740-rtc",
+       .probe   = jz4740_rtc_probe,
+       .remove  = __devexit_p(jz4740_rtc_remove),
+       .driver  = {
+               .name  = "jz4740-rtc",
                .owner = THIS_MODULE,
+               .pm    = JZ4740_RTC_PM_OPS,
        },
 };