| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/automation/automation_util.h" | 5 #include "chrome/browser/automation/automation_util.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| (...skipping 134 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 145 TabContents* contents, | 145 TabContents* contents, |
| 146 int* value_size, | 146 int* value_size, |
| 147 std::string* value) { | 147 std::string* value) { |
| 148 *value_size = -1; | 148 *value_size = -1; |
| 149 if (url.is_valid() && contents) { | 149 if (url.is_valid() && contents) { |
| 150 scoped_refptr<net::URLRequestContextGetter> context_getter = | 150 scoped_refptr<net::URLRequestContextGetter> context_getter = |
| 151 GetRequestContext(contents); | 151 GetRequestContext(contents); |
| 152 base::WaitableEvent event(true /* manual reset */, | 152 base::WaitableEvent event(true /* manual reset */, |
| 153 false /* not initially signaled */); | 153 false /* not initially signaled */); |
| 154 CHECK(BrowserThread::PostTask( | 154 CHECK(BrowserThread::PostTask( |
| 155 BrowserThread::IO, FROM_HERE, | 155 BrowserThread::IO, FROM_HERE, |
| 156 NewRunnableFunction(&GetCookiesOnIOThread, | 156 base::Bind(&GetCookiesOnIOThread, url, context_getter, &event, value))); |
| 157 url, context_getter, &event, value))); | |
| 158 event.Wait(); | 157 event.Wait(); |
| 159 | 158 |
| 160 *value_size = static_cast<int>(value->size()); | 159 *value_size = static_cast<int>(value->size()); |
| 161 } | 160 } |
| 162 } | 161 } |
| 163 | 162 |
| 164 void SetCookie(const GURL& url, | 163 void SetCookie(const GURL& url, |
| 165 const std::string& value, | 164 const std::string& value, |
| 166 TabContents* contents, | 165 TabContents* contents, |
| 167 int* response_value) { | 166 int* response_value) { |
| 168 *response_value = -1; | 167 *response_value = -1; |
| 169 | 168 |
| 170 if (url.is_valid() && contents) { | 169 if (url.is_valid() && contents) { |
| 171 scoped_refptr<net::URLRequestContextGetter> context_getter = | 170 scoped_refptr<net::URLRequestContextGetter> context_getter = |
| 172 GetRequestContext(contents); | 171 GetRequestContext(contents); |
| 173 base::WaitableEvent event(true /* manual reset */, | 172 base::WaitableEvent event(true /* manual reset */, |
| 174 false /* not initially signaled */); | 173 false /* not initially signaled */); |
| 175 bool success = false; | 174 bool success = false; |
| 176 CHECK(BrowserThread::PostTask( | 175 CHECK(BrowserThread::PostTask( |
| 177 BrowserThread::IO, FROM_HERE, | 176 BrowserThread::IO, FROM_HERE, |
| 178 NewRunnableFunction(&SetCookieOnIOThread, | 177 base::Bind(&SetCookieOnIOThread, url, value, context_getter, &event, |
| 179 url, value, context_getter, &event, | 178 &success))); |
| 180 &success))); | |
| 181 event.Wait(); | 179 event.Wait(); |
| 182 if (success) | 180 if (success) |
| 183 *response_value = 1; | 181 *response_value = 1; |
| 184 } | 182 } |
| 185 } | 183 } |
| 186 | 184 |
| 187 void DeleteCookie(const GURL& url, | 185 void DeleteCookie(const GURL& url, |
| 188 const std::string& cookie_name, | 186 const std::string& cookie_name, |
| 189 TabContents* contents, | 187 TabContents* contents, |
| 190 bool* success) { | 188 bool* success) { |
| 191 *success = false; | 189 *success = false; |
| 192 if (url.is_valid() && contents) { | 190 if (url.is_valid() && contents) { |
| 193 scoped_refptr<net::URLRequestContextGetter> context_getter = | 191 scoped_refptr<net::URLRequestContextGetter> context_getter = |
| 194 GetRequestContext(contents); | 192 GetRequestContext(contents); |
| 195 base::WaitableEvent event(true /* manual reset */, | 193 base::WaitableEvent event(true /* manual reset */, |
| 196 false /* not initially signaled */); | 194 false /* not initially signaled */); |
| 197 CHECK(BrowserThread::PostTask( | 195 CHECK(BrowserThread::PostTask( |
| 198 BrowserThread::IO, FROM_HERE, | 196 BrowserThread::IO, FROM_HERE, |
| 199 NewRunnableFunction(&DeleteCookieOnIOThread, | 197 base::Bind(&DeleteCookieOnIOThread, url, cookie_name, context_getter, |
| 200 url, cookie_name, context_getter, &event))); | 198 &event))); |
| 201 event.Wait(); | 199 event.Wait(); |
| 202 *success = true; | 200 *success = true; |
| 203 } | 201 } |
| 204 } | 202 } |
| 205 | 203 |
| 206 void GetCookiesJSON(AutomationProvider* provider, | 204 void GetCookiesJSON(AutomationProvider* provider, |
| 207 DictionaryValue* args, | 205 DictionaryValue* args, |
| 208 IPC::Message* reply_message) { | 206 IPC::Message* reply_message) { |
| 209 AutomationJSONReply reply(provider, reply_message); | 207 AutomationJSONReply reply(provider, reply_message); |
| 210 std::string url; | 208 std::string url; |
| 211 if (!args->GetString("url", &url)) { | 209 if (!args->GetString("url", &url)) { |
| 212 reply.SendError("'url' missing or invalid"); | 210 reply.SendError("'url' missing or invalid"); |
| 213 return; | 211 return; |
| 214 } | 212 } |
| 215 | 213 |
| 216 // Since we may be on the UI thread don't call GetURLRequestContext(). | 214 // Since we may be on the UI thread don't call GetURLRequestContext(). |
| 217 scoped_refptr<net::URLRequestContextGetter> context_getter = | 215 scoped_refptr<net::URLRequestContextGetter> context_getter = |
| 218 provider->profile()->GetRequestContext(); | 216 provider->profile()->GetRequestContext(); |
| 219 | 217 |
| 220 net::CookieList cookie_list; | 218 net::CookieList cookie_list; |
| 221 base::WaitableEvent event(true /* manual reset */, | 219 base::WaitableEvent event(true /* manual reset */, |
| 222 false /* not initially signaled */); | 220 false /* not initially signaled */); |
| 223 Task* task = NewRunnableFunction( | 221 base::Closure task = base::Bind(&GetCanonicalCookiesOnIOThread, GURL(url), |
| 224 &GetCanonicalCookiesOnIOThread, | 222 context_getter, &event, &cookie_list); |
| 225 GURL(url), context_getter, &event, &cookie_list); | |
| 226 if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, task)) { | 223 if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, task)) { |
| 227 reply.SendError("Couldn't post task to get the cookies"); | 224 reply.SendError("Couldn't post task to get the cookies"); |
| 228 return; | 225 return; |
| 229 } | 226 } |
| 230 event.Wait(); | 227 event.Wait(); |
| 231 | 228 |
| 232 ListValue* list = new ListValue(); | 229 ListValue* list = new ListValue(); |
| 233 for (size_t i = 0; i < cookie_list.size(); ++i) { | 230 for (size_t i = 0; i < cookie_list.size(); ++i) { |
| 234 const net::CookieMonster::CanonicalCookie& cookie = cookie_list[i]; | 231 const net::CookieMonster::CanonicalCookie& cookie = cookie_list[i]; |
| 235 DictionaryValue* cookie_dict = new DictionaryValue(); | 232 DictionaryValue* cookie_dict = new DictionaryValue(); |
| (...skipping 25 matching lines...) Expand all Loading... |
| 261 reply.SendError("'name' missing or invalid"); | 258 reply.SendError("'name' missing or invalid"); |
| 262 return; | 259 return; |
| 263 } | 260 } |
| 264 | 261 |
| 265 // Since we may be on the UI thread don't call GetURLRequestContext(). | 262 // Since we may be on the UI thread don't call GetURLRequestContext(). |
| 266 scoped_refptr<net::URLRequestContextGetter> context_getter = | 263 scoped_refptr<net::URLRequestContextGetter> context_getter = |
| 267 provider->profile()->GetRequestContext(); | 264 provider->profile()->GetRequestContext(); |
| 268 | 265 |
| 269 base::WaitableEvent event(true /* manual reset */, | 266 base::WaitableEvent event(true /* manual reset */, |
| 270 false /* not initially signaled */); | 267 false /* not initially signaled */); |
| 271 Task* task = NewRunnableFunction( | 268 base::Closure task = base::Bind(&DeleteCookieOnIOThread, GURL(url), name, |
| 272 &DeleteCookieOnIOThread, | 269 context_getter, &event); |
| 273 GURL(url), name, context_getter, &event); | |
| 274 if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, task)) { | 270 if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, task)) { |
| 275 reply.SendError("Couldn't post task to delete the cookie"); | 271 reply.SendError("Couldn't post task to delete the cookie"); |
| 276 return; | 272 return; |
| 277 } | 273 } |
| 278 event.Wait(); | 274 event.Wait(); |
| 279 reply.SendSuccess(NULL); | 275 reply.SendSuccess(NULL); |
| 280 } | 276 } |
| 281 | 277 |
| 282 void SetCookieJSON(AutomationProvider* provider, | 278 void SetCookieJSON(AutomationProvider* provider, |
| 283 DictionaryValue* args, | 279 DictionaryValue* args, |
| (...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 351 return; | 347 return; |
| 352 } | 348 } |
| 353 | 349 |
| 354 // Since we may be on the UI thread don't call GetURLRequestContext(). | 350 // Since we may be on the UI thread don't call GetURLRequestContext(). |
| 355 scoped_refptr<net::URLRequestContextGetter> context_getter = | 351 scoped_refptr<net::URLRequestContextGetter> context_getter = |
| 356 provider->profile()->GetRequestContext(); | 352 provider->profile()->GetRequestContext(); |
| 357 | 353 |
| 358 base::WaitableEvent event(true /* manual reset */, | 354 base::WaitableEvent event(true /* manual reset */, |
| 359 false /* not initially signaled */); | 355 false /* not initially signaled */); |
| 360 bool success = false; | 356 bool success = false; |
| 361 Task* task = NewRunnableFunction( | 357 base::Closure task = base::Bind( |
| 362 &SetCookieWithDetailsOnIOThread, | 358 &SetCookieWithDetailsOnIOThread, GURL(url), *cookie.get(), domain, |
| 363 GURL(url), *cookie.get(), domain, context_getter, &event, &success); | 359 context_getter, &event, &success); |
| 364 if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, task)) { | 360 if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, task)) { |
| 365 reply.SendError("Couldn't post task to set the cookie"); | 361 reply.SendError("Couldn't post task to set the cookie"); |
| 366 return; | 362 return; |
| 367 } | 363 } |
| 368 event.Wait(); | 364 event.Wait(); |
| 369 | 365 |
| 370 if (!success) { | 366 if (!success) { |
| 371 reply.SendError("Could not set the cookie"); | 367 reply.SendError("Could not set the cookie"); |
| 372 return; | 368 return; |
| 373 } | 369 } |
| 374 reply.SendSuccess(NULL); | 370 reply.SendSuccess(NULL); |
| 375 } | 371 } |
| 376 | 372 |
| 377 bool SendErrorIfModalDialogActive(AutomationProvider* provider, | 373 bool SendErrorIfModalDialogActive(AutomationProvider* provider, |
| 378 IPC::Message* message) { | 374 IPC::Message* message) { |
| 379 bool active = AppModalDialogQueue::GetInstance()->HasActiveDialog(); | 375 bool active = AppModalDialogQueue::GetInstance()->HasActiveDialog(); |
| 380 if (active) { | 376 if (active) { |
| 381 AutomationJSONReply(provider, message).SendError( | 377 AutomationJSONReply(provider, message).SendError( |
| 382 "Command cannot be performed because a modal dialog is active"); | 378 "Command cannot be performed because a modal dialog is active"); |
| 383 } | 379 } |
| 384 return active; | 380 return active; |
| 385 } | 381 } |
| 386 | 382 |
| 387 } // namespace automation_util | 383 } // namespace automation_util |
| OLD | NEW |