mirror of https://github.com/ehang-io/nps
31 lines
496 B
Go
31 lines
496 B
Go
package tool
|
|
|
|
import (
|
|
"github.com/cnlh/nps/lib/common"
|
|
"github.com/cnlh/nps/vender/github.com/astaxie/beego"
|
|
)
|
|
|
|
var ports []int
|
|
|
|
func InitAllowPort() {
|
|
p := beego.AppConfig.String("allow_ports")
|
|
ports = common.GetPorts(p)
|
|
}
|
|
|
|
func TestServerPort(p int, m string) (b bool) {
|
|
if p > 65535 || p < 0 {
|
|
return false
|
|
}
|
|
if len(ports) != 0 {
|
|
if !common.InIntArr(ports, p) {
|
|
return false
|
|
}
|
|
}
|
|
if m == "udp" {
|
|
b = common.TestUdpPort(p)
|
|
} else {
|
|
b = common.TestTcpPort(p)
|
|
}
|
|
return
|
|
}
|