KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > proguard > obfuscate > SourceFileRenamer


1 /*
2  * ProGuard -- shrinking, optimization, obfuscation, and preverification
3  * of Java bytecode.
4  *
5  * Copyright (c) 2002-2007 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.obfuscate;
22
23 import proguard.classfile.*;
24 import proguard.classfile.attribute.*;
25 import proguard.classfile.attribute.visitor.AttributeVisitor;
26 import proguard.classfile.editor.ConstantPoolEditor;
27 import proguard.classfile.util.SimplifiedVisitor;
28 import proguard.classfile.visitor.ClassVisitor;
29
30 /**
31  * This ClassVisitor changes the name stored in the source file attributes
32  * and source dir attributes of the classes that it visits, if the
33  * attributes are present.
34  *
35  * @author Eric Lafortune
36  */
37 public class SourceFileRenamer
38 extends SimplifiedVisitor
39 implements ClassVisitor,
40              AttributeVisitor
41 {
42     private ConstantPoolEditor constantPoolEditor = new ConstantPoolEditor();
43
44     private String JavaDoc newSourceFileAttribute;
45
46
47     /**
48      * Creates a new SourceFileRenamer.
49      * @param newSourceFileAttribute the new string to be put in the source file
50      * attributes.
51      */
52     public SourceFileRenamer(String JavaDoc newSourceFileAttribute)
53     {
54         this.newSourceFileAttribute = newSourceFileAttribute;
55     }
56
57
58     // Implementations for ClassVisitor.
59
60     public void visitProgramClass(ProgramClass programClass)
61     {
62         // Only visit the class attributes.
63 programClass.attributesAccept(this);
64     }
65
66
67     public void visitLibraryClass(LibraryClass libraryClass)
68     {
69         // Library classes don't have attributes.
70 }
71
72
73     // Implementations for AttributeVisitor.
74
75     public void visitAnyAttribute(Clazz clazz, Attribute attribute) {}
76
77
78     public void visitSourceFileAttribute(Clazz clazz, SourceFileAttribute sourceFileAttribute)
79     {
80         // Fix the source file attribute.
81 sourceFileAttribute.u2sourceFileIndex =
82             constantPoolEditor.addUtf8Constant((ProgramClass)clazz,
83                                                newSourceFileAttribute);
84     }
85
86
87     public void visitSourceDirAttribute(Clazz clazz, SourceDirAttribute sourceDirAttribute)
88     {
89         // Fix the source file attribute.
90 sourceDirAttribute.u2sourceDirIndex =
91             constantPoolEditor.addUtf8Constant((ProgramClass)clazz,
92                                                newSourceFileAttribute);
93     }
94 }
95
Popular Tags