f9306f6de3
C++20 comes with std::erase to erase a value from std::vector. This patch renames llvm::erase_value to llvm::erase for consistency with C++20. We could make llvm::erase more similar to std::erase by having it return the number of elements removed, but I'm not doing that for now because nobody seems to care about that in our code base. Since there are only 50 occurrences of erase_value in our code base, this patch replaces all of them with llvm::erase and deprecates llvm::erase_value.
46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
//===-- yaml-numeric-parser-fuzzer.cpp - Fuzzer for YAML numeric parser ---===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Support/Regex.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
#include <string>
|
|
|
|
inline bool isNumericRegex(llvm::StringRef S) {
|
|
static llvm::Regex Infinity("^[-+]?(\\.inf|\\.Inf|\\.INF)$");
|
|
static llvm::Regex Base8("^0o[0-7]+$");
|
|
static llvm::Regex Base16("^0x[0-9a-fA-F]+$");
|
|
static llvm::Regex Float(
|
|
"^[-+]?(\\.[0-9]+|[0-9]+(\\.[0-9]*)?)([eE][-+]?[0-9]+)?$");
|
|
|
|
if (S.equals(".nan") || S.equals(".NaN") || S.equals(".NAN"))
|
|
return true;
|
|
|
|
if (Infinity.match(S))
|
|
return true;
|
|
|
|
if (Base8.match(S))
|
|
return true;
|
|
|
|
if (Base16.match(S))
|
|
return true;
|
|
|
|
if (Float.match(S))
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
|
|
std::string Input(reinterpret_cast<const char *>(Data), Size);
|
|
llvm::erase(Input, 0);
|
|
if (!Input.empty() && llvm::yaml::isNumeric(Input) != isNumericRegex(Input))
|
|
LLVM_BUILTIN_TRAP;
|
|
return 0;
|
|
}
|