Skip to content

Commit 6833c8c

Browse files
committed
dd: fix ISO-8859-1 case conversion for conv=lcase/ucase
1 parent b3f6e3b commit 6833c8c

File tree

3 files changed

+198
-38
lines changed

3 files changed

+198
-38
lines changed

src/uu/dd/src/conversion_tables.rs

Lines changed: 49 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -2,50 +2,63 @@
22
//
33
// For the full copyright and license information, please view the LICENSE
44
// file that was distributed with this source code.
5+
// spell-checker:ignore tolower toupper CTYPE
56

67
// Note: Conversion tables are just lookup tables.
78
// eg. The ASCII->EBCDIC table stores the EBCDIC code at the index
89
// obtained by treating the ASCII representation as a number.
910

1011
pub type ConversionTable = [u8; 256];
1112

12-
pub const ASCII_UCASE_TO_LCASE: ConversionTable = [
13-
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
14-
0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
15-
0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
16-
0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,
17-
0x40, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f,
18-
0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, 0x7a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f,
19-
0x60, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f,
20-
0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, 0x7a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f,
21-
0x80, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
22-
0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f,
23-
0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf,
24-
0xb0, 0xb1, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf,
25-
0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xcb, 0xcc, 0xcd, 0xce, 0xcf,
26-
0xd0, 0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xdb, 0xdc, 0xdd, 0xde, 0xdf,
27-
0xe0, 0xe1, 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef,
28-
0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff,
29-
];
13+
/// Builds a lowercase conversion table using locale-aware libc::tolower
14+
/// This function builds the table dynamically based on the current locale.
15+
/// The nix crate doesn't provide safe wrappers for locale functions, so we use libc directly.
16+
pub fn build_lcase_table() -> ConversionTable {
17+
// Initialize locale from environment if not already done
18+
// SAFETY: setlocale is called with a valid C string and is used to initialize
19+
// the locale for character conversion functions
20+
unsafe { libc::setlocale(libc::LC_CTYPE, c"".as_ptr()) };
3021

31-
pub const ASCII_LCASE_TO_UCASE: ConversionTable = [
32-
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
33-
0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
34-
0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
35-
0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,
36-
0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
37-
0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f,
38-
0x60, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
39-
0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f,
40-
0x80, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
41-
0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f,
42-
0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf,
43-
0xb0, 0xb1, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf,
44-
0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xcb, 0xcc, 0xcd, 0xce, 0xcf,
45-
0xd0, 0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xdb, 0xdc, 0xdd, 0xde, 0xdf,
46-
0xe0, 0xe1, 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef,
47-
0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff,
48-
];
22+
let mut table = [0u8; 256];
23+
for (i, item) in table.iter_mut().enumerate() {
24+
// SAFETY: tolower is called with a valid byte value and returns a valid byte
25+
*item = unsafe { libc::tolower(i as libc::c_int) } as u8;
26+
}
27+
table
28+
}
29+
30+
/// Builds an uppercase conversion table using locale-aware libc::toupper
31+
/// This function builds the table dynamically based on the current locale.
32+
/// The nix crate doesn't provide safe wrappers for locale functions, so we use libc directly.
33+
pub fn build_ucase_table() -> ConversionTable {
34+
// Initialize locale from environment if not already done
35+
// SAFETY: setlocale is called with a valid C string and is used to initialize
36+
// the locale for character conversion functions
37+
unsafe { libc::setlocale(libc::LC_CTYPE, c"".as_ptr()) };
38+
39+
let mut table = [0u8; 256];
40+
for (i, item) in table.iter_mut().enumerate() {
41+
// SAFETY: toupper is called with a valid byte value and returns a valid byte
42+
*item = unsafe { libc::toupper(i as libc::c_int) } as u8;
43+
}
44+
table
45+
}
46+
47+
/// Gets the uppercase to lowercase conversion table using current locale
48+
pub fn get_ucase_to_lcase_table() -> &'static ConversionTable {
49+
// For now, simply build a fresh table each time and leak it
50+
// This is simpler and safer than trying to manage mutable static state
51+
let table = build_lcase_table();
52+
Box::leak(Box::new(table))
53+
}
54+
55+
/// Gets the lowercase to uppercase conversion table using current locale
56+
pub fn get_lcase_to_ucase_table() -> &'static ConversionTable {
57+
// For now, simply build a fresh table each time and leak it
58+
// This is simpler and safer than trying to manage mutable static state
59+
let table = build_ucase_table();
60+
Box::leak(Box::new(table))
61+
}
4962

5063
pub const ASCII_TO_EBCDIC: ConversionTable = [
5164
0x00, 0x01, 0x02, 0x03, 0x37, 0x2d, 0x2e, 0x2f, 0x16, 0x05, 0x25, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,

src/uu/dd/src/parseargs.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -552,8 +552,8 @@ fn get_ctable(
552552
Conversion::Ibm => &ASCII_TO_IBM,
553553
},
554554
(None, Some(case)) => match case {
555-
Case::Lower => &ASCII_UCASE_TO_LCASE,
556-
Case::Upper => &ASCII_LCASE_TO_UCASE,
555+
Case::Lower => get_ucase_to_lcase_table(),
556+
Case::Upper => get_lcase_to_ucase_table(),
557557
},
558558
(Some(conv), Some(case)) => match (conv, case) {
559559
(Conversion::Ascii, Case::Upper) => &EBCDIC_TO_ASCII_LCASE_TO_UCASE,

tests/by-util/test_dd.rs

Lines changed: 147 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1945,3 +1945,150 @@ fn test_nocache_eof_fadvise_zero_length() {
19451945
"Expected len=0 at EOF: {strace}"
19461946
);
19471947
}
1948+
1949+
#[test]
1950+
#[cfg(not(target_os = "openbsd"))]
1951+
fn test_iso8859_1_case_conversion() {
1952+
use std::process::Command;
1953+
// Test ISO-8859-1 case conversion for accented characters
1954+
// Skip test if required locale is not available (common in CI environments)
1955+
let locale_test = Command::new("locale")
1956+
.arg("-a")
1957+
.output()
1958+
.ok()
1959+
.and_then(|output| String::from_utf8(output.stdout).ok())
1960+
.is_some_and(|locales| locales.contains("fr_FR"));
1961+
1962+
if !locale_test {
1963+
eprintln!("Skipping ISO-8859-1 test: French locale not available");
1964+
return;
1965+
}
1966+
1967+
let locale = "fr_FR";
1968+
1969+
// É (0xC9) should convert to é (0xE9) with lcase
1970+
let input = vec![0xC9, 0x0A]; // É\n in ISO-8859-1
1971+
let expected = vec![0xE9, 0x0A]; // é\n in ISO-8859-1
1972+
let result = new_ucmd!()
1973+
.args(&["conv=lcase", "status=none"])
1974+
.env("LC_ALL", locale)
1975+
.pipe_in(input)
1976+
.succeeds();
1977+
assert_eq!(result.stdout(), expected);
1978+
1979+
// é (0xE9) should convert to É (0xC9) with ucase
1980+
let input = vec![0xE9, 0x0A]; // é\n in ISO-8859-1
1981+
let expected = vec![0xC9, 0x0A]; // É\n in ISO-8859-1
1982+
let result = new_ucmd!()
1983+
.args(&["conv=ucase", "status=none"])
1984+
.env("LC_ALL", locale)
1985+
.pipe_in(input)
1986+
.succeeds();
1987+
assert_eq!(result.stdout(), expected);
1988+
}
1989+
1990+
#[test]
1991+
fn test_locale_aware_case_conversion() {
1992+
// Test that case conversion respects different single-byte locales
1993+
1994+
// Test Turkish (ISO-8859-9) where 'I' has special behavior
1995+
// Turkish has İ (0xDD) ↔ i (0xFD) and I (0x49) ↔ ı (0xFD in some positions)
1996+
// For simplicity, test some basic accented characters that differ between locales
1997+
1998+
// Test with ISO-8859-9 (Turkish) - Ğ (0xD0) should convert to ğ (0xF0)
1999+
let input = vec![0xD0, 0x0A]; // Ğ\n in ISO-8859-9
2000+
let expected = vec![0xF0, 0x0A]; // ğ\n in ISO-8859-9
2001+
let result = new_ucmd!()
2002+
.args(&["conv=lcase", "status=none"])
2003+
.env("LC_ALL", "tr_TR.iso8859-9")
2004+
.pipe_in(input)
2005+
.succeeds();
2006+
2007+
// Note: This test may not work if the system doesn't have Turkish locale installed
2008+
// In that case, it should fall back to C locale behavior
2009+
if result.stdout() == expected {
2010+
println!("Turkish locale case conversion working correctly");
2011+
} else {
2012+
println!("Turkish locale not available, using fallback behavior");
2013+
// Test that it at least doesn't crash and produces some output
2014+
assert!(!result.stdout().is_empty());
2015+
}
2016+
}
2017+
2018+
#[test]
2019+
fn test_french_locale_case_conversion() {
2020+
// Test French (ISO-8859-1) case conversion for French accented characters
2021+
// This test uses the same charset as the previous ISO-8859-1 test but with French locale
2022+
2023+
// Test French accented characters: À (0xC0) should convert to à (0xE0) with lcase
2024+
let input = vec![0xC0, 0x0A]; // À\n in ISO-8859-1
2025+
let expected = vec![0xE0, 0x0A]; // à\n in ISO-8859-1
2026+
let result = new_ucmd!()
2027+
.args(&["conv=lcase", "status=none"])
2028+
.env("LC_ALL", "fr_FR.iso8859-1")
2029+
.pipe_in(input)
2030+
.succeeds();
2031+
2032+
// Note: This test may not work if the system doesn't have French locale installed
2033+
// In that case, it should fall back to C locale behavior
2034+
if result.stdout() == expected {
2035+
println!("French locale case conversion working correctly for À -> à");
2036+
} else {
2037+
println!("French locale not available, using fallback behavior");
2038+
// Test that it at least doesn't crash and produces some output
2039+
assert!(!result.stdout().is_empty());
2040+
}
2041+
2042+
// Test reverse conversion: à (0xE0) should convert to À (0xC0) with ucase
2043+
let input = vec![0xE0, 0x0A]; // à\n in ISO-8859-1
2044+
let expected = vec![0xC0, 0x0A]; // À\n in ISO-8859-1
2045+
let result = new_ucmd!()
2046+
.args(&["conv=ucase", "status=none"])
2047+
.env("LC_ALL", "fr_FR.iso8859-1")
2048+
.pipe_in(input)
2049+
.succeeds();
2050+
2051+
if result.stdout() == expected {
2052+
println!("French locale case conversion working correctly for à -> À");
2053+
} else {
2054+
println!("French locale not available for reverse conversion, using fallback behavior");
2055+
assert!(!result.stdout().is_empty());
2056+
}
2057+
2058+
// Test another French character: Ç (0xC7) should convert to ç (0xE7) with lcase
2059+
let input = vec![0xC7, 0x0A]; // Ç\n in ISO-8859-1
2060+
let expected = vec![0xE7, 0x0A]; // ç\n in ISO-8859-1
2061+
let result = new_ucmd!()
2062+
.args(&["conv=lcase", "status=none"])
2063+
.env("LC_ALL", "fr_FR.iso8859-1")
2064+
.pipe_in(input)
2065+
.succeeds();
2066+
2067+
if result.stdout() == expected {
2068+
println!("French locale case conversion working correctly for Ç -> ç");
2069+
} else {
2070+
println!("French locale not available for Ç conversion, using fallback behavior");
2071+
assert!(!result.stdout().is_empty());
2072+
}
2073+
}
2074+
2075+
#[test]
2076+
fn test_ascii_case_conversion_fallback() {
2077+
// Test that ASCII characters always convert correctly regardless of locale
2078+
let input = vec![b'A', b'B', b'C', 0x0A]; // ABC\n
2079+
let expected = vec![b'a', b'b', b'c', 0x0A]; // abc\n
2080+
let result = new_ucmd!()
2081+
.args(&["conv=lcase", "status=none"])
2082+
.env("LC_ALL", "C")
2083+
.pipe_in(input.clone())
2084+
.succeeds();
2085+
assert_eq!(result.stdout(), expected);
2086+
2087+
// Test reverse conversion
2088+
let result = new_ucmd!()
2089+
.args(&["conv=ucase", "status=none"])
2090+
.env("LC_ALL", "C")
2091+
.pipe_in(expected)
2092+
.succeeds();
2093+
assert_eq!(result.stdout(), input);
2094+
}

0 commit comments

Comments
 (0)