Changeset View
Changeset View
Standalone View
Standalone View
src/test/fuzz/buffered_file.cpp
- This file was added.
// Copyright (c) 2020 The Bitcoin Core developers | |||||
// Distributed under the MIT software license, see the accompanying | |||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php. | |||||
#include <streams.h> | |||||
#include <test/fuzz/FuzzedDataProvider.h> | |||||
#include <test/fuzz/fuzz.h> | |||||
#include <test/fuzz/util.h> | |||||
#include <array> | |||||
#include <cstdint> | |||||
#include <iostream> | |||||
#include <optional> | |||||
#include <string> | |||||
#include <vector> | |||||
void test_one_input(const std::vector<uint8_t> &buffer) { | |||||
FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()}; | |||||
FuzzedFileProvider fuzzed_file_provider = ConsumeFile(fuzzed_data_provider); | |||||
std::optional<CBufferedFile> opt_buffered_file; | |||||
FILE *fuzzed_file = fuzzed_file_provider.open(); | |||||
try { | |||||
opt_buffered_file.emplace( | |||||
fuzzed_file, | |||||
fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096), | |||||
fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096), | |||||
fuzzed_data_provider.ConsumeIntegral<int>(), | |||||
fuzzed_data_provider.ConsumeIntegral<int>()); | |||||
} catch (const std::ios_base::failure &) { | |||||
if (fuzzed_file != nullptr) { | |||||
fclose(fuzzed_file); | |||||
} | |||||
} | |||||
if (opt_buffered_file && fuzzed_file != nullptr) { | |||||
bool setpos_fail = false; | |||||
while (fuzzed_data_provider.ConsumeBool()) { | |||||
switch (fuzzed_data_provider.ConsumeIntegralInRange<int>(0, 5)) { | |||||
case 0: { | |||||
std::array<uint8_t, 4096> arr{}; | |||||
try { | |||||
opt_buffered_file->read( | |||||
(char *)arr.data(), | |||||
fuzzed_data_provider.ConsumeIntegralInRange<size_t>( | |||||
0, 4096)); | |||||
} catch (const std::ios_base::failure &) { | |||||
} | |||||
break; | |||||
} | |||||
case 1: { | |||||
opt_buffered_file->Seek( | |||||
fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>( | |||||
0, 4096)); | |||||
break; | |||||
} | |||||
case 2: { | |||||
opt_buffered_file->SetLimit( | |||||
fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>( | |||||
0, 4096)); | |||||
break; | |||||
} | |||||
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 { | |||||
opt_buffered_file->FindByte( | |||||
fuzzed_data_provider.ConsumeIntegral<char>()); | |||||
} catch (const std::ios_base::failure &) { | |||||
} | |||||
break; | |||||
} | |||||
case 5: { | |||||
ReadFromStream(fuzzed_data_provider, *opt_buffered_file); | |||||
break; | |||||
} | |||||
} | |||||
} | |||||
opt_buffered_file->GetPos(); | |||||
opt_buffered_file->GetType(); | |||||
opt_buffered_file->GetVersion(); | |||||
} | |||||
} |