File: //lib/python3/dist-packages/apparmor/__pycache__/config.cpython-310.pyc
o
%`b�2 � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZejdk r1d dl Z
G dd� de
j �Z nd dl
Z
d dl m
Z
mZ G dd� de�Zd d
� ZdS ) � )�with_statementN�� r c @ s e Zd Zdd� ZdS )�configparser_py2c C s* | � |�}t� }|D ]\}}|||< q
|S )N)�items�dict)�self�section� section_val�section_options�option�value� r �1/usr/lib/python3/dist-packages/apparmor/config.py� __getitem__ s
z configparser_py2.__getitem__N)�__name__�
__module__� __qualname__r r r r r r s r )�AppArmorException�open_file_readc @ sV e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )�Config�
/etc/apparmorc C s. || _ |dks |dkr|| _d | _d S td��)N�shell�iniz Unknown configuration file type)�CONF_DIR� conf_type�
input_filer )r r �conf_dirr r r �__init__- s
zConfig.__init__c C s. | j dkr dt� i}|S | j dkrt�� }|S )Nr � r )r r � configparser� ConfigParser)r �configr r r �
new_config6 s
�zConfig.new_configc C s� t j�| j|�}|| _| jdkr| �|�}|S | jdkrVtjdkr&t �
� }nt � }t |_
tjdkr8|�|� |S z|�|� W |S t jyU t|�}|�|j� Y |S w |S )zGReads the file and returns a config[section][attribute]=property objectr r r )�os�path�joinr r r �
read_shell�sys� version_infor r! r �str� optionxform�read� ParsingError�
py2_parser�name)r �filename�filepathr" � tmp_filepathr r r � read_config= s,
�
� ��zConfig.read_configc C s� t j�| j|�}tjtjB }z@tjddd| jd�}t j� | j
�r)t � | j
|j
� nt �|j
|� | jdkr=| �|||� n | jdkrI| �|||� |�� W n
ty[ td| ��w t �|j
|� dS ) z-Writes the given config to the specified file�w�aa_tempF)�prefix�delete�dirr r zUnable to write to %sN)r$ r% r& r �stat�S_IRUSR�S_IWUSR�tempfile�NamedTemporaryFile�existsr �shutil�copymoder/ �chmodr � write_shell�write_configparser�close�IOErrorr �rename)r r0 r" r1 �permission_600� config_filer r r � write_configU s
�zConfig.write_configc C �0 d}|r|� � D ]
}tj�|�r|} |S q|S )z2Returns name of first matching file None otherwiseN)�splitr$ r% �isfile)r � file_listr0 �fr r r �find_first_filem � �zConfig.find_first_filec C rJ )z7Returns name of first matching directory None otherwiseN)rK r$ r% �isdir)r �dir_list�dirname�direcr r r �find_first_dirw rP zConfig.find_first_dirc C s� dt � i}t|��2}|D ]&}t�|d�}|r2d|d v r&|d �d�\}}n|d }d}||d |< q W d � |S 1 s>w Y |S )zDReads the shell type conf files and returns config[''][option]=valuer T�=r N)r r �shlexrK )r r1 r" � conf_file�line�resultr r
r r r r' � s
��
�
�
zConfig.read_shellc C s� dd� |d � � D �}tj�| j�r�t| j���}|D ]�}t�|d�}|r�d|d v r�|d �d�\}} d|v rD| �dd�d }
d|
}
nd}
||v r�| |d | krw|d | } | d urqd
|v rfd
| d
} |d | |
d }n||
d }|�|� |� |� q|d }d } ||v r�|d | d ur�|d | } |d | d }|�|� |� |� q|�|� qW d � n1 s�w Y |r�|D ] }|d | } | d u r�|d }n|d | d }|�|� q�d S d S ) z-Writes the config object in shell file formatc S s g | ]}|�qS r r )�.0�keyr r r �
<listcomp>� s z&Config.write_shell.<locals>.<listcomp>r TrV r �#� N�"�
)
�keysr$ r% r> r r rW rK �write�remove) r r1 �f_outr" �options�f_inrY rZ r r
�comment� value_newr r r rB � sX
�
� � �.
�zConfig.write_shellc
C sP |� � }d}d}g }tj�| j�r�t| j���}|D ]�} | �� �d�rc|r>|D ]}
d|
d || |
d } |�| � q&g }||v rG|� |� | �
� dd� }||v r`d}|� |�}|�| � qd }q|r�t �
| d�} | ss|�| � q| �
d
d�\}
} z| �
d d�\} }
d |
}
W n ty� d }
Y nw |
�
� |v r�|| |
�
� | �
� kr�| �| || |
�
� �} |
d
| |
} |�| � |� |
�
� � qW d � n1 s�w Y |r�|D ]}
d|
d || |
d } |�| � q�g }||v r�|� |� |D ]'}|�d
| � |� |�}|D ]}
d|
d || |
d } |�| � �qq�dS )z6Writes/updates the given file with given config objectTN�[� z = ra r_ ���FrV r^ r z
[%s]
)�sectionsr$ r% r> r r �lstrip�
startswithrc rd �striprf rW rK �
ValueError�replace)r r1 re r" rm rc r rf rg rY r �line_newr
rh r r r rC � sp
�
�� �,
��zConfig.write_configparserN)r ) r r r r r# r3 rI rO rU r' rB rC r r r r r , s
?r c C s� t �d�}t|jd�}tj�| �rJt| ��-}|D ]"}|dd� dkr)|dd� }n |d dkr5|dd� }|�|� qW d � n1 sEw Y |� � |S ) z6Returns the de-dented ini file from the new format ini�rwr4 N� rk r � r_ )
r< r= �openr/ r$ r% r> r rc �flush)r0 �tmpre rg rY r r r r. s
� � r. )�
__future__r r$ rW r? r9 r( r< r) r! r r �apparmor.commonr r �objectr r. r r r r �<module> s
l