content/browser/renderer_host/p2p/socket_host.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/renderer_host/p2p/socket_host.h"
   6
   7 #include "base/sys_byteorder.h"
   8 #include "content/browser/renderer_host/p2p/socket_host_tcp.h"
   9 #include "content/browser/renderer_host/p2p/socket_host_tcp_server.h"
  10 #include "content/browser/renderer_host/p2p/socket_host_udp.h"
  11
  12 namespace {
  13 const uint32 kStunMagicCookie = 0x2112A442;
  14 }  // namespace
  15
  16 namespace content {
  17
  18 P2PSocketHost::P2PSocketHost(IPC::Sender* message_sender,
  19                              int id)
  20     : message_sender_(message_sender),
  21       id_(id),
  22       state_(STATE_UNINITIALIZED) {
  23 }
  24
  25 P2PSocketHost::~P2PSocketHost() { }
  26
  27 // Verifies that the packet |data| has a valid STUN header.
  28 // static
  29 bool P2PSocketHost::GetStunPacketType(
  30     const char* data, int data_size, StunMessageType* type) {
  31
  32   if (data_size < kStunHeaderSize)
  33     return false;
  34
  35   uint32 cookie = base::NetToHost32(*reinterpret_cast<const uint32*>(data + 4));
  36   if (cookie != kStunMagicCookie)
  37     return false;
  38
  39   uint16 length = base::NetToHost16(*reinterpret_cast<const uint16*>(data + 2));
  40   if (length != data_size - kStunHeaderSize)
  41     return false;
  42
  43   int message_type = base::NetToHost16(*reinterpret_cast<const uint16*>(data));
  44
  45   // Verify that the type is known:
  46   switch (message_type) {
  47     case STUN_BINDING_REQUEST:
  48     case STUN_BINDING_RESPONSE:
  49     case STUN_BINDING_ERROR_RESPONSE:
  50     case STUN_SHARED_SECRET_REQUEST:
  51     case STUN_SHARED_SECRET_RESPONSE:
  52     case STUN_SHARED_SECRET_ERROR_RESPONSE:
  53     case STUN_ALLOCATE_REQUEST:
  54     case STUN_ALLOCATE_RESPONSE:
  55     case STUN_ALLOCATE_ERROR_RESPONSE:
  56     case STUN_SEND_REQUEST:
  57     case STUN_SEND_RESPONSE:
  58     case STUN_SEND_ERROR_RESPONSE:
  59     case STUN_DATA_INDICATION:
  60       *type = static_cast<StunMessageType>(message_type);
  61       return true;
  62
  63     default:
  64       return false;
  65   }
  66 }
  67
  68 // static
  69 bool P2PSocketHost::IsRequestOrResponse(StunMessageType type) {
  70   return type == STUN_BINDING_REQUEST || type == STUN_BINDING_RESPONSE ||
  71       type == STUN_ALLOCATE_REQUEST || type == STUN_ALLOCATE_RESPONSE;
  72 }
  73
  74 // static
  75 P2PSocketHost* P2PSocketHost::Create(
  76     IPC::Sender* message_sender, int id, P2PSocketType type,
  77     net::URLRequestContextGetter* url_context,
  78     P2PMessageThrottler* throttler) {
  79   switch (type) {
  80     case P2P_SOCKET_UDP:
  81       return new P2PSocketHostUdp(message_sender, id, throttler);
  82     case P2P_SOCKET_TCP_SERVER:
  83       return new P2PSocketHostTcpServer(
  84           message_sender, id, P2P_SOCKET_TCP_CLIENT);
  85
  86     case P2P_SOCKET_STUN_TCP_SERVER:
  87       return new P2PSocketHostTcpServer(
  88           message_sender, id, P2P_SOCKET_STUN_TCP_CLIENT);
  89
  90     case P2P_SOCKET_TCP_CLIENT:
  91     case P2P_SOCKET_SSLTCP_CLIENT:
  92     case P2P_SOCKET_TLS_CLIENT:
  93       return new P2PSocketHostTcp(message_sender, id, type, url_context);
  94
  95     case P2P_SOCKET_STUN_TCP_CLIENT:
  96     case P2P_SOCKET_STUN_SSLTCP_CLIENT:
  97     case P2P_SOCKET_STUN_TLS_CLIENT:
  98       return new P2PSocketHostStunTcp(message_sender, id, type, url_context);
  99   }
 100
 101   NOTREACHED();
 102   return NULL;
 103 }
 104
 105 }  // namespace content