File: //usr/local/lib/python3.7/test/libregrtest/__pycache__/win_utils.cpython-37.opt-1.pyc
B
��g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
dZ dZ de�
e d � ZdZdZG d d
� d
�ZdS ) � N)�support)�
print_warningi � �< � zDSOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\CurrentLanguagec @ sR e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z dd� Z
dd� Z dS )�WindowsLoadTrackerz�
This class asynchronously interacts with the `typeperf` command to read
the system load on Windows. Multiprocessing and threads can't be used
here because they interfere with the test suite's cases for those
modules.
c C s$ g | _ d | _d| _d | _| �� d S )N� )�_values�_load�_buffer�_popen�start)�self� r �6/usr/local/lib/python3.7/test/libregrtest/win_utils.py�__init__$ s
zWindowsLoadTracker.__init__c
C s� dt t�� � }tj}|tjO }|tjO }t�||tjdt t tj
tj �| _ t�
|tjdtj tjdtj �}t�|d�}tj| j dd�}|�d� | �� }d|dt t�g}tjd�|�|tjd �| _t � |� d S )
Nz\\.\pipe\typeperf_output_r r T)�
overlappedZtypeperfz-si� )�stdout�cwd) �str�uuidZuuid4�_winapiZPIPE_ACCESS_INBOUNDZ FILE_FLAG_FIRST_PIPE_INSTANCEZFILE_FLAG_OVERLAPPEDZCreateNamedPipeZ PIPE_WAIT�BUFSIZEZNMPWAIT_WAIT_FOREVERZNULL�pipeZ
CreateFileZ
GENERIC_WRITEZ
OPEN_EXISTING�msvcrtZopen_osfhandleZConnectNamedPipe�GetOverlappedResult�_get_counter_name�SAMPLING_INTERVAL�
subprocess�Popen�joinr ZSAVEDCWDr �os�close)r Z pipe_nameZ open_modeZpipe_write_endZcommand_stdoutZoverlapZ counter_nameZcommandr r r r
+ s"
zWindowsLoadTracker.startc C sb t �t jt��}t �|d�d }W d Q R X t|�}tt||��}|d }|d }d|� d|� d�S )N�Counterr �2Z44z"\�\�")�winreg�OpenKey�HKEY_LOCAL_MACHINE�COUNTER_REGISTRY_KEYZ QueryValueEx�iter�dict�zip)r ZperfkeyZcountersZ
counters_dict�systemZprocess_queue_lengthr r r r M s z$WindowsLoadTracker._get_counter_nameTc C s6 | j d krd S d | _|r"| j �� | j �� d | _ d S )N)r r
�kill�wait)r r0 r r r r# \ s
zWindowsLoadTracker.closec C s | � � d S )N)r# )r r r r �__del__g s zWindowsLoadTracker.__del__c C sN |� d�}t|�dkrt�|d }|�d�r6|�d�s:t�|dd� }t|�S )N�,� r r' ���)�split�len�
ValueError�
startswith�endswith�float)r �line�tokens�valuer r r � _parse_linej s
z WindowsLoadTracker._parse_linec C s� t �| jtd�\}}|�d�\}}|dkr.dS |�� }|�dd�}| j| }|�d�}y| � |d � W n t
k
r� |� d�| _Y nX d| _|S ) NTFr r Zoem�replacer5 r ) r ZReadFiler r r � getbuffer�decoder �
splitlinesr? r8 �pop)r r �_Z
bytes_readZres�output�linesr r r � _read_linesx s
z WindowsLoadTracker._read_linesc C s | j d krd S | j �� }|d k r0| jdd� d S y | �� }W n tk
rX | �� d S X x�|D ]�}|�� }d|krvq`|s|q`y| �|�}W n" tk
r� td| � w`Y nX | j d k r�| j t
|dt
| _ q`t | j �t
k r�| j �|� q`t| j �t | j � | _ q`W | j S )NF)r0 zPDH-CSVz#Failed to parse typeperf output: %ag �?)r Zpollr# rH �BrokenPipeError�rstripr? r8 r r
�
LOAD_FACTOR_1r7 r �NVALUE�append�sum)r �
returncoderG r< Zprocessor_queue_lengthr r r �
getloadavg� s:
z WindowsLoadTracker.getloadavgN)T) �__name__�
__module__� __qualname__�__doc__r r
r r# r2 r? rH rP r r r r r s "
r )r Zmathr r" r r r( Ztestr Ztest.libregrtest.utilsr r r ZexprK rL r+ r r r r r �<module> s