Implement websocket throttling.

net/socket_stream/socket_stream.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 NET_SOCKET_STREAM_SOCKET_STREAM_H_
 #define NET_SOCKET_STREAM_SOCKET_STREAM_H_
 
 #include <deque>
 #include <map>
 #include <string>
@@ skipping 13 matching lines @@
 #include "net/socket/tcp_client_socket.h"
 #include "net/url_request/url_request_context.h"
 
 namespace net {
 
 class AuthChallengeInfo;
 class ClientSocketFactory;
 class HostResolver;
 class SSLConfigService;
 class SingleRequestHostResolver;
+class SocketStreamThrottle;
 
 // SocketStream is used to implement Web Sockets.
 // It provides plain full-duplex stream with proxy and SSL support.
 // For proxy authentication, only basic mechanisum is supported.  It will try
 // authentication identity for proxy URL first.  If server requires proxy
 // authentication, it will try authentication identity for realm that server
 // requests.
 class SocketStream : public base::RefCountedThreadSafe<SocketStream> {
  public:
   // Derive from this class and add your own data members to associate extra
@@ skipping 45 matching lines @@
   SocketStream(const GURL& url, Delegate* delegate);
 
   // The user data allows the clients to associate data with this job.
   // Multiple user data values can be stored under different keys.
   // This job will TAKE OWNERSHIP of the given data pointer, and will
   // delete the object if it is changed or the job is destroyed.
   UserData* GetUserData(const void* key) const;
   void SetUserData(const void* key, UserData* data);
 
   const GURL& url() const { return url_; }
+  const AddressList& address_list() const { return addresses_; }
   Delegate* delegate() const { return delegate_; }
   int max_pending_send_allowed() const { return max_pending_send_allowed_; }
 
   URLRequestContext* context() const { return context_.get(); }
   void set_context(URLRequestContext* context);
 
   // Opens the connection on the IO thread.
   // Once the connection is established, calls delegate's OnConnected.
   void Connect();
 
@@ skipping 75 matching lines @@
   enum ProxyMode {
     kDirectConnection,  // If using a direct connection
     kTunnelProxy,  // If using a tunnel (CONNECT method as HTTPS)
     kSOCKSProxy,  // If using a SOCKS proxy
   };
 
   typedef std::deque< scoped_refptr<IOBufferWithSize> > PendingDataQueue;
   friend class base::RefCountedThreadSafe<SocketStream>;
   ~SocketStream();
 
+  friend class WebSocketThrottleTest;
+
+  // Copies the given addrinfo list in |addresses_|.
+  // Used for WebSocketThrottleTest.
+  void CopyAddrInfo(struct addrinfo* head);
+
   // Finishes the job.
   // Calls OnError and OnClose of delegate, and no more
   // notifications will be sent to delegate.
   void Finish(int result);
 
   int DidEstablishConnection();
-  void DidReceiveData(int result);
-  void DidSendData(int result);
+  int DidReceiveData(int result);
+  int DidSendData(int result);
 
   void OnIOCompleted(int result);
   void OnReadCompleted(int result);
   void OnWriteCompleted(int result);
 
   void DoLoop(int result);
 
   int DoResolveProxy();
   int DoResolveProxyComplete(int result);
   int DoResolveHost();
@@ skipping 70 matching lines @@
   // |max_pending_send_allowed_|.
   // |write_buf_| holds requested data and |current_write_buf_| is used
   // for Write operation, that is, |current_write_buf_| is
   // |write_buf_| + |write_buf_offset_|.
   scoped_refptr<IOBuffer> write_buf_;
   scoped_refptr<DrainableIOBuffer> current_write_buf_;
   int write_buf_offset_;
   int write_buf_size_;
   PendingDataQueue pending_write_bufs_;
 
+  SocketStreamThrottle* throttle_;
+
   DISALLOW_COPY_AND_ASSIGN(SocketStream);
 };
 
 }  // namespace net
 
 #endif  // NET_SOCKET_STREAM_SOCKET_STREAM_H_