
    b                         d dl mZ d dlZd dlZd dlmZmZmZ d dlmZm	Z	 d dl
Z
d dlmZmZmZ 	  G d de          ZdS )    )TestCaseN)CredentialsDONT_USE_KERBEROSMUST_USE_KERBEROS)NTSTATUSErrorntstatus)srvsvcsamrlsac                   4     e Zd Z fdZ fdZd Zd Z xZS )NtlmDisabledTestsc                 J   t          t          |                                            |                                 | _        t          j        d          | _        t                      | _	        | j	        
                    | j                   | j	                            t          j        d                     | j	                            | j                   | j	                            t          j        d                     | j	                            t                     d S )NSERVERUSERNAMEPASSWORD)superr   setUpget_loadparmlposgetenvserverr   credsguessset_username
set_domainset_passwordset_kerberos_stater   self	__class__s    :/usr/lib/python3/dist-packages/samba/tests/ntlmdisabled.pyr   zNtlmDisabledTests.setUp$   s    &&,,...,,..9X.. ]]

!!!
	* 5 5666
dk***
	* 5 5666
%%&788888    c                 V    t          t          |                                            d S )N)r   r   tearDownr   s    r"   r%   zNtlmDisabledTests.tearDown1   s&    &&//11111r#   c                 H   	 t          j         d| j        z  | j        | j                  }|                     |           d S # t
          $ rV}t          j        |j        d                   j	        }|t          j        k    r|                     d           n Y d }~d S d }~ww xY w)Nzncacn_np:%s[smb2,ntlm]r   NTLM is disabled on this server)r	   r   r   r   assertIsNotNoner   ctypesc_uint32argsvaluer   NT_STATUS_NTLM_BLOCKEDfail)r    conneenums       r"   test_ntlm_connectionz&NtlmDisabledTests.test_ntlm_connection4   s    
	=!9DK!GRVR\]]D  &&&&& 	 	 	?16!9--3Dx666		;<<<< =<<<<<		s   =A 
B!ABB!c           	      X   | j                             t                     t          j        dt	          j        d          z            }t          j                    }| j        |_	        t          j                    }t	          j        d          |_	        	 |
                    ||d d dd d            d S # t          $ rm}t          j        |j        d                   j        }|t           j        k    r|                     d           n|t           j        k    rn Y d }~d S Y d }~d S d }~ww xY w)Nzncacn_np:%s[krb5,seal,smb2]r   r   Tr   r'   )r   r   r   r
   r   r   r   Stringr   stringChangePasswordUser2r   r)   r*   r+   r,   r   r-   r.   NT_STATUS_WRONG_PASSWORD)r    r/   r   usernamer0   r1   s         r"   test_samr_change_passwordz+NtlmDisabledTests.test_samr_change_passwordA   s'   
%%&7888y689L9LLMM :<<)J//	$$VXtT4tTTTTT 		 		 		?16!9--3Dx666		;<<<<::: =<<<<< 		s   B2 2
D)<AD$$D))__name__
__module____qualname__r   r%   r2   r9   __classcell__)r!   s   @r"   r   r   "   so        9 9 9 9 92 2 2 2 2        r#   r   )samba.testsr   r   sambasamba.credentialsr   r   r   r   r   r)   samba.dcerpcr	   r
   r   r    r#   r"   <module>rC      s   $ !           				  O O O O O O O O O O ) ) ) ) ) ) ) )  * * * * * * * * * *
5 5 5 5 5 5 5 5 5 5r#   