diff --git a/python/trx/ctrl_if.py b/python/trx/ctrl_if.py index 7dee0f87..ae5cf050 100644 --- a/python/trx/ctrl_if.py +++ b/python/trx/ctrl_if.py @@ -25,15 +25,15 @@ from grgsm.trx import udp_link class ctrl_if(udp_link): - def handle_rx(self, data): + def handle_rx(self, data, remote): if self.verify_req(data): request = self.prepare_req(data) rc = self.parse_cmd(request) if type(rc) is tuple: - self.send_response(request, rc[0], rc[1]) + self.send_response(request, remote, rc[0], rc[1]) else: - self.send_response(request, rc) + self.send_response(request, remote, rc) else: print("[!] Wrong data on CTRL interface") @@ -65,7 +65,7 @@ def verify_cmd(self, request, cmd, argc): return True - def send_response(self, request, response_code, params = None): + def send_response(self, request, remote, response_code, params = None): # Include status code, for example ["TXTUNE", "0", "941600"] request.insert(1, str(response_code)) @@ -76,7 +76,7 @@ def send_response(self, request, response_code, params = None): # Add the response signature, and join back to string response = "RSP " + " ".join(request) + "\0" # Now we have something like "RSP TXTUNE 0 941600" - self.send(response) + self.send(response, remote) def parse_cmd(self, request): raise NotImplementedError diff --git a/python/trx/udp_link.py b/python/trx/udp_link.py index 1fae8b43..d96a6aa5 100644 --- a/python/trx/udp_link.py +++ b/python/trx/udp_link.py @@ -45,13 +45,16 @@ def loop(self): # Check for incoming data if self.sock in r_event: data, addr = self.sock.recvfrom(128) - self.handle_rx(data.decode()) + self.handle_rx(data.decode(), addr) - def send(self, data): + def send(self, data, remote = None): if type(data) not in [bytearray, bytes]: data = data.encode() - self.sock.sendto(data, (self.remote_addr, self.remote_port)) + if remote is None: + remote = (self.remote_addr, self.remote_port) - def handle_rx(self, data): + self.sock.sendto(data, remote) + + def handle_rx(self, data, remote): raise NotImplementedError