Skip to content

Commit 1b9634b

Browse files
author
‘niuerzhuang’
committed
feature: custom tag config.
1 parent 5f9d9a7 commit 1b9634b

File tree

3 files changed

+9
-18
lines changed

3 files changed

+9
-18
lines changed

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/models/taint/tag/TaintTag.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ public enum TaintTag {
6363

6464
private final String key;
6565

66-
private static final Map<String, TaintTag> LOOKUP = new HashMap<String, TaintTag>();
66+
private static final Map<String, TaintTag> LOOKUP = new HashMap<>();
6767

6868
static {
6969
for (TaintTag t : TaintTag.values()) {

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/vulscan/VulnType.java

Lines changed: 4 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
public enum VulnType {
88

99
/**
10-
* 漏洞
10+
* 漏洞类型
1111
*/
1212
SQL_OVER_POWER("sql-over-power", "info", false),
1313
CRYPTO_WEAK_RANDOMNESS("crypto-weak-randomness", "low", false),
@@ -40,9 +40,9 @@ public String getName() {
4040
/**
4141
* 漏洞类型 值
4242
*/
43-
String name;
44-
String weight;
45-
boolean tracked;
43+
final String name;
44+
final String weight;
45+
final boolean tracked;
4646

4747
VulnType(String name, String weight, boolean tracked) {
4848
this.name = name;
@@ -54,13 +54,4 @@ public String getName() {
5454
public boolean equals(String name) {
5555
return this.name.equals(name);
5656
}
57-
58-
public static VulnType getTypeByName(String name) {
59-
for (VulnType vType : VulnType.values()) {
60-
if (vType.equals(name)) {
61-
return vType;
62-
}
63-
}
64-
return null;
65-
}
6657
}

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/vulscan/dynamic/DynamicPropagatorScanner.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,18 +23,18 @@
2323
* @author dongzhiyong@huoxian.cn
2424
*/
2525
public class DynamicPropagatorScanner implements IVulScan {
26-
private final static Set<SinkSafeChecker> SAFE_CHECKERS = new HashSet<SinkSafeChecker>(Arrays.asList(
26+
private final static Set<SinkSafeChecker> SAFE_CHECKERS = new HashSet<>(Arrays.asList(
2727
new FastjsonCheck(),
2828
new XXECheck()
2929
));
3030

31-
private final static Set<SinkSourceChecker> SOURCE_CHECKERS = new HashSet<SinkSourceChecker>(Arrays.asList(
31+
private final static Set<SinkSourceChecker> SOURCE_CHECKERS = new HashSet<>(Arrays.asList(
3232
new PathTraversalCheck(),
3333
new SSRFSourceCheck(),
3434
new UnvalidatedRedirectCheck()
3535
));
3636

37-
private static final Set<ServiceTrace> SERVICE_TRACES = new HashSet<ServiceTrace>(Collections.singletonList(
37+
private static final Set<ServiceTrace> SERVICE_TRACES = new HashSet<>(Collections.singletonList(
3838
new HttpService()
3939
));
4040

@@ -170,7 +170,7 @@ private boolean sinkSourceHitTaintPool(MethodEvent event, SinkNode sinkNode) {
170170
}
171171
}
172172

173-
List<Object> sourceInstances = new ArrayList<Object>();
173+
List<Object> sourceInstances = new ArrayList<>();
174174
boolean hasTaint = false;
175175
boolean objHasTaint = false;
176176
Set<TaintPosition> sources = sinkNode.getSources();

0 commit comments

Comments
 (0)