diff --git a/src/cmd/compile/internal/gc/escape.go b/src/cmd/compile/internal/gc/escape.go index e25c79998cdbf9..66440674d999a7 100644 --- a/src/cmd/compile/internal/gc/escape.go +++ b/src/cmd/compile/internal/gc/escape.go @@ -471,8 +471,8 @@ func (e *Escape) exprSkipInit(k EscHole, n *Node) { e.discard(max) case OCONV, OCONVNOP: - if checkPtr(e.curfn) && n.Type.Etype == TUNSAFEPTR && n.Left.Type.IsPtr() { - // When -d=checkptr is enabled, treat + if checkPtr(e.curfn, 2) && n.Type.Etype == TUNSAFEPTR && n.Left.Type.IsPtr() { + // When -d=checkptr=2 is enabled, treat // conversions to unsafe.Pointer as an // escaping operation. This allows better // runtime instrumentation, since we can more diff --git a/src/cmd/compile/internal/gc/main.go b/src/cmd/compile/internal/gc/main.go index e7131f10a2e6a6..771b4fe973f225 100644 --- a/src/cmd/compile/internal/gc/main.go +++ b/src/cmd/compile/internal/gc/main.go @@ -94,6 +94,11 @@ const debugHelpHeader = `usage: -d arg[,arg]* and arg is [=] const debugHelpFooter = ` is key-specific. +Key "checkptr" supports values: + "0": instrumentation disabled + "1": conversions involving unsafe.Pointer are instrumented + "2": conversions to unsafe.Pointer force heap allocation + Key "pctab" supports values: "pctospadj", "pctofile", "pctoline", "pctoinline", "pctopcdata" ` diff --git a/src/cmd/compile/internal/gc/walk.go b/src/cmd/compile/internal/gc/walk.go index ebae392808bb01..4f5fa38a334cf8 100644 --- a/src/cmd/compile/internal/gc/walk.go +++ b/src/cmd/compile/internal/gc/walk.go @@ -951,7 +951,7 @@ opswitch: case OCONV, OCONVNOP: n.Left = walkexpr(n.Left, init) - if n.Op == OCONVNOP && checkPtr(Curfn) { + if n.Op == OCONVNOP && checkPtr(Curfn, 1) { if n.Type.IsPtr() && n.Left.Type.Etype == TUNSAFEPTR { // unsafe.Pointer to *T n = walkCheckPtrAlignment(n, init) break @@ -3976,7 +3976,8 @@ func walkCheckPtrArithmetic(n *Node, init *Nodes) *Node { } // checkPtr reports whether pointer checking should be enabled for -// function fn. -func checkPtr(fn *Node) bool { - return Debug_checkptr != 0 && fn.Func.Pragma&NoCheckPtr == 0 +// function fn at a given level. See debugHelpFooter for defined +// levels. +func checkPtr(fn *Node, level int) bool { + return Debug_checkptr >= level && fn.Func.Pragma&NoCheckPtr == 0 }