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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/*
* =========================================================================================
* Copyright © 2013-2014 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.system.native
import java.io._
import java.util
import java.util.logging.Logger
import java.util.{ ArrayList, List }
import org.hyperic.sigar.{ SigarProxy, SigarProxyCache }
import scala.annotation.tailrec
import scala.collection.JavaConversions._
import scala.io.Source
object SigarLoader {
val Version = "1.6.4"
val JavaLibraryPath = "java.library.path"
val TmpDir = "java.io.tmpdir"
val IndexFile = "/kamon/system/native/index"
val UsrPathField = "usr_paths"
private val log = Logger.getLogger("SigarLoader")
def sigarProxy = init(new File(System.getProperty(TmpDir)))
private[native] def init(baseTmp: File): SigarProxy = {
val tmpDir = createTmpDir(baseTmp)
for (lib ← loadIndex) copy(lib, tmpDir)
attachToLibraryPath(tmpDir)
try {
val sigar = SigarProxyCache.newInstance()
sigar.getPid
sigar
} catch {
case t: Throwable ⇒ {
log.severe("Failed to load sigar")
throw new RuntimeException(t)
}
}
}
private[native] val usrPathField = {
val usrPathField = classOf[ClassLoader].getDeclaredField(UsrPathField)
usrPathField.setAccessible(true)
usrPathField
}
private[native] def attachToLibraryPath(dir: File): Unit = {
val dirAbsolute = dir.getAbsolutePath
System.setProperty(JavaLibraryPath, newLibraryPath(dirAbsolute))
var paths = usrPathField.get(null).asInstanceOf[Array[String]]
if (paths == null) paths = new Array[String](0)
for (path ← paths) if (path == dirAbsolute) return
val newPaths = util.Arrays.copyOf(paths, paths.length + 1)
newPaths(newPaths.length - 1) = dirAbsolute
usrPathField.set(null, newPaths)
}
private[native] def newLibraryPath(dirAbsolutePath: String): String = {
Option(System.getProperty(JavaLibraryPath)).fold(dirAbsolutePath)(oldValue ⇒ s"$dirAbsolutePath${File.pathSeparator}$oldValue")
}
private[native] def copy(lib: String, tmpDir: File) {
val target = new File(tmpDir, lib)
if (target.exists()) return
write(classOf[Loader].getResourceAsStream(lib), target)
}
private[native] def createTmpDir(baseTmp: File): File = {
val tmpDir = new File(baseTmp, s"sigar-$Version")
if (!tmpDir.exists()) {
if (!tmpDir.mkdirs()) throw new RuntimeException(s"Could not create temp sigar directory: ${tmpDir.getAbsolutePath}")
}
if (!tmpDir.isDirectory) throw new RuntimeException(s"sigar temp directory path is not a directory: ${tmpDir.getAbsolutePath}")
if (!tmpDir.canWrite()) throw new RuntimeException(s"sigar temp directory not writeable: ${tmpDir.getAbsolutePath}")
tmpDir
}
private[native] def loadIndex(): List[String] = {
val libs = new ArrayList[String]()
val is = classOf[Loader].getResourceAsStream(IndexFile)
for (line ← Source.fromInputStream(is).getLines()) {
val currentLine = line.trim()
libs add currentLine
}
libs
}
private[native] def write(input: InputStream, to: File) {
val out = new FileOutputStream(to)
try {
transfer(input, out)
} finally {
out.close()
}
}
private[native] def transfer(input: InputStream, out: OutputStream) {
val buffer = new Array[Byte](8192)
@tailrec def transfer() {
val read = input.read(buffer)
if (read >= 0) {
out.write(buffer, 0, read)
transfer()
}
}
transfer()
}
class Loader private[native]
}
|
