ssg.build_cpe

Complexity

Total Complexity

12

Size/Duplication

Total Lines	57
Duplicated Lines	0 %

Test Coverage

Coverage

0%

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	extract_env_obj()	0	10	3
A	extract_subelement()	0	15	4
A	extract_referred_nodes()	0	19	5

"""
Common functions for building CPEs
"""

from __future__ import absolute_import
from __future__ import print_function


def extract_subelement(objects, sub_elem_type):
    """
    From a collection of element objects, return the value of
    the first attribute of name sub_elem_type found.

    This is useful when the object is a single element and
    we wish to query some external reference identifier
    in the subtree of that element.
    """

    for obj in objects:
        for subelement in obj.getiterator():
            if subelement.get(sub_elem_type):
                sub_element = subelement.get(sub_elem_type)
                return sub_element


def extract_env_obj(objects, local_var):
    """
    From a collection of objects, return the object with id matching
    the object_ref of the local variable.
    """

    for obj in objects:
        env_id = extract_subelement(local_var, 'object_ref')
        if env_id == obj.get('id'):
            return obj


def extract_referred_nodes(tree_with_refs, tree_with_ids, attrname):
    """
    Return the elements in tree_with_ids which are referenced
    from tree_with_refs via the element attribute 'attrname'.
    """

    reflist = []
    elementlist = []

    for element in tree_with_refs.getiterator():
        value = element.get(attrname)
        if value is not None:
            reflist.append(value)

    for element in tree_with_ids.getiterator():
        if element.get("id") in reflist:
            elementlist.append(element)

    return elementlist