fix default value for IAST_SCAN_INSTANCE_COUNT

newrelic · Nov 6, 2024 · ecd14d7 · ecd14d7
1 parent 5826ac7
commit ecd14d7
Showing 1 changed file with 5 additions and 3 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
@@ -102,11 +102,13 @@ public long instantiate() throws RestrictionModeException {
         // Set required LogLevel
         logLevel = applyRequiredLogLevel();
 
-        if(NewRelic.getAgent().getConfig().getValue(IUtilConstants.IAST_SCAN_INSTANCE_COUNT) instanceof Integer) {
-            scanControllers.setScanInstanceCount(NewRelic.getAgent().getConfig().getValue(IUtilConstants.IAST_SCAN_INSTANCE_COUNT));
-        } else {
+        try {
+            scanControllers.setScanInstanceCount(NewRelic.getAgent().getConfig().getValue(IUtilConstants.IAST_SCAN_INSTANCE_COUNT, 0));
+        } catch (NumberFormatException | ClassCastException e){
+            logger.log(LogLevel.FINEST, String.format("Error while reading IAST_SCAN_INSTANCE_COUNT %s , setting to default", NewRelic.getAgent().getConfig().getValue(IUtilConstants.IAST_SCAN_INSTANCE_COUNT)), AgentConfig.class.getName());
             scanControllers.setScanInstanceCount(0);
         }
+
         //Always set IAST Test Identifier after IAST_SCAN_INSTANCE_COUNT
         if(NewRelic.getAgent().getConfig().getValue(IUtilConstants.IAST_TEST_IDENTIFIER) instanceof String) {
             scanControllers.setIastTestIdentifier(NewRelic.getAgent().getConfig().getValue(IUtilConstants.IAST_TEST_IDENTIFIER));