1 15 16 package javassist.compiler; 17 18 import javassist.*; 19 import javassist.bytecode.*; 20 import java.util.HashMap ; 21 22 26 public class AccessorMaker { 27 private CtClass clazz; 28 private int uniqueNumber; 29 private HashMap accessors; 30 31 static final String lastParamType = "javassist.runtime.Inner"; 32 33 public AccessorMaker(CtClass c) { 34 clazz = c; 35 uniqueNumber = 1; 36 accessors = new HashMap (); 37 } 38 39 public String getConstructor(CtClass c, String desc, MethodInfo orig) 40 throws CompileError 41 { 42 String key = "<init>:" + desc; 43 String consDesc = (String )accessors.get(key); 44 if (consDesc != null) 45 return consDesc; 47 consDesc = Descriptor.appendParameter(lastParamType, desc); 48 ClassFile cf = clazz.getClassFile(); try { 50 ConstPool cp = cf.getConstPool(); 51 ClassPool pool = clazz.getClassPool(); 52 MethodInfo minfo 53 = new MethodInfo(cp, MethodInfo.nameInit, consDesc); 54 minfo.setAccessFlags(0); 55 minfo.addAttribute(new SyntheticAttribute(cp)); 56 ExceptionsAttribute ea = orig.getExceptionsAttribute(); 57 if (ea != null) 58 minfo.addAttribute(ea.copy(cp, null)); 59 60 CtClass[] params = Descriptor.getParameterTypes(desc, pool); 61 Bytecode code = new Bytecode(cp); 62 code.addAload(0); 63 int regno = 1; 64 for (int i = 0; i < params.length; ++i) 65 regno += code.addLoad(regno, params[i]); 66 code.setMaxLocals(regno + 1); code.addInvokespecial(clazz, MethodInfo.nameInit, desc); 68 69 code.addReturn(null); 70 minfo.setCodeAttribute(code.toCodeAttribute()); 71 cf.addMethod(minfo); 72 } 73 catch (CannotCompileException e) { 74 throw new CompileError(e); 75 } 76 catch (NotFoundException e) { 77 throw new CompileError(e); 78 } 79 80 accessors.put(key, consDesc); 81 return consDesc; 82 } 83 84 97 public String getMethodAccessor(String name, String desc, String accDesc, 98 MethodInfo orig) 99 throws CompileError 100 { 101 String key = name + ":" + desc; 102 String accName = (String )accessors.get(key); 103 if (accName != null) 104 return accName; 106 ClassFile cf = clazz.getClassFile(); accName = findAccessorName(cf); 108 try { 109 ConstPool cp = cf.getConstPool(); 110 ClassPool pool = clazz.getClassPool(); 111 MethodInfo minfo 112 = new MethodInfo(cp, accName, accDesc); 113 minfo.setAccessFlags(AccessFlag.STATIC); 114 minfo.addAttribute(new SyntheticAttribute(cp)); 115 ExceptionsAttribute ea = orig.getExceptionsAttribute(); 116 if (ea != null) 117 minfo.addAttribute(ea.copy(cp, null)); 118 119 CtClass[] params = Descriptor.getParameterTypes(accDesc, pool); 120 int regno = 0; 121 Bytecode code = new Bytecode(cp); 122 for (int i = 0; i < params.length; ++i) 123 regno += code.addLoad(regno, params[i]); 124 125 code.setMaxLocals(regno); 126 if (desc == accDesc) 127 code.addInvokestatic(clazz, name, desc); 128 else 129 code.addInvokevirtual(clazz, name, desc); 130 131 code.addReturn(Descriptor.getReturnType(desc, pool)); 132 minfo.setCodeAttribute(code.toCodeAttribute()); 133 cf.addMethod(minfo); 134 } 135 catch (CannotCompileException e) { 136 throw new CompileError(e); 137 } 138 catch (NotFoundException e) { 139 throw new CompileError(e); 140 } 141 142 accessors.put(key, accName); 143 return accName; 144 } 145 146 149 public MethodInfo getFieldGetter(FieldInfo finfo, boolean is_static) 150 throws CompileError 151 { 152 String fieldName = finfo.getName(); 153 String key = fieldName + ":getter"; 154 Object res = accessors.get(key); 155 if (res != null) 156 return (MethodInfo)res; 158 ClassFile cf = clazz.getClassFile(); String accName = findAccessorName(cf); 160 try { 161 ConstPool cp = cf.getConstPool(); 162 ClassPool pool = clazz.getClassPool(); 163 String fieldType = finfo.getDescriptor(); 164 String accDesc; 165 if (is_static) 166 accDesc = "()" + fieldType; 167 else 168 accDesc = "(" + Descriptor.of(clazz) + ")" + fieldType; 169 170 MethodInfo minfo = new MethodInfo(cp, accName, accDesc); 171 minfo.setAccessFlags(AccessFlag.STATIC); 172 minfo.addAttribute(new SyntheticAttribute(cp)); 173 Bytecode code = new Bytecode(cp); 174 if (is_static) { 175 code.addGetstatic(Bytecode.THIS, fieldName, fieldType); 176 } 177 else { 178 code.addAload(0); 179 code.addGetfield(Bytecode.THIS, fieldName, fieldType); 180 code.setMaxLocals(1); 181 } 182 183 code.addReturn(Descriptor.toCtClass(fieldType, pool)); 184 minfo.setCodeAttribute(code.toCodeAttribute()); 185 cf.addMethod(minfo); 186 accessors.put(key, minfo); 187 return minfo; 188 } 189 catch (CannotCompileException e) { 190 throw new CompileError(e); 191 } 192 catch (NotFoundException e) { 193 throw new CompileError(e); 194 } 195 } 196 197 200 public MethodInfo getFieldSetter(FieldInfo finfo, boolean is_static) 201 throws CompileError 202 { 203 String fieldName = finfo.getName(); 204 String key = fieldName + ":setter"; 205 Object res = accessors.get(key); 206 if (res != null) 207 return (MethodInfo)res; 209 ClassFile cf = clazz.getClassFile(); String accName = findAccessorName(cf); 211 try { 212 ConstPool cp = cf.getConstPool(); 213 ClassPool pool = clazz.getClassPool(); 214 String fieldType = finfo.getDescriptor(); 215 String accDesc; 216 if (is_static) 217 accDesc = "(" + fieldType + ")V"; 218 else 219 accDesc = "(" + Descriptor.of(clazz) + fieldType + ")V"; 220 221 MethodInfo minfo = new MethodInfo(cp, accName, accDesc); 222 minfo.setAccessFlags(AccessFlag.STATIC); 223 minfo.addAttribute(new SyntheticAttribute(cp)); 224 Bytecode code = new Bytecode(cp); 225 int reg; 226 if (is_static) { 227 reg = code.addLoad(0, Descriptor.toCtClass(fieldType, pool)); 228 code.addPutstatic(Bytecode.THIS, fieldName, fieldType); 229 } 230 else { 231 code.addAload(0); 232 reg = code.addLoad(1, Descriptor.toCtClass(fieldType, pool)) 233 + 1; 234 code.addPutfield(Bytecode.THIS, fieldName, fieldType); 235 } 236 237 code.addReturn(null); 238 code.setMaxLocals(reg); 239 minfo.setCodeAttribute(code.toCodeAttribute()); 240 cf.addMethod(minfo); 241 accessors.put(key, minfo); 242 return minfo; 243 } 244 catch (CannotCompileException e) { 245 throw new CompileError(e); 246 } 247 catch (NotFoundException e) { 248 throw new CompileError(e); 249 } 250 } 251 252 private String findAccessorName(ClassFile cf) { 253 String accName; 254 do { 255 accName = "access$" + uniqueNumber++; 256 } while (cf.getMethod(accName) != null); 257 return accName; 258 } 259 } 260 | Popular Tags |