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 #ifndef CHROME_BROWSER_GUEST_VIEW_GUEST_VIEW_MANAGER_H_ | 5 #ifndef CHROME_BROWSER_GUEST_VIEW_GUEST_VIEW_MANAGER_H_ |
6 #define CHROME_BROWSER_GUEST_VIEW_GUEST_VIEW_MANAGER_H_ | 6 #define CHROME_BROWSER_GUEST_VIEW_GUEST_VIEW_MANAGER_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 | 9 |
10 #include "base/gtest_prod_util.h" | 10 #include "base/gtest_prod_util.h" |
11 #include "base/lazy_instance.h" | 11 #include "base/lazy_instance.h" |
12 #include "base/macros.h" | 12 #include "base/macros.h" |
13 #include "content/public/browser/browser_plugin_guest_manager.h" | 13 #include "content/public/browser/browser_plugin_guest_manager.h" |
14 #include "content/public/browser/site_instance.h" | 14 #include "content/public/browser/site_instance.h" |
15 #include "content/public/browser/web_contents.h" | 15 #include "content/public/browser/web_contents.h" |
16 | 16 |
17 class GuestViewBase; | 17 class GuestViewBase; |
| 18 class GuestViewManagerFactory; |
18 class GuestWebContentsObserver; | 19 class GuestWebContentsObserver; |
19 class GURL; | 20 class GURL; |
20 | 21 |
21 namespace content { | 22 namespace content { |
22 class BrowserContext; | 23 class BrowserContext; |
23 } // namespace content | 24 } // namespace content |
24 | 25 |
| 26 namespace guestview { |
| 27 class TestGuestViewManager; |
| 28 } // namespace guestview |
| 29 |
25 class GuestViewManager : public content::BrowserPluginGuestManager, | 30 class GuestViewManager : public content::BrowserPluginGuestManager, |
26 public base::SupportsUserData::Data { | 31 public base::SupportsUserData::Data { |
27 public: | 32 public: |
28 explicit GuestViewManager(content::BrowserContext* context); | 33 explicit GuestViewManager(content::BrowserContext* context); |
29 virtual ~GuestViewManager(); | 34 virtual ~GuestViewManager(); |
30 | 35 |
31 static GuestViewManager* FromBrowserContext(content::BrowserContext* context); | 36 static GuestViewManager* FromBrowserContext(content::BrowserContext* context); |
32 | 37 |
| 38 // Overrides factory for testing. Default (NULL) value indicates regular |
| 39 // (non-test) environment. |
| 40 static void set_factory_for_testing(GuestViewManagerFactory* factory) { |
| 41 GuestViewManager::factory_ = factory; |
| 42 } |
33 // Returns the guest WebContents associated with the given |guest_instance_id| | 43 // Returns the guest WebContents associated with the given |guest_instance_id| |
34 // if the provided |embedder_render_process_id| is allowed to access it. | 44 // if the provided |embedder_render_process_id| is allowed to access it. |
35 // If the embedder is not allowed access, the embedder will be killed, and | 45 // If the embedder is not allowed access, the embedder will be killed, and |
36 // this method will return NULL. If no WebContents exists with the given | 46 // this method will return NULL. If no WebContents exists with the given |
37 // instance ID, then NULL will also be returned. | 47 // instance ID, then NULL will also be returned. |
38 content::WebContents* GetGuestByInstanceIDSafely( | 48 content::WebContents* GetGuestByInstanceIDSafely( |
39 int guest_instance_id, | 49 int guest_instance_id, |
40 int embedder_render_process_id); | 50 int embedder_render_process_id); |
41 | 51 |
42 // BrowserPluginGuestManager implementation. | 52 // BrowserPluginGuestManager implementation. |
43 virtual content::WebContents* CreateGuest( | 53 virtual content::WebContents* CreateGuest( |
44 content::SiteInstance* embedder_site_instance, | 54 content::SiteInstance* embedder_site_instance, |
45 int instance_id, | 55 int instance_id, |
46 scoped_ptr<base::DictionaryValue> extra_params) OVERRIDE; | 56 scoped_ptr<base::DictionaryValue> extra_params) OVERRIDE; |
47 virtual int GetNextInstanceID() OVERRIDE; | 57 virtual int GetNextInstanceID() OVERRIDE; |
48 virtual void MaybeGetGuestByInstanceIDOrKill( | 58 virtual void MaybeGetGuestByInstanceIDOrKill( |
49 int guest_instance_id, | 59 int guest_instance_id, |
50 int embedder_render_process_id, | 60 int embedder_render_process_id, |
51 const GuestByInstanceIDCallback& callback) OVERRIDE; | 61 const GuestByInstanceIDCallback& callback) OVERRIDE; |
52 virtual bool ForEachGuest(content::WebContents* embedder_web_contents, | 62 virtual bool ForEachGuest(content::WebContents* embedder_web_contents, |
53 const GuestCallback& callback) OVERRIDE; | 63 const GuestCallback& callback) OVERRIDE; |
54 | 64 |
55 private: | 65 protected: |
56 friend class GuestViewBase; | 66 friend class GuestViewBase; |
57 friend class GuestWebContentsObserver; | 67 friend class GuestWebContentsObserver; |
58 friend class TestGuestViewManager; | 68 friend class guestview::TestGuestViewManager; |
59 FRIEND_TEST_ALL_PREFIXES(GuestViewManagerTest, AddRemove); | 69 FRIEND_TEST_ALL_PREFIXES(GuestViewManagerTest, AddRemove); |
60 | 70 |
61 void AddGuest(int guest_instance_id, | 71 // Can be overriden in tests. |
62 content::WebContents* guest_web_contents); | 72 virtual void AddGuest(int guest_instance_id, |
| 73 content::WebContents* guest_web_contents); |
63 | 74 |
64 void RemoveGuest(int guest_instance_id); | 75 void RemoveGuest(int guest_instance_id); |
65 | 76 |
66 content::SiteInstance* GetGuestSiteInstance( | 77 content::SiteInstance* GetGuestSiteInstance( |
67 const GURL& guest_site); | 78 const GURL& guest_site); |
68 | 79 |
69 content::WebContents* GetGuestByInstanceID( | 80 content::WebContents* GetGuestByInstanceID( |
70 int guest_instance_id, | 81 int guest_instance_id, |
71 int embedder_render_process_id); | 82 int embedder_render_process_id); |
72 | 83 |
73 bool CanEmbedderAccessInstanceIDMaybeKill( | 84 bool CanEmbedderAccessInstanceIDMaybeKill( |
74 int embedder_render_process_id, | 85 int embedder_render_process_id, |
75 int guest_instance_id); | 86 int guest_instance_id); |
76 | 87 |
77 bool CanEmbedderAccessInstanceID(int embedder_render_process_id, | 88 bool CanEmbedderAccessInstanceID(int embedder_render_process_id, |
78 int guest_instance_id); | 89 int guest_instance_id); |
79 | 90 |
80 // Returns true if |guest_instance_id| can be used to add a new guest to this | 91 // Returns true if |guest_instance_id| can be used to add a new guest to this |
81 // manager. | 92 // manager. |
82 // We disallow adding new guest with instance IDs that were previously removed | 93 // We disallow adding new guest with instance IDs that were previously removed |
83 // from this manager using RemoveGuest. | 94 // from this manager using RemoveGuest. |
84 bool CanUseGuestInstanceID(int guest_instance_id); | 95 bool CanUseGuestInstanceID(int guest_instance_id); |
85 | 96 |
86 static bool CanEmbedderAccessGuest(int embedder_render_process_id, | 97 static bool CanEmbedderAccessGuest(int embedder_render_process_id, |
87 GuestViewBase* guest); | 98 GuestViewBase* guest); |
88 | 99 |
| 100 // Static factory instance (always NULL for non-test). |
| 101 static GuestViewManagerFactory* factory_; |
| 102 |
89 // Contains guests' WebContents, mapping from their instance ids. | 103 // Contains guests' WebContents, mapping from their instance ids. |
90 typedef std::map<int, content::WebContents*> GuestInstanceMap; | 104 typedef std::map<int, content::WebContents*> GuestInstanceMap; |
91 GuestInstanceMap guest_web_contents_by_instance_id_; | 105 GuestInstanceMap guest_web_contents_by_instance_id_; |
92 | 106 |
93 int current_instance_id_; | 107 int current_instance_id_; |
94 | 108 |
95 // Any instance ID whose number not greater than this was removed via | 109 // Any instance ID whose number not greater than this was removed via |
96 // RemoveGuest. | 110 // RemoveGuest. |
97 // This is used so that we don't have store all removed instance IDs in | 111 // This is used so that we don't have store all removed instance IDs in |
98 // |removed_instance_ids_|. | 112 // |removed_instance_ids_|. |
99 int last_instance_id_removed_; | 113 int last_instance_id_removed_; |
100 // The remaining instance IDs that are greater than | 114 // The remaining instance IDs that are greater than |
101 // |last_instance_id_removed_| are kept here. | 115 // |last_instance_id_removed_| are kept here. |
102 std::set<int> removed_instance_ids_; | 116 std::set<int> removed_instance_ids_; |
103 | 117 |
104 content::BrowserContext* context_; | 118 content::BrowserContext* context_; |
105 | 119 |
106 DISALLOW_COPY_AND_ASSIGN(GuestViewManager); | 120 DISALLOW_COPY_AND_ASSIGN(GuestViewManager); |
107 }; | 121 }; |
108 | 122 |
109 #endif // CHROME_BROWSER_GUEST_VIEW_GUEST_VIEW_MANAGER_H_ | 123 #endif // CHROME_BROWSER_GUEST_VIEW_GUEST_VIEW_MANAGER_H_ |
OLD | NEW |