47 lines
1.8 KiB
Rust
47 lines
1.8 KiB
Rust
use memchr::memchr;
|
|
|
|
use crate::gen::codepoints::ALPHANUMERIC_OR_EQUALS;
|
|
use crate::gen::entities::{EntityType, ENTITY};
|
|
use crate::pattern::TrieNodeMatch;
|
|
|
|
pub fn encode_ampersands(mut code: &[u8], in_attr_val: bool) -> Vec<u8> {
|
|
let mut res = Vec::<u8>::new();
|
|
while !code.is_empty() {
|
|
let (before, matched) = match memchr(b'&', code) {
|
|
None => (code.len(), false),
|
|
Some(n) => (n, true),
|
|
};
|
|
res.extend_from_slice(&code[..before]);
|
|
code = &code[before..];
|
|
if matched {
|
|
let len = match ENTITY.longest_matching_prefix(code) {
|
|
// Entity is malformed, so we can just ignore it.
|
|
TrieNodeMatch::NotFound { reached } => reached,
|
|
TrieNodeMatch::Found { len, value } => {
|
|
match value {
|
|
EntityType::Named(_)
|
|
if in_attr_val
|
|
&& code[len - 1] != b';'
|
|
&& code
|
|
.get(len)
|
|
.filter(|&&c| ALPHANUMERIC_OR_EQUALS[c])
|
|
.is_some() =>
|
|
{
|
|
// A named entity inside an attribute value that doesn't end with semicolon but is followed by an alphanumeric or `=` character is not decoded, so we don't need to encode.
|
|
// https://html.spec.whatwg.org/multipage/parsing.html#named-character-reference-state.
|
|
}
|
|
_ => {
|
|
res.extend_from_slice(b"&");
|
|
}
|
|
};
|
|
len
|
|
}
|
|
};
|
|
|
|
res.extend_from_slice(&code[..len]);
|
|
code = &code[len..];
|
|
};
|
|
}
|
|
res
|
|
}
|