Current File : //opt/alt/python37/lib/python3.7/site-packages/clcommon/__pycache__/clpwd.cpython-37.pyc
B
E!wdJ � @ s� d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dlmZ ddl m
Z
G dd � d �Z dd d �Z ed
d� �Z
dd� Zddd�ZdS )� )�absolute_import)�division)�print_functionN)�contextmanager� )�FormattedExceptionc @ s� e Zd ZdZG dd� de�Zd dd�Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d d �ZdS ) �ClPwdz/etc/login.defsc s e Zd Z� fdd�Z� ZS )zClPwd.NoSuchUserExceptionc s( t tj| �jdd|id�f|�|� d S )NzNo such user (%(user)s)�user)�message�context)�superr �NoSuchUserException�__init__)�selfr �args�kwargs)� __class__� �?/opt/alt/python37/lib/python3.7/site-packages/clcommon/clpwd.pyr s z"ClPwd.NoSuchUserException.__init__)�__name__�
__module__� __qualname__r �
__classcell__r r )r r r
s r
Nc C s8 i | _ i | _i | _i | _|d kr.| �d�| _n|| _d S )Ni� )�
_user_key_map� _uid_key_map�_user_full_map�
_uid_full_map�get_sys_min_uid�_min_uid)r Zmin_uidr r r r s zClPwd.__init__c C s | � � | jS )N)�_load_passwd_databaser )r r r r �
get_user_dict* s zClPwd.get_user_dictc C s | � � | jS )N)r r )r r r r � get_uid_dict. s zClPwd.get_uid_dictc C s | � � | jS )N)r r )r r r r �get_user_full_dict2 s zClPwd.get_user_full_dictc C s | � � | jS )N)r r )r r r r �get_uid_full_dict6 s zClPwd.get_uid_full_dictc C s0 y | � � | S tk
r* t�|��Y nX dS )z*
Return pw_entry for user
N)r" �KeyErrorr r
)r r r r r �get_pw_by_name: s zClPwd.get_pw_by_namec C s0 y | � � | S tk
r* t�|��Y nX dS )z7
Return list of passwd entries for uid
N)r# r$ r r
)r �uidr r r �
get_pw_by_uidC s zClPwd.get_pw_by_uidc C s2 y| � � | jS tk
r, t�|��Y nX dS )z&
Returns uid for user
N)r" �pw_uidr$ r r
)r r r r r �get_uidL s z
ClPwd.get_uidc C s2 y| � � | jS tk
r, t�|��Y nX dS )z`
Returns homedir for a user
@param user: string
@return: string
N)r" �pw_dirr$ r r
)r r r r r � get_homedirU s zClPwd.get_homedirc C s� | j s�x�t�� D ]|}|| j|j< |j| j kr8g | j |j< | j |j �|� |j| jkr|| j|j< |j| j krzg | j |j< | j |j �|� qW dS )zZ
Loads the passwd database and fills user_to_uid and user_to_homedir maps
N)
r �pwdZgetpwallr �pw_namer( �appendr r r )r �entryr r r r ` s zClPwd._load_passwd_databasec C s> y| � � | }W n tk
r. t�|��Y nX dd� |D �S )zs
Return names of users with uid specified
@param uid: int
@return: list of strings
c S s g | ]
}|j �qS r )r- )�.0r/ r r r �
<listcomp>{ s z#ClPwd.get_names.<locals>.<listcomp>)r# r$ r r
)r r& �entriesr r r � get_namesp s
zClPwd.get_names�� c C sv t j�| j�rrt| jd��}|�� }W dQ R X xB|D ]:}|�d�r4yt|�d�d � � �S t
k
rl Y q4X q4W |S )z�
Return system defined MIN_UID from /etc/login.def or def_min_uid
@param def_min_uid: int
@return: MIN_UID: int
�rNZUID_MINr ) �os�path�exists�LOGIN_DEF_FILE�open� readlines�
startswith�int�split�strip�
ValueError)r Z def_min_uid�f�lines�liner r r r } s
zClPwd.get_sys_min_uid)N)r4 )r r r r9 r r
r r r! r" r# r% r' r) r+ r r3 r r r r r r s
r Tc s� t �� }|dkr�� r�t�� �}� fdd�t�� D �}|j}||krL|�|� t �|� |rtt � |j� t �
|j � nt � |j� t �
|j � |r�� t jd< |jt jd< dS )z�
Drop current root privileges to user
:param effective_or_real: if True - drop euid, else - drop ruid
:param user: name of unix user
:param set_env: bool -> if true set $HOME and $USER env variables
:return: None
r c s g | ]}� |j kr|j�qS r )Zgr_memZgr_gid)r0 �group)r r r r1 � s z(drop_user_privileges.<locals>.<listcomp>ZUSER�HOMEN)r6 �geteuidr, �getpwnam�grpZgetgrallZpw_gidr. � setgroups�setegid�seteuidr( �setgid�setuid�environr* )r �effective_or_real�set_env� current_euidZuser_pwdZ user_groupsZmain_user_groupr )r r �drop_user_privileges� s
rR c c s~ t �� t �� t �� }}}yt| ddd� W n tk
rL t�| ��Y nX z
dV W dt �|� t � |� t �
|� X dS )zk
Temporary drop privileges to some user
:type user: str
:raises: ClPwd.NoSuchUserException
TF)rO rP N) r6 �getuid�getgid� getgroupsrR r$ r r
rK rJ rI )r Zold_uidZold_gidZ
old_groupsr r r �drop_privileges� s
rV c C sl | dkrt d��ddlm} ddlm} || �}y|d \}}W n& tk
rf |dd| id ���Y nX |S )
zw
Resolve document root by username
:param user: str -> name of unix user
:return: str -> document root
Nz-Cannot resolve docroot without specified userr )� userdomains)�NoDomainr z!No domain for user %(user)s foundr )r
r )r �cpapirW �cpapi.cpapiexceptionsrX �
IndexError)r rW rX Z domains_list�_�resultr r r �_resolve_doc_root_by_user� s r^ c C s� ddl m} ddlm} | }d}t�� }|dk r`||�\}}| dkrJ|}n| |krZ|d��|}n2| dk rtt| d�}n |dkr�t�|�j }t|d�}||fS )a
Resolve username and doc_root by domain,
or resolve document root by username,
or resolve document root and username by effective uid
:param user: str -> name of unix user
:param domain: str -> domain of panel user
:return: tuple -> user, doc_root
r )�docroot)�
IncorrectDataNz User and domain are in conflict)r r )
rY r_ rZ r` r6 rF r^ r, �getpwuidr- ) r Zdomainr_ r` Z result_userZresult_doc_rootrQ Zdoc_rootZ domain_userr r r � resolve_username_and_doc_root� s$
rb )NTT)NN)Z
__future__r r r r6 r, rH �
contextlibr Z clexceptionr r rR rV r^ rb r r r r �<module> s |
Mini Shell