mirror
/
micropython-lib
kopia lustrzana https://github.com/micropython/micropython-lib
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność

umqtt.simple: Allow to pass arbitrary params to ussl.wrap_socket().

pull/97/head
Paul Sokolovsky 2016-08-24 01:04:32 +03:00
rodzic c9a9b48ebe
commit 98c6fb9024
1 zmienionych plików z 3 dodań i 2 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

5
umqtt.simple/umqtt/simple.py
Wyświetl plik

@ -8,13 +8,14 @@ class MQTTException(Exception):
class MQTTClient: class MQTTClient:
def __init__(self, client_id, server, port=0, user=None, password=None, keepalive=0, def __init__(self, client_id, server, port=0, user=None, password=None, keepalive=0,
ssl=False): ssl=False, ssl_params={}):
if port == 0: if port == 0:
port = 8883 if ssl else 1883 port = 8883 if ssl else 1883
self.client_id = client_id self.client_id = client_id
self.sock = None self.sock = None
self.addr = socket.getaddrinfo(server, port)[0][-1] self.addr = socket.getaddrinfo(server, port)[0][-1]
self.ssl = ssl self.ssl = ssl
self.ssl_params = ssl_params
self.pid = 0 self.pid = 0
self.cb = None self.cb = None
self.user = user self.user = user
@ -43,7 +44,7 @@ class MQTTClient:
self.sock.connect(self.addr) self.sock.connect(self.addr)
if self.ssl: if self.ssl:
import ussl import ussl
self.sock = ussl.wrap_socket(self.sock) self.sock = ussl.wrap_socket(self.sock, **self.ssl_params)
msg = bytearray(b"\x10\0\0\x04MQTT\x04\x02\0\0") msg = bytearray(b"\x10\0\0\x04MQTT\x04\x02\0\0")
msg[1] = 10 + 2 + len(self.client_id) msg[1] = 10 + 2 + len(self.client_id)
msg[9] = clean_session << 1 msg[9] = clean_session << 1