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