| ... | ... | @@ -234,3 +234,114 @@ test "flate compress/decompress" { |
| 234 | 234 | } |
| 235 | 235 | } |
| 236 | 236 | } |
| 237 | |
| 238 | fn testDecompress(comptime container: Container, compressed: []const u8, expected_plain: []const u8) !void { |
| 239 | var in_stream = std.io.fixedBufferStream(compressed); |
| 240 | var al = std.ArrayList(u8).init(testing.allocator); |
| 241 | defer al.deinit(); |
| 242 | |
| 243 | try inflate.decompress(container, in_stream.reader(), al.writer()); |
| 244 | |
| 245 | try testing.expectEqualSlices(u8, expected_plain, al.items); |
| 246 | } |
| 247 | |
| 248 | test "flate don't read past deflate stream's end" { |
| 249 | try testDecompress(.zlib, &[_]u8{ |
| 250 | 0x08, 0xd7, 0x63, 0xf8, 0xcf, 0xc0, 0xc0, 0x00, 0xc1, 0xff, |
| 251 | 0xff, 0x43, 0x30, 0x03, 0x03, 0xc3, 0xff, 0xff, 0xff, 0x01, |
| 252 | 0x83, 0x95, 0x0b, 0xf5, |
| 253 | }, &[_]u8{ |
| 254 | 0x00, 0xff, 0x00, 0x00, 0x00, 0xff, 0x00, 0x00, 0x00, 0xff, |
| 255 | 0x00, 0xff, 0xff, 0xff, 0x00, 0xff, 0xff, 0xff, 0x00, 0x00, |
| 256 | 0x00, 0x00, 0xff, 0xff, 0xff, |
| 257 | }); |
| 258 | } |
| 259 | |
| 260 | test "flate zlib header" { |
| 261 | // Truncated header |
| 262 | try testing.expectError( |
| 263 | error.EndOfStream, |
| 264 | testDecompress(.zlib, &[_]u8{0x78}, ""), |
| 265 | ); |
| 266 | // Wrong CM |
| 267 | try testing.expectError( |
| 268 | error.BadZlibHeader, |
| 269 | testDecompress(.zlib, &[_]u8{ 0x79, 0x94 }, ""), |
| 270 | ); |
| 271 | // Wrong CINFO |
| 272 | try testing.expectError( |
| 273 | error.BadZlibHeader, |
| 274 | testDecompress(.zlib, &[_]u8{ 0x88, 0x98 }, ""), |
| 275 | ); |
| 276 | // Wrong checksum |
| 277 | try testing.expectError( |
| 278 | error.WrongZlibChecksum, |
| 279 | testDecompress(.zlib, &[_]u8{ 0x78, 0xda, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00 }, ""), |
| 280 | ); |
| 281 | // Truncated checksum |
| 282 | try testing.expectError( |
| 283 | error.EndOfStream, |
| 284 | testDecompress(.zlib, &[_]u8{ 0x78, 0xda, 0x03, 0x00, 0x00 }, ""), |
| 285 | ); |
| 286 | } |
| 287 | |
| 288 | test "flate gzip header" { |
| 289 | // Truncated header |
| 290 | try testing.expectError( |
| 291 | error.EndOfStream, |
| 292 | testDecompress(.gzip, &[_]u8{ 0x1f, 0x8B }, undefined), |
| 293 | ); |
| 294 | // Wrong CM |
| 295 | try testing.expectError( |
| 296 | error.BadGzipHeader, |
| 297 | testDecompress(.gzip, &[_]u8{ |
| 298 | 0x1f, 0x8b, 0x09, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 299 | 0x00, 0x03, |
| 300 | }, undefined), |
| 301 | ); |
| 302 | |
| 303 | // Wrong checksum |
| 304 | try testing.expectError( |
| 305 | error.WrongGzipChecksum, |
| 306 | testDecompress(.gzip, &[_]u8{ |
| 307 | 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 308 | 0x00, 0x03, 0x03, 0x00, 0x00, 0x00, 0x00, 0x01, |
| 309 | 0x00, 0x00, 0x00, 0x00, |
| 310 | }, undefined), |
| 311 | ); |
| 312 | // Truncated checksum |
| 313 | try testing.expectError( |
| 314 | error.EndOfStream, |
| 315 | testDecompress(.gzip, &[_]u8{ |
| 316 | 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 317 | 0x00, 0x03, 0x03, 0x00, 0x00, 0x00, 0x00, |
| 318 | }, undefined), |
| 319 | ); |
| 320 | // Wrong initial size |
| 321 | try testing.expectError( |
| 322 | error.WrongGzipSize, |
| 323 | testDecompress(.gzip, &[_]u8{ |
| 324 | 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 325 | 0x00, 0x03, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 326 | 0x00, 0x00, 0x00, 0x01, |
| 327 | }, undefined), |
| 328 | ); |
| 329 | // Truncated initial size field |
| 330 | try testing.expectError( |
| 331 | error.EndOfStream, |
| 332 | testDecompress(.gzip, &[_]u8{ |
| 333 | 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 334 | 0x00, 0x03, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 335 | 0x00, 0x00, 0x00, |
| 336 | }, undefined), |
| 337 | ); |
| 338 | |
| 339 | try testDecompress(.gzip, &[_]u8{ |
| 340 | // GZIP header |
| 341 | 0x1f, 0x8b, 0x08, 0x12, 0x00, 0x09, 0x6e, 0x88, 0x00, 0xff, 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x00, |
| 342 | // header.FHCRC (should cover entire header) |
| 343 | 0x99, 0xd6, |
| 344 | // GZIP data |
| 345 | 0x01, 0x00, 0x00, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 346 | }, ""); |
| 347 | } |
Igor Anić <igor.anic@gmail.com> 2024-02-14 22:12:54+01:00
lib