�
f � �� � d dl Zd dlZd dlZd dlZd dlZddlmZ d� Zd� Z d� Z
d� Z ej � Z
ej dfd�Z G d � d
e� Zdd �Zd Zd
� Z G d� de� Zy)� N� )�libc �x � g }t | d� 5 }|j � }|j d� D ]p }|j d� }|dk( r|j t |� � �4t |d | � }t ||dz d � }|j
t ||dz � � �r d d d � |S # 1 sw Y |S xY w)N�r�,�-���r )�open�read�split�find�append�int�extend�range)�path�cpus�f�cpus_range_str� cpu_range�rangeop�start�ends �+/usr/lib/python3/dist-packages/bcc/utils.py�_read_cpu_ranger s� � �
�D�
�d�C� � 1�A������'�-�-�c�2� 1�I� �n�n�S�)�G��"�}�� � �C� �N�+��I�h�w� /�0���)�G�A�I�J�/�0��� � �E�%��Q�� /�0� 1� 1� �K� 1� �K�s �BB/�/B9c � � t d� S )Nz /sys/devices/system/cpu/online�r � � r �get_online_cpusr $ s � � �;� <�<r c � � t d� S )Nz /sys/devices/system/cpu/possibler r r r �get_possible_cpusr" ' s � � �=� >�>r c � � t j |� }t j |t j � j
j
� }|| v r|S d S �N)r �bcc_procutils_language�ct�cast�c_char_p�value�decode)�
candidates�pid�res�languages r �detect_languager/ * sG � �
�
$�
$�S�
)�C��w�w�s�B�K�K�(�.�.�5�5�7�H� �:�-�8�7�4�7r c � � t |d� r |j n|}|j | � |r|j d� |j � y)zA
printb(s)
print a bytes object to stdout and flush
�buffer�
N)�hasattrr1 �write�flush)�s�file�nl�bufs r �printbr: 1 s; � � !��x�0�$�+�+�d�C��I�I�a�L� � � � �%���J�J�Lr c �" � e Zd ZdZd� Zd� Zd� Zy)� 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 �t � t j d dk\ r|| _ y |j t � | _ y )Nr � )�sys� version_infor6 r* �FILESYSTEMENCODING)�self�args r �__init__zArgString.__init__F s- � � � � �A� �!� #��D�F��Z�Z� 2�3�D�Fr c �@ � | j j t � S r$ )r6 �encoderA �rB s r � __bytes__zArgString.__bytes__L s � ��v�v�}�}� /�0�0r c � � | j S r$ )r6 rG s r �__str__zArgString.__str__O s � ��v�v�
r N)�__name__�
__module__� __qualname__�__doc__rD rH rJ r r r r< r< > s � ��4� 1�r r< c �� � t |d� r|nt j }t j t j
d� |�� |j
t j | ||||� � y )Nr4 � )r r7 ) r3 r? �stderr� traceback� print_stack� _getframer4 �warnings�
formatwarning)�message�category�filename�linenor7 �line�logs r �warn_with_tracebackr] R sN � ��$��(�$�c�j�j�C�
���C�M�M�!� ,�3�7��I�I�h�$�$�W�h��&�$�O�Pr Fc �� � | �| S t r t | � t u s
J d| z � �| S t | � t ur7t j d| z t
d� t
| � j � S | S )Nznot a bytes object: %rrP )�
_strict_bytes�type�bytesrU �warn�DeprecationWarningr< rH )rC s r �_assert_is_bytesrd [ sp � �
�{��
���C�y�E�!�A�#;�c�#A�A�!� �J�
�c��%� ��
�
�.��4�6H�!�L���~�'�'�)�)� �Jr c �, � e Zd Zed� � Zed� � Zy)�
StrcmpRewritec �2 � d|z }|d|�d| �d|�d�z
}||fS )Nzstreq_%dz
static inline bool z?(char const *ignored, uintptr_t str) {
char needle[] = z1;
char haystack[sizeof(needle)];
z�(&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�fnames r �_generate_streq_functionz&StrcmpRewrite._generate_streq_functionf s3 � � �W�$��� �f�o� /� /�� �o�%�%r c �0 � |rd}nd}t j || � }|D ]m }|j d� } d}
|s!|j d� j � |v rd}
t j | |
||� \ } }|dz
}| j
d| d� } �o | ||d�} | S ) NzSTRCMP\("([^"]+)\",(.+?)\)zSTRCMP\(("[^"]+\"),(.+?)\)r �bpf_probe_readrP �bpf_probe_read_user�STRCMP)�exprrj rk )�re�finditer�group�striprf rm �replace)
rr �bin_cmp�is_user�probe_user_listrj rk � STRCMP_RE�matches�matchrh ri rl �rdicts
r � rewrite_exprzStrcmpRewrite.rewrite_expry s� � � �8�I�7�I��+�+�i��.�� � 4�E��[�[��^�F� .�O� �� � �A��$�$�&�/�9�&;�O�%2�%K�%K�,2�O�,;�W�&F� "�E�?�
�q�L�G��<�<��%��3�D� 4� � .� �
��
� r N)rK rL rM � staticmethodrm r r r r rf rf e s( � ��&� �&�$ �� �r rf )NN)�ctypesr& r? rR rU rs �libbccr r r r" r/ �getfilesystemencodingrA �stdoutr: �objectr< r] r_ rd rf r r r �<module>r� sx �� �
� � � � � � =�?�8�
/�S�.�.�0� ��:�:�!� ��� �(Q� �
��.�F� .r |