chipmunkmc/mtkclient
1
0
Fork 0
mirror of https://github.com/bkerler/mtkclient.git synced 2024-11-14 19:25:05 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Improve read handling and fix write on mtk da legacy

Browse source
This commit is contained in:
info@revskills.de 2022-12-27 18:20:30 +01:00
parent 5714a30b43
commit 7c940e8443
2 changed files with 117 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
mtkclient/Library/mtk_dalegacy.py
View file

@ -14,6 +14,36 @@ from mtkclient.Library.partition import Partition
from mtkclient.config.payloads import pathconfig
from mtkclient.Library.legacy_ext import legacyext
from mtkclient.config.mtk_config import Mtk_Config
from queue import Queue
from threading import Thread, Event
rq = Queue()
wq = Queue()
def writedata(filename,rq,event):
pos=0
with open(filename, "wb") as wf:
while True:
if event.is_set():
break
data = rq.get()
pos+=len(data)
wf.write(data)
rq.task_done()
def readdata(filename,wq,pagesize,event):
pos=0
with open(filename, "rb") as rf:
bytestoread=os.stat(filename).st_size
while bytestoread>0:
if event.is_set():
break
data = rf.read(pagesize)
size = len(data)
wq.put(data)
pos += size
bytestoread-=size
wq.task_done()
class norinfo:
@ -1097,8 +1127,8 @@ class DALegacy(metaclass=LogBase):
if speed[0] == 0: # 1 = USB High Speed, 2= USB Ultra high speed
self.info("Reconnecting to preloader")
self.set_usb_cmd()
self.mtk.port.close(reset=False)
time.sleep(1)
self.mtk.port.close(reset=True)
time.sleep(2)
while not self.mtk.port.cdc.connect():
self.info("Waiting for reconnection")
time.sleep(0.5)
@ -1173,7 +1203,7 @@ class DALegacy(metaclass=LogBase):
return None
def set_usb_cmd(self):
if self.usbwrite(self.Cmd.USB_SETUP_PORT): # 72
if self.usbwrite(self.Cmd.USB_SETUP_PORT): # 70
if self.usbwrite(b"\x01"): # USB_HIGH_SPEED
res = self.usbread(1)
if len(res) > 0:
@ -1262,7 +1292,7 @@ class DALegacy(metaclass=LogBase):
self.usbwrite(b"\x08") # EMMC_PART_USER
self.usbwrite(pack(">Q", addr))
self.usbwrite(pack(">Q", length))
self.usbwrite(b"\x08") # index 8
self.usbwrite(b"\x06") # index 8
self.usbwrite(b"\x03")
packetsize = unpack(">I", self.usbread(4))[0]
ack = unpack(">B", self.usbread(1))[0]
@ -1366,6 +1396,7 @@ class DALegacy(metaclass=LogBase):
return length, parttype
def readflash(self, addr, length, filename, parttype=None, display=True):
global rq
self.mtk.daloader.progress.clear()
length, parttype = self.get_parttype(length, parttype)
self.check_usb_cmd()
@ -1409,14 +1440,17 @@ class DALegacy(metaclass=LogBase):
if display:
self.mtk.daloader.progress.show_progress("Read", 0, length, display)
if filename != "":
with open(filename, "wb") as wf:
event = Event()
worker = Thread(target=writedata, args=(filename, rq, event), daemon=True)
worker.start()
bytestoread = length
while bytestoread > 0:
size = bytestoread
if bytestoread > packetsize:
size = packetsize
wf.write(self.usbread(size))
bytestoread -= size
data=self.usbread(size)
bytestoread -= len(data)
rq.put(data)
checksum = unpack(">H", self.usbread(1) + self.usbread(1))[0]
self.debug("Checksum: %04X" % checksum)
self.usbwrite(self.Rsp.ACK)
@ -1426,6 +1460,7 @@ class DALegacy(metaclass=LogBase):
rpos = 0
self.mtk.daloader.progress.show_progress("Read", rpos, length, display)
self.mtk.daloader.progress.show_progress("Read", length, length, display)
event.set()
return True
else:
buffer = bytearray()

43
mtkclient/Library/mtk_daxflash.py
View file

@ -14,7 +14,36 @@ from mtkclient.Library.partition import Partition
from mtkclient.config.payloads import pathconfig
from mtkclient.Library.xflash_ext import xflashext, XCmd
from mtkclient.Library.settings import hwparam
from queue import Queue
from threading import Thread, Event
rq = Queue()
wq = Queue()
def writedata(filename,rq,event):
pos=0
with open(filename, "wb") as wf:
while True:
if event.is_set():
break
data = rq.get()
pos+=len(data)
wf.write(data)
rq.task_done()
def readdata(filename,wq,pagesize,event):
pos=0
with open(filename, "rb") as rf:
bytestoread=os.stat(filename).st_size
while bytestoread>0:
if event.is_set():
break
data = rf.read(pagesize)
size = len(data)
wq.put(data)
pos += size
bytestoread-=size
wq.task_done()
class NandExtension:
# uni=0, multi=1
@ -848,6 +877,7 @@ class DAXFlash(metaclass=LogBase):
return False
def readflash(self, addr, length, filename, parttype=None, display=True):
global rq
partinfo = self.getstorage(parttype, length)
if not partinfo:
return None
@ -859,14 +889,17 @@ class DAXFlash(metaclass=LogBase):
bytestoread = length
total = length
if filename != "":
with open(filename, "wb") as wf:
event = Event()
worker = Thread(target=writedata, args=(filename, rq, event), daemon=True)
worker.start()
resdata = bytearray()
while bytestoread > 0:
status = self.usbread(4 + 4 + 4)
magic, datatype, slength = unpack("<III", status)
if magic == 0xFEEEEEEF:
resdata = self.usbread(slength)
if slength > 4:
wf.write(resdata)
if len(resdata) > 4:
rq.put(resdata)
stmp = pack("<III", self.Cmd.MAGIC, self.DataType.DT_PROTOCOL_FLOW, 4)
data = pack("<I", 0)
self.usbwrite(stmp)
@ -886,7 +919,9 @@ class DAXFlash(metaclass=LogBase):
if unpack("<I", resdata)[0] == 0:
if display:
self.mtk.daloader.progress.show_progress("Read", total, total, display)
event.set()
return True
event.set()
return False
else:
buffer = bytearray()
@ -1135,7 +1170,7 @@ class DAXFlash(metaclass=LogBase):
self.info("Reconnecting to preloader")
self.config.set_gui_status(self.config.tr("Reconnecting to preloader"))
self.set_usb_speed()
self.mtk.port.close(reset=False)
self.mtk.port.close(reset=True)
time.sleep(2)
while not self.mtk.port.cdc.connect():
time.sleep(0.5)