xref: /aosp_15_r20/external/fmtlib/test/fuzzing/float.cc (revision 5c90c05cd622c0a81b57953a4d343e0e489f2e08)
1*5c90c05cSAndroid Build Coastguard Worker // A fuzzer for floating-point formatter.
2*5c90c05cSAndroid Build Coastguard Worker // For the license information refer to format.h.
3*5c90c05cSAndroid Build Coastguard Worker 
4*5c90c05cSAndroid Build Coastguard Worker #include <fmt/format.h>
5*5c90c05cSAndroid Build Coastguard Worker 
6*5c90c05cSAndroid Build Coastguard Worker #include <cstdint>
7*5c90c05cSAndroid Build Coastguard Worker #include <cstdlib>
8*5c90c05cSAndroid Build Coastguard Worker #include <limits>
9*5c90c05cSAndroid Build Coastguard Worker #include <stdexcept>
10*5c90c05cSAndroid Build Coastguard Worker 
11*5c90c05cSAndroid Build Coastguard Worker #include "fuzzer-common.h"
12*5c90c05cSAndroid Build Coastguard Worker 
check_round_trip(fmt::string_view format_str,double value)13*5c90c05cSAndroid Build Coastguard Worker void check_round_trip(fmt::string_view format_str, double value) {
14*5c90c05cSAndroid Build Coastguard Worker   auto buffer = fmt::memory_buffer();
15*5c90c05cSAndroid Build Coastguard Worker   fmt::format_to(std::back_inserter(buffer), format_str, value);
16*5c90c05cSAndroid Build Coastguard Worker 
17*5c90c05cSAndroid Build Coastguard Worker   if (std::isnan(value)) {
18*5c90c05cSAndroid Build Coastguard Worker     auto nan = std::signbit(value) ? "-nan" : "nan";
19*5c90c05cSAndroid Build Coastguard Worker     if (fmt::string_view(buffer.data(), buffer.size()) != nan)
20*5c90c05cSAndroid Build Coastguard Worker       throw std::runtime_error("round trip failure");
21*5c90c05cSAndroid Build Coastguard Worker     return;
22*5c90c05cSAndroid Build Coastguard Worker   }
23*5c90c05cSAndroid Build Coastguard Worker 
24*5c90c05cSAndroid Build Coastguard Worker   buffer.push_back('\0');
25*5c90c05cSAndroid Build Coastguard Worker   char* ptr = nullptr;
26*5c90c05cSAndroid Build Coastguard Worker   if (std::strtod(buffer.data(), &ptr) != value)
27*5c90c05cSAndroid Build Coastguard Worker     throw std::runtime_error("round trip failure");
28*5c90c05cSAndroid Build Coastguard Worker   if (ptr + 1 != buffer.end()) throw std::runtime_error("unparsed output");
29*5c90c05cSAndroid Build Coastguard Worker }
30*5c90c05cSAndroid Build Coastguard Worker 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)31*5c90c05cSAndroid Build Coastguard Worker extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
32*5c90c05cSAndroid Build Coastguard Worker   if (size <= sizeof(double) || !std::numeric_limits<double>::is_iec559)
33*5c90c05cSAndroid Build Coastguard Worker     return 0;
34*5c90c05cSAndroid Build Coastguard Worker   check_round_trip("{}", assign_from_buf<double>(data));
35*5c90c05cSAndroid Build Coastguard Worker   // A larger than necessary precision is used to trigger the fallback
36*5c90c05cSAndroid Build Coastguard Worker   // formatter.
37*5c90c05cSAndroid Build Coastguard Worker   check_round_trip("{:.50g}", assign_from_buf<double>(data));
38*5c90c05cSAndroid Build Coastguard Worker   return 0;
39*5c90c05cSAndroid Build Coastguard Worker }
40