fix(security): 修复 P4 封禁记录数据错位 + P5 枚举爆破宽限(方案 A)
P4 #17 手动封禁 Create 非 upsert 导致数据错位: - BanStore 新增 BanIP(ip, reason, duration):事务内清理该 IP 全部 既有记录(兼容历史脏数据)后插入单条封禁记录,计数清零; DisconnectConnection 改用(原裸 Create 为全仓库唯一调用点) - BanEntry.IPAddress 升级 uniqueIndex;InitDB 在 AutoMigrate 前 dedupeBanEntries 清理历史重复行(MySQL 1093 兼容写法) - IncrementFail 原子化:SQL 侧 fail_count+1,miss 时 OnConflict DoNothing 插入兜底并发竞态,回读计数 P5 #18 方案 A(按失败性质区分宽限): - RecordAuthFailure 新增 knownUser 参数:用户名不存在(枚举型 爆破)跳过 3 次宽限、首次触发即封第 1 档;用户名存在(真实 用户输错)保留宽限防误封 - 新增 UserStore.LoginExists(邮箱/裸用户名);五个失败调用点 接线(Web 登录查存在性;SMTP/IMAP/POP3 用登录名;LDAP 侧 存在性不可判定,保守按已知用户处理) - 封禁原因注明「未知用户名,跳过宽限」便于后台审计 新增 8 个测试(-race 通过):BanIP 单行 upsert、唯一索引约束、 16 协程并发计数精确、dedupe 清理/表不存在静默、未知用户即时 封禁、已知用户宽限回归、LoginExists 矩阵。
This commit is contained in:
@@ -2,6 +2,7 @@ package db
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
@@ -55,6 +56,11 @@ func InitDB(cfg config.DatabaseConfig, storageCfg config.StorageConfig) (*gorm.D
|
||||
return nil, fmt.Errorf("连接数据库失败: %w", err)
|
||||
}
|
||||
|
||||
// AutoMigrate 前清理 ban_entries 的历史重复行(保留每 IP 最大 id):
|
||||
// ip_address 将升级为唯一索引,重复行会使索引创建失败。
|
||||
// 首次安装表不存在时忽略错误(AutoMigrate 会建新表)。
|
||||
dedupeBanEntries(db)
|
||||
|
||||
// Auto-migrate all models
|
||||
if err := db.AutoMigrate(&User{}, &Domain{}, &Message{}, &Attachment{}, &BanEntry{}, &OutboundMessage{}, &ProtocolLog{}, &MailboxState{}, &Mailbox{}); err != nil {
|
||||
return nil, fmt.Errorf("数据库迁移失败: %w", err)
|
||||
@@ -62,3 +68,22 @@ func InitDB(cfg config.DatabaseConfig, storageCfg config.StorageConfig) (*gorm.D
|
||||
|
||||
return db, nil
|
||||
}
|
||||
|
||||
// dedupeBanEntries 删除 ban_entries 中同一 ip_address 的重复行(保留
|
||||
// 每组 id 最大的一条),为 ip_address 唯一索引的 AutoMigrate 扫清障碍。
|
||||
// 表不存在(首次安装)时静默跳过;清理失败仅告警,不阻断启动
|
||||
// (索引创建失败会在 AutoMigrate 中显式报错)。
|
||||
func dedupeBanEntries(db *gorm.DB) {
|
||||
// SQLite 与 MySQL 均支持;MySQL 不允许 DELETE 子查询直接引用同表
|
||||
// (1093),因此用派生表包一层。
|
||||
sql := "DELETE FROM ban_entries WHERE id NOT IN (" +
|
||||
"SELECT mid FROM (SELECT MAX(id) AS mid FROM ban_entries GROUP BY ip_address) AS t)"
|
||||
if err := db.Exec(sql).Error; err != nil {
|
||||
// 表不存在(首次安装)为预期情况
|
||||
msg := err.Error()
|
||||
if strings.Contains(msg, "no such table") || strings.Contains(msg, "doesn't exist") {
|
||||
return
|
||||
}
|
||||
log.Printf("清理 ban_entries 重复行失败(唯一索引可能无法创建): %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user