Reticulum/FPE/FlexPE.py

120 lines
3.3 KiB
Python
Raw Normal View History

2016-06-03 17:02:02 +00:00
from Interfaces import *
2018-03-19 15:39:08 +00:00
import FPE
2016-06-03 17:02:02 +00:00
import ConfigParser
import jsonpickle
from vendor.configobj import ConfigObj
import struct
import array
import os.path
import os
class FlexPE:
2018-03-16 09:50:37 +00:00
MTU = 600
2016-06-03 17:02:02 +00:00
router = None
config = None
destinations = []
interfaces = []
configdir = os.path.expanduser("~")+"/.flexpe"
configpath = configdir+"/config"
2018-03-16 10:40:37 +00:00
def __init__(self,config=None):
if config != None:
self.configpath = config
else:
self.configpath = FlexPE.configpath
if os.path.isfile(self.configpath):
self.config = ConfigObj(self.configpath)
2018-03-19 15:39:08 +00:00
FPE.log("Configuration loaded from "+self.configpath)
2016-06-03 17:02:02 +00:00
else:
2018-03-19 15:39:08 +00:00
FPE.log("Could not load config file, creating default configuration...")
2016-06-03 17:02:02 +00:00
self.createDefaultConfig()
self.applyConfig()
FlexPE.router = self
@staticmethod
def addDestination(destination):
destination.MTU = FlexPE.MTU
FlexPE.destinations.append(destination)
@staticmethod
def incoming(data):
2018-03-19 15:39:08 +00:00
packet = FPE.Packet(None, data)
packet.unpack()
2016-06-03 17:02:02 +00:00
for destination in FlexPE.destinations:
2018-03-19 15:39:08 +00:00
if destination.hash == packet.destination_hash and destination.type == packet.destination_type:
destination.receive(packet.data)
2016-06-03 17:02:02 +00:00
@staticmethod
def outbound(raw):
for interface in FlexPE.interfaces:
if interface.OUT:
2018-03-19 15:39:08 +00:00
FPE.log("Transmitting via: "+str(interface), FPE.LOG_DEBUG)
2016-06-03 17:02:02 +00:00
interface.processOutgoing(raw)
def applyConfig(self):
2018-03-19 15:39:08 +00:00
for option in self.config["logging"]:
value = self.config["logging"][option]
if option == "loglevel":
FPE.loglevel = int(value)
2016-06-03 17:02:02 +00:00
for name in self.config["interfaces"]:
c = self.config["interfaces"][name]
try:
if c["type"] == "UdpInterface":
interface = UdpInterface.UdpInterface(
self,
c["listen_ip"],
int(c["listen_port"]),
c["forward_ip"],
int(c["forward_port"])
)
if c["use_as_outgoing"].lower() == "true":
interface.OUT = True
interface.name = name
FlexPE.interfaces.append(interface)
if c["type"] == "SerialInterface":
interface = SerialInterface.SerialInterface(
self,
c["port"],
int(c["speed"]),
int(c["databits"]),
c["parity"],
int(c["stopbits"])
)
if c["use_as_outgoing"].lower() == "true":
interface.OUT = True
interface.name = name
FlexPE.interfaces.append(interface)
except Exception as e:
2018-03-19 15:39:08 +00:00
FPE.log("The interface \""+name+"\" could not be created. Check your configuration file for errors!", FPE.LOG_ERROR)
FPE.log("The contained exception was: "+str(e), FPE.LOG_ERROR)
2016-06-03 17:02:02 +00:00
def createDefaultConfig(self):
self.config = ConfigObj()
self.config.filename = FlexPE.configpath
self.config["interfaces"] = {}
self.config["interfaces"]["Default UDP Interface"] = {}
self.config["interfaces"]["Default UDP Interface"]["type"] = "UdpInterface"
self.config["interfaces"]["Default UDP Interface"]["listen_ip"] = "0.0.0.0"
self.config["interfaces"]["Default UDP Interface"]["listen_port"] = 7777
self.config["interfaces"]["Default UDP Interface"]["forward_ip"] = "255.255.255.255"
self.config["interfaces"]["Default UDP Interface"]["forward_port"] = 7777
self.config["interfaces"]["Default UDP Interface"]["use_as_outgoing"] = "true"
if not os.path.isdir(FlexPE.configdir):
os.makedirs(FlexPE.configdir)
self.config.write()
self.applyConfig()