1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
# Copyright (C) 2017 Jakob Kreuze, All Rights Reserved.
#
# This file is part of Hypodermic.
#
# Hypodermic is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation, either version 3 of the License, or (at your
# option) any later version.
#
# Hypodermic is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
# Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with Hypodermic. If not, see <http://www.gnu.org/licenses/>.
"""ctypes wrapper for ptrace."""
import ctypes
import os.path
from hypodermic.memory import maps
class Process(object):
"""Process attached via ptrace.
Note:
The process is implicitly detached from upon destruction of this
object, if appropriate.
Args:
pid (:obj:`int`, optional): The pid of the process to attach to.
Defaults to 0, which means that the argument will not be
used.
path (:obj:`str`, optional): The path of the binary to run.
Defaults to "", which will as the target if a pid is not
specified, either.
Raises:
TypeError: If the pid argument is not an int, or if the path
argument is not a string.
OSError: If the pid cannot be attached to, if the process could
not be created for the given binary, or if any wrapper
libraries could not be loaded.
"""
def __init__(self, pid=0, path=""):
if not isinstance(pid, int):
raise TypeError("pid argument must be an int")
elif not isinstance(path, str):
raise TypeError("path argument must be a string")
self._load_ffi_methods()
if pid != 0:
self._is_parent = False
if self._attach(ctypes.c_int(pid)):
raise OSError("Could not attach to pid {}".format(pid))
else:
self._is_parent = True
self.pid = self._new_proc(ctypes.c_char_p(path.encode()))
if self.pid < 0:
raise OSError("Could not create process {}".format(path))
def __del__(self):
if hasattr(self, "_is_parent") and not self._is_parent:
self.detach()
def _load_ffi_methods(self):
# setuptools/cython hack.
script_path = os.path.abspath(os.path.dirname(__file__))
for filename in os.listdir(os.path.join(script_path, "..")):
if filename.startswith("libhypodermicw"):
lib_path = os.path.join(script_path, "..", filename)
break
else:
raise OSError("Could not find wrapper library.")
self._so = ctypes.cdll.LoadLibrary(lib_path)
self._new_proc = self._so.new_proc
self._attach = self._so.attach
self._detach = self._so.detach
self._cont = self._so.cont
def detach(self):
"""Explicitly detaches from the process.
Raises:
OSError: If the process cannot be detached from.
"""
if not self._is_parent and self._detach(ctypes.c_int(self.pid)):
raise OSError("Could not detach from pid {}".format(self.pid))
def cont(self):
"""Continues until the program is haulted.
Raises:
OSError: If the process cannot be continued.
"""
if self._cont(ctypes.c_int(self.pid)):
raise OSError("Could not continue")
@property
def maps(self):
return maps(self.pid)
|
