From d17ecc9806691cd022e00c02a3af23f4f859feed Mon Sep 17 00:00:00 2001
From: root <email@example.com>
Date: Tue, 29 Nov 2022 18:54:09 +0000
Subject: [PATCH] Fix sftp logic

---
 sshim_patch.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/sshim_patch.py b/sshim_patch.py
index 3613a44..dc1c3f8 100644
--- a/sshim_patch.py
+++ b/sshim_patch.py
@@ -26,6 +26,14 @@ def check_channel_shell_request(self, channel):
     return True
 
 
+def check_channel_subsystem_request(self, channel, name):
+    if name == 'sftp':
+        self.runner.set_sftp_channel(channel)
+        return True
+    else:
+        return False
+
+
 def check_auth_none(self, username):
     if username == os.environ["SSH_USERNAME"]:
         return paramiko.AUTH_PARTIALLY_SUCCESSFUL
@@ -53,6 +61,7 @@ class Runner(threading.Thread):
         self.daemon = True
         self.client = client
         self.transport = transport
+        # self.transport.set_subsystem_handler('sftp', handler=paramiko.SFTPServer)
         self.shell_channel = None
         self.sftp_channel = None
 
@@ -103,12 +112,14 @@ class Runner(threading.Thread):
         self.shell_channel = channel
         self.shell_channel.settimeout(None)
 
-    def set_scp_channel(self, channel):
+    def set_sftp_channel(self, channel):
         self.sftp_channel = channel
         self.sftp_channel.settimeout(None)
 
 
+Handler.check_channel_request = check_channel_request
 Handler.check_channel_shell_request = check_channel_shell_request
+Handler.check_channel_subsystem_request = check_channel_subsystem_request
 Handler.check_auth_none = check_auth_none
 Handler.check_auth_password = check_auth_password
 Handler.check_auth_publickey = check_auth_publickey