Add files via upload

master
cheetah 4 years ago
parent a991bf62b0
commit df6ca8c39b

@ -0,0 +1,172 @@
#!/usr/bin/env python3
import time
import serial
import binascii
class TPLBirdySlim:
port = None
def __init__(self, serialPortName):
self.port = serial.Serial(
port=serialPortName,
baudrate=57600,
bytesize=serial.EIGHTBITS,
parity=serial.PARITY_NONE,
stopbits=serial.STOPBITS_ONE,
timeout=30,
xonxoff=False,
rtscts=False,
dsrdtr=False
)
print(self.port.name)
if self.port.is_open:
self.port.close()
self.port.open()
self.port.reset_input_buffer()
self.port.reset_output_buffer()
def _changeBaudrate(self, newRate):
if self.port.is_open:
self.port.close()
self.port.baudrate = newRate
self.port.open()
self.port.reset_input_buffer()
self.port.reset_output_buffer()
def checksum(self, data):
cs = 0x00
for x in data:
cs = (cs + x % 256) & 0xFF
return cs & 0xFF
def sendCMD(self, data):
cmd = []
cmd.append(0x02)
cmd.append(len(data))
cmd.extend(data)
cmd.append(0x03)
cmd.append(self.checksum(cmd.copy()))
cmd.append(0x04)
print(">", binascii.hexlify(bytes(cmd)))
self.port.write(cmd)
self.port.flush()
def readSingleByte(self):
return int.from_bytes(self.port.read(1), 'little')
def wait4Response(self):
#first Byte is always 0x02
if self.readSingleByte() != 0x02:
return None
dataLength = self.readSingleByte()
reply = []
for x in range(dataLength):
reply.append(self.readSingleByte())
if self.readSingleByte() != 0x03:
return None
pCRC = self.readSingleByte()
pCRCData = [0x02, dataLength]
pCRCData.extend(reply)
pCRCData.extend([0x03])
rCRC = self.checksum(pCRCData)
if self.readSingleByte() != 0x04:
return None
if pCRC != rCRC:
return None # crc mismatch
return reply
def askHighspeed(self):
self.sendCMD([0x25,0x01])
return self.wait4Response()
def switchToHighspeed(self):
resp = self.askHighspeed()
if len(resp) == 1 and resp[0] == 0x06:
self._changeBaudrate(115200)
return True
else:
return False
def readE2P(self, addr, size=0x80):
_addr = addr.to_bytes(2, byteorder='big')
_cmd = [0x29, 0x00, 0x00, _addr[0], _addr[1], size]
self.sendCMD(_cmd)
resp = self.wait4Response()
print(">", binascii.hexlify(bytes(resp)))
if resp[0:5] != _cmd[0:5]:
print(resp[0:5], _cmd[0:5])
return None # mismatch
return resp[5:]
def writeE2P(self, addr, data):
_addr = addr.to_bytes(2, byteorder='big')
_cmd = [0x2A, 0x00, 0x00, _addr[0], _addr[1]]
_cmd.extend(data)
self.sendCMD(_cmd)
resp = self.wait4Response()
if resp[0:4] != _cmd[0:4]:
print("resp mismatch", resp, _cmd)
return False # mismatch
if resp[5] != len(data):
print("length mismatch")
return False # mismatch
if resp[6:] != _cmd[5:]:
print("resp mismatch", resp, _cmd)
return False # mismatch
return True
#def boskrypt_mode(self):
# self.sendCMD([0xF3,0xF3])
def getDeviceInfo(self):
self.sendCMD([0x01,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff])
return self.wait4Response()
def writeSerialNumber(self, newSerial):
_cmd = [0x13]
_cmd.extend(newSerial)
self.sendCMD(_cmd)
resp = self.wait4Response()
if _cmd[2:] != resp[2:]:
return False
return True
def checkE2PData(self, data):
return len(data) / 0x80 == 0xCD
test = TPLBirdySlim('/dev/ttyUSB0')
#test.switchBootloaderSpeed()
#test._changeBaudrate(38700)
#print(bytes(test.getDeviceInfo()).decode('ascii'))
#test.sendCMD([0x01,0x12])
#test.sendCMD([0x18,0x08])
#test.sendCMD([0x10,0x08])
#print(bytes(test.wait4Response()))
#test.writeSerialNumber([0x42,0x36,0x31,0x33,0x33,0x37,0x31,0x33,0x33,0x37,0x31,0x33,0x33,0x37])
#print(bytes(test.getDeviceInfo()).decode('ascii'))
#test.sendCMD([0x13,0x43,0x48,0x45,0x45,0x54,0x41,0x48,0x43,0x48,0x45,0x45,0x54,0x41,0x48])
#print(
test.switchToHighspeed()
#)
"""
e2pread = []
for i in range(0xCD):
e2pread.extend(test.readE2P(i*0x80))
with open("dump_434_232500.bin", "wb") as e2pfile:
e2pfile.write(bytes(e2pread))
"""
e2pdata = []
#with open("BrainwvaebusinessDefualt.birdyIOT.rev2.bin", "rb") as e2pfile:
with open("dump2.bin", "rb") as e2pfile:
e2pdata = e2pfile.read()
if test.checkE2PData(e2pdata):
for i in range(0xCD):
chunk = e2pdata[ 0x80*i : 0x80*(i+1) ]
print(test.writeE2P(i*0x80, chunk))
#print(e2pdata)
#print(test.port.read(62))
Loading…
Cancel
Save