fake_trx/ctrl_if.py: allow adding custom params to response

Change-Id: I551bb425c25a5c978801d9e1e033b4ba352e259f
This commit is contained in:
Vadim Yanitskiy 2017-07-14 05:49:48 +07:00
parent ef31f1bff2
commit e39bb0f0d1
1 changed files with 11 additions and 2 deletions

View File

@ -30,7 +30,11 @@ class CTRLInterface(UDPLink):
if self.verify_req(data):
request = self.prepare_req(data)
rc = self.parse_cmd(request)
self.send_response(request, rc)
if type(rc) is tuple:
self.send_response(request, rc[0], rc[1])
else:
self.send_response(request, rc)
else:
print("[!] Wrong data on CTRL interface")
@ -62,9 +66,14 @@ class CTRLInterface(UDPLink):
return True
def send_response(self, request, response_code):
def send_response(self, request, response_code, params = None):
# Include status code, for example ["TXTUNE", "0", "941600"]
request.insert(1, str(response_code))
# Optionally append command specific parameters
if params is not None:
request += params
# Add the response signature, and join back to string
response = "RSP " + " ".join(request) + "\0"
# Now we have something like "RSP TXTUNE 0 941600"