isky/backend-springboot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

complete url verifications

Browse Source
This commit is contained in:
heyethereum
2024-08-13 21:03:15 +08:00
parent 63e2d299fd
commit 8665693642
3 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/com/safeqr/app/constants/CommonConstants.java
View File

@@ -23,10 +23,10 @@ public class CommonConstants {
public static final String CLASSIFY_UNSAFE = "UNSAFE";
public static final String CLASSIFY_UNKNOWN = "UNKNOWN";
public static final String CAT_BENIGN = "benign";
public static final String CAT_DEFACEMENT = "defacement";
public static final String CAT_MALWARE = "malware";
public static final String CAT_PHISHING = "phishing";
public static final String CAT_BENIGN = "Benign";
public static final String CAT_DEFACEMENT = "Defacement";
public static final String CAT_MALWARE = "Malware";
public static final String CAT_PHISHING = "Phishing";
public static final Integer GMAIL_ACTIVE = 1;
}

3
src/main/java/com/safeqr/app/qrcode/entity/URLEntity.java
View File

@@ -23,6 +23,9 @@ import java.util.UUID;
@NoArgsConstructor
@AllArgsConstructor
public class URLEntity {
@Column(name="classifications")
private String classifications;
@Id
@JsonIgnore
@GeneratedValue(generator = "UUID")

12
src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java
View File

@@ -432,11 +432,19 @@ public class URLVerificationService {
String category = predictionService.predict(urlModel);
//update in category in url table
urlModel.getDetails().setClassifications(category);
// return classification results
if (category.equals(CAT_BENIGN)) {
if (!urlModel.getDetails().getTrackingDescriptions().isEmpty()) {
if (!urlModel.getDetails().getTrackingDescriptions().isEmpty() || // contains tracking
urlModel.getData().getInfo().getPrefix().equalsIgnoreCase("http://") || // uses http
urlModel.getDetails().getSslStripping().contains(true) || // has SSL stripping
urlModel.getDetails().getHasExecutable().equalsIgnoreCase("yes") || // contains executable
!urlModel.getDetails().getJavascriptCheck().isEmpty() || // contains javascript
!urlModel.getDetails().getHasIpAddress().isEmpty() || // contains IP address
urlModel.getDetails().getHostnameEmbedding() != null // contains hostname embedding
) {
return CLASSIFY_WARNING;
}
return CLASSIFY_SAFE;