blob: fff237359b064c11c2e6fc4dc38aa01f61933e31 (
plain) (
tree)
|
|
/* =========================================================================================
* 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 }
}
|