2015-01-14 02:37:10 +00:00
|
|
|
package proctl
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
|
|
|
|
sys "golang.org/x/sys/unix"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Not actually used, but necessary
|
|
|
|
// to be defined.
|
2015-03-28 01:12:07 +00:00
|
|
|
type OSSpecificDetails struct {
|
|
|
|
registers sys.PtraceRegs
|
|
|
|
}
|
2015-01-14 02:37:10 +00:00
|
|
|
|
|
|
|
func (t *ThreadContext) Halt() error {
|
|
|
|
if stopped(t.Id) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
err := sys.Tgkill(t.Process.Pid, t.Id, sys.SIGSTOP)
|
|
|
|
if err != nil {
|
2015-02-27 23:11:13 +00:00
|
|
|
return fmt.Errorf("Halt err %s %d", err, t.Id)
|
2015-01-14 02:37:10 +00:00
|
|
|
}
|
2015-02-28 03:35:26 +00:00
|
|
|
_, _, err = wait(t.Id, 0)
|
2015-01-14 02:37:10 +00:00
|
|
|
if err != nil {
|
2015-02-27 23:11:13 +00:00
|
|
|
return fmt.Errorf("wait err %s %d", err, t.Id)
|
2015-01-14 02:37:10 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-03-01 03:14:22 +00:00
|
|
|
func (t *ThreadContext) resume() error {
|
2015-02-27 23:11:13 +00:00
|
|
|
return PtraceCont(t.Id, 0)
|
2015-01-14 02:37:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (t *ThreadContext) singleStep() error {
|
|
|
|
err := sys.PtraceSingleStep(t.Id)
|
2015-02-27 23:11:13 +00:00
|
|
|
if err != nil {
|
2015-01-14 02:37:10 +00:00
|
|
|
return err
|
|
|
|
}
|
2015-02-27 23:11:13 +00:00
|
|
|
_, _, err = wait(t.Id, 0)
|
2015-01-14 02:37:10 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2015-02-27 23:11:13 +00:00
|
|
|
func (t *ThreadContext) blocked() bool {
|
|
|
|
// TODO(dp) cache the func pc to remove this lookup
|
2015-04-23 15:40:33 +00:00
|
|
|
pc, _ := t.PC()
|
2015-04-03 16:10:35 +00:00
|
|
|
fn := t.Process.goSymTable.PCToFunc(pc)
|
2015-02-28 14:05:37 +00:00
|
|
|
if fn != nil && ((fn.Name == "runtime.futex") || (fn.Name == "runtime.usleep") || (fn.Name == "runtime.clone")) {
|
2015-02-27 23:11:13 +00:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2015-04-23 16:40:20 +00:00
|
|
|
func (thread *ThreadContext) saveRegisters() error {
|
|
|
|
return sys.PtraceGetRegs(thread.Id, &thread.os.registers)
|
2015-01-14 02:37:10 +00:00
|
|
|
}
|
|
|
|
|
2015-04-23 16:40:20 +00:00
|
|
|
func (thread *ThreadContext) restoreRegisters() error {
|
|
|
|
return sys.PtraceSetRegs(thread.Id, &thread.os.registers)
|
2015-01-14 02:37:10 +00:00
|
|
|
}
|
2015-03-28 01:12:07 +00:00
|
|
|
|
2015-04-23 16:40:20 +00:00
|
|
|
func writeMemory(thread *ThreadContext, addr uintptr, data []byte) (int, error) {
|
|
|
|
return sys.PtracePokeData(thread.Id, addr, data)
|
2015-03-28 01:12:07 +00:00
|
|
|
}
|
|
|
|
|
2015-04-23 16:40:20 +00:00
|
|
|
func readMemory(thread *ThreadContext, addr uintptr, data []byte) (int, error) {
|
|
|
|
return sys.PtracePeekData(thread.Id, addr, data)
|
2015-03-28 01:12:07 +00:00
|
|
|
}
|