1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-07-27 11:07:44 +00:00

tr: Raise error if set2 is too big on complemented class

This commit is contained in:
Dorian Péron 2024-07-12 16:12:35 +02:00 committed by Ben Wiederhake
parent ee75d5a5e1
commit 128335a24b
2 changed files with 19 additions and 3 deletions

View file

@ -238,12 +238,14 @@ impl Sequence {
set2_uniques.sort();
set2_uniques.dedup();
//If the complement flag is used in translate mode, only one unique character may appear in
//set2. Validate this with the set of uniques in set2 that we just generated.
// If the complement flag is used in translate mode, only one unique
// character may appear in set2. Validate this with the set of uniques
// in set2 that we just generated.
// Also, set2 must not overgrow set1, otherwise the mapping can't be 1:1.
if set1.iter().any(|x| matches!(x, Self::Class(_)))
&& translating
&& complement_flag
&& set2_uniques.len() > 1
&& (set2_uniques.len() > 1 || set2_solved.len() > set1_len)
{
return Err(BadSequence::ComplementMoreThanOneUniqueInSet2);
}

View file

@ -1413,3 +1413,17 @@ fn check_complement_1_unique_in_set2() {
.args(&["-c", "[:upper:]", arg.as_str()])
.succeeds();
}
#[test]
fn check_complement_set2_too_big() {
let x231 = "x".repeat(231);
let x230 = &x231[..230];
// The complement of [:upper:] expands to 230 characters,
// putting more characters in set2 should fail.
new_ucmd!().args(&["-c", "[:upper:]", x230]).succeeds();
new_ucmd!()
.args(&["-c", "[:upper:]", x231.as_str()])
.fails()
.stderr_contains("when translating with complemented character classes,\nstring2 must map all characters in the domain to one");
}