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 3 // found in the LICENSE file. 4 5 #include "chrome/browser/prerender/prerender_tracker.h" 6 7 #include "base/bind.h" 8 #include "base/logging.h" 9 #include "chrome/browser/prerender/prerender_pending_swap_throttle.h" 10 #include "content/public/browser/browser_thread.h" 11 #include "content/public/browser/render_process_host.h" 12 #include "net/url_request/url_request_context.h" 13 #include "net/url_request/url_request_context_getter.h" 14 15 using content::BrowserThread; 16 17 namespace prerender { 18 19 PrerenderTracker::PrerenderTracker() { 20 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); 21 } 22 23 PrerenderTracker::~PrerenderTracker() { 24 } 25 26 bool PrerenderTracker::IsPendingSwapRequestOnIOThread( 27 int render_process_id, int render_frame_id, const GURL& url) const { 28 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 29 30 ChildRouteIdPair render_frame_route_id_pair( 31 render_process_id, render_frame_id); 32 PendingSwapThrottleMap::const_iterator it = 33 pending_swap_throttle_map_.find(render_frame_route_id_pair); 34 return (it != pending_swap_throttle_map_.end() && it->second.url == url); 35 } 36 37 void PrerenderTracker::AddPendingSwapThrottleOnIOThread( 38 int render_process_id, 39 int render_frame_id, 40 const GURL& url, 41 const base::WeakPtr<PrerenderPendingSwapThrottle>& throttle) { 42 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 43 44 ChildRouteIdPair render_frame_route_id_pair( 45 render_process_id, render_frame_id); 46 PendingSwapThrottleMap::iterator it = 47 pending_swap_throttle_map_.find(render_frame_route_id_pair); 48 DCHECK(it != pending_swap_throttle_map_.end()); 49 if (it == pending_swap_throttle_map_.end()) 50 return; 51 CHECK(!it->second.throttle); 52 it->second.throttle = throttle; 53 } 54 55 void PrerenderTracker::AddPrerenderPendingSwapOnIOThread( 56 const ChildRouteIdPair& render_frame_route_id_pair, 57 const GURL& url) { 58 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 59 std::pair<PendingSwapThrottleMap::iterator, bool> insert_result = 60 pending_swap_throttle_map_.insert(std::make_pair( 61 render_frame_route_id_pair, PendingSwapThrottleData(url))); 62 DCHECK(insert_result.second); 63 } 64 65 void PrerenderTracker::RemovePrerenderPendingSwapOnIOThread( 66 const ChildRouteIdPair& render_frame_route_id_pair, 67 bool swap_successful) { 68 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 69 PendingSwapThrottleMap::iterator it = 70 pending_swap_throttle_map_.find(render_frame_route_id_pair); 71 DCHECK(it != pending_swap_throttle_map_.end()); 72 // Cancel or resume all throttled resources. 73 if (it->second.throttle) { 74 if (swap_successful) 75 it->second.throttle->Cancel(); 76 else 77 it->second.throttle->Resume(); 78 } 79 pending_swap_throttle_map_.erase(render_frame_route_id_pair); 80 } 81 82 void PrerenderTracker::AddPrerenderPendingSwap( 83 const ChildRouteIdPair& render_frame_route_id_pair, 84 const GURL& url) { 85 BrowserThread::PostTask( 86 BrowserThread::IO, FROM_HERE, 87 base::Bind(&PrerenderTracker::AddPrerenderPendingSwapOnIOThread, 88 base::Unretained(this), render_frame_route_id_pair, url)); 89 } 90 91 void PrerenderTracker::RemovePrerenderPendingSwap( 92 const ChildRouteIdPair& render_frame_route_id_pair, 93 bool swap_successful) { 94 BrowserThread::PostTask( 95 BrowserThread::IO, FROM_HERE, 96 base::Bind(&PrerenderTracker::RemovePrerenderPendingSwapOnIOThread, 97 base::Unretained(this), render_frame_route_id_pair, 98 swap_successful)); 99 } 100 101 PrerenderTracker::PendingSwapThrottleData::PendingSwapThrottleData( 102 const GURL& swap_url) 103 : url(swap_url) { 104 } 105 106 PrerenderTracker::PendingSwapThrottleData::~PendingSwapThrottleData() { 107 } 108 109 scoped_refptr<PrerenderCookieStore> 110 PrerenderTracker::GetPrerenderCookieStoreForRenderProcess( 111 int process_id) { 112 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 113 PrerenderCookieStoreMap::const_iterator it = 114 prerender_cookie_store_map_.find(process_id); 115 116 if (it == prerender_cookie_store_map_.end()) 117 return NULL; 118 119 return it->second; 120 } 121 122 void PrerenderTracker::OnCookieChangedForURL( 123 int process_id, 124 net::CookieMonster* cookie_monster, 125 const GURL& url) { 126 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 127 128 // We only care about cookie changes by non-prerender tabs, since only those 129 // get applied to the underlying cookie store. Therefore, if a cookie change 130 // originated from a prerender, there is nothing to do. 131 if (ContainsKey(prerender_cookie_store_map_, process_id)) 132 return; 133 134 // Since the cookie change did not come from a prerender, broadcast it too 135 // all prerenders so that they can be cancelled if there is a conflict. 136 for (PrerenderCookieStoreMap::iterator it = 137 prerender_cookie_store_map_.begin(); 138 it != prerender_cookie_store_map_.end(); 139 ++it) { 140 it->second->OnCookieChangedForURL(cookie_monster, url); 141 } 142 } 143 144 void PrerenderTracker::RemovePrerenderCookieStoreOnIOThread(int process_id, 145 bool was_swapped) { 146 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 147 148 PrerenderCookieStoreMap::iterator it = 149 prerender_cookie_store_map_.find(process_id); 150 151 if (it == prerender_cookie_store_map_.end()) 152 return; 153 154 std::vector<GURL> cookie_change_urls; 155 if (was_swapped) 156 it->second->ApplyChanges(&cookie_change_urls); 157 158 scoped_refptr<net::CookieMonster> cookie_monster( 159 it->second->default_cookie_monster()); 160 161 prerender_cookie_store_map_.erase(it); 162 163 // For each cookie updated by ApplyChanges, we need to call 164 // OnCookieChangedForURL so that any potentially conflicting prerenders 165 // will be aborted. 166 for (std::vector<GURL>::const_iterator url_it = cookie_change_urls.begin(); 167 url_it != cookie_change_urls.end(); 168 ++url_it) { 169 OnCookieChangedForURL(process_id, cookie_monster.get(), *url_it); 170 } 171 } 172 173 void PrerenderTracker::AddPrerenderCookieStoreOnIOThread( 174 int process_id, 175 scoped_refptr<net::URLRequestContextGetter> request_context, 176 const base::Closure& cookie_conflict_cb) { 177 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 178 DCHECK(request_context.get() != NULL); 179 net::CookieMonster* cookie_monster = 180 request_context->GetURLRequestContext()->cookie_store()-> 181 GetCookieMonster(); 182 DCHECK(cookie_monster != NULL); 183 bool exists = (prerender_cookie_store_map_.find(process_id) != 184 prerender_cookie_store_map_.end()); 185 DCHECK(!exists); 186 if (exists) 187 return; 188 prerender_cookie_store_map_[process_id] = 189 new PrerenderCookieStore(make_scoped_refptr(cookie_monster), 190 cookie_conflict_cb); 191 } 192 193 } // namespace prerender 194
