Adds socket.getInfo to the socket API

chrome/browser/extensions/api/socket/socket_api.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 "chrome/browser/extensions/api/socket/socket_api.h"
 
 #include "base/bind.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/api/dns/host_resolver_wrapper.h"
 #include "chrome/browser/extensions/api/socket/socket.h"
@@ skipping 429 matching lines @@
 void SocketSetNoDelayFunction::Work() {
   bool result = false;
   Socket* socket = manager_->Get(params_->socket_id);
   if (socket)
     result = socket->SetNoDelay(params_->no_delay);
   else
     error_ = kSocketNotFoundError;
   SetResult(Value::CreateBooleanValue(result));
 }
 
+SocketGetInfoFunction::SocketGetInfoFunction()
+    : params_(NULL) {

[benwells, 2012/07/30 00:47:58]

Nit: can use {} on the one line.

[thorogood, 2012/07/30 07:22:18]

Done, I've left the ": params_(NULL) {}" on the next line though (even though I
could fit it on the previous). Woo, nits.

+}
+
+SocketGetInfoFunction::~SocketGetInfoFunction() {}
+
+bool SocketGetInfoFunction::Prepare() {
+  params_ = api::experimental_socket::GetInfo::Params::Create(*args_);
+  EXTENSION_FUNCTION_VALIDATE(params_.get());
+  return true;
+}
+
+void SocketGetInfoFunction::Work() {
+  api::experimental_socket::SocketInfo info;
+  Socket* socket = manager_->Get(params_->socket_id);
+  if (socket) {
+    // This represents what we know about the socket, and does not call through
+    // to the system.
+    info.socket_type = (socket->IsTCPSocket() ? kTCPOption : kUDPOption);
+    info.connected = socket->IsConnected();
+
+    // Grab the peer address as known by the OS. This is racey along with the
+    // second call below; it is possible e.g. for a TCP socket to be connected
+    // here, but disconnected by the OS before we grab our localAddress, which
+    // will then be blank.
+    net::IPEndPoint peerAddress;
+    if (socket->GetPeerAddress(&peerAddress)) {
+      info.peer_address.reset(
+          new std::string(peerAddress.ToStringWithoutPort()));
+      info.peer_port.reset(new int(peerAddress.port()));
+    }
+
+    // Grab the local address as known by the OS.
+    net::IPEndPoint localAddress;
+    if (socket->GetLocalAddress(&localAddress)) {
+      info.local_address.reset(
+          new std::string(localAddress.ToStringWithoutPort()));
+      info.local_port.reset(new int(localAddress.port()));
+    }
+  } else {
+    error_ = kSocketNotFoundError;
+  }
+  SetResult(info.ToValue().release());
+}
+
 }  // namespace extensions