mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make communication thread safe

Browse source
This commit is contained in:
Daniel Høyer Iversen 2016-12-22 09:51:38 +01:00 committed by GitHub
parent 83f1c3fc93
commit 1296c3da48
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
broadlink/__init__.py
View file

@ -5,6 +5,7 @@ from Crypto.Cipher import AES
import time import time
import random import random
import socket import socket
import threading
def gendevice(devtype, host, mac): def gendevice(devtype, host, mac):
if devtype == 0: # SP1 if devtype == 0: # SP1
@ -138,6 +139,7 @@ class device:
self.cs.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1) self.cs.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
self.cs.bind(('',0)) self.cs.bind(('',0))
self.type = "Unknown" self.type = "Unknown"
self.lock = threading.Lock()
def auth(self): def auth(self):
payload = bytearray(0x50) payload = bytearray(0x50)
@ -228,16 +230,17 @@ class device:
packet[0x21] = checksum >> 8 packet[0x21] = checksum >> 8
starttime = time.time() starttime = time.time()
while True: with self.lock:
try: while True:
self.cs.sendto(packet, self.host) try:
self.cs.settimeout(1) self.cs.sendto(packet, self.host)
response = self.cs.recvfrom(1024) self.cs.settimeout(1)
break response = self.cs.recvfrom(1024)
except socket.timeout: break
if (time.time() - starttime) < self.timeout: except socket.timeout:
pass if (time.time() - starttime) < self.timeout:
raise pass
raise
return bytearray(response[0]) return bytearray(response[0])