123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875 |
- /*
- * Javassist, a Java-bytecode translator toolkit.
- * Copyright (C) 1999- Shigeru Chiba. All Rights Reserved.
- *
- * The contents of this file are subject to the Mozilla Public License Version
- * 1.1 (the "License"); you may not use this file except in compliance with
- * the License. Alternatively, the contents of this file may be used under
- * the terms of the GNU Lesser General Public License Version 2.1 or later,
- * or the Apache License Version 2.0.
- *
- * Software distributed under the License is distributed on an "AS IS" basis,
- * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
- * for the specific language governing rights and limitations under the
- * License.
- */
-
- package javassist;
-
- import java.io.BufferedInputStream;
- import java.io.ByteArrayInputStream;
- import java.io.ByteArrayOutputStream;
- import java.io.DataInputStream;
- import java.io.DataOutputStream;
- import java.io.IOException;
- import java.io.InputStream;
- import java.lang.ref.Reference;
- import java.lang.ref.WeakReference;
- import java.net.URL;
- import java.util.ArrayList;
- import java.util.HashMap;
- import java.util.Hashtable;
- import java.util.List;
- import java.util.Map;
- import java.util.Set;
-
- import javassist.bytecode.AccessFlag;
- import javassist.bytecode.AnnotationsAttribute;
- import javassist.bytecode.AttributeInfo;
- import javassist.bytecode.BadBytecode;
- import javassist.bytecode.Bytecode;
- import javassist.bytecode.ClassFile;
- import javassist.bytecode.CodeAttribute;
- import javassist.bytecode.CodeIterator;
- import javassist.bytecode.ConstPool;
- import javassist.bytecode.ConstantAttribute;
- import javassist.bytecode.Descriptor;
- import javassist.bytecode.EnclosingMethodAttribute;
- import javassist.bytecode.FieldInfo;
- import javassist.bytecode.InnerClassesAttribute;
- import javassist.bytecode.MethodInfo;
- import javassist.bytecode.ParameterAnnotationsAttribute;
- import javassist.bytecode.SignatureAttribute;
- import javassist.bytecode.annotation.Annotation;
- import javassist.compiler.AccessorMaker;
- import javassist.compiler.CompileError;
- import javassist.compiler.Javac;
- import javassist.expr.ExprEditor;
-
- /**
- * Class<?> types.
- */
- class CtClassType extends CtClass {
- ClassPool classPool;
- boolean wasChanged;
- private boolean wasFrozen;
- boolean wasPruned;
- boolean gcConstPool; // if true, the constant pool entries will be garbage collected.
- ClassFile classfile;
- byte[] rawClassfile; // backup storage
-
- private Reference<CtMember.Cache> memberCache;
- private AccessorMaker accessors;
-
- private FieldInitLink fieldInitializers;
- private Map<CtMethod,String> hiddenMethods; // must be synchronous
- private int uniqueNumberSeed;
-
- private boolean doPruning = ClassPool.doPruning;
- private int getCount;
- private static final int GET_THRESHOLD = 2; // see compress()
-
- CtClassType(String name, ClassPool cp) {
- super(name);
- classPool = cp;
- wasChanged = wasFrozen = wasPruned = gcConstPool = false;
- classfile = null;
- rawClassfile = null;
- memberCache = null;
- accessors = null;
- fieldInitializers = null;
- hiddenMethods = null;
- uniqueNumberSeed = 0;
- getCount = 0;
- }
-
- CtClassType(InputStream ins, ClassPool cp) throws IOException {
- this((String)null, cp);
- classfile = new ClassFile(new DataInputStream(ins));
- qualifiedName = classfile.getName();
- }
-
- CtClassType(ClassFile cf, ClassPool cp) {
- this((String)null, cp);
- classfile = cf;
- qualifiedName = classfile.getName();
- }
-
- @Override
- protected void extendToString(StringBuffer buffer) {
- if (wasChanged)
- buffer.append("changed ");
-
- if (wasFrozen)
- buffer.append("frozen ");
-
- if (wasPruned)
- buffer.append("pruned ");
-
- buffer.append(Modifier.toString(getModifiers()));
- buffer.append(" class ");
- buffer.append(getName());
-
- try {
- CtClass ext = getSuperclass();
- if (ext != null) {
- String name = ext.getName();
- if (!name.equals("java.lang.Object"))
- buffer.append(" extends " + ext.getName());
- }
- }
- catch (NotFoundException e) {
- buffer.append(" extends ??");
- }
-
- try {
- CtClass[] intf = getInterfaces();
- if (intf.length > 0)
- buffer.append(" implements ");
-
- for (int i = 0; i < intf.length; ++i) {
- buffer.append(intf[i].getName());
- buffer.append(", ");
- }
- }
- catch (NotFoundException e) {
- buffer.append(" extends ??");
- }
-
- CtMember.Cache memCache = getMembers();
- exToString(buffer, " fields=",
- memCache.fieldHead(), memCache.lastField());
- exToString(buffer, " constructors=",
- memCache.consHead(), memCache.lastCons());
- exToString(buffer, " methods=",
- memCache.methodHead(), memCache.lastMethod());
- }
-
- private void exToString(StringBuffer buffer, String msg,
- CtMember head, CtMember tail) {
- buffer.append(msg);
- while (head != tail) {
- head = head.next();
- buffer.append(head);
- buffer.append(", ");
- }
- }
-
- @Override
- public AccessorMaker getAccessorMaker() {
- if (accessors == null)
- accessors = new AccessorMaker(this);
-
- return accessors;
- }
-
- @Override
- public ClassFile getClassFile2() {
- return getClassFile3(true);
- }
-
- public ClassFile getClassFile3(boolean doCompress) {
- // quick path - no locking
- ClassFile cfile = classfile;
- if (cfile != null)
- return cfile;
-
- if (doCompress)
- classPool.compress();
-
- byte[] rcfile;
- synchronized (this) {
- // repeat under lock to make sure we get a consistent result (classfile might have been set by another thread)
- cfile = classfile;
- if (cfile != null)
- return cfile;
-
- rcfile = rawClassfile;
- }
-
- if (rcfile != null) {
- final ClassFile cf;
- try {
- cf = new ClassFile(new DataInputStream(new ByteArrayInputStream(rcfile)));
- }
- catch (IOException e) {
- throw new RuntimeException(e.toString(), e);
- }
- getCount = GET_THRESHOLD;
- synchronized (this) {
- rawClassfile = null;
- return setClassFile(cf);
- }
- }
-
- InputStream fin = null;
- try {
- fin = classPool.openClassfile(getName());
- if (fin == null)
- throw new NotFoundException(getName());
-
- fin = new BufferedInputStream(fin);
- ClassFile cf = new ClassFile(new DataInputStream(fin));
- if (!cf.getName().equals(qualifiedName))
- throw new RuntimeException("cannot find " + qualifiedName + ": "
- + cf.getName() + " found in "
- + qualifiedName.replace('.', '/') + ".class");
-
- return setClassFile(cf);
- }
- catch (NotFoundException e) {
- throw new RuntimeException(e.toString(), e);
- }
- catch (IOException e) {
- throw new RuntimeException(e.toString(), e);
- }
- finally {
- if (fin != null)
- try {
- fin.close();
- }
- catch (IOException e) {}
- }
- }
-
- /* Inherited from CtClass. Called by get() in ClassPool.
- *
- * @see javassist.CtClass#incGetCounter()
- * @see #toBytecode(DataOutputStream)
- */
- @Override
- final void incGetCounter() { ++getCount; }
-
- /**
- * Invoked from ClassPool#compress().
- * It releases the class files that have not been recently used
- * if they are unmodified.
- */
- @Override
- void compress() {
- if (getCount < GET_THRESHOLD)
- if (!isModified() && ClassPool.releaseUnmodifiedClassFile)
- removeClassFile();
- else if (isFrozen() && !wasPruned)
- saveClassFile();
-
- getCount = 0;
- }
-
- /**
- * Converts a ClassFile object into a byte array
- * for saving memory space.
- */
- private synchronized void saveClassFile() {
- /* getMembers() and removeClassFile() are also synchronized.
- */
- if (classfile == null || hasMemberCache() != null)
- return;
-
- ByteArrayOutputStream barray = new ByteArrayOutputStream();
- DataOutputStream out = new DataOutputStream(barray);
- try {
- classfile.write(out);
- barray.close();
- rawClassfile = barray.toByteArray();
- classfile = null;
- }
- catch (IOException e) {}
- }
-
- private synchronized void removeClassFile() {
- if (classfile != null && !isModified() && hasMemberCache() == null)
- classfile = null;
- }
-
- /**
- * Updates {@code classfile} if it is null.
- */
- private synchronized ClassFile setClassFile(ClassFile cf) {
- if (classfile == null)
- classfile = cf;
-
- return classfile;
- }
-
- @Override
- public ClassPool getClassPool() { return classPool; }
-
- void setClassPool(ClassPool cp) { classPool = cp; }
-
- @Override
- public URL getURL() throws NotFoundException {
- URL url = classPool.find(getName());
- if (url == null)
- throw new NotFoundException(getName());
- return url;
- }
-
- @Override
- public boolean isModified() { return wasChanged; }
-
- @Override
- public boolean isFrozen() { return wasFrozen; }
-
- @Override
- public void freeze() { wasFrozen = true; }
-
- @Override
- void checkModify() throws RuntimeException {
- if (isFrozen()) {
- String msg = getName() + " class is frozen";
- if (wasPruned)
- msg += " and pruned";
-
- throw new RuntimeException(msg);
- }
-
- wasChanged = true;
- }
-
- @Override
- public void defrost() {
- checkPruned("defrost");
- wasFrozen = false;
- }
-
- @Override
- public boolean subtypeOf(CtClass clazz) throws NotFoundException {
- int i;
- String cname = clazz.getName();
- if (this == clazz || getName().equals(cname))
- return true;
-
- ClassFile file = getClassFile2();
- String supername = file.getSuperclass();
- if (supername != null && supername.equals(cname))
- return true;
-
- String[] ifs = file.getInterfaces();
- int num = ifs.length;
- for (i = 0; i < num; ++i)
- if (ifs[i].equals(cname))
- return true;
-
- if (supername != null && classPool.get(supername).subtypeOf(clazz))
- return true;
-
- for (i = 0; i < num; ++i)
- if (classPool.get(ifs[i]).subtypeOf(clazz))
- return true;
-
- return false;
- }
-
- @Override
- public void setName(String name) throws RuntimeException {
- String oldname = getName();
- if (name.equals(oldname))
- return;
-
- // check this in advance although classNameChanged() below does.
- classPool.checkNotFrozen(name);
- ClassFile cf = getClassFile2();
- super.setName(name);
- cf.setName(name);
- nameReplaced();
- classPool.classNameChanged(oldname, this);
- }
-
- @Override
- public String getGenericSignature() {
- SignatureAttribute sa
- = (SignatureAttribute)getClassFile2().getAttribute(SignatureAttribute.tag);
- return sa == null ? null : sa.getSignature();
- }
-
- @Override
- public void setGenericSignature(String sig) {
- ClassFile cf = getClassFile();
- SignatureAttribute sa = new SignatureAttribute(cf.getConstPool(), sig);
- cf.addAttribute(sa);
- }
-
- @Override
- public void replaceClassName(ClassMap classnames)
- throws RuntimeException
- {
- String oldClassName = getName();
- String newClassName
- = classnames.get(Descriptor.toJvmName(oldClassName));
- if (newClassName != null) {
- newClassName = Descriptor.toJavaName(newClassName);
- // check this in advance although classNameChanged() below does.
- classPool.checkNotFrozen(newClassName);
- }
-
- super.replaceClassName(classnames);
- ClassFile cf = getClassFile2();
- cf.renameClass(classnames);
- nameReplaced();
-
- if (newClassName != null) {
- super.setName(newClassName);
- classPool.classNameChanged(oldClassName, this);
- }
- }
-
- @Override
- public void replaceClassName(String oldname, String newname)
- throws RuntimeException
- {
- String thisname = getName();
- if (thisname.equals(oldname))
- setName(newname);
- else {
- super.replaceClassName(oldname, newname);
- getClassFile2().renameClass(oldname, newname);
- nameReplaced();
- }
- }
-
- @Override
- public boolean isInterface() {
- return Modifier.isInterface(getModifiers());
- }
-
- @Override
- public boolean isAnnotation() {
- return Modifier.isAnnotation(getModifiers());
- }
-
- @Override
- public boolean isEnum() {
- return Modifier.isEnum(getModifiers());
- }
-
- @Override
- public int getModifiers() {
- ClassFile cf = getClassFile2();
- int acc = cf.getAccessFlags();
- acc = AccessFlag.clear(acc, AccessFlag.SUPER);
- int inner = cf.getInnerAccessFlags();
- if (inner != -1) {
- if ((inner & AccessFlag.STATIC) != 0)
- acc |= AccessFlag.STATIC;
- if ((inner & AccessFlag.PUBLIC) != 0)
- acc |= AccessFlag.PUBLIC;
- else {
- acc &= ~AccessFlag.PUBLIC; //clear PUBLIC
- if ((inner & AccessFlag.PROTECTED) != 0)
- acc |= AccessFlag.PROTECTED;
- else if ((inner & AccessFlag.PRIVATE) != 0)
- acc |= AccessFlag.PRIVATE;
- }
- }
- return AccessFlag.toModifier(acc);
- }
-
- @Override
- public CtClass[] getNestedClasses() throws NotFoundException {
- ClassFile cf = getClassFile2();
- InnerClassesAttribute ica
- = (InnerClassesAttribute)cf.getAttribute(InnerClassesAttribute.tag);
- if (ica == null)
- return new CtClass[0];
-
- String thisName = cf.getName() + "$";
- int n = ica.tableLength();
- List<CtClass> list = new ArrayList<CtClass>(n);
- for (int i = 0; i < n; i++) {
- String name = ica.innerClass(i);
- if (name != null)
- if (name.startsWith(thisName)) {
- // if it is an immediate nested class
- if (name.lastIndexOf('$') < thisName.length())
- list.add(classPool.get(name));
- }
- }
-
- return list.toArray(new CtClass[list.size()]);
- }
-
- @Override
- public void setModifiers(int mod) {
- checkModify();
- updateInnerEntry(mod, getName(), this, true);
- ClassFile cf = getClassFile2();
- cf.setAccessFlags(AccessFlag.of(mod & ~Modifier.STATIC));
- }
-
- private static void updateInnerEntry(int newMod, String name, CtClass clazz, boolean outer) {
- ClassFile cf = clazz.getClassFile2();
- InnerClassesAttribute ica
- = (InnerClassesAttribute)cf.getAttribute(InnerClassesAttribute.tag);
- if (ica != null) {
- // If the class is a static inner class, its modifier
- // does not contain the static bit. Its inner class attribute
- // contains the static bit.
- int mod = newMod & ~Modifier.STATIC;
- int i = ica.find(name);
- if (i >= 0) {
- int isStatic = ica.accessFlags(i) & AccessFlag.STATIC;
- if (isStatic != 0 || !Modifier.isStatic(newMod)) {
- clazz.checkModify();
- ica.setAccessFlags(i, AccessFlag.of(mod) | isStatic);
- String outName = ica.outerClass(i);
- if (outName != null && outer)
- try {
- CtClass parent = clazz.getClassPool().get(outName);
- updateInnerEntry(mod, name, parent, false);
- }
- catch (NotFoundException e) {
- throw new RuntimeException("cannot find the declaring class: "
- + outName);
- }
-
- return;
- }
- }
- }
-
- if (Modifier.isStatic(newMod))
- throw new RuntimeException("cannot change " + Descriptor.toJavaName(name)
- + " into a static class");
- }
-
- @Override
- public boolean hasAnnotation(String annotationName) {
- ClassFile cf = getClassFile2();
- AnnotationsAttribute ainfo = (AnnotationsAttribute)
- cf.getAttribute(AnnotationsAttribute.invisibleTag);
- AnnotationsAttribute ainfo2 = (AnnotationsAttribute)
- cf.getAttribute(AnnotationsAttribute.visibleTag);
- return hasAnnotationType(annotationName, getClassPool(), ainfo, ainfo2);
- }
-
- /**
- * @deprecated
- */
- @Deprecated
- static boolean hasAnnotationType(Class<?> clz, ClassPool cp,
- AnnotationsAttribute a1,
- AnnotationsAttribute a2)
- {
- return hasAnnotationType(clz.getName(), cp, a1, a2);
- }
-
- static boolean hasAnnotationType(String annotationTypeName, ClassPool cp,
- AnnotationsAttribute a1,
- AnnotationsAttribute a2)
- {
- Annotation[] anno1, anno2;
-
- if (a1 == null)
- anno1 = null;
- else
- anno1 = a1.getAnnotations();
-
- if (a2 == null)
- anno2 = null;
- else
- anno2 = a2.getAnnotations();
-
- if (anno1 != null)
- for (int i = 0; i < anno1.length; i++)
- if (anno1[i].getTypeName().equals(annotationTypeName))
- return true;
-
- if (anno2 != null)
- for (int i = 0; i < anno2.length; i++)
- if (anno2[i].getTypeName().equals(annotationTypeName))
- return true;
-
- return false;
- }
-
- @Override
- public Object getAnnotation(Class<?> clz) throws ClassNotFoundException {
- ClassFile cf = getClassFile2();
- AnnotationsAttribute ainfo = (AnnotationsAttribute)
- cf.getAttribute(AnnotationsAttribute.invisibleTag);
- AnnotationsAttribute ainfo2 = (AnnotationsAttribute)
- cf.getAttribute(AnnotationsAttribute.visibleTag);
- return getAnnotationType(clz, getClassPool(), ainfo, ainfo2);
- }
-
- static Object getAnnotationType(Class<?> clz, ClassPool cp,
- AnnotationsAttribute a1, AnnotationsAttribute a2)
- throws ClassNotFoundException
- {
- Annotation[] anno1, anno2;
-
- if (a1 == null)
- anno1 = null;
- else
- anno1 = a1.getAnnotations();
-
- if (a2 == null)
- anno2 = null;
- else
- anno2 = a2.getAnnotations();
-
- String typeName = clz.getName();
- if (anno1 != null)
- for (int i = 0; i < anno1.length; i++)
- if (anno1[i].getTypeName().equals(typeName))
- return toAnnoType(anno1[i], cp);
-
- if (anno2 != null)
- for (int i = 0; i < anno2.length; i++)
- if (anno2[i].getTypeName().equals(typeName))
- return toAnnoType(anno2[i], cp);
-
- return null;
- }
-
- @Override
- public Object[] getAnnotations() throws ClassNotFoundException {
- return getAnnotations(false);
- }
-
- @Override
- public Object[] getAvailableAnnotations(){
- try {
- return getAnnotations(true);
- }
- catch (ClassNotFoundException e) {
- throw new RuntimeException("Unexpected exception ", e);
- }
- }
-
- private Object[] getAnnotations(boolean ignoreNotFound)
- throws ClassNotFoundException
- {
- ClassFile cf = getClassFile2();
- AnnotationsAttribute ainfo = (AnnotationsAttribute)
- cf.getAttribute(AnnotationsAttribute.invisibleTag);
- AnnotationsAttribute ainfo2 = (AnnotationsAttribute)
- cf.getAttribute(AnnotationsAttribute.visibleTag);
- return toAnnotationType(ignoreNotFound, getClassPool(), ainfo, ainfo2);
- }
-
- static Object[] toAnnotationType(boolean ignoreNotFound, ClassPool cp,
- AnnotationsAttribute a1, AnnotationsAttribute a2)
- throws ClassNotFoundException
- {
- Annotation[] anno1, anno2;
- int size1, size2;
-
- if (a1 == null) {
- anno1 = null;
- size1 = 0;
- }
- else {
- anno1 = a1.getAnnotations();
- size1 = anno1.length;
- }
-
- if (a2 == null) {
- anno2 = null;
- size2 = 0;
- }
- else {
- anno2 = a2.getAnnotations();
- size2 = anno2.length;
- }
-
- if (!ignoreNotFound){
- Object[] result = new Object[size1 + size2];
- for (int i = 0; i < size1; i++)
- result[i] = toAnnoType(anno1[i], cp);
-
- for (int j = 0; j < size2; j++)
- result[j + size1] = toAnnoType(anno2[j], cp);
-
- return result;
- }
- List<Object> annotations = new ArrayList<Object>();
- for (int i = 0 ; i < size1 ; i++)
- try{
- annotations.add(toAnnoType(anno1[i], cp));
- }
- catch(ClassNotFoundException e){}
- for (int j = 0; j < size2; j++)
- try{
- annotations.add(toAnnoType(anno2[j], cp));
- }
- catch(ClassNotFoundException e){}
-
- return annotations.toArray();
- }
-
- static Object[][] toAnnotationType(boolean ignoreNotFound, ClassPool cp,
- ParameterAnnotationsAttribute a1,
- ParameterAnnotationsAttribute a2,
- MethodInfo minfo)
- throws ClassNotFoundException
- {
- int numParameters = 0;
- if (a1 != null)
- numParameters = a1.numParameters();
- else if (a2 != null)
- numParameters = a2.numParameters();
- else
- numParameters = Descriptor.numOfParameters(minfo.getDescriptor());
-
- Object[][] result = new Object[numParameters][];
- for (int i = 0; i < numParameters; i++) {
- Annotation[] anno1, anno2;
- int size1, size2;
-
- if (a1 == null) {
- anno1 = null;
- size1 = 0;
- }
- else {
- anno1 = a1.getAnnotations()[i];
- size1 = anno1.length;
- }
-
- if (a2 == null) {
- anno2 = null;
- size2 = 0;
- }
- else {
- anno2 = a2.getAnnotations()[i];
- size2 = anno2.length;
- }
-
- if (!ignoreNotFound){
- result[i] = new Object[size1 + size2];
- for (int j = 0; j < size1; ++j)
- result[i][j] = toAnnoType(anno1[j], cp);
-
- for (int j = 0; j < size2; ++j)
- result[i][j + size1] = toAnnoType(anno2[j], cp);
- }
- else{
- List<Object> annotations = new ArrayList<Object>();
- for (int j = 0 ; j < size1 ; j++){
- try{
- annotations.add(toAnnoType(anno1[j], cp));
- }
- catch(ClassNotFoundException e){}
- }
- for (int j = 0; j < size2; j++){
- try{
- annotations.add(toAnnoType(anno2[j], cp));
- }
- catch(ClassNotFoundException e){}
- }
-
- result[i] = annotations.toArray();
- }
- }
-
- return result;
- }
-
- private static Object toAnnoType(Annotation anno, ClassPool cp)
- throws ClassNotFoundException
- {
- try {
- ClassLoader cl = cp.getClassLoader();
- return anno.toAnnotationType(cl, cp);
- }
- catch (ClassNotFoundException e) {
- ClassLoader cl2 = cp.getClass().getClassLoader();
- try {
- return anno.toAnnotationType(cl2, cp);
- }
- catch (ClassNotFoundException e2){
- try {
- Class<?> clazz = cp.get(anno.getTypeName()).toClass();
- return javassist.bytecode.annotation.AnnotationImpl.make(
- clazz.getClassLoader(),
- clazz, cp, anno);
- }
- catch (Throwable e3) {
- throw new ClassNotFoundException(anno.getTypeName());
- }
- }
- }
- }
-
- @Override
- public boolean subclassOf(CtClass superclass) {
- if (superclass == null)
- return false;
-
- String superName = superclass.getName();
- CtClass curr = this;
- try {
- while (curr != null) {
- if (curr.getName().equals(superName))
- return true;
-
- curr = curr.getSuperclass();
- }
- }
- catch (Exception ignored) {}
- return false;
- }
-
- @Override
- public CtClass getSuperclass() throws NotFoundException {
- String supername = getClassFile2().getSuperclass();
- if (supername == null)
- return null;
- return classPool.get(supername);
- }
-
- @Override
- public void setSuperclass(CtClass clazz) throws CannotCompileException {
- checkModify();
- if (isInterface())
- addInterface(clazz);
- else
- getClassFile2().setSuperclass(clazz.getName());
- }
-
- @Override
- public CtClass[] getInterfaces() throws NotFoundException {
- String[] ifs = getClassFile2().getInterfaces();
- int num = ifs.length;
- CtClass[] ifc = new CtClass[num];
- for (int i = 0; i < num; ++i)
- ifc[i] = classPool.get(ifs[i]);
-
- return ifc;
- }
-
- @Override
- public void setInterfaces(CtClass[] list) {
- checkModify();
- String[] ifs;
- if (list == null)
- ifs = new String[0];
- else {
- int num = list.length;
- ifs = new String[num];
- for (int i = 0; i < num; ++i)
- ifs[i] = list[i].getName();
- }
-
- getClassFile2().setInterfaces(ifs);
- }
-
- @Override
- public void addInterface(CtClass anInterface) {
- checkModify();
- if (anInterface != null)
- getClassFile2().addInterface(anInterface.getName());
- }
-
- @Override
- public CtClass getDeclaringClass() throws NotFoundException {
- ClassFile cf = getClassFile2();
- InnerClassesAttribute ica = (InnerClassesAttribute)cf.getAttribute(
- InnerClassesAttribute.tag);
- if (ica == null)
- return null;
-
- String name = getName();
- int n = ica.tableLength();
- for (int i = 0; i < n; ++i)
- if (name.equals(ica.innerClass(i))) {
- String outName = ica.outerClass(i);
- if (outName != null)
- return classPool.get(outName);
-
- // maybe anonymous or local class.
- EnclosingMethodAttribute ema
- = (EnclosingMethodAttribute)cf.getAttribute(
- EnclosingMethodAttribute.tag);
- if (ema != null)
- return classPool.get(ema.className());
-
- }
-
- return null;
- }
-
- @Override
- public CtBehavior getEnclosingBehavior() throws NotFoundException
- {
- ClassFile cf = getClassFile2();
- EnclosingMethodAttribute ema
- = (EnclosingMethodAttribute)cf.getAttribute(
- EnclosingMethodAttribute.tag);
- if (ema == null)
- return null;
- CtClass enc = classPool.get(ema.className());
- String name = ema.methodName();
- if (MethodInfo.nameInit.equals(name))
- return enc.getConstructor(ema.methodDescriptor());
- else if(MethodInfo.nameClinit.equals(name))
- return enc.getClassInitializer();
- else
- return enc.getMethod(name, ema.methodDescriptor());
- }
-
- @Override
- public CtClass makeNestedClass(String name, boolean isStatic)
- {
- if (!isStatic)
- throw new RuntimeException(
- "sorry, only nested static class is supported");
-
- checkModify();
- CtClass c = classPool.makeNestedClass(getName() + "$" + name);
- ClassFile cf = getClassFile2();
- ClassFile cf2 = c.getClassFile2();
- InnerClassesAttribute ica = (InnerClassesAttribute)cf.getAttribute(
- InnerClassesAttribute.tag);
- if (ica == null) {
- ica = new InnerClassesAttribute(cf.getConstPool());
- cf.addAttribute(ica);
- }
-
- ica.append(c.getName(), this.getName(), name,
- (cf2.getAccessFlags() & ~AccessFlag.SUPER) | AccessFlag.STATIC);
- cf2.addAttribute(ica.copy(cf2.getConstPool(), null));
- return c;
- }
-
- /* flush cached names.
- */
- private void nameReplaced() {
- CtMember.Cache cache = hasMemberCache();
- if (cache != null) {
- CtMember mth = cache.methodHead();
- CtMember tail = cache.lastMethod();
- while (mth != tail) {
- mth = mth.next();
- mth.nameReplaced();
- }
- }
- }
-
- /**
- * Returns null if members are not cached.
- */
- protected CtMember.Cache hasMemberCache() {
- if (memberCache != null)
- return memberCache.get();
- return null;
- }
-
- protected synchronized CtMember.Cache getMembers() {
- CtMember.Cache cache = null;
- if (memberCache == null
- || (cache = memberCache.get()) == null) {
- cache = new CtMember.Cache(this);
- makeFieldCache(cache);
- makeBehaviorCache(cache);
- memberCache = new WeakReference<CtMember.Cache>(cache);
- }
-
- return cache;
- }
-
- private void makeFieldCache(CtMember.Cache cache) {
- List<FieldInfo> fields = getClassFile3(false).getFields();
- for (FieldInfo finfo:fields)
- cache.addField(new CtField(finfo, this));
- }
-
- private void makeBehaviorCache(CtMember.Cache cache) {
- List<MethodInfo> methods = getClassFile3(false).getMethods();
- for (MethodInfo minfo:methods)
- if (minfo.isMethod())
- cache.addMethod(new CtMethod(minfo, this));
- else
- cache.addConstructor(new CtConstructor(minfo, this));
- }
-
- @Override
- public CtField[] getFields() {
- List<CtMember> alist = new ArrayList<CtMember>();
- getFields(alist, this);
- return alist.toArray(new CtField[alist.size()]);
- }
-
- private static void getFields(List<CtMember> alist, CtClass cc) {
- if (cc == null)
- return;
-
- try {
- getFields(alist, cc.getSuperclass());
- }
- catch (NotFoundException e) {}
-
- try {
- CtClass[] ifs = cc.getInterfaces();
- for (CtClass ctc : ifs)
- getFields(alist, ctc);
- }
- catch (NotFoundException e) {}
-
- CtMember.Cache memCache = ((CtClassType)cc).getMembers();
- CtMember field = memCache.fieldHead();
- CtMember tail = memCache.lastField();
- while (field != tail) {
- field = field.next();
- if (!Modifier.isPrivate(field.getModifiers()))
- alist.add(field);
- }
- }
-
- @Override
- public CtField getField(String name, String desc) throws NotFoundException {
- CtField f = getField2(name, desc);
- return checkGetField(f, name, desc);
- }
-
- private CtField checkGetField(CtField f, String name, String desc)
- throws NotFoundException
- {
- if (f == null) {
- String msg = "field: " + name;
- if (desc != null)
- msg += " type " + desc;
-
- throw new NotFoundException(msg + " in " + getName());
- }
- return f;
- }
-
- @Override
- CtField getField2(String name, String desc) {
- CtField df = getDeclaredField2(name, desc);
- if (df != null)
- return df;
-
- try {
- CtClass[] ifs = getInterfaces();
- for (CtClass ctc : ifs) {
- CtField f = ctc.getField2(name, desc);
- if (f != null)
- return f;
- }
-
- CtClass s = getSuperclass();
- if (s != null)
- return s.getField2(name, desc);
- }
- catch (NotFoundException e) {}
- return null;
- }
-
- @Override
- public CtField[] getDeclaredFields() {
- CtMember.Cache memCache = getMembers();
- CtMember field = memCache.fieldHead();
- CtMember tail = memCache.lastField();
- int num = CtMember.Cache.count(field, tail);
- CtField[] cfs = new CtField[num];
- int i = 0;
- while (field != tail) {
- field = field.next();
- cfs[i++] = (CtField)field;
- }
-
- return cfs;
- }
-
- @Override
- public CtField getDeclaredField(String name) throws NotFoundException {
- return getDeclaredField(name, null);
- }
-
- @Override
- public CtField getDeclaredField(String name, String desc) throws NotFoundException {
- CtField f = getDeclaredField2(name, desc);
- return checkGetField(f, name, desc);
- }
-
- private CtField getDeclaredField2(String name, String desc) {
- CtMember.Cache memCache = getMembers();
- CtMember field = memCache.fieldHead();
- CtMember tail = memCache.lastField();
- while (field != tail) {
- field = field.next();
- if (field.getName().equals(name)
- && (desc == null || desc.equals(field.getSignature())))
- return (CtField)field;
- }
-
- return null;
- }
-
- @Override
- public CtBehavior[] getDeclaredBehaviors() {
- CtMember.Cache memCache = getMembers();
- CtMember cons = memCache.consHead();
- CtMember consTail = memCache.lastCons();
- int cnum = CtMember.Cache.count(cons, consTail);
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
- int mnum = CtMember.Cache.count(mth, mthTail);
-
- CtBehavior[] cb = new CtBehavior[cnum + mnum];
- int i = 0;
- while (cons != consTail) {
- cons = cons.next();
- cb[i++] = (CtBehavior)cons;
- }
-
- while (mth != mthTail) {
- mth = mth.next();
- cb[i++] = (CtBehavior)mth;
- }
-
- return cb;
- }
-
- @Override
- public CtConstructor[] getConstructors() {
- CtMember.Cache memCache = getMembers();
- CtMember cons = memCache.consHead();
- CtMember consTail = memCache.lastCons();
-
- int n = 0;
- CtMember mem = cons;
- while (mem != consTail) {
- mem = mem.next();
- if (isPubCons((CtConstructor)mem))
- n++;
- }
-
- CtConstructor[] result = new CtConstructor[n];
- int i = 0;
- mem = cons;
- while (mem != consTail) {
- mem = mem.next();
- CtConstructor cc = (CtConstructor)mem;
- if (isPubCons(cc))
- result[i++] = cc;
- }
-
- return result;
- }
-
- private static boolean isPubCons(CtConstructor cons) {
- return !Modifier.isPrivate(cons.getModifiers())
- && cons.isConstructor();
- }
-
- @Override
- public CtConstructor getConstructor(String desc)
- throws NotFoundException
- {
- CtMember.Cache memCache = getMembers();
- CtMember cons = memCache.consHead();
- CtMember consTail = memCache.lastCons();
-
- while (cons != consTail) {
- cons = cons.next();
- CtConstructor cc = (CtConstructor)cons;
- if (cc.getMethodInfo2().getDescriptor().equals(desc)
- && cc.isConstructor())
- return cc;
- }
-
- return super.getConstructor(desc);
- }
-
- @Override
- public CtConstructor[] getDeclaredConstructors() {
- CtMember.Cache memCache = getMembers();
- CtMember cons = memCache.consHead();
- CtMember consTail = memCache.lastCons();
-
- int n = 0;
- CtMember mem = cons;
- while (mem != consTail) {
- mem = mem.next();
- CtConstructor cc = (CtConstructor)mem;
- if (cc.isConstructor())
- n++;
- }
-
- CtConstructor[] result = new CtConstructor[n];
- int i = 0;
- mem = cons;
- while (mem != consTail) {
- mem = mem.next();
- CtConstructor cc = (CtConstructor)mem;
- if (cc.isConstructor())
- result[i++] = cc;
- }
-
- return result;
- }
-
- @Override
- public CtConstructor getClassInitializer() {
- CtMember.Cache memCache = getMembers();
- CtMember cons = memCache.consHead();
- CtMember consTail = memCache.lastCons();
-
- while (cons != consTail) {
- cons = cons.next();
- CtConstructor cc = (CtConstructor)cons;
- if (cc.isClassInitializer())
- return cc;
- }
-
- return null;
- }
-
- @Override
- public CtMethod[] getMethods() {
- Map<String,CtMember> h = new HashMap<String,CtMember>();
- getMethods0(h, this);
- return h.values().toArray(new CtMethod[h.size()]);
- }
-
- private static void getMethods0(Map<String,CtMember> h, CtClass cc) {
- try {
- CtClass[] ifs = cc.getInterfaces();
- for (CtClass ctc : ifs)
- getMethods0(h, ctc);
- }
- catch (NotFoundException e) {}
-
- try {
- CtClass s = cc.getSuperclass();
- if (s != null)
- getMethods0(h, s);
- }
- catch (NotFoundException e) {}
-
- if (cc instanceof CtClassType) {
- CtMember.Cache memCache = ((CtClassType)cc).getMembers();
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
-
- while (mth != mthTail) {
- mth = mth.next();
- if (!Modifier.isPrivate(mth.getModifiers()))
- h.put(((CtMethod)mth).getStringRep(), mth);
- }
- }
- }
-
- @Override
- public CtMethod getMethod(String name, String desc)
- throws NotFoundException
- {
- CtMethod m = getMethod0(this, name, desc);
- if (m != null)
- return m;
- throw new NotFoundException(name + "(..) is not found in "
- + getName());
- }
-
- private static CtMethod getMethod0(CtClass cc,
- String name, String desc) {
- if (cc instanceof CtClassType) {
- CtMember.Cache memCache = ((CtClassType)cc).getMembers();
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
-
- while (mth != mthTail) {
- mth = mth.next();
- if (mth.getName().equals(name)
- && ((CtMethod)mth).getMethodInfo2().getDescriptor().equals(desc))
- return (CtMethod)mth;
- }
- }
-
- try {
- CtClass s = cc.getSuperclass();
- if (s != null) {
- CtMethod m = getMethod0(s, name, desc);
- if (m != null)
- return m;
- }
- }
- catch (NotFoundException e) {}
-
- try {
- CtClass[] ifs = cc.getInterfaces();
- for (CtClass ctc : ifs) {
- CtMethod m = getMethod0(ctc, name, desc);
- if (m != null)
- return m;
- }
- }
- catch (NotFoundException e) {}
- return null;
- }
-
- @Override
- public CtMethod[] getDeclaredMethods() {
- CtMember.Cache memCache = getMembers();
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
- List<CtMember> methods = new ArrayList<CtMember>();
- while (mth != mthTail) {
- mth = mth.next();
- methods.add(mth);
- }
-
- return methods.toArray(new CtMethod[methods.size()]);
- }
-
- @Override
- public CtMethod[] getDeclaredMethods(String name) throws NotFoundException {
- CtMember.Cache memCache = getMembers();
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
- List<CtMember> methods = new ArrayList<CtMember>();
- while (mth != mthTail) {
- mth = mth.next();
- if (mth.getName().equals(name))
- methods.add(mth);
- }
-
- return methods.toArray(new CtMethod[methods.size()]);
- }
-
- @Override
- public CtMethod getDeclaredMethod(String name) throws NotFoundException {
- CtMember.Cache memCache = getMembers();
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
- while (mth != mthTail) {
- mth = mth.next();
- if (mth.getName().equals(name))
- return (CtMethod)mth;
- }
-
- throw new NotFoundException(name + "(..) is not found in "
- + getName());
- }
-
- @Override
- public CtMethod getDeclaredMethod(String name, CtClass[] params)
- throws NotFoundException
- {
- String desc = Descriptor.ofParameters(params);
- CtMember.Cache memCache = getMembers();
- CtMember mth = memCache.methodHead();
- CtMember mthTail = memCache.lastMethod();
-
- while (mth != mthTail) {
- mth = mth.next();
- if (mth.getName().equals(name)
- && ((CtMethod)mth).getMethodInfo2().getDescriptor().startsWith(desc))
- return (CtMethod)mth;
- }
-
- throw new NotFoundException(name + "(..) is not found in "
- + getName());
- }
-
- @Override
- public void addField(CtField f, String init)
- throws CannotCompileException
- {
- addField(f, CtField.Initializer.byExpr(init));
- }
-
- @Override
- public void addField(CtField f, CtField.Initializer init)
- throws CannotCompileException
- {
- checkModify();
- if (f.getDeclaringClass() != this)
- throw new CannotCompileException("cannot add");
-
- if (init == null)
- init = f.getInit();
-
- if (init != null) {
- init.check(f.getSignature());
- int mod = f.getModifiers();
- if (Modifier.isStatic(mod) && Modifier.isFinal(mod))
- try {
- ConstPool cp = getClassFile2().getConstPool();
- int index = init.getConstantValue(cp, f.getType());
- if (index != 0) {
- f.getFieldInfo2().addAttribute(new ConstantAttribute(cp, index));
- init = null;
- }
- }
- catch (NotFoundException e) {}
- }
-
- getMembers().addField(f);
- getClassFile2().addField(f.getFieldInfo2());
-
- if (init != null) {
- FieldInitLink fil = new FieldInitLink(f, init);
- FieldInitLink link = fieldInitializers;
- if (link == null)
- fieldInitializers = fil;
- else {
- while (link.next != null)
- link = link.next;
-
- link.next = fil;
- }
- }
- }
-
- @Override
- public void removeField(CtField f) throws NotFoundException {
- checkModify();
- FieldInfo fi = f.getFieldInfo2();
- ClassFile cf = getClassFile2();
- if (cf.getFields().remove(fi)) {
- getMembers().remove(f);
- gcConstPool = true;
- }
- else
- throw new NotFoundException(f.toString());
- }
-
- @Override
- public CtConstructor makeClassInitializer()
- throws CannotCompileException
- {
- CtConstructor clinit = getClassInitializer();
- if (clinit != null)
- return clinit;
-
- checkModify();
- ClassFile cf = getClassFile2();
- Bytecode code = new Bytecode(cf.getConstPool(), 0, 0);
- modifyClassConstructor(cf, code, 0, 0);
- return getClassInitializer();
- }
-
- @Override
- public void addConstructor(CtConstructor c)
- throws CannotCompileException
- {
- checkModify();
- if (c.getDeclaringClass() != this)
- throw new CannotCompileException("cannot add");
-
- getMembers().addConstructor(c);
- getClassFile2().addMethod(c.getMethodInfo2());
- }
-
- @Override
- public void removeConstructor(CtConstructor m) throws NotFoundException {
- checkModify();
- MethodInfo mi = m.getMethodInfo2();
- ClassFile cf = getClassFile2();
- if (cf.getMethods().remove(mi)) {
- getMembers().remove(m);
- gcConstPool = true;
- }
- else
- throw new NotFoundException(m.toString());
- }
-
- @Override
- public void addMethod(CtMethod m) throws CannotCompileException {
- checkModify();
- if (m.getDeclaringClass() != this)
- throw new CannotCompileException("bad declaring class");
-
- int mod = m.getModifiers();
- if ((getModifiers() & Modifier.INTERFACE) != 0) {
- if (Modifier.isProtected(mod) || Modifier.isPrivate(mod))
- throw new CannotCompileException(
- "an interface method must be public: " + m.toString());
-
- m.setModifiers(mod | Modifier.PUBLIC);
- }
-
- getMembers().addMethod(m);
- getClassFile2().addMethod(m.getMethodInfo2());
- if ((mod & Modifier.ABSTRACT) != 0)
- setModifiers(getModifiers() | Modifier.ABSTRACT);
- }
-
- @Override
- public void removeMethod(CtMethod m) throws NotFoundException
- {
- checkModify();
- MethodInfo mi = m.getMethodInfo2();
- ClassFile cf = getClassFile2();
- if (cf.getMethods().remove(mi)) {
- getMembers().remove(m);
- gcConstPool = true;
- }
- else
- throw new NotFoundException(m.toString());
- }
-
- @Override
- public byte[] getAttribute(String name)
- {
- AttributeInfo ai = getClassFile2().getAttribute(name);
- if (ai == null)
- return null;
- return ai.get();
- }
-
- @Override
- public void setAttribute(String name, byte[] data)
- {
- checkModify();
- ClassFile cf = getClassFile2();
- cf.addAttribute(new AttributeInfo(cf.getConstPool(), name, data));
- }
-
- @Override
- public void instrument(CodeConverter converter)
- throws CannotCompileException
- {
- checkModify();
- ClassFile cf = getClassFile2();
- ConstPool cp = cf.getConstPool();
- List<MethodInfo> methods = cf.getMethods();
- for (MethodInfo minfo: methods.toArray(new MethodInfo[methods.size()]))
- converter.doit(this, minfo, cp);
- }
-
- @Override
- public void instrument(ExprEditor editor)
- throws CannotCompileException
- {
- checkModify();
- ClassFile cf = getClassFile2();
- List<MethodInfo> methods = cf.getMethods();
- for (MethodInfo minfo: methods.toArray(new MethodInfo[methods.size()]))
- editor.doit(this, minfo);
- }
-
- /**
- * @see javassist.CtClass#prune()
- * @see javassist.CtClass#stopPruning(boolean)
- */
- @Override
- public void prune()
- {
- if (wasPruned)
- return;
-
- wasPruned = wasFrozen = true;
- getClassFile2().prune();
- }
-
- @Override
- public void rebuildClassFile() { gcConstPool = true; }
-
- @Override
- public void toBytecode(DataOutputStream out)
- throws CannotCompileException, IOException
- {
- try {
- if (isModified()) {
- checkPruned("toBytecode");
- ClassFile cf = getClassFile2();
- if (gcConstPool) {
- cf.compact();
- gcConstPool = false;
- }
-
- modifyClassConstructor(cf);
- modifyConstructors(cf);
- if (debugDump != null)
- dumpClassFile(cf);
-
- cf.write(out);
- out.flush();
- fieldInitializers = null;
- if (doPruning) {
- // to save memory
- cf.prune();
- wasPruned = true;
- }
- }
- else {
- classPool.writeClassfile(getName(), out);
- // to save memory
- // classfile = null;
- }
-
- getCount = 0;
- wasFrozen = true;
- }
- catch (NotFoundException e) {
- throw new CannotCompileException(e);
- }
- catch (IOException e) {
- throw new CannotCompileException(e);
- }
- }
-
- private void dumpClassFile(ClassFile cf) throws IOException
- {
- DataOutputStream dump = makeFileOutput(debugDump);
- try {
- cf.write(dump);
- }
- finally {
- dump.close();
- }
- }
-
- /* See also checkModified()
- */
- private void checkPruned(String method)
- {
- if (wasPruned)
- throw new RuntimeException(method + "(): " + getName()
- + " was pruned.");
- }
-
- @Override
- public boolean stopPruning(boolean stop)
- {
- boolean prev = !doPruning;
- doPruning = !stop;
- return prev;
- }
-
- private void modifyClassConstructor(ClassFile cf)
- throws CannotCompileException, NotFoundException
- {
- if (fieldInitializers == null)
- return;
-
- Bytecode code = new Bytecode(cf.getConstPool(), 0, 0);
- Javac jv = new Javac(code, this);
- int stacksize = 0;
- boolean doInit = false;
- for (FieldInitLink fi = fieldInitializers; fi != null; fi = fi.next) {
- CtField f = fi.field;
- if (Modifier.isStatic(f.getModifiers())) {
- doInit = true;
- int s = fi.init.compileIfStatic(f.getType(), f.getName(),
- code, jv);
- if (stacksize < s)
- stacksize = s;
- }
- }
-
- if (doInit) // need an initializer for static fileds.
- modifyClassConstructor(cf, code, stacksize, 0);
- }
-
- private void modifyClassConstructor(ClassFile cf, Bytecode code,
- int stacksize, int localsize)
- throws CannotCompileException
- {
- MethodInfo m = cf.getStaticInitializer();
- if (m == null) {
- code.add(Bytecode.RETURN);
- code.setMaxStack(stacksize);
- code.setMaxLocals(localsize);
- m = new MethodInfo(cf.getConstPool(), "<clinit>", "()V");
- m.setAccessFlags(AccessFlag.STATIC);
- m.setCodeAttribute(code.toCodeAttribute());
- cf.addMethod(m);
- CtMember.Cache cache = hasMemberCache();
- if (cache != null)
- cache.addConstructor(new CtConstructor(m, this));
- }
- else {
- CodeAttribute codeAttr = m.getCodeAttribute();
- if (codeAttr == null)
- throw new CannotCompileException("empty <clinit>");
-
- try {
- CodeIterator it = codeAttr.iterator();
- int pos = it.insertEx(code.get());
- it.insert(code.getExceptionTable(), pos);
- int maxstack = codeAttr.getMaxStack();
- if (maxstack < stacksize)
- codeAttr.setMaxStack(stacksize);
-
- int maxlocals = codeAttr.getMaxLocals();
- if (maxlocals < localsize)
- codeAttr.setMaxLocals(localsize);
- }
- catch (BadBytecode e) {
- throw new CannotCompileException(e);
- }
- }
-
- try {
- m.rebuildStackMapIf6(classPool, cf);
- }
- catch (BadBytecode e) {
- throw new CannotCompileException(e);
- }
- }
-
- private void modifyConstructors(ClassFile cf)
- throws CannotCompileException, NotFoundException
- {
- if (fieldInitializers == null)
- return;
-
- ConstPool cp = cf.getConstPool();
- List<MethodInfo> methods = cf.getMethods();
- for (MethodInfo minfo:methods) {
- if (minfo.isConstructor()) {
- CodeAttribute codeAttr = minfo.getCodeAttribute();
- if (codeAttr != null)
- try {
- Bytecode init = new Bytecode(cp, 0,
- codeAttr.getMaxLocals());
- CtClass[] params
- = Descriptor.getParameterTypes(
- minfo.getDescriptor(),
- classPool);
- int stacksize = makeFieldInitializer(init, params);
- insertAuxInitializer(codeAttr, init, stacksize);
- minfo.rebuildStackMapIf6(classPool, cf);
- }
- catch (BadBytecode e) {
- throw new CannotCompileException(e);
- }
- }
- }
- }
-
- private static void insertAuxInitializer(CodeAttribute codeAttr,
- Bytecode initializer,
- int stacksize)
- throws BadBytecode
- {
- CodeIterator it = codeAttr.iterator();
- int index = it.skipSuperConstructor();
- if (index < 0) {
- index = it.skipThisConstructor();
- if (index >= 0)
- return; // this() is called.
-
- // Neither this() or super() is called.
- }
-
- int pos = it.insertEx(initializer.get());
- it.insert(initializer.getExceptionTable(), pos);
- int maxstack = codeAttr.getMaxStack();
- if (maxstack < stacksize)
- codeAttr.setMaxStack(stacksize);
- }
-
- private int makeFieldInitializer(Bytecode code, CtClass[] parameters)
- throws CannotCompileException, NotFoundException
- {
- int stacksize = 0;
- Javac jv = new Javac(code, this);
- try {
- jv.recordParams(parameters, false);
- }
- catch (CompileError e) {
- throw new CannotCompileException(e);
- }
-
- for (FieldInitLink fi = fieldInitializers; fi != null; fi = fi.next) {
- CtField f = fi.field;
- if (!Modifier.isStatic(f.getModifiers())) {
- int s = fi.init.compile(f.getType(), f.getName(), code,
- parameters, jv);
- if (stacksize < s)
- stacksize = s;
- }
- }
-
- return stacksize;
- }
-
- // Methods used by CtNewWrappedMethod
-
- Map<CtMethod,String> getHiddenMethods() {
- if (hiddenMethods == null)
- hiddenMethods = new Hashtable<CtMethod,String>();
-
- return hiddenMethods;
- }
-
- int getUniqueNumber() { return uniqueNumberSeed++; }
-
- @Override
- public String makeUniqueName(String prefix) {
- Map<Object,CtClassType> table = new HashMap<Object,CtClassType>();
- makeMemberList(table);
- Set<Object> keys = table.keySet();
- String[] methods = new String[keys.size()];
- keys.toArray(methods);
-
- if (notFindInArray(prefix, methods))
- return prefix;
-
- int i = 100;
- String name;
- do {
- if (i > 999)
- throw new RuntimeException("too many unique name");
-
- name = prefix + i++;
- } while (!notFindInArray(name, methods));
- return name;
- }
-
- private static boolean notFindInArray(String prefix, String[] values) {
- int len = values.length;
- for (int i = 0; i < len; i++)
- if (values[i].startsWith(prefix))
- return false;
-
- return true;
- }
-
- private void makeMemberList(Map<Object,CtClassType> table) {
- int mod = getModifiers();
- if (Modifier.isAbstract(mod) || Modifier.isInterface(mod))
- try {
- CtClass[] ifs = getInterfaces();
- for (CtClass ic : ifs)
- if (ic != null && ic instanceof CtClassType)
- ((CtClassType)ic).makeMemberList(table);
- }
- catch (NotFoundException e) {}
-
- try {
- CtClass s = getSuperclass();
- if (s != null && s instanceof CtClassType)
- ((CtClassType)s).makeMemberList(table);
- }
- catch (NotFoundException e) {}
-
- List<MethodInfo> methods = getClassFile2().getMethods();
- for (MethodInfo minfo:methods)
- table.put(minfo.getName(), this);
-
- List<FieldInfo> fields = getClassFile2().getFields();
- for (FieldInfo finfo:fields)
- table.put(finfo.getName(), this);
- }
- }
-
- class FieldInitLink {
- FieldInitLink next;
- CtField field;
- CtField.Initializer init;
-
- FieldInitLink(CtField f, CtField.Initializer i) {
- next = null;
- field = f;
- init = i;
- }
- }
|