Fix error while writing command to shell on windows 10 (#24)

- Parse response to int only if the response is digit and not a random string
SadParad1se · Jun 28, 2024 · 726f013 · 726f013
1 parent e1ea0f9
commit 726f013
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 14 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "snek-sploit"
-version = "0.8.1"
+version = "0.8.2"
 description = "Python RPC client for Metasploit Framework"
 authors = [
     "Jiří Rája <[email protected]>"

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:
         """