7x31/anylink
1
0
Fork 0
mirror of https://github.com/bjdgyc/anylink.git synced 2025-08-08 22:08:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

兼容历史单端口配置

Browse Source
This commit is contained in:
huweishan
2024-04-09 10:29:54 +08:00
parent 4f56ea49c3
commit 24e30509e4
2 changed files with 25 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
server/dbdata/group.go
View File

@@ -5,10 +5,9 @@ import (
"fmt"
"net"
"regexp"
"strings"
"strconv"
"strings"
"time"
"reflect"
"github.com/bjdgyc/anylink/base"
"golang.org/x/text/language"
@@ -26,12 +25,12 @@ const DsMaxLen = 20000
type GroupLinkAcl struct {
// 自上而下匹配 默认 allow * *
Action string `json:"action"` // allow、deny
Val string `json:"val"`
Port interface{} `json:"port"`
Ports map[uint16]int8 `json:"ports"`
IpNet *net.IPNet `json:"ip_net"`
Note string `json:"note"`
Action string `json:"action"` // allow、deny
Val string `json:"val"`
Port interface{} `json:"port"` //兼容单端口历史数据类型uint16
Ports map[uint16]int8 `json:"ports"`
IpNet *net.IPNet `json:"ip_net"`
Note string `json:"note"`
}
type ValData struct {
@@ -45,13 +44,6 @@ type GroupNameId struct {
Name string `json:"name"`
}
type PortData struct {
PortFrom uint16 `json:"port_from"`
PortTo uint16 `json:"port_to"`
}
// type Group struct {
// Id int `json:"id" xorm:"pk autoincr not null"`
// Name string `json:"name" xorm:"varchar(60) not null unique"`
@@ -172,14 +164,14 @@ func SetGroup(g *Group) error {
}
v.IpNet = ipNet
port:="";
//base.Debug("v.port:",v.Port,v.Ports,reflect.TypeOf(v.Port).Name())
switch v := v.Port.(type) {
port := ""
switch vp := v.Port.(type) {
case float64:
port = strconv.Itoa(int(v))
port = strconv.Itoa(int(vp))
case string:
port = v
port = vp
}
if regexp.MustCompile(`^\d{1,5}(-\d{1,5})?(,\d{1,5}(-\d{1,5})?)*$`).MatchString(port) {
ports := map[uint16]int8{}
for _, p := range strings.Split(port, ",") {
@@ -187,23 +179,23 @@ func SetGroup(g *Group) error {
continue
}
if regexp.MustCompile(`^\d{1,5}-\d{1,5}$`).MatchString(p) {
rp := strings.Split(p, "-");
portfrom, err := strconv.Atoi(rp[0])
rp := strings.Split(p, "-")
portfrom, err := strconv.Atoi(rp[0])
if err != nil {
return errors.New("端口:"+rp[0]+" 格式错误, " + err.Error())
return errors.New("端口:" + rp[0] + " 格式错误, " + err.Error())
}
portto, err := strconv.Atoi(rp[1])
if err != nil {
return errors.New("端口:"+rp[1]+" 格式错误, " + err.Error())
return errors.New("端口:" + rp[1] + " 格式错误, " + err.Error())
}
for i := portfrom; i <= portto; i++ {
ports[uint16(i)] = 1
}
}
} else {
port, err := strconv.Atoi(p)
if err != nil {
return errors.New("端口:"+p+" 格式错误, " + err.Error())
return errors.New("端口:" + p + " 格式错误, " + err.Error())
}
ports[uint16(port)] = 1
}
@@ -211,7 +203,7 @@ func SetGroup(g *Group) error {
v.Ports = ports
linkAcl = append(linkAcl, v)
} else {
return errors.New("端口: "+port+" 格式错误,请用逗号分隔的端口,比如: 22,80,443 连续端口用-,比如:1234-5678")
return errors.New("端口: " + port + " 格式错误,请用逗号分隔的端口,比如: 22,80,443 连续端口用-,比如:1234-5678")
}
}