librtlsdr.c 36 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576
  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. #ifndef _WIN32
  25. #include <unistd.h>
  26. #define min(a, b) (((a) < (b)) ? (a) : (b))
  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. /* two raised to the power of n */
  39. #define TWO_POW(n) ((double)(1ULL<<(n)))
  40. #include "rtl-sdr.h"
  41. #include "tuner_e4k.h"
  42. #include "tuner_fc0012.h"
  43. #include "tuner_fc0013.h"
  44. #include "tuner_fc2580.h"
  45. #include "tuner_r820t.h"
  46. typedef struct rtlsdr_tuner_iface {
  47. /* tuner interface */
  48. int (*init)(void *);
  49. int (*exit)(void *);
  50. int (*set_freq)(void *, uint32_t freq /* Hz */);
  51. int (*set_bw)(void *, int bw /* Hz */);
  52. int (*set_gain)(void *, int gain /* tenth dB */);
  53. int (*set_if_gain)(void *, int stage, int gain /* tenth dB */);
  54. int (*set_gain_mode)(void *, int manual);
  55. } rtlsdr_tuner_iface_t;
  56. enum rtlsdr_async_status {
  57. RTLSDR_INACTIVE = 0,
  58. RTLSDR_CANCELING,
  59. RTLSDR_RUNNING
  60. };
  61. struct rtlsdr_dev {
  62. libusb_context *ctx;
  63. struct libusb_device_handle *devh;
  64. uint32_t xfer_buf_num;
  65. uint32_t xfer_buf_len;
  66. struct libusb_transfer **xfer;
  67. unsigned char **xfer_buf;
  68. rtlsdr_read_async_cb_t cb;
  69. void *cb_ctx;
  70. enum rtlsdr_async_status async_status;
  71. /* rtl demod context */
  72. uint32_t rate; /* Hz */
  73. uint32_t rtl_xtal; /* Hz */
  74. int direct_sampling;
  75. /* tuner context */
  76. enum rtlsdr_tuner tuner_type;
  77. rtlsdr_tuner_iface_t *tuner;
  78. uint32_t tun_xtal; /* Hz */
  79. uint32_t freq; /* Hz */
  80. uint32_t offs_freq; /* Hz */
  81. int corr; /* ppm */
  82. int gain; /* tenth dB */
  83. struct e4k_state e4k_s;
  84. };
  85. void rtlsdr_set_gpio_bit(rtlsdr_dev_t *dev, uint8_t gpio, int val);
  86. /* generic tuner interface functions, shall be moved to the tuner implementations */
  87. int e4000_init(void *dev) {
  88. rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
  89. devt->e4k_s.i2c_addr = E4K_I2C_ADDR;
  90. devt->e4k_s.vco.fosc = devt->tun_xtal; /* no need to correct it here */
  91. devt->e4k_s.rtl_dev = dev;
  92. return e4k_init(&devt->e4k_s);
  93. }
  94. int e4000_exit(void *dev) { return 0; }
  95. int e4000_set_freq(void *dev, uint32_t freq) {
  96. rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
  97. return e4k_tune_freq(&devt->e4k_s, freq);
  98. }
  99. int e4000_set_bw(void *dev, int bw) {
  100. int r = 0;
  101. rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
  102. r |= e4k_if_filter_bw_set(&devt->e4k_s, E4K_IF_FILTER_MIX, bw);
  103. r |= e4k_if_filter_bw_set(&devt->e4k_s, E4K_IF_FILTER_RC, bw);
  104. r |= e4k_if_filter_bw_set(&devt->e4k_s, E4K_IF_FILTER_CHAN, bw);
  105. return r;
  106. }
  107. int e4000_set_gain(void *dev, int gain) {
  108. rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
  109. int mixgain = (gain > 340) ? 12 : 4;
  110. #if 0
  111. int enhgain = (gain - 420);
  112. #endif
  113. if(e4k_set_lna_gain(&devt->e4k_s, min(300, gain - mixgain * 10)) == -EINVAL)
  114. return -1;
  115. if(e4k_mixer_gain_set(&devt->e4k_s, mixgain) == -EINVAL)
  116. return -1;
  117. #if 0 /* enhanced mixer gain seems to have no effect */
  118. if(enhgain >= 0)
  119. if(e4k_set_enh_gain(&devt->e4k_s, enhgain) == -EINVAL)
  120. return -1;
  121. #endif
  122. return 0;
  123. }
  124. int e4000_set_if_gain(void *dev, int stage, int gain) {
  125. rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
  126. return e4k_if_gain_set(&devt->e4k_s, (uint8_t)stage, (int8_t)(gain / 10));
  127. }
  128. int e4000_set_gain_mode(void *dev, int manual) {
  129. rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
  130. return e4k_enable_manual_gain(&devt->e4k_s, manual);
  131. }
  132. int _fc0012_init(void *dev) { return fc0012_init(dev); }
  133. int fc0012_exit(void *dev) { return 0; }
  134. int fc0012_set_freq(void *dev, uint32_t freq) {
  135. /* select V-band/U-band filter */
  136. rtlsdr_set_gpio_bit(dev, 6, (freq > 300000000) ? 1 : 0);
  137. return fc0012_set_params(dev, freq, 6000000);
  138. }
  139. int fc0012_set_bw(void *dev, int bw) { return 0; }
  140. int _fc0012_set_gain(void *dev, int gain) { return fc0012_set_gain(dev, gain); }
  141. int fc0012_set_gain_mode(void *dev, int manual) { return 0; }
  142. int _fc0013_init(void *dev) { return fc0013_init(dev); }
  143. int fc0013_exit(void *dev) { return 0; }
  144. int fc0013_set_freq(void *dev, uint32_t freq) {
  145. return fc0013_set_params(dev, freq, 6000000);
  146. }
  147. int fc0013_set_bw(void *dev, int bw) { return 0; }
  148. int _fc0013_set_gain(void *dev, int gain) { return fc0013_set_lna_gain(dev, gain); }
  149. int fc2580_init(void *dev) { return fc2580_Initialize(dev); }
  150. int fc2580_exit(void *dev) { return 0; }
  151. int _fc2580_set_freq(void *dev, uint32_t freq) {
  152. return fc2580_SetRfFreqHz(dev, freq);
  153. }
  154. int fc2580_set_bw(void *dev, int bw) { return fc2580_SetBandwidthMode(dev, 1); }
  155. int fc2580_set_gain(void *dev, int gain) { return 0; }
  156. int fc2580_set_gain_mode(void *dev, int manual) { return 0; }
  157. int r820t_init(void *dev) {
  158. int r = R828_Init(dev);
  159. r820t_SetStandardMode(dev, DVB_T_6M);
  160. return r;
  161. }
  162. int r820t_exit(void *dev) { return 0; }
  163. int r820t_set_freq(void *dev, uint32_t freq) { return r820t_SetRfFreqHz(dev, freq); }
  164. int r820t_set_bw(void *dev, int bw) { return 0; }
  165. int r820t_set_gain(void *dev, int gain) { return 0; }
  166. int r820t_set_gain_mode(void *dev, int manual) { return 0; }
  167. /* definition order must match enum rtlsdr_tuner */
  168. static rtlsdr_tuner_iface_t tuners[] = {
  169. {
  170. NULL, NULL, NULL, NULL, NULL, NULL, NULL /* dummy for unknown tuners */
  171. },
  172. {
  173. e4000_init, e4000_exit,
  174. e4000_set_freq, e4000_set_bw, e4000_set_gain, e4000_set_if_gain,
  175. e4000_set_gain_mode
  176. },
  177. {
  178. _fc0012_init, fc0012_exit,
  179. fc0012_set_freq, fc0012_set_bw, _fc0012_set_gain, NULL,
  180. fc0012_set_gain_mode
  181. },
  182. {
  183. _fc0013_init, fc0013_exit,
  184. fc0013_set_freq, fc0013_set_bw, _fc0013_set_gain, NULL,
  185. fc0013_set_gain_mode
  186. },
  187. {
  188. fc2580_init, fc2580_exit,
  189. _fc2580_set_freq, fc2580_set_bw, fc2580_set_gain, NULL,
  190. fc2580_set_gain_mode
  191. },
  192. {
  193. r820t_init, r820t_exit,
  194. r820t_set_freq, r820t_set_bw, r820t_set_gain, NULL,
  195. r820t_set_gain_mode
  196. },
  197. };
  198. typedef struct rtlsdr_dongle {
  199. uint16_t vid;
  200. uint16_t pid;
  201. const char *name;
  202. } rtlsdr_dongle_t;
  203. /*
  204. * Please add your device here and send a patch to osmocom-sdr@lists.osmocom.org
  205. */
  206. static rtlsdr_dongle_t known_devices[] = {
  207. { 0x0bda, 0x2832, "Generic RTL2832U (e.g. hama nano)" },
  208. { 0x0bda, 0x2838, "ezcap USB 2.0 DVB-T/DAB/FM dongle" },
  209. { 0x0ccd, 0x00a9, "Terratec Cinergy T Stick Black (rev 1)" },
  210. { 0x0ccd, 0x00b3, "Terratec NOXON DAB/DAB+ USB dongle (rev 1)" },
  211. { 0x0ccd, 0x00b4, "Terratec NOXON DAB/DAB+ USB dongle (rev 1)" },
  212. { 0x0ccd, 0x00b7, "Terratec NOXON DAB/DAB+ USB dongle (rev 1)" },
  213. { 0x0ccd, 0x00c6, "Terratec NOXON DAB/DAB+ USB dongle (rev 1)" },
  214. { 0x0ccd, 0x00d3, "Terratec Cinergy T Stick RC (Rev.3)" },
  215. { 0x0ccd, 0x00d7, "Terratec T Stick PLUS" },
  216. { 0x0ccd, 0x00e0, "Terratec NOXON DAB/DAB+ USB dongle (rev 2)" },
  217. { 0x185b, 0x0620, "Compro Videomate U620F"},
  218. { 0x185b, 0x0650, "Compro Videomate U650F"},
  219. { 0x185b, 0x0680, "Compro Videomate U680F"},
  220. { 0x1f4d, 0xa803, "Sweex DVB-T USB" },
  221. { 0x1f4d, 0xb803, "GTek T803" },
  222. { 0x1f4d, 0xc803, "Lifeview LV5TDeluxe" },
  223. { 0x1f4d, 0xd286, "MyGica TD312" },
  224. { 0x1f4d, 0xd803, "PROlectrix DV107669" },
  225. { 0x1b80, 0xd398, "Zaapa ZT-MINDVBZP" },
  226. { 0x1b80, 0xd3a4, "Twintech UT-40" },
  227. { 0x1d19, 0x1101, "Dexatek DK DVB-T Dongle (Logilink VG0002A)" },
  228. { 0x1d19, 0x1102, "Dexatek DK DVB-T Dongle (MSI DigiVox mini II V3.0)" },
  229. { 0x1d19, 0x1103, "Dexatek Technology Ltd. DK 5217 DVB-T Dongle" },
  230. { 0x0458, 0x707f, "Genius TVGo DVB-T03 USB dongle (Ver. B)" },
  231. { 0x1b80, 0xd393, "GIGABYTE GT-U7300" },
  232. { 0x1b80, 0xd394, "DIKOM USB-DVBT HD" },
  233. { 0x1b80, 0xd395, "Peak 102569AGPK" },
  234. { 0x1b80, 0xd39d, "SVEON STV20 DVB-T USB & FM" },
  235. };
  236. #define DEFAULT_BUF_NUMBER 32
  237. #define DEFAULT_BUF_LENGTH (16 * 32 * 512)
  238. #define DEF_RTL_XTAL_FREQ 28800000
  239. #define MIN_RTL_XTAL_FREQ (DEF_RTL_XTAL_FREQ - 1000)
  240. #define MAX_RTL_XTAL_FREQ (DEF_RTL_XTAL_FREQ + 1000)
  241. #define MAX_SAMP_RATE 3200000
  242. #define CTRL_IN (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_IN)
  243. #define CTRL_OUT (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT)
  244. #define CTRL_TIMEOUT 300
  245. #define BULK_TIMEOUT 0
  246. enum usb_reg {
  247. USB_SYSCTL = 0x2000,
  248. USB_CTRL = 0x2010,
  249. USB_STAT = 0x2014,
  250. USB_EPA_CFG = 0x2144,
  251. USB_EPA_CTL = 0x2148,
  252. USB_EPA_MAXPKT = 0x2158,
  253. USB_EPA_MAXPKT_2 = 0x215a,
  254. USB_EPA_FIFO_CFG = 0x2160,
  255. };
  256. enum sys_reg {
  257. DEMOD_CTL = 0x3000,
  258. GPO = 0x3001,
  259. GPI = 0x3002,
  260. GPOE = 0x3003,
  261. GPD = 0x3004,
  262. SYSINTE = 0x3005,
  263. SYSINTS = 0x3006,
  264. GP_CFG0 = 0x3007,
  265. GP_CFG1 = 0x3008,
  266. SYSINTE_1 = 0x3009,
  267. SYSINTS_1 = 0x300a,
  268. DEMOD_CTL_1 = 0x300b,
  269. IR_SUSPEND = 0x300c,
  270. };
  271. enum blocks {
  272. DEMODB = 0,
  273. USBB = 1,
  274. SYSB = 2,
  275. TUNB = 3,
  276. ROMB = 4,
  277. IRB = 5,
  278. IICB = 6,
  279. };
  280. int rtlsdr_read_array(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint8_t *array, uint8_t len)
  281. {
  282. int r;
  283. uint16_t index = (block << 8);
  284. r = libusb_control_transfer(dev->devh, CTRL_IN, 0, addr, index, array, len, CTRL_TIMEOUT);
  285. #if 0
  286. if (r < 0)
  287. fprintf(stderr, "%s failed with %d\n", __FUNCTION__, r);
  288. #endif
  289. return r;
  290. }
  291. int rtlsdr_write_array(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint8_t *array, uint8_t len)
  292. {
  293. int r;
  294. uint16_t index = (block << 8) | 0x10;
  295. r = libusb_control_transfer(dev->devh, CTRL_OUT, 0, addr, index, array, len, CTRL_TIMEOUT);
  296. #if 0
  297. if (r < 0)
  298. fprintf(stderr, "%s failed with %d\n", __FUNCTION__, r);
  299. #endif
  300. return r;
  301. }
  302. int rtlsdr_i2c_write_reg(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t reg, uint8_t val)
  303. {
  304. uint16_t addr = i2c_addr;
  305. uint8_t data[2];
  306. data[0] = reg;
  307. data[1] = val;
  308. return rtlsdr_write_array(dev, IICB, addr, (uint8_t *)&data, 2);
  309. }
  310. uint8_t rtlsdr_i2c_read_reg(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t reg)
  311. {
  312. uint16_t addr = i2c_addr;
  313. uint8_t data = 0;
  314. rtlsdr_write_array(dev, IICB, addr, &reg, 1);
  315. rtlsdr_read_array(dev, IICB, addr, &data, 1);
  316. return data;
  317. }
  318. /* TODO clean this up again */
  319. int e4k_reg_write(struct e4k_state *e4k, uint8_t reg, uint8_t val)
  320. {
  321. return rtlsdr_i2c_write_reg((rtlsdr_dev_t*)e4k->rtl_dev, e4k->i2c_addr, reg, val);}
  322. uint8_t e4k_reg_read(struct e4k_state *e4k, uint8_t reg)
  323. {
  324. return rtlsdr_i2c_read_reg((rtlsdr_dev_t*)e4k->rtl_dev, e4k->i2c_addr, reg);
  325. }
  326. int rtlsdr_i2c_write(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t *buffer, int len)
  327. {
  328. uint16_t addr = i2c_addr;
  329. if (!dev)
  330. return -1;
  331. return rtlsdr_write_array(dev, IICB, addr, buffer, len);
  332. }
  333. int rtlsdr_i2c_read(rtlsdr_dev_t *dev, uint8_t i2c_addr, uint8_t *buffer, int len)
  334. {
  335. uint16_t addr = i2c_addr;
  336. if (!dev)
  337. return -1;
  338. return rtlsdr_read_array(dev, IICB, addr, buffer, len);
  339. }
  340. uint16_t rtlsdr_read_reg(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint8_t len)
  341. {
  342. int r;
  343. unsigned char data[2];
  344. uint16_t index = (block << 8);
  345. uint16_t reg;
  346. r = libusb_control_transfer(dev->devh, CTRL_IN, 0, addr, index, data, len, CTRL_TIMEOUT);
  347. if (r < 0)
  348. fprintf(stderr, "%s failed with %d\n", __FUNCTION__, r);
  349. reg = (data[1] << 8) | data[0];
  350. return reg;
  351. }
  352. int rtlsdr_write_reg(rtlsdr_dev_t *dev, uint8_t block, uint16_t addr, uint16_t val, uint8_t len)
  353. {
  354. int r;
  355. unsigned char data[2];
  356. uint16_t index = (block << 8) | 0x10;
  357. if (len == 1)
  358. data[0] = val & 0xff;
  359. else
  360. data[0] = val >> 8;
  361. data[1] = val & 0xff;
  362. r = libusb_control_transfer(dev->devh, CTRL_OUT, 0, addr, index, data, len, CTRL_TIMEOUT);
  363. if (r < 0)
  364. fprintf(stderr, "%s failed with %d\n", __FUNCTION__, r);
  365. return r;
  366. }
  367. uint16_t rtlsdr_demod_read_reg(rtlsdr_dev_t *dev, uint8_t page, uint16_t addr, uint8_t len)
  368. {
  369. int r;
  370. unsigned char data[2];
  371. uint16_t index = page;
  372. uint16_t reg;
  373. addr = (addr << 8) | 0x20;
  374. r = libusb_control_transfer(dev->devh, CTRL_IN, 0, addr, index, data, len, CTRL_TIMEOUT);
  375. if (r < 0)
  376. fprintf(stderr, "%s failed with %d\n", __FUNCTION__, r);
  377. reg = (data[1] << 8) | data[0];
  378. return reg;
  379. }
  380. int rtlsdr_demod_write_reg(rtlsdr_dev_t *dev, uint8_t page, uint16_t addr, uint16_t val, uint8_t len)
  381. {
  382. int r;
  383. unsigned char data[2];
  384. uint16_t index = 0x10 | page;
  385. addr = (addr << 8) | 0x20;
  386. if (len == 1)
  387. data[0] = val & 0xff;
  388. else
  389. data[0] = val >> 8;
  390. data[1] = val & 0xff;
  391. r = libusb_control_transfer(dev->devh, CTRL_OUT, 0, addr, index, data, len, CTRL_TIMEOUT);
  392. if (r < 0)
  393. fprintf(stderr, "%s failed with %d\n", __FUNCTION__, r);
  394. rtlsdr_demod_read_reg(dev, 0x0a, 0x01, 1);
  395. return (r == len) ? 0 : -1;
  396. }
  397. void rtlsdr_set_gpio_bit(rtlsdr_dev_t *dev, uint8_t gpio, int val)
  398. {
  399. uint8_t r;
  400. gpio = 1 << gpio;
  401. r = rtlsdr_read_reg(dev, SYSB, GPO, 1);
  402. r = val ? (r | gpio) : (r & ~gpio);
  403. rtlsdr_write_reg(dev, SYSB, GPO, r, 1);
  404. }
  405. void rtlsdr_set_gpio_output(rtlsdr_dev_t *dev, uint8_t gpio)
  406. {
  407. int r;
  408. gpio = 1 << gpio;
  409. r = rtlsdr_read_reg(dev, SYSB, GPD, 1);
  410. rtlsdr_write_reg(dev, SYSB, GPO, r & ~gpio, 1);
  411. r = rtlsdr_read_reg(dev, SYSB, GPOE, 1);
  412. rtlsdr_write_reg(dev, SYSB, GPOE, r | gpio, 1);
  413. }
  414. void rtlsdr_set_i2c_repeater(rtlsdr_dev_t *dev, int on)
  415. {
  416. rtlsdr_demod_write_reg(dev, 1, 0x01, on ? 0x18 : 0x10, 1);
  417. }
  418. void rtlsdr_init_baseband(rtlsdr_dev_t *dev)
  419. {
  420. unsigned int i;
  421. /* default FIR coefficients used for DAB/FM by the Windows driver,
  422. * the DVB driver uses different ones */
  423. uint8_t fir_coeff[] = {
  424. 0xca, 0xdc, 0xd7, 0xd8, 0xe0, 0xf2, 0x0e, 0x35, 0x06, 0x50,
  425. 0x9c, 0x0d, 0x71, 0x11, 0x14, 0x71, 0x74, 0x19, 0x41, 0xa5,
  426. };
  427. /* initialize USB */
  428. rtlsdr_write_reg(dev, USBB, USB_SYSCTL, 0x09, 1);
  429. rtlsdr_write_reg(dev, USBB, USB_EPA_MAXPKT, 0x0002, 2);
  430. rtlsdr_write_reg(dev, USBB, USB_EPA_CTL, 0x1002, 2);
  431. /* poweron demod */
  432. rtlsdr_write_reg(dev, SYSB, DEMOD_CTL_1, 0x22, 1);
  433. rtlsdr_write_reg(dev, SYSB, DEMOD_CTL, 0xe8, 1);
  434. /* reset demod (bit 3, soft_rst) */
  435. rtlsdr_demod_write_reg(dev, 1, 0x01, 0x14, 1);
  436. rtlsdr_demod_write_reg(dev, 1, 0x01, 0x10, 1);
  437. /* disable spectrum inversion and adjacent channel rejection */
  438. rtlsdr_demod_write_reg(dev, 1, 0x15, 0x00, 1);
  439. rtlsdr_demod_write_reg(dev, 1, 0x16, 0x0000, 2);
  440. /* clear both DDC shift and IF frequency registers */
  441. for (i = 0; i < 6; i++)
  442. rtlsdr_demod_write_reg(dev, 1, 0x16 + i, 0x00, 1);
  443. /* set FIR coefficients */
  444. for (i = 0; i < sizeof (fir_coeff); i++)
  445. rtlsdr_demod_write_reg(dev, 1, 0x1c + i, fir_coeff[i], 1);
  446. /* enable SDR mode, disable DAGC (bit 5) */
  447. rtlsdr_demod_write_reg(dev, 0, 0x19, 0x05, 1);
  448. /* init FSM state-holding register */
  449. rtlsdr_demod_write_reg(dev, 1, 0x93, 0xf0, 1);
  450. rtlsdr_demod_write_reg(dev, 1, 0x94, 0x0f, 1);
  451. /* disable AGC (en_dagc, bit 0) (this seems to have no effect) */
  452. rtlsdr_demod_write_reg(dev, 1, 0x11, 0x00, 1);
  453. /* disable RF and IF AGC loop */
  454. rtlsdr_demod_write_reg(dev, 1, 0x04, 0x00, 1);
  455. /* disable PID filter (enable_PID = 0) */
  456. rtlsdr_demod_write_reg(dev, 0, 0x61, 0x60, 1);
  457. /* opt_adc_iq = 0, default ADC_I/ADC_Q datapath */
  458. rtlsdr_demod_write_reg(dev, 0, 0x06, 0x80, 1);
  459. /* Enable Zero-IF mode (en_bbin bit), DC cancellation (en_dc_est),
  460. * IQ estimation/compensation (en_iq_comp, en_iq_est) */
  461. rtlsdr_demod_write_reg(dev, 1, 0xb1, 0x1b, 1);
  462. /* disable 4.096 MHz clock output on pin TP_CK0 */
  463. rtlsdr_demod_write_reg(dev, 0, 0x0d, 0x83, 1);
  464. }
  465. int rtlsdr_deinit_baseband(rtlsdr_dev_t *dev)
  466. {
  467. int r = 0;
  468. if (!dev)
  469. return -1;
  470. if (dev->tuner && dev->tuner->exit) {
  471. rtlsdr_set_i2c_repeater(dev, 1);
  472. r = dev->tuner->exit(dev); /* deinitialize tuner */
  473. rtlsdr_set_i2c_repeater(dev, 0);
  474. }
  475. /* poweroff demodulator and ADCs */
  476. rtlsdr_write_reg(dev, SYSB, DEMOD_CTL, 0x20, 1);
  477. return r;
  478. }
  479. int rtlsdr_set_if_freq(rtlsdr_dev_t *dev, uint32_t freq)
  480. {
  481. uint32_t rtl_xtal;
  482. int32_t if_freq;
  483. uint8_t tmp;
  484. int r;
  485. if (!dev)
  486. return -1;
  487. /* read corrected clock value */
  488. if (rtlsdr_get_xtal_freq(dev, &rtl_xtal, NULL))
  489. return -2;
  490. if_freq = ((freq * TWO_POW(22)) / rtl_xtal) * (-1);
  491. tmp = (if_freq >> 16) & 0x3f;
  492. r = rtlsdr_demod_write_reg(dev, 1, 0x19, tmp, 1);
  493. tmp = (if_freq >> 8) & 0xff;
  494. r |= rtlsdr_demod_write_reg(dev, 1, 0x1a, tmp, 1);
  495. tmp = if_freq & 0xff;
  496. r |= rtlsdr_demod_write_reg(dev, 1, 0x1b, tmp, 1);
  497. return r;
  498. }
  499. int rtlsdr_set_sample_freq_correction(rtlsdr_dev_t *dev, int ppm)
  500. {
  501. int r = 0;
  502. uint8_t tmp;
  503. int16_t offs = ppm * (-1) * TWO_POW(24) / 1000000;
  504. tmp = offs & 0xff;
  505. r |= rtlsdr_demod_write_reg(dev, 1, 0x3f, tmp, 1);
  506. tmp = (offs >> 8) & 0x3f;
  507. r |= rtlsdr_demod_write_reg(dev, 1, 0x3e, tmp, 1);
  508. return r;
  509. }
  510. int rtlsdr_set_xtal_freq(rtlsdr_dev_t *dev, uint32_t rtl_freq, uint32_t tuner_freq)
  511. {
  512. int r = 0;
  513. if (!dev)
  514. return -1;
  515. if (rtl_freq > 0 &&
  516. (rtl_freq < MIN_RTL_XTAL_FREQ || rtl_freq > MAX_RTL_XTAL_FREQ))
  517. return -2;
  518. if (rtl_freq > 0 && dev->rtl_xtal != rtl_freq) {
  519. dev->rtl_xtal = rtl_freq;
  520. /* update xtal-dependent settings */
  521. if (dev->rate)
  522. r = rtlsdr_set_sample_rate(dev, dev->rate);
  523. }
  524. if (dev->tun_xtal != tuner_freq) {
  525. if (0 == tuner_freq)
  526. dev->tun_xtal = dev->rtl_xtal;
  527. else
  528. dev->tun_xtal = tuner_freq;
  529. /* read corrected clock value into e4k structure */
  530. if (rtlsdr_get_xtal_freq(dev, NULL, &dev->e4k_s.vco.fosc))
  531. return -3;
  532. /* update xtal-dependent settings */
  533. if (dev->freq)
  534. r = rtlsdr_set_center_freq(dev, dev->freq);
  535. }
  536. return r;
  537. }
  538. int rtlsdr_get_xtal_freq(rtlsdr_dev_t *dev, uint32_t *rtl_freq, uint32_t *tuner_freq)
  539. {
  540. if (!dev)
  541. return -1;
  542. #define APPLY_PPM_CORR(val,ppm) (((val) * (1.0 + (ppm) / 1e6)))
  543. if (rtl_freq)
  544. *rtl_freq = (uint32_t) APPLY_PPM_CORR(dev->rtl_xtal, dev->corr);
  545. if (tuner_freq)
  546. *tuner_freq = (uint32_t) APPLY_PPM_CORR(dev->tun_xtal, dev->corr);
  547. return 0;
  548. }
  549. int rtlsdr_get_usb_strings(rtlsdr_dev_t *dev, char *manufact, char *product,
  550. char *serial)
  551. {
  552. struct libusb_device_descriptor dd;
  553. libusb_device *device = NULL;
  554. const int buf_max = 256;
  555. int r = 0;
  556. if (!dev || !dev->devh)
  557. return -1;
  558. device = libusb_get_device(dev->devh);
  559. r = libusb_get_device_descriptor(device, &dd);
  560. if (r < 0)
  561. return -1;
  562. if (manufact) {
  563. memset(manufact, 0, buf_max);
  564. libusb_get_string_descriptor_ascii(dev->devh, dd.iManufacturer,
  565. (unsigned char *)manufact,
  566. buf_max);
  567. }
  568. if (product) {
  569. memset(product, 0, buf_max);
  570. libusb_get_string_descriptor_ascii(dev->devh, dd.iProduct,
  571. (unsigned char *)product,
  572. buf_max);
  573. }
  574. if (serial) {
  575. memset(serial, 0, buf_max);
  576. libusb_get_string_descriptor_ascii(dev->devh, dd.iSerialNumber,
  577. (unsigned char *)serial,
  578. buf_max);
  579. }
  580. return 0;
  581. }
  582. int rtlsdr_set_center_freq(rtlsdr_dev_t *dev, uint32_t freq)
  583. {
  584. int r = -1;
  585. if (!dev || !dev->tuner)
  586. return -1;
  587. if (dev->direct_sampling) {
  588. r = rtlsdr_set_if_freq(dev, freq);
  589. } else if (dev->tuner && dev->tuner->set_freq) {
  590. rtlsdr_set_i2c_repeater(dev, 1);
  591. r = dev->tuner->set_freq(dev, freq - dev->offs_freq);
  592. rtlsdr_set_i2c_repeater(dev, 0);
  593. }
  594. if (!r)
  595. dev->freq = freq;
  596. else
  597. dev->freq = 0;
  598. return r;
  599. }
  600. uint32_t rtlsdr_get_center_freq(rtlsdr_dev_t *dev)
  601. {
  602. if (!dev)
  603. return 0;
  604. return dev->freq;
  605. }
  606. int rtlsdr_set_freq_correction(rtlsdr_dev_t *dev, int ppm)
  607. {
  608. int r = 0;
  609. if (!dev)
  610. return -1;
  611. if (dev->corr == ppm)
  612. return -2;
  613. dev->corr = ppm;
  614. r |= rtlsdr_set_sample_freq_correction(dev, ppm);
  615. /* read corrected clock value into e4k structure */
  616. if (rtlsdr_get_xtal_freq(dev, NULL, &dev->e4k_s.vco.fosc))
  617. return -3;
  618. if (dev->freq) /* retune to apply new correction value */
  619. r |= rtlsdr_set_center_freq(dev, dev->freq);
  620. return r;
  621. }
  622. int rtlsdr_get_freq_correction(rtlsdr_dev_t *dev)
  623. {
  624. if (!dev)
  625. return 0;
  626. return dev->corr;
  627. }
  628. enum rtlsdr_tuner rtlsdr_get_tuner_type(rtlsdr_dev_t *dev)
  629. {
  630. if (!dev)
  631. return RTLSDR_TUNER_UNKNOWN;
  632. return dev->tuner_type;
  633. }
  634. int rtlsdr_get_tuner_gains(rtlsdr_dev_t *dev, int *gains)
  635. {
  636. /* all gain values are expressed in tenths of a dB */
  637. const int e4k_gains[] = { -10, 15, 40, 65, 90, 115, 140, 165, 190, 215,
  638. 240, 290, 340, 420 };
  639. const int fc0012_gains[] = { -99, -40, 71, 179, 192 };
  640. const int fc0013_gains[] = { -99, -73, -65, -63, -60, -58, -54, 58, 61,
  641. 63, 65, 67, 68, 70, 71, 179, 181, 182,
  642. 184, 186, 188, 191, 197 };
  643. const int fc2580_gains[] = { 0 /* no gain values */ };
  644. const int r820t_gains[] = { 0 /* no gain values */ };
  645. const int unknown_gains[] = { 0 /* no gain values */ };
  646. int *ptr = NULL;
  647. int len = 0;
  648. if (!dev)
  649. return -1;
  650. switch (dev->tuner_type) {
  651. case RTLSDR_TUNER_E4000:
  652. ptr = (int *)e4k_gains; len = sizeof(e4k_gains);
  653. break;
  654. case RTLSDR_TUNER_FC0012:
  655. ptr = (int *)fc0012_gains; len = sizeof(fc0012_gains);
  656. break;
  657. case RTLSDR_TUNER_FC0013:
  658. ptr = (int *)fc0013_gains; len = sizeof(fc0013_gains);
  659. break;
  660. case RTLSDR_TUNER_FC2580:
  661. ptr = (int *)fc2580_gains; len = sizeof(fc2580_gains);
  662. break;
  663. case RTLSDR_TUNER_R820T:
  664. ptr = (int *)r820t_gains; len = sizeof(r820t_gains);
  665. break;
  666. default:
  667. ptr = (int *)unknown_gains; len = sizeof(unknown_gains);
  668. break;
  669. }
  670. if (!gains) { /* no buffer provided, just return the count */
  671. return len / sizeof(int);
  672. } else {
  673. if (len)
  674. memcpy(gains, ptr, len);
  675. return len / sizeof(int);
  676. }
  677. }
  678. int rtlsdr_set_tuner_gain(rtlsdr_dev_t *dev, int gain)
  679. {
  680. int r = 0;
  681. if (!dev || !dev->tuner)
  682. return -1;
  683. if (dev->tuner->set_gain) {
  684. rtlsdr_set_i2c_repeater(dev, 1);
  685. r = dev->tuner->set_gain((void *)dev, gain);
  686. rtlsdr_set_i2c_repeater(dev, 0);
  687. }
  688. if (!r)
  689. dev->gain = gain;
  690. else
  691. dev->gain = 0;
  692. return r;
  693. }
  694. int rtlsdr_get_tuner_gain(rtlsdr_dev_t *dev)
  695. {
  696. if (!dev)
  697. return 0;
  698. return dev->gain;
  699. }
  700. int rtlsdr_set_tuner_if_gain(rtlsdr_dev_t *dev, int stage, int gain)
  701. {
  702. int r = 0;
  703. if (!dev || !dev->tuner)
  704. return -1;
  705. if (dev->tuner->set_if_gain) {
  706. rtlsdr_set_i2c_repeater(dev, 1);
  707. r = dev->tuner->set_if_gain(dev, stage, gain);
  708. rtlsdr_set_i2c_repeater(dev, 0);
  709. }
  710. return r;
  711. }
  712. int rtlsdr_set_tuner_gain_mode(rtlsdr_dev_t *dev, int mode)
  713. {
  714. int r = 0;
  715. if (!dev || !dev->tuner)
  716. return -1;
  717. if (dev->tuner->set_gain_mode) {
  718. rtlsdr_set_i2c_repeater(dev, 1);
  719. r = dev->tuner->set_gain_mode((void *)dev, mode);
  720. rtlsdr_set_i2c_repeater(dev, 0);
  721. }
  722. return r;
  723. }
  724. int rtlsdr_set_sample_rate(rtlsdr_dev_t *dev, uint32_t samp_rate)
  725. {
  726. int r = 0;
  727. uint16_t tmp;
  728. uint32_t rsamp_ratio;
  729. double real_rate;
  730. uint32_t rtl_freq = dev->rtl_xtal;
  731. if (!dev)
  732. return -1;
  733. /* check for the maximum rate the resampler supports */
  734. if (samp_rate > MAX_SAMP_RATE)
  735. samp_rate = MAX_SAMP_RATE;
  736. rsamp_ratio = (rtl_freq * TWO_POW(22)) / samp_rate;
  737. rsamp_ratio &= ~3;
  738. real_rate = (rtl_freq * TWO_POW(22)) / rsamp_ratio;
  739. if ( ((double)samp_rate) != real_rate )
  740. fprintf(stderr, "Exact sample rate is: %f Hz\n", real_rate);
  741. if (dev->tuner && dev->tuner->set_bw) {
  742. rtlsdr_set_i2c_repeater(dev, 1);
  743. dev->tuner->set_bw(dev, (int)real_rate);
  744. rtlsdr_set_i2c_repeater(dev, 0);
  745. }
  746. dev->rate = (uint32_t)real_rate;
  747. tmp = (rsamp_ratio >> 16);
  748. r |= rtlsdr_demod_write_reg(dev, 1, 0x9f, tmp, 2);
  749. tmp = rsamp_ratio & 0xffff;
  750. r |= rtlsdr_demod_write_reg(dev, 1, 0xa1, tmp, 2);
  751. r |= rtlsdr_set_sample_freq_correction(dev, dev->corr);
  752. /* reset demod (bit 3, soft_rst) */
  753. r |= rtlsdr_demod_write_reg(dev, 1, 0x01, 0x14, 1);
  754. r |= rtlsdr_demod_write_reg(dev, 1, 0x01, 0x10, 1);
  755. return r;
  756. }
  757. uint32_t rtlsdr_get_sample_rate(rtlsdr_dev_t *dev)
  758. {
  759. if (!dev)
  760. return 0;
  761. return dev->rate;
  762. }
  763. int rtlsdr_set_testmode(rtlsdr_dev_t *dev, int on)
  764. {
  765. if (!dev)
  766. return -1;
  767. return rtlsdr_demod_write_reg(dev, 0, 0x19, on ? 0x03 : 0x05, 1);
  768. }
  769. int rtlsdr_set_agc_mode(rtlsdr_dev_t *dev, int on)
  770. {
  771. if (!dev)
  772. return -1;
  773. return rtlsdr_demod_write_reg(dev, 0, 0x19, on ? 0x25 : 0x05, 1);
  774. }
  775. int rtlsdr_set_direct_sampling(rtlsdr_dev_t *dev, int on)
  776. {
  777. int r = 0;
  778. if (!dev)
  779. return -1;
  780. if (on) {
  781. if (dev->tuner && dev->tuner->exit) {
  782. rtlsdr_set_i2c_repeater(dev, 1);
  783. r = dev->tuner->exit(dev);
  784. rtlsdr_set_i2c_repeater(dev, 0);
  785. }
  786. /* disable Zero-IF mode */
  787. r |= rtlsdr_demod_write_reg(dev, 1, 0xb1, 0x1a, 1);
  788. /* disable spectrum inversion */
  789. r |= rtlsdr_demod_write_reg(dev, 1, 0x15, 0x00, 1);
  790. /* only enable In-phase ADC input */
  791. r |= rtlsdr_demod_write_reg(dev, 0, 0x08, 0x4d, 1);
  792. /* swap I and Q ADC, this allows to select between two inputs */
  793. r |= rtlsdr_demod_write_reg(dev, 0, 0x06, (on > 1) ? 0x90 : 0x80, 1);
  794. fprintf(stderr, "Enabled direct sampling mode, input %i\n", on);
  795. dev->direct_sampling = on;
  796. } else {
  797. if (dev->tuner && dev->tuner->init) {
  798. rtlsdr_set_i2c_repeater(dev, 1);
  799. r |= dev->tuner->init(dev);
  800. rtlsdr_set_i2c_repeater(dev, 0);
  801. }
  802. if (dev->tuner_type == RTLSDR_TUNER_R820T) {
  803. r |= rtlsdr_set_if_freq(dev, R820T_IF_FREQ);
  804. /* enable spectrum inversion */
  805. r |= rtlsdr_demod_write_reg(dev, 1, 0x15, 0x01, 1);
  806. } else {
  807. r |= rtlsdr_set_if_freq(dev, 0);
  808. /* enable In-phase + Quadrature ADC input */
  809. r |= rtlsdr_demod_write_reg(dev, 0, 0x08, 0xcd, 1);
  810. /* Enable Zero-IF mode */
  811. r |= rtlsdr_demod_write_reg(dev, 1, 0xb1, 0x1b, 1);
  812. }
  813. /* opt_adc_iq = 0, default ADC_I/ADC_Q datapath */
  814. r |= rtlsdr_demod_write_reg(dev, 0, 0x06, 0x80, 1);
  815. fprintf(stderr, "Disabled direct sampling mode\n");
  816. dev->direct_sampling = 0;
  817. }
  818. r |= rtlsdr_set_center_freq(dev, dev->freq);
  819. return r;
  820. }
  821. int rtlsdr_set_offset_tuning(rtlsdr_dev_t *dev, int on)
  822. {
  823. int r = 0;
  824. if (!dev)
  825. return -1;
  826. if (dev->tuner_type == RTLSDR_TUNER_R820T)
  827. return -2;
  828. /* based on keenerds 1/f noise measurements */
  829. dev->offs_freq = on ? ((dev->rate / 2) * 170 / 100) : 0;
  830. r |= rtlsdr_set_if_freq(dev, dev->offs_freq);
  831. if (dev->tuner && dev->tuner->set_bw) {
  832. rtlsdr_set_i2c_repeater(dev, 1);
  833. dev->tuner->set_bw(dev, on ? (2 * dev->offs_freq) : dev->rate);
  834. rtlsdr_set_i2c_repeater(dev, 0);
  835. }
  836. r |= rtlsdr_set_center_freq(dev, dev->freq);
  837. return r;
  838. }
  839. static rtlsdr_dongle_t *find_known_device(uint16_t vid, uint16_t pid)
  840. {
  841. unsigned int i;
  842. rtlsdr_dongle_t *device = NULL;
  843. for (i = 0; i < sizeof(known_devices)/sizeof(rtlsdr_dongle_t); i++ ) {
  844. if (known_devices[i].vid == vid && known_devices[i].pid == pid) {
  845. device = &known_devices[i];
  846. break;
  847. }
  848. }
  849. return device;
  850. }
  851. uint32_t rtlsdr_get_device_count(void)
  852. {
  853. int i;
  854. libusb_context *ctx;
  855. libusb_device **list;
  856. uint32_t device_count = 0;
  857. struct libusb_device_descriptor dd;
  858. ssize_t cnt;
  859. libusb_init(&ctx);
  860. cnt = libusb_get_device_list(ctx, &list);
  861. for (i = 0; i < cnt; i++) {
  862. libusb_get_device_descriptor(list[i], &dd);
  863. if (find_known_device(dd.idVendor, dd.idProduct))
  864. device_count++;
  865. }
  866. libusb_free_device_list(list, 1);
  867. libusb_exit(ctx);
  868. return device_count;
  869. }
  870. const char *rtlsdr_get_device_name(uint32_t index)
  871. {
  872. int i;
  873. libusb_context *ctx;
  874. libusb_device **list;
  875. struct libusb_device_descriptor dd;
  876. rtlsdr_dongle_t *device = NULL;
  877. uint32_t device_count = 0;
  878. ssize_t cnt;
  879. libusb_init(&ctx);
  880. cnt = libusb_get_device_list(ctx, &list);
  881. for (i = 0; i < cnt; i++) {
  882. libusb_get_device_descriptor(list[i], &dd);
  883. device = find_known_device(dd.idVendor, dd.idProduct);
  884. if (device) {
  885. device_count++;
  886. if (index == device_count - 1)
  887. break;
  888. }
  889. }
  890. libusb_free_device_list(list, 1);
  891. libusb_exit(ctx);
  892. if (device)
  893. return device->name;
  894. else
  895. return "";
  896. }
  897. int rtlsdr_get_device_usb_strings(uint32_t index, char *manufact,
  898. char *product, char *serial)
  899. {
  900. int r = -2;
  901. int i;
  902. libusb_context *ctx;
  903. libusb_device **list;
  904. struct libusb_device_descriptor dd;
  905. rtlsdr_dongle_t *device = NULL;
  906. rtlsdr_dev_t devt;
  907. uint32_t device_count = 0;
  908. ssize_t cnt;
  909. libusb_init(&ctx);
  910. cnt = libusb_get_device_list(ctx, &list);
  911. for (i = 0; i < cnt; i++) {
  912. libusb_get_device_descriptor(list[i], &dd);
  913. device = find_known_device(dd.idVendor, dd.idProduct);
  914. if (device) {
  915. device_count++;
  916. if (index == device_count - 1) {
  917. r = libusb_open(list[i], &devt.devh);
  918. if (!r) {
  919. r = rtlsdr_get_usb_strings(&devt,
  920. manufact,
  921. product,
  922. serial);
  923. libusb_close(devt.devh);
  924. }
  925. break;
  926. }
  927. }
  928. }
  929. libusb_free_device_list(list, 1);
  930. libusb_exit(ctx);
  931. return r;
  932. }
  933. int rtlsdr_open(rtlsdr_dev_t **out_dev, uint32_t index)
  934. {
  935. int r;
  936. int i;
  937. libusb_device **list;
  938. rtlsdr_dev_t *dev = NULL;
  939. libusb_device *device = NULL;
  940. uint32_t device_count = 0;
  941. struct libusb_device_descriptor dd;
  942. uint8_t reg;
  943. ssize_t cnt;
  944. dev = malloc(sizeof(rtlsdr_dev_t));
  945. if (NULL == dev)
  946. return -ENOMEM;
  947. memset(dev, 0, sizeof(rtlsdr_dev_t));
  948. libusb_init(&dev->ctx);
  949. cnt = libusb_get_device_list(dev->ctx, &list);
  950. for (i = 0; i < cnt; i++) {
  951. device = list[i];
  952. libusb_get_device_descriptor(list[i], &dd);
  953. if (find_known_device(dd.idVendor, dd.idProduct)) {
  954. device_count++;
  955. }
  956. if (index == device_count - 1)
  957. break;
  958. device = NULL;
  959. }
  960. if (!device) {
  961. r = -1;
  962. goto err;
  963. }
  964. r = libusb_open(device, &dev->devh);
  965. if (r < 0) {
  966. libusb_free_device_list(list, 1);
  967. fprintf(stderr, "usb_open error %d\n", r);
  968. goto err;
  969. }
  970. libusb_free_device_list(list, 1);
  971. r = libusb_claim_interface(dev->devh, 0);
  972. if (r < 0) {
  973. fprintf(stderr, "usb_claim_interface error %d\n", r);
  974. goto err;
  975. }
  976. dev->rtl_xtal = DEF_RTL_XTAL_FREQ;
  977. rtlsdr_init_baseband(dev);
  978. /* Probe tuners */
  979. rtlsdr_set_i2c_repeater(dev, 1);
  980. reg = rtlsdr_i2c_read_reg(dev, E4K_I2C_ADDR, E4K_CHECK_ADDR);
  981. if (reg == E4K_CHECK_VAL) {
  982. fprintf(stderr, "Found Elonics E4000 tuner\n");
  983. dev->tuner_type = RTLSDR_TUNER_E4000;
  984. goto found;
  985. }
  986. reg = rtlsdr_i2c_read_reg(dev, FC0013_I2C_ADDR, FC0013_CHECK_ADDR);
  987. if (reg == FC0013_CHECK_VAL) {
  988. fprintf(stderr, "Found Fitipower FC0013 tuner\n");
  989. dev->tuner_type = RTLSDR_TUNER_FC0013;
  990. goto found;
  991. }
  992. reg = rtlsdr_i2c_read_reg(dev, R820T_I2C_ADDR, R820T_CHECK_ADDR);
  993. if (reg == R820T_CHECK_VAL) {
  994. fprintf(stderr, "Found Rafael Micro R820T tuner\n");
  995. dev->tuner_type = RTLSDR_TUNER_R820T;
  996. /* disable Zero-IF mode */
  997. rtlsdr_demod_write_reg(dev, 1, 0xb1, 0x1a, 1);
  998. /* only enable In-phase ADC input */
  999. rtlsdr_demod_write_reg(dev, 0, 0x08, 0x4d, 1);
  1000. /* the R820T uses 3.57 MHz IF for the DVB-T 6 MHz mode, and
  1001. * 4.57 MHz for the 8 MHz mode */
  1002. rtlsdr_set_if_freq(dev, R820T_IF_FREQ);
  1003. /* enable spectrum inversion */
  1004. rtlsdr_demod_write_reg(dev, 1, 0x15, 0x01, 1);
  1005. goto found;
  1006. }
  1007. /* initialise GPIOs */
  1008. rtlsdr_set_gpio_output(dev, 5);
  1009. /* reset tuner before probing */
  1010. rtlsdr_set_gpio_bit(dev, 5, 1);
  1011. rtlsdr_set_gpio_bit(dev, 5, 0);
  1012. reg = rtlsdr_i2c_read_reg(dev, FC2580_I2C_ADDR, FC2580_CHECK_ADDR);
  1013. if ((reg & 0x7f) == FC2580_CHECK_VAL) {
  1014. fprintf(stderr, "Found FCI 2580 tuner\n");
  1015. dev->tuner_type = RTLSDR_TUNER_FC2580;
  1016. goto found;
  1017. }
  1018. reg = rtlsdr_i2c_read_reg(dev, FC0012_I2C_ADDR, FC0012_CHECK_ADDR);
  1019. if (reg == FC0012_CHECK_VAL) {
  1020. fprintf(stderr, "Found Fitipower FC0012 tuner\n");
  1021. rtlsdr_set_gpio_output(dev, 6);
  1022. dev->tuner_type = RTLSDR_TUNER_FC0012;
  1023. goto found;
  1024. }
  1025. found:
  1026. if (dev->tuner_type == RTLSDR_TUNER_UNKNOWN) {
  1027. fprintf(stderr, "No supported tuner found\n");
  1028. rtlsdr_set_direct_sampling(dev, 1);
  1029. }
  1030. dev->tuner = &tuners[dev->tuner_type];
  1031. dev->tun_xtal = dev->rtl_xtal; /* use the rtl clock value by default */
  1032. if (dev->tuner->init)
  1033. r = dev->tuner->init(dev);
  1034. rtlsdr_set_i2c_repeater(dev, 0);
  1035. *out_dev = dev;
  1036. return 0;
  1037. err:
  1038. if (dev) {
  1039. if (dev->ctx)
  1040. libusb_exit(dev->ctx);
  1041. free(dev);
  1042. }
  1043. return r;
  1044. }
  1045. int rtlsdr_close(rtlsdr_dev_t *dev)
  1046. {
  1047. if (!dev)
  1048. return -1;
  1049. /* block until all async operations have been completed (if any) */
  1050. while (RTLSDR_INACTIVE != dev->async_status) {
  1051. #ifdef _WIN32
  1052. Sleep(1);
  1053. #else
  1054. usleep(1000);
  1055. #endif
  1056. }
  1057. rtlsdr_deinit_baseband(dev);
  1058. libusb_release_interface(dev->devh, 0);
  1059. libusb_close(dev->devh);
  1060. libusb_exit(dev->ctx);
  1061. free(dev);
  1062. return 0;
  1063. }
  1064. int rtlsdr_reset_buffer(rtlsdr_dev_t *dev)
  1065. {
  1066. if (!dev)
  1067. return -1;
  1068. rtlsdr_write_reg(dev, USBB, USB_EPA_CTL, 0x1002, 2);
  1069. rtlsdr_write_reg(dev, USBB, USB_EPA_CTL, 0x0000, 2);
  1070. return 0;
  1071. }
  1072. int rtlsdr_read_sync(rtlsdr_dev_t *dev, void *buf, int len, int *n_read)
  1073. {
  1074. if (!dev)
  1075. return -1;
  1076. return libusb_bulk_transfer(dev->devh, 0x81, buf, len, n_read, BULK_TIMEOUT);
  1077. }
  1078. static void LIBUSB_CALL _libusb_callback(struct libusb_transfer *xfer)
  1079. {
  1080. rtlsdr_dev_t *dev = (rtlsdr_dev_t *)xfer->user_data;
  1081. if (LIBUSB_TRANSFER_COMPLETED == xfer->status) {
  1082. if (dev->cb)
  1083. dev->cb(xfer->buffer, xfer->actual_length, dev->cb_ctx);
  1084. libusb_submit_transfer(xfer); /* resubmit transfer */
  1085. } else if (LIBUSB_TRANSFER_CANCELLED == xfer->status) {
  1086. /* nothing to do */
  1087. } else {
  1088. /*fprintf(stderr, "transfer status: %d\n", xfer->status);*/
  1089. }
  1090. }
  1091. int rtlsdr_wait_async(rtlsdr_dev_t *dev, rtlsdr_read_async_cb_t cb, void *ctx)
  1092. {
  1093. return rtlsdr_read_async(dev, cb, ctx, 0, 0);
  1094. }
  1095. static int _rtlsdr_alloc_async_buffers(rtlsdr_dev_t *dev)
  1096. {
  1097. unsigned int i;
  1098. if (!dev)
  1099. return -1;
  1100. if (!dev->xfer) {
  1101. dev->xfer = malloc(dev->xfer_buf_num *
  1102. sizeof(struct libusb_transfer *));
  1103. for(i = 0; i < dev->xfer_buf_num; ++i)
  1104. dev->xfer[i] = libusb_alloc_transfer(0);
  1105. }
  1106. if (!dev->xfer_buf) {
  1107. dev->xfer_buf = malloc(dev->xfer_buf_num *
  1108. sizeof(unsigned char *));
  1109. for(i = 0; i < dev->xfer_buf_num; ++i)
  1110. dev->xfer_buf[i] = malloc(dev->xfer_buf_len);
  1111. }
  1112. return 0;
  1113. }
  1114. static int _rtlsdr_free_async_buffers(rtlsdr_dev_t *dev)
  1115. {
  1116. unsigned int i;
  1117. if (!dev)
  1118. return -1;
  1119. if (dev->xfer) {
  1120. for(i = 0; i < dev->xfer_buf_num; ++i) {
  1121. if (dev->xfer[i]) {
  1122. libusb_free_transfer(dev->xfer[i]);
  1123. }
  1124. }
  1125. free(dev->xfer);
  1126. dev->xfer = NULL;
  1127. }
  1128. if (dev->xfer_buf) {
  1129. for(i = 0; i < dev->xfer_buf_num; ++i) {
  1130. if (dev->xfer_buf[i])
  1131. free(dev->xfer_buf[i]);
  1132. }
  1133. free(dev->xfer_buf);
  1134. dev->xfer_buf = NULL;
  1135. }
  1136. return 0;
  1137. }
  1138. int rtlsdr_read_async(rtlsdr_dev_t *dev, rtlsdr_read_async_cb_t cb, void *ctx,
  1139. uint32_t buf_num, uint32_t buf_len)
  1140. {
  1141. unsigned int i;
  1142. int r = 0;
  1143. struct timeval tv = { 1, 0 };
  1144. enum rtlsdr_async_status next_status = RTLSDR_INACTIVE;
  1145. if (!dev)
  1146. return -1;
  1147. if (RTLSDR_INACTIVE != dev->async_status)
  1148. return -2;
  1149. dev->async_status = RTLSDR_RUNNING;
  1150. dev->cb = cb;
  1151. dev->cb_ctx = ctx;
  1152. if (buf_num > 0)
  1153. dev->xfer_buf_num = buf_num;
  1154. else
  1155. dev->xfer_buf_num = DEFAULT_BUF_NUMBER;
  1156. if (buf_len > 0 && buf_len % 512 == 0) /* len must be multiple of 512 */
  1157. dev->xfer_buf_len = buf_len;
  1158. else
  1159. dev->xfer_buf_len = DEFAULT_BUF_LENGTH;
  1160. _rtlsdr_alloc_async_buffers(dev);
  1161. for(i = 0; i < dev->xfer_buf_num; ++i) {
  1162. libusb_fill_bulk_transfer(dev->xfer[i],
  1163. dev->devh,
  1164. 0x81,
  1165. dev->xfer_buf[i],
  1166. dev->xfer_buf_len,
  1167. _libusb_callback,
  1168. (void *)dev,
  1169. BULK_TIMEOUT);
  1170. libusb_submit_transfer(dev->xfer[i]);
  1171. }
  1172. while (RTLSDR_INACTIVE != dev->async_status) {
  1173. r = libusb_handle_events_timeout(dev->ctx, &tv);
  1174. if (r < 0) {
  1175. /*fprintf(stderr, "handle_events returned: %d\n", r);*/
  1176. if (r == LIBUSB_ERROR_INTERRUPTED) /* stray signal */
  1177. continue;
  1178. break;
  1179. }
  1180. if (RTLSDR_CANCELING == dev->async_status) {
  1181. next_status = RTLSDR_INACTIVE;
  1182. if (!dev->xfer)
  1183. break;
  1184. for(i = 0; i < dev->xfer_buf_num; ++i) {
  1185. if (!dev->xfer[i])
  1186. continue;
  1187. if (LIBUSB_TRANSFER_CANCELLED !=
  1188. dev->xfer[i]->status) {
  1189. libusb_cancel_transfer(dev->xfer[i]);
  1190. next_status = RTLSDR_CANCELING;
  1191. }
  1192. }
  1193. if (RTLSDR_INACTIVE == next_status)
  1194. break;
  1195. }
  1196. }
  1197. _rtlsdr_free_async_buffers(dev);
  1198. dev->async_status = next_status;
  1199. return r;
  1200. }
  1201. int rtlsdr_cancel_async(rtlsdr_dev_t *dev)
  1202. {
  1203. if (!dev)
  1204. return -1;
  1205. /* if streaming, try to cancel gracefully */
  1206. if (RTLSDR_RUNNING == dev->async_status) {
  1207. dev->async_status = RTLSDR_CANCELING;
  1208. return 0;
  1209. }
  1210. /* if called while in pending state, change the state forcefully */
  1211. if (RTLSDR_INACTIVE != dev->async_status) {
  1212. dev->async_status = RTLSDR_INACTIVE;
  1213. return 0;
  1214. }
  1215. return -2;
  1216. }
  1217. uint32_t rtlsdr_get_tuner_clock(void *dev)
  1218. {
  1219. uint32_t tuner_freq;
  1220. if (!dev)
  1221. return 0;
  1222. /* read corrected clock value */
  1223. if (rtlsdr_get_xtal_freq((rtlsdr_dev_t *)dev, NULL, &tuner_freq))
  1224. return 0;
  1225. return tuner_freq;
  1226. }
  1227. int rtlsdr_i2c_write_fn(void *dev, uint8_t addr, uint8_t *buf, int len)
  1228. {
  1229. if (dev)
  1230. return rtlsdr_i2c_write(((rtlsdr_dev_t *)dev), addr, buf, len);
  1231. return -1;
  1232. }
  1233. int rtlsdr_i2c_read_fn(void *dev, uint8_t addr, uint8_t *buf, int len)
  1234. {
  1235. if (dev)
  1236. return rtlsdr_i2c_read(((rtlsdr_dev_t *)dev), addr, buf, len);
  1237. return -1;
  1238. }