farol/farolluz: farolluz/cvrf.py annotate

annotate farolluz/cvrf.py @ 15:dcc946b30343

Consolidate productTree edition

author	Benoît Allard <benoit.allard@greenbone.net>
date	Thu, 09 Oct 2014 14:21:07 +0200
parents	640b88744523
children	90852c11fabd

rev	line source
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1 # -- coding: utf-8 --
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	2 #
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	3 # Authors:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	4 # Benoît Allard <benoit.allard@greenbone.net>
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	5 #
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	6 # Copyright:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	7 # Copyright (C) 2014 Greenbone Networks GmbH
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	8 #
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	9 # This program is free software; you can redistribute it and/or
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	10 # modify it under the terms of the GNU General Public License
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	11 # as published by the Free Software Foundation; either version 2
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	12 # of the License, or (at your option) any later version.
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	13 #
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	14 # This program is distributed in the hope that it will be useful,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	17 # GNU General Public License for more details.
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	18 #
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	19 # You should have received a copy of the GNU General Public License
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	20 # along with this program; if not, write to the Free Software
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	21 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	22
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	23 """\
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	24 Objects related to CVRF Documents
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	25 """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	26
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	27 class ValidationError(Exception): pass
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	28
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	29 class CVRFPublisher(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	30 TYPES = ('Vendor', 'Discoverer', 'Coordinator', 'User', 'Other')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	31 def __init__(self, _type, vendorid=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	32 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	33 self._vendorid = vendorid
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	34 self._contact = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	35 self._authority = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	36
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	37 def setContact(self, contact):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	38 self._contact = contact
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	39
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	40 def setAuthority(self, authority):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	41 self._authority = authority
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	42
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	43 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	44 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	45 raise ValidationError('Document Publisher needs to have a type')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	46 if self._type not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	47 raise ValidationError('Document Publisher Type needs to be one of %s' % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	48
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	49 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	50 s = 'CVRFPublisher: %s' % self._type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	51 if self._vendorid is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	52 s += ' ID: %s' % self._vendorid
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	53 if self._contact is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	54 s += ' Contact: "%s"' % self._contact
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	55 if self._authority is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	56 s += ' Authority: "%s"' % self._authority
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	57 return s
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	58
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	59
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	60 class CVRFTrackingID(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	61 def __init__(self, _id):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	62 self._id = _id
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	63 self._aliases = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	64
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	65 def addAlias(self, alias):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	66 self._aliases.append(alias)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	67
6 633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	68 def getId(self):
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	69 return self._id
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	70
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	71 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	72 if not self._id:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	73 raise ValidationError('Document ID cannot be left empty')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	74
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	75 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	76 if self._aliases:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	77 return "%s (%s)" % (self._id, ', '.join(self._aliases))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	78 return self._id
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	79
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	80
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	81 class CVRFTracking(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	82 STATUSES = ('Draft', 'Interim', 'Final')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	83 def __init__(self, _id, status, version, initial, current):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	84 self._identification = _id
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	85 self._status = status
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	86 self._version = version
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	87 self._history = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	88 self._initialDate = initial
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	89 self._currentDate = current
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	90 self._generator = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	91
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	92 def addRevision(self, revision):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	93 self._history.append(revision)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	94
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	95 def setGenerator(self, generator):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	96 self._generator = generator
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	97
6 633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	98 def getId(self):
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	99 return self._identification.getId()
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	100
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	101 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	102 if self._identification is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	103 raise ValidationError('Document Tracking needs to have an Identification')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	104 self._identification.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	105 if not self._status:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	106 raise ValidationError('Document status must be set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	107 if self._status not in self.STATUSES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	108 raise ValidationError('Document Status must be one of %s' % ', '.join(self.STATUSES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	109 if not self._version:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	110 raise ValidationError('Document Version must be set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	111 if len(self._version) > 4:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	112 raise ValidationError('Document Version must be comprised between `nn` and `nn.nn.nn.nn`')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	113 if not self._history:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	114 raise ValidationError('Document must have at least a revision')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	115 if not self._initialDate:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	116 raise ValidationError('Document must have an initial Release date set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	117 prev_date = self._initialDate
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	118 if self._history[0]._date < self._initialDate:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	119 # Documents could have revisions before being released
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	120 prev_date = self._history[0]._date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	121 prev = ()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	122 for revision in self._history:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	123 revision.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	124 if revision._number <= prev:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	125 raise ValidationError('Revision numbers must always be increasing')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	126 if revision._date < prev_date:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	127 raise ValidationError('Revision dates must always be increasing')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	128 prev = revision._number
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	129 prev_date = revision._date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	130 if not self._currentDate:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	131 raise ValidationError('Document must have a Current Release Date set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	132 if self._currentDate != self._history[-1]._date:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	133 raise ValidationError('Current Release Date must be the same as the Date from the last Revision')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	134 if self._initialDate > self._currentDate:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	135 raise ValidationError('Initial date must not be after current Date')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	136 if self._version != self._history[-1]._number:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	137 raise ValidationError('Document version must be the same as the number of the last Revision')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	138
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	139 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	140 s = "ID: %s" % self._identification
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	141 s += " Status: %s" % self._status
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	142 s += " v%s" % '.'.join('%d' % i for i in self._version)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	143 s += " %d revisions" % len(self._history)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	144 s += " Initial release: %s" % self._initialDate.isoformat()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	145 return s
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	146
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	147
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	148 class CVRFRevision(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	149 def __init__(self, number, date, description):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	150 self._number = number
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	151 self._date = date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	152 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	153
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	154 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	155 if not self._number:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	156 raise ValidationError('A Revision must have a Number')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	157 if not self._date:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	158 raise ValidationError('A Revision must have a Date')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	159 if not self._description:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	160 raise ValidationError('A Revision must have a Description')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	161
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	162 class CVRFGenerator(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	163 def __init__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	164 self._engine = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	165 self._date = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	166
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	167 def setEngine(self, engine):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	168 self._engine = engine
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	169
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	170 def setDate(self, date):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	171 self._date = date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	172
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	173 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	174 if (not self._engine) and (not self._date):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	175 raise ValidationError('The Generator must have at least an Engine or a Date')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	176
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	177
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	178 class CVRFNote(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	179 TYPES = ('General', 'Details', 'Description', 'Summary', 'FAQ',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	180 'Legal Disclaimer', 'Other')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	181 def __init__(self, _type, ordinal, note, title=None, audience=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	182 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	183 self._ordinal = ordinal
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	184 self._note = note
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	185 self._title = title
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	186 self._audience = audience
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	187
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	188 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	189 """ returns something that can be used as a title """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	190 if self._title is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	191 return "%s (#%d)" % (self._type, self._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	192 return "%s (%s)" % (self._title, self._type)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	193
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	194 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	195 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	196 raise ValidationError('A Note needs to have a Type set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	197 if self._type not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	198 raise ValidationError('A Note Type needs to be one of %s' % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	199 if self._ordinal < 0:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	200 raise ValidationError('A Note ordinal must be a positive integer')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	201 if not self._note:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	202 raise ValidationError('A Note must contain some text')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	203
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	204
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	205 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	206 return self._note
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	207
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	208
1 d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	209 class CVRFAggregateSeverity(object):
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	210 def __init__(self, severity):
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	211 self._severity = severity
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	212 self._namespace = None
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	213
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	214 def setNamespace(self, namespace):
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	215 self._namespace = namespace
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	216
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	217 class CVRFReference(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	218 TYPES = ('Self', 'External')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	219 def __init__(self, url, description, _type=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	220 self._url = url
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	221 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	222 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	223
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	224 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	225 if (self._type is not None) and (self._type not in self.TYPES):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	226 raise ValidationError('If a Reference type is set, it mist be one of %s' % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	227 if not self._url:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	228 raise ValidationError('A Reference must contain an URL')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	229 if not self._description:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	230 raise ValidationError('A Reference must contain a description')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	231
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	232
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	233 class CVRFAcknowledgment(object):
8 cb8b2a196f0b Allow Name and Organization in anAcknowledgment to be plural Benoît Allard <benoit.allard@greenbone.net> parents: 7 diff changeset	234 def __init__(self, names=[], organizations=[], description=None,
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	235 url=None):
8 cb8b2a196f0b Allow Name and Organization in anAcknowledgment to be plural Benoît Allard <benoit.allard@greenbone.net> parents: 7 diff changeset	236 self._names = names
cb8b2a196f0b Allow Name and Organization in anAcknowledgment to be plural Benoît Allard <benoit.allard@greenbone.net> parents: 7 diff changeset	237 self._organizations = organizations
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	238 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	239 self._url = url
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	240
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	241 def getTitle(self):
8 cb8b2a196f0b Allow Name and Organization in anAcknowledgment to be plural Benoît Allard <benoit.allard@greenbone.net> parents: 7 diff changeset	242 return "%s - %s" % (', '.join(self._names),
cb8b2a196f0b Allow Name and Organization in anAcknowledgment to be plural Benoît Allard <benoit.allard@greenbone.net> parents: 7 diff changeset	243 ', '.join(self._organizations))
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	244
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	245 def validate(self):
8 cb8b2a196f0b Allow Name and Organization in anAcknowledgment to be plural Benoît Allard <benoit.allard@greenbone.net> parents: 7 diff changeset	246 if (not self._names) and (not self._organizations) and (not self._description):
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	247 raise ValidationError('An Acknowledgment must have at least a Name, an Organization or a Description')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	248
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	249
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	250 class CVRFProductTree(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	251 def __init__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	252 # All the branches, they can be order with their `parent` attribute
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	253 self._branches = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	254 self._groups = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	255 self._relationships = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	256 self._products = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	257 self._groups = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	258
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	259 def addProduct(self, product):
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	260 """ Add to the product list """
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	261 self._products.append(product)
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	262
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	263 def addRelationship(self, rel):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	264 self._relationships.append(rel)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	265
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	266 def addGroup(self, group):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	267 self._groups.append(group)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	268
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	269 def getProductForID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	270 for product in self._products:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	271 if product._productid == productid:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	272 return product
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	273 raise KeyError(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	274
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	275 def getGroupForID(self, groupid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	276 for group in self._groups:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	277 if group._groupid == groupid:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	278 return group
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	279 raise KeyError(groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	280
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	281 def decomposeProduct(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	282 """ In case of product defined as a relationship (product X installed
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	283 on OS Y), this gives us the following tuple: (OS, product). """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	284 product = self.getProductForID(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	285 parent = product._parent
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	286 if parent is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	287 return (None, None)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	288 if not isinstance(parent, CVRFRelationship):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	289 return (None, None)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	290 relationtype = parent._relationtype.replace(' ', '').lower()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	291 if relationtype not in ('defaultcomponentof', 'installedon'):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	292 return (None, None)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	293 return (
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	294 self.getProductForID(parent._relatestoproductreference),
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	295 self.getProductForID(parent._productreference)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	296 )
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	297
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	298 def getBranch(self, path):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	299 if len(path) == 0:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	300 return self
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	301 branches = self._branches
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	302 node = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	303 for idx in path:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	304 node = branches[idx]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	305 branches = node._childs
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	306 return node
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	307
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	308 def getBranches(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	309 for branch in self._branches:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	310 yield branch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	311 for sub_branch in branch.getBranches():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	312 yield sub_branch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	313
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	314 def getPath(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	315 return ()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	316
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	317 def getNameOfRelationship(self, relationship):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	318 if relationship is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	319 return ''
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	320 return ' '.join((self.getProductForID(relationship._productreference)._name, 'as',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	321 relationship._relationtype.lower(),
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	322 self.getProductForID(relationship._relatestoproductreference)._name))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	323
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	324 def getOrphanedBranches(self, product=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	325 """ The branches that could accept `product` as Product Definition """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	326 white_list = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	327 if product is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	328 white_list = [product._parent]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	329 for branch in self.getBranches():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	330 if (branch in white_list) or branch.isOrphaned():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	331 yield branch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	332
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	333 def getNotTerminalBranches(self, b2=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	334 """\
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	335 The branches that could accept `b2` as new sub-branches
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	336 Note that b2 and all its sub-branches cannot be listed
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	337 """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	338 black_list = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	339 if b2 is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	340 black_list = [b2] + list(b2.getBranches())
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	341 for branch in self.getBranches():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	342 if branch in black_list:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	343 continue
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	344 if branch._product is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	345 yield branch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	346
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	347 def getOrphanedRelationships(self, product=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	348 """ The relationships that need a product defninition """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	349 white_list = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	350 if product is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	351 white_list = [product.getCurrentRelationship()]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	352 for i, relationship in enumerate(self._relationships):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	353 if (relationship in white_list) or relationship.isOrphaned():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	354 yield (i, relationship)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	355
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	356 def nbProducts(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	357 """ Amount of 'raw' Products """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	358 return len([p for p in self._products if p._parent is self])
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	359
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	360 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	361 for branch in self._branches:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	362 branch.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	363 productids = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	364 for product in self._products:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	365 product.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	366 if product._productid in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	367 raise ValidationError('Each ProductID must be unique (%s)' % product._productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	368 productids.add(product._productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	369 for relationship in self._relationships:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	370 relationship.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	371 for productid in (relationship._productreference,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	372 relationship._relatestoproductreference):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	373 if productid not in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	374 raise ValidationError('ProductID %s is unknown' % productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	375 groupids = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	376 for group in self._groups:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	377 group.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	378 if group._groupid in groupids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	379 raise ValidationError('Duplicated GroupID: %s' % group._groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	380 groupids.add(group._groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	381 for productid in group._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	382 if productid not in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	383 raise ValidationError('ProductID %s is unknown' % productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	384 return productids, groupids
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	385
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	386 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	387 return 'Products: %s' % '\n'.join(str(p) for p in self._products)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	388
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	389
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	390 class CVRFProductBranch(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	391 TYPES = ('Vendor', 'Product Family', 'Product Name', 'Product Version',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	392 'Patch Level', 'Service Pack', 'Architecture', 'Language',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	393 'Legacy', 'Specification')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	394 def __init__(self, _type, name, parentbranch):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	395 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	396 self._name = name
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	397 self._childs = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	398 self._product = None
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	399 self.link(parentbranch)
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	400
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	401 def getParent(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	402 return self._parentbranch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	403
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	404 def getPath(self, string=False):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	405 """ return the path to that branch element as a tuple """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	406 if self.isRoot():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	407 for i, b in enumerate(self._parentbranch._branches):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	408 if b is self:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	409 if string:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	410 return '%d' % i
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	411 return (i, )
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	412 else:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	413 for i, b in enumerate(self._parentbranch._childs):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	414 if b is self:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	415 if string:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	416 return '/'.join([self._parentbranch.getPath(string), '%d' % i])
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	417 return self._parentbranch.getPath(string) + (i,)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	418 if string:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	419 return ''
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	420 return ()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	421
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	422 def getTree(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	423 """ this returns a list of tuples (type, name) leading to here"""
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	424 if self.isRoot():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	425 return [(self._type, self._name)]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	426 return self._parentbranch.getTree() + [(self._type, self._name)]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	427
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	428 def getName(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	429 return ' / '.join("%s: %s" % (type_, name) for type_, name in self.getTree())
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	430
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	431 def getParentPath(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	432 """ return as string the path to the parent """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	433 return '/'.join('%s' % p for p in self.getPath()[:-1])
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	434
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	435 def isRoot(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	436 return isinstance(self._parentbranch, CVRFProductTree)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	437
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	438 def isOrphaned(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	439 """ Has no childs and no product """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	440 return len(self._childs) == 0 and (self._product is None)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	441
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	442 def getBranches(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	443 for branch in self._childs:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	444 yield branch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	445 for sub_branch in branch.getBranches():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	446 yield sub_branch
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	447
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	448 def unlink(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	449 """ Unset our _parent, and remove us from the _parent._childs """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	450 if self.isRoot():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	451 self.getParent()._branches.remove(self)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	452 else:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	453 self.getParent()._childs.remove(self)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	454 self._parentbranch = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	455
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	456 def link(self, parent):
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	457 """ Actually, only set the parent """
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	458 self._parentbranch = parent
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	459 if self.isRoot():
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	460 parent._branches.append(self)
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	461 else:
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	462 parent._childs.append(self)
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	463
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	464
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	465 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	466 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	467 raise ValidationError('A Branch must have a Type')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	468 if self._type not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	469 raise ValidationError('A Branch Type must be one of %s' % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	470 if not self._name:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	471 raise ValidationError('A Branch must have a Name')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	472 for branch in self._childs:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	473 branch.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	474 if self.isOrphaned():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	475 raise ValidationError('A Branch must have at least a sub-product or sub-branches')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	476
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	477 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	478 return "%s: %s" % (self._type, self._name)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	479
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	480
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	481 class CVRFFullProductName(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	482 def __init__(self, productid, name, parent, cpe=None):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	483 self._productid = productid
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	484 self._name = name
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	485 self._cpe = cpe
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	486 # Can be None (directly under the tree), a ProductBranch, or a
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	487 # Relationship
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	488 self.link(parent)
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	489
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	490 def isRoot(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	491 return isinstance(self._parent, CVRFProductTree)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	492
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	493 def isRelationship(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	494 return isinstance(self._parent, CVRFRelationship)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	495
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	496 def getTree(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	497 if not isinstance(self._parent, CVRFProductBranch):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	498 return []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	499 return self._parent.getTree()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	500
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	501 def getParentPath(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	502 if self.isRoot() or self.isRelationship():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	503 return ''
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	504 return self._parent.getPath(True)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	505
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	506 def getCurrentRelationship(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	507 if self.isRelationship():
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	508 return self._parent
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	509 return None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	510
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	511 def unlink(self):
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	512 """ Unset our _parent, and remove us from the _parent._childs
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	513 We are still in the product list.
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	514 """
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	515 if not self.isRoot():
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	516 self._parent._product = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	517 self._parent = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	518
15 dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	519 def link(self, parent):
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	520 self._parent = parent
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	521 if not self.isRoot():
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	522 parent._product = self
dcc946b30343 Consolidate productTree edition Benoît Allard <benoit.allard@greenbone.net> parents: 14 diff changeset	523
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	524 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	525 if not self._productid:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	526 raise ValidationError('A Product must have a ProductID')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	527 if not self._name:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	528 raise ValidationError('A Product must have a Name')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	529
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	530 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	531 return "%s (%s)" % (self._productid, self._name)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	532
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	533
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	534 class CVRFRelationship(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	535 TYPES = ('Default Component Of', 'Optional Component Of',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	536 'External Component Of', 'Installed On', 'Installed With')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	537 def __init__(self, productref, reltype, relatestoproductref):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	538 self._productreference = productref
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	539 self._relationtype = reltype
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	540 self._relatestoproductreference = relatestoproductref
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	541 self._product = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	542
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	543 def getParent(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	544 """ All parent element of a FullProductName should implement that
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	545 method """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	546 return None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	547
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	548 def isOrphaned(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	549 return self._product is None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	550
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	551 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	552 if not self._productreference:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	553 raise ValidationError('A Relationship must have a Product Reference')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	554 if not self._relationtype:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	555 raise ValidationError('A Relationship must have a Relation Type')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	556 if self._relationtype not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	557 raise ValidationError('Relation Type must be one of %s' % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	558 if not self._relatestoproductreference:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	559 raise ValidationError('A Relationship must have a "Relates To product Reference"')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	560 if self._productreference == self._relatestoproductreference:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	561 raise ValidationError('A Relationship cannot reference twice the same Product')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	562
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	563
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	564 class CVRFGroup(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	565 def __init__(self, groupid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	566 self._groupid = groupid
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	567 self._description = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	568 self._productids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	569
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	570 def setDescription(self, description):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	571 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	572
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	573 def addProductID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	574 self._productids.append(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	575
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	576 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	577 if self._description:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	578 return "%s (%d products)" % (self._description, len(self._productids))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	579 return "#%s (%d products)" % (self._groupid, len(self._productids))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	580
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	581 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	582 if not self._groupid:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	583 raise ValidationError('A Group must have a GroupID')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	584 if not self._productids or len(self._productids) < 2:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	585 raise ValidationError('A Group must contain at least two products')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	586
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	587
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	588 class CVRFVulnerabilityID(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	589 def __init__(self, systemname, value):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	590 self._systemname = systemname
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	591 self._value = value
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	592
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	593 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	594 if not self._systemname:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	595 raise ValidationError('A Vulnerability ID must have a System Name')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	596 if not self._value:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	597 raise ValidationError('A Vulnerability ID must have a value')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	598
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	599
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	600 class CVRFVulnerability(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	601 def __init__(self, ordinal):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	602 self._ordinal = ordinal
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	603 self._title = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	604 self._id = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	605 self._notes = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	606 self._discoverydate = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	607 self._releasedate = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	608 self._involvements = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	609 self._cve = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	610 self._cwes = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	611 self._productstatuses = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	612 self._threats = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	613 self._cvsss = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	614 self._remediations = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	615 self._references = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	616 self._acknowledgments = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	617
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	618 def setTitle(self, title):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	619 self._title = title
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	620
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	621 def setID(self, _id):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	622 self._id = _id
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	623
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	624 def addNote(self, note):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	625 self._notes.append(note)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	626
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	627 def setDiscoveryDate(self, date):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	628 self._discoverydate = date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	629
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	630 def setReleaseDate(self, date):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	631 self._releasedate = date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	632
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	633 def addInvolvement(self, involvement):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	634 self._involvements.append(involvement)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	635
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	636 def setCVE(self, cve):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	637 self._cve = cve
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	638
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	639 def addCWE(self, cwe):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	640 self._cwes.append(cwe)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	641
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	642 def addProductStatus(self, productstatus):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	643 self._productstatuses.append(productstatus)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	644
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	645 def addThreat(self, threat):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	646 self._threats.append(threat)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	647
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	648 def addCVSSSet(self, cvss_set):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	649 self._cvsss.append(cvss_set)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	650
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	651 def addRemediation(self, remediation):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	652 self._remediations.append(remediation)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	653
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	654 def addReference(self, ref):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	655 self._references.append(ref)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	656
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	657 def addAcknowledgment(self, ack):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	658 self._acknowledgments.append(ack)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	659
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	660 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	661 """ return something that can be used as a title """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	662 if self._title:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	663 if self._id:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	664 return "%s (%s)" % (self._title, self._id._value)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	665 return self._title
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	666 if self._id:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	667 return self._id._value
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	668 return "#%d" % self._ordinal
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	669
7 c924c15bd110 Add a method to get a Vulnerability Note per ordinal Benoît Allard <benoit.allard@greenbone.net> parents: 6 diff changeset	670 def getNote(self, ordinal):
c924c15bd110 Add a method to get a Vulnerability Note per ordinal Benoît Allard <benoit.allard@greenbone.net> parents: 6 diff changeset	671 for note in self._notes:
c924c15bd110 Add a method to get a Vulnerability Note per ordinal Benoît Allard <benoit.allard@greenbone.net> parents: 6 diff changeset	672 if note._ordinal == ordinal:
c924c15bd110 Add a method to get a Vulnerability Note per ordinal Benoît Allard <benoit.allard@greenbone.net> parents: 6 diff changeset	673 return note
c924c15bd110 Add a method to get a Vulnerability Note per ordinal Benoît Allard <benoit.allard@greenbone.net> parents: 6 diff changeset	674 return None
c924c15bd110 Add a method to get a Vulnerability Note per ordinal Benoît Allard <benoit.allard@greenbone.net> parents: 6 diff changeset	675
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	676 def validate(self, productids, groupids):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	677 if not self._ordinal:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	678 raise ValidationError('A Vulnerability must have an ordinal')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	679 if self._id is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	680 self._id.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	681 ordinals = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	682 for note in self._notes:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	683 note.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	684 if note._ordinal in ordinals:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	685 raise ValidationError('Vulnerability Note Ordinal %d duplicated' % note._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	686 ordinals.add(note._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	687 for involvement in self._involvements:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	688 involvement.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	689 for cwe in self._cwes:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	690 cwe.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	691 for status in self._productstatuses:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	692 status.validate(productids)
13 db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	693 pids = set()
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	694 for status in self._productstatuses:
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	695 for pid in status._productids:
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	696 if pid in pids:
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	697 raise ValidationError('ProductID %s mentionned in two different ProductStatuses for Vulnerability %d' % (pid, self._ordinal))
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	698 pids.add(pid)
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	699 for threat in self._threats:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	700 threat.validate(productids, groupids)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	701 for cvss in self._cvsss:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	702 cvss.validate(productids)
14 640b88744523 Fix issue in validation of CVSS Score Sets Benoît Allard <benoit.allard@greenbone.net> parents: 13 diff changeset	703 pids = set()
13 db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	704 for cvss in self._cvsss:
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	705 for pid in (cvss._productids or productids):
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	706 if pid in pids:
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	707 raise ValidationError('ProductID %s mentionned in two different CVSS Score Sets for Vulnerability %d' % (pid, self._ordinal))
db2a02fff101 Improve validation Benoît Allard <benoit.allard@greenbone.net> parents: 8 diff changeset	708 pids.add(pid)
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	709 for remediation in self._remediations:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	710 remediation.validate(productids, groupids)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	711 for reference in self._references:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	712 reference.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	713 for acknowledgment in self._acknowledgments:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	714 acknowledgment.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	715
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	716
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	717
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	718 class CVRFInvolvement(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	719 PARTIES = CVRFPublisher.TYPES
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	720 STATUSES = ('Open', 'Disputed', 'In Progress', 'Completed',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	721 'Contact Attempted', 'Not Contacted')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	722 def __init__(self, party, status):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	723 self._party = party
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	724 self._status = status
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	725 self._description = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	726
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	727 def setDescription(self, description):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	728 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	729
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	730 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	731 return "From %s: %s" % (self._party, self._status)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	732
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	733 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	734 if not self._party:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	735 raise ValidationError('An Involvement must have a Party')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	736 if self._party not in self.PARTIES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	737 raise ValidationError("An Involvement's Party must be one of %s" % ', '.join(self.PARTIES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	738 if not self._status:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	739 raise ValidationError('An Involvement must have a Status')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	740 if self._status not in self.STATUSES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	741 raise ValidationError("An Involvement's Status must be one of %s" % ', '.join(self.STATUSES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	742
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	743
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	744 class CVRFCWE(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	745 def __init__(self, _id, value):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	746 self._id = _id
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	747 self._value = value
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	748
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	749 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	750 if not self._id:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	751 raise ValidationError('A CWE must have an ID')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	752 if not self._value:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	753 raise ValidationError('A CWE must have a description')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	754
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	755
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	756 class CVRFProductStatus(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	757 TYPES = ('First Affected', 'Known Affected', 'Known Not Affected',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	758 'First Fixed', 'Fixed', 'Recommended', 'Last Affected')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	759 def __init__(self, _type):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	760 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	761 self._productids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	762
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	763 def addProductID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	764 self._productids.append(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	765
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	766 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	767 return "%s: %d products" % (self._type, len(self._productids))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	768
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	769 def validate(self, productids):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	770 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	771 raise ValidationError('A Product Status must have a Type')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	772 if self._type not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	773 raise ValidationError("A Product Status' Type must be one of %s" % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	774 if len(self._productids) < 1:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	775 raise ValidationError('A Product Status must mention at least one Product')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	776 for productid in self._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	777 if productid not in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	778 raise ValidationError('Unknown ProductID: %s' % productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	779
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	780
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	781 class CVRFThreat(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	782 TYPES = ('Impact', 'Exploit Status', 'Target Set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	783 def __init__(self, _type, description):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	784 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	785 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	786 self._date = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	787 self._productids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	788 self._groupids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	789
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	790 def setDate(self, date):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	791 self._date = date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	792
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	793 def addProductID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	794 self._productids.append(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	795
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	796 def addGroupID(self, groupid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	797 self._groupids.append(groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	798
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	799 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	800 return self._type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	801
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	802 def validate(self, productids, groupids):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	803 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	804 raise ValidationError('A Threat must have a Type')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	805 if self._type not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	806 raise ValidationError("A Threat's Type must be one of %s" % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	807 if not self._description:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	808 raise ValidationError('A Threat must have a Description')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	809 for productid in self._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	810 if productid not in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	811 raise ValidationError('Unknown ProductID: %s' % productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	812 for groupid in self._groupids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	813 if groupid not in groupids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	814 raise ValidationError('Unknown GroupID: %s' % groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	815
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	816
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	817 class CVRFCVSSSet(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	818 # To determine the base Score
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	819 VALUES = {'AV': {'L':0.395, 'A':0.646, 'N':1.0},
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	820 'AC': {'H':0.35, 'M':0.61 ,'L':0.71},
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	821 'Au': {'M':0.45, 'S':0.56, 'N':0.704},
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	822 'C': {'N':0.0, 'P':0.275, 'C':0.66},
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	823 'I': {'N':0.0, 'P':0.275, 'C':0.66},
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	824 'A': {'N':0.0, 'P':0.275, 'C':0.66}}
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	825 def __init__(self, basescore):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	826 self._basescore = basescore
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	827 self._temporalscore = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	828 self._environmentalscore = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	829 self._vector = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	830 self.vector = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	831 self._productids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	832
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	833 def setTemporalScore(self, tempscore):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	834 self._temporalscore = tempscore
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	835
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	836 def setEnvironmentalScore(self, envscore):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	837 self._environmentalscore = envscore
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	838
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	839 def setVector(self, vector):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	840 self._vector = vector
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	841 if vector is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	842 self.vector = vector
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	843 return
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	844 try:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	845 self.vector = {}
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	846 for component in vector[:26].split('/'):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	847 name, value = component.split(':')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	848 self.vector[name] = self.VALUES[name][value]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	849 except (KeyError, ValueError):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	850 self.vector = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	851
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	852 def addProductID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	853 self._productids.append(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	854
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	855 def baseScore(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	856 v = self.vector # make an alias for shorter lines
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	857 exploitability = 20 * v['AV'] * v['AC'] * v['Au']
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	858 impact = 10.41 * (1 - (1 - v['C']) * (1 - v['I']) * (1 - v['A']))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	859 def f(i): return 0 if i == 0 else 1.176
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	860 return ((0.6 * impact) + (0.4 * exploitability) - 1.5) * f(impact)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	861
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	862 def validate(self, productids):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	863 if not self._basescore:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	864 raise ValidationError('A CVSS Score Set must have a Base Score')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	865 if self._vector and not self.vector:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	866 raise ValidationError('Syntax Error in CVSS Vector')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	867 if abs(self._basescore - self.baseScore()) >= 0.05:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	868 raise ValidationError('Inconsistency in CVSS Score Set between Vector (%f) and Base Score (%f)' % (self.baseScore(), self._basescore))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	869 for productid in self._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	870 if productid not in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	871 raise ValidationError('Unknown ProductID: %s' % productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	872
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	873
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	874 class CVRFRemediation(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	875 TYPES = ('Workaround', 'Mitigation', 'Vendor Fix', 'None Available',
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	876 'Will Not Fix')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	877 def __init__(self, _type, description):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	878 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	879 self._description = description
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	880 self._date = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	881 self._entitlement = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	882 self._url = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	883 self._productids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	884 self._groupids = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	885
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	886 def setDate(self, date):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	887 self._date = date
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	888
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	889 def setEntitlement(self, entitlement):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	890 self._entitlement = entitlement
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	891
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	892 def setURL(self, url):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	893 self._url = url
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	894
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	895 def addProductID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	896 self._productids.append(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	897
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	898 def addGroupID(self, groupid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	899 self._groupids.append(groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	900
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	901 def getTitle(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	902 return self._type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	903
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	904 def validate(self, productids, groupids):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	905 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	906 raise ValidationError('A Remediation must have a Type')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	907 if self._type not in self.TYPES:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	908 raise ValidationError("A Remediation's Type must be one of %s" % ', '.join(self.TYPES))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	909 if not self._description:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	910 raise ValidationError('A Remediation must have a Description')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	911 for productid in self._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	912 if productid not in productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	913 raise ValidationError('Unknown ProductID: %s' % productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	914 for groupid in self._groupids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	915 if groupid not in groupids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	916 raise ValidationError('Unknown GroupID: %s' % groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	917
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	918 class CVRF(object):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	919 def __init__(self, title, _type):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	920 self._title = title
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	921 self._type = _type
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	922 self._publisher = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	923 self._tracking = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	924 self._notes = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	925 self._distribution = None
1 d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	926 self._aggregateseverity = None
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	927 self._references = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	928 self._acknowledgments = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	929 self._producttree = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	930 self._vulnerabilities = []
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	931
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	932 def setPublisher(self, publisher):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	933 self._publisher = publisher
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	934
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	935 def setTracking(self, tracking):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	936 self._tracking = tracking
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	937
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	938 def addNote(self, note):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	939 self._notes.append(note)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	940
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	941 def setDistribution(self, distribution):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	942 self._distribution = distribution
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	943
1 d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	944 def setAggregateSeverity(self, aggregateseverity):
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	945 self._aggregateseverity = aggregateseverity
d47e1164740f Add support for AggregateSeverity Benoît Allard <benoit.allard@greenbone.net> parents: 0 diff changeset	946
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	947 def addReference(self, ref):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	948 self._references.append(ref)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	949
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	950 def addAcknowledgment(self, ack):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	951 self._acknowledgments.append(ack)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	952
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	953 def createProductTree(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	954 """ only done if the element is there """
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	955 self._producttree = CVRFProductTree()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	956 return self._producttree
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	957
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	958 def addVulnerability(self, vuln):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	959 self._vulnerabilities.append(vuln)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	960
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	961 def getProductForID(self, productid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	962 if self._producttree is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	963 raise ValueError('No ProductTree')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	964 return self._producttree.getProductForID(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	965
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	966 def getGroupForID(self, groupid):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	967 if self._producttree is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	968 raise ValueError('No ProductTree')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	969 return self._producttree.getGroupForID(groupid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	970
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	971 def getHighestCVSS(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	972 highestBaseScore = 0
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	973 highest = None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	974 for vulnerability in self._vulnerabilities:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	975 for cvss in vulnerability._cvsss:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	976 if cvss._basescore <= highestBaseScore:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	977 continue
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	978 highestBaseScore = cvss._basescore
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	979 highest = cvss
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	980 return highest
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	981
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	982 def getProductList(self, type_='Fixed'):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	983 products = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	984 if type_ == 'Fixed':
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	985 # First try through the Remediation
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	986 for vulnerability in self._vulnerabilities:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	987 for remediation in vulnerability._remediations:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	988 if remediation._type != 'Vendor Fix':
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	989 continue
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	990 for productid in remediation._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	991 products.add(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	992 for groupid in remediation._groupids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	993 for productid in self.getGroupForID(groupid)._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	994 products.add(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	995 if not products:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	996 # If nothing there, try through the productstatuses
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	997 for vulnerability in self._vulnerabilities:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	998 for status in vulnerability._productstatuses:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	999 if status._type != type_:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1000 continue
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1001 for productid in status._productids:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1002 products.add(productid)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1003 return set(self.getProductForID(p) for p in products)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1004
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1005 def getNote(self, ordinal):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1006 for note in self._notes:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1007 if note._ordinal == ordinal:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1008 return note
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1009 return None
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1010
6 633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	1011 def getDocId(self):
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	1012 if self._tracking is not None:
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	1013 return self._tracking.getId()
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	1014 # Make up something ...
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	1015 return self._title.lower()
633ebfcff0d0 Add a method to request a Document ID Benoît Allard <benoit.allard@greenbone.net> parents: 1 diff changeset	1016
0 e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1017 def validate(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1018 if not self._title:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1019 raise ValidationError('Document Title cannot be empty')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1020 if not self._type:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1021 raise ValidationError('Document Type cannot be empty')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1022 if self._publisher is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1023 raise ValidationError('Document Publisher needs to be set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1024 self._publisher.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1025 if self._tracking is None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1026 raise ValidationError('Document Tracking needs to be set')
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1027 self._tracking.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1028 ordinals = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1029 for note in self._notes:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1030 note.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1031 if note._ordinal in ordinals:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1032 raise ValidationError('Document Note ordinal %d is issued twice' % note._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1033 ordinals.add(note._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1034 for reference in self._references:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1035 reference.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1036 for acknowledgment in self._acknowledgments:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1037 acknowledgment.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1038 productids = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1039 groupids = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1040 if self._producttree:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1041 productids, groupids = self._producttree.validate()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1042 ordinals = set()
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1043 for vulnerability in self._vulnerabilities:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1044 vulnerability.validate(productids, groupids)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1045 if vulnerability._ordinal in ordinals:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1046 raise ValidationError('Vulnerability ordinal %d is issued twice' % vulnerability._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1047 ordinals.add(vulnerability._ordinal)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1048
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1049 def __str__(self):
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1050 s = [
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1051 'Title: %s' % self._title,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1052 'Type: %s' % self._type,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1053 'Publisher: %s' % self._publisher,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1054 'tracking: %s' % self._tracking,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1055 '%d Notes: %s' % (len(self._notes), ', '.join(
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1056 str(n) for n in self._notes))
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1057 ]
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1058 if self._distribution is not None:
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1059 s.append('Distribution: %s' % self._distribution)
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1060 s.extend([
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1061 '%d Acknowledgments' % len(self._acknowledgments),
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1062 'Products: %s' % self._producttree,
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1063 ])
e18b61a73a68 Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	1064 return '\n'.join(s)

Mercurial > farol > farolluz

annotate farolluz/cvrf.py @ 15:dcc946b30343