aboutsummaryrefslogtreecommitdiff
path: root/kamon-autoweave/src/main
diff options
context:
space:
mode:
authorIvan Topolnjak <ivantopo@gmail.com>2017-04-24 13:54:40 +0200
committerIvan Topolnjak <ivantopo@gmail.com>2017-04-24 13:54:40 +0200
commit4d828e1a3195e55365c865aa3a78af9668742643 (patch)
tree07fff2683933c96297a8ba577bbdc89888da16e1 /kamon-autoweave/src/main
parent469c11dc1ddb140f407a33f48033e533bf60611c (diff)
downloadKamon-4d828e1a3195e55365c865aa3a78af9668742643.tar.gz
Kamon-4d828e1a3195e55365c865aa3a78af9668742643.tar.bz2
Kamon-4d828e1a3195e55365c865aa3a78af9668742643.zip
Prepare for the major cleanup
Moved all the original files from src/main to src/legacy-main, same with test files. Also removed the autoweave module, examples and bench as I'm planning to have them in separate repositories.
Diffstat (limited to 'kamon-autoweave/src/main')
-rw-r--r--kamon-autoweave/src/main/resources/reference.conf12
-rw-r--r--kamon-autoweave/src/main/scala/kamon/autoweave/Autoweave.scala33
-rw-r--r--kamon-autoweave/src/main/scala/kamon/autoweave/loader/AgentLoader.scala122
-rw-r--r--kamon-autoweave/src/main/scala/kamon/autoweave/loader/AttachmentProviders.scala76
4 files changed, 0 insertions, 243 deletions
diff --git a/kamon-autoweave/src/main/resources/reference.conf b/kamon-autoweave/src/main/resources/reference.conf
deleted file mode 100644
index c47b1a5d..00000000
--- a/kamon-autoweave/src/main/resources/reference.conf
+++ /dev/null
@@ -1,12 +0,0 @@
-# ====================================== #
-# Kamon-Autowave Reference Configuration #
-# ====================================== #
-
-kamon {
- autowave {
- options {
- verbose = false
- showWeaveInfo = false
- }
- }
-} \ No newline at end of file
diff --git a/kamon-autoweave/src/main/scala/kamon/autoweave/Autoweave.scala b/kamon-autoweave/src/main/scala/kamon/autoweave/Autoweave.scala
deleted file mode 100644
index 9c349c9e..00000000
--- a/kamon-autoweave/src/main/scala/kamon/autoweave/Autoweave.scala
+++ /dev/null
@@ -1,33 +0,0 @@
-/* =========================================================================================
- * Copyright © 2013-2016 the kamon project <http://kamon.io/>
- *
- * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
- * except in compliance with the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software distributed under the
- * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
- * either express or implied. See the License for the specific language governing permissions
- * and limitations under the License.
- * =========================================================================================
- */
-
-package kamon.autoweave
-
-import kamon.Kamon
-import kamon.autoweave.loader.AgentLoader
-import org.aspectj.weaver.loadtime.Agent
-
-class Autoweave {
- val config = Kamon.config.getConfig("kamon.autowave.options")
- val verbose = config.getBoolean("verbose")
- val showWeaveInfo = config.getBoolean("showWeaveInfo")
-
- def attach(): Unit = {
- System.setProperty("aj.weaving.verbose", verbose.toString)
- System.setProperty("org.aspectj.weaver.showWeaveInfo", showWeaveInfo.toString)
-
- AgentLoader.attachAgentToJVM(classOf[Agent])
- }
-}
diff --git a/kamon-autoweave/src/main/scala/kamon/autoweave/loader/AgentLoader.scala b/kamon-autoweave/src/main/scala/kamon/autoweave/loader/AgentLoader.scala
deleted file mode 100644
index 5271e987..00000000
--- a/kamon-autoweave/src/main/scala/kamon/autoweave/loader/AgentLoader.scala
+++ /dev/null
@@ -1,122 +0,0 @@
-/* =========================================================================================
- * Copyright © 2013-2015 the kamon project <http://kamon.io/>
- *
- * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
- * except in compliance with the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software distributed under the
- * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
- * either express or implied. See the License for the specific language governing permissions
- * and limitations under the License.
- * =========================================================================================
- */
-
-package kamon.autoweave.loader
-
-import java.io.{File, FileOutputStream, InputStream}
-import java.lang.management.ManagementFactory
-import java.util.jar.Attributes.Name
-import java.util.jar.{JarEntry, JarOutputStream, Manifest}
-
-import scala.util.control.NoStackTrace
-
-object AgentLoader {
-
- /**
- * Gets the current JVM PID
- *
- * @return Returns the PID
- */
-
- private def getPidFromRuntimeMBean: String = {
- val jvm = ManagementFactory.getRuntimeMXBean.getName
- val pid = jvm.substring(0, jvm.indexOf('@'))
- pid
- }
-
- /**
- * Loads an agent into a JVM.
- *
- * @param agent The main agent class.
- * @param resources Array of classes to be included with agent.
- */
- def attachAgentToJVM(agent: Class[_], resources: Seq[Class[_]] = Seq.empty): Unit = attachToRunningJVM(agent, resources)
-
- /**
- * Java variant
- *
- * @param agent
- */
- def attachAgentToJVM(agent: Class[_]): Unit = attachAgentToJVM(agent, Seq.empty)
-
- /**
- * Generates a temporary agent file to be loaded.
- *
- * @param agent The main agent class.
- * @param resources Array of classes to be included with agent.
- * @return Returns a temporary jar file with the specified classes included.
- */
- private def generateAgentJar(agent: Class[_], resources: Seq[Class[_]]): File = {
- val jarFile = File.createTempFile("agent", ".jar")
- jarFile.deleteOnExit()
-
- val manifest = new Manifest()
- val mainAttributes = manifest.getMainAttributes
- // Create manifest stating that agent is allowed to transform classes
- mainAttributes.put(Name.MANIFEST_VERSION, "1.0")
- mainAttributes.put(new Name("Agent-Class"), agent.getName)
- mainAttributes.put(new Name("Can-Retransform-Classes"), "true")
- mainAttributes.put(new Name("Can-Redefine-Classes"), "true")
- mainAttributes.put(new Name("Can-Set-Native-Method-Prefix"), "true")
-
- val jos = new JarOutputStream(new FileOutputStream(jarFile), manifest)
-
- jos.putNextEntry(new JarEntry(agent.getName.replace('.', '/') + ".class"))
-
- jos.write(getBytesFromStream(agent.getClassLoader.getResourceAsStream(unqualify(agent))))
- jos.closeEntry()
-
- for (clazz ← resources) {
- val name = unqualify(clazz)
- jos.putNextEntry(new JarEntry(name))
- jos.write(getBytesFromStream(clazz.getClassLoader.getResourceAsStream(name)))
- jos.closeEntry()
- }
-
- jos.close()
- jarFile
- }
-
- /**
- * Attach to the running JVM.
- *
- * @return
- * Returns the attached VirtualMachine
- */
- private def attachToRunningJVM(agent: Class[_], resources: Seq[Class[_]]): Unit = {
- AttachmentProviders.resolve() match {
- case Some(virtualMachine) ⇒
- val virtualMachineInstance = virtualMachine.getDeclaredMethod("attach", classOf[String]).invoke(null, getPidFromRuntimeMBean)
- virtualMachine.getDeclaredMethod("loadAgent", classOf[String], classOf[String]).invoke(virtualMachineInstance, generateAgentJar(agent, resources).getAbsolutePath, "")
- virtualMachine.getDeclaredMethod("detach").invoke(virtualMachineInstance)
- case None ⇒ throw new IllegalStateException("Cannot read the virtual machine type...") with NoStackTrace
- }
- }
-
- /**
- * Gets bytes from InputStream.
- *
- * @param stream
- * The InputStream.
- * @return
- * Returns a byte[] representation of given stream.
- */
- private def getBytesFromStream(stream: InputStream): Array[Byte] = {
- Stream.continually(stream.read).takeWhile(_ != -1).map(_.toByte).toArray
- }
-
- private def unqualify(clazz: Class[_]): String = clazz.getName.replace('.', '/') + ".class"
-}
-
diff --git a/kamon-autoweave/src/main/scala/kamon/autoweave/loader/AttachmentProviders.scala b/kamon-autoweave/src/main/scala/kamon/autoweave/loader/AttachmentProviders.scala
deleted file mode 100644
index 1640d2c6..00000000
--- a/kamon-autoweave/src/main/scala/kamon/autoweave/loader/AttachmentProviders.scala
+++ /dev/null
@@ -1,76 +0,0 @@
-/* =========================================================================================
- * Copyright © 2013-2016 the kamon project <http://kamon.io/>
- *
- * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
- * except in compliance with the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software distributed under the
- * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
- * either express or implied. See the License for the specific language governing permissions
- * and limitations under the License.
- * =========================================================================================
- */
-
-package kamon.autoweave.loader
-
-import java.io.File
-import java.net.{URL, URLClassLoader}
-import java.security.{AccessController, PrivilegedAction}
-
-import scala.util.control.NonFatal
-import scala.util.{Failure, Success, Try}
-
-object AttachmentProviders {
-
- sealed trait AttachmentProvider {
- def resolve(): Option[Class[_]]
- }
-
- trait DefaultAttachmentProvider extends AttachmentProvider {
- val VirtualMachineTyeName = "com.sun.tools.attach.VirtualMachine"
-
- def toolsJarPath: String
-
- /**
- * Gets the current HotSpotVirtualMachine implementation otherwise a None.
- *
- * @return
- * Returns the HotSpotVirtualMachine implementation of the running JVM.
- */
- def resolve(): Option[Class[_]] = {
- val toolsJar = new File(System.getProperty("java.home").replace('\\', '/') + "/" + toolsJarPath)
- if (toolsJar.isFile && toolsJar.canRead) {
- Try(AccessController.doPrivileged(new ClassLoaderCreationAction(toolsJar)).loadClass(VirtualMachineTyeName)) match {
- case Success(vm) ⇒ Some(vm)
- case Failure(NonFatal(_)) ⇒ None
- }
- } else None
- }
- }
-
- case object JVM extends DefaultAttachmentProvider { val toolsJarPath = "../lib/tools.jar" }
- case object JDK extends DefaultAttachmentProvider { val toolsJarPath = "lib/tools.jar" }
- case object MAC extends DefaultAttachmentProvider { val toolsJarPath = "../Classes/classes.jar" }
- case object IBM extends AttachmentProvider {
-
- val IBMVirtualMachineTyeName = "com.ibm.tools.attach.VirtualMachine"
-
- override def resolve(): Option[Class[_]] = {
- Try(ClassLoader.getSystemClassLoader.loadClass(IBMVirtualMachineTyeName)) match {
- case Success(vm) ⇒ Some(vm)
- case Failure(NonFatal(_)) ⇒ None
- }
- }
- }
-
- private val providers = Seq(JVM, JDK, MAC, IBM)
-
- private final class ClassLoaderCreationAction(toolsJar: File) extends PrivilegedAction[ClassLoader] {
- override def run(): ClassLoader = new URLClassLoader(Array[URL](toolsJar.toURI.toURL), null)
- }
-
- def resolve(): Option[Class[_]] =
- providers.iterator.map(_.resolve()).collectFirst { case Some(clazz) ⇒ clazz }
-}