[record-dvb] / record-dvb.c

#include <features.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/time.h>
#include <time.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/select.h>
#include <fcntl.h>
#include <unistd.h>
#include <strings.h>

#include "http.h"
#include "mcast.h"
#include "sap.h"

#define CHUNKSIZE	3000
#define GTOD_INTERVAL	100
#define MAX_ERROR_SLEEP	60

void record(int(*open_fn)(char *), char *url, char *outfile, int duration)
{
	struct timeval start, curr;
	int bytes, recvd, written, count = 0;
	int error_sleep = 0;
	char buffer[CHUNKSIZE];
	int in, out;
	fd_set rfds;
	struct timeval tv;
	int retval;

	if ((in = (*open_fn)(url)) < 0) {
		fprintf(stderr,"Can't open url %s!\n",url);
		exit(EXIT_FAILURE);
	}

	if ((out = open(outfile, O_CREAT|O_TRUNC|O_WRONLY|O_LARGEFILE, 00644)) < 0) {
		perror("open");
		exit(EXIT_FAILURE);
	}

	printf("Recording from %s for %d seconds...\n", url, duration);

	gettimeofday(&start, NULL);
	curr = start;

	do {
		if (error_sleep) {
			sleep(error_sleep);
			printf("Reconnecting... ");
			if ((in = (*open_fn)(url)) < 0) {
				if (error_sleep < MAX_ERROR_SLEEP)
					error_sleep *= 2;

				if (error_sleep > MAX_ERROR_SLEEP)
					error_sleep = MAX_ERROR_SLEEP;

				printf("failed\n");
				continue;
			} else {
				printf("succeeded\n");
				error_sleep = 0;
			}
		}

		FD_ZERO(&rfds);
		FD_SET(in, &rfds);

		tv.tv_sec = 1;
		tv.tv_usec = 0;

		if ((retval = select(in + 1, &rfds, NULL, NULL, &tv)) == -1) {
			error_sleep = 1;
			continue;
		}

		if (!retval) {
			gettimeofday(&curr, NULL);
			continue;
		}

		if ((recvd = recv(in, buffer, CHUNKSIZE, 0)) < 1) {
			error_sleep = 1;
			continue;
		}
		written = 0;
		do {
			if ((bytes = write(out, buffer, recvd-written)) < 0) {
				perror("write");
				exit(EXIT_FAILURE);
			}
			written += bytes;
		} while(written < recvd);

		if (!(++count % GTOD_INTERVAL))
			gettimeofday(&curr, NULL);
	} while (curr.tv_sec < start.tv_sec+duration);

	close(out);
	close(in);
	shutdown(in, SHUT_RDWR);
}

int main(int argc, char **argv)
{
	int duration;
	char *url;
	char *outfile;
	int(*open_fn)(char *);

	if (argc == 4) {
		url = argv[1];
		duration = atoi(argv[2])*60;
		outfile = argv[3];
	} else {
		fprintf(stderr,"Syntax: %s URL duration_in_minutes outfile\n", argv[0]);
		exit(EXIT_FAILURE);
	}

	if (!is_http(url) && !is_mcast(url)) {
		char *service_url;
		if ((service_url = get_url_from_sap(url))) {
			printf("SAP says: '%s' -> %s\n", url, service_url);
			url = service_url;
		}
	}

	if (is_http(url)) {
		open_fn = &open_http;
	} else if (is_mcast(url)) {
		open_fn = &open_mcast;
	} else {
		printf("URL '%s' not supported!\n", url);
		exit(EXIT_FAILURE);
	}

	record(open_fn, url, outfile, duration);

	return EXIT_SUCCESS;
}
Commit	Line	Data
5c2e3e44	1	#include <features.h>
	2	#include <stdio.h>
	3	#include <stdlib.h>
	4	#include <sys/time.h>
	5	#include <time.h>
	6	#include <sys/socket.h>
	7	#include <sys/types.h>
	8	#include <sys/stat.h>
89ec024f	9	#include <sys/select.h>
5c2e3e44	10	#include <fcntl.h>
	11	#include <unistd.h>
	12	#include <strings.h>
	13
	14	#include "http.h"
	15	#include "mcast.h"
8447845f	16	#include "sap.h"
5c2e3e44	17
657d048d	18	#define CHUNKSIZE 3000
	19	#define GTOD_INTERVAL 100
	20	#define MAX_ERROR_SLEEP 60
5c2e3e44	21
	22	void record(int(open_fn)(char ), char url, char outfile, int duration)
	23	{
	24	struct timeval start, curr;
4710a9f8	25	int bytes, recvd, written, count = 0;
34d3d284	26	int error_sleep = 0;
5c2e3e44	27	char buffer[CHUNKSIZE];
5c2e3e44	28	int in, out;
89ec024f	29	fd_set rfds;
	30	struct timeval tv;
	31	int retval;
5c2e3e44	32
4710a9f8	33	if ((in = (*open_fn)(url)) < 0) {
4710a9f8	34	fprintf(stderr,"Can't open url %s!\n",url);
5c2e3e44	35	exit(EXIT_FAILURE);
	36	}
	37
4710a9f8	38	if ((out = open(outfile, O_CREAT\|O_TRUNC\|O_WRONLY\|O_LARGEFILE, 00644)) < 0) {
4710a9f8	39	perror("open");
5c2e3e44	40	exit(EXIT_FAILURE);
	41	}
	42
	43	printf("Recording from %s for %d seconds...\n", url, duration);
	44
	45	gettimeofday(&start, NULL);
6ac47f90	46	curr = start;
5c2e3e44	47
5c2e3e44	48	do {
34d3d284	49	if (error_sleep) {
	50	sleep(error_sleep);
	51	printf("Reconnecting... ");
b973e875	52	if ((in = (*open_fn)(url)) < 0) {
657d048d	53	if (error_sleep < MAX_ERROR_SLEEP)
34d3d284	54	error_sleep *= 2;
34d3d284	55
657d048d	56	if (error_sleep > MAX_ERROR_SLEEP)
	57	error_sleep = MAX_ERROR_SLEEP;
	58
34d3d284	59	printf("failed\n");
b973e875	60	continue;
34d3d284	61	} else {
	62	printf("succeeded\n");
	63	error_sleep = 0;
b973e875	64	}
5c2e3e44	65	}
34d3d284	66
89ec024f	67	FD_ZERO(&rfds);
	68	FD_SET(in, &rfds);
	69
	70	tv.tv_sec = 1;
	71	tv.tv_usec = 0;
	72
	73	if ((retval = select(in + 1, &rfds, NULL, NULL, &tv)) == -1) {
	74	error_sleep = 1;
	75	continue;
	76	}
	77
	78	if (!retval) {
	79	gettimeofday(&curr, NULL);
	80	continue;
	81	}
	82
34d3d284	83	if ((recvd = recv(in, buffer, CHUNKSIZE, 0)) < 1) {
	84	error_sleep = 1;
	85	continue;
	86	}
5c2e3e44	87	written = 0;
5c2e3e44	88	do {
4710a9f8	89	if ((bytes = write(out, buffer, recvd-written)) < 0) {
5c2e3e44	90	perror("write");
	91	exit(EXIT_FAILURE);
	92	}
4710a9f8	93	written += bytes;
4710a9f8	94	} while(written < recvd);
6ac47f90	95
4710a9f8	96	if (!(++count % GTOD_INTERVAL))
6ac47f90	97	gettimeofday(&curr, NULL);
5c2e3e44	98	} while (curr.tv_sec < start.tv_sec+duration);
	99
	100	close(out);
	101	close(in);
	102	shutdown(in, SHUT_RDWR);
	103	}
	104
	105	int main(int argc, char **argv)
	106	{
	107	int duration;
	108	char *url;
	109	char *outfile;
	110	int(open_fn)(char );
	111
	112	if (argc == 4) {
	113	url = argv[1];
6a3dc096	114	duration = atoi(argv[2])*60;
5c2e3e44	115	outfile = argv[3];
5c2e3e44	116	} else {
460e6d22	117	fprintf(stderr,"Syntax: %s URL duration_in_minutes outfile\n", argv[0]);
5c2e3e44	118	exit(EXIT_FAILURE);
	119	}
	120
8447845f	121	if (!is_http(url) && !is_mcast(url)) {
	122	char *service_url;
	123	if ((service_url = get_url_from_sap(url))) {
	124	printf("SAP says: '%s' -> %s\n", url, service_url);
	125	url = service_url;
	126	}
	127	}
	128
5c2e3e44	129	if (is_http(url)) {
	130	open_fn = &open_http;
	131	} else if (is_mcast(url)) {
	132	open_fn = &open_mcast;
	133	} else {
8447845f	134	printf("URL '%s' not supported!\n", url);
5c2e3e44	135	exit(EXIT_FAILURE);
	136	}
	137
	138	record(open_fn, url, outfile, duration);
	139
	140	return EXIT_SUCCESS;
	141	}