1/*
2 * driver for Earthsoft PT1/PT2
3 *
4 * Copyright (C) 2009 HIRANO Takahito <hiranotaka@zng.info>
5 *
6 * based on pt1dvr - http://pt1dvr.sourceforge.jp/
7 * 	by Tomoaki Ishikawa <tomy@users.sourceforge.jp>
8 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 */
23
24#include <linux/kernel.h>
25#include <linux/module.h>
26#include <linux/slab.h>
27#include <linux/vmalloc.h>
28#include <linux/pci.h>
29#include <linux/kthread.h>
30#include <linux/freezer.h>
31#include <linux/ratelimit.h>
32
33#include "dvbdev.h"
34#include "dvb_demux.h"
35#include "dmxdev.h"
36#include "dvb_net.h"
37#include "dvb_frontend.h"
38
39#include "va1j5jf8007t.h"
40#include "va1j5jf8007s.h"
41
42#define DRIVER_NAME "earth-pt1"
43
44#define PT1_PAGE_SHIFT 12
45#define PT1_PAGE_SIZE (1 << PT1_PAGE_SHIFT)
46#define PT1_NR_UPACKETS 1024
47#define PT1_NR_BUFS 511
48
49struct pt1_buffer_page {
50	__le32 upackets[PT1_NR_UPACKETS];
51};
52
53struct pt1_table_page {
54	__le32 next_pfn;
55	__le32 buf_pfns[PT1_NR_BUFS];
56};
57
58struct pt1_buffer {
59	struct pt1_buffer_page *page;
60	dma_addr_t addr;
61};
62
63struct pt1_table {
64	struct pt1_table_page *page;
65	dma_addr_t addr;
66	struct pt1_buffer bufs[PT1_NR_BUFS];
67};
68
69#define PT1_NR_ADAPS 4
70
71struct pt1_adapter;
72
73struct pt1 {
74	struct pci_dev *pdev;
75	void __iomem *regs;
76	struct i2c_adapter i2c_adap;
77	int i2c_running;
78	struct pt1_adapter *adaps[PT1_NR_ADAPS];
79	struct pt1_table *tables;
80	struct task_struct *kthread;
81	int table_index;
82	int buf_index;
83
84	struct mutex lock;
85	int power;
86	int reset;
87};
88
89struct pt1_adapter {
90	struct pt1 *pt1;
91	int index;
92
93	u8 *buf;
94	int upacket_count;
95	int packet_count;
96	int st_count;
97
98	struct dvb_adapter adap;
99	struct dvb_demux demux;
100	int users;
101	struct dmxdev dmxdev;
102	struct dvb_frontend *fe;
103	int (*orig_set_voltage)(struct dvb_frontend *fe,
104				fe_sec_voltage_t voltage);
105	int (*orig_sleep)(struct dvb_frontend *fe);
106	int (*orig_init)(struct dvb_frontend *fe);
107
108	fe_sec_voltage_t voltage;
109	int sleep;
110};
111
112static void pt1_write_reg(struct pt1 *pt1, int reg, u32 data)
113{
114	writel(data, pt1->regs + reg * 4);
115}
116
117static u32 pt1_read_reg(struct pt1 *pt1, int reg)
118{
119	return readl(pt1->regs + reg * 4);
120}
121
122static int pt1_nr_tables = 8;
123module_param_named(nr_tables, pt1_nr_tables, int, 0);
124
125static void pt1_increment_table_count(struct pt1 *pt1)
126{
127	pt1_write_reg(pt1, 0, 0x00000020);
128}
129
130static void pt1_init_table_count(struct pt1 *pt1)
131{
132	pt1_write_reg(pt1, 0, 0x00000010);
133}
134
135static void pt1_register_tables(struct pt1 *pt1, u32 first_pfn)
136{
137	pt1_write_reg(pt1, 5, first_pfn);
138	pt1_write_reg(pt1, 0, 0x0c000040);
139}
140
141static void pt1_unregister_tables(struct pt1 *pt1)
142{
143	pt1_write_reg(pt1, 0, 0x08080000);
144}
145
146static int pt1_sync(struct pt1 *pt1)
147{
148	int i;
149	for (i = 0; i < 57; i++) {
150		if (pt1_read_reg(pt1, 0) & 0x20000000)
151			return 0;
152		pt1_write_reg(pt1, 0, 0x00000008);
153	}
154	dev_err(&pt1->pdev->dev, "could not sync\n");
155	return -EIO;
156}
157
158static u64 pt1_identify(struct pt1 *pt1)
159{
160	int i;
161	u64 id;
162	id = 0;
163	for (i = 0; i < 57; i++) {
164		id |= (u64)(pt1_read_reg(pt1, 0) >> 30 & 1) << i;
165		pt1_write_reg(pt1, 0, 0x00000008);
166	}
167	return id;
168}
169
170static int pt1_unlock(struct pt1 *pt1)
171{
172	int i;
173	pt1_write_reg(pt1, 0, 0x00000008);
174	for (i = 0; i < 3; i++) {
175		if (pt1_read_reg(pt1, 0) & 0x80000000)
176			return 0;
177		schedule_timeout_uninterruptible((HZ + 999) / 1000);
178	}
179	dev_err(&pt1->pdev->dev, "could not unlock\n");
180	return -EIO;
181}
182
183static int pt1_reset_pci(struct pt1 *pt1)
184{
185	int i;
186	pt1_write_reg(pt1, 0, 0x01010000);
187	pt1_write_reg(pt1, 0, 0x01000000);
188	for (i = 0; i < 10; i++) {
189		if (pt1_read_reg(pt1, 0) & 0x00000001)
190			return 0;
191		schedule_timeout_uninterruptible((HZ + 999) / 1000);
192	}
193	dev_err(&pt1->pdev->dev, "could not reset PCI\n");
194	return -EIO;
195}
196
197static int pt1_reset_ram(struct pt1 *pt1)
198{
199	int i;
200	pt1_write_reg(pt1, 0, 0x02020000);
201	pt1_write_reg(pt1, 0, 0x02000000);
202	for (i = 0; i < 10; i++) {
203		if (pt1_read_reg(pt1, 0) & 0x00000002)
204			return 0;
205		schedule_timeout_uninterruptible((HZ + 999) / 1000);
206	}
207	dev_err(&pt1->pdev->dev, "could not reset RAM\n");
208	return -EIO;
209}
210
211static int pt1_do_enable_ram(struct pt1 *pt1)
212{
213	int i, j;
214	u32 status;
215	status = pt1_read_reg(pt1, 0) & 0x00000004;
216	pt1_write_reg(pt1, 0, 0x00000002);
217	for (i = 0; i < 10; i++) {
218		for (j = 0; j < 1024; j++) {
219			if ((pt1_read_reg(pt1, 0) & 0x00000004) != status)
220				return 0;
221		}
222		schedule_timeout_uninterruptible((HZ + 999) / 1000);
223	}
224	dev_err(&pt1->pdev->dev, "could not enable RAM\n");
225	return -EIO;
226}
227
228static int pt1_enable_ram(struct pt1 *pt1)
229{
230	int i, ret;
231	int phase;
232	schedule_timeout_uninterruptible((HZ + 999) / 1000);
233	phase = pt1->pdev->device == 0x211a ? 128 : 166;
234	for (i = 0; i < phase; i++) {
235		ret = pt1_do_enable_ram(pt1);
236		if (ret < 0)
237			return ret;
238	}
239	return 0;
240}
241
242static void pt1_disable_ram(struct pt1 *pt1)
243{
244	pt1_write_reg(pt1, 0, 0x0b0b0000);
245}
246
247static void pt1_set_stream(struct pt1 *pt1, int index, int enabled)
248{
249	pt1_write_reg(pt1, 2, 1 << (index + 8) | enabled << index);
250}
251
252static void pt1_init_streams(struct pt1 *pt1)
253{
254	int i;
255	for (i = 0; i < PT1_NR_ADAPS; i++)
256		pt1_set_stream(pt1, i, 0);
257}
258
259static int pt1_filter(struct pt1 *pt1, struct pt1_buffer_page *page)
260{
261	u32 upacket;
262	int i;
263	int index;
264	struct pt1_adapter *adap;
265	int offset;
266	u8 *buf;
267	int sc;
268
269	if (!page->upackets[PT1_NR_UPACKETS - 1])
270		return 0;
271
272	for (i = 0; i < PT1_NR_UPACKETS; i++) {
273		upacket = le32_to_cpu(page->upackets[i]);
274		index = (upacket >> 29) - 1;
275		if (index < 0 || index >=  PT1_NR_ADAPS)
276			continue;
277
278		adap = pt1->adaps[index];
279		if (upacket >> 25 & 1)
280			adap->upacket_count = 0;
281		else if (!adap->upacket_count)
282			continue;
283
284		if (upacket >> 24 & 1)
285			printk_ratelimited(KERN_INFO "earth-pt1: device "
286				"buffer overflowing. table[%d] buf[%d]\n",
287				pt1->table_index, pt1->buf_index);
288		sc = upacket >> 26 & 0x7;
289		if (adap->st_count != -1 && sc != ((adap->st_count + 1) & 0x7))
290			printk_ratelimited(KERN_INFO "earth-pt1: data loss"
291				" in streamID(adapter)[%d]\n", index);
292		adap->st_count = sc;
293
294		buf = adap->buf;
295		offset = adap->packet_count * 188 + adap->upacket_count * 3;
296		buf[offset] = upacket >> 16;
297		buf[offset + 1] = upacket >> 8;
298		if (adap->upacket_count != 62)
299			buf[offset + 2] = upacket;
300
301		if (++adap->upacket_count >= 63) {
302			adap->upacket_count = 0;
303			if (++adap->packet_count >= 21) {
304				dvb_dmx_swfilter_packets(&adap->demux, buf, 21);
305				adap->packet_count = 0;
306			}
307		}
308	}
309
310	page->upackets[PT1_NR_UPACKETS - 1] = 0;
311	return 1;
312}
313
314static int pt1_thread(void *data)
315{
316	struct pt1 *pt1;
317	struct pt1_buffer_page *page;
318
319	pt1 = data;
320	set_freezable();
321
322	while (!kthread_should_stop()) {
323		try_to_freeze();
324
325		page = pt1->tables[pt1->table_index].bufs[pt1->buf_index].page;
326		if (!pt1_filter(pt1, page)) {
327			schedule_timeout_interruptible((HZ + 999) / 1000);
328			continue;
329		}
330
331		if (++pt1->buf_index >= PT1_NR_BUFS) {
332			pt1_increment_table_count(pt1);
333			pt1->buf_index = 0;
334			if (++pt1->table_index >= pt1_nr_tables)
335				pt1->table_index = 0;
336		}
337	}
338
339	return 0;
340}
341
342static void pt1_free_page(struct pt1 *pt1, void *page, dma_addr_t addr)
343{
344	dma_free_coherent(&pt1->pdev->dev, PT1_PAGE_SIZE, page, addr);
345}
346
347static void *pt1_alloc_page(struct pt1 *pt1, dma_addr_t *addrp, u32 *pfnp)
348{
349	void *page;
350	dma_addr_t addr;
351
352	page = dma_alloc_coherent(&pt1->pdev->dev, PT1_PAGE_SIZE, &addr,
353				  GFP_KERNEL);
354	if (page == NULL)
355		return NULL;
356
357	BUG_ON(addr & (PT1_PAGE_SIZE - 1));
358	BUG_ON(addr >> PT1_PAGE_SHIFT >> 31 >> 1);
359
360	*addrp = addr;
361	*pfnp = addr >> PT1_PAGE_SHIFT;
362	return page;
363}
364
365static void pt1_cleanup_buffer(struct pt1 *pt1, struct pt1_buffer *buf)
366{
367	pt1_free_page(pt1, buf->page, buf->addr);
368}
369
370static int
371pt1_init_buffer(struct pt1 *pt1, struct pt1_buffer *buf,  u32 *pfnp)
372{
373	struct pt1_buffer_page *page;
374	dma_addr_t addr;
375
376	page = pt1_alloc_page(pt1, &addr, pfnp);
377	if (page == NULL)
378		return -ENOMEM;
379
380	page->upackets[PT1_NR_UPACKETS - 1] = 0;
381
382	buf->page = page;
383	buf->addr = addr;
384	return 0;
385}
386
387static void pt1_cleanup_table(struct pt1 *pt1, struct pt1_table *table)
388{
389	int i;
390
391	for (i = 0; i < PT1_NR_BUFS; i++)
392		pt1_cleanup_buffer(pt1, &table->bufs[i]);
393
394	pt1_free_page(pt1, table->page, table->addr);
395}
396
397static int
398pt1_init_table(struct pt1 *pt1, struct pt1_table *table, u32 *pfnp)
399{
400	struct pt1_table_page *page;
401	dma_addr_t addr;
402	int i, ret;
403	u32 buf_pfn;
404
405	page = pt1_alloc_page(pt1, &addr, pfnp);
406	if (page == NULL)
407		return -ENOMEM;
408
409	for (i = 0; i < PT1_NR_BUFS; i++) {
410		ret = pt1_init_buffer(pt1, &table->bufs[i], &buf_pfn);
411		if (ret < 0)
412			goto err;
413
414		page->buf_pfns[i] = cpu_to_le32(buf_pfn);
415	}
416
417	pt1_increment_table_count(pt1);
418	table->page = page;
419	table->addr = addr;
420	return 0;
421
422err:
423	while (i--)
424		pt1_cleanup_buffer(pt1, &table->bufs[i]);
425
426	pt1_free_page(pt1, page, addr);
427	return ret;
428}
429
430static void pt1_cleanup_tables(struct pt1 *pt1)
431{
432	struct pt1_table *tables;
433	int i;
434
435	tables = pt1->tables;
436	pt1_unregister_tables(pt1);
437
438	for (i = 0; i < pt1_nr_tables; i++)
439		pt1_cleanup_table(pt1, &tables[i]);
440
441	vfree(tables);
442}
443
444static int pt1_init_tables(struct pt1 *pt1)
445{
446	struct pt1_table *tables;
447	int i, ret;
448	u32 first_pfn, pfn;
449
450	tables = vmalloc(sizeof(struct pt1_table) * pt1_nr_tables);
451	if (tables == NULL)
452		return -ENOMEM;
453
454	pt1_init_table_count(pt1);
455
456	i = 0;
457	if (pt1_nr_tables) {
458		ret = pt1_init_table(pt1, &tables[0], &first_pfn);
459		if (ret)
460			goto err;
461		i++;
462	}
463
464	while (i < pt1_nr_tables) {
465		ret = pt1_init_table(pt1, &tables[i], &pfn);
466		if (ret)
467			goto err;
468		tables[i - 1].page->next_pfn = cpu_to_le32(pfn);
469		i++;
470	}
471
472	tables[pt1_nr_tables - 1].page->next_pfn = cpu_to_le32(first_pfn);
473
474	pt1_register_tables(pt1, first_pfn);
475	pt1->tables = tables;
476	return 0;
477
478err:
479	while (i--)
480		pt1_cleanup_table(pt1, &tables[i]);
481
482	vfree(tables);
483	return ret;
484}
485
486static int pt1_start_polling(struct pt1 *pt1)
487{
488	int ret = 0;
489
490	mutex_lock(&pt1->lock);
491	if (!pt1->kthread) {
492		pt1->kthread = kthread_run(pt1_thread, pt1, "earth-pt1");
493		if (IS_ERR(pt1->kthread)) {
494			ret = PTR_ERR(pt1->kthread);
495			pt1->kthread = NULL;
496		}
497	}
498	mutex_unlock(&pt1->lock);
499	return ret;
500}
501
502static int pt1_start_feed(struct dvb_demux_feed *feed)
503{
504	struct pt1_adapter *adap;
505	adap = container_of(feed->demux, struct pt1_adapter, demux);
506	if (!adap->users++) {
507		int ret;
508
509		ret = pt1_start_polling(adap->pt1);
510		if (ret)
511			return ret;
512		pt1_set_stream(adap->pt1, adap->index, 1);
513	}
514	return 0;
515}
516
517static void pt1_stop_polling(struct pt1 *pt1)
518{
519	int i, count;
520
521	mutex_lock(&pt1->lock);
522	for (i = 0, count = 0; i < PT1_NR_ADAPS; i++)
523		count += pt1->adaps[i]->users;
524
525	if (count == 0 && pt1->kthread) {
526		kthread_stop(pt1->kthread);
527		pt1->kthread = NULL;
528	}
529	mutex_unlock(&pt1->lock);
530}
531
532static int pt1_stop_feed(struct dvb_demux_feed *feed)
533{
534	struct pt1_adapter *adap;
535	adap = container_of(feed->demux, struct pt1_adapter, demux);
536	if (!--adap->users) {
537		pt1_set_stream(adap->pt1, adap->index, 0);
538		pt1_stop_polling(adap->pt1);
539	}
540	return 0;
541}
542
543static void
544pt1_update_power(struct pt1 *pt1)
545{
546	int bits;
547	int i;
548	struct pt1_adapter *adap;
549	static const int sleep_bits[] = {
550		1 << 4,
551		1 << 6 | 1 << 7,
552		1 << 5,
553		1 << 6 | 1 << 8,
554	};
555
556	bits = pt1->power | !pt1->reset << 3;
557	mutex_lock(&pt1->lock);
558	for (i = 0; i < PT1_NR_ADAPS; i++) {
559		adap = pt1->adaps[i];
560		switch (adap->voltage) {
561		case SEC_VOLTAGE_13: /* actually 11V */
562			bits |= 1 << 1;
563			break;
564		case SEC_VOLTAGE_18: /* actually 15V */
565			bits |= 1 << 1 | 1 << 2;
566			break;
567		default:
568			break;
569		}
570
571		/* XXX: The bits should be changed depending on adap->sleep. */
572		bits |= sleep_bits[i];
573	}
574	pt1_write_reg(pt1, 1, bits);
575	mutex_unlock(&pt1->lock);
576}
577
578static int pt1_set_voltage(struct dvb_frontend *fe, fe_sec_voltage_t voltage)
579{
580	struct pt1_adapter *adap;
581
582	adap = container_of(fe->dvb, struct pt1_adapter, adap);
583	adap->voltage = voltage;
584	pt1_update_power(adap->pt1);
585
586	if (adap->orig_set_voltage)
587		return adap->orig_set_voltage(fe, voltage);
588	else
589		return 0;
590}
591
592static int pt1_sleep(struct dvb_frontend *fe)
593{
594	struct pt1_adapter *adap;
595
596	adap = container_of(fe->dvb, struct pt1_adapter, adap);
597	adap->sleep = 1;
598	pt1_update_power(adap->pt1);
599
600	if (adap->orig_sleep)
601		return adap->orig_sleep(fe);
602	else
603		return 0;
604}
605
606static int pt1_wakeup(struct dvb_frontend *fe)
607{
608	struct pt1_adapter *adap;
609
610	adap = container_of(fe->dvb, struct pt1_adapter, adap);
611	adap->sleep = 0;
612	pt1_update_power(adap->pt1);
613	schedule_timeout_uninterruptible((HZ + 999) / 1000);
614
615	if (adap->orig_init)
616		return adap->orig_init(fe);
617	else
618		return 0;
619}
620
621static void pt1_free_adapter(struct pt1_adapter *adap)
622{
623	adap->demux.dmx.close(&adap->demux.dmx);
624	dvb_dmxdev_release(&adap->dmxdev);
625	dvb_dmx_release(&adap->demux);
626	dvb_unregister_adapter(&adap->adap);
627	free_page((unsigned long)adap->buf);
628	kfree(adap);
629}
630
631DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);
632
633static struct pt1_adapter *
634pt1_alloc_adapter(struct pt1 *pt1)
635{
636	struct pt1_adapter *adap;
637	void *buf;
638	struct dvb_adapter *dvb_adap;
639	struct dvb_demux *demux;
640	struct dmxdev *dmxdev;
641	int ret;
642
643	adap = kzalloc(sizeof(struct pt1_adapter), GFP_KERNEL);
644	if (!adap) {
645		ret = -ENOMEM;
646		goto err;
647	}
648
649	adap->pt1 = pt1;
650
651	adap->voltage = SEC_VOLTAGE_OFF;
652	adap->sleep = 1;
653
654	buf = (u8 *)__get_free_page(GFP_KERNEL);
655	if (!buf) {
656		ret = -ENOMEM;
657		goto err_kfree;
658	}
659
660	adap->buf = buf;
661	adap->upacket_count = 0;
662	adap->packet_count = 0;
663	adap->st_count = -1;
664
665	dvb_adap = &adap->adap;
666	dvb_adap->priv = adap;
667	ret = dvb_register_adapter(dvb_adap, DRIVER_NAME, THIS_MODULE,
668				   &pt1->pdev->dev, adapter_nr);
669	if (ret < 0)
670		goto err_free_page;
671
672	demux = &adap->demux;
673	demux->dmx.capabilities = DMX_TS_FILTERING | DMX_SECTION_FILTERING;
674	demux->priv = adap;
675	demux->feednum = 256;
676	demux->filternum = 256;
677	demux->start_feed = pt1_start_feed;
678	demux->stop_feed = pt1_stop_feed;
679	demux->write_to_decoder = NULL;
680	ret = dvb_dmx_init(demux);
681	if (ret < 0)
682		goto err_unregister_adapter;
683
684	dmxdev = &adap->dmxdev;
685	dmxdev->filternum = 256;
686	dmxdev->demux = &demux->dmx;
687	dmxdev->capabilities = 0;
688	ret = dvb_dmxdev_init(dmxdev, dvb_adap);
689	if (ret < 0)
690		goto err_dmx_release;
691
692	return adap;
693
694err_dmx_release:
695	dvb_dmx_release(demux);
696err_unregister_adapter:
697	dvb_unregister_adapter(dvb_adap);
698err_free_page:
699	free_page((unsigned long)buf);
700err_kfree:
701	kfree(adap);
702err:
703	return ERR_PTR(ret);
704}
705
706static void pt1_cleanup_adapters(struct pt1 *pt1)
707{
708	int i;
709	for (i = 0; i < PT1_NR_ADAPS; i++)
710		pt1_free_adapter(pt1->adaps[i]);
711}
712
713static int pt1_init_adapters(struct pt1 *pt1)
714{
715	int i;
716	struct pt1_adapter *adap;
717	int ret;
718
719	for (i = 0; i < PT1_NR_ADAPS; i++) {
720		adap = pt1_alloc_adapter(pt1);
721		if (IS_ERR(adap)) {
722			ret = PTR_ERR(adap);
723			goto err;
724		}
725
726		adap->index = i;
727		pt1->adaps[i] = adap;
728	}
729	return 0;
730
731err:
732	while (i--)
733		pt1_free_adapter(pt1->adaps[i]);
734
735	return ret;
736}
737
738static void pt1_cleanup_frontend(struct pt1_adapter *adap)
739{
740	dvb_unregister_frontend(adap->fe);
741}
742
743static int pt1_init_frontend(struct pt1_adapter *adap, struct dvb_frontend *fe)
744{
745	int ret;
746
747	adap->orig_set_voltage = fe->ops.set_voltage;
748	adap->orig_sleep = fe->ops.sleep;
749	adap->orig_init = fe->ops.init;
750	fe->ops.set_voltage = pt1_set_voltage;
751	fe->ops.sleep = pt1_sleep;
752	fe->ops.init = pt1_wakeup;
753
754	ret = dvb_register_frontend(&adap->adap, fe);
755	if (ret < 0)
756		return ret;
757
758	adap->fe = fe;
759	return 0;
760}
761
762static void pt1_cleanup_frontends(struct pt1 *pt1)
763{
764	int i;
765	for (i = 0; i < PT1_NR_ADAPS; i++)
766		pt1_cleanup_frontend(pt1->adaps[i]);
767}
768
769struct pt1_config {
770	struct va1j5jf8007s_config va1j5jf8007s_config;
771	struct va1j5jf8007t_config va1j5jf8007t_config;
772};
773
774static const struct pt1_config pt1_configs[2] = {
775	{
776		{
777			.demod_address = 0x1b,
778			.frequency = VA1J5JF8007S_20MHZ,
779		},
780		{
781			.demod_address = 0x1a,
782			.frequency = VA1J5JF8007T_20MHZ,
783		},
784	}, {
785		{
786			.demod_address = 0x19,
787			.frequency = VA1J5JF8007S_20MHZ,
788		},
789		{
790			.demod_address = 0x18,
791			.frequency = VA1J5JF8007T_20MHZ,
792		},
793	},
794};
795
796static const struct pt1_config pt2_configs[2] = {
797	{
798		{
799			.demod_address = 0x1b,
800			.frequency = VA1J5JF8007S_25MHZ,
801		},
802		{
803			.demod_address = 0x1a,
804			.frequency = VA1J5JF8007T_25MHZ,
805		},
806	}, {
807		{
808			.demod_address = 0x19,
809			.frequency = VA1J5JF8007S_25MHZ,
810		},
811		{
812			.demod_address = 0x18,
813			.frequency = VA1J5JF8007T_25MHZ,
814		},
815	},
816};
817
818static int pt1_init_frontends(struct pt1 *pt1)
819{
820	int i, j;
821	struct i2c_adapter *i2c_adap;
822	const struct pt1_config *configs, *config;
823	struct dvb_frontend *fe[4];
824	int ret;
825
826	i = 0;
827	j = 0;
828
829	i2c_adap = &pt1->i2c_adap;
830	configs = pt1->pdev->device == 0x211a ? pt1_configs : pt2_configs;
831	do {
832		config = &configs[i / 2];
833
834		fe[i] = va1j5jf8007s_attach(&config->va1j5jf8007s_config,
835					    i2c_adap);
836		if (!fe[i]) {
837			ret = -ENODEV; /* This does not sound nice... */
838			goto err;
839		}
840		i++;
841
842		fe[i] = va1j5jf8007t_attach(&config->va1j5jf8007t_config,
843					    i2c_adap);
844		if (!fe[i]) {
845			ret = -ENODEV;
846			goto err;
847		}
848		i++;
849
850		ret = va1j5jf8007s_prepare(fe[i - 2]);
851		if (ret < 0)
852			goto err;
853
854		ret = va1j5jf8007t_prepare(fe[i - 1]);
855		if (ret < 0)
856			goto err;
857
858	} while (i < 4);
859
860	do {
861		ret = pt1_init_frontend(pt1->adaps[j], fe[j]);
862		if (ret < 0)
863			goto err;
864	} while (++j < 4);
865
866	return 0;
867
868err:
869	while (i-- > j)
870		fe[i]->ops.release(fe[i]);
871
872	while (j--)
873		dvb_unregister_frontend(fe[j]);
874
875	return ret;
876}
877
878static void pt1_i2c_emit(struct pt1 *pt1, int addr, int busy, int read_enable,
879			 int clock, int data, int next_addr)
880{
881	pt1_write_reg(pt1, 4, addr << 18 | busy << 13 | read_enable << 12 |
882		      !clock << 11 | !data << 10 | next_addr);
883}
884
885static void pt1_i2c_write_bit(struct pt1 *pt1, int addr, int *addrp, int data)
886{
887	pt1_i2c_emit(pt1, addr,     1, 0, 0, data, addr + 1);
888	pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, data, addr + 2);
889	pt1_i2c_emit(pt1, addr + 2, 1, 0, 0, data, addr + 3);
890	*addrp = addr + 3;
891}
892
893static void pt1_i2c_read_bit(struct pt1 *pt1, int addr, int *addrp)
894{
895	pt1_i2c_emit(pt1, addr,     1, 0, 0, 1, addr + 1);
896	pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 1, addr + 2);
897	pt1_i2c_emit(pt1, addr + 2, 1, 1, 1, 1, addr + 3);
898	pt1_i2c_emit(pt1, addr + 3, 1, 0, 0, 1, addr + 4);
899	*addrp = addr + 4;
900}
901
902static void pt1_i2c_write_byte(struct pt1 *pt1, int addr, int *addrp, int data)
903{
904	int i;
905	for (i = 0; i < 8; i++)
906		pt1_i2c_write_bit(pt1, addr, &addr, data >> (7 - i) & 1);
907	pt1_i2c_write_bit(pt1, addr, &addr, 1);
908	*addrp = addr;
909}
910
911static void pt1_i2c_read_byte(struct pt1 *pt1, int addr, int *addrp, int last)
912{
913	int i;
914	for (i = 0; i < 8; i++)
915		pt1_i2c_read_bit(pt1, addr, &addr);
916	pt1_i2c_write_bit(pt1, addr, &addr, last);
917	*addrp = addr;
918}
919
920static void pt1_i2c_prepare(struct pt1 *pt1, int addr, int *addrp)
921{
922	pt1_i2c_emit(pt1, addr,     1, 0, 1, 1, addr + 1);
923	pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 0, addr + 2);
924	pt1_i2c_emit(pt1, addr + 2, 1, 0, 0, 0, addr + 3);
925	*addrp = addr + 3;
926}
927
928static void
929pt1_i2c_write_msg(struct pt1 *pt1, int addr, int *addrp, struct i2c_msg *msg)
930{
931	int i;
932	pt1_i2c_prepare(pt1, addr, &addr);
933	pt1_i2c_write_byte(pt1, addr, &addr, msg->addr << 1);
934	for (i = 0; i < msg->len; i++)
935		pt1_i2c_write_byte(pt1, addr, &addr, msg->buf[i]);
936	*addrp = addr;
937}
938
939static void
940pt1_i2c_read_msg(struct pt1 *pt1, int addr, int *addrp, struct i2c_msg *msg)
941{
942	int i;
943	pt1_i2c_prepare(pt1, addr, &addr);
944	pt1_i2c_write_byte(pt1, addr, &addr, msg->addr << 1 | 1);
945	for (i = 0; i < msg->len; i++)
946		pt1_i2c_read_byte(pt1, addr, &addr, i == msg->len - 1);
947	*addrp = addr;
948}
949
950static int pt1_i2c_end(struct pt1 *pt1, int addr)
951{
952	pt1_i2c_emit(pt1, addr,     1, 0, 0, 0, addr + 1);
953	pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 0, addr + 2);
954	pt1_i2c_emit(pt1, addr + 2, 1, 0, 1, 1, 0);
955
956	pt1_write_reg(pt1, 0, 0x00000004);
957	do {
958		if (signal_pending(current))
959			return -EINTR;
960		schedule_timeout_interruptible((HZ + 999) / 1000);
961	} while (pt1_read_reg(pt1, 0) & 0x00000080);
962	return 0;
963}
964
965static void pt1_i2c_begin(struct pt1 *pt1, int *addrp)
966{
967	int addr;
968	addr = 0;
969
970	pt1_i2c_emit(pt1, addr,     0, 0, 1, 1, addr /* itself */);
971	addr = addr + 1;
972
973	if (!pt1->i2c_running) {
974		pt1_i2c_emit(pt1, addr,     1, 0, 1, 1, addr + 1);
975		pt1_i2c_emit(pt1, addr + 1, 1, 0, 1, 0, addr + 2);
976		addr = addr + 2;
977		pt1->i2c_running = 1;
978	}
979	*addrp = addr;
980}
981
982static int pt1_i2c_xfer(struct i2c_adapter *adap, struct i2c_msg *msgs, int num)
983{
984	struct pt1 *pt1;
985	int i;
986	struct i2c_msg *msg, *next_msg;
987	int addr, ret;
988	u16 len;
989	u32 word;
990
991	pt1 = i2c_get_adapdata(adap);
992
993	for (i = 0; i < num; i++) {
994		msg = &msgs[i];
995		if (msg->flags & I2C_M_RD)
996			return -ENOTSUPP;
997
998		if (i + 1 < num)
999			next_msg = &msgs[i + 1];
1000		else
1001			next_msg = NULL;
1002
1003		if (next_msg && next_msg->flags & I2C_M_RD) {
1004			i++;
1005
1006			len = next_msg->len;
1007			if (len > 4)
1008				return -ENOTSUPP;
1009
1010			pt1_i2c_begin(pt1, &addr);
1011			pt1_i2c_write_msg(pt1, addr, &addr, msg);
1012			pt1_i2c_read_msg(pt1, addr, &addr, next_msg);
1013			ret = pt1_i2c_end(pt1, addr);
1014			if (ret < 0)
1015				return ret;
1016
1017			word = pt1_read_reg(pt1, 2);
1018			while (len--) {
1019				next_msg->buf[len] = word;
1020				word >>= 8;
1021			}
1022		} else {
1023			pt1_i2c_begin(pt1, &addr);
1024			pt1_i2c_write_msg(pt1, addr, &addr, msg);
1025			ret = pt1_i2c_end(pt1, addr);
1026			if (ret < 0)
1027				return ret;
1028		}
1029	}
1030
1031	return num;
1032}
1033
1034static u32 pt1_i2c_func(struct i2c_adapter *adap)
1035{
1036	return I2C_FUNC_I2C;
1037}
1038
1039static const struct i2c_algorithm pt1_i2c_algo = {
1040	.master_xfer = pt1_i2c_xfer,
1041	.functionality = pt1_i2c_func,
1042};
1043
1044static void pt1_i2c_wait(struct pt1 *pt1)
1045{
1046	int i;
1047	for (i = 0; i < 128; i++)
1048		pt1_i2c_emit(pt1, 0, 0, 0, 1, 1, 0);
1049}
1050
1051static void pt1_i2c_init(struct pt1 *pt1)
1052{
1053	int i;
1054	for (i = 0; i < 1024; i++)
1055		pt1_i2c_emit(pt1, i, 0, 0, 1, 1, 0);
1056}
1057
1058static void pt1_remove(struct pci_dev *pdev)
1059{
1060	struct pt1 *pt1;
1061	void __iomem *regs;
1062
1063	pt1 = pci_get_drvdata(pdev);
1064	regs = pt1->regs;
1065
1066	if (pt1->kthread)
1067		kthread_stop(pt1->kthread);
1068	pt1_cleanup_tables(pt1);
1069	pt1_cleanup_frontends(pt1);
1070	pt1_disable_ram(pt1);
1071	pt1->power = 0;
1072	pt1->reset = 1;
1073	pt1_update_power(pt1);
1074	pt1_cleanup_adapters(pt1);
1075	i2c_del_adapter(&pt1->i2c_adap);
1076	kfree(pt1);
1077	pci_iounmap(pdev, regs);
1078	pci_release_regions(pdev);
1079	pci_disable_device(pdev);
1080}
1081
1082static int pt1_probe(struct pci_dev *pdev, const struct pci_device_id *ent)
1083{
1084	int ret;
1085	void __iomem *regs;
1086	struct pt1 *pt1;
1087	struct i2c_adapter *i2c_adap;
1088
1089	ret = pci_enable_device(pdev);
1090	if (ret < 0)
1091		goto err;
1092
1093	ret = pci_set_dma_mask(pdev, DMA_BIT_MASK(32));
1094	if (ret < 0)
1095		goto err_pci_disable_device;
1096
1097	pci_set_master(pdev);
1098
1099	ret = pci_request_regions(pdev, DRIVER_NAME);
1100	if (ret < 0)
1101		goto err_pci_disable_device;
1102
1103	regs = pci_iomap(pdev, 0, 0);
1104	if (!regs) {
1105		ret = -EIO;
1106		goto err_pci_release_regions;
1107	}
1108
1109	pt1 = kzalloc(sizeof(struct pt1), GFP_KERNEL);
1110	if (!pt1) {
1111		ret = -ENOMEM;
1112		goto err_pci_iounmap;
1113	}
1114
1115	mutex_init(&pt1->lock);
1116	pt1->pdev = pdev;
1117	pt1->regs = regs;
1118	pci_set_drvdata(pdev, pt1);
1119
1120	ret = pt1_init_adapters(pt1);
1121	if (ret < 0)
1122		goto err_kfree;
1123
1124	mutex_init(&pt1->lock);
1125
1126	pt1->power = 0;
1127	pt1->reset = 1;
1128	pt1_update_power(pt1);
1129
1130	i2c_adap = &pt1->i2c_adap;
1131	i2c_adap->algo = &pt1_i2c_algo;
1132	i2c_adap->algo_data = NULL;
1133	i2c_adap->dev.parent = &pdev->dev;
1134	strcpy(i2c_adap->name, DRIVER_NAME);
1135	i2c_set_adapdata(i2c_adap, pt1);
1136	ret = i2c_add_adapter(i2c_adap);
1137	if (ret < 0)
1138		goto err_pt1_cleanup_adapters;
1139
1140	pt1_i2c_init(pt1);
1141	pt1_i2c_wait(pt1);
1142
1143	ret = pt1_sync(pt1);
1144	if (ret < 0)
1145		goto err_i2c_del_adapter;
1146
1147	pt1_identify(pt1);
1148
1149	ret = pt1_unlock(pt1);
1150	if (ret < 0)
1151		goto err_i2c_del_adapter;
1152
1153	ret = pt1_reset_pci(pt1);
1154	if (ret < 0)
1155		goto err_i2c_del_adapter;
1156
1157	ret = pt1_reset_ram(pt1);
1158	if (ret < 0)
1159		goto err_i2c_del_adapter;
1160
1161	ret = pt1_enable_ram(pt1);
1162	if (ret < 0)
1163		goto err_i2c_del_adapter;
1164
1165	pt1_init_streams(pt1);
1166
1167	pt1->power = 1;
1168	pt1_update_power(pt1);
1169	schedule_timeout_uninterruptible((HZ + 49) / 50);
1170
1171	pt1->reset = 0;
1172	pt1_update_power(pt1);
1173	schedule_timeout_uninterruptible((HZ + 999) / 1000);
1174
1175	ret = pt1_init_frontends(pt1);
1176	if (ret < 0)
1177		goto err_pt1_disable_ram;
1178
1179	ret = pt1_init_tables(pt1);
1180	if (ret < 0)
1181		goto err_pt1_cleanup_frontends;
1182
1183	return 0;
1184
1185err_pt1_cleanup_frontends:
1186	pt1_cleanup_frontends(pt1);
1187err_pt1_disable_ram:
1188	pt1_disable_ram(pt1);
1189	pt1->power = 0;
1190	pt1->reset = 1;
1191	pt1_update_power(pt1);
1192err_i2c_del_adapter:
1193	i2c_del_adapter(i2c_adap);
1194err_pt1_cleanup_adapters:
1195	pt1_cleanup_adapters(pt1);
1196err_kfree:
1197	kfree(pt1);
1198err_pci_iounmap:
1199	pci_iounmap(pdev, regs);
1200err_pci_release_regions:
1201	pci_release_regions(pdev);
1202err_pci_disable_device:
1203	pci_disable_device(pdev);
1204err:
1205	return ret;
1206
1207}
1208
1209static struct pci_device_id pt1_id_table[] = {
1210	{ PCI_DEVICE(0x10ee, 0x211a) },
1211	{ PCI_DEVICE(0x10ee, 0x222a) },
1212	{ },
1213};
1214MODULE_DEVICE_TABLE(pci, pt1_id_table);
1215
1216static struct pci_driver pt1_driver = {
1217	.name		= DRIVER_NAME,
1218	.probe		= pt1_probe,
1219	.remove		= pt1_remove,
1220	.id_table	= pt1_id_table,
1221};
1222
1223module_pci_driver(pt1_driver);
1224
1225MODULE_AUTHOR("Takahito HIRANO <hiranotaka@zng.info>");
1226MODULE_DESCRIPTION("Earthsoft PT1/PT2 Driver");
1227MODULE_LICENSE("GPL");
1228