Onionr/src/onionrcommands/daemonlaunch/__init__.py

160 lines
4.5 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
from logger import log as logging
from logger import enable_file_logging
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
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=logging.info)
logging.debug('Python version %s' % platform.python_version())
if onionrvalues.DEVELOPMENT_MODE:
logging.warn('Development mode enabled')
2019-07-31 05:10:28 +00:00
def daemon():
"""Start Onionr's primary threads for communicator, API server, node, and LAN."""
if config.get('log.file.output', False):
enable_file_logging()
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()
logging.info(
f"Onionr daemon is running under pid {os.getpid()}")
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:
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"):
logging.warn(
f"Detected stale run file, deleting {filepaths.lock_file}")
2020-12-22 14:57:15 +00:00
try:
os.remove(filepaths.lock_file)
except FileNotFoundError:
pass
start(override=True)
return
logging.error('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).',
)
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