File: //snap/core20/2599/lib/python3.8/test/libregrtest/__pycache__/win_utils.cpython-38.pyc
U
���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 �0/usr/lib/python3.8/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 5 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 r2 r r r r$ \ s
zWindowsLoadTracker.closec C s | � � d S r1 )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�}z| � |d � W n t
k
r� |� d�| _Y nX d| _|S ) NTFr r Zoem�replacer7 r ) r ZReadFiler r r � getbuffer�decoder �
splitlinesrA r: �pop)r r �_Z
bytes_read�res�output�linesr r r � _read_linesx s
z WindowsLoadTracker._read_linesc C s | j d krd S | j �� }|d k r0| jdd� d S z | �� }W n tk
rZ | �� Y d S X |D ]�}|�� }d|krvq`|s|q`z| �|�}W n$ tk
r� td| � Y q`Y nX | j d k r�| j t
|dt
| _ q`t | j �t
k r�| j �|� q`t| j �t | j � | _ q`| j S )NF)r2 zPDH-CSVz#Failed to parse typeperf output: %ag �?)r Zpollr$ rK �BrokenPipeError�rstriprA r: r r
�
LOAD_FACTOR_1r9 r �NVALUE�append�sum)r �
returncoderJ r> Zprocessor_queue_lengthr r r �
getloadavg� s<
�z WindowsLoadTracker.getloadavgN)T) �__name__�
__module__� __qualname__�__doc__r r
r r$ r4 rA rK rS r r r r r s "
r )r Zmathr r# r r r) Ztestr Ztest.libregrtest.utilsr r r ZexprN rO r, r r r r r �<module> s