Fix index out of bound error

CybercentreCanada · Dec 23, 2024 · 75fe771 · 75fe771
1 parent 7da38af
commit 75fe771
Showing 1 changed file with 11 additions and 24 deletions.
diff --git a/src/ca/gc/cyber/kangooroo/utils/io/net/http/HarUtils.java b/src/ca/gc/cyber/kangooroo/utils/io/net/http/HarUtils.java
@@ -14,13 +14,16 @@
 import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.Base64;
+import java.util.Collections;
 import java.util.Date;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
+import java.util.stream.Collector;
+import java.util.stream.Collectors;
 
 import org.apache.commons.codec.digest.DigestUtils;
 import org.apache.commons.io.FileUtils;
@@ -404,38 +407,22 @@ public static void removeRequestUrlEntries(Har har, String pattern) {
 
         List<HarEntry> oldEntries = har.getLog().getEntries();
 
-        List<Integer> entryToRemove = new ArrayList<Integer>();
-        for (int i = 0; i < oldEntries.size(); i++) {
-            var curUrl = oldEntries.get(i).getRequest().getUrl();
-            if (curUrl.contains(pattern)) {
-                entryToRemove.add(i);
-            }
-        }
-
-        for (Integer entryIdx : entryToRemove) {
-            oldEntries.remove((int) entryIdx);
-
-        }
+        har.getLog().setEntries(
+            oldEntries.stream().filter(entry -> !entry.getRequest().getUrl().contains(pattern))
+            .collect(Collectors.toList())
+        );
 
 
     }
 
     public static void removeResponseEntries(Har har, int code) {
 
         List<HarEntry> oldEntries = har.getLog().getEntries();
-        List<Integer> entryToRemove = new ArrayList<Integer>();
-        for (int i = 0; i < oldEntries.size(); i++) {
-            var statusCode = oldEntries.get(i).getResponse().getStatus();
 
-            if (statusCode == code) {
-                entryToRemove.add(i);
-            }
-
-        }
-
-        for (Integer entryIdx : entryToRemove) {
-            oldEntries.remove((int) entryIdx);
-        }
+        har.getLog().setEntries(oldEntries.stream()
+        .filter(entry -> entry.getResponse().getStatus() != code)
+        .collect(Collectors.toList()));
+
 
     }