��V*{ �
@ s d Z d d l m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z y e Z e
Z
Wn e k
r� e Z e Z
Yn XGd d � d e � Z Gd d � d e e � Z Gd d � d e e � Z Gd
d � d � Z d d
� Z d d � Z d d � Z d d � Z d d � Z d d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z! d% d& � Z" d' Z# d( d) � Z$ e% � d* d+ � Z& d, d- � Z' d. d/ � Z( d0 d1 � Z) d2 d3 � Z* d d4 d5 � Z+ d S)6z Daemon process behaviour.
� )�absolute_import�unicode_literalsNc s4 e Z d Z d Z � f d d � Z d d � Z � S)� DaemonErrorz3 Base exception class for errors from this module. c s' | j � t t | � j | | � d S)N)�_chain_from_context�superr �__init__)�self�args�kwargs)� __class__� �
/daemon.pyr , s
zDaemonError.__init__c C s t | d d �d S)N�as_causeT)�0_chain_exception_from_existing_exception_context)r r r r
r 1 s z DaemonError._chain_from_context)�__name__�
__module__� __qualname__�__doc__r r r r )r r
r ) s r c @ s e Z d Z d Z d S)�DaemonOSEnvironmentErrorzC Exception raised when daemon OS environment setup receives error. N)r r r r r r r r
r 5 s r c @ s e Z d Z d Z d S)�DaemonProcessDetachErrorz- Exception raised when process detach fails. N)r r r r r r r r
r 9 s r c @ s� e Z d Z d Z e Z d d d d d d d d d d d d d d d d � Z e d d
� � Z d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d S)�
DaemonContextu. Context for turning the current program into a daemon process.
A `DaemonContext` instance represents the behaviour settings and
process context for the program when it becomes a daemon. The
behaviour and environment is customised by setting options on the
instance, before calling the `open` method.
Each option can be passed as a keyword argument to the `DaemonContext`
constructor, or subsequently altered by assigning to an attribute on
the instance at any time prior to calling `open`. That is, for
options named `wibble` and `wubble`, the following invocation::
foo = daemon.DaemonContext(wibble=bar, wubble=baz)
foo.open()
is equivalent to::
foo = daemon.DaemonContext()
foo.wibble = bar
foo.wubble = baz
foo.open()
The following options are defined.
`files_preserve`
:Default: ``None``
List of files that should *not* be closed when starting the
daemon. If ``None``, all open file descriptors will be closed.
Elements of the list are file descriptors (as returned by a file
object's `fileno()` method) or Python `file` objects. Each
specifies a file that is not to be closed during daemon start.
`chroot_directory`
:Default: ``None``
Full path to a directory to set as the effective root directory of
the process. If ``None``, specifies that the root directory is not
to be changed.
`working_directory`
:Default: ``'/'``
Full path of the working directory to which the process should
change on daemon start.
Since a filesystem cannot be unmounted if a process has its
current working directory on that filesystem, this should either
be left at default or set to a directory that is a sensible “home
directory” for the daemon while it is running.
`umask`
:Default: ``0``
File access creation mask (“umask”) to set for the process on
daemon start.
A daemon should not rely on the parent process's umask value,
which is beyond its control and may prevent creating a file with
the required access mode. So when the daemon context opens, the
umask is set to an explicit known value.
If the conventional value of 0 is too open, consider setting a
value such as 0o022, 0o027, 0o077, or another specific value.
Otherwise, ensure the daemon creates every file with an
explicit access mode for the purpose.
`pidfile`
:Default: ``None``
Context manager for a PID lock file. When the daemon context opens
and closes, it enters and exits the `pidfile` context manager.
`detach_process`
:Default: ``None``
If ``True``, detach the process context when opening the daemon
context; if ``False``, do not detach.
If unspecified (``None``) during initialisation of the instance,
this will be set to ``True`` by default, and ``False`` only if
detaching the process is determined to be redundant; for example,
in the case when the process was started by `init`, by `initd`, or
by `inetd`.
`signal_map`
:Default: system-dependent
Mapping from operating system signals to callback actions.
The mapping is used when the daemon context opens, and determines
the action for each signal's signal handler:
* A value of ``None`` will ignore the signal (by setting the
signal action to ``signal.SIG_IGN``).
* A string value will be used as the name of an attribute on the
``DaemonContext`` instance. The attribute's value will be used
as the action for the signal handler.
* Any other value will be used as the action for the
signal handler. See the ``signal.signal`` documentation
for details of the signal handler interface.
The default value depends on which signals are defined on the
running system. Each item from the list below whose signal is
actually defined in the ``signal`` module will appear in the
default map:
* ``signal.SIGTTIN``: ``None``
* ``signal.SIGTTOU``: ``None``
* ``signal.SIGTSTP``: ``None``
* ``signal.SIGTERM``: ``'terminate'``
Depending on how the program will interact with its child
processes, it may need to specify a signal map that
includes the ``signal.SIGCHLD`` signal (received when a
child process exits). See the specific operating system's
documentation for more detail on how to determine what
circumstances dictate the need for signal handlers.
`uid`
:Default: ``os.getuid()``
`gid`
:Default: ``os.getgid()``
The user ID (“UID”) value and group ID (“GID”) value to switch
the process to on daemon start.
The default values, the real UID and GID of the process, will
relinquish any effective privilege elevation inherited by the
process.
`initgroups`
:Default: ``False``
If true, set the daemon process's supplementary groups as
determined by the specified `uid`.
This will require that the current process UID has
permission to change the process's owning GID.
`prevent_core`
:Default: ``True``
If true, prevents the generation of core files, in order to avoid
leaking sensitive information from daemons run as `root`.
`stdin`
:Default: ``None``
`stdout`
:Default: ``None``
`stderr`
:Default: ``None``
Each of `stdin`, `stdout`, and `stderr` is a file-like object
which will be used as the new file for the standard I/O stream
`sys.stdin`, `sys.stdout`, and `sys.stderr` respectively. The file
should therefore be open, with a minimum of mode 'r' in the case
of `stdin`, and mimimum of mode 'w+' in the case of `stdout` and
`stderr`.
If the object has a `fileno()` method that returns a file
descriptor, the corresponding file will be excluded from being
closed during daemon start (that is, it will be treated as though
it were listed in `files_preserve`).
If ``None``, the corresponding system stream is re-bound to the
file named by `os.devnull`.
N�/r FTc C s� | | _ | | _ | | _ | | _ | | _ |
| _ | | _ | | _ |
| _ | d k ri t j
� } | | _ | d k r� t j � } | | _
| | _ | d k r� t � } | | _ | d k r� t � } | | _ d | _ d S)z Set up a new instance. NF)�chroot_directory�working_directory�umask� prevent_core�files_preserve�pidfile�stdin�stdout�stderr�os�getuid�uid�getgid�gid�
initgroups�"is_detach_process_context_required�detach_process�make_default_signal_map�
signal_map�_is_open)r r r r r# r% r& r r( r r r r r r* r r r
r � s. zDaemonContext.__init__c C s | j S)z- ``True`` if the instance is currently open. )r+ )r r r r
�is_open! s zDaemonContext.is_openc C s | j r
d S| j d k r) t | j � | j r9 t � t | j � t | j � t | j
| j | j � | j
r| t � | j � } t | � | j � } t d | � t t j | j � t t j | j � t t j | j � | j d k r | j j � d | _ t | j � d S)u0 Become a daemon process.
:return: ``None``.
Open the daemon context, turning the current program into a daemon
process. This performs the following steps:
* If this instance's `is_open` property is true, return
immediately. This makes it safe to call `open` multiple times on
an instance.
* If the `prevent_core` attribute is true, set the resource limits
for the process to prevent any core dump from the process.
* If the `chroot_directory` attribute is not ``None``, set the
effective root directory of the process to that directory (via
`os.chroot`).
This allows running the daemon process inside a “chroot gaol”
as a means of limiting the system's exposure to rogue behaviour
by the process. Note that the specified directory needs to
already be set up for this purpose.
* Set the process owner (UID and GID) to the `uid` and `gid`
attribute values.
If the `initgroups` attribute is true, also set the process's
supplementary groups to all the user's groups (i.e. those
groups whose membership includes the username corresponding
to `uid`).
* Close all open file descriptors. This excludes those listed in
the `files_preserve` attribute, and those that correspond to the
`stdin`, `stdout`, or `stderr` attributes.
* Change current working directory to the path specified by the
`working_directory` attribute.
* Reset the file access creation mask to the value specified by
the `umask` attribute.
* If the `detach_process` option is true, detach the current
process into its own process group, and disassociate from any
controlling terminal.
* Set signal handlers as specified by the `signal_map` attribute.
* If any of the attributes `stdin`, `stdout`, `stderr` are not
``None``, bind the system streams `sys.stdin`, `sys.stdout`,
and/or `sys.stderr` to the files represented by the
corresponding attributes. Where the attribute has a file
descriptor, the descriptor is duplicated (instead of re-binding
the name).
* If the `pidfile` attribute is not ``None``, enter its context
manager.
* Mark this instance as open (for the purpose of future `open` and
`close` calls).
* Register the `close` method to be called during Python's exit
processing.
When the function returns, the running program is a daemon
process.
N�excludeT) r, r �change_root_directoryr �prevent_core_dump�change_file_creation_maskr �change_working_directoryr �change_process_ownerr# r% r& r( �detach_process_context�_make_signal_handler_map�set_signal_handlers� _get_exclude_file_descriptors�close_all_open_files�redirect_stream�sysr r r r � __enter__r+ �register_atexit_function�close)r �signal_handler_mapZ exclude_fdsr r r
�open& s, D
zDaemonContext.openc C s | j � | S)z Context manager entry point. )r> )r r r r
r: � s
zDaemonContext.__enter__c C s? | j s
d S| j d k r2 | j j d d d � d | _ d S)a Exit the daemon process context.
:return: ``None``.
Close the daemon context. This performs the following steps:
* If this instance's `is_open` property is false, return
immediately. This makes it safe to call `close` multiple times
on an instance.
* If the `pidfile` attribute is not ``None``, exit its context
manager.
* Mark this instance as closed (for the purpose of future `open`
and `close` calls).
NF)r, r �__exit__r+ )r r r r
r< � s
zDaemonContext.closec C s | j � d S)z Context manager exit point. N)r< )r �exc_type� exc_value� tracebackr r r
r? � s zDaemonContext.__exit__c C s" t d j d | � � } | � d S)a� Signal handler for end-process signals.
:param signal_number: The OS signal number received.
:param stack_frame: The frame object at the point the
signal was received.
:return: ``None``.
Signal handler for the ``signal.SIGTERM`` signal. Performs the
following step:
* Raise a ``SystemExit`` exception explaining the signal.
z'Terminating on signal {signal_number!r}�
signal_numberN)�
SystemExit�format)r rC Z stack_frame� exceptionr r r
� terminate� s zDaemonContext.terminatec C s� | j } | d k r g } | j d d � | j | j | j g D� � t � } xR | D]J } | d k rl qW t | � } | d k r� | j | � qW | j | � qW W| S)ar Get the set of file descriptors to exclude closing.
:return: A set containing the file descriptors for the
files to be preserved.
The file descriptors to be preserved are those from the
items in `files_preserve`, and also each of `stdin`,
`stdout`, and `stderr`. For each item:
* If the item is ``None``, it is omitted from the return
set.
* If the item's ``fileno()`` method returns a value, that
value is in the return set.
* Otherwise, the item is in the return set verbatim.
Nc s s$ | ] } t | d � r | Vq d S)�filenoN)�hasattr)�.0�itemr r r
� <genexpr>� s z>DaemonContext._get_exclude_file_descriptors.<locals>.<genexpr>)r �extendr r r �set�_get_file_descriptor�add)r r Zexclude_descriptorsrK �file_descriptorr r r
r6 � s &
z+DaemonContext._get_exclude_file_descriptorsc C sI | d k r t j } n- t | t � r? | } t | | � } n | } | S)a� Make the signal handler for a specified target object.
:param target: A specification of the target for the
handler; see below.
:return: The value for use by `signal.signal()`.
If `target` is ``None``, return ``signal.SIG_IGN``. If `target`
is a text string, return the attribute of this instance named
by that string. Otherwise, return `target` itself.
N)�signal�SIG_IGN�
isinstance�
basestring�getattr)r �target�result�namer r r
�_make_signal_handler� s z"DaemonContext._make_signal_handlerc s, t � f d d � � j j � D� � } | S)a* Make the map from signals to handlers for this instance.
:return: The constructed signal map for this instance.
Construct a map from signal numbers to handlers for this
context instance, suitable for passing to
`set_signal_handlers`.
c 3 s* | ] \ } } | � j | � f Vq d S)N)rZ )rJ rC rW )r r r
rL
s z9DaemonContext._make_signal_handler_map.<locals>.<genexpr>)�dictr* �items)r r= r )r r
r4 � s
z&DaemonContext._make_signal_handler_map)r r r r �typeZ
__metaclass__r �propertyr, r> r: r< r? rG r6 rZ r4 r r r r
r = s2 � e & r c C s> d } t | d � r: y | j � } Wn t k
r9 Yn X| S)a� Get the file descriptor, if the object has one.
:param obj: The object expected to be a file-like object.
:return: The file descriptor iff the file supports it; otherwise
``None``.
The object may be a non-file object. It may also be a
file-like object with no support for a file descriptor. In
either case, return ``None``.
NrH )rI rH �
ValueError)�objrQ r r r
rO s
rO c C s[ y t j | � WnC t k
rV } z# t d j d | � � } | � WYd d } ~ Xn Xd S)z� Change the working directory of this process.
:param directory: The target directory path.
:return: ``None``.
z*Unable to change working directory ({exc})�excN)r! �chdir� Exceptionr rE )� directoryra �errorr r r
r1 & s r1 c C sh y t j | � t j | � WnC t k
rc } z# t d j d | � � } | � WYd d } ~ Xn Xd S)a6 Change the root directory of this process.
:param directory: The target directory path.
:return: ``None``.
Set the current working directory, then the process root directory,
to the specified `directory`. Requires appropriate OS privileges
for this process.
z'Unable to change root directory ({exc})ra N)r! rb �chrootrc r rE )rd ra re r r r
r. 5 s
r. c C s[ y t j | � WnC t k
rV } z# t d j d | � � } | � WYd d } ~ Xn Xd S)z� Change the file creation mask for this process.
:param mask: The numeric file creation mask to set.
:return: ``None``.
z+Unable to change file creation mask ({exc})ra N)r! r rc r rE )�maskra re r r r
r0 I s r0 c C s t j | � } | j } | S)z) Get the username for the specified UID. )�pwd�getpwuid�pw_name)r# Z passwd_entry�usernamer r r
�get_username_for_uidX s rl Fc C s� y t | � } Wn t k
r* d } Yn Xy7 | rG t j | | � n
t j | � t j | � WnC t k
r� } z# t d j d | � � } | � WYd d } ~ Xn Xd S)a Change the owning UID, GID, and groups of this process.
:param uid: The target UID for the daemon process.
:param gid: The target GID for the daemon process.
:param initgroups: If true, initialise the supplementary
groups of the process.
:return: ``None``.
Sets the owning GID and UID of the process (in that order, to
avoid permission errors) to the specified `gid` and `uid`
values.
If `initgroups` is true, the supplementary groups of the
process are also initialised, with those corresponding to the
username for the target UID.
All these operations require appropriate OS privileges. If
permission is denied, a ``DaemonOSEnvironmentError`` is
raised.
Fz&Unable to change process owner ({exc})ra N) rl �KeyErrorr! r& �setgid�setuidrc r rE )r# r% r& rk ra re r r r
r2 ` s
r2 c C s| t j } y t j | � } WnC t k
ra } z# t d j d | � � } | � WYd d } ~ Xn Xd } t j | | � d S)z� Prevent this process from generating a core dump.
:return: ``None``.
Set the soft and hard limits for core dump size to zero. On Unix,
this entirely prevents the process from creating core dump.
z:System does not support RLIMIT_CORE resource limit ({exc})ra Nr )r r )�resourceZ RLIMIT_CORE� getrlimitr_ r rE Z setrlimit)Z
core_resourceZcore_limit_prevra re Z
core_limitr r r
r/ � s r/ c C s4 d d � } | d d � t j � | d d � d S)u~ Detach the process context from parent and session.
:return: ``None``.
Detach from the parent process and session group, allowing the
parent to exit while this process continues running.
Reference: “Advanced Programming in the Unix Environment”,
section 13.3, by W. Richard Stevens, published 1993 by
Addison-Wesley.
c S sy y) t j � } | d k r( t j d � WnI t k
rt } z) t d j d | d | � � } | � WYd d } ~ Xn Xd S)a Fork a child process, then exit the parent process.
:param error_message: Message for the exception in case of a
detach failure.
:return: ``None``.
:raise DaemonProcessDetachError: If the fork fails.
r z){message}: [{exc.errno:d}] {exc.strerror}�messagera N)r! �fork�_exit�OSErrorr rE )�
error_message�pidra re r r r
�fork_then_exit_parent� s z5detach_process_context.<locals>.fork_then_exit_parentrv zFailed first forkzFailed second forkN)r! �setsid)rx r r r
r3 � s
r3 c C s( d } d } t j � | k r$ d } | S)z� Determine whether the current process is started by `init`.
:return: ``True`` iff the parent process is `init`; otherwise
``False``.
The `init` process is the one with process ID of 1.
F� T)r! �getppid)rX Zinit_pidr r r
�is_process_started_by_init� s
r| c C s� d } t j | t j t j � } y | j t j t j � } WnM t j k
r� } z* | j d } | t j
k rt n d } WYd d } ~ Xn Xd } | S)a1 Determine whether the file descriptor is a socket.
:param fd: The file descriptor to interrogate.
:return: ``True`` iff the file descriptor is a socket; otherwise
``False``.
Query the socket type of `fd`. If there is no error, the file is a
socket.
Fr TN) �socket�fromfd�AF_INET�SOCK_RAW�
getsockopt�
SOL_SOCKET�SO_TYPEre r �errnoZENOTSOCK)�fdrX Z file_socketZ socket_typera Z exc_errnor r r
� is_socket� s
r� c C s+ d } t j j � } t | � r' d } | S)a� Determine whether the current process is started by the superserver.
:return: ``True`` if this process was started by the internet
superserver; otherwise ``False``.
The internet superserver creates a network socket, and
attaches it to the standard streams of the child process. If
that is the case for this process, return ``True``, otherwise
``False``.
FT)r9 � __stdin__rH r� )rX Zstdin_fdr r r
�!is_process_started_by_superserver� s
r� c C s" d } t � s t � r d } | S)a� Determine whether detaching the process context is required.
:return: ``True`` iff the process is already detached; otherwise
``False``.
The process environment is interrogated for the following:
* Process was started by `init`; or
* Process was started by `inetd`.
If any of the above are true, the process is deemed to be already
detached.
TF)r| r� )rX r r r
r' s r' c C sv y t j | � Wn^ t k
rq } z> | j t j k r; n$ t d j d | d | � � } | � WYd d } ~ Xn Xd S)z� Close a file descriptor if already open.
:param fd: The file descriptor to close.
:return: ``None``.
Close the file descriptor `fd`, suppressing an error in the
case the file was not open.
z.Failed to close file descriptor {fd:d} ({exc})r� ra N)r! r< �EnvironmentErrorr� �EBADFr rE )r� ra re r r r
� close_file_descriptor_if_open' s
r� i c C s5 t j t j � } | d } | t j k r1 t } | S)uz Get the maximum number of open file descriptors for this process.
:return: The number (integer) to use as the maximum number of open
files for this process.
The maximum is the process hard resource limit of maximum number of
open file descriptors. If the limit is “infinity”, a default value
of ``MAXFD`` is returned.
rz )rp rq Z
RLIMIT_NOFILEZ
RLIM_INFINITY�MAXFD)ZlimitsrX r r r
� get_maximum_file_descriptors@ s
r� c C s@ t � } x0 t t | � � D] } | | k r t | � q Wd S)a< Close all open file descriptors.
:param exclude: Collection of file descriptors to skip when closing
files.
:return: ``None``.
Closes every file descriptor (if open) of this process. If
specified, `exclude` is a set of file descriptors to *not*
close.
N)r� �reversed�ranger� )r- Zmaxfdr� r r r
r7 R s r7 c C sM | d k r' t j t j t j � } n | j � } t j | | j � � d S)a} Redirect a system stream to a specified file.
:param standard_stream: A file object representing a standard I/O
stream.
:param target_stream: The target file object for the redirected
stream, or ``None`` to specify the null device.
:return: ``None``.
`system_stream` is a standard system stream such as
``sys.stdout``. `target_stream` is an open file object that
should replace the corresponding system stream object.
If `target_stream` is ``None``, defaults to opening the
operating system's null device and using its file descriptor.
N)r! r> �devnull�O_RDWRrH �dup2)Z
system_streamZ
target_streamZ target_fdr r r
r8 d s r8 c C sA d d d d d d d d i } t d d � | j � D� � } | S) z� Make the default signal map for this system.
:return: A mapping from signal number to handler object.
The signals available differ by system. The map will not contain
any signals not defined on the running system.
�SIGTSTPN�SIGTTIN�SIGTTOU�SIGTERMrG c s s9 | ]/ \ } } t t | � r t t | � | f Vq d S)N)rI rR rV )rJ rY rW r r r
rL � s z*make_default_signal_map.<locals>.<genexpr>)r[ r\ )Zname_mapr* r r r
r) | s
r) c C s1 x* | j � D] \ } } t j | | � q
Wd S)a Set the signal handlers as specified.
:param signal_handler_map: A map from signal number to handler
object.
:return: ``None``.
See the `signal` module for details on signal numbers and signal
handlers.
N)r\ rR )r= rC �handlerr r r
r5 � s r5 c C s t j | � d S)z� Register a function for processing at program exit.
:param func: A callable function expecting no arguments.
:return: ``None``.
The function `func` is registered for a call with no arguments
at program exit.
N)�atexit�register)�funcr r r
r; � s
r; c C s= t j � \ } } } | r' | | _ n | | _ | | _ d S)a, Decorate the specified exception with the existing exception context.
:param exc: The exception instance to decorate.
:param as_cause: If true, the existing context is declared to be
the cause of the exception.
:return: ``None``.
:PEP:`344` describes syntax and attributes (`__traceback__`,
`__context__`, `__cause__`) for use in exception chaining.
Python 2 does not have that syntax, so this function decorates
the exception with values from the current exception context.
N)r9 �exc_info� __cause__� __context__�
__traceback__)ra r Zexisting_exc_typeZ existing_excZexisting_tracebackr r r
r � s
r ),r Z
__future__r r r! r9 rh rp r� rR r} r� rU �unicode� NameError�strrc r ru r r r rO r1 r. r0 rl r2 r/ r3 r| r� r� r' r� r� r� rN r7 r8 r) r5 r; r r r r r
�<module> sR
� � ( & !