One Hat Cyber Team
Your IP :
18.118.140.120
Server IP :
192.145.235.60
Server :
Linux ngx365.inmotionhosting.com 5.14.0-427.33.1.el9_4.x86_64 #1 SMP PREEMPT_DYNAMIC Fri Aug 30 09:45:56 EDT 2024 x86_64
Server Software :
Apache
PHP Version :
8.2.27
Buat File
|
Buat Folder
Eksekusi
Dir :
~
/
opt
/
tier2c
/
__pycache__
/
View File Name :
safe_restorepkg.cpython-39.pyc
a �g�) � @ s@ d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl Z ddl Z ddl Z ddl m Z mZmZ ddlmZmZ ddlmZ dd lmZ ddlZe j�dd � ddlZdd lmZ dd l m Z m Z ed �� � r�d Z ned�� � r�dZ ndZ eG dd� d��Z!e!d�dd�Z"e#e#d�dd�Z$e#ed�dd�Z%d;e e#e&d�d d �Z'e e#e#e&d �d d �Z(e e#eedf e&d!�d"d#�Z)e e#ee#df e&d$�d%d&�Z*e eee#df e&d'�d(d)�Z+e!ee#ee,e# d*�d+d,�Z-eee.e#ef ddf d-�d.d/�Z/d0d1� Z0e1e#e,e# f d2�d3d4�Z2e,e# d5�d6d7�Z3e,e# d5�d8d9�Z4e5d:k�r<e0� dS )<z0Wrapper for /usr/local/cpanel/scripts/restorepkg� N)� dataclass)�ArgumentTypeError)�Path)�IO� Generator�Union)�whmapi1� CpAPIError)�Proc)� IPAddressz/opt/support/lib)� CPMOVE_RE)� MAIN_RESELLER�ROLEz /opt/sharedrads/hostsfilemodsz/opt/dedrads/hostsfilemodsc @ sz e Zd ZU dZeedf ed< eed< eed< eed< eed< eedf ed< eedf ed < eed < e ed < e ed < dS ) �ArgszType hint for get_argsN�newuser�owner�quiet�yes� host_mods�ipaddr�package�fixperms�path�log_dir) �__name__� __module__� __qualname__�__doc__r �str�__annotations__�boolr r � r! r! � ./opt/tier2c/safe_restorepkg.pyr s r )�returnc C sb t jtd�} | jddtjdd� tr,tdkrH| jddd ttd t� �d � | jd d ddd� | jdddddd� t r�| jdddddd� | jdddd d d � | jd d tj d!d� d"d#� t d$�� � D �}| jd%d&d'd(d)|d*|��d+� | jd,t d-d.� | �� }|j�� �r |j�r | �� t�d/� t�r6tdk�r6t|_t �sBd0|_t�rTt d1�|_n t d2�|_|S )3zParse sys.argv)� description� --newuserr zwAllows you to restore to the username in AMP without having to modify account. Will be ignored if restoring a directory)�dest�type�helpzshared:resellerz--ownerz-or z)Set Ownership to a reseller. Defaults to )r& �defaultr' r( z --no-fixpermsr Z store_falsez#Do not run fixperms after restoring)r& �actionr( �--quietz-q� print_logsz0Silence restorepkg output (it still gets logged)z --host-modsz-mr � store_truez=Print host file mod entries at the end for all restored usersz--yesz-yr zNo Confirmation Prompts)r* r& r( z--ipr zSet an IP addressc S s h | ]}|� � r|j�qS r! )�is_file�name)�.0�xr! r! r" � <setcomp>d s zget_args.<locals>.<setcomp>z/var/cpanel/packagesz--pkgz-pz-PZPKGr zSet a package type )�metavarr& �choicesr( r z4Path to the backup file or directory of backup files)r' r( z2 --newuser invalid when restoring from a directoryFz /home/t1binz/var/log/t1bin)�argparse�ArgumentParserr � add_argument� arg_typesZvalid_usernamer r �existing_reseller� HOSTFILEMODSZ ipaddressr �iterdir�restorable_path� parse_argsr �is_dirr Z print_help�sys�exitr r r )�parserZpackages�argsr! r! r" �get_args/ s� � ������ �� � rC )�userr# c C s� | s t d��| dkr| S t�| �s2t d| � d���zbtddd��B}|D ],}|�| � d��rF| W d � W S qFW d � n1 s�0 Y W n ty� td tjd � Y n0 t | � d ���d S ) zDArgparse type: validate a user as existing with reseller permissionszcannot be blank�rootz reseller z does not existz/var/cpanel/resellers�ascii)�encoding�:Nz$/var/cpanel/resellers does not exist��filez not setup as a reseller) �BadArg�rads� is_cpuser�open� startswith�FileNotFoundError�printr? �stderr)rD �handle�liner! r! r" r9 � s < r9 )�str_pathr# c C sv z t �| �W S ty Y n0 zt �| �}W n. ty\ } ztd�|�W Y d}~n d}~0 0 |td�krrtd��|S )zSArgparse type: validates a path as either a cpmove file or a directory in /homez9not a cPanel backup or directory in /home containing themNz/homezQinvalid path; when restoring from a directory, it must be a subdirectory of /home)r8 Zcpmove_file_typerK Z path_in_homer )rU r �excr! r! r" r< � s" �� �r< T�rS �msgZshowc C s2 |� d�s|� d�}| �|� |r.t|dd� dS )z$Writes to a log and prints to stdout� � )�endN)�endswith�writerQ rW r! r! r" � log_print� s r^ )�log_filerD r r# c C sn |dkr dS t | d|� d|� �� zt�||� W n6 tyh } z t | d|� �� W Y d}~dS d}~0 0 dS )zChange a user's ownerrE Tzsetting owner of � to zmodifyacct failed: NF)r^ r � set_ownerr )r_ rD r rV r! r! r" ra � s ra )r_ rD r r# c C sn |sdS t | d|� d|� �� zt�|t|�� W n6 tyh } z t | d|� �� W Y d}~dS d}~0 0 dS )zSet a user's IPTzsetting IP of r` zsetsiteip failed: NF)r^ r Z setsiteipr r )r_ rD r rV r! r! r" �set_ip� s rb )r_ rD �pkgr# c C sj |sdS t | d|� d|� �� zt�||� W n6 tyd } z t | d|� �� W Y d}~dS d}~0 0 dS )zSet a user's cPanel packageTzsetting package of r` zchangepackage failed: NF)r^ r Z changepackager )r_ rD rc rV r! r! r" � set_package� s rd )r_ �cpmover r, c C s� ddg}|r|� d|g� |�|� d}t|t�� ddtjtjd��4}|jD ] }t| ||� d|v rLd }qLW d � n1 s~0 Y | � d � |j d kr�t| d |j � �� d }|S )zExecute restorepkgz$/usr/local/cpanel/scripts/restorepkgz --skipresr% T�utf-8�replace)ZlimrG �errors�stdoutrR zAccount Restore FailedFNrY r zrestorepkg exit code was ) �extend�appendr �os� cpu_count�PIPEZSTDOUTri r^ r] � returncode)r_ re r r, �cmd�success�procrT r! r! r" � restorepkg� s. � $ rs )rB re rD �logr# c C s | j p|}| j|kr*t| j� d�� dgS t�|�rJt|dtjd� dgS td|� |jddd���}t||| j | j �s�dgW d � S g }t ||| j�s�|� d | j� �� t ||| j �s�|� d | j � �� t||| j�s�|� d | j� �� W d � n1 �s0 Y |S ) zgRestore a user (restorepkg + set owner/ip/package) and return a list of task(s) that failed, if anyz-: You cannot set a reseller to own themselvesrs zalready existsrI z Logging to:�arf )�moderG Nz set owner to z set ip to zset package to )r r rQ rL rM r? rR rN rs r, ra rk rb r rd r )rB re rD rt r_ �failedr! r! r" � restore_user� s& 2rx )r r# c c s2 | � � D ]$}t�|j� }r|�d�|fV qdS )z)Iterate over backups found in a directory� N)r; r �matchr/ �group)r �entryrz r! r! r" � iter_backups s r} c C s` t � } i }| jjddd� | j�� r�tt| j��}|sHt�d| j� �� t d� |D ]\}}t ||dd� qT| j r~t � d� nt �d �s�t�d � |D ](\}}| j�|� d ��}t| |||�}q�|D ]\}}|||< q�nJt�| jj��d �}| jr�| jn|}| j�|� d ��}t| | j||�||< t|� d d� |�� D �}| j�rLt|� | j�r\t|� dS )z"Wrapper around cPanel's restorepkgi T)rv �exist_okzNo backups in z'The following backups will be restored:�: ��sep� zWould you like to proceed?r z .restore.logry c S s g | ]\}}|d gkr|�qS )rs r! )r0 �k�vr! r! r" � <listcomp>7 � zmain.<locals>.<listcomp>N)rC r �mkdirr r> �listr} r? r@ rQ r �time�sleeprL Z prompt_y_n�joinpathrx r rz r/ r{ r � print_results�itemsr r �print_host_mods) rB � user_failsZbackupsrD r rt rw Z orig_user�restoredr! r! r" �main s: r� )r� c C sP t d� | �� D ]:\}}|r<t |dddd� t |ddi� qt |ddd� qd S ) z*Print results from each ``restore_user()``z== Restore Results ==rw r )r� r[ r� z, rq r� N)rQ r� )r� rD Zfailsr! r! r" r� > s r� �r� c C s | sdS t �ddg| � dS )z Runs fixperms on restored usersNz/usr/bin/fixpermsr+ )� subprocess�callr� r! r! r" r I s r c C s$ t d� | D ]}t�t|g� q dS )z$Runs the command at ``HOSTFILEMODS``z4Host file mod entries for all restored cPanel users:N)rQ r� r� r: )r� rD r! r! r" r� Q s r� �__main__)T)6r rl r5 Z dataclassesr r rK �pathlibr r� r? r� �typingr r r Zcpapisr r Zcprocr Znetaddrr rL r �insertr8 r Zserverr r r. r: r rC r r9 r< r r^ ra rb rd rs r� rx �tupler} r� �dictr� r r� r r! r! r! r" �<module> sR V � "&