DevComp/zip-rs-wasm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Include comments in fuzz_write

Browse source
This commit is contained in:
Chris Hennick 2023-05-21 15:34:22 -07:00
parent bef9fce30a
commit 235d2da745
No known key found for this signature in database
GPG key ID: 25653935CC8B6C74
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
fuzz/fuzz_targets/fuzz_write.rs
View file

@ -76,16 +76,19 @@ fn do_operation<T>(writer: &mut RefCell<zip_next::ZipWriter<T>>,
writer.borrow_mut().abort_file().unwrap(); writer.borrow_mut().abort_file().unwrap();
} }
if operation.reopen { if operation.reopen {
let mut new_writer = zip_next::ZipWriter::new_append(writer.borrow_mut().finish().unwrap()).unwrap(); let old_comment = writer.borrow().get_raw_comment().to_owned();
assert_eq!(Ok(""), new_writer.get_comment()); let new_writer = zip_next::ZipWriter::new_append(writer.borrow_mut().finish().unwrap()).unwrap();
assert_eq!(&old_comment, new_writer.get_raw_comment());
*writer = new_writer.into(); *writer = new_writer.into();
} }
Ok(()) Ok(())
} }
fuzz_target!(|data: Vec<(FileOperation, bool)>| { fuzz_target!(|data: (Vec<u8>, Vec<(FileOperation, bool)>)| {
let (comment, operations) = data;
let mut writer = RefCell::new(zip_next::ZipWriter::new(Cursor::new(Vec::new()))); let mut writer = RefCell::new(zip_next::ZipWriter::new(Cursor::new(Vec::new())));
for (operation, abort) in data { writer.borrow_mut().set_raw_comment(comment);
for (operation, abort) in operations {
let _ = do_operation(&mut writer, operation, abort); let _ = do_operation(&mut writer, operation, abort);
} }
let _ = zip_next::ZipArchive::new(writer.borrow_mut().finish().unwrap()); let _ = zip_next::ZipArchive::new(writer.borrow_mut().finish().unwrap());