Add GetBoundAddress to PPB_UDPSocket_Private

content/browser/renderer_host/pepper_udp_socket.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/pepper_udp_socket.h"
 
 #include <string.h>
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "content/browser/renderer_host/pepper_message_filter.h"
 #include "content/public/browser/browser_thread.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/udp/udp_server_socket.h"
 #include "ppapi/proxy/ppapi_messages.h"
 #include "ppapi/shared_impl/private/net_address_private_impl.h"
 
 using content::BrowserThread;
 using ppapi::NetAddressPrivateImpl;
 
 PepperUDPSocket::PepperUDPSocket(
     PepperMessageFilter* manager,
     int32 routing_id,
     uint32 plugin_dispatcher_id,
     uint32 socket_id)
     : manager_(manager),
       routing_id_(routing_id),
       plugin_dispatcher_id_(plugin_dispatcher_id),
-      socket_id_(socket_id) {
+      socket_id_(socket_id),
+      bound_port_(0) {
   DCHECK(manager);
 }
 
 PepperUDPSocket::~PepperUDPSocket() {
   // Make sure there are no further callbacks from socket_.
   if (socket_.get())
     socket_->Close();
 }
 
 void PepperUDPSocket::Bind(const PP_NetAddress_Private& addr) {
   socket_.reset(new net::UDPServerSocket(NULL, net::NetLog::Source()));
 
+  // Reset |bound_port_| to zero.
+  bound_port_ = 0;
+
   net::IPEndPoint address;
   if (!socket_.get() ||
       !NetAddressPrivateImpl::NetAddressToIPEndPoint(addr, &address)) {
     SendBindACK(false);
     return;
   }
 
   int result = socket_->Listen(address);
 
+  // If we are successful, get the local port that we bound to.
+  if (result == net::OK) {
+      net::IPEndPoint current_address;
+      socket_->GetLocalAddress(&current_address);
+      bound_port_ = current_address.port();
+  }
+
   SendBindACK(result == net::OK);
 }
 
 void PepperUDPSocket::RecvFrom(int32_t num_bytes) {
   if (recvfrom_buffer_.get()) {
     SendRecvFromACKError();
     return;
   }
 
   recvfrom_buffer_ = new net::IOBuffer(num_bytes);
@@ skipping 38 matching lines @@
       addr));
 }
 
 void PepperUDPSocket::SendSendToACKError() {
   manager_->Send(new PpapiMsg_PPBUDPSocket_SendToACK(
       routing_id_, plugin_dispatcher_id_, socket_id_, false, 0));
 }
 
 void PepperUDPSocket::SendBindACK(bool result) {
   manager_->Send(new PpapiMsg_PPBUDPSocket_BindACK(
-      routing_id_, plugin_dispatcher_id_, socket_id_, result));
+      routing_id_, plugin_dispatcher_id_, socket_id_, result, bound_port_));
 }
 
 void PepperUDPSocket::OnRecvFromCompleted(int result) {
   DCHECK(recvfrom_buffer_.get());
 
   // Convert IPEndPoint we get back from RecvFrom to a PP_NetAddress_Private,
   // to send back.
   PP_NetAddress_Private addr = NetAddressPrivateImpl::kInvalidNetAddress;
   if (result < 0 ||
       !NetAddressPrivateImpl::IPEndPointToNetAddress(recvfrom_address_,
                                                      &addr)) {
     SendRecvFromACKError();
   } else {
     manager_->Send(new PpapiMsg_PPBUDPSocket_RecvFromACK(
         routing_id_, plugin_dispatcher_id_, socket_id_, true,
         std::string(recvfrom_buffer_->data(), result), addr));
   }
 
   recvfrom_buffer_ = NULL;
 }
 
 void PepperUDPSocket::OnSendToCompleted(int result) {
   DCHECK(sendto_buffer_.get());
 
   manager_->Send(new PpapiMsg_PPBUDPSocket_SendToACK(
       routing_id_, plugin_dispatcher_id_, socket_id_, true, result));
 
   sendto_buffer_ = NULL;
 }