package pointer // This file implements Hash-Value Numbering (HVN), a pre-solver // constraint optimization described in Hardekopf & Lin, SAS'07 (see // doc.go) that analyses the graph topology to determine which sets of // variables are "pointer equivalent" (PE), i.e. must have identical // points-to sets in the solution. // // A separate ("offline") graph is constructed. Its nodes are those of // the main-graph, plus an additional node *X for each pointer node X. // With this graph we can reason about the unknown points-to set of // dereferenced pointers. (We do not generalize this to represent // unknown fields x->f, perhaps because such fields would be numerous, // though it might be worth an experiment.) // // Nodes whose points-to relations are not entirely captured by the // graph are marked as "indirect": the *X nodes, the parameters of // address-taken functions (which includes all functions in method // sets), or nodes updated by the solver rules for reflection, etc. // // All addr (y=&x) nodes are initially assigned a pointer-equivalence // (PE) label equal to x's nodeid in the main graph. (These are the // only PE labels that are less than len(a.nodes).) // // All offsetAddr (y=&x.f) constraints are initially assigned a PE // label; such labels are memoized, keyed by (x, f), so that equivalent // nodes y as assigned the same label. // // Then we process each strongly connected component (SCC) of the graph // in topological order, assigning it a PE label based on the set P of // PE labels that flow to it from its immediate dependencies. // // If any node in P is "indirect", the entire SCC is assigned a fresh PE // label. Otherwise: // // |P|=0 if P is empty, all nodes in the SCC are non-pointers (e.g. // uninitialized variables, or formal params of dead functions) // and the SCC is assigned the PE label of zero. // // |P|=1 if P is a singleton, the SCC is assigned the same label as the // sole element of P. // // |P|>1 if P contains multiple labels, a unique label representing P is // invented and recorded in an hash table, so that other // equivalent SCCs may also be assigned this label, akin to // conventional hash-value numbering in a compiler. // // Finally, a renumbering is computed such that each node is replaced by // the lowest-numbered node with the same PE label. All constraints are // renumbered, and any resulting duplicates are eliminated. // // The only nodes that are not renumbered are the objects x in addr // (y=&x) constraints, since the ids of these nodes (and fields derived // from them via offsetAddr rules) are the elements of all points-to // sets, so they must remain as they are if we want the same solution. // // The solverStates (node.solve) for nodes in the same equivalence class // are linked together so that all nodes in the class have the same // solution. This avoids the need to renumber nodeids buried in // Queries, cgnodes, etc (like (*analysis).renumber() does) since only // the solution is needed. // // The result of HVN is that the number of distinct nodes and // constraints is reduced, but the solution is identical (almost---see // CROSS-CHECK below). In particular, both linear and cyclic chains of // copies are each replaced by a single node. // // Nodes and constraints created "online" (e.g. while solving reflection // constraints) are not subject to this optimization. // // PERFORMANCE // // In two benchmarks (oracle and godoc), HVN eliminates about two thirds // of nodes, the majority accounted for by non-pointers: nodes of // non-pointer type, pointers that remain nil, formal parameters of dead // functions, nodes of untracked types, etc. It also reduces the number // of constraints, also by about two thirds, and the solving time by // 30--42%, although we must pay about 15% for the running time of HVN // itself. The benefit is greater for larger applications. // // There are many possible optimizations to improve the performance: // * Use fewer than 1:1 onodes to main graph nodes: many of the onodes // we create are not needed. // * HU (HVN with Union---see paper): coalesce "union" peLabels when // their expanded-out sets are equal. // * HR (HVN with deReference---see paper): this will require that we // apply HVN until fixed point, which may need more bookkeeping of the // correspondance of main nodes to onodes. // * Location Equivalence (see paper): have points-to sets contain not // locations but location-equivalence class labels, each representing // a set of locations. // * HVN with field-sensitive ref: model each of the fields of a // pointer-to-struct. // // CROSS-CHECK // // To verify the soundness of the optimization, when the // debugHVNCrossCheck option is enabled, we run the solver twice, once // before and once after running HVN, dumping the solution to disk, and // then we compare the results. If they are not identical, the analysis // panics. // // The solution dumped to disk includes only the N*N submatrix of the // complete solution where N is the number of nodes after generation. // In other words, we ignore pointer variables and objects created by // the solver itself, since their numbering depends on the solver order, // which is affected by the optimization. In any case, that's the only // part the client cares about. // // The cross-check is too strict and may fail spuriously. Although the // H&L paper describing HVN states that the solutions obtained should be // identical, this is not the case in practice because HVN can collapse // cycles involving *p even when pts(p)={}. Consider this example // distilled from testdata/hello.go: // // var x T // func f(p **T) { // t0 = *p // ... // t1 = φ(t0, &x) // *p = t1 // } // // If f is dead code, we get: // unoptimized: pts(p)={} pts(t0)={} pts(t1)={&x} // optimized: pts(p)={} pts(t0)=pts(t1)=pts(*p)={&x} // // It's hard to argue that this is a bug: the result is sound and the // loss of precision is inconsequential---f is dead code, after all. // But unfortunately it limits the usefulness of the cross-check since // failures must be carefully analyzed. Ben Hardekopf suggests (in // personal correspondence) some approaches to mitigating it: // // If there is a node with an HVN points-to set that is a superset // of the NORM points-to set, then either it's a bug or it's a // result of this issue. If it's a result of this issue, then in // the offline constraint graph there should be a REF node inside // some cycle that reaches this node, and in the NORM solution the // pointer being dereferenced by that REF node should be the empty // set. If that isn't true then this is a bug. If it is true, then // you can further check that in the NORM solution the "extra" // points-to info in the HVN solution does in fact come from that // purported cycle (if it doesn't, then this is still a bug). If // you're doing the further check then you'll need to do it for // each "extra" points-to element in the HVN points-to set. // // There are probably ways to optimize these checks by taking // advantage of graph properties. For example, extraneous points-to // info will flow through the graph and end up in many // nodes. Rather than checking every node with extra info, you // could probably work out the "origin point" of the extra info and // just check there. Note that the check in the first bullet is // looking for soundness bugs, while the check in the second bullet // is looking for precision bugs; depending on your needs, you may // care more about one than the other. // // which we should evaluate. The cross-check is nonetheless invaluable // for all but one of the programs in the pointer_test suite. import ( "fmt" "io" "reflect" "golang.org/x/tools/container/intsets" "golang.org/x/tools/go/types" ) // A peLabel is a pointer-equivalence label: two nodes with the same // peLabel have identical points-to solutions. // // The numbers are allocated consecutively like so: // 0 not a pointer // 1..N-1 addrConstraints (equals the constraint's .src field, hence sparse) // ... offsetAddr constraints // ... SCCs (with indirect nodes or multiple inputs) // // Each PE label denotes a set of pointers containing a single addr, a // single offsetAddr, or some set of other PE labels. // type peLabel int type hvn struct { a *analysis N int // len(a.nodes) immediately after constraint generation log io.Writer // (optional) log of HVN lemmas onodes []*onode // nodes of the offline graph label peLabel // the next available PE label hvnLabel map[string]peLabel // hash-value numbering (PE label) for each set of onodeids stack []onodeid // DFS stack index int32 // next onode.index, from Tarjan's SCC algorithm // For each distinct offsetAddrConstraint (src, offset) pair, // offsetAddrLabels records a unique PE label >= N. offsetAddrLabels map[offsetAddr]peLabel } // The index of an node in the offline graph. // (Currently the first N align with the main nodes, // but this may change with HRU.) type onodeid uint32 // An onode is a node in the offline constraint graph. // (Where ambiguous, members of analysis.nodes are referred to as // "main graph" nodes.) // // Edges in the offline constraint graph (edges and implicit) point to // the source, i.e. against the flow of values: they are dependencies. // Implicit edges are used for SCC computation, but not for gathering // incoming labels. // type onode struct { rep onodeid // index of representative of SCC in offline constraint graph edges intsets.Sparse // constraint edges X-->Y (this onode is X) implicit intsets.Sparse // implicit edges *X-->*Y (this onode is X) peLabels intsets.Sparse // set of peLabels are pointer-equivalent to this one indirect bool // node has points-to relations not represented in graph // Tarjan's SCC algorithm index, lowlink int32 // Tarjan numbering scc int32 // -ve => on stack; 0 => unvisited; +ve => node is root of a found SCC } type offsetAddr struct { ptr nodeid offset uint32 } // nextLabel issues the next unused pointer-equivalence label. func (h *hvn) nextLabel() peLabel { h.label++ return h.label } // ref(X) returns the index of the onode for *X. func (h *hvn) ref(id onodeid) onodeid { return id + onodeid(len(h.a.nodes)) } // hvn computes pointer-equivalence labels (peLabels) using the Hash-based // Value Numbering (HVN) algorithm described in Hardekopf & Lin, SAS'07. // func (a *analysis) hvn() { start("HVN") if a.log != nil { fmt.Fprintf(a.log, "\n\n==== Pointer equivalence optimization\n\n") } h := hvn{ a: a, N: len(a.nodes), log: a.log, hvnLabel: make(map[string]peLabel), offsetAddrLabels: make(map[offsetAddr]peLabel), } if h.log != nil { fmt.Fprintf(h.log, "\nCreating offline graph nodes...\n") } // Create offline nodes. The first N nodes correspond to main // graph nodes; the next N are their corresponding ref() nodes. h.onodes = make([]*onode, 2*h.N) for id := range a.nodes { id := onodeid(id) h.onodes[id] = &onode{} h.onodes[h.ref(id)] = &onode{indirect: true} } // Each node initially represents just itself. for id, o := range h.onodes { o.rep = onodeid(id) } h.markIndirectNodes() // Reserve the first N PE labels for addrConstraints. h.label = peLabel(h.N) // Add offline constraint edges. if h.log != nil { fmt.Fprintf(h.log, "\nAdding offline graph edges...\n") } for _, c := range a.constraints { if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "; %s\n", c) } c.presolve(&h) } // Find and collapse SCCs. if h.log != nil { fmt.Fprintf(h.log, "\nFinding SCCs...\n") } h.index = 1 for id, o := range h.onodes { if id > 0 && o.index == 0 { // Start depth-first search at each unvisited node. h.visit(onodeid(id)) } } // Dump the solution // (NB: somewhat redundant with logging from simplify().) if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\nPointer equivalences:\n") for id, o := range h.onodes { if id == 0 { continue } if id == int(h.N) { fmt.Fprintf(h.log, "---\n") } fmt.Fprintf(h.log, "o%d\t", id) if o.rep != onodeid(id) { fmt.Fprintf(h.log, "rep=o%d", o.rep) } else { fmt.Fprintf(h.log, "p%d", o.peLabels.Min()) if o.indirect { fmt.Fprint(h.log, " indirect") } } fmt.Fprintln(h.log) } } // Simplify the main constraint graph h.simplify() a.showCounts() stop("HVN") } // ---- constraint-specific rules ---- // dst := &src func (c *addrConstraint) presolve(h *hvn) { // Each object (src) is an initial PE label. label := peLabel(c.src) // label < N if debugHVNVerbose && h.log != nil { // duplicate log messages are possible fmt.Fprintf(h.log, "\tcreate p%d: {&n%d}\n", label, c.src) } odst := onodeid(c.dst) osrc := onodeid(c.src) // Assign dst this label. h.onodes[odst].peLabels.Insert(int(label)) if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d has p%d\n", odst, label) } h.addImplicitEdge(h.ref(odst), osrc) // *dst ~~> src. } // dst = src func (c *copyConstraint) presolve(h *hvn) { odst := onodeid(c.dst) osrc := onodeid(c.src) h.addEdge(odst, osrc) // dst --> src h.addImplicitEdge(h.ref(odst), h.ref(osrc)) // *dst ~~> *src } // dst = *src + offset func (c *loadConstraint) presolve(h *hvn) { odst := onodeid(c.dst) osrc := onodeid(c.src) if c.offset == 0 { h.addEdge(odst, h.ref(osrc)) // dst --> *src } else { // We don't interpret load-with-offset, e.g. results // of map value lookup, R-block of dynamic call, slice // copy/append, reflection. h.markIndirect(odst, "load with offset") } } // *dst + offset = src func (c *storeConstraint) presolve(h *hvn) { odst := onodeid(c.dst) osrc := onodeid(c.src) if c.offset == 0 { h.onodes[h.ref(odst)].edges.Insert(int(osrc)) // *dst --> src if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d --> o%d\n", h.ref(odst), osrc) } } else { // We don't interpret store-with-offset. // See discussion of soundness at markIndirectNodes. } } // dst = &src.offset func (c *offsetAddrConstraint) presolve(h *hvn) { // Give each distinct (addr, offset) pair a fresh PE label. // The cache performs CSE, effectively. key := offsetAddr{c.src, c.offset} label, ok := h.offsetAddrLabels[key] if !ok { label = h.nextLabel() h.offsetAddrLabels[key] = label if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\tcreate p%d: {&n%d.#%d}\n", label, c.src, c.offset) } } // Assign dst this label. h.onodes[c.dst].peLabels.Insert(int(label)) if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d has p%d\n", c.dst, label) } } // dst = src.(typ) where typ is an interface func (c *typeFilterConstraint) presolve(h *hvn) { h.markIndirect(onodeid(c.dst), "typeFilter result") } // dst = src.(typ) where typ is concrete func (c *untagConstraint) presolve(h *hvn) { odst := onodeid(c.dst) for end := odst + onodeid(h.a.sizeof(c.typ)); odst < end; odst++ { h.markIndirect(odst, "untag result") } } // dst = src.method(c.params...) func (c *invokeConstraint) presolve(h *hvn) { // All methods are address-taken functions, so // their formal P-blocks were already marked indirect. // Mark the caller's targets node as indirect. sig := c.method.Type().(*types.Signature) id := c.params h.markIndirect(onodeid(c.params), "invoke targets node") id++ id += nodeid(h.a.sizeof(sig.Params())) // Mark the caller's R-block as indirect. end := id + nodeid(h.a.sizeof(sig.Results())) for id < end { h.markIndirect(onodeid(id), "invoke R-block") id++ } } // markIndirectNodes marks as indirect nodes whose points-to relations // are not entirely captured by the offline graph, including: // // (a) All address-taken nodes (including the following nodes within // the same object). This is described in the paper. // // The most subtle cause of indirect nodes is the generation of // store-with-offset constraints since the offline graph doesn't // represent them. A global audit of constraint generation reveals the // following uses of store-with-offset: // // (b) genDynamicCall, for P-blocks of dynamically called functions, // to which dynamic copy edges will be added to them during // solving: from storeConstraint for standalone functions, // and from invokeConstraint for methods. // All such P-blocks must be marked indirect. // (c) MakeUpdate, to update the value part of a map object. // All MakeMap objects's value parts must be marked indirect. // (d) copyElems, to update the destination array. // All array elements must be marked indirect. // // Not all indirect marking happens here. ref() nodes are marked // indirect at construction, and each constraint's presolve() method may // mark additional nodes. // func (h *hvn) markIndirectNodes() { // (a) all address-taken nodes, plus all nodes following them // within the same object, since these may be indirectly // stored or address-taken. for _, c := range h.a.constraints { if c, ok := c.(*addrConstraint); ok { start := h.a.enclosingObj(c.src) end := start + nodeid(h.a.nodes[start].obj.size) for id := c.src; id < end; id++ { h.markIndirect(onodeid(id), "A-T object") } } } // (b) P-blocks of all address-taken functions. for id := 0; id < h.N; id++ { obj := h.a.nodes[id].obj // TODO(adonovan): opt: if obj.cgn.fn is a method and // obj.cgn is not its shared contour, this is an // "inlined" static method call. We needn't consider it // address-taken since no invokeConstraint will affect it. if obj != nil && obj.flags&otFunction != 0 && h.a.atFuncs[obj.cgn.fn] { // address-taken function if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "n%d is address-taken: %s\n", id, obj.cgn.fn) } h.markIndirect(onodeid(id), "A-T func identity") id++ sig := obj.cgn.fn.Signature psize := h.a.sizeof(sig.Params()) if sig.Recv() != nil { psize += h.a.sizeof(sig.Recv().Type()) } for end := id + int(psize); id < end; id++ { h.markIndirect(onodeid(id), "A-T func P-block") } id-- continue } } // (c) all map objects' value fields. for _, id := range h.a.mapValues { h.markIndirect(onodeid(id), "makemap.value") } // (d) all array element objects. // TODO(adonovan): opt: can we do better? for id := 0; id < h.N; id++ { // Identity node for an object of array type? if tArray, ok := h.a.nodes[id].typ.(*types.Array); ok { // Mark the array element nodes indirect. // (Skip past the identity field.) for _ = range h.a.flatten(tArray.Elem()) { id++ h.markIndirect(onodeid(id), "array elem") } } } } func (h *hvn) markIndirect(oid onodeid, comment string) { h.onodes[oid].indirect = true if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d is indirect: %s\n", oid, comment) } } // Adds an edge dst-->src. // Note the unusual convention: edges are dependency (contraflow) edges. func (h *hvn) addEdge(odst, osrc onodeid) { h.onodes[odst].edges.Insert(int(osrc)) if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d --> o%d\n", odst, osrc) } } func (h *hvn) addImplicitEdge(odst, osrc onodeid) { h.onodes[odst].implicit.Insert(int(osrc)) if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d ~~> o%d\n", odst, osrc) } } // visit implements the depth-first search of Tarjan's SCC algorithm. // Precondition: x is canonical. func (h *hvn) visit(x onodeid) { h.checkCanonical(x) xo := h.onodes[x] xo.index = h.index xo.lowlink = h.index h.index++ h.stack = append(h.stack, x) // push assert(xo.scc == 0, "node revisited") xo.scc = -1 var deps []int deps = xo.edges.AppendTo(deps) deps = xo.implicit.AppendTo(deps) for _, y := range deps { // Loop invariant: x is canonical. y := h.find(onodeid(y)) if x == y { continue // nodes already coalesced } xo := h.onodes[x] yo := h.onodes[y] switch { case yo.scc > 0: // y is already a collapsed SCC case yo.scc < 0: // y is on the stack, and thus in the current SCC. if yo.index < xo.lowlink { xo.lowlink = yo.index } default: // y is unvisited; visit it now. h.visit(y) // Note: x and y are now non-canonical. x = h.find(onodeid(x)) if yo.lowlink < xo.lowlink { xo.lowlink = yo.lowlink } } } h.checkCanonical(x) // Is x the root of an SCC? if xo.lowlink == xo.index { // Coalesce all nodes in the SCC. if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "scc o%d\n", x) } for { // Pop y from stack. i := len(h.stack) - 1 y := h.stack[i] h.stack = h.stack[:i] h.checkCanonical(x) xo := h.onodes[x] h.checkCanonical(y) yo := h.onodes[y] if xo == yo { // SCC is complete. xo.scc = 1 h.labelSCC(x) break } h.coalesce(x, y) } } } // Precondition: x is canonical. func (h *hvn) labelSCC(x onodeid) { h.checkCanonical(x) xo := h.onodes[x] xpe := &xo.peLabels // All indirect nodes get new labels. if xo.indirect { label := h.nextLabel() if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\tcreate p%d: indirect SCC\n", label) fmt.Fprintf(h.log, "\to%d has p%d\n", x, label) } // Remove pre-labeling, in case a direct pre-labeled node was // merged with an indirect one. xpe.Clear() xpe.Insert(int(label)) return } // Invariant: all peLabels sets are non-empty. // Those that are logically empty contain zero as their sole element. // No other sets contains zero. // Find all labels coming in to the coalesced SCC node. for _, y := range xo.edges.AppendTo(nil) { y := h.find(onodeid(y)) if y == x { continue // already coalesced } ype := &h.onodes[y].peLabels if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\tedge from o%d = %s\n", y, ype) } if ype.IsEmpty() { if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\tnode has no PE label\n") } } assert(!ype.IsEmpty(), "incoming node has no PE label") if ype.Has(0) { // {0} represents a non-pointer. assert(ype.Len() == 1, "PE set contains {0, ...}") } else { xpe.UnionWith(ype) } } switch xpe.Len() { case 0: // SCC has no incoming non-zero PE labels: it is a non-pointer. xpe.Insert(0) case 1: // already a singleton default: // SCC has multiple incoming non-zero PE labels. // Find the canonical label representing this set. // We use String() as a fingerprint consistent with Equals(). key := xpe.String() label, ok := h.hvnLabel[key] if !ok { label = h.nextLabel() if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\tcreate p%d: union %s\n", label, xpe.String()) } h.hvnLabel[key] = label } xpe.Clear() xpe.Insert(int(label)) } if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\to%d has p%d\n", x, xpe.Min()) } } // coalesce combines two nodes in the offline constraint graph. // Precondition: x and y are canonical. func (h *hvn) coalesce(x, y onodeid) { xo := h.onodes[x] yo := h.onodes[y] // x becomes y's canonical representative. yo.rep = x if debugHVNVerbose && h.log != nil { fmt.Fprintf(h.log, "\tcoalesce o%d into o%d\n", y, x) } // x accumulates y's edges. xo.edges.UnionWith(&yo.edges) yo.edges.Clear() // x accumulates y's implicit edges. xo.implicit.UnionWith(&yo.implicit) yo.implicit.Clear() // x accumulates y's pointer-equivalence labels. xo.peLabels.UnionWith(&yo.peLabels) yo.peLabels.Clear() // x accumulates y's indirect flag. if yo.indirect { xo.indirect = true } } // simplify computes a degenerate renumbering of nodeids from the PE // labels assigned by the hvn, and uses it to simplify the main // constraint graph, eliminating non-pointer nodes and duplicate // constraints. // func (h *hvn) simplify() { // canon maps each peLabel to its canonical main node. canon := make([]nodeid, h.label) for i := range canon { canon[i] = nodeid(h.N) // indicates "unset" } // mapping maps each main node index to the index of the canonical node. mapping := make([]nodeid, len(h.a.nodes)) for id := range h.a.nodes { id := nodeid(id) if id == 0 { canon[0] = 0 mapping[0] = 0 continue } oid := h.find(onodeid(id)) peLabels := &h.onodes[oid].peLabels assert(peLabels.Len() == 1, "PE class is not a singleton") label := peLabel(peLabels.Min()) canonId := canon[label] if canonId == nodeid(h.N) { // id becomes the representative of the PE label. canonId = id canon[label] = canonId if h.a.log != nil { fmt.Fprintf(h.a.log, "\tpts(n%d) is canonical : \t(%s)\n", id, h.a.nodes[id].typ) } } else { // Link the solver states for the two nodes. assert(h.a.nodes[canonId].solve != nil, "missing solver state") h.a.nodes[id].solve = h.a.nodes[canonId].solve if h.a.log != nil { // TODO(adonovan): debug: reorganize the log so it prints // one line: // pe y = x1, ..., xn // for each canonical y. Requires allocation. fmt.Fprintf(h.a.log, "\tpts(n%d) = pts(n%d) : %s\n", id, canonId, h.a.nodes[id].typ) } } mapping[id] = canonId } // Renumber the constraints, eliminate duplicates, and eliminate // any containing non-pointers (n0). addrs := make(map[addrConstraint]bool) copys := make(map[copyConstraint]bool) loads := make(map[loadConstraint]bool) stores := make(map[storeConstraint]bool) offsetAddrs := make(map[offsetAddrConstraint]bool) untags := make(map[untagConstraint]bool) typeFilters := make(map[typeFilterConstraint]bool) invokes := make(map[invokeConstraint]bool) nbefore := len(h.a.constraints) cc := h.a.constraints[:0] // in-situ compaction for _, c := range h.a.constraints { // Renumber. switch c := c.(type) { case *addrConstraint: // Don't renumber c.src since it is the label of // an addressable object and will appear in PT sets. c.dst = mapping[c.dst] default: c.renumber(mapping) } if c.ptr() == 0 { continue // skip: constraint attached to non-pointer } var dup bool switch c := c.(type) { case *addrConstraint: _, dup = addrs[*c] addrs[*c] = true case *copyConstraint: if c.src == c.dst { continue // skip degenerate copies } if c.src == 0 { continue // skip copy from non-pointer } _, dup = copys[*c] copys[*c] = true case *loadConstraint: if c.src == 0 { continue // skip load from non-pointer } _, dup = loads[*c] loads[*c] = true case *storeConstraint: if c.src == 0 { continue // skip store from non-pointer } _, dup = stores[*c] stores[*c] = true case *offsetAddrConstraint: if c.src == 0 { continue // skip offset from non-pointer } _, dup = offsetAddrs[*c] offsetAddrs[*c] = true case *untagConstraint: if c.src == 0 { continue // skip untag of non-pointer } _, dup = untags[*c] untags[*c] = true case *typeFilterConstraint: if c.src == 0 { continue // skip filter of non-pointer } _, dup = typeFilters[*c] typeFilters[*c] = true case *invokeConstraint: if c.params == 0 { panic("non-pointer invoke.params") } if c.iface == 0 { continue // skip invoke on non-pointer } _, dup = invokes[*c] invokes[*c] = true default: // We don't bother de-duping advanced constraints // (e.g. reflection) since they are uncommon. // Eliminate constraints containing non-pointer nodeids. // // We use reflection to find the fields to avoid // adding yet another method to constraint. // // TODO(adonovan): experiment with a constraint // method that returns a slice of pointers to // nodeids fields to enable uniform iteration; // the renumber() method could be removed and // implemented using the new one. // // TODO(adonovan): opt: this is unsound since // some constraints still have an effect if one // of the operands is zero: rVCall, rVMapIndex, // rvSetMapIndex. Handle them specially. rtNodeid := reflect.TypeOf(nodeid(0)) x := reflect.ValueOf(c).Elem() for i, nf := 0, x.NumField(); i < nf; i++ { f := x.Field(i) if f.Type() == rtNodeid { if f.Uint() == 0 { dup = true // skip it break } } } } if dup { continue // skip duplicates } cc = append(cc, c) } h.a.constraints = cc if h.log != nil { fmt.Fprintf(h.log, "#constraints: was %d, now %d\n", nbefore, len(h.a.constraints)) } } // find returns the canonical onodeid for x. // (The onodes form a disjoint set forest.) func (h *hvn) find(x onodeid) onodeid { // TODO(adonovan): opt: this is a CPU hotspot. Try "union by rank". xo := h.onodes[x] rep := xo.rep if rep != x { rep = h.find(rep) // simple path compression xo.rep = rep } return rep } func (h *hvn) checkCanonical(x onodeid) { if debugHVN { assert(x == h.find(x), "not canonical") } } func assert(p bool, msg string) { if debugHVN && !p { panic("assertion failed: " + msg) } }