1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
/* =========================================================================================
* 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 }
}
|