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