DMRlink/playback.py

172 lines
7.6 KiB
Python
Raw Normal View History

#!/usr/bin/env python
#
2016-11-23 08:50:56 -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
###############################################################################
# This is a sample application that "records" and replays transmissions for testing.
from __future__ import print_function
from twisted.internet import reactor
2016-12-18 22:51:13 -05:00
from binascii import b2a_hex as ahex
import sys, time
2017-05-16 14:50:56 -04:00
from dmrlink import IPSC, mk_ipsc_systems, systems, reportFactory, build_aliases, config_reports
2016-12-18 22:51:13 -05:00
from dmr_utils.utils import int_id, hex_str_3
2016-11-23 08:50:56 -05:00
__author__ = 'Cortney T. Buffington, N0MJS'
__copyright__ = 'Copyright (c) 2014 Cortney T. Buffington, N0MJS and the K0USY Group'
__credits__ = 'Adam Fast, KC0YLK; Dave Kierzkowski, KD8EYF'
__license__ = 'GNU GPLv3'
__maintainer__ = 'Cort Buffington, N0MJS'
__email__ = 'n0mjs@me.com'
2014-04-25 16:34:53 -04:00
try:
from playback_config import *
except ImportError:
sys.exit('Configuration file not found or invalid')
HEX_TGID = hex_str_3(TGID)
HEX_SUB = hex_str_3(SUB)
BOGUS_SUB = '\xFF\xFF\xFF'
class playbackIPSC(IPSC):
2017-05-16 14:50:56 -04:00
def __init__(self, _name, _config, _logger, _report):
IPSC.__init__(self, _name, _config, _logger, _report)
self.CALL_DATA = []
2016-12-18 22:51:13 -05:00
if GROUP_SRC_SUB:
self._logger.info('Playback: USING SUBSCRIBER ID: %s FOR GROUP REPEAT', GROUP_SRC_SUB)
self.GROUP_SRC_SUB = hex_str_3(GROUP_SRC_SUB)
if GROUP_REPEAT:
self._logger.info('Playback: GROUP REPEAT ENABLED')
if PRIVATE_REPEAT:
self._logger.info('Playback: PRIVATE REPEAT ENABLED')
#************************************************
# CALLBACK FUNCTIONS FOR USER PACKET TYPES
#************************************************
#
2014-08-17 10:35:50 -04:00
if GROUP_REPEAT:
2016-12-18 22:51:13 -05:00
def group_voice(self, _src_sub, _dst_sub, _ts, _end, _peerid, _data):
2014-08-17 10:35:50 -04:00
if HEX_TGID == _dst_sub and _ts in GROUP_TS:
if not _end:
if not self.CALL_DATA:
2016-12-18 22:51:13 -05:00
self._logger.info('(%s) Receiving transmission to be played back from subscriber: %s', self._system, int_id(_src_sub))
2014-08-17 12:45:37 -04:00
_tmp_data = _data
2016-12-18 22:51:13 -05:00
#_tmp_data = dmr_nat(_data, _src_sub, self._config['LOCAL']['RADIO_ID'])
2014-08-17 12:45:37 -04:00
self.CALL_DATA.append(_tmp_data)
if _end:
self.CALL_DATA.append(_data)
time.sleep(2)
2016-12-18 22:51:13 -05:00
self._logger.info('(%s) Playing back transmission from subscriber: %s', self._system, int_id(_src_sub))
2014-08-17 12:45:37 -04:00
for i in self.CALL_DATA:
_tmp_data = i
2016-12-18 22:51:13 -05:00
_tmp_data = _tmp_data.replace(_peerid, self._config['LOCAL']['RADIO_ID'])
if GROUP_SRC_SUB:
2016-12-18 22:51:13 -05:00
_tmp_data = _tmp_data.replace(_src_sub, self.GROUP_SRC_SUB)
2014-08-17 12:45:37 -04:00
# Send the packet to all peers in the target IPSC
self.send_to_ipsc(_tmp_data)
2014-08-17 12:45:37 -04:00
time.sleep(0.06)
self.CALL_DATA = []
2014-08-17 10:35:50 -04:00
if PRIVATE_REPEAT:
2016-12-18 22:51:13 -05:00
def private_voice(self, _src_sub, _dst_sub, _ts, _end, _peerid, _data):
2014-08-17 10:35:50 -04:00
if HEX_SUB == _dst_sub and _ts in PRIVATE_TS:
if not _end:
if not self.CALL_DATA:
2016-12-18 22:51:13 -05:00
self._logger.info('(%s) Receiving transmission to be played back from subscriber: %s, to subscriber: %s', self._system, int_id(_src_sub), int_id(_dst_sub))
2014-08-17 10:35:50 -04:00
_tmp_data = _data
self.CALL_DATA.append(_tmp_data)
if _end:
self.CALL_DATA.append(_data)
time.sleep(1)
2016-12-18 22:51:13 -05:00
self._logger.info('(%s) Playing back transmission from subscriber: %s, to subscriber %s', self._system, int_id(_src_sub), int_id(_dst_sub))
2014-08-17 10:35:50 -04:00
_orig_src = _src_sub
_orig_dst = _dst_sub
for i in self.CALL_DATA:
_tmp_data = i
2016-12-18 22:51:13 -05:00
_tmp_data = _tmp_data.replace(_peerid, self._config['LOCAL']['RADIO_ID'])
2014-08-17 10:35:50 -04:00
_tmp_data = _tmp_data.replace(_dst_sub, BOGUS_SUB)
_tmp_data = _tmp_data.replace(_src_sub, _orig_dst)
_tmp_data = _tmp_data.replace(BOGUS_SUB, _orig_src)
# Send the packet to all peers in the target IPSC
self.send_to_ipsc(_tmp_data)
2014-08-17 10:35:50 -04:00
time.sleep(0.06)
self.CALL_DATA = []
2016-12-18 22:51:13 -05:00
if __name__ == '__main__':
2016-12-18 22:51:13 -05:00
import argparse
import sys
2017-05-16 14:50:56 -04:00
import os
2016-12-18 22:51:13 -05:00
import signal
2017-05-16 17:55:08 -04:00
from ipsc.dmrlink_config import build_config
from ipsc.dmrlink_log import config_logging
2016-12-18 22:51:13 -05:00
# 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='CFG_FILE', help='/full/path/to/config.file (usually dmrlink.cfg)')
parser.add_argument('-ll', '--log_level', action='store', dest='LOG_LEVEL', help='Override config file logging level.')
parser.add_argument('-lh', '--log_handle', action='store', dest='LOG_HANDLERS', help='Override config file logging handler.')
2016-12-18 22:51:13 -05:00
cli_args = parser.parse_args()
if not cli_args.CFG_FILE:
cli_args.CFG_FILE = os.path.dirname(os.path.abspath(__file__))+'/dmrlink.cfg'
2017-05-16 14:50:56 -04:00
2016-12-18 22:51:13 -05:00
# Call the external routine to build the configuration dictionary
2017-05-16 14:50:56 -04:00
CONFIG = build_config(cli_args.CFG_FILE)
2016-12-18 22:51:13 -05:00
# Call the external routing to start the system logger
if cli_args.LOG_LEVEL:
CONFIG['LOGGER']['LOG_LEVEL'] = cli_args.LOG_LEVEL
if cli_args.LOG_HANDLERS:
CONFIG['LOGGER']['LOG_HANDLERS'] = cli_args.LOG_HANDLERS
2017-05-16 14:50:56 -04:00
logger = config_logging(CONFIG['LOGGER'])
logger.info('DMRlink \'dmrlink.py\' (c) 2013 - 2015 N0MJS & the K0USY Group - SYSTEM STARTING...')
2016-12-18 22:51:13 -05:00
2017-05-16 14:50:56 -04:00
# Set signal handers so that we can gracefully exit if need be
2016-12-18 22:51:13 -05:00
def sig_handler(_signal, _frame):
logger.info('*** DMRLINK IS TERMINATING WITH SIGNAL %s ***', str(_signal))
for system in systems:
2017-05-16 14:50:56 -04:00
systems[system].de_register_self()
2016-12-18 22:51:13 -05:00
reactor.stop()
2017-05-16 14:50:56 -04:00
2016-12-18 22:51:13 -05:00
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGQUIT]:
signal.signal(sig, sig_handler)
2017-05-16 14:50:56 -04:00
# INITIALIZE THE REPORTING LOOP
report_server = config_reports(CONFIG, logger, reportFactory)
2016-12-18 22:51:13 -05:00
2017-05-16 14:50:56 -04:00
# Build ID Aliases
peer_ids, subscriber_ids, talkgroup_ids, local_ids = build_aliases(CONFIG, logger)
# INITIALIZE AN IPSC OBJECT (SELF SUSTAINING) FOR EACH CONFIGRUED IPSC
systems = mk_ipsc_systems(CONFIG, logger, systems, playbackIPSC, report_server)
# INITIALIZATION COMPLETE -- START THE REACTOR
2014-08-17 11:23:10 -04:00
reactor.run()