Skip to content

Commit 1bd4721

Browse files
committed
Missing service in looking up vulns by refs now handled.
2 parents f2f94f1 + 8b10cbe commit 1bd4721

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

lib/msf/core/db_manager/vuln.rb

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,10 @@ def find_vuln_by_details(details_map, host, service=nil)
4747
def find_vuln_by_refs(refs, host, service=nil)
4848
ref_ids = refs.find_all { |ref| ref.name.starts_with? 'CVE-'}
4949
relation = host.vulns.includes(:refs)
50-
relation.where(service_id: service.try(:id), refs: { id: ref_ids}).first || relation.where(refs: { id: ref_ids}).first
50+
if !service.try(:id).nil?
51+
return relation.where(service_id: service.try(:id), refs: { id: ref_ids}).first
52+
end
53+
return relation.where(refs: { id: ref_ids}).first
5154
end
5255

5356
def get_vuln(wspace, host, service, name, data='')

0 commit comments

Comments
 (0)