1 files changed, 415 insertions, 0 deletions
diff --git a/src/main.c b/src/main.c
new file mode 100644
index 0000000..833d81f
--- /dev/null
+++ b/src/main.c
@@ -0,0 +1,415 @@
+/*
+ * rtl-sdr, a poor man's SDR using a Realtek RTL2832 based DVB-stick
+ * Copyright (C) 2012 by Steve Markgraf <steve@steve-m.de>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 2 of the License, or
+ *(at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <errno.h>
+#include <signal.h>
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include <libusb.h>
+
+#include "tuner_e4000.h"
+#include "tuner_fc0013.h"
+
+#define READLEN		(16 * 16384)
+#define CTRL_IN		(LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_IN)
+#define CTRL_OUT	(LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT)
+
+/* ezcap USB 2.0 DVB-T/DAB/FM stick */
+#define EZCAP_VID	0x0bda
+#define EZCAP_PID	0x2838
+
+/* Terratec NOXON DAB/DAB+ USB-Stick */
+#define NOXON_VID	0x0ccd
+#define NOXON_PID	0x00b3
+
+static struct libusb_device_handle *devh = NULL;
+static int do_exit = 0;
+
+enum TUNER_TYPE {
+	TUNER_E4000,
+	TUNER_FC0013
+} tuner_type;
+
+static int find_device(void)
+{
+	devh = libusb_open_device_with_vid_pid(NULL, EZCAP_VID, EZCAP_PID);
+	if (devh > 0) {
+		tuner_type = TUNER_E4000;
+		printf("Found ezcap stick with E4000 tuner\n");
+		return 0;
+	}
+
+	devh = libusb_open_device_with_vid_pid(NULL, NOXON_VID, NOXON_PID);
+	if (devh > 0) {
+		tuner_type = TUNER_FC0013;
+		printf("Found Terratec NOXON stick with FC0013 tuner\n");
+		return 0;
+	}
+
+	return -EIO;
+}
+
+enum usb_reg {
+	USB_SYSCTL		= 0x2000,
+	USB_CTRL		= 0x2010,
+	USB_STAT		= 0x2014,
+	USB_EPA_CFG		= 0x2144,
+	USB_EPA_CTL		= 0x2148,
+	USB_EPA_MAXPKT		= 0x2158,
+	USB_EPA_MAXPKT_2	= 0x215a,
+	USB_EPA_FIFO_CFG	= 0x2160,
+};
+
+enum sys_reg {
+	DEMOD_CTL		= 0x3000,
+	GPO			= 0x3001,
+	GPI			= 0x3002,
+	GPOE			= 0x3003,
+	GPD			= 0x3004,
+	SYSINTE			= 0x3005,
+	SYSINTS			= 0x3006,
+	GP_CFG0			= 0x3007,
+	GP_CFG1			= 0x3008,
+	SYSINTE_1		= 0x3009,
+	SYSINTS_1		= 0x300a,
+	DEMOD_CTL_1		= 0x300b,
+	IR_SUSPEND		= 0x300c,
+};
+
+enum blocks {
+	DEMODB			= 0,
+	USBB			= 1,
+	SYSB			= 2,
+	TUNB			= 3,
+	ROMB			= 4,
+	IRB			= 5,
+	IICB			= 6,
+};
+
+int rtl_read_array(uint8_t block, uint16_t addr, uint8_t *array, uint8_t len)
+{
+	int r;
+	uint16_t index = (block << 8);
+
+	r = libusb_control_transfer(devh, CTRL_IN, 0, addr, index, array, len, 0);
+
+	return r;
+}
+
+int rtl_write_array(uint8_t block, uint16_t addr, uint8_t *array, uint8_t len)
+{
+	int r;
+	uint16_t index = (block << 8) | 0x10;
+
+	r = libusb_control_transfer(devh, CTRL_OUT, 0, addr, index, array, len, 0);
+
+	return r;
+}
+
+int rtl_i2c_write(uint8_t i2c_addr, uint8_t *buffer, int len)
+{
+	uint16_t addr = i2c_addr;
+	return rtl_write_array(IICB, addr, buffer, len);
+}
+
+int rtl_i2c_read(uint8_t i2c_addr, uint8_t *buffer, int len)
+{
+	uint16_t addr = i2c_addr;
+	return rtl_read_array(IICB, addr, buffer, len);
+}
+
+uint16_t rtl_read_reg(uint8_t block, uint16_t addr, uint8_t len)
+{
+	int r;
+	unsigned char data[2];
+	uint16_t index = (block << 8);
+	uint16_t reg;
+
+	r = libusb_control_transfer(devh, CTRL_IN, 0, addr, index, data, len, 0);
+
+	if (r < 0)
+		printf("%s failed\n", __FUNCTION__);
+
+	reg = (data[1] << 8) | data[0];
+
+	return reg;
+}
+
+void rtl_write_reg(uint8_t block, uint16_t addr, uint16_t val, uint8_t len)
+{
+	int r;
+	unsigned char data[2];
+
+	uint16_t index = (block << 8) | 0x10;
+
+	if (len == 1)
+		data[0] = val & 0xff;
+	else
+		data[0] = val >> 8;
+
+	data[1] = val & 0xff;
+
+	r = libusb_control_transfer(devh, CTRL_OUT, 0, addr, index, data, len, 0);
+
+	if (r < 0)
+		printf("%s failed\n", __FUNCTION__);
+}
+
+uint16_t demod_read_reg(uint8_t page, uint8_t addr, uint8_t len)
+{
+	int r;
+	unsigned char data[2];
+
+	uint16_t index = page;
+	uint16_t reg;
+	addr = (addr << 8) | 0x20;
+
+	r = libusb_control_transfer(devh, CTRL_IN, 0, addr, index, data, len, 0);
+
+	if (r < 0)
+		printf("%s failed\n", __FUNCTION__);
+
+	reg = (data[1] << 8) | data[0];
+
+	return reg;
+}
+
+void demod_write_reg(uint8_t page, uint16_t addr, uint16_t val, uint8_t len)
+{
+	int r;
+	unsigned char data[2];
+	uint16_t index = 0x10 | page;
+	addr = (addr << 8) | 0x20;
+
+	if (len == 1)
+		data[0] = val & 0xff;
+	else
+		data[0] = val >> 8;
+
+	data[1] = val & 0xff;
+
+	r = libusb_control_transfer(devh, CTRL_OUT, 0, addr, index, data, len, 0);
+
+	if (r < 0)
+		printf("%s failed\n", __FUNCTION__);
+
+	demod_read_reg(0x0a, 0x01, 1);
+}
+
+void set_resampler(uint32_t rsamp_ratio)
+{
+	uint16_t tmp;
+	rsamp_ratio <<= 2;
+
+	tmp = (rsamp_ratio >> 16) & 0xffff;
+	demod_write_reg(1, 0x9f, tmp, 2);
+	tmp = rsamp_ratio & 0xffff;
+	demod_write_reg(1, 0xa1, tmp, 2);
+}
+
+void set_i2c_repeater(int on)
+{
+	demod_write_reg(1, 0x01, on ? 0x18 : 0x10, 1);
+}
+
+void rtl_init(void)
+{
+	unsigned int i;
+
+	/* default FIR coefficients used for DAB/FM by the Windows driver,
+	 * the DVB driver uses different ones */
+	uint8_t fir_coeff[] = {
+		0xca, 0xdc, 0xd7, 0xd8, 0xe0, 0xf2, 0x0e, 0x35, 0x06, 0x50,
+		0x9c, 0x0d, 0x71, 0x11, 0x14, 0x71, 0x74, 0x19, 0x41, 0x00,
+	};
+
+	/* initialize USB */
+	rtl_write_reg(USBB, USB_SYSCTL, 0x09, 1);
+	rtl_write_reg(USBB, USB_EPA_MAXPKT, 0x0002, 2);
+	rtl_write_reg(USBB, USB_EPA_CTL, 0x1002, 2);
+
+	/* poweron demod */
+	rtl_write_reg(SYSB, DEMOD_CTL_1, 0x22, 1);
+	rtl_write_reg(SYSB, DEMOD_CTL, 0xe8, 1);
+
+	/* reset demod (bit 3, soft_rst) */
+	demod_write_reg(1, 0x01, 0x14, 1);
+	demod_write_reg(1, 0x01, 0x10, 1);
+
+	/* disable spectrum inversion and adjacent channel rejection */
+	demod_write_reg(1, 0x15, 0x00, 1);
+	demod_write_reg(1, 0x16, 0x0000, 2);
+
+	/* set IF-frequency to 0 Hz */
+	demod_write_reg(1, 0x19, 0x0000, 2);
+
+	/* set FIR coefficients */
+	for (i = 0; i < sizeof (fir_coeff); i++)
+		demod_write_reg(1, 0x1c + i, fir_coeff[i], 1);
+
+	/* TODO setting resampler test value, max value is 0xC99999,
+	 * value for DAB/FM is 0xE10000*/
+	set_resampler(1 << 24);
+
+	demod_write_reg(0, 0x19, 0x25, 1);
+
+	/* init FSM state-holding register */
+	demod_write_reg(1, 0x93, 0xf0, 1);
+
+	/* disable AGC (en_dagc, bit 0) */
+	demod_write_reg(1, 0x11, 0x00, 1);
+
+	/* disable PID filter (enable_PID = 0) */
+	demod_write_reg(0, 0x61, 0x60, 1);
+
+	/* opt_adc_iq = 0, default ADC_I/ADC_Q datapath */
+	demod_write_reg(0, 0x06, 0x80, 1);
+
+	/* Enable Zero-IF mode (en_bbin bit), DC cancellation (en_dc_est),
+	 * IQ estimation/compensation (en_iq_comp, en_iq_est) */
+	demod_write_reg(1, 0xb1, 0x1b, 1);
+}
+
+void tuner_init(int frequency)
+{
+	set_i2c_repeater(1);
+
+	switch (tuner_type) {
+	case TUNER_E4000:
+		e4000_Initialize(1);
+		e4000_SetBandwidthHz(1, 80000);
+		e4000_SetRfFreqHz(1, frequency);
+		break;
+	case TUNER_FC0013:
+		FC0013_Open();
+		FC0013_SetFrequency(frequency/1000, 8);
+		break;
+	default:
+		printf("No valid tuner available!");
+		break;
+	}
+
+	printf("Tuned to %i Hz\n", frequency);
+	set_i2c_repeater(0);
+}
+
+void usage(void)
+{
+	printf("rtl-sdr, an I/Q recorder for RTL2832 based USB-sticks\n\n"
+		"Usage:\t-f frequency to tune to [Hz]\n"
+		"\toutput filename\n");
+	exit(1);
+}
+
+static void sighandler(int signum)
+{
+	do_exit = 1;
+}
+
+int main(int argc, char **argv)
+{
+	struct sigaction sigact;
+	int r, opt;
+	char *filename;
+	unsigned int frequency = 0;
+	uint8_t buffer[READLEN];
+	int n_read;
+	FILE *file;
+
+	while ((opt = getopt(argc, argv, "f:")) != -1) {
+		switch (opt) {
+		case 'f':
+			frequency = atoi(optarg);
+			break;
+		default:
+			usage();
+			break;
+		}
+	}
+
+	if (argc <= optind) {
+		usage();
+	} else {
+		filename = argv[optind];
+	}
+
+	r = libusb_init(NULL);
+	if (r < 0) {
+		fprintf(stderr, "Failed to initialize libusb\n");
+		exit(1);
+	}
+
+	r = find_device();
+	if (r < 0) {
+		fprintf(stderr, "Could not find/open device\n");
+		goto out;
+	}
+
+	r = libusb_claim_interface(devh, 0);
+	if (r < 0) {
+		fprintf(stderr, "usb_claim_interface error %d\n", r);
+		goto out;
+	}
+
+	sigact.sa_handler = sighandler;
+	sigemptyset(&sigact.sa_mask);
+	sigact.sa_flags = 0;
+	sigaction(SIGINT, &sigact, NULL);
+	sigaction(SIGTERM, &sigact, NULL);
+	sigaction(SIGQUIT, &sigact, NULL);
+
+	/* Initialize the RTL2832 */
+	rtl_init();
+
+	/* Initialize tuner & set frequency */
+	tuner_init(frequency);
+
+	file = fopen(filename, "wb");
+
+	if (!file) {
+		printf("Failed to open %s\n", filename);
+		goto out;
+	}
+
+	/* reset endpoint before we start reading */
+	rtl_write_reg(USBB, USB_EPA_CTL, 0x1002, 2);
+	rtl_write_reg(USBB, USB_EPA_CTL, 0x0000, 2);
+
+	printf("Reading samples...\n");
+	while (!do_exit) {
+		libusb_bulk_transfer(devh, 0x81, buffer, READLEN, &n_read, 3000);
+		fwrite(buffer, n_read, 1, file);
+
+		if (n_read < READLEN) {
+			printf("Short bulk read, samples lost, exiting!\n");
+			break;
+		}
+	}
+
+	fclose(file);
+	libusb_release_interface(devh, 0);
+
+out:
+	libusb_close(devh);
+	libusb_exit(NULL);
+	return r >= 0 ? r : -r;
+}
+