improved SIP event handling

mushorg · Sep 28, 2023 · 5e42e77 · 5e42e77
1 parent b87d8f2
commit 5e42e77
Showing 1 changed file with 60 additions and 36 deletions.
diff --git a/protocols/sip.go b/protocols/sip.go
@@ -14,56 +14,80 @@ import (
 
 const maxBufferSize = 1024
 
+type parsedSIP struct {
+ Direction string
+ Payload []byte
+ Message sip.Message
+}
+
+type sipServer struct {
+ events []parsedSIP
+}
+
 // HandleSIP takes a net.Conn and does basic SIP communication
 func HandleSIP(ctx context.Context, conn net.Conn, logger Logger, h Honeypot) error {
+ server := sipServer{
+ events: []parsedSIP{},
+ }
  defer func() {
+ md, err := h.MetadataByConnection(conn)
+ if err != nil {
+ logger.Error("failed to fetch meta data", zap.String("protocol", "sip"), zap.Error(err))
+ }
+ if err := h.Produce("sip", conn, md, firstOrEmpty[parsedSIP](server.events).Payload, server.events); err != nil {
+ logger.Error("failed to produce message", zap.String("protocol", "sip"), zap.Error(err))
+ }
  if err := conn.Close(); err != nil {
  logger.Error(fmt.Errorf("failed to close SIP connection: %w", err).Error())
  }
  }()
 
  buffer := make([]byte, maxBufferSize)
-
- _, err := conn.Read(buffer)
- if err != nil {
- return err
- }
-
  l := log.NewDefaultLogrusLogger()
  pp := parser.NewPacketParser(l)
- msg, err := pp.ParseMessage(buffer)
- if err != nil {
- return err
- }
 
- md, err := h.MetadataByConnection(conn)
- if err != nil {
- return err
- }
- if err := h.Produce("sip", conn, md, buffer, msg); err != nil {
- logger.Error("failed to produce message", zap.String("protocol", "sip"), zap.Error(err))
- }
+ for {
+ n, err := conn.Read(buffer)
+ if err != nil {
+ return err
+ }
+
+ msg, err := pp.ParseMessage(buffer[:n])
+ if err != nil {
+ return err
+ }
+
+ server.events = append(server.events, parsedSIP{
+ Direction: "read",
+ Message: msg,
+ Payload: buffer[:n],
+ })
 
- switch msg := msg.(type) {
- case sip.Request:
- switch msg.Method() {
- case sip.REGISTER:
- logger.Info("handling SIP register")
- case sip.INVITE:
- logger.Info("handling SIP invite")
- case sip.OPTIONS:
- logger.Info("handling SIP options")
- resp := sip.NewResponseFromRequest(
- msg.MessageID(),
- msg,
- http.StatusOK,
- "",
- "",
- )
- if _, err := conn.Write([]byte(resp.String())); err != nil {
- return err
+ switch msg := msg.(type) {
+ case sip.Request:
+ switch msg.Method() {
+ case sip.REGISTER:
+ logger.Info("handling SIP register")
+ case sip.INVITE:
+ logger.Info("handling SIP invite")
+ case sip.OPTIONS:
+ logger.Info("handling SIP options")
+ resp := sip.NewResponseFromRequest(
+ msg.MessageID(),
+ msg,
+ http.StatusOK,
+ "",
+ "",
+ )
+ server.events = append(server.events, parsedSIP{
+ Direction: "write",
+ Message: resp,
+ Payload: []byte(resp.String()),
+ })
+ if _, err := conn.Write([]byte(resp.String())); err != nil {
+ return err
+ }
  }
  }
  }
- return nil
 }