File: //usr/lib/python3/dist-packages/samba/__pycache__/getopt.cpython-310.pyc
o
�\ e @ � @ s� d Z dZddlZddlmZ ddlZddlmZmZmZm Z ddl
Z
G dd� dej �Z G dd � d ej �Z
d
d � Zd d
� ZG dd� dej �ZG dd� de�Zdd� ZG dd� dej�ZdS )z7Support for parsing Samba-related command-line options.�restructuredText� N)�copy)� Credentials�AUTO_USE_KERBEROS�DONT_USE_KERBEROS�MUST_USE_KERBEROSc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� SambaOptionsz+General Samba-related command line options.c C s� ddl m} |� ddlm} tj�| |d� | jdddtdd | j d
� | jd d dtd
d| j
d
� | jddtdd| j d
� | jddtdd| j d
� d | _
|� | _d | _d S )Nr )� fault_setup)�LoadParmzSamba Common Optionsz-sz --configfile�callback�FILEzConfiguration file)�action�type�metavar�helpr z-dz --debuglevel�
DEBUGLEVEL� debug levelz--option�OPTIONz%set smb.conf option from command linez--realm�REALMzset the realm name)�sambar � samba.paramr
�optparse� OptionGroup�__init__�
add_option�str�_load_configfile�_set_debuglevel� _set_option�
_set_realm� _configfile�_lp�realm)�self�parserr r
� r% �./usr/lib/python3/dist-packages/samba/getopt.pyr % s0
�
���
zSambaOptions.__init__c C s | j S )z?Return path to the smb.conf file specified on the command line.�r �r# r% r% r& �get_loadparm_path; s z SambaOptions.get_loadparm_pathc C �
|| _ d S �Nr' �r# �option�opt_str�argr$ r% r% r& r ? �
z SambaOptions._load_configfilec C s | j �d|� ||j_d S )Nr )r! �set�values�
debuglevelr, r% r% r& r B s z SambaOptions._set_debuglevelc C s | j �d|� || _d S )Nr" )r! r1 r" r, r% r% r& r F s
zSambaOptions._set_realmc
C sj |� d�dkr t�d��|�d�}z| j�|d |d � W d S ty4 } z t�d||f ��d }~ww )N�=���z&--option option takes a 'a=b' argumentr � z$invalid --option option value %r: %s)�findr �OptionValueError�splitr! r1 � Exception)r# r- r. r/ r$ �a�er% r% r& r J s �
���zSambaOptions._set_optionc C sT | j dur| j�| j � | jS t�d�dur"| j�t�d�� | jS | j�� | jS )z?Return loadparm object with data specified on the command line.N�
SMB_CONF_PATH)r r! �load�os�getenv� load_defaultr( r% r% r& � get_loadparmU s
�
�zSambaOptions.get_loadparmN) �__name__�
__module__� __qualname__�__doc__r r) r r r r rB r% r% r% r& r " s r c @ s e Zd ZdZdd� Zdd� ZdS )�VersionOptionsz/Command line option for printing Samba version.c C s* t j�| |d� | jddd| jdd� d S )NzVersion Optionsz-Vz --versionr zDisplay version number)r
r r )r r r r �_display_version�r# r$ r% r% r& r b s
�zVersionOptions.__init__c C s dd l }t|j� t�d� d S )Nr )r �print�version�sys�exit)r# r- r. r/ r$ r r% r% r& rH h s
z VersionOptions._display_versionN)rC rD rE rF r rH r% r% r% r& rG ` s rG c C sB | � � dv rtS | � � dv rtS | � � dv rtS t�d|| f ��)N)�yes�true�1)�no�false�0)�auto�invalid %s option value: %s)�lowerr r r r r8 �r/ r. r% r% r& �parse_kerberos_arg_legacyn � �rX c C sB | � � dkrtS | � � dkrtS | � � dkrtS t�d|| f ��)N�required�desired�offrU )rV r r r r r8 rW r% r% r& �parse_kerberos_argz rY r] c @ s� e Zd ZdZd!dd�Zd"dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd#d d �ZdS )$�CredentialsOptionsz0Command line options for specifying credentials.Nc C sD || _ |d ur
d| | _nd| _d| _d | _d| _tj�| || j� | jddd| j t
dd � | jd
d dd t
| j d
� | jddddt
d| j d� | jddddt
d| j
d� | jdddd| jd� | jdd d dt
d | jd� | jd d dd!| jd� | jd"d#dt
d$| jd� | jd%d&dt
d'| jd� | jd(d)d*dt
d+| jd� t� | _d S ),NzCredentials Options (%s)zCredentials OptionsTFz--simple-bind-dn�DNr �DN to use for a simple bind�r r
r r r z
--password�PASSWORD�Password�r r
r r r z-Uz
--username�USERNAME�Username�r r
r r r z-Wz --workgroup� WORKGROUP� Workgroupz-Nz --no-passzDon't ask for a password)r
r r � z --ipaddress� IPADDRESSzIP address of serverz-Pz--machine-passz#Use stored machine account passwordz--use-kerberos�desired|required|off�Use Kerberos authenticationz--use-krb5-ccache�
KRB5CCNAMEzKerberos Credentials cachez-kz
--kerberos�KERBEROSz%DEPRECATED: Migrate to --use-kerberos)� special_name�section�ask_for_password� ipaddress� machine_passr r r � _add_option�_set_simple_bind_dnr �
_set_password�_parse_username�_parse_workgroup�_set_no_password�_set_ipaddress�_set_machine_pass�
_set_kerberos�_set_krb5_ccache�_set_kerberos_legacyr �creds)r# r$ rp r% r% r& r � sf
�
�
�
�
�
����
� zCredentialsOptions.__init__�passwordc C s ddl }zddl}W n ty# d| }tj�|� tj�� Y dS w d|�|�|�|�f }|�|�}|� � } |�
| �sOd| }tj�|� tj�� dS d}
d||f } |� |d | �} | | krid||
f }
nd
||
f }
|� |d |
d | �}|�|� dS )
zC Make sure no sensitive data (e.g. password) resides in proctitle. r Nz^WARNING: Using %s on command line is insecure. Please install the setproctitle python module.
Fz(.*[ ]+)(%s[= ]%s)([ ]*.*)z;Unable to hide %s in proctitle. This is most likely a bug!
�xxxz%s=%sz\2z%s %sz\1z\3) �re� setproctitle�ModuleNotFoundErrorrL �stderr�write�flush�escape�compile� getproctitle�match�sub)r# r. � secret_data� data_typer� r� �msg�pass_opt_re_str� pass_opt_re�
cur_proctitle�secret_data_replacer�opt_pass_with_eq�
opt_pass_part� replace_str�
new_proctitler% r% r& �_ensure_secure_proctitle� s@ �
�
�
�z+CredentialsOptions._ensure_secure_proctitlec O s` | j d u r
| j|i |��S d}|D ]}|�d�sq||�dd| j �f7 }q| j|i |�� d S )Nr% z--z--%s-)rp r �
startswith�replace)r# �args1�kwargs�args2r; r% r% r& ru � s
z CredentialsOptions._add_optionc C � | j �|� d| _d S �NF)r� � parse_stringrt r, r% r% r& rx � �
z"CredentialsOptions._parse_usernamec C � | j �|� d S r+ )r� �
set_domainr, r% r% r& ry � � z#CredentialsOptions._parse_workgroupc C s* | � ||d� | j�|� d| _d| _d S )Nr� F)r� r� � set_passwordrr rt r, r% r% r& rw � s
z CredentialsOptions._set_passwordc C �
d| _ d S r� )rr r, r% r% r& rz � r0 z#CredentialsOptions._set_no_passwordc C r� )NT)rt r, r% r% r& r| r0 z$CredentialsOptions._set_machine_passc C r* r+ )rs r, r% r% r& r{ r0 z!CredentialsOptions._set_ipaddressc C s t d� | j�t||�� d S )Nz0WARNING: The option -k|--kerberos is deprecated!)rJ r� �set_kerberos_staterX r, r% r% r& r s z'CredentialsOptions._set_kerberos_legacyc C � | j �t||�� d S r+ )r� r� r] r, r% r% r& r} � z CredentialsOptions._set_kerberosc C r� r+ )r� � set_bind_dnr, r% r% r& rv r� z&CredentialsOptions._set_simple_bind_dnc C s | j �t� | j �|� d S r+ )r� r� r �set_named_ccacher, r% r% r& r~ s z#CredentialsOptions._set_krb5_ccacheFc C sp | j �|� | jr| j �|� n| jr| j �� |r5| j �� s5z
| j �|� W | j S ty4 Y | j S w | j S )z�Obtain the credentials set on the command-line.
:param lp: Loadparm object to use.
:return: Credentials object
)r� �guessrt �set_machine_accountrr �set_cmdline_callbacks�authentication_requestedr: )r# �lp�fallback_machiner% r% r& �get_credentials s
��z"CredentialsOptions.get_credentialsr+ )r� )F)rC rD rE rF r r� ru rx ry rw rz r| r{ r r} rv r~ r� r% r% r% r& r^ � s
// r^ c @ sR e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
ddd�Z dS )�CredentialsOptionsDoublez?Command line options for specifying credentials of two servers.c C s� t �| |� d| _| jddd| jtdd� | jdddd t| jd
� | jd d dtd
| jd� | jdddtd| jd� | jdddd� | jdddtd| j d� | jdddtd| j
d� t � | _ d S ) NTz--simple-bind-dn2�DN2r r` ra z --password2� PASSWORD2rc rd z --username2� USERNAME2zUsername for second serverrg z --workgroup2�
WORKGROUP2zWorkgroup for second serverz
--no-pass2�
store_truez.Don't ask for a password for the second server)r
r z--use-kerberos2rl rm z --kerberos2� KERBEROS2z Use Kerberos)
r^ r �no_pass2r �_set_simple_bind_dn2r �_set_password2�_parse_username2�_parse_workgroup2�_set_kerberos2�_set_kerberos2_legacyr �creds2rI r% r% r& r 0 s@
�
������ z!CredentialsOptionsDouble.__init__c C r� r+ )r� r� r, r% r% r& r� M r� z)CredentialsOptionsDouble._parse_username2c C r� r+ )r� r� r, r% r% r& r� P r� z*CredentialsOptionsDouble._parse_workgroup2c C r� r� )r� r� r� r, r% r% r& r� S r� z'CredentialsOptionsDouble._set_password2c C r� r+ �r� r� r] r, r% r% r& r� W r� z.CredentialsOptionsDouble._set_kerberos2_legacyc C r� r+ r� r, r% r% r& r� Z r� z'CredentialsOptionsDouble._set_kerberos2c C r� r+ )r� r� r, r% r% r& r� ] r� z-CredentialsOptionsDouble._set_simple_bind_dn2Tc C s< |r | j �|� n
| j �� s| j �� | jr| j �� | j S )z�Obtain the credentials set on the command-line.
:param lp: Loadparm object to use.
:param guess: Try guess Credentials from environment
:return: Credentials object
)r� r� � get_username�
set_anonymousr� r� )r# r� r� r% r% r& �get_credentials2` s
z)CredentialsOptionsDouble.get_credentials2N)T) rC rD rE rF r r� r� r� r� r� r� r� r% r% r% r& r� - s r� c
C s� ddddd�}|� dd�}d}|r|dd� �� s!|dd� d kr?||d 7 }|dd � }|r7|dd� �� s!|dd� d ks!zt|�}W n tyV d
�|�}t�|��w |�� � d d�}z||| W S ty} } z d �||�}t�|��d } ~ ww )
Nr6 i i i @)�B�KB�MB�GB� rj r �.zA{0} option requires a numeric value, with an optional unit suffix�Iz3{0} invalid suffix '{1}', should be B, Kb, Mb or Gb) r� �isdigit�float�
ValueError�formatr r8 �upper�KeyError)
r- �opt�value� multipliers�v�digits�mr� �suffix�kr% r% r& � check_bytest s: � $ $� �
��
��r� c @ s, e Zd Zejjd Zeejj�Zeed< dS )� SambaOption)�bytesr� N) rC rD rE r �Option�TYPESr � TYPE_CHECKERr� r% r% r% r& r� � s r� )rF �
__docformat__r r r? �samba.credentialsr r r r rL r r rG rX r] r^ r� r� r� r� r% r% r% r& �<module> s > (G#