NextGB, web demo powerd by vue

pkg/service/inbound.go

@@ -8,21 +8,13 @@ import (
 
 	"github.com/emiago/sipgo/sip"
 	"github.com/ossrs/go-oryx-lib/logger"
+	"github.com/ossrs/srs-sip/pkg/models"
 	"github.com/ossrs/srs-sip/pkg/service/stack"
 	"golang.org/x/net/html/charset"
 )
 
 const GB28181_ID_LENGTH = 20
 
-type VideoChannelStatus struct {
-	ID        string
-	ParentID  string
-	MediaHost string
-	MediaPort int
-	Ssrc      string
-	Status    string
-}
-
 func (s *UAS) onRegister(req *sip.Request, tx sip.ServerTransaction) {
 	id := req.From().Address.User
 	if len(id) != GB28181_ID_LENGTH {
@@ -30,6 +22,27 @@ func (s *UAS) onRegister(req *sip.Request, tx sip.ServerTransaction) {
 		return
 	}
 
+	if s.conf.GB28181.Auth.Enable {
+		// Check if Authorization header exists
+		authHeader := req.GetHeaders("Authorization")
+
+		// If no Authorization header, send 401 response to request authentication
+		if len(authHeader) == 0 {
+			nonce := GenerateNonce()
+			resp := stack.NewUnauthorizedResponse(req, http.StatusUnauthorized, "Unauthorized", nonce, s.conf.GB28181.Realm)
+			_ = tx.Respond(resp)
+			return
+		}
+
+		// Validate Authorization
+		authInfo := ParseAuthorization(authHeader[0].Value())
+		if !ValidateAuth(authInfo, s.conf.GB28181.Auth.Password) {
+			logger.Ef(s.ctx, "%s auth failed, source: %s", id, req.Source())
+			s.respondRegister(req, http.StatusForbidden, "Auth Failed", tx)
+			return
+		}
+	}
+
 	isUnregister := false
 	if exps := req.GetHeaders("Expires"); len(exps) > 0 {
 		exp := exps[0]
@@ -88,19 +101,7 @@ func (s *UAS) onMessage(req *sip.Request, tx sip.ServerTransaction) {
 
 	//logger.Tf(s.ctx, "Received MESSAGE: %s", req.String())
 
-	temp := &struct {
-		XMLName      xml.Name
-		CmdType      string
-		SN           int // 请求序列号，一般用于对应 request 和 response
-		DeviceID     string
-		DeviceName   string
-		Manufacturer string
-		Model        string
-		Channel      string
-		DeviceList   []ChannelInfo `xml:"DeviceList>Item"`
-		// RecordList   []*Record     `xml:"RecordList>Item"`
-		// SumNum       int
-	}{}
+	temp := &models.XmlMessageInfo{}
 	decoder := xml.NewDecoder(bytes.NewReader([]byte(req.Body())))
 	decoder.CharsetReader = charset.NewReaderLabel
 	if err := decoder.Decode(temp); err != nil {
@@ -122,6 +123,14 @@ func (s *UAS) onMessage(req *sip.Request, tx sip.ServerTransaction) {
 		//go s.AutoInvite(temp.DeviceID, temp.DeviceList...)
 	case "Alarm":
 		logger.T(s.ctx, "Alarm")
+	case "RecordInfo":
+		logger.T(s.ctx, "RecordInfo")
+		// 从 recordQueryResults 中获取对应通道的结果通道
+		if ch, ok := s.recordQueryResults.Load(temp.DeviceID); ok {
+			// 发送查询结果
+			resultChan := ch.(chan *models.XmlMessageInfo)
+			resultChan <- temp
+		}
 	default:
 		logger.Wf(s.ctx, "Not supported CmdType: %s", temp.CmdType)
 		response := sip.NewResponseFromRequest(req, http.StatusBadRequest, "", nil)
@@ -135,19 +144,3 @@ func (s *UAS) onNotify(req *sip.Request, tx sip.ServerTransaction) {
 	logger.T(s.ctx, "Received NOTIFY request")
 	tx.Respond(sip.NewResponseFromRequest(req, http.StatusOK, "OK", nil))
 }
-
-func (s *UAS) AddVideoChannelStatue(channelID string, status VideoChannelStatus) {
-	s.channelsStatue.Store(channelID, status)
-}
-
-func (s *UAS) GetVideoChannelStatue(channelID string) (VideoChannelStatus, bool) {
-	v, ok := s.channelsStatue.Load(channelID)
-	if !ok {
-		return VideoChannelStatus{}, false
-	}
-	return v.(VideoChannelStatus), true
-}
-
-func (s *UAS) RemoveVideoChannelStatue(channelID string) {
-	s.channelsStatue.Delete(channelID)
-}