コード例 #1
0
ファイル: runner.py プロジェクト: tsmithv11/checkov
def get_files_definitions(files: List[str], filepath_fn=None) \
        -> Tuple[Dict[str, DictNode], Dict[str, List[Tuple[int, str]]]]:
    results = parallel_runner.run_function(lambda f: (f, parse(f)), files)
    definitions = {}
    definitions_raw = {}
    for file, result in results:
        path = filepath_fn(file) if filepath_fn else file
        definitions[path], definitions_raw[path] = result

    return definitions, definitions_raw
コード例 #2
0
ファイル: runner.py プロジェクト: sammadu1/checkov
    def run(self,
            root_folder,
            external_checks_dir=None,
            files=None,
            runner_filter=RunnerFilter()):
        report = Report(self.check_type)
        definitions = {}
        definitions_raw = {}
        parsing_errors = {}
        files_list = []
        if external_checks_dir:
            for directory in external_checks_dir:
                arm_registry.load_external_checks(directory)

        if files:
            for file in files:
                (definitions[file], definitions_raw[file]) = parse(file)

        if root_folder:
            for root, d_names, f_names in os.walk(root_folder):
                filter_ignored_directories(d_names)
                for file in f_names:
                    file_ending = os.path.splitext(file)[1]
                    if file_ending in ARM_POSSIBLE_ENDINGS:
                        files_list.append(os.path.join(root, file))

            for file in files_list:
                relative_file_path = f'/{os.path.relpath(file, os.path.commonprefix((root_folder, file)))}'
                (definitions[relative_file_path],
                 definitions_raw[relative_file_path]) = parse(file)

        # Filter out empty files that have not been parsed successfully, and filter out non-CF template files
        definitions = {
            k: v
            for k, v in definitions.items()
            if v and v.__contains__("resources")
        }
        definitions_raw = {
            k: v
            for k, v in definitions_raw.items() if k in definitions.keys()
        }

        for arm_file in definitions.keys():
            if isinstance(
                    definitions[arm_file],
                    dict_node) and 'resources' in definitions[arm_file].keys():
                arm_context_parser = ContextParser(arm_file,
                                                   definitions[arm_file],
                                                   definitions_raw[arm_file])
                logging.debug("Template Dump for {}: {}".format(
                    arm_file, definitions[arm_file], indent=2))
                arm_context_parser.evaluate_default_parameters()
                for resource in definitions[arm_file]['resources']:
                    resource_id = arm_context_parser.extract_arm_resource_id(
                        resource)
                    resource_name = arm_context_parser.extract_arm_resource_name(
                        resource)
                    entity_lines_range, entity_code_lines = arm_context_parser.extract_arm_resource_code_lines(
                        resource)
                    if entity_lines_range and entity_code_lines:
                        # TODO - Variable Eval Message!
                        variable_evaluations = {}

                        skipped_checks = ContextParser.collect_skip_comments(
                            resource)

                        results = arm_registry.scan(arm_file,
                                                    {resource_name: resource},
                                                    skipped_checks,
                                                    runner_filter)
                        for check, check_result in results.items():
                            record = Record(
                                check_id=check.id,
                                check_name=check.name,
                                check_result=check_result,
                                code_block=entity_code_lines,
                                file_path=arm_file,
                                file_line_range=entity_lines_range,
                                resource=resource_id,
                                evaluations=variable_evaluations,
                                check_class=check.__class__.__module__)
                            report.add_record(record=record)
        return report
コード例 #3
0
ファイル: runner.py プロジェクト: xB-2048/checkov
    def run(self,
            root_folder,
            external_checks_dir=None,
            files=None,
            runner_filter=RunnerFilter(),
            collect_skip_comments=True):
        report = Report(self.check_type)
        definitions = {}
        definitions_raw = {}
        parsing_errors = {}
        files_list = []
        if external_checks_dir:
            for directory in external_checks_dir:
                arm_registry.load_external_checks(directory, runner_filter)

        if files:
            for file in files:
                (definitions[file], definitions_raw[file]) = parse(file)

        if root_folder:
            for root, d_names, f_names in os.walk(root_folder):
                filter_ignored_directories(d_names)
                for file in f_names:
                    file_ending = os.path.splitext(file)[1]
                    if file_ending in ARM_POSSIBLE_ENDINGS:
                        files_list.append(os.path.join(root, file))

            for file in files_list:
                relative_file_path = f'/{os.path.relpath(file, os.path.commonprefix((root_folder, file)))}'
                (definitions[relative_file_path],
                 definitions_raw[relative_file_path]) = parse(file)

        # Filter out empty files that have not been parsed successfully, and filter out non-CF template files
        definitions = {
            k: v
            for k, v in definitions.items()
            if v and v.__contains__("resources")
        }
        definitions_raw = {
            k: v
            for k, v in definitions_raw.items() if k in definitions.keys()
        }

        for arm_file in definitions.keys():

            # There are a few cases here. If -f was used, there could be a leading / because it's an absolute path,
            # or there will be no leading slash; root_folder will always be none.
            # If -d is used, root_folder will be the value given, and -f will start with a / (hardcoded above).
            # The goal here is simply to get a valid path to the file (which arm_file does not always give).
            if arm_file[0] == '/':
                path_to_convert = (root_folder +
                                   arm_file) if root_folder else arm_file
            else:
                path_to_convert = (os.path.join(
                    root_folder, arm_file)) if root_folder else arm_file

            file_abs_path = os.path.abspath(path_to_convert)

            if isinstance(
                    definitions[arm_file],
                    dict_node) and 'resources' in definitions[arm_file].keys():
                arm_context_parser = ContextParser(arm_file,
                                                   definitions[arm_file],
                                                   definitions_raw[arm_file])
                logging.debug("Template Dump for {}: {}".format(
                    arm_file, definitions[arm_file], indent=2))
                arm_context_parser.evaluate_default_parameters()

                # Split out nested resources from base resource
                for resource in definitions[arm_file]['resources']:
                    if "parent_name" in resource.keys():
                        continue
                    nested_resources = []
                    nested_resources = arm_context_parser.search_deep_keys(
                        "resources", resource, [])
                    if nested_resources:
                        for nr in nested_resources:
                            nr_element = nr.pop()
                            if nr_element:
                                for element in nr_element:
                                    new_resource = {}
                                    new_resource = element
                                    if isinstance(new_resource, dict):
                                        new_resource["parent_name"] = resource[
                                            "name"]
                                        new_resource["parent_type"] = resource[
                                            "type"]
                                        definitions[arm_file][
                                            'resources'].append(new_resource)

                for resource in definitions[arm_file]['resources']:
                    resource_id = arm_context_parser.extract_arm_resource_id(
                        resource)
                    resource_name = arm_context_parser.extract_arm_resource_name(
                        resource)
                    entity_lines_range, entity_code_lines = arm_context_parser.extract_arm_resource_code_lines(
                        resource)
                    if entity_lines_range and entity_code_lines:
                        # TODO - Variable Eval Message!
                        variable_evaluations = {}

                        skipped_checks = ContextParser.collect_skip_comments(
                            resource)

                        results = arm_registry.scan(arm_file,
                                                    {resource_name: resource},
                                                    skipped_checks,
                                                    runner_filter)
                        for check, check_result in results.items():
                            record = Record(
                                check_id=check.id,
                                check_name=check.name,
                                check_result=check_result,
                                code_block=entity_code_lines,
                                file_path=arm_file,
                                file_line_range=entity_lines_range,
                                resource=resource_id,
                                evaluations=variable_evaluations,
                                check_class=check.__class__.__module__,
                                file_abs_path=file_abs_path)
                            report.add_record(record=record)
        return report