|
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
|
#coding=utf-8
import getopt
import os, sys
import zipfile
from Crypto.Cipher import AES
import random, struct
#加密python3的代码
def transfer3(dir_pref):
os.system('cython -2 %s.py;'
'gcc -c -fPIC -I/usr/include/python3.5/ %s.c -o %s.o'
% (dir_pref, dir_pref, dir_pref))
os.system('gcc -shared %s.o -o %s.so' % (dir_pref, dir_pref))
if clear:
os.system('rm -f %s.c %s.o %s.py' % (dir_pref, dir_pref, dir_pref))
else:
os.system('rm -f %s.c %s.o' % (dir_pref, dir_pref))
#加密python2的代码
def transfer2(dir_pref):
os.system('cython -2 %s.py;'
'gcc -c -fPIC -I/usr/include/python2.7/ %s.c -o %s.o'
% (dir_pref, dir_pref, dir_pref))
os.system('gcc -shared %s.o -o %s.so' % (dir_pref, dir_pref))
if clear:
os.system('rm -f %s.c %s.o %s.py' % (dir_pref, dir_pref, dir_pref))
else:
os.system('rm -f %s.c %s.o' % (dir_pref, dir_pref))
#加密AI模型
def encrypt_file(in_filename, out_filename=None, chunksize=64*1024):
"""
使用AES（CBC模式）加密文件给定的密钥。
:param key: 加密密钥-必须是16、24或32字节长。长按键更安全。
:param in_filename: 输入的文件的名称
:param out_filename: 如果为None，将使用“<in_filename>.enc”。
:param chunksize: 设置函数用于读取和加密文件。大块一些文件和机器的大小可能更快。块大小必须可被16整除。
:return: None
"""
if not out_filename:
out_filename = in_filename + '.enc'
salt = '' # 盐值
key = "{: <32}".format(salt).encode("utf-8")
#iv = ''.join(chr(random.randint(0, 0xFF)) for i in range(16))
#encryptor = AES.new(key, AES.MODE_CBC, iv)
iv = b'0000000000000000'
encryptor = AES.new(key, AES.MODE_CBC, iv)
filesize = os.path.getsize(in_filename)
with open(in_filename, 'rb') as infile:
with open(out_filename, 'wb') as outfile:
outfile.write(struct.pack('<Q', filesize))
outfile.write(iv)
while True:
chunk = infile.read(chunksize)
if len(chunk) == 0:
break
elif len(chunk) % 16 != 0:
chunk += (' ' * (16 - len(chunk) % 16)).encode("utf-8")
outfile.write(encryptor.encrypt(chunk))
def zip_dir(dir_path,out_path):
"""
压缩指定文件夹
:param dir_path: 目标文件夹路径
:param out_path: 压缩文件保存路径+xxxx.zip
:return: 无
"""
zip = zipfile.ZipFile(out_path, "w", zipfile.ZIP_DEFLATED)
for path, dirnames, filenames in os.walk(dir_path):
# 去掉目标跟路径，只对目标文件夹下边的文件及文件夹进行压缩
fpath = path.replace(dir_path, '')
for filename in filenames:
zip.write(os.path.join(path, filename), os.path.join(fpath, filename))
zip.close()
if __name__ == '__main__':
help_show = '''
python version:
python3 该代码用于加密python3编写的代码,将.py文件转换成.so文件，达到加密的效果
python2 该代码用于加密python2编写的代码,将.py文件转换成.so文件，达到加密的效果
Options:
-h, --help 显示帮助
-d, --directory 你需要加密的文件夹
-o, --operation 你所需要执行的操作,python3 or python2 or model
-f, --file 加密单个py文件
-c, --clear 删除原始的py文件
-m, --maintain 列出你不需要加密的文件和文件夹，如果是文件夹的话需要加[]
例子: -m __init__.py,setup.py,[poc,resource,venv,interface]
-z, --zip 加密之后压缩文件
Example:
python setup.py -f test_file.py -o python2 加密单个文件
python setup.py -d test_dir -o python2 -m __init__.py,setup.py,[poc/,resource/,venv/,interface/] -c 加密文件夹
python3 setup.py -f test_file.py -o python3 加密单个文件
python3 setup.py -d test_dir -o python3 -m __init__.py,setup.py,[poc/,resource/,venv/,interface/] -c 加密文件夹
'''
clear = 0
is_zip = False
root_name = ''
operation = ''
file_name = ''
m_list = ''
try:
options,args = getopt.getopt(sys.argv[1:],"vh:d:f:cm:o:z:",["version","help","directory=","file=","operation=","zip","clear","maintain="])
except getopt.GetoptError:
print(help_show)
sys.exit(1)
for key, value in options:
if key in ['-h', '--help']:
print(help_show)
elif key in ['-c', '--clear']:
clear = 1
elif key in ['-d', '--directory']:
root_name = value
elif key in ['-f', '--file']:
file_name = value
elif key in ['-o', '--operation']:
operation = value
elif key in ['-z','--zip']:
is_zip = True
elif key in ['-m', '--maintain']:
m_list = value
file_flag = 0
dir_flag = 0
if m_list.find(',[') != -1:
tmp = m_list.split(',[')
file_list = tmp[0]
dir_list = tmp[1:-1]
file_flag = 1
dir_flag = 1
elif m_list.find('[') != -1:
dir_list = m_list[1:-1]
dir_flag = 1
else:
file_list = m_list.split(',')
file_flag = 1
if dir_flag == 1:
dir_tmp = dir_list.split(',')
dir_list=[]
for d in dir_tmp:
if d.startswith('./'):
dir_list.append(d[2:])
else:
dir_list.append(d)
if root_name != '':
if not os.path.exists(root_name):
print('No such Directory, please check or use the Absolute Path')
sys.exit(1)
if os.path.exists('%s_old' % root_name):
os.system('rm -rf %s_old' % root_name)
#os.system('cp -r %s %s_old' % (root_name, root_name)) #备份源文件
try:
for root, dirs, files in os.walk(root_name):
for file in files:
if m_list != '':
skip_flag = 0
if dir_flag == 1:
for dir in dir_list:
if (root+'/').startswith(root_name + '/' + dir):
skip_flag = 1
break
if skip_flag:
continue
if file_flag == 1:
if file in file_list:
continue
pref = file.split('.')[0]
dir_pref = root + '/' + pref
if file.endswith('.pyc'):
os.system('rm -f %s' % dir_pref+'.pyc')
elif file.endswith('.so'):
pass
elif file.endswith('.py'):
if operation == 'python3':
transfer3(dir_pref)
elif operation == 'python2':
transfer2(dir_pref)
else:
pass
except Exception as e:
print(e)
if file_name != '':
try:
dir_pref = file_name.split('.')[0]
if operation == 'python3':
transfer3(dir_pref)
elif operation == 'python2':
transfer2(dir_pref)
else:
pass
except Exception as e:
print(e)
if is_zip:
zip_dir(root_name,root_name+'.zip')
if operation == 'model':
encrypt_file(root_name)
|
以上就是python通过cython加密代码的详细内容，更多关于python cpython加密的资料请关注服务器之家其它相关文章！
原文链接：https://www.cnblogs.com/wuli1427102168/p/12418629.html