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`
This commit is contained in:
57
common.go
57
common.go
@@ -1,8 +1,13 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
"log"
|
"log"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
// CheckCommandExists 检测系统命令是否存在,不存在则退出程序
|
// CheckCommandExists 检测系统命令是否存在,不存在则退出程序
|
||||||
@@ -12,3 +17,55 @@ func CheckCommandExists(cmd string) {
|
|||||||
log.Fatalf("命令未找到: %s (请确保已安装该命令)", cmd)
|
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
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|||||||
BIN
denyip.upx
BIN
denyip.upx
Binary file not shown.
@@ -1,35 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"net"
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
||||||
@@ -1,31 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// 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
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
func main() {
|
|
||||||
REGION := "中国 内网"
|
|
||||||
|
|
||||||
fmt.Println("结果1:", ContainsPart("美国", REGION)) // true
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
@@ -1,83 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"encoding/json"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"net/http"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
WarningList = make(map[string]string) // 存储 IP 和位置的映射
|
|
||||||
WarningListLock = sync.RWMutex{}
|
|
||||||
|
|
||||||
dingtalkWebhook = "https://oapi.dingtalk.com/robot/send?access_token=7f069c672cb878987aa6772cca336740eece4ce36bde12b51b45e9f440e0565a" // 替换为你的 token
|
|
||||||
)
|
|
||||||
|
|
||||||
// 发送钉钉 Markdown 消息
|
|
||||||
func sendDingTalkMessage(title, text string) {
|
|
||||||
if len(text) > 1900 {
|
|
||||||
text = text[:1900] + "\n...(内容过长,已截断)"
|
|
||||||
}
|
|
||||||
|
|
||||||
msg := map[string]interface{}{
|
|
||||||
"msgtype": "markdown",
|
|
||||||
"markdown": map[string]string{
|
|
||||||
"title": title,
|
|
||||||
"text": "### Warning " + title + "\n" + text,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
jsonData, err := json.Marshal(msg)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("JSON 编码失败:", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
client := &http.Client{Timeout: 5 * time.Second}
|
|
||||||
resp, err := client.Post(dingtalkWebhook, "application/json", bytes.NewBuffer(jsonData))
|
|
||||||
if err != nil {
|
|
||||||
log.Println("发送钉钉消息失败:", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer resp.Body.Close()
|
|
||||||
|
|
||||||
body, _ := io.ReadAll(resp.Body)
|
|
||||||
var result map[string]interface{}
|
|
||||||
if err := json.Unmarshal(body, &result); err != nil {
|
|
||||||
log.Println("解析钉钉响应失败:", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if result["errcode"].(float64) != 0 {
|
|
||||||
log.Println("钉钉返回错误:", result["errmsg"])
|
|
||||||
} else {
|
|
||||||
log.Println("钉钉告警已发送成功")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func warning_ding(ipStr string, position string) {
|
|
||||||
WarningListLock.Lock()
|
|
||||||
defer WarningListLock.Unlock()
|
|
||||||
|
|
||||||
WarningList[ipStr] = position
|
|
||||||
|
|
||||||
if len(WarningList) == 1000 {
|
|
||||||
var content string
|
|
||||||
for ip, pos := range WarningList {
|
|
||||||
line := "- " + ip + ", " + pos + "\n"
|
|
||||||
content += line
|
|
||||||
log.Println(ip, pos)
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Println("警告列表已满,发送钉钉告警并清空 map")
|
|
||||||
|
|
||||||
//go sendDingTalkMessage("⚠️ 告警列表达到上限"+local_ipv4_addr, content)
|
|
||||||
|
|
||||||
// 清空 map
|
|
||||||
WarningList = make(map[string]string)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user