parser.rs 43 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096
  1. // Copyright 2013-2014 Simon Sapin.
  2. //
  3. // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
  4. // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
  5. // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
  6. // option. This file may not be copied, modified, or distributed
  7. // except according to those terms.
  8. use std::ascii::AsciiExt;
  9. use std::error::Error;
  10. use std::fmt::{self, Formatter, Write};
  11. use super::{Url, EncodingOverride};
  12. use host::{self, HostInternal};
  13. use percent_encoding::{
  14. utf8_percent_encode, percent_encode,
  15. SIMPLE_ENCODE_SET, DEFAULT_ENCODE_SET, USERINFO_ENCODE_SET, QUERY_ENCODE_SET,
  16. PATH_SEGMENT_ENCODE_SET
  17. };
  18. pub type ParseResult<T> = Result<T, ParseError>;
  19. macro_rules! simple_enum_error {
  20. ($($name: ident => $description: expr,)+) => {
  21. /// Errors that can occur during parsing.
  22. #[derive(PartialEq, Eq, Clone, Copy, Debug)]
  23. pub enum ParseError {
  24. $(
  25. $name,
  26. )+
  27. }
  28. impl Error for ParseError {
  29. fn description(&self) -> &str {
  30. match *self {
  31. $(
  32. ParseError::$name => $description,
  33. )+
  34. }
  35. }
  36. }
  37. }
  38. }
  39. simple_enum_error! {
  40. EmptyHost => "empty host",
  41. IdnaError => "invalid international domain name",
  42. InvalidPort => "invalid port number",
  43. InvalidIpv4Address => "invalid IPv4 address",
  44. InvalidIpv6Address => "invalid IPv6 address",
  45. InvalidDomainCharacter => "invalid domain character",
  46. RelativeUrlWithoutBase => "relative URL without a base",
  47. RelativeUrlWithNonRelativeBase => "relative URL with a non-relative base",
  48. Overflow => "URLs more than 4 GB are not supported",
  49. }
  50. impl fmt::Display for ParseError {
  51. fn fmt(&self, fmt: &mut Formatter) -> fmt::Result {
  52. self.description().fmt(fmt)
  53. }
  54. }
  55. impl From<::idna::uts46::Errors> for ParseError {
  56. fn from(_: ::idna::uts46::Errors) -> ParseError { ParseError::IdnaError }
  57. }
  58. #[derive(Copy, Clone)]
  59. pub enum SchemeType {
  60. File,
  61. SpecialNotFile,
  62. NotSpecial,
  63. }
  64. impl SchemeType {
  65. fn is_special(&self) -> bool {
  66. !matches!(*self, SchemeType::NotSpecial)
  67. }
  68. fn is_file(&self) -> bool {
  69. matches!(*self, SchemeType::File)
  70. }
  71. pub fn from(s: &str) -> Self {
  72. match s {
  73. "http" | "https" | "ws" | "wss" | "ftp" | "gopher" => SchemeType::SpecialNotFile,
  74. "file" => SchemeType::File,
  75. _ => SchemeType::NotSpecial,
  76. }
  77. }
  78. }
  79. pub fn default_port(scheme: &str) -> Option<u16> {
  80. match scheme {
  81. "http" | "ws" => Some(80),
  82. "https" | "wss" => Some(443),
  83. "ftp" => Some(21),
  84. "gopher" => Some(70),
  85. _ => None,
  86. }
  87. }
  88. pub struct Parser<'a> {
  89. pub serialization: String,
  90. pub base_url: Option<&'a Url>,
  91. pub query_encoding_override: EncodingOverride,
  92. pub log_syntax_violation: Option<&'a Fn(&'static str)>,
  93. pub context: Context,
  94. }
  95. #[derive(PartialEq, Eq)]
  96. pub enum Context {
  97. UrlParser,
  98. Setter,
  99. PathSegmentSetter,
  100. }
  101. impl<'a> Parser<'a> {
  102. pub fn for_setter(serialization: String) -> Parser<'a> {
  103. Parser {
  104. serialization: serialization,
  105. base_url: None,
  106. query_encoding_override: EncodingOverride::utf8(),
  107. log_syntax_violation: None,
  108. context: Context::Setter,
  109. }
  110. }
  111. fn syntax_violation(&self, reason: &'static str) {
  112. if let Some(log) = self.log_syntax_violation {
  113. log(reason)
  114. }
  115. }
  116. fn syntax_violation_if<F: Fn() -> bool>(&self, reason: &'static str, test: F) {
  117. // Skip test if not logging.
  118. if let Some(log) = self.log_syntax_violation {
  119. if test() {
  120. log(reason)
  121. }
  122. }
  123. }
  124. /// https://url.spec.whatwg.org/#concept-basic-url-parser
  125. pub fn parse_url(mut self, original_input: &str) -> ParseResult<Url> {
  126. let input = original_input.trim_matches(c0_control_or_space);
  127. if input.len() < original_input.len() {
  128. self.syntax_violation("leading or trailing control or space character")
  129. }
  130. if let Ok(remaining) = self.parse_scheme(input) {
  131. return self.parse_with_scheme(remaining)
  132. }
  133. // No-scheme state
  134. if let Some(base_url) = self.base_url {
  135. if input.starts_with("#") {
  136. self.fragment_only(base_url, input)
  137. } else if base_url.non_relative() {
  138. Err(ParseError::RelativeUrlWithNonRelativeBase)
  139. } else {
  140. let scheme_type = SchemeType::from(base_url.scheme());
  141. if scheme_type.is_file() {
  142. self.parse_file(input, Some(base_url))
  143. } else {
  144. self.parse_relative(input, scheme_type, base_url)
  145. }
  146. }
  147. } else {
  148. Err(ParseError::RelativeUrlWithoutBase)
  149. }
  150. }
  151. pub fn parse_scheme<'i>(&mut self, input: &'i str) -> Result<&'i str, ()> {
  152. if input.is_empty() || !input.starts_with(ascii_alpha) {
  153. return Err(())
  154. }
  155. debug_assert!(self.serialization.is_empty());
  156. for (i, c) in input.char_indices() {
  157. match c {
  158. 'a'...'z' | 'A'...'Z' | '0'...'9' | '+' | '-' | '.' => {
  159. self.serialization.push(c.to_ascii_lowercase())
  160. }
  161. ':' => return Ok(&input[i + 1..]),
  162. _ => {
  163. self.serialization.clear();
  164. return Err(())
  165. }
  166. }
  167. }
  168. // EOF before ':'
  169. if self.context == Context::Setter {
  170. Ok("")
  171. } else {
  172. self.serialization.clear();
  173. Err(())
  174. }
  175. }
  176. fn parse_with_scheme(mut self, input: &str) -> ParseResult<Url> {
  177. let scheme_end = try!(to_u32(self.serialization.len()));
  178. let scheme_type = SchemeType::from(&self.serialization);
  179. self.serialization.push(':');
  180. match scheme_type {
  181. SchemeType::File => {
  182. self.syntax_violation_if("expected // after file:", || !input.starts_with("//"));
  183. let base_file_url = self.base_url.and_then(|base| {
  184. if base.scheme() == "file" { Some(base) } else { None }
  185. });
  186. self.serialization.clear();
  187. self.parse_file(input, base_file_url)
  188. }
  189. SchemeType::SpecialNotFile => {
  190. // special relative or authority state
  191. let slashes_count = input.find(|c| !matches!(c, '/' | '\\')).unwrap_or(input.len());
  192. if let Some(base_url) = self.base_url {
  193. if slashes_count < 2 &&
  194. base_url.scheme() == &self.serialization[..scheme_end as usize] {
  195. // Non-relative URLs only happen with "not special" schemes.
  196. debug_assert!(!base_url.non_relative());
  197. self.serialization.clear();
  198. return self.parse_relative(input, scheme_type, base_url)
  199. }
  200. }
  201. // special authority slashes state
  202. self.syntax_violation_if("expected //", || &input[..slashes_count] != "//");
  203. self.after_double_slash(&input[slashes_count..], scheme_type, scheme_end)
  204. }
  205. SchemeType::NotSpecial => self.parse_non_special(input, scheme_type, scheme_end)
  206. }
  207. }
  208. /// Scheme other than file, http, https, ws, ws, ftp, gopher.
  209. fn parse_non_special(mut self, input: &str, scheme_type: SchemeType, scheme_end: u32)
  210. -> ParseResult<Url> {
  211. // path or authority state (
  212. if input.starts_with("//") {
  213. return self.after_double_slash(&input[2..], scheme_type, scheme_end)
  214. }
  215. // Anarchist URL (no authority)
  216. let path_start = try!(to_u32(self.serialization.len()));
  217. let username_end = path_start;
  218. let host_start = path_start;
  219. let host_end = path_start;
  220. let host = HostInternal::None;
  221. let port = None;
  222. let remaining = if input.starts_with("/") {
  223. let path_start = self.serialization.len();
  224. self.serialization.push('/');
  225. self.parse_path(scheme_type, &mut false, path_start, &input[1..])
  226. } else {
  227. self.parse_non_relative_path(input)
  228. };
  229. self.with_query_and_fragment(scheme_end, username_end, host_start,
  230. host_end, host, port, path_start, remaining)
  231. }
  232. fn parse_file(mut self, input: &str, mut base_file_url: Option<&Url>) -> ParseResult<Url> {
  233. // file state
  234. debug_assert!(self.serialization.is_empty());
  235. let c = input.chars().next();
  236. match c {
  237. None => {
  238. if let Some(base_url) = base_file_url {
  239. // Copy everything except the fragment
  240. let before_fragment = match base_url.fragment_start {
  241. Some(i) => &base_url.serialization[..i as usize],
  242. None => &*base_url.serialization,
  243. };
  244. self.serialization.push_str(before_fragment);
  245. Ok(Url {
  246. serialization: self.serialization,
  247. fragment_start: None,
  248. ..*base_url
  249. })
  250. } else {
  251. self.serialization.push_str("file:///");
  252. let scheme_end = "file".len() as u32;
  253. let path_start = "file://".len() as u32;
  254. Ok(Url {
  255. serialization: self.serialization,
  256. scheme_end: scheme_end,
  257. username_end: path_start,
  258. host_start: path_start,
  259. host_end: path_start,
  260. host: HostInternal::None,
  261. port: None,
  262. path_start: path_start,
  263. query_start: None,
  264. fragment_start: None,
  265. })
  266. }
  267. },
  268. Some('?') => {
  269. if let Some(base_url) = base_file_url {
  270. // Copy everything up to the query string
  271. let before_query = match (base_url.query_start, base_url.fragment_start) {
  272. (None, None) => &*base_url.serialization,
  273. (Some(i), _) |
  274. (None, Some(i)) => base_url.slice(..i)
  275. };
  276. self.serialization.push_str(before_query);
  277. let (query_start, fragment_start) =
  278. try!(self.parse_query_and_fragment(base_url.scheme_end, input));
  279. Ok(Url {
  280. serialization: self.serialization,
  281. query_start: query_start,
  282. fragment_start: fragment_start,
  283. ..*base_url
  284. })
  285. } else {
  286. self.serialization.push_str("file:///");
  287. let scheme_end = "file".len() as u32;
  288. let path_start = "file://".len() as u32;
  289. let (query_start, fragment_start) =
  290. try!(self.parse_query_and_fragment(scheme_end, input));
  291. Ok(Url {
  292. serialization: self.serialization,
  293. scheme_end: scheme_end,
  294. username_end: path_start,
  295. host_start: path_start,
  296. host_end: path_start,
  297. host: HostInternal::None,
  298. port: None,
  299. path_start: path_start,
  300. query_start: query_start,
  301. fragment_start: fragment_start,
  302. })
  303. }
  304. },
  305. Some('#') => {
  306. if let Some(base_url) = base_file_url {
  307. self.fragment_only(base_url, input)
  308. } else {
  309. self.serialization.push_str("file:///");
  310. let scheme_end = "file".len() as u32;
  311. let path_start = "file://".len() as u32;
  312. let fragment_start = "file:///".len() as u32;
  313. self.parse_fragment(&input[1..]);
  314. Ok(Url {
  315. serialization: self.serialization,
  316. scheme_end: scheme_end,
  317. username_end: path_start,
  318. host_start: path_start,
  319. host_end: path_start,
  320. host: HostInternal::None,
  321. port: None,
  322. path_start: path_start,
  323. query_start: None,
  324. fragment_start: Some(fragment_start),
  325. })
  326. }
  327. }
  328. Some('/') | Some('\\') => {
  329. self.syntax_violation_if("backslash", || c == Some('\\'));
  330. let input = &input[1..];
  331. // file slash state
  332. let c = input.chars().next();
  333. self.syntax_violation_if("backslash", || c == Some('\\'));
  334. if matches!(c, Some('/') | Some('\\')) {
  335. // file host state
  336. self.serialization.push_str("file://");
  337. let scheme_end = "file".len() as u32;
  338. let host_start = "file://".len() as u32;
  339. let (path_start, host, remaining) = try!(self.parse_file_host(&input[1..]));
  340. let host_end = try!(to_u32(self.serialization.len()));
  341. let mut has_host = !matches!(host, HostInternal::None);
  342. let remaining = if path_start {
  343. self.parse_path_start(SchemeType::File, &mut has_host, remaining)
  344. } else {
  345. let path_start = self.serialization.len();
  346. self.serialization.push('/');
  347. self.parse_path(SchemeType::File, &mut has_host, path_start, remaining)
  348. };
  349. // FIXME: deal with has_host
  350. let (query_start, fragment_start) =
  351. try!(self.parse_query_and_fragment(scheme_end, remaining));
  352. Ok(Url {
  353. serialization: self.serialization,
  354. scheme_end: scheme_end,
  355. username_end: host_start,
  356. host_start: host_start,
  357. host_end: host_end,
  358. host: host,
  359. port: None,
  360. path_start: host_end,
  361. query_start: query_start,
  362. fragment_start: fragment_start,
  363. })
  364. } else {
  365. self.serialization.push_str("file:///");
  366. let scheme_end = "file".len() as u32;
  367. let path_start = "file://".len();
  368. if let Some(base_url) = base_file_url {
  369. let first_segment = base_url.path_segments().unwrap().next().unwrap();
  370. // FIXME: *normalized* drive letter
  371. if is_windows_drive_letter(first_segment) {
  372. self.serialization.push_str(first_segment);
  373. self.serialization.push('/');
  374. }
  375. }
  376. let remaining = self.parse_path(
  377. SchemeType::File, &mut false, path_start, input);
  378. let (query_start, fragment_start) =
  379. try!(self.parse_query_and_fragment(scheme_end, remaining));
  380. let path_start = path_start as u32;
  381. Ok(Url {
  382. serialization: self.serialization,
  383. scheme_end: scheme_end,
  384. username_end: path_start,
  385. host_start: path_start,
  386. host_end: path_start,
  387. host: HostInternal::None,
  388. port: None,
  389. path_start: path_start,
  390. query_start: query_start,
  391. fragment_start: fragment_start,
  392. })
  393. }
  394. }
  395. _ => {
  396. if starts_with_windows_drive_letter_segment(input) {
  397. base_file_url = None;
  398. }
  399. if let Some(base_url) = base_file_url {
  400. let before_query = match (base_url.query_start, base_url.fragment_start) {
  401. (None, None) => &*base_url.serialization,
  402. (Some(i), _) |
  403. (None, Some(i)) => base_url.slice(..i)
  404. };
  405. self.serialization.push_str(before_query);
  406. self.pop_path(SchemeType::File, base_url.path_start as usize);
  407. let remaining = self.parse_path(
  408. SchemeType::File, &mut true, base_url.path_start as usize, input);
  409. self.with_query_and_fragment(
  410. base_url.scheme_end, base_url.username_end, base_url.host_start,
  411. base_url.host_end, base_url.host, base_url.port, base_url.path_start, remaining)
  412. } else {
  413. self.serialization.push_str("file:///");
  414. let scheme_end = "file".len() as u32;
  415. let path_start = "file://".len();
  416. let remaining = self.parse_path(
  417. SchemeType::File, &mut false, path_start, input);
  418. let (query_start, fragment_start) =
  419. try!(self.parse_query_and_fragment(scheme_end, remaining));
  420. let path_start = path_start as u32;
  421. Ok(Url {
  422. serialization: self.serialization,
  423. scheme_end: scheme_end,
  424. username_end: path_start,
  425. host_start: path_start,
  426. host_end: path_start,
  427. host: HostInternal::None,
  428. port: None,
  429. path_start: path_start,
  430. query_start: query_start,
  431. fragment_start: fragment_start,
  432. })
  433. }
  434. }
  435. }
  436. }
  437. fn parse_relative(mut self, input: &str, scheme_type: SchemeType, base_url: &Url)
  438. -> ParseResult<Url> {
  439. // relative state
  440. debug_assert!(self.serialization.is_empty());
  441. match input.chars().next() {
  442. None => {
  443. // Copy everything except the fragment
  444. let before_fragment = match base_url.fragment_start {
  445. Some(i) => &base_url.serialization[..i as usize],
  446. None => &*base_url.serialization,
  447. };
  448. self.serialization.push_str(before_fragment);
  449. Ok(Url {
  450. serialization: self.serialization,
  451. fragment_start: None,
  452. ..*base_url
  453. })
  454. },
  455. Some('?') => {
  456. // Copy everything up to the query string
  457. let before_query = match (base_url.query_start, base_url.fragment_start) {
  458. (None, None) => &*base_url.serialization,
  459. (Some(i), _) |
  460. (None, Some(i)) => base_url.slice(..i)
  461. };
  462. self.serialization.push_str(before_query);
  463. let (query_start, fragment_start) =
  464. try!(self.parse_query_and_fragment(base_url.scheme_end, input));
  465. Ok(Url {
  466. serialization: self.serialization,
  467. query_start: query_start,
  468. fragment_start: fragment_start,
  469. ..*base_url
  470. })
  471. },
  472. Some('#') => self.fragment_only(base_url, input),
  473. Some('/') | Some('\\') => {
  474. let slashes_count = input.find(|c| !matches!(c, '/' | '\\')).unwrap_or(input.len());
  475. if slashes_count >= 2 {
  476. self.syntax_violation_if("expected //", || &input[..slashes_count] != "//");
  477. let scheme_end = base_url.scheme_end;
  478. debug_assert!(base_url.byte_at(scheme_end) == b':');
  479. self.serialization.push_str(base_url.slice(..scheme_end + 1));
  480. return self.after_double_slash(&input[slashes_count..], scheme_type, scheme_end)
  481. }
  482. let path_start = base_url.path_start;
  483. debug_assert!(base_url.byte_at(path_start) == b'/');
  484. self.serialization.push_str(base_url.slice(..path_start + 1));
  485. let remaining = self.parse_path(
  486. scheme_type, &mut true, path_start as usize, &input[1..]);
  487. self.with_query_and_fragment(
  488. base_url.scheme_end, base_url.username_end, base_url.host_start,
  489. base_url.host_end, base_url.host, base_url.port, base_url.path_start, remaining)
  490. }
  491. _ => {
  492. let before_query = match (base_url.query_start, base_url.fragment_start) {
  493. (None, None) => &*base_url.serialization,
  494. (Some(i), _) |
  495. (None, Some(i)) => base_url.slice(..i)
  496. };
  497. self.serialization.push_str(before_query);
  498. // FIXME spec says just "remove last entry", not the "pop" algorithm
  499. self.pop_path(scheme_type, base_url.path_start as usize);
  500. let remaining = self.parse_path(
  501. scheme_type, &mut true, base_url.path_start as usize, input);
  502. self.with_query_and_fragment(
  503. base_url.scheme_end, base_url.username_end, base_url.host_start,
  504. base_url.host_end, base_url.host, base_url.port, base_url.path_start, remaining)
  505. }
  506. }
  507. }
  508. fn after_double_slash(mut self, input: &str, scheme_type: SchemeType, scheme_end: u32)
  509. -> ParseResult<Url> {
  510. self.serialization.push('/');
  511. self.serialization.push('/');
  512. // authority state
  513. let (username_end, remaining) = try!(self.parse_userinfo(input, scheme_type));
  514. // host state
  515. let host_start = try!(to_u32(self.serialization.len()));
  516. let (host_end, host, port, remaining) =
  517. try!(self.parse_host_and_port(remaining, scheme_end, scheme_type));
  518. // path state
  519. let path_start = try!(to_u32(self.serialization.len()));
  520. let remaining = self.parse_path_start(
  521. scheme_type, &mut true, remaining);
  522. self.with_query_and_fragment(scheme_end, username_end, host_start,
  523. host_end, host, port, path_start, remaining)
  524. }
  525. /// Return (username_end, remaining)
  526. fn parse_userinfo<'i>(&mut self, input: &'i str, scheme_type: SchemeType)
  527. -> ParseResult<(u32, &'i str)> {
  528. let mut last_at = None;
  529. for (i, c) in input.char_indices() {
  530. match c {
  531. '@' => {
  532. if last_at.is_some() {
  533. self.syntax_violation("unencoded @ sign in username or password")
  534. } else {
  535. self.syntax_violation(
  536. "embedding authentification information (username or password) \
  537. in an URL is not recommended")
  538. }
  539. last_at = Some(i)
  540. },
  541. '/' | '?' | '#' => break,
  542. '\\' if scheme_type.is_special() => break,
  543. _ => (),
  544. }
  545. }
  546. let (input, remaining) = match last_at {
  547. None => return Ok((try!(to_u32(self.serialization.len())), input)),
  548. Some(0) => return Ok((try!(to_u32(self.serialization.len())), &input[1..])),
  549. Some(at) => (&input[..at], &input[at + 1..]),
  550. };
  551. let mut username_end = None;
  552. for (i, c, next_i) in input.char_ranges() {
  553. match c {
  554. ':' if username_end.is_none() => {
  555. // Start parsing password
  556. username_end = Some(try!(to_u32(self.serialization.len())));
  557. self.serialization.push(':');
  558. },
  559. '\t' | '\n' | '\r' => {},
  560. _ => {
  561. self.check_url_code_point(input, i, c);
  562. let utf8_c = &input[i..next_i];
  563. self.serialization.extend(utf8_percent_encode(utf8_c, USERINFO_ENCODE_SET));
  564. }
  565. }
  566. }
  567. let username_end = match username_end {
  568. Some(i) => i,
  569. None => try!(to_u32(self.serialization.len())),
  570. };
  571. self.serialization.push('@');
  572. Ok((username_end, remaining))
  573. }
  574. pub fn parse_host_and_port<'i>(&mut self, input: &'i str,
  575. scheme_end: u32, scheme_type: SchemeType)
  576. -> ParseResult<(u32, HostInternal, Option<u16>, &'i str)> {
  577. let (host, remaining) = try!(self.parse_host(input, scheme_type));
  578. let host_end = try!(to_u32(self.serialization.len()));
  579. let (port, remaining) = if remaining.starts_with(":") {
  580. try!(self.parse_port(&remaining[1..], scheme_end))
  581. } else {
  582. (None, remaining)
  583. };
  584. Ok((host_end, host, port, remaining))
  585. }
  586. pub fn parse_host<'i>(&mut self, input: &'i str, scheme_type: SchemeType)
  587. -> ParseResult<(HostInternal, &'i str)> {
  588. let mut inside_square_brackets = false;
  589. let mut has_ignored_chars = false;
  590. let mut end = input.len();
  591. for (i, b) in input.bytes().enumerate() {
  592. match b {
  593. b':' if !inside_square_brackets => {
  594. end = i;
  595. break
  596. },
  597. b'/' | b'?' | b'#' => {
  598. end = i;
  599. break
  600. }
  601. b'\\' if scheme_type.is_special() => {
  602. end = i;
  603. break
  604. }
  605. b'\t' | b'\n' | b'\r' => {
  606. self.syntax_violation("invalid character");
  607. has_ignored_chars = true;
  608. }
  609. b'[' => inside_square_brackets = true,
  610. b']' => inside_square_brackets = false,
  611. _ => {}
  612. }
  613. }
  614. let replaced: String;
  615. let host_input = if has_ignored_chars {
  616. replaced = input[..end].chars().filter(|&c| !matches!(c, '\t' | '\n' | '\r')).collect();
  617. &*replaced
  618. } else {
  619. &input[..end]
  620. };
  621. if scheme_type.is_special() && host_input.is_empty() {
  622. return Err(ParseError::EmptyHost)
  623. }
  624. let host = try!(host::parse(&host_input, &mut self.serialization));
  625. Ok((host, &input[end..]))
  626. }
  627. pub fn parse_file_host<'i>(&mut self, input: &'i str)
  628. -> ParseResult<(bool, HostInternal, &'i str)> {
  629. let mut has_ignored_chars = false;
  630. let mut end = input.len();
  631. for (i, b) in input.bytes().enumerate() {
  632. match b {
  633. b'/' | b'\\' | b'?' | b'#' => {
  634. end = i;
  635. break
  636. }
  637. b'\t' | b'\n' | b'\r' => {
  638. self.syntax_violation("invalid character");
  639. has_ignored_chars = true;
  640. }
  641. _ => {}
  642. }
  643. }
  644. let replaced: String;
  645. let host_input = if has_ignored_chars {
  646. replaced = input[..end].chars().filter(|&c| !matches!(c, '\t' | '\n' | '\r')).collect();
  647. &*replaced
  648. } else {
  649. &input[..end]
  650. };
  651. if is_windows_drive_letter(host_input) {
  652. return Ok((false, HostInternal::None, input))
  653. }
  654. let mut host;
  655. if host_input.is_empty() {
  656. host = HostInternal::None;
  657. } else {
  658. let host_start = self.serialization.len();
  659. host = try!(host::parse(&host_input, &mut self.serialization));
  660. if &self.serialization[host_start..] == "localhost" {
  661. host = HostInternal::None;
  662. self.serialization.truncate(host_start);
  663. }
  664. }
  665. Ok((true, host, &input[end..]))
  666. }
  667. pub fn parse_port<'i>(&mut self, input: &'i str, scheme_end: u32)
  668. -> ParseResult<(Option<u16>, &'i str)> {
  669. let mut port = 0;
  670. let mut has_any_digit = false;
  671. let mut end = input.len();
  672. for (i, c) in input.char_indices() {
  673. if let Some(digit) = c.to_digit(10) {
  674. port = port * 10 + digit;
  675. if port > ::std::u16::MAX as u32 {
  676. return Err(ParseError::InvalidPort)
  677. }
  678. has_any_digit = true;
  679. } else {
  680. match c {
  681. '/' | '\\' | '?' | '#' => {
  682. end = i;
  683. break
  684. },
  685. '\t' | '\n' | '\r' => self.syntax_violation("invalid character"),
  686. _ => return Err(ParseError::InvalidPort)
  687. }
  688. }
  689. }
  690. let mut opt_port = Some(port as u16);
  691. if !has_any_digit || opt_port == default_port(&self.serialization[..scheme_end as usize]) {
  692. opt_port = None;
  693. } else {
  694. self.serialization.push(':');
  695. write!(&mut self.serialization, "{}", port).unwrap();
  696. }
  697. return Ok((opt_port, &input[end..]))
  698. }
  699. fn parse_path_start<'i>(&mut self, scheme_type: SchemeType, has_host: &mut bool,
  700. mut input: &'i str)
  701. -> &'i str {
  702. // Path start state
  703. let mut iter = input.chars();
  704. match iter.next() {
  705. Some('/') => input = iter.as_str(),
  706. Some('\\') => {
  707. self.syntax_violation("backslash");
  708. input = iter.as_str()
  709. }
  710. _ => {}
  711. }
  712. let path_start = self.serialization.len();
  713. self.serialization.push('/');
  714. self.parse_path(scheme_type, has_host, path_start, input)
  715. }
  716. pub fn parse_path<'i>(&mut self, scheme_type: SchemeType, has_host: &mut bool,
  717. path_start: usize, input: &'i str)
  718. -> &'i str {
  719. // Relative path state
  720. debug_assert!(self.serialization.ends_with("/"));
  721. let mut iter = input.char_ranges();
  722. let mut end;
  723. loop {
  724. let segment_start = self.serialization.len();
  725. let mut ends_with_slash = false;
  726. end = input.len();
  727. while let Some((i, c, next_i)) = iter.next() {
  728. match c {
  729. '/' => {
  730. ends_with_slash = true;
  731. end = i;
  732. break
  733. },
  734. '\\' if scheme_type.is_special() => {
  735. self.syntax_violation("backslash");
  736. ends_with_slash = true;
  737. end = i;
  738. break
  739. },
  740. '?' | '#' if self.context == Context::UrlParser => {
  741. end = i;
  742. break
  743. },
  744. '\t' | '\n' | '\r' => self.syntax_violation("invalid characters"),
  745. _ => {
  746. self.check_url_code_point(input, i, c);
  747. if c == '%' {
  748. let after_percent_sign = iter.clone();
  749. if matches!(iter.next(), Some((_, '2', _))) &&
  750. matches!(iter.next(), Some((_, 'E', _)) | Some((_, 'e', _))) {
  751. self.serialization.push('.');
  752. continue
  753. }
  754. iter = after_percent_sign
  755. }
  756. if self.context == Context::PathSegmentSetter {
  757. self.serialization.extend(utf8_percent_encode(
  758. &input[i..next_i], PATH_SEGMENT_ENCODE_SET));
  759. } else {
  760. self.serialization.extend(utf8_percent_encode(
  761. &input[i..next_i], DEFAULT_ENCODE_SET));
  762. }
  763. }
  764. }
  765. }
  766. match &self.serialization[segment_start..] {
  767. ".." => {
  768. debug_assert!(self.serialization.as_bytes()[segment_start - 1] == b'/');
  769. self.serialization.truncate(segment_start - 1); // Truncate "/.."
  770. self.pop_path(scheme_type, path_start);
  771. if !self.serialization[path_start..].ends_with("/") {
  772. self.serialization.push('/')
  773. }
  774. },
  775. "." => {
  776. self.serialization.truncate(segment_start);
  777. },
  778. _ => {
  779. if scheme_type.is_file() && is_windows_drive_letter(
  780. &self.serialization[path_start + 1..]
  781. ) {
  782. unsafe {
  783. *self.serialization.as_mut_vec().last_mut().unwrap() = b':'
  784. }
  785. if *has_host {
  786. self.syntax_violation("file: with host and Windows drive letter");
  787. *has_host = false; // FIXME account for this in callers
  788. }
  789. }
  790. if ends_with_slash {
  791. self.serialization.push('/')
  792. }
  793. }
  794. }
  795. if !ends_with_slash {
  796. break
  797. }
  798. }
  799. &input[end..]
  800. }
  801. /// https://url.spec.whatwg.org/#pop-a-urls-path
  802. fn pop_path(&mut self, scheme_type: SchemeType, path_start: usize) {
  803. if self.serialization.len() > path_start {
  804. let slash_position = self.serialization[path_start..].rfind('/').unwrap();
  805. // + 1 since rfind returns the position before the slash.
  806. let segment_start = path_start + slash_position + 1;
  807. // Don’t pop a Windows drive letter
  808. // FIXME: *normalized* Windows drive letter
  809. if !(
  810. scheme_type.is_file() &&
  811. is_windows_drive_letter(&self.serialization[segment_start..])
  812. ) {
  813. self.serialization.truncate(segment_start);
  814. }
  815. }
  816. }
  817. fn parse_non_relative_path<'i>(&mut self, input: &'i str) -> &'i str {
  818. for (i, c, next_i) in input.char_ranges() {
  819. match c {
  820. '?' | '#' => return &input[i..],
  821. '\t' | '\n' | '\r' => self.syntax_violation("invalid character"),
  822. _ => {
  823. self.check_url_code_point(input, i, c);
  824. self.serialization.extend(utf8_percent_encode(
  825. &input[i..next_i], SIMPLE_ENCODE_SET));
  826. }
  827. }
  828. }
  829. ""
  830. }
  831. fn with_query_and_fragment(mut self, scheme_end: u32, username_end: u32,
  832. host_start: u32, host_end: u32, host: HostInternal,
  833. port: Option<u16>, path_start: u32, remaining: &str)
  834. -> ParseResult<Url> {
  835. let (query_start, fragment_start) =
  836. try!(self.parse_query_and_fragment(scheme_end, remaining));
  837. Ok(Url {
  838. serialization: self.serialization,
  839. scheme_end: scheme_end,
  840. username_end: username_end,
  841. host_start: host_start,
  842. host_end: host_end,
  843. host: host,
  844. port: port,
  845. path_start: path_start,
  846. query_start: query_start,
  847. fragment_start: fragment_start
  848. })
  849. }
  850. /// Return (query_start, fragment_start)
  851. fn parse_query_and_fragment(&mut self, scheme_end: u32, mut input: &str)
  852. -> ParseResult<(Option<u32>, Option<u32>)> {
  853. let mut query_start = None;
  854. match input.chars().next() {
  855. Some('#') => {}
  856. Some('?') => {
  857. query_start = Some(try!(to_u32(self.serialization.len())));
  858. self.serialization.push('?');
  859. let remaining = self.parse_query(scheme_end, &input[1..]);
  860. if let Some(remaining) = remaining {
  861. input = remaining
  862. } else {
  863. return Ok((query_start, None))
  864. }
  865. }
  866. None => return Ok((None, None)),
  867. _ => panic!("Programming error. parse_query_and_fragment() should not \
  868. have been called with input \"{}\"", input)
  869. };
  870. let fragment_start = try!(to_u32(self.serialization.len()));
  871. self.serialization.push('#');
  872. debug_assert!(input.starts_with("#"));
  873. self.parse_fragment(&input[1..]);
  874. Ok((query_start, Some(fragment_start)))
  875. }
  876. pub fn parse_query<'i>(&mut self, scheme_end: u32, input: &'i str)
  877. -> Option<&'i str> {
  878. let mut query = String::new(); // FIXME: use a streaming decoder instead
  879. let mut remaining = None;
  880. for (i, c) in input.char_indices() {
  881. match c {
  882. '#' if self.context == Context::UrlParser => {
  883. remaining = Some(&input[i..]);
  884. break
  885. },
  886. '\t' | '\n' | '\r' => self.syntax_violation("invalid characters"),
  887. _ => {
  888. self.check_url_code_point(input, i, c);
  889. query.push(c);
  890. }
  891. }
  892. }
  893. let encoding = match &self.serialization[..scheme_end as usize] {
  894. "http" | "https" | "file" | "ftp" | "gopher" => self.query_encoding_override,
  895. _ => EncodingOverride::utf8(),
  896. };
  897. let query_bytes = encoding.encode(&query);
  898. self.serialization.extend(percent_encode(&query_bytes, QUERY_ENCODE_SET));
  899. remaining
  900. }
  901. fn fragment_only(mut self, base_url: &Url, input: &str) -> ParseResult<Url> {
  902. let before_fragment = match base_url.fragment_start {
  903. Some(i) => base_url.slice(..i),
  904. None => &*base_url.serialization,
  905. };
  906. debug_assert!(self.serialization.is_empty());
  907. self.serialization.reserve(before_fragment.len() + input.len());
  908. self.serialization.push_str(before_fragment);
  909. self.serialization.push('#');
  910. debug_assert!(input.starts_with("#"));
  911. self.parse_fragment(&input[1..]);
  912. Ok(Url {
  913. serialization: self.serialization,
  914. fragment_start: Some(try!(to_u32(before_fragment.len()))),
  915. ..*base_url
  916. })
  917. }
  918. pub fn parse_fragment(&mut self, input: &str) {
  919. for (i, c) in input.char_indices() {
  920. match c {
  921. '\0' | '\t' | '\n' | '\r' => self.syntax_violation("invalid character"),
  922. _ => {
  923. self.check_url_code_point(input, i, c);
  924. self.serialization.push(c); // No percent-encoding here.
  925. }
  926. }
  927. }
  928. }
  929. fn check_url_code_point(&self, input: &str, i: usize, c: char) {
  930. if let Some(log) = self.log_syntax_violation {
  931. if c == '%' {
  932. if !starts_with_2_hex(&input[i + 1..]) {
  933. log("expected 2 hex digits after %")
  934. }
  935. } else if !is_url_code_point(c) {
  936. log("non-URL code point")
  937. }
  938. }
  939. }
  940. }
  941. #[inline]
  942. fn is_ascii_hex_digit(byte: u8) -> bool {
  943. matches!(byte, b'a'...b'f' | b'A'...b'F' | b'0'...b'9')
  944. }
  945. #[inline]
  946. fn starts_with_2_hex(input: &str) -> bool {
  947. input.len() >= 2
  948. && is_ascii_hex_digit(input.as_bytes()[0])
  949. && is_ascii_hex_digit(input.as_bytes()[1])
  950. }
  951. // Non URL code points:
  952. // U+0000 to U+0020 (space)
  953. // " # % < > [ \ ] ^ ` { | }
  954. // U+007F to U+009F
  955. // surrogates
  956. // U+FDD0 to U+FDEF
  957. // Last two of each plane: U+__FFFE to U+__FFFF for __ in 00 to 10 hex
  958. #[inline]
  959. fn is_url_code_point(c: char) -> bool {
  960. matches!(c,
  961. 'a'...'z' |
  962. 'A'...'Z' |
  963. '0'...'9' |
  964. '!' | '$' | '&' | '\'' | '(' | ')' | '*' | '+' | ',' | '-' |
  965. '.' | '/' | ':' | ';' | '=' | '?' | '@' | '_' | '~' |
  966. '\u{A0}'...'\u{D7FF}' | '\u{E000}'...'\u{FDCF}' | '\u{FDF0}'...'\u{FFFD}' |
  967. '\u{10000}'...'\u{1FFFD}' | '\u{20000}'...'\u{2FFFD}' |
  968. '\u{30000}'...'\u{3FFFD}' | '\u{40000}'...'\u{4FFFD}' |
  969. '\u{50000}'...'\u{5FFFD}' | '\u{60000}'...'\u{6FFFD}' |
  970. '\u{70000}'...'\u{7FFFD}' | '\u{80000}'...'\u{8FFFD}' |
  971. '\u{90000}'...'\u{9FFFD}' | '\u{A0000}'...'\u{AFFFD}' |
  972. '\u{B0000}'...'\u{BFFFD}' | '\u{C0000}'...'\u{CFFFD}' |
  973. '\u{D0000}'...'\u{DFFFD}' | '\u{E1000}'...'\u{EFFFD}' |
  974. '\u{F0000}'...'\u{FFFFD}' | '\u{100000}'...'\u{10FFFD}')
  975. }
  976. pub trait StrCharRanges<'a> {
  977. fn char_ranges(&self) -> CharRanges<'a>;
  978. }
  979. impl<'a> StrCharRanges<'a> for &'a str {
  980. #[inline]
  981. fn char_ranges(&self) -> CharRanges<'a> {
  982. CharRanges { slice: *self, position: 0 }
  983. }
  984. }
  985. #[derive(Clone)]
  986. pub struct CharRanges<'a> {
  987. slice: &'a str,
  988. position: usize,
  989. }
  990. impl<'a> Iterator for CharRanges<'a> {
  991. type Item = (usize, char, usize);
  992. #[inline]
  993. fn next(&mut self) -> Option<(usize, char, usize)> {
  994. match self.slice[self.position..].chars().next() {
  995. Some(ch) => {
  996. let position = self.position;
  997. self.position = position + ch.len_utf8();
  998. Some((position, ch, position + ch.len_utf8()))
  999. }
  1000. None => None,
  1001. }
  1002. }
  1003. }
  1004. /// https://url.spec.whatwg.org/#c0-controls-and-space
  1005. #[inline]
  1006. fn c0_control_or_space(ch: char) -> bool {
  1007. ch <= ' ' // U+0000 to U+0020
  1008. }
  1009. /// https://url.spec.whatwg.org/#ascii-alpha
  1010. #[inline]
  1011. pub fn ascii_alpha(ch: char) -> bool {
  1012. matches!(ch, 'a'...'z' | 'A'...'Z')
  1013. }
  1014. #[inline]
  1015. pub fn to_u32(i: usize) -> ParseResult<u32> {
  1016. if i <= ::std::u32::MAX as usize {
  1017. Ok(i as u32)
  1018. } else {
  1019. Err(ParseError::Overflow)
  1020. }
  1021. }
  1022. /// Wether the scheme is file:, the path has a single segment, and that segment
  1023. /// is a Windows drive letter
  1024. fn is_windows_drive_letter(segment: &str) -> bool {
  1025. segment.len() == 2
  1026. && starts_with_windows_drive_letter(segment)
  1027. }
  1028. fn starts_with_windows_drive_letter(s: &str) -> bool {
  1029. ascii_alpha(s.as_bytes()[0] as char)
  1030. && matches!(s.as_bytes()[1], b':' | b'|')
  1031. }
  1032. fn starts_with_windows_drive_letter_segment(s: &str) -> bool {
  1033. s.len() >= 3
  1034. && starts_with_windows_drive_letter(s)
  1035. && matches!(s.as_bytes()[2], b'/' | b'\\' | b'?' | b'#')
  1036. }