Handled different extension unique id

Signed-off-by: Owais Kazi <owaiskazi19@gmail.com>
opensearch-project · Jun 23, 2023 · d2e28f0 · d2e28f0
1 parent ac39864
commit d2e28f0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server/src/main/java/org/opensearch/transport/TransportService.java b/server/src/main/java/org/opensearch/transport/TransportService.java
@@ -547,6 +547,9 @@ public ConnectionManager.ConnectionValidator connectionValidatorForExtensionConn
             if (Strings.isNullOrEmpty(threadPool.getThreadContext().getHeader("extension_unique_id"))) {
                 threadPool.getThreadContext().putHeader("extension_unique_id", extensionUniqueId);
             }
+            if (!extensionUniqueId.equals(threadPool.getThreadContext().getHeader("extension_unique_id"))) {
+                throw new ConnectTransportException(node, "validation failed. unexpected extension");
+            }
             handshake(newConnection, actualProfile.getHandshakeTimeout().millis(), cn -> true, ActionListener.map(listener, resp -> {
                 final DiscoveryNode remote = resp.discoveryNode;