fix: 验收修正
This commit is contained in:
@@ -1,114 +1,146 @@
|
||||
package ingest
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gosnmp/gosnmp"
|
||||
)
|
||||
|
||||
type timeWindow struct {
|
||||
Days []int `json:"days"`
|
||||
Start string `json:"start"`
|
||||
End string `json:"end"`
|
||||
}
|
||||
|
||||
func ipMatchesCIDR(ip net.IP, cidr string) bool {
|
||||
cidr = strings.TrimSpace(cidr)
|
||||
if cidr == "" {
|
||||
return false
|
||||
}
|
||||
if !strings.Contains(cidr, "/") {
|
||||
p := net.ParseIP(cidr)
|
||||
return p != nil && p.Equal(ip)
|
||||
}
|
||||
_, network, err := net.ParseCIDR(cidr)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
return network.Contains(ip)
|
||||
}
|
||||
|
||||
func inTimeWindows(now time.Time, jsonStr string) bool {
|
||||
s := strings.TrimSpace(jsonStr)
|
||||
if s == "" || s == "null" {
|
||||
return true
|
||||
}
|
||||
var windows []timeWindow
|
||||
if err := json.Unmarshal([]byte(s), &windows); err != nil || len(windows) == 0 {
|
||||
return false
|
||||
}
|
||||
tod := now.Hour()*60 + now.Minute()
|
||||
wd := int(now.Weekday())
|
||||
for _, w := range windows {
|
||||
if len(w.Days) > 0 {
|
||||
ok := false
|
||||
for _, d := range w.Days {
|
||||
if d == wd {
|
||||
ok = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
}
|
||||
start := parseHHMM(w.Start)
|
||||
end := parseHHMM(w.End)
|
||||
if start < 0 || end < 0 {
|
||||
continue
|
||||
}
|
||||
if start <= end {
|
||||
if tod >= start && tod <= end {
|
||||
return true
|
||||
}
|
||||
} else {
|
||||
if tod >= start || tod <= end {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func parseHHMM(s string) int {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
return -1
|
||||
}
|
||||
parts := strings.Split(s, ":")
|
||||
if len(parts) != 2 {
|
||||
return -1
|
||||
}
|
||||
h, err1 := strconv.Atoi(parts[0])
|
||||
m, err2 := strconv.Atoi(parts[1])
|
||||
if err1 != nil || err2 != nil || h < 0 || h > 23 || m < 0 || m > 59 {
|
||||
return -1
|
||||
}
|
||||
return h*60 + m
|
||||
}
|
||||
|
||||
func varbindFingerprint(pkt *gosnmp.SnmpPacket) string {
|
||||
var b strings.Builder
|
||||
for _, v := range pkt.Variables {
|
||||
b.WriteString(v.Name)
|
||||
b.WriteByte('=')
|
||||
b.WriteString(fmtVarbindValue(v))
|
||||
b.WriteByte(';')
|
||||
}
|
||||
return b.String()
|
||||
}
|
||||
|
||||
func fmtVarbindValue(v gosnmp.SnmpPDU) string {
|
||||
switch v.Type {
|
||||
case gosnmp.OctetString:
|
||||
if bb, ok := v.Value.([]byte); ok {
|
||||
return string(bb)
|
||||
}
|
||||
}
|
||||
return fmt.Sprintf("%v", v.Value)
|
||||
}
|
||||
package ingest
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gosnmp/gosnmp"
|
||||
)
|
||||
|
||||
type timeWindow struct {
|
||||
Days []int `json:"days"`
|
||||
Start string `json:"start"`
|
||||
End string `json:"end"`
|
||||
}
|
||||
|
||||
// ValidateTimeWindowsJSON 校验 Trap 屏蔽规则的生效时间窗。
|
||||
// 空字符串表示始终生效；非空时必须是至少包含一个有效时间段的 JSON 数组。
|
||||
func ValidateTimeWindowsJSON(jsonStr string) error {
|
||||
s := strings.TrimSpace(jsonStr)
|
||||
if s == "" {
|
||||
return nil
|
||||
}
|
||||
var windows []timeWindow
|
||||
if err := json.Unmarshal([]byte(s), &windows); err != nil {
|
||||
return fmt.Errorf("时间窗必须是 JSON 数组：%w", err)
|
||||
}
|
||||
if len(windows) == 0 {
|
||||
return fmt.Errorf("时间窗不能为空数组；不限制时间时请清空该字段")
|
||||
}
|
||||
for i, window := range windows {
|
||||
if parseHHMM(window.Start) < 0 || parseHHMM(window.End) < 0 {
|
||||
return fmt.Errorf("第 %d 个时间窗的开始或结束时间无效，应使用 HH:MM 格式", i+1)
|
||||
}
|
||||
seenDays := make(map[int]struct{}, len(window.Days))
|
||||
for _, day := range window.Days {
|
||||
if day < 0 || day > 6 {
|
||||
return fmt.Errorf("第 %d 个时间窗的星期值无效，应为 0 到 6", i+1)
|
||||
}
|
||||
if _, exists := seenDays[day]; exists {
|
||||
return fmt.Errorf("第 %d 个时间窗包含重复的星期值 %d", i+1, day)
|
||||
}
|
||||
seenDays[day] = struct{}{}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func ipMatchesCIDR(ip net.IP, cidr string) bool {
|
||||
cidr = strings.TrimSpace(cidr)
|
||||
if cidr == "" {
|
||||
return false
|
||||
}
|
||||
if !strings.Contains(cidr, "/") {
|
||||
p := net.ParseIP(cidr)
|
||||
return p != nil && p.Equal(ip)
|
||||
}
|
||||
_, network, err := net.ParseCIDR(cidr)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
return network.Contains(ip)
|
||||
}
|
||||
|
||||
func inTimeWindows(now time.Time, jsonStr string) bool {
|
||||
s := strings.TrimSpace(jsonStr)
|
||||
if s == "" {
|
||||
return true
|
||||
}
|
||||
var windows []timeWindow
|
||||
if err := json.Unmarshal([]byte(s), &windows); err != nil || len(windows) == 0 {
|
||||
return false
|
||||
}
|
||||
tod := now.Hour()*60 + now.Minute()
|
||||
wd := int(now.Weekday())
|
||||
for _, w := range windows {
|
||||
if len(w.Days) > 0 {
|
||||
ok := false
|
||||
for _, d := range w.Days {
|
||||
if d == wd {
|
||||
ok = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
}
|
||||
start := parseHHMM(w.Start)
|
||||
end := parseHHMM(w.End)
|
||||
if start < 0 || end < 0 {
|
||||
continue
|
||||
}
|
||||
if start <= end {
|
||||
if tod >= start && tod <= end {
|
||||
return true
|
||||
}
|
||||
} else {
|
||||
if tod >= start || tod <= end {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func parseHHMM(s string) int {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
return -1
|
||||
}
|
||||
parts := strings.Split(s, ":")
|
||||
if len(parts) != 2 {
|
||||
return -1
|
||||
}
|
||||
h, err1 := strconv.Atoi(parts[0])
|
||||
m, err2 := strconv.Atoi(parts[1])
|
||||
if err1 != nil || err2 != nil || h < 0 || h > 23 || m < 0 || m > 59 {
|
||||
return -1
|
||||
}
|
||||
return h*60 + m
|
||||
}
|
||||
|
||||
func varbindFingerprint(pkt *gosnmp.SnmpPacket) string {
|
||||
var b strings.Builder
|
||||
for _, v := range pkt.Variables {
|
||||
b.WriteString(v.Name)
|
||||
b.WriteByte('=')
|
||||
b.WriteString(fmtVarbindValue(v))
|
||||
b.WriteByte(';')
|
||||
}
|
||||
return b.String()
|
||||
}
|
||||
|
||||
func fmtVarbindValue(v gosnmp.SnmpPDU) string {
|
||||
switch v.Type {
|
||||
case gosnmp.OctetString:
|
||||
if bb, ok := v.Value.([]byte); ok {
|
||||
return string(bb)
|
||||
}
|
||||
}
|
||||
return fmt.Sprintf("%v", v.Value)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user