variables.go 36.9 KB
Newer Older
D
Derek Parker 已提交
1
package proc
2 3 4 5

import (
	"bytes"
	"encoding/binary"
6
	"errors"
7
	"fmt"
8
	"go/constant"
9 10
	"go/parser"
	"go/token"
11
	"golang.org/x/debug/dwarf"
12
	"reflect"
13 14 15 16
	"strings"
	"unsafe"

	"github.com/derekparker/delve/dwarf/op"
17
	"github.com/derekparker/delve/dwarf/reader"
18 19
)

20
const (
21 22
	maxVariableRecurse = 1  // How far to recurse when evaluating nested types.
	maxArrayValues     = 64 // Max value for reading large arrays.
23
	maxErrCount        = 3  // Max number of read errors to accept while evaluating slices, arrays and structs
24

25
	maxArrayStridePrefetch = 1024 // Maximum size of array stride for which we will prefetch the array contents
26

D
Derek Parker 已提交
27 28
	chanRecv = "chan receive"
	chanSend = "chan send"
A
aarzilli 已提交
29 30 31

	hashTophashEmpty = 0 // used by map reading code, indicates an empty bucket
	hashMinTopHash   = 4 // used by map reading code, indicates minimum value of tophash that isn't empty or evacuated
32 33
)

D
Derek Parker 已提交
34 35 36 37
// Variable represents a variable. It contains the address, name,
// type and other information parsed from both the Dwarf information
// and the memory of the debugged process.
// If OnlyAddr is true, the variables value has not been loaded.
38
type Variable struct {
39
	Addr      uintptr
A
aarzilli 已提交
40
	OnlyAddr  bool
41
	Name      string
42 43 44
	DwarfType dwarf.Type
	RealType  dwarf.Type
	Kind      reflect.Kind
45 46
	mem       memoryReadWriter
	dbp       *Process
47

48
	Value constant.Value
49 50 51 52

	Len int64
	Cap int64

53 54
	// Base address of arrays, Base address of the backing array for slices (0 for nil slices)
	// Base address of the backing byte array for strings
A
aarzilli 已提交
55
	// address of the struct backing chan and map variables
A
aarzilli 已提交
56
	// address of the function entry point for function variables (0 for nil function pointers)
57
	Base      uintptr
58 59
	stride    int64
	fieldType dwarf.Type
60

A
aarzilli 已提交
61 62 63
	// number of elements to skip when loading a map
	mapSkip int

64 65 66 67
	Children []Variable

	loaded     bool
	Unreadable error
68 69
}

D
Derek Parker 已提交
70
// M represents a runtime M (OS thread) structure.
71
type M struct {
72 73 74 75
	procid   int     // Thread ID or port.
	spinning uint8   // Busy looping.
	blocked  uint8   // Waiting on futex / semaphore.
	curg     uintptr // Current G running on this thread.
76 77
}

D
Derek Parker 已提交
78
// G status, from: src/runtime/runtime2.go
79
const (
D
Derek Parker 已提交
80 81 82 83 84 85 86 87 88
	Gidle           uint64 = iota // 0
	Grunnable                     // 1 runnable and on a run queue
	Grunning                      // 2
	Gsyscall                      // 3
	Gwaiting                      // 4
	GmoribundUnused               // 5 currently unused, but hardcoded in gdb scripts
	Gdead                         // 6
	Genqueue                      // 7 Only the Gscanenqueue is used.
	Gcopystack                    // 8 in this state when newstack is moving the stack
89 90
)

D
Derek Parker 已提交
91
// G represents a runtime G (goroutine) structure (at least the
92
// fields that Delve is interested in).
D
Derek Parker 已提交
93
type G struct {
D
Derek Parker 已提交
94
	ID         int    // Goroutine ID
95 96 97 98
	PC         uint64 // PC of goroutine when it was parked.
	SP         uint64 // SP of goroutine when it was parked.
	GoPC       uint64 // PC of 'go' statement that created this goroutine.
	WaitReason string // Reason for goroutine being parked.
99
	Status     uint64
100

101
	// Information on goroutine location
102
	CurrentLoc Location
103 104 105

	// PC of entry to top-most deferred function.
	DeferPC uint64
A
aarzilli 已提交
106 107 108

	// Thread that this goroutine is currently allocated to
	thread *Thread
109 110

	dbp *Process
111 112
}

D
Derek Parker 已提交
113 114
// EvalScope is the scope for variable evaluation. Contains the thread,
// current location (PC), and canonical frame address.
115 116 117 118 119 120
type EvalScope struct {
	Thread *Thread
	PC     uint64
	CFA    int64
}

D
Derek Parker 已提交
121
// IsNilErr is returned when a variable is nil.
122 123 124 125 126 127 128 129
type IsNilErr struct {
	name string
}

func (err *IsNilErr) Error() string {
	return fmt.Sprintf("%s is nil", err.name)
}

130 131 132 133
func (scope *EvalScope) newVariable(name string, addr uintptr, dwarfType dwarf.Type) *Variable {
	return newVariable(name, addr, dwarfType, scope.Thread.dbp, scope.Thread)
}

134 135 136 137
func (t *Thread) newVariable(name string, addr uintptr, dwarfType dwarf.Type) *Variable {
	return newVariable(name, addr, dwarfType, t.dbp, t)
}

138 139 140 141 142
func (v *Variable) newVariable(name string, addr uintptr, dwarfType dwarf.Type) *Variable {
	return newVariable(name, addr, dwarfType, v.dbp, v.mem)
}

func newVariable(name string, addr uintptr, dwarfType dwarf.Type, dbp *Process, mem memoryReadWriter) *Variable {
143 144 145
	v := &Variable{
		Name:      name,
		Addr:      addr,
146
		DwarfType: dwarfType,
147 148
		mem:       mem,
		dbp:       dbp,
149 150
	}

151
	v.RealType = resolveTypedef(v.DwarfType)
152 153 154

	switch t := v.RealType.(type) {
	case *dwarf.PtrType:
155 156 157
		v.Kind = reflect.Ptr
		if _, isvoid := t.Type.(*dwarf.VoidType); isvoid {
			v.Kind = reflect.UnsafePointer
158
		}
159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
	case *dwarf.ChanType:
		v.Kind = reflect.Chan
	case *dwarf.MapType:
		v.Kind = reflect.Map
	case *dwarf.StringType:
		v.Kind = reflect.String
		v.stride = 1
		v.fieldType = &dwarf.UintType{BasicType: dwarf.BasicType{CommonType: dwarf.CommonType{ByteSize: 1, Name: "byte"}, BitSize: 8, BitOffset: 0}}
		if v.Addr != 0 {
			v.Base, v.Len, v.Unreadable = readStringInfo(v.mem, v.dbp.arch, v.Addr)
		}
	case *dwarf.SliceType:
		v.Kind = reflect.Slice
		if v.Addr != 0 {
			v.loadSliceInfo(t)
		}
	case *dwarf.InterfaceType:
		v.Kind = reflect.Interface
	case *dwarf.StructType:
		v.Kind = reflect.Struct
179
	case *dwarf.ArrayType:
180
		v.Kind = reflect.Array
181
		v.Base = v.Addr
182 183 184 185 186 187 188 189
		v.Len = t.Count
		v.Cap = -1
		v.fieldType = t.Type
		v.stride = 0

		if t.Count > 0 {
			v.stride = t.ByteSize / t.Count
		}
190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217
	case *dwarf.ComplexType:
		switch t.ByteSize {
		case 8:
			v.Kind = reflect.Complex64
		case 16:
			v.Kind = reflect.Complex128
		}
	case *dwarf.IntType:
		v.Kind = reflect.Int
	case *dwarf.UintType:
		v.Kind = reflect.Uint
	case *dwarf.FloatType:
		switch t.ByteSize {
		case 4:
			v.Kind = reflect.Float32
		case 8:
			v.Kind = reflect.Float64
		}
	case *dwarf.BoolType:
		v.Kind = reflect.Bool
	case *dwarf.FuncType:
		v.Kind = reflect.Func
	case *dwarf.VoidType:
		v.Kind = reflect.Invalid
	case *dwarf.UnspecifiedType:
		v.Kind = reflect.Invalid
	default:
		v.Unreadable = fmt.Errorf("Unknown type: %T", t)
218 219
	}

220 221 222
	return v
}

223 224 225 226 227 228 229 230 231 232
func resolveTypedef(typ dwarf.Type) dwarf.Type {
	for {
		if tt, ok := typ.(*dwarf.TypedefType); ok {
			typ = tt.Type
		} else {
			return typ
		}
	}
}

233 234
func newConstant(val constant.Value, mem memoryReadWriter) *Variable {
	v := &Variable{Value: val, mem: mem, loaded: true}
A
aarzilli 已提交
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252
	switch val.Kind() {
	case constant.Int:
		v.Kind = reflect.Int
	case constant.Float:
		v.Kind = reflect.Float64
	case constant.Bool:
		v.Kind = reflect.Bool
	case constant.Complex:
		v.Kind = reflect.Complex128
	case constant.String:
		v.Kind = reflect.String
		v.Len = int64(len(constant.StringVal(val)))
	}
	return v
}

var nilVariable = &Variable{
	Addr:     0,
253
	Base:     0,
A
aarzilli 已提交
254 255 256 257
	Kind:     reflect.Ptr,
	Children: []Variable{{Addr: 0, OnlyAddr: true}},
}

258 259 260
func (v *Variable) clone() *Variable {
	r := *v
	return &r
261 262
}

D
Derek Parker 已提交
263 264
// TypeString returns the string representation
// of the type of this variable.
265 266 267 268 269 270 271 272 273 274
func (v *Variable) TypeString() string {
	if v == nilVariable {
		return "nil"
	}
	if v.DwarfType != nil {
		return v.DwarfType.String()
	}
	return v.Kind.String()
}

275
func (v *Variable) toField(field *dwarf.StructField) (*Variable, error) {
276 277 278
	if v.Unreadable != nil {
		return v.clone(), nil
	}
D
Derek Parker 已提交
279
	if v.Addr == 0 {
280
		return nil, &IsNilErr{v.Name}
D
Derek Parker 已提交
281 282
	}

283 284
	name := ""
	if v.Name != "" {
285 286 287 288
		parts := strings.Split(field.Name, ".")
		if len(parts) > 1 {
			name = fmt.Sprintf("%s.%s", v.Name, parts[1])
		} else {
D
Derek Parker 已提交
289
			name = fmt.Sprintf("%s.%s", v.Name, field.Name)
290
		}
291
	}
292
	return v.newVariable(name, uintptr(int64(v.Addr)+field.ByteOffset), field.Type), nil
293 294
}

D
Derek Parker 已提交
295 296
// DwarfReader returns the DwarfReader containing the
// Dwarf information for the target process.
297 298 299 300
func (scope *EvalScope) DwarfReader() *reader.Reader {
	return scope.Thread.dbp.DwarfReader()
}

D
Derek Parker 已提交
301
// Type returns the Dwarf type entry at `offset`.
302 303 304 305
func (scope *EvalScope) Type(offset dwarf.Offset) (dwarf.Type, error) {
	return scope.Thread.dbp.dwarf.Type(offset)
}

D
Derek Parker 已提交
306
// PtrSize returns the size of a pointer.
307 308 309 310
func (scope *EvalScope) PtrSize() int {
	return scope.Thread.dbp.arch.PtrSize()
}

A
aarzilli 已提交
311
// ChanRecvBlocked returns whether the goroutine is blocked on
312
// a channel read operation.
313
func (g *G) ChanRecvBlocked() bool {
A
aarzilli 已提交
314
	return (g.thread == nil) && (g.WaitReason == chanRecv)
315 316 317
}

// chanRecvReturnAddr returns the address of the return from a channel read.
D
Derek Parker 已提交
318
func (g *G) chanRecvReturnAddr(dbp *Process) (uint64, error) {
319
	locs, err := dbp.GoroutineStacktrace(g, 4)
320 321 322 323
	if err != nil {
		return 0, err
	}
	topLoc := locs[len(locs)-1]
324
	return topLoc.Current.PC, nil
D
Derek Parker 已提交
325 326
}

D
Derek Parker 已提交
327 328
// NoGError returned when a G could not be found
// for a specific thread.
329 330 331 332 333 334 335 336
type NoGError struct {
	tid int
}

func (ng NoGError) Error() string {
	return fmt.Sprintf("no G executing on thread %d", ng.tid)
}

337 338 339 340 341 342 343
func (gvar *Variable) parseG() (*G, error) {
	mem := gvar.mem
	dbp := gvar.dbp
	gaddr := uint64(gvar.Addr)
	_, deref := gvar.RealType.(*dwarf.PtrType)

	initialInstructions := make([]byte, dbp.arch.PtrSize()+1)
344 345 346
	initialInstructions[0] = op.DW_OP_addr
	binary.LittleEndian.PutUint64(initialInstructions[1:], gaddr)
	if deref {
347
		gaddrbytes, err := mem.readMemory(uintptr(gaddr), dbp.arch.PtrSize())
348 349 350 351 352 353
		if err != nil {
			return nil, fmt.Errorf("error derefing *G %s", err)
		}
		initialInstructions = append([]byte{op.DW_OP_addr}, gaddrbytes...)
		gaddr = binary.LittleEndian.Uint64(gaddrbytes)
		if gaddr == 0 {
354 355 356 357 358
			id := 0
			if thread, ok := mem.(*Thread); ok {
				id = thread.ID
			}
			return nil, NoGError{tid: id}
359
		}
360
	}
D
Derek Parker 已提交
361 362
	if gaddr == 0 {
		return nil, NoGError{}
363
	}
D
Derek Parker 已提交
364 365 366
	gvar.loadValue()
	if gvar.Unreadable != nil {
		return nil, gvar.Unreadable
367
	}
D
Derek Parker 已提交
368 369 370 371 372 373 374 375 376 377 378 379
	schedVar := gvar.toFieldNamed("sched")
	pc, _ := constant.Int64Val(schedVar.toFieldNamed("pc").Value)
	sp, _ := constant.Int64Val(schedVar.toFieldNamed("sp").Value)
	id, _ := constant.Int64Val(gvar.toFieldNamed("goid").Value)
	gopc, _ := constant.Int64Val(gvar.toFieldNamed("gopc").Value)
	waitReason := constant.StringVal(gvar.toFieldNamed("waitreason").Value)
	d := gvar.toFieldNamed("_defer")
	deferPC := int64(0)
	fnvar := d.toFieldNamed("fn")
	if fnvar != nil {
		fnvalvar := fnvar.toFieldNamed("fn")
		deferPC, _ = constant.Int64Val(fnvalvar.Value)
380
	}
D
Derek Parker 已提交
381 382 383 384 385 386 387 388 389 390 391 392
	status, _ := constant.Int64Val(gvar.toFieldNamed("atomicstatus").Value)
	f, l, fn := gvar.dbp.goSymTable.PCToLine(uint64(pc))
	g := &G{
		ID:         int(id),
		GoPC:       uint64(gopc),
		PC:         uint64(pc),
		SP:         uint64(sp),
		WaitReason: waitReason,
		DeferPC:    uint64(deferPC),
		Status:     uint64(status),
		CurrentLoc: Location{PC: uint64(pc), File: f, Line: l, Fn: fn},
		dbp:        gvar.dbp,
393
	}
D
Derek Parker 已提交
394 395
	return g, nil
}
396

D
Derek Parker 已提交
397 398
func (v *Variable) toFieldNamed(name string) *Variable {
	v, err := v.structMember(name)
399
	if err != nil {
D
Derek Parker 已提交
400
		return nil
401
	}
D
Derek Parker 已提交
402 403 404
	v.loadValue()
	if v.Unreadable != nil {
		return nil
405
	}
D
Derek Parker 已提交
406
	return v
407 408
}

409 410 411 412 413 414 415 416
// From $GOROOT/src/runtime/traceback.go:597
// isExportedRuntime reports whether name is an exported runtime function.
// It is only for runtime functions, so ASCII A-Z is fine.
func isExportedRuntime(name string) bool {
	const n = len("runtime.")
	return len(name) > n && name[:n] == "runtime." && 'A' <= name[n] && name[n] <= 'Z'
}

D
Derek Parker 已提交
417 418
// UserCurrent returns the location the users code is at,
// or was at before entering a runtime function.
419 420 421 422 423
func (g *G) UserCurrent() Location {
	pc, sp := g.PC, g.SP
	if g.thread != nil {
		regs, err := g.thread.Registers()
		if err != nil {
424
			return g.CurrentLoc
425 426 427 428 429 430 431 432 433 434 435
		}
		pc, sp = regs.PC(), regs.SP()
	}
	it := newStackIterator(g.dbp, pc, sp)
	for it.Next() {
		frame := it.Frame()
		name := frame.Call.Fn.Name
		if (strings.Index(name, ".") >= 0) && (!strings.HasPrefix(name, "runtime.") || isExportedRuntime(name)) {
			return frame.Call
		}
	}
436
	return g.CurrentLoc
437 438
}

D
Derek Parker 已提交
439 440
// Go returns the location of the 'go' statement
// that spawned this goroutine.
441 442 443 444 445
func (g *G) Go() Location {
	f, l, fn := g.dbp.goSymTable.PCToLine(g.GoPC)
	return Location{PC: g.GoPC, File: f, Line: l, Fn: fn}
}

D
Derek Parker 已提交
446
// EvalVariable returns the value of the given expression (backwards compatibility).
A
aarzilli 已提交
447 448 449
func (scope *EvalScope) EvalVariable(name string) (*Variable, error) {
	return scope.EvalExpression(name)
}
450

D
Derek Parker 已提交
451
// SetVariable sets the value of the named variable
A
aarzilli 已提交
452 453
func (scope *EvalScope) SetVariable(name, value string) error {
	t, err := parser.ParseExpr(name)
454
	if err != nil {
A
aarzilli 已提交
455
		return err
456 457
	}

A
aarzilli 已提交
458
	xv, err := scope.evalAST(t)
459
	if err != nil {
A
aarzilli 已提交
460
		return err
461
	}
462

A
aarzilli 已提交
463 464 465 466 467 468 469 470 471
	if xv.Addr == 0 {
		return fmt.Errorf("Can not assign to \"%s\"", name)
	}

	if xv.Unreadable != nil {
		return fmt.Errorf("Expression \"%s\" is unreadable: %v", name, xv.Unreadable)
	}

	t, err = parser.ParseExpr(value)
472 473 474
	if err != nil {
		return err
	}
A
aarzilli 已提交
475 476 477 478 479 480 481 482 483 484

	yv, err := scope.evalAST(t)
	if err != nil {
		return err
	}

	yv.loadValue()

	if err := yv.isType(xv.RealType, xv.Kind); err != nil {
		return err
485
	}
A
aarzilli 已提交
486 487 488 489 490 491

	if yv.Unreadable != nil {
		return fmt.Errorf("Expression \"%s\" is unreadable: %v", value, yv.Unreadable)
	}

	return xv.setValue(yv)
492 493
}

494 495
func (scope *EvalScope) extractVariableFromEntry(entry *dwarf.Entry) (*Variable, error) {
	rdr := scope.DwarfReader()
496
	v, err := scope.extractVarInfoFromEntry(entry, rdr)
497
	if err != nil {
D
Derek Parker 已提交
498 499
		return nil, err
	}
500 501
	v.loadValue()
	return v, nil
502
}
D
Derek Parker 已提交
503

504 505 506 507 508 509
func (scope *EvalScope) extractVarInfo(varName string) (*Variable, error) {
	reader := scope.DwarfReader()

	_, err := reader.SeekToFunction(scope.PC)
	if err != nil {
		return nil, err
510 511
	}

512
	for entry, err := reader.NextScopeVariable(); entry != nil; entry, err = reader.NextScopeVariable() {
D
Derek Parker 已提交
513
		if err != nil {
514
			return nil, err
D
Derek Parker 已提交
515 516 517 518 519 520 521
		}

		n, ok := entry.Val(dwarf.AttrName).(string)
		if !ok {
			continue
		}

522
		if n == varName {
523
			return scope.extractVarInfoFromEntry(entry, reader)
D
Derek Parker 已提交
524 525
		}
	}
526
	return nil, fmt.Errorf("could not find symbol value for %s", varName)
D
Derek Parker 已提交
527
}
528

529
// LocalVariables returns all local variables from the current function scope.
530 531
func (scope *EvalScope) LocalVariables() ([]*Variable, error) {
	return scope.variablesByTag(dwarf.TagVariable)
532 533 534
}

// FunctionArguments returns the name, value, and type of all current function arguments.
535 536
func (scope *EvalScope) FunctionArguments() ([]*Variable, error) {
	return scope.variablesByTag(dwarf.TagFormalParameter)
537 538 539
}

// PackageVariables returns the name, value, and type of all package variables in the application.
540
func (scope *EvalScope) PackageVariables() ([]*Variable, error) {
D
Derek Parker 已提交
541
	var vars []*Variable
542
	reader := scope.DwarfReader()
543 544 545 546 547 548 549

	for entry, err := reader.NextPackageVariable(); entry != nil; entry, err = reader.NextPackageVariable() {
		if err != nil {
			return nil, err
		}

		// Ignore errors trying to extract values
550
		val, err := scope.extractVariableFromEntry(entry)
551 552 553 554 555 556 557 558 559
		if err != nil {
			continue
		}
		vars = append(vars, val)
	}

	return vars, nil
}

D
Derek Parker 已提交
560 561
// EvalPackageVariable will evaluate the package level variable
// specified by 'name'.
562 563
func (dbp *Process) EvalPackageVariable(name string) (*Variable, error) {
	scope := &EvalScope{Thread: dbp.CurrentThread, PC: 0, CFA: 0}
564

565
	v, err := scope.packageVarAddr(name)
566 567 568
	if err != nil {
		return nil, err
	}
569 570
	v.loadValue()
	return v, nil
571 572 573 574
}

func (scope *EvalScope) packageVarAddr(name string) (*Variable, error) {
	reader := scope.DwarfReader()
575 576 577 578 579 580 581 582 583 584 585
	for entry, err := reader.NextPackageVariable(); entry != nil; entry, err = reader.NextPackageVariable() {
		if err != nil {
			return nil, err
		}

		n, ok := entry.Val(dwarf.AttrName).(string)
		if !ok {
			continue
		}

		if n == name {
586
			return scope.extractVarInfoFromEntry(entry, reader)
587 588 589 590 591
		}
	}
	return nil, fmt.Errorf("could not find symbol value for %s", name)
}

592
func (v *Variable) structMember(memberName string) (*Variable, error) {
593 594 595 596
	if v.Unreadable != nil {
		return v.clone(), nil
	}
	structVar := v.maybeDereference()
597
	structVar.Name = v.Name
598 599
	if structVar.Unreadable != nil {
		return structVar, nil
600
	}
601 602

	switch t := structVar.RealType.(type) {
603 604 605 606
	case *dwarf.StructType:
		for _, field := range t.Field {
			if field.Name != memberName {
				continue
607
			}
L
Luke Hoban 已提交
608
			return structVar.toField(field)
609 610 611 612 613 614
		}
		// Check for embedded field only if field was
		// not a regular struct member
		for _, field := range t.Field {
			isEmbeddedStructMember :=
				(field.Type.String() == ("struct " + field.Name)) ||
D
Derek Parker 已提交
615 616 617
					(len(field.Name) > 1 &&
						field.Name[0] == '*' &&
						field.Type.String()[1:] == ("struct "+field.Name[1:]))
618 619 620 621 622 623
			if !isEmbeddedStructMember {
				continue
			}
			// Check for embedded field referenced by type name
			parts := strings.Split(field.Name, ".")
			if len(parts) > 1 && parts[1] == memberName {
L
Luke Hoban 已提交
624
				embeddedVar, err := structVar.toField(field)
625 626 627 628 629 630
				if err != nil {
					return nil, err
				}
				return embeddedVar, nil
			}
			// Recursively check for promoted fields on the embedded field
L
Luke Hoban 已提交
631
			embeddedVar, err := structVar.toField(field)
632 633 634 635 636 637 638
			if err != nil {
				return nil, err
			}
			embeddedVar.Name = structVar.Name
			embeddedField, err := embeddedVar.structMember(memberName)
			if embeddedField != nil {
				return embeddedField, nil
639 640
			}
		}
641 642
		return nil, fmt.Errorf("%s has no member %s", v.Name, memberName)
	default:
643 644 645
		if v.Name == "" {
			return nil, fmt.Errorf("type %s is not a struct", structVar.TypeString())
		}
D
Derek Parker 已提交
646
		return nil, fmt.Errorf("%s (type %s) is not a struct", v.Name, structVar.TypeString())
647
	}
648 649
}

650 651 652
// Extracts the name and type of a variable from a dwarf entry
// then executes the instructions given in the  DW_AT_location attribute to grab the variable's address
func (scope *EvalScope) extractVarInfoFromEntry(entry *dwarf.Entry, rdr *reader.Reader) (*Variable, error) {
E
epipho 已提交
653
	if entry == nil {
D
Derek Parker 已提交
654
		return nil, fmt.Errorf("invalid entry")
E
epipho 已提交
655 656 657 658 659 660 661 662
	}

	if entry.Tag != dwarf.TagFormalParameter && entry.Tag != dwarf.TagVariable {
		return nil, fmt.Errorf("invalid entry tag, only supports FormalParameter and Variable, got %s", entry.Tag.String())
	}

	n, ok := entry.Val(dwarf.AttrName).(string)
	if !ok {
D
Derek Parker 已提交
663
		return nil, fmt.Errorf("type assertion failed")
E
epipho 已提交
664 665 666 667
	}

	offset, ok := entry.Val(dwarf.AttrType).(dwarf.Offset)
	if !ok {
D
Derek Parker 已提交
668
		return nil, fmt.Errorf("type assertion failed")
E
epipho 已提交
669 670
	}

671
	t, err := scope.Type(offset)
E
epipho 已提交
672 673 674 675 676 677
	if err != nil {
		return nil, err
	}

	instructions, ok := entry.Val(dwarf.AttrLocation).([]byte)
	if !ok {
D
Derek Parker 已提交
678
		return nil, fmt.Errorf("type assertion failed")
E
epipho 已提交
679 680
	}

681
	addr, err := op.ExecuteStackProgram(scope.CFA, instructions)
E
epipho 已提交
682 683 684 685
	if err != nil {
		return nil, err
	}

686
	return scope.newVariable(n, uintptr(addr), t), nil
E
epipho 已提交
687 688
}

689
// If v is a pointer a new variable is returned containing the value pointed by v.
690 691 692 693
func (v *Variable) maybeDereference() *Variable {
	if v.Unreadable != nil {
		return v
	}
694

695
	switch t := v.RealType.(type) {
696
	case *dwarf.PtrType:
697 698
		ptrval, err := readUintRaw(v.mem, uintptr(v.Addr), t.ByteSize)
		r := v.newVariable("", uintptr(ptrval), t.Type)
699
		if err != nil {
700
			r.Unreadable = err
701 702
		}

703
		return r
704
	default:
705
		return v
706 707
	}
}
708

709
// Extracts the value of the variable at the given address.
710 711
func (v *Variable) loadValue() {
	v.loadValueInternal(0)
712 713
}

714
func (v *Variable) loadValueInternal(recurseLevel int) {
715
	if v.Unreadable != nil || v.loaded || (v.Addr == 0 && v.Base == 0) {
716 717
		return
	}
718

719 720
	v.loaded = true
	switch v.Kind {
A
aarzilli 已提交
721
	case reflect.Ptr, reflect.UnsafePointer:
722 723
		v.Len = 1
		v.Children = []Variable{*v.maybeDereference()}
724
		// Don't increase the recursion level when dereferencing pointers
725 726
		v.Children[0].loadValueInternal(recurseLevel)

A
aarzilli 已提交
727
	case reflect.Chan:
728 729 730
		sv := v.clone()
		sv.RealType = resolveTypedef(&(sv.RealType.(*dwarf.ChanType).TypedefType))
		sv = sv.maybeDereference()
A
aarzilli 已提交
731 732 733
		sv.loadValueInternal(recurseLevel)
		v.Children = sv.Children
		v.Len = sv.Len
734
		v.Base = sv.Addr
A
aarzilli 已提交
735

A
aarzilli 已提交
736
	case reflect.Map:
A
aarzilli 已提交
737 738 739
		if recurseLevel <= maxVariableRecurse {
			v.loadMap(recurseLevel)
		}
A
aarzilli 已提交
740

741
	case reflect.String:
742
		var val string
743
		val, v.Unreadable = readStringValue(v.mem, v.Base, v.Len)
744
		v.Value = constant.MakeString(val)
745 746 747 748 749

	case reflect.Slice, reflect.Array:
		v.loadArrayValues(recurseLevel)

	case reflect.Struct:
750
		v.mem = cacheMemory(v.mem, v.Addr, int(v.RealType.Size()))
751 752 753 754 755 756 757 758 759 760 761 762
		t := v.RealType.(*dwarf.StructType)
		v.Len = int64(len(t.Field))
		// Recursively call extractValue to grab
		// the value of all the members of the struct.
		if recurseLevel <= maxVariableRecurse {
			v.Children = make([]Variable, 0, len(t.Field))
			for i, field := range t.Field {
				f, _ := v.toField(field)
				v.Children = append(v.Children, *f)
				v.Children[i].Name = field.Name
				v.Children[i].loadValueInternal(recurseLevel + 1)
			}
763 764
		}

765 766 767
	case reflect.Interface:
		v.loadInterface(recurseLevel, true)

768 769 770
	case reflect.Complex64, reflect.Complex128:
		v.readComplex(v.RealType.(*dwarf.ComplexType).ByteSize)
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
771
		var val int64
772
		val, v.Unreadable = readIntRaw(v.mem, v.Addr, v.RealType.(*dwarf.IntType).ByteSize)
773
		v.Value = constant.MakeInt64(val)
774
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
775
		var val uint64
776
		val, v.Unreadable = readUintRaw(v.mem, v.Addr, v.RealType.(*dwarf.UintType).ByteSize)
777 778
		v.Value = constant.MakeUint64(val)

779
	case reflect.Bool:
780
		val, err := v.mem.readMemory(v.Addr, 1)
781 782
		v.Unreadable = err
		if err == nil {
783
			v.Value = constant.MakeBool(val[0] != 0)
784
		}
785
	case reflect.Float32, reflect.Float64:
786 787 788
		var val float64
		val, v.Unreadable = v.readFloatRaw(v.RealType.(*dwarf.FloatType).ByteSize)
		v.Value = constant.MakeFloat64(val)
789 790
	case reflect.Func:
		v.readFunctionPtr()
791
	default:
792
		v.Unreadable = fmt.Errorf("unknown or unsupported kind: \"%s\"", v.Kind.String())
793 794 795
	}
}

A
aarzilli 已提交
796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813
func (v *Variable) setValue(y *Variable) error {
	var err error
	switch v.Kind {
	case reflect.Float32, reflect.Float64:
		f, _ := constant.Float64Val(y.Value)
		err = v.writeFloatRaw(f, v.RealType.Size())
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		n, _ := constant.Int64Val(y.Value)
		err = v.writeUint(uint64(n), v.RealType.Size())
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
		n, _ := constant.Uint64Val(y.Value)
		err = v.writeUint(n, v.RealType.Size())
	case reflect.Bool:
		err = v.writeBool(constant.BoolVal(y.Value))
	case reflect.Complex64, reflect.Complex128:
		real, _ := constant.Float64Val(constant.Real(y.Value))
		imag, _ := constant.Float64Val(constant.Imag(y.Value))
		err = v.writeComplex(real, imag, v.RealType.Size())
814
	default:
A
aarzilli 已提交
815
		fmt.Printf("default\n")
816 817
		if t, isptr := v.RealType.(*dwarf.PtrType); isptr {
			err = v.writeUint(uint64(y.Children[0].Addr), int64(t.ByteSize))
A
aarzilli 已提交
818 819 820
		} else {
			return fmt.Errorf("can not set variables of type %s (not implemented)", v.Kind.String())
		}
821
	}
A
aarzilli 已提交
822 823

	return err
824 825
}

826
func readStringInfo(mem memoryReadWriter, arch Arch, addr uintptr) (uintptr, int64, error) {
827 828 829
	// string data structure is always two ptrs in size. Addr, followed by len
	// http://research.swtch.com/godata

830 831
	mem = cacheMemory(mem, addr, arch.PtrSize()*2)

832
	// read len
833
	val, err := mem.readMemory(addr+uintptr(arch.PtrSize()), arch.PtrSize())
834
	if err != nil {
A
aarzilli 已提交
835
		return 0, 0, fmt.Errorf("could not read string len %s", err)
836
	}
837
	strlen := int64(binary.LittleEndian.Uint64(val))
838
	if strlen < 0 {
A
aarzilli 已提交
839
		return 0, 0, fmt.Errorf("invalid length: %d", strlen)
840
	}
841 842

	// read addr
843
	val, err = mem.readMemory(addr, arch.PtrSize())
844
	if err != nil {
A
aarzilli 已提交
845
		return 0, 0, fmt.Errorf("could not read string pointer %s", err)
846 847
	}
	addr = uintptr(binary.LittleEndian.Uint64(val))
D
Derek Parker 已提交
848
	if addr == 0 {
A
aarzilli 已提交
849
		return 0, 0, nil
D
Derek Parker 已提交
850
	}
D
Derek Parker 已提交
851

A
aarzilli 已提交
852 853 854
	return addr, strlen, nil
}

855
func readStringValue(mem memoryReadWriter, addr uintptr, strlen int64) (string, error) {
A
aarzilli 已提交
856 857 858 859 860
	count := strlen
	if count > maxArrayValues {
		count = maxArrayValues
	}

861
	val, err := mem.readMemory(addr, int(count))
862
	if err != nil {
A
aarzilli 已提交
863
		return "", fmt.Errorf("could not read string at %#v due to %s", addr, err)
864 865
	}

866 867
	retstr := *(*string)(unsafe.Pointer(&val))

A
aarzilli 已提交
868 869 870
	return retstr, nil
}

871 872
func readString(mem memoryReadWriter, arch Arch, addr uintptr) (string, int64, error) {
	addr, strlen, err := readStringInfo(mem, arch, addr)
A
aarzilli 已提交
873 874 875 876
	if err != nil {
		return "", 0, err
	}

877
	retstr, err := readStringValue(mem, addr, strlen)
A
aarzilli 已提交
878
	return retstr, strlen, err
879 880
}

881
func (v *Variable) loadSliceInfo(t *dwarf.SliceType) {
882 883
	v.mem = cacheMemory(v.mem, v.Addr, int(t.Size()))

884
	var err error
E
epipho 已提交
885 886 887
	for _, f := range t.Field {
		switch f.Name {
		case "array":
888
			var base uint64
889
			base, err = readUintRaw(v.mem, uintptr(int64(v.Addr)+f.ByteOffset), f.Type.Size())
890
			if err == nil {
891
				v.Base = uintptr(base)
892 893 894
				// Dereference array type to get value type
				ptrType, ok := f.Type.(*dwarf.PtrType)
				if !ok {
895 896
					v.Unreadable = fmt.Errorf("Invalid type %s in slice array", f.Type)
					return
897 898
				}
				v.fieldType = ptrType.Type
E
epipho 已提交
899 900
			}
		case "len":
901 902 903
			lstrAddr, _ := v.toField(f)
			lstrAddr.loadValue()
			err = lstrAddr.Unreadable
904
			if err == nil {
905
				v.Len, _ = constant.Int64Val(lstrAddr.Value)
E
epipho 已提交
906 907
			}
		case "cap":
908 909 910
			cstrAddr, _ := v.toField(f)
			cstrAddr.loadValue()
			err = cstrAddr.Unreadable
911
			if err == nil {
912
				v.Cap, _ = constant.Int64Val(cstrAddr.Value)
E
epipho 已提交
913 914
			}
		}
915 916 917 918
		if err != nil {
			v.Unreadable = err
			return
		}
919 920
	}

921
	v.stride = v.fieldType.Size()
922 923
	if t, ok := v.fieldType.(*dwarf.PtrType); ok {
		v.stride = t.ByteSize
924
	}
925

926
	return
927 928
}

929 930 931 932
func (v *Variable) loadArrayValues(recurseLevel int) {
	if v.Unreadable != nil {
		return
	}
933 934 935 936
	if v.Len < 0 {
		v.Unreadable = errors.New("Negative array length")
		return
	}
937

938 939 940 941 942
	count := v.Len
	// Cap number of elements
	if count > maxArrayValues {
		count = maxArrayValues
	}
943

944
	if v.stride < maxArrayStridePrefetch {
945
		v.mem = cacheMemory(v.mem, v.Base, int(v.stride*count))
946
	}
947

948 949 950
	errcount := 0

	for i := int64(0); i < count; i++ {
951
		fieldvar := v.newVariable("", uintptr(int64(v.Base)+(i*v.stride)), v.fieldType)
952 953 954
		fieldvar.loadValueInternal(recurseLevel + 1)

		if fieldvar.Unreadable != nil {
955
			errcount++
E
epipho 已提交
956
		}
957

958
		v.Children = append(v.Children, *fieldvar)
959 960 961
		if errcount > maxErrCount {
			break
		}
962 963 964
	}
}

965
func (v *Variable) readComplex(size int64) {
966 967 968 969 970 971 972
	var fs int64
	switch size {
	case 8:
		fs = 4
	case 16:
		fs = 8
	default:
973 974
		v.Unreadable = fmt.Errorf("invalid size (%d) for complex type", size)
		return
975
	}
976 977 978

	ftyp := &dwarf.FloatType{BasicType: dwarf.BasicType{CommonType: dwarf.CommonType{ByteSize: fs, Name: fmt.Sprintf("float%d", fs)}, BitSize: fs * 8, BitOffset: 0}}

979 980
	realvar := v.newVariable("real", v.Addr, ftyp)
	imagvar := v.newVariable("imaginary", v.Addr+uintptr(fs), ftyp)
981 982
	realvar.loadValue()
	imagvar.loadValue()
A
aarzilli 已提交
983
	v.Value = constant.BinaryOp(realvar.Value, token.ADD, constant.MakeImag(imagvar.Value))
984 985
}

A
aarzilli 已提交
986 987
func (v *Variable) writeComplex(real, imag float64, size int64) error {
	err := v.writeFloatRaw(real, int64(size/2))
988 989 990 991 992 993 994 995
	if err != nil {
		return err
	}
	imagaddr := *v
	imagaddr.Addr += uintptr(size / 2)
	return imagaddr.writeFloatRaw(imag, int64(size/2))
}

996
func readIntRaw(mem memoryReadWriter, addr uintptr, size int64) (int64, error) {
E
epipho 已提交
997
	var n int64
998

999
	val, err := mem.readMemory(addr, int(size))
1000
	if err != nil {
D
Derek Parker 已提交
1001
		return 0, err
1002 1003
	}

1004 1005
	switch size {
	case 1:
E
epipho 已提交
1006
		n = int64(val[0])
1007
	case 2:
E
epipho 已提交
1008
		n = int64(binary.LittleEndian.Uint16(val))
1009
	case 4:
E
epipho 已提交
1010
		n = int64(binary.LittleEndian.Uint32(val))
1011
	case 8:
E
epipho 已提交
1012
		n = int64(binary.LittleEndian.Uint64(val))
1013
	}
1014

D
Derek Parker 已提交
1015
	return n, nil
E
epipho 已提交
1016 1017
}

A
aarzilli 已提交
1018
func (v *Variable) writeUint(value uint64, size int64) error {
1019 1020 1021 1022
	val := make([]byte, size)

	switch size {
	case 1:
A
aarzilli 已提交
1023
		val[0] = byte(value)
1024
	case 2:
A
aarzilli 已提交
1025
		binary.LittleEndian.PutUint16(val, uint16(value))
1026
	case 4:
A
aarzilli 已提交
1027
		binary.LittleEndian.PutUint32(val, uint32(value))
1028
	case 8:
A
aarzilli 已提交
1029
		binary.LittleEndian.PutUint64(val, uint64(value))
1030 1031
	}

1032
	_, err := v.mem.writeMemory(v.Addr, val)
1033 1034 1035
	return err
}

1036
func readUintRaw(mem memoryReadWriter, addr uintptr, size int64) (uint64, error) {
E
epipho 已提交
1037 1038
	var n uint64

1039
	val, err := mem.readMemory(addr, int(size))
E
epipho 已提交
1040
	if err != nil {
D
Derek Parker 已提交
1041
		return 0, err
E
epipho 已提交
1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054
	}

	switch size {
	case 1:
		n = uint64(val[0])
	case 2:
		n = uint64(binary.LittleEndian.Uint16(val))
	case 4:
		n = uint64(binary.LittleEndian.Uint32(val))
	case 8:
		n = uint64(binary.LittleEndian.Uint64(val))
	}

D
Derek Parker 已提交
1055
	return n, nil
1056 1057
}

1058
func (v *Variable) readFloatRaw(size int64) (float64, error) {
1059
	val, err := v.mem.readMemory(v.Addr, int(size))
1060
	if err != nil {
1061
		return 0.0, err
1062 1063 1064
	}
	buf := bytes.NewBuffer(val)

D
Derek Parker 已提交
1065 1066 1067 1068
	switch size {
	case 4:
		n := float32(0)
		binary.Read(buf, binary.LittleEndian, &n)
1069
		return float64(n), nil
D
Derek Parker 已提交
1070 1071 1072
	case 8:
		n := float64(0)
		binary.Read(buf, binary.LittleEndian, &n)
1073
		return n, nil
D
Derek Parker 已提交
1074 1075
	}

1076
	return 0.0, fmt.Errorf("could not read float")
1077 1078
}

1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090
func (v *Variable) writeFloatRaw(f float64, size int64) error {
	buf := bytes.NewBuffer(make([]byte, 0, size))

	switch size {
	case 4:
		n := float32(f)
		binary.Write(buf, binary.LittleEndian, n)
	case 8:
		n := float64(f)
		binary.Write(buf, binary.LittleEndian, n)
	}

1091
	_, err := v.mem.writeMemory(v.Addr, buf.Bytes())
1092 1093 1094
	return err
}

A
aarzilli 已提交
1095
func (v *Variable) writeBool(value bool) error {
1096
	val := []byte{0}
A
aarzilli 已提交
1097
	val[0] = *(*byte)(unsafe.Pointer(&value))
1098
	_, err := v.mem.writeMemory(v.Addr, val)
1099 1100 1101
	return err
}

1102
func (v *Variable) readFunctionPtr() {
1103
	val, err := v.mem.readMemory(v.Addr, v.dbp.arch.PtrSize())
1104
	if err != nil {
1105 1106
		v.Unreadable = err
		return
1107 1108 1109
	}

	// dereference pointer to find function pc
1110 1111
	fnaddr := uintptr(binary.LittleEndian.Uint64(val))
	if fnaddr == 0 {
1112
		v.Base = 0
A
aarzilli 已提交
1113
		v.Value = constant.MakeString("")
1114
		return
1115
	}
1116

1117
	val, err = v.mem.readMemory(fnaddr, v.dbp.arch.PtrSize())
1118
	if err != nil {
1119 1120
		v.Unreadable = err
		return
1121 1122
	}

1123 1124
	v.Base = uintptr(binary.LittleEndian.Uint64(val))
	fn := v.dbp.goSymTable.PCToFunc(uint64(v.Base))
1125
	if fn == nil {
1126
		v.Unreadable = fmt.Errorf("could not find function for %#v", v.Base)
1127
		return
1128 1129
	}

1130
	v.Value = constant.MakeString(fn.Name)
1131 1132
}

A
aarzilli 已提交
1133 1134 1135 1136 1137 1138 1139 1140
func (v *Variable) loadMap(recurseLevel int) {
	it := v.mapIterator()
	if it == nil {
		return
	}

	for skip := 0; skip < v.mapSkip; skip++ {
		if ok := it.next(); !ok {
1141
			v.Unreadable = fmt.Errorf("map index out of bounds")
A
aarzilli 已提交
1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
			return
		}
	}

	count := 0
	errcount := 0
	for it.next() {
		if count >= maxArrayValues {
			break
		}
		key := it.key()
		val := it.value()
A
aarzilli 已提交
1154 1155
		key.loadValueInternal(recurseLevel + 1)
		val.loadValueInternal(recurseLevel + 1)
A
aarzilli 已提交
1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186
		if key.Unreadable != nil || val.Unreadable != nil {
			errcount++
		}
		v.Children = append(v.Children, *key)
		v.Children = append(v.Children, *val)
		count++
		if errcount > maxErrCount {
			break
		}
	}
}

type mapIterator struct {
	v          *Variable
	numbuckets uint64
	oldmask    uint64
	buckets    *Variable
	oldbuckets *Variable
	b          *Variable
	bidx       uint64

	tophashes *Variable
	keys      *Variable
	values    *Variable
	overflow  *Variable

	idx int64
}

// Code derived from go/src/runtime/hashmap.go
func (v *Variable) mapIterator() *mapIterator {
1187 1188 1189
	sv := v.clone()
	sv.RealType = resolveTypedef(&(sv.RealType.(*dwarf.MapType).TypedefType))
	sv = sv.maybeDereference()
1190
	v.Base = sv.Addr
A
aarzilli 已提交
1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204

	maptype, ok := sv.RealType.(*dwarf.StructType)
	if !ok {
		v.Unreadable = fmt.Errorf("wrong real type for map")
		return nil
	}

	it := &mapIterator{v: v, bidx: 0, b: nil, idx: 0}

	if sv.Addr == 0 {
		it.numbuckets = 0
		return it
	}

1205
	v.mem = cacheMemory(v.mem, v.Base, int(v.RealType.Size()))
1206

A
aarzilli 已提交
1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
	for _, f := range maptype.Field {
		var err error
		field, _ := sv.toField(f)
		switch f.Name {
		case "count":
			v.Len, err = field.asInt()
		case "B":
			var b uint64
			b, err = field.asUint()
			it.numbuckets = 1 << b
			it.oldmask = (1 << (b - 1)) - 1
		case "buckets":
			it.buckets = field.maybeDereference()
		case "oldbuckets":
			it.oldbuckets = field.maybeDereference()
		}
		if err != nil {
			v.Unreadable = err
			return nil
		}
	}

	return it
}

func (it *mapIterator) nextBucket() bool {
	if it.overflow != nil && it.overflow.Addr > 0 {
		it.b = it.overflow
	} else {
		it.b = nil

		for it.bidx < it.numbuckets {
			it.b = it.buckets.clone()
			it.b.Addr += uintptr(uint64(it.buckets.DwarfType.Size()) * it.bidx)

			if it.oldbuckets.Addr <= 0 {
				break
			}

			// if oldbuckets is not nil we are iterating through a map that is in
			// the middle of a grow.
			// if the bucket we are looking at hasn't been filled in we iterate
			// instead through its corresponding "oldbucket" (i.e. the bucket the
			// elements of this bucket are coming from) but only if this is the first
			// of the two buckets being created from the same oldbucket (otherwise we
			// would print some keys twice)

			oldbidx := it.bidx & it.oldmask
			oldb := it.oldbuckets.clone()
			oldb.Addr += uintptr(uint64(it.oldbuckets.DwarfType.Size()) * oldbidx)

			if mapEvacuated(oldb) {
				break
			}

			if oldbidx == it.bidx {
				it.b = oldb
				break
			}

			// oldbucket origin for current bucket has not been evacuated but we have already
			// iterated over it so we should just skip it
			it.b = nil
			it.bidx++
		}

		if it.b == nil {
			return false
		}
		it.bidx++
	}

	if it.b.Addr <= 0 {
		return false
	}

1283 1284
	it.b.mem = cacheMemory(it.b.mem, it.b.Addr, int(it.b.RealType.Size()))

A
aarzilli 已提交
1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382
	it.tophashes = nil
	it.keys = nil
	it.values = nil
	it.overflow = nil

	for _, f := range it.b.DwarfType.(*dwarf.StructType).Field {
		field, err := it.b.toField(f)
		if err != nil {
			it.v.Unreadable = err
			return false
		}
		if field.Unreadable != nil {
			it.v.Unreadable = field.Unreadable
			return false
		}

		switch f.Name {
		case "tophash":
			it.tophashes = field
		case "keys":
			it.keys = field
		case "values":
			it.values = field
		case "overflow":
			it.overflow = field.maybeDereference()
		}
	}

	// sanity checks
	if it.tophashes == nil || it.keys == nil || it.values == nil {
		it.v.Unreadable = fmt.Errorf("malformed map type")
		return false
	}

	if it.tophashes.Kind != reflect.Array || it.keys.Kind != reflect.Array || it.values.Kind != reflect.Array {
		it.v.Unreadable = fmt.Errorf("malformed map type: keys, values or tophash of a bucket is not an array")
		return false
	}

	if it.tophashes.Len != it.keys.Len || it.tophashes.Len != it.values.Len {
		it.v.Unreadable = fmt.Errorf("malformed map type: inconsistent array length in bucket")
		return false
	}

	return true
}

func (it *mapIterator) next() bool {
	for {
		if it.b == nil || it.idx >= it.tophashes.Len {
			r := it.nextBucket()
			if !r {
				return false
			}
			it.idx = 0
		}
		tophash, _ := it.tophashes.sliceAccess(int(it.idx))
		h, err := tophash.asUint()
		if err != nil {
			it.v.Unreadable = fmt.Errorf("unreadable tophash: %v", err)
			return false
		}
		it.idx++
		if h != hashTophashEmpty {
			return true
		}
	}
}

func (it *mapIterator) key() *Variable {
	k, _ := it.keys.sliceAccess(int(it.idx - 1))
	return k
}

func (it *mapIterator) value() *Variable {
	v, _ := it.values.sliceAccess(int(it.idx - 1))
	return v
}

func mapEvacuated(b *Variable) bool {
	if b.Addr == 0 {
		return true
	}
	for _, f := range b.DwarfType.(*dwarf.StructType).Field {
		if f.Name != "tophash" {
			continue
		}
		tophashes, _ := b.toField(f)
		tophash0var, _ := tophashes.sliceAccess(0)
		tophash0, err := tophash0var.asUint()
		if err != nil {
			return true
		}
		return tophash0 > hashTophashEmpty && tophash0 < hashMinTopHash
	}
	return true
}

1383 1384 1385 1386
func (v *Variable) loadInterface(recurseLevel int, loadData bool) {
	var typestring, data *Variable
	isnil := false

1387 1388
	v.mem = cacheMemory(v.mem, v.Addr, int(v.RealType.Size()))

1389 1390 1391
	ityp := resolveTypedef(&v.RealType.(*dwarf.InterfaceType).TypedefType).(*dwarf.StructType)

	for _, f := range ityp.Field {
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431
		switch f.Name {
		case "tab": // for runtime.iface
			tab, _ := v.toField(f)
			_type, err := tab.structMember("_type")
			if err != nil {
				_, isnil = err.(*IsNilErr)
				if !isnil {
					v.Unreadable = fmt.Errorf("invalid interface type: %v", err)
					return
				}
			} else {
				typestring, err = _type.structMember("_string")
				if err != nil {
					v.Unreadable = fmt.Errorf("invalid interface type: %v", err)
					return
				}
				typestring = typestring.maybeDereference()
			}
		case "_type": // for runtime.eface
			var err error
			_type, _ := v.toField(f)
			typestring, err = _type.structMember("_string")
			if err != nil {
				_, isnil = err.(*IsNilErr)
				if !isnil {
					v.Unreadable = fmt.Errorf("invalid interface type: %v", err)
					return
				}
			} else {
				typestring = typestring.maybeDereference()
			}
		case "data":
			data, _ = v.toField(f)
		}
	}

	if isnil {
		// interface to nil
		data = data.maybeDereference()
		v.Children = []Variable{*data}
A
aarzilli 已提交
1432 1433 1434
		if loadData {
			v.Children[0].loadValueInternal(recurseLevel)
		}
1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453
		return
	}

	if typestring == nil || data == nil || typestring.Addr == 0 || typestring.Kind != reflect.String {
		v.Unreadable = fmt.Errorf("invalid interface type")
		return
	}
	typestring.loadValue()
	if typestring.Unreadable != nil {
		v.Unreadable = fmt.Errorf("invalid interface type: %v", typestring.Unreadable)
		return
	}

	t, err := parser.ParseExpr(constant.StringVal(typestring.Value))
	if err != nil {
		v.Unreadable = fmt.Errorf("invalid interface type, unparsable data type: %v", err)
		return
	}

1454
	typ, err := v.dbp.findTypeExpr(t)
1455
	if err != nil {
1456
		v.Unreadable = fmt.Errorf("interface type \"%s\" not found for 0x%x: %v", constant.StringVal(typestring.Value), data.Addr, err)
1457 1458 1459
		return
	}

1460 1461 1462
	realtyp := resolveTypedef(typ)
	if _, isptr := realtyp.(*dwarf.PtrType); !isptr {
		// interface to non-pointer types are pointers even if the type says otherwise
1463
		typ = v.dbp.pointerTo(typ)
1464 1465
	}

1466
	data = data.newVariable("data", data.Addr, typ)
1467 1468 1469

	v.Children = []Variable{*data}
	if loadData {
1470
		v.Children[0].loadValueInternal(recurseLevel)
1471 1472 1473 1474
	}
	return
}

E
epipho 已提交
1475
// Fetches all variables of a specific type in the current function scope
1476 1477
func (scope *EvalScope) variablesByTag(tag dwarf.Tag) ([]*Variable, error) {
	reader := scope.DwarfReader()
E
epipho 已提交
1478

1479
	_, err := reader.SeekToFunction(scope.PC)
1480
	if err != nil {
E
epipho 已提交
1481 1482 1483
		return nil, err
	}

D
Derek Parker 已提交
1484
	var vars []*Variable
1485
	for entry, err := reader.NextScopeVariable(); entry != nil; entry, err = reader.NextScopeVariable() {
E
epipho 已提交
1486 1487 1488 1489 1490
		if err != nil {
			return nil, err
		}

		if entry.Tag == tag {
1491
			val, err := scope.extractVariableFromEntry(entry)
E
epipho 已提交
1492
			if err != nil {
E
epipho 已提交
1493 1494
				// skip variables that we can't parse yet
				continue
E
epipho 已提交
1495 1496 1497 1498 1499 1500 1501 1502
			}

			vars = append(vars, val)
		}
	}

	return vars, nil
}