librtlsdr.c 39 KB

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