qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-11-26 05:07:59 -07:00
Code Releases Activity

cmd/compile: be sure to wrap defer/go calls with arguments

Browse Source 
CL 298669 implemented wrapping for defer/go calls so the function
being called with defer or go statement has no arguments. This
simplifies the compiler and the runtime, especially with the
new ABI.

If the called function does not have any argument, we don't need
to wrap. But the code missed the cases of method receiver, as
well as some apparent argumentless builtin calls which may later
be rewritten to having arguments (e.g. recover). This CL makes
sure to wrap those cases. Also add a check to ensure that go and
defer calls are indeed argumentless.

Handle "defer recover()" specially, as recover() is lowered to
runtime.gorecover(FP) where FP is the frame's FP. FP needs to be
evaluated before wrapping.

Updates #40724.

Change-Id: I2758b6c69ab6aa02dd588441a457fe28ddd0d5a5
Reviewed-on: https://go-review.googlesource.com/c/go/+/304771
Trust: Cherry Zhang <cherryyz@google.com>
Reviewed-by: Than McIntosh <thanm@google.com>
This commit is contained in:
Cherry Zhang 2021-03-25 13:24:16 -04:00
parent bd6628e62d
commit e27f3966bb
4 changed files with 40 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/cmd/compile/internal/ssagen/ssa.go
View File

@ -4608,6 +4608,10 @@ func (s *state) openDeferRecord(n *ir.CallExpr) {
var args []*ssa.Value
var argNodes []*ir.Name
if objabi.Experiment.RegabiDefer && (len(n.Args) != 0 || n.Op() == ir.OCALLINTER) {
s.Fatalf("defer call with arguments: %v", n)
}
opendefer := &openDeferInfo{
n: n,
}
@ -4856,6 +4860,10 @@ func (s *state) call(n *ir.CallExpr, k callKind, returnResultAddr bool) *ssa.Val
}
}
if objabi.Experiment.RegabiDefer && k != callNormal && (len(n.Args) != 0 || n.Op() == ir.OCALLINTER) {
s.Fatalf("go/defer call with arguments: %v", n)
}
switch n.Op() {
case ir.OCALLFUNC:
if k == callNormal && fn.Op() == ir.ONAME && fn.(*ir.Name).Class == ir.PFUNC {

12
src/cmd/compile/internal/walk/builtin.go
View File

@ -653,6 +653,18 @@ func walkPrint(nn *ir.CallExpr, init *ir.Nodes) ir.Node {
return walkStmt(typecheck.Stmt(r))
}
// walkRecover walks an ORECOVER node.
func walkRecover(nn *ir.CallExpr, init *ir.Nodes) ir.Node {
// Call gorecover with the FP of this frame.
// FP is equal to caller's SP plus FixedFrameSize().
var fp ir.Node = mkcall("getcallersp", types.Types[types.TUINTPTR], init)
if off := base.Ctxt.FixedFrameSize(); off != 0 {
fp = ir.NewBinaryExpr(fp.Pos(), ir.OADD, fp, ir.NewInt(off))
}
fp = ir.NewConvExpr(fp.Pos(), ir.OCONVNOP, types.NewPtr(types.Types[types.TINT32]), fp)
return mkcall("gorecover", nn.Type(), init, fp)
}
func badtype(op ir.Op, tl, tr *types.Type) {
var s string
if tl != nil {

10
src/cmd/compile/internal/walk/expr.go
View File

@ -157,15 +157,7 @@ func walkExpr1(n ir.Node, init *ir.Nodes) ir.Node {
return mkcall("gopanic", nil, init, n.X)
case ir.ORECOVER:
n := n.(*ir.CallExpr)
// Call gorecover with the FP of this frame.
// FP is equal to caller's SP plus FixedFrameSize().
var fp ir.Node = mkcall("getcallersp", types.Types[types.TUINTPTR], init)
if off := base.Ctxt.FixedFrameSize(); off != 0 {
fp = ir.NewBinaryExpr(fp.Pos(), ir.OADD, fp, ir.NewInt(off))
}
fp = ir.NewConvExpr(fp.Pos(), ir.OCONVNOP, types.NewPtr(types.Types[types.TINT32]), fp)
return mkcall("gorecover", n.Type(), init, fp)
return walkRecover(n.(*ir.CallExpr), init)
case ir.OCFUNC:
return n

25
src/cmd/compile/internal/walk/order.go
View File

@ -732,6 +732,13 @@ func (o *orderState) stmt(n ir.Node) {
t := o.markTemp()
o.init(n.Call)
o.call(n.Call)
if n.Call.Op() == ir.ORECOVER {
// Special handling of "defer recover()". We need to evaluate the FP
// argument before wrapping.
var init ir.Nodes
n.Call = walkRecover(n.Call.(*ir.CallExpr), &init)
o.stmtList(init)
}
if objabi.Experiment.RegabiDefer {
o.wrapGoDefer(n)
}
@ -1481,7 +1488,7 @@ func (o *orderState) wrapGoDefer(n *ir.GoDeferStmt) {
callArgs = []ir.Node{x.X}
mkNewCall = func(pos src.XPos, op ir.Op, fun ir.Node, args []ir.Node) ir.Node {
if len(args) != 1 {
panic("internal error, expecting single arg to close")
panic("internal error, expecting single arg")
}
return ir.Node(ir.NewUnaryExpr(pos, op, args[0]))
}
@ -1497,11 +1504,17 @@ func (o *orderState) wrapGoDefer(n *ir.GoDeferStmt) {
panic("unhandled op")
}
// No need to wrap if called func has no args. However in the case
// of "defer func() { ... }()" we need to protect against the
// possibility of directClosureCall rewriting things so that the
// call does have arguments.
if len(callArgs) == 0 {
// No need to wrap if called func has no args and no receiver.
// However in the case of "defer func() { ... }()" we need to
// protect against the possibility of directClosureCall rewriting
// things so that the call does have arguments.
//
// Do wrap method calls (OCALLMETH, OCALLINTER), because it has
// a receiver.
//
// Also do wrap builtin functions, because they may be expanded to
// calls with arguments (e.g. ORECOVER).
if len(callArgs) == 0 && call.Op() == ir.OCALLFUNC {
if c, ok := call.(*ir.CallExpr); ok && callX != nil && callX.Op() == ir.OCLOSURE {
cloFunc := callX.(*ir.ClosureExpr).Func
cloFunc.SetClosureCalled(false)