1//===-- yaml-numeric-parser-fuzzer.cpp - Fuzzer for YAML numeric parser ---===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "llvm/ADT/StringRef.h"
10#include "llvm/Support/Regex.h"
11#include "llvm/Support/YAMLTraits.h"
12#include <string>
13
14inline bool isNumericRegex(llvm::StringRef S) {
15 static llvm::Regex Infinity("^[-+]?(\\.inf|\\.Inf|\\.INF)$");
16 static llvm::Regex Base8("^0o[0-7]+$");
17 static llvm::Regex Base16("^0x[0-9a-fA-F]+$");
18 static llvm::Regex Float(
19 "^[-+]?(\\.[0-9]+|[0-9]+(\\.[0-9]*)?)([eE][-+]?[0-9]+)?$");
20
21 if (S.equals(RHS: ".nan") || S.equals(RHS: ".NaN") || S.equals(RHS: ".NAN"))
22 return true;
23
24 if (Infinity.match(String: S))
25 return true;
26
27 if (Base8.match(String: S))
28 return true;
29
30 if (Base16.match(String: S))
31 return true;
32
33 if (Float.match(String: S))
34 return true;
35
36 return false;
37}
38
39extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
40 std::string Input(reinterpret_cast<const char *>(Data), Size);
41 llvm::erase(C&: Input, V: 0);
42 if (!Input.empty() && llvm::yaml::isNumeric(S: Input) != isNumericRegex(S: Input))
43 LLVM_BUILTIN_TRAP;
44 return 0;
45}
46

source code of llvm/tools/llvm-yaml-numeric-parser-fuzzer/yaml-numeric-parser-fuzzer.cpp