Mirrors/coreutils
2
0
Fork 0
mirror of https://github.com/uutils/coreutils synced 2024-12-18 00:53:25 +00:00
Code Issues Projects Releases Packages Wiki Activity

Don't convert octal if its not valid character

Browse source 
Signed-off-by: Hanif Bin Ariffin <hanif.ariffin.4326@gmail.com>
This commit is contained in:
Hanif Bin Ariffin 2021-08-01 12:32:35 +08:00
parent 3fa56eabce
commit d813e00588
2 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/uu/tr/src/convert.rs
View file

@ -13,8 +13,9 @@ fn parse_octal(input: &str) -> IResult<&str, char> {
preceded(tag("\\"), recognize(many_m_n(1, 3, one_of("01234567")))),
|out: &str| {
u32::from_str_radix(out, 8)
.map(|u| char::from_u32(u).unwrap())
.map(|u| char::from_u32(u))
.ok()
.flatten()
},
)(input)
}

10
src/uu/tr/src/operation.rs
View file

@ -22,6 +22,7 @@ pub enum BadSequence {
MultipleCharRepeatInSet2,
CharRepeatInSet1,
InvalidRepeatCount(String),
EmptySet2WhenNotTruncatingSet1,
}
impl Display for BadSequence {
@ -42,6 +43,9 @@ impl Display for BadSequence {
BadSequence::InvalidRepeatCount(count) => {
writeln!(f, "invalid repeat count '{}' in [c*n] construct", count)
}
BadSequence::EmptySet2WhenNotTruncatingSet1 => {
writeln!(f, "when not truncating set1, string2 must be non-empty")
}
}
}
}
@ -434,7 +438,7 @@ pub struct TranslateOperationStandard {
}
impl TranslateOperationStandard {
fn new(set1: Vec<char>, set2: Vec<char>) -> Result<TranslateOperationStandard, String> {
fn new(set1: Vec<char>, set2: Vec<char>) -> Result<TranslateOperationStandard, BadSequence> {
if let Some(fallback) = set2.last().copied() {
Ok(TranslateOperationStandard {
translation_map: set1
@ -447,7 +451,7 @@ impl TranslateOperationStandard {
translation_map: HashMap::new(),
})
} else {
Err("when not truncating set1, string2 must be non-empty".to_string())
Err(BadSequence::EmptySet2WhenNotTruncatingSet1)
}
}
}
@ -473,7 +477,7 @@ impl TranslateOperation {
set1: Vec<char>,
set2: Vec<char>,
complement: bool,
) -> Result<TranslateOperation, String> {
) -> Result<TranslateOperation, BadSequence> {
if complement {
Ok(TranslateOperation::Complement(
TranslateOperationComplement::new(set1, set2),