2016-07-20 17:16:27 -04:00
#!/usr/bin/env python
#
2016-11-21 20:13:32 -05:00
###############################################################################
# Copyright (C) 2016 Cortney T. Buffington, N0MJS <n0mjs@me.com>
#
# 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
###############################################################################
2016-07-20 17:16:27 -04:00
from __future__ import print_function
2016-07-20 22:25:47 -04:00
# Specifig functions from modules we need
2016-11-23 16:01:05 -05:00
from binascii import b2a_hex as ahex
from binascii import a2b_hex as bhex
2016-07-21 13:41:36 -04:00
from random import randint
2016-07-21 17:14:42 -04:00
from hashlib import sha256
2016-07-24 10:42:42 -04:00
from time import time
2016-11-23 16:01:05 -05:00
from bitstring import BitArray
2016-07-30 12:28:49 -04:00
import socket
2016-07-21 13:41:36 -04:00
2016-07-20 22:25:47 -04:00
# Twisted is pretty important, so I keep it separate
2016-07-20 17:16:27 -04:00
from twisted . internet . protocol import DatagramProtocol
from twisted . internet import reactor
2016-07-21 13:41:36 -04:00
from twisted . internet import task
2016-07-20 17:16:27 -04:00
2016-07-20 22:25:47 -04:00
# Other files we pull from -- this is mostly for readability and segmentation
import hb_log
import hb_config
2016-11-23 16:01:05 -05:00
from dmr_utils import int_id , hex_str_4
2016-07-20 22:25:47 -04:00
# Does anybody read this stuff? There's a PEP somewhere that says I should do this.
2016-07-20 17:16:27 -04:00
__author__ = ' Cortney T. Buffington, N0MJS '
2016-07-30 21:37:05 -04:00
__copyright__ = ' Copyright (c) 2016 Cortney T. Buffington, N0MJS and the K0USY Group '
2016-07-23 16:53:22 -04:00
__credits__ = ' Colin Durbridge, G4EML, Steve Zingman, N4IRS; Mike Zingman, N4IRR; Jonathan Naylor, G4KLX; Hans Barthen, DL5DI; Torsten Shultze, DG1HT '
2016-11-22 21:02:09 -05:00
__license__ = ' GNU GPLv3 '
2016-07-20 17:16:27 -04:00
__maintainer__ = ' Cort Buffington, N0MJS '
__email__ = ' n0mjs@me.com '
2016-07-20 22:25:47 -04:00
2016-07-31 10:41:21 -04:00
# Global variables used whether we are a module or __main__
2016-08-25 21:44:15 -04:00
systems = { }
2016-07-31 10:41:21 -04:00
2016-07-23 16:53:22 -04:00
# Shut ourselves down gracefully by disconnecting from the masters and clients.
2016-11-23 16:01:05 -05:00
def hblink_handler ( _signal , _frame , _logger ) :
2016-08-25 21:44:15 -04:00
for system in systems :
2016-11-23 16:01:05 -05:00
_logger . info ( ' SHUTDOWN: DE-REGISTER SYSTEM: %s ' , system )
systems [ system ] . dereg ( )
2016-07-20 17:16:27 -04:00
2016-07-24 10:42:42 -04:00
2016-07-30 12:28:49 -04:00
#************************************************
# AMBE CLASS: Used to parse out AMBE and send to gateway
#************************************************
class AMBE :
2016-11-21 20:54:23 -05:00
def __init__ ( self , _config ) :
self . _CONFIG = _config
self . _sock = socket . socket ( socket . AF_INET , socket . SOCK_DGRAM )
self . _exp_ip = self . _CONFIG [ ' AMBE ' ] [ ' EXPORT_IP ' ]
self . _exp_port = self . _CONFIG [ ' AMBE ' ] [ ' EXPORT_PORT ' ]
2016-09-01 15:33:47 -04:00
2016-07-30 12:28:49 -04:00
def parseAMBE ( self , _client , _data ) :
_seq = int_id ( _data [ 4 : 5 ] )
_srcID = int_id ( _data [ 5 : 8 ] )
_dstID = int_id ( _data [ 8 : 11 ] )
_rptID = int_id ( _data [ 11 : 15 ] )
_bits = int_id ( _data [ 15 : 16 ] ) # SCDV NNNN (Slot|Call type|Data|Voice|Seq or Data type)
_slot = 2 if _bits & 0x80 else 1
_callType = 1 if ( _bits & 0x40 ) else 0
_frameType = ( _bits & 0x30 ) >> 4
_voiceSeq = ( _bits & 0x0f )
_streamID = int_id ( _data [ 16 : 20 ] )
logger . debug ( ' ( %s ) seq: %d srcID: %d dstID: %d rptID: %d bits: %0X slot: %d callType: %d frameType: %d voiceSeq: %d streamID: %0X ' ,
_client , _seq , _srcID , _dstID , _rptID , _bits , _slot , _callType , _frameType , _voiceSeq , _streamID )
#logger.debug('Frame 1:(%s)', self.ByteToHex(_data))
2016-11-23 16:01:05 -05:00
_dmr_frame = BitArray ( ' 0x ' + ahex ( _data [ 20 : ] ) )
2016-07-30 12:28:49 -04:00
_ambe = _dmr_frame [ 0 : 108 ] + _dmr_frame [ 156 : 264 ]
#_sock.sendto(_ambe.tobytes(), ("127.0.0.1", 31000))
ambeBytes = _ambe . tobytes ( )
2016-07-30 21:37:05 -04:00
self . _sock . sendto ( ambeBytes [ 0 : 9 ] , ( self . _exp_ip , self . _exp_port ) )
self . _sock . sendto ( ambeBytes [ 9 : 18 ] , ( self . _exp_ip , self . _exp_port ) )
self . _sock . sendto ( ambeBytes [ 18 : 27 ] , ( self . _exp_ip , self . _exp_port ) )
2016-07-30 12:28:49 -04:00
2016-11-23 16:01:05 -05:00
2016-07-20 17:16:27 -04:00
#************************************************
2016-07-24 10:42:42 -04:00
# HB MASTER CLASS
2016-07-20 17:16:27 -04:00
#************************************************
2016-11-16 12:05:01 -05:00
class HBSYSTEM ( DatagramProtocol ) :
2016-11-23 16:01:05 -05:00
def __init__ ( self , _name , _config , _logger ) :
2016-11-21 20:46:07 -05:00
# Define a few shortcuts to make the rest of the class more readable
2016-11-21 20:54:23 -05:00
self . _CONFIG = _config
self . _system = _name
2016-11-23 16:01:05 -05:00
self . _logger = _logger
2016-11-21 20:54:23 -05:00
self . _config = self . _CONFIG [ ' SYSTEMS ' ] [ self . _system ]
2016-11-21 20:46:07 -05:00
# Define shortcuts and generic function names based on the type of system we are
if self . _config [ ' MODE ' ] == ' MASTER ' :
2016-11-21 20:54:23 -05:00
self . _clients = self . _CONFIG [ ' SYSTEMS ' ] [ self . _system ] [ ' CLIENTS ' ]
2016-11-21 20:46:07 -05:00
self . send_system = self . send_clients
self . maintenance_loop = self . master_maintenance_loop
self . datagramReceived = self . master_datagramReceived
2016-11-23 16:01:05 -05:00
self . dereg = self . master_dereg
2016-11-21 20:46:07 -05:00
elif self . _config [ ' MODE ' ] == ' CLIENT ' :
self . _stats = self . _config [ ' STATS ' ]
self . send_system = self . send_master
self . maintenance_loop = self . client_maintenance_loop
self . datagramReceived = self . client_datagramReceived
2016-11-23 16:01:05 -05:00
self . dereg = self . client_dereg
2016-11-21 20:46:07 -05:00
# Configure for AMBE audio export if enabled
if self . _config [ ' EXPORT_AMBE ' ] :
self . _ambe = AMBE ( )
2016-09-01 15:33:47 -04:00
2016-07-21 13:41:36 -04:00
def startProtocol ( self ) :
2016-07-24 10:42:42 -04:00
# Set up periodic loop for tracking pings from clients. Run every 'PING_TIME' seconds
2016-11-21 20:54:23 -05:00
self . _system_maintenance = task . LoopingCall ( self . maintenance_loop )
self . _system_maintenance_loop = self . _system_maintenance . start ( self . _CONFIG [ ' GLOBAL ' ] [ ' PING_TIME ' ] )
2016-11-16 12:05:01 -05:00
# Aliased in __init__ to maintenance_loop if system is a master
2016-07-24 10:42:42 -04:00
def master_maintenance_loop ( self ) :
2016-11-23 16:01:05 -05:00
self . _logger . debug ( ' ( %s ) Master maintenance loop started ' , self . _system )
2016-07-30 19:42:17 -04:00
for client in self . _clients :
2016-07-24 13:07:43 -04:00
_this_client = self . _clients [ client ]
2016-07-26 21:51:04 -04:00
# Check to see if any of the clients have been quiet (no ping) longer than allowed
2016-11-21 20:54:23 -05:00
if _this_client [ ' LAST_PING ' ] + self . _CONFIG [ ' GLOBAL ' ] [ ' PING_TIME ' ] * self . _CONFIG [ ' GLOBAL ' ] [ ' MAX_MISSED ' ] < time ( ) :
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Client %s ( %s ) has timed out ' , self . _system , _this_client [ ' CALLSIGN ' ] , _this_client [ ' RADIO_ID ' ] )
2016-09-01 15:36:31 -04:00
# Remove any timed out clients from the configuration
2016-11-21 20:54:23 -05:00
del self . _CONFIG [ ' SYSTEMS ' ] [ self . _system ] [ ' CLIENTS ' ] [ client ]
2016-11-16 12:05:01 -05:00
# Aliased in __init__ to maintenance_loop if system is a client
def client_maintenance_loop ( self ) :
2016-11-23 16:01:05 -05:00
self . _logger . debug ( ' ( %s ) Client maintenance loop started ' , self . _system )
2016-11-16 12:05:01 -05:00
# If we're not connected, zero out the stats and send a login request RPTL
if self . _stats [ ' CONNECTION ' ] == ' NO ' or self . _stats [ ' CONNECTION ' ] == ' RTPL_SENT ' :
self . _stats [ ' PINGS_SENT ' ] = 0
self . _stats [ ' PINGS_ACKD ' ] = 0
self . _stats [ ' CONNECTION ' ] = ' RTPL_SENT '
self . send_master ( ' RPTL ' + self . _config [ ' RADIO_ID ' ] )
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Sending login request to master %s : %s ' , self . _system , self . _config [ ' MASTER_IP ' ] , self . _config [ ' MASTER_PORT ' ] )
2016-11-16 12:05:01 -05:00
# If we are connected, sent a ping to the master and increment the counter
if self . _stats [ ' CONNECTION ' ] == ' YES ' :
self . send_master ( ' RPTPING ' + self . _config [ ' RADIO_ID ' ] )
self . _stats [ ' PINGS_SENT ' ] + = 1
2016-11-23 16:01:05 -05:00
self . _logger . debug ( ' ( %s ) RPTPING Sent to Master. Pings Since Connected: %s ' , self . _system , self . _stats [ ' PINGS_SENT ' ] )
2016-09-01 15:33:47 -04:00
2016-08-02 09:10:52 -04:00
def send_clients ( self , _packet ) :
for _client in self . _clients :
2016-08-25 21:44:15 -04:00
self . send_client ( _client , _packet )
2016-11-23 16:01:05 -05:00
#self._logger.debug('(%s) Packet sent to client %s', self._system, self._clients[_client]['RADIO_ID'])
2016-09-01 15:33:47 -04:00
2016-08-25 21:44:15 -04:00
def send_client ( self , _client , _packet ) :
2016-07-30 19:42:17 -04:00
_ip = self . _clients [ _client ] [ ' IP ' ]
_port = self . _clients [ _client ] [ ' PORT ' ]
self . transport . write ( _packet , ( _ip , _port ) )
2016-07-24 10:42:42 -04:00
# KEEP THE FOLLOWING COMMENTED OUT UNLESS YOU'RE DEBUGGING DEEPLY!!!!
2016-11-23 16:01:05 -05:00
#self._logger.debug('(%s) TX Packet to %s on port %s: %s', self._clients[_client]['RADIO_ID'], self._clients[_client]['IP'], self._clients[_client]['PORT'], ahex(_packet))
2016-09-01 15:33:47 -04:00
2016-11-16 12:05:01 -05:00
def send_master ( self , _packet ) :
self . transport . write ( _packet , ( self . _config [ ' MASTER_IP ' ] , self . _config [ ' MASTER_PORT ' ] ) )
# KEEP THE FOLLOWING COMMENTED OUT UNLESS YOU'RE DEBUGGING DEEPLY!!!!
2016-11-23 16:01:05 -05:00
#self._logger.debug('(%s) TX Packet to %s:%s -- %s', self._system, self._config['MASTER_IP'], self._config['MASTER_PORT'], ahex(_packet))
2016-09-01 15:33:47 -04:00
2016-09-25 17:36:01 -04:00
def dmrd_received ( self , _radio_id , _rf_src , _dst_id , _seq , _slot , _call_type , _frame_type , _dtype_vseq , _stream_id , _data ) :
2016-07-31 10:41:21 -04:00
pass
2016-11-16 12:05:01 -05:00
2016-11-23 16:01:05 -05:00
def master_dereg ( self ) :
for _client in self . _clients :
self . send_client ( _client , ' MSTCL ' + _client )
2016-11-25 09:14:36 -05:00
self . _logger . info ( ' ( %s ) De-Registration sent to Client: %s ( %s ) ' , self . _system , self . _clients [ _client ] [ ' CALLSIGN ' ] , self . _clients [ _client ] [ ' RADIO_ID ' ] )
2016-11-23 16:01:05 -05:00
def client_dereg ( self ) :
self . send_master ( ' RPTCL ' + self . _config [ ' RADIO_ID ' ] )
self . _logger . info ( ' ( %s ) De-Registeration sent to Master: %s : %s ' , self . _system , self . _config [ ' MASTER_IP ' ] , self . _config [ ' MASTER_PORT ' ] )
2016-11-16 12:05:01 -05:00
# Aliased in __init__ to datagramReceived if system is a master
def master_datagramReceived ( self , _data , ( _host , _port ) ) :
2016-08-01 22:30:40 -04:00
# Keep This Line Commented Unless HEAVILY Debugging!
2016-11-23 16:01:05 -05:00
#self._logger.debug('(%s) RX packet from %s:%s -- %s', self._system, _host, _port, ahex(_data))
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
# Extract the command, which is various length, all but one 4 significant characters -- RPTCL
_command = _data [ : 4 ]
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
if _command == ' DMRD ' : # DMRData -- encapsulated DMR data frame
_radio_id = _data [ 11 : 15 ]
if _radio_id in self . _clients \
and self . _clients [ _radio_id ] [ ' CONNECTION ' ] == ' YES ' \
and self . _clients [ _radio_id ] [ ' IP ' ] == _host \
and self . _clients [ _radio_id ] [ ' PORT ' ] == _port :
2016-08-15 18:18:16 -04:00
_seq = _data [ 4 ]
2016-08-01 22:30:40 -04:00
_rf_src = _data [ 5 : 8 ]
_dst_id = _data [ 8 : 11 ]
2016-08-15 18:18:16 -04:00
_bits = int_id ( _data [ 15 ] )
2016-08-26 22:31:20 -04:00
_slot = 2 if ( _bits & 0x80 ) else 1
2016-08-26 23:43:51 -04:00
_call_type = ' unit ' if ( _bits & 0x40 ) else ' group '
2016-11-01 20:49:13 -04:00
_frame_type = ( _bits & 0x30 ) >> 4
2016-09-22 10:12:38 -04:00
_dtype_vseq = ( _bits & 0xF ) # data, 1=voice header, 2=voice terminator; voice, 0=burst A ... 5=burst F
2016-08-15 18:18:16 -04:00
_stream_id = _data [ 16 : 20 ]
2016-11-23 16:01:05 -05:00
#self._logger.debug('(%s) DMRD - Seqence: %s, RF Source: %s, Destination ID: %s', self._system, int_id(_seq), int_id(_rf_src), int_id(_dst_id))
2016-09-01 15:33:47 -04:00
2016-09-01 15:36:31 -04:00
# If AMBE audio exporting is configured...
2016-08-01 22:30:40 -04:00
if self . _config [ ' EXPORT_AMBE ' ] :
2016-11-21 20:54:23 -05:00
self . _ambe . parseAMBE ( self . _system , _data )
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
# The basic purpose of a master is to repeat to the clients
if self . _config [ ' REPEAT ' ] == True :
for _client in self . _clients :
if _client != _radio_id :
2016-08-25 21:44:15 -04:00
self . send_client ( _client , _data )
2016-11-23 16:01:05 -05:00
self . _logger . debug ( ' ( %s ) Packet on TS %s from %s ( %s ) for destination ID %s repeated to client: %s ( %s ) [Stream ID: %s ] ' , self . _system , _slot , self . _clients [ _radio_id ] [ ' CALLSIGN ' ] , int_id ( _radio_id ) , int_id ( _dst_id ) , self . _clients [ _client ] [ ' CALLSIGN ' ] , int_id ( _client ) , int_id ( _stream_id ) )
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
# Userland actions -- typically this is the function you subclass for an application
2016-09-22 10:12:38 -04:00
self . dmrd_received ( _radio_id , _rf_src , _dst_id , _seq , _slot , _call_type , _frame_type , _dtype_vseq , _stream_id , _data )
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
elif _command == ' RPTL ' : # RPTLogin -- a repeater wants to login
_radio_id = _data [ 4 : 8 ]
if _radio_id : # Future check here for valid Radio ID
self . _clients . update ( { _radio_id : { # Build the configuration data strcuture for the client
' CONNECTION ' : ' RPTL-RECEIVED ' ,
' PINGS_RECEIVED ' : 0 ,
' LAST_PING ' : time ( ) ,
' IP ' : _host ,
' PORT ' : _port ,
' SALT ' : randint ( 0 , 0xFFFFFFFF ) ,
2016-11-23 16:01:05 -05:00
' RADIO_ID ' : str ( int ( ahex ( _radio_id ) , 16 ) ) ,
2016-08-01 22:30:40 -04:00
' CALLSIGN ' : ' ' ,
' RX_FREQ ' : ' ' ,
' TX_FREQ ' : ' ' ,
' TX_POWER ' : ' ' ,
' COLORCODE ' : ' ' ,
' LATITUDE ' : ' ' ,
' LONGITUDE ' : ' ' ,
' HEIGHT ' : ' ' ,
' LOCATION ' : ' ' ,
' DESCRIPTION ' : ' ' ,
' SLOTS ' : ' ' ,
' URL ' : ' ' ,
' SOFTWARE_ID ' : ' ' ,
' PACKAGE_ID ' : ' ' ,
} } )
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Repeater Logging in with Radio ID: %s , %s : %s ' , self . _system , int_id ( _radio_id ) , _host , _port )
2016-08-01 22:30:40 -04:00
_salt_str = hex_str_4 ( self . _clients [ _radio_id ] [ ' SALT ' ] )
2016-08-25 21:44:15 -04:00
self . send_client ( _radio_id , ' RPTACK ' + _salt_str )
2016-08-01 22:30:40 -04:00
self . _clients [ _radio_id ] [ ' CONNECTION ' ] = ' CHALLENGE_SENT '
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Sent Challenge Response to %s for login: %s ' , self . _system , int_id ( _radio_id ) , self . _clients [ _radio_id ] [ ' SALT ' ] )
2016-08-01 22:30:40 -04:00
else :
self . transport . write ( ' MSTNAK ' + _radio_id , ( _host , _port ) )
2016-11-23 16:01:05 -05:00
self . _logger . warning ( ' ( %s ) Invalid Login from Radio ID: %s ' , self . _system , int_id ( _radio_id ) )
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
elif _command == ' RPTK ' : # Repeater has answered our login challenge
_radio_id = _data [ 4 : 8 ]
if _radio_id in self . _clients \
and self . _clients [ _radio_id ] [ ' CONNECTION ' ] == ' CHALLENGE_SENT ' \
and self . _clients [ _radio_id ] [ ' IP ' ] == _host \
and self . _clients [ _radio_id ] [ ' PORT ' ] == _port :
_this_client = self . _clients [ _radio_id ]
_this_client [ ' LAST_PING ' ] = time ( )
_sent_hash = _data [ 8 : ]
_salt_str = hex_str_4 ( _this_client [ ' SALT ' ] )
2016-11-25 09:14:36 -05:00
_calc_hash = bhex ( sha256 ( _salt_str + self . _config [ ' PASSPHRASE ' ] ) . hexdigest ( ) )
2016-08-01 22:30:40 -04:00
if _sent_hash == _calc_hash :
_this_client [ ' CONNECTION ' ] = ' WAITING_CONFIG '
2016-08-25 21:44:15 -04:00
self . send_client ( _radio_id , ' RPTACK ' + _radio_id )
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Client %s has completed the login exchange successfully ' , self . _system , _this_client [ ' RADIO_ID ' ] )
2016-08-01 22:30:40 -04:00
else :
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Client %s has FAILED the login exchange successfully ' , self . _system , _this_client [ ' RADIO_ID ' ] )
2016-08-01 22:30:40 -04:00
self . transport . write ( ' MSTNAK ' + _radio_id , ( _host , _port ) )
del self . _clients [ _radio_id ]
else :
self . transport . write ( ' MSTNAK ' + _radio_id , ( _host , _port ) )
2016-11-23 16:01:05 -05:00
self . _logger . warning ( ' ( %s ) Login challenge from Radio ID that has not logged in: %s ' , self . _system , int_id ( _radio_id ) )
2016-09-01 15:33:47 -04:00
2016-08-01 22:30:40 -04:00
elif _command == ' RPTC ' : # Repeater is sending it's configuraiton OR disconnecting
if _data [ : 5 ] == ' RPTCL ' : # Disconnect command
_radio_id = _data [ 5 : 9 ]
2016-07-27 22:56:33 -04:00
if _radio_id in self . _clients \
and self . _clients [ _radio_id ] [ ' CONNECTION ' ] == ' YES ' \
and self . _clients [ _radio_id ] [ ' IP ' ] == _host \
and self . _clients [ _radio_id ] [ ' PORT ' ] == _port :
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Client is closing down: %s ( %s ) ' , self . _system , self . _clients [ _radio_id ] [ ' CALLSIGN ' ] , int_id ( _radio_id ) )
2016-07-26 21:51:04 -04:00
self . transport . write ( ' MSTNAK ' + _radio_id , ( _host , _port ) )
2016-08-01 22:30:40 -04:00
del self . _clients [ _radio_id ]
2016-11-18 09:27:57 -05:00
2016-09-01 15:36:31 -04:00
else :
2016-08-01 22:30:40 -04:00
_radio_id = _data [ 4 : 8 ] # Configure Command
2016-07-26 21:51:04 -04:00
if _radio_id in self . _clients \
2016-08-01 22:30:40 -04:00
and self . _clients [ _radio_id ] [ ' CONNECTION ' ] == ' WAITING_CONFIG ' \
2016-07-26 21:51:04 -04:00
and self . _clients [ _radio_id ] [ ' IP ' ] == _host \
and self . _clients [ _radio_id ] [ ' PORT ' ] == _port :
_this_client = self . _clients [ _radio_id ]
2016-08-01 22:30:40 -04:00
_this_client [ ' CONNECTION ' ] = ' YES '
2016-07-26 21:51:04 -04:00
_this_client [ ' LAST_PING ' ] = time ( )
2016-08-01 22:30:40 -04:00
_this_client [ ' CALLSIGN ' ] = _data [ 8 : 16 ]
_this_client [ ' RX_FREQ ' ] = _data [ 16 : 25 ]
_this_client [ ' TX_FREQ ' ] = _data [ 25 : 34 ]
_this_client [ ' TX_POWER ' ] = _data [ 34 : 36 ]
_this_client [ ' COLORCODE ' ] = _data [ 36 : 38 ]
2016-11-18 09:27:57 -05:00
_this_client [ ' LATITUDE ' ] = _data [ 38 : 46 ]
_this_client [ ' LONGITUDE ' ] = _data [ 46 : 55 ]
_this_client [ ' HEIGHT ' ] = _data [ 55 : 58 ]
_this_client [ ' LOCATION ' ] = _data [ 58 : 78 ]
_this_client [ ' DESCRIPTION ' ] = _data [ 78 : 97 ]
_this_client [ ' SLOTS ' ] = _data [ 97 : 98 ]
_this_client [ ' URL ' ] = _data [ 98 : 222 ]
_this_client [ ' SOFTWARE_ID ' ] = _data [ 222 : 262 ]
_this_client [ ' PACKAGE_ID ' ] = _data [ 262 : 302 ]
2016-08-01 22:30:40 -04:00
2016-08-25 21:44:15 -04:00
self . send_client ( _radio_id , ' RPTACK ' + _radio_id )
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Client %s ( %s ) has sent repeater configuration ' , self . _system , _this_client [ ' CALLSIGN ' ] , _this_client [ ' RADIO_ID ' ] )
2016-07-24 13:07:43 -04:00
else :
2016-07-25 19:55:23 -04:00
self . transport . write ( ' MSTNAK ' + _radio_id , ( _host , _port ) )
2016-11-23 16:01:05 -05:00
self . _logger . warning ( ' ( %s ) Client info from Radio ID that has not logged in: %s ' , self . _system , int_id ( _radio_id ) )
2016-08-01 22:30:40 -04:00
elif _command == ' RPTP ' : # RPTPing -- client is pinging us
_radio_id = _data [ 7 : 11 ]
if _radio_id in self . _clients \
and self . _clients [ _radio_id ] [ ' CONNECTION ' ] == " YES " \
and self . _clients [ _radio_id ] [ ' IP ' ] == _host \
and self . _clients [ _radio_id ] [ ' PORT ' ] == _port :
self . _clients [ _radio_id ] [ ' LAST_PING ' ] = time ( )
2016-08-25 21:44:15 -04:00
self . send_client ( _radio_id , ' MSTPONG ' + _radio_id )
2016-11-23 16:01:05 -05:00
self . _logger . debug ( ' ( %s ) Received and answered RPTPING from client %s ( %s ) ' , self . _system , self . _clients [ _radio_id ] [ ' CALLSIGN ' ] , int_id ( _radio_id ) )
2016-08-01 22:30:40 -04:00
else :
self . transport . write ( ' MSTNAK ' + _radio_id , ( _host , _port ) )
2016-11-23 16:01:05 -05:00
self . _logger . warning ( ' ( %s ) Client info from Radio ID that has not logged in: %s ' , self . _system , int_id ( _radio_id ) )
2016-09-01 15:33:47 -04:00
2016-07-21 13:41:36 -04:00
else :
2016-11-23 16:01:05 -05:00
self . _logger . error ( ' ( %s ) Unrecognized command from: %s . Packet: %s ' , self . _system , int_id ( _radio_id ) , ahex ( _data ) )
2016-11-16 12:05:01 -05:00
# Aliased in __init__ to datagramReceived if system is a client
def client_datagramReceived ( self , _data , ( _host , _port ) ) :
2016-08-01 22:30:40 -04:00
# Keep This Line Commented Unless HEAVILY Debugging!
2016-11-23 16:01:05 -05:00
# self._logger.debug('(%s) RX packet from %s:%s -- %s', self._system, _host, _port, ahex(_data))
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
# Validate that we receveived this packet from the master - security check!
if self . _config [ ' MASTER_IP ' ] == _host and self . _config [ ' MASTER_PORT ' ] == _port :
# Extract the command, which is various length, but only 4 significant characters
2016-09-01 15:36:31 -04:00
_command = _data [ : 4 ]
2016-07-26 21:51:04 -04:00
if _command == ' DMRD ' : # DMRData -- encapsulated DMR data frame
2016-07-27 22:56:33 -04:00
_radio_id = _data [ 11 : 15 ]
if _radio_id == self . _config [ ' RADIO_ID ' ] : # Validate the source and intended target
2016-07-31 10:41:21 -04:00
_seq = _data [ 4 : 5 ]
_rf_src = _data [ 5 : 8 ]
_dst_id = _data [ 8 : 11 ]
2016-08-26 22:31:20 -04:00
_bits = int_id ( _data [ 15 ] )
_slot = 2 if ( _bits & 0x80 ) else 1
2016-08-26 23:43:51 -04:00
_call_type = ' unit ' if ( _bits & 0x40 ) else ' group '
2016-11-16 12:05:01 -05:00
_frame_type = ( _bits & 0x30 ) >> 4
2016-09-22 10:12:38 -04:00
_dtype_vseq = ( _bits & 0xF ) # data, 1=voice header, 2=voice terminator; voice, 0=burst A ... 5=burst F
2016-08-26 22:31:20 -04:00
_stream_id = _data [ 16 : 20 ]
2016-11-23 16:01:05 -05:00
#self._logger.debug('(%s) DMRD - Seqence: %s, RF Source: %s, Destination ID: %s', self._system, int_id(_seq), int_id(_rf_src), int_id(_dst_id))
2016-09-01 15:33:47 -04:00
2016-07-31 10:41:21 -04:00
# If AMBE audio exporting is configured...
if self . _config [ ' EXPORT_AMBE ' ] :
2016-11-21 20:54:23 -05:00
self . _ambe . parseAMBE ( self . _system , _data )
2016-09-01 15:33:47 -04:00
2016-07-31 10:41:21 -04:00
# Userland actions -- typically this is the function you subclass for an application
2016-09-22 10:12:38 -04:00
self . dmrd_received ( _radio_id , _rf_src , _dst_id , _seq , _slot , _call_type , _frame_type , _dtype_vseq , _stream_id , _data )
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
elif _command == ' MSTN ' : # Actually MSTNAK -- a NACK from the master
2016-07-27 15:42:49 -04:00
_radio_id = _data [ 4 : 8 ]
2016-07-27 22:56:33 -04:00
if _radio_id == self . _config [ ' RADIO_ID ' ] : # Validate the source and intended target
2016-11-23 16:01:05 -05:00
self . _logger . warning ( ' ( %s ) MSTNAK Received ' , self . _system )
2016-07-26 21:51:04 -04:00
self . _stats [ ' CONNECTION ' ] = ' NO ' # Disconnect ourselves and re-register
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
elif _command == ' RPTA ' : # Actually RPTACK -- an ACK from the master
# Depending on the state, an RPTACK means different things, in each clause, we check and/or set the state
if self . _stats [ ' CONNECTION ' ] == ' RTPL_SENT ' : # If we've sent a login request...
_login_int32 = _data [ 6 : 10 ]
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Repeater Login ACK Received with 32bit ID: %s ' , self . _system , int_id ( _login_int32 ) )
2016-07-26 21:51:04 -04:00
_pass_hash = sha256 ( _login_int32 + self . _config [ ' PASSPHRASE ' ] ) . hexdigest ( )
2016-11-23 16:01:05 -05:00
_pass_hash = bhex ( _pass_hash )
2016-08-25 21:44:15 -04:00
self . send_master ( ' RPTK ' + self . _config [ ' RADIO_ID ' ] + _pass_hash )
2016-07-26 21:51:04 -04:00
self . _stats [ ' CONNECTION ' ] = ' AUTHENTICATED '
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
elif self . _stats [ ' CONNECTION ' ] == ' AUTHENTICATED ' : # If we've sent the login challenge...
if _data [ 6 : 10 ] == self . _config [ ' RADIO_ID ' ] :
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Repeater Authentication Accepted ' , self . _system )
2016-07-26 21:51:04 -04:00
_config_packet = self . _config [ ' RADIO_ID ' ] + \
self . _config [ ' CALLSIGN ' ] + \
self . _config [ ' RX_FREQ ' ] + \
self . _config [ ' TX_FREQ ' ] + \
self . _config [ ' TX_POWER ' ] + \
self . _config [ ' COLORCODE ' ] + \
self . _config [ ' LATITUDE ' ] + \
self . _config [ ' LONGITUDE ' ] + \
self . _config [ ' HEIGHT ' ] + \
self . _config [ ' LOCATION ' ] + \
self . _config [ ' DESCRIPTION ' ] + \
self . _config [ ' SLOTS ' ] + \
self . _config [ ' URL ' ] + \
self . _config [ ' SOFTWARE_ID ' ] + \
self . _config [ ' PACKAGE_ID ' ]
2016-09-01 15:33:47 -04:00
2016-08-25 21:44:15 -04:00
self . send_master ( ' RPTC ' + _config_packet )
2016-07-26 21:51:04 -04:00
self . _stats [ ' CONNECTION ' ] = ' CONFIG-SENT '
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Repeater Configuration Sent ' , self . _system )
2016-07-26 21:51:04 -04:00
else :
self . _stats [ ' CONNECTION ' ] = ' NO '
2016-11-23 16:01:05 -05:00
self . _logger . error ( ' ( %s ) Master ACK Contained wrong ID - Connection Reset ' , self . _system )
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
elif self . _stats [ ' CONNECTION ' ] == ' CONFIG-SENT ' : # If we've sent out configuration to the master
if _data [ 6 : 10 ] == self . _config [ ' RADIO_ID ' ] :
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Repeater Configuration Accepted ' , self . _system )
2016-07-26 21:51:04 -04:00
self . _stats [ ' CONNECTION ' ] = ' YES '
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) Connection to Master Completed ' , self . _system )
2016-07-26 21:51:04 -04:00
else :
self . _stats [ ' CONNECTION ' ] = ' NO '
2016-11-23 16:01:05 -05:00
self . _logger . error ( ' ( %s ) Master ACK Contained wrong ID - Connection Reset ' , self . _system )
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
elif _command == ' MSTP ' : # Actually MSTPONG -- a reply to RPTPING (send by client)
if _data [ 7 : 11 ] == self . _config [ ' RADIO_ID ' ] :
self . _stats [ ' PINGS_ACKD ' ] + = 1
2016-11-23 16:01:05 -05:00
self . _logger . debug ( ' ( %s ) MSTPONG Received. Pongs Since Connected: %s ' , self . _system , self . _stats [ ' PINGS_ACKD ' ] )
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
elif _command == ' MSTC ' : # Actually MSTCL -- notify us the master is closing down
if _data [ 5 : 9 ] == self . _config [ ' RADIO_ID ' ] :
self . _stats [ ' CONNECTION ' ] = ' NO '
2016-11-23 16:01:05 -05:00
self . _logger . info ( ' ( %s ) MSTCL Recieved ' , self . _system )
2016-09-01 15:33:47 -04:00
2016-07-26 21:51:04 -04:00
else :
2016-11-23 16:01:05 -05:00
self . _logger . error ( ' ( %s ) Received an invalid command in packet: %s ' , self . _system , ahex ( _data ) )
2016-07-20 17:16:27 -04:00
2016-07-20 22:25:47 -04:00
2016-07-20 17:16:27 -04:00
#************************************************
# MAIN PROGRAM LOOP STARTS HERE
#************************************************
if __name__ == ' __main__ ' :
2016-11-23 16:01:05 -05:00
# Python modules we need
import argparse
import sys
import os
import signal
#
# Parse the command line and make adjustments
#
# 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 hblink.cfg) ' )
parser . add_argument ( ' -l ' , ' --logging ' , action = ' store ' , dest = ' LOG_LEVEL ' , help = ' Override config file logging level. ' )
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__ ) ) + ' /hblink.cfg '
#
# Build the configuration file
#
# Call the external routine to build the configuration dictionary
CONFIG = hb_config . build_config ( cli_args . CONFIG_FILE )
#
# Start the system logger
#
# Call the external routing to start the system logger
if cli_args . LOG_LEVEL :
CONFIG [ ' LOGGER ' ] [ ' LOG_LEVEL ' ] = cli_args . LOG_LEVEL
logger = hb_log . config_logging ( CONFIG [ ' LOGGER ' ] )
logger . debug ( ' Logging system started, anything from here on gets logged ' )
#
# Set up the signal handler
#
def sig_handler ( _signal , _frame ) :
logger . info ( ' SHUTDOWN: HBLINK IS TERMINATING WITH SIGNAL %s ' , str ( _signal ) )
hblink_handler ( _signal , _frame , logger )
logger . info ( ' SHUTDOWN: ALL SYSTEM HANDLERS EXECUTED - STOPPING REACTOR ' )
reactor . stop ( )
# Set signal handers so that we can gracefully exit if need be
for sig in [ signal . SIGTERM , signal . SIGINT ] :
signal . signal ( sig , sig_handler )
#
# START HB_ROUTER
#
2016-09-01 15:33:47 -04:00
2016-08-25 21:44:15 -04:00
# HBlink instance creation
2016-11-23 16:01:05 -05:00
logger . info ( ' HBlink \' HBlink.py \' (c) 2016 N0MJS & the K0USY Group - SYSTEM STARTING... ' )
2016-08-25 21:44:15 -04:00
for system in CONFIG [ ' SYSTEMS ' ] :
if CONFIG [ ' SYSTEMS ' ] [ system ] [ ' ENABLED ' ] :
2016-11-23 16:01:05 -05:00
systems [ system ] = HBSYSTEM ( system , CONFIG , logger )
2016-08-25 21:44:15 -04:00
reactor . listenUDP ( CONFIG [ ' SYSTEMS ' ] [ system ] [ ' PORT ' ] , systems [ system ] , interface = CONFIG [ ' SYSTEMS ' ] [ system ] [ ' IP ' ] )
logger . debug ( ' %s instance created: %s , %s ' , CONFIG [ ' SYSTEMS ' ] [ system ] [ ' MODE ' ] , system , systems [ system ] )
2016-09-01 15:33:47 -04:00
2016-09-01 11:32:30 -04:00
reactor . run ( )