OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_BROWSER_PRERENDER_PRERENDER_TRACKER_H_ | 5 #ifndef CHROME_BROWSER_PRERENDER_PRERENDER_TRACKER_H_ |
6 #define CHROME_BROWSER_PRERENDER_PRERENDER_TRACKER_H_ | 6 #define CHROME_BROWSER_PRERENDER_PRERENDER_TRACKER_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 #include <set> | 9 #include <set> |
10 #include <utility> | 10 #include <utility> |
11 | 11 |
12 #include "base/containers/hash_tables.h" | 12 #include "base/containers/hash_tables.h" |
13 #include "base/memory/ref_counted.h" | 13 #include "base/memory/ref_counted.h" |
14 #include "base/memory/weak_ptr.h" | 14 #include "base/memory/weak_ptr.h" |
15 #include "base/synchronization/lock.h" | 15 #include "base/synchronization/lock.h" |
16 #include "chrome/browser/prerender/prerender_cookie_store.h" | 16 #include "chrome/browser/prerender/prerender_cookie_store.h" |
17 #include "content/public/browser/render_process_host_observer.h" | 17 #include "content/public/browser/render_process_host_observer.h" |
18 #include "url/gurl.h" | 18 #include "url/gurl.h" |
19 | 19 |
20 namespace net { | 20 namespace net { |
21 class URLRequestContextGetter; | 21 class URLRequestContextGetter; |
22 } | 22 } |
23 | 23 |
24 namespace prerender { | 24 namespace prerender { |
25 | 25 |
26 class PrerenderPendingSwapThrottle; | |
27 | |
28 // Global object for maintaining various prerender state on the IO thread. | 26 // Global object for maintaining various prerender state on the IO thread. |
29 class PrerenderTracker { | 27 class PrerenderTracker { |
30 public: | 28 public: |
31 typedef std::pair<int, int> ChildRouteIdPair; | 29 typedef std::pair<int, int> ChildRouteIdPair; |
32 | 30 |
33 PrerenderTracker(); | 31 PrerenderTracker(); |
34 virtual ~PrerenderTracker(); | 32 virtual ~PrerenderTracker(); |
35 | 33 |
36 // Returns whether or not a RenderFrame and URL are regarding a pending | |
37 // prerender swap. Can only be called on the IO thread. | |
38 bool IsPendingSwapRequestOnIOThread(int render_process_id, | |
39 int render_frame_id, | |
40 const GURL& url) const; | |
41 | |
42 // Called when a PrerenderPendingSwapThrottle defers a request. Cancel or | |
43 // Resume will be called on |throttle| when the prerender is canceled or used, | |
44 // respectively. | |
45 void AddPendingSwapThrottleOnIOThread( | |
46 int render_process_id, int render_frame_id, const GURL& url, | |
47 const base::WeakPtr<PrerenderPendingSwapThrottle>& throttle); | |
48 | |
49 // Called to add throttles for a pending prerender swap. | |
50 void AddPrerenderPendingSwap( | |
51 const ChildRouteIdPair& render_frame_route_id_pair, | |
52 const GURL& url); | |
53 | |
54 // Called to remove the throttles for a pending prerender swap. | |
55 void RemovePrerenderPendingSwap( | |
56 const ChildRouteIdPair& render_frame_route_id_pair, | |
57 bool swap_successful); | |
58 | |
59 // Gets the Prerender Cookie Store for a specific render process, if it | 34 // Gets the Prerender Cookie Store for a specific render process, if it |
60 // is a prerender. Only to be called from the IO thread. | 35 // is a prerender. Only to be called from the IO thread. |
61 scoped_refptr<PrerenderCookieStore> GetPrerenderCookieStoreForRenderProcess( | 36 scoped_refptr<PrerenderCookieStore> GetPrerenderCookieStoreForRenderProcess( |
62 int process_id); | 37 int process_id); |
63 | 38 |
64 // Called when a given render process has changed a cookie for |url|, | 39 // Called when a given render process has changed a cookie for |url|, |
65 // in |cookie_monster|. | 40 // in |cookie_monster|. |
66 // Only to be called from the IO thread. | 41 // Only to be called from the IO thread. |
67 void OnCookieChangedForURL(int process_id, | 42 void OnCookieChangedForURL(int process_id, |
68 net::CookieMonster* cookie_monster, | 43 net::CookieMonster* cookie_monster, |
69 const GURL& url); | 44 const GURL& url); |
70 | 45 |
71 void AddPrerenderCookieStoreOnIOThread( | 46 void AddPrerenderCookieStoreOnIOThread( |
72 int process_id, | 47 int process_id, |
73 scoped_refptr<net::URLRequestContextGetter> request_context, | 48 scoped_refptr<net::URLRequestContextGetter> request_context, |
74 const base::Closure& cookie_conflict_cb); | 49 const base::Closure& cookie_conflict_cb); |
75 | 50 |
76 void RemovePrerenderCookieStoreOnIOThread(int process_id, bool was_swapped); | 51 void RemovePrerenderCookieStoreOnIOThread(int process_id, bool was_swapped); |
77 | 52 |
78 private: | 53 private: |
79 // Add/remove prerenders pending swap on the IO Thread. | |
80 void AddPrerenderPendingSwapOnIOThread( | |
81 const ChildRouteIdPair& render_frame_route_id_pair, const GURL& url); | |
82 void RemovePrerenderPendingSwapOnIOThread( | |
83 const ChildRouteIdPair& render_frame_route_id_pair, | |
84 bool swap_successful); | |
85 | |
86 struct PendingSwapThrottleData { | |
87 explicit PendingSwapThrottleData(const GURL& swap_url); | |
88 ~PendingSwapThrottleData(); | |
89 GURL url; | |
90 base::WeakPtr<PrerenderPendingSwapThrottle> throttle; | |
91 }; | |
92 | |
93 // Map of pending prerender swaps and their associated throttles, | |
94 // maintained on the IO thread. The key is the routing ID pair | |
95 // of a RenderFrame. | |
96 typedef std::map<ChildRouteIdPair, PendingSwapThrottleData> | |
97 PendingSwapThrottleMap; | |
98 PendingSwapThrottleMap pending_swap_throttle_map_; | |
99 | |
100 // Map of prerendering render process ids to PrerenderCookieStore used for | 54 // Map of prerendering render process ids to PrerenderCookieStore used for |
101 // the prerender. Only to be used on the IO thread. | 55 // the prerender. Only to be used on the IO thread. |
102 typedef base::hash_map<int, scoped_refptr<PrerenderCookieStore> > | 56 typedef base::hash_map<int, scoped_refptr<PrerenderCookieStore> > |
103 PrerenderCookieStoreMap; | 57 PrerenderCookieStoreMap; |
104 PrerenderCookieStoreMap prerender_cookie_store_map_; | 58 PrerenderCookieStoreMap prerender_cookie_store_map_; |
105 | 59 |
106 DISALLOW_COPY_AND_ASSIGN(PrerenderTracker); | 60 DISALLOW_COPY_AND_ASSIGN(PrerenderTracker); |
107 }; | 61 }; |
108 | 62 |
109 } // namespace prerender | 63 } // namespace prerender |
110 | 64 |
111 #endif // CHROME_BROWSER_PRERENDER_PRERENDER_TRACKER_H_ | 65 #endif // CHROME_BROWSER_PRERENDER_PRERENDER_TRACKER_H_ |
OLD | NEW |