Server push cancellation: change the ownership of ServerPushDelegate to HttpNetworkSession.

net/http/http_network_session.cc

 // 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.
 
 #include "net/http/http_network_session.h"
 
 #include <utility>
 
 #include "base/atomic_sequence_num.h"
 #include "base/compiler_specific.h"
@@ skipping 126 matching lines @@
 HttpNetworkSession::Params::~Params() {}
 
 // TODO(mbelshe): Move the socket factories into HttpStreamFactory.
 HttpNetworkSession::HttpNetworkSession(const Params& params)
     : net_log_(params.net_log),
       http_server_properties_(params.http_server_properties),
       cert_verifier_(params.cert_verifier),
       http_auth_handler_factory_(params.http_auth_handler_factory),
       proxy_service_(params.proxy_service),
       ssl_config_service_(params.ssl_config_service),
+      push_delegate_(nullptr),
       quic_stream_factory_(
           params.net_log,
           params.host_resolver,
           params.ssl_config_service,
           params.client_socket_factory
               ? params.client_socket_factory
               : ClientSocketFactory::GetDefaultFactory(),
           params.http_server_properties,
           params.proxy_delegate,
           params.cert_verifier,
@@ skipping 200 matching lines @@
       return true;
     case kProtoHTTP2:
       return params_.enable_http2;
     case kProtoQUIC:
       return params_.enable_quic;
   }
   NOTREACHED();
   return false;
 }
 
+void HttpNetworkSession::SetServerPushDelegate(
+    std::unique_ptr<ServerPushDelegate> push_delegate) {
+  DCHECK(!push_delegate_ && push_delegate);
+
+  push_delegate_ = std::move(push_delegate);
+  spdy_session_pool_.set_server_push_delegate(push_delegate_.get());
+  quic_stream_factory_.set_server_push_delegate(push_delegate_.get());
+}
+
 void HttpNetworkSession::GetAlpnProtos(NextProtoVector* alpn_protos) const {
   *alpn_protos = next_protos_;
 }
 
 void HttpNetworkSession::GetSSLConfig(const HttpRequestInfo& request,
                                       SSLConfig* server_config,
                                       SSLConfig* proxy_config) const {
   ssl_config_service_->GetSSLConfig(server_config);
   GetAlpnProtos(&server_config->alpn_protos);
   *proxy_config = *server_config;
@@ skipping 42 matching lines @@
       break;
     case base::MemoryState::SUSPENDED:
     // Note: Not supported at present. Fall through.
     case base::MemoryState::UNKNOWN:
       NOTREACHED();
       break;
   }
 }
 
 }  // namespace net