[Android WebView] Implement initial settings and callback support for Service Workers.

android_webview/browser/net/aw_request_interceptor.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 "android_webview/browser/net/aw_request_interceptor.h"
 
 #include <utility>
 
 #include "android_webview/browser/aw_contents_io_thread_client.h"
 #include "android_webview/browser/input_stream.h"
@@ skipping 96 matching lines @@
     }
   }
 
   scoped_ptr<AwContentsIoThreadClient> io_thread_client_;
   Callback callback_;
   base::WeakPtrFactory<ShouldInterceptRequestAdaptor> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ShouldInterceptRequestAdaptor);
 };
 
+scoped_ptr<AwContentsIoThreadClient> GetCorrespondingIoThreadClient(
+    net::URLRequest* request) {
+  int render_process_id, render_frame_id;
+  if (!content::ResourceRequestInfo::GetRenderFrameForRequest(
+      request, &render_process_id, &render_frame_id)) {
+    // When there is no associated render frame, or the frame_id is
+    // invalid we assume the request come from a service worker.
+    // TODO: add a more explicit service worker check.
+    return AwContentsIoThreadClient::GetServiceWorkerIoThreadClient();
+  }
+
+  scoped_ptr<AwContentsIoThreadClient> io_thread_client =
+      AwContentsIoThreadClient::FromID(render_process_id, render_frame_id);
+
+  if (!io_thread_client) {
+    // This means the frame id is invalid/not set, this currently happens
+    // with e.g. fetch requests coming from service workers.
+    return AwContentsIoThreadClient::GetServiceWorkerIoThreadClient();
+  }
+
+  return io_thread_client;
+}
+
 }  // namespace
 
 AwRequestInterceptor::AwRequestInterceptor() {}
 
 AwRequestInterceptor::~AwRequestInterceptor() {}
 
 net::URLRequestJob* AwRequestInterceptor::MaybeInterceptRequest(
     net::URLRequest* request,
     net::NetworkDelegate* network_delegate) const {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
 
   // MaybeInterceptRequest can be called multiple times for the same request.
   if (request->GetUserData(kRequestAlreadyHasJobDataKey))
     return nullptr;
 
-  int render_process_id, render_frame_id;
-  if (!content::ResourceRequestInfo::GetRenderFrameForRequest(
-          request, &render_process_id, &render_frame_id)) {
-    return nullptr;
-  }
-
   scoped_ptr<AwContentsIoThreadClient> io_thread_client =
-      AwContentsIoThreadClient::FromID(render_process_id, render_frame_id);
+      GetCorrespondingIoThreadClient(request);
 
   if (!io_thread_client)
     return nullptr;
 
   GURL referrer(request->referrer());
   if (referrer.is_valid() &&
       (!request->is_pending() || request->is_redirecting())) {
     request->SetExtraRequestHeaderByName(net::HttpRequestHeaders::kReferer,
                                          referrer.spec(), true);
   }
   request->SetUserData(kRequestAlreadyHasJobDataKey,
                        new base::SupportsUserData::Data());
   return new AndroidStreamReaderURLRequestJob(
       request, network_delegate,
       make_scoped_ptr(
           new ShouldInterceptRequestAdaptor(std::move(io_thread_client))),
       true);
 }
 
 }  // namespace android_webview