Edit File: clpwd.cpython-37.pyc
B � eJ � @ 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__� ��/builddir/build/BUILDROOT/alt-python27-cllib-3.3.5-1.el8.cloudlinux.x86_64/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 |