Allow p2p UDP packages to set DSCP

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 23 matching lines @@
 namespace net {
 
 UDPSocketLibevent::UDPSocketLibevent(
     DatagramSocket::BindType bind_type,
     const RandIntCallback& rand_int_cb,
     net::NetLog* net_log,
     const net::NetLog::Source& source)
         : socket_(kInvalidSocket),
           addr_family_(0),
           socket_options_(SOCKET_OPTION_MULTICAST_LOOP),
+          dscp_(0),
           multicast_time_to_live_(1),
           bind_type_(bind_type),
           rand_int_cb_(rand_int_cb),
           read_watcher_(this),
           write_watcher_(this),
           read_buf_len_(0),
           recv_from_address_(NULL),
           write_buf_len_(0),
           net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_UDP_SOCKET)) {
   net_log_.BeginEvent(NetLog::TYPE_SOCKET_ALIVE,
@@ skipping 470 matching lines @@
                       &ttl, sizeof(ttl));
     } else {
       // Signed interger. -1 to use route default.
       int ttl = multicast_time_to_live_;
       rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
                       &ttl, sizeof(ttl));
     }
     if (rv < 0)
       return MapSystemError(errno);
   }
+
+  if (dscp_) {
+    int rv;
+    int dscp_and_ecn = dscp_ << 2;
+    if (addr_family_ == AF_INET) {
+      rv = setsockopt(socket_, IPPROTO_IP, IP_TOS,
+                      &dscp_and_ecn, sizeof(dscp_and_ecn));
+    } else {
+      rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_TCLASS,
+                      &dscp_and_ecn, sizeof(dscp_and_ecn));
+    }
+    if (rv < 0) {
+      LOG(INFO) << "Failed to set dscp, errno=" << errno;
+    }
+  }
   return OK;
 }
 
 int UDPSocketLibevent::DoBind(const IPEndPoint& address) {
   SockaddrStorage storage;
   if (!address.ToSockAddr(storage.addr, &storage.addr_len))
     return ERR_ADDRESS_INVALID;
   int rv = bind(socket_, storage.addr, storage.addr_len);
   return rv < 0 ? MapSystemError(errno) : rv;
 }
@@ skipping 100 matching lines @@
   DCHECK(CalledOnValidThread());
   if (is_connected())
     return ERR_SOCKET_IS_CONNECTED;
 
   if (loopback)
     socket_options_ |= SOCKET_OPTION_MULTICAST_LOOP;
   else
     socket_options_ &= ~SOCKET_OPTION_MULTICAST_LOOP;
   return OK;
 }
+
+int UDPSocketLibevent::SetToS(int dscp) {
+  DCHECK_GE(dscp, 0);
+  DCHECK_LT(dscp, 64);
+  if (is_connected())
+    return ERR_SOCKET_IS_CONNECTED;
+
+  dscp_ = dscp;
+  return OK;
+}
+
 }  // namespace net