zack3d/pylingual
1
0
Fork 0
mirror of https://github.com/syssec-utd/pylingual.git synced 2026-05-10 18:39:03 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix slice masking

Browse Source
This commit is contained in:
Josh Wiedemeier
2026-01-07 15:40:05 -06:00
parent ede802d9ee
commit 881d7b03f7
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pylingual/masking/global_masker.py
+4
View File
@@ -195,6 +195,8 @@ class Masker:
continue # don't mask None
elif type(const) in (list, tuple, frozenset):
consts[idx] = type(const)(replace_list(list(const)))
elif type(const) is slice:
consts[idx] = f"{self.mask(const.start)} : {self.mask(const.stop)} : {self.mask(const.step)}"
else:
consts[idx] = self.mask(const)
return consts
@@ -214,6 +216,8 @@ class Masker:
# demote quotes one layer
arg_repr = repr(type(inst.argval)(consts)).replace("'", "").replace('"', "'")
view = f"{inst.opname} , {arg_repr}"
elif type(inst.argval) is slice:
view = f"{inst.opname} , {self.mask(inst.argval.start)} : {self.mask(inst.argval.stop)} : {self.mask(inst.argval.step)}"
else:
view = f"{inst.opname} , {self.mask(inst.bytecode.resolve_namespace(inst.argval))}"
pylingual/masking/model_disasm.py
+3
View File
@@ -56,6 +56,9 @@ def create_global_masker(bytecode: EditableBytecode) -> Masker:
continue
if type(const) in (list, tuple, frozenset, set):
consts.extend(const)
elif type(const) is slice:
# decompose slice constants for 3.14+
consts.extend([const.start, const.stop, const.step])
else:
global_tab.update({bc.resolve_namespace(const): f"<mask_{global_idx}>"})
global_idx += 1