chrome.bluetooth API improvements.

device/bluetooth/bluetooth_socket_win.h

Index: device/bluetooth/bluetooth_socket_win.h
diff --git a/device/bluetooth/bluetooth_socket_win.h b/device/bluetooth/bluetooth_socket_win.h
index 41713e713097d7c7c377ddaee04cebe7a8f4cd60..94f3df213edca7a4adfe30cfa2e782ccd94b3136 100644
--- a/device/bluetooth/bluetooth_socket_win.h
+++ b/device/bluetooth/bluetooth_socket_win.h
@@ -7,42 +7,123 @@
 
 #include <WinSock2.h>
 
+#include <queue>
 #include <string>
 
+#include "base/memory/linked_ptr.h"
 #include "base/memory/ref_counted.h"
+#include "base/threading/thread_checker.h"
+#include "device/bluetooth/bluetooth_service_record_win.h"
 #include "device/bluetooth/bluetooth_socket.h"
+#include "net/base/net_log.h"
+#include "net/socket/tcp_socket.h"
 
 namespace net {
-
-class DrainableIOBuffer;
-class GrowableIOBuffer;
-
+class IOBuffer;
+class IOBufferWithSize;
 }  // namespace net
 
 namespace device {
 
 class BluetoothServiceRecord;
+class BluetoothSocketThreadWin;
 
-// This class is an implementation of BluetoothSocket class for Windows
-// platform.
+// This class is an implementation of BluetoothSocket class for the Windows
+// platform.  All public methods (including the factory method) must be called
+// on the UI thread, while underlying socket operations are performed on a
+// separated thread.
 class BluetoothSocketWin : public BluetoothSocket {
  public:
-  static scoped_refptr<BluetoothSocket> CreateBluetoothSocket(
-      const BluetoothServiceRecord& service_record);
-
-  // BluetoothSocket override
-  virtual bool Receive(net::GrowableIOBuffer* buffer) OVERRIDE;
-  virtual bool Send(net::DrainableIOBuffer* buffer) OVERRIDE;
-  virtual std::string GetLastErrorMessage() const OVERRIDE;
+  static scoped_refptr<BluetoothSocketWin> CreateBluetoothSocket(
+      const BluetoothServiceRecord& service_record,
+      scoped_refptr<base::SequencedTaskRunner> ui_task_runner,
+      scoped_refptr<BluetoothSocketThreadWin> socket_thread,
+      net::NetLog* net_log,
+      const net::NetLog::Source& source);
+
+  // Connect to the peer device and calls |success_callback| when the
+  // connection has been established successfully. If an error occurs, calls
+  // |error_callback| with a system error message.
+  void Connect(const base::Closure& success_callback,
+               const ErrorCompletionCallback& error_callback);
+
+  // Overriden from BluetoothSocket:
+  virtual void Close() OVERRIDE;
+
+  virtual void Disconnect(const base::Closure& callback) OVERRIDE;
+
+  virtual void Receive(int count,
+                       const ReceiveCompletionCallback& success_callback,
+                       const ReceiveErrorCompletionCallback& error_callback)
+      OVERRIDE;
+  virtual void Send(scoped_refptr<net::IOBuffer> buffer,
+                    int buffer_size,
+                    const SendCompletionCallback& success_callback,
+                    const ErrorCompletionCallback& error_callback) OVERRIDE;
 
  protected:
   virtual ~BluetoothSocketWin();
 
  private:
-  explicit BluetoothSocketWin(SOCKET fd);
-
-  const SOCKET fd_;
-  std::string error_message_;
+  struct WriteRequest {
+    scoped_refptr<net::IOBuffer> buffer;
+    int buffer_size;
+    SendCompletionCallback success_callback;
+    ErrorCompletionCallback error_callback;
+  };
+
+  BluetoothSocketWin(scoped_refptr<base::SequencedTaskRunner> ui_task_runner,
+                     scoped_refptr<BluetoothSocketThreadWin> socket_thread,
+                     net::NetLog* net_log,
+                     const net::NetLog::Source& source);
+
+  void DoClose();
+  void DoConnect(const base::Closure& success_callback,
+                 const ErrorCompletionCallback& error_callback);
+  void DoDisconnect(const base::Closure& callback);
+  void DoReceive(int count,
+                 const ReceiveCompletionCallback& success_callback,
+                 const ReceiveErrorCompletionCallback& error_callback);
+  void DoSend(scoped_refptr<net::IOBuffer> buffer,
+              int buffer_size,
+              const SendCompletionCallback& success_callback,
+              const ErrorCompletionCallback& error_callback);
+
+  void PostSuccess(const base::Closure& callback);
+  void PostErrorCompletion(const ErrorCompletionCallback& callback,
+                           const std::string& error);
+  void PostReceiveCompletion(const ReceiveCompletionCallback& callback,
+                             int io_buffer_size,
+                             scoped_refptr<net::IOBuffer> io_buffer);
+  void PostReceiveErrorCompletion(
+      const ReceiveErrorCompletionCallback& callback,
+      ErrorReason reason,
+      const std::string& error_message);
+  void PostSendCompletion(const SendCompletionCallback& callback,
+                          int bytes_written);
+
+  void SendFrontWriteRequest();
+  void OnSocketWriteComplete(const SendCompletionCallback& success_callback,
+                             const ErrorCompletionCallback& error_callback,
+                             int net_status);
+  void OnSocketReadComplete(
+      const ReceiveCompletionCallback& success_callback,
+      const ReceiveErrorCompletionCallback& error_callback,
+      int send_result);
+
+  scoped_refptr<base::SequencedTaskRunner> ui_task_runner_;
+  scoped_refptr<BluetoothSocketThreadWin> socket_thread_;
+  net::NetLog* net_log_;
+  const net::NetLog::Source source_;
+  std::string device_address_;
+  bool supports_rfcomm_;
+  uint8 rfcomm_channel_;
+  BTH_ADDR bth_addr_;
+  scoped_ptr<net::TCPSocket> tcp_socket_;
+  // Queue of pending writes. The buffer at the front of the queue is the one
+  // being written.
+  std::queue<linked_ptr<WriteRequest> > write_queue_;
+  scoped_refptr<net::IOBufferWithSize> read_buffer_;
 
   DISALLOW_COPY_AND_ASSIGN(BluetoothSocketWin);
 };