fix in6_addr problems

[record-dvb] / dump-stream.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 <fcntl.h>
#include <unistd.h>
#include <strings.h>
#include <netdb.h>

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

#define CHUNKSIZE 1500

int main(int argc, char **argv)
{
	char *url;
	int bytes, written, i;
	int in;
	char buffer[CHUNKSIZE];

	if (argc == 2) {
		url = argv[1];
	} else {
		fprintf(stderr,"Syntax: %s URL|SAPServiceName\n", argv[0]);
		exit(EXIT_FAILURE);
	}

	if (!is_url(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)) {
		if ((in = open_http(url)) < 0) {
			fprintf(stderr,"Can't open url %s!\n",url);
			exit(EXIT_FAILURE);
		}
	} else if (is_mcast(url)) {
		if ((in = open_mcast(url)) < 0) {
			fprintf(stderr,"Can't open url %s!\n",url);
			exit(EXIT_FAILURE);
		}
	} else {
		printf("URL '%s' not supported!\n", url);
		exit(EXIT_FAILURE);
	}

	while(1) {
		if ((bytes = recv(in, buffer, CHUNKSIZE, 0)) < 1) {
			perror("recv");
			exit(EXIT_FAILURE);
		}

		written = 0;
		do {
			if ((i = write(STDOUT_FILENO, buffer, bytes-written)) < 0) {
				perror("write");
				exit(EXIT_FAILURE);
			}
			written += i;
		} while(written < bytes);
	}

	close(in);

	return EXIT_SUCCESS;
}