first commit

1 files changed, 146 insertions, 0 deletions

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..a8a4e4e
--- /dev/null
+++ b/main.go
@@ -0,0 +1,146 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"go/ast"
+	"go/parser"
+	"go/token"
+	"io/fs"
+	"io/ioutil"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+
+	"golang.org/x/mod/modfile"
+)
+
+var (
+	file     = flag.String("mod", "go.mod", "go.mod file")
+	cache    = flag.String("modcache", getmodcache(), "location of go mod cache")
+	excheckm = flag.String("exempt", "golang.org", "domains exempt from the search, seperated by space")
+
+	checkInit   = flag.Bool("ci", false, "check for implementations of init()")
+	checkUnsafe = flag.Bool("cu", false, "check for imports of unsafe")
+	checkCgo    = flag.Bool("cc", false, "check for imports of cgo")
+	checkPprof  = flag.Bool("cp", false, "check for imports of net/http/pprof")
+
+	exceptions []string
+
+	relcache string
+)
+
+func getmodcache() string {
+	if c := os.Getenv("GOMODCACHE"); c != "" {
+		return c
+	} else if c = os.Getenv("GOPATH"); c != "" {
+		return c + "/pkg/mod"
+	} else {
+		return os.Getenv("HOME") + "/pkg/mod"
+	}
+
+}
+
+func main() {
+	flag.Parse()
+
+	exceptions = strings.Fields(*excheckm)
+	relcache = strings.Replace(*cache, os.Getenv("HOME"), "~", 1)
+
+	if !(*checkInit || *checkUnsafe || *checkCgo || *checkPprof) {
+		fmt.Println("Nothing to check. Use -ci|-cu|-cc|-cp")
+		return
+	}
+
+	data, err := ioutil.ReadFile(*file)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	mf, err := modfile.Parse(*file, data, nil)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	var wg sync.WaitGroup
+	wg.Add(len(mf.Require))
+	for _, r := range mf.Require {
+		path := strings.Join(r.Syntax.Token, "@")
+		go func() {
+			defer wg.Done()
+			checkPath(*cache + "/" + path)
+		}()
+	}
+
+	wg.Wait()
+
+}
+
+func isExempt(path string) bool {
+	for _, pattern := range exceptions {
+		if strings.Contains(path, pattern) {
+			return true
+		}
+	}
+	return false
+}
+
+func modpath(filename string) string {
+	return filepath.Join(relcache, strings.TrimPrefix(filename, *cache+"/"))
+}
+
+func checkPath(path string) {
+	var fset = token.NewFileSet()
+
+	filter := func(inf fs.FileInfo) bool {
+		return !strings.HasSuffix(inf.Name(), "_test.go")
+	}
+
+	filepath.WalkDir(path, func(p string, info fs.DirEntry, err error) error {
+		if err != nil {
+			return err
+		} else if !info.IsDir() {
+			return nil
+		} else if isExempt(p) {
+			return nil
+		}
+
+		pkgs, err := parser.ParseDir(fset, p, filter, parser.Mode(0))
+		if err != nil {
+			return err
+		}
+
+		for _, pkg := range pkgs {
+			for filename, file := range pkg.Files {
+				if *checkUnsafe || *checkCgo || *checkPprof {
+					for _, imp := range file.Imports {
+						if *checkUnsafe && imp.Path.Value == `"unsafe"` {
+							fmt.Println("unsafe in", modpath(filename))
+						} else if *checkCgo && imp.Path.Value == `"C"` {
+							fmt.Println("cgo    in", modpath(filename))
+						} else if *checkPprof && imp.Path.Value == `"net/http/pprof"` {
+							fmt.Println("pprof  in", modpath(filename))
+						}
+					}
+				}
+
+				if !*checkInit {
+					break
+				}
+			INIT:
+				for _, decl := range file.Decls {
+					if f, ok := decl.(*ast.FuncDecl); ok {
+						if f.Name.Name == "init" {
+							fmt.Println("init   in", modpath(filename))
+							break INIT
+						}
+					}
+				}
+			}
+		}
+
+		return nil
+	})
+}