1 files changed, 126 insertions, 0 deletions
diff --git a/extensions/browser/api/socket/socket.h b/extensions/browser/api/socket/socket.h
new file mode 100644
index 0000000..9ae45e7
--- /dev/null
+++ b/extensions/browser/api/socket/socket.h
@@ -0,0 +1,126 @@
+// 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_H_
+#define EXTENSIONS_BROWSER_API_SOCKET_SOCKET_H_
+
+#include <queue>
+#include <string>
+#include <utility>
+
+#include "base/callback.h"
+#include "base/memory/ref_counted.h"
+#include "extensions/browser/api/api_resource.h"
+#include "extensions/browser/api/api_resource_manager.h"
+#include "net/base/completion_callback.h"
+#include "net/base/io_buffer.h"
+#include "net/base/ip_endpoint.h"
+#include "net/socket/tcp_client_socket.h"
+
+namespace net {
+class AddressList;
+class IPEndPoint;
+class Socket;
+}
+
+namespace extensions {
+
+typedef base::Callback<void(int)> CompletionCallback;
+typedef base::Callback<void(int, scoped_refptr<net::IOBuffer> io_buffer)>
+    ReadCompletionCallback;
+typedef base::Callback<
+    void(int, scoped_refptr<net::IOBuffer> io_buffer, const std::string&, int)>
+    RecvFromCompletionCallback;
+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, };
+
+  virtual ~Socket();
+  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;
+
+  // The |callback| will be called with |byte_count| or a negative number if an
+  // error occurred.
+  void Write(scoped_refptr<net::IOBuffer> io_buffer,
+             int byte_count,
+             const CompletionCallback& callback);
+
+  virtual void RecvFrom(int count,
+                        const RecvFromCompletionCallback& callback) = 0;
+  virtual void SendTo(scoped_refptr<net::IOBuffer> io_buffer,
+                      int byte_count,
+                      const std::string& address,
+                      int port,
+                      const CompletionCallback& callback) = 0;
+
+  virtual bool SetKeepAlive(bool enable, int delay);
+  virtual bool SetNoDelay(bool no_delay);
+  virtual int Listen(const std::string& address,
+                     int port,
+                     int backlog,
+                     std::string* error_msg);
+  virtual void Accept(const AcceptCompletionCallback& callback);
+
+  virtual bool IsConnected() = 0;
+
+  virtual bool GetPeerAddress(net::IPEndPoint* address) = 0;
+  virtual bool GetLocalAddress(net::IPEndPoint* address) = 0;
+
+  virtual SocketType GetSocketType() const = 0;
+
+  static bool StringAndPortToAddressList(const std::string& ip_address_str,
+                                         int port,
+                                         net::AddressList* address_list);
+  static bool StringAndPortToIPEndPoint(const std::string& ip_address_str,
+                                        int port,
+                                        net::IPEndPoint* ip_end_point);
+  static void IPEndPointToStringAndPort(const net::IPEndPoint& address,
+                                        std::string* ip_address_str,
+                                        int* port);
+
+ 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_;
+  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  // EXTENSIONS_BROWSER_API_SOCKET_SOCKET_H_