File: //usr/lib/python3/dist-packages/twisted/scripts/__pycache__/_twistd_unix.cpython-38.pyc
U
��W[�? � @ s d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z m Z d dl
mZmZmZ d dlmZmZmZmZ d dlmZmZ d dlmZ d dlm Z m Z d d l m Z e d
kr�e d ��d d
� Z!G dd� dej"�Z"dd� Z#G dd� dej$�Z%dd� Z&G dd� dej'�Z(dS )� )�absolute_import�division�print_functionN)�log�logfile�usage)�
intToBytes�
_bytesRepr�_PY3)� switchUID�
uidFromString�
gidFromString�untilConcludes)�app�service)�IReactorDaemonize)� copyright�logger)� platformTypeZwin32z%_twistd_unix doesn't work on Windows.c C s
t | d�S )N� )�int)�value� r �>/usr/lib/python3/dist-packages/twisted/scripts/_twistd_unix.py�_umask s r c
@ s� e Zd ZdZdddgdddgddd gd
d d ggZd
dddgdd ddgddddgddddegddddegdddd eggZe j
e � d �e j d d �e j
d d �e jd!d �e jd"d �d#�d$�Zd%d&� Zd'd(� ZdS ))�
ServerOptionszUsage: twistd [options]�nodaemon�nz0don't daemonize, don't use default umask of 0077� originalnameNz$Don't try to change the process name�syslogzLog to syslog, not to file�euid� a Set only effective user-id rather than real user-id. (This option has no effect unless the server is running as root, in which case it means not to shed all privileges after binding ports, retaining the option to regain privileges in cases such as spawning processes. Use with caution.)�prefix�twistedz$use the given prefix when syslogging�pidfilez
twistd.pidzName of the pidfile�chrootz-Chroot to a supplied directory before running�uid�uzThe uid to run as.�gid�gzbThe gid to run as. If not specified, the default gid associated with the specified --uid is used.�umaskz(The (octal) file creation mask to apply.z*.pidzchroot directory)Zdescrz
gid to run asz
uid to run asz
syslog prefix)r$ r% r( r& r" )Z
optActionsc C s2 t d�tj�| jd� t tj| jd� t�� dS )z5
Print version information and exit.
z twistd (the Twisted daemon) {})�fileN)�print�formatr �version�stdout�sys�exit��selfr r r � opt_versionI s
�zServerOptions.opt_versionc C s, t j�| � | d r(tj�| d �| d< d S )Nr$ )r r � postOptions�os�path�abspathr2 r r r r5 S s zServerOptions.postOptions)�__name__�
__module__� __qualname__ZsynopsisZoptFlagsr r
r Z
optParametersr Z CompletionsZ
CompleteFilesZ CompleteDirsZCompleteGroupsZCompleteUsernamesZ CompleterZcompDatar4 r5 r r r r r sJ ��
��� � ��
��
r c
C s� | sd S t j�| �r�z$t| ��}t|�� �}W 5 Q R X W n$ tk
r\ t�d� | �� Y nX zt �
|d� W nb t k
r� } zD|j t j
kr�tjd� | �dd� t �| � nt�d� || |�� W 5 d }~X Y nX t�d� |�� d S )Nz%Pidfile {} contains non-numeric valuer zRemoving stale pidfile {}T)ZisErrorz0Can't check status of PID {} from pidfile {}: {}a Another twistd server is running, PID {}
This could either be a previously started instance of your application or a
different application entirely. To start a new one, either run it in some other
directory, or use the --pidfile and --logfile parameters to avoid clashes.
)r6 r7 �exists�openr �read�
ValueErrorr0 r1 r- �kill�OSError�errnoZESRCHr �msg�remove)r$ �f�pidZwhyr r r �checkPIDY s0
���rG c @ s e Zd ZdZdd� Zdd� ZdS )�
UnixAppLoggera�
A logger able to log to syslog, to files, and to stdout.
@ivar _syslog: A flag indicating whether to use syslog instead of file
logging.
@type _syslog: C{bool}
@ivar _syslogPrefix: If C{sysLog} is C{True}, the string prefix to use for
syslog messages.
@type _syslogPrefix: C{str}
@ivar _nodaemon: A flag indicating the process will not be daemonizing.
@type _nodaemon: C{bool}
c C s< t j�| |� |�dd�| _|�dd�| _|�dd�| _d S )Nr Fr" r! r )r � AppLogger�__init__�get�_syslog�
_syslogPrefix� _nodaemon)r3 Zoptionsr r r rJ � s zUnixAppLogger.__init__c s� | j r ddlm} |�| j�jS | jdkrB| js:t� d� tj
� nv| jrV| jsVtj
� nb| jsbd| _t j �
| j�� z ddl}W n tk
r� Y n(X |�|j�s�� fdd�}|�|j|� t�� �S ) a�
Create and return a suitable log observer for the given configuration.
The observer will go to syslog using the prefix C{_syslogPrefix} if
C{_syslog} is true. Otherwise, it will go to the file named
C{_logfilename} or, if C{_nodaemon} is true and C{_logfilename} is
C{"-"}, to stdout.
@return: An object suitable to be passed to C{log.addObserver}.
r )r �-z&Daemons cannot log to stdout, exiting!z
twistd.logNc s ddl m} |�� j� d S )Nr ��reactor)�twisted.internetrQ ZcallFromThread�rotate)�signal�framerQ �ZlogFiler r � rotateLog� s z0UnixAppLogger._getLogObserver.<locals>.rotateLog)rL �twisted.pythonr ZSyslogObserverrM �emitZ _logfilenamerN r0 r1 r/ r ZLogFileZ fromFullPathrT � ImportError� getsignal�SIGUSR1r ZtextFileLogObserver)r3 r rT rW r rV r �_getLogObserver� s(
z UnixAppLogger._getLogObserverN)r9 r: r; �__doc__rJ r] r r r r rH w s rH c C sX | rT| t jd krTtj�t j�}t�d| � t�|| t jd dgt jdd � � d S )Nr zChanging process name to z--originalname� ) r0 �argvr6 r7 �realpath�
executabler rC �execv)�name�exer r r �launchWithName� s rf c @ s\ e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� ZdS )�UnixApplicationRunnerzx
An ApplicationRunner which does Unix-specific things, like fork,
shed privileges, and maintain a PID file.
c C s< t | jd � | jd p | jd | jd< tj| _tj| _dS )z4
Do pre-application-creation setup.
r$ r �debugN)rG �configr0 r/ � oldstdout�stderr� oldstderrr2 r r r �preApplication� s
�z$UnixApplicationRunner.preApplicationc C s< t �|j|�d }d�|�� �}tr0|�dd�}|dd� S )aC
Format the C{exception} in preparation for writing to the
status pipe. This does the right thing on Python 2 if the
exception's message is Unicode, and in all cases limits the
length of the message afte* encoding to 100 bytes.
This means the returned message may be truncated in the middle
of a unicode escape.
@type exception: L{Exception}
@param exception: The exception to format.
@return: The formatted message, suitable for writing to the
status pipe.
@rtype: L{bytes}
���z1 {}�ascii�backslashreplaceN�d )� traceback�format_exception_only� __class__r- �stripr
�encode)r3 Z exceptionZ
exceptionLineZformattedMessager r r �_formatChildException� s ���
z+UnixApplicationRunner._formatChildExceptionc
C s� z| � | j� W nn tk
r~ } zP| j�dd�}|dk r\| �|�}ttj||� ttj |� | �
| jd � � W 5 d}~X Y n2X | j�dd�}|dk r�ttj|d� ttj |� | � d| j | j
� | �
| jd � dS )z�
To be called after the application is created: start the application
and run the reactor. After the reactor stops, clean up PID files and
such.
�
statusPipeNr$ � 0)�startApplication� application� Exceptionri rK rw r r6 �write�close� removePIDZ startReactorrj rl )r3 �exrx �messager r r �postApplication� s
z%UnixApplicationRunner.postApplicationc
C s� |sdS zt �|� W nh tk
rh } z4|jtjks@|jtjkrLt�d� n t�|d� W 5 d}~X Y n t�dd� Y nX dS )z�
Remove the specified PID file, if possible. Errors are logged, not
raised.
@type pidfile: C{str}
@param pidfile: The path to the PID tracking file.
Nz)Warning: No permission to delete pid filezFailed to unlink PID file:) r6 �unlinkrA rB ZEACCESZEPERMr rC �err)r3 r$ �er r r r s z UnixApplicationRunner.removePIDc C s� | }|dk r$t �|� |dkr$d}t �|� |r>|dkr>d}|dk rPt �|� |rpddlm} | �|�| jd< |r�t|d��}|� t
t � � �� W 5 Q R X dS ) a�
Set the filesystem root, the working directory, and daemonize.
@type chroot: C{str} or L{None}
@param chroot: If not None, a path to use as the filesystem root (using
L{os.chroot}).
@type rundir: C{str}
@param rundir: The path to set as the working directory.
@type nodaemon: C{bool}
@param nodaemon: A flag which, if set, indicates that daemonization
should not be done.
@type umask: C{int} or L{None}
@param umask: The value to which to change the process umask.
@type pidfile: C{str} or L{None}
@param pidfile: If not L{None}, the path to a file into which to put
the PID of this process.
N�.�/�? r rP rx �wb) r6 r% �chdirr* rR rQ � daemonizeri r= r} r �getpid) r3 r% �rundirr r* r$ ZdaemonrQ rE r r r �setupEnvironment% s
z&UnixApplicationRunner.setupEnvironmentc C s� t �|�r|�� t�� \}}t�� rD| �|�}t�|� t�|� t� � t�� r^t�d� t�
dtj �}t d�D ]F}zt�
||� W qt tk
r� } z|jtjkr�� W 5 d}~X Y qtX qtt�|� t �|�r�|�� |S )a7
Daemonizes the application on Unix. This is done by the usual double
forking approach.
@see: U{http://code.activestate.com/recipes/278731/}
@see: W. Richard Stevens,
"Advanced Programming in the Unix Environment",
1992, Addison-Wesley, ISBN 0-201-56317-7
@param reactor: The reactor in use. If it provides
L{IReactorDaemonize}, its daemonization-related callbacks will be
invoked.
@return: A writable pipe to be used to report errors.
@rtype: C{int}
r z /dev/null� N)r Z
providedByZbeforeDaemonizer6 �pipe�fork�
_waitForStartr~ �_exit�setsidr= �O_RDWR�range�dup2rA rB ZEBADFZafterDaemonize)r3 rQ �r�w�codeZnull�ir� r r r r� N s*
z UnixApplicationRunner.daemonizec C sF t tj|d�}t|dd� �}|dkrBd�|�}t tjj|� dS dS )a
Wait for the daemonization success.
@param readPipe: file descriptor to read start information from.
@type readPipe: C{int}
@return: code to be passed to C{os._exit}: 0 for success, 1 for error.
@rtype: C{int}
rq � Nry zHAn error has occurred: {}
Please look at log file for more information.
r_ r )r r6 r> r r- r0 �
__stderr__r} )r3 ZreadPipe�dataZdataReprrC r r r r� { s
�z#UnixApplicationRunner._waitForStartc
C s� |dk s|dk r�|rdpd}d� ||||�}zt|||� W n> tk
rz } z t�d� ||�� t�d� W 5 d}~X Y nX t�d� |�� dS )a�
Change the UID and GID or the EUID and EGID of this process.
@type euid: C{bool}
@param euid: A flag which, if set, indicates that only the I{effective}
UID and GID should be set.
@type uid: C{int} or L{None}
@param uid: If not L{None}, the UID to which to switch.
@type gid: C{int} or L{None}
@param gid: If not L{None}, the GID to which to switch.
Nr� r! z{}uid/{}gid {}/{}z0failed to set {}: {} (are you root?) -- exiting.r_ zset {})r- r rA r rC r0 r1 )r3 r r& r( ZextraZdescr� r r r �shedPrivileges� s � z$UnixApplicationRunner.shedPrivilegesc C s� t �|�}| jd s t|j� | �| jd | jd | jd | jd | jd � t �|��� | jd | jd }}|d kr�|j}|d kr�|j }|d k r�|d kr�t
� |�j }| �
| jd
||� t�|| jd � d S ) z�
Configure global process state based on the given application and run
the application.
@param application: An object which can be adapted to
L{service.IProcess} and L{service.IService}.
r r% r� r r* r$ r& r( Nr Zno_save)r ZIProcessri rf Z processNamer� ZIServiceZprivilegedStartServicer& r( �pwd�getpwuidZpw_gidr� r rz )r3 r{ Zprocessr& r( r r r rz � s(
� z&UnixApplicationRunner.startApplicationN)r9 r: r; r^ rH Z
loggerFactoryrm rw r� r r� r� r� r� rz r r r r rg � s ')-rg ))Z
__future__r r r rB r6 r� r0 rr rX r r r Ztwisted.python.compatr r r
Ztwisted.python.utilr r r
r Ztwisted.applicationr r Ztwisted.internet.interfacesr r# r r Ztwisted.python.runtimer rZ r r rG rI rH rf ZApplicationRunnerrg r r r r �<module> s( 9 ?