tamakyi
d690355cdc
以周/月为单位汇总蜜罐命中，把来源 IP 通过公共服务查地理位置，画到地图上，
生成 HTML 报告并发邮件。同时补齐 ipinfo 无法回答「这个 IP 在哪儿」的问题。
新增 IP 地理定位层 internal/security/geoip：
- 与 internal/security/ipinfo 正交。ipinfo 走 cdncheck 离线指纹，只认云厂商/CDN/WAF，
答的是「是不是代理」；肉鸡（家宽、企业宽带、移动网络）必然被判为
「未识别（普通网络）」，所以「在哪儿」必须另建一套。
- ip-api.com 批量打底（100 个/请求、1.4s 节流）+ 腾讯位置服务精修国内
（一 IP 一请求，只精修命中数最高的前 N 个，受 ip_geo_tencent_key 控制）。
- Result{Geo, Unresolved, Requests, Failures} 三态。服务明确回答「没有这个地址」
才允许写失败缓存；请求超时/限流/5xx 一律不写 —— 否则一次抖动会把整批 IP
钉成「查不到」三天，此后每份报告的定位覆盖率都莫名变成 0。
- addrs 按调用方给的顺序构建（不是 map 遍历）：腾讯精修只取前 N 个，乱序会让
每次运行精修到不同 IP，配额花在长尾上。
报告生成 internal/security/report：
- period：PreviousWeek / PreviousMonth，窗口一律 [start, end) 左闭右开，
相邻两期不重叠也不漏边界整点；period_key（week:2026-W44）做幂等。
- aggregate：按 shared.LimitKey 合并（IPv4 精确 / IPv6 折 /64）；国家、城市、
路径、每日趋势；显式统计未定位命中数（覆盖率 60% 的图看起来和 100% 一样）。
- mapdraw：自绘等距圆柱投影气泡图。零海岸线零国界 —— 国界/台湾/南海诸岛绘制
必须符合国家标准，自绘极易画错，真实底图交给腾讯。经纬网 + 度数刻度 +
标签防重叠（同一都市圈两城市坐标差 0.01°，曾把「AU 55」和「AU 30」叠成
看起来像 AU 305 的假数字）。
- html：自包含报告页，内嵌 SVG + 腾讯交互地图（无 key 时给申请指引，
代码里不预置任何 key）。
- countryname：港澳台统一显示「中国香港/中国澳门/中国台湾」，常见国家用中文名。
存储与调度：
- ip_geo_cache（主键 = LimitKey，成功 TTL 90 天 / 明确无数据 TTL 3 天）
- security_report_logs（period_key 唯一索引 → 重启、错过时间窗都能正确补发，
失败重试上限 5 次后停止）
- honeypot report --period week|month [--to] [--dir] [--no-email]
- web 启动时按配置挂 30 分钟一轮的调度器（与保留期清理共用停止信号，
defer 移出 if 外，避免「清理关掉、报告开着」时 goroutine 泄漏）
配置（[security]）：ip_geo_provider=auto|ipapi|tencent|none、ip_geo_tencent_key、
honeypot_report_enabled / _weekly / _monthly / _email / _dir。
顺带修复：
- route/honeypot_test.go 的 captureStore 缺 AggregateSources/AggregatePaths/
TimestampsBetween，整个 route 包编译不过。
- templates/render_test.go 的 mockLoggedUser 缺 TelegramChatID，3 个渲染测试失败。
102 lines
3.7 KiB
Go
102 lines
3.7 KiB
Go
// Copyright 2026 tamakyi. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package report
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestPreviousWeek 覆盖周报窗口：[周一 00:00, 下周一 00:00)，且与「现在是周几」无关。
|
|
func TestPreviousWeek(t *testing.T) {
|
|
loc := time.Local
|
|
|
|
cases := []struct {
|
|
name string
|
|
now time.Time
|
|
wantStart time.Time
|
|
wantEnd time.Time
|
|
}{
|
|
{
|
|
name: "周一凌晨（报告应在此时发出，覆盖上周）",
|
|
now: time.Date(2026, 10, 26, 1, 0, 0, 0, loc), // 周一
|
|
wantStart: time.Date(2026, 10, 19, 0, 0, 0, 0, loc),
|
|
wantEnd: time.Date(2026, 10, 26, 0, 0, 0, 0, loc),
|
|
},
|
|
{
|
|
name: "周三补发：仍然覆盖「上周」而不是本周至今",
|
|
now: time.Date(2026, 10, 28, 15, 30, 0, 0, loc), // 周三
|
|
wantStart: time.Date(2026, 10, 19, 0, 0, 0, 0, loc),
|
|
wantEnd: time.Date(2026, 10, 26, 0, 0, 0, 0, loc),
|
|
},
|
|
{
|
|
name: "周日晚间：还没到本周结束，仍覆盖上周",
|
|
now: time.Date(2026, 10, 25, 23, 59, 0, 0, loc), // 周日
|
|
wantStart: time.Date(2026, 10, 12, 0, 0, 0, 0, loc),
|
|
wantEnd: time.Date(2026, 10, 19, 0, 0, 0, 0, loc),
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
p := PreviousWeek(c.now)
|
|
require.Equal(t, c.wantStart, p.Start)
|
|
require.Equal(t, c.wantEnd, p.End)
|
|
// 窗口必须正好 7 天，且左闭右开
|
|
require.Equal(t, 7*24*time.Hour, p.End.Sub(p.Start))
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestPreviousWeekKey 覆盖 ISO 周号，特别是跨年那一周。
|
|
func TestPreviousWeekKey(t *testing.T) {
|
|
loc := time.Local
|
|
|
|
// 2027-01-01 是周五，属于 ISO 2026 年第 53 周；
|
|
// 2026-12-28（周一）~ 2027-01-03（周日）正是这一周。
|
|
// 若用 end 去算周号会得到 2027-W01，导致这一周的台账键错位。
|
|
p := PreviousWeek(time.Date(2027, 1, 4, 9, 0, 0, 0, loc)) // 周一
|
|
require.Equal(t, "week:2026-W53", p.Key)
|
|
require.Equal(t, time.Date(2026, 12, 28, 0, 0, 0, 0, loc), p.Start)
|
|
|
|
// 跨年另一侧：2026-01-01 所在周是 2026-W01，前一周属于 2025-W52。
|
|
p2 := PreviousWeek(time.Date(2026, 1, 5, 9, 0, 0, 0, loc))
|
|
require.Equal(t, "week:2026-W01", p2.Key)
|
|
}
|
|
|
|
// TestPreviousMonth 覆盖月报窗口，含 31 天/30 天/2 月。
|
|
func TestPreviousMonth(t *testing.T) {
|
|
loc := time.Local
|
|
|
|
p := PreviousMonth(time.Date(2026, 11, 3, 10, 0, 0, 0, loc))
|
|
require.Equal(t, "month:2026-10", p.Key)
|
|
require.Equal(t, time.Date(2026, 10, 1, 0, 0, 0, 0, loc), p.Start)
|
|
require.Equal(t, time.Date(2026, 11, 1, 0, 0, 0, 0, loc), p.End)
|
|
|
|
// 闰年 2 月必须有 29 天，且不能因为「减一个月」跑到 3 月去。
|
|
p2 := PreviousMonth(time.Date(2024, 3, 1, 0, 30, 0, 0, loc))
|
|
require.Equal(t, "month:2024-02", p2.Key)
|
|
require.Equal(t, time.Date(2024, 2, 1, 0, 0, 0, 0, loc), p2.Start)
|
|
require.Equal(t, 29*24*time.Hour, p2.End.Sub(p2.Start))
|
|
|
|
// 5 月 31 日往前一个月不能变成「4 月 31 日」这种不存在的日期：
|
|
// 窗口是 [4-01, 5-01)，不需要做任何日期加减。
|
|
p3 := PreviousMonth(time.Date(2026, 5, 31, 12, 0, 0, 0, loc))
|
|
require.Equal(t, time.Date(2026, 4, 1, 0, 0, 0, 0, loc), p3.Start)
|
|
require.Equal(t, time.Date(2026, 5, 1, 0, 0, 0, 0, loc), p3.End)
|
|
}
|
|
|
|
// TestPeriodLabel 覆盖展示用的窗口描述：右端点要退一天，别让人以为多算了一天。
|
|
func TestPeriodLabel(t *testing.T) {
|
|
loc := time.Local
|
|
p := PreviousWeek(time.Date(2026, 10, 26, 1, 0, 0, 0, loc))
|
|
require.Equal(t, "2026-10-19 ~ 2026-10-25", p.Label())
|
|
|
|
m := PreviousMonth(time.Date(2026, 11, 3, 10, 0, 0, 0, loc))
|
|
require.Equal(t, "2026-10", m.Label())
|
|
require.Equal(t, "security-month-2026-10", m.Filename())
|
|
}
|