mirror of
https://github.com/Control-D-Inc/ctrld.git
synced 2026-02-03 22:18:39 +00:00
0f3e8c7ada
So ctrld can record the raw/original client IP instead of looking up from MAC to IP, which may not the right choice in some network setup like using wireguard/vpn on Merlin router.
50 lines
787 B
Go
50 lines
787 B
Go
package clientinfo
|
|
|
|
import "sync"
|
|
|
|
type arpDiscover struct {
|
|
mac sync.Map // ip => mac
|
|
ip sync.Map // mac => ip
|
|
}
|
|
|
|
func (a *arpDiscover) refresh() error {
|
|
a.scan()
|
|
return nil
|
|
}
|
|
|
|
func (a *arpDiscover) LookupIP(mac string) string {
|
|
val, ok := a.ip.Load(mac)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
return val.(string)
|
|
}
|
|
|
|
func (a *arpDiscover) LookupMac(ip string) string {
|
|
val, ok := a.mac.Load(ip)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
return val.(string)
|
|
}
|
|
|
|
func (a *arpDiscover) String() string {
|
|
return "arp"
|
|
}
|
|
|
|
func (a *arpDiscover) List() []string {
|
|
if a == nil {
|
|
return nil
|
|
}
|
|
var ips []string
|
|
a.ip.Range(func(key, value any) bool {
|
|
ips = append(ips, value.(string))
|
|
return true
|
|
})
|
|
a.mac.Range(func(key, value any) bool {
|
|
ips = append(ips, key.(string))
|
|
return true
|
|
})
|
|
return ips
|
|
}
|