mirror of
https://github.com/beefytech/Beef.git
synced 2025-06-08 11:38:21 +02:00
Add BeefFuzz for Beef fuzzing
This commit is contained in:
parent
84bc0ad1de
commit
e16ed4bfbd
7 changed files with 1017 additions and 0 deletions
87
BeefFuzz/BeefFuzz.cpp
Normal file
87
BeefFuzz/BeefFuzz.cpp
Normal file
|
@ -0,0 +1,87 @@
|
|||
#pragma warning(disable:4996)
|
||||
|
||||
#include <iostream>
|
||||
#include "BeefySysLib/Common.h"
|
||||
#include "BeefySysLib/util/Array.h"
|
||||
#include "BeefySysLib/util/SizedArray.h"
|
||||
#include "BeefySysLib/util/Dictionary.h"
|
||||
#include "BeefySysLib/util/CabUtil.h"
|
||||
#include "BeefySysLib/util/BeefPerf.h"
|
||||
#include "BeefySysLib/util/Deque.h"
|
||||
#include "BeefySysLib/util/HashSet.h"
|
||||
#include "BeefySysLib/util/MultiHashSet.h"
|
||||
|
||||
#define BF_DBG_64
|
||||
#include "IDEHelper/StrHashMap.h"
|
||||
|
||||
using namespace Beefy;
|
||||
|
||||
#include "FuzzApp.h"
|
||||
|
||||
BF_IMPORT void BF_CALLTYPE IDEHelper_ProgramStart();
|
||||
BF_IMPORT void BF_CALLTYPE IDEHelper_ProgramDone();
|
||||
|
||||
static void FuzzInit()
|
||||
{
|
||||
BfpSystem_SetCommandLine(0, NULL);
|
||||
|
||||
BfpThread_SetName(NULL, "MainThread", NULL);
|
||||
|
||||
BfpSystem_Init(BFP_VERSION, BfpSystemInitFlag_None);
|
||||
|
||||
IDEHelper_ProgramStart();
|
||||
|
||||
gApp = new FuzzApp();
|
||||
gApp->SetTargetPath("fuzz_testd");
|
||||
gApp->AddDefine("CLI");
|
||||
gApp->AddDefine("DEBUG");
|
||||
gApp->SetStartupObject("fuzz_test.Program");
|
||||
gApp->SetLinkParams("./libBeefRT_d.a ./libBeefySysLib_d.so -ldl -lpthread -Wl,-rpath -Wl,$ORIGIN");
|
||||
|
||||
BF_ASSERT(gApp->Init());
|
||||
}
|
||||
|
||||
void trimwhitespace(const uint8_t*& str, size_t& len)
|
||||
{
|
||||
while (len > 0 && isspace(*str))
|
||||
{
|
||||
str++;
|
||||
len--;
|
||||
}
|
||||
|
||||
const uint8_t* end = str + len - 1;
|
||||
while (len > 0 && isspace(*end))
|
||||
{
|
||||
end--;
|
||||
len--;
|
||||
}
|
||||
}
|
||||
|
||||
static bool init = false;
|
||||
|
||||
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
|
||||
{
|
||||
if (!init)
|
||||
{
|
||||
init = true;
|
||||
FuzzInit();
|
||||
}
|
||||
|
||||
trimwhitespace(Data, Size);
|
||||
if (Size == 0)
|
||||
return 0;
|
||||
|
||||
gApp->PrepareCompiler();
|
||||
|
||||
bool ready = gApp->QueueFile((char*)Data, Size);
|
||||
|
||||
//if (ready)
|
||||
// ready = gApp->QueuePath("./corlib/src");
|
||||
|
||||
//if (ready)
|
||||
// gApp->Compile();
|
||||
|
||||
gApp->ReleaseCompiler();
|
||||
|
||||
return 0;
|
||||
}
|
180
BeefFuzz/BeefFuzz.h
Normal file
180
BeefFuzz/BeefFuzz.h
Normal file
|
@ -0,0 +1,180 @@
|
|||
#pragma once
|
||||
|
||||
#include "BeefySysLib/Common.h"
|
||||
#include "BeefySysLib/util/Array.h"
|
||||
#include "BeefySysLib/util/String.h"
|
||||
#include <set>
|
||||
//#include <direct.h>
|
||||
|
||||
#pragma warning(disable:4996)
|
||||
|
||||
NS_BF_BEGIN
|
||||
|
||||
#define APPEND2(VAL1, VAL2) VAL1##VAL2
|
||||
#define APPEND(VAL1, VAL2) APPEND2(VAL1, VAL2)
|
||||
#define ENUM_VAL_GENERATE(ENUM_ENTRY) APPEND(ENUM_TYPE, _##ENUM_ENTRY),
|
||||
#define ENUM_NAME_GENERATE(ENUM_ENTRY) #ENUM_ENTRY,
|
||||
#define ENUM_CREATE_DO2(EnumName) \
|
||||
static const char* EnumName##_Names[] = { ENUM_DECLARE(ENUM_NAME_GENERATE) }; \
|
||||
enum EnumName { ENUM_DECLARE(ENUM_VAL_GENERATE) }; \
|
||||
static bool EnumParse(const String& name, EnumName& result) \
|
||||
{ \
|
||||
for (int i = 0; i < sizeof(EnumName##_Names)/sizeof(const char*); i++) \
|
||||
if (name == EnumName##_Names[i]) { result = (EnumName)i; return true; } \
|
||||
return false; \
|
||||
} \
|
||||
static const char* EnumToString(EnumName enumVal) \
|
||||
{ \
|
||||
return EnumName##_Names[(int)enumVal]; \
|
||||
}
|
||||
#define ENUM_CREATE_DO(EnumType) ENUM_CREATE_DO2(EnumType)
|
||||
#define ENUM_CREATE ENUM_CREATE_DO(ENUM_TYPE)
|
||||
|
||||
class IDEUtils
|
||||
{
|
||||
public:
|
||||
static bool FixFilePath(String& filePath)
|
||||
{
|
||||
if (filePath.length() == 0)
|
||||
return false;
|
||||
|
||||
if (filePath[0] == '<')
|
||||
return false;
|
||||
|
||||
if ((filePath.length() > 1) && (filePath[1] == ':'))
|
||||
filePath[0] = tolower(filePath[0]);
|
||||
|
||||
bool prevWasSlash = false;
|
||||
for (int i = 0; i < filePath.length(); i++)
|
||||
{
|
||||
//if ((filePath[i] == '/') && (filePath[i - 1])
|
||||
|
||||
if (filePath[i] == DIR_SEP_CHAR_ALT)
|
||||
filePath[i] = DIR_SEP_CHAR;
|
||||
|
||||
if (filePath[i] == DIR_SEP_CHAR)
|
||||
{
|
||||
if ((prevWasSlash) && (i > 1))
|
||||
{
|
||||
filePath.Remove(i, 1);
|
||||
i--;
|
||||
continue;
|
||||
}
|
||||
|
||||
prevWasSlash = true;
|
||||
}
|
||||
else
|
||||
prevWasSlash = false;
|
||||
|
||||
if ((i >= 4) && (filePath[i - 3] == DIR_SEP_CHAR) && (filePath[i - 2] == '.') && (filePath[i - 1] == '.') && (filePath[i] == DIR_SEP_CHAR))
|
||||
{
|
||||
int prevSlash = (int)filePath.LastIndexOf(DIR_SEP_CHAR, i - 4);
|
||||
if (prevSlash != -1)
|
||||
{
|
||||
filePath.Remove(prevSlash, i - prevSlash);
|
||||
i = prevSlash;
|
||||
}
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static void GetDirWithSlash(String& dirName)
|
||||
{
|
||||
if (dirName.length() == 0)
|
||||
return;
|
||||
char lastC = dirName[dirName.length() - 1];
|
||||
if ((lastC != '\\') && (lastC != '/'))
|
||||
dirName += DIR_SEP_CHAR;
|
||||
}
|
||||
|
||||
static FILE* CreateFileWithDir(const String& fileName, const char* options)
|
||||
{
|
||||
FILE* fp = fopen(fileName.c_str(), options);
|
||||
if (fp == NULL)
|
||||
{
|
||||
String fileDir = GetFileDir(fileName);
|
||||
if (!fileDir.empty())
|
||||
{
|
||||
RecursiveCreateDirectory(fileDir);
|
||||
fp = fopen(fileName.c_str(), "w");
|
||||
}
|
||||
}
|
||||
return fp;
|
||||
}
|
||||
|
||||
static bool WriteAllText(const String& fileName, const String& data)
|
||||
{
|
||||
FILE* fp = CreateFileWithDir(fileName, "w");
|
||||
if (fp == NULL)
|
||||
return false;
|
||||
fwrite(data.c_str(), 1, data.length(), fp);
|
||||
fclose(fp);
|
||||
return true;
|
||||
}
|
||||
|
||||
static void GetFileNameWithoutExtension(const String& filePath, String& outFileName)
|
||||
{
|
||||
outFileName += GetFileName(filePath);
|
||||
int dot = (int)outFileName.LastIndexOf('.');
|
||||
if (dot != -1)
|
||||
outFileName.RemoveToEnd(dot);
|
||||
}
|
||||
|
||||
static void GetExtension(const String& filePath, String& ext)
|
||||
{
|
||||
int idx = (int)filePath.LastIndexOf('.');
|
||||
if (idx != -1)
|
||||
ext = filePath.Substring(idx);
|
||||
}
|
||||
|
||||
static void AppendWithOptionalQuotes(String& targetStr, const String& srcFileName)
|
||||
{
|
||||
if ((int)srcFileName.IndexOf(' ') == -1)
|
||||
targetStr += srcFileName;
|
||||
else
|
||||
targetStr += "\"" + srcFileName + "\"";
|
||||
}
|
||||
};
|
||||
|
||||
class ArgBuilder
|
||||
{
|
||||
public:
|
||||
String* mTarget;
|
||||
bool mDoLongBreak;
|
||||
int mLastBreak;
|
||||
std::multiset<String> mLinkPaths;
|
||||
|
||||
public:
|
||||
ArgBuilder(String& target, bool doLongBreak)
|
||||
{
|
||||
mTarget = ⌖
|
||||
mDoLongBreak = doLongBreak;
|
||||
if (mDoLongBreak)
|
||||
mLastBreak = (int)mTarget->LastIndexOf('\n');
|
||||
else
|
||||
mLastBreak = 0;
|
||||
}
|
||||
|
||||
void AddSep()
|
||||
{
|
||||
if (mDoLongBreak)
|
||||
{
|
||||
if (mTarget->length() - mLastBreak > 0x1F000)
|
||||
{
|
||||
mLastBreak = (int)mTarget->length();
|
||||
mTarget->Append('\n');
|
||||
return;
|
||||
}
|
||||
}
|
||||
mTarget->Append(' ');
|
||||
}
|
||||
|
||||
void AddFileName(const String& filePath)
|
||||
{
|
||||
IDEUtils::AppendWithOptionalQuotes(*mTarget, filePath);
|
||||
}
|
||||
};
|
||||
|
||||
NS_BF_END
|
||||
|
182
BeefFuzz/BeefFuzz.vcxproj
Normal file
182
BeefFuzz/BeefFuzz.vcxproj
Normal file
|
@ -0,0 +1,182 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="14.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
<PropertyGroup Label="Globals">
|
||||
<ProjectGuid>{F5FD6845-BBAE-47B6-AE4E-2C1D30F6E145}</ProjectGuid>
|
||||
<Keyword>Win32Proj</Keyword>
|
||||
<RootNamespace>BeefFuzz</RootNamespace>
|
||||
<WindowsTargetPlatformVersion>10.0.17763.0</WindowsTargetPlatformVersion>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v142</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v142</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v142</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v142</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
|
||||
<ImportGroup Label="ExtensionSettings">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="Shared">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<PropertyGroup Label="UserMacros" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<LinkIncremental>true</LinkIncremental>
|
||||
<OutDir>$(SolutionDir)\ide\dist\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<LinkIncremental>true</LinkIncremental>
|
||||
<TargetName>$(ProjectName)_d</TargetName>
|
||||
<OutDir>$(SolutionDir)\ide\dist\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
<OutDir>$(SolutionDir)\ide\dist\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
<OutDir>$(SolutionDir)\ide\dist\</OutDir>
|
||||
</PropertyGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Console</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>_DEBUG;_CONSOLE;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<AdditionalIncludeDirectories>../;../IDEHelper;../BeefySysLib/platform/win;../BeefySysLib/third_party;..\extern\llvm-project_13_0_1\llvm\include;..\extern\llvm_win64_13_0_1\include;..\extern\llvm-project_13_0_1\llvm\lib\Target;..\extern\llvm_win64_13_0_1\lib\Target\X86;..\extern\llvm-project_13_0_1\llvm\tools\clang\include</AdditionalIncludeDirectories>
|
||||
<RuntimeTypeInfo>true</RuntimeTypeInfo>
|
||||
<SupportJustMyCode>false</SupportJustMyCode>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<BufferSecurityCheck>false</BufferSecurityCheck>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Console</SubSystem>
|
||||
<GenerateDebugInformation>DebugFull</GenerateDebugInformation>
|
||||
<OutputFile>$(SolutionDir)\ide\dist\$(TargetName).exe</OutputFile>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_CONSOLE;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Console</SubSystem>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>NDEBUG;_CONSOLE;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<RuntimeTypeInfo>true</RuntimeTypeInfo>
|
||||
<AdditionalIncludeDirectories>../;../IDEHelper;../BeefySysLib/platform/win;../BeefySysLib/third_party;..\extern\llvm-project_13_0_1\llvm\include;..\extern\llvm_win64_13_0_1\include;..\extern\llvm-project_13_0_1\llvm\lib\Target;..\extern\llvm_win64_13_0_1\lib\Target\X86;..\extern\llvm\tools\clang\include</AdditionalIncludeDirectories>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<BufferSecurityCheck>false</BufferSecurityCheck>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Console</SubSystem>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<GenerateDebugInformation>DebugFull</GenerateDebugInformation>
|
||||
<OutputFile>$(SolutionDir)\ide\dist\$(TargetName).exe</OutputFile>
|
||||
<LinkTimeCodeGeneration>Default</LinkTimeCodeGeneration>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="BeefFuzz.cpp" />
|
||||
<ClCompile Include="FuzzApp.cpp" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\BeefySysLib\BeefySysLib_static.vcxproj">
|
||||
<Project>{eceab68d-2f15-495f-a29c-5ea9548aa23d}</Project>
|
||||
</ProjectReference>
|
||||
<ProjectReference Include="..\IDEHelper\IDEHelper.vcxproj">
|
||||
<Project>{f8d29c38-d37c-4af2-8540-2f6e543264f1}</Project>
|
||||
<Private>false</Private>
|
||||
</ProjectReference>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="BeefFuzz.h" />
|
||||
<ClInclude Include="FuzzApp.h" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
33
BeefFuzz/BeefFuzz.vcxproj.filters
Normal file
33
BeefFuzz/BeefFuzz.vcxproj.filters
Normal file
|
@ -0,0 +1,33 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup>
|
||||
<Filter Include="Source Files">
|
||||
<UniqueIdentifier>{4FC737F1-C7A5-4376-A066-2A32D752A2FF}</UniqueIdentifier>
|
||||
<Extensions>cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx</Extensions>
|
||||
</Filter>
|
||||
<Filter Include="Header Files">
|
||||
<UniqueIdentifier>{93995380-89BD-4b04-88EB-625FBE52EBFB}</UniqueIdentifier>
|
||||
<Extensions>h;hh;hpp;hxx;hm;inl;inc;xsd</Extensions>
|
||||
</Filter>
|
||||
<Filter Include="Resource Files">
|
||||
<UniqueIdentifier>{67DA6AB6-F800-4c08-8B7A-83BB121AAD01}</UniqueIdentifier>
|
||||
<Extensions>rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms</Extensions>
|
||||
</Filter>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="BeefFuzz.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="FuzzApp.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="FuzzApp.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="BeefFuzz.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
</ItemGroup>
|
||||
</Project>
|
169
BeefFuzz/CMakeLists.txt
Normal file
169
BeefFuzz/CMakeLists.txt
Normal file
|
@ -0,0 +1,169 @@
|
|||
cmake_minimum_required(VERSION 3.0.0 FATAL_ERROR)
|
||||
|
||||
################### Variables. ####################
|
||||
# Change if you want modify path or other values. #
|
||||
###################################################
|
||||
|
||||
set(PROJECT_NAME BeefFuzz)
|
||||
# Output Variables
|
||||
set(OUTPUT_DEBUG Debug/bin)
|
||||
set(OUTPUT_RELEASE Release/bin)
|
||||
|
||||
############## CMake Project ################
|
||||
# The main options of project #
|
||||
#############################################
|
||||
|
||||
project(${PROJECT_NAME} CXX C)
|
||||
|
||||
set (CMAKE_CXX_STANDARD 14)
|
||||
add_definitions(-D_GLIBCXX_USE_CXX11_ABI=1)
|
||||
|
||||
#set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
||||
|
||||
# Define Release by default.
|
||||
if(NOT CMAKE_BUILD_TYPE)
|
||||
set(CMAKE_BUILD_TYPE "Debug")
|
||||
message(STATUS "Build type not specified: Use Debug by default.")
|
||||
endif(NOT CMAKE_BUILD_TYPE)
|
||||
|
||||
# Definition of Macros
|
||||
add_definitions(
|
||||
-DIDEHELPER_EXPORTS
|
||||
-DBFSYSLIB_DYNAMIC
|
||||
-DUNICODE
|
||||
-D_UNICODE
|
||||
-DBF_NO_FBX
|
||||
-DFT2_BUILD_LIBRARY
|
||||
-DBFSYSLIB_DYNAMIC
|
||||
)
|
||||
|
||||
if (${APPLE})
|
||||
include_directories(
|
||||
.
|
||||
../
|
||||
../BeefySysLib/
|
||||
../BeefySysLib/third_party
|
||||
../BeefySysLib/third_party/freetype/include
|
||||
../extern/llvm-project_13_0_1/llvm/include
|
||||
../extern/llvm-project_13_0_1/llvm/lib/Target
|
||||
../IDEHelper
|
||||
|
||||
../BeefySysLib/platform/osx
|
||||
)
|
||||
else()
|
||||
include_directories(
|
||||
.
|
||||
../
|
||||
../BeefySysLib/
|
||||
../BeefySysLib/third_party
|
||||
../BeefySysLib/third_party/freetype/include
|
||||
../extern/llvm-project_13_0_1/llvm/include
|
||||
../extern/llvm-project_13_0_1/llvm/lib/Target
|
||||
../IDEHelper
|
||||
|
||||
../BeefySysLib/platform/linux
|
||||
)
|
||||
endif()
|
||||
|
||||
############## Artefacts Output #################
|
||||
# Defines outputs , depending Debug or Release. #
|
||||
#################################################
|
||||
|
||||
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
||||
add_definitions(
|
||||
-D_DEBUG
|
||||
)
|
||||
include_directories(
|
||||
../extern/llvm_linux_13_0_1/include
|
||||
../extern/llvm_linux_13_0_1/lib/Target/X86
|
||||
)
|
||||
set(EXECUTABLE_OUTPUT_PATH "${CMAKE_BINARY_DIR}/${OUTPUT_DEBUG}")
|
||||
set(LLVM_LIB "${CMAKE_CURRENT_SOURCE_DIR}/../extern/llvm_linux_13_0_1/lib")
|
||||
else()
|
||||
include_directories(
|
||||
../extern/llvm_linux_rel_13_0_1/include
|
||||
../extern/llvm_linux_rel_13_0_1/lib/Target/X86
|
||||
)
|
||||
set(EXECUTABLE_OUTPUT_PATH "${CMAKE_BINARY_DIR}/${OUTPUT_RELEASE}")
|
||||
set(LLVM_LIB "${CMAKE_CURRENT_SOURCE_DIR}/../extern/llvm_linux_rel_13_0_1/lib")
|
||||
endif()
|
||||
|
||||
################### Dependencies ##################
|
||||
# Add Dependencies to project. #
|
||||
###################################################
|
||||
|
||||
option(BUILD_DEPENDS
|
||||
"Build other CMake project."
|
||||
ON
|
||||
)
|
||||
|
||||
# Dependencies : disable BUILD_DEPENDS to link with lib already build.
|
||||
if(BUILD_DEPENDS)
|
||||
|
||||
else()
|
||||
|
||||
endif()
|
||||
|
||||
################# Flags ################
|
||||
# Defines Flags for Windows and Linux. #
|
||||
########################################
|
||||
|
||||
if(MSVC)
|
||||
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /W3 /MD /MDd /Od /EHsc")
|
||||
set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /W3 /GL /Od /Oi /Gy /EHsc")
|
||||
else()
|
||||
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17 -Wno-multichar -Wno-invalid-offsetof -fno-omit-frame-pointer -fsanitize=fuzzer")
|
||||
if (BF_ENABLE_ASAN)
|
||||
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fsanitize=address")
|
||||
endif()
|
||||
endif()
|
||||
|
||||
################ Files ################
|
||||
# -- Add files to project. -- #
|
||||
#######################################
|
||||
|
||||
file(GLOB SRC_FILES
|
||||
BeefFuzz.cpp
|
||||
FuzzApp.cpp
|
||||
)
|
||||
|
||||
# Add executable to build.
|
||||
add_executable(${PROJECT_NAME}
|
||||
${SRC_FILES}
|
||||
)
|
||||
|
||||
execute_process(
|
||||
COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/../extern/llvm_linux_13_0_1/bin/llvm-config --system-libs --link-static
|
||||
OUTPUT_VARIABLE LLVM_SYSTEM_LIBS
|
||||
OUTPUT_STRIP_TRAILING_WHITESPACE
|
||||
RESULT_VARIABLE EXEC_RESULT
|
||||
)
|
||||
|
||||
if (EXEC_RESULT AND NOT EXEC_RESULT EQUAL 0)
|
||||
if (EXEC_RESULT MATCHES "^[0-9]+$")
|
||||
message(FATAL_ERROR "llvm-config exited with code ${EXEC_RESULT}.")
|
||||
else()
|
||||
message(FATAL_ERROR "llvm-config couldn't be executed: ${EXEC_RESULT}")
|
||||
endif()
|
||||
endif()
|
||||
|
||||
if (${APPLE})
|
||||
set(TARGET_LIBS_OS "")
|
||||
else()
|
||||
#set(TARGET_LIBS_OS "curses")
|
||||
#set(TARGET_LIBS_OS "-Xlinker --no-demangle -v")
|
||||
|
||||
set(TARGET_LIBS_OS "${LLVM_SYSTEM_LIBS}")
|
||||
endif()
|
||||
|
||||
# Link with other dependencies.
|
||||
if(MSVC)
|
||||
target_link_libraries(${PROJECT_NAME} BeefySysLib IDEHelper kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib)
|
||||
else()
|
||||
target_link_libraries(${PROJECT_NAME} BeefySysLib
|
||||
IDEHelper
|
||||
${TARGET_LIBS_OS}
|
||||
|
||||
#${LLVM_LIB}/libLLVMSupport.a
|
||||
)
|
||||
endif()
|
300
BeefFuzz/FuzzApp.cpp
Normal file
300
BeefFuzz/FuzzApp.cpp
Normal file
|
@ -0,0 +1,300 @@
|
|||
#pragma warning(disable:4996)
|
||||
|
||||
#include "FuzzApp.h"
|
||||
#include <iostream>
|
||||
#include "BeefySysLib/util/String.h"
|
||||
#include "BeefySysLib/util/FileEnumerator.h"
|
||||
#include "BeefySysLib/util/WorkThread.h"
|
||||
#include "BeefySysLib/platform/PlatformHelper.h"
|
||||
#include "Compiler/BfSystem.h"
|
||||
|
||||
#ifdef BF_PLATFORM_WINDOWS
|
||||
#include <direct.h>
|
||||
#endif
|
||||
|
||||
BF_IMPORT void BF_CALLTYPE Targets_Create();
|
||||
BF_IMPORT void BF_CALLTYPE Targets_Delete();
|
||||
|
||||
BF_IMPORT void BF_CALLTYPE BfSystem_ReportMemory(void* bfSystem);
|
||||
BF_EXPORT void BF_CALLTYPE BfCompiler_ProgramDone();
|
||||
|
||||
BF_IMPORT void BF_CALLTYPE Debugger_FullReportMemory();
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
BF_IMPORT void BF_CALLTYPE BfCompiler_Delete(void* bfCompiler);
|
||||
BF_EXPORT void BF_CALLTYPE BfCompiler_SetOptions(void* bfCompiler, void* hotProject, int hotIdx,
|
||||
const char* targetTriple, const char* targetCPU, int toolsetType, int simdSetting, int allocStackCount, int maxWorkerThreads,
|
||||
Beefy::BfCompilerOptionFlags optionFlags, const char* mallocLinkName, const char* freeLinkName);
|
||||
BF_IMPORT void BF_CALLTYPE BfCompiler_ClearBuildCache(void* bfCompiler);
|
||||
BF_IMPORT bool BF_CALLTYPE BfCompiler_Compile(void* bfCompiler, void* bfPassInstance, const char* outputPath);
|
||||
BF_IMPORT float BF_CALLTYPE BfCompiler_GetCompletionPercentage(void* bfCompiler);
|
||||
BF_IMPORT const char* BF_CALLTYPE BfCompiler_GetUsedOutputFileNames(void* bfCompiler, void* bfProject, bool flushQueuedHotFiles, bool* hadOutputChanges);
|
||||
|
||||
BF_IMPORT void* BF_CALLTYPE BfSystem_CreateParser(void* bfSystem, void* bfProject);
|
||||
BF_IMPORT void BF_CALLTYPE BfParser_SetSource(void* bfParser, const char* data, int length, const char* fileName);
|
||||
BF_IMPORT void BF_CALLTYPE BfParser_SetCharIdData(void* bfParser, uint8* data, int length);
|
||||
BF_IMPORT bool BF_CALLTYPE BfParser_Parse(void* bfParser, void* bfPassInstance, bool compatMode);
|
||||
BF_IMPORT bool BF_CALLTYPE BfParser_Reduce(void* bfParser, void* bfPassInstance);
|
||||
BF_IMPORT bool BF_CALLTYPE BfParser_BuildDefs(void* bfParser, void* bfPassInstance, void* resolvePassData, bool fullRefresh);
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
BF_IMPORT void* BF_CALLTYPE BfSystem_Create();
|
||||
BF_EXPORT void BF_CALLTYPE BfSystem_Lock(void* bfSystem, int priority);
|
||||
BF_EXPORT void BF_CALLTYPE BfSystem_Unlock(void* bfSystem);
|
||||
BF_IMPORT void BF_CALLTYPE BfSystem_ReportMemory(void* bfSystem);
|
||||
BF_IMPORT void BF_CALLTYPE BfSystem_Delete(void* bfSystem);
|
||||
BF_IMPORT void* BF_CALLTYPE BfSystem_CreatePassInstance(void* bfSystem);
|
||||
BF_IMPORT void* BF_CALLTYPE BfSystem_CreateCompiler(void* bfSystem, bool isResolveOnly);
|
||||
BF_IMPORT void* BF_CALLTYPE BfSystem_CreateProject(void* bfSystem, const char* projectName, const char* projectDir);
|
||||
BF_IMPORT void BF_CALLTYPE BfParser_Delete(void* bfParser);
|
||||
BF_IMPORT void BF_CALLTYPE BfSystem_AddTypeOptions(void* bfSystem, const char* filter, int32 simdSetting, int32 optimizationLevel, int32 emitDebugInfo, int32 arrayBoundsCheck,
|
||||
int32 initLocalVariables, int32 emitDynamicCastCheck, int32 emitObjectAccessCheck, int32 allocStackTraceDepth);
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
BF_IMPORT void BF_CALLTYPE BfProject_SetDisabled(void* bfProject, bool disabled);
|
||||
BF_IMPORT void BF_CALLTYPE BfProject_SetOptions(void* bfProject, int targetType, const char* startupObject, const char* preprocessorMacros,
|
||||
int optLevel, int ltoType, int relocType, int picLevel, int32 flags);
|
||||
BF_IMPORT void BF_CALLTYPE BfProject_ClearDependencies(void* bfProject);
|
||||
BF_IMPORT void BF_CALLTYPE BfProject_AddDependency(void* bfProject, void* depProject);
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
BF_IMPORT const char* BF_CALLTYPE BfPassInstance_PopOutString(void* bfPassInstance);
|
||||
BF_IMPORT void BF_CALLTYPE BfPassInstance_Delete(void* bfPassInstance);
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
BF_IMPORT const char* BF_CALLTYPE VSSupport_Find();
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
USING_NS_BF;
|
||||
|
||||
FuzzApp* Beefy::gApp = NULL;
|
||||
|
||||
|
||||
FuzzApp::FuzzApp()
|
||||
{
|
||||
Targets_Create();
|
||||
|
||||
mTargetType = BfTargetType_BeefConsoleApplication;
|
||||
|
||||
mSystem = NULL;
|
||||
mCompiler = NULL;
|
||||
mProject = NULL;
|
||||
mCELibProject = NULL;
|
||||
mIsCERun = false;
|
||||
mStartupObject = "Program";
|
||||
|
||||
#ifdef BF_PLATFORM_WINDOWS
|
||||
mOptLevel = BfOptLevel_OgPlus;
|
||||
mToolset = BfToolsetType_Microsoft;
|
||||
#else
|
||||
mOptLevel = BfOptLevel_O0;
|
||||
mToolset = BfToolsetType_GNU;
|
||||
#endif
|
||||
|
||||
#ifdef BF_PLATFORM_WINDOWS
|
||||
mTargetTriple = "x86_64-pc-windows-msvc";
|
||||
#elif defined BF_PLATFORM_MACOS
|
||||
mTargetTriple = "x86_64-apple-macosx10.8.0";
|
||||
#else
|
||||
mTargetTriple = "x86_64-unknown-linux-gnu";
|
||||
#endif
|
||||
}
|
||||
|
||||
FuzzApp::~FuzzApp()
|
||||
{
|
||||
Targets_Delete();
|
||||
}
|
||||
|
||||
bool FuzzApp::Init()
|
||||
{
|
||||
char* cwdPtr = getcwd(NULL, 0);
|
||||
mWorkingDir = cwdPtr;
|
||||
free(cwdPtr);
|
||||
|
||||
if (mTargetPath.IsEmpty())
|
||||
return false;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool FuzzApp::QueueFile(const char* data, size_t len)
|
||||
{
|
||||
bool worked = true;
|
||||
void* bfParser = BfSystem_CreateParser(mSystem, (mCELibProject != NULL) ? mCELibProject : mProject);
|
||||
BfParser_SetSource(bfParser, data, len, "Fuzz.bf");
|
||||
//bfParser.SetCharIdData(charIdData);
|
||||
worked &= BfParser_Parse(bfParser, mPassInstance, false);
|
||||
worked &= BfParser_Reduce(bfParser, mPassInstance);
|
||||
worked &= BfParser_BuildDefs(bfParser, mPassInstance, NULL, false);
|
||||
return worked;
|
||||
}
|
||||
|
||||
bool FuzzApp::QueuePath(const StringImpl& path)
|
||||
{
|
||||
if (DirectoryExists(path))
|
||||
{
|
||||
for (auto& fileEntry : FileEnumerator(path, FileEnumerator::Flags_Files))
|
||||
{
|
||||
String filePath = fileEntry.GetFilePath();
|
||||
|
||||
String fileName;
|
||||
fileName = GetFileName(filePath);
|
||||
|
||||
String ext;
|
||||
ext = GetFileExtension(filePath);
|
||||
|
||||
if ((ext.Equals(".bf", StringImpl::CompareKind_OrdinalIgnoreCase)) ||
|
||||
(ext.Equals(".cs", StringImpl::CompareKind_OrdinalIgnoreCase)))
|
||||
{
|
||||
int len;
|
||||
const char* data = LoadTextData(filePath, &len);
|
||||
if (data != NULL)
|
||||
{
|
||||
bool success = QueueFile(data, len);
|
||||
delete[] data;
|
||||
|
||||
if (!success)
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (auto& fileEntry : FileEnumerator(path, FileEnumerator::Flags_Directories))
|
||||
{
|
||||
String childPath = fileEntry.GetFilePath();
|
||||
String dirName;
|
||||
dirName = GetFileName(childPath);
|
||||
|
||||
if (dirName == "build")
|
||||
continue;
|
||||
|
||||
if (!QueuePath(childPath))
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool FuzzApp::CopyFile(const StringImpl& srcPath, const StringImpl& destPath)
|
||||
{
|
||||
BfpFileResult result = BfpFileResult_Ok;
|
||||
for (int i = 0; i < 20; i++)
|
||||
{
|
||||
BfpFile_Copy(srcPath.c_str(), destPath.c_str(), BfpFileCopyKind_Always, &result);
|
||||
if (result == BfpFileResult_Ok)
|
||||
return true;
|
||||
BfpThread_Sleep(100);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void FuzzApp::PrepareCompiler()
|
||||
{
|
||||
mSystem = BfSystem_Create();
|
||||
|
||||
mCompiler = BfSystem_CreateCompiler(mSystem, false);
|
||||
|
||||
String projectName = GetFileName(mTargetPath);
|
||||
int dotPos = (int)projectName.IndexOf('.');
|
||||
if (dotPos != -1)
|
||||
projectName.RemoveToEnd(dotPos);
|
||||
if (projectName.IsEmpty())
|
||||
projectName.Append("BeefProject");
|
||||
|
||||
mProject = BfSystem_CreateProject(mSystem, projectName.c_str(), GetFileDir(mTargetPath).c_str());
|
||||
|
||||
if (mIsCERun)
|
||||
{
|
||||
mCELibProject = BfSystem_CreateProject(mSystem, "BeefLib", GetFileDir(mTargetPath).c_str());
|
||||
BfProject_SetOptions(mCELibProject, BfTargetType_BeefLib, "", mDefines.c_str(), mOptLevel, 0, 0, 0, BfProjectFlags_None);
|
||||
}
|
||||
|
||||
String defines = mDefines;
|
||||
if (!defines.IsEmpty())
|
||||
defines.Append("\n");
|
||||
defines.Append("BF_64_BIT");
|
||||
defines.Append("\nBF_LITTLE_ENDIAN");
|
||||
defines.Append("\n");
|
||||
defines.Append(BF_PLATFORM_NAME);
|
||||
|
||||
int ltoType = 0;
|
||||
BfProject_SetOptions(mProject, mTargetType, mStartupObject.c_str(), defines.c_str(), BfOptLevel_O0, ltoType, 0, 0, BfProjectFlags_None);
|
||||
|
||||
if (mCELibProject != NULL)
|
||||
BfProject_AddDependency(mProject, mCELibProject);
|
||||
|
||||
mPassInstance = BfSystem_CreatePassInstance(mSystem);
|
||||
|
||||
Beefy::String exePath;
|
||||
BfpGetStrHelper(exePath, [](char* outStr, int* inOutStrSize, BfpResult* result)
|
||||
{
|
||||
BfpSystem_GetExecutablePath(outStr, inOutStrSize, (BfpSystemResult*)result);
|
||||
});
|
||||
mBuildDir = GetFileDir(exePath) + "/build";
|
||||
|
||||
RecursiveCreateDirectory(mBuildDir + "/" + projectName);
|
||||
if (mIsCERun)
|
||||
RecursiveCreateDirectory(mBuildDir + "/BeefLib");
|
||||
|
||||
BfCompilerOptionFlags optionFlags = (BfCompilerOptionFlags)(BfCompilerOptionFlag_EmitDebugInfo | BfCompilerOptionFlag_EmitLineInfo | BfCompilerOptionFlag_GenerateOBJ | BfCompilerOptionFlag_OmitDebugHelpers);
|
||||
|
||||
//int maxWorkerThreads = BfpSystem_GetNumLogicalCPUs(NULL);
|
||||
//if (maxWorkerThreads <= 1)
|
||||
// maxWorkerThreads = 6;
|
||||
|
||||
BfCompiler_SetOptions(mCompiler, NULL, 0, mTargetTriple.c_str(), "", mToolset, BfSIMDSetting_SSE2, 1, 1, optionFlags, "malloc", "free");
|
||||
}
|
||||
|
||||
|
||||
bool FuzzApp::Compile()
|
||||
{
|
||||
BfCompiler_ClearBuildCache(mCompiler);
|
||||
|
||||
if (!BfCompiler_Compile(mCompiler, mPassInstance, mBuildDir.c_str()))
|
||||
return false;
|
||||
|
||||
if (!mCEDest.IsEmpty())
|
||||
{
|
||||
String ext;
|
||||
String srcResult = mBuildDir + "/BeefProject/BeefProject";
|
||||
srcResult += BF_OBJ_EXT;
|
||||
|
||||
if (!CopyFile(srcResult, mCEDest))
|
||||
return false;
|
||||
}
|
||||
|
||||
while (true)
|
||||
{
|
||||
const char* msg = BfPassInstance_PopOutString(mPassInstance);
|
||||
if (msg == NULL)
|
||||
break;
|
||||
|
||||
if ((strncmp(msg, ":error ", 7) == 0) ||
|
||||
(strncmp(msg, "ERROR(", 6) == 0) ||
|
||||
(strncmp(msg, "ERROR:", 6) == 0))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void FuzzApp::ReleaseCompiler()
|
||||
{
|
||||
BfPassInstance_Delete(mPassInstance);
|
||||
BfCompiler_Delete(mCompiler);
|
||||
|
||||
BfSystem_Delete(mSystem);
|
||||
}
|
||||
|
||||
|
66
BeefFuzz/FuzzApp.h
Normal file
66
BeefFuzz/FuzzApp.h
Normal file
|
@ -0,0 +1,66 @@
|
|||
#pragma once
|
||||
|
||||
#include "BeefFuzz.h"
|
||||
#include "BeefySysLib/FileStream.h"
|
||||
#include "BeefySysLib/util/CritSect.h"
|
||||
#include "BeefySysLib/util/String.h"
|
||||
#include "BeefySysLib/util/Array.h"
|
||||
#include "Compiler/BfSystem.h"
|
||||
|
||||
NS_BF_BEGIN
|
||||
|
||||
class FuzzApp
|
||||
{
|
||||
public:
|
||||
BfTargetType mTargetType;
|
||||
String mTargetTriple;
|
||||
BfOptLevel mOptLevel;
|
||||
BfToolsetType mToolset;
|
||||
String mBuildDir;
|
||||
String mWorkingDir;
|
||||
String mDefines;
|
||||
String mStartupObject;
|
||||
String mTargetPath;
|
||||
String mLinkParams;
|
||||
|
||||
void* mSystem;
|
||||
void* mCompiler;
|
||||
void* mProject;
|
||||
void* mPassInstance;
|
||||
|
||||
bool mIsCERun;
|
||||
void* mCELibProject;
|
||||
String mCEDest;
|
||||
|
||||
public:
|
||||
bool CopyFile(const StringImpl& srcPath, const StringImpl& destPath);
|
||||
|
||||
bool QueueFile(const char* data, size_t len);
|
||||
bool QueuePath(const StringImpl& path);
|
||||
|
||||
public:
|
||||
FuzzApp();
|
||||
~FuzzApp();
|
||||
|
||||
void SetTargetType(BfTargetType value) { mTargetType = value; }
|
||||
void SetTargetTriple(const String& value) { mTargetTriple = value; }
|
||||
void SetOptLevel(BfOptLevel value) { mOptLevel = value; }
|
||||
void SetToolset(BfToolsetType value) { mToolset = value; }
|
||||
void SetBuildDir(const String& value) { mBuildDir = value; }
|
||||
void SetWorkingDir(const String& value) { mWorkingDir = value; }
|
||||
void AddDefine(const String& value) { mDefines += mDefines.IsEmpty() ? value : "\n" + value; }
|
||||
void SetStartupObject(const String& value) { mStartupObject = value; }
|
||||
void SetTargetPath(const String& value) { mTargetPath = value; }
|
||||
void SetLinkParams(const String& value) { mLinkParams = value; }
|
||||
void SetCEDest(const String& value) { mCEDest = value; }
|
||||
|
||||
bool Init();
|
||||
|
||||
void PrepareCompiler();
|
||||
bool Compile();
|
||||
void ReleaseCompiler();
|
||||
};
|
||||
|
||||
extern FuzzApp* gApp;
|
||||
|
||||
NS_BF_END
|
Loading…
Add table
Add a link
Reference in a new issue