nyanloutre
/
tplink-smartplug
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #1 from LeeXiaolan/bind-receive-in-advance 

Binding reply receive socket before send request.
This commit is contained in:
softScheck GmbH 2016-09-26 11:33:23 +02:00 committed by GitHub
parent c59d92c202 c5eaa41b64
commit e342b7d688
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tddp-client/tddp-client.py
View File

@ -181,6 +181,10 @@ tddp_packet = "".join([tddp_ver, tddp_type, tddp_code, tddp_reply, tddp_length,
tddp_digest = hashlib.md5(binascii.unhexlify(tddp_packet)).hexdigest() tddp_digest = hashlib.md5(binascii.unhexlify(tddp_packet)).hexdigest()
tddp_packet = tddp_packet[:24] + tddp_digest + tddp_packet[56:] tddp_packet = tddp_packet[:24] + tddp_digest + tddp_packet[56:]
# Binding receive socket in advance in case reply comes fast.
sock_receive = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock_receive.bind(('', port_receive))
# Send a request # Send a request
sock_send = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock_send = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock_send.sendto(binascii.unhexlify(tddp_packet), (ip, port_send)) sock_send.sendto(binascii.unhexlify(tddp_packet), (ip, port_send))
@ -191,8 +195,6 @@ print "Request Data:\tVersion", t[0:2], "Type", t[2:4], "Status", t[6:8], "Lengt
sock_send.close() sock_send.close()
# Receive the reply # Receive the reply
sock_receive = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock_receive.bind(('', port_receive))
response, addr = sock_receive.recvfrom(1024) response, addr = sock_receive.recvfrom(1024)
r = response.encode('hex') r = response.encode('hex')
if args.verbose: if args.verbose: