Java tutorial
/* * This file is part of Applied Energistics 2. * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved. * * Applied Energistics 2 is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * Applied Energistics 2 is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with Applied Energistics 2. If not, see <http://www.gnu.org/licenses/lgpl>. */ package appeng.transformer.asm; import java.util.Iterator; import javax.annotation.Nullable; import org.apache.logging.log4j.Level; import org.objectweb.asm.ClassReader; import org.objectweb.asm.ClassWriter; import org.objectweb.asm.Type; import org.objectweb.asm.tree.AnnotationNode; import org.objectweb.asm.tree.ClassNode; import org.objectweb.asm.tree.MethodNode; import net.minecraft.launchwrapper.IClassTransformer; import cpw.mods.fml.relauncher.FMLRelaunchLog; import appeng.helpers.Reflected; import appeng.integration.IntegrationRegistry; import appeng.integration.IntegrationType; import appeng.transformer.annotations.Integration; @Reflected public final class ASMIntegration implements IClassTransformer { @Reflected public ASMIntegration() { /** * Side, Display Name, ModID ClassPostFix */ for (IntegrationType type : IntegrationType.values()) { IntegrationRegistry.INSTANCE.add(type); } // integrationModules.add( IntegrationSide.BOTH, "Thermal Expansion", "ThermalExpansion", IntegrationType.TE ); // integrationModules.add( IntegrationSide.BOTH, "Mystcraft", "Mystcraft", IntegrationType.Mystcraft ); // integrationModules.add( IntegrationSide.BOTH, "Greg Tech", "gregtech_addon", IntegrationType.GT ); // integrationModules.add( IntegrationSide.BOTH, "Universal Electricity", null, IntegrationType.UE ); // integrationModules.add( IntegrationSide.BOTH, "Logistics Pipes", "LogisticsPipes|Main", IntegrationType.LP ); // integrationModules.add( IntegrationSide.BOTH, "Better Storage", IntegrationType.betterstorage ); // integrationModules.add( IntegrationSide.BOTH, "Forestry", "Forestry", IntegrationType.Forestry ); // integrationModules.add( IntegrationSide.BOTH, "Mekanism", "Mekanism", IntegrationType.Mekanism ); } @Nullable @Override public byte[] transform(String name, String transformedName, byte[] basicClass) { if (basicClass == null || transformedName.startsWith("appeng.transformer")) { return basicClass; } if (transformedName.startsWith("appeng.")) { ClassNode classNode = new ClassNode(); ClassReader classReader = new ClassReader(basicClass); classReader.accept(classNode, 0); try { boolean reWrite = this.removeOptionals(classNode); if (reWrite) { ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_MAXS); classNode.accept(writer); return writer.toByteArray(); } } catch (Throwable t) { t.printStackTrace(); } } return basicClass; } private boolean removeOptionals(ClassNode classNode) { boolean changed = false; if (classNode.visibleAnnotations != null) { for (AnnotationNode an : classNode.visibleAnnotations) { if (this.hasAnnotation(an, Integration.Interface.class)) { if (this.stripInterface(classNode, Integration.Interface.class, an)) { changed = true; } } else if (this.hasAnnotation(an, Integration.InterfaceList.class)) { for (Object o : ((Iterable) an.values.get(1))) { if (this.stripInterface(classNode, Integration.InterfaceList.class, (AnnotationNode) o)) { changed = true; } } } } } Iterator<MethodNode> i = classNode.methods.iterator(); while (i.hasNext()) { MethodNode mn = i.next(); if (mn.visibleAnnotations != null) { for (AnnotationNode an : mn.visibleAnnotations) { if (this.hasAnnotation(an, Integration.Method.class)) { if (this.stripMethod(classNode, mn, i, Integration.Method.class, an)) { changed = true; } } } } } if (changed) { this.log("Updated " + classNode.name); } return changed; } private boolean hasAnnotation(AnnotationNode ann, Class<?> annotation) { return ann.desc.equals(Type.getDescriptor(annotation)); } private boolean stripInterface(ClassNode classNode, Class<?> class1, AnnotationNode an) { if (an.values.size() != 4) { throw new IllegalArgumentException("Unable to handle Interface annotation on " + classNode.name); } String iFace = null; String iName = null; if (an.values.get(0).equals("iface")) { iFace = (String) an.values.get(1); } else if (an.values.get(2).equals("iface")) { iFace = (String) an.values.get(3); } if (an.values.get(0).equals("iname")) { iName = (String) an.values.get(1); } else if (an.values.get(2).equals("iname")) { iName = (String) an.values.get(3); } if (iName != null && iFace != null) { final IntegrationType type = IntegrationType.valueOf(iName); if (!IntegrationRegistry.INSTANCE.isEnabled(type)) { this.log("Removing Interface " + iFace + " from " + classNode.name + " because " + iName + " integration is disabled."); classNode.interfaces.remove(iFace.replace('.', '/')); return true; } else { this.log("Allowing Interface " + iFace + " from " + classNode.name + " because " + iName + " integration is enabled."); } } else { throw new IllegalStateException("Unable to handle Method annotation on " + classNode.name); } return false; } private boolean stripMethod(ClassNode classNode, MethodNode mn, Iterator<MethodNode> i, Class class1, AnnotationNode an) { if (an.values.size() != 2) { throw new IllegalArgumentException("Unable to handle Method annotation on " + classNode.name); } String iName = null; if (an.values.get(0).equals("iname")) { iName = (String) an.values.get(1); } if (iName != null) { IntegrationType type = IntegrationType.valueOf(iName); if (!IntegrationRegistry.INSTANCE.isEnabled(type)) { this.log("Removing Method " + mn.name + " from " + classNode.name + " because " + iName + " integration is disabled."); i.remove(); return true; } else { this.log("Allowing Method " + mn.name + " from " + classNode.name + " because " + iName + " integration is enabled."); } } else { throw new IllegalStateException("Unable to handle Method annotation on " + classNode.name); } return false; } private void log(String string) { FMLRelaunchLog.log("AE2-CORE", Level.INFO, string); } }