improve escapes handling
authoralfadur
Tue, 19 May 2020 02:25:24 +0300
changeset 15597 a2e78f5907cc
parent 15596 ca5c7c1de985
child 15598 3be9c98ae190
improve escapes handling
rust/hedgewars-server/src/server/haskell.rs
--- a/rust/hedgewars-server/src/server/haskell.rs	Mon May 18 23:43:58 2020 +0300
+++ b/rust/hedgewars-server/src/server/haskell.rs	Tue May 19 02:25:24 2020 +0300
@@ -5,7 +5,7 @@
     combinator::{map, map_res},
     multi::separated_list,
     sequence::{delimited, pair, preceded, separated_pair},
-    IResult,
+    ExtendInto, IResult,
 };
 use std::{
     collections::HashMap,
@@ -53,12 +53,12 @@
     write!(f, "\"")?;
     for c in text.chars() {
         if c.is_ascii() && !(c as u8).is_ascii_control() {
-            write!(f, "{}", c);
+            write!(f, "{}", c)?;
         } else {
             let mut bytes = [0u8; 4];
             let size = c.encode_utf8(&mut bytes).len();
             for byte in &bytes[0..size] {
-                write!(f, "\\{:03}", byte);
+                write!(f, "\\{:03}", byte)?;
             }
         }
     }
@@ -73,15 +73,15 @@
             HaskellValue::Tuple(items) => write_sequence(f, b"()", items.iter()),
             HaskellValue::List(items) => write_sequence(f, b"[]", items.iter()),
             HaskellValue::AnonStruct { name, fields } => {
-                write!(f, "{} ", name);
+                write!(f, "{} ", name)?;
                 write_sequence(f, b" \0", fields.iter())
             }
             HaskellValue::Struct { name, fields } => {
-                write!(f, "{} {{", name);
+                write!(f, "{} {{", name)?;
                 let fields = fields.iter().collect::<Vec<_>>();
                 let mut items = fields.iter();
                 while let Some((field_name, value)) = items.next() {
-                    write!(f, "{} = {}", field_name, value);
+                    write!(f, "{} = {}", field_name, value)?;
                     if !items.as_slice().is_empty() {
                         write!(f, ", ")?;
                     }
@@ -126,51 +126,78 @@
     map(number_raw, HaskellValue::Number)(input)
 }
 
-const BYTES: &[u8] = b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f\x20\x21\x22\x23\x24\x25\x26\x27\x28\x29\x2a\x2b\x2c\x2d\x2e\x2f\x30\x31\x32\x33\x34\x35\x36\x37\x38\x39\x3a\x3b\x3c\x3d\x3e\x3f\x40\x41\x42\x43\x44\x45\x46\x47\x48\x49\x4a\x4b\x4c\x4d\x4e\x4f\x50\x51\x52\x53\x54\x55\x56\x57\x58\x59\x5a\x5b\x5c\x5d\x5e\x5f\x60\x61\x62\x63\x64\x65\x66\x67\x68\x69\x6a\x6b\x6c\x6d\x6e\x6f\x70\x71\x72\x73\x74\x75\x76\x77\x78\x79\x7a\x7b\x7c\x7d\x7e\x7f\x80\x81\x82\x83\x84\x85\x86\x87\x88\x89\x8a\x8b\x8c\x8d\x8e\x8f\x90\x91\x92\x93\x94\x95\x96\x97\x98\x99\x9a\x9b\x9c\x9d\x9e\x9f\xa0\xa1\xa2\xa3\xa4\xa5\xa6\xa7\xa8\xa9\xaa\xab\xac\xad\xae\xaf\xb0\xb1\xb2\xb3\xb4\xb5\xb6\xb7\xb8\xb9\xba\xbb\xbc\xbd\xbe\xbf\xc0\xc1\xc2\xc3\xc4\xc5\xc6\xc7\xc8\xc9\xca\xcb\xcc\xcd\xce\xcf\xd0\xd1\xd2\xd3\xd4\xd5\xd6\xd7\xd8\xd9\xda\xdb\xdc\xdd\xde\xdf\xe0\xe1\xe2\xe3\xe4\xe5\xe6\xe7\xe8\xe9\xea\xeb\xec\xed\xee\xef\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7\xf8\xf9\xfa\xfb\xfc\xfd\xfe\xff";
+enum Escape {
+    Empty,
+    Byte(u8),
+}
+
+impl ExtendInto for Escape {
+    type Item = u8;
+    type Extender = Vec<u8>;
+
+    fn new_builder(&self) -> Self::Extender {
+        Vec::new()
+    }
 
-fn string_escape(input: &[u8]) -> HaskellResult<&[u8]> {
+    fn extend_into(&self, acc: &mut Self::Extender) {
+        if let Escape::Byte(b) = self {
+            acc.push(*b);
+        }
+    }
+}
+
+impl Extend<Escape> for Vec<u8> {
+    fn extend<T: IntoIterator<Item = Escape>>(&mut self, iter: T) {
+        for item in iter {
+            item.extend_into(self);
+        }
+    }
+}
+
+fn string_escape(input: &[u8]) -> HaskellResult<Escape> {
+    use Escape::*;
     alt((
-        map(number_raw, |n| &BYTES[n as usize..(n + 1) as usize]),
+        map(number_raw, |n| Byte(n)),
         alt((
-            map(tag("\\"), |_| &b"\\"[..]),
-            map(tag("\""), |_| &b"\""[..]),
-            map(tag("'"), |_| &b"'"[..]),
-            map(tag("n"), |_| &b"\n"[..]),
-            map(tag("r"), |_| &b"\r"[..]),
-            map(tag("t"), |_| &b"\t"[..]),
-            map(tag("a"), |_| &b"\x07"[..]),
-            map(tag("b"), |_| &b"\x08"[..]),
-            map(tag("v"), |_| &b"\x0B"[..]),
-            map(tag("f"), |_| &b"\x0C"[..]),
-            map(tag("&"), |_| &b""[..]),
-            map(tag("NUL"), |_| &b"\x00"[..]),
-            map(tag("SOH"), |_| &b"\x01"[..]),
-            map(tag("STX"), |_| &b"\x02"[..]),
-            map(tag("ETX"), |_| &b"\x03"[..]),
-            map(tag("EOT"), |_| &b"\x04"[..]),
-            map(tag("ENQ"), |_| &b"\x05"[..]),
-            map(tag("ACK"), |_| &b"\x06"[..]),
+            map(tag("\\"), |_| Byte(b'\\')),
+            map(tag("\""), |_| Byte(b'\"')),
+            map(tag("'"), |_| Byte(b'\'')),
+            map(tag("n"), |_| Byte(b'\n')),
+            map(tag("r"), |_| Byte(b'\r')),
+            map(tag("t"), |_| Byte(b'\t')),
+            map(tag("a"), |_| Byte(b'\x07')),
+            map(tag("b"), |_| Byte(b'\x08')),
+            map(tag("v"), |_| Byte(b'\x0B')),
+            map(tag("f"), |_| Byte(b'\x0C')),
+            map(tag("&"), |_| Empty),
+            map(tag("NUL"), |_| Byte(b'\x00')),
+            map(tag("SOH"), |_| Byte(b'\x01')),
+            map(tag("STX"), |_| Byte(b'\x02')),
+            map(tag("ETX"), |_| Byte(b'\x03')),
+            map(tag("EOT"), |_| Byte(b'\x04')),
+            map(tag("ENQ"), |_| Byte(b'\x05')),
+            map(tag("ACK"), |_| Byte(b'\x06')),
         )),
         alt((
-            map(tag("SO"), |_| &b"\x0E"[..]),
-            map(tag("SI"), |_| &b"\x0F"[..]),
-            map(tag("DLE"), |_| &b"\x10"[..]),
-            map(tag("DC1"), |_| &b"\x11"[..]),
-            map(tag("DC2"), |_| &b"\x12"[..]),
-            map(tag("DC3"), |_| &b"\x13"[..]),
-            map(tag("DC4"), |_| &b"\x14"[..]),
-            map(tag("NAK"), |_| &b"\x15"[..]),
-            map(tag("SYN"), |_| &b"\x16"[..]),
-            map(tag("ETB"), |_| &b"\x17"[..]),
-            map(tag("CAN"), |_| &b"\x18"[..]),
-            map(tag("EM"), |_| &b"\x19"[..]),
-            map(tag("SUB"), |_| &b"\x1A"[..]),
-            map(tag("ESC"), |_| &b"\x1B"[..]),
-            map(tag("FS"), |_| &b"\x1C"[..]),
-            map(tag("GS"), |_| &b"\x1D"[..]),
-            map(tag("RS"), |_| &b"\x1E"[..]),
-            map(tag("US"), |_| &b"\x1F"[..]),
-            map(tag("DEL"), |_| &b"\x7F"[..]),
+            map(tag("SO"), |_| Byte(b'\x0E')),
+            map(tag("SI"), |_| Byte(b'\x0F')),
+            map(tag("DLE"), |_| Byte(b'\x10')),
+            map(tag("DC1"), |_| Byte(b'\x11')),
+            map(tag("DC2"), |_| Byte(b'\x12')),
+            map(tag("DC3"), |_| Byte(b'\x13')),
+            map(tag("DC4"), |_| Byte(b'\x14')),
+            map(tag("NAK"), |_| Byte(b'\x15')),
+            map(tag("SYN"), |_| Byte(b'\x16')),
+            map(tag("ETB"), |_| Byte(b'\x17')),
+            map(tag("CAN"), |_| Byte(b'\x18')),
+            map(tag("EM"), |_| Byte(b'\x19')),
+            map(tag("SUB"), |_| Byte(b'\x1A')),
+            map(tag("ESC"), |_| Byte(b'\x1B')),
+            map(tag("FS"), |_| Byte(b'\x1C')),
+            map(tag("GS"), |_| Byte(b'\x1D')),
+            map(tag("RS"), |_| Byte(b'\x1E')),
+            map(tag("US"), |_| Byte(b'\x1F')),
+            map(tag("DEL"), |_| Byte(b'\x7F')),
         )),
     ))(input)
 }
@@ -270,11 +297,14 @@
 
         let value = Tuple(vec![
             Number(64),
-            String("text".to_string()),
+            String("text\t1".to_string()),
             List(vec![Number(1), Number(2), Number(3)]),
         ]);
 
-        assert_eq!(tuple(b"(64, \"text\", [1 , 2, 3])"), Ok((&b""[..], value)));
+        assert_eq!(
+            tuple(b"(64, \"text\\t1\", [1 , 2, 3])"),
+            Ok((&b""[..], value))
+        );
     }
 
     #[test]