File: //usr/local/lib/python3.9/site-packages/celery/__pycache__/platforms.cpython-39.pyc
a
X>h
d � @ s d Z ddlZddlZddlZddlZddlZddlZddlZ ddl
Z
ddl Z ddl m
Z
ddlmZmZ ddlmZ ddlmZ ddlmZ dd lmZmZmZ dd
l m Z zdd l m Z W n e y� dZ Y n0 e d �Z!e d
�Z"e d�Z#e d�Z$e d�Z%dZ&e'edd�Z(dZ)e'edd�Z*e'edd�Z+e'edd�Z,e�-� Z.e.dkZ/e.dkZ0dZ1ej2ej3B ej4B Z5ej6ej7B d > ej6d > B ej6B Z8d Z9d Z:d Z;d!Z<d"d#� e=e �D �Z>d$d%� e>D �Z?d&d'� Z@d(d)� ZAG d*d+� d+eB�ZCG d,d-� d-�ZDeDZEd.d/� ZFd0d1� ZGd2d3� ZHG d4d5� d5�ZIded7d8�ZJd9d:� ZKd;d<� ZLd=d>� ZMd?d@� ZNdAdB� ZOdCdD� ZPdEdF� ZQdfdGdH�ZRdIdJ� ZSeTe dK��rtdLdM� ZUndNdM� ZUG dOdP� dP�ZVeV� ZWeWjXZYeWjZZ[eWj\Z]eWj^Z_dQdR� Z`dSdT� ZadUdV� ZdgdWdX�Zbejc�ddY��r�dZd[� Zen
dhd\d[�Zed]d^� Zfe
d_d`� �Zgdadb� Zhdcdd� ZidS )iziPlatforms.
Utilities dealing with platform specifics: signals, daemonization,
users, groups, and so on.
� N)�contextmanager)�close_open_fds� get_fdmax)�
set_pdeathsig)� maybe_fileno)�safe_str� )�
SecurityError�SecurityWarning�reraise)�
try_import)�current_process� setproctitle�resource�pwd�grpzmultiprocessing.util) �EX_OK�
EX_FAILURE�EX_UNAVAILABLE�EX_USAGE�SYSTEM�IS_macOS�
IS_WINDOWS�SIGMAP�pyimplementation�
LockFailedr �Pidfile�create_pidlockr �
DaemonContext�detached� parse_uid� parse_gid� setgroups�
initgroups�setgid�setuid�maybe_drop_privileges�signals� signal_name�set_process_title�set_mp_process_title�get_errno_name� ignore_errno�
fd_by_path�isattyr r �E r �@ � EX_CANTCREAT�I �Darwin�Windows�/� � zLERROR: Pidfile ({0}) already exists.
Seems we're already running? (pid: {1})a> Running a worker with superuser privileges when the
worker accepts messages serialized with pickle is a very bad idea!
If you really want to continue then you have to set the C_FORCE_ROOT
environment variable (but please think about this before you do).
User information: uid={uid} euid={euid} gid={gid} egid={egid}
z�You're running the worker with superuser privileges: this is
absolutely not recommended!
Please specify a different user using the --uid option.
User information: uid={uid} euid={euid} gid={gid} egid={egid}
ziAn entry for the specified gid or egid was not found.
We're assuming this is a potential security issue.
c C s" h | ]}|� d �rd|vr|�qS )�SIG�_)�
startswith��.0�sig� r> �:/usr/local/lib/python3.9/site-packages/celery/platforms.py� <setcomp>[ s �r@ c C s i | ]}t t|�|�qS r> )�getattr�_signal)r< �namer> r> r? �
<dictcomp>_ � rD c C s" z
| � � W S ty Y n0 dS )z6Return true if the process has a controlling terminal.N)r. �AttributeError)�fhr> r> r? r. b s
r. c C s� t td�rt�� S tj�d�r(dtj S t td�r�d�dd� tjdd � D ��} tjd d� r�| d
d �d d� tjd d� D �� 7 } d
| S dS dS )z<Return string identifying the current Python implementation.�python_implementation�javazJython �pypy_version_info�.c s s | ]}t |�V qd S �N��str�r< �pr> r> r? � <genexpr>q rE z#pyimplementation.<locals>.<genexpr>Nr7 �-� c s s | ]}t |�V qd S rL rM rO r> r> r? rQ s rE zPyPy �CPython)�hasattr� _platformrH �sys�platformr: �joinrJ )�vr> r> r? r j s
&r c @ s e Zd ZdZdS )r z'Raised if a PID lock can't be acquired.N)�__name__�
__module__� __qualname__�__doc__r> r> r> r? r y s r c @ s\ e Zd ZdZdZdd� Zdd� ZeZdd� Zd d
� Z e Z
d d � Z d
d� Z dd� Z
dd� ZdS )r aF Pidfile.
This is the type returned by :func:`create_pidlock`.
See Also:
Best practice is to not use this directly but rather use
the :func:`create_pidlock` function instead:
more convenient and also removes stale pidfiles (when
the process holding the lock is no longer running).
Nc C s t j�|�| _d S rL )�os�path�abspath)�selfr` r> r> r? �__init__� s zPidfile.__init__c
C sR z | � � W n@ tyL } z(tttt|��t�� d � W Y d}~n
d}~0 0 | S )z
Acquire lock.� N)� write_pid�OSErrorr r rN rW �exc_info)rb �excr> r> r? �acquire� s
2zPidfile.acquirec C s t j�| j�S )z#Return true if the pid lock exists.)r_ r` �exists�rb r> r> r? � is_locked� s zPidfile.is_lockedc G s | � � dS )z
Release lock.N)�remove�rb �argsr> r> r? �release� s zPidfile.releasec
C s� t d��� t| j���}|�� }|�� |kr:td| j� ���z*t|�� �W W d � W d � S ty� td| j� d���Y n0 W d � n1 s�0 Y W d � n1 s�0 Y dS )z Read and return the current pid.�ENOENTzPartial or invalid pidfile Nzpidfile z contents invalid.)r, �openr` �readline�strip�
ValueError�int)rb rG �liner> r> r? �read_pid� s
�* �zPidfile.read_pidc C s> t tjtj�� t�| j� W d � n1 s00 Y dS )zRemove the lock.N)r, �errnorq �EACCESr_ �unlinkr` rk r> r> r? rm � s zPidfile.removec
C s z | � � }W n* ty6 tdtjd� | �� Y dS 0 |sH| �� dS |t�� kr`| �� dS zt�|d� W n� t y� } zH|j
t
j ks�|j
t
j kr�tdtjd� | �� W Y d}~dS W Y d}~n4d}~0 t
� y� tdtjd� | �� Y dS 0 dS )zhRemove the lock if the process isn't running.
I.e. process does not respond to signal.
z#Broken pidfile found - Removing it.��fileTr z#Stale pidfile exists - Removing it.NF)rx ru �printrW �stderrrm r_ �getpid�killrf ry ZESRCH�EPERM� SystemError)rb �pidrh r> r> r? �remove_if_stale� s0 "zPidfile.remove_if_stalec C s� t �� }|� d�}t �| jtt�}t �|d�}z@|�|� |�� zt � |� W n t
yb Y n0 W |� � n
|� � 0 t| j�}z |� � |kr�t
d��W |� � n
|� � 0 d S )N�
�wz7Inconsistency: Pidfile content doesn't match at re-read)r_ r� rr r` �
PIDFILE_FLAGS� PIDFILE_MODE�fdopen�write�flush�fsyncrF �close�readr )rb r� �contentZ
pidfile_fd�pidfileZrfhr> r> r? re � s&
�zPidfile.write_pid)r[ r\ r] r^ r` rc ri � __enter__rl rp �__exit__rx rm r� re r> r> r> r? r } s r c C s t | �}t�|j� |S )a Create and verify pidfile.
If the pidfile already exists the program exits with an error message,
however if the process it refers to isn't running anymore, the pidfile
is deleted and the program continues.
This function will automatically install an :mod:`atexit` handler
to release the lock at exit, you can skip this by calling
:func:`_create_pidlock` instead.
Returns:
Pidfile: used to manage the lock.
Example:
>>> pidlock = create_pidlock('/var/run/app.pid')
)�_create_pidlock�atexit�registerrp �r� Zpidlockr> r> r? r � s r c C sF t | �}|�� r:|�� s:tt�| |�� �tjd� t t
��|� � |S )Nr| ) r rl r� r~ � PIDLOCKED�formatrx rW r �
SystemExitr1 ri r� r> r> r? r�
s r� c
s� t � �| D ]b}zt�|tj�}W n ty6 Y q
Y n0 z&��t�|�dd� � W t�|� q
t�|� 0 q
�fdd�� � fdd�tt d��D �S )a[ Return a list of file descriptors.
This method returns list of file descriptors corresponding to
file paths passed in paths variable.
Arguments:
paths: List[str]: List of file paths.
Returns:
List[int]: List of file descriptors.
Example:
>>> keep = fd_by_path(['/dev/urandom', '/my/precious/'])
r r7 c s2 zt �| �dd� � v W S ty, Y dS 0 d S )Nr r7 F)r_ �fstatrf )�fd)�statsr> r? � fd_in_stats- s z fd_by_path.<locals>.fd_in_statsc s g | ]}� |�r|�qS r> r> )r< Z_fd)r� r> r? �
<listcomp>3 rE z fd_by_path.<locals>.<listcomp>i )
�setr_ rr �O_RDONLYrf �addr� r� �ranger )�pathsr` r� r> )r� r� r? r- s
r- c @ sF e Zd ZdZdZddd�Zdd� Zd d
� ZeZd d � Z e Z
d
d� Z dS )r z(Context manager daemonizing the process.FNTc K sZ t |t�r"t||�d�r dnd�}|p(t| _|| _|| _|| _|| _ t
j t
j t
j
f| _d S )N�0� �
)�
isinstancerN rv r: �DAEMON_WORKDIR�workdir�umask�fake� after_chdir�
after_forkersrW �stdin�stdoutr �stdfds)rb r� r� r� r� r� r� �kwargsr> r> r? rc ; s
zDaemonContext.__init__c C s( |d ur$t �t jt j�}t �||� d S rL )r_ rr �devnull�O_RDWR�dup2)rb r� �destr> r> r? �redirect_to_nullH s z DaemonContext.redirect_to_nullc C s� | j s�| js| �� t�| j� | jd ur6t�| j� | jrD| �� | js�t| j �t
dg� }t |� | j D ]}| � t
|�� ql| jr�td ur�t�� d| _ d S )Nz /dev/urandomT)�_is_openr� �_detachr_ �chdirr� r� r� �listr� r- r r� r r� �mputilZ_run_after_forkers)rb Zkeepr� r> r> r? rr M s
zDaemonContext.openc G s | j r d| _ d S )NF)r� rn r> r> r? r� g s zDaemonContext.closec C s: t �� dkr,t �� t �� dkr6t �d� n
t �d� | S �Nr )r_ �fork�setsid�_exitrk r> r> r? r� m s
zDaemonContext._detach)NNNFNT) r[ r\ r] r^ r� rc r� rr r� r� r� r� r> r> r> r? r 6 s �
r Fc sT t s td��|du r t�� n|}t�d� t||d� � �fdd�}t||||d�S )a� Detach the current process in the background (daemonize).
Arguments:
logfile (str): Optional log file.
The ability to write to this file
will be verified before the process is detached.
pidfile (str): Optional pid file.
The pidfile won't be created,
as this is the responsibility of the child. But the process will
exit if the pid lock exists and the pid written is still running.
uid (int, str): Optional user id or user name to change
effective privileges to.
gid (int, str): Optional group id or group name to change
effective privileges to.
umask (str, int): Optional umask that'll be effective in
the child process.
workdir (str): Optional new working directory.
fake (bool): Don't actually detach, intended for debugging purposes.
**opts (Any): Ignored.
Example:
>>> from celery.platforms import detached, create_pidlock
>>> with detached(
... logfile='/var/log/app.log',
... pidfile='/var/run/app.pid',
... uid='nobody'):
... # Now in detached child process with effective user set to nobody,
... # and we know that our logfile can be written to, and that
... # the pidfile isn't locked.
... pidlock = create_pidlock('/var/run/app.pid')
...
... # Run the program
... program.run(logfile='/var/log/app.log')
z&This platform does not support detach.N�SIGCLD��uid�gidc s& � ot � d��� �r"t���� d S )N�a)rr r� r� rp r> ��logfiler� r> r? �after_chdir_do� s z detached.<locals>.after_chdir_do)r� r� r� r� )r � RuntimeErrorr_ �getcwdr' �resetr& r ) r� r� r� r� r� r� r� �optsr� r> r� r? r x s $
�r c C sZ z
t | �W S tyT zt�| �jW Y S ttfyN td| � ���Y n0 Y n0 dS )z�Parse user id.
Arguments:
uid (str, int): Actual uid, or the username of a user.
Returns:
int: The actual uid.
zUser does not exist: N)rv ru r �getpwnam�pw_uidrF �KeyError�r� r> r> r? r � s
r c C sZ z
t | �W S tyT zt�| �jW Y S ttfyN td| � ���Y n0 Y n0 dS )z�Parse group id.
Arguments:
gid (str, int): Actual gid, or the name of a group.
Returns:
int: The actual gid of the group.
zGroup does not exist: N)rv ru r �getgrnam�gr_gidrF r� �r� r> r> r? r! � s
r! c
C s� | d d � } z t �| �W S tyL t| �dkr4� | d d� | d d �< Y q ty� } z:|jtjksrt| �dkrt� | d d� | d d �< W Y d }~q d }~0 0 q d S )Nr ���)r_ r" ru �lenrf ry �EINVAL)�groupsrh r> r> r? �_setgroups_hack� s r� c
s� d}zt �d�}W n ty$ Y n0 zt� d|� �W S ty� } z6|jtjkrV� t� fdd�t �� D ��rr� W Y d}~n
d}~0 0 dS )z+Set active groups from a list of group ids.N�SC_NGROUPS_MAXc 3 s | ]}|� vV qd S rL r> �r< �group�r� r> r? rQ � rE z setgroups.<locals>.<genexpr>) r_ �sysconf� Exceptionr� rf ry r� �any� getgroups)r� Z
max_groupsrh r> r� r? r" � s r" c sN t sdS t �| �d � ttd�r,t�� |�S � fdd�t�� D �}t|� dS )zyInit process group permissions.
Compat version of :func:`os.initgroups` that was first
added to Python 2.7.
Nr r# c s g | ]}� |j v r|j�qS r> )�gr_memr� )r< Zgr��usernamer> r? r� s
�z initgroups.<locals>.<listcomp>)r �getpwuidrU r_ r# r �getgrallr" )r� r� r� r> r� r? r# � s
r# c C s t �t| �� dS )z4Version of :func:`os.setgid` supporting group names.N)r_ r$ r! r� r> r> r? r$ s r$ c C s t �t| �� dS )z2Version of :func:`os.setuid` supporting usernames.N)r_ r% r r� r> r> r? r% s r% c C s� t jdkrdS t�� r&t�� s&td��| o0t| �} |o<t|�}| rNt| |� n |oXt |� | rvt�� svt�� svtd��|r�t�
� s�t� � s�td��dS )a Change process privileges to new user/group.
If UID and GID is specified, the real user/group is changed.
If only UID is specified, the real user is changed, and the group is
changed to the users primary group.
If only GID is specified, only the group is changed.
�win32Nzcontact supportz%Still root uid after drop privileges!z%Still root gid after drop privileges!) rW rX r_ �geteuid�getuidr r r! �_setuidr$ �getgid�getegidr� r> r> r? r& s
r& c
C sz |st rt �| �j}t|� t| |� t| � z td� W n2 tyl } z|jtjkrX� W Y d }~nd }~0 0 t d��d S )Nr z6non-root user able to restore privileges after setuid.)
r r� �pw_gidr$ r# r% rf ry r� r )r� r� rh r> r> r? r� . s
�r� � setitimerc C s t �t j| � d S rL )rB r� � ITIMER_REAL��secondsr> r> r? �
_arm_alarmK s r� c C s t �t�| �� d S rL )rB �alarm�math�ceilr� r> r> r? r� N s c @ sf e Zd ZdZejZejZdd� Z dd� Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zddd�ZdS )�SignalsaE Convenience interface to :mod:`signals`.
If the requested signal isn't supported on the current platform,
the operation will be ignored.
Example:
>>> from celery.platforms import signals
>>> from proj.handlers import my_handler
>>> signals['INT'] = my_handler
>>> signals['INT']
my_handler
>>> signals.supported('INT')
True
>>> signals.signum('INT')
2
>>> signals.ignore('USR1')
>>> signals['USR1'] == signals.ignored
True
>>> signals.reset('USR1')
>>> signals['USR1'] == signals.default
True
>>> from proj.handlers import exit_handler, hup_handler
>>> signals.update(INT=exit_handler,
... TERM=exit_handler,
... HUP=hup_handler)
c C s t |�S rL )r� )rb r� r> r> r? � arm_alarmx s zSignals.arm_alarmc C s
t �d�S r� )rB r� rk r> r> r? � reset_alarm{ s zSignals.reset_alarmc C s, z| � |� W n ty" Y dS 0 dS dS )z@Return true value if signal by ``name`` exists on this platform.FTN)�signumrF �rb rC r> r> r? � supported~ s
zSignals.supportedc C sF t |tj�r|S t |t�r"|�� s*td��|�d�s<d| }tt|�S )zGet signal number by name.z%signal name must be uppercase string.r8 ) r� �numbers�IntegralrN �isupper� TypeErrorr: rA rB r� r> r> r? r� � s
�
zSignals.signumc s � � � fdd�|D �� dS )z�Reset signals to the default signal handler.
Does nothing if the platform has no support for signals,
or the specified signal in particular.
c 3 s | ]}|� j fV qd S rL )�defaultr; rk r> r? rQ � rE z Signals.reset.<locals>.<genexpr>N��update)rb Z signal_namesr> rk r? r� � s z
Signals.resetc s � � � fdd�|D �� dS )z�Ignore signal using :const:`SIG_IGN`.
Does nothing if the platform has no support for signals,
or the specified signal in particular.
c 3 s | ]}|� j fV qd S rL )�ignoredr; rk r> r? rQ � rE z!Signals.ignore.<locals>.<genexpr>Nr )rb �namesr> rk r? �ignore� s zSignals.ignorec C s t �| �|��S rL )rB � getsignalr� r� r> r> r? � __getitem__� s zSignals.__getitem__c C s2 zt �| �|�|� W n ttfy, Y n0 dS )z�Install signal handler.
Does nothing if the current platform has no support for signals,
or the specified signal in particular.
N)rB �signalr� rF ru )rb rC �handlerr> r> r? � __setitem__� s zSignals.__setitem__Nc K s. t |pi fi |���� D ]\}}|| |< qdS )z#Set signal handlers from a mapping.N)�dict�items)rb Z_d_ZsigmaprC r r> r> r? r � s zSignals.update)N)r[ r\ r] r^ rB �SIG_IGNr �SIG_DFLr� r� r� r� r� r� r r r r r> r> r> r? r� R s " r� c C s t | dd� S )z)Return name of signal from signal number.r7 N)r )r� r> r> r? r( � s r( c C s6 d| d v rdnd}t | �|kr2d�| |d � �S dS )NZmanager rd r � rS )r� rY )�argvZ arg_startr> r> r? �strargv� s r c C s4 t �d�r0ztt �d�� W n ty. Y n0 dS )z@Sends signal ``name`` to process when parent process terminates.�SIGKILLN)r' r� �_set_pdeathsigr� rf )rC r> r> r? r � s
r c C s8 d| � d�}|r |� d|� �n|}t r4t �t|�� |S )zxSet the :command:`ps` name for the currently running process.
Only works if :pypi:`setproctitle` is installed.
�[�]r )�
_setproctitler r )�progname�infoZ proctitler> r> r? r) � s
r) ZNOSETPSc O s dS )zDisabled feature.Nr> )r� �kr> r> r? r* � s r* c C s8 |r| � d|� �} t r t � jnd}t| � d|� �|d�S )z|Set the :command:`ps` name from the current process name.
Only works if :pypi:`setproctitle` is installed.
z: � MainProcess�:)r )r
rC r) )r r �hostnamerC r> r> r? r* � s c C s t | t�rtt| �S | S )z(Get errno for string (e.g., ``ENOENT``).)r� rN rA ry )�nr> r> r? r+ � s
r+ c
o sj |� d�ptf}dd� | D �} z
dV W n< |yd } z$t|d�sD� |j| vrP� W Y d}~n
d}~0 0 dS )a Context manager to ignore specific POSIX error codes.
Takes a list of error codes to ignore: this can be either
the name of the code, or the code integer itself::
>>> with ignore_errno('ENOENT'):
... with open('foo', 'r') as fh:
... return fh.read()
>>> with ignore_errno(errno.ENOENT, errno.EPERM):
... pass
Arguments:
types (Tuple[Exception]): A tuple of exceptions to ignore
(when the errno matches). Defaults to :exc:`Exception`.
�typesc S s g | ] }t |��qS r> )r+ )r< ry r> r> r? r�
rE z ignore_errno.<locals>.<listcomp>Nry )�getr� rU ry )Zerrnosr� r rh r> r> r? r, � s
r, c sL t d u std u rd S d| v p"d| v }ttd�r6t�� nd}ttd�rLt�� nd}ttd�rbt�� nd}ttd�rxt�� nd}ttd�r�td d
� d D ��s�t d ��zt �
|�}t �
|�}W n2 t y� t �
tt�� t|||||� Y d S 0 |d
}|d
} || f� d}
|d
k�p|d
k} | �s8t� fdd
�|
D ���rHt|||||� d S )N�picklez"application/group-python-serializer� i�� r� r� r� �fchownc s s | ]}t t|�V qd S rL )rU r_ )r< �attrr> r> r? rQ s �z#check_privileges.<locals>.<genexpr>)r� r� r� r� z$suspicious platform, contact supportr )Zsudo�wheelc 3 s | ]}|� v V qd S rL r> r� �Z gids_in_user> r? rQ ; s �)r r rU r_ r� r� r� r� �allr �getgrgidr� �warnings�warnr
�
ASSUMING_ROOT� _warn_or_raise_security_errorr� ) Zaccept_content�pickle_or_serializer� r� �euid�egidZ gid_entryZ
egid_entryZ gid_grp_nameZ
egid_grp_nameZgroups_with_security_riskZis_rootr> r# r? �check_privileges sD �
�
��
�r- c C sL t j�dd�}|r,|s,ttj|||| d���t�tt j|||| d��� d S )NZ C_FORCE_ROOTF)r� r+ r� r, )
r_ �environr r �ROOT_DISALLOWEDr� r& r' r
�ROOT_DISCOURAGED)r, r+ r� r� r* Z c_force_rootr> r> r? r) A s �
�r) )NNNNr NF)NN)N)NN)jr^ r� ry r� r� r_ rX rV r rB rW r& �
contextlibr Zbilliard.compatr r Z
billiard.utilr r Zkombu.utils.compatr Zkombu.utils.encodingr �
exceptionsr r
r �localr Zbilliard.processr
� ImportErrorr r r r r� �__all__rA r r r r r1 �systemr r r r� �O_CREAT�O_EXCL�O_WRONLYr� �R_OK�W_OKr� r� r/ r0 r( �dirZSIGNAMESr r. r r� r r ZPIDFiler r� r- r r r r! r� r" r# r$ r% r& r� rU r� r� r' r� Z
get_signalr Zinstall_signal_handlerr� Z reset_signalr Z
ignore_signalr( r r) r. r r* r+ r, r- r) r> r> r> r? �<module> s�
�t #B �
8
d
-