From 0716214a3121a3afed3b4f4a13e1c21684251d8f Mon Sep 17 00:00:00 2001 From: heyethereum Date: Sat, 17 Aug 2024 13:13:24 +0800 Subject: [PATCH] whitelist some url classified wrongly --- .../qrcode/service/URLVerificationService.java | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) 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 9845f41..877014d 100644 --- a/src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java +++ b/src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java @@ -429,6 +429,16 @@ public class URLVerificationService { // Get Classification using ML Model public String getClassification(URLModel urlModel){ + String content = urlModel.getData().getContents(); + // if in whitelist, return Benign and Safe + for (String domain : WHITELIST_DOMAINS) { + if (content.contains(domain)) { + // If in whitelist, set category to BENIGN and return SAFE + urlModel.getDetails().setClassifications(CAT_BENIGN); + return CLASSIFY_SAFE; + } + } + // Call ML model String category = predictionService.predict(urlModel); @@ -452,4 +462,12 @@ public class URLVerificationService { } return CLASSIFY_UNSAFE; } + + // Static array for whitelist domains + private static final List WHITELIST_DOMAINS = Arrays.asList( + "safeqr.github.io/marketing", + "uow.edu.au", + "nus.edu.sg", + "sim.edu.sg" + ); } \ No newline at end of file