| 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 #include "content/browser/plugin_content_origin_whitelist.h" | 5 #include "content/browser/plugin_content_origin_whitelist.h" |
| 6 | 6 |
| 7 #include "content/common/frame_messages.h" | 7 #include "content/common/frame_messages.h" |
| 8 #include "content/public/browser/navigation_details.h" | 8 #include "content/public/browser/navigation_handle.h" |
| 9 #include "content/public/browser/render_frame_host.h" | 9 #include "content/public/browser/render_frame_host.h" |
| 10 #include "content/public/browser/web_contents.h" | 10 #include "content/public/browser/web_contents.h" |
| 11 | 11 |
| 12 namespace content { | 12 namespace content { |
| 13 | 13 |
| 14 PluginContentOriginWhitelist::PluginContentOriginWhitelist( | 14 PluginContentOriginWhitelist::PluginContentOriginWhitelist( |
| 15 WebContents* web_contents) | 15 WebContents* web_contents) |
| 16 : WebContentsObserver(web_contents) { | 16 : WebContentsObserver(web_contents) { |
| 17 } | 17 } |
| 18 | 18 |
| (...skipping 14 matching lines...) Expand all Loading... |
| 33 bool handled = true; | 33 bool handled = true; |
| 34 IPC_BEGIN_MESSAGE_MAP(PluginContentOriginWhitelist, message) | 34 IPC_BEGIN_MESSAGE_MAP(PluginContentOriginWhitelist, message) |
| 35 IPC_MESSAGE_HANDLER(FrameHostMsg_PluginContentOriginAllowed, | 35 IPC_MESSAGE_HANDLER(FrameHostMsg_PluginContentOriginAllowed, |
| 36 OnPluginContentOriginAllowed) | 36 OnPluginContentOriginAllowed) |
| 37 IPC_MESSAGE_UNHANDLED(handled = false) | 37 IPC_MESSAGE_UNHANDLED(handled = false) |
| 38 IPC_END_MESSAGE_MAP() | 38 IPC_END_MESSAGE_MAP() |
| 39 | 39 |
| 40 return handled; | 40 return handled; |
| 41 } | 41 } |
| 42 | 42 |
| 43 void PluginContentOriginWhitelist::DidNavigateMainFrame( | 43 void PluginContentOriginWhitelist::DidFinishNavigation( |
| 44 const LoadCommittedDetails& details, | 44 NavigationHandle* navigation_handle) { |
| 45 const FrameNavigateParams& params) { | 45 if (!navigation_handle->IsInMainFrame() || |
| 46 if (details.is_navigation_to_different_page()) { | 46 !navigation_handle->HasCommitted() || |
| 47 // We expect RenderFrames to clear their replicated whitelist independently. | 47 navigation_handle->IsSamePage()) { |
| 48 whitelist_.clear(); | 48 return; |
| 49 } | 49 } |
| 50 |
| 51 // We expect RenderFrames to clear their replicated whitelist independently. |
| 52 whitelist_.clear(); |
| 50 } | 53 } |
| 51 | 54 |
| 52 void PluginContentOriginWhitelist::OnPluginContentOriginAllowed( | 55 void PluginContentOriginWhitelist::OnPluginContentOriginAllowed( |
| 53 const url::Origin& content_origin) { | 56 const url::Origin& content_origin) { |
| 54 whitelist_.insert(content_origin); | 57 whitelist_.insert(content_origin); |
| 55 | 58 |
| 56 web_contents()->SendToAllFrames( | 59 web_contents()->SendToAllFrames( |
| 57 new FrameMsg_UpdatePluginContentOriginWhitelist( | 60 new FrameMsg_UpdatePluginContentOriginWhitelist( |
| 58 MSG_ROUTING_NONE, whitelist_)); | 61 MSG_ROUTING_NONE, whitelist_)); |
| 59 } | 62 } |
| 60 | 63 |
| 61 } // namespace content | 64 } // namespace content |
| OLD | NEW |