FilesExpand file tree

main

/

test_obfuscator.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

80 lines (77 loc) · 3.32 KB

main

/

test_obfuscator.py

Copy path

Top

File metadata and controls

• Code
• Blame

80 lines (77 loc) · 3.32 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

from __future__ import annotations

import subprocess

import sys

from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent

sys.path.insert(0, str(ROOT))

from patchwork import Obfuscator

EXAMPLES = ROOT / 'examples'

EXAMPLE_NAMES = ['hello', 'fizzbuzz', 'classes', 'stress', 'modern', 'future_annotations', 'literals']

SEEDS = [1, 7, 42, 1234, 999999]

def run(path: Path) -> str:

res = subprocess.run([sys.executable, str(path)], capture_output=True, text=True, timeout=20)

if res.returncode != 0:

raise RuntimeError(f'{path} exited {res.returncode}\nstderr:\n{res.stderr}')

return res.stdout

def check_one(name: str, seed: int) -> None:

src_path = EXAMPLES / f'{name}.py'

expected = run(src_path)

obf_src = Obfuscator(seed=seed).obfuscate(src_path.read_text(encoding='utf-8'))

obf_path = EXAMPLES / f'{name}_test_{seed}.py'

obf_path.write_text(obf_src, encoding='utf-8')

try:

actual = run(obf_path)

finally:

obf_path.unlink(missing_ok=True)

if actual != expected:

raise AssertionError(f'{name} seed={seed}: output mismatch\n--- expected ---\n{expected}--- actual ---\n{actual}')

def main() -> int:

fails = 0

for name in EXAMPLE_NAMES:

for seed in SEEDS:

try:

check_one(name, seed)

print(f'ok: {name:10s} seed={seed}')

except Exception as e:

fails += 1

print(f'FAIL: {name:10s} seed={seed}: {e}')

src = (EXAMPLES / 'hello.py').read_text(encoding='utf-8')

a = Obfuscator(seed=1).obfuscate(src)

b = Obfuscator(seed=2).obfuscate(src)

c = Obfuscator(seed=1).obfuscate(src)

if a == b:

print('FAIL: polymorphism (seed=1 vs seed=2 produced identical output)')

fails += 1

else:

print('ok: polymorphism (seeds 1 vs 2 differ)')

if a != c:

print('FAIL: reproducibility (same seed gave different output)')

fails += 1

else:

print('ok: reproducibility (seed=1 stable)')

literals_src = (EXAMPLES / 'literals.py').read_text(encoding='utf-8')

literals_obf = Obfuscator(seed=2026).obfuscate(literals_src)

if 'PATCHWORK_LITERAL_MARKER_9f8b6b7d' in literals_obf or 'PATCHWORK_BYTES_MARKER' in literals_obf:

print('FAIL: literal markers were present in obfuscated output')

fails += 1

else:

print('ok: literal markers absent from obfuscated output')

future_src = (EXAMPLES / 'future_annotations.py').read_text(encoding='utf-8')

for kwargs in ({'encrypt_strings': False}, {'opaque_predicates': False, 'junk_branches': False}):

try:

obf = Obfuscator(seed=77, **kwargs).obfuscate(future_src)

obf_path = EXAMPLES / f'future_annotations_options_{len(kwargs)}.py'

obf_path.write_text(obf, encoding='utf-8')

try:

if run(obf_path) != run(EXAMPLES / 'future_annotations.py'):

raise AssertionError('output mismatch')

finally:

obf_path.unlink(missing_ok=True)

print(f'ok: future annotations options {kwargs}')

except Exception as e:

fails += 1

print(f'FAIL: future annotations options {kwargs}: {e}')

print(f'\n{fails} failures' if fails else '\nall good')

return 1 if fails else 0

if __name__ == '__main__':

raise SystemExit(main())