123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175 |
- // Copyright 2015 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- // Fuzzer launcher script tests.
- #include <string>
- #include <vector>
- #include "base/command_line.h"
- #include "base/path_service.h"
- #include "base/process/launch.h"
- #include "base/strings/string_split.h"
- #include "testing/gmock/include/gmock/gmock.h"
- #include "testing/gtest/include/gtest/gtest.h"
- TEST(FuzzerConfigTest, DictOnly) {
- // Test of automatically generated .options file for fuzzer with dict option.
- base::FilePath exe_path;
- base::PathService::Get(base::FILE_EXE, &exe_path);
- std::string launcher_path =
- exe_path.DirName().Append("check_fuzzer_config.py").value();
- std::string output;
- base::CommandLine cmd(
- std::vector<std::string>({launcher_path, "test_dict_only.options"}));
- bool success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> fuzzer_args = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(1UL, fuzzer_args.size());
- EXPECT_EQ(fuzzer_args[0], "dict=test_dict_only.dict");
- }
- TEST(FuzzerConfigTest, ConfigOnly) {
- // Test of .options file for fuzzer with libfuzzer_options and without dict.
- base::FilePath exe_path;
- base::PathService::Get(base::FILE_EXE, &exe_path);
- std::string launcher_path =
- exe_path.DirName().Append("check_fuzzer_config.py").value();
- std::string output;
- base::CommandLine cmd(
- std::vector<std::string>({launcher_path, "test_config_only.options"}));
- bool success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> fuzzer_args = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(2UL, fuzzer_args.size());
- EXPECT_EQ(fuzzer_args[0], "some_test_option=test_value");
- EXPECT_EQ(fuzzer_args[1], "max_len=1024");
- }
- TEST(FuzzerConfigTest, ConfigAndDict) {
- // Test of .options file for fuzzer with options file and dictionary.
- base::FilePath exe_path;
- base::PathService::Get(base::FILE_EXE, &exe_path);
- std::string launcher_path =
- exe_path.DirName().Append("check_fuzzer_config.py").value();
- std::string output;
- base::CommandLine cmd(std::vector<std::string>(
- {launcher_path, "test_config_and_dict.options"}));
- bool success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> fuzzer_args = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(4UL, fuzzer_args.size());
- EXPECT_EQ(fuzzer_args[0], "dict=test_config_and_dict.dict");
- EXPECT_EQ(fuzzer_args[1], "max_len=random(1337, 31337)");
- EXPECT_EQ(fuzzer_args[2], "timeout=666");
- EXPECT_EQ(fuzzer_args[3], "use_traces=1");
- }
- TEST(FuzzerConfigTest, ConfigAndSeedCorpus) {
- // Test of .options file for fuzzer with libfuzzer_options and seed corpus.
- base::FilePath exe_path;
- base::PathService::Get(base::FILE_EXE, &exe_path);
- std::string launcher_path =
- exe_path.DirName().Append("check_fuzzer_config.py").value();
- std::string output;
- base::CommandLine cmd(std::vector<std::string>(
- {launcher_path, "test_config_and_seed_corpus.options"}));
- bool success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> fuzzer_args = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(2UL, fuzzer_args.size());
- EXPECT_EQ(fuzzer_args[0], "some_test_option=test_value");
- EXPECT_EQ(fuzzer_args[1], "max_len=1024");
- // Test seed_corpus archive.
- launcher_path =
- exe_path.DirName().Append("check_seed_corpus_archive.py").value();
- cmd = base::CommandLine(std::vector<std::string>(
- {launcher_path, "test_config_and_seed_corpus_seed_corpus.zip"}));
- success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> seed_corpus_info = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(1UL, seed_corpus_info.size());
- EXPECT_EQ(seed_corpus_info[0], "3");
- }
- TEST(FuzzerConfigTest, ConfigAndSeedCorpuses) {
- // Test of .options file for fuzzer with libfuzzer_options and seed corpuses.
- base::FilePath exe_path;
- base::PathService::Get(base::FILE_EXE, &exe_path);
- std::string launcher_path =
- exe_path.DirName().Append("check_fuzzer_config.py").value();
- std::string output;
- base::CommandLine cmd(std::vector<std::string>(
- {launcher_path, "test_config_and_seed_corpuses.options"}));
- bool success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> fuzzer_args = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(2UL, fuzzer_args.size());
- EXPECT_EQ(fuzzer_args[0], "some_test_option=another_test_value");
- EXPECT_EQ(fuzzer_args[1], "max_len=1337");
- // Test seed_corpus archive.
- launcher_path =
- exe_path.DirName().Append("check_seed_corpus_archive.py").value();
- cmd = base::CommandLine(std::vector<std::string>(
- {launcher_path, "test_config_and_seed_corpuses_seed_corpus.zip"}));
- success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> seed_corpus_info = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(1UL, seed_corpus_info.size());
- EXPECT_EQ(seed_corpus_info[0], "5");
- }
- TEST(FuzzerConfigTest, DictSubdir) {
- // Test of auto-generated .options file for fuzzer with dict in sub-directory.
- base::FilePath exe_path;
- base::PathService::Get(base::FILE_EXE, &exe_path);
- std::string launcher_path =
- exe_path.DirName().Append("check_fuzzer_config.py").value();
- std::string output;
- base::CommandLine cmd(std::vector<std::string>(
- {launcher_path, "test_dict_from_subdir.options"}));
- bool success = base::GetAppOutputAndError(cmd, &output);
- EXPECT_TRUE(success);
- std::vector<std::string> fuzzer_args = base::SplitString(
- output, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
- EXPECT_EQ(1UL, fuzzer_args.size());
- EXPECT_EQ(fuzzer_args[0], "dict=test_dict_from_subdir.dict");
- }
|