Properly handle STUN requests and responses from relay servers.

content/browser/renderer_host/p2p/socket_host_udp.cc

 // Copyright (c) 2011 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 "content/browser/renderer_host/p2p/socket_host_udp.h"
 
 #include "content/common/p2p_messages.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/base/net_util.h"
@@ skipping 82 matching lines @@
 
 void P2PSocketHostUdp::DidCompleteRead(int result) {
   DCHECK_EQ(state_, STATE_OPEN);
 
   if (result > 0) {
     std::vector<char> data(recv_buffer_->data(), recv_buffer_->data() + result);
 
     if (authorized_peers_.find(recv_address_) == authorized_peers_.end()) {
       P2PSocketHost::StunMessageType type;
       bool stun = GetStunPacketType(&*data.begin(), data.size(), &type);
-      if (stun && (type == STUN_BINDING_REQUEST ||
-                   type == STUN_BINDING_RESPONSE)) {
+      if (stun && IsRequestOrResponse(type)) {
         authorized_peers_.insert(recv_address_);
       } else if (!stun || type == STUN_DATA_INDICATION) {
         LOG(ERROR) << "Received unexpected data packet from "
                    << recv_address_.ToString()
                    << " before STUN binding is finished.";
         return;
       }
     }
 
     message_sender_->Send(new P2PMsg_OnDataReceived(routing_id_, id_,
@@ skipping 48 matching lines @@
   if (result < 0)
     OnError();
 }
 
 P2PSocketHost* P2PSocketHostUdp::AcceptIncomingTcpConnection(
     const net::IPEndPoint& remote_address, int id) {
   NOTREACHED();
   OnError();
   return NULL;
 }