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 | |
package com.sun.tools.javafx.script; |
27 | |
|
28 | |
import java.io.File; |
29 | |
import java.net.MalformedURLException; |
30 | |
import java.net.URL; |
31 | |
import java.net.URLClassLoader; |
32 | |
import java.util.ArrayList; |
33 | |
import java.util.Map; |
34 | |
import java.util.List; |
35 | |
import java.util.StringTokenizer; |
36 | |
|
37 | |
|
38 | |
|
39 | |
|
40 | |
|
41 | |
|
42 | |
public final class MemoryClassLoader extends URLClassLoader { |
43 | |
private Map<String, byte[]> classBytes; |
44 | |
private URL source; |
45 | |
|
46 | |
public MemoryClassLoader(URL source, Map<String, |
47 | |
byte[]> classBytes, String classPath, ClassLoader parent) { |
48 | 18 | super(toURLs(classPath), parent); |
49 | 18 | this.source = source; |
50 | 18 | this.classBytes = classBytes; |
51 | 18 | } |
52 | |
|
53 | |
public Class load(String className) throws ClassNotFoundException { |
54 | 0 | return loadClass(className); |
55 | |
} |
56 | |
|
57 | |
public Iterable<Class> loadAll() throws ClassNotFoundException { |
58 | 18 | List<Class> classes = new ArrayList<Class>(classBytes.size()); |
59 | 18 | for (String name : classBytes.keySet()) { |
60 | 40 | classes.add(loadClass(name)); |
61 | |
} |
62 | 18 | return classes; |
63 | |
} |
64 | |
|
65 | |
@Override |
66 | |
protected Class findClass(String className) throws ClassNotFoundException { |
67 | 40 | byte[] buf = classBytes.get(className); |
68 | 40 | if (buf != null) { |
69 | |
|
70 | 40 | classBytes.put(className, null); |
71 | 40 | return defineClass(className, buf, 0, buf.length); |
72 | |
} else { |
73 | 0 | return super.findClass(className); |
74 | |
} |
75 | |
} |
76 | |
|
77 | |
@Override |
78 | |
public URL findResource(String name) { |
79 | 1 | if (name.endsWith(".class")) { |
80 | 1 | name = name.substring(0, name.length() - 6).replace('/', '.'); |
81 | 1 | if (classBytes.containsKey(name)) |
82 | 1 | return source; |
83 | |
} |
84 | 0 | return super.findResource(name); |
85 | |
} |
86 | |
|
87 | |
private static URL[] toURLs(String classPath) { |
88 | 18 | if (classPath == null) { |
89 | 0 | return new URL[0]; |
90 | |
} |
91 | |
|
92 | 18 | List<URL> list = new ArrayList<URL>(); |
93 | 18 | StringTokenizer st = new StringTokenizer(classPath, File.pathSeparator); |
94 | 378 | while (st.hasMoreTokens()) { |
95 | 360 | String token = st.nextToken(); |
96 | 360 | File file = new File(token); |
97 | 360 | if (file.exists()) { |
98 | |
try { |
99 | 360 | list.add(file.toURI().toURL()); |
100 | 360 | } catch (MalformedURLException mue) {} |
101 | |
} else { |
102 | |
try { |
103 | 0 | list.add(new URL(token)); |
104 | 0 | } catch (MalformedURLException mue) {} |
105 | |
} |
106 | 360 | } |
107 | 18 | URL[] res = new URL[list.size()]; |
108 | 18 | list.toArray(res); |
109 | 18 | return res; |
110 | |
} |
111 | |
} |