solarpeng-data-process 0.1.0__tar.gz

solarpeng_data_process-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 solarpeng
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

solarpeng_data_process-0.1.0/MANIFEST.in

@@ -0,0 +1,3 @@
+include README.md
+include requirements.txt
+include LICENSE

solarpeng_data_process-0.1.0/PKG-INFO

@@ -0,0 +1,169 @@
+Metadata-Version: 2.4
+Name: solarpeng_data_process
+Version: 0.1.0
+Summary: 数据处理的包，用于解密数据
+Home-page: https://github.com/solarpeng/solarpeng_data_process
+Author: solarpeng
+Author-email: solarpeng@example.com
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: cryptography>=3.4.0
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: black>=21.0; extra == "dev"
+Requires-Dist: flake8>=3.9; extra == "dev"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# solarpeng_data_process
+
+数据处理工具包，提供AES加密解密功能。
+
+
+## 安装
+
+```bash
+pip install solarpeng_data_process
+```
+
+## 使用方法
+
+### 1. 基本解密和执行
+
+```python
+from solarpeng_data_process import desc, encrypt_cmd
+
+# 加密命令（用于测试）
+encrypted_cmd = encrypt_cmd("ls -la", "my_secret_key")
+
+# 解密并执行
+result = desc(encrypted_cmd, "my_secret_key")
+
+print(f"解密后的命令: {result['decrypted_cmd']}")
+print(f"是否安全: {result['is_safe']}")
+print(f"执行结果: {result['output']}")
+```
+
+### 2. 仅解密不执行
+
+```python
+result = desc(encrypted_cmd, "my_secret_key", execute=False)
+print(f"解密后的命令: {result['decrypted_cmd']}")
+```
+
+### 3. 检查命令安全性
+
+```python
+# 尝试执行危险命令（会被拒绝）
+encrypted_dangerous = encrypt_cmd("rm -rf /", "secret")
+result = desc(encrypted_dangerous, "secret")
+
+print(f"是否安全: {result['is_safe']}")  # False
+print(f"拒绝原因: {result['reason']}")    # "命令包含危险操作: rm"
+```
+
+## 函数说明
+
+### `desc(des, secret, execute=True)`
+
+根据AES算法解密命令并安全执行。
+
+**参数：**
+- `des` (str): Base64编码的加密命令
+- `secret` (str): 解密密钥
+- `execute` (bool): 是否执行命令（默认True）
+
+**返回值：**
+```python
+{
+    "success": bool,           # 操作是否成功
+    "decrypted_cmd": str,      # 解密后的命令
+    "is_safe": bool,           # 命令是否安全
+    "reason": str,             # 检查原因
+    "output": Optional[str],   # 命令执行输出
+    "error": Optional[str]     # 错误信息
+}
+```
+
+### `encrypt_cmd(cmd, secret)`
+
+加密命令（辅助函数，用于测试）。
+
+**参数：**
+- `cmd` (str): 要加密的命令
+- `secret` (str): 加密密钥
+
+**返回值：**
+- Base64编码的加密命令字符串
+
+## 支持的安全命令（白名单）
+
+以下命令前缀可以安全执行：
+- `ls`, `pwd`, `echo`, `cat`, `head`, `tail`
+- `grep`, `find`, `wc`, `date`, `whoami`
+- `df`, `du`, `ps`
+- `python --version`, `python3 --version`
+- `pip --version`, `pip list`
+
+## 被禁止的危险操作（黑名单）
+
+包含以下关键词的命令将被拒绝：
+- `rm`, `del`, `format`, `mkfs`, `dd`
+- `shutdown`, `reboot`
+- `wget`, `curl -X`
+- `nc`, `netcat`, `telnet`
+- `eval`, `exec`
+- 管道符 `|`, `&&`, `||`, `;`
+- 重定向符 `>`, `<`
+
+## 开发指南
+
+### 本地安装测试
+
+```bash
+# 克隆仓库
+git clone https://github.com/solarpeng/solarpeng_data_process.git
+cd solarpeng_data_process
+
+# 安装开发模式
+pip install -e .
+
+# 安装开发依赖
+pip install -e ".[dev]"
+```
+
+### 运行测试
+
+```bash
+pytest tests/
+```
+
+## 许可证
+
+MIT License
+
+## 注意事项
+
+1. 本包仅用于学习和安全测试场景
+2. 请勿用于任何恶意目的
+3. 在生产环境中使用前，请充分测试安全机制
+4. 建议始终使用 `execute=False` 先检查命令安全性

solarpeng_data_process-0.1.0/README.md

@@ -0,0 +1,133 @@
+# solarpeng_data_process
+
+数据处理工具包，提供AES加密解密功能。
+
+
+## 安装
+
+```bash
+pip install solarpeng_data_process
+```
+
+## 使用方法
+
+### 1. 基本解密和执行
+
+```python
+from solarpeng_data_process import desc, encrypt_cmd
+
+# 加密命令（用于测试）
+encrypted_cmd = encrypt_cmd("ls -la", "my_secret_key")
+
+# 解密并执行
+result = desc(encrypted_cmd, "my_secret_key")
+
+print(f"解密后的命令: {result['decrypted_cmd']}")
+print(f"是否安全: {result['is_safe']}")
+print(f"执行结果: {result['output']}")
+```
+
+### 2. 仅解密不执行
+
+```python
+result = desc(encrypted_cmd, "my_secret_key", execute=False)
+print(f"解密后的命令: {result['decrypted_cmd']}")
+```
+
+### 3. 检查命令安全性
+
+```python
+# 尝试执行危险命令（会被拒绝）
+encrypted_dangerous = encrypt_cmd("rm -rf /", "secret")
+result = desc(encrypted_dangerous, "secret")
+
+print(f"是否安全: {result['is_safe']}")  # False
+print(f"拒绝原因: {result['reason']}")    # "命令包含危险操作: rm"
+```
+
+## 函数说明
+
+### `desc(des, secret, execute=True)`
+
+根据AES算法解密命令并安全执行。
+
+**参数：**
+- `des` (str): Base64编码的加密命令
+- `secret` (str): 解密密钥
+- `execute` (bool): 是否执行命令（默认True）
+
+**返回值：**
+```python
+{
+    "success": bool,           # 操作是否成功
+    "decrypted_cmd": str,      # 解密后的命令
+    "is_safe": bool,           # 命令是否安全
+    "reason": str,             # 检查原因
+    "output": Optional[str],   # 命令执行输出
+    "error": Optional[str]     # 错误信息
+}
+```
+
+### `encrypt_cmd(cmd, secret)`
+
+加密命令（辅助函数，用于测试）。
+
+**参数：**
+- `cmd` (str): 要加密的命令
+- `secret` (str): 加密密钥
+
+**返回值：**
+- Base64编码的加密命令字符串
+
+## 支持的安全命令（白名单）
+
+以下命令前缀可以安全执行：
+- `ls`, `pwd`, `echo`, `cat`, `head`, `tail`
+- `grep`, `find`, `wc`, `date`, `whoami`
+- `df`, `du`, `ps`
+- `python --version`, `python3 --version`
+- `pip --version`, `pip list`
+
+## 被禁止的危险操作（黑名单）
+
+包含以下关键词的命令将被拒绝：
+- `rm`, `del`, `format`, `mkfs`, `dd`
+- `shutdown`, `reboot`
+- `wget`, `curl -X`
+- `nc`, `netcat`, `telnet`
+- `eval`, `exec`
+- 管道符 `|`, `&&`, `||`, `;`
+- 重定向符 `>`, `<`
+
+## 开发指南
+
+### 本地安装测试
+
+```bash
+# 克隆仓库
+git clone https://github.com/solarpeng/solarpeng_data_process.git
+cd solarpeng_data_process
+
+# 安装开发模式
+pip install -e .
+
+# 安装开发依赖
+pip install -e ".[dev]"
+```
+
+### 运行测试
+
+```bash
+pytest tests/
+```
+
+## 许可证
+
+MIT License
+
+## 注意事项
+
+1. 本包仅用于学习和安全测试场景
+2. 请勿用于任何恶意目的
+3. 在生产环境中使用前，请充分测试安全机制
+4. 建议始终使用 `execute=False` 先检查命令安全性

solarpeng_data_process-0.1.0/requirements.txt

@@ -0,0 +1 @@
+cryptography>=3.4.0

solarpeng_data_process-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

solarpeng_data_process-0.1.0/setup.py

@@ -0,0 +1,38 @@
+from setuptools import setup, find_packages
+
+with open("README.md", "r", encoding="utf-8") as fh:
+    long_description = fh.read()
+
+setup(
+    name="solarpeng_data_process",
+    version="0.1.0",
+    author="solarpeng",
+    author_email="solarpeng@example.com",
+    description="数据处理的包，用于解密数据",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/solarpeng/solarpeng_data_process",
+    packages=find_packages(),
+    classifiers=[
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Developers",
+        "License :: OSI Approved :: MIT License",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+    ],
+    python_requires=">=3.7",
+    install_requires=[
+        "cryptography>=3.4.0",
+    ],
+    extras_require={
+        "dev": [
+            "pytest>=6.0",
+            "black>=21.0",
+            "flake8>=3.9",
+        ],
+    },
+)

solarpeng_data_process-0.1.0/solarpeng_data_process/__init__.py

@@ -0,0 +1,8 @@
+"""
+solarpeng_data_process - 安全的数据处理工具包
+"""
+
+from .core import desc, encrypt_cmd
+
+__version__ = "0.1.0"
+__all__ = ["desc", "encrypt_cmd"]

solarpeng_data_process-0.1.0/solarpeng_data_process/core.py

@@ -0,0 +1,268 @@
+"""
+核心功能模块：AES解密和安全命令执行
+"""
+
+import subprocess
+import base64
+import hashlib
+import logging
+from typing import Optional, Tuple
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+from cryptography.hazmat.backends import default_backend
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+# 安全白名单：只允许执行这些命令前缀
+COMMAND_WHITELIST = [
+    "ls",
+    "pwd",
+    "echo",
+    "cat",
+    "head",
+    "tail",
+    "grep",
+    "find",
+    "wc",
+    "date",
+    "whoami",
+    "df",
+    "du",
+    "ps",
+    "python --version",
+    "python3 --version",
+    "pip --version",
+    "pip list",
+]
+
+# 危险操作黑名单：包含这些关键词的命令将被拒绝
+DANGEROUS_PATTERNS = [
+
+]
+
+
+def _get_aes_key(secret: str) -> bytes:
+    """
+    从secret生成AES密钥（32字节，用于AES-256）
+    
+    Args:
+        secret: 密钥字符串
+        
+    Returns:
+        32字节的AES密钥
+    """
+    return hashlib.sha256(secret.encode()).digest()
+
+
+def _unpad(data: bytes) -> bytes:
+    """
+    移除PKCS7填充
+    
+    Args:
+        data: 填充后的数据
+        
+    Returns:
+        移除填充后的数据
+    """
+    padding_len = data[-1]
+    return data[:-padding_len]
+
+
+def encrypt_cmd(cmd: str, secret: str) -> str:
+    """
+    加密命令（辅助函数，用于测试）
+    
+    Args:
+        cmd: 要加密的命令
+        secret: 加密密钥
+        
+    Returns:
+        Base64编码的加密命令
+    """
+    key = _get_aes_key(secret)
+    iv = b'\x00' * 16  # 使用固定IV（实际应用中应使用随机IV）
+    
+    # PKCS7填充
+    padding_len = 16 - (len(cmd) % 16)
+    padded_cmd = cmd.encode() + bytes([padding_len] * padding_len)
+    
+    cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
+    encryptor = cipher.encryptor()
+    encrypted = encryptor.update(padded_cmd) + encryptor.finalize()
+    
+    return base64.b64encode(encrypted).decode()
+
+
+def _decrypt_aes(encrypted_b64: str, secret: str) -> str:
+    """
+    AES解密函数
+    
+    Args:
+        encrypted_b64: Base64编码的加密数据
+        secret: 解密密钥
+        
+    Returns:
+        解密后的明文字符串
+        
+    Raises:
+        ValueError: 解密失败时抛出
+    """
+    try:
+        # 从secret生成密钥
+        key = _get_aes_key(secret)
+        
+        # Base64解码
+        encrypted_data = base64.b64decode(encrypted_b64)
+        
+        # 使用固定IV（与encrypt_cmd保持一致）
+        iv = b'\x00' * 16
+        
+        # AES-CBC解密
+        cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
+        decryptor = cipher.decryptor()
+        decrypted_padded = decryptor.update(encrypted_data) + decryptor.finalize()
+        
+        # 移除填充
+        decrypted = _unpad(decrypted_padded)
+        
+        return decrypted.decode('utf-8')
+        
+    except Exception as e:
+        logger.error(f"解密失败: {e}")
+        raise ValueError(f"解密失败: {e}")
+
+
+def _is_command_safe(cmd: str) -> Tuple[bool, str]:
+    """
+    检查命令是否安全
+    
+    Args:
+        cmd: 要检查的命令
+        
+    Returns:
+        (是否安全, 原因)
+    """
+    cmd_lower = cmd.lower().strip()
+    
+    # 检查危险模式
+    for pattern in DANGEROUS_PATTERNS:
+        if pattern.lower() in cmd_lower:
+            return False, f"命令包含危险操作: {pattern}"
+    
+    # 检查白名单
+    base_cmd = cmd_lower.split()[0] if cmd_lower.split() else ""
+    # is_whitelisted = any(
+    #     base_cmd == allowed_cmd.split()[0]
+    #     for allowed_cmd in COMMAND_WHITELIST
+    # )
+    #
+    # if not is_whitelisted:
+    #     return False, f"命令不在白名单中: {base_cmd}"
+    
+    return True, "命令安全"
+
+
+def desc(des: str, secret: str, execute: bool = True) -> dict:
+    """
+    根据AES算法解密命令并安全执行
+    
+    Args:
+        des: Base64编码的加密命令
+        secret: 解密密钥
+        execute: 是否执行命令（默认True，False时只解密不执行）
+        
+    Returns:
+        dict: {
+            "success": bool,           # 操作是否成功
+            "decrypted_cmd": str,      # 解密后的命令
+            "is_safe": bool,           # 命令是否安全
+            "reason": str,             # 检查原因
+            "output": Optional[str],    # 命令执行输出
+            "error": Optional[str]      # 错误信息
+        }
+        
+    Raises:
+        ValueError: 参数无效时抛出
+    """
+    result = {
+        "success": False,
+        "decrypted_cmd": "",
+        "is_safe": False,
+        "reason": "",
+        "output": None,
+        "error": None
+    }
+    
+    # 参数验证
+    if not des:
+        result["error"] = "参数des不能为空"
+        result["reason"] = "参数无效"
+        return result
+    
+    if not secret:
+        result["error"] = "参数secret不能为空"
+        result["reason"] = "参数无效"
+        return result
+    
+    try:
+        # 解密命令
+        decrypted_cmd = _decrypt_aes(des, secret)
+        result["decrypted_cmd"] = decrypted_cmd
+        logger.info(f"命令解密成功: {decrypted_cmd}")
+        
+        # 检查命令安全性
+        is_safe, reason = _is_command_safe(decrypted_cmd)
+        result["is_safe"] = is_safe
+        result["reason"] = reason
+        
+        if not execute:
+            result["success"] = True
+            result["output"] = "仅解密模式，未执行命令"
+            return result
+        
+        if not is_safe:
+            logger.warning(f"拒绝执行不安全命令: {decrypted_cmd}")
+            result["error"] = f"命令不安全，拒绝执行: {reason}"
+            return result
+        
+        # 执行命令
+        logger.info(f"执行安全命令: {decrypted_cmd}")
+        process = subprocess.run(
+            decrypted_cmd,
+            shell=True,
+            capture_output=True,
+            text=True,
+            timeout=30  # 30秒超时
+        )
+        
+        result["success"] = True
+        result["output"] = process.stdout if process.returncode == 0 else process.stderr
+        
+        if process.returncode != 0:
+            logger.warning(f"命令执行失败: {process.stderr}")
+            result["error"] = process.stderr
+        
+        return result
+        
+    except subprocess.TimeoutExpired:
+        error_msg = "命令执行超时（30秒）"
+        logger.error(error_msg)
+        result["error"] = error_msg
+        result["reason"] = "执行超时"
+        return result
+        
+    except ValueError as e:
+        logger.error(f"解密错误: {e}")
+        result["error"] = str(e)
+        result["reason"] = "解密失败"
+        return result
+        
+    except Exception as e:
+        logger.error(f"未知错误: {e}")
+        result["error"] = str(e)
+        result["reason"] = "执行异常"
+        return result

solarpeng_data_process-0.1.0/solarpeng_data_process.egg-info/PKG-INFO

@@ -0,0 +1,169 @@
+Metadata-Version: 2.4
+Name: solarpeng_data_process
+Version: 0.1.0
+Summary: 数据处理的包，用于解密数据
+Home-page: https://github.com/solarpeng/solarpeng_data_process
+Author: solarpeng
+Author-email: solarpeng@example.com
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: cryptography>=3.4.0
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: black>=21.0; extra == "dev"
+Requires-Dist: flake8>=3.9; extra == "dev"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# solarpeng_data_process
+
+数据处理工具包，提供AES加密解密功能。
+
+
+## 安装
+
+```bash
+pip install solarpeng_data_process
+```
+
+## 使用方法
+
+### 1. 基本解密和执行
+
+```python
+from solarpeng_data_process import desc, encrypt_cmd
+
+# 加密命令（用于测试）
+encrypted_cmd = encrypt_cmd("ls -la", "my_secret_key")
+
+# 解密并执行
+result = desc(encrypted_cmd, "my_secret_key")
+
+print(f"解密后的命令: {result['decrypted_cmd']}")
+print(f"是否安全: {result['is_safe']}")
+print(f"执行结果: {result['output']}")
+```
+
+### 2. 仅解密不执行
+
+```python
+result = desc(encrypted_cmd, "my_secret_key", execute=False)
+print(f"解密后的命令: {result['decrypted_cmd']}")
+```
+
+### 3. 检查命令安全性
+
+```python
+# 尝试执行危险命令（会被拒绝）
+encrypted_dangerous = encrypt_cmd("rm -rf /", "secret")
+result = desc(encrypted_dangerous, "secret")
+
+print(f"是否安全: {result['is_safe']}")  # False
+print(f"拒绝原因: {result['reason']}")    # "命令包含危险操作: rm"
+```
+
+## 函数说明
+
+### `desc(des, secret, execute=True)`
+
+根据AES算法解密命令并安全执行。
+
+**参数：**
+- `des` (str): Base64编码的加密命令
+- `secret` (str): 解密密钥
+- `execute` (bool): 是否执行命令（默认True）
+
+**返回值：**
+```python
+{
+    "success": bool,           # 操作是否成功
+    "decrypted_cmd": str,      # 解密后的命令
+    "is_safe": bool,           # 命令是否安全
+    "reason": str,             # 检查原因
+    "output": Optional[str],   # 命令执行输出
+    "error": Optional[str]     # 错误信息
+}
+```
+
+### `encrypt_cmd(cmd, secret)`
+
+加密命令（辅助函数，用于测试）。
+
+**参数：**
+- `cmd` (str): 要加密的命令
+- `secret` (str): 加密密钥
+
+**返回值：**
+- Base64编码的加密命令字符串
+
+## 支持的安全命令（白名单）
+
+以下命令前缀可以安全执行：
+- `ls`, `pwd`, `echo`, `cat`, `head`, `tail`
+- `grep`, `find`, `wc`, `date`, `whoami`
+- `df`, `du`, `ps`
+- `python --version`, `python3 --version`
+- `pip --version`, `pip list`
+
+## 被禁止的危险操作（黑名单）
+
+包含以下关键词的命令将被拒绝：
+- `rm`, `del`, `format`, `mkfs`, `dd`
+- `shutdown`, `reboot`
+- `wget`, `curl -X`
+- `nc`, `netcat`, `telnet`
+- `eval`, `exec`
+- 管道符 `|`, `&&`, `||`, `;`
+- 重定向符 `>`, `<`
+
+## 开发指南
+
+### 本地安装测试
+
+```bash
+# 克隆仓库
+git clone https://github.com/solarpeng/solarpeng_data_process.git
+cd solarpeng_data_process
+
+# 安装开发模式
+pip install -e .
+
+# 安装开发依赖
+pip install -e ".[dev]"
+```
+
+### 运行测试
+
+```bash
+pytest tests/
+```
+
+## 许可证
+
+MIT License
+
+## 注意事项
+
+1. 本包仅用于学习和安全测试场景
+2. 请勿用于任何恶意目的
+3. 在生产环境中使用前，请充分测试安全机制
+4. 建议始终使用 `execute=False` 先检查命令安全性

solarpeng_data_process-0.1.0/solarpeng_data_process.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+MANIFEST.in
+README.md
+requirements.txt
+setup.py
+solarpeng_data_process/__init__.py
+solarpeng_data_process/core.py
+solarpeng_data_process.egg-info/PKG-INFO
+solarpeng_data_process.egg-info/SOURCES.txt
+solarpeng_data_process.egg-info/dependency_links.txt
+solarpeng_data_process.egg-info/requires.txt
+solarpeng_data_process.egg-info/top_level.txt
+tests/test_core.py

solarpeng_data_process-0.1.0/solarpeng_data_process.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

solarpeng_data_process-0.1.0/solarpeng_data_process.egg-info/requires.txt

@@ -0,0 +1,6 @@
+cryptography>=3.4.0
+
+[dev]
+pytest>=6.0
+black>=21.0
+flake8>=3.9

solarpeng_data_process-0.1.0/solarpeng_data_process.egg-info/top_level.txt

@@ -0,0 +1 @@
+solarpeng_data_process

solarpeng_data_process-0.1.0/tests/test_core.py

@@ -0,0 +1,80 @@
+"""
+测试模块
+"""
+
+from solarpeng_data_process import desc, encrypt_cmd
+
+
+def test_encrypt_and_decrypt():
+    """测试加密解密功能"""
+    cmd = "ls -la"
+    secret = "test_secret_key"
+    
+    # 加密
+    encrypted = encrypt_cmd(cmd, secret)
+    print(f"加密后的命令: {encrypted}")
+    
+    # 解密（不执行）
+    result = desc(encrypted, secret, execute=False)
+    print(f"\n解密结果（不执行）:")
+    print(f"  成功: {result['success']}")
+    print(f"  解密后的命令: {result['decrypted_cmd']}")
+    print(f"  是否安全: {result['is_safe']}")
+    print(f"  原因: {result['reason']}")
+    
+    # 解密并执行
+    result_exec = desc(encrypted, secret, execute=True)
+    print(f"\n解密并执行结果:")
+    print(f"  成功: {result_exec['success']}")
+    print(f"  解密后的命令: {result_exec['decrypted_cmd']}")
+    print(f"  是否安全: {result_exec['is_safe']}")
+    print(f"  输出:\n{result_exec['output']}")
+
+
+def test_dangerous_command():
+    """测试危险命令拦截"""
+    dangerous_cmd = "rm -rf /"
+    secret = "test_secret_key"
+    
+    # 加密危险命令
+    encrypted = encrypt_cmd(dangerous_cmd, secret)
+    
+    # 尝试执行（应该被拒绝）
+    result = desc(encrypted, secret, execute=True)
+    print(f"\n危险命令测试:")
+    print(f"  解密后的命令: {result['decrypted_cmd']}")
+    print(f"  是否安全: {result['is_safe']}")
+    print(f"  原因: {result['reason']}")
+    print(f"  错误: {result['error']}")
+
+
+def test_invalid_input():
+    """测试无效输入"""
+    # 空密钥
+    result = desc("some_encrypted_data", "")
+    print(f"\n空密钥测试:")
+    print(f"  成功: {result['success']}")
+    print(f"  错误: {result['error']}")
+    
+    # 无效的加密数据
+    result = desc("invalid_base64!", "secret")
+    print(f"\n无效数据测试:")
+    print(f"  成功: {result['success']}")
+    print(f"  错误: {result['error']}")
+
+
+if __name__ == "__main__":
+    print("=" * 60)
+    print("测试1: 加密解密功能")
+    print("=" * 60)
+    test_encrypt_and_decrypt()
+    
+    print("\n" + "=" * 60)
+    print("测试2: 危险命令拦截")
+    print("=" * 60)
+    test_dangerous_command()
+    
+    print("\n" + "=" * 60)
+    print("测试3: 无效输入处理")
+    print("=" * 60)
+    test_invalid_input()