diff options
author | Tudor.Suciu <tudor.suciu@gmail.com> | 2012-01-29 15:24:59 +0100 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2012-03-12 06:01:09 +0100 |
commit | 9abd47fe90a592822b326345726626a018641d04 (patch) | |
tree | 2193cf675649743b940b01a90c02accc01f11781 /libavformat/udp.c | |
parent | 3c9a86df0e20dd1c90a9b390a6555b22f87aa8cd (diff) | |
download | ffmpeg-9abd47fe90a592822b326345726626a018641d04.tar.gz |
udp: add multicast input interface selection
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavformat/udp.c')
-rw-r--r-- | libavformat/udp.c | 30 |
1 files changed, 28 insertions, 2 deletions
diff --git a/libavformat/udp.c b/libavformat/udp.c index cfc5ae51db..72ff7905d5 100644 --- a/libavformat/udp.c +++ b/libavformat/udp.c @@ -35,6 +35,16 @@ #include <unistd.h> #include "internal.h" #include "network.h" + + +#ifdef WIN32 +# define if_nametoindex( str ) atoi( str ) +#else +# include <unistd.h> +# include <net/if.h> +#endif + + #include "os_support.h" #include "url.h" @@ -100,11 +110,12 @@ static int udp_set_multicast_ttl(int sockfd, int mcastTTL, return 0; } -static int udp_join_multicast_group(int sockfd, struct sockaddr *addr) +static int udp_join_multicast_group(int sockfd, struct sockaddr *addr, int miface_nr) { #ifdef IP_ADD_MEMBERSHIP if (addr->sa_family == AF_INET) { struct ip_mreq mreq; + struct group_req gr; mreq.imr_multiaddr.s_addr = ((struct sockaddr_in *)addr)->sin_addr.s_addr; mreq.imr_interface.s_addr= INADDR_ANY; @@ -112,6 +123,16 @@ static int udp_join_multicast_group(int sockfd, struct sockaddr *addr) av_log(NULL, AV_LOG_ERROR, "setsockopt(IP_ADD_MEMBERSHIP): %s\n", strerror(errno)); return -1; } + +#ifdef MCAST_JOIN_GROUP + memset(&gr, 0, sizeof(struct group_req)); + gr.gr_interface = miface_nr; + memcpy(&gr.gr_group, addr, sizeof(struct sockaddr_in)); + if (setsockopt(sockfd, SOL_IP, MCAST_JOIN_GROUP, &gr, sizeof(struct group_req)) < 0) { + av_log(NULL, AV_LOG_ERROR, "setsockopt(MCAST_JOIN_GROUP): %s\n", strerror(errno)); + return -1; + } +#endif } #endif #if HAVE_STRUCT_IPV6_MREQ && defined(IPPROTO_IPV6) @@ -259,6 +280,7 @@ static int udp_port(struct sockaddr_storage *addr, int addr_len) * option: 'ttl=n' : set the ttl value (for multicast only) * 'localport=n' : set the local port * 'pkt_size=n' : set max packet size + * 'miface=if' : set multicast input interface * 'reuse=1' : enable reusing the socket * * @param h media file context @@ -401,6 +423,7 @@ static int udp_open(URLContext *h, const char *uri, int flags) struct sockaddr_storage my_addr; int len; int reuse_specified = 0; + int miface_nr = 0; h->is_streamed = 1; h->max_packet_size = 1472; @@ -443,6 +466,9 @@ static int udp_open(URLContext *h, const char *uri, int flags) if (av_find_info_tag(buf, sizeof(buf), "localaddr", p)) { av_strlcpy(localaddr, buf, sizeof(localaddr)); } + if (av_find_info_tag(buf, sizeof(buf), "miface", p)) { + miface_nr = if_nametoindex (buf); + } } /* fill the dest addr */ @@ -500,7 +526,7 @@ static int udp_open(URLContext *h, const char *uri, int flags) } if (h->flags & AVIO_FLAG_READ) { /* input */ - if (udp_join_multicast_group(udp_fd, (struct sockaddr *)&s->dest_addr) < 0) + if (udp_join_multicast_group(udp_fd, (struct sockaddr *)&s->dest_addr, miface_nr) < 0) goto fail; } } |