diff --git a/src/cmd/gc/go.h b/src/cmd/gc/go.h index 5a2b774316c..ca313a0478b 100644 --- a/src/cmd/gc/go.h +++ b/src/cmd/gc/go.h @@ -982,6 +982,7 @@ EXTERN int funcdepth; EXTERN int typecheckok; EXTERN int compiling_runtime; EXTERN int compiling_wrappers; +EXTERN int inl_nonlocal; EXTERN int use_writebarrier; EXTERN int pure_go; EXTERN char* flag_installsuffix; diff --git a/src/cmd/gc/inl.c b/src/cmd/gc/inl.c index cf89b009028..45e15bb9b7c 100644 --- a/src/cmd/gc/inl.c +++ b/src/cmd/gc/inl.c @@ -804,9 +804,12 @@ inlvar(Node *var) n->curfn = curfn; // the calling function, not the called one n->addrtaken = var->addrtaken; - // esc pass wont run if we're inlining into a iface wrapper - // luckily, we can steal the results from the target func - if(var->esc == EscHeap) + // Esc pass wont run if we're inlining into a iface wrapper. + // Luckily, we can steal the results from the target func. + // If inlining a function defined in another package after + // escape analysis is done, treat all local vars as escaping. + // See issue 9537. + if(var->esc == EscHeap || (inl_nonlocal && var->op == ONAME)) addrescapes(n); curfn->dcl = list(curfn->dcl, n); diff --git a/src/cmd/gc/subr.c b/src/cmd/gc/subr.c index a8acae1fd27..d46e44d3134 100644 --- a/src/cmd/gc/subr.c +++ b/src/cmd/gc/subr.c @@ -2617,7 +2617,16 @@ genwrapper(Type *rcvr, Type *method, Sym *newnam, int iface) fn->dupok = 1; typecheck(&fn, Etop); typechecklist(fn->nbody, Etop); + + // Set inl_nonlocal to whether we are calling a method on a + // type defined in a different package. Checked in inlvar. + if(!methodrcvr->local) + inl_nonlocal = 1; + inlcalls(fn); + + inl_nonlocal = 0; + curfn = nil; funccompile(fn, 0); } diff --git a/test/fixedbugs/issue9537.dir/a.go b/test/fixedbugs/issue9537.dir/a.go new file mode 100644 index 00000000000..818c9eb4ab8 --- /dev/null +++ b/test/fixedbugs/issue9537.dir/a.go @@ -0,0 +1,25 @@ +// Copyright 2015 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +package a + +type X struct { + T [32]byte +} + +func (x *X) Get() []byte { + t := x.T + return t[:] +} + +func (x *X) RetPtr(i int) *int { + i++ + return &i +} + +func (x *X) RetRPtr(i int) (r1 int, r2 *int) { + r1 = i + 1 + r2 = &r1 + return +} diff --git a/test/fixedbugs/issue9537.dir/b.go b/test/fixedbugs/issue9537.dir/b.go new file mode 100644 index 00000000000..52e64c81f1f --- /dev/null +++ b/test/fixedbugs/issue9537.dir/b.go @@ -0,0 +1,43 @@ +// Copyright 2015 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +package main + +import ( + "bytes" + + "./a" +) + +type X struct { + *a.X +} + +type Intf interface { + Get() []byte + RetPtr(int) *int + RetRPtr(int) (int, *int) +} + +func main() { + x := &a.X{T: [32]byte{1, 2, 3, 4}} + var ix Intf = X{x} + t1 := ix.Get() + t2 := x.Get() + if !bytes.Equal(t1, t2) { + panic(t1) + } + + p1 := ix.RetPtr(5) + p2 := x.RetPtr(7) + if *p1 != 6 || *p2 != 8 { + panic(*p1) + } + + r1, r2 := ix.RetRPtr(10) + r3, r4 := x.RetRPtr(13) + if r1 != 11 || *r2 != 11 || r3 != 14 || *r4 != 14 { + panic("bad RetRPtr") + } +} diff --git a/test/fixedbugs/issue9537.go b/test/fixedbugs/issue9537.go new file mode 100644 index 00000000000..ac2d41b1234 --- /dev/null +++ b/test/fixedbugs/issue9537.go @@ -0,0 +1,10 @@ +// rundir + +// Copyright 2015 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +// Issue 9537: Compiler does not run escape analysis on an inlined +// generated method wrapper. + +package ignored