Skip to content

Commit

Permalink
feat(java): define imported classes
Browse files Browse the repository at this point in the history
  • Loading branch information
didroe committed Jul 1, 2024
1 parent d8df782 commit fb0ac3c
Show file tree
Hide file tree
Showing 5 changed files with 95 additions and 0 deletions.
30 changes: 30 additions & 0 deletions pkg/languages/java/.snapshots/TestImport-import.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
high:
- rule:
cwe_ids:
- "42"
id: import_test
title: Test import handling
description: Test import handling
documentation_url: ""
line_number: 7
full_filename: import.java
filename: import.java
source:
location:
start: 7
end: 7
column:
start: 9
end: 21
sink:
location:
start: 7
end: 7
column:
start: 9
end: 21
content: ""
parent_line_number: 7
fingerprint: fd41a77f77dc09c75355f0d8bf69d976_0
old_fingerprint: fd41a77f77dc09c75355f0d8bf69d976_0

26 changes: 26 additions & 0 deletions pkg/languages/java/analyzer/analyzer.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package analyzer

import (
"slices"
"strings"

sitter "github.com/smacker/go-tree-sitter"

Expand Down Expand Up @@ -48,6 +49,8 @@ func (analyzer *analyzer) Analyze(node *sitter.Node, visitChildren func() error)
return analyzer.withScope(language.NewScope(analyzer.scope), func() error {
return visitChildren()
})
case "import_declaration":
return analyzer.analyzeImport(node, visitChildren)
case "assignment_expression":
return analyzer.analyzeAssignment(node, visitChildren)
case "variable_declarator":
Expand Down Expand Up @@ -84,6 +87,29 @@ func (analyzer *analyzer) Analyze(node *sitter.Node, visitChildren func() error)
}
}

// import foo.Bar;
// import foo.*;
// import static foo.Bar;
func (analyzer *analyzer) analyzeImport(node *sitter.Node, visitChildren func() error) error {
// package import
if node.NamedChildCount() != 1 {
return nil
}

identifier := node.NamedChild(0)

content := analyzer.builder.ContentFor(node)
prefix := content[:identifier.StartByte()-node.StartByte()]
if strings.Contains(prefix, "static") {
return nil
}

name := identifier.ChildByFieldName("name")
analyzer.scope.Declare(analyzer.builder.ContentFor(name), name)
analyzer.builder.Alias(name, identifier)
return nil
}

// foo = a
// foo += a
func (analyzer *analyzer) analyzeAssignment(node *sitter.Node, visitChildren func() error) error {
Expand Down
7 changes: 7 additions & 0 deletions pkg/languages/java/java_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,19 @@ import (
patternquerybuilder "github.com/bearer/bearer/pkg/scanner/detectors/customrule/patternquery/builder"
)

//go:embed testdata/import.yml
var importRule []byte

//go:embed testdata/logger.yml
var loggerRule []byte

//go:embed testdata/scope_rule.yml
var scopeRule []byte

func TestImport(t *testing.T) {
testhelper.GetRunner(t, importRule, java.Get()).RunTest(t, "./testdata/import", ".snapshots/")
}

func TestFlow(t *testing.T) {
testhelper.GetRunner(t, loggerRule, java.Get()).RunTest(t, "./testdata/testcases/flow", ".snapshots/flow/")
}
Expand Down
21 changes: 21 additions & 0 deletions pkg/languages/java/testdata/import.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
languages:
- java
patterns:
- pattern: sink($<IMPORT>)
filters:
- variable: IMPORT
detection: flow_test_source
scope: cursor
auxiliary:
- id: flow_test_source
patterns:
- import $<!>foo.Import
- import $<!>foo.Import2
- import $<!>foo.Import3
severity: high
metadata:
description: Test import handling
remediation_message: Test import handling
cwe_id:
- 42
id: import_test
11 changes: 11 additions & 0 deletions pkg/languages/java/testdata/import/import.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
import foo.Import;
import foo.Import2.*;
import static foo.Import3;

class A {
public void exec() {
sink(Import);
sink(Import2); // no match
sink(Import3); // no match
}
}

0 comments on commit fb0ac3c

Please sign in to comment.