diff --git a/src/target/trx_toolkit/ctrl_if.py b/src/target/trx_toolkit/ctrl_if.py index b533746b0..45bfa2b92 100644 --- a/src/target/trx_toolkit/ctrl_if.py +++ b/src/target/trx_toolkit/ctrl_if.py @@ -27,7 +27,11 @@ import logging as log from udp_link import UDPLink class CTRLInterface(UDPLink): - def handle_rx(self, data, remote): + def handle_rx(self): + # Read data from socket + data, remote = self.sock.recvfrom(128) + data = data.decode() + if not self.verify_req(data): log.error("Wrong data on CTRL interface") return diff --git a/src/target/trx_toolkit/fake_trx.py b/src/target/trx_toolkit/fake_trx.py index d73b56626..95261dfe0 100755 --- a/src/target/trx_toolkit/fake_trx.py +++ b/src/target/trx_toolkit/fake_trx.py @@ -112,13 +112,11 @@ class Application(ApplicationBase): # CTRL commands from BTS if self.bts_ctrl.sock in r_event: - data, addr = self.bts_ctrl.sock.recvfrom(128) - self.bts_ctrl.handle_rx(data.decode(), addr) + self.bts_ctrl.handle_rx() # CTRL commands from BB if self.bb_ctrl.sock in r_event: - data, addr = self.bb_ctrl.sock.recvfrom(128) - self.bb_ctrl.handle_rx(data.decode(), addr) + self.bb_ctrl.handle_rx() def shutdown(self): log.info("Shutting down...")