base/test/confirm_fuzztest_init.cc

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <string>
#include <vector>

#include "base/compiler_specific.h"
#include "base/no_destructor.h"
#include "base/test/confirm_fuzztest_init_buildflags.h"
#include "base/test/fuzztest_init_helper.h"
#include "third_party/fuzztest/src/fuzztest/init_fuzztest.h"

namespace {

static void RealInitFunction(int argc, char** argv) {
  static base::NoDestructor<std::vector<std::string>> fuzztest_argv_strings;
  static base::NoDestructor<std::vector<char*>> fuzztest_argv_data;
  static int fuzztest_argc;
  static char** fuzztest_argv;
  // Fuzztest might refer to the command line later, by which time Chromium
  // may have altered it. Keep our own copy so that the data
  // we pass into fuzztest remains valid, no matter what Chromium code
  // does to the original.
  fuzztest_argv_strings->reserve(argc);
  fuzztest_argv_data->reserve(argc);
  for (int i = 0; i < argc; i++) {
    // SAFETY: this function relies upon paired argc and argv
    // per command-line norms. Spanification of this family
    // of functions has been attempted, but because the
    // provider and consumer of the data both work in terms of
    // old-fashioned argc/argv pairs, it seemed to introduce
    // more complexity than it eliminated.
    UNSAFE_BUFFERS(fuzztest_argv_strings->push_back(argv[i]));
    fuzztest_argv_data->push_back(fuzztest_argv_strings->back().data());
  }
  fuzztest_argc = argc;
  fuzztest_argv = fuzztest_argv_data->data();
  fuzztest::ParseAbslFlags(fuzztest_argc, fuzztest_argv);
#if BUILDFLAG(REGISTER_FUZZTESTS_IN_TEST_SUITES)
  fuzztest::InitFuzzTest(&fuzztest_argc, &fuzztest_argv);
#endif
}

// base/test:test_support is used in test suites containing fuzztests
// and those without. In those without, we want to avoid depending
// on fuzztest's complex dependencies, but on those with fuzztests
// we need to call InitFuzzTest. So, use a static initializer to fill
// in a function pointer in those cases.
class FuzztestInitializer {
 public:
  FuzztestInitializer() {
    fuzztest_init_helper::initialization_function = RealInitFunction;
  }
};

FuzztestInitializer static_initializer;  // NOLINT

}  // namespace