| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_LAST_DOWNLOAD_FINDER_H_ | |
| 6 #define CHROME_BROWSER_SAFE_BROWSING_LAST_DOWNLOAD_FINDER_H_ | |
| 7 | |
| 8 #include <vector> | |
| 9 | |
| 10 #include "base/callback.h" | |
| 11 #include "base/compiler_specific.h" | |
| 12 #include "base/macros.h" | |
| 13 #include "base/memory/scoped_ptr.h" | |
| 14 #include "base/memory/weak_ptr.h" | |
| 15 #include "chrome/browser/history/download_row.h" | |
| 16 #include "content/public/browser/notification_observer.h" | |
| 17 #include "content/public/browser/notification_registrar.h" | |
| 18 | |
| 19 class HistoryService; | |
| 20 class Profile; | |
| 21 | |
| 22 namespace content { | |
| 23 class NotificationDetails; | |
| 24 class NotificationSource; | |
| 25 } | |
| 26 | |
| 27 namespace history { | |
| 28 struct DownloadRow; | |
| 29 } | |
| 30 | |
| 31 namespace safe_browsing { | |
| 32 | |
| 33 class ClientIncidentReport_DownloadDetails; | |
| 34 | |
| 35 // Finds the most recent executable downloaded by any on-the-record profile with | |
| 36 // history that participates in safe browsing. | |
| 37 class LastDownloadFinder : public content::NotificationObserver { | |
| 38 public: | |
| 39 // The type of a callback run by the finder upon completion. The argument is a | |
| 40 // protobuf containing details of the download that was found, or an empty | |
| 41 // pointer if none was found. | |
| 42 typedef base::Callback<void(scoped_ptr<ClientIncidentReport_DownloadDetails>)> | |
| 43 LastDownloadCallback; | |
| 44 | |
| 45 virtual ~LastDownloadFinder(); | |
| 46 | |
| 47 // Initiates an asynchronous search for the most recent download. |callback| | |
| 48 // will be run when the search is complete. The returned instance can be | |
| 49 // deleted to terminate the search, in which case |callback| is not invoked. | |
| 50 // Returns NULL without running |callback| if there are no eligible profiles | |
| 51 // to search. | |
| 52 static scoped_ptr<LastDownloadFinder> Create( | |
| 53 const LastDownloadCallback& callback); | |
| 54 | |
| 55 protected: | |
| 56 // Protected constructor so that unit tests can create a fake finder. | |
| 57 LastDownloadFinder(); | |
| 58 | |
| 59 private: | |
| 60 LastDownloadFinder(const std::vector<Profile*>& profiles, | |
| 61 const LastDownloadCallback& callback); | |
| 62 | |
| 63 // Adds |profile| to the set of profiles to be searched if it is an | |
| 64 // on-the-record profile with history that participates in safe browsing. The | |
| 65 // search is initiated if the profile has already loaded history. | |
| 66 void SearchInProfile(Profile* profile); | |
| 67 | |
| 68 // Initiates a search in |profile| if it is in the set of profiles to be | |
| 69 // searched. | |
| 70 void OnProfileHistoryLoaded(Profile* profile, | |
| 71 HistoryService* history_service); | |
| 72 | |
| 73 // Abandons the search for downloads in |profile|, reporting results if there | |
| 74 // are no more pending queries. | |
| 75 void AbandonSearchInProfile(Profile* profile); | |
| 76 | |
| 77 // HistoryService::DownloadQueryCallback. Retrieves the most recent completed | |
| 78 // executable download from |downloads| and reports results if there are no | |
| 79 // more pending queries. | |
| 80 void OnDownloadQuery( | |
| 81 Profile* profile, | |
| 82 scoped_ptr<std::vector<history::DownloadRow> > downloads); | |
| 83 | |
| 84 // Removes the profile pointed to by |it| from profiles_ and reports results | |
| 85 // if there are no more pending queries. | |
| 86 void RemoveProfileAndReportIfDone(std::vector<Profile*>::iterator it); | |
| 87 | |
| 88 // Invokes the caller-supplied callback with the download found. | |
| 89 void ReportResults(); | |
| 90 | |
| 91 // content::NotificationObserver methods. | |
| 92 virtual void Observe(int type, | |
| 93 const content::NotificationSource& source, | |
| 94 const content::NotificationDetails& details) OVERRIDE; | |
| 95 | |
| 96 // Caller-supplied callback to be invoked when the most recent download is | |
| 97 // found. | |
| 98 LastDownloadCallback callback_; | |
| 99 | |
| 100 // The profiles for which a download query is pending. | |
| 101 std::vector<Profile*> profiles_; | |
| 102 | |
| 103 // Registrar for observing profile lifecycle notifications. | |
| 104 content::NotificationRegistrar notification_registrar_; | |
| 105 | |
| 106 // The most recent download, updated progressively as query results arrive. | |
| 107 history::DownloadRow most_recent_row_; | |
| 108 | |
| 109 // A factory for asynchronous operations on profiles' HistoryService. | |
| 110 base::WeakPtrFactory<LastDownloadFinder> weak_ptr_factory_; | |
| 111 | |
| 112 DISALLOW_COPY_AND_ASSIGN(LastDownloadFinder); | |
| 113 }; | |
| 114 | |
| 115 } // namespace safe_browsing | |
| 116 | |
| 117 #endif // CHROME_BROWSER_SAFE_BROWSING_LAST_DOWNLOAD_FINDER_H_ | |
| OLD | NEW |