3
:�)g � @ s� d dl Zd dlZd dlZd dlZd dlZddlmZ dd� Zdd� Z dd � Z
d
d � Z ej � Z
ejdfd d
�ZG dd� de�Zddd�ZdZdd� ZG dd� de�ZdS )� N� )�libc
C s� g }t | d��|}|j� }xl|jd�D ]^}|jd�}|dkrJ|jt|�� q$t|d |� �}t||d d � �}|jt||d �� q$W W d Q R X |S )N�r�,�-r ���)�open�read�split�find�append�int�extend�range)�pathZcpus�fZcpus_range_strZ cpu_rangeZrangeop�start�end� r �/usr/lib/python3.6/utils.py�_read_cpu_range s
"r 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/possible)r r r r r �get_possible_cpus' s r c C s. t j|�}tj|tj�jj� }|| kr*|S d S )N)r Zbcc_procutils_language�ct�castZc_char_p�value�decode)Z
candidates�pid�resZlanguager r r �detect_language* s
r c C s8 t |d�r|jn|}|j| � |r,|jd� |j� dS )zA
printb(s)
print a bytes object to stdout and flush
�buffer�
N)�hasattrr �write�flush)�s�file�nlZbufr 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 |jt�| _d S )Nr � )�sys� version_infor% r �FILESYSTEMENCODING)�self�argr r r �__init__F s zArgString.__init__c C s | j jt�S )N)r% �encoder- )r. r r r � __bytes__L s zArgString.__bytes__c C s | j S )N)r% )r. r r r �__str__O s zArgString.__str__N)�__name__�
__module__� __qualname__�__doc__r0 r2 r3 r r r r r) > s r) c C sD t |d�r|ntj}tjtjd�|d� |jtj| ||||�� 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 rE Fc C sX | d kr | S t r*t| �tksTtd| ��n*t| �tk rTtjd| td� t| �j� S | S )Nznot a bytes object: %rr8 ) �
_strict_bytes�type�bytes�AssertionErrorr= �warn�DeprecationWarningr) r2 )r/ r r r �_assert_is_bytes[ s rL 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 j|| �}x^|D ]V}|jd�} d}
|sH|jd�j� |krLd}
tj| |
||�\} }|d7 }| jd| d�} q W | ||d�} | S ) NzSTRCMP\("([^"]+)\",(.+?)\)zSTRCMP\(("[^"]+\"),(.+?)\)r Zbpf_probe_readr8 Zbpf_probe_read_userZSTRCMP)�exprrP rQ )�re�finditer�group�striprM rS �replace)
rT Zbin_cmpZis_userZprobe_user_listrP rQ Z STRCMP_REZmatches�matchrN rO rR Zrdictr r r � rewrite_expry s&
zStrcmpRewrite.rewrite_exprN)r4 r5 r6 � staticmethodrS r[ r r r r rM e s rM )NN)Zctypesr r+ r: r= rU Zlibbccr r r r r �getfilesystemencodingr- �stdoutr( �objectr) rE rF rL rM r r r r �<module> s