gst/udp/gstudpnetutils.c

   1 /* GStreamer UDP network utility functions
   2  * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>
   3  * Copyright (C) 2006 Joni Valtanen <joni.valtanen@movial.fi>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24
  25 #include <errno.h>
  26 #include <stdio.h>
  27 #include <memory.h>
  28
  29 /* EAI_ADDRFAMILY was obsoleted in BSD at some point */
  30 #ifndef EAI_ADDRFAMILY
  31 #define EAI_ADDRFAMILY 1
  32 #endif
  33
  34 #include "gstudpnetutils.h"
  35
  36 #ifdef G_OS_WIN32
  37
  38 gboolean
  39 gst_udp_net_utils_win32_wsa_startup (GstObject * obj)
  40 {
  41   WSADATA w;
  42   int error;
  43
  44   error = WSAStartup (0x0202, &w);
  45
  46   if (error) {
  47     GST_WARNING_OBJECT (obj, "WSAStartup error: %d", error);
  48     return FALSE;
  49   }
  50
  51   if (w.wVersion != 0x0202) {
  52     WSACleanup ();
  53     GST_WARNING_OBJECT (obj, "Winsock version wrong : 0x%x", w.wVersion);
  54     return FALSE;
  55   }
  56
  57   return TRUE;
  58 }
  59
  60 #endif
  61
  62 int
  63 gst_udp_get_sockaddr_length (struct sockaddr_storage *addr)
  64 {
  65   /* MacOS is picky about passing precisely the correct length,
  66    * so we calculate it here for the given socket type.
  67    */
  68   switch (addr->ss_family) {
  69     case AF_INET:
  70       return sizeof (struct sockaddr_in);
  71     case AF_INET6:
  72       return sizeof (struct sockaddr_in6);
  73     default:
  74       /* don't know, Screw MacOS and use the full length */
  75       return sizeof (*addr);
  76   }
  77 }
  78
  79 int
  80 gst_udp_get_addr (const char *hostname, int port, struct sockaddr_storage *addr)
  81 {
  82   struct addrinfo hints, *res, *nres;
  83   char service[NI_MAXSERV];
  84   int ret;
  85
  86   memset (&hints, 0, sizeof (hints));
  87   hints.ai_family = AF_UNSPEC;
  88   hints.ai_socktype = SOCK_DGRAM;
  89   g_snprintf (service, sizeof (service) - 1, "%d", port);
  90   service[sizeof (service) - 1] = '\0';
  91
  92   if ((ret = getaddrinfo (hostname, (port == -1) ? NULL : service, &hints,
  93               &res)) < 0) {
  94     return ret;
  95   }
  96
  97   nres = res;
  98   while (nres) {
  99     if (nres->ai_family == AF_INET || nres->ai_family == AF_INET6)
 100       break;
 101     nres = nres->ai_next;
 102   }
 103
 104   if (nres) {
 105     memcpy (addr, nres->ai_addr, nres->ai_addrlen);
 106   } else {
 107     errno = EAI_ADDRFAMILY;
 108     ret = -1;
 109   }
 110   freeaddrinfo (res);
 111
 112   return ret;
 113 }
 114
 115 int
 116 gst_udp_set_loop_ttl (int sockfd, gboolean loop, int ttl)
 117 {
 118   int ret = -1;
 119
 120 #if 0
 121   int l = (loop == FALSE) ? 0 : 1;
 122
 123   switch (addr->ss_family) {
 124     case AF_INET:
 125     {
 126       if ((ret =
 127               setsockopt (sockfd, IPPROTO_IP, IP_MULTICAST_LOOP, &l,
 128                   sizeof (l))) < 0)
 129         return ret;
 130
 131       if ((ret =
 132               setsockopt (sockfd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl,
 133                   sizeof (ttl))) < 0)
 134         return ret;
 135       break;
 136     }
 137     case AF_INET6:
 138     {
 139       if ((ret =
 140               setsockopt (sockfd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, &l,
 141                   sizeof (l))) < 0)
 142         return ret;
 143
 144       if ((ret =
 145               setsockopt (sockfd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &ttl,
 146                   sizeof (ttl))) < 0)
 147         return ret;
 148
 149       break;
 150     }
 151     default:
 152       errno = EAFNOSUPPORT;
 153   }
 154 #endif
 155   return ret;
 156 }
 157
 158 int
 159 gst_udp_join_group (int sockfd, struct sockaddr_storage *addr)
 160 {
 161   int ret = -1;
 162
 163   switch (addr->ss_family) {
 164     case AF_INET:
 165     {
 166       struct ip_mreq mreq4;
 167
 168       mreq4.imr_multiaddr.s_addr =
 169           ((struct sockaddr_in *) addr)->sin_addr.s_addr;
 170       mreq4.imr_interface.s_addr = INADDR_ANY;
 171
 172       if ((ret =
 173               setsockopt (sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP,
 174                   (const void *) &mreq4, sizeof (mreq4))) < 0)
 175         return ret;
 176
 177       break;
 178     }
 179     case AF_INET6:
 180     {
 181       struct ipv6_mreq mreq6;
 182
 183       memcpy (&mreq6.ipv6mr_multiaddr,
 184           &(((struct sockaddr_in6 *) addr)->sin6_addr),
 185           sizeof (struct in6_addr));
 186       mreq6.ipv6mr_interface = 0;
 187
 188       if ((ret =
 189               setsockopt (sockfd, IPPROTO_IPV6, IPV6_JOIN_GROUP,
 190                   (const void *) &mreq6, sizeof (mreq6))) < 0)
 191         return ret;
 192
 193       break;
 194     }
 195     default:
 196 #ifdef G_OS_WIN32
 197       WSASetLastError (WSAEAFNOSUPPORT);
 198 #else
 199       errno = EAFNOSUPPORT;
 200 #endif
 201   }
 202   return ret;
 203 }
 204
 205 int
 206 gst_udp_leave_group (int sockfd, struct sockaddr_storage *addr)
 207 {
 208   int ret = -1;
 209
 210   switch (addr->ss_family) {
 211     case AF_INET:
 212     {
 213       struct ip_mreq mreq4;
 214
 215       mreq4.imr_multiaddr.s_addr =
 216           ((struct sockaddr_in *) addr)->sin_addr.s_addr;
 217       mreq4.imr_interface.s_addr = INADDR_ANY;
 218
 219       if ((ret =
 220               setsockopt (sockfd, IPPROTO_IP, IP_DROP_MEMBERSHIP,
 221                   (const void *) &mreq4, sizeof (mreq4))) < 0)
 222         return ret;
 223     }
 224       break;
 225
 226     case AF_INET6:
 227     {
 228       struct ipv6_mreq mreq6;
 229
 230       memcpy (&mreq6.ipv6mr_multiaddr,
 231           &(((struct sockaddr_in6 *) addr)->sin6_addr),
 232           sizeof (struct in6_addr));
 233       mreq6.ipv6mr_interface = 0;
 234
 235       if ((ret =
 236               setsockopt (sockfd, IPPROTO_IPV6, IPV6_LEAVE_GROUP,
 237                   (const void *) &mreq6, sizeof (mreq6))) < 0)
 238         return ret;
 239     }
 240       break;
 241
 242     default:
 243 #ifdef G_OS_WIN32
 244       WSASetLastError (WSAEAFNOSUPPORT);
 245 #else
 246       errno = EAFNOSUPPORT;
 247 #endif
 248   }
 249
 250   return ret;
 251 }
 252
 253 int
 254 gst_udp_is_multicast (struct sockaddr_storage *addr)
 255 {
 256   int ret = -1;
 257
 258   switch (addr->ss_family) {
 259     case AF_INET:
 260     {
 261       struct sockaddr_in *addr4 = (struct sockaddr_in *) addr;
 262
 263       ret = IN_MULTICAST (g_ntohl (addr4->sin_addr.s_addr));
 264     }
 265       break;
 266
 267     case AF_INET6:
 268     {
 269       struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *) addr;
 270
 271       ret = IN6_IS_ADDR_MULTICAST (&addr6->sin6_addr);
 272     }
 273       break;
 274
 275     default:
 276 #ifdef G_OS_WIN32
 277       WSASetLastError (WSAEAFNOSUPPORT);
 278 #else
 279       errno = EAFNOSUPPORT;
 280 #endif
 281   }
 282
 283   return ret;
 284 }