forked from soot-oss/soot
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathCoffiClassSource.java
More file actions
126 lines (110 loc) · 3.66 KB
/
CoffiClassSource.java
File metadata and controls
126 lines (110 loc) · 3.66 KB
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
package soot;
/*-
* #%L
* Soot - a J*va Optimization Framework
* %%
* Copyright (C) 2004 Ondrej Lhotak
* %%
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation, either version 2.1 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Lesser Public License for more details.
*
* You should have received a copy of the GNU General Lesser Public
* License along with this program. If not, see
* <http://www.gnu.org/licenses/lgpl-2.1.html>.
* #L%
*/
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import soot.javaToJimple.IInitialResolver;
import soot.javaToJimple.IInitialResolver.Dependencies;
import soot.options.Options;
import soot.tagkit.SourceFileTag;
/**
* A class source for resolving from .class files through coffi.
*/
public class CoffiClassSource extends ClassSource {
private static final Logger logger = LoggerFactory.getLogger(CoffiClassSource.class);
private IFoundFile foundFile;
private InputStream classFile;
private final String fileName;
private final String zipFileName;
public CoffiClassSource(String className, IFoundFile foundFile) {
super(className);
if (foundFile == null) {
throw new IllegalStateException("Error: The FoundFile must not be null.");
}
this.foundFile = foundFile;
this.classFile = foundFile.inputStream();
this.fileName = foundFile.getFile().getAbsolutePath();
this.zipFileName = !foundFile.isZipFile() ? null : foundFile.getFilePath();
}
public CoffiClassSource(String className, InputStream classFile, String fileName) {
super(className);
if (classFile == null || fileName == null) {
throw new IllegalStateException("Error: The class file input strean and file name must not be null.");
}
this.classFile = classFile;
this.fileName = fileName;
this.zipFileName = null;
this.foundFile = null;
}
@Override
public Dependencies resolve(SootClass sc) {
if (Options.v().verbose()) {
logger.debug("resolving [from .class]: " + className);
}
List<Type> references = new ArrayList<Type>();
try {
soot.coffi.Util.v().resolveFromClassFile(sc, classFile, fileName, references);
} finally {
close();
}
addSourceFileTag(sc);
IInitialResolver.Dependencies deps = new IInitialResolver.Dependencies();
deps.typesToSignature.addAll(references);
return deps;
}
private void addSourceFileTag(soot.SootClass sc) {
if (fileName == null && zipFileName == null) {
return;
}
SourceFileTag tag = (SourceFileTag) sc.getTag(SourceFileTag.NAME);
if (tag == null) {
tag = new SourceFileTag();
sc.addTag(tag);
}
// Sets sourceFile only when it hasn't been set before
if (tag.getSourceFile() == null) {
String name = zipFileName == null ? new File(fileName).getName() : new File(zipFileName).getName();
tag.setSourceFile(name);
}
}
@Override
public void close() {
try {
if (classFile != null) {
classFile.close();
classFile = null;
}
} catch (IOException e) {
throw new RuntimeException("Error: Failed to close source input stream.", e);
} finally {
if (foundFile != null) {
foundFile.close();
foundFile = null;
}
}
}
}