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 31KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970
  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.isVMGreaterOrEqual(9)) {
  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 in the internal form of fully qualified class and interface names as defined
  287. * in <i>The Java Virtual Machine Specification</i>. For example, <code>"java/util/List"</code>.
  288. * @param bytes the input byte buffer in class file format - must not be modified
  289. * @param mustWeave if true then this class <i>must</i> get woven (used for concrete aspects generated from XML)
  290. *
  291. * @return a well-formed class file buffer (the weaving result), or {@code null} if no weaving was performed
  292. *
  293. * @throws IOException weave failed
  294. *
  295. * @see java.lang.instrument.ClassFileTransformer#transform(ClassLoader, String, Class, ProtectionDomain, byte[])
  296. */
  297. public byte[] weaveClass(String name, final byte[] bytes, boolean mustWeave) throws IOException {
  298. if (trace == null) {
  299. // Pr231945: we are likely to be under tomcat and ENABLE_CLEAR_REFERENCES hasn't been set
  300. System.err
  301. .println("AspectJ Weaver cannot continue to weave, static state has been cleared. Are you under Tomcat? In order to weave '"
  302. + name
  303. + "' 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).");
  304. return null;
  305. }
  306. if (weaverRunning.get()) {
  307. // System.out.println("AJC: avoiding re-entrant call to transform " + name);
  308. return null;
  309. }
  310. try {
  311. byte[] newBytes = null;
  312. weaverRunning.set(true);
  313. if (trace.isTraceEnabled()) {
  314. trace.enter("weaveClass", this, new Object[] { name, bytes });
  315. }
  316. if (!enabled) {
  317. if (trace.isTraceEnabled()) {
  318. trace.exit("weaveClass", false);
  319. }
  320. return null;
  321. }
  322. boolean debugOn = !messageHandler.isIgnoring(Message.DEBUG);
  323. name = name.replace('/', '.');
  324. byte[] wovenBytes = wovenWithGeneratedClass(name);
  325. if (wovenBytes != null) {
  326. if (debugOn) {
  327. debug("returning woven bytes for '" + name + "' that were generated by a previous weaving process");
  328. }
  329. return wovenBytes;
  330. }
  331. try {
  332. delegateForCurrentClass = null;
  333. if (shouldWeaveName(name)) {
  334. if (accept(name, bytes)) {
  335. // Determine if we have the weaved class cached
  336. CachedClassReference cacheKey = null;
  337. if (cache != null && !mustWeave) {
  338. cacheKey = cache.createCacheKey(name, bytes);
  339. CachedClassEntry entry = cache.get(cacheKey, bytes);
  340. if (entry != null) {
  341. // If the entry has been explicitly ignored
  342. // return the original bytes
  343. if (entry.isIgnored()) {
  344. return null;
  345. }
  346. return entry.getBytes();
  347. }
  348. }
  349. // TODO @AspectJ problem
  350. // Annotation style aspects need to be included regardless in order to get
  351. // a valid aspectOf()/hasAspect() generated in them. However - if they are excluded
  352. // (via include/exclude in aop.xml) they really should only get aspectOf()/hasAspect()
  353. // and not be included in the full set of aspects being applied by 'this' weaver
  354. if (debugOn) {
  355. debug("weaving '" + name + "'");
  356. }
  357. newBytes = getWovenBytes(name, bytes);
  358. // TODO: Is this OK performance-wise?
  359. if (Arrays.equals(bytes, newBytes)) {
  360. // null means unchanged in java.lang.instrument.ClassFileTransformer::transform
  361. newBytes = null;
  362. }
  363. // temporarily out - searching for @Aspect annotated types is a slow thing to do - we should
  364. // expect the user to name them if they want them woven - just like code style
  365. // } else if (shouldWeaveAnnotationStyleAspect(name, bytes)) {
  366. // if (mustWeave) {
  367. // if (bcelWorld.getLint().mustWeaveXmlDefinedAspects.isEnabled()) {
  368. // bcelWorld.getLint().mustWeaveXmlDefinedAspects.signal(name, null);
  369. // }
  370. // }
  371. // // an @AspectJ aspect needs to be at least munged by the aspectOf munger
  372. // if (debugOn) {
  373. // debug("weaving '" + name + "'");
  374. // }
  375. // bytes = getAtAspectJAspectBytes(name, bytes);
  376. // Add the weaved class to the cache only if there
  377. // has been an actual change
  378. // JVK: Is there a better way to check if the class has
  379. // been transformed without carrying up some value
  380. // from the depths?
  381. if (cacheKey != null) {
  382. // If no transform has been applied, mark the class
  383. // as ignored.
  384. if (newBytes == null) {
  385. cache.ignore(cacheKey, bytes);
  386. } else {
  387. cache.put(cacheKey, bytes, newBytes);
  388. }
  389. }
  390. } else if (debugOn) {
  391. debug("not weaving '" + name + "'");
  392. }
  393. } else if (debugOn) {
  394. debug("cannot weave '" + name + "'");
  395. }
  396. } finally {
  397. delegateForCurrentClass = null;
  398. }
  399. if (trace.isTraceEnabled()) {
  400. trace.exit("weaveClass", newBytes);
  401. }
  402. return newBytes;
  403. } finally {
  404. weaverRunning.remove();
  405. }
  406. }
  407. /**
  408. * Return the bytes from a (parallel?) weaving process that generated an inner class, e.g. to support Around closures.
  409. * This is done instead of weaving again, as weaving would generate another inner class.
  410. * @param name
  411. * @return the cached bytes of a previously woven class, or null if not found
  412. */
  413. private byte[] wovenWithGeneratedClass(String name) {
  414. IUnwovenClassFile woven = generatedClasses.get(name);
  415. if (woven == null) {
  416. return null;
  417. }
  418. return woven.getBytes();
  419. }
  420. // ATAJ
  421. protected boolean accept(String name, byte[] bytes) {
  422. return true;
  423. }
  424. protected boolean shouldDump(String name, boolean before) {
  425. return false;
  426. }
  427. private boolean shouldWeaveName(String name) {
  428. if (PACKAGE_INITIAL_CHARS.indexOf(name.charAt(0)) != -1) {
  429. if ((weavingSpecialTypes & INITIALIZED) == 0) {
  430. weavingSpecialTypes |= INITIALIZED;
  431. // initialize it
  432. Properties p = weaver.getWorld().getExtraConfiguration();
  433. if (p != null) {
  434. boolean b = p.getProperty(World.xsetWEAVE_JAVA_PACKAGES, "false").equalsIgnoreCase("true");
  435. if (b) {
  436. weavingSpecialTypes |= WEAVE_JAVA_PACKAGE;
  437. }
  438. b = p.getProperty(World.xsetWEAVE_JAVAX_PACKAGES, "false").equalsIgnoreCase("true");
  439. if (b) {
  440. weavingSpecialTypes |= WEAVE_JAVAX_PACKAGE;
  441. }
  442. }
  443. }
  444. if (name.startsWith(ASPECTJ_BASE_PACKAGE)) {
  445. return false;
  446. }
  447. if (name.startsWith("sun.reflect.")) {// JDK reflect
  448. return false;
  449. }
  450. if (name.startsWith("javax.")) {
  451. if ((weavingSpecialTypes & WEAVE_JAVAX_PACKAGE) != 0) {
  452. return true;
  453. } else {
  454. if (!haveWarnedOnJavax) {
  455. haveWarnedOnJavax = true;
  456. warn("javax.* types are not being woven because the weaver option '-Xset:weaveJavaxPackages=true' has not been specified");
  457. }
  458. return false;
  459. }
  460. }
  461. if (name.startsWith("java.")) {
  462. if ((weavingSpecialTypes & WEAVE_JAVA_PACKAGE) != 0) {
  463. return true;
  464. } else {
  465. return false;
  466. }
  467. }
  468. }
  469. // boolean should = !(name.startsWith("org.aspectj.")
  470. // || (name.startsWith("java.") && (weavingSpecialTypes & WEAVE_JAVA_PACKAGE) == 0)
  471. // || (name.startsWith("javax.") && (weavingSpecialTypes & WEAVE_JAVAX_PACKAGE) == 0)
  472. // // || name.startsWith("$Proxy")//JDK proxies//FIXME AV is that 1.3 proxy ? fe. ataspect.$Proxy0 is a java5 proxy...
  473. // || name.startsWith("sun.reflect."));
  474. return true;
  475. }
  476. /**
  477. * We allow @AJ aspect weaving so that we can add aspectOf() as part of the weaving (and not part of the source compilation)
  478. *
  479. * @param name
  480. * @param bytes bytecode (from classloader), allow to NOT lookup stuff on disk again during resolve
  481. * @return true if @Aspect
  482. */
  483. private boolean shouldWeaveAnnotationStyleAspect(String name, byte[] bytes) {
  484. if (delegateForCurrentClass == null) {
  485. // if (weaver.getWorld().isASMAround()) return asmCheckAnnotationStyleAspect(bytes);
  486. // else
  487. ensureDelegateInitialized(name, bytes);
  488. }
  489. return (delegateForCurrentClass.isAnnotationStyleAspect());
  490. }
  491. // private boolean asmCheckAnnotationStyleAspect(byte[] bytes) {
  492. // IsAtAspectAnnotationVisitor detector = new IsAtAspectAnnotationVisitor();
  493. //
  494. // ClassReader cr = new ClassReader(bytes);
  495. // try {
  496. // cr.accept(detector, true);//, ClassReader.SKIP_DEBUG | ClassReader.SKIP_CODE | ClassReader.SKIP_FRAMES);
  497. // } catch (Exception spe) {
  498. // // if anything goes wrong, e.g., an NPE, then assume it's NOT an @AspectJ aspect...
  499. // System.err.println("Unexpected problem parsing bytes to discover @Aspect annotation");
  500. // spe.printStackTrace();
  501. // return false;
  502. // }
  503. //
  504. // return detector.isAspect();
  505. // }
  506. protected void ensureDelegateInitialized(String name, byte[] bytes) {
  507. if (delegateForCurrentClass == null) {
  508. BcelWorld world = (BcelWorld) weaver.getWorld();
  509. delegateForCurrentClass = world.addSourceObjectType(name, bytes, false);
  510. }
  511. }
  512. /**
  513. * Weave a set of bytes defining a class.
  514. *
  515. * @param name the name of the class being woven
  516. * @param bytes the bytes that define the class
  517. * @return byte[] the woven bytes for the class
  518. * @throws IOException
  519. */
  520. private byte[] getWovenBytes(String name, byte[] bytes) throws IOException {
  521. WeavingClassFileProvider wcp = new WeavingClassFileProvider(name, bytes);
  522. weaver.weave(wcp);
  523. return wcp.getBytes();
  524. }
  525. /**
  526. * Weave a set of bytes defining a class for only what is needed to turn @AspectJ aspect in a usefull form ie with aspectOf
  527. * method - see #113587
  528. *
  529. * @param name the name of the class being woven
  530. * @param bytes the bytes that define the class
  531. * @return byte[] the woven bytes for the class
  532. * @throws IOException
  533. */
  534. private byte[] getAtAspectJAspectBytes(String name, byte[] bytes) throws IOException {
  535. WeavingClassFileProvider wcp = new WeavingClassFileProvider(name, bytes);
  536. wcp.setApplyAtAspectJMungersOnly();
  537. weaver.weave(wcp);
  538. return wcp.getBytes();
  539. }
  540. private void registerAspectLibraries(List<String> aspectPath) {
  541. // System.err.println("? WeavingAdaptor.registerAspectLibraries(" + aspectPath + ")");
  542. for (String libName : aspectPath) {
  543. addAspectLibrary(libName);
  544. }
  545. weaver.prepareForWeave();
  546. }
  547. /*
  548. * Register an aspect library with this classloader for use during weaving. This class loader will also return (unmodified) any
  549. * of the classes in the library in response to a <code>findClass()</code> request. The library is not required to be on the
  550. * weavingClasspath given when this classloader was constructed.
  551. *
  552. * @param aspectLibraryJarFile a jar file representing an aspect library
  553. *
  554. * @throws IOException
  555. */
  556. private void addAspectLibrary(String aspectLibraryName) {
  557. File aspectLibrary = new File(aspectLibraryName);
  558. if (aspectLibrary.isDirectory() || (FileUtil.isZipFile(aspectLibrary))) {
  559. try {
  560. info("adding aspect library: '" + aspectLibrary + "'");
  561. weaver.addLibraryJarFile(aspectLibrary);
  562. } catch (IOException ex) {
  563. error("exception adding aspect library: '" + ex + "'");
  564. }
  565. } else {
  566. error("bad aspect library: '" + aspectLibrary + "'");
  567. }
  568. }
  569. private static List<String> makeClasspath(String cp) {
  570. List<String> ret = new ArrayList<>();
  571. if (cp != null) {
  572. StringTokenizer tok = new StringTokenizer(cp, File.pathSeparator);
  573. while (tok.hasMoreTokens()) {
  574. ret.add(tok.nextToken());
  575. }
  576. }
  577. return ret;
  578. }
  579. protected boolean debug(String message) {
  580. return MessageUtil.debug(messageHandler, message);
  581. }
  582. protected boolean info(String message) {
  583. return MessageUtil.info(messageHandler, message);
  584. }
  585. protected boolean warn(String message) {
  586. return MessageUtil.warn(messageHandler, message);
  587. }
  588. protected boolean warn(String message, Throwable th) {
  589. return messageHandler.handleMessage(new Message(message, IMessage.WARNING, th, null));
  590. }
  591. protected boolean error(String message) {
  592. return MessageUtil.error(messageHandler, message);
  593. }
  594. protected boolean error(String message, Throwable th) {
  595. return messageHandler.handleMessage(new Message(message, IMessage.ERROR, th, null));
  596. }
  597. public String getContextId() {
  598. return "WeavingAdaptor";
  599. }
  600. /**
  601. * Dump the given bytcode in _dump/... (dev mode)
  602. *
  603. * @param name
  604. * @param b
  605. * @param before whether we are dumping before weaving
  606. */
  607. protected void dump(String name, byte[] b, boolean before) {
  608. String dirName = getDumpDir();
  609. if (before) {
  610. dirName = dirName + File.separator + "_before";
  611. }
  612. String className = name.replace('.', '/');
  613. final File dir;
  614. if (className.indexOf('/') > 0) {
  615. dir = new File(dirName + File.separator + className.substring(0, className.lastIndexOf('/')));
  616. } else {
  617. dir = new File(dirName);
  618. }
  619. dir.mkdirs();
  620. String fileName = dirName + File.separator + className + ".class";
  621. try {
  622. // System.out.println("WeavingAdaptor.dump() fileName=" + new File(fileName).getAbsolutePath());
  623. FileOutputStream os = new FileOutputStream(fileName);
  624. os.write(b);
  625. os.close();
  626. } catch (IOException ex) {
  627. warn("unable to dump class " + name + " in directory " + dirName, ex);
  628. }
  629. }
  630. /**
  631. * @return the directory in which to dump - default is _ajdump but it
  632. */
  633. protected String getDumpDir() {
  634. return "_ajdump";
  635. }
  636. /**
  637. * Processes messages arising from weaver operations. Tell weaver to abort on any message more severe than warning.
  638. */
  639. protected class WeavingAdaptorMessageHolder extends MessageHandler {
  640. private IMessageHandler delegate;
  641. private List<IMessage> savedMessages;
  642. protected boolean traceMessages = Boolean.getBoolean(TRACE_MESSAGES_PROPERTY);
  643. public WeavingAdaptorMessageHolder(PrintWriter writer) {
  644. this.delegate = new WeavingAdaptorMessageWriter(writer);
  645. super.dontIgnore(IMessage.WEAVEINFO);
  646. }
  647. private void traceMessage(IMessage message) {
  648. if (message instanceof WeaveMessage) {
  649. trace.debug(render(message));
  650. } else if (message.isDebug()) {
  651. trace.debug(render(message));
  652. } else if (message.isInfo()) {
  653. trace.info(render(message));
  654. } else if (message.isWarning()) {
  655. trace.warn(render(message), message.getThrown());
  656. } else if (message.isError()) {
  657. trace.error(render(message), message.getThrown());
  658. } else if (message.isFailed()) {
  659. trace.fatal(render(message), message.getThrown());
  660. } else if (message.isAbort()) {
  661. trace.fatal(render(message), message.getThrown());
  662. } else {
  663. trace.error(render(message), message.getThrown());
  664. }
  665. }
  666. protected String render(IMessage message) {
  667. return "[" + getContextId() + "] " + message.toString();
  668. }
  669. public void flushMessages() {
  670. if (savedMessages == null) {
  671. savedMessages = new ArrayList<>();
  672. savedMessages.addAll(super.getUnmodifiableListView());
  673. clearMessages();
  674. for (IMessage message : savedMessages) {
  675. delegate.handleMessage(message);
  676. }
  677. }
  678. // accumulating = false;
  679. // messages.clear();
  680. }
  681. public void setDelegate(IMessageHandler messageHandler) {
  682. delegate = messageHandler;
  683. }
  684. /*
  685. * IMessageHandler
  686. */
  687. @Override
  688. public boolean handleMessage(IMessage message) throws AbortException {
  689. if (traceMessages) {
  690. traceMessage(message);
  691. }
  692. super.handleMessage(message);
  693. if (abortOnError && 0 <= message.getKind().compareTo(IMessage.ERROR)) {
  694. throw new AbortException(message);
  695. }
  696. // if (accumulating) {
  697. // boolean result = addMessage(message);
  698. // if (abortOnError && 0 <= message.getKind().compareTo(IMessage.ERROR)) {
  699. // throw new AbortException(message);
  700. // }
  701. // return result;
  702. // }
  703. // else return delegate.handleMessage(message);
  704. if (savedMessages != null) {
  705. delegate.handleMessage(message);
  706. }
  707. return true;
  708. }
  709. @Override
  710. public boolean isIgnoring(Kind kind) {
  711. return delegate.isIgnoring(kind);
  712. }
  713. @Override
  714. public void dontIgnore(IMessage.Kind kind) {
  715. if (null != kind && delegate != null) {
  716. delegate.dontIgnore(kind);
  717. }
  718. }
  719. @Override
  720. public void ignore(Kind kind) {
  721. if (null != kind && delegate != null) {
  722. delegate.ignore(kind);
  723. }
  724. }
  725. /*
  726. * IMessageHolder
  727. */
  728. @Override
  729. public List<IMessage> getUnmodifiableListView() {
  730. // System.err.println("? WeavingAdaptorMessageHolder.getUnmodifiableListView() savedMessages=" + savedMessages);
  731. List<IMessage> allMessages = new ArrayList<>();
  732. allMessages.addAll(savedMessages);
  733. allMessages.addAll(super.getUnmodifiableListView());
  734. return allMessages;
  735. }
  736. }
  737. protected class WeavingAdaptorMessageWriter extends MessageWriter {
  738. private final Set<IMessage.Kind> ignoring = new HashSet<>();
  739. private final IMessage.Kind failKind;
  740. public WeavingAdaptorMessageWriter(PrintWriter writer) {
  741. super(writer, true);
  742. ignore(IMessage.WEAVEINFO);
  743. ignore(IMessage.DEBUG);
  744. ignore(IMessage.INFO);
  745. this.failKind = IMessage.ERROR;
  746. }
  747. @Override
  748. public boolean handleMessage(IMessage message) throws AbortException {
  749. // boolean result =
  750. super.handleMessage(message);
  751. if (abortOnError && 0 <= message.getKind().compareTo(failKind)) {
  752. throw new AbortException(message);
  753. }
  754. return true;
  755. }
  756. @Override
  757. public boolean isIgnoring(Kind kind) {
  758. return ((null != kind) && (ignoring.contains(kind)));
  759. }
  760. /**
  761. * Set a message kind to be ignored from now on
  762. */
  763. @Override
  764. public void ignore(IMessage.Kind kind) {
  765. if ((null != kind) && (!ignoring.contains(kind))) {
  766. ignoring.add(kind);
  767. }
  768. }
  769. /**
  770. * Remove a message kind from the list of those ignored from now on.
  771. */
  772. @Override
  773. public void dontIgnore(IMessage.Kind kind) {
  774. if (null != kind) {
  775. ignoring.remove(kind);
  776. }
  777. }
  778. @Override
  779. protected String render(IMessage message) {
  780. return "[" + getContextId() + "] " + super.render(message);
  781. }
  782. }
  783. private class WeavingClassFileProvider implements IClassFileProvider {
  784. private final UnwovenClassFile unwovenClass;
  785. private final List<UnwovenClassFile> unwovenClasses = new ArrayList<>();
  786. private IUnwovenClassFile wovenClass;
  787. private boolean isApplyAtAspectJMungersOnly = false;
  788. public WeavingClassFileProvider(String name, byte[] bytes) {
  789. ensureDelegateInitialized(name, bytes);
  790. this.unwovenClass = new UnwovenClassFile(name, delegateForCurrentClass.getResolvedTypeX().getName(), bytes);
  791. this.unwovenClasses.add(unwovenClass);
  792. if (shouldDump(name.replace('/', '.'), true)) {
  793. dump(name, bytes, true);
  794. }
  795. }
  796. public void setApplyAtAspectJMungersOnly() {
  797. isApplyAtAspectJMungersOnly = true;
  798. }
  799. public boolean isApplyAtAspectJMungersOnly() {
  800. return isApplyAtAspectJMungersOnly;
  801. }
  802. public byte[] getBytes() {
  803. if (wovenClass != null) {
  804. return wovenClass.getBytes();
  805. } else {
  806. return unwovenClass.getBytes();
  807. }
  808. }
  809. public Iterator<UnwovenClassFile> getClassFileIterator() {
  810. return unwovenClasses.iterator();
  811. }
  812. public IWeaveRequestor getRequestor() {
  813. return new WeaveRequestor();
  814. }
  815. private class WeaveRequestor implements IWeaveRequestor {
  816. @Override
  817. public void acceptResult(IUnwovenClassFile result) {
  818. if (wovenClass == null) {
  819. wovenClass = result;
  820. String name = result.getClassName();
  821. if (shouldDump(name.replace('/', '.'), false)) {
  822. dump(name, result.getBytes(), false);
  823. }
  824. } else {
  825. // Classes generated by weaver e.g. around closure advice
  826. String className = result.getClassName();
  827. byte[] resultBytes = result.getBytes();
  828. if (SimpleCacheFactory.isEnabled()) {
  829. SimpleCache lacache=SimpleCacheFactory.createSimpleCache();
  830. lacache.put(result.getClassName(), wovenClass.getBytes(), result.getBytes());
  831. lacache.addGeneratedClassesNames(wovenClass.getClassName(), wovenClass.getBytes(), result.getClassName());
  832. }
  833. generatedClasses.put(className, result);
  834. generatedClasses.put(wovenClass.getClassName(), wovenClass);
  835. generatedClassHandler.acceptClass(className, null, resultBytes);
  836. }
  837. }
  838. public void processingReweavableState() {}
  839. public void addingTypeMungers() {}
  840. public void weavingAspects() {}
  841. public void weavingClasses() {}
  842. public void weaveCompleted() {
  843. // ResolvedType.resetPrimitives();
  844. if (delegateForCurrentClass != null) {
  845. delegateForCurrentClass.weavingCompleted();
  846. }
  847. // ResolvedType.resetPrimitives();
  848. // bcelWorld.discardType(typeBeingProcessed.getResolvedTypeX()); // work in progress
  849. }
  850. }
  851. }
  852. public void setActiveProtectionDomain(ProtectionDomain protectionDomain) {
  853. activeProtectionDomain = protectionDomain;
  854. }
  855. }