Index: chrome/browser/net/connect_interceptor.cc |
diff --git a/chrome/browser/net/connect_interceptor.cc b/chrome/browser/net/connect_interceptor.cc |
index 494e4f57fb8e82f3b1b0e4cd6c8793297c3e9a5b..40bf61ce876094ae785fc5df95dd1ce7ec75706c 100644 |
--- a/chrome/browser/net/connect_interceptor.cc |
+++ b/chrome/browser/net/connect_interceptor.cc |
@@ -5,6 +5,8 @@ |
#include "chrome/browser/net/connect_interceptor.h" |
#include "chrome/browser/net/predictor.h" |
+#include "content/public/browser/resource_request_info.h" |
+#include "content/public/common/resource_type.h" |
#include "net/base/load_flags.h" |
#include "net/url_request/url_request.h" |
@@ -25,11 +27,22 @@ void ConnectInterceptor::WitnessURLRequest(net::URLRequest* request) { |
if (request_scheme_host == GURL::EmptyGURL()) |
return; |
+ const content::ResourceRequestInfo* info = |
+ content::ResourceRequestInfo::ForRequest(request); |
+ bool is_main_frame = false; |
+ bool is_sub_frame = false; |
+ // TODO(mmenke): Should the predictor really be fed requests without a |
+ // ResourceRequestInfo? |
+ if (info) { |
+ content::ResourceType resource_type = info->GetResourceType(); |
+ is_main_frame = (resource_type == content::RESOURCE_TYPE_MAIN_FRAME); |
+ is_sub_frame = (resource_type == content::RESOURCE_TYPE_SUB_FRAME); |
+ } |
+ |
// Learn what URLs are likely to be needed during next startup. |
predictor_->LearnAboutInitialNavigation(request_scheme_host); |
bool redirected_host = false; |
- bool is_subresource = !(request->load_flags() & net::LOAD_MAIN_FRAME); |
if (request->referrer().empty()) { |
if (request->url() != request->original_url()) { |
// This request was completed with a redirect. |
@@ -57,9 +70,10 @@ void ConnectInterceptor::WitnessURLRequest(net::URLRequest* request) { |
} else { |
GURL referring_scheme_host = GURL(request->referrer()).GetWithEmptyPath(); |
// Learn about our referring URL, for use in the future. |
- if (is_subresource && timed_cache_.WasRecentlySeen(referring_scheme_host)) |
+ if (!is_main_frame && timed_cache_.WasRecentlySeen(referring_scheme_host)) { |
predictor_->LearnFromNavigation(referring_scheme_host, |
request_scheme_host); |
+ } |
if (referring_scheme_host == request_scheme_host) { |
// We've already made any/all predictions when we navigated to the |
// referring host, so we can bail out here. |
@@ -74,9 +88,10 @@ void ConnectInterceptor::WitnessURLRequest(net::URLRequest* request) { |
// Subresources for main frames usually get predicted when we detected the |
// main frame request - way back in RenderViewHost::Navigate. So only handle |
// predictions now for subresources or for redirected hosts. |
- if ((request->load_flags() & net::LOAD_SUB_FRAME) || redirected_host) |
+ if (is_sub_frame || redirected_host) { |
predictor_->PredictFrameSubresources(request_scheme_host, |
request->first_party_for_cookies()); |
+ } |
return; |
} |