230 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			230 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python
 | 
						|
#
 | 
						|
###############################################################################
 | 
						|
#   Copyright (C) 2016-2019  Cortney T. Buffington, N0MJS <n0mjs@me.com> (and Mike Zingman N4IRR)
 | 
						|
#
 | 
						|
#   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
 | 
						|
###############################################################################
 | 
						|
 | 
						|
 | 
						|
# Python modules we need
 | 
						|
import sys
 | 
						|
from bitarray import bitarray
 | 
						|
from time import time, sleep
 | 
						|
from importlib import import_module
 | 
						|
 | 
						|
# Twisted is pretty important, so I keep it separate
 | 
						|
from twisted.internet.protocol import Factory, Protocol
 | 
						|
from twisted.protocols.basic import NetstringReceiver
 | 
						|
from twisted.internet import reactor, task
 | 
						|
 | 
						|
# Things we import from the main hblink module
 | 
						|
from hblink import HBSYSTEM, systems, hblink_handler, reportFactory, REPORT_OPCODES, config_reports, mk_aliases
 | 
						|
from dmr_utils3.utils import bytes_3, int_id, get_alias
 | 
						|
from dmr_utils3 import decode, bptc, const
 | 
						|
import config
 | 
						|
import log
 | 
						|
import const
 | 
						|
 | 
						|
# The module needs logging logging, but handlers, etc. are controlled by the parent
 | 
						|
import logging
 | 
						|
logger = logging.getLogger(__name__)
 | 
						|
 | 
						|
 | 
						|
# Does anybody read this stuff? There's a PEP somewhere that says I should do this.
 | 
						|
__author__     = 'Cortney T. Buffington, N0MJS and Mike Zingman, N4IRR'
 | 
						|
__copyright__  = 'Copyright (c) 2016-2019 Cortney T. Buffington, N0MJS and the K0USY Group'
 | 
						|
__license__    = 'GNU GPLv3'
 | 
						|
__maintainer__ = 'Cort Buffington, N0MJS'
 | 
						|
__email__      = 'n0mjs@me.com'
 | 
						|
__status__     = 'pre-alpha'
 | 
						|
 | 
						|
# Module gobal varaibles
 | 
						|
 | 
						|
class playback(HBSYSTEM):
 | 
						|
 | 
						|
    def __init__(self, _name, _config, _report):
 | 
						|
        HBSYSTEM.__init__(self, _name, _config, _report)
 | 
						|
 | 
						|
        # Status information for the system, TS1 & TS2
 | 
						|
        # 1 & 2 are "timeslot"
 | 
						|
        # In TX_EMB_LC, 2-5 are burst B-E
 | 
						|
        self.STATUS = {
 | 
						|
            1: {
 | 
						|
                'RX_START':     time(),
 | 
						|
                'RX_SEQ':       '\x00',
 | 
						|
                'RX_RFS':       '\x00',
 | 
						|
                'TX_RFS':       '\x00',
 | 
						|
                'RX_STREAM_ID': '\x00',
 | 
						|
                'TX_STREAM_ID': '\x00',
 | 
						|
                'RX_TGID':      '\x00\x00\x00',
 | 
						|
                'TX_TGID':      '\x00\x00\x00',
 | 
						|
                'RX_TIME':      time(),
 | 
						|
                'TX_TIME':      time(),
 | 
						|
                'RX_TYPE':      const.HBPF_SLT_VTERM,
 | 
						|
                'RX_LC':        '\x00',
 | 
						|
                'TX_H_LC':      '\x00',
 | 
						|
                'TX_T_LC':      '\x00',
 | 
						|
                'TX_EMB_LC': {
 | 
						|
                    1: '\x00',
 | 
						|
                    2: '\x00',
 | 
						|
                    3: '\x00',
 | 
						|
                    4: '\x00',
 | 
						|
                }
 | 
						|
                },
 | 
						|
            2: {
 | 
						|
                'RX_START':     time(),
 | 
						|
                'RX_SEQ':       '\x00',
 | 
						|
                'RX_RFS':       '\x00',
 | 
						|
                'TX_RFS':       '\x00',
 | 
						|
                'RX_STREAM_ID': '\x00',
 | 
						|
                'TX_STREAM_ID': '\x00',
 | 
						|
                'RX_TGID':      '\x00\x00\x00',
 | 
						|
                'TX_TGID':      '\x00\x00\x00',
 | 
						|
                'RX_TIME':      time(),
 | 
						|
                'TX_TIME':      time(),
 | 
						|
                'RX_TYPE':      const.HBPF_SLT_VTERM,
 | 
						|
                'RX_LC':        '\x00',
 | 
						|
                'TX_H_LC':      '\x00',
 | 
						|
                'TX_T_LC':      '\x00',
 | 
						|
                'TX_EMB_LC': {
 | 
						|
                    1: '\x00',
 | 
						|
                    2: '\x00',
 | 
						|
                    3: '\x00',
 | 
						|
                    4: '\x00',
 | 
						|
                }
 | 
						|
            }
 | 
						|
        }
 | 
						|
        self.CALL_DATA = []
 | 
						|
 | 
						|
    def dmrd_received(self, _peer_id, _rf_src, _dst_id, _seq, _slot, _call_type, _frame_type, _dtype_vseq, _stream_id, _data):
 | 
						|
        pkt_time = time()
 | 
						|
        dmrpkt = _data[20:53]
 | 
						|
        _bits = _data[15]
 | 
						|
        if _call_type == 'group':
 | 
						|
            
 | 
						|
            # Is this is a new call stream?
 | 
						|
            if (_stream_id != self.STATUS[_slot]['RX_STREAM_ID']):
 | 
						|
                self.STATUS['RX_START'] = pkt_time
 | 
						|
                logger.info('(%s) *START RECORDING* STREAM ID: %s SUB: %s (%s) REPEATER: %s (%s) TGID %s (%s), TS %s', \
 | 
						|
                                  self._system, int_id(_stream_id), get_alias(_rf_src, subscriber_ids), int_id(_rf_src), get_alias(_peer_id, peer_ids), int_id(_peer_id), get_alias(_dst_id, talkgroup_ids), int_id(_dst_id), _slot)
 | 
						|
                self.CALL_DATA.append(_data)
 | 
						|
                self.STATUS[_slot]['RX_STREAM_ID'] = _stream_id
 | 
						|
                return
 | 
						|
 | 
						|
            # Final actions - Is this a voice terminator?
 | 
						|
            if (_frame_type == const.HBPF_DATA_SYNC) and (_dtype_vseq == const.HBPF_SLT_VTERM) and (self.STATUS[_slot]['RX_TYPE'] != const.HBPF_SLT_VTERM) and (self.CALL_DATA):
 | 
						|
                call_duration = pkt_time - self.STATUS['RX_START']
 | 
						|
                self.CALL_DATA.append(_data)
 | 
						|
                logger.info('(%s) *END   RECORDING* STREAM ID: %s', self._system, int_id(_stream_id))
 | 
						|
                sleep(2)
 | 
						|
                logger.info('(%s) *START  PLAYBACK* STREAM ID: %s SUB: %s (%s) REPEATER: %s (%s) TGID %s (%s), TS %s, Duration: %s', \
 | 
						|
                                  self._system, int_id(_stream_id), get_alias(_rf_src, subscriber_ids), int_id(_rf_src), get_alias(_peer_id, peer_ids), int_id(_peer_id), get_alias(_dst_id, talkgroup_ids), int_id(_dst_id), _slot, call_duration)
 | 
						|
                for i in self.CALL_DATA:
 | 
						|
                    self.send_system(i)
 | 
						|
                    #print(i)
 | 
						|
                    sleep(0.06)
 | 
						|
                self.CALL_DATA = []
 | 
						|
                logger.info('(%s) *END    PLAYBACK* STREAM ID: %s', self._system, int_id(_stream_id))
 | 
						|
 | 
						|
            else:
 | 
						|
                if self.CALL_DATA:
 | 
						|
                    self.CALL_DATA.append(_data)
 | 
						|
 | 
						|
 | 
						|
            # Mark status variables for use later
 | 
						|
            self.STATUS[_slot]['RX_RFS']       = _rf_src
 | 
						|
            self.STATUS[_slot]['RX_TYPE']      = _dtype_vseq
 | 
						|
            self.STATUS[_slot]['RX_TGID']      = _dst_id
 | 
						|
            self.STATUS[_slot]['RX_TIME']      = pkt_time
 | 
						|
            self.STATUS[_slot]['RX_STREAM_ID'] = _stream_id
 | 
						|
 | 
						|
 | 
						|
#************************************************
 | 
						|
#      MAIN PROGRAM LOOP STARTS HERE
 | 
						|
#************************************************
 | 
						|
 | 
						|
if __name__ == '__main__':
 | 
						|
    
 | 
						|
    import argparse
 | 
						|
    import sys
 | 
						|
    import os
 | 
						|
    import signal
 | 
						|
    from dmr_utils3.utils import try_download, mk_id_dict
 | 
						|
    
 | 
						|
    # 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 default (top of file)
 | 
						|
    if not cli_args.CONFIG_FILE:
 | 
						|
        cli_args.CONFIG_FILE = os.path.dirname(os.path.abspath(__file__))+'/hblink.cfg'
 | 
						|
 | 
						|
    # Call the external routine to build the configuration dictionary
 | 
						|
    CONFIG = config.build_config(cli_args.CONFIG_FILE)
 | 
						|
    
 | 
						|
    # Start the system logger
 | 
						|
    if cli_args.LOG_LEVEL:
 | 
						|
        CONFIG['LOGGER']['LOG_LEVEL'] = cli_args.LOG_LEVEL
 | 
						|
    logger = log.config_logging(CONFIG['LOGGER'])
 | 
						|
    logger.info('\n\nCopyright (c) 2013, 2014, 2015, 2016, 2018, 2019\n\tThe Founding Members of the K0USY Group. All rights reserved.\n')
 | 
						|
    logger.debug('Logging system started, anything from here on gets logged')
 | 
						|
    
 | 
						|
    # Set up the signal handler
 | 
						|
    def sig_handler(_signal, _frame):
 | 
						|
        logger.info('SHUTDOWN: HBROUTER IS TERMINATING WITH SIGNAL %s', str(_signal))
 | 
						|
        hblink_handler(_signal, _frame)
 | 
						|
        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)
 | 
						|
    
 | 
						|
    # ID ALIAS CREATION
 | 
						|
    # Download
 | 
						|
    if CONFIG['ALIASES']['TRY_DOWNLOAD'] == True:
 | 
						|
        # Try updating peer aliases file
 | 
						|
        result = try_download(CONFIG['ALIASES']['PATH'], CONFIG['ALIASES']['PEER_FILE'], CONFIG['ALIASES']['PEER_URL'], CONFIG['ALIASES']['STALE_TIME'])
 | 
						|
        logger.info(result)
 | 
						|
        # Try updating subscriber aliases file
 | 
						|
        result = try_download(CONFIG['ALIASES']['PATH'], CONFIG['ALIASES']['SUBSCRIBER_FILE'], CONFIG['ALIASES']['SUBSCRIBER_URL'], CONFIG['ALIASES']['STALE_TIME'])
 | 
						|
        logger.info(result)
 | 
						|
        
 | 
						|
    # Create the name-number mapping dictionaries
 | 
						|
    peer_ids, subscriber_ids, talkgroup_ids = mk_aliases(CONFIG)
 | 
						|
        
 | 
						|
    # INITIALIZE THE REPORTING LOOP
 | 
						|
    report_server = config_reports(CONFIG, reportFactory)    
 | 
						|
    
 | 
						|
    # HBlink instance creation
 | 
						|
    logger.info('HBlink \'playback.py\' (c) 2017-2019 Cort Buffington, N0MJS & Mike Zingman, N4IRR -- SYSTEM STARTING...')
 | 
						|
    for system in CONFIG['SYSTEMS']:
 | 
						|
        if CONFIG['SYSTEMS'][system]['ENABLED']:
 | 
						|
            if CONFIG['SYSTEMS'][system]['MODE'] == 'OPENBRIDGE':
 | 
						|
                logger.critical('%s FATAL: Instance is mode \'OPENBRIDGE\', \n\t\t...Which would be tragic for playback, since it carries multiple call\n\t\tstreams simultaneously. playback.py onlyl works with MMDVM-based systems', system)
 | 
						|
                sys.exit('playback.py cannot function with systems that are not MMDVM devices. System {} is configured as an OPENBRIDGE'.format(system))
 | 
						|
            else:
 | 
						|
                systems[system] = playback(system, CONFIG, report_server)
 | 
						|
            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])
 | 
						|
 | 
						|
    reactor.run()
 |