1 /* 2 * ProGuard -- shrinking, optimization, obfuscation, and preverification 3 * of Java bytecode. 4 * 5 * Copyright (c) 2002-2014 Eric Lafortune (eric@graphics.cornell.edu) 6 * 7 * This program is free software; you can redistribute it and/or modify it 8 * under the terms of the GNU General Public License as published by the Free 9 * Software Foundation; either version 2 of the License, or (at your option) 10 * any later version. 11 * 12 * This program is distributed in the hope that it will be useful, but WITHOUT 13 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 14 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 15 * more details. 16 * 17 * You should have received a copy of the GNU General Public License along 18 * with this program; if not, write to the Free Software Foundation, Inc., 19 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 20 */ 21 package proguard.classfile.editor; 22 23 import proguard.classfile.*; 24 import proguard.classfile.attribute.*; 25 import proguard.classfile.attribute.visitor.AttributeVisitor; 26 import proguard.classfile.util.SimplifiedVisitor; 27 import proguard.classfile.visitor.*; 28 29 import java.util.*; 30 31 /** 32 * This ClassVisitor sorts the attributes of the classes that it visits. 33 * The sorting order is based on the types of the attributes. 34 * 35 * @author Eric Lafortune 36 */ 37 public class AttributeSorter 38 extends SimplifiedVisitor 39 implements ClassVisitor, MemberVisitor, AttributeVisitor, Comparator 40 { 41 // Implementations for ClassVisitor. 42 visitProgramClass(ProgramClass programClass)43 public void visitProgramClass(ProgramClass programClass) 44 { 45 // Sort the attributes. 46 Arrays.sort(programClass.attributes, 0, programClass.u2attributesCount, this); 47 48 // Sort the attributes of the class members. 49 programClass.fieldsAccept(this); 50 programClass.methodsAccept(this); 51 } 52 53 54 // Implementations for MemberVisitor. 55 visitProgramMember(ProgramClass programClass, ProgramMember programMember)56 public void visitProgramMember(ProgramClass programClass, ProgramMember programMember) 57 { 58 // Sort the attributes. 59 Arrays.sort(programMember.attributes, 0, programMember.u2attributesCount, this); 60 61 // Sort the attributes of the attributes. 62 programMember.attributesAccept(programClass, this); 63 } 64 65 66 // Implementations for AttributeVisitor. 67 visitAnyAttribute(Clazz clazz, Attribute attribute)68 public void visitAnyAttribute(Clazz clazz, Attribute attribute) {} 69 70 visitCodeAttribute(Clazz clazz, Method method, CodeAttribute codeAttribute)71 public void visitCodeAttribute(Clazz clazz, Method method, CodeAttribute codeAttribute) 72 { 73 // Sort the attributes. 74 Arrays.sort(codeAttribute.attributes, 0, codeAttribute.u2attributesCount, this); 75 } 76 77 78 // Implementations for Comparator. 79 compare(Object object1, Object object2)80 public int compare(Object object1, Object object2) 81 { 82 Attribute attribute1 = (Attribute)object1; 83 Attribute attribute2 = (Attribute)object2; 84 85 return attribute1.u2attributeNameIndex < attribute2.u2attributeNameIndex ? -1 : 86 attribute1.u2attributeNameIndex > attribute2.u2attributeNameIndex ? 1 : 87 0; 88 } 89 } 90