add SAP/SDP support
[record-dvb] / record-dvb.c
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 }
Impressum, Datenschutz