Edit File: syscall_argument.cpython-36.pyc
3 � eU" � @ s� d dl mZ d dlmZmZmZ d dlmZ d dlm Z d dl m Z m Z d dl mZmZ d dlmZ d dlmZmZmZmZmZmZ d d lmZ d d lm Z m Z m Z m Z m Z d d l!m"Z" d d l#Z#d d l$Z$d d l%m&Z&m'Z' d dl(m)Z) d dl!m*Z* e&�r(d dl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 d dl2m3Z3m4Z4 ne'�r<d dl5m3Z3 ni Z3g Z6e&�r`e6j7e,e-e.e/e0e1f� e8dd� e6D ��Z6deieed�eeed�deideideideide*id e id � Z9d e iZ:e e d �Z;e<d,�Z=d(d)� Z>G d*d+� d+e�Z?d S )-� )� CPU_WORD_SIZE)�uint2int� formatWordHex� formatAddress)� signalName)�c_int)� PTRACE_ERRORS� writeError)� getLogger�INFO)�FunctionArgument)�formatMmapProt�formatAccessMode�formatOpenFlags�formatCloneFlags� formatDirFd�formatOpenMode)� FunctionCall)�setupSocketCall� formatOptVal�formatSockaddr�formatSockaddrInStruct�formatSockaddrIn6Struct)� SOCKETCALLN)� RUNNING_LINUX�RUNNING_FREEBSD)�FILENAME_ARGUMENTS)�formatSocketType)�timeval�timespec�pollfd�rlimit� new_utsname� user_desc)�SYSCALL_ARG_DICT� FD_SETSIZE)r$ c c s | ]}|j |fV qd S )N)�__name__)�.0�struct� r) �H/tmp/pip-build-uv1wutoy/python-ptrace/ptrace/syscall/syscall_argument.py� <genexpr>" s r+ �mode)�flagsr, )�dirfdr- r, r. Zprotr- �type�optval) �access�openZopenatZname_to_handle_atZmmapZmmap2�clone�socket� setsockoptZsockaddr)Z sockaddr_inZ sockaddr_in6�int�size_t� clockid_t�long� socklen_t�pid_t�uid_t�gid_tc c sB x<| D ]4}t t|��}x"td�D ]}||? d@ dkV q W qW d S )N� � )�ord�chr�range)�data�charZbyte�indexr) r) r* �iterBitsB s rF c @ sH e Zd Zdd� Zdd� Zefdd�Zdd� Zd d � Zd d � Z d d� Z dS )�SyscallArgumentc * C s� | j }| j}| j}| s | r(t| j �S | jj}yt| | | S tk rT Y nX yt| | }W n tk r~ d }Y nX |r�|| �S |dkr�|dkr�| j|�S |dk�r$|dkr�yt | S tk r� t |�S X |dk�r$t d| j �}t || jj| jd | j � d|j� }| j|| j �S |d k�rf|d k�rf| jd j }|d k �rf| jd j } | j|| �S |d k�rxt|�S |jd��r�|dd � }|tk�r�|dk�r�| j|�S |jd��r$y$| j|d d� j� �}|�r�|S W n: tk �r } z tt� | d| t d� W Y d d } ~ X nX t | j �S t j d|�} | �rh| j d�} t!| j d ��} | dk�rh| j"| j t#| �S |d k�r|t | j �S |t$k�r�t t%| j ��S t| j �S ) N�execve�argv�envpZ socketcall�call�argsr z<%s>�write�buf� � �signumzconst � zchar *�*r? z Warning: Format %r value error)Z log_levelz(.*)\[([0-9])+\]r6 � unsigned int� unsigned long�u32)rI rJ ���)rT rU rV )&�valuer/ �namer �functionr$ �KeyError�ARGUMENT_CALLBACK�readCStringArrayr �strr �optionsr �process�format� formatPointer� readStringr � startswithr � readCString�endswith�formatValuePointer�stripr r r r r �re�match�groupr6 Z readArrayr � INTEGER_TYPESr )�selfrX �argtyperY �syscall�callbackZ func_call�text�fd�length�errrj Zbasetype�countr) r) r* � createTextK s| & zSyscallArgument.createTextc sB | j }|sdS |jd�r$|dd � }y t| }W n tk rH d }Y nX |rX|| |�S |dkr�| jjj|t�}| jd| |�S |t kr�t | }| j||�S t r�|dkr�| jj d j � t � fdd �| j |t��}| jd d j|� |�S | jj}|d ko�|d k�r>| jd j d }dd� } | j ||| d�}| jdd j|� |�S d S )N�NULLzstruct � r6 z<%s>�fd_setr c s t | �� k S )N)r6 )�x)�nfdsr) r* �<lambda>� s z4SyscallArgument.formatValuePointer.<locals>.<lambda>z[%s]� Zrt_sigprocmaskZsigset_tZ sigsetsizer> c S s | d7 } t | �S )Nr? )r )�keyr) r) r* � formatter� s z5SyscallArgument.formatValuePointer.<locals>.formatter)ra z <sigset=(%s)>)rX rd �POINTER_CALLBACKr[ rZ r` Z readStructr rb � KNOWN_STRUCTSr Z arguments�filter�readBitsr% �joinrY ) rm rn �addressrp Zpointeer( ry ro �sizer r) )r{ r* rg � s8 z"SyscallArgument.formatValuePointerc s2 | j jj||d �}� fdd�tt|��D �}|S )Nr> c s g | ]\}}|r� |��qS r) r) )r' rE �bit)ra r) r* � <listcomp>� s z,SyscallArgument.readBits.<locals>.<listcomp>)rZ r` � readBytes� enumeraterF )rm r� ru ra �bytesry r) )ra r* r� � s zSyscallArgument.readBitsc C sP |r@| j j}| jjj||�\}}tj|�}t|�}|rD|d7 }nd}| j||�S )Nz...rw ) r_ �string_max_lengthrZ r` re �os�fsdecode�reprrb )rm r� �max_sizerC � truncatedrq r) r) r* re � s zSyscallArgument.readCStringc C s^ |rN| j j}||k }t||�}| jjj||�}tj|�}t|�}|rR|d7 }nd}| j ||�S )Nz...rw ) r_ r� �minrZ r` r� r� r� r� rb )rm r� r� �max_lenr� rC rq r) r) r* rc � s zSyscallArgument.readStringc C s� |sdS |}| j j}g }xL| jjj|�}|t7 }|j| j|�� |sFP |t|�kr|jd| � P qW ddj |� }| j ||�S )Nrw z (... more than %s strings ...)z<(%s)>z, ) r_ Zmax_array_countrZ r` ZreadWordr �appendre �lenr� rb )rm r� Zaddress0� max_countrq Zstr_addrr) r) r* r] � s z SyscallArgument.readCStringArrayc C s |t kr t | }|| ||�S d S )N)�STRUCT_CALLBACK)rm r( rY rX rp r) r) r* �formatStructValue� s z!SyscallArgument.formatStructValueN) r& � __module__� __qualname__rv rg r^ r� re rc r] r� r) r) r) r* rG I s O' rG )r6 r7 r8 r9 r: r; r< r= )@Zptrace.cpu_infor Zptrace.ctypes_toolsr r r Zptrace.signamesr �ctypesr Z ptrace.errorr r �loggingr r Zptrace.func_argr Zptrace.syscall.posix_argr r r r r r Zptrace.func_callr Zptrace.syscall.socketcallr r r r r Z#ptrace.syscall.socketcall_constantsr r� ri Zptrace.os_toolsr r Zptrace.syscallr r Zptrace.syscall.linux_structr r r r! r" r# Z ptrace.syscall.linux_constantsr$ r% Z ptrace.syscall.freebsd_constantsr� �extend�dictr\ r� r� �setrl rF rG r) r) r) r* �<module> sV