111 lines
1.9 KiB
Go
111 lines
1.9 KiB
Go
package frame
|
|
|
|
import (
|
|
"debug/elf"
|
|
"debug/gosym"
|
|
"encoding/binary"
|
|
"os"
|
|
"path/filepath"
|
|
"syscall"
|
|
"testing"
|
|
|
|
"github.com/derekparker/dbg/_helper"
|
|
"github.com/derekparker/dbg/proctl"
|
|
)
|
|
|
|
var testfile string
|
|
|
|
func init() {
|
|
testfile, _ = filepath.Abs("../_fixtures/testprog")
|
|
}
|
|
|
|
func parseGoSym(t *testing.T, exe *elf.File) *gosym.Table {
|
|
symdat, err := exe.Section(".gosymtab").Data()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
pclndat, err := exe.Section(".gopclntab").Data()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
pcln := gosym.NewLineTable(pclndat, exe.Section(".text").Addr)
|
|
tab, err := gosym.NewTable(symdat, pcln)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return tab
|
|
}
|
|
|
|
func gosymData(t *testing.T) *gosym.Table {
|
|
f, err := os.Open(testfile)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
e, err := elf.NewFile(f)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return parseGoSym(t, e)
|
|
}
|
|
|
|
func TestFindReturnAddress(t *testing.T) {
|
|
var (
|
|
dbframe = grabDebugFrameSection(testfile, t)
|
|
fdes = Parse(dbframe)
|
|
gsd = gosymData(t)
|
|
)
|
|
|
|
helper.WithTestProcess("testprog", t, func(p *proctl.DebuggedProcess) {
|
|
testsourcefile := testfile + ".go"
|
|
start, _, err := gsd.LineToPC(testsourcefile, 9)
|
|
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)
|
|
}
|
|
|
|
end, _, err := gsd.LineToPC(testsourcefile, 19)
|
|
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 != end {
|
|
t.Fatalf("return address not found correctly, expected %#v got %#v", end, ret)
|
|
}
|
|
})
|
|
}
|