rtl2832_sdr.c 39 KB

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