Merge pull request #255 from sipcapture/ping

Ping

Author: lmangani

decoder/decoder.go

@@ -531,3 +531,16 @@ func (d *Decoder) ProcessHEPPacket(data []byte) {
 
 	PacketQueue <- pkt
 }
+
+func (d *Decoder) SendPingHEPPacket() {
+
+	var data = []byte{0x48, 0x45, 0x50, 0x33, 0x3, 0xa}
+	pkt := &Packet{
+		Version: 0,
+		Payload: data,
+	}
+
+	atomic.AddUint64(&d.hepCount, 1)
+
+	PacketQueue <- pkt
+}

main.go

@@ -12,7 +12,7 @@ import (
 	"github.com/sipcapture/heplify/sniffer"
 )
 
-const version = "heplify 1.65.2"
+const version = "heplify 1.65.3"
 
 func createFlags() {
 
@@ -139,6 +139,11 @@ func main() {
 
 		wg.Add(1)
 		go func() {
+
+			if config.Cfg.HepNodePW != "" {
+				capture.SendPing()
+			}
+
 			err = capture.Run()
 			checkCritErr(err)
 			wg.Done()

sniffer/sniffer.go

@@ -51,11 +51,13 @@ type SnifferSetup struct {
 type MainWorker struct {
 	publisher *publish.Publisher
 	decoder   *decoder.Decoder
+	ping      *decoder.Decoder
 }
 
 type Worker interface {
 	OnPacket(data []byte, ci *gopacket.CaptureInfo)
 	OnHEPPacket(data []byte)
+	SendPingHEPPacket()
 }
 
 type stats struct {
@@ -95,6 +97,10 @@ func (mw *MainWorker) OnHEPPacket(data []byte) {
 	mw.decoder.ProcessHEPPacket(data)
 }
 
+func (mw *MainWorker) SendPingHEPPacket() {
+	mw.decoder.SendPingHEPPacket()
+}
+
 func (sniffer *SnifferSetup) setFromConfig() error {
 	var err error
 
@@ -315,6 +321,11 @@ func New(cfgMain *config.Config) (*SnifferSetup, error) {
 	return sniffer, nil
 }
 
+func (sniffer *SnifferSetup) SendPing() error {
+	sniffer.worker.SendPingHEPPacket()
+	return nil
+}
+
 func (sniffer *SnifferSetup) Run() error {
 	var (
 		loopCount   = 1