CasperSecurity
o
U�7e � @ sL d dl Z z
d dlmZmZ W n ey d dlmZmZ Y nw d dlZd dlZd dlZd dlZd dl Z
d dl Z
zd dl m
Z
W n eyO d dlm
Z
Y nw d dlZG dd� de j�Zedkr�e
j�� Ze�e
j�� � e�d�Ze�� �� Z e � � Z e � � dej j!ej j"d j!e f Z#ee$ej j%�&� �e#e � � e �Z'e'�(� dS dS ) � N)�Empty�Queue)�urlopenc @ sf e Zd ZdZG dd� dej�Zddd�Zdd� Zdd d �Z d
d� Z
ddd�Z
ddd�Z dd� Z
dS )�
MirrorTestz@Determines the best mirrors by perfoming ping and download test.c @ s" e Zd ZdZd dd�Zdd� ZdS )
zMirrorTest.PingWorkerz�Use the command line command ping to determine the server's
response time. Using multiple threads allows to run several
test simultaneously.�r � �r r c C s@ || _ || _|| _|| _|| _|| _t�d�| _t j
� | � d S )Nz ^rtt .* = [\.\d]+/([\.\d]+)/.*) �borders�mod�parent�id�jobs�results�re�compile� match_result� threading�Thread�__init__)�selfr
r r r r r
� r �?/usr/lib/python3/dist-packages/softwareproperties/MirrorTest.pyr s z MirrorTest.PingWorker.__init__c C s0 d }| j �� s�| jj�� r�z5| j �d�}|j}| j�d| � tj ddddd|gtj
tj dd �j } |�
� }|s9nt�| j|�}q2W n tyL Y d S | j�d
| � Y tj�� t jd 7 _| j�tjtj| j| j� |r�| j�t |d �||g� tj� � | j �� s�| jj�� s
d S d S d S d S )
NFz
Pinging %s...�pingz-qz-c 2z-W 1z-i 0.5T)�stdout�stderr�universal_newlinesz Skipping %sr r ) r
�emptyr �running�isSet�get�hostname�
report_action�
subprocess�Popen�PIPE�STDOUTr �readliner �findallr r r �completed_lock�acquire� completed�report_progress�todor r
r �append�float�release)r �result�mirror�host�commando�liner r r �run# sF ���
�
&�zMirrorTest.PingWorker.runN)r r )�__name__�
__module__� __qualname__�__doc__r r5 r r r r �
PingWorker s
r: Nc C sl t j�| � d| _d| _|| _d | _|| _g | _dt _
t � � t _ t
|�t _|| _|s1t �� | _d S || _d S )N� )r r g r )r r r �action�progress�event�best� test_file�threadsr r* �Lockr( �lenr, �mirrors�Eventr )r rD r@ r> r r r r r C s
zMirrorTest.__init__c C s || _ | jr
| j�� d S d S �N)r< r> �set)r �textr r r r! T s �zMirrorTest.report_action�r �d r c C sP ||d ||d |d |d |d | | f| _ | jr&| j�� dS dS )zTSubclasses should override this method to receive
progress status updatesr r N)r= r> rG )r �current�maxr r
r r r r+ Y s
��zMirrorTest.report_progressc
C s� | j dddd�}t| j�}|dkr3|�dd| jt�d|d � g� |�dd| jt�d|d � g� | jdd � |D �d
tjtjfd �}|D ]
\}}t d |j
|f � qF|sXd S |d d j
}t d
| � |S )N� )r � �?)r � )rL r r
� r r c S s g | ]}|d �qS )rP r )�.0�rr r r �
<listcomp>j s z,MirrorTest.run_full_test.<locals>.<listcomp>�rN r )r r
zmirror: %s - time: %szand the winner is: %s) �
run_ping_testrC rD r- �random�randint�run_download_testr r, �printr )r � results_ping�sizer �t�h�winnerr r r �
run_full_testb s$
""�� zMirrorTest.run_full_testr c
C s� |dkr| j }t� }|D ]}|�|� q g }td�D ]}t�|||| ||�} | j�| � | �� q| jD ]} | � � q4|�
� |d|� S )z�Performs ping tests of the given mirrors and returns the
best results (specified by max).
Mod and borders could be used to tweak the reported result if
the download test is only a part of a whole series of tests.N� r ) rD r �put�ranger r: rA r- �start�join�sort)
r rD rL r r
r
�mr �ir\ r r r rU w s
zMirrorTest.run_ping_testc s� � fdd�}|dkr
� j }g }|D ]'}� j�� s n ||�}|dkr)|�||g� � �|�|�d t|�d|� q|�� |d|� S )z�Performs download tests of the given mirrors and returns the
best results (specified by max).
Mod and borders could be used to tweak the reported result if
the download test is only a part of a whole series of tests.c sZ d| � � d � jf }� �d| � t�� }zt|dd��d� t�� | W S Y dS )Nz%s/%sr zDownloading %s...rP )�timeouti � )�
get_repo_urlsr@ r! �timer �read)r1 �urlrc �r r r �test_download_speed� s �z9MirrorTest.run_download_test.<locals>.test_download_speedNr r rT )rD r r r- r+ �indexrC re ) r rD rL r r
rn r rf �
download_timer rm r rX � s
z MirrorTest.run_download_testc C s | � � | _| j�� dS )z/Complete test exercise, set self.best when doneN)r_ r? r �clearrm r r r r5 � s
zMirrorTest.runrF )rI r )NNr r )r6 r7 r8 r9 r r r: r r! r+ r_ rU rX r5 r r r r r s
-
� r �__main__zdpkg --print-architecturez!dists/%s/%s/binary-%s/Packages.gz))r �queuer r � ImportErrorrj r r" �os�aptsources.distro�
aptsources�aptsources.sourceslist�urllib.requestr �urllib2rV r r r6 �distro�
get_distro� get_sources� sourceslist� SourcesList�popen�piperk �strip�archrE r rG �source_template�name�
componentsr@ �list�
mirror_set�values�appr_ r r r r �<module> sL � �
��
� �