Edit File: ptrace_signal.cpython-36.pyc
3 � e& � @ s2 d dl mZ d dlmZ d dlmZmZmZ yd dlmZ W n e k rT dZY nX yd dlm Z W n e k r~ dZ Y nX d dl m Z d dl mZ d d lmZ d d lmZ d d lmZ d d lmZmZmZmZmZmZm Z m Z d d l m Z d dl Z dZ!ddddd�Z"dZ#dd� Z$dd� Z%G dd� de�Z&dS )� )�error)�HAS_DISASSEMBLER)�SIGFPE�SIGSEGV�SIGABRT)�SIGCHLDN)�SIGBUS)� RUNNING_LINUX)� CPU_64BITS)� ProcessEvent)� PtraceError)� signalName)�DivisionByZero�Abort� StackOverflow�InvalidMemoryAccess� InvalidRead� InvalidWrite�InstructionError� ChildExit)�parseExpressionz>(?P<deref_size>(BYTE|WORD|DWORD|DQWORD) )?\[(?P<deref>[^]]+)\]� � � � )ZBYTEZWORDZDWORDZDQWORDz(?:[A-Z]{3,10})c C s, | j d�}yt| S tk r& d S X d S )NZ deref_size)�group�NAMED_WORD_SIZE�KeyError)�match�name� r �F/tmp/pip-build-uv1wutoy/python-ptrace/ptrace/debugger/ptrace_signal.py� findDerefSize& s r" c C sN |j d�}|sd S y t| |�S tk rH } ztd| � d S d }~X nX d S )NZderefzerr: %s)r r � ValueErrorr )�processr �expr�errr r r! �evalFaultAddress. s r' c @ s^ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z ddd�Z dS )� ProcessSignalc C s. t |�| _tj| |d| j � || _d | _d S )Nz Signal %s)r r r �__init__�signum�reason)�selfr* r$ r r r! r) ; s zProcessSignal.__init__c C sX | j ttfkr| j� n:| j tkr,| j� n&| j tkr@| j� n| j tkrRt � | _ | j S )N) r* r r � memoryFaultr � mathErrorr � childExitr r r+ )r, r r r! �_analyzeB s zProcessSignal._analyzec C s, t sd S y | jj� S tk r& d S X d S )N)r r$ ZdisassembleOner )r, r r r! �getInstructionM s z ProcessSignal.getInstructionc C s |j }tjdt |�}|rJ|d kr.t| j|�}t|t|�|| jd�| _d S tjdt tf |�}|r�|d krvt| j|�}t |t|�|| jd�| _d S tj dt tf |�}|r�|d kr�t| j|�}t |t|�|| jd�| _d S tjd|�}|r�| j |||�| _d S d S )Nz^(?:MOV|TEST)[A-Z]* %s,)�size�instrr$ z^%s %s,z %s [^,]+, %sz:^(?:REP(?:NZ)? )?(?P<operator>MOVS|SCAS)(?P<suffix>[BWD])?) �text�re�search� DEREF_REGEXr' r$ r r" r+ � INSTR_REGEXr r � movsInstr)r, r3 � fault_addressZasmr r r r! �memoryFaultInstrU s6 z ProcessSignal.memoryFaultInstrc C s� |j d�}|j d�}dddd�j|�}t}y�| j}tr@d} d} nd } d } |j| �} | | i} |d k} | rx|j| �}|| | < |d k r�|| kr�t}| r�||kr�t}n| r�| |f}n| f}W n tk r� i } Y nX ||||| | jd �S ) N�operator�suffixr r r )�B�W�DZrsiZrdiZesiZediZMOVS)r2 r3 � registersr$ ) r �getr r$ r Zgetregr r r )r, r: r3 r r< r= r2 Z error_clsr$ Z source_regZdest_regZ source_addrrA �writeZ dest_addrr r r! r9 z s: zProcessSignal.movsInstrc C s t r| jj� S d S d S )N)r r$ Z getsiginfo)r, r r r! � getSignalInfo� s zProcessSignal.getSignalInfoc C s� | j � }|r |jj}|s"d}nd }| j� }|d k rry*| jj� }||krZt|| jd�| _d S W n tk rp Y nX | jj � }|r�| jj � }|j | ko�|j kn s�t |||| jd�| _d S |r�| j||� | jr�d S t||| jd�| _d S )Nr )r$ )r3 r$ )rD Z _sigfaultZ_addrr1 r$ ZgetInstrPointerr r+ r Z findStackZgetStackPointer�start�endr r; r )r, �siginfor: r3 �ip�stack�spr r r! r- � s8 zProcessSignal.memoryFaultc C s: | j � }|sd S tjd|j�}|s&d S t|| jd�| _d S )Nz I?DIV (.*))r3 r$ )r1 r5 r r4 r r$ r+ )r, r3 r r r r! r. � s zProcessSignal.mathErrorc C s4 | j � }|r(|j}t|j|j|j�| _nt� | _d S )N)rD Z_sigchldr �pid�status�uidr+ )r, rG �childr r r! r/ � s zProcessSignal.childExitNc C s\ | j � |st}|dd � |d| jj � |d| j � | jrL| jj|� |dd � d S )N�-�< zPID: %sz Signal: %s)r0 r r$ rK r r+ �display)r, �logr r r! rQ � s zProcessSignal.display)N) �__name__� __module__� __qualname__r) r0 r1 r; r9 rD r- r. r/ rQ r r r r! r( 9 s %#* r( )'�loggingr Z ptrace.disasmr �signalr r r r � ImportErrorr Zptrace.os_toolsr Zptrace.cpu_infor Zptrace.debuggerr Z ptrace.errorr Zptracer Z ptrace.debugger.signal_reasonr r r r r r r r Zptrace.debugger.parse_exprr r5 r7 r r8 r" r' r( r r r r! �<module> s6 (