Reticulum/RNS/__init__.py

231 lines
6.6 KiB
Python
Raw Normal View History

2022-04-01 09:18:18 -06:00
# MIT License
#
# Copyright (c) 2016-2022 Mark Qvist / unsigned.io
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
2016-06-03 11:02:02 -06:00
import os
2018-04-16 14:04:28 -06:00
import sys
2016-06-03 11:02:02 -06:00
import glob
2018-03-19 09:39:08 -06:00
import time
import random
import threading
2016-06-03 11:02:02 -06:00
2021-08-19 06:10:37 -06:00
from ._version import __version__
from .Reticulum import Reticulum
2018-03-16 03:50:37 -06:00
from .Identity import Identity
2021-09-02 12:35:42 -06:00
from .Link import Link, RequestReceipt
2018-03-16 03:50:37 -06:00
from .Transport import Transport
2018-03-19 09:39:08 -06:00
from .Destination import Destination
from .Packet import Packet
2018-04-17 09:46:48 -06:00
from .Packet import PacketReceipt
from .Resource import Resource, ResourceAdvertisement
from .Cryptography import HKDF
from .Cryptography import Hashes
2018-03-16 03:50:37 -06:00
2016-06-03 11:02:02 -06:00
modules = glob.glob(os.path.dirname(__file__)+"/*.py")
2018-03-19 09:39:08 -06:00
__all__ = [ os.path.basename(f)[:-3] for f in modules if not f.endswith('__init__.py')]
LOG_CRITICAL = 0
LOG_ERROR = 1
LOG_WARNING = 2
LOG_NOTICE = 3
LOG_INFO = 4
LOG_VERBOSE = 5
LOG_DEBUG = 6
2018-04-23 15:42:16 -06:00
LOG_EXTREME = 7
2018-03-19 09:39:08 -06:00
2020-08-13 04:15:56 -06:00
LOG_STDOUT = 0x91
2018-03-19 09:39:08 -06:00
LOG_FILE = 0x92
2021-10-08 11:23:10 -06:00
LOG_MAXSIZE = 5*1024*1024
2020-08-13 04:15:56 -06:00
loglevel = LOG_NOTICE
2018-03-19 09:39:08 -06:00
logfile = None
logdest = LOG_STDOUT
logtimefmt = "%Y-%m-%d %H:%M:%S"
instance_random = random.Random()
instance_random.seed(os.urandom(10))
_always_override_destination = False
logging_lock = threading.Lock()
2018-03-19 09:39:08 -06:00
def loglevelname(level):
2020-08-13 04:15:56 -06:00
if (level == LOG_CRITICAL):
return "Critical"
if (level == LOG_ERROR):
return "Error"
if (level == LOG_WARNING):
return "Warning"
if (level == LOG_NOTICE):
return "Notice"
if (level == LOG_INFO):
return "Info"
if (level == LOG_VERBOSE):
return "Verbose"
if (level == LOG_DEBUG):
return "Debug"
if (level == LOG_EXTREME):
return "Extra"
return "Unknown"
2018-03-19 09:39:08 -06:00
2021-08-19 06:10:37 -06:00
def version():
return __version__
2021-12-01 03:40:44 -07:00
def host_os():
from .vendor.platformutils import get_platform
return get_platform()
def timestamp_str(time_s):
timestamp = time.localtime(time_s)
return time.strftime(logtimefmt, timestamp)
def log(msg, level=3, _override_destination = False):
global _always_override_destination
2020-08-13 04:15:56 -06:00
if loglevel >= level:
logstring = "["+timestamp_str(time.time())+"] ["+loglevelname(level)+"] "+msg
logging_lock.acquire()
2018-03-19 09:39:08 -06:00
2021-08-19 06:10:37 -06:00
if (logdest == LOG_STDOUT or _always_override_destination or _override_destination):
2020-08-13 04:15:56 -06:00
print(logstring)
logging_lock.release()
2018-03-19 09:39:08 -06:00
elif (logdest == LOG_FILE and logfile != None):
try:
file = open(logfile, "a")
file.write(logstring+"\n")
file.close()
2021-10-08 11:23:10 -06:00
if os.path.getsize(logfile) > LOG_MAXSIZE:
prevfile = logfile+".1"
if os.path.isfile(prevfile):
os.unlink(prevfile)
os.rename(logfile, prevfile)
logging_lock.release()
except Exception as e:
logging_lock.release()
_always_override_destination = True
log("Exception occurred while writing log message to log file: "+str(e), LOG_CRITICAL)
log("Dumping future log events to console!", LOG_CRITICAL)
log(msg, level)
2018-03-19 11:11:50 -06:00
def rand():
result = instance_random.random()
2020-08-13 04:15:56 -06:00
return result
2018-03-19 11:11:50 -06:00
def hexrep(data, delimit=True):
try:
iter(data)
except TypeError:
data = [data]
2020-08-13 04:15:56 -06:00
delimiter = ":"
if not delimit:
delimiter = ""
hexrep = delimiter.join("{:02x}".format(c) for c in data)
return hexrep
2018-03-20 05:32:41 -06:00
def prettyhexrep(data):
2020-08-13 04:15:56 -06:00
delimiter = ""
hexrep = "<"+delimiter.join("{:02x}".format(c) for c in data)+">"
return hexrep
2018-04-16 14:04:28 -06:00
2022-06-09 06:46:36 -06:00
def prettysize(num, suffix='B'):
units = ['','K','M','G','T','P','E','Z']
last_unit = 'Y'
if suffix == 'b':
num *= 8
units = ['','K','M','G','T','P','E','Z']
last_unit = 'Y'
for unit in units:
if abs(num) < 1000.0:
if unit == "":
return "%.0f %s%s" % (num, unit, suffix)
else:
return "%.2f %s%s" % (num, unit, suffix)
num /= 1000.0
return "%.2f%s%s" % (num, last_unit, suffix)
2022-07-02 05:12:54 -06:00
def prettytime(time, verbose=False):
days = int(time // (24 * 3600))
time = time % (24 * 3600)
hours = int(time // 3600)
time %= 3600
minutes = int(time // 60)
time %= 60
seconds = round(time, 2)
ss = "" if seconds == 1 else "s"
sm = "" if minutes == 1 else "s"
sh = "" if hours == 1 else "s"
sd = "" if days == 1 else "s"
components = []
if days > 0:
components.append(str(days)+" day"+sd if verbose else str(days)+"d")
if hours > 0:
components.append(str(hours)+" hour"+sh if verbose else str(hours)+"h")
if minutes > 0:
components.append(str(minutes)+" minute"+sm if verbose else str(minutes)+"m")
if seconds > 0:
components.append(str(seconds)+" second"+ss if verbose else str(seconds)+"s")
i = 0
tstr = ""
for c in components:
i += 1
if i == 1:
pass
elif i < len(components):
tstr += ", "
elif i == len(components):
tstr += " and "
tstr += c
return tstr
2022-06-30 06:02:57 -06:00
def phyparams():
print("Required Physical Layer MTU : "+str(Reticulum.MTU)+" bytes")
print("Plaintext Packet MDU : "+str(Packet.PLAIN_MDU)+" bytes")
print("Encrypted Packet MDU : "+str(Packet.ENCRYPTED_MDU)+" bytes")
print("Link Curve : "+str(Link.CURVE))
print("Link Packet MDU : "+str(Packet.ENCRYPTED_MDU)+" bytes")
print("Link Public Key Size : "+str(Link.ECPUBSIZE*8)+" bits")
print("Link Private Key Size : "+str(Link.KEYSIZE*8)+" bits")
2018-04-16 14:04:28 -06:00
def panic():
os._exit(255)
def exit():
print("")
sys.exit(0)