Merge pull request #617 from bytedance/fix-cloader

Fix cloader
bytedance · May 22, 2024 · 1ee508c · 1ee508c
2 parents 4e3f188 + 572fa71
commit 1ee508c
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/rasp/jvm/JVMAgent/src/main/java/com/security/smithloader/SmithAgent.java b/rasp/jvm/JVMAgent/src/main/java/com/security/smithloader/SmithAgent.java
@@ -43,6 +43,10 @@ public static Object getSmithProbeProxy() {
         return SmithProberProxyObj;
     }
 
+    public static Object getSmithProbe() {
+        return SmithProberObj;
+    }
+
     public static void PreProxy(Object MethodNameObj,int classID, int methodID, Object[] args) {
 
         if(SmithProberProxyObj != null) {
@@ -73,6 +77,17 @@ public static Object ExceptionProxy(Object MethodNameObj,int classID, int method
         return null;
     }
 
+    public static Object PassProxyToProbe(Object Obj,Object MethodNameObj,int functionID, Object[] args) {
+
+          if(Obj != null) {
+            String MethodName = (String)MethodNameObj;
+            Class<?>[]  argType = new Class[]{int.class,int.class,Object[].class};
+            return Reflection.invokeMethod(Obj,MethodName,argType,functionID,args);
+        }
+
+        return null;
+    }
+
     private static boolean loadSmithProber(String proberPath, Instrumentation inst) {
         boolean bret = false;
         boolean bexception = false;
@@ -268,13 +283,11 @@ public static void agentmain(String agentArgs, Instrumentation inst) {
                 SmithAgentLogger.logger.info("checksumStr:" + checksumStr);
                 SmithAgentLogger.logger.info("proberPath:" + proberPath); 
 
-                /* 
                 if (!JarUtil.checkJarFile(proberPath,checksumStr)) {
                     System.setProperty("smith.status", proberPath + " check fail");
                     SmithAgentLogger.logger.warning(proberPath + " check fail!");
                     return ;
                 }
-                */
 
                 if(instrumentation == null) {
                     instrumentation = inst;