mirror/MikroTikPatch
1
0
Fork 0
mirror of https://huihui.cat/mirrors/MikroTikPatch.git synced 2025-12-10 03:24:37 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Update npk.py

Browse source 
Signed-off-by: elseif <elseif@live.cn>
This commit is contained in:
elseif 2025-08-22 00:26:03 +08:00 • committed by GitHub
parent 7195dae015
commit a85cbf1548
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 30 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
npk.py
View file

@ -208,14 +208,26 @@ class NovaPackage(Package):
else: else:
self._parts.append(NpkPartItem(NpkPartID(part_id),part_data)) self._parts.append(NpkPartItem(NpkPartID(part_id),part_data))
def set_null_block(self): def set_null_block(self):
def rebuild_squashfs(data):
with open('squashfs.sfs', 'wb') as f:
f.write(data)
os.system('unsquashfs -d squashfs-root squashfs.sfs')
os.system('rm squashfs.sfs')
os.system('mksquashfs squashfs-root squashfs.sfs -comp xz -no-xattrs -b 256k')
os.system('rm -rf squashfs-root')
with open('squashfs.sfs', 'rb') as f:
return f.read()
if len(self._packages) > 0:
has_squashfs = False has_squashfs = False
for part in self._parts: for package in self._packages:
for part in package._parts:
if part.id == NpkPartID.SQUASHFS: if part.id == NpkPartID.SQUASHFS:
part.data = rebuild_squashfs(part.data)
has_squashfs = True has_squashfs = True
break
if not has_squashfs: if not has_squashfs:
return return
if len(self._packages) > 0:
for package in self._packages: for package in self._packages:
count = 8 count = 8
for part in package._parts: for part in package._parts:
@ -224,9 +236,17 @@ class NovaPackage(Package):
break break
count += len(part.data) count += len(part.data)
count += 6 count += 6
package[NpkPartID.NULL_BLOCK].data = b'\x00' * (4096-count) pad_len = (4096 - (count % 4096)) % 4096
package[NpkPartID.NULL_BLOCK].data = b'\x00' * pad_len
else: else:
has_squashfs = False
for part in self._parts:
if part.id == NpkPartID.SQUASHFS:
part.data = rebuild_squashfs(part.data)
has_squashfs = True
if not has_squashfs:
return
count = 8 count = 8
for part in self._parts: for part in self._parts:
count += 6 count += 6
@ -234,7 +254,8 @@ class NovaPackage(Package):
break break
count += len(part.data) count += len(part.data)
count += 6 count += 6
self[NpkPartID.NULL_BLOCK].data = b'\x00' * (4096-count) pad_len = (4096 - (count % 4096)) % 4096
self[NpkPartID.NULL_BLOCK].data = b'\x00' * pad_len
def get_digest(self,hash_fnc,package:Package=None)->bytes: def get_digest(self,hash_fnc,package:Package=None)->bytes:
parts = package._parts if package else self._parts parts = package._parts if package else self._parts
for part in parts: for part in parts: