1 //===- FuzzerUtil.h - Internal header for the Fuzzer Utils ------*- C++ -* ===//
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
7 //===----------------------------------------------------------------------===//
9 //===----------------------------------------------------------------------===//
11 #ifndef LLVM_FUZZER_UTIL_H
12 #define LLVM_FUZZER_UTIL_H
14 #include "FuzzerBuiltins.h"
15 #include "FuzzerBuiltinsMsvc.h"
16 #include "FuzzerCommand.h"
17 #include "FuzzerDefs.h"
21 void PrintHexArray(const Unit &U, const char *PrintAfter = "");
23 void PrintHexArray(const uint8_t *Data, size_t Size,
24 const char *PrintAfter = "");
26 void PrintASCII(const uint8_t *Data, size_t Size, const char *PrintAfter = "");
28 void PrintASCII(const Unit &U, const char *PrintAfter = "");
30 // Changes U to contain only ASCII (isprint+isspace) characters.
31 // Returns true iff U has been changed.
32 bool ToASCII(uint8_t *Data, size_t Size);
34 bool IsASCII(const Unit &U);
36 bool IsASCII(const uint8_t *Data, size_t Size);
38 std::string Base64(const Unit &U);
40 void PrintPC(const char *SymbolizedFMT, const char *FallbackFMT, uintptr_t PC);
42 std::string DescribePC(const char *SymbolizedFMT, uintptr_t PC);
44 void PrintStackTrace();
46 void PrintMemoryProfile();
48 unsigned NumberOfCpuCores();
50 // Platform specific functions.
51 void SetSignalHandler(const FuzzingOptions& Options);
53 void SleepSeconds(int Seconds);
55 bool Mprotect(void *Ptr, size_t Size, bool AllowReadWrite);
57 unsigned long GetPid();
59 size_t GetPeakRSSMb();
61 int ExecuteCommand(const Command &Cmd);
63 FILE *OpenProcessPipe(const char *Command, const char *Mode);
65 const void *SearchMemory(const void *haystack, size_t haystacklen,
66 const void *needle, size_t needlelen);
68 std::string CloneArgsWithoutX(const Vector<std::string> &Args,
69 const char *X1, const char *X2);
71 inline std::string CloneArgsWithoutX(const Vector<std::string> &Args,
73 return CloneArgsWithoutX(Args, X, X);
76 inline std::pair<std::string, std::string> SplitBefore(std::string X,
79 if (Pos == std::string::npos)
80 return std::make_pair(S, "");
81 return std::make_pair(S.substr(0, Pos), S.substr(Pos));
84 std::string DisassembleCmd(const std::string &FileName);
86 std::string SearchRegexCmd(const std::string &Regex);
88 size_t SimpleFastHash(const uint8_t *Data, size_t Size);
90 inline uint32_t Log(uint32_t X) { return 32 - Clz(X) - 1; }
92 inline size_t PageSize() { return 4096; }
93 inline uint8_t *RoundUpByPage(uint8_t *P) {
94 uintptr_t X = reinterpret_cast<uintptr_t>(P);
95 size_t Mask = PageSize() - 1;
96 X = (X + Mask) & ~Mask;
97 return reinterpret_cast<uint8_t *>(X);
99 inline uint8_t *RoundDownByPage(uint8_t *P) {
100 uintptr_t X = reinterpret_cast<uintptr_t>(P);
101 size_t Mask = PageSize() - 1;
103 return reinterpret_cast<uint8_t *>(X);
106 } // namespace fuzzer
108 #endif // LLVM_FUZZER_UTIL_H