]> git.sesse.net Git - vlc/blob - src/network/udp.c
Add SSM definition for Vista (closes #313)
[vlc] / src / network / udp.c
1 /*****************************************************************************
2  * udp.c:
3  *****************************************************************************
4  * Copyright (C) 2004-2006 the VideoLAN team
5  * Copyright © 2006 Rémi Denis-Courmont
6  *
7  * $Id$
8  *
9  * Authors: Laurent Aimar <fenrir@videolan.org>
10  *          Rémi Denis-Courmont <rem # videolan.org>
11  *
12  * This program is free software; you can redistribute it and/or modify
13  * it under the terms of the GNU General Public License as published by
14  * the Free Software Foundation; either version 2 of the License, or
15  * (at your option) any later version.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software
24  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25  *****************************************************************************/
26
27 /*****************************************************************************
28  * Preamble
29  *****************************************************************************/
30 #include <stdlib.h>
31 #include <vlc/vlc.h>
32
33 #include <errno.h>
34 #include <assert.h>
35
36 #ifdef HAVE_SYS_TIME_H
37 #    include <sys/time.h>
38 #endif
39
40 #include <vlc_network.h>
41
42 #ifdef WIN32
43 #   if defined(UNDER_CE)
44 #       undef IP_MULTICAST_TTL
45 #       define IP_MULTICAST_TTL 3
46 #       undef IP_ADD_MEMBERSHIP
47 #       define IP_ADD_MEMBERSHIP 5
48 #   endif
49 #   define EAFNOSUPPORT WSAEAFNOSUPPORT
50 #   define if_nametoindex( str ) atoi( str )
51 #else
52 #   include <unistd.h>
53 #   ifdef HAVE_NET_IF_H
54 #       include <net/if.h>
55 #   endif
56 #endif
57
58 #ifndef SOL_IP
59 # define SOL_IP IPPROTO_IP
60 #endif
61 #ifndef SOL_IPV6
62 # define SOL_IPV6 IPPROTO_IPV6
63 #endif
64 #ifndef IPPROTO_IPV6
65 # define IPPROTO_IPV6 41
66 #endif
67
68 extern int net_Socket( vlc_object_t *p_this, int i_family, int i_socktype,
69                        int i_protocol );
70
71
72 /*
73  * XXX: I am too lazy to put all these dual functions in “next generation”
74  * network plugins.
75  */
76
77 static int net_SetMcastHopLimit( vlc_object_t *p_this,
78                                  int fd, int family, int hlim )
79 {
80     int proto, cmd;
81
82     /* There is some confusion in the world whether IP_MULTICAST_TTL 
83      * takes a byte or an int as an argument.
84      * BSD seems to indicate byte so we are going with that and use
85      * int as a fallback to be safe */
86     switch( family )
87     {
88 #ifdef IP_MULTICAST_TTL
89         case AF_INET:
90             proto = SOL_IP;
91             cmd = IP_MULTICAST_TTL;
92             break;
93 #endif
94
95 #ifdef IPV6_MULTICAST_HOPS
96         case AF_INET6:
97             proto = SOL_IPV6;
98             cmd = IPV6_MULTICAST_HOPS;
99             break;
100 #endif
101
102         default:
103             msg_Warn( p_this, "%s", strerror( EAFNOSUPPORT ) );
104             return VLC_EGENERIC;
105     }
106
107     if( setsockopt( fd, proto, cmd, &hlim, sizeof( hlim ) ) < 0 )
108     {
109         /* BSD compatibility */
110         unsigned char buf;
111
112         buf = (unsigned char)(( hlim > 255 ) ? 255 : hlim);
113         if( setsockopt( fd, proto, cmd, &buf, sizeof( buf ) ) )
114             return VLC_EGENERIC;
115     }
116
117     return VLC_SUCCESS;
118 }
119
120
121 static int net_SetMcastOutIface (int fd, int family, int scope)
122 {
123     switch (family)
124     {
125 #ifdef IPV6_MULTICAST_IF
126         case AF_INET6:
127             return setsockopt (fd, SOL_IPV6, IPV6_MULTICAST_IF,
128                                &scope, sizeof (scope));
129 #endif
130
131 #ifdef __linux__
132         case AF_INET:
133         {
134             struct ip_mreqn req = { .imr_ifindex = scope };
135
136             return setsockopt (fd, SOL_IP, IP_MULTICAST_IF, &req,
137                                sizeof (req));
138         }
139 #endif
140     }
141
142     errno = EAFNOSUPPORT;
143     return -1;
144 }
145
146
147 static inline int net_SetMcastOutIPv4 (int fd, struct in_addr ipv4)
148 {
149 #ifdef IP_MULTICAST_IF
150     return setsockopt( fd, SOL_IP, IP_MULTICAST_IF, &ipv4, sizeof (ipv4));
151 #else
152     errno = EAFNOSUPPORT;
153     return -1;
154 #endif
155 }
156
157
158 static int net_SetMcastOut (vlc_object_t *p_this, int fd, int family,
159                             const char *iface, const char *addr)
160 {
161     if (iface != NULL)
162     {
163         int scope = if_nametoindex (iface);
164         if (scope == 0)
165         {
166             msg_Err (p_this, "invalid multicast interface: %s", iface);
167             return -1;
168         }
169
170         if (net_SetMcastOutIface (fd, family, scope) == 0)
171             return 0;
172
173         msg_Err (p_this, "%s: %s", iface, net_strerror (net_errno));
174     }
175
176     if (addr != NULL)
177     {
178         if (family == AF_INET)
179         {
180             struct in_addr ipv4;
181             if (inet_pton (AF_INET, addr, &ipv4) <= 0)
182             {
183                 msg_Err (p_this, "invalid IPv4 address for multicast: %s",
184                          addr);
185                 return -1;
186             }
187
188             if (net_SetMcastOutIPv4 (fd, ipv4) == 0)
189                 return 0;
190
191             msg_Err (p_this, "%s: %s", addr, net_strerror (net_errno));
192         }
193     }
194
195     return -1;
196 }
197
198
199 /**
200  * Old-style any-source multicast join.
201  * In use on Windows XP/2003 and older.
202  */
203 static int
204 net_IPv4Join (vlc_object_t *obj, int fd,
205               const struct sockaddr_in *src, const struct sockaddr_in *grp)
206 {
207 #ifdef IP_ADD_MEMBERSHIP
208     union
209     {
210         struct ip_mreq gr4;
211 # ifdef IP_ADD_SOURCE_MEMBERSHIP
212         struct ip_mreq_source gsr4;
213 # endif
214     } opt;
215     int cmd;
216     struct in_addr id = { .s_addr = INADDR_ANY };
217     socklen_t optlen;
218
219     /* Multicast interface IPv4 address */
220     char *iface = var_CreateGetString (obj, "miface-addr");
221     if (iface != NULL)
222     {
223         if ((*iface)
224          && (inet_pton (AF_INET, iface, &id) <= 0))
225         {
226             msg_Err (obj, "invalid multicast interface address %s", iface);
227             free (iface);
228             goto error;
229         }
230     }
231
232     memset (&opt, 0, sizeof (opt));
233     if (src != NULL)
234     {
235 # ifdef IP_ADD_SOURCE_MEMBERSHIP
236         cmd = IP_ADD_SOURCE_MEMBERSHIP;
237         opt.gsr4.imr_multiaddr = grp->sin_addr;
238         opt.gsr4.imr_sourceaddr = src->sin_addr;
239         opt.gsr4.imr_interface = id;
240         optlen = sizeof (opt.gsr4);
241 # else
242         errno = ENOSYS;
243         goto error;
244 # endif
245     }
246     else
247     {
248         cmd = IP_ADD_MEMBERSHIP;
249         opt.gr4.imr_multiaddr = grp->sin_addr;
250         opt.gr4.imr_interface = id;
251         optlen = sizeof (opt.gr4);
252     }
253
254     msg_Dbg (obj, "IP_ADD_%sMEMBERSHIP multicast request",
255              (src != NULL) ? "SOURCE_" : "");
256
257     if (setsockopt (fd, SOL_IP, cmd, &opt, optlen) == 0)
258         return 0;
259
260 error:
261 #endif
262
263     msg_Err (obj, "cannot join IPv4 multicast group (%s)",
264              net_strerror (net_errno));
265     return -1;
266 }
267
268
269 static int
270 net_IPv6Join (vlc_object_t *obj, int fd, const struct sockaddr_in6 *src)
271 {
272 #ifdef IPV6_JOIN_GROUP
273     struct ipv6_mreq gr6;
274     memset (&gr6, 0, sizeof (gr6));
275     gr6.ipv6mr_interface = src->sin6_scope_id;
276     memcpy (&gr6.ipv6mr_multiaddr, &src->sin6_addr, 16);
277
278     msg_Dbg (obj, "IPV6_JOIN_GROUP multicast request");
279
280     if (!setsockopt (fd, IPPROTO_IPV6, IPV6_JOIN_GROUP, &gr6, sizeof (gr6)))
281         return 0;
282 #else
283     errno = ENOSYS;
284 #endif
285
286     msg_Err (obj, "cannot join IPv6 any-source multicast group (%s)",
287              net_strerror (net_errno));
288     return -1;
289 }
290
291
292 #if defined (WIN32) && !defined (MCAST_JOIN_SOURCE_GROUP)
293 /*
294  * I hate manual definitions: Error-prone. Portability hell.
295  * Developers shall use UP-TO-DATE compilers. Full point.
296  * If you remove the warning, you remove the whole ifndef.
297  */
298 #  warning Your C headers are out-of-date. Please update.
299
300 #  define MCAST_JOIN_GROUP 41
301 struct group_req
302 {
303     ULONG gr_interface;
304     struct sockaddr_storage gr_group;
305 };
306
307 #  define MCAST_JOIN_SOURCE_GROUP 45 /* from <ws2ipdef.h> */
308 struct group_source_req
309 {
310     uint32_t gsr_interface;
311     struct sockaddr_storage gsr_group;
312     struct sockaddr_storage gsr_source;
313 };
314 #endif
315
316 /**
317  * IP-agnostic multicast join,
318  * with fallback to old APIs, and fallback from SSM to ASM.
319  */
320 static int
321 net_SourceSubscribe (vlc_object_t *obj, int fd,
322                      const struct sockaddr *src, socklen_t srclen,
323                      const struct sockaddr *grp, socklen_t grplen)
324 {
325     int level, iid = 0;
326
327     char *iface = var_CreateGetString (obj, "miface");
328     if (iface != NULL)
329     {
330         if (*iface)
331         {
332             iid = if_nametoindex (iface);
333             if (iid == 0)
334             {
335                 msg_Err (obj, "invalid multicast interface: %s", iface);
336                 free (iface);
337                 return -1;
338             }
339         }
340         free (iface);
341     }
342
343     switch (grp->sa_family)
344     {
345 #ifdef AF_INET6
346         case AF_INET6:
347             level = SOL_IPV6;
348             if (((const struct sockaddr_in6 *)grp)->sin6_scope_id)
349                 iid = ((const struct sockaddr_in6 *)grp)->sin6_scope_id;
350             break;
351 #endif
352
353         case AF_INET:
354             level = SOL_IP;
355             break;
356
357         default:
358             errno = EAFNOSUPPORT;
359             return -1;
360     }
361
362     if (src != NULL)
363         switch (src->sa_family)
364         {
365 #ifdef AF_INET6
366             case AF_INET6:
367                 if (memcmp (&((const struct sockaddr_in6 *)src)->sin6_addr,
368                             &in6addr_any, sizeof (in6addr_any)) == 0)
369                     src = NULL;
370             break;
371 #endif
372
373             case AF_INET:
374                 if (((const struct sockaddr_in *)src)->sin_addr.s_addr
375                      == INADDR_ANY)
376                     src = NULL;
377                 break;
378         }
379
380
381     /* Agnostic ASM/SSM multicast join */
382 #ifdef MCAST_JOIN_SOURCE_GROUP
383     union
384     {
385         struct group_req gr;
386         struct group_source_req gsr;
387     } opt;
388     socklen_t optlen;
389
390     memset (&opt, 0, sizeof (opt));
391
392     if (src != NULL)
393     {
394         if ((grplen > sizeof (opt.gsr.gsr_group))
395          || (srclen > sizeof (opt.gsr.gsr_source)))
396             return -1;
397
398         opt.gsr.gsr_interface = iid;
399         memcpy (&opt.gsr.gsr_source, src, srclen);
400         memcpy (&opt.gsr.gsr_group,  grp, grplen);
401         optlen = sizeof (opt.gsr);
402     }
403     else
404     {
405         if (grplen > sizeof (opt.gr.gr_group))
406             return -1;
407
408         opt.gr.gr_interface = iid;
409         memcpy (&opt.gr.gr_group, grp, grplen);
410         optlen = sizeof (opt.gr);
411     }
412
413     msg_Dbg (obj, "Multicast %sgroup join request", src ? "source " : "");
414
415     if (setsockopt (fd, level, 
416                     src ? MCAST_JOIN_SOURCE_GROUP : MCAST_JOIN_GROUP,
417                     (void *)&opt, optlen) == 0)
418         return 0;
419 #endif
420
421     /* Fallback to IPv-specific APIs */
422     if ((src != NULL) && (src->sa_family != grp->sa_family))
423         return -1;
424
425     switch (grp->sa_family)
426     {
427         case AF_INET:
428             if ((grplen < sizeof (struct sockaddr_in))
429              || ((src != NULL) && (srclen < sizeof (struct sockaddr_in))))
430                 return -1;
431
432             if (net_IPv4Join (obj, fd, (const struct sockaddr_in *)src,
433                               (const struct sockaddr_in *)grp) == 0)
434                 return 0;
435             break;
436
437 #ifdef AF_INET6
438         case AF_INET6:
439             if ((grplen < sizeof (struct sockaddr_in6))
440              || ((src != NULL) && (srclen < sizeof (struct sockaddr_in6))))
441                 return -1;
442
443             /* IPv6-specific SSM API does not exist. So if we're here
444              * it means IPv6 SSM is not supported on this OS and we
445              * directly fallback to ASM */
446
447             if (net_IPv6Join (obj, fd, (const struct sockaddr_in6 *)grp) == 0)
448                 return 0;
449             break;
450 #endif
451     }
452
453     msg_Err (obj, "Multicast group join error (%s)",
454              net_strerror (net_errno));
455
456     if (src != NULL)
457     {
458         msg_Warn (obj, "Trying ASM instead of SSM...");
459         return net_Subscribe (obj, fd, grp, grplen);
460     }
461
462     msg_Err (obj, "Multicast not supported");
463     return -1;
464 }
465
466
467 int net_Subscribe (vlc_object_t *obj, int fd,
468                    const struct sockaddr *addr, socklen_t addrlen)
469 {
470     return net_SourceSubscribe (obj, fd, NULL, 0, addr, addrlen);
471 }
472
473
474 int net_SetDSCP( int fd, uint8_t dscp )
475 {
476     struct sockaddr_storage addr;
477     if( getsockname( fd, (struct sockaddr *)&addr, &(socklen_t){ sizeof (addr) }) )
478         return -1;
479
480     int level, cmd;
481
482     switch( addr.ss_family )
483     {
484 #ifdef IPV6_TCLASS
485         case AF_INET6:
486             level = SOL_IPV6;
487             cmd = IPV6_TCLASS;
488             break;
489 #endif
490
491         case AF_INET:
492             level = SOL_IP;
493             cmd = IP_TOS;
494             break;
495
496         default:
497 #ifdef ENOPROTOOPT
498             errno = ENOPROTOOPT;
499 #endif
500             return -1;
501     }
502
503     return setsockopt( fd, level, cmd, &(int){ dscp }, sizeof (int));
504 }
505
506
507 /*****************************************************************************
508  * __net_ConnectDgram:
509  *****************************************************************************
510  * Open a datagram socket to send data to a defined destination, with an
511  * optional hop limit.
512  *****************************************************************************/
513 int __net_ConnectDgram( vlc_object_t *p_this, const char *psz_host, int i_port,
514                         int i_hlim, int proto )
515 {
516     struct addrinfo hints, *res, *ptr;
517     int             i_val, i_handle = -1;
518     vlc_bool_t      b_unreach = VLC_FALSE;
519
520     if( i_port == 0 )
521         i_port = 1234; /* historical VLC thing */
522
523     if( i_hlim < 1 )
524         i_hlim = var_CreateGetInteger( p_this, "ttl" );
525
526     memset( &hints, 0, sizeof( hints ) );
527     hints.ai_socktype = SOCK_DGRAM;
528
529     msg_Dbg( p_this, "net: connecting to %s port %d", psz_host, i_port );
530
531     i_val = vlc_getaddrinfo( p_this, psz_host, i_port, &hints, &res );
532     if( i_val )
533     {
534         msg_Err( p_this, "cannot resolve %s port %d : %s", psz_host, i_port,
535                  vlc_gai_strerror( i_val ) );
536         return -1;
537     }
538
539     for( ptr = res; ptr != NULL; ptr = ptr->ai_next )
540     {
541         char *str;
542         int fd = net_Socket (p_this, ptr->ai_family, ptr->ai_socktype,
543                              proto ?: ptr->ai_protocol);
544         if (fd == -1)
545             continue;
546
547 #if !defined( SYS_BEOS )
548         /* Increase the receive buffer size to 1/2MB (8Mb/s during 1/2s)
549         * to avoid packet loss caused by scheduling problems */
550         setsockopt (fd, SOL_SOCKET, SO_RCVBUF, &(int){ 0x80000 }, sizeof (int));
551         setsockopt (fd, SOL_SOCKET, SO_SNDBUF, &(int){ 0x80000 }, sizeof (int));
552
553         /* Allow broadcast sending */
554         setsockopt (fd, SOL_SOCKET, SO_BROADCAST, &(int){ 1 }, sizeof (int));
555 #endif
556
557         if( i_hlim > 0 )
558             net_SetMcastHopLimit( p_this, fd, ptr->ai_family, i_hlim );
559
560         str = var_CreateGetString (p_this, "miface");
561         if (str != NULL)
562         {
563             if (*str)
564                 net_SetMcastOut (p_this, fd, ptr->ai_family, str, NULL);
565             free (str);
566         }
567
568         str = var_CreateGetString (p_this, "miface-addr");
569         if (str != NULL)
570         {
571             if (*str)
572                 net_SetMcastOut (p_this, fd, ptr->ai_family, NULL, str);
573             free (str);
574         }
575
576         net_SetDSCP (fd, var_CreateGetInteger (p_this, "dscp"));
577
578         if( connect( fd, ptr->ai_addr, ptr->ai_addrlen ) == 0 )
579         {
580             /* success */
581             i_handle = fd;
582             break;
583         }
584
585 #if defined( WIN32 ) || defined( UNDER_CE )
586         if( WSAGetLastError () == WSAENETUNREACH )
587 #else
588         if( errno == ENETUNREACH )
589 #endif
590             b_unreach = VLC_TRUE;
591         else
592         {
593             msg_Warn( p_this, "%s port %d : %s", psz_host, i_port,
594                       strerror( errno ) );
595             net_Close( fd );
596             continue;
597         }
598     }
599
600     vlc_freeaddrinfo( res );
601
602     if( i_handle == -1 )
603     {
604         if( b_unreach )
605             msg_Err( p_this, "Host %s port %d is unreachable", psz_host,
606                      i_port );
607         return -1;
608     }
609
610     return i_handle;
611 }
612
613
614 /*****************************************************************************
615  * __net_OpenDgram:
616  *****************************************************************************
617  * OpenDgram a datagram socket and return a handle
618  *****************************************************************************/
619 int __net_OpenDgram( vlc_object_t *obj, const char *psz_bind, int i_bind,
620                      const char *psz_server, int i_server,
621                      int family, int protocol )
622 {
623     struct addrinfo hints, *loc, *rem;
624     int val;
625
626     if( !*psz_server )
627         return net_ListenSingle (obj, psz_bind, i_bind,
628                                  family, SOCK_DGRAM, protocol);
629
630     msg_Dbg( obj, "net: connecting to [%s]:%d from [%s]:%d",
631              psz_server, i_server, psz_bind, i_bind );
632
633     memset (&hints, 0, sizeof (hints));
634     hints.ai_family = family;
635     hints.ai_socktype = SOCK_DGRAM;
636     hints.ai_flags = AI_PASSIVE;
637
638     val = vlc_getaddrinfo (obj, psz_server, i_server, &hints, &rem);
639     if (val)
640     {
641         msg_Err (obj, "cannot resolve %s port %d : %s", psz_bind, i_bind,
642                  vlc_gai_strerror (val));
643         return -1;
644     }
645
646     val = vlc_getaddrinfo (obj, psz_bind, i_bind, &hints, &loc);
647     if (val)
648     {
649         msg_Err (obj, "cannot resolve %s port %d : %s", psz_bind, i_bind,
650                  vlc_gai_strerror (val));
651         vlc_freeaddrinfo (rem);
652         return -1;
653     }
654
655     for (struct addrinfo *ptr = loc; ptr != NULL; ptr = ptr->ai_next)
656     {
657         int fd = net_Socket (obj, ptr->ai_family, ptr->ai_socktype,
658                              protocol ?: ptr->ai_protocol);
659         if (fd == -1)
660             continue; // usually, address family not supported
661
662 #ifdef SO_REUSEPORT
663         setsockopt (fd, SOL_SOCKET, SO_REUSEPORT, &(int){ 1 }, sizeof (int));
664 #endif
665
666 #ifdef SO_RCVBUF
667         /* Increase the receive buffer size to 1/2MB (8Mb/s during 1/2s)
668          * to avoid packet loss caused in case of scheduling hiccups */
669         setsockopt (fd, SOL_SOCKET, SO_RCVBUF,
670                     (void *)&(int){ 0x80000 }, sizeof (int));
671         setsockopt (fd, SOL_SOCKET, SO_SNDBUF,
672                     (void *)&(int){ 0x80000 }, sizeof (int));
673 #endif
674
675 #if defined (WIN32) || defined (UNDER_CE)
676         if (net_SockAddrIsMulticast (ptr->ai_addr, ptr->ai_addrlen)
677          && (sizeof (struct sockaddr_storage) >= ptr->ai_addrlen))
678         {
679             // This works for IPv4 too - don't worry!
680             struct sockaddr_in6 dumb =
681             {
682                 .sin6_family = ptr->ai_addr->sa_family,
683                 .sin6_port =  ((struct sockaddr_in *)(ptr->ai_addr))->sin_port
684             };
685
686             bind (fd, (struct sockaddr *)&dumb, ptr->ai_addrlen);
687         }
688         else
689 #endif
690         if (bind (fd, ptr->ai_addr, ptr->ai_addrlen))
691         {
692             net_Close (fd);
693             continue;
694         }
695
696         val = -1;
697         for (struct addrinfo *ptr2 = rem; ptr2 != NULL; ptr2 = ptr2->ai_next)
698         {
699             if ((ptr2->ai_family != ptr->ai_family)
700              || (ptr2->ai_socktype != ptr->ai_socktype)
701              || (ptr2->ai_protocol != ptr->ai_protocol))
702                 continue;
703
704             if (net_SockAddrIsMulticast (ptr->ai_addr, ptr->ai_addrlen)
705               ? net_SourceSubscribe (obj, fd,
706                                      ptr2->ai_addr, ptr2->ai_addrlen,
707                                      ptr->ai_addr, ptr->ai_addrlen)
708               : connect (fd, ptr2->ai_addr, ptr2->ai_addrlen))
709             {
710                 msg_Err (obj, "cannot connect to %s port %d: %s",
711                          psz_server, i_server, net_strerror (net_errno));
712                 continue;
713             }
714             val = fd;
715             break;
716         }
717
718         if (val != -1)
719             break;
720
721         close (fd);
722     }
723
724     vlc_freeaddrinfo (rem);
725     vlc_freeaddrinfo (loc);
726     return val;
727 }
728