From 49e7eceeb7fa6d754807f89dfb9b1eb78303d881 Mon Sep 17 00:00:00 2001
From: lovesh-ap <lbaya@newrelic.com>
Date: Mon, 16 Dec 2024 19:01:42 +0530
Subject: [PATCH] Fix for status code check in 4xx list

---
 .../intcodeagent/apache/httpclient/IastHttpClient.java        | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/newrelic-security-agent/src/main/java/com/newrelic/agent/security/intcodeagent/apache/httpclient/IastHttpClient.java b/newrelic-security-agent/src/main/java/com/newrelic/agent/security/intcodeagent/apache/httpclient/IastHttpClient.java
index f858ed575..97130132d 100644
--- a/newrelic-security-agent/src/main/java/com/newrelic/agent/security/intcodeagent/apache/httpclient/IastHttpClient.java
+++ b/newrelic-security-agent/src/main/java/com/newrelic/agent/security/intcodeagent/apache/httpclient/IastHttpClient.java
@@ -78,7 +78,9 @@ public void tryToEstablishApplicationEndpoint(HttpRequest request) {
                 try {
                     ReadResult result = httpClient.execute(request, endpoint.getValue(), null, true);
                     int statusCode = result.getStatusCode();
-                    if ((statusCode >= 200 && statusCode < 300) || (statusCode >= 400 && statusCode < 500)) {
+                    if ((statusCode >= 200 && statusCode < 300) ||
+                            statusCode == 401 || statusCode == 402 ||
+                            statusCode == 406 || statusCode == 409) {
                         ServerConnectionConfiguration serverConnectionConfiguration = new ServerConnectionConfiguration(serverPort, endpoint.getKey(), endpoint.getValue(), true);
                         AppServerInfo appServerInfo = AppServerInfoHelper.getAppServerInfo();
                         appServerInfo.getConnectionConfiguration().put(serverPort, serverConnectionConfiguration);