pax_global_header 0000666 0000000 0000000 00000000064 14143644277 0014526 g ustar 00root root 0000000 0000000 52 comment=b0c878199db2898ea9d08c6a2e0b652971662bab
agentencryption/ 0000775 0000000 0000000 00000000000 14143644277 0014303 5 ustar 00root root 0000000 0000000 agentencryption/.github/ 0000775 0000000 0000000 00000000000 14143644277 0015643 5 ustar 00root root 0000000 0000000 agentencryption/.github/workflows/ 0000775 0000000 0000000 00000000000 14143644277 0017700 5 ustar 00root root 0000000 0000000 agentencryption/.github/workflows/whl.yml 0000664 0000000 0000000 00000001166 14143644277 0021221 0 ustar 00root root 0000000 0000000 name: Whl Action
on:
release:
types: [published]
workflow_dispatch:
jobs:
deploy:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: '3.x'
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install build
- name: Build package
run: python -m build
- name: Publish package
uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
with:
user: __token__
password: ${{ secrets.PYPI_API_TOKEN }} agentencryption/.gitignore 0000664 0000000 0000000 00000003407 14143644277 0016277 0 ustar 00root root 0000000 0000000 # Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class
# C extensions
*.so
# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
pip-wheel-metadata/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST
# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec
# Installer logs
pip-log.txt
pip-delete-this-directory.txt
# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py,cover
.hypothesis/
.pytest_cache/
# Translations
*.mo
*.pot
# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal
# Flask stuff:
instance/
.webassets-cache
# Scrapy stuff:
.scrapy
# Sphinx documentation
docs/_build/
# PyBuilder
target/
# Jupyter Notebook
.ipynb_checkpoints
# IPython
profile_default/
ipython_config.py
# pyenv
.python-version
# pipenv
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
# However, in case of collaboration, if having platform-specific dependencies or dependencies
# having no cross-platform support, pipenv may install dependencies that don't work, or not
# install all needed dependencies.
#Pipfile.lock
# PEP 582; used by e.g. github.com/David-OConnor/pyflow
__pypackages__/
# Celery stuff
celerybeat-schedule
celerybeat.pid
# SageMath parsed files
*.sage.py
# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/
# Spyder project settings
.spyderproject
.spyproject
# Rope project settings
.ropeproject
# mkdocs documentation
/site
# mypy
.mypy_cache/
.dmypy.json
dmypy.json
# Pyre type checker
.pyre/
agentencryption/LICENSE 0000664 0000000 0000000 00000026135 14143644277 0015317 0 ustar 00root root 0000000 0000000 Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
1. Definitions.
"License" shall mean the terms and conditions for use, reproduction,
and distribution as defined by Sections 1 through 9 of this document.
"Licensor" shall mean the copyright owner or entity authorized by
the copyright owner that is granting the License.
"Legal Entity" shall mean the union of the acting entity and all
other entities that control, are controlled by, or are under common
control with that entity. For the purposes of this definition,
"control" means (i) the power, direct or indirect, to cause the
direction or management of such entity, whether by contract or
otherwise, or (ii) ownership of fifty percent (50%) or more of the
outstanding shares, or (iii) beneficial ownership of such entity.
"You" (or "Your") shall mean an individual or Legal Entity
exercising permissions granted by this License.
"Source" form shall mean the preferred form for making modifications,
including but not limited to software source code, documentation
source, and configuration files.
"Object" form shall mean any form resulting from mechanical
transformation or translation of a Source form, including but
not limited to compiled object code, generated documentation,
and conversions to other media types.
"Work" shall mean the work of authorship, whether in Source or
Object form, made available under the License, as indicated by a
copyright notice that is included in or attached to the work
(an example is provided in the Appendix below).
"Derivative Works" shall mean any work, whether in Source or Object
form, that is based on (or derived from) the Work and for which the
editorial revisions, annotations, elaborations, or other modifications
represent, as a whole, an original work of authorship. For the purposes
of this License, Derivative Works shall not include works that remain
separable from, or merely link (or bind by name) to the interfaces of,
the Work and Derivative Works thereof.
"Contribution" shall mean any work of authorship, including
the original version of the Work and any modifications or additions
to that Work or Derivative Works thereof, that is intentionally
submitted to Licensor for inclusion in the Work by the copyright owner
or by an individual or Legal Entity authorized to submit on behalf of
the copyright owner. For the purposes of this definition, "submitted"
means any form of electronic, verbal, or written communication sent
to the Licensor or its representatives, including but not limited to
communication on electronic mailing lists, source code control systems,
and issue tracking systems that are managed by, or on behalf of, the
Licensor for the purpose of discussing and improving the Work, but
excluding communication that is conspicuously marked or otherwise
designated in writing by the copyright owner as "Not a Contribution."
"Contributor" shall mean Licensor and any individual or Legal Entity
on behalf of whom a Contribution has been received by Licensor and
subsequently incorporated within the Work.
2. Grant of Copyright License. Subject to the terms and conditions of
this License, each Contributor hereby grants to You a perpetual,
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
copyright license to reproduce, prepare Derivative Works of,
publicly display, publicly perform, sublicense, and distribute the
Work and such Derivative Works in Source or Object form.
3. Grant of Patent License. Subject to the terms and conditions of
this License, each Contributor hereby grants to You a perpetual,
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
(except as stated in this section) patent license to make, have made,
use, offer to sell, sell, import, and otherwise transfer the Work,
where such license applies only to those patent claims licensable
by such Contributor that are necessarily infringed by their
Contribution(s) alone or by combination of their Contribution(s)
with the Work to which such Contribution(s) was submitted. If You
institute patent litigation against any entity (including a
cross-claim or counterclaim in a lawsuit) alleging that the Work
or a Contribution incorporated within the Work constitutes direct
or contributory patent infringement, then any patent licenses
granted to You under this License for that Work shall terminate
as of the date such litigation is filed.
4. Redistribution. You may reproduce and distribute copies of the
Work or Derivative Works thereof in any medium, with or without
modifications, and in Source or Object form, provided that You
meet the following conditions:
(a) You must give any other recipients of the Work or
Derivative Works a copy of this License; and
(b) You must cause any modified files to carry prominent notices
stating that You changed the files; and
(c) You must retain, in the Source form of any Derivative Works
that You distribute, all copyright, patent, trademark, and
attribution notices from the Source form of the Work,
excluding those notices that do not pertain to any part of
the Derivative Works; and
(d) If the Work includes a "NOTICE" text file as part of its
distribution, then any Derivative Works that You distribute must
include a readable copy of the attribution notices contained
within such NOTICE file, excluding those notices that do not
pertain to any part of the Derivative Works, in at least one
of the following places: within a NOTICE text file distributed
as part of the Derivative Works; within the Source form or
documentation, if provided along with the Derivative Works; or,
within a display generated by the Derivative Works, if and
wherever such third-party notices normally appear. The contents
of the NOTICE file are for informational purposes only and
do not modify the License. You may add Your own attribution
notices within Derivative Works that You distribute, alongside
or as an addendum to the NOTICE text from the Work, provided
that such additional attribution notices cannot be construed
as modifying the License.
You may add Your own copyright statement to Your modifications and
may provide additional or different license terms and conditions
for use, reproduction, or distribution of Your modifications, or
for any such Derivative Works as a whole, provided Your use,
reproduction, and distribution of the Work otherwise complies with
the conditions stated in this License.
5. Submission of Contributions. Unless You explicitly state otherwise,
any Contribution intentionally submitted for inclusion in the Work
by You to the Licensor shall be under the terms and conditions of
this License, without any additional terms or conditions.
Notwithstanding the above, nothing herein shall supersede or modify
the terms of any separate license agreement you may have executed
with Licensor regarding such Contributions.
6. Trademarks. This License does not grant permission to use the trade
names, trademarks, service marks, or product names of the Licensor,
except as required for reasonable and customary use in describing the
origin of the Work and reproducing the content of the NOTICE file.
7. Disclaimer of Warranty. Unless required by applicable law or
agreed to in writing, Licensor provides the Work (and each
Contributor provides its Contributions) on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
implied, including, without limitation, any warranties or conditions
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
PARTICULAR PURPOSE. You are solely responsible for determining the
appropriateness of using or redistributing the Work and assume any
risks associated with Your exercise of permissions under this License.
8. Limitation of Liability. In no event and under no legal theory,
whether in tort (including negligence), contract, or otherwise,
unless required by applicable law (such as deliberate and grossly
negligent acts) or agreed to in writing, shall any Contributor be
liable to You for damages, including any direct, indirect, special,
incidental, or consequential damages of any character arising as a
result of this License or out of the use or inability to use the
Work (including but not limited to damages for loss of goodwill,
work stoppage, computer failure or malfunction, or any and all
other commercial damages or losses), even if such Contributor
has been advised of the possibility of such damages.
9. Accepting Warranty or Additional Liability. While redistributing
the Work or Derivative Works thereof, You may choose to offer,
and charge a fee for, acceptance of support, warranty, indemnity,
or other liability obligations and/or rights consistent with this
License. However, in accepting such obligations, You may act only
on Your own behalf and on Your sole responsibility, not on behalf
of any other Contributor, and only if You agree to indemnify,
defend, and hold each Contributor harmless for any liability
incurred by, or claims asserted against, such Contributor by reason
of your accepting any such warranty or additional liability.
END OF TERMS AND CONDITIONS
APPENDIX: How to apply the Apache License to your work.
To apply the Apache License to your work, attach the following
boilerplate notice, with the fields enclosed by brackets "[]"
replaced with your own identifying information. (Don't include
the brackets!) The text should be enclosed in the appropriate
comment syntax for the file format. We also recommend that a
file or class name and description of purpose be included on the
same "printed page" as the copyright notice for easier
identification within third-party archives.
Copyright [yyyy] [name of copyright owner]
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
agentencryption/README.md 0000664 0000000 0000000 00000007063 14143644277 0015570 0 ustar 00root root 0000000 0000000 # AgentEncryption
飞桨模型加密库 - EAP阶段
在分发加密模型时，请务必进行不限于以下的操作方可避免轻易破解
- [x] 加密深度学习框架以及其相关的Python库，至少为pyd格式
- [x] 加密AgentEncryption库，至少为pyd格式
- [x] 限制用户只能使用你提供的Python环境
> 飞桨推理模型加密可参考[从内存中加载加密飞桨推理模型](./example/paddlepaddle)
> 1. [加密飞桨模型](./example/paddlepaddle/make_model.py)
> 2. [借助PPQI推理加密的飞桨模型](./example/paddlepaddle/test_model_ppqi.py)
> 3. [借助PaddleInference推理加密的飞桨模型](./example/paddlepaddle/test_model_paddleinference.py)
## 快速使用
* 如下是几个简单的示例代码：
* 数据加密：
```python
# 导入 RSA 加密器
from agentenc import RSAEncryptor
# 初始化 RSA 加密器
encryptor = RSAEncryptor(bits=1024)
# 构建原始数据
pure_data = {
'int': 100,
'float': 0.01,
'str': 'hello',
'list': [100, 0.01, 'hello']
}
# 数据加密
key = encryptor.encode(
input=pure_data,
output='out',
format='pkl',
keys_saving_path='key'
)
# 打印随机生成的密钥
print(key)
'''
{
'private_key': b'-----BEGIN RSA PRIVATE KEY-----\nMIICXAIBAAKBgQDQq3mzdfDYjg8ool1Jl5WrmFAkJWarokIQAzq/3wT+cbNUy/zv\nqxHCn7bYsifvx5nLnfCL7cm3BVygnB4clP8p6EAlO8KcocC/6WfCyTW5gw23z5Tc\ntzSzAERtNTGwDst3RAnEDeJ2crNi/xSs2Wa6k1bNkxNFUehEd4ExpYFwLQIDAQAB\nAoGAWmYG9aOBANfeIkzgnBqSyQFVqqsXRQConPZBM9EigTZxqakrfQq/yXBWjp7z\npTFz452bEYukqOimPaAUfW5g8ZYxndN2G1ctWMPg3KrJ++7FYFJJhJ223IOmSmB4\nKf2FyVjBNUTzKchBdxHcLqAQX1DNXnWrJ5cWc2pS0olz+BMCQQDTZ8hqIT1UPclk\noPTX6gC+874D+GNa5r4CsMMiUZNrN7hCXEwZJ+3kR14vcJkDd2QVg8cR6VSxP9Vg\nBg5XuD2jAkEA/K/3mtKMDhX5HjNUfoBzLBSJreAVMOcGGrLCLFob+8t64456SQ8I\nOKABsocpKaRp+s3loi6obGoRwIQ5PBTX7wJBAMp60pfj8kunSidZimjqtYAvEEXZ\nN8Au1Lra9mr+WwYMPi1BHZnShqVoPauOWt/ZEEETEC31n6qNCx+HbWFTE6UCQEfF\n16ezPDLYDO2GGO7hn1Ua9ExeBMbiJ/q3Ya3lXmNz1ZEDLDrKOUSUNkc2WvvIBo5F\no38gj5hTvH0ZUYR+SyMCQEnnTVZXhtU7jtyOfEXsTqUtZIVo01sJ5SoekGrnZ0Ht\n8PxyjyOc5IFWieKGUfZQ3ZdKDm95dvAyKpkDa8tqMCM=\n-----END RSA PRIVATE KEY-----',
'public_key': b'-----BEGIN PUBLIC KEY-----\nMIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDQq3mzdfDYjg8ool1Jl5WrmFAk\nJWarokIQAzq/3wT+cbNUy/zvqxHCn7bYsifvx5nLnfCL7cm3BVygnB4clP8p6EAl\nO8KcocC/6WfCyTW5gw23z5TctzSzAERtNTGwDst3RAnEDeJ2crNi/xSs2Wa6k1bN\nkxNFUehEd4ExpYFwLQIDAQAB\n-----END PUBLIC KEY-----'
}
'''
# 使用解密函数对文件进行解密
data = RSAEncryptor.decode(
inp='out.pkl',
private_key=key['private_key']
)
# 输入与输出数据对比
print(data == pure_data)
'''
True
'''
```
* 数据解密：
```python
# 导入基础加密器
from agentenc import Encryptor
# 读取私钥
private_key = open('key.PRIVATE', 'rb').read()
# 使用解密函数对文件进行解密
data = Encryptor.decode(
inp='out.pkl',
private_key=private_key
)
# 输入与输出数据对比
print(data)
'''
{
'int': 100,
'float': 0.01,
'str': 'hello',
'list': [100, 0.01, 'hello']
}
'''
```
## Contact us
Email : [agentmaker@163.com]()
QQ Group : 1005109853
agentencryption/Server/ 0000775 0000000 0000000 00000000000 14143644277 0015551 5 ustar 00root root 0000000 0000000 agentencryption/Server/app.go 0000664 0000000 0000000 00000000321 14143644277 0016654 0 ustar 00root root 0000000 0000000 package main
import (
"agentencryption/controller"
"log"
)
func main() {
router := controller.InitServer()
err := router.Run(":6666")
if err != nil {
log.Fatalln("Failed to start server: ", err)
}
}
agentencryption/Server/assets/ 0000775 0000000 0000000 00000000000 14143644277 0017053 5 ustar 00root root 0000000 0000000 agentencryption/Server/assets/1.pdiparams 0000664 0000000 0000000 00000000072 14143644277 0021114 0 ustar 00root root 0000000 0000000 agentencryption/Server/assets/1.pdiparams.info 0000664 0000000 0000000 00000000224 14143644277 0022045 0 ustar 00root root 0000000 0000000 }q (X linear_0.w_0q}q(X structured_nameqX weightqX
stop_gradientqX trainablequX linear_0.b_0q}q(hX biasq hhuu. agentencryption/Server/assets/1.pdmodel 0000664 0000000 0000000 00000032553 14143644277 0020571 0 ustar 00root root 0000000 0000000
^
linear_0.w_0
linear_0.b_0
+
linear_1.tmp_1
+
linear_1.tmp_0
input_1
fetch
feed "
Xfeed
Outinput_1feed"
op_role "
op_callstackBd File "/Users/zhanghongji/PycharmProjects/AgentEncryption/test/make_model.py", line 13, in BK paddle.jit.save(fc, path="./sample_model/model1", input_spec=[input_1])B- File "", line 2, in saveB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 40, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/jit.py", line 864, in saveB clip_extra=False)B= File "", line 2, in save_inference_modelB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 238, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/io.py", line 1432, in save_inference_modelB4 prepend_feed_ops(main_program, feeded_var_names)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/io.py", line 1224, in prepend_feed_opsB attrs={'col': i})B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 3263, in _prepend_opB$ attrs=kwargs.get("attrs", None))B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2220, in __init__B+ for frame in traceback.extract_stack():"
col "
op_device* "
op_role_var"
with_quant_attrP "
op_namescope*/"
Xinput_1
Y linear_0.w_0
Outlinear_1.tmp_0 matmul_v2"
with_quant_attrP "
use_mkldnnP "
op_namescope*/"
op_role_var"
op_role "
mkldnn_data_type*float32"
trans_yP "
trans_xP "
op_device* "
op_callstackBd File "/Users/zhanghongji/PycharmProjects/AgentEncryption/test/make_model.py", line 13, in BK paddle.jit.save(fc, path="./sample_model/model1", input_spec=[input_1])B- File "", line 2, in saveB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 40, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/jit.py", line 754, in saveB6 concrete_program = static_forward.concrete_programB File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 458, in concrete_programBD return self.concrete_program_specify_input_spec(input_spec=None)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 496, in concrete_program_specify_input_specB *desired_input_spec)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 406, in get_concrete_programBL concrete_program, partial_program_layer = self._program_cache[cache_key]B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 723, in __getitem__B/ self._caches[item] = self._build_once(item)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 714, in _build_onceB **cache_key.kwargs)B6 File "", line 2, in from_func_specB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 40, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 662, in from_func_specB" outputs = static_func(*inputs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/nn/layer/common.py", line 172, in forwardB@ x=input, weight=self.weight, bias=self.bias, name=self.name)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/nn/functional/common.py", line 1492, in linearBG type='matmul_v2', inputs=inputs, outputs={'Out': tmp}, attrs=attrs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/layer_helper.py", line 43, in append_opBG return self.main_program.current_block().append_op(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 3180, in append_opB$ attrs=kwargs.get("attrs", None))B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2220, in __init__B+ for frame in traceback.extract_stack():"
Xlinear_1.tmp_0
Y linear_0.b_0
Outlinear_1.tmp_1elementwise_add"
op_role_var"
use_mkldnnP "
with_quant_attrP "
op_device* "
op_callstackBd File "/Users/zhanghongji/PycharmProjects/AgentEncryption/test/make_model.py", line 13, in BK paddle.jit.save(fc, path="./sample_model/model1", input_spec=[input_1])B- File "", line 2, in saveB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 40, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/jit.py", line 754, in saveB6 concrete_program = static_forward.concrete_programB File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 458, in concrete_programBD return self.concrete_program_specify_input_spec(input_spec=None)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 496, in concrete_program_specify_input_specB *desired_input_spec)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 406, in get_concrete_programBL concrete_program, partial_program_layer = self._program_cache[cache_key]B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 723, in __getitem__B/ self._caches[item] = self._build_once(item)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 714, in _build_onceB **cache_key.kwargs)B6 File "", line 2, in from_func_specB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 40, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 662, in from_func_specB" outputs = static_func(*inputs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/nn/layer/common.py", line 172, in forwardB@ x=input, weight=self.weight, bias=self.bias, name=self.name)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/nn/functional/common.py", line 1500, in linearB% attrs={'axis': len(x.shape) - 1})B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/layer_helper.py", line 43, in append_opBG return self.main_program.current_block().append_op(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 3180, in append_opB$ attrs=kwargs.get("attrs", None))B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2220, in __init__B+ for frame in traceback.extract_stack():"
op_namescope*/"
mkldnn_data_type*float32"
y_data_format* "
Scale_y% ?"
use_quantizerP "
Scale_out% ?"
axis "
op_role "
Scale_x% ?"
x_data_format* "
Xlinear_1.tmp_1
Outfetchfetch"
op_role "
op_callstackBd File "/Users/zhanghongji/PycharmProjects/AgentEncryption/test/make_model.py", line 13, in BK paddle.jit.save(fc, path="./sample_model/model1", input_spec=[input_1])B- File "", line 2, in saveB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 40, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/dygraph/jit.py", line 864, in saveB clip_extra=False)B= File "", line 2, in save_inference_modelB B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in __impl__B( return wrapped_func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 238, in __impl__B return func(*args, **kwargs)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/io.py", line 1433, in save_inference_modelB3 append_fetch_ops(main_program, fetch_var_names)B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/io.py", line 1241, in append_fetch_opsB attrs={'col': i})B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 3180, in append_opB$ attrs=kwargs.get("attrs", None))B File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2220, in __init__B+ for frame in traceback.extract_stack():"
col "
op_device* "
op_role_var"
with_quant_attrP "
op_namescope*/"Иz*
equal
not_equal
greater_equal
less_equal
generate_proposal_labels
matrix_nms
yolo_box
generate_proposals_v2
elementwise_add
distribute_fpn_proposals
elementwise_div
elementwise_max
elementwise_min
elementwise_pow
fusion_gru
momentum
collect_fpn_proposals
softplus
gather
depthwise_conv2d_transpose
fill_constant
allclose
arg_max
elementwise_sub
affine_grid
unique
arg_min
coalesce_tensor
conv3d_transpose
leaky_relu
instance_norm
clip
flip
data_norm
conv3d
greater_than
roi_pool
conv2d_transpose
generate_proposals
conv2d
conv_transpose
hard_shrink
grid_sampler
cudnn_lstm
dequantize
lookup_table_v2
*
$fake_channel_wise_dequantize_max_abs
adam
(
"fake_channel_wise_quantize_abs_max
depthwise_conv2d
"
moving_average_abs_max_scale
rank_attention
lookup_table
trace
p_norm
pixel_shuffle
linspace
print
elementwise_floordiv
elementwise_mod
for_pybind_test__
matmul
requantize
lamb
cumsum
gaussian_random
roi_align
warpctc
elementwise_mul
roll
set_value
unique_consecutive
quantize
less_than
softmax_with_cross_entropy agentencryption/Server/common/ 0000775 0000000 0000000 00000000000 14143644277 0017041 5 ustar 00root root 0000000 0000000 agentencryption/Server/common/common.go 0000664 0000000 0000000 00000002673 14143644277 0020670 0 ustar 00root root 0000000 0000000 package common
import (
"bytes"
"crypto/rand"
"crypto/rsa"
"crypto/x509"
"encoding/pem"
"log"
)
func PartEncrypt(PublicKey *rsa.PublicKey, File []byte) (data []byte, err error) {
FileSize := len(File)
OnceStep := PublicKey.Size() - 11
FileOffset := 0
var FileBuffer bytes.Buffer
// Model Encrypt
for FileOffset < FileSize {
EndIndex := FileOffset + OnceStep
if EndIndex > FileSize { // 当EndIndex比文件大的时候
EndIndex = FileSize
}
// 分段加密
Part, err := rsa.EncryptPKCS1v15(rand.Reader, PublicKey, File[FileOffset:EndIndex])
if err != nil {
return nil, err
}
FileBuffer.Write(Part)
FileOffset = EndIndex
}
return FileBuffer.Bytes(), nil
}
func CreateRsaKey(bits int) (prikey string, pubkey string, err error) {
// 生成私钥文件
privateKey, err := rsa.GenerateKey(rand.Reader, bits)
if err != nil {
log.Println("CreateKeyError: ", err)
return "", "", err
}
derStream := x509.MarshalPKCS1PrivateKey(privateKey)
block := &pem.Block{
Type: "RSA PRIVATE KEY",
Bytes: derStream,
}
// 向内存输出私钥
prikey = string(pem.EncodeToMemory(block))
//生成公钥文件
publicKey := &privateKey.PublicKey
derPkix, err := x509.MarshalPKIXPublicKey(publicKey)
if err != nil {
log.Println("CreateKeyError: ", err)
return "", "", err
}
block = &pem.Block{
Type: "PUBLIC KEY",
Bytes: derPkix,
}
// 向内存输出公钥
pubkey = string(pem.EncodeToMemory(block))
return
}
agentencryption/Server/controller/ 0000775 0000000 0000000 00000000000 14143644277 0017734 5 ustar 00root root 0000000 0000000 agentencryption/Server/controller/init.go 0000664 0000000 0000000 00000000447 14143644277 0021233 0 ustar 00root root 0000000 0000000 package controller
import (
"agentencryption/core"
"github.com/gin-contrib/cors"
"github.com/gin-gonic/gin"
)
func InitServer() *gin.Engine {
route := gin.Default()
route.Use(cors.Default())
route.POST("/Register", core.Register)
route.POST("/GetModel", core.GetModel)
return route
}
agentencryption/Server/core/ 0000775 0000000 0000000 00000000000 14143644277 0016501 5 ustar 00root root 0000000 0000000 agentencryption/Server/core/core.go 0000664 0000000 0000000 00000007540 14143644277 0017766 0 ustar 00root root 0000000 0000000 package core
import (
"agentencryption/common"
"agentencryption/model"
"crypto/rsa"
"crypto/x509"
"encoding/base64"
"encoding/pem"
"fmt"
"io/ioutil"
"log"
"net/http"
"github.com/gin-gonic/gin"
"github.com/gin-gonic/gin/binding"
uuid "github.com/satori/go.uuid"
)
func JSONFail(ctx *gin.Context, retCode int, errorMsg string) {
ctx.JSON(http.StatusOK, gin.H{
"Code": retCode,
"Status": "Fail",
"RequestID": uuid.NewV4(),
"ErrorMsg": errorMsg,
})
}
func JSONSuccess(ctx *gin.Context, result interface{}) {
ctx.JSON(http.StatusOK, gin.H{
"Code": 0,
"Status": "Success",
"RequestID": uuid.NewV4(),
"Result": result,
})
}
func Register(ctx *gin.Context) {
var RegParams model.RegisterParams
err := ctx.ShouldBindBodyWith(&RegParams, binding.JSON)
if err != nil {
JSONFail(ctx, -1, err.Error())
log.Println("RequestParamsError: ", err)
return
}
if RegParams.Username != "TestUser" || RegParams.Password != "test123" {
JSONFail(ctx, -1, "Auth Error: Username or Password doesn't match.")
log.Println("Auth Error: ", "UNAME OR PASSWD DOESN'T MATCH")
return
}
prikey, pubkey, err := common.CreateRsaKey(2048) // Default: 8192
if err != nil {
JSONFail(ctx, -2, err.Error())
log.Println("CreateRSAError: ", err)
return
}
var Response model.RegisterResp
Response.RSAPrivate = prikey
err = ioutil.WriteFile("./keys/"+fmt.Sprint(RegParams.Username)+".AgentRSAPriv", []byte(prikey), 0700)
if err != nil {
JSONFail(ctx, -2, err.Error())
log.Println("CreateRSAError: ", err)
return
}
err = ioutil.WriteFile("./keys/"+fmt.Sprint(RegParams.Username)+".AgentRSAPub", []byte(pubkey), 0700)
if err != nil {
JSONFail(ctx, -2, err.Error())
log.Println("CreateRSAError: ", err)
return
}
JSONSuccess(ctx, Response)
}
func GetModel(ctx *gin.Context) {
var GetParams model.GetModelParams
err := ctx.ShouldBindBodyWith(&GetParams, binding.JSON)
if err != nil {
JSONFail(ctx, -1, err.Error())
log.Println("RequestParamsError: ", err)
return
}
if GetParams.AccountInfo.Username != "TestUser" || GetParams.AccountInfo.Password != "test123" {
JSONFail(ctx, -1, "Auth Error: Username or Password doesn't match.")
log.Println("Auth Error: ", "UNAME OR PASSWD DOESN'T MATCH")
return
}
ModelID := fmt.Sprint(GetParams.ID)
ModelParams, err := ioutil.ReadFile("./assets/" + ModelID + ".pdiparams")
if err != nil {
JSONFail(ctx, -3, err.Error())
log.Println(err)
return
}
ModelParamInfo, err := ioutil.ReadFile("./assets/" + ModelID + ".pdiparams.info")
if err != nil {
JSONFail(ctx, -3, err.Error())
log.Println(err)
return
}
Model, err := ioutil.ReadFile("./assets/" + ModelID + ".pdmodel")
if err != nil {
JSONFail(ctx, -3, err.Error())
log.Println(err)
return
}
PublicKeyFile, err := ioutil.ReadFile("./keys/" + GetParams.AccountInfo.Username + ".AgentRSAPub")
if err != nil {
JSONFail(ctx, -4, err.Error())
log.Println(err)
return
}
Block, _ := pem.Decode(PublicKeyFile)
KeyInterface, err := x509.ParsePKIXPublicKey(Block.Bytes)
if err != nil {
JSONFail(ctx, -4, err.Error())
log.Println(err)
return
}
PublicKey := KeyInterface.(*rsa.PublicKey)
Model, err = common.PartEncrypt(PublicKey, Model)
if err != nil {
JSONFail(ctx, -5, err.Error())
log.Println(err)
return
}
ModelParams, err = common.PartEncrypt(PublicKey, ModelParams)
if err != nil {
JSONFail(ctx, -5, err.Error())
log.Println(err)
return
}
ModelParamInfo, err = common.PartEncrypt(PublicKey, ModelParamInfo)
if err != nil {
JSONFail(ctx, -5, err.Error())
log.Println(err)
return
}
var ModelData model.ModelResp
ModelData.ID = GetParams.ID
ModelData.Model = base64.StdEncoding.EncodeToString(Model)
ModelData.ModelParam = base64.StdEncoding.EncodeToString(ModelParams)
ModelData.ModelParamInfo = base64.StdEncoding.EncodeToString(ModelParamInfo)
JSONSuccess(ctx, ModelData)
}
agentencryption/Server/go.mod 0000664 0000000 0000000 00000001575 14143644277 0016667 0 ustar 00root root 0000000 0000000 module agentencryption
go 1.17
require (
github.com/gin-contrib/cors v1.3.1
github.com/gin-gonic/gin v1.5.0
)
require (
github.com/gin-contrib/sse v0.1.0 // indirect
github.com/go-playground/locales v0.12.1 // indirect
github.com/go-playground/universal-translator v0.16.0 // indirect
github.com/golang/protobuf v1.3.2 // indirect
github.com/json-iterator/go v1.1.7 // indirect
github.com/leodido/go-urn v1.1.0 // indirect
github.com/mattn/go-isatty v0.0.9 // indirect
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 // indirect
github.com/satori/go.uuid v1.2.0 // indirect
github.com/ugorji/go/codec v1.1.7 // indirect
golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a // indirect
gopkg.in/go-playground/validator.v9 v9.29.1 // indirect
gopkg.in/yaml.v2 v2.2.2 // indirect
)
agentencryption/Server/go.sum 0000664 0000000 0000000 00000011415 14143644277 0016706 0 ustar 00root root 0000000 0000000 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/gin-contrib/cors v1.3.1 h1:doAsuITavI4IOcd0Y19U4B+O0dNWihRyX//nn4sEmgA=
github.com/gin-contrib/cors v1.3.1/go.mod h1:jjEJ4268OPZUcU7k9Pm653S7lXUGcqMADzFA61xsmDk=
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
github.com/gin-gonic/gin v1.5.0 h1:fi+bqFAx/oLK54somfCtEZs9HeH1LHVoEPUgARpTqyc=
github.com/gin-gonic/gin v1.5.0/go.mod h1:Nd6IXA8m5kNZdNEHMBd93KT+mdY3+bewLgRvmCsR2Do=
github.com/go-playground/locales v0.12.1 h1:2FITxuFt/xuCNP1Acdhv62OzaCiviiE4kotfhkmOqEc=
github.com/go-playground/locales v0.12.1/go.mod h1:IUMDtCfWo/w/mtMfIE/IG2K+Ey3ygWanZIBtBW0W2TM=
github.com/go-playground/universal-translator v0.16.0 h1:X++omBR/4cE2MNg91AoC3rmGrCjJ8eAeUP/K/EKx4DM=
github.com/go-playground/universal-translator v0.16.0/go.mod h1:1AnU7NaIRDWWzGEKwgtJRd2xk99HeFyHw3yid4rvQIY=
github.com/golang/protobuf v1.3.2 h1:6nsPYzhq5kReh6QImI3k5qWzO4PEbvbIW2cwSfR/6xs=
github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
github.com/json-iterator/go v1.1.7 h1:KfgG9LzI+pYjr4xvmz/5H4FXjokeP+rlHLhv3iH62Fo=
github.com/json-iterator/go v1.1.7/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
github.com/leodido/go-urn v1.1.0 h1:Sm1gr51B1kKyfD2BlRcLSiEkffoG96g6TPv6eRoEiB8=
github.com/leodido/go-urn v1.1.0/go.mod h1:+cyI34gQWZcE1eQU7NVgKkkzdXDQHr1dBMtdAPozLkw=
github.com/mattn/go-isatty v0.0.9 h1:d5US/mDsogSGW37IV293h//ZFaeajb69h+EHFsv2xGg=
github.com/mattn/go-isatty v0.0.9/go.mod h1:YNRxwqDuOph6SZLI9vUUz6OYw3QyUt7WiY2yME+cCiQ=
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLDQ0W1YjYsBW+p8U2u7vzgW2SQVmlNazg=
github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/satori/go.uuid v1.2.0 h1:0uYX9dsZ2yD7q2RtLRtPSdGDWzjeM3TbMJP9utgA0ww=
github.com/satori/go.uuid v1.2.0/go.mod h1:dA0hQrYB0VpLJoorglMZABFdXlWrHn1NEOzdhQKdks0=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
github.com/ugorji/go v1.1.7 h1:/68gy2h+1mWMrwZFeD1kQialdSzAb432dtpeJ42ovdo=
github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
github.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs=
github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=
golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a h1:aYOabOQFp6Vj6W1F80affTUvO9UxmJRx8K0gsfABByQ=
golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/go-playground/assert.v1 v1.2.1 h1:xoYuJVE7KT85PYWrN730RguIQO0ePzVRfFMXadIrXTM=
gopkg.in/go-playground/assert.v1 v1.2.1/go.mod h1:9RXL0bg/zibRAgZUYszZSwO/z8Y/a8bDuhia5mkpMnE=
gopkg.in/go-playground/validator.v9 v9.29.1 h1:SvGtYmN60a5CVKTOzMSyfzWDeZRxRuGvRQyEAKbw1xc=
gopkg.in/go-playground/validator.v9 v9.29.1/go.mod h1:+c9/zcJMFNgbLvly1L1V+PpxWdVbfP1avr/N00E2vyQ=
gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
agentencryption/Server/keys/ 0000775 0000000 0000000 00000000000 14143644277 0016524 5 ustar 00root root 0000000 0000000 agentencryption/Server/keys/TestUser.AgentRSAPriv 0000664 0000000 0000000 00000003213 14143644277 0022470 0 ustar 00root root 0000000 0000000 -----BEGIN RSA PRIVATE KEY-----
MIIEowIBAAKCAQEApI5g5z/9TbJh/SVHMGSGwXCMcycgN3xoq4ni4dFvdI+t0Xmg
qdszayPlKjTUQr/MrILlRFXW7U6xXGL4gMCY9YRbz/cHjfYAfk4+xVSN2PBO08bV
B7pf9njBckgkGlm6SOAoePk9c5t08qkOYoGh0pqJQsVo8nIu75kO5B6oh1Lj0YCG
yV0zrOT6fWawjkNG6n/zUzKmXCXx5FxRW6EH0n43kYg7gacqjCfmaClU7C/fMCq1
61lKHhaHfBV1Zpg/aHY67j6UCBvAAypSdl/ML260IQNv0tlQgLRdz+eBDNiRQIWW
PWJkAV9y1iquvzKJR1VJM26Xkzc/9MxaWeGXvQIDAQABAoIBACeyeUoQM8lEmUL6
dTwFZETSI++7ppf8GTwYnXUgkySpbKM2Fj4jtTnJhQUzjWk03+lWAn0DwD+rVjoV
XiGhVMUV0xhOXbIvgFr4AUiVZ0rIXhvUwZuFqJTK+M/a+2qn7rqZ6OmWJu72t7W6
ds/oueMWiLpk6TZjuVLvt5nsk3dhcq7Qsap+8hzwSp4M9D1yaixFbv69jte86SXd
msM4QJ1LF1IOPJ7eU5/k/tNDBgMtB8CZqhSlbt9Btfdt3J/2GIIuxQIJK5nxmjZI
wKzRtkw6cz3Y+e7Az5/QgWDIasRDcLDNhou3vl5a1gJN4HZRjjAVlpiavFayEKfZ
0cXry4ECgYEAyqNGUnC1/1PlVejjoV/j2mwi78NVcMTlZFQk+D61f3WQ/jCoIQ8O
crEDqu4kkEMhsyJz3oI3WMB4TvwEID3WS7jbRDLWAmv2jezkUeF32zMWOFXGfnGx
Hnm3hgbJmJQlsAYrlMA2DKbApMISf4ocKPrlCSF8FaZ5QNVrEeFYP60CgYEAz+PZ
oDt/Q7Mmk4pRWCMXoiTLQ0q/GTNyaS7S2T7PbPDMoqRfO/KhOjNIQ8NOrnlFcACb
oJxnvUe3dfrnt7VTzevJnFVtNKXwNkQ2JpNgr3C5Qdpj+Egk4tY/LfjzwCyPLTK8
2qPFW146V5jnyhl+1zArncm1SRESNPq+0NixelECgYAlZOgPQplzWsoRhn6rQwTL
Q/VWhZd4zfz+/MAtx3S65rkxD02XfC3QYHYqzvOUY68OXadAQotwJ2SMPtmMEZ1A
MAbWmQMBQ2cmXbbsnUz++Hh7QOj2YIis9Q03AH2nWH9z8k+MjcPrpcIbJW1Ad/4m
WdZ2TuWVITlzC29Mx2b/qQKBgQCkHxKhUDdhoB61Xf5oIZQ32uppnqoNYDNT3LZy
8gZZ0CUNxcy1jBBC3VeQv7DvRyhNiv1SKAgyQ1sQFlw3CT03FY3eEQcdCGACx4kk
nTc4ajqnRZ2k1nfgaRoJ1j/BGKyUFzlHNwHI5zQSEdH0GHqtsgbNFAA9xe9kygw4
3soGcQKBgFoi/f61G1ZTl+RQf+Qrzo3J3C/Q7ivA8XdLoshtXttp13IwEh3I+vxx
QLMaCzC2h8lBhyEuTlGnZLIQ1S7gunglmh7i/AiRxPrYwr0zI6EmkYK5GzE3chN3
DqwvN7Hk4OIXJplQGPIWtog9eZG+wsz4WN6AWx7N7tDaOL+Z+4Nq
-----END RSA PRIVATE KEY-----
agentencryption/Server/keys/TestUser.AgentRSAPub 0000664 0000000 0000000 00000000703 14143644277 0022277 0 ustar 00root root 0000000 0000000 -----BEGIN PUBLIC KEY-----
MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEApI5g5z/9TbJh/SVHMGSG
wXCMcycgN3xoq4ni4dFvdI+t0XmgqdszayPlKjTUQr/MrILlRFXW7U6xXGL4gMCY
9YRbz/cHjfYAfk4+xVSN2PBO08bVB7pf9njBckgkGlm6SOAoePk9c5t08qkOYoGh
0pqJQsVo8nIu75kO5B6oh1Lj0YCGyV0zrOT6fWawjkNG6n/zUzKmXCXx5FxRW6EH
0n43kYg7gacqjCfmaClU7C/fMCq161lKHhaHfBV1Zpg/aHY67j6UCBvAAypSdl/M
L260IQNv0tlQgLRdz+eBDNiRQIWWPWJkAV9y1iquvzKJR1VJM26Xkzc/9MxaWeGX
vQIDAQAB
-----END PUBLIC KEY-----
agentencryption/Server/model/ 0000775 0000000 0000000 00000000000 14143644277 0016651 5 ustar 00root root 0000000 0000000 agentencryption/Server/model/model.go 0000664 0000000 0000000 00000000760 14143644277 0020303 0 ustar 00root root 0000000 0000000 package model
type RegisterParams struct {
Username string `json:"Username"`
Password string `json:"Password"`
}
type RegisterResp struct {
RSAPrivate string `json:"PrivateKey"`
}
type GetModelParams struct {
AccountInfo RegisterParams `json:"AccountInfo"`
ID int `json:"ID"`
}
type ModelResp struct {
ID int `json:"ID"`
Model string `json:"Model"`
ModelParam string `json:"ModelParam"`
ModelParamInfo string `json:"ModelParamInfo"`
}
agentencryption/agentenc/ 0000775 0000000 0000000 00000000000 14143644277 0016067 5 ustar 00root root 0000000 0000000 agentencryption/agentenc/__init__.py 0000664 0000000 0000000 00000000303 14143644277 0020174 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/10/27
# Copyright belongs to the author.
# Please indicate the source for reprinting.
from agentenc.encryptors import Encryptor, RSAEncryptor, AESEncryptor
agentencryption/agentenc/encryptors/ 0000775 0000000 0000000 00000000000 14143644277 0020277 5 ustar 00root root 0000000 0000000 agentencryption/agentenc/encryptors/__init__.py 0000664 0000000 0000000 00000000221 14143644277 0022403 0 ustar 00root root 0000000 0000000 from agentenc.encryptors.base import Encryptor
from agentenc.encryptors.rsa import RSAEncryptor
from agentenc.encryptors.aes import AESEncryptor
agentencryption/agentenc/encryptors/aes.py 0000664 0000000 0000000 00000002375 14143644277 0021430 0 ustar 00root root 0000000 0000000 from agentenc.encryptors import Encryptor
from agentenc.ops import AESEncryptOp
class AESEncryptor(Encryptor):
def __init__(self, bits: int = 128, mode: str = 'ECB', key: bytes = None, **kwargs):
"""
AES 加密器
:param
bits(int: 128): 加密使用的 bit 数
mode(str: ECB): 加密类型，可选：['ECB', 'CBC', 'CFB', 'OFB', 'CTR', 'OPENPGP', 'CCM', 'EAX', 'SIV', 'GCM', 'OCB']
iv(bytes: None): 偏移值，长度为 16 bytes ，默认随机生成
key(bytes: None): AES 密钥，长度为 (bits // 8) bytes ，默认随机生成
"""
super(AESEncryptor, self).__init__(
AESEncryptOp(
bits=bits,
mode=mode,
key=key,
**kwargs
)
)
@staticmethod
def decode(inp: str, key: bytes, **kwargs) -> any:
"""
解密函数
:param
inp(str): 输入的文件路径
key(bytes): AES 密钥
iv(bytes): 偏移值
:return
pure_datas(any): 原始数据
"""
return Encryptor.decode(
inp=inp,
key=key,
decode=AESEncryptOp.decode,
**kwargs
)
agentencryption/agentenc/encryptors/base.py 0000664 0000000 0000000 00000016533 14143644277 0021573 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/10/27
# Copyright belongs to the author.
# Please indicate the source for reprinting.
import os
import json
import base64
import pickle
from agentenc.ops import EncryptOp
class Encryptor:
def __init__(self, encrypt_op: EncryptOp):
"""
加密器基类
:param
encrypt_op(EncryptOp): 加密相关OP
"""
self.encrypt_op = encrypt_op
@staticmethod
def bytes2str(input: bytes) -> str:
"""
bytes to base64 str (inp -> data:data/agt;base64,{base64[inp]})
:param
inp(bytes): 输入
:return
output(str): 输出
"""
output = base64.b64encode(input).decode('UTF-8')
return f'data:data/agt;base64,{output}'
@staticmethod
def str2bytes(input: str) -> bytes:
"""
base64 str to bytes (data:data/agt;base64,{base64[inp]} -> inp)
:param
inp(bytes): 输入
:return
output(str): 输出
"""
if isinstance(input, str) and input[:21] == 'data:data/agt;base64,':
input = base64.b64decode(input[21:].encode('UTF-8'))
assert type(input) == bytes
return input
@staticmethod
def check_and_convert(input: any) -> any:
"""
检查并递归转换 bytes -> bytes_str
:param
inp(any): 输入
:return
output(any): 输出
"""
if isinstance(input, dict):
_input = input.copy()
for k in input.keys():
_input[k] = Encryptor.check_and_convert(input[k])
return _input
elif isinstance(input, (list, tuple)):
_input = input.copy()
for i in range(len(input)):
_input[i] = Encryptor.check_and_convert(input[i])
return _input
elif isinstance(input, bytes):
return Encryptor.bytes2str(input)
elif isinstance(input, (str, int, float, bool, None)):
return input
else:
raise ValueError('Please check inp data type.')
@staticmethod
def resume_and_convert(input: any) -> any:
"""
恢复并递归转换 bytes_str -> bytes
:param
inp(any): 输入
:return
output(any): 输出
"""
if isinstance(input, dict):
for k in input.keys():
input[k] = Encryptor.resume_and_convert(input[k])
return input
elif isinstance(input, (list, tuple)):
for i in range(len(input)):
input[i] = Encryptor.resume_and_convert(input[i])
return input
elif isinstance(input, str):
return Encryptor.str2bytes(input)
elif isinstance(input, (int, float, bool, None)):
return input
else:
raise ValueError('Please check inp data type.')
def encode(self, inp: any, output: str, format: str = 'pkl', keys_saving_path: str = None) -> dict:
"""
加密函数
:param
inp(any): 输入的需要加密的数据
output(str): 输出的文件名称
format(str: pkl [pkl / json / pdmodel]): 输出的数据格式
:return
private_params(dict): 加密器的私密参数，如密钥等
:format details
pkl: 此格式支持的加密数据类型较多，并且附带 python 解密函数，可依靠自身进行解密，但只可以在 python 端进行解密操作
json: 此格式支持如下几种数据类型 (dict, list, tuple, str, int, float, bool, bytes->bytes_str, None), 可以在任意语言中读取和解密，需搭配对应语言的解密函数进行解密操作
"""
if format == 'pkl':
encrypt_data = self.encrypt_op.encode(
pickle.dumps(inp, protocol=4))
with open(output + '.pkl', "wb") as file:
pickle.dump({
'datas': Encryptor.bytes2str(encrypt_data),
'params': self.encrypt_op.get_public_params(),
'decode': self.encrypt_op.decode
}, file, protocol=4)
elif format == 'json':
encrypt_data = self.encrypt_op.encode(
json.dumps(Encryptor.check_and_convert(inp)).encode('UTF-8'))
with open(output + '.json', "w") as file:
json.dump({
'datas': Encryptor.bytes2str(encrypt_data),
'params': Encryptor.check_and_convert(self.encrypt_op.get_public_params())
}, file)
elif format == "pdmodel":
pdiparams = inp + ".pdiparams"
pdmodel = inp + ".pdmodel"
assert os.path.exists(pdiparams), FileExistsError(pdiparams + "文件不存在，请检查输入路径是否正确")
assert os.path.exists(pdmodel), FileExistsError(pdmodel + "文件不存在，请检查输入路径是否正确")
with open(pdiparams, "rb") as f:
pdiparams = f.read()
with open(pdmodel, "rb") as f:
pdmodel = f.read()
encrypt_data = pdiparams + b"AgentEncryptionFlags" + pdmodel
encrypt_data = self.encrypt_op.encode(encrypt_data)
with open(output + '.EncPdModel', "wb") as file:
pickle.dump({
'data': Encryptor.bytes2str(encrypt_data),
'params': self.encrypt_op.get_public_params(),
'decode': self.encrypt_op.decode
}, file, protocol=4)
else:
raise ValueError('Please check the format type.')
return self.encrypt_op.get_private_params(keys_saving_path)
@staticmethod
def decode(inp: str, decode=None, **kwargs) -> any:
"""
解密函数
:param
inp(str): 输入的文件路径
decode(func): 解密函数
**kwargs: 解密所需的一些其他参数
:return
pure_datas(any): 原始数据
"""
ext = os.path.splitext(inp)[1]
# 加载机密数据包
if ext == '.pkl' or ext == ".EncPdModel":
with open(inp, "rb") as file:
encrypt_package = pickle.load(file)
elif ext == '.json':
with open(inp, "r") as file:
encrypt_package = json.load(file)
else:
raise ValueError('Please check inp path.')
# 解码公开参数
params = encrypt_package['params']
params = Encryptor.resume_and_convert(params)
# 解码加密数据
encrypt_data = Encryptor.str2bytes(encrypt_package['data'])
decode = encrypt_package.get('decode', decode)
pure_data = decode(encrypt_data, **kwargs, **params)
# 重新加载原始数据
if ext == '.pkl':
output = pickle.loads(pure_data)
elif ext == '.json':
output = json.loads(pure_data.decode('UTF-8'))
output = Encryptor.resume_and_convert(output)
elif ext == ".EncPdModel":
pdiparams, pdmodel = pure_data.split(b"AgentEncryptionFlags")
from paddle.inference import Config
output = Config()
output.set_model_buffer(pdmodel, len(pdmodel), pdiparams, len(pdiparams))
return output
else:
raise ValueError('Please check inp data type.')
return output
agentencryption/agentenc/encryptors/rsa.py 0000664 0000000 0000000 00000002022 14143644277 0021432 0 ustar 00root root 0000000 0000000 from agentenc.encryptors import Encryptor
from agentenc.ops import RSAEncryptOp
class RSAEncryptor(Encryptor):
def __init__(self, bits: int = 1024, public_key: bytes = None):
"""
RSA 加密器
:param
bits(int: 1024): 加密使用的 bit 数
public_key(bytes: None): RSA 公钥，格式为 b'-----BEGIN PUBLIC KEY-----...-----END PUBLIC KEY----- 的 bytes，默认随机生成
"""
super(RSAEncryptor, self).__init__(
RSAEncryptOp(
bits=bits,
public_key=public_key
)
)
@staticmethod
def decode(inp: str, private_key: bytes) -> any:
"""
解密函数
:param
inp(str): 输入的文件路径
private_key(bytes): RSA 私钥用于数据解密
:return
pure_datas(any): 原始数据
"""
return Encryptor.decode(
inp=inp,
private_key=private_key,
decode=RSAEncryptOp.decode
)
agentencryption/agentenc/ops/ 0000775 0000000 0000000 00000000000 14143644277 0016670 5 ustar 00root root 0000000 0000000 agentencryption/agentenc/ops/__init__.py 0000664 0000000 0000000 00000000174 14143644277 0021003 0 ustar 00root root 0000000 0000000 from agentenc.ops.base import EncryptOp
from agentenc.ops.rsa import RSAEncryptOp
from agentenc.ops.aes import AESEncryptOp
agentencryption/agentenc/ops/aes.py 0000664 0000000 0000000 00000006627 14143644277 0020025 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/10/27
# Copyright belongs to the author.
# Please indicate the source for reprinting.
from os import urandom
from Crypto.Cipher import AES
from agentenc.ops import EncryptOp
IV_FILE = "IV"
KEY_FILE = "KEY"
# AES Mode Switch
ASE_MODES = {
'ECB': AES.MODE_ECB,
'CBC': AES.MODE_CBC,
'CFB': AES.MODE_CFB,
'OFB': AES.MODE_OFB,
'CTR': AES.MODE_CTR,
'CCM': AES.MODE_CCM,
'EAX': AES.MODE_EAX,
'GCM': AES.MODE_GCM,
'OCB': AES.MODE_OCB
}
class AESEncryptOp(EncryptOp):
def __init__(self, bits: int = 128, mode: str = 'ECB', key: bytes = None, **kwargs):
"""
AES 加密算子
:param
bits(int: 128): 加密 bit 数
mode(str: ECB): 加密类型，可选：['ECB', 'CBC', 'CFB', 'OFB', 'CTR', 'CCM', 'EAX', 'GCM', 'OCB']
key(bytes: None): AES 密钥，长度为 (bits // 8) bytes ，默认随机生成
**kwargs: 一些其他的加密参数，如 iv, nonce 等
"""
super().__init__()
self.length = bits // 8
self.mode = mode
if key is not None:
self.key = key
else:
self.key = urandom(self.length)
self.aes = AES.new(
key=self.key,
mode=ASE_MODES[self.mode],
**kwargs
)
def get_private_params(self, save_path: str = None) -> dict:
"""
获取并保存密钥
:param
save_path(str: None): 密钥保存目录
:return
private_params(dict): {'key': AES 密钥}
:save file details
"{save_path}.key": AES 密钥
"""
if save_path:
with open(f'{save_path}.{KEY_FILE}', "wb") as f:
f.write(self.key)
return {
'key': self.key
}
def get_public_params(self) -> dict:
"""
获取解密所需的公开参数
:return
public_params(dict): {'mode': 加密类型}
"""
if hasattr(self.aes, 'iv'):
return {
'mode': self.mode,
'iv': self.aes.iv
}
elif hasattr(self.aes, 'nonce'):
return {
'mode': self.mode,
'nonce': self.aes.nonce
}
else:
return {
'mode': self.mode
}
def encode(self, input: bytes) -> bytes:
"""
AES 加密
:param
inp(bytes): 输入数据
:return
output(bytes): 加密数据
"""
count = len(input)
if count % self.length != 0:
add = self.length - (count % self.length)
else:
add = 0
input = input + (b'\x00' * add)
output = self.aes.encrypt(input)
return output
@staticmethod
def decode(input: bytes, mode: str, key: bytes, **kwargs) -> bytes:
"""
AES 解密
:param
inp(bytes): 加密数据
mode(str): 加密类型，可选：['ECB', 'CBC', 'CFB', 'OFB', 'CTR', 'CCM', 'EAX', 'GCM', 'OCB']
key(bytes): AES 密钥
**kwargs: 一些其他的加密参数，如 iv, nonce 等
:return
output(bytes): 原始数据
"""
aes = AES.new(key=key, mode=ASE_MODES[mode], **kwargs)
output = aes.decrypt(input)
output = output.rstrip(b'\x00')
return output
agentencryption/agentenc/ops/base.py 0000664 0000000 0000000 00000001511 14143644277 0020152 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/10/27
# Copyright belongs to the author.
# Please indicate the source for reprinting.
class EncryptOp:
def __init__(self):
"""
加密算子基类
"""
pass
def get_private_params(self, *args, **kwargs) -> dict:
"""
获取隐私参数并储存，如获取并保存密钥等信息
"""
pass
def get_public_params(self, *args, **kwargs) -> dict:
"""
客户端额外所需的解密信息，此处请勿返回任何公钥与私钥内容
"""
pass
def encode(self, input: bytes, **kwargs) -> bytes:
"""
定义加密流程
"""
pass
@staticmethod
def decode(input: bytes, **kwargs) -> bytes:
"""
定义解密流程
"""
pass
agentencryption/agentenc/ops/rsa.py 0000664 0000000 0000000 00000006643 14143644277 0020040 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/10/27
# Copyright belongs to the author.
# Please indicate the source for reprinting.
from Crypto import Random
from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_v1_5
from agentenc.ops import EncryptOp
PRIVATE_FILE = "PRIVATE"
PUBLIC_FILE = "PUBLIC"
class RSAEncryptOp(EncryptOp):
def __init__(self, bits: int = 1024, public_key: bytes = None):
"""
RSA 加密算子
:param
bits(int: 1024): 加密 bit 数
public_key(bytes: None): RSA 公钥，格式为 b'-----BEGIN PUBLIC KEY-----...-----END PUBLIC KEY----- 的 bytes，默认随机生成
"""
super().__init__()
self.bits = bits
if public_key is not None:
self.private_key = None
self.public_key = public_key
else:
self.rsa = RSA.generate(self.bits, Random.new().read)
self.private_key = self.rsa.exportKey()
self.public_key = self.rsa.publickey().exportKey()
self.length = bits // 8 - 11
def get_private_params(self, save_path: str = None) -> dict:
"""
获取并保存密钥
:param
save_path(str: None): 密钥保存目录
:return
private_params(dict): {'private_key': RSA 私钥, 'public_key': RSA 公钥}
:save file details
"{save_path}.PUBLIC": RSA 公钥
"{save_path}.PRIVATE": RSA 私钥
"""
if save_path:
with open(f'{save_path}.{PUBLIC_FILE}', "wb") as f:
f.write(self.public_key)
if self.private_key is not None:
with open(f'{save_path}.{PRIVATE_FILE}', "wb") as f:
f.write(self.private_key)
if self.private_key is not None:
return {
'public_key': self.public_key,
'private_key': self.private_key
}
else:
return {
'public_key': self.public_key
}
def get_public_params(self) -> dict:
"""
获取解密所需的公开参数
:return
public_params(dict): {'length': 加密长度}
"""
return {"length": self.length + 11}
def encode(self, input: bytes) -> bytes:
"""
RSA 加密
:param
inp(bytes): 输入数据
:return
output(bytes): 加密数据
"""
rsa_key = RSA.importKey(self.public_key)
cipher = PKCS1_v1_5.new(rsa_key)
cipher_text_ = []
for i in range(0, len(input), self.length):
cipher_text_.append(cipher.encrypt(input[i:i + self.length]))
output = b""
for item in cipher_text_:
output += item
return output
@staticmethod
def decode(input: bytes, length: int, private_key: bytes) -> bytes:
"""
RSA 解密
:param
inp(bytes): 加密数据
length(int): 加密长度
private_key(bytes): 私钥
:return
output(bytes): 原始数据
"""
rsa_key = RSA.importKey(private_key)
cipher = PKCS1_v1_5.new(rsa_key)
plain_text_ = []
for i in range(0, len(input), length):
plain_text_.append(cipher.decrypt(
input[i:i + length], "Decode error."))
output = b""
for item in plain_text_:
output += item
return output
agentencryption/example/ 0000775 0000000 0000000 00000000000 14143644277 0015736 5 ustar 00root root 0000000 0000000 agentencryption/example/paddlepaddle/ 0000775 0000000 0000000 00000000000 14143644277 0020341 5 ustar 00root root 0000000 0000000 agentencryption/example/paddlepaddle/key.PRIVATE 0000664 0000000 0000000 00000001566 14143644277 0022175 0 ustar 00root root 0000000 0000000 -----BEGIN RSA PRIVATE KEY-----
MIICWwIBAAKBgQCdHPMo50Scf/K668g/kc+/rdmhbuNpe/qGH2W5xjnRimirV/59
kUIv9wV3PEB71/WjNb13UcR/98Xc6+4VKD5ZsL0zW98YlflY7VjLPw5qx5egEXO1
Z/ANSQxCu3fu/mi1mp69XJoPyj+bpLx5cI+RL1r/i1FqaUf+ATD6MyhaXwIDAQAB
AoGAGGw1IlEc+cTr6wGW7cbRSAcS0wTlRfL3RR0hV4kIWNnOnn/9t/A6HTGlgFz1
2yp9bSL/8kncJVZTRB3KszA/zLvp+9ozoGLIGNPnyc1dAsSv6JO3cDz8pXM6jqtC
ZHJ+G6vAIPITKArPyfOiCRFIrVqoWrvKG5nZG5PCV3ukYsUCQQC6z0SW3ikSnwcN
u9pRmD8R1AQnT+9LQuBIq2JcBcLEFC0x/cThl7uhz+qrJSv25rfnQsaEY44oCio6
KDxRWXnNAkEA103xepKUguai8GVtIrbckwgXTn3JV8uCA/WHVVrvm4CESoeMQhCJ
QYgcI4s/kiql9crdyxBNXGAa2FzN3UnI2wJAJS5jxHNrUV19leoxxdO7mdzNBtYO
Fri000HpTc+HuGcbp7uOmCgPHGG5n17CpV2+dLUBtg/ql2WOigzAFKDucQJAQAhT
UBP2DTRQs1RG7/MYiLX4UITCc9T3lvvXwv/nkQTY012azi8i0xWK4NtSDcOXVhF+
gx0XFOkUMtLu+ugtzwJABRUvEa5vKyCqkDYiRWl+WZbm9DyOJ7NJd5Sc9dWKujeJ
MJbPX6UhWeI3yn8nzTUGegECySS42vtlvHHxmu8W8g==
-----END RSA PRIVATE KEY----- agentencryption/example/paddlepaddle/key.PUBLIC 0000664 0000000 0000000 00000000417 14143644277 0022033 0 ustar 00root root 0000000 0000000 -----BEGIN PUBLIC KEY-----
MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCdHPMo50Scf/K668g/kc+/rdmh
buNpe/qGH2W5xjnRimirV/59kUIv9wV3PEB71/WjNb13UcR/98Xc6+4VKD5ZsL0z
W98YlflY7VjLPw5qx5egEXO1Z/ANSQxCu3fu/mi1mp69XJoPyj+bpLx5cI+RL1r/
i1FqaUf+ATD6MyhaXwIDAQAB
-----END PUBLIC KEY----- agentencryption/example/paddlepaddle/make_model.py 0000664 0000000 0000000 00000002127 14143644277 0023012 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/11/5
# Copyright belongs to the author.
# Please indicate the source for reprinting.
import paddle
# 导入 RSA 加密器
from agentenc import RSAEncryptor
SAVE_ORI_MODEL_PATH = "./model/ori_model"
SAVE_ENC_MODEL_PATH = "./model/enc_model"
"""
制作一个简单的模型
"""
# 模拟一个线性层组成的模型
fc = paddle.nn.Linear(1, 1)
# 转静态图
input_1 = paddle.static.InputSpec(shape=[-1, 1], name="input_1")
# 保存模型
paddle.jit.save(fc, path=SAVE_ORI_MODEL_PATH, input_spec=[input_1])
# 输入一个常量，查看结果
test_ipt = paddle.to_tensor([1.], dtype="float32")
out = fc(test_ipt).numpy()
# 保持该结果，用于解密后的验证
with open(SAVE_ENC_MODEL_PATH + "_one_test.txt", "w") as f:
f.write(str(out))
print("当输入1时，程序应得到的结果为：", out)
"""
加密模型
"""
# 初始化 RSA 加密器
encryptor = RSAEncryptor(bits=1024)
# 数据加密
key = encryptor.encode(
inp=SAVE_ORI_MODEL_PATH,
output=SAVE_ENC_MODEL_PATH,
format='pdmodel',
keys_saving_path='key'
)
print("加密完毕")
agentencryption/example/paddlepaddle/test_model_paddleinference.py 0000664 0000000 0000000 00000003014 14143644277 0026240 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/11/5
# Copyright belongs to the author.
# Please indicate the source for reprinting.
import numpy as np
import paddle.inference as paddle_infer
from agentenc import Encryptor
SAVE_ENC_MODEL_PATH = "./model/enc_model"
SAVE_ENC_MODEL_FILE_PATH = SAVE_ENC_MODEL_PATH + ".EncPdModel"
# 读取私钥
private_key = open('key.PRIVATE', 'rb').read()
# 使用解密函数对文件进行解密
config = Encryptor.decode(
inp=SAVE_ENC_MODEL_FILE_PATH,
private_key=private_key
)
# 创建 config
config = paddle_infer.Config(config)
# 根据 config 创建 predictor
predictor = paddle_infer.create_predictor(config)
# 获取输入的名称
input_names = predictor.get_input_names()
input_handle = predictor.get_input_handle(input_names[0])
# 设置输入
fake_input = np.array([1]).astype(np.float32)
input_handle.copy_from_cpu(fake_input)
# 运行predictor
predictor.run()
# 获取输出
output_names = predictor.get_output_names()
output_handle = predictor.get_output_handle(output_names[0])
output_data = output_handle.copy_to_cpu() # numpy.ndarray类型
print("Output data size is {}".format(output_data.size))
print("Output data shape is {}".format(output_data.shape))
"""
验证部分
"""
# 打开先前准备的测试结果数据
with open(SAVE_ENC_MODEL_PATH + "_one_test.txt", "r") as f:
one = eval(f.read())
loss = abs(output_data - one)
print("LOSS:", loss)
if loss < 1e-5:
print("误差较低，通过测试")
else:
raise Exception("误差较高，请手动评估是否通过")
agentencryption/example/paddlepaddle/test_model_ppqi.py 0000664 0000000 0000000 00000001750 14143644277 0024106 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/11/5
# Copyright belongs to the author.
# Please indicate the source for reprinting.
import numpy as np
from ppqi import InferenceModel
from agentenc import Encryptor
SAVE_ENC_MODEL_PATH = "./model/enc_model"
SAVE_ENC_MODEL_FILE_PATH = SAVE_ENC_MODEL_PATH + ".EncPdModel"
# 读取私钥
private_key = open('key.PRIVATE', 'rb').read()
# 使用解密函数对文件进行解密
config = Encryptor.decode(
inp=SAVE_ENC_MODEL_FILE_PATH,
private_key=private_key
)
# 加载推理模型
model = InferenceModel(config)
model.eval()
# 准备数据
inputs = np.array([1]).astype(np.float32)
# 前向计算
outputs = model(inputs)
"""
验证部分
"""
# 打开先前准备的测试结果数据
with open(SAVE_ENC_MODEL_PATH + "_one_test.txt", "r") as f:
one = eval(f.read())
loss = abs(outputs - one)
print("LOSS:", loss)
if loss < 1e-5:
print("误差较低，通过测试")
else:
raise Exception("误差较高，请手动评估是否通过")
agentencryption/setup.py 0000664 0000000 0000000 00000001210 14143644277 0016007 0 ustar 00root root 0000000 0000000 # Author: Acer Zhang
# Datetime: 2021/10/27
# Copyright belongs to the author.
# Please indicate the source for reprinting.
from setuptools import setup
from setuptools import find_packages
__version__ = "0.1"
setup(
name='AgentEnc',
version=__version__,
packages=['agentenc', 'agentenc.ops', 'agentenc.encryptors'],
url='https://github.com/AgentMaker/AgentEncryption',
license='Apache2',
author='AgentMaker',
author_email='agentmaker@163.com',
description='飞桨模型加密库',
install_requires=["ppqi",
"pycryptodome"],
python_requires='>3.5',
include_package_data=True,
)