Reticulum/RNS/Bundle.py

117 lines
3.5 KiB
Python
Raw Normal View History

2020-06-14 18:18:46 +00:00
import RNS
2020-08-11 18:15:23 +00:00
import time
import os
import os.path
from .vendor import umsgpack as umsgpack
2020-06-14 18:18:46 +00:00
class Bundle:
2020-08-11 18:15:23 +00:00
NO_CUSTODY = 0x00;
TAKING_CUSTODY = 0x01;
FULL_CUSTODY = 0x02;
CHUNK_SIZE = RNS.Resource.MAX_EFFICIENT_SIZE / 4
def __init__(self, destination = None, data = None, filepath = None, advertised_id = None):
self.destination = destination
self.state = None
self.data_file = None
self.meta_file = None
self.id = None
self.storagepath = None
self.size = None
self.chunks = 0
self.heartbeat = time.time()
self.resources = []
try:
if data != None or filepath != None:
if filepath == None and data != None:
try:
self.id = RNS.Identity.fullHash(data)
self.storagepath = Reticulum.bundlepath+"/"+self.id.hex()
self.datapath = self.storagepath+"/data"
self.metadatapath = self.storagepath+"/metadata"
if not os.path.isdir(self.storagepath):
os.makedirs(self.storagepath)
else:
RNS.log("Warning, bundle already exists in storage location, recreating", RNS.LOG_DEBUG)
self.data_file = open(self.datapath, "wb")
self.data_file.write(data)
self.data_file.close()
except Exception as e:
RNS.log("Error while initialising bundle from data, the contained exception was:", RNS.LOG_ERROR)
RNS.log(str(e))
self.state = Bundle.FULL_CUSTODY
elif data == None and filepath != None:
try:
input_file = open(filepath, "rb")
self.id = RNS.Identity.fullHash(input_file.read())
input_file.seek(0)
self.storagepath = RNS.Reticulum.bundlepath+"/"+self.id.hex()
self.datapath = self.storagepath+"/data"
self.metadatapath = self.storagepath+"/metadata"
if not os.path.isdir(self.storagepath):
os.makedirs(self.storagepath)
else:
RNS.log("Warning, bundle already exists in storage location, recreating", RNS.LOG_DEBUG)
self.data_file = open(self.datapath, "wb")
self.data_file.write(input_file.read())
self.data_file.close()
input_file.close()
except Exception as e:
RNS.log("Error while reading input file for bundle, the contained exception was:", RNS.LOG_ERROR)
RNS.log(str(e))
self.state = Bundle.FULL_CUSTODY
else:
raise ValueError("Bundle cannot be created from data and file path at the same time")
elif advertised_id != None:
# Incoming bundle transfer
self.state = Bundle.TAKING_CUSTODY
else:
raise ValueError("No source of data specified for bundle initialisation")
# Prepare file handles and metadata
self.size = os.stat(self.datapath).st_size
if self.size < 1:
raise IOError("Bundle data is empty")
self.chunks = ((self.size-1)//Bundle.CHUNK_SIZE)+1
self.data_file = open(self.datapath, "rb")
self.flush_metadata()
except Exception as e:
RNS.log("Error while initialising bundle. The contained exception was:", RNS.LOG_ERROR)
RNS.log(str(e), RNS.LOG_ERROR)
raise e
def flush_metadata(self):
try:
metadata = {
"destination": self.destination,
"heartbeat": self.heartbeat,
"size": self.size,
"chunks": self.chunks,
"state": self.state}
self.meta_file = open(self.metadatapath, "wb")
self.meta_file.write(umsgpack.packb(metadata))
self.meta_file.close()
except Exception as e:
RNS.log("Error while flushing metadata for bundle "+RNS.prettyhexrep(self.id), RNS.LOG_ERROR)
RNS.log("The contained exception was: "+str(e), RNS.LOG_ERROR)
2020-06-14 18:18:46 +00:00
class BundleAdvertisement:
pass