From 795454fb81af3ff894c42cc3da13bab1616af0fb Mon Sep 17 00:00:00 2001
From: Sam Gibson <sam@ifdown.net>
Date: Wed, 24 Jun 2015 11:30:08 -0700
Subject: [PATCH] Adds control message for HSTS headers

---
 components/net/resource_task.rs | 7 +++++++
 components/net_traits/lib.rs    | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/components/net/resource_task.rs b/components/net/resource_task.rs
index 41b23a2eb82..6968da6a94f 100644
--- a/components/net/resource_task.rs
+++ b/components/net/resource_task.rs
@@ -366,6 +366,13 @@ impl ResourceChannelManager {
               ControlMsg::GetCookiesForUrl(url, consumer, source) => {
                   consumer.send(self.resource_manager.cookie_storage.cookies_for_url(&url, source)).unwrap();
               }
+              ControlMsg::SetHSTSEntryForHost(host, include_subdomains, max_age) => {
+                  match HSTSEntry::new(host, include_subdomains, max_age) {
+                      Some(entry) => self.resource_manager.add_hsts_entry(entry),
+                      /// Invalid entries (e.g. IP's don't matter)
+                      None => ()
+                  }
+              }
               ControlMsg::Exit => {
                   break
               }
diff --git a/components/net_traits/lib.rs b/components/net_traits/lib.rs
index 620d6c39aaf..5da6357d333 100644
--- a/components/net_traits/lib.rs
+++ b/components/net_traits/lib.rs
@@ -126,6 +126,8 @@ pub enum ControlMsg {
     SetCookiesForUrl(Url, String, CookieSource),
     /// Retrieve the stored cookies for a given URL
     GetCookiesForUrl(Url, Sender<Option<String>>, CookieSource),
+    /// Store a domain's STS information
+    SetHSTSEntryForHost(String, bool, Option<u64>),
     Exit
 }