/*
 * osmo-fl2k, turns FL2000-based USB 3.0 to VGA adapters into
 * low cost DACs
 *
 * Copyright (C) 2016-2018 by Steve Markgraf <steve@steve-m.de>
 *
 * SPDX-License-Identifier: GPL-2.0+
 *
 * 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>

#ifndef _WIN32
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <netinet/tcp.h> /* for TCP_NODELAY */
#include <fcntl.h>
#define sleep_ms(ms)	usleep(ms*1000)
#else
#include <windows.h>
#include <winsock2.h>
#include "getopt/getopt.h"
#define sleep_ms(ms)	Sleep(ms)
#endif

#include "osmo-fl2k.h"

#ifdef _WIN32
#pragma comment(lib, "ws2_32.lib")

typedef int socklen_t;

#else
#define closesocket close
#define SOCKADDR struct sockaddr
#define SOCKET int
#define SOCKET_ERROR -1
#endif

static SOCKET s;
static fl2k_dev_t *dev = NULL;
static volatile int do_exit = 0;
static volatile int connected = 0;
static char *txbuf = NULL;
static fd_set readfds;
static SOCKET sock;

void usage(void)
{
	fprintf(stderr,
		"fl2k_tcp, a spectrum client for FL2K VGA dongles\n\n"
		"Usage:\t[-a server address]\n"
		"\t[-d device index (default: 0)]\n"
		"\t[-p port (default: 1234)]\n"
		"\t[-s samplerate in Hz (default: 100 MS/s)]\n"
		"\t[-b number of buffers (default: 4)]\n"
	);
	exit(1);
}

#ifdef _WIN32
BOOL WINAPI
sighandler(int signum)
{
	if (CTRL_C_EVENT == signum) {
		fprintf(stderr, "Signal caught, exiting!\n");
		fl2k_stop_tx(dev);
		do_exit = 1;
		return TRUE;
	}
	return FALSE;
}
#else
static void sighandler(int signum)
{
	fprintf(stderr, "Signal caught, exiting!\n");
	do_exit = 1;
	fl2k_stop_tx(dev);
}
#endif

void fl2k_callback(fl2k_data_info_t *data_info)
{
	int left = FL2K_BUF_LEN;
	int received;
	int r;
	struct timeval tv = { 1, 0 };

	if (data_info->device_error) {
		fprintf(stderr, "Device error, exiting.\n");
		do_exit = 1;
		return;
	}

	if (!connected)
		return;

	data_info->sampletype_signed = 1;
	data_info->r_buf = txbuf;

	while (!do_exit && (left > 0)) {
		FD_ZERO(&readfds);
		FD_SET(sock, &readfds);
		tv.tv_sec = 1;
		tv.tv_usec = 0;
		r = select(sock + 1, &readfds, NULL, NULL, &tv);

		if (r) {
			received = recv(sock, txbuf + (FL2K_BUF_LEN - left), left, 0);
			if (!received) {
				fprintf(stderr, "Connection was closed!\n");
				fl2k_stop_tx(dev);
				do_exit = 1;
			}

			left -= received;
		}
	}
}

int main(int argc, char **argv)
{
	int r, opt, i;
	char *addr = "127.0.0.1";
	int port = 1234;
	uint32_t samp_rate = 100000000;
	struct sockaddr_in local, remote;
	uint32_t buf_num = 0;
	int dev_index = 0;
	int dev_given = 0;
	int flag = 1;

#ifdef _WIN32
	WSADATA wsd;
	i = WSAStartup(MAKEWORD(2,2), &wsd);
#else
	struct sigaction sigact, sigign;
#endif

	while ((opt = getopt(argc, argv, "d:s:a:p:b:")) != -1) {
		switch (opt) {
		case 'd':
			dev_index = (uint32_t)atoi(optarg);
			dev_given = 1;
			break;
		case 's':
			samp_rate = (uint32_t)atof(optarg);
			break;
		case 'a':
			addr = optarg;
			break;
		case 'p':
			port = atoi(optarg);
			break;
		case 'b':
			buf_num = atoi(optarg);
			break;
		default:
			usage();
			break;
		}
	}

	if (argc < optind)
		usage();

	if (dev_index < 0) {
		exit(1);
	}

	txbuf = malloc(FL2K_BUF_LEN);

	if (!txbuf) {
		fprintf(stderr, "malloc error!\n");
		exit(1);
	}

	fl2k_open(&dev, (uint32_t)dev_index);
	if (NULL == dev) {
		fprintf(stderr, "Failed to open fl2k device #%d.\n", dev_index);
		exit(1);
	}

	r = fl2k_start_tx(dev, fl2k_callback, NULL, buf_num);

	/* Set the sample rate */
	r = fl2k_set_sample_rate(dev, samp_rate);
	if (r < 0)
		fprintf(stderr, "WARNING: Failed to set sample rate.\n");

#ifndef _WIN32
	sigact.sa_handler = sighandler;
	sigemptyset(&sigact.sa_mask);
	sigact.sa_flags = 0;
	sigign.sa_handler = SIG_IGN;
	sigaction(SIGINT, &sigact, NULL);
	sigaction(SIGTERM, &sigact, NULL);
	sigaction(SIGQUIT, &sigact, NULL);
	sigaction(SIGPIPE, &sigact, NULL);
#else
	SetConsoleCtrlHandler( (PHANDLER_ROUTINE) sighandler, TRUE );
#endif

	sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

	memset(&remote, 0, sizeof(remote));

	remote.sin_family = AF_INET;
	remote.sin_port = htons(port);
	remote.sin_addr.s_addr = inet_addr(addr);

	fprintf(stderr, "Connecting to %s:%d...\n", addr, port);
	while (connect(sock, (struct sockaddr *)&remote, sizeof(remote)) != 0) {
		sleep_ms(500);
		if (do_exit)
			goto out;
	}

	setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, (char *)&flag,sizeof(flag));
	fprintf(stderr, "Connected\n");
	connected = 1;

	while (!do_exit)
		sleep_ms(500);

out:
	fl2k_close(dev);
	free(txbuf);
	closesocket(s);
#ifdef _WIN32
	WSACleanup();
#endif

	return 0;
}