YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-01 02:05:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

access list: use hex crate for info hash parsing

Browse source
This commit is contained in:
Joakim Frostegård 2021-10-15 10:09:02 +02:00
parent 752873280b
commit f370dac330
4 changed files with 16 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

28
aquatic_common/src/lib.rs
View file

@ -36,22 +36,9 @@ pub struct AccessList(HashSet<[u8; 20]>);
impl AccessList {
fn parse_info_hash(line: String) -> anyhow::Result<[u8; 20]> {
if line.len() != 20 {
return Err(anyhow::anyhow!("Info hash is not 20 bytes long: {}", line));
}
let mut bytes = [0u8; 20];
for (byte, c) in bytes.iter_mut().zip(line.chars()) {
if c as u32 > 255 {
return Err(anyhow::anyhow!(
"Info hash character is not ASCII: {:#?}",
c
));
}
*byte = c as u8;
}
hex::decode_to_slice(line, &mut bytes)?;
Ok(bytes)
}
@ -170,3 +157,16 @@ pub fn convert_ipv4_mapped_ipv6(ip_address: IpAddr) -> IpAddr {
ip_address
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_access_list_parse_info_hash(){
assert!(AccessList::parse_info_hash("aaaabbbbccccddddeeeeaaaabbbbccccddddeeee".to_owned()).is_ok());
assert!(AccessList::parse_info_hash("aaaabbbbccccddddeeeeaaaabbbbccccddddeeeef".to_owned()).is_err());
assert!(AccessList::parse_info_hash("aaaabbbbccccddddeeeeaaaabbbbccccddddeee".to_owned()).is_err());
assert!(AccessList::parse_info_hash("aaaabbbbccccddddeeeeaaaabbbbccccddddeeeö".to_owned()).is_err());
}
}