An implementation of chrome.socket.secure().

extensions/browser/api/socket/socket_api.h

 // Copyright 2014 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.
 
 #ifndef EXTENSIONS_BROWSER_API_SOCKET_SOCKET_API_H_
 #define EXTENSIONS_BROWSER_API_SOCKET_SOCKET_API_H_
 
 #include <string>
 
 #include "base/gtest_prod_util.h"
 #include "base/memory/ref_counted.h"
 #include "extensions/browser/api/api_resource_manager.h"
 #include "extensions/browser/api/async_api_function.h"
 #include "extensions/browser/extension_function.h"
 #include "extensions/common/api/socket.h"
 #include "net/base/address_list.h"
 #include "net/dns/host_resolver.h"
 #include "net/socket/tcp_client_socket.h"
 
 namespace content {
 class BrowserContext;
 class ResourceContext;
 }
 
 namespace net {
 class IOBuffer;
+class URLRequestContextGetter;
+class SSLClientSocket;
 }
 
 namespace extensions {
-
+class TLSSocket;
 class Socket;
 
 // A simple interface to ApiResourceManager<Socket> or derived class. The goal
 // of this interface is to allow Socket API functions to use distinct instances
 // of ApiResourceManager<> depending on the type of socket (old version in
 // "socket" namespace vs new version in "socket.xxx" namespaces).
 class SocketResourceManagerInterface {
  public:
   virtual ~SocketResourceManagerInterface() {}
 
   virtual bool SetBrowserContext(content::BrowserContext* context) = 0;
-  virtual int Add(Socket* socket) = 0;
-  virtual Socket* Get(const std::string& extension_id, int api_resource_id) = 0;
-  virtual void Remove(const std::string& extension_id, int api_resource_id) = 0;
+  virtual int Add(Socket *socket) = 0;
+  virtual Socket* Get(const std::string& extension_id,
+                      int api_resource_id) = 0;
+  virtual void Remove(const std::string& extension_id,
+                      int api_resource_id) = 0;
+  virtual void Replace(const std::string& extension_id,
+                       int api_resource_id,
+                       Socket* socket) = 0;
   virtual base::hash_set<int>* GetResourceIds(
       const std::string& extension_id) = 0;
 };
 
 // Implementation of SocketResourceManagerInterface using an
 // ApiResourceManager<T> instance (where T derives from Socket).
 template <typename T>
 class SocketResourceManager : public SocketResourceManagerInterface {
  public:
   SocketResourceManager() : manager_(NULL) {}
@@ skipping 11 matching lines @@
   virtual int Add(Socket* socket) OVERRIDE {
     // Note: Cast needed here, because "T" may be a subclass of "Socket".
     return manager_->Add(static_cast<T*>(socket));
   }
 
   virtual Socket* Get(const std::string& extension_id,
                       int api_resource_id) OVERRIDE {
     return manager_->Get(extension_id, api_resource_id);
   }
 
+  virtual void Replace(const std::string& extension_id,
+                       int api_resource_id,
+                       Socket* socket) OVERRIDE {
+    manager_->Replace(extension_id, api_resource_id, static_cast<T*>(socket));
+  }
+
   virtual void Remove(const std::string& extension_id,
                       int api_resource_id) OVERRIDE {
     manager_->Remove(extension_id, api_resource_id);
   }
 
   virtual base::hash_set<int>* GetResourceIds(const std::string& extension_id)
       OVERRIDE {
     return manager_->GetResourceIds(extension_id);
   }
 
@@ skipping 10 matching lines @@
 
   // AsyncApiFunction:
   virtual bool PrePrepare() OVERRIDE;
   virtual bool Respond() OVERRIDE;
 
   virtual scoped_ptr<SocketResourceManagerInterface>
       CreateSocketResourceManager();
 
   int AddSocket(Socket* socket);
   Socket* GetSocket(int api_resource_id);
+  void ReplaceSocket(int api_resource_id, Socket* socket);
   void RemoveSocket(int api_resource_id);
   base::hash_set<int>* GetSocketIds();
 
  private:
   scoped_ptr<SocketResourceManagerInterface> manager_;
 };
 
 class SocketExtensionWithDnsLookupFunction : public SocketAsyncApiFunction {
  protected:
   SocketExtensionWithDnsLookupFunction();
@@ skipping 379 matching lines @@
  protected:
   virtual ~SocketGetJoinedGroupsFunction();
 
   // AsyncApiFunction
   virtual bool Prepare() OVERRIDE;
   virtual void Work() OVERRIDE;
 
  private:
   scoped_ptr<core_api::socket::GetJoinedGroups::Params> params_;
 };
+
+class SocketSecureFunction : public SocketAsyncApiFunction {
+ public:
+  DECLARE_EXTENSION_FUNCTION("socket.secure", SOCKET_SECURE);
+  SocketSecureFunction();
+
+ protected:
+  virtual ~SocketSecureFunction();
+
+  // AsyncApiFunction
+  virtual bool Prepare() OVERRIDE;
+  virtual void AsyncWorkStart() OVERRIDE;
+
+ private:
+  // Callback from TLSSocket::UpgradeSocketToTLS().
+  void TlsConnectDone(scoped_ptr<TLSSocket> socket, int result);
+
+  scoped_ptr<core_api::socket::Secure::Params> params_;
+  scoped_refptr<net::URLRequestContextGetter> url_request_getter_;
+
+  DISALLOW_COPY_AND_ASSIGN(SocketSecureFunction);
+};
+
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_API_SOCKET_SOCKET_API_H_