Update GetBinaryPath: use Libc to check if gnu is available

Author: bitterpanda63

agent_api/src/main/java/dev/aikido/agent_api/vulnerabilities/sql_injection/GetBinaryPath.java

@@ -2,23 +2,27 @@
 
 import dev.aikido.agent_api.helpers.logging.LogManager;
 import dev.aikido.agent_api.helpers.logging.Logger;
+import jnr.a64asm.INST_CODE;
+import jnr.ffi.Library;
+import jnr.ffi.LibraryLoader;
 
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.io.InterruptedIOException;
+import java.lang.annotation.Native;
 
 public final class GetBinaryPath {
     private GetBinaryPath() {}
     private static final Logger logger = LogManager.getLogger(GetBinaryPath.class);
 
     public static String getPathForBinary() {
-      String fileName = getFileName();
-      String aikidoDirectory = System.getProperty("AIK_agent_dir");
-      if (aikidoDirectory == null) {
-          return null;
-      }
-      return aikidoDirectory + "/binaries/" + fileName;
+        String fileName = getFileName();
+        String aikidoDirectory = System.getProperty("AIK_agent_dir");
+        if (aikidoDirectory == null) {
+            return null;
+        }
+        return aikidoDirectory + "/binaries/" + fileName;
     }
     private static String getFileName() {
         String os = System.getProperty("os.name").toLowerCase();
@@ -35,34 +39,29 @@ private static String getFileName() {
             fileName.append("x86_64-"); // Default to x86-64
         }
 
-        boolean isMusl = os.contains("musl") || architecture.contains("musl");
-
         if (os.contains("win")) {
             fileName.append("pc-windows-gnu.dll"); // Windows
         } else if (os.contains("mac")) {
             fileName.append("apple-darwin.dylib"); // macOS
-        } else { // os.contains("nix") || os.contains("nux")
+        } else {
             // Default to linux
             fileName.append(String.format("unknown-linux-%s.so", getLibCVariant()));
         }
         return fileName.toString();
     }
 
+    public interface Libc {
+        Libc INSTANCE = LibraryLoader.create(Libc.class).load("c");
+        String gnu_get_libc_version();
+    }
+
     private static String getLibCVariant() {
+        // gnu_get_libc_version only works for systems with gnu installed.
         try {
-            Process process = new ProcessBuilder("uname", "-o").start();
-            BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
-            String line = reader.readLine();
-            if (line != null) {
-                if (line.toLowerCase().contains("gnu")) {
-                    return "gnu";
-                } else {
-                    return "musl";
-                }
-            }
-        } catch (IOException e) {
-            logger.trace(e);
+            Libc.INSTANCE.gnu_get_libc_version();
+        } catch (UnsatisfiedLinkError e) {
+            return "musl";
         }
-        return "gnu"; // Default to gnu
+        return "gnu";
     }
 }