🔐 SSL termination handling (#37)

docs/public/static/changelog/iotFtpsClient.md

@@ -4,6 +4,10 @@ title: iot-ftps-client Changelog
 
 # Release History for iot-ftps-client
 
+### 1.1.0 (09/07/2022)
+
+- Override `storbinary` function to prevent implicit SSL shutdown
+
 ### 1.0.1 (08/30/2022)
 
 - Bug fix for uploading byte streams from within a context

iot-ftps-client/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Release History
 
+## 1.1.0 (09/07/2022)
+
+- Override `storbinary` function to prevent implicit SSL shutdown
+
 ## 1.0.1 (08/30/2022)
 
 - Bug fix for uploading byte streams from within a context

iot-ftps-client/iot/ftps/client/_client.py

@@ -24,6 +24,25 @@ def sock(self, value):
             value = self.context.wrap_socket(value)
         self._sock = value
 
+    def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
+        """
+        override storbinary to prevent SSL shutdown
+        (https://github.com/python/cpython/blob/main/Lib/ftplib.py)
+        """
+        self.voidcmd("TYPE I")
+        with self.transfercmd(cmd, rest) as conn:
+            while 1:
+                buf = fp.read(blocksize)
+                if not buf:
+                    break
+                conn.sendall(buf)
+                if callback:
+                    callback(buf)
+            # THIS IS WHERE WE OVERRIDE
+            # if _SSLSocket is not None and isinstance(conn, _SSLSocket):
+            #   conn.unwrap()
+        return self.voidresp()
+
 
 class IoTFTPSClient:
     """iot ftps client"""

iot-ftps-client/iot/ftps/client/_version.py

@@ -1,3 +1,3 @@
-VERSION = "1.0.1"
+VERSION = "1.1.0"
 
 __version__ = VERSION