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
|
/*
* Copyright 2007 Google Inc.
*
* 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 com.tonicsystems.jarjar;
import com.tonicsystems.jarjar.util.*;
import java.io.*;
import java.util.*;
import org.objectweb.asm.*;
import org.objectweb.asm.commons.*;
// TODO: this can probably be refactored into JarClassVisitor, etc.
class KeepProcessor extends Remapper implements JarProcessor {
private final ClassVisitor cv = new ClassRemapper(new EmptyClassVisitor(), this);
private final List<Wildcard> wildcards;
private final List<String> roots = new ArrayList<String>();
private final Map<String, Set<String>> depend = new HashMap<String, Set<String>>();
public KeepProcessor(List<Keep> patterns) {
wildcards = PatternElement.createWildcards(patterns);
}
public boolean isEnabled() {
return !wildcards.isEmpty();
}
public Set<String> getExcludes() {
Set<String> closure = new HashSet<String>();
closureHelper(closure, roots);
Set<String> removable = new HashSet<String>(depend.keySet());
removable.removeAll(closure);
return removable;
}
private void closureHelper(Set<String> closure, Collection<String> process) {
if (process == null) {
return;
}
for (String name : process) {
if (closure.add(name)) {
closureHelper(closure, depend.get(name));
}
}
}
private Set<String> curSet;
private byte[] buf = new byte[0x2000];
public boolean process(EntryStruct struct) throws IOException {
try {
if (struct.name.endsWith(".class")) {
String name = struct.name.substring(0, struct.name.length() - 6);
for (Wildcard wildcard : wildcards)
if (wildcard.matches(name)) {
roots.add(name);
}
depend.put(name, curSet = new HashSet<String>());
new ClassReader(new ByteArrayInputStream(struct.data))
.accept(cv, ClassReader.EXPAND_FRAMES);
curSet.remove(name);
}
} catch (Exception e) {
System.err.println("Error reading " + struct.name + ": " + e.getMessage());
}
return true;
}
public String map(String key) {
if (key.startsWith("java/") || key.startsWith("javax/")) {
return null;
}
curSet.add(key);
return null;
}
public Object mapValue(Object value) {
if (value instanceof String) {
String s = (String) value;
if (PackageRemapper.isArrayForName(s)) {
mapDesc(s.replace('.', '/'));
} else if (isForName(s)) {
map(s.replace('.', '/'));
}
return value;
} else {
return super.mapValue(value);
}
}
// TODO: use this for package remapping too?
private static boolean isForName(String value) {
if (value.equals("")) {
return false;
}
for (int i = 0, len = value.length(); i < len; i++) {
char c = value.charAt(i);
if (c != '.' && !Character.isJavaIdentifierPart(c)) {
return false;
}
}
return true;
}
}
|
