telnet: fix prompt handling for config commands

telekom · Feb 7, 2024 · fb37eba · fb37eba
1 parent d63dceb
commit fb37eba
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mb_netmgmt/__init__.py b/mb_netmgmt/__init__.py
@@ -18,7 +18,7 @@
 # along with mb-netmgmt. If not, see <https://www.gnu.org/licenses/
 
 """Network Management Protocols for Mountebank"""
-__version__ = "0.0.80"
+__version__ = "0.0.81"
 
 import os
 import subprocess

diff --git a/mb_netmgmt/telnet.py b/mb_netmgmt/telnet.py
@@ -54,7 +54,7 @@ def handle_command(self, command, request_id):
         return self.handle_request({"command": command.decode()}, request_id)
 
     def send_upstream(self, request, request_id):
-        self.telnet.execute(request["command"])
+        self.telnet.send(request["command"])
 
     def read_request(self):
         return self.rfile.readline(), None
@@ -65,7 +65,8 @@ def respond(self, response, request_id):
             self.stopped = True
 
     def read_proxy_response(self):
-        return {"response": self.telnet.response}
+        _, match = self.telnet.expect_prompt()
+        return {"response": match.string}
 
     def handle_username_prompt(self):
         username_prompt = b"Username: "