Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(509)

Unified Diff: chrome/browser/safe_browsing/malware_details_cache.h

Issue 6611023: The optional Malware Details also collects HTTP cache information. See design... (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src/
Patch Set: '' Created 9 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: chrome/browser/safe_browsing/malware_details_cache.h
===================================================================
--- chrome/browser/safe_browsing/malware_details_cache.h (revision 0)
+++ chrome/browser/safe_browsing/malware_details_cache.h (revision 0)
@@ -0,0 +1,118 @@
+// 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.
+
+#ifndef CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_CACHE_H_
+#define CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_CACHE_H_
+#pragma once
+
+// A class that gets malware details from the HTTP Cache.
+// An instance of this class is generated by MalwareDetails.
+
+#include <string>
+#include <vector>
+
+#include "base/hash_tables.h"
+#include "base/memory/linked_ptr.h"
+#include "base/memory/ref_counted.h"
+#include "chrome/browser/safe_browsing/report.pb.h"
+#include "chrome/common/net/url_fetcher.h"
+#include "net/base/completion_callback.h"
+
+namespace net {
+class URLRequestContext;
+}
+
+class MalwareDetailsFactory;
+
+namespace safe_browsing {
+
+// Maps a URL to its Resource.
+typedef base::hash_map<
+ std::string,
+ linked_ptr<safe_browsing::ClientMalwareReportRequest::Resource> > ResourceMap;
+}
+
+class MalwareDetailsCacheCollector
+ : public base::RefCountedThreadSafe<MalwareDetailsCacheCollector>,
+ public URLFetcher::Delegate {
+
+ public:
+ MalwareDetailsCacheCollector();
+
+ // friend class base::RefCountedThreadSafe<MalwareDetailsCacheCollector>;
+ virtual ~MalwareDetailsCacheCollector();
+
+ // We use |request_context_getter|, we modify |resources| and
+ // |result|, and we call |callback|, so they must all remain alive
+ // for the lifetime of this object.
+ void StartCacheCollection(
+ net::URLRequestContextGetter* request_context_getter,
+ safe_browsing::ResourceMap* resources,
+ bool* result,
+ Task* callback);
+
+ // Returns whether or not StartCacheCollection has been called.
+ bool HasStarted();
+
+ // Creates a new URLFetcher and starts it. Public because we create
+ // a task to it, but not to be used by other classes.
+ // tODO: needs to be public?
mattm 2011/04/18 02:19:24 shouldn't need to be public
panayiotis 2011/04/18 17:26:40 Done.
+ void OpenEntry();
+
+ protected:
+ // Implementation of URLFetcher::Delegate. Called after the request
+ // completes (either successfully or with failure).
+ virtual void OnURLFetchComplete(const URLFetcher* source,
+ const GURL& url,
+ const net::URLRequestStatus& status,
+ int response_code,
+ const ResponseCookies& cookies,
+ const std::string& data);
+
+ private:
+ // Points to the url for which we are fetching the HTTP cache entry or
+ // redirect chain.
+ safe_browsing::ResourceMap::iterator resources_it_;
+
+ // Points to the resources_ map in the MalwareDetails.
+ safe_browsing::ResourceMap* resources_;
+
+ // Points to the cache_result_ in the MalwareDetails.
+ bool* result_;
+
+ // Method we call when we are done. The caller must be alive for the
+ // whole time, we are modifying its state (see above).
+ Task* callback_;
+
+ // Set to true as soon as StartCacheCollection is called.
+ bool has_started_;
+
+ // Used to get a pointer to the HTTP cache.
+ net::URLRequestContextGetter* request_context_getter_;
+
+ // The current URLFetcher.
+ scoped_ptr<URLFetcher> current_fetch_;
+
+ // Returns the resource from resources_ that corresponds to |url|
+ safe_browsing::ClientMalwareReportRequest::Resource* GetResource(
+ const GURL& url);
+
+ // Read the HTTP response from |source| and add it to |pb_resource|.
+ void ReadResponse(
+ safe_browsing::ClientMalwareReportRequest::Resource* pb_resource,
+ const URLFetcher* source);
+
+ // Read the body |data| and add it to |pb_resource|.
+ void ReadData(
+ safe_browsing::ClientMalwareReportRequest::Resource* pb_resource,
+ const std::string& data);
+
+ // Called when we are done.
+ void AllDone(bool success);
+
+ // Advances to the next entry in resources_it_.
+ void AdvanceEntry();
+};
+
+#endif // CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_CACHE_H_
Property changes on: chrome/browser/safe_browsing/malware_details_cache.h
___________________________________________________________________
Added: svn:eol-style
+ LF

Powered by Google App Engine
This is Rietveld 408576698