Handle origin when blocking content

servo · May 30, 2016 · 6a59e9e · 6a59e9e
1 parent 8e61ae3
commit 6a59e9e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/components/net/http_loader.rs b/components/net/http_loader.rs
@@ -862,10 +862,15 @@ pub fn load<A, B>(load_data: &LoadData,
 
         let mut block_cookies = false;
         if let Some(ref rules) = *http_state.blocked_content {
+            let same_origin =
+                load_data.referrer_url.as_ref()
+                         .map(|url| url.origin() == doc_url.origin())
+                         .unwrap_or(false);
+            let load_type = if same_origin { LoadType::FirstParty } else { LoadType::ThirdParty };
             let actions = process_rules_for_request(rules, &CBRequest {
                 url: &doc_url,
                 resource_type: to_resource_type(&load_data.context),
-                load_type: LoadType::FirstParty, //FIXME need request origin
+                load_type: load_type,
             });
             for action in actions {
                 match action {