Files
DenyIP-go/common.go
aixiao f13283126f feat(common): 添加获取本机公网IP及区域模糊匹配功能
新增以下功能:
- 添加 `GetLocalIpv4Addr` 函数用于获取本机公网IPv4地址,通过外部API实现,并增加基本校验逻辑
- 添加 `ContainsPart` 和 `SplitBySpace` 函数,支持按空格分割字符串并进行区域关键词模糊匹配
- 引入标准库依赖:`fmt`、`io`、`net`、`net/http`、`strings`

同时移除已废弃的文件:
- 删除 `local_addr.go`,相关逻辑迁移至 `common.go`
- 删除 `region_name_match.go`,相关函数整合进 `common.go`
- 删除 `warning_ding.go`,原钉钉告警模块暂不启用

更新二进制文件 `denyip`,删除压缩版本 `denyip.upx`
2025-11-18 16:12:08 +08:00

72 lines
1.5 KiB
Go

package main
import (
"fmt"
"io"
"log"
"net"
"net/http"
"os/exec"
"strings"
)
// CheckCommandExists 检测系统命令是否存在,不存在则退出程序
func CheckCommandExists(cmd string) {
_, err := exec.LookPath(cmd)
if err != nil {
log.Fatalf("命令未找到: %s (请确保已安装该命令)", cmd)
}
}
// SplitBySpace 按空格分割字符串
func SplitBySpace(s string) []string {
return strings.Split(s, " ")
}
// ContainsPart 判断 target 是否包含 regionParts 中的任意片段(模糊匹配)
func ContainsPart(target string, REGION string) bool {
regionParts := SplitBySpace(REGION)
for _, rp := range regionParts {
//fmt.Println("regionParts:", rp)
if strings.Contains(target, rp) {
return true
}
}
return false
}
// 获取本机公网 IPv4 地址
func GetLocalIpv4Addr() string {
resp, err := http.Get("https://inet-ip.aixiao.me/")
if err != nil {
fmt.Printf("请求失败: %s\n", err)
return "NULL"
}
defer resp.Body.Close()
ip, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Printf("读取响应失败: %s\n", err)
return "NULL"
}
ipStr := strings.TrimSpace(string(ip))
// ✅ 验证是否是合法 IPv4
if net.ParseIP(ipStr) == nil || strings.Contains(ipStr, "<") {
fmt.Printf("无效的IP响应: %q\n", ipStr)
return "NULL"
}
log.Printf(" 公网IP: %s\n", ipStr)
return ipStr
}
/*
func main() {
REGION := "中国 内网"
fmt.Println("结果1:", ContainsPart("美国", REGION)) // true
}
*/