CasperSecurity
o
4��f/8 � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZmZ d dl Z d dl
Z
d dl
m Z dd� Z
G dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�Zdd� ZG dd� de�Zedkr�edd�Ze�ej� ejD ]VZe�dej � e�dej � e�dej � e�dej � ej i kr�e�de ej �!� � � ej"D ]Z#e�de#je#j e#j$f � q�ej%D ]
Z&e�d e&j � q�qndS dS ) � )�print_functionN)�Popen�PIPE)�gettextc # sl � t � }t�| ��P}|D ],}d|v r |�|d �d�� d|v r/d|d v r( n|�|d � t|��� E dH q W d � dS |D ]
}d|v rQ|�|d � qDW d � n1 s\w Y t|��}t t
t � |���}W d � n1 syw Y |D ]l}|d |v r�q�dV |d �
d d�|d< t�| ��I}|D ]>}d|vs�d|d vr�q�d
|v r�|d � t� fd d �|d
�d�D ��r�q�t|�jd
i |���� D ] }|�d
�r�q�|V q�q�W d � n1 s�w Y q�t�| ��9}|D ]}d|v �rd|d v �r nq�|D ]}d|v �rd|d v �r�qt|��� E dH �qW d � dS 1 �s/w Y dS )a~ Expand the given template.
A template file consists of a header, followed by paragraphs
of templated suites, followed by a footer. A templated suite
is any paragraph where the Suite field contains {.
This function expands all templated suites using the information
found in the CSV file supplied by distro-info-data.
It yields lines of template info.
zX-Exclude-Suitesz, �Suite�{N�series� �versionz LTSz X-Versionc 3 sT � | ]%}|� d �rt�|dd� � �dk p%|� d�o%t�|dd� � �dkV qdS )�le� Nr �ge)�
startswith�apt_pkg�version_compare)�.0�field��ver� �5/usr/lib/python3/dist-packages/aptsources/distinfo.py� <genexpr>Z s �
��
�z#_expand_template.<locals>.<genexpr>r )�setr �TagFile�update�split�add�str�
splitlines�open�reversed�list�csv�
DictReader�replace�any�formatr ) �template�csv_path� known_suites�tmpl�section�
csv_object�releases�rel�liner r r �_expand_template& sp �
��� �
� �
�� �� ��$�r0 c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�Templatec C sX d | _ d| _g | _d | _d | _d | _d | _g | _g | _d | _ i | _
d | _ d| _ d| _
d S �NFT)�name�child�parents�
match_name� description�base_uri�type�
components�children� match_uri�
mirror_set� distribution� available�official��selfr r r �__init__y s
zTemplate.__init__c C s |dd� | j D �v S )z8 Check if the distribution provides the given component c s s � | ]}|j V qd S �N)r4 )r �cr r r r � s � z)Template.has_component.<locals>.<genexpr>)r; )rC �compr r r �
has_component� s zTemplate.has_componentc C s. t |�\}}}|| jv r| j| �||�S dS )z8 Check if a given url of a repository is a valid mirror F)� split_urlr> �has_repository)rC �url�proto�hostname�dirr r r � is_mirror� s
zTemplate.is_mirrorN)�__name__�
__module__� __qualname__rD rH rO r r r r r2 w s r2 c @ sF e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )� ComponentNc C s || _ || _|| _|| _d S rE )r4 r8 �description_long�parent_component)rC r4 �desc� long_descrU r r r rD � s
zComponent.__init__c C � | j S rE �rU rB r r r �get_parent_component� � z Component.get_parent_componentc C �
|| _ d S rE rY )rC �parentr r r �set_parent_component� �
z Component.set_parent_componentc C s$ | j d ur| j S | jd ur| jS d S rE )rT r8 rB r r r �get_description� s
zComponent.get_descriptionc C r\ rE )r8 �rC rV r r r �set_description� r_ zComponent.set_descriptionc C r\ rE �rT ra r r r �set_description_long� r_ z Component.set_description_longc C rX rE rc rB r r r �get_description_long� r[ z Component.get_description_long)NNN)
rP rQ rR rD rZ r^ r` rb rd re r r r r rS � s
rS c @ sJ e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�Mirrorz( Storage for mirror related information Nc C s" || _ g | _| �||� || _d S rE )rM � repositories�add_repository�location)rC rL rM rN ri r r r rD � s
zMirror.__init__c C s | j �t||�� d S rE )rg �append�
Repository�rC rL rN r r r rh � s zMirror.add_repositoryc s � fdd�| j D �S )Nc s g | ] }|j � kr|�qS r �rL �r �rrm r r �
<listcomp>� s z5Mirror.get_repositories_for_proto.<locals>.<listcomp>�rg )rC rL r rm r �get_repositories_for_proto� � z!Mirror.get_repositories_for_protoc C s6 |d u rdS | j D ]}|j|kr||jv r dS q dS r3 )rg rL rN )rC rL rN ro r r r rJ � s
�zMirror.has_repositoryc s � fdd�� j D �S )Nc s g | ]}|� � j��qS r )�get_urlrM rn rB r r rp � s z(Mirror.get_repo_urls.<locals>.<listcomp>rq rB r rB r �
get_repo_urls� rs zMirror.get_repo_urlsc C rX rE �ri rB r r r � get_location� r[ zMirror.get_locationc C r\ rE rv )rC ri r r r � set_location� r_ zMirror.set_locationrE ) rP rQ rR �__doc__rD rh rr rJ ru rw rx r r r r rf � s
rf c @ r1 )rk c C s || _ || _d S rE �rL rN rl r r r rD � s
zRepository.__init__c C s | j | jfS rE rz rB r r r �get_info� s zRepository.get_infoc C s d| j || jf S )Nz
%s://%s/%srz )rC rM r r r rt � s zRepository.get_urlN)rP rQ rR rD r{ rt r r r r rk � s rk c C s6 t jd| dd�}t|�dk r|�d� t|�dk s|S )zE split a given URL into the protocoll, the hostname and the dir part z:*\/+� )�maxsplitr N)�rer �lenrj )rK r r r r rI � s
�rI c @ s e Zd Zddd�Zdd� ZdS )�DistInfoN� /usr/share/python-apt/templatesc
C s8 d| _ g | _tj�d�| _d }t�d�}t�d�}|sPztg d�dt d��
� d � � }W n# t t
fyO } z|jtjkrCt�d | � d
}W Y d }~nd }~ww || _i }d ||f }d �|�� �} d }
d } t|| �D �]�} | �d
d�}
t|
�dk r{qk|
d � � }|
d � � }|dkr�t|�| _qk|dkr�|| _ qk|dkr�| �|
| � d } t� }
||
_ ||
_ d| |
_ qk|dkr�||
_ qk|dkr�d|
_ | jD ]}|j |kr�|
j �!|� |j"�!|
� q�qk|dkr�t�#|�|
_$qk|dkr�t�#|�|
_%qk|dkr�||
_&qk|dk�r|
j'�s||
_'qk|d| j k�r||
_'qk|dk�r |
j(�s ||
_(qk|d | j k�r+||
_(qk|d k�s8|d | j k�r�t)j*�+|��rB|�pLt)j*�,t)j*�-||��}||v�r�i }z$t.|��}t/t0|j1d d � |D ���}W d � n 1 �srw Y W n t2�y� t3d!� g }Y nw |D ]0} | �4d"��r�|�5d#| �}�q�t6| �\}}}||v �r�|| �7||� �q�t8||||�||< �q�|||< || |
_9qk|d$k�r�t|�|
_:qk|d%k�r�| �r�|
�;| j ��s�|
j<�!| � t=|�} qk|d&k�r�| �>t|�� qk|d'k�r| �?t|�� qk|d(k�r| �@|� qk| �|
| � d }
d } d S ))Nr zAPT::Architecturez ^#LOC:(.+)$zS^(#LOC:.+)|(((http)|(ftp)|(rsync)|(file)|(mirror)|(https))://[A-Za-z0-9/\.:\-_@]+)$)� lsb_releasez-iz-sT)�universal_newlines�stdoutr z#lsb_release failed, using defaults:�Debianz
%s/%s.infoz /usr/share/distro-info/{}.csv�:� r| � ChangelogURI�MetaReleaseURIr z^%s$� MatchName� ParentSuite� Available�Official�RepositoryType�BaseURIz
BaseURI-%s�MatchURIz MatchURI-%s� MirrorsFilezMirrorsFile-%sc S s g | ]}|� � �qS r )�strip)r �xr r r rp A s z%DistInfo.__init__.<locals>.<listcomp>z#WARNING: Failed to read mirror filez#LOC:z\1� DescriptionrS �CompDescription�CompDescriptionLong�ParentComponent)A�metarelease_uri� templatesr �config�find�archr~ �compiler r � communicater� �OSError�IOError�errno�ENOENT�logging�warning�distr& �lowerr0 r r �_�changelogs_uri�finish_templater2 r4 r? r7 r5 r6 rj r< �string_to_boolr@ rA r: r9 r= �os�path�isabs�abspath�joinr r! �filter�match� Exception�printr �subrI rh rf r> r8 rH r; rS rb rd r^ )rC r� �base_dirri � match_loc�match_mirror_line�exc�map_mirror_sets�
dist_fname� csv_fnamer' � componentr/ �tokensr �value�nannyr> �value_f� mirror_datarL rM rN r r r rD � s�
���� ���
��
�
� ���
�
�
� zDistInfo.__init__c C s� |sdS |j du r|jr|jD ] }|j r|j |_ nq|ji kr2|jr2|jD ] }|j r1|j|_ nq&|rB|�|j�sB|j�|� d}|jD ]}|j|_qE| j �|� dS )z finish the current tempalte N)
r= r5 r6 r> rH r4 r; rj rA r� )rC r' r� �tr r r r� b s(
�
�
zDistInfo.finish_template)Nr� )rP rQ rR rD r� r r r r r� � s
tr� �__main__�Ubuntur� z
Suite: %szDesc: %sz BaseURI: %sz MatchURI: %sz Mirrors: %sz %s -%s -%sz %s)'�
__future__r r" r� r� r� �
subprocessr r r~ r r r� r0 �objectr2 rS rf rk rI r� rP �d�infor� r� r' r4 r8 r9 r= r> r! �keysr; rG rT r< r5 r r r r �<module> sJ Q !
�
��