// Copyright (c) 2010 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_HTTP_HTTP_NETWORK_SESSION_H_
#define NET_HTTP_HTTP_NETWORK_SESSION_H_
#pragma once

#include <map>
#include "base/non_thread_safe.h"
#include "base/ref_counted.h"
#include "base/scoped_ptr.h"
#include "net/base/host_port_pair.h"
#include "net/base/host_resolver.h"
#include "net/base/ssl_client_auth_cache.h"
#include "net/base/ssl_config_service.h"
#include "net/http/http_alternate_protocols.h"
#include "net/http/http_auth_cache.h"
#include "net/http/http_network_delegate.h"
#include "net/http/http_network_transaction.h"
#include "net/http/http_proxy_client_socket_pool.h"
#include "net/proxy/proxy_service.h"
#include "net/socket/client_socket_pool_histograms.h"
#include "net/socket/socks_client_socket_pool.h"
#include "net/socket/ssl_client_socket_pool.h"
#include "net/socket/tcp_client_socket_pool.h"
#include "net/spdy/spdy_settings_storage.h"

namespace net {

class ClientSocketFactory;
class HttpAuthHandlerFactory;
class HttpNetworkDelegate;
class HttpNetworkSessionPeer;
class SpdySessionPool;

// This class holds session objects used by HttpNetworkTransaction objects.
class HttpNetworkSession : public base::RefCounted<HttpNetworkSession>,
                           public NonThreadSafe {
 public:
  HttpNetworkSession(
      HostResolver* host_resolver,
      ProxyService* proxy_service,
      ClientSocketFactory* client_socket_factory,
      SSLConfigService* ssl_config_service,
      SpdySessionPool* spdy_session_pool,
      HttpAuthHandlerFactory* http_auth_handler_factory,
      HttpNetworkDelegate* network_delegate,
      NetLog* net_log);

  HttpAuthCache* auth_cache() { return &auth_cache_; }
  SSLClientAuthCache* ssl_client_auth_cache() {
    return &ssl_client_auth_cache_;
  }

  const HttpAlternateProtocols& alternate_protocols() const {
    return alternate_protocols_;
  }
  HttpAlternateProtocols* mutable_alternate_protocols() {
    return &alternate_protocols_;
  }

  // Access to the SpdySettingsStorage
  const SpdySettingsStorage& spdy_settings() const {
    return spdy_settings_;
  }
  SpdySettingsStorage* mutable_spdy_settings() {
    return &spdy_settings_;
  }

  // TCP sockets come from the tcp_socket_pool().
  const scoped_refptr<TCPClientSocketPool>& tcp_socket_pool() {
    return tcp_socket_pool_;
  }

  const scoped_refptr<SSLClientSocketPool>& ssl_socket_pool() {
    return ssl_socket_pool_;
  }

  const scoped_refptr<SOCKSClientSocketPool>& GetSocketPoolForSOCKSProxy(
      const HostPortPair& socks_proxy);

  const scoped_refptr<HttpProxyClientSocketPool>& GetSocketPoolForHTTPProxy(
      const HostPortPair& http_proxy);

  const scoped_refptr<SSLClientSocketPool>& GetSocketPoolForSSLWithProxy(
      const HostPortPair& proxy_server);

  // SSL sockets come from the socket_factory().
  ClientSocketFactory* socket_factory() { return socket_factory_; }
  HostResolver* host_resolver() { return host_resolver_; }
  ProxyService* proxy_service() { return proxy_service_; }
  SSLConfigService* ssl_config_service() { return ssl_config_service_; }
  const scoped_refptr<SpdySessionPool>& spdy_session_pool() {
    return spdy_session_pool_;
  }
  HttpAuthHandlerFactory* http_auth_handler_factory() {
    return http_auth_handler_factory_;
  }
  HttpNetworkDelegate* network_delegate() {
    return network_delegate_;
  }

  static int max_sockets_per_group();
  static void set_max_sockets_per_group(int socket_count);
  static void set_max_sockets_per_proxy_server(int socket_count);

#ifdef UNIT_TEST
  void FlushSocketPools() {
    if (ssl_socket_pool_.get())
      ssl_socket_pool_->Flush();
    if (tcp_socket_pool_.get())
      tcp_socket_pool_->Flush();

    for (SSLSocketPoolMap::const_iterator it =
        ssl_socket_pools_for_proxies_.begin();
        it != ssl_socket_pools_for_proxies_.end();
        it++)
      it->second->Flush();

    for (SOCKSSocketPoolMap::const_iterator it =
        socks_socket_pools_.begin();
        it != socks_socket_pools_.end();
        it++)
      it->second->Flush();

    for (HTTPProxySocketPoolMap::const_iterator it =
        http_proxy_socket_pools_.begin();
        it != http_proxy_socket_pools_.end();
        it++)
      it->second->Flush();
  }
#endif

 private:
  typedef std::map<HostPortPair, scoped_refptr<HttpProxyClientSocketPool> >
      HTTPProxySocketPoolMap;
  typedef std::map<HostPortPair, scoped_refptr<SOCKSClientSocketPool> >
      SOCKSSocketPoolMap;
  typedef std::map<HostPortPair, scoped_refptr<SSLClientSocketPool> >
      SSLSocketPoolMap;

  friend class base::RefCounted<HttpNetworkSession>;
  friend class HttpNetworkSessionPeer;

  ~HttpNetworkSession();

  HttpAuthCache auth_cache_;
  SSLClientAuthCache ssl_client_auth_cache_;
  HttpAlternateProtocols alternate_protocols_;
  scoped_refptr<ClientSocketPoolHistograms> tcp_pool_histograms_;
  scoped_refptr<ClientSocketPoolHistograms> tcp_for_http_proxy_pool_histograms_;
  scoped_refptr<ClientSocketPoolHistograms> http_proxy_pool_histograms_;
  scoped_refptr<ClientSocketPoolHistograms> tcp_for_socks_pool_histograms_;
  scoped_refptr<ClientSocketPoolHistograms> socks_pool_histograms_;
  scoped_refptr<ClientSocketPoolHistograms> ssl_pool_histograms_;
  scoped_refptr<TCPClientSocketPool> tcp_socket_pool_;
  scoped_refptr<SSLClientSocketPool> ssl_socket_pool_;
  HTTPProxySocketPoolMap http_proxy_socket_pools_;
  SOCKSSocketPoolMap socks_socket_pools_;
  SSLSocketPoolMap ssl_socket_pools_for_proxies_;
  ClientSocketFactory* socket_factory_;
  scoped_refptr<HostResolver> host_resolver_;
  scoped_refptr<ProxyService> proxy_service_;
  scoped_refptr<SSLConfigService> ssl_config_service_;
  scoped_refptr<SpdySessionPool> spdy_session_pool_;
  HttpAuthHandlerFactory* http_auth_handler_factory_;
  HttpNetworkDelegate* const network_delegate_;
  NetLog* net_log_;
  SpdySettingsStorage spdy_settings_;
};

}  // namespace net

#endif  // NET_HTTP_HTTP_NETWORK_SESSION_H_