Edit File: ptrace_syscall.cpython-36.pyc
3 � e� � @ s: d dl mZ d dlmZ d dlmZmZmZmZm Z d dl m Z m Z m Z d dlmZ d dlmZmZmZ d dlmZ d dlmZmZ d d lmZ d d lmZ er�d Z n6er�d Z n,e r�d Z n"er�er�dZ q�dZ ner�dZ ndZ er�dZ n4e r�dZ n*er�dZ n e�rdZ ne�rdZ ne d��d d d d�Z G dd� de�Z dS ) � )�strerror)� errorcode)� CPU_X86_64� CPU_POWERPC�CPU_I386� CPU_ARM32� CPU_AARCH64)� ulong2long� formatAddress� formatWordHex)� FunctionCall)� SYSCALL_NAMES�SYSCALL_PROTOTYPES�SyscallArgument)�setupSocketCall)� RUNNING_LINUX� RUNNING_BSD)� CPU_WORD_SIZE)�CPU_INSTR_POINTERZgpr0�r7�r8Zorig_raxZorig_eaxZraxZeax�r0�resultz Unsupported CPU architecture� � � )�select�execve�clonec @ sH e Zd Zddd�Zddd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dS )� PtraceSyscallNc C sJ t j| d|t� || _d| _d | _d | _d | _|s<| jj� }| j |� d S )N�syscall�long) r �__init__r �process�restyper � result_text� instr_pointer�getregs� readSyscall)�selfr# �options�regs� r, �F/tmp/pip-build-uv1wutoy/python-ptrace/ptrace/syscall/ptrace_syscall.pyr" 4 s zPtraceSyscall.__init__c C s� |s| j j� }| j|�}| j|� | jdkrN| jjrNt| | j | d | d j� | jt kr~x$t | j D ]}| j | }|j � qdW | jj r�t |t�| _ d S )NZ socketcallr r )r# r' �readArgumentValues� readArguments�namer* Zreplace_socketcallr �value�PREFORMAT_ARGUMENTS� arguments�formatr& �getattrr )r) r+ �argument_values�index�argumentr, r, r- �enter? s zPtraceSyscall.enterc C s& t |t�| _tj| jd| j �| _d S )Nz syscall<%s>)r5 �SYSCALL_REGISTERr r �getr0 )r) r+ r, r, r- r( Q s zPtraceSyscall.readSyscallc s� t r |j|j|j|j|j|jfS trD|j|j |j |j |j |j |jfS trl|j|j |j |j |j |j |j|jfS tr�� jj� �� �fdd�tdd�D �S tr�|j|j|j|j|j|jfS t r�|j |j |j |j |j!|j"fS t#� �d S )Nc s g | ]}� j j�|t ��qS r, )r# ZreadWordr )�.0r7 )r) �spr, r- � <listcomp>a s z4PtraceSyscall.readArgumentValues.<locals>.<listcomp>r � � )$r ZrdiZrsiZrdxZr10r Zr9r r Zr1Zr2Zr3Zr4Zr5Zr6r r r r# ZgetStackPointer�ranger ZebxZecxZedxZesiZediZebpr Zgpr3Zgpr4Zgpr5Zgpr6Zgpr7Zgpr8�NotImplementedError)r) r+ r, )r) r= r- r. X s $ z PtraceSyscall.readArgumentValuesc C sj | j tkrLt| j \| _}xJt||�D ] \}}|\}}| j|||d� q&W nx|D ]}| j|d� qRW d S )N)r1 r0 �type)r1 )r0 r r$ �zipZ addArgument)r) r6 �formatsr1 r4 ZargtypeZargnamer, r, r- r/ i s zPtraceSyscall.readArgumentsc C s | j tkrtt| j �}nt� }x<t| j�D ].\}}||kr>q,|jrT|jjd� rTq,d |_q,W | jj t �| _ | j jd�r�t | j �}n�| j }t| j �| _ | j dk r�| j tkr�| j }d| j t| t|�f }n8d| j ko�dkn s�d| j t|�f }n t| j �}|| _|S )N�*r z %s %s (%s)� z%s (%s))r0 r2 �set� enumerater3 rC �endswith�textr# Zgetreg�RETURN_VALUE_REGISTERr r$ r r r r r �strr% )r) Z preformatr7 r8 rK Zuresult�errcoder, r, r- �exits s. zPtraceSyscall.exitc C s d| j S )Nz<Syscall name=%r>)r0 )r) r, r, r- �__str__� s zPtraceSyscall.__str__)N)N) �__name__� __module__� __qualname__r" r9 r( r. r/ rO rP r, r, r, r- r 2 s "r N)r r r )r r r )r r )!�osr �errnor Zptrace.cpu_infor r r r r Zptrace.ctypes_toolsr r r Zptrace.func_callr Zptrace.syscallr r r Zptrace.syscall.socketcallr Zptrace.os_toolsr r r Zptrace.binding.cpur r: rL rB r2 r r, r, r, r- �<module> sJ