fixing scan manage

Mixeway · Jun 21, 2024 · ae5909e · ae5909e
1 parent f1f019f
commit ae5909e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/io/mixeway/api/vulnmanage/model/Vuln.java b/src/main/java/io/mixeway/api/vulnmanage/model/Vuln.java
@@ -88,6 +88,7 @@ public <S extends VulnSource> Vuln(ProjectVulnerability projectVulnerability, St
 			case Constants.API_SCANNER_WEBAPP:
 				if (target instanceof WebApp && projectVulnerability.getVulnerabilitySource().getName().equals(Constants.VULN_TYPE_WEBAPP)) {
 					this.setGrade(projectVulnerability.getGrade());
+					this.setType(Constants.API_SCANNER_WEBAPP);
 					this.setVulnerabilityName(projectVulnerability.getVulnerability().getName());
 					this.setDescription(Objects.toString(projectVulnerability.getDescription(), "Description missing") + "\n\n" + Objects.toString(projectVulnerability.getRecommendation(), ""));
 					this.setBaseURL(projectVulnerability.getWebApp().getUrl());
@@ -105,6 +106,7 @@ public <S extends VulnSource> Vuln(ProjectVulnerability projectVulnerability, St
 			case Constants.API_SCANNER_OPENVAS:
 				if (target instanceof Interface && projectVulnerability.getVulnerabilitySource().getName().equals(Constants.VULN_TYPE_NETWORK)) {
 					this.setGrade(projectVulnerability.getGrade());
+					this.setType(Constants.API_SCANNER_OPENVAS);
 					this.setVulnerabilityName(projectVulnerability.getVulnerability().getName());
 					this.setIpAddress(projectVulnerability.getAnInterface().getPrivateip() != null && !projectVulnerability.getAnInterface().getPrivateip().equals("") ?
 							projectVulnerability.getAnInterface().getPrivateip() :