[chromedriver] Write websocket client and sync websocket client.

chrome/test/chromedriver/net/sync_websocket.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_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_H_
+#define CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_H_
+
+#include <list>
+#include <string>
+
+#include "base/basictypes.h"
+#include "base/compiler_specific.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/synchronization/condition_variable.h"
+#include "base/synchronization/lock.h"
+#include "chrome/test/chromedriver/net/websocket.h"
+#include "net/base/completion_callback.h"
+#include "net/socket_stream/socket_stream.h"
+
+namespace base {
+class WaitableEvent;
+}
+
+namespace net {
+class URLRequestContextGetter;
+}
+
+class GURL;
+
+// Proxy for using a WebSocket running on a background thread synchronously.
+class SyncWebSocket {
+ public:
+  explicit SyncWebSocket(net::URLRequestContextGetter* context_getter);
+  virtual ~SyncWebSocket();
+
+  // Connects to the WebSocket server. Returns true on success.
+  bool Connect(const GURL& url);
+
+  // Sends message. Returns true on success.
+  bool Send(const std::string& message);
+
+  // Receives next message. Blocks until at least one message is received or
+  // the socket is closed. Returns true on success and modifies |message|.
+  bool ReceiveNextMessage(std::string* message);
+
+ private:
+  struct CoreTraits;
+  class Core : public WebSocketListener,
+               public base::RefCountedThreadSafe<Core, CoreTraits> {
+   public:
+    explicit Core(net::URLRequestContextGetter* context_getter);
+
+    bool Connect(const GURL& url);
+
+    bool Send(const std::string& message);
+
+    bool ReceiveNextMessage(std::string* message);
+
+    // Overriden from WebSocketListener:
+    virtual void OnMessageReceived(const std::string& message) OVERRIDE;
+    virtual void OnClose() OVERRIDE;
+
+   private:
+    friend class base::RefCountedThreadSafe<Core, CoreTraits>;
+    friend class base::DeleteHelper<Core>;
+    friend struct CoreTraits;
+
+    virtual ~Core();
+
+    void ConnectOnIO(const GURL& url,
+                     bool* success,
+                     base::WaitableEvent* event);
+    void OnConnectCompletedOnIO(bool* connected,
+                                base::WaitableEvent* event,
+                                int error);
+    void SendOnIO(const std::string& message,
+                  bool* result,
+                  base::WaitableEvent* event);
+
+    // OnDestruct is meant to ensure deletion on the IO thread.
+    void OnDestruct() const;
+
+    scoped_refptr<net::URLRequestContextGetter> context_getter_;
+
+    // Only accessed on IO thread.
+    scoped_ptr<WebSocket> socket_;
+
+    base::Lock lock_;
+
+    // Protected by |lock_|.
+    bool closed_;
+
+    // Protected by |lock_|.
+    std::list<std::string> received_queue_;
+
+    // Protected by |lock_|.
+    // Signaled when the socket closes or a message is received.
+    base::ConditionVariable on_update_event_;
+  };
+
+  scoped_refptr<Core> core_;
+};
+
+struct SyncWebSocket::CoreTraits {
+  static void Destruct(const SyncWebSocket::Core* core) {
+    core->OnDestruct();
+  }
+};
+
+#endif  // CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_H_