Send the referral URL with the client-side phishing detection request.

chrome/browser/safe_browsing/browser_feature_extractor.cc

 // Copyright (c) 2011 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/browser_feature_extractor.h"
 
 #include <map>
 #include <utility>
 
 #include "base/stl_util-inl.h"
@@ skipping 15 matching lines @@
 const char kUrlHistoryTypedCount[] = "UrlHistoryTypedCount";
 const char kUrlHistoryLinkCount[] = "UrlHistoryLinkCount";
 const char kUrlHistoryVisitCountMoreThan24hAgo[] =
     "UrlHistoryVisitCountMoreThan24hAgo";
 const char kHttpHostVisitCount[] = "HttpHostVisitCount";
 const char kHttpsHostVisitCount[] = "HttpsHostVisitCount";
 const char kFirstHttpHostVisitMoreThan24hAgo[] =
     "FirstHttpHostVisitMoreThan24hAgo";
 const char kFirstHttpsHostVisitMoreThan24hAgo[] =
     "FirstHttpsHostVisitMoreThan24hAgo";
+const char kHasSSLReferrer[] = "HasSSLReferrer";
+const char kPageTransitionType[] = "PageTransitionType";
 }  // namespace features
 
 static void AddFeature(const std::string& feature_name,
                        double feature_value,
                        ClientPhishingRequest* request) {
   DCHECK(request);
   ClientPhishingRequest::Feature* feature =
       request->add_non_model_feature_map();
   feature->set_name(feature_name);
   feature->set_value(feature_value);
@@ skipping 21 matching lines @@
     if (history) {
       history->CancelRequest(it->first);
     }
     ExtractionData& extraction = it->second;
     delete extraction.first;  // delete request
     delete extraction.second;  // delete callback
   }
   pending_queries_.clear();
 }
 
-void BrowserFeatureExtractor::ExtractFeatures(ClientPhishingRequest* request,
+void BrowserFeatureExtractor::ExtractFeatures(const BrowseInfo& info,
+                                              ClientPhishingRequest* request,
                                               DoneCallback* callback) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   DCHECK(request);
   DCHECK_EQ(0U, request->url().find("http:"));
   DCHECK(callback);
   if (!callback) {
     DLOG(ERROR) << "ExtractFeatures called without a callback object";
     return;
   }
+  bool is_secure_referrer = info.referrer.SchemeIsSecure();
+  if (!is_secure_referrer) {
+    request->set_referrer_url(info.referrer.spec());
+  }
+  AddFeature(features::kHasSSLReferrer,
+             is_secure_referrer ? 1.0 : 0.0,
+             request);
+  AddFeature(features::kPageTransitionType,
+             static_cast<double>(
+                 PageTransition::StripQualifier(info.transition)),
+             request);
+
   pending_extractions_.insert(std::make_pair(request, callback));
   MessageLoop::current()->PostTask(
       FROM_HERE,
       method_factory_.NewRunnableMethod(
           &BrowserFeatureExtractor::StartExtractFeatures,
           request, callback));
 }
 
 void BrowserFeatureExtractor::StartExtractFeatures(
     ClientPhishingRequest* request,
@@ skipping 198 matching lines @@
   if (tab_ && tab_->profile()) {
     *history = tab_->profile()->GetHistoryService(Profile::EXPLICIT_ACCESS);
     if (*history) {
       return true;
     }
   }
   VLOG(2) << "Unable to query history.  No history service available.";
   return false;
 }
 };  // namespace safe_browsing