fix(backend): 补齐三处漏洞——暂停租户拦截/金额不符落审计终态/邀请码列表滤失效
P0 暂停租户是空开关:admin 能设 suspended,但 preflight 只查预算+余额、不看 租户 status → 暂停后照样能提交烧积分。加 TenantSuspended 校验(活跃租户 + 分叉时 的计费租户都拦),403 拒绝。 P1 金额不符只刷日志:回调/查单判了不符却没落审计、订单永远卡 pending 被补偿定时器 每轮重扫刷屏。加 disputed 终态 + MarkOrderDisputed(CAS 只挂一次) + 审计(首次写一次); disputed 不在 pending 扫描内,停止无限重扫。admin /orders?status=disputed 可查。 P1 邀请码列表混入失效码:ListInvites 只按 status=active 过滤,过期/满员的码仍显示为 有效、误导邀请人。有效列表加 expires_at>now 且 used<max 过滤(RedeemInvite 本就会拒, 这里修的是展示一致性)。 三处均带 store 单测(TenantSuspended/MarkOrderDisputed CAS/ListInvites 过滤)。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,84 @@
|
||||
package store
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
// 暂停租户是个真管控,不能是空开关:TenantSuspended 必须如实反映 status。
|
||||
func TestTenantSuspended(t *testing.T) {
|
||||
p := newTestStore(t)
|
||||
ctx := context.Background()
|
||||
seedTenant(t, p, "t1") // seedTenant 建的是 active
|
||||
if p.TenantSuspended(ctx, "t1") {
|
||||
t.Fatal("active 租户不该报暂停")
|
||||
}
|
||||
if err := p.SetTenantStatus(ctx, "t1", "suspended"); err != nil {
|
||||
t.Fatalf("置暂停失败: %v", err)
|
||||
}
|
||||
if !p.TenantSuspended(ctx, "t1") {
|
||||
t.Fatal("suspended 租户应报暂停")
|
||||
}
|
||||
// 查不到的租户按未暂停处理(宁放行不误封)。
|
||||
if p.TenantSuspended(ctx, "nope") {
|
||||
t.Fatal("不存在的租户不该报暂停")
|
||||
}
|
||||
}
|
||||
|
||||
// 金额不符挂起:CAS 只挂一次,且不动已 paid 单。
|
||||
func TestMarkOrderDisputed_CAS(t *testing.T) {
|
||||
p := newTestStore(t)
|
||||
ctx := context.Background()
|
||||
seedTenant(t, p, "t1")
|
||||
o := &PaymentOrder{TenantID: "t1", UserID: "u1", AmountFen: 990, Channel: ChannelWechat, Status: OrderPending}
|
||||
if err := p.CreateOrder(ctx, o); err != nil {
|
||||
t.Fatalf("建单失败: %v", err)
|
||||
}
|
||||
changed, err := p.MarkOrderDisputed(ctx, o.ID)
|
||||
if err != nil || !changed {
|
||||
t.Fatalf("首次挂起应 changed=true: %v %v", changed, err)
|
||||
}
|
||||
again, _ := p.MarkOrderDisputed(ctx, o.ID)
|
||||
if again {
|
||||
t.Fatal("重复挂起应 changed=false(审计只写一次)")
|
||||
}
|
||||
got, _ := p.GetOrder(ctx, o.ID)
|
||||
if got.Status != OrderDisputed {
|
||||
t.Fatalf("状态应 disputed,得 %q", got.Status)
|
||||
}
|
||||
|
||||
// 已 paid 的单不能被挂起(CAS 只认 pending)。
|
||||
paid := &PaymentOrder{TenantID: "t1", UserID: "u1", AmountFen: 990, Channel: ChannelWechat, Status: OrderPaid}
|
||||
p.CreateOrder(ctx, paid)
|
||||
if c, _ := p.MarkOrderDisputed(ctx, paid.ID); c {
|
||||
t.Fatal("已 paid 单不该能挂起")
|
||||
}
|
||||
}
|
||||
|
||||
// 有效邀请码列表要滤掉过期/满员,否则误导邀请人。
|
||||
func TestListInvites_FiltersDeadCodes(t *testing.T) {
|
||||
p := newTestStore(t)
|
||||
ctx := context.Background()
|
||||
seedTenant(t, p, "t1")
|
||||
|
||||
good, _ := p.CreateInvite(ctx, "t1", "inv", RoleMember, time.Now().Add(time.Hour), 0)
|
||||
expired, _ := p.CreateInvite(ctx, "t1", "inv", RoleMember, time.Now().Add(-time.Hour), 0)
|
||||
full, _ := p.CreateInvite(ctx, "t1", "inv", RoleMember, time.Now().Add(time.Hour), 1)
|
||||
// 把 full 灌满
|
||||
p.db.Model(&TenantInvite{}).Where("id = ?", full.ID).UpdateColumn("used_count", 1)
|
||||
|
||||
active := p.ListInvites(ctx, "t1", true)
|
||||
if len(active) != 1 || active[0].ID != good.ID {
|
||||
ids := make([]string, len(active))
|
||||
for i, a := range active {
|
||||
ids[i] = a.ID
|
||||
}
|
||||
t.Fatalf("有效列表应只剩 good(%s),得 %v", good.ID, ids)
|
||||
}
|
||||
_ = expired
|
||||
// onlyActive=false 仍应看到全部三条(管理/审计用途)。
|
||||
if all := p.ListInvites(ctx, "t1", false); len(all) != 3 {
|
||||
t.Fatalf("全量列表应 3 条,得 %d", len(all))
|
||||
}
|
||||
}
|
||||
@@ -100,7 +100,10 @@ func (p *Postgres) ListInvites(ctx context.Context, tenantID string, onlyActive
|
||||
}
|
||||
q := p.db.WithContext(ctx).Where("tenant_id = ?", tenantID)
|
||||
if onlyActive {
|
||||
q = q.Where("status = ?", InviteActive)
|
||||
// 「有效」= 未撤销 + 未过期 + 未满员。只看 status 会把过期/满员的码当有效展示、误导邀请人。
|
||||
q = q.Where("status = ?", InviteActive).
|
||||
Where("expires_at > ?", time.Now()).
|
||||
Where("max_uses = 0 OR used_count < max_uses")
|
||||
}
|
||||
var out []TenantInvite
|
||||
q.Order("created_at desc").Limit(100).Find(&out)
|
||||
|
||||
@@ -32,6 +32,7 @@ const (
|
||||
OrderFailed = "failed"
|
||||
OrderExpired = "expired"
|
||||
OrderRefunded = "refunded"
|
||||
OrderDisputed = "disputed" // 渠道已付但金额与订单不符:不入账、挂起待人工核对(终态,不再重扫)
|
||||
)
|
||||
|
||||
// 渠道名。P5.1 只有 redeem;wechat 在 P5.2 挂上。
|
||||
@@ -220,6 +221,21 @@ func (p *Postgres) GetOrder(ctx context.Context, id string) (*PaymentOrder, erro
|
||||
return &o, nil
|
||||
}
|
||||
|
||||
// MarkOrderDisputed 把一张 pending 订单 CAS 置为 disputed(渠道已付但金额不符)。
|
||||
// 返回 changed:仅首次转移为 true,供调用方决定是否只写一次审计。CAS 保证并发/重复回调只挂起一次。
|
||||
// disputed 是终态:补偿定时器只扫 pending,从此不再重复查它、不再刷屏。
|
||||
func (p *Postgres) MarkOrderDisputed(ctx context.Context, orderID string) (bool, error) {
|
||||
if p.db == nil {
|
||||
return false, errStoreDisabled
|
||||
}
|
||||
res := p.db.WithContext(WithoutTenant(ctx)).Model(&PaymentOrder{}).
|
||||
Where("id = ? AND status = ?", orderID, OrderPending).Update("status", OrderDisputed)
|
||||
if res.Error != nil {
|
||||
return false, res.Error
|
||||
}
|
||||
return res.RowsAffected > 0, nil
|
||||
}
|
||||
|
||||
// MarkOrderPaid 渠道确认已支付后的入账:一个事务里「订单 CAS(pending→paid) → 分录 → 物化余额」。
|
||||
// 返回 changed=false 表示这单已被处理过(回调重复推送/回调与主动查单赛跑),幂等直接成功。
|
||||
// 双闸:CAS 是主闸;credit_ledger (kind,ref=订单号) 唯一索引兜底。
|
||||
|
||||
@@ -171,6 +171,18 @@ func (p *Postgres) GetTenant(ctx context.Context, id string) (*Tenant, error) {
|
||||
return &t, nil
|
||||
}
|
||||
|
||||
// TenantSuspended 报告租户是否被暂停(轻查询,供提交前门控)。查不到/出错按未暂停处理,
|
||||
// 宁可放行也不误封(暂停是显式管控动作,缺数据时不该凭空拦人)。
|
||||
func (p *Postgres) TenantSuspended(ctx context.Context, tenantID string) bool {
|
||||
if p.db == nil || tenantID == "" {
|
||||
return false
|
||||
}
|
||||
var status string
|
||||
p.db.WithContext(WithoutTenant(ctx)).Model(&Tenant{}).
|
||||
Where("id = ?", tenantID).Select("status").Scan(&status)
|
||||
return status == "suspended"
|
||||
}
|
||||
|
||||
// MemberRole 返回用户在某租户的角色(无成员关系返回空)。
|
||||
func (p *Postgres) MemberRole(ctx context.Context, tenantID, userID string) string {
|
||||
if p.db == nil {
|
||||
|
||||
Reference in New Issue
Block a user