// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * driver for Earthsoft PT1/PT2
 *
 * Copyright (C) 2009 HIRANO Takahito <hiranotaka@zng.info>
 *
 * based on pt1dvr - http://pt1dvr.sourceforge.jp/
 *      by Tomoaki Ishikawa <tomy@users.sourceforge.jp>
 */

#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/sched/signal.h>
#include <linux/hrtimer.h>
#include <linux/delay.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/vmalloc.h>
#include <linux/pci.h>
#include <linux/kthread.h>
#include <linux/freezer.h>
#include <linux/ratelimit.h>
#include <linux/string.h>
#include <linux/i2c.h>

#include <media/dvbdev.h>
#include <media/dvb_demux.h>
#include <media/dmxdev.h>
#include <media/dvb_net.h>
#include <media/dvb_frontend.h>

#include "tc90522.h"
#include "qm1d1b0004.h"
#include "dvb-pll.h"

#define DRIVER_NAME "earth-pt1"

#define PT1_PAGE_SHIFT 12
#define PT1_PAGE_SIZE (1 << PT1_PAGE_SHIFT)
#define PT1_NR_UPACKETS 1024
#define PT1_NR_BUFS 511

struct pt1_buffer_page {
        __le32 upackets[PT1_NR_UPACKETS];
};

struct pt1_table_page {
        __le32 next_pfn;
        __le32 buf_pfns[PT1_NR_BUFS];
};

struct pt1_buffer {
        struct pt1_buffer_page *page;
        dma_addr_t addr;
};

struct pt1_table {
        struct pt1_table_page *page;
        dma_addr_t addr;
        struct pt1_buffer bufs[PT1_NR_BUFS];
};

enum pt1_fe_clk {
        PT1_FE_CLK_20MHZ,       /* PT1 */
        PT1_FE_CLK_25MHZ,       /* PT2 */
};

#define PT1_NR_ADAPS 4

struct pt1_adapter;

struct pt1 {
        struct pci_dev *pdev;
        void __iomem *regs;
        struct i2c_adapter i2c_adap;
        int i2c_running;
        struct pt1_adapter *adaps[PT1_NR_ADAPS];
        struct pt1_table *tables;
        struct task_struct *kthread;
        int table_index;
        int buf_index;

        struct mutex lock;
        int power;
        int reset;

        enum pt1_fe_clk fe_clk;
};

struct pt1_adapter {
        struct pt1 *pt1;
        int index;

        u8 *buf;
        int upacket_count;
        int packet_count;
        int st_count;

        struct dvb_adapter adap;
        struct dvb_demux demux;
        int users;
        struct dmxdev dmxdev;
        struct dvb_frontend *fe;
        struct i2c_client *demod_i2c_client;
        struct i2c_client *tuner_i2c_client;
        int (*orig_set_voltage)(struct dvb_frontend *fe,
                                enum fe_sec_voltage voltage);
        int (*orig_sleep)(struct dvb_frontend *fe);
        int (*orig_init)(struct dvb_frontend *fe);

        enum fe_sec_voltage voltage;
        int sleep;
};

union pt1_tuner_config {
        struct qm1d1b0004_config qm1d1b0004;
        struct dvb_pll_config tda6651;
};

struct pt1_config {
        struct i2c_board_info demod_info;
        struct tc90522_config demod_cfg;

        struct i2c_board_info tuner_info;
        union pt1_tuner_config tuner_cfg;
};

static const struct pt1_config pt1_configs[PT1_NR_ADAPS] = {
        {
                .demod_info = {
                        I2C_BOARD_INFO(TC90522_I2C_DEV_SAT, 0x1b),
                },
                .tuner_info = {
                        I2C_BOARD_INFO("qm1d1b0004", 0x60),
                },
        },
        {
                .demod_info = {
                        I2C_BOARD_INFO(TC90522_I2C_DEV_TER, 0x1a),
                },
                .tuner_info = {
                        I2C_BOARD_INFO("tda665x_earthpt1", 0x61),
                },
        },
        {
                .demod_info = {
                        I2C_BOARD_INFO(TC90522_I2C_DEV_SAT, 0x19),
                },
                .tuner_info = {
                        I2C_BOARD_INFO("qm1d1b0004", 0x60),
                },
        },
        {
                .demod_info = {
                        I2C_BOARD_INFO(TC90522_I2C_DEV_TER, 0x18),
                },
                .tuner_info = {
                        I2C_BOARD_INFO("tda665x_earthpt1", 0x61),
                },
        },
};

static const u8 va1j5jf8007s_20mhz_configs[][2] = {
        {0x04, 0x02}, {0x0d, 0x55}, {0x11, 0x40}, {0x13, 0x80}, {0x17, 0x01},
        {0x1c, 0x0a}, {0x1d, 0xaa}, {0x1e, 0x20}, {0x1f, 0x88}, {0x51, 0xb0},
        {0x52, 0x89}, {0x53, 0xb3}, {0x5a, 0x2d}, {0x5b, 0xd3}, {0x85, 0x69},
        {0x87, 0x04}, {0x8e, 0x02}, {0xa3, 0xf7}, {0xa5, 0xc0},
};

static const u8 va1j5jf8007s_25mhz_configs[][2] = {
        {0x04, 0x02}, {0x11, 0x40}, {0x13, 0x80}, {0x17, 0x01}, {0x1c, 0x0a},
        {0x1d, 0xaa}, {0x1e, 0x20}, {0x1f, 0x88}, {0x51, 0xb0}, {0x52, 0x89},
        {0x53, 0xb3}, {0x5a, 0x2d}, {0x5b, 0xd3}, {0x85, 0x69}, {0x87, 0x04},
        {0x8e, 0x26}, {0xa3, 0xf7}, {0xa5, 0xc0},
};

static const u8 va1j5jf8007t_20mhz_configs[][2] = {
        {0x03, 0x90}, {0x14, 0x8f}, {0x1c, 0x2a}, {0x1d, 0xa8}, {0x1e, 0xa2},
        {0x22, 0x83}, {0x31, 0x0d}, {0x32, 0xe0}, {0x39, 0xd3}, {0x3a, 0x00},
        {0x3b, 0x11}, {0x3c, 0x3f},
        {0x5c, 0x40}, {0x5f, 0x80}, {0x75, 0x02}, {0x76, 0x4e}, {0x77, 0x03},
        {0xef, 0x01}
};

static const u8 va1j5jf8007t_25mhz_configs[][2] = {
        {0x03, 0x90}, {0x1c, 0x2a}, {0x1d, 0xa8}, {0x1e, 0xa2}, {0x22, 0x83},
        {0x3a, 0x04}, {0x3b, 0x11}, {0x3c, 0x3f}, {0x5c, 0x40}, {0x5f, 0x80},
        {0x75, 0x0a}, {0x76, 0x4c}, {0x77, 0x03}, {0xef, 0x01}
};

static int config_demod(struct i2c_client *cl, enum pt1_fe_clk clk)
{
        int ret;
        bool is_sat;
        const u8 (*cfg_data)[2];
        int i, len;

        is_sat = !strncmp(cl->name, TC90522_I2C_DEV_SAT,
                          strlen(TC90522_I2C_DEV_SAT));
        if (is_sat) {
                struct i2c_msg msg[2];
                u8 wbuf, rbuf;

                wbuf = 0x07;
                msg[0].addr = cl->addr;
                msg[0].flags = 0;
                msg[0].len = 1;
                msg[0].buf = &wbuf;

                msg[1].addr = cl->addr;
                msg[1].flags = I2C_M_RD;
                msg[1].len = 1;
                msg[1].buf = &rbuf;
                ret = i2c_transfer(cl->adapter, msg, 2);
                if (ret < 0)
                        return ret;
                if (rbuf != 0x41)
                        return -EIO;
        }

        /* frontend init */
        if (clk == PT1_FE_CLK_20MHZ) {
                if (is_sat) {
                        cfg_data = va1j5jf8007s_20mhz_configs;
                        len = ARRAY_SIZE(va1j5jf8007s_20mhz_configs);
                } else {
                        cfg_data = va1j5jf8007t_20mhz_configs;
                        len = ARRAY_SIZE(va1j5jf8007t_20mhz_configs);
                }
        } else {
                if (is_sat) {
                        cfg_data = va1j5jf8007s_25mhz_configs;
                        len = ARRAY_SIZE(va1j5jf8007s_25mhz_configs);
                } else {
                        cfg_data = va1j5jf8007t_25mhz_configs;
                        len = ARRAY_SIZE(va1j5jf8007t_25mhz_configs);
                }
        }

        for (i = 0; i < len; i++) {
                ret = i2c_master_send(cl, cfg_data[i], 2);
                if (ret < 0)
                        return ret;
        }
        return 0;
}

/*
 * Init registers for (each pair of) terrestrial/satellite block in demod.
 * Note that resetting terr. block also resets its peer sat. block as well.
 * This function must be called before configuring any demod block
 * (before pt1_wakeup(), fe->ops.init()).
 */
static int pt1_demod_block_init(struct pt1 *pt1)
{
        struct i2c_client *cl;
        u8 buf[2] = {0x01, 0x80};
        int ret;
        int i;

        /* reset all terr. & sat. pairs first */
        for (i = 0; i < PT1_NR_ADAPS; i++) {
                cl = pt1->adaps[i]->demod_i2c_client;
                if (strncmp(cl->name, TC90522_I2C_DEV_TER,
                            strlen(TC90522_I2C_DEV_TER)))
                        continue;

                ret = i2c_master_send(cl, buf, 2);
                if (ret < 0)
                        return ret;
                usleep_range(30000, 50000);
        }

        for (i = 0; i < PT1_NR_ADAPS; i++) {
                cl = pt1->adaps[i]->demod_i2c_client;
                if (strncmp(cl->name, TC90522_I2C_DEV_SAT,
                            strlen(TC90522_I2C_DEV_SAT)))
                        continue;

                ret = i2c_master_send(cl, buf, 2);
                if (ret < 0)
                        return ret;
                usleep_range(30000, 50000);
        }
        return 0;
}

static void pt1_write_reg(struct pt1 *pt1, int reg, u32 data)
{
        writel(data, pt1->regs + reg * 4);
}

static u32 pt1_read_reg(struct pt1 *pt1, int reg)
{
        return readl(pt1->regs + reg * 4);
}

static unsigned int pt1_nr_tables = 8;
module_param_named(nr_tables, pt1_nr_tables, uint, 0);

static void pt1_increment_table_count(struct pt1 *pt1)
{
        pt1_write_reg(pt1, 0, 0x00000020);
}

static void pt1_init_table_count(struct pt1 *pt1)
{
        pt1_write_reg(pt1, 0, 0x00000010);
}

static void pt1_register_tables(struct pt1 *pt1, u32 first_pfn)
{
        pt1_write_reg(pt1, 5, first_pfn);
        pt1_write_reg(pt1, 0, 0x0c000040);
}

static void pt1_unregister_tables(struct pt1 *pt1)
{
        pt1_write_reg(pt1, 0, 0x08080000);
}

static int pt1_sync(struct pt1 *pt1)
{
        int i;
        for (i = 0; i < 57; i++) {
                if (pt1_read_reg(pt1, 0) & 0x20000000)
                        return 0;
                pt1_write_reg(pt1, 0, 0x00000008);
        }
        dev_err(&pt1->pdev->dev, "could not sync\n");
        return -EIO;
}

static u64 pt1_identify(struct pt1 *pt1)
{
        int i;
        u64 id = 0;
        for (i = 0; i < 57; i++) {
                id |= (u64)(pt1_read_reg(pt1, 0) >> 30 & 1) << i;
                pt1_write_reg(pt1, 0, 0x00000008);
        }
        return id;
}

static int pt1_unlock(struct pt1 *pt1)
{
        int i;
        pt1_write_reg(pt1, 0, 0x00000008);
        for (i = 0; i < 3; i++) {
                if (pt1_read_reg(pt1, 0) & 0x80000000)
                        return 0;
                usleep_range(1000, 2000);
        }
        dev_err(&pt1->pdev->dev, "could not unlock\n");
        return -EIO;
}

static int pt1_reset_pci(struct pt1 *pt1)
{
        int i;
        pt1_write_reg(pt1, 0, 0x01010000);
        pt1_write_reg(pt1, 0, 0x01000000);
        for (i = 0; i < 10; i++) {
                if (pt1_read_reg(pt1, 0) & 0x00000001)
                        return 0;
                usleep_range(1000, 2000);
        }
        dev_err(&pt1->pdev->dev, "could not reset PCI\n");
        return -EIO;
}

static int pt1_reset_ram(struct pt1 *pt1)
{
        int i;
        pt1_write_reg(pt1, 0, 0x02020000);
        pt1_write_reg(pt1, 0, 0x02000000);
        for (i = 0; i < 10; i++) {
                if (pt1_read_reg(pt1, 0) & 0x00000002)
                        return 0;
                usleep_range(1000, 2000);
        }
        dev_err(&pt1->pdev->dev, "could not reset RAM\n");
        return -EIO;
}

static int pt1_do_enable_ram(struct pt1 *pt1)
{
        int i, j;
        u32 status;
        status = pt1_read_reg(pt1, 0) & 0x00000004;
        pt1_write_reg(pt1, 0, 0x00000002);
        for (i = 0; i < 10; i++) {
                for (j = 0; j < 1024; j++) {
                        if ((pt1_read_reg(pt1, 0) & 0x00000004) != status)
                                return 0;
                }
                usleep_range(1000, 2000);
        }
        dev_err(&pt1->pdev->dev, "could not enable RAM\n");
        return -EIO;
}

static int pt1_enable_ram(struct pt1 *pt1)
{
        int i, ret;
        int phase;
        usleep_range(1000, 2000);
        phase = pt1->pdev->device == 0x211a ? 128 : 166;
        for (i = 0; i < phase; i++) {
                ret = pt1_do_enable_ram(pt1);
                if (ret < 0)
                        return ret;
        }
        return 0;
}

static void pt1_disable_ram(struct pt1 *pt1)
{
        pt1_write_reg(pt1, 0, 0x0b0b0000);
}

static void pt1_set_stream(struct pt1 *pt1, int index, int enabled)
{
        pt1_write_reg(pt1, 2, 1 << (index + 8) | enabled << index);
}

static void pt1_init_streams(struct pt1 *pt1)
{
        int i;
        for (i = 0; i < PT1_NR_ADAPS; i++)
                pt1_set_stream(pt1, i, 0);
}

static int pt1_filter(struct pt1 *pt1, struct pt1_buffer_page *page)
{
        u32 upacket;
        int i;
        int index;
        struct pt1_adapter *adap;
        int offset;
        u8 *buf;
        int sc;

        if (!page->upackets[PT1_NR_UPACKETS - 1])
                return 0;

        for (i = 0; i < PT1_NR_UPACKETS; i++) {
                upacket = le32_to_cpu(page->upackets[i]);
                index = (upacket >> 29) - 1;
                if (index < 0 || index >=  PT1_NR_ADAPS)
                        continue;

                adap = pt1->adaps[index];
                if (upacket >> 25 & 1)
                        adap->upacket_count = 0;
                else if (!adap->upacket_count)
                        continue;

                if (upacket >> 24 & 1)
                        printk_ratelimited(KERN_INFO "earth-pt1: device buffer overflowing. table[%d] buf[%d]\n",
                                pt1->table_index, pt1->buf_index);
                sc = upacket >> 26 & 0x7;
                if (adap->st_count != -1 && sc != ((adap->st_count + 1) & 0x7))
                        printk_ratelimited(KERN_INFO "earth-pt1: data loss in streamID(adapter)[%d]\n",
                                           index);
                adap->st_count = sc;

                buf = adap->buf;
                offset = adap->packet_count * 188 + adap->upacket_count * 3;
                buf[offset] = upacket >> 16;
                buf[offset + 1] = upacket >> 8;
                if (adap->upacket_count != 62)
                        buf[offset + 2] = upacket;

                if (++adap->upacket_count >= 63) {
                        adap->upacket_count = 0;
                        if (++adap->packet_count >= 21) {
                                dvb_dmx_swfilter_packets(&adap->demux, buf, 21);
                                adap->packet_count = 0;
                        }
                }
        }

        page->upackets[PT1_NR_UPACKETS - 1] = 0;
        return 1;
}

static int pt1_thread(void *data)
{
        struct pt1 *pt1;
        struct pt1_buffer_page *page;
        bool was_frozen;

#define PT1_FETCH_DELAY 10
#define PT1_FETCH_DELAY_DELTA 2

        pt1 = data;
        set_freezable();

        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)) {
                        ktime_t delay;

                        delay = ktime_set(0, PT1_FETCH_DELAY * NSEC_PER_MSEC);
                        set_current_state(TASK_INTERRUPTIBLE);
                        schedule_hrtimeout_range(&delay,
                                        PT1_FETCH_DELAY_DELTA * NSEC_PER_MSEC,
                                        HRTIMER_MODE_REL);
                        continue;
                }

                if (++pt1->buf_index >= PT1_NR_BUFS) {
                        pt1_increment_table_count(pt1);
                        pt1->buf_index = 0;
                        if (++pt1->table_index >= pt1_nr_tables)
                                pt1->table_index = 0;
                }
        }

        return 0;
}

static void pt1_free_page(struct pt1 *pt1, void *page, dma_addr_t addr)
{
        dma_free_coherent(&pt1->pdev->dev, PT1_PAGE_SIZE, page, addr);
}

static void *pt1_alloc_page(struct pt1 *pt1, dma_addr_t *addrp, u32 *pfnp)
{
        void *page;
        dma_addr_t addr;

        page = dma_alloc_coherent(&pt1->pdev->dev, PT1_PAGE_SIZE, &addr,
                                  GFP_KERNEL);
        if (page == NULL)
                return NULL;

        BUG_ON(addr & (PT1_PAGE_SIZE - 1));
        BUG_ON(addr >> PT1_PAGE_SHIFT >> 31 >> 1);

        *addrp = addr;
        *pfnp = addr >> PT1_PAGE_SHIFT;
        return page;
}

static void pt1_cleanup_buffer(struct pt1 *pt1, struct pt1_buffer *buf)
{
        pt1_free_page(pt1, buf->page, buf->addr);
}

static int
pt1_init_buffer(struct pt1 *pt1, struct pt1_buffer *buf,  u32 *pfnp)
{
        struct pt1_buffer_page *page;
        dma_addr_t addr;

        page = pt1_alloc_page(pt1, &addr, pfnp);
        if (page == NULL)
                return -ENOMEM;

        page->upackets[PT1_NR_UPACKETS - 1] = 0;

        buf->page = page;
        buf->addr = addr;
        return 0;
}

static void pt1_cleanup_table(struct pt1 *pt1, struct pt1_table *table)
{
        int i;

        for (i = 0; i < PT1_NR_BUFS; i++)
                pt1_cleanup_buffer(pt1, &table->bufs[i]);

        pt1_free_page(pt1, table->page, table->addr);
}

static int
pt1_init_table(struct pt1 *pt1, struct pt1_table *table, u32 *pfnp)
{
        struct pt1_table_page *page;
        dma_addr_t addr;
        int i, ret;
        u32 buf_pfn;

        page = pt1_alloc_page(pt1, &addr, pfnp);
        if (page == NULL)
                return -ENOMEM;

        for (i = 0; i < PT1_NR_BUFS; i++) {
                ret = pt1_init_buffer(pt1, &table->bufs[i], &buf_pfn);
                if (ret < 0)
                        goto err;

                page->buf_pfns[i] = cpu_to_le32(buf_pfn);
        }

        pt1_increment_table_count(pt1);
        table->page = page;
        table->addr = addr;
        return 0;

err:
        while (i--)
                pt1_cleanup_buffer(pt1, &table->bufs[i]);

        pt1_free_page(pt1, page, addr);
        return ret;
}

static void pt1_cleanup_tables(struct pt1 *pt1)
{
        struct pt1_table *tables;
        int i;

        tables = pt1->tables;
        pt1_unregister_tables(pt1);

        for (i = 0; i < pt1_nr_tables; i++)
                pt1_cleanup_table(pt1, &tables[i]);

        vfree(tables);
}

static int pt1_init_tables(struct pt1 *pt1)
{
        struct pt1_table *tables;
        int i, ret;
        u32 first_pfn, pfn;

        if (!pt1_nr_tables)
                return 0;

        tables = vmalloc_array(pt1_nr_tables, sizeof(struct pt1_table));
        if (tables == NULL)
                return -ENOMEM;

        pt1_init_table_count(pt1);

        i = 0;
        ret = pt1_init_table(pt1, &tables[0], &first_pfn);
        if (ret)
                goto err;
        i++;

        while (i < pt1_nr_tables) {
                ret = pt1_init_table(pt1, &tables[i], &pfn);
                if (ret)
                        goto err;
                tables[i - 1].page->next_pfn = cpu_to_le32(pfn);
                i++;
        }

        tables[pt1_nr_tables - 1].page->next_pfn = cpu_to_le32(first_pfn);

        pt1_register_tables(pt1, first_pfn);
        pt1->tables = tables;
        return 0;

err:
        while (i--)
                pt1_cleanup_table(pt1, &tables[i]);

        vfree(tables);
        return ret;
}

static int pt1_start_polling(struct pt1 *pt1)
{
        int ret = 0;

        mutex_lock(&pt1->lock);
        if (!pt1->kthread) {
                pt1->kthread = kthread_run(pt1_thread, pt1, "earth-pt1");
                if (IS_ERR(pt1->kthread)) {
                        ret = PTR_ERR(pt1->kthread);
                        pt1->kthread = NULL;
                }
        }
        mutex_unlock(&pt1->lock);
        return ret;
}

static int pt1_start_feed(struct dvb_demux_feed *feed)
{
        struct pt1_adapter *adap;
        adap = container_of(feed->demux, struct pt1_adapter, demux);
        if (!adap->users++) {
                int ret;

                ret = pt1_start_polling(adap->pt1);
                if (ret)
                        return ret;
                pt1_set_stream(adap->pt1, adap->index, 1);
        }
        return 0;
}

static void pt1_stop_polling(struct pt1 *pt1)
{
        int i, count;

        mutex_lock(&pt1->lock);
        for (i = 0, count = 0; i < PT1_NR_ADAPS; i++)
                count += pt1->adaps[i]->users;

        if (count == 0 && pt1->kthread) {
                kthread_stop(pt1->kthread);
                pt1->kthread = NULL;
        }
        mutex_unlock(&pt1->lock);
}

static int pt1_stop_feed(struct dvb_demux_feed *feed)
{
        struct pt1_adapter *adap;
        adap = container_of(feed->demux, struct pt1_adapter, demux);
        if (!--adap->users) {
                pt1_set_stream(adap->pt1, adap->index, 0);
                pt1_stop_polling(adap->pt1);
        }
        return 0;
}

static void
pt1_update_power(struct pt1 *pt1)
{
        int bits;
        int i;
        struct pt1_adapter *adap;
        static const int sleep_bits[] = {
                1 << 4,
                1 << 6 | 1 << 7,
                1 << 5,
                1 << 6 | 1 << 8,
        };

        bits = pt1->power | !pt1->reset << 3;
        mutex_lock(&pt1->lock);
        for (i = 0; i < PT1_NR_ADAPS; i++) {
                adap = pt1->adaps[i];
                switch (adap->voltage) {
                case SEC_VOLTAGE_13: /* actually 11V */
                        bits |= 1 << 2;
                        break;
                case SEC_VOLTAGE_18: /* actually 15V */
                        bits |= 1 << 1 | 1 << 2;
                        break;
                default:
                        break;
                }

                /* XXX: The bits should be changed depending on adap->sleep. */
                bits |= sleep_bits[i];
        }
        pt1_write_reg(pt1, 1, bits);
        mutex_unlock(&pt1->lock);
}

static int pt1_set_voltage(struct dvb_frontend *fe, enum fe_sec_voltage voltage)
{
        struct pt1_adapter *adap;

        adap = container_of(fe->dvb, struct pt1_adapter, adap);
        adap->voltage = voltage;
        pt1_update_power(adap->pt1);

        if (adap->orig_set_voltage)
                return adap->orig_set_voltage(fe, voltage);
        else
                return 0;
}

static int pt1_sleep(struct dvb_frontend *fe)
{
        struct pt1_adapter *adap;
        int ret;

        adap = container_of(fe->dvb, struct pt1_adapter, adap);

        ret = 0;
        if (adap->orig_sleep)
                ret = adap->orig_sleep(fe);

        adap->sleep = 1;
        pt1_update_power(adap->pt1);
        return ret;
}

static int pt1_wakeup(struct dvb_frontend *fe)
{
        struct pt1_adapter *adap;
        int ret;

        adap = container_of(fe->dvb, struct pt1_adapter, adap);
        adap->sleep = 0;
        pt1_update_power(adap->pt1);
        usleep_range(1000, 2000);

        ret = config_demod(adap->demod_i2c_client, adap->pt1->fe_clk);
        if (ret == 0 && adap->orig_init)
                ret = adap->orig_init(fe);
        return ret;
}

static void pt1_free_adapter(struct pt1_adapter *adap)
{
        adap->demux.dmx.close(&adap->demux.dmx);
        dvb_dmxdev_release(&adap->dmxdev);
        dvb_dmx_release(&adap->demux);
        dvb_unregister_adapter(&adap->adap);
        free_page((unsigned long)adap->buf);
        kfree(adap);
}

DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);

static struct pt1_adapter *
pt1_alloc_adapter(struct pt1 *pt1)
{
        struct pt1_adapter *adap;
        void *buf;
        struct dvb_adapter *dvb_adap;
        struct dvb_demux *demux;
        struct dmxdev *dmxdev;
        int ret;

        adap = kzalloc_obj(struct pt1_adapter);
        if (!adap) {
                ret = -ENOMEM;
                goto err;
        }

        adap->pt1 = pt1;

        adap->voltage = SEC_VOLTAGE_OFF;
        adap->sleep = 1;

        buf = (u8 *)__get_free_page(GFP_KERNEL);
        if (!buf) {
                ret = -ENOMEM;
                goto err_kfree;
        }

        adap->buf = buf;
        adap->upacket_count = 0;
        adap->packet_count = 0;
        adap->st_count = -1;

        dvb_adap = &adap->adap;
        dvb_adap->priv = adap;
        ret = dvb_register_adapter(dvb_adap, DRIVER_NAME, THIS_MODULE,
                                   &pt1->pdev->dev, adapter_nr);
        if (ret < 0)
                goto err_free_page;

        demux = &adap->demux;
        demux->dmx.capabilities = DMX_TS_FILTERING | DMX_SECTION_FILTERING;
        demux->priv = adap;
        demux->feednum = 256;
        demux->filternum = 256;
        demux->start_feed = pt1_start_feed;
        demux->stop_feed = pt1_stop_feed;
        demux->write_to_decoder = NULL;
        ret = dvb_dmx_init(demux);
        if (ret < 0)
                goto err_unregister_adapter;

        dmxdev = &adap->dmxdev;
        dmxdev->filternum = 256;
        dmxdev->demux = &demux->dmx;
        dmxdev->capabilities = 0;
        ret = dvb_dmxdev_init(dmxdev, dvb_adap);
        if (ret < 0)
                goto err_dmx_release;

        return adap;

err_dmx_release:
        dvb_dmx_release(demux);
err_unregister_adapter:
        dvb_unregister_adapter(dvb_adap);
err_free_page:
        free_page((unsigned long)buf);
err_kfree:
        kfree(adap);
err:
        return ERR_PTR(ret);
}

static void pt1_cleanup_adapters(struct pt1 *pt1)
{
        int i;
        for (i = 0; i < PT1_NR_ADAPS; i++)
                pt1_free_adapter(pt1->adaps[i]);
}

static int pt1_init_adapters(struct pt1 *pt1)
{
        int i;
        struct pt1_adapter *adap;
        int ret;

        for (i = 0; i < PT1_NR_ADAPS; i++) {
                adap = pt1_alloc_adapter(pt1);
                if (IS_ERR(adap)) {
                        ret = PTR_ERR(adap);
                        goto err;
                }

                adap->index = i;
                pt1->adaps[i] = adap;
        }
        return 0;

err:
        while (i--)
                pt1_free_adapter(pt1->adaps[i]);

        return ret;
}

static void pt1_cleanup_frontend(struct pt1_adapter *adap)
{
        dvb_unregister_frontend(adap->fe);
        dvb_module_release(adap->tuner_i2c_client);
        dvb_module_release(adap->demod_i2c_client);
}

static int pt1_init_frontend(struct pt1_adapter *adap, struct dvb_frontend *fe)
{
        int ret;

        adap->orig_set_voltage = fe->ops.set_voltage;
        adap->orig_sleep = fe->ops.sleep;
        adap->orig_init = fe->ops.init;
        fe->ops.set_voltage = pt1_set_voltage;
        fe->ops.sleep = pt1_sleep;
        fe->ops.init = pt1_wakeup;

        ret = dvb_register_frontend(&adap->adap, fe);
        if (ret < 0)
                return ret;

        adap->fe = fe;
        return 0;
}

static void pt1_cleanup_frontends(struct pt1 *pt1)
{
        int i;
        for (i = 0; i < PT1_NR_ADAPS; i++)
                pt1_cleanup_frontend(pt1->adaps[i]);
}

static int pt1_init_frontends(struct pt1 *pt1)
{
        int i;
        int ret;

        for (i = 0; i < ARRAY_SIZE(pt1_configs); i++) {
                const struct i2c_board_info *info;
                struct tc90522_config dcfg;
                struct i2c_client *cl;

                info = &pt1_configs[i].demod_info;
                dcfg = pt1_configs[i].demod_cfg;
                dcfg.tuner_i2c = NULL;

                ret = -ENODEV;
                cl = dvb_module_probe("tc90522", info->type, &pt1->i2c_adap,
                                      info->addr, &dcfg);
                if (!cl)
                        goto fe_unregister;
                pt1->adaps[i]->demod_i2c_client = cl;

                if (!strncmp(cl->name, TC90522_I2C_DEV_SAT,
                             strlen(TC90522_I2C_DEV_SAT))) {
                        struct qm1d1b0004_config tcfg;

                        info = &pt1_configs[i].tuner_info;
                        tcfg = pt1_configs[i].tuner_cfg.qm1d1b0004;
                        tcfg.fe = dcfg.fe;
                        cl = dvb_module_probe("qm1d1b0004",
                                              info->type, dcfg.tuner_i2c,
                                              info->addr, &tcfg);
                } else {
                        struct dvb_pll_config tcfg;

                        info = &pt1_configs[i].tuner_info;
                        tcfg = pt1_configs[i].tuner_cfg.tda6651;
                        tcfg.fe = dcfg.fe;
                        cl = dvb_module_probe("dvb_pll",
                                              info->type, dcfg.tuner_i2c,
                                              info->addr, &tcfg);
                }
                if (!cl)
                        goto demod_release;
                pt1->adaps[i]->tuner_i2c_client = cl;

                ret = pt1_init_frontend(pt1->adaps[i], dcfg.fe);
                if (ret < 0)
                        goto tuner_release;
        }

        ret = pt1_demod_block_init(pt1);
        if (ret < 0)
                goto fe_unregister;

        return 0;

tuner_release:
        dvb_module_release(pt1->adaps[i]->tuner_i2c_client);
demod_release:
        dvb_module_release(pt1->adaps[i]->demod_i2c_client);
fe_unregister:
        dev_warn(&pt1->pdev->dev, "failed to init FE(%d).\n", i);
        i--;
        for (; i >= 0; i--) {
                dvb_unregister_frontend(pt1->adaps[i]->fe);
                dvb_module_release(pt1->adaps[i]->tuner_i2c_client);
                dvb_module_release(pt1->adaps[i]->demod_i2c_client);
        }
        return ret;
}

static void pt1_i2c_emit(struct pt1 *pt1, int addr, int busy, int read_enable,
                         int clock, int data, int next_addr)
{
        pt1_write_reg(pt1, 4, addr << 18 | busy << 13 | read_enable << 12 |
                      !clock << 11 | !data << 10 | next_addr);
}

static void pt1_i2c_write_bit(struct pt1 *pt1, int addr, int *addrp, int data)
{
        pt1_i2c_emit(pt1, addr,     1, 0, 0, data, addr + 1);
        pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, data, addr + 2);
        pt1_i2c_emit(pt1, addr + 2, 1, 0, 0, data, addr + 3);
        *addrp = addr + 3;
}

static void pt1_i2c_read_bit(struct pt1 *pt1, int addr, int *addrp)
{
        pt1_i2c_emit(pt1, addr,     1, 0, 0, 1, addr + 1);
        pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 1, addr + 2);
        pt1_i2c_emit(pt1, addr + 2, 1, 1, 1, 1, addr + 3);
        pt1_i2c_emit(pt1, addr + 3, 1, 0, 0, 1, addr + 4);
        *addrp = addr + 4;
}

static void pt1_i2c_write_byte(struct pt1 *pt1, int addr, int *addrp, int data)
{
        int i;
        for (i = 0; i < 8; i++)
                pt1_i2c_write_bit(pt1, addr, &addr, data >> (7 - i) & 1);
        pt1_i2c_write_bit(pt1, addr, &addr, 1);
        *addrp = addr;
}

static void pt1_i2c_read_byte(struct pt1 *pt1, int addr, int *addrp, int last)
{
        int i;
        for (i = 0; i < 8; i++)
                pt1_i2c_read_bit(pt1, addr, &addr);
        pt1_i2c_write_bit(pt1, addr, &addr, last);
        *addrp = addr;
}

static void pt1_i2c_prepare(struct pt1 *pt1, int addr, int *addrp)
{
        pt1_i2c_emit(pt1, addr,     1, 0, 1, 1, addr + 1);
        pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 0, addr + 2);
        pt1_i2c_emit(pt1, addr + 2, 1, 0, 0, 0, addr + 3);
        *addrp = addr + 3;
}

static void
pt1_i2c_write_msg(struct pt1 *pt1, int addr, int *addrp, struct i2c_msg *msg)
{
        int i;
        pt1_i2c_prepare(pt1, addr, &addr);
        pt1_i2c_write_byte(pt1, addr, &addr, msg->addr << 1);
        for (i = 0; i < msg->len; i++)
                pt1_i2c_write_byte(pt1, addr, &addr, msg->buf[i]);
        *addrp = addr;
}

static void
pt1_i2c_read_msg(struct pt1 *pt1, int addr, int *addrp, struct i2c_msg *msg)
{
        int i;
        pt1_i2c_prepare(pt1, addr, &addr);
        pt1_i2c_write_byte(pt1, addr, &addr, msg->addr << 1 | 1);
        for (i = 0; i < msg->len; i++)
                pt1_i2c_read_byte(pt1, addr, &addr, i == msg->len - 1);
        *addrp = addr;
}

static int pt1_i2c_end(struct pt1 *pt1, int addr)
{
        pt1_i2c_emit(pt1, addr,     1, 0, 0, 0, addr + 1);
        pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 0, addr + 2);
        pt1_i2c_emit(pt1, addr + 2, 1, 0, 1, 1, 0);

        pt1_write_reg(pt1, 0, 0x00000004);
        do {
                if (signal_pending(current))
                        return -EINTR;
                usleep_range(1000, 2000);
        } while (pt1_read_reg(pt1, 0) & 0x00000080);
        return 0;
}

static void pt1_i2c_begin(struct pt1 *pt1, int *addrp)
{
        int addr = 0;

        pt1_i2c_emit(pt1, addr,     0, 0, 1, 1, addr /* itself */);
        addr = addr + 1;

        if (!pt1->i2c_running) {
                pt1_i2c_emit(pt1, addr,     1, 0, 1, 1, addr + 1);
                pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 0, addr + 2);
                addr = addr + 2;
                pt1->i2c_running = 1;
        }
        *addrp = addr;
}

static int pt1_i2c_xfer(struct i2c_adapter *adap, struct i2c_msg *msgs, int num)
{
        struct pt1 *pt1;
        int i;
        struct i2c_msg *msg, *next_msg;
        int addr, ret;
        u16 len;
        u32 word;

        pt1 = i2c_get_adapdata(adap);

        for (i = 0; i < num; i++) {
                msg = &msgs[i];
                if (msg->flags & I2C_M_RD)
                        return -ENOTSUPP;

                if (i + 1 < num)
                        next_msg = &msgs[i + 1];
                else
                        next_msg = NULL;

                if (next_msg && next_msg->flags & I2C_M_RD) {
                        i++;

                        len = next_msg->len;
                        if (len > 4)
                                return -ENOTSUPP;

                        pt1_i2c_begin(pt1, &addr);
                        pt1_i2c_write_msg(pt1, addr, &addr, msg);
                        pt1_i2c_read_msg(pt1, addr, &addr, next_msg);
                        ret = pt1_i2c_end(pt1, addr);
                        if (ret < 0)
                                return ret;

                        word = pt1_read_reg(pt1, 2);
                        while (len--) {
                                next_msg->buf[len] = word;
                                word >>= 8;
                        }
                } else {
                        pt1_i2c_begin(pt1, &addr);
                        pt1_i2c_write_msg(pt1, addr, &addr, msg);
                        ret = pt1_i2c_end(pt1, addr);
                        if (ret < 0)
                                return ret;
                }
        }

        return num;
}

static u32 pt1_i2c_func(struct i2c_adapter *adap)
{
        return I2C_FUNC_I2C;
}

static const struct i2c_algorithm pt1_i2c_algo = {
        .master_xfer = pt1_i2c_xfer,
        .functionality = pt1_i2c_func,
};

static void pt1_i2c_wait(struct pt1 *pt1)
{
        int i;
        for (i = 0; i < 128; i++)
                pt1_i2c_emit(pt1, 0, 0, 0, 1, 1, 0);
}

static void pt1_i2c_init(struct pt1 *pt1)
{
        int i;
        for (i = 0; i < 1024; i++)
                pt1_i2c_emit(pt1, i, 0, 0, 1, 1, 0);
}

#ifdef CONFIG_PM_SLEEP

static int pt1_suspend(struct device *dev)
{
        struct pt1 *pt1 = dev_get_drvdata(dev);

        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 pt1 *pt1 = dev_get_drvdata(dev);
        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);

        ret = pt1_demod_block_init(pt1);
        if (ret < 0)
                goto resume_err;

        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;
        void __iomem *regs;

        pt1 = pci_get_drvdata(pdev);
        regs = pt1->regs;

        if (pt1->kthread)
                kthread_stop(pt1->kthread);
        pt1_cleanup_tables(pt1);
        pt1_cleanup_frontends(pt1);
        pt1_disable_ram(pt1);
        pt1->power = 0;
        pt1->reset = 1;
        pt1_update_power(pt1);
        pt1_cleanup_adapters(pt1);
        i2c_del_adapter(&pt1->i2c_adap);
        kfree(pt1);
        pci_iounmap(pdev, regs);
        pci_release_regions(pdev);
        pci_disable_device(pdev);
}

static int pt1_probe(struct pci_dev *pdev, const struct pci_device_id *ent)
{
        int ret;
        void __iomem *regs;
        struct pt1 *pt1;
        struct i2c_adapter *i2c_adap;

        ret = pci_enable_device(pdev);
        if (ret < 0)
                goto err;

        ret = dma_set_mask(&pdev->dev, DMA_BIT_MASK(32));
        if (ret < 0)
                goto err_pci_disable_device;

        pci_set_master(pdev);

        ret = pci_request_regions(pdev, DRIVER_NAME);
        if (ret < 0)
                goto err_pci_disable_device;

        regs = pci_iomap(pdev, 0, 0);
        if (!regs) {
                ret = -EIO;
                goto err_pci_release_regions;
        }

        pt1 = kzalloc_obj(struct pt1);
        if (!pt1) {
                ret = -ENOMEM;
                goto err_pci_iounmap;
        }

        mutex_init(&pt1->lock);
        pt1->pdev = pdev;
        pt1->regs = regs;
        pt1->fe_clk = (pdev->device == 0x211a) ?
                                PT1_FE_CLK_20MHZ : PT1_FE_CLK_25MHZ;
        pci_set_drvdata(pdev, pt1);

        ret = pt1_init_adapters(pt1);
        if (ret < 0)
                goto err_kfree;

        mutex_init(&pt1->lock);

        pt1->power = 0;
        pt1->reset = 1;
        pt1_update_power(pt1);

        i2c_adap = &pt1->i2c_adap;
        i2c_adap->algo = &pt1_i2c_algo;
        i2c_adap->algo_data = NULL;
        i2c_adap->dev.parent = &pdev->dev;
        strscpy(i2c_adap->name, DRIVER_NAME, sizeof(i2c_adap->name));
        i2c_set_adapdata(i2c_adap, pt1);
        ret = i2c_add_adapter(i2c_adap);
        if (ret < 0)
                goto err_pt1_cleanup_adapters;

        pt1_i2c_init(pt1);
        pt1_i2c_wait(pt1);

        ret = pt1_sync(pt1);
        if (ret < 0)
                goto err_i2c_del_adapter;

        pt1_identify(pt1);

        ret = pt1_unlock(pt1);
        if (ret < 0)
                goto err_i2c_del_adapter;

        ret = pt1_reset_pci(pt1);
        if (ret < 0)
                goto err_i2c_del_adapter;

        ret = pt1_reset_ram(pt1);
        if (ret < 0)
                goto err_i2c_del_adapter;

        ret = pt1_enable_ram(pt1);
        if (ret < 0)
                goto err_i2c_del_adapter;

        pt1_init_streams(pt1);

        pt1->power = 1;
        pt1_update_power(pt1);
        msleep(20);

        pt1->reset = 0;
        pt1_update_power(pt1);
        usleep_range(1000, 2000);

        ret = pt1_init_frontends(pt1);
        if (ret < 0)
                goto err_pt1_disable_ram;

        ret = pt1_init_tables(pt1);
        if (ret < 0)
                goto err_pt1_cleanup_frontends;

        return 0;

err_pt1_cleanup_frontends:
        pt1_cleanup_frontends(pt1);
err_pt1_disable_ram:
        pt1_disable_ram(pt1);
        pt1->power = 0;
        pt1->reset = 1;
        pt1_update_power(pt1);
err_i2c_del_adapter:
        i2c_del_adapter(i2c_adap);
err_pt1_cleanup_adapters:
        pt1_cleanup_adapters(pt1);
err_kfree:
        kfree(pt1);
err_pci_iounmap:
        pci_iounmap(pdev, regs);
err_pci_release_regions:
        pci_release_regions(pdev);
err_pci_disable_device:
        pci_disable_device(pdev);
err:
        return ret;

}

static const struct pci_device_id pt1_id_table[] = {
        { PCI_DEVICE(0x10ee, 0x211a) },
        { PCI_DEVICE(0x10ee, 0x222a) },
        { },
};
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,
        .driver.pm      = &pt1_pm_ops,
};

module_pci_driver(pt1_driver);

MODULE_AUTHOR("Takahito HIRANO <hiranotaka@zng.info>");
MODULE_DESCRIPTION("Earthsoft PT1/PT2 Driver");
MODULE_LICENSE("GPL");