File tree Expand file tree Collapse file tree 2 files changed +21
-14
lines changed Expand file tree Collapse file tree 2 files changed +21
-14
lines changed Original file line number Diff line number Diff line change @@ -50,6 +50,26 @@ class CommandInjectionSinkType extends SinkType {
50
50
CommandInjectionSinkType ( ) { this = "command-injection" }
51
51
}
52
52
53
+ /** A sink relevant to file storage. */
54
+ class FileContentStoreSinkType extends SinkType {
55
+ FileContentStoreSinkType ( ) { this = "file-content-store" }
56
+ }
57
+
58
+ /** A sink relevant to HTML injection. */
59
+ class HtmlInjectionSinkType extends SinkType {
60
+ HtmlInjectionSinkType ( ) { this = "html-injection" }
61
+ }
62
+
63
+ /** A sink relevant to LDAP injection. */
64
+ class LdapInjectionSinkType extends SinkType {
65
+ LdapInjectionSinkType ( ) { this = "ldap-injection" }
66
+ }
67
+
68
+ /** A sink relevant to URL redirection. */
69
+ class UrlRedirectionSinkType extends SinkType {
70
+ UrlRedirectionSinkType ( ) { this = "url-redirection" }
71
+ }
72
+
53
73
/** A class for source types that can be predicted by a classifier. */
54
74
abstract class SourceType extends EndpointType {
55
75
bindingset [ this ]
Original file line number Diff line number Diff line change @@ -28,20 +28,7 @@ class DollarAtString extends string {
28
28
* descriptions.
29
29
*/
30
30
predicate isKnownKind ( string kind , AutomodelEndpointTypes:: EndpointType type ) {
31
- kind = "path-injection" and
32
- type instanceof AutomodelEndpointTypes:: PathInjectionSinkType
33
- or
34
- kind = "sql-injection" and
35
- type instanceof AutomodelEndpointTypes:: SqlInjectionSinkType
36
- or
37
- kind = "request-forgery" and
38
- type instanceof AutomodelEndpointTypes:: RequestForgerySinkType
39
- or
40
- kind = "command-injection" and
41
- type instanceof AutomodelEndpointTypes:: CommandInjectionSinkType
42
- or
43
- kind = "remote" and
44
- type instanceof AutomodelEndpointTypes:: RemoteSourceType
31
+ kind = type .getKind ( )
45
32
}
46
33
47
34
/**
You can’t perform that action at this time.
0 commit comments