Add path to skipScanParameters

newrelic · Dec 18, 2024 · 13bf17c · 13bf17c
1 parent fe0ce16
commit 13bf17c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/newrelic-security-agent/src/main/java/com/newrelic/agent/security/AgentConfig.java b/newrelic-security-agent/src/main/java/com/newrelic/agent/security/AgentConfig.java
@@ -273,6 +273,11 @@ private void readIastRestrictedConfig() throws RestrictionModeException {
             this.agentMode.getIastScan().getRestrictionCriteria().getMappingParameters().getHeader().setLocations(NewRelic.getAgent().getConfig().getValue(RESTRICTION_CRITERIA_MAPPING_PARAMETERS_HEADER_LOCATION, Collections.emptyList()).stream()
                     .map(Object::toString)
                     .collect(Collectors.toList()));
+            if(restrictionCriteria.getMappingParameters().getPath().isEnabled()) {
+                this.agentMode.getIastScan().getRestrictionCriteria().getMappingParameters().getPath().setLocations(restrictionCriteria.getAccountInfo().getAccountIds().stream()
+                        .map(Object::toString)
+                        .collect(Collectors.toList()));
+            }
 
             //Strict Criteria
             List<Map<String, String>> strictCriteria = NewRelic.getAgent().getConfig().getValue(RESTRICTION_CRITERIA_STRICT, Collections.emptyList());

diff --git a/...y-api/src/main/java/com/newrelic/api/agent/security/schema/policy/SkipScanParameters.java b/...y-api/src/main/java/com/newrelic/api/agent/security/schema/policy/SkipScanParameters.java
@@ -11,6 +11,8 @@ public class SkipScanParameters {
 
     private List<String> body = new ArrayList<>();
 
+    private List<String> path = new ArrayList<>();
+
     public SkipScanParameters() {
     }
 
@@ -26,6 +28,14 @@ public List<String> getBody() {
         return body;
     }
 
+    public List<String> getPath() {
+        return path;
+    }
+
+    public void setPath(List<String> path) {
+        this.path = path;
+    }
+
     public void setHeader(List<String> header) {
         this.header = header;
     }
@@ -42,6 +52,10 @@ public void addHeader(String header) {
         this.header.add(header);
     }
 
+    public void addPath(String path) {
+        this.path.add(path);
+    }
+
     public void addQuery(String query) {
         this.query.add(query);
     }