Mirrors/servo
1
0
Fork 0
mirror of https://github.com/servo/servo.git synced 2025-08-05 21:50:18 +01:00
Code Issues Projects Releases Packages Wiki Activity

General cleanup for HSTS/Cookie Arc

Browse source
This commit is contained in:
Sam Gibson 2015-09-19 18:23:03 +12:00
parent 6f573d5d44
commit 6d20b388f6
3 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

26
components/net/http_loader.rs
View file

@ -323,7 +323,7 @@ fn set_default_accept(headers: &mut Headers) {
} }
} }
fn set_request_cookies(url: Url, headers: &mut Headers, cookie_jar: Arc<RwLock<CookieStorage>>) { fn set_request_cookies(url: Url, headers: &mut Headers, cookie_jar: &Arc<RwLock<CookieStorage>>) {
let mut cookie_jar = cookie_jar.write().unwrap(); let mut cookie_jar = cookie_jar.write().unwrap();
if let Some(cookie_list) = cookie_jar.cookies_for_url(&url, CookieSource::HTTP) { if let Some(cookie_list) = cookie_jar.cookies_for_url(&url, CookieSource::HTTP) {
let mut v = Vec::new(); let mut v = Vec::new();
@ -332,7 +332,7 @@ fn set_request_cookies(url: Url, headers: &mut Headers, cookie_jar: Arc<RwLock<C
} }
} }
fn set_cookie_for_url(cookie_jar: Arc<RwLock<CookieStorage>>, fn set_cookie_for_url(cookie_jar: &Arc<RwLock<CookieStorage>>,
request: Url, request: Url,
cookie_val: String) { cookie_val: String) {
let mut cookie_jar = cookie_jar.write().unwrap(); let mut cookie_jar = cookie_jar.write().unwrap();
@ -348,11 +348,11 @@ fn set_cookie_for_url(cookie_jar: Arc<RwLock<CookieStorage>>,
} }
} }
fn set_cookies_from_response(url: Url, response: &HttpResponse, cookie_jar: Arc<RwLock<CookieStorage>>) { fn set_cookies_from_response(url: Url, response: &HttpResponse, cookie_jar: &Arc<RwLock<CookieStorage>>) {
if let Some(cookies) = response.headers().get_raw("set-cookie") { if let Some(cookies) = response.headers().get_raw("set-cookie") {
for cookie in cookies.iter() { for cookie in cookies.iter() {
if let Ok(cookie_value) = String::from_utf8(cookie.clone()) { if let Ok(cookie_value) = String::from_utf8(cookie.clone()) {
set_cookie_for_url(cookie_jar.clone(), set_cookie_for_url(&cookie_jar,
url.clone(), url.clone(),
cookie_value); cookie_value);
} }
@ -360,7 +360,7 @@ fn set_cookies_from_response(url: Url, response: &HttpResponse, cookie_jar: Arc<
} }
} }
fn update_sts_list_from_response(url: &Url, response: &HttpResponse, hsts_list: Arc<RwLock<HSTSList>>) { fn update_sts_list_from_response(url: &Url, response: &HttpResponse, hsts_list: &Arc<RwLock<HSTSList>>) {
if url.scheme != "https" { if url.scheme != "https" {
return; return;
} }
@ -466,6 +466,13 @@ fn send_response_to_devtools(devtools_chan: Option<Sender<DevtoolsControlMsg>>,
} }
} }
fn request_must_be_secured(url: &Url, hsts_list: &Arc<RwLock<HSTSList>>) -> bool {
match url.domain() {
Some(domain) => hsts_list.read().unwrap().is_host_secure(domain),
None => false
}
}
pub fn load<A>(load_data: LoadData, pub fn load<A>(load_data: LoadData,
hsts_list: Arc<RwLock<HSTSList>>, hsts_list: Arc<RwLock<HSTSList>>,
cookie_jar: Arc<RwLock<CookieStorage>>, cookie_jar: Arc<RwLock<CookieStorage>>,
@ -500,8 +507,7 @@ pub fn load<A>(load_data: LoadData,
loop { loop {
iters = iters + 1; iters = iters + 1;
// if &*url.scheme == "http" && request_must_be_secured(&url, &resource_mgr_chan) { if &*url.scheme == "http" && request_must_be_secured(&url, &hsts_list) {
if &*url.scheme == "http" && hsts_list.read().unwrap().is_host_secure(url.domain().unwrap()) {
info!("{} is in the strict transport security list, requesting secure host", url); info!("{} is in the strict transport security list, requesting secure host", url);
url = secure_url(&url); url = secure_url(&url);
} }
@ -540,7 +546,7 @@ pub fn load<A>(load_data: LoadData,
set_default_accept(&mut request_headers); set_default_accept(&mut request_headers);
set_default_accept_encoding(&mut request_headers); set_default_accept_encoding(&mut request_headers);
set_request_cookies(doc_url.clone(), &mut request_headers, cookie_jar.clone()); set_request_cookies(doc_url.clone(), &mut request_headers, &cookie_jar);
let request_id = uuid::Uuid::new_v4().to_simple_string(); let request_id = uuid::Uuid::new_v4().to_simple_string();
@ -618,8 +624,8 @@ pub fn load<A>(load_data: LoadData,
} }
} }
set_cookies_from_response(doc_url.clone(), &response, cookie_jar.clone()); set_cookies_from_response(doc_url.clone(), &response, &cookie_jar);
update_sts_list_from_response(&url, &response, hsts_list.clone()); update_sts_list_from_response(&url, &response, &hsts_list);
// --- Loop if there's a redirect // --- Loop if there's a redirect
if response.status().class() == StatusClass::Redirection { if response.status().class() == StatusClass::Redirection {

4
components/net/resource_task.rs
View file

@ -173,7 +173,7 @@ impl ResourceChannelManager {
self.resource_manager.set_cookies_for_url(request, cookie_list, source) self.resource_manager.set_cookies_for_url(request, cookie_list, source)
} }
ControlMsg::GetCookiesForUrl(url, consumer, source) => { ControlMsg::GetCookiesForUrl(url, consumer, source) => {
let cookie_jar = self.resource_manager.cookie_storage.clone(); let ref cookie_jar = self.resource_manager.cookie_storage;
let mut cookie_jar = cookie_jar.write().unwrap(); let mut cookie_jar = cookie_jar.write().unwrap();
consumer.send(cookie_jar.cookies_for_url(&url, source)).unwrap(); consumer.send(cookie_jar.cookies_for_url(&url, source)).unwrap();
} }
@ -215,7 +215,7 @@ impl ResourceManager {
if let Ok(SetCookie(cookies)) = header { if let Ok(SetCookie(cookies)) = header {
for bare_cookie in cookies { for bare_cookie in cookies {
if let Some(cookie) = cookie::Cookie::new_wrapped(bare_cookie, &request, source) { if let Some(cookie) = cookie::Cookie::new_wrapped(bare_cookie, &request, source) {
let cookie_jar = self.cookie_storage.clone(); let ref cookie_jar = self.cookie_storage;
let mut cookie_jar = cookie_jar.write().unwrap(); let mut cookie_jar = cookie_jar.write().unwrap();
cookie_jar.push(cookie, source); cookie_jar.push(cookie, source);
} }

2
tests/unit/net/http_loader.rs
View file

@ -481,7 +481,7 @@ fn test_load_sets_requests_cookies_header_for_url_by_getting_cookies_from_the_re
{ {
let mut cookie_jar = cookie_jar.write().unwrap(); let mut cookie_jar = cookie_jar.write().unwrap();
let cookie_url = Url::parse("http://mozilla.com").unwrap(); let cookie_url = url.clone();
let cookie = Cookie::new_wrapped( let cookie = Cookie::new_wrapped(
cookie_rs::Cookie::parse("mozillaIs=theBest").unwrap(), cookie_rs::Cookie::parse("mozillaIs=theBest").unwrap(),
&cookie_url, &cookie_url,