| Current File : //usr/local/lib/python3.6/site-packages/ptrace/debugger/__pycache__/debugger.cpython-36.pyc |
3
�d~& � @ s� d dl mZmZmZ d dlmZ d dlmZmZ d dl m
Z
m Z d dl m
Z
d dlmZmZ d dlmZ d dlmZ er�d d lmZmZmZmZmZmZ G d
d � d e�Z G d d
� d
e �Z dS )� )�info�warning�error)� PtraceError)�waitpid�WNOHANG)�SIGTRAP�SIGSTOP)�ECHILD)�
PtraceProcess�
ProcessSignal)�HAS_PTRACE_EVENTS)�sleep)�PTRACE_O_TRACEFORK�PTRACE_O_TRACEVFORK�PTRACE_O_TRACEEXEC�PTRACE_O_TRACESYSGOOD�PTRACE_O_TRACECLONE�THREAD_TRACE_FLAGSc @ s e Zd ZdS )�
DebuggerErrorN)�__name__�
__module__� __qualname__� r r �A/tmp/pip-build-in2aja_b/python-ptrace/ptrace/debugger/debugger.pyr s r c @ s� e Zd ZdZdd� Zd'dd�Zdd � Zd(d d �Zd)d
d�Zd*dd�Z d+dd�Z
dd� Z d,dd�Z d-dd�Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� ZdS ).�PtraceDebuggera�
Debugger managing one or multiple processes at the same time.
Methods
=======
* Process list:
- addProcess(): add a new process
- deleteProcess(): remove a process from the debugger
* Wait for an event:
- waitProcessEvent(): wait for a process event
- waitSignals(): wait for a signal
- waitSyscall(): wait for the next syscall event
* Options:
- traceFork(): enable fork() tracing
- traceExec(): enable exec() tracing
- traceClone(): enable clone() tracing
- enableSysgood(): enable sysgood option
* Other:
- quit(): quit the debugger, terminate all processes
Operations
==========
- iterate on all processes: "for process in debugger: ..."
- get a process by its identifier: "process = debugger[pid]"
- get the number of processes: len(debugger)
Attributes
==========
- dict: processes dictionary (pid -> PtraceProcess)
- list: processes list
- options: ptrace options
- trace_fork (bool): fork() tracing is enabled?
- trace_exec (bool): exec() tracing is enabled?
- trace_clone (bool): clone() tracing is enabled?
- use_sysgood (bool): sysgood option is enabled?
c C s6 i | _ g | _d| _d| _d| _d| _d| _| j� d S )Nr F)�dict�list�options�
trace_fork�
trace_exec� trace_clone� use_sysgood�
enableSysgood)�selfr r r �__init__@ s zPtraceDebugger.__init__NFc C s� || j krtd| ��t| ||||d�}td| � || j |< | jj|� y|jtt� W nb t k
rz t
d| � Y nD t k
r� } z|j � W Y dd}~X n d|_
|j� � Y nX tr�| jr�|j| j� |S )z�
Add a new process using its identifier. Use is_attached=False to
attach an existing (running) process, and is_attached=True to trace
a new (stopped) process.
z%The process %s is already registered!)�parent� is_threadzAttach %s to debuggerzEUser interrupt! Force the process %s attach (don't wait for signals).NF)r �KeyErrorr r r �append� waitSignalsr r �KeyboardInterruptr r �display� is_attached�detachr
r Z
setoptions)r$ �pidr- r& r' �process�eventr r r �
addProcessJ s,
zPtraceDebugger.addProcessc C s8 t d� t| j�}x t|�D ]}|j� |j� q W dS )zN
Quit the debugger: terminate all processes in reverse order.
z
Quit debuggerN)r r �reversed� terminater. )r$ Z processesr0 r r r �quith s
zPtraceDebugger.quitTc C s� d}|s|t O }|rV|| jkr.td| |d��| j| }|jrF|tO }t||�\}}ntd|�\}}|sl|r�|r�||kr�td||f |d��||fS )a'
Wait for a process event from a specific process (if wanted_pid is
set) or any process (wanted_pid=None). The call is blocking is
blocking option is True. Return the tuple (pid, status).
See os.waitpid() documentation for explanations about the result.
r zUnknown PID: %r)r/ � z Unwanted PID: %r (instead of %s)���)r r r r' r r )r$ �
wanted_pid�blocking�flagsr0 r/ �statusr r r �_waitpidt s
zPtraceDebugger._waitpidc C s� d}x�|s�y| j ||�\}}W nB tk
r` } z&|jtkrL| j| }|j� S |�W Y dd}~X nX | rr| rrdS y| j| }W q tk
r� td| � Y qX qW |j|�S )z�
Wait for a process event from the specified process identifier. If
blocking=False, return None if there is no new event, otherwise return
an object based on ProcessEvent.
Nz!waitpid() warning: Unknown PID %r) r< �OSError�errnor
r ZprocessTerminatedr( r Z
processStatus)r$ r8 r9 r0 r/ r; �errr r r �_wait_event_pid� s
z PtraceDebugger._wait_event_pidc C s� |d k r| j ||�S d}xpt| j�}t|�dkrvx$|D ] }| j |d�}|d k r6|S q6W |s^d S t|d d�}t|� q| j |d |�S qW d S )Ng����MbP?r6 F� g �?r )r@ �tupler �len�minr )r$ r8 r9 �pauseZpidsr/ r0 r r r � _wait_event� s
zPtraceDebugger._wait_eventc C s | j ||�S )z�
Wait for a process event from a specific process (if pid option is
set) or any process (default). If blocking=False, return None if there
is no new event, otherwise return an object based on ProcessEvent.
)rF )r$ r/ r9 r r r �waitProcessEvent� s z PtraceDebugger.waitProcessEventc O sJ |j dd�}x8| j|�}|jtkr&|�|j}||ks:| r>|S |�qW dS )z�
Wait for any signal or some specific signals (if specified) from a
specific process (if pid keyword is set) or any process (default).
Return a ProcessSignal object or raise an unexpected ProcessEvent.
r/ N)�getrF � __class__r �signum)r$ Zsignals�kwr/ r1 rJ r r r r* � s
zPtraceDebugger.waitSignalsc C s4 t }| jr|dO }|r&| j||jd�S | j|�S dS )z�
Wait for the next syscall event (enter or exit) for a specific process
(if specified) or any process (default). Return a ProcessSignal object
or raise an unexpected ProcessEvent.
� )r/ N)r r" r* r/ )r$ r0 rJ r r r � waitSyscall� s zPtraceDebugger.waitSyscallc C sv |s(y| j | }W n tk
r& dS X y| j |j= W n tk
rJ Y nX y| jj|� W n tk
rp Y nX dS )z9
Delete a process from the process list.
N)r r( r/ r �remove�
ValueError)r$ r0 r/ r r r �
deleteProcess� s z PtraceDebugger.deleteProcessc C s6 t s td��| jttB O _d| _td| j � dS )zJ
Enable fork() tracing. Do nothing if it's not supported.
zMTracing fork events is not supported on this architecture or operating systemTz!Debugger trace forks (options=%s)N)r
r r r r r r )r$ r r r � traceFork� s zPtraceDebugger.traceForkc C s t sdS d| _| jtO _dS )zJ
Enable exec() tracing. Do nothing if it's not supported.
NT)r
r r r )r$ r r r � traceExec� s zPtraceDebugger.traceExecc C s t sdS d| _| jtO _dS )zK
Enable clone() tracing. Do nothing if it's not supported.
NT)r
r! r r )r$ r r r �
traceClone s zPtraceDebugger.traceClonec C s t sdS d| _| jtO _dS )z�
Enable sysgood option: ask the kernel to set bit #7 of the signal
number if the signal comes from the kernel space. If the signal comes
from the user space, the bit is unset.
NT)r
r" r r )r$ r r r r# s z PtraceDebugger.enableSysgoodc C s
| j | S )N)r )r$ r/ r r r � __getitem__ s zPtraceDebugger.__getitem__c C s
t | j�S )N)�iterr )r$ r r r �__iter__! s zPtraceDebugger.__iter__c C s
t | j�S )N)rC r )r$ r r r �__len__$ s zPtraceDebugger.__len__)NF)T)T)T)NT)N)NN)r r r �__doc__r% r2 r5 r< r@ rF rG r* rM rP rQ rR rS r# rT rV rW r r r r r s$ *
r N) �loggingr r r Zptracer �osr r �signalr r r> r
Zptrace.debuggerr r Zptrace.bindingr
�timer Zptrace.binding.funcr r r r r r r �objectr r r r r �<module> s