mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-21 09:15:35 +00:00
20c67c2367
Changed to_string calls to to_owned calls where was a need. Source-Repo: https://github.com/servo/servo Source-Revision: d588ce8d8503bf0b340abfb13e2bd6a228cab900
167 lines
5.3 KiB
Rust
167 lines
5.3 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/. */
|
|
|
|
//! Implementation of cookie creation and matching as specified by
|
|
//! http://tools.ietf.org/html/rfc6265
|
|
|
|
use cookie_rs;
|
|
use net_traits::CookieSource;
|
|
use pub_domains::PUB_DOMAINS;
|
|
use std::borrow::ToOwned;
|
|
use std::net::{Ipv4Addr, Ipv6Addr};
|
|
use time::{Tm, now, at, Duration};
|
|
use url::Url;
|
|
|
|
/// A stored cookie that wraps the definition in cookie-rs. This is used to implement
|
|
/// various behaviours defined in the spec that rely on an associated request URL,
|
|
/// which cookie-rs and hyper's header parsing do not support.
|
|
#[derive(Clone, Debug)]
|
|
pub struct Cookie {
|
|
pub cookie: cookie_rs::Cookie,
|
|
pub host_only: bool,
|
|
pub persistent: bool,
|
|
pub creation_time: Tm,
|
|
pub last_access: Tm,
|
|
pub expiry_time: Option<Tm>,
|
|
}
|
|
|
|
impl Cookie {
|
|
/// http://tools.ietf.org/html/rfc6265#section-5.3
|
|
pub fn new_wrapped(mut cookie: cookie_rs::Cookie, request: &Url, source: CookieSource)
|
|
-> Option<Cookie> {
|
|
// Step 3
|
|
let (persistent, expiry_time) = match (&cookie.max_age, &cookie.expires) {
|
|
(&Some(max_age), _) => {
|
|
(true, Some(at(now().to_timespec() + Duration::seconds(max_age as i64))))
|
|
}
|
|
(_, &Some(expires)) => (true, Some(expires)),
|
|
_ => (false, None)
|
|
};
|
|
|
|
let url_host = request.host().map(|host| host.serialize()).unwrap_or("".to_owned());
|
|
|
|
// Step 4
|
|
let mut domain = cookie.domain.clone().unwrap_or("".to_owned());
|
|
|
|
// Step 5
|
|
match PUB_DOMAINS.iter().find(|&x| domain == *x) {
|
|
Some(val) if *val == url_host => domain = "".to_owned(),
|
|
Some(_) => return None,
|
|
None => {}
|
|
}
|
|
|
|
// Step 6
|
|
let host_only = if !domain.is_empty() {
|
|
if !Cookie::domain_match(&url_host, &domain) {
|
|
return None;
|
|
} else {
|
|
cookie.domain = Some(domain);
|
|
false
|
|
}
|
|
} else {
|
|
cookie.domain = Some(url_host);
|
|
true
|
|
};
|
|
|
|
// Step 7
|
|
let mut path = cookie.path.unwrap_or("".to_owned());
|
|
if path.is_empty() || path.as_bytes()[0] != b'/' {
|
|
let url_path = request.serialize_path();
|
|
let url_path = url_path.as_ref().map(|path| &**path);
|
|
path = Cookie::default_path(url_path.unwrap_or("")).to_owned();
|
|
}
|
|
cookie.path = Some(path);
|
|
|
|
|
|
// Step 10
|
|
if cookie.httponly && source != CookieSource::HTTP {
|
|
return None;
|
|
}
|
|
|
|
Some(Cookie {
|
|
cookie: cookie,
|
|
host_only: host_only,
|
|
persistent: persistent,
|
|
creation_time: now(),
|
|
last_access: now(),
|
|
expiry_time: expiry_time,
|
|
})
|
|
}
|
|
|
|
pub fn touch(&mut self) {
|
|
self.last_access = now();
|
|
}
|
|
|
|
// http://tools.ietf.org/html/rfc6265#section-5.1.4
|
|
pub fn default_path(request_path: &str) -> &str {
|
|
// Step 2
|
|
if request_path.is_empty() || !request_path.starts_with("/") {
|
|
return "/";
|
|
}
|
|
|
|
// Step 3
|
|
let rightmost_slash_idx = request_path.rfind("/").unwrap();
|
|
if rightmost_slash_idx == 0 {
|
|
// There's only one slash; it's the first character
|
|
return "/";
|
|
}
|
|
|
|
// Step 4
|
|
&request_path[..rightmost_slash_idx]
|
|
}
|
|
|
|
// http://tools.ietf.org/html/rfc6265#section-5.1.4
|
|
pub fn path_match(request_path: &str, cookie_path: &str) -> bool {
|
|
request_path == cookie_path ||
|
|
( request_path.starts_with(cookie_path) &&
|
|
( request_path.ends_with("/") || request_path.as_bytes()[cookie_path.len() - 1] == b'/' )
|
|
)
|
|
}
|
|
|
|
// http://tools.ietf.org/html/rfc6265#section-5.1.3
|
|
pub fn domain_match(string: &str, domain_string: &str) -> bool {
|
|
if string == domain_string {
|
|
return true;
|
|
}
|
|
if string.ends_with(domain_string)
|
|
&& string.as_bytes()[string.len()-domain_string.len()-1] == b'.'
|
|
&& string.parse::<Ipv4Addr>().is_err()
|
|
&& string.parse::<Ipv6Addr>().is_err() {
|
|
return true;
|
|
}
|
|
false
|
|
}
|
|
|
|
// http://tools.ietf.org/html/rfc6265#section-5.4 step 1
|
|
pub fn appropriate_for_url(&self, url: &Url, source: CookieSource) -> bool {
|
|
let domain = url.host().map(|host| host.serialize());
|
|
if self.host_only {
|
|
if self.cookie.domain != domain {
|
|
return false;
|
|
}
|
|
} else {
|
|
if let (Some(ref domain), &Some(ref cookie_domain)) = (domain, &self.cookie.domain) {
|
|
if !Cookie::domain_match(domain, cookie_domain) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
if let (Some(ref path), &Some(ref cookie_path)) = (url.serialize_path(), &self.cookie.path) {
|
|
if !Cookie::path_match(path, cookie_path) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if self.cookie.secure && url.scheme != "https".to_owned() {
|
|
return false;
|
|
}
|
|
if self.cookie.httponly && source == CookieSource::NonHTTP {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|