+++ /dev/null
-// Copyright 2013 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 ir
-
-// This file defines algorithms related to dominance.
-
-// Dominator tree construction ----------------------------------------
-//
-// We use the algorithm described in Lengauer & Tarjan. 1979. A fast
-// algorithm for finding dominators in a flowgraph.
-// http://doi.acm.org/10.1145/357062.357071
-//
-// We also apply the optimizations to SLT described in Georgiadis et
-// al, Finding Dominators in Practice, JGAA 2006,
-// http://jgaa.info/accepted/2006/GeorgiadisTarjanWerneck2006.10.1.pdf
-// to avoid the need for buckets of size > 1.
-
-import (
- "bytes"
- "fmt"
- "io"
- "math/big"
- "os"
- "sort"
-)
-
-// Idom returns the block that immediately dominates b:
-// its parent in the dominator tree, if any.
-// The entry node (b.Index==0) does not have a parent.
-//
-func (b *BasicBlock) Idom() *BasicBlock { return b.dom.idom }
-
-// Dominees returns the list of blocks that b immediately dominates:
-// its children in the dominator tree.
-//
-func (b *BasicBlock) Dominees() []*BasicBlock { return b.dom.children }
-
-// Dominates reports whether b dominates c.
-func (b *BasicBlock) Dominates(c *BasicBlock) bool {
- return b.dom.pre <= c.dom.pre && c.dom.post <= b.dom.post
-}
-
-type byDomPreorder []*BasicBlock
-
-func (a byDomPreorder) Len() int { return len(a) }
-func (a byDomPreorder) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
-func (a byDomPreorder) Less(i, j int) bool { return a[i].dom.pre < a[j].dom.pre }
-
-// DomPreorder returns a new slice containing the blocks of f in
-// dominator tree preorder.
-//
-func (f *Function) DomPreorder() []*BasicBlock {
- n := len(f.Blocks)
- order := make(byDomPreorder, n)
- copy(order, f.Blocks)
- sort.Sort(order)
- return order
-}
-
-// domInfo contains a BasicBlock's dominance information.
-type domInfo struct {
- idom *BasicBlock // immediate dominator (parent in domtree)
- children []*BasicBlock // nodes immediately dominated by this one
- pre, post int32 // pre- and post-order numbering within domtree
-}
-
-// buildDomTree computes the dominator tree of f using the LT algorithm.
-// Precondition: all blocks are reachable (e.g. optimizeBlocks has been run).
-//
-func buildDomTree(fn *Function) {
- // The step numbers refer to the original LT paper; the
- // reordering is due to Georgiadis.
-
- // Clear any previous domInfo.
- for _, b := range fn.Blocks {
- b.dom = domInfo{}
- }
-
- idoms := make([]*BasicBlock, len(fn.Blocks))
-
- order := make([]*BasicBlock, 0, len(fn.Blocks))
- seen := fn.blockset(0)
- var dfs func(b *BasicBlock)
- dfs = func(b *BasicBlock) {
- if !seen.Add(b) {
- return
- }
- for _, succ := range b.Succs {
- dfs(succ)
- }
- if fn.fakeExits.Has(b) {
- dfs(fn.Exit)
- }
- order = append(order, b)
- b.post = len(order) - 1
- }
- dfs(fn.Blocks[0])
-
- for i := 0; i < len(order)/2; i++ {
- o := len(order) - i - 1
- order[i], order[o] = order[o], order[i]
- }
-
- idoms[fn.Blocks[0].Index] = fn.Blocks[0]
- changed := true
- for changed {
- changed = false
- // iterate over all nodes in reverse postorder, except for the
- // entry node
- for _, b := range order[1:] {
- var newIdom *BasicBlock
- do := func(p *BasicBlock) {
- if idoms[p.Index] == nil {
- return
- }
- if newIdom == nil {
- newIdom = p
- } else {
- finger1 := p
- finger2 := newIdom
- for finger1 != finger2 {
- for finger1.post < finger2.post {
- finger1 = idoms[finger1.Index]
- }
- for finger2.post < finger1.post {
- finger2 = idoms[finger2.Index]
- }
- }
- newIdom = finger1
- }
- }
- for _, p := range b.Preds {
- do(p)
- }
- if b == fn.Exit {
- for _, p := range fn.Blocks {
- if fn.fakeExits.Has(p) {
- do(p)
- }
- }
- }
-
- if idoms[b.Index] != newIdom {
- idoms[b.Index] = newIdom
- changed = true
- }
- }
- }
-
- for i, b := range idoms {
- fn.Blocks[i].dom.idom = b
- if b == nil {
- // malformed CFG
- continue
- }
- if i == b.Index {
- continue
- }
- b.dom.children = append(b.dom.children, fn.Blocks[i])
- }
-
- numberDomTree(fn.Blocks[0], 0, 0)
-
- // printDomTreeDot(os.Stderr, fn) // debugging
- // printDomTreeText(os.Stderr, root, 0) // debugging
-
- if fn.Prog.mode&SanityCheckFunctions != 0 {
- sanityCheckDomTree(fn)
- }
-}
-
-// buildPostDomTree is like buildDomTree, but builds the post-dominator tree instead.
-func buildPostDomTree(fn *Function) {
- // The step numbers refer to the original LT paper; the
- // reordering is due to Georgiadis.
-
- // Clear any previous domInfo.
- for _, b := range fn.Blocks {
- b.pdom = domInfo{}
- }
-
- idoms := make([]*BasicBlock, len(fn.Blocks))
-
- order := make([]*BasicBlock, 0, len(fn.Blocks))
- seen := fn.blockset(0)
- var dfs func(b *BasicBlock)
- dfs = func(b *BasicBlock) {
- if !seen.Add(b) {
- return
- }
- for _, pred := range b.Preds {
- dfs(pred)
- }
- if b == fn.Exit {
- for _, p := range fn.Blocks {
- if fn.fakeExits.Has(p) {
- dfs(p)
- }
- }
- }
- order = append(order, b)
- b.post = len(order) - 1
- }
- dfs(fn.Exit)
-
- for i := 0; i < len(order)/2; i++ {
- o := len(order) - i - 1
- order[i], order[o] = order[o], order[i]
- }
-
- idoms[fn.Exit.Index] = fn.Exit
- changed := true
- for changed {
- changed = false
- // iterate over all nodes in reverse postorder, except for the
- // exit node
- for _, b := range order[1:] {
- var newIdom *BasicBlock
- do := func(p *BasicBlock) {
- if idoms[p.Index] == nil {
- return
- }
- if newIdom == nil {
- newIdom = p
- } else {
- finger1 := p
- finger2 := newIdom
- for finger1 != finger2 {
- for finger1.post < finger2.post {
- finger1 = idoms[finger1.Index]
- }
- for finger2.post < finger1.post {
- finger2 = idoms[finger2.Index]
- }
- }
- newIdom = finger1
- }
- }
- for _, p := range b.Succs {
- do(p)
- }
- if fn.fakeExits.Has(b) {
- do(fn.Exit)
- }
-
- if idoms[b.Index] != newIdom {
- idoms[b.Index] = newIdom
- changed = true
- }
- }
- }
-
- for i, b := range idoms {
- fn.Blocks[i].pdom.idom = b
- if b == nil {
- // malformed CFG
- continue
- }
- if i == b.Index {
- continue
- }
- b.pdom.children = append(b.pdom.children, fn.Blocks[i])
- }
-
- numberPostDomTree(fn.Exit, 0, 0)
-
- // printPostDomTreeDot(os.Stderr, fn) // debugging
- // printPostDomTreeText(os.Stderr, fn.Exit, 0) // debugging
-
- if fn.Prog.mode&SanityCheckFunctions != 0 { // XXX
- sanityCheckDomTree(fn) // XXX
- }
-}
-
-// numberDomTree sets the pre- and post-order numbers of a depth-first
-// traversal of the dominator tree rooted at v. These are used to
-// answer dominance queries in constant time.
-//
-func numberDomTree(v *BasicBlock, pre, post int32) (int32, int32) {
- v.dom.pre = pre
- pre++
- for _, child := range v.dom.children {
- pre, post = numberDomTree(child, pre, post)
- }
- v.dom.post = post
- post++
- return pre, post
-}
-
-// numberPostDomTree sets the pre- and post-order numbers of a depth-first
-// traversal of the post-dominator tree rooted at v. These are used to
-// answer post-dominance queries in constant time.
-//
-func numberPostDomTree(v *BasicBlock, pre, post int32) (int32, int32) {
- v.pdom.pre = pre
- pre++
- for _, child := range v.pdom.children {
- pre, post = numberPostDomTree(child, pre, post)
- }
- v.pdom.post = post
- post++
- return pre, post
-}
-
-// Testing utilities ----------------------------------------
-
-// sanityCheckDomTree checks the correctness of the dominator tree
-// computed by the LT algorithm by comparing against the dominance
-// relation computed by a naive Kildall-style forward dataflow
-// analysis (Algorithm 10.16 from the "Dragon" book).
-//
-func sanityCheckDomTree(f *Function) {
- n := len(f.Blocks)
-
- // D[i] is the set of blocks that dominate f.Blocks[i],
- // represented as a bit-set of block indices.
- D := make([]big.Int, n)
-
- one := big.NewInt(1)
-
- // all is the set of all blocks; constant.
- var all big.Int
- all.Set(one).Lsh(&all, uint(n)).Sub(&all, one)
-
- // Initialization.
- for i := range f.Blocks {
- if i == 0 {
- // A root is dominated only by itself.
- D[i].SetBit(&D[0], 0, 1)
- } else {
- // All other blocks are (initially) dominated
- // by every block.
- D[i].Set(&all)
- }
- }
-
- // Iteration until fixed point.
- for changed := true; changed; {
- changed = false
- for i, b := range f.Blocks {
- if i == 0 {
- continue
- }
- // Compute intersection across predecessors.
- var x big.Int
- x.Set(&all)
- for _, pred := range b.Preds {
- x.And(&x, &D[pred.Index])
- }
- if b == f.Exit {
- for _, p := range f.Blocks {
- if f.fakeExits.Has(p) {
- x.And(&x, &D[p.Index])
- }
- }
- }
- x.SetBit(&x, i, 1) // a block always dominates itself.
- if D[i].Cmp(&x) != 0 {
- D[i].Set(&x)
- changed = true
- }
- }
- }
-
- // Check the entire relation. O(n^2).
- ok := true
- for i := 0; i < n; i++ {
- for j := 0; j < n; j++ {
- b, c := f.Blocks[i], f.Blocks[j]
- actual := b.Dominates(c)
- expected := D[j].Bit(i) == 1
- if actual != expected {
- fmt.Fprintf(os.Stderr, "dominates(%s, %s)==%t, want %t\n", b, c, actual, expected)
- ok = false
- }
- }
- }
-
- preorder := f.DomPreorder()
- for _, b := range f.Blocks {
- if got := preorder[b.dom.pre]; got != b {
- fmt.Fprintf(os.Stderr, "preorder[%d]==%s, want %s\n", b.dom.pre, got, b)
- ok = false
- }
- }
-
- if !ok {
- panic("sanityCheckDomTree failed for " + f.String())
- }
-
-}
-
-// Printing functions ----------------------------------------
-
-// printDomTree prints the dominator tree as text, using indentation.
-//lint:ignore U1000 used during debugging
-func printDomTreeText(buf *bytes.Buffer, v *BasicBlock, indent int) {
- fmt.Fprintf(buf, "%*s%s\n", 4*indent, "", v)
- for _, child := range v.dom.children {
- printDomTreeText(buf, child, indent+1)
- }
-}
-
-// printDomTreeDot prints the dominator tree of f in AT&T GraphViz
-// (.dot) format.
-//lint:ignore U1000 used during debugging
-func printDomTreeDot(buf io.Writer, f *Function) {
- fmt.Fprintln(buf, "//", f)
- fmt.Fprintln(buf, "digraph domtree {")
- for i, b := range f.Blocks {
- v := b.dom
- fmt.Fprintf(buf, "\tn%d [label=\"%s (%d, %d)\",shape=\"rectangle\"];\n", v.pre, b, v.pre, v.post)
- // TODO(adonovan): improve appearance of edges
- // belonging to both dominator tree and CFG.
-
- // Dominator tree edge.
- if i != 0 {
- fmt.Fprintf(buf, "\tn%d -> n%d [style=\"solid\",weight=100];\n", v.idom.dom.pre, v.pre)
- }
- // CFG edges.
- for _, pred := range b.Preds {
- fmt.Fprintf(buf, "\tn%d -> n%d [style=\"dotted\",weight=0];\n", pred.dom.pre, v.pre)
- }
- }
- fmt.Fprintln(buf, "}")
-}
-
-// printDomTree prints the dominator tree as text, using indentation.
-//lint:ignore U1000 used during debugging
-func printPostDomTreeText(buf io.Writer, v *BasicBlock, indent int) {
- fmt.Fprintf(buf, "%*s%s\n", 4*indent, "", v)
- for _, child := range v.pdom.children {
- printPostDomTreeText(buf, child, indent+1)
- }
-}
-
-// printDomTreeDot prints the dominator tree of f in AT&T GraphViz
-// (.dot) format.
-//lint:ignore U1000 used during debugging
-func printPostDomTreeDot(buf io.Writer, f *Function) {
- fmt.Fprintln(buf, "//", f)
- fmt.Fprintln(buf, "digraph pdomtree {")
- for _, b := range f.Blocks {
- v := b.pdom
- fmt.Fprintf(buf, "\tn%d [label=\"%s (%d, %d)\",shape=\"rectangle\"];\n", v.pre, b, v.pre, v.post)
- // TODO(adonovan): improve appearance of edges
- // belonging to both dominator tree and CFG.
-
- // Dominator tree edge.
- if b != f.Exit {
- fmt.Fprintf(buf, "\tn%d -> n%d [style=\"solid\",weight=100];\n", v.idom.pdom.pre, v.pre)
- }
- // CFG edges.
- for _, pred := range b.Preds {
- fmt.Fprintf(buf, "\tn%d -> n%d [style=\"dotted\",weight=0];\n", pred.pdom.pre, v.pre)
- }
- }
- fmt.Fprintln(buf, "}")
-}