def wrapper(db, *args, **kwargs):
issue_objs = []
for issue in issues:
issue_objs.append(db.get_or_create(CVE, id=issue))
max_severity = highest_severity([issue.severity for issue in issue_objs])
for num in range(1, count + 1):
group = CVEGroup()
if id:
group.id = id if count <= 1 else '{}{}'.format(id, num)
group.status = status if status else affected_to_status(Affected.affected, packages[0], fixed)
group.severity = severity if severity else max_severity
group.affected = affected
group.fixed = fixed
group.bug_ticket = bug_ticket
group.reference = reference
group.notes = notes
group.created = created
group.advisory_qualified = advisory_qualified
db.session.add(group)
db.session.commit()
for issue in issue_objs:
db.get_or_create(CVEGroupEntry, group=group, cve=issue)
for pkgname in packages:
db.get_or_create(CVEGroupPackage, pkgname=pkgname, group=group)
db.session.commit()
func(db=db, *args, **kwargs)
def wrapper(db, *args, **kwargs):
group = CVEGroup()
if id:
group.id = id
group.status = status if status else affected_to_status(Affected.affected, packages[0], fixed)
group.severity = severity
group.affected = affected
group.fixed = fixed
group.bug_ticket = bug_ticket
group.reference = reference
group.notes = notes
group.created = created
group.advisory_qualified = advisory_qualified
db.session.add(group)
db.session.commit()
for issue in issues:
cve = db.get_or_create(CVE, id=issue)
db.get_or_create(CVEGroupEntry, group=group, cve=cve)
for pkgname in packages:
db.get_or_create(CVEGroupPackage, pkgname=pkgname, group=group)
db.session.commit()
func(db=db, *args, **kwargs)
def add_group():
form = GroupForm()
if not form.validate_on_submit():
return render_template('form/group.html',
title='Add AVG',
form=form,
CVEGroup=CVEGroup)
issues = []
cve_ids = multiline_to_list(form.cve.data)
cve_ids = set(filter(lambda s: s.startswith('CVE-'), cve_ids))
issues = CVE.query.filter(CVE.id.in_(cve_ids)).all()
issue_ids = [issue.id for issue in issues]
for cve_id in list(filter(lambda issue: issue not in issue_ids, cve_ids)):
cve = db.create(CVE, id=cve_id)
issues.append(cve)
flash('Added {}'.format(cve.id))
pkgnames = multiline_to_list(form.pkgnames.data)
fixed = form.fixed.data
affected = Affected.fromstring(form.status.data)
status = affected_to_status(affected, pkgnames[0], fixed)
severity = highest_severity([issue.severity for issue in issues])
advisory_qualified = form.advisory_qualified and status is not Status.not_affected
group = db.create(CVEGroup,
affected=form.affected.data,
status=status,
fixed=fixed,
bug_ticket=form.bug_ticket.data,
reference=form.reference.data,
notes=form.notes.data,
severity=severity,
advisory_qualified=advisory_qualified)
db.session.commit()
for cve in issues:
db.create(CVEGroupEntry, group=group, cve=cve)
for pkgname in pkgnames:
db.get_or_create(CVEGroupPackage, pkgname=pkgname, group=group)
flash('Added {}'.format(pkgname))
db.session.commit()
flash('Added {}'.format(group.name))
return redirect('/{}'.format(group.name))
def edit_group(avg):
group_id = avg.replace('AVG-', '')
group_data = (db.session.query(
CVEGroup, CVE, func.group_concat(CVEGroupPackage.pkgname, ' '),
Advisory).filter(CVEGroup.id == group_id).join(CVEGroupEntry).join(
CVE).join(CVEGroupPackage).outerjoin(
Advisory,
Advisory.group_package_id == CVEGroupPackage.id).group_by(
CVEGroup.id).group_by(CVE.id).order_by(CVE.id)).all()
if not group_data:
return not_found()
group = group_data[0][0]
issues = [cve for (group, cve, pkg, advisory) in group_data]
issue_ids = [cve.id for cve in issues]
pkgnames = set(
chain.from_iterable(
[pkg.split(' ') for (group, cve, pkg, advisory) in group_data]))
advisories = set(advisory for (group, cve, pkg, advisory) in group_data
if advisory)
if not user_can_edit_group(advisories):
return forbidden()
form = GroupForm(pkgnames)
if not form.is_submitted():
form.affected.data = group.affected
form.fixed.data = group.fixed
form.pkgnames.data = "\n".join(sorted(pkgnames))
form.status.data = status_to_affected(group.status).name
form.reference.data = group.reference
form.notes.data = group.notes
form.bug_ticket.data = group.bug_ticket
form.advisory_qualified.data = group.advisory_qualified and group.status is not Status.not_affected
form.cve.data = "\n".join(issue_ids)
if not form.validate_on_submit():
if advisories:
flash(
'WARNING: This is referenced by an already published advisory!',
'warning')
return render_template('form/group.html',
title='Edit {}'.format(avg),
form=form,
CVEGroup=CVEGroup)
pkgnames_edited = multiline_to_list(form.pkgnames.data)
group.affected = form.affected.data
group.fixed = form.fixed.data
group.status = affected_to_status(Affected.fromstring(form.status.data),
pkgnames_edited[0], group.fixed)
group.bug_ticket = form.bug_ticket.data
group.reference = form.reference.data
group.notes = form.notes.data
group.advisory_qualified = form.advisory_qualified.data and group.status is not Status.not_affected
cve_ids = multiline_to_list(form.cve.data)
cve_ids = set(filter(lambda s: s.startswith('CVE-'), cve_ids))
issues_removed = set(filter(lambda issue: issue not in cve_ids, issue_ids))
issues_added = set(filter(lambda issue: issue not in issue_ids, cve_ids))
issues_final = set(
filter(lambda issue: issue.id not in issues_removed, issues))
if issues_removed:
(db.session.query(CVEGroupEntry).filter(
CVEGroupEntry.group_id == group.id).filter(
CVEGroupEntry.cve_id.in_(issues_removed)).delete(
synchronize_session=False))
for removed in issues_removed:
flash('Removed {}'.format(removed))
severities = [
issue.severity for issue in list(
filter(lambda issue: issue.id not in issues_removed, issues))
]
for cve_id in issues_added:
cve = db.get_or_create(CVE, id=cve_id)
db.get_or_create(CVEGroupEntry, group=group, cve=cve)
severities.append(cve.severity)
issues_final.add(cve)
flash('Added {}'.format(cve.id))
group.severity = highest_severity(severities)
pkgnames_removed = set(
filter(lambda pkgname: pkgname not in pkgnames_edited, pkgnames))
pkgnames_added = set(
filter(lambda pkgname: pkgname not in pkgnames, pkgnames_edited))
if pkgnames_removed:
(db.session.query(CVEGroupPackage).filter(
CVEGroupPackage.group_id == group.id).filter(
CVEGroupPackage.pkgname.in_(pkgnames_removed)).delete(
synchronize_session=False))
for removed in pkgnames_removed:
flash('Removed {}'.format(removed))
for pkgname in pkgnames_added:
db.get_or_create(CVEGroupPackage, pkgname=pkgname, group=group)
flash('Added {}'.format(pkgname))
# update scheduled advisories
for advisory in advisories:
if Publication.published == advisory.publication:
continue
issue_type = 'multiple issues' if len(
set([issue.issue_type for issue in issues_final])) > 1 else next(
iter(issues_final)).issue_type
advisory.advisory_type = issue_type
db.session.commit()
flash('Edited {}'.format(group.name))
return redirect('/{}'.format(group.name))
def add_group():
form = GroupForm()
if not form.validate_on_submit():
return render_template('form/group.html',
title='Add AVG',
form=form,
CVEGroup=CVEGroup)
issue_ids = multiline_to_list(form.cve.data)
issue_ids = set(filter(lambda s: s.startswith('CVE-'), issue_ids))
existing_issues = CVE.query.filter(CVE.id.in_(issue_ids)).all()
existing_issue_ids = [issue.id for issue in existing_issues]
pkgnames = multiline_to_list(form.pkgnames.data)
# check if a package with a CVE clashes with an existing group
if not form.force_submit.data:
same_group = (db.session.query(CVEGroup, CVE, CVEGroupPackage).join(
CVEGroupEntry).join(CVE).join(CVEGroupPackage).filter(
CVEGroupPackage.pkgname.in_(pkgnames)))
if issue_ids:
same_group = same_group.filter(CVE.id.in_(issue_ids))
same_group = same_group.all()
if same_group:
for group, cve, package in same_group:
flash(
ERROR_GROUP_WITH_ISSUE_EXISTS.format(
group.id, cve.id, package.pkgname), 'warning')
return render_template('form/group.html',
title='Add AVG',
form=form,
CVEGroup=CVEGroup,
show_force=True)
for cve_id in list(
filter(lambda issue: issue not in existing_issue_ids, issue_ids)):
cve = db.create(CVE, id=cve_id)
existing_issues.append(cve)
flash('Added {}'.format(cve.id))
fixed = form.fixed.data
affected = Affected.fromstring(form.status.data)
status = affected_to_status(affected, pkgnames[0], fixed)
severity = highest_severity([issue.severity for issue in existing_issues])
advisory_qualified = form.advisory_qualified.data and status is not Status.not_affected
group = db.create(CVEGroup,
affected=form.affected.data,
status=status,
fixed=fixed,
bug_ticket=form.bug_ticket.data,
reference=form.reference.data,
notes=form.notes.data,
severity=severity,
advisory_qualified=advisory_qualified)
db.session.commit()
for cve in existing_issues:
db.create(CVEGroupEntry, group=group, cve=cve)
for pkgname in pkgnames:
db.get_or_create(CVEGroupPackage, pkgname=pkgname, group=group)
flash('Added {}'.format(pkgname))
db.session.commit()
flash('Added {}'.format(group.name))
return redirect('/{}'.format(group.name))