OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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 #include "chrome/browser/prerender/prerender_cookie_store.h" | 5 #include "chrome/browser/prerender/prerender_cookie_store.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "base/stl_util.h" | 8 #include "base/stl_util.h" |
9 #include "content/public/browser/browser_thread.h" | 9 #include "content/public/browser/browser_thread.h" |
10 | 10 |
11 using content::BrowserThread; | 11 using content::BrowserThread; |
12 | 12 |
13 namespace prerender { | 13 namespace prerender { |
14 | 14 |
15 PrerenderCookieStore::PrerenderCookieStore( | 15 PrerenderCookieStore::PrerenderCookieStore( |
16 scoped_refptr<net::CookieMonster> default_cookie_monster, | 16 scoped_refptr<net::CookieMonster> default_cookie_monster, |
17 const base::Closure& cookie_conflict_cb) | 17 const base::Closure& cookie_conflict_cb) |
18 : in_forwarding_mode_(false), | 18 : in_forwarding_mode_(false), |
19 default_cookie_monster_(default_cookie_monster), | 19 default_cookie_monster_(default_cookie_monster), |
20 changes_cookie_monster_(new net::CookieMonster(NULL, NULL)), | 20 changes_cookie_monster_(new net::CookieMonster(NULL, NULL)), |
21 cookie_conflict_cb_(cookie_conflict_cb), | 21 cookie_conflict_cb_(cookie_conflict_cb), |
22 cookie_conflict_(false) { | 22 cookie_conflict_(false) { |
23 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 23 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
24 DCHECK(default_cookie_monster_ != NULL); | 24 DCHECK(default_cookie_monster_.get() != NULL); |
25 DCHECK(default_cookie_monster_->loaded()); | 25 DCHECK(default_cookie_monster_->loaded()); |
26 } | 26 } |
27 | 27 |
28 PrerenderCookieStore::~PrerenderCookieStore() { | 28 PrerenderCookieStore::~PrerenderCookieStore() { |
29 } | 29 } |
30 | 30 |
31 void PrerenderCookieStore::SetCookieWithOptionsAsync( | 31 void PrerenderCookieStore::SetCookieWithOptionsAsync( |
32 const GURL& url, | 32 const GURL& url, |
33 const std::string& cookie_line, | 33 const std::string& cookie_line, |
34 const net::CookieOptions& options, | 34 const net::CookieOptions& options, |
(...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 | 112 |
113 net::CookieStore* PrerenderCookieStore::GetCookieStoreForCookieOpAndLog( | 113 net::CookieStore* PrerenderCookieStore::GetCookieStoreForCookieOpAndLog( |
114 const CookieOperation& op) { | 114 const CookieOperation& op) { |
115 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 115 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
116 | 116 |
117 std::string key = default_cookie_monster_->GetKey(op.url.host()); | 117 std::string key = default_cookie_monster_->GetKey(op.url.host()); |
118 bool is_read_only = (op.op == COOKIE_OP_GET_COOKIES_WITH_OPTIONS_ASYNC || | 118 bool is_read_only = (op.op == COOKIE_OP_GET_COOKIES_WITH_OPTIONS_ASYNC || |
119 op.op == COOKIE_OP_GET_ALL_COOKIES_FOR_URL_ASYNC); | 119 op.op == COOKIE_OP_GET_ALL_COOKIES_FOR_URL_ASYNC); |
120 | 120 |
121 if (in_forwarding_mode_) | 121 if (in_forwarding_mode_) |
122 return default_cookie_monster_; | 122 return default_cookie_monster_.get(); |
123 | 123 |
124 DCHECK(changes_cookie_monster_ != NULL); | 124 DCHECK(changes_cookie_monster_.get() != NULL); |
125 | 125 |
126 cookie_ops_.push_back(op); | 126 cookie_ops_.push_back(op); |
127 | 127 |
128 bool key_copied = ContainsKey(copied_keys_, key); | 128 bool key_copied = ContainsKey(copied_keys_, key); |
129 | 129 |
130 if (key_copied) | 130 if (key_copied) |
131 return changes_cookie_monster_; | 131 return changes_cookie_monster_.get(); |
132 | 132 |
133 if (is_read_only) { | 133 if (is_read_only) { |
134 // Insert this key into the set of read keys, if it doesn't exist yet. | 134 // Insert this key into the set of read keys, if it doesn't exist yet. |
135 if (!ContainsKey(read_keys_, key)) | 135 if (!ContainsKey(read_keys_, key)) |
136 read_keys_.insert(key); | 136 read_keys_.insert(key); |
137 return default_cookie_monster_; | 137 return default_cookie_monster_.get(); |
138 } | 138 } |
139 | 139 |
140 // If this method hasn't returned yet, the key has not been copied yet, | 140 // If this method hasn't returned yet, the key has not been copied yet, |
141 // and we must copy it due to the requested write operation. | 141 // and we must copy it due to the requested write operation. |
142 | 142 |
143 bool copy_success = default_cookie_monster_-> | 143 bool copy_success = |
144 CopyCookiesForKeyToOtherCookieMonster(key, changes_cookie_monster_); | 144 default_cookie_monster_->CopyCookiesForKeyToOtherCookieMonster( |
| 145 key, changes_cookie_monster_.get()); |
145 | 146 |
146 // The copy must succeed. | 147 // The copy must succeed. |
147 DCHECK(copy_success); | 148 DCHECK(copy_success); |
148 | 149 |
149 copied_keys_.insert(key); | 150 copied_keys_.insert(key); |
150 | 151 |
151 return changes_cookie_monster_; | 152 return changes_cookie_monster_.get(); |
152 } | 153 } |
153 | 154 |
154 void PrerenderCookieStore::ApplyChanges(std::vector<GURL>* cookie_change_urls) { | 155 void PrerenderCookieStore::ApplyChanges(std::vector<GURL>* cookie_change_urls) { |
155 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 156 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
156 | 157 |
157 if (in_forwarding_mode_) | 158 if (in_forwarding_mode_) |
158 return; | 159 return; |
159 | 160 |
160 // Apply all changes to the underlying cookie store. | 161 // Apply all changes to the underlying cookie store. |
161 for (std::vector<CookieOperation>::const_iterator it = cookie_ops_.begin(); | 162 for (std::vector<CookieOperation>::const_iterator it = cookie_ops_.begin(); |
(...skipping 29 matching lines...) Expand all Loading... |
191 changes_cookie_monster_ = NULL; | 192 changes_cookie_monster_ = NULL; |
192 } | 193 } |
193 | 194 |
194 void PrerenderCookieStore::OnCookieChangedForURL( | 195 void PrerenderCookieStore::OnCookieChangedForURL( |
195 net::CookieMonster* cookie_monster, | 196 net::CookieMonster* cookie_monster, |
196 const GURL& url) { | 197 const GURL& url) { |
197 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 198 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
198 | 199 |
199 // If the cookie was changed in a different cookie monster than the one | 200 // If the cookie was changed in a different cookie monster than the one |
200 // being decorated, there is nothing to do). | 201 // being decorated, there is nothing to do). |
201 if (cookie_monster != default_cookie_monster_) | 202 if (cookie_monster != default_cookie_monster_.get()) |
202 return; | 203 return; |
203 | 204 |
204 if (in_forwarding_mode_) | 205 if (in_forwarding_mode_) |
205 return; | 206 return; |
206 | 207 |
207 // If we have encountered a conflict before, it has already been recorded | 208 // If we have encountered a conflict before, it has already been recorded |
208 // and the cb has been issued, so nothing to do. | 209 // and the cb has been issued, so nothing to do. |
209 if (cookie_conflict_) | 210 if (cookie_conflict_) |
210 return; | 211 return; |
211 | 212 |
(...skipping 20 matching lines...) Expand all Loading... |
232 } | 233 } |
233 } | 234 } |
234 | 235 |
235 PrerenderCookieStore::CookieOperation::CookieOperation() { | 236 PrerenderCookieStore::CookieOperation::CookieOperation() { |
236 } | 237 } |
237 | 238 |
238 PrerenderCookieStore::CookieOperation::~CookieOperation() { | 239 PrerenderCookieStore::CookieOperation::~CookieOperation() { |
239 } | 240 } |
240 | 241 |
241 } // namespace prerender | 242 } // namespace prerender |
OLD | NEW |