3
�ft`W. � @ sZ d Z ddlmZmZ ddlmZ ddlmZmZmZ ddl T ddl
Z ddl Z
ddlZ
ddlZ
ddlZ
ddlZddlZddlZddlZddlZddlZddlZddlZddlZdgZdadd � Z da d
d � Z d d
� Z G dd� de �Z!dd� Z"dd� Z#d,dd�Z$dd� Z%dd� Z&dd � Z'd-d d �Z(d d!� Z)da*d"d#� Z+d.d$d%�Z,d&d'� Z-d(d)� Z.d/d*d+�Z/dS )0z%
Assorted utility functions for yum.
� )�print_function�absolute_import)�unicode_literals)�base64_decodebytes�
basestring�unicode)�*NZsha256c C s t dkrtjd�ja t | �S )z( Tests if a string is a shell wildcard. Nz [*?]|\[.+\])�_re_compiled_glob_match�re�compile�search)�s� r �/usr/lib/python3.6/misc.py�re_glob. s r c C sF t dkr(tjd�j}tjd�j}||fa xt D ]}|| �r.dS q.W dS )zC Tests if a string needs a full nevra match, instead of just name. Nz.*([-.*?]|\[.+\]).z[0-9]+:TF)�_re_compiled_full_matchr
r �match)r
ZoneZtwoZrecr r r �re_full_search_needed6 s
r c C s t d S )Nr )�_default_checksumsr r r r �get_default_chksum_typeD s r c @ s: e Zd ZdZd
dd�Zdd� Zdd� Zd d
� Zd d � ZdS )�
GenericHolderz�Generic Holder class used to hold other objects of known types
It exists purely to be able to do object.somestuff, object.someotherstuff
or object[key] and pass object to another function that will
understand itNc C s
|| _ d S )N)�_GenericHolder__iter)�self�iterr r r �__init__M s zGenericHolder.__init__c C s | j d k rt| | j �S d S )N)r r )r r r r �__iter__P s
zGenericHolder.__iter__c C s t | |�rt| |�S t|��d S )N)�hasattr�getattr�KeyError)r �itemr r r � __getitem__T s
zGenericHolder.__getitem__c C s dd� t | �j� D �S )z!Return a dictionary of all lists.c S s" i | ]\}}t |�tkr||�qS r )�type�list)�.0�keyZlist_r r r �
\ s z+GenericHolder.all_lists..)�vars�items)r r r r � all_listsZ s zGenericHolder.all_listsc C s4 x.|j � j� D ] \}}t| �j|g �j|� qW | S )z7 Concatenate the list attributes from 'other' to ours. )r( r' r& �
setdefault�extend)r �otherr$ �valr r r � merge_lists_ s zGenericHolder.merge_lists)N) �__name__�
__module__� __qualname__�__doc__r r r r( r- r r r r r G s
r c C s� t jdd| �} tj� }d}d}xn| jd�D ]`}|jd�r>d}q*|rT|j� dkrTd}q*|rf|jd�rfP q*|rx|jd�rxP q*|r*|j|d � q*W t|j � �S ) z,Convert ASCII-armored GPG key to binary
s
?�
r s$ -----BEGIN PGP PUBLIC KEY BLOCK-----� � s" -----END PGP PUBLIC KEY BLOCK-----� =)
r
�sub�io�BytesIO�split�
startswith�strip�writer �getvalue)�rawkey�blockZinblockZ pastheaders�liner r r �
procgpgkeye s
rA c C sP xJ| j dd�D ]:}|d |krt|d d�}||kr8dS ||k rDdS dS qW d S )
ab
Return if the GPG key described by the given keyid and timestamp are
installed in the rpmdb.
The keyid and timestamp should both be passed as integers.
The ts is an rpm transaction set object
Return values:
- -1 key is not installed
- 0 key with matching ID and timestamp is installed
- 1 key with matching ID is installed but has an older timestamp
- 2 key with matching ID is installed but has a newer timestamp
No effort is made to handle duplicates. The first matching keyid is used to
calculate the return result.
�namez
gpg-pubkey�version�release� r r3 � �����)ZdbMatch�int)Zts�keyidZ timestampZhdrZ installedtsr r r � keyInstalled� s rJ Tc C s t jj|�st j|� tjj|��� tjj� ��}tt jj |d�d��}|j
d� W d Q R X |j | � |�r|d }t jj|��st j|dd� xFt j |d �D ]4}t jj
|�}|d | } tj|| � t j| d� q�W d }
tt jj |d�d
d��}|j
|
� W d Q R X d S Q R X W d Q R X d S ) Nzgpg.conf�wbr4 z-roi� )�modez/*�/z�lock-never
no-auto-check-trustdb
trust-model direct
no-expensive-trust-checks
no-permission-warning
preserve-permissions
�wT)�os�path�exists�makedirs�dnfZcryptoZ pubring_dirZContext�open�joinr< Z op_import�glob�basename�shutil�copy�chmod) r> rI ZgpgdirZ make_ro_copyZctx�fpZrodir�frW Zro_fZoptsr r r �import_key_to_pubring� s&
r] c C s� t j� } y.tj| �}tjj|d �}dtjj|f }W n$ t k
rZ dtjj| f }Y nX dtjj
|f }t t j |��}xB|D ]:}t j
|�}t|d �r�t|d �dkr�|d | kr�|S q�W tj|tjj
d�}|S )zqreturn a path to a valid and safe cachedir - only used when not running
as root or when --tempcache is setr z%s-%s-z%s/%s*i� � )�prefix�dir)rO �geteuid�pwd�getpwuidrS Zi18nZucd�constZPREFIXr ZTMPDIR�sortedrV �lstat�S_ISDIR�S_IMODE�tempfileZmkdtemp) ZuidZusertupZusernamer_ �dirpathZ cachedirsZthisdirZstatsZcachedirr r r � getCacheDir� s
(rk c C sf g }t | �}t| �} d}x6||krN|| }|j| ||� � ||7 }||8 }qW |j| |d� � |S )zE Given a seq, split into a list of lists of length max_entries each. r N)�lenr" �append)�seqZ max_entries�retZnumZbeg�endr r r �
seq_max_split� s
rq c C sD yt j| � W n0 tk
r> } z|jtjkr.� W Y dd}~X nX dS )z| Call os.unlink, but don't die if the file isn't there. This is the main
difference between "rm -f" and plain "rm". N)rO �unlink�OSError�errno�ENOENT)�filename�er r r �unlink_f� s
rx Fc C s^ y
t j| �S tk
rX } z2|jtjtjfkr2dS |rF|jtjkrFdS � W Y dd}~X nX dS )zF Call os.stat(), but don't die if the file isn't there. Returns None. N)rO �statrs rt ru �ENOTDIRZEACCES)rv Z
ignore_EACCESrw r r r �stat_f� s
r{ c C sF y$t d��} | j� }t|�S Q R X W n ttfk
r@ tj� S X dS )z� Get the audit-uid/login-uid, if available. os.getuid() is returned
instead if there was a problem. Note that no caching is done here. z/proc/self/loginuidN)rT �readrH �IOError�
ValueErrorrO �getuid)�fo�datar r r � _getloginuid� s
r� c C s t dkrt� a t S )z� Get the audit-uid/login-uid, if available. os.getuid() is returned
instead if there was a problem. The value is cached, so you don't
have to save it. N)�_cached_getloginuidr� r r r r � getloginuid s r� c C s� |r
|}nJd}| j d�}|dkr@| |d� }|d kr@| d|� }|dkrTtjjd
��|r�t| �}t|�}|r�|r�|j|jkr�|S ytjj| |d � W n2 t k
r� } ztjjt
|���W Y dd}~X nX |r�|r�t j ||j|jf� |S )
z�take a filename and decompress it into the same relative location.
When the compression type is not recognized (or file is not compressed),
the content of the file is copied to the destinationN�.r �.zck�.xz�.bz2�.gz�.lzma�.zstz(Could not determine destination filenamei� )r� r� r� r� r� r� )
�rfindrS �
exceptionsZ MiscErrorr{ �st_mtime�libdnfZutils�
decompress� RuntimeError�strrO �utime) rv �dest�check_timestamps�outZdot_posZextZfir� rw r r r r� s,
"r� c C s: t jj| �}|d7 }t jj|�s.t j|dd� |d | S )Nz/geni� )rL rM )rO rP �dirnamerQ rR )rv �generated_namer� r r r �calculate_repo_gen_dest8 s
r� c C s t | |�}t| |dd�S )z� This is a wrapper around decompress, where we work out a cached
generated name, and use check_timestamps. filename _must_ be from
a repo. and generated_name is the type of the file. T)r� r� )r� r� )rv r� r� r r r �repo_gen_decompress@ s
r� c C s� g }x�t j | �D ]�}t|��n}xf|D ]^}tjd|�r6q$|j� }|j� }|sLq$|rx|jdd�}|jdd�}|j|j� � q$|j |� q$W W dQ R X qW |S )a( Takes a glob of a dir (like /etc/foo.d/\*.foo) returns a list of all
the lines in all the files matching that glob, ignores comments and blank
lines, optional paramater 'line_as_list tells whether to treat each line
as a space or comma-separated list, defaults to True.
z\s*(#|$)�
� �,N)
rV rT r
r �rstrip�lstrip�replacer* r9 rm )ZthisglobZ line_as_list�resultsZfnamer\ r@ r r r �read_in_items_from_dot_dirH s"
r� )NT)F)NF)T)0r1 Z
__future__r r r Z
dnf.pycompr r r ry Z libdnf.utilsr� Z dnf.constrS Z
dnf.cryptoZdnf.exceptionsZdnf.i18nrt rV r7 rO Zos.pathrb r
rX ri r r r r r r �objectr rA rJ r] rk rq rx r{ r� r� r� r� r� r� r� r r r r � sN
#
#