b7019c2535
**Do not merge yet:** rust-url 1.0 is not published yet and may still get breaking changes. The goal of this PR for now is to demonstrate API usage. Depends on: * <s>https://github.com/servo/rust-url/pull/176</s> * <s>https://github.com/alexcrichton/cookie-rs/pull/42</s> * <s>https://github.com/hyperium/hyper/pull/740</s> * https://github.com/cyderize/rust-websocket/pull/70 * https://github.com/jgraham/webdriver-rust/pull/28 Source-Repo: https://github.com/servo/servo Source-Revision: 84ab7e9fe8f4a6528995eff3eb6e814cb724c364
54 lines
1.9 KiB
Rust
54 lines
1.9 KiB
Rust
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
use file_loader;
|
|
use mime_classifier::MIMEClassifier;
|
|
use net_traits::{LoadConsumer, LoadData, NetworkError};
|
|
use resource_thread::{CancellationListener, send_error};
|
|
use std::fs::canonicalize;
|
|
use std::sync::Arc;
|
|
use url::Url;
|
|
use url::percent_encoding::percent_decode;
|
|
use util::resource_files::resources_dir_path;
|
|
|
|
pub fn resolve_chrome_url(url: &Url) -> Result<Url, ()> {
|
|
assert_eq!(url.scheme(), "chrome");
|
|
if url.host_str() != Some("resources") {
|
|
return Err(())
|
|
}
|
|
let resources = canonicalize(resources_dir_path())
|
|
.expect("Error canonicalizing path to the resources directory");
|
|
let mut path = resources.clone();
|
|
for segment in url.path_segments().unwrap() {
|
|
match percent_decode(segment.as_bytes()).decode_utf8() {
|
|
// Check ".." to prevent access to files outside of the resources directory.
|
|
Ok(segment) => path.push(&*segment),
|
|
_ => return Err(())
|
|
}
|
|
}
|
|
match canonicalize(path) {
|
|
Ok(ref path) if path.starts_with(&resources) && path.exists() => {
|
|
Ok(Url::from_file_path(path).unwrap())
|
|
}
|
|
_ => Err(())
|
|
}
|
|
}
|
|
|
|
pub fn factory(mut load_data: LoadData,
|
|
start_chan: LoadConsumer,
|
|
classifier: Arc<MIMEClassifier>,
|
|
cancel_listener: CancellationListener) {
|
|
let file_url = match resolve_chrome_url(&load_data.url) {
|
|
Ok(url) => url,
|
|
Err(_) => {
|
|
send_error(load_data.url,
|
|
NetworkError::Internal("Invalid chrome URL.".to_owned()),
|
|
start_chan);
|
|
return;
|
|
}
|
|
};
|
|
load_data.url = file_url;
|
|
file_loader::factory(load_data, start_chan, classifier, cancel_listener)
|
|
}
|