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/extensions/user_script_master.h" | 5 #include "chrome/browser/extensions/user_script_master.h" |
6 | 6 |
7 #include <map> | 7 #include <map> |
8 #include <string> | 8 #include <string> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/bind.h" | 11 #include "base/bind.h" |
12 #include "base/file_path.h" | 12 #include "base/file_path.h" |
13 #include "base/file_util.h" | 13 #include "base/file_util.h" |
14 #include "base/pickle.h" | 14 #include "base/pickle.h" |
15 #include "base/stl_util.h" | 15 #include "base/stl_util.h" |
16 #include "base/string_util.h" | 16 #include "base/string_util.h" |
17 #include "base/threading/thread.h" | 17 #include "base/threading/thread.h" |
18 #include "base/version.h" | 18 #include "base/version.h" |
19 #include "chrome/browser/extensions/extension_service.h" | 19 #include "chrome/browser/extensions/extension_service.h" |
20 #include "chrome/browser/profiles/profile.h" | 20 #include "chrome/browser/profiles/profile.h" |
21 #include "chrome/common/chrome_notification_types.h" | 21 #include "chrome/common/chrome_notification_types.h" |
22 #include "chrome/common/extensions/extension.h" | 22 #include "chrome/common/extensions/extension.h" |
23 #include "chrome/common/extensions/extension_file_util.h" | 23 #include "chrome/common/extensions/extension_file_util.h" |
24 #include "chrome/common/extensions/extension_message_bundle.h" | 24 #include "chrome/common/extensions/extension_message_bundle.h" |
25 #include "chrome/common/extensions/extension_resource.h" | 25 #include "chrome/common/extensions/extension_resource.h" |
26 #include "chrome/common/extensions/extension_set.h" | 26 #include "chrome/common/extensions/extension_set.h" |
27 #include "content/browser/renderer_host/render_process_host.h" | |
28 #include "content/public/browser/notification_service.h" | 27 #include "content/public/browser/notification_service.h" |
| 28 #include "content/public/browser/render_process_host.h" |
29 | 29 |
30 using content::BrowserThread; | 30 using content::BrowserThread; |
31 | 31 |
32 // Helper function to parse greasesmonkey headers | 32 // Helper function to parse greasesmonkey headers |
33 static bool GetDeclarationValue(const base::StringPiece& line, | 33 static bool GetDeclarationValue(const base::StringPiece& line, |
34 const base::StringPiece& prefix, | 34 const base::StringPiece& prefix, |
35 std::string* value) { | 35 std::string* value) { |
36 base::StringPiece::size_type index = line.find(prefix); | 36 base::StringPiece::size_type index = line.find(prefix); |
37 if (index == base::StringPiece::npos) | 37 if (index == base::StringPiece::npos) |
38 return false; | 38 return false; |
(...skipping 274 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
313 // While we were loading, there were further changes. Don't bother | 313 // While we were loading, there were further changes. Don't bother |
314 // notifying about these scripts and instead just immediately reload. | 314 // notifying about these scripts and instead just immediately reload. |
315 pending_load_ = false; | 315 pending_load_ = false; |
316 StartLoad(); | 316 StartLoad(); |
317 } else { | 317 } else { |
318 // We're no longer loading. | 318 // We're no longer loading. |
319 script_reloader_ = NULL; | 319 script_reloader_ = NULL; |
320 // We've got scripts ready to go. | 320 // We've got scripts ready to go. |
321 shared_memory_.swap(handle_deleter); | 321 shared_memory_.swap(handle_deleter); |
322 | 322 |
323 for (RenderProcessHost::iterator i(RenderProcessHost::AllHostsIterator()); | 323 for (content::RenderProcessHost::iterator i( |
| 324 content::RenderProcessHost::AllHostsIterator()); |
324 !i.IsAtEnd(); i.Advance()) { | 325 !i.IsAtEnd(); i.Advance()) { |
325 SendUpdate(i.GetCurrentValue(), handle); | 326 SendUpdate(i.GetCurrentValue(), handle); |
326 } | 327 } |
327 | 328 |
328 content::NotificationService::current()->Notify( | 329 content::NotificationService::current()->Notify( |
329 chrome::NOTIFICATION_USER_SCRIPTS_UPDATED, | 330 chrome::NOTIFICATION_USER_SCRIPTS_UPDATED, |
330 content::Source<Profile>(profile_), | 331 content::Source<Profile>(profile_), |
331 content::Details<base::SharedMemory>(handle)); | 332 content::Details<base::SharedMemory>(handle)); |
332 } | 333 } |
333 } | 334 } |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
373 } | 374 } |
374 user_scripts_ = new_user_scripts; | 375 user_scripts_ = new_user_scripts; |
375 should_start_load = true; | 376 should_start_load = true; |
376 | 377 |
377 // TODO(aa): Do we want to do something smarter for the scripts that have | 378 // TODO(aa): Do we want to do something smarter for the scripts that have |
378 // already been injected? | 379 // already been injected? |
379 | 380 |
380 break; | 381 break; |
381 } | 382 } |
382 case content::NOTIFICATION_RENDERER_PROCESS_CREATED: { | 383 case content::NOTIFICATION_RENDERER_PROCESS_CREATED: { |
383 RenderProcessHost* process = | 384 content::RenderProcessHost* process = |
384 content::Source<RenderProcessHost>(source).ptr(); | 385 content::Source<content::RenderProcessHost>(source).ptr(); |
385 Profile* profile = Profile::FromBrowserContext( | 386 Profile* profile = Profile::FromBrowserContext( |
386 process->browser_context()); | 387 process->GetBrowserContext()); |
387 if (!profile_->IsSameProfile(profile)) | 388 if (!profile_->IsSameProfile(profile)) |
388 return; | 389 return; |
389 if (ScriptsReady()) | 390 if (ScriptsReady()) |
390 SendUpdate(process, GetSharedMemory()); | 391 SendUpdate(process, GetSharedMemory()); |
391 break; | 392 break; |
392 } | 393 } |
393 default: | 394 default: |
394 DCHECK(false); | 395 DCHECK(false); |
395 } | 396 } |
396 | 397 |
397 if (should_start_load) { | 398 if (should_start_load) { |
398 if (script_reloader_) { | 399 if (script_reloader_) { |
399 pending_load_ = true; | 400 pending_load_ = true; |
400 } else { | 401 } else { |
401 StartLoad(); | 402 StartLoad(); |
402 } | 403 } |
403 } | 404 } |
404 } | 405 } |
405 | 406 |
406 void UserScriptMaster::StartLoad() { | 407 void UserScriptMaster::StartLoad() { |
407 if (!script_reloader_) | 408 if (!script_reloader_) |
408 script_reloader_ = new ScriptReloader(this); | 409 script_reloader_ = new ScriptReloader(this); |
409 | 410 |
410 script_reloader_->StartLoad(user_scripts_, extensions_info_); | 411 script_reloader_->StartLoad(user_scripts_, extensions_info_); |
411 } | 412 } |
412 | 413 |
413 void UserScriptMaster::SendUpdate(RenderProcessHost* process, | 414 void UserScriptMaster::SendUpdate(content::RenderProcessHost* process, |
414 base::SharedMemory* shared_memory) { | 415 base::SharedMemory* shared_memory) { |
415 Profile* profile = Profile::FromBrowserContext(process->browser_context()); | 416 Profile* profile = Profile::FromBrowserContext(process->GetBrowserContext()); |
416 // Make sure we only send user scripts to processes in our profile. | 417 // Make sure we only send user scripts to processes in our profile. |
417 if (!profile_->IsSameProfile(profile)) | 418 if (!profile_->IsSameProfile(profile)) |
418 return; | 419 return; |
419 | 420 |
420 // If the process is being started asynchronously, early return. We'll end up | 421 // If the process is being started asynchronously, early return. We'll end up |
421 // calling InitUserScripts when it's created which will call this again. | 422 // calling InitUserScripts when it's created which will call this again. |
422 base::ProcessHandle handle = process->GetHandle(); | 423 base::ProcessHandle handle = process->GetHandle(); |
423 if (!handle) | 424 if (!handle) |
424 return; | 425 return; |
425 | 426 |
426 base::SharedMemoryHandle handle_for_process; | 427 base::SharedMemoryHandle handle_for_process; |
427 if (!shared_memory->ShareToProcess(handle, &handle_for_process)) | 428 if (!shared_memory->ShareToProcess(handle, &handle_for_process)) |
428 return; // This can legitimately fail if the renderer asserts at startup. | 429 return; // This can legitimately fail if the renderer asserts at startup. |
429 | 430 |
430 if (base::SharedMemory::IsHandleValid(handle_for_process)) | 431 if (base::SharedMemory::IsHandleValid(handle_for_process)) |
431 process->Send(new ExtensionMsg_UpdateUserScripts(handle_for_process)); | 432 process->Send(new ExtensionMsg_UpdateUserScripts(handle_for_process)); |
432 } | 433 } |
OLD | NEW |