3
:�)g5$ � @ 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.6/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)�selfZargumentr 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| jj� S | j tj@ dkr�| j tj@ dkr�| j tj@ dkr�d| j j� }| j tj
@ dkr�|d| j 7 }nd}| j dkr�dnd}d | jj� |t
| j �|f S | j tj@ dk�r8| j tj@ dk�r8| j tj@ dk�r8| jd
k�r8| j dk�rdnd}d | jj� |t
| j �f S d | j S )
Nr z%dz%sz + %sz * %d� �+�-z
*(%s %s %d%s)Zipz *(&%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_offsetZsignr r r �_format/ s,
zUSDTProbeArgument._formatc C s d| j � | j� 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! r+ r, �locationr r r r" Q s
zUSDTProbeLocation.__init__c C s d| j j� | jf S )Nz%s 0x%x)r/ r' r. )r! r r r r) X s zUSDTProbeLocation.__str__c C sP t � }tj| jj| jj| jj| j|tj |��}|dkrHt
d|| jf ��t |�S )Nr z.error retrieving probe argument %d location %d) r r Zbcc_usdt_get_argumentr+ �context�provider�namer, �ct�byrefr
r )r! r, �arg�resr r r � get_argument[ s z USDTProbeLocation.get_argumentN)r r r
r" r) r8 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 )N)r1 r2 r3 r/ � semaphoreZ
num_locationsr- )r! r1 r+ r r r r" g s zUSDTProbe.__init__c C s d| j j� | jj� | jf S )Nz%s:%s [sema 0x%x])r2 r' r3 r: )r! r r r r) p s zUSDTProbe.__str__c C s d| j j� | jj� f S )Nz%s:%s)r2 r' r3 )r! r r r �
short_namet s zUSDTProbe.short_namec C sD t � }tj| j| j| j|tj|��}|dkr8td| ��t | ||�S )Nr z"error retrieving probe location %d)
r r Zbcc_usdt_get_locationr1 r2 r3 r4 r5 r
r* )r! r, �locr7 r r r � get_locationw s zUSDTProbe.get_locationN)r r r
r" r) r; r= r r r r r9 f s r9 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,tj||jd��| _ntj|tjd��| _| jd kr�td| ��n<|r�|| _tj |jd��| _| jd kr�td| ��ntd��d S )Nr �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�encoder1 r4 Zc_char_pr
�pathZbcc_usdt_new_frompath)r! rA rC r r r r" � s
z
USDT.__init__c C s t j| j� d S )N)r Zbcc_usdt_closer1 )r! r r r �__del__� s z USDT.__del__c C sz |j dd�}t|�dkr6tj| j|jd�|jd��}n,|\}}tj| j|jd�|jd�|jd��}|dkrvtd| ��d S )N�:r r? 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_prober1 rB Z%bcc_usdt_enable_fully_specified_prober
)r! r+ �fn_name� probe_parts�ret�
provider_name�
probe_namer r r � enable_probe� s zUSDT.enable_probec C sP y| j ||� W n: tk
rJ } z t|tjd� tjd� W Y d d }~X nX d S )N)�filer )rM r
�print�sys�stderr�exit)r! r+ rH �er r r �enable_probe_or_bail� s
zUSDT.enable_probe_or_bailc C s | j S )N)r1 )r! r r r � get_context� s zUSDT.get_contextc C s, t jd � }t j| j�|d< tj|d�j� S )Nr r )r4 Zc_void_pr1 r Zbcc_usdt_genargsr' )r! Z ctx_arrayr r r �get_text� s z
USDT.get_textc C s` |j dd�}t|�dkr2tj| j|jd�|�j� S |\}}tj| j|jd�|jd�|�j� S d S )NrE r r? )rF rG r Zbcc_usdt_get_probe_argctyper1 rB r' Z+bcc_usdt_get_fully_specified_probe_argctype)r! rL Z arg_indexrI rK r+ r r r �get_probe_arg_ctype� s zUSDT.get_probe_arg_ctypec s( g � � �fdd�}t j�jt|�� � S )Nc s � j t�j| j�� d S )N)�appendr9 r1 �contents)r+ )�probesr! r r �
_add_probe� s z)USDT.enumerate_probes.<locals>._add_probe)r Zbcc_usdt_foreachr1 r )r! r[ r )rZ r! r �enumerate_probes� s zUSDT.enumerate_probesc C sD | j � }x6|D ].\}}}}|r"d}|j|j� |j� ||d� qW d S )Nr )r3 rH �addrrA r@ )�enumerate_active_probesZ
attach_uprober' )r! ZbpfZattach_usdt_ignore_pidrZ �binpathrH r] rA r r r �attach_uprobes� s zUSDT.attach_uprobesc s& g � � fdd�}t j| jt|�� � S )Nc s � j | |||f� d S )N)rX )r_ rH r] rA )rZ r r r[ � s z0USDT.enumerate_active_probes.<locals>._add_probe)r Zbcc_usdt_foreach_uprober1 r )r! r[ r )rZ r r^ � s z USDT.enumerate_active_probes)NN)
r r r
r" rD rM rT rU rV rW r\ r` r^ r r r r r> s
r> )Z
__future__r Zctypesr4 rP Zlibbccr r r r r r � Exceptionr
�objectr r* r9 r> r r r r �<module> s 7