OwlCyberSecurity - MANAGER
Edit File: 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