package controller import ( "net" // log "github.com/sirupsen/logrus" "dev.sum7.eu/genofire/wifictld-analyzer/data" ) func (c *Controller) Handler(addr *net.UDPAddr, msg *data.SocketMSG) (*data.SocketMSG, error) { ignore := false if msg.Types.Is(data.SocketMSGTypeClient) && msg.Client != nil { ignore = c.db.LearnClient(addr.IP, msg.Client) } if !msg.Types.Is(data.SocketMSGTypeRequest) { return nil, nil } msg = &data.SocketMSG{ Types: (data.SocketMSGTypeResponse | data.SocketMSGTypeClient), Client: c.db.GetClient(msg.Client.Addr), } if !ignore { return msg, nil } return nil, nil }