about summary refs log blame commit diff
path: root/tvix/nix-compat/src/derivation/escape.rs
blob: 06b550bbf02d895a8109753c68aa92de6a629956 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
12
                    
 
                                                             
                                                      
                                             






                                  
     



            
                            

                             


                                               
                                              
                                                 

     
use bstr::ByteSlice;

/// Escapes a byte sequence. Does not add surrounding quotes.
pub fn escape_bytes<P: AsRef<[u8]>>(s: P) -> Vec<u8> {
    let mut s: Vec<u8> = s.as_ref().to_vec();

    s = s.replace(b"\\", b"\\\\");
    s = s.replace(b"\n", b"\\n");
    s = s.replace(b"\r", b"\\r");
    s = s.replace(b"\t", b"\\t");
    s = s.replace(b"\"", b"\\\"");

    s
}

#[cfg(test)]
mod tests {
    use super::escape_bytes;
    use test_case::test_case;

    #[test_case(b"", b""; "empty")]
    #[test_case(b"\"", b"\\\""; "doublequote")]
    #[test_case(b":", b":"; "colon")]
    fn escape(input: &[u8], expected: &[u8]) {
        assert_eq!(expected, escape_bytes(input))
    }
}