forked from OSchip/llvm-project
47 lines
1.3 KiB
C++
47 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 <cassert>
|
|
#include <string>
|
|
|
|
llvm::Regex Infinity("^[-+]?(\\.inf|\\.Inf|\\.INF)$");
|
|
llvm::Regex Base8("^0o[0-7]+$");
|
|
llvm::Regex Base16("^0x[0-9a-fA-F]+$");
|
|
llvm::Regex Float("^[-+]?(\\.[0-9]+|[0-9]+(\\.[0-9]*)?)([eE][-+]?[0-9]+)?$");
|
|
|
|
inline bool isNumericRegex(llvm::StringRef S) {
|
|
|
|
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_value(Input, 0);
|
|
if (!Input.empty() && llvm::yaml::isNumeric(Input) != isNumericRegex(Input))
|
|
LLVM_BUILTIN_TRAP;
|
|
return 0;
|
|
}
|