forked from x/icebergs
98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
package tcp
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/aaa"
|
|
kit "github.com/shylinux/toolkits"
|
|
)
|
|
|
|
func _host_list(m *ice.Message, name string) {
|
|
if ifs, e := net.Interfaces(); m.Assert(e) {
|
|
for _, v := range ifs {
|
|
if name != "" && !strings.Contains(v.Name, name) {
|
|
continue
|
|
}
|
|
if len(v.HardwareAddr.String()) == 0 {
|
|
continue
|
|
}
|
|
|
|
if ips, e := v.Addrs(); m.Assert(e) {
|
|
for _, x := range ips {
|
|
ip := strings.Split(x.String(), "/")
|
|
if strings.Contains(ip[0], ":") || len(ip) == 0 {
|
|
continue
|
|
}
|
|
|
|
m.Push(kit.MDB_INDEX, v.Index)
|
|
m.Push(kit.MDB_NAME, v.Name)
|
|
m.Push(IP, ip[0])
|
|
m.Push(MASK, ip[1])
|
|
m.Push(HARD, v.HardwareAddr.String())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(m.Appendv(IP)) == 0 {
|
|
m.Push(kit.MDB_INDEX, -1)
|
|
m.Push(kit.MDB_NAME, LOCALHOST)
|
|
m.Push(IP, "127.0.0.1")
|
|
m.Push(MASK, "255.0.0.0")
|
|
m.Push(HARD, "")
|
|
}
|
|
}
|
|
|
|
func _islocalhost(m *ice.Message, ip string) (ok bool) {
|
|
if ip == "::1" || strings.HasPrefix(ip, "127.") {
|
|
return true
|
|
}
|
|
if m.Richs(HOST, kit.Keym(aaa.BLACK), ip, nil) != nil {
|
|
return false
|
|
}
|
|
if m.Richs(HOST, kit.Keym(aaa.WHITE), ip, nil) != nil {
|
|
m.Log_AUTH(aaa.WHITE, ip)
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
func IsLocalHost(m *ice.Message, ip string) bool { return _islocalhost(m, ip) }
|
|
|
|
const (
|
|
HOSTPORT = "hostport"
|
|
HOSTNAME = "hostname"
|
|
PROTOCOL = "protocol"
|
|
|
|
LOCALHOST = "localhost"
|
|
|
|
HARD = "hard"
|
|
MASK = "mask"
|
|
IP = "ip"
|
|
)
|
|
const HOST = "host"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
HOST: {Name: HOST, Help: "主机", Value: kit.Data(
|
|
aaa.BLACK, kit.Data(kit.MDB_SHORT, kit.MDB_TEXT),
|
|
aaa.WHITE, kit.Data(kit.MDB_SHORT, kit.MDB_TEXT),
|
|
)},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
HOST: {Name: "host name auto", Help: "主机", Action: map[string]*ice.Action{
|
|
aaa.BLACK: {Name: "black", Help: "黑名单", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Rich(HOST, kit.Keym(aaa.BLACK), kit.Dict(kit.MDB_TEXT, arg[0]))
|
|
}},
|
|
aaa.WHITE: {Name: "white", Help: "白名单", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Rich(HOST, kit.Keym(aaa.WHITE), kit.Dict(kit.MDB_TEXT, arg[0]))
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
_host_list(m, kit.Select("", arg, 0))
|
|
}},
|
|
},
|
|
})
|
|
}
|