summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVadim Yanitskiy <axilirator@gmail.com>2018-01-27 20:55:57 +0700
committerVadim Yanitskiy <axilirator@gmail.com>2018-01-29 04:19:01 +0700
commitb84699096a3fc40717b792ddf51b678ee7503d91 (patch)
treede5a9549308d138c08577a052fddd9e8bb3d6934
parentd9553faf62a6e5eec339dd0c324b1a31a71350ab (diff)
fake_trx/burst_gen.py: check argv separately
-rwxr-xr-xsrc/target/fake_trx/burst_gen.py19
1 files changed, 12 insertions, 7 deletions
diff --git a/src/target/fake_trx/burst_gen.py b/src/target/fake_trx/burst_gen.py
index 68ba0ece..66d8e89e 100755
--- a/src/target/fake_trx/burst_gen.py
+++ b/src/target/fake_trx/burst_gen.py
@@ -53,6 +53,7 @@ class Application:
def __init__(self):
self.print_copyright()
self.parse_argv()
+ self.check_argv()
# Set up signal handlers
signal.signal(signal.SIGINT, self.sig_handler)
@@ -65,9 +66,6 @@ class Application:
elif self.conn_mode == "L1":
self.data_if = DATAInterface(self.remote_addr,
self.base_port + 102, self.base_port + 2)
- else:
- self.print_help("[!] Unknown connection type")
- sys.exit(2)
# Init random burst generator
self.gen = RandBurstGen()
@@ -89,10 +87,6 @@ class Application:
buf = self.gen.gen_sb()
elif self.burst_type == "AB":
buf = self.gen.gen_ab()
- else:
- self.print_help("[!] Unknown burst type")
- self.shutdown()
- sys.exit(2)
print("[i] Sending %d/%d %s burst (fn=%u) to %s..."
% (i + 1, self.burst_count, self.burst_type,
@@ -178,6 +172,17 @@ class Application:
elif o in ("-l", "--power-level"):
self.pwr = abs(int(v))
+ def check_argv(self):
+ # Check connection mode
+ if self.conn_mode not in ("TRX", "L1"):
+ self.print_help("[!] Unknown connection type")
+ sys.exit(2)
+
+ # Check connection mode
+ if self.burst_type not in ("NB", "FB", "SB", "AB"):
+ self.print_help("[!] Unknown burst type")
+ sys.exit(2)
+
def shutdown(self):
self.data_if.shutdown()