Index: chrome/browser/extensions/api/messaging/extension_message_port.cc |
diff --git a/chrome/browser/extensions/api/messaging/extension_message_port.cc b/chrome/browser/extensions/api/messaging/extension_message_port.cc |
index 3225143c9fb01729dbd1af9d8defaa5e6e2189de..bdb95669eaf0034fcf72c82a2fe39ace6cf5f77f 100644 |
--- a/chrome/browser/extensions/api/messaging/extension_message_port.cc |
+++ b/chrome/browser/extensions/api/messaging/extension_message_port.cc |
@@ -99,6 +99,7 @@ ExtensionMessagePort::ExtensionMessagePort( |
extension_id_(extension_id), |
browser_context_(extension_process->GetBrowserContext()), |
extension_process_(extension_process), |
+ opener_tab_id_(-1), |
did_create_port_(false), |
background_host_ptr_(nullptr), |
frame_tracker_(new FrameTracker(this)) { |
@@ -121,6 +122,7 @@ ExtensionMessagePort::ExtensionMessagePort( |
extension_id_(extension_id), |
browser_context_(rfh->GetProcess()->GetBrowserContext()), |
extension_process_(nullptr), |
+ opener_tab_id_(-1), |
did_create_port_(false), |
background_host_ptr_(nullptr), |
frame_tracker_(new FrameTracker(this)) { |
@@ -158,6 +160,16 @@ ExtensionMessagePort::ExtensionMessagePort( |
ExtensionMessagePort::~ExtensionMessagePort() {} |
+void ExtensionMessagePort::RevalidatePort() { |
+ // Only opener ports need to be revalidated, because these are created in the |
+ // renderer before the browser knows about them. |
+ DCHECK(!extension_process_); |
+ DCHECK_LE(frames_.size(), 1U); |
+ |
+ SendToPort(base::WrapUnique( |
+ new ExtensionMsg_CheckHasMessagePort(MSG_ROUTING_NONE, port_id_))); |
+} |
+ |
void ExtensionMessagePort::RemoveCommonFrames(const MessagePort& port) { |
// Avoid overlap in the set of frames to make sure that it does not matter |
// when UnregisterFrame is called. |
@@ -190,8 +202,10 @@ void ExtensionMessagePort::DispatchOnConnect( |
const GURL& source_url, |
const std::string& tls_channel_id) { |
ExtensionMsg_TabConnectionInfo source; |
- if (source_tab) |
+ if (source_tab) { |
source.tab.Swap(source_tab.get()); |
+ source.tab.GetInteger("id", &opener_tab_id_); |
+ } |
source.frame_id = source_frame_id; |
ExtensionMsg_ExternalConnectionInfo info; |
@@ -212,8 +226,8 @@ void ExtensionMessagePort::DispatchOnDisconnect( |
} |
void ExtensionMessagePort::DispatchOnMessage(const Message& message) { |
- SendToPort(base::WrapUnique( |
- new ExtensionMsg_DeliverMessage(MSG_ROUTING_NONE, port_id_, message))); |
+ SendToPort(base::WrapUnique(new ExtensionMsg_DeliverMessage( |
+ MSG_ROUTING_NONE, port_id_, opener_tab_id_, message))); |
} |
void ExtensionMessagePort::IncrementLazyKeepaliveCount() { |