Edit File: debugger.cpython-36.pyc
3 � e~& � @ 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-uv1wutoy/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