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 // Implements the Chrome Extensions Cookies API. | 5 // Implements the Chrome Extensions Cookies API. |
6 | 6 |
7 #include "chrome/browser/extensions/api/cookies/cookies_api.h" | 7 #include "chrome/browser/extensions/api/cookies/cookies_api.h" |
8 | 8 |
9 #include <vector> | |
10 | |
9 #include "base/bind.h" | 11 #include "base/bind.h" |
10 #include "base/json/json_writer.h" | 12 #include "base/json/json_writer.h" |
13 #include "base/memory/linked_ptr.h" | |
14 #include "base/memory/scoped_ptr.h" | |
11 #include "base/values.h" | 15 #include "base/values.h" |
12 #include "chrome/browser/extensions/api/cookies/cookies_api_constants.h" | 16 #include "chrome/browser/extensions/api/cookies/cookies_api_constants.h" |
13 #include "chrome/browser/extensions/api/cookies/cookies_helpers.h" | 17 #include "chrome/browser/extensions/api/cookies/cookies_helpers.h" |
14 #include "chrome/browser/extensions/extension_event_router.h" | 18 #include "chrome/browser/extensions/extension_event_router.h" |
15 #include "chrome/browser/profiles/profile.h" | 19 #include "chrome/browser/profiles/profile.h" |
16 #include "chrome/browser/ui/browser.h" | 20 #include "chrome/browser/ui/browser.h" |
17 #include "chrome/browser/ui/browser_list.h" | 21 #include "chrome/browser/ui/browser_list.h" |
18 #include "chrome/common/chrome_notification_types.h" | 22 #include "chrome/common/chrome_notification_types.h" |
23 #include "chrome/common/extensions/api/cookies.h" | |
19 #include "chrome/common/extensions/extension.h" | 24 #include "chrome/common/extensions/extension.h" |
20 #include "chrome/common/extensions/extension_error_utils.h" | 25 #include "chrome/common/extensions/extension_error_utils.h" |
21 #include "content/public/browser/browser_thread.h" | 26 #include "content/public/browser/browser_thread.h" |
22 #include "content/public/browser/notification_service.h" | 27 #include "content/public/browser/notification_service.h" |
23 #include "net/cookies/cookie_monster.h" | 28 #include "net/cookies/cookie_monster.h" |
24 #include "net/url_request/url_request_context.h" | 29 #include "net/url_request/url_request_context.h" |
25 #include "net/url_request/url_request_context_getter.h" | 30 #include "net/url_request/url_request_context_getter.h" |
26 | 31 |
27 using content::BrowserThread; | 32 using content::BrowserThread; |
33 using extensions::api::cookies::Cookie; | |
34 using extensions::api::cookies::CookieStore; | |
35 | |
36 namespace Get = extensions::api::cookies::Get; | |
37 namespace GetAll = extensions::api::cookies::GetAll; | |
38 namespace GetAllCookieStores = extensions::api::cookies::GetAllCookieStores; | |
39 namespace Remove = extensions::api::cookies::Remove; | |
40 namespace Set = extensions::api::cookies::Set; | |
28 | 41 |
29 namespace extensions { | 42 namespace extensions { |
30 namespace keys = cookies_api_constants; | 43 namespace keys = cookies_api_constants; |
31 | 44 |
32 ExtensionCookiesEventRouter::ExtensionCookiesEventRouter(Profile* profile) | 45 ExtensionCookiesEventRouter::ExtensionCookiesEventRouter(Profile* profile) |
33 : profile_(profile) {} | 46 : profile_(profile) { |
47 } | |
34 | 48 |
35 ExtensionCookiesEventRouter::~ExtensionCookiesEventRouter() {} | 49 ExtensionCookiesEventRouter::~ExtensionCookiesEventRouter() { |
50 } | |
36 | 51 |
37 void ExtensionCookiesEventRouter::Init() { | 52 void ExtensionCookiesEventRouter::Init() { |
38 CHECK(registrar_.IsEmpty()); | 53 CHECK(registrar_.IsEmpty()); |
39 registrar_.Add(this, | 54 registrar_.Add(this, |
40 chrome::NOTIFICATION_COOKIE_CHANGED, | 55 chrome::NOTIFICATION_COOKIE_CHANGED, |
41 content::NotificationService::AllBrowserContextsAndSources()); | 56 content::NotificationService::AllBrowserContextsAndSources()); |
42 } | 57 } |
43 | 58 |
44 void ExtensionCookiesEventRouter::Observe( | 59 void ExtensionCookiesEventRouter::Observe( |
45 int type, | 60 int type, |
(...skipping 14 matching lines...) Expand all Loading... | |
60 NOTREACHED(); | 75 NOTREACHED(); |
61 } | 76 } |
62 } | 77 } |
63 | 78 |
64 void ExtensionCookiesEventRouter::CookieChanged( | 79 void ExtensionCookiesEventRouter::CookieChanged( |
65 Profile* profile, | 80 Profile* profile, |
66 ChromeCookieDetails* details) { | 81 ChromeCookieDetails* details) { |
67 ListValue args; | 82 ListValue args; |
68 DictionaryValue* dict = new DictionaryValue(); | 83 DictionaryValue* dict = new DictionaryValue(); |
69 dict->SetBoolean(keys::kRemovedKey, details->removed); | 84 dict->SetBoolean(keys::kRemovedKey, details->removed); |
70 dict->Set( | 85 |
71 keys::kCookieKey, | 86 scoped_ptr<Cookie> cookie( |
72 cookies_helpers::CreateCookieValue(*details->cookie, | 87 cookies_helpers::CreateCookie(*details->cookie, |
73 cookies_helpers::GetStoreIdFromProfile(profile))); | 88 cookies_helpers::GetStoreIdFromProfile(profile))); |
89 dict->Set(keys::kCookieKey, cookie->ToValue().release()); | |
74 | 90 |
75 // Map the interal cause to an external string. | 91 // Map the interal cause to an external string. |
76 std::string cause; | 92 std::string cause; |
77 switch (details->cause) { | 93 switch (details->cause) { |
78 case net::CookieMonster::Delegate::CHANGE_COOKIE_EXPLICIT: | 94 case net::CookieMonster::Delegate::CHANGE_COOKIE_EXPLICIT: |
79 cause = keys::kExplicitChangeCause; | 95 cause = keys::kExplicitChangeCause; |
80 break; | 96 break; |
81 | 97 |
82 case net::CookieMonster::Delegate::CHANGE_COOKIE_OVERWRITE: | 98 case net::CookieMonster::Delegate::CHANGE_COOKIE_OVERWRITE: |
83 cause = keys::kOverwriteChangeCause; | 99 cause = keys::kOverwriteChangeCause; |
(...skipping 28 matching lines...) Expand all Loading... | |
112 void ExtensionCookiesEventRouter::DispatchEvent(Profile* profile, | 128 void ExtensionCookiesEventRouter::DispatchEvent(Profile* profile, |
113 const char* event_name, | 129 const char* event_name, |
114 const std::string& json_args, | 130 const std::string& json_args, |
115 GURL& cookie_domain) { | 131 GURL& cookie_domain) { |
116 if (profile && profile->GetExtensionEventRouter()) { | 132 if (profile && profile->GetExtensionEventRouter()) { |
117 profile->GetExtensionEventRouter()->DispatchEventToRenderers( | 133 profile->GetExtensionEventRouter()->DispatchEventToRenderers( |
118 event_name, json_args, profile, cookie_domain, EventFilteringInfo()); | 134 event_name, json_args, profile, cookie_domain, EventFilteringInfo()); |
119 } | 135 } |
120 } | 136 } |
121 | 137 |
122 bool CookiesFunction::ParseUrl(const DictionaryValue* details, GURL* url, | 138 bool CookiesFunction::ParseUrl(const std::string& url_string, GURL* url, |
123 bool check_host_permissions) { | 139 bool check_host_permissions) { |
124 DCHECK(details && url); | |
125 std::string url_string; | |
126 // Get the URL string or return false. | |
127 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kUrlKey, &url_string)); | |
128 *url = GURL(url_string); | 140 *url = GURL(url_string); |
129 if (!url->is_valid()) { | 141 if (!url->is_valid()) { |
130 error_ = ExtensionErrorUtils::FormatErrorMessage( | 142 error_ = ExtensionErrorUtils::FormatErrorMessage( |
131 keys::kInvalidUrlError, url_string); | 143 keys::kInvalidUrlError, url_string); |
132 return false; | 144 return false; |
133 } | 145 } |
134 // Check against host permissions if needed. | 146 // Check against host permissions if needed. |
135 if (check_host_permissions && | 147 if (check_host_permissions && |
136 !GetExtension()->HasHostPermission(*url)) { | 148 !GetExtension()->HasHostPermission(*url)) { |
137 error_ = ExtensionErrorUtils::FormatErrorMessage( | 149 error_ = ExtensionErrorUtils::FormatErrorMessage( |
138 keys::kNoHostPermissionsError, url->spec()); | 150 keys::kNoHostPermissionsError, url->spec()); |
139 return false; | 151 return false; |
140 } | 152 } |
141 return true; | 153 return true; |
142 } | 154 } |
143 | 155 |
144 bool CookiesFunction::ParseStoreContext(const DictionaryValue* details, | 156 bool CookiesFunction::ParseStoreContext(net::URLRequestContextGetter** context, |
145 net::URLRequestContextGetter** context, | |
146 std::string* store_id) { | 157 std::string* store_id) { |
147 DCHECK(details && (context || store_id)); | 158 DCHECK((context || store_id->empty())); |
148 Profile* store_profile = NULL; | 159 Profile* store_profile = NULL; |
149 if (details->HasKey(keys::kStoreIdKey)) { | 160 if (!store_id->empty()) { |
150 // The store ID was explicitly specified in the details dictionary. | |
151 // Retrieve its corresponding cookie store. | |
152 std::string store_id_value; | |
153 // Get the store ID string or return false. | |
154 EXTENSION_FUNCTION_VALIDATE( | |
155 details->GetString(keys::kStoreIdKey, &store_id_value)); | |
156 store_profile = cookies_helpers::ChooseProfileFromStoreId( | 161 store_profile = cookies_helpers::ChooseProfileFromStoreId( |
157 store_id_value, profile(), include_incognito()); | 162 *store_id, profile(), include_incognito()); |
158 if (!store_profile) { | 163 if (!store_profile) { |
159 error_ = ExtensionErrorUtils::FormatErrorMessage( | 164 error_ = ExtensionErrorUtils::FormatErrorMessage( |
160 keys::kInvalidStoreIdError, store_id_value); | 165 keys::kInvalidStoreIdError, *store_id); |
161 return false; | 166 return false; |
162 } | 167 } |
163 } else { | 168 } else { |
164 // The store ID was not specified; use the current execution context's | 169 // The store ID was not specified; use the current execution context's |
165 // cookie store by default. | 170 // cookie store by default. |
166 // GetCurrentBrowser() already takes into account incognito settings. | 171 // GetCurrentBrowser() already takes into account incognito settings. |
167 Browser* current_browser = GetCurrentBrowser(); | 172 Browser* current_browser = GetCurrentBrowser(); |
168 if (!current_browser) { | 173 if (!current_browser) { |
169 error_ = keys::kNoCookieStoreFoundError; | 174 error_ = keys::kNoCookieStoreFoundError; |
170 return false; | 175 return false; |
171 } | 176 } |
172 store_profile = current_browser->profile(); | 177 store_profile = current_browser->profile(); |
178 *store_id = cookies_helpers::GetStoreIdFromProfile(store_profile); | |
173 } | 179 } |
174 DCHECK(store_profile); | |
175 | 180 |
176 if (context) | 181 if (context) |
177 *context = store_profile->GetRequestContext(); | 182 *context = store_profile->GetRequestContext(); |
178 if (store_id) | 183 DCHECK(context); |
179 *store_id = cookies_helpers::GetStoreIdFromProfile(store_profile); | |
180 | 184 |
181 return true; | 185 return true; |
182 } | 186 } |
183 | 187 |
184 GetCookieFunction::GetCookieFunction() {} | 188 GetCookieFunction::GetCookieFunction() { |
189 } | |
185 | 190 |
186 GetCookieFunction::~GetCookieFunction() {} | 191 GetCookieFunction::~GetCookieFunction() { |
192 } | |
187 | 193 |
188 bool GetCookieFunction::RunImpl() { | 194 bool GetCookieFunction::RunImpl() { |
189 // Return false if the arguments are malformed. | 195 parsed_args_ = Get::Params::Create(*args_); |
190 DictionaryValue* details; | 196 EXTENSION_FUNCTION_VALIDATE(parsed_args_.get()); |
191 EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &details)); | |
192 DCHECK(details); | |
193 | 197 |
194 // Read/validate input parameters. | 198 // Read/validate input parameters. |
195 if (!ParseUrl(details, &url_, true)) | 199 if (!ParseUrl(parsed_args_->details.url, &url_, true)) |
196 return false; | 200 return false; |
197 | 201 |
198 // Get the cookie name string or return false. | 202 net::URLRequestContextGetter* store_context = NULL; |
199 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kNameKey, &name_)); | 203 std::string* store_id = parsed_args_->details.store_id.get() ? |
204 parsed_args_->details.store_id.get() : new std::string(""); | |
205 if (!ParseStoreContext(&store_context, store_id)) | |
206 return false; | |
207 store_context_ = store_context; | |
208 if (!parsed_args_->details.store_id.get()) | |
209 parsed_args_->details.store_id.reset(store_id); | |
210 DCHECK(parsed_args_->details.store_id.get()); | |
Aaron Boodman
2012/07/09 17:34:48
Looks like you always expect this value to get pas
mitchellwrosen
2012/07/09 19:36:12
I realized there's definitely no way for this chec
| |
200 | 211 |
201 net::URLRequestContextGetter* store_context = NULL; | 212 DCHECK(store_context && !parsed_args_->details.store_id->empty()); |
Aaron Boodman
2012/07/09 17:34:49
I don't think it's necessary to DCHECK either of t
mitchellwrosen
2012/07/09 19:36:12
Done.
| |
202 if (!ParseStoreContext(details, &store_context, &store_id_)) | |
203 return false; | |
204 | |
205 DCHECK(store_context && !store_id_.empty()); | |
206 store_context_ = store_context; | 213 store_context_ = store_context; |
207 | 214 |
208 bool rv = BrowserThread::PostTask( | 215 bool rv = BrowserThread::PostTask( |
209 BrowserThread::IO, FROM_HERE, | 216 BrowserThread::IO, FROM_HERE, |
210 base::Bind(&GetCookieFunction::GetCookieOnIOThread, this)); | 217 base::Bind(&GetCookieFunction::GetCookieOnIOThread, this)); |
211 DCHECK(rv); | 218 DCHECK(rv); |
212 | 219 |
213 // Will finish asynchronously. | 220 // Will finish asynchronously. |
214 return true; | 221 return true; |
215 } | 222 } |
216 | 223 |
217 void GetCookieFunction::GetCookieOnIOThread() { | 224 void GetCookieFunction::GetCookieOnIOThread() { |
218 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 225 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
219 net::CookieStore* cookie_store = | 226 net::CookieStore* cookie_store = |
220 store_context_->GetURLRequestContext()->cookie_store(); | 227 store_context_->GetURLRequestContext()->cookie_store(); |
221 cookies_helpers::GetCookieListFromStore( | 228 cookies_helpers::GetCookieListFromStore( |
222 cookie_store, url_, | 229 cookie_store, url_, |
223 base::Bind(&GetCookieFunction::GetCookieCallback, this)); | 230 base::Bind(&GetCookieFunction::GetCookieCallback, this)); |
224 } | 231 } |
225 | 232 |
226 void GetCookieFunction::GetCookieCallback(const net::CookieList& cookie_list) { | 233 void GetCookieFunction::GetCookieCallback(const net::CookieList& cookie_list) { |
227 net::CookieList::const_iterator it; | 234 net::CookieList::const_iterator it; |
228 for (it = cookie_list.begin(); it != cookie_list.end(); ++it) { | 235 for (it = cookie_list.begin(); it != cookie_list.end(); ++it) { |
229 // Return the first matching cookie. Relies on the fact that the | 236 // Return the first matching cookie. Relies on the fact that the |
230 // CookieMonster returns them in canonical order (longest path, then | 237 // CookieMonster returns them in canonical order (longest path, then |
231 // earliest creation time). | 238 // earliest creation time). |
232 if (it->Name() == name_) { | 239 if (it->Name() == parsed_args_->details.name) { |
233 result_.reset( | 240 scoped_ptr<Cookie> cookie( |
234 cookies_helpers::CreateCookieValue(*it, store_id_)); | 241 cookies_helpers::CreateCookie(*it, *parsed_args_->details.store_id)); |
242 result_.reset(Get::Result::Create(*cookie)); | |
235 break; | 243 break; |
236 } | 244 } |
237 } | 245 } |
238 | 246 |
239 // The cookie doesn't exist; return null. | 247 // The cookie doesn't exist; return null. |
240 if (it == cookie_list.end()) | 248 if (it == cookie_list.end()) |
241 result_.reset(Value::CreateNullValue()); | 249 result_.reset(Value::CreateNullValue()); |
242 | 250 |
243 bool rv = BrowserThread::PostTask( | 251 bool rv = BrowserThread::PostTask( |
244 BrowserThread::UI, FROM_HERE, | 252 BrowserThread::UI, FROM_HERE, |
245 base::Bind(&GetCookieFunction::RespondOnUIThread, this)); | 253 base::Bind(&GetCookieFunction::RespondOnUIThread, this)); |
246 DCHECK(rv); | 254 DCHECK(rv); |
247 } | 255 } |
248 | 256 |
249 void GetCookieFunction::RespondOnUIThread() { | 257 void GetCookieFunction::RespondOnUIThread() { |
250 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 258 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
251 SendResponse(true); | 259 SendResponse(true); |
252 } | 260 } |
253 | 261 |
254 GetAllCookiesFunction::GetAllCookiesFunction() : details_(NULL) {} | 262 GetAllCookiesFunction::GetAllCookiesFunction() { |
263 } | |
255 | 264 |
256 GetAllCookiesFunction::~GetAllCookiesFunction() {} | 265 GetAllCookiesFunction::~GetAllCookiesFunction() { |
266 } | |
257 | 267 |
258 bool GetAllCookiesFunction::RunImpl() { | 268 bool GetAllCookiesFunction::RunImpl() { |
259 // Return false if the arguments are malformed. | 269 parsed_args_ = GetAll::Params::Create(*args_); |
260 EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &details_)); | 270 EXTENSION_FUNCTION_VALIDATE(parsed_args_.get()); |
261 DCHECK(details_); | |
262 | 271 |
263 // Read/validate input parameters. | 272 // Read/validate input parameters. |
264 if (details_->HasKey(keys::kUrlKey) && !ParseUrl(details_, &url_, false)) | 273 scoped_ptr<GetAll::Params> params(GetAll::Params::Create(*args_)); |
Aaron Boodman
2012/07/09 17:34:49
Why are you creating another instance of this? Isn
mitchellwrosen
2012/07/09 19:36:12
Sorry, I re-factored a couple times and forgot to
| |
265 return false; | 274 if (params->details.url.get()) { |
275 if (!ParseUrl(*params->details.url, &url_, false)) | |
276 return false; | |
277 } | |
266 | 278 |
267 net::URLRequestContextGetter* store_context = NULL; | 279 net::URLRequestContextGetter* store_context = NULL; |
268 if (!ParseStoreContext(details_, &store_context, &store_id_)) | 280 std::string* store_id = parsed_args_->details.store_id.get() ? |
281 parsed_args_->details.store_id.get() : | |
282 new std::string(""); // Wrapped in a scoped_ptr below. | |
283 if (!ParseStoreContext(&store_context, store_id)) | |
269 return false; | 284 return false; |
Aaron Boodman
2012/07/09 17:34:49
The new string leaks here. I think safer/easier to
mitchellwrosen
2012/07/09 19:36:12
Done.
| |
270 DCHECK(store_context); | |
271 store_context_ = store_context; | 285 store_context_ = store_context; |
286 if (!parsed_args_->details.store_id.get()) | |
287 parsed_args_->details.store_id.reset(store_id); | |
272 | 288 |
273 bool rv = BrowserThread::PostTask( | 289 bool rv = BrowserThread::PostTask( |
274 BrowserThread::IO, FROM_HERE, | 290 BrowserThread::IO, FROM_HERE, |
275 base::Bind(&GetAllCookiesFunction::GetAllCookiesOnIOThread, this)); | 291 base::Bind(&GetAllCookiesFunction::GetAllCookiesOnIOThread, this)); |
276 DCHECK(rv); | 292 DCHECK(rv); |
277 | 293 |
278 // Will finish asynchronously. | 294 // Will finish asynchronously. |
279 return true; | 295 return true; |
280 } | 296 } |
281 | 297 |
282 void GetAllCookiesFunction::GetAllCookiesOnIOThread() { | 298 void GetAllCookiesFunction::GetAllCookiesOnIOThread() { |
283 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 299 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
284 net::CookieStore* cookie_store = | 300 net::CookieStore* cookie_store = |
285 store_context_->GetURLRequestContext()->cookie_store(); | 301 store_context_->GetURLRequestContext()->cookie_store(); |
286 cookies_helpers::GetCookieListFromStore( | 302 cookies_helpers::GetCookieListFromStore( |
287 cookie_store, url_, | 303 cookie_store, url_, |
288 base::Bind(&GetAllCookiesFunction::GetAllCookiesCallback, this)); | 304 base::Bind(&GetAllCookiesFunction::GetAllCookiesCallback, this)); |
289 } | 305 } |
290 | 306 |
291 void GetAllCookiesFunction::GetAllCookiesCallback( | 307 void GetAllCookiesFunction::GetAllCookiesCallback( |
292 const net::CookieList& cookie_list) { | 308 const net::CookieList& cookie_list) { |
293 const extensions::Extension* extension = GetExtension(); | 309 const extensions::Extension* extension = GetExtension(); |
294 if (extension) { | 310 if (extension) { |
295 ListValue* matching_list = new ListValue(); | 311 std::vector<linked_ptr<Cookie> > match_vector; |
296 cookies_helpers::AppendMatchingCookiesToList( | 312 cookies_helpers::AppendMatchingCookiesToVector( |
297 cookie_list, store_id_, url_, details_, | 313 cookie_list, url_, &parsed_args_->details, |
298 GetExtension(), matching_list); | 314 GetExtension(), &match_vector); |
299 result_.reset(matching_list); | 315 result_.reset(GetAll::Result::Create(match_vector)); |
300 } | 316 } |
301 bool rv = BrowserThread::PostTask( | 317 bool rv = BrowserThread::PostTask( |
302 BrowserThread::UI, FROM_HERE, | 318 BrowserThread::UI, FROM_HERE, |
303 base::Bind(&GetAllCookiesFunction::RespondOnUIThread, this)); | 319 base::Bind(&GetAllCookiesFunction::RespondOnUIThread, this)); |
304 DCHECK(rv); | 320 DCHECK(rv); |
305 } | 321 } |
306 | 322 |
307 void GetAllCookiesFunction::RespondOnUIThread() { | 323 void GetAllCookiesFunction::RespondOnUIThread() { |
308 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 324 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
309 SendResponse(true); | 325 SendResponse(true); |
310 } | 326 } |
311 | 327 |
312 SetCookieFunction::SetCookieFunction() | 328 SetCookieFunction::SetCookieFunction() { |
313 : secure_(false), | |
314 http_only_(false), | |
315 success_(false) { | |
316 } | 329 } |
317 | 330 |
318 SetCookieFunction::~SetCookieFunction() { | 331 SetCookieFunction::~SetCookieFunction() { |
319 } | 332 } |
320 | 333 |
321 bool SetCookieFunction::RunImpl() { | 334 bool SetCookieFunction::RunImpl() { |
322 // Return false if the arguments are malformed. | 335 parsed_args_ = Set::Params::Create(*args_); |
323 DictionaryValue* details; | 336 EXTENSION_FUNCTION_VALIDATE(parsed_args_.get()); |
324 EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &details)); | |
325 DCHECK(details); | |
326 | 337 |
327 // Read/validate input parameters. | 338 // Read/validate input parameters. |
328 if (!ParseUrl(details, &url_, true)) | 339 if (!ParseUrl(parsed_args_->details.url, &url_, true)) |
329 return false; | 340 return false; |
330 // The macros below return false if argument types are not as expected. | |
331 if (details->HasKey(keys::kNameKey)) | |
332 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kNameKey, &name_)); | |
333 if (details->HasKey(keys::kValueKey)) | |
334 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kValueKey, &value_)); | |
335 if (details->HasKey(keys::kDomainKey)) | |
336 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kDomainKey, &domain_)); | |
337 if (details->HasKey(keys::kPathKey)) | |
338 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kPathKey, &path_)); | |
339 | |
340 if (details->HasKey(keys::kSecureKey)) { | |
341 EXTENSION_FUNCTION_VALIDATE( | |
342 details->GetBoolean(keys::kSecureKey, &secure_)); | |
343 } | |
344 if (details->HasKey(keys::kHttpOnlyKey)) { | |
345 EXTENSION_FUNCTION_VALIDATE( | |
346 details->GetBoolean(keys::kHttpOnlyKey, &http_only_)); | |
347 } | |
348 if (details->HasKey(keys::kExpirationDateKey)) { | |
349 Value* expiration_date_value; | |
350 EXTENSION_FUNCTION_VALIDATE(details->Get(keys::kExpirationDateKey, | |
351 &expiration_date_value)); | |
352 double expiration_date; | |
353 if (expiration_date_value->IsType(Value::TYPE_INTEGER)) { | |
354 int expiration_date_int; | |
355 EXTENSION_FUNCTION_VALIDATE( | |
356 expiration_date_value->GetAsInteger(&expiration_date_int)); | |
357 expiration_date = static_cast<double>(expiration_date_int); | |
358 } else { | |
359 EXTENSION_FUNCTION_VALIDATE( | |
360 expiration_date_value->GetAsDouble(&expiration_date)); | |
361 } | |
362 // Time::FromDoubleT converts double time 0 to empty Time object. So we need | |
363 // to do special handling here. | |
364 expiration_time_ = (expiration_date == 0) ? | |
365 base::Time::UnixEpoch() : base::Time::FromDoubleT(expiration_date); | |
366 } | |
367 | 341 |
368 net::URLRequestContextGetter* store_context = NULL; | 342 net::URLRequestContextGetter* store_context = NULL; |
369 if (!ParseStoreContext(details, &store_context, NULL)) | 343 std::string* store_id = parsed_args_->details.store_id.get() ? |
344 parsed_args_->details.store_id.get() : new std::string(""); | |
345 if (!ParseStoreContext(&store_context, store_id)) | |
370 return false; | 346 return false; |
371 DCHECK(store_context); | |
372 store_context_ = store_context; | 347 store_context_ = store_context; |
348 if (!parsed_args_->details.store_id.get()) | |
349 parsed_args_->details.store_id.reset(store_id); | |
373 | 350 |
374 bool rv = BrowserThread::PostTask( | 351 bool rv = BrowserThread::PostTask( |
375 BrowserThread::IO, FROM_HERE, | 352 BrowserThread::IO, FROM_HERE, |
376 base::Bind(&SetCookieFunction::SetCookieOnIOThread, this)); | 353 base::Bind(&SetCookieFunction::SetCookieOnIOThread, this)); |
377 DCHECK(rv); | 354 DCHECK(rv); |
378 | 355 |
379 // Will finish asynchronously. | 356 // Will finish asynchronously. |
380 return true; | 357 return true; |
381 } | 358 } |
382 | 359 |
383 void SetCookieFunction::SetCookieOnIOThread() { | 360 void SetCookieFunction::SetCookieOnIOThread() { |
384 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 361 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
385 net::CookieMonster* cookie_monster = | 362 net::CookieMonster* cookie_monster = |
386 store_context_->GetURLRequestContext()->cookie_store()-> | 363 store_context_->GetURLRequestContext()->cookie_store()-> |
387 GetCookieMonster(); | 364 GetCookieMonster(); |
365 | |
366 base::Time expiration_time; | |
367 if (parsed_args_->details.expiration_date.get()) { | |
368 // Time::FromDoubleT converts double time 0 to empty Time object. So we need | |
369 // to do special handling here. | |
370 expiration_time = (*parsed_args_->details.expiration_date == 0) ? | |
371 base::Time::UnixEpoch() : | |
372 base::Time::FromDoubleT(*parsed_args_->details.expiration_date); | |
373 } | |
374 | |
375 if (parsed_args_->details.name.get()) | |
376 LOG(INFO) << "Cookie name: " << *parsed_args_->details.name; | |
377 | |
388 cookie_monster->SetCookieWithDetailsAsync( | 378 cookie_monster->SetCookieWithDetailsAsync( |
389 url_, name_, value_, domain_, path_, expiration_time_, | 379 url_, |
390 secure_, http_only_, base::Bind(&SetCookieFunction::PullCookie, this)); | 380 parsed_args_->details.name.get() ? *parsed_args_->details.name : "", |
381 parsed_args_->details.value.get() ? *parsed_args_->details.value : "", | |
382 parsed_args_->details.domain.get() ? *parsed_args_->details.domain : "", | |
383 parsed_args_->details.path.get() ? *parsed_args_->details.path : "", | |
384 expiration_time, | |
385 parsed_args_->details.secure.get() ? | |
386 *parsed_args_->details.secure.get() : | |
387 false, | |
388 parsed_args_->details.http_only.get() ? | |
389 *parsed_args_->details.http_only : | |
390 false, | |
391 base::Bind(&SetCookieFunction::PullCookie, this)); | |
391 } | 392 } |
392 | 393 |
393 void SetCookieFunction::PullCookie(bool set_cookie_result) { | 394 void SetCookieFunction::PullCookie(bool set_cookie_result) { |
394 // Pull the newly set cookie. | 395 // Pull the newly set cookie. |
395 net::CookieMonster* cookie_monster = | 396 net::CookieMonster* cookie_monster = |
396 store_context_->GetURLRequestContext()->cookie_store()-> | 397 store_context_->GetURLRequestContext()->cookie_store()-> |
397 GetCookieMonster(); | 398 GetCookieMonster(); |
398 success_ = set_cookie_result; | 399 success_ = set_cookie_result; |
399 cookies_helpers::GetCookieListFromStore( | 400 cookies_helpers::GetCookieListFromStore( |
400 cookie_monster, url_, | 401 cookie_monster, url_, |
401 base::Bind(&SetCookieFunction::PullCookieCallback, this)); | 402 base::Bind(&SetCookieFunction::PullCookieCallback, this)); |
402 } | 403 } |
403 | 404 |
404 void SetCookieFunction::PullCookieCallback(const net::CookieList& cookie_list) { | 405 void SetCookieFunction::PullCookieCallback(const net::CookieList& cookie_list) { |
405 net::CookieList::const_iterator it; | 406 net::CookieList::const_iterator it; |
406 for (it = cookie_list.begin(); it != cookie_list.end(); ++it) { | 407 for (it = cookie_list.begin(); it != cookie_list.end(); ++it) { |
407 // Return the first matching cookie. Relies on the fact that the | 408 // Return the first matching cookie. Relies on the fact that the |
408 // CookieMonster returns them in canonical order (longest path, then | 409 // CookieMonster returns them in canonical order (longest path, then |
409 // earliest creation time). | 410 // earliest creation time). |
410 if (it->Name() == name_) { | 411 std::string name = parsed_args_->details.name.get() ? |
411 result_.reset( | 412 *parsed_args_->details.name : ""; |
412 cookies_helpers::CreateCookieValue(*it, store_id_)); | 413 if (it->Name() == name) { |
414 scoped_ptr<Cookie> cookie( | |
415 cookies_helpers::CreateCookie(*it, *parsed_args_->details.store_id)); | |
416 result_.reset(Set::Result::Create(*cookie)); | |
413 break; | 417 break; |
414 } | 418 } |
415 } | 419 } |
416 | 420 |
417 bool rv = BrowserThread::PostTask( | 421 bool rv = BrowserThread::PostTask( |
418 BrowserThread::UI, FROM_HERE, | 422 BrowserThread::UI, FROM_HERE, |
419 base::Bind(&SetCookieFunction::RespondOnUIThread, this)); | 423 base::Bind(&SetCookieFunction::RespondOnUIThread, this)); |
420 DCHECK(rv); | 424 DCHECK(rv); |
421 } | 425 } |
422 | 426 |
423 void SetCookieFunction::RespondOnUIThread() { | 427 void SetCookieFunction::RespondOnUIThread() { |
424 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 428 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
425 if (!success_) { | 429 if (!success_) { |
430 std::string name = parsed_args_->details.name.get() ? | |
431 *parsed_args_->details.name : ""; | |
426 error_ = ExtensionErrorUtils::FormatErrorMessage( | 432 error_ = ExtensionErrorUtils::FormatErrorMessage( |
427 keys::kCookieSetFailedError, name_); | 433 keys::kCookieSetFailedError, name); |
428 } | 434 } |
429 SendResponse(success_); | 435 SendResponse(success_); |
430 } | 436 } |
431 | 437 |
432 RemoveCookieFunction::RemoveCookieFunction() : success_(false) { | 438 RemoveCookieFunction::RemoveCookieFunction() { |
433 } | 439 } |
434 | 440 |
435 RemoveCookieFunction::~RemoveCookieFunction() { | 441 RemoveCookieFunction::~RemoveCookieFunction() { |
436 } | 442 } |
437 | 443 |
438 bool RemoveCookieFunction::RunImpl() { | 444 bool RemoveCookieFunction::RunImpl() { |
439 // Return false if the arguments are malformed. | 445 parsed_args_ = Remove::Params::Create(*args_); |
440 DictionaryValue* details; | 446 EXTENSION_FUNCTION_VALIDATE(parsed_args_.get()); |
441 EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &details)); | |
442 DCHECK(details); | |
443 | 447 |
444 // Read/validate input parameters. | 448 // Read/validate input parameters. |
445 if (!ParseUrl(details, &url_, true)) | 449 if (!ParseUrl(parsed_args_->details.url, &url_, true)) |
446 return false; | 450 return false; |
447 | 451 |
448 // Get the cookie name string or return false. | |
449 EXTENSION_FUNCTION_VALIDATE(details->GetString(keys::kNameKey, &name_)); | |
450 | |
451 net::URLRequestContextGetter* store_context = NULL; | 452 net::URLRequestContextGetter* store_context = NULL; |
452 if (!ParseStoreContext(details, &store_context, &store_id_)) | 453 std::string* store_id = parsed_args_->details.store_id.get() ? |
454 parsed_args_->details.store_id.get() : new std::string(""); | |
455 if (!ParseStoreContext(&store_context, store_id)) | |
453 return false; | 456 return false; |
454 DCHECK(store_context); | |
455 store_context_ = store_context; | 457 store_context_ = store_context; |
458 if (!parsed_args_->details.store_id.get()) | |
459 parsed_args_->details.store_id.reset(store_id); | |
456 | 460 |
457 // Pass the work off to the IO thread. | 461 // Pass the work off to the IO thread. |
458 bool rv = BrowserThread::PostTask( | 462 bool rv = BrowserThread::PostTask( |
459 BrowserThread::IO, FROM_HERE, | 463 BrowserThread::IO, FROM_HERE, |
460 base::Bind(&RemoveCookieFunction::RemoveCookieOnIOThread, this)); | 464 base::Bind(&RemoveCookieFunction::RemoveCookieOnIOThread, this)); |
461 DCHECK(rv); | 465 DCHECK(rv); |
462 | 466 |
463 // Will return asynchronously. | 467 // Will return asynchronously. |
464 return true; | 468 return true; |
465 } | 469 } |
466 | 470 |
467 void RemoveCookieFunction::RemoveCookieOnIOThread() { | 471 void RemoveCookieFunction::RemoveCookieOnIOThread() { |
468 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 472 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
469 | 473 |
470 // Remove the cookie | 474 // Remove the cookie |
471 net::CookieStore* cookie_store = | 475 net::CookieStore* cookie_store = |
472 store_context_->GetURLRequestContext()->cookie_store(); | 476 store_context_->GetURLRequestContext()->cookie_store(); |
473 cookie_store->DeleteCookieAsync( | 477 cookie_store->DeleteCookieAsync( |
474 url_, name_, | 478 url_, parsed_args_->details.name, |
475 base::Bind(&RemoveCookieFunction::RemoveCookieCallback, this)); | 479 base::Bind(&RemoveCookieFunction::RemoveCookieCallback, this)); |
476 } | 480 } |
477 | 481 |
478 void RemoveCookieFunction::RemoveCookieCallback() { | 482 void RemoveCookieFunction::RemoveCookieCallback() { |
479 // Build the callback result | 483 // Build the callback result |
480 DictionaryValue* resultDictionary = new DictionaryValue(); | 484 Remove::Result::Details details; |
481 resultDictionary->SetString(keys::kNameKey, name_); | 485 details.name = parsed_args_->details.name; |
482 resultDictionary->SetString(keys::kUrlKey, url_.spec()); | 486 details.url = url_.spec(); |
483 resultDictionary->SetString(keys::kStoreIdKey, store_id_); | 487 details.store_id = *parsed_args_->details.store_id; |
484 result_.reset(resultDictionary); | 488 result_.reset(Remove::Result::Create(details)); |
485 | 489 |
486 // Return to UI thread | 490 // Return to UI thread |
487 bool rv = BrowserThread::PostTask( | 491 bool rv = BrowserThread::PostTask( |
488 BrowserThread::UI, FROM_HERE, | 492 BrowserThread::UI, FROM_HERE, |
489 base::Bind(&RemoveCookieFunction::RespondOnUIThread, this)); | 493 base::Bind(&RemoveCookieFunction::RespondOnUIThread, this)); |
490 DCHECK(rv); | 494 DCHECK(rv); |
491 } | 495 } |
492 | 496 |
493 void RemoveCookieFunction::RespondOnUIThread() { | 497 void RemoveCookieFunction::RespondOnUIThread() { |
494 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 498 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
(...skipping 13 matching lines...) Expand all Loading... | |
508 } | 512 } |
509 DCHECK(original_profile != incognito_profile); | 513 DCHECK(original_profile != incognito_profile); |
510 | 514 |
511 // Iterate through all browser instances, and for each browser, | 515 // Iterate through all browser instances, and for each browser, |
512 // add its tab IDs to either the regular or incognito tab ID list depending | 516 // add its tab IDs to either the regular or incognito tab ID list depending |
513 // whether the browser is regular or incognito. | 517 // whether the browser is regular or incognito. |
514 for (BrowserList::const_iterator iter = BrowserList::begin(); | 518 for (BrowserList::const_iterator iter = BrowserList::begin(); |
515 iter != BrowserList::end(); ++iter) { | 519 iter != BrowserList::end(); ++iter) { |
516 Browser* browser = *iter; | 520 Browser* browser = *iter; |
517 if (browser->profile() == original_profile) { | 521 if (browser->profile() == original_profile) { |
518 cookies_helpers::AppendToTabIdList(browser, | 522 cookies_helpers::AppendToTabIdList(browser, original_tab_ids.get()); |
519 original_tab_ids.get()); | |
520 } else if (incognito_tab_ids.get() && | 523 } else if (incognito_tab_ids.get() && |
521 browser->profile() == incognito_profile) { | 524 browser->profile() == incognito_profile) { |
522 cookies_helpers::AppendToTabIdList(browser, | 525 cookies_helpers::AppendToTabIdList(browser, incognito_tab_ids.get()); |
523 incognito_tab_ids.get()); | |
524 } | 526 } |
525 } | 527 } |
526 // Return a list of all cookie stores with at least one open tab. | 528 // Return a list of all cookie stores with at least one open tab. |
527 ListValue* cookie_store_list = new ListValue(); | 529 std::vector<linked_ptr<CookieStore> > cookie_stores; |
528 if (original_tab_ids->GetSize() > 0) { | 530 if (original_tab_ids->GetSize() > 0) { |
529 cookie_store_list->Append( | 531 linked_ptr<CookieStore> cookie_store( |
530 cookies_helpers::CreateCookieStoreValue( | 532 cookies_helpers::CreateCookieStore( |
531 original_profile, original_tab_ids.release())); | 533 original_profile, original_tab_ids.release())); |
534 cookie_stores.push_back(cookie_store); | |
532 } | 535 } |
533 if (incognito_tab_ids.get() && incognito_tab_ids->GetSize() > 0 && | 536 if (incognito_tab_ids.get() && incognito_tab_ids->GetSize() > 0 && |
534 incognito_profile) { | 537 incognito_profile) { |
535 cookie_store_list->Append( | 538 linked_ptr<CookieStore> cookie_store( |
536 cookies_helpers::CreateCookieStoreValue( | 539 cookies_helpers::CreateCookieStore( |
537 incognito_profile, incognito_tab_ids.release())); | 540 incognito_profile, incognito_tab_ids.release())); |
541 cookie_stores.push_back(cookie_store); | |
538 } | 542 } |
539 result_.reset(cookie_store_list); | 543 result_.reset(GetAllCookieStores::Result::Create(cookie_stores)); |
540 return true; | 544 return true; |
541 } | 545 } |
542 | 546 |
543 void GetAllCookieStoresFunction::Run() { | 547 void GetAllCookieStoresFunction::Run() { |
544 SendResponse(RunImpl()); | 548 SendResponse(RunImpl()); |
545 } | 549 } |
546 | 550 |
547 } // namespace extensions | 551 } // namespace extensions |
OLD | NEW |