diff --git a/denyip b/denyip index 09f4488..0bc2034 100644 Binary files a/denyip and b/denyip differ diff --git a/go.mod b/go.mod index 3cff896..4fb006b 100644 --- a/go.mod +++ b/go.mod @@ -6,6 +6,6 @@ require github.com/google/gopacket v1.1.19 require ( github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20241220152942-06eb5c6e8230 - golang.org/x/net v0.33.0 // indirect - golang.org/x/sys v0.28.0 // indirect + golang.org/x/net v0.34.0 // indirect + golang.org/x/sys v0.29.0 // indirect ) diff --git a/go.sum b/go.sum index 7a5dbd5..2a2e482 100644 --- a/go.sum +++ b/go.sum @@ -10,11 +10,16 @@ golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I= golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4= +golang.org/x/net v0.34.0 h1:Mb7Mrk043xzHgnRM88suvJFwzVrRfHEHJEl5/71CKw0= +golang.org/x/net v0.34.0/go.mod h1:di0qlW3YNM5oh6GqDGQr92MyTozJPmybPK4Ev/Gm31k= golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA= golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU= +golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k= golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28= golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= diff --git a/main.go b/main.go index ccb773e..615139b 100644 --- a/main.go +++ b/main.go @@ -36,8 +36,8 @@ var ( IpList = list.New() // 存储 IPv4 地址的链表 IpMutex sync.Mutex // 保护 ipList 的互斥锁 - ProcessedIPMap = make(map[string]string) // 使用 map 存储已处理的 IP - ProcessedMutex sync.Mutex // 互斥锁保护 ProcessedIPMap + ProcessedIPMap = map[string]time.Time{} // 使用 map 存储已处理的 IP + ProcessedMutex sync.Mutex // 互斥锁保护 ProcessedIPMap ) // 启动子进程 @@ -187,13 +187,28 @@ func loadFromFile(filePath string, logMessage string) error { } defer file.Close() + // 尝试用新格式解码 decoder := json.NewDecoder(file) - if err := decoder.Decode(&ProcessedIPMap); err != nil { - return fmt.Errorf("解码 Map 失败: %w", err) + temp := make(map[string]time.Time) + if err := decoder.Decode(&temp); err == nil { + ProcessedIPMap = temp + log.Println(logMessage) + return nil } - log.Println(logMessage) - return nil + // 如果失败,尝试旧格式 + file.Seek(0, 0) // 重置读取位置 + decoder = json.NewDecoder(file) + oldTemp := make(map[string]string) + if err := decoder.Decode(&oldTemp); err == nil { + for ip := range oldTemp { + ProcessedIPMap[ip] = time.Now() // 给旧 IP 打个当前时间戳 + } + log.Println(logMessage + "(从旧格式转换)") + return nil + } + + return fmt.Errorf("解码 Map 失败: %w", err) } func InitMap() { @@ -220,7 +235,7 @@ func WriteLocalAddr() { if local_ipv4_addr != "NULL" { //log.Printf("\033[33m %s 本机地址 \033[0m\n", ipStr) ProcessedMutex.Lock() - ProcessedIPMap[local_ipv4_addr] = "" + ProcessedIPMap[local_ipv4_addr] = time.Now() ProcessedMutex.Unlock() } // 写入json文件 @@ -287,7 +302,7 @@ func RunMainProcess() { // 主进程逻辑 // 将 IP 标记为已处理,国内地址 ProcessedMutex.Lock() - ProcessedIPMap[ipStr] = "" + ProcessedIPMap[ipStr] = time.Now() ProcessedMutex.Unlock() // 写入json文件 @@ -309,28 +324,45 @@ func RunMainProcess() { // 主进程逻辑 IpMutex.Unlock() // 解锁互斥锁 // 打印当前info - /* - log.Printf(" 当前Ip链表长度:%d, Ipset名:%s, 长:%d, ProcessedIPMap:[%s]当前长度:%d\n", IpList.Len(), IPSET_NAME, func() int { // 打印 当前 Ipset 链长度 - _len, _err := NumIPSet(IPSET_NAME) - if _err == nil { - return _len - } - return 0 - }(), ProcessedIPMap, len(ProcessedIPMap)) - */ - - // 清理 ProcessedIPMap - if len(ProcessedIPMap) >= 1024 { - ProcessedMutex.Lock() // 锁定互斥锁 - ProcessedIPMap = make(map[string]string) // 创建一个新的空 map - ProcessedMutex.Unlock() // 解锁 - log.Println("ProcessedIPMap 已清空") - } + log.Printf(" 当前Ip链表长度:%d, Ipset名:%s, 长:%d, ProcessedIPMap:[%s]当前长度:%d\n", IpList.Len(), IPSET_NAME, func() int { // 打印 当前 Ipset 链长度 + _len, _err := NumIPSet(IPSET_NAME) + if _err == nil { + return _len + } + return 0 + }(), ProcessedIPMap, len(ProcessedIPMap)) time.Sleep(1 * time.Second) // 防止高频运行 } }() + // 启动一个 goroutine(后台任务),用于定期清理 ProcessedIPMap 中过期的 IP 条目 + go func() { + for { + time.Sleep(1 * time.Minute) + + // 获取当前时间,用于与 map 中的时间戳进行比较 + now := time.Now() + + // 加锁,确保并发安全地访问全局变量 ProcessedIPMap + ProcessedMutex.Lock() + + // 遍历 ProcessedIPMap,检查每个 IP 的记录是否已超过 24 小时 + for ip, t := range ProcessedIPMap { + // 如果当前时间减去记录时间大于 1 小时,则删除该条目 + if now.Sub(t) > 10*time.Minute { + delete(ProcessedIPMap, ip) + } + } + + // 解锁,允许其他 goroutine 访问 ProcessedIPMap + ProcessedMutex.Unlock() + + // 打印日志,表示本次清理已完成 + log.Println("已清理过期 ProcessedIPMap 项") + } + }() + // 启动防火墙管理 go func() { for {