Onionr/src/onionrcommands/daemonlaunch/__init__.py

189 lines
5.8 KiB
Python
Raw Normal View History

"""Onionr - Private P2P Communication.
launch the api servers and communicator
"""
import os
import queue
import sys
import platform
import signal
2019-03-08 01:08:06 +00:00
from threading import Thread
2020-02-03 01:00:49 +00:00
2020-02-02 06:55:39 +00:00
from stem.connection import IncorrectPassword
import stem
2019-08-02 23:00:04 +00:00
import toomanyobjs
2020-02-03 01:00:49 +00:00
import filenuke
from deadsimplekv import DeadSimpleKV
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 apiservers
import logger
from onionrplugins import onionrevents as events
2022-01-11 07:20:15 +00:00
from onionrutils import localcommand
2020-02-03 01:16:30 +00:00
from utils import identifyhome
2019-07-19 04:59:44 +00:00
import filepaths
import onionrvalues
from onionrutils import cleanup
2019-07-20 00:01:16 +00:00
from onionrcrypto import getourkeypair
2019-09-26 07:43:03 +00:00
import runtests
2020-01-02 02:07:34 +00:00
from httpapi import daemoneventsapi
from .. import version
from utils.bettersleep import better_sleep
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
2020-07-30 01:23:48 +00:00
from setupkvvars import setup_kv
"""
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/>.
"""
2019-07-27 02:42:55 +00:00
2019-07-31 05:10:28 +00:00
def _proper_shutdown():
2019-07-19 04:59:44 +00:00
localcommand.local_command('shutdown')
sys.exit(1)
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)
logger.info('Using public key: %s' %
(logger.colors.underline +
getourkeypair.get_keypair()[0][:52]))
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):
2020-12-01 22:24:10 +00:00
pid = str(os.getpid())
main_pid = localcommand.local_command('/getpid')
#logger.info(main_pid, terminal=True)
if main_pid and main_pid == pid:
logger.info(
f"Received sigterm, shutting down gracefully. PID: {pid}", terminal=True)
2022-02-10 23:38:22 +00:00
localcommand.local_command('/shutdown')
2020-12-01 22:24:10 +00:00
else:
logger.info(
f"Recieved sigterm in child process or fork, exiting. PID: {pid}")
sys.exit(0)
signal.signal(signal.SIGTERM, _handle_sig_term)
2019-09-26 07:43:03 +00:00
# Create shared objects
2019-08-04 04:52:57 +00:00
shared_state = toomanyobjs.TooMany()
2019-03-08 01:08:06 +00:00
# Add DeadSimpleKV for quasi-global variables (ephemeral key-value)
shared_state.get(DeadSimpleKV)
2020-07-30 01:23:48 +00:00
# Initialize the quasi-global variables
setup_kv(shared_state.get(DeadSimpleKV))
2020-01-02 02:07:34 +00:00
shared_state.get(daemoneventsapi.DaemonEventsBP)
# Init run time tester
# (ensures Onionr is running right, for testing purposes)
# Run time tests are not normally run
2019-09-26 07:43:03 +00:00
shared_state.get(runtests.OnionrRunTestManager)
2022-01-31 05:59:34 +00:00
# initialize clientAPI but dont start it yet
shared_state.get(apiservers.ClientAPI)
shared_state.share_object() # share the parent object to the threads
2019-03-08 01:08:06 +00:00
show_logo()
security_level = config.get('general.security_level', 1)
_show_info_messages()
logger.info(
"Onionr daemon is running under " + str(os.getpid()), terminal=True)
events.event('init', threaded=False)
2019-07-31 05:10:28 +00:00
events.event('daemon_start')
2022-01-11 07:20:15 +00:00
shared_state.get(apiservers.ClientAPI).gossip_peer_set = set()
shared_state.get(apiservers.ClientAPI).gossip_block_queue = queue.Queue()
gossip.start_gossip_threads(
shared_state.get(apiservers.ClientAPI).gossip_peer_set,
shared_state.get(apiservers.ClientAPI).gossip_block_queue)
2022-02-11 06:56:19 +00:00
2022-01-31 05:59:34 +00:00
try:
shared_state.get(apiservers.ClientAPI).start()
except KeyboardInterrupt:
pass
2019-07-31 05:10:28 +00:00
cleanup.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.info(
f"Detected stale run file, deleting {filepaths.lock_file}", terminal=True)
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