media: dvb: earth-pt1: add support for suspend/resume

Without this patch, re-loading of the module was required after resume.

Signed-off-by: Akihiro Tsukada <tskd08@gmail.com>
Signed-off-by: Mauro Carvalho Chehab <mchehab+samsung@kernel.org>

diff --git a/drivers/media/pci/pt1/pt1.c b/drivers/media/pci/pt1/pt1.c
index 40b6c0ac34226f548895f14656d07dec73292a33..b169175d85e22e9d2a6a842dbf87ab142c3a6c1e 100644 (file)
--- a/drivers/media/pci/pt1/pt1.c
+++ b/drivers/media/pci/pt1/pt1.c
@@ -461,12 +461,18 @@ static int pt1_thread(void *data)
 {
        struct pt1 *pt1;
        struct pt1_buffer_page *page;
+       bool was_frozen;
 
        pt1 = data;
        set_freezable();
 
-       while (!kthread_should_stop()) {
-               try_to_freeze();
+       while (!kthread_freezable_should_stop(&was_frozen)) {
+               if (was_frozen) {
+                       int i;
+
+                       for (i = 0; i < PT1_NR_ADAPS; i++)
+                               pt1_set_stream(pt1, i, !!pt1->adaps[i]->users);
+               }
 
                page = pt1->tables[pt1->table_index].bufs[pt1->buf_index].page;
                if (!pt1_filter(pt1, page)) {
@@ -1165,6 +1171,98 @@ static void pt1_i2c_init(struct pt1 *pt1)
                pt1_i2c_emit(pt1, i, 0, 0, 1, 1, 0);
 }
 
+#ifdef CONFIG_PM_SLEEP
+
+static int pt1_suspend(struct device *dev)
+{
+       struct pci_dev *pdev = to_pci_dev(dev);
+       struct pt1 *pt1 = pci_get_drvdata(pdev);
+
+       pt1_init_streams(pt1);
+       pt1_disable_ram(pt1);
+       pt1->power = 0;
+       pt1->reset = 1;
+       pt1_update_power(pt1);
+       return 0;
+}
+
+static int pt1_resume(struct device *dev)
+{
+       struct pci_dev *pdev = to_pci_dev(dev);
+       struct pt1 *pt1 = pci_get_drvdata(pdev);
+       int ret;
+       int i;
+
+       pt1->power = 0;
+       pt1->reset = 1;
+       pt1_update_power(pt1);
+
+       pt1_i2c_init(pt1);
+       pt1_i2c_wait(pt1);
+
+       ret = pt1_sync(pt1);
+       if (ret < 0)
+               goto resume_err;
+
+       pt1_identify(pt1);
+
+       ret = pt1_unlock(pt1);
+       if (ret < 0)
+               goto resume_err;
+
+       ret = pt1_reset_pci(pt1);
+       if (ret < 0)
+               goto resume_err;
+
+       ret = pt1_reset_ram(pt1);
+       if (ret < 0)
+               goto resume_err;
+
+       ret = pt1_enable_ram(pt1);
+       if (ret < 0)
+               goto resume_err;
+
+       pt1_init_streams(pt1);
+
+       pt1->power = 1;
+       pt1_update_power(pt1);
+       msleep(20);
+
+       pt1->reset = 0;
+       pt1_update_power(pt1);
+       usleep_range(1000, 2000);
+
+       for (i = 0; i < PT1_NR_ADAPS; i++)
+               dvb_frontend_reinitialise(pt1->adaps[i]->fe);
+
+       pt1_init_table_count(pt1);
+       for (i = 0; i < pt1_nr_tables; i++) {
+               int j;
+
+               for (j = 0; j < PT1_NR_BUFS; j++)
+                       pt1->tables[i].bufs[j].page->upackets[PT1_NR_UPACKETS-1]
+                               = 0;
+               pt1_increment_table_count(pt1);
+       }
+       pt1_register_tables(pt1, pt1->tables[0].addr >> PT1_PAGE_SHIFT);
+
+       pt1->table_index = 0;
+       pt1->buf_index = 0;
+       for (i = 0; i < PT1_NR_ADAPS; i++) {
+               pt1->adaps[i]->upacket_count = 0;
+               pt1->adaps[i]->packet_count = 0;
+               pt1->adaps[i]->st_count = -1;
+       }
+
+       return 0;
+
+resume_err:
+       dev_info(&pt1->pdev->dev, "failed to resume PT1/PT2.");
+       return 0;       /* resume anyway */
+}
+
+#endif /* CONFIG_PM_SLEEP */
+
 static void pt1_remove(struct pci_dev *pdev)
 {
        struct pt1 *pt1;
@@ -1325,11 +1423,16 @@ static const struct pci_device_id pt1_id_table[] = {
 };
 MODULE_DEVICE_TABLE(pci, pt1_id_table);
 
+static SIMPLE_DEV_PM_OPS(pt1_pm_ops, pt1_suspend, pt1_resume);
+
 static struct pci_driver pt1_driver = {
        .name           = DRIVER_NAME,
        .probe          = pt1_probe,
        .remove         = pt1_remove,
        .id_table       = pt1_id_table,
+#if CONFIG_PM_SLEEP
+       .driver.pm      = &pt1_pm_ops,
+#endif
 };
 
 module_pci_driver(pt1_driver);