This commit is contained in:
David Konsumer 2017-09-05 02:31:56 +00:00 committed by GitHub
commit 234cb00ee5
2 changed files with 61 additions and 71 deletions

View file

@ -2,9 +2,7 @@
-- For decrypting local network traffic between TP-Link
-- Smart Home Devices and the Kasa Smart Home App
--
-- Install under:
-- (Windows) %APPDATA%\Wireshark\plugins\
-- (Linux, Mac) $HOME/.wireshark/plugins
-- Install in the location listed in About Wireshark/Folders/Personal Plugins
--
-- by Lubomir Stroetmann
-- Copyright 2016 softScheck GmbH
@ -24,65 +22,45 @@
--
-- Create TP-Link Smart Home protocol and its fields
p_tplink = Proto ("TPLink-SmartHome","TP-Link Smart Home Protocol")
hs1x0_proto_TCP = Proto ("TPLink-SmartHome-TCP", "TP-Link Smart Home Protocol (TCP")
hs1x0_proto_UDP = Proto ("TPLink-SmartHome-UDP", "TP-Link Smart Home Protocol (UDP)")
-- Dissector function
function p_tplink.dissector (buf, pkt, root)
-- Validate packet length
if buf:len() == 0 then return end
pkt.cols.protocol = p_tplink.name
-- Decode data
local ascii = ""
local hex = ""
-- Skip first 4 bytes (header)
start = 4
endPosition = buf:len() - 1
-- Decryption key is -85 (256-85=171)
local key = 171
-- Decrypt string Autokey XOR to ByteArray
function tpdecode(buf, start)
local key = 171
local size = buf:len()-1
local decoded = ""
for i=start,size do
local c = buf(i,1):uint()
decoded = decoded .. string.format("%x", bit.bxor(c,key))
key = c
end
return ByteArray.new(decoded)
end
-- Decrypt Autokey XOR
-- Save results as ascii and hex
for index = start, endPosition do
local c = buf(index,1):uint()
-- XOR first byte with key
d = bit32.bxor(c,key)
-- Use byte as next key
key = c
hex = hex .. string.format("%x", d)
-- Convert to printable characters
if d >= 0x20 and d <= 0x7E then
ascii = ascii .. string.format("%c", d)
else
-- Use dot for non-printable bytes
ascii = ascii .. "."
end
end
-- Create subtree
subtree = root:add(p_tplink, buf(0))
-- Add data to subtree
subtree:add(ascii)
-- Description of payload
function hs1x0_proto_TCP.dissector (buf, pkt, root)
pkt.cols.protocol = "TPLink-SmartHome (TCP)"
local subtree = root:add(hs1x0_proto_TCP, buf() ,"TPLink-SmartHome")
local decoded = tpdecode(buf, 4)
subtree:add(decoded:raw())
subtree:append_text(" (decrypted)")
-- Call JSON Dissector with decrypted data
local b = ByteArray.new(hex)
local tvb = ByteArray.tvb(b, "JSON TVB")
local tvb = ByteArray.tvb(decoded, "JSON TVB")
Dissector.get("json"):call(tvb, pkt, root)
end
-- Initialization routine
function p_tplink.init()
function hs1x0_proto_UDP.dissector (buf, pkt, root)
pkt.cols.protocol = "TPLink-SmartHome (UDP)"
local subtree = root:add(hs1x0_proto_UDP, buf() ,"TPLink-SmartHome")
local decoded = tpdecode(buf, 0)
subtree:add(decoded:raw())
subtree:append_text(" (decrypted)")
local tvb = ByteArray.tvb(decoded, "JSON TVB")
Dissector.get("json"):call(tvb, pkt, root)
end
-- Register a chained dissector for port 9999
local tcp_dissector_table = DissectorTable.get("tcp.port")
dissector = tcp_dissector_table:get_dissector(9999)
tcp_dissector_table:add(9999, p_tplink)
tcp_table = DissectorTable.get ("tcp.port")
udp_table = DissectorTable.get ("udp.port")
-- register the protocol to port 9999
tcp_table:add (9999, hs1x0_proto_TCP)
udp_table:add (9999, hs1x0_proto_UDP)

View file

@ -47,20 +47,30 @@ commands = {'info' : '{"system":{"get_sysinfo":{}}}',
'reset' : '{"system":{"reset":{"delay":1}}}'
}
light_commands = {
'on': '{"smartlife.iot.smartbulb.lightingservice":{"transition_light_state":{"on_off":1,"transition_period":0}}}',
'off': '{"smartlife.iot.smartbulb.lightingservice":{"transition_light_state":{"on_off":0,"transition_period":0}}}'
}
# Encryption and Decryption of TP-Link Smart Home Protocol
# XOR Autokey Cipher with starting key = 171
def encrypt(string):
def encrypt(string, doHeader=True):
key = 171
result = "\0\0\0\0"
if doHeader:
result = "\0\0\0\0"
else:
result = ""
for i in string:
a = key ^ ord(i)
key = a
result += chr(a)
return result
def decrypt(string):
def decrypt(string, doHeader=True):
key = 171
result = ""
if doHeader:
string = string[4:]
for i in string:
a = key ^ ord(i)
key = ord(i)
@ -70,6 +80,7 @@ def decrypt(string):
# Parse commandline arguments
parser = argparse.ArgumentParser(description="TP-Link Wi-Fi Smart Plug Client v" + str(version))
parser.add_argument("-t", "--target", metavar="<ip>", required=True, help="Target IP Address", type=validIP)
parser.add_argument("-l", "--lightbulb", dest="lightbulb", help="Enable lightbulb mode", action="store_true")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("-c", "--command", metavar="<command>", help="Preset command to send. Choices are: "+", ".join(commands), choices=commands)
group.add_argument("-j", "--json", metavar="<JSON string>", help="Full JSON string of command to send")
@ -81,19 +92,20 @@ port = 9999
if args.command is None:
cmd = args.json
else:
cmd = commands[args.command]
if args.lightbulb:
cmd = light_commands[args.command]
else:
cmd = commands[args.command]
# Send command and receive reply
try:
sock_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock_tcp.connect((ip, port))
sock_tcp.send(encrypt(cmd))
data = sock_tcp.recv(2048)
sock_tcp.close()
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM if args.lightbulb else socket.SOCK_STREAM)
sock.settimeout(1)
sock.sendto(encrypt(cmd, not args.lightbulb), (ip, port))
data = decrypt(sock.recv(2048), not args.lightbulb)
sock.close()
print "Sent: ", cmd
print "Received: ", decrypt(data[4:])
print "Received: ", data
except socket.error:
quit("Cound not connect to host " + ip + ":" + str(port))