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

feat: support servletNameFilter

Browse Source
This commit is contained in:
ReaJason
2026-01-12 02:12:34 +08:00
parent 3fe9a932ba
commit 117b48a9d0
16 changed files with 287 additions and 277 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/main/java/com/reajason/javaweb/probe/payload/FilterProbeFactory.java
+2
View File
@@ -45,6 +45,8 @@ public class FilterProbeFactory {
return WebLogicFilterProbe.class;
case Server.Undertow:
return UndertowFilterProbe.class;
case Server.Resin:
return ResinFilterProbe.class;
default:
throw new GenerationException("filterProbe not supported for server: " + server);
}
generator/src/main/java/com/reajason/javaweb/probe/payload/filter/ApusicFilterProbe.java
+57 -66
View File
@@ -13,73 +13,58 @@ public class ApusicFilterProbe {
@Override
public String toString() {
String msg = "";
StringBuilder msg = new StringBuilder();
Map<String, List<Map<String, String>>> allFiltersData = new LinkedHashMap<String, List<Map<String, String>>>();
Set<Object> contexts = null;
try {
contexts = getContext();
} catch (Throwable throwable) {
msg += "context error: " + getErrorMessage(throwable);
msg.append("context error: ").append(getErrorMessage(throwable));
}
if (contexts == null || contexts.isEmpty()) {
msg += "context not found\n";
msg.append("context not found\n");
} else {
for (Object context : contexts) {
String contextRoot = getContextRoot(context);
List<Map<String, String>> filters = collectFiltersData(context);
allFiltersData.put(contextRoot, filters);
try {
List<Map<String, String>> filters = collectFiltersData(context);
allFiltersData.put(contextRoot, filters);
} catch (Throwable e) {
msg.append(contextRoot).append(" failed ").append(getErrorMessage(e)).append("\n");
}
}
msg += formatFiltersData(allFiltersData);
msg.append(formatFiltersData(allFiltersData));
}
return msg;
return msg.toString();
}
@SuppressWarnings("all")
private List<Map<String, String>> collectFiltersData(Object context) {
private List<Map<String, String>> collectFiltersData(Object context) throws Exception {
Map<String, Map<String, Object>> aggregatedData = new LinkedHashMap<>();
Object webModule = getFieldValue(context, "webapp");
Object[] filterMappings = (Object[]) invokeMethod(webModule, "getAllFilterMappings");
Map<String, String> filterClassMap = new HashMap<>();
try {
Object webModule = getFieldValue(context, "webapp");
if (webModule == null) {
return Collections.emptyList();
for (Object fm : filterMappings) {
String name = (String) invokeMethod(fm, "getFilterName");
if (!aggregatedData.containsKey(name)) {
Map<String, Object> info = new HashMap<>();
Object filterModel = invokeMethod(webModule, "getFilter", new Class[]{String.class}, new Object[]{name});
info.put("filterName", name);
info.put("filterClass", invokeMethod(filterModel, "getFilterClass"));
info.put("urlPatterns", new LinkedHashSet<String>());
info.put("servletNames", new LinkedHashSet<String>());
aggregatedData.put(name, info);
}
Object[] filterMappings = (Object[]) invokeMethod(webModule, "getAllFilterMappings");
if (filterMappings == null || filterMappings.length == 0) {
return Collections.emptyList();
Map<String, Object> info = aggregatedData.get(name);
String urlPattern = (String) invokeMethod(fm, "getUrlPattern");
if (urlPattern != null) {
((Set<String>) info.get("urlPatterns")).add(urlPattern);
}
Object[] filters = (Object[]) invokeMethod(webModule, "getFilterList");
Map<String, String> filterClassMap = new HashMap<>();
if (filters != null) {
for (Object filter : filters) {
String name = (String) invokeMethod(filter, "getName");
String className = (String) invokeMethod(filter, "getFilterClass");
if (name != null && className != null) {
filterClassMap.put(name, className);
}
}
String servletName = (String) invokeMethod(fm, "getServletName");
if (servletName != null) {
((Set<String>) info.get("servletNames")).add(servletName);
}
for (Object fm : filterMappings) {
String name = (String) invokeMethod(fm, "getFilterName");
if (name == null) {
continue;
}
if (!aggregatedData.containsKey(name)) {
Map<String, Object> info = new HashMap<>();
info.put("filterName", name);
info.put("filterClass", filterClassMap.getOrDefault(name, "N/A"));
info.put("urlPatterns", new LinkedHashSet<String>());
aggregatedData.put(name, info);
}
Map<String, Object> info = aggregatedData.get(name);
String urlPattern = (String) invokeMethod(fm, "getUrlPattern");
if (urlPattern != null) {
((Set<String>) info.get("urlPatterns")).add(urlPattern);
}
}
} catch (Exception ignored) {
}
List<Map<String, String>> result = new ArrayList<>();
@@ -89,6 +74,8 @@ public class ApusicFilterProbe {
finalInfo.put("filterClass", (String) entry.get("filterClass"));
Set<?> urls = (Set<?>) entry.get("urlPatterns");
finalInfo.put("urlPatterns", urls.isEmpty() ? "" : urls.toString());
Set<?> servletNames = (Set<?>) entry.get("servletNames");
finalInfo.put("servletNames", servletNames.isEmpty() ? "" : servletNames.toString());
result.add(finalInfo);
}
return result;
@@ -108,6 +95,7 @@ public class ApusicFilterProbe {
appendIfPresent(output, "", info.get("filterName"), "");
appendIfPresent(output, " -> ", info.get("filterClass"), "");
appendIfPresent(output, " -> URL:", info.get("urlPatterns"), "");
appendIfPresent(output, " -> Servlet:", info.get("servletNames"), "");
output.append("\n");
}
}
@@ -176,27 +164,30 @@ public class ApusicFilterProbe {
throw new NoSuchFieldException(fieldName + " for " + obj.getClass().getName());
}
@SuppressWarnings("all")
public static Object invokeMethod(Object obj, String methodName) {
try {
Class<?> clazz = (obj instanceof Class) ? (Class<?>) obj : obj.getClass();
Method method = null;
while (clazz != null && method == null) {
try {
method = clazz.getDeclaredMethod(methodName);
} catch (NoSuchMethodException e) {
clazz = clazz.getSuperclass();
}
}
if (method == null) {
throw new NoSuchMethodException("Method not found: " + methodName);
}
public static Object invokeMethod(Object obj, String methodName) throws Exception {
return invokeMethod(obj, methodName, null, null);
}
method.setAccessible(true);
return method.invoke(obj instanceof Class ? null : obj);
} catch (Exception e) {
throw new RuntimeException("Error invoking method: " + methodName, e);
@SuppressWarnings("all")
public static Object invokeMethod(Object obj, String methodName, Class<?>[] paramClazz, Object[] param) throws Exception {
Class<?> clazz = (obj instanceof Class) ? (Class<?>) obj : obj.getClass();
Method method = null;
while (clazz != null && method == null) {
try {
if (paramClazz == null) {
method = clazz.getDeclaredMethod(methodName);
} else {
method = clazz.getDeclaredMethod(methodName, paramClazz);
}
} catch (NoSuchMethodException e) {
clazz = clazz.getSuperclass();
}
}
if (method == null) {
throw new NoSuchMethodException("Method not found: " + methodName);
}
method.setAccessible(true);
return method.invoke(obj instanceof Class ? null : obj, param);
}
@SuppressWarnings("all")
generator/src/main/java/com/reajason/javaweb/probe/payload/filter/ResinFilterProbe.java
+64 -43
View File
@@ -13,73 +13,93 @@ public class ResinFilterProbe {
@Override
public String toString() {
String msg = "";
StringBuilder msg = new StringBuilder();
Map<String, List<Map<String, String>>> allFiltersData = new LinkedHashMap<String, List<Map<String, String>>>();
Set<Object> contexts = null;
try {
contexts = getContext();
} catch (Throwable throwable) {
msg += "context error: " + getErrorMessage(throwable);
msg.append("context error: ").append(getErrorMessage(throwable));
}
if (contexts == null || contexts.isEmpty()) {
msg += "context not found\n";
msg.append("context not found\n");
} else {
for (Object context : contexts) {
String contextRoot = getContextRoot(context);
List<Map<String, String>> filters = collectFiltersData(context);
allFiltersData.put(contextRoot, filters);
try {
List<Map<String, String>> filters = collectFiltersData(context);
allFiltersData.put(contextRoot, filters);
} catch (Throwable e) {
msg.append(contextRoot).append(" failed ").append(getErrorMessage(e)).append("\n");
}
}
msg += formatFiltersData(allFiltersData);
msg.append(formatFiltersData(allFiltersData));
}
return msg;
return msg.toString();
}
private List<Map<String, String>> collectFiltersData(Object context) {
List<Map<String, String>> result = new ArrayList<>();
try {
Object filterMapper = getFieldValue(context, "_filterMapper");
Object filterManager = getFieldValue(context, "_filterManager");
if (filterMapper == null) return Collections.emptyList();
ArrayList<Object> filterMappings = (ArrayList<Object>) getFieldValue(filterMapper, "_filterMap");
for (Object filterMapping : filterMappings) {
Map<String, String> info = new HashMap<>();
String filterName = (String) invokeMethod(filterMapping, "getFilterName", null, null);
@SuppressWarnings("unchecked")
private List<Map<String, String>> collectFiltersData(Object context) throws Exception {
Map<String, Map<String, Object>> aggregatedData = new LinkedHashMap<>();
Object filterMapper = getFieldValue(context, "_filterMapper");
Object filterManager = getFieldValue(context, "_filterManager");
if (filterMapper == null) return Collections.emptyList();
ArrayList<Object> filterMappings = (ArrayList<Object>) getFieldValue(filterMapper, "_filterMap");
for (Object filterMapping : filterMappings) {
String filterName = (String) invokeMethod(filterMapping, "getFilterName", null, null);
if (aggregatedData.get(filterName) == null) {
Map<String, Object> info = new HashMap<>();
info.put("filterName", filterName);
String filterClassName = (String) invokeMethod(filterMapping, "getFilterClassName", null, null);
try {
if (filterClassName == null) {
Class<?> filterClass = (Class<?>) invokeMethod(filterMapping, "getFilterClass", null, null);
if (filterClass != null) {
filterClassName = filterClass.getName();
} else {
Object filter = ((Map<String, Object>) getFieldValue(filterManager, "_instances")).get(filterName);
if (filter != null) {
filterClassName = filter.getClass().getName();
}
if (filterClassName == null) {
Class<?> filterClass = (Class<?>) invokeMethod(filterMapping, "getFilterClass", null, null);
if (filterClass != null) {
filterClassName = filterClass.getName();
} else {
Object filter = ((Map<String, Object>) getFieldValue(filterManager, "_instances")).get(filterName);
if (filter != null) {
filterClassName = filter.getClass().getName();
}
}
} catch (Exception ignored) {
}
info.put("filterClass", filterClassName != null ? filterClassName : "N/A");
List<String> urlPatterns = new ArrayList<>();
String urlPattern = invokeMethod(filterMapping, "getURLPattern", null, null).toString();
if (urlPattern == null || urlPattern.isEmpty()) {
List<Object> matchList = (List<Object>) getFieldValue(filterMapping, "_matchList");
if (matchList != null && !matchList.isEmpty()) {
for (Object match : matchList) {
if (((Integer) getFieldValue(match, "_value")) == 1) {
urlPatterns.add(getFieldValue(match, "_regex").toString());
}
info.put("urlPatterns", new LinkedHashSet<String>());
info.put("servletNames", new LinkedHashSet<String>());
aggregatedData.put(filterName, info);
}
Map<String, Object> info = aggregatedData.get(filterName);
List<String> urlPatterns = new ArrayList<>();
String urlPattern = (String) invokeMethod(filterMapping, "getURLPattern", null, null);
if (urlPattern == null || urlPattern.isEmpty()) {
List<Object> matchList = (List<Object>) getFieldValue(filterMapping, "_matchList");
if (matchList != null && !matchList.isEmpty()) {
for (Object match : matchList) {
if (((Integer) getFieldValue(match, "_value")) == 1) {
urlPatterns.add(getFieldValue(match, "_regex").toString());
}
}
} else {
urlPatterns.add(urlPattern);
}
info.put("urlPatterns", Arrays.toString(urlPatterns.toArray()));
result.add(info);
} else {
urlPatterns.add(urlPattern);
}
} catch (Exception ignored) {
if (!urlPatterns.isEmpty()) {
((Set) info.get("urlPatterns")).addAll(urlPatterns);
}
List<String> servletNames = (List<String>) getFieldValue(filterMapping, "_servletNames");
if (servletNames != null && !servletNames.isEmpty()) {
((Set) info.get("servletNames")).addAll(servletNames);
}
}
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());
Set<?> servletNames = (Set<?>) entry.get("servletNames");
finalInfo.put("servletNames", servletNames.isEmpty() ? "" : servletNames.toString());
result.add(finalInfo);
}
return result;
}
@@ -100,6 +120,7 @@ public class ResinFilterProbe {
appendIfPresent(output, "", info.get("filterName"), "");
appendIfPresent(output, " -> ", info.get("filterClass"), "");
appendIfPresent(output, " -> URL:", info.get("urlPatterns"), "");
appendIfPresent(output, " -> Servlet:", info.get("servletNames"), "");
output.append("\n");
}
}
generator/src/main/java/com/reajason/javaweb/probe/payload/filter/WebSphereFilterProbe.java
+73 -41
View File
@@ -13,89 +13,114 @@ public class WebSphereFilterProbe {
@Override
public String toString() {
String msg = "";
StringBuilder msg = new StringBuilder();
Map<String, List<Map<String, String>>> allFiltersData = new LinkedHashMap<String, List<Map<String, String>>>();
Set<Object> contexts = null;
try {
contexts = getContext();
} catch (Throwable throwable) {
msg += "context error: " + getErrorMessage(throwable);
msg.append("context error: ").append(getErrorMessage(throwable));
}
if (contexts == null || contexts.isEmpty()) {
msg += "context not found\n";
msg.append("context not found\n");
} else {
for (Object context : contexts) {
String contextRoot = getContextRoot(context);
List<Map<String, String>> filters = collectFiltersData(context);
allFiltersData.put(contextRoot, filters);
try {
List<Map<String, String>> filters = collectFiltersData(context);
allFiltersData.put(contextRoot, filters);
} catch (Throwable e) {
msg.append(contextRoot).append(" failed ").append(getErrorMessage(e)).append("\n");
}
}
msg += formatFiltersData(allFiltersData);
msg.append(formatFiltersData(allFiltersData));
}
return msg;
return msg.toString();
}
private List<Map<String, String>> collectFiltersData(Object context) {
private List<Map<String, String>> collectFiltersData(Object context) throws Exception {
Map<String, Map<String, Object>> aggregatedData = new LinkedHashMap<>();
try {
Object filterManager = getFieldValue(context, "filterManager");
Object webAppConfig = getFieldValue(context, "config");
try {
List uriFilterMappingInfos = (List) getFieldValue(webAppConfig, "uriFilterMappingInfos");
if (uriFilterMappingInfos == null || uriFilterMappingInfos.isEmpty()) {
return Collections.emptyList();
}
for (Object uriFilterMappingInfo : uriFilterMappingInfos) {
Object filterConfig = getFieldValue(uriFilterMappingInfo, "filterConfig");
String filterName = (String) invokeMethod(filterConfig, "getFilterName", null, null);
Collection urlPatternMappings = (Collection) invokeMethod(filterConfig, "getUrlPatternMappings", null, null);
String urlPattern = (String) invokeMethod(uriFilterMappingInfo, "getUrlPattern", null, null);
if (!aggregatedData.containsKey(filterName)) {
String filterClassName = (String) invokeMethod(filterConfig, "getFilterClassName", null, null);
Map<String, Object> info = new HashMap<>();
info.put("filterName", filterName);
info.put("filterClass", filterClassName);
LinkedHashSet<String> urlPatterns = new LinkedHashSet<>();
if (urlPattern != null && !urlPattern.isEmpty()) {
urlPatterns.add(urlPattern);
}
if (urlPatternMappings != null) {
urlPatterns.addAll(urlPatternMappings);
}
info.put("urlPatterns", urlPatterns);
info.put("urlPatterns", new LinkedHashSet<>());
info.put("servletNames", new LinkedHashSet<>());
aggregatedData.put(filterName, info);
} else {
Set urlPatterns = (Set) aggregatedData.get(filterName).get("urlPatterns");
if (urlPattern != null && !urlPattern.isEmpty()) {
urlPatterns.add(urlPattern);
}
if (urlPatternMappings != null) {
urlPatterns.addAll(urlPatternMappings);
}
}
Map<String, Object> info = aggregatedData.get(filterName);
String urlPattern = (String) invokeMethod(uriFilterMappingInfo, "getUrlPattern", null, null);
if (urlPattern != null && !urlPattern.isEmpty()) {
((Set) info.get("urlPatterns")).add(urlPattern);
}
}
List servletFilterMappingInfos = (List) getFieldValue(webAppConfig, "servletFilterMappingInfos");
for (Object servletFilterMappingInfo : servletFilterMappingInfos) {
Object filterConfig = getFieldValue(servletFilterMappingInfo, "filterConfig");
String filterName = (String) invokeMethod(filterConfig, "getFilterName", null, null);
if (!aggregatedData.containsKey(filterName)) {
String filterClassName = (String) invokeMethod(filterConfig, "getFilterClassName", null, null);
Map<String, Object> info = new HashMap<>();
info.put("filterName", filterName);
info.put("filterClass", filterClassName);
info.put("urlPatterns", new LinkedHashSet<>());
info.put("servletNames", new LinkedHashSet<>());
aggregatedData.put(filterName, info);
}
Map<String, Object> info = aggregatedData.get(filterName);
String servletName = (String) invokeMethod(invokeMethod(servletFilterMappingInfo, "getServletConfig"), "getServletName");
if (servletName != null && !servletName.isEmpty()) {
((Set) info.get("servletNames")).add(servletName);
}
}
} catch (Throwable throwable) {
throwable.printStackTrace();
// WebLogic 10.3.6
List uriFilterMappings = (List) getFieldValue(filterManager, "_uriFilterMappings");
for (Object uriFilterMapping : uriFilterMappings) {
String filterName = (String) getFieldValue(uriFilterMapping, "_filterName");
String urlPattern = (String) getFieldValue(uriFilterMapping, "_filterURI");
if (!aggregatedData.containsKey(filterName)) {
Object filterConfig = invokeMethod(webAppConfig, "getFilterInfo", new Class[]{String.class}, new Object[]{filterName});
String filterClassName = (String) invokeMethod(filterConfig, "getFilterClassName", null, null);
Map<String, Object> info = new HashMap<>();
info.put("filterName", filterName);
info.put("filterClass", filterClassName);
LinkedHashSet<String> urlPatterns = new LinkedHashSet<>();
if (urlPattern != null && !urlPattern.isEmpty()) {
urlPatterns.add(urlPattern);
}
info.put("urlPatterns", urlPatterns);
info.put("urlPatterns", new LinkedHashSet<>());
info.put("servletNames", new LinkedHashSet<>());
aggregatedData.put(filterName, info);
} else {
if (urlPattern != null && !urlPattern.isEmpty()) {
((Set) aggregatedData.get(filterName).get("urlPatterns")).add(urlPattern);
}
}
Map<String, Object> info = aggregatedData.get(filterName);
String urlPattern = (String) getFieldValue(uriFilterMapping, "_filterURI");
if (urlPattern != null && !urlPattern.isEmpty()) {
((Set) info.get("urlPatterns")).add(urlPattern);
}
}
List servletFilterMappings = (List) getFieldValue(filterManager, "_servletFilterMappings");
for (Object servletFilterMapping : servletFilterMappings) {
String filterName = (String) getFieldValue(servletFilterMapping, "_filterName");
if (!aggregatedData.containsKey(filterName)) {
Object filterConfig = invokeMethod(webAppConfig, "getFilterInfo", new Class[]{String.class}, new Object[]{filterName});
String filterClassName = (String) invokeMethod(filterConfig, "getFilterClassName", null, null);
Map<String, Object> info = new HashMap<>();
info.put("filterName", filterName);
info.put("filterClass", filterClassName);
info.put("urlPatterns", new LinkedHashSet<>());
info.put("servletNames", new LinkedHashSet<>());
aggregatedData.put(filterName, info);
}
Map<String, Object> info = aggregatedData.get(filterName);
String servletName = (String) getFieldValue(servletFilterMapping, "_filterServlet");
if (servletName != null && !servletName.isEmpty()) {
((Set) info.get("servletNames")).add(servletName);
}
}
}
@@ -108,7 +133,9 @@ public class WebSphereFilterProbe {
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() ? "[/*]" : Arrays.toString(urls.toArray()));
finalInfo.put("urlPatterns", urls.isEmpty() ? "" : urls.toString());
Set<?> servletNames = (Set<?>) entry.get("servletNames");
finalInfo.put("servletNames", servletNames.isEmpty() ? "" : servletNames.toString());
result.add(finalInfo);
}
return result;
@@ -130,6 +157,7 @@ public class WebSphereFilterProbe {
appendIfPresent(output, "", info.get("filterName"), "");
appendIfPresent(output, " -> ", info.get("filterClass"), "");
appendIfPresent(output, " -> URL:", info.get("urlPatterns"), "");
appendIfPresent(output, " -> Servlet:", info.get("servletNames"), "");
output.append("\n");
}
}
@@ -208,6 +236,10 @@ public class WebSphereFilterProbe {
return contexts;
}
public static Object invokeMethod(Object obj, String methodName) {
return invokeMethod(obj, methodName, null, null);
}
@SuppressWarnings("all")
public static Object invokeMethod(Object obj, String methodName, Class<?>[] paramClazz, Object[] param) {
try {