import json from datetime import datetime, timedelta from jira import JIRA import requests from bottle import template import re class JiraAPI(object): #NamedLogger): __logname__="vjira" #TODO implement logging def __init__(self, hostname=None, username=None, password=None, debug=False, clean_obsolete=True, max_time_window=6): #self.setup_logger(debug=debug) if "https://" not in hostname: hostname = "https://{}".format(hostname) self.username = username self.password = password self.jira = JIRA(options={'server': hostname}, basic_auth=(self.username, self.password)) #self.logger.info("Created vjira service for {}".format(server)) self.all_tickets = [] self.JIRA_REOPEN_ISSUE = "Reopen Issue" self.JIRA_CLOSE_ISSUE = "Close Issue" self.max_time_tracking = max_time_window #in months # self.JIRA_RESOLUTION_OBSOLETE = "Obsolete" self.JIRA_RESOLUTION_FIXED = "Fixed" self.clean_obsolete = clean_obsolete self.template_path = 'vulnwhisp/reporting/resources/ticket.tpl' def create_ticket(self, title, desc, project="IS", components=[], tags=[]): labels = ['vulnerability_management'] for tag in tags: labels.append(str(tag)) #self.logger.info("creating ticket for project {} title[20] {}".format(project, title[:20])) #self.logger.info("project {} has a component requirement: {}".format(project, self.PROJECT_COMPONENT_TABLE[project])) project_obj = self.jira.project(project) components_ticket = [] for component in components: exists = False for c in project_obj.components: if component == c.name: #self.logger.debug("resolved component name {} to id {}".format(component_name, c.id)ra python) components_ticket.append({ "id": c.id }) exists=True if not exists: print "[ERROR] Error creating Ticket: component {} not found".format(component) return 0 new_issue = self.jira.create_issue(project=project, summary=title, description=desc, issuetype={'name': 'Bug'}, labels=labels, components=components_ticket) print "[SUCCESS] Ticket {} has been created".format(new_issue) return new_issue #Basic JIRA Metrics def metrics_open_tickets(self, project=None): jql = "labels= vulnerability_management and resolution = Unresolved" if project: jql += " and (project='{}')".format(project) print jql return len(self.jira.search_issues(jql, maxResults=0)) def metrics_closed_tickets(self, project=None): jql = "labels= vulnerability_management and NOT resolution = Unresolved" if project: jql += " and (project='{}')".format(project) return len(self.jira.search_issues(jql, maxResults=0)) def sync(self, vulnerabilities, project, components=[]): #JIRA structure of each vulnerability: [source, scan_name, title, diagnosis, consequence, solution, ips, risk, references] print "JIRA Sync started" # [HIGIENE] close tickets older than 6 months as obsolete # Higiene clean up affects to all tickets created by the module, filters by label 'vulnerability_management' if self.clean_obsolete: self.close_obsolete_tickets() for vuln in vulnerabilities: # JIRA doesn't allow labels with spaces, so making sure that the scan_name doesn't have spaces # if it has, they will be replaced by "_" if " " in vuln['scan_name']: vuln['scan_name'] = "_".join(vuln['scan_name'].split(" ")) exists = False to_update = False ticketid = "" ticket_assets = [] exists, to_update, ticketid, ticket_assets = self.check_vuln_already_exists(vuln) if exists: # If ticket "resolved" -> reopen, as vulnerability is still existent self.reopen_ticket(ticketid) continue elif to_update: self.ticket_update_assets(vuln, ticketid, ticket_assets) continue try: tpl = template(self.template_path, vuln) except Exception as e: print e return 0 self.create_ticket(title=vuln['title'], desc=tpl, project=project, components=components, tags=[vuln['source'], vuln['scan_name'], 'vulnerability', vuln['risk']]) self.close_fixed_tickets(vulnerabilities) # we reinitialize so the next sync redoes the query with their specific variables self.all_tickets = [] return True def check_vuln_already_exists(self, vuln): # we need to return if the vulnerability has already been reported and the ID of the ticket for further processing #function returns array [duplicated(bool), update(bool), ticketid, ticket_assets] title = vuln['title'] labels = [vuln['source'], vuln['scan_name'], 'vulnerability_management', 'vulnerability'] #list(set()) to remove duplicates assets = list(set(re.findall(r"\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b", ",".join(vuln['ips'])))) if not self.all_tickets: print "Retrieving all JIRA tickets with the following tags {}".format(labels) # we want to check all JIRA tickets, to include tickets moved to other queues # will exclude tickets older than 6 months, old tickets will get closed for higiene and recreated if still vulnerable jql = "{} AND NOT labels=advisory AND created >=startOfMonth(-{})".format(" AND ".join(["labels={}".format(label) for label in labels]), self.max_time_tracking) self.all_tickets = self.jira.search_issues(jql, maxResults=0) #WARNING: function IGNORES DUPLICATES, after finding a "duplicate" will just return it exists #it wont iterate over the rest of tickets looking for other possible duplicates/similar issues print "Comparing Vulnerabilities to created tickets" for index in range(len(self.all_tickets)-1): checking_ticketid, checking_title, checking_assets = self.ticket_get_unique_fields(self.all_tickets[index]) if title == checking_title: difference = list(set(assets).symmetric_difference(checking_assets)) #to check intersection - set(assets) & set(checking_assets) if difference: print "Asset mismatch, ticket to update. TickedID: {}".format(checking_ticketid) return False, True, checking_ticketid, checking_assets #this will automatically validate else: print "Confirmed duplicated. TickedID: {}".format(checking_ticketid) return True, False, checking_ticketid, [] #this will automatically validate return False, False, "", [] def ticket_get_unique_fields(self, ticket): title = ticket.raw.get('fields', {}).get('summary').encode("ascii").strip() ticketid = ticket.key.encode("ascii") try: affected_assets_section = ticket.raw.get('fields', {}).get('description').encode("ascii").split("{panel:title=Affected Assets}")[1].split("{panel}")[0] assets = list(set(re.findall(r"\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b", affected_assets_section))) except: print "[ERROR] Ticket IPs regex failed. Ticket ID: {}".format(ticketid) assets = [] return ticketid, title, assets def ticket_update_assets(self, vuln, ticketid, ticket_assets): # correct description will always be in the vulnerability to report, only needed to update description to new one print "Ticket {} exists, UPDATE requested".format(ticketid) if self.is_ticket_resolved(self.jira.issue(ticketid)): self.reopen_ticket(ticketid) try: tpl = template(self.template_path, vuln) except Exception as e: print e return 0 ticket_obj = self.jira.issue(ticketid) ticket_obj.update() assets = list(set(re.findall(r"\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b", ",".join(vuln['ips'])))) difference = list(set(assets).symmetric_difference(ticket_assets)) comment = '' #put a comment with the assets that have been added/removed for asset in difference: if asset in assets: comment += "Asset {} have been added to the ticket as vulnerability *has been newly detected*.\n".format(asset) elif asset in ticket_assets: comment += "Asset {} have been removed from the ticket as vulnerability *has been resolved*.\n".format(asset) ticket_obj.fields.labels.append('updated') try: ticket_obj.update(description=tpl, comment=comment, fields={"labels":ticket_obj.fields.labels}) print "Ticket {} updated successfully".format(ticketid) except: print "[ERROR] Error while trying up update ticket {}".format(ticketid) return 0 def close_fixed_tickets(self, vulnerabilities): # close tickets which vulnerabilities have been resolved and are still open found_vulns = [] for vuln in vulnerabilities: found_vulns.append(vuln['title']) comment = '''This ticket is being closed as it appears that the vulnerability no longer exists. If the vulnerability reappears, a new ticket will be opened.''' for ticket in self.all_tickets: if ticket.raw['fields']['summary'].strip() in found_vulns: print "Ticket {} is still vulnerable".format(ticket) continue print "Ticket {} is no longer vulnerable".format(ticket) self.close_ticket(ticket, self.JIRA_RESOLUTION_FIXED, comment) return 0 def is_ticket_reopenable(self, ticket_obj): transitions = self.jira.transitions(ticket_obj) for transition in transitions: if transition.get('name') == self.JIRA_REOPEN_ISSUE: #print "ticket is reopenable" return True print "[ERROR] Ticket can't be opened. Check Jira transitions." return False def is_ticket_closeable(self, ticket_obj): transitions = self.jira.transitions(ticket_obj) for transition in transitions: if transition.get('name') == self.JIRA_CLOSE_ISSUE: return True print "[ERROR] Ticket can't closed. Check Jira transitions." return False def is_ticket_resolved(self, ticket_obj): #Checks if a ticket is resolved or not if ticket_obj is not None: if ticket_obj.raw['fields'].get('resolution') is not None: if ticket_obj.raw['fields'].get('resolution').get('name') != 'Unresolved': print "Checked ticket {} is already closed".format(ticket_obj) #logger.info("ticket {} is closed".format(ticketid)) return True print "Checked ticket {} is already open".format(ticket_obj) return False def is_risk_accepted(self, ticket_obj): if ticket_obj is not None: if ticket_obj.raw['fields'].get('labels') is not None: labels = ticket_obj.raw['fields'].get('labels') print labels if "risk_accepted" in labels: print "Ticket {} accepted risk, will be ignored".format(ticket_obj) return True elif "server_decomission" in labels: print "Ticket {} server decomissioned, will be ignored".format(ticket_obj) return True print "Ticket {} risk has not been accepted".format(ticket_obj) return False def reopen_ticket(self, ticketid): print "Ticket {} exists, REOPEN requested".format(ticketid) # this will reopen a ticket by ticketid ticket_obj = self.jira.issue(ticketid) if self.is_ticket_resolved(ticket_obj): #print "ticket is resolved" if not self.is_risk_accepted(ticket_obj): try: if self.is_ticket_reopenable(ticket_obj): comment = '''This ticket has been reopened due to the vulnerability not having been fixed (if multiple assets are affected, all need to be fixed; if the server is down, lastest known vulnerability might be the one reported). In the case of the team accepting the risk and wanting to close the ticket, please add the label "*risk_accepted*" to the ticket before closing it. If server has been decomissioned, please add the label "*server_decomission*" to the ticket before closing it. If you have further doubts, please contact the Security Team.''' error = self.jira.transition_issue(issue=ticketid, transition=self.JIRA_REOPEN_ISSUE, comment = comment) print "[SUCCESS] ticket {} reopened successfully".format(ticketid) #logger.info("ticket {} reopened successfully".format(ticketid)) return 1 except Exception as e: # continue with ticket data so that a new ticket is created in place of the "lost" one print "[ERROR] error reopening ticket {}: {}".format(ticketid, e) #logger.error("error reopening ticket {}: {}".format(ticketid, e)) return 0 return 0 def close_ticket(self, ticketid, resolution, comment): # this will close a ticket by ticketid print "Ticket {} exists, CLOSE requested".format(ticketid) ticket_obj = self.jira.issue(ticketid) if not self.is_ticket_resolved(ticket_obj): try: if self.is_ticket_closeable(ticket_obj): error = self.jira.transition_issue(issue=ticketid, transition=self.JIRA_CLOSE_ISSUE, comment = comment, resolution = {"name": resolution }) print "[SUCCESS] ticket {} closed successfully".format(ticketid) #logger.info("ticket {} reopened successfully".format(ticketid)) return 1 except Exception as e: # continue with ticket data so that a new ticket is created in place of the "lost" one print "[ERROR] error closing ticket {}: {}".format(ticketid, e) #logger.error("error closing ticket {}: {}".format(ticketid, e)) return 0 return 0 def close_obsolete_tickets(self): # Close tickets older than 6 months, vulnerabilities not solved will get created a new ticket print "Closing obsolete tickets older than {} months".format(self.max_time_tracking) jql = "labels=vulnerability_management AND created