1 extern crate inflate;
2 
get_test_file_data(name: &str) -> Vec<u8>3 fn get_test_file_data(name: &str) -> Vec<u8> {
4     use std::fs::File;
5     use std::io::Read;
6     let mut input = Vec::new();
7     let mut f = File::open(name).unwrap();
8 
9     f.read_to_end(&mut input).unwrap();
10     input
11 }
12 
13 #[test]
14 /// See https://github.com/PistonDevelopers/inflate/issues/14
issue_14()15 fn issue_14() {
16     let test_data = get_test_file_data("tests/issue_14.zlib");
17     let res = inflate::inflate_bytes_zlib(&test_data);
18     // This should fail as the file specifies code lengths that won't work.
19     assert!(res.is_err());
20 }
21 
22 #[test]
23 /// Another input that produce invalid code lengths.
issue_16()24 fn issue_16() {
25     let data = b"M\xff\xffM*\xad\xad\xad\xad\xad\xad\xad\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xcd\xad\xad\xad\xad\xad\xad\xad\xad\xad\xad\xad\xadMCMMMM\x00\x00\x00\x00\x00\x00\x00\x00\x00!\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xd9\xcb!\x0a";
26     let mut stream = inflate::InflateStream::new();
27     let res = stream.update(data);
28     assert!(res.is_err());
29 }
30 
31 #[test]
32 /// Similar to 16 but with CLENS being invalid.
issue_17()33 fn issue_17() {
34     let data = b"\xdd\xff\xff*M\x94ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff\x01\x09\x00\x00\xf2\xf2MM\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00*M\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00MMMM\xff\xffM\xff\x00;MM0*\x00\x00\x00\x00\x00\x00\x0a";
35     let mut stream = inflate::InflateStream::new();
36     let res = stream.update(data);
37     assert!(res.is_err());
38 }
39 
40 #[test]
issue_30_realworld()41 fn issue_30_realworld() {
42     // In this case, the LZ77 window size is misreported as 256B. The first two bytes should be
43     // e.g. [0x78, 0xda] to set the window size to a maximum value (32kiB).
44     let data = &[
45         0x8, 0xd7, 99, 100, 48, 158, 201, 128, 3, 196, 91, 158, 48, 231, 153, 127, 242, 75,
46         226, 194, 227, 22, 184, 212, 48, 51, 72, 249, 98, 213, 57, 47, 180, 153, 225, 195, 133,
47         148, 64, 230, 87, 247, 206, 53, 249, 111, 249, 201, 40, 122, 241, 137, 12, 166, 74, 70,
48         172, 246, 255, 154, 150, 149, 186, 56, 14, 110, 109, 188, 229, 137, 217, 177, 139, 216,
49         178, 166, 49, 50, 48, 252, 199, 175, 63, 222, 242, 68, 130, 238, 124, 199, 89, 88, 12, 221,
50         159, 150, 190, 224, 50, 186, 95, 152, 208, 20, 213, 56, 207, 95, 112, 57, 17, 171, 87, 23,
51         92, 78, 172, 113, 158, 143, 207, 255, 241, 150, 39, 126, 190, 190, 208, 118, 48, 19, 171,
52         254, 139, 79, 100, 52, 217, 183, 136, 136, 163, 4, 4, 138, 253, 230, 60, 243, 83, 2, 153,
53         25, 209, 253, 136, 0, 41, 129, 204, 230, 60, 40, 78, 96, 65, 230, 156, 252, 146, 200, 176,
54         126, 254, 127, 6, 70, 92, 209, 57, 103, 253, 223, 147, 95, 18, 113, 250, 127, 225, 113, 11,
55         87, 115, 134, 120, 203, 19, 88, 245, 31, 186, 165, 234, 106, 206, 64, 32, 252, 90, 246, 38,
56         38, 232, 206, 199, 170, 127, 94, 112, 85, 203, 222, 68, 2, 241, 207, 200, 192, 192, 192,
57         240, 255, 231, 180, 108, 204, 248, 103, 207, 154, 198, 128, 17, 255, 44, 104, 230, 253,
58         103, 96, 96, 96, 96, 76, 93, 28, 151, 160, 59, 31, 18, 156, 115, 214, 255, 213, 84, 100,
59         72, 93, 156, 136, 53, 84, 25, 41, 76, 255, 0, 17, 24, 113, 221];
60     inflate::inflate_bytes_zlib(data);
61 }
62 
63 #[test]
issue_30_small()64 fn issue_30_small() {
65     // Invalid data, but should not panic
66     // generated by cargo-fuzz
67     let data = &[0x08, 0xd7, 0x3a, 0xff, 0x3b, 0x8, 0xd7, 0xfd, 0xff, 0xff, 0xff];
68     inflate::inflate_bytes_zlib(data);
69 }
70 
71 #[test]
72 // no checksum present at the end of the data stream (cargo-fuzz test-case)
no_checksum()73 fn no_checksum() {
74     let data = b"\x13\xff\xed\xff\xff\x12\xbfM\x00\x00\x00\x00\xd1";
75     let mut stream = inflate::InflateStream::new();
76     let res = stream.update(data);
77     // This is not an error, because the checksum may be included in the next
78     // call to `stream.update`. See issue #27.
79     assert!(res.is_ok());
80 }
81 
82 #[test]
83 /// The first byte of the CRC is already read into the BitStream buffer.
issue_26()84 fn issue_26() {
85     let data = &[120, 156, 189, 138, 65, 13, 0, 32, 16, 195, 64, 2, 22, 176, 128, 5, 44, 96, 1, 11,
86         216, 103, 19, 176, 123, 118, 73, 155, 61, 218, 155, 54, 10, 136, 192, 170, 32, 130, 41,
87         249, 36, 136, 96, 73, 62, 9, 34, 216, 146, 79, 130, 8, 142, 228, 147, 32, 130, 43, 249, 36,
88         136, 224, 73, 62, 9, 32, 248, 250, 192, 22, 113, 123];
89     let mut stream = inflate::InflateStream::from_zlib();
90     let res = stream.update(data);
91     assert!(res.is_ok());
92 }
93 
94 #[test]
95 /// InflateStream::update() should try to process bytes in the Bits state.
issue_38_update_bits()96 fn issue_38_update_bits() {
97     // compressed final block 011, literal 111111111=255, end 0000000
98     let data = &[0b1111_1011, 0b0000_1111, 0b0000_0000];
99     let bytes = inflate::inflate_bytes(data).unwrap();
100     assert_eq!(bytes.len(), 1);
101 }
102 
103 #[test]
104 /// InflateStream::update() should try to process bytes in the LenDist state.
issue_38_update_lendist()105 fn issue_38_update_lendist() {
106     // compressed final block 011, literal 00110000=0 x 32766, len 0000001=3, dist 00000=1, end 0000000
107     let mut data = vec![0b0110_0011];
108     data.extend(std::iter::repeat(0b0110_0000).take(32765));
109     data.extend(vec![0b0000_0000, 0b0000_0010, 0b0000_00000]);
110     let bytes = inflate::inflate_bytes(&data).unwrap();
111     assert_eq!(bytes.len(), 32769);
112 }
113 
114 #[test]
115 /// InflateStream::update() should return if it can't make progress
issue_38_update_no_progress()116 fn issue_38_update_no_progress() {
117     // compressed final block 011, 110010000 x 5, 110010000 truncated to 8 bits
118     let data = &[0b1001_1011, 0b0011_0000, 0b0110_0001, 0b1100_0010, 0b10000100, 0b0000_1001, 0b00010011];
119     let bytes = inflate::inflate_bytes(data).unwrap();
120     assert_eq!(bytes.len(), 5);
121 }
122 
123 #[test]
124 /// inflate() should loop until no data is produced.
issue_38_inflate()125 fn issue_38_inflate() {
126     // compressed final block 011, literal 00110000=0 x 32769, end 0000000
127     let mut data = vec![0b0110_0011];
128     data.extend(std::iter::repeat(0b0110_0000).take(32768));
129     data.extend(vec![0b0000_0000, 0b0000_0000]);
130     let bytes = inflate::inflate_bytes(&data).unwrap();
131     assert_eq!(bytes.len(), 32769);
132 }
133 
134 #[test]
135 /// InflateStream::update() should handle end of input between uncompressed len and nlen.
issue_47_uncompressed()136 fn issue_47_uncompressed() {
137     let len = 50;
138     let hilen = (len >> 8) as u8;
139     let lolen = (len & 0xff) as u8;
140     // uncompressed block 001
141     let mut data = vec![0b001, lolen, hilen, !lolen, !hilen];
142     data.extend(std::iter::repeat(0).take(len + 1));
143     let mut stream = inflate::InflateStream::new();
144     {
145         // Update with input to the end of len.
146         let (update_len, bytes) = stream.update(&data[..3]).unwrap();
147         assert_eq!(update_len, 3);
148         assert_eq!(bytes, &[]);
149     }
150     {
151         // Update with remainder of input.
152         let (update_len, bytes) = stream.update(&data[3..]).unwrap();
153         assert_eq!(update_len, data.len() - 3);
154         assert_eq!(bytes.len(), len);
155     }
156 }
157