WebSocket implementation in Chromium.

chrome/browser/renderer_host/socket_stream_host.h

+// Copyright (c) 2009 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_RENDERER_HOST_SOCKET_STREAM_HOST_H_
+#define CHROME_BROWSER_RENDERER_HOST_SOCKET_STREAM_HOST_H_
+
+#include <vector>
+
+#include "base/ref_counted.h"
+#include "net/socket_stream/socket_stream.h"
+
+class GURL;
+
+// Host of SocketStreamHandle.
+// Each SocketStramHandle will have an unique socket_id assigned by
+// SocketStreamHost constructor. If socket id is chrome_common_net::kNoSocketId,
+// there is no SocketStreamHost.
+// Each SocketStreamHost has SocketStream to manage bi-directional
+// communication over socket stream.
+// The lifetime of an instance of this class is completely controlled by the
+// SocketStreamDispatcherHost.
+class SocketStreamHost {
+  public:
+    SocketStreamHost(net::SocketStream::Delegate* delegate,
+                     int socket_id);
+    ~SocketStreamHost();
+
+    // Get socket_id associated with |socket|.
+    static int SocketIdFromSocketStream(net::SocketStream* socket);
+
+    int socket_id() const { return socket_id_; }
+
+    // Start to open connection to |url|.
+    void Connect(const GURL& url);
+
+    // Send |data| over the socket stream.
+    // socket stream must be open to send data.
+    // Returns true if the data is put in transmit buffer in socket stream.
+    // Returns false otherwise (transmit buffer exceeds limit, or socket
+    // stream is closed).
+    bool SendData(const std::vector<char>& data);
+
+    // Close the socket stream.
+    void Close();
+
+  private:
+    net::SocketStream::Delegate* delegate_;
+    int socket_id_;
+
+    scoped_refptr<net::SocketStream> socket_;
+
+    DISALLOW_COPY_AND_ASSIGN(SocketStreamHost);
+};
+
+#endif  // CHROME_BROWSER_RENDERER_HOST_SOCKET_STREAM_HOST_H_