fix(agent)★: relay 的 budget_charged 回传撒谎 —— Agent↔Agent 明明扣了却说没扣

## 实测发现的(Agent↔Agent 全场景回归中)

    relay→Agent : budget_charged=False   ← 错
    relay→人类  : budget_charged=False   ← 对
    普通→Agent  : (无该字段 = 正常扣费)

查库确认 relay→Agent 的会话 **used_rounds=1(确实扣了)**,
而同一响应里回传 `budget_charged=false`。

## 根因

    if relay != "" {
        resp["relay"] = relay
        resp["budget_charged"] = false    // ← 只看 relay 非空
    }

而真正的计费判定在别处,且是对的:

    relayFree := false
    if relay != "" {
        human, _ := repo.IsHumanUser(ctx, to.Name)
        relayFree = human       // ← 只有发给人类才免
    }

## 为什么这个字段的错不是文案瑕疵

`relayFree` 的注释写着:「Agent→Agent 的 relay 照样扣会话预算,
max_rounds 就能截断它」,理由是「Agent↔Agent 双方都不在做决定,
整个回路没有任何一处在计数」——生产上跑出过**41 封、间隔从 15 分钟
缩到 5 秒的无穷循环**(会话 f3d824ce)。

`budget_charged` 正是那条链上的判定依据之一:插件/宿主据此认为
「还有额度」而继续发。

## 顺带实测确认的三道 Agent↔Agent 专属防线

1. **新建会话速率限制**:24 次 `.new` 里 18 次被拦
   (1 小时上限 20 条),文案含剩余秒数。
2. **冷静期**:普通 Agent↔Agent 回信第 1 封进冷静期,
   后续被拒 —— 提示「上一次连续 N 封」。
3. **relay 连转发上限 5**:★ relay **不是**绕过冷静期的后门 ——
   实测第 6 封被「本会话已连续 5 封自动转发(上限 5)」挡住,
   并说明「若确实需要继续,请由模型主动调 send_mail(不带 relay)」。
   这三道防线都在,41 封循环的结构性防护完整。

## 判据(2 格,2 个变异都转红)

- relay→人类:确实免配额 + 回传 false
- relay→Agent:**要扣** + 回传 true + 库里 used_rounds=1

变异①改回 `= false` → 转红;
变异②让 `relayFree` 对 Agent 也为 true → 转红
(第一次试这个变异时 0 红,查下去是**编译失败**(`human` 变量
 变成 unused)—— 变异无效而不是判据无效。改成 `human || true`
  才真正生效。**「0 红」的第一反应必须是「变异没生效」。**)

14 包全绿。
This commit is contained in:
2026-10-04 16:42:43 +08:00
parent a6b7994a45
commit 1324232946
2 changed files with 142 additions and 2 deletions

View File

@ -658,9 +658,22 @@ func SendMail(w http.ResponseWriter, r *http.Request) {
resp["budget_max"] = budget.Max
}
if relay != "" {
// 告知本次未扣预算,否则插件看到 budget_remaining 没变会以为数据错了
resp["relay"] = relay
resp["budget_charged"] = false
// ★ 必须用 relayFree,而不是 relay != ""(2026-10-04 实测修)。
//
// 原写法在**任何** relay 上都回 budget_charged=false,而实际上:
// relay 发给**人类** → relayFree=true → 确实没扣 ✓
// relay 发给**另一个 Agent** → relayFree=false → **扣了**
// (见上面 relayFree 的注释:Agent↔Agent 双方都不在做决定,
// 整个回路没有任何一处在计数 —— 生产上跑出过 41 封、间隔从
// 15 分钟缩到 5 秒的无穷循环。免配额只给发往人类的 relay。)
//
// 后果:Agent↔Agent 的 relay 明明扣了预算,却告诉调用方没扣。
// 插件据此判断「还有额度」而继续发 —— 正是那个循环的判定依据。
//
// 实测证据:relay→Agent 的会话 used_rounds=1(确实扣了),
// 而同一响应里 budget_charged=false。
resp["budget_charged"] = !relayFree
}
if proposal != nil {
// 回传规范化后的别名:Agent 提的名字可能含非法字符被改写过,

View File

@ -0,0 +1,127 @@
package handler
/*
relay 的配额回传必须与实际扣费一致(2026-10-04 实测驱动)。
★ 缺陷:`resp["budget_charged"] = false` 只看 `relay != ""`,
于是**任何** relay 都回「没扣预算」。而实际:
relay → 人类 relayFree=true 确实没扣 回传 false ✓
relay → 另一个 Agent relayFree=false **扣了** 回传 false ✗ 撒谎
★ 为什么这个字段的错很要紧(不是文案瑕疵):
插件据 budget_charged 判断「还有额度」,Agent↔Agent 的 relay
明明扣了却被告知没扣 —— 而「Agent↔Agent 双方都不计数」正是
生产上那个 41 封、间隔 15 分钟→5 秒的无穷循环(会话 f3d824ce)。
这个字段就是那条链上的判定依据之一。
★ 实测证据:relay→Agent 的会话 used_rounds=1(确实扣了),
同一响应里 budget_charged=false。
*/
import (
"context"
"net/http"
"net/http/httptest"
"path/filepath"
"strings"
"testing"
"github.com/agentmail/gateway/internal/db"
"github.com/agentmail/gateway/internal/middleware"
"github.com/agentmail/gateway/internal/models"
)
func setupRelayBudgetDB(t *testing.T) {
t.Helper()
db.Close()
p := filepath.Join(t.TempDir(), "relay-budget.db")
if err := db.Connect(context.Background(), "sqlite://"+p); err != nil {
t.Fatalf("连接测试库: %v", err)
}
if err := db.Migrate(context.Background()); err != nil {
t.Fatalf("迁移: %v", err)
}
t.Cleanup(db.Close)
}
func mustAgentForBudget(t *testing.T, name string) {
t.Helper()
if _, err := db.DB.ExecContext(context.Background(),
`INSERT INTO agents (agent_name, secret, platform) VALUES ($1,'x','test')
ON CONFLICT (agent_name) DO NOTHING`, name); err != nil {
t.Fatalf("建 agent %s: %v", name, err)
}
}
func mustHumanForBudget(t *testing.T, name string) {
t.Helper()
if _, err := db.DB.ExecContext(context.Background(),
`INSERT INTO users (username, password_hash, role) VALUES ($1,'x','user')
ON CONFLICT (username) DO NOTHING`, name); err != nil {
t.Fatalf("建用户 %s: %v", name, err)
}
}
func postMailAs(t *testing.T, agent, body string) *httptest.ResponseRecorder {
t.Helper()
req := httptest.NewRequest(http.MethodPost, "/api/v1/mail/send", strings.NewReader(body))
req = req.WithContext(context.WithValue(req.Context(), middleware.AgentNameKey, agent))
rec := httptest.NewRecorder()
SendMail(rec, req)
return rec
}
// relay 发给人类:确实免配额,且回传 budget_charged=false
func TestRelayToHumanIsFreeAndReportsSo(t *testing.T) {
setupRelayBudgetDB(t)
mustAgentForBudget(t, "sender-h")
mustHumanForBudget(t, "human-h")
rec := postMailAs(t, "sender-h",
`{"to":"human-h@/tmp.new","subject":"总结","body":"x","relay":"summary","relay_key":"h1"}`)
if rec.Code != http.StatusOK {
t.Fatalf("应 200,实际 %d:%s", rec.Code, rec.Body.String())
}
body := rec.Body.String()
if !strings.Contains(body, `"budget_charged":false`) {
t.Errorf("relay 发给人类应回 budget_charged=false,实际:%s", body)
}
}
// ★ relay 发给另一个 Agent:**要扣**预算,且回传必须说 true
func TestRelayToAgentIsChargedAndReportsTruthfully(t *testing.T) {
setupRelayBudgetDB(t)
mustAgentForBudget(t, "sender-a")
mustAgentForBudget(t, "receiver-a")
rec := postMailAs(t, "sender-a",
`{"to":"receiver-a@/tmp.new","subject":"relay给Agent","body":"x","relay":"summary","relay_key":"a1"}`)
if rec.Code != http.StatusOK {
t.Fatalf("应 200,实际 %d:%s", rec.Code, rec.Body.String())
}
body := rec.Body.String()
// ① 回传不能说谎
if !strings.Contains(body, `"budget_charged":true`) {
t.Errorf("★ relay 发给 Agent **扣了**预算,回传却说 budget_charged=false"+
"(插件据此以为还有额度而继续发):%s", body)
}
// ② 而且库里的计数必须真的动了
var sid string
if err := db.DB.QueryRowContext(context.Background(),
`SELECT session_id FROM mails ORDER BY created_at DESC LIMIT 1`).Scan(&sid); err != nil {
t.Fatalf("取会话: %v", err)
}
var used int
if err := db.DB.QueryRowContext(context.Background(),
`SELECT used_rounds FROM sessions WHERE session_id = $1`, sid).Scan(&used); err != nil {
t.Fatalf("读 used_rounds: %v", err)
}
if used != 1 {
t.Errorf("★ relay 发给 Agent 应扣 1 次预算,实际 used_rounds=%d", used)
}
}
var _ = models.Mail{}