Onionr/static-data/default-plugins/torgossip/main.py
2021-02-16 05:01:01 +00:00

77 lines
2.4 KiB
Python
Executable File

"""
Onionr - Private P2P Communication
Default plugin which allows users to encrypt/decrypt messages w/o using blocks
"""
from base64 import b32encode
import locale
from netcontroller.torcontrol import torcontroller
locale.setlocale(locale.LC_ALL, '')
import sys
import os
import traceback
from threading import Thread
from netcontroller.torcontrol import onionservice
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))
from constants import SERVER_SOCKET, GOSSIP_PORT, HOSTNAME_FILE
"""
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/>.
"""
plugin_name = 'torgossip'
import logger # noqa
try:
from server import start_server
from client import start_client
from peerdb import TorGossipPeers
from runtest import torgossip_runtest
except Exception as _: # noqa
logger.error(traceback.format_exc(), terminal=True)
def on_init(api, data=None):
shared_state = data
shared_state.get_by_string(
"OnionrRunTestManager").plugin_tests.append(torgossip_runtest)
shared_state.get(TorGossipPeers)
hs = b""
try:
with open(HOSTNAME_FILE, "rb") as f:
hs = f.read()
if not hs:
raise FileNotFoundError
except FileNotFoundError:
with torcontroller.get_controller() as c:
try:
hs = onionservice.run_new_and_store_service(
c, onionservice.OnionServiceTarget(
GOSSIP_PORT, SERVER_SOCKET))
except Exception:
logger.error(traceback.format_exc(), terminal=True)
raise
with open(HOSTNAME_FILE, "wb") as hf:
hf.write(hs)
logger.info("TorGossip server on " + b32encode(hs).lower().decode('utf-8'), terminal=True)
Thread(target=start_server, daemon=True, args=[shared_state]).start()
Thread(target=start_client, daemon=True, args=[shared_state]).start()