コード例 #1
0
 def setUp(self):
     self.temp_input_dir = os.path.join(self.TEMP_DIR, 'afl_input_dir')
     super(AflFuzzInputDirectoryTest, self).setUp()
     test_helpers.patch(self,
                        ['bot.fuzzers.engine_common.is_lpm_fuzz_target'])
     self.mock.is_lpm_fuzz_target.return_value = True
     self.strategies = launcher.FuzzingStrategies(None)
コード例 #2
0
  def setUp(self):
    test_utils.set_up_pyfakefs(self)
    for file_num in range(self.NUM_FILES):
      self.fs.create_file(os.path.join(self.INPUT_DIR, str(file_num)))

    test_helpers.patch(self, ['bot.fuzzers.engine_common.is_lpm_fuzz_target'])
    self.mock.is_lpm_fuzz_target.return_value = True
    self.strategies = launcher.FuzzingStrategies(None)
コード例 #3
0
    def prepare(self, corpus_dir, target_path, build_dir):  # pylint: disable=unused-argument
        """Prepare for a fuzzing session, by generating options. Returns a
    FuzzOptions object.

    Args:
      corpus_dir: The main corpus directory.
      target_path: Path to the target.
      build_dir: Path to the build directory.

    Returns:
      A FuzzOptions object.
    """
        afl_config = launcher.AFLConfig.from_target_path(target_path)
        arguments = afl_config.additional_afl_arguments
        # TODO(mbarbella): Select all strategies here instead of deferring to fuzz.
        strategies = launcher.FuzzingStrategies(target_path).to_strategy_dict()
        return engine.FuzzOptions(corpus_dir, arguments, strategies)
コード例 #4
0
    def setUp(self):
        def get_data_path(filename, is_in_output=True):
            """Returns absolute path of data files used in unittests."""
            input_or_output_dir = 'output' if is_in_output else 'input'
            return os.path.join(self.data_dir, input_or_output_dir, filename)

        self.data_dir = os.path.join(
            os.path.abspath(os.path.dirname(__file__)), 'stats_data')
        self.fuzzer_stats_path = get_data_path('fuzzer_stats')
        self.fuzzer_stats_invalid_path = get_data_path('fuzzer_stats_invalid')
        self.dict_path = get_data_path('sample.dict', False)

        self.stats_getter = stats.StatsGetter(self.fuzzer_stats_path,
                                              self.dict_path)

        self.stats_getter.set_afl_stats()
        dont_use_strategies(self)
        self.strategies = launcher.FuzzingStrategies()

        self.maxDiff = None  # pylint: disable=invalid-name
コード例 #5
0
 def setUp(self):
     self.strategies = launcher.FuzzingStrategies()
コード例 #6
0
 def setUp(self):
     self.temp_input_dir = os.path.join(self.TEMP_DIR, 'afl_input_dir')
     super(AflFuzzInputDirectoryTest, self).setUp()
     self.strategies = launcher.FuzzingStrategies()
コード例 #7
0
 def setUp(self):
   test_helpers.patch(self, ['bot.fuzzers.engine_common.is_lpm_fuzz_target'])
   self.mock.is_lpm_fuzz_target.return_value = True
   self.strategies = launcher.FuzzingStrategies(None)