shell bypass 403
o
` b� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z dd� Z
dd� Z G dd� d�Z G d d
� d
e �Z
G d d � d e �ZG d
d� d�ZG dd� deej�ZG dd� deej�ZG dd� deej�ZG dd� deej�Zedkrye� dS dS )� N)�findfilec C s t j�t| dd��S )NZ
dtracedata)Zsubdir)�os�path�abspathr )�filename� r �9/usr/local/python-3.10/lib/python3.10/test/test_dtrace.pyr s r c C sl t �dd| �} z dd� | �� D �}|jdd� d� dd� |D �}d �|�W S ttfy5 td
�| ���w ) aM Normalize DTrace output for comparison.
DTrace keeps a per-CPU buffer, and when showing the fired probes, buffers
are concatenated. So if the operating system moves our thread around, the
straight result can be "non-causal". So we add timestamps to the probe
firing, sort by that field, then strip it from the outputz\[[0-9]+ refs\]� c S s$ g | ]}|r|� d �s|�d��qS )�#� )�
startswith�split��.0�rowr r r �
<listcomp> s ���z*normalize_trace_output.<locals>.<listcomp>c S s t | d �S )Nr )�int)r r r r �<lambda> s z(normalize_trace_output.<locals>.<lambda>)�keyc S s g | ]}|d �qS )� r r r r r r ! s �
z%tracer produced unparsable output:
{}) �re�sub�
splitlines�sort�join�
IndexError�
ValueError�AssertionError�format)�output�resultr r r �normalize_trace_output s � ��r" c @ sH e Zd ZdZdZg Zd dd�Zd dd�Zd dd�Zd dd �Z d
d � Z
dS )
� TraceBackendNc C sv t | jt|| j �t|d �|d��}tt|| j d ���}|�� �� }W d � ||fS 1 s2w Y ||fS )Nz.py)� script_file� python_file�optimize_pythonz .expected)r" � trace_pythonr � EXTENSION�open�read�rstrip)�self�namer& �
actual_output�f�expected_outputr r r �run_case. s
�
��zTraceBackend.run_casec C s | j |g }|r|d|g7 }|S )Nz-c)�COMMAND)r, r$ �
subcommand�commandr r r �generate_trace_command9 s z#TraceBackend.generate_trace_commandc C s. | � ||�}tj|tjtjdd��� \}}|S )NT)�stdout�stderr�universal_newlines)r5 �
subprocess�Popen�PIPE�STDOUT� communicate)r, r$ r3 r4 r6 �_r r r �trace? s ��zTraceBackend.tracec C s< g }|r |� dg| � d�tjg| |g �}| �||�S )Nz-O� )�extendr �sys�
executabler? )r, r$ r% r& Z python_flagsr3 r r r r' G s
zTraceBackend.trace_pythonc
C sx z| � td| j ��}|�� }W n tttfy) } z
t|�}W Y d }~nd }~ww |dkr:t� d�
| j d |���d S )N�
assert_usablezprobe: successz{}(1) failed: {}r ) r? r r( �strip�FileNotFoundError�NotADirectoryError�PermissionError�str�unittestZSkipTestr r2 )r, r Zfnfer r r rD N s ����zTraceBackend.assert_usable�N) �__name__�
__module__� __qualname__r( r2 Z COMMAND_ARGSr1 r5 r? r' rD r r r r r# ) s
r# c @ s e Zd ZdZg d�ZdS )�
DTraceBackendz.d)Zdtracez-qz-sN�rL rM rN r( r2 r r r r rO Z � rO c @ s e Zd ZdZddgZdS )�SystemTapBackendz.stpZstapz-gNrP r r r r rR _ rQ rR c @ sL e Zd ZdZdZdZedd� �Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�
TraceTestsNr c C s | j �� d S rK )�backendrD �r, r r r �
setUpClassl s zTraceTests.setUpClassc C s& | j j|| jd�\}}| �||� d S )N)r& )rT r1 r& � assertEqual)r, r- r. r0 r r r r1 p s
�zTraceTests.run_casec C � | � d� d S )NZ
call_stack�r1 rU r r r �test_function_entry_returnu � z%TraceTests.test_function_entry_returnc sz t g d��}ttd��� }|�� � W d � n1 sw Y � �fdd�}|d�D ]}|�|j� q+��t � |� dS )z9Ensure our call stack test hits all function call opcodes)Z
CALL_FUNCTIONZCALL_FUNCTION_EXZCALL_FUNCTION_KWz
call_stack.pyNc sF t � dd�jd�}|jD ]}t|tj�r |j| kr t�|� S q g S )Nz<string>�exec)�sourcer �mode�optimize) �compiler& � co_consts�
isinstance�types�CodeType�co_name�disZget_instructions)�funcname�code�c�Z code_stringr, r r �get_function_instructions� s �
�zFTraceTests.test_verify_call_opcodes.<locals>.get_function_instructions�start)�setr) r r* �discard�opnamerW )r, Zopcodesr/ rk Z instructionr rj r �test_verify_call_opcodesx s
� z#TraceTests.test_verify_call_opcodesc C rX )N�gcrY rU r r r �test_gc� r[ zTraceTests.test_gcc C rX )N�linerY rU r r r � test_line� r[ zTraceTests.test_line)
rL rM rN ZmaxDiffrT r& � classmethodrV r1 rZ rp rr rt r r r r rS d s
rS c @ � e Zd Ze� ZdZdS )�DTraceNormalTestsr N�rL rM rN rO rT r& r r r r rw � � rw c @ rv )�DTraceOptimizedTests� Nrx r r r r rz � ry rz c @ rv )�SystemTapNormalTestsr N�rL rM rN rR rT r& r r r r r| � ry r| c @ rv )�SystemTapOptimizedTestsr{ Nr} r r r r r~ � ry r~ �__main__)rf Zos.pathr r r9 rB rc rJ Z test.supportr r r" r# rO rR rS ZTestCaserw rz r| r~ rL Z test_mainr r r r �<module> s* 14
�