From 8665693642caff33695f4c6aa5321b7716cc4582 Mon Sep 17 00:00:00 2001 From: heyethereum Date: Tue, 13 Aug 2024 21:03:15 +0800 Subject: [PATCH] complete url verifications --- .../com/safeqr/app/constants/CommonConstants.java | 8 ++++---- .../java/com/safeqr/app/qrcode/entity/URLEntity.java | 3 +++ .../app/qrcode/service/URLVerificationService.java | 12 ++++++++++-- 3 files changed, 17 insertions(+), 6 deletions(-) diff --git a/src/main/java/com/safeqr/app/constants/CommonConstants.java b/src/main/java/com/safeqr/app/constants/CommonConstants.java index 7f9c176..93fbf40 100644 --- a/src/main/java/com/safeqr/app/constants/CommonConstants.java +++ b/src/main/java/com/safeqr/app/constants/CommonConstants.java @@ -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; } diff --git a/src/main/java/com/safeqr/app/qrcode/entity/URLEntity.java b/src/main/java/com/safeqr/app/qrcode/entity/URLEntity.java index 38d1619..0d22e74 100644 --- a/src/main/java/com/safeqr/app/qrcode/entity/URLEntity.java +++ b/src/main/java/com/safeqr/app/qrcode/entity/URLEntity.java @@ -23,6 +23,9 @@ import java.util.UUID; @NoArgsConstructor @AllArgsConstructor public class URLEntity { + @Column(name="classifications") + private String classifications; + @Id @JsonIgnore @GeneratedValue(generator = "UUID") diff --git a/src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java b/src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java index c225799..d1165b1 100644 --- a/src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java +++ b/src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java @@ -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;