Mercurial > libavformat.hg
annotate rtpproto.c @ 3228:697a11d7e3cf libavformat
Expose max_packet_size from the rtp protocol
author | lu_zero |
---|---|
date | Tue, 15 Apr 2008 19:27:39 +0000 |
parents | e17b25e8a34e |
children | 510fab17efc3 |
rev | line source |
---|---|
0 | 1 /* |
2 * RTP network protocol | |
3 * Copyright (c) 2002 Fabrice Bellard. | |
4 * | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1332
diff
changeset
|
5 * This file is part of FFmpeg. |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1332
diff
changeset
|
6 * |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1332
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
0 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1332
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
0 | 11 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1332
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
0 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1332
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
896
edbe5c3717f9
Update licensing information: The FSF changed postal address.
diego
parents:
887
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 20 */ |
21 #include "avformat.h" | |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
22 #include "avstring.h" |
0 | 23 |
24 #include <unistd.h> | |
25 #include <stdarg.h> | |
1754 | 26 #include "network.h" |
2782
419e121ce80a
Add #include "os_support.h" to restore OS/2 support.
diego
parents:
2274
diff
changeset
|
27 #include "os_support.h" |
0 | 28 #include <fcntl.h> |
29 | |
30 #define RTP_TX_BUF_SIZE (64 * 1024) | |
31 #define RTP_RX_BUF_SIZE (128 * 1024) | |
32 | |
33 typedef struct RTPContext { | |
34 URLContext *rtp_hd, *rtcp_hd; | |
35 int rtp_fd, rtcp_fd; | |
36 } RTPContext; | |
37 | |
38 /** | |
39 * If no filename is given to av_open_input_file because you want to | |
40 * get the local port first, then you must call this function to set | |
41 * the remote server address. | |
42 * | |
43 * @param s1 media file context | |
44 * @param uri of the remote server | |
45 * @return zero if no error. | |
46 */ | |
47 int rtp_set_remote_url(URLContext *h, const char *uri) | |
48 { | |
49 RTPContext *s = h->priv_data; | |
50 char hostname[256]; | |
51 int port; | |
52 | |
53 char buf[1024]; | |
54 char path[1024]; | |
885 | 55 |
56 url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, | |
0 | 57 path, sizeof(path), uri); |
58 | |
59 snprintf(buf, sizeof(buf), "udp://%s:%d%s", hostname, port, path); | |
60 udp_set_remote_url(s->rtp_hd, buf); | |
61 | |
62 snprintf(buf, sizeof(buf), "udp://%s:%d%s", hostname, port + 1, path); | |
63 udp_set_remote_url(s->rtcp_hd, buf); | |
64 return 0; | |
65 } | |
66 | |
67 | |
68 /* add option to url of the form: | |
69 "http://host:port/path?option1=val1&option2=val2... */ | |
64 | 70 static void url_add_option(char *buf, int buf_size, const char *fmt, ...) |
0 | 71 { |
72 char buf1[1024]; | |
73 va_list ap; | |
74 | |
75 va_start(ap, fmt); | |
76 if (strchr(buf, '?')) | |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
77 av_strlcat(buf, "&", buf_size); |
0 | 78 else |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
79 av_strlcat(buf, "?", buf_size); |
0 | 80 vsnprintf(buf1, sizeof(buf1), fmt, ap); |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
81 av_strlcat(buf, buf1, buf_size); |
0 | 82 va_end(ap); |
83 } | |
84 | |
64 | 85 static void build_udp_url(char *buf, int buf_size, |
887 | 86 const char *hostname, int port, |
3228 | 87 int local_port, int ttl, |
88 int max_packet_size) | |
0 | 89 { |
90 snprintf(buf, buf_size, "udp://%s:%d", hostname, port); | |
91 if (local_port >= 0) | |
92 url_add_option(buf, buf_size, "localport=%d", local_port); | |
93 if (ttl >= 0) | |
94 url_add_option(buf, buf_size, "ttl=%d", ttl); | |
3228 | 95 if (max_packet_size >=0) |
96 url_add_option(buf, buf_size, "pkt_size=%d", max_packet_size); | |
0 | 97 } |
98 | |
99 /* | |
100 * url syntax: rtp://host:port[?option=val...] | |
3221 | 101 * option: 'ttl=n' : set the ttl value (for multicast only) |
0 | 102 * 'localport=n' : set the local port to n |
3228 | 103 * 'pkt_size=n' : set max packet size |
0 | 104 * |
105 */ | |
106 static int rtp_open(URLContext *h, const char *uri, int flags) | |
107 { | |
108 RTPContext *s; | |
3228 | 109 int port, is_output, ttl, local_port, max_packet_size; |
0 | 110 char hostname[256]; |
111 char buf[1024]; | |
112 char path[1024]; | |
113 const char *p; | |
885 | 114 |
0 | 115 is_output = (flags & URL_WRONLY); |
116 | |
117 s = av_mallocz(sizeof(RTPContext)); | |
118 if (!s) | |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
119 return AVERROR(ENOMEM); |
0 | 120 h->priv_data = s; |
885 | 121 |
122 url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, | |
0 | 123 path, sizeof(path), uri); |
124 /* extract parameters */ | |
125 ttl = -1; | |
126 local_port = -1; | |
3228 | 127 max_packet_size = -1; |
128 | |
0 | 129 p = strchr(uri, '?'); |
130 if (p) { | |
131 if (find_info_tag(buf, sizeof(buf), "ttl", p)) { | |
132 ttl = strtol(buf, NULL, 10); | |
133 } | |
134 if (find_info_tag(buf, sizeof(buf), "localport", p)) { | |
135 local_port = strtol(buf, NULL, 10); | |
136 } | |
3228 | 137 if (find_info_tag(buf, sizeof(buf), "pkt_size", p)) { |
138 max_packet_size = strtol(buf, NULL, 10); | |
139 } | |
0 | 140 } |
141 | |
142 build_udp_url(buf, sizeof(buf), | |
3228 | 143 hostname, port, local_port, ttl, max_packet_size); |
0 | 144 if (url_open(&s->rtp_hd, buf, flags) < 0) |
145 goto fail; | |
146 local_port = udp_get_local_port(s->rtp_hd); | |
2079 | 147 /* XXX: need to open another connection if the port is not even */ |
0 | 148 |
149 /* well, should suppress localport in path */ | |
885 | 150 |
0 | 151 build_udp_url(buf, sizeof(buf), |
3228 | 152 hostname, port + 1, local_port + 1, ttl, max_packet_size); |
0 | 153 if (url_open(&s->rtcp_hd, buf, flags) < 0) |
154 goto fail; | |
885 | 155 |
0 | 156 /* just to ease handle access. XXX: need to suppress direct handle |
157 access */ | |
158 s->rtp_fd = udp_get_file_handle(s->rtp_hd); | |
159 s->rtcp_fd = udp_get_file_handle(s->rtcp_hd); | |
160 | |
885 | 161 h->max_packet_size = url_get_max_packet_size(s->rtp_hd); |
0 | 162 h->is_streamed = 1; |
163 return 0; | |
164 | |
165 fail: | |
166 if (s->rtp_hd) | |
167 url_close(s->rtp_hd); | |
168 if (s->rtcp_hd) | |
169 url_close(s->rtcp_hd); | |
170 av_free(s); | |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
171 return AVERROR(EIO); |
0 | 172 } |
173 | |
65 | 174 static int rtp_read(URLContext *h, uint8_t *buf, int size) |
0 | 175 { |
176 RTPContext *s = h->priv_data; | |
177 struct sockaddr_in from; | |
1332 | 178 socklen_t from_len; |
179 int len, fd_max, n; | |
0 | 180 fd_set rfds; |
181 #if 0 | |
182 for(;;) { | |
183 from_len = sizeof(from); | |
184 len = recvfrom (s->rtp_fd, buf, size, 0, | |
185 (struct sockaddr *)&from, &from_len); | |
186 if (len < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
187 if (ff_neterrno() == FF_NETERROR(EAGAIN) || |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
188 ff_neterrno() == FF_NETERROR(EINTR)) |
0 | 189 continue; |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
190 return AVERROR(EIO); |
0 | 191 } |
192 break; | |
193 } | |
194 #else | |
195 for(;;) { | |
196 /* build fdset to listen to RTP and RTCP packets */ | |
197 FD_ZERO(&rfds); | |
198 fd_max = s->rtp_fd; | |
199 FD_SET(s->rtp_fd, &rfds); | |
200 if (s->rtcp_fd > fd_max) | |
201 fd_max = s->rtcp_fd; | |
202 FD_SET(s->rtcp_fd, &rfds); | |
203 n = select(fd_max + 1, &rfds, NULL, NULL, NULL); | |
204 if (n > 0) { | |
205 /* first try RTCP */ | |
206 if (FD_ISSET(s->rtcp_fd, &rfds)) { | |
207 from_len = sizeof(from); | |
208 len = recvfrom (s->rtcp_fd, buf, size, 0, | |
209 (struct sockaddr *)&from, &from_len); | |
210 if (len < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
211 if (ff_neterrno() == FF_NETERROR(EAGAIN) || |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
212 ff_neterrno() == FF_NETERROR(EINTR)) |
0 | 213 continue; |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
214 return AVERROR(EIO); |
0 | 215 } |
216 break; | |
217 } | |
218 /* then RTP */ | |
219 if (FD_ISSET(s->rtp_fd, &rfds)) { | |
220 from_len = sizeof(from); | |
221 len = recvfrom (s->rtp_fd, buf, size, 0, | |
222 (struct sockaddr *)&from, &from_len); | |
223 if (len < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
224 if (ff_neterrno() == FF_NETERROR(EAGAIN) || |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
225 ff_neterrno() == FF_NETERROR(EINTR)) |
0 | 226 continue; |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
227 return AVERROR(EIO); |
0 | 228 } |
229 break; | |
230 } | |
231 } | |
232 } | |
233 #endif | |
234 return len; | |
235 } | |
236 | |
65 | 237 static int rtp_write(URLContext *h, uint8_t *buf, int size) |
0 | 238 { |
239 RTPContext *s = h->priv_data; | |
240 int ret; | |
241 URLContext *hd; | |
885 | 242 |
0 | 243 if (buf[1] >= 200 && buf[1] <= 204) { |
244 /* RTCP payload type */ | |
245 hd = s->rtcp_hd; | |
246 } else { | |
247 /* RTP payload type */ | |
248 hd = s->rtp_hd; | |
249 } | |
250 | |
251 ret = url_write(hd, buf, size); | |
252 #if 0 | |
253 { | |
254 struct timespec ts; | |
255 ts.tv_sec = 0; | |
256 ts.tv_nsec = 10 * 1000000; | |
257 nanosleep(&ts, NULL); | |
258 } | |
259 #endif | |
260 return ret; | |
261 } | |
262 | |
263 static int rtp_close(URLContext *h) | |
264 { | |
265 RTPContext *s = h->priv_data; | |
266 | |
267 url_close(s->rtp_hd); | |
268 url_close(s->rtcp_hd); | |
269 av_free(s); | |
270 return 0; | |
271 } | |
272 | |
273 /** | |
2079 | 274 * Return the local port used by the RTP connection |
0 | 275 * @param s1 media file context |
276 * @return the local port number | |
277 */ | |
278 int rtp_get_local_port(URLContext *h) | |
279 { | |
280 RTPContext *s = h->priv_data; | |
281 return udp_get_local_port(s->rtp_hd); | |
282 } | |
283 | |
284 /** | |
285 * Return the rtp and rtcp file handles for select() usage to wait for several RTP | |
286 * streams at the same time. | |
287 * @param h media file context | |
288 */ | |
289 void rtp_get_file_handles(URLContext *h, int *prtp_fd, int *prtcp_fd) | |
290 { | |
291 RTPContext *s = h->priv_data; | |
292 | |
293 *prtp_fd = s->rtp_fd; | |
294 *prtcp_fd = s->rtcp_fd; | |
295 } | |
296 | |
297 URLProtocol rtp_protocol = { | |
298 "rtp", | |
299 rtp_open, | |
300 rtp_read, | |
301 rtp_write, | |
302 NULL, /* seek */ | |
303 rtp_close, | |
304 }; |