CURRENT PATH:
/
usr
/
lib
/
python3.9
/
site-packages
/
bcc
/
__pycache__
/
[ ⬅ KEMBALI ]
|
[ HOME ]
Upload File:
Upload Ke Sini
Dir Baru
File Baru
Editing:
usdt.cpython-39.pyc
a u*�h#$ � @ s� d dl mZ d dlZd dlZddlmZmZmZm Z m Z m Z G dd� de �Z G dd� de�ZG d d � d e�ZG d d � d e�ZG d d� de�ZdS )� )�print_functionN� )�lib�_USDT_CB�_USDT_PROBE_CB�bcc_usdt_location�bcc_usdt_argument�BCC_USDT_ARGUMENT_FLAGSc @ s e Zd ZdS )� USDTExceptionN)�__name__� __module__� __qualname__� r r �,/usr/lib/python3.9/site-packages/bcc/usdt.pyr s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �USDTProbeArgumentc C s� |j dk | _t|j �| _ |j| _| jtj@ dkr8|j| _| jtj@ dkrP|j| _| jtj @ dkrh|j | _ | jtj @ dkr�|j | _ | jtj @ dkr�|j| _| jtj@ dkr�|j| _d S )Nr )�size�signed�abs�validr �CONSTANT�constant� DEREF_OFFSET� deref_offset� DEREF_IDENT� deref_ident�BASE_REGISTER_NAME�base_register_name�INDEX_REGISTER_NAME�index_register_name�SCALE�scale)�self�argumentr r r �__init__ s zUSDTProbeArgument.__init__c C s d| j | jrdndf S )Nz %d %s byteszsigned Zunsigned)r r �r! r r r � _size_prefix+ s �z USDTProbeArgument._size_prefixc C sB | j tj@ dkrd| j S | j tj@ dkr8d| j�� S | j tj@ dkr�| j tj@ dkr�| j tj@ dkr�d| j �� }| j tj @ dkr�|d| j 7 }nd}| j dkr�dnd}d | j�� |t | j �|f S | j tj@ dk�r8| j tj@ dk�r8| j tj@ dk�r8| jd k�r8| j dk�rdnd}d | j�� |t | j �f S d | j S ) Nr z%dz%sz + %sz * %d� �+�-z *(%s %s %d%s)�ipz *(&%s %s %d)z&unrecognized argument format, flags %d)r r r r r r �decoder r r r r r r r r )r! Z index_offset�signr r r �_format/ s8 � ���� �zUSDTProbeArgument._formatc C s d| � � | �� f S )Nz%s @ %s)r% r, r$ r r r �__str__M s zUSDTProbeArgument.__str__N)r r r r# r% r, r- r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�USDTProbeLocationc C s( || _ || _|j| _|j| _|j| _d S �N)�probe�index� num_arguments�address�bin_path)r! r0 r1 �locationr r r r# Q s zUSDTProbeLocation.__init__c C s d| j �� | jf S )Nz%s 0x%x)r4 r* r3 r$ r r r r- X s zUSDTProbeLocation.__str__c C sP t � }t�| jj| jj| jj| j|t� |��}|dkrHt d|| jf ��t |�S )Nr z.error retrieving probe argument %d location %d) r r Zbcc_usdt_get_argumentr0 �context�provider�namer1 �ct�byrefr r )r! r1 �arg�resr r r � get_argument[ s ���z USDTProbeLocation.get_argumentN)r r r r# r- r= r r r r r. P s r. c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � USDTProbec C s: || _ |j| _|j| _|j| _|j| _|j| _|j| _d S r/ )r6 r7 r8 r4 � semaphoreZ num_locationsr2 )r! r6 r0 r r r r# g s zUSDTProbe.__init__c C s d| j �� | j�� | jf S )Nz%s:%s [sema 0x%x])r7 r* r8 r? r$ r r r r- p s �zUSDTProbe.__str__c C s d| j �� | j�� f S )Nz%s:%s)r7 r* r8 r$ r r r � short_namet s zUSDTProbe.short_namec C sD t � }t�| j| j| j|t�|��}|dkr8td| ��t | ||�S )Nr z"error retrieving probe location %d) r r Zbcc_usdt_get_locationr6 r7 r8 r9 r: r r. )r! r1 �locr< r r r � get_locationw s � zUSDTProbe.get_locationN)r r r r# r- r@ rB r r r r r> f s r> c @ s^ e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dS )�USDTNc C s� |rX|dkrX|| _ |r,t�||�d��| _nt�|t�d��| _| jd kr�td| ��n<|r�|| _t� |�d��| _| jd kr�td| ��ntd��d S )N����asciir z USDT failed to instrument PID %dz!USDT failed to instrument path %sz/either a pid or a binary path must be specified) �pidr Zbcc_usdt_new_frompid�encoder6 r9 Zc_char_pr �pathZbcc_usdt_new_frompath)r! rF rH r r r r# � s �z USDT.__init__c C s t �| j� d S r/ )r Zbcc_usdt_closer6 r$ r r r �__del__� s z USDT.__del__c C sz |� dd�}t|�dkr6t�| j|�d�|�d��}n,|\}}t�| j|�d�|�d�|�d��}|dkrvtd| ��d S )N�:r rE r a. Failed to enable USDT probe '%s': the specified pid might not contain the given language's runtime, or the runtime was not built with the required USDT probes. Look for a configure flag similar to --with-dtrace or --enable-dtrace. To check which probes are present in the process, use the tplist tool. )�split�lenr Zbcc_usdt_enable_prober6 rG Z%bcc_usdt_enable_fully_specified_prober )r! r0 �fn_name� probe_parts�ret� provider_name� probe_namer r r � enable_probe� s ����zUSDT.enable_probec C sR z| � ||� W n< tyL } z$t|tjd� t�d� W Y d }~n d }~0 0 d S )N)�filer )rR r �print�sys�stderr�exit)r! r0 rM �er r r �enable_probe_or_bail� s zUSDT.enable_probe_or_bailc C s | j S r/ )r6 r$ r r r � get_context� s zUSDT.get_contextc C s, t jd � }t �| j�|d< t�|d��� S )Nr r )r9 Zc_void_pr6 r Zbcc_usdt_genargsr* )r! Z ctx_arrayr r r �get_text� s z USDT.get_textc C s` |� dd�}t|�dkr2t�| j|�d�|��� S |\}}t�| j|�d�|�d�|��� S d S )NrJ r rE )rK rL r Zbcc_usdt_get_probe_argctyper6 rG r* Z+bcc_usdt_get_fully_specified_probe_argctype)r! rQ Z arg_indexrN rP r0 r r r �get_probe_arg_ctype� s ��zUSDT.get_probe_arg_ctypec s( g � � �fdd�}t ��jt|�� � S )Nc s � � t�j| j�� d S r/ )�appendr> r6 �contents)r0 ��probesr! r r � _add_probe� s z)USDT.enumerate_probes.<locals>._add_probe)r Zbcc_usdt_foreachr6 r �r! ra r r_ r �enumerate_probes� s zUSDT.enumerate_probesc C s8 | � � }|D ]&\}}}}|r d}|j||||d� q d S )NrD )r8 rM �addrrF )�enumerate_active_probesZ attach_uprobe)r! ZbpfZattach_usdt_ignore_pidr` �binpathrM rd rF r r r �attach_uprobes� s �zUSDT.attach_uprobesc s& g � � fdd�}t �| jt|�� � S )Nc s � � | |||f� d S r/ )r] )rf rM rd rF �r` r r ra � s z0USDT.enumerate_active_probes.<locals>._add_probe)r Zbcc_usdt_foreach_uprober6 r rb r rh r re � s z USDT.enumerate_active_probes)NN) r r r r# rI rR rY rZ r[ r\ rc rg re r r r r rC s rC )Z __future__r Zctypesr9 rU Zlibbccr r r r r r � Exceptionr �objectr r. r> rC r r r r �<module> s 7
SIMPAN
[ BATAL ]