mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-29 23:30:04 +01:00
1817 lines
45 KiB
Go
1817 lines
45 KiB
Go
|
// Package simple contains a linter for Go source code.
|
||
|
package simple // import "honnef.co/go/tools/simple"
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"go/ast"
|
||
|
"go/constant"
|
||
|
"go/token"
|
||
|
"go/types"
|
||
|
"reflect"
|
||
|
"sort"
|
||
|
"strings"
|
||
|
|
||
|
"golang.org/x/tools/go/analysis"
|
||
|
"golang.org/x/tools/go/analysis/passes/inspect"
|
||
|
"golang.org/x/tools/go/ast/inspector"
|
||
|
"golang.org/x/tools/go/types/typeutil"
|
||
|
. "honnef.co/go/tools/arg"
|
||
|
"honnef.co/go/tools/internal/passes/buildssa"
|
||
|
"honnef.co/go/tools/internal/sharedcheck"
|
||
|
"honnef.co/go/tools/lint"
|
||
|
. "honnef.co/go/tools/lint/lintdsl"
|
||
|
)
|
||
|
|
||
|
func LintSingleCaseSelect(pass *analysis.Pass) (interface{}, error) {
|
||
|
isSingleSelect := func(node ast.Node) bool {
|
||
|
v, ok := node.(*ast.SelectStmt)
|
||
|
if !ok {
|
||
|
return false
|
||
|
}
|
||
|
return len(v.Body.List) == 1
|
||
|
}
|
||
|
|
||
|
seen := map[ast.Node]struct{}{}
|
||
|
fn := func(node ast.Node) {
|
||
|
switch v := node.(type) {
|
||
|
case *ast.ForStmt:
|
||
|
if len(v.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if !isSingleSelect(v.Body.List[0]) {
|
||
|
return
|
||
|
}
|
||
|
if _, ok := v.Body.List[0].(*ast.SelectStmt).Body.List[0].(*ast.CommClause).Comm.(*ast.SendStmt); ok {
|
||
|
// Don't suggest using range for channel sends
|
||
|
return
|
||
|
}
|
||
|
seen[v.Body.List[0]] = struct{}{}
|
||
|
ReportNodefFG(pass, node, "should use for range instead of for { select {} }")
|
||
|
case *ast.SelectStmt:
|
||
|
if _, ok := seen[v]; ok {
|
||
|
return
|
||
|
}
|
||
|
if !isSingleSelect(v) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, node, "should use a simple channel send/receive instead of select with a single case")
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.ForStmt)(nil), (*ast.SelectStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintLoopCopy(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
loop := node.(*ast.RangeStmt)
|
||
|
|
||
|
if loop.Key == nil {
|
||
|
return
|
||
|
}
|
||
|
if len(loop.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
stmt, ok := loop.Body.List[0].(*ast.AssignStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if stmt.Tok != token.ASSIGN || len(stmt.Lhs) != 1 || len(stmt.Rhs) != 1 {
|
||
|
return
|
||
|
}
|
||
|
lhs, ok := stmt.Lhs[0].(*ast.IndexExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if _, ok := pass.TypesInfo.TypeOf(lhs.X).(*types.Slice); !ok {
|
||
|
return
|
||
|
}
|
||
|
lidx, ok := lhs.Index.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
key, ok := loop.Key.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if pass.TypesInfo.TypeOf(lhs) == nil || pass.TypesInfo.TypeOf(stmt.Rhs[0]) == nil {
|
||
|
return
|
||
|
}
|
||
|
if pass.TypesInfo.ObjectOf(lidx) != pass.TypesInfo.ObjectOf(key) {
|
||
|
return
|
||
|
}
|
||
|
if !types.Identical(pass.TypesInfo.TypeOf(lhs), pass.TypesInfo.TypeOf(stmt.Rhs[0])) {
|
||
|
return
|
||
|
}
|
||
|
if _, ok := pass.TypesInfo.TypeOf(loop.X).(*types.Slice); !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if rhs, ok := stmt.Rhs[0].(*ast.IndexExpr); ok {
|
||
|
rx, ok := rhs.X.(*ast.Ident)
|
||
|
_ = rx
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
ridx, ok := rhs.Index.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if pass.TypesInfo.ObjectOf(ridx) != pass.TypesInfo.ObjectOf(key) {
|
||
|
return
|
||
|
}
|
||
|
} else if rhs, ok := stmt.Rhs[0].(*ast.Ident); ok {
|
||
|
value, ok := loop.Value.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if pass.TypesInfo.ObjectOf(rhs) != pass.TypesInfo.ObjectOf(value) {
|
||
|
return
|
||
|
}
|
||
|
} else {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, loop, "should use copy() instead of a loop")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.RangeStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintIfBoolCmp(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
expr := node.(*ast.BinaryExpr)
|
||
|
if expr.Op != token.EQL && expr.Op != token.NEQ {
|
||
|
return
|
||
|
}
|
||
|
x := IsBoolConst(pass, expr.X)
|
||
|
y := IsBoolConst(pass, expr.Y)
|
||
|
if !x && !y {
|
||
|
return
|
||
|
}
|
||
|
var other ast.Expr
|
||
|
var val bool
|
||
|
if x {
|
||
|
val = BoolConst(pass, expr.X)
|
||
|
other = expr.Y
|
||
|
} else {
|
||
|
val = BoolConst(pass, expr.Y)
|
||
|
other = expr.X
|
||
|
}
|
||
|
basic, ok := pass.TypesInfo.TypeOf(other).Underlying().(*types.Basic)
|
||
|
if !ok || basic.Kind() != types.Bool {
|
||
|
return
|
||
|
}
|
||
|
op := ""
|
||
|
if (expr.Op == token.EQL && !val) || (expr.Op == token.NEQ && val) {
|
||
|
op = "!"
|
||
|
}
|
||
|
r := op + Render(pass, other)
|
||
|
l1 := len(r)
|
||
|
r = strings.TrimLeft(r, "!")
|
||
|
if (l1-len(r))%2 == 1 {
|
||
|
r = "!" + r
|
||
|
}
|
||
|
if IsInTest(pass, node) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, expr, "should omit comparison to bool constant, can be simplified to %s", r)
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.BinaryExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintBytesBufferConversions(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
if len(call.Args) != 1 {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
argCall, ok := call.Args[0].(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
sel, ok := argCall.Fun.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
typ := pass.TypesInfo.TypeOf(call.Fun)
|
||
|
if typ == types.Universe.Lookup("string").Type() && IsCallToAST(pass, call.Args[0], "(*bytes.Buffer).Bytes") {
|
||
|
ReportNodefFG(pass, call, "should use %v.String() instead of %v", Render(pass, sel.X), Render(pass, call))
|
||
|
} else if typ, ok := typ.(*types.Slice); ok && typ.Elem() == types.Universe.Lookup("byte").Type() && IsCallToAST(pass, call.Args[0], "(*bytes.Buffer).String") {
|
||
|
ReportNodefFG(pass, call, "should use %v.Bytes() instead of %v", Render(pass, sel.X), Render(pass, call))
|
||
|
}
|
||
|
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintStringsContains(pass *analysis.Pass) (interface{}, error) {
|
||
|
// map of value to token to bool value
|
||
|
allowed := map[int64]map[token.Token]bool{
|
||
|
-1: {token.GTR: true, token.NEQ: true, token.EQL: false},
|
||
|
0: {token.GEQ: true, token.LSS: false},
|
||
|
}
|
||
|
fn := func(node ast.Node) {
|
||
|
expr := node.(*ast.BinaryExpr)
|
||
|
switch expr.Op {
|
||
|
case token.GEQ, token.GTR, token.NEQ, token.LSS, token.EQL:
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
|
||
|
value, ok := ExprToInt(pass, expr.Y)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
allowedOps, ok := allowed[value]
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
b, ok := allowedOps[expr.Op]
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
call, ok := expr.X.(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
sel, ok := call.Fun.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
pkgIdent, ok := sel.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
funIdent := sel.Sel
|
||
|
if pkgIdent.Name != "strings" && pkgIdent.Name != "bytes" {
|
||
|
return
|
||
|
}
|
||
|
newFunc := ""
|
||
|
switch funIdent.Name {
|
||
|
case "IndexRune":
|
||
|
newFunc = "ContainsRune"
|
||
|
case "IndexAny":
|
||
|
newFunc = "ContainsAny"
|
||
|
case "Index":
|
||
|
newFunc = "Contains"
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
|
||
|
prefix := ""
|
||
|
if !b {
|
||
|
prefix = "!"
|
||
|
}
|
||
|
ReportNodefFG(pass, node, "should use %s%s.%s(%s) instead", prefix, pkgIdent.Name, newFunc, RenderArgs(pass, call.Args))
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.BinaryExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintBytesCompare(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
expr := node.(*ast.BinaryExpr)
|
||
|
if expr.Op != token.NEQ && expr.Op != token.EQL {
|
||
|
return
|
||
|
}
|
||
|
call, ok := expr.X.(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !IsCallToAST(pass, call, "bytes.Compare") {
|
||
|
return
|
||
|
}
|
||
|
value, ok := ExprToInt(pass, expr.Y)
|
||
|
if !ok || value != 0 {
|
||
|
return
|
||
|
}
|
||
|
args := RenderArgs(pass, call.Args)
|
||
|
prefix := ""
|
||
|
if expr.Op == token.NEQ {
|
||
|
prefix = "!"
|
||
|
}
|
||
|
ReportNodefFG(pass, node, "should use %sbytes.Equal(%s) instead", prefix, args)
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.BinaryExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintForTrue(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
loop := node.(*ast.ForStmt)
|
||
|
if loop.Init != nil || loop.Post != nil {
|
||
|
return
|
||
|
}
|
||
|
if !IsBoolConst(pass, loop.Cond) || !BoolConst(pass, loop.Cond) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, loop, "should use for {} instead of for true {}")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.ForStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintRegexpRaw(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
if !IsCallToAST(pass, call, "regexp.MustCompile") &&
|
||
|
!IsCallToAST(pass, call, "regexp.Compile") {
|
||
|
return
|
||
|
}
|
||
|
sel, ok := call.Fun.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if len(call.Args) != 1 {
|
||
|
// invalid function call
|
||
|
return
|
||
|
}
|
||
|
lit, ok := call.Args[Arg("regexp.Compile.expr")].(*ast.BasicLit)
|
||
|
if !ok {
|
||
|
// TODO(dominikh): support string concat, maybe support constants
|
||
|
return
|
||
|
}
|
||
|
if lit.Kind != token.STRING {
|
||
|
// invalid function call
|
||
|
return
|
||
|
}
|
||
|
if lit.Value[0] != '"' {
|
||
|
// already a raw string
|
||
|
return
|
||
|
}
|
||
|
val := lit.Value
|
||
|
if !strings.Contains(val, `\\`) {
|
||
|
return
|
||
|
}
|
||
|
if strings.Contains(val, "`") {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
bs := false
|
||
|
for _, c := range val {
|
||
|
if !bs && c == '\\' {
|
||
|
bs = true
|
||
|
continue
|
||
|
}
|
||
|
if bs && c == '\\' {
|
||
|
bs = false
|
||
|
continue
|
||
|
}
|
||
|
if bs {
|
||
|
// backslash followed by non-backslash -> escape sequence
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ReportNodefFG(pass, call, "should use raw string (`...`) with regexp.%s to avoid having to escape twice", sel.Sel.Name)
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintIfReturn(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
block := node.(*ast.BlockStmt)
|
||
|
l := len(block.List)
|
||
|
if l < 2 {
|
||
|
return
|
||
|
}
|
||
|
n1, n2 := block.List[l-2], block.List[l-1]
|
||
|
|
||
|
if len(block.List) >= 3 {
|
||
|
if _, ok := block.List[l-3].(*ast.IfStmt); ok {
|
||
|
// Do not flag a series of if statements
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
// if statement with no init, no else, a single condition
|
||
|
// checking an identifier or function call and just a return
|
||
|
// statement in the body, that returns a boolean constant
|
||
|
ifs, ok := n1.(*ast.IfStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if ifs.Else != nil || ifs.Init != nil {
|
||
|
return
|
||
|
}
|
||
|
if len(ifs.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if op, ok := ifs.Cond.(*ast.BinaryExpr); ok {
|
||
|
switch op.Op {
|
||
|
case token.EQL, token.LSS, token.GTR, token.NEQ, token.LEQ, token.GEQ:
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
ret1, ok := ifs.Body.List[0].(*ast.ReturnStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if len(ret1.Results) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if !IsBoolConst(pass, ret1.Results[0]) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ret2, ok := n2.(*ast.ReturnStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if len(ret2.Results) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if !IsBoolConst(pass, ret2.Results[0]) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if ret1.Results[0].(*ast.Ident).Name == ret2.Results[0].(*ast.Ident).Name {
|
||
|
// we want the function to return true and false, not the
|
||
|
// same value both times.
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ReportNodefFG(pass, n1, "should use 'return <expr>' instead of 'if <expr> { return <bool> }; return <bool>'")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.BlockStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
// LintRedundantNilCheckWithLen checks for the following reduntant nil-checks:
|
||
|
//
|
||
|
// if x == nil || len(x) == 0 {}
|
||
|
// if x != nil && len(x) != 0 {}
|
||
|
// if x != nil && len(x) == N {} (where N != 0)
|
||
|
// if x != nil && len(x) > N {}
|
||
|
// if x != nil && len(x) >= N {} (where N != 0)
|
||
|
//
|
||
|
func LintRedundantNilCheckWithLen(pass *analysis.Pass) (interface{}, error) {
|
||
|
isConstZero := func(expr ast.Expr) (isConst bool, isZero bool) {
|
||
|
_, ok := expr.(*ast.BasicLit)
|
||
|
if ok {
|
||
|
return true, IsZero(expr)
|
||
|
}
|
||
|
id, ok := expr.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return false, false
|
||
|
}
|
||
|
c, ok := pass.TypesInfo.ObjectOf(id).(*types.Const)
|
||
|
if !ok {
|
||
|
return false, false
|
||
|
}
|
||
|
return true, c.Val().Kind() == constant.Int && c.Val().String() == "0"
|
||
|
}
|
||
|
|
||
|
fn := func(node ast.Node) {
|
||
|
// check that expr is "x || y" or "x && y"
|
||
|
expr := node.(*ast.BinaryExpr)
|
||
|
if expr.Op != token.LOR && expr.Op != token.LAND {
|
||
|
return
|
||
|
}
|
||
|
eqNil := expr.Op == token.LOR
|
||
|
|
||
|
// check that x is "xx == nil" or "xx != nil"
|
||
|
x, ok := expr.X.(*ast.BinaryExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if eqNil && x.Op != token.EQL {
|
||
|
return
|
||
|
}
|
||
|
if !eqNil && x.Op != token.NEQ {
|
||
|
return
|
||
|
}
|
||
|
xx, ok := x.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !IsNil(pass, x.Y) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// check that y is "len(xx) == 0" or "len(xx) ... "
|
||
|
y, ok := expr.Y.(*ast.BinaryExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if eqNil && y.Op != token.EQL { // must be len(xx) *==* 0
|
||
|
return
|
||
|
}
|
||
|
yx, ok := y.X.(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
yxFun, ok := yx.Fun.(*ast.Ident)
|
||
|
if !ok || yxFun.Name != "len" || len(yx.Args) != 1 {
|
||
|
return
|
||
|
}
|
||
|
yxArg, ok := yx.Args[Arg("len.v")].(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if yxArg.Name != xx.Name {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if eqNil && !IsZero(y.Y) { // must be len(x) == *0*
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if !eqNil {
|
||
|
isConst, isZero := isConstZero(y.Y)
|
||
|
if !isConst {
|
||
|
return
|
||
|
}
|
||
|
switch y.Op {
|
||
|
case token.EQL:
|
||
|
// avoid false positive for "xx != nil && len(xx) == 0"
|
||
|
if isZero {
|
||
|
return
|
||
|
}
|
||
|
case token.GEQ:
|
||
|
// avoid false positive for "xx != nil && len(xx) >= 0"
|
||
|
if isZero {
|
||
|
return
|
||
|
}
|
||
|
case token.NEQ:
|
||
|
// avoid false positive for "xx != nil && len(xx) != <non-zero>"
|
||
|
if !isZero {
|
||
|
return
|
||
|
}
|
||
|
case token.GTR:
|
||
|
// ok
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// finally check that xx type is one of array, slice, map or chan
|
||
|
// this is to prevent false positive in case if xx is a pointer to an array
|
||
|
var nilType string
|
||
|
switch pass.TypesInfo.TypeOf(xx).(type) {
|
||
|
case *types.Slice:
|
||
|
nilType = "nil slices"
|
||
|
case *types.Map:
|
||
|
nilType = "nil maps"
|
||
|
case *types.Chan:
|
||
|
nilType = "nil channels"
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, expr, "should omit nil check; len() for %s is defined as zero", nilType)
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.BinaryExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintSlicing(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
n := node.(*ast.SliceExpr)
|
||
|
if n.Max != nil {
|
||
|
return
|
||
|
}
|
||
|
s, ok := n.X.(*ast.Ident)
|
||
|
if !ok || s.Obj == nil {
|
||
|
return
|
||
|
}
|
||
|
call, ok := n.High.(*ast.CallExpr)
|
||
|
if !ok || len(call.Args) != 1 || call.Ellipsis.IsValid() {
|
||
|
return
|
||
|
}
|
||
|
fun, ok := call.Fun.(*ast.Ident)
|
||
|
if !ok || fun.Name != "len" {
|
||
|
return
|
||
|
}
|
||
|
if _, ok := pass.TypesInfo.ObjectOf(fun).(*types.Builtin); !ok {
|
||
|
return
|
||
|
}
|
||
|
arg, ok := call.Args[Arg("len.v")].(*ast.Ident)
|
||
|
if !ok || arg.Obj != s.Obj {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, n, "should omit second index in slice, s[a:len(s)] is identical to s[a:]")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.SliceExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func refersTo(pass *analysis.Pass, expr ast.Expr, ident *ast.Ident) bool {
|
||
|
found := false
|
||
|
fn := func(node ast.Node) bool {
|
||
|
ident2, ok := node.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return true
|
||
|
}
|
||
|
if pass.TypesInfo.ObjectOf(ident) == pass.TypesInfo.ObjectOf(ident2) {
|
||
|
found = true
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
ast.Inspect(expr, fn)
|
||
|
return found
|
||
|
}
|
||
|
|
||
|
func LintLoopAppend(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
loop := node.(*ast.RangeStmt)
|
||
|
if !IsBlank(loop.Key) {
|
||
|
return
|
||
|
}
|
||
|
val, ok := loop.Value.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if len(loop.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
stmt, ok := loop.Body.List[0].(*ast.AssignStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if stmt.Tok != token.ASSIGN || len(stmt.Lhs) != 1 || len(stmt.Rhs) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if refersTo(pass, stmt.Lhs[0], val) {
|
||
|
return
|
||
|
}
|
||
|
call, ok := stmt.Rhs[0].(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if len(call.Args) != 2 || call.Ellipsis.IsValid() {
|
||
|
return
|
||
|
}
|
||
|
fun, ok := call.Fun.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
obj := pass.TypesInfo.ObjectOf(fun)
|
||
|
fn, ok := obj.(*types.Builtin)
|
||
|
if !ok || fn.Name() != "append" {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
src := pass.TypesInfo.TypeOf(loop.X)
|
||
|
dst := pass.TypesInfo.TypeOf(call.Args[Arg("append.slice")])
|
||
|
// TODO(dominikh) remove nil check once Go issue #15173 has
|
||
|
// been fixed
|
||
|
if src == nil {
|
||
|
return
|
||
|
}
|
||
|
if !types.Identical(src, dst) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if Render(pass, stmt.Lhs[0]) != Render(pass, call.Args[Arg("append.slice")]) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
el, ok := call.Args[Arg("append.elems")].(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if pass.TypesInfo.ObjectOf(val) != pass.TypesInfo.ObjectOf(el) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, loop, "should replace loop with %s = append(%s, %s...)",
|
||
|
Render(pass, stmt.Lhs[0]), Render(pass, call.Args[Arg("append.slice")]), Render(pass, loop.X))
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.RangeStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintTimeSince(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
sel, ok := call.Fun.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !IsCallToAST(pass, sel.X, "time.Now") {
|
||
|
return
|
||
|
}
|
||
|
if sel.Sel.Name != "Sub" {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, call, "should use time.Since instead of time.Now().Sub")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintTimeUntil(pass *analysis.Pass) (interface{}, error) {
|
||
|
if !IsGoVersion(pass, 8) {
|
||
|
return nil, nil
|
||
|
}
|
||
|
fn := func(node ast.Node) {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
if !IsCallToAST(pass, call, "(time.Time).Sub") {
|
||
|
return
|
||
|
}
|
||
|
if !IsCallToAST(pass, call.Args[Arg("(time.Time).Sub.u")], "time.Now") {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, call, "should use time.Until instead of t.Sub(time.Now())")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintUnnecessaryBlank(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn1 := func(node ast.Node) {
|
||
|
assign := node.(*ast.AssignStmt)
|
||
|
if len(assign.Lhs) != 2 || len(assign.Rhs) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if !IsBlank(assign.Lhs[1]) {
|
||
|
return
|
||
|
}
|
||
|
switch rhs := assign.Rhs[0].(type) {
|
||
|
case *ast.IndexExpr:
|
||
|
// The type-checker should make sure that it's a map, but
|
||
|
// let's be safe.
|
||
|
if _, ok := pass.TypesInfo.TypeOf(rhs.X).Underlying().(*types.Map); !ok {
|
||
|
return
|
||
|
}
|
||
|
case *ast.UnaryExpr:
|
||
|
if rhs.Op != token.ARROW {
|
||
|
return
|
||
|
}
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
cp := *assign
|
||
|
cp.Lhs = cp.Lhs[0:1]
|
||
|
ReportNodefFG(pass, assign, "should write %s instead of %s", Render(pass, &cp), Render(pass, assign))
|
||
|
}
|
||
|
|
||
|
fn2 := func(node ast.Node) {
|
||
|
stmt := node.(*ast.AssignStmt)
|
||
|
if len(stmt.Lhs) != len(stmt.Rhs) {
|
||
|
return
|
||
|
}
|
||
|
for i, lh := range stmt.Lhs {
|
||
|
rh := stmt.Rhs[i]
|
||
|
if !IsBlank(lh) {
|
||
|
continue
|
||
|
}
|
||
|
expr, ok := rh.(*ast.UnaryExpr)
|
||
|
if !ok {
|
||
|
continue
|
||
|
}
|
||
|
if expr.Op != token.ARROW {
|
||
|
continue
|
||
|
}
|
||
|
ReportNodefFG(pass, lh, "'_ = <-ch' can be simplified to '<-ch'")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fn3 := func(node ast.Node) {
|
||
|
rs := node.(*ast.RangeStmt)
|
||
|
|
||
|
// for x, _
|
||
|
if !IsBlank(rs.Key) && IsBlank(rs.Value) {
|
||
|
ReportNodefFG(pass, rs.Value, "should omit value from range; this loop is equivalent to `for %s %s range ...`", Render(pass, rs.Key), rs.Tok)
|
||
|
}
|
||
|
// for _, _ || for _
|
||
|
if IsBlank(rs.Key) && (IsBlank(rs.Value) || rs.Value == nil) {
|
||
|
ReportNodefFG(pass, rs.Key, "should omit values from range; this loop is equivalent to `for range ...`")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.AssignStmt)(nil)}, fn1)
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.AssignStmt)(nil)}, fn2)
|
||
|
if IsGoVersion(pass, 4) {
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.RangeStmt)(nil)}, fn3)
|
||
|
}
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintSimplerStructConversion(pass *analysis.Pass) (interface{}, error) {
|
||
|
var skip ast.Node
|
||
|
fn := func(node ast.Node) {
|
||
|
// Do not suggest type conversion between pointers
|
||
|
if unary, ok := node.(*ast.UnaryExpr); ok && unary.Op == token.AND {
|
||
|
if lit, ok := unary.X.(*ast.CompositeLit); ok {
|
||
|
skip = lit
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if node == skip {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
lit, ok := node.(*ast.CompositeLit)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
typ1, _ := pass.TypesInfo.TypeOf(lit.Type).(*types.Named)
|
||
|
if typ1 == nil {
|
||
|
return
|
||
|
}
|
||
|
s1, ok := typ1.Underlying().(*types.Struct)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
var typ2 *types.Named
|
||
|
var ident *ast.Ident
|
||
|
getSelType := func(expr ast.Expr) (types.Type, *ast.Ident, bool) {
|
||
|
sel, ok := expr.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return nil, nil, false
|
||
|
}
|
||
|
ident, ok := sel.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return nil, nil, false
|
||
|
}
|
||
|
typ := pass.TypesInfo.TypeOf(sel.X)
|
||
|
return typ, ident, typ != nil
|
||
|
}
|
||
|
if len(lit.Elts) == 0 {
|
||
|
return
|
||
|
}
|
||
|
if s1.NumFields() != len(lit.Elts) {
|
||
|
return
|
||
|
}
|
||
|
for i, elt := range lit.Elts {
|
||
|
var t types.Type
|
||
|
var id *ast.Ident
|
||
|
var ok bool
|
||
|
switch elt := elt.(type) {
|
||
|
case *ast.SelectorExpr:
|
||
|
t, id, ok = getSelType(elt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if i >= s1.NumFields() || s1.Field(i).Name() != elt.Sel.Name {
|
||
|
return
|
||
|
}
|
||
|
case *ast.KeyValueExpr:
|
||
|
var sel *ast.SelectorExpr
|
||
|
sel, ok = elt.Value.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if elt.Key.(*ast.Ident).Name != sel.Sel.Name {
|
||
|
return
|
||
|
}
|
||
|
t, id, ok = getSelType(elt.Value)
|
||
|
}
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
// All fields must be initialized from the same object
|
||
|
if ident != nil && ident.Obj != id.Obj {
|
||
|
return
|
||
|
}
|
||
|
typ2, _ = t.(*types.Named)
|
||
|
if typ2 == nil {
|
||
|
return
|
||
|
}
|
||
|
ident = id
|
||
|
}
|
||
|
|
||
|
if typ2 == nil {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if typ1.Obj().Pkg() != typ2.Obj().Pkg() {
|
||
|
// Do not suggest type conversions between different
|
||
|
// packages. Types in different packages might only match
|
||
|
// by coincidence. Furthermore, if the dependency ever
|
||
|
// adds more fields to its type, it could break the code
|
||
|
// that relies on the type conversion to work.
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s2, ok := typ2.Underlying().(*types.Struct)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if typ1 == typ2 {
|
||
|
return
|
||
|
}
|
||
|
if IsGoVersion(pass, 8) {
|
||
|
if !types.IdenticalIgnoreTags(s1, s2) {
|
||
|
return
|
||
|
}
|
||
|
} else {
|
||
|
if !types.Identical(s1, s2) {
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
ReportNodefFG(pass, node, "should convert %s (type %s) to %s instead of using struct literal",
|
||
|
ident.Name, typ2.Obj().Name(), typ1.Obj().Name())
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.UnaryExpr)(nil), (*ast.CompositeLit)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintTrim(pass *analysis.Pass) (interface{}, error) {
|
||
|
sameNonDynamic := func(node1, node2 ast.Node) bool {
|
||
|
if reflect.TypeOf(node1) != reflect.TypeOf(node2) {
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
switch node1 := node1.(type) {
|
||
|
case *ast.Ident:
|
||
|
return node1.Obj == node2.(*ast.Ident).Obj
|
||
|
case *ast.SelectorExpr:
|
||
|
return Render(pass, node1) == Render(pass, node2)
|
||
|
case *ast.IndexExpr:
|
||
|
return Render(pass, node1) == Render(pass, node2)
|
||
|
}
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
isLenOnIdent := func(fn ast.Expr, ident ast.Expr) bool {
|
||
|
call, ok := fn.(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return false
|
||
|
}
|
||
|
if fn, ok := call.Fun.(*ast.Ident); !ok || fn.Name != "len" {
|
||
|
return false
|
||
|
}
|
||
|
if len(call.Args) != 1 {
|
||
|
return false
|
||
|
}
|
||
|
return sameNonDynamic(call.Args[Arg("len.v")], ident)
|
||
|
}
|
||
|
|
||
|
fn := func(node ast.Node) {
|
||
|
var pkg string
|
||
|
var fun string
|
||
|
|
||
|
ifstmt := node.(*ast.IfStmt)
|
||
|
if ifstmt.Init != nil {
|
||
|
return
|
||
|
}
|
||
|
if ifstmt.Else != nil {
|
||
|
return
|
||
|
}
|
||
|
if len(ifstmt.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
condCall, ok := ifstmt.Cond.(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
switch {
|
||
|
case IsCallToAST(pass, condCall, "strings.HasPrefix"):
|
||
|
pkg = "strings"
|
||
|
fun = "HasPrefix"
|
||
|
case IsCallToAST(pass, condCall, "strings.HasSuffix"):
|
||
|
pkg = "strings"
|
||
|
fun = "HasSuffix"
|
||
|
case IsCallToAST(pass, condCall, "strings.Contains"):
|
||
|
pkg = "strings"
|
||
|
fun = "Contains"
|
||
|
case IsCallToAST(pass, condCall, "bytes.HasPrefix"):
|
||
|
pkg = "bytes"
|
||
|
fun = "HasPrefix"
|
||
|
case IsCallToAST(pass, condCall, "bytes.HasSuffix"):
|
||
|
pkg = "bytes"
|
||
|
fun = "HasSuffix"
|
||
|
case IsCallToAST(pass, condCall, "bytes.Contains"):
|
||
|
pkg = "bytes"
|
||
|
fun = "Contains"
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
|
||
|
assign, ok := ifstmt.Body.List[0].(*ast.AssignStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if assign.Tok != token.ASSIGN {
|
||
|
return
|
||
|
}
|
||
|
if len(assign.Lhs) != 1 || len(assign.Rhs) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if !sameNonDynamic(condCall.Args[0], assign.Lhs[0]) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
switch rhs := assign.Rhs[0].(type) {
|
||
|
case *ast.CallExpr:
|
||
|
if len(rhs.Args) < 2 || !sameNonDynamic(condCall.Args[0], rhs.Args[0]) || !sameNonDynamic(condCall.Args[1], rhs.Args[1]) {
|
||
|
return
|
||
|
}
|
||
|
if IsCallToAST(pass, condCall, "strings.HasPrefix") && IsCallToAST(pass, rhs, "strings.TrimPrefix") ||
|
||
|
IsCallToAST(pass, condCall, "strings.HasSuffix") && IsCallToAST(pass, rhs, "strings.TrimSuffix") ||
|
||
|
IsCallToAST(pass, condCall, "strings.Contains") && IsCallToAST(pass, rhs, "strings.Replace") ||
|
||
|
IsCallToAST(pass, condCall, "bytes.HasPrefix") && IsCallToAST(pass, rhs, "bytes.TrimPrefix") ||
|
||
|
IsCallToAST(pass, condCall, "bytes.HasSuffix") && IsCallToAST(pass, rhs, "bytes.TrimSuffix") ||
|
||
|
IsCallToAST(pass, condCall, "bytes.Contains") && IsCallToAST(pass, rhs, "bytes.Replace") {
|
||
|
ReportNodefFG(pass, ifstmt, "should replace this if statement with an unconditional %s", CallNameAST(pass, rhs))
|
||
|
}
|
||
|
return
|
||
|
case *ast.SliceExpr:
|
||
|
slice := rhs
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if slice.Slice3 {
|
||
|
return
|
||
|
}
|
||
|
if !sameNonDynamic(slice.X, condCall.Args[0]) {
|
||
|
return
|
||
|
}
|
||
|
var index ast.Expr
|
||
|
switch fun {
|
||
|
case "HasPrefix":
|
||
|
// TODO(dh) We could detect a High that is len(s), but another
|
||
|
// rule will already flag that, anyway.
|
||
|
if slice.High != nil {
|
||
|
return
|
||
|
}
|
||
|
index = slice.Low
|
||
|
case "HasSuffix":
|
||
|
if slice.Low != nil {
|
||
|
n, ok := ExprToInt(pass, slice.Low)
|
||
|
if !ok || n != 0 {
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
index = slice.High
|
||
|
}
|
||
|
|
||
|
switch index := index.(type) {
|
||
|
case *ast.CallExpr:
|
||
|
if fun != "HasPrefix" {
|
||
|
return
|
||
|
}
|
||
|
if fn, ok := index.Fun.(*ast.Ident); !ok || fn.Name != "len" {
|
||
|
return
|
||
|
}
|
||
|
if len(index.Args) != 1 {
|
||
|
return
|
||
|
}
|
||
|
id3 := index.Args[Arg("len.v")]
|
||
|
switch oid3 := condCall.Args[1].(type) {
|
||
|
case *ast.BasicLit:
|
||
|
if pkg != "strings" {
|
||
|
return
|
||
|
}
|
||
|
lit, ok := id3.(*ast.BasicLit)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
s1, ok1 := ExprToString(pass, lit)
|
||
|
s2, ok2 := ExprToString(pass, condCall.Args[1])
|
||
|
if !ok1 || !ok2 || s1 != s2 {
|
||
|
return
|
||
|
}
|
||
|
default:
|
||
|
if !sameNonDynamic(id3, oid3) {
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
case *ast.BasicLit, *ast.Ident:
|
||
|
if fun != "HasPrefix" {
|
||
|
return
|
||
|
}
|
||
|
if pkg != "strings" {
|
||
|
return
|
||
|
}
|
||
|
string, ok1 := ExprToString(pass, condCall.Args[1])
|
||
|
int, ok2 := ExprToInt(pass, slice.Low)
|
||
|
if !ok1 || !ok2 || int != int64(len(string)) {
|
||
|
return
|
||
|
}
|
||
|
case *ast.BinaryExpr:
|
||
|
if fun != "HasSuffix" {
|
||
|
return
|
||
|
}
|
||
|
if index.Op != token.SUB {
|
||
|
return
|
||
|
}
|
||
|
if !isLenOnIdent(index.X, condCall.Args[0]) ||
|
||
|
!isLenOnIdent(index.Y, condCall.Args[1]) {
|
||
|
return
|
||
|
}
|
||
|
default:
|
||
|
return
|
||
|
}
|
||
|
|
||
|
var replacement string
|
||
|
switch fun {
|
||
|
case "HasPrefix":
|
||
|
replacement = "TrimPrefix"
|
||
|
case "HasSuffix":
|
||
|
replacement = "TrimSuffix"
|
||
|
}
|
||
|
ReportNodefFG(pass, ifstmt, "should replace this if statement with an unconditional %s.%s", pkg, replacement)
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.IfStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintLoopSlide(pass *analysis.Pass) (interface{}, error) {
|
||
|
// TODO(dh): detect bs[i+offset] in addition to bs[offset+i]
|
||
|
// TODO(dh): consider merging this function with LintLoopCopy
|
||
|
// TODO(dh): detect length that is an expression, not a variable name
|
||
|
// TODO(dh): support sliding to a different offset than the beginning of the slice
|
||
|
|
||
|
fn := func(node ast.Node) {
|
||
|
/*
|
||
|
for i := 0; i < n; i++ {
|
||
|
bs[i] = bs[offset+i]
|
||
|
}
|
||
|
|
||
|
↓
|
||
|
|
||
|
copy(bs[:n], bs[offset:offset+n])
|
||
|
*/
|
||
|
|
||
|
loop := node.(*ast.ForStmt)
|
||
|
if len(loop.Body.List) != 1 || loop.Init == nil || loop.Cond == nil || loop.Post == nil {
|
||
|
return
|
||
|
}
|
||
|
assign, ok := loop.Init.(*ast.AssignStmt)
|
||
|
if !ok || len(assign.Lhs) != 1 || len(assign.Rhs) != 1 || !IsZero(assign.Rhs[0]) {
|
||
|
return
|
||
|
}
|
||
|
initvar, ok := assign.Lhs[0].(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
post, ok := loop.Post.(*ast.IncDecStmt)
|
||
|
if !ok || post.Tok != token.INC {
|
||
|
return
|
||
|
}
|
||
|
postvar, ok := post.X.(*ast.Ident)
|
||
|
if !ok || pass.TypesInfo.ObjectOf(postvar) != pass.TypesInfo.ObjectOf(initvar) {
|
||
|
return
|
||
|
}
|
||
|
bin, ok := loop.Cond.(*ast.BinaryExpr)
|
||
|
if !ok || bin.Op != token.LSS {
|
||
|
return
|
||
|
}
|
||
|
binx, ok := bin.X.(*ast.Ident)
|
||
|
if !ok || pass.TypesInfo.ObjectOf(binx) != pass.TypesInfo.ObjectOf(initvar) {
|
||
|
return
|
||
|
}
|
||
|
biny, ok := bin.Y.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
assign, ok = loop.Body.List[0].(*ast.AssignStmt)
|
||
|
if !ok || len(assign.Lhs) != 1 || len(assign.Rhs) != 1 || assign.Tok != token.ASSIGN {
|
||
|
return
|
||
|
}
|
||
|
lhs, ok := assign.Lhs[0].(*ast.IndexExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
rhs, ok := assign.Rhs[0].(*ast.IndexExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
bs1, ok := lhs.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
bs2, ok := rhs.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
obj1 := pass.TypesInfo.ObjectOf(bs1)
|
||
|
obj2 := pass.TypesInfo.ObjectOf(bs2)
|
||
|
if obj1 != obj2 {
|
||
|
return
|
||
|
}
|
||
|
if _, ok := obj1.Type().Underlying().(*types.Slice); !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
index1, ok := lhs.Index.(*ast.Ident)
|
||
|
if !ok || pass.TypesInfo.ObjectOf(index1) != pass.TypesInfo.ObjectOf(initvar) {
|
||
|
return
|
||
|
}
|
||
|
index2, ok := rhs.Index.(*ast.BinaryExpr)
|
||
|
if !ok || index2.Op != token.ADD {
|
||
|
return
|
||
|
}
|
||
|
add1, ok := index2.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
add2, ok := index2.Y.(*ast.Ident)
|
||
|
if !ok || pass.TypesInfo.ObjectOf(add2) != pass.TypesInfo.ObjectOf(initvar) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ReportNodefFG(pass, loop, "should use copy(%s[:%s], %s[%s:]) instead", Render(pass, bs1), Render(pass, biny), Render(pass, bs1), Render(pass, add1))
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.ForStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintMakeLenCap(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
if fn, ok := call.Fun.(*ast.Ident); !ok || fn.Name != "make" {
|
||
|
// FIXME check whether make is indeed the built-in function
|
||
|
return
|
||
|
}
|
||
|
switch len(call.Args) {
|
||
|
case 2:
|
||
|
// make(T, len)
|
||
|
if _, ok := pass.TypesInfo.TypeOf(call.Args[Arg("make.t")]).Underlying().(*types.Slice); ok {
|
||
|
break
|
||
|
}
|
||
|
if IsZero(call.Args[Arg("make.size[0]")]) {
|
||
|
ReportNodefFG(pass, call.Args[Arg("make.size[0]")], "should use make(%s) instead", Render(pass, call.Args[Arg("make.t")]))
|
||
|
}
|
||
|
case 3:
|
||
|
// make(T, len, cap)
|
||
|
if Render(pass, call.Args[Arg("make.size[0]")]) == Render(pass, call.Args[Arg("make.size[1]")]) {
|
||
|
ReportNodefFG(pass, call.Args[Arg("make.size[0]")],
|
||
|
"should use make(%s, %s) instead",
|
||
|
Render(pass, call.Args[Arg("make.t")]), Render(pass, call.Args[Arg("make.size[0]")]))
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintAssertNotNil(pass *analysis.Pass) (interface{}, error) {
|
||
|
isNilCheck := func(ident *ast.Ident, expr ast.Expr) bool {
|
||
|
xbinop, ok := expr.(*ast.BinaryExpr)
|
||
|
if !ok || xbinop.Op != token.NEQ {
|
||
|
return false
|
||
|
}
|
||
|
xident, ok := xbinop.X.(*ast.Ident)
|
||
|
if !ok || xident.Obj != ident.Obj {
|
||
|
return false
|
||
|
}
|
||
|
if !IsNil(pass, xbinop.Y) {
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
isOKCheck := func(ident *ast.Ident, expr ast.Expr) bool {
|
||
|
yident, ok := expr.(*ast.Ident)
|
||
|
if !ok || yident.Obj != ident.Obj {
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
fn1 := func(node ast.Node) {
|
||
|
ifstmt := node.(*ast.IfStmt)
|
||
|
assign, ok := ifstmt.Init.(*ast.AssignStmt)
|
||
|
if !ok || len(assign.Lhs) != 2 || len(assign.Rhs) != 1 || !IsBlank(assign.Lhs[0]) {
|
||
|
return
|
||
|
}
|
||
|
assert, ok := assign.Rhs[0].(*ast.TypeAssertExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
binop, ok := ifstmt.Cond.(*ast.BinaryExpr)
|
||
|
if !ok || binop.Op != token.LAND {
|
||
|
return
|
||
|
}
|
||
|
assertIdent, ok := assert.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
assignIdent, ok := assign.Lhs[1].(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !(isNilCheck(assertIdent, binop.X) && isOKCheck(assignIdent, binop.Y)) &&
|
||
|
!(isNilCheck(assertIdent, binop.Y) && isOKCheck(assignIdent, binop.X)) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, ifstmt, "when %s is true, %s can't be nil", Render(pass, assignIdent), Render(pass, assertIdent))
|
||
|
}
|
||
|
fn2 := func(node ast.Node) {
|
||
|
// Check that outer ifstmt is an 'if x != nil {}'
|
||
|
ifstmt := node.(*ast.IfStmt)
|
||
|
if ifstmt.Init != nil {
|
||
|
return
|
||
|
}
|
||
|
if ifstmt.Else != nil {
|
||
|
return
|
||
|
}
|
||
|
if len(ifstmt.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
binop, ok := ifstmt.Cond.(*ast.BinaryExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if binop.Op != token.NEQ {
|
||
|
return
|
||
|
}
|
||
|
lhs, ok := binop.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !IsNil(pass, binop.Y) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// Check that inner ifstmt is an `if _, ok := x.(T); ok {}`
|
||
|
ifstmt, ok = ifstmt.Body.List[0].(*ast.IfStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
assign, ok := ifstmt.Init.(*ast.AssignStmt)
|
||
|
if !ok || len(assign.Lhs) != 2 || len(assign.Rhs) != 1 || !IsBlank(assign.Lhs[0]) {
|
||
|
return
|
||
|
}
|
||
|
assert, ok := assign.Rhs[0].(*ast.TypeAssertExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
assertIdent, ok := assert.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if lhs.Obj != assertIdent.Obj {
|
||
|
return
|
||
|
}
|
||
|
assignIdent, ok := assign.Lhs[1].(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !isOKCheck(assignIdent, ifstmt.Cond) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, ifstmt, "when %s is true, %s can't be nil", Render(pass, assignIdent), Render(pass, assertIdent))
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.IfStmt)(nil)}, fn1)
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.IfStmt)(nil)}, fn2)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintDeclareAssign(pass *analysis.Pass) (interface{}, error) {
|
||
|
hasMultipleAssignments := func(root ast.Node, ident *ast.Ident) bool {
|
||
|
num := 0
|
||
|
ast.Inspect(root, func(node ast.Node) bool {
|
||
|
if num >= 2 {
|
||
|
return false
|
||
|
}
|
||
|
assign, ok := node.(*ast.AssignStmt)
|
||
|
if !ok {
|
||
|
return true
|
||
|
}
|
||
|
for _, lhs := range assign.Lhs {
|
||
|
if oident, ok := lhs.(*ast.Ident); ok {
|
||
|
if oident.Obj == ident.Obj {
|
||
|
num++
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return true
|
||
|
})
|
||
|
return num >= 2
|
||
|
}
|
||
|
fn := func(node ast.Node) {
|
||
|
block := node.(*ast.BlockStmt)
|
||
|
if len(block.List) < 2 {
|
||
|
return
|
||
|
}
|
||
|
for i, stmt := range block.List[:len(block.List)-1] {
|
||
|
_ = i
|
||
|
decl, ok := stmt.(*ast.DeclStmt)
|
||
|
if !ok {
|
||
|
continue
|
||
|
}
|
||
|
gdecl, ok := decl.Decl.(*ast.GenDecl)
|
||
|
if !ok || gdecl.Tok != token.VAR || len(gdecl.Specs) != 1 {
|
||
|
continue
|
||
|
}
|
||
|
vspec, ok := gdecl.Specs[0].(*ast.ValueSpec)
|
||
|
if !ok || len(vspec.Names) != 1 || len(vspec.Values) != 0 {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
assign, ok := block.List[i+1].(*ast.AssignStmt)
|
||
|
if !ok || assign.Tok != token.ASSIGN {
|
||
|
continue
|
||
|
}
|
||
|
if len(assign.Lhs) != 1 || len(assign.Rhs) != 1 {
|
||
|
continue
|
||
|
}
|
||
|
ident, ok := assign.Lhs[0].(*ast.Ident)
|
||
|
if !ok {
|
||
|
continue
|
||
|
}
|
||
|
if vspec.Names[0].Obj != ident.Obj {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if refersTo(pass, assign.Rhs[0], ident) {
|
||
|
continue
|
||
|
}
|
||
|
if hasMultipleAssignments(block, ident) {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
ReportNodefFG(pass, decl, "should merge variable declaration with assignment on next line")
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.BlockStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintRedundantBreak(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn1 := func(node ast.Node) {
|
||
|
clause := node.(*ast.CaseClause)
|
||
|
if len(clause.Body) < 2 {
|
||
|
return
|
||
|
}
|
||
|
branch, ok := clause.Body[len(clause.Body)-1].(*ast.BranchStmt)
|
||
|
if !ok || branch.Tok != token.BREAK || branch.Label != nil {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, branch, "redundant break statement")
|
||
|
}
|
||
|
fn2 := func(node ast.Node) {
|
||
|
var ret *ast.FieldList
|
||
|
var body *ast.BlockStmt
|
||
|
switch x := node.(type) {
|
||
|
case *ast.FuncDecl:
|
||
|
ret = x.Type.Results
|
||
|
body = x.Body
|
||
|
case *ast.FuncLit:
|
||
|
ret = x.Type.Results
|
||
|
body = x.Body
|
||
|
default:
|
||
|
panic(fmt.Sprintf("unreachable: %T", node))
|
||
|
}
|
||
|
// if the func has results, a return can't be redundant.
|
||
|
// similarly, if there are no statements, there can be
|
||
|
// no return.
|
||
|
if ret != nil || body == nil || len(body.List) < 1 {
|
||
|
return
|
||
|
}
|
||
|
rst, ok := body.List[len(body.List)-1].(*ast.ReturnStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
// we don't need to check rst.Results as we already
|
||
|
// checked x.Type.Results to be nil.
|
||
|
ReportNodefFG(pass, rst, "redundant return statement")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CaseClause)(nil)}, fn1)
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.FuncDecl)(nil), (*ast.FuncLit)(nil)}, fn2)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func isStringer(T types.Type, msCache *typeutil.MethodSetCache) bool {
|
||
|
ms := msCache.MethodSet(T)
|
||
|
sel := ms.Lookup(nil, "String")
|
||
|
if sel == nil {
|
||
|
return false
|
||
|
}
|
||
|
fn, ok := sel.Obj().(*types.Func)
|
||
|
if !ok {
|
||
|
// should be unreachable
|
||
|
return false
|
||
|
}
|
||
|
sig := fn.Type().(*types.Signature)
|
||
|
if sig.Params().Len() != 0 {
|
||
|
return false
|
||
|
}
|
||
|
if sig.Results().Len() != 1 {
|
||
|
return false
|
||
|
}
|
||
|
if !IsType(sig.Results().At(0).Type(), "string") {
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
func LintRedundantSprintf(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
if !IsCallToAST(pass, call, "fmt.Sprintf") {
|
||
|
return
|
||
|
}
|
||
|
if len(call.Args) != 2 {
|
||
|
return
|
||
|
}
|
||
|
if s, ok := ExprToString(pass, call.Args[Arg("fmt.Sprintf.format")]); !ok || s != "%s" {
|
||
|
return
|
||
|
}
|
||
|
arg := call.Args[Arg("fmt.Sprintf.a[0]")]
|
||
|
typ := pass.TypesInfo.TypeOf(arg)
|
||
|
|
||
|
ssapkg := pass.ResultOf[buildssa.Analyzer].(*buildssa.SSA).Pkg
|
||
|
if isStringer(typ, &ssapkg.Prog.MethodSets) {
|
||
|
ReportNodef(pass, call, "should use String() instead of fmt.Sprintf")
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if typ.Underlying() == types.Universe.Lookup("string").Type() {
|
||
|
if typ == types.Universe.Lookup("string").Type() {
|
||
|
ReportNodefFG(pass, call, "the argument is already a string, there's no need to use fmt.Sprintf")
|
||
|
} else {
|
||
|
ReportNodefFG(pass, call, "the argument's underlying type is a string, should use a simple conversion instead of fmt.Sprintf")
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintErrorsNewSprintf(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
if !IsCallToAST(pass, node, "errors.New") {
|
||
|
return
|
||
|
}
|
||
|
call := node.(*ast.CallExpr)
|
||
|
if !IsCallToAST(pass, call.Args[Arg("errors.New.text")], "fmt.Sprintf") {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, node, "should use fmt.Errorf(...) instead of errors.New(fmt.Sprintf(...))")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.CallExpr)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintRangeStringRunes(pass *analysis.Pass) (interface{}, error) {
|
||
|
return sharedcheck.CheckRangeStringRunes(pass)
|
||
|
}
|
||
|
|
||
|
func LintNilCheckAroundRange(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
ifstmt := node.(*ast.IfStmt)
|
||
|
cond, ok := ifstmt.Cond.(*ast.BinaryExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if cond.Op != token.NEQ || !IsNil(pass, cond.Y) || len(ifstmt.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
loop, ok := ifstmt.Body.List[0].(*ast.RangeStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
ifXIdent, ok := cond.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
rangeXIdent, ok := loop.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if ifXIdent.Obj != rangeXIdent.Obj {
|
||
|
return
|
||
|
}
|
||
|
switch pass.TypesInfo.TypeOf(rangeXIdent).(type) {
|
||
|
case *types.Slice, *types.Map:
|
||
|
ReportNodefFG(pass, node, "unnecessary nil check around range")
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.IfStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func isPermissibleSort(pass *analysis.Pass, node ast.Node) bool {
|
||
|
call := node.(*ast.CallExpr)
|
||
|
typeconv, ok := call.Args[0].(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
sel, ok := typeconv.Fun.(*ast.SelectorExpr)
|
||
|
if !ok {
|
||
|
return true
|
||
|
}
|
||
|
name := SelectorName(pass, sel)
|
||
|
switch name {
|
||
|
case "sort.IntSlice", "sort.Float64Slice", "sort.StringSlice":
|
||
|
default:
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
func LintSortHelpers(pass *analysis.Pass) (interface{}, error) {
|
||
|
type Error struct {
|
||
|
node ast.Node
|
||
|
msg string
|
||
|
}
|
||
|
var allErrors []Error
|
||
|
fn := func(node ast.Node) {
|
||
|
var body *ast.BlockStmt
|
||
|
switch node := node.(type) {
|
||
|
case *ast.FuncLit:
|
||
|
body = node.Body
|
||
|
case *ast.FuncDecl:
|
||
|
body = node.Body
|
||
|
default:
|
||
|
panic(fmt.Sprintf("unreachable: %T", node))
|
||
|
}
|
||
|
if body == nil {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
var errors []Error
|
||
|
permissible := false
|
||
|
fnSorts := func(node ast.Node) bool {
|
||
|
if permissible {
|
||
|
return false
|
||
|
}
|
||
|
if !IsCallToAST(pass, node, "sort.Sort") {
|
||
|
return true
|
||
|
}
|
||
|
if isPermissibleSort(pass, node) {
|
||
|
permissible = true
|
||
|
return false
|
||
|
}
|
||
|
call := node.(*ast.CallExpr)
|
||
|
typeconv := call.Args[Arg("sort.Sort.data")].(*ast.CallExpr)
|
||
|
sel := typeconv.Fun.(*ast.SelectorExpr)
|
||
|
name := SelectorName(pass, sel)
|
||
|
|
||
|
switch name {
|
||
|
case "sort.IntSlice":
|
||
|
errors = append(errors, Error{node, "should use sort.Ints(...) instead of sort.Sort(sort.IntSlice(...))"})
|
||
|
case "sort.Float64Slice":
|
||
|
errors = append(errors, Error{node, "should use sort.Float64s(...) instead of sort.Sort(sort.Float64Slice(...))"})
|
||
|
case "sort.StringSlice":
|
||
|
errors = append(errors, Error{node, "should use sort.Strings(...) instead of sort.Sort(sort.StringSlice(...))"})
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
ast.Inspect(body, fnSorts)
|
||
|
|
||
|
if permissible {
|
||
|
return
|
||
|
}
|
||
|
allErrors = append(allErrors, errors...)
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.FuncLit)(nil), (*ast.FuncDecl)(nil)}, fn)
|
||
|
sort.Slice(allErrors, func(i, j int) bool {
|
||
|
return allErrors[i].node.Pos() < allErrors[j].node.Pos()
|
||
|
})
|
||
|
var prev token.Pos
|
||
|
for _, err := range allErrors {
|
||
|
if err.node.Pos() == prev {
|
||
|
continue
|
||
|
}
|
||
|
prev = err.node.Pos()
|
||
|
ReportNodefFG(pass, err.node, "%s", err.msg)
|
||
|
}
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintGuardedDelete(pass *analysis.Pass) (interface{}, error) {
|
||
|
isCommaOkMapIndex := func(stmt ast.Stmt) (b *ast.Ident, m ast.Expr, key ast.Expr, ok bool) {
|
||
|
// Has to be of the form `_, <b:*ast.Ident> = <m:*types.Map>[<key>]
|
||
|
|
||
|
assign, ok := stmt.(*ast.AssignStmt)
|
||
|
if !ok {
|
||
|
return nil, nil, nil, false
|
||
|
}
|
||
|
if len(assign.Lhs) != 2 || len(assign.Rhs) != 1 {
|
||
|
return nil, nil, nil, false
|
||
|
}
|
||
|
if !IsBlank(assign.Lhs[0]) {
|
||
|
return nil, nil, nil, false
|
||
|
}
|
||
|
ident, ok := assign.Lhs[1].(*ast.Ident)
|
||
|
if !ok {
|
||
|
return nil, nil, nil, false
|
||
|
}
|
||
|
index, ok := assign.Rhs[0].(*ast.IndexExpr)
|
||
|
if !ok {
|
||
|
return nil, nil, nil, false
|
||
|
}
|
||
|
if _, ok := pass.TypesInfo.TypeOf(index.X).(*types.Map); !ok {
|
||
|
return nil, nil, nil, false
|
||
|
}
|
||
|
key = index.Index
|
||
|
return ident, index.X, key, true
|
||
|
}
|
||
|
fn := func(node ast.Node) {
|
||
|
stmt := node.(*ast.IfStmt)
|
||
|
if len(stmt.Body.List) != 1 {
|
||
|
return
|
||
|
}
|
||
|
if stmt.Else != nil {
|
||
|
return
|
||
|
}
|
||
|
expr, ok := stmt.Body.List[0].(*ast.ExprStmt)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
call, ok := expr.X.(*ast.CallExpr)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if !IsCallToAST(pass, call, "delete") {
|
||
|
return
|
||
|
}
|
||
|
b, m, key, ok := isCommaOkMapIndex(stmt.Init)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
if cond, ok := stmt.Cond.(*ast.Ident); !ok || pass.TypesInfo.ObjectOf(cond) != pass.TypesInfo.ObjectOf(b) {
|
||
|
return
|
||
|
}
|
||
|
if Render(pass, call.Args[0]) != Render(pass, m) || Render(pass, call.Args[1]) != Render(pass, key) {
|
||
|
return
|
||
|
}
|
||
|
ReportNodefFG(pass, stmt, "unnecessary guard around call to delete")
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.IfStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func LintSimplifyTypeSwitch(pass *analysis.Pass) (interface{}, error) {
|
||
|
fn := func(node ast.Node) {
|
||
|
stmt := node.(*ast.TypeSwitchStmt)
|
||
|
if stmt.Init != nil {
|
||
|
// bailing out for now, can't anticipate how type switches with initializers are being used
|
||
|
return
|
||
|
}
|
||
|
expr, ok := stmt.Assign.(*ast.ExprStmt)
|
||
|
if !ok {
|
||
|
// the user is in fact assigning the result
|
||
|
return
|
||
|
}
|
||
|
assert := expr.X.(*ast.TypeAssertExpr)
|
||
|
ident, ok := assert.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
x := pass.TypesInfo.ObjectOf(ident)
|
||
|
var allOffenders []ast.Node
|
||
|
for _, clause := range stmt.Body.List {
|
||
|
clause := clause.(*ast.CaseClause)
|
||
|
if len(clause.List) != 1 {
|
||
|
continue
|
||
|
}
|
||
|
hasUnrelatedAssertion := false
|
||
|
var offenders []ast.Node
|
||
|
ast.Inspect(clause, func(node ast.Node) bool {
|
||
|
assert2, ok := node.(*ast.TypeAssertExpr)
|
||
|
if !ok {
|
||
|
return true
|
||
|
}
|
||
|
ident, ok := assert2.X.(*ast.Ident)
|
||
|
if !ok {
|
||
|
hasUnrelatedAssertion = true
|
||
|
return false
|
||
|
}
|
||
|
if pass.TypesInfo.ObjectOf(ident) != x {
|
||
|
hasUnrelatedAssertion = true
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
if !types.Identical(pass.TypesInfo.TypeOf(clause.List[0]), pass.TypesInfo.TypeOf(assert2.Type)) {
|
||
|
hasUnrelatedAssertion = true
|
||
|
return false
|
||
|
}
|
||
|
offenders = append(offenders, assert2)
|
||
|
return true
|
||
|
})
|
||
|
if !hasUnrelatedAssertion {
|
||
|
// don't flag cases that have other type assertions
|
||
|
// unrelated to the one in the case clause. often
|
||
|
// times, this is done for symmetry, when two
|
||
|
// different values have to be asserted to the same
|
||
|
// type.
|
||
|
allOffenders = append(allOffenders, offenders...)
|
||
|
}
|
||
|
}
|
||
|
if len(allOffenders) != 0 {
|
||
|
at := ""
|
||
|
for _, offender := range allOffenders {
|
||
|
pos := lint.DisplayPosition(pass.Fset, offender.Pos())
|
||
|
at += "\n\t" + pos.String()
|
||
|
}
|
||
|
ReportNodefFG(pass, expr, "assigning the result of this type assertion to a variable (switch %s := %s.(type)) could eliminate the following type assertions:%s", Render(pass, ident), Render(pass, ident), at)
|
||
|
}
|
||
|
}
|
||
|
pass.ResultOf[inspect.Analyzer].(*inspector.Inspector).Preorder([]ast.Node{(*ast.TypeSwitchStmt)(nil)}, fn)
|
||
|
return nil, nil
|
||
|
}
|