Allow p2p UDP packages to set DSCP

net/udp/udp_socket_win.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_win.h"
 
 #include <mstcpip.h>
 
 #include "base/callback.h"
 #include "base/logging.h"
@@ skipping 148 matching lines @@
 //-----------------------------------------------------------------------------
 
 UDPSocketWin::UDPSocketWin(DatagramSocket::BindType bind_type,
                            const RandIntCallback& rand_int_cb,
                            net::NetLog* net_log,
                            const net::NetLog::Source& source)
     : socket_(INVALID_SOCKET),
       addr_family_(0),
       socket_options_(SOCKET_OPTION_MULTICAST_LOOP),
       multicast_time_to_live_(1),
+      dscp_(DSCP_CS0),
       bind_type_(bind_type),
       rand_int_cb_(rand_int_cb),
       recv_from_address_(NULL),
       net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_UDP_SOCKET)) {
   EnsureWinsockInit();
   net_log_.BeginEvent(NetLog::TYPE_SOCKET_ALIVE,
                       source.ToEventParametersCallback());
   if (bind_type == DatagramSocket::RANDOM_BIND)
     DCHECK(!rand_int_cb.is_null());
 }
@@ skipping 435 matching lines @@
     DWORD hops = multicast_time_to_live_;
     int protocol_level =
         addr_family_ == AF_INET ? IPPROTO_IP : IPPROTO_IPV6;
     int option =
         addr_family_ == AF_INET ? IP_MULTICAST_TTL: IPV6_MULTICAST_HOPS;
     int rv = setsockopt(socket_, protocol_level, option,
                         reinterpret_cast<const char*>(&hops), sizeof(hops));
     if (rv < 0)
       return MapSystemError(WSAGetLastError());
   }
+  if (dscp_) {
+    int rw;
+    DWORD dscp_and_ecn = dscp_ << 2;
+    if (addr_family_ == AF_INET) {
+      rv = setsockopt(socket_, IPPROTO_IP, IP_TOS,
+                      reinterpret_cast<const char*>(&dscp_and_ecn),
+                      sizeof(dscp_and_ecn));
+    } else {
+      rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_TCLASS,
+                      reinterpret_cast<const char*>(&dscp_and_ecn),
+                      sizeof(dscp_and_ecn));
+    }
+    if (rv < 0) {
+      LOG(INFO) << "Failed to set dscp, errno=" << errno;
+    }
+  }
   return OK;
 }
 
 int UDPSocketWin::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(WSAGetLastError()) : rv;
 }
@@ skipping 112 matching lines @@
     return ERR_SOCKET_IS_CONNECTED;
 
   if (loopback)
     socket_options_ |= SOCKET_OPTION_MULTICAST_LOOP;
   else
     socket_options_ &= ~SOCKET_OPTION_MULTICAST_LOOP;
   return OK;
 }
 
 }  // namespace net