mod big5_decode;
mod big5_encode;
mod euc_kr_decode;
mod euc_kr_encode;
mod gb18030_decode;
mod gb18030_encode;
mod gb18030_ranges;
mod iso2022jp_katakana;
mod jis0208_decode;
mod jis0208_encode_full;
mod jis0208_encode_shift_jis;
mod jis0212_decode;
mod legacy_single_byte;
pub(crate) use iso2022jp_katakana::ISO2022JP_KATAKANA;
pub(crate) use jis0208_decode::JIS0208_DECODE;
pub(crate) use jis0208_encode_full::JIS0208_ENCODE_FULL;
pub(crate) use jis0208_encode_shift_jis::JIS0208_ENCODE_SHIFT_JIS;
pub(crate) use jis0212_decode::JIS0212_DECODE;
use big5_decode::BIG5_DECODE;
use big5_encode::BIG5_ENCODE;
use euc_kr_decode::EUC_KR_DECODE;
use euc_kr_encode::EUC_KR_ENCODE;
use gb18030_decode::GB18030_DECODE;
use gb18030_encode::GB18030_ENCODE;
use gb18030_ranges::GB18030_RANGES;
use crate::label::Encoding;
pub(crate) const UNMAPPED: u32 = u32::MAX;
pub(crate) const LATIN1_CODEPOINTS: [u32; 256] = {
let mut table = [0u32; 256];
let mut i = 0usize;
while i < 256 {
table[i] = i as u32;
i += 1;
}
table
};
pub(crate) const WINDOWS_1252_CODEPOINTS: [u32; 256] = [
0x0000, 0x0001, 0x0002, 0x0003, 0x0004, 0x0005, 0x0006, 0x0007, 0x0008, 0x0009, 0x000A, 0x000B,
0x000C, 0x000D, 0x000E, 0x000F, 0x0010, 0x0011, 0x0012, 0x0013, 0x0014, 0x0015, 0x0016, 0x0017,
0x0018, 0x0019, 0x001A, 0x001B, 0x001C, 0x001D, 0x001E, 0x001F, 0x0020, 0x0021, 0x0022, 0x0023,
0x0024, 0x0025, 0x0026, 0x0027, 0x0028, 0x0029, 0x002A, 0x002B, 0x002C, 0x002D, 0x002E, 0x002F,
0x0030, 0x0031, 0x0032, 0x0033, 0x0034, 0x0035, 0x0036, 0x0037, 0x0038, 0x0039, 0x003A, 0x003B,
0x003C, 0x003D, 0x003E, 0x003F, 0x0040, 0x0041, 0x0042, 0x0043, 0x0044, 0x0045, 0x0046, 0x0047,
0x0048, 0x0049, 0x004A, 0x004B, 0x004C, 0x004D, 0x004E, 0x004F, 0x0050, 0x0051, 0x0052, 0x0053,
0x0054, 0x0055, 0x0056, 0x0057, 0x0058, 0x0059, 0x005A, 0x005B, 0x005C, 0x005D, 0x005E, 0x005F,
0x0060, 0x0061, 0x0062, 0x0063, 0x0064, 0x0065, 0x0066, 0x0067, 0x0068, 0x0069, 0x006A, 0x006B,
0x006C, 0x006D, 0x006E, 0x006F, 0x0070, 0x0071, 0x0072, 0x0073, 0x0074, 0x0075, 0x0076, 0x0077,
0x0078, 0x0079, 0x007A, 0x007B, 0x007C, 0x007D, 0x007E, 0x007F,
0x20AC, 0x0081, 0x201A, 0x0192, 0x201E, 0x2026, 0x2020, 0x2021, 0x02C6, 0x2030, 0x0160, 0x2039,
0x0152, 0x008D, 0x017D, 0x008F, 0x0090, 0x2018, 0x2019, 0x201C, 0x201D, 0x2022, 0x2013, 0x2014,
0x02DC, 0x2122, 0x0161, 0x203A, 0x0153, 0x009D, 0x017E, 0x0178,
0x00A0, 0x00A1, 0x00A2, 0x00A3, 0x00A4, 0x00A5, 0x00A6, 0x00A7, 0x00A8, 0x00A9, 0x00AA, 0x00AB,
0x00AC, 0x00AD, 0x00AE, 0x00AF, 0x00B0, 0x00B1, 0x00B2, 0x00B3, 0x00B4, 0x00B5, 0x00B6, 0x00B7,
0x00B8, 0x00B9, 0x00BA, 0x00BB, 0x00BC, 0x00BD, 0x00BE, 0x00BF, 0x00C0, 0x00C1, 0x00C2, 0x00C3,
0x00C4, 0x00C5, 0x00C6, 0x00C7, 0x00C8, 0x00C9, 0x00CA, 0x00CB, 0x00CC, 0x00CD, 0x00CE, 0x00CF,
0x00D0, 0x00D1, 0x00D2, 0x00D3, 0x00D4, 0x00D5, 0x00D6, 0x00D7, 0x00D8, 0x00D9, 0x00DA, 0x00DB,
0x00DC, 0x00DD, 0x00DE, 0x00DF, 0x00E0, 0x00E1, 0x00E2, 0x00E3, 0x00E4, 0x00E5, 0x00E6, 0x00E7,
0x00E8, 0x00E9, 0x00EA, 0x00EB, 0x00EC, 0x00ED, 0x00EE, 0x00EF, 0x00F0, 0x00F1, 0x00F2, 0x00F3,
0x00F4, 0x00F5, 0x00F6, 0x00F7, 0x00F8, 0x00F9, 0x00FA, 0x00FB, 0x00FC, 0x00FD, 0x00FE, 0x00FF,
];
#[inline]
pub(crate) fn codepoint_to_char(codepoint: u32) -> char {
char::from_u32(codepoint).unwrap_or(char::REPLACEMENT_CHARACTER)
}
#[inline]
pub(crate) fn decode_single_byte(table: &[u32; 256], byte: u8) -> char {
codepoint_to_char(table[byte as usize])
}
#[inline]
pub(crate) fn encode_single_byte(table: &[u32; 256], scalar: u32) -> Option<u8> {
if scalar < 0x80 {
return Some(scalar as u8);
}
table[0x80..=0xFF]
.iter()
.position(|&cp| cp == scalar)
.map(|offset| (0x80 + offset) as u8)
}
#[inline]
pub(crate) fn decode_single_byte_upper(table: &[u32; 128], byte: u8) -> Option<char> {
let pointer = byte - 0x80;
let codepoint = table[pointer as usize];
if codepoint == UNMAPPED {
None
} else {
Some(codepoint_to_char(codepoint))
}
}
#[inline]
pub(crate) fn encode_single_byte_upper(table: &[u32; 128], scalar: u32) -> Option<u8> {
if scalar < 0x80 {
return Some(scalar as u8);
}
table
.iter()
.position(|&cp| cp == scalar)
.map(|offset| (0x80 + offset) as u8)
}
pub(crate) fn legacy_single_byte_table(encoding: Encoding) -> &'static [u32; 128] {
use legacy_single_byte::*;
match encoding {
Encoding::Ibm866 => &IBM866_UPPER,
Encoding::Iso8859_2 => &ISO_8859_2_UPPER,
Encoding::Iso8859_3 => &ISO_8859_3_UPPER,
Encoding::Iso8859_4 => &ISO_8859_4_UPPER,
Encoding::Iso8859_5 => &ISO_8859_5_UPPER,
Encoding::Iso8859_6 => &ISO_8859_6_UPPER,
Encoding::Iso8859_7 => &ISO_8859_7_UPPER,
Encoding::Iso8859_8 => &ISO_8859_8_UPPER,
Encoding::Iso8859_9 => &ISO_8859_9_UPPER,
Encoding::Iso8859_10 => &ISO_8859_10_UPPER,
Encoding::Iso8859_13 => &ISO_8859_13_UPPER,
Encoding::Iso8859_14 => &ISO_8859_14_UPPER,
Encoding::Iso8859_15 => &ISO_8859_15_UPPER,
Encoding::Iso8859_16 => &ISO_8859_16_UPPER,
Encoding::Koi8R => &KOI8_R_UPPER,
Encoding::Koi8U => &KOI8_U_UPPER,
Encoding::Macintosh => &MACINTOSH_UPPER,
Encoding::Windows1250 => &WINDOWS_1250_UPPER,
Encoding::Windows1251 => &WINDOWS_1251_UPPER,
Encoding::Windows1253 => &WINDOWS_1253_UPPER,
Encoding::Windows1254 => &WINDOWS_1254_UPPER,
Encoding::Windows1255 => &WINDOWS_1255_UPPER,
Encoding::Windows1256 => &WINDOWS_1256_UPPER,
Encoding::Windows1257 => &WINDOWS_1257_UPPER,
Encoding::Windows1258 => &WINDOWS_1258_UPPER,
Encoding::XMacCyrillic => &X_MAC_CYRILLIC_UPPER,
Encoding::Utf8
| Encoding::Utf16Le
| Encoding::Utf16Be
| Encoding::Latin1
| Encoding::Windows1252
| Encoding::Ascii
| Encoding::ShiftJis
| Encoding::EucJp
| Encoding::Iso2022Jp
| Encoding::Gbk
| Encoding::Gb18030
| Encoding::Big5
| Encoding::EucKr => unreachable!(
"legacy_single_byte_table called for non-single-byte-table encoding {encoding:?}"
),
}
}
#[inline]
pub(crate) fn jis0208_decode(pointer: u32) -> Option<char> {
let index = usize::try_from(pointer).ok()?;
let codepoint = *JIS0208_DECODE.get(index)?;
if codepoint == UNMAPPED {
None
} else {
Some(codepoint_to_char(codepoint))
}
}
#[inline]
pub(crate) fn jis0212_decode(pointer: u32) -> Option<char> {
let index = usize::try_from(pointer).ok()?;
let codepoint = *JIS0212_DECODE.get(index)?;
if codepoint == UNMAPPED {
None
} else {
Some(codepoint_to_char(codepoint))
}
}
#[inline]
pub(crate) fn jis0208_encode_full(scalar: u32) -> Option<u16> {
JIS0208_ENCODE_FULL
.binary_search_by_key(&scalar, |&(cp, _)| cp)
.ok()
.map(|i| JIS0208_ENCODE_FULL[i].1)
}
#[inline]
pub(crate) fn jis0208_encode_shift_jis(scalar: u32) -> Option<u16> {
JIS0208_ENCODE_SHIFT_JIS
.binary_search_by_key(&scalar, |&(cp, _)| cp)
.ok()
.map(|i| JIS0208_ENCODE_SHIFT_JIS[i].1)
}
#[inline]
pub(crate) fn iso2022jp_katakana_replacement(codepoint: u32) -> u32 {
let index = (codepoint - 0xFF61) as usize;
ISO2022JP_KATAKANA[index]
}
#[inline]
pub(crate) fn gb18030_decode_2byte(pointer: u32) -> Option<char> {
let index = usize::try_from(pointer).ok()?;
let codepoint = *GB18030_DECODE.get(index)?;
if codepoint == UNMAPPED {
None
} else {
Some(codepoint_to_char(codepoint))
}
}
#[inline]
pub(crate) fn gb18030_encode_2byte(scalar: u32) -> Option<u16> {
GB18030_ENCODE
.binary_search_by_key(&scalar, |&(cp, _)| cp)
.ok()
.map(|i| GB18030_ENCODE[i].1)
}
#[inline]
pub(crate) fn gb18030_ranges_decode(pointer: u32) -> Option<char> {
if (pointer > 39419 && pointer < 189000) || pointer > 1_237_575 {
return None;
}
if pointer == 7457 {
return Some('\u{E7C7}');
}
let i = GB18030_RANGES.partition_point(|&(p, _)| p <= pointer) - 1;
let (pointer_offset, codepoint_offset) = GB18030_RANGES[i];
char::from_u32(codepoint_offset + (pointer - pointer_offset))
}
#[inline]
pub(crate) fn gb18030_ranges_encode(scalar: u32) -> Option<u32> {
if scalar == 0xE7C7 {
return Some(7457);
}
let i = GB18030_RANGES.partition_point(|&(_, cp)| cp <= scalar);
if i == 0 {
return None;
}
let (pointer_offset, codepoint_offset) = GB18030_RANGES[i - 1];
Some(pointer_offset + (scalar - codepoint_offset))
}
#[inline]
pub(crate) fn big5_decode(pointer: u32) -> Option<char> {
let index = usize::try_from(pointer).ok()?;
let codepoint = *BIG5_DECODE.get(index)?;
if codepoint == UNMAPPED {
None
} else {
Some(codepoint_to_char(codepoint))
}
}
#[inline]
pub(crate) fn big5_encode(scalar: u32) -> Option<u16> {
BIG5_ENCODE
.binary_search_by_key(&scalar, |&(cp, _)| cp)
.ok()
.map(|i| BIG5_ENCODE[i].1)
}
#[inline]
pub(crate) fn euc_kr_decode(pointer: u32) -> Option<char> {
let index = usize::try_from(pointer).ok()?;
let codepoint = *EUC_KR_DECODE.get(index)?;
if codepoint == UNMAPPED {
None
} else {
Some(codepoint_to_char(codepoint))
}
}
#[inline]
pub(crate) fn euc_kr_encode(scalar: u32) -> Option<u16> {
EUC_KR_ENCODE
.binary_search_by_key(&scalar, |&(cp, _)| cp)
.ok()
.map(|i| EUC_KR_ENCODE[i].1)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn latin1_table_is_identity() {
for b in 0..=255u8 {
assert_eq!(LATIN1_CODEPOINTS[b as usize], b as u32);
}
}
#[test]
fn windows_1252_agrees_with_latin1_outside_c1_block() {
for b in 0..0x80usize {
assert_eq!(WINDOWS_1252_CODEPOINTS[b], LATIN1_CODEPOINTS[b]);
}
for b in 0xA0..=0xFFusize {
assert_eq!(WINDOWS_1252_CODEPOINTS[b], LATIN1_CODEPOINTS[b]);
}
}
#[test]
fn windows_1252_euro_sign_at_0x80() {
assert_eq!(WINDOWS_1252_CODEPOINTS[0x80], 0x20AC);
assert_eq!(codepoint_to_char(WINDOWS_1252_CODEPOINTS[0x80]), '\u{20AC}');
}
#[test]
fn windows_1252_unassigned_slots_fall_back_to_latin1() {
for b in [0x81usize, 0x8D, 0x8F, 0x90, 0x9D] {
assert_eq!(WINDOWS_1252_CODEPOINTS[b], b as u32, "byte 0x{b:02X}");
}
}
#[test]
fn every_256_table_entry_is_a_valid_scalar_value() {
for &cp in LATIN1_CODEPOINTS
.iter()
.chain(WINDOWS_1252_CODEPOINTS.iter())
{
assert!(
char::from_u32(cp).is_some(),
"0x{cp:04X} is not a valid scalar value"
);
}
}
#[test]
fn encode_single_byte_round_trips_windows_1252() {
for b in 0..=255u8 {
let scalar = WINDOWS_1252_CODEPOINTS[b as usize];
assert_eq!(
encode_single_byte(&WINDOWS_1252_CODEPOINTS, scalar),
Some(b)
);
}
}
#[test]
fn encode_single_byte_rejects_unmappable_scalar() {
assert_eq!(encode_single_byte(&WINDOWS_1252_CODEPOINTS, 0x3042), None);
assert_eq!(encode_single_byte(&LATIN1_CODEPOINTS, 0x3042), None);
}
#[test]
fn every_128_table_entry_is_unmapped_or_a_valid_scalar_value() {
for &encoding in Encoding::ALL {
if !encoding.is_legacy_single_byte_table() {
continue;
}
let table = legacy_single_byte_table(encoding);
for &cp in table.iter() {
assert!(
cp == UNMAPPED || char::from_u32(cp).is_some(),
"{encoding}: 0x{cp:04X} is neither UNMAPPED nor a valid scalar value"
);
}
}
}
#[test]
fn legacy_single_byte_table_partitions_every_encoding_variant_without_panicking() {
for &encoding in Encoding::ALL {
if encoding.is_legacy_single_byte_table() {
let _ = legacy_single_byte_table(encoding);
}
}
let non_table: Vec<Encoding> = Encoding::ALL
.iter()
.copied()
.filter(|e| !e.is_legacy_single_byte_table())
.collect();
assert_eq!(
non_table,
vec![
Encoding::Utf8,
Encoding::Utf16Le,
Encoding::Utf16Be,
Encoding::Latin1,
Encoding::Windows1252,
Encoding::Ascii,
Encoding::ShiftJis,
Encoding::EucJp,
Encoding::Iso2022Jp,
Encoding::Gbk,
Encoding::Gb18030,
Encoding::Big5,
Encoding::EucKr,
]
);
}
#[test]
fn upper_table_decode_encode_round_trips_for_every_mapped_byte() {
for &encoding in Encoding::ALL {
if !encoding.is_legacy_single_byte_table() {
continue;
}
let table = legacy_single_byte_table(encoding);
for pointer in 0u8..128 {
if table[pointer as usize] == UNMAPPED {
continue;
}
let byte = 0x80 + pointer;
let c = decode_single_byte_upper(table, byte)
.unwrap_or_else(|| panic!("{encoding}: byte 0x{byte:02X} must decode"));
assert_eq!(
encode_single_byte_upper(table, c as u32),
Some(byte),
"{encoding}: round trip for byte 0x{byte:02X} (U+{:04X})",
c as u32
);
}
}
}
#[test]
fn decode_single_byte_upper_indexes_from_the_start_of_the_upper_half() {
let table = legacy_single_byte_table(Encoding::Windows1250);
assert_eq!(
decode_single_byte_upper(table, 0x80),
Some(codepoint_to_char(table[0]))
);
assert_eq!(
decode_single_byte_upper(table, 0xFF),
Some(codepoint_to_char(table[127]))
);
}
#[test]
fn jis0208_decode_out_of_range_pointer_is_none() {
assert_eq!(jis0208_decode(JIS0208_DECODE.len() as u32), None);
assert_eq!(jis0208_decode(u32::MAX), None);
}
#[test]
fn jis0212_decode_out_of_range_pointer_is_none() {
assert_eq!(jis0212_decode(JIS0212_DECODE.len() as u32), None);
assert_eq!(jis0212_decode(u32::MAX), None);
}
#[test]
fn jis0208_hiragana_a_round_trips_at_well_known_pointer() {
assert_eq!(jis0208_decode(283), Some('\u{3042}'));
assert_eq!(jis0208_encode_full(0x3042), Some(283));
assert_eq!(jis0208_encode_shift_jis(0x3042), Some(283));
}
#[test]
fn iso2022jp_katakana_table_covers_the_full_half_width_katakana_block() {
assert_eq!(ISO2022JP_KATAKANA.len(), 63);
assert_eq!(iso2022jp_katakana_replacement(0xFF66), 0x30F2);
}
}