blob: 0467583eefa0f8ea73936c0adbadc51d79ad58b9 [file] [log] [blame]
// 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 NET_SPDY_SPDY_SESSION_POOL_H_
#define NET_SPDY_SPDY_SESSION_POOL_H_
#include <map>
#include <list>
#include <string>
#include "base/basictypes.h"
#include "base/gtest_prod_util.h"
#include "base/memory/ref_counted.h"
#include "net/base/cert_database.h"
#include "net/base/host_port_pair.h"
#include "net/base/ip_endpoint.h"
#include "net/base/net_errors.h"
#include "net/base/net_export.h"
#include "net/base/network_change_notifier.h"
#include "net/base/ssl_config_service.h"
#include "net/proxy/proxy_config.h"
#include "net/proxy/proxy_server.h"
#include "net/socket/next_proto.h"
namespace net {
class AddressList;
class BoundNetLog;
class ClientSocketHandle;
class HostResolver;
class HttpServerProperties;
class SpdySession;
namespace test_spdy2 {
class SpdySessionPoolPeer;
} // namespace test_spdy
namespace test_spdy3 {
class SpdySessionPoolPeer;
} // namespace test_spdy
// This is a very simple pool for open SpdySessions.
class NET_EXPORT SpdySessionPool
: public NetworkChangeNotifier::IPAddressObserver,
public SSLConfigService::Observer,
public CertDatabase::Observer {
public:
typedef base::TimeTicks (*TimeFunc)(void);
SpdySessionPool(HostResolver* host_resolver,
SSLConfigService* ssl_config_service,
HttpServerProperties* http_server_properties,
size_t max_sessions_per_domain,
bool force_single_domain,
bool enable_ip_pooling,
bool enable_credential_frames,
bool enable_compression,
bool enable_ping_based_connection_checking,
NextProto default_protocol,
size_t default_initial_recv_window_size,
size_t initial_max_concurrent_streams,
size_t max_concurrent_streams_limit,
SpdySessionPool::TimeFunc time_func,
const std::string& trusted_spdy_proxy);
virtual ~SpdySessionPool();
// Either returns an existing SpdySession or creates a new SpdySession for
// use.
scoped_refptr<SpdySession> Get(
const HostPortProxyPair& host_port_proxy_pair,
const BoundNetLog& net_log);
// Only returns a SpdySession if it already exists.
scoped_refptr<SpdySession> GetIfExists(
const HostPortProxyPair& host_port_proxy_pair,
const BoundNetLog& net_log);
// Builds a SpdySession from an existing SSL socket. Users should try
// calling Get() first to use an existing SpdySession so we don't get
// multiple SpdySessions per domain. Note that ownership of |connection| is
// transferred from the caller to the SpdySession.
// |certificate_error_code| is used to indicate the certificate error
// encountered when connecting the SSL socket. OK means there was no error.
// For testing, setting is_secure to false allows Spdy to connect with a
// pre-existing TCP socket.
// Returns OK on success, and the |spdy_session| will be provided.
// Returns an error on failure, and |spdy_session| will be NULL.
net::Error GetSpdySessionFromSocket(
const HostPortProxyPair& host_port_proxy_pair,
ClientSocketHandle* connection,
const BoundNetLog& net_log,
int certificate_error_code,
scoped_refptr<SpdySession>* spdy_session,
bool is_secure);
// TODO(willchan): Consider renaming to HasReusableSession, since perhaps we
// should be creating a new session. WARNING: Because of IP connection pooling
// using the HostCache, if HasSession() returns true at one point, it does not
// imply the SpdySessionPool will still have a matching session in the near
// future, since the HostCache's entry may have expired.
bool HasSession(const HostPortProxyPair& host_port_proxy_pair) const;
// Close all SpdySessions, including any new ones created in the process of
// closing the current ones.
void CloseAllSessions();
// Close only the currently existing SpdySessions with |error|.
// Let any new ones created continue to live.
void CloseCurrentSessions(net::Error error);
// Close only the idle SpdySessions.
void CloseIdleSessions();
// Removes a SpdySession from the SpdySessionPool. This should only be called
// by SpdySession, because otherwise session->state_ is not set to CLOSED.
void Remove(const scoped_refptr<SpdySession>& session);
// Creates a Value summary of the state of the spdy session pool. The caller
// responsible for deleting the returned value.
base::Value* SpdySessionPoolInfoToValue() const;
HttpServerProperties* http_server_properties() {
return http_server_properties_;
}
// NetworkChangeNotifier::IPAddressObserver methods:
// We flush all idle sessions and release references to the active ones so
// they won't get re-used. The active ones will either complete successfully
// or error out due to the IP address change.
virtual void OnIPAddressChanged() OVERRIDE;
// SSLConfigService::Observer methods:
// We perform the same flushing as described above when SSL settings change.
virtual void OnSSLConfigChanged() OVERRIDE;
// CertDatabase::Observer methods:
virtual void OnCertAdded(const X509Certificate* cert) OVERRIDE;
virtual void OnCertTrustChanged(const X509Certificate* cert) OVERRIDE;
private:
friend class test_spdy2::SpdySessionPoolPeer; // For testing.
friend class test_spdy3::SpdySessionPoolPeer; // For testing.
friend class SpdyNetworkTransactionSpdy2Test; // For testing.
friend class SpdyNetworkTransactionSpdy3Test; // For testing.
FRIEND_TEST_ALL_PREFIXES(SpdyNetworkTransactionSpdy2Test,
WindowUpdateOverflow);
FRIEND_TEST_ALL_PREFIXES(SpdyNetworkTransactionSpdy3Test,
WindowUpdateOverflow);
typedef std::list<scoped_refptr<SpdySession> > SpdySessionList;
typedef std::map<HostPortProxyPair, SpdySessionList*> SpdySessionsMap;
typedef std::map<IPEndPoint, HostPortProxyPair> SpdyAliasMap;
scoped_refptr<SpdySession> GetInternal(
const HostPortProxyPair& host_port_proxy_pair,
const BoundNetLog& net_log,
bool only_use_existing_sessions);
scoped_refptr<SpdySession> GetExistingSession(
SpdySessionList* list,
const BoundNetLog& net_log) const;
scoped_refptr<SpdySession> GetFromAlias(
const HostPortProxyPair& host_port_proxy_pair,
const BoundNetLog& net_log,
bool record_histograms) const;
// Helper functions for manipulating the lists.
const HostPortProxyPair& NormalizeListPair(
const HostPortProxyPair& host_port_proxy_pair) const;
SpdySessionList* AddSessionList(
const HostPortProxyPair& host_port_proxy_pair);
SpdySessionList* GetSessionList(
const HostPortProxyPair& host_port_proxy_pair) const;
void RemoveSessionList(const HostPortProxyPair& host_port_proxy_pair);
// Does a DNS cache lookup for |pair|, and returns the |addresses| found.
// Returns true if addresses found, false otherwise.
bool LookupAddresses(const HostPortProxyPair& pair,
const BoundNetLog& net_log,
AddressList* addresses) const;
// Add |address| as an IP-equivalent address for |pair|.
void AddAlias(const IPEndPoint& address, const HostPortProxyPair& pair);
// Remove all aliases for |pair| from the aliases table.
void RemoveAliases(const HostPortProxyPair& pair);
// Removes |session| from the session list associated with |pair|.
// Returns true if the session was removed, false otherwise.
bool RemoveFromSessionList(const scoped_refptr<SpdySession>& session,
const HostPortProxyPair& pair);
HttpServerProperties* const http_server_properties_;
// This is our weak session pool - one session per domain.
SpdySessionsMap sessions_;
// A map of IPEndPoint aliases for sessions.
SpdyAliasMap aliases_;
static bool g_force_single_domain;
const scoped_refptr<SSLConfigService> ssl_config_service_;
HostResolver* const resolver_;
// Defaults to true. May be controlled via SpdySessionPoolPeer for tests.
bool verify_domain_authentication_;
bool enable_sending_initial_settings_;
size_t max_sessions_per_domain_;
bool force_single_domain_;
bool enable_ip_pooling_;
bool enable_credential_frames_;
bool enable_compression_;
bool enable_ping_based_connection_checking_;
NextProto default_protocol_;
size_t initial_recv_window_size_;
size_t initial_max_concurrent_streams_;
size_t max_concurrent_streams_limit_;
TimeFunc time_func_;
// This SPDY proxy is allowed to push resources from origins that are
// different from those of their associated streams.
HostPortPair trusted_spdy_proxy_;
DISALLOW_COPY_AND_ASSIGN(SpdySessionPool);
};
} // namespace net
#endif // NET_SPDY_SPDY_SESSION_POOL_H_