diff options
| -rwxr-xr-x | src/ukify/ukify.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/ukify/ukify.py b/src/ukify/ukify.py index 63e7202d3d..8f8537b30e 100755 --- a/src/ukify/ukify.py +++ b/src/ukify/ukify.py @@ -424,6 +424,18 @@ def join_initrds(initrds): assert False +def pe_validate(filename): + import pefile + + pe = pefile.PE(filename) + + sections = sorted(pe.sections, key=lambda s: (s.VirtualAddress, s.Misc_VirtualSize)) + + for l, r in itertools.pairwise(sections): + if l.VirtualAddress + l.Misc_VirtualSize > r.VirtualAddress + r.Misc_VirtualSize: + raise ValueError(f'Section "{l.Name.decode()}" ({l.VirtualAddress}, {l.Misc_VirtualSize}) overlaps with section "{r.Name.decode()}" ({r.VirtualAddress}, {r.Misc_VirtualSize})') + + def make_uki(opts): # kernel payload signing @@ -540,6 +552,8 @@ def make_uki(opts): print('+', shell_join(cmd)) subprocess.check_call(cmd) + pe_validate(output) + # UKI signing if opts.sb_key: |