rtl-sdr.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855
  1. /*
  2. * rtl-sdr, turns your Realtek RTL2832 based DVB dongle into a SDR receiver
  3. * Copyright (C) 2012 by Steve Markgraf <steve@steve-m.de>
  4. * Copyright (C) 2012 by Dimitri Stolnikov <horiz0n@gmx.net>
  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
  17. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. */
  19. #include <errno.h>
  20. #include <signal.h>
  21. #include <string.h>
  22. #include <stdio.h>
  23. #include <stdlib.h>
  24. #include <math.h>
  25. #ifndef _WIN32
  26. #include <unistd.h>
  27. #endif
  28. #include <libusb.h>
  29. /*
  30. * All libusb callback functions should be marked with the LIBUSB_CALL macro
  31. * to ensure that they are compiled with the same calling convention as libusb.
  32. *
  33. * If the macro isn't available in older libusb versions, we simply define it.
  34. */
  35. #ifndef LIBUSB_CALL
  36. #define LIBUSB_CALL
  37. #endif
  38. #include <rtl-sdr.h>
  39. #include "tuner_e4000.h"
  40. #include "tuner_fc0012.h"
  41. #include "tuner_fc0013.h"
  42. #include "tuner_fc2580.h"
  43. typedef struct rtlsdr_tuner {
  44. int(*init)(void *);
  45. int(*exit)(void *);
  46. int(*tune)(void *, int freq /* Hz */);
  47. int(*set_bw)(void *, int bw /* Hz */);
  48. int(*set_gain)(void *, int gain /* dB */);
  49. int freq; /* Hz */
  50. int corr; /* ppm */
  51. int gain; /* dB */
  52. } rtlsdr_tuner_t;
  53. void rtlsdr_set_gpio_bit(rtlsdr_dev_t *dev, uint8_t gpio, int val);
  54. /* generic tuner interface functions, shall be moved to the tuner implementations */
  55. int e4k_init(void *dev) { return e4000_Initialize(dev); }
  56. int e4k_exit(void *dev) { return 0; }
  57. int e4k_tune(void *dev, int freq) { return e4000_SetRfFreqHz(dev, freq); }
  58. int e4k_set_bw(void *dev, int bw) { return e4000_SetBandwidthHz(dev, 8000000); }
  59. int e4k_set_gain(void *dev, int gain) { return 0; }
  60. int fc0012_init(void *dev) { return FC0012_Open(dev); }
  61. int fc0012_exit(void *dev) { return 0; }
  62. int fc0012_tune(void *dev, int freq) {
  63. unsigned int bw = 6;
  64. /* select V-band/U-band filter */
  65. rtlsdr_set_gpio_bit(dev, 6, (freq > 300000000) ? 1 : 0);
  66. return FC0012_SetFrequency(dev, freq/1000, bw & 0xff);
  67. }
  68. int fc0012_set_bw(void *dev, int bw) {
  69. unsigned long freq = ((rtlsdr_tuner_t *)dev)->freq;
  70. return FC0013_SetFrequency(dev, freq/1000, bw/1000000);
  71. }
  72. int fc0012_set_gain(void *dev, int gain) { return 0; }
  73. int fc0013_init(void *dev) { return FC0013_Open(dev); }
  74. int fc0013_exit(void *dev) { return 0; }
  75. int fc0013_tune(void *dev, int freq) {
  76. unsigned int bw = 6;
  77. return FC0013_SetFrequency(dev, freq/1000, bw & 0xff);
  78. }
  79. int fc0013_set_bw(void *dev, int bw) {
  80. unsigned long freq = ((rtlsdr_tuner_t *)dev)->freq;
  81. return FC0013_SetFrequency(dev, freq/1000, bw/1000000);
  82. }
  83. int fc0013_set_gain(void *dev, int gain) { return 0; }
  84. int fc2580_init(void *dev) { return fc2580_Initialize(dev); }
  85. int fc2580_exit(void *dev) { return 0; }
  86. int fc2580_tune(void *dev, int freq) { return fc2580_SetRfFreqHz(dev, freq); }
  87. int fc2580_set_bw(void *dev, int bw) { return fc2580_SetBandwidthMode(dev, 1); }
  88. int fc2580_set_gain(void *dev, int gain) { return 0; }
  89. enum rtlsdr_tuners {
  90. RTLSDR_TUNER_E4000,
  91. RTLSDR_TUNER_FC0012,
  92. RTLSDR_TUNER_FC0013,
  93. RTLSDR_TUNER_FC2580
  94. };
  95. static rtlsdr_tuner_t tuners[] = {
  96. { e4k_init, e4k_exit, e4k_tune, e4k_set_bw, e4k_set_gain, 0, 0, 0 },
  97. { fc0012_init, fc0012_exit, fc0012_tune, fc0012_set_bw, fc0012_set_gain, 0, 0, 0 },
  98. { fc0013_init, fc0013_exit, fc0013_tune, fc0013_set_bw, fc0013_set_gain, 0, 0, 0 },
  99. { fc2580_init, fc2580_exit, fc2580_tune, fc2580_set_bw, fc2580_set_gain, 0, 0, 0 },
  100. };
  101. typedef struct rtlsdr_device {
  102. uint16_t vid;
  103. uint16_t pid;
  104. const char *name;
  105. } rtlsdr_device_t;
  106. static rtlsdr_device_t devices[] = {
  107. { 0x0bda, 0x2832, "Generic RTL2832U (e.g. hama nano)" },
  108. { 0x0bda, 0x2838, "ezcap USB 2.0 DVB-T/DAB/FM dongle" },
  109. { 0x0ccd, 0x00a9, "Terratec Cinergy T Stick Black (rev 1)" },
  110. { 0x0ccd, 0x00b3, "Terratec NOXON DAB/DAB+ USB dongle (rev 1)" },
  111. { 0x0ccd, 0x00e0, "Terratec NOXON DAB/DAB+ USB dongle (rev 2)" },
  112. { 0x1f4d, 0xb803, "GTek T803" },
  113. { 0x1f4d, 0xc803, "Lifeview LV5TDeluxe" },
  114. { 0x1b80, 0xd3a4, "Twintech UT-40" },
  115. { 0x1d19, 0x1101, "Dexatek DK DVB-T Dongle (Logilink VG0002A)" },
  116. { 0x1d19, 0x1102, "Dexatek DK DVB-T Dongle (MSI DigiVox mini II V3.0)" },
  117. { 0x0458, 0x707f, "Genius TVGo DVB-T03 USB dongle (Ver. B)" },
  118. { 0x1b80, 0xd393, "GIGABYTE GT-U7300" },
  119. { 0x1b80, 0xd395, "Peak 102569AGPK" },
  120. { 0x1b80, 0xd39d, "SVEON STV20 DVB-T USB & FM" },
  121. };
  122. #define BUF_COUNT 32
  123. #define BUF_LENGTH (16 * 16384)
  124. struct rtlsdr_dev {
  125. libusb_context *ctx;
  126. struct libusb_device_handle *devh;
  127. struct libusb_transfer *xfer[BUF_COUNT];
  128. unsigned char *xfer_buf[BUF_COUNT];
  129. rtlsdr_async_read_cb_t cb;
  130. void *cb_ctx;
  131. int run_async;
  132. rtlsdr_tuner_t *tuner;
  133. int rate; /* Hz */
  134. };
  135. #define CRYSTAL_FREQ 28800000
  136. #define MAX_SAMP_RATE 3200000
  137. #define CTRL_IN (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_IN)
  138. #define CTRL_OUT (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT)
  139. #define CTRL_TIMEOUT 300
  140. enum usb_reg {
  141. USB_SYSCTL = 0x2000,
  142. USB_CTRL = 0x2010,
  143. USB_STAT = 0x2014,
  144. USB_EPA_CFG = 0x2144,
  145. USB_EPA_CTL = 0x2148,
  146. USB_EPA_MAXPKT = 0x2158,
  147. USB_EPA_MAXPKT_2 = 0x215a,
  148. USB_EPA_FIFO_CFG = 0x2160,
  149. };
  150. enum sys_reg {
  151. DEMOD_CTL = 0x3000,
  152. GPO = 0x3001,
  153. GPI = 0x3002,
  154. GPOE = 0x3003,
  155. GPD = 0x3004,
  156. SYSINTE = 0x3005,
  157. SYSINTS = 0x3006,
  158. GP_CFG0 = 0x3007,
  159. GP_CFG1 = 0x3008,
  160. SYSINTE_1 = 0x3009,
  161. SYSINTS_1 = 0x300a,
  162. DEMOD_CTL_1 = 0x300b,
  163. IR_SUSPEND = 0x300c,
  164. };
  165. enum blocks {
  166. DEMODB = 0,
  167. USBB = 1,
  168. SYSB = 2,
  169. TUNB = 3,
  170. ROMB = 4,
  171. IRB = 5,
  172. IICB = 6,
  173. };
  174. int rtlsdr_read_array(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint8_t *array, uint8_t len)
  175. {
  176. int r;
  177. uint16_t index = (block << 8);
  178. r = libusb_control_transfer(dev->devh, CTRL_IN, 0, addr, index, array, len, CTRL_TIMEOUT);
  179. return r;
  180. }
  181. int rtlsdr_write_array(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint8_t *array, uint8_t len)
  182. {
  183. int r;
  184. uint16_t index = (block << 8) | 0x10;
  185. r = libusb_control_transfer(dev->devh, CTRL_OUT, 0, addr, index, array, len, CTRL_TIMEOUT);
  186. return r;
  187. }
  188. int rtlsdr_i2c_write_reg(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t reg, uint8_t val)
  189. {
  190. uint16_t addr = i2c_addr;
  191. uint8_t data[2];
  192. data[0] = reg;
  193. data[1] = val;
  194. return rtlsdr_write_array(dev, IICB, addr, (uint8_t *)&data, 2);
  195. }
  196. uint8_t rtlsdr_i2c_read_reg(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t reg)
  197. {
  198. uint16_t addr = i2c_addr;
  199. uint8_t data;
  200. rtlsdr_write_array(dev, IICB, addr, &reg, 1);
  201. rtlsdr_read_array(dev, IICB, addr, &data, 1);
  202. return data;
  203. }
  204. int rtlsdr_i2c_write(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t *buffer, int len)
  205. {
  206. uint16_t addr = i2c_addr;
  207. if (!dev)
  208. return -1;
  209. return rtlsdr_write_array(dev, IICB, addr, buffer, len);
  210. }
  211. int rtlsdr_i2c_read(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t *buffer, int len)
  212. {
  213. uint16_t addr = i2c_addr;
  214. if (!dev)
  215. return -1;
  216. return rtlsdr_read_array(dev, IICB, addr, buffer, len);
  217. }
  218. uint16_t rtlsdr_read_reg(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint8_t len)
  219. {
  220. int r;
  221. unsigned char data[2];
  222. uint16_t index = (block << 8);
  223. uint16_t reg;
  224. r = libusb_control_transfer(dev->devh, CTRL_IN, 0, addr, index, data, len, CTRL_TIMEOUT);
  225. if (r < 0)
  226. fprintf(stderr, "%s failed\n", __FUNCTION__);
  227. reg = (data[1] << 8) | data[0];
  228. return reg;
  229. }
  230. void rtlsdr_write_reg(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint16_t val, uint8_t len)
  231. {
  232. int r;
  233. unsigned char data[2];
  234. uint16_t index = (block << 8) | 0x10;
  235. if (len == 1)
  236. data[0] = val & 0xff;
  237. else
  238. data[0] = val >> 8;
  239. data[1] = val & 0xff;
  240. r = libusb_control_transfer(dev->devh, CTRL_OUT, 0, addr, index, data, len, CTRL_TIMEOUT);
  241. if (r < 0)
  242. fprintf(stderr, "%s failed\n", __FUNCTION__);
  243. }
  244. uint16_t rtlsdr_demod_read_reg(rtlsdr_dev_t *dev, uint8_t page, uint8_t addr, uint8_t len)
  245. {
  246. int r;
  247. unsigned char data[2];
  248. uint16_t index = page;
  249. uint16_t reg;
  250. addr = (addr << 8) | 0x20;
  251. r = libusb_control_transfer(dev->devh, CTRL_IN, 0, addr, index, data, len, CTRL_TIMEOUT);
  252. if (r < 0)
  253. fprintf(stderr, "%s failed\n", __FUNCTION__);
  254. reg = (data[1] << 8) | data[0];
  255. return reg;
  256. }
  257. void rtlsdr_demod_write_reg(rtlsdr_dev_t *dev, uint8_t page, uint16_t addr, uint16_t val, uint8_t len)
  258. {
  259. int r;
  260. unsigned char data[2];
  261. uint16_t index = 0x10 | page;
  262. addr = (addr << 8) | 0x20;
  263. if (len == 1)
  264. data[0] = val & 0xff;
  265. else
  266. data[0] = val >> 8;
  267. data[1] = val & 0xff;
  268. r = libusb_control_transfer(dev->devh, CTRL_OUT, 0, addr, index, data, len, CTRL_TIMEOUT);
  269. if (r < 0)
  270. fprintf(stderr, "%s failed\n", __FUNCTION__);
  271. rtlsdr_demod_read_reg(dev, 0x0a, 0x01, 1);
  272. }
  273. void rtlsdr_set_gpio_bit(rtlsdr_dev_t *dev, uint8_t gpio, int val)
  274. {
  275. uint8_t r;
  276. gpio = 1 << gpio;
  277. r = rtlsdr_read_reg(dev, SYSB, GPO, 1);
  278. r = val ? (r | gpio) : (r & ~gpio);
  279. rtlsdr_write_reg(dev, SYSB, GPO, r, 1);
  280. }
  281. void rtlsdr_set_gpio_output(rtlsdr_dev_t *dev, uint8_t gpio)
  282. {
  283. int r;
  284. gpio = 1 << gpio;
  285. r = rtlsdr_read_reg(dev, SYSB, GPD, 1);
  286. rtlsdr_write_reg(dev, SYSB, GPO, r & ~gpio, 1);
  287. r = rtlsdr_read_reg(dev, SYSB, GPOE, 1);
  288. rtlsdr_write_reg(dev, SYSB, GPOE, r | gpio, 1);
  289. }
  290. void rtlsdr_set_i2c_repeater(rtlsdr_dev_t *dev, int on)
  291. {
  292. rtlsdr_demod_write_reg(dev, 1, 0x01, on ? 0x18 : 0x10, 1);
  293. }
  294. void rtlsdr_init_baseband(rtlsdr_dev_t *dev)
  295. {
  296. unsigned int i;
  297. /* default FIR coefficients used for DAB/FM by the Windows driver,
  298. * the DVB driver uses different ones */
  299. uint8_t fir_coeff[] = {
  300. 0xca, 0xdc, 0xd7, 0xd8, 0xe0, 0xf2, 0x0e, 0x35, 0x06, 0x50,
  301. 0x9c, 0x0d, 0x71, 0x11, 0x14, 0x71, 0x74, 0x19, 0x41, 0x00,
  302. };
  303. /* initialize USB */
  304. rtlsdr_write_reg(dev, USBB, USB_SYSCTL, 0x09, 1);
  305. rtlsdr_write_reg(dev, USBB, USB_EPA_MAXPKT, 0x0002, 2);
  306. rtlsdr_write_reg(dev, USBB, USB_EPA_CTL, 0x1002, 2);
  307. /* poweron demod */
  308. rtlsdr_write_reg(dev, SYSB, DEMOD_CTL_1, 0x22, 1);
  309. rtlsdr_write_reg(dev, SYSB, DEMOD_CTL, 0xe8, 1);
  310. /* reset demod (bit 3, soft_rst) */
  311. rtlsdr_demod_write_reg(dev, 1, 0x01, 0x14, 1);
  312. rtlsdr_demod_write_reg(dev, 1, 0x01, 0x10, 1);
  313. /* disable spectrum inversion and adjacent channel rejection */
  314. rtlsdr_demod_write_reg(dev, 1, 0x15, 0x00, 1);
  315. rtlsdr_demod_write_reg(dev, 1, 0x16, 0x0000, 2);
  316. /* set IF-frequency to 0 Hz */
  317. rtlsdr_demod_write_reg(dev, 1, 0x19, 0x0000, 2);
  318. /* set FIR coefficients */
  319. for (i = 0; i < sizeof (fir_coeff); i++)
  320. rtlsdr_demod_write_reg(dev, 1, 0x1c + i, fir_coeff[i], 1);
  321. rtlsdr_demod_write_reg(dev, 0, 0x19, 0x25, 1);
  322. /* init FSM state-holding register */
  323. rtlsdr_demod_write_reg(dev, 1, 0x93, 0xf0, 1);
  324. /* disable AGC (en_dagc, bit 0) */
  325. rtlsdr_demod_write_reg(dev, 1, 0x11, 0x00, 1);
  326. /* disable PID filter (enable_PID = 0) */
  327. rtlsdr_demod_write_reg(dev, 0, 0x61, 0x60, 1);
  328. /* opt_adc_iq = 0, default ADC_I/ADC_Q datapath */
  329. rtlsdr_demod_write_reg(dev, 0, 0x06, 0x80, 1);
  330. /* Enable Zero-IF mode (en_bbin bit), DC cancellation (en_dc_est),
  331. * IQ estimation/compensation (en_iq_comp, en_iq_est) */
  332. rtlsdr_demod_write_reg(dev, 1, 0xb1, 0x1b, 1);
  333. }
  334. void rtlsdr_deinit_baseband(rtlsdr_dev_t *dev)
  335. {
  336. /* deinitialize tuner */
  337. rtlsdr_set_i2c_repeater(dev, 1);
  338. dev->tuner->exit(dev);
  339. rtlsdr_set_i2c_repeater(dev, 0);
  340. /* poweroff demodulator and ADCs */
  341. rtlsdr_write_reg(dev, SYSB, DEMOD_CTL, 0x20, 1);
  342. }
  343. int rtlsdr_set_center_freq(rtlsdr_dev_t *dev, uint32_t freq)
  344. {
  345. int r;
  346. double f = (double) freq;
  347. if (!dev || !dev->tuner)
  348. return -1;
  349. rtlsdr_set_i2c_repeater(dev, 1);
  350. f *= 1.0 + dev->tuner->corr / 1e6;
  351. r = dev->tuner->tune((void *)dev, (int) f);
  352. rtlsdr_set_i2c_repeater(dev, 0);
  353. if (!r)
  354. dev->tuner->freq = freq;
  355. return r;
  356. }
  357. int rtlsdr_get_center_freq(rtlsdr_dev_t *dev)
  358. {
  359. if (!dev || !dev->tuner)
  360. return -1;
  361. return dev->tuner->freq;
  362. }
  363. int rtlsdr_set_freq_correction(rtlsdr_dev_t *dev, int ppm)
  364. {
  365. int r;
  366. if (!dev || !dev->tuner)
  367. return -1;
  368. if (dev->tuner->corr == ppm)
  369. return -1;
  370. dev->tuner->corr = ppm;
  371. /* retune to apply new correction value */
  372. r = rtlsdr_set_center_freq(dev, dev->tuner->freq);
  373. return r;
  374. }
  375. int rtlsdr_get_freq_correction(rtlsdr_dev_t *dev)
  376. {
  377. if (!dev || !dev->tuner)
  378. return -1;
  379. return dev->tuner->corr;
  380. }
  381. int rtlsdr_set_tuner_gain(rtlsdr_dev_t *dev, int gain)
  382. {
  383. int r;
  384. if (!dev || !dev->tuner)
  385. return -1;
  386. r = dev->tuner->set_gain((void *)dev, gain);
  387. if (!r)
  388. dev->tuner->gain = gain;
  389. return r;
  390. }
  391. int rtlsdr_get_tuner_gain(rtlsdr_dev_t *dev)
  392. {
  393. if (!dev || !dev->tuner)
  394. return -1;
  395. return dev->tuner->gain;
  396. }
  397. int rtlsdr_set_sample_rate(rtlsdr_dev_t *dev, uint32_t samp_rate)
  398. {
  399. uint16_t tmp;
  400. uint32_t rsamp_ratio;
  401. double real_rate;
  402. if (!dev)
  403. return -1;
  404. /* check for the maximum rate the resampler supports */
  405. if (samp_rate > MAX_SAMP_RATE)
  406. samp_rate = MAX_SAMP_RATE;
  407. rsamp_ratio = (CRYSTAL_FREQ * pow(2, 22)) / samp_rate;
  408. rsamp_ratio &= ~3;
  409. real_rate = (CRYSTAL_FREQ * pow(2, 22)) / rsamp_ratio;
  410. fprintf(stderr, "Setting sample rate: %.3f Hz\n", real_rate);
  411. if (dev->tuner)
  412. dev->tuner->set_bw((void *)dev, real_rate);
  413. dev->rate = samp_rate;
  414. tmp = (rsamp_ratio >> 16);
  415. rtlsdr_demod_write_reg(dev, 1, 0x9f, tmp, 2);
  416. tmp = rsamp_ratio & 0xffff;
  417. rtlsdr_demod_write_reg(dev, 1, 0xa1, tmp, 2);
  418. return 0;
  419. }
  420. int rtlsdr_get_sample_rate(rtlsdr_dev_t *dev)
  421. {
  422. if (!dev)
  423. return -1;
  424. return dev->rate;
  425. }
  426. rtlsdr_device_t *find_known_device(uint16_t vid, uint16_t pid)
  427. {
  428. int i;
  429. rtlsdr_device_t *device = NULL;
  430. for (i = 0; i < sizeof(devices)/sizeof(rtlsdr_device_t); i++ ) {
  431. if (devices[i].vid == vid && devices[i].pid == pid) {
  432. device = &devices[i];
  433. break;
  434. }
  435. }
  436. return device;
  437. }
  438. uint32_t rtlsdr_get_device_count(void)
  439. {
  440. int i;
  441. libusb_device **list;
  442. uint32_t device_count = 0;
  443. struct libusb_device_descriptor dd;
  444. ssize_t cnt;
  445. libusb_init(NULL);
  446. cnt = libusb_get_device_list(NULL, &list);
  447. for (i = 0; i < cnt; i++) {
  448. libusb_get_device_descriptor(list[i], &dd);
  449. if (find_known_device(dd.idVendor, dd.idProduct))
  450. device_count++;
  451. }
  452. libusb_free_device_list(list, 0);
  453. libusb_exit(NULL);
  454. return device_count;
  455. }
  456. const char *rtlsdr_get_device_name(uint32_t index)
  457. {
  458. int i;
  459. libusb_device **list;
  460. struct libusb_device_descriptor dd;
  461. rtlsdr_device_t *device = NULL;
  462. uint32_t device_count = 0;
  463. ssize_t cnt;
  464. libusb_init(NULL);
  465. cnt = libusb_get_device_list(NULL, &list);
  466. for (i = 0; i < cnt; i++) {
  467. libusb_get_device_descriptor(list[i], &dd);
  468. device = find_known_device(dd.idVendor, dd.idProduct);
  469. if (device) {
  470. device_count++;
  471. if (index == device_count - 1)
  472. break;
  473. }
  474. }
  475. libusb_free_device_list(list, 0);
  476. libusb_exit(NULL);
  477. if (device)
  478. return device->name;
  479. else
  480. return "";
  481. }
  482. int rtlsdr_open(rtlsdr_dev_t **out_dev, uint32_t index)
  483. {
  484. int r;
  485. int i;
  486. libusb_device **list;
  487. rtlsdr_dev_t *dev = NULL;
  488. libusb_device *device = NULL;
  489. uint32_t device_count = 0;
  490. struct libusb_device_descriptor dd;
  491. uint8_t reg;
  492. ssize_t cnt;
  493. dev = malloc(sizeof(rtlsdr_dev_t));
  494. memset(dev, 0, sizeof(rtlsdr_dev_t));
  495. libusb_init(&dev->ctx);
  496. cnt = libusb_get_device_list(dev->ctx, &list);
  497. for (i = 0; i < cnt; i++) {
  498. device = list[i];
  499. libusb_get_device_descriptor(list[i], &dd);
  500. if (find_known_device(dd.idVendor, dd.idProduct)) {
  501. device_count++;
  502. }
  503. if (index == device_count - 1)
  504. break;
  505. device = NULL;
  506. }
  507. if (!device) {
  508. r = -1;
  509. goto err;
  510. }
  511. r = libusb_open(device, &dev->devh);
  512. if (r < 0) {
  513. libusb_free_device_list(list, 0);
  514. fprintf(stderr, "usb_open error %d\n", r);
  515. goto err;
  516. }
  517. libusb_free_device_list(list, 0);
  518. r = libusb_claim_interface(dev->devh, 0);
  519. if (r < 0) {
  520. fprintf(stderr, "usb_claim_interface error %d\n", r);
  521. goto err;
  522. }
  523. rtlsdr_init_baseband(dev);
  524. /* Probe tuners */
  525. rtlsdr_set_i2c_repeater(dev, 1);
  526. reg = rtlsdr_i2c_read_reg(dev, E4K_I2C_ADDR, E4K_CHECK_ADDR);
  527. if (reg == E4K_CHECK_VAL) {
  528. fprintf(stderr, "Found Elonics E4000 tuner\n");
  529. dev->tuner = &tuners[RTLSDR_TUNER_E4000];
  530. goto found;
  531. }
  532. reg = rtlsdr_i2c_read_reg(dev, FC0013_I2C_ADDR, FC0013_CHECK_ADDR);
  533. if (reg == FC0013_CHECK_VAL) {
  534. fprintf(stderr, "Found Fitipower FC0013 tuner\n");
  535. dev->tuner = &tuners[RTLSDR_TUNER_FC0013];
  536. goto found;
  537. }
  538. /* initialise GPIOs */
  539. rtlsdr_set_gpio_output(dev, 5);
  540. /* reset tuner before probing */
  541. rtlsdr_set_gpio_bit(dev, 5, 1);
  542. rtlsdr_set_gpio_bit(dev, 5, 0);
  543. reg = rtlsdr_i2c_read_reg(dev, FC2580_I2C_ADDR, FC2580_CHECK_ADDR);
  544. if ((reg & 0x7f) == FC2580_CHECK_VAL) {
  545. fprintf(stderr, "Found FCI 2580 tuner\n");
  546. dev->tuner = &tuners[RTLSDR_TUNER_FC2580];
  547. goto found;
  548. }
  549. reg = rtlsdr_i2c_read_reg(dev, FC0012_I2C_ADDR, FC0012_CHECK_ADDR);
  550. if (reg == FC0012_CHECK_VAL) {
  551. fprintf(stderr, "Found Fitipower FC0012 tuner\n");
  552. rtlsdr_set_gpio_output(dev, 6);
  553. dev->tuner = &tuners[RTLSDR_TUNER_FC0012];
  554. goto found;
  555. }
  556. found:
  557. if (dev->tuner)
  558. r =dev->tuner->init(dev);
  559. rtlsdr_set_i2c_repeater(dev, 0);
  560. *out_dev = dev;
  561. return 0;
  562. err:
  563. if (dev) {
  564. if (dev->ctx)
  565. libusb_exit(dev->ctx);
  566. free(dev);
  567. }
  568. return r;
  569. }
  570. int rtlsdr_close(rtlsdr_dev_t *dev)
  571. {
  572. int i;
  573. if (!dev)
  574. return -1;
  575. rtlsdr_deinit_baseband(dev);
  576. libusb_release_interface(dev->devh, 0);
  577. libusb_close(dev->devh);
  578. for(i = 0; i < BUF_COUNT; ++i) {
  579. if (dev->xfer[i])
  580. libusb_free_transfer(dev->xfer[i]);
  581. if (dev->xfer_buf[i])
  582. free(dev->xfer_buf[i]);
  583. }
  584. libusb_exit(dev->ctx);
  585. free(dev);
  586. return 0;
  587. }
  588. int rtlsdr_reset_buffer(rtlsdr_dev_t *dev)
  589. {
  590. if (!dev)
  591. return -1;
  592. rtlsdr_write_reg(dev, USBB, USB_EPA_CTL, 0x1002, 2);
  593. rtlsdr_write_reg(dev, USBB, USB_EPA_CTL, 0x0000, 2);
  594. return 0;
  595. }
  596. int rtlsdr_read_sync(rtlsdr_dev_t *dev, void *buf, int len, int *n_read)
  597. {
  598. if (!dev)
  599. return -1;
  600. return libusb_bulk_transfer(dev->devh, 0x81, buf, len, n_read, 3000);
  601. }
  602. static void LIBUSB_CALL _libusb_callback(struct libusb_transfer *transfer)
  603. {
  604. if (LIBUSB_TRANSFER_COMPLETED == transfer->status) {
  605. rtlsdr_dev_t *dev = (rtlsdr_dev_t *)transfer->user_data;
  606. dev->cb(transfer->buffer, transfer->actual_length, dev->cb_ctx);
  607. libusb_submit_transfer(transfer); /* resubmit transfer */
  608. } else {
  609. /*fprintf(stderr, "transfer %d\n", transfer->status);*/
  610. }
  611. }
  612. int rtlsdr_wait_async(rtlsdr_dev_t *dev, rtlsdr_async_read_cb_t cb, void *ctx)
  613. {
  614. int i, r;
  615. if (!dev)
  616. return -1;
  617. dev->cb = cb;
  618. dev->cb_ctx = ctx;
  619. for(i = 0; i < BUF_COUNT; ++i) {
  620. if (dev->xfer[i])
  621. continue;
  622. dev->xfer[i] = libusb_alloc_transfer(0);
  623. }
  624. for(i = 0; i < BUF_COUNT; ++i) {
  625. if (dev->xfer_buf[i])
  626. continue;
  627. dev->xfer_buf[i] = (unsigned char *)malloc(BUF_LENGTH);
  628. }
  629. for(i = 0; i < BUF_COUNT; ++i) {
  630. libusb_fill_bulk_transfer(dev->xfer[i],
  631. dev->devh,
  632. 0x81,
  633. dev->xfer_buf[i], BUF_LENGTH,
  634. _libusb_callback,
  635. (void *)dev, 0);
  636. libusb_submit_transfer(dev->xfer[i]);
  637. }
  638. dev->run_async = 1;
  639. while (dev->run_async) {
  640. struct timeval tv = { 1, 0 };
  641. r = libusb_handle_events_timeout(dev->ctx, &tv);
  642. if (r < 0) {
  643. /*fprintf(stderr, "handle_events %d\n", r);*/
  644. break;
  645. }
  646. }
  647. return r;
  648. }
  649. int rtlsdr_cancel_async(rtlsdr_dev_t *dev)
  650. {
  651. if (!dev)
  652. return -1;
  653. if (dev->run_async) {
  654. dev->run_async = 0;
  655. return 0;
  656. }
  657. return -2;
  658. }