a
    I_}                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm	Z
 d dlmZ d dlmZ d dlmZ d dlZd dlmZ G d	d
 d
ejjZdS )    N)system_session)MUST_USE_KERBEROSDONT_USE_KERBEROS)traffic_packets)traffic)Packet)SamDB)sd_utilsc                       s  e Zd Z fddZ fddZdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Z d=d> Z!d?d@ Z"dAdB Z#dCdD Z$dEdF Z%dGdH Z&dIdJ Z'dKdL Z(dMdN Z)dOdP Z*dQdR Z+dSdT Z,dUdV Z-dWdX Z.dYdZ Z/d[d\ Z0d]d^ Z1d_d` Z2dadb Z3dcdd Z4dedf Z5dgdh Z6didj Z7dkdl Z8dmdn Z9dodp Z:dqdr Z;dsdt Z<dudv Z=dwdx Z>dydz Z?d{d| Z@d}d~ ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPdd ZQdd ZRdd ZSdd ZTdd ZUdd ZVdd ZWdd ZXdd ZYdd ZZdd Z[dd Z\dd Z]  Z^S )TrafficEmulatorPacketTestsc              
      s  t t|   tjd | _tjd | _tjd | _|  | _	t
 | _|  | _td| j | j| j| j	d| _| j | _t| jd tjdd| _tj| j| j	| j| jt| jd| jd	dd
| _t | _d| j_d| _tdd| _ d| _!tdd| _"t#| j| j | j!| j"}t$| jd t%| jd| j| j  t&| jd| j!| j" | j'|| j t()| j}d}|*| jj+| d S )NZSERVERZDOMAINZ	SERVER_IPz	ldap://%s)ZurlZsession_infocredentialslp   Ztraffic_packet_test_)prefix   )serverr   credstempdirZou
domain_sidZtotal_conversationsZinstance_idzSTGM-1-1    zSTGU-1-1z(A;;WP;;;PS)),superr
   setUposenvironr   ZdomainZhostZget_loadparmr   r   ZsessionZget_credentialsr   r   ldbZget_domain_sidr   r   clean_up_accountstempfileZmkdtempr   ZReplayContextZou_namecontextZConversationconversationZconversation_idZmachinenamesambaZgenerate_random_passwordZmachinepassZusernameZuserpassZConversationAccountsZ	create_ouZcreate_machine_accountZcreate_user_accountZgenerate_process_local_configr	   ZSDUtilsZdacl_add_aceZuser_dn)selfZaccountZsdutilsmod	__class__ D/usr/lib/python3/dist-packages/samba/tests/emulate/traffic_packet.pyr   $   sf    


	
z TrafficEmulatorPacketTests.setUpc                    s0   t t|   t| jd | `t| j d S )Nr   )	r   r
   tearDownr   r   r   shutilZrmtreer   )r   r!   r#   r$   r%   [   s    z#TrafficEmulatorPacketTests.tearDownc                 C   s&   t d}| t|| j| j d S )Nz#0.0	11	1	2	1	cldap	3	searchRequest	)r   	from_line
assertTruepZpacket_cldap_3r   r   r   Zpacketr#   r#   r$   test_packet_cldap_03a   s    
z/TrafficEmulatorPacketTests.test_packet_cldap_03c                 C   s&   t d}| t|| j| j d S )Nz#0.0	11	1	1	2	cldap	5	searchResDone	)r   r'   assertFalser)   Zpacket_cldap_5r   r   r*   r#   r#   r$   test_packet_cldap_05h   s    
z/TrafficEmulatorPacketTests.test_packet_cldap_05c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	2	1	dcerpc	0	Request	)r   r'   r,   r)   Zpacket_dcerpc_0r   r   r*   r#   r#   r$   test_packet_dcerpc_00o   s
    

z0TrafficEmulatorPacketTests.test_packet_dcerpc_00c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	1	2	dcerpc	2	Response	)r   r'   r,   r)   Zpacket_dcerpc_2r   r   r*   r#   r#   r$   test_packet_dcerpc_02u   s
    

z0TrafficEmulatorPacketTests.test_packet_dcerpc_02c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	1	2	dcerpc	3		)r   r'   r,   r)   Zpacket_dcerpc_3r   r   r*   r#   r#   r$   test_packet_dcerpc_03{   s
    

z0TrafficEmulatorPacketTests.test_packet_dcerpc_03c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	2	1	dcerpc	11	Bind	)r   r'   r,   r)   Zpacket_dcerpc_11r   r   r*   r#   r#   r$   test_packet_dcerpc_11   s
    

z0TrafficEmulatorPacketTests.test_packet_dcerpc_11c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	2	1	dcerpc	13		)r   r'   r,   r)   Zpacket_dcerpc_13r   r   r*   r#   r#   r$   test_packet_dcerpc_13   s
    

z0TrafficEmulatorPacketTests.test_packet_dcerpc_13c                 C   s&   t d}| t|| j| j d S )Nz%0.0	11	1	2	1	dcerpc	14	Alter_context	)r   r'   r,   r)   Zpacket_dcerpc_14r   r   r*   r#   r#   r$   test_packet_dcerpc_14   s    
z0TrafficEmulatorPacketTests.test_packet_dcerpc_14c                 C   s4   t d}| jjt | t|| j	| j d S )Nz*0.0	11	1	1	2	dcerpc	15	Alter_context_resp	)
r   r'   r   
user_credsset_kerberos_stater   r,   r)   Zpacket_dcerpc_15r   r*   r#   r#   r$   test_packet_dcerpc_15   s    
z0TrafficEmulatorPacketTests.test_packet_dcerpc_15c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	1	2	dcerpc	16	AUTH3	)r   r'   r,   r)   Zpacket_dcerpc_16r   r   r*   r#   r#   r$   test_packet_dcerpc_16   s    
z0TrafficEmulatorPacketTests.test_packet_dcerpc_16c                 C   s&   t d}| t|| j| j d S )Nz0.0	11	1	1	2	dns	1	response	)r   r'   r,   r)   Zpacket_dns_1r   r   r*   r#   r#   r$   test_packet_dns_01   s    
z-TrafficEmulatorPacketTests.test_packet_dns_01c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	1	2	drsuapi	0	DsBind	)r   r'   r(   r)   Zpacket_drsuapi_0r   r   r*   r#   r#   r$   test_packet_drsuapi_00   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_00c                 C   s&   t d}| t|| j| j d S )Nz 0.0	06	1	1	2	drsuapi	1	DsUnBind	)r   r'   r(   r)   Zpacket_drsuapi_1r   r   r*   r#   r#   r$   test_packet_drsuapi_01   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_01c                 C   s&   t d}| t|| j| j d S )Nz%0.0	06	1	1	2	drsuapi	2	DsReplicaSync	)r   r'   r,   r)   Zpacket_drsuapi_2r   r   r*   r#   r#   r$   test_packet_drsuapi_02   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_02c                 C   s&   t d}| t|| j| j d S )Nz&0.0	06	1	1	2	drsuapi	3	DsGetNCChanges	)r   r'   r,   r)   Zpacket_drsuapi_3r   r   r*   r#   r#   r$   test_packet_drsuapi_03   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_03c                 C   s&   t d}| t|| j| j d S )Nz+0.0	06	1	1	2	drsuapi	4	DsReplicaUpdateRefs	)r   r'   r,   r)   Zpacket_drsuapi_4r   r   r*   r#   r#   r$   test_packet_drsuapi_04   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_04c                 C   s&   t d}| t|| j| j d S )Nz%0.0	06	1	1	2	drsuapi	12	DsCrackNames	)r   r'   r(   r)   Zpacket_drsuapi_12r   r   r*   r#   r#   r$   test_packet_drsuapi_12   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_12c                 C   s&   t d}| t|| j| j d S )Nz*0.0	06	1	1	2	drsuapi	13	DsWriteAccountSpn	)r   r'   r(   r)   Zpacket_drsuapi_13r   r   r*   r#   r#   r$   test_packet_drsuapi_13   s    
z1TrafficEmulatorPacketTests.test_packet_drsuapi_13c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	1	2	epm	3	Map	)r   r'   r,   r)   Zpacket_epm_3r   r   r*   r#   r#   r$   test_packet_epm_03   s    
z-TrafficEmulatorPacketTests.test_packet_epm_03c                 C   s   t d}| t|| j| j | t| jj	
  | t| jj
  | t| jj
  | t| jj
  | t| jj
  | jt dS )z_Kerberos packets are not generated, but are used as a hint to
        favour kerberos.
        z0.0	11	1	1	2	kerberos			N)r   r'   r,   r)   Zpacket_kerberos_r   r   ZassertEqualr   r4   Zget_kerberos_stateZuser_creds_badZmachine_credsZmachine_creds_badr   r   r5   r   r*   r#   r#   r$   test_packet_kerberos   s.    





z/TrafficEmulatorPacketTests.test_packet_kerberosc                 C   s&   t d}| t|| j| j d S )Nz#0.0	06	1	1	2	ldap		*** Unknown ***	)r   r'   r,   r)   Zpacket_ldap_r   r   r*   r#   r#   r$   test_packet_ldap   s    
z+TrafficEmulatorPacketTests.test_packet_ldapc                 C   s&   t d}| t|| j| j d S )Nz80.0	06	1	2	1	ldap	0	bindRequest					3	sasl	1.3.6.1.5.5.2r   r'   r(   r)   Zpacket_ldap_0r   r   r*   r#   r#   r$   test_packet_ldap_00_sasl  s    
z3TrafficEmulatorPacketTests.test_packet_ldap_00_saslc                 C   s&   t d}| t|| j| j d S )Nz-0.0	06	1	2	1	ldap	0	bindRequest					0	simple	rC   r*   r#   r#   r$   test_packet_ldap_00_simple
  s    
z5TrafficEmulatorPacketTests.test_packet_ldap_00_simplec                 C   s&   t d}| t|| j| j d S )Nz!0.0	06	1	1	2	ldap	1	bindResponse	)r   r'   r,   r)   Zpacket_ldap_1r   r   r*   r#   r#   r$   test_packet_ldap_01  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_01c                 C   s&   t d}| t|| j| j d S )Nz"0.0	06	1	2	1	ldap	2	unbindRequest	)r   r'   r,   r)   Zpacket_ldap_2r   r   r*   r#   r#   r$   test_packet_ldap_02  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_02c                 C   s&   t d}| t|| j| j d S )Nz00.0	06	1	2	1	ldap	3	searchRequest	2	DC,DC		cn			)r   r'   r(   r)   Zpacket_ldap_3r   r   r*   r#   r#   r$   test_packet_ldap_03   s    
z.TrafficEmulatorPacketTests.test_packet_ldap_03c                 C   s&   t d}| t|| j| j d S )Nz#0.0	06	1	1	2	ldap	4	searchResEntry	)r   r'   r,   r)   Zpacket_ldap_4r   r   r*   r#   r#   r$   test_packet_ldap_04(  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_04c                 C   s&   t d}| t|| j| j d S )Nz"0.0	06	1	1	2	ldap	5	searchResDone	)r   r'   r,   r)   Zpacket_ldap_5r   r   r*   r#   r#   r$   test_packet_ldap_05/  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_05c                 C   s&   t d}| t|| j| j d S )Nz+0.0	06	1	2	1	ldap	6	modifyRequest					0	add)r   r'   r,   r)   Zpacket_ldap_6r   r   r*   r#   r#   r$   test_packet_ldap_066  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_06c                 C   s&   t d}| t|| j| j d S )Nz#0.0	06	1	1	2	ldap	7	modifyResponse	)r   r'   r,   r)   Zpacket_ldap_7r   r   r*   r#   r#   r$   test_packet_ldap_07>  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_07c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	ldap	8	addRequest	)r   r'   r,   r)   Zpacket_ldap_8r   r   r*   r#   r#   r$   test_packet_ldap_08E  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_08c                 C   s&   t d}| t|| j| j d S )Nz 0.0	06	1	1	2	ldap	9	addResponse	)r   r'   r,   r)   Zpacket_ldap_9r   r   r*   r#   r#   r$   test_packet_ldap_09L  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_09c                 C   s&   t d}| t|| j| j d S )Nz$0.0	06	1	2	1	ldap	16	abandonRequest	)r   r'   r,   r)   Zpacket_ldap_16r   r   r*   r#   r#   r$   test_packet_ldap_16S  s    
z.TrafficEmulatorPacketTests.test_packet_ldap_16c                 C   s&   t d}| t|| j| j d S )Nz 0.0	06	1	2	1	lsarpc	0	lsa_Close	r   r'   r,   r)   Zpacket_lsarpc_1r   r   r*   r#   r#   r$   test_packet_lsarpc_00Z  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_00c                 C   s&   t d}| t|| j| j d S )Nz!0.0	06	1	2	1	lsarpc	1	lsa_Delete	rP   r*   r#   r#   r$   test_packet_lsarpc_01a  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_01c                 C   s&   t d}| t|| j| j d S )Nz.0.0	06	1	2	1	lsarpc	2	lsa_EnumeratePrivileges	)r   r'   r,   r)   Zpacket_lsarpc_2r   r   r*   r#   r#   r$   test_packet_lsarpc_02h  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_02c                 C   s&   t d}| t|| j| j d S )Nz.0.0	06	1	2	1	lsarpc	3	lsa_QuerySecurityObject	)r   r'   r,   r)   Zpacket_lsarpc_3r   r   r*   r#   r#   r$   test_packet_lsarpc_03o  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_03c                 C   s&   t d}| t|| j| j d S )Nz,0.0	06	1	2	1	lsarpc	4	lsa_SetSecurityObject	)r   r'   r,   r)   Zpacket_lsarpc_4r   r   r*   r#   r#   r$   test_packet_lsarpc_04v  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_04c                 C   s&   t d}| t|| j| j d S )Nz)0.0	06	1	2	1	lsarpc	5	lsa_ChangePassword	)r   r'   r,   r)   Zpacket_lsarpc_5r   r   r*   r#   r#   r$   test_packet_lsarpc_05}  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_05c                 C   s&   t d}| t|| j| j d S )Nz%0.0	06	1	2	1	lsarpc	6	lsa_OpenPolicy	)r   r'   r,   r)   Zpacket_lsarpc_6r   r   r*   r#   r#   r$   test_packet_lsarpc_06  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_06c                 C   s&   t d}| t|| j| j d S )Nz'0.0	06	1	2	1	lsarpc	14	lsa_LookupNames	)r   r'   r(   r)   Zpacket_lsarpc_14r   r   r*   r#   r#   r$   test_packet_lsarpc_14  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_14c                 C   s&   t d}| t|| j| j d S )Nz&0.0	06	1	2	1	lsarpc	15	lsa_LookupSids	)r   r'   r(   r)   Zpacket_lsarpc_15r   r   r*   r#   r#   r$   test_packet_lsarpc_15  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_15c                 C   s&   t d}| t|| j| j d S )Nz70.0	06	1	2	1	lsarpc	39	lsa_QueryTrustedDomainInfoBySid	)r   r'   r(   r)   Zpacket_lsarpc_39r   r   r*   r#   r#   r$   test_packet_lsarpc_39  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_39c                 C   s&   t d}| t|| j| j d S )Nz00.0	06	1	2	1	lsarpc	40	lsa_SetTrustedDomainInfo	)r   r'   r,   r)   Zpacket_lsarpc_40r   r   r*   r#   r#   r$   test_packet_lsarpc_40  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_40c                 C   s&   t d}| t|| j| j d S )Nz,0.0	06	1	2	1	lsarpc	43	lsa_StorePrivateData	)r   r'   r,   r)   Zpacket_lsarpc_43r   r   r*   r#   r#   r$   test_packet_lsarpc_43  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_43c                 C   s&   t d}| t|| j| j d S )Nz/0.0	06	1	2	1	lsarpc	44	lsa_RetrievePrivateData	)r   r'   r,   r)   Zpacket_lsarpc_44r   r   r*   r#   r#   r$   test_packet_lsarpc_44  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_44c                 C   s&   t d}| t|| j| j d S )Nz(0.0	06	1	2	1	lsarpc	68	lsa_LookupNames3	)r   r'   r,   r)   Zpacket_lsarpc_68r   r   r*   r#   r#   r$   test_packet_lsarpc_68  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_68c                 C   s&   t d}| t|| j| j d S )Nz'0.0	06	1	2	1	lsarpc	76	lsa_LookupSids3	)r   r'   r(   r)   Zpacket_lsarpc_76r   r   r*   r#   r#   r$   test_packet_lsarpc_76  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_76c                 C   s&   t d}| t|| j| j d S )Nz(0.0	06	1	2	1	lsarpc	77	lsa_LookupNames4	)r   r'   r(   r)   Zpacket_lsarpc_77r   r   r*   r#   r#   r$   test_packet_lsarpc_77  s    
z0TrafficEmulatorPacketTests.test_packet_lsarpc_77c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	nbns	0	query	r   r'   r(   r)   Zpacket_nbns_0r   r   r*   r#   r#   r$   test_packet_nbns_00  s    
z.TrafficEmulatorPacketTests.test_packet_nbns_00c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	1	2	nbns	1	response	ra   r*   r#   r#   r$   test_packet_nbns_01  s    
z.TrafficEmulatorPacketTests.test_packet_nbns_01c                 C   s&   t d}| t|| j| j d S )Nz.0.0	06	1	2	1	rpc_netlogon	0	NetrLogonUasLogon	)r   r'   r,   r)   Zpacket_rpc_netlogon_0r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_00  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_00c                 C   s&   t d}| t|| j| j d S )Nz/0.0	06	1	2	1	rpc_netlogon	1	NetrLogonUasLogoff	)r   r'   r,   r)   Zpacket_rpc_netlogon_1r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_01  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_01c                 C   s&   t d}| t|| j| j d S )Nz30.0	06	1	2	1	rpc_netlogon	4	NetrServerReqChallenge	)r   r'   r,   r)   Zpacket_rpc_netlogon_4r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_04  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_04c                 C   s&   t d}| t|| j| j d S )Nz/0.0	06	1	2	1	rpc_netlogon	14	NetrLogonControl2	)r   r'   r,   r)   Zpacket_rpc_netlogon_14r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_14  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_14c                 C   s&   t d}| t|| j| j d S )Nz50.0	06	1	2	1	rpc_netlogon	15	NetrServerAuthenticate2	)r   r'   r,   r)   Zpacket_rpc_netlogon_15r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_15  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_15c                 C   s&   t d}| t|| j| j d S )Nz40.0	06	1	2	1	rpc_netlogon	21	NetrLogonDummyRoutine1	)r   r'   r,   r)   Zpacket_rpc_netlogon_21r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_21  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_21c                 C   s&   t d}| t|| j| j d S )Nz50.0	06	1	2	1	rpc_netlogon	26	NetrServerAuthenticate3	)r   r'   r,   r)   Zpacket_rpc_netlogon_26r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_26  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_26c                 C   s&   t d}| t|| j| j d S )Nz40.0	06	1	2	1	rpc_netlogon	29	NetrLogonGetDomainInfo	)r   r'   r(   r)   Zpacket_rpc_netlogon_29r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_29	  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_29c                 C   s&   t d}| t|| j| j d S )Nz40.0	06	1	2	1	rpc_netlogon	30	NetrServerPasswordSet2	)r   r'   r(   r)   Zpacket_rpc_netlogon_30r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_30  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_30c                 C   s&   t d}| t|| j| j d S )Nz-0.0	06	1	2	1	rpc_netlogon	34	DsrGetDcNameEx2	)r   r'   r,   r)   Zpacket_rpc_netlogon_34r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_34  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_34c                 C   s&   t d}| t|| j| j d S )Nz10.0	06	1	2	1	rpc_netlogon	39	NetrLogonSamLogonEx	)r   r'   r(   r)   Zpacket_rpc_netlogon_39r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_39  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_39c                 C   s&   t d}| t|| j| j d S )Nz60.0	06	1	2	1	rpc_netlogon	40	DsrEnumerateDomainTrusts	)r   r'   r(   r)   Zpacket_rpc_netlogon_40r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_40%  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_40c                 C   s&   t d}| t|| j| j d S )Nz80.0	06	1	2	1	rpc_netlogon	45	NetrLogonSamLogonWithFlags	)r   r'   r(   r)   Zpacket_rpc_netlogon_45r   r   r*   r#   r#   r$   test_packet_rpc_netlogon_45,  s    
z6TrafficEmulatorPacketTests.test_packet_rpc_netlogon_45c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	0	Connect	)r   r'   r(   r)   Zpacket_samr_0r   r   r*   r#   r#   r$   test_packet_samr_003  s    
z.TrafficEmulatorPacketTests.test_packet_samr_00c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	1	Close	)r   r'   r(   r)   Zpacket_samr_1r   r   r*   r#   r#   r$   test_packet_samr_01:  s    
z.TrafficEmulatorPacketTests.test_packet_samr_01c                 C   s&   t d}| t|| j| j d S )Nz"0.0	06	1	2	1	samr	3	QuerySecurity	)r   r'   r(   r)   Zpacket_samr_3r   r   r*   r#   r#   r$   test_packet_samr_03A  s    
z.TrafficEmulatorPacketTests.test_packet_samr_03c                 C   s&   t d}| t|| j| j d S )Nz!0.0	06	1	2	1	samr	5	LookupDomain	)r   r'   r(   r)   Zpacket_samr_5r   r   r*   r#   r#   r$   test_packet_samr_05H  s    
z.TrafficEmulatorPacketTests.test_packet_samr_05c                 C   s&   t d}| t|| j| j d S )Nz 0.0	06	1	2	1	samr	6	EnumDomains	)r   r'   r(   r)   Zpacket_samr_6r   r   r*   r#   r#   r$   test_packet_samr_06O  s    
z.TrafficEmulatorPacketTests.test_packet_samr_06c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	7	OpenDomain	)r   r'   r(   r)   Zpacket_samr_7r   r   r*   r#   r#   r$   test_packet_samr_07V  s    
z.TrafficEmulatorPacketTests.test_packet_samr_07c                 C   s&   t d}| t|| j| j d S )Nz%0.0	06	1	2	1	samr	8	QueryDomainInfo'	)r   r'   r(   r)   Zpacket_samr_8r   r   r*   r#   r#   r$   test_packet_samr_08]  s    
z.TrafficEmulatorPacketTests.test_packet_samr_08c                 C   s&   t d}| t|| j| j d S )Nz$0.0	06	1	2	1	samr	14	CreateDomAlias	)r   r'   r,   r)   Zpacket_samr_14r   r   r*   r#   r#   r$   test_packet_samr_14d  s    
z.TrafficEmulatorPacketTests.test_packet_samr_14c                 C   s&   t d}| t|| j| j d S )Nz'0.0	06	1	2	1	samr	15	EnumDomainAliases	)r   r'   r(   r)   Zpacket_samr_15r   r   r*   r#   r#   r$   test_packet_samr_15k  s    
z.TrafficEmulatorPacketTests.test_packet_samr_15c                 C   s&   t d}| t|| j| j d S )Nz(0.0	06	1	2	1	samr	16	GetAliasMembership	)r   r'   r(   r)   Zpacket_samr_16r   r   r*   r#   r#   r$   test_packet_samr_16r  s    
z.TrafficEmulatorPacketTests.test_packet_samr_16c                 C   s&   t d}| t|| j| j d S )Nz!0.0	06	1	2	1	samr	17	LookupNames	)r   r'   r(   r)   Zpacket_samr_17r   r   r*   r#   r#   r$   test_packet_samr_17y  s    
z.TrafficEmulatorPacketTests.test_packet_samr_17c                 C   s&   t d}| t|| j| j d S )Nz 0.0	06	1	2	1	samr	18	LookupRids	)r   r'   r(   r)   Zpacket_samr_18r   r   r*   r#   r#   r$   test_packet_samr_18  s    
z.TrafficEmulatorPacketTests.test_packet_samr_18c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	19	OpenGroup	)r   r'   r(   r)   Zpacket_samr_19r   r   r*   r#   r#   r$   test_packet_samr_19  s    
z.TrafficEmulatorPacketTests.test_packet_samr_19c                 C   s&   t d}| t|| j| j d S )Nz&0.0	06	1	2	1	samr	25	QueryGroupMember	)r   r'   r(   r)   Zpacket_samr_25r   r   r*   r#   r#   r$   test_packet_samr_25  s    
z.TrafficEmulatorPacketTests.test_packet_samr_25c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	34	OpenUser	)r   r'   r(   r)   Zpacket_samr_34r   r   r*   r#   r#   r$   test_packet_samr_34  s    
z.TrafficEmulatorPacketTests.test_packet_samr_34c                 C   s&   t d}| t|| j| j d S )Nz#0.0	06	1	2	1	samr	36	QueryUserInfo	)r   r'   r(   r)   Zpacket_samr_36r   r   r*   r#   r#   r$   test_packet_samr_36  s    
z.TrafficEmulatorPacketTests.test_packet_samr_36c                 C   s&   t d}| t|| j| j d S )Nz!0.0	06	1	2	1	samr	37	SetUserInfo	)r   r'   r,   r)   Zpacket_samr_37r   r   r*   r#   r#   r$   test_packet_samr_37  s    
z.TrafficEmulatorPacketTests.test_packet_samr_37c                 C   s&   t d}| t|| j| j d S )Nz&0.0	06	1	2	1	samr	39	GetGroupsForUser	)r   r'   r(   r)   Zpacket_samr_39r   r   r*   r#   r#   r$   test_packet_samr_39  s    
z.TrafficEmulatorPacketTests.test_packet_samr_39c                 C   s&   t d}| t|| j| j d S )Nz&0.0	06	1	2	1	samr	40	QueryDisplayInfo	)r   r'   r,   r)   Zpacket_samr_40r   r   r*   r#   r#   r$   test_packet_samr_40  s    
z.TrafficEmulatorPacketTests.test_packet_samr_40c                 C   s&   t d}| t|| j| j d S )Nz#0.0	06	1	2	1	samr	44	GetUserPwInfo	)r   r'   r,   r)   Zpacket_samr_44r   r   r*   r#   r#   r$   test_packet_samr_44  s    
z.TrafficEmulatorPacketTests.test_packet_samr_44c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	57	Connect2	)r   r'   r(   r)   Zpacket_samr_57r   r   r*   r#   r#   r$   test_packet_samr_57  s    
z.TrafficEmulatorPacketTests.test_packet_samr_57c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	64	Connect5	)r   r'   r(   r)   Zpacket_samr_64r   r   r*   r#   r#   r$   test_packet_samr_64  s    
z.TrafficEmulatorPacketTests.test_packet_samr_64c                 C   s&   t d}| t|| j| j d S )Nz0.0	06	1	2	1	samr	68		)r   r'   r,   r)   Zpacket_samr_68r   r   r*   r#   r#   r$   test_packet_samr_68  s    
z.TrafficEmulatorPacketTests.test_packet_samr_68c                 C   s&   t d}| t|| j| j d S )Nz'0.0	06	1	2	1	srvsvc	16	NetShareGetInfo	)r   r'   r(   r)   Zpacket_srvsvc_16r   r   r*   r#   r#   r$   test_packet_srvsvc_16  s    
z0TrafficEmulatorPacketTests.test_packet_srvsvc_16c                 C   s&   t d}| t|| j| j d S )Nz%0.0	06	1	2	1	srvsvc	21	NetSrvGetInfo	)r   r'   r(   r)   Zpacket_srvsvc_21r   r   r*   r#   r#   r$   test_packet_srvsvc_21  s    
z0TrafficEmulatorPacketTests.test_packet_srvsvc_21)___name__
__module____qualname__r   r%   r+   r-   r.   r/   r0   r1   r2   r3   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   r   __classcell__r#   r#   r!   r$   r
   #   s   7	r
   )r   r&   r   Z
samba.authr   Zsamba.credentialsr   r   Zsamba.emulater   r)   r   Zsamba.emulate.trafficr   Zsamba.samdbr   Zsamba.testsr   r	   ZtestsZTestCaser
   r#   r#   r#   r$   <module>   s   