PR comments, except one

Author: alexlivekit

go.mod

@@ -142,3 +142,5 @@ require (
 	google.golang.org/grpc v1.76.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 )
+
+replace github.com/livekit/psrpc => ../psrpc

pkg/sip/inbound.go

@@ -30,7 +30,7 @@ import (
 
 	msdk "github.com/livekit/media-sdk"
 	"github.com/livekit/protocol/rpc"
-	uuid "github.com/satori/go.uuid"
+	"github.com/livekit/protocol/utils"
 
 	"github.com/frostbyte73/core"
 	"github.com/icholy/digest"
@@ -138,7 +138,11 @@ func (s *Server) getCallInfo(id string) *inboundCallInfo {
 }
 
 func (s *Server) getInvite(sipCallID, toTag, fromTag string) *inProgressInvite {
-	key := fmt.Sprintf("%s:%s:%s", sipCallID, toTag, fromTag)
+	key := dialogKey{
+		sipCallID: sipCallID,
+		toTag:     toTag,
+		fromTag:   fromTag,
+	}
 	s.imu.Lock()
 	defer s.imu.Unlock()
 	is, ok := s.inProgressInvites[key]
@@ -325,6 +329,7 @@ func (s *Server) processInvite(req *sip.Request, tx sip.ServerTransaction) (retE
 	log = LoggerWithHeaders(log, cc)
 
 	if err := cc.ValidateInvite(); err != nil {
+		log.Errorw("invalid invite", err)
 		if s.conf.HideInboundPort {
 			cc.Drop()
 		} else {
@@ -335,13 +340,16 @@ func (s *Server) processInvite(req *sip.Request, tx sip.ServerTransaction) (retE
 
 	// Establish ID
 	fromTag, _ := req.From().Params.Get("tag") // always exists, via ValidateInvite() check
-	toTag, ok := req.To().Params.Get("tag")    // To() always exists, via ValidateInvite() check
+	toParams := req.To().Params                // To() always exists, via ValidateInvite() check
+	if toParams == nil {
+		toParams = sip.NewParams()
+		req.To().Params = toParams
+	}
+	toTag, ok := toParams.Get("tag")
 	if !ok {
 		// No to-tag on the invite means we need to generate one per RFC 3261 section 12.
 		// Generate a new to-tag early, to make sure both INVITES have the same ID.
-		uuid, _ := uuid.NewV4() // Same as NewResponseFromRequest in sipgo
-		toTag = uuid.String()
-		req.To().Params.Add("tag", toTag)
+		toParams.Add("tag", utils.NewGuid(""))
 	}
 	inviteProgress := s.getInvite(sipCallID, toTag, fromTag)
 	callID := inviteProgress.lkCallID

pkg/sip/server.go

@@ -122,6 +122,12 @@ type Handler interface {
 	OnSessionEnd(ctx context.Context, callIdentifier *CallIdentifier, callInfo *livekit.SIPCallInfo, reason string)
 }
 
+type dialogKey struct {
+	sipCallID string
+	toTag     string
+	fromTag   string
+}
+
 type Server struct {
 	log          logger.Logger
 	mon          *stats.Monitor
@@ -132,7 +138,7 @@ type Server struct {
 	sipUnhandled RequestHandler
 
 	imu               sync.Mutex
-	inProgressInvites map[string]*inProgressInvite
+	inProgressInvites map[dialogKey]*inProgressInvite
 
 	closing     core.Fuse
 	cmu         sync.RWMutex
@@ -167,7 +173,7 @@ func NewServer(region string, conf *config.Config, log logger.Logger, mon *stats
 		region:            region,
 		mon:               mon,
 		getIOClient:       getIOClient,
-		inProgressInvites: make(map[string]*inProgressInvite),
+		inProgressInvites: make(map[dialogKey]*inProgressInvite),
 		activeCalls:       make(map[RemoteTag]*inboundCall),
 		byLocal:           make(map[LocalTag]*inboundCall),
 	}