mirror of
https://mau.dev/maunium/synapse.git
synced 2024-10-01 01:36:05 -04:00
23740eaa3d
During the migration the automated script to update the copyright headers accidentally got rid of some of the existing copyright lines. Reinstate them.
150 lines
5.1 KiB
Python
150 lines
5.1 KiB
Python
#
|
|
# This file is licensed under the Affero General Public License (AGPL) version 3.
|
|
#
|
|
# Copyright 2020 The Matrix.org Foundation C.I.C.
|
|
# Copyright (c) 2012, 2013, 2014 Ilya Otyutskiy <ilya.otyutskiy@icloud.com>
|
|
# Copyright (C) 2023 New Vector, Ltd
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# See the GNU Affero General Public License for more details:
|
|
# <https://www.gnu.org/licenses/agpl-3.0.html>.
|
|
#
|
|
# Originally licensed under the Apache License, Version 2.0:
|
|
# <http://www.apache.org/licenses/LICENSE-2.0>.
|
|
#
|
|
# [This file includes modifications made by New Vector Limited]
|
|
#
|
|
#
|
|
|
|
import atexit
|
|
import fcntl
|
|
import logging
|
|
import os
|
|
import signal
|
|
import sys
|
|
from types import FrameType, TracebackType
|
|
from typing import NoReturn, Optional, Type
|
|
|
|
|
|
def daemonize_process(pid_file: str, logger: logging.Logger, chdir: str = "/") -> None:
|
|
"""daemonize the current process
|
|
|
|
This calls fork(), and has the main process exit. When it returns we will be
|
|
running in the child process.
|
|
"""
|
|
|
|
# If pidfile already exists, we should read pid from there; to overwrite it, if
|
|
# locking will fail, because locking attempt somehow purges the file contents.
|
|
if os.path.isfile(pid_file):
|
|
with open(pid_file) as pid_fh:
|
|
old_pid = pid_fh.read()
|
|
|
|
# Create a lockfile so that only one instance of this daemon is running at any time.
|
|
try:
|
|
lock_fh = open(pid_file, "w")
|
|
except OSError:
|
|
print("Unable to create the pidfile.")
|
|
sys.exit(1)
|
|
|
|
try:
|
|
# Try to get an exclusive lock on the file. This will fail if another process
|
|
# has the file locked.
|
|
fcntl.flock(lock_fh, fcntl.LOCK_EX | fcntl.LOCK_NB)
|
|
except OSError:
|
|
print("Unable to lock on the pidfile.")
|
|
# We need to overwrite the pidfile if we got here.
|
|
#
|
|
# XXX better to avoid overwriting it, surely. this looks racey as the pid file
|
|
# could be created between us trying to read it and us trying to lock it.
|
|
with open(pid_file, "w") as pid_fh:
|
|
pid_fh.write(old_pid)
|
|
sys.exit(1)
|
|
|
|
# Fork, creating a new process for the child.
|
|
process_id = os.fork()
|
|
|
|
if process_id != 0:
|
|
# parent process: exit.
|
|
|
|
# we use os._exit to avoid running the atexit handlers. In particular, that
|
|
# means we don't flush the logs. This is important because if we are using
|
|
# a MemoryHandler, we could have logs buffered which are now buffered in both
|
|
# the main and the child process, so if we let the main process flush the logs,
|
|
# we'll get two copies.
|
|
os._exit(0)
|
|
|
|
# This is the child process. Continue.
|
|
|
|
# Stop listening for signals that the parent process receives.
|
|
# This is done by getting a new process id.
|
|
# setpgrp() is an alternative to setsid().
|
|
# setsid puts the process in a new parent group and detaches its controlling
|
|
# terminal.
|
|
|
|
os.setsid()
|
|
|
|
# point stdin, stdout, stderr at /dev/null
|
|
devnull = "/dev/null"
|
|
if hasattr(os, "devnull"):
|
|
# Python has set os.devnull on this system, use it instead as it might be
|
|
# different than /dev/null.
|
|
devnull = os.devnull
|
|
|
|
devnull_fd = os.open(devnull, os.O_RDWR)
|
|
os.dup2(devnull_fd, 0)
|
|
os.dup2(devnull_fd, 1)
|
|
os.dup2(devnull_fd, 2)
|
|
os.close(devnull_fd)
|
|
|
|
# now that we have redirected stderr to /dev/null, any uncaught exceptions will
|
|
# get sent to /dev/null, so make sure we log them.
|
|
#
|
|
# (we don't normally expect reactor.run to raise any exceptions, but this will
|
|
# also catch any other uncaught exceptions before we get that far.)
|
|
|
|
def excepthook(
|
|
type_: Type[BaseException],
|
|
value: BaseException,
|
|
traceback: Optional[TracebackType],
|
|
) -> None:
|
|
logger.critical("Unhanded exception", exc_info=(type_, value, traceback))
|
|
|
|
sys.excepthook = excepthook
|
|
|
|
# Set umask to default to safe file permissions when running as a root daemon. 027
|
|
# is an octal number which we are typing as 0o27 for Python3 compatibility.
|
|
os.umask(0o27)
|
|
|
|
# Change to a known directory. If this isn't done, starting a daemon in a
|
|
# subdirectory that needs to be deleted results in "directory busy" errors.
|
|
os.chdir(chdir)
|
|
|
|
try:
|
|
lock_fh.write("%s" % (os.getpid()))
|
|
lock_fh.flush()
|
|
except OSError:
|
|
logger.error("Unable to write pid to the pidfile.")
|
|
print("Unable to write pid to the pidfile.")
|
|
sys.exit(1)
|
|
|
|
# write a log line on SIGTERM.
|
|
def sigterm(signum: int, frame: Optional[FrameType]) -> NoReturn:
|
|
logger.warning("Caught signal %s. Stopping daemon." % signum)
|
|
sys.exit(0)
|
|
|
|
signal.signal(signal.SIGTERM, sigterm)
|
|
|
|
# Cleanup pid file at exit.
|
|
def exit() -> None:
|
|
logger.warning("Stopping daemon.")
|
|
os.remove(pid_file)
|
|
sys.exit(0)
|
|
|
|
atexit.register(exit)
|
|
|
|
logger.warning("Starting daemon.")
|