File: //lib/python3.9/site-packages/bcc/__pycache__/utils.cpython-39.opt-1.pyc
a
u*�h � @ s� d dl Zd dlZd dlZd dlZd dlZddlmZ dd� Zdd� Z dd � Z
d
d � Z e� � Z
ejdfd d
�ZG dd� de�Zddd�ZdZdd� ZG dd� de�ZdS )� N� )�libc C s� g }t | d���}|�� }|�d�D ]^}|�d�}|dkrH|�t|�� q"t|d |� �}t||d d � �}|�t||d �� q"W d � n1 s�0 Y |S )N�r�,�-���r )�open�read�split�find�append�int�extend�range)�pathZcpus�fZcpus_range_strZ cpu_rangeZrangeop�start�end� r �-/usr/lib/python3.9/site-packages/bcc/utils.py�_read_cpu_range s
4r c C s t d�S )Nz /sys/devices/system/cpu/online�r r r r r �get_online_cpus$ s r c C s t d�S )Nz /sys/devices/system/cpu/possibler r r r r �get_possible_cpus' s r c C s. t �|�}t�|tj�j�� }|| v r*|S d S �N)r Zbcc_procutils_language�ct�castZc_char_p�value�decode)Z
candidates�pid�res�languager r r �detect_language* s
r" c C s8 t |d�r|jn|}|�| � |r,|�d� |�� dS )zA
printb(s)
print a bytes object to stdout and flush
�buffer�
N)�hasattrr# �write�flush)�s�file�nl�bufr r r �printb1 s
r, c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ArgStringz�
ArgString(arg)
encapsulate a system argument that can be easily coerced to a bytes()
object, which is better for comparing to kernel or probe data (which should
never be en/decode()'ed).
c C s& t jd dkr|| _n |�t�| _d S )Nr � )�sys� version_infor( r �FILESYSTEMENCODING)�self�argr r r �__init__F s zArgString.__init__c C s | j �t�S r )r( �encoder1 �r2 r r r � __bytes__L s zArgString.__bytes__c C s | j S r )r( r6 r r r �__str__O s zArgString.__str__N)�__name__�
__module__� __qualname__�__doc__r4 r7 r8 r r r r r- > s r- c C sD t |d�r|ntj}tjt�d�|d� |�t�| ||||�� d S )Nr& � )r r) ) r% r/ �stderr� traceback� print_stack� _getframer& �warnings�
formatwarning)�message�category�filename�linenor) �line�logr r r �warn_with_tracebackR s rJ Fc C s@ | d u r | S t rn*t| �tur<t�d| td� t| ��� S | S )Nznot a bytes object: %rr= )�
_strict_bytes�type�bytesrB �warn�DeprecationWarningr- r7 )r3 r r r �_assert_is_bytes[ s rP c @ s$ e Zd Zedd� �Zedd� �ZdS )�
StrcmpRewritec C s" d| }|d|| |f 7 }||fS )Nzstreq_%da~
static inline bool %s(char const *ignored, uintptr_t str) {
char needle[] = %s;
char haystack[sizeof(needle)];
%s(&haystack, sizeof(haystack), (void *)str);
for (int i = 0; i < sizeof(needle) - 1; ++i) {
if (needle[i] != haystack[i]) {
return false;
}
}
return true;
}
r )�string�probe_read_func�streq_functions�probeid�fnamer r r �_generate_streq_functionf s
�
z&StrcmpRewrite._generate_streq_functionc
C s� |r
d}nd}t �|| �}|D ]V}|�d�} d}
|sF|�d��� |v rJd}
t�| |
||�\} }|d7 }| �d| d�} q | ||d�} | S ) NzSTRCMP\("([^"]+)\",(.+?)\)zSTRCMP\(("[^"]+\"),(.+?)\)r Zbpf_probe_readr= Zbpf_probe_read_userZSTRCMP)�exprrT rU )�re�finditer�group�striprQ rW �replace)
rX Zbin_cmpZis_userZprobe_user_listrT rU Z STRCMP_RE�matches�matchrR rS rV Zrdictr r r � rewrite_expry s,
���zStrcmpRewrite.rewrite_exprN)r9 r: r; � staticmethodrW r` r r r r rQ e s
rQ )NN)Zctypesr r/ r? rB rY Zlibbccr r r r r" �getfilesystemencodingr1 �stdoutr, �objectr- rJ rK rP rQ r r r r �<module> s