diff --git a/core/analyzer.go b/core/analyzer.go index 0fa8a48..c55c901 100644 --- a/core/analyzer.go +++ b/core/analyzer.go @@ -34,6 +34,14 @@ type ImportInfo struct { InitOnly map[string]bool } +func NewImportInfo() *ImportInfo { + return &ImportInfo{ + make(map[string]string), + make(map[string]string), + make(map[string]bool), + } +} + // The Context is populated with data parsed from the source code as it is scanned. // It is passed through to all rule functions as they are called. Rules may use // this data in conjunction withe the encoutered AST node. @@ -92,11 +100,7 @@ func NewAnalyzer(conf map[string]interface{}, logger *log.Logger) Analyzer { nil, nil, nil, - &ImportInfo{ - make(map[string]string), - make(map[string]string), - make(map[string]bool), - }, + nil, }, logger: logger, } @@ -130,6 +134,7 @@ func (gas *Analyzer) process(filename string, source interface{}) error { return err } + gas.context.Imports = NewImportInfo() for _, pkg := range gas.context.Pkg.Imports() { gas.context.Imports.Imported[pkg.Path()] = pkg.Name() } diff --git a/rules/rand.go b/rules/rand.go index 47ac55d..6626cad 100644 --- a/rules/rand.go +++ b/rules/rand.go @@ -15,6 +15,7 @@ package rules import ( + "fmt" "go/ast" gas "github.com/GoASTScanner/gas/core" @@ -28,6 +29,16 @@ type WeakRand struct { func (w *WeakRand) Match(n ast.Node, c *gas.Context) (*gas.Issue, error) { if _, matched := gas.MatchCallByPackage(n, c, w.packagePath, w.funcName); matched { + fmt.Println("Imported:") + for k, v := range c.Imports.Imported { + fmt.Printf("%s => %s\n", k, v) + } + fmt.Println("Aliased:") + for k, v := range c.Imports.Aliased { + fmt.Printf("%s => %s\n", k, v) + } + fmt.Println("----------------------------------------") + return gas.NewIssue(c, n, w.What, w.Severity, w.Confidence), nil }