centipede_test.cc

// Copyright 2022 The Centipede Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <algorithm>
#include <cstddef>
#include <cstdint>
#include <cstdlib>
#include <cstring>
#include <filesystem>  // NOLINT
#include <iterator>
#include <limits>
#include <set>
#include <string>
#include <string_view>
#include <thread>  // NOLINT: For thread::get_id() only.
#include <utility>
#include <vector>

#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "absl/base/nullability.h"
#include "absl/container/flat_hash_set.h"
#include "absl/strings/str_cat.h"
#include "absl/time/time.h"
#include "./centipede/centipede_callbacks.h"
#include "./centipede/centipede_default_callbacks.h"
#include "./centipede/centipede_interface.h"
#include "./centipede/environment.h"
#include "./centipede/feature.h"
#include "./centipede/mutation_input.h"
#include "./centipede/runner_result.h"
#include "./centipede/stop.h"
#include "./centipede/util.h"
#include "./centipede/workdir.h"
#include "./common/defs.h"
#include "./common/hash.h"
#include "./common/logging.h"
#include "./common/test_util.h"

namespace fuzztest::internal {
namespace {

using ::testing::AllOf;
using ::testing::Contains;
using ::testing::Each;
using ::testing::HasSubstr;
using ::testing::IsEmpty;
using ::testing::IsSupersetOf;
using ::testing::Le;
using ::testing::Not;
using ::testing::SizeIs;

// A mock for CentipedeCallbacks.
class CentipedeMock : public CentipedeCallbacks {
 public:
  CentipedeMock(const Environment &env) : CentipedeCallbacks(env) {}
  // Doesn't execute anything
  // Sets `batch_result.results()` based on the values of `inputs`:
  // Collects various stats about the inputs, to be checked in tests.
  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    batch_result.results().clear();
    // For every input, we create a 256-element array `counters`, where
    // i-th element is the number of bytes with the value 'i' in the input.
    // `counters` is converted to FeatureVec and added to
    // `batch_result.results()`.
    for (auto &input : inputs) {
      ByteArray counters(256);
      for (uint8_t byte : input) {
        counters[byte]++;
      }
      FeatureVec features;
      for (size_t i = 0; i < counters.size(); ++i) {
        if (counters[i] == 0) continue;
        features.push_back(feature_domains::k8bitCounters.ConvertToMe(
            Convert8bitCounterToNumber(i, counters[i])));
      }
      batch_result.results().emplace_back(ExecutionResult{features});
      if (input.size() == 1) {
        observed_1byte_inputs_.insert(input[0]);
      } else {
        EXPECT_EQ(input.size(), 2);
        uint16_t input2bytes = (input[0] << 8) | input[1];
        observed_2byte_inputs_.insert(input2bytes);
      }
      num_inputs_++;
    }
    num_executions_++;
    max_batch_size_ = std::max(max_batch_size_, inputs.size());
    min_batch_size_ = std::min(min_batch_size_, inputs.size());
    return true;
  }
  // Makes predictable mutants:
  // first 255 mutations are 1-byte sequences {1} ... {255}.
  // (the value {0} is produced by the default GetSeeds()).
  // Next 65536 mutations are 2-byte sequences {0,0} ... {255, 255}.
  // Then repeat 2-byte sequences.
  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    std::vector<ByteArray> mutants;
    mutants.reserve(num_mutants);
    for (size_t i = 0; i < num_mutants; ++i) {
      num_mutations_++;
      if (num_mutations_ < 256) {
        mutants.push_back({static_cast<uint8_t>(num_mutations_)});
        continue;
      }
      uint8_t byte0 = (num_mutations_ - 256) / 256;
      uint8_t byte1 = (num_mutations_ - 256) % 256;
      mutants.push_back({byte0, byte1});
    }
    return mutants;
  }

  absl::flat_hash_set<uint8_t> observed_1byte_inputs_;
  absl::flat_hash_set<uint16_t> observed_2byte_inputs_;

  size_t num_executions_ = 0;
  size_t num_inputs_ = 0;
  size_t num_mutations_ = 0;
  size_t max_batch_size_ = 0;
  size_t min_batch_size_ = -1;
};

// Returns the same CentipedeCallbacks object every time, never destroys it.
class MockFactory : public CentipedeCallbacksFactory {
 public:
  explicit MockFactory(CentipedeCallbacks &cb) : cb_(cb) {}
  CentipedeCallbacks *absl_nonnull create(const Environment &env) override {
    return &cb_;
  }
  void destroy(CentipedeCallbacks *cb) override { EXPECT_EQ(cb, &cb_); }

 private:
  CentipedeCallbacks &cb_;
};

TEST(Centipede, MockTest) {
  TempCorpusDir tmp_dir{test_info_->name()};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = tmp_dir.path();
  env.num_runs = 100000;  // Enough to run through all 1- and 2-byte inputs.
  env.batch_size = 7;     // Just some small number.
  env.require_pc_table = false;  // No PC table here.
  CentipedeMock mock(env);
  MockFactory factory(mock);
  CentipedeMain(env, factory);  // Run fuzzing with num_runs inputs.
  EXPECT_EQ(mock.num_inputs_, env.num_runs + 1);  // num_runs and one dummy.
  EXPECT_EQ(mock.num_mutations_, env.num_runs);
  EXPECT_EQ(mock.max_batch_size_, env.batch_size);
  EXPECT_EQ(mock.min_batch_size_, 1);  // 1 for dummy.
  EXPECT_EQ(tmp_dir.CountElementsInCorpusFile(0), 512);
  EXPECT_EQ(mock.observed_1byte_inputs_.size(), 256);    // all 1-byte seqs.
  EXPECT_EQ(mock.observed_2byte_inputs_.size(), 65536);  // all 2-byte seqs.
}

static size_t CountFilesInDir(std::string_view dir_path) {
  const std::filesystem::directory_iterator dir_iter{dir_path};
  return std::distance(std::filesystem::begin(dir_iter),
                       std::filesystem::end(dir_iter));
}

TEST(Centipede, ReadFirstCorpusDir) {
  TempDir workdir_1{test_info_->name(), "workdir_1"};
  TempDir workdir_2{test_info_->name(), "workdir_2"};
  TempDir corpus_dir{test_info_->name(), "corpus"};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = workdir_1.path();
  env.num_runs = 100000;  // Enough to run through all 1- and 2-byte inputs.
  env.batch_size = 7;     // Just some small number.
  env.require_pc_table = false;  // No PC table here.
  env.corpus_dir.push_back(corpus_dir.path());

  // Need to wrap each CentipedeMain in a scope to make sure the shmem is
  // released before the next call. Otherwise it may fail in MacOS.
  {
    // First, generate corpus files in corpus_dir.
    CentipedeMock mock_1(env);
    MockFactory factory_1(mock_1);
    CentipedeMain(env, factory_1);
    ASSERT_EQ(mock_1.observed_1byte_inputs_.size(), 256);    // all 1-byte seqs.
    ASSERT_EQ(mock_1.observed_2byte_inputs_.size(), 65536);  // all 2-byte seqs.
    ASSERT_EQ(CountFilesInDir(env.corpus_dir[0]),
              512);  // All 1-byte and 2-byte inputs.
  }

  {
    // Second, run without fuzzing using the same corpus_dir.
    env.workdir = workdir_2.path();
    env.num_runs = 0;
    CentipedeMock mock_2(env);
    MockFactory factory_2(mock_2);
    CentipedeMain(env, factory_2);
    // Should observe all inputs in corpus_dir, plus the dummy seed input {0}.
    EXPECT_EQ(mock_2.num_inputs_, 513);
  }
}

TEST(Centipede, DoesNotReadFirstCorpusDirIfOutputOnly) {
  TempDir workdir_1{test_info_->name(), "workdir_1"};
  TempDir workdir_2{test_info_->name(), "workdir_2"};
  TempDir corpus_dir{test_info_->name(), "corpus"};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = workdir_1.path();
  env.num_runs = 100000;  // Enough to run through all 1- and 2-byte inputs.
  env.batch_size = 7;     // Just some small number.
  env.require_pc_table = false;  // No PC table here.
  env.corpus_dir.push_back(corpus_dir.path());

  {
    // First, generate corpus files in corpus_dir.
    CentipedeMock mock_1(env);
    MockFactory factory_1(mock_1);
    CentipedeMain(env, factory_1);
    ASSERT_EQ(mock_1.observed_1byte_inputs_.size(), 256);    // all 1-byte seqs.
    ASSERT_EQ(mock_1.observed_2byte_inputs_.size(), 65536);  // all 2-byte seqs.
    ASSERT_EQ(CountFilesInDir(env.corpus_dir[0]),
              512);  // All 1-byte and 2-byte inputs.
  }

  {
    // Second, run without fuzzing using the same corpus_dir, but as
    // output-only.
    env.workdir = workdir_2.path();
    env.num_runs = 0;
    env.first_corpus_dir_output_only = true;
    CentipedeMock mock_2(env);
    MockFactory factory_2(mock_2);
    CentipedeMain(env, factory_2);
    // Should observe no inputs other than the seed input {0}.
    EXPECT_EQ(mock_2.num_inputs_, 1);
  }
}

TEST(Centipede, SkipsOutputIfFirstCorpusDirIsEmptyPath) {
  TempCorpusDir tmp_dir{test_info_->name()};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = tmp_dir.path();
  env.num_runs = 100000;  // Enough to run through all 1- and 2-byte inputs.
  env.batch_size = 7;     // Just some small number.
  env.require_pc_table = false;  // No PC table here.
  // Set the first corpus_dir entry to empty path to skip output.
  env.corpus_dir.push_back("");
  env.corpus_dir.push_back(tmp_dir.CreateSubdir("cd"));

  CentipedeMock mock(env);
  MockFactory factory(mock);
  CentipedeMain(env, factory);  // Run fuzzing with num_runs inputs.
  EXPECT_EQ(mock.observed_1byte_inputs_.size(), 256);    // all 1-byte seqs.
  EXPECT_EQ(mock.observed_2byte_inputs_.size(), 65536);  // all 2-byte seqs.
  // No output should be in other entires of corpus_dir.
  EXPECT_EQ(CountFilesInDir(env.corpus_dir[1]), 0);
}

// Tests fuzzing and distilling in multiple shards.
TEST(Centipede, ShardsAndDistillTest) {
  TempCorpusDir tmp_dir{test_info_->name()};
  Environment env;
  env.workdir = tmp_dir.path();
  env.log_level = 0;  // Disable most of the logging in the test.
  size_t combined_num_runs = 100000;  // Enough to run through all inputs.
  env.total_shards = 20;
  env.num_runs = combined_num_runs / env.total_shards;
  env.require_pc_table = false;  // No PC table here.

  // Create two empty dirs and add them to corpus_dir.
  env.corpus_dir.push_back(tmp_dir.CreateSubdir("cd1"));
  env.corpus_dir.push_back(tmp_dir.CreateSubdir("cd2"));

  CentipedeMock mock(env);
  // First round of runs: do the actual fuzzing, compute the features.
  size_t max_shard_size = 0;
  for (size_t shard_index = 0; shard_index < env.total_shards; shard_index++) {
    env.my_shard_index = shard_index;
    MockFactory factory(mock);
    CentipedeMain(env, factory);  // Run fuzzing in shard `shard_index`.
    auto corpus_size = tmp_dir.CountElementsInCorpusFile(shard_index);
    // Every byte should be present at least once.
    // With 2-byte inputs, we get at least 128 inputs covering 256 features.
    EXPECT_GT(corpus_size, 128);
    max_shard_size = std::max(max_shard_size, corpus_size);
  }
  EXPECT_EQ(mock.observed_1byte_inputs_.size(), 256);    // all 1-byte seqs.
  EXPECT_EQ(mock.observed_2byte_inputs_.size(), 65536);  // all 2-byte seqs.

  EXPECT_GT(CountFilesInDir(env.corpus_dir[0]), 128);
  EXPECT_EQ(CountFilesInDir(env.corpus_dir[1]), 0);

  // Second round of runs. Don't fuzz, only distill.
  // Don't distill in the last one to test the flag behaviour.
  env.distill = true;
  env.num_threads = env.total_shards - 1;
  env.my_shard_index = 0;
  // Empty the corpus_dir[0]
  std::filesystem::remove_all(env.corpus_dir[0]);
  std::filesystem::create_directory(env.corpus_dir[0]);
  MockFactory factory(mock);
  CentipedeMain(env, factory);  // Run distilling in shard `shard_index`.
  EXPECT_EQ(CountFilesInDir(env.corpus_dir[0]), 0);
  size_t distilled_size = 0;
  for (size_t shard_index = 0; shard_index < env.total_shards; shard_index++) {
    SCOPED_TRACE(absl::StrCat("Shard ", shard_index));
    const auto shard_distilled_size =
        tmp_dir.CountElementsInCorpusFile(shard_index, "distilled-.");
    if (shard_index == env.total_shards - 1) {
      // Didn't distill in the last shard.
      EXPECT_EQ(shard_distilled_size, 0);
    }
    distilled_size += shard_distilled_size;
  }
  // Distillation is expected to find more inputs than any individual shard.
  EXPECT_GT(distilled_size, max_shard_size);
  // And since we are expecting 512 features, with 2-byte inputs,
  // we get at least 512/2 corpus elements after distillation.
  EXPECT_GT(distilled_size, 256);
}

// Tests --input_filter. test_input_filter filters out inputs with 'b' in them.
TEST(Centipede, InputFilter) {
  TempCorpusDir tmp_dir{test_info_->name()};
  Environment env;
  env.workdir = tmp_dir.path();
  env.num_runs = 256;            // Enough to run through all 1- byte inputs.
  env.log_level = 0;             // Disable most of the logging in the test.
  env.require_pc_table = false;  // No PC table here.
  // Add %f so that test_input_filter doesn't need to be linked with forkserver.
  env.input_filter = "%f" + std::string{GetDataDependencyFilepath(
                                "centipede/testing/test_input_filter")};
  CentipedeMock mock(env);
  MockFactory factory(mock);
  CentipedeMain(env, factory);  // Run fuzzing.
  auto corpus = tmp_dir.GetCorpus(0);
  std::set<ByteArray> corpus_set(corpus.begin(), corpus.end());
  EXPECT_FALSE(corpus_set.count({'b'}));
  EXPECT_TRUE(corpus_set.count({'a'}));
  EXPECT_TRUE(corpus_set.count({'c'}));
}

// Callbacks for MutateViaExternalBinary test.
class MutateCallbacks : public CentipedeCallbacks {
 public:
  explicit MutateCallbacks(const Environment &env) : CentipedeCallbacks(env) {}
  // Will not be called.
  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    FUZZTEST_LOG(FATAL);
    return false;
  }

  // Will not be called.
  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    FUZZTEST_LOG(FATAL);
  }

  // Redeclare a protected member function as public so the tests can call it.
  using CentipedeCallbacks::MutateViaExternalBinary;
};

// Maintains `TemporaryLocalDirPath()` during the lifetime.
//
// Some parts of Centipede rely on `TemporaryLocalDirPath()` being set up as a
// global resource. Tests that exercise such parts of Centipede should use this
// fixture.
//
// TODO(b/391433873): Get rid of this once the design of
// `TemporaryLocalDirPath()` is revisited.
class CentipedeWithTemporaryLocalDir : public testing::Test {
 public:
  CentipedeWithTemporaryLocalDir() {
    std::filesystem::path tmp_dir = TemporaryLocalDirPath();
    std::filesystem::remove_all(tmp_dir);
    std::filesystem::create_directory(tmp_dir);
  }

  ~CentipedeWithTemporaryLocalDir() override {
    std::filesystem::remove_all(TemporaryLocalDirPath());
  }
};

TEST_F(CentipedeWithTemporaryLocalDir, MutateViaExternalBinary) {
  // This binary contains a test-friendly custom mutator.
  const std::string binary_with_custom_mutator =
      GetDataDependencyFilepath("centipede/testing/test_fuzz_target");
  // This binary does not contain a custom mutator.
  const std::string binary_without_custom_mutator =
      GetDataDependencyFilepath("centipede/testing/abort_fuzz_target");
  // Mutate a couple of different inputs.
  std::vector<ByteArray> inputs = {{0, 1, 2}, {3, 4}};
  // The custom mutator in the test binary will revert the order of bytes
  // and sometimes add a number in [100-107) at the end.
  // Periodically, the custom mutator will fall back to LLVMFuzzerMutate,
  // which in turn will sometimes shrink the inputs.
  std::vector<ByteArray> some_of_expected_mutants = {
      // Reverted inputs, sometimes with an extra byte at the end.
      {2, 1, 0},
      {2, 1, 0, 100},
      {2, 1, 0, 101},
      {2, 1, 0, 102},
      {4, 3},
      {4, 3, 103},
      {4, 3, 104},
      {4, 3, 105},
      // Shrunk inputs.
      {0, 1},
      {4}};

  std::vector<ByteArray> expected_crossover_mutants = {
      // Crossed-over mutants.
      {0, 1, 2, 42, 3, 4},
      {3, 4, 42, 0, 1, 2},
  };

  auto all_expected_mutants = some_of_expected_mutants;
  all_expected_mutants.insert(all_expected_mutants.end(),
                              expected_crossover_mutants.begin(),
                              expected_crossover_mutants.end());

  // Test with crossover enabled (default).
  {
    Environment env;
    MutateCallbacks callbacks(env);
    {
      const MutationResult result = callbacks.MutateViaExternalBinary(
          binary_without_custom_mutator,
          GetMutationInputRefsFromDataInputs(inputs), 1);
      EXPECT_EQ(result.exit_code(), EXIT_SUCCESS);
      EXPECT_FALSE(result.has_custom_mutator());
    }

    {
      const MutationResult result = callbacks.MutateViaExternalBinary(
          binary_with_custom_mutator,
          GetMutationInputRefsFromDataInputs(inputs), 10000);
      EXPECT_EQ(result.exit_code(), EXIT_SUCCESS);
      EXPECT_TRUE(result.has_custom_mutator());
      EXPECT_THAT(result.mutants(), AllOf(IsSupersetOf(all_expected_mutants),
                                          Each(Not(IsEmpty()))));
    }
  }

  // Test with a max_len of 10
  {
    Environment env;
    env.max_len = 10;
    MutateCallbacks callbacks(env);
    const MutationResult result = callbacks.MutateViaExternalBinary(
        binary_with_custom_mutator, GetMutationInputRefsFromDataInputs(inputs),
        10000);
    EXPECT_EQ(result.exit_code(), EXIT_SUCCESS);
    EXPECT_TRUE(result.has_custom_mutator());
    EXPECT_THAT(result.mutants(), AllOf(IsSupersetOf(all_expected_mutants),
                                        Each(Not(IsEmpty()))));
    EXPECT_THAT(result.mutants(),
                AllOf(IsSupersetOf(all_expected_mutants), Each(Not(IsEmpty())),
                      // The byte_array_mutator may insert up to 20 bytes to an
                      // input, which may push the size over the max_len.
                      Each(SizeIs(Le(30)))));
  }

  // Test with crossover disabled.
  {
    Environment env_no_crossover;
    env_no_crossover.crossover_level = 0;
    MutateCallbacks callbacks_no_crossover(env_no_crossover);

    const MutationResult result =
        callbacks_no_crossover.MutateViaExternalBinary(
            binary_with_custom_mutator,
            GetMutationInputRefsFromDataInputs(inputs), 10000);
    // Must contain normal mutants, but not the ones from crossover.
    EXPECT_THAT(result.mutants(), IsSupersetOf(some_of_expected_mutants));
    for (const auto &crossover_mutant : expected_crossover_mutants) {
      EXPECT_THAT(result.mutants(), Not(Contains(crossover_mutant)));
    }
  }
}

// A mock for MergeFromOtherCorpus test.
class MergeMock : public CentipedeCallbacks {
 public:
  explicit MergeMock(const Environment &env) : CentipedeCallbacks(env) {}

  // Doesn't execute anything.
  // All inputs are 1-byte long.
  // For an input {X}, the feature output is {X}.
  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    batch_result.results().resize(inputs.size());
    for (size_t i = 0, n = inputs.size(); i < n; ++i) {
      FUZZTEST_CHECK_EQ(inputs[i].size(), 1);
      batch_result.results()[i].mutable_features() = {inputs[i][0]};
    }
    return true;
  }

  // Every consecutive mutation is {number_of_mutations_} (starting from 1).
  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    std::vector<ByteArray> mutants{num_mutants};
    for (auto &mutant : mutants) {
      mutant.resize(1);
      mutant[0] = ++number_of_mutations_;
    }
    return mutants;
  }

  void Reset() { number_of_mutations_ = 0; }

 private:
  size_t number_of_mutations_ = 0;
};

TEST(Centipede, MergeFromOtherCorpus) {
  using Corpus = std::vector<ByteArray>;

  // Set up the workdir, create a 2-shard corpus with 3 inputs plus the seed {0}
  // each.
  TempCorpusDir work_tmp_dir{test_info_->name(), "workdir"};
  Environment env;
  env.workdir = work_tmp_dir.path();
  env.num_runs = 3;              // Just a few runs.
  env.require_pc_table = false;  // No PC table here.
  MergeMock mock(env);
  MockFactory factory(mock);
  for (env.my_shard_index = 0; env.my_shard_index < 2; ++env.my_shard_index) {
    CentipedeMain(env, factory);
  }
  CentipedeMain(env, factory);
  EXPECT_EQ(work_tmp_dir.GetCorpus(0), Corpus({{0}, {1}, {2}, {3}}));
  EXPECT_EQ(work_tmp_dir.GetCorpus(1), Corpus({{0}, {4}, {5}, {6}}));

  // Set up another workdir, create a 2-shard corpus there, with 4 inputs plus
  // the seed {0} each.
  TempCorpusDir merge_tmp_dir(test_info_->name(), "merge_from");
  Environment merge_env;
  merge_env.workdir = merge_tmp_dir.path();
  merge_env.num_runs = 4;
  merge_env.require_pc_table = false;  // No PC table here.
  mock.Reset();
  for (merge_env.my_shard_index = 0; merge_env.my_shard_index < 2;

       ++merge_env.my_shard_index) {
    CentipedeMain(merge_env, factory);
  }
  EXPECT_EQ(merge_tmp_dir.GetCorpus(0), Corpus({{0}, {1}, {2}, {3}, {4}}));
  EXPECT_EQ(merge_tmp_dir.GetCorpus(1), Corpus({{0}, {5}, {6}, {7}, {8}}));

  // Merge shards of `merge_env` into shards of `env`.
  // Shard 0 will receive one extra input: {4}
  // Shard 1 will receive two extra inputs: {7}, {8}
  env.merge_from = merge_tmp_dir.path();
  env.num_runs = 0;
  for (env.my_shard_index = 0; env.my_shard_index < 2; ++env.my_shard_index) {
    CentipedeMain(env, factory);
  }
  EXPECT_EQ(work_tmp_dir.GetCorpus(0), Corpus({{0}, {1}, {2}, {3}, {4}}));
  EXPECT_EQ(work_tmp_dir.GetCorpus(1), Corpus({{0}, {4}, {5}, {6}, {7}, {8}}));
}

// A mock for FunctionFilter test.
class FunctionFilterMock : public CentipedeCallbacks {
 public:
  explicit FunctionFilterMock(const Environment &env)
      : CentipedeCallbacks(env) {
    std::vector<ByteArray> seed_inputs;
    const size_t num_seeds_available = GetSeeds(/*num_seeds=*/1, seed_inputs);
    FUZZTEST_CHECK_EQ(num_seeds_available, 1)
        << "Default seeds must have size one.";
    FUZZTEST_CHECK_EQ(seed_inputs.size(), 1)
        << "Default seeds must have size one.";
    seed_inputs_.insert(seed_inputs.begin(), seed_inputs.end());
  }

  // Executes the target in the normal way.
  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    return ExecuteCentipedeSancovBinaryWithShmem(
               env_.binary, inputs, batch_result, deadline) == EXIT_SUCCESS;
  }

  // Sets the inputs to one of 3 pre-defined values.
  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    for (auto &input : inputs) {
      if (!seed_inputs_.contains(input.data)) {
        observed_inputs_.insert(input.data);
      }
    }
    std::vector<ByteArray> mutants;
    mutants.reserve(num_mutants);
    for (size_t i = 0; i < num_mutants; ++i) {
      mutants.push_back(GetMutant(++number_of_mutations_));
    }
    return mutants;
  }

  // Returns one of 3 pre-defined values, that trigger different code paths in
  // the test target.
  static ByteArray GetMutant(size_t idx) {
    const char *mutants[3] = {"func1", "func2-A", "foo"};
    const char *mutant = mutants[idx % 3];
    return {mutant, mutant + strlen(mutant)};
  }

  // Seed inputs generated from GetSeeds().
  absl::flat_hash_set<ByteArray> seed_inputs_;
  // Set of inputs observed by Mutate(), except for seed inputs.
  absl::flat_hash_set<ByteArray> observed_inputs_;

 private:
  size_t number_of_mutations_ = 0;
};

// Runs a short fuzzing session with the provided `function_filter`.
// Returns a sorted array of observed inputs.
static std::vector<ByteArray> RunWithFunctionFilter(
    std::string_view function_filter, const TempDir &tmp_dir) {
  Environment env;
  env.workdir = tmp_dir.path();
  env.seed = 1;  // make the runs predictable.
  env.num_runs = 100;
  env.batch_size = 10;
  env.binary = GetDataDependencyFilepath("centipede/testing/test_fuzz_target");
  env.coverage_binary = env.binary;
  // Must symbolize in order for the filter to work.
  env.symbolizer_path = GetLLVMSymbolizerPath();
  env.objdump_path = GetObjDumpPath();
  env.log_level = 0;
  env.function_filter = function_filter;
  FunctionFilterMock mock(env);
  MockFactory factory(mock);
  CentipedeMain(env, factory);
  FUZZTEST_LOG(INFO) << mock.observed_inputs_.size();
  std::vector<ByteArray> res(mock.observed_inputs_.begin(),
                             mock.observed_inputs_.end());
  std::sort(res.begin(), res.end());
  return res;
}

// Tests --function_filter.
TEST(Centipede, FunctionFilter) {
  // Run with empty function filter.
  {
    TempDir tmp_dir{test_info_->name(), "none"};
    auto observed_empty = RunWithFunctionFilter("", tmp_dir);
    ASSERT_EQ(observed_empty.size(), 3);
  }

  // Run with a one-function filter
  {
    TempDir tmp_dir{test_info_->name(), "single"};
    auto observed_single = RunWithFunctionFilter("SingleEdgeFunc", tmp_dir);
    ASSERT_EQ(observed_single.size(), 1);
    EXPECT_EQ(observed_single[0], FunctionFilterMock::GetMutant(0));
  }

  // Run with a two-function filter.
  {
    TempDir tmp_dir{test_info_->name(), "single_multi"};
    auto observed_both =
        RunWithFunctionFilter("SingleEdgeFunc,MultiEdgeFunc", tmp_dir);
    ASSERT_EQ(observed_both.size(), 2);
    EXPECT_EQ(observed_both[0], FunctionFilterMock::GetMutant(0));
    EXPECT_EQ(observed_both[1], FunctionFilterMock::GetMutant(1));
  }
}

struct Crash {
  std::string binary;
  unsigned char input = 0;
  std::string description;
  std::string signature;
};

// A mock for ExtraBinaries test.
class ExtraBinariesMock : public CentipedeCallbacks {
 public:
  explicit ExtraBinariesMock(const Environment &env, std::vector<Crash> crashes)
      : CentipedeCallbacks(env), crashes_(std::move(crashes)) {}

  // Doesn't execute anything.
  // On certain combinations of {binary,input} returns false.
  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    bool res = true;
    for (const auto &input : inputs) {
      if (input.size() != 1) continue;
      for (const Crash &crash : crashes_) {
        if (binary == crash.binary && input[0] == crash.input) {
          batch_result.failure_description() = crash.description;
          batch_result.failure_signature() = crash.signature;
          res = false;
        }
      }
    }
    batch_result.results().resize(inputs.size());
    return res;
  }

  // Sets the mutants to different 1-byte values.
  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    std::vector<ByteArray> mutants{num_mutants};
    for (auto &mutant : mutants) {
      mutant.resize(1);
      mutant[0] = ++number_of_mutations_;
    }
    return mutants;
  }

 private:
  size_t number_of_mutations_ = 0;
  std::vector<Crash> crashes_;
};

struct FileAndContents {
  std::string file;
  std::string contents;

  bool operator==(const FileAndContents &other) const {
    return file == other.file && contents == other.contents;
  }

  template <typename Sink>
  friend void AbslStringify(Sink &sink, const FileAndContents &f) {
    absl::Format(&sink, "FileAndContents{%s, \"%s\"}", f.file, f.contents);
  }
};

MATCHER_P(HasFilesWithContents, expected_files_and_contents, "") {
  const std::string &dir_path = arg;
  std::vector<FileAndContents> files_and_contents;
  for (const auto &dir_ent : std::filesystem::directory_iterator(dir_path)) {
    auto file_and_contents = FileAndContents{dir_ent.path().filename()};
    ReadFromLocalFile(dir_ent.path().c_str(), file_and_contents.contents);
    files_and_contents.push_back(std::move(file_and_contents));
  }
  return ExplainMatchResult(expected_files_and_contents, files_and_contents,
                            result_listener);
}

// Tests --extra_binaries.
// Executes one main binary (--binary) and 3 extra ones (--extra_binaries).
// Expects the main binary and two extra ones to generate one crash each.
TEST(Centipede, ExtraBinaries) {
  TempDir tmp_dir{test_info_->name()};
  Environment env;
  env.workdir = tmp_dir.path();
  env.num_runs = 100;
  env.batch_size = 10;
  env.log_level = 1;
  env.binary = "b1";
  env.extra_binaries = {"b2", "b3", "b4"};
  env.require_pc_table = false;  // No PC table here.
  ExtraBinariesMock mock(env, {Crash{"b1", 10, "b1-crash", "b1-sig"},
                               Crash{"b2", 30, "b2-crash", "b2-sig"},
                               Crash{"b3", 50, "b3-crash", "b3-sig"}});
  MockFactory factory(mock);
  CentipedeMain(env, factory);

  // Verify that we see the expected crashes.
  // The "crashes" dir must contain 3 crashy inputs, one for each binary.
  auto crashes_dir_path = WorkDir{env}.CrashReproducerDirPaths().MyShard();
  ASSERT_TRUE(std::filesystem::exists(crashes_dir_path))
      << VV(crashes_dir_path);
  EXPECT_THAT(crashes_dir_path,
              HasFilesWithContents(testing::UnorderedElementsAre(
                  FileAndContents{Hash({10}), AsString({10})},
                  FileAndContents{Hash({30}), AsString({30})},
                  FileAndContents{Hash({50}), AsString({50})})));

  // Verify that we see the expected crash metadata.
  // The "crash-metadata" dir must contain 3 crash metadata files, one for each
  // crashy input.
  auto crash_metadata_dir_path = WorkDir{env}.CrashMetadataDirPaths().MyShard();
  ASSERT_TRUE(std::filesystem::exists(crash_metadata_dir_path))
      << VV(crash_metadata_dir_path);
  EXPECT_THAT(
      crash_metadata_dir_path,
      HasFilesWithContents(testing::UnorderedElementsAre(
          FileAndContents{absl::StrCat(Hash({10}), ".desc"), "b1-crash"},
          FileAndContents{absl::StrCat(Hash({10}), ".sig"), "b1-sig"},
          FileAndContents{absl::StrCat(Hash({30}), ".desc"), "b2-crash"},
          FileAndContents{absl::StrCat(Hash({30}), ".sig"), "b2-sig"},
          FileAndContents{absl::StrCat(Hash({50}), ".desc"), "b3-crash"},
          FileAndContents{absl::StrCat(Hash({50}), ".sig"), "b3-sig"})));
}

// A mock for UndetectedCrashingInput test.
class UndetectedCrashingInputMock : public CentipedeCallbacks {
 public:
  explicit UndetectedCrashingInputMock(const Environment &env,
                                       size_t crashing_input_idx)
      : CentipedeCallbacks{env}, crashing_input_idx_{crashing_input_idx} {
    FUZZTEST_CHECK_LE(crashing_input_idx_, std::numeric_limits<uint8_t>::max());
  }

  // Doesn't execute anything.
  // Crash when 0th char of input to binary b1 equals `crashing_input_idx_`, but
  // only on 1st exec.
  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    batch_result.ClearAndResize(inputs.size());
    bool res = true;
    if (!first_pass_) {
      num_inputs_triaged_ += inputs.size();
    }
    for (const auto &input : inputs) {
      FUZZTEST_CHECK_EQ(input.size(), 1);  // By construction in `Mutate()`.
      // The contents of each mutant is its sequential number.
      if (input[0] == crashing_input_idx_) {
        if (first_pass_) {
          first_pass_ = false;
          crashing_input_ = input;
          // TODO(b/274705740): `num_outputs_read()` is the number of outputs
          //  that Centipede engine *expects* to have been read from *the
          //  current BatchResult* by the *particular* implementation of
          //  `CentipedeCallbacks` (and `DefaultCentipedeCallbacks` fits the
          //  bill). `fuzztest::internal::ReportCrash()` then uses this value as
          //  a hint for the crashing input's index, and in our case saves the
          //  batch's inputs from 0 up to and including the crasher to a subdir.
          //  See the bug for details. All of this is horribly convoluted and
          //  misplaced here. Implement a cleaner solution.
          batch_result.num_outputs_read() =
              crashing_input_idx_ % env_.batch_size;
          res = false;
        }
      }
    }
    return res;
  }

  // Sets the mutants to different 1-byte values.
  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    std::vector<ByteArray> mutants;
    mutants.reserve(num_mutants);
    for (size_t i = 0; i < num_mutants; ++i) {
      // The contents of each mutant is simply its sequential number.
      mutants.push_back({static_cast<uint8_t>(curr_input_idx_++)});
    }
    return mutants;
  }

  // Gets the input that triggered the crash.
  ByteArray crashing_input() const { return crashing_input_; }

  size_t num_inputs_triaged() const { return num_inputs_triaged_; }

 private:
  const size_t crashing_input_idx_;
  size_t curr_input_idx_ = 0;
  size_t num_inputs_triaged_ = 0;
  ByteArray crashing_input_ = {};
  bool first_pass_ = true;
};

// Test for preserving a crashing batch when 1-by-1 exec fails to reproduce.
// Executes one main binary (--binary).
// Expects the binary to crash once and 1-by-1 reproduction to fail.
TEST(Centipede, UndetectedCrashingInput) {
  constexpr size_t kNumBatches = 7;
  constexpr size_t kBatchSize = 11;
  constexpr size_t kCrashingInputIdxInBatch = kBatchSize / 2;
  constexpr size_t kCrashingInputIdx =
      (kNumBatches / 2) * kBatchSize + kCrashingInputIdxInBatch;

  FUZZTEST_LOG(INFO) << VV(kNumBatches) << VV(kBatchSize)
                     << VV(kCrashingInputIdxInBatch) VV(kCrashingInputIdx);

  TempDir temp_dir{test_info_->name()};
  Environment env;
  env.workdir = temp_dir.path();
  env.num_runs = kBatchSize * kNumBatches;
  env.batch_size = kBatchSize;
  // No real binary: prevent attempts by Centipede to read a PCtable from it.
  env.require_pc_table = false;
  env.exit_on_crash = true;

  {
    UndetectedCrashingInputMock mock(env, kCrashingInputIdx);
    MockFactory factory(mock);
    CentipedeMain(env, factory);

    // Verify that we see the expected inputs from the batch.
    // The "crashes/unreliable_batch-<HASH>" dir must contain all inputs from
    // the batch that were executing during the session. We simply verify the
    // number of saved inputs matches the number of executed inputs.
    const auto crashing_input_hash = Hash(mock.crashing_input());
    const auto crashes_dir_path =
        std::filesystem::path{
            WorkDir{env}.CrashReproducerDirPaths().MyShard()} /
        absl::StrCat("crashing_batch-", crashing_input_hash);
    EXPECT_TRUE(std::filesystem::exists(crashes_dir_path)) << crashes_dir_path;
    std::vector<std::string> found_crash_file_names;
    for (auto const &dir_ent :
         std::filesystem::directory_iterator(crashes_dir_path)) {
      found_crash_file_names.push_back(dir_ent.path().filename());
    }
    // TODO(ussuri): Verify exact names/contents of the files, not just count.
    EXPECT_EQ(found_crash_file_names.size(), kCrashingInputIdxInBatch + 1);
    // Suspected input first, then every input in the batch (including the
    // suspected input again).
    EXPECT_EQ(mock.num_inputs_triaged(), kBatchSize + 1);
  }

  // Verify that when `env.batch_triage_suspect_only` is set, only triage the
  // suspect.
  TempDir suspect_only_temp_dir{test_info_->name()};
  env.workdir = suspect_only_temp_dir.path();
  env.batch_triage_suspect_only = true;
  UndetectedCrashingInputMock suspect_only_mock(env, kCrashingInputIdx);
  MockFactory suspect_only_factory(suspect_only_mock);
  CentipedeMain(env, suspect_only_factory);

  EXPECT_EQ(suspect_only_mock.num_inputs_triaged(), 1);
}

TEST_F(CentipedeWithTemporaryLocalDir, GetsSeedInputs) {
  Environment env;
  env.binary =
      GetDataDependencyFilepath("centipede/testing/seeded_fuzz_target");
  CentipedeDefaultCallbacks callbacks(env);

  std::vector<ByteArray> seeds;
  EXPECT_EQ(callbacks.GetSeeds(10, seeds), 10);
  EXPECT_THAT(seeds, testing::ContainerEq(std::vector<ByteArray>{
                         {0}, {1}, {2}, {3}, {4}, {5}, {6}, {7}, {8}, {9}}));
  EXPECT_EQ(callbacks.GetSeeds(5, seeds), 10);
  EXPECT_THAT(seeds, testing::ContainerEq(
                         std::vector<ByteArray>{{0}, {1}, {2}, {3}, {4}}));
  EXPECT_EQ(callbacks.GetSeeds(100, seeds), 10);
  EXPECT_THAT(seeds, testing::ContainerEq(std::vector<ByteArray>{
                         {0}, {1}, {2}, {3}, {4}, {5}, {6}, {7}, {8}, {9}}));
}

TEST_F(CentipedeWithTemporaryLocalDir, GetsSerializedTargetConfig) {
  Environment env;
  env.binary =
      GetDataDependencyFilepath("centipede/testing/fuzz_target_with_config");
  CentipedeDefaultCallbacks callbacks(env);

  const auto serialized_config = callbacks.GetSerializedTargetConfig();
  ASSERT_TRUE(serialized_config.ok());
  EXPECT_EQ(*serialized_config, "fake serialized config");
}

TEST_F(CentipedeWithTemporaryLocalDir,
       GetSerializedTargetConfigProducesFailure) {
  Environment env;
  env.binary = absl::StrCat(
      GetDataDependencyFilepath("centipede/testing/fuzz_target_with_config")
          .c_str(),
      " --simulate_failure");
  CentipedeDefaultCallbacks callbacks(env);

  const auto serialized_config = callbacks.GetSerializedTargetConfig();
  EXPECT_FALSE(serialized_config.ok());
}

TEST_F(CentipedeWithTemporaryLocalDir, CleansUpMetadataAfterStartup) {
  Environment env;
  env.binary = GetDataDependencyFilepath(
      "centipede/testing/expensive_startup_fuzz_target");
  CentipedeDefaultCallbacks callbacks(env);

  BatchResult batch_result;
  const std::vector<ByteArray> inputs = {{0}};
  ASSERT_TRUE(callbacks.Execute(env.binary, inputs, batch_result,
                                /*deadline=*/absl::InfiniteFuture()));
  ASSERT_EQ(batch_result.results().size(), 1);
  bool found_startup_cmp_entry = false;
  batch_result.results()[0].metadata().ForEachCmpEntry(
      [&](ByteSpan a, ByteSpan b) {
        if (a == ByteArray{'F', 'u', 'z', 'z'}) found_startup_cmp_entry = true;
        if (b == ByteArray{'F', 'u', 'z', 'z'}) found_startup_cmp_entry = true;
      });
  EXPECT_FALSE(found_startup_cmp_entry);
}

class FakeCentipedeCallbacksForThreadChecking : public CentipedeCallbacks {
 public:
  FakeCentipedeCallbacksForThreadChecking(const Environment &env,
                                          std::thread::id execute_thread_id)
      : CentipedeCallbacks(env), execute_thread_id_(execute_thread_id) {}

  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    batch_result.ClearAndResize(inputs.size());
    thread_check_passed_ = thread_check_passed_ &&
                           std::this_thread::get_id() == execute_thread_id_;
    return true;
  }

  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    return {num_mutants, {0}};
  }

  bool thread_check_passed() { return thread_check_passed_; }

 private:
  std::thread::id execute_thread_id_;
  bool thread_check_passed_ = true;
};

TEST(Centipede, RunsExecuteCallbackInTheCurrentThreadWhenFuzzingWithOneThread) {
  TempDir temp_dir{test_info_->name()};
  Environment env;
  env.workdir = temp_dir.path();
  env.require_pc_table = false;
  ASSERT_EQ(env.num_threads, 1);
  FakeCentipedeCallbacksForThreadChecking callbacks(env,
                                                    std::this_thread::get_id());
  BatchResult batch_result;
  const std::vector<ByteArray> inputs = {{0}};
  env.num_runs = 100;
  MockFactory factory(callbacks);
  EXPECT_EQ(CentipedeMain(env, factory), EXIT_SUCCESS);
  EXPECT_TRUE(callbacks.thread_check_passed());
}

TEST_F(CentipedeWithTemporaryLocalDir, DetectsStackOverflow) {
  Environment env;
  env.binary = GetDataDependencyFilepath("centipede/testing/test_fuzz_target");
  env.stack_limit_kb = 64;
  CentipedeDefaultCallbacks callbacks(env);

  BatchResult batch_result;
  const std::vector<ByteArray> inputs = {ByteArray{'s', 't', 'k'}};

  ASSERT_FALSE(callbacks.Execute(env.binary, inputs, batch_result,
                                 /*deadline=*/absl::InfiniteFuture()));
  EXPECT_THAT(batch_result.log(), HasSubstr("Stack limit exceeded"));
  EXPECT_EQ(batch_result.failure_description(), "stack-limit-exceeded");
}

class SetupFailureCallbacks : public CentipedeCallbacks {
 public:
  using CentipedeCallbacks::CentipedeCallbacks;

  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    ++execute_count_;
    batch_result.ClearAndResize(inputs.size());
    batch_result.exit_code() = EXIT_FAILURE;
    batch_result.failure_description() = "SETUP FAILURE: something went wrong";
    return false;
  }

  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    return {num_mutants, {0}};
  }

  int execute_count() const { return execute_count_; }

 private:
  int execute_count_ = 0;
};

TEST(Centipede, ReturnsFailureOnSetupFailure) {
  TempDir temp_dir{test_info_->name()};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = temp_dir.path();
  env.batch_size = 7;            // Just some small number.
  env.require_pc_table = false;  // No PC table here.
  SetupFailureCallbacks mock(env);
  MockFactory factory(mock);
  EXPECT_EQ(CentipedeMain(env, factory), EXIT_FAILURE);
  EXPECT_EQ(mock.execute_count(), 1);
}

class SkippedTestCallbacks : public CentipedeCallbacks {
 public:
  using CentipedeCallbacks::CentipedeCallbacks;

  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time deadline) override {
    ++execute_count_;
    batch_result.ClearAndResize(inputs.size());
    batch_result.exit_code() = EXIT_FAILURE;
    batch_result.failure_description() =
        "SKIPPED TEST: test skipped on purpose";
    return false;
  }

  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    return {num_mutants, {0}};
  }

  int execute_count() const { return execute_count_; }

 private:
  int execute_count_ = 0;
};

TEST(Centipede, ReturnsSuccessOnSkippedTest) {
  TempDir temp_dir{test_info_->name()};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = temp_dir.path();
  env.batch_size = 7;            // Just some small number.
  env.require_pc_table = false;  // No PC table here.
  SkippedTestCallbacks mock(env);
  MockFactory factory(mock);
  EXPECT_EQ(CentipedeMain(env, factory), EXIT_SUCCESS);
  EXPECT_EQ(mock.execute_count(), 1);
}

class IgnoredFailureCallbacks : public CentipedeCallbacks {
 public:
  using CentipedeCallbacks::CentipedeCallbacks;

  bool Execute(std::string_view binary, const std::vector<ByteArray>& inputs,
               BatchResult& batch_result, absl::Time daedline) override {
    ++execute_count_;
    batch_result.ClearAndResize(inputs.size());
    batch_result.exit_code() = EXIT_FAILURE;
    batch_result.failure_description() =
        "IGNORED FAILURE: failure ignored on purpose";
    return false;
  }

  std::vector<ByteArray> Mutate(const std::vector<MutationInputRef> &inputs,
                                size_t num_mutants) override {
    return {num_mutants, {0}};
  }

  int execute_count() const { return execute_count_; }

 private:
  int execute_count_ = 0;
};

TEST(Centipede, KeepsRunningAndReturnsSuccessWithIgnoredFailures) {
  TempDir temp_dir{test_info_->name()};
  Environment env;
  env.log_level = 0;  // Disable most of the logging in the test.
  env.workdir = temp_dir.path();
  env.batch_size = 7;  // Just some small number.
  env.num_runs = 100;
  env.require_pc_table = false;  // No PC table here.
  env.exit_on_crash = true;
  IgnoredFailureCallbacks mock(env);
  MockFactory factory(mock);
  EXPECT_EQ(CentipedeMain(env, factory), EXIT_SUCCESS);
  EXPECT_GE(mock.execute_count(), 2);
}

TEST_F(CentipedeWithTemporaryLocalDir, UsesProvidedCustomMutator) {
  Environment env;
  env.binary = GetDataDependencyFilepath(
      "centipede/testing/fuzz_target_with_custom_mutator");
  CentipedeDefaultCallbacks callbacks(env);

  const std::vector<ByteArray> inputs = {{1}, {2}, {3}, {4}, {5}, {6}};
  const std::vector<ByteArray> mutants = callbacks.Mutate(
      GetMutationInputRefsFromDataInputs(inputs), inputs.size());

  // The custom mutator just returns the original inputs as mutants.
  EXPECT_EQ(inputs, mutants);
}

TEST_F(CentipedeWithTemporaryLocalDir, FailsOnMisbehavingCustomMutator) {
  Environment env;
  env.binary =
      absl::StrCat(GetDataDependencyFilepath(
                       "centipede/testing/fuzz_target_with_custom_mutator")
                       .c_str(),
                   " --simulate_failure");
  CentipedeDefaultCallbacks callbacks(env);

  const std::vector<ByteArray> inputs = {{1}, {2}, {3}, {4}, {5}, {6}};
  // Previous stop condition could interfere here.
  ClearEarlyStopRequestAndSetStopTime(absl::InfiniteFuture());
  EXPECT_THAT(callbacks.Mutate(GetMutationInputRefsFromDataInputs(inputs),
                               inputs.size()),
              IsEmpty());
  EXPECT_TRUE(EarlyStopRequested());
  EXPECT_EQ(ExitCode(), EXIT_FAILURE);
}

TEST_F(CentipedeWithTemporaryLocalDir,
       FallsBackToBuiltInMutatorWhenCustomMutatorNotProvided) {
  Environment env;
  env.binary = GetDataDependencyFilepath("centipede/testing/abort_fuzz_target");
  CentipedeDefaultCallbacks callbacks(env);

  const std::vector<ByteArray> inputs = {{1}, {2}, {3}, {4}, {5}, {6}};
  const std::vector<ByteArray> mutants = callbacks.Mutate(
      GetMutationInputRefsFromDataInputs(inputs), inputs.size());

  // The built-in mutator performs non-trivial mutations.
  EXPECT_EQ(inputs.size(), mutants.size());
  EXPECT_NE(inputs, mutants);
}

TEST_F(CentipedeWithTemporaryLocalDir, HangingFuzzTargetExitsAfterTimeout) {
  Environment env;
  env.binary =
      GetDataDependencyFilepath("centipede/testing/hanging_fuzz_target");
  BatchResult batch_result;
  const std::vector<ByteArray> inputs = {{0}};
  CentipedeDefaultCallbacks callbacks(env);

  env.timeout_per_batch = 1;
  env.fork_server = false;

  // Test that the process does not get stuck and exits promptly.
  EXPECT_FALSE(callbacks.Execute(env.binary, {{0}}, batch_result,
                                 /*deadline=*/absl::InfiniteFuture()));
}

TEST_F(CentipedeWithTemporaryLocalDir, RunnerExitsAfterFirstCustomFailure) {
  Environment env;
  env.binary = GetDataDependencyFilepath("centipede/testing/test_fuzz_target");
  CentipedeDefaultCallbacks callbacks(env);
  BatchResult result;
  std::vector<ByteArray> inputs = {
      {'c', 'u', 's', 't', 'o', 'm'},
      {'c', 'u', 's', 't', 'o', 'm'},
  };
  EXPECT_FALSE(callbacks.Execute(env.binary, inputs, result,
                                 /*deadline=*/absl::InfiniteFuture()));
  EXPECT_THAT(result.failure_description(), HasSubstr("custom"));
  EXPECT_THAT(result.log(), AllOf(HasSubstr("custom failure 0"),
                                  Not(HasSubstr("custom failure 1"))));
}

}  // namespace
}  // namespace fuzztest::internal