64 lines
1.1 KiB
Go
64 lines
1.1 KiB
Go
|
package internal
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"hash/crc32"
|
||
|
|
"net"
|
||
|
|
"net/netip"
|
||
|
|
"sync/atomic"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
)
|
||
|
|
|
||
|
|
var messageID = atomic.Uint64{}
|
||
|
|
|
||
|
|
type Message struct {
|
||
|
|
ID uint64
|
||
|
|
Data []byte
|
||
|
|
Len int
|
||
|
|
|
||
|
|
Src *net.UDPAddr
|
||
|
|
Dst *net.UDPAddr
|
||
|
|
|
||
|
|
Checksum string
|
||
|
|
Timestamp time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMessage(data []byte, src net.Addr, dst net.Addr) (msg Message, err error) {
|
||
|
|
msg.SetData(data)
|
||
|
|
|
||
|
|
var addr netip.AddrPort
|
||
|
|
addr, err = netip.ParseAddrPort(src.String())
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
msg.Src = net.UDPAddrFromAddrPort(addr)
|
||
|
|
|
||
|
|
addr, err = netip.ParseAddrPort(dst.String())
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
msg.Dst = net.UDPAddrFromAddrPort(addr)
|
||
|
|
msg.Timestamp = time.Now()
|
||
|
|
msg.ID = messageID.Add(1)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m Message) Log(event *zerolog.Event) *zerolog.Event {
|
||
|
|
return event.
|
||
|
|
Uint64("id", m.ID).
|
||
|
|
Int("length", m.Len).
|
||
|
|
Str("src", m.Src.String()).
|
||
|
|
Str("dst", m.Dst.String()).
|
||
|
|
Str("checksum", m.Checksum)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Message) SetData(data []byte) {
|
||
|
|
m.Data = make([]byte, len(data))
|
||
|
|
copy(m.Data, data)
|
||
|
|
m.Len = len(m.Data)
|
||
|
|
m.Checksum = fmt.Sprintf("%04X", crc32.ChecksumIEEE(m.Data))
|
||
|
|
}
|