Mercurial > libavformat.hg
annotate udp.c @ 4021:6390b29b59f2 libavformat
Allow the UDP socket buffer size to be adjusted using a
'buffer_size' URL option.
Patch by Jindrich Makovicka, makovick at gmail dot com
author | andoma |
---|---|
date | Wed, 12 Nov 2008 21:40:53 +0000 |
parents | 8329c0373f69 |
children | 459dd8d5ad86 |
rev | line source |
---|---|
0 | 1 /* |
2 * UDP prototype streaming system | |
3 * Copyright (c) 2000, 2001, 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:
885
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 20 */ |
3231 | 21 |
22 /** | |
23 * @file udp.c | |
24 * UDP protocol | |
25 */ | |
26 | |
3776 | 27 #define _BSD_SOURCE /* Needed for using struct ip_mreq with recent glibc */ |
0 | 28 #include "avformat.h" |
29 #include <unistd.h> | |
1754 | 30 #include "network.h" |
2773 | 31 #include "os_support.h" |
0 | 32 |
834 | 33 #ifndef IPV6_ADD_MEMBERSHIP |
34 #define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP | |
35 #define IPV6_DROP_MEMBERSHIP IPV6_LEAVE_GROUP | |
36 #endif | |
3221 | 37 #ifndef IN_MULTICAST |
38 #define IN_MULTICAST(a) ((((uint32_t)(a)) & 0xf0000000) == 0xe0000000) | |
39 #endif | |
40 #ifndef IN6_IS_ADDR_MULTICAST | |
41 #define IN6_IS_ADDR_MULTICAST(a) (((uint8_t *) (a))[0] == 0xff) | |
42 #endif | |
834 | 43 |
0 | 44 typedef struct { |
45 int udp_fd; | |
46 int ttl; | |
4021
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
47 int buffer_size; |
0 | 48 int is_multicast; |
49 int local_port; | |
1428
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
50 int reuse_socket; |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
51 #ifndef CONFIG_IPV6 |
0 | 52 struct sockaddr_in dest_addr; |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
53 #else |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
54 struct sockaddr_storage dest_addr; |
2738
1f5c5c223764
Remove some ifdefs by using the "dest_addr_len" field in both the IPv4-only
lucabe
parents:
2689
diff
changeset
|
55 #endif |
3287
8570df039c75
Fix type of dest_addr_len to respect return value of udp_set_url.
cehoyos
parents:
3231
diff
changeset
|
56 int dest_addr_len; |
0 | 57 } UDPContext; |
58 | |
59 #define UDP_TX_BUF_SIZE 32768 | |
2391 | 60 #define UDP_MAX_PKT_SIZE 65536 |
0 | 61 |
2740
e23eaab1a894
Give better names to multicast functions (they are not IPv6-only)
lucabe
parents:
2739
diff
changeset
|
62 static int udp_set_multicast_ttl(int sockfd, int mcastTTL, struct sockaddr *addr) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
63 #ifdef IP_MULTICAST_TTL |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
64 if (addr->sa_family == AF_INET) { |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
65 if (setsockopt(sockfd, IPPROTO_IP, IP_MULTICAST_TTL, &mcastTTL, sizeof(mcastTTL)) < 0) { |
2768 | 66 av_log(NULL, AV_LOG_ERROR, "setsockopt(IP_MULTICAST_TTL): %s\n", strerror(errno)); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
67 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
68 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
69 } |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
70 #endif |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
71 #ifdef CONFIG_IPV6 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
72 if (addr->sa_family == AF_INET6) { |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
73 if (setsockopt(sockfd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &mcastTTL, sizeof(mcastTTL)) < 0) { |
2768 | 74 av_log(NULL, AV_LOG_ERROR, "setsockopt(IPV6_MULTICAST_HOPS): %s\n", strerror(errno)); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
75 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
76 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
77 } |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
78 #endif |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
79 return 0; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
80 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
81 |
2740
e23eaab1a894
Give better names to multicast functions (they are not IPv6-only)
lucabe
parents:
2739
diff
changeset
|
82 static int udp_join_multicast_group(int sockfd, struct sockaddr *addr) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
83 #ifdef IP_ADD_MEMBERSHIP |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
84 if (addr->sa_family == AF_INET) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
85 struct ip_mreq mreq; |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
86 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
87 mreq.imr_multiaddr.s_addr = ((struct sockaddr_in *)addr)->sin_addr.s_addr; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
88 mreq.imr_interface.s_addr= INADDR_ANY; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
89 if (setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const void *)&mreq, sizeof(mreq)) < 0) { |
2768 | 90 av_log(NULL, AV_LOG_ERROR, "setsockopt(IP_ADD_MEMBERSHIP): %s\n", strerror(errno)); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
91 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
92 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
93 } |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
94 #endif |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
95 #ifdef CONFIG_IPV6 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
96 if (addr->sa_family == AF_INET6) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
97 struct ipv6_mreq mreq6; |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
98 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
99 memcpy(&mreq6.ipv6mr_multiaddr, &(((struct sockaddr_in6 *)addr)->sin6_addr), sizeof(struct in6_addr)); |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
100 mreq6.ipv6mr_interface= 0; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
101 if (setsockopt(sockfd, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, &mreq6, sizeof(mreq6)) < 0) { |
2768 | 102 av_log(NULL, AV_LOG_ERROR, "setsockopt(IPV6_ADD_MEMBERSHIP): %s\n", strerror(errno)); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
103 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
104 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
105 } |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
106 #endif |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
107 return 0; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
108 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
109 |
2740
e23eaab1a894
Give better names to multicast functions (they are not IPv6-only)
lucabe
parents:
2739
diff
changeset
|
110 static int udp_leave_multicast_group(int sockfd, struct sockaddr *addr) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
111 #ifdef IP_DROP_MEMBERSHIP |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
112 if (addr->sa_family == AF_INET) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
113 struct ip_mreq mreq; |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
114 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
115 mreq.imr_multiaddr.s_addr = ((struct sockaddr_in *)addr)->sin_addr.s_addr; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
116 mreq.imr_interface.s_addr= INADDR_ANY; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
117 if (setsockopt(sockfd, IPPROTO_IP, IP_DROP_MEMBERSHIP, (const void *)&mreq, sizeof(mreq)) < 0) { |
2768 | 118 av_log(NULL, AV_LOG_ERROR, "setsockopt(IP_DROP_MEMBERSHIP): %s\n", strerror(errno)); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
119 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
120 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
121 } |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
122 #endif |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
123 #ifdef CONFIG_IPV6 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
124 if (addr->sa_family == AF_INET6) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
125 struct ipv6_mreq mreq6; |
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
126 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
127 memcpy(&mreq6.ipv6mr_multiaddr, &(((struct sockaddr_in6 *)addr)->sin6_addr), sizeof(struct in6_addr)); |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
128 mreq6.ipv6mr_interface= 0; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
129 if (setsockopt(sockfd, IPPROTO_IPV6, IPV6_DROP_MEMBERSHIP, &mreq6, sizeof(mreq6)) < 0) { |
2768 | 130 av_log(NULL, AV_LOG_ERROR, "setsockopt(IPV6_DROP_MEMBERSHIP): %s\n", strerror(errno)); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
131 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
132 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
133 } |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
134 #endif |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
135 return 0; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
136 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
137 |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
138 #ifdef CONFIG_IPV6 |
1124
d3aff2c607f9
Add const to (mostly) char* and make some functions static, which aren't used
diego
parents:
896
diff
changeset
|
139 static struct addrinfo* udp_ipv6_resolve_host(const char *hostname, int port, int type, int family, int flags) { |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
140 struct addrinfo hints, *res = 0; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
141 int error; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
142 char sport[16]; |
2687
f4d24b10d33d
Resolve hosts and bind sockets even when the local_port is not set (0)
lucabe
parents:
2391
diff
changeset
|
143 const char *node = 0, *service = "0"; |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
144 |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
145 if (port > 0) { |
644 | 146 snprintf(sport, sizeof(sport), "%d", port); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
147 service = sport; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
148 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
149 if ((hostname) && (hostname[0] != '\0') && (hostname[0] != '?')) { |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
150 node = hostname; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
151 } |
2688 | 152 memset(&hints, 0, sizeof(hints)); |
153 hints.ai_socktype = type; | |
154 hints.ai_family = family; | |
155 hints.ai_flags = flags; | |
156 if ((error = getaddrinfo(node, service, &hints, &res))) { | |
157 av_log(NULL, AV_LOG_ERROR, "udp_ipv6_resolve_host: %s\n", gai_strerror(error)); | |
158 } | |
159 | |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
160 return res; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
161 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
162 |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
163 static int udp_set_url(struct sockaddr_storage *addr, const char *hostname, int port) { |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
164 struct addrinfo *res0; |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
165 int addr_len; |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
166 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
167 res0 = udp_ipv6_resolve_host(hostname, port, SOCK_DGRAM, AF_UNSPEC, 0); |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2162
diff
changeset
|
168 if (res0 == 0) return AVERROR(EIO); |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
169 memcpy(addr, res0->ai_addr, res0->ai_addrlen); |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
170 addr_len = res0->ai_addrlen; |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
171 freeaddrinfo(res0); |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
172 |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
173 return addr_len; |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
174 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
175 |
3221 | 176 static int is_multicast_address(struct sockaddr_storage *addr) |
177 { | |
178 if (addr->ss_family == AF_INET) { | |
179 return IN_MULTICAST(ntohl(((struct sockaddr_in *)addr)->sin_addr.s_addr)); | |
180 } | |
181 if (addr->ss_family == AF_INET6) { | |
182 return IN6_IS_ADDR_MULTICAST(&((struct sockaddr_in6 *)addr)->sin6_addr); | |
183 } | |
184 | |
185 return 0; | |
186 } | |
187 | |
2752 | 188 static int udp_socket_create(UDPContext *s, struct sockaddr_storage *addr, int *addr_len) |
189 { | |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
190 int udp_fd = -1; |
825
c8d4a65729c5
udp ipv6 localhost resolving patch by ("Hans Zandbelt": Hans Zandbelt, telin nl)
michael
parents:
815
diff
changeset
|
191 struct addrinfo *res0 = NULL, *res = NULL; |
2689
7d25b8de708d
Take the target address family in account when determining the family of
lucabe
parents:
2688
diff
changeset
|
192 int family = AF_UNSPEC; |
885 | 193 |
2689
7d25b8de708d
Take the target address family in account when determining the family of
lucabe
parents:
2688
diff
changeset
|
194 if (((struct sockaddr *) &s->dest_addr)->sa_family) |
7d25b8de708d
Take the target address family in account when determining the family of
lucabe
parents:
2688
diff
changeset
|
195 family = ((struct sockaddr *) &s->dest_addr)->sa_family; |
7d25b8de708d
Take the target address family in account when determining the family of
lucabe
parents:
2688
diff
changeset
|
196 res0 = udp_ipv6_resolve_host(0, s->local_port, SOCK_DGRAM, family, AI_PASSIVE); |
2688 | 197 if (res0 == 0) |
198 goto fail; | |
199 for (res = res0; res; res=res->ai_next) { | |
200 udp_fd = socket(res->ai_family, SOCK_DGRAM, 0); | |
201 if (udp_fd > 0) break; | |
2768 | 202 av_log(NULL, AV_LOG_ERROR, "socket: %s\n", strerror(errno)); |
2688 | 203 } |
825
c8d4a65729c5
udp ipv6 localhost resolving patch by ("Hans Zandbelt": Hans Zandbelt, telin nl)
michael
parents:
815
diff
changeset
|
204 |
c8d4a65729c5
udp ipv6 localhost resolving patch by ("Hans Zandbelt": Hans Zandbelt, telin nl)
michael
parents:
815
diff
changeset
|
205 if (udp_fd < 0) |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
206 goto fail; |
885 | 207 |
2752 | 208 memcpy(addr, res->ai_addr, res->ai_addrlen); |
209 *addr_len = res->ai_addrlen; | |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
210 |
2752 | 211 freeaddrinfo(res0); |
885 | 212 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
213 return udp_fd; |
885 | 214 |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
215 fail: |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
216 if (udp_fd >= 0) |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
217 closesocket(udp_fd); |
683
095009fc2f35
kill warnings patch by (M«©ns Rullg«©rd <mru inprovide com>)
michael
parents:
644
diff
changeset
|
218 if(res0) |
095009fc2f35
kill warnings patch by (M«©ns Rullg«©rd <mru inprovide com>)
michael
parents:
644
diff
changeset
|
219 freeaddrinfo(res0); |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
220 return -1; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
221 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
222 |
2752 | 223 static int udp_port(struct sockaddr_storage *addr, int addr_len) |
224 { | |
3025
41d68d056417
Do not use GNU-specific (or BSD-specific or whatever they may be)
rfelker
parents:
2773
diff
changeset
|
225 char sbuf[sizeof(int)*3+1]; |
2752 | 226 |
3025
41d68d056417
Do not use GNU-specific (or BSD-specific or whatever they may be)
rfelker
parents:
2773
diff
changeset
|
227 if (getnameinfo((struct sockaddr *)addr, addr_len, NULL, 0, sbuf, sizeof(sbuf), NI_NUMERICSERV) != 0) { |
2768 | 228 av_log(NULL, AV_LOG_ERROR, "getnameinfo: %s\n", strerror(errno)); |
2752 | 229 return -1; |
230 } | |
231 | |
232 return strtol(sbuf, NULL, 10); | |
233 } | |
234 | |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
235 #else |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
236 |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
237 static int udp_set_url(struct sockaddr_in *addr, const char *hostname, int port) |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
238 { |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
239 /* set the destination address */ |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
240 if (resolve_host(&addr->sin_addr, hostname) < 0) |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
241 return AVERROR(EIO); |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
242 addr->sin_family = AF_INET; |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
243 addr->sin_port = htons(port); |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
244 |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
245 return sizeof(struct sockaddr_in); |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
246 } |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
247 |
3221 | 248 static int is_multicast_address(struct sockaddr_in *addr) |
249 { | |
250 return IN_MULTICAST(ntohl(addr->sin_addr.s_addr)); | |
251 } | |
252 | |
2752 | 253 static int udp_socket_create(UDPContext *s, struct sockaddr_in *addr, int *addr_len) |
254 { | |
255 int fd; | |
256 | |
257 fd = socket(AF_INET, SOCK_DGRAM, 0); | |
258 if (fd < 0) | |
259 return -1; | |
260 | |
261 addr->sin_family = AF_INET; | |
262 addr->sin_addr.s_addr = htonl (INADDR_ANY); | |
263 addr->sin_port = htons(s->local_port); | |
264 *addr_len = sizeof(struct sockaddr_in); | |
265 | |
266 return fd; | |
267 } | |
268 | |
269 static int udp_port(struct sockaddr_in *addr, int len) | |
270 { | |
271 return ntohs(addr->sin_port); | |
272 } | |
2162 | 273 #endif /* CONFIG_IPV6 */ |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
274 |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
275 |
0 | 276 /** |
277 * If no filename is given to av_open_input_file because you want to | |
278 * get the local port first, then you must call this function to set | |
279 * the remote server address. | |
280 * | |
281 * url syntax: udp://host:port[?option=val...] | |
3221 | 282 * option: 'ttl=n' : set the ttl value (for multicast only) |
0 | 283 * 'localport=n' : set the local port |
62 | 284 * 'pkt_size=n' : set max packet size |
1428
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
285 * 'reuse=1' : enable reusing the socket |
0 | 286 * |
287 * @param s1 media file context | |
288 * @param uri of the remote server | |
289 * @return zero if no error. | |
290 */ | |
291 int udp_set_remote_url(URLContext *h, const char *uri) | |
292 { | |
293 UDPContext *s = h->priv_data; | |
294 char hostname[256]; | |
295 int port; | |
885 | 296 |
511
056991ab9f10
HTTP Authentication Patch by (Petr Doubek <doubek at vision dot ee dot ethz dot ch>)
michael
parents:
482
diff
changeset
|
297 url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, NULL, 0, uri); |
0 | 298 |
299 /* set the destination address */ | |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
300 s->dest_addr_len = udp_set_url(&s->dest_addr, hostname, port); |
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
301 if (s->dest_addr_len < 0) { |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2162
diff
changeset
|
302 return AVERROR(EIO); |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
303 } |
3221 | 304 s->is_multicast = is_multicast_address(&s->dest_addr); |
2743
a847a9eda9b2
Simplify set_remote_url(), and remove some code duplication
lucabe
parents:
2742
diff
changeset
|
305 |
0 | 306 return 0; |
307 } | |
308 | |
309 /** | |
310 * Return the local port used by the UDP connexion | |
311 * @param s1 media file context | |
312 * @return the local port number | |
313 */ | |
314 int udp_get_local_port(URLContext *h) | |
315 { | |
316 UDPContext *s = h->priv_data; | |
317 return s->local_port; | |
318 } | |
319 | |
320 /** | |
321 * Return the udp file handle for select() usage to wait for several RTP | |
322 * streams at the same time. | |
323 * @param h media file context | |
324 */ | |
325 int udp_get_file_handle(URLContext *h) | |
326 { | |
327 UDPContext *s = h->priv_data; | |
328 return s->udp_fd; | |
329 } | |
330 | |
331 /* put it in UDP context */ | |
332 /* return non zero if error */ | |
333 static int udp_open(URLContext *h, const char *uri, int flags) | |
334 { | |
335 char hostname[1024]; | |
336 int port, udp_fd = -1, tmp; | |
337 UDPContext *s = NULL; | |
683
095009fc2f35
kill warnings patch by (M«©ns Rullg«©rd <mru inprovide com>)
michael
parents:
644
diff
changeset
|
338 int is_output; |
0 | 339 const char *p; |
340 char buf[256]; | |
683
095009fc2f35
kill warnings patch by (M«©ns Rullg«©rd <mru inprovide com>)
michael
parents:
644
diff
changeset
|
341 #ifndef CONFIG_IPV6 |
2751 | 342 struct sockaddr_in my_addr; |
2752 | 343 #else |
344 struct sockaddr_storage my_addr; | |
345 #endif | |
683
095009fc2f35
kill warnings patch by (M«©ns Rullg«©rd <mru inprovide com>)
michael
parents:
644
diff
changeset
|
346 int len; |
0 | 347 |
348 h->is_streamed = 1; | |
62 | 349 h->max_packet_size = 1472; |
0 | 350 |
351 is_output = (flags & URL_WRONLY); | |
885 | 352 |
3753
42ed3629601d
Fix memleak on some OSes in case network initialization fails. See
rbultje
parents:
3287
diff
changeset
|
353 if(!ff_network_init()) |
42ed3629601d
Fix memleak on some OSes in case network initialization fails. See
rbultje
parents:
3287
diff
changeset
|
354 return AVERROR(EIO); |
42ed3629601d
Fix memleak on some OSes in case network initialization fails. See
rbultje
parents:
3287
diff
changeset
|
355 |
2689
7d25b8de708d
Take the target address family in account when determining the family of
lucabe
parents:
2688
diff
changeset
|
356 s = av_mallocz(sizeof(UDPContext)); |
0 | 357 if (!s) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
358 return AVERROR(ENOMEM); |
0 | 359 |
360 h->priv_data = s; | |
361 s->ttl = 16; | |
4021
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
362 s->buffer_size = is_output ? UDP_TX_BUF_SIZE : UDP_MAX_PKT_SIZE; |
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
363 |
0 | 364 p = strchr(uri, '?'); |
365 if (p) { | |
1428
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
366 s->reuse_socket = find_info_tag(buf, sizeof(buf), "reuse", p); |
0 | 367 if (find_info_tag(buf, sizeof(buf), "ttl", p)) { |
368 s->ttl = strtol(buf, NULL, 10); | |
369 } | |
370 if (find_info_tag(buf, sizeof(buf), "localport", p)) { | |
371 s->local_port = strtol(buf, NULL, 10); | |
372 } | |
62 | 373 if (find_info_tag(buf, sizeof(buf), "pkt_size", p)) { |
374 h->max_packet_size = strtol(buf, NULL, 10); | |
375 } | |
4021
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
376 if (find_info_tag(buf, sizeof(buf), "buffer_size", p)) { |
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
377 s->buffer_size = strtol(buf, NULL, 10); |
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
378 } |
0 | 379 } |
380 | |
381 /* fill the dest addr */ | |
511
056991ab9f10
HTTP Authentication Patch by (Petr Doubek <doubek at vision dot ee dot ethz dot ch>)
michael
parents:
482
diff
changeset
|
382 url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, NULL, 0, uri); |
885 | 383 |
0 | 384 /* XXX: fix url_split */ |
385 if (hostname[0] == '\0' || hostname[0] == '?') { | |
386 /* only accepts null hostname if input */ | |
3221 | 387 if (flags & URL_WRONLY) |
0 | 388 goto fail; |
389 } else { | |
390 udp_set_remote_url(h, uri); | |
391 } | |
392 | |
2750
b3a11a0966af
Use the same code to set local_port in the IPv4-only case and in the
lucabe
parents:
2744
diff
changeset
|
393 if (s->is_multicast && !(h->flags & URL_WRONLY)) |
b3a11a0966af
Use the same code to set local_port in the IPv4-only case and in the
lucabe
parents:
2744
diff
changeset
|
394 s->local_port = port; |
2752 | 395 udp_fd = udp_socket_create(s, &my_addr, &len); |
0 | 396 if (udp_fd < 0) |
397 goto fail; | |
398 | |
1428
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
399 if (s->reuse_socket) |
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
400 if (setsockopt (udp_fd, SOL_SOCKET, SO_REUSEADDR, &(s->reuse_socket), sizeof(s->reuse_socket)) != 0) |
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
401 goto fail; |
7316227e64eb
Make it possible to reuse UDP socket (optional, disabled by default)
gpoirier
parents:
1358
diff
changeset
|
402 |
0 | 403 /* the bind is needed to give a port to the socket now */ |
2752 | 404 if (bind(udp_fd,(struct sockaddr *)&my_addr, len) < 0) |
0 | 405 goto fail; |
406 | |
2751 | 407 len = sizeof(my_addr); |
408 getsockname(udp_fd, (struct sockaddr *)&my_addr, &len); | |
2752 | 409 s->local_port = udp_port(&my_addr, len); |
410 | |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
411 if (s->is_multicast) { |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
412 if (h->flags & URL_WRONLY) { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
413 /* output */ |
2740
e23eaab1a894
Give better names to multicast functions (they are not IPv6-only)
lucabe
parents:
2739
diff
changeset
|
414 if (udp_set_multicast_ttl(udp_fd, s->ttl, (struct sockaddr *)&s->dest_addr) < 0) |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
415 goto fail; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
416 } else { |
2739
091af9f47edf
Avoid to duplicate the multicast code between the IPv4-only and
lucabe
parents:
2738
diff
changeset
|
417 /* input */ |
2740
e23eaab1a894
Give better names to multicast functions (they are not IPv6-only)
lucabe
parents:
2739
diff
changeset
|
418 if (udp_join_multicast_group(udp_fd, (struct sockaddr *)&s->dest_addr) < 0) |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
419 goto fail; |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
420 } |
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
421 } |
0 | 422 |
423 if (is_output) { | |
424 /* limit the tx buf size to limit latency */ | |
4021
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
425 tmp = s->buffer_size; |
0 | 426 if (setsockopt(udp_fd, SOL_SOCKET, SO_SNDBUF, &tmp, sizeof(tmp)) < 0) { |
2768 | 427 av_log(NULL, AV_LOG_ERROR, "setsockopt(SO_SNDBUF): %s\n", strerror(errno)); |
0 | 428 goto fail; |
429 } | |
2391 | 430 } else { |
431 /* set udp recv buffer size to the largest possible udp packet size to | |
432 * avoid losing data on OSes that set this too low by default. */ | |
4021
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
433 tmp = s->buffer_size; |
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
434 if (setsockopt(udp_fd, SOL_SOCKET, SO_RCVBUF, &tmp, sizeof(tmp)) < 0) { |
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
435 av_log(NULL, AV_LOG_WARNING, "setsockopt(SO_RECVBUF): %s\n", strerror(errno)); |
6390b29b59f2
Allow the UDP socket buffer size to be adjusted using a
andoma
parents:
3776
diff
changeset
|
436 } |
0 | 437 } |
438 | |
439 s->udp_fd = udp_fd; | |
440 return 0; | |
441 fail: | |
442 if (udp_fd >= 0) | |
443 closesocket(udp_fd); | |
444 av_free(s); | |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2162
diff
changeset
|
445 return AVERROR(EIO); |
0 | 446 } |
447 | |
65 | 448 static int udp_read(URLContext *h, uint8_t *buf, int size) |
0 | 449 { |
450 UDPContext *s = h->priv_data; | |
1332 | 451 int len; |
0 | 452 |
453 for(;;) { | |
2742
c51f5ad5d131
Use recv() instead of recvfrom() (removes some other differences between
lucabe
parents:
2741
diff
changeset
|
454 len = recv(s->udp_fd, buf, size, 0); |
0 | 455 if (len < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1810
diff
changeset
|
456 if (ff_neterrno() != FF_NETERROR(EAGAIN) && |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1810
diff
changeset
|
457 ff_neterrno() != FF_NETERROR(EINTR)) |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2162
diff
changeset
|
458 return AVERROR(EIO); |
0 | 459 } else { |
460 break; | |
461 } | |
462 } | |
463 return len; | |
464 } | |
465 | |
65 | 466 static int udp_write(URLContext *h, uint8_t *buf, int size) |
0 | 467 { |
468 UDPContext *s = h->priv_data; | |
469 int ret; | |
470 | |
471 for(;;) { | |
885 | 472 ret = sendto (s->udp_fd, buf, size, 0, |
0 | 473 (struct sockaddr *) &s->dest_addr, |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
474 s->dest_addr_len); |
0 | 475 if (ret < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1810
diff
changeset
|
476 if (ff_neterrno() != FF_NETERROR(EINTR) && |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
1810
diff
changeset
|
477 ff_neterrno() != FF_NETERROR(EAGAIN)) |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2162
diff
changeset
|
478 return AVERROR(EIO); |
0 | 479 } else { |
480 break; | |
481 } | |
482 } | |
483 return size; | |
484 } | |
485 | |
486 static int udp_close(URLContext *h) | |
487 { | |
488 UDPContext *s = h->priv_data; | |
489 | |
579
117ece7c24a6
IPv6 support patch by ("Hans Zandbelt" <Hans.Zandbelt <at> telin {dot} nl>)
michael
parents:
511
diff
changeset
|
490 if (s->is_multicast && !(h->flags & URL_WRONLY)) |
2740
e23eaab1a894
Give better names to multicast functions (they are not IPv6-only)
lucabe
parents:
2739
diff
changeset
|
491 udp_leave_multicast_group(s->udp_fd, (struct sockaddr *)&s->dest_addr); |
0 | 492 closesocket(s->udp_fd); |
2351
b9a881c0967e
Add initialization and cleanup functions for Winsock
ramiro
parents:
2274
diff
changeset
|
493 ff_network_close(); |
0 | 494 av_free(s); |
495 return 0; | |
496 } | |
497 | |
498 URLProtocol udp_protocol = { | |
499 "udp", | |
500 udp_open, | |
501 udp_read, | |
502 udp_write, | |
503 NULL, /* seek */ | |
504 udp_close, | |
505 }; |