WebUI for WKWebView

ios/web/web_state/ui/crw_ui_web_view_web_controller.mm

 // Copyright 2014 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.
 
 #import "ios/web/web_state/ui/crw_ui_web_view_web_controller.h"
 
 #import "base/ios/weak_nsobject.h"
 #include "base/json/json_reader.h"
 #include "base/json/string_escape.h"
 #include "base/mac/bind_objc_block.h"
@@ skipping 513 matching lines @@
   NSString* documentType =
       [_uiWebView stringByEvaluatingJavaScriptFromString:
           @"'' + document"];
   if ([documentType isEqualToString:@"[object HTMLDocument]"])
     return web::WEB_VIEW_DOCUMENT_TYPE_HTML;
   else if ([documentType isEqualToString:@"[object Document]"])
     return web::WEB_VIEW_DOCUMENT_TYPE_GENERIC;
   return web::WEB_VIEW_DOCUMENT_TYPE_UNKNOWN;
 }
 
-- (void)loadWebRequest:(NSURLRequest*)request {
+- (void)loadRequest:(NSMutableURLRequest*)request {
+  DCHECK(web::GetWebClient());
+  GURL requestURL = net::GURLWithNSURL(request.URL);
+  // If the request is for WebUI, add information to let the network stack
+  // access the requestGroupID.
+  if (web::GetWebClient()->IsAppSpecificURL(requestURL)) {
+    // Sub requests of a chrome:// page will not contain the user agent.
+    // Instead use the username part of the URL to allow the network stack to
+    // associate a request to the correct tab.
+    request.URL = web::AddRequestGroupIDToURL(
+        request.URL, self.webStateImpl->GetRequestGroupID());
+  }
   [_uiWebView loadRequest:request];
 }
 
 - (void)loadWebHTMLString:(NSString*)html forURL:(const GURL&)URL {
   [_uiWebView loadHTMLString:html baseURL:net::NSURLWithGURL(URL)];
 }
 
 - (BOOL)scriptHasBeenInjectedForClass:(Class)jsInjectionManagerClass
                        presenceBeacon:(NSString*)beacon {
   NSString* beaconCheckScript = [NSString stringWithFormat:
@@ skipping 880 matching lines @@
 }
 
 #pragma mark -
 #pragma mark Testing methods
 
 -(id<CRWRecurringTaskDelegate>)recurringTaskDelegate {
   return _recurringTaskDelegate;
 }
 
 @end