0N/A/*
3261N/A * Copyright (c) 2008, 2011, Oracle and/or its affiliates. All rights reserved.
0N/A * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
0N/A *
0N/A * This code is free software; you can redistribute it and/or modify it
0N/A * under the terms of the GNU General Public License version 2 only, as
2362N/A * published by the Free Software Foundation. Oracle designates this
0N/A * particular file as subject to the "Classpath" exception as provided
2362N/A * by Oracle in the LICENSE file that accompanied this code.
0N/A *
0N/A * This code is distributed in the hope that it will be useful, but WITHOUT
0N/A * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
0N/A * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
0N/A * version 2 for more details (a copy is included in the LICENSE file that
0N/A * accompanied this code).
0N/A *
0N/A * You should have received a copy of the GNU General Public License version
0N/A * 2 along with this work; if not, write to the Free Software Foundation,
0N/A * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
0N/A *
2362N/A * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
2362N/A * or visit www.oracle.com if you need additional information or have any
2362N/A * questions.
0N/A */
0N/A
0N/Apackage com.sun.tools.classfile;
0N/A
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Class_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Double_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Fieldref_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Float_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Integer_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_InterfaceMethodref_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_InvokeDynamic_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Long_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_MethodHandle_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_MethodType_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Methodref_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_NameAndType_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_String_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CONSTANT_Utf8_info;
0N/Aimport com.sun.tools.classfile.ConstantPool.CPInfo;
0N/Aimport java.util.Map;
0N/A
2507N/A/**
0N/A * Rewrites a class file using a map of translations.
0N/A *
0N/A * <p><b>This is NOT part of any supported API.
0N/A * If you write code that depends on this, you do so at your own risk.
0N/A * This code and its internal interfaces are subject to change or
0N/A * deletion without notice.</b>
0N/A */
0N/Apublic class ClassTranslator
0N/A implements ConstantPool.Visitor<ConstantPool.CPInfo,Map<Object,Object>> {
0N/A /**
0N/A * Create a new ClassFile from {@code cf}, such that for all entries
0N/A * {@code k&nbsp;-\&gt;&nbsp;v} in {@code translations},
0N/A * each occurrence of {@code k} in {@code cf} will be replaced by {@code v}.
0N/A * in
0N/A * @param cf the class file to be processed
0N/A * @param translations the set of translations to be applied
0N/A * @return a copy of {@code} with the values in {@code translations} substituted
0N/A */
0N/A public ClassFile translate(ClassFile cf, Map<Object,Object> translations) {
0N/A ClassFile cf2 = (ClassFile) translations.get(cf);
0N/A if (cf2 == null) {
0N/A ConstantPool constant_pool2 = translate(cf.constant_pool, translations);
0N/A Field[] fields2 = translate(cf.fields, cf.constant_pool, translations);
0N/A Method[] methods2 = translateMethods(cf.methods, cf.constant_pool, translations);
0N/A Attributes attributes2 = translateAttributes(cf.attributes, cf.constant_pool,
0N/A translations);
0N/A
0N/A if (constant_pool2 == cf.constant_pool &&
0N/A fields2 == cf.fields &&
0N/A methods2 == cf.methods &&
0N/A attributes2 == cf.attributes)
0N/A cf2 = cf;
0N/A else
0N/A cf2 = new ClassFile(
0N/A cf.magic,
0N/A cf.minor_version,
0N/A cf.major_version,
0N/A constant_pool2,
0N/A cf.access_flags,
0N/A cf.this_class,
0N/A cf.super_class,
0N/A cf.interfaces,
0N/A fields2,
0N/A methods2,
0N/A attributes2);
0N/A translations.put(cf, cf2);
0N/A }
0N/A return cf2;
0N/A }
0N/A
0N/A ConstantPool translate(ConstantPool cp, Map<Object,Object> translations) {
0N/A ConstantPool cp2 = (ConstantPool) translations.get(cp);
0N/A if (cp2 == null) {
0N/A ConstantPool.CPInfo[] pool2 = new ConstantPool.CPInfo[cp.size()];
0N/A boolean eq = true;
0N/A for (int i = 0; i < cp.size(); ) {
0N/A ConstantPool.CPInfo cpInfo;
0N/A try {
0N/A cpInfo = cp.get(i);
0N/A } catch (ConstantPool.InvalidIndex e) {
0N/A throw new IllegalStateException(e);
0N/A }
0N/A ConstantPool.CPInfo cpInfo2 = translate(cpInfo, translations);
0N/A eq &= (cpInfo == cpInfo2);
0N/A pool2[i] = cpInfo2;
0N/A if (cpInfo.getTag() != cpInfo2.getTag())
0N/A throw new IllegalStateException();
0N/A i += cpInfo.size();
0N/A }
0N/A
0N/A if (eq)
0N/A cp2 = cp;
2507N/A else
0N/A cp2 = new ConstantPool(pool2);
0N/A
0N/A translations.put(cp, cp2);
0N/A }
0N/A return cp2;
0N/A }
0N/A
0N/A ConstantPool.CPInfo translate(ConstantPool.CPInfo cpInfo, Map<Object,Object> translations) {
0N/A ConstantPool.CPInfo cpInfo2 = (ConstantPool.CPInfo) translations.get(cpInfo);
0N/A if (cpInfo2 == null) {
0N/A cpInfo2 = cpInfo.accept(this, translations);
0N/A translations.put(cpInfo, cpInfo2);
0N/A }
0N/A return cpInfo2;
0N/A }
0N/A
0N/A Field[] translate(Field[] fields, ConstantPool constant_pool, Map<Object,Object> translations) {
0N/A Field[] fields2 = (Field[]) translations.get(fields);
0N/A if (fields2 == null) {
0N/A fields2 = new Field[fields.length];
0N/A for (int i = 0; i < fields.length; i++)
0N/A fields2[i] = translate(fields[i], constant_pool, translations);
0N/A if (equal(fields, fields2))
0N/A fields2 = fields;
0N/A translations.put(fields, fields2);
0N/A }
0N/A return fields2;
0N/A }
0N/A
0N/A Field translate(Field field, ConstantPool constant_pool, Map<Object,Object> translations) {
0N/A Field field2 = (Field) translations.get(field);
0N/A if (field2 == null) {
0N/A Attributes attributes2 = translateAttributes(field.attributes, constant_pool,
0N/A translations);
0N/A
0N/A if (attributes2 == field.attributes)
0N/A field2 = field;
0N/A else
0N/A field2 = new Field(
0N/A field.access_flags,
0N/A field.name_index,
0N/A field.descriptor,
0N/A attributes2);
0N/A translations.put(field, field2);
0N/A }
0N/A return field2;
0N/A }
0N/A
0N/A Method[] translateMethods(Method[] methods, ConstantPool constant_pool, Map<Object,Object> translations) {
0N/A Method[] methods2 = (Method[]) translations.get(methods);
0N/A if (methods2 == null) {
0N/A methods2 = new Method[methods.length];
0N/A for (int i = 0; i < methods.length; i++)
0N/A methods2[i] = translate(methods[i], constant_pool, translations);
0N/A if (equal(methods, methods2))
0N/A methods2 = methods;
0N/A translations.put(methods, methods2);
0N/A }
0N/A return methods2;
0N/A }
0N/A
0N/A Method translate(Method method, ConstantPool constant_pool, Map<Object,Object> translations) {
0N/A Method method2 = (Method) translations.get(method);
0N/A if (method2 == null) {
0N/A Attributes attributes2 = translateAttributes(method.attributes, constant_pool,
0N/A translations);
0N/A
0N/A if (attributes2 == method.attributes)
0N/A method2 = method;
0N/A else
0N/A method2 = new Method(
0N/A method.access_flags,
0N/A method.name_index,
0N/A method.descriptor,
0N/A attributes2);
0N/A translations.put(method, method2);
0N/A }
0N/A return method2;
0N/A }
0N/A
0N/A Attributes translateAttributes(Attributes attributes,
0N/A ConstantPool constant_pool, Map<Object,Object> translations) {
0N/A Attributes attributes2 = (Attributes) translations.get(attributes);
0N/A if (attributes2 == null) {
0N/A Attribute[] attrArray2 = new Attribute[attributes.size()];
0N/A ConstantPool constant_pool2 = translate(constant_pool, translations);
0N/A boolean attrsEqual = true;
0N/A for (int i = 0; i < attributes.size(); i++) {
0N/A Attribute attr = attributes.get(i);
0N/A Attribute attr2 = translate(attr, translations);
0N/A if (attr2 != attr)
0N/A attrsEqual = false;
0N/A attrArray2[i] = attr2;
0N/A }
0N/A if ((constant_pool2 == constant_pool) && attrsEqual)
0N/A attributes2 = attributes;
0N/A else
0N/A attributes2 = new Attributes(constant_pool2, attrArray2);
0N/A translations.put(attributes, attributes2);
0N/A }
0N/A return attributes2;
0N/A }
0N/A
0N/A Attribute translate(Attribute attribute, Map<Object,Object> translations) {
0N/A Attribute attribute2 = (Attribute) translations.get(attribute);
0N/A if (attribute2 == null) {
0N/A attribute2 = attribute; // don't support translation within attributes yet
0N/A // (what about Code attribute)
0N/A translations.put(attribute, attribute2);
0N/A }
0N/A return attribute2;
0N/A }
0N/A
0N/A private static <T> boolean equal(T[] a1, T[] a2) {
0N/A if (a1 == null || a2 == null)
0N/A return (a1 == a2);
0N/A if (a1.length != a2.length)
0N/A return false;
0N/A for (int i = 0; i < a1.length; i++) {
0N/A if (a1[i] != a2[i])
0N/A return false;
0N/A }
0N/A return true;
0N/A }
0N/A
0N/A public CPInfo visitClass(CONSTANT_Class_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Class_info info2 = (CONSTANT_Class_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp)
0N/A info2 = info;
0N/A else
0N/A info2 = new CONSTANT_Class_info(cp2, info.name_index);
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitDouble(CONSTANT_Double_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Double_info info2 = (CONSTANT_Double_info) translations.get(info);
0N/A if (info2 == null) {
0N/A info2 = info;
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitFieldref(CONSTANT_Fieldref_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Fieldref_info info2 = (CONSTANT_Fieldref_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp)
0N/A info2 = info;
0N/A else
0N/A info2 = new CONSTANT_Fieldref_info(cp2, info.class_index, info.name_and_type_index);
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitFloat(CONSTANT_Float_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Float_info info2 = (CONSTANT_Float_info) translations.get(info);
0N/A if (info2 == null) {
0N/A info2 = info;
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitInteger(CONSTANT_Integer_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Integer_info info2 = (CONSTANT_Integer_info) translations.get(info);
0N/A if (info2 == null) {
0N/A info2 = info;
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitInterfaceMethodref(CONSTANT_InterfaceMethodref_info info, Map<Object, Object> translations) {
0N/A CONSTANT_InterfaceMethodref_info info2 = (CONSTANT_InterfaceMethodref_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp)
0N/A info2 = info;
0N/A else
0N/A info2 = new CONSTANT_InterfaceMethodref_info(cp2, info.class_index, info.name_and_type_index);
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitInvokeDynamic(CONSTANT_InvokeDynamic_info info, Map<Object, Object> translations) {
0N/A CONSTANT_InvokeDynamic_info info2 = (CONSTANT_InvokeDynamic_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp) {
0N/A info2 = info;
0N/A } else {
0N/A info2 = new CONSTANT_InvokeDynamic_info(cp2, info.bootstrap_method_attr_index, info.name_and_type_index);
0N/A }
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitLong(CONSTANT_Long_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Long_info info2 = (CONSTANT_Long_info) translations.get(info);
0N/A if (info2 == null) {
0N/A info2 = info;
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitNameAndType(CONSTANT_NameAndType_info info, Map<Object, Object> translations) {
0N/A CONSTANT_NameAndType_info info2 = (CONSTANT_NameAndType_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp)
0N/A info2 = info;
0N/A else
0N/A info2 = new CONSTANT_NameAndType_info(cp2, info.name_index, info.type_index);
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitMethodref(CONSTANT_Methodref_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Methodref_info info2 = (CONSTANT_Methodref_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp)
0N/A info2 = info;
0N/A else
0N/A info2 = new CONSTANT_Methodref_info(cp2, info.class_index, info.name_and_type_index);
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitMethodHandle(CONSTANT_MethodHandle_info info, Map<Object, Object> translations) {
0N/A CONSTANT_MethodHandle_info info2 = (CONSTANT_MethodHandle_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp) {
0N/A info2 = info;
0N/A } else {
0N/A info2 = new CONSTANT_MethodHandle_info(cp2, info.reference_kind, info.reference_index);
0N/A }
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitMethodType(CONSTANT_MethodType_info info, Map<Object, Object> translations) {
0N/A CONSTANT_MethodType_info info2 = (CONSTANT_MethodType_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp) {
0N/A info2 = info;
0N/A } else {
0N/A info2 = new CONSTANT_MethodType_info(cp2, info.descriptor_index);
0N/A }
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitString(CONSTANT_String_info info, Map<Object, Object> translations) {
0N/A CONSTANT_String_info info2 = (CONSTANT_String_info) translations.get(info);
0N/A if (info2 == null) {
0N/A ConstantPool cp2 = translate(info.cp, translations);
0N/A if (cp2 == info.cp)
0N/A info2 = info;
0N/A else
0N/A info2 = new CONSTANT_String_info(cp2, info.string_index);
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A public CPInfo visitUtf8(CONSTANT_Utf8_info info, Map<Object, Object> translations) {
0N/A CONSTANT_Utf8_info info2 = (CONSTANT_Utf8_info) translations.get(info);
0N/A if (info2 == null) {
0N/A info2 = info;
0N/A translations.put(info, info2);
0N/A }
0N/A return info;
0N/A }
0N/A
0N/A}
0N/A