Wire up download attribution enable finch experiment

chrome/browser/safe_browsing/safe_browsing_navigation_observer_manager.cc

 // Copyright 2016 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 "chrome/browser/safe_browsing/safe_browsing_navigation_observer_manager.h"
 
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/strings/stringprintf.h"
 #include "base/time/time.h"
 #include "base/timer/timer.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/safe_browsing/safe_browsing_navigation_observer.h"
+#include "chrome/browser/safe_browsing/safe_browsing_service.h"
 #include "chrome/browser/sessions/session_tab_helper.h"
 #include "chrome/browser/tab_contents/retargeting_details.h"
+#include "chrome/common/pref_names.h"
+#include "components/prefs/pref_service.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 
 using content::WebContents;
 
 namespace safe_browsing {
@@ skipping 14 matching lines @@
 }  // namespace
 
 // The expiration period of a user gesture. Any user gesture that happened 1.0
 // second ago is considered as expired and not relevant to upcoming navigation
 // events.
 static const double kUserGestureTTLInSecond = 1.0;
 // The expiration period of navigation events and resolved IP addresses. Any
 // navigation related records that happened 2 minutes ago are considered as
 // expired. So we clean up these navigation footprints every 2 minutes.
 static const double kNavigationFootprintTTLInSecond = 120.0;
-// The number of user gestures we trace back for download attribution.
-static const int kDownloadAttributionUserGestureLimit = 2;
 
 // static
+const base::Feature
+SafeBrowsingNavigationObserverManager::kDownloadAttribution {
+    "DownloadAttribution", base::FEATURE_DISABLED_BY_DEFAULT
+};
+// static
 bool SafeBrowsingNavigationObserverManager::IsUserGestureExpired(
     const base::Time& timestamp) {
   return IsEventExpired(timestamp, kUserGestureTTLInSecond);
 }
 
 // static
 GURL SafeBrowsingNavigationObserverManager::ClearEmptyRef(const GURL& url) {
   if (url.has_ref() && url.ref().empty()) {
     url::Replacements<char> replacements;
     replacements.ClearRef();
     return url.ReplaceComponents(replacements);
   }
   return url;
 }
 
+// static
+bool SafeBrowsingNavigationObserverManager::IsEnabledAndReady(
+    Profile* profile) {
+  return base::FeatureList::IsEnabled(
+      SafeBrowsingNavigationObserverManager::kDownloadAttribution) &&
+      profile->GetPrefs()->GetBoolean(prefs::kSafeBrowsingEnabled) &&
+      g_browser_process->safe_browsing_service() &&
+      g_browser_process->safe_browsing_service()->navigation_observer_manager();
+}
+
 SafeBrowsingNavigationObserverManager::SafeBrowsingNavigationObserverManager() {
   registrar_.Add(this, chrome::NOTIFICATION_RETARGETING,
                  content::NotificationService::AllSources());
 
-  // TODO(jialiul): call ScheduleNextCleanUpAfterInterval() when this class is
-  // ready to be hooked into SafeBrowsingService.
+  // Schedule clean up in 2 minutes.
+  ScheduleNextCleanUpAfterInterval(
+      base::TimeDelta::FromSecondsD(kNavigationFootprintTTLInSecond));
 }
 
 void SafeBrowsingNavigationObserverManager::RecordNavigationEvent(
     const GURL& nav_event_key,
     NavigationEvent* nav_event) {
   auto insertion_result = navigation_map_.insert(
       std::make_pair(nav_event_key, std::vector<NavigationEvent>()));
 
   insertion_result.first->second.push_back(std::move(*nav_event));
 }
@@ skipping 116 matching lines @@
       AddToReferrerChain(out_referrer_chain, nav_event,
                          ReferrerChainEntry::LANDING_REFERRER);
       result = SUCCESS_LANDING_REFERRER;
     } else {
       NOTREACHED();
     }
   }
   return result;
 }
 
-void SafeBrowsingNavigationObserverManager::
-    AddReferrerChainToClientDownloadRequest(
-        const GURL& download_url,
-        content::WebContents* source_contents,
-        ClientDownloadRequest* out_request) {
-  int download_tab_id = SessionTabHelper::IdForTab(source_contents);
-  UMA_HISTOGRAM_BOOLEAN(
-      "SafeBrowsing.ReferrerHasInvalidTabID.DownloadAttribution",
-      download_tab_id == -1);
-  std::vector<ReferrerChainEntry> attribution_chain;
-  AttributionResult result = IdentifyReferrerChain(
-      download_url, download_tab_id, kDownloadAttributionUserGestureLimit,
-      &attribution_chain);
-  UMA_HISTOGRAM_COUNTS_100(
-      "SafeBrowsing.ReferrerURLChainSize.DownloadAttribution",
-      attribution_chain.size());
-  UMA_HISTOGRAM_ENUMERATION(
-      "SafeBrowsing.ReferrerAttributionResult.DownloadAttribution", result,
-      SafeBrowsingNavigationObserverManager::ATTRIBUTION_FAILURE_TYPE_MAX);
-  for (auto entry : attribution_chain)
-    *out_request->add_referrer_chain() = entry;
-}
-
 SafeBrowsingNavigationObserverManager::
     ~SafeBrowsingNavigationObserverManager() {}
 
 void SafeBrowsingNavigationObserverManager::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   if (type == chrome::NOTIFICATION_RETARGETING)
     RecordRetargeting(details);
 }
@@ skipping 151 matching lines @@
         nav_event->source_main_frame_url.spec());
   }
   referrer_chain_entry.set_is_retargeting(nav_event->source_tab_id !=
                                           nav_event->target_tab_id);
   referrer_chain_entry.set_navigation_time_msec(
       nav_event->last_updated.ToJavaTime());
   referrer_chain->push_back(referrer_chain_entry);
 }
 
 }  // namespace safe_browsing