OwlCyberSecurity - MANAGER
Edit File: DistUpgradeAptCdrom.cpython-38.pyc
U ٪zc�1 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z G dd� de �Z G dd� de�ZdS )� N)�gettextc @ s e Zd ZdZdS )� AptCdromErrorz% base exception for apt cdrom errors N)�__name__� __module__� __qualname__�__doc__� r r �A/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeAptCdrom.pyr # s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z d"dd�Z d#dd�Z dd� Zd$dd�Zdd � Zd%d d �Zd d � Zejd!k r�eZdS )&�AptCdromz represents a apt cdrom object c C s0 || _ || _t� | _t� | _t� | _t�� d S )N)�view� cdrompath�set�packages� signatures�i18n�apt_pkgZ init_config)�selfr �pathr r r �__init__+ s zAptCdrom.__init__c C sB t j�tj�d�tj�d��}t j�|| �r>t� || |� dS )z\ restore the backup copy of the cdroms.list file (*not* sources.list)! � Dir::State�Dir::State::cdromsN) �osr �joinr �config�find_dir�find�exists�shutil�copy)r � backup_ext� cdromstater r r �restore_backup4 s �zAptCdrom.restore_backupc C sr | � � }| �|| j�}tj�d�}t|��}|�� }W 5 Q R X |�|d| �}t|d��}|� |� W 5 Q R X dS )z comment out the cdrom entry �Dir::Etc::sourcelistz# %s�wN) � _readDiskName�_generateSourcesListLiner r r � find_file�open�read�replace�write)r �diskname�pentry� sourceslist�f�contentr r r �comment_out_cdrom_entry? s z AptCdrom.comment_out_cdrom_entryc C s< t � }t � }t � }tj| jdd�D �] \}}}|�d�sB|�d�rP|dd�= q"q"d|kr\q"q"d|krz|�tj�|d�� q"d|kr�|�tj�|d�� q"d |ks�d |kr�t� d | � q"d |kr�|�tj�|d �� q"d |k�rt� tj�|d ��D ]}|�tj�|d |�� q�q"tj� |�d dkr"|dd�= q"|||fS )z| scan the CD for interessting files and return them as: (packagesfiles, signaturefiles, i18nfiles) T)�topdownzdebian-installerz dist-upgraderNz.aptignrZPackagesz Packages.gzZSourcesz Sources.gzz+Sources entry found in %s but not supportedz Release.gpgr � Zpool) r r �walkr �endswith�addr r �logging�error�listdir�split)r r r r �root�dirs�filesr. r r r �_scanCDJ s6 � � zAptCdrom._scanCDc C sl t j�d�}t �� }|�tjj�� �}| � � }t |d��*}|� d||f � |� d||f � W 5 Q R X dS )z update apts cdrom.list r �az CD::%s "%s"; zCD::%s::Label "%s"; N) r r r& ZCdromZident�aptZprogress�baseZ CdromProgressr$ r' r* )r ZdbfileZcdrom�idZlabel�outr r r �_writeDatabasek s zAptCdrom._writeDatabasec C s@ t |�}tj�d�}t |�D ] }d|kr ||kr |�|� q |S )z& drop architectures that are not ours zAPT::Architecturez/binary-)r r r r �remove)r r Zarch�dr r r � _dropArchu s zAptCdrom._dropArchc C sZ | j }tj�| j dd�}tj�|�rVt|��}|�� }W 5 Q R X dD ]}|�|d�}qD|S )Nz.disk�info)�"�]�[�_rK )r r r r r r' r( r) )r r+ rG r. Zspecialr r r r$ � s zAptCdrom._readDiskNamec C s� d}d}g }|D ]D}t �d|�}|s0ttd���|�d�}|�d�}|�|�d�� q|r^|sbd S |�� d||d�|�f }|S ) N� z (.*)/dists/([^/]*)/(.*)/binary-*z&Could not calculate sources.list entryr2 � � zdeb cdrom:[%s]/ %s %s� )�re�matchr rK �group�append�sortr ) r r+ r r Zdist�compsrE �mr, r r r r% � s" � z!AptCdrom._generateSourcesListLineNc C s� |st j�d�}| �� }|D ]�}t �d|||�d�d � f �}tj�|tj� |�d �}|� d�r�t � |��<}t |d��&}|� d�} |�| � | dkr|q�q|W 5 Q R X W 5 Q R X q t�||� q d S � N�Dir::State::lists� cdrom:[%s]/%s�distsr z.gz�wbi � � T�r r r r$ �uri_to_filenamer r r r �splitextr4 �gzipr' r( r* r r ) r Z translations� targetdirr+ r. �fname�outf�grB �sr r r �_copyTranslations� s � zAptCdrom._copyTranslationsc C s� |st j�d�}| �� }|D ]�}t �d|||�d�d � f �}tj�|tj� |�d �}|� d�r�t � |��<}t |d��&}|� d�} |�| � | dkr|q�q|W 5 Q R X W 5 Q R X q t�||� q d S rW r] ) r r ra r+ r. rb rc rd rB re r r r � _copyPackages� s � zAptCdrom._copyPackagesc C sL |D �]@}t j�|�d }t j�|�d }t j�� }ddd||g}t�� �.}|�t j � � � |j |d< t j||d�}W 5 Q R X |dks� dS t|��"} t �| �} | �� | jd } W 5 Q R X | �d �D ]~} | �� \} }}t j�||�} t j�| ��st�d | � q�t| ��}t �|�}W 5 Q R X || ks�t�d | |f � dS q�qd S ) z! verify the signatues and hashes r zapt-keyz--quietZverifyZ APT_CONFIG)�envFZSHA256� zignoring missing '%s'z(hash sum mismatch expected %s but got %sT)r r r9 r_ �environr �tempfileZNamedTemporaryFiler* r r �dump�name� subprocessZcallr' ZTagFile�stepZsectionr r r6 rG Z sha256sumr7 )r r �sigZbasepath�releasefZ verify_env�cmd�fpZretr. �tZsha256_section�entry�hash�sizerm �sumr r r �_verifyRelease� s@ �� zAptCdrom._verifyReleasec C sz |st j�d�}| �� }|D ]X}tj�|�d }||fD ]:}t �d|||�d�d� f �}t � |tj� ||�� q8q dS )z copy the release file rX r rY rZ NT) r r r r$ r r r_ r^ r r r r )r r ra r+ rp rq r. rb r r r � _copyRelease� s �zAptCdrom._copyReleasec C s� | � � \| _| _| _| �| j�| _t| j�dkrFt�d� tt d���| � | j�r^| � | j� | � | j� | � | j� | �� }|s�t�d� dS | �|| j�}tj�d�}t|��}|�� }W 5 Q R X t|d��}|�d||f � W 5 Q R X | �� d S ) z' reimplement pkgCdrom::Add() in python r z+no useable indexes found on CD, wrong ARCH?z`Unable to locate any package files, perhaps this is not a Ubuntu Disc or the wrong architecture?zno .disk/ directory foundFr" r# z%# added by the release upgrader %s %sT)r= r r r rF �lenr6 r7 r rK ry rz rg rf r$ r% r r r& r' r( r* rC )r r+ Zdebliner- r. r/ r r r �_doAdd� s6 � ��zAptCdrom._doAddc C s� t �d| j� |rLtj�tj�d�tj� d��}tj� |�rLt � ||| � tj� d| j� tj� dd� z | �� }W nT ttfk r� } z2t �d| � | j�td�td �| � W Y �d S d }~X Y nX t �d | � |S ) z add a cdrom to apt's database z AptCdrom.add() called with '%s'r r zAcquire::cdrom::mountzAPT::CDROM::NoMount�truezcan't add cdrom: %szFailed to add the CDz�There was a error adding the CD, the upgrade will abort. Please report this as a bug if this is a valid Ubuntu CD. The error message was: '%s'FNzAptCdrom.add() returned: %s)r6 �debugr r r r r r r r r r r r r| � SystemErrorr r7 r rK )r r r �res�er r r r5 # s, � ��z AptCdrom.addc C s | j dk S )z# helper to use this as 'if cdrom:' N)r )r r r r �__bool__? s zAptCdrom.__bool__�3)N)N)N)N)r r r r r r! r0 r= rC rF r$ r% rf rg ry rz r| r5 r� �sys�versionZ __nonzero__r r r r r ( s$ ! " ) r )rP r r? r r6 r` r rn r� rk r rK � Exceptionr �objectr r r r r �<module> s