Add UDP support to dart:io

runtime/bin/socket_linux.cc

Index: runtime/bin/socket_linux.cc
diff --git a/runtime/bin/socket_linux.cc b/runtime/bin/socket_linux.cc
index c36a66e3e26ff76e49bed10a17dba9f4e60dca59..9870a0786adc904a7e1834de5b9e604d4d012471 100644
--- a/runtime/bin/socket_linux.cc
+++ b/runtime/bin/socket_linux.cc
@@ -39,6 +39,21 @@ SocketAddress::SocketAddress(struct sockaddr* sa) {
 }
 
 
+bool Socket::FormatNumericAddress(RawAddr* addr, char* address, int len) {
+  socklen_t salen = SocketAddress::GetAddrLength(addr);
+  if (TEMP_FAILURE_RETRY(getnameinfo(&addr->addr,
+                                     salen,
+                                     address,
+                                     len,
+                                     NULL,
+                                     0,
+                                     NI_NUMERICHOST)) != 0) {
+    return false;
+  }
+  return true;
+}
+
+
 bool Socket::Initialize() {
   // Nothing to do on Linux.
   return true;
@@ -106,6 +121,21 @@ int Socket::Read(intptr_t fd, void* buffer, intptr_t num_bytes) {
 }
 
 
+int Socket::RecvFrom(intptr_t fd, void* buffer, intptr_t num_bytes,
+                     RawAddr* addr) {
+  ASSERT(fd >= 0);
+  socklen_t addr_len = sizeof(addr->addr);
+  ssize_t read_bytes =
+      recvfrom(fd, buffer, num_bytes, 0, &addr->addr, &addr_len);
+  if (read_bytes == -1 && errno == EWOULDBLOCK) {
+    // If the read would block we need to retry and therefore return 0
+    // as the number of bytes written.
+    read_bytes = 0;
+  }
+  return read_bytes;
+}
+
+
 int Socket::Write(intptr_t fd, const void* buffer, intptr_t num_bytes) {
   ASSERT(fd >= 0);
   ssize_t written_bytes = TEMP_FAILURE_RETRY(write(fd, buffer, num_bytes));
@@ -119,6 +149,23 @@ int Socket::Write(intptr_t fd, const void* buffer, intptr_t num_bytes) {
 }
 
 
+int Socket::SendTo(intptr_t fd, const void* buffer, intptr_t num_bytes,
+                   RawAddr addr) {
+  ASSERT(fd >= 0);
+  ssize_t written_bytes =
+      TEMP_FAILURE_RETRY(
+          sendto(fd, buffer, num_bytes, 0,
+                 &addr.addr, SocketAddress::GetAddrLength(&addr)));
+  ASSERT(EAGAIN == EWOULDBLOCK);
+  if (written_bytes == -1 && errno == EWOULDBLOCK) {
+    // If the would block we need to retry and therefore return 0 as
+    // the number of bytes written.
+    written_bytes = 0;
+  }
+  return written_bytes;
+}
+
+
 intptr_t Socket::GetPort(intptr_t fd) {
   ASSERT(fd >= 0);
   RawAddr raw;
@@ -256,6 +303,30 @@ bool Socket::ParseAddress(int type, const char* address, RawAddr* addr) {
 }
 
 
+intptr_t Socket::CreateBindDatagram(RawAddr* addr, intptr_t port) {
+  intptr_t fd;
+
+  fd = TEMP_FAILURE_RETRY(socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP));
+  if (fd < 0) return -1;
+
+  FDUtils::SetCloseOnExec(fd);
+
+  if (addr != NULL) {
+    SocketAddress::SetAddrPort(addr, port);
+    if (TEMP_FAILURE_RETRY(
+            bind(fd,
+                 &addr->addr,
+                 SocketAddress::GetAddrLength(addr))) < 0) {
+      TEMP_FAILURE_RETRY(close(fd));
+      return -1;
+    }
+  }
+
+  Socket::SetNonBlocking(fd);
+  return fd;
+}
+
+
 static bool ShouldIncludeIfaAddrs(struct ifaddrs* ifa, int lookup_family) {
   if (ifa->ifa_addr == NULL) {
     // OpenVPN's virtual device tun0.
@@ -409,6 +480,21 @@ bool Socket::SetBlocking(intptr_t fd) {
 }
 
 
+bool Socket::GetNoDelay(intptr_t fd, bool* enabled) {
+  int on;
+  socklen_t len = sizeof(on);
+  int err = TEMP_FAILURE_RETRY(getsockopt(fd,
+                                          IPPROTO_TCP,
+                                          TCP_NODELAY,
+                                          reinterpret_cast<void *>(&on),
+                                          &len));
+  if (err == 0) {
+    *enabled = on == 1;
+  }
+  return err == 0;
+}
+
+
 bool Socket::SetNoDelay(intptr_t fd, bool enabled) {
   int on = enabled ? 1 : 0;
   return TEMP_FAILURE_RETRY(setsockopt(fd,
@@ -418,6 +504,99 @@ bool Socket::SetNoDelay(intptr_t fd, bool enabled) {
                                        sizeof(on))) == 0;
 }
 
+
+bool Socket::GetMulticastLoop(intptr_t fd, bool* enabled) {
+  uint8_t on;
+  socklen_t len = sizeof(on);
+  int err = TEMP_FAILURE_RETRY(getsockopt(fd,
+                                          IPPROTO_IP,
+                                          IP_MULTICAST_LOOP,
+                                          reinterpret_cast<char *>(&on),
+                                          &len));
+  if (err == 0) {
+    *enabled = on == 1;
+  }
+  return err == 0;
+}
+
+
+bool Socket::SetMulticastLoop(intptr_t fd, bool enabled) {
+  uint8_t on = enabled ? 1 : 0;
+  return TEMP_FAILURE_RETRY(setsockopt(fd,
+                                       IPPROTO_IP,
+                                       IP_MULTICAST_LOOP,
+                                       reinterpret_cast<char *>(&on),
+                                       sizeof(on))) == 0;
+}
+
+
+bool Socket::GetMulticastTTL(intptr_t fd, int* value) {
+  uint8_t v;
+  socklen_t len = sizeof(v);
+  int err = TEMP_FAILURE_RETRY(getsockopt(fd,
+                                          IPPROTO_IP,
+                                          IP_MULTICAST_TTL,
+                                          reinterpret_cast<char *>(&v),
+                                          &len));
+  if (err == 0) {
+    *value = v;
+  }
+  return err == 0;
+}
+
+
+bool Socket::SetMulticastTTL(intptr_t fd, int value) {
+  uint8_t v = value;
+  return TEMP_FAILURE_RETRY(setsockopt(fd,
+                                       IPPROTO_IP,
+                                       IP_MULTICAST_TTL,
+                                       reinterpret_cast<char *>(&v),
+                                       sizeof(v))) == 0;
+}
+
+
+bool Socket::GetBroadcast(intptr_t fd, bool* enabled) {
+  int on;
+  socklen_t len = sizeof(on);
+  int err = TEMP_FAILURE_RETRY(getsockopt(fd,
+                                          SOL_SOCKET,
+                                          SO_BROADCAST,
+                                          reinterpret_cast<char *>(&on),
+                                          &len));
+  if (err == 0) {
+    *enabled = on == 1;
+  }
+  return err == 0;
+}
+
+
+bool Socket::SetBroadcast(intptr_t fd, bool enabled) {
+  int on = enabled ? 1 : 0;
+  return TEMP_FAILURE_RETRY(setsockopt(fd,
+                                       SOL_SOCKET,
+                                       SO_BROADCAST,
+                                       reinterpret_cast<char *>(&on),
+                                       sizeof(on))) == 0;
+}
+
+
+bool Socket::JoinMulticast(intptr_t fd, RawAddr* addr, int interfaceIndex) {
+  struct group_req mreq;
+  mreq.gr_interface = interfaceIndex;
+  memmove(&mreq.gr_group, &addr->ss, SocketAddress::GetAddrLength(addr));
+  return TEMP_FAILURE_RETRY(setsockopt(
+      fd, IPPROTO_IP, MCAST_JOIN_GROUP, &mreq, sizeof(mreq))) == 0;
+}
+
+
+bool Socket::LeaveMulticast(intptr_t fd, RawAddr* addr, int interfaceIndex) {
+  struct group_req mreq;
+  mreq.gr_interface = interfaceIndex;
+  memmove(&mreq.gr_group, &addr->ss, SocketAddress::GetAddrLength(addr));
+  return TEMP_FAILURE_RETRY(setsockopt(
+      fd, IPPROTO_IP, MCAST_LEAVE_GROUP, &mreq, sizeof(mreq))) == 0;
+}
+
 }  // namespace bin
 }  // namespace dart