trx_toolkit/fake_trx.py: handle multiple transceivers
Since Ice44e2b22566b3652ef6d43896055963b13ab185 is merged, the class hierarchy has become much more flexible, so it's possible to create multiple Transceiver / FakeTRX instances and distribute bursts using a single instance of BurstForwarder. This change introduces a new command line option, that can be used to specify additional transceivers. Please note that fake_trx.py still initializes a pair of BTS and BB transceivers by default. The new option has the following format: --trx REMOTE_ADDR:BASE_PORT[/TRX_NUM] Some examples for IPv4 and IPv6: --trx 127.0.0.1:5703 --trx [2001:0db8:85a3:0000:0000:8a2e:0370:7334]:6703 If optional TRX_NUM > 0 is specified, e.g.: --trx 127.0.0.1:5700/1 then this transceiver is considered as a child of another one. See I7e97b7f32dde7ab74779133e9d7504f1d0fce60c for details. Change-Id: Ibff31fb3a958a714c828d0dea7e87d47f778fd80
This commit is contained in:
Vadim Yanitskiy
parent
42bce2bd5b
commit
786e97a0dc
|
|
@ -30,11 +30,13 @@ import argparse
|
|||
import random
|
||||
import select
|
||||
import sys
|
||||
import re
|
||||
|
||||
from app_common import ApplicationBase
|
||||
from burst_fwd import BurstForwarder
|
||||
from transceiver import Transceiver
|
||||
from clck_gen import CLCKGen
|
||||
from trx_list import TRXList
|
||||
from fake_pm import FakePM
|
||||
|
||||
class FakeTRX(Transceiver):
|
||||
|
|
@ -270,6 +272,34 @@ class Application(ApplicationBase):
|
|||
# Configure logging
|
||||
self.app_init_logging(self.argv)
|
||||
|
||||
# List of all transceivers
|
||||
self.trx_list = TRXList()
|
||||
|
||||
def append_trx(self, remote_addr, base_port):
|
||||
trx = FakeTRX(self.argv.trx_bind_addr, remote_addr, base_port,
|
||||
clck_gen = self.clck_gen, pwr_meas = self.fake_pm)
|
||||
self.trx_list.add_trx(trx)
|
||||
|
||||
def append_child_trx(self, remote_addr, base_port, child_idx):
|
||||
# Index 0 corresponds to the first transceiver
|
||||
if child_idx is 0:
|
||||
self.append_trx(remote_addr, base_port)
|
||||
return
|
||||
|
||||
# Find 'parent' transceiver for a new child
|
||||
trx_parent = self.trx_list.find_trx(remote_addr, base_port)
|
||||
if trx_parent is None:
|
||||
raise IndexError("Couldn't find parent transceiver "
|
||||
"for '%s:%d/%d'" % (remote_addr, base_port, child_idx))
|
||||
|
||||
# Allocate a new child
|
||||
trx_child = FakeTRX(self.argv.trx_bind_addr, remote_addr, base_port,
|
||||
child_idx = child_idx, pwr_meas = self.fake_pm)
|
||||
self.trx_list.add_trx(trx_child)
|
||||
|
||||
# Link a new 'child' with its 'parent'
|
||||
trx_parent.child_trx_list.add_trx(trx_child)
|
||||
|
||||
def run(self):
|
||||
# Init shared clock generator
|
||||
self.clck_gen = CLCKGen([])
|
||||
|
|
@ -280,49 +310,44 @@ class Application(ApplicationBase):
|
|||
self.fake_pm = FakePM(-120, -105, -75, -50)
|
||||
|
||||
# Init TRX instance for BTS
|
||||
self.bts_trx = FakeTRX(self.argv.trx_bind_addr,
|
||||
self.argv.bts_addr, self.argv.bts_base_port,
|
||||
clck_gen = self.clck_gen)
|
||||
self.append_trx(self.argv.bts_addr, self.argv.bts_base_port)
|
||||
|
||||
# Init TRX instance for BB
|
||||
self.bb_trx = FakeTRX(self.argv.trx_bind_addr,
|
||||
self.argv.bb_addr, self.argv.bb_base_port,
|
||||
pwr_meas = self.fake_pm)
|
||||
self.append_trx(self.argv.bb_addr, self.argv.bb_base_port)
|
||||
|
||||
# Additional transceivers (optional)
|
||||
if self.argv.trx_list is not None:
|
||||
for trx_def in self.argv.trx_list:
|
||||
(addr, port, idx) = trx_def
|
||||
self.append_child_trx(addr, port, idx)
|
||||
|
||||
# Burst forwarding between transceivers
|
||||
self.burst_fwd = BurstForwarder()
|
||||
self.burst_fwd.add_trx(self.bts_trx)
|
||||
self.burst_fwd.add_trx(self.bb_trx)
|
||||
self.burst_fwd = BurstForwarder(self.trx_list)
|
||||
|
||||
# Compose list of to be monitored sockets
|
||||
self.sock_list = []
|
||||
for trx in self.trx_list:
|
||||
self.sock_list.append(trx.ctrl_if.sock)
|
||||
self.sock_list.append(trx.data_if.sock)
|
||||
|
||||
log.info("Init complete")
|
||||
|
||||
# Enter main loop
|
||||
while True:
|
||||
socks = [self.bts_trx.ctrl_if.sock, self.bb_trx.ctrl_if.sock,
|
||||
self.bts_trx.data_if.sock, self.bb_trx.data_if.sock]
|
||||
|
||||
# Wait until we get any data on any socket
|
||||
r_event, w_event, x_event = select.select(socks, [], [])
|
||||
r_event, _, _ = select.select(self.sock_list, [], [])
|
||||
|
||||
# Downlink: BTS -> BB
|
||||
if self.bts_trx.data_if.sock in r_event:
|
||||
msg = self.bts_trx.recv_data_msg()
|
||||
if msg is not None:
|
||||
self.burst_fwd.forward_msg(self.bts_trx, msg)
|
||||
# Iterate over all transceivers
|
||||
for trx in self.trx_list:
|
||||
# DATA interface
|
||||
if trx.data_if.sock in r_event:
|
||||
msg = trx.recv_data_msg()
|
||||
if msg is not None:
|
||||
self.burst_fwd.forward_msg(trx, msg)
|
||||
|
||||
# Uplink: BB -> BTS
|
||||
if self.bb_trx.data_if.sock in r_event:
|
||||
msg = self.bb_trx.recv_data_msg()
|
||||
if msg is not None:
|
||||
self.burst_fwd.forward_msg(self.bb_trx, msg)
|
||||
|
||||
# CTRL commands from BTS
|
||||
if self.bts_trx.ctrl_if.sock in r_event:
|
||||
self.bts_trx.ctrl_if.handle_rx()
|
||||
|
||||
# CTRL commands from BB
|
||||
if self.bb_trx.ctrl_if.sock in r_event:
|
||||
self.bb_trx.ctrl_if.handle_rx()
|
||||
# CTRL interface
|
||||
if trx.ctrl_if.sock in r_event:
|
||||
trx.ctrl_if.handle_rx()
|
||||
|
||||
def shutdown(self):
|
||||
log.info("Shutting down...")
|
||||
|
|
@ -330,6 +355,25 @@ class Application(ApplicationBase):
|
|||
# Stop clock generator
|
||||
self.clck_gen.stop()
|
||||
|
||||
# Parses a TRX definition of the following
|
||||
# format: REMOTE_ADDR:BIND_PORT[/TRX_NUM]
|
||||
# e.g. [2001:0db8:85a3:0000:0000:8a2e:0370:7334]:5700/5
|
||||
# e.g. 127.0.0.1:5700 or 127.0.0.1:5700/1
|
||||
@staticmethod
|
||||
def trx_def(val):
|
||||
try:
|
||||
result = re.match("(.+):([0-9]+)(\/[0-9]+)?", val)
|
||||
(addr, port, idx) = result.groups()
|
||||
except:
|
||||
raise argparse.ArgumentTypeError("Invalid TRX definition: %s" % val)
|
||||
|
||||
if idx is not None:
|
||||
idx = int(idx[1:])
|
||||
else:
|
||||
idx = 0
|
||||
|
||||
return (addr, int(port), idx)
|
||||
|
||||
def parse_argv(self):
|
||||
parser = argparse.ArgumentParser(prog = "fake_trx",
|
||||
description = "Virtual Um-interface (fake transceiver)")
|
||||
|
|
@ -354,6 +398,12 @@ class Application(ApplicationBase):
|
|||
dest = "bb_base_port", type = int, default = 6700,
|
||||
help = "Set BB base port number (default %(default)s)")
|
||||
|
||||
mtrx_group = parser.add_argument_group("Additional transceivers")
|
||||
mtrx_group.add_argument("--trx",
|
||||
metavar = "REMOTE_ADDR:BASE_PORT[/TRX_NUM]",
|
||||
dest = "trx_list", type = self.trx_def, action = "append",
|
||||
help = "Add a transceiver for BTS or MS (e.g. 127.0.0.1:5703)")
|
||||
|
||||
argv = parser.parse_args()
|
||||
|
||||
# Make sure there is no overlap between ports
|
||||
|
|
|
|||
Loading…
Reference in New Issue