feat scanclass debug log

bytedance · Nov 25, 2023 · b12c02a · b12c02a
1 parent 2d5f5fa
commit b12c02a
Show file tree

Hide file tree

Showing 5 changed files with 40 additions and 0 deletions.
diff --git a/rasp/jvm/JVMProbe/src/main/java/com/security/smith/SmithProbe.java b/rasp/jvm/JVMProbe/src/main/java/com/security/smith/SmithProbe.java
@@ -422,6 +422,7 @@ private void checkClassFilter(ClassLoader loader, String className, byte[] class
                 classFilter.setStackTrace(Thread.currentThread().getStackTrace());
 
                 client.write(Operate.SCANCLASS, classFilter);
+                SmithLogger.logger.info("send metadata: " + classFilter.toString());
                 sendByte(classfileBuffer, classFilter.getTransId());
             }
         } catch(Exception e) {
@@ -742,6 +743,7 @@ public void onScanAllClass() {
                         classFilter.setStackTrace(Thread.currentThread().getStackTrace());
 
                         client.write(Operate.SCANCLASS, classFilter);
+                        SmithLogger.logger.info("send metadata: " + classFilter.toString());
                         sendClass(clazz, classFilter.getTransId());
                     }
 
@@ -811,6 +813,7 @@ private void sendByte(byte[] data, String transId) {
             classUpload.setClassData(data);
 
             client.write(Operate.CLASSUPLOAD, classUpload);
+            SmithLogger.logger.info("send classdata: " + classUpload.toString());
         //}
     }
 

diff --git a/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/ClassUploadTransformer.java b/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/ClassUploadTransformer.java
@@ -75,6 +75,7 @@ private void sendClass(Class<?> clazz, byte[] data) {
 
                 if (client != null) {
                     client.write(Operate.CLASSUPLOAD, classUpload);
+                    SmithLogger.logger.info("send classdata: " + classUpload.toString());
                 }
             }
 

diff --git a/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/Client.java b/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/Client.java
@@ -86,6 +86,15 @@ public void write(Operate operate, Object object) {
 
         Message message = new Message();
 
+        try {
+            if (operate == Operate.CLASSUPLOAD) {
+                Thread.sleep(1000);
+            }
+
+        } catch (Exception e) {
+            // TODO: handle exception
+        }
+
         message.setOperate(operate);
         message.setData(objectMapper.valueToTree(object));
 

diff --git a/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/message/ClassFilter.java b/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/message/ClassFilter.java
@@ -1,6 +1,7 @@
 package com.security.smith.client.message;
 
 
+import java.time.Instant;
 import java.util.UUID;
 
 import com.fasterxml.jackson.databind.annotation.JsonSerialize;
@@ -90,5 +91,19 @@ public StackTraceElement[] getStackTrace() {
     public void setStackTrace(StackTraceElement[] stackTrace) {
         this.stackTrace = stackTrace;
     }
+    @Override
+    public String toString() {
+        return "{" +
+                "transId: '" + transId + '\'' +
+                ", className: '" + className + '\'' +
+                ", classPath: '" + classPath + '\'' +
+                ", interfacesName: '" + interfacesName + '\'' +
+                ", classLoaderName: '" + classLoaderName + '\'' +
+                ", parentClassName: '" + parentClassName + '\'' +
+                ", parentClassLoaderName: '" + parentClassLoaderName + '\'' +
+                ", ruleId: " + ruleId +
+                ", timestamp: " + Instant.now().getEpochSecond() +
+                '}';
+    }
 
 }
diff --git a/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/message/ClassUpload.java b/rasp/jvm/JVMProbe/src/main/java/com/security/smith/client/message/ClassUpload.java
@@ -1,5 +1,7 @@
 package com.security.smith.client.message;
 
+import java.time.Instant;
+import java.util.Arrays;
 
 public class ClassUpload {
 
@@ -49,4 +51,14 @@ public void setClassData(byte[] class_data) {
         this.classData = class_data;
     }
 
+    @Override
+    public String toString() {
+        return "{" +
+                "transId: '" + transId + '\'' +
+                ", byteTotalLength: " + byteTotalLength +
+                ", byteOffset: " + byteOffset +
+                ", byteLength: " + byteLength +
+                ", timestamp: " + Instant.now().getEpochSecond() +
+                '}';
+    }
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -75,6 +75,7 @@ private void sendClass(Class<?> clazz, byte[] data) { @@
                     if (client != null) {
                         client.write(Operate.CLASSUPLOAD, classUpload);
+                        SmithLogger.logger.info("send classdata: " + classUpload.toString());
                     }
                 }
@@ Expand Down @@