Simon
3 years ago
parent
6312699d07
commit
bff5f8a186
@ -1,264 +0,0 @@
|
|||||||
###############################################################################
|
|
||||||
# Copyright (C) 2020 Simon Adlem, G7RZU <g7rzu@gb7fr.org.uk>
|
|
||||||
#
|
|
||||||
# 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, write to the Free Software Foundation,
|
|
||||||
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
||||||
###############################################################################
|
|
||||||
|
|
||||||
from twisted.internet.protocol import DatagramProtocol
|
|
||||||
from twisted.internet import reactor, task
|
|
||||||
from time import time
|
|
||||||
from dmr_utils3.utils import int_id
|
|
||||||
import random
|
|
||||||
import ipaddress
|
|
||||||
import os
|
|
||||||
from setproctitle import setproctitle
|
|
||||||
from datetime import datetime
|
|
||||||
#from ipsc_const import *
|
|
||||||
|
|
||||||
# Does anybody read this stuff? There's a PEP somewhere that says I should do this.
|
|
||||||
__author__ = 'Simon Adlem - G7RZU'
|
|
||||||
__copyright__ = 'Copyright (c) Simon Adlem, G7RZU 2022'
|
|
||||||
__credits__ = 'Jon Lee, G4TSN; Norman Williams, M6NBP'
|
|
||||||
__license__ = 'GNU GPLv3'
|
|
||||||
__maintainer__ = 'Simon Adlem G7RZU'
|
|
||||||
__email__ = 'simon@gb7fr.org.uk'
|
|
||||||
|
|
||||||
def IsIPv4Address(ip):
|
|
||||||
try:
|
|
||||||
ipaddress.IPv4Address(ip)
|
|
||||||
return True
|
|
||||||
except ValueError as errorCode:
|
|
||||||
pass
|
|
||||||
return False
|
|
||||||
|
|
||||||
def IsIPv6Address(ip):
|
|
||||||
try:
|
|
||||||
ipaddress.IPv6Address(ip)
|
|
||||||
return True
|
|
||||||
except ValueError as errorCode:
|
|
||||||
pass
|
|
||||||
|
|
||||||
class Proxy(DatagramProtocol):
|
|
||||||
|
|
||||||
def __init__(self,Master,ListenPort,connTrack,peerTrack,blackList,IPBlackList,Timeout,Debug,ClientInfo,DestportStart,DestPortEnd):
|
|
||||||
self.master = Master
|
|
||||||
self.connTrack = connTrack
|
|
||||||
self.peerTrack = peerTrack
|
|
||||||
self.timeout = Timeout
|
|
||||||
self.debug = Debug
|
|
||||||
self.clientinfo = ClientInfo
|
|
||||||
self.blackList = blackList
|
|
||||||
self.IPBlackList = IPBlackList
|
|
||||||
self.destPortStart = DestportStart
|
|
||||||
self.destPortEnd = DestPortEnd
|
|
||||||
self.numPorts = DestPortEnd - DestportStart
|
|
||||||
|
|
||||||
|
|
||||||
def reaper(self,_peer_id):
|
|
||||||
if self.debug:
|
|
||||||
print("dead",_peer_id)
|
|
||||||
if self.clientinfo:
|
|
||||||
print(f"{datetime.now().replace(microsecond=0)} Client: ID:{str(int_id(_peer_id)).rjust(9)} IP:{self.peerTrack[_peer_id]['shost'].rjust(15)} Port:{self.peerTrack[_peer_id]['sport']} Removed.")
|
|
||||||
self.connTrack[self.peerTrack[_peer_id]['dport']] = False
|
|
||||||
del self.peerTrack[_peer_id]
|
|
||||||
|
|
||||||
def datagramReceived(self, data, addr):
|
|
||||||
|
|
||||||
_peer_id = False
|
|
||||||
|
|
||||||
host,port = addr
|
|
||||||
|
|
||||||
nowtime = time()
|
|
||||||
|
|
||||||
Debug = self.debug
|
|
||||||
|
|
||||||
if host in self.IPBlackList:
|
|
||||||
return
|
|
||||||
|
|
||||||
#If the packet comes from the master
|
|
||||||
if host == self.master:
|
|
||||||
|
|
||||||
_command = data[0:1]
|
|
||||||
_peer_id = data[1:5]
|
|
||||||
|
|
||||||
if self.debug:
|
|
||||||
print(data)
|
|
||||||
if _peer_id in self.peerTrack:
|
|
||||||
self.transport.write(data,(self.peerTrack[_peer_id]['shost'],self.peerTrack[_peer_id]['sport']))
|
|
||||||
|
|
||||||
return
|
|
||||||
|
|
||||||
|
|
||||||
else:
|
|
||||||
_command = data[0:1]
|
|
||||||
_peer_id = data[1:5]
|
|
||||||
|
|
||||||
if _peer_id in self.peerTrack:
|
|
||||||
_dport = self.peerTrack[_peer_id]['dport']
|
|
||||||
self.peerTrack[_peer_id]['sport'] = port
|
|
||||||
self.peerTrack[_peer_id]['shost'] = host
|
|
||||||
self.transport.write(data, (self.master,_dport))
|
|
||||||
self.peerTrack[_peer_id]['timer'].reset(self.timeout)
|
|
||||||
if self.debug:
|
|
||||||
print(data)
|
|
||||||
return
|
|
||||||
|
|
||||||
else:
|
|
||||||
if int_id(_peer_id) in self.blackList:
|
|
||||||
return
|
|
||||||
# Make a list with the available ports
|
|
||||||
_ports_avail = [port for port in self.connTrack if not self.connTrack[port]]
|
|
||||||
if _ports_avail:
|
|
||||||
_dport = random.choice(_ports_avail)
|
|
||||||
else:
|
|
||||||
return
|
|
||||||
self.connTrack[_dport] = _peer_id
|
|
||||||
self.peerTrack[_peer_id] = {}
|
|
||||||
self.peerTrack[_peer_id]['dport'] = _dport
|
|
||||||
self.peerTrack[_peer_id]['sport'] = port
|
|
||||||
self.peerTrack[_peer_id]['shost'] = host
|
|
||||||
self.peerTrack[_peer_id]['timer'] = reactor.callLater(self.timeout,self.reaper,_peer_id)
|
|
||||||
self.transport.write(data, (self.master,_dport))
|
|
||||||
|
|
||||||
if self.clientinfo:
|
|
||||||
print(f'{datetime.now().replace(microsecond=0)} New client: ID:{str(int_id(_peer_id)).rjust(9)} IP:{host.rjust(15)} Port:{port}, assigned to port:{_dport}.')
|
|
||||||
if self.debug:
|
|
||||||
print(data)
|
|
||||||
return
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
|
|
||||||
import signal
|
|
||||||
import configparser
|
|
||||||
import argparse
|
|
||||||
import sys
|
|
||||||
import json
|
|
||||||
|
|
||||||
#Set process title early
|
|
||||||
setproctitle(__file__)
|
|
||||||
|
|
||||||
# Change the current directory to the location of the application
|
|
||||||
os.chdir(os.path.dirname(os.path.realpath(sys.argv[0])))
|
|
||||||
|
|
||||||
# CLI argument parser - handles picking up the config file from the command line, and sending a "help" message
|
|
||||||
parser = argparse.ArgumentParser()
|
|
||||||
parser.add_argument('-c', '--config', action='store', dest='CONFIG_FILE', help='/full/path/to/config.file (usually freedmr.cfg)')
|
|
||||||
cli_args = parser.parse_args()
|
|
||||||
|
|
||||||
|
|
||||||
# Ensure we have a path for the config file, if one wasn't specified, then use the execution directory
|
|
||||||
if not cli_args.CONFIG_FILE:
|
|
||||||
cli_args.CONFIG_FILE = os.path.dirname(os.path.abspath(__file__))+'/freedmr.cfg'
|
|
||||||
|
|
||||||
_config_file = cli_args.CONFIG_FILE
|
|
||||||
|
|
||||||
config = configparser.ConfigParser()
|
|
||||||
|
|
||||||
if not config.read(_config_file):
|
|
||||||
print('Configuration file \''+_config_file+'\' is not a valid configuration file!')
|
|
||||||
|
|
||||||
try:
|
|
||||||
|
|
||||||
Master = config.get('IPSC_PROXY','Master')
|
|
||||||
ListenPort = config.getint('IPSC_PROXY','ListenPort')
|
|
||||||
ListenIP = config.get('IPSC_PROXY','ListenIP')
|
|
||||||
DestportStart = config.getint('IPSC_PROXY','DestportStart')
|
|
||||||
DestPortEnd = config.getint('IPSC_PROXY','DestPortEnd')
|
|
||||||
Timeout = config.getint('IPSC_PROXY','Timeout')
|
|
||||||
Stats = config.getboolean('IPSC_PROXY','Stats')
|
|
||||||
Debug = config.getboolean('IPSC_PROXY','Debug')
|
|
||||||
ClientInfo = config.getboolean('IPSC_PROXY','ClientInfo')
|
|
||||||
BlackList = json.loads(config.get('IPSC_PROXY','BlackList'))
|
|
||||||
IPBlackList = json.loads(config.get('IPSC_PROXY','IPBlackList'))
|
|
||||||
|
|
||||||
except configparser.Error as err:
|
|
||||||
print('Error processing configuration file -- {}'.format(err))
|
|
||||||
|
|
||||||
print('Using default config')
|
|
||||||
#*** CONFIG HERE ***
|
|
||||||
|
|
||||||
Master = "127.0.0.1"
|
|
||||||
ListenPort = 55005
|
|
||||||
#'' = all IPv4, '::' = all IPv4 and IPv6 (Dual Stack)
|
|
||||||
ListenIP = ''
|
|
||||||
DestportStart = 59000
|
|
||||||
DestPortEnd = 59001
|
|
||||||
Timeout = 30
|
|
||||||
Stats = True
|
|
||||||
Debug = True
|
|
||||||
ClientInfo = True
|
|
||||||
BlackList = [1234567]
|
|
||||||
#e.g. {10.0.0.1: 0, 10.0.0.2: 0}
|
|
||||||
IPBlackList = {}
|
|
||||||
|
|
||||||
#*******************
|
|
||||||
|
|
||||||
CONNTRACK = {}
|
|
||||||
PEERTRACK = {}
|
|
||||||
|
|
||||||
# Set up the signal handler
|
|
||||||
def sig_handler(_signal, _frame):
|
|
||||||
print('(GLOBAL) SHUTDOWN: IPSC_PROXY IS TERMINATING WITH SIGNAL {}'.format(str(_signal)))
|
|
||||||
reactor.stop()
|
|
||||||
|
|
||||||
# Set signal handers so that we can gracefully exit if need be
|
|
||||||
for sig in [signal.SIGINT, signal.SIGTERM]:
|
|
||||||
signal.signal(sig, sig_handler)
|
|
||||||
|
|
||||||
#Override static config from Environment
|
|
||||||
if 'FDIPSC_PROXY_STATS' in os.environ:
|
|
||||||
Stats = bool(os.environ['FDIPSC_PROXY_STATS'])
|
|
||||||
#if 'FDIPSC_PROXY_DEBUG' in os.environ:
|
|
||||||
# Debug = bool(os.environ['FDIPSC_PROXY_DEBUG'])
|
|
||||||
if 'FDIPSC_PROXY_CLIENTINFO' in os.environ:
|
|
||||||
ClientInfo = bool(os.environ['FDIPSC_PROXY_CLIENTINFO'])
|
|
||||||
if 'FDIPSC_PROXY_LISTENPORT' in os.environ:
|
|
||||||
ListenPort = int(os.environ['FDIPSC_PROXY_LISTENPORT'])
|
|
||||||
|
|
||||||
for port in range(DestportStart,DestPortEnd+1,1):
|
|
||||||
CONNTRACK[port] = False
|
|
||||||
|
|
||||||
#If we are listening IPv6 and Master is an IPv4 IPv4Address
|
|
||||||
#IPv6ify the address.
|
|
||||||
if ListenIP == '::' and IsIPv4Address(Master):
|
|
||||||
Master = '::ffff:' + Master
|
|
||||||
|
|
||||||
reactor.listenUDP(ListenPort,Proxy(Master,ListenPort,CONNTRACK,PEERTRACK,BlackList,IPBlackList,Timeout,Debug,ClientInfo,DestportStart,DestPortEnd),interface=ListenIP)
|
|
||||||
|
|
||||||
def loopingErrHandle(failure):
|
|
||||||
print('(GLOBAL) STOPPING REACTOR TO AVOID MEMORY LEAK: Unhandled error innowtimed loop.\n {}'.format(failure))
|
|
||||||
reactor.stop()
|
|
||||||
|
|
||||||
def stats():
|
|
||||||
count = 0
|
|
||||||
nowtime = time()
|
|
||||||
for port in CONNTRACK:
|
|
||||||
if CONNTRACK[port]:
|
|
||||||
count = count+1
|
|
||||||
|
|
||||||
totalPorts = DestPortEnd - DestportStart
|
|
||||||
freePorts = totalPorts - count
|
|
||||||
|
|
||||||
print("{} ports out of {} in use ({} free)".format(count,totalPorts,freePorts))
|
|
||||||
|
|
||||||
|
|
||||||
if Stats == True:
|
|
||||||
stats_task = task.LoopingCall(stats)
|
|
||||||
statsa = stats_task.start(30)
|
|
||||||
statsa.addErrback(loopingErrHandle)
|
|
||||||
|
|
||||||
reactor.run()
|
|
||||||
|
|
||||||
Loading…
Reference in new issue