Fix SSL and set_socket

Author: justmobilize

adafruit_azureiot/device_registration.py

@@ -112,7 +112,7 @@ def _connect_to_mqtt(self) -> None:
             " - device_registration :: connect :: created mqtt client. connecting.."
         )
         while not self._auth_response_received:
-            self._mqtt.loop()
+            self._mqtt.loop(2)
 
         self._logger.info(
             " - device_registration :: connect :: on_connect must be fired. Connected ?"
@@ -139,7 +139,7 @@ def _start_registration(self) -> None:
         while self._operation_id is None and retry < 10:
             time.sleep(1)
             retry += 1
-            self._mqtt.loop()
+            self._mqtt.loop(2)
 
         if self._operation_id is None:
             raise DeviceRegistrationError(
@@ -159,7 +159,7 @@ def _wait_for_operation(self) -> None:
         while self._hostname is None and retry < 10:
             time.sleep(1)
             retry += 1
-            self._mqtt.loop()
+            self._mqtt.loop(2)
 
         if self._hostname is None:
             raise DeviceRegistrationError(
@@ -194,15 +194,15 @@ def register_device(self, expiry: int) -> str:
             "&skn=registration"
         )
 
-        MQTT.set_socket(self._socket, self._iface)
-
         self._mqtt = MQTT.MQTT(
             broker=constants.DPS_END_POINT,
+            port=8883,
             username=username,
             password=auth_string,
-            port=8883,
-            keep_alive=120,
             client_id=self._device_id,
+            is_ssl=True,
+            keep_alive=120,
+            socket_pool=self._socket,
             ssl_context=ssl.create_default_context(),
         )
 

adafruit_azureiot/iot_mqtt.py

@@ -120,8 +120,6 @@ def _gen_sas_token(self) -> str:
         )
 
     def _create_mqtt_client(self) -> None:
-        MQTT.set_socket(self._socket, self._iface)
-
         log_text = (
             f"- iot_mqtt :: _on_connect :: username = {self._username}, password = "
             + f"{self._passwd}"
@@ -135,11 +133,13 @@ def _create_mqtt_client(self) -> None:
 
         self._mqtts = MQTT.MQTT(
             broker=self._hostname,
+            port=8883,
             username=self._username,
             password=self._passwd,
-            port=8883,
-            keep_alive=120,
             client_id=self._device_id,
+            is_ssl=True,
+            keep_alive=120,
+            socket_pool=self._socket,
             ssl_context=ssl.create_default_context(),
         )
 
@@ -457,7 +457,7 @@ def loop(self) -> None:
         if not self.is_connected():
             return
 
-        self._mqtts.loop()
+        self._mqtts.loop(2)
         gc.collect()
 
     def send_device_to_cloud_message(