o

%`b�2�@s�dd
lm
Z

ddlZddlZddlZddlZddlZddlZejdkr1ddl	Z
Gdd�de
j	�Znddl
Z
ddlm
Z
mZ
Gdd�de�Zd	d
�ZdS)�)
�with_statementN��r
c@seZ
dZd
d�ZdS)�configparser_py2cCs*|�|
�
}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







zconfigparser_py2.__getitem__N)�__name__�
__module__�__qualname__rrrrrrs
r)�AppArmorException�open_file_readc@sVeZ
dZddd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�ZdS)�Config�
/etc/apparmorcCs.||_|
d
ks|
dkr|
|_
d|_dStd�
�
)N�shell�inizUnknown configuration file type)�CONF_DIR�	conf_type�
input_filer)rr�conf_dirrrr�__init__-s




zConfig.__init__c
Cs.|jd
krdt
�i
}
|
S|jdkrt��}
|
S)Nr�r)rr�configparser�ConfigParser)r�configrrr�
new_config6s




�

zConfig.new_configcCs�tj
�|j|
�}||_|jd
kr|�|�
}|S|jdkrVtjdkr&t	�
�}nt�}t|_
tjdkr8|�|�

|Sz|�|�

W|St	jyU


t|�
}|�|j�

Y|Sw|S)zGReads the file and returns a config[section][attribute]=property objectrrr)�os�path�joinrrr�
read_shell�sys�version_infor r!r�str�optionxform�read�ParsingError�
py2_parser�name)r�filename�filepathr"�tmp_filepathrrr�read_config=s,





�







�
�

�zConfig.read_configcCs�tj
�|j|
�}tjtjB}z@tjd
dd|jd�}tj
�	|j
�
r)t�|j
|j
�
nt�|j
|�
|jdkr=|�|||�
n|jdkrI|�|||�
|��
Wn
ty[


td|
�
�
wt�|j
|�
dS)	z-Writes the given config to the specified file�
w�aa_tempF)�prefix�delete�dirrrzUnable 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)rr0r"r1�permission_600�config_filerrr�write_configUs 











�zConfig.write_configcC�0d
}|
r|
��D]
}t
j�|�
r|}
|Sq|S)z2Returns name of first matching file None otherwiseN)�splitr$r%�isfile)r�	file_listr0�
frrr�find_first_filem�





�zConfig.find_first_filecCrJ)z7Returns name of first matching directory None otherwiseN)rKr$r%�isdir)r�dir_list�dirname�direcrrr�find_first_dirwrPzConfig.find_first_dircCs�d
t�i
}t
|
�
�2}|D]&}t�|d�}|r2d|dvr&|d�d�
\}}n|d}d}||d
|<qWd�
|S1s>w



Y
|S)zDReads the shell type conf files and returns config[''][option]=valuerT�
=r
N)rr�shlexrK)rr1r"�	conf_file�line�resultrr
rrrr'�s 







��
�
�
zConfig.read_shellcCs�
d
d�|d��D�
}t
j�|j�
r�t|j�
��}|D]�}t�|d�}|r�d|dvr�|d�d�
\}}	d|vrD|	�dd�d}
d|
}
nd}
||vr�|	|d|krw|d|}|d	u
rqd
|vrfd
|d
}|d||
d}n||
d}|�|�

|�	|�

q|d}d	}	||vr�|d|d	u
r�|d|}	|d|	d}|�|�

|�	|�

q|�|�

qWd	�
n1s�w



Y
|r�|D] }|d|}	|	d	ur�|d}n|d|	d}|�|�

q�d	Sd	S)z-Writes the config object in shell file formatc
Ssg|]}
|
�qSrr)�.0�keyrrr�
<listcomp>�sz&Config.write_shell.<locals>.<listcomp>rTrVr
�
#�
N�
"�

)
�keysr$r%r>rrrWrK�write�remove)rr1�f_outr"�options�f_inrYrZrr
�comment�	value_newrrrrB�sX














�





���.




�
zConfig.write_shellc
CsP|��}d
}d}g}t
j�|j�
r�t|j�
��}|D]�}	|	���d�
rc|r>|D]}
d|
d|||
d}|�|�

q&g}||vrG|�	|�

|	�
�dd�}||vr`d
}|�|�
}|�|	�

qd	}q|r�t�
|	d
�}|ss|�|	�

q|	�
d
d�\}
}z|�
dd�\}}
d|
}
Wnty�


d}
Yn
w|
�
�|vr�|||
�
�|�
�kr�|�||||
�
��}|
d
||
}	|�|	�

|�	|
�
��

qWd�
n1s�w



Y
|r�|D]}
d|
d|||
d}	|�|	�

q�g}||vr�|�	|�

|D]'}|�d
|�

|�|�
}|D]}
d|
d|||
d}	|�|	�

�
qq�dS)z6Writes/updates the given file with given config objectTN�
[�  z = rar_���FrVr^rz
[%s]
)�sectionsr$r%r>rr�lstrip�
startswithrcrd�striprfrWrK�
ValueError�replace)rr1rer"rmrcr	rfrgrYr�line_newr
rhrrrrC�sp




















�





���,












��zConfig.write_configparserN)
r)rrrrr#r3rIrOrUr'rBrCrrrrr,s

	

?rc
Cs�t�
d
�
}
t|
jd�}tj�|�
rJt|�
�-}|D]"}|dd�dkr)|dd�}n|ddkr5|dd�}|�|�

qWd�
n1sEw



Y
|�	�
|
S)	z6Returns the de-dented ini file from the new format ini�rwr4N�rkr
�
	r_)
r<r=�openr/r$r%r>rrc�flush)r0�tmprergrYrrrr.
s









��	
r.)�
__future__rr$rWr?r9r(r<r)r!r r�apparmor.commonrr�objectrr.rrrr�<module>
s










l