post-message: remove unnecessary call to constellation

gterzian · gterzian · commit beffc43ef827 · 2020-07-04T18:49:30.000+08:00
diff --git a/components/constellation/constellation.rs b/components/constellation/constellation.rs
@@ -4149,14 +4149,34 @@ where
             },
             Some(browsing_context) => browsing_context.pipeline_id,
         };
-        let source_browsing_context = match self.pipelines.get(&source_pipeline) {
-            Some(pipeline) => pipeline.top_level_browsing_context_id,
-            None => return warn!("PostMessage from closed pipeline {:?}", source_pipeline),
-        };
+        let (source_top_level_browsing_context_id, source_browsing_context_id) =
+            match self.pipelines.get(&source_pipeline) {
+                Some(pipeline) => (
+                    pipeline.top_level_browsing_context_id,
+                    pipeline.browsing_context_id,
+                ),
+                None => return warn!("PostMessage from closed pipeline {:?}", source_pipeline),
+            };
+        let source_parent_pipeline_id =
+            match self.browsing_contexts.get(&source_browsing_context_id) {
+                Some(bc) => bc.parent_pipeline_id,
+                None => return warn!("PostMessage from closed pipeline {:?}", source_pipeline),
+            };
+
+        let source_parent = source_parent_pipeline_id.and_then(|parent_id| {
+            let parent_bc = match self.pipelines.get(&parent_id) {
+                Some(pipeline) => pipeline.browsing_context_id,
+                None => return None,
+            };
+            Some(parent_bc)
+        });
+
         let msg = ConstellationControlMsg::PostMessage {
             target: pipeline_id,
+            source_parent,
             source: source_pipeline,
-            source_browsing_context: source_browsing_context,
+            source_browsing_context: source_browsing_context_id,
+            source_top_level_browsing_context: source_top_level_browsing_context_id,
             target_origin: origin,
             source_origin,
             data,
diff --git a/components/script/script_thread.rs b/components/script/script_thread.rs
@@ -1951,15 +1951,19 @@ impl ScriptThread {
             ),
             ConstellationControlMsg::PostMessage {
                 target: target_pipeline_id,
+                source_parent,
                 source: source_pipeline_id,
                 source_browsing_context,
+                source_top_level_browsing_context,
                 target_origin: origin,
                 source_origin,
                 data,
             } => self.handle_post_message_msg(
                 target_pipeline_id,
+                source_parent,
                 source_pipeline_id,
                 source_browsing_context,
+                source_top_level_browsing_context,
                 origin,
                 source_origin,
                 data,
@@ -2605,8 +2609,10 @@ impl ScriptThread {
     fn handle_post_message_msg(
         &self,
         pipeline_id: PipelineId,
+        source_parent: Option<BrowsingContextId>,
         source_pipeline_id: PipelineId,
-        source_browsing_context: TopLevelBrowsingContextId,
+        source_browsing_context: BrowsingContextId,
+        source_top_level_browsing_context: TopLevelBrowsingContextId,
         origin: Option<ImmutableOrigin>,
         source_origin: ImmutableOrigin,
         data: StructuredSerializedData,
@@ -2619,8 +2625,9 @@ impl ScriptThread {
                 // send the required info as part of postmessage instead.
                 let source = match self.remote_window_proxy(
                     &*window.global(),
+                    source_top_level_browsing_context,
                     source_browsing_context,
-                    source_pipeline_id,
+                    source_parent,
                     None,
                 ) {
                     None => {
@@ -3082,20 +3089,20 @@ impl ScriptThread {
         &self,
         global_to_clone: &GlobalScope,
         top_level_browsing_context_id: TopLevelBrowsingContextId,
-        pipeline_id: PipelineId,
+        browsing_context_id: BrowsingContextId,
+        parent: Option<BrowsingContextId>,
         opener: Option<BrowsingContextId>,
     ) -> Option<DomRoot<WindowProxy>> {
-        let (browsing_context_id, parent_pipeline_id) =
-            self.ask_constellation_for_browsing_context_info(pipeline_id)?;
         if let Some(window_proxy) = self.window_proxies.borrow().get(&browsing_context_id) {
             return Some(DomRoot::from_ref(window_proxy));
         }
 
-        let parent_browsing_context = parent_pipeline_id.and_then(|parent_id| {
+        let parent_browsing_context = parent.and_then(|parent_bc| {
             self.remote_window_proxy(
                 global_to_clone,
                 top_level_browsing_context_id,
-                parent_id,
+                parent_bc,
+                None,
                 opener,
             )
         });
@@ -3147,12 +3154,18 @@ impl ScriptThread {
         });
         let parent_browsing_context = match (parent_info, iframe.as_ref()) {
             (_, Some(iframe)) => Some(window_from_node(&**iframe).window_proxy()),
-            (Some(parent_id), _) => self.remote_window_proxy(
-                window.upcast(),
-                top_level_browsing_context_id,
-                parent_id,
-                opener,
-            ),
+            (Some(parent_id), _) => {
+                let (browsing_context_id, _) = self
+                    .ask_constellation_for_browsing_context_info(parent_id)
+                    .expect("Couldn't get bc info for parent.");
+                self.remote_window_proxy(
+                    window.upcast(),
+                    top_level_browsing_context_id,
+                    browsing_context_id,
+                    None,
+                    opener,
+                )
+            },
             _ => None,
         };
 
diff --git a/components/script_traits/lib.rs b/components/script_traits/lib.rs
@@ -340,10 +340,14 @@ pub enum ConstellationControlMsg {
     PostMessage {
         /// The target of the message.
         target: PipelineId,
+        /// The parent of the source, if any.
+        source_parent: Option<BrowsingContextId>,
         /// The source of the message.
         source: PipelineId,
+        /// The browsing context associated with the source pipeline.
+        source_browsing_context: BrowsingContextId,
         /// The top level browsing context associated with the source pipeline.
-        source_browsing_context: TopLevelBrowsingContextId,
+        source_top_level_browsing_context: TopLevelBrowsingContextId,
         /// The expected origin of the target.
         target_origin: Option<ImmutableOrigin>,
         /// The source origin of the message.
