Fixed edge case when msf returns random string instead of int

SadParad1se · Jun 28, 2024 · 296f53b · 296f53b
1 parent e1ea0f9
commit 296f53b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/snek_sploit/lib/rpc/sessions.py b/snek_sploit/lib/rpc/sessions.py
@@ -218,7 +218,7 @@ def shell_write(self, session_id: int, data: str) -> int:
 
         response = self._context.call(self.SHELL_WRITE, [session_id, data])
 
-        return int(response[constants.B_WRITE_COUNT])
+        return int(response[constants.B_WRITE_COUNT]) if response.isdigit() else -1
 
     def shell_upgrade(self, session_id: int, local_host: str, local_port: int) -> bool:
         """
@@ -379,7 +379,7 @@ def ring_put(self, session_id: int, data: str) -> int:
         """
         response = self._context.call(self.RING_PUT, [session_id, data])
 
-        return int(response[constants.B_WRITE_COUNT])
+        return int(response[constants.B_WRITE_COUNT]) if response.isdigit() else -1
 
     def ring_last(self, session_id: int) -> int:
         """