Mercurial > libavformat.hg
annotate rtpproto.c @ 5725:334f223fc455 libavformat
Include rtpdec.h, it contains prototypes for the following functions:
rtp_set_remote_url(), rtp_get_local_port(), rtp_get_file_handles()
author | cehoyos |
---|---|
date | Sun, 28 Feb 2010 03:59:15 +0000 |
parents | b34d9614b887 |
children | 0f9c0db923e5 |
rev | line source |
---|---|
0 | 1 /* |
2 * RTP network protocol | |
4251
77e0c7511d41
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
4206
diff
changeset
|
3 * Copyright (c) 2002 Fabrice Bellard |
0 | 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 */ |
3229 | 21 |
22 /** | |
4331
49c1d3b27727
Use full internal pathname in doxygen @file directives.
diego
parents:
4251
diff
changeset
|
23 * @file libavformat/rtpproto.c |
3229 | 24 * RTP protocol |
25 */ | |
26 | |
3286 | 27 #include "libavutil/avstring.h" |
0 | 28 #include "avformat.h" |
5725
334f223fc455
Include rtpdec.h, it contains prototypes for the following functions:
cehoyos
parents:
4640
diff
changeset
|
29 #include "rtpdec.h" |
0 | 30 |
31 #include <unistd.h> | |
32 #include <stdarg.h> | |
1754 | 33 #include "network.h" |
2782
419e121ce80a
Add #include "os_support.h" to restore OS/2 support.
diego
parents:
2274
diff
changeset
|
34 #include "os_support.h" |
0 | 35 #include <fcntl.h> |
4206 | 36 #if HAVE_SYS_SELECT_H |
3712
30d4d95e068f
Add needed include, make it compile without -D_BSD_SOURCE.
michael
parents:
3286
diff
changeset
|
37 #include <sys/select.h> |
3720
1968303c7566
Surround '#include <sys/select>' by HAVE_SYS_SELECT_H.
diego
parents:
3712
diff
changeset
|
38 #endif |
0 | 39 |
40 #define RTP_TX_BUF_SIZE (64 * 1024) | |
41 #define RTP_RX_BUF_SIZE (128 * 1024) | |
42 | |
43 typedef struct RTPContext { | |
44 URLContext *rtp_hd, *rtcp_hd; | |
45 int rtp_fd, rtcp_fd; | |
46 } RTPContext; | |
47 | |
48 /** | |
49 * If no filename is given to av_open_input_file because you want to | |
50 * get the local port first, then you must call this function to set | |
51 * the remote server address. | |
52 * | |
53 * @param s1 media file context | |
54 * @param uri of the remote server | |
55 * @return zero if no error. | |
56 */ | |
3229 | 57 |
0 | 58 int rtp_set_remote_url(URLContext *h, const char *uri) |
59 { | |
60 RTPContext *s = h->priv_data; | |
61 char hostname[256]; | |
62 int port; | |
63 | |
64 char buf[1024]; | |
65 char path[1024]; | |
885 | 66 |
67 url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, | |
0 | 68 path, sizeof(path), uri); |
69 | |
70 snprintf(buf, sizeof(buf), "udp://%s:%d%s", hostname, port, path); | |
71 udp_set_remote_url(s->rtp_hd, buf); | |
72 | |
73 snprintf(buf, sizeof(buf), "udp://%s:%d%s", hostname, port + 1, path); | |
74 udp_set_remote_url(s->rtcp_hd, buf); | |
75 return 0; | |
76 } | |
77 | |
78 | |
3229 | 79 /** |
80 * add option to url of the form: | |
81 * "http://host:port/path?option1=val1&option2=val2... | |
82 */ | |
83 | |
64 | 84 static void url_add_option(char *buf, int buf_size, const char *fmt, ...) |
0 | 85 { |
86 char buf1[1024]; | |
87 va_list ap; | |
88 | |
89 va_start(ap, fmt); | |
90 if (strchr(buf, '?')) | |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
91 av_strlcat(buf, "&", buf_size); |
0 | 92 else |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
93 av_strlcat(buf, "?", buf_size); |
0 | 94 vsnprintf(buf1, sizeof(buf1), fmt, ap); |
2193
5ce5fad0dfac
replace the uses of old string functions that Reimar missed
mru
parents:
2079
diff
changeset
|
95 av_strlcat(buf, buf1, buf_size); |
0 | 96 va_end(ap); |
97 } | |
98 | |
64 | 99 static void build_udp_url(char *buf, int buf_size, |
887 | 100 const char *hostname, int port, |
3228 | 101 int local_port, int ttl, |
102 int max_packet_size) | |
0 | 103 { |
104 snprintf(buf, buf_size, "udp://%s:%d", hostname, port); | |
105 if (local_port >= 0) | |
106 url_add_option(buf, buf_size, "localport=%d", local_port); | |
107 if (ttl >= 0) | |
108 url_add_option(buf, buf_size, "ttl=%d", ttl); | |
3228 | 109 if (max_packet_size >=0) |
110 url_add_option(buf, buf_size, "pkt_size=%d", max_packet_size); | |
0 | 111 } |
112 | |
3229 | 113 /** |
0 | 114 * url syntax: rtp://host:port[?option=val...] |
3221 | 115 * option: 'ttl=n' : set the ttl value (for multicast only) |
0 | 116 * 'localport=n' : set the local port to n |
3228 | 117 * 'pkt_size=n' : set max packet size |
0 | 118 * |
119 */ | |
3229 | 120 |
0 | 121 static int rtp_open(URLContext *h, const char *uri, int flags) |
122 { | |
123 RTPContext *s; | |
3228 | 124 int port, is_output, ttl, local_port, max_packet_size; |
0 | 125 char hostname[256]; |
126 char buf[1024]; | |
127 char path[1024]; | |
128 const char *p; | |
885 | 129 |
0 | 130 is_output = (flags & URL_WRONLY); |
131 | |
132 s = av_mallocz(sizeof(RTPContext)); | |
133 if (!s) | |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
134 return AVERROR(ENOMEM); |
0 | 135 h->priv_data = s; |
885 | 136 |
137 url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, | |
0 | 138 path, sizeof(path), uri); |
139 /* extract parameters */ | |
140 ttl = -1; | |
141 local_port = -1; | |
3228 | 142 max_packet_size = -1; |
143 | |
0 | 144 p = strchr(uri, '?'); |
145 if (p) { | |
146 if (find_info_tag(buf, sizeof(buf), "ttl", p)) { | |
147 ttl = strtol(buf, NULL, 10); | |
148 } | |
149 if (find_info_tag(buf, sizeof(buf), "localport", p)) { | |
150 local_port = strtol(buf, NULL, 10); | |
151 } | |
3228 | 152 if (find_info_tag(buf, sizeof(buf), "pkt_size", p)) { |
153 max_packet_size = strtol(buf, NULL, 10); | |
154 } | |
0 | 155 } |
156 | |
157 build_udp_url(buf, sizeof(buf), | |
3228 | 158 hostname, port, local_port, ttl, max_packet_size); |
0 | 159 if (url_open(&s->rtp_hd, buf, flags) < 0) |
160 goto fail; | |
161 local_port = udp_get_local_port(s->rtp_hd); | |
2079 | 162 /* XXX: need to open another connection if the port is not even */ |
0 | 163 |
164 /* well, should suppress localport in path */ | |
885 | 165 |
0 | 166 build_udp_url(buf, sizeof(buf), |
3228 | 167 hostname, port + 1, local_port + 1, ttl, max_packet_size); |
0 | 168 if (url_open(&s->rtcp_hd, buf, flags) < 0) |
169 goto fail; | |
885 | 170 |
0 | 171 /* just to ease handle access. XXX: need to suppress direct handle |
172 access */ | |
4640
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
173 s->rtp_fd = url_get_file_handle(s->rtp_hd); |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
174 s->rtcp_fd = url_get_file_handle(s->rtcp_hd); |
0 | 175 |
885 | 176 h->max_packet_size = url_get_max_packet_size(s->rtp_hd); |
0 | 177 h->is_streamed = 1; |
178 return 0; | |
179 | |
180 fail: | |
181 if (s->rtp_hd) | |
182 url_close(s->rtp_hd); | |
183 if (s->rtcp_hd) | |
184 url_close(s->rtcp_hd); | |
185 av_free(s); | |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
186 return AVERROR(EIO); |
0 | 187 } |
188 | |
65 | 189 static int rtp_read(URLContext *h, uint8_t *buf, int size) |
0 | 190 { |
191 RTPContext *s = h->priv_data; | |
192 struct sockaddr_in from; | |
1332 | 193 socklen_t from_len; |
194 int len, fd_max, n; | |
0 | 195 fd_set rfds; |
196 #if 0 | |
197 for(;;) { | |
198 from_len = sizeof(from); | |
199 len = recvfrom (s->rtp_fd, buf, size, 0, | |
200 (struct sockaddr *)&from, &from_len); | |
201 if (len < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
202 if (ff_neterrno() == FF_NETERROR(EAGAIN) || |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
203 ff_neterrno() == FF_NETERROR(EINTR)) |
0 | 204 continue; |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
205 return AVERROR(EIO); |
0 | 206 } |
207 break; | |
208 } | |
209 #else | |
210 for(;;) { | |
211 /* build fdset to listen to RTP and RTCP packets */ | |
212 FD_ZERO(&rfds); | |
213 fd_max = s->rtp_fd; | |
214 FD_SET(s->rtp_fd, &rfds); | |
215 if (s->rtcp_fd > fd_max) | |
216 fd_max = s->rtcp_fd; | |
217 FD_SET(s->rtcp_fd, &rfds); | |
218 n = select(fd_max + 1, &rfds, NULL, NULL, NULL); | |
219 if (n > 0) { | |
220 /* first try RTCP */ | |
221 if (FD_ISSET(s->rtcp_fd, &rfds)) { | |
222 from_len = sizeof(from); | |
223 len = recvfrom (s->rtcp_fd, buf, size, 0, | |
224 (struct sockaddr *)&from, &from_len); | |
225 if (len < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
226 if (ff_neterrno() == FF_NETERROR(EAGAIN) || |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
227 ff_neterrno() == FF_NETERROR(EINTR)) |
0 | 228 continue; |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
229 return AVERROR(EIO); |
0 | 230 } |
231 break; | |
232 } | |
233 /* then RTP */ | |
234 if (FD_ISSET(s->rtp_fd, &rfds)) { | |
235 from_len = sizeof(from); | |
236 len = recvfrom (s->rtp_fd, buf, size, 0, | |
237 (struct sockaddr *)&from, &from_len); | |
238 if (len < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
239 if (ff_neterrno() == FF_NETERROR(EAGAIN) || |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1930
diff
changeset
|
240 ff_neterrno() == FF_NETERROR(EINTR)) |
0 | 241 continue; |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2200
diff
changeset
|
242 return AVERROR(EIO); |
0 | 243 } |
244 break; | |
245 } | |
246 } | |
247 } | |
248 #endif | |
249 return len; | |
250 } | |
251 | |
65 | 252 static int rtp_write(URLContext *h, uint8_t *buf, int size) |
0 | 253 { |
254 RTPContext *s = h->priv_data; | |
255 int ret; | |
256 URLContext *hd; | |
885 | 257 |
0 | 258 if (buf[1] >= 200 && buf[1] <= 204) { |
259 /* RTCP payload type */ | |
260 hd = s->rtcp_hd; | |
261 } else { | |
262 /* RTP payload type */ | |
263 hd = s->rtp_hd; | |
264 } | |
265 | |
266 ret = url_write(hd, buf, size); | |
267 #if 0 | |
268 { | |
269 struct timespec ts; | |
270 ts.tv_sec = 0; | |
271 ts.tv_nsec = 10 * 1000000; | |
272 nanosleep(&ts, NULL); | |
273 } | |
274 #endif | |
275 return ret; | |
276 } | |
277 | |
278 static int rtp_close(URLContext *h) | |
279 { | |
280 RTPContext *s = h->priv_data; | |
281 | |
282 url_close(s->rtp_hd); | |
283 url_close(s->rtcp_hd); | |
284 av_free(s); | |
285 return 0; | |
286 } | |
287 | |
288 /** | |
2079 | 289 * Return the local port used by the RTP connection |
0 | 290 * @param s1 media file context |
291 * @return the local port number | |
292 */ | |
3229 | 293 |
0 | 294 int rtp_get_local_port(URLContext *h) |
295 { | |
296 RTPContext *s = h->priv_data; | |
297 return udp_get_local_port(s->rtp_hd); | |
298 } | |
299 | |
4640
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
300 #if (LIBAVFORMAT_VERSION_MAJOR <= 52) |
0 | 301 /** |
3229 | 302 * Return the rtp and rtcp file handles for select() usage to wait for |
303 * several RTP streams at the same time. | |
0 | 304 * @param h media file context |
305 */ | |
3229 | 306 |
0 | 307 void rtp_get_file_handles(URLContext *h, int *prtp_fd, int *prtcp_fd) |
308 { | |
309 RTPContext *s = h->priv_data; | |
310 | |
311 *prtp_fd = s->rtp_fd; | |
312 *prtcp_fd = s->rtcp_fd; | |
313 } | |
4640
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
314 #endif |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
315 |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
316 static int rtp_get_file_handle(URLContext *h) |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
317 { |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
318 RTPContext *s = h->priv_data; |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
319 return s->rtp_fd; |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
320 } |
0 | 321 |
322 URLProtocol rtp_protocol = { | |
323 "rtp", | |
324 rtp_open, | |
325 rtp_read, | |
326 rtp_write, | |
327 NULL, /* seek */ | |
328 rtp_close, | |
4640
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4331
diff
changeset
|
329 .url_get_file_handle = rtp_get_file_handle, |
0 | 330 }; |