Add generated file
This PR adds generated files under pkg/client and vendor folder.
This commit is contained in:
95
vendor/golang.org/x/tools/go/ssa/ssautil/load.go
generated
vendored
Normal file
95
vendor/golang.org/x/tools/go/ssa/ssautil/load.go
generated
vendored
Normal file
@@ -0,0 +1,95 @@
|
||||
// Copyright 2015 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 ssautil
|
||||
|
||||
// This file defines utility functions for constructing programs in SSA form.
|
||||
|
||||
import (
|
||||
"go/ast"
|
||||
"go/token"
|
||||
"go/types"
|
||||
|
||||
"golang.org/x/tools/go/loader"
|
||||
"golang.org/x/tools/go/ssa"
|
||||
)
|
||||
|
||||
// CreateProgram returns a new program in SSA form, given a program
|
||||
// loaded from source. An SSA package is created for each transitively
|
||||
// error-free package of lprog.
|
||||
//
|
||||
// Code for bodies of functions is not built until Build is called
|
||||
// on the result.
|
||||
//
|
||||
// mode controls diagnostics and checking during SSA construction.
|
||||
//
|
||||
func CreateProgram(lprog *loader.Program, mode ssa.BuilderMode) *ssa.Program {
|
||||
prog := ssa.NewProgram(lprog.Fset, mode)
|
||||
|
||||
for _, info := range lprog.AllPackages {
|
||||
if info.TransitivelyErrorFree {
|
||||
prog.CreatePackage(info.Pkg, info.Files, &info.Info, info.Importable)
|
||||
}
|
||||
}
|
||||
|
||||
return prog
|
||||
}
|
||||
|
||||
// BuildPackage builds an SSA program with IR for a single package.
|
||||
//
|
||||
// It populates pkg by type-checking the specified file ASTs. All
|
||||
// dependencies are loaded using the importer specified by tc, which
|
||||
// typically loads compiler export data; SSA code cannot be built for
|
||||
// those packages. BuildPackage then constructs an ssa.Program with all
|
||||
// dependency packages created, and builds and returns the SSA package
|
||||
// corresponding to pkg.
|
||||
//
|
||||
// The caller must have set pkg.Path() to the import path.
|
||||
//
|
||||
// The operation fails if there were any type-checking or import errors.
|
||||
//
|
||||
// See ../ssa/example_test.go for an example.
|
||||
//
|
||||
func BuildPackage(tc *types.Config, fset *token.FileSet, pkg *types.Package, files []*ast.File, mode ssa.BuilderMode) (*ssa.Package, *types.Info, error) {
|
||||
if fset == nil {
|
||||
panic("no token.FileSet")
|
||||
}
|
||||
if pkg.Path() == "" {
|
||||
panic("package has no import path")
|
||||
}
|
||||
|
||||
info := &types.Info{
|
||||
Types: make(map[ast.Expr]types.TypeAndValue),
|
||||
Defs: make(map[*ast.Ident]types.Object),
|
||||
Uses: make(map[*ast.Ident]types.Object),
|
||||
Implicits: make(map[ast.Node]types.Object),
|
||||
Scopes: make(map[ast.Node]*types.Scope),
|
||||
Selections: make(map[*ast.SelectorExpr]*types.Selection),
|
||||
}
|
||||
if err := types.NewChecker(tc, fset, pkg, info).Files(files); err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
prog := ssa.NewProgram(fset, mode)
|
||||
|
||||
// Create SSA packages for all imports.
|
||||
// Order is not significant.
|
||||
created := make(map[*types.Package]bool)
|
||||
var createAll func(pkgs []*types.Package)
|
||||
createAll = func(pkgs []*types.Package) {
|
||||
for _, p := range pkgs {
|
||||
if !created[p] {
|
||||
created[p] = true
|
||||
prog.CreatePackage(p, nil, nil, true)
|
||||
createAll(p.Imports())
|
||||
}
|
||||
}
|
||||
}
|
||||
createAll(pkg.Imports())
|
||||
|
||||
// Create and build the primary package.
|
||||
ssapkg := prog.CreatePackage(pkg, files, info, false)
|
||||
ssapkg.Build()
|
||||
return ssapkg, info, nil
|
||||
}
|
64
vendor/golang.org/x/tools/go/ssa/ssautil/load_test.go
generated
vendored
Normal file
64
vendor/golang.org/x/tools/go/ssa/ssautil/load_test.go
generated
vendored
Normal file
@@ -0,0 +1,64 @@
|
||||
// Copyright 2015 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 ssautil_test
|
||||
|
||||
import (
|
||||
"go/ast"
|
||||
"go/importer"
|
||||
"go/parser"
|
||||
"go/token"
|
||||
"go/types"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/tools/go/ssa/ssautil"
|
||||
)
|
||||
|
||||
const hello = `package main
|
||||
|
||||
import "fmt"
|
||||
|
||||
func main() {
|
||||
fmt.Println("Hello, world")
|
||||
}
|
||||
`
|
||||
|
||||
func TestBuildPackage(t *testing.T) {
|
||||
// There is a more substantial test of BuildPackage and the
|
||||
// SSA program it builds in ../ssa/builder_test.go.
|
||||
|
||||
fset := token.NewFileSet()
|
||||
f, err := parser.ParseFile(fset, "hello.go", hello, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
pkg := types.NewPackage("hello", "")
|
||||
ssapkg, _, err := ssautil.BuildPackage(&types.Config{Importer: importer.Default()}, fset, pkg, []*ast.File{f}, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if pkg.Name() != "main" {
|
||||
t.Errorf("pkg.Name() = %s, want main", pkg.Name())
|
||||
}
|
||||
if ssapkg.Func("main") == nil {
|
||||
ssapkg.WriteTo(os.Stderr)
|
||||
t.Errorf("ssapkg has no main function")
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildPackage_MissingImport(t *testing.T) {
|
||||
fset := token.NewFileSet()
|
||||
f, err := parser.ParseFile(fset, "bad.go", `package bad; import "missing"`, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
pkg := types.NewPackage("bad", "")
|
||||
ssapkg, _, err := ssautil.BuildPackage(new(types.Config), fset, pkg, []*ast.File{f}, 0)
|
||||
if err == nil || ssapkg != nil {
|
||||
t.Fatal("BuildPackage succeeded unexpectedly")
|
||||
}
|
||||
}
|
234
vendor/golang.org/x/tools/go/ssa/ssautil/switch.go
generated
vendored
Normal file
234
vendor/golang.org/x/tools/go/ssa/ssautil/switch.go
generated
vendored
Normal file
@@ -0,0 +1,234 @@
|
||||
// Copyright 2013 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 ssautil
|
||||
|
||||
// This file implements discovery of switch and type-switch constructs
|
||||
// from low-level control flow.
|
||||
//
|
||||
// Many techniques exist for compiling a high-level switch with
|
||||
// constant cases to efficient machine code. The optimal choice will
|
||||
// depend on the data type, the specific case values, the code in the
|
||||
// body of each case, and the hardware.
|
||||
// Some examples:
|
||||
// - a lookup table (for a switch that maps constants to constants)
|
||||
// - a computed goto
|
||||
// - a binary tree
|
||||
// - a perfect hash
|
||||
// - a two-level switch (to partition constant strings by their first byte).
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"go/token"
|
||||
"go/types"
|
||||
|
||||
"golang.org/x/tools/go/ssa"
|
||||
)
|
||||
|
||||
// A ConstCase represents a single constant comparison.
|
||||
// It is part of a Switch.
|
||||
type ConstCase struct {
|
||||
Block *ssa.BasicBlock // block performing the comparison
|
||||
Body *ssa.BasicBlock // body of the case
|
||||
Value *ssa.Const // case comparand
|
||||
}
|
||||
|
||||
// A TypeCase represents a single type assertion.
|
||||
// It is part of a Switch.
|
||||
type TypeCase struct {
|
||||
Block *ssa.BasicBlock // block performing the type assert
|
||||
Body *ssa.BasicBlock // body of the case
|
||||
Type types.Type // case type
|
||||
Binding ssa.Value // value bound by this case
|
||||
}
|
||||
|
||||
// A Switch is a logical high-level control flow operation
|
||||
// (a multiway branch) discovered by analysis of a CFG containing
|
||||
// only if/else chains. It is not part of the ssa.Instruction set.
|
||||
//
|
||||
// One of ConstCases and TypeCases has length >= 2;
|
||||
// the other is nil.
|
||||
//
|
||||
// In a value switch, the list of cases may contain duplicate constants.
|
||||
// A type switch may contain duplicate types, or types assignable
|
||||
// to an interface type also in the list.
|
||||
// TODO(adonovan): eliminate such duplicates.
|
||||
//
|
||||
type Switch struct {
|
||||
Start *ssa.BasicBlock // block containing start of if/else chain
|
||||
X ssa.Value // the switch operand
|
||||
ConstCases []ConstCase // ordered list of constant comparisons
|
||||
TypeCases []TypeCase // ordered list of type assertions
|
||||
Default *ssa.BasicBlock // successor if all comparisons fail
|
||||
}
|
||||
|
||||
func (sw *Switch) String() string {
|
||||
// We represent each block by the String() of its
|
||||
// first Instruction, e.g. "print(42:int)".
|
||||
var buf bytes.Buffer
|
||||
if sw.ConstCases != nil {
|
||||
fmt.Fprintf(&buf, "switch %s {\n", sw.X.Name())
|
||||
for _, c := range sw.ConstCases {
|
||||
fmt.Fprintf(&buf, "case %s: %s\n", c.Value, c.Body.Instrs[0])
|
||||
}
|
||||
} else {
|
||||
fmt.Fprintf(&buf, "switch %s.(type) {\n", sw.X.Name())
|
||||
for _, c := range sw.TypeCases {
|
||||
fmt.Fprintf(&buf, "case %s %s: %s\n",
|
||||
c.Binding.Name(), c.Type, c.Body.Instrs[0])
|
||||
}
|
||||
}
|
||||
if sw.Default != nil {
|
||||
fmt.Fprintf(&buf, "default: %s\n", sw.Default.Instrs[0])
|
||||
}
|
||||
fmt.Fprintf(&buf, "}")
|
||||
return buf.String()
|
||||
}
|
||||
|
||||
// Switches examines the control-flow graph of fn and returns the
|
||||
// set of inferred value and type switches. A value switch tests an
|
||||
// ssa.Value for equality against two or more compile-time constant
|
||||
// values. Switches involving link-time constants (addresses) are
|
||||
// ignored. A type switch type-asserts an ssa.Value against two or
|
||||
// more types.
|
||||
//
|
||||
// The switches are returned in dominance order.
|
||||
//
|
||||
// The resulting switches do not necessarily correspond to uses of the
|
||||
// 'switch' keyword in the source: for example, a single source-level
|
||||
// switch statement with non-constant cases may result in zero, one or
|
||||
// many Switches, one per plural sequence of constant cases.
|
||||
// Switches may even be inferred from if/else- or goto-based control flow.
|
||||
// (In general, the control flow constructs of the source program
|
||||
// cannot be faithfully reproduced from the SSA representation.)
|
||||
//
|
||||
func Switches(fn *ssa.Function) []Switch {
|
||||
// Traverse the CFG in dominance order, so we don't
|
||||
// enter an if/else-chain in the middle.
|
||||
var switches []Switch
|
||||
seen := make(map[*ssa.BasicBlock]bool) // TODO(adonovan): opt: use ssa.blockSet
|
||||
for _, b := range fn.DomPreorder() {
|
||||
if x, k := isComparisonBlock(b); x != nil {
|
||||
// Block b starts a switch.
|
||||
sw := Switch{Start: b, X: x}
|
||||
valueSwitch(&sw, k, seen)
|
||||
if len(sw.ConstCases) > 1 {
|
||||
switches = append(switches, sw)
|
||||
}
|
||||
}
|
||||
|
||||
if y, x, T := isTypeAssertBlock(b); y != nil {
|
||||
// Block b starts a type switch.
|
||||
sw := Switch{Start: b, X: x}
|
||||
typeSwitch(&sw, y, T, seen)
|
||||
if len(sw.TypeCases) > 1 {
|
||||
switches = append(switches, sw)
|
||||
}
|
||||
}
|
||||
}
|
||||
return switches
|
||||
}
|
||||
|
||||
func valueSwitch(sw *Switch, k *ssa.Const, seen map[*ssa.BasicBlock]bool) {
|
||||
b := sw.Start
|
||||
x := sw.X
|
||||
for x == sw.X {
|
||||
if seen[b] {
|
||||
break
|
||||
}
|
||||
seen[b] = true
|
||||
|
||||
sw.ConstCases = append(sw.ConstCases, ConstCase{
|
||||
Block: b,
|
||||
Body: b.Succs[0],
|
||||
Value: k,
|
||||
})
|
||||
b = b.Succs[1]
|
||||
if len(b.Instrs) > 2 {
|
||||
// Block b contains not just 'if x == k',
|
||||
// so it may have side effects that
|
||||
// make it unsafe to elide.
|
||||
break
|
||||
}
|
||||
if len(b.Preds) != 1 {
|
||||
// Block b has multiple predecessors,
|
||||
// so it cannot be treated as a case.
|
||||
break
|
||||
}
|
||||
x, k = isComparisonBlock(b)
|
||||
}
|
||||
sw.Default = b
|
||||
}
|
||||
|
||||
func typeSwitch(sw *Switch, y ssa.Value, T types.Type, seen map[*ssa.BasicBlock]bool) {
|
||||
b := sw.Start
|
||||
x := sw.X
|
||||
for x == sw.X {
|
||||
if seen[b] {
|
||||
break
|
||||
}
|
||||
seen[b] = true
|
||||
|
||||
sw.TypeCases = append(sw.TypeCases, TypeCase{
|
||||
Block: b,
|
||||
Body: b.Succs[0],
|
||||
Type: T,
|
||||
Binding: y,
|
||||
})
|
||||
b = b.Succs[1]
|
||||
if len(b.Instrs) > 4 {
|
||||
// Block b contains not just
|
||||
// {TypeAssert; Extract #0; Extract #1; If}
|
||||
// so it may have side effects that
|
||||
// make it unsafe to elide.
|
||||
break
|
||||
}
|
||||
if len(b.Preds) != 1 {
|
||||
// Block b has multiple predecessors,
|
||||
// so it cannot be treated as a case.
|
||||
break
|
||||
}
|
||||
y, x, T = isTypeAssertBlock(b)
|
||||
}
|
||||
sw.Default = b
|
||||
}
|
||||
|
||||
// isComparisonBlock returns the operands (v, k) if a block ends with
|
||||
// a comparison v==k, where k is a compile-time constant.
|
||||
//
|
||||
func isComparisonBlock(b *ssa.BasicBlock) (v ssa.Value, k *ssa.Const) {
|
||||
if n := len(b.Instrs); n >= 2 {
|
||||
if i, ok := b.Instrs[n-1].(*ssa.If); ok {
|
||||
if binop, ok := i.Cond.(*ssa.BinOp); ok && binop.Block() == b && binop.Op == token.EQL {
|
||||
if k, ok := binop.Y.(*ssa.Const); ok {
|
||||
return binop.X, k
|
||||
}
|
||||
if k, ok := binop.X.(*ssa.Const); ok {
|
||||
return binop.Y, k
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// isTypeAssertBlock returns the operands (y, x, T) if a block ends with
|
||||
// a type assertion "if y, ok := x.(T); ok {".
|
||||
//
|
||||
func isTypeAssertBlock(b *ssa.BasicBlock) (y, x ssa.Value, T types.Type) {
|
||||
if n := len(b.Instrs); n >= 4 {
|
||||
if i, ok := b.Instrs[n-1].(*ssa.If); ok {
|
||||
if ext1, ok := i.Cond.(*ssa.Extract); ok && ext1.Block() == b && ext1.Index == 1 {
|
||||
if ta, ok := ext1.Tuple.(*ssa.TypeAssert); ok && ta.Block() == b {
|
||||
// hack: relies upon instruction ordering.
|
||||
if ext0, ok := b.Instrs[n-3].(*ssa.Extract); ok {
|
||||
return ext0, ta.X, ta.AssertedType
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
74
vendor/golang.org/x/tools/go/ssa/ssautil/switch_test.go
generated
vendored
Normal file
74
vendor/golang.org/x/tools/go/ssa/ssautil/switch_test.go
generated
vendored
Normal file
@@ -0,0 +1,74 @@
|
||||
// Copyright 2013 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.
|
||||
|
||||
// No testdata on Android.
|
||||
|
||||
// +build !android
|
||||
|
||||
package ssautil_test
|
||||
|
||||
import (
|
||||
"go/parser"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/tools/go/loader"
|
||||
"golang.org/x/tools/go/ssa"
|
||||
"golang.org/x/tools/go/ssa/ssautil"
|
||||
)
|
||||
|
||||
func TestSwitches(t *testing.T) {
|
||||
conf := loader.Config{ParserMode: parser.ParseComments}
|
||||
f, err := conf.ParseFile("testdata/switches.go", nil)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
conf.CreateFromFiles("main", f)
|
||||
iprog, err := conf.Load()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
prog := ssautil.CreateProgram(iprog, 0)
|
||||
mainPkg := prog.Package(iprog.Created[0].Pkg)
|
||||
mainPkg.Build()
|
||||
|
||||
for _, mem := range mainPkg.Members {
|
||||
if fn, ok := mem.(*ssa.Function); ok {
|
||||
if fn.Synthetic != "" {
|
||||
continue // e.g. init()
|
||||
}
|
||||
// Each (multi-line) "switch" comment within
|
||||
// this function must match the printed form
|
||||
// of a ConstSwitch.
|
||||
var wantSwitches []string
|
||||
for _, c := range f.Comments {
|
||||
if fn.Syntax().Pos() <= c.Pos() && c.Pos() < fn.Syntax().End() {
|
||||
text := strings.TrimSpace(c.Text())
|
||||
if strings.HasPrefix(text, "switch ") {
|
||||
wantSwitches = append(wantSwitches, text)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
switches := ssautil.Switches(fn)
|
||||
if len(switches) != len(wantSwitches) {
|
||||
t.Errorf("in %s, found %d switches, want %d", fn, len(switches), len(wantSwitches))
|
||||
}
|
||||
for i, sw := range switches {
|
||||
got := sw.String()
|
||||
if i >= len(wantSwitches) {
|
||||
continue
|
||||
}
|
||||
want := wantSwitches[i]
|
||||
if got != want {
|
||||
t.Errorf("in %s, found switch %d: got <<%s>>, want <<%s>>", fn, i, got, want)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
357
vendor/golang.org/x/tools/go/ssa/ssautil/testdata/switches.go
generated
vendored
Normal file
357
vendor/golang.org/x/tools/go/ssa/ssautil/testdata/switches.go
generated
vendored
Normal file
@@ -0,0 +1,357 @@
|
||||
// +build ignore
|
||||
|
||||
package main
|
||||
|
||||
// This file is the input to TestSwitches in switch_test.go.
|
||||
// Each multiway conditional with constant or type cases (Switch)
|
||||
// discovered by Switches is printed, and compared with the
|
||||
// comments.
|
||||
//
|
||||
// The body of each case is printed as the value of its first
|
||||
// instruction.
|
||||
|
||||
// -------- Value switches --------
|
||||
|
||||
func SimpleSwitch(x, y int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: print(23:int)
|
||||
// case 3:int: print(23:int)
|
||||
// case 4:int: print(3:int)
|
||||
// default: x == y
|
||||
// }
|
||||
switch x {
|
||||
case 1:
|
||||
print(1)
|
||||
case 2, 3:
|
||||
print(23)
|
||||
fallthrough
|
||||
case 4:
|
||||
print(3)
|
||||
default:
|
||||
print(4)
|
||||
case y:
|
||||
print(5)
|
||||
}
|
||||
print(6)
|
||||
}
|
||||
|
||||
func four() int { return 4 }
|
||||
|
||||
// A non-constant case makes a switch "impure", but its pure
|
||||
// cases form two separate switches.
|
||||
func SwitchWithNonConstantCase(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: print(23:int)
|
||||
// case 3:int: print(23:int)
|
||||
// default: four()
|
||||
// }
|
||||
|
||||
// switch x {
|
||||
// case 5:int: print(5:int)
|
||||
// case 6:int: print(6:int)
|
||||
// default: print("done":string)
|
||||
// }
|
||||
switch x {
|
||||
case 1:
|
||||
print(1)
|
||||
case 2, 3:
|
||||
print(23)
|
||||
case four():
|
||||
print(3)
|
||||
case 5:
|
||||
print(5)
|
||||
case 6:
|
||||
print(6)
|
||||
}
|
||||
print("done")
|
||||
}
|
||||
|
||||
// Switches may be found even where the source
|
||||
// program doesn't have a switch statement.
|
||||
|
||||
func ImplicitSwitches(x, y int) {
|
||||
// switch x {
|
||||
// case 1:int: print(12:int)
|
||||
// case 2:int: print(12:int)
|
||||
// default: x < 5:int
|
||||
// }
|
||||
if x == 1 || 2 == x || x < 5 {
|
||||
print(12)
|
||||
}
|
||||
|
||||
// switch x {
|
||||
// case 3:int: print(34:int)
|
||||
// case 4:int: print(34:int)
|
||||
// default: x == y
|
||||
// }
|
||||
if x == 3 || 4 == x || x == y {
|
||||
print(34)
|
||||
}
|
||||
|
||||
// Not a switch: no consistent variable.
|
||||
if x == 5 || y == 6 {
|
||||
print(56)
|
||||
}
|
||||
|
||||
// Not a switch: only one constant comparison.
|
||||
if x == 7 || x == y {
|
||||
print(78)
|
||||
}
|
||||
}
|
||||
|
||||
func IfElseBasedSwitch(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: print(2:int)
|
||||
// default: print("else":string)
|
||||
// }
|
||||
if x == 1 {
|
||||
print(1)
|
||||
} else if x == 2 {
|
||||
print(2)
|
||||
} else {
|
||||
print("else")
|
||||
}
|
||||
}
|
||||
|
||||
func GotoBasedSwitch(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: print(2:int)
|
||||
// default: print("else":string)
|
||||
// }
|
||||
if x == 1 {
|
||||
goto L1
|
||||
}
|
||||
if x == 2 {
|
||||
goto L2
|
||||
}
|
||||
print("else")
|
||||
L1:
|
||||
print(1)
|
||||
goto end
|
||||
L2:
|
||||
print(2)
|
||||
end:
|
||||
}
|
||||
|
||||
func SwitchInAForLoop(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: print(2:int)
|
||||
// default: print("head":string)
|
||||
// }
|
||||
loop:
|
||||
for {
|
||||
print("head")
|
||||
switch x {
|
||||
case 1:
|
||||
print(1)
|
||||
break loop
|
||||
case 2:
|
||||
print(2)
|
||||
break loop
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// This case is a switch in a for-loop, both constructed using goto.
|
||||
// As before, the default case points back to the block containing the
|
||||
// switch, but that's ok.
|
||||
func SwitchInAForLoopUsingGoto(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: print(2:int)
|
||||
// default: print("head":string)
|
||||
// }
|
||||
loop:
|
||||
print("head")
|
||||
if x == 1 {
|
||||
goto L1
|
||||
}
|
||||
if x == 2 {
|
||||
goto L2
|
||||
}
|
||||
goto loop
|
||||
L1:
|
||||
print(1)
|
||||
goto end
|
||||
L2:
|
||||
print(2)
|
||||
end:
|
||||
}
|
||||
|
||||
func UnstructuredSwitchInAForLoop(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 2:int: x == 1:int
|
||||
// default: print("end":string)
|
||||
// }
|
||||
for {
|
||||
if x == 1 {
|
||||
print(1)
|
||||
return
|
||||
}
|
||||
if x == 2 {
|
||||
continue
|
||||
}
|
||||
break
|
||||
}
|
||||
print("end")
|
||||
}
|
||||
|
||||
func CaseWithMultiplePreds(x int) {
|
||||
for {
|
||||
if x == 1 {
|
||||
print(1)
|
||||
return
|
||||
}
|
||||
loop:
|
||||
// This block has multiple predecessors,
|
||||
// so can't be treated as a switch case.
|
||||
if x == 2 {
|
||||
goto loop
|
||||
}
|
||||
break
|
||||
}
|
||||
print("end")
|
||||
}
|
||||
|
||||
func DuplicateConstantsAreNotEliminated(x int) {
|
||||
// switch x {
|
||||
// case 1:int: print(1:int)
|
||||
// case 1:int: print("1a":string)
|
||||
// case 2:int: print(2:int)
|
||||
// default: return
|
||||
// }
|
||||
if x == 1 {
|
||||
print(1)
|
||||
} else if x == 1 { // duplicate => unreachable
|
||||
print("1a")
|
||||
} else if x == 2 {
|
||||
print(2)
|
||||
}
|
||||
}
|
||||
|
||||
// Interface values (created by comparisons) are not constants,
|
||||
// so ConstSwitch.X is never of interface type.
|
||||
func MakeInterfaceIsNotAConstant(x interface{}) {
|
||||
if x == "foo" {
|
||||
print("foo")
|
||||
} else if x == 1 {
|
||||
print(1)
|
||||
}
|
||||
}
|
||||
|
||||
func ZeroInitializedVarsAreConstants(x int) {
|
||||
// switch x {
|
||||
// case 0:int: print(1:int)
|
||||
// case 2:int: print(2:int)
|
||||
// default: print("end":string)
|
||||
// }
|
||||
var zero int // SSA construction replaces zero with 0
|
||||
if x == zero {
|
||||
print(1)
|
||||
} else if x == 2 {
|
||||
print(2)
|
||||
}
|
||||
print("end")
|
||||
}
|
||||
|
||||
// -------- Select --------
|
||||
|
||||
// NB, potentially fragile reliance on register number.
|
||||
func SelectDesugarsToSwitch(ch chan int) {
|
||||
// switch t1 {
|
||||
// case 0:int: extract t0 #2
|
||||
// case 1:int: println(0:int)
|
||||
// case 2:int: println(1:int)
|
||||
// default: println("default":string)
|
||||
// }
|
||||
select {
|
||||
case x := <-ch:
|
||||
println(x)
|
||||
case <-ch:
|
||||
println(0)
|
||||
case ch <- 1:
|
||||
println(1)
|
||||
default:
|
||||
println("default")
|
||||
}
|
||||
}
|
||||
|
||||
// NB, potentially fragile reliance on register number.
|
||||
func NonblockingSelectDefaultCasePanics(ch chan int) {
|
||||
// switch t1 {
|
||||
// case 0:int: extract t0 #2
|
||||
// case 1:int: println(0:int)
|
||||
// case 2:int: println(1:int)
|
||||
// default: make interface{} <- string ("blocking select m...":string)
|
||||
// }
|
||||
select {
|
||||
case x := <-ch:
|
||||
println(x)
|
||||
case <-ch:
|
||||
println(0)
|
||||
case ch <- 1:
|
||||
println(1)
|
||||
}
|
||||
}
|
||||
|
||||
// -------- Type switches --------
|
||||
|
||||
// NB, reliance on fragile register numbering.
|
||||
func SimpleTypeSwitch(x interface{}) {
|
||||
// switch x.(type) {
|
||||
// case t3 int: println(x)
|
||||
// case t7 bool: println(x)
|
||||
// case t10 string: println(t10)
|
||||
// default: println(x)
|
||||
// }
|
||||
switch y := x.(type) {
|
||||
case nil:
|
||||
println(y)
|
||||
case int, bool:
|
||||
println(y)
|
||||
case string:
|
||||
println(y)
|
||||
default:
|
||||
println(y)
|
||||
}
|
||||
}
|
||||
|
||||
// NB, potentially fragile reliance on register number.
|
||||
func DuplicateTypesAreNotEliminated(x interface{}) {
|
||||
// switch x.(type) {
|
||||
// case t1 string: println(1:int)
|
||||
// case t5 interface{}: println(t5)
|
||||
// case t9 int: println(3:int)
|
||||
// default: return
|
||||
// }
|
||||
switch y := x.(type) {
|
||||
case string:
|
||||
println(1)
|
||||
case interface{}:
|
||||
println(y)
|
||||
case int:
|
||||
println(3) // unreachable!
|
||||
}
|
||||
}
|
||||
|
||||
// NB, potentially fragile reliance on register number.
|
||||
func AdHocTypeSwitch(x interface{}) {
|
||||
// switch x.(type) {
|
||||
// case t1 int: println(t1)
|
||||
// case t5 string: println(t5)
|
||||
// default: print("default":string)
|
||||
// }
|
||||
if i, ok := x.(int); ok {
|
||||
println(i)
|
||||
} else if s, ok := x.(string); ok {
|
||||
println(s)
|
||||
} else {
|
||||
print("default")
|
||||
}
|
||||
}
|
79
vendor/golang.org/x/tools/go/ssa/ssautil/visit.go
generated
vendored
Normal file
79
vendor/golang.org/x/tools/go/ssa/ssautil/visit.go
generated
vendored
Normal file
@@ -0,0 +1,79 @@
|
||||
// Copyright 2013 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 ssautil // import "golang.org/x/tools/go/ssa/ssautil"
|
||||
|
||||
import "golang.org/x/tools/go/ssa"
|
||||
|
||||
// This file defines utilities for visiting the SSA representation of
|
||||
// a Program.
|
||||
//
|
||||
// TODO(adonovan): test coverage.
|
||||
|
||||
// AllFunctions finds and returns the set of functions potentially
|
||||
// needed by program prog, as determined by a simple linker-style
|
||||
// reachability algorithm starting from the members and method-sets of
|
||||
// each package. The result may include anonymous functions and
|
||||
// synthetic wrappers.
|
||||
//
|
||||
// Precondition: all packages are built.
|
||||
//
|
||||
func AllFunctions(prog *ssa.Program) map[*ssa.Function]bool {
|
||||
visit := visitor{
|
||||
prog: prog,
|
||||
seen: make(map[*ssa.Function]bool),
|
||||
}
|
||||
visit.program()
|
||||
return visit.seen
|
||||
}
|
||||
|
||||
type visitor struct {
|
||||
prog *ssa.Program
|
||||
seen map[*ssa.Function]bool
|
||||
}
|
||||
|
||||
func (visit *visitor) program() {
|
||||
for _, pkg := range visit.prog.AllPackages() {
|
||||
for _, mem := range pkg.Members {
|
||||
if fn, ok := mem.(*ssa.Function); ok {
|
||||
visit.function(fn)
|
||||
}
|
||||
}
|
||||
}
|
||||
for _, T := range visit.prog.RuntimeTypes() {
|
||||
mset := visit.prog.MethodSets.MethodSet(T)
|
||||
for i, n := 0, mset.Len(); i < n; i++ {
|
||||
visit.function(visit.prog.MethodValue(mset.At(i)))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (visit *visitor) function(fn *ssa.Function) {
|
||||
if !visit.seen[fn] {
|
||||
visit.seen[fn] = true
|
||||
var buf [10]*ssa.Value // avoid alloc in common case
|
||||
for _, b := range fn.Blocks {
|
||||
for _, instr := range b.Instrs {
|
||||
for _, op := range instr.Operands(buf[:0]) {
|
||||
if fn, ok := (*op).(*ssa.Function); ok {
|
||||
visit.function(fn)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// MainPackages returns the subset of the specified packages
|
||||
// named "main" that define a main function.
|
||||
// The result may include synthetic "testmain" packages.
|
||||
func MainPackages(pkgs []*ssa.Package) []*ssa.Package {
|
||||
var mains []*ssa.Package
|
||||
for _, pkg := range pkgs {
|
||||
if pkg.Pkg.Name() == "main" && pkg.Func("main") != nil {
|
||||
mains = append(mains, pkg)
|
||||
}
|
||||
}
|
||||
return mains
|
||||
}
|
Reference in New Issue
Block a user