Implement chrome.socket.bind/listen/accept for TCP server socket.

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 206 matching lines @@
   int result = -1;
   Socket* socket = manager_->Get(socket_id_);
   if (socket)
     result = socket->Bind(address_, port_);
   else
     error_ = kSocketNotFoundError;
 
   SetResult(Value::CreateIntegerValue(result));
 }
 
+bool SocketListenFunction::Prepare() {
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(0, &socket_id_));

[miket_OOO, 2012/08/31 23:18:01]

You can do an EXTENSION_FUNCTION_VALIDATE(params_.get()) if that's easier, and
then use fewer custom member variables.

[justinlin, 2012/09/11 04:32:32]

Done.

+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(1, &backlog_));
+  return true;
+}
+
+void SocketListenFunction::Work() {
+  int result = -1;
+  Socket* socket = manager_->Get(socket_id_);
+  if (socket)
+    result = socket->Listen(backlog_);
+  else
+    error_ = kSocketNotFoundError;
+
+  SetResult(Value::CreateIntegerValue(result));
+}
+
+SocketAcceptFunction::SocketAcceptFunction()
+  : params_(NULL),
+    event_notifier_(NULL),
+    src_id_(-1) {
+
+}

[miket_OOO, 2012/08/31 23:18:01]

vertical whitespace

[justinlin, 2012/09/11 04:32:32]

Done.

+SocketAcceptFunction::~SocketAcceptFunction() {}
+
+bool SocketAcceptFunction::Prepare() {
+  params_ = api::socket::Accept::Params::Create(*args_);
+  EXTENSION_FUNCTION_VALIDATE(params_.get());
+
+  event_notifier_ = CreateEventNotifier(src_id_);
+  return true;
+}
+
+void SocketAcceptFunction::Work() {
+  Socket* socket = manager_->Get(params_->socket_id);
+  if (socket) {
+    socket->Accept(base::Bind(&SocketAcceptFunction::OnAccept, this));
+  } else {
+    error_ = kSocketNotFoundError;
+  }
+}
+
+void SocketAcceptFunction::OnAccept(int result_code,
+                                    net::TCPClientSocket *socket) {
+  DCHECK(socket);
+  Socket *client_socket = new TCPSocket(event_notifier_, socket);
+
+  DictionaryValue* result = new DictionaryValue();
+  result->SetInteger(kResultCodeKey, result_code);
+  result->SetInteger(kSocketIdKey, manager_->Add(client_socket));

[miket_OOO, 2012/08/31 23:18:01]

You might want to add a note to
http://code.google.com/p/chromium/issues/detail?id=142521 and make sure that
these sockets are added to whatever is-OK list is part of the solution to that
bug. Charles is going to be working on that bug very soon, and he might make
your code mysteriously stop working if you guys don't keep in sync.

[justinlin, 2012/09/11 04:32:32]

I'll keep an eye out for that issue. I'm not entirely sure how that will be
implemented, but there shouldn't be issues since this will look like any other
socket.

On 2012/08/31 23:18:01, miket wrote:

+  SetResult(result);
+}
+
 SocketReadFunction::SocketReadFunction()
     : params_(NULL) {
 }
 
 SocketReadFunction::~SocketReadFunction() {}
 
 bool SocketReadFunction::Prepare() {
   params_ = api::socket::Read::Params::Create(*args_);
   EXTENSION_FUNCTION_VALIDATE(params_.get());
   return true;
@@ skipping 254 matching lines @@
           new std::string(localAddress.ToStringWithoutPort()));
       info.local_port.reset(new int(localAddress.port()));
     }
   } else {
     error_ = kSocketNotFoundError;
   }
   SetResult(info.ToValue().release());
 }
 
 }  // namespace extensions