3
�^h � @ st d dl mZ ddlmZ ddlmZ ddlmZmZ ddl m
Z
edd d
d g�Z G d d
� d
e�Z G dd� de�Z
dS )� )�
namedtuple� )�BoundsRuletype� )�DiffResultDescriptor)�
Difference�Wrapper)�type_wrapper_factoryZmodified_bound�ruleZ added_boundZ
removed_boundc @ sH e Zd ZdZed�Zed�Zed�ZdZdZ dd� Z
dd� Z dd � Z dS )
�BoundsDifferencez9Determine the difference in *bounds between two policies.�diff_typeboundsNc C s� | j jdj| �� | jdks&| jdkr.| j� | jdd� | jD �dd� | jD �dd� d�\| _| _}g | _ x<|D ]4\}}t
|j �t
|j �krp| j j t
||j |j �� qpW dS ) z@Generate the difference in typebound rules between the policies.zJGenerating typebounds differences from {0.left_policy} to {0.right_policy}Nc s s | ]}t |�V qd S )N)�
BoundsWrapper)�.0�c� r � /usr/lib64/python3.6/bounds.py� <genexpr>5 s z3BoundsDifference.diff_typebounds.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r
)r r r r r r 6 s c S s
t | j�S )N)�str�child)�br r r �<lambda>7 s z2BoundsDifference.diff_typebounds.<locals>.<lambda>)�key)�log�info�format�_left_typebounds�_right_typebounds�_create_typebound_listsZ _set_diff�added_typebounds�removed_typebounds�modified_typeboundsr �parent�append�modified_bounds_record)�selfZmatched_typeboundsZ
left_boundZ right_boundr r r r + s
z BoundsDifference.diff_typeboundsc C s� g | _ x@| jj� D ]2}|jtjkr0| j j|� q| jjdj |j�� qW g | _
x@| j j� D ]2}|jtjkrx| j
j|� qZ| jjdj |j�� qZW dS )z$Create rule lists for both policies.z/Unknown rule type: {0} (This is an SETools bug)N) r Z left_policyZbounds�ruletyper Z
typeboundsr" r �errorr r Z right_policy)r$ r
r r r r C s
z(BoundsDifference._create_typebound_listsc C s( | j jd� d| _d| _d| _d| _dS )z%Reset diff results on policy changes.z!Resetting all *bounds differencesN)r �debugr r r r )r$ r r r � _reset_diffU s
z BoundsDifference._reset_diff)
�__name__�
__module__� __qualname__�__doc__r r r r r r r r r( r r r r r s r c @ s4 e Zd ZdZdZdd� Zdd� Zd d
� Zd d � Zd
S )r
z Wrap *bounds for diff purposes.r% r! r c C s4 || _ |j| _t|j�| _t|j�| _t|�| _d S )N)�originr% r r! r �hashr )r$ r
r r r �__init__f s
zBoundsWrapper.__init__c C s | j S )N)r )r$ r r r �__hash__m s zBoundsWrapper.__hash__c C s | j |j k S )N)r )r$ �otherr r r �__lt__p s zBoundsWrapper.__lt__c C s | j |j ko| j|jkS )N)r% r )r$ r1 r r r �__eq__s s zBoundsWrapper.__eq__N)r% r! r ) r) r* r+ r, � __slots__r/ r0 r2 r3 r r r r r
` s r
N)� collectionsr Z policyrepr Z descriptorsr �
differencer r �typesr r# r r
r r r r �<module> s A