CasperSecurity
o
�b� � @ sF d Z ddlmZ ddlmZmZ ddlmZmZ G dd� dej �Z
dS )z
An in-memory caching resolver.
� )�defer)�common�dns)�failure�logc @ sR e Zd ZdZdZddd�Zdd� Zdd � Zd
d � Zdd d
�Z ddd�Z
dd� Z dS )�
CacheResolverz�
A resolver that serves records from a local, memory cache.
@ivar _reactor: A provider of L{interfaces.IReactorTime}.
Nr c C sh t j�| � i | _|| _i | _|d u rddlm} || _|r0|� � D ]\}\}}| �
|||� q"d S d S )Nr )�reactor) r � ResolverBase�__init__�cache�verbose�cancel�twisted.internetr �_reactor�items� cacheResult)�selfr r r �query�seconds�payload� r �5/usr/lib/python3/dist-packages/twisted/names/cache.pyr
s �zCacheResolver.__init__c
C sb || _ | j�� }| j�� D ]!\}\}\}}}|| }|| | D ]
} | j|k r-| j|= nq q
d S �N)�__dict__r r r r �ttl)
r �state�now�k�when�ans�add�ns�diff�recr r r � __setstate__& s
���zCacheResolver.__setstate__c C s( | j �� D ]}|� � q| j �� | jS r )r
�values�clearr )r �cr r r � __getstate__1 s
zCacheResolver.__getstate__c
s | j �� }t�|||�}z | j| \}\}} }
W n" ty: | jdkr-t�dt |� � t
� t �
t�|��� Y S w | jrGt�dt |� � || � z � fdd�|D �� fdd�| D �� fdd�|
D �f} W n ty{ t
� t �
t�|��� Y S w t
�| �S )N� zCache miss for zCache hit for c �. g | ]}t �|jj|j|j|j� |j��qS r �r �RRHeader�name�type�clsr r ��.0�r�r" r r �
<listcomp>G � ���z)CacheResolver._lookup.<locals>.<listcomp>c r* r r+ r0 r3 r r r4 M r5 c r* r r+ r0 r3 r r r4 S r5 )r r r �Queryr �KeyErrorr r �msg�reprr �failr �Failure� DomainError�
ValueError�succeed) r r- r/ r. �timeoutr �qr r �authr �resultr r3 r �_lookup7 s6
�
�
�
�� �
zCacheResolver._lookupc C s t �t�t�|���S r )r r: r r; r r< )r r- r? r r r �lookupAllRecords_ s z CacheResolver.lookupAllRecordsc C s� | j dkr t�d| � |p| j�� |f| j|< || jv r$| j| �� t|d �t|d � t|d � }|rI|d j}|D ]}t ||j�}q?nd}| j�
|| j |�| j|< dS )a�
Cache a DNS entry.
@param query: a L{dns.Query} instance.
@param payload: a 3-tuple of lists of L{dns.RRHeader} records, the
matching result of the query (answers, authority and additional).
@param cacheTime: The time (seconds since epoch) at which the entry is
considered to have been added to the cache. If L{None} is given,
the current time is used.
r) zAdding %r to cacher � N) r r r8 r r r r
�listr �min� callLater�
clearEntry)r r r � cacheTime�s�mr2 r r r r b s
$
� zCacheResolver.cacheResultc C s | j |= | j|= d S r )r r
)r r r r r rI � s zCacheResolver.clearEntry)Nr Nr ) �__name__�
__module__� __qualname__�__doc__r r
r$ r( rC rD r rI r r r r r s
(
r N) rP r r �
twisted.namesr r �twisted.pythonr r r r r r r r �<module> s