GNU Linux-libre 4.14.332-gnu1
[releases.git] / drivers / media / usb / dvb-usb / af9005.c
1 /* DVB USB compliant Linux driver for the Afatech 9005
2  * USB1.1 DVB-T receiver.
3  *
4  * Copyright (C) 2007 Luca Olivetti (luca@ventoso.org)
5  *
6  * Thanks to Afatech who kindly provided information.
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * see Documentation/dvb/README.dvb-usb for more information
19  */
20 #include "af9005.h"
21
22 /* debug */
23 int dvb_usb_af9005_debug;
24 module_param_named(debug, dvb_usb_af9005_debug, int, 0644);
25 MODULE_PARM_DESC(debug,
26                  "set debugging level (1=info,xfer=2,rc=4,reg=8,i2c=16,fw=32 (or-able))."
27                  DVB_USB_DEBUG_STATUS);
28 /* enable obnoxious led */
29 bool dvb_usb_af9005_led = true;
30 module_param_named(led, dvb_usb_af9005_led, bool, 0644);
31 MODULE_PARM_DESC(led, "enable led (default: 1).");
32
33 /* eeprom dump */
34 static int dvb_usb_af9005_dump_eeprom;
35 module_param_named(dump_eeprom, dvb_usb_af9005_dump_eeprom, int, 0);
36 MODULE_PARM_DESC(dump_eeprom, "dump contents of the eeprom.");
37
38 DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);
39
40 /* remote control decoder */
41 static int (*rc_decode) (struct dvb_usb_device *d, u8 *data, int len,
42                 u32 *event, int *state);
43 static void *rc_keys;
44 static int *rc_keys_size;
45
46 u8 regmask[8] = { 0x01, 0x03, 0x07, 0x0f, 0x1f, 0x3f, 0x7f, 0xff };
47
48 struct af9005_device_state {
49         u8 sequence;
50         int led_state;
51         unsigned char data[256];
52 };
53
54 static int af9005_generic_read_write(struct dvb_usb_device *d, u16 reg,
55                               int readwrite, int type, u8 * values, int len)
56 {
57         struct af9005_device_state *st = d->priv;
58         u8 command, seq;
59         int i, ret;
60
61         if (len < 1) {
62                 err("generic read/write, less than 1 byte. Makes no sense.");
63                 return -EINVAL;
64         }
65         if (len > 8) {
66                 err("generic read/write, more than 8 bytes. Not supported.");
67                 return -EINVAL;
68         }
69
70         mutex_lock(&d->data_mutex);
71         st->data[0] = 14;               /* rest of buffer length low */
72         st->data[1] = 0;                /* rest of buffer length high */
73
74         st->data[2] = AF9005_REGISTER_RW;       /* register operation */
75         st->data[3] = 12;               /* rest of buffer length */
76
77         st->data[4] = seq = st->sequence++;     /* sequence number */
78
79         st->data[5] = (u8) (reg >> 8);  /* register address */
80         st->data[6] = (u8) (reg & 0xff);
81
82         if (type == AF9005_OFDM_REG) {
83                 command = AF9005_CMD_OFDM_REG;
84         } else {
85                 command = AF9005_CMD_TUNER;
86         }
87
88         if (len > 1)
89                 command |=
90                     AF9005_CMD_BURST | AF9005_CMD_AUTOINC | (len - 1) << 3;
91         command |= readwrite;
92         if (readwrite == AF9005_CMD_WRITE)
93                 for (i = 0; i < len; i++)
94                         st->data[8 + i] = values[i];
95         else if (type == AF9005_TUNER_REG)
96                 /* read command for tuner, the first byte contains the i2c address */
97                 st->data[8] = values[0];
98         st->data[7] = command;
99
100         ret = dvb_usb_generic_rw(d, st->data, 16, st->data, 17, 0);
101         if (ret)
102                 goto ret;
103
104         /* sanity check */
105         if (st->data[2] != AF9005_REGISTER_RW_ACK) {
106                 err("generic read/write, wrong reply code.");
107                 ret = -EIO;
108                 goto ret;
109         }
110         if (st->data[3] != 0x0d) {
111                 err("generic read/write, wrong length in reply.");
112                 ret = -EIO;
113                 goto ret;
114         }
115         if (st->data[4] != seq) {
116                 err("generic read/write, wrong sequence in reply.");
117                 ret = -EIO;
118                 goto ret;
119         }
120         /*
121          * In thesis, both input and output buffers should have
122          * identical values for st->data[5] to st->data[8].
123          * However, windows driver doesn't check these fields, in fact
124          * sometimes the register in the reply is different that what
125          * has been sent
126          */
127         if (st->data[16] != 0x01) {
128                 err("generic read/write wrong status code in reply.");
129                 ret = -EIO;
130                 goto ret;
131         }
132
133         if (readwrite == AF9005_CMD_READ)
134                 for (i = 0; i < len; i++)
135                         values[i] = st->data[8 + i];
136
137 ret:
138         mutex_unlock(&d->data_mutex);
139         return ret;
140
141 }
142
143 int af9005_read_ofdm_register(struct dvb_usb_device *d, u16 reg, u8 * value)
144 {
145         int ret;
146         deb_reg("read register %x ", reg);
147         ret = af9005_generic_read_write(d, reg,
148                                         AF9005_CMD_READ, AF9005_OFDM_REG,
149                                         value, 1);
150         if (ret)
151                 deb_reg("failed\n");
152         else
153                 deb_reg("value %x\n", *value);
154         return ret;
155 }
156
157 int af9005_read_ofdm_registers(struct dvb_usb_device *d, u16 reg,
158                                u8 * values, int len)
159 {
160         int ret;
161         deb_reg("read %d registers %x ", len, reg);
162         ret = af9005_generic_read_write(d, reg,
163                                         AF9005_CMD_READ, AF9005_OFDM_REG,
164                                         values, len);
165         if (ret)
166                 deb_reg("failed\n");
167         else
168                 debug_dump(values, len, deb_reg);
169         return ret;
170 }
171
172 int af9005_write_ofdm_register(struct dvb_usb_device *d, u16 reg, u8 value)
173 {
174         int ret;
175         u8 temp = value;
176         deb_reg("write register %x value %x ", reg, value);
177         ret = af9005_generic_read_write(d, reg,
178                                         AF9005_CMD_WRITE, AF9005_OFDM_REG,
179                                         &temp, 1);
180         if (ret)
181                 deb_reg("failed\n");
182         else
183                 deb_reg("ok\n");
184         return ret;
185 }
186
187 int af9005_write_ofdm_registers(struct dvb_usb_device *d, u16 reg,
188                                 u8 * values, int len)
189 {
190         int ret;
191         deb_reg("write %d registers %x values ", len, reg);
192         debug_dump(values, len, deb_reg);
193
194         ret = af9005_generic_read_write(d, reg,
195                                         AF9005_CMD_WRITE, AF9005_OFDM_REG,
196                                         values, len);
197         if (ret)
198                 deb_reg("failed\n");
199         else
200                 deb_reg("ok\n");
201         return ret;
202 }
203
204 int af9005_read_register_bits(struct dvb_usb_device *d, u16 reg, u8 pos,
205                               u8 len, u8 * value)
206 {
207         u8 temp;
208         int ret;
209         deb_reg("read bits %x %x %x", reg, pos, len);
210         ret = af9005_read_ofdm_register(d, reg, &temp);
211         if (ret) {
212                 deb_reg(" failed\n");
213                 return ret;
214         }
215         *value = (temp >> pos) & regmask[len - 1];
216         deb_reg(" value %x\n", *value);
217         return 0;
218
219 }
220
221 int af9005_write_register_bits(struct dvb_usb_device *d, u16 reg, u8 pos,
222                                u8 len, u8 value)
223 {
224         u8 temp, mask;
225         int ret;
226         deb_reg("write bits %x %x %x value %x\n", reg, pos, len, value);
227         if (pos == 0 && len == 8)
228                 return af9005_write_ofdm_register(d, reg, value);
229         ret = af9005_read_ofdm_register(d, reg, &temp);
230         if (ret)
231                 return ret;
232         mask = regmask[len - 1] << pos;
233         temp = (temp & ~mask) | ((value << pos) & mask);
234         return af9005_write_ofdm_register(d, reg, temp);
235
236 }
237
238 static int af9005_usb_read_tuner_registers(struct dvb_usb_device *d,
239                                            u16 reg, u8 * values, int len)
240 {
241         return af9005_generic_read_write(d, reg,
242                                          AF9005_CMD_READ, AF9005_TUNER_REG,
243                                          values, len);
244 }
245
246 static int af9005_usb_write_tuner_registers(struct dvb_usb_device *d,
247                                             u16 reg, u8 * values, int len)
248 {
249         return af9005_generic_read_write(d, reg,
250                                          AF9005_CMD_WRITE,
251                                          AF9005_TUNER_REG, values, len);
252 }
253
254 int af9005_write_tuner_registers(struct dvb_usb_device *d, u16 reg,
255                                  u8 * values, int len)
256 {
257         /* don't let the name of this function mislead you: it's just used
258            as an interface from the firmware to the i2c bus. The actual
259            i2c addresses are contained in the data */
260         int ret, i, done = 0, fail = 0;
261         u8 temp;
262         ret = af9005_usb_write_tuner_registers(d, reg, values, len);
263         if (ret)
264                 return ret;
265         if (reg != 0xffff) {
266                 /* check if write done (0xa40d bit 1) or fail (0xa40d bit 2) */
267                 for (i = 0; i < 200; i++) {
268                         ret =
269                             af9005_read_ofdm_register(d,
270                                                       xd_I2C_i2c_m_status_wdat_done,
271                                                       &temp);
272                         if (ret)
273                                 return ret;
274                         done = temp & (regmask[i2c_m_status_wdat_done_len - 1]
275                                        << i2c_m_status_wdat_done_pos);
276                         if (done)
277                                 break;
278                         fail = temp & (regmask[i2c_m_status_wdat_fail_len - 1]
279                                        << i2c_m_status_wdat_fail_pos);
280                         if (fail)
281                                 break;
282                         msleep(50);
283                 }
284                 if (i == 200)
285                         return -ETIMEDOUT;
286                 if (fail) {
287                         /* clear write fail bit */
288                         af9005_write_register_bits(d,
289                                                    xd_I2C_i2c_m_status_wdat_fail,
290                                                    i2c_m_status_wdat_fail_pos,
291                                                    i2c_m_status_wdat_fail_len,
292                                                    1);
293                         return -EIO;
294                 }
295                 /* clear write done bit */
296                 ret =
297                     af9005_write_register_bits(d,
298                                                xd_I2C_i2c_m_status_wdat_fail,
299                                                i2c_m_status_wdat_done_pos,
300                                                i2c_m_status_wdat_done_len, 1);
301                 if (ret)
302                         return ret;
303         }
304         return 0;
305 }
306
307 int af9005_read_tuner_registers(struct dvb_usb_device *d, u16 reg, u8 addr,
308                                 u8 * values, int len)
309 {
310         /* don't let the name of this function mislead you: it's just used
311            as an interface from the firmware to the i2c bus. The actual
312            i2c addresses are contained in the data */
313         int ret, i;
314         u8 temp, buf[2];
315
316         buf[0] = addr;          /* tuner i2c address */
317         buf[1] = values[0];     /* tuner register */
318
319         values[0] = addr + 0x01;        /* i2c read address */
320
321         if (reg == APO_REG_I2C_RW_SILICON_TUNER) {
322                 /* write tuner i2c address to tuner, 0c00c0 undocumented, found by sniffing */
323                 ret = af9005_write_tuner_registers(d, 0x00c0, buf, 2);
324                 if (ret)
325                         return ret;
326         }
327
328         /* send read command to ofsm */
329         ret = af9005_usb_read_tuner_registers(d, reg, values, 1);
330         if (ret)
331                 return ret;
332
333         /* check if read done */
334         for (i = 0; i < 200; i++) {
335                 ret = af9005_read_ofdm_register(d, 0xa408, &temp);
336                 if (ret)
337                         return ret;
338                 if (temp & 0x01)
339                         break;
340                 msleep(50);
341         }
342         if (i == 200)
343                 return -ETIMEDOUT;
344
345         /* clear read done bit (by writing 1) */
346         ret = af9005_write_ofdm_register(d, xd_I2C_i2c_m_data8, 1);
347         if (ret)
348                 return ret;
349
350         /* get read data (available from 0xa400) */
351         for (i = 0; i < len; i++) {
352                 ret = af9005_read_ofdm_register(d, 0xa400 + i, &temp);
353                 if (ret)
354                         return ret;
355                 values[i] = temp;
356         }
357         return 0;
358 }
359
360 static int af9005_i2c_write(struct dvb_usb_device *d, u8 i2caddr, u8 reg,
361                             u8 * data, int len)
362 {
363         int ret, i;
364         u8 buf[3];
365         deb_i2c("i2c_write i2caddr %x, reg %x, len %d data ", i2caddr,
366                 reg, len);
367         debug_dump(data, len, deb_i2c);
368
369         for (i = 0; i < len; i++) {
370                 buf[0] = i2caddr;
371                 buf[1] = reg + (u8) i;
372                 buf[2] = data[i];
373                 ret =
374                     af9005_write_tuner_registers(d,
375                                                  APO_REG_I2C_RW_SILICON_TUNER,
376                                                  buf, 3);
377                 if (ret) {
378                         deb_i2c("i2c_write failed\n");
379                         return ret;
380                 }
381         }
382         deb_i2c("i2c_write ok\n");
383         return 0;
384 }
385
386 static int af9005_i2c_read(struct dvb_usb_device *d, u8 i2caddr, u8 reg,
387                            u8 * data, int len)
388 {
389         int ret, i;
390         u8 temp;
391         deb_i2c("i2c_read i2caddr %x, reg %x, len %d\n ", i2caddr, reg, len);
392         for (i = 0; i < len; i++) {
393                 temp = reg + i;
394                 ret =
395                     af9005_read_tuner_registers(d,
396                                                 APO_REG_I2C_RW_SILICON_TUNER,
397                                                 i2caddr, &temp, 1);
398                 if (ret) {
399                         deb_i2c("i2c_read failed\n");
400                         return ret;
401                 }
402                 data[i] = temp;
403         }
404         deb_i2c("i2c data read: ");
405         debug_dump(data, len, deb_i2c);
406         return 0;
407 }
408
409 static int af9005_i2c_xfer(struct i2c_adapter *adap, struct i2c_msg msg[],
410                            int num)
411 {
412         /* only implements what the mt2060 module does, don't know how
413            to make it really generic */
414         struct dvb_usb_device *d = i2c_get_adapdata(adap);
415         int ret;
416         u8 reg, addr;
417         u8 *value;
418
419         if (mutex_lock_interruptible(&d->i2c_mutex) < 0)
420                 return -EAGAIN;
421
422         if (num > 2)
423                 warn("more than 2 i2c messages at a time is not handled yet. TODO.");
424
425         if (num == 2) {
426                 /* reads a single register */
427                 reg = *msg[0].buf;
428                 addr = msg[0].addr;
429                 value = msg[1].buf;
430                 ret = af9005_i2c_read(d, addr, reg, value, 1);
431                 if (ret == 0)
432                         ret = 2;
433         } else {
434                 if (msg[0].len < 2) {
435                         ret = -EOPNOTSUPP;
436                         goto unlock;
437                 }
438                 /* write one or more registers */
439                 reg = msg[0].buf[0];
440                 addr = msg[0].addr;
441                 value = &msg[0].buf[1];
442                 ret = af9005_i2c_write(d, addr, reg, value, msg[0].len - 1);
443                 if (ret == 0)
444                         ret = 1;
445         }
446
447 unlock:
448         mutex_unlock(&d->i2c_mutex);
449         return ret;
450 }
451
452 static u32 af9005_i2c_func(struct i2c_adapter *adapter)
453 {
454         return I2C_FUNC_I2C;
455 }
456
457 static struct i2c_algorithm af9005_i2c_algo = {
458         .master_xfer = af9005_i2c_xfer,
459         .functionality = af9005_i2c_func,
460 };
461
462 int af9005_send_command(struct dvb_usb_device *d, u8 command, u8 * wbuf,
463                         int wlen, u8 * rbuf, int rlen)
464 {
465         struct af9005_device_state *st = d->priv;
466
467         int ret, i, packet_len;
468         u8 seq;
469
470         if (wlen < 0) {
471                 err("send command, wlen less than 0 bytes. Makes no sense.");
472                 return -EINVAL;
473         }
474         if (wlen > 54) {
475                 err("send command, wlen more than 54 bytes. Not supported.");
476                 return -EINVAL;
477         }
478         if (rlen > 54) {
479                 err("send command, rlen more than 54 bytes. Not supported.");
480                 return -EINVAL;
481         }
482         packet_len = wlen + 5;
483
484         mutex_lock(&d->data_mutex);
485
486         st->data[0] = (u8) (packet_len & 0xff);
487         st->data[1] = (u8) ((packet_len & 0xff00) >> 8);
488
489         st->data[2] = 0x26;             /* packet type */
490         st->data[3] = wlen + 3;
491         st->data[4] = seq = st->sequence++;
492         st->data[5] = command;
493         st->data[6] = wlen;
494         for (i = 0; i < wlen; i++)
495                 st->data[7 + i] = wbuf[i];
496         ret = dvb_usb_generic_rw(d, st->data, wlen + 7, st->data, rlen + 7, 0);
497         if (st->data[2] != 0x27) {
498                 err("send command, wrong reply code.");
499                 ret = -EIO;
500         } else if (st->data[4] != seq) {
501                 err("send command, wrong sequence in reply.");
502                 ret = -EIO;
503         } else if (st->data[5] != 0x01) {
504                 err("send command, wrong status code in reply.");
505                 ret = -EIO;
506         } else if (st->data[6] != rlen) {
507                 err("send command, invalid data length in reply.");
508                 ret = -EIO;
509         }
510         if (!ret) {
511                 for (i = 0; i < rlen; i++)
512                         rbuf[i] = st->data[i + 7];
513         }
514
515         mutex_unlock(&d->data_mutex);
516         return ret;
517 }
518
519 int af9005_read_eeprom(struct dvb_usb_device *d, u8 address, u8 * values,
520                        int len)
521 {
522         struct af9005_device_state *st = d->priv;
523         u8 seq;
524         int ret, i;
525
526         mutex_lock(&d->data_mutex);
527
528         memset(st->data, 0, sizeof(st->data));
529
530         st->data[0] = 14;               /* length of rest of packet low */
531         st->data[1] = 0;                /* length of rest of packer high */
532
533         st->data[2] = 0x2a;             /* read/write eeprom */
534
535         st->data[3] = 12;               /* size */
536
537         st->data[4] = seq = st->sequence++;
538
539         st->data[5] = 0;                /* read */
540
541         st->data[6] = len;
542         st->data[7] = address;
543         ret = dvb_usb_generic_rw(d, st->data, 16, st->data, 14, 0);
544         if (st->data[2] != 0x2b) {
545                 err("Read eeprom, invalid reply code");
546                 ret = -EIO;
547         } else if (st->data[3] != 10) {
548                 err("Read eeprom, invalid reply length");
549                 ret = -EIO;
550         } else if (st->data[4] != seq) {
551                 err("Read eeprom, wrong sequence in reply ");
552                 ret = -EIO;
553         } else if (st->data[5] != 1) {
554                 err("Read eeprom, wrong status in reply ");
555                 ret = -EIO;
556         }
557
558         if (!ret) {
559                 for (i = 0; i < len; i++)
560                         values[i] = st->data[6 + i];
561         }
562         mutex_unlock(&d->data_mutex);
563
564         return ret;
565 }
566
567 static int af9005_boot_packet(struct usb_device *udev, int type, u8 *reply,
568                               u8 *buf, int size)
569 {
570         u16 checksum;
571         int act_len = 0, i, ret;
572
573         memset(buf, 0, size);
574         buf[0] = (u8) (FW_BULKOUT_SIZE & 0xff);
575         buf[1] = (u8) ((FW_BULKOUT_SIZE >> 8) & 0xff);
576         switch (type) {
577         case FW_CONFIG:
578                 buf[2] = 0x11;
579                 buf[3] = 0x04;
580                 buf[4] = 0x00;  /* sequence number, original driver doesn't increment it here */
581                 buf[5] = 0x03;
582                 checksum = buf[4] + buf[5];
583                 buf[6] = (u8) ((checksum >> 8) & 0xff);
584                 buf[7] = (u8) (checksum & 0xff);
585                 break;
586         case FW_CONFIRM:
587                 buf[2] = 0x11;
588                 buf[3] = 0x04;
589                 buf[4] = 0x00;  /* sequence number, original driver doesn't increment it here */
590                 buf[5] = 0x01;
591                 checksum = buf[4] + buf[5];
592                 buf[6] = (u8) ((checksum >> 8) & 0xff);
593                 buf[7] = (u8) (checksum & 0xff);
594                 break;
595         case FW_BOOT:
596                 buf[2] = 0x10;
597                 buf[3] = 0x08;
598                 buf[4] = 0x00;  /* sequence number, original driver doesn't increment it here */
599                 buf[5] = 0x97;
600                 buf[6] = 0xaa;
601                 buf[7] = 0x55;
602                 buf[8] = 0xa5;
603                 buf[9] = 0x5a;
604                 checksum = 0;
605                 for (i = 4; i <= 9; i++)
606                         checksum += buf[i];
607                 buf[10] = (u8) ((checksum >> 8) & 0xff);
608                 buf[11] = (u8) (checksum & 0xff);
609                 break;
610         default:
611                 err("boot packet invalid boot packet type");
612                 return -EINVAL;
613         }
614         deb_fw(">>> ");
615         debug_dump(buf, FW_BULKOUT_SIZE + 2, deb_fw);
616
617         ret = usb_bulk_msg(udev,
618                            usb_sndbulkpipe(udev, 0x02),
619                            buf, FW_BULKOUT_SIZE + 2, &act_len, 2000);
620         if (ret)
621                 err("boot packet bulk message failed: %d (%d/%d)", ret,
622                     FW_BULKOUT_SIZE + 2, act_len);
623         else
624                 ret = act_len != FW_BULKOUT_SIZE + 2 ? -1 : 0;
625         if (ret)
626                 return ret;
627         memset(buf, 0, 9);
628         ret = usb_bulk_msg(udev,
629                            usb_rcvbulkpipe(udev, 0x01), buf, 9, &act_len, 2000);
630         if (ret) {
631                 err("boot packet recv bulk message failed: %d", ret);
632                 return ret;
633         }
634         deb_fw("<<< ");
635         debug_dump(buf, act_len, deb_fw);
636         checksum = 0;
637         switch (type) {
638         case FW_CONFIG:
639                 if (buf[2] != 0x11) {
640                         err("boot bad config header.");
641                         return -EIO;
642                 }
643                 if (buf[3] != 0x05) {
644                         err("boot bad config size.");
645                         return -EIO;
646                 }
647                 if (buf[4] != 0x00) {
648                         err("boot bad config sequence.");
649                         return -EIO;
650                 }
651                 if (buf[5] != 0x04) {
652                         err("boot bad config subtype.");
653                         return -EIO;
654                 }
655                 for (i = 4; i <= 6; i++)
656                         checksum += buf[i];
657                 if (buf[7] * 256 + buf[8] != checksum) {
658                         err("boot bad config checksum.");
659                         return -EIO;
660                 }
661                 *reply = buf[6];
662                 break;
663         case FW_CONFIRM:
664                 if (buf[2] != 0x11) {
665                         err("boot bad confirm header.");
666                         return -EIO;
667                 }
668                 if (buf[3] != 0x05) {
669                         err("boot bad confirm size.");
670                         return -EIO;
671                 }
672                 if (buf[4] != 0x00) {
673                         err("boot bad confirm sequence.");
674                         return -EIO;
675                 }
676                 if (buf[5] != 0x02) {
677                         err("boot bad confirm subtype.");
678                         return -EIO;
679                 }
680                 for (i = 4; i <= 6; i++)
681                         checksum += buf[i];
682                 if (buf[7] * 256 + buf[8] != checksum) {
683                         err("boot bad confirm checksum.");
684                         return -EIO;
685                 }
686                 *reply = buf[6];
687                 break;
688         case FW_BOOT:
689                 if (buf[2] != 0x10) {
690                         err("boot bad boot header.");
691                         return -EIO;
692                 }
693                 if (buf[3] != 0x05) {
694                         err("boot bad boot size.");
695                         return -EIO;
696                 }
697                 if (buf[4] != 0x00) {
698                         err("boot bad boot sequence.");
699                         return -EIO;
700                 }
701                 if (buf[5] != 0x01) {
702                         err("boot bad boot pattern 01.");
703                         return -EIO;
704                 }
705                 if (buf[6] != 0x10) {
706                         err("boot bad boot pattern 10.");
707                         return -EIO;
708                 }
709                 for (i = 4; i <= 6; i++)
710                         checksum += buf[i];
711                 if (buf[7] * 256 + buf[8] != checksum) {
712                         err("boot bad boot checksum.");
713                         return -EIO;
714                 }
715                 break;
716
717         }
718
719         return 0;
720 }
721
722 static int af9005_download_firmware(struct usb_device *udev, const struct firmware *fw)
723 {
724         int i, packets, ret, act_len;
725
726         u8 *buf;
727         u8 reply;
728
729         buf = kmalloc(FW_BULKOUT_SIZE + 2, GFP_KERNEL);
730         if (!buf)
731                 return -ENOMEM;
732
733         ret = af9005_boot_packet(udev, FW_CONFIG, &reply, buf,
734                                  FW_BULKOUT_SIZE + 2);
735         if (ret)
736                 goto err;
737         if (reply != 0x01) {
738                 err("before downloading firmware, FW_CONFIG expected 0x01, received 0x%x", reply);
739                 ret = -EIO;
740                 goto err;
741         }
742         packets = fw->size / FW_BULKOUT_SIZE;
743         buf[0] = (u8) (FW_BULKOUT_SIZE & 0xff);
744         buf[1] = (u8) ((FW_BULKOUT_SIZE >> 8) & 0xff);
745         for (i = 0; i < packets; i++) {
746                 memcpy(&buf[2], fw->data + i * FW_BULKOUT_SIZE,
747                        FW_BULKOUT_SIZE);
748                 deb_fw(">>> ");
749                 debug_dump(buf, FW_BULKOUT_SIZE + 2, deb_fw);
750                 ret = usb_bulk_msg(udev,
751                                    usb_sndbulkpipe(udev, 0x02),
752                                    buf, FW_BULKOUT_SIZE + 2, &act_len, 1000);
753                 if (ret) {
754                         err("firmware download failed at packet %d with code %d", i, ret);
755                         goto err;
756                 }
757         }
758         ret = af9005_boot_packet(udev, FW_CONFIRM, &reply,
759                                  buf, FW_BULKOUT_SIZE + 2);
760         if (ret)
761                 goto err;
762         if (reply != (u8) (packets & 0xff)) {
763                 err("after downloading firmware, FW_CONFIRM expected 0x%x, received 0x%x", packets & 0xff, reply);
764                 ret = -EIO;
765                 goto err;
766         }
767         ret = af9005_boot_packet(udev, FW_BOOT, &reply, buf,
768                                  FW_BULKOUT_SIZE + 2);
769         if (ret)
770                 goto err;
771         ret = af9005_boot_packet(udev, FW_CONFIG, &reply, buf,
772                                  FW_BULKOUT_SIZE + 2);
773         if (ret)
774                 goto err;
775         if (reply != 0x02) {
776                 err("after downloading firmware, FW_CONFIG expected 0x02, received 0x%x", reply);
777                 ret = -EIO;
778                 goto err;
779         }
780
781 err:
782         kfree(buf);
783         return ret;
784
785 }
786
787 int af9005_led_control(struct dvb_usb_device *d, int onoff)
788 {
789         struct af9005_device_state *st = d->priv;
790         int temp, ret;
791
792         if (onoff && dvb_usb_af9005_led)
793                 temp = 1;
794         else
795                 temp = 0;
796         if (st->led_state != temp) {
797                 ret =
798                     af9005_write_register_bits(d, xd_p_reg_top_locken1,
799                                                reg_top_locken1_pos,
800                                                reg_top_locken1_len, temp);
801                 if (ret)
802                         return ret;
803                 ret =
804                     af9005_write_register_bits(d, xd_p_reg_top_lock1,
805                                                reg_top_lock1_pos,
806                                                reg_top_lock1_len, temp);
807                 if (ret)
808                         return ret;
809                 st->led_state = temp;
810         }
811         return 0;
812 }
813
814 static int af9005_frontend_attach(struct dvb_usb_adapter *adap)
815 {
816         u8 buf[8];
817         int i;
818
819         /* without these calls the first commands after downloading
820            the firmware fail. I put these calls here to simulate
821            what it is done in dvb-usb-init.c.
822          */
823         struct usb_device *udev = adap->dev->udev;
824         usb_clear_halt(udev, usb_sndbulkpipe(udev, 2));
825         usb_clear_halt(udev, usb_rcvbulkpipe(udev, 1));
826         if (dvb_usb_af9005_dump_eeprom) {
827                 printk("EEPROM DUMP\n");
828                 for (i = 0; i < 255; i += 8) {
829                         af9005_read_eeprom(adap->dev, i, buf, 8);
830                         debug_dump(buf, 8, printk);
831                 }
832         }
833         adap->fe_adap[0].fe = af9005_fe_attach(adap->dev);
834         return 0;
835 }
836
837 static int af9005_rc_query(struct dvb_usb_device *d, u32 * event, int *state)
838 {
839         struct af9005_device_state *st = d->priv;
840         int ret, len;
841         u8 seq;
842
843         *state = REMOTE_NO_KEY_PRESSED;
844         if (rc_decode == NULL) {
845                 /* it shouldn't never come here */
846                 return 0;
847         }
848
849         mutex_lock(&d->data_mutex);
850
851         /* deb_info("rc_query\n"); */
852         st->data[0] = 3;                /* rest of packet length low */
853         st->data[1] = 0;                /* rest of packet lentgh high */
854         st->data[2] = 0x40;             /* read remote */
855         st->data[3] = 1;                /* rest of packet length */
856         st->data[4] = seq = st->sequence++;     /* sequence number */
857         ret = dvb_usb_generic_rw(d, st->data, 5, st->data, 256, 0);
858         if (ret) {
859                 err("rc query failed");
860                 goto ret;
861         }
862         if (st->data[2] != 0x41) {
863                 err("rc query bad header.");
864                 ret = -EIO;
865                 goto ret;
866         } else if (st->data[4] != seq) {
867                 err("rc query bad sequence.");
868                 ret = -EIO;
869                 goto ret;
870         }
871         len = st->data[5];
872         if (len > 246) {
873                 err("rc query invalid length");
874                 ret = -EIO;
875                 goto ret;
876         }
877         if (len > 0) {
878                 deb_rc("rc data (%d) ", len);
879                 debug_dump((st->data + 6), len, deb_rc);
880                 ret = rc_decode(d, &st->data[6], len, event, state);
881                 if (ret) {
882                         err("rc_decode failed");
883                         goto ret;
884                 } else {
885                         deb_rc("rc_decode state %x event %x\n", *state, *event);
886                         if (*state == REMOTE_KEY_REPEAT)
887                                 *event = d->last_event;
888                 }
889         }
890
891 ret:
892         mutex_unlock(&d->data_mutex);
893         return ret;
894 }
895
896 static int af9005_power_ctrl(struct dvb_usb_device *d, int onoff)
897 {
898
899         return 0;
900 }
901
902 static int af9005_pid_filter_control(struct dvb_usb_adapter *adap, int onoff)
903 {
904         int ret;
905         deb_info("pid filter control  onoff %d\n", onoff);
906         if (onoff) {
907                 ret =
908                     af9005_write_ofdm_register(adap->dev, XD_MP2IF_DMX_CTRL, 1);
909                 if (ret)
910                         return ret;
911                 ret =
912                     af9005_write_register_bits(adap->dev,
913                                                XD_MP2IF_DMX_CTRL, 1, 1, 1);
914                 if (ret)
915                         return ret;
916                 ret =
917                     af9005_write_ofdm_register(adap->dev, XD_MP2IF_DMX_CTRL, 1);
918         } else
919                 ret =
920                     af9005_write_ofdm_register(adap->dev, XD_MP2IF_DMX_CTRL, 0);
921         if (ret)
922                 return ret;
923         deb_info("pid filter control ok\n");
924         return 0;
925 }
926
927 static int af9005_pid_filter(struct dvb_usb_adapter *adap, int index,
928                              u16 pid, int onoff)
929 {
930         u8 cmd = index & 0x1f;
931         int ret;
932         deb_info("set pid filter, index %d, pid %x, onoff %d\n", index,
933                  pid, onoff);
934         if (onoff) {
935                 /* cannot use it as pid_filter_ctrl since it has to be done
936                    before setting the first pid */
937                 if (adap->feedcount == 1) {
938                         deb_info("first pid set, enable pid table\n");
939                         ret = af9005_pid_filter_control(adap, onoff);
940                         if (ret)
941                                 return ret;
942                 }
943                 ret =
944                     af9005_write_ofdm_register(adap->dev,
945                                                XD_MP2IF_PID_DATA_L,
946                                                (u8) (pid & 0xff));
947                 if (ret)
948                         return ret;
949                 ret =
950                     af9005_write_ofdm_register(adap->dev,
951                                                XD_MP2IF_PID_DATA_H,
952                                                (u8) (pid >> 8));
953                 if (ret)
954                         return ret;
955                 cmd |= 0x20 | 0x40;
956         } else {
957                 if (adap->feedcount == 0) {
958                         deb_info("last pid unset, disable pid table\n");
959                         ret = af9005_pid_filter_control(adap, onoff);
960                         if (ret)
961                                 return ret;
962                 }
963         }
964         ret = af9005_write_ofdm_register(adap->dev, XD_MP2IF_PID_IDX, cmd);
965         if (ret)
966                 return ret;
967         deb_info("set pid ok\n");
968         return 0;
969 }
970
971 static int af9005_identify_state(struct usb_device *udev,
972                                  struct dvb_usb_device_properties *props,
973                                  struct dvb_usb_device_description **desc,
974                                  int *cold)
975 {
976         int ret;
977         u8 reply, *buf;
978
979         buf = kmalloc(FW_BULKOUT_SIZE + 2, GFP_KERNEL);
980         if (!buf)
981                 return -ENOMEM;
982
983         ret = af9005_boot_packet(udev, FW_CONFIG, &reply,
984                                  buf, FW_BULKOUT_SIZE + 2);
985         if (ret)
986                 goto err;
987         deb_info("result of FW_CONFIG in identify state %d\n", reply);
988         if (reply == 0x01)
989                 *cold = 1;
990         else if (reply == 0x02)
991                 *cold = 0;
992         else
993                 ret = -EIO;
994         if (!ret)
995                 deb_info("Identify state cold = %d\n", *cold);
996
997 err:
998         kfree(buf);
999         return ret;
1000 }
1001
1002 static struct dvb_usb_device_properties af9005_properties;
1003
1004 static int af9005_usb_probe(struct usb_interface *intf,
1005                             const struct usb_device_id *id)
1006 {
1007         return dvb_usb_device_init(intf, &af9005_properties,
1008                                   THIS_MODULE, NULL, adapter_nr);
1009 }
1010
1011 enum af9005_usb_table_entry {
1012         AFATECH_AF9005,
1013         TERRATEC_AF9005,
1014         ANSONIC_AF9005,
1015 };
1016
1017 static struct usb_device_id af9005_usb_table[] = {
1018         [AFATECH_AF9005] = {USB_DEVICE(USB_VID_AFATECH,
1019                                 USB_PID_AFATECH_AF9005)},
1020         [TERRATEC_AF9005] = {USB_DEVICE(USB_VID_TERRATEC,
1021                                 USB_PID_TERRATEC_CINERGY_T_USB_XE)},
1022         [ANSONIC_AF9005] = {USB_DEVICE(USB_VID_ANSONIC,
1023                                 USB_PID_ANSONIC_DVBT_USB)},
1024         { }
1025 };
1026
1027 MODULE_DEVICE_TABLE(usb, af9005_usb_table);
1028
1029 static struct dvb_usb_device_properties af9005_properties = {
1030         .caps = DVB_USB_IS_AN_I2C_ADAPTER,
1031
1032         .usb_ctrl = DEVICE_SPECIFIC,
1033         .firmware = "/*(DEBLOBBED)*/",
1034         .download_firmware = af9005_download_firmware,
1035         .no_reconnect = 1,
1036
1037         .size_of_priv = sizeof(struct af9005_device_state),
1038
1039         .num_adapters = 1,
1040         .adapter = {
1041                     {
1042                     .num_frontends = 1,
1043                     .fe = {{
1044                      .caps =
1045                      DVB_USB_ADAP_HAS_PID_FILTER |
1046                      DVB_USB_ADAP_PID_FILTER_CAN_BE_TURNED_OFF,
1047                      .pid_filter_count = 32,
1048                      .pid_filter = af9005_pid_filter,
1049                      /* .pid_filter_ctrl = af9005_pid_filter_control, */
1050                      .frontend_attach = af9005_frontend_attach,
1051                      /* .tuner_attach     = af9005_tuner_attach, */
1052                      /* parameter for the MPEG2-data transfer */
1053                      .stream = {
1054                                 .type = USB_BULK,
1055                                 .count = 10,
1056                                 .endpoint = 0x04,
1057                                 .u = {
1058                                       .bulk = {
1059                                                .buffersize = 4096,      /* actual size seen is 3948 */
1060                                                }
1061                                       }
1062                                 },
1063                      }},
1064                      }
1065                     },
1066         .power_ctrl = af9005_power_ctrl,
1067         .identify_state = af9005_identify_state,
1068
1069         .i2c_algo = &af9005_i2c_algo,
1070
1071         .rc.legacy = {
1072                 .rc_interval = 200,
1073                 .rc_map_table = NULL,
1074                 .rc_map_size = 0,
1075                 .rc_query = af9005_rc_query,
1076         },
1077
1078         .generic_bulk_ctrl_endpoint          = 2,
1079         .generic_bulk_ctrl_endpoint_response = 1,
1080
1081         .num_device_descs = 3,
1082         .devices = {
1083                     {.name = "Afatech DVB-T USB1.1 stick",
1084                      .cold_ids = {&af9005_usb_table[AFATECH_AF9005], NULL},
1085                      .warm_ids = {NULL},
1086                      },
1087                     {.name = "TerraTec Cinergy T USB XE",
1088                      .cold_ids = {&af9005_usb_table[TERRATEC_AF9005], NULL},
1089                      .warm_ids = {NULL},
1090                      },
1091                     {.name = "Ansonic DVB-T USB1.1 stick",
1092                      .cold_ids = {&af9005_usb_table[ANSONIC_AF9005], NULL},
1093                      .warm_ids = {NULL},
1094                      },
1095                     {NULL},
1096                     }
1097 };
1098
1099 /* usb specific object needed to register this driver with the usb subsystem */
1100 static struct usb_driver af9005_usb_driver = {
1101         .name = "dvb_usb_af9005",
1102         .probe = af9005_usb_probe,
1103         .disconnect = dvb_usb_device_exit,
1104         .id_table = af9005_usb_table,
1105 };
1106
1107 /* module stuff */
1108 static int __init af9005_usb_module_init(void)
1109 {
1110         int result;
1111         if ((result = usb_register(&af9005_usb_driver))) {
1112                 err("usb_register failed. (%d)", result);
1113                 return result;
1114         }
1115 #if IS_MODULE(CONFIG_DVB_USB_AF9005) || defined(CONFIG_DVB_USB_AF9005_REMOTE)
1116         /* FIXME: convert to todays kernel IR infrastructure */
1117         rc_decode = symbol_request(af9005_rc_decode);
1118         rc_keys = symbol_request(rc_map_af9005_table);
1119         rc_keys_size = symbol_request(rc_map_af9005_table_size);
1120 #endif
1121         if (rc_decode == NULL || rc_keys == NULL || rc_keys_size == NULL) {
1122                 err("af9005_rc_decode function not found, disabling remote");
1123                 af9005_properties.rc.legacy.rc_query = NULL;
1124         } else {
1125                 af9005_properties.rc.legacy.rc_map_table = rc_keys;
1126                 af9005_properties.rc.legacy.rc_map_size = *rc_keys_size;
1127         }
1128
1129         return 0;
1130 }
1131
1132 static void __exit af9005_usb_module_exit(void)
1133 {
1134         /* release rc decode symbols */
1135         if (rc_decode != NULL)
1136                 symbol_put(af9005_rc_decode);
1137         if (rc_keys != NULL)
1138                 symbol_put(rc_map_af9005_table);
1139         if (rc_keys_size != NULL)
1140                 symbol_put(rc_map_af9005_table_size);
1141         /* deregister this driver from the USB subsystem */
1142         usb_deregister(&af9005_usb_driver);
1143 }
1144
1145 module_init(af9005_usb_module_init);
1146 module_exit(af9005_usb_module_exit);
1147
1148 MODULE_AUTHOR("Luca Olivetti <luca@ventoso.org>");
1149 MODULE_DESCRIPTION("Driver for Afatech 9005 DVB-T USB1.1 stick");
1150 MODULE_VERSION("1.0");
1151 MODULE_LICENSE("GPL");