Python set_ptracer Examples

Example #1

File: test_prctl.py Project: wqh872081365/python-prctl

 def test_ptracer(self):
     """Test manipulation of the ptracer setting"""
     if not os.path.exists('/proc/sys/kernel/yama'):
         return
     self.assertEqual(prctl.get_ptracer(), os.getppid())
     prctl.set_ptracer(1)
     self.assertEqual(prctl.get_ptracer(), 1)
     new_pid = os.fork()
     if new_pid:
         os.waitpid(new_pid, 0)
     else:
         os._exit(0)
     self.assertRaises(OSError, prctl.set_ptracer, new_pid)

Example #2

File: test_prctl.py Project: bugchecker/python-prctl

 def test_ptracer(self):
     """Test manipulation of the ptracer setting"""
     if not os.path.exists('/proc/sys/kernel/yama'):
         return
     self.assertEqual(prctl.get_ptracer(), os.getppid())
     prctl.set_ptracer(1)
     self.assertEqual(prctl.get_ptracer(), 1)
     new_pid = os.fork()
     if new_pid:
         os.waitpid(new_pid, 0)
     else:
         os._exit(0)
     self.assertRaises(OSError, prctl.set_ptracer, new_pid)

Example #3

import pexpect
import os
import sys

if os.getenv("NO_TEST_STEAL") is not None:
    print("Skipping tty-stealing tests because $NO_TEST_STEAL is set.")
    sys.exit(0)

did_prctl = False
try:
    import prctl
    PR_SET_PTRACER_ANY = 0xffffffff
    if hasattr(prctl, 'set_ptracer'):
        did_prctl = True
        prctl.set_ptracer(PR_SET_PTRACER_ANY)
except ImportError:
    pass

if not did_prctl:
    print("Unable to find `prctl.set_ptracer`, skipping `PR_SET_PTRACER`.")

child = pexpect.spawn("test/victim")
child.setecho(False)
child.sendline("hello")
child.expect("ECHO: hello")

reptyr = pexpect.spawn("./reptyr -V -T %d" % (child.pid, ))
print("spawned children: me={} victim={} reptyr={}".format(
    os.getpid(), child.pid, reptyr.pid))
reptyr.logfile = sys.stdout

Example #4

File: tty-steal.py Project: 40a/reptyr

import pexpect
import os
import sys

if os.getenv("NO_TEST_STEAL") is not None:
    print("Skipping tty-stealing tests because $NO_TEST_STEAL is set.")
    sys.exit(0)

try:
    import prctl
    PR_SET_PTRACER_ANY = 0xffffffff
    if hasattr(prctl, 'set_ptracer'):
        prctl.set_ptracer(PR_SET_PTRACER_ANY)
except ImportError:
    print("Unable to import `prctl`, skipping `PR_SET_PTRACER`.")

child = pexpect.spawn("test/victim")
child.setecho(False)
child.sendline("hello")
child.expect("ECHO: hello")

reptyr = pexpect.spawn("./reptyr -T %d" % (child.pid,))
reptyr.sendline("world")
reptyr.expect("ECHO: world")

child.sendline("final")
child.expect(pexpect.EOF)
assert os.stat("/dev/null").st_rdev == os.fstat(child.fileno()).st_rdev

reptyr.sendeof()
reptyr.expect(pexpect.EOF)