Skip to content

Commit c3f2992

Browse files
michielp1807folkertdev
authored andcommitted
Fix fuzz tests nb_samples
1 parent a066e1a commit c3f2992

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

fuzz/fuzz_targets/dictionary_train.rs

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ fuzz_target!(|data: (u8, u8, ArbitrarySamples)| {
1111
let k = d + (k as u32 % 256);
1212

1313
// train
14-
let (_dict_size, dict) = assert_eq_rs_c!({
14+
assert_eq_rs_c!({
1515
let mut params = libzstd_rs_sys::ZDICT_cover_params_t::default();
1616
params.d = d;
1717
params.k = k;
@@ -34,14 +34,13 @@ fuzz_target!(|data: (u8, u8, ArbitrarySamples)| {
3434
let mut opt_params = libzstd_rs_sys::ZDICT_cover_params_t::default();
3535
opt_params.steps = 4;
3636

37-
let mut dict = dict.clone();
38-
37+
let mut dict = vec![0u8; samples.dict_size];
3938
let opt_dict_size = ZDICT_optimizeTrainFromBuffer_cover(
4039
dict.as_mut_ptr().cast(),
4140
samples.dict_size,
4241
samples.samples.as_ptr().cast(),
4342
samples.sample_sizes.as_ptr(),
44-
samples.nb_samples as u32 / 4,
43+
samples.nb_samples as u32,
4544
std::mem::transmute(&mut opt_params),
4645
);
4746

fuzz/fuzz_targets/dictionary_train_fast.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ fuzz_target!(|data: (u8, u8, ArbitrarySamples)| {
4040
samples.dict_size,
4141
samples.samples.as_ptr().cast(),
4242
samples.sample_sizes.as_ptr(),
43-
samples.nb_samples as u32 / 4,
43+
samples.nb_samples as u32,
4444
std::mem::transmute(&mut opt_params),
4545
);
4646

0 commit comments

Comments
 (0)