????
Your IP : 18.221.20.252
a
z
f, � @ sp d Z ddlmZ ddlmZ ejr.ddlmZ G dd� d�Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )ai
Classes and functions for the output of reference policy modules.
This module takes a refpolicy.Module object and formats it for
output using the ModuleWriter object. By separating the output
in this way the other parts of Madison can focus solely on
generating policy. This keeps the semantic / syntactic issues
cleanly separated from the formatting issues.
� )� refpolicy)�util)�cmpc @ s e Zd Zdd� Zdd� ZdS )� ModuleWriterc C s d | _ d | _d| _d| _d S )NT)�fd�module�sort�requires)�self� r �3/usr/lib/python3.9/site-packages/sepolgen/output.py�__init__& s zModuleWriter.__init__c C sF || _ | jrt| j � tj| j dd�D ]\}}|�dt|� � q&d S )NT)Z showdepthz%s
)r r � sort_filterr Zwalktree�write�str)r
r r �node�depthr r r r , s
zModuleWriter.writeN)�__name__�
__module__� __qualname__r
r r r r r r % s r c C s� t �| �}|�� t �|�}|�� t|�t|�krFt|d |d �S t||�D ]*}|d |d krPt|d |d � S qPdS )N� r )r Z set_to_listr �lenr �zip)�x�yZxlZyl�vr r r �
id_set_cmp= s
r c C sR t | j|j�}|dkr|S t | j|j�}|dkr4|S t | j|j�}|dkrN|S dS �Nr )r � src_typesZ tgt_typesZ obj_classes)�a�b�retr r r �
avrule_cmpK s r" c C s8 | j d |j d kr*t| j d |j d �S t| j|j�S r )�argsr Zifname�r r r r r �
ifcall_cmpZ s r% c C sr t | tj�r8t |tj�r"t| |�S t| jd g|j�S n6t |tj�sPt |tj�rZt | |�S t| j|jd g�S d S r )
�
isinstancer �
InterfaceCallr% r r# r ZAVRuleZ AVExtRuler" r$ r r r �rule_cmp` s
r( c C s t | j|j�S )N)r Zroler$ r r r �
role_type_cmpl s r) c C s" dd� }| � � D ] }||� qdS )z/Sort and group the output for readability.
c S s� g }| � � D ] }|�|� |�t�� � q | �� D ]}|�|� q2|�t�� � g }|�| �� � |�| �� � |�| �� � |j t
� t �d� d }g }|D ]t}t
|tj�r�|jd }n t
�|j�}||k�r|r�|�t�� � |}t�� } | j�d| � |�| � |�|� q�|�|� g }
|
�| �� � |
j t
� t�d� t|
��rht�� } | j�d� |�| � |�|
� | jD ]} | |v�rx|�| � �qx|| _d S )N)�keyr z ============= %s ==============z"============= ROLES ==============)Zmodule_declarations�appendr �Commentr �extendZavrulesZ
avextrulesZinterface_callsr r �
cmp_to_keyr( r& r' r# �firstr �linesZ
role_typesr) r Zchildren) r �c�modZrequireZrulesZcurZ sep_rulesZruler ZcommentZras�childr r r � sort_noder sN
z sort_filter.<locals>.sort_nodeN)Znodes)r r4 r r r r r o s = r N)
�__doc__� r r ZPY3r r r r" r% r( r) r r r r r �<module> s