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.ExceptionsAttribute;
25 import proguard.classfile.constant.ClassConstant;
26 import proguard.classfile.constant.visitor.ConstantVisitor;
27 import proguard.classfile.util.SimplifiedVisitor;
28 
29 /**
30  * This ConstantVisitor adds all class constants that it visits to the given
31  * target exceptions attribute.
32  *
33  * @author Eric Lafortune
34  */
35 public class ExceptionAdder
36 extends      SimplifiedVisitor
37 implements   ConstantVisitor
38 {
39     private final ConstantAdder             constantAdder;
40     private final ExceptionsAttributeEditor exceptionsAttributeEditor;
41 
42 
43     /**
44      * Creates a new ExceptionAdder that will copy classes into the given
45      * target exceptions attribute.
46      */
ExceptionAdder(ProgramClass targetClass, ExceptionsAttribute targetExceptionsAttribute)47     public ExceptionAdder(ProgramClass        targetClass,
48                           ExceptionsAttribute targetExceptionsAttribute)
49     {
50         constantAdder             = new ConstantAdder(targetClass);
51         exceptionsAttributeEditor = new ExceptionsAttributeEditor(targetExceptionsAttribute);
52     }
53 
54 
55     // Implementations for ConstantVisitor.
56 
visitClassConstant(Clazz clazz, ClassConstant classConstant)57     public void visitClassConstant(Clazz clazz, ClassConstant classConstant)
58     {
59         // Add a class constant to the constant pool.
60         constantAdder.visitClassConstant(clazz, classConstant);
61 
62         // Add the index of the class constant to the list of exceptions.
63         exceptionsAttributeEditor.addException(constantAdder.getConstantIndex());
64     }
65 }
66