rtl_tcp: proper linux support

as well as some cosmetic fixes

Signed-off-by: Steve Markgraf <steve@steve-m.de>

1 files changed, 26 insertions, 17 deletions

diff --git a/src/rtl_tcp.c b/src/rtl_tcp.c
index 007482e..f1f05ca 100644
--- a/src/rtl_tcp.c
+++ b/src/rtl_tcp.c
@@ -77,13 +77,14 @@ void usage(void)
 	#ifdef _WIN32
 	printf("rtl-sdr, an I/Q recorder for RTL2832 based USB-sticks\n\n"
 		"Usage:\t rtl-sdr-win.exe [listen addr] [listen port] "
-		"[samplerate in kHz] [frequency in hz]\n");
+		"[samplerate in kHz] [frequency in hz] [device index]\n");
 	#else
 	printf("rtl-sdr, an I/Q recorder for RTL2832 based USB-sticks\n\n"
 		"Usage:\t -a listen address\n"
 		"\t[-p listen port (default: 1234)\n"
 		"\t -f frequency to tune to [Hz]\n"
 		"\t[-s samplerate in kHz (default: 2048 kHz)]\n"
+		"\t[-d device index (default: 0)]\n"
 		"\toutput filename\n");
 	#endif
 	exit(1);
@@ -200,7 +201,9 @@ static void *tcp_worker(void *arg)
 			while(bytesleft > 0) {
 				FD_ZERO(&writefds);
 				FD_SET(s, &writefds);
-				r = select(0+1, NULL, &writefds, NULL, &tv);
+				tv.tv_sec = 1;
+				tv.tv_usec = 0;
+				r = select(s+1, NULL, &writefds, NULL, &tv);
 				if(r) {
 					bytessent = send(s,  &curelem->data[index], bytesleft, 0);
 					if (bytessent == SOCKET_ERROR || do_exit) {
@@ -250,7 +253,9 @@ static void *command_worker(void *arg)
 		while(left >0) {
 			FD_ZERO(&readfds);
 			FD_SET(s, &readfds);
-			r = select(0+1, &readfds, NULL, NULL, &tv);
+			tv.tv_sec = 1;
+			tv.tv_usec = 0;
+			r = select(s+1, &readfds, NULL, NULL, &tv);
 			if(r) {
 				received = recv(s, (char*)&cmd+(sizeof(cmd)-left), left, 0);
 				if(received == SOCKET_ERROR || do_exit){
@@ -283,7 +288,7 @@ static void *command_worker(void *arg)
 int main(int argc, char **argv)
 {
 	int r, opt, i;
-	char* addr;
+	char* addr = "127.0.0.1";
 	int port = 1234;
 	uint32_t frequency = 0, samp_rate = 2048000;
 	struct sockaddr_in local, remote;
@@ -303,13 +308,13 @@ int main(int argc, char **argv)
 #endif
 #ifndef _WIN32
 	struct sigaction sigact;
-	while ((opt = getopt(argc, argv, "a:p:f:s:")) != -1) {
+	while ((opt = getopt(argc, argv, "a:p:f:s:d:")) != -1) {
 		switch (opt) {
 		case 'f':
 			frequency = atoi(optarg);
 			break;
 		case 's':
-			samp_rate = atoi(optarg);
+			samp_rate = atoi(optarg)*1000;
 			break;
 		case 'a':
 			addr = optarg;
@@ -317,13 +322,16 @@ int main(int argc, char **argv)
 		case 'p':
 			port = atoi(optarg);
 			break;
+		case 'd':
+			dev_index = atoi(optarg);
+			break;
 		default:
 			usage();
 			break;
 		}
 	}
 
-	if (argc <= optind)
+	if (argc < optind)
 		usage();
 
 #else
@@ -407,7 +415,7 @@ int main(int argc, char **argv)
 	ioctlsocket(listensocket, FIONBIO, &blockmode);
 	#else
 	r = fcntl(listensocket, F_GETFL, 0);
-	fcntl(listensocket, F_SETFL, r | O_NONBLOCK);
+	r = fcntl(listensocket, F_SETFL, r | O_NONBLOCK);
 	#endif
 
 	while(1) {
@@ -417,7 +425,9 @@ int main(int argc, char **argv)
 		while(1) {
 			FD_ZERO(&readfds);
 			FD_SET(listensocket, &readfds);
-			r = select(0+1, &readfds, NULL, NULL, &tv);
+			tv.tv_sec = 1;
+			tv.tv_usec = 0;
+			r = select(listensocket+1, &readfds, NULL, NULL, &tv);
 			if(do_exit) {
 				goto out;
 			} else if(r) {
@@ -430,7 +440,7 @@ int main(int argc, char **argv)
 		setsockopt(s, SOL_SOCKET, SO_LINGER, (char *)&ling, sizeof(ling));
 
 		printf("client accepted!\n");
-		
+
 		pthread_attr_init(&attr);
 		pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
 		r = pthread_create(&tcp_worker_thread, &attr, tcp_worker, NULL);
@@ -440,22 +450,21 @@ int main(int argc, char **argv)
 		rtlsdr_wait_async(dev, rtlsdr_callback, (void *)0);
 
 		closesocket(s);
-		if(!dead[0]){
-		pthread_join(tcp_worker_thread, &status);
-		}
-		if(!dead[1]){
-		pthread_join(command_thread, &status);
-		}
+		if(!dead[0])
+			pthread_join(tcp_worker_thread, &status);
+
+		if(!dead[1])
+			pthread_join(command_thread, &status);
 
 		printf("all threads dead..\n");
 		curelem = ll_buffers;
 		ll_buffers = 0;
+
 		while(curelem != 0) {
 			prev = curelem;
 			curelem = curelem->next;
 			free(prev->data);
 			free(prev);
-			
 		}
 
 		do_exit = 0;