mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-01-20 15:19:07 +01:00
8bb34f07df
There is no way to report a close error from `AutoFile` destructor. Such an error could be serious if the file has been written to because it may mean the file is now corrupted (same as if write fails). So, change all users of `AutoFile` that use it to write data to explicitly close the file and handle a possible error.
69 lines
2.1 KiB
C++
69 lines
2.1 KiB
C++
// Copyright (c) 2020-2022 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 <span.h>
|
|
#include <streams.h>
|
|
#include <test/fuzz/FuzzedDataProvider.h>
|
|
#include <test/fuzz/fuzz.h>
|
|
#include <test/fuzz/util.h>
|
|
|
|
#include <array>
|
|
#include <cstddef>
|
|
#include <cstdio>
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
FUZZ_TARGET(autofile)
|
|
{
|
|
FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
|
|
FuzzedFileProvider fuzzed_file_provider{fuzzed_data_provider};
|
|
AutoFile auto_file{
|
|
fuzzed_file_provider.open(),
|
|
ConsumeRandomLengthByteVector<std::byte>(fuzzed_data_provider),
|
|
};
|
|
LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 100)
|
|
{
|
|
CallOneOf(
|
|
fuzzed_data_provider,
|
|
[&] {
|
|
std::array<std::byte, 4096> arr{};
|
|
try {
|
|
auto_file.read({arr.data(), fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096)});
|
|
} catch (const std::ios_base::failure&) {
|
|
}
|
|
},
|
|
[&] {
|
|
const std::array<std::byte, 4096> arr{};
|
|
try {
|
|
auto_file.write({arr.data(), fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096)});
|
|
} catch (const std::ios_base::failure&) {
|
|
}
|
|
},
|
|
[&] {
|
|
try {
|
|
auto_file.ignore(fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096));
|
|
} catch (const std::ios_base::failure&) {
|
|
}
|
|
},
|
|
[&] {
|
|
(void)auto_file.fclose();
|
|
},
|
|
[&] {
|
|
ReadFromStream(fuzzed_data_provider, auto_file);
|
|
},
|
|
[&] {
|
|
WriteToStream(fuzzed_data_provider, auto_file);
|
|
});
|
|
}
|
|
(void)auto_file.IsNull();
|
|
if (fuzzed_data_provider.ConsumeBool()) {
|
|
FILE* f = auto_file.release();
|
|
if (f != nullptr) {
|
|
fclose(f);
|
|
}
|
|
} else {
|
|
(void)auto_file.fclose();
|
|
}
|
|
}
|