drivers/media/dvb-frontends/rtl2832_sdr.c

   1 /*
   2  * Realtek RTL2832U SDR driver
   3  *
   4  * Copyright (C) 2013 Antti Palosaari <crope@iki.fi>
   5  *
   6  *    This program is free software; you can redistribute it and/or modify
   7  *    it under the terms of the GNU General Public License as published by
   8  *    the Free Software Foundation; either version 2 of the License, or
   9  *    (at your option) any later version.
  10  *
  11  *    This program is distributed in the hope that it will be useful,
  12  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *    GNU General Public License for more details.
  15  *
  16  *    You should have received a copy of the GNU General Public License along
  17  *    with this program; if not, write to the Free Software Foundation, Inc.,
  18  *    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  *
  20  * GNU Radio plugin "gr-kernel" for device usage will be on:
  21  * http://git.linuxtv.org/anttip/gr-kernel.git
  22  *
  23  */
  24
  25 #include "dvb_frontend.h"
  26 #include "rtl2832_sdr.h"
  27 #include "dvb_usb.h"
  28
  29 #include <media/v4l2-device.h>
  30 #include <media/v4l2-ioctl.h>
  31 #include <media/v4l2-ctrls.h>
  32 #include <media/v4l2-event.h>
  33 #include <media/videobuf2-vmalloc.h>
  34
  35 #include <linux/jiffies.h>
  36 #include <linux/math64.h>
  37
  38 static bool rtl2832_sdr_emulated_fmt;
  39 module_param_named(emulated_formats, rtl2832_sdr_emulated_fmt, bool, 0644);
  40 MODULE_PARM_DESC(emulated_formats, "enable emulated formats (disappears in future)");
  41
  42 #define MAX_BULK_BUFS            (10)
  43 #define BULK_BUFFER_SIZE         (128 * 512)
  44
  45 static const struct v4l2_frequency_band bands_adc[] = {
  46         {
  47                 .tuner = 0,
  48                 .type = V4L2_TUNER_ADC,
  49                 .index = 0,
  50                 .capability = V4L2_TUNER_CAP_1HZ | V4L2_TUNER_CAP_FREQ_BANDS,
  51                 .rangelow   =  300000,
  52                 .rangehigh  =  300000,
  53         },
  54         {
  55                 .tuner = 0,
  56                 .type = V4L2_TUNER_ADC,
  57                 .index = 1,
  58                 .capability = V4L2_TUNER_CAP_1HZ | V4L2_TUNER_CAP_FREQ_BANDS,
  59                 .rangelow   =  900001,
  60                 .rangehigh  = 2800000,
  61         },
  62         {
  63                 .tuner = 0,
  64                 .type = V4L2_TUNER_ADC,
  65                 .index = 2,
  66                 .capability = V4L2_TUNER_CAP_1HZ | V4L2_TUNER_CAP_FREQ_BANDS,
  67                 .rangelow   = 3200000,
  68                 .rangehigh  = 3200000,
  69         },
  70 };
  71
  72 static const struct v4l2_frequency_band bands_fm[] = {
  73         {
  74                 .tuner = 1,
  75                 .type = V4L2_TUNER_RF,
  76                 .index = 0,
  77                 .capability = V4L2_TUNER_CAP_1HZ | V4L2_TUNER_CAP_FREQ_BANDS,
  78                 .rangelow   =    50000000,
  79                 .rangehigh  =  2000000000,
  80         },
  81 };
  82
  83 /* stream formats */
  84 struct rtl2832_sdr_format {
  85         char    *name;
  86         u32     pixelformat;
  87         u32     buffersize;
  88 };
  89
  90 static struct rtl2832_sdr_format formats[] = {
  91         {
  92                 .name           = "Complex U8",
  93                 .pixelformat    = V4L2_SDR_FMT_CU8,
  94                 .buffersize     = BULK_BUFFER_SIZE,
  95         }, {
  96                 .name           = "Complex U16LE (emulated)",
  97                 .pixelformat    = V4L2_SDR_FMT_CU16LE,
  98                 .buffersize     = BULK_BUFFER_SIZE * 2,
  99         },
 100 };
 101
 102 static const unsigned int NUM_FORMATS = ARRAY_SIZE(formats);
 103
 104 /* intermediate buffers with raw data from the USB device */
 105 struct rtl2832_sdr_frame_buf {
 106         struct vb2_buffer vb;   /* common v4l buffer stuff -- must be first */
 107         struct list_head list;
 108 };
 109
 110 struct rtl2832_sdr_state {
 111 #define POWER_ON           (1 << 1)
 112 #define URB_BUF            (1 << 2)
 113         unsigned long flags;
 114
 115         const struct rtl2832_config *cfg;
 116         struct dvb_frontend *fe;
 117         struct dvb_usb_device *d;
 118         struct i2c_adapter *i2c;
 119         u8 bank;
 120
 121         struct video_device vdev;
 122         struct v4l2_device v4l2_dev;
 123
 124         /* videobuf2 queue and queued buffers list */
 125         struct vb2_queue vb_queue;
 126         struct list_head queued_bufs;
 127         spinlock_t queued_bufs_lock; /* Protects queued_bufs */
 128         unsigned sequence;           /* buffer sequence counter */
 129
 130         /* Note if taking both locks v4l2_lock must always be locked first! */
 131         struct mutex v4l2_lock;      /* Protects everything else */
 132         struct mutex vb_queue_lock;  /* Protects vb_queue and capt_file */
 133
 134         /* Pointer to our usb_device, will be NULL after unplug */
 135         struct usb_device *udev; /* Both mutexes most be hold when setting! */
 136
 137         unsigned int vb_full; /* vb is full and packets dropped */
 138
 139         struct urb     *urb_list[MAX_BULK_BUFS];
 140         int            buf_num;
 141         unsigned long  buf_size;
 142         u8             *buf_list[MAX_BULK_BUFS];
 143         dma_addr_t     dma_addr[MAX_BULK_BUFS];
 144         int urbs_initialized;
 145         int urbs_submitted;
 146
 147         unsigned int f_adc, f_tuner;
 148         u32 pixelformat;
 149         u32 buffersize;
 150         unsigned int num_formats;
 151
 152         /* Controls */
 153         struct v4l2_ctrl_handler hdl;
 154         struct v4l2_ctrl *bandwidth_auto;
 155         struct v4l2_ctrl *bandwidth;
 156
 157         /* for sample rate calc */
 158         unsigned int sample;
 159         unsigned int sample_measured;
 160         unsigned long jiffies_next;
 161 };
 162
 163 /* write multiple hardware registers */
 164 static int rtl2832_sdr_wr(struct rtl2832_sdr_state *s, u8 reg, const u8 *val,
 165                 int len)
 166 {
 167         int ret;
 168 #define MAX_WR_LEN 24
 169 #define MAX_WR_XFER_LEN (MAX_WR_LEN + 1)
 170         u8 buf[MAX_WR_XFER_LEN];
 171         struct i2c_msg msg[1] = {
 172                 {
 173                         .addr = s->cfg->i2c_addr,
 174                         .flags = 0,
 175                         .len = 1 + len,
 176                         .buf = buf,
 177                 }
 178         };
 179
 180         if (WARN_ON(len > MAX_WR_LEN))
 181                 return -EINVAL;
 182
 183         buf[0] = reg;
 184         memcpy(&buf[1], val, len);
 185
 186         ret = i2c_transfer(s->i2c, msg, 1);
 187         if (ret == 1) {
 188                 ret = 0;
 189         } else {
 190                 dev_err(&s->i2c->dev,
 191                         "%s: I2C wr failed=%d reg=%02x len=%d\n",
 192                         KBUILD_MODNAME, ret, reg, len);
 193                 ret = -EREMOTEIO;
 194         }
 195         return ret;
 196 }
 197
 198 /* read multiple hardware registers */
 199 static int rtl2832_sdr_rd(struct rtl2832_sdr_state *s, u8 reg, u8 *val, int len)
 200 {
 201         int ret;
 202         struct i2c_msg msg[2] = {
 203                 {
 204                         .addr = s->cfg->i2c_addr,
 205                         .flags = 0,
 206                         .len = 1,
 207                         .buf = &reg,
 208                 }, {
 209                         .addr = s->cfg->i2c_addr,
 210                         .flags = I2C_M_RD,
 211                         .len = len,
 212                         .buf = val,
 213                 }
 214         };
 215
 216         ret = i2c_transfer(s->i2c, msg, 2);
 217         if (ret == 2) {
 218                 ret = 0;
 219         } else {
 220                 dev_err(&s->i2c->dev,
 221                                 "%s: I2C rd failed=%d reg=%02x len=%d\n",
 222                                 KBUILD_MODNAME, ret, reg, len);
 223                 ret = -EREMOTEIO;
 224         }
 225         return ret;
 226 }
 227
 228 /* write multiple registers */
 229 static int rtl2832_sdr_wr_regs(struct rtl2832_sdr_state *s, u16 reg,
 230                 const u8 *val, int len)
 231 {
 232         int ret;
 233         u8 reg2 = (reg >> 0) & 0xff;
 234         u8 bank = (reg >> 8) & 0xff;
 235
 236         /* switch bank if needed */
 237         if (bank != s->bank) {
 238                 ret = rtl2832_sdr_wr(s, 0x00, &bank, 1);
 239                 if (ret)
 240                         return ret;
 241
 242                 s->bank = bank;
 243         }
 244
 245         return rtl2832_sdr_wr(s, reg2, val, len);
 246 }
 247
 248 /* read multiple registers */
 249 static int rtl2832_sdr_rd_regs(struct rtl2832_sdr_state *s, u16 reg, u8 *val,
 250                 int len)
 251 {
 252         int ret;
 253         u8 reg2 = (reg >> 0) & 0xff;
 254         u8 bank = (reg >> 8) & 0xff;
 255
 256         /* switch bank if needed */
 257         if (bank != s->bank) {
 258                 ret = rtl2832_sdr_wr(s, 0x00, &bank, 1);
 259                 if (ret)
 260                         return ret;
 261
 262                 s->bank = bank;
 263         }
 264
 265         return rtl2832_sdr_rd(s, reg2, val, len);
 266 }
 267
 268 /* write single register */
 269 static int rtl2832_sdr_wr_reg(struct rtl2832_sdr_state *s, u16 reg, u8 val)
 270 {
 271         return rtl2832_sdr_wr_regs(s, reg, &val, 1);
 272 }
 273
 274 #if 0
 275 /* read single register */
 276 static int rtl2832_sdr_rd_reg(struct rtl2832_sdr_state *s, u16 reg, u8 *val)
 277 {
 278         return rtl2832_sdr_rd_regs(s, reg, val, 1);
 279 }
 280 #endif
 281
 282 /* write single register with mask */
 283 static int rtl2832_sdr_wr_reg_mask(struct rtl2832_sdr_state *s, u16 reg,
 284                 u8 val, u8 mask)
 285 {
 286         int ret;
 287         u8 tmp;
 288
 289         /* no need for read if whole reg is written */
 290         if (mask != 0xff) {
 291                 ret = rtl2832_sdr_rd_regs(s, reg, &tmp, 1);
 292                 if (ret)
 293                         return ret;
 294
 295                 val &= mask;
 296                 tmp &= ~mask;
 297                 val |= tmp;
 298         }
 299
 300         return rtl2832_sdr_wr_regs(s, reg, &val, 1);
 301 }
 302
 303 #if 0
 304 /* read single register with mask */
 305 static int rtl2832_sdr_rd_reg_mask(struct rtl2832_sdr_state *s, u16 reg,
 306                 u8 *val, u8 mask)
 307 {
 308         int ret, i;
 309         u8 tmp;
 310
 311         ret = rtl2832_sdr_rd_regs(s, reg, &tmp, 1);
 312         if (ret)
 313                 return ret;
 314
 315         tmp &= mask;
 316
 317         /* find position of the first bit */
 318         for (i = 0; i < 8; i++) {
 319                 if ((mask >> i) & 0x01)
 320                         break;
 321         }
 322         *val = tmp >> i;
 323
 324         return 0;
 325 }
 326 #endif
 327
 328 /* Private functions */
 329 static struct rtl2832_sdr_frame_buf *rtl2832_sdr_get_next_fill_buf(
 330                 struct rtl2832_sdr_state *s)
 331 {
 332         unsigned long flags = 0;
 333         struct rtl2832_sdr_frame_buf *buf = NULL;
 334
 335         spin_lock_irqsave(&s->queued_bufs_lock, flags);
 336         if (list_empty(&s->queued_bufs))
 337                 goto leave;
 338
 339         buf = list_entry(s->queued_bufs.next,
 340                         struct rtl2832_sdr_frame_buf, list);
 341         list_del(&buf->list);
 342 leave:
 343         spin_unlock_irqrestore(&s->queued_bufs_lock, flags);
 344         return buf;
 345 }
 346
 347 static unsigned int rtl2832_sdr_convert_stream(struct rtl2832_sdr_state *s,
 348                 void *dst, const u8 *src, unsigned int src_len)
 349 {
 350         unsigned int dst_len;
 351
 352         if (s->pixelformat ==  V4L2_SDR_FMT_CU8) {
 353                 /* native stream, no need to convert */
 354                 memcpy(dst, src, src_len);
 355                 dst_len = src_len;
 356         } else if (s->pixelformat == V4L2_SDR_FMT_CU16LE) {
 357                 /* convert u8 to u16 */
 358                 unsigned int i;
 359                 u16 *u16dst = dst;
 360
 361                 for (i = 0; i < src_len; i++)
 362                         *u16dst++ = (src[i] << 8) | (src[i] >> 0);
 363                 dst_len = 2 * src_len;
 364         } else {
 365                 dst_len = 0;
 366         }
 367
 368         /* calculate samping rate and output it in 10 seconds intervals */
 369         if (unlikely(time_is_before_jiffies(s->jiffies_next))) {
 370 #define MSECS 10000UL
 371                 unsigned int samples = s->sample - s->sample_measured;
 372
 373                 s->jiffies_next = jiffies + msecs_to_jiffies(MSECS);
 374                 s->sample_measured = s->sample;
 375                 dev_dbg(&s->udev->dev,
 376                                 "slen=%d samples=%u msecs=%lu sampling rate=%lu\n",
 377                                 src_len, samples, MSECS,
 378                                 samples * 1000UL / MSECS);
 379         }
 380
 381         /* total number of I+Q pairs */
 382         s->sample += src_len / 2;
 383
 384         return dst_len;
 385 }
 386
 387 /*
 388  * This gets called for the bulk stream pipe. This is done in interrupt
 389  * time, so it has to be fast, not crash, and not stall. Neat.
 390  */
 391 static void rtl2832_sdr_urb_complete(struct urb *urb)
 392 {
 393         struct rtl2832_sdr_state *s = urb->context;
 394         struct rtl2832_sdr_frame_buf *fbuf;
 395
 396         dev_dbg_ratelimited(&s->udev->dev,
 397                         "%s: status=%d length=%d/%d errors=%d\n",
 398                         __func__, urb->status, urb->actual_length,
 399                         urb->transfer_buffer_length, urb->error_count);
 400
 401         switch (urb->status) {
 402         case 0:             /* success */
 403         case -ETIMEDOUT:    /* NAK */
 404                 break;
 405         case -ECONNRESET:   /* kill */
 406         case -ENOENT:
 407         case -ESHUTDOWN:
 408                 return;
 409         default:            /* error */
 410                 dev_err_ratelimited(&s->udev->dev, "urb failed=%d\n",
 411                                 urb->status);
 412                 break;
 413         }
 414
 415         if (likely(urb->actual_length > 0)) {
 416                 void *ptr;
 417                 unsigned int len;
 418                 /* get free framebuffer */
 419                 fbuf = rtl2832_sdr_get_next_fill_buf(s);
 420                 if (unlikely(fbuf == NULL)) {
 421                         s->vb_full++;
 422                         dev_notice_ratelimited(&s->udev->dev,
 423                                         "videobuf is full, %d packets dropped\n",
 424                                         s->vb_full);
 425                         goto skip;
 426                 }
 427
 428                 /* fill framebuffer */
 429                 ptr = vb2_plane_vaddr(&fbuf->vb, 0);
 430                 len = rtl2832_sdr_convert_stream(s, ptr, urb->transfer_buffer,
 431                                 urb->actual_length);
 432                 vb2_set_plane_payload(&fbuf->vb, 0, len);
 433                 v4l2_get_timestamp(&fbuf->vb.v4l2_buf.timestamp);
 434                 fbuf->vb.v4l2_buf.sequence = s->sequence++;
 435                 vb2_buffer_done(&fbuf->vb, VB2_BUF_STATE_DONE);
 436         }
 437 skip:
 438         usb_submit_urb(urb, GFP_ATOMIC);
 439 }
 440
 441 static int rtl2832_sdr_kill_urbs(struct rtl2832_sdr_state *s)
 442 {
 443         int i;
 444
 445         for (i = s->urbs_submitted - 1; i >= 0; i--) {
 446                 dev_dbg(&s->udev->dev, "%s: kill urb=%d\n", __func__, i);
 447                 /* stop the URB */
 448                 usb_kill_urb(s->urb_list[i]);
 449         }
 450         s->urbs_submitted = 0;
 451
 452         return 0;
 453 }
 454
 455 static int rtl2832_sdr_submit_urbs(struct rtl2832_sdr_state *s)
 456 {
 457         int i, ret;
 458
 459         for (i = 0; i < s->urbs_initialized; i++) {
 460                 dev_dbg(&s->udev->dev, "%s: submit urb=%d\n", __func__, i);
 461                 ret = usb_submit_urb(s->urb_list[i], GFP_ATOMIC);
 462                 if (ret) {
 463                         dev_err(&s->udev->dev,
 464                                         "Could not submit urb no. %d - get them all back\n",
 465                                         i);
 466                         rtl2832_sdr_kill_urbs(s);
 467                         return ret;
 468                 }
 469                 s->urbs_submitted++;
 470         }
 471
 472         return 0;
 473 }
 474
 475 static int rtl2832_sdr_free_stream_bufs(struct rtl2832_sdr_state *s)
 476 {
 477         if (s->flags & USB_STATE_URB_BUF) {
 478                 while (s->buf_num) {
 479                         s->buf_num--;
 480                         dev_dbg(&s->udev->dev, "%s: free buf=%d\n",
 481                                         __func__, s->buf_num);
 482                         usb_free_coherent(s->udev, s->buf_size,
 483                                           s->buf_list[s->buf_num],
 484                                           s->dma_addr[s->buf_num]);
 485                 }
 486         }
 487         s->flags &= ~USB_STATE_URB_BUF;
 488
 489         return 0;
 490 }
 491
 492 static int rtl2832_sdr_alloc_stream_bufs(struct rtl2832_sdr_state *s)
 493 {
 494         s->buf_num = 0;
 495         s->buf_size = BULK_BUFFER_SIZE;
 496
 497         dev_dbg(&s->udev->dev,
 498                         "%s: all in all I will use %u bytes for streaming\n",
 499                         __func__,  MAX_BULK_BUFS * BULK_BUFFER_SIZE);
 500
 501         for (s->buf_num = 0; s->buf_num < MAX_BULK_BUFS; s->buf_num++) {
 502                 s->buf_list[s->buf_num] = usb_alloc_coherent(s->udev,
 503                                 BULK_BUFFER_SIZE, GFP_ATOMIC,
 504                                 &s->dma_addr[s->buf_num]);
 505                 if (!s->buf_list[s->buf_num]) {
 506                         dev_dbg(&s->udev->dev, "%s: alloc buf=%d failed\n",
 507                                         __func__, s->buf_num);
 508                         rtl2832_sdr_free_stream_bufs(s);
 509                         return -ENOMEM;
 510                 }
 511
 512                 dev_dbg(&s->udev->dev, "%s: alloc buf=%d %p (dma %llu)\n",
 513                                 __func__, s->buf_num,
 514                                 s->buf_list[s->buf_num],
 515                                 (long long)s->dma_addr[s->buf_num]);
 516                 s->flags |= USB_STATE_URB_BUF;
 517         }
 518
 519         return 0;
 520 }
 521
 522 static int rtl2832_sdr_free_urbs(struct rtl2832_sdr_state *s)
 523 {
 524         int i;
 525
 526         rtl2832_sdr_kill_urbs(s);
 527
 528         for (i = s->urbs_initialized - 1; i >= 0; i--) {
 529                 if (s->urb_list[i]) {
 530                         dev_dbg(&s->udev->dev, "%s: free urb=%d\n",
 531                                         __func__, i);
 532                         /* free the URBs */
 533                         usb_free_urb(s->urb_list[i]);
 534                 }
 535         }
 536         s->urbs_initialized = 0;
 537
 538         return 0;
 539 }
 540
 541 static int rtl2832_sdr_alloc_urbs(struct rtl2832_sdr_state *s)
 542 {
 543         int i, j;
 544
 545         /* allocate the URBs */
 546         for (i = 0; i < MAX_BULK_BUFS; i++) {
 547                 dev_dbg(&s->udev->dev, "%s: alloc urb=%d\n", __func__, i);
 548                 s->urb_list[i] = usb_alloc_urb(0, GFP_ATOMIC);
 549                 if (!s->urb_list[i]) {
 550                         dev_dbg(&s->udev->dev, "%s: failed\n", __func__);
 551                         for (j = 0; j < i; j++)
 552                                 usb_free_urb(s->urb_list[j]);
 553                         return -ENOMEM;
 554                 }
 555                 usb_fill_bulk_urb(s->urb_list[i],
 556                                 s->udev,
 557                                 usb_rcvbulkpipe(s->udev, 0x81),
 558                                 s->buf_list[i],
 559                                 BULK_BUFFER_SIZE,
 560                                 rtl2832_sdr_urb_complete, s);
 561
 562                 s->urb_list[i]->transfer_flags = URB_NO_TRANSFER_DMA_MAP;
 563                 s->urb_list[i]->transfer_dma = s->dma_addr[i];
 564                 s->urbs_initialized++;
 565         }
 566
 567         return 0;
 568 }
 569
 570 /* Must be called with vb_queue_lock hold */
 571 static void rtl2832_sdr_cleanup_queued_bufs(struct rtl2832_sdr_state *s)
 572 {
 573         unsigned long flags = 0;
 574
 575         dev_dbg(&s->udev->dev, "%s:\n", __func__);
 576
 577         spin_lock_irqsave(&s->queued_bufs_lock, flags);
 578         while (!list_empty(&s->queued_bufs)) {
 579                 struct rtl2832_sdr_frame_buf *buf;
 580
 581                 buf = list_entry(s->queued_bufs.next,
 582                                 struct rtl2832_sdr_frame_buf, list);
 583                 list_del(&buf->list);
 584                 vb2_buffer_done(&buf->vb, VB2_BUF_STATE_ERROR);
 585         }
 586         spin_unlock_irqrestore(&s->queued_bufs_lock, flags);
 587 }
 588
 589 /* The user yanked out the cable... */
 590 static void rtl2832_sdr_release_sec(struct dvb_frontend *fe)
 591 {
 592         struct rtl2832_sdr_state *s = fe->sec_priv;
 593
 594         dev_dbg(&s->udev->dev, "%s:\n", __func__);
 595
 596         mutex_lock(&s->vb_queue_lock);
 597         mutex_lock(&s->v4l2_lock);
 598         /* No need to keep the urbs around after disconnection */
 599         s->udev = NULL;
 600
 601         v4l2_device_disconnect(&s->v4l2_dev);
 602         video_unregister_device(&s->vdev);
 603         mutex_unlock(&s->v4l2_lock);
 604         mutex_unlock(&s->vb_queue_lock);
 605
 606         v4l2_device_put(&s->v4l2_dev);
 607
 608         fe->sec_priv = NULL;
 609 }
 610
 611 static int rtl2832_sdr_querycap(struct file *file, void *fh,
 612                 struct v4l2_capability *cap)
 613 {
 614         struct rtl2832_sdr_state *s = video_drvdata(file);
 615
 616         dev_dbg(&s->udev->dev, "%s:\n", __func__);
 617
 618         strlcpy(cap->driver, KBUILD_MODNAME, sizeof(cap->driver));
 619         strlcpy(cap->card, s->vdev.name, sizeof(cap->card));
 620         usb_make_path(s->udev, cap->bus_info, sizeof(cap->bus_info));
 621         cap->device_caps = V4L2_CAP_SDR_CAPTURE | V4L2_CAP_STREAMING |
 622                         V4L2_CAP_READWRITE | V4L2_CAP_TUNER;
 623         cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
 624         return 0;
 625 }
 626
 627 /* Videobuf2 operations */
 628 static int rtl2832_sdr_queue_setup(struct vb2_queue *vq,
 629                 const struct v4l2_format *fmt, unsigned int *nbuffers,
 630                 unsigned int *nplanes, unsigned int sizes[], void *alloc_ctxs[])
 631 {
 632         struct rtl2832_sdr_state *s = vb2_get_drv_priv(vq);
 633
 634         dev_dbg(&s->udev->dev, "%s: *nbuffers=%d\n", __func__, *nbuffers);
 635
 636         /* Need at least 8 buffers */
 637         if (vq->num_buffers + *nbuffers < 8)
 638                 *nbuffers = 8 - vq->num_buffers;
 639         *nplanes = 1;
 640         sizes[0] = PAGE_ALIGN(s->buffersize);
 641         dev_dbg(&s->udev->dev, "%s: nbuffers=%d sizes[0]=%d\n",
 642                         __func__, *nbuffers, sizes[0]);
 643         return 0;
 644 }
 645
 646 static int rtl2832_sdr_buf_prepare(struct vb2_buffer *vb)
 647 {
 648         struct rtl2832_sdr_state *s = vb2_get_drv_priv(vb->vb2_queue);
 649
 650         /* Don't allow queing new buffers after device disconnection */
 651         if (!s->udev)
 652                 return -ENODEV;
 653
 654         return 0;
 655 }
 656
 657 static void rtl2832_sdr_buf_queue(struct vb2_buffer *vb)
 658 {
 659         struct rtl2832_sdr_state *s = vb2_get_drv_priv(vb->vb2_queue);
 660         struct rtl2832_sdr_frame_buf *buf =
 661                         container_of(vb, struct rtl2832_sdr_frame_buf, vb);
 662         unsigned long flags = 0;
 663
 664         /* Check the device has not disconnected between prep and queuing */
 665         if (!s->udev) {
 666                 vb2_buffer_done(&buf->vb, VB2_BUF_STATE_ERROR);
 667                 return;
 668         }
 669
 670         spin_lock_irqsave(&s->queued_bufs_lock, flags);
 671         list_add_tail(&buf->list, &s->queued_bufs);
 672         spin_unlock_irqrestore(&s->queued_bufs_lock, flags);
 673 }
 674
 675 static int rtl2832_sdr_set_adc(struct rtl2832_sdr_state *s)
 676 {
 677         struct dvb_frontend *fe = s->fe;
 678         int ret;
 679         unsigned int f_sr, f_if;
 680         u8 buf[4], u8tmp1, u8tmp2;
 681         u64 u64tmp;
 682         u32 u32tmp;
 683
 684         dev_dbg(&s->udev->dev, "%s: f_adc=%u\n", __func__, s->f_adc);
 685
 686         if (!test_bit(POWER_ON, &s->flags))
 687                 return 0;
 688
 689         if (s->f_adc == 0)
 690                 return 0;
 691
 692         f_sr = s->f_adc;
 693
 694         ret = rtl2832_sdr_wr_regs(s, 0x13e, "\x00\x00", 2);
 695         if (ret)
 696                 goto err;
 697
 698         ret = rtl2832_sdr_wr_regs(s, 0x115, "\x00\x00\x00\x00", 4);
 699         if (ret)
 700                 goto err;
 701
 702         /* get IF from tuner */
 703         if (fe->ops.tuner_ops.get_if_frequency)
 704                 ret = fe->ops.tuner_ops.get_if_frequency(fe, &f_if);
 705         else
 706                 ret = -EINVAL;
 707
 708         if (ret)
 709                 goto err;
 710
 711         /* program IF */
 712         u64tmp = f_if % s->cfg->xtal;
 713         u64tmp *= 0x400000;
 714         u64tmp = div_u64(u64tmp, s->cfg->xtal);
 715         u64tmp = -u64tmp;
 716         u32tmp = u64tmp & 0x3fffff;
 717
 718         dev_dbg(&s->udev->dev, "%s: f_if=%u if_ctl=%08x\n",
 719                         __func__, f_if, u32tmp);
 720
 721         buf[0] = (u32tmp >> 16) & 0xff;
 722         buf[1] = (u32tmp >>  8) & 0xff;
 723         buf[2] = (u32tmp >>  0) & 0xff;
 724
 725         ret = rtl2832_sdr_wr_regs(s, 0x119, buf, 3);
 726         if (ret)
 727                 goto err;
 728
 729         /* BB / IF mode */
 730         /* POR: 0x1b1=0x1f, 0x008=0x0d, 0x006=0x80 */
 731         if (f_if) {
 732                 u8tmp1 = 0x1a; /* disable Zero-IF */
 733                 u8tmp2 = 0x8d; /* enable ADC I */
 734         } else {
 735                 u8tmp1 = 0x1b; /* enable Zero-IF, DC, IQ */
 736                 u8tmp2 = 0xcd; /* enable ADC I, ADC Q */
 737         }
 738
 739         ret = rtl2832_sdr_wr_reg(s, 0x1b1, u8tmp1);
 740         if (ret)
 741                 goto err;
 742
 743         ret = rtl2832_sdr_wr_reg(s, 0x008, u8tmp2);
 744         if (ret)
 745                 goto err;
 746
 747         ret = rtl2832_sdr_wr_reg(s, 0x006, 0x80);
 748         if (ret)
 749                 goto err;
 750
 751         /* program sampling rate (resampling down) */
 752         u32tmp = div_u64(s->cfg->xtal * 0x400000ULL, f_sr * 4U);
 753         u32tmp <<= 2;
 754         buf[0] = (u32tmp >> 24) & 0xff;
 755         buf[1] = (u32tmp >> 16) & 0xff;
 756         buf[2] = (u32tmp >>  8) & 0xff;
 757         buf[3] = (u32tmp >>  0) & 0xff;
 758         ret = rtl2832_sdr_wr_regs(s, 0x19f, buf, 4);
 759         if (ret)
 760                 goto err;
 761
 762         /* low-pass filter */
 763         ret = rtl2832_sdr_wr_regs(s, 0x11c,
 764                         "\xca\xdc\xd7\xd8\xe0\xf2\x0e\x35\x06\x50\x9c\x0d\x71\x11\x14\x71\x74\x19\x41\xa5",
 765                         20);
 766         if (ret)
 767                 goto err;
 768
 769         ret = rtl2832_sdr_wr_regs(s, 0x017, "\x11\x10", 2);
 770         if (ret)
 771                 goto err;
 772
 773         /* mode */
 774         ret = rtl2832_sdr_wr_regs(s, 0x019, "\x05", 1);
 775         if (ret)
 776                 goto err;
 777
 778         ret = rtl2832_sdr_wr_regs(s, 0x01a, "\x1b\x16\x0d\x06\x01\xff", 6);
 779         if (ret)
 780                 goto err;
 781
 782         /* FSM */
 783         ret = rtl2832_sdr_wr_regs(s, 0x192, "\x00\xf0\x0f", 3);
 784         if (ret)
 785                 goto err;
 786
 787         /* PID filter */
 788         ret = rtl2832_sdr_wr_regs(s, 0x061, "\x60", 1);
 789         if (ret)
 790                 goto err;
 791
 792         /* used RF tuner based settings */
 793         switch (s->cfg->tuner) {
 794         case RTL2832_TUNER_E4000:
 795                 ret = rtl2832_sdr_wr_regs(s, 0x112, "\x5a", 1);
 796                 ret = rtl2832_sdr_wr_regs(s, 0x102, "\x40", 1);
 797                 ret = rtl2832_sdr_wr_regs(s, 0x103, "\x5a", 1);
 798                 ret = rtl2832_sdr_wr_regs(s, 0x1c7, "\x30", 1);
 799                 ret = rtl2832_sdr_wr_regs(s, 0x104, "\xd0", 1);
 800                 ret = rtl2832_sdr_wr_regs(s, 0x105, "\xbe", 1);
 801                 ret = rtl2832_sdr_wr_regs(s, 0x1c8, "\x18", 1);
 802                 ret = rtl2832_sdr_wr_regs(s, 0x106, "\x35", 1);
 803                 ret = rtl2832_sdr_wr_regs(s, 0x1c9, "\x21", 1);
 804                 ret = rtl2832_sdr_wr_regs(s, 0x1ca, "\x21", 1);
 805                 ret = rtl2832_sdr_wr_regs(s, 0x1cb, "\x00", 1);
 806                 ret = rtl2832_sdr_wr_regs(s, 0x107, "\x40", 1);
 807                 ret = rtl2832_sdr_wr_regs(s, 0x1cd, "\x10", 1);
 808                 ret = rtl2832_sdr_wr_regs(s, 0x1ce, "\x10", 1);
 809                 ret = rtl2832_sdr_wr_regs(s, 0x108, "\x80", 1);
 810                 ret = rtl2832_sdr_wr_regs(s, 0x109, "\x7f", 1);
 811                 ret = rtl2832_sdr_wr_regs(s, 0x10a, "\x80", 1);
 812                 ret = rtl2832_sdr_wr_regs(s, 0x10b, "\x7f", 1);
 813                 ret = rtl2832_sdr_wr_regs(s, 0x00e, "\xfc", 1);
 814                 ret = rtl2832_sdr_wr_regs(s, 0x00e, "\xfc", 1);
 815                 ret = rtl2832_sdr_wr_regs(s, 0x011, "\xd4", 1);
 816                 ret = rtl2832_sdr_wr_regs(s, 0x1e5, "\xf0", 1);
 817                 ret = rtl2832_sdr_wr_regs(s, 0x1d9, "\x00", 1);
 818                 ret = rtl2832_sdr_wr_regs(s, 0x1db, "\x00", 1);
 819                 ret = rtl2832_sdr_wr_regs(s, 0x1dd, "\x14", 1);
 820                 ret = rtl2832_sdr_wr_regs(s, 0x1de, "\xec", 1);
 821                 ret = rtl2832_sdr_wr_regs(s, 0x1d8, "\x0c", 1);
 822                 ret = rtl2832_sdr_wr_regs(s, 0x1e6, "\x02", 1);
 823                 ret = rtl2832_sdr_wr_regs(s, 0x1d7, "\x09", 1);
 824                 ret = rtl2832_sdr_wr_regs(s, 0x00d, "\x83", 1);
 825                 ret = rtl2832_sdr_wr_regs(s, 0x010, "\x49", 1);
 826                 ret = rtl2832_sdr_wr_regs(s, 0x00d, "\x87", 1);
 827                 ret = rtl2832_sdr_wr_regs(s, 0x00d, "\x85", 1);
 828                 ret = rtl2832_sdr_wr_regs(s, 0x013, "\x02", 1);
 829                 break;
 830         case RTL2832_TUNER_FC0012:
 831         case RTL2832_TUNER_FC0013:
 832                 ret = rtl2832_sdr_wr_regs(s, 0x112, "\x5a", 1);
 833                 ret = rtl2832_sdr_wr_regs(s, 0x102, "\x40", 1);
 834                 ret = rtl2832_sdr_wr_regs(s, 0x103, "\x5a", 1);
 835                 ret = rtl2832_sdr_wr_regs(s, 0x1c7, "\x2c", 1);
 836                 ret = rtl2832_sdr_wr_regs(s, 0x104, "\xcc", 1);
 837                 ret = rtl2832_sdr_wr_regs(s, 0x105, "\xbe", 1);
 838                 ret = rtl2832_sdr_wr_regs(s, 0x1c8, "\x16", 1);
 839                 ret = rtl2832_sdr_wr_regs(s, 0x106, "\x35", 1);
 840                 ret = rtl2832_sdr_wr_regs(s, 0x1c9, "\x21", 1);
 841                 ret = rtl2832_sdr_wr_regs(s, 0x1ca, "\x21", 1);
 842                 ret = rtl2832_sdr_wr_regs(s, 0x1cb, "\x00", 1);
 843                 ret = rtl2832_sdr_wr_regs(s, 0x107, "\x40", 1);
 844                 ret = rtl2832_sdr_wr_regs(s, 0x1cd, "\x10", 1);
 845                 ret = rtl2832_sdr_wr_regs(s, 0x1ce, "\x10", 1);
 846                 ret = rtl2832_sdr_wr_regs(s, 0x108, "\x80", 1);
 847                 ret = rtl2832_sdr_wr_regs(s, 0x109, "\x7f", 1);
 848                 ret = rtl2832_sdr_wr_regs(s, 0x10a, "\x80", 1);
 849                 ret = rtl2832_sdr_wr_regs(s, 0x10b, "\x7f", 1);
 850                 ret = rtl2832_sdr_wr_regs(s, 0x00e, "\xfc", 1);
 851                 ret = rtl2832_sdr_wr_regs(s, 0x00e, "\xfc", 1);
 852                 ret = rtl2832_sdr_wr_regs(s, 0x011, "\xe9\xbf", 2);
 853                 ret = rtl2832_sdr_wr_regs(s, 0x1e5, "\xf0", 1);
 854                 ret = rtl2832_sdr_wr_regs(s, 0x1d9, "\x00", 1);
 855                 ret = rtl2832_sdr_wr_regs(s, 0x1db, "\x00", 1);
 856                 ret = rtl2832_sdr_wr_regs(s, 0x1dd, "\x11", 1);
 857                 ret = rtl2832_sdr_wr_regs(s, 0x1de, "\xef", 1);
 858                 ret = rtl2832_sdr_wr_regs(s, 0x1d8, "\x0c", 1);
 859                 ret = rtl2832_sdr_wr_regs(s, 0x1e6, "\x02", 1);
 860                 ret = rtl2832_sdr_wr_regs(s, 0x1d7, "\x09", 1);
 861                 break;
 862         case RTL2832_TUNER_R820T:
 863                 ret = rtl2832_sdr_wr_regs(s, 0x112, "\x5a", 1);
 864                 ret = rtl2832_sdr_wr_regs(s, 0x102, "\x40", 1);
 865                 ret = rtl2832_sdr_wr_regs(s, 0x115, "\x01", 1);
 866                 ret = rtl2832_sdr_wr_regs(s, 0x103, "\x80", 1);
 867                 ret = rtl2832_sdr_wr_regs(s, 0x1c7, "\x24", 1);
 868                 ret = rtl2832_sdr_wr_regs(s, 0x104, "\xcc", 1);
 869                 ret = rtl2832_sdr_wr_regs(s, 0x105, "\xbe", 1);
 870                 ret = rtl2832_sdr_wr_regs(s, 0x1c8, "\x14", 1);
 871                 ret = rtl2832_sdr_wr_regs(s, 0x106, "\x35", 1);
 872                 ret = rtl2832_sdr_wr_regs(s, 0x1c9, "\x21", 1);
 873                 ret = rtl2832_sdr_wr_regs(s, 0x1ca, "\x21", 1);
 874                 ret = rtl2832_sdr_wr_regs(s, 0x1cb, "\x00", 1);
 875                 ret = rtl2832_sdr_wr_regs(s, 0x107, "\x40", 1);
 876                 ret = rtl2832_sdr_wr_regs(s, 0x1cd, "\x10", 1);
 877                 ret = rtl2832_sdr_wr_regs(s, 0x1ce, "\x10", 1);
 878                 ret = rtl2832_sdr_wr_regs(s, 0x108, "\x80", 1);
 879                 ret = rtl2832_sdr_wr_regs(s, 0x109, "\x7f", 1);
 880                 ret = rtl2832_sdr_wr_regs(s, 0x10a, "\x80", 1);
 881                 ret = rtl2832_sdr_wr_regs(s, 0x10b, "\x7f", 1);
 882                 ret = rtl2832_sdr_wr_regs(s, 0x00e, "\xfc", 1);
 883                 ret = rtl2832_sdr_wr_regs(s, 0x00e, "\xfc", 1);
 884                 ret = rtl2832_sdr_wr_regs(s, 0x011, "\xf4", 1);
 885                 break;
 886         default:
 887                 dev_notice(&s->udev->dev, "Unsupported tuner\n");
 888         }
 889
 890         /* software reset */
 891         ret = rtl2832_sdr_wr_reg_mask(s, 0x101, 0x04, 0x04);
 892         if (ret)
 893                 goto err;
 894
 895         ret = rtl2832_sdr_wr_reg_mask(s, 0x101, 0x00, 0x04);
 896         if (ret)
 897                 goto err;
 898 err:
 899         return ret;
 900 };
 901
 902 static void rtl2832_sdr_unset_adc(struct rtl2832_sdr_state *s)
 903 {
 904         int ret;
 905
 906         dev_dbg(&s->udev->dev, "%s:\n", __func__);
 907
 908         /* PID filter */
 909         ret = rtl2832_sdr_wr_regs(s, 0x061, "\xe0", 1);
 910         if (ret)
 911                 goto err;
 912
 913         /* mode */
 914         ret = rtl2832_sdr_wr_regs(s, 0x019, "\x20", 1);
 915         if (ret)
 916                 goto err;
 917
 918         ret = rtl2832_sdr_wr_regs(s, 0x017, "\x11\x10", 2);
 919         if (ret)
 920                 goto err;
 921
 922         /* FSM */
 923         ret = rtl2832_sdr_wr_regs(s, 0x192, "\x00\x0f\xff", 3);
 924         if (ret)
 925                 goto err;
 926
 927         ret = rtl2832_sdr_wr_regs(s, 0x13e, "\x40\x00", 2);
 928         if (ret)
 929                 goto err;
 930
 931         ret = rtl2832_sdr_wr_regs(s, 0x115, "\x06\x3f\xce\xcc", 4);
 932         if (ret)
 933                 goto err;
 934 err:
 935         return;
 936 };
 937
 938 static int rtl2832_sdr_set_tuner_freq(struct rtl2832_sdr_state *s)
 939 {
 940         struct dvb_frontend *fe = s->fe;
 941         struct dtv_frontend_properties *c = &fe->dtv_property_cache;
 942         struct v4l2_ctrl *bandwidth_auto;
 943         struct v4l2_ctrl *bandwidth;
 944
 945         /*
 946          * tuner RF (Hz)
 947          */
 948         if (s->f_tuner == 0)
 949                 return 0;
 950
 951         /*
 952          * bandwidth (Hz)
 953          */
 954         bandwidth_auto = v4l2_ctrl_find(&s->hdl,
 955                                         V4L2_CID_RF_TUNER_BANDWIDTH_AUTO);
 956         bandwidth = v4l2_ctrl_find(&s->hdl, V4L2_CID_RF_TUNER_BANDWIDTH);
 957         if (v4l2_ctrl_g_ctrl(bandwidth_auto)) {
 958                 c->bandwidth_hz = s->f_adc;
 959                 v4l2_ctrl_s_ctrl(bandwidth, s->f_adc);
 960         } else {
 961                 c->bandwidth_hz = v4l2_ctrl_g_ctrl(bandwidth);
 962         }
 963
 964         c->frequency = s->f_tuner;
 965         c->delivery_system = SYS_DVBT;
 966
 967         dev_dbg(&s->udev->dev, "%s: frequency=%u bandwidth=%d\n",
 968                         __func__, c->frequency, c->bandwidth_hz);
 969
 970         if (!test_bit(POWER_ON, &s->flags))
 971                 return 0;
 972
 973         if (fe->ops.tuner_ops.set_params)
 974                 fe->ops.tuner_ops.set_params(fe);
 975
 976         return 0;
 977 };
 978
 979 static int rtl2832_sdr_set_tuner(struct rtl2832_sdr_state *s)
 980 {
 981         struct dvb_frontend *fe = s->fe;
 982
 983         dev_dbg(&s->udev->dev, "%s:\n", __func__);
 984
 985         if (fe->ops.tuner_ops.init)
 986                 fe->ops.tuner_ops.init(fe);
 987
 988         return 0;
 989 };
 990
 991 static void rtl2832_sdr_unset_tuner(struct rtl2832_sdr_state *s)
 992 {
 993         struct dvb_frontend *fe = s->fe;
 994
 995         dev_dbg(&s->udev->dev, "%s:\n", __func__);
 996
 997         if (fe->ops.tuner_ops.sleep)
 998                 fe->ops.tuner_ops.sleep(fe);
 999
1000         return;
1001 };
1002
1003 static int rtl2832_sdr_start_streaming(struct vb2_queue *vq, unsigned int count)
1004 {
1005         struct rtl2832_sdr_state *s = vb2_get_drv_priv(vq);
1006         int ret;
1007
1008         dev_dbg(&s->udev->dev, "%s:\n", __func__);
1009
1010         if (!s->udev)
1011                 return -ENODEV;
1012
1013         if (mutex_lock_interruptible(&s->v4l2_lock))
1014                 return -ERESTARTSYS;
1015
1016         if (s->d->props->power_ctrl)
1017                 s->d->props->power_ctrl(s->d, 1);
1018
1019         set_bit(POWER_ON, &s->flags);
1020
1021         ret = rtl2832_sdr_set_tuner(s);
1022         if (ret)
1023                 goto err;
1024
1025         ret = rtl2832_sdr_set_tuner_freq(s);
1026         if (ret)
1027                 goto err;
1028
1029         ret = rtl2832_sdr_set_adc(s);
1030         if (ret)
1031                 goto err;
1032
1033         ret = rtl2832_sdr_alloc_stream_bufs(s);
1034         if (ret)
1035                 goto err;
1036
1037         ret = rtl2832_sdr_alloc_urbs(s);
1038         if (ret)
1039                 goto err;
1040
1041         s->sequence = 0;
1042
1043         ret = rtl2832_sdr_submit_urbs(s);
1044         if (ret)
1045                 goto err;
1046
1047 err:
1048         mutex_unlock(&s->v4l2_lock);
1049
1050         return ret;
1051 }
1052
1053 static void rtl2832_sdr_stop_streaming(struct vb2_queue *vq)
1054 {
1055         struct rtl2832_sdr_state *s = vb2_get_drv_priv(vq);
1056
1057         dev_dbg(&s->udev->dev, "%s:\n", __func__);
1058
1059         mutex_lock(&s->v4l2_lock);
1060
1061         rtl2832_sdr_kill_urbs(s);
1062         rtl2832_sdr_free_urbs(s);
1063         rtl2832_sdr_free_stream_bufs(s);
1064         rtl2832_sdr_cleanup_queued_bufs(s);
1065         rtl2832_sdr_unset_adc(s);
1066         rtl2832_sdr_unset_tuner(s);
1067
1068         clear_bit(POWER_ON, &s->flags);
1069
1070         if (s->d->props->power_ctrl)
1071                 s->d->props->power_ctrl(s->d, 0);
1072
1073         mutex_unlock(&s->v4l2_lock);
1074 }
1075
1076 static struct vb2_ops rtl2832_sdr_vb2_ops = {
1077         .queue_setup            = rtl2832_sdr_queue_setup,
1078         .buf_prepare            = rtl2832_sdr_buf_prepare,
1079         .buf_queue              = rtl2832_sdr_buf_queue,
1080         .start_streaming        = rtl2832_sdr_start_streaming,
1081         .stop_streaming         = rtl2832_sdr_stop_streaming,
1082         .wait_prepare           = vb2_ops_wait_prepare,
1083         .wait_finish            = vb2_ops_wait_finish,
1084 };
1085
1086 static int rtl2832_sdr_g_tuner(struct file *file, void *priv,
1087                 struct v4l2_tuner *v)
1088 {
1089         struct rtl2832_sdr_state *s = video_drvdata(file);
1090
1091         dev_dbg(&s->udev->dev, "%s: index=%d type=%d\n",
1092                         __func__, v->index, v->type);
1093
1094         if (v->index == 0) {
1095                 strlcpy(v->name, "ADC: Realtek RTL2832", sizeof(v->name));
1096                 v->type = V4L2_TUNER_ADC;
1097                 v->capability = V4L2_TUNER_CAP_1HZ | V4L2_TUNER_CAP_FREQ_BANDS;
1098                 v->rangelow =   300000;
1099                 v->rangehigh = 3200000;
1100         } else if (v->index == 1) {
1101                 strlcpy(v->name, "RF: <unknown>", sizeof(v->name));
1102                 v->type = V4L2_TUNER_RF;
1103                 v->capability = V4L2_TUNER_CAP_1HZ | V4L2_TUNER_CAP_FREQ_BANDS;
1104                 v->rangelow =    50000000;
1105                 v->rangehigh = 2000000000;
1106         } else {
1107                 return -EINVAL;
1108         }
1109
1110         return 0;
1111 }
1112
1113 static int rtl2832_sdr_s_tuner(struct file *file, void *priv,
1114                 const struct v4l2_tuner *v)
1115 {
1116         struct rtl2832_sdr_state *s = video_drvdata(file);
1117
1118         dev_dbg(&s->udev->dev, "%s:\n", __func__);
1119
1120         if (v->index > 1)
1121                 return -EINVAL;
1122         return 0;
1123 }
1124
1125 static int rtl2832_sdr_enum_freq_bands(struct file *file, void *priv,
1126                 struct v4l2_frequency_band *band)
1127 {
1128         struct rtl2832_sdr_state *s = video_drvdata(file);
1129
1130         dev_dbg(&s->udev->dev, "%s: tuner=%d type=%d index=%d\n",
1131                         __func__, band->tuner, band->type, band->index);
1132
1133         if (band->tuner == 0) {
1134                 if (band->index >= ARRAY_SIZE(bands_adc))
1135                         return -EINVAL;
1136
1137                 *band = bands_adc[band->index];
1138         } else if (band->tuner == 1) {
1139                 if (band->index >= ARRAY_SIZE(bands_fm))
1140                         return -EINVAL;
1141
1142                 *band = bands_fm[band->index];
1143         } else {
1144                 return -EINVAL;
1145         }
1146
1147         return 0;
1148 }
1149
1150 static int rtl2832_sdr_g_frequency(struct file *file, void *priv,
1151                 struct v4l2_frequency *f)
1152 {
1153         struct rtl2832_sdr_state *s = video_drvdata(file);
1154         int ret  = 0;
1155
1156         dev_dbg(&s->udev->dev, "%s: tuner=%d type=%d\n",
1157                         __func__, f->tuner, f->type);
1158
1159         if (f->tuner == 0) {
1160                 f->frequency = s->f_adc;
1161                 f->type = V4L2_TUNER_ADC;
1162         } else if (f->tuner == 1) {
1163                 f->frequency = s->f_tuner;
1164                 f->type = V4L2_TUNER_RF;
1165         } else {
1166                 return -EINVAL;
1167         }
1168
1169         return ret;
1170 }
1171
1172 static int rtl2832_sdr_s_frequency(struct file *file, void *priv,
1173                 const struct v4l2_frequency *f)
1174 {
1175         struct rtl2832_sdr_state *s = video_drvdata(file);
1176         int ret, band;
1177
1178         dev_dbg(&s->udev->dev, "%s: tuner=%d type=%d frequency=%u\n",
1179                         __func__, f->tuner, f->type, f->frequency);
1180
1181         /* ADC band midpoints */
1182         #define BAND_ADC_0 ((bands_adc[0].rangehigh + bands_adc[1].rangelow) / 2)
1183         #define BAND_ADC_1 ((bands_adc[1].rangehigh + bands_adc[2].rangelow) / 2)
1184
1185         if (f->tuner == 0 && f->type == V4L2_TUNER_ADC) {
1186                 if (f->frequency < BAND_ADC_0)
1187                         band = 0;
1188                 else if (f->frequency < BAND_ADC_1)
1189                         band = 1;
1190                 else
1191                         band = 2;
1192
1193                 s->f_adc = clamp_t(unsigned int, f->frequency,
1194                                 bands_adc[band].rangelow,
1195                                 bands_adc[band].rangehigh);
1196
1197                 dev_dbg(&s->udev->dev, "%s: ADC frequency=%u Hz\n",
1198                                 __func__, s->f_adc);
1199                 ret = rtl2832_sdr_set_adc(s);
1200         } else if (f->tuner == 1) {
1201                 s->f_tuner = clamp_t(unsigned int, f->frequency,
1202                                 bands_fm[0].rangelow,
1203                                 bands_fm[0].rangehigh);
1204                 dev_dbg(&s->udev->dev, "%s: RF frequency=%u Hz\n",
1205                                 __func__, f->frequency);
1206
1207                 ret = rtl2832_sdr_set_tuner_freq(s);
1208         } else {
1209                 ret = -EINVAL;
1210         }
1211
1212         return ret;
1213 }
1214
1215 static int rtl2832_sdr_enum_fmt_sdr_cap(struct file *file, void *priv,
1216                 struct v4l2_fmtdesc *f)
1217 {
1218         struct rtl2832_sdr_state *s = video_drvdata(file);
1219
1220         dev_dbg(&s->udev->dev, "%s:\n", __func__);
1221
1222         if (f->index >= s->num_formats)
1223                 return -EINVAL;
1224
1225         strlcpy(f->description, formats[f->index].name, sizeof(f->description));
1226         f->pixelformat = formats[f->index].pixelformat;
1227
1228         return 0;
1229 }
1230
1231 static int rtl2832_sdr_g_fmt_sdr_cap(struct file *file, void *priv,
1232                 struct v4l2_format *f)
1233 {
1234         struct rtl2832_sdr_state *s = video_drvdata(file);
1235
1236         dev_dbg(&s->udev->dev, "%s:\n", __func__);
1237
1238         f->fmt.sdr.pixelformat = s->pixelformat;
1239         f->fmt.sdr.buffersize = s->buffersize;
1240
1241         memset(f->fmt.sdr.reserved, 0, sizeof(f->fmt.sdr.reserved));
1242
1243         return 0;
1244 }
1245
1246 static int rtl2832_sdr_s_fmt_sdr_cap(struct file *file, void *priv,
1247                 struct v4l2_format *f)
1248 {
1249         struct rtl2832_sdr_state *s = video_drvdata(file);
1250         struct vb2_queue *q = &s->vb_queue;
1251         int i;
1252
1253         dev_dbg(&s->udev->dev, "%s: pixelformat fourcc %4.4s\n", __func__,
1254                         (char *)&f->fmt.sdr.pixelformat);
1255
1256         if (vb2_is_busy(q))
1257                 return -EBUSY;
1258
1259         memset(f->fmt.sdr.reserved, 0, sizeof(f->fmt.sdr.reserved));
1260         for (i = 0; i < s->num_formats; i++) {
1261                 if (formats[i].pixelformat == f->fmt.sdr.pixelformat) {
1262                         s->pixelformat = formats[i].pixelformat;
1263                         s->buffersize = formats[i].buffersize;
1264                         f->fmt.sdr.buffersize = formats[i].buffersize;
1265                         return 0;
1266                 }
1267         }
1268
1269         s->pixelformat = formats[0].pixelformat;
1270         s->buffersize = formats[0].buffersize;
1271         f->fmt.sdr.pixelformat = formats[0].pixelformat;
1272         f->fmt.sdr.buffersize = formats[0].buffersize;
1273
1274         return 0;
1275 }
1276
1277 static int rtl2832_sdr_try_fmt_sdr_cap(struct file *file, void *priv,
1278                 struct v4l2_format *f)
1279 {
1280         struct rtl2832_sdr_state *s = video_drvdata(file);
1281         int i;
1282
1283         dev_dbg(&s->udev->dev, "%s: pixelformat fourcc %4.4s\n", __func__,
1284                         (char *)&f->fmt.sdr.pixelformat);
1285
1286         memset(f->fmt.sdr.reserved, 0, sizeof(f->fmt.sdr.reserved));
1287         for (i = 0; i < s->num_formats; i++) {
1288                 if (formats[i].pixelformat == f->fmt.sdr.pixelformat) {
1289                         f->fmt.sdr.buffersize = formats[i].buffersize;
1290                         return 0;
1291                 }
1292         }
1293
1294         f->fmt.sdr.pixelformat = formats[0].pixelformat;
1295         f->fmt.sdr.buffersize = formats[0].buffersize;
1296
1297         return 0;
1298 }
1299
1300 static const struct v4l2_ioctl_ops rtl2832_sdr_ioctl_ops = {
1301         .vidioc_querycap          = rtl2832_sdr_querycap,
1302
1303         .vidioc_enum_fmt_sdr_cap  = rtl2832_sdr_enum_fmt_sdr_cap,
1304         .vidioc_g_fmt_sdr_cap     = rtl2832_sdr_g_fmt_sdr_cap,
1305         .vidioc_s_fmt_sdr_cap     = rtl2832_sdr_s_fmt_sdr_cap,
1306         .vidioc_try_fmt_sdr_cap   = rtl2832_sdr_try_fmt_sdr_cap,
1307
1308         .vidioc_reqbufs           = vb2_ioctl_reqbufs,
1309         .vidioc_create_bufs       = vb2_ioctl_create_bufs,
1310         .vidioc_prepare_buf       = vb2_ioctl_prepare_buf,
1311         .vidioc_querybuf          = vb2_ioctl_querybuf,
1312         .vidioc_qbuf              = vb2_ioctl_qbuf,
1313         .vidioc_dqbuf             = vb2_ioctl_dqbuf,
1314
1315         .vidioc_streamon          = vb2_ioctl_streamon,
1316         .vidioc_streamoff         = vb2_ioctl_streamoff,
1317
1318         .vidioc_g_tuner           = rtl2832_sdr_g_tuner,
1319         .vidioc_s_tuner           = rtl2832_sdr_s_tuner,
1320
1321         .vidioc_enum_freq_bands   = rtl2832_sdr_enum_freq_bands,
1322         .vidioc_g_frequency       = rtl2832_sdr_g_frequency,
1323         .vidioc_s_frequency       = rtl2832_sdr_s_frequency,
1324
1325         .vidioc_subscribe_event   = v4l2_ctrl_subscribe_event,
1326         .vidioc_unsubscribe_event = v4l2_event_unsubscribe,
1327         .vidioc_log_status        = v4l2_ctrl_log_status,
1328 };
1329
1330 static const struct v4l2_file_operations rtl2832_sdr_fops = {
1331         .owner                    = THIS_MODULE,
1332         .open                     = v4l2_fh_open,
1333         .release                  = vb2_fop_release,
1334         .read                     = vb2_fop_read,
1335         .poll                     = vb2_fop_poll,
1336         .mmap                     = vb2_fop_mmap,
1337         .unlocked_ioctl           = video_ioctl2,
1338 };
1339
1340 static struct video_device rtl2832_sdr_template = {
1341         .name                     = "Realtek RTL2832 SDR",
1342         .release                  = video_device_release_empty,
1343         .fops                     = &rtl2832_sdr_fops,
1344         .ioctl_ops                = &rtl2832_sdr_ioctl_ops,
1345 };
1346
1347 static int rtl2832_sdr_s_ctrl(struct v4l2_ctrl *ctrl)
1348 {
1349         struct rtl2832_sdr_state *s =
1350                         container_of(ctrl->handler, struct rtl2832_sdr_state,
1351                                         hdl);
1352         struct dvb_frontend *fe = s->fe;
1353         struct dtv_frontend_properties *c = &fe->dtv_property_cache;
1354         int ret;
1355
1356         dev_dbg(&s->udev->dev,
1357                         "%s: id=%d name=%s val=%d min=%lld max=%lld step=%lld\n",
1358                         __func__, ctrl->id, ctrl->name, ctrl->val,
1359                         ctrl->minimum, ctrl->maximum, ctrl->step);
1360
1361         switch (ctrl->id) {
1362         case V4L2_CID_RF_TUNER_BANDWIDTH_AUTO:
1363         case V4L2_CID_RF_TUNER_BANDWIDTH:
1364                 /* TODO: these controls should be moved to tuner drivers */
1365                 if (s->bandwidth_auto->val) {
1366                         /* Round towards the closest legal value */
1367                         s32 val = s->f_adc + div_u64(s->bandwidth->step, 2);
1368                         u32 offset;
1369
1370                         val = clamp_t(s32, val, s->bandwidth->minimum,
1371                                       s->bandwidth->maximum);
1372                         offset = val - s->bandwidth->minimum;
1373                         offset = s->bandwidth->step *
1374                                 div_u64(offset, s->bandwidth->step);
1375                         s->bandwidth->val = s->bandwidth->minimum + offset;
1376                 }
1377                 c->bandwidth_hz = s->bandwidth->val;
1378
1379                 if (!test_bit(POWER_ON, &s->flags))
1380                         return 0;
1381
1382                 if (fe->ops.tuner_ops.set_params)
1383                         ret = fe->ops.tuner_ops.set_params(fe);
1384                 else
1385                         ret = 0;
1386                 break;
1387         default:
1388                 ret = -EINVAL;
1389         }
1390
1391         return ret;
1392 }
1393
1394 static const struct v4l2_ctrl_ops rtl2832_sdr_ctrl_ops = {
1395         .s_ctrl = rtl2832_sdr_s_ctrl,
1396 };
1397
1398 static void rtl2832_sdr_video_release(struct v4l2_device *v)
1399 {
1400         struct rtl2832_sdr_state *s =
1401                         container_of(v, struct rtl2832_sdr_state, v4l2_dev);
1402
1403         v4l2_ctrl_handler_free(&s->hdl);
1404         v4l2_device_unregister(&s->v4l2_dev);
1405         kfree(s);
1406 }
1407
1408 struct dvb_frontend *rtl2832_sdr_attach(struct dvb_frontend *fe,
1409                 struct i2c_adapter *i2c, const struct rtl2832_config *cfg,
1410                 struct v4l2_subdev *sd)
1411 {
1412         int ret;
1413         struct rtl2832_sdr_state *s;
1414         const struct v4l2_ctrl_ops *ops = &rtl2832_sdr_ctrl_ops;
1415         struct dvb_usb_device *d = i2c_get_adapdata(i2c);
1416
1417         s = kzalloc(sizeof(struct rtl2832_sdr_state), GFP_KERNEL);
1418         if (s == NULL) {
1419                 dev_err(&d->udev->dev,
1420                                 "Could not allocate memory for rtl2832_sdr_state\n");
1421                 return NULL;
1422         }
1423
1424         /* setup the state */
1425         s->fe = fe;
1426         s->d = d;
1427         s->udev = d->udev;
1428         s->i2c = i2c;
1429         s->cfg = cfg;
1430         s->f_adc = bands_adc[0].rangelow;
1431         s->f_tuner = bands_fm[0].rangelow;
1432         s->pixelformat = formats[0].pixelformat;
1433         s->buffersize = formats[0].buffersize;
1434         s->num_formats = NUM_FORMATS;
1435         if (rtl2832_sdr_emulated_fmt == false)
1436                 s->num_formats -= 1;
1437
1438         mutex_init(&s->v4l2_lock);
1439         mutex_init(&s->vb_queue_lock);
1440         spin_lock_init(&s->queued_bufs_lock);
1441         INIT_LIST_HEAD(&s->queued_bufs);
1442
1443         /* Init videobuf2 queue structure */
1444         s->vb_queue.type = V4L2_BUF_TYPE_SDR_CAPTURE;
1445         s->vb_queue.io_modes = VB2_MMAP | VB2_USERPTR | VB2_READ;
1446         s->vb_queue.drv_priv = s;
1447         s->vb_queue.buf_struct_size = sizeof(struct rtl2832_sdr_frame_buf);
1448         s->vb_queue.ops = &rtl2832_sdr_vb2_ops;
1449         s->vb_queue.mem_ops = &vb2_vmalloc_memops;
1450         s->vb_queue.timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
1451         ret = vb2_queue_init(&s->vb_queue);
1452         if (ret) {
1453                 dev_err(&s->udev->dev, "Could not initialize vb2 queue\n");
1454                 goto err_free_mem;
1455         }
1456
1457         /* Register controls */
1458         switch (s->cfg->tuner) {
1459         case RTL2832_TUNER_E4000:
1460                 v4l2_ctrl_handler_init(&s->hdl, 9);
1461                 if (sd)
1462                         v4l2_ctrl_add_handler(&s->hdl, sd->ctrl_handler, NULL);
1463                 break;
1464         case RTL2832_TUNER_R820T:
1465                 v4l2_ctrl_handler_init(&s->hdl, 2);
1466                 s->bandwidth_auto = v4l2_ctrl_new_std(&s->hdl, ops,
1467                                                       V4L2_CID_RF_TUNER_BANDWIDTH_AUTO,
1468                                                       0, 1, 1, 1);
1469                 s->bandwidth = v4l2_ctrl_new_std(&s->hdl, ops,
1470                                                  V4L2_CID_RF_TUNER_BANDWIDTH,
1471                                                  0, 8000000, 100000, 0);
1472                 v4l2_ctrl_auto_cluster(2, &s->bandwidth_auto, 0, false);
1473                 break;
1474         case RTL2832_TUNER_FC0012:
1475         case RTL2832_TUNER_FC0013:
1476                 v4l2_ctrl_handler_init(&s->hdl, 2);
1477                 s->bandwidth_auto = v4l2_ctrl_new_std(&s->hdl, ops,
1478                                                       V4L2_CID_RF_TUNER_BANDWIDTH_AUTO,
1479                                                       0, 1, 1, 1);
1480                 s->bandwidth = v4l2_ctrl_new_std(&s->hdl, ops,
1481                                                  V4L2_CID_RF_TUNER_BANDWIDTH,
1482                                                  6000000, 8000000, 1000000,
1483                                                  6000000);
1484                 v4l2_ctrl_auto_cluster(2, &s->bandwidth_auto, 0, false);
1485                 break;
1486         default:
1487                 v4l2_ctrl_handler_init(&s->hdl, 0);
1488                 dev_notice(&s->udev->dev, "%s: Unsupported tuner\n",
1489                                 KBUILD_MODNAME);
1490                 goto err_free_controls;
1491         }
1492
1493         if (s->hdl.error) {
1494                 ret = s->hdl.error;
1495                 dev_err(&s->udev->dev, "Could not initialize controls\n");
1496                 goto err_free_controls;
1497         }
1498
1499         /* Init video_device structure */
1500         s->vdev = rtl2832_sdr_template;
1501         s->vdev.queue = &s->vb_queue;
1502         s->vdev.queue->lock = &s->vb_queue_lock;
1503         video_set_drvdata(&s->vdev, s);
1504
1505         /* Register the v4l2_device structure */
1506         s->v4l2_dev.release = rtl2832_sdr_video_release;
1507         ret = v4l2_device_register(&s->udev->dev, &s->v4l2_dev);
1508         if (ret) {
1509                 dev_err(&s->udev->dev,
1510                                 "Failed to register v4l2-device (%d)\n", ret);
1511                 goto err_free_controls;
1512         }
1513
1514         s->v4l2_dev.ctrl_handler = &s->hdl;
1515         s->vdev.v4l2_dev = &s->v4l2_dev;
1516         s->vdev.lock = &s->v4l2_lock;
1517         s->vdev.vfl_dir = VFL_DIR_RX;
1518
1519         ret = video_register_device(&s->vdev, VFL_TYPE_SDR, -1);
1520         if (ret) {
1521                 dev_err(&s->udev->dev,
1522                                 "Failed to register as video device (%d)\n",
1523                                 ret);
1524                 goto err_unregister_v4l2_dev;
1525         }
1526         dev_info(&s->udev->dev, "Registered as %s\n",
1527                         video_device_node_name(&s->vdev));
1528
1529         fe->sec_priv = s;
1530         fe->ops.release_sec = rtl2832_sdr_release_sec;
1531
1532         dev_info(&s->i2c->dev, "%s: Realtek RTL2832 SDR attached\n",
1533                         KBUILD_MODNAME);
1534         dev_notice(&s->udev->dev,
1535                         "%s: SDR API is still slightly experimental and functionality changes may follow\n",
1536                         KBUILD_MODNAME);
1537         return fe;
1538
1539 err_unregister_v4l2_dev:
1540         v4l2_device_unregister(&s->v4l2_dev);
1541 err_free_controls:
1542         v4l2_ctrl_handler_free(&s->hdl);
1543 err_free_mem:
1544         kfree(s);
1545         return NULL;
1546 }
1547 EXPORT_SYMBOL(rtl2832_sdr_attach);
1548
1549 MODULE_AUTHOR("Antti Palosaari <crope@iki.fi>");
1550 MODULE_DESCRIPTION("Realtek RTL2832 SDR driver");
1551 MODULE_LICENSE("GPL");