From 2d6de5f19875a4db65d3f4139717ac0eb19da466 Mon Sep 17 00:00:00 2001
From: Gulshan <kumargu@amazon.com>
Date: Fri, 11 Apr 2025 18:18:38 +0530
Subject: [PATCH] Support phasing off SecurityManager usage in favor of Java
 Agent

Signed-off-by: Gulshan <kumargu@amazon.com>
---
 build.gradle | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/build.gradle b/build.gradle
index 9bf6834c..8daa3e7e 100644
--- a/build.gradle
+++ b/build.gradle
@@ -164,6 +164,7 @@ publishing {
 
 configurations {
     zipArchive
+    agent
 }
 
 //****************************************************************************/
@@ -183,6 +184,9 @@ dependencies {
     zipArchive group: 'org.opensearch.plugin', name:'opensearch-job-scheduler', version: "${opensearch_build}"
     compileOnly "org.opensearch:opensearch-job-scheduler-spi:${opensearch_build}"
     implementation "com.github.seancfoley:ipaddress:5.4.2"
+    agent "org.opensearch:opensearch-agent-bootstrap:${opensearch_version}"
+    agent "org.opensearch:opensearch-agent:${opensearch_version}"
+    agent "net.bytebuddy:byte-buddy:${versions.bytebuddy}"
 }
 
 licenseHeaders.enabled = true
@@ -351,3 +355,13 @@ task updateVersion {
         ant.replaceregexp(file:'build.gradle', match: '"opensearch.version", "\\d.*"', replace: '"opensearch.version", "' + newVersion.tokenize('-')[0] + '-SNAPSHOT"', flags:'g', byline:true)
     }
 }
+
+task prepareAgent(type: Copy) {
+  from(configurations.agent)
+  into "$buildDir/agent"
+}
+
+tasks.withType(Test) {
+  dependsOn prepareAgent
+  jvmArgs += ["-javaagent:" + project.layout.buildDirectory.file("agent/opensearch-agent-${opensearch_version}.jar").get()]
+}