github.com/tidwall/go@v0.0.0-20170415222209-6694a6888b7d/src/cmd/compile/internal/gc/noder.go (about)

     1  // Copyright 2016 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package gc
     6  
     7  import (
     8  	"fmt"
     9  	"os"
    10  	"strconv"
    11  	"strings"
    12  	"unicode/utf8"
    13  
    14  	"cmd/compile/internal/syntax"
    15  	"cmd/compile/internal/types"
    16  	"cmd/internal/obj"
    17  	"cmd/internal/src"
    18  )
    19  
    20  func parseFiles(filenames []string) uint {
    21  	var lines uint
    22  	var noders []*noder
    23  
    24  	for _, filename := range filenames {
    25  		p := &noder{err: make(chan syntax.Error)}
    26  		noders = append(noders, p)
    27  
    28  		go func(filename string) {
    29  			defer close(p.err)
    30  			base := src.NewFileBase(filename, absFilename(filename))
    31  
    32  			f, err := os.Open(filename)
    33  			if err != nil {
    34  				p.error(syntax.Error{Pos: src.MakePos(base, 0, 0), Msg: err.Error()})
    35  				return
    36  			}
    37  			defer f.Close()
    38  
    39  			p.file, _ = syntax.Parse(base, f, p.error, p.pragma, 0) // errors are tracked via p.error
    40  		}(filename)
    41  	}
    42  
    43  	for _, p := range noders {
    44  		for e := range p.err {
    45  			yyerrorpos(e.Pos, "%s", e.Msg)
    46  		}
    47  
    48  		p.node()
    49  		lines += p.file.Lines
    50  		p.file = nil // release memory
    51  
    52  		if nsyntaxerrors != 0 {
    53  			errorexit()
    54  		}
    55  		// Always run testdclstack here, even when debug_dclstack is not set, as a sanity measure.
    56  		testdclstack()
    57  	}
    58  
    59  	return lines
    60  }
    61  
    62  func yyerrorpos(pos src.Pos, format string, args ...interface{}) {
    63  	yyerrorl(Ctxt.PosTable.XPos(pos), format, args...)
    64  }
    65  
    66  var pathPrefix string
    67  
    68  func absFilename(name string) string {
    69  	return obj.AbsFile(Ctxt.Pathname, name, pathPrefix)
    70  }
    71  
    72  // noder transforms package syntax's AST into a Node tree.
    73  type noder struct {
    74  	file       *syntax.File
    75  	linknames  []linkname
    76  	pragcgobuf string
    77  	err        chan syntax.Error
    78  }
    79  
    80  // linkname records a //go:linkname directive.
    81  type linkname struct {
    82  	pos    src.Pos
    83  	local  string
    84  	remote string
    85  }
    86  
    87  func (p *noder) node() {
    88  	types.Block = 1
    89  	imported_unsafe = false
    90  
    91  	p.lineno(p.file.PkgName)
    92  	mkpackage(p.file.PkgName.Value)
    93  
    94  	xtop = append(xtop, p.decls(p.file.DeclList)...)
    95  
    96  	for _, n := range p.linknames {
    97  		if imported_unsafe {
    98  			lookup(n.local).Linkname = n.remote
    99  		} else {
   100  			yyerrorpos(n.pos, "//go:linkname only allowed in Go files that import \"unsafe\"")
   101  		}
   102  	}
   103  
   104  	pragcgobuf += p.pragcgobuf
   105  	lineno = src.NoXPos
   106  	clearImports()
   107  }
   108  
   109  func (p *noder) decls(decls []syntax.Decl) (l []*Node) {
   110  	var cs constState
   111  
   112  	for _, decl := range decls {
   113  		p.lineno(decl)
   114  		switch decl := decl.(type) {
   115  		case *syntax.ImportDecl:
   116  			p.importDecl(decl)
   117  
   118  		case *syntax.VarDecl:
   119  			l = append(l, p.varDecl(decl)...)
   120  
   121  		case *syntax.ConstDecl:
   122  			l = append(l, p.constDecl(decl, &cs)...)
   123  
   124  		case *syntax.TypeDecl:
   125  			l = append(l, p.typeDecl(decl))
   126  
   127  		case *syntax.FuncDecl:
   128  			l = append(l, p.funcDecl(decl))
   129  
   130  		default:
   131  			panic("unhandled Decl")
   132  		}
   133  	}
   134  
   135  	return
   136  }
   137  
   138  func (p *noder) importDecl(imp *syntax.ImportDecl) {
   139  	val := p.basicLit(imp.Path)
   140  	ipkg := importfile(&val)
   141  
   142  	if ipkg == nil {
   143  		if nerrors == 0 {
   144  			Fatalf("phase error in import")
   145  		}
   146  		return
   147  	}
   148  
   149  	ipkg.Direct = true
   150  
   151  	var my *types.Sym
   152  	if imp.LocalPkgName != nil {
   153  		my = p.name(imp.LocalPkgName)
   154  	} else {
   155  		my = lookup(ipkg.Name)
   156  	}
   157  
   158  	pack := p.nod(imp, OPACK, nil, nil)
   159  	pack.Sym = my
   160  	pack.Name.Pkg = ipkg
   161  
   162  	if my.Name == "." {
   163  		importdot(ipkg, pack)
   164  		return
   165  	}
   166  	if my.Name == "init" {
   167  		yyerrorl(pack.Pos, "cannot import package as init - init must be a func")
   168  		return
   169  	}
   170  	if my.Name == "_" {
   171  		return
   172  	}
   173  	if my.Def != nil {
   174  		lineno = pack.Pos
   175  		redeclare(my, "as imported package name")
   176  	}
   177  	my.Def = asTypesNode(pack)
   178  	my.Lastlineno = pack.Pos
   179  	my.Block = 1 // at top level
   180  }
   181  
   182  func (p *noder) varDecl(decl *syntax.VarDecl) []*Node {
   183  	names := p.declNames(decl.NameList)
   184  	typ := p.typeExprOrNil(decl.Type)
   185  
   186  	var exprs []*Node
   187  	if decl.Values != nil {
   188  		exprs = p.exprList(decl.Values)
   189  	}
   190  
   191  	p.lineno(decl)
   192  	return variter(names, typ, exprs)
   193  }
   194  
   195  // constState tracks state between constant specifiers within a
   196  // declaration group. This state is kept separate from noder so nested
   197  // constant declarations are handled correctly (e.g., issue 15550).
   198  type constState struct {
   199  	group  *syntax.Group
   200  	typ    *Node
   201  	values []*Node
   202  	iota   int64
   203  }
   204  
   205  func (p *noder) constDecl(decl *syntax.ConstDecl, cs *constState) []*Node {
   206  	if decl.Group == nil || decl.Group != cs.group {
   207  		*cs = constState{
   208  			group: decl.Group,
   209  		}
   210  	}
   211  
   212  	names := p.declNames(decl.NameList)
   213  	typ := p.typeExprOrNil(decl.Type)
   214  
   215  	var values []*Node
   216  	if decl.Values != nil {
   217  		values = p.exprList(decl.Values)
   218  		cs.typ, cs.values = typ, values
   219  	} else {
   220  		if typ != nil {
   221  			yyerror("const declaration cannot have type without expression")
   222  		}
   223  		typ, values = cs.typ, cs.values
   224  	}
   225  
   226  	var nn []*Node
   227  	for i, n := range names {
   228  		if i >= len(values) {
   229  			yyerror("missing value in const declaration")
   230  			break
   231  		}
   232  		v := values[i]
   233  		if decl.Values == nil {
   234  			v = treecopy(v, n.Pos)
   235  		}
   236  
   237  		n.Op = OLITERAL
   238  		declare(n, dclcontext)
   239  
   240  		n.Name.Param.Ntype = typ
   241  		n.Name.Defn = v
   242  		n.SetIota(cs.iota)
   243  
   244  		nn = append(nn, p.nod(decl, ODCLCONST, n, nil))
   245  	}
   246  
   247  	if len(values) > len(names) {
   248  		yyerror("extra expression in const declaration")
   249  	}
   250  
   251  	cs.iota++
   252  
   253  	return nn
   254  }
   255  
   256  func (p *noder) typeDecl(decl *syntax.TypeDecl) *Node {
   257  	n := p.declName(decl.Name)
   258  	n.Op = OTYPE
   259  	declare(n, dclcontext)
   260  	n.SetLocal(true)
   261  
   262  	// decl.Type may be nil but in that case we got a syntax error during parsing
   263  	typ := p.typeExprOrNil(decl.Type)
   264  
   265  	param := n.Name.Param
   266  	param.Ntype = typ
   267  	param.Pragma = decl.Pragma
   268  	param.Alias = decl.Alias
   269  	if param.Alias && param.Pragma != 0 {
   270  		yyerror("cannot specify directive with type alias")
   271  		param.Pragma = 0
   272  	}
   273  
   274  	return p.nod(decl, ODCLTYPE, n, nil)
   275  
   276  }
   277  
   278  func (p *noder) declNames(names []*syntax.Name) []*Node {
   279  	var nodes []*Node
   280  	for _, name := range names {
   281  		nodes = append(nodes, p.declName(name))
   282  	}
   283  	return nodes
   284  }
   285  
   286  func (p *noder) declName(name *syntax.Name) *Node {
   287  	// TODO(mdempsky): Set lineno?
   288  	return dclname(p.name(name))
   289  }
   290  
   291  func (p *noder) funcDecl(fun *syntax.FuncDecl) *Node {
   292  	name := p.name(fun.Name)
   293  	t := p.signature(fun.Recv, fun.Type)
   294  	f := p.nod(fun, ODCLFUNC, nil, nil)
   295  
   296  	if fun.Recv == nil {
   297  		if name.Name == "init" {
   298  			name = renameinit()
   299  			if t.List.Len() > 0 || t.Rlist.Len() > 0 {
   300  				yyerrorl(f.Pos, "func init must have no arguments and no return values")
   301  			}
   302  		}
   303  
   304  		if localpkg.Name == "main" && name.Name == "main" {
   305  			if t.List.Len() > 0 || t.Rlist.Len() > 0 {
   306  				yyerrorl(f.Pos, "func main must have no arguments and no return values")
   307  			}
   308  		}
   309  	} else {
   310  		f.Func.Shortname = name
   311  		name = nblank.Sym // filled in by typecheckfunc
   312  	}
   313  
   314  	f.Func.Nname = newfuncname(name)
   315  	f.Func.Nname.Name.Defn = f
   316  	f.Func.Nname.Name.Param.Ntype = t
   317  
   318  	pragma := fun.Pragma
   319  	f.Func.Pragma = fun.Pragma
   320  	f.SetNoescape(pragma&Noescape != 0)
   321  	if pragma&Systemstack != 0 && pragma&Nosplit != 0 {
   322  		yyerrorl(f.Pos, "go:nosplit and go:systemstack cannot be combined")
   323  	}
   324  
   325  	if fun.Recv == nil {
   326  		declare(f.Func.Nname, PFUNC)
   327  	}
   328  
   329  	funchdr(f)
   330  
   331  	if fun.Body != nil {
   332  		if f.Noescape() {
   333  			yyerrorl(f.Pos, "can only use //go:noescape with external func implementations")
   334  		}
   335  
   336  		body := p.stmts(fun.Body.List)
   337  		if body == nil {
   338  			body = []*Node{p.nod(fun, OEMPTY, nil, nil)}
   339  		}
   340  		f.Nbody.Set(body)
   341  
   342  		lineno = Ctxt.PosTable.XPos(fun.Body.Rbrace)
   343  		f.Func.Endlineno = lineno
   344  	} else {
   345  		if pure_go || strings.HasPrefix(f.Func.Nname.Sym.Name, "init.") {
   346  			yyerrorl(f.Pos, "missing function body for %q", f.Func.Nname.Sym.Name)
   347  		}
   348  	}
   349  
   350  	funcbody(f)
   351  
   352  	return f
   353  }
   354  
   355  func (p *noder) signature(recv *syntax.Field, typ *syntax.FuncType) *Node {
   356  	n := p.nod(typ, OTFUNC, nil, nil)
   357  	if recv != nil {
   358  		n.Left = p.param(recv, false, false)
   359  	}
   360  	n.List.Set(p.params(typ.ParamList, true))
   361  	n.Rlist.Set(p.params(typ.ResultList, false))
   362  	return n
   363  }
   364  
   365  func (p *noder) params(params []*syntax.Field, dddOk bool) []*Node {
   366  	var nodes []*Node
   367  	for i, param := range params {
   368  		p.lineno(param)
   369  		nodes = append(nodes, p.param(param, dddOk, i+1 == len(params)))
   370  	}
   371  	return nodes
   372  }
   373  
   374  func (p *noder) param(param *syntax.Field, dddOk, final bool) *Node {
   375  	var name *Node
   376  	if param.Name != nil {
   377  		name = p.newname(param.Name)
   378  	}
   379  
   380  	typ := p.typeExpr(param.Type)
   381  	n := p.nod(param, ODCLFIELD, name, typ)
   382  
   383  	// rewrite ...T parameter
   384  	if typ.Op == ODDD {
   385  		if !dddOk {
   386  			yyerror("cannot use ... in receiver or result parameter list")
   387  		} else if !final {
   388  			yyerror("can only use ... with final parameter in list")
   389  		}
   390  		typ.Op = OTARRAY
   391  		typ.Right = typ.Left
   392  		typ.Left = nil
   393  		n.SetIsddd(true)
   394  		if n.Left != nil {
   395  			n.Left.SetIsddd(true)
   396  		}
   397  	}
   398  
   399  	return n
   400  }
   401  
   402  func (p *noder) exprList(expr syntax.Expr) []*Node {
   403  	if list, ok := expr.(*syntax.ListExpr); ok {
   404  		return p.exprs(list.ElemList)
   405  	}
   406  	return []*Node{p.expr(expr)}
   407  }
   408  
   409  func (p *noder) exprs(exprs []syntax.Expr) []*Node {
   410  	var nodes []*Node
   411  	for _, expr := range exprs {
   412  		nodes = append(nodes, p.expr(expr))
   413  	}
   414  	return nodes
   415  }
   416  
   417  func (p *noder) expr(expr syntax.Expr) *Node {
   418  	p.lineno(expr)
   419  	switch expr := expr.(type) {
   420  	case nil, *syntax.BadExpr:
   421  		return nil
   422  	case *syntax.Name:
   423  		return p.mkname(expr)
   424  	case *syntax.BasicLit:
   425  		return p.setlineno(expr, nodlit(p.basicLit(expr)))
   426  
   427  	case *syntax.CompositeLit:
   428  		n := p.nod(expr, OCOMPLIT, nil, nil)
   429  		if expr.Type != nil {
   430  			n.Right = p.expr(expr.Type)
   431  		}
   432  		l := p.exprs(expr.ElemList)
   433  		for i, e := range l {
   434  			l[i] = p.wrapname(expr.ElemList[i], e)
   435  		}
   436  		n.List.Set(l)
   437  		lineno = Ctxt.PosTable.XPos(expr.Rbrace)
   438  		return n
   439  	case *syntax.KeyValueExpr:
   440  		return p.nod(expr, OKEY, p.expr(expr.Key), p.wrapname(expr.Value, p.expr(expr.Value)))
   441  	case *syntax.FuncLit:
   442  		closurehdr(p.typeExpr(expr.Type))
   443  		body := p.stmts(expr.Body.List)
   444  		lineno = Ctxt.PosTable.XPos(expr.Body.Rbrace)
   445  		return p.setlineno(expr, closurebody(body))
   446  	case *syntax.ParenExpr:
   447  		return p.nod(expr, OPAREN, p.expr(expr.X), nil)
   448  	case *syntax.SelectorExpr:
   449  		// parser.new_dotname
   450  		obj := p.expr(expr.X)
   451  		if obj.Op == OPACK {
   452  			obj.SetUsed(true)
   453  			return oldname(restrictlookup(expr.Sel.Value, obj.Name.Pkg))
   454  		}
   455  		return p.setlineno(expr, nodSym(OXDOT, obj, p.name(expr.Sel)))
   456  	case *syntax.IndexExpr:
   457  		return p.nod(expr, OINDEX, p.expr(expr.X), p.expr(expr.Index))
   458  	case *syntax.SliceExpr:
   459  		op := OSLICE
   460  		if expr.Full {
   461  			op = OSLICE3
   462  		}
   463  		n := p.nod(expr, op, p.expr(expr.X), nil)
   464  		var index [3]*Node
   465  		for i, x := range expr.Index {
   466  			if x != nil {
   467  				index[i] = p.expr(x)
   468  			}
   469  		}
   470  		n.SetSliceBounds(index[0], index[1], index[2])
   471  		return n
   472  	case *syntax.AssertExpr:
   473  		if expr.Type == nil {
   474  			panic("unexpected AssertExpr")
   475  		}
   476  		// TODO(mdempsky): parser.pexpr uses p.expr(), but
   477  		// seems like the type field should be parsed with
   478  		// ntype? Shrug, doesn't matter here.
   479  		return p.nod(expr, ODOTTYPE, p.expr(expr.X), p.expr(expr.Type))
   480  	case *syntax.Operation:
   481  		x := p.expr(expr.X)
   482  		if expr.Y == nil {
   483  			if expr.Op == syntax.And {
   484  				x = unparen(x) // TODO(mdempsky): Needed?
   485  				if x.Op == OCOMPLIT {
   486  					// Special case for &T{...}: turn into (*T){...}.
   487  					// TODO(mdempsky): Switch back to p.nod after we
   488  					// get rid of gcCompat.
   489  					x.Right = nod(OIND, x.Right, nil)
   490  					x.Right.SetImplicit(true)
   491  					return x
   492  				}
   493  			}
   494  			return p.nod(expr, p.unOp(expr.Op), x, nil)
   495  		}
   496  		return p.nod(expr, p.binOp(expr.Op), x, p.expr(expr.Y))
   497  	case *syntax.CallExpr:
   498  		n := p.nod(expr, OCALL, p.expr(expr.Fun), nil)
   499  		n.List.Set(p.exprs(expr.ArgList))
   500  		n.SetIsddd(expr.HasDots)
   501  		return n
   502  
   503  	case *syntax.ArrayType:
   504  		var len *Node
   505  		if expr.Len != nil {
   506  			len = p.expr(expr.Len)
   507  		} else {
   508  			len = p.nod(expr, ODDD, nil, nil)
   509  		}
   510  		return p.nod(expr, OTARRAY, len, p.typeExpr(expr.Elem))
   511  	case *syntax.SliceType:
   512  		return p.nod(expr, OTARRAY, nil, p.typeExpr(expr.Elem))
   513  	case *syntax.DotsType:
   514  		return p.nod(expr, ODDD, p.typeExpr(expr.Elem), nil)
   515  	case *syntax.StructType:
   516  		return p.structType(expr)
   517  	case *syntax.InterfaceType:
   518  		return p.interfaceType(expr)
   519  	case *syntax.FuncType:
   520  		return p.signature(nil, expr)
   521  	case *syntax.MapType:
   522  		return p.nod(expr, OTMAP, p.typeExpr(expr.Key), p.typeExpr(expr.Value))
   523  	case *syntax.ChanType:
   524  		n := p.nod(expr, OTCHAN, p.typeExpr(expr.Elem), nil)
   525  		n.Etype = types.EType(p.chanDir(expr.Dir))
   526  		return n
   527  
   528  	case *syntax.TypeSwitchGuard:
   529  		n := p.nod(expr, OTYPESW, nil, p.expr(expr.X))
   530  		if expr.Lhs != nil {
   531  			n.Left = p.declName(expr.Lhs)
   532  			if isblank(n.Left) {
   533  				yyerror("invalid variable name %v in type switch", n.Left)
   534  			}
   535  		}
   536  		return n
   537  	}
   538  	panic("unhandled Expr")
   539  }
   540  
   541  func (p *noder) typeExpr(typ syntax.Expr) *Node {
   542  	// TODO(mdempsky): Be stricter? typecheck should handle errors anyway.
   543  	return p.expr(typ)
   544  }
   545  
   546  func (p *noder) typeExprOrNil(typ syntax.Expr) *Node {
   547  	if typ != nil {
   548  		return p.expr(typ)
   549  	}
   550  	return nil
   551  }
   552  
   553  func (p *noder) chanDir(dir syntax.ChanDir) types.ChanDir {
   554  	switch dir {
   555  	case 0:
   556  		return types.Cboth
   557  	case syntax.SendOnly:
   558  		return types.Csend
   559  	case syntax.RecvOnly:
   560  		return types.Crecv
   561  	}
   562  	panic("unhandled ChanDir")
   563  }
   564  
   565  func (p *noder) structType(expr *syntax.StructType) *Node {
   566  	var l []*Node
   567  	for i, field := range expr.FieldList {
   568  		p.lineno(field)
   569  		var n *Node
   570  		if field.Name == nil {
   571  			n = p.embedded(field.Type)
   572  		} else {
   573  			n = p.nod(field, ODCLFIELD, p.newname(field.Name), p.typeExpr(field.Type))
   574  		}
   575  		if i < len(expr.TagList) && expr.TagList[i] != nil {
   576  			n.SetVal(p.basicLit(expr.TagList[i]))
   577  		}
   578  		l = append(l, n)
   579  	}
   580  
   581  	p.lineno(expr)
   582  	n := p.nod(expr, OTSTRUCT, nil, nil)
   583  	n.List.Set(l)
   584  	return n
   585  }
   586  
   587  func (p *noder) interfaceType(expr *syntax.InterfaceType) *Node {
   588  	var l []*Node
   589  	for _, method := range expr.MethodList {
   590  		p.lineno(method)
   591  		var n *Node
   592  		if method.Name == nil {
   593  			n = p.nod(method, ODCLFIELD, nil, oldname(p.packname(method.Type)))
   594  		} else {
   595  			mname := p.newname(method.Name)
   596  			sig := p.typeExpr(method.Type)
   597  			sig.Left = fakethis()
   598  			n = p.nod(method, ODCLFIELD, mname, sig)
   599  			ifacedcl(n)
   600  		}
   601  		l = append(l, n)
   602  	}
   603  
   604  	n := p.nod(expr, OTINTER, nil, nil)
   605  	n.List.Set(l)
   606  	return n
   607  }
   608  
   609  func (p *noder) packname(expr syntax.Expr) *types.Sym {
   610  	switch expr := expr.(type) {
   611  	case *syntax.Name:
   612  		name := p.name(expr)
   613  		if n := oldname(name); n.Name != nil && n.Name.Pack != nil {
   614  			n.Name.Pack.SetUsed(true)
   615  		}
   616  		return name
   617  	case *syntax.SelectorExpr:
   618  		name := p.name(expr.X.(*syntax.Name))
   619  		var pkg *types.Pkg
   620  		if asNode(name.Def) == nil || asNode(name.Def).Op != OPACK {
   621  			yyerror("%v is not a package", name)
   622  			pkg = localpkg
   623  		} else {
   624  			asNode(name.Def).SetUsed(true)
   625  			pkg = asNode(name.Def).Name.Pkg
   626  		}
   627  		return restrictlookup(expr.Sel.Value, pkg)
   628  	}
   629  	panic(fmt.Sprintf("unexpected packname: %#v", expr))
   630  }
   631  
   632  func (p *noder) embedded(typ syntax.Expr) *Node {
   633  	op, isStar := typ.(*syntax.Operation)
   634  	if isStar {
   635  		if op.Op != syntax.Mul || op.Y != nil {
   636  			panic("unexpected Operation")
   637  		}
   638  		typ = op.X
   639  	}
   640  	n := embedded(p.packname(typ), localpkg)
   641  	if isStar {
   642  		n.Right = p.nod(op, OIND, n.Right, nil)
   643  	}
   644  	return n
   645  }
   646  
   647  func (p *noder) stmts(stmts []syntax.Stmt) []*Node {
   648  	var nodes []*Node
   649  	for _, stmt := range stmts {
   650  		s := p.stmt(stmt)
   651  		if s == nil {
   652  		} else if s.Op == OBLOCK && s.Ninit.Len() == 0 {
   653  			nodes = append(nodes, s.List.Slice()...)
   654  		} else {
   655  			nodes = append(nodes, s)
   656  		}
   657  	}
   658  	return nodes
   659  }
   660  
   661  func (p *noder) stmt(stmt syntax.Stmt) *Node {
   662  	p.lineno(stmt)
   663  	switch stmt := stmt.(type) {
   664  	case *syntax.EmptyStmt:
   665  		return nil
   666  	case *syntax.LabeledStmt:
   667  		return p.labeledStmt(stmt)
   668  	case *syntax.BlockStmt:
   669  		l := p.blockStmt(stmt)
   670  		if len(l) == 0 {
   671  			// TODO(mdempsky): Line number?
   672  			return nod(OEMPTY, nil, nil)
   673  		}
   674  		return liststmt(l)
   675  	case *syntax.ExprStmt:
   676  		return p.wrapname(stmt, p.expr(stmt.X))
   677  	case *syntax.SendStmt:
   678  		return p.nod(stmt, OSEND, p.expr(stmt.Chan), p.expr(stmt.Value))
   679  	case *syntax.DeclStmt:
   680  		return liststmt(p.decls(stmt.DeclList))
   681  	case *syntax.AssignStmt:
   682  		if stmt.Op != 0 && stmt.Op != syntax.Def {
   683  			n := p.nod(stmt, OASOP, p.expr(stmt.Lhs), p.expr(stmt.Rhs))
   684  			n.SetImplicit(stmt.Rhs == syntax.ImplicitOne)
   685  			n.Etype = types.EType(p.binOp(stmt.Op))
   686  			return n
   687  		}
   688  
   689  		lhs := p.exprList(stmt.Lhs)
   690  		rhs := p.exprList(stmt.Rhs)
   691  
   692  		n := p.nod(stmt, OAS, nil, nil) // assume common case
   693  
   694  		if stmt.Op == syntax.Def {
   695  			n.SetColas(true)
   696  			colasdefn(lhs, n) // modifies lhs, call before using lhs[0] in common case
   697  		}
   698  
   699  		if len(lhs) == 1 && len(rhs) == 1 {
   700  			// common case
   701  			n.Left = lhs[0]
   702  			n.Right = rhs[0]
   703  		} else {
   704  			n.Op = OAS2
   705  			n.List.Set(lhs)
   706  			n.Rlist.Set(rhs)
   707  		}
   708  		return n
   709  
   710  	case *syntax.BranchStmt:
   711  		var op Op
   712  		switch stmt.Tok {
   713  		case syntax.Break:
   714  			op = OBREAK
   715  		case syntax.Continue:
   716  			op = OCONTINUE
   717  		case syntax.Fallthrough:
   718  			op = OXFALL
   719  		case syntax.Goto:
   720  			op = OGOTO
   721  		default:
   722  			panic("unhandled BranchStmt")
   723  		}
   724  		n := p.nod(stmt, op, nil, nil)
   725  		if stmt.Label != nil {
   726  			n.Left = p.newname(stmt.Label)
   727  		}
   728  		if op == OGOTO {
   729  			n.Sym = types.Dclstack // context, for goto restriction
   730  		}
   731  		if op == OXFALL {
   732  			n.Xoffset = int64(types.Block)
   733  		}
   734  		return n
   735  	case *syntax.CallStmt:
   736  		var op Op
   737  		switch stmt.Tok {
   738  		case syntax.Defer:
   739  			op = ODEFER
   740  		case syntax.Go:
   741  			op = OPROC
   742  		default:
   743  			panic("unhandled CallStmt")
   744  		}
   745  		return p.nod(stmt, op, p.expr(stmt.Call), nil)
   746  	case *syntax.ReturnStmt:
   747  		var results []*Node
   748  		if stmt.Results != nil {
   749  			results = p.exprList(stmt.Results)
   750  		}
   751  		n := p.nod(stmt, ORETURN, nil, nil)
   752  		n.List.Set(results)
   753  		if n.List.Len() == 0 && Curfn != nil {
   754  			for _, ln := range Curfn.Func.Dcl {
   755  				if ln.Class == PPARAM {
   756  					continue
   757  				}
   758  				if ln.Class != PPARAMOUT {
   759  					break
   760  				}
   761  				if asNode(ln.Sym.Def) != ln {
   762  					yyerror("%s is shadowed during return", ln.Sym.Name)
   763  				}
   764  			}
   765  		}
   766  		return n
   767  	case *syntax.IfStmt:
   768  		return p.ifStmt(stmt)
   769  	case *syntax.ForStmt:
   770  		return p.forStmt(stmt)
   771  	case *syntax.SwitchStmt:
   772  		return p.switchStmt(stmt)
   773  	case *syntax.SelectStmt:
   774  		return p.selectStmt(stmt)
   775  	}
   776  	panic("unhandled Stmt")
   777  }
   778  
   779  func (p *noder) blockStmt(stmt *syntax.BlockStmt) []*Node {
   780  	types.Markdcl(lineno)
   781  	nodes := p.stmts(stmt.List)
   782  	types.Popdcl()
   783  	return nodes
   784  }
   785  
   786  func (p *noder) ifStmt(stmt *syntax.IfStmt) *Node {
   787  	types.Markdcl(lineno)
   788  	n := p.nod(stmt, OIF, nil, nil)
   789  	if stmt.Init != nil {
   790  		n.Ninit.Set1(p.stmt(stmt.Init))
   791  	}
   792  	if stmt.Cond != nil {
   793  		n.Left = p.expr(stmt.Cond)
   794  	}
   795  	n.Nbody.Set(p.blockStmt(stmt.Then))
   796  	if stmt.Else != nil {
   797  		e := p.stmt(stmt.Else)
   798  		if e.Op == OBLOCK && e.Ninit.Len() == 0 {
   799  			n.Rlist.Set(e.List.Slice())
   800  		} else {
   801  			n.Rlist.Set1(e)
   802  		}
   803  	}
   804  	types.Popdcl()
   805  	return n
   806  }
   807  
   808  func (p *noder) forStmt(stmt *syntax.ForStmt) *Node {
   809  	types.Markdcl(lineno)
   810  	var n *Node
   811  	if r, ok := stmt.Init.(*syntax.RangeClause); ok {
   812  		if stmt.Cond != nil || stmt.Post != nil {
   813  			panic("unexpected RangeClause")
   814  		}
   815  
   816  		n = p.nod(r, ORANGE, nil, p.expr(r.X))
   817  		if r.Lhs != nil {
   818  			lhs := p.exprList(r.Lhs)
   819  			n.List.Set(lhs)
   820  			if r.Def {
   821  				n.SetColas(true)
   822  				colasdefn(lhs, n)
   823  			}
   824  		}
   825  	} else {
   826  		n = p.nod(stmt, OFOR, nil, nil)
   827  		if stmt.Init != nil {
   828  			n.Ninit.Set1(p.stmt(stmt.Init))
   829  		}
   830  		if stmt.Cond != nil {
   831  			n.Left = p.expr(stmt.Cond)
   832  		}
   833  		if stmt.Post != nil {
   834  			n.Right = p.stmt(stmt.Post)
   835  		}
   836  	}
   837  	n.Nbody.Set(p.blockStmt(stmt.Body))
   838  	types.Popdcl()
   839  	return n
   840  }
   841  
   842  func (p *noder) switchStmt(stmt *syntax.SwitchStmt) *Node {
   843  	types.Markdcl(lineno)
   844  	n := p.nod(stmt, OSWITCH, nil, nil)
   845  	if stmt.Init != nil {
   846  		n.Ninit.Set1(p.stmt(stmt.Init))
   847  	}
   848  	if stmt.Tag != nil {
   849  		n.Left = p.expr(stmt.Tag)
   850  	}
   851  
   852  	tswitch := n.Left
   853  	if tswitch != nil && (tswitch.Op != OTYPESW || tswitch.Left == nil) {
   854  		tswitch = nil
   855  	}
   856  
   857  	n.List.Set(p.caseClauses(stmt.Body, tswitch))
   858  
   859  	types.Popdcl()
   860  	return n
   861  }
   862  
   863  func (p *noder) caseClauses(clauses []*syntax.CaseClause, tswitch *Node) []*Node {
   864  	var nodes []*Node
   865  	for _, clause := range clauses {
   866  		p.lineno(clause)
   867  		types.Markdcl(lineno)
   868  		n := p.nod(clause, OXCASE, nil, nil)
   869  		if clause.Cases != nil {
   870  			n.List.Set(p.exprList(clause.Cases))
   871  		}
   872  		if tswitch != nil {
   873  			nn := newname(tswitch.Left.Sym)
   874  			declare(nn, dclcontext)
   875  			n.Rlist.Set1(nn)
   876  			// keep track of the instances for reporting unused
   877  			nn.Name.Defn = tswitch
   878  		}
   879  		n.Xoffset = int64(types.Block)
   880  		n.Nbody.Set(p.stmts(clause.Body))
   881  		types.Popdcl()
   882  		nodes = append(nodes, n)
   883  	}
   884  	return nodes
   885  }
   886  
   887  func (p *noder) selectStmt(stmt *syntax.SelectStmt) *Node {
   888  	n := p.nod(stmt, OSELECT, nil, nil)
   889  	n.List.Set(p.commClauses(stmt.Body))
   890  	return n
   891  }
   892  
   893  func (p *noder) commClauses(clauses []*syntax.CommClause) []*Node {
   894  	var nodes []*Node
   895  	for _, clause := range clauses {
   896  		p.lineno(clause)
   897  		types.Markdcl(lineno)
   898  		n := p.nod(clause, OXCASE, nil, nil)
   899  		if clause.Comm != nil {
   900  			n.List.Set1(p.stmt(clause.Comm))
   901  		}
   902  		n.Xoffset = int64(types.Block)
   903  		n.Nbody.Set(p.stmts(clause.Body))
   904  		types.Popdcl()
   905  		nodes = append(nodes, n)
   906  	}
   907  	return nodes
   908  }
   909  
   910  func (p *noder) labeledStmt(label *syntax.LabeledStmt) *Node {
   911  	lhs := p.nod(label, OLABEL, p.newname(label.Label), nil)
   912  	lhs.Sym = types.Dclstack // context, for goto restriction
   913  
   914  	var ls *Node
   915  	if label.Stmt != nil { // TODO(mdempsky): Should always be present.
   916  		ls = p.stmt(label.Stmt)
   917  	}
   918  
   919  	lhs.Name.Defn = ls
   920  	l := []*Node{lhs}
   921  	if ls != nil {
   922  		if ls.Op == OBLOCK && ls.Ninit.Len() == 0 {
   923  			l = append(l, ls.List.Slice()...)
   924  		} else {
   925  			l = append(l, ls)
   926  		}
   927  	}
   928  	return liststmt(l)
   929  }
   930  
   931  var unOps = [...]Op{
   932  	syntax.Recv: ORECV,
   933  	syntax.Mul:  OIND,
   934  	syntax.And:  OADDR,
   935  
   936  	syntax.Not: ONOT,
   937  	syntax.Xor: OCOM,
   938  	syntax.Add: OPLUS,
   939  	syntax.Sub: OMINUS,
   940  }
   941  
   942  func (p *noder) unOp(op syntax.Operator) Op {
   943  	if uint64(op) >= uint64(len(unOps)) || unOps[op] == 0 {
   944  		panic("invalid Operator")
   945  	}
   946  	return unOps[op]
   947  }
   948  
   949  var binOps = [...]Op{
   950  	syntax.OrOr:   OOROR,
   951  	syntax.AndAnd: OANDAND,
   952  
   953  	syntax.Eql: OEQ,
   954  	syntax.Neq: ONE,
   955  	syntax.Lss: OLT,
   956  	syntax.Leq: OLE,
   957  	syntax.Gtr: OGT,
   958  	syntax.Geq: OGE,
   959  
   960  	syntax.Add: OADD,
   961  	syntax.Sub: OSUB,
   962  	syntax.Or:  OOR,
   963  	syntax.Xor: OXOR,
   964  
   965  	syntax.Mul:    OMUL,
   966  	syntax.Div:    ODIV,
   967  	syntax.Rem:    OMOD,
   968  	syntax.And:    OAND,
   969  	syntax.AndNot: OANDNOT,
   970  	syntax.Shl:    OLSH,
   971  	syntax.Shr:    ORSH,
   972  }
   973  
   974  func (p *noder) binOp(op syntax.Operator) Op {
   975  	if uint64(op) >= uint64(len(binOps)) || binOps[op] == 0 {
   976  		panic("invalid Operator")
   977  	}
   978  	return binOps[op]
   979  }
   980  
   981  func (p *noder) basicLit(lit *syntax.BasicLit) Val {
   982  	// TODO: Don't try to convert if we had syntax errors (conversions may fail).
   983  	//       Use dummy values so we can continue to compile. Eventually, use a
   984  	//       form of "unknown" literals that are ignored during type-checking so
   985  	//       we can continue type-checking w/o spurious follow-up errors.
   986  	switch s := lit.Value; lit.Kind {
   987  	case syntax.IntLit:
   988  		x := new(Mpint)
   989  		x.SetString(s)
   990  		return Val{U: x}
   991  
   992  	case syntax.FloatLit:
   993  		x := newMpflt()
   994  		x.SetString(s)
   995  		return Val{U: x}
   996  
   997  	case syntax.ImagLit:
   998  		x := new(Mpcplx)
   999  		x.Imag.SetString(strings.TrimSuffix(s, "i"))
  1000  		return Val{U: x}
  1001  
  1002  	case syntax.RuneLit:
  1003  		var r rune
  1004  		if u, err := strconv.Unquote(s); err == nil && len(u) > 0 {
  1005  			// Package syntax already reported any errors.
  1006  			// Check for them again though because 0 is a
  1007  			// better fallback value for invalid rune
  1008  			// literals than 0xFFFD.
  1009  			if len(u) == 1 {
  1010  				r = rune(u[0])
  1011  			} else {
  1012  				r, _ = utf8.DecodeRuneInString(u)
  1013  			}
  1014  		}
  1015  		x := new(Mpint)
  1016  		x.SetInt64(int64(r))
  1017  		x.Rune = true
  1018  		return Val{U: x}
  1019  
  1020  	case syntax.StringLit:
  1021  		if len(s) > 0 && s[0] == '`' {
  1022  			// strip carriage returns from raw string
  1023  			s = strings.Replace(s, "\r", "", -1)
  1024  		}
  1025  		// Ignore errors because package syntax already reported them.
  1026  		u, _ := strconv.Unquote(s)
  1027  		return Val{U: u}
  1028  
  1029  	default:
  1030  		panic("unhandled BasicLit kind")
  1031  	}
  1032  }
  1033  
  1034  func (p *noder) name(name *syntax.Name) *types.Sym {
  1035  	return lookup(name.Value)
  1036  }
  1037  
  1038  func (p *noder) mkname(name *syntax.Name) *Node {
  1039  	// TODO(mdempsky): Set line number?
  1040  	return mkname(p.name(name))
  1041  }
  1042  
  1043  func (p *noder) newname(name *syntax.Name) *Node {
  1044  	// TODO(mdempsky): Set line number?
  1045  	return newname(p.name(name))
  1046  }
  1047  
  1048  func (p *noder) wrapname(n syntax.Node, x *Node) *Node {
  1049  	// These nodes do not carry line numbers.
  1050  	// Introduce a wrapper node to give them the correct line.
  1051  	switch x.Op {
  1052  	case OTYPE, OLITERAL:
  1053  		if x.Sym == nil {
  1054  			break
  1055  		}
  1056  		fallthrough
  1057  	case ONAME, ONONAME, OPACK:
  1058  		x = p.nod(n, OPAREN, x, nil)
  1059  		x.SetImplicit(true)
  1060  	}
  1061  	return x
  1062  }
  1063  
  1064  func (p *noder) nod(orig syntax.Node, op Op, left, right *Node) *Node {
  1065  	return p.setlineno(orig, nod(op, left, right))
  1066  }
  1067  
  1068  func (p *noder) setlineno(src_ syntax.Node, dst *Node) *Node {
  1069  	pos := src_.Pos()
  1070  	if !pos.IsKnown() {
  1071  		// TODO(mdempsky): Shouldn't happen. Fix package syntax.
  1072  		return dst
  1073  	}
  1074  	dst.Pos = Ctxt.PosTable.XPos(pos)
  1075  	return dst
  1076  }
  1077  
  1078  func (p *noder) lineno(n syntax.Node) {
  1079  	if n == nil {
  1080  		return
  1081  	}
  1082  	pos := n.Pos()
  1083  	if !pos.IsKnown() {
  1084  		// TODO(mdempsky): Shouldn't happen. Fix package syntax.
  1085  		return
  1086  	}
  1087  	lineno = Ctxt.PosTable.XPos(pos)
  1088  }
  1089  
  1090  // error is called concurrently if files are parsed concurrently.
  1091  func (p *noder) error(err error) {
  1092  	p.err <- err.(syntax.Error)
  1093  }
  1094  
  1095  // pragma is called concurrently if files are parsed concurrently.
  1096  func (p *noder) pragma(pos src.Pos, text string) syntax.Pragma {
  1097  	switch {
  1098  	case strings.HasPrefix(text, "line "):
  1099  		// line directives are handled by syntax package
  1100  		panic("unreachable")
  1101  
  1102  	case strings.HasPrefix(text, "go:linkname "):
  1103  		f := strings.Fields(text)
  1104  		if len(f) != 3 {
  1105  			p.error(syntax.Error{Pos: pos, Msg: "usage: //go:linkname localname linkname"})
  1106  			break
  1107  		}
  1108  		p.linknames = append(p.linknames, linkname{pos, f[1], f[2]})
  1109  
  1110  	case strings.HasPrefix(text, "go:cgo_"):
  1111  		p.pragcgobuf += p.pragcgo(pos, text)
  1112  		fallthrough // because of //go:cgo_unsafe_args
  1113  	default:
  1114  		verb := text
  1115  		if i := strings.Index(text, " "); i >= 0 {
  1116  			verb = verb[:i]
  1117  		}
  1118  		prag := pragmaValue(verb)
  1119  		const runtimePragmas = Systemstack | Nowritebarrier | Nowritebarrierrec | Yeswritebarrierrec
  1120  		if !compiling_runtime && prag&runtimePragmas != 0 {
  1121  			p.error(syntax.Error{Pos: pos, Msg: fmt.Sprintf("//%s only allowed in runtime", verb)})
  1122  		}
  1123  		return prag
  1124  	}
  1125  
  1126  	return 0
  1127  }
  1128  
  1129  func mkname(sym *types.Sym) *Node {
  1130  	n := oldname(sym)
  1131  	if n.Name != nil && n.Name.Pack != nil {
  1132  		n.Name.Pack.SetUsed(true)
  1133  	}
  1134  	return n
  1135  }
  1136  
  1137  func unparen(x *Node) *Node {
  1138  	for x.Op == OPAREN {
  1139  		x = x.Left
  1140  	}
  1141  	return x
  1142  }