3a096f59a9
- 核心功能从 store/ 迁移至 oss/core/ 框架层 - 实现 NBPF 包格式：多重签名（Ed25519+RSA-PSS+HMAC）+ 多重加密（AES-256-GCM） - 实现 NIR 编译器：基于 compile()+marshal 的跨平台中间表示 - 新增 nebula nbpf CLI 命令组（pack/unpack/verify/sign/keygen） - 新增 19 个 NBPF 测试用例，覆盖全链路 - 彻底重写 README，大型项目标准框架风格，所有图表使用 SVG - 更新 LICENSE 版权声明 - 清理旧版 store 插件目录（已迁移至 oss/core）
361 lines
14 KiB
Python
361 lines
14 KiB
Python
""".nbpf 加载器 — 加载 .nbpf 文件到运行时环境
|
||
|
||
加载流程：
|
||
1. 打开 .nbpf (ZIP) 文件
|
||
2. 外层验签：用 Ed25519 公钥验证包签名
|
||
3. 外层解密：用 RSA 私钥解密密钥1，解密 META-INF/
|
||
4. 中层验签：用 RSA-4096 公钥验证 NIR 签名
|
||
5. 中层解密：用 RSA 私钥解密密钥2，解密 NIR 数据
|
||
6. 内层验签：用 HMAC 验证每个模块签名
|
||
7. 反序列化 NIR 为 code object
|
||
8. 在受限沙箱中执行
|
||
9. 内存擦除所有密钥
|
||
"""
|
||
import json
|
||
import zipfile
|
||
import sys
|
||
import types
|
||
import hashlib
|
||
import base64
|
||
from pathlib import Path
|
||
from typing import Any, Optional
|
||
|
||
from oss.logger.logger import Log
|
||
from .crypto import NBPCrypto, NBPCryptoError
|
||
from .compiler import NIRCompiler, NIRCompileError
|
||
from .format import NBPFFormatter, NBPFFormatError
|
||
|
||
|
||
class NBPFLoadError(Exception):
|
||
""".nbpf 加载错误"""
|
||
pass
|
||
|
||
|
||
class NBPFLoader:
|
||
""".nbpf 加载器"""
|
||
|
||
def __init__(
|
||
self,
|
||
crypto: NBPCrypto = None,
|
||
compiler: NIRCompiler = None,
|
||
trusted_ed25519_keys: dict[str, bytes] = None,
|
||
trusted_rsa_keys: dict[str, bytes] = None,
|
||
rsa_private_key: bytes = None,
|
||
):
|
||
"""
|
||
Args:
|
||
crypto: 加密工具实例
|
||
compiler: 编译器实例
|
||
trusted_ed25519_keys: {signer_name: ed25519_public_key_bytes}
|
||
trusted_rsa_keys: {signer_name: rsa_public_key_pem}
|
||
rsa_private_key: RSA 私钥 PEM（用于解密 AES 密钥）
|
||
"""
|
||
self.crypto = crypto or NBPCrypto()
|
||
self.compiler = compiler or NIRCompiler()
|
||
self.trusted_ed25519_keys = trusted_ed25519_keys or {}
|
||
self.trusted_rsa_keys = trusted_rsa_keys or {}
|
||
self.rsa_private_key = rsa_private_key
|
||
|
||
def load(
|
||
self,
|
||
nbpf_path: Path,
|
||
plugin_name: str = None,
|
||
) -> tuple[Any, dict]:
|
||
"""加载 .nbpf 插件
|
||
|
||
Args:
|
||
nbpf_path: .nbpf 文件路径
|
||
plugin_name: 插件名称（用于日志，默认从 manifest 读取）
|
||
|
||
Returns:
|
||
(plugin_instance, plugin_info_dict)
|
||
|
||
Raises:
|
||
NBPFLoadError: 加载失败
|
||
"""
|
||
if not nbpf_path.exists():
|
||
raise NBPFLoadError(f".nbpf 文件不存在: {nbpf_path}")
|
||
|
||
try:
|
||
with zipfile.ZipFile(nbpf_path, 'r') as zf:
|
||
# 1. 外层验签
|
||
signer_name = self._verify_outer_signature(zf)
|
||
Log.info("NBPF", f"外层签名验证通过 (signer: {signer_name})")
|
||
|
||
# 2. 外层解密
|
||
key1, meta_inf = self._decrypt_outer(zf)
|
||
key1_buf = bytearray(key1)
|
||
|
||
# 3. 中层验签
|
||
rsa_signer = self._verify_inner_signature(zf, meta_inf)
|
||
Log.info("NBPF", f"中层签名验证通过 (signer: {rsa_signer})")
|
||
|
||
# 4. 中层解密
|
||
key2 = self._decrypt_inner(meta_inf)
|
||
key2_buf = bytearray(key2)
|
||
|
||
# 5. 派生 HMAC 密钥
|
||
hmac_key = self.crypto.derive_hmac_key(key1, key2)
|
||
self.crypto._secure_wipe(key1_buf)
|
||
self.crypto._secure_wipe(key2_buf)
|
||
|
||
# 6. 解密 NIR 数据
|
||
nir_data = self._decrypt_nir_data(zf, key2)
|
||
|
||
# 7. 内层验签
|
||
self._verify_module_signatures(nir_data, meta_inf, hmac_key)
|
||
Log.info("NBPF", "内层模块签名验证通过")
|
||
|
||
# 8. 获取插件名称
|
||
manifest = meta_inf.get("manifest", {})
|
||
meta = manifest.get("metadata", {})
|
||
name = plugin_name or meta.get("name", nbpf_path.stem)
|
||
|
||
# 9. 反序列化并执行
|
||
instance, module = self._deserialize_and_exec(nir_data, name)
|
||
|
||
# 10. 构建插件信息
|
||
info = {
|
||
"name": name,
|
||
"version": meta.get("version", ""),
|
||
"author": meta.get("author", ""),
|
||
"description": meta.get("description", ""),
|
||
"manifest": manifest,
|
||
"nbpf_path": str(nbpf_path),
|
||
"signer": signer_name,
|
||
}
|
||
|
||
Log.ok("NBPF", f"插件 '{name}' 加载成功")
|
||
return instance, info
|
||
|
||
except (NBPFFormatError, NBPCryptoError, NIRCompileError) as e:
|
||
raise NBPFLoadError(str(e)) from e
|
||
except zipfile.BadZipFile as e:
|
||
raise NBPFLoadError(f".nbpf 文件损坏: {e}") from e
|
||
except Exception as e:
|
||
raise NBPFLoadError(f"加载失败: {type(e).__name__}: {e}") from e
|
||
|
||
# ── 外层验签 ──
|
||
|
||
def _verify_outer_signature(self, zf: zipfile.ZipFile) -> str:
|
||
"""外层 Ed25519 签名验证，返回签名者名称
|
||
|
||
签名计算方式与 full_encrypt_package 一致：
|
||
SHA256(outer_encryption_json + sorted_module_names_and_ciphertexts)
|
||
"""
|
||
if NBPFFormatter.SIGNATURE not in zf.namelist():
|
||
raise NBPFLoadError("缺少外层签名文件")
|
||
if NBPFFormatter.SIGNER_PEM not in zf.namelist():
|
||
raise NBPFLoadError("缺少签名者公钥文件")
|
||
|
||
signature_b64 = zf.read(NBPFFormatter.SIGNATURE).decode().strip()
|
||
signer_pub_key = zf.read(NBPFFormatter.SIGNER_PEM)
|
||
|
||
# 查找匹配的信任公钥
|
||
signer_name = None
|
||
for name, trusted_key in self.trusted_ed25519_keys.items():
|
||
if trusted_key == signer_pub_key:
|
||
signer_name = name
|
||
break
|
||
|
||
if signer_name is None:
|
||
raise NBPFLoadError("签名者公钥不在信任列表中")
|
||
|
||
# 计算包摘要（与 full_encrypt_package 一致）
|
||
encryption_data = json.loads(zf.read(NBPFFormatter.ENCRYPTION).decode("utf-8"))
|
||
digest = hashlib.sha256()
|
||
digest.update(json.dumps(encryption_data["data"]).encode())
|
||
|
||
# 按模块名排序，添加模块名和密文
|
||
nir_modules = {}
|
||
for info in zf.infolist():
|
||
if info.filename.startswith(NBPFFormatter.NIR_DIR) and not info.filename.endswith("/"):
|
||
mod_name = info.filename[len(NBPFFormatter.NIR_DIR):]
|
||
mod_data = json.loads(zf.read(info.filename).decode("utf-8"))
|
||
nir_modules[mod_name] = mod_data
|
||
|
||
for mod_name in sorted(nir_modules.keys()):
|
||
digest.update(mod_name.encode())
|
||
digest.update(nir_modules[mod_name]["ciphertext"].encode())
|
||
|
||
# 验签
|
||
signature = base64.b64decode(signature_b64)
|
||
if not self.crypto.outer_verify(digest.digest(), signature, signer_pub_key):
|
||
raise NBPFLoadError("外层签名验证失败，包可能被篡改")
|
||
|
||
return signer_name
|
||
|
||
# ── 外层解密 ──
|
||
|
||
def _decrypt_outer(self, zf: zipfile.ZipFile) -> tuple[bytes, dict]:
|
||
"""外层解密，返回 (key1, meta_inf_dict)"""
|
||
if NBPFFormatter.ENCRYPTION not in zf.namelist():
|
||
raise NBPFLoadError("缺少外层加密信息")
|
||
|
||
encryption_info = json.loads(zf.read(NBPFFormatter.ENCRYPTION).decode("utf-8"))
|
||
|
||
# 用 RSA 私钥解密 key1
|
||
if self.rsa_private_key is None:
|
||
raise NBPFLoadError("未配置 RSA 私钥，无法解密")
|
||
key1 = self.crypto.decrypt_key(encryption_info["encrypted_key"], self.rsa_private_key)
|
||
|
||
# 解密 META-INF 数据
|
||
meta_inf_bytes = self.crypto.outer_decrypt(encryption_info["data"], key1)
|
||
meta_inf = json.loads(meta_inf_bytes.decode("utf-8"))
|
||
|
||
return key1, meta_inf
|
||
|
||
# ── 中层验签 ──
|
||
|
||
def _verify_inner_signature(self, zf: zipfile.ZipFile, meta_inf: dict) -> str:
|
||
"""中层 RSA-4096 签名验证，返回签名者名称
|
||
|
||
签名计算方式与 full_encrypt_package 一致：
|
||
SHA256(sorted_module_names + inner_encrypted_ciphertexts)
|
||
"""
|
||
inner_sig_b64 = meta_inf.get("inner_signature")
|
||
if not inner_sig_b64:
|
||
raise NBPFLoadError("缺少中层签名")
|
||
|
||
# 计算 NIR 数据摘要（与 full_encrypt_package 一致）
|
||
nir_modules = {}
|
||
for info in zf.infolist():
|
||
if info.filename.startswith(NBPFFormatter.NIR_DIR) and not info.filename.endswith("/"):
|
||
mod_name = info.filename[len(NBPFFormatter.NIR_DIR):]
|
||
mod_data = json.loads(zf.read(info.filename).decode("utf-8"))
|
||
nir_modules[mod_name] = mod_data
|
||
|
||
nir_digest = hashlib.sha256()
|
||
for mod_name in sorted(nir_modules.keys()):
|
||
nir_digest.update(mod_name.encode())
|
||
nir_digest.update(nir_modules[mod_name]["ciphertext"].encode())
|
||
|
||
# 查找匹配的 RSA 公钥
|
||
inner_sig = base64.b64decode(inner_sig_b64)
|
||
for name, rsa_pub_key in self.trusted_rsa_keys.items():
|
||
if self.crypto.inner_verify(nir_digest.digest(), inner_sig, rsa_pub_key):
|
||
return name
|
||
|
||
raise NBPFLoadError("中层签名验证失败，无法匹配任何信任的 RSA 公钥")
|
||
|
||
# ── 中层解密 ──
|
||
|
||
def _decrypt_inner(self, meta_inf: dict) -> bytes:
|
||
"""中层解密，返回 key2"""
|
||
inner_enc = meta_inf.get("inner_encryption", {})
|
||
encrypted_key = inner_enc.get("encrypted_key")
|
||
if not encrypted_key:
|
||
raise NBPFLoadError("缺少中层加密密钥")
|
||
|
||
if self.rsa_private_key is None:
|
||
raise NBPFLoadError("未配置 RSA 私钥，无法解密")
|
||
return self.crypto.decrypt_key(encrypted_key, self.rsa_private_key)
|
||
|
||
# ── 解密 NIR 数据 ──
|
||
|
||
def _decrypt_nir_data(self, zf: zipfile.ZipFile, key2: bytes) -> dict[str, bytes]:
|
||
"""解密 NIR 数据，返回 {module_name: nir_bytes}"""
|
||
nir_data = {}
|
||
for info in zf.infolist():
|
||
if not info.filename.startswith(NBPFFormatter.NIR_DIR):
|
||
continue
|
||
if info.filename.endswith("/"):
|
||
continue
|
||
|
||
module_name = info.filename[len(NBPFFormatter.NIR_DIR):]
|
||
enc_info = json.loads(zf.read(info.filename).decode("utf-8"))
|
||
mod_data = self.crypto.inner_decrypt(enc_info, key2)
|
||
nir_data[module_name] = mod_data
|
||
|
||
return nir_data
|
||
|
||
# ── 内层验签 ──
|
||
|
||
def _verify_module_signatures(self, nir_data: dict, meta_inf: dict, hmac_key: bytes):
|
||
"""内层 HMAC 模块签名验证"""
|
||
module_sigs = meta_inf.get("module_signatures", {})
|
||
if not module_sigs:
|
||
Log.warn("NBPF", "未找到模块签名，跳过内层验签")
|
||
return
|
||
|
||
for mod_name, mod_data in nir_data.items():
|
||
expected_sig = module_sigs.get(mod_name)
|
||
if expected_sig:
|
||
if not self.crypto.module_verify(mod_data, expected_sig, hmac_key):
|
||
raise NBPFLoadError(f"模块 '{mod_name}' HMAC 签名验证失败")
|
||
|
||
# ── 反序列化并执行 ──
|
||
|
||
def _deserialize_and_exec(
|
||
self,
|
||
nir_data: dict[str, bytes],
|
||
plugin_name: str,
|
||
) -> tuple[Any, types.ModuleType]:
|
||
"""反序列化 NIR 并执行，返回 (instance, module)"""
|
||
|
||
# 构建安全的全局命名空间
|
||
safe_globals = self._build_safe_globals(plugin_name)
|
||
|
||
# 按依赖顺序执行模块
|
||
main_module = None
|
||
for mod_name in sorted(nir_data.keys()):
|
||
nir_bytes = nir_data[mod_name]
|
||
code = self.compiler.deserialize_nir(nir_bytes)
|
||
|
||
# 创建模块
|
||
module_name = f"nbpf.{plugin_name}.{mod_name}"
|
||
if mod_name == NBPFFormatter.ENTRY_POINT:
|
||
module_name = f"nbpf.{plugin_name}"
|
||
|
||
module = types.ModuleType(module_name)
|
||
module.__package__ = f"nbpf.{plugin_name}"
|
||
module.__path__ = []
|
||
module.__file__ = f"<nbpf:{plugin_name}/{mod_name}>"
|
||
sys.modules[module_name] = module
|
||
|
||
# 执行 code object
|
||
exec(code, module.__dict__)
|
||
|
||
if mod_name == NBPFFormatter.ENTRY_POINT:
|
||
main_module = module
|
||
|
||
# 调用 New() 创建实例
|
||
if main_module is None:
|
||
raise NBPFLoadError(f"缺少入口模块 '{NBPFFormatter.ENTRY_POINT}'")
|
||
|
||
if not hasattr(main_module, "New"):
|
||
raise NBPFLoadError("插件缺少 New() 函数")
|
||
|
||
try:
|
||
instance = main_module.New()
|
||
except Exception as e:
|
||
raise NBPFLoadError(f"创建插件实例失败: {e}") from e
|
||
|
||
return instance, main_module
|
||
|
||
def _build_safe_globals(self, plugin_name: str) -> dict:
|
||
"""构建安全的全局命名空间"""
|
||
safe_builtins = {
|
||
'True': True, 'False': False, 'None': None,
|
||
'dict': dict, 'list': list, 'str': str, 'int': int,
|
||
'float': float, 'bool': bool, 'tuple': tuple, 'set': set,
|
||
'len': len, 'range': range, 'enumerate': enumerate,
|
||
'zip': zip, 'map': map, 'filter': filter,
|
||
'sorted': sorted, 'reversed': reversed,
|
||
'min': min, 'max': max, 'sum': sum, 'abs': abs,
|
||
'round': round, 'isinstance': isinstance, 'issubclass': issubclass,
|
||
'type': type, 'id': id, 'hash': hash, 'repr': repr,
|
||
'print': print, 'object': object, 'property': property,
|
||
'staticmethod': staticmethod, 'classmethod': classmethod,
|
||
'super': super, 'iter': iter, 'next': next,
|
||
'any': any, 'all': all, 'callable': callable,
|
||
'hasattr': hasattr, 'getattr': getattr,
|
||
'ValueError': ValueError, 'TypeError': TypeError,
|
||
'KeyError': KeyError, 'IndexError': IndexError,
|
||
'Exception': Exception, 'BaseException': BaseException,
|
||
}
|
||
return {
|
||
'__builtins__': safe_builtins,
|
||
'__name__': f'nbpf.{plugin_name}',
|
||
}
|