Page MenuHomePhabricator
Differential D8969 Diff 27093 src/test/fuzz/multiplication_overflow.cpp

Changeset View

Standalone View

View Options

src/test/fuzz/multiplication_overflow.cpp

// Copyright (c) 2020 The Bitcoin Core developers // Copyright (c) 2020 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying // Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php. // file COPYING or http://www.opensource.org/licenses/mit-license.php.
#include <test/fuzz/FuzzedDataProvider.h> #include <test/fuzz/FuzzedDataProvider.h>
#include <test/fuzz/fuzz.h> #include <test/fuzz/fuzz.h>
#include <test/fuzz/util.h> #include <test/fuzz/util.h>
#include <cstdint> #include <cstdint>
#include <string> #include <string>
#include <vector> #include <vector>
#if defined(__has_builtin)
#if __has_builtin(__builtin_mul_overflow)
#define HAVE_BUILTIN_MUL_OVERFLOW
#endif
#elif defined(__GNUC__) && (__GNUC__ >= 5)
#define HAVE_BUILTIN_MUL_OVERFLOW
#endif
namespace { namespace {
template <typename T> template <typename T>
void TestMultiplicationOverflow(FuzzedDataProvider &fuzzed_data_provider) { void TestMultiplicationOverflow(FuzzedDataProvider &fuzzed_data_provider) {
const T i = fuzzed_data_provider.ConsumeIntegral<T>(); const T i = fuzzed_data_provider.ConsumeIntegral<T>();
const T j = fuzzed_data_provider.ConsumeIntegral<T>(); const T j = fuzzed_data_provider.ConsumeIntegral<T>();
const bool is_multiplication_overflow_custom = MultiplicationOverflow(i, j); const bool is_multiplication_overflow_custom = MultiplicationOverflow(i, j);
#if defined(HAVE_BUILTIN_MUL_OVERFLOW)
T result_builtin; T result_builtin;
const bool is_multiplication_overflow_builtin = const bool is_multiplication_overflow_builtin =
__builtin_mul_overflow(i, j, &result_builtin); __builtin_mul_overflow(i, j, &result_builtin);
assert(is_multiplication_overflow_custom == assert(is_multiplication_overflow_custom ==
is_multiplication_overflow_builtin); is_multiplication_overflow_builtin);
if (!is_multiplication_overflow_custom) { if (!is_multiplication_overflow_custom) {
assert(i * j == result_builtin); assert(i * j == result_builtin);
} }
#else
if (!is_multiplication_overflow_custom) {
(void)(i * j);
}
#endif
} }
} // namespace } // namespace
void test_one_input(const std::vector<uint8_t> &buffer) { void test_one_input(const std::vector<uint8_t> &buffer) {
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size()); FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
TestMultiplicationOverflow<int64_t>(fuzzed_data_provider); TestMultiplicationOverflow<int64_t>(fuzzed_data_provider);
TestMultiplicationOverflow<uint64_t>(fuzzed_data_provider); TestMultiplicationOverflow<uint64_t>(fuzzed_data_provider);
TestMultiplicationOverflow<int32_t>(fuzzed_data_provider); TestMultiplicationOverflow<int32_t>(fuzzed_data_provider);
TestMultiplicationOverflow<uint32_t>(fuzzed_data_provider); TestMultiplicationOverflow<uint32_t>(fuzzed_data_provider);
TestMultiplicationOverflow<int16_t>(fuzzed_data_provider); TestMultiplicationOverflow<int16_t>(fuzzed_data_provider);
TestMultiplicationOverflow<uint16_t>(fuzzed_data_provider); TestMultiplicationOverflow<uint16_t>(fuzzed_data_provider);
TestMultiplicationOverflow<char>(fuzzed_data_provider); TestMultiplicationOverflow<char>(fuzzed_data_provider);
TestMultiplicationOverflow<uint8_t>(fuzzed_data_provider); TestMultiplicationOverflow<uint8_t>(fuzzed_data_provider);
TestMultiplicationOverflow<signed char>(fuzzed_data_provider); TestMultiplicationOverflow<signed char>(fuzzed_data_provider);
TestMultiplicationOverflow<bool>(fuzzed_data_provider);
} }