You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
2.8 KiB
67 lines
2.8 KiB
// Copyright 2024 Google LLC
|
|
//
|
|
// 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
|
|
//
|
|
// http://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.
|
|
|
|
#ifndef FUZZTEST_E2E_TESTS_TEST_BINARY_UTIL_H_
|
|
#define FUZZTEST_E2E_TESTS_TEST_BINARY_UTIL_H_
|
|
|
|
#include <string>
|
|
|
|
#include "absl/container/flat_hash_map.h"
|
|
#include "absl/strings/string_view.h"
|
|
#include "absl/time/time.h"
|
|
#include "./fuzztest/internal/subprocess.h"
|
|
|
|
namespace fuzztest::internal {
|
|
|
|
// Returns a string of the form "--flag_name=flag_value" (or "--flag_name" if
|
|
// `flag_value` is empty), but with the flag name transformed to take into
|
|
// account any internal flag prefix.
|
|
std::string CreateFuzzTestFlag(absl::string_view flag_name,
|
|
absl::string_view flag_value);
|
|
|
|
// Returns the full path to the test binary based on the path relative to the
|
|
// e2e_tests directory. The test binary's name must end with the suffix
|
|
// ".stripped", but the function doesn't require the suffix in `relative_path`.
|
|
// If the suffix is missing, it will be added.
|
|
std::string BinaryPath(absl::string_view relative_path);
|
|
|
|
// Returns the full path to the Centipede binary.
|
|
std::string CentipedePath();
|
|
|
|
struct RunOptions {
|
|
// General flags to pass to the binary. Useful when passing flags to
|
|
// binaries like Centipede, e.g., {"binary", SOME_BINARY_PATH}
|
|
absl::flat_hash_map<std::string, std::string> flags;
|
|
// Flags to pass to a FuzzTest binary. These should be given without the
|
|
// internal flag prefix, e.g., just ("fuzz_for", "1s").
|
|
absl::flat_hash_map<std::string, std::string> fuzztest_flags;
|
|
// Raw args to append after `fuzztest_flags` and `flags`. Useful when passing
|
|
// positional arguments or non-standard flags, e.g. "--" for ending absl flag
|
|
// processing, and `-NAME=VALUE` for libFuzzer flags.
|
|
std::vector<std::string> raw_args;
|
|
// Environment variables to pass to the binary.
|
|
absl::flat_hash_map<std::string, std::string> env;
|
|
// Duration after which the binary will be terminated.
|
|
absl::Duration timeout = absl::Minutes(10);
|
|
};
|
|
|
|
// Runs the binary given by its full `binary_path` in a subprocess and returns
|
|
// the results. The `options` parameter can be used to pass flags and
|
|
// environment variables to the binary.
|
|
RunResults RunBinary(absl::string_view binary_path,
|
|
const RunOptions& options = {});
|
|
|
|
} // namespace fuzztest::internal
|
|
|
|
#endif // FUZZTEST_E2E_TESTS_TEST_BINARY_UTIL_H_
|
|
|