OwlCyberSecurity - MANAGER
Edit File: trace.cpython-310.pyc
o �?Og-r � @ s� d Z ddgZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZ ddlZdZG dd� d�Zd d � Zd d � ZG d d� d�Zdd� Zdd� Zddd�Zdd� ZG dd� d�Zdd� Ze dkrye� dS dS )a� program/module to trace Python program or function execution Sample use, command line: trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs trace.py -t --ignore-dir '$prefix' spam.py eggs trace.py --trackcalls spam.py eggs Sample use, programmatically import sys # create a Trace object, telling it what to ignore, and whether to # do tracing or line-counting or both. tracer = trace.Trace(ignoredirs=[sys.base_prefix, sys.base_exec_prefix,], trace=0, count=1) # run the new command using the given tracer tracer.run('main()') # make a report, placing output in /tmp r = tracer.results() r.write_results(show_missing=True, coverdir="/tmp") �Trace�CoverageResults� N)� monotonicz#pragma NO COVERc @ s e Zd Zddd�Zdd� ZdS )�_IgnoreNc C s: |st � nt |�| _|sg ndd� |D �| _ddi| _d S )Nc S s g | ]}t j�|��qS � )�os�path�normpath��.0�dr r �,/opt/alt/python310/lib64/python3.10/trace.py� <listcomp>H s �z$_Ignore.__init__.<locals>.<listcomp>z<string>� )�set�_mods�_dirs�_ignore)�self�modules�dirsr r r �__init__F s �z_Ignore.__init__c C s� || j v r | j | S || jv rd| j |< dS | jD ]}|�|d �r*d| j |< dS q|d u r6d| j |< dS | jD ]}|�|tj �rKd| j |< dS q9d| j |< dS )Nr �.r )r r � startswithr r �sep)r �filename� modulename�modr r r r �namesL s( � � z _Ignore.names�NN)�__name__� __module__� __qualname__r r r r r r r E s r c C s t j�| �}t j�|�\}}|S )�,Return a plausible module name for the path.)r r �basename�splitext)r �baser �extr r r �_modnamew s r( c C s� t j�| �}d}tjD ] }t j�|�}|�|�r+|t|� t jkr+t|�t|�kr+|}q |r9| t|�d d� }n| }t j�|�\}}|�t jd�}t j rT|�t j d�}t j� |�\}}|� d�S )r# � r Nr ) r r �normcase�sysr �lenr � splitdrive�replace�altsepr% �lstrip)r Z comparepathZlongest�dirr& Zdriver r' r r r � _fullmodname~ s" � r2 c @ s> e Zd Z ddd�Zdd� Zdd� Zdd d �Zdd d �ZdS )r Nc C s || _ | j d u r i | _ | j �� | _|| _| jd u r i | _| j�� | _|| _| jd u r-i | _| j�� | _|| _|| _| jr�z,t| jd��}t� |�\}}}W d � n1 sVw Y | � | j |||d�� W d S t t tfy� } ztd| j|f tjd� W Y d }~d S d }~ww d S )N�rb)�callerszSkipping counts file %r: %s��file)�counts�copyZcounter� calledfuncsr4 �infile�outfile�open�pickle�load�update� __class__�OSError�EOFError� ValueError�printr+ �stderr)r r7 r9 r: r4 r; �f�errr r r r � s: � �����zCoverageResults.__init__c C s |� d�o |�d�S )z_Return True if the filename does not refer to a file we want to have reported. �<�>)r �endswith)r r r r r �is_ignored_filename� s z#CoverageResults.is_ignored_filenamec C sn | j }| j}| j}|j }|j}|j}|D ]}|�|d�|| ||< q|D ]}d||< q%|D ]}d||< q.dS )z.Merge in the data from another CoverageResultsr r N)r7 r9 r4 �get) r �otherr7 r9 r4 Z other_countsZother_calledfuncsZ other_callers�keyr r r r? � s �zCoverageResults.updateTFc " C s | j r t� td� | j }t|�D ]\}}}td|||f � q| jrht� td� d }} t| j�D ]4\\} } } \} }}| |krNt� td| d� | }d} | | kr]| | kr]td| � | } td| | ||f � q3i }| jD ]\}}|�|i � }||< | j||f ||< qmi }|�� D ]�\}}| �|�r�q�|�d�r�|d d � }|d u r�t j � t j � |��}t |�}n|}t j �|�s�t �|� t|�}|r�t|�}ni }t�|�}t j �||d �}t|d ��}t�|j�\}}W d � n1 s�w Y | �|||||�\} } |�r| �rtd | | �} | | ||f||< q�|�r<|�r<td� t|�D ]} || \} } }}td|| � �q(| j�r�z)t| jd��} t � | j| j | jf| d� W d � W d S 1 �sbw Y W d S t �y� }! ztd|! t j d� W Y d }!~!d S d }!~!ww d S )af Write the coverage results. :param show_missing: Show lines that had no hits. :param summary: Include coverage summary per module. :param coverdir: If None, the results of each module are placed in its directory, otherwise it is included in the directory specified. zfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r) z***z -->z %s.%s -> %s.%sz.pycN���z.coverr3 �d z lines cov% module (path)z%5d %3d%% %s (%s)�wbr z"Can't save counts files because %sr5 )!r9 rD �sortedr4 r7 rL �itemsrK rJ r r �dirname�abspathr( �exists�makedirsr2 �_find_executable_linenos� linecache�getlines�joinr< �tokenize�detect_encoding�readline�write_results_file�intr; r= �dumprA r+ rE )"r Z show_missing�summary�coverdirZcallsr r �funcnameZlastfileZ lastcfileZpfileZpmodZpfunc�cfileZcmodZcfuncZper_file�lineno� lines_hitZsums�countr1 �lnotab�sourceZ coverpath�fp�encoding�_�n_hits�n_linesZpercent�mrF rG r r r � write_results� s� �� �� � �(� ���z CoverageResults.write_resultsc C s z t |d|d�}W n ty' } ztd||f tjd� W Y d}~dS d}~ww d}d} |�J t|d�D ]9\} } | |v rN|�d || � | d7 } |d7 }n| |v r`t| vr`|�d � |d7 }n|�d � |�| �d �� q4W d � | |fS 1 s{w Y | |fS ) z'Return a coverage results file in path.�w�rl z3trace: Could not open %r for writing: %s - skippingr5 N)r r r r z%5d: z>>>>>> z � ) r< rA rD r+ rE � enumerate�write�PRAGMA_NOCOVER� expandtabs) r r �linesri rg rl r; rG ro rn rf �liner r r r_ ) s: ���� � ��z"CoverageResults.write_results_file)NNNNN)TFN�N)r r! r" r rK r? rq r_ r r r r r � s � \c C s, i }t �| �D ] \}}||vrd||< q|S )z:Return dict where keys are lines in the line number table.r )�disZfindlinestarts)�code�strs�linenosrm rf r r r �_find_lines_from_codeI s �r� c C s4 t | |�}| jD ]}t�|�r|�t||�� q|S )z<Return lineno dict for all code objects reachable from code.)r� � co_consts�inspectZiscoder? � _find_lines)r} r~ r �cr r r r� S s �r� c C s� i }t j}t| |d��;}t�|j�}|D ])\}}}} } |t jkr;|t jkr;|\} } | \} }t| | d �D ]}d||< q4|}qW d � |S 1 sIw Y |S )z�Return a dict of possible docstring positions. The dict maps line numbers to strings. There is an entry for line that contains only a string or a part of a triple-quoted string. rs r N)�token�INDENTr< r\ �generate_tokensr^ �STRING�range)r rl r Z prev_ttyperF �tokZttypeZtstr�start�endrz ZslineZscolZelineZecol�ir r r � _find_strings_ s" � � � r� c C s� z t �| ��}|�� }|j}W d � n1 sw Y W n ty= } ztd| |f tjd� i W Y d}~S d}~ww t|| d�}t | |�}t ||�S )zAReturn dict where keys are line numbers in the line number table.Nz%Not printing coverage data for %r: %sr5 �exec) r\ r< �readrl rA rD r+ rE �compiler� r� )r rF �progrl rG r} r~ r r r rX v s$ ������ rX c @ sv e Zd Z d dd�Zdd � Zd d d �Zd d � Zdd� Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z d d � ZdS ) r r r r NFc C s� || _ || _t||�| _i | _i | _d| _|| _i | _i | _ i | _ d| _ | r*t � | _ |r2| j | _dS |r:| j| _dS |rH|rH| j| _| j| _dS |rT| j| _| j| _dS |r`| j| _| j| _dS d| _dS )ax @param count true iff it should count number of times each line is executed @param trace true iff it should print out each line that is being counted @param countfuncs true iff it should just output a list of (filename, modulename, funcname,) for functions that were called at least once; This overrides `count' and `trace' @param ignoremods a list of the names of modules to ignore @param ignoredirs a list of the names of directories to ignore all of the (recursive) contents of @param infile file from which to read stored counts to be added into the results @param outfile file in which to write the results @param timing true iff timing information be displayed r Nr )r: r; r �ignorer7 Zpathtobasename� donothing�trace� _calledfuncs�_callers� _caller_cache� start_time�_time�globaltrace_trackcallers� globaltrace�globaltrace_countfuncs�globaltrace_lt�localtrace_trace_and_count� localtrace�localtrace_trace�localtrace_count) r rh r� � countfuncs� countcallers� ignoremods� ignoredirsr: r; �timingr r r r � s6 zTrace.__init__c C s dd l }|j}| �|||� d S )Nr )�__main__�__dict__�runctx)r �cmdr� �dictr r r �run� s z Trace.runc C s� |d u ri }|d u r i }| j st�| j� t�| j� zt|||� W | j s2t�d � t�d � d S d S | j sBt�d � t�d � w w r{ )r� � threading�settracer� r+ r� )r r� �globals�localsr r r r� � s � �z Trace.runctxc O sP d }| j s t�| j� z||i |��}W | j s t�d � |S | j s't�d � w w r{ )r� r+ r� r� )r �func�args�kw�resultr r r �runfunc� s � �z Trace.runfuncc C s� |j }|j}|r t|�}nd }|j}d }|| jv r&| j| d ur%| j| }nCd | j|< dd� t�|�D �}t|�dkridd� t�|d �D �}t|�dkridd� t�|d �D �} t| �dkri| d j}|| j|< |d ursd||f }|||fS )Nc S s g | ] }t �|�r|�qS r )r� Z isfunction)r rF r r r r � � �z1Trace.file_module_function_of.<locals>.<listcomp>r c S s g | ] }t |t�r|�qS r )� isinstancer� r r r r r � r� r c S s g | ] }t |d �r|�qS )� __bases__)�hasattr)r r� r r r r � r� z%s.%s) �f_code� co_filenamer( �co_namer� �gcZ get_referrersr, r ) r �framer} r r rd ZclsnameZfuncsZdictsZclassesr r r �file_module_function_of� s. � z Trace.file_module_function_ofc C s4 |dkr| � |�}| � |j�}d| j||f< dS dS )zkHandler for call events. Adds information about who called who to the self._callers dict. �callr N)r� �f_backr� )r r� �why�arg� this_funcZ parent_funcr r r r� � s �z Trace.globaltrace_trackcallersc C s$ |dkr| � |�}d| j|< dS dS )zoHandler for call events. Adds (filename, modulename, funcname) to the self._calledfuncs dict. r� r N)r� r� )r r� r� r� r� r r r r� s �z Trace.globaltrace_countfuncsc C sp |dkr6|j }|j�dd�}|r4t|�}|dur0| j�||�}|s2| jr-td||jf � | j S dS dS dS dS )z�Handler for call events. If the code block being entered is to be ignored, returns `None', else returns self.localtrace. r� �__file__Nz! --- modulename: %s, funcname: %s) r� � f_globalsrL r( r� r r� rD r� r� )r r� r� r� r} r r Z ignore_itr r r r� s" ���zTrace.globaltrace_ltc C s� |dkr?|j j}|j}||f}| j�|d�d | j|< | jr*tdt� | j dd� tj � |�}td||t � ||�f dd� | j S ) Nrz r r �%.2f� �r� � %s(%d): %sr) )r� r� �f_linenor7 rL r� rD r� r r r$ rY �getliner� )r r� r� r� r rf rN �bnamer r r r� ( s ��z Trace.localtrace_trace_and_countc C sd |dkr/|j j}|j}| jrtdt� | j dd� tj�|�}td||t � ||�f dd� | j S )Nrz r� r� r� r� r) ) r� r� r� r� rD r� r r r$ rY r� r� )r r� r� r� r rf r� r r r r� 7 s ��zTrace.localtrace_tracec C s< |dkr|j j}|j}||f}| j�|d�d | j|< | jS )Nrz r r )r� r� r� r7 rL r� )r r� r� r� r rf rN r r r r� D s zTrace.localtrace_countc C s t | j| j| j| j| jd�S )N)r: r; r9 r4 )r r7 r: r; r� r� )r r r r �resultsL s �z Trace.results) r r r r r r NNFr )r r! r" r r� r� r� r� r� r� r� r� r� r� r� r r r r r � s �2 ) c s� dd l } | �� }|jdddd� |�dd�}|jdd d d d � |jd dd dd � |jddd dd � |jddd dd � |�d�}|�� }|jddd dd � |jddd d d � |jd d d d � |jd!d"d#d � |jd$d%d d&d � |jd'd(d d)d � |jd*d+d d,d � |�d-d.�}|jd/d0g d1d2� |jd3d0g d4d2� |jd5d d6d7d2� |jd8d9d:d;� |jd<| jd=d;� |�� }|jr�t� d>��t� d?�� � �fd@dA��dBdC� |j D �|_ �fdDdC�|jD �|_|j r�|j s�|� dE� t|j |j dF�}|�|j|j|j�S t|j|j|j|jg��s |� dG� |j�r |j�s|j�r |� dH� |j�r+|j�s+|� dI� |jd u �r6|� dJ� t|j|j|j|j|j |j|j |j |jdK� }zk|j�rudd l }|j}|� |�\} } } | j g|j �t _!dL| j | j"| j#| d dM�} n:|jg|j �t _!t$j%�&|j�t j%d< t'�(|j��} t)| �*� |jdN�} W d � n 1 �s�w Y |jdLd d dO�} |�+| | | � W n( t,�y� } zt �-dPt j!d |f � W Y d }~nd }~w t.�y� Y nw |�/� }|j0�s�|�|j|j|j� d S d S )QNr z --version�versionz trace 2.0)�actionr� z Main optionsz(One of these (or --report) must be givenz-cz--count� store_truez�Count the number of times each line is executed and write the counts to <module>.cover for each module executed, in the module's directory. See also --coverdir, --file, --no-report below.)r� �helpz-tz--tracez3Print each line to sys.stdout before it is executedz-lz --listfuncsz�Keep track of which functions are executed at least once and write the results to sys.stdout after the program exits. Cannot be specified alongside --trace or --count.z-Tz --trackcallsz^Keep track of caller/called pairs and write the results to sys.stdout after the program exits.Z Modifiersz-rz--reportz�Generate a report from a counts file; does not execute any code. --file must specify the results file to read, which must have been created in a previous run with --count --file=FILEz-Rz --no-reportz^Do not generate the coverage report files. Useful if you want to accumulate over several runs.z-fz--filez+File to accumulate counts over several runs)r� z-Cz --coverdirz�Directory where the report files go. The coverage report for <package>.<module> will be written to file <dir>/<package>/<module>.coverz-mz --missingz?Annotate executable lines that were not executed with ">>>>>> "z-sz --summaryz\Write a brief summary for each file to sys.stdout. Can only be used with --count or --reportz-gz--timingzQPrefix each line with the time since the program started. Only used while tracingZFiltersz Can be specified multiple timesz--ignore-module�appendzqIgnore the given module(s) and its submodules (if it is a package). Accepts comma separated list of module names.)r� �defaultr� z --ignore-dirzWIgnore files in the given directory (multiple directories can be joined by os.pathsep).z--moduleFzTrace a module. �progname�?zfile to run as main program)�nargsr� � argumentszarguments to the programZstdlibZ platstdlibc s4 t j�t j�| ��} | �d���d� �} t j�| �S )Nz$prefixz $exec_prefix)r r � expanduser� expandvarsr. r )�s)� _exec_prefix�_prefixr r �parse_ignore_dir� s z main.<locals>.parse_ignore_dirc S s$ g | ]}|� d �D ]}|�� �q qS )�,)�split�strip)r r� r r r r r � s ��zmain.<locals>.<listcomp>c s& g | ]}|� tj�D ]}� |��q qS r )r� r �pathsep)r r� r� )r� r r r � s ��z -r/--report requires -f/--file)r: r; zLmust specify one of --trace, --count, --report, --listfuncs, or --trackcallsz8cannot specify both --listfuncs and (--trace or --count)z3--summary can only be used with --count or --reportz3progname is missing: required with the main options)r� r� r� r� r: r; r� r� )r r� � __package__� __loader__�__spec__� __cached__r� )r� r r� r� z Cannot run file %r because: %s)1�argparse�ArgumentParser� add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER� parse_argsZ ignore_dir� sysconfigZget_pathZ ignore_moduleZreportr6 �errorr rq Zmissingrb rc �anyr� rh Z listfuncsZ trackcallsr� r r� �module�runpyZ_get_module_detailsr� r� r+ �argv�parent�loaderr r rT �io� open_coder� r� r� rA �exit� SystemExitr� Z no_report)r� �parserZgrpZ_grpZoptsr� �tr� Z module_nameZmod_nameZmod_specr} Zglobsrk rG r )r� r� r� r �mainR s� � � � � � � ��� � � �� � � �� � � � �� ��$���r� r� r{ ) �__doc__�__all__r� rY r r+ r� r� r\ r� r� r| r= �timer r� r� rw r r( r2 r r� r� r� rX r r� r r r r r �<module> s@ 2 0 O �