Fix build due to phasing off SecurityManager usage in favor of Java Agent

Signed-off-by: Andriy Redko <[email protected]>

Author: reta

build.gradle

@@ -83,6 +83,7 @@ buildscript {
 //****************************************************************************/
 
 plugins {
+    id 'eclipse'
     id 'java-library'
     id 'java-test-fixtures'
     id 'idea'
@@ -244,6 +245,26 @@ allprojects {
         project.noticeFile = rootProject.file('NOTICE.txt')
         project.forbiddenApis.ignoreFailures = true
     }
+    
+    configurations {
+        agent
+    }
+
+    task prepareAgent(type: Copy) {
+        from(configurations.agent)
+        into "$buildDir/agent"
+    }
+
+    dependencies {
+        agent "org.opensearch:opensearch-agent-bootstrap:${opensearch_version}"
+        agent "org.opensearch:opensearch-agent:${opensearch_version}"
+        agent "net.bytebuddy:byte-buddy:${versions.bytebuddy}"
+    }
+
+    tasks.withType(Test) {
+      dependsOn prepareAgent
+      jvmArgs += ["-javaagent:" + project.layout.buildDirectory.file("agent/opensearch-agent-${opensearch_version}.jar").get()]
+    }
 }
 
 configurations {
@@ -610,3 +631,4 @@ task updateVersion {
         }
     }
 }
+

src/main/java/org/opensearch/knn/jni/PlatformUtils.java

@@ -15,6 +15,7 @@
 import org.apache.commons.lang.StringUtils;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+
 import oshi.util.platform.mac.SysctlUtil;
 
 import java.nio.file.Files;
@@ -27,7 +28,8 @@
 import java.util.stream.Stream;
 
 public class PlatformUtils {
-
+    private static volatile Boolean isAVX2Supported;
+    private static volatile Boolean isAVX512Supported;
     private static final Logger logger = LogManager.getLogger(PlatformUtils.class);
 
     /**
@@ -41,22 +43,26 @@ public class PlatformUtils {
      */
     public static boolean isAVX2SupportedBySystem() {
         if (!Platform.isIntel() || Platform.isWindows()) {
-            return false;
+            isAVX2Supported = false;
         }
 
-        if (Platform.isMac()) {
+        if (isAVX2Supported != null) {
+            return isAVX2Supported;
+        }
 
+        if (Platform.isMac()) {
             // sysctl or system control retrieves system info and allows processes with appropriate privileges
             // to set system info. This system info contains the machine dependent cpu features that are supported by it.
             // On MacOS, if the underlying processor supports AVX2 instruction set, it will be listed under the "leaf7"
             // subset of instructions ("sysctl -a | grep machdep.cpu.leaf7_features").
             // https://developer.apple.com/library/archive/documentation/System/Conceptual/ManPages_iPhoneOS/man3/sysctl.3.html
             try {
-                return AccessController.doPrivileged((PrivilegedExceptionAction<Boolean>) () -> {
+                isAVX2Supported = AccessController.doPrivileged((PrivilegedExceptionAction<Boolean>) () -> {
                     String flags = SysctlUtil.sysctl("machdep.cpu.leaf7_features", "empty");
                     return (flags.toLowerCase(Locale.ROOT)).contains("avx2");
                 });
             } catch (Exception e) {
+                isAVX2Supported = false;
                 logger.error("[KNN] Error fetching cpu flags info. [{}]", e.getMessage(), e);
             }
 
@@ -70,17 +76,18 @@ public static boolean isAVX2SupportedBySystem() {
             // https://ark.intel.com/content/www/us/en/ark/products/199285/intel-pentium-gold-g6600-processor-4m-cache-4-20-ghz.html
             String fileName = "/proc/cpuinfo";
             try {
-                return AccessController.doPrivileged(
+                isAVX2Supported = AccessController.doPrivileged(
                     (PrivilegedExceptionAction<Boolean>) () -> (Boolean) Files.lines(Paths.get(fileName))
                         .filter(s -> s.startsWith("flags"))
                         .anyMatch(s -> StringUtils.containsIgnoreCase(s, "avx2"))
                 );
 
             } catch (Exception e) {
+                isAVX2Supported = false;
                 logger.error("[KNN] Error reading file [{}]. [{}]", fileName, e.getMessage(), e);
             }
         }
-        return false;
+        return isAVX2Supported;
     }
 
     public static boolean isAVX512SupportedBySystem() {
@@ -97,7 +104,11 @@ private static boolean areAVX512FlagsAvailable(String[] avx512) {
         // https://github.com/facebookresearch/faiss/blob/main/faiss/CMakeLists.txt
 
         if (!Platform.isIntel() || Platform.isMac() || Platform.isWindows()) {
-            return false;
+            isAVX512Supported = false;
+        }
+
+        if (isAVX512Supported != null) {
+            return isAVX512Supported;
         }
 
         if (Platform.isLinux()) {
@@ -110,16 +121,16 @@ private static boolean areAVX512FlagsAvailable(String[] avx512) {
             String fileName = "/proc/cpuinfo";
 
             try {
-                return AccessController.doPrivileged((PrivilegedExceptionAction<Boolean>) () -> {
+                isAVX512Supported = AccessController.doPrivileged((PrivilegedExceptionAction<Boolean>) () -> {
                     Stream<String> linestream = Files.lines(Paths.get(fileName));
                     String flags = linestream.filter(line -> line.startsWith("flags")).findFirst().orElse("");
                     return Arrays.stream(avx512).allMatch(flags::contains);
                 });
-
             } catch (PrivilegedActionException e) {
+                isAVX512Supported = false;
                 logger.error("[KNN] Error reading file [{}]. [{}]", fileName, e.getMessage(), e);
             }
         }
-        return false;
+        return isAVX512Supported;
     }
 }

src/main/java/org/opensearch/knn/plugin/KNNPlugin.java

@@ -47,6 +47,7 @@
 import org.opensearch.knn.indices.ModelCache;
 import org.opensearch.knn.indices.ModelDao;
 import org.opensearch.knn.indices.ModelGraveyard;
+import org.opensearch.knn.jni.PlatformUtils;
 import org.opensearch.knn.plugin.rest.RestClearCacheHandler;
 import org.opensearch.knn.plugin.rest.RestDeleteModelHandler;
 import org.opensearch.knn.plugin.rest.RestGetModelHandler;
@@ -117,6 +118,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
+import java.util.concurrent.ForkJoinPool;
 import java.util.function.Supplier;
 
 import static java.util.Collections.singletonList;
@@ -174,6 +176,13 @@ public class KNNPlugin extends Plugin
     private ClusterService clusterService;
     private Supplier<RepositoriesService> repositoriesServiceSupplier;
 
+    static {
+        ForkJoinPool.commonPool().execute(() -> {
+            PlatformUtils.isAVX2SupportedBySystem();
+            PlatformUtils.isAVX512SPRSupportedBySystem();
+        });
+    }
+
     @Override
     public Map<String, Mapper.TypeParser> getMappers() {
         return Collections.singletonMap(