tests: Add fuzzing harness for CBufferedFile::{SetPos,GetPos,GetType,GetVersion} (stream.h)

This commit is contained in:
practicalswift 2020-07-15 00:04:46 +00:00
parent 7bcc71e5f8
commit 614e0807a8

View file

@ -29,8 +29,9 @@ void test_one_input(const std::vector<uint8_t>& buffer)
} }
} }
if (opt_buffered_file && fuzzed_file != nullptr) { if (opt_buffered_file && fuzzed_file != nullptr) {
bool setpos_fail = false;
while (fuzzed_data_provider.ConsumeBool()) { while (fuzzed_data_provider.ConsumeBool()) {
switch (fuzzed_data_provider.ConsumeIntegralInRange<int>(0, 4)) { switch (fuzzed_data_provider.ConsumeIntegralInRange<int>(0, 5)) {
case 0: { case 0: {
std::array<uint8_t, 4096> arr{}; std::array<uint8_t, 4096> arr{};
try { try {
@ -48,17 +49,30 @@ void test_one_input(const std::vector<uint8_t>& buffer)
break; break;
} }
case 3: { case 3: {
if (!opt_buffered_file->SetPos(fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096))) {
setpos_fail = true;
}
break;
}
case 4: {
if (setpos_fail) {
// Calling FindByte(...) after a failed SetPos(...) call may result in an infinite loop.
break;
}
try { try {
opt_buffered_file->FindByte(fuzzed_data_provider.ConsumeIntegral<char>()); opt_buffered_file->FindByte(fuzzed_data_provider.ConsumeIntegral<char>());
} catch (const std::ios_base::failure&) { } catch (const std::ios_base::failure&) {
} }
break; break;
} }
case 4: { case 5: {
ReadFromStream(fuzzed_data_provider, *opt_buffered_file); ReadFromStream(fuzzed_data_provider, *opt_buffered_file);
break; break;
} }
} }
} }
opt_buffered_file->GetPos();
opt_buffered_file->GetType();
opt_buffered_file->GetVersion();
} }
} }