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
 }