// 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 pointer // This file defines the constraint generation phase. // TODO(adonovan): move the constraint definitions and the store() etc // functions which add them (and are also used by the solver) into a // new file, constraints.go. import ( "fmt" "go/ast" "go/token" "code.google.com/p/go.tools/go/types" "code.google.com/p/go.tools/ssa" ) var ( tEface = types.NewInterface(nil) tInvalid = types.Typ[types.Invalid] tUnsafePtr = types.Typ[types.UnsafePointer] ) // ---------- Node creation ---------- // nextNode returns the index of the next unused node. func (a *analysis) nextNode() nodeid { return nodeid(len(a.nodes)) } // addNodes creates nodes for all scalar elements in type typ, and // returns the id of the first one, or zero if the type was // analytically uninteresting. // // comment explains the origin of the nodes, as a debugging aid. // func (a *analysis) addNodes(typ types.Type, comment string) nodeid { id := a.nextNode() for _, fi := range a.flatten(typ) { a.addOneNode(fi.typ, comment, fi) } if id == a.nextNode() { return 0 // type contained no pointers } return id } // addOneNode creates a single node with type typ, and returns its id. // // typ should generally be scalar (except for tagged.T nodes // and struct/array identity nodes). Use addNodes for non-scalar types. // // comment explains the origin of the nodes, as a debugging aid. // subelement indicates the subelement, e.g. ".a.b[*].c". // func (a *analysis) addOneNode(typ types.Type, comment string, subelement *fieldInfo) nodeid { id := a.nextNode() a.nodes = append(a.nodes, &node{typ: typ, subelement: subelement}) if a.log != nil { fmt.Fprintf(a.log, "\tcreate n%d %s for %s%s\n", id, typ, comment, subelement.path()) } return id } // setValueNode associates node id with the value v. // cgn identifies the context iff v is a local variable. // func (a *analysis) setValueNode(v ssa.Value, id nodeid, cgn *cgnode) { if cgn != nil { a.localval[v] = id } else { a.globalval[v] = id } if a.log != nil { fmt.Fprintf(a.log, "\tval[%s] = n%d (%T)\n", v.Name(), id, v) } // Record the (v, id) relation if the client has queried v. if indirect, ok := a.config.Queries[v]; ok { if indirect { tmp := a.addNodes(v.Type(), "query.indirect") a.genLoad(cgn, tmp, v, 0, a.sizeof(v.Type())) id = tmp } a.queries[v] = append(a.queries[v], ptr{a, cgn, id}) } } // endObject marks the end of a sequence of calls to addNodes denoting // a single object allocation. // // obj is the start node of the object, from a prior call to nextNode. // Its size, flags and (optionally) data will be updated. // func (a *analysis) endObject(obj nodeid, cgn *cgnode, val ssa.Value) *object { // Ensure object is non-empty by padding; // the pad will be the object node. size := uint32(a.nextNode() - obj) if size == 0 { a.addOneNode(tInvalid, "padding", nil) } objNode := a.nodes[obj] o := &object{ size: size, // excludes padding cgn: cgn, val: val, } objNode.obj = o if val != nil && a.log != nil { fmt.Fprintf(a.log, "\tobj[%s] = n%d\n", val, obj) } return o } // makeFunctionObject creates and returns a new function object for // fn, and returns the id of its first node. It also enqueues fn for // subsequent constraint generation. // func (a *analysis) makeFunctionObject(fn *ssa.Function) nodeid { if a.log != nil { fmt.Fprintf(a.log, "\t---- makeFunctionObject %s\n", fn) } // obj is the function object (identity, params, results). obj := a.nextNode() cgn := a.makeCGNode(fn, obj) sig := fn.Signature a.addOneNode(sig, "func.cgnode", nil) // (scalar with Signature type) if recv := sig.Recv(); recv != nil { a.addNodes(recv.Type(), "func.recv") } a.addNodes(sig.Params(), "func.params") a.addNodes(sig.Results(), "func.results") a.endObject(obj, cgn, fn).flags |= otFunction if a.log != nil { fmt.Fprintf(a.log, "\t----\n") } // Queue it up for constraint processing. a.genq = append(a.genq, cgn) return obj } // makeTagged creates a tagged object of type typ. func (a *analysis) makeTagged(typ types.Type, cgn *cgnode, val ssa.Value) nodeid { obj := a.addOneNode(typ, "tagged.T", nil) // NB: type may be non-scalar! a.addNodes(typ, "tagged.v") a.endObject(obj, cgn, val).flags |= otTagged return obj } // makeRtype returns the canonical tagged object of type *rtype whose // payload points to the sole rtype object for T. func (a *analysis) makeRtype(T types.Type) nodeid { if v := a.rtypes.At(T); v != nil { return v.(nodeid) } // Create the object for the reflect.rtype itself, which is // ordinarily a large struct but here a single node will do. obj := a.nextNode() a.addOneNode(T, "reflect.rtype", nil) a.endObject(obj, nil, nil).rtype = T id := a.makeTagged(a.reflectRtypePtr, nil, nil) a.nodes[id].obj.rtype = T a.nodes[id+1].typ = T // trick (each *rtype tagged object is a singleton) a.addressOf(id+1, obj) a.rtypes.Set(T, id) return id } // rtypeValue returns the type of the *reflect.rtype-tagged object obj. func (a *analysis) rtypeTaggedValue(obj nodeid) types.Type { tDyn, t, _ := a.taggedValue(obj) if tDyn != a.reflectRtypePtr { panic(fmt.Sprintf("not a *reflect.rtype-tagged value: obj=n%d tag=%v payload=n%d", obj, tDyn, t)) } return a.nodes[t].typ } // valueNode returns the id of the value node for v, creating it (and // the association) as needed. It may return zero for uninteresting // values containing no pointers. // func (a *analysis) valueNode(v ssa.Value) nodeid { // Value nodes for locals are created en masse by genFunc. if id, ok := a.localval[v]; ok { return id } // Value nodes for globals are created on demand. id, ok := a.globalval[v] if !ok { var comment string if a.log != nil { comment = v.String() } id = a.addOneNode(v.Type(), comment, nil) if obj := a.objectNode(nil, v); obj != 0 { a.addressOf(id, obj) } a.setValueNode(v, id, nil) } return id } // valueOffsetNode ascertains the node for tuple/struct value v, // then returns the node for its subfield #index. // func (a *analysis) valueOffsetNode(v ssa.Value, index int) nodeid { id := a.valueNode(v) if id == 0 { panic(fmt.Sprintf("cannot offset within n0: %s = %s", v.Name(), v)) } return id + nodeid(a.offsetOf(v.Type(), index)) } // taggedValue returns the dynamic type tag, the (first node of the) // payload, and the indirect flag of the tagged object starting at id. // It returns tDyn==nil if obj is not a tagged object. // func (a *analysis) taggedValue(id nodeid) (tDyn types.Type, v nodeid, indirect bool) { n := a.nodes[id] flags := n.obj.flags if flags&otTagged != 0 { return n.typ, id + 1, flags&otIndirect != 0 } return } // funcParams returns the first node of the params block of the // function whose object node (obj.flags&otFunction) is id. // func (a *analysis) funcParams(id nodeid) nodeid { n := a.nodes[id] if n.obj == nil || n.obj.flags&otFunction == 0 { panic(fmt.Sprintf("funcParams(n%d): not a function object block", id)) } return id + 1 } // funcResults returns the first node of the results block of the // function whose object node (obj.flags&otFunction) is id. // func (a *analysis) funcResults(id nodeid) nodeid { n := a.nodes[id] if n.obj == nil || n.obj.flags&otFunction == 0 { panic(fmt.Sprintf("funcResults(n%d): not a function object block", id)) } sig := n.typ.(*types.Signature) id += 1 + nodeid(a.sizeof(sig.Params())) if sig.Recv() != nil { id += nodeid(a.sizeof(sig.Recv().Type())) } return id } // ---------- Constraint creation ---------- // copy creates a constraint of the form dst = src. // sizeof is the width (in logical fields) of the copied type. // func (a *analysis) copy(dst, src nodeid, sizeof uint32) { if src == dst || sizeof == 0 { return // trivial } if src == 0 || dst == 0 { panic(fmt.Sprintf("ill-typed copy dst=n%d src=n%d", dst, src)) } for i := uint32(0); i < sizeof; i++ { a.addConstraint(©Constraint{dst, src}) src++ dst++ } } // addressOf creates a constraint of the form id = &obj. func (a *analysis) addressOf(id, obj nodeid) { if id == 0 { panic("addressOf: zero id") } if obj == 0 { panic("addressOf: zero obj") } a.addConstraint(&addrConstraint{id, obj}) } // load creates a load constraint of the form dst = src[offset]. // offset is the pointer offset in logical fields. // sizeof is the width (in logical fields) of the loaded type. // func (a *analysis) load(dst, src nodeid, offset, sizeof uint32) { if dst == 0 { return // load of non-pointerlike value } if src == 0 && dst == 0 { return // non-pointerlike operation } if src == 0 || dst == 0 { panic(fmt.Sprintf("ill-typed load dst=n%d src=n%d", dst, src)) } for i := uint32(0); i < sizeof; i++ { a.addConstraint(&loadConstraint{offset, dst, src}) offset++ dst++ } } // store creates a store constraint of the form dst[offset] = src. // offset is the pointer offset in logical fields. // sizeof is the width (in logical fields) of the stored type. // func (a *analysis) store(dst, src nodeid, offset uint32, sizeof uint32) { if src == 0 { return // store of non-pointerlike value } if src == 0 && dst == 0 { return // non-pointerlike operation } if src == 0 || dst == 0 { panic(fmt.Sprintf("ill-typed store dst=n%d src=n%d", dst, src)) } for i := uint32(0); i < sizeof; i++ { a.addConstraint(&storeConstraint{offset, dst, src}) offset++ src++ } } // offsetAddr creates an offsetAddr constraint of the form dst = &src.#offset. // offset is the field offset in logical fields. // func (a *analysis) offsetAddr(dst, src nodeid, offset uint32) { if offset == 0 { // Simplify dst = &src->f0 // to dst = src // (NB: this optimisation is defeated by the identity // field prepended to struct and array objects.) a.copy(dst, src, 1) } else { a.addConstraint(&offsetAddrConstraint{offset, dst, src}) } } // typeAssert creates a typeAssert constraint of the form dst = src.(T). func (a *analysis) typeAssert(T types.Type, dst, src nodeid) { a.addConstraint(&typeAssertConstraint{T, dst, src}) } // addConstraint adds c to the constraint set. func (a *analysis) addConstraint(c constraint) { a.constraints = append(a.constraints, c) if a.log != nil { fmt.Fprintf(a.log, "\t%s\n", c) } } // copyElems generates load/store constraints for *dst = *src, // where src and dst are slices or *arrays. // func (a *analysis) copyElems(cgn *cgnode, typ types.Type, dst, src ssa.Value) { tmp := a.addNodes(typ, "copy") sz := a.sizeof(typ) a.genLoad(cgn, tmp, src, 1, sz) a.genStore(cgn, dst, tmp, 1, sz) } // ---------- Constraint generation ---------- // genConv generates constraints for the conversion operation conv. func (a *analysis) genConv(conv *ssa.Convert, cgn *cgnode) { res := a.valueNode(conv) if res == 0 { return // result is non-pointerlike } tSrc := conv.X.Type() tDst := conv.Type() switch utSrc := tSrc.Underlying().(type) { case *types.Slice: // []byte/[]rune -> string? return case *types.Pointer: // *T -> unsafe.Pointer? if tDst == tUnsafePtr { // ignore for now // a.copy(res, a.valueNode(conv.X), 1) return } case *types.Basic: switch utDst := tDst.Underlying().(type) { case *types.Pointer: // unsafe.Pointer -> *T? (currently unsound) if utSrc == tUnsafePtr { // For now, suppress unsafe.Pointer conversion // warnings on "syscall" package. // TODO(adonovan): audit for soundness. if conv.Parent().Pkg.Object.Path() != "syscall" { a.warnf(conv.Pos(), "unsound: %s contains an unsafe.Pointer conversion (to %s)", conv.Parent(), tDst) } // For now, we treat unsafe.Pointer->*T // conversion like new(T) and create an // unaliased object. In future we may handle // unsafe conversions soundly; see TODO file. obj := a.addNodes(mustDeref(tDst), "unsafe.Pointer conversion") a.endObject(obj, cgn, conv) a.addressOf(res, obj) return } case *types.Slice: // string -> []byte/[]rune (or named aliases)? if utSrc.Info()&types.IsString != 0 { obj := a.addNodes(sliceToArray(tDst), "convert") a.endObject(obj, cgn, conv) a.addressOf(res, obj) return } case *types.Basic: // TODO(adonovan): // unsafe.Pointer -> uintptr? // uintptr -> unsafe.Pointer // // The language doesn't adequately specify the // behaviour of these operations, but almost // all uses of these conversions (even in the // spec) seem to imply a non-moving garbage // collection strategy, or implicit "pinning" // semantics for unsafe.Pointer conversions. // TODO(adonovan): we need more work before we can handle // cryptopointers well. if utSrc == tUnsafePtr || utDst == tUnsafePtr { // Ignore for now. See TODO file for ideas. return } return // ignore all other basic type conversions } } panic(fmt.Sprintf("illegal *ssa.Convert %s -> %s: %s", tSrc, tDst, conv.Parent())) } // genAppend generates constraints for a call to append. func (a *analysis) genAppend(instr *ssa.Call, cgn *cgnode) { // Consider z = append(x, y). y is optional. // This may allocate a new [1]T array; call its object w. // We get the following constraints: // z = x // z = &w // *z = *y x := instr.Call.Args[0] z := instr a.copy(a.valueNode(z), a.valueNode(x), 1) // z = x if len(instr.Call.Args) == 1 { return // no allocation for z = append(x) or _ = append(x). } // TODO(adonovan): test append([]byte, ...string) []byte. y := instr.Call.Args[1] tArray := sliceToArray(instr.Call.Args[0].Type()) var w nodeid w = a.nextNode() a.addNodes(tArray, "append") a.endObject(w, cgn, instr) a.copyElems(cgn, tArray.Elem(), z, y) // *z = *y a.addressOf(a.valueNode(z), w) // z = &w } // genBuiltinCall generates contraints for a call to a built-in. func (a *analysis) genBuiltinCall(instr ssa.CallInstruction, cgn *cgnode) { call := instr.Common() switch call.Value.(*ssa.Builtin).Object().Name() { case "append": // Safe cast: append cannot appear in a go or defer statement. a.genAppend(instr.(*ssa.Call), cgn) case "copy": tElem := call.Args[0].Type().Underlying().(*types.Slice).Elem() a.copyElems(cgn, tElem, call.Args[0], call.Args[1]) case "panic": a.copy(a.panicNode, a.valueNode(call.Args[0]), 1) case "recover": if v := instr.Value(); v != nil { a.copy(a.valueNode(v), a.panicNode, 1) } case "print": // Analytically print is a no-op, but it's a convenient hook // for testing the pts of an expression, so we notify the client. // Existing uses in Go core libraries are few and harmless. if Print := a.config.Print; Print != nil { // Due to context-sensitivity, we may encounter // the same print() call in many contexts, so // we merge them to a canonical node. probe := a.probes[call] t := call.Args[0].Type() // First time? Create the canonical probe node. if probe == 0 { probe = a.addNodes(t, "print") a.probes[call] = probe Print(call, ptr{a, nil, probe}) // notify client } a.copy(probe, a.valueNode(call.Args[0]), a.sizeof(t)) } default: // No-ops: close len cap real imag complex println delete. } } // shouldUseContext defines the context-sensitivity policy. It // returns true if we should analyse all static calls to fn anew. // // Obviously this interface rather limits how much freedom we have to // choose a policy. The current policy, rather arbitrarily, is true // for intrinsics and accessor methods (actually: short, single-block, // call-free functions). This is just a starting point. // func (a *analysis) shouldUseContext(fn *ssa.Function) bool { if a.findIntrinsic(fn) != nil { return true // treat intrinsics context-sensitively } if len(fn.Blocks) != 1 { return false // too expensive } blk := fn.Blocks[0] if len(blk.Instrs) > 10 { return false // too expensive } if fn.Synthetic != "" && (fn.Pkg == nil || fn != fn.Pkg.Func("init")) { return true // treat synthetic wrappers context-sensitively } for _, instr := range blk.Instrs { switch instr := instr.(type) { case ssa.CallInstruction: // Disallow function calls (except to built-ins) // because of the danger of unbounded recursion. if _, ok := instr.Common().Value.(*ssa.Builtin); !ok { return false } } } return true } // genStaticCall generates constraints for a statically dispatched // function call. It returns a node whose pts() will be the set of // possible call targets (in this case, a singleton). // func (a *analysis) genStaticCall(call *ssa.CallCommon, result nodeid) nodeid { // Ascertain the context (contour/CGNode) for a particular call. var obj nodeid fn := call.StaticCallee() if a.shouldUseContext(fn) { obj = a.makeFunctionObject(fn) // new contour for this call } else { obj = a.objectNode(nil, fn) // ordinary (shared) contour } sig := call.Signature() targets := a.addOneNode(sig, "call.targets", nil) a.addressOf(targets, obj) // (a singleton) // Copy receiver, if any. params := a.funcParams(obj) args := call.Args if sig.Recv() != nil { sz := a.sizeof(sig.Recv().Type()) a.copy(params, a.valueNode(args[0]), sz) params += nodeid(sz) args = args[1:] } // Copy actual parameters into formal params block. // Must loop, since the actuals aren't contiguous. for i, arg := range args { sz := a.sizeof(sig.Params().At(i).Type()) a.copy(params, a.valueNode(arg), sz) params += nodeid(sz) } // Copy formal results block to actual result. if result != 0 { a.copy(result, a.funcResults(obj), a.sizeof(sig.Results())) } return targets } // genDynamicCall generates constraints for a dynamic function call. // It returns a node whose pts() will be the set of possible call targets. // func (a *analysis) genDynamicCall(caller *cgnode, call *ssa.CallCommon, result nodeid) nodeid { fn := a.valueNode(call.Value) sig := call.Signature() // We add dynamic closure rules that store the arguments into, // and load the results from, the P/R block of each function // discovered in pts(fn). var offset uint32 = 1 // P/R block starts at offset 1 for i, arg := range call.Args { sz := a.sizeof(sig.Params().At(i).Type()) a.genStore(caller, call.Value, a.valueNode(arg), offset, sz) offset += sz } if result != 0 { a.genLoad(caller, result, call.Value, offset, a.sizeof(sig.Results())) } return fn } // genInvoke generates constraints for a dynamic method invocation. // It returns a node whose pts() will be the set of possible call targets. // func (a *analysis) genInvoke(call *ssa.CallCommon, result nodeid) nodeid { if call.Value.Type() == a.reflectType { return a.genInvokeReflectType(call, result) } sig := call.Signature() // Allocate a contiguous targets/params/results block for this call. block := a.nextNode() targets := a.addOneNode(sig, "invoke.targets", nil) p := a.addNodes(sig.Params(), "invoke.params") r := a.addNodes(sig.Results(), "invoke.results") // Copy the actual parameters into the call's params block. for i, n := 0, sig.Params().Len(); i < n; i++ { sz := a.sizeof(sig.Params().At(i).Type()) a.copy(p, a.valueNode(call.Args[i]), sz) p += nodeid(sz) } // Copy the call's results block to the actual results. if result != 0 { a.copy(result, r, a.sizeof(sig.Results())) } // We add a dynamic invoke constraint that will add // edges from the caller's P/R block to the callee's // P/R block for each discovered call target. a.addConstraint(&invokeConstraint{call.Method, a.valueNode(call.Value), block}) return targets } // genInvokeReflectType is a specialization of genInvoke where the // receiver type is a reflect.Type, under the assumption that there // can be at most one implementation of this interface, *reflect.rtype. // // (Though this may appear to be an instance of a pattern---method // calls on interfaces known to have exactly one implementation---in // practice it occurs rarely, so we special case for reflect.Type.) // // In effect we treat this: // var rt reflect.Type = ... // rt.F() // as this: // rt.(*reflect.rtype).F() // // It returns a node whose pts() will be the (singleton) set of // possible call targets. // func (a *analysis) genInvokeReflectType(call *ssa.CallCommon, result nodeid) nodeid { // Unpack receiver into rtype rtype := a.addOneNode(a.reflectRtypePtr, "rtype.recv", nil) recv := a.valueNode(call.Value) a.typeAssert(a.reflectRtypePtr, rtype, recv) // Look up the concrete method. meth := a.reflectRtypePtr.MethodSet().Lookup(call.Method.Pkg(), call.Method.Name()) fn := a.prog.Method(meth) obj := a.makeFunctionObject(fn) // new contour for this call // From now on, it's essentially a static call, but little is // gained by factoring together the code for both cases. sig := fn.Signature // concrete method targets := a.addOneNode(sig, "call.targets", nil) a.addressOf(targets, obj) // (a singleton) // Copy receiver. params := a.funcParams(obj) a.copy(params, rtype, 1) params++ // Copy actual parameters into formal params block. // Must loop, since the actuals aren't contiguous. for i, arg := range call.Args { sz := a.sizeof(sig.Params().At(i).Type()) a.copy(params, a.valueNode(arg), sz) params += nodeid(sz) } // Copy formal results block to actual result. if result != 0 { a.copy(result, a.funcResults(obj), a.sizeof(sig.Results())) } return obj } // genCall generates contraints for call instruction instr. func (a *analysis) genCall(caller *cgnode, instr ssa.CallInstruction) { call := instr.Common() // Intrinsic implementations of built-in functions. if _, ok := call.Value.(*ssa.Builtin); ok { a.genBuiltinCall(instr, caller) return } var result nodeid if v := instr.Value(); v != nil { result = a.valueNode(v) } // The node whose pts(ยท) will contain all targets of the call. var targets nodeid switch { case call.StaticCallee() != nil: targets = a.genStaticCall(call, result) case call.IsInvoke(): targets = a.genInvoke(call, result) default: targets = a.genDynamicCall(caller, call, result) } site := &callsite{ targets: targets, instr: instr, } caller.sites = append(caller.sites, site) if a.log != nil { fmt.Fprintf(a.log, "\t%s to targets %s from %s\n", site, site.targets, caller) } } // objectNode returns the object to which v points, if known. // In other words, if the points-to set of v is a singleton, it // returns the sole label, zero otherwise. // // We exploit this information to make the generated constraints less // dynamic. For example, a complex load constraint can be replaced by // a simple copy constraint when the sole destination is known a priori. // // Some SSA instructions always have singletons points-to sets: // Alloc, Function, Global, MakeChan, MakeClosure, MakeInterface, MakeMap, MakeSlice. // Others may be singletons depending on their operands: // Capture, Const, Convert, FieldAddr, IndexAddr, Slice. // // Idempotent. Objects are created as needed, possibly via recursion // down the SSA value graph, e.g IndexAddr(FieldAddr(Alloc))). // func (a *analysis) objectNode(cgn *cgnode, v ssa.Value) nodeid { if cgn == nil { // Global object. obj, ok := a.globalobj[v] if !ok { switch v := v.(type) { case *ssa.Global: obj = a.nextNode() a.addNodes(mustDeref(v.Type()), "global") a.endObject(obj, nil, v) case *ssa.Function: obj = a.makeFunctionObject(v) case *ssa.Const: if t, ok := v.Type().Underlying().(*types.Slice); ok && !v.IsNil() { // Non-nil []byte or []rune constant. obj = a.nextNode() a.addNodes(sliceToArray(t), "array in slice constant") a.endObject(obj, nil, v) } case *ssa.Capture: // For now, Captures have the same cardinality as globals. // TODO(adonovan): treat captures context-sensitively. } a.globalobj[v] = obj } return obj } // Local object. obj, ok := a.localobj[v] if !ok { switch v := v.(type) { case *ssa.Alloc: obj = a.nextNode() a.addNodes(mustDeref(v.Type()), "alloc") a.endObject(obj, cgn, v) case *ssa.MakeSlice: obj = a.nextNode() a.addNodes(sliceToArray(v.Type()), "makeslice") a.endObject(obj, cgn, v) case *ssa.MakeChan: obj = a.nextNode() a.addNodes(v.Type().Underlying().(*types.Chan).Elem(), "makechan") a.endObject(obj, cgn, v) case *ssa.MakeMap: obj = a.nextNode() tmap := v.Type().Underlying().(*types.Map) a.addNodes(tmap.Key(), "makemap.key") a.addNodes(tmap.Elem(), "makemap.value") a.endObject(obj, cgn, v) case *ssa.MakeInterface: tConc := v.X.Type() // Create nodes and constraints for all methods of the type. // Ascertaining which will be needed is undecidable in general. mset := tConc.MethodSet() for i, n := 0, mset.Len(); i < n; i++ { a.valueNode(a.prog.Method(mset.At(i))) } obj = a.makeTagged(tConc, cgn, v) // Copy the value into it, if nontrivial. if x := a.valueNode(v.X); x != 0 { a.copy(obj+1, x, a.sizeof(tConc)) } case *ssa.FieldAddr: if xobj := a.objectNode(cgn, v.X); xobj != 0 { obj = xobj + nodeid(a.offsetOf(mustDeref(v.X.Type()), v.Field)) } case *ssa.IndexAddr: if xobj := a.objectNode(cgn, v.X); xobj != 0 { obj = xobj + 1 } case *ssa.Slice: obj = a.objectNode(cgn, v.X) case *ssa.Convert: // TODO(adonovan): opt: handle these cases too: // - unsafe.Pointer->*T conversion acts like Alloc // - string->[]byte/[]rune conversion acts like MakeSlice } a.localobj[v] = obj } return obj } // genLoad generates constraints for result = *(ptr + val). func (a *analysis) genLoad(cgn *cgnode, result nodeid, ptr ssa.Value, offset, sizeof uint32) { if obj := a.objectNode(cgn, ptr); obj != 0 { // Pre-apply loadConstraint.solve(). a.copy(result, obj+nodeid(offset), sizeof) } else { a.load(result, a.valueNode(ptr), offset, sizeof) } } // genOffsetAddr generates constraints for a 'v=ptr.field' (FieldAddr) // or 'v=ptr[*]' (IndexAddr) instruction v. func (a *analysis) genOffsetAddr(cgn *cgnode, v ssa.Value, ptr nodeid, offset uint32) { dst := a.valueNode(v) if obj := a.objectNode(cgn, v); obj != 0 { // Pre-apply offsetAddrConstraint.solve(). a.addressOf(dst, obj) } else { a.offsetAddr(dst, ptr, offset) } } // genStore generates constraints for *(ptr + offset) = val. func (a *analysis) genStore(cgn *cgnode, ptr ssa.Value, val nodeid, offset, sizeof uint32) { if obj := a.objectNode(cgn, ptr); obj != 0 { // Pre-apply storeConstraint.solve(). a.copy(obj+nodeid(offset), val, sizeof) } else { a.store(a.valueNode(ptr), val, offset, sizeof) } } // genInstr generates contraints for instruction instr in context cgn. func (a *analysis) genInstr(cgn *cgnode, instr ssa.Instruction) { if a.log != nil { var prefix string if val, ok := instr.(ssa.Value); ok { prefix = val.Name() + " = " } fmt.Fprintf(a.log, "; %s%s\n", prefix, instr) } switch instr := instr.(type) { case *ssa.DebugRef: // no-op. case *ssa.UnOp: switch instr.Op { case token.ARROW: // <-x // We can ignore instr.CommaOk because the node we're // altering is always at zero offset relative to instr a.genLoad(cgn, a.valueNode(instr), instr.X, 0, a.sizeof(instr.Type())) case token.MUL: // *x a.genLoad(cgn, a.valueNode(instr), instr.X, 0, a.sizeof(instr.Type())) default: // NOT, SUB, XOR: no-op. } case *ssa.BinOp: // All no-ops. case ssa.CallInstruction: // *ssa.Call, *ssa.Go, *ssa.Defer a.genCall(cgn, instr) case *ssa.ChangeType: a.copy(a.valueNode(instr), a.valueNode(instr.X), 1) case *ssa.Convert: a.genConv(instr, cgn) case *ssa.Extract: a.copy(a.valueNode(instr), a.valueOffsetNode(instr.Tuple, instr.Index), a.sizeof(instr.Type())) case *ssa.FieldAddr: a.genOffsetAddr(cgn, instr, a.valueNode(instr.X), a.offsetOf(mustDeref(instr.X.Type()), instr.Field)) case *ssa.IndexAddr: a.genOffsetAddr(cgn, instr, a.valueNode(instr.X), 1) case *ssa.Field: a.copy(a.valueNode(instr), a.valueOffsetNode(instr.X, instr.Field), a.sizeof(instr.Type())) case *ssa.Index: a.copy(a.valueNode(instr), 1+a.valueNode(instr.X), a.sizeof(instr.Type())) case *ssa.Select: recv := a.valueOffsetNode(instr, 2) // instr : (index, recvOk, recv0, ... recv_n-1) for _, st := range instr.States { elemSize := a.sizeof(st.Chan.Type().Underlying().(*types.Chan).Elem()) switch st.Dir { case ast.RECV: a.genLoad(cgn, recv, st.Chan, 0, elemSize) recv++ case ast.SEND: a.genStore(cgn, st.Chan, a.valueNode(st.Send), 0, elemSize) } } case *ssa.Ret: results := a.funcResults(cgn.obj) for _, r := range instr.Results { sz := a.sizeof(r.Type()) a.copy(results, a.valueNode(r), sz) results += nodeid(sz) } case *ssa.Send: a.genStore(cgn, instr.Chan, a.valueNode(instr.X), 0, a.sizeof(instr.X.Type())) case *ssa.Store: a.genStore(cgn, instr.Addr, a.valueNode(instr.Val), 0, a.sizeof(instr.Val.Type())) case *ssa.Alloc, *ssa.MakeSlice, *ssa.MakeChan, *ssa.MakeMap, *ssa.MakeInterface: v := instr.(ssa.Value) a.addressOf(a.valueNode(v), a.objectNode(cgn, v)) case *ssa.ChangeInterface: a.copy(a.valueNode(instr), a.valueNode(instr.X), 1) case *ssa.TypeAssert: a.typeAssert(instr.AssertedType, a.valueNode(instr), a.valueNode(instr.X)) case *ssa.Slice: a.copy(a.valueNode(instr), a.valueNode(instr.X), 1) case *ssa.If, *ssa.Jump: // no-op. case *ssa.Phi: sz := a.sizeof(instr.Type()) for _, e := range instr.Edges { a.copy(a.valueNode(instr), a.valueNode(e), sz) } case *ssa.MakeClosure: fn := instr.Fn.(*ssa.Function) a.copy(a.valueNode(instr), a.valueNode(fn), 1) // Free variables are treated like global variables. for i, b := range instr.Bindings { a.copy(a.valueNode(fn.FreeVars[i]), a.valueNode(b), a.sizeof(b.Type())) } case *ssa.RunDefers: // The analysis is flow insensitive, so we just "call" // defers as we encounter them. case *ssa.Range: // Do nothing. Next{Iter: *ssa.Range} handles this case. case *ssa.Next: if !instr.IsString { // map // Assumes that Next is always directly applied to a Range result. theMap := instr.Iter.(*ssa.Range).X tMap := theMap.Type().Underlying().(*types.Map) ksize := a.sizeof(tMap.Key()) vsize := a.sizeof(tMap.Elem()) // Load from the map's (k,v) into the tuple's (ok, k, v). a.genLoad(cgn, a.valueNode(instr)+1, theMap, 0, ksize+vsize) } case *ssa.Lookup: if tMap, ok := instr.X.Type().Underlying().(*types.Map); ok { // CommaOk can be ignored: field 0 is a no-op. ksize := a.sizeof(tMap.Key()) vsize := a.sizeof(tMap.Elem()) a.genLoad(cgn, a.valueNode(instr), instr.X, ksize, vsize) } case *ssa.MapUpdate: tmap := instr.Map.Type().Underlying().(*types.Map) ksize := a.sizeof(tmap.Key()) vsize := a.sizeof(tmap.Elem()) a.genStore(cgn, instr.Map, a.valueNode(instr.Key), 0, ksize) a.genStore(cgn, instr.Map, a.valueNode(instr.Value), ksize, vsize) case *ssa.Panic: a.copy(a.panicNode, a.valueNode(instr.X), 1) default: panic(fmt.Sprintf("unimplemented: %T", instr)) } } func (a *analysis) makeCGNode(fn *ssa.Function, obj nodeid) *cgnode { cgn := &cgnode{fn: fn, obj: obj} a.cgnodes = append(a.cgnodes, cgn) return cgn } // genRootCalls generates the synthetic root of the callgraph and the // initial calls from it to the analysis scope, such as main, a test // or a library. // func (a *analysis) genRootCalls() *cgnode { r := ssa.NewFunction("", new(types.Signature), "root of callgraph") r.Prog = a.prog // hack. r.Enclosing = r // hack, so Function.String() doesn't crash r.String() // (asserts that it doesn't crash) root := a.makeCGNode(r, 0) // For each main package, call main.init(), main.main(). for _, mainPkg := range a.config.Mains { main := mainPkg.Func("main") if main == nil { panic(fmt.Sprintf("%s has no main function", mainPkg)) } targets := a.addOneNode(main.Signature, "root.targets", nil) site := &callsite{targets: targets} root.sites = append(root.sites, site) for _, fn := range [2]*ssa.Function{mainPkg.Func("init"), main} { if a.log != nil { fmt.Fprintf(a.log, "\troot call to %s:\n", fn) } a.copy(targets, a.valueNode(fn), 1) } } return root } // genFunc generates constraints for function fn. func (a *analysis) genFunc(cgn *cgnode) { fn := cgn.fn impl := a.findIntrinsic(fn) if a.log != nil { fmt.Fprintln(a.log) fmt.Fprintln(a.log) // Hack: don't display body if intrinsic. if impl != nil { fn2 := *cgn.fn // copy fn2.Locals = nil fn2.Blocks = nil fn2.DumpTo(a.log) } else { cgn.fn.DumpTo(a.log) } } if impl != nil { impl(a, cgn) return } if fn.Blocks == nil { // External function with no intrinsic treatment. // We'll warn about calls to such functions at the end. return } if a.log != nil { fmt.Fprintln(a.log, "; Creating nodes for local values") } a.localval = make(map[ssa.Value]nodeid) a.localobj = make(map[ssa.Value]nodeid) // The value nodes for the params are in the func object block. params := a.funcParams(cgn.obj) for _, p := range fn.Params { a.setValueNode(p, params, cgn) params += nodeid(a.sizeof(p.Type())) } // Free variables are treated like global variables: // the outer function sets them with MakeClosure; // the inner function accesses them with Capture. // Create value nodes for all value instructions // since SSA may contain forward references. for _, b := range fn.Blocks { for _, instr := range b.Instrs { switch instr := instr.(type) { case *ssa.Range: // do nothing: it has a funky type, // and *ssa.Next does all the work. case ssa.Value: var comment string if a.log != nil { comment = instr.Name() } id := a.addNodes(instr.Type(), comment) a.setValueNode(instr, id, cgn) } } } // Generate constraints for instructions. for _, b := range fn.Blocks { for _, instr := range b.Instrs { a.genInstr(cgn, instr) } } a.localval = nil a.localobj = nil } // generate generates offline constraints for the entire program. // It returns the synthetic root of the callgraph. // func (a *analysis) generate() *cgnode { // Create a dummy node since we use the nodeid 0 for // non-pointerlike variables. a.addNodes(tInvalid, "(zero)") // Create the global node for panic values. a.panicNode = a.addNodes(tEface, "panic") // Create nodes and constraints for all methods of reflect.rtype. // (Shared contours are used by dynamic calls to reflect.Type // methods---typically just String().) if rtype := a.reflectRtypePtr; rtype != nil { mset := rtype.MethodSet() for i, n := 0, mset.Len(); i < n; i++ { a.valueNode(a.prog.Method(mset.At(i))) } } root := a.genRootCalls() // Generate constraints for entire program. // (Actually just the RTA-reachable portion of the program. // See Bacon & Sweeney, OOPSLA'96). for len(a.genq) > 0 { cgn := a.genq[0] a.genq = a.genq[1:] a.genFunc(cgn) } return root }