use std::borrow::Cow;
use std::error::Error;
use std::fmt::{self, Display};
use std::string::FromUtf8Error;
#[inline]
pub(crate) fn from_hex_digit(digit: u8) -> Option<u8> {
match digit {
b'0'..=b'9' => Some(digit - b'0'),
b'A'..=b'F' => Some(digit - b'A' + 10),
b'a'..=b'f' => Some(digit - b'a' + 10),
_ => None,
}
}
#[inline]
pub fn decode(urlencoded: &str) -> Result<String, FromUrlEncodingError> {
let data = urlencoded.as_bytes();
String::from_utf8(decode_binary(data).into_owned())
.map_err(|error| FromUrlEncodingError::Utf8CharacterError {error})
}
pub fn decode_binary(mut data: &[u8]) -> Cow<[u8]> {
let mut out: Vec<u8> = Vec::with_capacity(data.len());
loop {
let mut parts = data.splitn(2, |&c| c == b'%');
out.extend_from_slice(parts.next().unwrap());
match parts.next() {
None => {
if out.is_empty() {
return data.into();
}
break;
},
Some(rest) => match rest.get(0..2) {
Some(&[first, second]) => match from_hex_digit(first) {
Some(first_val) => match from_hex_digit(second) {
Some(second_val) => {
out.push((first_val << 4) | second_val);
data = &rest[2..];
},
None => {
out.extend_from_slice(&[b'%', first]);
data = &rest[1..];
},
},
None => {
out.push(b'%');
data = rest;
},
},
_ => {
out.push(b'%');
out.extend_from_slice(rest);
break;
},
},
};
}
Cow::Owned(out)
}
#[derive(Debug)]
pub enum FromUrlEncodingError {
UriCharacterError { character: char, index: usize },
Utf8CharacterError { error: FromUtf8Error },
}
impl Error for FromUrlEncodingError {
fn source(&self) -> Option<&(dyn Error + 'static)> {
match self {
FromUrlEncodingError::UriCharacterError {character: _, index: _} => None,
FromUrlEncodingError::Utf8CharacterError {error} => Some(error)
}
}
}
impl Display for FromUrlEncodingError {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
match self {
FromUrlEncodingError::UriCharacterError {character, index} =>
write!(f, "invalid URI char [{}] at [{}]", character, index),
FromUrlEncodingError::Utf8CharacterError {error} =>
write!(f, "invalid utf8 char: {}", error)
}
}
}