torcontrol.cpp raw

   1  // Copyright (c) 2020-2021 The Limenka developers
   2  // Distributed under the MIT software license, see the accompanying
   3  // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   4  
   5  #include <test/fuzz/FuzzedDataProvider.h>
   6  #include <test/fuzz/fuzz.h>
   7  #include <test/fuzz/util.h>
   8  #include <test/util/setup_common.h>
   9  #include <torcontrol.h>
  10  
  11  #include <cstdint>
  12  #include <string>
  13  #include <vector>
  14  
  15  class DummyTorControlConnection : public TorControlConnection
  16  {
  17  public:
  18      DummyTorControlConnection() : TorControlConnection{nullptr}
  19      {
  20      }
  21  
  22      bool Connect(const std::string&, const ConnectionCB&, const ConnectionCB&)
  23      {
  24          return true;
  25      }
  26  
  27      void Disconnect()
  28      {
  29      }
  30  
  31      bool Command(const std::string&, const ReplyHandlerCB&)
  32      {
  33          return true;
  34      }
  35  };
  36  
  37  void initialize_torcontrol()
  38  {
  39      static const auto testing_setup = MakeNoLogFileContext<>();
  40  }
  41  
  42  FUZZ_TARGET(torcontrol, .init = initialize_torcontrol)
  43  {
  44      FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
  45  
  46      TorController tor_controller;
  47      LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
  48          TorControlReply tor_control_reply;
  49          CallOneOf(
  50              fuzzed_data_provider,
  51              [&] {
  52                  tor_control_reply.code = 250;
  53              },
  54              [&] {
  55                  tor_control_reply.code = 510;
  56              },
  57              [&] {
  58                  tor_control_reply.code = TOR_REPLY_SYNTAX_ERROR;
  59              },
  60              [&] {
  61                  tor_control_reply.code = fuzzed_data_provider.ConsumeIntegral<int>();
  62              });
  63          tor_control_reply.lines = ConsumeRandomLengthStringVector(fuzzed_data_provider);
  64          if (tor_control_reply.lines.empty()) {
  65              break;
  66          }
  67          DummyTorControlConnection dummy_tor_control_connection;
  68          CallOneOf(
  69              fuzzed_data_provider,
  70              [&] {
  71                  tor_controller.add_onion_cb(dummy_tor_control_connection, tor_control_reply, /*pow_was_enabled=*/true);
  72              },
  73              [&] {
  74                  tor_controller.add_onion_cb(dummy_tor_control_connection, tor_control_reply, /*pow_was_enabled=*/false);
  75              },
  76              [&] {
  77                  tor_controller.auth_cb(dummy_tor_control_connection, tor_control_reply);
  78              },
  79              [&] {
  80                  tor_controller.authchallenge_cb(dummy_tor_control_connection, tor_control_reply);
  81              },
  82              [&] {
  83                  tor_controller.protocolinfo_cb(dummy_tor_control_connection, tor_control_reply);
  84              });
  85      }
  86  }
  87