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

refactor: simplify code

Browse Source
This commit is contained in:
ReaJason
2026-01-12 02:12:34 +08:00
parent b4f8e62fc7
commit c0a059fbdc
1 changed files with 20 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/main/java/com/reajason/javaweb/probe/payload/filter/JettyFilterProbe.java
+20 -39
View File
@@ -36,7 +36,7 @@ public class JettyFilterProbe {
} }
private List<Map<String, String>> collectFiltersData(Object context) { private List<Map<String, String>> collectFiltersData(Object context) {
Map<String, Map<String, Object>> aggregatedData = new LinkedHashMap<>(); List<Map<String, String>> result = new ArrayList<>();
try { try {
Object servletHandler = getFieldValue(context, "_servletHandler"); Object servletHandler = getFieldValue(context, "_servletHandler");
if (servletHandler == null) return Collections.emptyList(); if (servletHandler == null) return Collections.emptyList();
@@ -46,54 +46,35 @@ public class JettyFilterProbe {
for (Object mapping : filterMappings) { for (Object mapping : filterMappings) {
String name = (String) invokeMethod(mapping, "getFilterName"); String name = (String) invokeMethod(mapping, "getFilterName");
if (name == null) continue; if (name == null) continue;
if (!aggregatedData.containsKey(name)) { String filterClass = "N/A";
String filterClass = "N/A"; if (filterHolders != null) {
if (filterHolders != null) { for (Object holder : filterHolders) {
for (Object holder : filterHolders) { String holderName = (String) invokeMethod(holder, "getName");
String holderName = (String) invokeMethod(holder, "getName"); if (!name.equals(holderName)) continue;
if (!name.equals(holderName)) continue; String cls = (String) invokeMethod(holder, "getClassName");
String cls = (String) invokeMethod(holder, "getClassName"); if (cls == null) {
if (cls == null) { Object filterInstance = invokeMethod(holder, "getFilter");
Object filterInstance = invokeMethod(holder, "getFilter"); if (filterInstance != null) {
if (filterInstance != null) { cls = filterInstance.getClass().getName();
cls = filterInstance.getClass().getName();
}
} }
if (cls != null) filterClass = cls;
break;
} }
if (cls != null) filterClass = cls;
break;
} }
Map<String, Object> info = new HashMap<>();
info.put("filterName", name);
info.put("filterClass", filterClass);
info.put("urlPatterns", new LinkedHashSet<String>());
aggregatedData.put(name, info);
} }
Map<String, Object> info = aggregatedData.get(name); Map<String, String> info = new HashMap<>();
String[] pathSpecs = null; info.put("filterName", name);
info.put("filterClass", filterClass);
try { try {
pathSpecs = (String[]) invokeMethod(mapping, "getPathSpecs"); String[] pathSpecs = (String[]) invokeMethod(mapping, "getPathSpecs");
info.put("urlPatterns", pathSpecs.length == 0 ? "" : Arrays.toString(pathSpecs));
} catch (Exception ignored) { } catch (Exception ignored) {
} }
result.add(info);
if (pathSpecs != null) {
((Set<String>) info.get("urlPatterns")).addAll(Arrays.asList(pathSpecs));
}
} }
} catch (Exception ignored) { } catch (Exception ignored) {
} }
List<Map<String, String>> result = new ArrayList<>();
for (Map<String, Object> entry : aggregatedData.values()) {
Map<String, String> finalInfo = new HashMap<>();
finalInfo.put("filterName", (String) entry.get("filterName"));
finalInfo.put("filterClass", (String) entry.get("filterClass"));
Set<?> urls = (Set<?>) entry.get("urlPatterns");
finalInfo.put("urlPatterns", urls.isEmpty() ? "" : urls.toString());
result.add(finalInfo);
}
return result; return result;
} }