Plumbing for safe browsing reporting for Android WebView.

android_webview/browser/aw_safe_browsing_ui_manager.cc

 // Copyright 2017 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/aw_safe_browsing_ui_manager.h"
 
 #include "android_webview/browser/aw_safe_browsing_blocking_page.h"
+#include "android_webview/browser/net/aw_url_request_context_getter.h"
+#include "android_webview/common/aw_paths.h"
+#include "base/command_line.h"
+#include "base/path_service.h"
+#include "components/safe_browsing/base_ping_manager.h"
+#include "components/safe_browsing/base_ui_manager.h"
+#include "components/safe_browsing/browser/safe_browsing_url_request_context_getter.h"
+#include "components/safe_browsing/common/safebrowsing_constants.h"
+#include "components/safe_browsing/common/safebrowsing_switches.h"
 #include "content/public/browser/browser_thread.h"
 
 using content::BrowserThread;
 using content::WebContents;
 
+namespace {
+
+// TODO(timvolodine): consider unifying with SafeBrowsingService
+std::string GetProtocolConfigClientName() {
+  std::string client_name;
+#if defined(GOOGLE_CHROME_BUILD)
+  client_name = "googlechrome";
+#else
+  client_name = "chromium";

[Nate Fischer, 2017/06/08 22:21:47]

Do these names still make sense for webview? Not sure if this matters to Safe
Browsing team. If we change the names, we can remove the "OS_ANDROID" section
below.

[timvolodine, 2017/06/09 11:53:21]

good point, will try to discuss this during the meeting today.

+#endif
+
+// Mark client string to allow server to differentiate mobile.
+#if defined(OS_ANDROID)
+  client_name.append("-a");
+#endif
+
+  return client_name;
+}
+
+}  // namespace
+
 namespace android_webview {
 
-AwSafeBrowsingUIManager::AwSafeBrowsingUIManager() {
+AwSafeBrowsingUIManager::AwSafeBrowsingUIManager(
+    AwURLRequestContextGetter* browser_url_request_context_getter) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
+
+  // TODO(timvolodine): verify this is what we want regarding the directory.
+  base::FilePath user_data_dir;
+  bool result =
+      PathService::Get(android_webview::DIR_SAFE_BROWSING, &user_data_dir);
+  DCHECK(result);
+
+  url_request_context_getter_ =
+      new safe_browsing::SafeBrowsingURLRequestContextGetter(
+          browser_url_request_context_getter, user_data_dir);
 }
 
 AwSafeBrowsingUIManager::~AwSafeBrowsingUIManager() {}
 
 void AwSafeBrowsingUIManager::DisplayBlockingPage(
     const UnsafeResource& resource) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
   WebContents* web_contents = resource.web_contents_getter.Run();
   // Check the size of the view
@@ skipping 13 matching lines @@
 }
 
 int AwSafeBrowsingUIManager::GetErrorUiType(
     const UnsafeResource& resource) const {
   WebContents* web_contents = resource.web_contents_getter.Run();
   UIManagerClient* client = UIManagerClient::FromWebContents(web_contents);
   DCHECK(client);
   return client->GetErrorUiType();
 }
 
+void AwSafeBrowsingUIManager::SendSerializedThreatDetails(
+    const std::string& serialized) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+  if (!ping_manager_) {
+    // Lazy creation of ping manager, needs to happen on IO thread.
+    safe_browsing::SafeBrowsingProtocolConfig config;
+    config.client_name = GetProtocolConfigClientName();
+    base::CommandLine* cmdline = ::base::CommandLine::ForCurrentProcess();
+    config.disable_auto_update =
+        cmdline->HasSwitch(::safe_browsing::switches::kSbDisableAutoUpdate);
+    config.url_prefix = ::safe_browsing::kSbDefaultURLPrefix;
+    config.backup_connect_error_url_prefix =
+        ::safe_browsing::kSbBackupConnectErrorURLPrefix;
+    config.backup_http_error_url_prefix =
+        ::safe_browsing::kSbBackupHttpErrorURLPrefix;
+    config.backup_network_error_url_prefix =
+        ::safe_browsing::kSbBackupNetworkErrorURLPrefix;
+    ping_manager_ = ::safe_browsing::BasePingManager::Create(
+        url_request_context_getter_.get(), config);
+  }
+
+  if (!serialized.empty()) {
+    DVLOG(1) << "Sending serialized threat details";
+    ping_manager_->ReportThreatDetails(serialized);
+  }
+}
+
 }  // namespace android_webview