/*
* Copyright 2009 Sun Microsystems, Inc. All Rights Reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code 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 General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
* CA 95054 USA or visit www.sun.com if you need additional information or
* have any questions.
*/
package com.sun.classanalyzer;
import com.sun.tools.classfile.*;
import com.sun.tools.classfile.ConstantPool.*;
import static com.sun.tools.classfile.ConstantPool.*;
/**
*
* @author Mandy Chung
*/
public class ConstantPoolParser {
private final ClassFileParser cfparser;
private final StringValueVisitor visitor;
private final ConstantPool cpool;
ConstantPoolParser(ClassFileParser parser) {
this.cfparser = parser;
this.cpool = cfparser.classfile.constant_pool;
this.visitor = new StringValueVisitor();
}
public String stringValue(CPInfo cpInfo) {
return visitor.visit(cpInfo);
}
public String stringValue(int constant_pool_index) {
try {
return stringValue(cpool.get(constant_pool_index));
} catch (ConstantPool.InvalidIndex e) {
throw new RuntimeException(e);
}
}
public void parseDependency() {
ConstantPool.Visitor<Integer, Void> v = new ConstantPool.Visitor<Integer, Void>() {
public Integer visitClass(CONSTANT_Class_info info, Void p) {
try {
String classname = cfparser.checkClassName(info.getName());
if (classname.isEmpty()) {
return 1;
}
Klass from = cfparser.this_klass;
Klass to = Klass.getKlass(classname);
ResolutionInfo resInfo = ResolutionInfo.resolvedConstantPool(from, to, info.name_index);
from.addDep(to, resInfo);
to.addReferrer(from, resInfo);
} catch (ConstantPoolException ex) {
throw new RuntimeException(ex);
}
return 1;
}
public Integer visitDouble(CONSTANT_Double_info info, Void p) {
// skip
return 2;
}
public Integer visitFieldref(CONSTANT_Fieldref_info info, Void p) {
// skip
return 1;
}
public Integer visitFloat(CONSTANT_Float_info info, Void p) {
// skip
return 1;
}
public Integer visitInteger(CONSTANT_Integer_info info, Void p) {
// skip
return 1;
}
public Integer visitInterfaceMethodref(CONSTANT_InterfaceMethodref_info info, Void p) {
// skip
return 1;
}
public Integer visitLong(CONSTANT_Long_info info, Void p) {
// skip
return 2;
}
public Integer visitNameAndType(CONSTANT_NameAndType_info info, Void p) {
// skip
return 1;
}
public Integer visitMethodref(CONSTANT_Methodref_info info, Void p) {
// skip
return 1;
}
public Integer visitString(CONSTANT_String_info info, Void p) {
// skip
return 1;
}
public Integer visitUtf8(CONSTANT_Utf8_info info, Void p) {
// skip
return 1;
}
};
int cpx = 1;
while (cpx < cpool.size()) {
try {
CPInfo cpInfo = cpool.get(cpx);
cpx += cpInfo.accept(v, null);
} catch (ConstantPool.InvalidIndex ex) {
throw new RuntimeException(ex);
}
}
}
int getTag(int index) {
try {
return cpool.get(index).getTag();
} catch (ConstantPoolException ex) {
throw new RuntimeException(ex);
}
}
String getDescriptor(int index) {
CPInfo cpInfo;
try {
cpInfo = cpool.get(index);
} catch (ConstantPoolException ex) {
throw new RuntimeException(ex);
}
int tag = cpInfo.getTag();
switch (tag) {
case CONSTANT_Methodref:
case CONSTANT_InterfaceMethodref:
case CONSTANT_Fieldref:
// simplify references within this class
CPRefInfo ref = (CPRefInfo) cpInfo;
try {
return ref.getNameAndTypeInfo().getType();
} catch (ConstantPoolException ex) {
}
}
return stringValue(cpInfo);
}
String getMethodName(int index) {
try {
CPInfo cpInfo = cpool.get(index);
if (cpInfo.getTag() == CONSTANT_Methodref ||
cpInfo.getTag() == CONSTANT_InterfaceMethodref) {
// simplify references within this class
CPRefInfo ref = (CPRefInfo) cpInfo;
String classname;
if (ref.class_index == cfparser.classfile.this_class) {
classname = cfparser.this_klass.getClassName();
} else {
classname = cfparser.checkClassName(ref.getClassName()).replace('/', '.');
}
String methodname = ref.getNameAndTypeInfo().getName();
return classname + "." + methodname;
} else {
return null;
}
} catch (InvalidIndex ex) {
throw new RuntimeException(ex);
} catch (ConstantPoolException ex) {
throw new RuntimeException(ex);
}
}
class StringValueVisitor implements ConstantPool.Visitor<String, Void> {
public StringValueVisitor() {
}
public String visit(CPInfo info) {
return info.accept(this, null);
}
public String visitClass(CONSTANT_Class_info info, Void p) {
return getCheckedName(info);
}
String getCheckedName(CONSTANT_Class_info info) {
try {
return checkName(info.getName());
} catch (ConstantPoolException e) {
throw new RuntimeException(e);
}
}
public String visitDouble(CONSTANT_Double_info info, Void p) {
return info.value + "d";
}
public String visitFieldref(CONSTANT_Fieldref_info info, Void p) {
return visitRef(info, p);
}
public String visitFloat(CONSTANT_Float_info info, Void p) {
return info.value + "f";
}
public String visitInteger(CONSTANT_Integer_info info, Void p) {
return String.valueOf(info.value);
}
public String visitInterfaceMethodref(CONSTANT_InterfaceMethodref_info info, Void p) {
return visitRef(info, p);
}
public String visitLong(CONSTANT_Long_info info, Void p) {
return info.value + "l";
}
public String visitNameAndType(CONSTANT_NameAndType_info info, Void p) {
return getCheckedName(info) + ":" + getType(info);
}
String getCheckedName(CONSTANT_NameAndType_info info) {
try {
return checkName(info.getName());
} catch (ConstantPoolException e) {
throw new RuntimeException(e);
}
}
String getType(CONSTANT_NameAndType_info info) {
try {
return info.getType();
} catch (ConstantPoolException e) {
throw new RuntimeException(e);
}
}
public String visitMethodref(CONSTANT_Methodref_info info, Void p) {
return visitRef(info, p);
}
public String visitString(CONSTANT_String_info info, Void p) {
try {
int string_index = info.string_index;
return cpool.getUTF8Info(string_index).accept(this, p);
} catch (ConstantPoolException e) {
throw new RuntimeException(e);
}
}
public String visitUtf8(CONSTANT_Utf8_info info, Void p) {
String s = info.value;
StringBuilder sb = new StringBuilder();
for (int i = 0; i < s.length(); i++) {
char c = s.charAt(i);
switch (c) {
case '\t':
sb.append('\\').append('t');
break;
case '\n':
sb.append('\\').append('n');
break;
case '\r':
sb.append('\\').append('r');
break;
case '\"':
sb.append('\\').append('\"');
break;
default:
sb.append(c);
}
}
return sb.toString();
}
String visitRef(CPRefInfo info, Void p) {
String cn = getCheckedClassName(info);
String nat;
try {
nat = info.getNameAndTypeInfo().accept(this, p);
} catch (ConstantPoolException e) {
nat = e.getMessage();
}
return cn + "." + nat;
}
String getCheckedClassName(CPRefInfo info) {
try {
return checkName(info.getClassName());
} catch (ConstantPoolException e) {
throw new RuntimeException(e);
}
}
}
/* If name is a valid binary name, return it; otherwise quote it. */
private static String checkName(String name) {
if (name == null) {
return "null";
}
int len = name.length();
if (len == 0) {
return "\"\"";
}
int cc = '/';
int cp;
for (int k = 0; k < len; k += Character.charCount(cp)) {
cp = name.codePointAt(k);
if ((cc == '/' && !Character.isJavaIdentifierStart(cp)) || (cp != '/' && !Character.isJavaIdentifierPart(cp))) {
return "\"" + name + "\"";
}
cc = cp;
}
return name;
}
String tagName(int index) {
try {
int tag = cpool.get(index).getTag();
switch (tag) {
case CONSTANT_Utf8:
return "Utf8";
case CONSTANT_Integer:
return "int";
case CONSTANT_Float:
return "float";
case CONSTANT_Long:
return "long";
case CONSTANT_Double:
return "double";
case CONSTANT_Class:
return "class";
case CONSTANT_String:
return "String";
case CONSTANT_Fieldref:
return "Field";
case CONSTANT_Methodref:
return "Method";
case CONSTANT_InterfaceMethodref:
return "InterfaceMethod";
case CONSTANT_NameAndType:
return "NameAndType";
default:
return "(unknown tag)";
}
} catch (InvalidIndex e) {
throw new RuntimeException(e);
}
}
}