Onionr/src/onionrcommands/daemonlaunch/__init__.py

228 lines
7.1 KiB
Python
Raw Normal View History

"""Onionr - Private P2P Communication.
launch the api servers and communicator
"""
import os
import sys
import platform
import sqlite3
2019-03-08 01:08:06 +00:00
from threading import Thread
from gevent import time
from gevent import spawn
2020-02-02 06:55:39 +00:00
from stem.connection import IncorrectPassword
2019-08-02 23:00:04 +00:00
import toomanyobjs
import config
2020-01-27 08:20:09 +00:00
import onionrstatistics
from onionrstatistics import serializeddata
import apiservers
import logger
import communicator
from onionrplugins import onionrevents as events
2019-03-08 01:08:06 +00:00
from netcontroller import NetController
from netcontroller import get_open_port
from onionrutils import localcommand
2019-07-19 04:59:44 +00:00
import filepaths
2019-07-31 05:10:28 +00:00
from etc import onionrvalues, cleanup
2019-07-20 00:01:16 +00:00
from onionrcrypto import getourkeypair
2019-07-30 05:19:22 +00:00
from utils import hastor, logoheader
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 .getapihost import get_api_host_until_available
from utils.bettersleep import better_sleep
from netcontroller.torcontrol.onionservicecreator import create_onion_service
"""
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)
2019-07-31 05:10:28 +00:00
def daemon():
"""Start the Onionr communication daemon."""
offline_mode = config.get('general.offline_mode', False)
2019-07-27 02:42:55 +00:00
if not hastor.has_tor():
offline_mode = True
logger.error("Tor is not present in system path or Onionr directory",
terminal=True)
2019-03-08 01:08:06 +00:00
# remove runcheck if it exists
2019-07-19 04:59:44 +00:00
if os.path.isfile(filepaths.run_check_file):
logger.debug('Runcheck file found on daemon start, deleting.')
2019-07-19 04:59:44 +00:00
os.remove(filepaths.run_check_file)
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
2020-01-02 02:07:34 +00:00
shared_state.get(daemoneventsapi.DaemonEventsBP)
Thread(target=shared_state.get(apiservers.ClientAPI).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()
2019-09-26 07:43:03 +00:00
# Init run time tester
# (ensures Onionr is running right, for testing purposes)
2019-09-26 07:43:03 +00:00
shared_state.get(runtests.OnionrRunTestManager)
2019-08-05 04:08:56 +00:00
shared_state.get(serializeddata.SerializedData)
2020-01-02 02:07:34 +00:00
shared_state.share_object() # share the parent object to the threads
2019-03-08 01:08:06 +00:00
apiHost = ''
if not offline_mode:
apiHost = get_api_host_until_available()
2019-03-08 01:08:06 +00:00
logger.raw('', terminal=True)
# print nice header thing :)
2019-07-31 05:10:28 +00:00
if config.get('general.display_header', True):
2019-07-30 05:19:22 +00:00
logoheader.header()
version.version(verbosity=5, function=logger.info)
logger.debug('Python version %s' % platform.python_version())
2019-07-31 05:10:28 +00:00
if onionrvalues.DEVELOPMENT_MODE:
logger.warn('Development mode enabled', timestamp=False, terminal=True)
2019-08-04 04:52:57 +00:00
net = NetController(config.get('client.public.port', 59497),
apiServerIP=apiHost)
2019-08-04 04:52:57 +00:00
shared_state.add(net)
2020-01-27 08:20:09 +00:00
shared_state.get(onionrstatistics.tor.TorStats)
security_level = config.get('general.security_level', 1)
use_existing_tor = config.get('tor.use_existing_tor', False)
if not offline_mode:
if use_existing_tor:
net.socksPort = config.get('tor.existing_socks_port')
2020-02-02 06:55:39 +00:00
try:
net.myID = create_onion_service(
port=net.apiServerIP + ':' + str(net.hsPort))[0]
except IncorrectPassword:
logger.error('Invalid Tor control password', terminal=True)
localcommand.local_command('shutdown')
cleanup.delete_run_files()
sys.exit(1)
2020-02-01 04:14:26 +00:00
if not net.myID.endswith('.onion'):
net.myID += '.onion'
with open(filepaths.tor_hs_address_file, 'w') as tor_file:
tor_file.write(net.myID)
else:
logger.info('Tor is starting...', terminal=True)
if not net.startTor():
localcommand.local_command('shutdown')
cleanup.delete_run_files()
sys.exit(1)
if len(net.myID) > 0 and security_level == 0:
logger.debug('Started .onion service: %s' %
(logger.colors.underline + net.myID))
else:
logger.debug('.onion service disabled')
logger.info('Using public key: %s' %
(logger.colors.underline +
getourkeypair.get_keypair()[0][:52]))
better_sleep(1)
events.event('init', threaded=False)
2019-07-31 05:10:28 +00:00
events.event('daemon_start')
2019-08-04 04:52:57 +00:00
communicator.startCommunicator(shared_state)
2019-03-08 01:08:06 +00:00
if not offline_mode and not use_existing_tor:
net.killTor()
2020-02-01 04:14:26 +00:00
else:
try:
os.remove(filepaths.tor_hs_address_file)
except FileNotFoundError:
pass
better_sleep(5)
2019-07-31 05:10:28 +00:00
cleanup.delete_run_files()
2019-03-08 01:08:06 +00:00
def _ignore_sigint(sig, frame): # pylint: disable=W0612,W0613
"""Space intentionally left blank."""
return
2019-07-31 05:10:28 +00:00
def kill_daemon():
"""Shutdown the Onionr daemon (communicator)."""
logger.warn('Stopping the running daemon...', timestamp=False,
terminal=True)
# On platforms where we can, fork out to prevent locking
2019-03-08 01:08:06 +00:00
try:
pid = os.fork()
if pid != 0:
return
except (AttributeError, OSError):
pass
events.event('daemon_stop')
net = NetController(config.get('client.port', 59496))
try:
spawn(
localcommand.local_command,
'/shutdownclean'
).get(timeout=5)
except sqlite3.OperationalError:
pass
2019-03-08 01:08:06 +00:00
net.killTor()
kill_daemon.onionr_help = "Gracefully stops the " # type: ignore
kill_daemon.onionr_help += "Onionr API servers" # type: ignore
2019-09-21 05:06:49 +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:
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:
2019-07-31 05:10:28 +00:00
if not onionrvalues.DEVELOPMENT_MODE:
lock_file = open(filepaths.lock_file, 'w')
lock_file.write('delete at your own risk')
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