OLD | NEW |
| (Empty) |
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/renderer/extensions/chrome_v8_context_set.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "base/message_loop/message_loop.h" | |
9 #include "base/tracked_objects.h" | |
10 #include "base/values.h" | |
11 #include "chrome/common/url_constants.h" | |
12 #include "chrome/renderer/extensions/chrome_v8_context.h" | |
13 #include "content/public/renderer/render_thread.h" | |
14 #include "content/public/renderer/render_view.h" | |
15 #include "extensions/common/constants.h" | |
16 #include "extensions/common/extension.h" | |
17 #include "third_party/WebKit/public/platform/WebURL.h" | |
18 #include "third_party/WebKit/public/platform/WebURLRequest.h" | |
19 #include "third_party/WebKit/public/web/WebDocument.h" | |
20 #include "third_party/WebKit/public/web/WebFrame.h" | |
21 #include "third_party/WebKit/public/web/WebView.h" | |
22 #include "v8/include/v8.h" | |
23 | |
24 using content::RenderThread; | |
25 | |
26 namespace extensions { | |
27 | |
28 ChromeV8ContextSet::ChromeV8ContextSet() { | |
29 } | |
30 ChromeV8ContextSet::~ChromeV8ContextSet() { | |
31 } | |
32 | |
33 int ChromeV8ContextSet::size() const { | |
34 return static_cast<int>(contexts_.size()); | |
35 } | |
36 | |
37 void ChromeV8ContextSet::Add(ChromeV8Context* context) { | |
38 #if DCHECK_IS_ON | |
39 // It's OK to insert the same context twice, but we should only ever have | |
40 // one ChromeV8Context per v8::Context. | |
41 for (ContextSet::iterator iter = contexts_.begin(); iter != contexts_.end(); | |
42 ++iter) { | |
43 ChromeV8Context* candidate = *iter; | |
44 if (candidate != context) | |
45 DCHECK(candidate->v8_context() != context->v8_context()); | |
46 } | |
47 #endif | |
48 contexts_.insert(context); | |
49 } | |
50 | |
51 void ChromeV8ContextSet::Remove(ChromeV8Context* context) { | |
52 if (contexts_.erase(context)) { | |
53 context->Invalidate(); | |
54 base::MessageLoop::current()->DeleteSoon(FROM_HERE, context); | |
55 } | |
56 } | |
57 | |
58 ChromeV8ContextSet::ContextSet ChromeV8ContextSet::GetAll() const { | |
59 return contexts_; | |
60 } | |
61 | |
62 ChromeV8Context* ChromeV8ContextSet::GetCurrent() const { | |
63 v8::Isolate* isolate = v8::Isolate::GetCurrent(); | |
64 return isolate->InContext() ? GetByV8Context(isolate->GetCurrentContext()) | |
65 : NULL; | |
66 } | |
67 | |
68 ChromeV8Context* ChromeV8ContextSet::GetCalling() const { | |
69 v8::Isolate* isolate = v8::Isolate::GetCurrent(); | |
70 v8::Local<v8::Context> calling = isolate->GetCallingContext(); | |
71 return calling.IsEmpty() ? NULL : GetByV8Context(calling); | |
72 } | |
73 | |
74 ChromeV8Context* ChromeV8ContextSet::GetByV8Context( | |
75 v8::Handle<v8::Context> v8_context) const { | |
76 for (ContextSet::const_iterator iter = contexts_.begin(); | |
77 iter != contexts_.end(); ++iter) { | |
78 if ((*iter)->v8_context() == v8_context) | |
79 return *iter; | |
80 } | |
81 | |
82 return NULL; | |
83 } | |
84 | |
85 void ChromeV8ContextSet::ForEach( | |
86 const std::string& extension_id, | |
87 content::RenderView* render_view, | |
88 const base::Callback<void(ChromeV8Context*)>& callback) const { | |
89 // We copy the context list, because calling into javascript may modify it | |
90 // out from under us. | |
91 ContextSet contexts = GetAll(); | |
92 | |
93 for (ContextSet::iterator it = contexts.begin(); it != contexts.end(); | |
94 ++it) { | |
95 ChromeV8Context* context = *it; | |
96 | |
97 // For the same reason as above, contexts may become invalid while we run. | |
98 if (!context->is_valid()) | |
99 continue; | |
100 | |
101 if (!extension_id.empty()) { | |
102 const Extension* extension = context->extension(); | |
103 if (!extension || (extension_id != extension->id())) | |
104 continue; | |
105 } | |
106 | |
107 content::RenderView* context_render_view = context->GetRenderView(); | |
108 if (!context_render_view) | |
109 continue; | |
110 | |
111 if (render_view && render_view != context_render_view) | |
112 continue; | |
113 | |
114 callback.Run(context); | |
115 } | |
116 } | |
117 | |
118 ChromeV8ContextSet::ContextSet ChromeV8ContextSet::OnExtensionUnloaded( | |
119 const std::string& extension_id) { | |
120 ContextSet contexts = GetAll(); | |
121 ContextSet removed; | |
122 | |
123 // Clean up contexts belonging to the unloaded extension. This is done so | |
124 // that content scripts (which remain injected into the page) don't continue | |
125 // receiving events and sending messages. | |
126 for (ContextSet::iterator it = contexts.begin(); it != contexts.end(); | |
127 ++it) { | |
128 if ((*it)->extension() && (*it)->extension()->id() == extension_id) { | |
129 (*it)->DispatchOnUnloadEvent(); | |
130 removed.insert(*it); | |
131 Remove(*it); | |
132 } | |
133 } | |
134 | |
135 return removed; | |
136 } | |
137 | |
138 } // namespace extensions | |
OLD | NEW |