You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

WeavingAdaptor.java 30KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947
  1. /* *******************************************************************
  2. * Copyright (c) 2004 IBM Corporation
  3. * All rights reserved.
  4. * This program and the accompanying materials are made available
  5. * under the terms of the Eclipse Public License v 2.0
  6. * which accompanies this distribution and is available at
  7. * https://www.eclipse.org/org/documents/epl-2.0/EPL-2.0.txt
  8. *
  9. * Contributors:
  10. * Matthew Webster, Adrian Colyer, John Kew + Lyor Goldstein (caching)
  11. * Martin Lippert initial implementation
  12. * ******************************************************************/
  13. package org.aspectj.weaver.tools;
  14. import java.io.File;
  15. import java.io.FileOutputStream;
  16. import java.io.IOException;
  17. import java.io.PrintWriter;
  18. import java.net.URL;
  19. import java.net.URLClassLoader;
  20. import java.security.ProtectionDomain;
  21. import java.util.ArrayList;
  22. import java.util.Arrays;
  23. import java.util.HashMap;
  24. import java.util.HashSet;
  25. import java.util.Iterator;
  26. import java.util.LinkedList;
  27. import java.util.List;
  28. import java.util.Map;
  29. import java.util.Properties;
  30. import java.util.Set;
  31. import java.util.StringTokenizer;
  32. import org.aspectj.bridge.AbortException;
  33. import org.aspectj.bridge.IMessage;
  34. import org.aspectj.bridge.IMessage.Kind;
  35. import org.aspectj.bridge.IMessageContext;
  36. import org.aspectj.bridge.IMessageHandler;
  37. import org.aspectj.bridge.IMessageHolder;
  38. import org.aspectj.bridge.Message;
  39. import org.aspectj.bridge.MessageHandler;
  40. import org.aspectj.bridge.MessageUtil;
  41. import org.aspectj.bridge.MessageWriter;
  42. import org.aspectj.bridge.Version;
  43. import org.aspectj.bridge.WeaveMessage;
  44. import org.aspectj.util.FileUtil;
  45. import org.aspectj.util.LangUtil;
  46. import org.aspectj.weaver.IClassFileProvider;
  47. import org.aspectj.weaver.IUnwovenClassFile;
  48. import org.aspectj.weaver.IWeaveRequestor;
  49. import org.aspectj.weaver.World;
  50. import org.aspectj.weaver.bcel.BcelObjectType;
  51. import org.aspectj.weaver.bcel.BcelWeaver;
  52. import org.aspectj.weaver.bcel.BcelWorld;
  53. import org.aspectj.weaver.bcel.UnwovenClassFile;
  54. import org.aspectj.weaver.tools.cache.CachedClassEntry;
  55. import org.aspectj.weaver.tools.cache.CachedClassReference;
  56. import org.aspectj.weaver.tools.cache.SimpleCache;
  57. import org.aspectj.weaver.tools.cache.SimpleCacheFactory;
  58. import org.aspectj.weaver.tools.cache.WeavedClassCache;
  59. // OPTIMIZE add guards for all the debug/info/etc
  60. /**
  61. * This adaptor allows the AspectJ compiler to be embedded in an existing system to facilitate load-time weaving. It provides an
  62. * interface for a weaving class loader to provide a classpath to be woven by a set of aspects. A callback is supplied to allow a
  63. * class loader to define classes generated by the compiler during the weaving process.
  64. * <p>
  65. * A weaving class loader should create a <code>WeavingAdaptor</code> before any classes are defined, typically during construction.
  66. * The set of aspects passed to the adaptor is fixed for the lifetime of the adaptor although the classpath can be augmented. A
  67. * system property can be set to allow verbose weaving messages to be written to the console.
  68. *
  69. */
  70. public class WeavingAdaptor implements IMessageContext {
  71. /**
  72. * System property used to turn on verbose weaving messages
  73. */
  74. public static final String WEAVING_ADAPTOR_VERBOSE = "aj.weaving.verbose";
  75. public static final String SHOW_WEAVE_INFO_PROPERTY = "org.aspectj.weaver.showWeaveInfo";
  76. public static final String TRACE_MESSAGES_PROPERTY = "org.aspectj.tracing.messages";
  77. private final static String ASPECTJ_BASE_PACKAGE = "org.aspectj.";
  78. private final static String PACKAGE_INITIAL_CHARS = ASPECTJ_BASE_PACKAGE.charAt(0) + "sj";
  79. private boolean enabled = false;
  80. protected boolean verbose = getVerbose();
  81. protected BcelWorld bcelWorld;
  82. protected BcelWeaver weaver;
  83. private IMessageHandler messageHandler;
  84. private WeavingAdaptorMessageHolder messageHolder;
  85. private boolean abortOnError = false;
  86. protected GeneratedClassHandler generatedClassHandler;
  87. protected Map<String, IUnwovenClassFile> generatedClasses = new HashMap<>();
  88. public BcelObjectType delegateForCurrentClass; // lazily initialized, should be used to prevent parsing bytecode multiple
  89. // times
  90. protected ProtectionDomain activeProtectionDomain;
  91. private boolean haveWarnedOnJavax = false;
  92. protected WeavedClassCache cache;
  93. private int weavingSpecialTypes = 0;
  94. private static final int INITIALIZED = 0x1;
  95. private static final int WEAVE_JAVA_PACKAGE = 0x2;
  96. private static final int WEAVE_JAVAX_PACKAGE = 0x4;
  97. private static Trace trace = TraceFactory.getTraceFactory().getTrace(WeavingAdaptor.class);
  98. protected WeavingAdaptor() {
  99. }
  100. /**
  101. * Construct a WeavingAdaptor with a reference to a weaving class loader. The adaptor will automatically search the class loader
  102. * hierarchy to resolve classes. The adaptor will also search the hierarchy for WeavingClassLoader instances to determine the
  103. * set of aspects to be used for weaving.
  104. *
  105. * @param loader instance of <code>ClassLoader</code>
  106. */
  107. public WeavingAdaptor(WeavingClassLoader loader) {
  108. // System.err.println("? WeavingAdaptor.<init>(" + loader +"," + aspectURLs.length + ")");
  109. generatedClassHandler = loader;
  110. init((ClassLoader)loader, getFullClassPath((ClassLoader) loader), getFullAspectPath((ClassLoader) loader/* ,aspectURLs */));
  111. }
  112. /**
  113. * Construct a WeavingAdaptor with a reference to a <code>GeneratedClassHandler</code>, a full search path for resolving classes
  114. * and a complete set of aspects. The search path must include classes loaded by the class loader constructing the
  115. * WeavingAdaptor and all its parents in the hierarchy.
  116. *
  117. * @param handler <code>GeneratedClassHandler</code>
  118. * @param classURLs the URLs from which to resolve classes
  119. * @param aspectURLs the aspects used to weave classes defined by this class loader
  120. */
  121. public WeavingAdaptor(GeneratedClassHandler handler, URL[] classURLs, URL[] aspectURLs) {
  122. // System.err.println("? WeavingAdaptor.<init>()");
  123. generatedClassHandler = handler;
  124. init(null, FileUtil.makeClasspath(classURLs), FileUtil.makeClasspath(aspectURLs));
  125. }
  126. protected List<String> getFullClassPath(ClassLoader loader) {
  127. List<String> list = new LinkedList<>();
  128. for (; loader != null; loader = loader.getParent()) {
  129. if (loader instanceof URLClassLoader) {
  130. URL[] urls = ((URLClassLoader) loader).getURLs();
  131. list.addAll(0, FileUtil.makeClasspath(urls));
  132. } else {
  133. warn("cannot determine classpath");
  134. }
  135. }
  136. // On Java9 it is possible to fail to find a URLClassLoader from which to derive a suitable classpath
  137. // For now we can determine it from the java.class.path:
  138. if (LangUtil.is9VMOrGreater()) {
  139. list.add(0, LangUtil.getJrtFsFilePath());
  140. List<String> javaClassPathEntries = makeClasspath(System.getProperty("java.class.path"));
  141. for (int i=javaClassPathEntries.size()-1;i>=0;i--) {
  142. String javaClassPathEntry = javaClassPathEntries.get(i);
  143. if (!list.contains(javaClassPathEntry)) {
  144. list.add(0,javaClassPathEntry);
  145. }
  146. }
  147. }
  148. // On Java9 the sun.boot.class.path won't be set. System classes accessible through JRT filesystem
  149. list.addAll(0, makeClasspath(System.getProperty("sun.boot.class.path")));
  150. return list;
  151. }
  152. private List<String> getFullAspectPath(ClassLoader loader) {
  153. List<String> list = new LinkedList<>();
  154. for (; loader != null; loader = loader.getParent()) {
  155. if (loader instanceof WeavingClassLoader) {
  156. URL[] urls = ((WeavingClassLoader) loader).getAspectURLs();
  157. list.addAll(0, FileUtil.makeClasspath(urls));
  158. }
  159. }
  160. return list;
  161. }
  162. private static boolean getVerbose() {
  163. try {
  164. return Boolean.getBoolean(WEAVING_ADAPTOR_VERBOSE);
  165. } catch (Throwable t) {
  166. // security exception
  167. return false;
  168. }
  169. }
  170. /**
  171. * Initialize the WeavingAdapter
  172. * @param loader ClassLoader used by this adapter; which can be null
  173. * @param classPath classpath of this adapter
  174. * @param aspectPath list of aspect paths
  175. */
  176. private void init(ClassLoader loader, List<String> classPath, List<String> aspectPath) {
  177. abortOnError = true;
  178. createMessageHandler();
  179. info("using classpath: " + classPath);
  180. info("using aspectpath: " + aspectPath);
  181. bcelWorld = new BcelWorld(classPath, messageHandler, null);
  182. bcelWorld.setXnoInline(false);
  183. bcelWorld.getLint().loadDefaultProperties();
  184. bcelWorld.setBehaveInJava5Way(true);
  185. weaver = new BcelWeaver(bcelWorld);
  186. registerAspectLibraries(aspectPath);
  187. initializeCache(loader, aspectPath, null, getMessageHandler());
  188. enabled = true;
  189. }
  190. /**
  191. * If the cache is enabled, initialize it and swap out the existing classhandler
  192. * for the caching one -
  193. *
  194. * @param loader classloader for this adapter, may be null
  195. * @param aspects List of strings representing aspects managed by the adapter; these could be urls or classnames
  196. * @param existingClassHandler current class handler
  197. * @param myMessageHandler current message handler
  198. */
  199. protected void initializeCache(ClassLoader loader, List<String> aspects, GeneratedClassHandler existingClassHandler, IMessageHandler myMessageHandler) {
  200. if (WeavedClassCache.isEnabled()) {
  201. cache = WeavedClassCache.createCache(loader, aspects, existingClassHandler, myMessageHandler);
  202. // Wrap the existing class handler so that any generated classes are also cached
  203. if (cache != null) {
  204. this.generatedClassHandler = cache.getCachingClassHandler();
  205. }
  206. }
  207. }
  208. protected void createMessageHandler() {
  209. messageHolder = new WeavingAdaptorMessageHolder(new PrintWriter(System.err));
  210. messageHandler = messageHolder;
  211. if (verbose) {
  212. messageHandler.dontIgnore(IMessage.INFO);
  213. }
  214. if (Boolean.getBoolean(SHOW_WEAVE_INFO_PROPERTY)) {
  215. messageHandler.dontIgnore(IMessage.WEAVEINFO);
  216. }
  217. info("AspectJ Weaver Version " + Version.getText() + " built on " + Version.getTimeText()); //$NON-NLS-1$
  218. }
  219. protected IMessageHandler getMessageHandler() {
  220. return messageHandler;
  221. }
  222. public IMessageHolder getMessageHolder() {
  223. return messageHolder;
  224. }
  225. protected void setMessageHandler(IMessageHandler mh) {
  226. if (mh instanceof ISupportsMessageContext) {
  227. ISupportsMessageContext smc = (ISupportsMessageContext) mh;
  228. smc.setMessageContext(this);
  229. }
  230. if (mh != messageHolder) {
  231. messageHolder.setDelegate(mh);
  232. }
  233. messageHolder.flushMessages();
  234. }
  235. protected void disable() {
  236. if (trace.isTraceEnabled()) {
  237. trace.enter("disable", this);
  238. }
  239. enabled = false;
  240. messageHolder.flushMessages();
  241. if (trace.isTraceEnabled()) {
  242. trace.exit("disable");
  243. }
  244. }
  245. protected void enable() {
  246. enabled = true;
  247. messageHolder.flushMessages();
  248. }
  249. protected boolean isEnabled() {
  250. return enabled;
  251. }
  252. /**
  253. * Appends URL to path used by the WeavingAdptor to resolve classes
  254. *
  255. * @param url to be appended to search path
  256. */
  257. public void addURL(URL url) {
  258. File libFile = new File(url.getPath());
  259. try {
  260. weaver.addLibraryJarFile(libFile);
  261. } catch (IOException ex) {
  262. warn("bad library: '" + libFile + "'");
  263. }
  264. }
  265. /**
  266. * Weave a class using aspects previously supplied to the adaptor.
  267. *
  268. * @param name the name of the class
  269. * @param bytes the class bytes
  270. * @return the woven bytes
  271. * @exception IOException weave failed
  272. */
  273. public byte[] weaveClass(String name, byte[] bytes) throws IOException {
  274. return weaveClass(name, bytes, false);
  275. }
  276. // Track if the weaver is already running on this thread - don't allow re-entrant calls
  277. private ThreadLocal<Boolean> weaverRunning = new ThreadLocal<Boolean>() {
  278. @Override
  279. protected Boolean initialValue() {
  280. return Boolean.FALSE;
  281. }
  282. };
  283. /**
  284. * Weave a class using aspects previously supplied to the adaptor.
  285. *
  286. * @param name the name of the class
  287. * @param bytes the class bytes
  288. * @param mustWeave if true then this class *must* get woven (used for concrete aspects generated from XML)
  289. * @return the woven bytes
  290. * @exception IOException weave failed
  291. */
  292. public byte[] weaveClass(String name, byte[] bytes, boolean mustWeave) throws IOException {
  293. if (trace == null) {
  294. // Pr231945: we are likely to be under tomcat and ENABLE_CLEAR_REFERENCES hasn't been set
  295. System.err
  296. .println("AspectJ Weaver cannot continue to weave, static state has been cleared. Are you under Tomcat? In order to weave '"
  297. + name
  298. + "' during shutdown, 'org.apache.catalina.loader.WebappClassLoader.ENABLE_CLEAR_REFERENCES=false' must be set (see https://bugs.eclipse.org/bugs/show_bug.cgi?id=231945).");
  299. return bytes;
  300. }
  301. if (weaverRunning.get()) {
  302. // System.out.println("AJC: avoiding re-entrant call to transform " + name);
  303. return bytes;
  304. }
  305. try {
  306. weaverRunning.set(true);
  307. if (trace.isTraceEnabled()) {
  308. trace.enter("weaveClass", this, new Object[] { name, bytes });
  309. }
  310. if (!enabled) {
  311. if (trace.isTraceEnabled()) {
  312. trace.exit("weaveClass", false);
  313. }
  314. return bytes;
  315. }
  316. boolean debugOn = !messageHandler.isIgnoring(Message.DEBUG);
  317. try {
  318. delegateForCurrentClass = null;
  319. name = name.replace('/', '.');
  320. if (couldWeave(name, bytes)) {
  321. if (accept(name, bytes)) {
  322. // Determine if we have the weaved class cached
  323. CachedClassReference cacheKey = null;
  324. final byte[] original_bytes = bytes;
  325. if (cache != null && !mustWeave) {
  326. cacheKey = cache.createCacheKey(name, original_bytes);
  327. CachedClassEntry entry = cache.get(cacheKey, original_bytes);
  328. if (entry != null) {
  329. // If the entry has been explicitly ignored
  330. // return the original bytes
  331. if (entry.isIgnored()) {
  332. return bytes;
  333. }
  334. return entry.getBytes();
  335. }
  336. }
  337. // TODO @AspectJ problem
  338. // Annotation style aspects need to be included regardless in order to get
  339. // a valid aspectOf()/hasAspect() generated in them. However - if they are excluded
  340. // (via include/exclude in aop.xml) they really should only get aspectOf()/hasAspect()
  341. // and not be included in the full set of aspects being applied by 'this' weaver
  342. if (debugOn) {
  343. debug("weaving '" + name + "'");
  344. }
  345. bytes = getWovenBytes(name, bytes);
  346. // temporarily out - searching for @Aspect annotated types is a slow thing to do - we should
  347. // expect the user to name them if they want them woven - just like code style
  348. // } else if (shouldWeaveAnnotationStyleAspect(name, bytes)) {
  349. // if (mustWeave) {
  350. // if (bcelWorld.getLint().mustWeaveXmlDefinedAspects.isEnabled()) {
  351. // bcelWorld.getLint().mustWeaveXmlDefinedAspects.signal(name, null);
  352. // }
  353. // }
  354. // // an @AspectJ aspect needs to be at least munged by the aspectOf munger
  355. // if (debugOn) {
  356. // debug("weaving '" + name + "'");
  357. // }
  358. // bytes = getAtAspectJAspectBytes(name, bytes);
  359. // Add the weaved class to the cache only if there
  360. // has been an actual change
  361. // JVK: Is there a better way to check if the class has
  362. // been transformed without carrying up some value
  363. // from the depths?
  364. if (cacheKey != null) {
  365. // If no transform has been applied, mark the class
  366. // as ignored.
  367. if (Arrays.equals(original_bytes, bytes)) {
  368. cache.ignore(cacheKey, original_bytes);
  369. } else {
  370. cache.put(cacheKey, original_bytes, bytes);
  371. }
  372. }
  373. } else if (debugOn) {
  374. debug("not weaving '" + name + "'");
  375. }
  376. } else if (debugOn) {
  377. debug("cannot weave '" + name + "'");
  378. }
  379. } finally {
  380. delegateForCurrentClass = null;
  381. }
  382. if (trace.isTraceEnabled()) {
  383. trace.exit("weaveClass", bytes);
  384. }
  385. return bytes;
  386. } finally {
  387. weaverRunning.remove();
  388. }
  389. }
  390. /**
  391. * @param name
  392. * @return true if even valid to weave: either with an accept check or to munge it for @AspectJ aspectof support
  393. */
  394. private boolean couldWeave(String name, byte[] bytes) {
  395. return !generatedClasses.containsKey(name) && shouldWeaveName(name);
  396. }
  397. // ATAJ
  398. protected boolean accept(String name, byte[] bytes) {
  399. return true;
  400. }
  401. protected boolean shouldDump(String name, boolean before) {
  402. return false;
  403. }
  404. private boolean shouldWeaveName(String name) {
  405. if (PACKAGE_INITIAL_CHARS.indexOf(name.charAt(0)) != -1) {
  406. if ((weavingSpecialTypes & INITIALIZED) == 0) {
  407. weavingSpecialTypes |= INITIALIZED;
  408. // initialize it
  409. Properties p = weaver.getWorld().getExtraConfiguration();
  410. if (p != null) {
  411. boolean b = p.getProperty(World.xsetWEAVE_JAVA_PACKAGES, "false").equalsIgnoreCase("true");
  412. if (b) {
  413. weavingSpecialTypes |= WEAVE_JAVA_PACKAGE;
  414. }
  415. b = p.getProperty(World.xsetWEAVE_JAVAX_PACKAGES, "false").equalsIgnoreCase("true");
  416. if (b) {
  417. weavingSpecialTypes |= WEAVE_JAVAX_PACKAGE;
  418. }
  419. }
  420. }
  421. if (name.startsWith(ASPECTJ_BASE_PACKAGE)) {
  422. return false;
  423. }
  424. if (name.startsWith("sun.reflect.")) {// JDK reflect
  425. return false;
  426. }
  427. if (name.startsWith("javax.")) {
  428. if ((weavingSpecialTypes & WEAVE_JAVAX_PACKAGE) != 0) {
  429. return true;
  430. } else {
  431. if (!haveWarnedOnJavax) {
  432. haveWarnedOnJavax = true;
  433. warn("javax.* types are not being woven because the weaver option '-Xset:weaveJavaxPackages=true' has not been specified");
  434. }
  435. return false;
  436. }
  437. }
  438. if (name.startsWith("java.")) {
  439. if ((weavingSpecialTypes & WEAVE_JAVA_PACKAGE) != 0) {
  440. return true;
  441. } else {
  442. return false;
  443. }
  444. }
  445. }
  446. // boolean should = !(name.startsWith("org.aspectj.")
  447. // || (name.startsWith("java.") && (weavingSpecialTypes & WEAVE_JAVA_PACKAGE) == 0)
  448. // || (name.startsWith("javax.") && (weavingSpecialTypes & WEAVE_JAVAX_PACKAGE) == 0)
  449. // // || name.startsWith("$Proxy")//JDK proxies//FIXME AV is that 1.3 proxy ? fe. ataspect.$Proxy0 is a java5 proxy...
  450. // || name.startsWith("sun.reflect."));
  451. return true;
  452. }
  453. /**
  454. * We allow @AJ aspect weaving so that we can add aspectOf() as part of the weaving (and not part of the source compilation)
  455. *
  456. * @param name
  457. * @param bytes bytecode (from classloader), allow to NOT lookup stuff on disk again during resolve
  458. * @return true if @Aspect
  459. */
  460. private boolean shouldWeaveAnnotationStyleAspect(String name, byte[] bytes) {
  461. if (delegateForCurrentClass == null) {
  462. // if (weaver.getWorld().isASMAround()) return asmCheckAnnotationStyleAspect(bytes);
  463. // else
  464. ensureDelegateInitialized(name, bytes);
  465. }
  466. return (delegateForCurrentClass.isAnnotationStyleAspect());
  467. }
  468. // private boolean asmCheckAnnotationStyleAspect(byte[] bytes) {
  469. // IsAtAspectAnnotationVisitor detector = new IsAtAspectAnnotationVisitor();
  470. //
  471. // ClassReader cr = new ClassReader(bytes);
  472. // try {
  473. // cr.accept(detector, true);//, ClassReader.SKIP_DEBUG | ClassReader.SKIP_CODE | ClassReader.SKIP_FRAMES);
  474. // } catch (Exception spe) {
  475. // // if anything goes wrong, e.g., an NPE, then assume it's NOT an @AspectJ aspect...
  476. // System.err.println("Unexpected problem parsing bytes to discover @Aspect annotation");
  477. // spe.printStackTrace();
  478. // return false;
  479. // }
  480. //
  481. // return detector.isAspect();
  482. // }
  483. protected void ensureDelegateInitialized(String name, byte[] bytes) {
  484. if (delegateForCurrentClass == null) {
  485. BcelWorld world = (BcelWorld) weaver.getWorld();
  486. delegateForCurrentClass = world.addSourceObjectType(name, bytes, false);
  487. }
  488. }
  489. /**
  490. * Weave a set of bytes defining a class.
  491. *
  492. * @param name the name of the class being woven
  493. * @param bytes the bytes that define the class
  494. * @return byte[] the woven bytes for the class
  495. * @throws IOException
  496. */
  497. private byte[] getWovenBytes(String name, byte[] bytes) throws IOException {
  498. WeavingClassFileProvider wcp = new WeavingClassFileProvider(name, bytes);
  499. weaver.weave(wcp);
  500. return wcp.getBytes();
  501. }
  502. /**
  503. * Weave a set of bytes defining a class for only what is needed to turn @AspectJ aspect in a usefull form ie with aspectOf
  504. * method - see #113587
  505. *
  506. * @param name the name of the class being woven
  507. * @param bytes the bytes that define the class
  508. * @return byte[] the woven bytes for the class
  509. * @throws IOException
  510. */
  511. private byte[] getAtAspectJAspectBytes(String name, byte[] bytes) throws IOException {
  512. WeavingClassFileProvider wcp = new WeavingClassFileProvider(name, bytes);
  513. wcp.setApplyAtAspectJMungersOnly();
  514. weaver.weave(wcp);
  515. return wcp.getBytes();
  516. }
  517. private void registerAspectLibraries(List<String> aspectPath) {
  518. // System.err.println("? WeavingAdaptor.registerAspectLibraries(" + aspectPath + ")");
  519. for (String libName : aspectPath) {
  520. addAspectLibrary(libName);
  521. }
  522. weaver.prepareForWeave();
  523. }
  524. /*
  525. * Register an aspect library with this classloader for use during weaving. This class loader will also return (unmodified) any
  526. * of the classes in the library in response to a <code>findClass()</code> request. The library is not required to be on the
  527. * weavingClasspath given when this classloader was constructed.
  528. *
  529. * @param aspectLibraryJarFile a jar file representing an aspect library
  530. *
  531. * @throws IOException
  532. */
  533. private void addAspectLibrary(String aspectLibraryName) {
  534. File aspectLibrary = new File(aspectLibraryName);
  535. if (aspectLibrary.isDirectory() || (FileUtil.isZipFile(aspectLibrary))) {
  536. try {
  537. info("adding aspect library: '" + aspectLibrary + "'");
  538. weaver.addLibraryJarFile(aspectLibrary);
  539. } catch (IOException ex) {
  540. error("exception adding aspect library: '" + ex + "'");
  541. }
  542. } else {
  543. error("bad aspect library: '" + aspectLibrary + "'");
  544. }
  545. }
  546. private static List<String> makeClasspath(String cp) {
  547. List<String> ret = new ArrayList<>();
  548. if (cp != null) {
  549. StringTokenizer tok = new StringTokenizer(cp, File.pathSeparator);
  550. while (tok.hasMoreTokens()) {
  551. ret.add(tok.nextToken());
  552. }
  553. }
  554. return ret;
  555. }
  556. protected boolean debug(String message) {
  557. return MessageUtil.debug(messageHandler, message);
  558. }
  559. protected boolean info(String message) {
  560. return MessageUtil.info(messageHandler, message);
  561. }
  562. protected boolean warn(String message) {
  563. return MessageUtil.warn(messageHandler, message);
  564. }
  565. protected boolean warn(String message, Throwable th) {
  566. return messageHandler.handleMessage(new Message(message, IMessage.WARNING, th, null));
  567. }
  568. protected boolean error(String message) {
  569. return MessageUtil.error(messageHandler, message);
  570. }
  571. protected boolean error(String message, Throwable th) {
  572. return messageHandler.handleMessage(new Message(message, IMessage.ERROR, th, null));
  573. }
  574. public String getContextId() {
  575. return "WeavingAdaptor";
  576. }
  577. /**
  578. * Dump the given bytcode in _dump/... (dev mode)
  579. *
  580. * @param name
  581. * @param b
  582. * @param before whether we are dumping before weaving
  583. */
  584. protected void dump(String name, byte[] b, boolean before) {
  585. String dirName = getDumpDir();
  586. if (before) {
  587. dirName = dirName + File.separator + "_before";
  588. }
  589. String className = name.replace('.', '/');
  590. final File dir;
  591. if (className.indexOf('/') > 0) {
  592. dir = new File(dirName + File.separator + className.substring(0, className.lastIndexOf('/')));
  593. } else {
  594. dir = new File(dirName);
  595. }
  596. dir.mkdirs();
  597. String fileName = dirName + File.separator + className + ".class";
  598. try {
  599. // System.out.println("WeavingAdaptor.dump() fileName=" + new File(fileName).getAbsolutePath());
  600. FileOutputStream os = new FileOutputStream(fileName);
  601. os.write(b);
  602. os.close();
  603. } catch (IOException ex) {
  604. warn("unable to dump class " + name + " in directory " + dirName, ex);
  605. }
  606. }
  607. /**
  608. * @return the directory in which to dump - default is _ajdump but it
  609. */
  610. protected String getDumpDir() {
  611. return "_ajdump";
  612. }
  613. /**
  614. * Processes messages arising from weaver operations. Tell weaver to abort on any message more severe than warning.
  615. */
  616. protected class WeavingAdaptorMessageHolder extends MessageHandler {
  617. private IMessageHandler delegate;
  618. private List<IMessage> savedMessages;
  619. protected boolean traceMessages = Boolean.getBoolean(TRACE_MESSAGES_PROPERTY);
  620. public WeavingAdaptorMessageHolder(PrintWriter writer) {
  621. this.delegate = new WeavingAdaptorMessageWriter(writer);
  622. super.dontIgnore(IMessage.WEAVEINFO);
  623. }
  624. private void traceMessage(IMessage message) {
  625. if (message instanceof WeaveMessage) {
  626. trace.debug(render(message));
  627. } else if (message.isDebug()) {
  628. trace.debug(render(message));
  629. } else if (message.isInfo()) {
  630. trace.info(render(message));
  631. } else if (message.isWarning()) {
  632. trace.warn(render(message), message.getThrown());
  633. } else if (message.isError()) {
  634. trace.error(render(message), message.getThrown());
  635. } else if (message.isFailed()) {
  636. trace.fatal(render(message), message.getThrown());
  637. } else if (message.isAbort()) {
  638. trace.fatal(render(message), message.getThrown());
  639. } else {
  640. trace.error(render(message), message.getThrown());
  641. }
  642. }
  643. protected String render(IMessage message) {
  644. return "[" + getContextId() + "] " + message.toString();
  645. }
  646. public void flushMessages() {
  647. if (savedMessages == null) {
  648. savedMessages = new ArrayList<>();
  649. savedMessages.addAll(super.getUnmodifiableListView());
  650. clearMessages();
  651. for (IMessage message : savedMessages) {
  652. delegate.handleMessage(message);
  653. }
  654. }
  655. // accumulating = false;
  656. // messages.clear();
  657. }
  658. public void setDelegate(IMessageHandler messageHandler) {
  659. delegate = messageHandler;
  660. }
  661. /*
  662. * IMessageHandler
  663. */
  664. @Override
  665. public boolean handleMessage(IMessage message) throws AbortException {
  666. if (traceMessages) {
  667. traceMessage(message);
  668. }
  669. super.handleMessage(message);
  670. if (abortOnError && 0 <= message.getKind().compareTo(IMessage.ERROR)) {
  671. throw new AbortException(message);
  672. }
  673. // if (accumulating) {
  674. // boolean result = addMessage(message);
  675. // if (abortOnError && 0 <= message.getKind().compareTo(IMessage.ERROR)) {
  676. // throw new AbortException(message);
  677. // }
  678. // return result;
  679. // }
  680. // else return delegate.handleMessage(message);
  681. if (savedMessages != null) {
  682. delegate.handleMessage(message);
  683. }
  684. return true;
  685. }
  686. @Override
  687. public boolean isIgnoring(Kind kind) {
  688. return delegate.isIgnoring(kind);
  689. }
  690. @Override
  691. public void dontIgnore(IMessage.Kind kind) {
  692. if (null != kind && delegate != null) {
  693. delegate.dontIgnore(kind);
  694. }
  695. }
  696. @Override
  697. public void ignore(Kind kind) {
  698. if (null != kind && delegate != null) {
  699. delegate.ignore(kind);
  700. }
  701. }
  702. /*
  703. * IMessageHolder
  704. */
  705. @Override
  706. public List<IMessage> getUnmodifiableListView() {
  707. // System.err.println("? WeavingAdaptorMessageHolder.getUnmodifiableListView() savedMessages=" + savedMessages);
  708. List<IMessage> allMessages = new ArrayList<>();
  709. allMessages.addAll(savedMessages);
  710. allMessages.addAll(super.getUnmodifiableListView());
  711. return allMessages;
  712. }
  713. }
  714. protected class WeavingAdaptorMessageWriter extends MessageWriter {
  715. private final Set<IMessage.Kind> ignoring = new HashSet<>();
  716. private final IMessage.Kind failKind;
  717. public WeavingAdaptorMessageWriter(PrintWriter writer) {
  718. super(writer, true);
  719. ignore(IMessage.WEAVEINFO);
  720. ignore(IMessage.DEBUG);
  721. ignore(IMessage.INFO);
  722. this.failKind = IMessage.ERROR;
  723. }
  724. @Override
  725. public boolean handleMessage(IMessage message) throws AbortException {
  726. // boolean result =
  727. super.handleMessage(message);
  728. if (abortOnError && 0 <= message.getKind().compareTo(failKind)) {
  729. throw new AbortException(message);
  730. }
  731. return true;
  732. }
  733. @Override
  734. public boolean isIgnoring(Kind kind) {
  735. return ((null != kind) && (ignoring.contains(kind)));
  736. }
  737. /**
  738. * Set a message kind to be ignored from now on
  739. */
  740. @Override
  741. public void ignore(IMessage.Kind kind) {
  742. if ((null != kind) && (!ignoring.contains(kind))) {
  743. ignoring.add(kind);
  744. }
  745. }
  746. /**
  747. * Remove a message kind from the list of those ignored from now on.
  748. */
  749. @Override
  750. public void dontIgnore(IMessage.Kind kind) {
  751. if (null != kind) {
  752. ignoring.remove(kind);
  753. }
  754. }
  755. @Override
  756. protected String render(IMessage message) {
  757. return "[" + getContextId() + "] " + super.render(message);
  758. }
  759. }
  760. private class WeavingClassFileProvider implements IClassFileProvider {
  761. private final UnwovenClassFile unwovenClass;
  762. private final List<UnwovenClassFile> unwovenClasses = new ArrayList<>();
  763. private IUnwovenClassFile wovenClass;
  764. private boolean isApplyAtAspectJMungersOnly = false;
  765. public WeavingClassFileProvider(String name, byte[] bytes) {
  766. ensureDelegateInitialized(name, bytes);
  767. this.unwovenClass = new UnwovenClassFile(name, delegateForCurrentClass.getResolvedTypeX().getName(), bytes);
  768. this.unwovenClasses.add(unwovenClass);
  769. if (shouldDump(name.replace('/', '.'), true)) {
  770. dump(name, bytes, true);
  771. }
  772. }
  773. public void setApplyAtAspectJMungersOnly() {
  774. isApplyAtAspectJMungersOnly = true;
  775. }
  776. public boolean isApplyAtAspectJMungersOnly() {
  777. return isApplyAtAspectJMungersOnly;
  778. }
  779. public byte[] getBytes() {
  780. if (wovenClass != null) {
  781. return wovenClass.getBytes();
  782. } else {
  783. return unwovenClass.getBytes();
  784. }
  785. }
  786. public Iterator<UnwovenClassFile> getClassFileIterator() {
  787. return unwovenClasses.iterator();
  788. }
  789. public IWeaveRequestor getRequestor() {
  790. return new IWeaveRequestor() {
  791. public void acceptResult(IUnwovenClassFile result) {
  792. if (wovenClass == null) {
  793. wovenClass = result;
  794. String name = result.getClassName();
  795. if (shouldDump(name.replace('/', '.'), false)) {
  796. dump(name, result.getBytes(), false);
  797. }
  798. } else {
  799. // Classes generated by weaver e.g. around closure advice
  800. String className = result.getClassName();
  801. byte[] resultBytes = result.getBytes();
  802. if (SimpleCacheFactory.isEnabled()) {
  803. SimpleCache lacache=SimpleCacheFactory.createSimpleCache();
  804. lacache.put(result.getClassName(), wovenClass.getBytes(), result.getBytes());
  805. lacache.addGeneratedClassesNames(wovenClass.getClassName(), wovenClass.getBytes(), result.getClassName());
  806. }
  807. generatedClasses.put(className, result);
  808. generatedClasses.put(wovenClass.getClassName(), result);
  809. generatedClassHandler.acceptClass(className, null, resultBytes);
  810. }
  811. }
  812. public void processingReweavableState() {
  813. }
  814. public void addingTypeMungers() {
  815. }
  816. public void weavingAspects() {
  817. }
  818. public void weavingClasses() {
  819. }
  820. public void weaveCompleted() {
  821. // ResolvedType.resetPrimitives();
  822. if (delegateForCurrentClass != null) {
  823. delegateForCurrentClass.weavingCompleted();
  824. }
  825. // ResolvedType.resetPrimitives();
  826. // bcelWorld.discardType(typeBeingProcessed.getResolvedTypeX()); // work in progress
  827. }
  828. };
  829. }
  830. }
  831. public void setActiveProtectionDomain(ProtectionDomain protectionDomain) {
  832. activeProtectionDomain = protectionDomain;
  833. }
  834. }