mirror of https://github.com/beego/bee.git
97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
package proc
|
||
|
||
import (
|
||
"encoding/binary"
|
||
"syscall"
|
||
"unsafe"
|
||
|
||
sys "golang.org/x/sys/unix"
|
||
)
|
||
|
||
// PtraceAttach executes the sys.PtraceAttach call.
|
||
func PtraceAttach(pid int) error {
|
||
return sys.PtraceAttach(pid)
|
||
}
|
||
|
||
// PtraceDetach calls ptrace(PTRACE_DETACH).
|
||
func PtraceDetach(tid, sig int) error {
|
||
_, _, err := sys.Syscall6(sys.SYS_PTRACE, sys.PTRACE_DETACH, uintptr(tid), 1, uintptr(sig), 0, 0)
|
||
if err != syscall.Errno(0) {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// PtraceCont executes ptrace PTRACE_CONT
|
||
func PtraceCont(tid, sig int) error {
|
||
return sys.PtraceCont(tid, sig)
|
||
}
|
||
|
||
// PtraceSingleStep executes ptrace PTRACE_SINGLE_STEP.
|
||
func PtraceSingleStep(tid int) error {
|
||
return sys.PtraceSingleStep(tid)
|
||
}
|
||
|
||
// PtracePokeUser execute ptrace PTRACE_POKE_USER.
|
||
func PtracePokeUser(tid int, off, addr uintptr) error {
|
||
_, _, err := sys.Syscall6(sys.SYS_PTRACE, sys.PTRACE_POKEUSR, uintptr(tid), uintptr(off), uintptr(addr), 0, 0)
|
||
if err != syscall.Errno(0) {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// PtracePeekUser execute ptrace PTRACE_PEEK_USER.
|
||
func PtracePeekUser(tid int, off uintptr) (uintptr, error) {
|
||
var val uintptr
|
||
_, _, err := syscall.Syscall6(syscall.SYS_PTRACE, syscall.PTRACE_PEEKUSR, uintptr(tid), uintptr(off), uintptr(unsafe.Pointer(&val)), 0, 0)
|
||
if err != syscall.Errno(0) {
|
||
return 0, err
|
||
}
|
||
return val, nil
|
||
}
|
||
|
||
// PtraceGetRegset returns floating point registers of the specified thread
|
||
// using PTRACE.
|
||
// See amd64_linux_fetch_inferior_registers in gdb/amd64-linux-nat.c.html
|
||
// and amd64_supply_xsave in gdb/amd64-tdep.c.html
|
||
// and Section 13.1 (and following) of Intel® 64 and IA-32 Architectures Software Developer’s Manual, Volume 1: Basic Architecture
|
||
func PtraceGetRegset(tid int) (regset PtraceXsave, err error) {
|
||
_, _, err = syscall.Syscall6(syscall.SYS_PTRACE, sys.PTRACE_GETFPREGS, uintptr(tid), uintptr(0), uintptr(unsafe.Pointer(®set.PtraceFpRegs)), 0, 0)
|
||
if err == syscall.Errno(0) {
|
||
err = nil
|
||
}
|
||
|
||
var xstateargs [_X86_XSTATE_MAX_SIZE]byte
|
||
iov := sys.Iovec{Base: &xstateargs[0], Len: _X86_XSTATE_MAX_SIZE}
|
||
_, _, err = syscall.Syscall6(syscall.SYS_PTRACE, sys.PTRACE_GETREGSET, uintptr(tid), _NT_X86_XSTATE, uintptr(unsafe.Pointer(&iov)), 0, 0)
|
||
if err != syscall.Errno(0) {
|
||
return
|
||
} else {
|
||
err = nil
|
||
}
|
||
|
||
if _XSAVE_HEADER_START+_XSAVE_HEADER_LEN >= iov.Len {
|
||
return
|
||
}
|
||
xsaveheader := xstateargs[_XSAVE_HEADER_START : _XSAVE_HEADER_START+_XSAVE_HEADER_LEN]
|
||
xstate_bv := binary.LittleEndian.Uint64(xsaveheader[0:8])
|
||
xcomp_bv := binary.LittleEndian.Uint64(xsaveheader[8:16])
|
||
|
||
if xcomp_bv&(1<<63) != 0 {
|
||
// compact format not supported
|
||
return
|
||
}
|
||
|
||
if xstate_bv&(1<<2) == 0 {
|
||
// AVX state not present
|
||
return
|
||
}
|
||
|
||
avxstate := xstateargs[_XSAVE_EXTENDED_REGION_START:iov.Len]
|
||
regset.AvxState = true
|
||
copy(regset.YmmSpace[:], avxstate[:len(regset.YmmSpace)])
|
||
|
||
return
|
||
}
|