cvs.zerfleddert.de Git - record-dvb/blame_incremental

... / ...

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