o

�/a0�@s�dd
lZdd
l
Z
ddlmZmZ
ddlmZmZ
ddlm	Z	
ddl
mZ
ddlm
Z

dd�Zd	d
�Zdd�Zd
d�Zdd�Zdd�Zdd�Zdd�Zddd�
Zd
S)�N)�
SCOPE_BASE�LdbError)�nttime2unix�dsdb)
�CommandError)
�SamDB)
�KCCcCs\tt
�
��
}t|d
d�}|�||
|�
|��}t|�
}t|�
t|�
kr*tdtj	d�
||fS)z>Get a readonly KCC object and the list of DSAs it knows about.T)
�readonlyzThere seem to be duplicate dsas�
�file)
�int�timer�
load_samdb�	list_dsas�set�len�print�sys�stderr)�url�lp�creds�unix_now�kcc�dsa_list�dsas�r�4/usr/lib/python3/dist-packages/samba/uptodateness.py�get_kcc_and_dsass




rc
CsV|��}
|
t
|���
d
|��d|
d|
d�}i}|��D]\}}|||<q||fS)zOGenerate dictionaries mapping short partition names to the
    appropriate DNs.zCN=Schema,%szDC=DomainDnsZones,%szDC=ForestDnsZones,%s)�DOMAIN�
CONFIGURATION�SCHEMA�	DNSDOMAIN�	DNSFOREST)�	domain_dn�str�get_config_basedn�items)�samdb�base_dn�
short_to_long�
long_to_short�
s�
lrrr�get_partition_maps-s





�


r.cCs<|
du
rt|�
\}}|�
|
��|
�}
|
|v
rtd
|
�
�
|
S)Nzunknown partition %s)r.�get�upperr)r(�part�short_partitions�long_partitionsrrr�
get_partition@s




r4cCs�g}|��}t
�||
�D]S}t|j�
}|j|d
|ddgd�}zt|ddd�
}|�dd�\}}	WntyE
}

zWYd}
~
qd}
~
w
w|d	krPtd
|�
�
|�	|	|t
|j�
t|j
�
f�

q|S)z3This finds the uptodateness vector in the database.z)(&(invocationId=%s)(objectClass=nTDSDSA))�distinguishedName�invocationId)�base�
expression�attrsr
�
,�
NzCN=NTDS Settingsz!Expected NTDS Settings DN, got %s)r&r�_dsdb_load_udv_v2r%�source_dsa_invocation_id�search�split�
IndexErrorr�appendr�highest_usnr�last_sync_success)r(�dn�cursors�	config_dn�
c�inv_id�res�settings_dn�prefix�dsa_dn�
errr�get_utdvKs6






��



��

�


�rNc
Cs:|jd
t
dg
d�}
t|
ddd�
}tt���
}||fS)N��highestCommittedUSN)r7�scoper9r
)r>rrr
)r(rI�usn�nowrrr�get_own_cursores


�

rTcCs�i}|
D]]}|jj
|td
g
d�}d|dd
d}z%t|||d�}	t|	|�}
t|	�
\}}||i
}
|
D]
\}}}}||
|<q3Wnty\
}
ztd||ftj	d�
WYd}~qd}~w
w|
||<q|S)N�dNSHostName)rQr9z	ldap://%sr
)r�credentialsrzCould not contact %s (%s)r
)
r(r>rrrNrTrrrr)�	local_kccr�part_dnrr�
utdv_edgesrLrI�ldap_urlr(rE�own_usn�own_time�remotesrD�guidrR�
trMrrr�get_utdv_edgesns0



�








�

���

r`c	Cs�i}|
D]T}z|||}Wnty"
}
zd
}WYd}~nd}~w
wi}|||<|
D],}||vrN|||vrD||||}|||<q+t
d|tjd�
q+t
d|tjd�
q+q|S)Nr
zMissing dn %s from UTD vectorr
z"missing dn %s from UTD vector list)�KeyErrorrrr)	rYr�	distances�dn1�peakrM�
d�dn2�distrrr�get_utdv_distances�s.





��







�
��rhc
Cs.d
}
|��D]}|��D]}t
|
|�}
qq|
S)Nr
)�values�max)rb�max_distance�vector�distancerrr�get_utdv_max_distance�s



�rnc
s�d
}}d}g}|��D]\}}|��D]
\}}	||kr!|�
|	�

qq|rk|��
|d}t|�
}
|
dd
krO|
dd}||||dd}t|d�}n|
dd}||}t|�
}t|�
t|�
d}||
}|||d��|
r|�f
dd	�|
D�
S�S)
Nr
g����r;g@)�maximum�median�failurec
si|]}
|
�|
�qSrr)�.0�key�
�summaryrr�
<dictcomp>�sz$get_utdv_summary.<locals>.<dictcomp>)r'rA�sortr�round�float)
rb�filtersrqrsrrri�dn_outerrl�dn_innerrm�length�index�expected_lengthrrvr�get_utdv_summary�s:






��











�
r�)
N)rr
�ldbrr�sambarr�samba.netcmdr�samba.samdbr�	samba.kccrrr.r4rNrTr`rhrnr�rrrr�<module>
s