h
���_ c @ 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)� os_helper)�
print_warningi � i< i zDSOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\CurrentLanguagec @ sP e Zd Z dd� Zdd� Zdd� Zddd �Zd
d � Zd d
� Zdd� Z dd� Z
dS )�WindowsLoadTrackerc C s$ g | _ d | _d| _d | _| �� d S )N� )�_values�_load�_buffer�_popen�start��self� r
�3/usr/lib64/python3.10/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-siz )ZstdoutZcwd) Zstr�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�
subprocessZPopenZjoinr ZSAVEDCWDr �os�close)r Z pipe_nameZ open_modeZpipe_write_endZcommand_stdoutZoverlapZ counter_nameZcommandr
r
r r
+ s.
�
�
���zWindowsLoadTracker.startc C sv t �t jt�� }t �|d�d }W d � n1 s40 Y t|�}tt||��}|d }|d }d|� d|� d�S )NZCounterr Z2Z44z"\z\�")�winregZOpenKeyZHKEY_LOCAL_MACHINE�COUNTER_REGISTRY_KEYZ QueryValueExZiterZdictZzip)r ZperfkeyZcountersZ
counters_dictZsystemZprocess_queue_lengthr
r
r r O s .��z$WindowsLoadTracker._get_counter_nameTc C s6 | j d u rd S d | _|r"| j �� | j �� d | _ d S �N)r r �killZwait)r r r
r
r r ^ s
��zWindowsLoadTracker.closec C s | � � d S r )r r r
r
r �__del__i s ��zWindowsLoadTracker.__del__c C sN |� d�}t|�dkrt�|d }|�d�r6|�d�s:t�|dd� }t|�S )Nz,i r r �����)Zsplit�len�
ValueErrorZ
startswithZendswithZfloat)r �lineZtokensZvaluer
r
r � _parse_linel s
��z WindowsLoadTracker._parse_linec C s� t �| jtd�\}}|�d�\}}|dkr.dS |�� }|�dd�}| j| }|�d�}z| � |d � W n t
y� |� d�| _Y |S 0 d| _|S ) NTFr r
ZoemZreplacer" r ) r ZReadFiler r r Z getbufferZdecoder Z
splitlinesr& r$ Zpop)r r Z_Z
bytes_readZresZoutput�linesr
r
r � _read_linesz s"
���z WindowsLoadTracker._read_linesc C s | j d u rd S | j �� }|d ur0| jdd� d S z | �� }W n tyX | �� Y d S 0 |D ]�}|�� }d|v rtq^|szq^z| �|�}W n ty� td| � Y q^0 | j d ur�| j t
|dt
| _ q^t | j �t
k r�| j �|� q^t| j �t | j � | _ q^| j S )NF)r zPDH-CSVz#Failed to parse typeperf output: %ag �?)r Zpollr r( ZBrokenPipeErrorZrstripr& r$ r r �
LOAD_FACTOR_1r# r �NVALUEZappendZsum)r Z
returncoder' r% Zprocessor_queue_lengthr
r
r �
getloadavg� s>
���z WindowsLoadTracker.getloadavgN)T) Z__name__Z
__module__Z __qualname__r r
r r r! r& r( r+ r
r
r
r r s $
��r )r Zmathr r r r r Z test.supportr Ztest.libregrtest.utilsr r r Zexpr) r* r r r
r
r
r Z<module> s ��