mirror of
https://github.com/servo/servo.git
synced 2025-08-03 20:50:07 +01:00
Added global state struct
Replace hsts_list, auth_cache, and cookie_jar with http_state Reverted a few changes Moved http_state back to factory Removed unnecessary Arc Removed Arc for http_state
This commit is contained in:
parent
7f944afee8
commit
b09570b2b6
3 changed files with 122 additions and 204 deletions
|
@ -78,9 +78,7 @@ pub fn create_http_connector() -> Arc<Pool<Connector>> {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn factory(user_agent: String,
|
pub fn factory(user_agent: String,
|
||||||
hsts_list: Arc<RwLock<HSTSList>>,
|
http_state: HttpState,
|
||||||
cookie_jar: Arc<RwLock<CookieStorage>>,
|
|
||||||
auth_cache: Arc<RwLock<HashMap<Url, AuthCacheEntry>>>,
|
|
||||||
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
||||||
connector: Arc<Pool<Connector>>)
|
connector: Arc<Pool<Connector>>)
|
||||||
-> Box<FnBox(LoadData,
|
-> Box<FnBox(LoadData,
|
||||||
|
@ -93,9 +91,7 @@ pub fn factory(user_agent: String,
|
||||||
senders,
|
senders,
|
||||||
classifier,
|
classifier,
|
||||||
connector,
|
connector,
|
||||||
hsts_list,
|
http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
devtools_chan,
|
devtools_chan,
|
||||||
cancel_listener,
|
cancel_listener,
|
||||||
user_agent)
|
user_agent)
|
||||||
|
@ -126,13 +122,27 @@ fn inner_url(url: &Url) -> Url {
|
||||||
Url::parse(inner_url).unwrap()
|
Url::parse(inner_url).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct HttpState {
|
||||||
|
pub hsts_list: Arc<RwLock<HSTSList>>,
|
||||||
|
pub cookie_jar: Arc<RwLock<CookieStorage>>,
|
||||||
|
pub auth_cache: Arc<RwLock<HashMap<Url, AuthCacheEntry>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl HttpState {
|
||||||
|
pub fn new() -> HttpState {
|
||||||
|
HttpState {
|
||||||
|
hsts_list: Arc::new(RwLock::new(HSTSList::new())),
|
||||||
|
cookie_jar: Arc::new(RwLock::new(CookieStorage::new())),
|
||||||
|
auth_cache: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn load_for_consumer(load_data: LoadData,
|
fn load_for_consumer(load_data: LoadData,
|
||||||
start_chan: LoadConsumer,
|
start_chan: LoadConsumer,
|
||||||
classifier: Arc<MIMEClassifier>,
|
classifier: Arc<MIMEClassifier>,
|
||||||
connector: Arc<Pool<Connector>>,
|
connector: Arc<Pool<Connector>>,
|
||||||
hsts_list: Arc<RwLock<HSTSList>>,
|
http_state: HttpState,
|
||||||
cookie_jar: Arc<RwLock<CookieStorage>>,
|
|
||||||
auth_cache: Arc<RwLock<HashMap<Url, AuthCacheEntry>>>,
|
|
||||||
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
||||||
cancel_listener: CancellationListener,
|
cancel_listener: CancellationListener,
|
||||||
user_agent: String) {
|
user_agent: String) {
|
||||||
|
@ -141,8 +151,7 @@ fn load_for_consumer(load_data: LoadData,
|
||||||
connector: connector,
|
connector: connector,
|
||||||
};
|
};
|
||||||
let context = load_data.context.clone();
|
let context = load_data.context.clone();
|
||||||
match load::<WrappedHttpRequest>(load_data, hsts_list,
|
match load::<WrappedHttpRequest>(load_data, &http_state,
|
||||||
cookie_jar, auth_cache,
|
|
||||||
devtools_chan, &factory,
|
devtools_chan, &factory,
|
||||||
user_agent, &cancel_listener) {
|
user_agent, &cancel_listener) {
|
||||||
Err(LoadError::UnsupportedScheme(url)) => {
|
Err(LoadError::UnsupportedScheme(url)) => {
|
||||||
|
@ -686,9 +695,7 @@ pub fn obtain_response<A>(request_factory: &HttpRequestFactory<R=A>,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn load<A>(load_data: LoadData,
|
pub fn load<A>(load_data: LoadData,
|
||||||
hsts_list: Arc<RwLock<HSTSList>>,
|
http_state: &HttpState,
|
||||||
cookie_jar: Arc<RwLock<CookieStorage>>,
|
|
||||||
auth_cache: Arc<RwLock<HashMap<Url, AuthCacheEntry>>>,
|
|
||||||
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
||||||
request_factory: &HttpRequestFactory<R=A>,
|
request_factory: &HttpRequestFactory<R=A>,
|
||||||
user_agent: String,
|
user_agent: String,
|
||||||
|
@ -721,7 +728,7 @@ pub fn load<A>(load_data: LoadData,
|
||||||
loop {
|
loop {
|
||||||
iters = iters + 1;
|
iters = iters + 1;
|
||||||
|
|
||||||
if &*doc_url.scheme == "http" && request_must_be_secured(&doc_url, &hsts_list) {
|
if &*doc_url.scheme == "http" && request_must_be_secured(&doc_url, &http_state.hsts_list) {
|
||||||
info!("{} is in the strict transport security list, requesting secure host", doc_url);
|
info!("{} is in the strict transport security list, requesting secure host", doc_url);
|
||||||
doc_url = secure_url(&doc_url);
|
doc_url = secure_url(&doc_url);
|
||||||
}
|
}
|
||||||
|
@ -755,14 +762,14 @@ pub fn load<A>(load_data: LoadData,
|
||||||
let request_id = uuid::Uuid::new_v4().to_simple_string();
|
let request_id = uuid::Uuid::new_v4().to_simple_string();
|
||||||
|
|
||||||
modify_request_headers(&mut request_headers, &doc_url,
|
modify_request_headers(&mut request_headers, &doc_url,
|
||||||
&user_agent, &cookie_jar,
|
&user_agent, &http_state.cookie_jar,
|
||||||
&auth_cache, &load_data);
|
&http_state.auth_cache, &load_data);
|
||||||
|
|
||||||
let response = try!(obtain_response(request_factory, &doc_url, &method, &request_headers,
|
let response = try!(obtain_response(request_factory, &doc_url, &method, &request_headers,
|
||||||
&cancel_listener, &load_data.data, &load_data.method,
|
&cancel_listener, &load_data.data, &load_data.method,
|
||||||
&load_data.pipeline_id, iters, &devtools_chan, &request_id));
|
&load_data.pipeline_id, iters, &devtools_chan, &request_id));
|
||||||
|
|
||||||
process_response_headers(&response, &doc_url, &cookie_jar, &hsts_list, &load_data);
|
process_response_headers(&response, &doc_url, &http_state.cookie_jar, &http_state.hsts_list, &load_data);
|
||||||
|
|
||||||
// --- Loop if there's a redirect
|
// --- Loop if there's a redirect
|
||||||
if response.status().class() == StatusClass::Redirection {
|
if response.status().class() == StatusClass::Redirection {
|
||||||
|
|
|
@ -11,7 +11,7 @@ use data_loader;
|
||||||
use devtools_traits::{DevtoolsControlMsg};
|
use devtools_traits::{DevtoolsControlMsg};
|
||||||
use file_loader;
|
use file_loader;
|
||||||
use hsts::{HSTSList, preload_hsts_domains};
|
use hsts::{HSTSList, preload_hsts_domains};
|
||||||
use http_loader::{self, Connector, create_http_connector};
|
use http_loader::{self, Connector, create_http_connector, HttpState};
|
||||||
use hyper::client::pool::Pool;
|
use hyper::client::pool::Pool;
|
||||||
use hyper::header::{ContentType, Header, SetCookie};
|
use hyper::header::{ContentType, Header, SetCookie};
|
||||||
use hyper::mime::{Mime, SubLevel, TopLevel};
|
use hyper::mime::{Mime, SubLevel, TopLevel};
|
||||||
|
@ -185,7 +185,7 @@ impl ResourceChannelManager {
|
||||||
ControlMsg::SetCookiesForUrl(request, cookie_list, source) =>
|
ControlMsg::SetCookiesForUrl(request, cookie_list, source) =>
|
||||||
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;
|
let cookie_jar = &self.resource_manager.cookie_jar;
|
||||||
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();
|
||||||
}
|
}
|
||||||
|
@ -276,7 +276,7 @@ pub struct AuthCacheEntry {
|
||||||
|
|
||||||
pub struct ResourceManager {
|
pub struct ResourceManager {
|
||||||
user_agent: String,
|
user_agent: String,
|
||||||
cookie_storage: Arc<RwLock<CookieStorage>>,
|
cookie_jar: Arc<RwLock<CookieStorage>>,
|
||||||
auth_cache: Arc<RwLock<HashMap<Url, AuthCacheEntry>>>,
|
auth_cache: Arc<RwLock<HashMap<Url, AuthCacheEntry>>>,
|
||||||
mime_classifier: Arc<MIMEClassifier>,
|
mime_classifier: Arc<MIMEClassifier>,
|
||||||
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
devtools_chan: Option<Sender<DevtoolsControlMsg>>,
|
||||||
|
@ -292,7 +292,7 @@ impl ResourceManager {
|
||||||
devtools_channel: Option<Sender<DevtoolsControlMsg>>) -> ResourceManager {
|
devtools_channel: Option<Sender<DevtoolsControlMsg>>) -> ResourceManager {
|
||||||
ResourceManager {
|
ResourceManager {
|
||||||
user_agent: user_agent,
|
user_agent: user_agent,
|
||||||
cookie_storage: Arc::new(RwLock::new(CookieStorage::new())),
|
cookie_jar: Arc::new(RwLock::new(CookieStorage::new())),
|
||||||
auth_cache: Arc::new(RwLock::new(HashMap::new())),
|
auth_cache: Arc::new(RwLock::new(HashMap::new())),
|
||||||
mime_classifier: Arc::new(MIMEClassifier::new()),
|
mime_classifier: Arc::new(MIMEClassifier::new()),
|
||||||
devtools_chan: devtools_channel,
|
devtools_chan: devtools_channel,
|
||||||
|
@ -308,7 +308,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;
|
let cookie_jar = &self.cookie_jar;
|
||||||
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);
|
||||||
}
|
}
|
||||||
|
@ -344,13 +344,17 @@ impl ResourceManager {
|
||||||
let cancel_listener = CancellationListener::new(cancel_resource);
|
let cancel_listener = CancellationListener::new(cancel_resource);
|
||||||
let loader = match &*load_data.url.scheme {
|
let loader = match &*load_data.url.scheme {
|
||||||
"file" => from_factory(file_loader::factory),
|
"file" => from_factory(file_loader::factory),
|
||||||
"http" | "https" | "view-source" =>
|
"http" | "https" | "view-source" => {
|
||||||
|
let http_state = HttpState {
|
||||||
|
hsts_list: self.hsts_list.clone(),
|
||||||
|
cookie_jar: self.cookie_jar.clone(),
|
||||||
|
auth_cache: self.auth_cache.clone()
|
||||||
|
};
|
||||||
http_loader::factory(self.user_agent.clone(),
|
http_loader::factory(self.user_agent.clone(),
|
||||||
self.hsts_list.clone(),
|
http_state,
|
||||||
self.cookie_storage.clone(),
|
|
||||||
self.auth_cache.clone(),
|
|
||||||
self.devtools_chan.clone(),
|
self.devtools_chan.clone(),
|
||||||
self.connector.clone()),
|
self.connector.clone())
|
||||||
|
},
|
||||||
"data" => from_factory(data_loader::factory),
|
"data" => from_factory(data_loader::factory),
|
||||||
"about" => from_factory(about_loader::factory),
|
"about" => from_factory(about_loader::factory),
|
||||||
_ => {
|
_ => {
|
||||||
|
@ -370,6 +374,6 @@ impl ResourceManager {
|
||||||
fn websocket_connect(&self,
|
fn websocket_connect(&self,
|
||||||
connect: WebSocketCommunicate,
|
connect: WebSocketCommunicate,
|
||||||
connect_data: WebSocketConnectData) {
|
connect_data: WebSocketConnectData) {
|
||||||
websocket_loader::init(connect, connect_data, self.cookie_storage.clone());
|
websocket_loader::init(connect, connect_data, self.cookie_jar.clone());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -20,7 +20,7 @@ use msg::constellation_msg::PipelineId;
|
||||||
use net::cookie::Cookie;
|
use net::cookie::Cookie;
|
||||||
use net::cookie_storage::CookieStorage;
|
use net::cookie_storage::CookieStorage;
|
||||||
use net::hsts::{HSTSList, HSTSEntry};
|
use net::hsts::{HSTSList, HSTSEntry};
|
||||||
use net::http_loader::{load, LoadError, HttpRequestFactory, HttpRequest, HttpResponse};
|
use net::http_loader::{load, LoadError, HttpRequestFactory, HttpRequest, HttpResponse, HttpState};
|
||||||
use net::resource_thread::{AuthCacheEntry, CancellationListener};
|
use net::resource_thread::{AuthCacheEntry, CancellationListener};
|
||||||
use net_traits::{LoadData, CookieSource, LoadContext, IncludeSubdomains};
|
use net_traits::{LoadData, CookieSource, LoadContext, IncludeSubdomains};
|
||||||
use std::borrow::Cow;
|
use std::borrow::Cow;
|
||||||
|
@ -375,9 +375,7 @@ fn expect_devtools_http_response(devtools_port: &Receiver<DevtoolsControlMsg>) -
|
||||||
fn test_check_default_headers_loaded_in_every_request() {
|
fn test_check_default_headers_loaded_in_every_request() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.data = None;
|
load_data.data = None;
|
||||||
|
@ -398,8 +396,7 @@ fn test_check_default_headers_loaded_in_every_request() {
|
||||||
headers.set(UserAgent(DEFAULT_USER_AGENT.to_owned()));
|
headers.set(UserAgent(DEFAULT_USER_AGENT.to_owned()));
|
||||||
|
|
||||||
// Testing for method.GET
|
// Testing for method.GET
|
||||||
let _ = load::<AssertRequestMustHaveHeaders>(load_data.clone(), hsts_list.clone(),
|
let _ = load::<AssertRequestMustHaveHeaders>(load_data.clone(), &http_state, None,
|
||||||
cookie_jar.clone(), auth_cache.clone(), None,
|
|
||||||
&AssertMustHaveHeadersRequestFactory {
|
&AssertMustHaveHeadersRequestFactory {
|
||||||
expected_headers: headers.clone(),
|
expected_headers: headers.clone(),
|
||||||
body: <[_]>::to_vec(&[])
|
body: <[_]>::to_vec(&[])
|
||||||
|
@ -410,7 +407,7 @@ fn test_check_default_headers_loaded_in_every_request() {
|
||||||
|
|
||||||
headers.set(ContentLength(0 as u64));
|
headers.set(ContentLength(0 as u64));
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustHaveHeaders>(load_data.clone(), hsts_list, cookie_jar, auth_cache.clone(), None,
|
let _ = load::<AssertRequestMustHaveHeaders>(load_data.clone(), &http_state, None,
|
||||||
&AssertMustHaveHeadersRequestFactory {
|
&AssertMustHaveHeadersRequestFactory {
|
||||||
expected_headers: headers,
|
expected_headers: headers,
|
||||||
body: <[_]>::to_vec(&[])
|
body: <[_]>::to_vec(&[])
|
||||||
|
@ -421,9 +418,7 @@ fn test_check_default_headers_loaded_in_every_request() {
|
||||||
fn test_load_when_request_is_not_get_or_head_and_there_is_no_body_content_length_should_be_set_to_0() {
|
fn test_load_when_request_is_not_get_or_head_and_there_is_no_body_content_length_should_be_set_to_0() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.data = None;
|
load_data.data = None;
|
||||||
|
@ -433,7 +428,7 @@ fn test_load_when_request_is_not_get_or_head_and_there_is_no_body_content_length
|
||||||
content_length.set(ContentLength(0));
|
content_length.set(ContentLength(0));
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(
|
let _ = load::<AssertRequestMustIncludeHeaders>(
|
||||||
load_data.clone(), hsts_list, cookie_jar, auth_cache,
|
load_data.clone(), &http_state,
|
||||||
None, &AssertMustIncludeHeadersRequestFactory {
|
None, &AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: content_length,
|
expected_headers: content_length,
|
||||||
body: <[_]>::to_vec(&[])
|
body: <[_]>::to_vec(&[])
|
||||||
|
@ -456,9 +451,7 @@ fn test_request_and_response_data_with_network_messages() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let url = url!("https://mozilla.com");
|
let url = url!("https://mozilla.com");
|
||||||
let (devtools_chan, devtools_port) = mpsc::channel::<DevtoolsControlMsg>();
|
let (devtools_chan, devtools_port) = mpsc::channel::<DevtoolsControlMsg>();
|
||||||
|
@ -468,7 +461,7 @@ fn test_request_and_response_data_with_network_messages() {
|
||||||
let mut request_headers = Headers::new();
|
let mut request_headers = Headers::new();
|
||||||
request_headers.set(Host { hostname: "bar.foo".to_owned(), port: None });
|
request_headers.set(Host { hostname: "bar.foo".to_owned(), port: None });
|
||||||
load_data.headers = request_headers.clone();
|
load_data.headers = request_headers.clone();
|
||||||
let _ = load::<MockRequest>(load_data, hsts_list, cookie_jar, auth_cache, Some(devtools_chan), &Factory,
|
let _ = load::<MockRequest>(load_data, &http_state, Some(devtools_chan), &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None));
|
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None));
|
||||||
|
|
||||||
// notification received from devtools
|
// notification received from devtools
|
||||||
|
@ -532,14 +525,12 @@ fn test_request_and_response_message_from_devtool_without_pipeline_id() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let url = url!("https://mozilla.com");
|
let url = url!("https://mozilla.com");
|
||||||
let (devtools_chan, devtools_port) = mpsc::channel::<DevtoolsControlMsg>();
|
let (devtools_chan, devtools_port) = mpsc::channel::<DevtoolsControlMsg>();
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
let _ = load::<MockRequest>(load_data, hsts_list, cookie_jar, auth_cache, Some(devtools_chan), &Factory,
|
let _ = load::<MockRequest>(load_data, &http_state, Some(devtools_chan), &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None));
|
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None));
|
||||||
|
|
||||||
// notification received from devtools
|
// notification received from devtools
|
||||||
|
@ -570,11 +561,9 @@ fn test_load_when_redirecting_from_a_post_should_rewrite_next_request_as_get() {
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.method = Method::Post;
|
load_data.method = Method::Post;
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<MockRequest>(load_data, hsts_list, cookie_jar, auth_cache, None, &Factory,
|
let _ = load::<MockRequest>(load_data, &http_state, None, &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None));
|
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -599,12 +588,10 @@ fn test_load_should_decode_the_response_as_deflate_when_response_headers_have_co
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let mut response = load::<MockRequest>(
|
let mut response = load::<MockRequest>(
|
||||||
load_data, hsts_list, cookie_jar, auth_cache, None,
|
load_data, &http_state, None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None))
|
&CancellationListener::new(None))
|
||||||
|
@ -633,15 +620,11 @@ fn test_load_should_decode_the_response_as_gzip_when_response_headers_have_conte
|
||||||
|
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let mut response = load::<MockRequest>(
|
let mut response = load::<MockRequest>(
|
||||||
load_data,
|
load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None, &Factory,
|
None, &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None))
|
&CancellationListener::new(None))
|
||||||
|
@ -680,13 +663,10 @@ fn test_load_doesnt_send_request_body_on_any_redirect() {
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.data = Some(<[_]>::to_vec("Body on POST!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Body on POST!".as_bytes()));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<AssertMustHaveBodyRequest>(
|
let _ = load::<AssertMustHaveBodyRequest>(
|
||||||
load_data, hsts_list, cookie_jar,
|
load_data, &http_state,
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
|
@ -712,20 +692,16 @@ fn test_load_doesnt_add_host_to_sts_list_when_url_is_http_even_if_sts_headers_ar
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<MockRequest>(load_data,
|
let _ = load::<MockRequest>(load_data,
|
||||||
hsts_list.clone(),
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None));
|
&CancellationListener::new(None));
|
||||||
|
|
||||||
assert_eq!(hsts_list.read().unwrap().is_host_secure("mozilla.com"), false);
|
assert_eq!(http_state.hsts_list.read().unwrap().is_host_secure("mozilla.com"), false);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -747,20 +723,16 @@ fn test_load_adds_host_to_sts_list_when_url_is_https_and_sts_headers_are_present
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<MockRequest>(load_data,
|
let _ = load::<MockRequest>(load_data,
|
||||||
hsts_list.clone(),
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None));
|
&CancellationListener::new(None));
|
||||||
|
|
||||||
assert!(hsts_list.read().unwrap().is_host_secure("mozilla.com"));
|
assert!(http_state.hsts_list.read().unwrap().is_host_secure("mozilla.com"));
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -780,24 +752,20 @@ fn test_load_sets_cookies_in_the_resource_manager_when_it_get_set_cookie_header_
|
||||||
|
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
assert_cookie_for_domain(cookie_jar.clone(), "http://mozilla.com", "");
|
assert_cookie_for_domain(http_state.cookie_jar.clone(), "http://mozilla.com", "");
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let _ = load::<MockRequest>(load_data,
|
let _ = load::<MockRequest>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar.clone(),
|
|
||||||
auth_cache.clone(),
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None));
|
&CancellationListener::new(None));
|
||||||
|
|
||||||
assert_cookie_for_domain(cookie_jar.clone(), "http://mozilla.com", "mozillaIs=theBest");
|
assert_cookie_for_domain(http_state.cookie_jar.clone(), "http://mozilla.com", "mozillaIs=theBest");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -807,12 +775,10 @@ fn test_load_sets_requests_cookies_header_for_url_by_getting_cookies_from_the_re
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut cookie_jar = cookie_jar.write().unwrap();
|
let mut cookie_jar = http_state.cookie_jar.write().unwrap();
|
||||||
let cookie_url = url.clone();
|
let cookie_url = url.clone();
|
||||||
let cookie = Cookie::new_wrapped(
|
let cookie = Cookie::new_wrapped(
|
||||||
CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned()),
|
CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned()),
|
||||||
|
@ -825,7 +791,7 @@ fn test_load_sets_requests_cookies_header_for_url_by_getting_cookies_from_the_re
|
||||||
let mut cookie = Headers::new();
|
let mut cookie = Headers::new();
|
||||||
cookie.set(CookieHeader(vec![CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned())]));
|
cookie.set(CookieHeader(vec![CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned())]));
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(load_data.clone(), hsts_list, cookie_jar, auth_cache, None,
|
let _ = load::<AssertRequestMustIncludeHeaders>(load_data.clone(), &http_state, None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: cookie,
|
expected_headers: cookie,
|
||||||
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
||||||
|
@ -838,12 +804,9 @@ fn test_load_sends_secure_cookie_if_http_changed_to_https_due_to_entry_in_hsts_s
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
let secured_url = url!("https://mozilla.com");
|
let secured_url = url!("https://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut hsts_list = hsts_list.write().unwrap();
|
let mut hsts_list = http_state.hsts_list.write().unwrap();
|
||||||
let entry = HSTSEntry::new(
|
let entry = HSTSEntry::new(
|
||||||
"mozilla.com".to_owned(), IncludeSubdomains::Included, Some(1000000)
|
"mozilla.com".to_owned(), IncludeSubdomains::Included, Some(1000000)
|
||||||
).unwrap();
|
).unwrap();
|
||||||
|
@ -851,7 +814,7 @@ fn test_load_sends_secure_cookie_if_http_changed_to_https_due_to_entry_in_hsts_s
|
||||||
}
|
}
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut cookie_jar = cookie_jar.write().unwrap();
|
let mut cookie_jar = http_state.cookie_jar.write().unwrap();
|
||||||
let cookie_url = secured_url.clone();
|
let cookie_url = secured_url.clone();
|
||||||
let mut cookie_pair = CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned());
|
let mut cookie_pair = CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned());
|
||||||
cookie_pair.secure = true;
|
cookie_pair.secure = true;
|
||||||
|
@ -871,7 +834,7 @@ fn test_load_sends_secure_cookie_if_http_changed_to_https_due_to_entry_in_hsts_s
|
||||||
headers.set_raw("Cookie".to_owned(), vec![<[_]>::to_vec("mozillaIs=theBest".as_bytes())]);
|
headers.set_raw("Cookie".to_owned(), vec![<[_]>::to_vec("mozillaIs=theBest".as_bytes())]);
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(
|
let _ = load::<AssertRequestMustIncludeHeaders>(
|
||||||
load_data.clone(), hsts_list, cookie_jar, auth_cache, None,
|
load_data.clone(), &http_state, None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: headers,
|
expected_headers: headers,
|
||||||
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
||||||
|
@ -882,12 +845,10 @@ fn test_load_sends_secure_cookie_if_http_changed_to_https_due_to_entry_in_hsts_s
|
||||||
fn test_load_sends_cookie_if_nonhttp() {
|
fn test_load_sends_cookie_if_nonhttp() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut cookie_jar = cookie_jar.write().unwrap();
|
let mut cookie_jar = http_state.cookie_jar.write().unwrap();
|
||||||
let cookie_url = url.clone();
|
let cookie_url = url.clone();
|
||||||
let cookie = Cookie::new_wrapped(
|
let cookie = Cookie::new_wrapped(
|
||||||
CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned()),
|
CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned()),
|
||||||
|
@ -904,7 +865,7 @@ fn test_load_sends_cookie_if_nonhttp() {
|
||||||
headers.set_raw("Cookie".to_owned(), vec![<[_]>::to_vec("mozillaIs=theBest".as_bytes())]);
|
headers.set_raw("Cookie".to_owned(), vec![<[_]>::to_vec("mozillaIs=theBest".as_bytes())]);
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(
|
let _ = load::<AssertRequestMustIncludeHeaders>(
|
||||||
load_data.clone(), hsts_list, cookie_jar, auth_cache, None,
|
load_data.clone(), &http_state, None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: headers,
|
expected_headers: headers,
|
||||||
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
||||||
|
@ -928,20 +889,17 @@ fn test_cookie_set_with_httponly_should_not_be_available_using_getcookiesforurl(
|
||||||
|
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
let _ = load::<MockRequest>(load_data, hsts_list,
|
let _ = load::<MockRequest>(load_data,
|
||||||
cookie_jar.clone(),
|
&http_state,
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None));
|
&CancellationListener::new(None));
|
||||||
|
|
||||||
let mut cookie_jar = cookie_jar.write().unwrap();
|
let mut cookie_jar = http_state.cookie_jar.write().unwrap();
|
||||||
assert!(cookie_jar.cookies_for_url(&url, CookieSource::NonHTTP).is_none());
|
assert!(cookie_jar.cookies_for_url(&url, CookieSource::NonHTTP).is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -960,20 +918,17 @@ fn test_when_cookie_received_marked_secure_is_ignored_for_http() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url!("http://mozilla.com"), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url!("http://mozilla.com"), None);
|
||||||
let _ = load::<MockRequest>(load_data, hsts_list,
|
let _ = load::<MockRequest>(load_data,
|
||||||
cookie_jar.clone(),
|
&http_state,
|
||||||
auth_cache.clone(),
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
&CancellationListener::new(None));
|
&CancellationListener::new(None));
|
||||||
|
|
||||||
assert_cookie_for_domain(cookie_jar, "http://mozilla.com", "");
|
assert_cookie_for_domain(http_state.cookie_jar.clone(), "http://mozilla.com", "");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -982,12 +937,10 @@ fn test_when_cookie_set_marked_httpsonly_secure_isnt_sent_on_http_request() {
|
||||||
let sec_url = url!("https://mozilla.com");
|
let sec_url = url!("https://mozilla.com");
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut cookie_jar = cookie_jar.write().unwrap();
|
let mut cookie_jar = http_state.cookie_jar.write().unwrap();
|
||||||
let cookie_url = sec_url.clone();
|
let cookie_url = sec_url.clone();
|
||||||
let cookie = Cookie::new_wrapped(
|
let cookie = Cookie::new_wrapped(
|
||||||
CookiePair::parse("mozillaIs=theBest; Secure;").unwrap(),
|
CookiePair::parse("mozillaIs=theBest; Secure;").unwrap(),
|
||||||
|
@ -1000,10 +953,10 @@ fn test_when_cookie_set_marked_httpsonly_secure_isnt_sent_on_http_request() {
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url, None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url, None);
|
||||||
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
||||||
|
|
||||||
assert_cookie_for_domain(cookie_jar.clone(), "https://mozilla.com", "mozillaIs=theBest");
|
assert_cookie_for_domain(http_state.cookie_jar.clone(), "https://mozilla.com", "mozillaIs=theBest");
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustNotIncludeHeaders>(
|
let _ = load::<AssertRequestMustNotIncludeHeaders>(
|
||||||
load_data.clone(), hsts_list, cookie_jar, auth_cache, None,
|
load_data.clone(), &http_state, None,
|
||||||
&AssertMustNotIncludeHeadersRequestFactory {
|
&AssertMustNotIncludeHeadersRequestFactory {
|
||||||
headers_not_expected: vec!["Cookie".to_owned()],
|
headers_not_expected: vec!["Cookie".to_owned()],
|
||||||
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
||||||
|
@ -1021,11 +974,9 @@ fn test_load_sets_content_length_to_length_of_request_body() {
|
||||||
let mut content_len_headers = Headers::new();
|
let mut content_len_headers = Headers::new();
|
||||||
content_len_headers.set(ContentLength(content.as_bytes().len() as u64));
|
content_len_headers.set(ContentLength(content.as_bytes().len() as u64));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(load_data.clone(), hsts_list, cookie_jar, auth_cache,
|
let _ = load::<AssertRequestMustIncludeHeaders>(load_data.clone(), &http_state,
|
||||||
None, &AssertMustIncludeHeadersRequestFactory {
|
None, &AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: content_len_headers,
|
expected_headers: content_len_headers,
|
||||||
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
body: <[_]>::to_vec(&*load_data.data.unwrap())
|
||||||
|
@ -1045,14 +996,10 @@ fn test_load_uses_explicit_accept_from_headers_in_load_data() {
|
||||||
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
||||||
load_data.headers.set(Accept(vec![text_html.clone()]));
|
load_data.headers.set(Accept(vec![text_html.clone()]));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: accept_headers,
|
expected_headers: accept_headers,
|
||||||
|
@ -1075,14 +1022,10 @@ fn test_load_sets_default_accept_to_html_xhtml_xml_and_then_anything_else() {
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: accept_headers,
|
expected_headers: accept_headers,
|
||||||
|
@ -1101,14 +1044,10 @@ fn test_load_uses_explicit_accept_encoding_from_load_data_headers() {
|
||||||
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
||||||
load_data.headers.set(AcceptEncoding(vec![qitem(Encoding::Chunked)]));
|
load_data.headers.set(AcceptEncoding(vec![qitem(Encoding::Chunked)]));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: accept_encoding_headers,
|
expected_headers: accept_encoding_headers,
|
||||||
|
@ -1128,14 +1067,10 @@ fn test_load_sets_default_accept_encoding_to_gzip_and_deflate() {
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
load_data.data = Some(<[_]>::to_vec("Yay!".as_bytes()));
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
let _ = load::<AssertRequestMustIncludeHeaders>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&AssertMustIncludeHeadersRequestFactory {
|
&AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: accept_encoding_headers,
|
expected_headers: accept_encoding_headers,
|
||||||
|
@ -1165,11 +1100,9 @@ fn test_load_errors_when_there_a_redirect_loop() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<MockRequest>(load_data, hsts_list, cookie_jar, auth_cache, None, &Factory,
|
match load::<MockRequest>(load_data, &http_state, None, &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None)) {
|
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None)) {
|
||||||
Err(LoadError::InvalidRedirect(_, msg)) => {
|
Err(LoadError::InvalidRedirect(_, msg)) => {
|
||||||
assert_eq!(msg, "redirect loop");
|
assert_eq!(msg, "redirect loop");
|
||||||
|
@ -1197,11 +1130,9 @@ fn test_load_errors_when_there_is_too_many_redirects() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<MockRequest>(load_data, hsts_list, cookie_jar, auth_cache, None, &Factory,
|
match load::<MockRequest>(load_data, &http_state, None, &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None)) {
|
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None)) {
|
||||||
Err(LoadError::MaxRedirects(url)) => {
|
Err(LoadError::MaxRedirects(url)) => {
|
||||||
assert_eq!(url.domain().unwrap(), "mozilla.com")
|
assert_eq!(url.domain().unwrap(), "mozilla.com")
|
||||||
|
@ -1237,11 +1168,9 @@ fn test_load_follows_a_redirect() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<MockRequest>(load_data, hsts_list, cookie_jar, auth_cache, None, &Factory,
|
match load::<MockRequest>(load_data, &http_state, None, &Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None)) {
|
DEFAULT_USER_AGENT.to_owned(), &CancellationListener::new(None)) {
|
||||||
Err(e) => panic!("expected to follow a redirect {:?}", e),
|
Err(e) => panic!("expected to follow a redirect {:?}", e),
|
||||||
Ok(mut lr) => {
|
Ok(mut lr) => {
|
||||||
|
@ -1266,14 +1195,10 @@ fn test_load_errors_when_scheme_is_not_http_or_https() {
|
||||||
let url = url!("ftp://not-supported");
|
let url = url!("ftp://not-supported");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<MockRequest>(load_data,
|
match load::<MockRequest>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&DontConnectFactory,
|
&DontConnectFactory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
|
@ -1288,14 +1213,10 @@ fn test_load_errors_when_viewing_source_and_inner_url_scheme_is_not_http_or_http
|
||||||
let url = url!("view-source:ftp://not-supported");
|
let url = url!("view-source:ftp://not-supported");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<MockRequest>(load_data,
|
match load::<MockRequest>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&DontConnectFactory,
|
&DontConnectFactory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
|
@ -1333,14 +1254,10 @@ fn test_load_errors_when_cancelled() {
|
||||||
|
|
||||||
let url = url!("https://mozilla.com");
|
let url = url!("https://mozilla.com");
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<MockRequest>(load_data,
|
match load::<MockRequest>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
|
@ -1384,12 +1301,10 @@ fn test_redirect_from_x_to_y_provides_y_cookies_from_y() {
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url_x.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url_x.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut cookie_jar = cookie_jar.write().unwrap();
|
let mut cookie_jar = http_state.cookie_jar.write().unwrap();
|
||||||
let cookie_x_url = url_x.clone();
|
let cookie_x_url = url_x.clone();
|
||||||
let cookie_x = Cookie::new_wrapped(
|
let cookie_x = Cookie::new_wrapped(
|
||||||
CookiePair::new("mozillaIsNot".to_owned(), "dotCom".to_owned()),
|
CookiePair::new("mozillaIsNot".to_owned(), "dotCom".to_owned()),
|
||||||
|
@ -1409,9 +1324,7 @@ fn test_redirect_from_x_to_y_provides_y_cookies_from_y() {
|
||||||
}
|
}
|
||||||
|
|
||||||
match load::<AssertRequestMustIncludeHeaders>(load_data,
|
match load::<AssertRequestMustIncludeHeaders>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
|
@ -1455,14 +1368,10 @@ fn test_redirect_from_x_to_x_provides_x_with_cookie_from_first_response() {
|
||||||
|
|
||||||
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), None);
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
match load::<AssertRequestMustIncludeHeaders>(load_data,
|
match load::<AssertRequestMustIncludeHeaders>(load_data,
|
||||||
hsts_list,
|
&http_state,
|
||||||
cookie_jar,
|
|
||||||
auth_cache,
|
|
||||||
None,
|
None,
|
||||||
&Factory,
|
&Factory,
|
||||||
DEFAULT_USER_AGENT.to_owned(),
|
DEFAULT_USER_AGENT.to_owned(),
|
||||||
|
@ -1479,16 +1388,14 @@ fn test_redirect_from_x_to_x_provides_x_with_cookie_from_first_response() {
|
||||||
fn test_if_auth_creds_not_in_url_but_in_cache_it_sets_it() {
|
fn test_if_auth_creds_not_in_url_but_in_cache_it_sets_it() {
|
||||||
let url = url!("http://mozilla.com");
|
let url = url!("http://mozilla.com");
|
||||||
|
|
||||||
let hsts_list = Arc::new(RwLock::new(HSTSList::new()));
|
let http_state = HttpState::new();
|
||||||
let cookie_jar = Arc::new(RwLock::new(CookieStorage::new()));
|
|
||||||
let auth_cache = Arc::new(RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
let auth_entry = AuthCacheEntry {
|
let auth_entry = AuthCacheEntry {
|
||||||
user_name: "username".to_owned(),
|
user_name: "username".to_owned(),
|
||||||
password: "test".to_owned(),
|
password: "test".to_owned(),
|
||||||
};
|
};
|
||||||
|
|
||||||
auth_cache.write().unwrap().insert(url.clone(), auth_entry);
|
http_state.auth_cache.write().unwrap().insert(url.clone(), auth_entry);
|
||||||
|
|
||||||
let mut load_data = LoadData::new(LoadContext::Browsing, url, None);
|
let mut load_data = LoadData::new(LoadContext::Browsing, url, None);
|
||||||
load_data.credentials_flag = true;
|
load_data.credentials_flag = true;
|
||||||
|
@ -1505,7 +1412,7 @@ fn test_if_auth_creds_not_in_url_but_in_cache_it_sets_it() {
|
||||||
);
|
);
|
||||||
|
|
||||||
let _ = load::<AssertRequestMustIncludeHeaders>(
|
let _ = load::<AssertRequestMustIncludeHeaders>(
|
||||||
load_data.clone(), hsts_list, cookie_jar, auth_cache,
|
load_data.clone(), &http_state,
|
||||||
None, &AssertMustIncludeHeadersRequestFactory {
|
None, &AssertMustIncludeHeadersRequestFactory {
|
||||||
expected_headers: auth_header,
|
expected_headers: auth_header,
|
||||||
body: <[_]>::to_vec(&[])
|
body: <[_]>::to_vec(&[])
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue