blob: 16bd25ff2bb271fc56484399feeeb89ebba0640f [file] [log] [blame]
// Copyright (c) 2012 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.
// This class gets redirect chain for urls from the history service.
#include <string>
#include <vector>
#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/scoped_observer.h"
#include "base/sequenced_task_runner_helpers.h"
#include "base/task/cancelable_task_tracker.h"
#include "components/history/core/browser/history_service_observer.h"
#include "content/public/browser/browser_thread.h"
namespace safe_browsing {
typedef std::vector<GURL> RedirectChain;
class ThreatDetailsRedirectsCollector
: public base::RefCounted<ThreatDetailsRedirectsCollector>,
public history::HistoryServiceObserver {
explicit ThreatDetailsRedirectsCollector(
const base::WeakPtr<history::HistoryService>& history_service);
// Collects urls' redirects chain information from the history service.
// We get access to history service via web_contents in UI thread.
void StartHistoryCollection(const std::vector<GURL>& urls,
const base::Closure& callback);
// Returns whether or not StartCacheCollection has been called.
bool HasStarted() const;
// Returns the redirect urls we get from history service
const std::vector<RedirectChain>& GetCollectedUrls() const;
// history::HistoryServiceObserver
void HistoryServiceBeingDeleted(
history::HistoryService* history_service) override;
friend class base::RefCounted<ThreatDetailsRedirectsCollector>;
~ThreatDetailsRedirectsCollector() override;
void StartGetRedirects(const std::vector<GURL>& urls);
void GetRedirects(const GURL& url);
void OnGotQueryRedirectsTo(const GURL& url,
const history::RedirectList* redirect_list);
// Runs the callback when redirects collecting is all done.
void AllDone();
base::CancelableTaskTracker request_tracker_;
// Method we call when we are done. The caller must be alive for the
// whole time, we are modifying its state (see above).
base::Closure callback_;
// Sets to true once StartHistoryCollection is called
bool has_started_;
// The urls we need to get redirects for
std::vector<GURL> urls_;
// The iterator goes over urls_
std::vector<GURL>::iterator urls_it_;
// The collected directs from history service
std::vector<RedirectChain> redirects_urls_;
base::WeakPtr<history::HistoryService> history_service_;
ScopedObserver<history::HistoryService, history::HistoryServiceObserver>
} // namespace safe_browsing