make daemon launch obey offline mode and pep8 refactor it

This commit is contained in:
Kevin Froman 2019-12-18 04:02:08 -06:00
parent a36885c806
commit 8f1df8c371

View File

@ -25,33 +25,56 @@ from utils import hastor, logoheader
from . import version from . import version
import serializeddata import serializeddata
import runtests import runtests
"""
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/>.
"""
def _proper_shutdown(): def _proper_shutdown():
localcommand.local_command('shutdown') localcommand.local_command('shutdown')
sys.exit(1) sys.exit(1)
def daemon(): def daemon():
''' """
Starts the Onionr communication daemon Starts the Onionr communication daemon
''' """
offline_mode = config.get('general.offline_mode', False)
if not hastor.has_tor(): if not hastor.has_tor():
logger.error("Tor is not present in system path or Onionr directory", terminal=True) offline_mode = True
cleanup.delete_run_files() logger.error("Tor is not present in system path or Onionr directory",
sys.exit(1) terminal=True)
# remove runcheck if it exists # remove runcheck if it exists
if os.path.isfile(filepaths.run_check_file): if os.path.isfile(filepaths.run_check_file):
logger.debug('Runcheck file found on daemon start, deleting in advance.') logger.debug('Runcheck file found on daemon start, deleting.')
os.remove(filepaths.run_check_file) os.remove(filepaths.run_check_file)
# Create shared objects # Create shared objects
shared_state = toomanyobjs.TooMany() shared_state = toomanyobjs.TooMany()
Thread(target=shared_state.get(apiservers.ClientAPI).start, daemon=True, name='client HTTP API').start() Thread(target=shared_state.get(apiservers.ClientAPI).start,
Thread(target=shared_state.get(apiservers.PublicAPI).start, daemon=True, name='public HTTP API').start() daemon=True, name='client HTTP API').start()
if not offline_mode:
Thread(target=shared_state.get(apiservers.PublicAPI).start,
daemon=True, name='public HTTP API').start()
# Init run time tester (ensures Onionr is running right, for testing purposes) # Init run time tester
# (ensures Onionr is running right, for testing purposes)
shared_state.get(runtests.OnionrRunTestManager) shared_state.get(runtests.OnionrRunTestManager)
shared_state.get(serializeddata.SerializedData) shared_state.get(serializeddata.SerializedData)
@ -70,59 +93,73 @@ def daemon():
# print nice header thing :) # print nice header thing :)
if config.get('general.display_header', True): if config.get('general.display_header', True):
logoheader.header() logoheader.header()
version.version(verbosity = 5, function = logger.info) version.version(verbosity=5, function=logger.info)
logger.debug('Python version %s' % platform.python_version()) logger.debug('Python version %s' % platform.python_version())
if onionrvalues.DEVELOPMENT_MODE: if onionrvalues.DEVELOPMENT_MODE:
logger.warn('Development mode enabled', timestamp = False, terminal=True) logger.warn('Development mode enabled', timestamp=False, terminal=True)
net = NetController(config.get('client.public.port', 59497), apiServerIP=apiHost) net = NetController(config.get('client.public.port', 59497),
apiServerIP=apiHost)
shared_state.add(net) shared_state.add(net)
if not offline_mode:
logger.info('Tor is starting...', terminal=True) logger.info('Tor is starting...', terminal=True)
if not net.startTor(): if not net.startTor():
localcommand.local_command('shutdown') localcommand.local_command('shutdown')
cleanup.delete_run_files() cleanup.delete_run_files()
sys.exit(1) sys.exit(1)
if len(net.myID) > 0 and config.get('general.security_level', 1) == 0: if len(net.myID) > 0 and config.get('general.security_level', 1) == 0:
logger.debug('Started .onion service: %s' % (logger.colors.underline + net.myID)) logger.debug('Started .onion service: %s' %
(logger.colors.underline + net.myID))
else: else:
logger.debug('.onion service disabled') logger.debug('.onion service disabled')
logger.info('Using public key: %s' % (logger.colors.underline + getourkeypair.get_keypair()[0][:52]))
logger.info('Using public key: %s' %
(logger.colors.underline +
getourkeypair.get_keypair()[0][:52]))
try: try:
time.sleep(1) time.sleep(1)
except KeyboardInterrupt: except KeyboardInterrupt:
pass pass
events.event('init', threaded = False) events.event('init', threaded=False)
events.event('daemon_start') events.event('daemon_start')
communicator.startCommunicator(shared_state) communicator.startCommunicator(shared_state)
localcommand.local_command('shutdown') localcommand.local_command('shutdown')
if not offline_mode:
net.killTor() net.killTor()
try: try:
time.sleep(5) # Time to allow threads to finish, if not any "daemon" threads will be slaughtered http://docs.python.org/library/threading.html#threading.Thread.daemon # Time to allow threads to finish,
# if not any "daemon" threads will be slaughtered
# http://docs.python.org/library/threading.html#threading.Thread.daemon
time.sleep(5)
except KeyboardInterrupt: except KeyboardInterrupt:
pass pass
cleanup.delete_run_files() cleanup.delete_run_files()
def _ignore_sigint(sig, frame): def _ignore_sigint(sig, frame):
'''This space intentionally left blank''' """This space intentionally left blank"""
return return
def kill_daemon(): def kill_daemon():
''' """
Shutdown the Onionr daemon (communicator) Shutdown the Onionr daemon (communicator)
''' """
logger.warn('Stopping the running daemon...', timestamp = False, terminal=True) logger.warn('Stopping the running daemon...', timestamp=False,
terminal=True)
try: try:
# On platforms where we can, fork out to prevent locking # On platforms where we can, fork out to prevent locking
try: try:
pid = os.fork() pid = os.fork()
if pid != 0: return if pid != 0: return
except (AttributeError, OSError) as e: pass except (AttributeError, OSError): pass
events.event('daemon_stop') events.event('daemon_stop')
net = NetController(config.get('client.port', 59496)) net = NetController(config.get('client.port', 59496))
@ -133,15 +170,22 @@ def kill_daemon():
net.killTor() net.killTor()
except Exception as e: except Exception as e:
logger.error('Failed to shutdown daemon: ' + str(e), error = e, timestamp = False, terminal=True) logger.error('Failed to shutdown daemon: ' + str(e),
error=e, timestamp=False, terminal=True)
return return
kill_daemon.onionr_help = "Gracefully stops the Onionr API servers" kill_daemon.onionr_help = "Gracefully stops the Onionr API servers"
def start(input: bool = False, override: bool = False): def start(input: bool = False, override: bool = False):
"""If no lock file, make one and start onionr, error if there is and its not overridden""" """If no lock file, make one and start onionr,
error if there is and its not overridden"""
if os.path.exists(filepaths.lock_file) and not override: if os.path.exists(filepaths.lock_file) and not override:
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, delete onionr.lock & try again).', terminal=True) 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,'
+ ' delete onionr.lock & try again).', terminal=True)
else: else:
if not onionrvalues.DEVELOPMENT_MODE: if not onionrvalues.DEVELOPMENT_MODE:
lockFile = open(filepaths.lock_file, 'w') lockFile = open(filepaths.lock_file, 'w')
@ -153,4 +197,5 @@ def start(input: bool = False, override: bool = False):
except FileNotFoundError: except FileNotFoundError:
pass pass
start.onionr_help = "Start Onionr node (public and clients API servers)" start.onionr_help = "Start Onionr node (public and clients API servers)"