new: new can module for CAN-bus

network/can.go

@@ -0,0 +1,110 @@
+package network
+
+import (
+	"encoding/json"
+	"sync"
+	"time"
+
+	"github.com/evilsocket/islazy/data"
+)
+
+type CANDevNewCallback func(dev *CANDevice)
+type CANDevLostCallback func(dev *CANDevice)
+
+type CAN struct {
+	sync.RWMutex
+	devices map[string]*CANDevice
+	newCb   CANDevNewCallback
+	lostCb  CANDevLostCallback
+}
+
+type canJSON struct {
+	Devices []*CANDevice `json:"devices"`
+}
+
+func NewCAN(aliases *data.UnsortedKV, newcb CANDevNewCallback, lostcb CANDevLostCallback) *CAN {
+	return &CAN{
+		devices: make(map[string]*CANDevice),
+		newCb:   newcb,
+		lostCb:  lostcb,
+	}
+}
+
+func (h *CAN) MarshalJSON() ([]byte, error) {
+	doc := canJSON{
+		Devices: make([]*CANDevice, 0),
+	}
+
+	for _, dev := range h.devices {
+		doc.Devices = append(doc.Devices, dev)
+	}
+
+	return json.Marshal(doc)
+}
+
+func (b *CAN) Get(id string) (dev *CANDevice, found bool) {
+	b.RLock()
+	defer b.RUnlock()
+	dev, found = b.devices[id]
+	return
+}
+
+func (b *CAN) AddIfNew(name string, description string, payload []byte) (bool, *CANDevice) {
+	b.Lock()
+	defer b.Unlock()
+
+	id := name
+
+	if dev, found := b.devices[id]; found {
+		dev.LastSeen = time.Now()
+		dev.AddPayload(payload)
+		return false, dev
+	}
+
+	newDev := NewCANDevice(name, description, payload)
+	b.devices[id] = newDev
+
+	if b.newCb != nil {
+		b.newCb(newDev)
+	}
+
+	return true, newDev
+}
+
+func (b *CAN) 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 *CAN) Devices() (devices []*CANDevice) {
+	b.Lock()
+	defer b.Unlock()
+
+	devices = make([]*CANDevice, 0)
+	for _, dev := range b.devices {
+		devices = append(devices, dev)
+	}
+	return
+}
+
+func (b *CAN) EachDevice(cb func(mac string, d *CANDevice)) {
+	b.Lock()
+	defer b.Unlock()
+
+	for m, dev := range b.devices {
+		cb(m, dev)
+	}
+}
+
+func (b *CAN) Clear() {
+	b.Lock()
+	defer b.Unlock()
+	b.devices = make(map[string]*CANDevice)
+}

network/can_device.go

@@ -0,0 +1,57 @@
+package network
+
+import (
+	"encoding/json"
+	"sync"
+	"time"
+)
+
+type CANDevice struct {
+	sync.Mutex
+	LastSeen    time.Time
+	Name        string
+	Description string
+	Read        uint64
+}
+
+type canDeviceJSON struct {
+	LastSeen    time.Time `json:"last_seen"`
+	Name        string    `json:"name"`
+	Description string    `json:"description"`
+	Read        uint64    `json:"description"`
+}
+
+func NewCANDevice(name string, description string, payload []byte) *CANDevice {
+	dev := &CANDevice{
+		LastSeen:    time.Now(),
+		Name:        name,
+		Description: description,
+		Read:        uint64(len(payload)),
+	}
+
+	return dev
+}
+
+func (dev *CANDevice) MarshalJSON() ([]byte, error) {
+	dev.Lock()
+	defer dev.Unlock()
+
+	doc := canDeviceJSON{
+		LastSeen:    dev.LastSeen,
+		Name:        dev.Name,
+		Description: dev.Description,
+		Read:        dev.Read,
+	}
+
+	return json.Marshal(doc)
+}
+
+func (dev *CANDevice) AddPayload(payload []byte) {
+	dev.Lock()
+	defer dev.Unlock()
+
+	sz := len(payload)
+	if payload != nil && sz > 0 {
+		dev.Read += uint64(sz)
+	}
+}