kaiyga/MikroTikPatch
1
0
Fork 0
mirror of https://github.com/elseif/MikroTikPatch synced 2026-07-17 18:49:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update npk.py

Browse Source 
Signed-off-by: elseif <elseif@live.cn>
This commit is contained in:
elseif 2025-08-21 20:52:04 +08:00 committed by GitHub
parent 808b637ebd
commit 3c5ed64380
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
npk.py
View File

@ -207,7 +207,27 @@ class NovaPackage(Package):
self._parts.append(NpkPartItem(NpkPartID(part_id),NpkInfo.unserialize_from(part_data))) self._parts.append(NpkPartItem(NpkPartID(part_id),NpkInfo.unserialize_from(part_data)))
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):
if len(self._packages) > 0:
for package in self._packages:
count = 8
for part in package._parts:
count += 6
if part.id == NpkPartID.NULL_BLOCK:
break
count += len(part.data)
count += 6
package[NpkPartID.NULL_BLOCK].data = b'\x00' * (4096-count)
else:
count = 8
for part in self._parts:
count += 6
if part.id == NpkPartID.NULL_BLOCK:
break
count += len(part.data)
count += 6
self[NpkPartID.NULL_BLOCK].data = b'\x00' * (4096-count)
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:
@ -229,6 +249,7 @@ class NovaPackage(Package):
import hashlib import hashlib
from mikro import mikro_kcdsa_sign,mikro_eddsa_sign from mikro import mikro_kcdsa_sign,mikro_eddsa_sign
build_time = os.getenv('BUILD_TIME',None) build_time = os.getenv('BUILD_TIME',None)
self.set_null_block()
if len(self._packages) > 0: if len(self._packages) > 0:
if build_time: if build_time:
self[NpkPartID.PKG_INFO].data._build_time = int(build_time) self[NpkPartID.PKG_INFO].data._build_time = int(build_time)
@ -356,4 +377,4 @@ if __name__=='__main__':
option_npk.save(args.output) option_npk.save(args.output)
print(f'Created {args.output}') print(f'Created {args.output}')
else: else:
parser.print_help() parser.print_help()