Fixing telnet read until (threat9#466)

devkw · Jun 14, 2018 · 6d6d172 · 6d6d172
1 parent f61d9eb
commit 6d6d172
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/routersploit/core/telnet/telnet_client.py b/routersploit/core/telnet/telnet_client.py
@@ -81,20 +81,26 @@ def telnet_interactive(self, telnet):
 
     def telnet_read_until(self, telnet_client, data):
         if telnet_client:
-            if type(data) is str:
-                data = bytes(data, "utf-8")
+            try:
+                if type(data) is str:
+                    data = bytes(data, "utf-8")
 
-            response = telnet_client.read_until(data, 5)
-            return str(response, "utf-8")
+                response = telnet_client.read_until(data, 5)
+                return str(response, "utf-8")
+            except Exception:
+                pass
 
         return None
 
     def telnet_write(self, telnet_client, data):
         if telnet_client:
-            if type(data) is str:
-                data = bytes(data, "utf-8")
+            try:
+                if type(data) is str:
+                    data = bytes(data, "utf-8")
 
-            return telnet_client.write(data, 5)
+                return telnet_client.write(data, 5)
+            except Exception:
+                pass
 
         return None