mirror of
https://github.com/markqvist/Reticulum.git
synced 2024-10-01 03:15:44 -04:00
Destination table is now stored on exit
This commit is contained in:
parent
5f0f53eba1
commit
49fce1e1cb
@ -1,3 +1,5 @@
|
|||||||
|
import RNS
|
||||||
|
|
||||||
class Interface:
|
class Interface:
|
||||||
IN = False
|
IN = False
|
||||||
OUT = False
|
OUT = False
|
||||||
@ -7,3 +9,7 @@ class Interface:
|
|||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def get_hash(self):
|
||||||
|
# TODO: Maybe expand this to something more unique
|
||||||
|
return RNS.Identity.fullHash(str(self).encode("utf-8"))
|
@ -48,8 +48,13 @@ class Reticulum:
|
|||||||
os.makedirs(Reticulum.cachepath)
|
os.makedirs(Reticulum.cachepath)
|
||||||
|
|
||||||
if os.path.isfile(self.configpath):
|
if os.path.isfile(self.configpath):
|
||||||
|
try:
|
||||||
self.config = ConfigObj(self.configpath)
|
self.config = ConfigObj(self.configpath)
|
||||||
RNS.log("Configuration loaded from "+self.configpath)
|
RNS.log("Configuration loaded from "+self.configpath)
|
||||||
|
except Exception as e:
|
||||||
|
RNS.log("Could not parse the configuration at "+self.configpath, RNS.LOG_ERROR)
|
||||||
|
RNS.log("Check your configuration file for errors!", RNS.LOG_ERROR)
|
||||||
|
RNS.panic()
|
||||||
else:
|
else:
|
||||||
RNS.log("Could not load config file, creating default configuration file...")
|
RNS.log("Could not load config file, creating default configuration file...")
|
||||||
self.createDefaultConfig()
|
self.createDefaultConfig()
|
||||||
@ -103,8 +108,9 @@ class Reticulum:
|
|||||||
RNS.log("", RNS.LOG_CRITICAL)
|
RNS.log("", RNS.LOG_CRITICAL)
|
||||||
Reticulum.__allow_unencrypted = True
|
Reticulum.__allow_unencrypted = True
|
||||||
|
|
||||||
|
interface_names = []
|
||||||
for name in self.config["interfaces"]:
|
for name in self.config["interfaces"]:
|
||||||
|
if not name in interface_names:
|
||||||
c = self.config["interfaces"][name]
|
c = self.config["interfaces"][name]
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@ -303,7 +309,10 @@ class Reticulum:
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
RNS.log("The interface \""+name+"\" could not be created. Check your configuration file for errors!", RNS.LOG_ERROR)
|
RNS.log("The interface \""+name+"\" could not be created. Check your configuration file for errors!", RNS.LOG_ERROR)
|
||||||
RNS.log("The contained exception was: "+str(e), RNS.LOG_ERROR)
|
RNS.log("The contained exception was: "+str(e), RNS.LOG_ERROR)
|
||||||
raise e
|
RNS.panic()
|
||||||
|
else:
|
||||||
|
RNS.log("The interface name \""+name+"\" was already used. Check your configuration file for errors!", RNS.LOG_ERROR)
|
||||||
|
RNS.panic()
|
||||||
|
|
||||||
|
|
||||||
def createDefaultConfig(self):
|
def createDefaultConfig(self):
|
||||||
|
108
RNS/Transport.py
108
RNS/Transport.py
@ -70,25 +70,25 @@ class Transport:
|
|||||||
@staticmethod
|
@staticmethod
|
||||||
def start():
|
def start():
|
||||||
if Transport.identity == None:
|
if Transport.identity == None:
|
||||||
transport_identity_path = RNS.Reticulum.configdir+"/transportidentity"
|
transport_identity_path = RNS.Reticulum.storagepath+"/transport_identity"
|
||||||
if os.path.isfile(transport_identity_path):
|
if os.path.isfile(transport_identity_path):
|
||||||
Transport.identity = RNS.Identity.from_file(transport_identity_path)
|
Transport.identity = RNS.Identity.from_file(transport_identity_path)
|
||||||
|
|
||||||
if Transport.identity == None:
|
if Transport.identity == None:
|
||||||
RNS.log("No valid Transport Identity on disk, creating...", RNS.LOG_VERBOSE)
|
RNS.log("No valid Transport Identity in storage, creating...", RNS.LOG_VERBOSE)
|
||||||
Transport.identity = RNS.Identity()
|
Transport.identity = RNS.Identity()
|
||||||
Transport.identity.save(transport_identity_path)
|
Transport.identity.save(transport_identity_path)
|
||||||
else:
|
else:
|
||||||
RNS.log("Loaded Transport Identity from disk", RNS.LOG_VERBOSE)
|
RNS.log("Loaded Transport Identity from storage", RNS.LOG_VERBOSE)
|
||||||
|
|
||||||
packet_hashlist_path = RNS.Reticulum.configdir+"/packet_hashlist"
|
packet_hashlist_path = RNS.Reticulum.storagepath+"/packet_hashlist"
|
||||||
if os.path.isfile(packet_hashlist_path):
|
if os.path.isfile(packet_hashlist_path):
|
||||||
try:
|
try:
|
||||||
file = open(packet_hashlist_path, "rb")
|
file = open(packet_hashlist_path, "rb")
|
||||||
Transport.packet_hashlist = umsgpack.unpackb(file.read())
|
Transport.packet_hashlist = umsgpack.unpackb(file.read())
|
||||||
file.close()
|
file.close()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
RNS.log("Could not load packet hashlist from disk, the contained exception was: "+str(e), RNS.LOG_ERROR)
|
RNS.log("Could not load packet hashlist from storage, the contained exception was: "+str(e), RNS.LOG_ERROR)
|
||||||
|
|
||||||
# Create transport-specific destinations
|
# Create transport-specific destinations
|
||||||
path_request_destination = RNS.Destination(None, RNS.Destination.IN, RNS.Destination.PLAIN, Transport.APP_NAME, "path", "request")
|
path_request_destination = RNS.Destination(None, RNS.Destination.IN, RNS.Destination.PLAIN, Transport.APP_NAME, "path", "request")
|
||||||
@ -99,6 +99,38 @@ class Transport:
|
|||||||
thread.start()
|
thread.start()
|
||||||
|
|
||||||
if RNS.Reticulum.transport_enabled():
|
if RNS.Reticulum.transport_enabled():
|
||||||
|
destination_table_path = RNS.Reticulum.storagepath+"/destination_table"
|
||||||
|
if os.path.isfile(destination_table_path):
|
||||||
|
serialised_destinations = []
|
||||||
|
try:
|
||||||
|
file = open(destination_table_path, "rb")
|
||||||
|
serialised_destinations = umsgpack.unpackb(file.read())
|
||||||
|
file.close()
|
||||||
|
|
||||||
|
for serialised_entry in serialised_destinations:
|
||||||
|
destination_hash = serialised_entry[0]
|
||||||
|
receiving_interface = Transport.find_interface_from_hash(serialised_entry[6])
|
||||||
|
announce_packet = Transport.get_cached_packet(serialised_entry[7])
|
||||||
|
|
||||||
|
if announce_packet != None and receiving_interface != None:
|
||||||
|
announce_packet.unpack()
|
||||||
|
timestamp = serialised_entry[1]
|
||||||
|
received_from = serialised_entry[2]
|
||||||
|
hops = serialised_entry[3]
|
||||||
|
expires = serialised_entry[4]
|
||||||
|
random_blobs = serialised_entry[5]
|
||||||
|
Transport.destination_table[destination_hash] = [timestamp, received_from, hops, expires, random_blobs, receiving_interface, announce_packet]
|
||||||
|
RNS.log("Loaded destination table entry for "+RNS.prettyhexrep(destination_hash)+" from storage", RNS.LOG_DEBUG)
|
||||||
|
else:
|
||||||
|
RNS.log("Could not reconstruct destination table entry from storage for "+RNS.prettyhexrep(destination_hash), RNS.LOG_DEBUG)
|
||||||
|
if announce_packet == None:
|
||||||
|
RNS.log("The announce packet could not be loaded from cache", RNS.LOG_DEBUG)
|
||||||
|
if receiving_interface == None:
|
||||||
|
RNS.log("The interface is no longer available", RNS.LOG_DEBUG)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
RNS.log("Could not load destination table from storage, the contained exception was: "+str(e), RNS.LOG_ERROR)
|
||||||
|
|
||||||
RNS.log("Transport instance "+str(Transport.identity)+" started")
|
RNS.log("Transport instance "+str(Transport.identity)+" started")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@ -599,6 +631,13 @@ class Transport:
|
|||||||
else:
|
else:
|
||||||
RNS.log("Attempted to activate a link that was not in the pending table", RNS.LOG_ERROR)
|
RNS.log("Attempted to activate a link that was not in the pending table", RNS.LOG_ERROR)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def find_interface_from_hash(interface_hash):
|
||||||
|
for interface in Transport.interfaces:
|
||||||
|
if interface.get_hash() == interface_hash:
|
||||||
|
return interface
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def shouldCache(packet):
|
def shouldCache(packet):
|
||||||
@ -610,8 +649,8 @@ class Transport:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def cache(packet):
|
def cache(packet, force_cache=False):
|
||||||
if RNS.Transport.shouldCache(packet):
|
if RNS.Transport.shouldCache(packet) or force_cache:
|
||||||
try:
|
try:
|
||||||
packet_hash = RNS.hexrep(packet.getHash(), delimit=False)
|
packet_hash = RNS.hexrep(packet.getHash(), delimit=False)
|
||||||
file = open(RNS.Reticulum.cachepath+"/"+packet_hash, "wb")
|
file = open(RNS.Reticulum.cachepath+"/"+packet_hash, "wb")
|
||||||
@ -622,20 +661,36 @@ class Transport:
|
|||||||
RNS.log("Error writing packet to cache", RNS.LOG_ERROR)
|
RNS.log("Error writing packet to cache", RNS.LOG_ERROR)
|
||||||
RNS.log("The contained exception was: "+str(e))
|
RNS.log("The contained exception was: "+str(e))
|
||||||
|
|
||||||
# TODO: Implement cache requests. Needs methodology
|
|
||||||
# rethinking. This is skeleton code.
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def cache_request_packet(packet):
|
def get_cached_packet(packet_hash):
|
||||||
if len(packet.data) == RNS.Identity.HASHLENGTH/8:
|
try:
|
||||||
packet_hash = RNS.hexrep(packet.data, delimit=False)
|
packet_hash = RNS.hexrep(packet_hash, delimit=False)
|
||||||
path = RNS.Reticulum.cachepath+"/"+packet_hash
|
path = RNS.Reticulum.cachepath+"/"+packet_hash
|
||||||
if os.path.isfile(path):
|
if os.path.isfile(path):
|
||||||
file = open(path, "rb")
|
file = open(path, "rb")
|
||||||
raw = file.read()
|
raw = file.read()
|
||||||
file.close()
|
file.close()
|
||||||
packet = RNS.Packet(None, raw)
|
packet = RNS.Packet(None, raw)
|
||||||
# TODO: Implement outbound for this
|
return packet
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
except Exception as e:
|
||||||
|
RNS.log("Exception occurred while getting cached packet.", RNS.LOG_ERROR)
|
||||||
|
RNS.log("The contained exception was: "+str(e), RNS.LOG_ERROR)
|
||||||
|
|
||||||
|
# TODO: Implement cache requests. Needs methodology
|
||||||
|
# rethinking. This is skeleton code.
|
||||||
|
@staticmethod
|
||||||
|
def cache_request_packet(packet):
|
||||||
|
if len(packet.data) == RNS.Identity.HASHLENGTH/8:
|
||||||
|
packet_hash = RNS.hexrep(packet.data, delimit=False)
|
||||||
|
packet = Transport.get_cached_packet(packet_hash)
|
||||||
|
|
||||||
|
if packet != None:
|
||||||
|
# TODO: Implement outbound for this
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
pass
|
||||||
|
|
||||||
# TODO: Implement cache requests. Needs methodology
|
# TODO: Implement cache requests. Needs methodology
|
||||||
# rethinking. This is skeleton code.
|
# rethinking. This is skeleton code.
|
||||||
@ -704,10 +759,33 @@ class Transport:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def exitHandler():
|
def exitHandler():
|
||||||
|
RNS.log("Saving packet hashlist to storage...", RNS.LOG_VERBOSE)
|
||||||
try:
|
try:
|
||||||
packet_hashlist_path = RNS.Reticulum.configdir+"/packet_hashlist"
|
packet_hashlist_path = RNS.Reticulum.storagepath+"/packet_hashlist"
|
||||||
file = open(packet_hashlist_path, "wb")
|
file = open(packet_hashlist_path, "wb")
|
||||||
file.write(umsgpack.packb(Transport.packet_hashlist))
|
file.write(umsgpack.packb(Transport.packet_hashlist))
|
||||||
file.close()
|
file.close()
|
||||||
|
RNS.log("Done packet hashlist to storage", RNS.LOG_VERBOSE)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
RNS.log("Could not save packet hashlist to disk, the contained exception was: "+str(e), RNS.LOG_ERROR)
|
RNS.log("Could not save packet hashlist to storage, the contained exception was: "+str(e), RNS.LOG_ERROR)
|
||||||
|
|
||||||
|
RNS.log("Saving destination table to storage...", RNS.LOG_VERBOSE)
|
||||||
|
try:
|
||||||
|
serialised_destinations = []
|
||||||
|
for destination_hash in Transport.destination_table:
|
||||||
|
destination_entry = Transport.destination_table[destination_hash]
|
||||||
|
|
||||||
|
de = destination_entry
|
||||||
|
Transport.cache(de[6], force_cache=True)
|
||||||
|
interface_hash = de[5].get_hash()
|
||||||
|
packet_hash = de[6].getHash()
|
||||||
|
serialised_entry = [destination_hash, de[0], de[1], de[2], de[3], de[4], interface_hash, packet_hash]
|
||||||
|
serialised_destinations.append(serialised_entry)
|
||||||
|
|
||||||
|
destination_table_path = RNS.Reticulum.storagepath+"/destination_table"
|
||||||
|
file = open(destination_table_path, "wb")
|
||||||
|
file.write(umsgpack.packb(serialised_destinations))
|
||||||
|
file.close()
|
||||||
|
RNS.log("Done saving destination table to storage", RNS.LOG_VERBOSE)
|
||||||
|
except Exception as e:
|
||||||
|
RNS.log("Could not save destination table to storage, the contained exception was: "+str(e), RNS.LOG_ERROR)
|
||||||
|
Loading…
Reference in New Issue
Block a user