| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/service_worker/link_header_support.h" | 5 #include "content/browser/service_worker/link_header_support.h" |
| 6 | 6 |
| 7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
| 8 #include "base/strings/string_split.h" | 8 #include "base/strings/string_split.h" |
| 9 #include "base/strings/string_util.h" | 9 #include "base/strings/string_util.h" |
| 10 #include "components/link_header_util/link_header_util.h" | 10 #include "components/link_header_util/link_header_util.h" |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 | 27 |
| 28 namespace { | 28 namespace { |
| 29 | 29 |
| 30 void RegisterServiceWorkerFinished(int64_t trace_id, bool result) { | 30 void RegisterServiceWorkerFinished(int64_t trace_id, bool result) { |
| 31 TRACE_EVENT_ASYNC_END1("ServiceWorker", | 31 TRACE_EVENT_ASYNC_END1("ServiceWorker", |
| 32 "LinkHeaderResourceThrottle::HandleServiceWorkerLink", | 32 "LinkHeaderResourceThrottle::HandleServiceWorkerLink", |
| 33 trace_id, "Success", result); | 33 trace_id, "Success", result); |
| 34 } | 34 } |
| 35 | 35 |
| 36 void HandleServiceWorkerLink( | 36 void HandleServiceWorkerLink( |
| 37 const net::URLRequest* request, | 37 net::URLRequest* request, |
| 38 const std::string& url, | 38 const std::string& url, |
| 39 const std::unordered_map<std::string, base::Optional<std::string>>& params, | 39 const std::unordered_map<std::string, base::Optional<std::string>>& params, |
| 40 ServiceWorkerContextWrapper* service_worker_context_for_testing) { | 40 ServiceWorkerContextWrapper* service_worker_context_for_testing) { |
| 41 DCHECK_CURRENTLY_ON(BrowserThread::IO); | 41 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 42 | 42 |
| 43 if (!base::CommandLine::ForCurrentProcess()->HasSwitch( | 43 if (!base::CommandLine::ForCurrentProcess()->HasSwitch( |
| 44 switches::kEnableExperimentalWebPlatformFeatures) && | 44 switches::kEnableExperimentalWebPlatformFeatures) && |
| 45 !TrialTokenValidator::RequestEnablesFeature(request, "ForeignFetch")) { | 45 !TrialTokenValidator::RequestEnablesFeature(request, "ForeignFetch")) { |
| 46 // TODO(mek): Log attempt to use without having correct token? | 46 // TODO(mek): Log attempt to use without having correct token? |
| 47 return; | 47 return; |
| 48 } | 48 } |
| 49 | 49 |
| 50 if (ContainsKey(params, "anchor")) | 50 if (ContainsKey(params, "anchor")) |
| 51 return; | 51 return; |
| 52 | 52 |
| 53 const ResourceRequestInfoImpl* request_info = | 53 ResourceRequestInfoImpl* request_info = |
| 54 ResourceRequestInfoImpl::ForRequest(request); | 54 ResourceRequestInfoImpl::ForRequest(request); |
| 55 ResourceMessageFilter* filter = request_info->filter(); | |
| 56 ServiceWorkerContext* service_worker_context = | 55 ServiceWorkerContext* service_worker_context = |
| 57 filter ? filter->service_worker_context() | 56 service_worker_context_for_testing |
| 58 : service_worker_context_for_testing; | 57 ? service_worker_context_for_testing |
| 59 if (IsBrowserSideNavigationEnabled() && | 58 : request_info->requester_info().service_worker_context(); |
| 60 ServiceWorkerUtils::IsMainResourceType(request_info->GetResourceType()) && | |
| 61 !service_worker_context) { | |
| 62 service_worker_context = request_info->service_worker_context(); | |
| 63 } | |
| 64 | 59 |
| 65 if (!service_worker_context) | 60 if (!service_worker_context) |
| 66 return; | 61 return; |
| 67 | 62 |
| 68 ServiceWorkerProviderHost* provider_host = | 63 ServiceWorkerProviderHost* provider_host = |
| 69 ServiceWorkerRequestHandler::GetProviderHost(request); | 64 ServiceWorkerRequestHandler::GetProviderHost(request); |
| 70 | 65 |
| 71 // If fetched from a service worker, make sure fetching service worker is | 66 // If fetched from a service worker, make sure fetching service worker is |
| 72 // controlling at least one client to prevent a service worker from spawning | 67 // controlling at least one client to prevent a service worker from spawning |
| 73 // new service workers in the background. | 68 // new service workers in the background. |
| (...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 126 static int64_t trace_id = 0; | 121 static int64_t trace_id = 0; |
| 127 TRACE_EVENT_ASYNC_BEGIN2( | 122 TRACE_EVENT_ASYNC_BEGIN2( |
| 128 "ServiceWorker", "LinkHeaderResourceThrottle::HandleServiceWorkerLink", | 123 "ServiceWorker", "LinkHeaderResourceThrottle::HandleServiceWorkerLink", |
| 129 ++trace_id, "Pattern", scope_url.spec(), "Script URL", script_url.spec()); | 124 ++trace_id, "Pattern", scope_url.spec(), "Script URL", script_url.spec()); |
| 130 service_worker_context->RegisterServiceWorker( | 125 service_worker_context->RegisterServiceWorker( |
| 131 scope_url, script_url, | 126 scope_url, script_url, |
| 132 base::Bind(&RegisterServiceWorkerFinished, trace_id)); | 127 base::Bind(&RegisterServiceWorkerFinished, trace_id)); |
| 133 } | 128 } |
| 134 | 129 |
| 135 void ProcessLinkHeaderValueForRequest( | 130 void ProcessLinkHeaderValueForRequest( |
| 136 const net::URLRequest* request, | 131 net::URLRequest* request, |
| 137 std::string::const_iterator value_begin, | 132 std::string::const_iterator value_begin, |
| 138 std::string::const_iterator value_end, | 133 std::string::const_iterator value_end, |
| 139 ServiceWorkerContextWrapper* service_worker_context_for_testing) { | 134 ServiceWorkerContextWrapper* service_worker_context_for_testing) { |
| 140 DCHECK_CURRENTLY_ON(BrowserThread::IO); | 135 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 141 | 136 |
| 142 std::string url; | 137 std::string url; |
| 143 std::unordered_map<std::string, base::Optional<std::string>> params; | 138 std::unordered_map<std::string, base::Optional<std::string>> params; |
| 144 if (!link_header_util::ParseLinkHeaderValue(value_begin, value_end, &url, | 139 if (!link_header_util::ParseLinkHeaderValue(value_begin, value_end, &url, |
| 145 ¶ms)) | 140 ¶ms)) |
| 146 return; | 141 return; |
| 147 | 142 |
| 148 auto rel_param = params.find("rel"); | 143 auto rel_param = params.find("rel"); |
| 149 if (rel_param == params.end() || !rel_param->second) | 144 if (rel_param == params.end() || !rel_param->second) |
| 150 return; | 145 return; |
| 151 for (const auto& rel : base::SplitStringPiece(rel_param->second.value(), | 146 for (const auto& rel : base::SplitStringPiece(rel_param->second.value(), |
| 152 HTTP_LWS, base::TRIM_WHITESPACE, | 147 HTTP_LWS, base::TRIM_WHITESPACE, |
| 153 base::SPLIT_WANT_NONEMPTY)) { | 148 base::SPLIT_WANT_NONEMPTY)) { |
| 154 if (base::EqualsCaseInsensitiveASCII(rel, "serviceworker")) | 149 if (base::EqualsCaseInsensitiveASCII(rel, "serviceworker")) |
| 155 HandleServiceWorkerLink(request, url, params, | 150 HandleServiceWorkerLink(request, url, params, |
| 156 service_worker_context_for_testing); | 151 service_worker_context_for_testing); |
| 157 } | 152 } |
| 158 } | 153 } |
| 159 | 154 |
| 160 } // namespace | 155 } // namespace |
| 161 | 156 |
| 162 void ProcessRequestForLinkHeaders(const net::URLRequest* request) { | 157 void ProcessRequestForLinkHeaders(net::URLRequest* request) { |
| 163 std::string link_header; | 158 std::string link_header; |
| 164 request->GetResponseHeaderByName("link", &link_header); | 159 request->GetResponseHeaderByName("link", &link_header); |
| 165 if (link_header.empty()) | 160 if (link_header.empty()) |
| 166 return; | 161 return; |
| 167 | 162 |
| 168 ProcessLinkHeaderForRequest(request, link_header); | 163 ProcessLinkHeaderForRequest(request, link_header); |
| 169 } | 164 } |
| 170 | 165 |
| 171 void ProcessLinkHeaderForRequest( | 166 void ProcessLinkHeaderForRequest( |
| 172 const net::URLRequest* request, | 167 net::URLRequest* request, |
| 173 const std::string& link_header, | 168 const std::string& link_header, |
| 174 ServiceWorkerContextWrapper* service_worker_context_for_testing) { | 169 ServiceWorkerContextWrapper* service_worker_context_for_testing) { |
| 175 for (const auto& value : link_header_util::SplitLinkHeader(link_header)) { | 170 for (const auto& value : link_header_util::SplitLinkHeader(link_header)) { |
| 176 ProcessLinkHeaderValueForRequest(request, value.first, value.second, | 171 ProcessLinkHeaderValueForRequest(request, value.first, value.second, |
| 177 service_worker_context_for_testing); | 172 service_worker_context_for_testing); |
| 178 } | 173 } |
| 179 } | 174 } |
| 180 | 175 |
| 181 } // namespace content | 176 } // namespace content |
| OLD | NEW |