Update references to Crypto with Cryptodome in discovery and smartconfig components

pull/482/head
Colin Kuebler 2020-01-02 16:29:28 -05:00
parent 61927f766b
commit a9abd469ea
2 changed files with 4 additions and 4 deletions

View File

@ -9,7 +9,7 @@ multicast strategy reverse engineered by kueblc
from crc import crc_32
from Crypto.Cipher import AES
from Cryptodome.Cipher import AES
pad = lambda data, block_size : data + ('\0' * ( (block_size - len(data)) % block_size ) )
aes = AES.new( b'a3c6794oiu876t54', AES.MODE_ECB )
encrypt = lambda data : aes.encrypt( pad(data, 16).encode() )

View File

@ -9,11 +9,11 @@ Discover Tuya devices on the LAN via UDP broadcast
import asyncio
import json
from Crypto.Cipher import AES
from Cryptodome.Cipher import AES
pad = lambda s: s + (16 - len(s) % 16) * chr(16 - len(s) % 16)
unpad = lambda s: s[:-ord(s[len(s) - 1:])]
encrypt = lambda msg, key: AES.new(key, AES.MODE_ECB).encrypt(pad(msg))
decrypt = lambda msg, key: unpad(AES.new(key, AES.MODE_ECB).decrypt(msg))
encrypt = lambda msg, key: AES.new(key.encode(), AES.MODE_ECB).encrypt(pad(msg).encode())
decrypt = lambda msg, key: unpad(AES.new(key.encode(), AES.MODE_ECB).decrypt(msg.encode()))
from hashlib import md5
udpkey = md5(b"yGAdlopoPVldABfn").digest()