refactor: simplify CommandProbe

Author: ReaJason

generator/build.gradle.kts

@@ -41,6 +41,7 @@ dependencies {
     implementation(libs.bcel)
     implementation(libs.jackson.databind)
     testImplementation(libs.junit.jupiter)
+    testImplementation(libs.hamcrest)
     testRuntimeOnly(libs.junit.platform.launcher)
     testImplementation(libs.bundles.mockito)
 }

generator/src/main/java/com/reajason/javaweb/probe/payload/CommandProbe.java

@@ -3,7 +3,6 @@
 import lombok.SneakyThrows;
 import net.bytebuddy.asm.Advice;
 
-import java.util.NoSuchElementException;
 import java.util.Scanner;
 
 /**
@@ -20,12 +19,8 @@ public CommandProbe(String command) {
     @Advice.OnMethodExit
     public static String exit(@Advice.Argument(0) String data, @Advice.Return(readOnly = false) String ret) throws Exception {
         String[] cmd = System.getProperty("os.name").toLowerCase().contains("window") ? new String[]{"cmd.exe", "/c", data} : new String[]{"/bin/sh", "-c", data};
-        Process process = new ProcessBuilder(cmd).start();
-        try {
-            return ret = new Scanner(process.getInputStream()).useDelimiter("\\A").next();
-        } catch (NoSuchElementException e) {
-            return ret = new Scanner(process.getErrorStream()).useDelimiter("\\A").next();
-        }
+        Process process = new ProcessBuilder(cmd).redirectErrorStream(true).start();
+        return ret = new Scanner(process.getInputStream()).useDelimiter("\\A").next();
     }
 
     @Override

generator/src/test/java/com/reajason/javaweb/probe/payload/CommandProbeTest.java

@@ -0,0 +1,24 @@
+package com.reajason.javaweb.probe.payload;
+
+import org.junit.jupiter.api.Disabled;
+import org.junit.jupiter.api.Test;
+
+import static org.hamcrest.CoreMatchers.anyOf;
+import static org.hamcrest.CoreMatchers.containsString;
+import static org.hamcrest.MatcherAssert.assertThat;
+
+/**
+ * @author ReaJason
+ * @since 2025/8/26
+ */
+class CommandProbeTest {
+
+    @Test
+    @Disabled
+    void test() {
+        String result = new CommandProbe("hello").toString();
+        assertThat(result, anyOf(
+                containsString("not found")
+        ));
+    }
+}