@kenhkl/agent-moss 0.7.13 → 0.8.1

package/dist/agents/__init__.py

@@ -0,0 +1,92 @@
+"""三方 agent 定制化注册中心
+
+按 `agent_id` 加载该 agent 的专属规则集（JSON 单一事实源）。
+规则文件位置：`agent_moss/agents/<agent_id>/rules.json`
+
+Python 与 TypeScript 两端共享同一份 rules.json：
+- py: agent_moss/agents/__init__.py:get_agent_extensions
+- ts: ts/src/agents/index.ts:getAgentExtensions
+
+使用：
+    from agent_moss.agents import get_agent_extensions
+    ext = get_agent_extensions("xiaoo")
+    if ext:
+        # 把 ext.extra_command_patterns 追加到通用规则之后
+        ...
+
+无 agent_id 或 rules.json 不存在时返回 None（通用模式）。
+"""
+
+from pathlib import Path
+import threading
+import logging
+
+from .base import AgentExtensions, InjectionKeyword, CommandPattern, SensitivePath
+
+logger = logging.getLogger(__name__)
+
+_AGENTS_DIR = Path(__file__).parent
+_cache: dict[str, AgentExtensions | None] = {}
+_lock = threading.Lock()
+
+
+def get_agent_extensions(agent_id: str) -> AgentExtensions | None:
+    """按 agent_id 加载该 agent 的专属规则集。
+
+    Args:
+        agent_id: 调用方 agent 标识，如 "xiaoo" / "opencode" / "opendesk"。
+                  空字符串返回 None（通用模式）。
+
+    Returns:
+        AgentExtensions 实例，或 None（无 agent 专属规则）。
+    """
+    if not agent_id:
+        return None
+
+    # 双重检查锁定，避免并发重复 IO
+    if agent_id in _cache:
+        return _cache[agent_id]
+
+    with _lock:
+        if agent_id in _cache:
+            return _cache[agent_id]
+
+        rules_file = _AGENTS_DIR / agent_id / "rules.json"
+        if not rules_file.exists():
+            logger.debug("agent %s 无 rules.json，走通用模式", agent_id)
+            _cache[agent_id] = None
+            return None
+
+        try:
+            ext = AgentExtensions.from_json(rules_file)
+            _cache[agent_id] = ext
+            logger.info(
+                "加载 agent %s 定制规则: %d cmd patterns, %d injection keywords, "
+                "%d sensitive paths, %d builtin safe skills",
+                agent_id,
+                len(ext.extra_command_patterns),
+                len(ext.extra_injection_keywords),
+                len(ext.extra_sensitive_paths),
+                len(ext.builtin_safe_skills),
+            )
+            return ext
+        except Exception as e:
+            logger.error("加载 agent %s rules.json 失败: %s", agent_id, e)
+            _cache[agent_id] = None
+            return None
+
+
+def clear_agent_extensions_cache() -> None:
+    """清空 agent 扩展缓存（用于测试或 rules.json 热加载）"""
+    with _lock:
+        _cache.clear()
+
+
+__all__ = [
+    "AgentExtensions",
+    "InjectionKeyword",
+    "CommandPattern",
+    "SensitivePath",
+    "get_agent_extensions",
+    "clear_agent_extensions_cache",
+]

package/dist/agents/base.py

@@ -0,0 +1,116 @@
+"""三方 agent 定制化扩展 — 数据载体
+
+`AgentExtensions` 描述某个宿主 agent（xiaoO / opencode / opendesk）在通用安全规则
+之上需要追加的专属规则。规则以 JSON 单一事实源存储（agent_moss/agents/<id>/rules.json），
+Python 与 TypeScript 两端共享同一份文件。
+
+设计原则：
+- 所有字段以 `extra_` 为前缀，表示"追加到通用规则之后"，不覆盖通用规则。
+- 通用规则先匹配；命中即短路，避免重复检测。
+- agent 专属规则只做"通用规则没覆盖的宿主 agent 特有风险"。
+
+JSON schema（agent_moss/agents/<id>/rules.json）：
+{
+  "agent_id": "xiaoo",
+  "description": "xiaoO 宿主 agent 专属安全规则",
+  "extra_command_patterns": [CommandPattern...],
+  "extra_injection_keywords": [InjectionKeyword...],
+  "extra_sensitive_paths": [SensitivePath...],
+  "builtin_safe_skills": ["xiaoo-guardian"],
+  "extra_skills": []
+}
+"""
+
+from dataclasses import dataclass, field
+from pathlib import Path
+import json
+import logging
+
+from agent_moss.profiles.base import CommandPattern, SensitivePath
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass
+class InjectionKeyword:
+    """Prompt 注入关键词条目"""
+    keyword: str
+    risk_level: str      # low / medium / high / critical
+    lang: str            # en / zh
+
+
+@dataclass
+class AgentExtensions:
+    """三方 agent 定制化扩展
+
+    由 `agent_moss.agents.get_agent_extensions(agent_id)` 按需加载。
+    缺失 rules.json 时返回 None（通用模式）。
+    """
+    agent_id: str
+    description: str = ""
+    extra_command_patterns: list[CommandPattern] = field(default_factory=list)
+    extra_injection_keywords: list[InjectionKeyword] = field(default_factory=list)
+    extra_sensitive_paths: list[SensitivePath] = field(default_factory=list)
+    builtin_safe_skills: list[str] = field(default_factory=list)
+    extra_skills: list[dict] = field(default_factory=list)
+
+    @classmethod
+    def from_json(cls, rules_file: Path) -> "AgentExtensions":
+        """从 rules.json 加载 AgentExtensions。
+
+        Args:
+            rules_file: agent_moss/agents/<id>/rules.json 的绝对路径
+
+        Returns:
+            AgentExtensions 实例
+
+        Raises:
+            ValueError: JSON 结构不合法
+        """
+        with open(rules_file, "r", encoding="utf-8") as f:
+            data = json.load(f)
+
+        agent_id = data.get("agent_id", "")
+        if not agent_id:
+            raise ValueError(f"rules.json 缺少 agent_id 字段: {rules_file}")
+
+        # extra_command_patterns: dict → CommandPattern
+        extra_cmd_patterns: list[CommandPattern] = []
+        for p in data.get("extra_command_patterns", []):
+            extra_cmd_patterns.append(CommandPattern(
+                pattern=p["pattern"],
+                risk_level=p["risk_level"],
+                risk_type=p["risk_type"],
+                reason=p["reason"],
+            ))
+
+        # extra_injection_keywords: dict → InjectionKeyword
+        extra_inj_keywords: list[InjectionKeyword] = []
+        for k in data.get("extra_injection_keywords", []):
+            extra_inj_keywords.append(InjectionKeyword(
+                keyword=k["keyword"],
+                risk_level=k["risk_level"],
+                lang=k.get("lang", "en"),
+            ))
+
+        # extra_sensitive_paths: dict → SensitivePath
+        extra_sensitive_paths: list[SensitivePath] = []
+        for sp in data.get("extra_sensitive_paths", []):
+            extra_sensitive_paths.append(SensitivePath(
+                path=sp["path"],
+                risk_level=sp["risk_level"],
+                desc=sp["desc"],
+                credential=sp.get("credential", False),
+                read_only=sp.get("read_only", False),
+                deny_mode=sp.get("deny_mode", ""),
+            ))
+
+        return cls(
+            agent_id=agent_id,
+            description=data.get("description", ""),
+            extra_command_patterns=extra_cmd_patterns,
+            extra_injection_keywords=extra_inj_keywords,
+            extra_sensitive_paths=extra_sensitive_paths,
+            builtin_safe_skills=data.get("builtin_safe_skills", []),
+            extra_skills=data.get("extra_skills", []),
+        )

package/dist/agents/xiaoo/README.md

@@ -0,0 +1,41 @@
+# xiaoO 宿主 agent 定制化规则
+
+## 文件
+
+- `rules.json` — xiaoO 专属规则的单一事实源，Python 与 TypeScript 两端共享
+
+## 规则内容
+
+### extra_command_patterns（5 条）
+
+同步自 `xiaoO/plugins/hookers/audit_agent/audit_policy_checker/audit_policy_checker/security/heuristic_detector.py` 行 98–132：
+
+1. `xiaoo\.env|xiaoo\.toml|llm_secrets\.json` — 访问 xiaoO 配置/密钥文件（high）
+2. `/usr/lib/\.xiaoo/skills/xiaoo-guardian/.*<写命令>` — 系统级 xiaoO-guardian 目录写保护（critical）
+3. `<写命令>.* /usr/lib/\.xiaoo/skills/xiaoo-guardian/` — 反序匹配（critical）
+4. `\.xiaoo/skills/xiaoo-guardian/.*<写命令>` — 用户级 xiaoO-guardian 目录写保护（critical）
+5. `<写命令>.*\.xiaoo/skills/xiaoo-guardian/` — 反序匹配（critical）
+
+写命令集合：`rm|del|delete|remove|修改|删除|chmod|chown|write|save|保存|modify|更新|覆盖|replace|truncate|追加|append`
+
+### extra_injection_keywords（6 条）
+
+xiaoO 专属绕过词（同步自 audit_agent heuristic_detector.py 行 361–367）：
+
+- EN: `bypass audit` / `skip security check` / `disable guard`（全 critical）
+- ZH: `绕过审计` / `禁用安全检查` / `关闭防护`（全 critical）
+
+### builtin_safe_skills
+
+`["xiaoo-guardian"]` — xiaoO 内置安全 Skill，coordinator 直接放行（不递归分析其内容）。
+
+## 加载机制
+
+- py: `agent_moss.agents.get_agent_extensions("xiaoo")` → `AgentExtensions`
+- ts: `ts/src/agents/index.ts:getAgentExtensions("xiaoo")` → `AgentExtensions`
+
+detection pipeline 把 `extra_*` 追加到通用规则之后；通用规则先匹配命中即短路。
+
+## 调用方
+
+`hooks/xiaoO/bridge.py` 在 POST `/api/v1/analyze` 时带 `agent_id: "xiaoo"`，AgentMoss 服务端据此加载本规则集。

package/dist/agents/xiaoo/rules.json

@@ -0,0 +1,47 @@
+{
+  "agent_id": "xiaoo",
+  "description": "xiaoO 宿主 agent 专属安全规则。在通用 Linux profile 之上追加 xiaoO 配置/密钥/安全 Skill 目录的写保护，以及 xiaoO 专属绕过词。同步自 audit_agent heuristic_detector.py 行 98-132 + 行 361-367。",
+  "extra_command_patterns": [
+    {
+      "pattern": "xiaoo\\.env|xiaoo\\.toml|llm_secrets\\.json",
+      "risk_level": "high",
+      "risk_type": "config_tampering",
+      "reason": "检测到尝试访问或修改 xiaoO 配置文件"
+    },
+    {
+      "pattern": "/usr/lib/\\.xiaoo/skills/xiaoo-guardian/.*\\b(rm|del|delete|remove|修改|删除|chmod|chown|write|save|保存|modify|更新|覆盖|replace|truncate|追加|append)\\b",
+      "risk_level": "critical",
+      "risk_type": "config_tampering",
+      "reason": "检测到尝试修改或删除 xiaoO 系统级安全防护 Skill 目录"
+    },
+    {
+      "pattern": "\\b(rm|del|delete|remove|修改|删除|chmod|chown|write|save|保存|modify|更新|覆盖|replace|truncate|追加|append)\\b.* /usr/lib/\\.xiaoo/skills/xiaoo-guardian/",
+      "risk_level": "critical",
+      "risk_type": "config_tampering",
+      "reason": "检测到尝试修改或删除 xiaoO 系统级安全防护 Skill 目录"
+    },
+    {
+      "pattern": "\\.xiaoo/skills/xiaoo-guardian/.*\\b(rm|del|delete|remove|修改|删除|chmod|chown|write|save|保存|modify|更新|覆盖|replace|truncate|追加|append)\\b",
+      "risk_level": "critical",
+      "risk_type": "config_tampering",
+      "reason": "检测到尝试修改或删除 xiaoO 用户级安全防护 Skill 目录"
+    },
+    {
+      "pattern": "\\b(rm|del|delete|remove|修改|删除|chmod|chown|write|save|保存|modify|更新|覆盖|replace|truncate|追加|append)\\b.*\\.xiaoo/skills/xiaoo-guardian/",
+      "risk_level": "critical",
+      "risk_type": "config_tampering",
+      "reason": "检测到尝试修改或删除 xiaoO 用户级安全防护 Skill 目录"
+    }
+  ],
+  "extra_injection_keywords": [
+    { "keyword": "bypass audit", "risk_level": "critical", "lang": "en" },
+    { "keyword": "skip security check", "risk_level": "critical", "lang": "en" },
+    { "keyword": "disable guard", "risk_level": "critical", "lang": "en" },
+    { "keyword": "绕过审计", "risk_level": "critical", "lang": "zh" },
+    { "keyword": "禁用安全检查", "risk_level": "critical", "lang": "zh" },
+    { "keyword": "关闭防护", "risk_level": "critical", "lang": "zh" }
+  ],
+  "extra_sensitive_paths": [],
+  "builtin_safe_skills": ["xiaoo-guardian"],
+  "extra_skills": []
+}