[Python] 파이썬 바이트 코드 디컴파일

컴파일된 pyc가 포함된 zip파일에서 python 코드 디컴파일


문제 출처 :

Hack.lu 2013 [ What's wrong with this?]

- zip file : library.zip
- pyc file : __main__hello__.pyc

1. >>> import zipimport
2. >>>
3. >>> importer = zipimport.zipimporter('library.zip')
4. >>> code = importer.get_code('__main__hello__')
5. >>>
6. >>> print dir(code)
7. ['__class__', '__cmp__', '__delattr__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__gt__', '__hash__', '__init__', '__le__', '__lt__','__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'co_argcount', 'co_cellvars','co_code', 'co_consts', 'co_filename', 'co_firstlineno', 'co_flags', 'co_freevars', 'co_lnotab', 'co_name', 'co_names', 'co_nlocals', 'co_stacksize','co_varnames']
8. >>> print code.co_names # 사용된 오브젝트 이름
9. ('sys', 'hashlib', 'sha256', 'dis', 'multiprocessing', 'UserList', 'encrypt_string', 'rot_chr', 'SECRET', 'argv')
10. >>>
11. >>> import dis
12. >>> dis.dis(code)# 파이썬 바이너리 코드를 disassemble함
13.   1           0 LOAD_CONST               0 (-1)
14.               3 LOAD_CONST               1 (None)
15.               6 IMPORT_NAME              0 (sys)
16.               9 STORE_NAME               0 (sys)
18.   2          12 LOAD_CONST               0 (-1)
19.              15 LOAD_CONST               2 (('sha256',))
20.              18 IMPORT_NAME              1 (hashlib)
21.              21 IMPORT_FROM              2 (sha256)
22.              24 STORE_NAME               2 (sha256)
23.              27 ROT_TWO
25.   3          28 LOAD_CONST               0 (-1)
26.              31 LOAD_CONST               1 (None)
27.              34 IMPORT_NAME              3 (dis)
28.              37 STORE_NAME               3 (dis)
29.              40 LOAD_CONST               0 (-1)
30.              43 LOAD_CONST               1 (None)
31.              46 IMPORT_NAME              4 (multiprocessing)
32.              49 STORE_NAME               4 (multiprocessing)
33.              52 LOAD_CONST               0 (-1)
34.              55 LOAD_CONST               1 (None)
35.              58 IMPORT_NAME              5 (UserList)
36.              61 STORE_NAME               5 (UserList)
38.   5          64 LOAD_CONST               3 (<code object encrypt_string at 0x7fa62ded7cb0, file "chall.py", line 5>) # 첫번째 함수
39.              67 MAKE_FUNCTION            0
40.              70 STORE_NAME               6 (encrypt_string)
42.  16          73 LOAD_CONST               4 (<code object rot_chr at 0x7fa62dedcb30, file "chall.py", line 16>) # 두번째 함수
43.              76 MAKE_FUNCTION            0
44.              79 STORE_NAME               7 (rot_chr)
46.  19          82 LOAD_CONST               5 ('w*0;CNU[\\gwPWk}3:PWk"#&:ABu/:Hi,M')
47.              85 STORE_NAME               8 (SECRET)
49.  20          88 LOAD_NAME                6 (encrypt_string)
50.              91 LOAD_NAME                0 (sys)
51.              94 LOAD_ATTR                9 (argv)
52.              97 LOAD_CONST               6 (1)
53.             100 BINARY_SUBSCR
54.             101 CALL_FUNCTION            1
55.             104 LOAD_NAME                8 (SECRET)
56.             107 COMPARE_OP               2 (==)
57.             110 POP_JUMP_IF_FALSE      121
59.  21         113 LOAD_CONST               7 ('Yup')
60.             116 PRINT_ITEM
61.             117 PRINT_NEWLINE
62.             118 JUMP_FORWARD             5 (to 126)
64.  23     >>  121 LOAD_CONST               8 ('Nope')
65.             124 PRINT_ITEM
66.             125 PRINT_NEWLINE
67.         >>  126 LOAD_CONST               1 (None)
68.             129 RETURN_VALUE
69. >>> print code.co_consts
70. (-1, None, ('sha256',), <code object encrypt_string at 0x7fa62ded7cb0, file "chall.py", line 5>, <code object rot_chr at 0x7fa62dedcb30, file"chall.py", line 16>, 'w*0;CNU[\\gwPWk}3:PWk"#&:ABu/:Hi,M', 1, 'Yup', 'Nope')
71. >>> print code.co_consts[3]
72. <code object encrypt_string at 0x7fa62ded7cb0, file "chall.py", line 5>
73. >>>
74. >>> dir(code.co_consts[3])
75. ['__class__', '__cmp__', '__delattr__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__gt__', '__hash__', '__init__', '__le__', '__lt__','__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'co_argcount', 'co_cellvars','co_code', 'co_consts', 'co_filename', 'co_firstlineno', 'co_flags', 'co_freevars', 'co_lnotab', 'co_name', 'co_names', 'co_nlocals', 'co_stacksize','co_varnames']
76. >>>
77. >>> print code.co_consts[3].co_code
78. g}xct|▒E]U\}}|dkrD|jt|d▒▒q|jt|t||d▒▒▒qXdj|▒T
79. >>> dis.dis(code.co_consts[3].co_code)
80.           0 BUILD_LIST          0
81.           3 STORE_FAST          1 (1) # '1' = new_str = []
82.           6 SETUP_LOOP         99 (to 108) # for
83.           9 LOAD_GLOBAL         0 (0) # 'global_0' = parameter : s
84.          12 LOAD_FAST           0 (0) # '0' = s
85.          15 CALL_FUNCTION       1 # enumerate(s)
86.          18 GET_ITER
87.     >>   19 FOR_ITER           85 (to 107)
88.          22 UNPACK_SEQUENCE     2 # (index, c)
89.          25 STORE_FAST          2 (2) # '2' = index
90.          28 STORE_FAST          3 (3) # '3' = c
91.          31 LOAD_FAST           2 (2) # index
92.          34 LOAD_CONST          1 (1) # 'const_1' = 0
93.          37 COMPARE_OP          2 (==) # ( '2' == '1' ) <-> ( index == 1 )
94.          40 POP_JUMP_IF_FALSE    68
95.          43 LOAD_FAST           1 (1) # '1' (new_str)
96.          46 LOAD_ATTR           1 (1) # '1'.'1' (new_str.append)
97.          49 LOAD_GLOBAL         2 (2) # 'global_2' = 'rot_chr'
98.          52 LOAD_FAST           3 (3) # 'c'
99.          55 LOAD_CONST          2 (2) # 'const_2' = 10
100.          58 CALL_FUNCTION       2 # rot_chr(c, 10)
101.          61 CALL_FUNCTION       1 # new_str.append(rot_chr(c, 10))
102.          64 POP_TOP
103.          65 JUMP_ABSOLUTE      19
104.     >>   68 LOAD_FAST           1 (1) # '1' (new_str)
105.          71 LOAD_ATTR           1 (1) # '1'.'1' (new_str.append)
106.          74 LOAD_GLOBAL         2 (2) # 'global_2' = 'rot_chr'
107.          77 LOAD_FAST           3 (3) # 'c'
108.          80 LOAD_GLOBAL         3 (3) # 'global_3' = 'ord'
109.          83 LOAD_FAST           1 (1) # '1' (new_str)
110.          86 LOAD_FAST           2 (2) # '2' = index
111.          89 LOAD_CONST          3 (3) # 'const_2' = 1
112.          92 BINARY_SUBTRACT  # '2' - 'const_2' <-> index - 1
113.          93 BINARY_SUBSCR  # new_str[index - 1]
114.          94 CALL_FUNCTION       1  # ord(new_str[index - 1])
115.          97 CALL_FUNCTION       2  # rot_chr(c, ord(new_str[index - 1]))
116.         100 CALL_FUNCTION       1  # new_str.append(rot_chr(c, ord(new_str[index - 1])))
117.         103 POP_TOP
118.         104 JUMP_ABSOLUTE      19
119.     >>  107 POP_BLOCK
120.     >>  108 LOAD_CONST          4 (4)
121.         111 LOAD_ATTR           4 (4)
122.         114 LOAD_FAST           1 (1)
123.         117 CALL_FUNCTION       1
124.         120 RETURN_VALUE

<code object encrypt_string at 0x7fa62ded7cb0, file "chall.py", line 5> 디컴파일 결과

1. def encrypt_string(s):
2.     new_str = []
3.     for (index, c) in enumerate(s):
4.         if (index == 0):
5.             new_str.append(rot_chr(c,10))
6.         else:
7.             new_str.append(rot_chr(c,ord(new_str[index - 1])))
8.     return ''.join(new_str)

전체 소스 디컴파일 결과

1. import sys
2. import dis
3. import multiprocessing
4. import UserList
6. def encrypt_string(s):
7.     new_str = []
8.     for (index, c) in enumerate(s):
9.         if (index == 0):
10.             new_str.append(rot_chr(c,10))
11.         else:
12.             new_str.append(rot_chr(c,ord(new_str[index - 1])))
13.     return ''.join(new_str)
15. def rot_chr(c, amount):
16.     return chr((((ord(c) - 33) + amount) % 94) + 33)
18. SECRET = 'w*0;CNU[\\gwPWk}3:PWk"#&:ABu/:Hi,M'
20. if encrypt_string(sys.argv[1]) == SECRET:
21.     print
22.     print >>'Yup'
23. else:
24.     print
25.     print >>'Nope'

** key : modified_in7erpreters_are_3vil!!! **


Reference :

- http://docs.python.org/2/library/dis.html#python-bytecode-instructions
- http://pymotw.com/2/dis/
- http://stackoverflow.com/questions/16219245/how-do-you-remove-the-nth-index-of-a-nested-list
-  http://stackoverflow.com/questions/13905741/accessing-class-variables-from-a-list-comprehension-in-the-class-definition