mirror of
https://github.com/bettercap/bettercap
synced 2025-07-05 20:42:09 -07:00
118 lines
2 KiB
Go
118 lines
2 KiB
Go
package network
|
|
|
|
import (
|
|
"encoding/json"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/evilsocket/islazy/data"
|
|
)
|
|
|
|
type HIDDevNewCallback func(dev *HIDDevice)
|
|
type HIDDevLostCallback func(dev *HIDDevice)
|
|
|
|
type HID struct {
|
|
sync.RWMutex
|
|
aliases *data.UnsortedKV
|
|
devices map[string]*HIDDevice
|
|
newCb HIDDevNewCallback
|
|
lostCb HIDDevLostCallback
|
|
}
|
|
|
|
type hidJSON struct {
|
|
Devices []*HIDDevice `json:"devices"`
|
|
}
|
|
|
|
func NewHID(aliases *data.UnsortedKV, newcb HIDDevNewCallback, lostcb HIDDevLostCallback) *HID {
|
|
return &HID{
|
|
devices: make(map[string]*HIDDevice),
|
|
aliases: aliases,
|
|
newCb: newcb,
|
|
lostCb: lostcb,
|
|
}
|
|
}
|
|
|
|
func (h *HID) MarshalJSON() ([]byte, error) {
|
|
doc := hidJSON{
|
|
Devices: make([]*HIDDevice, 0),
|
|
}
|
|
|
|
for _, dev := range h.devices {
|
|
doc.Devices = append(doc.Devices, dev)
|
|
}
|
|
|
|
return json.Marshal(doc)
|
|
}
|
|
|
|
func (b *HID) Get(id string) (dev *HIDDevice, found bool) {
|
|
b.RLock()
|
|
defer b.RUnlock()
|
|
dev, found = b.devices[id]
|
|
return
|
|
}
|
|
|
|
func (b *HID) AddIfNew(address []byte, channel int, payload []byte) (bool, *HIDDevice) {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
|
|
id := HIDAddress(address)
|
|
alias := b.aliases.GetOr(id, "")
|
|
|
|
if dev, found := b.devices[id]; found {
|
|
dev.LastSeen = time.Now()
|
|
dev.AddChannel(channel)
|
|
dev.AddPayload(payload)
|
|
if alias != "" {
|
|
dev.Alias = alias
|
|
}
|
|
return false, dev
|
|
}
|
|
|
|
newDev := NewHIDDevice(address, channel, payload)
|
|
newDev.Alias = alias
|
|
b.devices[id] = newDev
|
|
|
|
if b.newCb != nil {
|
|
b.newCb(newDev)
|
|
}
|
|
|
|
return true, newDev
|
|
}
|
|
|
|
func (b *HID) Remove(id string) {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
|
|
if dev, found := b.devices[id]; found {
|
|
delete(b.devices, id)
|
|
if b.lostCb != nil {
|
|
b.lostCb(dev)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (b *HID) Devices() (devices []*HIDDevice) {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
|
|
devices = make([]*HIDDevice, 0)
|
|
for _, dev := range b.devices {
|
|
devices = append(devices, dev)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (b *HID) EachDevice(cb func(mac string, d *HIDDevice)) {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
|
|
for m, dev := range b.devices {
|
|
cb(m, dev)
|
|
}
|
|
}
|
|
|
|
func (b *HID) Clear() {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
b.devices = make(map[string]*HIDDevice)
|
|
}
|