实现python源代码加密

加密方法

最近一直在研究python加密，网上总结包括以下几种:

1、将py文件编译成pyc（放弃）

Python 标准库中提供了一个名为 compileall 的库，可以轻松地进行编译。
优点：
简单方便，提高了一点源码破解门槛
平台兼容性好，.py 能在哪里运行，.pyc 就能在哪里运行
缺点：
有现成的反编译工具（python-uncompyle6 ），破解成本低

2、代码混淆（放弃）

使用Python代码混淆库pyobfuscate
优点：
简单方便，提高了一点源码破解门槛
兼容性好，只要源码逻辑能做到兼容，混淆代码亦能
缺点：
只能对单个文件混淆，无法做到多个互相有联系的源码文件的联动混淆
代码结构未发生变化，也能获取字节码，破解难度不大

3、修改python解释器（放弃）

由于技术有限，只能暂时放弃

由于发行商业 Python 程序到客户环境时通常会包含一个 Python 解释器，如果改造解释器能解决源码保护的问题，那么也是可选的一条路。

4、将py转化成so文件（采用）

使用Cython将py转成so文件

1、安装包

python安装Cython
pip3 install Cython
linux安装python-devel,gcc
yum install python-devel
yum install gcc

2、编写测试脚本

在/home目录下新建test目录，在test目录下新建一个测试文件jiami.py’，写入如下内容

#!/usr/bin/env python
# -*- coding: utf-8 -*-
__author__ = "***"

import datetime
class test:
    def say(self):
        print('hello',datetime.datetime.now())

在新建一个用来调用jiami.py中的test类的文件–run.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from jiami import test

test_obj = test()
test_obj.say()

最后新建一个setup.py用来加密jiami.py文件，将jiami.py文件改成.so文件

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from distutils.core import setup
from Cython.Build import cythonize

setup(ext_modules = cythonize(['jiami.py']))

3、运行setup.py对jiami.py进行加密

切换到/home/test下运行

python3 setup.py build_ext

发现增加了一个build文件夹和jiami.c的文件

执行python3 run.py

加密文件在/home/test/build/lib.linux-x86_64-3.7下

后缀为.so的就是加密后的文件

4、替换

将build/lib.linux-x86_64-3.7文件夹下的.so文件移动到/home/test文件夹下，也就是替换jiami.py文件

mv /home/test/build/lib.linux-x86_64-3.7/jiami.cpython-37m-x86_64-linux-gnu.so /home/test

删除jiami.py文件，运行run.py，看能否打印结果

rm -rf /home/test/jiami.py
python3 run.py

直接执行某so文件内函数：

python -c "from hello import hello;hello()"

5、文件夹下文件加密

一个一个加密太麻烦了，github上有人写好了
链接地址：github:https://github.com/ArvinMei/py2so.git

新建一个setup.py文件，填写如下内容：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/9/10 14:38
# @Author  : 
# @FILE    : setup.py
# @DESC   :


import sys, os, shutil, time
from distutils.core import setup
from Cython.Build import cythonize

starttime = time.time()
setupfile= os.path.join(os.path.abspath('.'), __file__)

def getpy(basepath=os.path.abspath('.'), parentpath='', name='', build_dir="build",
          excepts=(), copyOther=False, delC=False):
    """
    获取py文件的路径
    :param basepath: 根路径
    :param parentpath: 父路径
    :param name: 文件/夹
    :param excepts: 排除文件
    :param copy: 是否copy其他文件
    :return: py文件的迭代器
    """
    fullpath = os.path.join(basepath, parentpath, name)
    for fname in os.listdir(fullpath):
        ffile = os.path.join(fullpath, fname)
        if os.path.isdir(ffile) and ffile != os.path.join(basepath, build_dir) and not fname.startswith('.'):
            for f in getpy(basepath, os.path.join(parentpath, name), fname, build_dir, excepts, copyOther, delC):
                yield f
        elif os.path.isfile(ffile):
            # print("\t", basepath, parentpath, name, ffile)
            ext = os.path.splitext(fname)[1]
            if ext == ".c":
                if delC and os.stat(ffile).st_mtime > starttime:
                    os.remove(ffile)
            elif ffile not in excepts and ext not in('.pyc', '.pyx'):
                # print("\t\t", basepath, parentpath, name, ffile)
                if ext in('.py', '.pyx') and not fname.startswith('__'):
                    yield os.path.join(parentpath, name, fname)
                elif copyOther:
                        dstdir = os.path.join(basepath, build_dir, parentpath, name)
                        if not os.path.isdir(dstdir): os.makedirs(dstdir)
                        shutil.copyfile(ffile, os.path.join(dstdir, fname))
        else:
            pass

if __name__ == "__main__":
    currdir = os.path.abspath('.')
    parentpath = sys.argv[1] if len(sys.argv)>1 else "."

    currdir, parentpath = os.path.split(currdir if parentpath == "." else os.path.abspath(parentpath))
    build_dir = os.path.join(parentpath, "build")
    build_tmp_dir = os.path.join(build_dir, "temp")
    print("start:", currdir, parentpath, build_dir)
    os.chdir(currdir)
    try:
        #获取py列表
        module_list = list(getpy(basepath=currdir,parentpath=parentpath, build_dir=build_dir, excepts=(setupfile)))
        print(module_list)
        setup(ext_modules = cythonize(module_list),script_args=["build_ext", "-b", build_dir, "-t", build_tmp_dir])
        module_list = list(getpy(basepath=currdir, parentpath=parentpath, build_dir=build_dir, excepts=(setupfile), copyOther=True))
    except Exception as ex:
        print("error! ", ex)
    finally:
        print("cleaning...")
        module_list = list(getpy(basepath=currdir, parentpath=parentpath, build_dir=build_dir, excepts=(setupfile), delC=True))
        if os.path.exists(build_tmp_dir): shutil.rmtree(build_tmp_dir)

    print("complate! time:", time.time()-starttime, 's')

运行命令为：

python3 setup.py 文件夹路径

运行结束后，会生成一个build文件夹，将该文件夹下的.os文件全部移动到被加密的文件夹，删除原文件夹下的py文件

总结：

Cython方法加密就是将py文件转为so文件，用so文件替换py文件，执行速率会比python快一些，但是偶尔会遇到部分代码不好使，只能针对进行测试，将失效的不进行加密或者采取其他方式加密