diff options
author | Vadim Yanitskiy <axilirator@gmail.com> | 2018-01-27 20:55:57 +0700 |
---|---|---|
committer | Vadim Yanitskiy <axilirator@gmail.com> | 2018-01-29 04:19:01 +0700 |
commit | b84699096a3fc40717b792ddf51b678ee7503d91 (patch) | |
tree | de5a9549308d138c08577a052fddd9e8bb3d6934 /src/target | |
parent | d9553faf62a6e5eec339dd0c324b1a31a71350ab (diff) |
fake_trx/burst_gen.py: check argv separately
Change-Id: I35b5475d3b6df6dc92a1981c693afb63df866c87
Diffstat (limited to 'src/target')
-rwxr-xr-x | src/target/fake_trx/burst_gen.py | 19 |
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() |