Add general fuzzer. (#159)

cmake/ftxui_fuzzer.cmake

@@ -1,17 +1,23 @@
 set(CMAKE_C_COMPILER clang)
 set(CMAKE_CXX_COMPILER clang++)
-add_executable(terminal_input_parser_test_fuzz
-  src/ftxui/component/terminal_input_parser_test_fuzz.cpp
-)
-target_include_directories(terminal_input_parser_test_fuzz
-  PRIVATE src
-)
-target_link_libraries(terminal_input_parser_test_fuzz
-  PRIVATE component
-)
-target_compile_options(terminal_input_parser_test_fuzz
-  PRIVATE -fsanitize=fuzzer,address
-)
-target_link_libraries(terminal_input_parser_test_fuzz
-  PRIVATE -fsanitize=fuzzer,address
-)
+
+function(fuzz name)
+  add_executable(${name}
+    src/ftxui/component/${name}.cpp
+  )
+  target_include_directories(${name}
+    PRIVATE src
+  )
+  target_link_libraries(${name}
+    PRIVATE component
+  )
+  target_compile_options(${name}
+    PRIVATE -fsanitize=fuzzer,address
+  )
+  target_link_libraries(${name}
+    PRIVATE -fsanitize=fuzzer,address
+  )
+endfunction(fuzz)
+
+fuzz(terminal_input_parser_test_fuzzer)
+fuzz(component_fuzzer)

include/ftxui/util/ref.hpp

@@ -46,6 +46,7 @@ class StringRef {
   StringRef(std::wstring ref) : owned_(std::move(ref)) {}
   StringRef(const wchar_t* ref) : StringRef(std::wstring(ref)) {}
   StringRef(const char* ref) : StringRef(to_wstring(std::string(ref))) {}
+  StringRef(std::string ref) : StringRef(to_wstring(std::move(ref))) {}
   std::wstring& operator*() { return address_ ? *address_ : owned_; }
   std::wstring* operator->() { return address_ ? address_ : &owned_; }
 
@@ -63,6 +64,8 @@ class ConstStringRef {
   ConstStringRef(const wchar_t* ref) : ConstStringRef(std::wstring(ref)) {}
   ConstStringRef(const char* ref)
       : ConstStringRef(to_wstring(std::string(ref))) {}
+  ConstStringRef(std::string ref)
+      : ConstStringRef(to_wstring(std::move(ref))) {}
   const std::wstring& operator*() { return address_ ? *address_ : owned_; }
   const std::wstring* operator->() { return address_ ? address_ : &owned_; }
 

src/ftxui/component/component_fuzzer.cpp

@@ -0,0 +1,147 @@
+//#include "ftxui/component/event.hpp"
+//#include "ftxui/component/receiver.hpp"
+#include <vector>
+#include "ftxui/component/component.hpp"
+#include "ftxui/component/terminal_input_parser.hpp"
+
+using namespace ftxui;
+namespace {
+
+bool GeneratorBool(const char*& data, size_t& size) {
+  if (size == 0)
+    return false;
+
+  auto out = bool(data[0] % 2);
+  data++;
+  size--;
+  return out;
+}
+
+std::wstring GeneratorString(const char*& data, size_t& size) {
+  int index = 0;
+  while (index < size && data[index])
+    ++index;
+
+  auto out = std::wstring(data, data + index);
+  data += index;
+  size -= index;
+  return std::move(out);
+}
+
+int GeneratorInt(const char* data, size_t size) {
+  if (size == 0)
+    return 0;
+  auto out = int(data[0]);
+  data++;
+  size--;
+  return out;
+}
+
+bool g_bool;
+int g_int;
+std::vector<std::wstring> g_list;
+
+Components GeneratorComponents(const char*& data, size_t& size, int depth);
+
+Component GeneratorComponent(const char*& data, size_t& size, int depth) {
+  depth--;
+  int value = GeneratorInt(data, size);
+  if (depth <= 0)
+    return Button(GeneratorString(data, size), [] {});
+
+  switch (value % 16) {
+    case 1:
+      return Checkbox(GeneratorString(data, size), &g_bool);
+    case 2:
+      return Input(GeneratorString(data, size), GeneratorString(data, size));
+    case 3:
+      return Menu(&g_list, &g_int);
+    case 4:
+      return Radiobox(&g_list, &g_int);
+    case 5:
+      return Toggle(&g_list, &g_int);
+    case 6:
+      return Slider(GeneratorString(data, size), &g_int,
+                    GeneratorInt(data, size), GeneratorInt(data, size),
+                    GeneratorInt(data, size));
+    case 7:
+      return ResizableSplitLeft(GeneratorComponent(data, size, depth - 1),
+                                GeneratorComponent(data, size, depth - 1),
+                                &g_int);
+    case 8:
+      return ResizableSplitRight(GeneratorComponent(data, size, depth - 1),
+                                 GeneratorComponent(data, size, depth - 1),
+                                 &g_int);
+    case 9:
+      return ResizableSplitTop(GeneratorComponent(data, size, depth - 1),
+                               GeneratorComponent(data, size, depth - 1),
+                               &g_int);
+    case 10:
+      return ResizableSplitBottom(GeneratorComponent(data, size, depth - 1),
+                                  GeneratorComponent(data, size, depth - 1),
+                                  &g_int);
+    case 11:
+      return Container::Vertical(GeneratorComponents(data, size, depth - 1));
+
+    case 12:
+      return Container::Vertical(GeneratorComponents(data, size, depth - 1),
+                                 &g_int);
+
+    case 13:
+      return Container::Horizontal(GeneratorComponents(data, size, depth - 1));
+    case 14:
+      return Container::Horizontal(GeneratorComponents(data, size, depth - 1),
+                                   &g_int);
+    case 15:
+      return Container::Tab(GeneratorComponents(data, size, depth - 1), &g_int);
+    default:
+      return Button(GeneratorString(data, size), [] {});
+  }
+}
+
+Components GeneratorComponents(const char*& data, size_t& size, int depth) {
+  Components out;
+  if (depth > 0) {
+    while (size && GeneratorInt(data, size) % 2) {
+      out.push_back(GeneratorComponent(data, size, depth - 1));
+    }
+  }
+  return std::move(out);
+}
+
+}  // namespace
+extern "C" int LLVMFuzzerTestOneInput(const char* data, size_t size) {
+  g_bool = GeneratorBool(data, size);
+  g_int = GeneratorInt(data, size);
+  g_list = {
+      L"test_1", L"test_2", L"test_3", L"test_4", L"test_5",
+  };
+
+  int depth = 10;
+  auto component = GeneratorComponent(data, size, depth);
+
+  int width = GeneratorInt(data, size);
+  int height = GeneratorInt(data, size);
+
+  auto screen =
+      Screen::Create(Dimension::Fixed(width), Dimension::Fixed(height));
+
+  auto event_receiver = MakeReceiver<Event>();
+  {
+    auto parser = TerminalInputParser(event_receiver->MakeSender());
+    for (size_t i = 0; i < size; ++i)
+      parser.Add(data[i]);
+  }
+
+  Event event;
+  while (event_receiver->Receive(&event)) {
+    component->OnEvent(event);
+    auto document = component->Render();
+    Render(screen, document);
+  }
+  return 0;  // Non-zero return values are reserved for future use.
+}
+
+// Copyright 2021 Arthur Sonzogni. All rights reserved.
+// Use of this source code is governed by the MIT license that can be found in
+// the LICENSE file.