package data import ( "encoding/binary" "fmt" "net" "time" log "github.com/sirupsen/logrus" ) var DEBUG = false // SocketMSGType kind of packages type SocketMSGType uint32 const ( SocketMSGTypeRequest SocketMSGType = (1 << 0) SocketMSGTypeResponse SocketMSGType = (1 << 1) SocketMSGTypeClient SocketMSGType = (1 << 2) SocketMSGTypeStats SocketMSGType = (1 << 3) ) func (a SocketMSGType) Is(b SocketMSGType) bool { if DEBUG { log.Debugf("SocketType: %x & %x = %x -> %b", a, b, (a & b), (a&b) > 0) } return (a & b) > 0 } // SocketMSG package of wifictld format type SocketMSG struct { Types SocketMSGType Client *WifiClient } func NewSocketMSG(obj []byte) (*SocketMSG, error) { msg := &SocketMSG{} if err := msg.Unmarshal(obj); err != nil { return nil, err } return msg, nil } func (msg *SocketMSG) Marshal() ([]byte, error) { obj := make([]byte, MaxDataGramSize) pos := 0 binary.BigEndian.PutUint32(obj[pos:(pos+4)], uint32(msg.Types)) pos += 4 if msg.Types.Is(SocketMSGTypeClient) { obj[pos] = msg.Client.Addr[0] pos++ obj[pos] = msg.Client.Addr[1] pos++ obj[pos] = msg.Client.Addr[2] pos++ obj[pos] = msg.Client.Addr[3] pos++ obj[pos] = msg.Client.Addr[4] pos++ obj[pos] = msg.Client.Addr[5] pos++ binary.BigEndian.PutUint32(obj[pos:(pos+4)], uint32(msg.Client.Time.Unix())) pos += 4 binary.BigEndian.PutUint16(obj[pos:(pos+2)], msg.Client.TryProbe) pos += 2 binary.BigEndian.PutUint16(obj[pos:(pos+2)], msg.Client.TryAuth) pos += 2 if msg.Client.Connected { obj[pos] = byte(1) } pos++ if msg.Client.Authed { obj[pos] = byte(1) } pos++ binary.BigEndian.PutUint16(obj[pos:(pos+2)], msg.Client.FreqHighest) pos += 2 binary.BigEndian.PutUint16(obj[pos:(pos+2)], uint16(msg.Client.SignalLowFreq)) pos += 2 binary.BigEndian.PutUint16(obj[pos:(pos+2)], uint16(msg.Client.SignalHighFreq)) //pos += 2 } return obj, nil } func (msg *SocketMSG) Unmarshal(obj []byte) error { if DEBUG { log.Debugf("hex unmarshal: %x", obj) } pos := 0 msg.Types = SocketMSGType(binary.BigEndian.Uint32(obj[pos:(pos + 4)])) pos += 4 if msg.Types.Is(SocketMSGTypeClient) { msg.Client = &WifiClient{ Addr: net.HardwareAddr(obj[pos:(pos + 6)]), Time: time.Unix(int64(binary.BigEndian.Uint32(obj[(pos+6):(pos+10)])), 0), TryProbe: binary.BigEndian.Uint16(obj[(pos + 10):(pos + 12)]), TryAuth: binary.BigEndian.Uint16(obj[(pos + 12):(pos + 14)]), Connected: (obj[(pos+14)] == 1), Authed: (obj[(pos+15)] == 1), FreqHighest: binary.BigEndian.Uint16(obj[(pos + 16):(pos + 18)]), SignalLowFreq: int16(binary.BigEndian.Uint16(obj[(pos + 18):(pos + 20)])), SignalHighFreq: int16(binary.BigEndian.Uint16(obj[(pos + 20):(pos + 22)])), } //pos += 22 } return nil } func (msg *SocketMSG) String() string { answer := "" if msg.Types.Is(SocketMSGTypeRequest) { answer += "request " } if msg.Types.Is(SocketMSGTypeResponse) { answer += "response " } if msg.Types.Is(SocketMSGTypeClient) { answer += fmt.Sprintf("client(mac=%s freq=%d ssi_l=%d ssi_h=%d tprobe=%d tauth=%d authed=%v time=%s) ", msg.Client.Addr, msg.Client.FreqHighest, msg.Client.SignalLowFreq, msg.Client.SignalHighFreq, msg.Client.TryProbe, msg.Client.TryAuth, msg.Client.Authed, msg.Client.Time.String()) } return answer }