delve/dwarf/frame/frame_table_test.go

package frame_test

import (
	"encoding/binary"
	"path/filepath"
	"syscall"
	"testing"

	"github.com/derekparker/dbg/_helper"
	"github.com/derekparker/dbg/dwarf/_helper"
	"github.com/derekparker/dbg/dwarf/frame"
	"github.com/derekparker/dbg/proctl"
)

func TestFindReturnAddress(t *testing.T) {
	var testfile, _ = filepath.Abs("../../_fixtures/testnextprog")

	helper.WithTestProcess(testfile, t, func(p *proctl.DebuggedProcess) {
		var (
			dbframe = dwarfhelper.GrabDebugFrameSection(testfile, t)
			fdes    = frame.Parse(dbframe)
			gsd     = dwarfhelper.GosymData(testfile, t)
		)

		testsourcefile := testfile + ".go"
		start, _, err := gsd.LineToPC(testsourcefile, 22)
		if err != nil {
			t.Fatal(err)
		}

		_, err = p.Break(uintptr(start))
		if err != nil {
			t.Fatal(err)
		}

		err = p.Continue()
		if err != nil {
			t.Fatal(err)
		}

		regs, err := p.Registers()
		if err != nil {
			t.Fatal(err)
		}

		fde, err := fdes.FDEForPC(start)
		if err != nil {
			t.Fatal(err)
		}

		ret := fde.ReturnAddressOffset(start)
		if err != nil {
			t.Fatal(err)
		}

		addr := uint64(int64(regs.Rsp) + ret)
		data := make([]byte, 8)

		syscall.PtracePeekText(p.Pid, uintptr(addr), data)
		addr = binary.LittleEndian.Uint64(data)

		if addr != 0x400dff {
			t.Fatalf("return address not found correctly, expected %#v got %#v", uintptr(0x400dff), addr)
		}
	})
}