Multicast socket API

net/udp/udp_socket_libevent.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/udp/udp_socket_libevent.h"
 
 #include <errno.h>
 #include <fcntl.h>
 #include <netdb.h>
 #include <sys/socket.h>
@@ skipping 345 matching lines @@
         CreateNetLogUDPDataTranferCallback(
             result, bytes,
             is_address_valid ? &address : NULL));
   }
 
   base::StatsCounter read_bytes("udp.read_bytes");
   read_bytes.Add(result);
 }
 
 int UDPSocketLibevent::CreateSocket(const IPEndPoint& address) {
-  socket_ = socket(address.GetSockAddrFamily(), SOCK_DGRAM, 0);
+  sock_addr_family_ = address.GetSockAddrFamily();
+  socket_ = socket(sock_addr_family_, SOCK_DGRAM, 0);
   if (socket_ == kInvalidSocket)
     return MapSystemError(errno);
   if (SetNonBlocking(socket_)) {
     const int err = MapSystemError(errno);
     Close();
     return err;
   }
   return OK;
 }
 
@@ skipping 125 matching lines @@
   IPAddressNumber ip(address.address().size());
 
   for (int i = 0; i < kBindRetries; ++i) {
     int rv = DoBind(IPEndPoint(ip, rand_int_cb_.Run(kPortStart, kPortEnd)));
     if (rv == OK || rv != ERR_ADDRESS_IN_USE)
       return rv;
   }
   return DoBind(IPEndPoint(ip, 0));
 }
 
+int UDPSocketLibevent::JoinGroup(
+    const net::IPAddressNumber& group_address) const {
+  DCHECK(CalledOnValidThread());
+  DCHECK(is_connected());
+
+  int rv = -1;
+  switch(group_address.size()) {
+    case kIPv4AddressSize: {
+      if (sock_addr_family_ != AF_INET) {
+        return ERR_ADDRESS_INVALID;
+      }
+      ip_mreq mreq;
+      memset(&mreq, 0, sizeof(mreq));
+      memcpy(&mreq.imr_multiaddr, &group_address[0], kIPv4AddressSize);
+      rv = setsockopt(socket_, IPPROTO_IP, IP_ADD_MEMBERSHIP,
+                      &mreq, sizeof(mreq));
+      break;
+    }
+    case kIPv6AddressSize: {
+      if (sock_addr_family_ != AF_INET6) {
+        return ERR_ADDRESS_INVALID;
+      }
+      ipv6_mreq mreq;
+      memset(&mreq, 0, sizeof(mreq));
+      memcpy(&mreq.ipv6mr_multiaddr, &group_address[0], kIPv6AddressSize);
+      rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_JOIN_GROUP,
+                      &mreq, sizeof(mreq));
+      break;
+    }
+  }
+
+  return MapSystemError(rv);
+}
+
+int UDPSocketLibevent::LeaveGroup(
+    const net::IPAddressNumber& group_address) const {
+  DCHECK(CalledOnValidThread());
+  DCHECK(is_connected());
+
+  int rv = -1;
+  switch(group_address.size()) {
+    case kIPv4AddressSize: {
+      if (sock_addr_family_ != AF_INET) {
+        return ERR_ADDRESS_INVALID;
+      }
+      ip_mreq mreq;
+      memset(&mreq, 0, sizeof(mreq));
+      memcpy(&mreq.imr_multiaddr, &group_address[0], kIPv4AddressSize);
+      rv = setsockopt(socket_, IPPROTO_IP, IP_DROP_MEMBERSHIP,
+                          &mreq, sizeof(mreq));
+      break;
+    }
+    case kIPv6AddressSize: {
+      if (sock_addr_family_ != AF_INET6) {
+        return ERR_ADDRESS_INVALID;
+      }
+      ipv6_mreq mreq;
+      memset(&mreq, 0, sizeof(mreq));
+      memcpy(&mreq.ipv6mr_multiaddr, &group_address[0], kIPv6AddressSize);
+      rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_LEAVE_GROUP,
+                          &mreq, sizeof(mreq));
+      break;
+    }
+  }
+
+  return MapSystemError(rv);
+}
+
+int UDPSocketLibevent::SetMulticastTimeToLive(int timeToLive) {
+  DCHECK(CalledOnValidThread());
+  DCHECK(is_connected());
+
+  uint8 ttl = timeToLive;
+  int ip_family = sock_addr_family_ == AF_INET ? IPPROTO_IP : IPPROTO_IPV6;
+  return MapSystemError(setsockopt(socket_,
+      ip_family,
+      IP_MULTICAST_TTL,
+      &ttl, sizeof(ttl)));
+}
+
+int UDPSocketLibevent::SetMulticastLoopbackMode(bool loopback) {
+  DCHECK(CalledOnValidThread());
+  DCHECK(is_connected());
+
+  uint8 loop = loopback;
+  int ip_family = sock_addr_family_ == AF_INET ? IPPROTO_IP : IPPROTO_IPV6;
+  return MapSystemError(setsockopt(socket_,
+      ip_family,
+      IP_MULTICAST_LOOP,
+      &loop, sizeof(loop)));
+}
 }  // namespace net