1
0
mirror of https://github.com/golang/go synced 2024-11-21 20:44:39 -07:00

go/printer: fix a couple of hidden crashes that become

visible only when enabling internal debug mode:
- in rare cases expression depth can underflow
- when printing a single labeled statement, indentation
  may underflow if not setup correctly

R=rsc
CC=golang-dev
https://golang.org/cl/484041
This commit is contained in:
Robert Griesemer 2010-03-12 14:01:52 -08:00
parent f7c27b9af9
commit ee0c35be85
5 changed files with 20 additions and 3 deletions

View File

@ -562,6 +562,15 @@ func diffPrec(expr ast.Expr, prec int) int {
}
func reduceDepth(depth int) int {
depth--
if depth < 1 {
depth = 1
}
return depth
}
// Format the binary expression: decide the cutoff and then format.
// Let's call depth == 1 Normal mode, and depth > 1 Compact mode.
// (Algorithm suggestion by Russ Cox.)
@ -604,7 +613,7 @@ func (p *printer) binaryExpr(x *ast.BinaryExpr, prec1, cutoff, depth int, multiL
// Note: The parser inserts an ast.ParenExpr node; thus this case
// can only occur if the AST is created in a different way.
p.print(token.LPAREN)
p.expr0(x, depth-1, multiLine) // parentheses undo one level of depth
p.expr0(x, reduceDepth(depth), multiLine) // parentheses undo one level of depth
p.print(token.RPAREN)
return
}
@ -707,7 +716,7 @@ func (p *printer) expr1(expr ast.Expr, prec1, depth int, ctxt exprContext, multi
case *ast.ParenExpr:
p.print(token.LPAREN)
p.expr0(x.X, depth-1, multiLine) // parentheses undo one level of depth
p.expr0(x.X, reduceDepth(depth), multiLine) // parentheses undo one level of depth
p.print(x.Rparen, token.RPAREN)
case *ast.SelectorExpr:
@ -925,7 +934,7 @@ func (p *printer) stmt(stmt ast.Stmt, multiLine *bool) {
case *ast.LabeledStmt:
// a "correcting" unindent immediately following a line break
// is applied before the line break if there is no comment
// is applied before the line break if there is no comment
// between (see writeWhitespace)
p.print(unindent)
p.expr(s.Label, multiLine)

View File

@ -1002,6 +1002,11 @@ func (cfg *Config) Fprint(output io.Writer, node interface{}) (int, os.Error) {
p.expr(n, ignoreMultiLine)
case ast.Stmt:
p.useNodeComments = true
// A labeled statement will un-indent to position the
// label. Set indent to 1 so we don't get indent "underflow".
if _, labeledStmt := n.(*ast.LabeledStmt); labeledStmt {
p.indent = 1
}
p.stmt(n, ignoreMultiLine)
case ast.Decl:
p.useNodeComments = true

View File

@ -20,6 +20,7 @@ var (
func _() {
// no spaces around simple or parenthesized expressions
_ = (a + 0)
_ = a + b
_ = a + b + c
_ = a + b - c

View File

@ -20,6 +20,7 @@ var (
func _() {
// no spaces around simple or parenthesized expressions
_ = (a+0)
_ = a+b
_ = a+b+c
_ = a+b-c

View File

@ -20,6 +20,7 @@ var (
func _() {
// no spaces around simple or parenthesized expressions
_ = (a + 0)
_ = a + b
_ = a + b + c
_ = a + b - c