An implementation of chrome.socket.secure().

chrome/browser/extensions/api/socket/socket.h

 // 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.
 
 #ifndef CHROME_BROWSER_EXTENSIONS_API_SOCKET_SOCKET_H_
 #define CHROME_BROWSER_EXTENSIONS_API_SOCKET_SOCKET_H_
 
 #include <queue>
 #include <string>
 #include <utility>
@@ skipping 24 matching lines @@
 typedef base::Callback<
   void(int, net::TCPClientSocket*)> AcceptCompletionCallback;
 
 // A Socket wraps a low-level socket and includes housekeeping information that
 // we need to manage it in the context of an extension.
 class Socket : public ApiResource {
  public:
   enum SocketType {
     TYPE_TCP,
     TYPE_UDP,
+    TYPE_TLS
   };
 
   virtual ~Socket();
+
+  const std::string& hostname() const {
+    return hostname_;
+  }
+
+  void set_hostname(const std::string& hostname) {
+    hostname_ = hostname;
+  }
+
+  // |address| is a resolved IP address. Call set_hostname() to save the
+  // original DNS name of the host if this socket may later get upgraded to
+  // TLS (see TLSSocket::UpgradeSocketToTLS()).
   virtual void Connect(const std::string& address,
                        int port,
                        const CompletionCallback& callback) = 0;
   virtual void Disconnect() = 0;
   virtual int Bind(const std::string& address, int port) = 0;
 
   // The |callback| will be called with the number of bytes read into the
   // buffer, or a negative number if an error occurred.
   virtual void Read(int count,
                     const ReadCompletionCallback& callback) = 0;
@@ skipping 37 matching lines @@
 
  protected:
   explicit Socket(const std::string& owner_extension_id_);
 
   void WriteData();
   virtual int WriteImpl(net::IOBuffer* io_buffer,
                         int io_buffer_size,
                         const net::CompletionCallback& callback) = 0;
   virtual void OnWriteComplete(int result);
 
-  const std::string address_;
+  // The DNS name of the host that this socket is connected to, used when
+  // verifying a TLS certificate.
+  std::string hostname_;
   bool is_connected_;
 
  private:
   friend class ApiResourceManager<Socket>;
   static const char* service_name() {
     return "SocketManager";
   }
 
   struct WriteRequest {
     WriteRequest(scoped_refptr<net::IOBuffer> io_buffer,
                  int byte_count,
                  const CompletionCallback& callback);
     ~WriteRequest();
     scoped_refptr<net::IOBuffer> io_buffer;
     int byte_count;
     CompletionCallback callback;
     int bytes_written;
   };
   std::queue<WriteRequest> write_queue_;
   scoped_refptr<net::IOBuffer> io_buffer_write_;
 };
 
 }  //  namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_SOCKET_SOCKET_H_