diff --git a/lldb/tools/lldb-fuzzer/CMakeLists.txt b/lldb/tools/lldb-fuzzer/CMakeLists.txt --- a/lldb/tools/lldb-fuzzer/CMakeLists.txt +++ b/lldb/tools/lldb-fuzzer/CMakeLists.txt @@ -1,2 +1,3 @@ add_subdirectory(lldb-target-fuzzer) +add_subdirectory(lldb-commandinterpreter-fuzzer) add_subdirectory(utils) diff --git a/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/CMakeLists.txt b/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/CMakeLists.txt new file mode 100644 --- /dev/null +++ b/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/CMakeLists.txt @@ -0,0 +1,24 @@ +set(LLVM_LINK_COMPONENTS + Support + ObjectYAML + ) + +add_llvm_fuzzer(lldb-commandinterpreter-fuzzer + EXCLUDE_FROM_ALL + lldb-commandinterpreter-fuzzer.cpp + ) + +if(TARGET lldb-commandinterpreter-fuzzer) + target_include_directories(lldb-commandinterpreter-fuzzer PRIVATE ..) + target_link_libraries(lldb-commandinterpreter-fuzzer + PRIVATE + liblldb + lldbFuzzerUtils + ) + + add_custom_target(fuzz-lldb-commandinterpreter + COMMENT "Running the LLDB command interpreter fuzzer..." + COMMAND cd ${CMAKE_CURRENT_SOURCE_DIR} && $ -dict=inputdictionary.txt -only_ascii=1 + USES_TERMINAL + ) +endif() diff --git a/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/inputdictionary.txt b/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/inputdictionary.txt new file mode 100644 --- /dev/null +++ b/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/inputdictionary.txt @@ -0,0 +1,4 @@ +kw1="breakpoint set" +kw2="target" +kw3="run" +kw4="frame info" diff --git a/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/lldb-commandinterpreter-fuzzer.cpp b/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/lldb-commandinterpreter-fuzzer.cpp new file mode 100644 --- /dev/null +++ b/lldb/tools/lldb-fuzzer/lldb-commandinterpreter-fuzzer/lldb-commandinterpreter-fuzzer.cpp @@ -0,0 +1,53 @@ +//===-- lldb-commandinterpreter-fuzzer.cpp - Fuzz LLDB's command interpreter +//---------------------===// +// +// 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 + +#include "lldb/API/SBCommandInterpreter.h" +#include "lldb/API/SBCommandInterpreterRunOptions.h" +#include "lldb/API/SBCommandReturnObject.h" +#include "lldb/API/SBDebugger.h" +#include "lldb/API/SBTarget.h" +#include "utils/TempFile.h" + +using namespace lldb; +using namespace lldb_fuzzer; +using namespace llvm; + +extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) { + SBDebugger::Initialize(); + return 0; +} + +extern "C" int LLVMFuzzerTestOneInput(uint8_t *data, size_t size) { + // Convert the data into a null-terminated string + std::string str((char *)data, size); + + // Create a debugger and a dummy target + SBDebugger debugger = SBDebugger::Create(false); + SBTarget target = debugger.GetDummyTarget(); + + // Create a command interpreter for the current debugger + // A return object is needed to run the command interpreter + SBCommandReturnObject ro = SBCommandReturnObject(); + SBCommandInterpreter thisinterpreter = debugger.GetCommandInterpreter(); + + // Create a breakpoint in the target program and then use the fuzzer + // generated input as input for the command interpreter + if (thisinterpreter.IsValid()) { + thisinterpreter.HandleCommand("breakpoint set --name main", ro, false); + thisinterpreter.HandleCommand(str.c_str(), ro, false); + } + + debugger.DeleteTarget(target); + SBDebugger::Destroy(debugger); + SBModule::GarbageCollectAllocatedModules(); + + return 0; +}