Onionr/src/onionrcommands/daemonlaunch/__init__.py

164 lines
4.7 KiB
Python
Raw Normal View History

"""Onionr - Private P2P Communication.
launch the api servers and communicator
"""
import os
from time import sleep
import sys
import platform
import signal
from threading import Thread
2020-02-03 01:00:49 +00:00
import filenuke
2020-12-22 14:57:15 +00:00
import psutil
2019-08-02 23:00:04 +00:00
import config
2022-01-11 07:20:15 +00:00
import logger
from onionrplugins import onionrevents as events
2022-01-11 07:20:15 +00:00
2020-02-03 01:16:30 +00:00
from utils import identifyhome
2019-07-19 04:59:44 +00:00
import filepaths
import onionrvalues
2022-07-31 05:32:43 +00:00
from onionrthreads import add_onionr_thread
from blockdb.blockcleaner import clean_block_database
from .. import version
from .killdaemon import kill_daemon # noqa
from .showlogo import show_logo
2022-02-11 06:56:19 +00:00
import gossip
2022-01-11 07:20:15 +00:00
"""
2022-01-11 07:20:15 +00:00
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
2022-09-26 20:06:05 +00:00
def _safe_remove(path):
try:
os.remove(path)
except FileNotFoundError:
pass
def delete_run_files():
"""Delete run files, do not error if not found.
Test: test_cleanup.py
"""
_safe_remove(filepaths.lock_file)
_safe_remove(filepaths.gossip_server_socket_file)
_safe_remove(filepaths.pid_file)
2019-07-27 02:42:55 +00:00
def _show_info_messages():
version.version(verbosity=5, function=logger.info)
logger.debug('Python version %s' % platform.python_version())
if onionrvalues.DEVELOPMENT_MODE:
logger.warn('Development mode enabled', timestamp=False, terminal=True)
2019-07-31 05:10:28 +00:00
def daemon():
"""Start Onionr's primary threads for communicator, API server, node, and LAN."""
def _handle_sig_term(signum, frame):
sys.exit(0)
2022-08-16 14:54:54 +00:00
with open(filepaths.pid_file, 'w') as f:
f.write(str(os.getpid()))
2020-07-30 01:23:48 +00:00
signal.signal(signal.SIGTERM, _handle_sig_term)
2019-03-08 01:08:06 +00:00
show_logo()
security_level = config.get('general.security_level', 1)
_show_info_messages()
logger.info(
f"Onionr daemon is running under pid {os.getpid()}", terminal=True)
events.event('init', threaded=False)
2022-08-16 14:54:54 +00:00
events.event('afterinit', threaded=False)
2019-07-31 05:10:28 +00:00
events.event('daemon_start')
2022-07-31 05:32:43 +00:00
add_onionr_thread(
clean_block_database, 60, 'clean_block_database', initial_sleep=0)
Thread(
2022-08-16 14:54:54 +00:00
target=gossip.start_gossip_threads,
daemon=True,
name='start_gossip_threads').start()
2022-01-31 05:59:34 +00:00
try:
while True:
2022-09-17 04:24:44 +00:00
# Mainly for things like repls
events.event('primary_loop', threaded=False)
sleep(60)
2022-01-31 05:59:34 +00:00
except KeyboardInterrupt:
pass
2022-09-17 04:24:44 +00:00
2022-09-26 20:06:05 +00:00
delete_run_files()
if security_level >= 2:
2020-02-03 01:00:49 +00:00
filenuke.nuke.clean_tree(identifyhome.identify_home())
2019-03-08 01:08:06 +00:00
def start(override: bool = False):
"""If no lock file, make one and start onionr.
Error exit if there is and its not overridden
"""
if os.path.exists(filepaths.lock_file) and not override:
if os.path.exists(filepaths.restarting_indicator):
try:
os.remove(filepaths.restarting_indicator)
except FileNotFoundError:
pass
else:
return
2020-12-22 14:57:15 +00:00
with open(filepaths.lock_file, 'r') as lock_file:
try:
proc = psutil.Process(int(lock_file.read())).name()
except psutil.NoSuchProcess:
proc = ""
if not proc.startswith("python"):
logger.warn(
f"Detected stale run file, deleting {filepaths.lock_file}",
terminal=True)
2020-12-22 14:57:15 +00:00
try:
os.remove(filepaths.lock_file)
except FileNotFoundError:
pass
start(override=True)
return
logger.fatal('Cannot start. Daemon is already running,'
+ ' or it did not exit cleanly.\n'
+ ' (if you are sure that there is not a daemon running,'
2020-02-10 14:01:49 +00:00
+ f' delete {filepaths.lock_file} & try again).',
terminal=True)
else:
2019-07-31 05:10:28 +00:00
if not onionrvalues.DEVELOPMENT_MODE:
lock_file = open(filepaths.lock_file, 'w')
2020-12-22 14:57:15 +00:00
lock_file.write(str(os.getpid()))
lock_file.close()
# Start Onionr daemon
2019-07-31 05:10:28 +00:00
daemon()
2019-10-08 22:26:44 +00:00
try:
os.remove(filepaths.lock_file)
except FileNotFoundError:
pass
2019-09-21 05:06:49 +00:00
start.onionr_help = "Start Onionr node " # type: ignore
start.onionr_help += "(public and clients API servers)" # type: ignore