| // Copyright 2022 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package govulncheck |
| |
| import ( |
| "fmt" |
| "strings" |
| |
| "golang.org/x/tools/go/packages" |
| "golang.org/x/vuln/vulncheck" |
| ) |
| |
| // A PackageError contains errors from loading a set of packages. |
| type PackageError struct { |
| Errors []packages.Error |
| } |
| |
| func (e *PackageError) Error() string { |
| var b strings.Builder |
| fmt.Fprintln(&b, "Packages contain errors:") |
| for _, e := range e.Errors { |
| fmt.Println(&b, e) |
| } |
| return b.String() |
| } |
| |
| // LoadPackages loads the packages matching patterns using cfg, after setting |
| // the cfg mode flags that vulncheck needs for analysis. |
| // If the packages contain errors, a PackageError is returned containing a list of the errors, |
| // along with the packages themselves. |
| func LoadPackages(cfg *packages.Config, patterns ...string) ([]*vulncheck.Package, error) { |
| cfg.Mode |= packages.NeedName | packages.NeedFiles | packages.NeedCompiledGoFiles | |
| packages.NeedImports | packages.NeedTypes | packages.NeedTypesSizes | |
| packages.NeedSyntax | packages.NeedTypesInfo | packages.NeedDeps | |
| packages.NeedModule |
| |
| pkgs, err := packages.Load(cfg, patterns...) |
| vpkgs := vulncheck.Convert(pkgs) |
| if err != nil { |
| return nil, err |
| } |
| var perrs []packages.Error |
| packages.Visit(pkgs, nil, func(p *packages.Package) { |
| perrs = append(perrs, p.Errors...) |
| }) |
| if len(perrs) > 0 { |
| err = &PackageError{perrs} |
| } |
| return vpkgs, err |
| } |