Source code for pytw.core_vuln

import datetime
import json
import copy

import constants as Constants
import rating
import product
import remediation
import patch
import exploit

[docs]class CoreVuln(object): """ Core Vulnerability object. :param vuln_json: Vulnerability JSON in canonical form """ def __init__(self, vuln_json): self.__vuln_json = vuln_json self.__id = self.__vuln_json[Constants.VULN_ID] self.__title = self.__vuln_json[Constants.VULN_TITLE] if self.__vuln_json.get(Constants.VULN_TITLE) is not None else None self.__summary = self.__vuln_json[Constants.VULN_SUMMARY] if self.__vuln_json.get(Constants.VULN_SUMMARY) is not None else None self.__vuln_types = self.__vuln_json[Constants.VULN_VULNERABILITY_TYPES] if self.__vuln_json.get(Constants.VULN_VULNERABILITY_TYPES) is not None else None self.__cvss_score = self.__vuln_json[Constants.VULN_CVSS_SCORE] if self.__vuln_json.get(Constants.VULN_CVSS_SCORE) is not None else None self.__cvss_vector = self.__vuln_json[Constants.VULN_CVSS_VECTOR] if self.__vuln_json.get(Constants.VULN_CVSS_VECTOR) is not None else None self.__rating = rating.Rating(int(self.__vuln_json[Constants.RATING])) if self.__vuln_json.get(Constants.RATING) is not None else rating.Rating.Unknown self.__publisher = self.__vuln_json[Constants.VULN_PUBLISHER] if self.__vuln_json.get(Constants.VULN_PUBLISHER) is not None else None self.__published_datetime = datetime.datetime.strptime(self.__vuln_json[Constants.VULN_PUBLISHED_DATETIME], "%Y-%m-%d %H:%M:%S") if self.__vuln_json.get(Constants.VULN_PUBLISHED_DATETIME) is not None else None self.__last_modified_datetime = datetime.datetime.strptime(self.__vuln_json[Constants.VULN_LAST_MODIFIED_DATETIME], "%Y-%m-%d %H:%M:%S") if self.__vuln_json.get(Constants.VULN_LAST_MODIFIED_DATETIME) is not None else None self.__last_change = self.__vuln_json[Constants.VULN_LAST_CHANGE] if self.__vuln_json.get(Constants.VULN_LAST_CHANGE) is not None else None self.__references = self.__vuln_json[Constants.VULN_REFERENCES] if self.__vuln_json.get(Constants.VULN_REFERENCES) is not None else None self.__products = product.json2products(self.__vuln_json[Constants.VULN_PRODUCTS]) if self.__vuln_json.get(Constants.VULN_PRODUCTS) is not None else None self.__exploits = exploit.json2exploits(self.__vuln_json[Constants.VULN_EXPLOITS]) if self.__vuln_json.get(Constants.VULN_EXPLOITS) is not None else None self.__remediations = remediation.json2remediations(self.__vuln_json[Constants.VULN_REMEDIATIONS]) if self.__vuln_json.get(Constants.VULN_REMEDIATIONS) is not None else None self.__patches = patch.json2patches(self.__vuln_json[Constants.VULN_PATCHES]) if self.__vuln_json.get(Constants.VULN_PATCHES) is not None else None
[docs] def get_id(self): """ :Returns a string containing the vulnerability ID """ return self.__id
[docs] def get_title(self): """ :Returns a string containing the vulnerability title """ return self.__title
[docs] def get_summary(self): """ :Returns a string containing the vulnerability summary """ return self.__summary
[docs] def get_vuln_types(self): """ :Returns an array containing strings of vulnerability types """ return self.__vuln_types
[docs] def get_cvss_score(self): """ :Returns the CVSS score """ return self.__cvss_score
[docs] def get_cvss_vector(self): """ :Returns the CVSS vector """ return self.__cvss_vector
[docs] def get_rating(self): """ :Returns the rating as VulnRating enum """ return self.__rating
[docs] def get_rating_as_int(self): """ :Returns the rating as an integer """ return self.__rating.value
[docs] def get_rating_as_str(self): """ :Returns the rating as a string for display purposes """ return self.__rating.name
[docs] def get_publisher(self): """ :Returns the publisher as a string """ return self.__publisher
[docs] def get_published_datetime(self): """ :Returns the published date as datetime """ return self.__published_datetime
[docs] def get_last_modified_datetime(self): """ :Returns the last modified date as datetime """ return self.__last_modified_datetime
[docs] def get_last_change(self): """ :Returns the list of last changed fields """ return self.__last_change
[docs] def get_references(self): """ :Returns references as an array of URL strings """ return self.__references
[docs] def get_products(self): """ :Returns the list of products """ return self.__products
[docs] def get_exploits(self): """ :Returns the list of exploits """ return self.__exploits
[docs] def get_remediations(self, publisher=None): """ :Returns the list of remediations """ if (publisher is None): return self.__remediations else: if (publisher == self.__publisher): return self.__remediations else: return []
[docs] def get_patches(self, publisher=None): """ :Returns the list of patches """ if (publisher is None): return self.__patches else: if (publisher == self.__publisher): return self.__patches else: return []
[docs] def is_new(self): """ :Returns True if this vulnerability is newly discovered False if it has been updated atleast once """ if 'new' in self.__last_change: return True return False
[docs] def to_json(self): """ :Returns JSON representation of the object """ return copy.deepcopy(self.__vuln_json)
def __str__(self): return json.dumps(self.__vuln_json)