SafeBrowsing support for WebSocket (post-network-servicification)

android_webview/renderer/aw_content_renderer_client.h

Index: android_webview/renderer/aw_content_renderer_client.h
diff --git a/android_webview/renderer/aw_content_renderer_client.h b/android_webview/renderer/aw_content_renderer_client.h
index 4bfae8684ccea24b99a85af06ee07bb79041f7cb..6576d0eff3e396e455bc0e219d31d5c2602655dd 100644
--- a/android_webview/renderer/aw_content_renderer_client.h
+++ b/android_webview/renderer/aw_content_renderer_client.h
@@ -11,6 +11,7 @@
 #include "android_webview/renderer/aw_render_thread_observer.h"
 #include "base/compiler_specific.h"
 #include "base/memory/weak_ptr.h"
+#include "components/safe_browsing/common/safe_browsing.mojom.h"
 #include "components/spellcheck/spellcheck_build_features.h"
 #include "components/web_restrictions/interfaces/web_restrictions.mojom.h"
 #include "content/public/renderer/content_renderer_client.h"
@@ -46,6 +47,8 @@ class AwContentRendererClient : public content::ContentRendererClient {
   void AddSupportedKeySystems(
       std::vector<std::unique_ptr<::media::KeySystemProperties>>* key_systems)
       override;
+  std::unique_ptr<blink::WebSocketHandshakeThrottle>
+  CreateWebSocketHandshakeThrottle() override;
 
   bool HandleNavigation(content::RenderFrame* render_frame,
                         bool is_content_initiated,
@@ -58,9 +61,14 @@ class AwContentRendererClient : public content::ContentRendererClient {
   bool ShouldUseMediaPlayerForURL(const GURL& url) override;
 
  private:
+  // Returns |true| if we should use the SafeBrowsing mojo service. Initialises
+  // |safe_browsing_| on the first call as a side-effect.
+  bool UsingSafeBrowsingMojoService();
+
   std::unique_ptr<AwRenderThreadObserver> aw_render_thread_observer_;
   std::unique_ptr<visitedlink::VisitedLinkSlave> visited_link_slave_;
   web_restrictions::mojom::WebRestrictionsPtr web_restrictions_service_;
+  safe_browsing::mojom::SafeBrowsingPtr safe_browsing_;
 
 #if BUILDFLAG(ENABLE_SPELLCHECK)
   std::unique_ptr<SpellCheck> spellcheck_;